提交 b24a2c20 编写于 作者: Y yangyaming

Bug fix which results main process hanging.

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