code: def __init__(
      self, optimizer, base_lr, lr_steps, warmup_steps, last_epoch=-1
  ):
      self.base_lr = base_lr
      self.warmup_lr_init = 0.0001
      self.lr_steps = lr_steps
      self.warmup_steps: int = warmup_steps
      super(StepScheduler, self).__init__(optimizer, last_epoch, False)
  
file_path: insightface/reconstruction/jmlr/lr_scheduler.py
imports: import torch
  from torch.optim.lr_scheduler import _LRScheduler
label_desc: Use image node_id 1 to create a new StepScheduler object from inherited base classes: _LRScheduler with example: obj = StepScheduler(optimizer, base_lr, lr_steps, warmup_steps, last_epoch)