[Enhance] Auto set the `end` of param schedulers. (#361)
* [Enhance] Auto set the `end` of param schedulers. * Add log output and unit test * Update docstring * Update unit tests of `CosineAnnealingParamScheduler`.
Showing
- mmengine/optim/scheduler/param_scheduler.py 4 additions, 3 deletionsmmengine/optim/scheduler/param_scheduler.py
- mmengine/runner/runner.py 16 additions, 0 deletionsmmengine/runner/runner.py
- tests/test_optim/test_scheduler/test_lr_scheduler.py 5 additions, 0 deletionstests/test_optim/test_scheduler/test_lr_scheduler.py
- tests/test_optim/test_scheduler/test_momentum_scheduler.py 5 additions, 0 deletionstests/test_optim/test_scheduler/test_momentum_scheduler.py
- tests/test_optim/test_scheduler/test_param_scheduler.py 5 additions, 0 deletionstests/test_optim/test_scheduler/test_param_scheduler.py
- tests/test_runner/test_runner.py 19 additions, 0 deletionstests/test_runner/test_runner.py
Loading
Please register or sign in to comment