longdnk commited on
Commit
36ae85e
·
verified ·
1 Parent(s): 52efbd5

Upload 10 files

Browse files
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6f5204db756e34f799719358f88672c8b4c6026f91eebda145d29f6b5a859364
3
  size 377851056
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb6f65f29ee2dd243f8694ffbcb7bbaa69f03ab6b54e8e0304daed4f8c389605
3
  size 377851056
optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8e5c4216be93855fbc013153efec4d3f85293e3472110511180806826aaf4fe2
3
  size 722217338
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eefd1ccbd216487e6fef630abe43cad9988fd32f3971fdd0e671c57ca15ee35b
3
  size 722217338
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5c63dd48407ceec67dee01431e0a481bb387803d9ef4f6a5541ecfead1b611e8
3
  size 377899102
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4ba484b726cd999ba2ea5517937ac5134e6dd5ba461411ddcfcc27176ea6f26
3
  size 377899102
rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b3913ae9493cc4baad846c1b707d069986928d1245d876521c5d6546dda5464f
3
  size 14244
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2f26a93007e1df9fe2d108b521dc1e9f1a2714dcc89ec6017185eb07f7a2e36
3
  size 14244
scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ea297f7ab56af8b17d03b96627bb16862b6bf5fc7088f8cd1f0a0f4093d4464c
3
  size 1064
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:82aca86b1ee054cd7ebba1ff6f00f6a548e6c9f0be9128f918ee36a8e37a51bd
3
  size 1064
trainer_state.json CHANGED
@@ -1,298 +1,398 @@
1
  {
2
- "best_metric": 0.05392587557435036,
3
- "best_model_checkpoint": "./wav2vec2-base-demo/checkpoint-7000",
4
- "epoch": 14.553014553014552,
5
  "eval_steps": 500,
6
- "global_step": 7000,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
  "epoch": 1.0395010395010396,
13
- "grad_norm": 2.4276885986328125,
14
- "learning_rate": 4.99e-05,
15
- "loss": 0.0935,
16
  "step": 500
17
  },
18
  {
19
  "epoch": 1.0395010395010396,
20
- "eval_cer": 0.03704485933935964,
21
- "eval_loss": 0.07600321620702744,
22
- "eval_mer": 0.10765687967760507,
23
- "eval_runtime": 34.3532,
24
- "eval_samples_per_second": 44.741,
25
- "eval_steps_per_second": 5.618,
26
- "eval_wer": 0.1080737444373808,
27
- "eval_wil": 0.19400502379268658,
28
- "eval_wip": 0.8059949762073134,
29
  "step": 500
30
  },
31
  {
32
  "epoch": 2.079002079002079,
33
- "grad_norm": 1.2353624105453491,
34
- "learning_rate": 4.8671291355389544e-05,
35
- "loss": 0.0914,
36
  "step": 1000
37
  },
38
  {
39
  "epoch": 2.079002079002079,
40
- "eval_cer": 0.037467263664779926,
41
- "eval_loss": 0.08513512462377548,
42
- "eval_mer": 0.1085740644640489,
43
- "eval_runtime": 34.2139,
44
- "eval_samples_per_second": 44.923,
45
- "eval_steps_per_second": 5.641,
46
- "eval_wer": 0.10882505923828238,
47
- "eval_wil": 0.19587239193616368,
48
- "eval_wip": 0.8041276080638363,
49
  "step": 1000
50
  },
51
  {
52
  "epoch": 3.1185031185031185,
53
- "grad_norm": 0.4545057415962219,
54
- "learning_rate": 4.733724653148346e-05,
55
- "loss": 0.1007,
56
  "step": 1500
57
  },
58
  {
59
  "epoch": 3.1185031185031185,
60
- "eval_cer": 0.03667877559066205,
61
- "eval_loss": 0.08273813873529434,
62
- "eval_mer": 0.10847477196628565,
63
- "eval_runtime": 34.1503,
64
- "eval_samples_per_second": 45.007,
65
- "eval_steps_per_second": 5.651,
66
- "eval_wer": 0.10859388545338959,
67
- "eval_wil": 0.1960486347087158,
68
- "eval_wip": 0.8039513652912842,
69
  "step": 1500
70
  },
71
  {
72
  "epoch": 4.158004158004158,
73
- "grad_norm": 0.7860220074653625,
74
- "learning_rate": 4.600586979722519e-05,
75
- "loss": 0.0883,
76
  "step": 2000
77
  },
78
  {
79
  "epoch": 4.158004158004158,
80
- "eval_cer": 0.03734054236715384,
81
- "eval_loss": 0.08130443096160889,
82
- "eval_mer": 0.10836359444348377,
83
- "eval_runtime": 34.1136,
84
- "eval_samples_per_second": 45.055,
85
- "eval_steps_per_second": 5.658,
86
- "eval_wer": 0.10865167889961279,
87
- "eval_wil": 0.19535745081720401,
88
- "eval_wip": 0.804642549182796,
89
  "step": 2000
90
  },
91
  {
92
  "epoch": 5.197505197505198,
93
- "grad_norm": 5.241793632507324,
94
- "learning_rate": 4.46718249733191e-05,
95
- "loss": 0.0988,
96
  "step": 2500
97
  },
98
  {
99
  "epoch": 5.197505197505198,
100
- "eval_cer": 0.03660837486975867,
101
- "eval_loss": 0.07141255587339401,
102
- "eval_mer": 0.10702128887093983,
103
- "eval_runtime": 34.0472,
104
- "eval_samples_per_second": 45.143,
105
- "eval_steps_per_second": 5.669,
106
- "eval_wer": 0.10720684274403283,
107
- "eval_wil": 0.1936232791597472,
108
- "eval_wip": 0.8063767208402528,
109
  "step": 2500
110
  },
111
  {
112
  "epoch": 6.237006237006237,
113
- "grad_norm": 1.3025474548339844,
114
- "learning_rate": 4.334044823906083e-05,
115
- "loss": 0.0962,
116
  "step": 3000
117
  },
118
  {
119
  "epoch": 6.237006237006237,
120
- "eval_cer": 0.035636844921291995,
121
- "eval_loss": 0.06766052544116974,
122
- "eval_mer": 0.1056455798396863,
123
- "eval_runtime": 34.1329,
124
- "eval_samples_per_second": 45.03,
125
- "eval_steps_per_second": 5.654,
126
- "eval_wer": 0.10587759348089927,
127
- "eval_wil": 0.1915670539174873,
128
- "eval_wip": 0.8084329460825127,
129
  "step": 3000
130
  },
131
  {
132
  "epoch": 7.276507276507276,
133
- "grad_norm": 1.1820029020309448,
134
- "learning_rate": 4.200907150480256e-05,
135
- "loss": 0.0951,
136
  "step": 3500
137
  },
138
  {
139
  "epoch": 7.276507276507276,
140
- "eval_cer": 0.03551012362366591,
141
- "eval_loss": 0.06142089143395424,
142
- "eval_mer": 0.10593122547888299,
143
- "eval_runtime": 34.0865,
144
- "eval_samples_per_second": 45.091,
145
- "eval_steps_per_second": 5.662,
146
- "eval_wer": 0.10610876726579206,
147
- "eval_wil": 0.19198195094690818,
148
- "eval_wip": 0.8080180490530918,
149
  "step": 3500
150
  },
151
  {
152
  "epoch": 8.316008316008316,
153
- "grad_norm": 3.633211851119995,
154
- "learning_rate": 4.0677694770544295e-05,
155
- "loss": 0.1161,
156
  "step": 4000
157
  },
158
  {
159
  "epoch": 8.316008316008316,
160
- "eval_cer": 0.03551012362366591,
161
- "eval_loss": 0.06067756190896034,
162
- "eval_mer": 0.10610691424946658,
163
- "eval_runtime": 34.0596,
164
- "eval_samples_per_second": 45.127,
165
- "eval_steps_per_second": 5.667,
166
- "eval_wer": 0.10633994105068485,
167
- "eval_wil": 0.19244783032392676,
168
- "eval_wip": 0.8075521696760732,
169
  "step": 4000
170
  },
171
  {
172
  "epoch": 9.355509355509355,
173
- "grad_norm": 1.8992377519607544,
174
- "learning_rate": 3.934364994663821e-05,
175
- "loss": 0.1202,
176
  "step": 4500
177
  },
178
  {
179
  "epoch": 9.355509355509355,
180
- "eval_cer": 0.036453493283771225,
181
- "eval_loss": 0.07183138281106949,
182
- "eval_mer": 0.10711813567143516,
183
- "eval_runtime": 34.0394,
184
- "eval_samples_per_second": 45.154,
185
- "eval_steps_per_second": 5.67,
186
- "eval_wer": 0.10732242963647923,
187
- "eval_wil": 0.19300207421517157,
188
- "eval_wip": 0.8069979257848284,
189
  "step": 4500
190
  },
191
  {
192
  "epoch": 10.395010395010395,
193
- "grad_norm": 1.5443813800811768,
194
- "learning_rate": 3.800960512273213e-05,
195
- "loss": 0.1431,
196
  "step": 5000
197
  },
198
  {
199
  "epoch": 10.395010395010395,
200
- "eval_cer": 0.036031088958350936,
201
- "eval_loss": 0.0653519257903099,
202
- "eval_mer": 0.10687726984492996,
203
- "eval_runtime": 34.4227,
204
- "eval_samples_per_second": 44.651,
205
- "eval_steps_per_second": 5.607,
206
- "eval_wer": 0.10714904929780963,
207
- "eval_wil": 0.19299958643561255,
208
- "eval_wip": 0.8070004135643875,
209
  "step": 5000
210
  },
211
  {
212
  "epoch": 11.434511434511435,
213
- "grad_norm": 0.9764054417610168,
214
- "learning_rate": 3.667556029882604e-05,
215
- "loss": 0.1356,
216
  "step": 5500
217
  },
218
  {
219
  "epoch": 11.434511434511435,
220
- "eval_cer": 0.03546788319112388,
221
- "eval_loss": 0.07734616100788116,
222
- "eval_mer": 0.10516307479543621,
223
- "eval_runtime": 34.1476,
224
- "eval_samples_per_second": 45.01,
225
- "eval_steps_per_second": 5.652,
226
- "eval_wer": 0.10547303935733687,
227
- "eval_wil": 0.18957490448529868,
228
- "eval_wip": 0.8104250955147013,
229
  "step": 5500
230
  },
231
  {
232
  "epoch": 12.474012474012474,
233
- "grad_norm": 1.104174017906189,
234
- "learning_rate": 3.5341515474919964e-05,
235
- "loss": 0.1561,
236
  "step": 6000
237
  },
238
  {
239
  "epoch": 12.474012474012474,
240
- "eval_cer": 0.035862127228182814,
241
- "eval_loss": 0.06988665461540222,
242
- "eval_mer": 0.10506015773415463,
243
- "eval_runtime": 34.0048,
244
- "eval_samples_per_second": 45.2,
245
- "eval_steps_per_second": 5.676,
246
- "eval_wer": 0.10547303935733687,
247
- "eval_wil": 0.1890727356961288,
248
- "eval_wip": 0.8109272643038712,
249
  "step": 6000
250
  },
251
  {
252
  "epoch": 13.513513513513514,
253
- "grad_norm": 4.416908264160156,
254
- "learning_rate": 3.400747065101387e-05,
255
- "loss": 0.134,
256
  "step": 6500
257
  },
258
  {
259
  "epoch": 13.513513513513514,
260
- "eval_cer": 0.0349750781448002,
261
- "eval_loss": 0.0634101927280426,
262
- "eval_mer": 0.10508709193678625,
263
- "eval_runtime": 34.1003,
264
- "eval_samples_per_second": 45.073,
265
- "eval_steps_per_second": 5.66,
266
- "eval_wer": 0.10529965901866728,
267
- "eval_wil": 0.19050757507724403,
268
- "eval_wip": 0.809492424922756,
269
  "step": 6500
270
  },
271
  {
272
  "epoch": 14.553014553014552,
273
- "grad_norm": 0.7748957276344299,
274
- "learning_rate": 3.267342582710779e-05,
275
- "loss": 0.1304,
276
  "step": 7000
277
  },
278
  {
279
  "epoch": 14.553014553014552,
280
- "eval_cer": 0.03490467742389682,
281
- "eval_loss": 0.05392587557435036,
282
- "eval_mer": 0.10498097544102386,
283
- "eval_runtime": 34.1635,
284
- "eval_samples_per_second": 44.99,
285
- "eval_steps_per_second": 5.649,
286
- "eval_wer": 0.10524186557244408,
287
- "eval_wil": 0.18975677659084123,
288
- "eval_wip": 0.8102432234091588,
289
  "step": 7000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
290
  }
291
  ],
292
  "logging_steps": 500,
293
- "max_steps": 19240,
294
  "num_input_tokens_seen": 0,
295
- "num_train_epochs": 40,
296
  "save_steps": 500,
297
  "stateful_callbacks": {
298
  "TrainerControl": {
@@ -301,12 +401,12 @@
301
  "should_evaluate": false,
302
  "should_log": false,
303
  "should_save": true,
304
- "should_training_stop": false
305
  },
306
  "attributes": {}
307
  }
308
  },
309
- "total_flos": 4.4824764428392996e+18,
310
  "train_batch_size": 16,
311
  "trial_name": null,
312
  "trial_params": null
 
1
  {
2
+ "best_metric": 0.04381340742111206,
3
+ "best_model_checkpoint": "./wav2vec2-base-demo/checkpoint-7500",
4
+ "epoch": 20.0,
5
  "eval_steps": 500,
6
+ "global_step": 9620,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
  "epoch": 1.0395010395010396,
13
+ "grad_norm": 2.0576889514923096,
14
+ "learning_rate": 5e-05,
15
+ "loss": 0.0875,
16
  "step": 500
17
  },
18
  {
19
  "epoch": 1.0395010395010396,
20
+ "eval_cer": 0.035779866801874644,
21
+ "eval_loss": 0.06287091970443726,
22
+ "eval_mer": 0.10693836308521387,
23
+ "eval_runtime": 38.7834,
24
+ "eval_samples_per_second": 39.63,
25
+ "eval_steps_per_second": 4.976,
26
+ "eval_wer": 0.1073481347909388,
27
+ "eval_wil": 0.19249465427434787,
28
+ "eval_wip": 0.8075053457256521,
29
  "step": 500
30
  },
31
  {
32
  "epoch": 2.079002079002079,
33
+ "grad_norm": 2.8653175830841064,
34
+ "learning_rate": 4.7258771929824566e-05,
35
+ "loss": 0.0756,
36
  "step": 1000
37
  },
38
  {
39
  "epoch": 2.079002079002079,
40
+ "eval_cer": 0.03587579137775098,
41
+ "eval_loss": 0.07413601130247116,
42
+ "eval_mer": 0.10741198965245755,
43
+ "eval_runtime": 38.5405,
44
+ "eval_samples_per_second": 39.88,
45
+ "eval_steps_per_second": 5.008,
46
+ "eval_wer": 0.10762988842556069,
47
+ "eval_wil": 0.1943762121634257,
48
+ "eval_wip": 0.8056237878365743,
49
  "step": 1000
50
  },
51
  {
52
  "epoch": 3.1185031185031185,
53
+ "grad_norm": 2.5419676303863525,
54
+ "learning_rate": 4.451754385964912e-05,
55
+ "loss": 0.0846,
56
  "step": 1500
57
  },
58
  {
59
  "epoch": 3.1185031185031185,
60
+ "eval_cer": 0.03573875626935621,
61
+ "eval_loss": 0.06114184856414795,
62
+ "eval_mer": 0.10740366250983036,
63
+ "eval_runtime": 38.4569,
64
+ "eval_samples_per_second": 39.967,
65
+ "eval_steps_per_second": 5.019,
66
+ "eval_wer": 0.10774258987940945,
67
+ "eval_wil": 0.19442229772038444,
68
+ "eval_wip": 0.8055777022796156,
69
  "step": 1500
70
  },
71
  {
72
  "epoch": 4.158004158004158,
73
+ "grad_norm": 0.5870616436004639,
74
+ "learning_rate": 4.178179824561404e-05,
75
+ "loss": 0.0702,
76
  "step": 2000
77
  },
78
  {
79
  "epoch": 4.158004158004158,
80
+ "eval_cer": 0.03601282648614575,
81
+ "eval_loss": 0.06569497287273407,
82
+ "eval_mer": 0.1072850653356514,
83
+ "eval_runtime": 38.397,
84
+ "eval_samples_per_second": 40.029,
85
+ "eval_steps_per_second": 5.026,
86
+ "eval_wer": 0.10779894060633383,
87
+ "eval_wil": 0.19276907013348765,
88
+ "eval_wip": 0.8072309298665123,
89
  "step": 2000
90
  },
91
  {
92
  "epoch": 5.197505197505198,
93
+ "grad_norm": 0.7855449318885803,
94
+ "learning_rate": 3.90405701754386e-05,
95
+ "loss": 0.0783,
96
  "step": 2500
97
  },
98
  {
99
  "epoch": 5.197505197505198,
100
+ "eval_cer": 0.03561542467180091,
101
+ "eval_loss": 0.059767913073301315,
102
+ "eval_mer": 0.10633408071748879,
103
+ "eval_runtime": 38.4232,
104
+ "eval_samples_per_second": 40.002,
105
+ "eval_steps_per_second": 5.023,
106
+ "eval_wer": 0.10689732897554378,
107
+ "eval_wil": 0.19087190993998937,
108
+ "eval_wip": 0.8091280900600106,
109
  "step": 2500
110
  },
111
  {
112
  "epoch": 6.237006237006237,
113
+ "grad_norm": 1.664757251739502,
114
+ "learning_rate": 3.630482456140351e-05,
115
+ "loss": 0.0743,
116
  "step": 3000
117
  },
118
  {
119
  "epoch": 6.237006237006237,
120
+ "eval_cer": 0.03521802285745608,
121
+ "eval_loss": 0.058559250086545944,
122
+ "eval_mer": 0.10588895120583286,
123
+ "eval_runtime": 38.4379,
124
+ "eval_samples_per_second": 39.987,
125
+ "eval_steps_per_second": 5.021,
126
+ "eval_wer": 0.10639017243322439,
127
+ "eval_wil": 0.1914301768679808,
128
+ "eval_wip": 0.8085698231320192,
129
  "step": 3000
130
  },
131
  {
132
  "epoch": 7.276507276507276,
133
+ "grad_norm": 1.1673611402511597,
134
+ "learning_rate": 3.356907894736842e-05,
135
+ "loss": 0.0706,
136
  "step": 3500
137
  },
138
  {
139
  "epoch": 7.276507276507276,
140
+ "eval_cer": 0.033943596349384715,
141
+ "eval_loss": 0.0515948086977005,
142
+ "eval_mer": 0.10461261868644306,
143
+ "eval_runtime": 38.4144,
144
+ "eval_samples_per_second": 40.011,
145
+ "eval_steps_per_second": 5.024,
146
+ "eval_wer": 0.10492505353319058,
147
+ "eval_wil": 0.18992491039188297,
148
+ "eval_wip": 0.810075089608117,
149
  "step": 3500
150
  },
151
  {
152
  "epoch": 8.316008316008316,
153
+ "grad_norm": 1.4960205554962158,
154
+ "learning_rate": 3.0833333333333335e-05,
155
+ "loss": 0.0861,
156
  "step": 4000
157
  },
158
  {
159
  "epoch": 8.316008316008316,
160
+ "eval_cer": 0.03432729465289007,
161
+ "eval_loss": 0.05460890755057335,
162
+ "eval_mer": 0.1054821478774248,
163
+ "eval_runtime": 38.5397,
164
+ "eval_samples_per_second": 39.881,
165
+ "eval_steps_per_second": 5.008,
166
+ "eval_wer": 0.10571396371013186,
167
+ "eval_wil": 0.19116659580295647,
168
+ "eval_wip": 0.8088334041970435,
169
  "step": 4000
170
  },
171
  {
172
  "epoch": 9.355509355509355,
173
+ "grad_norm": 1.2034363746643066,
174
+ "learning_rate": 2.80921052631579e-05,
175
+ "loss": 0.0815,
176
  "step": 4500
177
  },
178
  {
179
  "epoch": 9.355509355509355,
180
+ "eval_cer": 0.033710636665113605,
181
+ "eval_loss": 0.04747992008924484,
182
+ "eval_mer": 0.10431998202348183,
183
+ "eval_runtime": 38.4001,
184
+ "eval_samples_per_second": 40.026,
185
+ "eval_steps_per_second": 5.026,
186
+ "eval_wer": 0.10464329989856869,
187
+ "eval_wil": 0.18925902394749305,
188
+ "eval_wip": 0.810740976052507,
189
  "step": 4500
190
  },
191
  {
192
  "epoch": 10.395010395010395,
193
+ "grad_norm": 1.3969591856002808,
194
+ "learning_rate": 2.535087719298246e-05,
195
+ "loss": 0.0945,
196
  "step": 5000
197
  },
198
  {
199
  "epoch": 10.395010395010395,
200
+ "eval_cer": 0.03349138049168197,
201
+ "eval_loss": 0.04973128065466881,
202
+ "eval_mer": 0.10454392081880554,
203
+ "eval_runtime": 38.8567,
204
+ "eval_samples_per_second": 39.556,
205
+ "eval_steps_per_second": 4.967,
206
+ "eval_wer": 0.10475600135241744,
207
+ "eval_wil": 0.19024756262985354,
208
+ "eval_wip": 0.8097524373701465,
209
  "step": 5000
210
  },
211
  {
212
  "epoch": 11.434511434511435,
213
+ "grad_norm": 1.1010193824768066,
214
+ "learning_rate": 2.260964912280702e-05,
215
+ "loss": 0.0887,
216
  "step": 5500
217
  },
218
  {
219
  "epoch": 11.434511434511435,
220
+ "eval_cer": 0.03318990325321347,
221
+ "eval_loss": 0.047050829976797104,
222
+ "eval_mer": 0.10347152005392653,
223
+ "eval_runtime": 38.4763,
224
+ "eval_samples_per_second": 39.947,
225
+ "eval_steps_per_second": 5.016,
226
+ "eval_wer": 0.10379803899470304,
227
+ "eval_wil": 0.1878148971119521,
228
+ "eval_wip": 0.8121851028880479,
229
  "step": 5500
230
  },
231
  {
232
  "epoch": 12.474012474012474,
233
+ "grad_norm": 0.8793329000473022,
234
+ "learning_rate": 1.986842105263158e-05,
235
+ "loss": 0.1057,
236
  "step": 6000
237
  },
238
  {
239
  "epoch": 12.474012474012474,
240
+ "eval_cer": 0.03343656644832406,
241
+ "eval_loss": 0.044334810227155685,
242
+ "eval_mer": 0.1031848564848621,
243
+ "eval_runtime": 38.3629,
244
+ "eval_samples_per_second": 40.065,
245
+ "eval_steps_per_second": 5.031,
246
+ "eval_wer": 0.10351628536008115,
247
+ "eval_wil": 0.1874799681437187,
248
+ "eval_wip": 0.8125200318562813,
249
  "step": 6000
250
  },
251
  {
252
  "epoch": 13.513513513513514,
253
+ "grad_norm": 2.145364761352539,
254
+ "learning_rate": 1.712719298245614e-05,
255
+ "loss": 0.0852,
256
  "step": 6500
257
  },
258
  {
259
  "epoch": 13.513513513513514,
260
+ "eval_cer": 0.03316249623153452,
261
+ "eval_loss": 0.05074913427233696,
262
+ "eval_mer": 0.10275280898876404,
263
+ "eval_runtime": 38.4426,
264
+ "eval_samples_per_second": 39.982,
265
+ "eval_steps_per_second": 5.02,
266
+ "eval_wer": 0.10306547954468613,
267
+ "eval_wil": 0.18655687555087064,
268
+ "eval_wip": 0.8134431244491294,
269
  "step": 6500
270
  },
271
  {
272
  "epoch": 14.553014553014552,
273
+ "grad_norm": 1.7652612924575806,
274
+ "learning_rate": 1.4385964912280702e-05,
275
+ "loss": 0.0839,
276
  "step": 7000
277
  },
278
  {
279
  "epoch": 14.553014553014552,
280
+ "eval_cer": 0.03292953654726341,
281
+ "eval_loss": 0.04428742080926895,
282
+ "eval_mer": 0.1023710529272952,
283
+ "eval_runtime": 38.4249,
284
+ "eval_samples_per_second": 40.0,
285
+ "eval_steps_per_second": 5.023,
286
+ "eval_wer": 0.10267102445621548,
287
+ "eval_wil": 0.18581708506190875,
288
+ "eval_wip": 0.8141829149380913,
289
  "step": 7000
290
+ },
291
+ {
292
+ "epoch": 15.592515592515593,
293
+ "grad_norm": 0.4182071089744568,
294
+ "learning_rate": 1.1655701754385966e-05,
295
+ "loss": 0.114,
296
+ "step": 7500
297
+ },
298
+ {
299
+ "epoch": 15.592515592515593,
300
+ "eval_cer": 0.03295694356894236,
301
+ "eval_loss": 0.04381340742111206,
302
+ "eval_mer": 0.10235380034829504,
303
+ "eval_runtime": 38.4514,
304
+ "eval_samples_per_second": 39.973,
305
+ "eval_steps_per_second": 5.019,
306
+ "eval_wer": 0.10267102445621548,
307
+ "eval_wil": 0.18560348362381918,
308
+ "eval_wip": 0.8143965163761808,
309
+ "step": 7500
310
+ },
311
+ {
312
+ "epoch": 16.632016632016633,
313
+ "grad_norm": 0.8875317573547363,
314
+ "learning_rate": 8.914473684210526e-06,
315
+ "loss": 0.1054,
316
+ "step": 8000
317
+ },
318
+ {
319
+ "epoch": 16.632016632016633,
320
+ "eval_cer": 0.03250472771123962,
321
+ "eval_loss": 0.04421629384160042,
322
+ "eval_mer": 0.10242860355295705,
323
+ "eval_runtime": 38.4795,
324
+ "eval_samples_per_second": 39.943,
325
+ "eval_steps_per_second": 5.016,
326
+ "eval_wer": 0.10267102445621548,
327
+ "eval_wil": 0.18642151998260104,
328
+ "eval_wip": 0.813578480017399,
329
+ "step": 8000
330
+ },
331
+ {
332
+ "epoch": 17.671517671517673,
333
+ "grad_norm": 1.46517813205719,
334
+ "learning_rate": 6.1787280701754385e-06,
335
+ "loss": 0.108,
336
+ "step": 8500
337
+ },
338
+ {
339
+ "epoch": 17.671517671517673,
340
+ "eval_cer": 0.03273768739551073,
341
+ "eval_loss": 0.044399794191122055,
342
+ "eval_mer": 0.1021680521231184,
343
+ "eval_runtime": 38.4253,
344
+ "eval_samples_per_second": 40.0,
345
+ "eval_steps_per_second": 5.023,
346
+ "eval_wer": 0.10250197227544236,
347
+ "eval_wil": 0.18531456248554645,
348
+ "eval_wip": 0.8146854375144535,
349
+ "step": 8500
350
+ },
351
+ {
352
+ "epoch": 18.71101871101871,
353
+ "grad_norm": 0.9103938341140747,
354
+ "learning_rate": 3.4429824561403515e-06,
355
+ "loss": 0.1081,
356
+ "step": 9000
357
+ },
358
+ {
359
+ "epoch": 18.71101871101871,
360
+ "eval_cer": 0.03260065228711596,
361
+ "eval_loss": 0.044667795300483704,
362
+ "eval_mer": 0.10185861081475658,
363
+ "eval_runtime": 38.5859,
364
+ "eval_samples_per_second": 39.833,
365
+ "eval_steps_per_second": 5.002,
366
+ "eval_wer": 0.10222021864082047,
367
+ "eval_wil": 0.18466398681705298,
368
+ "eval_wip": 0.815336013182947,
369
+ "step": 9000
370
+ },
371
+ {
372
+ "epoch": 19.75051975051975,
373
+ "grad_norm": 1.2316771745681763,
374
+ "learning_rate": 7.017543859649123e-07,
375
+ "loss": 0.1122,
376
+ "step": 9500
377
+ },
378
+ {
379
+ "epoch": 19.75051975051975,
380
+ "eval_cer": 0.03240880313536328,
381
+ "eval_loss": 0.04449043795466423,
382
+ "eval_mer": 0.10180347210517444,
383
+ "eval_runtime": 38.5699,
384
+ "eval_samples_per_second": 39.85,
385
+ "eval_steps_per_second": 5.004,
386
+ "eval_wer": 0.10210751718697171,
387
+ "eval_wil": 0.1850184634725608,
388
+ "eval_wip": 0.8149815365274392,
389
+ "step": 9500
390
  }
391
  ],
392
  "logging_steps": 500,
393
+ "max_steps": 9620,
394
  "num_input_tokens_seen": 0,
395
+ "num_train_epochs": 20,
396
  "save_steps": 500,
397
  "stateful_callbacks": {
398
  "TrainerControl": {
 
401
  "should_evaluate": false,
402
  "should_log": false,
403
  "should_save": true,
404
+ "should_training_stop": true
405
  },
406
  "attributes": {}
407
  }
408
  },
409
+ "total_flos": 6.154380357141717e+18,
410
  "train_batch_size": 16,
411
  "trial_name": null,
412
  "trial_params": null
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:775696d645371d4025f4adc46331697961085adc3d0812695e8af16d5dc92a47
3
  size 5176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:38d136ab42c7a1566f953cd7f5347049927e6445ad163737d0a0ffc5e5be98b2
3
  size 5176