Fix Documentation issue in BertForMaskedLM forward (#3855)

7d40901c · Bharat Raghunathan · GitHub · b1ff0b2a · 7d40901c
Unverified Commit 7d40901c authored Apr 21, 2020 by Bharat Raghunathan Committed by GitHub Apr 21, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/modeling_bert.py src/transformers/modeling_bert.py +1 -1

No files found.
--- a/src/transformers/modeling_bert.py
+++ b/src/transformers/modeling_bert.py
@@ -957,7 +957,7 @@ class BertForMaskedLM(BertPreTrainedModel):
            ltr_lm_loss = loss_fct(prediction_scores.view(-1, self.config.vocab_size), lm_labels.view(-1))
            outputs = (ltr_lm_loss,) + outputs

-        return outputs  # (masked_lm_loss), (ltr_lm_loss), prediction_scores, (hidden_states), (attentions)
+        return outputs  # (ltr_lm_loss), (masked_lm_loss), prediction_scores, (hidden_states), (attentions)


 @add_start_docstrings(