ColossalAI/tests/test_trainer/test_pipeline
Frank Lee 3601b2bad0
[test] fixed rerun_on_exception and adapted test cases (#487)
2022-03-25 17:25:12 +08:00
..
model pipeline last stage supports multi output (#151) 2022-01-17 15:57:47 +08:00
resnet_config.py Hotfix/Colossalai layers (#92) 2021-12-29 23:32:10 +08:00
test_p2p.py [test] fixed rerun_on_exception and adapted test cases (#487) 2022-03-25 17:25:12 +08:00
test_partition.py [test] fixed rerun_on_exception and adapted test cases (#487) 2022-03-25 17:25:12 +08:00
test_pipeline_schedule.py [test] fixed rerun_on_exception and adapted test cases (#487) 2022-03-25 17:25:12 +08:00