这个屌丝很懒，什么也没留下！

权重pt文件pth文件如何查看？_怎么查看.pth

作者：繁依Fanyi0 | 2024-06-30 13:15:27

对于在深度学习过程中，模型生成的权重文件是没办法直接打开的，这里展示利用代码来读取网络层和相关参数。


import torch
 
pthfile = r'./last_ckpt.pt'  # .pth文件的路径
model = torch.load(pthfile, torch.device('cpu'))  # 设置在cpu环境下查询
print('type:')
print(type(model))  # 查看模型字典长度
print('length:')
print(len(model))
print('key:')
for k in model.keys():  # 查看模型字典里面的key
    print(k)
print('value:')
for k in model:  # 查看模型字典里面的value
    print(k, model[k])