[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
}

 

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