Commit 59057abe authored by LysandreJik's avatar LysandreJik
Browse files

typo

parent bac332fe
......@@ -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.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment