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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-15 15:29:48 +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