基于pytorch的保存和加载模型参数的方法


Posted in Python onAugust 17, 2019

当我们花费大量的精力训练完网络,下次预测数据时不想再(有时也不必再)训练一次时,这时候torch.save(),torch.load()就要登场了。

保存和加载模型参数有两种方式:

方式一:

torch.save(net.state_dict(),path):

功能:保存训练完的网络的各层参数(即weights和bias)

其中:net.state_dict()获取各层参数,path是文件存放路径(通常保存文件格式为.pt或.pth)

net2.load_state_dict(torch.load(path)):

功能:加载保存到path中的各层参数到神经网络

注意:不可以直接为torch.load_state_dict(path),此函数不能直接接收字符串类型参数

方式二:

torch.save(net,path):

功能:保存训练完的整个网络模型(不止weights和bias)

net2=torch.load(path):

功能:加载保存到path中的整个神经网络

说明:官方推荐方式一,原因自然是保存的内容少,速度会更快。

以上这篇基于pytorch的保存和加载模型参数的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python自动化测试之从命令行运行测试用例with verbosity
Sep 28 Python
在Python中操作时间之tzset()方法的使用教程
May 22 Python
基于Python如何使用AIML搭建聊天机器人
Jan 27 Python
python 创建弹出式菜单的实现代码
Jul 11 Python
解决tensorflow1.x版本加载saver.restore目录报错的问题
Jul 26 Python
Python的argparse库使用详解
Oct 09 Python
python 梯度法求解函数极值的实例
Jul 10 Python
Python实现串口通信(pyserial)过程解析
Sep 25 Python
Django中从mysql数据库中获取数据传到echarts方式
Apr 07 Python
解决安装新版PyQt5、PyQT5-tool后打不开并Designer.exe提示no Qt platform plugin的问题
Apr 24 Python
Python3+selenium配置常见报错解决方案
Aug 28 Python
Python使用scapy模块发包收包
May 07 Python
pytorch 固定部分参数训练的方法
Aug 17 #Python
python之PyQt按钮右键菜单功能的实现代码
Aug 17 #Python
pytorch 在网络中添加可训练参数,修改预训练权重文件的方法
Aug 17 #Python
python PyQt5/Pyside2 按钮右击菜单实例代码
Aug 17 #Python
Pytorch 实现自定义参数层的例子
Aug 17 #Python
Python中PyQt5/PySide2的按钮控件使用实例
Aug 17 #Python
画pytorch模型图,以及参数计算的方法
Aug 17 #Python
You might like
php5中date()得出的时间为什么不是当前时间的解决方法
2008/06/30 PHP
php 友好URL的实现(吐血推荐)
2008/10/04 PHP
PHP 错误处理机制
2015/07/06 PHP
PHP实现根据数组某个键值大小进行排序的方法
2018/03/13 PHP
js jquery做的图片连续滚动代码
2008/01/06 Javascript
Jquery 数据选择插件Pickerbox使用介绍
2012/08/24 Javascript
jQuery$命名冲突怎么办如何解决
2014/01/16 Javascript
jquery的ajax跨域请求原理和示例
2014/05/08 Javascript
JavaScript实现同步于本地时间的动态时间显示方法
2015/02/02 Javascript
详解JavaScript中的forEach()方法的使用
2015/06/08 Javascript
JS使用post提交的两种方式
2015/12/03 Javascript
jQuery添加和删除指定标签的方法
2015/12/16 Javascript
Angular中使用ui router实现系统权限控制及开发遇到问题
2016/09/23 Javascript
JavaScript 继承详解(六)
2016/10/11 Javascript
JavaScript仿支付宝6位数字密码输入框
2016/12/29 Javascript
ztree实现左边动态生成树右边为内容详情功能
2017/11/03 Javascript
vue中el-upload上传图片到七牛的示例代码
2018/10/19 Javascript
js实现鼠标拖拽缩放div实例代码
2019/03/25 Javascript
vue项目启动出现cannot GET /服务错误的解决方法
2020/04/26 Javascript
vue实现简单计算商品价格
2020/09/14 Javascript
python实现的各种排序算法代码
2013/03/04 Python
python如何读写csv数据
2018/03/21 Python
Python拼接字符串的7种方法总结
2018/11/01 Python
python射线法判断一个点在图形区域内外
2019/06/28 Python
基于pytorch的保存和加载模型参数的方法
2019/08/17 Python
kafka监控获取指定topic的消息总量示例
2019/12/23 Python
Python3 pickle对象串行化代码实例解析
2020/03/23 Python
解决pymysql cursor.fetchall() 获取不到数据的问题
2020/05/15 Python
LN-CC中国:高端男装和女装的奢侈时尚目的地
2019/09/14 全球购物
Mybag美国/加拿大:英国奢华包包和名牌手袋网站
2020/02/16 全球购物
母亲节演讲稿范文
2014/01/02 职场文书
作弊检讨书1000字
2014/02/01 职场文书
竞争与合作演讲稿
2014/05/12 职场文书
小学生植树节活动总结
2014/07/04 职场文书
2016年心理学教育培训学习心得体会
2016/01/12 职场文书
js 实现Material UI点击涟漪效果示例
2022/09/23 Javascript