ColossalAI/colossalai/zero
Jiarui Fang 90d3aef62c [zero] yet an improved sharded param (#311) 2022-03-11 15:50:28 +08:00
..
shard_utils [zero] polish shard strategy (#310) 2022-03-11 15:50:28 +08:00
sharded_model fix sharded param hook and unit test 2022-03-11 15:50:28 +08:00
sharded_optim polish code 2022-03-11 15:50:28 +08:00
sharded_param [zero] yet an improved sharded param (#311) 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