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

华墨1024

8268人浏览 · 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社区

更多推荐

为什么你的手机里藏着一颗“偏科”的芯片？——NPU 芯片科普

NPU是专为神经网络推理设计的芯片，通过脉动阵列架构和低精度计算，实现高能效、低延迟的AI运算。它与CPU、GPU分工明确：前者负责调度，后者承担训练与通用并行，而NPU则在手机、PC、汽车及可穿戴设备中实现本地化AI体验。2026年，算力虽飙升至百TOPS，但瓶颈已从算力转向内存带宽。存算一体、3D堆叠与芯粒集成成为突破方向，未来芯片价值不仅看TOPS，更取决于数据搬运效率与场景适配能力。