Ошибка несоответствия размера с помощью PyTorch. - PullRequest
Новая
       52

Ошибка несоответствия размера с помощью PyTorch.

0 голосов
/ 23 января 2019

Я изучаю учебник под названием «Глубокое обучение с PyTorch: 60-минутный блиц» на веб-сайте PyTorch.Мои коды такие же, как и у него, но есть ошибка несоответствия размера, как показано ниже.Может ли кто-нибудь сказать мне, почему и как это решить?Спасибо:)

RuntimeError: несоответствие размера, m1: [80 x 5], м2: [400 x 120] вc: \ a \ w \ 1 \ s \ tmp_conda_3.7_110509 \ conda \ conda-bld \ pytorch_1544094576194 \ work \ aten \ src \ th \ generic / THTensorMath.cpp: 940 

import torch.nn as nn
import torch.nn.functional as F  

class Net(nn.Module):
def __init__(self):
    super(Net,self).__init__()
    self.conv1=nn.Conv2d(1,6,5)
    self.conv2=nn.Conv2d(6,16,5)
    self.fc1=nn.Linear(16*5*5,120)
    self.fc2=nn.Linear(120,84)
    self.fc3=nn.Linear(84,10)

def forward(self,x):
    x=F.max_pool2d(F.relu(self.conv1(x)),(2,2))
    x=F.max_pool2d(F.relu(self.conv2(x)),2)
    x.view(-1,self.num_flat_features(x))
    x=F.relu(self.fc1(x))
    x=F.relu(self.fc2(x))
    x=self.fc3(x)
    return x

def num_flat_features(self,x):
    size=x.size()[1:]
    num_features=1
    for s in size:
        num_features*=s
    return num_features

net=Net()
input=torch.randn(1,1,32,32)
out=net(input)
print(out)

1 Ответ

0 голосов
/ 23 января 2019

Извините за беспокойство, я обнаружил ошибку. Я пропустил x= в x=x.view(-1,self.num_flat_features(x)) ...

...