无法在Google Colab上使用torch.load()加载.pth文件(预训练的神经网络)

诺埃·阿切奇

我的Google驱动器已链接到我的Google Colab笔记本。使用pytorch库torch.load($ PATH)无法加载此219 Mo文件(预训练的神经网络)(https://drive.google.com/drive/folders/1-9m4aVg8Hze0IsZRyxvm5gLybuRLJHv-)驾驶。但是,当我在计算机上本地执行此操作时,它工作正常。我在Google Collab上遇到的错误是:(设置:Python 3.6,pytorch 1.3.1):

state_dict = torch.load(model_path)['state_dict']
File "/usr/local/lib/python3.6/dist-packages/torch/serialization.py", line 303, in load
return _load(f, map_location, pickle_module)
File "/usr/local/lib/python3.6/dist-packages/torch/serialization.py", line 454, in _load
return legacy_load(f)
File "/usr/local/lib/python3.6/dist-packages/torch/serialization.py", line 380, in legacy_load
with closing(tarfile.open(fileobj=f, mode='r:', format=tarfile.PAX_FORMAT)) as tar,
File "/usr/lib/python3.6/tarfile.py", line 1589, in open
return func(name, filemode, fileobj, **kwargs)
File "/usr/lib/python3.6/tarfile.py", line 1619, in taropen
return cls(name, mode, fileobj, **kwargs)
File "/usr/lib/python3.6/tarfile.py", line 1482, in init
self.firstmember = self.next()
File "/usr/lib/python3.6/tarfile.py", line 2297, in next
tarinfo = self.tarinfo.fromtarfile(self)
File "/usr/lib/python3.6/tarfile.py", line 1092, in fromtarfile
buf = tarfile.fileobj.read(BLOCKSIZE)
OSError: [Errno 5] Input/output error```   




Any help would be much appreciated!
诺埃·阿切奇

通过将文件直接上传到google colab而不是使用以下方法从google驱动器加载文件来工作:

from google.colab import files
uploaded= files.upload()

我猜这个解决方案类似于@Yuri提出的解决方案

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

从任何平台加载和使用预训练的神经网络

步数未知-在Google Colab Pro上训练卷积神经网络

如何使用 Torch Vision 在 Google Colab 上加载 CelebA 数据集,而不会耗尽内存?

OpenCV 3.0无法加载神经网络

Python神经网络无法训练

无法正确训练神经网络

如何使用带有灰度图像的预训练神经网络?

如何使用预训练模型为图像分类优化神经网络

使用预训练卷积神经网络了解特征提取

在 Google Colab CPU 中训练神经网络 - 第二个纪元没有开始

使用我的电脑中的训练和评估文件在谷歌云中训练神经网络

在预训练嵌入的基础上构建用于二元分类的神经网络不起作用?

加载不同大小的图像以训练卷积神经网络

torch7神经网络训练错误

无法加载数据 Google Colab

无法训练神经网络解决异或映射

具有隐藏层的神经网络无法训练

在keras中的CPU上并行训练多个神经网络

在Brain.js神经网络模型上恢复训练

在表类型的数据上训练神经网络

使用Pytorch训练神经网络时了解类的使用

Torch / Lua,如何将训练有素的神经网络模型保存到文件中?

如果〜/ .bashrc文件不存在,该如何在Google Colab上安装Torch?

如何使用张量流数据集训练神经网络?

如何使用 Q-Learning 训练神经网络

使用TensorFlow加速GPU训练循环神经网络

使用TensorFlow训练神经网络时出错

在FANN中使用验证数据集训练神经网络

使用单个优化器训练多个连接的神经网络