csikasote commited on
Commit
fae06b8
·
verified ·
1 Parent(s): b1a3f09

End of training

Browse files
README.md CHANGED
@@ -4,11 +4,23 @@ license: apache-2.0
4
  base_model: openai/whisper-medium
5
  tags:
6
  - generated_from_trainer
 
 
7
  metrics:
8
  - wer
9
  model-index:
10
  - name: whisper-medium-swagen-combined-15hrs-model
11
- results: []
 
 
 
 
 
 
 
 
 
 
12
  ---
13
 
14
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,10 +28,10 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  # whisper-medium-swagen-combined-15hrs-model
18
 
19
- This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.4299
22
- - Wer: 0.2526
23
 
24
  ## Model description
25
 
 
4
  base_model: openai/whisper-medium
5
  tags:
6
  - generated_from_trainer
7
+ datasets:
8
+ - swagen
9
  metrics:
10
  - wer
11
  model-index:
12
  - name: whisper-medium-swagen-combined-15hrs-model
13
+ results:
14
+ - task:
15
+ name: Automatic Speech Recognition
16
+ type: automatic-speech-recognition
17
+ dataset:
18
+ name: swagen
19
+ type: swagen
20
+ metrics:
21
+ - name: Wer
22
+ type: wer
23
+ value: 0.27171266233766234
24
  ---
25
 
26
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
28
 
29
  # whisper-medium-swagen-combined-15hrs-model
30
 
31
+ This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on the swagen dataset.
32
  It achieves the following results on the evaluation set:
33
+ - Loss: 0.4103
34
+ - Wer: 0.2717
35
 
36
  ## Model description
37
 
all_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.4796361381021295,
3
+ "eval_loss": 0.41031718254089355,
4
+ "eval_runtime": 563.4746,
5
+ "eval_samples": 1087,
6
+ "eval_samples_per_second": 1.929,
7
+ "eval_steps_per_second": 0.965,
8
+ "eval_wer": 0.27171266233766234,
9
+ "total_flos": 2.448227052158976e+19,
10
+ "train_loss": 1.2735130944252013,
11
+ "train_runtime": 14573.655,
12
+ "train_samples": 9674,
13
+ "train_samples_per_second": 19.914,
14
+ "train_steps_per_second": 2.489
15
+ }
eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.4796361381021295,
3
+ "eval_loss": 0.41031718254089355,
4
+ "eval_runtime": 563.4746,
5
+ "eval_samples": 1087,
6
+ "eval_samples_per_second": 1.929,
7
+ "eval_steps_per_second": 0.965,
8
+ "eval_wer": 0.27171266233766234
9
+ }
runs/Jan05_08-45-15_srvrocgpu011.uct.ac.za/events.out.tfevents.1736074925.srvrocgpu011.uct.ac.za ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2534fe8dfa19182f409e17efc7ad7d391429c159a90ac107e29a5a4c0df88bb4
3
+ size 40
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.4796361381021295,
3
+ "total_flos": 2.448227052158976e+19,
4
+ "train_loss": 1.2735130944252013,
5
+ "train_runtime": 14573.655,
6
+ "train_samples": 9674,
7
+ "train_samples_per_second": 19.914,
8
+ "train_steps_per_second": 2.489
9
+ }
trainer_state.json ADDED
@@ -0,0 +1,1026 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.41031718254089355,
3
+ "best_model_checkpoint": "/scratch/skscla001/speech/results/whisper-medium-swagen-combined-15hrs-model/checkpoint-2400",
4
+ "epoch": 2.4796361381021295,
5
+ "eval_steps": 200,
6
+ "global_step": 3000,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.02067397146991937,
13
+ "grad_norm": 101.30694580078125,
14
+ "learning_rate": 4.0000000000000003e-07,
15
+ "loss": 10.3278,
16
+ "step": 25
17
+ },
18
+ {
19
+ "epoch": 0.04134794293983874,
20
+ "grad_norm": 101.91423034667969,
21
+ "learning_rate": 9.000000000000001e-07,
22
+ "loss": 8.3674,
23
+ "step": 50
24
+ },
25
+ {
26
+ "epoch": 0.062021914409758114,
27
+ "grad_norm": 75.16177368164062,
28
+ "learning_rate": 1.4000000000000001e-06,
29
+ "loss": 5.6124,
30
+ "step": 75
31
+ },
32
+ {
33
+ "epoch": 0.08269588587967748,
34
+ "grad_norm": 90.76732635498047,
35
+ "learning_rate": 1.9000000000000002e-06,
36
+ "loss": 4.0776,
37
+ "step": 100
38
+ },
39
+ {
40
+ "epoch": 0.10336985734959686,
41
+ "grad_norm": 97.34427642822266,
42
+ "learning_rate": 2.4000000000000003e-06,
43
+ "loss": 3.6242,
44
+ "step": 125
45
+ },
46
+ {
47
+ "epoch": 0.12404382881951623,
48
+ "grad_norm": 64.82703399658203,
49
+ "learning_rate": 2.9e-06,
50
+ "loss": 2.9814,
51
+ "step": 150
52
+ },
53
+ {
54
+ "epoch": 0.1447178002894356,
55
+ "grad_norm": 71.11066436767578,
56
+ "learning_rate": 3.4000000000000005e-06,
57
+ "loss": 2.9502,
58
+ "step": 175
59
+ },
60
+ {
61
+ "epoch": 0.16539177175935496,
62
+ "grad_norm": 61.207664489746094,
63
+ "learning_rate": 3.900000000000001e-06,
64
+ "loss": 2.6268,
65
+ "step": 200
66
+ },
67
+ {
68
+ "epoch": 0.16539177175935496,
69
+ "eval_loss": 0.8031158447265625,
70
+ "eval_runtime": 560.9375,
71
+ "eval_samples_per_second": 1.938,
72
+ "eval_steps_per_second": 0.97,
73
+ "eval_wer": 0.46053165584415584,
74
+ "step": 200
75
+ },
76
+ {
77
+ "epoch": 0.18606574322927436,
78
+ "grad_norm": 65.30712890625,
79
+ "learning_rate": 4.4e-06,
80
+ "loss": 2.6405,
81
+ "step": 225
82
+ },
83
+ {
84
+ "epoch": 0.20673971469919372,
85
+ "grad_norm": 66.03694915771484,
86
+ "learning_rate": 4.9000000000000005e-06,
87
+ "loss": 2.2504,
88
+ "step": 250
89
+ },
90
+ {
91
+ "epoch": 0.2274136861691131,
92
+ "grad_norm": 53.01826095581055,
93
+ "learning_rate": 5.400000000000001e-06,
94
+ "loss": 2.3635,
95
+ "step": 275
96
+ },
97
+ {
98
+ "epoch": 0.24808765763903246,
99
+ "grad_norm": 63.38921356201172,
100
+ "learning_rate": 5.9e-06,
101
+ "loss": 2.4033,
102
+ "step": 300
103
+ },
104
+ {
105
+ "epoch": 0.26876162910895185,
106
+ "grad_norm": 57.252105712890625,
107
+ "learning_rate": 6.4000000000000006e-06,
108
+ "loss": 2.2953,
109
+ "step": 325
110
+ },
111
+ {
112
+ "epoch": 0.2894356005788712,
113
+ "grad_norm": 72.04216766357422,
114
+ "learning_rate": 6.9e-06,
115
+ "loss": 2.1778,
116
+ "step": 350
117
+ },
118
+ {
119
+ "epoch": 0.3101095720487906,
120
+ "grad_norm": 67.5660629272461,
121
+ "learning_rate": 7.4e-06,
122
+ "loss": 2.1933,
123
+ "step": 375
124
+ },
125
+ {
126
+ "epoch": 0.3307835435187099,
127
+ "grad_norm": 58.30109405517578,
128
+ "learning_rate": 7.9e-06,
129
+ "loss": 2.0712,
130
+ "step": 400
131
+ },
132
+ {
133
+ "epoch": 0.3307835435187099,
134
+ "eval_loss": 0.614786684513092,
135
+ "eval_runtime": 564.8568,
136
+ "eval_samples_per_second": 1.924,
137
+ "eval_steps_per_second": 0.963,
138
+ "eval_wer": 0.382913961038961,
139
+ "step": 400
140
+ },
141
+ {
142
+ "epoch": 0.3514575149886293,
143
+ "grad_norm": 54.52381134033203,
144
+ "learning_rate": 8.400000000000001e-06,
145
+ "loss": 1.9964,
146
+ "step": 425
147
+ },
148
+ {
149
+ "epoch": 0.3721314864585487,
150
+ "grad_norm": 45.91225814819336,
151
+ "learning_rate": 8.900000000000001e-06,
152
+ "loss": 2.0653,
153
+ "step": 450
154
+ },
155
+ {
156
+ "epoch": 0.39280545792846805,
157
+ "grad_norm": 46.43616485595703,
158
+ "learning_rate": 9.4e-06,
159
+ "loss": 1.8897,
160
+ "step": 475
161
+ },
162
+ {
163
+ "epoch": 0.41347942939838744,
164
+ "grad_norm": 49.05242919921875,
165
+ "learning_rate": 9.9e-06,
166
+ "loss": 1.7375,
167
+ "step": 500
168
+ },
169
+ {
170
+ "epoch": 0.4341534008683068,
171
+ "grad_norm": 35.908485412597656,
172
+ "learning_rate": 9.994408722393068e-06,
173
+ "loss": 1.6212,
174
+ "step": 525
175
+ },
176
+ {
177
+ "epoch": 0.4548273723382262,
178
+ "grad_norm": 60.85798263549805,
179
+ "learning_rate": 9.987419625384402e-06,
180
+ "loss": 1.8667,
181
+ "step": 550
182
+ },
183
+ {
184
+ "epoch": 0.47550134380814557,
185
+ "grad_norm": 43.83357238769531,
186
+ "learning_rate": 9.980430528375734e-06,
187
+ "loss": 1.7175,
188
+ "step": 575
189
+ },
190
+ {
191
+ "epoch": 0.4961753152780649,
192
+ "grad_norm": 46.343505859375,
193
+ "learning_rate": 9.973441431367069e-06,
194
+ "loss": 1.7302,
195
+ "step": 600
196
+ },
197
+ {
198
+ "epoch": 0.4961753152780649,
199
+ "eval_loss": 0.5561510920524597,
200
+ "eval_runtime": 570.0484,
201
+ "eval_samples_per_second": 1.907,
202
+ "eval_steps_per_second": 0.954,
203
+ "eval_wer": 0.349025974025974,
204
+ "step": 600
205
+ },
206
+ {
207
+ "epoch": 0.5168492867479842,
208
+ "grad_norm": 70.89414978027344,
209
+ "learning_rate": 9.966452334358401e-06,
210
+ "loss": 1.6704,
211
+ "step": 625
212
+ },
213
+ {
214
+ "epoch": 0.5375232582179037,
215
+ "grad_norm": 41.292442321777344,
216
+ "learning_rate": 9.959463237349735e-06,
217
+ "loss": 1.6499,
218
+ "step": 650
219
+ },
220
+ {
221
+ "epoch": 0.558197229687823,
222
+ "grad_norm": 58.775917053222656,
223
+ "learning_rate": 9.952474140341068e-06,
224
+ "loss": 1.8108,
225
+ "step": 675
226
+ },
227
+ {
228
+ "epoch": 0.5788712011577424,
229
+ "grad_norm": 38.21028518676758,
230
+ "learning_rate": 9.945485043332402e-06,
231
+ "loss": 1.6487,
232
+ "step": 700
233
+ },
234
+ {
235
+ "epoch": 0.5995451726276618,
236
+ "grad_norm": 52.19435119628906,
237
+ "learning_rate": 9.938495946323736e-06,
238
+ "loss": 1.6182,
239
+ "step": 725
240
+ },
241
+ {
242
+ "epoch": 0.6202191440975812,
243
+ "grad_norm": 53.34971618652344,
244
+ "learning_rate": 9.931506849315069e-06,
245
+ "loss": 1.5108,
246
+ "step": 750
247
+ },
248
+ {
249
+ "epoch": 0.6408931155675005,
250
+ "grad_norm": 41.846012115478516,
251
+ "learning_rate": 9.924517752306403e-06,
252
+ "loss": 1.5143,
253
+ "step": 775
254
+ },
255
+ {
256
+ "epoch": 0.6615670870374198,
257
+ "grad_norm": 44.0091438293457,
258
+ "learning_rate": 9.917528655297737e-06,
259
+ "loss": 1.5735,
260
+ "step": 800
261
+ },
262
+ {
263
+ "epoch": 0.6615670870374198,
264
+ "eval_loss": 0.5102764964103699,
265
+ "eval_runtime": 561.2293,
266
+ "eval_samples_per_second": 1.937,
267
+ "eval_steps_per_second": 0.969,
268
+ "eval_wer": 0.3105722402597403,
269
+ "step": 800
270
+ },
271
+ {
272
+ "epoch": 0.6822410585073393,
273
+ "grad_norm": 48.56166458129883,
274
+ "learning_rate": 9.91053955828907e-06,
275
+ "loss": 1.5532,
276
+ "step": 825
277
+ },
278
+ {
279
+ "epoch": 0.7029150299772586,
280
+ "grad_norm": 40.50489044189453,
281
+ "learning_rate": 9.903550461280404e-06,
282
+ "loss": 1.5039,
283
+ "step": 850
284
+ },
285
+ {
286
+ "epoch": 0.723589001447178,
287
+ "grad_norm": 62.77304458618164,
288
+ "learning_rate": 9.896561364271738e-06,
289
+ "loss": 1.4893,
290
+ "step": 875
291
+ },
292
+ {
293
+ "epoch": 0.7442629729170974,
294
+ "grad_norm": 52.9476432800293,
295
+ "learning_rate": 9.88957226726307e-06,
296
+ "loss": 1.4988,
297
+ "step": 900
298
+ },
299
+ {
300
+ "epoch": 0.7649369443870168,
301
+ "grad_norm": 40.155426025390625,
302
+ "learning_rate": 9.882583170254404e-06,
303
+ "loss": 1.4247,
304
+ "step": 925
305
+ },
306
+ {
307
+ "epoch": 0.7856109158569361,
308
+ "grad_norm": 40.80839538574219,
309
+ "learning_rate": 9.875594073245739e-06,
310
+ "loss": 1.4519,
311
+ "step": 950
312
+ },
313
+ {
314
+ "epoch": 0.8062848873268554,
315
+ "grad_norm": 42.46361541748047,
316
+ "learning_rate": 9.868604976237071e-06,
317
+ "loss": 1.4005,
318
+ "step": 975
319
+ },
320
+ {
321
+ "epoch": 0.8269588587967749,
322
+ "grad_norm": 30.87221336364746,
323
+ "learning_rate": 9.861615879228405e-06,
324
+ "loss": 1.5623,
325
+ "step": 1000
326
+ },
327
+ {
328
+ "epoch": 0.8269588587967749,
329
+ "eval_loss": 0.4683079719543457,
330
+ "eval_runtime": 551.982,
331
+ "eval_samples_per_second": 1.969,
332
+ "eval_steps_per_second": 0.986,
333
+ "eval_wer": 0.2775974025974026,
334
+ "step": 1000
335
+ },
336
+ {
337
+ "epoch": 0.8476328302666942,
338
+ "grad_norm": 30.2112979888916,
339
+ "learning_rate": 9.854626782219738e-06,
340
+ "loss": 1.5792,
341
+ "step": 1025
342
+ },
343
+ {
344
+ "epoch": 0.8683068017366136,
345
+ "grad_norm": 35.51711654663086,
346
+ "learning_rate": 9.847637685211072e-06,
347
+ "loss": 1.3541,
348
+ "step": 1050
349
+ },
350
+ {
351
+ "epoch": 0.888980773206533,
352
+ "grad_norm": 35.0782356262207,
353
+ "learning_rate": 9.840648588202404e-06,
354
+ "loss": 1.4297,
355
+ "step": 1075
356
+ },
357
+ {
358
+ "epoch": 0.9096547446764524,
359
+ "grad_norm": 36.1882438659668,
360
+ "learning_rate": 9.833659491193739e-06,
361
+ "loss": 1.5218,
362
+ "step": 1100
363
+ },
364
+ {
365
+ "epoch": 0.9303287161463717,
366
+ "grad_norm": 48.197357177734375,
367
+ "learning_rate": 9.826670394185071e-06,
368
+ "loss": 1.3869,
369
+ "step": 1125
370
+ },
371
+ {
372
+ "epoch": 0.9510026876162911,
373
+ "grad_norm": 30.1329288482666,
374
+ "learning_rate": 9.819681297176405e-06,
375
+ "loss": 1.4091,
376
+ "step": 1150
377
+ },
378
+ {
379
+ "epoch": 0.9716766590862105,
380
+ "grad_norm": 36.65972900390625,
381
+ "learning_rate": 9.81269220016774e-06,
382
+ "loss": 1.208,
383
+ "step": 1175
384
+ },
385
+ {
386
+ "epoch": 0.9923506305561298,
387
+ "grad_norm": 41.54533386230469,
388
+ "learning_rate": 9.805703103159072e-06,
389
+ "loss": 1.2713,
390
+ "step": 1200
391
+ },
392
+ {
393
+ "epoch": 0.9923506305561298,
394
+ "eval_loss": 0.44389376044273376,
395
+ "eval_runtime": 555.064,
396
+ "eval_samples_per_second": 1.958,
397
+ "eval_steps_per_second": 0.98,
398
+ "eval_wer": 0.2687702922077922,
399
+ "step": 1200
400
+ },
401
+ {
402
+ "epoch": 1.0124043828819516,
403
+ "grad_norm": 25.689674377441406,
404
+ "learning_rate": 9.798714006150406e-06,
405
+ "loss": 0.8989,
406
+ "step": 1225
407
+ },
408
+ {
409
+ "epoch": 1.033078354351871,
410
+ "grad_norm": 21.258874893188477,
411
+ "learning_rate": 9.79172490914174e-06,
412
+ "loss": 0.717,
413
+ "step": 1250
414
+ },
415
+ {
416
+ "epoch": 1.0537523258217905,
417
+ "grad_norm": 26.262178421020508,
418
+ "learning_rate": 9.784735812133073e-06,
419
+ "loss": 0.6577,
420
+ "step": 1275
421
+ },
422
+ {
423
+ "epoch": 1.0744262972917098,
424
+ "grad_norm": 38.216026306152344,
425
+ "learning_rate": 9.777746715124407e-06,
426
+ "loss": 0.7783,
427
+ "step": 1300
428
+ },
429
+ {
430
+ "epoch": 1.0951002687616291,
431
+ "grad_norm": 22.054594039916992,
432
+ "learning_rate": 9.770757618115741e-06,
433
+ "loss": 0.8445,
434
+ "step": 1325
435
+ },
436
+ {
437
+ "epoch": 1.1157742402315485,
438
+ "grad_norm": 28.307518005371094,
439
+ "learning_rate": 9.763768521107074e-06,
440
+ "loss": 0.7671,
441
+ "step": 1350
442
+ },
443
+ {
444
+ "epoch": 1.1364482117014678,
445
+ "grad_norm": 32.70164489746094,
446
+ "learning_rate": 9.756779424098408e-06,
447
+ "loss": 0.7347,
448
+ "step": 1375
449
+ },
450
+ {
451
+ "epoch": 1.1571221831713872,
452
+ "grad_norm": 27.47947883605957,
453
+ "learning_rate": 9.74979032708974e-06,
454
+ "loss": 0.7209,
455
+ "step": 1400
456
+ },
457
+ {
458
+ "epoch": 1.1571221831713872,
459
+ "eval_loss": 0.4601219892501831,
460
+ "eval_runtime": 556.2377,
461
+ "eval_samples_per_second": 1.954,
462
+ "eval_steps_per_second": 0.978,
463
+ "eval_wer": 0.27323457792207795,
464
+ "step": 1400
465
+ },
466
+ {
467
+ "epoch": 1.1777961546413067,
468
+ "grad_norm": 35.48779296875,
469
+ "learning_rate": 9.742801230081074e-06,
470
+ "loss": 0.8649,
471
+ "step": 1425
472
+ },
473
+ {
474
+ "epoch": 1.198470126111226,
475
+ "grad_norm": 39.03016662597656,
476
+ "learning_rate": 9.735812133072407e-06,
477
+ "loss": 0.7049,
478
+ "step": 1450
479
+ },
480
+ {
481
+ "epoch": 1.2191440975811454,
482
+ "grad_norm": 32.857749938964844,
483
+ "learning_rate": 9.728823036063741e-06,
484
+ "loss": 0.8108,
485
+ "step": 1475
486
+ },
487
+ {
488
+ "epoch": 1.2398180690510647,
489
+ "grad_norm": 33.16276550292969,
490
+ "learning_rate": 9.721833939055075e-06,
491
+ "loss": 0.7099,
492
+ "step": 1500
493
+ },
494
+ {
495
+ "epoch": 1.260492040520984,
496
+ "grad_norm": 33.7196159362793,
497
+ "learning_rate": 9.714844842046408e-06,
498
+ "loss": 0.7708,
499
+ "step": 1525
500
+ },
501
+ {
502
+ "epoch": 1.2811660119909034,
503
+ "grad_norm": 25.693979263305664,
504
+ "learning_rate": 9.707855745037742e-06,
505
+ "loss": 0.8099,
506
+ "step": 1550
507
+ },
508
+ {
509
+ "epoch": 1.3018399834608227,
510
+ "grad_norm": 19.993423461914062,
511
+ "learning_rate": 9.700866648029076e-06,
512
+ "loss": 0.7314,
513
+ "step": 1575
514
+ },
515
+ {
516
+ "epoch": 1.3225139549307423,
517
+ "grad_norm": 38.857879638671875,
518
+ "learning_rate": 9.693877551020408e-06,
519
+ "loss": 0.6856,
520
+ "step": 1600
521
+ },
522
+ {
523
+ "epoch": 1.3225139549307423,
524
+ "eval_loss": 0.43911561369895935,
525
+ "eval_runtime": 558.8109,
526
+ "eval_samples_per_second": 1.945,
527
+ "eval_steps_per_second": 0.973,
528
+ "eval_wer": 0.25953733766233766,
529
+ "step": 1600
530
+ },
531
+ {
532
+ "epoch": 1.3431879264006616,
533
+ "grad_norm": 34.80131530761719,
534
+ "learning_rate": 9.686888454011743e-06,
535
+ "loss": 0.7398,
536
+ "step": 1625
537
+ },
538
+ {
539
+ "epoch": 1.363861897870581,
540
+ "grad_norm": 30.94778823852539,
541
+ "learning_rate": 9.679899357003077e-06,
542
+ "loss": 0.741,
543
+ "step": 1650
544
+ },
545
+ {
546
+ "epoch": 1.3845358693405003,
547
+ "grad_norm": 39.64469909667969,
548
+ "learning_rate": 9.67291025999441e-06,
549
+ "loss": 0.7306,
550
+ "step": 1675
551
+ },
552
+ {
553
+ "epoch": 1.4052098408104197,
554
+ "grad_norm": 35.8865966796875,
555
+ "learning_rate": 9.665921162985743e-06,
556
+ "loss": 0.6787,
557
+ "step": 1700
558
+ },
559
+ {
560
+ "epoch": 1.425883812280339,
561
+ "grad_norm": 29.526947021484375,
562
+ "learning_rate": 9.658932065977078e-06,
563
+ "loss": 0.6757,
564
+ "step": 1725
565
+ },
566
+ {
567
+ "epoch": 1.4465577837502583,
568
+ "grad_norm": 35.52757263183594,
569
+ "learning_rate": 9.65194296896841e-06,
570
+ "loss": 0.7509,
571
+ "step": 1750
572
+ },
573
+ {
574
+ "epoch": 1.467231755220178,
575
+ "grad_norm": 15.845513343811035,
576
+ "learning_rate": 9.644953871959744e-06,
577
+ "loss": 0.7827,
578
+ "step": 1775
579
+ },
580
+ {
581
+ "epoch": 1.4879057266900972,
582
+ "grad_norm": 33.226444244384766,
583
+ "learning_rate": 9.637964774951077e-06,
584
+ "loss": 0.7661,
585
+ "step": 1800
586
+ },
587
+ {
588
+ "epoch": 1.4879057266900972,
589
+ "eval_loss": 0.43962281942367554,
590
+ "eval_runtime": 565.8494,
591
+ "eval_samples_per_second": 1.921,
592
+ "eval_steps_per_second": 0.961,
593
+ "eval_wer": 0.2754667207792208,
594
+ "step": 1800
595
+ },
596
+ {
597
+ "epoch": 1.5085796981600166,
598
+ "grad_norm": 41.39881896972656,
599
+ "learning_rate": 9.630975677942411e-06,
600
+ "loss": 0.6939,
601
+ "step": 1825
602
+ },
603
+ {
604
+ "epoch": 1.529253669629936,
605
+ "grad_norm": 22.1778564453125,
606
+ "learning_rate": 9.623986580933743e-06,
607
+ "loss": 0.7105,
608
+ "step": 1850
609
+ },
610
+ {
611
+ "epoch": 1.5499276410998553,
612
+ "grad_norm": 26.528127670288086,
613
+ "learning_rate": 9.616997483925078e-06,
614
+ "loss": 0.693,
615
+ "step": 1875
616
+ },
617
+ {
618
+ "epoch": 1.5706016125697748,
619
+ "grad_norm": 35.542049407958984,
620
+ "learning_rate": 9.61000838691641e-06,
621
+ "loss": 0.7716,
622
+ "step": 1900
623
+ },
624
+ {
625
+ "epoch": 1.591275584039694,
626
+ "grad_norm": 22.44879913330078,
627
+ "learning_rate": 9.603019289907744e-06,
628
+ "loss": 0.6435,
629
+ "step": 1925
630
+ },
631
+ {
632
+ "epoch": 1.6119495555096135,
633
+ "grad_norm": 28.783998489379883,
634
+ "learning_rate": 9.596030192899078e-06,
635
+ "loss": 0.6641,
636
+ "step": 1950
637
+ },
638
+ {
639
+ "epoch": 1.6326235269795326,
640
+ "grad_norm": 19.634614944458008,
641
+ "learning_rate": 9.589041095890411e-06,
642
+ "loss": 0.8506,
643
+ "step": 1975
644
+ },
645
+ {
646
+ "epoch": 1.6532974984494522,
647
+ "grad_norm": 39.75117492675781,
648
+ "learning_rate": 9.582051998881745e-06,
649
+ "loss": 0.8113,
650
+ "step": 2000
651
+ },
652
+ {
653
+ "epoch": 1.6532974984494522,
654
+ "eval_loss": 0.42616990208625793,
655
+ "eval_runtime": 564.8279,
656
+ "eval_samples_per_second": 1.924,
657
+ "eval_steps_per_second": 0.963,
658
+ "eval_wer": 0.2643060064935065,
659
+ "step": 2000
660
+ },
661
+ {
662
+ "epoch": 1.6739714699193715,
663
+ "grad_norm": 38.3372917175293,
664
+ "learning_rate": 9.57506290187308e-06,
665
+ "loss": 0.7083,
666
+ "step": 2025
667
+ },
668
+ {
669
+ "epoch": 1.6946454413892909,
670
+ "grad_norm": 31.000165939331055,
671
+ "learning_rate": 9.568073804864413e-06,
672
+ "loss": 0.8346,
673
+ "step": 2050
674
+ },
675
+ {
676
+ "epoch": 1.7153194128592104,
677
+ "grad_norm": 29.758399963378906,
678
+ "learning_rate": 9.561084707855746e-06,
679
+ "loss": 0.7429,
680
+ "step": 2075
681
+ },
682
+ {
683
+ "epoch": 1.7359933843291295,
684
+ "grad_norm": 30.983566284179688,
685
+ "learning_rate": 9.55409561084708e-06,
686
+ "loss": 0.6273,
687
+ "step": 2100
688
+ },
689
+ {
690
+ "epoch": 1.756667355799049,
691
+ "grad_norm": 44.77085876464844,
692
+ "learning_rate": 9.547106513838413e-06,
693
+ "loss": 0.7398,
694
+ "step": 2125
695
+ },
696
+ {
697
+ "epoch": 1.7773413272689682,
698
+ "grad_norm": 19.01654815673828,
699
+ "learning_rate": 9.540117416829747e-06,
700
+ "loss": 0.7235,
701
+ "step": 2150
702
+ },
703
+ {
704
+ "epoch": 1.7980152987388878,
705
+ "grad_norm": 24.96238136291504,
706
+ "learning_rate": 9.53312831982108e-06,
707
+ "loss": 0.7483,
708
+ "step": 2175
709
+ },
710
+ {
711
+ "epoch": 1.818689270208807,
712
+ "grad_norm": 31.377376556396484,
713
+ "learning_rate": 9.526139222812413e-06,
714
+ "loss": 0.77,
715
+ "step": 2200
716
+ },
717
+ {
718
+ "epoch": 1.818689270208807,
719
+ "eval_loss": 0.4175179898738861,
720
+ "eval_runtime": 569.934,
721
+ "eval_samples_per_second": 1.907,
722
+ "eval_steps_per_second": 0.954,
723
+ "eval_wer": 0.26785714285714285,
724
+ "step": 2200
725
+ },
726
+ {
727
+ "epoch": 1.8393632416787264,
728
+ "grad_norm": 29.9045352935791,
729
+ "learning_rate": 9.519150125803746e-06,
730
+ "loss": 0.7315,
731
+ "step": 2225
732
+ },
733
+ {
734
+ "epoch": 1.860037213148646,
735
+ "grad_norm": 49.298458099365234,
736
+ "learning_rate": 9.51216102879508e-06,
737
+ "loss": 0.6964,
738
+ "step": 2250
739
+ },
740
+ {
741
+ "epoch": 1.8807111846185651,
742
+ "grad_norm": 25.912410736083984,
743
+ "learning_rate": 9.505171931786414e-06,
744
+ "loss": 0.6415,
745
+ "step": 2275
746
+ },
747
+ {
748
+ "epoch": 1.9013851560884847,
749
+ "grad_norm": 20.445037841796875,
750
+ "learning_rate": 9.498182834777747e-06,
751
+ "loss": 0.5379,
752
+ "step": 2300
753
+ },
754
+ {
755
+ "epoch": 1.922059127558404,
756
+ "grad_norm": 26.04856300354004,
757
+ "learning_rate": 9.49119373776908e-06,
758
+ "loss": 0.8237,
759
+ "step": 2325
760
+ },
761
+ {
762
+ "epoch": 1.9427330990283234,
763
+ "grad_norm": 31.83847999572754,
764
+ "learning_rate": 9.484204640760415e-06,
765
+ "loss": 0.7311,
766
+ "step": 2350
767
+ },
768
+ {
769
+ "epoch": 1.9634070704982427,
770
+ "grad_norm": 20.37737464904785,
771
+ "learning_rate": 9.477215543751747e-06,
772
+ "loss": 0.8047,
773
+ "step": 2375
774
+ },
775
+ {
776
+ "epoch": 1.984081041968162,
777
+ "grad_norm": 25.79982566833496,
778
+ "learning_rate": 9.470226446743082e-06,
779
+ "loss": 0.6942,
780
+ "step": 2400
781
+ },
782
+ {
783
+ "epoch": 1.984081041968162,
784
+ "eval_loss": 0.41031718254089355,
785
+ "eval_runtime": 570.9309,
786
+ "eval_samples_per_second": 1.904,
787
+ "eval_steps_per_second": 0.953,
788
+ "eval_wer": 0.27171266233766234,
789
+ "step": 2400
790
+ },
791
+ {
792
+ "epoch": 2.004134794293984,
793
+ "grad_norm": 5.95644474029541,
794
+ "learning_rate": 9.463237349734416e-06,
795
+ "loss": 0.5701,
796
+ "step": 2425
797
+ },
798
+ {
799
+ "epoch": 2.024808765763903,
800
+ "grad_norm": 21.82009506225586,
801
+ "learning_rate": 9.456248252725748e-06,
802
+ "loss": 0.3277,
803
+ "step": 2450
804
+ },
805
+ {
806
+ "epoch": 2.0454827372338227,
807
+ "grad_norm": 16.10843849182129,
808
+ "learning_rate": 9.449259155717082e-06,
809
+ "loss": 0.2838,
810
+ "step": 2475
811
+ },
812
+ {
813
+ "epoch": 2.066156708703742,
814
+ "grad_norm": 12.50637435913086,
815
+ "learning_rate": 9.442270058708417e-06,
816
+ "loss": 0.2812,
817
+ "step": 2500
818
+ },
819
+ {
820
+ "epoch": 2.0868306801736614,
821
+ "grad_norm": 21.421022415161133,
822
+ "learning_rate": 9.435280961699749e-06,
823
+ "loss": 0.3007,
824
+ "step": 2525
825
+ },
826
+ {
827
+ "epoch": 2.107504651643581,
828
+ "grad_norm": 28.00385284423828,
829
+ "learning_rate": 9.428291864691083e-06,
830
+ "loss": 0.2837,
831
+ "step": 2550
832
+ },
833
+ {
834
+ "epoch": 2.1281786231135,
835
+ "grad_norm": 17.902904510498047,
836
+ "learning_rate": 9.421302767682416e-06,
837
+ "loss": 0.2927,
838
+ "step": 2575
839
+ },
840
+ {
841
+ "epoch": 2.1488525945834196,
842
+ "grad_norm": 19.412660598754883,
843
+ "learning_rate": 9.41431367067375e-06,
844
+ "loss": 0.2814,
845
+ "step": 2600
846
+ },
847
+ {
848
+ "epoch": 2.1488525945834196,
849
+ "eval_loss": 0.4295479953289032,
850
+ "eval_runtime": 572.2047,
851
+ "eval_samples_per_second": 1.9,
852
+ "eval_steps_per_second": 0.951,
853
+ "eval_wer": 0.2616680194805195,
854
+ "step": 2600
855
+ },
856
+ {
857
+ "epoch": 2.1695265660533387,
858
+ "grad_norm": 21.472017288208008,
859
+ "learning_rate": 9.407324573665082e-06,
860
+ "loss": 0.3163,
861
+ "step": 2625
862
+ },
863
+ {
864
+ "epoch": 2.1902005375232583,
865
+ "grad_norm": 25.40437889099121,
866
+ "learning_rate": 9.400335476656417e-06,
867
+ "loss": 0.3067,
868
+ "step": 2650
869
+ },
870
+ {
871
+ "epoch": 2.2108745089931774,
872
+ "grad_norm": 19.886768341064453,
873
+ "learning_rate": 9.393346379647749e-06,
874
+ "loss": 0.2705,
875
+ "step": 2675
876
+ },
877
+ {
878
+ "epoch": 2.231548480463097,
879
+ "grad_norm": 18.539291381835938,
880
+ "learning_rate": 9.386357282639083e-06,
881
+ "loss": 0.2948,
882
+ "step": 2700
883
+ },
884
+ {
885
+ "epoch": 2.2522224519330165,
886
+ "grad_norm": 10.278417587280273,
887
+ "learning_rate": 9.379368185630417e-06,
888
+ "loss": 0.2911,
889
+ "step": 2725
890
+ },
891
+ {
892
+ "epoch": 2.2728964234029356,
893
+ "grad_norm": 25.13685417175293,
894
+ "learning_rate": 9.372379088621752e-06,
895
+ "loss": 0.3339,
896
+ "step": 2750
897
+ },
898
+ {
899
+ "epoch": 2.293570394872855,
900
+ "grad_norm": 28.98663902282715,
901
+ "learning_rate": 9.365389991613084e-06,
902
+ "loss": 0.3182,
903
+ "step": 2775
904
+ },
905
+ {
906
+ "epoch": 2.3142443663427743,
907
+ "grad_norm": 17.335172653198242,
908
+ "learning_rate": 9.358400894604418e-06,
909
+ "loss": 0.3171,
910
+ "step": 2800
911
+ },
912
+ {
913
+ "epoch": 2.3142443663427743,
914
+ "eval_loss": 0.4301378130912781,
915
+ "eval_runtime": 557.7802,
916
+ "eval_samples_per_second": 1.949,
917
+ "eval_steps_per_second": 0.975,
918
+ "eval_wer": 0.2432021103896104,
919
+ "step": 2800
920
+ },
921
+ {
922
+ "epoch": 2.334918337812694,
923
+ "grad_norm": 22.960308074951172,
924
+ "learning_rate": 9.351411797595752e-06,
925
+ "loss": 0.3493,
926
+ "step": 2825
927
+ },
928
+ {
929
+ "epoch": 2.3555923092826134,
930
+ "grad_norm": 21.65870475769043,
931
+ "learning_rate": 9.344422700587085e-06,
932
+ "loss": 0.3179,
933
+ "step": 2850
934
+ },
935
+ {
936
+ "epoch": 2.3762662807525325,
937
+ "grad_norm": 18.237104415893555,
938
+ "learning_rate": 9.337433603578419e-06,
939
+ "loss": 0.3325,
940
+ "step": 2875
941
+ },
942
+ {
943
+ "epoch": 2.396940252222452,
944
+ "grad_norm": 23.409774780273438,
945
+ "learning_rate": 9.330444506569753e-06,
946
+ "loss": 0.3483,
947
+ "step": 2900
948
+ },
949
+ {
950
+ "epoch": 2.4176142236923712,
951
+ "grad_norm": 16.235368728637695,
952
+ "learning_rate": 9.323455409561086e-06,
953
+ "loss": 0.3094,
954
+ "step": 2925
955
+ },
956
+ {
957
+ "epoch": 2.438288195162291,
958
+ "grad_norm": 15.572430610656738,
959
+ "learning_rate": 9.31646631255242e-06,
960
+ "loss": 0.3099,
961
+ "step": 2950
962
+ },
963
+ {
964
+ "epoch": 2.45896216663221,
965
+ "grad_norm": 16.10034942626953,
966
+ "learning_rate": 9.309477215543752e-06,
967
+ "loss": 0.3394,
968
+ "step": 2975
969
+ },
970
+ {
971
+ "epoch": 2.4796361381021295,
972
+ "grad_norm": 31.92362403869629,
973
+ "learning_rate": 9.302488118535085e-06,
974
+ "loss": 0.3495,
975
+ "step": 3000
976
+ },
977
+ {
978
+ "epoch": 2.4796361381021295,
979
+ "eval_loss": 0.42990124225616455,
980
+ "eval_runtime": 559.9267,
981
+ "eval_samples_per_second": 1.941,
982
+ "eval_steps_per_second": 0.972,
983
+ "eval_wer": 0.252637987012987,
984
+ "step": 3000
985
+ },
986
+ {
987
+ "epoch": 2.4796361381021295,
988
+ "step": 3000,
989
+ "total_flos": 2.448227052158976e+19,
990
+ "train_loss": 1.2735130944252013,
991
+ "train_runtime": 14573.655,
992
+ "train_samples_per_second": 19.914,
993
+ "train_steps_per_second": 2.489
994
+ }
995
+ ],
996
+ "logging_steps": 25,
997
+ "max_steps": 36270,
998
+ "num_input_tokens_seen": 0,
999
+ "num_train_epochs": 30,
1000
+ "save_steps": 200,
1001
+ "stateful_callbacks": {
1002
+ "EarlyStoppingCallback": {
1003
+ "args": {
1004
+ "early_stopping_patience": 3,
1005
+ "early_stopping_threshold": 0.0
1006
+ },
1007
+ "attributes": {
1008
+ "early_stopping_patience_counter": 3
1009
+ }
1010
+ },
1011
+ "TrainerControl": {
1012
+ "args": {
1013
+ "should_epoch_stop": false,
1014
+ "should_evaluate": false,
1015
+ "should_log": false,
1016
+ "should_save": true,
1017
+ "should_training_stop": true
1018
+ },
1019
+ "attributes": {}
1020
+ }
1021
+ },
1022
+ "total_flos": 2.448227052158976e+19,
1023
+ "train_batch_size": 2,
1024
+ "trial_name": null,
1025
+ "trial_params": null
1026
+ }