Skip to content

Commit bce08d1

Browse files
committed
Python wrapper for MultiPassReader
1 parent 3bcffd4 commit bce08d1

File tree

1 file changed

+5
-0
lines changed
  • python/paddle/fluid/layers

1 file changed

+5
-0
lines changed

python/paddle/fluid/layers/io.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -314,6 +314,11 @@ def create_double_buffer_reader(reader, place=None):
314314
attrs)
315315

316316

317+
def create_multi_pass_reader(reader, pass_num):
318+
return __create_decorated_reader__('create_multi_pass_reader', reader,
319+
{'pass_num': int(pass_num)})
320+
321+
317322
def read_file(file_obj):
318323
helper = LayerHelper('read_file')
319324
out = [

0 commit comments

Comments
 (0)