Bootstrap

Pytorch使用手册-Save and Load the Model(专题八)

保存和加载模型

在本节中,我们将学习如何通过保存、加载模型的状态来持久化模型,并运行模型预测。


一、 保存和加载模型权重

PyTorch 模型将学习到的参数存储在一个内部状态字典中,称为 state_dict
这些参数可以通过 torch.save 方法保存下来:

model = models.vgg16(weights='IMAGENET1K_V1')
torch.save(model.state_dict(
;