summaryrefslogtreecommitdiff
path: root/training/dtrain/parallelize.rb
diff options
context:
space:
mode:
authorPatrick Simianer <simianer@cl.uni-heidelberg.de>2013-03-03 12:06:25 +0100
committerPatrick Simianer <simianer@cl.uni-heidelberg.de>2013-03-03 12:06:25 +0100
commit9d306b30c9abba995ba35243e5cb461bb472a61f (patch)
treee576d1d5e367364fd3b5d8b1446908ad2e2511f7 /training/dtrain/parallelize.rb
parentf051e98fb35c4ce69e489ef6550840bd6cd617fd (diff)
dtrain parallelize.rb fixes
Diffstat (limited to 'training/dtrain/parallelize.rb')
-rwxr-xr-xtraining/dtrain/parallelize.rb3
1 files changed, 2 insertions, 1 deletions
diff --git a/training/dtrain/parallelize.rb b/training/dtrain/parallelize.rb
index 9b0923f6..23f2a7ed 100755
--- a/training/dtrain/parallelize.rb
+++ b/training/dtrain/parallelize.rb
@@ -1,7 +1,7 @@
#!/usr/bin/env ruby
-if ARGV.size != 7
+if ARGV.size != 8
STDERR.write "Usage: "
STDERR.write "ruby parallelize.rb <dtrain.ini> <epochs> <rand=true|false> <#shards|predef> <at once> <input> <refs> <qsub>\n"
exit
@@ -95,6 +95,7 @@ end
remaining_shards = num_shards
while remaining_shards > 0
shards_at_once.times {
+ break if remaining_shards==0
qsub_str_start = qsub_str_end = ''
local_end = ''
if use_qsub