code: def set_bn_param(self, momentum, eps):
      for m in self.modules():
          if isinstance(m, nn.BatchNorm2d) or isinstance(
              m, nn.BatchNorm1d
          ):
              m.momentum = momentum
              m.eps = eps
      return
  
file_path: nni/examples/nas/legacy/oneshot/proxylessnas/model.py
imports: import torch
  import nni.retiarii.nn.pytorch
  import math
  import ops
  import putils
  from nni.retiarii.nn.pytorch import LayerChoice
label_desc: Use image node_id 3 for calling the SearchMobileNet obj's underlying member method code with example usage: obj.set_bn_param(momentum, eps) without return types