Skip to content

Commit 99a11a5

Browse files
committed
Revert tensor names to torch 1.10.2 in CorrdConvAC sample
Signed-off-by: Rajeev Rao <rajeevrao@nvidia.com>
1 parent f238f86 commit 99a11a5

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

requirements.txt

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,12 @@
1-
onnx==1.9.0; (platform_machine=="x86_64" and sys.platform=="linux")
2-
onnx==1.10.0; (platform_machine=="aarch64" and sys.platform=="linux")
1+
onnx==1.10.2
32
tensorflow-gpu==2.5.1; (platform_machine=="x86_64" and sys.platform=="linux")
43
onnxruntime==1.8.1
54
-f https://download.pytorch.org/whl/cu113/torch_stable.html
65
torch==1.10.2+cu113
6+
torchvision==0.11.3
77
Pillow==8.3.2
88
numpy
99
pycuda<2021.1
1010
pytest
11+
--extra-index-url https://pypi.ngc.nvidia.com
12+
onnx-graphsurgeon

samples/sampleOnnxMnistCoordConvAC/modify_onnx_ac.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,11 +57,11 @@ def main():
5757
tmap = graph.tensors()
5858
# You can figure out the input and output tensors using Netron.
5959
inputs = [tmap["conv1"]]
60-
outputs = [tmap["input"]]
60+
outputs = [tmap["90"]]
6161
replace_with_coordconvac(graph, inputs, outputs)
6262

63-
inputs = [tmap["onnx::Cast_90"]]
64-
outputs = [tmap["input.3"]]
63+
inputs = [tmap["92"]]
64+
outputs = [tmap["170"]]
6565
replace_with_coordconvac(graph, inputs, outputs)
6666

6767
# Remove the now-dangling subgraph.

0 commit comments

Comments
 (0)