fixed model compilation in model loading

aertslab · Jun 21, 2024 · df5ec90 · df5ec90
1 parent 9f32496
commit df5ec90
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/crested/tl/_crested.py b/src/crested/tl/_crested.py
@@ -182,16 +182,18 @@ def _initialize_logger(logger_type: str | None, project_name: str, run_name: str
 
         return run, callbacks
 
-    def load_model(self, model_path: os.PathLike) -> None:
+    def load_model(self, model_path: os.PathLike, compile: bool = False) -> None:
         """
         Load a (pretrained) model from a file.
 
         Parameters
         ----------
         model_path : os.PathLike
             Path to the model file.
+        compile: bool
+            Compile model after loading.
         """
-        self.model = tf.keras.models.load_model(model_path, compile=True)
+        self.model = tf.keras.models.load_model(model_path, compile=compile)
 
     def fit(
         self,