哪里使用BN

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).

你可能感兴趣的:(哪里使用BN)