diff --git a/ppdet/data/parallel_map.py b/ppdet/data/parallel_map.py index 789fda1f2ed1c18a162e287562bbf09315c5211e..8d1bc6d0b24a8fbe1c245f7841c13f546f86897c 100644 --- a/ppdet/data/parallel_map.py +++ b/ppdet/data/parallel_map.py @@ -119,7 +119,7 @@ class ParallelMap(object): self._producer = threading.Thread( target=self._produce, args=('producer-' + id, self._source, self._inq)) - self._producer.daemon = True + self._producer.daemon = False self._consumers = [] self._consumer_endsig = {} @@ -130,7 +130,7 @@ class ParallelMap(object): target=self._consume, args=(consumer_id, self._inq, self._outq, self._worker)) self._consumers.append(p) - p.daemon = True + p.daemon = use_process setattr(p, 'id', consumer_id) if use_process: worker_set.add(p)