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。广州

文件“C:/Users/Nico/PycharmProjects/PyTorch/mnist.py”，第13行，在转换中。正常化（（0.1307，），（0.3081，）））），

文件“C:\Users\Nico\AppData\Local\Programs\Python\Python37\lib\site packages\torchvision\datasets\mnist.py”，第68行，在init self中。下载（）

文件"C：\用户\Nico\AppData\本地\程序\Python\Python37\lib\site-pack\torchvision\datasets\mnist.py"，第143行，在下载download_url（url，root=self.raw_folder，filename=filename，md5=无）

文件“C:\Users\Nico\AppData\Local\Programs\Python\Python37\lib\site packages\torchvision\dataset\utils.py”，第73行，位于下载\u url reporthook=gen\u bar\u更新程序（tqdm（）中）

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

你们中有谁知道我需要改变什么，或者我如何下载/使用它们？正如我之前所说的，我对它还不熟悉，而且我没有任何线索。

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

你好尼科·阿卡。Myridor

Answer 1

所以问题不在于代码、命名或其他什么。

这是火炬视觉版本。我有0.2.2.post2，它与0.2.1工作！