ColossalAI/tests/test_trainer/test_pipeline
Frank Lee bb2790cf0b
optimize engine and trainer test (#448)
2022-03-17 15:44:17 +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 optimize engine and trainer test (#448) 2022-03-17 15:44:17 +08:00
test_partition.py Hotfix/Colossalai layers (#92) 2021-12-29 23:32:10 +08:00
test_pipeline_schedule.py optimize engine and trainer test (#448) 2022-03-17 15:44:17 +08:00