未验证 提交 683f8190 编写于 作者: R Ruibiao Chen 提交者: GitHub

Do not set return_merge in test_parallel_executor_run_cinn (#44809)

上级 221e1376
......@@ -99,11 +99,8 @@ def train(dot_save_dir, prefix, seed=1234):
feed = rand_data(img.name, label.name, iters)
loss_values = []
for step in range(iters):
loss_v = exe.run(compiled_program,
feed=feed[step],
fetch_list=[loss],
return_merged=False)
loss_values.append(loss_v[0][0][0])
loss_v = exe.run(compiled_program, feed=feed[step], fetch_list=[loss])
loss_values.append(loss_v[0][0])
return loss_values
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册