import tensorflow as tf

      session_config=config)

  if FLAGS.use_hvd and hvd.rank() != 0 and not FLAGS.auto_recover:
    run_config = tf.estimator.RunConfig(
        model_dir=FLAGS.output_dir,
        keep_checkpoint_max=FLAGS.keep_checkpoint_max,
