You can save model via torch.save but there are issues like pickling gpu-located models and if you decide to tinker with some functions or classes model is based on you’ll encounter that this is not possible.
Saving through model.state_dict() is a much more stable approach