1. 首先导入torch库
import torch
2. 然后,指明pth权重路径
这里路径用你自己的pth文件路径
pthfile = r'path\to\demo.pth'
3.接着,采用torch.load()函数来调出参数
net = torch.load(pthfile)
4. 最后print()打印参数
print(net)
完整代码如下
import torch
pthfile = r'path\to\demo.pth'
net = torch.load(pthfile)
print(net)