number of workers as config parameter

2026-09-09 11:16:00 +08:00 · 2020-05-15 10:19:52 +02:00
parent 7a9247f91f
commit 65b9c7d3d6
2 changed files with 2 additions and 1 deletions
@@ -44,7 +44,7 @@ def setup_loader(ap, is_val=False, verbose=False):
        loader = DataLoader(dataset,
                            batch_size=c.num_speakers_in_batch,
                            shuffle=False,
-                            num_workers=0,
+                            num_workers=c.num_loader_workers,
                            collate_fn=dataset.collate_fn)
    return loader