Может кто-нибудь помочь мне решить эту проблему, искать много решений, но они не работают для меня

Question

ошибка al entrenar размер изображения ([64, 3, 224, 224]) Я пытался изменить размер пакета или размер изображения, но все равно получаю ошибки

Epoch 1/30
----------
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-160-dbcdb17ea6ee> in <module>()
      1 epochs = 30
      2 net.to(device)
----> 3 net = train_model(net, criterion, optimizer, sched, epochs)

2 frames
<ipython-input-157-d34ea1683b12> in forward(self, x)
     12     x = self.pool(F.relu(self.conv1(x)))
     13     x = self.pool(F.relu(self.conv2(x)))
---> 14     x = x.view(x.size(0), 16 * 38 * 38)
     15     x = F.relu(self.fc1(x))
     16     x = F.relu(self.fc2(x))

RuntimeError: shape '[64, 23104]' is invalid for input of size 2876416

Answer 1

Это потому, что произведение ваших пространственных и канальных измерений не эквивалентно 23104, а скорее равно 2876416. Чтобы сгладить ваш тензор, вы можете вместо этого попробовать out = out.view(out.size(0), -1), что должно работать нормально.