import tensorflow as tf

            config=run_config,
            params={
                "batch_size": FLAGS.train_batch_size
                if FLAGS.do_train
                else FLAGS.eval_batch_size,
            },
        )
    if FLAGS.do_train and FLAGS.do_eval:
        tf.logging.info("***** Running training *****")
        tf.logging.info("  Training batch size = %d", FLAGS.train_batch_size)
        train_input_fn = input_fn_builder(
            input_files=input_files,
            max_seq_length=FLAGS.max_seq_length,
            max_predictions_per_seq=FLAGS.max_predictions_per_seq,
            is_training=True,
        )
        eval_input_fn = input_fn_builder(
            input_files=input_files,
