ubuntu中使用终端命令行快速查看torch.save保存的pth / pth.tar格式训练数据的方法
2021/10/31 7:12:25
本文主要是介绍ubuntu中使用终端命令行快速查看torch.save保存的pth / pth.tar格式训练数据的方法,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
在使用pytorch框架训练深度学习网络的时候,我们可以很方便地使用torch.save()方法对训练过程中的网络参数等信息进行保存。比如这里,我们保存成的文件格式为pth.tar(如下图所示),咋一看以为是一个压缩包,需要先解压一下。其实不用,直接使用torch.load就可以了。下面讲一下其数据查看方法。
首先,在这些pth.tar文件的目录下打开一个终端,执行:
python
进入python命令行模式。比如我想查看net_5000_checkpoint.pth.tar文件里的内容,那么,依次执行如下命令:
import torch checkpoint = torch.load('net_5000_checkpoint.pth.tar') print(checkpoint.keys()) #输出该文件里保存的内容的keys #比如我这里的输出为:dict_keys(['epoch', 'state_dict']) #表示这个文件里保存了名为epoch和state_dict这两个内容 print(checkpoint['epoch']) #查看key为epoch的内容 #我这里输出为5001
以上。其实输出的内容都是你在save时候自己写入的内容,而查看keys目的就是为了看一下保存了哪些项,名称是啥。
这篇关于ubuntu中使用终端命令行快速查看torch.save保存的pth / pth.tar格式训练数据的方法的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-07-03微信支付提示下单账户与支付账户不一致-icode9专业技术文章分享
- 2024-07-03微信支付提示订单号重复-icode9专业技术文章分享
- 2024-07-02微服务启动nacos注册上去了,但是一直没有收到请求-icode9专业技术文章分享
- 2024-07-02如何检查文件的编码格式-icode9专业技术文章分享
- 2024-07-02sublime 更改编码格式-icode9专业技术文章分享
- 2024-06-30uniAPP 实现全屏左右滚动滚动的效果-icode9专业技术文章分享
- 2024-06-30如何在本地使用授权或插件-icode9专业技术文章分享
- 2024-06-30伪静态规则配置方法汇总-icode9专业技术文章分享
- 2024-06-29易优CMS安装常见问题汇总-icode9专业技术文章分享
- 2024-06-28易优新手必读安装教程-icode9专业技术文章分享