PyTorch 模型可以保存为两种格式:.pt:此格式保存整个模型,
包括其架构和学习参数。.pth:此格式仅保存模型的状态字典,其中包括模型的学习参数和一些元数据。
PyTorch
格式基于 Python 的 pickle 模块,该模块用于序列化 Python 对象。
为了理解 pickle
的工作原理,让我们看以下示例:
import pickle
model_state_dict = { "layer1": "hello", "layer2": "world" }
pickle.dump(model_state_dict, open("model.pkl", "wb"))The pickle.dump() 函数将 model_state_dict 字典序列化并保存到名为
model.pkl. 的文件中。
输出文件现在包含字典的
二进制表示:
model.pkl hex view要
将序列化的字典加载回
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。