diff --git a/examples/run_glue.py b/examples/run_glue.py index 25a487156e5f3a..b383bbcb808856 100644 --- a/examples/run_glue.py +++ b/examples/run_glue.py @@ -256,7 +256,7 @@ def load_and_cache_examples(args, task, tokenizer, evaluate=False): cls_token_at_end=bool(args.model_type in ['xlnet']), # xlnet has a cls token at the end cls_token=tokenizer.cls_token, sep_token=tokenizer.sep_token, - cls_token_segment_id=2 if args.model_type in ['xlnet'] else 1, + cls_token_segment_id=2 if args.model_type in ['xlnet'] else 0, pad_on_left=bool(args.model_type in ['xlnet']), # pad on the left for xlnet pad_token_segment_id=4 if args.model_type in ['xlnet'] else 0) if args.local_rank in [-1, 0]: