ColossalAI/colossalai/zero
jiaruifang 7977422aeb add bert for unitest and sharded model is not able to pass the bert case 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 add bert for unitest and sharded model is not able to pass the bert case 2022-03-11 15:50:28 +08:00
sharded_optim [zero] cpu adam kernel (#288) 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