nn.Graph optimizer part 2: add L2, pass job complete, refactor (#5604)
* add valid optimizer case to pass job complete
* refactor and fix optimizer
* support l2 in variable op
* merge master
* format
* fix merge
* rm useless include
* address review
* auto format by CI
* rm internal tensor check after merge master
* restore add loss after merge master
Co-authored-by:
oneflow-ci-bot <ci-bot@oneflow.org>
Showing
- oneflow/core/framework/op_interpreter/lazy_op_interpreter.cpp 4 additions, 1 deletion...low/core/framework/op_interpreter/lazy_op_interpreter.cpp
- python/oneflow/framework/graph_build_util.py 31 additions, 6 deletionspython/oneflow/framework/graph_build_util.py
- python/oneflow/framework/tensor.py 8 additions, 1 deletionpython/oneflow/framework/tensor.py
- python/oneflow/nn/graph.py 57 additions, 29 deletionspython/oneflow/nn/graph.py
- python/oneflow/nn/graph_optimizer.py 25 additions, 0 deletionspython/oneflow/nn/graph_optimizer.py
- python/oneflow/nn/optimizer/optimizer.py 3 additions, 2 deletionspython/oneflow/nn/optimizer/optimizer.py
- python/oneflow/nn/optimizer/sgd.py 9 additions, 4 deletionspython/oneflow/nn/optimizer/sgd.py
- python/oneflow/test/graph/test_graph_optimizer.py 62 additions, 13 deletionspython/oneflow/test/graph/test_graph_optimizer.py
Please register or sign in to comment