MatricariaV commited on
Commit
60f058e
·
verified ·
1 Parent(s): 23db635

Duplicated repository

Browse files
README.md ADDED
@@ -0,0 +1,88 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: cc-by-nc-4.0
4
+ base_model: facebook/mms-1b-all
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - wer
9
+ model-index:
10
+ - name: kbd-asr-colab2
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # kbd-asr-colab2
18
+
19
+ This model is a fine-tuned version of [facebook/mms-1b-all](https://huggingface.co/facebook/mms-1b-all) on the None dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.1829
22
+ - Wer: 0.3182
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - learning_rate: 0.001
42
+ - train_batch_size: 8
43
+ - eval_batch_size: 8
44
+ - seed: 42
45
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
46
+ - lr_scheduler_type: linear
47
+ - lr_scheduler_warmup_steps: 100
48
+ - num_epochs: 5
49
+ - mixed_precision_training: Native AMP
50
+
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Validation Loss | Wer |
54
+ |:-------------:|:------:|:----:|:---------------:|:------:|
55
+ | 0.7968 | 0.1871 | 200 | 0.4521 | 0.7382 |
56
+ | 0.6536 | 0.3742 | 400 | 0.3965 | 0.7499 |
57
+ | 0.5989 | 0.5613 | 600 | 0.3850 | 0.7644 |
58
+ | 0.5212 | 0.7484 | 800 | 0.3473 | 0.6373 |
59
+ | 0.5649 | 0.9355 | 1000 | 0.2951 | 0.5633 |
60
+ | 0.4996 | 1.1225 | 1200 | 0.2771 | 0.4806 |
61
+ | 0.5247 | 1.3096 | 1400 | 0.2553 | 0.4592 |
62
+ | 0.4559 | 1.4967 | 1600 | 0.2537 | 0.4396 |
63
+ | 0.5153 | 1.6838 | 1800 | 0.2479 | 0.4364 |
64
+ | 0.4622 | 1.8709 | 2000 | 0.2363 | 0.4251 |
65
+ | 0.4533 | 2.0580 | 2200 | 0.2280 | 0.4091 |
66
+ | 0.4529 | 2.2451 | 2400 | 0.2182 | 0.4066 |
67
+ | 0.4453 | 2.4322 | 2600 | 0.2191 | 0.3902 |
68
+ | 0.4339 | 2.6193 | 2800 | 0.2135 | 0.3815 |
69
+ | 0.4179 | 2.8064 | 3000 | 0.2151 | 0.3906 |
70
+ | 0.3983 | 2.9935 | 3200 | 0.2059 | 0.3725 |
71
+ | 0.3868 | 3.1805 | 3400 | 0.2058 | 0.3614 |
72
+ | 0.3804 | 3.3676 | 3600 | 0.1959 | 0.3558 |
73
+ | 0.3934 | 3.5547 | 3800 | 0.2015 | 0.3555 |
74
+ | 0.3693 | 3.7418 | 4000 | 0.1995 | 0.3587 |
75
+ | 0.3678 | 3.9289 | 4200 | 0.2017 | 0.3876 |
76
+ | 0.4097 | 4.1160 | 4400 | 0.1893 | 0.3355 |
77
+ | 0.34 | 4.3031 | 4600 | 0.1877 | 0.3291 |
78
+ | 0.3553 | 4.4902 | 4800 | 0.1857 | 0.3222 |
79
+ | 0.3133 | 4.6773 | 5000 | 0.1839 | 0.3224 |
80
+ | 0.3077 | 4.8644 | 5200 | 0.1829 | 0.3182 |
81
+
82
+
83
+ ### Framework versions
84
+
85
+ - Transformers 4.50.0.dev0
86
+ - Pytorch 2.5.1+cu121
87
+ - Datasets 3.3.2
88
+ - Tokenizers 0.21.0
adapter.kbd.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6904a9fc8edf02a6cf85f1511b73580044307e84cbc5453f777967e8e3922b7c
3
+ size 8839532
added_tokens.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "</s>": 38,
3
+ "<s>": 37
4
+ }
config.json ADDED
@@ -0,0 +1,108 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/mms-1b-all",
3
+ "activation_dropout": 0.05,
4
+ "adapter_attn_dim": 16,
5
+ "adapter_kernel_size": 3,
6
+ "adapter_stride": 2,
7
+ "add_adapter": false,
8
+ "apply_spec_augment": true,
9
+ "architectures": [
10
+ "Wav2Vec2ForCTC"
11
+ ],
12
+ "attention_dropout": 0.0,
13
+ "bos_token_id": 1,
14
+ "classifier_proj_size": 256,
15
+ "codevector_dim": 1024,
16
+ "contrastive_logits_temperature": 0.1,
17
+ "conv_bias": true,
18
+ "conv_dim": [
19
+ 512,
20
+ 512,
21
+ 512,
22
+ 512,
23
+ 512,
24
+ 512,
25
+ 512
26
+ ],
27
+ "conv_kernel": [
28
+ 10,
29
+ 3,
30
+ 3,
31
+ 3,
32
+ 3,
33
+ 2,
34
+ 2
35
+ ],
36
+ "conv_stride": [
37
+ 5,
38
+ 2,
39
+ 2,
40
+ 2,
41
+ 2,
42
+ 2,
43
+ 2
44
+ ],
45
+ "ctc_loss_reduction": "mean",
46
+ "ctc_zero_infinity": false,
47
+ "diversity_loss_weight": 0.1,
48
+ "do_stable_layer_norm": true,
49
+ "eos_token_id": 2,
50
+ "feat_extract_activation": "gelu",
51
+ "feat_extract_dropout": 0.0,
52
+ "feat_extract_norm": "layer",
53
+ "feat_proj_dropout": 0.0,
54
+ "feat_quantizer_dropout": 0.0,
55
+ "final_dropout": 0.05,
56
+ "hidden_act": "gelu",
57
+ "hidden_dropout": 0.0,
58
+ "hidden_size": 1280,
59
+ "initializer_range": 0.02,
60
+ "intermediate_size": 5120,
61
+ "layer_norm_eps": 1e-05,
62
+ "layerdrop": 0.0,
63
+ "mask_feature_length": 10,
64
+ "mask_feature_min_masks": 0,
65
+ "mask_feature_prob": 0.0,
66
+ "mask_time_length": 10,
67
+ "mask_time_min_masks": 2,
68
+ "mask_time_prob": 0.05,
69
+ "model_type": "wav2vec2",
70
+ "num_adapter_layers": 3,
71
+ "num_attention_heads": 16,
72
+ "num_codevector_groups": 2,
73
+ "num_codevectors_per_group": 320,
74
+ "num_conv_pos_embedding_groups": 16,
75
+ "num_conv_pos_embeddings": 128,
76
+ "num_feat_extract_layers": 7,
77
+ "num_hidden_layers": 48,
78
+ "num_negatives": 100,
79
+ "output_hidden_size": 1280,
80
+ "pad_token_id": 36,
81
+ "proj_codevector_dim": 1024,
82
+ "tdnn_dilation": [
83
+ 1,
84
+ 2,
85
+ 3,
86
+ 1,
87
+ 1
88
+ ],
89
+ "tdnn_dim": [
90
+ 512,
91
+ 512,
92
+ 512,
93
+ 512,
94
+ 1500
95
+ ],
96
+ "tdnn_kernel": [
97
+ 5,
98
+ 3,
99
+ 3,
100
+ 1,
101
+ 1
102
+ ],
103
+ "torch_dtype": "float32",
104
+ "transformers_version": "4.50.0.dev0",
105
+ "use_weighted_layer_sum": false,
106
+ "vocab_size": 39,
107
+ "xvector_output_dim": 512
108
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6f083d60000afa5c299a27313a26dd929fcb689974ffe00ecf6234e900c5e958
3
+ size 3858931916
preprocessor_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "feature_extractor_type": "Wav2Vec2FeatureExtractor",
4
+ "feature_size": 1,
5
+ "padding_side": "right",
6
+ "padding_value": 0.0,
7
+ "return_attention_mask": true,
8
+ "sampling_rate": 16000
9
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<s>",
3
+ "eos_token": "</s>",
4
+ "pad_token": "[PAD]",
5
+ "unk_token": "[UNK]"
6
+ }
tokenizer_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "35": {
4
+ "content": "[UNK]",
5
+ "lstrip": true,
6
+ "normalized": false,
7
+ "rstrip": true,
8
+ "single_word": false,
9
+ "special": false
10
+ },
11
+ "36": {
12
+ "content": "[PAD]",
13
+ "lstrip": true,
14
+ "normalized": false,
15
+ "rstrip": true,
16
+ "single_word": false,
17
+ "special": false
18
+ },
19
+ "37": {
20
+ "content": "<s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "38": {
28
+ "content": "</s>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ }
35
+ },
36
+ "bos_token": "<s>",
37
+ "clean_up_tokenization_spaces": false,
38
+ "do_lower_case": false,
39
+ "eos_token": "</s>",
40
+ "extra_special_tokens": {},
41
+ "model_max_length": 1000000000000000019884624838656,
42
+ "pad_token": "[PAD]",
43
+ "replace_word_delimiter_char": " ",
44
+ "target_lang": "kbd",
45
+ "tokenizer_class": "Wav2Vec2CTCTokenizer",
46
+ "unk_token": "[UNK]",
47
+ "word_delimiter_token": "*"
48
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61d77501bc4982904d3a78466efd119013e60105421b6db82cc6bc452a79e922
3
+ size 5304
vocab.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "kbd": {
3
+ "*": 0,
4
+ "[PAD]": 36,
5
+ "[UNK]": 35,
6
+ "а": 1,
7
+ "б": 2,
8
+ "в": 3,
9
+ "г": 4,
10
+ "д": 5,
11
+ "е": 6,
12
+ "ж": 7,
13
+ "з": 8,
14
+ "и": 9,
15
+ "й": 10,
16
+ "к": 11,
17
+ "л": 12,
18
+ "м": 13,
19
+ "н": 14,
20
+ "о": 15,
21
+ "п": 16,
22
+ "р": 17,
23
+ "с": 18,
24
+ "т": 19,
25
+ "у": 20,
26
+ "ф": 21,
27
+ "х": 22,
28
+ "ц": 23,
29
+ "ч": 24,
30
+ "ш": 25,
31
+ "щ": 26,
32
+ "ъ": 27,
33
+ "ы": 28,
34
+ "ь": 29,
35
+ "э": 30,
36
+ "ю": 31,
37
+ "я": 32,
38
+ "ё": 33,
39
+ "ӏ": 34
40
+ }
41
+ }