after training the model, I use torch.save to save the model with .pth format.
And in another python file, I try to use torch.load to load the model and do the prediction, but seems like it retrain the model and then do the prediction,
Can someone explains that for me ?