File size: 10,101 Bytes
ccd244e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 1.8808777429467085,
  "eval_steps": 50,
  "global_step": 1200,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.07836990595611286,
      "grad_norm": 0.9672619104385376,
      "learning_rate": 2.413127413127413e-05,
      "loss": 0.5333,
      "step": 50
    },
    {
      "epoch": 0.07836990595611286,
      "eval_loss": 0.21812455356121063,
      "eval_runtime": 136.5294,
      "eval_samples_per_second": 5.23,
      "eval_steps_per_second": 0.659,
      "step": 50
    },
    {
      "epoch": 0.15673981191222572,
      "grad_norm": 0.8239838480949402,
      "learning_rate": 2.3166023166023168e-05,
      "loss": 0.1948,
      "step": 100
    },
    {
      "epoch": 0.15673981191222572,
      "eval_loss": 0.180569127202034,
      "eval_runtime": 136.8617,
      "eval_samples_per_second": 5.217,
      "eval_steps_per_second": 0.658,
      "step": 100
    },
    {
      "epoch": 0.23510971786833856,
      "grad_norm": 0.4985473155975342,
      "learning_rate": 2.2200772200772202e-05,
      "loss": 0.1772,
      "step": 150
    },
    {
      "epoch": 0.23510971786833856,
      "eval_loss": 0.17201179265975952,
      "eval_runtime": 136.9237,
      "eval_samples_per_second": 5.215,
      "eval_steps_per_second": 0.657,
      "step": 150
    },
    {
      "epoch": 0.31347962382445144,
      "grad_norm": 0.5671184659004211,
      "learning_rate": 2.1235521235521236e-05,
      "loss": 0.1674,
      "step": 200
    },
    {
      "epoch": 0.31347962382445144,
      "eval_loss": 0.16670115292072296,
      "eval_runtime": 137.1006,
      "eval_samples_per_second": 5.208,
      "eval_steps_per_second": 0.656,
      "step": 200
    },
    {
      "epoch": 0.39184952978056425,
      "grad_norm": 0.5542048215866089,
      "learning_rate": 2.0270270270270273e-05,
      "loss": 0.1639,
      "step": 250
    },
    {
      "epoch": 0.39184952978056425,
      "eval_loss": 0.16387563943862915,
      "eval_runtime": 136.9694,
      "eval_samples_per_second": 5.213,
      "eval_steps_per_second": 0.657,
      "step": 250
    },
    {
      "epoch": 0.4702194357366771,
      "grad_norm": 0.5355677604675293,
      "learning_rate": 1.9305019305019306e-05,
      "loss": 0.1603,
      "step": 300
    },
    {
      "epoch": 0.4702194357366771,
      "eval_loss": 0.16061373054981232,
      "eval_runtime": 136.8175,
      "eval_samples_per_second": 5.219,
      "eval_steps_per_second": 0.658,
      "step": 300
    },
    {
      "epoch": 0.54858934169279,
      "grad_norm": 0.648248553276062,
      "learning_rate": 1.833976833976834e-05,
      "loss": 0.1563,
      "step": 350
    },
    {
      "epoch": 0.54858934169279,
      "eval_loss": 0.1591891348361969,
      "eval_runtime": 136.7712,
      "eval_samples_per_second": 5.22,
      "eval_steps_per_second": 0.658,
      "step": 350
    },
    {
      "epoch": 0.6269592476489029,
      "grad_norm": 0.4232325851917267,
      "learning_rate": 1.7374517374517377e-05,
      "loss": 0.1595,
      "step": 400
    },
    {
      "epoch": 0.6269592476489029,
      "eval_loss": 0.15778718888759613,
      "eval_runtime": 137.2186,
      "eval_samples_per_second": 5.203,
      "eval_steps_per_second": 0.656,
      "step": 400
    },
    {
      "epoch": 0.7053291536050157,
      "grad_norm": 0.4626815915107727,
      "learning_rate": 1.640926640926641e-05,
      "loss": 0.1508,
      "step": 450
    },
    {
      "epoch": 0.7053291536050157,
      "eval_loss": 0.1561730057001114,
      "eval_runtime": 136.858,
      "eval_samples_per_second": 5.217,
      "eval_steps_per_second": 0.658,
      "step": 450
    },
    {
      "epoch": 0.7836990595611285,
      "grad_norm": 0.4221360385417938,
      "learning_rate": 1.5444015444015444e-05,
      "loss": 0.1581,
      "step": 500
    },
    {
      "epoch": 0.7836990595611285,
      "eval_loss": 0.15424229204654694,
      "eval_runtime": 137.1103,
      "eval_samples_per_second": 5.207,
      "eval_steps_per_second": 0.656,
      "step": 500
    },
    {
      "epoch": 0.8620689655172413,
      "grad_norm": 0.3882145583629608,
      "learning_rate": 1.4478764478764478e-05,
      "loss": 0.1441,
      "step": 550
    },
    {
      "epoch": 0.8620689655172413,
      "eval_loss": 0.15357084572315216,
      "eval_runtime": 136.9739,
      "eval_samples_per_second": 5.213,
      "eval_steps_per_second": 0.657,
      "step": 550
    },
    {
      "epoch": 0.9404388714733543,
      "grad_norm": 0.4727325141429901,
      "learning_rate": 1.3513513513513515e-05,
      "loss": 0.1487,
      "step": 600
    },
    {
      "epoch": 0.9404388714733543,
      "eval_loss": 0.15231835842132568,
      "eval_runtime": 136.9176,
      "eval_samples_per_second": 5.215,
      "eval_steps_per_second": 0.657,
      "step": 600
    },
    {
      "epoch": 1.0188087774294672,
      "grad_norm": 0.4438364803791046,
      "learning_rate": 1.2548262548262549e-05,
      "loss": 0.14,
      "step": 650
    },
    {
      "epoch": 1.0188087774294672,
      "eval_loss": 0.15219952166080475,
      "eval_runtime": 137.0305,
      "eval_samples_per_second": 5.211,
      "eval_steps_per_second": 0.657,
      "step": 650
    },
    {
      "epoch": 1.09717868338558,
      "grad_norm": 0.5468264818191528,
      "learning_rate": 1.1583011583011584e-05,
      "loss": 0.1366,
      "step": 700
    },
    {
      "epoch": 1.09717868338558,
      "eval_loss": 0.1532224416732788,
      "eval_runtime": 137.4744,
      "eval_samples_per_second": 5.194,
      "eval_steps_per_second": 0.655,
      "step": 700
    },
    {
      "epoch": 1.1755485893416928,
      "grad_norm": 0.609539270401001,
      "learning_rate": 1.0617760617760618e-05,
      "loss": 0.1352,
      "step": 750
    },
    {
      "epoch": 1.1755485893416928,
      "eval_loss": 0.15247634053230286,
      "eval_runtime": 137.6071,
      "eval_samples_per_second": 5.189,
      "eval_steps_per_second": 0.654,
      "step": 750
    },
    {
      "epoch": 1.2539184952978055,
      "grad_norm": 0.5422230958938599,
      "learning_rate": 9.652509652509653e-06,
      "loss": 0.1278,
      "step": 800
    },
    {
      "epoch": 1.2539184952978055,
      "eval_loss": 0.15182016789913177,
      "eval_runtime": 137.443,
      "eval_samples_per_second": 5.195,
      "eval_steps_per_second": 0.655,
      "step": 800
    },
    {
      "epoch": 1.3322884012539185,
      "grad_norm": 0.574143648147583,
      "learning_rate": 8.687258687258689e-06,
      "loss": 0.1315,
      "step": 850
    },
    {
      "epoch": 1.3322884012539185,
      "eval_loss": 0.15190598368644714,
      "eval_runtime": 137.5049,
      "eval_samples_per_second": 5.193,
      "eval_steps_per_second": 0.655,
      "step": 850
    },
    {
      "epoch": 1.4106583072100314,
      "grad_norm": 0.5425278544425964,
      "learning_rate": 7.722007722007722e-06,
      "loss": 0.1274,
      "step": 900
    },
    {
      "epoch": 1.4106583072100314,
      "eval_loss": 0.15125001966953278,
      "eval_runtime": 137.6226,
      "eval_samples_per_second": 5.188,
      "eval_steps_per_second": 0.654,
      "step": 900
    },
    {
      "epoch": 1.489028213166144,
      "grad_norm": 0.48568734526634216,
      "learning_rate": 6.7567567567567575e-06,
      "loss": 0.1347,
      "step": 950
    },
    {
      "epoch": 1.489028213166144,
      "eval_loss": 0.15007726848125458,
      "eval_runtime": 137.6801,
      "eval_samples_per_second": 5.186,
      "eval_steps_per_second": 0.654,
      "step": 950
    },
    {
      "epoch": 1.567398119122257,
      "grad_norm": 0.5619414448738098,
      "learning_rate": 5.791505791505792e-06,
      "loss": 0.1332,
      "step": 1000
    },
    {
      "epoch": 1.567398119122257,
      "eval_loss": 0.15013186633586884,
      "eval_runtime": 137.5165,
      "eval_samples_per_second": 5.192,
      "eval_steps_per_second": 0.654,
      "step": 1000
    },
    {
      "epoch": 1.64576802507837,
      "grad_norm": 0.5540875196456909,
      "learning_rate": 4.8262548262548266e-06,
      "loss": 0.1263,
      "step": 1050
    },
    {
      "epoch": 1.64576802507837,
      "eval_loss": 0.14901334047317505,
      "eval_runtime": 137.5107,
      "eval_samples_per_second": 5.192,
      "eval_steps_per_second": 0.654,
      "step": 1050
    },
    {
      "epoch": 1.7241379310344827,
      "grad_norm": 0.5203628540039062,
      "learning_rate": 3.861003861003861e-06,
      "loss": 0.1291,
      "step": 1100
    },
    {
      "epoch": 1.7241379310344827,
      "eval_loss": 0.14880172908306122,
      "eval_runtime": 137.6519,
      "eval_samples_per_second": 5.187,
      "eval_steps_per_second": 0.654,
      "step": 1100
    },
    {
      "epoch": 1.8025078369905956,
      "grad_norm": 0.5150081515312195,
      "learning_rate": 2.895752895752896e-06,
      "loss": 0.1287,
      "step": 1150
    },
    {
      "epoch": 1.8025078369905956,
      "eval_loss": 0.14855234324932098,
      "eval_runtime": 137.7759,
      "eval_samples_per_second": 5.182,
      "eval_steps_per_second": 0.653,
      "step": 1150
    },
    {
      "epoch": 1.8808777429467085,
      "grad_norm": 0.5428478717803955,
      "learning_rate": 1.9305019305019305e-06,
      "loss": 0.1278,
      "step": 1200
    },
    {
      "epoch": 1.8808777429467085,
      "eval_loss": 0.14838644862174988,
      "eval_runtime": 137.7999,
      "eval_samples_per_second": 5.181,
      "eval_steps_per_second": 0.653,
      "step": 1200
    }
  ],
  "logging_steps": 50,
  "max_steps": 1300,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 50,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 2.1478967485661184e+17,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}