diff --git a/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py b/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py index 125582c..551942c 100644 --- 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 # ->