ColossalAI/colossalai/zero
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