MNIST объект не имеет данных атрибута - PullRequest
Новая
       29

MNIST объект не имеет данных атрибута

0 голосов
/ 08 июня 2019

Я пытаюсь запустить фрагмент кода, приведенный ниже, но, к сожалению, я получаю следующую ошибку: «Объект MNIST не имеет данных атрибута».Ошибка из-за строки "mnist_train_set.data.view (-1, 1, 28, 28) .float ()".Может кто-нибудь пролить свет на то, как это исправить?Благодарю.

import torch
from torchvision import datasets

...

mnist_train_set = datasets.MNIST(data_dir + '/mnist/', train = True, download = True)
mnist_test_set = datasets.MNIST(data_dir + '/mnist/', train = False, download = True)

train_input = mnist_train_set.data.view(-1, 1, 28, 28).float()
train_target = mnist_train_set.targets
test_input = mnist_test_set.data.view(-1, 1, 28, 28).float()
test_target = mnist_test_set.targets
...