style(Optimzier): refine optimizer interface (#5145)
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/python/nn/optimizer/adam.py 10 additions, 10 deletionsoneflow/python/nn/optimizer/adam.py
- oneflow/python/nn/optimizer/adamw.py 10 additions, 10 deletionsoneflow/python/nn/optimizer/adamw.py
- oneflow/python/nn/optimizer/lambda_lr.py 3 additions, 3 deletionsoneflow/python/nn/optimizer/lambda_lr.py
- oneflow/python/nn/optimizer/lr_scheduler.py 7 additions, 9 deletionsoneflow/python/nn/optimizer/lr_scheduler.py
- oneflow/python/nn/optimizer/optimizer.py 8 additions, 2 deletionsoneflow/python/nn/optimizer/optimizer.py
- oneflow/python/nn/optimizer/rmsprop.py 11 additions, 11 deletionsoneflow/python/nn/optimizer/rmsprop.py
- oneflow/python/nn/optimizer/sgd.py 10 additions, 10 deletionsoneflow/python/nn/optimizer/sgd.py
- oneflow/python/nn/optimizer/step_lr.py 2 additions, 5 deletionsoneflow/python/nn/optimizer/step_lr.py
Please register or sign in to comment