[Pytorch --- 6] Roberta分类任务,出现RuntimeError: Creating MTGP constants failed

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
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章