mirror of https://github.com/hpcaitech/ColossalAI
d86ddd9b29
* improve stablility of zero * fix wrong index * add record stream |
||
---|---|---|
.. | ||
test_grad_acc.py | ||
test_zero1_2.py | ||
test_zero_ckpt.py | ||
test_zero_init.py | ||
test_zero_tp.py |