Use config options in model init

Noble-Lab · Sep 22, 2023 · 0ea1b7d · 0ea1b7d
1 parent 0d312e6
commit 0ea1b7d
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/casanovo/denovo/model_runner.py b/casanovo/denovo/model_runner.py
@@ -252,16 +252,13 @@ def initialize_model(self, train: bool) -> None:
         # otherwise use the provided configuration.
         device = torch.empty(1).device  # Use the default device.
         try:
-            self.model = Spec2Pep.load_from_checkpoint(
-                self.model_filename,
-                map_location=device,
-            )
-        except RuntimeError:
             self.model = Spec2Pep.load_from_checkpoint(
                 self.model_filename,
                 map_location=device,
                 **model_params,
             )
+        except RuntimeError:
+            raise RuntimeError("Mismatching parameters between loaded model and config file")
 
     def initialize_data_module(
         self,