import tensorflow as tf

  else:
    tf.logging.info('Mode not found.')

  if FLAGS.export_dir is not None:
    tf.logging.info('Starting exporting saved model ...')
    serving_shape = [hparams.image_size, hparams.image_size, 3]
    export_path = image_classifier.export_saved_model(
        export_dir_base=FLAGS.export_dir,
