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 (most recent call last):

Downloading http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz to data\MNIST\raw\train-images-idx3-ubyte.gz

File "C:/Users/Nico/PycharmProjects/PyTorch/mnist.py", line 13, in transforms.Normalize((0.1307,), (0.3081,))])),

File "C:\Users\Nico\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\datasets\mnist.py", line 68, in init self.download()

File "C:\Users\Nico\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\datasets\mnist.py", line 143, in download download_url(url, root=self.raw_folder, filename=filename, md5=None)

File "C:\Users\Nico\AppData\Local\Programs\Python\Python37\lib\site-packages\torchvision\datasets\utils.py", line 73, in download_url reporthook=gen_bar_updater(tqdm())

TypeError: init() missing 1 required positional argument: 'total'

你们中有人知道我必须改变什么，或者我如何才能 download/use 他们吗？正如我之前所说，我是新手，我没有任何线索。

希望大家能帮帮我，先谢谢了。

你好，又名 Nico。无数

Answer 1

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

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

Pytorch torchvision MNIST 下载

Pytorch torchvision MNIST download

python

pytorch

mnist

torchvision