基于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读取mp3中ID3信息的方法
Mar 05 Python
Python检测字符串中是否包含某字符集合中的字符
May 21 Python
Python实现进程同步和通信的方法
Jan 02 Python
Python简单实现的代理服务器端口映射功能示例
Apr 08 Python
Python实现iOS自动化打包详解步骤
Oct 03 Python
浅谈Python3中strip()、lstrip()、rstrip()用法详解
Apr 29 Python
Python高级特性 切片 迭代解析
Aug 23 Python
Python求解正态分布置信区间教程
Nov 20 Python
解决python中显示图片的plt.imshow plt.show()内存泄漏问题
Apr 24 Python
Python在线和离线安装第三方库的方法
Oct 31 Python
使用django自带的user做外键的方法
Nov 30 Python
Pandas搭配lambda组合使用详解
Jan 22 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
IIS php环境配置PHP5 MySQL5 ZendOptimizer phpmyadmin安装与配置
2008/11/18 PHP
php 删除cookie和浏览器重定向
2009/03/16 PHP
php实现多维数组中每个单元值(数字)翻倍的方法
2015/02/16 PHP
jquery easyui滚动条部分设置介绍
2013/09/12 Javascript
鼠标滚轴控制文本框值的JS代码
2013/11/19 Javascript
[原创]推荐10款最热门jQuery UI框架
2014/08/19 Javascript
JavaScript Math.round() 方法
2015/12/18 Javascript
基于Bootstrap里面的Button dropdown打造自定义select
2016/05/30 Javascript
Bootstrap3.0建站教程(一)之bootstrap表单元素排版
2016/06/01 Javascript
js字符串操作总结(必看篇)
2016/11/22 Javascript
理解javascript中的闭包
2017/01/11 Javascript
JSONP跨域请求
2017/03/02 Javascript
微信小程序中使用javascript 回调函数
2017/05/11 Javascript
AngularJS创建一个上传照片的指令实例代码
2018/02/24 Javascript
layer弹出层 iframe层去掉滚动条的实例代码
2018/08/17 Javascript
详解微信小程序回到顶部的两种方式
2019/05/09 Javascript
详解VUE调用本地json的使用方法
2019/05/15 Javascript
使用apifm-wxapi快速开发小程序过程详解
2019/08/05 Javascript
Vue 实现把表单form数据 转化成json格式的数据
2019/10/29 Javascript
vue+elementui通用弹窗的实现(新增+编辑)
2021/01/07 Vue.js
[01:11:21]DOTA2-DPC中国联赛 正赛 Phoenix vs CDEC BO3 第三场 3月7日
2021/03/11 DOTA
Django ORM框架的定时任务如何使用详解
2017/10/19 Python
pandas多级分组实现排序的方法
2018/04/20 Python
python中利用h5py模块读取h5文件中的主键方法
2018/06/05 Python
python实现求两个字符串的最长公共子串方法
2018/07/20 Python
基于OpenCV的路面质量检测的实现
2020/11/04 Python
python中封包建立过程实例
2021/02/18 Python
意大利奢侈品综合电商网站:MODES
2019/12/14 全球购物
莫斯科制造商的廉价皮大衣:Fursk
2020/06/09 全球购物
Ejb技术面试题
2015/04/29 面试题
校长竞聘演讲稿
2014/05/16 职场文书
公司会议策划方案
2014/05/17 职场文书
庆祝儿童节标语
2014/10/09 职场文书
当你焦虑迷茫时,请读读这6句话
2019/07/24 职场文书
解析python中的jsonpath 提取器
2022/01/18 Python
深入浅出讲解Java8函数式编程
2022/01/18 Java/Android