RuntimeError: фигура '[100, 1, 28, 28]' недопустима для ввода размером 62720

Question

PyTorch

Я пытаюсь создать CNN на своем собственном наборе данных, который имеет 39 различных классов.

Мне грозит одна ошибка выполнения

Ошибка

---

RuntimeError                              Traceback (most recent call last)
<ipython-input-42-67f1329ebfd8> in <module>()
     16         images, labels = images.to(device), labels.to(device)
     17 
---> 18         train = Variable(images.view(100, 1, 28, 28))
     19         labels = Variable(labels)
     20 

RuntimeError: shape '[100, 1, 28, 28]' is invalid for input of size 62720

RuntimeError: фигура '[100, 1, 28, 28]' недопустима для ввода размером 62720.

Answer 1

Если ваши изображения имеют размер 28x28, то, вероятно, размер пакета равен 80 вместо 100.

train = Variable(images.view(80, 1, 28, 28))