Pytorch model with multi batchnorm1d layers get errors during inference - "expected 2D or 3D in...

Batchnorm2d Pytorch - Why pass number of channels to batchnorm?...

error in BatchNorm2d in pytorch CNN model...