Pytorch torchvision MNIST скачать

Я новичок в Pytorch и torchvision. Я следил за учебником примерно годовой давности, в котором он пытался загрузить mnist через python и torchvision.

Вот как:

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 = имя файла, 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'

Кто-нибудь из вас знает, что мне нужно изменить, или как я могу их скачать / использовать? Как я сказал ранее, я новичок в этом и понятия не имею.

Надеюсь, вы можете мне помочь, заранее спасибо.

Привет Нико ака. Миридор


python pytorch torchvision mnist
person Myridor    schedule 01.03.2019    source источник
comment
Прекрасно работает на PyTorch 1.0.0.   -  person kHarshit    schedule 01.03.2019
comment
Я получил PyTorch 1.0.1, попробую с 1.0.0   -  person Myridor    schedule 01.03.2019
comment
Какая у вас версия torchvision? Получил 0.2.2.post2   -  person Myridor    schedule 01.03.2019
comment
torchvision - 0,2,1   -  person kHarshit    schedule 01.03.2019

Ответы (1)


arrow_upward
0
arrow_downward

Так что проблема заключалась не в коде, названии или чем-то еще.

Это была версия torchvision. У меня был 0.2.2.post2, и он работал с 0.2.1!

person Myridor    schedule 01.03.2019
comment
Я думаю, это должно было работать и на 0.2.2. Вы можете поднять проблему. - person kHarshit; 01.03.2019