train_set = torchvision.datasets.CIFAR10(root='./data', train=True,
download=True, transform=transform)
速度超级慢,慢得直接报错:
urllib.error.URLError: urlopen error [Errno 11001] getaddrinfo failed
官网下载超级快http://www.cs.toronto.edu/~kriz/cifar.html
下载第一个:
解压它,然后目录会是这样的
cifar-10-python/cifar-10-batches-py/*
其中*号就是下面这些文件。
我们只需要将上述文件复制到data目录下:
data/cifar-10-batches-py/*
然后使用torchvision处理,并设置download=False
即可。
train_set = torchvision.datasets.CIFAR10(root='./data', train=True,
download=False, transform=transform)
成功。
更多参考
https://blog.csdn.net/qq_53144843/article/details/122887758