remove non tensors from middle layers (#160)

* remove non tensors from middle layers * 0.0.140
tensorleap · Jul 28, 2024 · 75a232a · 75a232a
1 parent 1d4a483
commit 75a232a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/onnx2kerastl/converter.py b/onnx2kerastl/converter.py
@@ -277,7 +277,8 @@ def onnx_to_keras(onnx_model, input_names, name_policy=None, verbose=True, chang
             for inp in node.input:
                 keras_middle_outputs.pop(inp, None)
             # add node to middle map
-            keras_middle_outputs[node_name] = layers[node_name]
+            if tf.is_tensor(layers[node_name]):
+                keras_middle_outputs[node_name] = layers[node_name]
 
             if isinstance(keras_names, list):
                 keras_names = keras_names[0]

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "onnx2kerastl"
-version = "0.0.139"
+version = "0.0.140"
 description = ""
 authors = ["dorhar <[email protected]>"]
 license = "MIT"