保存

根据PyTorch文档，在把PyTorch模型保存成文件的时候有两种方法，第一种是推荐的：

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

对应地，加载模型这样做：

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

阅读更多