ofirzaf commited on
Commit
4426b15
1 Parent(s): 8378a24

Fix config file

Browse files
Files changed (1) hide show
  1. config.json +1 -13
config.json CHANGED
@@ -1,25 +1,13 @@
1
  {
2
  "architectures": [
3
- "BertForSequenceClassification"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
- "finetuning_task": "mnli",
7
- "gradient_checkpointing": false,
8
  "hidden_act": "gelu",
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 768,
11
- "id2label": {
12
- "0": "LABEL_0",
13
- "1": "LABEL_1",
14
- "2": "LABEL_2"
15
- },
16
  "initializer_range": 0.02,
17
  "intermediate_size": 3072,
18
- "label2id": {
19
- "LABEL_0": 0,
20
- "LABEL_1": 1,
21
- "LABEL_2": 2
22
- },
23
  "layer_norm_eps": 1e-12,
24
  "max_position_embeddings": 512,
25
  "model_type": "bert",
1
  {
2
  "architectures": [
3
+ "BertForMaskedLM"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
 
 
6
  "hidden_act": "gelu",
7
  "hidden_dropout_prob": 0.1,
8
  "hidden_size": 768,
 
 
 
 
 
9
  "initializer_range": 0.02,
10
  "intermediate_size": 3072,
 
 
 
 
 
11
  "layer_norm_eps": 1e-12,
12
  "max_position_embeddings": 512,
13
  "model_type": "bert",