2019年12月7日
RuntimeError: size mismatch, m1: [32 x 2304], m2: [400 x 100] at C:/w/1/s/windows/pytorch/aten/src\THC/generic/THCTensorMathBlas.cu:268
摘要： 这个错误一般是卷积和全连接直接对接的时候，大小不匹配导致的。 可以查看卷积结束后 图片大小 ，然后对应修改全连接的输入大小即可。 def forward(self,x): out = self.conv(x) out = out.view(out.size(0),-1) # print("-->{} 阅读全文
posted @ 2019-12-07 16:44 闪存第一菜鸡 阅读(5768) 评论(0) 推荐(0)