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

2022-04-15 14:41:31 +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

[gemini] init genimi individual directory (#754)

2022-04-14 16:40:26 +08:00

sharded_optim

polish sharded optim docstr and warning (#770)

2022-04-14 21:03:59 +08:00

sharded_param

[hotfix] fix reuse_fp16_shard of sharded model (#756)

2022-04-14 14:56:46 +08:00

utils

[gemini] init genimi individual directory (#754)

2022-04-14 16:40:26 +08:00

__init__.py

[refactor] remove old zero code (#517)

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