Pytorch火炬视觉MNIST下载

时间:2019-03-01 18:35:10

标签: python pytorch mnist torchvision

我是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,filename = filename,md5 = None)

     

文件“ 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”

你们中有人知道我必须更改什么,或者我如何下载/使用它们吗?如前所述,我是新手,没有任何线索。

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

问候Nico又名。迈里德(Myridor)

1 个答案:

答案 0 :(得分:0)

所以问题不是代码,命名或其他任何问题。

这是火炬手的版本。我有0.2.2.post2,它与0.2.1兼容!