日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后,達者為師

網站首頁 編程語言 正文

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

作者:甘霖佳佳 更新時間: 2022-01-31 編程語言

在這里插入圖片描述
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層。

修改成功,不報錯了。

原文鏈接:https://blog.csdn.net/weixin_45928096/article/details/122387754

欄目分類
最近更新