[Feature] Support convert epoch-based schedulers to iter-based. (#221)
* [Feature] Support convert epoch-based schedulers to iter-based. * Support convert and refactor LR and Momentum to mixin. * Add unit tests * fix args and add runner ut * resolve comments
Showing
- mmengine/optim/scheduler/lr_scheduler.py 15 additions, 148 deletionsmmengine/optim/scheduler/lr_scheduler.py
- mmengine/optim/scheduler/momentum_scheduler.py 16 additions, 148 deletionsmmengine/optim/scheduler/momentum_scheduler.py
- mmengine/optim/scheduler/param_scheduler.py 171 additions, 0 deletionsmmengine/optim/scheduler/param_scheduler.py
- mmengine/runner/runner.py 23 additions, 4 deletionsmmengine/runner/runner.py
- tests/test_optim/test_scheduler/test_lr_scheduler.py 138 additions, 0 deletionstests/test_optim/test_scheduler/test_lr_scheduler.py
- tests/test_optim/test_scheduler/test_param_scheduler.py 161 additions, 2 deletionstests/test_optim/test_scheduler/test_param_scheduler.py
- tests/test_runner/test_runner.py 22 additions, 0 deletionstests/test_runner/test_runner.py
Loading
Please register or sign in to comment