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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-12 07:02:22 +00:00

Files

History

HELSON 22c4b88d56 [zero] refactor ShardedParamV2 for convenience (#742 )

2022-04-13 14:54:26 +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] refactor ShardedParamV2 for convenience (#742 )

2022-04-13 14:54:26 +08:00