Add friendly error msg to py_reader

python/paddle/fluid/layers/io.py

@@ -28,6 +28,7 @@
     default_startup_program, program_guard, Program, Variable
 from ..layer_helper import LayerHelper
 from ..unique_name import generate as unique_name
+import logging
 
 __all__ = [
     'data', 'open_files', 'read_file', 'shuffle', 'batch', 'double_buffer',
@@ -590,6 +591,7 @@ def __provider_thread__():
                 feed_queue.close()
             except Exception as ex:
                 feed_queue.close()
+                logging.warn('Your decorated reader has raised an exception!')
                 raise ex
 
         reader.thread = threading.Thread(target=__provider_thread__)

python/paddle/fluid/reader.py

@@ -23,6 +23,7 @@
 from .data_feeder import DataFeeder, BatchedTensorProvider, ListTensorProvider
 from .layers.io import monkey_patch_reader_methods, _copy_reader_var_, double_buffer
 from .unique_name import UniqueNameGenerator
+import logging
 
 __all__ = ['PyReader']
 
@@ -441,6 +442,7 @@ def __thread_main__():
                 self._queue.close()
             except Exception as ex:
                 self._queue.close()
+                logging.warn('Your decorated reader has raised an exception!')
                 raise ex
 
         self._thread = threading.Thread(target=__thread_main__)