Solved by Dwight_Foster in post #4 
              
              
                Ok then you still need to define the model in this file and then do load state dict. The model load function won’t work because you didn’t save the entire model.