ColossalAI/colossalai/zero
ver217 6e553748a7
polish sharded optim docstr and warning (#770)
2022-04-14 21:03:59 +08:00
..
init_ctx [zero] refactor ShardedParamV2 for convenience (#742) 2022-04-13 14:54:26 +08:00
shard_utils [refactor] zero directory (#724) 2022-04-11 23:13:02 +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