ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 8,..

1.问题描述

报错：ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 8, 1, 1])
大致意思是，由模型中BN层属性决定，训练完train样本后，生成的模型model要用来测试样本。在model(test)之前，需要加上model.eval()，否则的话，有输入数据，即使不训练，它也会改变权值。

2.解决办法

在测试时model.train()之前加上model.eval()