Pytorch torchvision MNIST 下载

Question

我是 Pytorch 和 torchvision 的新手。我遵循了大约一年前的教程，他试图通过 python 和 torchvision 下载 mnist。

这是如何：

import torch

from torchvision import datasets, transforms

kwargs = {'num_workers': 1, 'pin_memory': True}

train = torch.utils.data.DataLoader(

    datasets.MNIST('data', train=True, download=True,

                   transform=transforms.Compose([transforms.ToTensor(),

                   transforms.Normalize((0.1307,), (0.3081,))])),

    batch_size=64, shuffle=True, **kwargs)

test = torch.utils.data.DataLoader(

    datasets.MNIST('data', train=False,

                   transform=transforms.Compose([transforms.ToTensor(),

                   transforms.Normalize((0.1307,), (0.3081,))])),

    batch_size=64, shuffle=True, **kwargs)

现在我的问题是我收到此错误：

回溯（最近一次调用最后一次）：

下载http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz到 data\MNIST\raw\train-images-idx3-ubyte.gz

文件“C:/Users/Nico/PycharmProjects/PyTorch/mnist.py”，第 13 行，在 transforms.Normalize((0.1307,), (0.3081,))]))，

文件“C:\Users\Nico\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\datasets\mnist.py”，第 68 行，在init self.download()

文件“C:\Users\Nico\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\datasets\mnist.py”，第 143 行，下载中的 download_url(url, root=self.raw_folder, filename = 文件名，md5=无）

文件“C:\Users\Nico\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\datasets\utils.py”，第 73 行，在 download_url reporthook=gen_bar_updater(tqdm())

TypeError: init () 缺少 1 个必需的位置参数：'total'

你们中的任何人都知道我必须改变什么，或者我如何下载/使用它们？正如我之前所说，我是新手，我没有任何线索。

我希望你们能帮助我，在此先感谢。

Answer 1

所以问题不在于代码或命名或任何东西。这是torchvision 版本。我有 0.2.2.post2 并且它与 0.2.1 一起工作！