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读取mp3中ID3信息的方法
Mar 05 Python
Python简单操作sqlite3的方法示例
Mar 22 Python
Python算法输出1-9数组形成的结果为100的所有运算式
Nov 03 Python
Python实现自动发送邮件功能
Mar 02 Python
Python反射用法实例简析
Dec 22 Python
tensorflow 获取模型所有参数总和数量的方法
Jun 14 Python
Python实现快速计算词频功能示例
Jun 25 Python
python requests爬取高德地图数据的实例
Nov 10 Python
对python 命令的-u参数详解
Dec 03 Python
Python多项式回归的实现方法
Mar 11 Python
在django-xadmin中APScheduler的启动初始化实例
Nov 15 Python
Python3标准库之dbm UNIX键-值数据库问题
Mar 24 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设计模式 Decorator(装饰模式)
2011/06/26 PHP
php Session存储到Redis的方法
2013/11/04 PHP
php获取QQ头像并显示的方法
2014/12/23 PHP
zend framework中使用memcache的方法
2016/03/04 PHP
PHP 二维关联数组根据其中一个字段排序(推荐)
2017/04/04 PHP
jQuery 获取对象 定位子对象
2010/05/31 Javascript
使用js判断TextBox控件值改变然后出发事件
2014/03/07 Javascript
JavaScript代码实现禁止右键、禁选择、禁粘贴、禁shift、禁ctrl、禁alt
2015/11/17 Javascript
js密码强度检测
2016/01/07 Javascript
js实现的彩色方块飞舞奇幻效果
2016/01/27 Javascript
jQuery中通过ajax的get()函数读取页面的方法
2016/02/29 Javascript
Ext JS框架中日期函数的用法及日期选择控件的实现
2016/05/21 Javascript
JS实现刷新父页面不弹出提示框的方法
2016/06/22 Javascript
详解Vue路由History mode模式中页面无法渲染的原因及解决
2017/09/28 Javascript
基于jQuery ztree实现表格风格的树状结构
2018/08/31 jQuery
微信小程序背景音乐开发详解
2019/12/12 Javascript
优化Vue中date format的性能详解
2020/01/13 Javascript
[01:36:57]【09DOTA2第一视角】小骷髅
2014/04/16 DOTA
使用python实现拉钩网上的FizzBuzzWhizz问题示例
2014/05/05 Python
Python编程中运用闭包时所需要注意的一些地方
2015/05/02 Python
Python编程实现蚁群算法详解
2017/11/13 Python
numpy找出array中的最大值,最小值实例
2018/04/03 Python
对python 通过ssh访问数据库的实例详解
2019/02/19 Python
在cmd中查看python的安装路径方法
2019/07/03 Python
python实现单张图像拼接与批量图片拼接
2020/03/23 Python
python小技巧——将变量保存在本地及读取
2020/11/13 Python
详解使用HTML5的classList属性操作CSS类
2017/10/13 HTML / CSS
日语求职信范文
2013/12/17 职场文书
红领巾广播站广播稿
2014/02/01 职场文书
《蝙蝠和雷达》教学反思
2014/04/23 职场文书
校外活动方案
2014/08/28 职场文书
2015年党务公开工作总结
2015/05/19 职场文书
python爬取新闻门户网站的示例
2021/04/25 Python
Java实现斗地主之洗牌发牌
2021/06/14 Java/Android
postgresql 删除重复数据案例详解
2021/08/02 PostgreSQL
Python中Matplotlib的点、线形状、颜色以及绘制散点图
2022/04/07 Python