diff --git a/python/paddle/reader/decorator.py b/python/paddle/reader/decorator.py index e06c151d309bc01f4686aaf19c14880643abf4d4..7b73a3a930cc2c53a54bf06bb16007e01ed5c9c3 100644 --- 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