错误提示:
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的要求。