jiaruifang d984dff88c add bert for unitest and sharded model is not able to pass the bert case

2022-03-09 13:30:38 +08:00

..

init_ctx

[zero] Update sharded model v2 using sharded param v2 (#323)

2022-03-08 18:18:06 +08:00

shard_utils

[zero] Update sharded model v2 using sharded param v2 (#323)

2022-03-08 18:18:06 +08:00

sharded_model

add bert for unitest and sharded model is not able to pass the bert case

2022-03-09 13:30:38 +08:00

sharded_optim

[zero] cpu adam kernel (#288)

2022-03-04 16:05:15 +08:00

sharded_param

[zero] Update sharded model v2 using sharded param v2 (#323)

2022-03-08 18:18:06 +08:00

__init__.py

added buffer sync to naive amp model wrapper (#291)

2022-03-02 16:47:17 +08:00