RangiLyu 8d25dbdeda

[Fix] Fix EMAHook trigger train loop and AveragedModel sync buffer. (#467) ...

* [Fix] Fix EMAHook trigger train loop init during testing.

* fix sync buffer

* update ut

* fix sync buffer

* fix sync buffer

2022-08-26 14:21:56 +08:00

..

__init__.py

[Feature]: add base model, ddp model wrapper and unit test (#268)

2022-06-07 22:13:53 +08:00

checkpoint_hook.py

[Refactor] Refactor data flow to make the interface more natural (#468)

2022-08-24 22:04:55 +08:00

ema_hook.py

[Fix] Fix EMAHook trigger train loop and AveragedModel sync buffer. (#467)

2022-08-26 14:21:56 +08:00

empty_cache_hook.py

[Refactor] Refactor data flow to make the interface more natural (#468)

2022-08-24 22:04:55 +08:00

hook.py

[Refactor] Refactor data flow to make the interface more natural (#468)

2022-08-24 22:04:55 +08:00

iter_timer_hook.py

[Refactor] Refactor data flow to make the interface more natural (#468)

2022-08-24 22:04:55 +08:00

logger_hook.py

[Refactor] Refactor data flow to make the interface more natural (#468)

2022-08-24 22:04:55 +08:00

naive_visualization_hook.py

[Refactor] Refactor data flow to make the interface more natural (#468)

2022-08-24 22:04:55 +08:00

param_scheduler_hook.py

[Refactor] Refactor data flow to make the interface more natural (#468)

2022-08-24 22:04:55 +08:00

runtime_info_hook.py

[Refactor] Refactor data flow to make the interface more natural (#468)

2022-08-24 22:04:55 +08:00

sampler_seed_hook.py

[Docs] Simplify hook docs (#428)

2022-08-23 16:20:47 +08:00

sync_buffer_hook.py

[Refactor] Refactor the import rule (#459)

2022-08-23 18:58:36 +08:00