[fix] Make queue.put() blocking again (OpenNMT#1455)

Fix OpenNMT#1454 .
AMusketeer · Jun 4, 2019 · 065c99f · 065c99f
1 parent b731e04
commit 065c99f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -124,7 +124,7 @@ def next_batch(device_id):
 
         # hack to dodge unpicklable `dict_keys`
         b.fields = list(b.fields)
-        q.put(b, False)
+        q.put(b)
         b = next_batch(device_id)