Merge pull request #145 from wlhgtc/master

Correct the wrong note
2019-01-07 12:23:05 +01:00
parent 2e8c5c00ec e626eecc25
commit bcd607542c
1 changed files with 1 additions and 1 deletions
--- a/pytorch_pretrained_bert/modeling.py
+++ b/pytorch_pretrained_bert/modeling.py
@@ -730,7 +730,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].