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
Изменял входную форму, все равно ошибка