[Feature] Support CosineRestartParamScheduler. (#397)
* [Feature] Support CosineRestartParamScheduler. * add ut and docstring * add docstring
Showing
- mmengine/optim/scheduler/__init__.py 12 additions, 5 deletionsmmengine/optim/scheduler/__init__.py
- mmengine/optim/scheduler/lr_scheduler.py 33 additions, 0 deletionsmmengine/optim/scheduler/lr_scheduler.py
- mmengine/optim/scheduler/momentum_scheduler.py 34 additions, 0 deletionsmmengine/optim/scheduler/momentum_scheduler.py
- mmengine/optim/scheduler/param_scheduler.py 173 additions, 1 deletionmmengine/optim/scheduler/param_scheduler.py
- tests/test_optim/test_scheduler/test_lr_scheduler.py 44 additions, 2 deletionstests/test_optim/test_scheduler/test_lr_scheduler.py
- tests/test_optim/test_scheduler/test_momentum_scheduler.py 52 additions, 0 deletionstests/test_optim/test_scheduler/test_momentum_scheduler.py
- tests/test_optim/test_scheduler/test_param_scheduler.py 49 additions, 1 deletiontests/test_optim/test_scheduler/test_param_scheduler.py
Loading
Please register or sign in to comment