Skip to content

Commit 3a1d4a3

Browse files
committed
fix ci
1 parent eb89a8a commit 3a1d4a3

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

beginner_source/onnx/export_simple_model_to_onnx_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ def forward(self, x: torch.Tensor):
174174
"./image_classifier_model.onnx", providers=["CPUExecutionProvider"]
175175
)
176176

177-
onnxruntime_input = dict(zip(ort_session.get_inputs(), onnx_inputs))
177+
onnxruntime_input = {input_arg.name: input_value for input_arg, input_value in zip(ort_session.get_inputs(), onnx_inputs)}
178178

179179
# onnxruntime returns a list of outputs
180180
onnxruntime_outputs = ort_session.run(None, onnxruntime_input)[0]

beginner_source/onnx/intro_onnx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
.. note::
2323
Currently, you can choose either through `TorchScript https://pytorch.org/docs/stable/jit.html`_ or
2424
`ExportedProgram https://pytorch.org/docs/stable/export.html`_ to export the model to ONNX by the
25-
boolean parameter dynamo in `torch.onnx.export <https://pytorch.org/docs/stable/generated/torch.onnx.export.html>`_.
25+
boolean parameter dynamo in `torch.onnx.export <https://pytorch.org/docs/stable/onnx_torchscript.html#torch.onnx.export>`_.
2626
In this tutorial, we will focus on the ``ExportedProgram`` approach.
2727
2828
When setting ``dynamo=True``, the exporter will use `torch.export <https://pytorch.org/docs/stable/export.html>`_ to capture an ``ExportedProgram``,

0 commit comments

Comments
 (0)