From 59057abe523297711bd3edcab64b6c20a32c46de Mon Sep 17 00:00:00 2001 From: LysandreJik Date: Mon, 2 Sep 2019 22:03:39 -0400 Subject: [PATCH] typo --- examples/utils_glue.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/utils_glue.py b/examples/utils_glue.py index 841d7376cf..96023e149c 100644 --- a/examples/utils_glue.py +++ b/examples/utils_glue.py @@ -415,7 +415,7 @@ def convert_examples_to_features(examples, label_list, max_seq_length, if ex_index % 10000 == 0: logger.info("Writing example %d of %d" % (ex_index, len(examples))) - input_ids, input_mask = tokenizer.encode(example.text_a, example.text_b, add_special_tokens=True, output_mask=True) + input_ids, segment_ids = tokenizer.encode(example.text_a, example.text_b, add_special_tokens=True, output_mask=True) # The mask has 1 for real tokens and 0 for padding tokens. Only real # tokens are attended to.