gokuls commited on
Commit
631b362
·
verified ·
1 Parent(s): 7e0bb3b

End of training

Browse files
README.md ADDED
@@ -0,0 +1,96 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: facebook/hubert-base-ls960
4
+ tags:
5
+ - generated_from_trainer
6
+ datasets:
7
+ - speech_commands
8
+ metrics:
9
+ - accuracy
10
+ model-index:
11
+ - name: hubert-base-ls960-speech-commands
12
+ results:
13
+ - task:
14
+ name: Audio Classification
15
+ type: audio-classification
16
+ dataset:
17
+ name: speech_commands
18
+ type: speech_commands
19
+ config: v0.02
20
+ split: None
21
+ args: v0.02
22
+ metrics:
23
+ - name: Accuracy
24
+ type: accuracy
25
+ value: 0.8057553956834532
26
+ ---
27
+
28
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
29
+ should probably proofread and complete it, then remove this comment. -->
30
+
31
+ # hubert-base-ls960-speech-commands
32
+
33
+ This model is a fine-tuned version of [facebook/hubert-base-ls960](https://huggingface.co/facebook/hubert-base-ls960) on the speech_commands dataset.
34
+ It achieves the following results on the evaluation set:
35
+ - Loss: 1.0829
36
+ - Accuracy: 0.8058
37
+
38
+ ## Model description
39
+
40
+ More information needed
41
+
42
+ ## Intended uses & limitations
43
+
44
+ More information needed
45
+
46
+ ## Training and evaluation data
47
+
48
+ More information needed
49
+
50
+ ## Training procedure
51
+
52
+ ### Training hyperparameters
53
+
54
+ The following hyperparameters were used during training:
55
+ - learning_rate: 5e-05
56
+ - train_batch_size: 48
57
+ - eval_batch_size: 48
58
+ - seed: 42
59
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
60
+ - lr_scheduler_type: linear
61
+ - lr_scheduler_warmup_ratio: 0.1
62
+ - num_epochs: 20
63
+ - mixed_precision_training: Native AMP
64
+
65
+ ### Training results
66
+
67
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
68
+ |:-------------:|:-----:|:-----:|:---------------:|:--------:|
69
+ | 1.8285 | 1.0 | 824 | 1.9509 | 0.7167 |
70
+ | 0.5292 | 2.0 | 1648 | 1.3813 | 0.7909 |
71
+ | 0.3554 | 3.0 | 2472 | 1.1773 | 0.7941 |
72
+ | 0.2873 | 4.0 | 3296 | 1.2437 | 0.7981 |
73
+ | 0.2525 | 5.0 | 4120 | 1.2514 | 0.8004 |
74
+ | 0.2941 | 6.0 | 4944 | 1.2243 | 0.7995 |
75
+ | 0.1809 | 7.0 | 5768 | 1.1965 | 0.8008 |
76
+ | 0.2313 | 8.0 | 6592 | 1.0694 | 0.8022 |
77
+ | 0.1917 | 9.0 | 7416 | 1.0618 | 0.7995 |
78
+ | 0.1212 | 10.0 | 8240 | 1.0972 | 0.8026 |
79
+ | 0.185 | 11.0 | 9064 | 1.0868 | 0.8017 |
80
+ | 0.143 | 12.0 | 9888 | 1.1558 | 0.8031 |
81
+ | 0.2227 | 13.0 | 10712 | 1.0550 | 0.8040 |
82
+ | 0.1884 | 14.0 | 11536 | 1.0384 | 0.8022 |
83
+ | 0.1183 | 15.0 | 12360 | 1.0169 | 0.8035 |
84
+ | 0.1849 | 16.0 | 13184 | 1.0061 | 0.8035 |
85
+ | 0.141 | 17.0 | 14008 | 1.0337 | 0.8053 |
86
+ | 0.1328 | 18.0 | 14832 | 1.0829 | 0.8058 |
87
+ | 0.1238 | 19.0 | 15656 | 1.0576 | 0.8053 |
88
+ | 0.0932 | 20.0 | 16480 | 1.0641 | 0.8053 |
89
+
90
+
91
+ ### Framework versions
92
+
93
+ - Transformers 4.43.3
94
+ - Pytorch 2.2.2+cu121
95
+ - Datasets 2.18.0
96
+ - Tokenizers 0.19.1
config.json ADDED
@@ -0,0 +1,152 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "facebook/hubert-base-ls960",
3
+ "activation_dropout": 0.1,
4
+ "apply_spec_augment": true,
5
+ "architectures": [
6
+ "HubertForSequenceClassification"
7
+ ],
8
+ "attention_dropout": 0.1,
9
+ "bos_token_id": 1,
10
+ "classifier_proj_size": 256,
11
+ "conv_bias": false,
12
+ "conv_dim": [
13
+ 512,
14
+ 512,
15
+ 512,
16
+ 512,
17
+ 512,
18
+ 512,
19
+ 512
20
+ ],
21
+ "conv_kernel": [
22
+ 10,
23
+ 3,
24
+ 3,
25
+ 3,
26
+ 3,
27
+ 2,
28
+ 2
29
+ ],
30
+ "conv_stride": [
31
+ 5,
32
+ 2,
33
+ 2,
34
+ 2,
35
+ 2,
36
+ 2,
37
+ 2
38
+ ],
39
+ "ctc_loss_reduction": "sum",
40
+ "ctc_zero_infinity": false,
41
+ "do_stable_layer_norm": false,
42
+ "eos_token_id": 2,
43
+ "feat_extract_activation": "gelu",
44
+ "feat_extract_dropout": 0.0,
45
+ "feat_extract_norm": "group",
46
+ "feat_proj_dropout": 0.1,
47
+ "feat_proj_layer_norm": true,
48
+ "final_dropout": 0.1,
49
+ "gradient_checkpointing": false,
50
+ "hidden_act": "gelu",
51
+ "hidden_dropout": 0.1,
52
+ "hidden_dropout_prob": 0.1,
53
+ "hidden_size": 768,
54
+ "id2label": {
55
+ "0": "yes",
56
+ "1": "no",
57
+ "10": "zero",
58
+ "11": "one",
59
+ "12": "two",
60
+ "13": "three",
61
+ "14": "four",
62
+ "15": "five",
63
+ "16": "six",
64
+ "17": "seven",
65
+ "18": "eight",
66
+ "19": "nine",
67
+ "2": "up",
68
+ "20": "bed",
69
+ "21": "bird",
70
+ "22": "cat",
71
+ "23": "dog",
72
+ "24": "happy",
73
+ "25": "house",
74
+ "26": "marvin",
75
+ "27": "sheila",
76
+ "28": "tree",
77
+ "29": "wow",
78
+ "3": "down",
79
+ "30": "backward",
80
+ "31": "forward",
81
+ "32": "follow",
82
+ "33": "learn",
83
+ "34": "visual",
84
+ "35": "_silence_",
85
+ "4": "left",
86
+ "5": "right",
87
+ "6": "on",
88
+ "7": "off",
89
+ "8": "stop",
90
+ "9": "go"
91
+ },
92
+ "initializer_range": 0.02,
93
+ "intermediate_size": 3072,
94
+ "label2id": {
95
+ "_silence_": "35",
96
+ "backward": "30",
97
+ "bed": "20",
98
+ "bird": "21",
99
+ "cat": "22",
100
+ "dog": "23",
101
+ "down": "3",
102
+ "eight": "18",
103
+ "five": "15",
104
+ "follow": "32",
105
+ "forward": "31",
106
+ "four": "14",
107
+ "go": "9",
108
+ "happy": "24",
109
+ "house": "25",
110
+ "learn": "33",
111
+ "left": "4",
112
+ "marvin": "26",
113
+ "nine": "19",
114
+ "no": "1",
115
+ "off": "7",
116
+ "on": "6",
117
+ "one": "11",
118
+ "right": "5",
119
+ "seven": "17",
120
+ "sheila": "27",
121
+ "six": "16",
122
+ "stop": "8",
123
+ "three": "13",
124
+ "tree": "28",
125
+ "two": "12",
126
+ "up": "2",
127
+ "visual": "34",
128
+ "wow": "29",
129
+ "yes": "0",
130
+ "zero": "10"
131
+ },
132
+ "layer_norm_eps": 1e-05,
133
+ "layerdrop": 0.1,
134
+ "mask_feature_length": 10,
135
+ "mask_feature_min_masks": 0,
136
+ "mask_feature_prob": 0.0,
137
+ "mask_time_length": 10,
138
+ "mask_time_min_masks": 2,
139
+ "mask_time_prob": 0.05,
140
+ "model_type": "hubert",
141
+ "num_attention_heads": 12,
142
+ "num_conv_pos_embedding_groups": 16,
143
+ "num_conv_pos_embeddings": 128,
144
+ "num_feat_extract_layers": 7,
145
+ "num_hidden_layers": 12,
146
+ "pad_token_id": 0,
147
+ "tokenizer_class": "Wav2Vec2CTCTokenizer",
148
+ "torch_dtype": "float32",
149
+ "transformers_version": "4.43.3",
150
+ "use_weighted_layer_sum": false,
151
+ "vocab_size": 32
152
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e23706f3a820cb4ed25d98a9e0cb2ea2032dfea0f63b20722f69038ec53efb92
3
+ size 378336904
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,
7
+ "return_attention_mask": false,
8
+ "sampling_rate": 16000
9
+ }
runs/Oct01_09-39-27_ki-g0010/events.out.tfevents.1727775572.ki-g0010.1555826.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0d0282c5e7ae248a4ea355f7c3776cc16eef4295e7beb4d8c21740a413db90ab
3
+ size 13880
runs/Oct01_09-41-54_ki-g0010/events.out.tfevents.1727775719.ki-g0010.1558223.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b98ac2fe1d84a6d29b21d1f0f308f6a4f9c22849e0fb92f1261f91b08b8db9ff
3
+ size 709490
runs/Oct01_09-41-54_ki-g0010/events.out.tfevents.1727776568.ki-g0010.1558223.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23ccf8e62d3c3f242753631b01ba3cb1f2610879c81d0236d5929181718d2891
3
+ size 417
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc7a797b97bc4aa390371970b1809629e0ee56acbe55467eb33f8259bb6c9eff
3
+ size 5240