maps_dir VS maps_path

aramis-lab · Sep 30, 2024 · 745fab5 · 745fab5
1 parent ecbcb35
commit 745fab5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/clinicadl/trainer/trainer.py b/clinicadl/trainer/trainer.py
@@ -78,7 +78,7 @@ def _init_maps_manager(self, config) -> MapsManager:
                 maps_path, verbose=None
             )  # TODO : precise which parameters in config are useful
         else:
-            parameters["maps_path"] = maps_path
+            # parameters["maps_path"] = maps_path
             return MapsManager(
                 maps_path, parameters, verbose=None
             )  # TODO : precise which parameters in config are useful
@@ -110,7 +110,7 @@ def from_json(cls, config_file: str | Path, maps_path: str | Path) -> Trainer:
         if not (config_file).is_file():
             raise FileNotFoundError(f"No file found at {str(config_file)}.")
         config_dict = patch_to_read_json(read_json(config_file))  # TODO : remove patch
-        config_dict["maps_path"] = maps_path
+        config_dict["maps_dir"] = maps_path
         config_object = create_training_config(config_dict["network_task"])(
             **config_dict
         )
@@ -240,10 +240,10 @@ def train(
                 split_df_dict = split_manager[split]
 
                 if self.config.model.multi_network:
-                    resume, first_network = self.init_first_network(resume, split)
+                    resume, first_network = self.init_first_network(False, split)
                     for network in range(first_network, self.maps_manager.num_networks):
                         self._train_single(
-                            split, split_df_dict, network=network, resume=False
+                            split, split_df_dict, network=network, resume=resume
                         )
                 else:
                     self._train_single(split, split_df_dict, resume=False)