code: def _set_momenta(module, momenta):
      if issubclass(
          module.__class__, torch.nn.modules.batchnorm._BatchNorm
      ):
          module.momentum = momenta[module]
  
imports: import warnings
  from collections import defaultdict
  from typing import Optional, Union
  import torch
  from pytext.config.component import create_optimizer
  from pytext.optimizer.lamb import Lamb
  from pytext.optimizer.madgrad import MADGRAD
  from pytext.optimizer.optimizers import Adagrad, Adam, AdamW, Optimizer, SGD
  from pytext.optimizer.radam import RAdam
  from torch.optim import Optimizer
label_desc: Use image node_id 5 for calling a global function with example usage: _set_momenta(module, momenta) without return types