Change layer checkpoint dependencies to empty dict (#2249)

keras-team · Jan 4, 2024 · 60f5fa0 · 60f5fa0
1 parent 6315e97
commit 60f5fa0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_cv/utils/preset_utils.py b/keras_cv/utils/preset_utils.py
@@ -216,6 +216,6 @@ def legacy_load_weights(layer, weights_path):
         if cls.__name__ == "Functional":
             functional_cls = cls
     property = functional_cls._layer_checkpoint_dependencies
-    functional_cls._layer_checkpoint_dependencies = []
+    functional_cls._layer_checkpoint_dependencies = {}
     layer.load_weights(weights_path)
     functional_cls._layer_checkpoint_dependencies = property