RuntimeError: mat1 dim 1 must match mat2 dim 0 解决方法

• Python

RuntimeError: mat1 dim 1 must match mat2 dim 0

这个错误的大概意思是：矩阵mat1 的第二维度要与mat2的第一维度不匹配

在新增别的数据集进行训练时报当前错误，原因是输入的图像大小与之前不一样，这是新手在学习时常会遇到的问题。

解决方法：

先看报错信息，确定报错位置

在这里插入图片描述

我的这个代码是utils.py文件的104行 x=self.fc1(x)这个代码出现了问题，查找之后发现fc1是个全连接层

在这里插入图片描述

咱们在104报错的位置打上断点，用debug进行调试

在这里插入图片描述

下面是调试后显示的变量情况

在这里插入图片描述

展开x，将shape里的2304填入之前的fc1全连接层中，

在这里插入图片描述在这里插入图片描述

问题解决

本文来自网络，不代表协通编程立场，如若转载，请注明出处：https://www.net2asp.com/58dbb07a85.html

赞 (0)
« 上一篇 1天前
下一篇 » 1天前

相关推荐