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()