From 78742b4bb6ad2621915c5a465eab688b106a89ee Mon Sep 17 00:00:00 2001
From: wangxiao1021 <zko1021@foxmail.com>
Date: Tue, 18 Feb 2020 19:45:51 +0800
Subject: [PATCH] fix bugs

---
 paddlepalm/utils/reader_helper.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/paddlepalm/utils/reader_helper.py b/paddlepalm/utils/reader_helper.py
index 05e5c5a..4dc5e19 100644
--- a/paddlepalm/utils/reader_helper.py
+++ b/paddlepalm/utils/reader_helper.py
@@ -140,11 +140,8 @@ def create_iterator_fn(iterator, iterator_prefix, shape_and_dtypes, outname_to_p
     
     def iterator_fn():
         v = verbose
-        while True:
-            # results = _zero_batch(shape_and_dtypes)
+        for outputs in iterator:
             results = [None] * len(outname_to_pos)
-
-            outputs = next(iterator) # dict type
             prefix = iterator_prefix
             for outname, val in outputs.items():
                 task_outname = prefix + '.' + outname
-- 
GitLab