Separated the base context inference for the multiprocessor for the t…

…rainer Issue drivendataorg#240
sambujangfofana · Apr 26, 2024 · 879a776 · 879a776
1 parent 70e79f2
commit 879a776
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 (multiprocessing_context is None) or (num_workers == 0)
-            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)