提交 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): ...@@ -3118,7 +3118,7 @@ class _GeneratorWorker(multiprocessing.Process):
""" """
Get function for worker result queue. Block with timeout. Get function for worker result queue. Block with timeout.
""" """
return self.res_queue.get(timeout=5) return self.res_queue.get()
def __del__(self): def __del__(self):
self.terminate() self.terminate()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册