[Enhance] Support eta_min_ratio in CosineAnnealingParamScheduler (#725)
* [Enhance] support eta_min_ratio in CosineAnnealingParamScheduler
* [doc] fix docstring
* [Enhance] add ut for eta_min_ratio
* [doc] update docstring
* avoid bc-breaking of eta_min
* [doc] add docstring in CosineAnnealingParamScheduler and CosineAnnealingMomentum
* Apply suggestions from code review
Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Showing
- mmengine/optim/scheduler/lr_scheduler.py 5 additions, 1 deletionmmengine/optim/scheduler/lr_scheduler.py
- mmengine/optim/scheduler/momentum_scheduler.py 5 additions, 1 deletionmmengine/optim/scheduler/momentum_scheduler.py
- mmengine/optim/scheduler/param_scheduler.py 30 additions, 11 deletionsmmengine/optim/scheduler/param_scheduler.py
- tests/test_optim/test_scheduler/test_param_scheduler.py 32 additions, 4 deletionstests/test_optim/test_scheduler/test_param_scheduler.py
Loading
Please register or sign in to comment