RDS-867: Remove num_workers and related params from dgan DataLoader

GitOrigin-RevId: 312d736944acc69e7068f5fe70cd89ca529fa0d4
gretelai · Mar 19, 2024 · 3175102 · 3175102
1 parent 5be7b76
commit 3175102
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/src/gretel_synthetics/timeseries_dgan/dgan.py b/src/gretel_synthetics/timeseries_dgan/dgan.py
@@ -788,11 +788,7 @@ def _train(
             self.config.batch_size,
             shuffle=True,
             drop_last=drop_last,
-            num_workers=2,
-            prefetch_factor=4,
-            persistent_workers=True,
             pin_memory=True,
-            multiprocessing_context="fork",
         )
 
         opt_discriminator = torch.optim.Adam(