Skip to content

Commit 859f4a9

Browse files
committed
h
1 parent ce5c1bf commit 859f4a9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnx_diagnostic/torch_onnx/sbs_dataclasses.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -342,7 +342,7 @@ def dump(
342342
)
343343
onnx.save(model_inputs, model_inputs_file)
344344

345-
model_outputs_file = os.path.join(folder, "model.inputs.onnx")
345+
model_outputs_file = os.path.join(folder, "model.outputs.onnx")
346346
model_outputs = select_model_inputs_outputs(
347347
model, outputs=[i.name for i in submodel.graph.output]
348348
)

0 commit comments

Comments
 (0)