Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/latte_pytorch.git" did not exist on "601eab0669698bf2f261ccf2a55c6c12b090702a"
Commit ebaacba3 authored by thomwolf's avatar thomwolf
Browse files

fixing typo in docstring

parent 870d7163
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pytorch_pretrained_bert/modeling.py
View file @ ebaacba3
...@@ -781,7 +781,7 @@ class BertForNextSentencePrediction(PreTrainedBertModel): ...@@ -781,7 +781,7 @@ class BertForNextSentencePrediction(PreTrainedBertModel):
# Already been converted into WordPiece token ids # Already been converted into WordPiece token ids
input_ids = torch.LongTensor([[31, 51, 99], [15, 5, 0]]) input_ids = torch.LongTensor([[31, 51, 99], [15, 5, 0]])
input_mask = torch.LongTensor([[1, 1, 1], [1, 1, 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, config = BertConfig(vocab_size=32000, hidden_size=512,
num_hidden_layers=8, num_attention_heads=6, intermediate_size=1024) num_hidden_layers=8, num_attention_heads=6, intermediate_size=1024)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment