Fix XLNet Classifier.

PiperOrigin-RevId: 334642892

official/nlp/xlnet/classifier_utils.py

@@ -153,7 +153,7 @@ def convert_single_example(example_index, example, label_list, max_seq_length,
     logging.info("input_ids: %s", " ".join([str(x) for x in input_ids]))
     logging.info("input_mask: %s", " ".join([str(x) for x in input_mask]))
     logging.info("segment_ids: %s", " ".join([str(x) for x in segment_ids]))
-    logging.info("label: %d (id = %d)", example.label, label_id)
+    logging.info("label: %s (id = %d)", example.label, label_id)
 
   feature = InputFeatures(
       input_ids=input_ids,

official/nlp/xlnet/run_classifier.py

@@ -155,7 +155,7 @@ def main(unused_argv):
       adam_epsilon=FLAGS.adam_epsilon)
   model_config = xlnet_config.XLNetConfig(FLAGS)
   run_config = xlnet_config.create_run_config(True, False, FLAGS)
-  model_fn = functools.partial(modeling.classification_model, model_config,
+  model_fn = functools.partial(get_classificationxlnet_model, model_config,
                                run_config, FLAGS.n_class, FLAGS.summary_type)
   input_meta_data = {}
   input_meta_data["d_model"] = FLAGS.d_model