ColossalAI/colossalai/nn/optimizer
ver217 e619a651fb
polish optimizer docstring (#619)
2022-04-01 16:27:03 +08:00
..
__init__.py [hotfix] add hybrid adam to __init__ (#584) 2022-03-31 19:08:34 +08:00
colossalai_optimizer.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
cpu_adam.py polish optimizer docstring (#619) 2022-04-01 16:27:03 +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 polish optimizer docstring (#619) 2022-04-01 16:27:03 +08:00
hybrid_adam.py polish optimizer docstring (#619) 2022-04-01 16:27:03 +08:00
lamb.py Fixed docstring in colossalai (#171) 2022-01-21 10:44:30 +08:00
lars.py Fixed docstring in colossalai (#171) 2022-01-21 10:44:30 +08:00