mirror of https://github.com/hpcaitech/ColossalAI
![]() * remove fp16 from lamb * remove d2h copy in checking states --------- Co-authored-by: Edenzzzz <wtan45@wisc.edu> |
||
---|---|---|
.. | ||
test_gemini | ||
test_low_level |
![]() * remove fp16 from lamb * remove d2h copy in checking states --------- Co-authored-by: Edenzzzz <wtan45@wisc.edu> |
||
---|---|---|
.. | ||
test_gemini | ||
test_low_level |