From 8dae87ef6006cb15f803d64abe4bc6b1f4b98fbb Mon Sep 17 00:00:00 2001
From: tangwei <tangwei12@baidu.com>
Date: Thu, 14 May 2020 11:43:06 +0800
Subject: [PATCH] add paddle cloud run

---
 core/trainers/cluster_trainer.py  | 3 +--
 example/mpi/{worker.sh => job.sh} | 0
 example/mpi/submit.sh             | 2 +-
 3 files changed, 2 insertions(+), 3 deletions(-)
 rename example/mpi/{worker.sh => job.sh} (100%)

diff --git a/core/trainers/cluster_trainer.py b/core/trainers/cluster_trainer.py
index 8d71280c..a23ba61c 100755
--- a/core/trainers/cluster_trainer.py
+++ b/core/trainers/cluster_trainer.py
@@ -31,8 +31,7 @@ from paddlerec.core.trainers.transpiler_trainer import TranspileTrainer
 
 class ClusterTrainer(TranspileTrainer):
     def processor_register(self):
-        #role = PaddleCloudRoleMaker()
-        role = MPISymetricRoleMaker()
+        role = PaddleCloudRoleMaker()
         fleet.init(role)
 
         if fleet.is_server():
diff --git a/example/mpi/worker.sh b/example/mpi/job.sh
similarity index 100%
rename from example/mpi/worker.sh
rename to example/mpi/job.sh
diff --git a/example/mpi/submit.sh b/example/mpi/submit.sh
index 56b5f879..33eba839 100644
--- a/example/mpi/submit.sh
+++ b/example/mpi/submit.sh
@@ -65,7 +65,7 @@ function submit() {
   g_run_stage="submit"
   g_job_name="paddle_rec_mpi"
   g_hdfs_path=$g_hdfs_path
-  g_job_entry="worker.sh"
+  g_job_entry="job.sh"
 
   engine_hdfs_output=${engine_hdfs_output}/$(date +%Y%m%d%H%M%S)
 
-- 
GitLab