Pytorch - 模型保存与加载

原文：Saving and Loading Models
作者：Matthew Inkawhich

介绍一系列关于 PyTorch 模型保存与加载的应用场景，主要包括三个核心函数：

[1] - torch.save

保存序列化的对象(Serialized object)到磁盘.

其中，应用了 Python 的 pickle 包，进行序列化，可适用于模型Models，张量Tensors，以及各种类型的字典对象的序列化保存.

[2] - torch.load

采用 Python 的 pickle 的 unpickling 函数，对磁盘 pickled 的对象文件进行反序列化(deserialize)，加载到内存.

[3] - torch.nn.Module.load_state_dict

采用序列化的 state_dict 加载模型参数(字典).

懒得搬运了，看这里