ver217 12b4887097 [hotfix] fix CPUAdam kernel nullptr (#1410)

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

..

__init__.py

[hotfix] fix CPUAdam kernel nullptr (#1410)

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

colossalai_optimizer.py

[checkpoint] add ColoOptimizer checkpointing (#1316)

2022-07-15 09:52:55 +08:00

cpu_adam.py

[hotfix] fix CPUAdam kernel nullptr (#1410)

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

fused_adam.py

[hotfix] adapt ProcessGroup and Optimizer to ColoTensor (#1388)

2022-07-29 19:33:24 +08:00

fused_lamb.py

[hotfix] adapt ProcessGroup and Optimizer to ColoTensor (#1388)

2022-07-29 19:33:24 +08:00

fused_sgd.py

[hotfix] adapt ProcessGroup and Optimizer to ColoTensor (#1388)

2022-07-29 19:33:24 +08:00

hybrid_adam.py

[hotfix] fix CPUAdam kernel nullptr (#1410)

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

lamb.py

[hotfix] adapt ProcessGroup and Optimizer to ColoTensor (#1388)

2022-07-29 19:33:24 +08:00

lars.py

Fixed docstring in colossalai (#171)

2022-01-21 10:44:30 +08:00

nvme_optimizer.py

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

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