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)

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

Traceback（最近一次调用最后一次）：

将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行，in transforms.Normalize（（0.1307，），（0.3081，））]）），

在init self.download（）中的文件“C：\ Users \ Nico \ AppData \ Local \ Programs \ Python \ Python37 \ lib \ site-packages \ torchvision \ datasets \ mnist.py”，第68行

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

在download_url reporthook = gen_bar_updater（tqdm（））中输入文件“C：\ Users \ Nico \ AppData \ Local \ Programs \ Python \ Python37 \ lib \ site-packages \ torchvision \ datasets \ utils.py”，第73行

TypeError：init（）缺少1个必需的位置参数：'total'

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

我希望你们能帮助我，谢谢你们。

问候Nico aka。 Myridor

Answer 1

所以问题不在于代码或命名或任何东西。

这是火炬的版本。我有0.2.2.post2，它与0.2.1一起工作！