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

华墨1024

8256人浏览 · 2023-01-31 12:44:57

华墨1024 · 2023-01-31 12:44:57 发布

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全连接层中，
在这里插入图片描述
问题解决

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

ops-transformer 是什么：五句话让一个完全不懂的人听明白

我有个朋友是做后端的老程序员，最近想转大模型训练方向，跟我说想了解一下昇腾 NPU 的算子生态。他对 PyTorch 熟悉，但没接触过 CANN，问了我一个问题：“ops-transformer 这个仓库到底解决了什么问题？我给他讲了大概二十分钟，最后他跟我说："你能不能用五句话概括？"我试了一下，发现做不到——因为这个仓库解决的不是一个问题，而是串联起了一整条链路上的多个问题。但我可以换一种方式