国外镜像 vue组件 反射 razor text Vanilla JS vue入门 jquery触发点击事件 打印缩放怎么设置 android常用布局 kafka消费不到数据 oracle查询数据库 mysql自然连接 python运算符优先级 python位操作 java中的数据结构 jdk环境配置 java时间转时间戳 java中文文档 linux安装系统 js数组删除指定元素 volist shutil labview宝典 python游戏代码 qq免安装版 ram容量是什么意思 git命令 gunzip iframe跨域 战地2地图包下载 ps祛痘 字符串分割 方正兰亭大黑简体 js观察者模式 cad圆形变成多边形 this关键字 qupzilla 1000kbps xshow
当前位置: 首页 > 学习教程  > 编程语言

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

2021/1/22 23:09:22 文章标签:

更多python教程请到: 菜鸟教程 https://www.piaodoo.com/ 当我们花费大量的精力训练完网络,下次预测数据时不想再(有时也不必再)训练一次时,这时候torch.save(),torch.load()就要登场了。 保存和加载模型参数有两种方…

更多python教程请到: 菜鸟教程 https://www.piaodoo.com/


当我们花费大量的精力训练完网络,下次预测数据时不想再(有时也不必再)训练一次时,这时候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的保存和加载模型参数的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持菜鸟教程www.piaodoo.com。


本文链接: http://www.dtmao.cc/news_show_630878.shtml

附件下载

相关教程

    暂无相关的数据...

共有条评论 网友评论

验证码: 看不清楚?