code: def get_warmup_lr(self):
      alpha = float(self.last_epoch) / float(self.warmup_steps)
      # _lr = max(self.base_lr * alpha, self.warmup_lr_init)
      _lr = self.base_lr * alpha
      return [_lr for _ in self.optimizer.param_groups]
  
file_path: insightface/reconstruction/jmlr/lr_scheduler.py
imports: import torch
  from torch.optim.lr_scheduler import _LRScheduler
label_desc: Use image node_id 2 for calling the StepScheduler obj's underlying member method code with example usage: obj.get_warmup_lr() and returns: unknown