ColossalAI/tests/test_ddp at ae1b58cd16ca6363c31bf4681607414c831c32d4 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-03 09:23:58 +00:00

Files

History

HELSON 4e98e938ce [zero] alleviate memory usage in ZeRODDP state_dict (#1398 )

2022-08-02 15:49:13 +08:00

..

test_ddp_ignore_params.py

[hotfix] ZeroDDP use new process group (#1333 )

2022-07-18 14:14:52 +08:00

test_ddp_state_dict.py

[zero] alleviate memory usage in ZeRODDP state_dict (#1398 )

2022-08-02 15:49:13 +08:00

test_reducer.py

[refactor] move chunk and chunkmgr to directory gemini (#1182 )

2022-06-29 13:31:02 +08:00