It is recommended to save and load the state_dict(s) instead of the complete model as explained in the serialization docs 8.
However, I’m not sure, if you are trying to load the model or any other object, as the file names changed between torch.save and torch.load.