Update config.json

#2
by lewtun HF staff - opened
Files changed (1) hide show
  1. config.json +9 -1
config.json CHANGED
@@ -8,7 +8,15 @@
8
  "dim": 768,
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
 
 
 
 
11
  "initializer_range": 0.02,
 
 
 
 
12
  "max_position_embeddings": 512,
13
  "model_type": "distilbert",
14
  "n_heads": 12,
@@ -20,6 +28,6 @@
20
  "sinusoidal_pos_embds": false,
21
  "tie_weights_": true,
22
  "torch_dtype": "float32",
23
- "transformers_version": "4.21.1",
24
  "vocab_size": 30522
25
  }
 
8
  "dim": 768,
9
  "dropout": 0.1,
10
  "hidden_dim": 3072,
11
+ "id2label": {
12
+ "0": "not_equivalent",
13
+ "1": "equivalent"
14
+ },
15
  "initializer_range": 0.02,
16
+ "label2id": {
17
+ "equivalent": 1,
18
+ "not_equivalent": 0
19
+ },
20
  "max_position_embeddings": 512,
21
  "model_type": "distilbert",
22
  "n_heads": 12,
 
28
  "sinusoidal_pos_embds": false,
29
  "tie_weights_": true,
30
  "torch_dtype": "float32",
31
+ "transformers_version": "4.20.1",
32
  "vocab_size": 30522
33
  }