yyx123 commited on
Commit
3a7e084
1 Parent(s): 123fb42

Model save

Browse files
Files changed (5) hide show
  1. README.md +22 -25
  2. all_results.json +8 -8
  3. eval_results.json +4 -4
  4. train_results.json +4 -4
  5. trainer_state.json +164 -151
README.md CHANGED
@@ -2,13 +2,9 @@
2
  license: other
3
  library_name: peft
4
  tags:
5
- - alignment-handbook
6
- - generated_from_trainer
7
  - trl
8
  - sft
9
  - generated_from_trainer
10
- datasets:
11
- - ruozhiba
12
  base_model: 01-ai/Yi-6B
13
  model-index:
14
  - name: Yi-6B-ruozhiba
@@ -20,9 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  # Yi-6B-ruozhiba
22
 
23
- This model is a fine-tuned version of [01-ai/Yi-6B](https://huggingface.co/01-ai/Yi-6B) on the ruozhiba dataset.
24
  It achieves the following results on the evaluation set:
25
- - Loss: 4.2568
26
 
27
  ## Model description
28
 
@@ -54,25 +50,26 @@ The following hyperparameters were used during training:
54
 
55
  | Training Loss | Epoch | Step | Validation Loss |
56
  |:-------------:|:-----:|:----:|:---------------:|
57
- | 2.7616 | 1.0 | 55 | 2.3938 |
58
- | 2.0613 | 2.0 | 110 | 1.9861 |
59
- | 1.8236 | 3.0 | 165 | 1.9593 |
60
- | 1.2817 | 5.0 | 275 | 2.2494 |
61
- | 0.9519 | 6.0 | 330 | 2.5876 |
62
- | 0.6733 | 7.0 | 385 | 3.0864 |
63
- | 0.4873 | 8.0 | 440 | 3.1206 |
64
- | 0.3163 | 9.0 | 495 | 3.3577 |
65
- | 0.2422 | 10.0 | 550 | 3.5398 |
66
- | 0.1929 | 11.0 | 605 | 3.7080 |
67
- | 0.1429 | 12.0 | 660 | 3.8363 |
68
- | 0.1237 | 13.0 | 715 | 3.9905 |
69
- | 0.1161 | 14.0 | 770 | 4.0962 |
70
- | 0.1112 | 15.0 | 825 | 4.1331 |
71
- | 0.1041 | 16.0 | 880 | 4.2006 |
72
- | 0.1038 | 17.0 | 935 | 4.2211 |
73
- | 0.12 | 18.0 | 990 | 4.2481 |
74
- | 0.1096 | 19.0 | 1045 | 4.2556 |
75
- | 0.1086 | 20.0 | 1100 | 4.2568 |
 
76
 
77
 
78
  ### Framework versions
 
2
  license: other
3
  library_name: peft
4
  tags:
 
 
5
  - trl
6
  - sft
7
  - generated_from_trainer
 
 
8
  base_model: 01-ai/Yi-6B
9
  model-index:
10
  - name: Yi-6B-ruozhiba
 
16
 
17
  # Yi-6B-ruozhiba
18
 
19
+ This model is a fine-tuned version of [01-ai/Yi-6B](https://huggingface.co/01-ai/Yi-6B) on the None dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 4.1664
22
 
23
  ## Model description
24
 
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss |
52
  |:-------------:|:-----:|:----:|:---------------:|
53
+ | 2.333 | 1.0 | 55 | 2.0215 |
54
+ | 1.8575 | 2.0 | 110 | 1.8371 |
55
+ | 1.6435 | 3.0 | 165 | 1.8182 |
56
+ | 1.4138 | 4.0 | 220 | 1.9196 |
57
+ | 1.0749 | 5.0 | 275 | 2.2265 |
58
+ | 0.6715 | 6.0 | 330 | 2.5187 |
59
+ | 0.4252 | 7.0 | 385 | 2.8304 |
60
+ | 0.2572 | 8.0 | 440 | 3.1702 |
61
+ | 0.1543 | 9.0 | 495 | 3.3739 |
62
+ | 0.1095 | 10.0 | 550 | 3.4964 |
63
+ | 0.0875 | 11.0 | 605 | 3.6468 |
64
+ | 0.0692 | 12.0 | 660 | 3.8002 |
65
+ | 0.0573 | 13.0 | 715 | 3.9182 |
66
+ | 0.0527 | 14.0 | 770 | 4.0117 |
67
+ | 0.0494 | 15.0 | 825 | 4.0920 |
68
+ | 0.0459 | 16.0 | 880 | 4.1272 |
69
+ | 0.0456 | 17.0 | 935 | 4.1514 |
70
+ | 0.0435 | 18.0 | 990 | 4.1618 |
71
+ | 0.0446 | 19.0 | 1045 | 4.1660 |
72
+ | 0.0428 | 20.0 | 1100 | 4.1664 |
73
 
74
 
75
  ### Framework versions
all_results.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
  "epoch": 20.0,
3
- "eval_loss": 4.256848335266113,
4
- "eval_runtime": 4.8946,
5
  "eval_samples": 23,
6
- "eval_samples_per_second": 4.699,
7
- "eval_steps_per_second": 1.226,
8
- "train_loss": 0.1145730891010978,
9
- "train_runtime": 6855.7487,
10
  "train_samples": 217,
11
- "train_samples_per_second": 0.633,
12
- "train_steps_per_second": 0.16
13
  }
 
1
  {
2
  "epoch": 20.0,
3
+ "eval_loss": 4.166376113891602,
4
+ "eval_runtime": 4.9235,
5
  "eval_samples": 23,
6
+ "eval_samples_per_second": 4.672,
7
+ "eval_steps_per_second": 1.219,
8
+ "train_loss": 0.5259156136621128,
9
+ "train_runtime": 7800.8143,
10
  "train_samples": 217,
11
+ "train_samples_per_second": 0.556,
12
+ "train_steps_per_second": 0.141
13
  }
eval_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 20.0,
3
- "eval_loss": 4.256848335266113,
4
- "eval_runtime": 4.8946,
5
  "eval_samples": 23,
6
- "eval_samples_per_second": 4.699,
7
- "eval_steps_per_second": 1.226
8
  }
 
1
  {
2
  "epoch": 20.0,
3
+ "eval_loss": 4.166376113891602,
4
+ "eval_runtime": 4.9235,
5
  "eval_samples": 23,
6
+ "eval_samples_per_second": 4.672,
7
+ "eval_steps_per_second": 1.219
8
  }
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 20.0,
3
- "train_loss": 0.1145730891010978,
4
- "train_runtime": 6855.7487,
5
  "train_samples": 217,
6
- "train_samples_per_second": 0.633,
7
- "train_steps_per_second": 0.16
8
  }
 
1
  {
2
  "epoch": 20.0,
3
+ "train_loss": 0.5259156136621128,
4
+ "train_runtime": 7800.8143,
5
  "train_samples": 217,
6
+ "train_samples_per_second": 0.556,
7
+ "train_steps_per_second": 0.141
8
  }
trainer_state.json CHANGED
@@ -11,135 +11,148 @@
11
  {
12
  "epoch": 0.02,
13
  "learning_rate": 4.545454545454545e-07,
14
- "loss": 2.7383,
15
  "step": 1
16
  },
17
  {
18
  "epoch": 0.36,
19
  "learning_rate": 9.090909090909091e-06,
20
- "loss": 2.7445,
21
  "step": 20
22
  },
23
  {
24
  "epoch": 0.73,
25
  "learning_rate": 1.8181818181818182e-05,
26
- "loss": 2.7616,
27
  "step": 40
28
  },
29
  {
30
  "epoch": 1.0,
31
- "gpt4_scores": 0.0,
32
  "step": 55
33
  },
34
  {
35
  "epoch": 1.0,
36
- "eval_loss": 2.393845319747925,
37
- "eval_runtime": 4.9416,
38
- "eval_samples_per_second": 4.654,
39
- "eval_steps_per_second": 1.214,
40
  "step": 55
41
  },
42
  {
43
  "epoch": 1.09,
44
  "learning_rate": 2.7272727272727273e-05,
45
- "loss": 2.3983,
46
  "step": 60
47
  },
48
  {
49
  "epoch": 1.45,
50
  "learning_rate": 3.6363636363636364e-05,
51
- "loss": 2.1924,
52
  "step": 80
53
  },
54
  {
55
  "epoch": 1.82,
56
  "learning_rate": 4.545454545454546e-05,
57
- "loss": 2.0613,
58
  "step": 100
59
  },
60
  {
61
  "epoch": 2.0,
62
- "gpt4_scores": 0.5,
63
  "step": 110
64
  },
65
  {
66
  "epoch": 2.0,
67
- "eval_loss": 1.9860810041427612,
68
- "eval_runtime": 4.9615,
69
- "eval_samples_per_second": 4.636,
70
  "eval_steps_per_second": 1.209,
71
  "step": 110
72
  },
73
  {
74
  "epoch": 2.18,
75
  "learning_rate": 4.9987413559579636e-05,
76
- "loss": 1.8572,
77
  "step": 120
78
  },
79
  {
80
  "epoch": 2.55,
81
  "learning_rate": 4.988679806432712e-05,
82
- "loss": 1.8734,
83
  "step": 140
84
  },
85
  {
86
  "epoch": 2.91,
87
  "learning_rate": 4.968597221690986e-05,
88
- "loss": 1.8236,
89
  "step": 160
90
  },
91
  {
92
  "epoch": 3.0,
93
- "gpt4_scores": 0.6,
94
  "step": 165
95
  },
96
  {
97
  "epoch": 3.0,
98
- "eval_loss": 1.9592580795288086,
99
- "eval_runtime": 4.964,
100
- "eval_samples_per_second": 4.633,
101
  "eval_steps_per_second": 1.209,
102
  "step": 165
103
  },
104
  {
105
  "epoch": 3.27,
106
  "learning_rate": 4.938574467213518e-05,
107
- "loss": 1.6908,
108
  "step": 180
109
  },
110
  {
111
  "epoch": 3.64,
112
  "learning_rate": 4.898732434036244e-05,
113
- "loss": 1.6501,
114
  "step": 200
115
  },
116
  {
117
  "epoch": 4.0,
118
  "learning_rate": 4.849231551964771e-05,
119
- "loss": 1.617,
 
 
 
 
 
 
 
 
 
 
 
 
 
120
  "step": 220
121
  },
122
  {
123
  "epoch": 4.36,
124
  "learning_rate": 4.790271143580174e-05,
125
- "loss": 1.3484,
126
  "step": 240
127
  },
128
  {
129
  "epoch": 4.73,
130
  "learning_rate": 4.722088621637309e-05,
131
- "loss": 1.2817,
132
  "step": 260
133
  },
134
  {
135
  "epoch": 5.0,
136
- "gpt4_scores": 0.7166666666666667,
137
  "step": 275
138
  },
139
  {
140
  "epoch": 5.0,
141
- "eval_loss": 2.249370574951172,
142
- "eval_runtime": 4.9357,
143
  "eval_samples_per_second": 4.66,
144
  "eval_steps_per_second": 1.216,
145
  "step": 275
@@ -147,342 +160,342 @@
147
  {
148
  "epoch": 5.09,
149
  "learning_rate": 4.644958533087443e-05,
150
- "loss": 1.219,
151
  "step": 280
152
  },
153
  {
154
  "epoch": 5.45,
155
  "learning_rate": 4.559191453574582e-05,
156
- "loss": 0.9792,
157
  "step": 300
158
  },
159
  {
160
  "epoch": 5.82,
161
  "learning_rate": 4.465132736856969e-05,
162
- "loss": 0.9519,
163
  "step": 320
164
  },
165
  {
166
  "epoch": 6.0,
167
- "gpt4_scores": 0.4333333333333333,
168
  "step": 330
169
  },
170
  {
171
  "epoch": 6.0,
172
- "eval_loss": 2.587601661682129,
173
- "eval_runtime": 4.9102,
174
- "eval_samples_per_second": 4.684,
175
- "eval_steps_per_second": 1.222,
176
  "step": 330
177
  },
178
  {
179
  "epoch": 6.18,
180
  "learning_rate": 4.3631611241893874e-05,
181
- "loss": 0.7862,
182
  "step": 340
183
  },
184
  {
185
  "epoch": 6.55,
186
  "learning_rate": 4.2536872192658036e-05,
187
- "loss": 0.6587,
188
  "step": 360
189
  },
190
  {
191
  "epoch": 6.91,
192
  "learning_rate": 4.137151834863213e-05,
193
- "loss": 0.6733,
194
  "step": 380
195
  },
196
  {
197
  "epoch": 7.0,
198
- "gpt4_scores": 0.6666666666666666,
199
  "step": 385
200
  },
201
  {
202
  "epoch": 7.0,
203
- "eval_loss": 3.0864293575286865,
204
- "eval_runtime": 4.9167,
205
- "eval_samples_per_second": 4.678,
206
- "eval_steps_per_second": 1.22,
207
  "step": 385
208
  },
209
  {
210
  "epoch": 7.27,
211
  "learning_rate": 4.014024217844167e-05,
212
- "loss": 0.4403,
213
  "step": 400
214
  },
215
  {
216
  "epoch": 7.64,
217
  "learning_rate": 3.884800159665276e-05,
218
- "loss": 0.4879,
219
  "step": 420
220
  },
221
  {
222
  "epoch": 8.0,
223
  "learning_rate": 3.7500000000000003e-05,
224
- "loss": 0.4873,
225
  "step": 440
226
  },
227
  {
228
  "epoch": 8.0,
229
- "gpt4_scores": 0.7000000000000001,
230
  "step": 440
231
  },
232
  {
233
  "epoch": 8.0,
234
- "eval_loss": 3.120582342147827,
235
- "eval_runtime": 4.9178,
236
- "eval_samples_per_second": 4.677,
237
- "eval_steps_per_second": 1.22,
238
  "step": 440
239
  },
240
  {
241
  "epoch": 8.36,
242
  "learning_rate": 3.610166531514436e-05,
243
- "loss": 0.3402,
244
  "step": 460
245
  },
246
  {
247
  "epoch": 8.73,
248
  "learning_rate": 3.465862814232822e-05,
249
- "loss": 0.3163,
250
  "step": 480
251
  },
252
  {
253
  "epoch": 9.0,
254
- "gpt4_scores": 0.5666666666666667,
255
  "step": 495
256
  },
257
  {
258
  "epoch": 9.0,
259
- "eval_loss": 3.35768461227417,
260
- "eval_runtime": 4.9205,
261
- "eval_samples_per_second": 4.674,
262
- "eval_steps_per_second": 1.219,
263
  "step": 495
264
  },
265
  {
266
  "epoch": 9.09,
267
  "learning_rate": 3.3176699082935545e-05,
268
- "loss": 0.3176,
269
  "step": 500
270
  },
271
  {
272
  "epoch": 9.45,
273
  "learning_rate": 3.166184534225087e-05,
274
- "loss": 0.2303,
275
  "step": 520
276
  },
277
  {
278
  "epoch": 9.82,
279
  "learning_rate": 3.012016670162977e-05,
280
- "loss": 0.2422,
281
  "step": 540
282
  },
283
  {
284
  "epoch": 10.0,
285
- "gpt4_scores": 0.31666666666666665,
286
  "step": 550
287
  },
288
  {
289
  "epoch": 10.0,
290
- "eval_loss": 3.5397582054138184,
291
- "eval_runtime": 4.9098,
292
- "eval_samples_per_second": 4.685,
293
- "eval_steps_per_second": 1.222,
294
  "step": 550
295
  },
296
  {
297
  "epoch": 10.18,
298
  "learning_rate": 2.8557870956832132e-05,
299
- "loss": 0.1994,
300
  "step": 560
301
  },
302
  {
303
  "epoch": 10.55,
304
  "learning_rate": 2.698124892141971e-05,
305
- "loss": 0.1625,
306
  "step": 580
307
  },
308
  {
309
  "epoch": 10.91,
310
  "learning_rate": 2.5396649095870202e-05,
311
- "loss": 0.1929,
312
  "step": 600
313
  },
314
  {
315
  "epoch": 11.0,
316
- "gpt4_scores": 0.5666666666666667,
317
  "step": 605
318
  },
319
  {
320
  "epoch": 11.0,
321
- "eval_loss": 3.7079503536224365,
322
- "eval_runtime": 4.915,
323
- "eval_samples_per_second": 4.68,
324
- "eval_steps_per_second": 1.221,
325
  "step": 605
326
  },
327
  {
328
  "epoch": 11.27,
329
  "learning_rate": 2.3810452104406444e-05,
330
- "loss": 0.1475,
331
  "step": 620
332
  },
333
  {
334
  "epoch": 11.64,
335
  "learning_rate": 2.222904500247473e-05,
336
- "loss": 0.1554,
337
  "step": 640
338
  },
339
  {
340
  "epoch": 12.0,
341
  "learning_rate": 2.0658795558326743e-05,
342
- "loss": 0.1429,
343
  "step": 660
344
  },
345
  {
346
  "epoch": 12.0,
347
- "gpt4_scores": 0.5333333333333333,
348
  "step": 660
349
  },
350
  {
351
  "epoch": 12.0,
352
- "eval_loss": 3.8362960815429688,
353
- "eval_runtime": 4.9205,
354
- "eval_samples_per_second": 4.674,
355
  "eval_steps_per_second": 1.219,
356
  "step": 660
357
  },
358
  {
359
  "epoch": 12.36,
360
  "learning_rate": 1.9106026612264316e-05,
361
- "loss": 0.1338,
362
  "step": 680
363
  },
364
  {
365
  "epoch": 12.73,
366
  "learning_rate": 1.7576990616793137e-05,
367
- "loss": 0.1237,
368
  "step": 700
369
  },
370
  {
371
  "epoch": 13.0,
372
- "gpt4_scores": 0.3666666666666667,
373
  "step": 715
374
  },
375
  {
376
  "epoch": 13.0,
377
- "eval_loss": 3.9904778003692627,
378
- "eval_runtime": 4.9311,
379
- "eval_samples_per_second": 4.664,
380
- "eval_steps_per_second": 1.217,
381
  "step": 715
382
  },
383
  {
384
  "epoch": 13.09,
385
  "learning_rate": 1.6077844460203206e-05,
386
- "loss": 0.1228,
387
  "step": 720
388
  },
389
  {
390
  "epoch": 13.45,
391
  "learning_rate": 1.4614624674952842e-05,
392
- "loss": 0.1116,
393
  "step": 740
394
  },
395
  {
396
  "epoch": 13.82,
397
  "learning_rate": 1.3193223130682936e-05,
398
- "loss": 0.1161,
399
  "step": 760
400
  },
401
  {
402
  "epoch": 14.0,
403
- "gpt4_scores": 0.3499999999999999,
404
  "step": 770
405
  },
406
  {
407
  "epoch": 14.0,
408
- "eval_loss": 4.096155166625977,
409
- "eval_runtime": 4.9519,
410
- "eval_samples_per_second": 4.645,
411
- "eval_steps_per_second": 1.212,
412
  "step": 770
413
  },
414
  {
415
  "epoch": 14.18,
416
  "learning_rate": 1.181936330973744e-05,
417
- "loss": 0.1262,
418
  "step": 780
419
  },
420
  {
421
  "epoch": 14.55,
422
  "learning_rate": 1.049857726072005e-05,
423
- "loss": 0.1215,
424
  "step": 800
425
  },
426
  {
427
  "epoch": 14.91,
428
  "learning_rate": 9.236183322886945e-06,
429
- "loss": 0.1112,
430
  "step": 820
431
  },
432
  {
433
  "epoch": 15.0,
434
- "gpt4_scores": 0.5333333333333333,
435
  "step": 825
436
  },
437
  {
438
  "epoch": 15.0,
439
- "eval_loss": 4.133083820343018,
440
- "eval_runtime": 4.8899,
441
- "eval_samples_per_second": 4.704,
442
- "eval_steps_per_second": 1.227,
443
  "step": 825
444
  },
445
  {
446
  "epoch": 15.27,
447
  "learning_rate": 8.0372647110717e-06,
448
- "loss": 0.1219,
449
  "step": 840
450
  },
451
  {
452
  "epoch": 15.64,
453
  "learning_rate": 6.906649047373246e-06,
454
- "loss": 0.0985,
455
  "step": 860
456
  },
457
  {
458
  "epoch": 16.0,
459
  "learning_rate": 5.848888922025553e-06,
460
- "loss": 0.1041,
461
  "step": 880
462
  },
463
  {
464
  "epoch": 16.0,
465
- "gpt4_scores": 0.43333333333333335,
466
  "step": 880
467
  },
468
  {
469
  "epoch": 16.0,
470
- "eval_loss": 4.200608730316162,
471
- "eval_runtime": 4.9125,
472
- "eval_samples_per_second": 4.682,
473
- "eval_steps_per_second": 1.221,
474
  "step": 880
475
  },
476
  {
477
  "epoch": 16.36,
478
  "learning_rate": 4.868243561723535e-06,
479
- "loss": 0.1105,
480
  "step": 900
481
  },
482
  {
483
  "epoch": 16.73,
484
  "learning_rate": 3.968661679220468e-06,
485
- "loss": 0.1038,
486
  "step": 920
487
  },
488
  {
@@ -492,113 +505,113 @@
492
  },
493
  {
494
  "epoch": 17.0,
495
- "eval_loss": 4.221064567565918,
496
- "eval_runtime": 4.9085,
497
- "eval_samples_per_second": 4.686,
498
- "eval_steps_per_second": 1.222,
499
  "step": 935
500
  },
501
  {
502
  "epoch": 17.09,
503
  "learning_rate": 3.1537655732553768e-06,
504
- "loss": 0.102,
505
  "step": 940
506
  },
507
  {
508
  "epoch": 17.45,
509
  "learning_rate": 2.4268365428344736e-06,
510
- "loss": 0.0959,
511
  "step": 960
512
  },
513
  {
514
  "epoch": 17.82,
515
  "learning_rate": 1.790801674598186e-06,
516
- "loss": 0.12,
517
  "step": 980
518
  },
519
  {
520
  "epoch": 18.0,
521
- "gpt4_scores": 0.4166666666666667,
522
  "step": 990
523
  },
524
  {
525
  "epoch": 18.0,
526
- "eval_loss": 4.2480902671813965,
527
- "eval_runtime": 4.9416,
528
- "eval_samples_per_second": 4.654,
529
- "eval_steps_per_second": 1.214,
530
  "step": 990
531
  },
532
  {
533
  "epoch": 18.18,
534
  "learning_rate": 1.248222056476367e-06,
535
- "loss": 0.0944,
536
  "step": 1000
537
  },
538
  {
539
  "epoch": 18.55,
540
  "learning_rate": 8.012824650910938e-07,
541
- "loss": 0.1038,
542
  "step": 1020
543
  },
544
  {
545
  "epoch": 18.91,
546
  "learning_rate": 4.517825684323324e-07,
547
- "loss": 0.1096,
548
  "step": 1040
549
  },
550
  {
551
  "epoch": 19.0,
552
- "gpt4_scores": 0.43333333333333335,
553
  "step": 1045
554
  },
555
  {
556
  "epoch": 19.0,
557
- "eval_loss": 4.255565166473389,
558
- "eval_runtime": 4.9309,
559
- "eval_samples_per_second": 4.664,
560
- "eval_steps_per_second": 1.217,
561
  "step": 1045
562
  },
563
  {
564
  "epoch": 19.27,
565
  "learning_rate": 2.011296792301165e-07,
566
- "loss": 0.0992,
567
  "step": 1060
568
  },
569
  {
570
  "epoch": 19.64,
571
  "learning_rate": 5.033308820289184e-08,
572
- "loss": 0.0993,
573
  "step": 1080
574
  },
575
  {
576
  "epoch": 20.0,
577
  "learning_rate": 0.0,
578
- "loss": 0.1086,
579
  "step": 1100
580
  },
581
  {
582
  "epoch": 20.0,
583
- "gpt4_scores": 0.3499999999999999,
584
  "step": 1100
585
  },
586
  {
587
  "epoch": 20.0,
588
- "eval_loss": 4.256848335266113,
589
- "eval_runtime": 4.9124,
590
- "eval_samples_per_second": 4.682,
591
- "eval_steps_per_second": 1.221,
592
  "step": 1100
593
  },
594
  {
595
  "epoch": 20.0,
596
  "step": 1100,
597
- "total_flos": 3.804893675237376e+16,
598
- "train_loss": 0.1145730891010978,
599
- "train_runtime": 6855.7487,
600
- "train_samples_per_second": 0.633,
601
- "train_steps_per_second": 0.16
602
  }
603
  ],
604
  "logging_steps": 20,
@@ -606,7 +619,7 @@
606
  "num_input_tokens_seen": 0,
607
  "num_train_epochs": 20,
608
  "save_steps": 20,
609
- "total_flos": 3.804893675237376e+16,
610
  "train_batch_size": 4,
611
  "trial_name": null,
612
  "trial_params": null
 
11
  {
12
  "epoch": 0.02,
13
  "learning_rate": 4.545454545454545e-07,
14
+ "loss": 2.3833,
15
  "step": 1
16
  },
17
  {
18
  "epoch": 0.36,
19
  "learning_rate": 9.090909090909091e-06,
20
+ "loss": 2.345,
21
  "step": 20
22
  },
23
  {
24
  "epoch": 0.73,
25
  "learning_rate": 1.8181818181818182e-05,
26
+ "loss": 2.333,
27
  "step": 40
28
  },
29
  {
30
  "epoch": 1.0,
31
+ "gpt4_scores": 0.21666666666666667,
32
  "step": 55
33
  },
34
  {
35
  "epoch": 1.0,
36
+ "eval_loss": 2.0215015411376953,
37
+ "eval_runtime": 4.9229,
38
+ "eval_samples_per_second": 4.672,
39
+ "eval_steps_per_second": 1.219,
40
  "step": 55
41
  },
42
  {
43
  "epoch": 1.09,
44
  "learning_rate": 2.7272727272727273e-05,
45
+ "loss": 2.1124,
46
  "step": 60
47
  },
48
  {
49
  "epoch": 1.45,
50
  "learning_rate": 3.6363636363636364e-05,
51
+ "loss": 1.9167,
52
  "step": 80
53
  },
54
  {
55
  "epoch": 1.82,
56
  "learning_rate": 4.545454545454546e-05,
57
+ "loss": 1.8575,
58
  "step": 100
59
  },
60
  {
61
  "epoch": 2.0,
62
+ "gpt4_scores": 0.7166666666666667,
63
  "step": 110
64
  },
65
  {
66
  "epoch": 2.0,
67
+ "eval_loss": 1.8371087312698364,
68
+ "eval_runtime": 4.9624,
69
+ "eval_samples_per_second": 4.635,
70
  "eval_steps_per_second": 1.209,
71
  "step": 110
72
  },
73
  {
74
  "epoch": 2.18,
75
  "learning_rate": 4.9987413559579636e-05,
76
+ "loss": 1.7124,
77
  "step": 120
78
  },
79
  {
80
  "epoch": 2.55,
81
  "learning_rate": 4.988679806432712e-05,
82
+ "loss": 1.7283,
83
  "step": 140
84
  },
85
  {
86
  "epoch": 2.91,
87
  "learning_rate": 4.968597221690986e-05,
88
+ "loss": 1.6435,
89
  "step": 160
90
  },
91
  {
92
  "epoch": 3.0,
93
+ "gpt4_scores": 0.7166666666666667,
94
  "step": 165
95
  },
96
  {
97
  "epoch": 3.0,
98
+ "eval_loss": 1.8182122707366943,
99
+ "eval_runtime": 4.9633,
100
+ "eval_samples_per_second": 4.634,
101
  "eval_steps_per_second": 1.209,
102
  "step": 165
103
  },
104
  {
105
  "epoch": 3.27,
106
  "learning_rate": 4.938574467213518e-05,
107
+ "loss": 1.4997,
108
  "step": 180
109
  },
110
  {
111
  "epoch": 3.64,
112
  "learning_rate": 4.898732434036244e-05,
113
+ "loss": 1.4507,
114
  "step": 200
115
  },
116
  {
117
  "epoch": 4.0,
118
  "learning_rate": 4.849231551964771e-05,
119
+ "loss": 1.4138,
120
+ "step": 220
121
+ },
122
+ {
123
+ "epoch": 4.0,
124
+ "gpt4_scores": 0.7999999999999999,
125
+ "step": 220
126
+ },
127
+ {
128
+ "epoch": 4.0,
129
+ "eval_loss": 1.9195586442947388,
130
+ "eval_runtime": 4.9623,
131
+ "eval_samples_per_second": 4.635,
132
+ "eval_steps_per_second": 1.209,
133
  "step": 220
134
  },
135
  {
136
  "epoch": 4.36,
137
  "learning_rate": 4.790271143580174e-05,
138
+ "loss": 1.0929,
139
  "step": 240
140
  },
141
  {
142
  "epoch": 4.73,
143
  "learning_rate": 4.722088621637309e-05,
144
+ "loss": 1.0749,
145
  "step": 260
146
  },
147
  {
148
  "epoch": 5.0,
149
+ "gpt4_scores": 0.6333333333333333,
150
  "step": 275
151
  },
152
  {
153
  "epoch": 5.0,
154
+ "eval_loss": 2.2265050411224365,
155
+ "eval_runtime": 4.9358,
156
  "eval_samples_per_second": 4.66,
157
  "eval_steps_per_second": 1.216,
158
  "step": 275
 
160
  {
161
  "epoch": 5.09,
162
  "learning_rate": 4.644958533087443e-05,
163
+ "loss": 0.9318,
164
  "step": 280
165
  },
166
  {
167
  "epoch": 5.45,
168
  "learning_rate": 4.559191453574582e-05,
169
+ "loss": 0.6893,
170
  "step": 300
171
  },
172
  {
173
  "epoch": 5.82,
174
  "learning_rate": 4.465132736856969e-05,
175
+ "loss": 0.6715,
176
  "step": 320
177
  },
178
  {
179
  "epoch": 6.0,
180
+ "gpt4_scores": 0.7666666666666666,
181
  "step": 330
182
  },
183
  {
184
  "epoch": 6.0,
185
+ "eval_loss": 2.5187041759490967,
186
+ "eval_runtime": 4.9618,
187
+ "eval_samples_per_second": 4.635,
188
+ "eval_steps_per_second": 1.209,
189
  "step": 330
190
  },
191
  {
192
  "epoch": 6.18,
193
  "learning_rate": 4.3631611241893874e-05,
194
+ "loss": 0.5555,
195
  "step": 340
196
  },
197
  {
198
  "epoch": 6.55,
199
  "learning_rate": 4.2536872192658036e-05,
200
+ "loss": 0.4251,
201
  "step": 360
202
  },
203
  {
204
  "epoch": 6.91,
205
  "learning_rate": 4.137151834863213e-05,
206
+ "loss": 0.4252,
207
  "step": 380
208
  },
209
  {
210
  "epoch": 7.0,
211
+ "gpt4_scores": 0.7833333333333333,
212
  "step": 385
213
  },
214
  {
215
  "epoch": 7.0,
216
+ "eval_loss": 2.830350875854492,
217
+ "eval_runtime": 4.9687,
218
+ "eval_samples_per_second": 4.629,
219
+ "eval_steps_per_second": 1.208,
220
  "step": 385
221
  },
222
  {
223
  "epoch": 7.27,
224
  "learning_rate": 4.014024217844167e-05,
225
+ "loss": 0.295,
226
  "step": 400
227
  },
228
  {
229
  "epoch": 7.64,
230
  "learning_rate": 3.884800159665276e-05,
231
+ "loss": 0.2544,
232
  "step": 420
233
  },
234
  {
235
  "epoch": 8.0,
236
  "learning_rate": 3.7500000000000003e-05,
237
+ "loss": 0.2572,
238
  "step": 440
239
  },
240
  {
241
  "epoch": 8.0,
242
+ "gpt4_scores": 0.5833333333333334,
243
  "step": 440
244
  },
245
  {
246
  "epoch": 8.0,
247
+ "eval_loss": 3.170214891433716,
248
+ "eval_runtime": 4.9207,
249
+ "eval_samples_per_second": 4.674,
250
+ "eval_steps_per_second": 1.219,
251
  "step": 440
252
  },
253
  {
254
  "epoch": 8.36,
255
  "learning_rate": 3.610166531514436e-05,
256
+ "loss": 0.1568,
257
  "step": 460
258
  },
259
  {
260
  "epoch": 8.73,
261
  "learning_rate": 3.465862814232822e-05,
262
+ "loss": 0.1543,
263
  "step": 480
264
  },
265
  {
266
  "epoch": 9.0,
267
+ "gpt4_scores": 0.5833333333333334,
268
  "step": 495
269
  },
270
  {
271
  "epoch": 9.0,
272
+ "eval_loss": 3.3739395141601562,
273
+ "eval_runtime": 4.953,
274
+ "eval_samples_per_second": 4.644,
275
+ "eval_steps_per_second": 1.211,
276
  "step": 495
277
  },
278
  {
279
  "epoch": 9.09,
280
  "learning_rate": 3.3176699082935545e-05,
281
+ "loss": 0.1501,
282
  "step": 500
283
  },
284
  {
285
  "epoch": 9.45,
286
  "learning_rate": 3.166184534225087e-05,
287
+ "loss": 0.1172,
288
  "step": 520
289
  },
290
  {
291
  "epoch": 9.82,
292
  "learning_rate": 3.012016670162977e-05,
293
+ "loss": 0.1095,
294
  "step": 540
295
  },
296
  {
297
  "epoch": 10.0,
298
+ "gpt4_scores": 0.75,
299
  "step": 550
300
  },
301
  {
302
  "epoch": 10.0,
303
+ "eval_loss": 3.496412515640259,
304
+ "eval_runtime": 4.961,
305
+ "eval_samples_per_second": 4.636,
306
+ "eval_steps_per_second": 1.209,
307
  "step": 550
308
  },
309
  {
310
  "epoch": 10.18,
311
  "learning_rate": 2.8557870956832132e-05,
312
+ "loss": 0.1012,
313
  "step": 560
314
  },
315
  {
316
  "epoch": 10.55,
317
  "learning_rate": 2.698124892141971e-05,
318
+ "loss": 0.0825,
319
  "step": 580
320
  },
321
  {
322
  "epoch": 10.91,
323
  "learning_rate": 2.5396649095870202e-05,
324
+ "loss": 0.0875,
325
  "step": 600
326
  },
327
  {
328
  "epoch": 11.0,
329
+ "gpt4_scores": 0.48333333333333334,
330
  "step": 605
331
  },
332
  {
333
  "epoch": 11.0,
334
+ "eval_loss": 3.64684796333313,
335
+ "eval_runtime": 4.9227,
336
+ "eval_samples_per_second": 4.672,
337
+ "eval_steps_per_second": 1.219,
338
  "step": 605
339
  },
340
  {
341
  "epoch": 11.27,
342
  "learning_rate": 2.3810452104406444e-05,
343
+ "loss": 0.0739,
344
  "step": 620
345
  },
346
  {
347
  "epoch": 11.64,
348
  "learning_rate": 2.222904500247473e-05,
349
+ "loss": 0.0683,
350
  "step": 640
351
  },
352
  {
353
  "epoch": 12.0,
354
  "learning_rate": 2.0658795558326743e-05,
355
+ "loss": 0.0692,
356
  "step": 660
357
  },
358
  {
359
  "epoch": 12.0,
360
+ "gpt4_scores": 0.7166666666666667,
361
  "step": 660
362
  },
363
  {
364
  "epoch": 12.0,
365
+ "eval_loss": 3.800189256668091,
366
+ "eval_runtime": 4.9225,
367
+ "eval_samples_per_second": 4.672,
368
  "eval_steps_per_second": 1.219,
369
  "step": 660
370
  },
371
  {
372
  "epoch": 12.36,
373
  "learning_rate": 1.9106026612264316e-05,
374
+ "loss": 0.0617,
375
  "step": 680
376
  },
377
  {
378
  "epoch": 12.73,
379
  "learning_rate": 1.7576990616793137e-05,
380
+ "loss": 0.0573,
381
  "step": 700
382
  },
383
  {
384
  "epoch": 13.0,
385
+ "gpt4_scores": 0.5,
386
  "step": 715
387
  },
388
  {
389
  "epoch": 13.0,
390
+ "eval_loss": 3.918210506439209,
391
+ "eval_runtime": 4.9555,
392
+ "eval_samples_per_second": 4.641,
393
+ "eval_steps_per_second": 1.211,
394
  "step": 715
395
  },
396
  {
397
  "epoch": 13.09,
398
  "learning_rate": 1.6077844460203206e-05,
399
+ "loss": 0.058,
400
  "step": 720
401
  },
402
  {
403
  "epoch": 13.45,
404
  "learning_rate": 1.4614624674952842e-05,
405
+ "loss": 0.0512,
406
  "step": 740
407
  },
408
  {
409
  "epoch": 13.82,
410
  "learning_rate": 1.3193223130682936e-05,
411
+ "loss": 0.0527,
412
  "step": 760
413
  },
414
  {
415
  "epoch": 14.0,
416
+ "gpt4_scores": 0.5166666666666667,
417
  "step": 770
418
  },
419
  {
420
  "epoch": 14.0,
421
+ "eval_loss": 4.011707782745361,
422
+ "eval_runtime": 4.9674,
423
+ "eval_samples_per_second": 4.63,
424
+ "eval_steps_per_second": 1.208,
425
  "step": 770
426
  },
427
  {
428
  "epoch": 14.18,
429
  "learning_rate": 1.181936330973744e-05,
430
+ "loss": 0.0498,
431
  "step": 780
432
  },
433
  {
434
  "epoch": 14.55,
435
  "learning_rate": 1.049857726072005e-05,
436
+ "loss": 0.0489,
437
  "step": 800
438
  },
439
  {
440
  "epoch": 14.91,
441
  "learning_rate": 9.236183322886945e-06,
442
+ "loss": 0.0494,
443
  "step": 820
444
  },
445
  {
446
  "epoch": 15.0,
447
+ "gpt4_scores": 0.65,
448
  "step": 825
449
  },
450
  {
451
  "epoch": 15.0,
452
+ "eval_loss": 4.091977596282959,
453
+ "eval_runtime": 4.9428,
454
+ "eval_samples_per_second": 4.653,
455
+ "eval_steps_per_second": 1.214,
456
  "step": 825
457
  },
458
  {
459
  "epoch": 15.27,
460
  "learning_rate": 8.0372647110717e-06,
461
+ "loss": 0.0471,
462
  "step": 840
463
  },
464
  {
465
  "epoch": 15.64,
466
  "learning_rate": 6.906649047373246e-06,
467
+ "loss": 0.0461,
468
  "step": 860
469
  },
470
  {
471
  "epoch": 16.0,
472
  "learning_rate": 5.848888922025553e-06,
473
+ "loss": 0.0459,
474
  "step": 880
475
  },
476
  {
477
  "epoch": 16.0,
478
+ "gpt4_scores": 0.5833333333333334,
479
  "step": 880
480
  },
481
  {
482
  "epoch": 16.0,
483
+ "eval_loss": 4.127155303955078,
484
+ "eval_runtime": 4.9427,
485
+ "eval_samples_per_second": 4.653,
486
+ "eval_steps_per_second": 1.214,
487
  "step": 880
488
  },
489
  {
490
  "epoch": 16.36,
491
  "learning_rate": 4.868243561723535e-06,
492
+ "loss": 0.0453,
493
  "step": 900
494
  },
495
  {
496
  "epoch": 16.73,
497
  "learning_rate": 3.968661679220468e-06,
498
+ "loss": 0.0456,
499
  "step": 920
500
  },
501
  {
 
505
  },
506
  {
507
  "epoch": 17.0,
508
+ "eval_loss": 4.1514201164245605,
509
+ "eval_runtime": 4.9682,
510
+ "eval_samples_per_second": 4.629,
511
+ "eval_steps_per_second": 1.208,
512
  "step": 935
513
  },
514
  {
515
  "epoch": 17.09,
516
  "learning_rate": 3.1537655732553768e-06,
517
+ "loss": 0.0469,
518
  "step": 940
519
  },
520
  {
521
  "epoch": 17.45,
522
  "learning_rate": 2.4268365428344736e-06,
523
+ "loss": 0.0447,
524
  "step": 960
525
  },
526
  {
527
  "epoch": 17.82,
528
  "learning_rate": 1.790801674598186e-06,
529
+ "loss": 0.0435,
530
  "step": 980
531
  },
532
  {
533
  "epoch": 18.0,
534
+ "gpt4_scores": 0.6333333333333333,
535
  "step": 990
536
  },
537
  {
538
  "epoch": 18.0,
539
+ "eval_loss": 4.1617751121521,
540
+ "eval_runtime": 4.9681,
541
+ "eval_samples_per_second": 4.63,
542
+ "eval_steps_per_second": 1.208,
543
  "step": 990
544
  },
545
  {
546
  "epoch": 18.18,
547
  "learning_rate": 1.248222056476367e-06,
548
+ "loss": 0.0484,
549
  "step": 1000
550
  },
551
  {
552
  "epoch": 18.55,
553
  "learning_rate": 8.012824650910938e-07,
554
+ "loss": 0.0436,
555
  "step": 1020
556
  },
557
  {
558
  "epoch": 18.91,
559
  "learning_rate": 4.517825684323324e-07,
560
+ "loss": 0.0446,
561
  "step": 1040
562
  },
563
  {
564
  "epoch": 19.0,
565
+ "gpt4_scores": 0.6166666666666667,
566
  "step": 1045
567
  },
568
  {
569
  "epoch": 19.0,
570
+ "eval_loss": 4.1660075187683105,
571
+ "eval_runtime": 4.9125,
572
+ "eval_samples_per_second": 4.682,
573
+ "eval_steps_per_second": 1.221,
574
  "step": 1045
575
  },
576
  {
577
  "epoch": 19.27,
578
  "learning_rate": 2.011296792301165e-07,
579
+ "loss": 0.0463,
580
  "step": 1060
581
  },
582
  {
583
  "epoch": 19.64,
584
  "learning_rate": 5.033308820289184e-08,
585
+ "loss": 0.0473,
586
  "step": 1080
587
  },
588
  {
589
  "epoch": 20.0,
590
  "learning_rate": 0.0,
591
+ "loss": 0.0428,
592
  "step": 1100
593
  },
594
  {
595
  "epoch": 20.0,
596
+ "gpt4_scores": 0.6833333333333332,
597
  "step": 1100
598
  },
599
  {
600
  "epoch": 20.0,
601
+ "eval_loss": 4.166376113891602,
602
+ "eval_runtime": 4.9632,
603
+ "eval_samples_per_second": 4.634,
604
+ "eval_steps_per_second": 1.209,
605
  "step": 1100
606
  },
607
  {
608
  "epoch": 20.0,
609
  "step": 1100,
610
+ "total_flos": 3.792163606268314e+16,
611
+ "train_loss": 0.5259156136621128,
612
+ "train_runtime": 7800.8143,
613
+ "train_samples_per_second": 0.556,
614
+ "train_steps_per_second": 0.141
615
  }
616
  ],
617
  "logging_steps": 20,
 
619
  "num_input_tokens_seen": 0,
620
  "num_train_epochs": 20,
621
  "save_steps": 20,
622
+ "total_flos": 3.792163606268314e+16,
623
  "train_batch_size": 4,
624
  "trial_name": null,
625
  "trial_params": null