基于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的Django框架下搭建的BLOG添加RSS功能的教程
Apr 08 Python
在Python下利用OpenCV来旋转图像的教程
Apr 16 Python
python的依赖管理的实现
May 14 Python
在Python中过滤Windows文件名中的非法字符方法
Jun 10 Python
Python:Numpy 求平均向量的实例
Jun 29 Python
django 单表操作实例详解
Jul 30 Python
Python中pymysql 模块的使用详解
Aug 12 Python
python实现生成Word、docx文件的方法分析
Aug 30 Python
python进程的状态、创建及使用方法详解
Dec 06 Python
python实现替换word中的关键文字(使用通配符)
Feb 13 Python
使用Dajngo 通过代码添加xadmin用户和权限(组)
Jul 03 Python
python如何调用php文件中的函数详解
Dec 29 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
Win2003下APACHE+PHP5+MYSQL4+PHPMYADMIN 的简易安装配置
2006/11/18 PHP
php 三维饼图的实现代码
2008/09/28 PHP
简单的php中文转拼音的实现代码
2014/02/11 PHP
PHP实现搜索地理位置及计算两点地理位置间距离的实例
2016/01/08 PHP
Laravel实现autoload方法详解
2017/05/07 PHP
找到一点可怜的关于dojo资料,谢谢作者!
2006/12/06 Javascript
javascript 学习之旅 (2)
2009/02/05 Javascript
JavaScript去掉数组中的重复元素
2011/01/13 Javascript
为你的网站增加亮点的9款jQuery插件推荐
2011/05/03 Javascript
JavaScript中的apply()方法和call()方法使用介绍
2012/07/25 Javascript
DOM基础教程之模型中的模型节点
2015/01/19 Javascript
jQueryMobile之Helloworld与页面切换的方法
2015/02/04 Javascript
JavaScript实现对下拉列表值进行排序的方法
2015/07/15 Javascript
11种ASP连接数据库的方法
2015/09/18 Javascript
详解Vue.js入门环境搭建
2017/03/17 Javascript
JS实现简单的浮动碰撞效果示例
2017/12/28 Javascript
puppeteer库入门初探
2019/01/09 Javascript
vue插件mescroll.js实现移动端上拉加载和下拉刷新
2019/03/07 Javascript
vscode 插件开发 + vue的操作方法
2020/06/05 Javascript
Python简单进程锁代码实例
2015/04/27 Python
Python元组 tuple的概念与基本操作详解【定义、创建、访问、计数、推导式等】
2019/10/30 Python
Tensorflow轻松实现XOR运算的方式
2020/02/03 Python
Python正则表达式高级使用方法汇总
2020/06/18 Python
Python脚本实现Zabbix多行日志监控过程解析
2020/08/26 Python
CSS3制作炫酷的下拉菜单及弹起式选单的实例分享
2016/05/17 HTML / CSS
使用postMessage让 iframe自适应高度的方法示例
2019/10/08 HTML / CSS
美国半成品食材配送服务商:Home Chef
2018/01/25 全球购物
德国Discount-Apotheke中文官网:DC德式康线上药房
2020/02/18 全球购物
早晨薰衣草在线女性精品店:Morning Lavender
2021/01/04 全球购物
应届生会计电算化求职信
2013/10/03 职场文书
建筑人员岗位职责
2013/12/25 职场文书
大学生自我评价范文分享
2014/02/21 职场文书
期末个人总结范文
2015/02/13 职场文书
大学生逃课检讨书
2015/05/04 职场文书
二手手机买卖合同范本(2019年版)
2019/10/28 职场文书
Python中的 Set 与 dict
2022/03/13 Python