ver217 367c615818 fix nvme docstring (#1450)		2022-08-12 18:01:02 +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	fix nvme docstring (#1450)	2022-08-12 18:01:02 +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	fix nvme docstring (#1450)	2022-08-12 18:01:02 +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