Accelerate processing

PINTO0309 · Jan 2, 2023 · 1f5a674 · 1f5a674
1 parent acf637d
commit 1f5a674
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/snc4onnx/__init__.py b/snc4onnx/__init__.py
@@ -1,3 +1,3 @@
 from snc4onnx.onnx_network_combine import combine, main
 
-__version__ = '1.0.10'
+__version__ = '1.0.11'
diff --git a/snc4onnx/onnx_network_combine.py b/snc4onnx/onnx_network_combine.py
@@ -497,11 +497,12 @@ def has_duplicates(seq):
             print(f'{Color.YELLOW}WARNING:{Color.RESET} {tracetxt}')
 
     ## 5. Restore a node's custom domain
-    combined_model_graph_nodes = combined_model.graph.node
-    for combined_model_graph_node in combined_model_graph_nodes:
-        for custom_domain_check_onnx_node in custom_domain_check_onnx_nodes:
-            if combined_model_graph_node.name == custom_domain_check_onnx_node.name:
-                combined_model_graph_node.domain = custom_domain_check_onnx_node.domain
+    if contains_custom_domain:
+        combined_model_graph_nodes = combined_model.graph.node
+        for combined_model_graph_node in combined_model_graph_nodes:
+            for custom_domain_check_onnx_node in custom_domain_check_onnx_nodes:
+                if combined_model_graph_node.name == custom_domain_check_onnx_node.name:
+                    combined_model_graph_node.domain = custom_domain_check_onnx_node.domain
 
     ## 6. Final save
     if output_onnx_file_path: