Skip to content

Commit 275e62a

Browse files
committed
[ehealth] add parameters of ehealth
1 parent 7153318 commit 275e62a

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

paddlenlp/transformers/electra/modeling.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -225,6 +225,21 @@ class ElectraPretrainedModel(PretrainedModel):
225225
"type_vocab_size": 2,
226226
"vocab_size": 21128,
227227
},
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+
},
228243
}
229244
resource_files_names = {"model_state": "model_state.pdparams"}
230245
pretrained_resource_files_map = {
@@ -239,6 +254,8 @@ class ElectraPretrainedModel(PretrainedModel):
239254
"https://bj.bcebos.com/paddlenlp/models/transformers/chinese-electra-small/chinese-electra-small.pdparams",
240255
"chinese-electra-base":
241256
"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"
242259
}
243260
}
244261

0 commit comments

Comments
 (0)