5.4PyTorch模型保存与读取

文档摘要

5.4 PyTorch模型保存与读取在前面几节的内容中，我们介绍了如何构建和修改PyTorch模型。本节我们来讨论PyTorch如何保存和读取训练好的模型和参数。另外，在很多场景下我们都会使用多GPU训练。这种情况下，模型会分布于各个GPU上（参加2.3节分布数据式训练，这里暂不考虑分布模型式训练），模型的保存和读取与单GPU训练情景下是否有所不同？经过本节的学习，你将收获： PyTorch的模型的存储格式 PyTorch如何存储模型单卡与多卡训练下模型的保存与加载方法优化器模型参数的保存 5.4.1 模型存储格式 PyTorch存储模型主要采用pkl，pt，pth三种格式。就使用层面来说没有区别，这里不做具体的讨论。