ColossalAI/colossalai/zero/init_ctx at 055fbf5be680dfde20be1c51302f3c8b154a93e4 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-20 09:34:48 +00:00

Files

History

HELSON 055fbf5be6 [zero] adapt zero for unsharded paramters (Optimizer part) (#601 )

2022-04-01 20:10:47 +08:00

..

__init__.py

[zero] add zero context manager to change config during initialization (#546 )

2022-03-29 17:57:59 +08:00

init_context.py

[zero] adapt zero for unsharded paramters (Optimizer part) (#601 )

2022-04-01 20:10:47 +08:00