pytorch模型存储的2种实现方法
更新时间：2020年02月14日 12:00:56 作者：慢行厚积
今天小编就为大家分享一篇pytorch模型存储的2种实现方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
1、保存整个网络结构信息和模型参数信息：
torch.save(model_object, './model.pth')
直接加载即可使用：
model = torch.load('./model.pth')
2、只保存网络的模型参数-推荐使用
torch.save(model_object.state_dict(), './params.pth')
加载则要先从本地网络模块导入网络，然后再加载参数：
from models import AgeModel
model = AgeModel()
model.load_state_dict(torch.load('./params.pth'))
以上这篇pytorch模型存储的2种实现方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
浅谈Python中的可迭代对象、迭代器、For循环工作机制、生成器
这篇文章主要介绍了Python中的可迭代对象、迭代器、For循环工作机制、生成器，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-03-03
今天小编就为大家分享一篇Python实现钉钉发送报警消息的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-02-02
这篇文章主要为大家介绍了python使用TCP实现的对话客户端和服务器实现示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-03-03
这篇文章主要介绍了Python命令行click参数用法解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-12-12
这篇文章主要介绍了Python实现字符串与数组相互转换功能,结合具体实例形式分析了Python字符串与数组相关转换功能的相关实现技巧与注意事项,需要的朋友可以参考下2017-09-09
这篇文章主要介绍了Python线程条件变量Condition原理解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-01-01
本文给大家讲解的是使用Python的Testlink实现将实现将xml导入至excel表格中，方法非常的简单，另外附上其他小伙伴的方法，有需要的童鞋们可以参考下。2015-11-11
这篇文章主要为大家详细介绍了python的数据类型，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下希望能够给你带来帮助2022-01-01
这篇文章主要介绍了使用Python发送各种形式的邮件的方法汇总,包括HTML形式的或者带图片以及带附件的邮件等等,需要的朋友可以参考下2015-11-11
这篇文章主要为大家介绍了python使用turtle实现最简单简单奥运五环绘图，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-05-05
最新评论