Jiarui Fang 0fcfb1e00d [test] make zero engine test really work (#447)

2022-03-17 17:24:25 +08:00

..

init_ctx

use double buffer to handle grad

2022-03-16 14:24:09 +08:00

shard_utils

[test] make zero engine test really work (#447)

2022-03-17 17:24:25 +08:00

sharded_model

[hotfix] fix initialize bug with zero (#442)

2022-03-17 13:16:22 +08:00

sharded_optim

[test] make zero engine test really work (#447)

2022-03-17 17:24:25 +08:00

sharded_param

use double buffer to handle grad

2022-03-16 14:24:09 +08:00

__init__.py

[zero] hybrid cpu adam (#445)

2022-03-17 15:05:41 +08:00