pytorch 训练 报错 输入大小（128x1x1） 输出太小

pytorch训练 cifar 10 练习 不知道这个报错为什么
报错为 RuntimeError: Given input size: (128x1x1). Calculated output size: (128x0x0). Output size is too small