Python wrapper for MultiPassReader

7 years ago · bce08d19cc
parent 3bcffd495e
commit bce08d19cc
1 changed files with 5 additions and 0 deletions
--- a/python/paddle/fluid/layers/io.py
+++ b/python/paddle/fluid/layers/io.py
@ -314,6 +314,11 @@ def create_double_buffer_reader(reader, place=None):
                                       attrs)


+def create_multi_pass_reader(reader, pass_num):
+    return __create_decorated_reader__('create_multi_pass_reader', reader,
+                                       {'pass_num': int(pass_num)})
+
+
 def read_file(file_obj):
    helper = LayerHelper('read_file')
    out = [