预训练的 roberta 关系提取属性错误答案

【问题标题】：pretrained roberta relation extraction attribute error预训练的 roberta 关系提取属性错误
【发布时间】：2021-12-09 02:59:10
【问题描述】：

我正在尝试使用以下预训练的拥抱脸模型：https://huggingface.co/mmoradi/Robust-Biomed-RoBERTa-RelationClassification

我使用以下代码：

from transformers import AutoTokenizer, AutoModel
  
tokenizer = AutoTokenizer.from_pretrained("mmoradi/Robust-Biomed-RoBERTa-RelationClassification")

model = AutoModel.from_pretrained("mmoradi/Robust-Biomed-RoBERTa-RelationClassification")

inputs = tokenizer("""The colorectal cancer was caused by mutations in angina""")
outputs = model(**inputs)

由于某种原因，我在尝试生成输出时收到以下错误，所以在我的代码的最后一行：

--> 796 input_shape = input_ids.size() 797 elif inputs_embeds 不是无：第798章

AttributeError: 'list' 对象没有属性 'size'

输入如下所示：

{'input_ids': [0, 133, 11311, 1688, 3894, 337, 1668, 21, 1726, 30, 28513, 11, 1480, 347, 2], 'attention_mask': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]}

我不知道如何调试它，所以欢迎任何帮助或提示！

【问题讨论】：

标签： nlp huggingface-transformers roberta

【解决方案1】：

你必须指定你想要的张量类型以换取tokenizer。如果不这样做，它将返回一个包含两个列表（input_ids 和 attention_mask）的字典：

inputs = tokenizer("""The colorectal cancer was caused by mutations in angina""", return_tensors="pt")

【讨论】：