mirror of https://github.com/hpcaitech/ColossalAI
0ed7042f42
* 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 |
||
---|---|---|
.. | ||
__init__.py | ||
_base_schedule.py | ||
_non_pipeline_schedule.py | ||
_pipeline_schedule.py |