How to freeze BN layers while training the rest of network (mean and var wont freeze)

[ptrblck]

The code looks alright, but again I would recommend to use isinstance instead of the name check.
Note that model.eval() will also disable dropout layers (any maybe change the behavior of custom layers), so you might want to call eval() only on the batchnorm layers.