From 20dfcdb4ed74c1183b3c365a5b2469f1393e0aed Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 19 Sep 2023 14:07:46 +0200
Subject: [PATCH] [data.datamodule] Experiment with fork/spawn setup
---
src/ptbench/data/datamodule.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/ptbench/data/datamodule.py b/src/ptbench/data/datamodule.py
index 8b158c28..cd90f828 100644
--- a/src/ptbench/data/datamodule.py
+++ b/src/ptbench/data/datamodule.py
@@ -584,7 +584,7 @@ class ConcatDataModule(lightning.LightningDataModule):
num_workers = value or multiprocessing.cpu_count()
self._dataloader_multiproc["num_workers"] = num_workers
- if num_workers > 0 and sys.platform == "darwin":
+ if num_workers > 0: # and sys.platform == "darwin":
self._dataloader_multiproc[
"multiprocessing_context"
] = multiprocessing.get_context("spawn")
--
GitLab