mirror of https://github.com/hpcaitech/ColossalAI
ed19290560
* [booster] implemented mixed precision class * polish code |
||
---|---|---|
.. | ||
__init__.py | ||
optimizer.py |