Update simple_cerebros_random_search.py

Bug fix, get_best_model
david-thrower · Jun 12, 2024 · 3e89828 · 3e89828
1 parent 92dc2f1
commit 3e89828
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py b/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py
@@ -571,7 +571,9 @@ def run_random_search(self):
         return best
 
     def get_best_model(self):
-        best_model = tf.keras.models.load_model(self.best_model_path)
+        best_model_path = self.best_model_path
+        best_model = clone_model(tf.keras.models.load_model(best_model_path)
+        # best_model = tf.keras.models.load_model(self.best_model_path)
         return best_model
 
 # ->