提交 78742b4b 编写于 作者: W wangxiao1021

fix bugs

上级 36bbc8e7
...@@ -140,11 +140,8 @@ def create_iterator_fn(iterator, iterator_prefix, shape_and_dtypes, outname_to_p ...@@ -140,11 +140,8 @@ def create_iterator_fn(iterator, iterator_prefix, shape_and_dtypes, outname_to_p
def iterator_fn(): def iterator_fn():
v = verbose v = verbose
while True: for outputs in iterator:
# results = _zero_batch(shape_and_dtypes)
results = [None] * len(outname_to_pos) results = [None] * len(outname_to_pos)
outputs = next(iterator) # dict type
prefix = iterator_prefix prefix = iterator_prefix
for outname, val in outputs.items(): for outname, val in outputs.items():
task_outname = prefix + '.' + outname task_outname = prefix + '.' + outname
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册