Solved by RaLo4 in post #2 
              
              
                I am assuming this is your code for testing the model whose parameters are saved in base_model.pth. 
So I am guessing you want to run this: 

With the parameters saved in this: 

But you load your model parameters at the very end and do nothing with it: 

Try putting these two lines: 

After this li…