add workers to data loaders

dsethz · Oct 27, 2024 · 7682a53 · 7682a53
1 parent 0b3a71c
commit 7682a53
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/nuclai/utils/datamodule.py b/src/nuclai/utils/datamodule.py
@@ -280,12 +280,12 @@ def setup(self, stage: Optional[str] = None):
 
     def train_dataloader(self):
         return DataLoader(
-            self.data, batch_size=self.batch_size, shuffle=True, num_workers=0
+            self.data, batch_size=self.batch_size, shuffle=True, num_workers=4
         )
 
     def val_dataloader(self):
         return DataLoader(
-            self.data_val, batch_size=self.batch_size, num_workers=0
+            self.data_val, batch_size=self.batch_size, num_workers=4
         )
 
     def test_dataloader(self):