Fix make_videodataset when num_workers=0

zlu8899 · Feb 18, 2024 · 37a20f8 · 37a20f8
1 parent 2dfdfac
commit 37a20f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/datasets/video_dataset.py b/src/datasets/video_dataset.py
@@ -94,7 +94,7 @@ def make_videodataset(
         drop_last=drop_last,
         pin_memory=pin_mem,
         num_workers=num_workers,
-        persistent_workers=True)
+        persistent_workers=num_workers > 0)
     logger.info('VideoDataset unsupervised data loader created')
 
     return dataset, data_loader, dist_sampler