Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
71766748
"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "c11b3e2926740d134dd83ce96b6ec64d6f29d49b"
Commit
71766748
authored
Dec 20, 2018
by
Grégory Châtel
Browse files
Fixing various class documentations.
parent
7fb94ab9
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
10 deletions
+4
-10
pytorch_pretrained_bert/modeling.py
pytorch_pretrained_bert/modeling.py
+4
-10
No files found.
pytorch_pretrained_bert/modeling.py
View file @
71766748
...
@@ -456,7 +456,9 @@ class PreTrainedBertModel(nn.Module):
...
@@ -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):
...
@@ -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]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment