关于BN层

[war0927]

请问当我应用iaff模块时候存在以下情况：1、F的batch_size为1；2、全局平均池化将特征图的面积变为1*1。这导致了经过BN层的时候会报错ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 256, 1, 1])，请问这个有处理方法吗

[Niche180]

同样地问题，这样设置是否合理呢？

[liyiersan]

batch size设置为1，使用BN层没有意义，把Batch size改为大于1即可。