Files
ColossalAI/colossalai/engine/schedule/__init__.py
2021-10-28 18:21:23 +02:00

6 lines
196 B
Python

from ._base_schedule import BaseSchedule
from ._no_pipeline import NoPipelineSchedule
from ._pipeline import PipelineSchedule
__all__ = ['BaseSchedule', 'NoPipelineSchedule', 'PipelineSchedule']