search

nlintz / TensorFlow-Tutorials

Simple tutorials using Google's TensorFlow Framework
6.01k stars 1.51k forks source link

Exhaust training set #49

Closed eywalker closed 8 years ago

eywalker commented 8 years ago

Previously, when the len(trX) is completely divisible by batch_size, the training skipped over the last batch because end=len(trX) would never occur. Corrected this and now the loop will go over the entire training set if batch_size can cleanly fit in.

nlintz commented 8 years ago

I think this looks good. Can you take a look @hunkim

hunkim commented 8 years ago

+1. @eywalker We have the same code in other files. Do you have time to fix them as well?

eywalker commented 8 years ago

Sure - I'll do a quick scan.

hunkim commented 8 years ago

Wonderful! Thanks!!