fixing typo in docstring

ebaacba3 · thomwolf · 870d7163 · ebaacba3
Commit ebaacba3 authored Nov 26, 2018 by thomwolf
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

pytorch_pretrained_bert/modeling.py pytorch_pretrained_bert/modeling.py +1 -1

No files found.
--- a/pytorch_pretrained_bert/modeling.py
+++ b/pytorch_pretrained_bert/modeling.py
@@ -781,7 +781,7 @@ class BertForNextSentencePrediction(PreTrainedBertModel):
    # Already been converted into WordPiece token ids
    input_ids = torch.LongTensor([[31, 51, 99], [15, 5, 0]])
    input_mask = torch.LongTensor([[1, 1, 1], [1, 1, 0]])
-    token_type_ids = torch.LongTensor([[0, 0, 1], [0, 2, 0]])
+    token_type_ids = torch.LongTensor([[0, 0, 1], [0, 1, 0]])

    config = BertConfig(vocab_size=32000, hidden_size=512,
        num_hidden_layers=8, num_attention_heads=6, intermediate_size=1024)