linux系统pkl,pytorch模型 .pt, .pth, .pkl有什么区别及如何保存
pytorch模型 .pt, .pth, .pkl有什么区别及如何保存
发布时间:2020-07-22 10:47:44
来源:亿速云
阅读:371
作者:小猪
小编这次要给大家分享的是pytorch模型 .pt, .pth, .pkl有什么区别及如何保存,文章内容丰富,感兴趣的小伙伴可以来了解一下,希望大家阅读完这篇文章之后能够有所收获。
我们经常会看到后缀名为.pt, .pth, .pkl的pytorch模型文件,这几种模型文件在格式上有什么区别吗?
其实它们并不是在格式上有区别,只是后缀不同而已(仅此而已),在用torch.save()函数保存模型文件时,各人有不同的喜好,有些人喜欢用.pt后缀,有些人喜欢用.pth或.pkl.用相同的torch.save()语句保存出来的模型文件没有什么不同。
在pytorch官方的文档/代码里,有用.pt的,也有用.pth的。一般惯例是使用.pth,但是官方文档里貌似.pt更多,而且官方也不是很在意固定用一种。
模型保存与调用方式一:
保存:torch.save(model.state_dict(), mymodel.pth)#只保存模型权重参数,不保存模型结构
调用:
model = My_model(*args, **kwargs) #这里需要重新模型结构,My_model
model.load_state_dict(torch.load(mymodel.pth))#这里根据模型结构,调用存储的模型参数
model.eval()
模型保存与调用方式一:
保存:torch.save(model, mymodel.pth)#保存整个model的状态
调用:model=torch.load(mymodel.pth)#这里已经不需要重构模型结构了,直接load就可以
model.eval()
补充知识:关于python 导入包 .pth文件
Python在遍历已知的库文件目录过程中,如果见到一个.pth 文件,就会将文件中所记录的路径加入到 sys.path 设置中,于是 .pth 文件说指明的库也就可以被 Python 运行环境找到了。
python中有一个.pth文件,该文件的用法是:
首先xxx.pth文件里面会书写一些路径,一行一个。
将xxx.pth文件放在特定位置,则可以让python在加载模块时,读取xxx.pth中指定的路径。
*******************************************
有时,在用import导入项目文件夹里自己写的python文件时,常常说找不到该文件,这时可以将项目路径添加到PYTHONPATH下。
暂时添加:import sys
sys.path.append(项目路径)
这种方法只会存在内存里,一旦退出python就没有了
一劳永逸:
在 XXX/pythonXXX/site-packages/ (python 包安装路径) 下新建后缀为 .pth的文件,在其中加入你的项目路径,比如K:\Desktop\face_detect\DFace-win64-master\src
K:\Desktop\face_detect\DFace-win64-master\src\core
*******************************************
有时候我们正在修改或调试的程序会是一个库,为修改方便,我们可能不大希望把它放到 site-packages 下面,而是更愿意把它保留在原始的工程目录中,以方便 IDE 和版本控制工具进行管理。那么怎么能让 Python 运行环境找到这个库呢?
原理上, Python 运行环境查找库文件时本质是对 sys.path 列表的遍历,如果我们想给运行环境注册新的类库进来,
要么得用代码给 sys.path 列表增加新路径;
要么得调整 PYTHONPATH 环境变量;
要么就得把库文件复制到已经在 sys.path 设置中的路径中去(比如 site-packages 目录);
这些方法都不够方便。最简单的办法是用 .pth 文件来实现。Python 在遍历已知的库文件目录过程中,如果见到一个 .pth 文件,就会将文件中所记录的路径加入到 sys.path 设置中,于是 .pth 文件说指明的库也就可以被 Python 运行环境找到了。
其实,easy_install 所依赖的 egg 包安装就是靠 site-packages 目录下的 .pth 文件添加对 egg 包的引用实现的。所以修改对应的 .pth 文件内容,就可以实现对 egg 包的卸载。
看完这篇关于pytorch模型 .pt, .pth, .pkl有什么区别及如何保存的文章,如果觉得文章内容写得不错的话,可以把它分享出去给更多人看到。
linux系统pkl,pytorch模型 .pt, .pth, .pkl有什么区别及如何保存相关推荐
- 浅谈pytorch 模型 .pt, .pth, .pkl的区别及模型保存方式 pth中的路径加载使用
首先xxx.pth文件里面会书写一些路径,一行一个. 将xxx.pth文件放在特定位置,则可以让python在加载模块时,读取xxx.pth中指定的路径. Python客栈送红包.纸质书 有时,在用i ...
- pytorch保存模型pth_浅谈pytorch 模型 .pt, .pth, .pkl的区别及模型保存方式
我们经常会看到后缀名为.pt, .pth, .pkl的pytorch模型文件,这几种模型文件在格式上有什么区别吗? 其实它们并不是在格式上有区别,只是后缀不同而已(仅此而已),在用torch.save ...
- 3月3 pytorch模型保存的.pt, .pth, .pkl的pytorch模型文件,只是后缀不同而已(仅此而已),打开方式
pytorch模型保存的格式 首先讲讲保存模型或权重参数的后缀格式,权重参数和模型参数的后缀格式一样,pytorch中最常见的模型保存使用 .pt 或者是 .pth 作为模型文件扩展名.还有其他的保存 ...
- pytorch 模型保存方式(.pt, .pth, .pkl)
模型不同后缀名的区别 经常会看到后缀名为.pt, .pth, .pkl的pytorch模型文件,其实它们并不是在格式上有区别,只是后缀不同而已(仅此而已),在用torch.save()函数保存模型文件 ...
- 将训练好的pytorch模型的pth文件转换成onnx模型(亲测成功)
将训练好的pytorch模型的pth文件转换成onnx模型(亲测成功) 模型转换 声明:本文原创,未经许可严禁转载,原文地址https://blog.csdn.net/hutao1030813002/ ...
- Linux系统I/O模型和网络I/O模型
计算机I/O I/O在计算机中指Input/Output, Linux服务器 I/O 原理和流程 内核空间和用户空间数据交换 具体参见:https://blog.csdn.net/timonium/a ...
- pytorch模型(.pt)转onnx模型(.onnx)的方法详解(1)
1. pytorch模型转换到onnx模型 2.运行onnx模型 3.比对onnx模型和pytorch模型的输出结果 我这里重点是第一点和第二点,第三部分 比较容易 首先你要安装 依赖库:onnx ...
- Linux属于软件还是硬件,Linux系统教程_Linux系统软件处理和硬件处理的区别有什么不一样?...
在我的印象当中系统里的硬件和软件是一定的区别的,特别是Linux系统中的各种计算和处理的,可分为两类,一类是硬件处理,有一种软件处理,硬件处理和软件处理原理是类似的.但可以肯定的是差异化.下面小编就来 ...
- 用VMware创建Linux系统时用户名,全名,虚拟机名的区别
用VMware创建linux系统的时候,会有多个命名的地方 linux的全名和用户名如下图: 用户名:是用于限制权限的多用户操作系统管理工具,是登录是要输入的 个性化Linux全名:虚拟机的个性化名称 ...
最新文章
- 云炬Android开发笔记 6启动图功能开发与封装
- JavaScript——max-age
- Facebook 发币 Libra;谷歌十亿美金为穷人造房;第四代树莓派 Raspberry Pi 4 发布 | 开发者周刊...
- Python基础知识 D3
- 基于Python编写的倒计时工具
- i510300h和i79750h参数对比哪个好?
- Docker镜像的制作、上传、拉取和部署(利用阿里云)
- snmp-cmds:最简单的调用 Net-SNMP 二进制文件作为子进程的库可移植python SNMP 库
- 利用paddlepaddle爬取丁香园实时统计数据,并生成疫情地图!!
- 高端蓝牙耳机哪个牌子好?四款高音质不错的蓝牙耳机推荐
- 推荐一个类似于国内知乎国外网站-Quora
- 广告公司到底干什么的?欣奥诚分享
- 自建OTA服务器实现设备固件自动更新
- 菜刀 mysql_chopper菜刀一句话操作mysql数据库乱码问题脚本安全 -电脑资料
- 2016中国国际石墨烯创新大会展商名录抢先看
- 提高影响力:职权+威信
- 关于一些不为人知的小秘密
- 麻里麻里哄,还原基本法
- Kafka生产与消费脚本工具
- 牛客刷题错题(一)——测试知识
热门文章
- 基于java的扫雷论文_毕业论文基于JAVA的扫雷游戏设计
- 关于2022年6月青少年等级考试的几点说明:
- MacBook安装Docker
- 如何通过JQuery动态触发oninput事件
- 我在 Chrome 上玩了下云游戏,它能让我们以后都不抢 RTX3060Ti 了吗?
- ubuntu下显卡(独显,集显)切换
- 235、一个带宽如何连接两个路由器?详解两种方法
- 中位数及带权中位数问题
- 天涯[经济杂谈]一位创业者的商海感悟(语录版,连载中,持续更新)【力荐】...
- 适合直播无线蓝牙耳机什么牌子好,618音质最好蓝牙耳机测评