diff options
Diffstat (limited to 'training/dtrain')
-rwxr-xr-x | training/dtrain/parallelize.rb | 1 |
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 |