pytorch训练时报input size错误问题

错误提示：

Expected more than 1 value per channel when training, got input size torch.Size([1, 256, 1, 1])

在用pytorch做模型训练的，之前可以正常运行的代码，突然出现这个问题，一开始被吓一跳。从字面意思看，是输入的size不能为1。

通过一番查找发现，是在训练阶段batch_size不能为1，则就需要总的训练图片数量/batch_size不能余1，因此调整一下train batch_size，则可解决。

而在model.eval()阶段，则没有batch_size一定需要为1的要求。