ryanmarten commited on
Commit
51f8d8f
·
verified ·
1 Parent(s): 9c18580

End of training

Browse files
Files changed (5) hide show
  1. README.md +2 -1
  2. all_results.json +6 -6
  3. train_results.json +6 -6
  4. trainer_state.json +209 -209
  5. training_loss.png +0 -0
README.md CHANGED
@@ -4,6 +4,7 @@ license: apache-2.0
4
  base_model: Qwen/Qwen2.5-7B-Instruct
5
  tags:
6
  - llama-factory
 
7
  - generated_from_trainer
8
  model-index:
9
  - name: s1
@@ -15,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  # s1
17
 
18
- This model is a fine-tuned version of [Qwen/Qwen2.5-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-7B-Instruct) on an unknown dataset.
19
 
20
  ## Model description
21
 
 
4
  base_model: Qwen/Qwen2.5-7B-Instruct
5
  tags:
6
  - llama-factory
7
+ - full
8
  - generated_from_trainer
9
  model-index:
10
  - name: s1
 
16
 
17
  # s1
18
 
19
+ This model is a fine-tuned version of [Qwen/Qwen2.5-7B-Instruct](https://huggingface.co/Qwen/Qwen2.5-7B-Instruct) on the mlfoundations-dev/s1 dataset.
20
 
21
  ## Model description
22
 
all_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "epoch": 6.768,
3
- "total_flos": 1.9932435535468954e+17,
4
- "train_loss": 0.9179985097476414,
5
- "train_runtime": 8939.9445,
6
- "train_samples_per_second": 0.783,
7
- "train_steps_per_second": 0.008
8
  }
 
1
  {
2
+ "epoch": 6.72,
3
+ "total_flos": 2.0848443398710886e+17,
4
+ "train_loss": 0.8195367446967534,
5
+ "train_runtime": 7873.6613,
6
+ "train_samples_per_second": 0.889,
7
+ "train_steps_per_second": 0.009
8
  }
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "epoch": 6.768,
3
- "total_flos": 1.9932435535468954e+17,
4
- "train_loss": 0.9179985097476414,
5
- "train_runtime": 8939.9445,
6
- "train_samples_per_second": 0.783,
7
- "train_steps_per_second": 0.008
8
  }
 
1
  {
2
+ "epoch": 6.72,
3
+ "total_flos": 2.0848443398710886e+17,
4
+ "train_loss": 0.8195367446967534,
5
+ "train_runtime": 7873.6613,
6
+ "train_samples_per_second": 0.889,
7
+ "train_steps_per_second": 0.009
8
  }
trainer_state.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
- "epoch": 6.768,
5
  "eval_steps": 500,
6
  "global_step": 70,
7
  "is_hyper_param_search": false,
@@ -10,505 +10,505 @@
10
  "log_history": [
11
  {
12
  "epoch": 0.096,
13
- "grad_norm": 7.4363104030057015,
14
  "learning_rate": 2.8571428571428573e-06,
15
- "loss": 1.2719,
16
  "step": 1
17
  },
18
  {
19
  "epoch": 0.192,
20
- "grad_norm": 7.250506606510911,
21
  "learning_rate": 5.7142857142857145e-06,
22
- "loss": 1.2273,
23
  "step": 2
24
  },
25
  {
26
  "epoch": 0.288,
27
- "grad_norm": 6.832063539613698,
28
  "learning_rate": 8.571428571428571e-06,
29
- "loss": 1.2782,
30
  "step": 3
31
  },
32
  {
33
  "epoch": 0.384,
34
- "grad_norm": 3.06741712124664,
35
  "learning_rate": 1.1428571428571429e-05,
36
- "loss": 1.1462,
37
  "step": 4
38
  },
39
  {
40
  "epoch": 0.48,
41
- "grad_norm": 4.505384780183221,
42
  "learning_rate": 1.4285714285714287e-05,
43
- "loss": 1.1577,
44
  "step": 5
45
  },
46
  {
47
  "epoch": 0.576,
48
- "grad_norm": 4.74796133360235,
49
  "learning_rate": 1.7142857142857142e-05,
50
- "loss": 1.1802,
51
  "step": 6
52
  },
53
  {
54
  "epoch": 0.672,
55
- "grad_norm": 5.275344422603644,
56
  "learning_rate": 2e-05,
57
- "loss": 1.1361,
58
  "step": 7
59
  },
60
  {
61
  "epoch": 0.768,
62
- "grad_norm": 3.6077121515672523,
63
  "learning_rate": 1.9987569212189224e-05,
64
- "loss": 1.0833,
65
  "step": 8
66
  },
67
  {
68
  "epoch": 0.864,
69
- "grad_norm": 2.3494976692411593,
70
  "learning_rate": 1.9950307753654016e-05,
71
- "loss": 1.0261,
72
  "step": 9
73
  },
74
  {
75
  "epoch": 0.96,
76
- "grad_norm": 2.0474852082338257,
77
  "learning_rate": 1.9888308262251286e-05,
78
- "loss": 1.019,
79
  "step": 10
80
  },
81
  {
82
- "epoch": 1.064,
83
- "grad_norm": 2.3101496294813555,
84
  "learning_rate": 1.9801724878485438e-05,
85
- "loss": 1.6489,
86
  "step": 11
87
  },
88
  {
89
- "epoch": 1.16,
90
- "grad_norm": 1.1632982074724751,
91
  "learning_rate": 1.969077286229078e-05,
92
- "loss": 1.0,
93
  "step": 12
94
  },
95
  {
96
- "epoch": 1.256,
97
- "grad_norm": 1.7306728680836048,
98
  "learning_rate": 1.955572805786141e-05,
99
- "loss": 0.9872,
100
  "step": 13
101
  },
102
  {
103
- "epoch": 1.3519999999999999,
104
- "grad_norm": 1.1781453231197772,
105
  "learning_rate": 1.9396926207859085e-05,
106
- "loss": 0.9577,
107
  "step": 14
108
  },
109
  {
110
- "epoch": 1.448,
111
- "grad_norm": 1.5370410163051034,
112
  "learning_rate": 1.921476211870408e-05,
113
- "loss": 0.9674,
114
  "step": 15
115
  },
116
  {
117
- "epoch": 1.544,
118
- "grad_norm": 1.3382455593337474,
119
  "learning_rate": 1.900968867902419e-05,
120
- "loss": 0.97,
121
  "step": 16
122
  },
123
  {
124
- "epoch": 1.6400000000000001,
125
- "grad_norm": 1.5556961280785029,
126
  "learning_rate": 1.8782215733702286e-05,
127
- "loss": 0.9541,
128
  "step": 17
129
  },
130
  {
131
- "epoch": 1.736,
132
- "grad_norm": 1.3147958958896104,
133
  "learning_rate": 1.8532908816321557e-05,
134
- "loss": 1.017,
135
  "step": 18
136
  },
137
  {
138
- "epoch": 1.8319999999999999,
139
- "grad_norm": 1.0697757940622286,
140
  "learning_rate": 1.826238774315995e-05,
141
- "loss": 0.8853,
142
  "step": 19
143
  },
144
  {
145
- "epoch": 1.928,
146
- "grad_norm": 1.2428076064637823,
147
  "learning_rate": 1.7971325072229227e-05,
148
- "loss": 0.9426,
149
  "step": 20
150
  },
151
  {
152
- "epoch": 2.032,
153
- "grad_norm": 1.5419571729367296,
154
  "learning_rate": 1.766044443118978e-05,
155
- "loss": 1.5667,
156
  "step": 21
157
  },
158
  {
159
- "epoch": 2.128,
160
- "grad_norm": 1.0330628917349387,
161
  "learning_rate": 1.7330518718298263e-05,
162
- "loss": 0.9308,
163
  "step": 22
164
  },
165
  {
166
- "epoch": 2.224,
167
- "grad_norm": 0.6738740135160787,
168
  "learning_rate": 1.698236818086073e-05,
169
- "loss": 0.8726,
170
  "step": 23
171
  },
172
  {
173
- "epoch": 2.32,
174
- "grad_norm": 0.851517903670043,
175
  "learning_rate": 1.6616858375968596e-05,
176
- "loss": 0.8916,
177
  "step": 24
178
  },
179
  {
180
- "epoch": 2.416,
181
- "grad_norm": 0.6658092812186402,
182
  "learning_rate": 1.6234898018587336e-05,
183
- "loss": 0.8827,
184
  "step": 25
185
  },
186
  {
187
- "epoch": 2.512,
188
- "grad_norm": 0.8302815653461936,
189
  "learning_rate": 1.5837436722347902e-05,
190
- "loss": 0.9593,
191
  "step": 26
192
  },
193
  {
194
- "epoch": 2.608,
195
- "grad_norm": 0.5805361472724696,
196
  "learning_rate": 1.5425462638657597e-05,
197
- "loss": 0.7634,
198
  "step": 27
199
  },
200
  {
201
- "epoch": 2.7039999999999997,
202
- "grad_norm": 0.7599957573097837,
203
  "learning_rate": 1.5000000000000002e-05,
204
- "loss": 0.9058,
205
  "step": 28
206
  },
207
  {
208
- "epoch": 2.8,
209
- "grad_norm": 0.6866513188504731,
210
  "learning_rate": 1.4562106573531632e-05,
211
- "loss": 0.8964,
212
  "step": 29
213
  },
214
  {
215
- "epoch": 2.896,
216
- "grad_norm": 0.9189397890158396,
217
  "learning_rate": 1.4112871031306118e-05,
218
- "loss": 0.8493,
219
  "step": 30
220
  },
221
  {
222
- "epoch": 2.992,
223
- "grad_norm": 1.0314524760639647,
224
  "learning_rate": 1.3653410243663953e-05,
225
- "loss": 1.4997,
226
  "step": 31
227
  },
228
  {
229
- "epoch": 3.096,
230
- "grad_norm": 0.8849240483310309,
231
  "learning_rate": 1.3184866502516846e-05,
232
- "loss": 0.8486,
233
  "step": 32
234
  },
235
  {
236
- "epoch": 3.192,
237
- "grad_norm": 0.8187560626368209,
238
  "learning_rate": 1.2708404681430054e-05,
239
- "loss": 0.8693,
240
  "step": 33
241
  },
242
  {
243
- "epoch": 3.288,
244
- "grad_norm": 0.6184119410179473,
245
  "learning_rate": 1.2225209339563144e-05,
246
- "loss": 0.8269,
247
  "step": 34
248
  },
249
  {
250
- "epoch": 3.384,
251
- "grad_norm": 0.6637146017943384,
252
  "learning_rate": 1.1736481776669307e-05,
253
- "loss": 0.7745,
254
  "step": 35
255
  },
256
  {
257
- "epoch": 3.48,
258
- "grad_norm": 0.5735188827410304,
259
  "learning_rate": 1.1243437046474854e-05,
260
- "loss": 0.8454,
261
  "step": 36
262
  },
263
  {
264
- "epoch": 3.576,
265
- "grad_norm": 0.5780493513130233,
266
  "learning_rate": 1.0747300935864245e-05,
267
- "loss": 0.7832,
268
  "step": 37
269
  },
270
  {
271
- "epoch": 3.672,
272
- "grad_norm": 0.5936174643312693,
273
  "learning_rate": 1.0249306917380731e-05,
274
- "loss": 0.8042,
275
  "step": 38
276
  },
277
  {
278
- "epoch": 3.768,
279
- "grad_norm": 0.5020247171477261,
280
  "learning_rate": 9.750693082619274e-06,
281
- "loss": 0.807,
282
  "step": 39
283
  },
284
  {
285
- "epoch": 3.864,
286
- "grad_norm": 0.546897890031181,
287
  "learning_rate": 9.252699064135759e-06,
288
- "loss": 0.8172,
289
  "step": 40
290
  },
291
  {
292
- "epoch": 3.96,
293
- "grad_norm": 0.5251209746881235,
294
  "learning_rate": 8.756562953525151e-06,
295
- "loss": 0.833,
296
  "step": 41
297
  },
298
  {
299
- "epoch": 4.064,
300
- "grad_norm": 0.8078880511356095,
301
  "learning_rate": 8.263518223330698e-06,
302
- "loss": 1.2994,
303
  "step": 42
304
  },
305
  {
306
- "epoch": 4.16,
307
- "grad_norm": 0.45809720361024925,
308
  "learning_rate": 7.774790660436857e-06,
309
- "loss": 0.79,
310
  "step": 43
311
  },
312
  {
313
- "epoch": 4.256,
314
- "grad_norm": 0.47948993795584427,
315
  "learning_rate": 7.291595318569951e-06,
316
- "loss": 0.7785,
317
  "step": 44
318
  },
319
  {
320
- "epoch": 4.352,
321
- "grad_norm": 0.4418844117866036,
322
  "learning_rate": 6.815133497483157e-06,
323
- "loss": 0.7446,
324
  "step": 45
325
  },
326
  {
327
- "epoch": 4.448,
328
- "grad_norm": 0.4379523378890173,
329
  "learning_rate": 6.34658975633605e-06,
330
- "loss": 0.7455,
331
  "step": 46
332
  },
333
  {
334
- "epoch": 4.5440000000000005,
335
- "grad_norm": 0.45226105944653855,
336
  "learning_rate": 5.887128968693887e-06,
337
- "loss": 0.7859,
338
  "step": 47
339
  },
340
  {
341
- "epoch": 4.64,
342
- "grad_norm": 0.42317389902034425,
343
  "learning_rate": 5.43789342646837e-06,
344
- "loss": 0.6996,
345
  "step": 48
346
  },
347
  {
348
- "epoch": 4.736,
349
- "grad_norm": 0.43884919881361373,
350
  "learning_rate": 5.000000000000003e-06,
351
- "loss": 0.8119,
352
  "step": 49
353
  },
354
  {
355
- "epoch": 4.832,
356
- "grad_norm": 0.38397369222110406,
357
  "learning_rate": 4.5745373613424075e-06,
358
- "loss": 0.7318,
359
  "step": 50
360
  },
361
  {
362
- "epoch": 4.928,
363
- "grad_norm": 0.4009575586053713,
364
  "learning_rate": 4.162563277652104e-06,
365
- "loss": 0.776,
366
  "step": 51
367
  },
368
  {
369
- "epoch": 5.032,
370
- "grad_norm": 0.7990275577512086,
371
  "learning_rate": 3.7651019814126656e-06,
372
- "loss": 1.283,
373
  "step": 52
374
  },
375
  {
376
- "epoch": 5.128,
377
- "grad_norm": 0.41089217358223823,
378
  "learning_rate": 3.3831416240314085e-06,
379
- "loss": 0.8399,
380
  "step": 53
381
  },
382
  {
383
- "epoch": 5.224,
384
- "grad_norm": 0.3698577830646195,
385
  "learning_rate": 3.017631819139273e-06,
386
- "loss": 0.5846,
387
  "step": 54
388
  },
389
  {
390
- "epoch": 5.32,
391
- "grad_norm": 0.47574350194370274,
392
  "learning_rate": 2.669481281701739e-06,
393
- "loss": 0.7821,
394
  "step": 55
395
  },
396
  {
397
- "epoch": 5.416,
398
- "grad_norm": 0.34266807007529126,
399
  "learning_rate": 2.339555568810221e-06,
400
- "loss": 0.7378,
401
  "step": 56
402
  },
403
  {
404
- "epoch": 5.5120000000000005,
405
- "grad_norm": 0.33634280269178407,
406
  "learning_rate": 2.0286749277707783e-06,
407
- "loss": 0.7592,
408
  "step": 57
409
  },
410
  {
411
- "epoch": 5.608,
412
- "grad_norm": 0.3345979756869023,
413
  "learning_rate": 1.7376122568400533e-06,
414
- "loss": 0.7171,
415
  "step": 58
416
  },
417
  {
418
- "epoch": 5.704,
419
- "grad_norm": 0.35973384323335894,
420
  "learning_rate": 1.467091183678444e-06,
421
- "loss": 0.7126,
422
  "step": 59
423
  },
424
  {
425
- "epoch": 5.8,
426
- "grad_norm": 0.34513887529701814,
427
  "learning_rate": 1.2177842662977136e-06,
428
- "loss": 0.7747,
429
  "step": 60
430
  },
431
  {
432
- "epoch": 5.896,
433
- "grad_norm": 0.35888960062599257,
434
  "learning_rate": 9.903113209758098e-07,
435
- "loss": 0.7731,
436
  "step": 61
437
  },
438
  {
439
- "epoch": 5.992,
440
- "grad_norm": 0.5581950902164758,
441
  "learning_rate": 7.852378812959227e-07,
442
- "loss": 1.1821,
443
  "step": 62
444
  },
445
  {
446
- "epoch": 6.096,
447
- "grad_norm": 0.31817763350335815,
448
  "learning_rate": 6.030737921409169e-07,
449
- "loss": 0.7462,
450
  "step": 63
451
  },
452
  {
453
- "epoch": 6.192,
454
- "grad_norm": 0.2938154468380693,
455
  "learning_rate": 4.4427194213859216e-07,
456
- "loss": 0.7164,
457
  "step": 64
458
  },
459
  {
460
- "epoch": 6.288,
461
- "grad_norm": 0.29565655135478536,
462
  "learning_rate": 3.0922713770922155e-07,
463
- "loss": 0.715,
464
  "step": 65
465
  },
466
  {
467
- "epoch": 6.384,
468
- "grad_norm": 0.29842209686367416,
469
  "learning_rate": 1.9827512151456175e-07,
470
- "loss": 0.7515,
471
  "step": 66
472
  },
473
  {
474
- "epoch": 6.48,
475
- "grad_norm": 0.283354724873026,
476
  "learning_rate": 1.1169173774871478e-07,
477
- "loss": 0.7416,
478
  "step": 67
479
  },
480
  {
481
- "epoch": 6.576,
482
- "grad_norm": 0.2785218564599255,
483
  "learning_rate": 4.9692246345985905e-08,
484
- "loss": 0.7276,
485
  "step": 68
486
  },
487
  {
488
- "epoch": 6.672,
489
- "grad_norm": 0.29935887630355656,
490
  "learning_rate": 1.2430787810776556e-08,
491
- "loss": 0.7376,
492
  "step": 69
493
  },
494
  {
495
- "epoch": 6.768,
496
- "grad_norm": 0.2945945618574963,
497
  "learning_rate": 0.0,
498
- "loss": 0.7305,
499
  "step": 70
500
  },
501
  {
502
- "epoch": 6.768,
503
  "step": 70,
504
- "total_flos": 1.9932435535468954e+17,
505
- "train_loss": 0.9179985097476414,
506
- "train_runtime": 8939.9445,
507
- "train_samples_per_second": 0.783,
508
- "train_steps_per_second": 0.008
509
  }
510
  ],
511
- "logging_steps": 1.0,
512
  "max_steps": 70,
513
  "num_input_tokens_seen": 0,
514
  "num_train_epochs": 7,
@@ -525,7 +525,7 @@
525
  "attributes": {}
526
  }
527
  },
528
- "total_flos": 1.9932435535468954e+17,
529
  "train_batch_size": 1,
530
  "trial_name": null,
531
  "trial_params": null
 
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
+ "epoch": 6.72,
5
  "eval_steps": 500,
6
  "global_step": 70,
7
  "is_hyper_param_search": false,
 
10
  "log_history": [
11
  {
12
  "epoch": 0.096,
13
+ "grad_norm": 7.183526298197059,
14
  "learning_rate": 2.8571428571428573e-06,
15
+ "loss": 1.2159,
16
  "step": 1
17
  },
18
  {
19
  "epoch": 0.192,
20
+ "grad_norm": 7.031646024711875,
21
  "learning_rate": 5.7142857142857145e-06,
22
+ "loss": 1.1803,
23
  "step": 2
24
  },
25
  {
26
  "epoch": 0.288,
27
+ "grad_norm": 6.726449054019057,
28
  "learning_rate": 8.571428571428571e-06,
29
+ "loss": 1.2637,
30
  "step": 3
31
  },
32
  {
33
  "epoch": 0.384,
34
+ "grad_norm": 3.019251490604518,
35
  "learning_rate": 1.1428571428571429e-05,
36
+ "loss": 1.0997,
37
  "step": 4
38
  },
39
  {
40
  "epoch": 0.48,
41
+ "grad_norm": 4.372464279714231,
42
  "learning_rate": 1.4285714285714287e-05,
43
+ "loss": 1.134,
44
  "step": 5
45
  },
46
  {
47
  "epoch": 0.576,
48
+ "grad_norm": 4.668877413562089,
49
  "learning_rate": 1.7142857142857142e-05,
50
+ "loss": 1.1521,
51
  "step": 6
52
  },
53
  {
54
  "epoch": 0.672,
55
+ "grad_norm": 5.315985360539466,
56
  "learning_rate": 2e-05,
57
+ "loss": 1.0869,
58
  "step": 7
59
  },
60
  {
61
  "epoch": 0.768,
62
+ "grad_norm": 3.531866863804273,
63
  "learning_rate": 1.9987569212189224e-05,
64
+ "loss": 1.0384,
65
  "step": 8
66
  },
67
  {
68
  "epoch": 0.864,
69
+ "grad_norm": 2.360792496559264,
70
  "learning_rate": 1.9950307753654016e-05,
71
+ "loss": 0.9982,
72
  "step": 9
73
  },
74
  {
75
  "epoch": 0.96,
76
+ "grad_norm": 2.0730886631612866,
77
  "learning_rate": 1.9888308262251286e-05,
78
+ "loss": 0.9549,
79
  "step": 10
80
  },
81
  {
82
+ "epoch": 1.056,
83
+ "grad_norm": 2.0949143881823304,
84
  "learning_rate": 1.9801724878485438e-05,
85
+ "loss": 0.9849,
86
  "step": 11
87
  },
88
  {
89
+ "epoch": 1.152,
90
+ "grad_norm": 1.4418707389397343,
91
  "learning_rate": 1.969077286229078e-05,
92
+ "loss": 0.9482,
93
  "step": 12
94
  },
95
  {
96
+ "epoch": 1.248,
97
+ "grad_norm": 1.4176378835481556,
98
  "learning_rate": 1.955572805786141e-05,
99
+ "loss": 0.9402,
100
  "step": 13
101
  },
102
  {
103
+ "epoch": 1.3439999999999999,
104
+ "grad_norm": 1.213085102759749,
105
  "learning_rate": 1.9396926207859085e-05,
106
+ "loss": 0.9275,
107
  "step": 14
108
  },
109
  {
110
+ "epoch": 1.44,
111
+ "grad_norm": 1.20171438446904,
112
  "learning_rate": 1.921476211870408e-05,
113
+ "loss": 0.9148,
114
  "step": 15
115
  },
116
  {
117
+ "epoch": 1.536,
118
+ "grad_norm": 1.1294101542052173,
119
  "learning_rate": 1.900968867902419e-05,
120
+ "loss": 0.9246,
121
  "step": 16
122
  },
123
  {
124
+ "epoch": 1.6320000000000001,
125
+ "grad_norm": 1.0994307104874157,
126
  "learning_rate": 1.8782215733702286e-05,
127
+ "loss": 0.9171,
128
  "step": 17
129
  },
130
  {
131
+ "epoch": 1.728,
132
+ "grad_norm": 1.020464421286651,
133
  "learning_rate": 1.8532908816321557e-05,
134
+ "loss": 0.9417,
135
  "step": 18
136
  },
137
  {
138
+ "epoch": 1.8239999999999998,
139
+ "grad_norm": 0.7681952849638046,
140
  "learning_rate": 1.826238774315995e-05,
141
+ "loss": 0.8709,
142
  "step": 19
143
  },
144
  {
145
+ "epoch": 1.92,
146
+ "grad_norm": 0.9490121535512666,
147
  "learning_rate": 1.7971325072229227e-05,
148
+ "loss": 0.9026,
149
  "step": 20
150
  },
151
  {
152
+ "epoch": 2.016,
153
+ "grad_norm": 0.6572000695564753,
154
  "learning_rate": 1.766044443118978e-05,
155
+ "loss": 0.8912,
156
  "step": 21
157
  },
158
  {
159
+ "epoch": 2.112,
160
+ "grad_norm": 0.7890941812184483,
161
  "learning_rate": 1.7330518718298263e-05,
162
+ "loss": 0.8605,
163
  "step": 22
164
  },
165
  {
166
+ "epoch": 2.208,
167
+ "grad_norm": 0.645276849098759,
168
  "learning_rate": 1.698236818086073e-05,
169
+ "loss": 0.833,
170
  "step": 23
171
  },
172
  {
173
+ "epoch": 2.304,
174
+ "grad_norm": 0.6944310244083031,
175
  "learning_rate": 1.6616858375968596e-05,
176
+ "loss": 0.8128,
177
  "step": 24
178
  },
179
  {
180
+ "epoch": 2.4,
181
+ "grad_norm": 0.5769863100807223,
182
  "learning_rate": 1.6234898018587336e-05,
183
+ "loss": 0.8633,
184
  "step": 25
185
  },
186
  {
187
+ "epoch": 2.496,
188
+ "grad_norm": 0.7467973758778769,
189
  "learning_rate": 1.5837436722347902e-05,
190
+ "loss": 0.8878,
191
  "step": 26
192
  },
193
  {
194
+ "epoch": 2.592,
195
+ "grad_norm": 0.7020191157406843,
196
  "learning_rate": 1.5425462638657597e-05,
197
+ "loss": 0.7629,
198
  "step": 27
199
  },
200
  {
201
+ "epoch": 2.6879999999999997,
202
+ "grad_norm": 0.7102558998323939,
203
  "learning_rate": 1.5000000000000002e-05,
204
+ "loss": 0.8378,
205
  "step": 28
206
  },
207
  {
208
+ "epoch": 2.784,
209
+ "grad_norm": 1.5903597029550902,
210
  "learning_rate": 1.4562106573531632e-05,
211
+ "loss": 0.795,
212
  "step": 29
213
  },
214
  {
215
+ "epoch": 2.88,
216
+ "grad_norm": 1.2308988723660046,
217
  "learning_rate": 1.4112871031306118e-05,
218
+ "loss": 0.8061,
219
  "step": 30
220
  },
221
  {
222
+ "epoch": 2.976,
223
+ "grad_norm": 1.8813596533220647,
224
  "learning_rate": 1.3653410243663953e-05,
225
+ "loss": 0.8583,
226
  "step": 31
227
  },
228
  {
229
+ "epoch": 3.072,
230
+ "grad_norm": 0.8623655173556376,
231
  "learning_rate": 1.3184866502516846e-05,
232
+ "loss": 0.7933,
233
  "step": 32
234
  },
235
  {
236
+ "epoch": 3.168,
237
+ "grad_norm": 276.41503463731976,
238
  "learning_rate": 1.2708404681430054e-05,
239
+ "loss": 0.8317,
240
  "step": 33
241
  },
242
  {
243
+ "epoch": 3.2640000000000002,
244
+ "grad_norm": 0.6904839668045957,
245
  "learning_rate": 1.2225209339563144e-05,
246
+ "loss": 0.764,
247
  "step": 34
248
  },
249
  {
250
+ "epoch": 3.36,
251
+ "grad_norm": 0.5664303765415496,
252
  "learning_rate": 1.1736481776669307e-05,
253
+ "loss": 0.7052,
254
  "step": 35
255
  },
256
  {
257
+ "epoch": 3.456,
258
+ "grad_norm": 0.5220759676992732,
259
  "learning_rate": 1.1243437046474854e-05,
260
+ "loss": 0.7776,
261
  "step": 36
262
  },
263
  {
264
+ "epoch": 3.552,
265
+ "grad_norm": 0.48786029357197297,
266
  "learning_rate": 1.0747300935864245e-05,
267
+ "loss": 0.7307,
268
  "step": 37
269
  },
270
  {
271
+ "epoch": 3.648,
272
+ "grad_norm": 0.637056641485944,
273
  "learning_rate": 1.0249306917380731e-05,
274
+ "loss": 0.7632,
275
  "step": 38
276
  },
277
  {
278
+ "epoch": 3.7439999999999998,
279
+ "grad_norm": 0.5274190066086378,
280
  "learning_rate": 9.750693082619274e-06,
281
+ "loss": 0.7752,
282
  "step": 39
283
  },
284
  {
285
+ "epoch": 3.84,
286
+ "grad_norm": 0.4742218489863218,
287
  "learning_rate": 9.252699064135759e-06,
288
+ "loss": 0.7477,
289
  "step": 40
290
  },
291
  {
292
+ "epoch": 3.936,
293
+ "grad_norm": 0.4906516507530535,
294
  "learning_rate": 8.756562953525151e-06,
295
+ "loss": 0.7471,
296
  "step": 41
297
  },
298
  {
299
+ "epoch": 4.032,
300
+ "grad_norm": 0.42574380900812253,
301
  "learning_rate": 8.263518223330698e-06,
302
+ "loss": 0.7555,
303
  "step": 42
304
  },
305
  {
306
+ "epoch": 4.128,
307
+ "grad_norm": 0.4612817312885407,
308
  "learning_rate": 7.774790660436857e-06,
309
+ "loss": 0.7426,
310
  "step": 43
311
  },
312
  {
313
+ "epoch": 4.224,
314
+ "grad_norm": 0.4048890809208518,
315
  "learning_rate": 7.291595318569951e-06,
316
+ "loss": 0.7229,
317
  "step": 44
318
  },
319
  {
320
+ "epoch": 4.32,
321
+ "grad_norm": 0.5950686810700512,
322
  "learning_rate": 6.815133497483157e-06,
323
+ "loss": 0.7084,
324
  "step": 45
325
  },
326
  {
327
+ "epoch": 4.416,
328
+ "grad_norm": 0.4083958557164403,
329
  "learning_rate": 6.34658975633605e-06,
330
+ "loss": 0.7176,
331
  "step": 46
332
  },
333
  {
334
+ "epoch": 4.5120000000000005,
335
+ "grad_norm": 0.3889024722667518,
336
  "learning_rate": 5.887128968693887e-06,
337
+ "loss": 0.6971,
338
  "step": 47
339
  },
340
  {
341
+ "epoch": 4.608,
342
+ "grad_norm": 0.38627888537778465,
343
  "learning_rate": 5.43789342646837e-06,
344
+ "loss": 0.6593,
345
  "step": 48
346
  },
347
  {
348
+ "epoch": 4.704,
349
+ "grad_norm": 0.41107376175312915,
350
  "learning_rate": 5.000000000000003e-06,
351
+ "loss": 0.7331,
352
  "step": 49
353
  },
354
  {
355
+ "epoch": 4.8,
356
+ "grad_norm": 0.368851573498416,
357
  "learning_rate": 4.5745373613424075e-06,
358
+ "loss": 0.6982,
359
  "step": 50
360
  },
361
  {
362
+ "epoch": 4.896,
363
+ "grad_norm": 0.3655873384272704,
364
  "learning_rate": 4.162563277652104e-06,
365
+ "loss": 0.7143,
366
  "step": 51
367
  },
368
  {
369
+ "epoch": 4.992,
370
+ "grad_norm": 0.36698174251614346,
371
  "learning_rate": 3.7651019814126656e-06,
372
+ "loss": 0.7085,
373
  "step": 52
374
  },
375
  {
376
+ "epoch": 5.088,
377
+ "grad_norm": 0.3565085992843195,
378
  "learning_rate": 3.3831416240314085e-06,
379
+ "loss": 0.7289,
380
  "step": 53
381
  },
382
  {
383
+ "epoch": 5.184,
384
+ "grad_norm": 0.33850857151098473,
385
  "learning_rate": 3.017631819139273e-06,
386
+ "loss": 0.677,
387
  "step": 54
388
  },
389
  {
390
+ "epoch": 5.28,
391
+ "grad_norm": 0.34277177833492384,
392
  "learning_rate": 2.669481281701739e-06,
393
+ "loss": 0.6235,
394
  "step": 55
395
  },
396
  {
397
+ "epoch": 5.376,
398
+ "grad_norm": 0.35409138917491984,
399
  "learning_rate": 2.339555568810221e-06,
400
+ "loss": 0.6884,
401
  "step": 56
402
  },
403
  {
404
+ "epoch": 5.4719999999999995,
405
+ "grad_norm": 0.3835765040879716,
406
  "learning_rate": 2.0286749277707783e-06,
407
+ "loss": 0.6943,
408
  "step": 57
409
  },
410
  {
411
+ "epoch": 5.568,
412
+ "grad_norm": 0.3116079973081498,
413
  "learning_rate": 1.7376122568400533e-06,
414
+ "loss": 0.6462,
415
  "step": 58
416
  },
417
  {
418
+ "epoch": 5.664,
419
+ "grad_norm": 0.31059743253395206,
420
  "learning_rate": 1.467091183678444e-06,
421
+ "loss": 0.6899,
422
  "step": 59
423
  },
424
  {
425
+ "epoch": 5.76,
426
+ "grad_norm": 0.3333135479564294,
427
  "learning_rate": 1.2177842662977136e-06,
428
+ "loss": 0.6851,
429
  "step": 60
430
  },
431
  {
432
+ "epoch": 5.856,
433
+ "grad_norm": 0.3545171656487777,
434
  "learning_rate": 9.903113209758098e-07,
435
+ "loss": 0.7027,
436
  "step": 61
437
  },
438
  {
439
+ "epoch": 5.952,
440
+ "grad_norm": 0.34270948410594704,
441
  "learning_rate": 7.852378812959227e-07,
442
+ "loss": 0.6706,
443
  "step": 62
444
  },
445
  {
446
+ "epoch": 6.048,
447
+ "grad_norm": 0.313217720743749,
448
  "learning_rate": 6.030737921409169e-07,
449
+ "loss": 0.693,
450
  "step": 63
451
  },
452
  {
453
+ "epoch": 6.144,
454
+ "grad_norm": 0.31090294988670286,
455
  "learning_rate": 4.4427194213859216e-07,
456
+ "loss": 0.6914,
457
  "step": 64
458
  },
459
  {
460
+ "epoch": 6.24,
461
+ "grad_norm": 0.3052177834188217,
462
  "learning_rate": 3.0922713770922155e-07,
463
+ "loss": 0.6366,
464
  "step": 65
465
  },
466
  {
467
+ "epoch": 6.336,
468
+ "grad_norm": 0.2906275385092298,
469
  "learning_rate": 1.9827512151456175e-07,
470
+ "loss": 0.6899,
471
  "step": 66
472
  },
473
  {
474
+ "epoch": 6.432,
475
+ "grad_norm": 0.2881426610713051,
476
  "learning_rate": 1.1169173774871478e-07,
477
+ "loss": 0.6732,
478
  "step": 67
479
  },
480
  {
481
+ "epoch": 6.5280000000000005,
482
+ "grad_norm": 0.2736825814044455,
483
  "learning_rate": 4.9692246345985905e-08,
484
+ "loss": 0.7035,
485
  "step": 68
486
  },
487
  {
488
+ "epoch": 6.624,
489
+ "grad_norm": 0.2908985031551779,
490
  "learning_rate": 1.2430787810776556e-08,
491
+ "loss": 0.6699,
492
  "step": 69
493
  },
494
  {
495
+ "epoch": 6.72,
496
+ "grad_norm": 0.2973945837536026,
497
  "learning_rate": 0.0,
498
+ "loss": 0.7055,
499
  "step": 70
500
  },
501
  {
502
+ "epoch": 6.72,
503
  "step": 70,
504
+ "total_flos": 2.0848443398710886e+17,
505
+ "train_loss": 0.8195367446967534,
506
+ "train_runtime": 7873.6613,
507
+ "train_samples_per_second": 0.889,
508
+ "train_steps_per_second": 0.009
509
  }
510
  ],
511
+ "logging_steps": 1,
512
  "max_steps": 70,
513
  "num_input_tokens_seen": 0,
514
  "num_train_epochs": 7,
 
525
  "attributes": {}
526
  }
527
  },
528
+ "total_flos": 2.0848443398710886e+17,
529
  "train_batch_size": 1,
530
  "trial_name": null,
531
  "trial_params": null
training_loss.png CHANGED