Raise errors when num_workers == 0 in DataLoader (#4019)

Erotemic · soumith · commit 7763c6f87109 · 2018-02-06T20:51:28.000-08:00
diff --git a/torch/utils/data/dataloader.py b/torch/utils/data/dataloader.py
@@ -295,6 +295,10 @@ def __init__(self, dataset, batch_size=1, shuffle=False, sampler=None, batch_sam
         if sampler is not None and shuffle:
             raise ValueError('sampler is mutually exclusive with shuffle')
 
+        if self.num_workers < 0:
+            raise ValueError('num_workers cannot be negative; '
+                             'use num_workers=0 to disable multiprocessing.')
+
         if batch_sampler is None:
             if sampler is None:
                 if shuffle:
