Jiarui Fang cec05b25c9 [zero] update zero context init with the updated test utils (#327)

2022-03-08 14:45:01 +08:00

..

init_ctx

[zero] update zero context init with the updated test utils (#327)

2022-03-08 14:45:01 +08:00

shard_utils

[zero] update zero context init with the updated test utils (#327)

2022-03-08 14:45:01 +08:00

sharded_model

fix sharded param hook and unit test

2022-03-04 15:27:39 +08:00

sharded_optim

[zero] cpu adam kernel (#288)

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

sharded_param

[zero] update zero context init with the updated test utils (#327)

2022-03-08 14:45:01 +08:00

__init__.py

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

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