From e25cf6487c2ee7e5e5edb0afbea459e5a0427e0c Mon Sep 17 00:00:00 2001 From: Patrick Simianer Date: Thu, 20 Jun 2013 02:36:26 +0200 Subject: extra qsub arg for training/dtrain/parallelize.rb --- training/dtrain/parallelize.rb | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'training/dtrain') diff --git a/training/dtrain/parallelize.rb b/training/dtrain/parallelize.rb index 5177c6a5..e54eba92 100755 --- a/training/dtrain/parallelize.rb +++ b/training/dtrain/parallelize.rb @@ -20,6 +20,7 @@ opts = Trollop::options do opt :references, "references", :type => :string opt :qsub, "use qsub", :type => :bool, :default => false opt :dtrain_binary, "path to dtrain binary", :type => :string + opt :extra_qsub, "extra qsub args", :type => :string, :default => "" # e.g. "-l virtual_free=32000M" end usage if not opts[:config]&&opts[:shards]&&opts[:input]&&opts[:references] @@ -119,7 +120,7 @@ end qsub_str_start = qsub_str_end = '' local_end = '' if use_qsub - qsub_str_start = "qsub -cwd -sync y -b y -j y -o work/out.#{shard}.#{epoch} -N dtrain.#{shard}.#{epoch} \"" + qsub_str_start = "qsub #{opts[:extra_qsub]} -cwd -sync y -b y -j y -o work/out.#{shard}.#{epoch} -N dtrain.#{shard}.#{epoch} \"" qsub_str_end = "\"" local_end = '' else -- cgit v1.2.3