ColossalAI/colossalai/zero
ver217 88804aee49 add bucket tensor shard strategy 2022-03-14 14:48:32 +08:00
..
init_ctx [bug] shard param during initializing the ShardedModelV2 (#381) 2022-03-11 15:50:28 +08:00
shard_utils add bucket tensor shard strategy 2022-03-14 14:48:32 +08:00
sharded_model [zero] polish ShardedOptimV2 unittest (#385) 2022-03-11 15:50:28 +08:00
sharded_optim [zero] polish ShardedOptimV2 unittest (#385) 2022-03-11 15:50:28 +08:00
sharded_param [zero] find miss code (#378) 2022-03-11 15:50:28 +08:00
__init__.py added buffer sync to naive amp model wrapper (#291) 2022-03-11 15:50:28 +08:00