Solved by Quinn in post #2 
              
              
                Hi, Talita: 
You can save your model by either of the following methods. 
# Method 1
torch.save(model, 'best-model.pt') 
# Method 2
torch.save(model.state_dict(), 'best-model-parameters.pt') # official recommended

The difference between two methods is that the first one saves the whole model which …