You shouldn’t save the net directly, as it would depend on the actual source files and could easily break.
The recommended way it so save the state_dict and create a new model object before loading the state_dict back.
Are you saving both, the actual model object as well as the state_dict?