add qsub submit

fleet_rec/core/engine/cluster/cluster.py

@@ -21,6 +21,7 @@ import os
 import copy
 
 from fleetrec.core.engine.engine import Engine
+from fleetrec.core.factory import TrainerFactory
 from fleetrec.core.utils import envs
 
 
@@ -30,16 +31,8 @@ class ClusterEngine(Engine):
         self.submit_script = os.path.join(abs_dir, "master.sh")
 
     def start_worker_procs(self):
-        default_env = os.environ.copy()
-        current_env = copy.copy(default_env)
-        current_env.pop("http_proxy", None)
-        current_env.pop("https_proxy", None)
-
-        cmd = ("bash {}".format(self.submit_script)).split(" ")
-        proc = subprocess.Popen(cmd, env=current_env, cwd=os.getcwd())
-        proc.wait()
-
-        print("all workers and parameter servers already completed", file=sys.stderr)
+        trainer = TrainerFactory.create(self.trainer)
+        trainer.run()
 
     def start_master_procs(self):
         default_env = os.environ.copy()
@@ -51,8 +44,6 @@ class ClusterEngine(Engine):
         proc = subprocess.Popen(cmd, env=current_env, cwd=os.getcwd())
         proc.wait()
 
-        print("all workers and parameter servers already completed", file=sys.stderr)
-
     def run(self):
         role = envs.get_runtime_environ("engine_role")
 

fleet_rec/core/engine/cluster/master.sh

@@ -19,7 +19,6 @@ declare g_run_stage=""
 declare -r CALL="x"
 ################################################################################
 
-
 #-----------------------------------------------------------------------------------------------------------------
 # Function: get_cur_path
 # Description: get churrent path
@@ -31,8 +30,7 @@ declare -r CALL="x"
 # Return: 0 -- success; not 0 -- failure
 # Others: N/A
 #-----------------------------------------------------------------------------------------------------------------
-get_cur_path()
-{
+get_cur_path() {
   g_run_stage="get_cur_path"
   cd "$(dirname "${BASH_SOURCE-$0}")"
   g_curPath="${PWD}"
@@ -45,15 +43,13 @@ get_cur_path()
 #param : N/A
 #return : 0 -- success; not 0 -- failure
 #-----------------------------------------------------------------------------------------------------------------
-function check_error()
-{
-    if [ ${?} -ne 0 ]
-    then
+function check_error() {
+  if [ ${?} -ne 0 ]; then
     echo "execute " + $g_run_stage + " raise exception! please check ..."
     exit 1
   fi
 }
 
-source ${engine_scrpit}
+source ${engine_submit_scrpit}
 
 main

fleet_rec/run.py

@@ -154,8 +154,10 @@ def cluster_engine(args):
         cluster_envs["train.trainer.platform"] = envs.get_platform()
         print("launch {} engine with cluster to with model: {}".format(trainer, args.model))
         set_runtime_envs(cluster_envs, args.model)
-        launch = ClusterEngine(cluster_envs, args.model)
+
+        launch = LocalClusterEngine(cluster_envs, args.model)
         return launch
+
     if args.role == "worker":
         return worker()
     else:

models/rank/dnn/submit.sh

@@ -63,13 +63,11 @@ function after_submit() {
 #-----------------------------------------------------------------------------------------------------------------
 function submit() {
   g_run_stage="submit"
-
   g_job_name="paddle_rec_mpi"
   g_hdfs_path=$g_hdfs_path
-
   g_job_entry="worker.sh"
 
-  engine_hdfs_output=${engine_hdfs_output}/`date +%Y%m%d%H%M%S`
+  engine_hdfs_output=${engine_hdfs_output}/$(date +%Y%m%d%H%M%S)
 
   cd ${engine_temp_path}
 
@@ -79,9 +77,8 @@ function submit() {
     --hdfs ${engine_hdfs_name} \
     --ugi ${engine_hdfs_ugi} \
     --hout ${engine_hdfs_output} \
-    --files ${engine_temp_path} \
+    --files ./package \
     -l nodes=${engine_submit_nodes},walltime=1000:00:00,resource=full ${g_job_entry}
-
 }
 
 function main() {

models/rank/dnn/worker.sh

@@ -41,24 +41,24 @@ function check_error() {
 #-----------------------------------------------------------------------------------------------------------------
 function env_prepare() {
   g_run_stage="env_prepare"
-}
+  WORKDIR=$(pwd)
+  mpirun -npernode 1 mv package/* ./
+  echo "current:"$WORKDIR
+  export LIBRARY_PATH=$WORKDIR/python/lib:$LIBRARY_PATH
 
-function user_define_variables() {
-  echo "user_define_variables"
-  g_run_stage="user_define_variables"
+  mpirun -npernode 1 python/bin/python -m pip install whl/fleet_rec-0.0.2-py2-none-any.whl --index-url=http://pip.baidu.com/pypi/simple --trusted-host pip.baidu.com >/dev/null
+  check_error
 }
 
-function job() {
-  echo "job"
-  g_run_stage="job"
-
-  #  mpirun -npernode 2 -timestamp-output -tag-output -machinefile ${PBS_NODEFILE} python -u ${g_job_entry}
+function run() {
+  echo "run"
+  g_run_stage="run"
+  mpirun -npernode 2 -timestamp-output -tag-output -machinefile ${PBS_NODEFILE} python/bin/python -u -m fleetrec.run -m fleetrec.models.rank.dnn --engine cluster --role worker
 }
 
 function main() {
-  user_define_variables
   env_prepare
-  job
+  run
 }
 
 main