来源：https://pytorch.org/tutorials/beginner/saving_loading_models.html

保存/加载整个模型

保存模型

torch.save(model, PATH)

加载模型

# Model class must be defined somewhere
model = torch.load(PATH)
model.eval()

保存 state_dict

保存 state_dict

torch.save(model.state_dict(), PATH)

加载 state_dict

model = TheModelClass(*args, **kwargs)
model.load_state_dict(torch.load(PATH))
model.eval()