ColossalAI/colossalai/zero/sharded_optim at dfe423ae42d28a0fc1929fdbebd264e706dceefc - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-15 15:29:48 +00:00

Files

History

HELSON 17e73e62cc [hotfix] fix bugs for unsharded parameters when restore data (#664 )

2022-04-03 22:02:11 +08:00

..

__init__.py

[refactor] remove old zero code (#517 )

2022-03-25 14:54:39 +08:00

_utils.py

Feature/zero (#279 )

2022-03-11 15:50:28 +08:00

sharded_optim_v2.py

[hotfix] fix bugs for unsharded parameters when restore data (#664 )

2022-04-03 22:02:11 +08:00