!3639 fix GeneratorDataset time out

Merge pull request !3639 from yanghaitao/yht_generator_timeout_r0.6
2020-07-29 14:42:38 +08:00 · 2020-07-29 14:42:38 +08:00 · 924a34acb8
parent db01f3eafe 248130e5d1
commit 924a34acb8
1 changed files with 1 additions and 1 deletions
--- a/mindspore/dataset/engine/datasets.py
+++ b/mindspore/dataset/engine/datasets.py
@ -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()