github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-03 13:16:53 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1cf6d92d7c93a26e29cadeb71bb34ee96b149a28
ColossalAI/tests/test_optimizer/test_fused_adam_kernel.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

2.6 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink