Merge pull request facebookresearch#8 from bryant1410/patch-2

Fix `make_videodataset` when `num_workers=0`
secutron · Feb 19, 2024 · dc8396a · dc8396a
2 parents 5073cf0 + 37a20f8
commit dc8396a
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