change num_workers to 0

janrth · Feb 20, 2023 · ed05768 · ed05768
1 parent edf9922
commit ed05768
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_chatbot.py b/train_chatbot.py
@@ -75,7 +75,7 @@ def detokenize_fct(x):
 train_loader = DataLoader(dataset=dataset,
                           batch_size=batch_size,
                           shuffle=True,
-                          num_workers=2)
+                          num_workers=0)
 
 # Loss and optimizer
 criterion = nn.CrossEntropyLoss()