Browse Source

!3624 fix GeneratorDataset time out

Merge pull request !3624 from yanghaitao/yht_generator_timeout
tags/v0.7.0-beta
mindspore-ci-bot Gitee 6 years ago
parent
commit
9ccc6889eb
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/dataset/engine/datasets.py

+ 1
- 1
mindspore/dataset/engine/datasets.py View File

@@ -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()


Loading…
Cancel
Save