Shortcuts

ConstantParamScheduler

class mmengine.optim.ConstantParamScheduler(optimizer, param_name, factor=0.3333333333333333, begin=0, end=1000000000, last_step=-1, by_epoch=True, verbose=False)[源代码]

Decays the parameter value of each parameter group by a small constant factor until the number of epoch reaches a pre-defined milestone: end. Notice that such decay can happen simultaneously with other changes to the parameter value from outside this scheduler.

参数:
  • optimizer (Optimizer or BaseOptimWrapper) – optimizer or Wrapped optimizer.

  • param_name (str) – Name of the parameter to be adjusted, such as lr, momentum.

  • factor (float) – The number we multiply parameter value until the milestone. Defaults to 1./3.

  • begin (int) – Step at which to start updating the parameters. Defaults to 0.

  • end (int) – Step at which to stop updating the parameters. Defaults to INF.

  • last_step (int) – The index of last step. Used for resume without state dict. Defaults to -1.

  • by_epoch (bool) – Whether the scheduled parameters are updated by epochs. Defaults to True.

  • verbose (bool) – Whether to print the value for each update. Defaults to False.

classmethod build_iter_from_epoch(*args, begin=0, end=1000000000, by_epoch=True, epoch_length=None, **kwargs)[源代码]

Build an iter-based instance of this scheduler from an epoch-based config.