File tree Expand file tree Collapse file tree 1 file changed +17
-0
lines changed
paddlenlp/transformers/electra Expand file tree Collapse file tree 1 file changed +17
-0
lines changed Original file line number Diff line number Diff line change @@ -225,6 +225,21 @@ class ElectraPretrainedModel(PretrainedModel):
225
225
"type_vocab_size" : 2 ,
226
226
"vocab_size" : 21128 ,
227
227
},
228
+ "chinese-ehealth" : {
229
+ "attention_probs_dropout_prob" : 0.1 ,
230
+ "embedding_size" : 768 ,
231
+ "hidden_act" : "gelu" ,
232
+ "hidden_dropout_prob" : 0.1 ,
233
+ "hidden_size" : 768 ,
234
+ "initializer_range" : 0.02 ,
235
+ "intermediate_size" : 3072 ,
236
+ "max_position_embeddings" : 512 ,
237
+ "num_attention_heads" : 12 ,
238
+ "num_hidden_layers" : 12 ,
239
+ "pad_token_id" : 0 ,
240
+ "type_vocab_size" : 2 ,
241
+ "vocab_size" : 22608 ,
242
+ },
228
243
}
229
244
resource_files_names = {"model_state" : "model_state.pdparams" }
230
245
pretrained_resource_files_map = {
@@ -239,6 +254,8 @@ class ElectraPretrainedModel(PretrainedModel):
239
254
"https://bj.bcebos.com/paddlenlp/models/transformers/chinese-electra-small/chinese-electra-small.pdparams" ,
240
255
"chinese-electra-base" :
241
256
"https://bj.bcebos.com/paddlenlp/models/transformers/chinese-electra-base/chinese-electra-base.pdparams" ,
257
+ "chinese-ehealth" :
258
+ "https://paddlenlp.bj.bcebos.com/models/transformers/chinese-ehealth.pdparams"
242
259
}
243
260
}
244
261
You can’t perform that action at this time.
0 commit comments