From 4c0a2796e194eabd62ebbbcb1a0e91f6c0cc97b6 Mon Sep 17 00:00:00 2001 From: David Thrower Date: Wed, 12 Jun 2024 19:05:56 -0400 Subject: [PATCH] Update simple_cerebros_random_search.py --- .../simple_cerebros_random_search.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py b/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py index e461ba7..a398e05 100644 --- a/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py +++ b/cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py @@ -572,10 +572,14 @@ def run_random_search(self): def get_best_model(self): best_model_path = self.best_model_path + # best_model =\ + # tf.keras.models.clone_model( + # tf.keras.models.load_model(best_model_path, custom_objects={"TernaryDenseLayer": units.units.TernaryDenseLayer})) best_model =\ - tf.keras.models.clone_model( - tf.keras.models.load_model(best_model_path)) - # best_model = tf.keras.models.load_model(self.best_model_path) + tf.keras.models.load_model( + self.best_model_path, + custom_objects=\ + {"TernaryDenseLayer": units.units.TernaryDenseLayer}) return best_model # ->