save chief's state before starting oracle server (keras-team#977)

invernizzi · Nov 17, 2023 · 92d713a · 92d713a
1 parent 338b505
commit 92d713a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keras_tuner/engine/base_tuner.py b/keras_tuner/engine/base_tuner.py
@@ -215,6 +215,7 @@ def search(self, *fit_args, **fit_kwargs):
             # Avoid import at the top, to avoid inconsistent protobuf versions.
             from keras_tuner.distribute import oracle_chief
 
+            self.save()
             oracle_chief.start_server(self.oracle)
             return