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
test_p2p.py optimize engine and trainer test (#448) 2022-03-17 15:44:17 +08:00
test_partition.py
test_pipeline_schedule.py optimize engine and trainer test (#448) 2022-03-17 15:44:17 +08:00