提交 856f650a 编写于 作者: K Kexin Zhao

fix bug

上级 6edbf138
......@@ -198,10 +198,11 @@ def prepend_feed_ops(inference_program, feeded_var_names):
name='feed', type=core.VarDesc.VarType.FEED_MINIBATCH, persistable=True)
for i, name in enumerate(feeded_var_names):
out = global_block.var(name)
global_block.prepend_op(
type='feed',
inputs={'X': [feed_var]},
outputs={'Out': [name]},
outputs={'Out': [out]},
attrs={'col': i})
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册