这个屌丝很懒，什么也没留下！

Expected more than 1 value per channel when training, got input size torch.Size

作者：知新_RL | 2024-07-15 06:13:25

expected more than 1 value per channel when training

训练的时候，batch size必须大于1

但是预测的时候，batch size可以等于1

1. 加大batch_size

2.网络设置eval模式：


model=MyModel()
model.eval()

2.网上查找的原因为模型中用了batchnomolization，训练中用batch训练的时候当前batch恰好只含一个sample，而由于BatchNorm操作需要多于一个数据计算平均值，因此造成该错误。

声明：本文内容由网友自发贡献，转载请注明出处：【wpsshop】