Merge pull request #50 from LennartPhil/model-fix

fixes a bug that occurs when trying to open the downloaded model weights
BrainLesion · Apr 22, 2024 · 5ad95d1 · 5ad95d1
2 parents 19c4895 + a53b5e1
commit 5ad95d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/brainles_aurora/inferer/model.py b/brainles_aurora/inferer/model.py
@@ -85,7 +85,7 @@ def _load_model(self, num_input_modalities: int) -> torch.nn.Module:
         # load weights
         weights_path = os.path.join(
             self.model_weights_folder,
-            f"{self.inference_mode}_{self.config.model_selection}.tar",
+            f"{self.inference_mode.value}_{self.config.model_selection.value}.tar",
         )
         if not os.path.exists(weights_path):
             raise NotImplementedError(