pytorch模型保存的格式

首先讲讲保存模型或权重参数的后缀格式，权重参数和模型参数的后缀格式一样，pytorch中最常见的模型保存使用 .pt 或者是 .pth 作为模型文件扩展名。还有其他的保存数据的格式为.t7或者.pkl格式。t7文件是沿用torch7中读取模型权重的方式，而pth文件是python中存储文件的常用格式，而在keras中则是使用.h5文件。

我们经常会看到后缀名为.pt, .pth, .pkl的pytorch模型文件，这几种模型文件在格式上有什么区别吗？其实它们并不是在格式上有区别，只是后缀不同而已（仅此而已），在用torch.save()函数保存模型文件时，各人有不同的喜好，有些人喜欢用.pt后缀，有些人喜欢用.pth或.pkl.用相同的torch.save（）语句保存出来的模型文件没有什么不同。

在pytorch官方的文档/代码里，有用.pt的，也有用.pth的。一般惯例是使用.pth,但是官方文档里貌似.pt更多，而且官方也不是很在意固定用一种。

打开pkl的方式

.pkl文件是python保存文件的一种文件格式，如果直接打开会显示一堆序列化的东西。
需要使用rb类型来打开
rb – 读取2进制文件，r – 读取文本文件

# cPickle是python2系列用的，3系列已经不用了，直接用pickle就好了
import pickle
data = pickle.load( open('a.pkl','rb'))  #记得加上'rb'
print(data)import pickle
if __name__ == '__main__':a = [1, 2, 3]# 以二进制的形式打开文件with open("./1.pkl", "wb") as f:# 将列表a序列化后写入文件pickle.dump(a, f)with open("./1.pkl", 'rb') as f:# 反序列化解析成列表aa = pickle.load(f)print(a, a.__class__)  # [1, 2, 3] <class 'list'>import os
import pickledef read_pickle(work_path):data_list = []with open(work_path, "rb") as f:while True :try:data = pickle.load(f)    #data_list.append(data)print(data.__class__)print(data.keys())print(set(data))print(data['infos'].__class__)print(data['metadata'].__class__)print(set(data['metadata']))print(data['metadata']['version'].__class__)print(data['infos'][0])print(data['infos'][0].__class__)print(data['infos'][0].keys())except EOFError:breakreturn data_listdef main():pkl_path = 'nuscenes_infos_temporal_test.pkl'data_list = read_pickle(pkl_path)print(data_list)
if __name__=="__main__":main()

3月3 pytorch模型保存的.pt, .pth, .pkl的pytorch模型文件，只是后缀不同而已（仅此而已），打开方式相关推荐

pytorch 模型保存方式(.pt, .pth, .pkl)
模型不同后缀名的区别经常会看到后缀名为.pt, .pth, .pkl的pytorch模型文件,其实它们并不是在格式上有区别,只是后缀不同而已(仅此而已),在用torch.save()函数保存模型文件 ...
pytorch保存模型pth_浅谈pytorch 模型 .pt, .pth, .pkl的区别及模型保存方式
我们经常会看到后缀名为.pt, .pth, .pkl的pytorch模型文件,这几种模型文件在格式上有什么区别吗? 其实它们并不是在格式上有区别,只是后缀不同而已(仅此而已),在用torch.save ...
linux系统pkl,pytorch模型 .pt, .pth, .pkl有什么区别及如何保存
pytorch模型 .pt, .pth, .pkl有什么区别及如何保存发布时间:2020-07-22 10:47:44 来源:亿速云阅读:371 作者:小猪小编这次要给大家分享的是pytorch ...
浅谈pytorch 模型 .pt, .pth, .pkl的区别及模型保存方式 pth中的路径加载使用
首先xxx.pth文件里面会书写一些路径,一行一个. 将xxx.pth文件放在特定位置,则可以让python在加载模块时,读取xxx.pth中指定的路径. Python客栈送红包.纸质书有时,在用i ...
Tensorflow 2.x(keras)源码详解之第十章：keras中的模型保存与加载(详解Checkpointmd5模型序列化)
大家好,我是爱编程的喵喵.双985硕士毕业,现担任全栈工程师一职,热衷于将数据思维应用到工作与生活中.从事机器学习以及相关的前后端开发工作.曾在阿里云.科大讯飞.CCF等比赛获得多次Top名次.现 ...
深度学习模型保存_解读计算机视觉的深度学习模型
作者 | Dipanjan(DJ)Sarkar 来源 | Medium 编辑 | 代码医生团队介绍人工智能(AI)不再仅限于研究论文和学术界.业内不同领域的企业和组织正在构建由AI支持的大规模应用 ...
Pytorch学习 - 保存模型和重新加载
Pytorch学习 - 保存和加载模型 1. 3个函数 2. 模型不同后缀名的区别 3. 保存和重载模型参考资料: Pytorch官方文档链接某博客 1. 3个函数 torch.save() : ...
Tensorflow2 图像分类-Flowers数据深度学习模型保存、读取、参数查看和图像预测
目录 1.原文完整代码 1.1 模型运行参数总结 1.2模型训练效果编辑2.模型的保存 3.读取模型model 4.使用模型进行图片预测 5.补充如何查看保存模型参数 5.1 model_wei ...
tensor和模型保存与加载 PyTorch
PyTorch教程-7:PyTorch中保存与加载tensor和模型详解保存和读取Tensor PyTorch中的tensor可以保存成 .pt 或者 .pth 格式的文件,使用torch.save ...

3月3 pytorch模型保存的.pt, .pth, .pkl的pytorch模型文件，只是后缀不同而已（仅此而已），打开方式

pytorch模型保存的格式

打开pkl的方式

3月3 pytorch模型保存的.pt, .pth, .pkl的pytorch模型文件，只是后缀不同而已（仅此而已），打开方式相关推荐

最新文章

热门文章