File size: 8,735 Bytes
3d66cce
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
{
  "best_metric": 0.04373383894562721,
  "best_model_checkpoint": "w2v-bert-final-v2/checkpoint-10000",
  "epoch": 7.716049382716049,
  "eval_steps": 1000,
  "global_step": 10000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.23148148148148148,
      "grad_norm": 7.229776859283447,
      "learning_rate": 7.425e-06,
      "loss": 3.4047,
      "step": 300
    },
    {
      "epoch": 0.46296296296296297,
      "grad_norm": 11.000567436218262,
      "learning_rate": 1.4925e-05,
      "loss": 0.8648,
      "step": 600
    },
    {
      "epoch": 0.6944444444444444,
      "grad_norm": 11.643492698669434,
      "learning_rate": 2.2400000000000002e-05,
      "loss": 0.6263,
      "step": 900
    },
    {
      "epoch": 0.7716049382716049,
      "eval_loss": 0.5938597321510315,
      "eval_runtime": 49.2872,
      "eval_samples_per_second": 38.408,
      "eval_steps_per_second": 4.809,
      "eval_wer": 0.509299543180335,
      "step": 1000
    },
    {
      "epoch": 0.9259259259259259,
      "grad_norm": 16.209674835205078,
      "learning_rate": 2.9900000000000002e-05,
      "loss": 0.5512,
      "step": 1200
    },
    {
      "epoch": 1.1574074074074074,
      "grad_norm": 2.7776882648468018,
      "learning_rate": 3.74e-05,
      "loss": 0.4844,
      "step": 1500
    },
    {
      "epoch": 1.3888888888888888,
      "grad_norm": 5.179060459136963,
      "learning_rate": 4.4875e-05,
      "loss": 0.4641,
      "step": 1800
    },
    {
      "epoch": 1.5432098765432098,
      "eval_loss": 0.5058629512786865,
      "eval_runtime": 49.9367,
      "eval_samples_per_second": 37.908,
      "eval_steps_per_second": 4.746,
      "eval_wer": 0.45061996954535566,
      "step": 2000
    },
    {
      "epoch": 1.6203703703703702,
      "grad_norm": 2.7534990310668945,
      "learning_rate": 4.943236137667304e-05,
      "loss": 0.4629,
      "step": 2100
    },
    {
      "epoch": 1.8518518518518519,
      "grad_norm": 3.1153225898742676,
      "learning_rate": 4.763981835564054e-05,
      "loss": 0.4393,
      "step": 2400
    },
    {
      "epoch": 2.0833333333333335,
      "grad_norm": 2.560495376586914,
      "learning_rate": 4.584727533460803e-05,
      "loss": 0.3782,
      "step": 2700
    },
    {
      "epoch": 2.314814814814815,
      "grad_norm": 2.7176785469055176,
      "learning_rate": 4.4054732313575525e-05,
      "loss": 0.3054,
      "step": 3000
    },
    {
      "epoch": 2.314814814814815,
      "eval_loss": 0.34695935249328613,
      "eval_runtime": 49.4937,
      "eval_samples_per_second": 38.247,
      "eval_steps_per_second": 4.788,
      "eval_wer": 0.3308135740700457,
      "step": 3000
    },
    {
      "epoch": 2.5462962962962963,
      "grad_norm": 2.319594383239746,
      "learning_rate": 4.226816443594647e-05,
      "loss": 0.3079,
      "step": 3300
    },
    {
      "epoch": 2.7777777777777777,
      "grad_norm": 2.165224552154541,
      "learning_rate": 4.047562141491396e-05,
      "loss": 0.301,
      "step": 3600
    },
    {
      "epoch": 3.009259259259259,
      "grad_norm": 3.609168529510498,
      "learning_rate": 3.8683078393881456e-05,
      "loss": 0.2837,
      "step": 3900
    },
    {
      "epoch": 3.0864197530864197,
      "eval_loss": 0.2659013867378235,
      "eval_runtime": 50.1986,
      "eval_samples_per_second": 37.71,
      "eval_steps_per_second": 4.721,
      "eval_wer": 0.2583206438981945,
      "step": 4000
    },
    {
      "epoch": 3.240740740740741,
      "grad_norm": 1.6436126232147217,
      "learning_rate": 3.689053537284895e-05,
      "loss": 0.2191,
      "step": 4200
    },
    {
      "epoch": 3.4722222222222223,
      "grad_norm": 1.704137921333313,
      "learning_rate": 3.5103967495219884e-05,
      "loss": 0.2103,
      "step": 4500
    },
    {
      "epoch": 3.7037037037037037,
      "grad_norm": 6.373330593109131,
      "learning_rate": 3.331142447418738e-05,
      "loss": 0.2174,
      "step": 4800
    },
    {
      "epoch": 3.8580246913580245,
      "eval_loss": 0.19009661674499512,
      "eval_runtime": 49.6613,
      "eval_samples_per_second": 38.118,
      "eval_steps_per_second": 4.772,
      "eval_wer": 0.195072873613226,
      "step": 5000
    },
    {
      "epoch": 3.935185185185185,
      "grad_norm": 2.4362969398498535,
      "learning_rate": 3.151888145315488e-05,
      "loss": 0.1939,
      "step": 5100
    },
    {
      "epoch": 4.166666666666667,
      "grad_norm": 1.4299020767211914,
      "learning_rate": 2.9726338432122373e-05,
      "loss": 0.1666,
      "step": 5400
    },
    {
      "epoch": 4.398148148148148,
      "grad_norm": 1.3548126220703125,
      "learning_rate": 2.793379541108987e-05,
      "loss": 0.1569,
      "step": 5700
    },
    {
      "epoch": 4.62962962962963,
      "grad_norm": 1.5247106552124023,
      "learning_rate": 2.6141252390057363e-05,
      "loss": 0.152,
      "step": 6000
    },
    {
      "epoch": 4.62962962962963,
      "eval_loss": 0.15165844559669495,
      "eval_runtime": 49.8658,
      "eval_samples_per_second": 37.962,
      "eval_steps_per_second": 4.753,
      "eval_wer": 0.1540678703502284,
      "step": 6000
    },
    {
      "epoch": 4.861111111111111,
      "grad_norm": 3.228041410446167,
      "learning_rate": 2.434870936902486e-05,
      "loss": 0.1481,
      "step": 6300
    },
    {
      "epoch": 5.092592592592593,
      "grad_norm": 1.5059906244277954,
      "learning_rate": 2.2562141491395794e-05,
      "loss": 0.1256,
      "step": 6600
    },
    {
      "epoch": 5.324074074074074,
      "grad_norm": 3.538102626800537,
      "learning_rate": 2.0769598470363287e-05,
      "loss": 0.1089,
      "step": 6900
    },
    {
      "epoch": 5.401234567901234,
      "eval_loss": 0.11081259697675705,
      "eval_runtime": 49.8031,
      "eval_samples_per_second": 38.01,
      "eval_steps_per_second": 4.759,
      "eval_wer": 0.11404176636937133,
      "step": 7000
    },
    {
      "epoch": 5.555555555555555,
      "grad_norm": 0.6531468033790588,
      "learning_rate": 1.8977055449330787e-05,
      "loss": 0.1022,
      "step": 7200
    },
    {
      "epoch": 5.787037037037037,
      "grad_norm": 1.7070139646530151,
      "learning_rate": 1.718451242829828e-05,
      "loss": 0.1022,
      "step": 7500
    },
    {
      "epoch": 6.018518518518518,
      "grad_norm": 1.612425446510315,
      "learning_rate": 1.5397944550669215e-05,
      "loss": 0.0986,
      "step": 7800
    },
    {
      "epoch": 6.172839506172839,
      "eval_loss": 0.08366883546113968,
      "eval_runtime": 51.7773,
      "eval_samples_per_second": 36.56,
      "eval_steps_per_second": 4.577,
      "eval_wer": 0.08902545138133565,
      "step": 8000
    },
    {
      "epoch": 6.25,
      "grad_norm": 0.7614215612411499,
      "learning_rate": 1.3605401529636713e-05,
      "loss": 0.0766,
      "step": 8100
    },
    {
      "epoch": 6.481481481481482,
      "grad_norm": 2.542365550994873,
      "learning_rate": 1.181883365200765e-05,
      "loss": 0.0758,
      "step": 8400
    },
    {
      "epoch": 6.712962962962963,
      "grad_norm": 1.319675326347351,
      "learning_rate": 1.0026290630975144e-05,
      "loss": 0.0726,
      "step": 8700
    },
    {
      "epoch": 6.944444444444445,
      "grad_norm": 0.8109455704689026,
      "learning_rate": 8.233747609942639e-06,
      "loss": 0.0648,
      "step": 9000
    },
    {
      "epoch": 6.944444444444445,
      "eval_loss": 0.05808680132031441,
      "eval_runtime": 51.2984,
      "eval_samples_per_second": 36.902,
      "eval_steps_per_second": 4.62,
      "eval_wer": 0.06031107243854688,
      "step": 9000
    },
    {
      "epoch": 7.175925925925926,
      "grad_norm": 1.2560392618179321,
      "learning_rate": 6.441204588910134e-06,
      "loss": 0.051,
      "step": 9300
    },
    {
      "epoch": 7.407407407407407,
      "grad_norm": 3.7403156757354736,
      "learning_rate": 4.654636711281071e-06,
      "loss": 0.0473,
      "step": 9600
    },
    {
      "epoch": 7.638888888888889,
      "grad_norm": 0.7289965748786926,
      "learning_rate": 2.862093690248566e-06,
      "loss": 0.0499,
      "step": 9900
    },
    {
      "epoch": 7.716049382716049,
      "eval_loss": 0.04373383894562721,
      "eval_runtime": 50.2864,
      "eval_samples_per_second": 37.644,
      "eval_steps_per_second": 4.713,
      "eval_wer": 0.04606264955405699,
      "step": 10000
    }
  ],
  "logging_steps": 300,
  "max_steps": 10368,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 8,
  "save_steps": 1000,
  "total_flos": 2.075992650724199e+19,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}