0

我是 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, 文件名=文件名，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'

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

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

问候尼科又名。万里多

4

1 回答 1

0

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

这是torchvision版本。我有 0.2.2.post2，它适用于 0.2.1！

于 2019-03-01T19:05:52.733 回答