【如何查看checkpoint.pth文件中的key名】

一般pytorch保存的checkpoint.pth文件内部都是字典的形式，通常在加载模型的权重的时候会用到文件中的键的名称，而每一个模型创建key的名称都不一样，所以在使用pytorch框架中的load_stat_dict加载权重之前，最好查看一下pth文件中的键的名称，代码如下。

load_state_dict（ckpt['key']) #加载模型权重

import torch

# 加载 Checkpoint 文件
ckpt = torch.load('Path to/ckpt.pth')

# 打印字典的键值对
for key, value in ckpt.items():
    print(key)