Update modeling.py

2018-12-22 20:26:05 +08:00
parent 8da280ebbe
commit e626eecc25
1 changed files with 1 additions and 1 deletions
--- a/pytorch_pretrained_bert/modeling.py
+++ b/pytorch_pretrained_bert/modeling.py
@@ -728,7 +728,7 @@ class BertForMaskedLM(PreTrainedBertModel):
            is only computed for the labels set in [0, ..., vocab_size]

    Outputs:
-        if `masked_lm_labels` is `None`:
+        if `masked_lm_labels` is  not `None`:
            Outputs the masked language modeling loss.
        if `masked_lm_labels` is `None`:
            Outputs the masked language modeling logits of shape [batch_size, sequence_length, vocab_size].