Pytorch使用手册-Save and Load the Model（专题八）
保存和加载模型
在本节中，我们将学习如何通过保存、加载模型的状态来持久化模型，并运行模型预测。
一、 保存和加载模型权重
PyTorch 模型将学习到的参数存储在一个内部状态字典中，称为 state_dict。
这些参数可以通过 torch.save 方法保存下来：
model = models.vgg16(weights='IMAGENET1K_V1')
torch.save(model.state_dict(原文地址：https://blog.csdn.net/qq_39698985/article/details/144003496
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！