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

浪矢\n / 2023-07-20 / 原文

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

问题原因:

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