From 7d574013ad9c684019802ac1a082cb22b355bc91 Mon Sep 17 00:00:00 2001
From: Benedikt Hilmes <benedikt.hilmes@rwth-aachen.de>
Date: Wed, 2 Apr 2025 14:55:29 +0200
Subject: [PATCH] update dataloader

---
 returnn/torch/engine.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/returnn/torch/engine.py b/returnn/torch/engine.py
index 4e1b18c608..73180a0b4c 100644
--- a/returnn/torch/engine.py
+++ b/returnn/torch/engine.py
@@ -377,8 +377,8 @@ def _create_data_loader(self, dataset: Dataset) -> DataLoader:
             dataset=batches_dataset,
             batch_size=None,
             num_workers=num_workers,
-            multiprocessing_context="spawn",
-            persistent_workers=True,
+            multiprocessing_context="spawn" if num_workers > 0 else None,
+            persistent_workers=num_workers > 0,
         )
 
     def run_train_step(