Skip to content
Snippets Groups Projects
Unverified Commit 49d53fec authored by Yinggang Wang's avatar Yinggang Wang Committed by GitHub
Browse files

refactor SGDUpdate and MomentumUpdate UserOp (#4930)


* refactor(ModelUpdate): SGDUpdate and MomentumUpdate use optional input
                       for learning rate

* fix(*): fix bugs

* style(*): refine code

Co-authored-by: default avataroneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
parent 8b0bdcdc
No related branches found
No related tags found
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment