mirror of https://github.com/hpcaitech/ColossalAI
a00644079e
* reorgnize colotensor directory * polish code |
||
---|---|---|
.. | ||
__init__.py | ||
colo_optimizer.py | ||
colossalai_optimizer.py | ||
cpu_adam.py | ||
fused_adam.py | ||
fused_lamb.py | ||
fused_sgd.py | ||
hybrid_adam.py | ||
lamb.py | ||
lars.py | ||
utils.py |