summaryrefslogtreecommitdiff
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
commit349ee7d5599bb973506c8bbb56926cf9d366b564 (patch)
treeb18cb3e0c27af2f6bf99606c573f26908f8512c1
parentb89fd90083b22e6d4ab469af001a1f15fbcd7da9 (diff)
dtrain parallelize.rb fixes
-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