365bet比分网-365bet888-beat365官方网站登录

Pytorch无法下载/读取数据集Error downloading(不必改源代码)

Pytorch无法下载/读取数据集Error downloading(不必改源代码)

不必改源代码!不必改源代码!不必改源代码! (看到许多教程说改源代码但是何必呢QAQ) 报错: Error downloading~~

原因: 程序发现数据集没下载=》于是去目标网站下载数据集=》结果由于网速/墙下载不下来=》报错。 因此, 手动下载数据集放到目标文件夹,程序检测到已经下载好,就不会再去联网下载了。 以MNIST数据集为例

STEP 1:在想要存数据的地方,新建“{数据集名称}/raw/”路径。

如下载MNIST,需要新建MNIST文件夹,再在该文件夹下新建raw文件夹。

STEP 2:去官网手动下载对应数据集,存放在上述路径。

STEP 3:把root参数设置为存数据的地址(注意是数据集名称的父目录),直接加载数据集。

import torch

from torchvision import datasets

# Download training data from datasets.

training_data = datasets.MNIST(

root="D:/classic_dataset",

train=True,

download=True,

transform=ToTensor(),

)

# Download test data from datasets.

test_data = datasets.MNIST(

root="D:/classic_dataset",

train=False,

download=True,

transform=ToTensor(),

)

运行结果

代码运行输出: 文件夹变化: 表明文件已经成功加载解析! 芜湖!

相关推荐