pth文件解析
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
原创
2020-05-24 20:37:08 ·
5729 阅读 ·
2 评论