提交 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,
else:
yield sample
for i in range(len(p_list)):
p_list[i].terminate()
p_list[i].join()
if p_list[i].is_alive():
p_list[i].join()
return queue_reader
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册