Try using:
Save:
torch.save(model.state_dict(), PATH)
Load:
model = TheModelClass(*args, **kwargs) 
model.load_state_dict(torch.load(PATH)) 
model.eval()

https://pytorch.org/tutorials/beginner/saving_loading_models.html 1
I had the same issue with loading the entire model. But just loading the state_dict performed very well on future training/evaluation.