Updated workers config

Author: C-Achard

napari_cellseg3d/code_models/worker_inference.py

@@ -181,7 +181,6 @@ def log_parameters(self):
     def load_folder(self):
         images_dict = self.create_inference_dict(self.config.images_filepaths)
 
-        # TODO : better solution than loading first image always ?
         data_check = LoadImaged(keys=["image"])(images_dict[0])
         check = data_check["image"].shape
         pad = utils.get_padding_dim(check)

napari_cellseg3d/code_models/worker_training.py

@@ -1273,14 +1273,14 @@ def get_patch_loader_func(num_samples):
                 train_dataset,
                 batch_size=self.config.batch_size,
                 shuffle=True,
-                num_workers=2,
+                num_workers=self.config.num_workers,
                 collate_fn=pad_list_data_collate,
             )
 
             validation_loader = DataLoader(
                 validation_dataset,
                 batch_size=self.config.batch_size,
-                num_workers=2,
+                num_workers=self.config.num_workers,
             )
             logger.info("\nDone")