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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-25 17:22:12 +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