update

examples/conv_tasnet/step_2_train_model.py

@@ -144,7 +144,7 @@ def main():
         max_wave_value=32768.0,
         min_snr_db=config.min_snr_db,
         max_snr_db=config.max_snr_db,
-        # skip=825000,
+        skip=225000,
     )
     valid_dataset = DenoiseJsonlDataset(
         jsonl_file=args.valid_dataset,

examples/conv_tasnet_gan/step_2_train_model.py

@@ -169,7 +169,7 @@ def main():
         num_workers=0 if platform.system() == "Windows" else os.cpu_count() // 2,
         collate_fn=collate_fn,
         pin_memory=False,
-        prefetch_factor=8,
+        prefetch_factor=2,
     )
     valid_data_loader = DataLoader(
         dataset=valid_dataset,
@@ -180,7 +180,7 @@ def main():
         num_workers=0 if platform.system() == "Windows" else os.cpu_count() // 2,
         collate_fn=collate_fn,
         pin_memory=False,
-        prefetch_factor=8,
+        prefetch_factor=2,
     )
 
     # models