!20332 Fix multi-processing Event set hang problem

Merge pull request !20332 from xiefangqi/md_fix_multiprocessing_event_hang
2021-07-15 11:18:59 +00:00 · 2021-07-15 11:18:59 +00:00 · 6fb544d86e
parent 51cd4215be 3e6489ea69
commit 6fb544d86e
1 changed files with 1 additions and 1 deletions
--- a/mindspore/dataset/engine/datasets.py
+++ b/mindspore/dataset/engine/datasets.py
@ -3683,7 +3683,7 @@ class SamplerFn:


 def _subprocess_handle(eof, signum, frame):
-    eof.set()
+    threading.Thread(target=eof.set()).start()


 def _generator_worker_loop(dataset, idx_queue, result_queue, eof, is_multiprocessing):