__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 |