不必改源代码!不必改源代码!不必改源代码!
(看到许多教程说改源代码但是何必呢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(),
)
运行结果
代码运行输出:
文件夹变化:
表明文件已经成功加载解析!
芜湖!