summaryrefslogtreecommitdiff
path: root/training/dtrain
diff options
context:
space:
mode:
Diffstat (limited to 'training/dtrain')
-rwxr-xr-xtraining/dtrain/parallelize.rb1
1 files changed, 1 insertions, 0 deletions
diff --git a/training/dtrain/parallelize.rb b/training/dtrain/parallelize.rb
index 128f3f98..a1c2108f 100755
--- a/training/dtrain/parallelize.rb
+++ b/training/dtrain/parallelize.rb
@@ -76,6 +76,7 @@ def make_shards input, num_shards, epoch, rand
}
while leftover > 0
j = index.pop
+ break if !j
shard_in_files[-1].write in_lines[j]
leftover -= 1
end