Decays the momentum of each parameter group by linearly changing small multiplicative 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 from outside this scheduler.
optimizer (Optimizer or OptimWrapper) – optimizer or Wrapped optimizer.
start_factor (float) – The number we multiply momentum in the first epoch. The multiplication factor changes towards end_factor in the following epochs. Defaults to 1./3.
end_factor (float) – The number we multiply momentum at the end of linear changing process. Defaults to 1.0.
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.
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4