Update utils.py - Setting multiprocessor base context - Issue#240

aaronphilip19 · Apr 26, 2024 · 80fe01b · 80fe01b
1 parent 3d2a72b
commit 80fe01b
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/zamba/pytorch_lightning/utils.py b/zamba/pytorch_lightning/utils.py
@@ -65,11 +65,8 @@ def __init__(
             transform=transform,
             video_loader_config=video_loader_config,
         )
-        self.multiprocessing_context: BaseContext = (
-            None
-            if (num_workers == 0) or (multiprocessing_context is None)
-            else multiprocessing_context
-        )
+        self.multiprocessing_context: BaseContext = multiprocessing_context #Modified the multiprocessing context to not factor in num_workers decoupliing it
+
 
         super().__init__(*args, **kwargs)