import torch
path = '/media/workstation/4.0T/master/model.pth'
pretrained_dict = torch.load(path)
for k, v in pretrained_dict.items(): # k 参数名 v 对应参数值
print(k)
查看参数模型.pth 里的参数
最新推荐文章于 2024-05-26 15:41:14 发布
import torch
path = '/media/workstation/4.0T/master/model.pth'
pretrained_dict = torch.load(path)
for k, v in pretrained_dict.items(): # k 参数名 v 对应参数值
print(k)