We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ce5c1bf commit 859f4a9Copy full SHA for 859f4a9
onnx_diagnostic/torch_onnx/sbs_dataclasses.py
@@ -342,7 +342,7 @@ def dump(
342
)
343
onnx.save(model_inputs, model_inputs_file)
344
345
- model_outputs_file = os.path.join(folder, "model.inputs.onnx")
+ model_outputs_file = os.path.join(folder, "model.outputs.onnx")
346
model_outputs = select_model_inputs_outputs(
347
model, outputs=[i.name for i in submodel.graph.output]
348
0 commit comments