ColossalAI/colossalai/zero
HELSON 7c079d9c33
[hotfix] fixed bugs in ShardStrategy and PcieProfiler (#394)
2022-03-11 18:12:46 +08:00
..
init_ctx [bug] shard param during initializing the ShardedModelV2 (#381) 2022-03-11 15:50:28 +08:00
shard_utils [hotfix] fixed bugs in ShardStrategy and PcieProfiler (#394) 2022-03-11 18:12:46 +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