experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix a python exception issue on using Sequence object of keras with MirroredStrategy

This commit is contained in:
Kim, Young Soo 2020-05-28 13:51:35 +09:00
parent dedac5053f
commit 4bc4fa3cb7
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tensorflow/python/keras/utils/data_utils.py
View File

@ -886,15 +886,18 @@ class OrderedEnqueuer(SequenceEnqueuer):
`(inputs, targets)` or `(inputs, targets)` or
`(inputs, targets, sample_weights)`. `(inputs, targets, sample_weights)`.
""" """
try: while self.is_running():
while self.is_running(): try:
inputs = self.queue.get(block=True).get() inputs = self.queue.get(block=True, timeout=5).get()
self.queue.task_done() if self.is_running():
self.queue.task_done()
if inputs is not None: if inputs is not None:
yield inputs yield inputs
except Exception: # pylint: disable=broad-except except queue.Empty:
self.stop() pass
six.reraise(*sys.exc_info()) except Exception: # pylint: disable=broad-except
self.stop()
six.reraise(*sys.exc_info())
def init_pool_generator(gens, random_seed=None, id_queue=None): def init_pool_generator(gens, random_seed=None, id_queue=None):