Added optimization process for final graph

PINTO0309 · May 7, 2022 · 5f0cd81 · 5f0cd81
1 parent 9115344
commit 5f0cd81
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 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.5'
+__version__ = '1.0.6'
diff --git a/snc4onnx/onnx_network_combine.py b/snc4onnx/onnx_network_combine.py
@@ -386,7 +386,9 @@ def has_duplicates(seq):
                 continue
             break
         gs_combined_model.inputs[0].name = gs_combined_model.inputs[0].name.lstrip(src_prefix)
-        combined_model = gs.export_onnx(gs_combined_model)
+
+    gs_combined_model.cleanup().toposort()
+    combined_model = gs.export_onnx(gs_combined_model)
 
     ## 4. Optimize
     try: