github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-02-02 23:49:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
2a7fa2e7d08bc2cb5cd67438489793ddff742ee4
ColossalAI/tests/test_optimizer/_utils.py
Edenzzzz 79f7a7b211 [misc] Accelerate CI for zero and dist optim (#5758) 
* remove fp16 from lamb

* remove d2h copy in checking states

---------

Co-authored-by: Edenzzzz <wtan45@wisc.edu>
2024-06-05 11:25:19 +08:00

9.8 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink