pb with init model ? or DDP

aramis-lab · Oct 8, 2024 · 40b2452 · 40b2452
1 parent a0cabba
commit 40b2452
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/clinicadl/maps_manager/maps_manager.py b/clinicadl/maps_manager/maps_manager.py
@@ -575,7 +575,7 @@ def _init_model(
         nb_unfrozen_layer: int = 0,
         split: Optional[int] = None,
         resume: bool = False,
-        gpu: bool = False,
+        gpu: Optional[bool] = None,
         network: Optional[int] = None,
     ):
         """

diff --git a/clinicadl/trainer/trainer.py b/clinicadl/trainer/trainer.py
@@ -450,7 +450,7 @@ def _train_single(
                 self.config.validation.selection_metrics,
             )
 
-            self._erase_tmp(split)
+            self.maps_manager._erase_tmp(split)
 
     def _train_ssda(
         self,
@@ -662,7 +662,7 @@ def _train_ssda(
                 self.config.validation.selection_metrics,
             )
 
-            self._erase_tmp(split)
+            self.maps_manager._erase_tmp(split)
 
     def _train(
         self,