Your line of code would re-initialize the batchnorm layer, which seems to be wrong, as you’ll lose all trained parameters and estimated stats.
So no, don’t use this line of code, just freeze the parameters using their requires_grad flag.