pytorch报错:ValueError: Expected more than 1 value per channel when training
pytorch报错:ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,512,1,1])2.网上查找下原因为模型中用了batchnomolization,训练中用batch训练的时候当前batch恰好只含一个sample,训练中用batch训练的时候,应该...
原创
2019-12-26 11:04:09 ·
1723 阅读 ·
4 评论