RuntimeError: size mismatch, m1: [10 x 13], m2: [130 x 10] at ..\aten\src\TH/generic/THTensorMath.cpp:41

Написана следующая сетка:

class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.fc1 = nn.Linear(13*10, 10)
        self.fc2 = nn.Linear(10, 2)

    def forward(self,x):
        x = F.relu(self.fc1(x))
        x = self.fc2(x)
        return F.logsigmoid(x)

net = Net()
print(net)

learning_rate = 0.01

optimizer = optim.SGD(net.parameters(), lr=learning_rate, momentum=0.9)
# Создаем функцию потерь
criterion = nn.NLLLoss()

epochs = 5
log_interval = 10

carset = carDataset()
train_loader = torch.utils.data.DataLoader(carset, batch_size = 10, shuffle = True )

# запускаем главный тренировочный цикл
for epoch in range(epochs):
   running_loss = 0
   for features, labels in train_loader:
       output = net(features)

Которая выдает следующую ошибку:

Traceback (most recent call last):
  File "nn.py", line 72, in <module>
    output = net(features)
  File "C:\Users\usr\PycharmProjects\pythonProject\venv\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
    result = self.forward(*input, **kwargs)
  File "nn.py", line 49, in forward
    x = F.relu(self.fc1(x))
  File "C:\Users\usr\PycharmProjects\pythonProject\venv\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
    result = self.forward(*input, **kwargs)
  File "C:\Users\usr\PycharmProjects\pythonProject\venv\lib\site-packages\torch\nn\modules\linear.py", line 91, in forward
    return F.linear(input, self.weight, self.bias)
  File "C:\Users\usr\PycharmProjects\pythonProject\venv\lib\site-packages\torch\nn\functional.py", line 1674, in linear
    ret = torch.addmm(bias, input, weight.t())
RuntimeError: size mismatch, m1: [10 x 13], m2: [130 x 10] at ..\aten\src\TH/generic/THTensorMath.cpp:41

Изменял входную форму, все равно ошибка


Ответы (0 шт):