File size: 10,109 Bytes
b7af886
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 15.992800575953924,
  "eval_steps": 500,
  "global_step": 11104,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.48,
      "learning_rate": 1.9399615754082615e-05,
      "loss": 2.0304,
      "step": 500
    },
    {
      "epoch": 0.96,
      "learning_rate": 1.879923150816523e-05,
      "loss": 1.781,
      "step": 1000
    },
    {
      "epoch": 2.0,
      "eval_f1": 0.9088,
      "eval_gen_len": 26.88909090909091,
      "eval_loss": 1.579687476158142,
      "eval_precision": 0.908,
      "eval_recall": 0.91,
      "eval_rouge1": 0.4708,
      "eval_rouge2": 0.2219,
      "eval_rougeL": 0.3892,
      "eval_rougeLsum": 0.389,
      "eval_runtime": 1186.1406,
      "eval_samples_per_second": 4.637,
      "eval_steps_per_second": 0.29,
      "step": 1388
    },
    {
      "epoch": 2.16,
      "learning_rate": 1.729827089337176e-05,
      "loss": 1.7026,
      "step": 1500
    },
    {
      "epoch": 2.88,
      "learning_rate": 1.6397694524495677e-05,
      "loss": 1.6618,
      "step": 2000
    },
    {
      "epoch": 3.0,
      "eval_f1": 0.91,
      "eval_gen_len": 26.728181818181817,
      "eval_loss": 1.5411016941070557,
      "eval_precision": 0.9094,
      "eval_recall": 0.9111,
      "eval_rouge1": 0.4776,
      "eval_rouge2": 0.2303,
      "eval_rougeL": 0.3977,
      "eval_rougeLsum": 0.3973,
      "eval_runtime": 1083.838,
      "eval_samples_per_second": 5.075,
      "eval_steps_per_second": 0.317,
      "step": 2083
    },
    {
      "epoch": 3.6,
      "learning_rate": 1.5497118155619597e-05,
      "loss": 1.626,
      "step": 2500
    },
    {
      "epoch": 4.0,
      "eval_f1": 0.911,
      "eval_gen_len": 26.759636363636364,
      "eval_loss": 1.5170917510986328,
      "eval_precision": 0.9102,
      "eval_recall": 0.9121,
      "eval_rouge1": 0.4834,
      "eval_rouge2": 0.2345,
      "eval_rougeL": 0.402,
      "eval_rougeLsum": 0.402,
      "eval_runtime": 1053.82,
      "eval_samples_per_second": 5.219,
      "eval_steps_per_second": 0.326,
      "step": 2776
    },
    {
      "epoch": 4.32,
      "learning_rate": 1.4596541786743516e-05,
      "loss": 1.5918,
      "step": 3000
    },
    {
      "epoch": 5.0,
      "eval_f1": 0.9112,
      "eval_gen_len": 26.647636363636362,
      "eval_loss": 1.500138521194458,
      "eval_precision": 0.9106,
      "eval_recall": 0.9122,
      "eval_rouge1": 0.4853,
      "eval_rouge2": 0.2365,
      "eval_rougeL": 0.4045,
      "eval_rougeLsum": 0.4045,
      "eval_runtime": 1079.0919,
      "eval_samples_per_second": 5.097,
      "eval_steps_per_second": 0.319,
      "step": 3471
    },
    {
      "epoch": 5.04,
      "learning_rate": 1.3695965417867436e-05,
      "loss": 1.5798,
      "step": 3500
    },
    {
      "epoch": 5.76,
      "learning_rate": 1.2795389048991355e-05,
      "loss": 1.5586,
      "step": 4000
    },
    {
      "epoch": 6.0,
      "eval_f1": 0.9116,
      "eval_gen_len": 26.777818181818184,
      "eval_loss": 1.4880452156066895,
      "eval_precision": 0.9108,
      "eval_recall": 0.9127,
      "eval_rouge1": 0.4875,
      "eval_rouge2": 0.2373,
      "eval_rougeL": 0.4063,
      "eval_rougeLsum": 0.4063,
      "eval_runtime": 1027.5441,
      "eval_samples_per_second": 5.353,
      "eval_steps_per_second": 0.335,
      "step": 4164
    },
    {
      "epoch": 6.48,
      "learning_rate": 1.1894812680115276e-05,
      "loss": 1.5375,
      "step": 4500
    },
    {
      "epoch": 7.0,
      "eval_f1": 0.912,
      "eval_gen_len": 26.39909090909091,
      "eval_loss": 1.4768402576446533,
      "eval_precision": 0.9116,
      "eval_recall": 0.9128,
      "eval_rouge1": 0.4898,
      "eval_rouge2": 0.24,
      "eval_rougeL": 0.4083,
      "eval_rougeLsum": 0.4083,
      "eval_runtime": 922.1893,
      "eval_samples_per_second": 5.964,
      "eval_steps_per_second": 0.373,
      "step": 4858
    },
    {
      "epoch": 7.2,
      "learning_rate": 1.0994236311239194e-05,
      "loss": 1.5228,
      "step": 5000
    },
    {
      "epoch": 7.92,
      "learning_rate": 1.0093659942363115e-05,
      "loss": 1.5146,
      "step": 5500
    },
    {
      "epoch": 8.0,
      "eval_f1": 0.9126,
      "eval_gen_len": 26.156,
      "eval_loss": 1.4685654640197754,
      "eval_precision": 0.9123,
      "eval_recall": 0.9133,
      "eval_rouge1": 0.4907,
      "eval_rouge2": 0.241,
      "eval_rougeL": 0.4088,
      "eval_rougeLsum": 0.4089,
      "eval_runtime": 865.3485,
      "eval_samples_per_second": 6.356,
      "eval_steps_per_second": 0.398,
      "step": 5553
    },
    {
      "epoch": 8.64,
      "learning_rate": 9.193083573487034e-06,
      "loss": 1.5006,
      "step": 6000
    },
    {
      "epoch": 9.0,
      "eval_f1": 0.9127,
      "eval_gen_len": 26.26290909090909,
      "eval_loss": 1.4636152982711792,
      "eval_precision": 0.9122,
      "eval_recall": 0.9135,
      "eval_rouge1": 0.4914,
      "eval_rouge2": 0.2419,
      "eval_rougeL": 0.4097,
      "eval_rougeLsum": 0.4099,
      "eval_runtime": 874.612,
      "eval_samples_per_second": 6.289,
      "eval_steps_per_second": 0.393,
      "step": 6247
    },
    {
      "epoch": 9.36,
      "learning_rate": 8.29250720461095e-06,
      "loss": 1.49,
      "step": 6500
    },
    {
      "epoch": 10.0,
      "eval_f1": 0.9127,
      "eval_gen_len": 26.027272727272727,
      "eval_loss": 1.4580360651016235,
      "eval_precision": 0.9125,
      "eval_recall": 0.9133,
      "eval_rouge1": 0.4911,
      "eval_rouge2": 0.2429,
      "eval_rougeL": 0.4109,
      "eval_rougeLsum": 0.411,
      "eval_runtime": 855.8845,
      "eval_samples_per_second": 6.426,
      "eval_steps_per_second": 0.402,
      "step": 6942
    },
    {
      "epoch": 10.08,
      "learning_rate": 7.391930835734871e-06,
      "loss": 1.485,
      "step": 7000
    },
    {
      "epoch": 10.8,
      "learning_rate": 6.491354466858791e-06,
      "loss": 1.4749,
      "step": 7500
    },
    {
      "epoch": 11.0,
      "eval_f1": 0.9131,
      "eval_gen_len": 26.230363636363638,
      "eval_loss": 1.4546109437942505,
      "eval_precision": 0.9127,
      "eval_recall": 0.9138,
      "eval_rouge1": 0.4932,
      "eval_rouge2": 0.244,
      "eval_rougeL": 0.4121,
      "eval_rougeLsum": 0.4123,
      "eval_runtime": 871.4205,
      "eval_samples_per_second": 6.312,
      "eval_steps_per_second": 0.395,
      "step": 7636
    },
    {
      "epoch": 11.52,
      "learning_rate": 5.590778097982709e-06,
      "loss": 1.4661,
      "step": 8000
    },
    {
      "epoch": 12.0,
      "eval_f1": 0.9132,
      "eval_gen_len": 25.87781818181818,
      "eval_loss": 1.4514495134353638,
      "eval_precision": 0.9133,
      "eval_recall": 0.9136,
      "eval_rouge1": 0.4937,
      "eval_rouge2": 0.2448,
      "eval_rougeL": 0.4126,
      "eval_rougeLsum": 0.4127,
      "eval_runtime": 867.3574,
      "eval_samples_per_second": 6.341,
      "eval_steps_per_second": 0.397,
      "step": 8331
    },
    {
      "epoch": 12.24,
      "learning_rate": 4.690201729106629e-06,
      "loss": 1.4626,
      "step": 8500
    },
    {
      "epoch": 12.96,
      "learning_rate": 3.7896253602305477e-06,
      "loss": 1.4575,
      "step": 9000
    },
    {
      "epoch": 13.0,
      "eval_f1": 0.9133,
      "eval_gen_len": 26.11509090909091,
      "eval_loss": 1.4499082565307617,
      "eval_precision": 0.913,
      "eval_recall": 0.914,
      "eval_rouge1": 0.4947,
      "eval_rouge2": 0.2453,
      "eval_rougeL": 0.4139,
      "eval_rougeLsum": 0.414,
      "eval_runtime": 860.9844,
      "eval_samples_per_second": 6.388,
      "eval_steps_per_second": 0.4,
      "step": 9025
    },
    {
      "epoch": 13.68,
      "learning_rate": 2.8890489913544673e-06,
      "loss": 1.4511,
      "step": 9500
    },
    {
      "epoch": 14.0,
      "eval_f1": 0.9133,
      "eval_gen_len": 26.028727272727274,
      "eval_loss": 1.44780433177948,
      "eval_precision": 0.9131,
      "eval_recall": 0.9138,
      "eval_rouge1": 0.4939,
      "eval_rouge2": 0.2451,
      "eval_rougeL": 0.4133,
      "eval_rougeLsum": 0.4134,
      "eval_runtime": 862.0827,
      "eval_samples_per_second": 6.38,
      "eval_steps_per_second": 0.399,
      "step": 9720
    },
    {
      "epoch": 14.4,
      "learning_rate": 1.988472622478386e-06,
      "loss": 1.4519,
      "step": 10000
    },
    {
      "epoch": 15.0,
      "eval_f1": 0.9133,
      "eval_gen_len": 25.907818181818183,
      "eval_loss": 1.4471020698547363,
      "eval_precision": 0.9132,
      "eval_recall": 0.9137,
      "eval_rouge1": 0.4938,
      "eval_rouge2": 0.2451,
      "eval_rougeL": 0.4134,
      "eval_rougeLsum": 0.4134,
      "eval_runtime": 855.2673,
      "eval_samples_per_second": 6.431,
      "eval_steps_per_second": 0.402,
      "step": 10414
    },
    {
      "epoch": 15.12,
      "learning_rate": 1.0878962536023055e-06,
      "loss": 1.4475,
      "step": 10500
    },
    {
      "epoch": 15.84,
      "learning_rate": 1.8731988472622478e-07,
      "loss": 1.4439,
      "step": 11000
    },
    {
      "epoch": 15.99,
      "eval_f1": 0.9134,
      "eval_gen_len": 25.96290909090909,
      "eval_loss": 1.4468724727630615,
      "eval_precision": 0.9133,
      "eval_recall": 0.9138,
      "eval_rouge1": 0.4939,
      "eval_rouge2": 0.2453,
      "eval_rougeL": 0.4133,
      "eval_rougeLsum": 0.4134,
      "eval_runtime": 864.4194,
      "eval_samples_per_second": 6.363,
      "eval_steps_per_second": 0.398,
      "step": 11104
    },
    {
      "epoch": 15.99,
      "step": 11104,
      "total_flos": 2.2405705733792072e+18,
      "train_loss": 0.8767006197992594,
      "train_runtime": 37037.5595,
      "train_samples_per_second": 43.199,
      "train_steps_per_second": 0.3
    }
  ],
  "logging_steps": 500,
  "max_steps": 11104,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 16,
  "save_steps": 500,
  "total_flos": 2.2405705733792072e+18,
  "train_batch_size": 24,
  "trial_name": null,
  "trial_params": null
}