github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-02 08:37:36 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
85e045b063a70cd36ccc0405acc245d86f2a1621
ColossalAI/colossalai/nn/optimizer/zero_optimizer.py
HELSON e7d3afc9cc [optimizer] add div_scale for optimizers (#2117) 
* [optimizer] add div_scale for optimizers

* [zero] use div_scale in zero optimizer

* fix testing error
2022-12-12 17:58:57 +08:00

13 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink