From 5def3302f455434e4aeb1aa225bacda63d9817c9 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Wed, 25 Sep 2019 12:38:08 +0200 Subject: [PATCH] update run_glue --- examples/run_glue.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/run_glue.py b/examples/run_glue.py index 496d4e937a..278f5c723a 100644 --- a/examples/run_glue.py +++ b/examples/run_glue.py @@ -279,10 +279,10 @@ def load_and_cache_examples(args, task, tokenizer, evaluate=False): label_list[1], label_list[2] = label_list[2], label_list[1] examples = processor.get_dev_examples(args.data_dir) if evaluate else processor.get_train_examples(args.data_dir) features = convert_examples_to_features(examples, - label_list, - args.max_seq_length, tokenizer, - output_mode, + label_list=label_list, + max_length=args.max_seq_length, + output_mode=output_mode, pad_on_left=bool(args.model_type in ['xlnet']), # pad on the left for xlnet pad_token=tokenizer.convert_tokens_to_ids([tokenizer.pad_token])[0], pad_token_segment_id=4 if args.model_type in ['xlnet'] else 0,