added a warning in 09_dataloader.py for num_workers fail

patrickloeber · patrickloeber · commit 3e574310a5e0 · 2020-04-27T16:24:31.000+02:00
diff --git a/09_dataloader.py b/09_dataloader.py
@@ -58,6 +58,7 @@ def __len__(self):
 # Load whole dataset with DataLoader
 # shuffle: shuffle data, good for training
 # num_workers: faster loading with multiple subprocesses
+# !!! IF YOU GET AN ERROR DURING LOADING, SET num_workers TO 0 !!!
 train_loader = DataLoader(dataset=dataset,
                           batch_size=4,
                           shuffle=True,
