ColossalAI/colossalai/zero
ver217 d0ae0f2215 [zero] update sharded optim v2 (#334) 2022-03-11 15:50:28 +08:00
..
init_ctx [zero] Update sharded model v2 using sharded param v2 (#323) 2022-03-11 15:50:28 +08:00
shard_utils [zero] Update sharded model v2 using sharded param v2 (#323) 2022-03-11 15:50:28 +08:00
sharded_model [zero] update sharded optim v2 (#334) 2022-03-11 15:50:28 +08:00
sharded_optim [zero] update sharded optim v2 (#334) 2022-03-11 15:50:28 +08:00
sharded_param [zero] Update sharded model v2 using sharded param v2 (#323) 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