提交 b24a2c20 编写于 作者: Y yangyaming

Bug fix which results main process hanging.

上级 4da0c0c6
...@@ -224,7 +224,7 @@ class DataReader(object): ...@@ -224,7 +224,7 @@ class DataReader(object):
finished_process_num = 0 finished_process_num = 0
while True: while finished_process_num < process_num:
sample = sample_queue.get() sample = sample_queue.get()
if isinstance(sample, EpochEndSignal): if isinstance(sample, EpochEndSignal):
finished_process_num += 1 finished_process_num += 1
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册