diff --git a/keras_wrapper/dataset.py b/keras_wrapper/dataset.py index f1db239..cef72a0 100644 --- 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):