1f1270f1908e1f0804bf9cfa47f099c8a6cb41fb,cloud_tpu/models/ptb/ptb.py,,main,#Any#,194

Before Change

      use_tpu=FLAGS.use_tpu,
      config=run_config,
      train_batch_size=FLAGS.batch_size)
  estimator.train(input_fn=input_fn, max_steps=FLAGS.train_steps)


if __name__ == "__main__":
  tf.app.run()

After Change

  epoch_size = FLAGS.train_steps // 10
  for train_steps in range(epoch_size, FLAGS.train_steps, epoch_size):
    estimator.train(input_fn=get_input_fn(FLAGS.train_file),
                    max_steps=train_steps)
    estimator.evaluate(input_fn=get_input_fn(FLAGS.valid_file), steps=15)


if __name__ == "__main__":
  tf.app.run()

Instances