From 2c9a3115b71aed07bf0745828e62b6f5ce1fca72 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Tue, 23 Jul 2019 16:45:55 +0200 Subject: [PATCH] fix #858 --- examples/run_glue.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/run_glue.py b/examples/run_glue.py index 25a487156e..b383bbcb80 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]: