code: def __init__(
      self,
      model,
      bn_lambda,
      last_epoch=-1,
      setter=set_bn_momentum_default,
  ):
      if not isinstance(model, nn.Module):
          raise RuntimeError(
              "Class '{}' is not a PyTorch nn Module".format(
                  type(model).__name__
              )
          )
  
      self.model = model
      self.setter = setter
      self.lmbd = bn_lambda
  
      self.step(last_epoch + 1)
      self.last_epoch = last_epoch
  
imports: import numpy
  import matplotlib.pyplot
  from mpl_toolkits.mplot3d import Axes3D
  import random
  import torch
  import torch.nn
  import torch.nn.functional
  import os
  from collections import abc
label_desc: Use image node_id 1 to create a new BNMomentumScheduler object from inherited base classes: object with example: obj = BNMomentumScheduler(model, bn_lambda, last_epoch, setter)