ColossalAI/colossalai/engine/__init__.py

8 lines
143 B
Python

from .amp_type import AMP_TYPE
from ._base_engine import Engine
from .gradient_handler import *
from .schedule import *
__all__ = ['Engine']