File tree Expand file tree Collapse file tree 2 files changed +20
-0
lines changed
paddlenlp/transformers/ernie Expand file tree Collapse file tree 2 files changed +20
-0
lines changed Original file line number Diff line number Diff line change @@ -134,6 +134,19 @@ class ErniePretrainedModel(PretrainedModel):
134
134
"vocab_size" : 30522 ,
135
135
"pad_token_id" : 0 ,
136
136
},
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
+ },
137
150
"ernie-2.0-large-en" : {
138
151
"attention_probs_dropout_prob" : 0.1 ,
139
152
"intermediate_size" : 4096 , # special for ernie-2.0-large-en
@@ -158,6 +171,8 @@ class ErniePretrainedModel(PretrainedModel):
158
171
"https://paddlenlp.bj.bcebos.com/models/transformers/ernie_tiny/ernie_tiny.pdparams" ,
159
172
"ernie-2.0-en" :
160
173
"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" ,
161
176
"ernie-2.0-large-en" :
162
177
"https://paddlenlp.bj.bcebos.com/models/transformers/ernie_v2_large/ernie_v2_eng_large.pdparams" ,
163
178
}
Original file line number Diff line number Diff line change @@ -76,6 +76,8 @@ class ErnieTokenizer(PretrainedTokenizer):
76
76
"https://paddlenlp.bj.bcebos.com/models/transformers/ernie_tiny/vocab.txt" ,
77
77
"ernie-2.0-en" :
78
78
"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" ,
79
81
"ernie-2.0-large-en" :
80
82
"https://paddlenlp.bj.bcebos.com/models/transformers/ernie_v2_large/vocab.txt" ,
81
83
"ernie-gen-base-en" :
@@ -96,6 +98,9 @@ class ErnieTokenizer(PretrainedTokenizer):
96
98
"ernie-2.0-en" : {
97
99
"do_lower_case" : True
98
100
},
101
+ "ernie-2.0-en-finetuned-squad" : {
102
+ "do_lower_case" : True
103
+ },
99
104
"ernie-2.0-large-en" : {
100
105
"do_lower_case" : True
101
106
},
You can’t perform that action at this time.
0 commit comments