Updating running_mean and running_var in a custom Batchnorm?

Answered in this topic.