RuntimeError:mat1 和 mat2 形状不能相乘(25×340 和 360×1)
pytorch 1188
原文标题 :RuntimeError: mat1 and mat2 shapes cannot be multiplied (25×340 and 360×1)
我收到此错误消息,但不知道为什么。我的输入是来自表格数据的 (batch, 1, 312),这个 CNN 是为回归预测而构建的。我使用公式 (输入 + 2*padding – filter size)/stride + 1 如下评论所示。问题出现在 x = self.fc(x) 处,我不知道为什么。非常感谢您的帮助。谢谢。
class CNNWeather(nn.Module):
# input (batch, 1, 312)
def __init__(self):
super(CNNWeather, self).__init__()
self.conv1 = nn.Conv1d(in_channels=1, out_channels=8, kernel_size=9, stride=1, padding='valid') # (312+2*0-9)/1 + 1 = 304
self.pool1 = nn.AvgPool1d(kernel_size=2, stride=2) # 304/2 = 302
self.conv2 = nn.Conv1d(in_channels=8, out_channels=12, kernel_size=3, stride=1, padding='valid') # (302-3)/1+1 = 300
self.pool2 = nn.AvgPool1d(kernel_size=2, stride=2) # 300/2 = 150
self.conv3 = nn.Conv1d(in_channels=12, out_channels=16, kernel_size=3, stride=1, padding='valid') # (150-3)/1+1 = 76
self.pool3 = nn.AvgPool1d(kernel_size=2, stride=2) # 76/2 = 38
self.conv4 = nn.Conv1d(in_channels=16, out_channels=20, kernel_size=3, stride=1, padding='valid') # (38-3)/1+1 = 36
self.pool4 = nn.AvgPool1d(kernel_size=2, stride=2) # 36/2 = 18 (batch, 20, 18)
self.fc = nn.Linear(in_features=20*18, out_features=1)
def forward(self, x):
x = self.pool1(F.relu(self.conv1(x)))
x = self.pool2(F.relu(self.conv2(x)))
x = self.pool3(F.relu(self.conv3(x)))
x = self.pool4(F.relu(self.conv4(x)))
print(x.size())
x = x.view(x.size(0), -1) # flatten (batch, 20*18)
x = self.fc(x)
return x
回复
我来回复-
aaossa 评论
该回答已被采纳!
问题似乎与您的 FC 层的输入大小有关:
self.fc = nn.Linear(in_features=20*18, out_features=1)
上一层的输出是 340,所以必须使用 in_features=340 。
这些是第三层和第四层的输出形状。
torch.Size([5, 16, 73]) conv3 out torch.Size([5, 16, 36]) pool3 out torch.Size([5, 20, 34]) conv4 out torch.Size([5, 20, 17]) pool4 out
请注意,“pool4”层中有 20×17,即 340 个元素。
1年前