From 5c9b0657df00911a90818d633d8eec26efcb789d Mon Sep 17 00:00:00 2001 From: yanghaitao Date: Tue, 28 Jul 2020 21:59:21 +0800 Subject: [PATCH] fix GeneratorDataset time out --- mindspore/dataset/engine/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py index afacb68089c..471565f7cec 100644 --- 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()