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("-->{}".format(out.size()))
        out= self.fc(out)
        return out

　　查看大小，然后修改自己的模型全連接的輸入既可