mirror of https://github.com/hpcaitech/ColossalAI
8 lines
166 B
Python
8 lines
166 B
Python
|
from .base import PipelineSchedule
|
||
|
|
from .one_f_one_b import OneForwardOneBackwardSchedule
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
'PipelineSchedule',
|
||
|
|
'OneForwardOneBackwardSchedule',
|
||
|
|
]
|