RuntimeError: mat1 dim 1 must match mat2 dim 0
这个错误的大概意思是:矩阵mat1 的第二维度要与mat2的第一维度不匹配
在新增别的数据集进行训练时报当前错误,原因是输入的图像大小与之前不一样,这是新手在学习时常会遇到的问题。文章来源:https://www.toymoban.com/news/detail-612187.html
解决方法:
先看报错信息,确定报错位置
我的这个代码是utils.py
文件的104
行 x=self.fc1(x)
这个代码出现了问题,查找之后发现fc1
是个全连接层
咱们在104报错的位置打上断点,用debug进行调试
下面是调试后显示的变量情况
展开x
,将shape里的2304
填入之前的fc1
全连接层中,
问题解决文章来源地址https://www.toymoban.com/news/detail-612187.html
到了这里,关于RuntimeError: mat1 dim 1 must match mat2 dim 0 解决方法的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!