提交 9ccc6889 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!3624 fix GeneratorDataset time out

Merge pull request !3624 from yanghaitao/yht_generator_timeout
......@@ -3118,7 +3118,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.
先完成此消息的编辑!
想要评论请 注册