mirror of https://github.com/hpcaitech/ColossalAI
c415240db6
* impl nvme optimizer * update cpu adam * add unit test * update hybrid adam * update docstr * add TODOs * update CI * fix CI * fix CI * fix CI path * fix CI path * fix CI path * fix install tensornvme * fix CI * fix CI path * fix CI env variables * test CI * test CI * fix CI * fix nvme optim __del__ * fix adam __del__ * fix nvme optim * fix CI env variables * fix nvme optim import * test CI * test CI * fix CI |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| colossalai_optimizer.py | ||
| cpu_adam.py | ||
| fused_adam.py | ||
| fused_lamb.py | ||
| fused_sgd.py | ||
| hybrid_adam.py | ||
| lamb.py | ||
| lars.py | ||
| nvme_optimizer.py | ||
| utils.py | ||