onnxruntime
Fix memory consumption discrepancy
#12266
Merged
Overview
Commits
8
Changes
View On GitHub

Fix memory consumption discrepancy #12266

pengwa merged 8 commits into master from pengwa/export_memory
pengwa
pengwa release cached cuda memory after temp model_copy run
e5d3b07e
pengwa op schema change only: remove PythonOp forward output from PythonOpGr…
168bf9c4
pengwa always export model using torch.no_grad
33398475
pengwa 1.update PythonOP's "input_requires_grads" attribute according to ORT…
eaa2caec
pengwa fix uts
41c321d1
pengwa pengwa requested a review from wschin wschin 3 years ago
pengwa pengwa requested a review from ytaous ytaous 3 years ago
pengwa pengwa requested a review from Lafi7e Lafi7e 3 years ago
pengwa pengwa requested a review from askhade askhade 3 years ago
pengwa pengwa requested a review from baijumeswani baijumeswani 3 years ago
pengwa pengwa added training
wschin
wschin dismissed these changes on 2022-07-22
pengwa refine basde on wschin's comments && fix pylint
fff8c9ae
pengwa pengwa dismissed their stale review via fff8c9ae 3 years ago
pengwa fix comments
27460c2f
lgtm-com
ytaous
ytaous commented on 2022-07-22
pengwa fix unused variable
775c9835
ytaous
ytaous approved these changes on 2022-07-22
ytaous
pengwa pengwa merged feabafe5 into master 3 years ago
pengwa pengwa deleted the pengwa/export_memory branch 3 years ago
thiagocrepaldi

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
training
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum