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的了,使用池化层就出错了
RuntimeError: Given input size: (128x1x1). Calculated output size: (128x0x0). Output size is too small
问题原因:
在池化层之前,我的图像就变成1x1的了,使用池化层就出错了