ColossalAI/colossalai/engine
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
..
gradient_handler fix format (#586) 2022-04-06 11:40:59 +08:00
ophooks fix format (#586) 2022-04-06 11:40:59 +08:00
paramhooks Fix/format colossalai/engine/paramhooks/(#350) 2022-03-11 15:50:28 +08:00
schedule [pipeline] refactor pipeline (#679) 2022-04-07 15:54:14 +08:00
__init__.py fix format (#580) 2022-04-06 11:40:59 +08:00
_base_engine.py '[NFC] polish <colossalai/engine/_base_engine.py> code style' (#631) 2022-04-06 11:40:59 +08:00