查看.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)

最后的输出结果为：

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

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