Best way to save a trained model in PyTorch?

I was looking for alternative ways to save a trained model in PyTorch. So far, I have found two alternatives. torch.save() to save a model and torch.load() to load a model. model.state_dict() to save a trained model and model.load_state_dict() to load the saved model. I have come across to this discussion where approach 2 is … Read more