Add opset to folder name

Author: xadupre

onnx_diagnostic/torch_models/validate.py

@@ -113,6 +113,7 @@ def _make_folder_name(
     dtype: Optional[Union[str, torch.dtype]] = None,
     device: Optional[Union[str, torch.device]] = None,
     subfolder: Optional[str] = None,
+    opset: Optional[int] = None,
 ) -> str:
     "Creates a filename unique based on the given options."
     els = [model_id.replace("/", "_")]
@@ -136,6 +137,8 @@ def _make_folder_name(
         else:
             raise AssertionError(f"unexpected value for device={device}, sdev={sdev!r}")
         els.append(sdev)
+    if opset is not None:
+        els.append(f"op{opset}")
     return "-".join(els)
 
 
@@ -412,7 +415,13 @@ def validate_model(
     folder_name = None
     if dump_folder:
         folder_name = _make_folder_name(
-            model_id, exporter, optimization, dtype=dtype, device=device, subfolder=subfolder
+            model_id,
+            exporter,
+            optimization,
+            dtype=dtype,
+            device=device,
+            subfolder=subfolder,
+            opset=opset,
         )
         dump_folder = os.path.join(dump_folder, folder_name)
         if not os.path.exists(dump_folder):