SaiPatil commited on
Commit
70d98db
1 Parent(s): b17bed3

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +13 -19
config.json CHANGED
@@ -1,20 +1,14 @@
1
- {
2
- "architectures": [
3
- "TFWav2Vec2ForSequenceClassification"
4
- ],
5
- "model_type": "wav2vec2",
6
- "hidden_dim": 768,
7
- "num_labels": 10,
8
- "id2label": {
9
- "0": "label_0",
10
- "1": "label_1",
11
- ...
12
- "9": "label_9"
13
- },
14
- "label2id": {
15
- "label_0": 0,
16
- "label_1": 1,
17
- ...
18
- "label_9": 9
19
- }
20
  }
 
1
+ config = {
2
+ "model_type": "transformer",
3
+ "num_hidden_layers": 12,
4
+ "num_attention_heads": 12,
5
+ "hidden_size": 768,
6
+ "intermediate_size": 3072,
7
+ "vocab_size": 30522,
8
+ "max_position_embeddings": 512,
9
+ "type_vocab_size": 2,
10
+ "initializer_range": 0.02,
11
+ "layer_norm_eps": 1e-12,
12
+ "pad_token_id": 0,
13
+ # ... other relevant parameters ...
 
 
 
 
 
 
14
  }