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

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

问题原因:

  在池化层之前,我的图像就变成1x1的了,使用池化层就出错了

posted @ 2023-07-20 15:28  浪矢-CL  阅读(361)  评论(0编辑  收藏  举报