[Feature] Support multiple optimizers (#235)
* Support multiple optimizers * minor refinement * improve unit tests * minor fix * Update unit tests for resuming or saving ckpt for multiple optimizers * refine docstring * refine docstring * fix typo * update docstring * refactor the logic to build multiple optimizers * resolve comments * ParamSchedulers spports multiple optimizers * refine docstring * RuntimeInfoHook supports printing multi learning rates * resolve comments * fix typo
Showing
- mmengine/hooks/param_scheduler_hook.py 34 additions, 6 deletionsmmengine/hooks/param_scheduler_hook.py
- mmengine/hooks/runtime_info_hook.py 7 additions, 2 deletionsmmengine/hooks/runtime_info_hook.py
- mmengine/runner/runner.py 207 additions, 55 deletionsmmengine/runner/runner.py
- tests/test_hook/test_param_scheduler_hook.py 52 additions, 0 deletionstests/test_hook/test_param_scheduler_hook.py
- tests/test_hook/test_runtime_info_hook.py 20 additions, 0 deletionstests/test_hook/test_runtime_info_hook.py
- tests/test_runner/test_runner.py 193 additions, 28 deletionstests/test_runner/test_runner.py
Loading
Please register or sign in to comment