RuntimeError:Given input size:(256x1x1). Calculated output size: (256x0x0).Output size is too small

RuntimeError:Given input size:(256x1x1). Calculated output size: (256x0x0).Output size is too small_第1张图片
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层。

修改成功,不报错了。

你可能感兴趣的:(python项目,#记录报错,深度学习,计算机视觉,神经网络)