ColossalAI/colossalai/nn/optimizer
ver217 367c615818
fix nvme docstring (#1450)
2022-08-12 18:01:02 +08:00
..
__init__.py
colossalai_optimizer.py
cpu_adam.py fix nvme docstring (#1450) 2022-08-12 18:01:02 +08:00
fused_adam.py
fused_lamb.py
fused_sgd.py
hybrid_adam.py fix nvme docstring (#1450) 2022-08-12 18:01:02 +08:00
lamb.py
lars.py
nvme_optimizer.py