未验证 提交 d47e15e0 编写于 作者: R ruri 提交者: GitHub

set lr.persistable=True in image classification (#2469)

* set lr.persistable=True

* comment fuse flag
上级 43696bbe
......@@ -328,6 +328,7 @@ def build_program(is_train, main_prog, startup_prog, args):
else:
optimizer.minimize(avg_cost)
global_lr = optimizer._global_learning_rate()
global_lr.persistable=True
build_program_out.append(global_lr)
return build_program_out
......@@ -429,7 +430,7 @@ def train(args):
build_strategy = fluid.BuildStrategy()
build_strategy.memory_optimize = args.with_mem_opt
build_strategy.enable_inplace = args.with_inplace
build_strategy.fuse_all_reduce_ops=1
#build_strategy.fuse_all_reduce_ops=1
exec_strategy = fluid.ExecutionStrategy()
exec_strategy.num_iteration_per_drop_scope = 10
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册