diff --git a/brainles_aurora/inferer/model.py b/brainles_aurora/inferer/model.py index 71b69a0..97b6154 100644 --- 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(