Browse Source

fix pool not running bug

tags/v1.1.0
liyong 5 years ago
parent
commit
8e3fea27d7
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

@@ -2211,7 +2211,7 @@ class _PythonCallable:
self.idx = idx

def __call__(self, *args):
if self.pool is not None:
if self.pool is not None and self.pool._state == 0: # pylint: disable=W0212
# This call will send the tensors along with Python callable index to the process pool.
# Block, yield GIL. Current thread will reacquire GIL once result is returned.
result = self.pool.apply_async(_pyfunc_worker_exec, [self.idx, *args])


Loading…
Cancel
Save