Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on 24 days ago

Commit

1c32993

1 Parent(s): 63d537a

update

Files changed (1) hide show

examples/conv_tasnet_gan/step_2_train_model.py CHANGED Viewed

@@ -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=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=2,
     )
     # models

         num_workers=0 if platform.system() == "Windows" else os.cpu_count() // 2,
         collate_fn=collate_fn,
         pin_memory=False,
+        prefetch_factor=8,
     )
     valid_data_loader = DataLoader(
         dataset=valid_dataset,
         num_workers=0 if platform.system() == "Windows" else os.cpu_count() // 2,
         collate_fn=collate_fn,
         pin_memory=False,
+        prefetch_factor=8,
     )
     # models