further logging improvements

add further descriptive logging to try and isolate issue with 4 GPUs per node runs
MannLabs · Dec 28, 2023 · e19e265 · e19e265
1 parent d0058f7
commit e19e265
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/sparcscore/pipeline/segmentation.py b/src/sparcscore/pipeline/segmentation.py
@@ -819,6 +819,7 @@ def process(self, input_image):
 
         if available_GPUs >= 1:
             n_processes = processes_per_GPU * available_GPUs  
+            self.log(f"Proceeding in segmentation with {n_processes} number of processes.")
         else:
             n_processes = self.config["threads"]
             available_GPUs = 1 #default to 1 GPU if non are available and a CPU only method is run

diff --git a/src/sparcscore/pipeline/workflows.py b/src/sparcscore/pipeline/workflows.py
@@ -730,9 +730,9 @@ def cellpose_segmentation(self, input_image):
             gpu_id_list = current.gpu_id_list
             self.log(f"gpu id list: {gpu_id_list}")
             cpu_id = int(cpu_name[cpu_name.find('-') + 1:]) - 1
-            self.log("cpu id: {cpu_id}")
+            self.log(f"cpu id: {cpu_id}")
             gpu_id = gpu_id_list[cpu_id]
-            self.log("gpu id: {gpu_id}")
+            self.log(f"gpu id: {gpu_id}")
             self.log(f'starting process on GPU {gpu_id}')
             status = "multi_GPU"
         except: