fix GeneratorDataset time out

2020-07-28 21:59:21 +08:00 · 2020-07-28 21:59:21 +08:00 · 5c9b0657df
parent efb12d5e1e
commit 5c9b0657df
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()