code: def _reset_bn(module):
      if issubclass(
          module.__class__, torch.nn.modules.batchnorm._BatchNorm
      ):
          module.running_mean = torch.zeros_like(module.running_mean)
          module.running_var = torch.ones_like(module.running_var)
  
imports: from collections import defaultdict
  from torch.optim.optimizer import Optimizer
  import torch
  import warnings
label_desc: Use image node_id 3 for calling a global function with example usage: _reset_bn(module) without return types