【error】RuntimeError: size mismatch与全连接fc层

今天跑代码的时候遇到了这个错误：
RuntimeError: size mismatch, m1:[1152 x 1] ,m2:[576 x 128] ,at /opt/conda/conda-bld/pytorch_1524

原因分析:这里是最后一层卷积层铺平后(conv_out.view(conv_out.size(0), -1))的shape与全连接层的输入不匹配导致的,因此要将维度对齐.

参考文献:

1.https://blog.csdn.net/u012422524/article/details/88573687

2.https://blog.csdn.net/lingzhou33/article/details/88312502