onnxruntime
[python] [orttraining] Add utility to export a graph to compute gradients
#8125
Merged

[python] [orttraining] Add utility to export a graph to compute gradients #8125

baijumeswani merged 61 commits into microsoft:master from pytorch-gradient-graph
juharris
juharris orttraining: Start to add some Python bindings to call GradientGraphB…
ea154c9f
juharris Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
101f63e1
juharris Set up save method.
88412c50
juharris Set up save method.
2c55537d
juharris Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
62b31273
juharris Add a test.
a57b7328
juharris Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
fc098ce8
juharris Get the test to actually run.
c14c8f46
juharris Set up test.
d62fe214
juharris test: Convert Path to str.
0ee9ceb5
ryotatomioka
juharris
Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
57ddcb51
Merge branch 'pytorch-gradient-graph' of github.com:juharris/onnxrunt…
32899b4b
juharris Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
97145204
edgchen1 edgchen1 assigned SherlockNoMad SherlockNoMad 4 years ago
juharris Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
fe2c153a
juharris Add comment about where seg fault happens.
b93e9acc
juharris Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
54eee5c3
juharris Add print statements and notes to try to debug.
5396c4aa
juharris GraphTransformerManager::ApplyTransformers: check if `level_to_transf…
ef29fdfa
juharris Add notes about issues.
72bcaaa2
juharris Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
e0cb8cec
juharris Get loss_node_arg_name_ to keep it's value by making a string paramet…
40fc321c
juharris Add FIXME for Build not working via Python.
4c135866
juharris Correct Python <-> C++ issues using pointers.
0cc2abd2
juharris Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
3b687f30
juharris Start to compare outputs.
db37d92e
Try to get logger working.
8915b792
Got valid gradient graph built but it doesn't output gradients.
0d5e1b8d
Try to keep gradient_graph_config.set_gradients_as_graph_outputs = true.
a0389b79
WORKSs
d0ecc036
Make sure the gradients can be output by the ONNX graph.
cd7165ce
juharris juharris marked this pull request as ready for review 4 years ago
juharris juharris requested a review from baijumeswani baijumeswani 4 years ago
juharris juharris requested a review from BowenBao BowenBao 4 years ago
juharris juharris requested a review from liqunfu liqunfu 4 years ago
juharris juharris requested a review from SherlockNoMad SherlockNoMad 4 years ago
juharris juharris requested a review from thiagocrepaldi thiagocrepaldi 4 years ago
juharris juharris requested a review from tlh20 tlh20 4 years ago
Undo unnecessary change.
f2326291
Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
8f019bbb
Start export_gradient_graph.
4ed7a41b
Implement export_gradient_graph.
3793350c
Try to get build to import new Python file.
58899f60
Use WrapperModule.
f7931da3
Got test working.
86c76a70
juharris Add docs.
7a8b80e9
juharris Start to try to pass model weights.
d5602cdf
juharris Support passing model weights as inputs.
394ef2f8
juharris Support passing model weights as inputs.
27115a19
juharris Force giving model params as input to the gradient graph.
5ad361d1
juharris output None
bb813b19
juharris Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
0d49e564
juharris juharris requested a review from xadupre xadupre 4 years ago
xadupre
juharris
juharris juharris changed the title [python] [orttraining] Add utility to export a gradient graph [python] [orttraining] Add utility to export a graph to compute weight updates 4 years ago
xadupre
baijumeswani
baijumeswani commented on 2022-01-20
juharris Address PR comments.
8f1b2ef1
juharris Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
3dc7f226
xadupre
xadupre commented on 2022-01-31
xadupre
xadupre commented on 2022-01-31
xadupre
xadupre commented on 2022-01-31
xadupre
xadupre commented on 2022-01-31
juharris Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
6bdc1a23
juharris Correct some indentation.
76c3ce9a
juharris build.py: Correct test file name.
0d255013
juharris build: Correct some paths for CMake.
9c5463a8
juharris Remove model.train() call because it's not needed.
72bd9428
juharris setup.py: Correct package path.
fc5e36c0
juharris build.py: Add set current working directory for running test.
1ec24ed9
juharris
juharris commented on 2022-02-03
juharris Check if requires_grad before requesting gradients.
15b2e2a4
juharris build.py: Run orttraining_test_experimental_gradient_graph even witho…
03983bc7
juharris build.py: Don't run run_training_python_utility_tests in Windows.
13ab2cf5
juharris test: Move call to run orttraining_test_experimental_gradient_graph.p…
b2e822ef
juharris Correct typo in log statement.
b43677ef
juharris test: Fallback on getting available providers for InferenceSession.
049441f4
juharris gradient-graph: Avoid making a temp ONNX graph file. Use bytes instead.
41434c8f
juharris Merge branch 'master' of github.com:microsoft/onnxruntime into pytorc…
2dccdfca
baijumeswani
baijumeswani
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
azure-pipelines
azure-pipelines
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
microsoft microsoft deleted a comment from azure-pipelines on 2022-02-18
baijumeswani
baijumeswani commented on 2022-02-18
baijumeswani
baijumeswani approved these changes on 2022-02-18
juharris juharris changed the title [python] [orttraining] Add utility to export a graph to compute weight updates [python] [orttraining] Add utility to export a graph to compute gradients. 4 years ago
juharris juharris changed the title [python] [orttraining] Add utility to export a graph to compute gradients. [python] [orttraining] Add utility to export a graph to compute gradients 4 years ago
baijumeswani baijumeswani merged 742694f6 into master 4 years ago
juharris juharris deleted the pytorch-gradient-graph branch 4 years ago

Login to write a write a comment.

Login via GitHub

Assignees
Labels
Milestone