基于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网络编程学习笔记(一)
Jun 09 Python
Python中给List添加元素的4种方法分享
Nov 28 Python
分享一下Python 开发者节省时间的10个方法
Oct 02 Python
python列表的常用操作方法小结
May 21 Python
Python自动化运维之IP地址处理模块详解
Dec 10 Python
windows 下python+numpy安装实用教程
Dec 23 Python
OpenCV2.3.1+Python2.7.3+Numpy等的配置解析
Jan 05 Python
python使用turtle绘制国际象棋棋盘
May 23 Python
在Pandas中处理NaN值的方法
Jun 25 Python
详解如何用TensorFlow训练和识别/分类自定义图片
Aug 05 Python
django实现web接口 python3模拟Post请求方式
Nov 19 Python
python 字段拆分详解
Dec 17 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
PHP网站提速三大“软”招
2006/10/09 PHP
PHP中的Session对象如何使用
2015/09/25 PHP
PHP编程文件处理类SplFileObject和SplFileInfo用法实例分析
2017/07/22 PHP
Laravel框架文件上传功能实现方法示例
2019/04/16 PHP
php变量与字符串的增删改查操作示例
2020/05/07 PHP
jquery滚动条插件jScrollPane的使用介绍
2013/11/08 Javascript
Ajax请求在数据量大的时候出现超时的解决方法
2014/02/27 Javascript
深入理解JavaScript系列(21):S.O.L.I.D五大原则之接口隔离原则ISP详解
2015/03/05 Javascript
分享jQuery封装好的一些常用操作
2016/07/28 Javascript
jQuery基本选择器和层次选择器学习使用
2017/02/27 Javascript
AngularJs定时器$interval 和 $timeout详解
2017/05/25 Javascript
Bootstrap Tooltip显示换行和左对齐的解决方案
2017/10/11 Javascript
vue的常用组件操作方法应用分析
2018/04/13 Javascript
解决layui富文本编辑器图片上传无法回显的问题
2019/09/18 Javascript
vue瀑布流组件实现上拉加载更多
2020/03/10 Javascript
python基于itchat实现微信群消息同步机器人
2017/02/27 Python
Python中关键字nonlocal和global的声明与解析
2017/03/12 Python
Python创建对称矩阵的方法示例【基于numpy模块】
2017/10/12 Python
如何在python开发工具PyCharm中搭建QtPy环境(教程详解)
2020/02/04 Python
tensorflow 利用expand_dims和squeeze扩展和压缩tensor维度方式
2020/02/07 Python
15行Python代码实现免费发送手机短信推送消息功能
2020/02/27 Python
Python Django2 model 查询介绍(条件、范围、模糊查询)
2020/03/16 Python
Opencv求取连通区域重心实例
2020/06/04 Python
为什么说python适合写爬虫
2020/06/11 Python
使用python库xlsxwriter库来输出各种xlsx文件的示例
2020/09/01 Python
Python 将代码转换为可执行文件脱离python环境运行(步骤详解)
2021/01/25 Python
如何设置Java的运行环境
2013/04/05 面试题
大学生职业规划论文
2014/01/11 职场文书
工作失误检讨书范文大全
2014/01/13 职场文书
优秀求职信范文分享
2014/01/26 职场文书
晨会主持词
2014/03/17 职场文书
中国梦读书活动总结
2014/07/10 职场文书
3.15消费者权益日活动总结
2015/02/09 职场文书
雷锋的观后感
2015/06/10 职场文书
iPhone13再次曝光
2021/04/15 数码科技
《遗弃》开发商删推文要跑路?官方回应:还在开发
2022/04/03 其他游戏