ColossalAI/colossalai/nn/optimizer
HELSON 0f8c7f9804
Fixed docstring in colossalai (#171)
2022-01-21 10:44:30 +08:00
..
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
colossalai_optimizer.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
fused_adam.py refactor kernel (#142) 2022-01-13 16:47:17 +08:00
fused_lamb.py Fixed docstring in colossalai (#171) 2022-01-21 10:44:30 +08:00
fused_sgd.py refactor kernel (#142) 2022-01-13 16:47:17 +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