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训练的时候,应该是
相关文章
相关标签/搜索