mirror of https://github.com/hpcaitech/ColossalAI
a3269de5c9
* Added CPU Adam * finished the cpu adam * updated the license * delete useless parameters, removed resnet * modified the method off cpu adam unittest * deleted some useless codes * removed useless codes Co-authored-by: ver217 <lhx0217@gmail.com> Co-authored-by: Frank Lee <somerlee.9@gmail.com> Co-authored-by: jiaruifang <fangjiarui123@gmail.com> |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| colossalai_optimizer.py | ||
| cpu_adam.py | ||
| fused_adam.py | ||
| fused_lamb.py | ||
| fused_sgd.py | ||
| lamb.py | ||
| lars.py | ||