diff --git a/data_loaders.py b/data_loaders.py index fb7eee8..0b5c5de 100644 --- a/data_loaders.py +++ b/data_loaders.py @@ -60,7 +60,7 @@ def get_data_loaders(args, whole_audio=False): data_train , batch_size=args.train.batch_size if not whole_audio else 1, shuffle=True, - num_workers=1, + num_workers=0, pin_memory=True ) data_valid = AudioDataset( @@ -74,7 +74,7 @@ def get_data_loaders(args, whole_audio=False): data_valid, batch_size=1, shuffle=False, - num_workers=1, + num_workers=0, pin_memory=True ) return loader_train, loader_valid @@ -202,4 +202,4 @@ def get_data(self, name, data_buffer): return dict(audio=audio, f0=f0_frames, volume=volume_frames, units=units, name=name) def __len__(self): - return len(self.paths) \ No newline at end of file + return len(self.paths)