ColossalAI/colossalai/engine/schedule
Frank Lee 6f82ac9bcb
[pipeline] supported more flexible dataflow control for pipeline parallel training (#1108)
* [pipeline] supported more flexible dataflow control for pipeline parallel training

* polish code

* polish code

* polish code
2022-06-15 10:41:28 +08:00
..
__init__.py [pipeline] refactor pipeline (#679) 2022-04-07 15:54:14 +08:00
_base_schedule.py [pipeline] supported more flexible dataflow control for pipeline parallel training (#1108) 2022-06-15 10:41:28 +08:00
_non_pipeline_schedule.py [pipeline] supported more flexible dataflow control for pipeline parallel training (#1108) 2022-06-15 10:41:28 +08:00
_pipeline_schedule.py [pipeline] supported more flexible dataflow control for pipeline parallel training (#1108) 2022-06-15 10:41:28 +08:00