reset yield_num after one pass

bda05dc3 · qiaolongfei · 198689c3 · bda05dc3
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

python/paddle/reader/decorator.py python/paddle/reader/decorator.py +1 -0

未找到文件。
--- 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