f80592836b5c17fd3e6be14d3f1ad8a3a7aa3663,cloud_tpu/models/alexnet/alexnet.py,,input_fn,#Any#,152

Before Change

    train_files = ["%s/train-%05d-of-"
                   "01024" % (FLAGS.data_dir, num) for num in range(1024)]
    dataset = tf.contrib.data.TFRecordDataset(train_files)
    dataset = dataset.repeat().map(parser).batch(batch_size)
    images, labels = dataset.make_one_shot_iterator().get_next()
    return (
        tf.reshape(images, [batch_size, 224, 224, 3]),

After Change

    train_files = ["%s/train-%05d-of-"
                   "01024" % (FLAGS.data_dir, num) for num in range(1024)]
    dataset = tf.contrib.data.TFRecordDataset(train_files).repeat()
    if FLAGS.num_preprocessing_threads == 0:
      dataset = dataset.map(parser)
    else:
      dataset = dataset.map(parser,
                            FLAGS.num_preprocessing_threads)
    if FLAGS.prefetch_buffer_size != 0:
      dataset = dataset.prefetch(FLAGS.prefetch_buffer_size)
    dataset = dataset.batch(batch_size)
    images, labels = dataset.make_one_shot_iterator().get_next()
    return (
        tf.reshape(images, [batch_size, 224, 224, 3]),

Instances