check.py

import torch

# 加载模型参数
model_path = '/home/aiops/wangzh/zss/AlphaCLIP/train/final-negative-large-wiseconv/ckpt/iter_10000.pth'  # 你的模型路径
checkpoint = torch.load(model_path)

# 输出 checkpoint 内容
# print("Checkpoint content:", checkpoint)
import pdb;pdb.set_trace()
# 如果 checkpoint 是一个包含模型参数的字典（例如state_dict）
if isinstance(checkpoint, dict):
    for key, value in checkpoint.items():
        print(f"{key}: {value.shape}")
        if 'cpe' in key:
            print(value)
else:
    print("The checkpoint does not contain a dictionary with parameters.")