ColossalAI/tests/test_trainer/test_pipeline
YuliangLiu0306 b167258b6a
[pipeline]refactor ppschedule to support tensor list (#1050)
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* refactor ppschedule to support tensor list

* polish
2022-06-02 13:48:59 +08:00
..
model
resnet_config.py [refactor] pipeline, put runtime schedule into engine. (#627) 2022-04-03 20:46:45 +08:00
test_p2p.py [pipeline]refactor ppschedule to support tensor list (#1050) 2022-06-02 13:48:59 +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 [refactor] pipeline, put runtime schedule into engine. (#627) 2022-04-03 20:46:45 +08:00