Are you able to load the state_dict without any modifications to the model?
The current error message seems to indicate that torch.load might load the complete model, not a state_dict.
Could this be the case, i.e. did you save the complete model instead of the state_dict?