0
点赞
收藏
分享

微信扫一扫

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

Yaphets_巍 2022-05-05 阅读 78

废话不多说,直接上代码

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 条评论