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

1.初始網(wǎng)絡(luò)模型

原始網(wǎng)絡(luò)模型對于cifar10使用了六層卷積，且每層卷積后接一層池化。
池化大小為（2*2，stride=2）。

2.報(bào)錯(cuò)原因

圖片尺寸太小（初始尺寸32323，而32=2^5，經(jīng)過五層池化后尺寸就變?yōu)?），進(jìn)入網(wǎng)絡(luò)后池化不夠大。

3.解決方法

①修改pooling層，步長變?yōu)?。
②減少pooling層，兩層卷積后一層池化，池化一共3層。

修改成功，不報(bào)錯(cuò)了。