
这个错误一般是卷积和全连接直接对接的时候,大小不匹配导致的。
可以查看卷积结束后 图片大小 ,然后对应修改全连接的输入大小即可。
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
查看大小,然后修改自己的模型全连接的输入既可