diff options
Diffstat (limited to 'tensorflow/contrib/tpu/python/tpu/keras_support.py')
-rw-r--r-- | tensorflow/contrib/tpu/python/tpu/keras_support.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/tpu/python/tpu/keras_support.py b/tensorflow/contrib/tpu/python/tpu/keras_support.py index dd7f8b678f..08e0465b71 100644 --- a/tensorflow/contrib/tpu/python/tpu/keras_support.py +++ b/tensorflow/contrib/tpu/python/tpu/keras_support.py @@ -1657,7 +1657,7 @@ class KerasTPUModel(models.Model): 'make sure your paths are correct and you have ' 'permissions to read the files. Skipping validation') - for step_index in range(steps_per_epoch - 1): + for step_index in range(steps_per_epoch): batch_logs = {'batch': step_index, 'size': 1} callbacks.on_batch_begin(step_index, batch_logs) try: |