File tree Expand file tree Collapse file tree 1 file changed +5
-0
lines changed
onnx_diagnostic/torch_models Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Original file line number Diff line number Diff 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 ):
You can’t perform that action at this time.
0 commit comments