ColossalAI/colossalai/engine/schedule
YuliangLiu0306 0ed7042f42
[pipeline] refactor pipeline (#679)
* refactor pipeline---put runtime schedule into engine.

* add type hint for schedule Optional[BaseSchedule]

* preprocess schedule during engine initializing

* infer pipeline schedule params from config
2022-04-07 15:54:14 +08:00
..
__init__.py [pipeline] refactor pipeline (#679) 2022-04-07 15:54:14 +08:00
_base_schedule.py 'fix/format' (#573) 2022-04-06 11:40:59 +08:00
_non_pipeline_schedule.py [refactor] pipeline, put runtime schedule into engine. (#627) 2022-04-03 20:46:45 +08:00
_pipeline_schedule.py [pipeline] refactor pipeline (#679) 2022-04-07 15:54:14 +08:00