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

阅读 78

2022-05-05

废话不多说,直接上代码

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)

最后的输出结果为:

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

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

精彩评论(0)

0 0 举报