diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py index caf3857e2060e8e863faf9ebff987544fe539d01..fa9c1099fcf20ba1b6193d6b2ba8dd2116c41ce0 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()