DistillBert Documentation Code Example fixes

2019-10-03 21:22:36 +05:30
parent e2ae9c0b73
commit e1b2949ae6
2 changed files with 3 additions and 5 deletions
--- a/transformers/modeling_distilbert.py
+++ b/transformers/modeling_distilbert.py
@@ -649,7 +649,7 @@ class DistilBertForQuestionAnswering(DistilBertPreTrainedModel):
        start_positions = torch.tensor([1])
        end_positions = torch.tensor([3])
        outputs = model(input_ids, start_positions=start_positions, end_positions=end_positions)
-        loss, start_scores, end_scores = outputs[:2]
+        loss, start_scores, end_scores = outputs[:3]

    """
    def __init__(self, config):