From 079b78f659ef9e06cb6b38db634c2f7daa5aba13 Mon Sep 17 00:00:00 2001 From: Patrick Simianer Date: Fri, 8 Apr 2016 14:10:57 +0200 Subject: dtrain: bugfix --- training/dtrain/parallelize.rb | 1 + 1 file changed, 1 insertion(+) (limited to 'training/dtrain') 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 -- cgit v1.2.3