reset yield_num after one pass

7 years ago · bda05dc387
parent 198689c3bb
commit bda05dc387
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/reader/decorator.py
+++ b/python/paddle/reader/decorator.py
@ -522,6 +522,7 @@ def fake(reader, data_num):
        while fake_reader.yield_num < data_num:
            yield fake_reader.data
            fake_reader.yield_num += 1
+        fake_reader.yield_num = 0

    fake_reader.data = None
    fake_reader.yield_num = 0