• Derek Murray's avatar
    Update mnist_tpu.py with recommended tf.data APIs (#5853) · aead3912
    Derek Murray authored
    1. `tf.contrib.data.batch_and_drop_remainder()` has been deprecated for a while now.
    2. `Dataset.make_one_shot_iterator()` is no longer necessary in input functions, and avoiding it lead to better performance on TPUs. (It will also be removed in TF 2.0.)
    aead3912
mnist_tpu.py 7.19 KB