diff --git a/snc4onnx/__init__.py b/snc4onnx/__init__.py index 2e60d27..061cac1 100644 --- 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 index 85c69db..51455af 100644 --- 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: