Jiarui Fang 4d9332b4c5 [refactor] moving memtracer to gemini (#801)

2022-04-19 10:13:08 +08:00

..

init_ctx

[zero] refactor ShardedParamV2 for convenience (#742)

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

shard_utils

[zero] refactor shard and gather operation (#773)

2022-04-15 14:41:31 +08:00

sharded_model

[refactor] moving memtracer to gemini (#801)

2022-04-19 10:13:08 +08:00

sharded_optim

[refactor] moving memtracer to gemini (#801)

2022-04-19 10:13:08 +08:00

sharded_param

[hotfix] fix memory leak in zero (#781)

2022-04-18 13:57:03 +08:00

utils

[refactor] moving memtracer to gemini (#801)

2022-04-19 10:13:08 +08:00

__init__.py

[refactor] remove old zero code (#517)

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