diff options
Diffstat (limited to 'tensorflow/contrib/eager/python/examples/revnet/cifar_input.py')
| -rw-r--r-- | tensorflow/contrib/eager/python/examples/revnet/cifar_input.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/eager/python/examples/revnet/cifar_input.py b/tensorflow/contrib/eager/python/examples/revnet/cifar_input.py index e1d8b3a055..e9672f13e1 100644 --- a/tensorflow/contrib/eager/python/examples/revnet/cifar_input.py +++ b/tensorflow/contrib/eager/python/examples/revnet/cifar_input.py @@ -35,7 +35,7 @@ def get_ds_from_tfrecords(data_dir, epochs=None, shuffle=True, data_format="channels_first", - num_parallel_calls=8, + num_parallel_calls=12, prefetch=0, div255=True, dtype=tf.float32): @@ -111,6 +111,6 @@ def get_ds_from_tfrecords(data_dir, }[split] dataset = dataset.shuffle(size) - dataset = dataset.batch(batch_size) + dataset = dataset.batch(batch_size, drop_remainder=True) return dataset |