提交 701599cb 编写于 作者: Z Zeng Jinle 提交者: guru4elephant

fix ctr startup run bug, test=develop (#2794)

上级 e281b530
...@@ -147,6 +147,7 @@ def train_loop(args, train_program, py_reader, loss, auc_var, batch_auc_var, ...@@ -147,6 +147,7 @@ def train_loop(args, train_program, py_reader, loss, auc_var, batch_auc_var,
fluid.BuildStrategy.ReduceStrategy.Reduce if cpu_num > 1 \ fluid.BuildStrategy.ReduceStrategy.Reduce if cpu_num > 1 \
else fluid.BuildStrategy.ReduceStrategy.AllReduce else fluid.BuildStrategy.ReduceStrategy.AllReduce
exe.run(fluid.default_startup_program())
pe = fluid.ParallelExecutor( pe = fluid.ParallelExecutor(
use_cuda=False, use_cuda=False,
loss_name=loss.name, loss_name=loss.name,
...@@ -154,8 +155,6 @@ def train_loop(args, train_program, py_reader, loss, auc_var, batch_auc_var, ...@@ -154,8 +155,6 @@ def train_loop(args, train_program, py_reader, loss, auc_var, batch_auc_var,
build_strategy=build_strategy, build_strategy=build_strategy,
exec_strategy=exec_strategy) exec_strategy=exec_strategy)
exe.run(fluid.default_startup_program())
total_time = 0 total_time = 0
for pass_id in range(args.num_passes): for pass_id in range(args.num_passes):
pass_start = time.time() pass_start = time.time()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册