ColossalAI/colossalai/zero
HELSON 17e73e62cc
[hotfix] fix bugs for unsharded parameters when restore data (#664)
2022-04-03 22:02:11 +08:00
..
init_ctx Revert "[zero] polish init context (#645)" (#657) 2022-04-02 18:30:06 +08:00
shard_utils [refactor] memory utils (#577) 2022-04-01 09:22:33 +08:00
sharded_model polish docstring of zero (#612) 2022-04-01 14:50:56 +08:00
sharded_optim [hotfix] fix bugs for unsharded parameters when restore data (#664) 2022-04-03 22:02:11 +08:00
sharded_param [refactor] memory utils (#577) 2022-04-01 09:22:33 +08:00
__init__.py [refactor] remove old zero code (#517) 2022-03-25 14:54:39 +08:00