ColossalAI/colossalai/engine/__init__.py

mirror of https://github.com/hpcaitech/ColossalAI
from ._base_engine import Engine
from .gradient_handler import *
from .schedule import *
from .amp import *


__all__ = ['Engine']
Migrated project 2021-10-28 16:21:23 +00:00			`from ._base_engine import Engine`
			`from .gradient_handler import *`
			`from .schedule import *`
fix FP16 optimizer and adapted torch amp with tensor parallel (#18) 2021-11-08 08:47:32 +00:00			`from .amp import *`
Migrated project 2021-10-28 16:21:23 +00:00

			`__all__ = ['Engine']`