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