问题描述及解决
在读取.pkl
文件的时候,采用下面的方式进行读取,
with open('/xxxxx.pkl', 'rb') as file:
data = pkl.load(file)
data.x
报错如下所示:
通过查看解决办法,得知我保存为.pkl
文件的时候采用的torch方法存储的,所以采用torch.load()
就可以顺利读取了。
如下所示:
# 保存数据的时候采用的下面方式
torch.save(pkl_data, 'xxxxx.pkl')
所以读取的时候用下述方式进行读取:
loaded_data = torch.load('xxxxx.pkl')
整理一下.pkl
文件读取
看Pkl储存的内容使用的什么方式
- 1、使用numpy存储的读取办法
import numpy as np
file_path= 'xxxxx.pkl'
loaded_data = np.load(file_path, allow_pickle=True)
- 2、使用torch存储数据的情况
import torch
loaded_data = torch.load('xxxxx.pkl')
- 正常读取
with open('/home/featurize/work/AS_GAE/pkl/BlogCatalog.pkl', 'rb') as file:
data = pkl.load(file)
这里是参考:
1、(20200720已解决)_pickle.UnpicklingError: A load persistent id instruction was encountered
2、读取pkl文件报错_pickle.UnpicklingError: A load persistent id instruction was encountered