fixes a bug that occurs when trying to open the downloaded model weights

BrainLesion · Apr 22, 2024 · a53b5e1 · a53b5e1
1 parent 19c4895
commit a53b5e1
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(