code: def __init__(
      self,
      lr,
      params=None,
      momentum=0,
      dampening=0,
      weight_decay=0,
      nesterov=False,
  ):
      TorchOptimizer.__init__(self)
      self.param_dict["lr"] = lr
      self.param_dict["momentum"] = momentum
      self.param_dict["dampening"] = dampening
      self.param_dict["weight_decay"] = weight_decay
      self.param_dict["nesterov"] = nesterov
      self.torch_class = type(self).__bases__[0]
  
      if params is None:
          return
  
      params = self.check_params(params)
  
      self.torch_class.__init__(self, params, **self.param_dict)
  
imports: from torch import optim
  from fate.components.components.nn.torch.base import TorchOptimizer
label_desc: Use image node_id 1 to create a new SGD object from inherited base classes: optim, TorchOptimizer with example: obj = SGD(lr, params, momentum, dampening, weight_decay, nesterov)