sijunhe commited on
Commit
d36b98f
1 Parent(s): 7a1a4d3

fix model_config

Browse files
Files changed (1) hide show
  1. model_config.json +14 -20
model_config.json CHANGED
@@ -1,22 +1,16 @@
1
  {
2
- "num_classes": 11,
3
- "init_args": [
4
- {
5
- "attention_probs_dropout_prob": 0.1,
6
- "hidden_act": "gelu",
7
- "hidden_dropout_prob": 0.1,
8
- "hidden_size": 312,
9
- "intermediate_size": 1248,
10
- "initializer_range": 0.02,
11
- "max_position_embeddings": 2048,
12
- "num_attention_heads": 12,
13
- "num_hidden_layers": 4,
14
- "task_type_vocab_size": 16,
15
- "type_vocab_size": 4,
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
  }