ptrblck:

Yeah, you should be a bit careful about using this argument, as it’s ignoring the (valid) error and is thus not used by default.


Thanks, will keep this in mind!



 ptrblck:

or store the state_dict from the underlying model using torch.save(model.module.state_dict(), PATH).


I am not sure what you mean by .module here, do you mean to pass torch.save() as
torch.save(m.module.state_dcit(), PATH)

where m is an instance of my model class?