ColossalAI/colossalai/legacy/engine/__init__.py

5 lines
87 B
Python
Raw Normal View History

2021-10-28 16:21:23 +00:00
from ._base_engine import Engine
from .gradient_handler import *
__all__ = ['Engine']