feat eager optimizer (#4680)
* feat(Optim): add sgd
* fix(*): add no_grad guard in optimizer
* fix(*): fix bug in optimizer test
* style(*): format codes
* style(Optimizer): refine codes
* style(Optimizer): refine codes
* fix(Optimizer): fix bug in ParamGroup
* fix(*): add __init__.py
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/nn/optimizer/__init__.py 0 additions, 0 deletionsoneflow/python/nn/optimizer/__init__.py
- oneflow/python/nn/optimizer/optimizer.py 173 additions, 0 deletionsoneflow/python/nn/optimizer/optimizer.py
- oneflow/python/test/modules/test_optimizers.py 94 additions, 0 deletionsoneflow/python/test/modules/test_optimizers.py
Please register or sign in to comment