maybe you can just subclass this:
https://pytorch.org/docs/master/_modules/torch/nn/modules/batchnorm.html 185
and override the forward function setting self.running_var.fill_(1) before calling F.batch_norm(blah blah)