Skip to content

Commit aa70c63

Browse files
authored
Add finetuned ernie model (#639)
* remove dead code * Add finetuned ernie model * fix nameing
1 parent 9a45f83 commit aa70c63

File tree

2 files changed

+20
-0
lines changed

2 files changed

+20
-0
lines changed

paddlenlp/transformers/ernie/modeling.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -134,6 +134,19 @@ class ErniePretrainedModel(PretrainedModel):
134134
"vocab_size": 30522,
135135
"pad_token_id": 0,
136136
},
137+
"ernie-2.0-en-finetuned-squad": {
138+
"attention_probs_dropout_prob": 0.1,
139+
"hidden_act": "gelu",
140+
"hidden_dropout_prob": 0.1,
141+
"hidden_size": 768,
142+
"initializer_range": 0.02,
143+
"max_position_embeddings": 512,
144+
"num_attention_heads": 12,
145+
"num_hidden_layers": 12,
146+
"type_vocab_size": 4,
147+
"vocab_size": 30522,
148+
"pad_token_id": 0,
149+
},
137150
"ernie-2.0-large-en": {
138151
"attention_probs_dropout_prob": 0.1,
139152
"intermediate_size": 4096, # special for ernie-2.0-large-en
@@ -158,6 +171,8 @@ class ErniePretrainedModel(PretrainedModel):
158171
"https://paddlenlp.bj.bcebos.com/models/transformers/ernie_tiny/ernie_tiny.pdparams",
159172
"ernie-2.0-en":
160173
"https://paddlenlp.bj.bcebos.com/models/transformers/ernie_v2_base/ernie_v2_eng_base.pdparams",
174+
"ernie-2.0-en-finetuned-squad":
175+
"https://paddlenlp.bj.bcebos.com/models/transformers/ernie_v2_base/ernie_v2_eng_base_finetuned_squad.pdparams",
161176
"ernie-2.0-large-en":
162177
"https://paddlenlp.bj.bcebos.com/models/transformers/ernie_v2_large/ernie_v2_eng_large.pdparams",
163178
}

paddlenlp/transformers/ernie/tokenizer.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,8 @@ class ErnieTokenizer(PretrainedTokenizer):
7676
"https://paddlenlp.bj.bcebos.com/models/transformers/ernie_tiny/vocab.txt",
7777
"ernie-2.0-en":
7878
"https://paddlenlp.bj.bcebos.com/models/transformers/ernie_v2_base/vocab.txt",
79+
"ernie-2.0-en-finetuned-squad":
80+
"https://paddlenlp.bj.bcebos.com/models/transformers/ernie_v2_base/vocab.txt",
7981
"ernie-2.0-large-en":
8082
"https://paddlenlp.bj.bcebos.com/models/transformers/ernie_v2_large/vocab.txt",
8183
"ernie-gen-base-en":
@@ -96,6 +98,9 @@ class ErnieTokenizer(PretrainedTokenizer):
9698
"ernie-2.0-en": {
9799
"do_lower_case": True
98100
},
101+
"ernie-2.0-en-finetuned-squad": {
102+
"do_lower_case": True
103+
},
99104
"ernie-2.0-large-en": {
100105
"do_lower_case": True
101106
},

0 commit comments

Comments
 (0)