提交 b720b3a5 编写于 作者: Y Yancey1989

fix fluid benchmark script

上级 2fd1bf2e
...@@ -91,7 +91,8 @@ def dist_transpile(trainer_id, args, train_prog, startup_prog): ...@@ -91,7 +91,8 @@ def dist_transpile(trainer_id, args, train_prog, startup_prog):
program=train_prog, program=train_prog,
pservers=pserver_endpoints, pservers=pserver_endpoints,
trainers=trainers, trainers=trainers,
sync_mode=not args.async_mode) sync_mode=not args.async_mode,
startup_program=startup_prog)
if training_role == "PSERVER": if training_role == "PSERVER":
pserver_program = t.get_pserver_program(current_endpoint) pserver_program = t.get_pserver_program(current_endpoint)
pserver_startup_program = t.get_startup_program( pserver_startup_program = t.get_startup_program(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册