mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-23 10:11:37 +00:00
f51ce1bc8e
* [api] update optimizer wrapper to fit pipeline * [pipeline] add base schedule * [pipeline] add 1f1b schedule * [test] add pipeline schedule utils test * [pipeline] fix import
8 lines
166 B
Python
8 lines
166 B
Python
from .base import PipelineSchedule
|
|
from .one_f_one_b import OneForwardOneBackwardSchedule
|
|
|
|
__all__ = [
|
|
'PipelineSchedule',
|
|
'OneForwardOneBackwardSchedule',
|
|
]
|