keras.models.functional.functional_from_config: Fix loading nested fu…

…nctional models
keras-team · Apr 14, 2024 · 13c3c91 · 13c3c91
1 parent b267f93
commit 13c3c91
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/keras/models/functional.py b/keras/models/functional.py
@@ -553,6 +553,10 @@ def process_layer(layer_data):
     def get_tensor(layer_name, node_index, tensor_index):
         assert layer_name in created_layers
         layer = created_layers[layer_name]
+        if issubclass(layer.__class__, Functional):
+            # Functional models start with a pre-existing node
+            # linking their input to output.
+            node_index -= 1
         layer_output_tensors = layer._inbound_nodes[node_index].output_tensors
         return layer_output_tensors[tensor_index]