Pytorch .pth权重文件的使用解析


Posted in Python onFebruary 14, 2020

pytorch最后的权重文件是.pth格式的。

经常遇到的问题:

进行finutune时,改配置文件中的学习率,发现程序跑起来后竟然保持了以前的学习率, 并没有使用新的学习率。

原因:

首先查看.pth文件中的内容,我们发现它其实是一个字典格式的文件

Pytorch .pth权重文件的使用解析

其中保存了optimizer和scheduler,所以再次加载此文件时会使用之前的学习率。

我们只需要权重,也就是model部分,将其导出就可以了

import torch
 
original = torch.load('path/to/your/checkpoint.pth')
 
new = {"model": original["model"]}
torch.save(new, 'path/to/new/checkpoint.pth')

以上这篇Pytorch .pth权重文件的使用解析就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python optparse模块使用实例
Apr 09 Python
Python3 Random模块代码详解
Dec 04 Python
解决python大批量读写.doc文件的问题
May 08 Python
python的concat等多种用法详解
Nov 28 Python
Form表单及django的form表单的补充
Jul 25 Python
python单向链表的基本实现与使用方法【定义、遍历、添加、删除、查找等】
Oct 24 Python
在Django中实现添加user到group并查看
Nov 18 Python
Python3 解决读取中文文件txt编码的问题
Dec 20 Python
Python描述符descriptor使用原理解析
Mar 21 Python
使用Python matplotlib作图时,设置横纵坐标轴数值以百分比(%)显示
May 16 Python
基于Python把网站域名解析成ip地址
May 25 Python
vscode+PyQt5安装详解步骤
Aug 12 Python
Python netmiko模块的使用
Feb 14 #Python
pytorch模型存储的2种实现方法
Feb 14 #Python
Python3运算符常见用法分析
Feb 14 #Python
pytorch实现从本地加载 .pth 格式模型
Feb 14 #Python
python 实现线程之间的通信示例
Feb 14 #Python
Python3变量与基本数据类型用法实例分析
Feb 14 #Python
OpenCV中VideoCapture类的使用详解
Feb 14 #Python
You might like
php 采集书并合成txt格式的实现代码
2009/03/01 PHP
PHP动态输出JavaScript代码实例
2015/02/12 PHP
使用PHP接受文件并获得其后缀名的方法
2015/08/05 PHP
php进程间通讯实例分析
2016/07/11 PHP
PHP实现针对日期,月数,天数,周数,小时,分,秒等的加减运算示例【基于strtotime】
2017/04/19 PHP
thinkphp5引入公共部分header、footer的方法详解
2018/09/14 PHP
学习ExtJS table布局
2009/10/08 Javascript
Javascript合并表格中具有相同内容单元格示例
2013/08/11 Javascript
js实现在字符串中提取数字
2013/11/05 Javascript
JavaScript中创建对象和继承示例解读
2014/02/12 Javascript
Javascript+CSS实现影像卷帘效果思路及代码
2014/10/20 Javascript
jQuery动态效果显示人物结构关系图的方法
2015/05/07 Javascript
javascript+css3 实现动态按钮菜单特效
2016/02/06 Javascript
JS控制FileUpload的上传文件类型实例代码
2016/10/07 Javascript
百度地图JavascriptApi Marker平滑移动及车头指向行径方向
2017/03/13 Javascript
实例教学如何写vue插件
2017/11/30 Javascript
在Vue组件上动态添加和删除属性方法
2018/02/23 Javascript
小谈angular ng deploy的实现
2020/04/07 Javascript
[17:36]VG战队纪录片
2014/08/21 DOTA
python 远程统计文件代码分享
2015/05/14 Python
Flask web开发处理POST请求实现(登录案例)
2018/07/26 Python
Python面向对象基础入门之设置对象属性
2018/12/11 Python
使用Python进行体育竞技分析(预测球队成绩)
2019/05/16 Python
python使用paramiko模块通过ssh2协议对交换机进行配置的方法
2019/07/25 Python
python scrapy重复执行实现代码详解
2019/12/28 Python
python实现录屏功能(亲测好用)
2020/03/02 Python
keras 两种训练模型方式详解fit和fit_generator(节省内存)
2020/07/03 Python
细说NumPy数组的四种乘法的使用
2020/12/18 Python
Oracle中delete,truncate和drop的区别
2016/05/05 面试题
大学专科生推荐信范文
2013/11/23 职场文书
银行员工辞职信范文
2014/01/20 职场文书
大学毕业感言50字
2014/02/07 职场文书
求职意向书
2014/04/01 职场文书
师德自我剖析材料范文
2014/10/06 职场文书
2015年艾滋病防治工作总结
2015/05/22 职场文书
护士年终工作总结不会写?各科护士模板总结
2020/01/02 职场文书