Solved by roaffix in post #5 
              
              
                So, as I understood: 

I save a model this way

# ... some code here
torch.save(model.state_dict(), "{}.pt".format(output_name))
# NOTE: output_name is modelX_name in next steps


Then I load model this way

model1 = ... # your model1. E.g, model1 = CatDogClassifier().to('cuda')
model2 = ... # your…