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
..
test_base_model	[Refactor] Refactor data flow to make the interface more natural (#468)	2022-08-24 22:04:55 +08:00
test_wrappers	[Refactor] Refactor data flow to make the interface more natural (#468)	2022-08-24 22:04:55 +08:00
test_averaged_model.py	[Feature] Add autocast wrapper (#307)	2022-06-22 19:49:20 +08:00
test_base_module.py	[Feature] Add ModuleList Sequential and ModuleDict (#299)	2022-06-13 13:51:07 +08:00
test_model_utils.py	[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