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_hook.py | ||
_checkpoint_hook.py | ||
_log_hook.py | ||
_lr_scheduler_hook.py | ||
_mem_tracer_hook.py | ||
_metric_hook.py |