From 695e7029a5408d7be56127d110f32d51c2a91482 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 19 Sep 2023 14:24:58 +0200
Subject: [PATCH] Revert "[data.datamodule] Experiment with fork/spawn setup"
This reverts commit 20dfcdb4ed74c1183b3c365a5b2469f1393e0aed.
---
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 cd90f828..8b158c28 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