Skip to content

Commit 9d91ae5

Browse files
authored
Merge pull request #7669 from putcn/develop
update doc and dist test due to transpiler change
2 parents 58be41f + 95d6dce commit 9d91ae5

File tree

6 files changed

+15
-8
lines changed

6 files changed

+15
-8
lines changed

doc/howto/usage/cluster/fluid_cluster_train_en.md

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,9 +101,11 @@ t.transpile(optimize_ops, params_grads, pservers=pserver_endpoints, trainers=2)
101101
... #create executor
102102

103103
# in pserver, run this
104-
exe.run(fluid.default_startup_program())
105104
#current_endpoint here means current pserver IP:PORT you wish to run on
106-
exe.run(t.get_pserver_program(current_endpoint))
105+
pserver_prog = t.get_pserver_program(current_endpoint)
106+
pserver_startup = t.get_startup_program(current_endpoint, pserver_prog)
107+
exe.run(pserver_startup)
108+
exe.run(pserver_prog)
107109

108110
# in trainer, run this
109111
... # define data reader

python/paddle/v2/fluid/tests/book_distribute/notest_dist_fit_a_line.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,8 @@
5454
print("need env SERVER_ENDPOINT")
5555
exit(1)
5656
pserver_prog = t.get_pserver_program(current_endpoint)
57-
exe.run(fluid.default_startup_program())
57+
pserver_startup = t.get_startup_program(current_endpoint, pserver_prog)
58+
exe.run(pserver_startup)
5859
exe.run(pserver_prog)
5960
else:
6061
trainer_prog = t.get_trainer_program()

python/paddle/v2/fluid/tests/book_distribute/notest_dist_label_semantic_roles.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,8 @@ def main():
198198
print("need env SERVER_ENDPOINT")
199199
exit(1)
200200
pserver_prog = t.get_pserver_program(current_endpoint)
201-
exe.run(fluid.default_startup_program())
201+
pserver_startup = t.get_startup_program(current_endpoint, pserver_prog)
202+
exe.run(pserver_startup)
202203
exe.run(pserver_prog)
203204
elif training_role == "TRAINER":
204205
trainer_prog = t.get_trainer_program()

python/paddle/v2/fluid/tests/book_distribute/notest_dist_word2vec.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,8 @@
8888
print("need env SERVER_ENDPOINT")
8989
exit(1)
9090
pserver_prog = t.get_pserver_program(current_endpoint)
91-
exe.run(fluid.default_startup_program())
91+
pserver_startup = t.get_startup_program(current_endpoint, pserver_prog)
92+
exe.run(pserver_startup)
9293
exe.run(pserver_prog)
9394
elif training_role == "TRAINER":
9495
feeder = fluid.DataFeeder(

python/paddle/v2/fluid/tests/book_distribute/notest_recognize_digits_mlp_dist.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,8 @@
6161
print("need env SERVER_ENDPOINT")
6262
exit(1)
6363
pserver_prog = t.get_pserver_program(current_endpoint)
64-
exe.run(fluid.default_startup_program())
64+
pserver_startup = t.get_startup_program(current_endpoint, pserver_prog)
65+
exe.run(pserver_startup)
6566
exe.run(pserver_prog)
6667
elif training_role == "TRAINER":
6768
trainer_prog = t.get_trainer_program()

python/paddle/v2/fluid/tests/book_distribute/notest_understand_sentiment_conv_dist.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,15 +92,16 @@ def main():
9292
t.transpile(
9393
optimize_ops, params_grads, pservers=pserver_endpoints, trainers=2)
9494

95-
exe.run(fluid.default_startup_program())
96-
9795
if training_role == "PSERVER":
9896
if not current_endpoint:
9997
print("need env SERVER_ENDPOINT")
10098
exit(1)
10199
pserver_prog = t.get_pserver_program(current_endpoint)
100+
pserver_startup = t.get_startup_program(current_endpoint, pserver_prog)
101+
exe.run(pserver_startup)
102102
exe.run(pserver_prog)
103103
elif training_role == "TRAINER":
104+
exe.run(fluid.default_startup_program())
104105
trainer_prog = t.get_trainer_program()
105106
feeder = fluid.DataFeeder(feed_list=[data, label], place=place)
106107

0 commit comments

Comments
 (0)