Explicit opset version in resnet50 export test, formatting

mortont · mortont · commit 3eaa9139bcaf · 2024-11-04T16:30:32.000-05:00
diff --git a/python/export_resnet.py b/python/export_resnet.py
@@ -15,4 +15,5 @@
     output_names=["output"],
     dynamic_axes={"input": {0: "batch_size"}, "output": {0: "batch_size"}},
     export_params=True,
+    opset_version=19,
 )
diff --git a/python/multi_input.py b/python/multi_input.py
@@ -37,5 +37,5 @@ def forward(self, x, y):
         "x": {0: "batch_size"},
         "y": {0: "batch_size"},
     },
-    opset_version=19
+    opset_version=19,
 )

Original file line number	Diff line number	Diff line change
`@@ -15,4 +15,5 @@`
`15`	`15`	`output_names=["output"],`
`16`	`16`	`dynamic_axes={"input": {0: "batch_size"}, "output": {0: "batch_size"}},`
`17`	`17`	`export_params=True,`
	`18`	`+ opset_version=19,`
`18`	`19`	`)`