ColossalAI/tests/test_trainer
Frank Lee 526a318032 [unit test] Refactored test cases with component func (#339)
* refactored test with component func

* fixed bug
2022-03-11 15:50:28 +08:00
..
test_pipeline pipeline last stage supports multi output (#151) 2022-01-17 15:57:47 +08:00
test_trainer_with_non_pipe_schedule.py [unit test] Refactored test cases with component func (#339) 2022-03-11 15:50:28 +08:00
test_trainer_with_pipe_schedule.py Hotfix/Colossalai layers (#92) 2021-12-29 23:32:10 +08:00