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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-18 16:46:08 +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