Jiarui Fang 036404ca8a Revert "[zero] polish init context (#645)" (#657)

2022-04-02 18:30:06 +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

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

2022-04-01 20:10:47 +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