提交 b8b478f0 编写于 作者: X xixiaoyao

Merge branch 'master' of https://github.com/xixiaoyao/PALM

......@@ -118,12 +118,10 @@ def create_joint_iterator_fn(iterators, iterator_prefixes, joint_shape_and_dtype
results = _zero_batch(joint_shape_and_dtypes)
outbuf = {}
for id in task_ids:
print(id)
outputs = next(iterators[id]) # dict type
outbuf[id] = outputs
prefix = iterator_prefixes[id]
for outname, val in outputs.items():
print(outname)
task_outname = prefix + '/' + outname
if outname in outname_to_pos:
......@@ -135,7 +133,6 @@ def create_joint_iterator_fn(iterators, iterator_prefixes, joint_shape_and_dtype
idx = outname_to_pos[task_outname]
val = _check_and_adapt_shape_dtype(val, joint_shape_and_dtypes[idx], message=task_outname+': ')
results[idx] = val
print('ok')
fake_batch = results
dev_count_bak = dev_count
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册