mmengine/tests/test_hook
Mashiro 38e78d5549
[Fix] Fix ema hook and add unit test (#327)
* Fix ema hook and add unit test

* save state_dict of ema.module

save state_dict of ema.module

* replace warning.warn with MMLogger.warn

* fix as comment

* fix bug

* fix bug
2022-07-04 14:23:23 +08:00
..
test_checkpoint_hook.py [Fix] Fix bug of not save-best in iteration-based training (#341) 2022-06-30 14:51:31 +08:00
test_ema_hook.py [Fix] Fix ema hook and add unit test (#327) 2022-07-04 14:23:23 +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 [Feat] Support save best ckpt (#310) 2022-06-22 19:48:46 +08:00
test_naive_visualization_hook.py refactor visualization (#147) 2022-04-15 15:56:06 +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 [Fix] fix build train_loop during test (#295) 2022-06-13 21:23:46 +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