mirror of https://github.com/hpcaitech/ColossalAI
105c5301c3
* [zero]added hybrid adam, removed loss scale of adam * remove useless code |
||
---|---|---|
.. | ||
unittest_cpu_adam.py | ||
unittest_fused_adam.py | ||
unittest_fused_adam_kernel.py | ||
unittest_hybrid_adam.py |