code: def __init__(
      self,
      optimizer,
      *,
      max_steps,
      min_lr=0,
      last_epoch=-1,
      max_steps_for_lr_sched=None,
      **kwargs,
  ):
      super().__init__(
          optimizer=optimizer,
          max_steps=max_steps,
          last_epoch=last_epoch,
          min_lr=min_lr,
          **kwargs,
      )
      if max_steps_for_lr_sched:
          self.max_steps = max_steps_for_lr_sched
          self.decay_steps = self.max_steps - (
              self.constant_steps + self.warmup_steps
          )
  
label_desc: Use image node_id 1 to create a new CosineAnnealingExp object from inherited base classes: CosineAnnealing with example: obj = CosineAnnealingExp(optimizer)