“88b9202c48ec206fb4a71720cfaca22519b59a7e”上不存在“python/paddle/v2/fluid/tests/unittests/test_program.py”
未验证 提交 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): ...@@ -99,11 +99,8 @@ def train(dot_save_dir, prefix, seed=1234):
feed = rand_data(img.name, label.name, iters) feed = rand_data(img.name, label.name, iters)
loss_values = [] loss_values = []
for step in range(iters): for step in range(iters):
loss_v = exe.run(compiled_program, loss_v = exe.run(compiled_program, feed=feed[step], fetch_list=[loss])
feed=feed[step], loss_values.append(loss_v[0][0])
fetch_list=[loss],
return_merged=False)
loss_values.append(loss_v[0][0][0])
return loss_values return loss_values
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册