Add debug info

7 years ago · ce4ddd7865
parent 19639e3106
commit ce4ddd7865
1 changed files with 8 additions and 0 deletions
--- a/python/paddle/reader/decorator.py
+++ b/python/paddle/reader/decorator.py
@ -325,6 +325,14 @@ def xmap_readers(mapper, reader, process_num, buffer_size, order=False):
            else:
                yield sample

+        assert finish == process_num
+        for w in workers:
+            print('join worker: ', w)
+            print('join worker: ', w.is_alive())
+            w.join()
+            print('join worker: ', w.is_alive())
+        del workers
+
    return xreader