Skip to content

Commit 02900ae

Browse files
committed
fix output names
1 parent 54a975b commit 02900ae

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

onnx_diagnostic/_command_lines_parser.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -548,7 +548,9 @@ def _cmd_validate(argv: List[Any]):
548548
repeat=args.repeat,
549549
warmup=args.warmup,
550550
inputs2=args.inputs2,
551-
output_names=args.outnames.strip().split(","),
551+
output_names=(
552+
None if len(args.outnames.strip()) < 2 else args.outnames.strip().split(",")
553+
),
552554
)
553555
print("")
554556
print("-- summary --")

onnx_diagnostic/torch_models/validate.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -435,6 +435,7 @@ def validate_model(
435435
)
436436
print(f"[validate_model] exporter={exporter!r}, optimization={optimization!r}")
437437
print(f"[validate_model] dump_folder={dump_folder!r}")
438+
print(f"[validate_model] output_names={output_names}")
438439
summary["model_id"] = model_id
439440
summary["model_subfolder"] = subfolder or ""
440441

0 commit comments

Comments
 (0)