mirror of https://github.com/hpcaitech/ColossalAI
![]() * [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 |