diff --git a/src/gretel_synthetics/timeseries_dgan/dgan.py b/src/gretel_synthetics/timeseries_dgan/dgan.py index 68e48880..35afee66 100644 --- 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(