1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-05 23:18:22 +00:00
ColossalAI/colossalai/booster/interface/__init__.py
Frank Lee ed19290560
[booster] implemented mixed precision class ()
* [booster] implemented mixed precision class

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

4 lines
72 B
Python

from .optimizer import OptimizerWrapper
__all__ = ['OptimizerWrapper']