ptrblck:

Are you saving both, the actual model object as well as the state_dict?


Yes, I’m saving both. Since I haven’t changed my model and its dependencies for a while, I’ve tended to load the net directly on GPU (though of course I did make sure to load the .state_dict  to a new model object on GPU this time).