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

Fixing various class documentations.

Browse Source
This commit is contained in:
Grégory Châtel
2018-12-20 13:11:17 +01:00
parent 7fb94ab934
commit 7176674849
1 changed files with 4 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pytorch_pretrained_bert/modeling.py
View File

@@ -456,7 +456,9 @@ class PreTrainedBertModel(nn.Module):
. `bert-base-uncased` . `bert-base-uncased`
. `bert-large-uncased` . `bert-large-uncased`
. `bert-base-cased` . `bert-base-cased`
. `bert-base-multilingual` . `bert-large-cased`
. `bert-base-multilingual-uncased`
. `bert-base-multilingual-cased`
. `bert-base-chinese` . `bert-base-chinese`
- a path or url to a pretrained model archive containing: - a path or url to a pretrained model archive containing:
. `bert_config.json` a configuration file for the model . `bert_config.json` a configuration file for the model
@@ -1035,15 +1037,7 @@ class BertForQuestionAnswering(PreTrainedBertModel):
the sequence output that computes start_logits and end_logits the sequence output that computes start_logits and end_logits
Params: Params:
`config`: either `config`: a BertConfig class instance with the configuration to build a new model.
- a BertConfig class instance with the configuration to build a new model, or
- a str with the name of a pre-trained model to load selected in the list of:
. `bert-base-uncased`
. `bert-large-uncased`
. `bert-base-cased`
. `bert-base-multilingual`
. `bert-base-chinese`
The pre-trained model will be downloaded and cached if needed.
Inputs: Inputs:
`input_ids`: a torch.LongTensor of shape [batch_size, sequence_length] `input_ids`: a torch.LongTensor of shape [batch_size, sequence_length]