Shortcuts

ConstantMomentum

class mmengine.optim.ConstantMomentum(optimizer, *args, **kwargs)[source]

Decays the momentum 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 momentum value from outside this scheduler.

Parameters
  • optimizer (Optimizer or OptimWrapper) – optimizer or Wrapped optimizer.

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

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

  • end (int) – Step at which to stop updating the momentum. 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 momentum is updated by epochs. Defaults to True.

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

Read the Docs v: v0.8.3
Versions
latest
stable
v0.8.3
v0.8.2
v0.8.1
v0.8.0
v0.7.4
v0.7.3
v0.7.2
v0.7.1
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.