Solved by smth in post #2 
              
              
                batchnorm weight, batchnorm bias, batchnorm running_mean, batchnorm running_std 
(batchnorm has an affine transform by default)