Mashiro 6b1b8a3751

[Fix] Fix unit test in windows (#515)

2022-09-13 11:46:21 +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

[Fix] Fix unit test in windows (#515)

2022-09-13 11:46:21 +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