ColossalAI/colossalai/trainer/hooks
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 [model checkpoint] updated checkpoint hook (#598) 2022-04-01 16:53:03 +08:00
_base_hook.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
_checkpoint_hook.py [model checkpoint] updated checkpoint hook (#598) 2022-04-01 16:53:03 +08:00
_log_hook.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
_lr_scheduler_hook.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
_mem_tracer_hook.py [profiler] add MemProfiler (#356) 2022-03-29 12:48:34 +08:00
_metric_hook.py [pipeline] refactor pipeline (#679) 2022-04-07 15:54:14 +08:00