HELSON b31daed4cf fix bugs in CPU adam (#633) ...

* add cpu adam counter for all cpu adam

* fixed updating error in adam kernel

2022-04-02 17:04:05 +08:00

..

test_grad_handler.py

Refactored docstring to google style

2022-03-29 17:17:47 +08:00

test_kernel.py

Refactored docstring to google style

2022-03-29 17:17:47 +08:00

test_moe_group.py

Refactored docstring to google style

2022-03-29 17:17:47 +08:00

test_moe_zero_init.py

[zero] adapt zero for unsharded paramters (Optimizer part) (#601)

2022-04-01 20:10:47 +08:00

test_moe_zero_model.py

[zero] adapt zero for unsharded paramters (Optimizer part) (#601)

2022-04-01 20:10:47 +08:00

test_moe_zero_optim.py

fix bugs in CPU adam (#633)

2022-04-02 17:04:05 +08:00