Bug fix which results main process hanging.

b24a2c20 · yangyaming · 4da0c0c6 · b24a2c20
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

fluid/DeepASR/data_utils/parallel_reader.py fluid/DeepASR/data_utils/parallel_reader.py +1 -1

未找到文件。
--- a/fluid/DeepASR/data_utils/parallel_reader.py
+++ b/fluid/DeepASR/data_utils/parallel_reader.py
@@ -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