github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/test_zero/low_level_zero
History
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
..
test_grad_acc.py [zero] test gradient accumulation (#1964) 2022-11-29 13:00:30 +08:00
test_zero1_2.py [zero] test gradient accumulation (#1964) 2022-11-29 13:00:30 +08:00