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 |
||
|---|---|---|
| .. | ||
| gradient_handler | ||
| ophooks | ||
| paramhooks | ||
| schedule | ||
| __init__.py | ||
| _base_engine.py | ||