code: def _set_momenta(module, momenta):
      if issubclass(
          module.__class__, torch.nn.modules.batchnorm._BatchNorm
      ):
          module.momentum = momenta[module]
  
imports: from collections import defaultdict
  from torch.optim.optimizer import Optimizer
  import torch
  import warnings
label_desc: Use image node_id 5 for calling a global function with example usage: _set_momenta(module, momenta) without return types