1. 原因
Roberta 模型的输入,不能包括token_type_ids,否则就会出现上述bug
2. 解决方案
将token_type_ids输入设置为None
inputs = {
"input_ids": input_ids,
"token_type_ids": None,
"attention_mask": attention_mask,
"labels": labels
}
1. 原因
Roberta 模型的输入,不能包括token_type_ids,否则就会出现上述bug
2. 解决方案
将token_type_ids输入设置为None
inputs = {
"input_ids": input_ids,
"token_type_ids": None,
"attention_mask": attention_mask,
"labels": labels
}