ПИТОРЧ ФАКЕЛ ВИДЕНИЕ МНИСТ скачать

Question

Я новичок в 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)

Теперь моя проблема в том, что я получаю эту ошибку:

Traceback (последний вызов был последним):

Загрузка http://yann.lecun.com/exdb/mnist/train-images-idx3-ubyte.gz в данные \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 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())

Ошибка типа: init() отсутствует 1 обязательный позиционный аргумент: "всего"

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

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

Привет Нико ака. Myridor

Answer 1

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

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