废话不多说,直接上代码
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)
最后的输出结果为:
-------------------------------------
说我文章篇幅太短,我也不知道说什么好了,,,,我就随便加点文字吧。。。