fix fluid benchmark script

7 years ago · b720b3a58f
parent 2fd1bf2ea6
commit b720b3a58f
1 changed files with 2 additions and 1 deletions
--- a/benchmark/fluid/fluid_benchmark.py
+++ b/benchmark/fluid/fluid_benchmark.py
@ -91,7 +91,8 @@ def dist_transpile(trainer_id, args, train_prog, startup_prog):
        program=train_prog,
        pservers=pserver_endpoints,
        trainers=trainers,
-        sync_mode=not args.async_mode)
+        sync_mode=not args.async_mode,
+        startup_program=startup_prog)
    if training_role == "PSERVER":
        pserver_program = t.get_pserver_program(current_endpoint)
        pserver_startup_program = t.get_startup_program(