bkbj commited on
Commit
fed2be3
1 Parent(s): df21b00

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +13 -13
config.json CHANGED
@@ -1,15 +1,15 @@
1
  {
2
- "model_type": "bert",
3
- "architectures": ["BertForMaskedLM"],
4
- "hidden_size": 768,
5
- "num_attention_heads": 12,
6
- "num_hidden_layers": 12,
7
- "intermediate_size": 3072,
8
- "hidden_dropout_prob": 0.1,
9
- "attention_probs_dropout_prob": 0.1,
10
- "max_position_embeddings": 512,
11
- "vocab_size": 30522,
12
- "type_vocab_size": 2,
13
- "initializer_range": 0.02,
14
- "layer_norm_eps": 1e-12
15
  }
 
1
  {
2
+ "model_type": "bert",
3
+ "num_labels": 2,
4
+ "architectures": [
5
+ "BertForSequenceClassification"
6
+ ],
7
+ "hidden_size": 768,
8
+ "intermediate_size": 3072,
9
+ "layer_norm_eps": 1e-12,
10
+ "max_position_embeddings": 512,
11
+ "num_attention_heads": 12,
12
+ "num_hidden_layers": 12,
13
+ "type_vocab_size": 2,
14
+ "vocab_size": 30522
15
  }