RangiLyu 0279ac2e8d [Feature] Support EMA and SWA. (#239) ... * [Feature] Support EMA and SWA. * add ema hook * add avg model ut * add more unit tests * resolve comments * fix warmup ema * rename * fix comments * add assert * fix typehint * add comments		2022-05-19 18:53:04 +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	[Refactor] Make loop-related attributes to be runner's properties. (#236)	2022-05-18 22:35:10 +08:00
test_iter_timer_hook.py	[Refactor] Make loop-related attributes to be runner's properties. (#236)	2022-05-18 22:35:10 +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	[Refactor] Add batch_idx to hook input. (#140)	2022-03-29 11:40:38 +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