擅长:python、mysql、java
<p><code>BertForQuestionAnswering</code>返回一个<a href="https://huggingface.co/transformers/_modules/transformers/modeling_outputs.html#QuestionAnsweringModelOutput" rel="nofollow noreferrer">^{<cd2>}</a>对象</p>
<p>由于将<code>BertForQuestionAnswering</code>的输出设置为<code>start_scores, end_scores</code>,因此返回的<code>QuestionAnsweringModelOutput</code>对象被强制转换为字符串的元组<code>('start_logits', 'end_logits')</code>,从而导致类型不匹配错误</p>
<p>以下方面应起作用:</p>
<pre><code>outputs = model(input_ids=torch.tensor([inputs]), token_type_ids=torch.tensor([sentence_embedding]))
start_index = torch.argmax(outputs.start_logits)
</code></pre>