mirror of https://github.com/hpcaitech/ColossalAI
dd7cc58299
* add state dict for zero * fix unit test * polish |
||
---|---|---|
.. | ||
test_grad_acc.py | ||
test_zero1_2.py | ||
test_zero_ckpt.py | ||
test_zero_init.py | ||
test_zero_tp.py |