update logging for debugging

MannLabs · Dec 28, 2023 · d0058f7 · d0058f7
1 parent b69be6d
commit d0058f7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/sparcscore/pipeline/workflows.py b/src/sparcscore/pipeline/workflows.py
@@ -724,10 +724,15 @@ def cellpose_segmentation(self, input_image):
 
         try:
             current = multiprocessing.current_process()
+            self.log(f"current process: {current}")
             cpu_name = current.name
+            self.log(f"cpu name: {cpu_name}")
             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}")
             gpu_id = gpu_id_list[cpu_id]
+            self.log("gpu id: {gpu_id}")
             self.log(f'starting process on GPU {gpu_id}')
             status = "multi_GPU"
         except: