【PyTorch 查看 pth 文件】

pth文件是PyTorch中用于保存和加载模型参数、网络结构和训练过程中产生的元数据的一种重要文件格式
以下是查看pth文件的代码

# coding:utf-8
import torch
content = torch.load('output_dir/1-pretrained-model.pth', map_location='cpu')
print(content.keys())
print(content['model'])
print(content['optimizer'])
print(content['epoch'])
print(content['args'])