查看生成的checkpoints的pth格式文件

一：

现在对应目录建一个txt程序，为

import torch

pthfile = r'./model_0.pth'  # .pth文件的路径（这里要改成你的文件名字）
model = torch.load(pthfile)  # 设置在cpu环境下查询

# 打开一个文件用于写入
with open('weight.txt', 'w') as f:
    # 写入模型类型和长度
    f.write('type:\n')
    f.write(str(type(model)) + '\n')
    f.write('length:\n')
    f.write(str(len(model)) + '\n\n')

    # 写入模型字典的key
    f.write('key:\n')
    for k in model.keys():
        f.write(k + '\n')
    f.write('\n')

    # 写入模型字典的value
    f.write('value:\n')
    for k in model:
        # 将键和值转换为字符串，并写入文件
        # 使用repr函数可以获取值的准确表示，适合于复杂的数据结构
        f.write(k + ': ' + repr(model[k]) + '\n')
运行生成txt文件