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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-02-21 14:32:09 +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