修复span编码bug

isLouisHsu · Oct 23, 2021 · db95b4a · db95b4a
1 parent 6a0c57e
commit db95b4a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_span.py b/run_span.py
@@ -817,7 +817,7 @@ def _convert_example_to_feature(self, example):
         inputs["input_len"] = inputs["attention_mask"].sum(dim=1)  # for special tokens
         input_len = inputs["input_len"].item()
         inputs["spans"], inputs["span_mask"] = self._encode_span(
-            input_len, input_len, sent_start, sent_end)  # dynamic batch
+            input_len, input_len, sent_start + 1, sent_end + 1)  # dynamic batch
         inputs["sent_start"] = torch.tensor([sent_start])
         inputs["sent_end"] = torch.tensor([sent_end])