Skip to content

Commit 403fb44

Browse files
committed
fix dump folder name
1 parent 80e2ebf commit 403fb44

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

onnx_diagnostic/torch_models/validate.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,7 @@ def _make_folder_name(
112112
device: Optional[Union[str, torch.device]] = None,
113113
subfolder: Optional[str] = None,
114114
opset: Optional[int] = None,
115+
drop_inputs: Optional[List[str]] = None,
115116
) -> str:
116117
"Creates a filename unique based on the given options."
117118
els = [model_id.replace("/", "_")]
@@ -137,6 +138,9 @@ def _make_folder_name(
137138
els.append(sdev)
138139
if opset is not None:
139140
els.append(f"op{opset}")
141+
if drop_inputs:
142+
ii = "-".join(f"{s[0]}{s[-1]}" for s in drop_inputs)
143+
els.append(f"I-{ii.upper()}")
140144
return "-".join(els)
141145

142146

@@ -449,6 +453,7 @@ def validate_model(
449453
device=device,
450454
subfolder=subfolder,
451455
opset=opset,
456+
drop_inputs=drop_inputs,
452457
)
453458
dump_folder = os.path.join(dump_folder, folder_name)
454459
if not os.path.exists(dump_folder):

0 commit comments

Comments
 (0)