fsicoli commited on
Commit
69a743d
1 Parent(s): 336cd76

437916f31f88890597dfd8c3b75f1625274c8ecf732ff04740052eba3919ec22

Browse files
checkpoint-5000/model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0a3553cfea4d55a22ddc4554e797d5f41b8d4ec64eff37633af911cf87725f80
3
+ size 1180663192
checkpoint-5000/model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-5000/preprocessor_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "chunk_length": 30,
3
+ "feature_extractor_type": "WhisperFeatureExtractor",
4
+ "feature_size": 128,
5
+ "hop_length": 160,
6
+ "n_fft": 400,
7
+ "n_samples": 480000,
8
+ "nb_max_frames": 3000,
9
+ "padding_side": "right",
10
+ "padding_value": 0.0,
11
+ "processor_class": "WhisperProcessor",
12
+ "return_attention_mask": false,
13
+ "sampling_rate": 16000
14
+ }
checkpoint-5000/rng_state.pth ADDED
Binary file (14.3 kB). View file
 
checkpoint-5000/scheduler.pt ADDED
Binary file (1.06 kB). View file
 
checkpoint-5000/trainer_state.json ADDED
@@ -0,0 +1,1266 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 1.2903225806451613,
5
+ "eval_steps": 1000,
6
+ "global_step": 5000,
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.01,
13
+ "learning_rate": 1.1499999999999999e-08,
14
+ "loss": 0.7822,
15
+ "step": 25
16
+ },
17
+ {
18
+ "epoch": 0.01,
19
+ "learning_rate": 2.4e-08,
20
+ "loss": 1.3149,
21
+ "step": 50
22
+ },
23
+ {
24
+ "epoch": 0.02,
25
+ "learning_rate": 3.6499999999999996e-08,
26
+ "loss": 0.7809,
27
+ "step": 75
28
+ },
29
+ {
30
+ "epoch": 0.03,
31
+ "learning_rate": 4.9e-08,
32
+ "loss": 1.2916,
33
+ "step": 100
34
+ },
35
+ {
36
+ "epoch": 0.03,
37
+ "learning_rate": 6.15e-08,
38
+ "loss": 0.7645,
39
+ "step": 125
40
+ },
41
+ {
42
+ "epoch": 0.04,
43
+ "learning_rate": 7.399999999999999e-08,
44
+ "loss": 1.3379,
45
+ "step": 150
46
+ },
47
+ {
48
+ "epoch": 0.05,
49
+ "learning_rate": 8.649999999999999e-08,
50
+ "loss": 0.7722,
51
+ "step": 175
52
+ },
53
+ {
54
+ "epoch": 0.05,
55
+ "learning_rate": 9.9e-08,
56
+ "loss": 1.281,
57
+ "step": 200
58
+ },
59
+ {
60
+ "epoch": 0.06,
61
+ "learning_rate": 1.115e-07,
62
+ "loss": 0.6998,
63
+ "step": 225
64
+ },
65
+ {
66
+ "epoch": 0.06,
67
+ "learning_rate": 1.24e-07,
68
+ "loss": 1.1532,
69
+ "step": 250
70
+ },
71
+ {
72
+ "epoch": 0.07,
73
+ "learning_rate": 1.365e-07,
74
+ "loss": 0.583,
75
+ "step": 275
76
+ },
77
+ {
78
+ "epoch": 0.08,
79
+ "learning_rate": 1.49e-07,
80
+ "loss": 0.6693,
81
+ "step": 300
82
+ },
83
+ {
84
+ "epoch": 0.08,
85
+ "learning_rate": 1.615e-07,
86
+ "loss": 0.3395,
87
+ "step": 325
88
+ },
89
+ {
90
+ "epoch": 0.09,
91
+ "learning_rate": 1.7399999999999997e-07,
92
+ "loss": 0.3597,
93
+ "step": 350
94
+ },
95
+ {
96
+ "epoch": 0.1,
97
+ "learning_rate": 1.8649999999999998e-07,
98
+ "loss": 0.22,
99
+ "step": 375
100
+ },
101
+ {
102
+ "epoch": 0.1,
103
+ "learning_rate": 1.99e-07,
104
+ "loss": 0.3449,
105
+ "step": 400
106
+ },
107
+ {
108
+ "epoch": 0.11,
109
+ "learning_rate": 2.1149999999999998e-07,
110
+ "loss": 0.2123,
111
+ "step": 425
112
+ },
113
+ {
114
+ "epoch": 0.12,
115
+ "learning_rate": 2.24e-07,
116
+ "loss": 0.3362,
117
+ "step": 450
118
+ },
119
+ {
120
+ "epoch": 0.12,
121
+ "learning_rate": 2.3649999999999998e-07,
122
+ "loss": 0.175,
123
+ "step": 475
124
+ },
125
+ {
126
+ "epoch": 0.13,
127
+ "learning_rate": 2.4899999999999997e-07,
128
+ "loss": 0.3051,
129
+ "step": 500
130
+ },
131
+ {
132
+ "epoch": 0.14,
133
+ "learning_rate": 2.615e-07,
134
+ "loss": 0.176,
135
+ "step": 525
136
+ },
137
+ {
138
+ "epoch": 0.14,
139
+ "learning_rate": 2.74e-07,
140
+ "loss": 0.2517,
141
+ "step": 550
142
+ },
143
+ {
144
+ "epoch": 0.15,
145
+ "learning_rate": 2.865e-07,
146
+ "loss": 0.1863,
147
+ "step": 575
148
+ },
149
+ {
150
+ "epoch": 0.15,
151
+ "learning_rate": 2.9899999999999996e-07,
152
+ "loss": 0.2614,
153
+ "step": 600
154
+ },
155
+ {
156
+ "epoch": 0.16,
157
+ "learning_rate": 3.115e-07,
158
+ "loss": 0.1942,
159
+ "step": 625
160
+ },
161
+ {
162
+ "epoch": 0.17,
163
+ "learning_rate": 3.24e-07,
164
+ "loss": 0.2901,
165
+ "step": 650
166
+ },
167
+ {
168
+ "epoch": 0.17,
169
+ "learning_rate": 3.3650000000000003e-07,
170
+ "loss": 0.1587,
171
+ "step": 675
172
+ },
173
+ {
174
+ "epoch": 0.18,
175
+ "learning_rate": 3.4899999999999996e-07,
176
+ "loss": 0.2128,
177
+ "step": 700
178
+ },
179
+ {
180
+ "epoch": 0.19,
181
+ "learning_rate": 3.6149999999999995e-07,
182
+ "loss": 0.1647,
183
+ "step": 725
184
+ },
185
+ {
186
+ "epoch": 0.19,
187
+ "learning_rate": 3.74e-07,
188
+ "loss": 0.2473,
189
+ "step": 750
190
+ },
191
+ {
192
+ "epoch": 0.2,
193
+ "learning_rate": 3.8649999999999997e-07,
194
+ "loss": 0.1415,
195
+ "step": 775
196
+ },
197
+ {
198
+ "epoch": 0.21,
199
+ "learning_rate": 3.99e-07,
200
+ "loss": 0.2367,
201
+ "step": 800
202
+ },
203
+ {
204
+ "epoch": 0.21,
205
+ "learning_rate": 4.1149999999999995e-07,
206
+ "loss": 0.1498,
207
+ "step": 825
208
+ },
209
+ {
210
+ "epoch": 0.22,
211
+ "learning_rate": 4.24e-07,
212
+ "loss": 0.2161,
213
+ "step": 850
214
+ },
215
+ {
216
+ "epoch": 0.23,
217
+ "learning_rate": 4.3649999999999997e-07,
218
+ "loss": 0.144,
219
+ "step": 875
220
+ },
221
+ {
222
+ "epoch": 0.23,
223
+ "learning_rate": 4.49e-07,
224
+ "loss": 0.2316,
225
+ "step": 900
226
+ },
227
+ {
228
+ "epoch": 0.24,
229
+ "learning_rate": 4.615e-07,
230
+ "loss": 0.1449,
231
+ "step": 925
232
+ },
233
+ {
234
+ "epoch": 0.25,
235
+ "learning_rate": 4.7399999999999993e-07,
236
+ "loss": 0.1513,
237
+ "step": 950
238
+ },
239
+ {
240
+ "epoch": 0.25,
241
+ "learning_rate": 4.864999999999999e-07,
242
+ "loss": 0.1855,
243
+ "step": 975
244
+ },
245
+ {
246
+ "epoch": 0.26,
247
+ "learning_rate": 4.99e-07,
248
+ "loss": 0.199,
249
+ "step": 1000
250
+ },
251
+ {
252
+ "epoch": 0.26,
253
+ "eval_loss": 0.15628743171691895,
254
+ "eval_runtime": 184706.1006,
255
+ "eval_samples_per_second": 0.051,
256
+ "eval_steps_per_second": 0.006,
257
+ "eval_wer": 0.1124272786037492,
258
+ "step": 1000
259
+ },
260
+ {
261
+ "epoch": 0.26,
262
+ "learning_rate": 5.114999999999999e-07,
263
+ "loss": 0.1301,
264
+ "step": 1025
265
+ },
266
+ {
267
+ "epoch": 0.27,
268
+ "learning_rate": 5.24e-07,
269
+ "loss": 0.2193,
270
+ "step": 1050
271
+ },
272
+ {
273
+ "epoch": 0.28,
274
+ "learning_rate": 5.365e-07,
275
+ "loss": 0.1552,
276
+ "step": 1075
277
+ },
278
+ {
279
+ "epoch": 0.28,
280
+ "learning_rate": 5.490000000000001e-07,
281
+ "loss": 0.1865,
282
+ "step": 1100
283
+ },
284
+ {
285
+ "epoch": 0.29,
286
+ "learning_rate": 5.614999999999999e-07,
287
+ "loss": 0.1618,
288
+ "step": 1125
289
+ },
290
+ {
291
+ "epoch": 0.3,
292
+ "learning_rate": 5.739999999999999e-07,
293
+ "loss": 0.2259,
294
+ "step": 1150
295
+ },
296
+ {
297
+ "epoch": 0.3,
298
+ "learning_rate": 5.865e-07,
299
+ "loss": 0.1418,
300
+ "step": 1175
301
+ },
302
+ {
303
+ "epoch": 0.31,
304
+ "learning_rate": 5.989999999999999e-07,
305
+ "loss": 0.1916,
306
+ "step": 1200
307
+ },
308
+ {
309
+ "epoch": 0.32,
310
+ "learning_rate": 6.115e-07,
311
+ "loss": 0.1295,
312
+ "step": 1225
313
+ },
314
+ {
315
+ "epoch": 0.32,
316
+ "learning_rate": 6.24e-07,
317
+ "loss": 0.2112,
318
+ "step": 1250
319
+ },
320
+ {
321
+ "epoch": 0.33,
322
+ "learning_rate": 6.365e-07,
323
+ "loss": 0.1231,
324
+ "step": 1275
325
+ },
326
+ {
327
+ "epoch": 0.34,
328
+ "learning_rate": 6.49e-07,
329
+ "loss": 0.1914,
330
+ "step": 1300
331
+ },
332
+ {
333
+ "epoch": 0.34,
334
+ "learning_rate": 6.614999999999999e-07,
335
+ "loss": 0.1485,
336
+ "step": 1325
337
+ },
338
+ {
339
+ "epoch": 0.35,
340
+ "learning_rate": 6.74e-07,
341
+ "loss": 0.1958,
342
+ "step": 1350
343
+ },
344
+ {
345
+ "epoch": 0.35,
346
+ "learning_rate": 6.865e-07,
347
+ "loss": 0.1452,
348
+ "step": 1375
349
+ },
350
+ {
351
+ "epoch": 0.36,
352
+ "learning_rate": 6.989999999999999e-07,
353
+ "loss": 0.1624,
354
+ "step": 1400
355
+ },
356
+ {
357
+ "epoch": 0.37,
358
+ "learning_rate": 7.115e-07,
359
+ "loss": 0.1518,
360
+ "step": 1425
361
+ },
362
+ {
363
+ "epoch": 0.37,
364
+ "learning_rate": 7.24e-07,
365
+ "loss": 0.1935,
366
+ "step": 1450
367
+ },
368
+ {
369
+ "epoch": 0.38,
370
+ "learning_rate": 7.365e-07,
371
+ "loss": 0.1138,
372
+ "step": 1475
373
+ },
374
+ {
375
+ "epoch": 0.39,
376
+ "learning_rate": 7.489999999999999e-07,
377
+ "loss": 0.16,
378
+ "step": 1500
379
+ },
380
+ {
381
+ "epoch": 0.39,
382
+ "learning_rate": 7.614999999999999e-07,
383
+ "loss": 0.1279,
384
+ "step": 1525
385
+ },
386
+ {
387
+ "epoch": 0.4,
388
+ "learning_rate": 7.74e-07,
389
+ "loss": 0.1862,
390
+ "step": 1550
391
+ },
392
+ {
393
+ "epoch": 0.41,
394
+ "learning_rate": 7.864999999999999e-07,
395
+ "loss": 0.1537,
396
+ "step": 1575
397
+ },
398
+ {
399
+ "epoch": 0.41,
400
+ "learning_rate": 7.99e-07,
401
+ "loss": 0.1915,
402
+ "step": 1600
403
+ },
404
+ {
405
+ "epoch": 0.42,
406
+ "learning_rate": 8.115e-07,
407
+ "loss": 0.1303,
408
+ "step": 1625
409
+ },
410
+ {
411
+ "epoch": 0.43,
412
+ "learning_rate": 8.24e-07,
413
+ "loss": 0.1967,
414
+ "step": 1650
415
+ },
416
+ {
417
+ "epoch": 0.43,
418
+ "learning_rate": 8.365e-07,
419
+ "loss": 0.1418,
420
+ "step": 1675
421
+ },
422
+ {
423
+ "epoch": 0.44,
424
+ "learning_rate": 8.489999999999999e-07,
425
+ "loss": 0.194,
426
+ "step": 1700
427
+ },
428
+ {
429
+ "epoch": 0.45,
430
+ "learning_rate": 8.615e-07,
431
+ "loss": 0.1377,
432
+ "step": 1725
433
+ },
434
+ {
435
+ "epoch": 0.45,
436
+ "learning_rate": 8.739999999999999e-07,
437
+ "loss": 0.1534,
438
+ "step": 1750
439
+ },
440
+ {
441
+ "epoch": 0.46,
442
+ "learning_rate": 8.864999999999999e-07,
443
+ "loss": 0.1416,
444
+ "step": 1775
445
+ },
446
+ {
447
+ "epoch": 0.46,
448
+ "learning_rate": 8.99e-07,
449
+ "loss": 0.1869,
450
+ "step": 1800
451
+ },
452
+ {
453
+ "epoch": 0.47,
454
+ "learning_rate": 9.115e-07,
455
+ "loss": 0.1507,
456
+ "step": 1825
457
+ },
458
+ {
459
+ "epoch": 0.48,
460
+ "learning_rate": 9.24e-07,
461
+ "loss": 0.1571,
462
+ "step": 1850
463
+ },
464
+ {
465
+ "epoch": 0.48,
466
+ "learning_rate": 9.365e-07,
467
+ "loss": 0.1348,
468
+ "step": 1875
469
+ },
470
+ {
471
+ "epoch": 0.49,
472
+ "learning_rate": 9.489999999999999e-07,
473
+ "loss": 0.1981,
474
+ "step": 1900
475
+ },
476
+ {
477
+ "epoch": 0.5,
478
+ "learning_rate": 9.615e-07,
479
+ "loss": 0.1324,
480
+ "step": 1925
481
+ },
482
+ {
483
+ "epoch": 0.5,
484
+ "learning_rate": 9.74e-07,
485
+ "loss": 0.1712,
486
+ "step": 1950
487
+ },
488
+ {
489
+ "epoch": 0.51,
490
+ "learning_rate": 9.865e-07,
491
+ "loss": 0.1169,
492
+ "step": 1975
493
+ },
494
+ {
495
+ "epoch": 0.52,
496
+ "learning_rate": 9.989999999999999e-07,
497
+ "loss": 0.1654,
498
+ "step": 2000
499
+ },
500
+ {
501
+ "epoch": 0.52,
502
+ "eval_loss": 0.1500108540058136,
503
+ "eval_runtime": 188234.7285,
504
+ "eval_samples_per_second": 0.05,
505
+ "eval_steps_per_second": 0.006,
506
+ "eval_wer": 0.10521978021978022,
507
+ "step": 2000
508
+ },
509
+ {
510
+ "epoch": 0.52,
511
+ "learning_rate": 9.923333333333332e-07,
512
+ "loss": 0.1202,
513
+ "step": 2025
514
+ },
515
+ {
516
+ "epoch": 0.53,
517
+ "learning_rate": 9.84e-07,
518
+ "loss": 0.186,
519
+ "step": 2050
520
+ },
521
+ {
522
+ "epoch": 0.54,
523
+ "learning_rate": 9.756666666666666e-07,
524
+ "loss": 0.1106,
525
+ "step": 2075
526
+ },
527
+ {
528
+ "epoch": 0.54,
529
+ "learning_rate": 9.673333333333332e-07,
530
+ "loss": 0.1926,
531
+ "step": 2100
532
+ },
533
+ {
534
+ "epoch": 0.55,
535
+ "learning_rate": 9.589999999999998e-07,
536
+ "loss": 0.1178,
537
+ "step": 2125
538
+ },
539
+ {
540
+ "epoch": 0.55,
541
+ "learning_rate": 9.506666666666667e-07,
542
+ "loss": 0.1744,
543
+ "step": 2150
544
+ },
545
+ {
546
+ "epoch": 0.56,
547
+ "learning_rate": 9.423333333333333e-07,
548
+ "loss": 0.1263,
549
+ "step": 2175
550
+ },
551
+ {
552
+ "epoch": 0.57,
553
+ "learning_rate": 9.34e-07,
554
+ "loss": 0.1818,
555
+ "step": 2200
556
+ },
557
+ {
558
+ "epoch": 0.57,
559
+ "learning_rate": 9.256666666666666e-07,
560
+ "loss": 0.1466,
561
+ "step": 2225
562
+ },
563
+ {
564
+ "epoch": 0.58,
565
+ "learning_rate": 9.173333333333333e-07,
566
+ "loss": 0.1357,
567
+ "step": 2250
568
+ },
569
+ {
570
+ "epoch": 0.59,
571
+ "learning_rate": 9.09e-07,
572
+ "loss": 0.148,
573
+ "step": 2275
574
+ },
575
+ {
576
+ "epoch": 0.59,
577
+ "learning_rate": 9.006666666666666e-07,
578
+ "loss": 0.1684,
579
+ "step": 2300
580
+ },
581
+ {
582
+ "epoch": 0.6,
583
+ "learning_rate": 8.923333333333333e-07,
584
+ "loss": 0.1368,
585
+ "step": 2325
586
+ },
587
+ {
588
+ "epoch": 0.61,
589
+ "learning_rate": 8.839999999999999e-07,
590
+ "loss": 0.188,
591
+ "step": 2350
592
+ },
593
+ {
594
+ "epoch": 0.61,
595
+ "learning_rate": 8.756666666666666e-07,
596
+ "loss": 0.1299,
597
+ "step": 2375
598
+ },
599
+ {
600
+ "epoch": 0.62,
601
+ "learning_rate": 8.673333333333332e-07,
602
+ "loss": 0.1461,
603
+ "step": 2400
604
+ },
605
+ {
606
+ "epoch": 0.63,
607
+ "learning_rate": 8.59e-07,
608
+ "loss": 0.1569,
609
+ "step": 2425
610
+ },
611
+ {
612
+ "epoch": 0.63,
613
+ "learning_rate": 8.506666666666667e-07,
614
+ "loss": 0.1527,
615
+ "step": 2450
616
+ },
617
+ {
618
+ "epoch": 0.64,
619
+ "learning_rate": 8.423333333333334e-07,
620
+ "loss": 0.1041,
621
+ "step": 2475
622
+ },
623
+ {
624
+ "epoch": 0.65,
625
+ "learning_rate": 8.34e-07,
626
+ "loss": 0.157,
627
+ "step": 2500
628
+ },
629
+ {
630
+ "epoch": 0.65,
631
+ "learning_rate": 8.256666666666666e-07,
632
+ "loss": 0.1364,
633
+ "step": 2525
634
+ },
635
+ {
636
+ "epoch": 0.66,
637
+ "learning_rate": 8.173333333333333e-07,
638
+ "loss": 0.1685,
639
+ "step": 2550
640
+ },
641
+ {
642
+ "epoch": 0.66,
643
+ "learning_rate": 8.09e-07,
644
+ "loss": 0.1344,
645
+ "step": 2575
646
+ },
647
+ {
648
+ "epoch": 0.67,
649
+ "learning_rate": 8.006666666666666e-07,
650
+ "loss": 0.1589,
651
+ "step": 2600
652
+ },
653
+ {
654
+ "epoch": 0.68,
655
+ "learning_rate": 7.923333333333333e-07,
656
+ "loss": 0.1334,
657
+ "step": 2625
658
+ },
659
+ {
660
+ "epoch": 0.68,
661
+ "learning_rate": 7.84e-07,
662
+ "loss": 0.1297,
663
+ "step": 2650
664
+ },
665
+ {
666
+ "epoch": 0.69,
667
+ "learning_rate": 7.756666666666665e-07,
668
+ "loss": 0.1326,
669
+ "step": 2675
670
+ },
671
+ {
672
+ "epoch": 0.7,
673
+ "learning_rate": 7.673333333333332e-07,
674
+ "loss": 0.179,
675
+ "step": 2700
676
+ },
677
+ {
678
+ "epoch": 0.7,
679
+ "learning_rate": 7.59e-07,
680
+ "loss": 0.1242,
681
+ "step": 2725
682
+ },
683
+ {
684
+ "epoch": 0.71,
685
+ "learning_rate": 7.506666666666667e-07,
686
+ "loss": 0.1714,
687
+ "step": 2750
688
+ },
689
+ {
690
+ "epoch": 0.72,
691
+ "learning_rate": 7.423333333333333e-07,
692
+ "loss": 0.1503,
693
+ "step": 2775
694
+ },
695
+ {
696
+ "epoch": 0.72,
697
+ "learning_rate": 7.34e-07,
698
+ "loss": 0.131,
699
+ "step": 2800
700
+ },
701
+ {
702
+ "epoch": 0.73,
703
+ "learning_rate": 7.256666666666667e-07,
704
+ "loss": 0.118,
705
+ "step": 2825
706
+ },
707
+ {
708
+ "epoch": 0.74,
709
+ "learning_rate": 7.173333333333333e-07,
710
+ "loss": 0.1729,
711
+ "step": 2850
712
+ },
713
+ {
714
+ "epoch": 0.74,
715
+ "learning_rate": 7.089999999999999e-07,
716
+ "loss": 0.1193,
717
+ "step": 2875
718
+ },
719
+ {
720
+ "epoch": 0.75,
721
+ "learning_rate": 7.006666666666666e-07,
722
+ "loss": 0.1723,
723
+ "step": 2900
724
+ },
725
+ {
726
+ "epoch": 0.75,
727
+ "learning_rate": 6.923333333333333e-07,
728
+ "loss": 0.1393,
729
+ "step": 2925
730
+ },
731
+ {
732
+ "epoch": 0.76,
733
+ "learning_rate": 6.84e-07,
734
+ "loss": 0.1259,
735
+ "step": 2950
736
+ },
737
+ {
738
+ "epoch": 0.77,
739
+ "learning_rate": 6.756666666666666e-07,
740
+ "loss": 0.1228,
741
+ "step": 2975
742
+ },
743
+ {
744
+ "epoch": 0.77,
745
+ "learning_rate": 6.673333333333334e-07,
746
+ "loss": 0.1794,
747
+ "step": 3000
748
+ },
749
+ {
750
+ "epoch": 0.77,
751
+ "eval_loss": 0.13788852095603943,
752
+ "eval_runtime": 198193.9708,
753
+ "eval_samples_per_second": 0.047,
754
+ "eval_steps_per_second": 0.006,
755
+ "eval_wer": 0.09970911441499677,
756
+ "step": 3000
757
+ },
758
+ {
759
+ "epoch": 0.78,
760
+ "learning_rate": 6.59e-07,
761
+ "loss": 0.1334,
762
+ "step": 3025
763
+ },
764
+ {
765
+ "epoch": 0.79,
766
+ "learning_rate": 6.506666666666666e-07,
767
+ "loss": 0.1584,
768
+ "step": 3050
769
+ },
770
+ {
771
+ "epoch": 0.79,
772
+ "learning_rate": 6.423333333333333e-07,
773
+ "loss": 0.1331,
774
+ "step": 3075
775
+ },
776
+ {
777
+ "epoch": 0.8,
778
+ "learning_rate": 6.346666666666666e-07,
779
+ "loss": 0.1226,
780
+ "step": 3100
781
+ },
782
+ {
783
+ "epoch": 0.81,
784
+ "learning_rate": 6.263333333333332e-07,
785
+ "loss": 0.135,
786
+ "step": 3125
787
+ },
788
+ {
789
+ "epoch": 0.81,
790
+ "learning_rate": 6.18e-07,
791
+ "loss": 0.1538,
792
+ "step": 3150
793
+ },
794
+ {
795
+ "epoch": 0.82,
796
+ "learning_rate": 6.096666666666667e-07,
797
+ "loss": 0.1405,
798
+ "step": 3175
799
+ },
800
+ {
801
+ "epoch": 0.83,
802
+ "learning_rate": 6.013333333333334e-07,
803
+ "loss": 0.1534,
804
+ "step": 3200
805
+ },
806
+ {
807
+ "epoch": 0.83,
808
+ "learning_rate": 5.93e-07,
809
+ "loss": 0.1628,
810
+ "step": 3225
811
+ },
812
+ {
813
+ "epoch": 0.84,
814
+ "learning_rate": 5.846666666666667e-07,
815
+ "loss": 0.2002,
816
+ "step": 3250
817
+ },
818
+ {
819
+ "epoch": 0.85,
820
+ "learning_rate": 5.763333333333333e-07,
821
+ "loss": 0.1155,
822
+ "step": 3275
823
+ },
824
+ {
825
+ "epoch": 0.85,
826
+ "learning_rate": 5.679999999999999e-07,
827
+ "loss": 0.179,
828
+ "step": 3300
829
+ },
830
+ {
831
+ "epoch": 0.86,
832
+ "learning_rate": 5.596666666666666e-07,
833
+ "loss": 0.1471,
834
+ "step": 3325
835
+ },
836
+ {
837
+ "epoch": 0.86,
838
+ "learning_rate": 5.513333333333333e-07,
839
+ "loss": 0.1386,
840
+ "step": 3350
841
+ },
842
+ {
843
+ "epoch": 0.87,
844
+ "learning_rate": 5.43e-07,
845
+ "loss": 0.1185,
846
+ "step": 3375
847
+ },
848
+ {
849
+ "epoch": 0.88,
850
+ "learning_rate": 5.346666666666666e-07,
851
+ "loss": 0.1418,
852
+ "step": 3400
853
+ },
854
+ {
855
+ "epoch": 0.88,
856
+ "learning_rate": 5.263333333333333e-07,
857
+ "loss": 0.1015,
858
+ "step": 3425
859
+ },
860
+ {
861
+ "epoch": 0.89,
862
+ "learning_rate": 5.18e-07,
863
+ "loss": 0.1525,
864
+ "step": 3450
865
+ },
866
+ {
867
+ "epoch": 0.9,
868
+ "learning_rate": 5.096666666666667e-07,
869
+ "loss": 0.1212,
870
+ "step": 3475
871
+ },
872
+ {
873
+ "epoch": 0.9,
874
+ "learning_rate": 5.013333333333333e-07,
875
+ "loss": 0.1623,
876
+ "step": 3500
877
+ },
878
+ {
879
+ "epoch": 0.91,
880
+ "learning_rate": 4.93e-07,
881
+ "loss": 0.1178,
882
+ "step": 3525
883
+ },
884
+ {
885
+ "epoch": 0.92,
886
+ "learning_rate": 4.846666666666667e-07,
887
+ "loss": 0.1618,
888
+ "step": 3550
889
+ },
890
+ {
891
+ "epoch": 0.92,
892
+ "learning_rate": 4.763333333333333e-07,
893
+ "loss": 0.1187,
894
+ "step": 3575
895
+ },
896
+ {
897
+ "epoch": 0.93,
898
+ "learning_rate": 4.68e-07,
899
+ "loss": 0.1382,
900
+ "step": 3600
901
+ },
902
+ {
903
+ "epoch": 0.94,
904
+ "learning_rate": 4.5966666666666667e-07,
905
+ "loss": 0.1211,
906
+ "step": 3625
907
+ },
908
+ {
909
+ "epoch": 0.94,
910
+ "learning_rate": 4.5133333333333327e-07,
911
+ "loss": 0.1582,
912
+ "step": 3650
913
+ },
914
+ {
915
+ "epoch": 0.95,
916
+ "learning_rate": 4.43e-07,
917
+ "loss": 0.1201,
918
+ "step": 3675
919
+ },
920
+ {
921
+ "epoch": 0.95,
922
+ "learning_rate": 4.3466666666666664e-07,
923
+ "loss": 0.1571,
924
+ "step": 3700
925
+ },
926
+ {
927
+ "epoch": 0.96,
928
+ "learning_rate": 4.263333333333333e-07,
929
+ "loss": 0.1247,
930
+ "step": 3725
931
+ },
932
+ {
933
+ "epoch": 0.97,
934
+ "learning_rate": 4.1799999999999996e-07,
935
+ "loss": 0.1648,
936
+ "step": 3750
937
+ },
938
+ {
939
+ "epoch": 0.97,
940
+ "learning_rate": 4.0966666666666667e-07,
941
+ "loss": 0.1313,
942
+ "step": 3775
943
+ },
944
+ {
945
+ "epoch": 0.98,
946
+ "learning_rate": 4.0133333333333333e-07,
947
+ "loss": 0.1528,
948
+ "step": 3800
949
+ },
950
+ {
951
+ "epoch": 0.99,
952
+ "learning_rate": 3.93e-07,
953
+ "loss": 0.1252,
954
+ "step": 3825
955
+ },
956
+ {
957
+ "epoch": 0.99,
958
+ "learning_rate": 3.8466666666666664e-07,
959
+ "loss": 0.1674,
960
+ "step": 3850
961
+ },
962
+ {
963
+ "epoch": 1.0,
964
+ "learning_rate": 3.7633333333333335e-07,
965
+ "loss": 0.1192,
966
+ "step": 3875
967
+ },
968
+ {
969
+ "epoch": 1.01,
970
+ "learning_rate": 3.6799999999999996e-07,
971
+ "loss": 0.1054,
972
+ "step": 3900
973
+ },
974
+ {
975
+ "epoch": 1.01,
976
+ "learning_rate": 3.5966666666666667e-07,
977
+ "loss": 0.1353,
978
+ "step": 3925
979
+ },
980
+ {
981
+ "epoch": 1.02,
982
+ "learning_rate": 3.5133333333333333e-07,
983
+ "loss": 0.1004,
984
+ "step": 3950
985
+ },
986
+ {
987
+ "epoch": 1.03,
988
+ "learning_rate": 3.43e-07,
989
+ "loss": 0.1382,
990
+ "step": 3975
991
+ },
992
+ {
993
+ "epoch": 1.03,
994
+ "learning_rate": 3.3466666666666665e-07,
995
+ "loss": 0.0821,
996
+ "step": 4000
997
+ },
998
+ {
999
+ "epoch": 1.03,
1000
+ "eval_loss": 0.13208560645580292,
1001
+ "eval_runtime": 194476.4747,
1002
+ "eval_samples_per_second": 0.048,
1003
+ "eval_steps_per_second": 0.006,
1004
+ "eval_wer": 1.000743374272786,
1005
+ "step": 4000
1006
+ },
1007
+ {
1008
+ "epoch": 1.04,
1009
+ "learning_rate": 3.263333333333333e-07,
1010
+ "loss": 0.1092,
1011
+ "step": 4025
1012
+ },
1013
+ {
1014
+ "epoch": 1.05,
1015
+ "learning_rate": 3.18e-07,
1016
+ "loss": 0.0928,
1017
+ "step": 4050
1018
+ },
1019
+ {
1020
+ "epoch": 1.05,
1021
+ "learning_rate": 3.096666666666666e-07,
1022
+ "loss": 0.1481,
1023
+ "step": 4075
1024
+ },
1025
+ {
1026
+ "epoch": 1.06,
1027
+ "learning_rate": 3.0133333333333333e-07,
1028
+ "loss": 0.1043,
1029
+ "step": 4100
1030
+ },
1031
+ {
1032
+ "epoch": 1.06,
1033
+ "learning_rate": 2.93e-07,
1034
+ "loss": 0.1018,
1035
+ "step": 4125
1036
+ },
1037
+ {
1038
+ "epoch": 1.07,
1039
+ "learning_rate": 2.8466666666666665e-07,
1040
+ "loss": 0.104,
1041
+ "step": 4150
1042
+ },
1043
+ {
1044
+ "epoch": 1.08,
1045
+ "learning_rate": 2.763333333333333e-07,
1046
+ "loss": 0.1394,
1047
+ "step": 4175
1048
+ },
1049
+ {
1050
+ "epoch": 1.08,
1051
+ "learning_rate": 2.68e-07,
1052
+ "loss": 0.0962,
1053
+ "step": 4200
1054
+ },
1055
+ {
1056
+ "epoch": 1.09,
1057
+ "learning_rate": 2.596666666666666e-07,
1058
+ "loss": 0.1058,
1059
+ "step": 4225
1060
+ },
1061
+ {
1062
+ "epoch": 1.1,
1063
+ "learning_rate": 2.5133333333333333e-07,
1064
+ "loss": 0.0837,
1065
+ "step": 4250
1066
+ },
1067
+ {
1068
+ "epoch": 1.1,
1069
+ "learning_rate": 2.43e-07,
1070
+ "loss": 0.1396,
1071
+ "step": 4275
1072
+ },
1073
+ {
1074
+ "epoch": 1.11,
1075
+ "learning_rate": 2.3466666666666665e-07,
1076
+ "loss": 0.1005,
1077
+ "step": 4300
1078
+ },
1079
+ {
1080
+ "epoch": 1.12,
1081
+ "learning_rate": 2.263333333333333e-07,
1082
+ "loss": 0.131,
1083
+ "step": 4325
1084
+ },
1085
+ {
1086
+ "epoch": 1.12,
1087
+ "learning_rate": 2.18e-07,
1088
+ "loss": 0.1069,
1089
+ "step": 4350
1090
+ },
1091
+ {
1092
+ "epoch": 1.13,
1093
+ "learning_rate": 2.0966666666666665e-07,
1094
+ "loss": 0.136,
1095
+ "step": 4375
1096
+ },
1097
+ {
1098
+ "epoch": 1.14,
1099
+ "learning_rate": 2.0133333333333334e-07,
1100
+ "loss": 0.0954,
1101
+ "step": 4400
1102
+ },
1103
+ {
1104
+ "epoch": 1.14,
1105
+ "learning_rate": 1.93e-07,
1106
+ "loss": 0.1276,
1107
+ "step": 4425
1108
+ },
1109
+ {
1110
+ "epoch": 1.15,
1111
+ "learning_rate": 1.8466666666666665e-07,
1112
+ "loss": 0.1033,
1113
+ "step": 4450
1114
+ },
1115
+ {
1116
+ "epoch": 1.15,
1117
+ "learning_rate": 1.7633333333333334e-07,
1118
+ "loss": 0.1547,
1119
+ "step": 4475
1120
+ },
1121
+ {
1122
+ "epoch": 1.16,
1123
+ "learning_rate": 1.68e-07,
1124
+ "loss": 0.1093,
1125
+ "step": 4500
1126
+ },
1127
+ {
1128
+ "epoch": 1.17,
1129
+ "learning_rate": 1.5966666666666668e-07,
1130
+ "loss": 0.1435,
1131
+ "step": 4525
1132
+ },
1133
+ {
1134
+ "epoch": 1.17,
1135
+ "learning_rate": 1.513333333333333e-07,
1136
+ "loss": 0.1085,
1137
+ "step": 4550
1138
+ },
1139
+ {
1140
+ "epoch": 1.18,
1141
+ "learning_rate": 1.4299999999999997e-07,
1142
+ "loss": 0.1105,
1143
+ "step": 4575
1144
+ },
1145
+ {
1146
+ "epoch": 1.19,
1147
+ "learning_rate": 1.3466666666666665e-07,
1148
+ "loss": 0.0858,
1149
+ "step": 4600
1150
+ },
1151
+ {
1152
+ "epoch": 1.19,
1153
+ "learning_rate": 1.263333333333333e-07,
1154
+ "loss": 0.1336,
1155
+ "step": 4625
1156
+ },
1157
+ {
1158
+ "epoch": 1.2,
1159
+ "learning_rate": 1.1799999999999998e-07,
1160
+ "loss": 0.0857,
1161
+ "step": 4650
1162
+ },
1163
+ {
1164
+ "epoch": 1.21,
1165
+ "learning_rate": 1.0966666666666666e-07,
1166
+ "loss": 0.1146,
1167
+ "step": 4675
1168
+ },
1169
+ {
1170
+ "epoch": 1.21,
1171
+ "learning_rate": 1.0133333333333333e-07,
1172
+ "loss": 0.097,
1173
+ "step": 4700
1174
+ },
1175
+ {
1176
+ "epoch": 1.22,
1177
+ "learning_rate": 9.3e-08,
1178
+ "loss": 0.1146,
1179
+ "step": 4725
1180
+ },
1181
+ {
1182
+ "epoch": 1.23,
1183
+ "learning_rate": 8.466666666666667e-08,
1184
+ "loss": 0.0932,
1185
+ "step": 4750
1186
+ },
1187
+ {
1188
+ "epoch": 1.23,
1189
+ "learning_rate": 7.633333333333333e-08,
1190
+ "loss": 0.1245,
1191
+ "step": 4775
1192
+ },
1193
+ {
1194
+ "epoch": 1.24,
1195
+ "learning_rate": 6.8e-08,
1196
+ "loss": 0.0978,
1197
+ "step": 4800
1198
+ },
1199
+ {
1200
+ "epoch": 1.25,
1201
+ "learning_rate": 5.966666666666666e-08,
1202
+ "loss": 0.1283,
1203
+ "step": 4825
1204
+ },
1205
+ {
1206
+ "epoch": 1.25,
1207
+ "learning_rate": 5.133333333333333e-08,
1208
+ "loss": 0.1091,
1209
+ "step": 4850
1210
+ },
1211
+ {
1212
+ "epoch": 1.26,
1213
+ "learning_rate": 4.2999999999999995e-08,
1214
+ "loss": 0.1106,
1215
+ "step": 4875
1216
+ },
1217
+ {
1218
+ "epoch": 1.26,
1219
+ "learning_rate": 3.4666666666666666e-08,
1220
+ "loss": 0.1131,
1221
+ "step": 4900
1222
+ },
1223
+ {
1224
+ "epoch": 1.27,
1225
+ "learning_rate": 2.633333333333333e-08,
1226
+ "loss": 0.1219,
1227
+ "step": 4925
1228
+ },
1229
+ {
1230
+ "epoch": 1.28,
1231
+ "learning_rate": 1.8e-08,
1232
+ "loss": 0.1205,
1233
+ "step": 4950
1234
+ },
1235
+ {
1236
+ "epoch": 1.28,
1237
+ "learning_rate": 9.666666666666667e-09,
1238
+ "loss": 0.1408,
1239
+ "step": 4975
1240
+ },
1241
+ {
1242
+ "epoch": 1.29,
1243
+ "learning_rate": 1.3333333333333333e-09,
1244
+ "loss": 0.1292,
1245
+ "step": 5000
1246
+ },
1247
+ {
1248
+ "epoch": 1.29,
1249
+ "eval_loss": 0.132488414645195,
1250
+ "eval_runtime": 160662.2084,
1251
+ "eval_samples_per_second": 0.059,
1252
+ "eval_steps_per_second": 0.007,
1253
+ "eval_wer": 0.9998545572074984,
1254
+ "step": 5000
1255
+ }
1256
+ ],
1257
+ "logging_steps": 25,
1258
+ "max_steps": 5000,
1259
+ "num_input_tokens_seen": 0,
1260
+ "num_train_epochs": 2,
1261
+ "save_steps": 1000,
1262
+ "total_flos": 1.3589313726578688e+20,
1263
+ "train_batch_size": 8,
1264
+ "trial_name": null,
1265
+ "trial_params": null
1266
+ }
checkpoint-5000/training_args.bin ADDED
Binary file (4.86 kB). View file
 
eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.29,
3
+ "eval_loss": 0.132488414645195,
4
+ "eval_runtime": 171706.1904,
5
+ "eval_samples": 9399,
6
+ "eval_samples_per_second": 0.055,
7
+ "eval_steps_per_second": 0.007,
8
+ "eval_wer": 0.9998545572074984
9
+ }
generation_config.json CHANGED
@@ -55,7 +55,7 @@
55
  ],
56
  [
57
  2,
58
- 50360
59
  ]
60
  ],
61
  "is_multilingual": true,
@@ -161,10 +161,11 @@
161
  "<|yue|>": 50358,
162
  "<|zh|>": 50260
163
  },
164
- "max_initial_timestamp_index": 1,
165
  "max_length": 448,
166
  "no_timestamps_token_id": 50364,
167
  "pad_token_id": 50257,
 
168
  "return_timestamps": false,
169
  "suppress_tokens": [
170
  1,
 
55
  ],
56
  [
57
  2,
58
+ 50359
59
  ]
60
  ],
61
  "is_multilingual": true,
 
161
  "<|yue|>": 50358,
162
  "<|zh|>": 50260
163
  },
164
+ "max_initial_timestamp_index": 50,
165
  "max_length": 448,
166
  "no_timestamps_token_id": 50364,
167
  "pad_token_id": 50257,
168
+ "prev_sot_token_id": 50362,
169
  "return_timestamps": false,
170
  "suppress_tokens": [
171
  1,