code: def reset_bn(module):
      classname = module.__class__.__name__
      if (
          classname.find("BatchNorm") != -1
          or classname.find("InPlaceABNSync") != -1
      ):
          module.running_mean = torch.zeros_like(module.running_mean)
          module.running_var = torch.ones_like(module.running_var)
  
file_path: insightface/parsing/dml_csr/utils/schp.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import torch
label_desc: Use image node_id 4 for calling a global function with example usage: reset_bn(module) without return types