class_docstr: StepScheduler
  Step decay
class_docstr_tok: ["StepScheduler","Step","decay"]
docstr: Args:
      init_lr (float): Initial learning rate.
      gamma (float): Multiplier.
      iter_steps (list of int): Iteration at which gamma is multiplied.
  
  Example:
      StepScheduler(0.1, 0.1, [150000, 300000, 400000])
docstr_tok: ["Args",":","init_lr","(","float",")",":","Initial","learning","rate",".","gamma","(","float",")",":","Multiplier",".","iter_steps","(","list","of","int",")",":","Iteration","at","which","gamma","is","multiplied",".","Example",":","StepScheduler","(","0.1",",","0.1",",","[","150000",",","300000",",","400000","]",")"]
code: def __init__(self, init_lr, gamma, iter_steps):
      """
      Args:
          init_lr (float): Initial learning rate.
          gamma (float): Multiplier.
          iter_steps (list of int): Iteration at which gamma is multiplied.
  
      Example:
          StepScheduler(0.1, 0.1, [150000, 300000, 400000])
      """
      self.init_lr = init_lr
      self.gamma = gamma
      self.iter_steps = iter_steps
  
imports: import math
label_desc: Use image node_id 1 to create a new StepScheduler object from inherited base classes: BaseLearningRateScheduler with example: obj = StepScheduler(init_lr, gamma, iter_steps)