mmengine/tests/test_hook
Zaida Zhou f1da9a1d7f
[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
2022-05-31 16:54:39 +08:00
..
test_checkpoint_hook.py [Fix] change CheckPointHook before_run to before train (#214) 2022-05-05 20:08:07 +08:00
test_ema_hook.py [Feature] Support EMA and SWA. (#239) 2022-05-19 18:53:04 +08:00
test_empty_cache_hook.py [Refactor] Add batch_idx to hook input. (#140) 2022-03-29 11:40:38 +08:00
test_hook.py [Enhance] Add RuntimeInfoHook to update runtime information. (#254) 2022-05-26 14:35:37 +08:00
test_iter_timer_hook.py [Fix] Fix some bugs in hooks and runner. (#242) 2022-05-20 17:18:24 +08:00
test_logger_hook.py [Refactor] Make loop-related attributes to be runner's properties. (#236) 2022-05-18 22:35:10 +08:00
test_naive_visualization_hook.py refactor visualization (#147) 2022-04-15 15:56:06 +08:00
test_optimizer_hook.py [Fix] resolve conflict betweem adapt and main. (#198) 2022-04-26 00:37:16 +08:00
test_param_scheduler_hook.py [Feature] Support multiple optimizers (#235) 2022-05-31 16:54:39 +08:00
test_runtime_info_hook.py [Feature] Support multiple optimizers (#235) 2022-05-31 16:54:39 +08:00
test_sampler_seed_hook.py [Refactor] Refactor data_batch type and remove cur_dataloader in runner. (#171) 2022-04-08 15:57:10 +08:00
test_sync_buffers_hook.py [Refactor] Add batch_idx to hook input. (#140) 2022-03-29 11:40:38 +08:00