The batch norm layer is used after linear layers (ie: FC, conv), and before the non-linear layers (relu).
There is actually 2 batch norm implementations one for FC layer and the other for conv layers (Spatial batch-norm).
The batch norm layer is used after linear layers (ie: FC, conv), and before the non-linear layers (relu).
There is actually 2 batch norm implementations one for FC layer and the other for conv layers (Spatial batch-norm).