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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 01:35:21 +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