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

fix docstring

Browse Source
This commit is contained in:
thomwolf
2019-02-05 16:34:32 +01:00
parent bd74632687
commit 822915142b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_pretrained_bert/modeling.py
View File

@@ -985,7 +985,7 @@ class BertForTokenClassification(PreTrainedBertModel):
selected in [0, 1]. It's a mask to be used if the input sequence length is smaller than the max selected in [0, 1]. It's a mask to be used if the input sequence length is smaller than the max
input sequence length in the current batch. It's the mask that we typically use for attention when input sequence length in the current batch. It's the mask that we typically use for attention when
a batch has varying length sentences. a batch has varying length sentences.
`labels`: labels for the classification output: torch.LongTensor of shape [batch_size] `labels`: labels for the classification output: torch.LongTensor of shape [batch_size, sequence_length]
with indices selected in [0, ..., num_labels]. with indices selected in [0, ..., num_labels].
Outputs: Outputs: