Jiarui Fang 9e4c6449b0 [checkpoint] add ColoOptimizer checkpointing (#1316)

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

..

__init__.py

[Optimizer] Remove useless ColoOptimizer (#1312)

2022-07-14 16:57:48 +08:00

colossalai_optimizer.py

[checkpoint] add ColoOptimizer checkpointing (#1316)

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

cpu_adam.py

[zero] improve adaptability for not-shard parameters (#708)

2022-04-11 13:38:51 +08:00

fused_adam.py

polish optimizer docstring (#619)

2022-04-01 16:27:03 +08:00

fused_lamb.py

polish optimizer docstring (#619)

2022-04-01 16:27:03 +08:00

fused_sgd.py

[optim] refactor fused sgd (#1134)

2022-06-20 11:19:38 +08:00

hybrid_adam.py

fix bugs in CPU adam (#633)

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

lamb.py

[NFC] polish colossalai/nn/optimizer/lamb.py code style (#1275)

2022-07-13 12:08:21 +08:00

lars.py

Fixed docstring in colossalai (#171)

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

utils.py

fix bugs in CPU adam (#633)

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