How to save and load models in PyTorch?

In PyTorch, saving and loading models can be achieved through the following steps:

1. save the torch model

torch.save(model.state_dict(), 'model.pth')

1. load the model using torch

model = Model()
model.load_state_dict(torch.load('model.pth'))
model.eval()

注意：在加载模型时，一定要确保其结构与保存时一致，否则可能会无法加载成功。