Bugfix data load queue.

MarcBS · Sep 14, 2019 · c2820e1 · c2820e1
1 parent 09da3e4
commit c2820e1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/keras_wrapper/dataset.py b/keras_wrapper/dataset.py
@@ -6,6 +6,7 @@
 import ntpath
 import os
 import random
+import time
 import re
 import sys
 #import threading
@@ -269,6 +270,7 @@ def generator(self):
             while out_queue.qsize() > 0:
                 data = out_queue.get()
                 yield(data)
+                time.sleep(0.1)
 
 
 class Homogeneous_Data_Batch_Generator(object):