Hi,
can you explain again in more detail what you want to do? So you replaced the last layer to get the desired number of out_features and now you want to replace the batchnorm weights/biases, but not the running statistics (running_mean, running_var), with your own or what?
Greetings.