19 feature request host model weights on zenodo

.gitignore

@@ -130,3 +130,4 @@ dmypy.json
 .vscode
 poetry.lock
 .DS_Store
+brainles_aurora/model_weights/*

brainles_aurora/inferer/inferer.py

@@ -221,4 +221,5 @@ def infer(
             self.data_handler.save_as_nifti(
                 postproc_data=out, output_file_mapping=output_file_mapping
             )
+        logger.info(f"{' Finished inference run ':=^80}")
         return out

brainles_aurora/inferer/model.py

@@ -43,7 +43,7 @@ def __init__(
         self.lib_path: str = Path(os.path.dirname(os.path.abspath(__file__)))
         self.model_weights_folder = self.lib_path.parent / "model_weights"
         if not self.model_weights_folder.exists():
-            download_model_weights(target_folder=str(self.lib_path.parent))
+            download_model_weights(target_folder=str(self.model_weights_folder))
 
     def load_model(
         self, inference_mode: InferenceMode, num_input_modalities: int
@@ -85,8 +85,7 @@ def _load_model(self, num_input_modalities: int) -> torch.nn.Module:
         # load weights
         weights_path = os.path.join(
             self.model_weights_folder,
-            self.inference_mode,
-            f"{self.config.model_selection}.tar",
+            f"{self.inference_mode}_{self.config.model_selection}.tar",
         )
         if not os.path.exists(weights_path):
             raise NotImplementedError(