github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-03 01:19:15 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
f6178728a0319fe58d02936e55dc08dfc19aca73
ColossalAI/colossalai/zero/sharded_optim/_utils.py
HELSON a1ce02d740 [zero] test gradient accumulation (#1964) 
* [zero] fix memory leak for zero2

* [zero] test gradient accumulation

* [zero] remove grad clip test
2022-11-29 13:00:30 +08:00

9.2 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink