fix(seg): increase the number of workers for dataloaders

archai/algos/evolution_pareto_image_seg/segmentation_trainer.py

@@ -172,8 +172,8 @@ class SegmentationTrainer():
         self.val_dataset = FaceSynthetics(self.data_dir, subset='validation', val_size=val_size,
                                           img_size=(img_size, img_size), augmentation=augmentation)
 
-        self.tr_dataloader = DataLoader(self.tr_dataset, batch_size=batch_size, num_workers=4, shuffle=True, pin_memory=True)
-        self.val_dataloader = DataLoader(self.val_dataset, batch_size=batch_size, num_workers=4, shuffle=False, pin_memory=True)
+        self.tr_dataloader = DataLoader(self.tr_dataset, batch_size=batch_size, num_workers=8, shuffle=True)
+        self.val_dataloader = DataLoader(self.val_dataset, batch_size=batch_size, num_workers=8, shuffle=False)
 
         self.model = LightningModelWrapper(model, criterion_name=criterion_name, lr=lr,
                                            img_size=img_size, lr_exp_decay_gamma=lr_exp_decay_gamma)