hiwden00 commited on
Commit
29a6316
1 Parent(s): cd02f53

Model save

Browse files
Files changed (2) hide show
  1. README.md +73 -0
  2. generation_config.json +167 -0
README.md ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: openai/whisper-base
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - wer
9
+ model-index:
10
+ - name: fs-w-he-base
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
+ # fs-w-he-base
18
+
19
+ This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on an unknown dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 1.1453
22
+ - Wer: 110.8262
23
+ - Cer: 93.0350
24
+
25
+ ## Model description
26
+
27
+ More information needed
28
+
29
+ ## Intended uses & limitations
30
+
31
+ More information needed
32
+
33
+ ## Training and evaluation data
34
+
35
+ More information needed
36
+
37
+ ## Training procedure
38
+
39
+ ### Training hyperparameters
40
+
41
+ The following hyperparameters were used during training:
42
+ - learning_rate: 1e-05
43
+ - train_batch_size: 16
44
+ - eval_batch_size: 8
45
+ - seed: 42
46
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
+ - lr_scheduler_type: linear
48
+ - lr_scheduler_warmup_steps: 500
49
+ - training_steps: 5000
50
+ - mixed_precision_training: Native AMP
51
+
52
+ ### Training results
53
+
54
+ | Training Loss | Epoch | Step | Validation Loss | Wer | Cer |
55
+ |:-------------:|:-------:|:----:|:---------------:|:--------:|:--------:|
56
+ | 7.2383 | 4.5872 | 500 | 7.6104 | 178.9174 | 140.9825 |
57
+ | 1.348 | 9.1743 | 1000 | 2.0785 | 113.0579 | 96.7795 |
58
+ | 0.4859 | 13.7615 | 1500 | 1.1437 | 132.2412 | 113.7066 |
59
+ | 0.2809 | 18.3486 | 2000 | 0.9320 | 156.9801 | 142.7087 |
60
+ | 0.1568 | 22.9358 | 2500 | 0.9841 | 124.2640 | 115.1666 |
61
+ | 0.041 | 27.5229 | 3000 | 1.0475 | 130.6743 | 110.1941 |
62
+ | 0.0091 | 32.1101 | 3500 | 1.0947 | 124.5014 | 103.2205 |
63
+ | 0.0016 | 36.6972 | 4000 | 1.1475 | 137.9392 | 119.5981 |
64
+ | 0.0001 | 41.2844 | 4500 | 1.1431 | 124.0741 | 103.3494 |
65
+ | 0.0001 | 45.8716 | 5000 | 1.1453 | 110.8262 | 93.0350 |
66
+
67
+
68
+ ### Framework versions
69
+
70
+ - Transformers 4.45.1
71
+ - Pytorch 2.4.1+cu121
72
+ - Datasets 3.0.1
73
+ - Tokenizers 0.20.0
generation_config.json ADDED
@@ -0,0 +1,167 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alignment_heads": [
3
+ [
4
+ 3,
5
+ 1
6
+ ],
7
+ [
8
+ 4,
9
+ 2
10
+ ],
11
+ [
12
+ 4,
13
+ 3
14
+ ],
15
+ [
16
+ 4,
17
+ 7
18
+ ],
19
+ [
20
+ 5,
21
+ 1
22
+ ],
23
+ [
24
+ 5,
25
+ 2
26
+ ],
27
+ [
28
+ 5,
29
+ 4
30
+ ],
31
+ [
32
+ 5,
33
+ 6
34
+ ]
35
+ ],
36
+ "begin_suppress_tokens": [
37
+ 220,
38
+ 50257
39
+ ],
40
+ "bos_token_id": 50257,
41
+ "decoder_start_token_id": 50258,
42
+ "eos_token_id": 50257,
43
+ "forced_decoder_ids": [
44
+ [
45
+ 1,
46
+ null
47
+ ],
48
+ [
49
+ 2,
50
+ 50359
51
+ ]
52
+ ],
53
+ "is_multilingual": true,
54
+ "lang_to_id": {
55
+ "<|af|>": 50327,
56
+ "<|am|>": 50334,
57
+ "<|ar|>": 50272,
58
+ "<|as|>": 50350,
59
+ "<|az|>": 50304,
60
+ "<|ba|>": 50355,
61
+ "<|be|>": 50330,
62
+ "<|bg|>": 50292,
63
+ "<|bn|>": 50302,
64
+ "<|bo|>": 50347,
65
+ "<|br|>": 50309,
66
+ "<|bs|>": 50315,
67
+ "<|ca|>": 50270,
68
+ "<|cs|>": 50283,
69
+ "<|cy|>": 50297,
70
+ "<|da|>": 50285,
71
+ "<|de|>": 50261,
72
+ "<|el|>": 50281,
73
+ "<|en|>": 50259,
74
+ "<|es|>": 50262,
75
+ "<|et|>": 50307,
76
+ "<|eu|>": 50310,
77
+ "<|fa|>": 50300,
78
+ "<|fi|>": 50277,
79
+ "<|fo|>": 50338,
80
+ "<|fr|>": 50265,
81
+ "<|gl|>": 50319,
82
+ "<|gu|>": 50333,
83
+ "<|haw|>": 50352,
84
+ "<|ha|>": 50354,
85
+ "<|he|>": 50279,
86
+ "<|hi|>": 50276,
87
+ "<|hr|>": 50291,
88
+ "<|ht|>": 50339,
89
+ "<|hu|>": 50286,
90
+ "<|hy|>": 50312,
91
+ "<|id|>": 50275,
92
+ "<|is|>": 50311,
93
+ "<|it|>": 50274,
94
+ "<|ja|>": 50266,
95
+ "<|jw|>": 50356,
96
+ "<|ka|>": 50329,
97
+ "<|kk|>": 50316,
98
+ "<|km|>": 50323,
99
+ "<|kn|>": 50306,
100
+ "<|ko|>": 50264,
101
+ "<|la|>": 50294,
102
+ "<|lb|>": 50345,
103
+ "<|ln|>": 50353,
104
+ "<|lo|>": 50336,
105
+ "<|lt|>": 50293,
106
+ "<|lv|>": 50301,
107
+ "<|mg|>": 50349,
108
+ "<|mi|>": 50295,
109
+ "<|mk|>": 50308,
110
+ "<|ml|>": 50296,
111
+ "<|mn|>": 50314,
112
+ "<|mr|>": 50320,
113
+ "<|ms|>": 50282,
114
+ "<|mt|>": 50343,
115
+ "<|my|>": 50346,
116
+ "<|ne|>": 50313,
117
+ "<|nl|>": 50271,
118
+ "<|nn|>": 50342,
119
+ "<|no|>": 50288,
120
+ "<|oc|>": 50328,
121
+ "<|pa|>": 50321,
122
+ "<|pl|>": 50269,
123
+ "<|ps|>": 50340,
124
+ "<|pt|>": 50267,
125
+ "<|ro|>": 50284,
126
+ "<|ru|>": 50263,
127
+ "<|sa|>": 50344,
128
+ "<|sd|>": 50332,
129
+ "<|si|>": 50322,
130
+ "<|sk|>": 50298,
131
+ "<|sl|>": 50305,
132
+ "<|sn|>": 50324,
133
+ "<|so|>": 50326,
134
+ "<|sq|>": 50317,
135
+ "<|sr|>": 50303,
136
+ "<|su|>": 50357,
137
+ "<|sv|>": 50273,
138
+ "<|sw|>": 50318,
139
+ "<|ta|>": 50287,
140
+ "<|te|>": 50299,
141
+ "<|tg|>": 50331,
142
+ "<|th|>": 50289,
143
+ "<|tk|>": 50341,
144
+ "<|tl|>": 50348,
145
+ "<|tr|>": 50268,
146
+ "<|tt|>": 50351,
147
+ "<|uk|>": 50280,
148
+ "<|ur|>": 50290,
149
+ "<|uz|>": 50337,
150
+ "<|vi|>": 50278,
151
+ "<|yi|>": 50335,
152
+ "<|yo|>": 50325,
153
+ "<|zh|>": 50260
154
+ },
155
+ "max_initial_timestamp_index": 50,
156
+ "max_length": 448,
157
+ "no_timestamps_token_id": 50363,
158
+ "pad_token_id": 50257,
159
+ "prev_sot_token_id": 50361,
160
+ "return_timestamps": false,
161
+ "suppress_tokens": [],
162
+ "task_to_id": {
163
+ "transcribe": 50359,
164
+ "translate": 50358
165
+ },
166
+ "transformers_version": "4.45.1"
167
+ }