python 读取pkl和pt模型文件

 '''*************************读取pkl文件****************************'''
import pickle
path = 'train.pkl'
f = open(path,'rb')
data=pickle.load(f)
print(data)
print(len(data))

 '''*************************读取pt文件****************************'''
import torch
cat_dict = torch.load('pretrain1.pt')
for k, v in cat_dict.items():  # k 参数名 v 对应参数值
    print(k, v)