报错:ValueError('Expected more than 1 value per channel when training, got input size {}'.format(size))

错误原因:batch中仅有一个样本,调用batch_norm就会有这个报错:

解决办法:重新计算num_samples和batchsize,保证num_samples%batchsi不是1即可

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐