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

RuntimeError: mat1 dim 1 must match mat2 dim 0
这个错误的大概意思是:矩阵mat1 的第二维度要与mat2的第一维度不匹配

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

解决方法:

先看报错信息,确定报错位置
RuntimeError: mat1 dim 1 must match mat2 dim 0 解决方法_第1张图片
我的这个代码是utils.py文件的104x=self.fc1(x)这个代码出现了问题,查找之后发现fc1是个全连接层
RuntimeError: mat1 dim 1 must match mat2 dim 0 解决方法_第2张图片
咱们在104报错的位置打上断点,用debug进行调试
RuntimeError: mat1 dim 1 must match mat2 dim 0 解决方法_第3张图片
下面是调试后显示的变量情况
RuntimeError: mat1 dim 1 must match mat2 dim 0 解决方法_第4张图片
展开x,将shape里的2304填入之前的fc1全连接层中,
RuntimeError: mat1 dim 1 must match mat2 dim 0 解决方法_第5张图片RuntimeError: mat1 dim 1 must match mat2 dim 0 解决方法_第6张图片
问题解决

你可能感兴趣的:(日常记录,python,深度学习,开发语言)