加载numpy对象时Pickle数据被截断

b5lpy0ml  于 2023-04-06  发布在  其他
关注(0)|答案(1)|浏览(140)

当我尝试上传像这样的大型numpy对象时,在Google Collab上:
data = np.load('sample_data/train_data.npy', allow_pickle = True)
我得到一个错误:
UnpicklingError: pickle data was truncated
当我尝试在本地系统上执行此操作时,没有出现这样的错误。我已经在Collab notebook中上传了train_data.npy。此外,本网站上与相同错误相关的其他答案似乎不适用于我的特定情况。

xmjla07d

xmjla07d1#

我不知道是什么原因导致了这个错误,但是在Google Drive上上传numpy对象并通过Drive访问它解决了这个问题。
通过运行以下代码给予访问驱动器:

from google.colab import drive
drive.mount('/content/drive')

并按如下方式加载数据:
data = np.load('drive/MyDrive/train_data.npy', allow_pickle = True)

相关问题