pth = r'./model/iresnet100.pth'
sta_dic = torch.load(pth)
print('.pth type:', type(sta_dic))
print('.pth len:', len(sta_dic))
print('--------------------------')
for k in sta_dic.keys():
print(k, type(sta_dic[k]), sta_dic[k].shape)
输出:
pth文件是dict,key为各层参数名称,value为参数值,数据类型为tensor