ColossalAI/tests/test_zero/low_level_zero at 4953b4ace15da59f4d122c5e433b08ad9db13054 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 17:53:08 +00:00

Files

History

HELSON 21c88220ce [zero] add unit test for low-level zero init (#2474 )

2023-01-15 10:42:01 +08:00

..

test_grad_acc.py

[zero] polish low level optimizer (#2473 )

2023-01-13 14:56:17 +08:00

test_zero1_2.py

[zero] polish low level optimizer (#2473 )

2023-01-13 14:56:17 +08:00

test_zero_init.py

[zero] add unit test for low-level zero init (#2474 )

2023-01-15 10:42:01 +08:00