（已解决torchvision中CIFAR10下载速度慢）如何下载以及使用torchvision导入

2024-07-17 19:52| 来源: 网络整理| 查看: 265

背景 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)

成功。

【本文地址】

今日新闻