Files
ColossalAI/colossalai/booster/__init__.py
Frank Lee ed19290560 [booster] implemented mixed precision class (#3151)
* [booster] implemented mixed precision class

* polish code
2023-03-17 11:00:15 +08:00

5 lines
141 B
Python

from .accelerator import Accelerator
from .booster import Booster
from .environment_table import EnvironmentTable
from .plugin import Plugin