RuntimeError: Given input size: (256x1x1). Calculated output size: (256x0x0). Output size is too small
1.初始网络模型
原始网络模型对于cifar10使用了六层卷积，且每层卷积后接一层池化。
池化大小为（2*2，stride=2）。
2.报错原因
图片尺寸太小（初始尺寸32323，而32=2^5，经过五层池化后尺寸就变为1），进入网络后池化不够大。
3.解决方法
①修改pooling层，步长变为1。
②减少pooling层，两层卷积后一层池化，池化一共3层。
修改成功，不报错了。
版权声明：本文为weixin_45928096原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。