Expected more than 1 value per channel when training, got input size torch.Size
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操作
·
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()
华为开发者空间，是为全球开发者打造的专属开发空间，汇聚了华为优质开发资源及工具，致力于让每一位开发者拥有一台云主机，基于华为根生态开发、创新。
更多推荐
6
0- 0
已为社区贡献6条内容
所有评论(0)