Skip to content

Commit

Permalink
Revert tensor names to torch 1.10.2 in CorrdConvAC sample
Browse files Browse the repository at this point in the history
Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
  • Loading branch information
rajeevsrao committed May 13, 2022
1 parent f238f86 commit 99a11a5
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 5 deletions.
6 changes: 4 additions & 2 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,10 +1,12 @@
onnx==1.9.0; (platform_machine=="x86_64" and sys.platform=="linux")
onnx==1.10.0; (platform_machine=="aarch64" and sys.platform=="linux")
onnx==1.10.2
tensorflow-gpu==2.5.1; (platform_machine=="x86_64" and sys.platform=="linux")
onnxruntime==1.8.1
-f https://download.pytorch.org/whl/cu113/torch_stable.html
torch==1.10.2+cu113
torchvision==0.11.3
Pillow==8.3.2
numpy
pycuda<2021.1
pytest
--extra-index-url https://pypi.ngc.nvidia.com
onnx-graphsurgeon
6 changes: 3 additions & 3 deletions samples/sampleOnnxMnistCoordConvAC/modify_onnx_ac.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,11 +57,11 @@ def main():
tmap = graph.tensors()
# You can figure out the input and output tensors using Netron.
inputs = [tmap["conv1"]]
outputs = [tmap["input"]]
outputs = [tmap["90"]]
replace_with_coordconvac(graph, inputs, outputs)

inputs = [tmap["onnx::Cast_90"]]
outputs = [tmap["input.3"]]
inputs = [tmap["92"]]
outputs = [tmap["170"]]
replace_with_coordconvac(graph, inputs, outputs)

# Remove the now-dangling subgraph.
Expand Down

0 comments on commit 99a11a5

Please sign in to comment.