ColossalAI/tests/test_ddp at 3263cdf57f6be9b61fc2f4e64b8f3e2e1dde8b3f - 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 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