SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix Documentation issue in BertForMaskedLM forward (#3855)

Browse Source
This commit is contained in:
Bharat Raghunathan
2020-04-21 12:38:20 +05:30
committed by GitHub
parent b1ff0b2ae7
commit 7d40901ce3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_bert.py
View File

@@ -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(