Expected more than 1 value per channel when training 报错,根源是 BatchNorm 训练时单通道样本数不

在 PyTorch 训练带 BatchNorm 的网络时,出现Expected more than 1 value per channel when training, got input size tor