Makkoen commited on
Commit
1662b97
1 Parent(s): 0cd5fd2

Model save

Browse files
README.md ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: openai/whisper-medium.en
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - wer
8
+ model-index:
9
+ - name: whisper-medium.en-cit-do015-wd0-lr1e-06-1000
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ # whisper-medium.en-cit-do015-wd0-lr1e-06-1000
17
+
18
+ This model is a fine-tuned version of [openai/whisper-medium.en](https://huggingface.co/openai/whisper-medium.en) on the None dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.6953
21
+ - Wer Ortho: 26.2768
22
+ - Wer: 14.7572
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: 1e-06
42
+ - train_batch_size: 4
43
+ - eval_batch_size: 8
44
+ - seed: 42
45
+ - distributed_type: multi-GPU
46
+ - gradient_accumulation_steps: 4
47
+ - total_train_batch_size: 16
48
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
49
+ - lr_scheduler_type: linear
50
+ - lr_scheduler_warmup_steps: 100
51
+ - training_steps: 500
52
+ - mixed_precision_training: Native AMP
53
+
54
+ ### Training results
55
+
56
+ | Training Loss | Epoch | Step | Validation Loss | Wer Ortho | Wer |
57
+ |:-------------:|:------:|:----:|:---------------:|:---------:|:-------:|
58
+ | No log | 0.4444 | 25 | 1.5811 | 45.2632 | 31.9044 |
59
+ | 1.7463 | 0.8889 | 50 | 1.3848 | 39.1033 | 27.0106 |
60
+ | 1.7463 | 1.3333 | 75 | 1.2178 | 35.7505 | 23.0273 |
61
+ | 1.3387 | 1.7778 | 100 | 1.0166 | 36.1014 | 23.4446 |
62
+ | 1.3387 | 2.2222 | 125 | 0.8784 | 31.9298 | 19.1958 |
63
+ | 0.988 | 2.6667 | 150 | 0.8340 | 30.8382 | 18.4750 |
64
+ | 0.988 | 3.1111 | 175 | 0.8027 | 30.3314 | 17.7162 |
65
+ | 0.8856 | 3.5556 | 200 | 0.7812 | 29.6686 | 17.4127 |
66
+ | 0.8856 | 4.0 | 225 | 0.7651 | 30.1365 | 17.6783 |
67
+ | 0.7927 | 4.4444 | 250 | 0.7515 | 29.2008 | 16.8816 |
68
+ | 0.7927 | 4.8889 | 275 | 0.7402 | 28.2651 | 15.6677 |
69
+ | 0.7482 | 5.3333 | 300 | 0.7300 | 27.9922 | 15.5159 |
70
+ | 0.7482 | 5.7778 | 325 | 0.7217 | 27.8752 | 15.6677 |
71
+ | 0.7275 | 6.2222 | 350 | 0.7153 | 27.4854 | 15.4021 |
72
+ | 0.7275 | 6.6667 | 375 | 0.7085 | 27.3684 | 15.3642 |
73
+ | 0.7003 | 7.1111 | 400 | 0.7041 | 26.6277 | 14.6813 |
74
+ | 0.7003 | 7.5556 | 425 | 0.7002 | 26.3158 | 14.7572 |
75
+ | 0.6763 | 8.0 | 450 | 0.6973 | 26.2378 | 14.6055 |
76
+ | 0.6763 | 8.4444 | 475 | 0.6963 | 26.4327 | 14.7951 |
77
+ | 0.6687 | 8.8889 | 500 | 0.6953 | 26.2768 | 14.7572 |
78
+
79
+
80
+ ### Framework versions
81
+
82
+ - Transformers 4.42.3
83
+ - Pytorch 1.13.1+cu117
84
+ - Datasets 2.20.0
85
+ - Tokenizers 0.19.1
generation_config.json ADDED
@@ -0,0 +1,188 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alignment_heads": [
3
+ [
4
+ 11,
5
+ 4
6
+ ],
7
+ [
8
+ 14,
9
+ 1
10
+ ],
11
+ [
12
+ 14,
13
+ 12
14
+ ],
15
+ [
16
+ 14,
17
+ 14
18
+ ],
19
+ [
20
+ 15,
21
+ 4
22
+ ],
23
+ [
24
+ 16,
25
+ 0
26
+ ],
27
+ [
28
+ 16,
29
+ 4
30
+ ],
31
+ [
32
+ 16,
33
+ 9
34
+ ],
35
+ [
36
+ 17,
37
+ 12
38
+ ],
39
+ [
40
+ 17,
41
+ 14
42
+ ],
43
+ [
44
+ 18,
45
+ 7
46
+ ],
47
+ [
48
+ 18,
49
+ 10
50
+ ],
51
+ [
52
+ 18,
53
+ 15
54
+ ],
55
+ [
56
+ 20,
57
+ 0
58
+ ],
59
+ [
60
+ 20,
61
+ 3
62
+ ],
63
+ [
64
+ 20,
65
+ 9
66
+ ],
67
+ [
68
+ 20,
69
+ 14
70
+ ],
71
+ [
72
+ 21,
73
+ 12
74
+ ]
75
+ ],
76
+ "begin_suppress_tokens": [
77
+ 220,
78
+ 50256
79
+ ],
80
+ "bos_token_id": 50257,
81
+ "decoder_start_token_id": 50257,
82
+ "eos_token_id": 50256,
83
+ "is_multilingual": false,
84
+ "max_initial_timestamp_index": 50,
85
+ "max_length": 448,
86
+ "no_timestamps_token_id": 50362,
87
+ "pad_token_id": 50256,
88
+ "prev_sot_token_id": 50360,
89
+ "return_timestamps": false,
90
+ "suppress_tokens": [
91
+ 1,
92
+ 2,
93
+ 7,
94
+ 8,
95
+ 9,
96
+ 10,
97
+ 14,
98
+ 25,
99
+ 26,
100
+ 27,
101
+ 28,
102
+ 29,
103
+ 31,
104
+ 58,
105
+ 59,
106
+ 60,
107
+ 61,
108
+ 62,
109
+ 63,
110
+ 90,
111
+ 91,
112
+ 92,
113
+ 93,
114
+ 357,
115
+ 366,
116
+ 438,
117
+ 532,
118
+ 685,
119
+ 705,
120
+ 796,
121
+ 930,
122
+ 1058,
123
+ 1220,
124
+ 1267,
125
+ 1279,
126
+ 1303,
127
+ 1343,
128
+ 1377,
129
+ 1391,
130
+ 1635,
131
+ 1782,
132
+ 1875,
133
+ 2162,
134
+ 2361,
135
+ 2488,
136
+ 3467,
137
+ 4008,
138
+ 4211,
139
+ 4600,
140
+ 4808,
141
+ 5299,
142
+ 5855,
143
+ 6329,
144
+ 7203,
145
+ 9609,
146
+ 9959,
147
+ 10563,
148
+ 10786,
149
+ 11420,
150
+ 11709,
151
+ 11907,
152
+ 13163,
153
+ 13697,
154
+ 13700,
155
+ 14808,
156
+ 15306,
157
+ 16410,
158
+ 16791,
159
+ 17992,
160
+ 19203,
161
+ 19510,
162
+ 20724,
163
+ 22305,
164
+ 22935,
165
+ 27007,
166
+ 30109,
167
+ 30420,
168
+ 33409,
169
+ 34949,
170
+ 40283,
171
+ 40493,
172
+ 40549,
173
+ 47282,
174
+ 49146,
175
+ 50257,
176
+ 50357,
177
+ 50358,
178
+ 50359,
179
+ 50360,
180
+ 50361
181
+ ],
182
+ "task": "transcribe",
183
+ "task_to_id": {
184
+ "transcribe": 50359,
185
+ "translate": 50358
186
+ },
187
+ "transformers_version": "4.42.3"
188
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2782a43fb7f18db93df1ddb9cb239ece45285f77b32824240527094609e67c2f
3
  size 1634043280
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4ac9f52ebc724709d0eda485449a3f2b1303aa88ac63a5e1fafadaa6c9bc0b0
3
  size 1634043280
runs/Jul02_10-38-19_whisper-finetune2/events.out.tfevents.1719916708.whisper-finetune2 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c78c2a8edb04b6f414b0ad2d308b83977e07aba0773d2c9569387270d4352943
3
- size 16136
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e7d8efc14f8e226db226b8fd8b08b2eed992512d3f654fb6aeac87bd272ff37
3
+ size 16490