output_token_type -> token_type_ids
This commit is contained in:
1 parent
3927d7756c
commit
c832f43a4d
3 files changed
+5
-5
No files matched your search
@@ -413,7 +413,7 @@ def convert_examples_to_features(examples, label_list, max_seq_length,
|
||||
max_length=max_seq_length,
|
||||
truncate_first_sequence=True # We're truncating the first sequence as a priority
|
||||
)
|
||||
input_ids, segment_ids = inputs["input_ids"], inputs["output_token_type"]
|
||||
input_ids, segment_ids = inputs["input_ids"], inputs["token_type_ids"]
|
||||
|
||||
# The mask has 1 for real tokens and 0 for padding tokens. Only real
|
||||
# tokens are attended to.
|
||||
|
||||
Reference in new issue
Block a user