Refine optimizer (#4840)
* refactor(Optim): refine optimizer codes
* docs(SGD): add document for SGD
* docs(SGD): fix code
* test(Adam): fix test_optim_adam bug
Co-authored-by: 
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/nn/optimizer/adam.py 3 additions, 10 deletionsoneflow/python/nn/optimizer/adam.py
- oneflow/python/nn/optimizer/optimizer.py 1 addition, 92 deletionsoneflow/python/nn/optimizer/optimizer.py
- oneflow/python/nn/optimizer/sgd.py 132 additions, 0 deletionsoneflow/python/nn/optimizer/sgd.py
- oneflow/python/test/modules/test_optim_adam.py 4 additions, 4 deletionsoneflow/python/test/modules/test_optim_adam.py
- oneflow/python/test/modules/test_optim_sgd.py 2 additions, 3 deletionsoneflow/python/test/modules/test_optim_sgd.py
Please register or sign in to comment