Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'master' of ssh://github.com/dmlc/rabit

Browse Source
This commit is contained in:
tqchen 2015-04-27 15:55:58 -07:00
commit e1ddcc2eb7
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tracker/rabit_sge.py
View File

@ -38,6 +38,7 @@ else:
runscript = '%s/runrabit.sh' % args.logdir runscript = '%s/runrabit.sh' % args.logdir
fo = open(runscript, 'w') fo = open(runscript, 'w')
fo.write('source ~/.bashrc\n')
fo.write('\"$@\"\n') fo.write('\"$@\"\n')
fo.close() fo.close()
# #
@ -51,7 +52,7 @@ def sge_submit(nslave, worker_args, worker_envs):
nslave number of slave process to start up nslave number of slave process to start up
args arguments to launch each job args arguments to launch each job
this usually includes the parameters of master_uri and parameters passed into submit this usually includes the parameters of master_uri and parameters passed into submit
""" """
env_arg = ','.join(['%s=\"%s\"' % (k, str(v)) for k, v in worker_envs.items()]) env_arg = ','.join(['%s=\"%s\"' % (k, str(v)) for k, v in worker_envs.items()])
cmd = 'qsub -cwd -t 1-%d -S /bin/bash' % nslave cmd = 'qsub -cwd -t 1-%d -S /bin/bash' % nslave
if args.queue != 'default': if args.queue != 'default':