HELSON e7d3afc9cc [optimizer] add div_scale for optimizers (#2117) ...

* [optimizer] add div_scale for optimizers

* [zero] use div_scale in zero optimizer

* fix testing error

2022-12-12 17:58:57 +08:00

..

test_cpu_adam.py

[kernel] move all symlinks of kernel to colossalai._C (#1971)

2022-11-17 13:42:33 +08:00

test_fused_adam_kernel.py

[optimizer] add div_scale for optimizers (#2117)

2022-12-12 17:58:57 +08:00

test_fused_adam.py

[hotfix] fix CPUAdam kernel nullptr (#1410)

2022-08-05 19:45:45 +08:00

test_hybrid_adam.py

[hotfix] fix CPUAdam kernel nullptr (#1410)

2022-08-05 19:45:45 +08:00

test_nvme.py

[nvme] CPUAdam and HybridAdam support NVMe offload (#1360)

2022-07-26 17:25:24 +08:00