speed up training

yxlllc · Mar 5, 2023 · 659ae71 · 659ae71
1 parent 7e00eed
commit 659ae71
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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)
+        return len(self.paths)