We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 035e85a commit 82fd0d0Copy full SHA for 82fd0d0
tibert/bertcoref.py
@@ -902,7 +902,10 @@ def coreference_documents(
902
antecedent_span_i = int(
903
self.top_antecedents_index[b_i][m_j][top_antecedent_idx].item()
904
)
905
- antecedent_coords = spans_idx[antecedent_span_i]
+ try:
906
+ antecedent_coords = spans_idx[antecedent_span_i]
907
+ except IndexError:
908
+ continue
909
910
antecedent_mention_score = float(
911
self.mentions_scores[b_i][antecedent_span_i].item()
0 commit comments