mat1 and mat2 shapes cannot be multiplied (128x432 and 576x64)的解决

仅作为记录,大佬请跳过。

今天博主在跑大佬博主的pytorch cnn分类mnist数据集的程序——大佬博主文章(第四个网络)

出现问题:

mat1 and mat2 shapes cannot be multiplied (128x432 and 576x64)的解决_第1张图片

经过尝试,发现是在定义网络中的问题

mat1 and mat2 shapes cannot be multiplied (128x432 and 576x64)的解决_第2张图片
将x = x.view(128,-1)改为x = x.view(-1,64*3*3)即可。

最后,再次感谢强大的大佬博主

你可能感兴趣的:(python)