summaryrefslogtreecommitdiff
path: root/training/dtrain
diff options
context:
space:
mode:
authorPatrick Simianer <p@simianer.de>2016-04-08 14:10:57 +0200
committerPatrick Simianer <p@simianer.de>2016-04-08 14:10:57 +0200
commit079b78f659ef9e06cb6b38db634c2f7daa5aba13 (patch)
treea9a0e8767ed01c2c47585b995df5ed1bbe10b958 /training/dtrain
parent851f1582e839c272cbc0b6e29f5f911e281911d9 (diff)
dtrain: bugfix
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