ColossalAI/tests/test_trainer/test_pipeline
ver217 8f02a88db2
add interleaved pipeline, fix naive amp and update pipeline model initializer (#80)
2021-12-20 23:26:19 +08:00
..
model added CI for unit testing (#69) 2021-12-16 10:32:08 +08:00
resnet_config.py added CI for unit testing (#69) 2021-12-16 10:32:08 +08:00
test_p2p.py added CI for unit testing (#69) 2021-12-16 10:32:08 +08:00
test_partition.py add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 2021-12-20 23:26:19 +08:00
test_pipeline_schedule.py add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 2021-12-20 23:26:19 +08:00