summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-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