Update examples/get_started/torch-loader.py

iterative · Dec 12, 2024 · b92a9dd · b92a9dd
1 parent f3a384e
commit b92a9dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/get_started/torch-loader.py b/examples/get_started/torch-loader.py
@@ -68,7 +68,7 @@ def forward(self, x):
     train_loader = DataLoader(
         ds.to_pytorch(transform=transform),
         batch_size=25,
-        num_workers=4,
+        num_workers=max(4, os.cpu_count() or 2),
         persistent_workers=True,
         multiprocessing_context=multiprocessing.get_context("spawn"),
     )