背景

    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)

成功。

Logo

讨论HarmonyOS开发技术,专注于API与组件、DevEco Studio、测试、元服务和应用上架分发等。

更多推荐