!3624 fix GeneratorDataset time out

Merge pull request !3624 from yanghaitao/yht_generator_timeout
2020-07-29 14:27:18 +08:00 · 2020-07-29 14:27:18 +08:00 · 9ccc6889eb
parent 4834a6b347 5c9b0657df
commit 9ccc6889eb
1 changed files with 1 additions and 1 deletions
--- a/mindspore/dataset/engine/datasets.py
+++ b/mindspore/dataset/engine/datasets.py
@ -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()