ft-wav2vec2-with-minds
Browse files- README.md +56 -56
- config.json +20 -23
- model.safetensors +2 -2
- training_args.bin +1 -1
README.md
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
---
|
2 |
-
license:
|
3 |
-
base_model:
|
4 |
tags:
|
5 |
- generated_from_trainer
|
6 |
metrics:
|
@@ -15,10 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
|
|
15 |
|
16 |
# ft-wav2vec2-with-minds
|
17 |
|
18 |
-
This model is a fine-tuned version of [
|
19 |
It achieves the following results on the evaluation set:
|
20 |
-
- Loss: 0.
|
21 |
-
- Accuracy: 0.
|
22 |
|
23 |
## Model description
|
24 |
|
@@ -37,7 +37,7 @@ More information needed
|
|
37 |
### Training hyperparameters
|
38 |
|
39 |
The following hyperparameters were used during training:
|
40 |
-
- learning_rate:
|
41 |
- train_batch_size: 120
|
42 |
- eval_batch_size: 120
|
43 |
- seed: 42
|
@@ -53,56 +53,56 @@ The following hyperparameters were used during training:
|
|
53 |
|
54 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
55 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
56 |
-
| 2.
|
57 |
-
| 2.
|
58 |
-
| 2.
|
59 |
-
|
|
60 |
-
|
|
61 |
-
|
|
62 |
-
|
|
63 |
-
| 1.
|
64 |
-
|
|
65 |
-
|
|
66 |
-
|
|
67 |
-
|
|
68 |
-
|
|
69 |
-
|
|
70 |
-
|
|
71 |
-
|
|
72 |
-
|
|
73 |
-
| 0.
|
74 |
-
| 0.
|
75 |
-
| 0.
|
76 |
-
| 0.
|
77 |
-
| 0.
|
78 |
-
| 0.
|
79 |
-
| 0.
|
80 |
-
| 0.
|
81 |
-
| 0.
|
82 |
-
| 0.
|
83 |
-
| 0.
|
84 |
-
| 0.
|
85 |
-
| 0.
|
86 |
-
| 0.
|
87 |
-
| 0.
|
88 |
-
| 0.
|
89 |
-
| 0.
|
90 |
-
| 0.
|
91 |
-
| 0.
|
92 |
-
| 0.
|
93 |
-
| 0.
|
94 |
-
| 0.
|
95 |
-
| 0.
|
96 |
-
| 0.
|
97 |
-
| 0.
|
98 |
-
| 0.
|
99 |
-
| 0.
|
100 |
-
| 0.
|
101 |
-
| 0.
|
102 |
-
| 0.
|
103 |
-
| 0.
|
104 |
-
| 0.
|
105 |
-
| 0.
|
106 |
|
107 |
|
108 |
### Framework versions
|
|
|
1 |
---
|
2 |
+
license: apache-2.0
|
3 |
+
base_model: facebook/wav2vec2-large
|
4 |
tags:
|
5 |
- generated_from_trainer
|
6 |
metrics:
|
|
|
15 |
|
16 |
# ft-wav2vec2-with-minds
|
17 |
|
18 |
+
This model is a fine-tuned version of [facebook/wav2vec2-large](https://huggingface.co/facebook/wav2vec2-large) on the None dataset.
|
19 |
It achieves the following results on the evaluation set:
|
20 |
+
- Loss: 0.0732
|
21 |
+
- Accuracy: 0.9822
|
22 |
|
23 |
## Model description
|
24 |
|
|
|
37 |
### Training hyperparameters
|
38 |
|
39 |
The following hyperparameters were used during training:
|
40 |
+
- learning_rate: 3e-05
|
41 |
- train_batch_size: 120
|
42 |
- eval_batch_size: 120
|
43 |
- seed: 42
|
|
|
53 |
|
54 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
55 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
56 |
+
| 2.0814 | 1.0 | 9 | 2.0883 | 0.1143 |
|
57 |
+
| 2.064 | 2.0 | 18 | 2.0619 | 0.1678 |
|
58 |
+
| 2.0232 | 3.0 | 27 | 1.9712 | 0.2709 |
|
59 |
+
| 1.861 | 4.0 | 36 | 1.7455 | 0.3880 |
|
60 |
+
| 1.6003 | 5.0 | 45 | 1.5115 | 0.4724 |
|
61 |
+
| 1.4972 | 6.0 | 54 | 1.2623 | 0.5998 |
|
62 |
+
| 1.2332 | 7.0 | 63 | 1.0138 | 0.6935 |
|
63 |
+
| 1.081 | 8.0 | 72 | 0.8169 | 0.7601 |
|
64 |
+
| 0.9925 | 9.0 | 81 | 0.7757 | 0.7873 |
|
65 |
+
| 0.8516 | 10.0 | 90 | 0.6470 | 0.8163 |
|
66 |
+
| 0.7544 | 11.0 | 99 | 0.7208 | 0.7873 |
|
67 |
+
| 0.7006 | 12.0 | 108 | 0.5074 | 0.8557 |
|
68 |
+
| 0.591 | 13.0 | 117 | 0.4326 | 0.8782 |
|
69 |
+
| 0.5155 | 14.0 | 126 | 0.3707 | 0.9053 |
|
70 |
+
| 0.4715 | 15.0 | 135 | 0.3116 | 0.9091 |
|
71 |
+
| 0.4461 | 16.0 | 144 | 0.3167 | 0.9138 |
|
72 |
+
| 0.445 | 17.0 | 153 | 0.2963 | 0.9250 |
|
73 |
+
| 0.3899 | 18.0 | 162 | 0.2499 | 0.9353 |
|
74 |
+
| 0.3656 | 19.0 | 171 | 0.2756 | 0.9194 |
|
75 |
+
| 0.3255 | 20.0 | 180 | 0.2280 | 0.9297 |
|
76 |
+
| 0.2756 | 21.0 | 189 | 0.2178 | 0.9438 |
|
77 |
+
| 0.3119 | 22.0 | 198 | 0.1858 | 0.9513 |
|
78 |
+
| 0.2595 | 23.0 | 207 | 0.1794 | 0.9475 |
|
79 |
+
| 0.2713 | 24.0 | 216 | 0.1737 | 0.9466 |
|
80 |
+
| 0.2336 | 25.0 | 225 | 0.1758 | 0.9531 |
|
81 |
+
| 0.2359 | 26.0 | 234 | 0.1690 | 0.9485 |
|
82 |
+
| 0.2229 | 27.0 | 243 | 0.1336 | 0.9606 |
|
83 |
+
| 0.2145 | 28.0 | 252 | 0.1338 | 0.9700 |
|
84 |
+
| 0.1986 | 29.0 | 261 | 0.1525 | 0.9625 |
|
85 |
+
| 0.1811 | 30.0 | 270 | 0.1415 | 0.9653 |
|
86 |
+
| 0.165 | 31.0 | 279 | 0.1208 | 0.9672 |
|
87 |
+
| 0.1755 | 32.0 | 288 | 0.1266 | 0.9634 |
|
88 |
+
| 0.175 | 33.0 | 297 | 0.1269 | 0.9672 |
|
89 |
+
| 0.149 | 34.0 | 306 | 0.1072 | 0.9728 |
|
90 |
+
| 0.1606 | 35.0 | 315 | 0.1183 | 0.9738 |
|
91 |
+
| 0.161 | 36.0 | 324 | 0.1009 | 0.9719 |
|
92 |
+
| 0.1533 | 37.0 | 333 | 0.1000 | 0.9728 |
|
93 |
+
| 0.1239 | 38.0 | 342 | 0.1109 | 0.9691 |
|
94 |
+
| 0.1353 | 39.0 | 351 | 0.0905 | 0.9775 |
|
95 |
+
| 0.1287 | 40.0 | 360 | 0.0920 | 0.9738 |
|
96 |
+
| 0.223 | 41.0 | 369 | 0.0855 | 0.9775 |
|
97 |
+
| 0.1302 | 42.0 | 378 | 0.0748 | 0.9794 |
|
98 |
+
| 0.1249 | 43.0 | 387 | 0.0732 | 0.9822 |
|
99 |
+
| 0.1552 | 44.0 | 396 | 0.0688 | 0.9822 |
|
100 |
+
| 0.098 | 45.0 | 405 | 0.0777 | 0.9766 |
|
101 |
+
| 0.1459 | 46.0 | 414 | 0.0634 | 0.9813 |
|
102 |
+
| 0.1267 | 47.0 | 423 | 0.0653 | 0.9822 |
|
103 |
+
| 0.149 | 48.0 | 432 | 0.0709 | 0.9794 |
|
104 |
+
| 0.1135 | 49.0 | 441 | 0.0660 | 0.9813 |
|
105 |
+
| 0.118 | 50.0 | 450 | 0.0652 | 0.9813 |
|
106 |
|
107 |
|
108 |
### Framework versions
|
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"activation_dropout": 0.1,
|
4 |
"adapter_attn_dim": null,
|
5 |
"adapter_kernel_size": 3,
|
@@ -14,7 +14,7 @@
|
|
14 |
"classifier_proj_size": 256,
|
15 |
"codevector_dim": 768,
|
16 |
"contrastive_logits_temperature": 0.1,
|
17 |
-
"conv_bias":
|
18 |
"conv_dim": [
|
19 |
512,
|
20 |
512,
|
@@ -45,15 +45,14 @@
|
|
45 |
"ctc_loss_reduction": "sum",
|
46 |
"ctc_zero_infinity": false,
|
47 |
"diversity_loss_weight": 0.1,
|
48 |
-
"do_stable_layer_norm":
|
49 |
"eos_token_id": 2,
|
50 |
"feat_extract_activation": "gelu",
|
51 |
"feat_extract_dropout": 0.0,
|
52 |
-
"feat_extract_norm": "
|
53 |
"feat_proj_dropout": 0.1,
|
54 |
"feat_quantizer_dropout": 0.0,
|
55 |
"final_dropout": 0.1,
|
56 |
-
"finetuning_task": "wav2vec2_reg",
|
57 |
"gradient_checkpointing": false,
|
58 |
"hidden_act": "gelu",
|
59 |
"hidden_dropout": 0.1,
|
@@ -61,25 +60,25 @@
|
|
61 |
"hidden_size": 1024,
|
62 |
"id2label": {
|
63 |
"0": "stop",
|
64 |
-
"1": "
|
65 |
-
"2": "
|
66 |
-
"3": "
|
67 |
-
"4": "
|
68 |
-
"5": "
|
69 |
-
"6": "
|
70 |
-
"7": "
|
71 |
},
|
72 |
"initializer_range": 0.02,
|
73 |
"intermediate_size": 4096,
|
74 |
"label2id": {
|
75 |
-
"down": "
|
76 |
-
"go": "
|
77 |
-
"left": "
|
78 |
-
"no": "
|
79 |
-
"right": "
|
80 |
"stop": "0",
|
81 |
-
"up": "
|
82 |
-
"yes": "
|
83 |
},
|
84 |
"layer_norm_eps": 1e-05,
|
85 |
"layerdrop": 0.1,
|
@@ -97,12 +96,10 @@
|
|
97 |
"num_conv_pos_embedding_groups": 16,
|
98 |
"num_conv_pos_embeddings": 128,
|
99 |
"num_feat_extract_layers": 7,
|
100 |
-
"num_hidden_layers":
|
101 |
"num_negatives": 100,
|
102 |
"output_hidden_size": 1024,
|
103 |
"pad_token_id": 0,
|
104 |
-
"pooling_mode": "mean",
|
105 |
-
"problem_type": "regression",
|
106 |
"proj_codevector_dim": 768,
|
107 |
"tdnn_dilation": [
|
108 |
1,
|
@@ -128,6 +125,6 @@
|
|
128 |
"torch_dtype": "float32",
|
129 |
"transformers_version": "4.35.2",
|
130 |
"use_weighted_layer_sum": false,
|
131 |
-
"vocab_size":
|
132 |
"xvector_output_dim": 512
|
133 |
}
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "facebook/wav2vec2-large",
|
3 |
"activation_dropout": 0.1,
|
4 |
"adapter_attn_dim": null,
|
5 |
"adapter_kernel_size": 3,
|
|
|
14 |
"classifier_proj_size": 256,
|
15 |
"codevector_dim": 768,
|
16 |
"contrastive_logits_temperature": 0.1,
|
17 |
+
"conv_bias": false,
|
18 |
"conv_dim": [
|
19 |
512,
|
20 |
512,
|
|
|
45 |
"ctc_loss_reduction": "sum",
|
46 |
"ctc_zero_infinity": false,
|
47 |
"diversity_loss_weight": 0.1,
|
48 |
+
"do_stable_layer_norm": false,
|
49 |
"eos_token_id": 2,
|
50 |
"feat_extract_activation": "gelu",
|
51 |
"feat_extract_dropout": 0.0,
|
52 |
+
"feat_extract_norm": "group",
|
53 |
"feat_proj_dropout": 0.1,
|
54 |
"feat_quantizer_dropout": 0.0,
|
55 |
"final_dropout": 0.1,
|
|
|
56 |
"gradient_checkpointing": false,
|
57 |
"hidden_act": "gelu",
|
58 |
"hidden_dropout": 0.1,
|
|
|
60 |
"hidden_size": 1024,
|
61 |
"id2label": {
|
62 |
"0": "stop",
|
63 |
+
"1": "down",
|
64 |
+
"2": "no",
|
65 |
+
"3": "left",
|
66 |
+
"4": "right",
|
67 |
+
"5": "yes",
|
68 |
+
"6": "go",
|
69 |
+
"7": "up"
|
70 |
},
|
71 |
"initializer_range": 0.02,
|
72 |
"intermediate_size": 4096,
|
73 |
"label2id": {
|
74 |
+
"down": "1",
|
75 |
+
"go": "6",
|
76 |
+
"left": "3",
|
77 |
+
"no": "2",
|
78 |
+
"right": "4",
|
79 |
"stop": "0",
|
80 |
+
"up": "7",
|
81 |
+
"yes": "5"
|
82 |
},
|
83 |
"layer_norm_eps": 1e-05,
|
84 |
"layerdrop": 0.1,
|
|
|
96 |
"num_conv_pos_embedding_groups": 16,
|
97 |
"num_conv_pos_embeddings": 128,
|
98 |
"num_feat_extract_layers": 7,
|
99 |
+
"num_hidden_layers": 24,
|
100 |
"num_negatives": 100,
|
101 |
"output_hidden_size": 1024,
|
102 |
"pad_token_id": 0,
|
|
|
|
|
103 |
"proj_codevector_dim": 768,
|
104 |
"tdnn_dilation": [
|
105 |
1,
|
|
|
125 |
"torch_dtype": "float32",
|
126 |
"transformers_version": "4.35.2",
|
127 |
"use_weighted_layer_sum": false,
|
128 |
+
"vocab_size": 32,
|
129 |
"xvector_output_dim": 512
|
130 |
}
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4a398bcfd35483061316b87c40c842401a1160ced1b8a5ae390e39a02ca46c6f
|
3 |
+
size 1262824104
|
training_args.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 4143
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a2eed9312a29a1bf16959421f1afa06489073a52b944433d18f7e5f37fd0d6e8
|
3 |
size 4143
|