pytorch模型存储的2种实现方法


Posted in Python onFebruary 14, 2020

1、保存整个网络结构信息和模型参数信息:

torch.save(model_object, './model.pth')

直接加载即可使用:

model = torch.load('./model.pth')

2、只保存网络的模型参数-推荐使用

torch.save(model_object.state_dict(), './params.pth')

加载则要先从本地网络模块导入网络,然后再加载参数:

from models import AgeModel
model = AgeModel()
model.load_state_dict(torch.load('./params.pth'))

以上这篇pytorch模型存储的2种实现方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
删除目录下相同文件的python代码(逐级优化)
May 25 Python
使用Python压缩和解压缩zip文件的教程
May 06 Python
Python设置默认编码为utf8的方法
Jul 01 Python
python 删除大文件中的某一行(最有效率的方法)
Aug 19 Python
python3 判断列表是一个空列表的方法
May 04 Python
python使用turtle库绘制树
Jun 25 Python
详解Django中六个常用的自定义装饰器
Jul 04 Python
python通过zabbix api获取主机
Sep 17 Python
python 通过SSHTunnelForwarder隧道连接redis的方法
Feb 19 Python
Python3.6+Django2.0以上 xadmin站点的配置和使用教程图解
Jun 04 Python
Python GUI自动化实现绕过验证码登录
Jan 10 Python
Python快速优雅的批量修改Word文档样式
May 20 Python
Python3运算符常见用法分析
Feb 14 #Python
pytorch实现从本地加载 .pth 格式模型
Feb 14 #Python
python 实现线程之间的通信示例
Feb 14 #Python
Python3变量与基本数据类型用法实例分析
Feb 14 #Python
OpenCV中VideoCapture类的使用详解
Feb 14 #Python
Python networkx包的实现
Feb 14 #Python
python常用运维脚本实例小结
Feb 14 #Python
You might like
为什么《星际争霸》是测试人工智能的理想战场
2019/12/03 星际争霸
重置版游戏视频
2020/04/09 魔兽争霸
php获取字段名示例分享
2014/03/03 PHP
php准确计算复活节日期的方法
2015/04/18 PHP
使用PHP连接多种数据库的实现代码(mysql,access,sqlserver,Oracle)
2016/12/21 PHP
php获取客户端IP及URL的方法示例
2017/02/03 PHP
php实现文件与16进制相互转换的方法示例
2017/02/16 PHP
js综合应用实例简单的表格统计
2013/09/03 Javascript
jquery实现右键菜单插件
2015/03/29 Javascript
jquery实现页面虚拟键盘特效
2015/08/08 Javascript
JS实现黑色风格的网页TAB选项卡效果代码
2015/10/09 Javascript
基于canvas实现的钟摆效果完整实例
2016/01/26 Javascript
jQuery+formdata实现上传进度特效遇到的问题
2016/02/24 Javascript
AngularJS 避繁就简的路由
2016/07/01 Javascript
Vuejs第十篇之vuejs父子组件通信
2016/09/06 Javascript
Js获取当前日期时间及格式化代码
2016/09/17 Javascript
JS实现类似51job上的地区选择效果示例
2016/11/17 Javascript
微信小程序实现图片自适应(支持多图)
2017/01/25 Javascript
js实现横向拖拽导航条功能
2017/02/17 Javascript
Angular.JS去掉访问路径URL中的#号详解
2017/03/30 Javascript
JavaScript设计模式之装饰者模式定义与应用示例
2018/07/25 Javascript
python登陆asp网站页面的实现代码
2015/01/14 Python
python中set常用操作汇总
2016/06/30 Python
PyQt5实现让QScrollArea支持鼠标拖动的操作方法
2019/06/19 Python
python递归法解决棋盘分割问题
2019/07/17 Python
redis数据库及与python交互用法简单示例
2019/11/01 Python
python中tkinter窗口位置\坐标\大小等实现示例
2020/07/09 Python
解决python对齐错误的方法
2020/07/16 Python
意大利网上购书网站:Libraccio.it
2021/02/03 全球购物
简单的JAVA编程面试题
2013/03/19 面试题
计算机售后服务承诺书
2014/05/30 职场文书
小学课外阅读总结
2014/07/09 职场文书
竞选班干部演讲稿400字
2014/08/20 职场文书
2015年会计年终工作总结
2015/05/26 职场文书
幼儿园教师暑期培训心得体会
2016/01/09 职场文书
Win11应用商店打开闪退怎么解决? win11应用商店打不开的多种解决办法
2022/04/05 数码科技