class_docstr: Anneals the learning rate.
code: def state_dict(self):
      state_dict = {
          "max_lr": self.max_lr,
          "warmup_steps": self.warmup_steps,
          "num_steps": self.num_steps,
          "decay_style": self.decay_style,
          "decay_steps": self.decay_steps,
          "min_lr": self.min_lr,
      }
      return state_dict
  
file_path: colossalai/examples/tutorial/sequence_parallel/lr_scheduler/annealing_lr.py
imports: import math
label_desc: Use image node_id 4 for calling the AnnealingLR obj's underlying member method code with example usage: obj.state_dict() and returns: state_dict