提交 924a34ac 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!3639 fix GeneratorDataset time out

Merge pull request !3639 from yanghaitao/yht_generator_timeout_r0.6
......@@ -3115,7 +3115,7 @@ class _GeneratorWorker(multiprocessing.Process):
"""
Get function for worker result queue. Block with timeout.
"""
return self.res_queue.get(timeout=5)
return self.res_queue.get()
def __del__(self):
self.terminate()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册