From 7d40901ce3ad9e1c79fd9bb117f5b84bff42c33f Mon Sep 17 00:00:00 2001 From: Bharat Raghunathan Date: Tue, 21 Apr 2020 12:38:20 +0530 Subject: [PATCH] Fix Documentation issue in BertForMaskedLM forward (#3855) --- src/transformers/modeling_bert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/modeling_bert.py b/src/transformers/modeling_bert.py index 3cfa13acdd..8c74ff7b59 100644 --- 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(