Mashiro 576e5c8f91

[Fix] Regist pytorch ddp and dp to MODEL_WRAPPERS, add unit test to is_model_wrapper (#474) ...

* regist pytorch ddp and dp, add unit test

* minor refine

* Support check custom wrapper

* enhance ut

2022-08-28 17:09:41 +08:00

..

base_model

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

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

wrappers

[Fix] Regist pytorch ddp and dp to MODEL_WRAPPERS, add unit test to is_model_wrapper (#474)

2022-08-28 17:09:41 +08:00

__init__.py

[Refactor] Refactor code structure (#395)

2022-08-24 19:14:07 +08:00

averaged_model.py

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

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

base_module.py

[Refactor] Refactor code structure (#395)

2022-08-24 19:14:07 +08:00

utils.py

[Refactor] Refactor code structure (#395)

2022-08-24 19:14:07 +08:00

weight_init.py

[Refactor] Refactor code structure (#395)

2022-08-24 19:14:07 +08:00