Functional Model: Bring the comment relevant to the method to the start of the method.
@@ -147,9 +147,10 @@ class Functional(training_lib.Model):
@tf.__internal__.tracking.no_automatic_dependency_tracking
def _init_graph_network(self, inputs, outputs):
- base_layer.keras_api_gauge.get_cell('Functional').set(True)
# This method is needed for Sequential to reinitialize graph network when
# layer is added or removed.
+
+ base_layer.keras_api_gauge.get_cell('Functional').set(True)
self._is_graph_network = True
# Normalize and set self.inputs, self.outputs.
