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
b0ee7c7d
"...resnet50_tensorflow.git" did not exist on "1b40311f660360ee0ce9006217cae4ad5508b8a2"
Commit
b0ee7c7d
authored
Nov 29, 2019
by
maxvidal
Committed by
Julien Chaumond
Nov 29, 2019
Browse files
Added Camembert to available models
parent
ecf15ebf
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
3 deletions
+5
-3
examples/run_lm_finetuning.py
examples/run_lm_finetuning.py
+5
-3
No files found.
examples/run_lm_finetuning.py
View file @
b0ee7c7d
...
...
@@ -47,7 +47,8 @@ from transformers import (WEIGHTS_NAME, AdamW, get_linear_schedule_with_warmup,
GPT2Config
,
GPT2LMHeadModel
,
GPT2Tokenizer
,
OpenAIGPTConfig
,
OpenAIGPTLMHeadModel
,
OpenAIGPTTokenizer
,
RobertaConfig
,
RobertaForMaskedLM
,
RobertaTokenizer
,
DistilBertConfig
,
DistilBertForMaskedLM
,
DistilBertTokenizer
)
DistilBertConfig
,
DistilBertForMaskedLM
,
DistilBertTokenizer
,
CamembertConfig
,
CamembertForMaskedLM
,
CamembertTokenizer
)
logger
=
logging
.
getLogger
(
__name__
)
...
...
@@ -58,7 +59,8 @@ MODEL_CLASSES = {
'openai-gpt'
:
(
OpenAIGPTConfig
,
OpenAIGPTLMHeadModel
,
OpenAIGPTTokenizer
),
'bert'
:
(
BertConfig
,
BertForMaskedLM
,
BertTokenizer
),
'roberta'
:
(
RobertaConfig
,
RobertaForMaskedLM
,
RobertaTokenizer
),
'distilbert'
:
(
DistilBertConfig
,
DistilBertForMaskedLM
,
DistilBertTokenizer
)
'distilbert'
:
(
DistilBertConfig
,
DistilBertForMaskedLM
,
DistilBertTokenizer
),
'camembert'
:
(
CamembertConfig
,
CamembertForMaskedLM
,
CamembertTokenizer
)
}
...
...
@@ -432,7 +434,7 @@ def main():
parser
.
add_argument
(
'--server_port'
,
type
=
str
,
default
=
''
,
help
=
"For distant debugging."
)
args
=
parser
.
parse_args
()
if
args
.
model_type
in
[
"bert"
,
"roberta"
,
"distilbert"
]
and
not
args
.
mlm
:
if
args
.
model_type
in
[
"bert"
,
"roberta"
,
"distilbert"
,
"camembert"
]
and
not
args
.
mlm
:
raise
ValueError
(
"BERT and RoBERTa do not have LM heads but masked LM heads. They must be run using the --mlm "
"flag (masked language modeling)."
)
if
args
.
eval_data_file
is
None
and
args
.
do_eval
:
...
...
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