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 16 Python
Ubuntu 下 vim 搭建python 环境 配置
Jun 12 Python
windows下python安装paramiko模块和pycrypto模块(简单三步)
Jul 06 Python
Python2.7基于笛卡尔积算法实现N个数组的排列组合运算示例
Nov 23 Python
python实现键盘输入的实操方法
Jul 16 Python
python datetime中strptime用法详解
Aug 29 Python
Python使用APScheduler实现定时任务过程解析
Sep 11 Python
Python生成六万个随机,唯一的8位数字和数字组成的随机字符串实例
Mar 03 Python
python中的socket实现ftp客户端和服务器收发文件及md5加密文件
Apr 01 Python
tensorflow中tf.reduce_mean函数的使用
Apr 19 Python
django 外键创建注意事项说明
May 20 Python
python链表类中获取元素实例方法
Feb 23 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
用js进行url编码后用php反解以及用php实现js的escape功能函数总结
2010/02/08 PHP
PHP实现长文章分页实例代码(附源码)
2016/02/03 PHP
PHP7扩展开发教程之Hello World实现方法示例
2017/08/03 PHP
jquery学习笔记二 实现可编辑的表格
2010/04/09 Javascript
原生Js页面滚动延迟加载图片实现原理及过程
2013/06/24 Javascript
jquery click([data],fn)使用方法实例介绍
2013/07/08 Javascript
一个jquery实现的不错的多行文字图片滚动效果
2014/09/28 Javascript
实例讲解javascript注册事件处理函数
2016/01/09 Javascript
jQuery基础知识点总结(必看)
2016/05/31 Javascript
Vue input控件通过value绑定动态属性及修饰符的方法
2017/05/03 Javascript
js禁止浏览器页面后退功能的实例(推荐)
2017/09/01 Javascript
parabola.js抛物线与加入购物车效果的示例代码
2017/10/25 Javascript
Vue press 支持图片放大功能的实例代码
2018/11/09 Javascript
详解vuex中action何时完成以及如何正确调用dispatch的思考
2019/01/21 Javascript
jQuery实现表格的增、删、改操作示例
2019/01/27 jQuery
微信小程序开发实现的IP地址查询功能示例
2019/03/28 Javascript
详解JavaScript之ES5的继承
2020/07/08 Javascript
maptalks+three.js+vue webpack实现二维地图上贴三维模型操作
2020/08/10 Javascript
jQuery实现简单评论功能
2020/08/19 jQuery
Python更新数据库脚本两种方法及对比介绍
2017/07/27 Python
python去除扩展名的实例讲解
2018/04/23 Python
对Python中list的倒序索引和切片实例讲解
2018/11/15 Python
python程序 创建多线程过程详解
2019/09/23 Python
opencv 图像滤波(均值,方框,高斯,中值)
2020/07/08 Python
python进度条显示之tqmd模块
2020/08/22 Python
css3 仿写阿里云水纹效果的示例代码
2018/02/10 HTML / CSS
采用专利算法搜索最廉价的机票:CheapAir
2016/09/10 全球购物
韩国女装NO.1网店:STYLENANDA
2016/09/16 全球购物
在使用非全零作为空指针内部表达的机器上, NULL是如何定义
2014/11/09 面试题
医务人员竞聘职务自我评价分享
2013/11/08 职场文书
期末考试动员演讲稿
2014/01/10 职场文书
创建服务型党组织实施方案
2014/02/25 职场文书
求职自荐信怎么写
2014/03/06 职场文书
关于梦想的演讲稿
2014/05/05 职场文书
爱心募捐感谢信
2015/01/22 职场文书
行政人事专员岗位职责
2015/04/07 职场文书