Pytorch火炬視覺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

在transforms.Normalize（（0.1307，），（0.3081，））]））中的文件“ C：/Users/Nico/PycharmProjects/PyTorch/mnist.py”，第13行

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中） =文件名，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”

你們中有人知道我必須更改什么或如何下載/使用它們嗎？ 正如我之前說的，我是新手，沒有任何線索。

希望大家能幫助我，在此先感謝。

問候尼科又名。 Myridor

Answer 1

因此，問題不是代碼，命名或其他任何問題。

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