RuntimeError: size mismatch, m1: [80 x 4], m2: [320 x 50] at ..\aten\src\TH/generic/THTensorMath.cpp

最新推荐文章于 2023-02-25 20:39:40 发布

我是一颗棒棒糖

最新推荐文章于 2023-02-25 20:39:40 发布

阅读量1.5k

点赞数 1

分类专栏： DeepLearning学习文章标签：人工智能深度学习机器学习

本文链接：https://blog.csdn.net/qq_42255269/article/details/108298702

版权

DeepLearning学习专栏收录该内容

77 篇文章 19 订阅

订阅专栏

RuntimeError: size mismatch, m1: [80 x 4], m2: [320 x 50] at …\aten\src\TH/generic/THTensorMath.cpp:41

使用pytorch进行深度学习的训练会出现这种问题，原因是fc全连接层的输入维度问题，由于输入是二维的数据，很多时候在输入全连接层的时候我们没有调整维度
例如：

def forward(self, input):
        x = self.pool1(F.relu(self.conv1(input)))
        x = self.pool2(F.relu(self.conv2(x)))
        x = self.fc1(x)
        x = self.fc2(x)
        return x

我们的fc1的输入将是804的情况，而需要的是3201的情况，这时会报错

改为：

def forward(self, input):
        x = self.pool1(F.relu(self.conv1(input)))
        x = self.pool2(F.relu(self.conv2(x))).view(320)
        x = self.fc1(x)
        x = self.fc2(x)
        return x

使用view调整维度即可