提交 860aaae2 编写于 作者: S SunGaofeng

fix error when join thread in the end of data reading

上级 397e3a8a
...@@ -332,7 +332,7 @@ def make_multi_reader(filelist, batch_size, sample_times, is_training, shuffle, ...@@ -332,7 +332,7 @@ def make_multi_reader(filelist, batch_size, sample_times, is_training, shuffle,
else: else:
yield sample yield sample
for i in range(len(p_list)): for i in range(len(p_list)):
p_list[i].terminate() if p_list[i].is_alive():
p_list[i].join() p_list[i].join()
return queue_reader return queue_reader
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册