查看.pth文件里面保存了哪些权重及那些权重的shape

废话不多说，直接上代码

import torch  content = torch.load('logs/ep100-loss0.253-val_loss1.945.pth',map_location=torch.device('cpu'))
#torch.load('路径') 但是我的电脑没有GPU，是集成显卡呜呜呜，所以还得加个后面那部分map_location=torch.device('cpu')print(content.keys())   print(content['extractor.0.weight'].shape)

最后的输出结果为：

-------------------------------------

说我文章篇幅太短，我也不知道说什么好了，，，，我就随便加点文字吧。。。

查看.pth文件里面保存了哪些权重及那些权重的shape相关推荐

将训练好的pytorch模型的pth文件转换成onnx模型（亲测成功）
将训练好的pytorch模型的pth文件转换成onnx模型(亲测成功) 模型转换声明:本文原创,未经许可严禁转载,原文地址https://blog.csdn.net/hutao1030813002/ ...
为什么有的.pth文件储存的是数字有些.pth储存的是网络结构？
对自己之前的一段代码的.pth文件里面储存的内容有些疑惑,因为有时候.pth文件储存的是网络结构,而有些储存的是数据,例如: from collections import OrderedDict i ...
《Adobe Acrobat DC经典教程》—第1章1.11节在阅读模式下查看PDF文件
本节书摘来自异步社区<Adobe Acrobat DC经典教程>一书中的第1章1.11节在阅读模式下查看PDF文件,作者[美]Lisa Fridsma(丽莎弗里斯玛) , Brie Gy ...
FPGA管脚分配文件的保存
使用别人的工程时,有时找不到他的管脚文件,但可以把他已经绑定好的管脚保存下来,输出到文件里. 方法一: 查看引脚绑定情况,quartus -> assignment -> Pins,打开F ...
python怎么读取txt文件数据保存数组中-Python Numpy中数据的常用的保存与读取方法...
在经常性读取大量的数值文件时(比如深度学习训练数据),可以考虑现将数据存储为Numpy格式,然后直接使用Numpy去读取,速度相比为转化前快很多. 下面就常用的保存数据到二进制文件和保存数据到文本文件 ...
用UltraEdit判断打开文件的编码类型用UltraEdit或notepad记事本查看文件编码格式用UltraEdit查看当前文件编码...
用UltraEdit查看当前文件编码想判断文件的编码类型? 用强大的UltraEdit-32软件: UltraEdit-32的状态栏可以显示文件的编码类型,详细情况如下: ANSI/ANSCI--- ...
附件文件无法保存到服务器,可能是目录属性设置问题,请与管理员联系,Discuz不能上传过大文件的解决办法...
烈火网(LieHuo.Net)教程今天论坛附件无法上传,进服务器去查看,后来发展所有DISCUZ论坛都无法上传附件,这个问题很严重,应该说非常之严重,上网找解决方案没有找到,去discuz论坛提问,斑 ...
使用C语言查看一个文件夹中所有文件及目录
1.前言 1.1声明文章中的文字可能存在语法错语以及标点错误,请谅解: 如果在文章中发现代码错误或其它问题请告知,感谢! 2.实现方法为了使用C语言实现查看一个文件夹中的文件数量功能,可以使用di ...
linux view查看日志命令,linux查看日志文件内容命令tail、cat、tac、head、echo、vi
linux查看日志文件内容命令tail.cat.tac.head.echo tail -f test.log 你会看到屏幕不断有内容被打印出来. 这时候中断第一个进程Ctrl-C, --------- ...
7 centos 查看程序文件数量_「动手打造家庭媒体网络平台」安装篇-centos搭建DLNA媒体服务...
大家好,我是路程lucky,热爱开发.设计.学习.生活.爱捣鼓的web前后端工程师~ 本文的重点是在之前文章提到创建的centos基础上继续搭建DLNA媒体服务.由于centos的系统是从零开始初步安 ...

查看.pth文件里面保存了哪些权重及那些权重的shape

查看.pth文件里面保存了哪些权重及那些权重的shape相关推荐

最新文章

热门文章