问题:
pytorch用torchvision下载数据集时速度极慢,而且大概率下载失败,就很恶心人,心态炸裂,故写本文提出解决此问题的方法。
原因:
下载速度极慢主要原因是pytorch下载数据集的网站如下图所示,属于国外网站,访问受限且极不稳定,解决此问题的最好方法就是自己把mnist等数据集下载到本地,然后把download参数改为False,但此时仍然会报错!原因是没有在./MNIST/processed/目录下生成(.pt)文件。
torchvision加载mnist数据集的代码如下:
train_dataset = datasets.MNIST(
root=r'./mnist', #此处为自己定义的下载存在目录
train=