fix model_config
Browse files- model_config.json +14 -20
model_config.json
CHANGED
@@ -1,22 +1,16 @@
|
|
1 |
{
|
2 |
-
"
|
3 |
-
"
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
"use_task_id": true,
|
17 |
-
"vocab_size": 40000,
|
18 |
-
"init_class": "ErnieModel"
|
19 |
-
}
|
20 |
-
],
|
21 |
-
"init_class": "ErnieForSequenceClassification"
|
22 |
}
|
|
|
1 |
{
|
2 |
+
"attention_probs_dropout_prob": 0.1,
|
3 |
+
"hidden_act": "gelu",
|
4 |
+
"hidden_dropout_prob": 0.1,
|
5 |
+
"hidden_size": 312,
|
6 |
+
"intermediate_size": 1248,
|
7 |
+
"initializer_range": 0.02,
|
8 |
+
"max_position_embeddings": 2048,
|
9 |
+
"num_attention_heads": 12,
|
10 |
+
"num_hidden_layers": 4,
|
11 |
+
"task_type_vocab_size": 16,
|
12 |
+
"type_vocab_size": 4,
|
13 |
+
"use_task_id": true,
|
14 |
+
"vocab_size": 40000,
|
15 |
+
"init_class": "ErnieModel"
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
}
|