不必改源代码!不必改源代码!不必改源代码!
(看到许多教程说改源代码但是何必呢QAQ)
报错: Error downloading~~
原因: 程序发现数据集没下载=》于是去目标网站下载数据集=》结果由于网速/墙下载不下来=》报错。
因此, 手动下载数据集放到目标文件夹,程序检测到已经下载好,就不会再去联网下载了。
以MNIST数据集为例
如下载MNIST,需要新建MNIST文件夹,再在该文件夹下新建raw文件夹。
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(),
)