add paddle cloud run

8dae87ef · tangwei · 67869ede · 8dae87ef · 8dae87ef · 8dae87ef
隐藏空白更改
内联并排

Showing with 2 addition and 3 deletion

core/trainers/cluster_trainer.py core/trainers/cluster_trainer.py +1 -2

example/mpi/job.sh example/mpi/job.sh +0 -0

example/mpi/submit.sh example/mpi/submit.sh +1 -1

未找到文件。
--- 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():

--- a/example/mpi/worker.sh
+++ b/example/mpi/worker.sh
--- 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)