ColossalAI/colossalai/zero/gemini at 990d4c3e4e433fe08c3fd7f3a86f678bd0381f86 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 01:35:21 +00:00

Files

History

YH bcf0cbcbe7 [doc] Add docs for clip args in zero optim (#3504 )

2023-04-10 11:11:28 +08:00

..

[zero] reorganize zero/gemini folder structure (#3424 )

2023-04-04 13:48:16 +08:00

[zero] reorganize zero/gemini folder structure (#3424 )

2023-04-04 13:48:16 +08:00

__init__.py

[zero] reorganize zero/gemini folder structure (#3424 )

2023-04-04 13:48:16 +08:00

colo_init_context.py

[zero] reorganize zero/gemini folder structure (#3424 )

2023-04-04 13:48:16 +08:00

gemini_ddp.py

[zero] reorganize zero/gemini folder structure (#3424 )

2023-04-04 13:48:16 +08:00

gemini_hook.py

[zero] reorganize zero/gemini folder structure (#3424 )

2023-04-04 13:48:16 +08:00

gemini_mgr.py

[zero] reorganize zero/gemini folder structure (#3424 )

2023-04-04 13:48:16 +08:00

gemini_optimizer.py

[doc] Add docs for clip args in zero optim (#3504 )

2023-04-10 11:11:28 +08:00

placement_policy.py

[zero] reorganize zero/gemini folder structure (#3424 )

2023-04-04 13:48:16 +08:00

utils.py

[zero] reorganize zero/gemini folder structure (#3424 )

2023-04-04 13:48:16 +08:00