Expected more than 1 value per channel when training, got input size torch.Size
训练的时候，batch size必须大于1，但是预测的时候，batch size可以等于1
原因：
因为模型中用了batchnomolization，训练中用batch训练的时候当前batch恰好只含一个sample，而由于BatchNorm操作需要多于一个数据计算平均值，因此造成该错误。
解决方法：
1. 加大batch_size
2.网络设置eval模式：
model=MyModel() model.eval()
