yyx123 commited on
Commit
4cfbbcc
1 Parent(s): ae4fe06

Model save

Browse files
Files changed (5) hide show
  1. README.md +10 -12
  2. all_results.json +8 -8
  3. eval_results.json +4 -4
  4. train_results.json +4 -4
  5. trainer_state.json +216 -212
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-1e-5-50
@@ -20,9 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  # Yi-6B-ruozhiba-1e-5-50
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: 3.8564
26
 
27
  ## Model description
28
 
@@ -54,12 +50,14 @@ The following hyperparameters were used during training:
54
 
55
  | Training Loss | Epoch | Step | Validation Loss |
56
  |:-------------:|:-----:|:----:|:---------------:|
57
- | 1.8713 | 1.0 | 55 | 1.8389 |
58
- | 0.1631 | 5.0 | 275 | 2.9787 |
59
- | 0.0765 | 6.0 | 330 | 3.0907 |
60
- | 0.0489 | 7.0 | 385 | 3.2638 |
61
- | 0.0559 | 8.0 | 440 | 3.3750 |
62
- | 0.0424 | 9.0 | 495 | 3.4014 |
 
 
63
 
64
 
65
  ### 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-1e-5-50
 
16
 
17
  # Yi-6B-ruozhiba-1e-5-50
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: 3.5129
22
 
23
  ## Model description
24
 
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss |
52
  |:-------------:|:-----:|:----:|:---------------:|
53
+ | 1.5902 | 2.0 | 110 | 2.0347 |
54
+ | 0.4146 | 4.0 | 220 | 2.7832 |
55
+ | 0.2193 | 5.0 | 275 | 3.0240 |
56
+ | 0.1547 | 6.0 | 330 | 3.2444 |
57
+ | 0.1096 | 7.0 | 385 | 3.3953 |
58
+ | 0.1053 | 8.0 | 440 | 3.4807 |
59
+ | 0.0969 | 9.0 | 495 | 3.5086 |
60
+ | 0.1768 | 10.0 | 550 | 3.5129 |
61
 
62
 
63
  ### Framework versions
all_results.json CHANGED
@@ -1,13 +1,13 @@
1
  {
2
  "epoch": 10.0,
3
- "eval_loss": 3.856374740600586,
4
- "eval_runtime": 6.9621,
5
  "eval_samples": 23,
6
- "eval_samples_per_second": 3.304,
7
- "eval_steps_per_second": 0.862,
8
- "train_loss": 0.0,
9
- "train_runtime": 13.7041,
10
  "train_samples": 217,
11
- "train_samples_per_second": 158.347,
12
- "train_steps_per_second": 40.134
13
  }
 
1
  {
2
  "epoch": 10.0,
3
+ "eval_loss": 3.5129005908966064,
4
+ "eval_runtime": 4.948,
5
  "eval_samples": 23,
6
+ "eval_samples_per_second": 4.648,
7
+ "eval_steps_per_second": 1.213,
8
+ "train_loss": 0.1272620278596878,
9
+ "train_runtime": 22126.0761,
10
  "train_samples": 217,
11
+ "train_samples_per_second": 0.098,
12
+ "train_steps_per_second": 0.025
13
  }
eval_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 10.0,
3
- "eval_loss": 3.856374740600586,
4
- "eval_runtime": 6.9621,
5
  "eval_samples": 23,
6
- "eval_samples_per_second": 3.304,
7
- "eval_steps_per_second": 0.862
8
  }
 
1
  {
2
  "epoch": 10.0,
3
+ "eval_loss": 3.5129005908966064,
4
+ "eval_runtime": 4.948,
5
  "eval_samples": 23,
6
+ "eval_samples_per_second": 4.648,
7
+ "eval_steps_per_second": 1.213
8
  }
train_results.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
  "epoch": 10.0,
3
- "train_loss": 0.0,
4
- "train_runtime": 13.7041,
5
  "train_samples": 217,
6
- "train_samples_per_second": 158.347,
7
- "train_steps_per_second": 40.134
8
  }
 
1
  {
2
  "epoch": 10.0,
3
+ "train_loss": 0.1272620278596878,
4
+ "train_runtime": 22126.0761,
5
  "train_samples": 217,
6
+ "train_samples_per_second": 0.098,
7
+ "train_steps_per_second": 0.025
8
  }
trainer_state.json CHANGED
@@ -11,947 +11,951 @@
11
  {
12
  "epoch": 0.02,
13
  "learning_rate": 9.090909090909091e-06,
14
- "loss": 2.3833,
15
  "step": 1
16
  },
17
  {
18
  "epoch": 0.07,
19
  "learning_rate": 3.6363636363636364e-05,
20
- "loss": 2.4676,
21
  "step": 4
22
  },
23
  {
24
  "epoch": 0.15,
25
  "learning_rate": 7.272727272727273e-05,
26
- "loss": 2.2207,
27
  "step": 8
28
  },
29
  {
30
  "epoch": 0.22,
31
  "learning_rate": 0.00010909090909090909,
32
- "loss": 2.1377,
33
  "step": 12
34
  },
35
  {
36
  "epoch": 0.29,
37
  "learning_rate": 0.00014545454545454546,
38
- "loss": 1.9925,
39
  "step": 16
40
  },
41
  {
42
  "epoch": 0.36,
43
  "learning_rate": 0.00018181818181818183,
44
- "loss": 1.9958,
45
  "step": 20
46
  },
47
  {
48
  "epoch": 0.44,
49
  "learning_rate": 0.00021818181818181818,
50
- "loss": 1.8572,
51
  "step": 24
52
  },
53
  {
54
  "epoch": 0.51,
55
  "learning_rate": 0.0002545454545454545,
56
- "loss": 1.9631,
57
  "step": 28
58
  },
59
  {
60
  "epoch": 0.58,
61
  "learning_rate": 0.0002909090909090909,
62
- "loss": 1.8855,
63
  "step": 32
64
  },
65
  {
66
  "epoch": 0.65,
67
  "learning_rate": 0.00032727272727272726,
68
- "loss": 1.8563,
69
  "step": 36
70
  },
71
  {
72
  "epoch": 0.73,
73
  "learning_rate": 0.00036363636363636367,
74
- "loss": 1.77,
75
  "step": 40
76
  },
77
  {
78
  "epoch": 0.8,
79
  "learning_rate": 0.0004,
80
- "loss": 1.8378,
81
  "step": 44
82
  },
83
  {
84
  "epoch": 0.87,
85
  "learning_rate": 0.00043636363636363637,
86
- "loss": 1.9476,
87
  "step": 48
88
  },
89
  {
90
  "epoch": 0.95,
91
  "learning_rate": 0.0004727272727272727,
92
- "loss": 1.8713,
93
  "step": 52
94
  },
95
- {
96
- "epoch": 1.0,
97
- "gpt4_scores": 0.5359999999999999,
98
- "step": 55
99
- },
100
- {
101
- "epoch": 1.0,
102
- "std": 0.0580868315541483,
103
- "step": 55
104
- },
105
- {
106
- "epoch": 1.0,
107
- "eval_loss": 1.8388524055480957,
108
- "eval_runtime": 4.9686,
109
- "eval_samples_per_second": 4.629,
110
- "eval_steps_per_second": 1.208,
111
- "step": 55
112
- },
113
  {
114
  "epoch": 1.02,
115
  "learning_rate": 0.0004999949650182266,
116
- "loss": 1.7168,
117
  "step": 56
118
  },
119
  {
120
  "epoch": 1.09,
121
  "learning_rate": 0.0004998741355957963,
122
- "loss": 1.6773,
123
  "step": 60
124
  },
125
  {
126
  "epoch": 1.16,
127
  "learning_rate": 0.0004995922759815339,
128
- "loss": 1.6162,
129
  "step": 64
130
  },
131
  {
132
  "epoch": 1.24,
133
  "learning_rate": 0.0004991495678185201,
134
- "loss": 1.5625,
135
  "step": 68
136
  },
137
  {
138
  "epoch": 1.31,
139
  "learning_rate": 0.0004985462964079136,
140
- "loss": 1.4995,
141
  "step": 72
142
  },
143
  {
144
  "epoch": 1.38,
145
  "learning_rate": 0.0004977828505250904,
146
- "loss": 1.4863,
147
  "step": 76
148
  },
149
  {
150
  "epoch": 1.45,
151
  "learning_rate": 0.0004968597221690986,
152
- "loss": 1.4958,
153
  "step": 80
154
  },
155
  {
156
  "epoch": 1.53,
157
  "learning_rate": 0.0004957775062455933,
158
- "loss": 1.6001,
159
  "step": 84
160
  },
161
  {
162
  "epoch": 1.6,
163
  "learning_rate": 0.0004945369001834514,
164
- "loss": 1.5667,
165
  "step": 88
166
  },
167
  {
168
  "epoch": 1.67,
169
  "learning_rate": 0.0004931387034853173,
170
- "loss": 1.4828,
171
  "step": 92
172
  },
173
  {
174
  "epoch": 1.75,
175
  "learning_rate": 0.0004915838172123671,
176
- "loss": 1.4508,
177
  "step": 96
178
  },
179
  {
180
  "epoch": 1.82,
181
  "learning_rate": 0.0004898732434036243,
182
- "loss": 1.4932,
183
  "step": 100
184
  },
185
  {
186
  "epoch": 1.89,
187
  "learning_rate": 0.0004880080844302004,
188
- "loss": 1.4789,
189
  "step": 104
190
  },
191
  {
192
  "epoch": 1.96,
193
  "learning_rate": 0.0004859895422848767,
194
- "loss": 1.3297,
195
  "step": 108
196
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
197
  {
198
  "epoch": 2.04,
199
  "learning_rate": 0.00048381891780748665,
200
- "loss": 1.0117,
201
  "step": 112
202
  },
203
  {
204
  "epoch": 2.11,
205
  "learning_rate": 0.0004814976098465951,
206
- "loss": 0.7652,
207
  "step": 116
208
  },
209
  {
210
  "epoch": 2.18,
211
  "learning_rate": 0.0004790271143580174,
212
- "loss": 0.8033,
213
  "step": 120
214
  },
215
  {
216
  "epoch": 2.25,
217
  "learning_rate": 0.0004764090234407577,
218
- "loss": 0.6905,
219
  "step": 124
220
  },
221
  {
222
  "epoch": 2.33,
223
  "learning_rate": 0.0004736450243109884,
224
- "loss": 0.7312,
225
  "step": 128
226
  },
227
  {
228
  "epoch": 2.4,
229
  "learning_rate": 0.00047073689821473173,
230
- "loss": 0.761,
231
  "step": 132
232
  },
233
  {
234
  "epoch": 2.47,
235
  "learning_rate": 0.00046768651927994433,
236
- "loss": 0.6884,
237
  "step": 136
238
  },
239
  {
240
  "epoch": 2.55,
241
  "learning_rate": 0.0004644958533087443,
242
- "loss": 0.623,
243
  "step": 140
244
  },
245
  {
246
  "epoch": 2.62,
247
  "learning_rate": 0.0004611669565105596,
248
- "loss": 0.7938,
249
  "step": 144
250
  },
251
  {
252
  "epoch": 2.69,
253
  "learning_rate": 0.00045770197417701366,
254
- "loss": 0.7256,
255
  "step": 148
256
  },
257
  {
258
  "epoch": 2.76,
259
  "learning_rate": 0.00045410313929940244,
260
- "loss": 0.8236,
261
  "step": 152
262
  },
263
  {
264
  "epoch": 2.84,
265
  "learning_rate": 0.00045037277112965383,
266
- "loss": 0.9053,
267
  "step": 156
268
  },
269
  {
270
  "epoch": 2.91,
271
  "learning_rate": 0.0004465132736856969,
272
- "loss": 0.8822,
273
  "step": 160
274
  },
275
  {
276
  "epoch": 2.98,
277
  "learning_rate": 0.00044252713420220394,
278
- "loss": 0.847,
279
  "step": 164
280
  },
281
  {
282
  "epoch": 3.05,
283
  "learning_rate": 0.00043841692152770415,
284
- "loss": 0.3486,
285
  "step": 168
286
  },
287
  {
288
  "epoch": 3.13,
289
  "learning_rate": 0.00043418528446910123,
290
- "loss": 0.3387,
291
  "step": 172
292
  },
293
  {
294
  "epoch": 3.2,
295
  "learning_rate": 0.0004298349500846628,
296
- "loss": 0.3827,
297
  "step": 176
298
  },
299
  {
300
  "epoch": 3.27,
301
  "learning_rate": 0.00042536872192658034,
302
- "loss": 0.3494,
303
  "step": 180
304
  },
305
  {
306
  "epoch": 3.35,
307
  "learning_rate": 0.00042078947823423365,
308
- "loss": 0.3773,
309
  "step": 184
310
  },
311
  {
312
  "epoch": 3.42,
313
  "learning_rate": 0.0004161001700793231,
314
- "loss": 0.327,
315
  "step": 188
316
  },
317
  {
318
  "epoch": 3.49,
319
  "learning_rate": 0.00041130381946406574,
320
- "loss": 0.4188,
321
  "step": 192
322
  },
323
  {
324
  "epoch": 3.56,
325
  "learning_rate": 0.0004064035173736804,
326
- "loss": 0.3763,
327
  "step": 196
328
  },
329
  {
330
  "epoch": 3.64,
331
  "learning_rate": 0.00040140242178441667,
332
- "loss": 0.3837,
333
  "step": 200
334
  },
335
  {
336
  "epoch": 3.71,
337
  "learning_rate": 0.0003963037556284129,
338
- "loss": 0.4003,
339
  "step": 204
340
  },
341
  {
342
  "epoch": 3.78,
343
  "learning_rate": 0.0003911108047166924,
344
- "loss": 0.32,
345
  "step": 208
346
  },
347
  {
348
  "epoch": 3.85,
349
  "learning_rate": 0.00038582691562163827,
350
- "loss": 0.3738,
351
  "step": 212
352
  },
353
  {
354
  "epoch": 3.93,
355
  "learning_rate": 0.0003804554935203115,
356
- "loss": 0.3584,
357
  "step": 216
358
  },
359
  {
360
  "epoch": 4.0,
361
  "learning_rate": 0.000375,
362
- "loss": 0.3469,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
363
  "step": 220
364
  },
365
  {
366
  "epoch": 4.07,
367
  "learning_rate": 0.0003694639508274158,
368
- "loss": 0.1782,
369
  "step": 224
370
  },
371
  {
372
  "epoch": 4.15,
373
  "learning_rate": 0.0003638509136829758,
374
- "loss": 0.1614,
375
  "step": 228
376
  },
377
  {
378
  "epoch": 4.22,
379
  "learning_rate": 0.00035816450586162706,
380
- "loss": 0.2127,
381
  "step": 232
382
  },
383
  {
384
  "epoch": 4.29,
385
  "learning_rate": 0.00035240839194169884,
386
- "loss": 0.173,
387
  "step": 236
388
  },
389
  {
390
  "epoch": 4.36,
391
  "learning_rate": 0.00034658628142328216,
392
- "loss": 0.1831,
393
  "step": 240
394
  },
395
  {
396
  "epoch": 4.44,
397
  "learning_rate": 0.00034070192633766023,
398
- "loss": 0.2235,
399
  "step": 244
400
  },
401
  {
402
  "epoch": 4.51,
403
  "learning_rate": 0.0003347591188293301,
404
- "loss": 0.1765,
405
  "step": 248
406
  },
407
  {
408
  "epoch": 4.58,
409
  "learning_rate": 0.00032876168871217323,
410
- "loss": 0.1828,
411
  "step": 252
412
  },
413
  {
414
  "epoch": 4.65,
415
  "learning_rate": 0.00032271350100134975,
416
- "loss": 0.1773,
417
  "step": 256
418
  },
419
  {
420
  "epoch": 4.73,
421
  "learning_rate": 0.0003166184534225087,
422
- "loss": 0.1496,
423
  "step": 260
424
  },
425
  {
426
  "epoch": 4.8,
427
  "learning_rate": 0.0003104804738999169,
428
- "loss": 0.176,
429
  "step": 264
430
  },
431
  {
432
  "epoch": 4.87,
433
  "learning_rate": 0.00030430351802512693,
434
- "loss": 0.1545,
435
  "step": 268
436
  },
437
  {
438
  "epoch": 4.95,
439
  "learning_rate": 0.00029809156650781527,
440
- "loss": 0.1631,
441
  "step": 272
442
  },
443
  {
444
  "epoch": 5.0,
445
- "gpt4_scores": 0.6274,
446
- "step": 275
447
- },
448
- {
449
- "epoch": 5.0,
450
- "std": 0.04891773502524415,
451
  "step": 275
452
  },
453
  {
454
  "epoch": 5.0,
455
- "eval_loss": 2.9787437915802,
456
- "eval_runtime": 4.9451,
457
- "eval_samples_per_second": 4.651,
458
- "eval_steps_per_second": 1.213,
459
  "step": 275
460
  },
461
  {
462
  "epoch": 5.02,
463
  "learning_rate": 0.0002918486226104327,
464
- "loss": 0.123,
465
  "step": 276
466
  },
467
  {
468
  "epoch": 5.09,
469
  "learning_rate": 0.00028557870956832135,
470
- "loss": 0.1251,
471
  "step": 280
472
  },
473
  {
474
  "epoch": 5.16,
475
  "learning_rate": 0.0002792858679969596,
476
- "loss": 0.0895,
477
  "step": 284
478
  },
479
  {
480
  "epoch": 5.24,
481
  "learning_rate": 0.0002729741532880069,
482
- "loss": 0.0952,
483
  "step": 288
484
  },
485
  {
486
  "epoch": 5.31,
487
  "learning_rate": 0.000266647632995826,
488
- "loss": 0.0914,
489
  "step": 292
490
  },
491
  {
492
  "epoch": 5.38,
493
  "learning_rate": 0.00026031038421616684,
494
- "loss": 0.1204,
495
  "step": 296
496
  },
497
  {
498
  "epoch": 5.45,
499
  "learning_rate": 0.000253966490958702,
500
- "loss": 0.1004,
501
  "step": 300
502
  },
503
  {
504
  "epoch": 5.53,
505
  "learning_rate": 0.00024762004151510585,
506
- "loss": 0.1251,
507
  "step": 304
508
  },
509
  {
510
  "epoch": 5.6,
511
  "learning_rate": 0.00024127512582437484,
512
- "loss": 0.0999,
513
  "step": 308
514
  },
515
  {
516
  "epoch": 5.67,
517
  "learning_rate": 0.00023493583283708543,
518
- "loss": 0.0932,
519
  "step": 312
520
  },
521
  {
522
  "epoch": 5.75,
523
  "learning_rate": 0.00022860624788029015,
524
- "loss": 0.0786,
525
  "step": 316
526
  },
527
  {
528
  "epoch": 5.82,
529
  "learning_rate": 0.00022229045002474727,
530
- "loss": 0.0984,
531
  "step": 320
532
  },
533
  {
534
  "epoch": 5.89,
535
  "learning_rate": 0.000215992509456184,
536
- "loss": 0.0769,
537
  "step": 324
538
  },
539
  {
540
  "epoch": 5.96,
541
  "learning_rate": 0.000209716484852284,
542
- "loss": 0.0765,
543
  "step": 328
544
  },
545
  {
546
  "epoch": 6.0,
547
- "gpt4_scores": 0.5669999999999998,
 
548
  "step": 330
549
  },
550
  {
551
  "epoch": 6.0,
552
- "std": 0.05915420526048846,
553
- "step": 330
554
- },
555
- {
556
- "epoch": 6.0,
557
- "eval_loss": 3.0907390117645264,
558
- "eval_runtime": 4.9453,
559
- "eval_samples_per_second": 4.651,
560
- "eval_steps_per_second": 1.213,
561
  "step": 330
562
  },
563
  {
564
  "epoch": 6.04,
565
  "learning_rate": 0.0002034664207670925,
566
- "loss": 0.0632,
567
  "step": 332
568
  },
569
  {
570
  "epoch": 6.11,
571
  "learning_rate": 0.0001972463450245226,
572
- "loss": 0.0566,
573
  "step": 336
574
  },
575
  {
576
  "epoch": 6.18,
577
  "learning_rate": 0.00019106026612264316,
578
- "loss": 0.054,
579
  "step": 340
580
  },
581
  {
582
  "epoch": 6.25,
583
  "learning_rate": 0.00018491217065042198,
584
- "loss": 0.0482,
585
  "step": 344
586
  },
587
  {
588
  "epoch": 6.33,
589
  "learning_rate": 0.00017880602071858692,
590
- "loss": 0.0482,
591
  "step": 348
592
  },
593
  {
594
  "epoch": 6.4,
595
  "learning_rate": 0.00017274575140626317,
596
- "loss": 0.0575,
597
  "step": 352
598
  },
599
  {
600
  "epoch": 6.47,
601
  "learning_rate": 0.00016673526822502983,
602
- "loss": 0.0506,
603
  "step": 356
604
  },
605
  {
606
  "epoch": 6.55,
607
  "learning_rate": 0.00016077844460203207,
608
- "loss": 0.0536,
609
  "step": 360
610
  },
611
  {
612
  "epoch": 6.62,
613
  "learning_rate": 0.00015487911938376925,
614
- "loss": 0.0503,
615
  "step": 364
616
  },
617
  {
618
  "epoch": 6.69,
619
  "learning_rate": 0.00014904109436216883,
620
- "loss": 0.0507,
621
  "step": 368
622
  },
623
  {
624
  "epoch": 6.76,
625
  "learning_rate": 0.00014326813182453956,
626
- "loss": 0.0572,
627
  "step": 372
628
  },
629
  {
630
  "epoch": 6.84,
631
  "learning_rate": 0.0001375639521289836,
632
- "loss": 0.0575,
633
  "step": 376
634
  },
635
  {
636
  "epoch": 6.91,
637
  "learning_rate": 0.00013193223130682935,
638
- "loss": 0.0546,
639
  "step": 380
640
  },
641
  {
642
  "epoch": 6.98,
643
  "learning_rate": 0.00012637659869363084,
644
- "loss": 0.0489,
645
  "step": 384
646
  },
647
  {
648
  "epoch": 7.0,
649
- "gpt4_scores": 0.5259999999999999,
 
650
  "step": 385
651
  },
652
  {
653
  "epoch": 7.0,
654
- "std": 0.05718811065247741,
655
- "step": 385
656
- },
657
- {
658
- "epoch": 7.0,
659
- "eval_loss": 3.2638301849365234,
660
- "eval_runtime": 4.9708,
661
- "eval_samples_per_second": 4.627,
662
- "eval_steps_per_second": 1.207,
663
  "step": 385
664
  },
665
  {
666
  "epoch": 7.05,
667
  "learning_rate": 0.00012090063459025954,
668
- "loss": 0.0396,
669
  "step": 388
670
  },
671
  {
672
  "epoch": 7.13,
673
  "learning_rate": 0.0001155078679555969,
674
- "loss": 0.0441,
675
  "step": 392
676
  },
677
  {
678
  "epoch": 7.2,
679
  "learning_rate": 0.00011020177413231333,
680
- "loss": 0.0413,
681
  "step": 396
682
  },
683
  {
684
  "epoch": 7.27,
685
  "learning_rate": 0.00010498577260720049,
686
- "loss": 0.0436,
687
  "step": 400
688
  },
689
  {
690
  "epoch": 7.35,
691
  "learning_rate": 9.986322480749927e-05,
692
- "loss": 0.0532,
693
  "step": 404
694
  },
695
  {
696
  "epoch": 7.42,
697
  "learning_rate": 9.483743193464408e-05,
698
- "loss": 0.0375,
699
  "step": 408
700
  },
701
  {
702
  "epoch": 7.49,
703
  "learning_rate": 8.991163283681945e-05,
704
- "loss": 0.0431,
705
  "step": 412
706
  },
707
  {
708
  "epoch": 7.56,
709
  "learning_rate": 8.508900192169963e-05,
710
- "loss": 0.0449,
711
  "step": 416
712
  },
713
  {
714
  "epoch": 7.64,
715
  "learning_rate": 8.037264711071699e-05,
716
- "loss": 0.0502,
717
  "step": 420
718
  },
719
  {
720
  "epoch": 7.71,
721
  "learning_rate": 7.576560783617667e-05,
722
- "loss": 0.038,
723
  "step": 424
724
  },
725
  {
726
  "epoch": 7.78,
727
  "learning_rate": 7.127085308250913e-05,
728
- "loss": 0.0483,
729
  "step": 428
730
  },
731
  {
732
  "epoch": 7.85,
733
  "learning_rate": 6.689127947292231e-05,
734
- "loss": 0.0411,
735
  "step": 432
736
  },
737
  {
738
  "epoch": 7.93,
739
  "learning_rate": 6.262970940268654e-05,
740
- "loss": 0.0529,
741
  "step": 436
742
  },
743
  {
744
  "epoch": 8.0,
745
  "learning_rate": 5.848888922025553e-05,
746
- "loss": 0.0559,
747
- "step": 440
748
- },
749
- {
750
- "epoch": 8.0,
751
- "gpt4_scores": 0.574,
752
  "step": 440
753
  },
754
  {
755
  "epoch": 8.0,
756
- "std": 0.056200355870759375,
 
757
  "step": 440
758
  },
759
  {
760
  "epoch": 8.0,
761
- "eval_loss": 3.374971389770508,
762
- "eval_runtime": 4.9393,
763
- "eval_samples_per_second": 4.657,
764
- "eval_steps_per_second": 1.215,
765
  "step": 440
766
  },
767
  {
768
  "epoch": 8.07,
769
  "learning_rate": 5.4471487457395216e-05,
770
- "loss": 0.0389,
771
  "step": 444
772
  },
773
  {
774
  "epoch": 8.15,
775
  "learning_rate": 5.058009310946118e-05,
776
- "loss": 0.0445,
777
  "step": 448
778
  },
779
  {
780
  "epoch": 8.22,
781
  "learning_rate": 4.6817213966933034e-05,
782
- "loss": 0.0467,
783
  "step": 452
784
  },
785
  {
786
  "epoch": 8.29,
787
  "learning_rate": 4.318527499928074e-05,
788
- "loss": 0.0357,
789
  "step": 456
790
  },
791
  {
792
  "epoch": 8.36,
793
  "learning_rate": 3.968661679220467e-05,
794
- "loss": 0.0469,
795
  "step": 460
796
  },
797
  {
798
  "epoch": 8.44,
799
  "learning_rate": 3.632349403925664e-05,
800
- "loss": 0.0503,
801
  "step": 464
802
  },
803
  {
804
  "epoch": 8.51,
805
  "learning_rate": 3.309807408881269e-05,
806
- "loss": 0.0419,
807
  "step": 468
808
  },
809
  {
810
  "epoch": 8.58,
811
  "learning_rate": 3.0012435547336736e-05,
812
- "loss": 0.0437,
813
  "step": 472
814
  },
815
  {
816
  "epoch": 8.65,
817
  "learning_rate": 2.7068566939831645e-05,
818
- "loss": 0.0388,
819
  "step": 476
820
  },
821
  {
822
  "epoch": 8.73,
823
  "learning_rate": 2.4268365428344735e-05,
824
- "loss": 0.0379,
825
  "step": 480
826
  },
827
  {
828
  "epoch": 8.8,
829
  "learning_rate": 2.1613635589349755e-05,
830
- "loss": 0.0442,
831
  "step": 484
832
  },
833
  {
834
  "epoch": 8.87,
835
  "learning_rate": 1.9106088250797264e-05,
836
- "loss": 0.0529,
837
  "step": 488
838
  },
839
  {
840
  "epoch": 8.95,
841
  "learning_rate": 1.674733938957873e-05,
842
- "loss": 0.0424,
843
  "step": 492
844
  },
845
  {
846
  "epoch": 9.0,
847
- "gpt4_scores": 0.586,
848
- "step": 495
849
- },
850
- {
851
- "epoch": 9.0,
852
- "std": 0.05261254603229158,
853
  "step": 495
854
  },
855
  {
856
  "epoch": 9.0,
857
- "eval_loss": 3.401412010192871,
858
- "eval_runtime": 4.9704,
859
- "eval_samples_per_second": 4.627,
860
- "eval_steps_per_second": 1.207,
861
  "step": 495
862
  },
863
  {
864
  "epoch": 9.02,
865
  "learning_rate": 1.4538909090118846e-05,
866
- "loss": 0.0488,
867
  "step": 496
868
  },
869
  {
870
  "epoch": 9.09,
871
  "learning_rate": 1.2482220564763668e-05,
872
- "loss": 0.0406,
873
  "step": 500
874
  },
875
  {
876
  "epoch": 9.16,
877
  "learning_rate": 1.0578599236598707e-05,
878
- "loss": 0.0387,
879
  "step": 504
880
  },
881
  {
882
  "epoch": 9.24,
883
  "learning_rate": 8.829271885286095e-06,
884
- "loss": 0.0413,
885
  "step": 508
886
  },
887
  {
888
  "epoch": 9.31,
889
  "learning_rate": 7.235365856472442e-06,
890
- "loss": 0.0512,
891
  "step": 512
892
  },
893
  {
894
  "epoch": 9.38,
895
  "learning_rate": 5.797908335276214e-06,
896
- "loss": 0.0451,
897
  "step": 516
898
  },
899
  {
900
  "epoch": 9.45,
901
  "learning_rate": 4.517825684323323e-06,
902
- "loss": 0.0385,
903
  "step": 520
904
  },
905
  {
906
  "epoch": 9.53,
907
  "learning_rate": 3.3959428467570664e-06,
908
- "loss": 0.0377,
909
  "step": 524
910
  },
911
  {
912
  "epoch": 9.6,
913
  "learning_rate": 2.4329828146074094e-06,
914
- "loss": 0.0434,
915
  "step": 528
916
  },
917
  {
918
  "epoch": 9.67,
919
  "learning_rate": 1.6295661628624448e-06,
920
- "loss": 0.0442,
921
  "step": 532
922
  },
923
  {
924
  "epoch": 9.75,
925
  "learning_rate": 9.862106495415469e-07,
926
- "loss": 0.0418,
927
  "step": 536
928
  },
929
  {
930
  "epoch": 9.82,
931
  "learning_rate": 5.033308820289185e-07,
932
- "loss": 0.0457,
933
  "step": 540
934
  },
935
  {
936
  "epoch": 9.89,
937
  "learning_rate": 1.8123804988159908e-07,
938
- "loss": 0.0428,
939
  "step": 544
940
  },
941
  {
942
  "epoch": 9.96,
943
  "learning_rate": 2.0139724285161975e-08,
944
- "loss": 0.0449,
945
  "step": 548
946
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
947
  {
948
  "epoch": 10.0,
949
  "step": 550,
950
- "total_flos": 1.879358707856179e+16,
951
- "train_loss": 0.0,
952
- "train_runtime": 13.7041,
953
- "train_samples_per_second": 158.347,
954
- "train_steps_per_second": 40.134
955
  }
956
  ],
957
  "logging_steps": 4,
@@ -959,7 +963,7 @@
959
  "num_input_tokens_seen": 0,
960
  "num_train_epochs": 10,
961
  "save_steps": 55,
962
- "total_flos": 1.879358707856179e+16,
963
  "train_batch_size": 4,
964
  "trial_name": null,
965
  "trial_params": null
 
11
  {
12
  "epoch": 0.02,
13
  "learning_rate": 9.090909090909091e-06,
14
+ "loss": 2.7431,
15
  "step": 1
16
  },
17
  {
18
  "epoch": 0.07,
19
  "learning_rate": 3.6363636363636364e-05,
20
+ "loss": 2.8776,
21
  "step": 4
22
  },
23
  {
24
  "epoch": 0.15,
25
  "learning_rate": 7.272727272727273e-05,
26
+ "loss": 2.634,
27
  "step": 8
28
  },
29
  {
30
  "epoch": 0.22,
31
  "learning_rate": 0.00010909090909090909,
32
+ "loss": 2.5152,
33
  "step": 12
34
  },
35
  {
36
  "epoch": 0.29,
37
  "learning_rate": 0.00014545454545454546,
38
+ "loss": 2.3073,
39
  "step": 16
40
  },
41
  {
42
  "epoch": 0.36,
43
  "learning_rate": 0.00018181818181818183,
44
+ "loss": 2.2473,
45
  "step": 20
46
  },
47
  {
48
  "epoch": 0.44,
49
  "learning_rate": 0.00021818181818181818,
50
+ "loss": 2.1606,
51
  "step": 24
52
  },
53
  {
54
  "epoch": 0.51,
55
  "learning_rate": 0.0002545454545454545,
56
+ "loss": 2.1845,
57
  "step": 28
58
  },
59
  {
60
  "epoch": 0.58,
61
  "learning_rate": 0.0002909090909090909,
62
+ "loss": 2.0583,
63
  "step": 32
64
  },
65
  {
66
  "epoch": 0.65,
67
  "learning_rate": 0.00032727272727272726,
68
+ "loss": 2.0335,
69
  "step": 36
70
  },
71
  {
72
  "epoch": 0.73,
73
  "learning_rate": 0.00036363636363636367,
74
+ "loss": 1.924,
75
  "step": 40
76
  },
77
  {
78
  "epoch": 0.8,
79
  "learning_rate": 0.0004,
80
+ "loss": 1.9703,
81
  "step": 44
82
  },
83
  {
84
  "epoch": 0.87,
85
  "learning_rate": 0.00043636363636363637,
86
+ "loss": 1.9575,
87
  "step": 48
88
  },
89
  {
90
  "epoch": 0.95,
91
  "learning_rate": 0.0004727272727272727,
92
+ "loss": 2.0255,
93
  "step": 52
94
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
95
  {
96
  "epoch": 1.02,
97
  "learning_rate": 0.0004999949650182266,
98
+ "loss": 1.9794,
99
  "step": 56
100
  },
101
  {
102
  "epoch": 1.09,
103
  "learning_rate": 0.0004998741355957963,
104
+ "loss": 1.789,
105
  "step": 60
106
  },
107
  {
108
  "epoch": 1.16,
109
  "learning_rate": 0.0004995922759815339,
110
+ "loss": 1.7866,
111
  "step": 64
112
  },
113
  {
114
  "epoch": 1.24,
115
  "learning_rate": 0.0004991495678185201,
116
+ "loss": 1.7061,
117
  "step": 68
118
  },
119
  {
120
  "epoch": 1.31,
121
  "learning_rate": 0.0004985462964079136,
122
+ "loss": 1.6782,
123
  "step": 72
124
  },
125
  {
126
  "epoch": 1.38,
127
  "learning_rate": 0.0004977828505250904,
128
+ "loss": 1.5574,
129
  "step": 76
130
  },
131
  {
132
  "epoch": 1.45,
133
  "learning_rate": 0.0004968597221690986,
134
+ "loss": 1.5883,
135
  "step": 80
136
  },
137
  {
138
  "epoch": 1.53,
139
  "learning_rate": 0.0004957775062455933,
140
+ "loss": 1.7155,
141
  "step": 84
142
  },
143
  {
144
  "epoch": 1.6,
145
  "learning_rate": 0.0004945369001834514,
146
+ "loss": 1.6924,
147
  "step": 88
148
  },
149
  {
150
  "epoch": 1.67,
151
  "learning_rate": 0.0004931387034853173,
152
+ "loss": 1.6414,
153
  "step": 92
154
  },
155
  {
156
  "epoch": 1.75,
157
  "learning_rate": 0.0004915838172123671,
158
+ "loss": 1.6989,
159
  "step": 96
160
  },
161
  {
162
  "epoch": 1.82,
163
  "learning_rate": 0.0004898732434036243,
164
+ "loss": 1.7026,
165
  "step": 100
166
  },
167
  {
168
  "epoch": 1.89,
169
  "learning_rate": 0.0004880080844302004,
170
+ "loss": 1.538,
171
  "step": 104
172
  },
173
  {
174
  "epoch": 1.96,
175
  "learning_rate": 0.0004859895422848767,
176
+ "loss": 1.5902,
177
  "step": 108
178
  },
179
+ {
180
+ "epoch": 2.0,
181
+ "pls_score": 62.8,
182
+ "std": 4.561052510112109,
183
+ "step": 110
184
+ },
185
+ {
186
+ "epoch": 2.0,
187
+ "eval_loss": 2.0346944332122803,
188
+ "eval_runtime": 4.9191,
189
+ "eval_samples_per_second": 4.676,
190
+ "eval_steps_per_second": 1.22,
191
+ "step": 110
192
+ },
193
  {
194
  "epoch": 2.04,
195
  "learning_rate": 0.00048381891780748665,
196
+ "loss": 1.5041,
197
  "step": 112
198
  },
199
  {
200
  "epoch": 2.11,
201
  "learning_rate": 0.0004814976098465951,
202
+ "loss": 0.9413,
203
  "step": 116
204
  },
205
  {
206
  "epoch": 2.18,
207
  "learning_rate": 0.0004790271143580174,
208
+ "loss": 0.9638,
209
  "step": 120
210
  },
211
  {
212
  "epoch": 2.25,
213
  "learning_rate": 0.0004764090234407577,
214
+ "loss": 0.8674,
215
  "step": 124
216
  },
217
  {
218
  "epoch": 2.33,
219
  "learning_rate": 0.0004736450243109884,
220
+ "loss": 0.8621,
221
  "step": 128
222
  },
223
  {
224
  "epoch": 2.4,
225
  "learning_rate": 0.00047073689821473173,
226
+ "loss": 0.86,
227
  "step": 132
228
  },
229
  {
230
  "epoch": 2.47,
231
  "learning_rate": 0.00046768651927994433,
232
+ "loss": 0.7993,
233
  "step": 136
234
  },
235
  {
236
  "epoch": 2.55,
237
  "learning_rate": 0.0004644958533087443,
238
+ "loss": 0.7921,
239
  "step": 140
240
  },
241
  {
242
  "epoch": 2.62,
243
  "learning_rate": 0.0004611669565105596,
244
+ "loss": 0.9535,
245
  "step": 144
246
  },
247
  {
248
  "epoch": 2.69,
249
  "learning_rate": 0.00045770197417701366,
250
+ "loss": 0.823,
251
  "step": 148
252
  },
253
  {
254
  "epoch": 2.76,
255
  "learning_rate": 0.00045410313929940244,
256
+ "loss": 0.8705,
257
  "step": 152
258
  },
259
  {
260
  "epoch": 2.84,
261
  "learning_rate": 0.00045037277112965383,
262
+ "loss": 0.9627,
263
  "step": 156
264
  },
265
  {
266
  "epoch": 2.91,
267
  "learning_rate": 0.0004465132736856969,
268
+ "loss": 1.0076,
269
  "step": 160
270
  },
271
  {
272
  "epoch": 2.98,
273
  "learning_rate": 0.00044252713420220394,
274
+ "loss": 0.9706,
275
  "step": 164
276
  },
277
  {
278
  "epoch": 3.05,
279
  "learning_rate": 0.00043841692152770415,
280
+ "loss": 0.4535,
281
  "step": 168
282
  },
283
  {
284
  "epoch": 3.13,
285
  "learning_rate": 0.00043418528446910123,
286
+ "loss": 0.3861,
287
  "step": 172
288
  },
289
  {
290
  "epoch": 3.2,
291
  "learning_rate": 0.0004298349500846628,
292
+ "loss": 0.4925,
293
  "step": 176
294
  },
295
  {
296
  "epoch": 3.27,
297
  "learning_rate": 0.00042536872192658034,
298
+ "loss": 0.4753,
299
  "step": 180
300
  },
301
  {
302
  "epoch": 3.35,
303
  "learning_rate": 0.00042078947823423365,
304
+ "loss": 0.4131,
305
  "step": 184
306
  },
307
  {
308
  "epoch": 3.42,
309
  "learning_rate": 0.0004161001700793231,
310
+ "loss": 0.3977,
311
  "step": 188
312
  },
313
  {
314
  "epoch": 3.49,
315
  "learning_rate": 0.00041130381946406574,
316
+ "loss": 0.4816,
317
  "step": 192
318
  },
319
  {
320
  "epoch": 3.56,
321
  "learning_rate": 0.0004064035173736804,
322
+ "loss": 0.4944,
323
  "step": 196
324
  },
325
  {
326
  "epoch": 3.64,
327
  "learning_rate": 0.00040140242178441667,
328
+ "loss": 0.4889,
329
  "step": 200
330
  },
331
  {
332
  "epoch": 3.71,
333
  "learning_rate": 0.0003963037556284129,
334
+ "loss": 0.449,
335
  "step": 204
336
  },
337
  {
338
  "epoch": 3.78,
339
  "learning_rate": 0.0003911108047166924,
340
+ "loss": 0.5103,
341
  "step": 208
342
  },
343
  {
344
  "epoch": 3.85,
345
  "learning_rate": 0.00038582691562163827,
346
+ "loss": 0.4971,
347
  "step": 212
348
  },
349
  {
350
  "epoch": 3.93,
351
  "learning_rate": 0.0003804554935203115,
352
+ "loss": 0.4087,
353
  "step": 216
354
  },
355
  {
356
  "epoch": 4.0,
357
  "learning_rate": 0.000375,
358
+ "loss": 0.4146,
359
+ "step": 220
360
+ },
361
+ {
362
+ "epoch": 4.0,
363
+ "pls_score": 63.6,
364
+ "std": 4.067038234391212,
365
+ "step": 220
366
+ },
367
+ {
368
+ "epoch": 4.0,
369
+ "eval_loss": 2.783196449279785,
370
+ "eval_runtime": 4.9791,
371
+ "eval_samples_per_second": 4.619,
372
+ "eval_steps_per_second": 1.205,
373
  "step": 220
374
  },
375
  {
376
  "epoch": 4.07,
377
  "learning_rate": 0.0003694639508274158,
378
+ "loss": 0.2699,
379
  "step": 224
380
  },
381
  {
382
  "epoch": 4.15,
383
  "learning_rate": 0.0003638509136829758,
384
+ "loss": 0.24,
385
  "step": 228
386
  },
387
  {
388
  "epoch": 4.22,
389
  "learning_rate": 0.00035816450586162706,
390
+ "loss": 0.2368,
391
  "step": 232
392
  },
393
  {
394
  "epoch": 4.29,
395
  "learning_rate": 0.00035240839194169884,
396
+ "loss": 0.1893,
397
  "step": 236
398
  },
399
  {
400
  "epoch": 4.36,
401
  "learning_rate": 0.00034658628142328216,
402
+ "loss": 0.2246,
403
  "step": 240
404
  },
405
  {
406
  "epoch": 4.44,
407
  "learning_rate": 0.00034070192633766023,
408
+ "loss": 0.3669,
409
  "step": 244
410
  },
411
  {
412
  "epoch": 4.51,
413
  "learning_rate": 0.0003347591188293301,
414
+ "loss": 0.2311,
415
  "step": 248
416
  },
417
  {
418
  "epoch": 4.58,
419
  "learning_rate": 0.00032876168871217323,
420
+ "loss": 0.213,
421
  "step": 252
422
  },
423
  {
424
  "epoch": 4.65,
425
  "learning_rate": 0.00032271350100134975,
426
+ "loss": 0.2392,
427
  "step": 256
428
  },
429
  {
430
  "epoch": 4.73,
431
  "learning_rate": 0.0003166184534225087,
432
+ "loss": 0.2525,
433
  "step": 260
434
  },
435
  {
436
  "epoch": 4.8,
437
  "learning_rate": 0.0003104804738999169,
438
+ "loss": 0.2326,
439
  "step": 264
440
  },
441
  {
442
  "epoch": 4.87,
443
  "learning_rate": 0.00030430351802512693,
444
+ "loss": 0.2374,
445
  "step": 268
446
  },
447
  {
448
  "epoch": 4.95,
449
  "learning_rate": 0.00029809156650781527,
450
+ "loss": 0.2193,
451
  "step": 272
452
  },
453
  {
454
  "epoch": 5.0,
455
+ "pls_score": 62.0,
456
+ "std": 3.676955262170047,
 
 
 
 
457
  "step": 275
458
  },
459
  {
460
  "epoch": 5.0,
461
+ "eval_loss": 3.024041175842285,
462
+ "eval_runtime": 4.9575,
463
+ "eval_samples_per_second": 4.639,
464
+ "eval_steps_per_second": 1.21,
465
  "step": 275
466
  },
467
  {
468
  "epoch": 5.02,
469
  "learning_rate": 0.0002918486226104327,
470
+ "loss": 0.2179,
471
  "step": 276
472
  },
473
  {
474
  "epoch": 5.09,
475
  "learning_rate": 0.00028557870956832135,
476
+ "loss": 0.1443,
477
  "step": 280
478
  },
479
  {
480
  "epoch": 5.16,
481
  "learning_rate": 0.0002792858679969596,
482
+ "loss": 0.1345,
483
  "step": 284
484
  },
485
  {
486
  "epoch": 5.24,
487
  "learning_rate": 0.0002729741532880069,
488
+ "loss": 0.1421,
489
  "step": 288
490
  },
491
  {
492
  "epoch": 5.31,
493
  "learning_rate": 0.000266647632995826,
494
+ "loss": 0.1247,
495
  "step": 292
496
  },
497
  {
498
  "epoch": 5.38,
499
  "learning_rate": 0.00026031038421616684,
500
+ "loss": 0.1541,
501
  "step": 296
502
  },
503
  {
504
  "epoch": 5.45,
505
  "learning_rate": 0.000253966490958702,
506
+ "loss": 0.1539,
507
  "step": 300
508
  },
509
  {
510
  "epoch": 5.53,
511
  "learning_rate": 0.00024762004151510585,
512
+ "loss": 0.1498,
513
  "step": 304
514
  },
515
  {
516
  "epoch": 5.6,
517
  "learning_rate": 0.00024127512582437484,
518
+ "loss": 0.1678,
519
  "step": 308
520
  },
521
  {
522
  "epoch": 5.67,
523
  "learning_rate": 0.00023493583283708543,
524
+ "loss": 0.2514,
525
  "step": 312
526
  },
527
  {
528
  "epoch": 5.75,
529
  "learning_rate": 0.00022860624788029015,
530
+ "loss": 0.1418,
531
  "step": 316
532
  },
533
  {
534
  "epoch": 5.82,
535
  "learning_rate": 0.00022229045002474727,
536
+ "loss": 0.1455,
537
  "step": 320
538
  },
539
  {
540
  "epoch": 5.89,
541
  "learning_rate": 0.000215992509456184,
542
+ "loss": 0.1259,
543
  "step": 324
544
  },
545
  {
546
  "epoch": 5.96,
547
  "learning_rate": 0.000209716484852284,
548
+ "loss": 0.1547,
549
  "step": 328
550
  },
551
  {
552
  "epoch": 6.0,
553
+ "pls_score": 60.8,
554
+ "std": 4.155382052230576,
555
  "step": 330
556
  },
557
  {
558
  "epoch": 6.0,
559
+ "eval_loss": 3.2443737983703613,
560
+ "eval_runtime": 4.9618,
561
+ "eval_samples_per_second": 4.635,
562
+ "eval_steps_per_second": 1.209,
 
 
 
 
 
563
  "step": 330
564
  },
565
  {
566
  "epoch": 6.04,
567
  "learning_rate": 0.0002034664207670925,
568
+ "loss": 0.1248,
569
  "step": 332
570
  },
571
  {
572
  "epoch": 6.11,
573
  "learning_rate": 0.0001972463450245226,
574
+ "loss": 0.1031,
575
  "step": 336
576
  },
577
  {
578
  "epoch": 6.18,
579
  "learning_rate": 0.00019106026612264316,
580
+ "loss": 0.1077,
581
  "step": 340
582
  },
583
  {
584
  "epoch": 6.25,
585
  "learning_rate": 0.00018491217065042198,
586
+ "loss": 0.1014,
587
  "step": 344
588
  },
589
  {
590
  "epoch": 6.33,
591
  "learning_rate": 0.00017880602071858692,
592
+ "loss": 0.1129,
593
  "step": 348
594
  },
595
  {
596
  "epoch": 6.4,
597
  "learning_rate": 0.00017274575140626317,
598
+ "loss": 0.1099,
599
  "step": 352
600
  },
601
  {
602
  "epoch": 6.47,
603
  "learning_rate": 0.00016673526822502983,
604
+ "loss": 0.1073,
605
  "step": 356
606
  },
607
  {
608
  "epoch": 6.55,
609
  "learning_rate": 0.00016077844460203207,
610
+ "loss": 0.1967,
611
  "step": 360
612
  },
613
  {
614
  "epoch": 6.62,
615
  "learning_rate": 0.00015487911938376925,
616
+ "loss": 0.1046,
617
  "step": 364
618
  },
619
  {
620
  "epoch": 6.69,
621
  "learning_rate": 0.00014904109436216883,
622
+ "loss": 0.1225,
623
  "step": 368
624
  },
625
  {
626
  "epoch": 6.76,
627
  "learning_rate": 0.00014326813182453956,
628
+ "loss": 0.1107,
629
  "step": 372
630
  },
631
  {
632
  "epoch": 6.84,
633
  "learning_rate": 0.0001375639521289836,
634
+ "loss": 0.106,
635
  "step": 376
636
  },
637
  {
638
  "epoch": 6.91,
639
  "learning_rate": 0.00013193223130682935,
640
+ "loss": 0.1025,
641
  "step": 380
642
  },
643
  {
644
  "epoch": 6.98,
645
  "learning_rate": 0.00012637659869363084,
646
+ "loss": 0.1096,
647
  "step": 384
648
  },
649
  {
650
  "epoch": 7.0,
651
+ "pls_score": 59.183673469387756,
652
+ "std": 4.240153693923696,
653
  "step": 385
654
  },
655
  {
656
  "epoch": 7.0,
657
+ "eval_loss": 3.395291566848755,
658
+ "eval_runtime": 4.9577,
659
+ "eval_samples_per_second": 4.639,
660
+ "eval_steps_per_second": 1.21,
 
 
 
 
 
661
  "step": 385
662
  },
663
  {
664
  "epoch": 7.05,
665
  "learning_rate": 0.00012090063459025954,
666
+ "loss": 0.0881,
667
  "step": 388
668
  },
669
  {
670
  "epoch": 7.13,
671
  "learning_rate": 0.0001155078679555969,
672
+ "loss": 0.0862,
673
  "step": 392
674
  },
675
  {
676
  "epoch": 7.2,
677
  "learning_rate": 0.00011020177413231333,
678
+ "loss": 0.1033,
679
  "step": 396
680
  },
681
  {
682
  "epoch": 7.27,
683
  "learning_rate": 0.00010498577260720049,
684
+ "loss": 0.0953,
685
  "step": 400
686
  },
687
  {
688
  "epoch": 7.35,
689
  "learning_rate": 9.986322480749927e-05,
690
+ "loss": 0.0981,
691
  "step": 404
692
  },
693
  {
694
  "epoch": 7.42,
695
  "learning_rate": 9.483743193464408e-05,
696
+ "loss": 0.1042,
697
  "step": 408
698
  },
699
  {
700
  "epoch": 7.49,
701
  "learning_rate": 8.991163283681945e-05,
702
+ "loss": 0.0992,
703
  "step": 412
704
  },
705
  {
706
  "epoch": 7.56,
707
  "learning_rate": 8.508900192169963e-05,
708
+ "loss": 0.0904,
709
  "step": 416
710
  },
711
  {
712
  "epoch": 7.64,
713
  "learning_rate": 8.037264711071699e-05,
714
+ "loss": 0.1085,
715
  "step": 420
716
  },
717
  {
718
  "epoch": 7.71,
719
  "learning_rate": 7.576560783617667e-05,
720
+ "loss": 0.0924,
721
  "step": 424
722
  },
723
  {
724
  "epoch": 7.78,
725
  "learning_rate": 7.127085308250913e-05,
726
+ "loss": 0.213,
727
  "step": 428
728
  },
729
  {
730
  "epoch": 7.85,
731
  "learning_rate": 6.689127947292231e-05,
732
+ "loss": 0.0925,
733
  "step": 432
734
  },
735
  {
736
  "epoch": 7.93,
737
  "learning_rate": 6.262970940268654e-05,
738
+ "loss": 0.1037,
739
  "step": 436
740
  },
741
  {
742
  "epoch": 8.0,
743
  "learning_rate": 5.848888922025553e-05,
744
+ "loss": 0.1053,
 
 
 
 
 
745
  "step": 440
746
  },
747
  {
748
  "epoch": 8.0,
749
+ "pls_score": 58.0,
750
+ "std": 3.7629775444453557,
751
  "step": 440
752
  },
753
  {
754
  "epoch": 8.0,
755
+ "eval_loss": 3.4806602001190186,
756
+ "eval_runtime": 4.9487,
757
+ "eval_samples_per_second": 4.648,
758
+ "eval_steps_per_second": 1.212,
759
  "step": 440
760
  },
761
  {
762
  "epoch": 8.07,
763
  "learning_rate": 5.4471487457395216e-05,
764
+ "loss": 0.0906,
765
  "step": 444
766
  },
767
  {
768
  "epoch": 8.15,
769
  "learning_rate": 5.058009310946118e-05,
770
+ "loss": 0.1902,
771
  "step": 448
772
  },
773
  {
774
  "epoch": 8.22,
775
  "learning_rate": 4.6817213966933034e-05,
776
+ "loss": 0.093,
777
  "step": 452
778
  },
779
  {
780
  "epoch": 8.29,
781
  "learning_rate": 4.318527499928074e-05,
782
+ "loss": 0.0911,
783
  "step": 456
784
  },
785
  {
786
  "epoch": 8.36,
787
  "learning_rate": 3.968661679220467e-05,
788
+ "loss": 0.0958,
789
  "step": 460
790
  },
791
  {
792
  "epoch": 8.44,
793
  "learning_rate": 3.632349403925664e-05,
794
+ "loss": 0.0956,
795
  "step": 464
796
  },
797
  {
798
  "epoch": 8.51,
799
  "learning_rate": 3.309807408881269e-05,
800
+ "loss": 0.094,
801
  "step": 468
802
  },
803
  {
804
  "epoch": 8.58,
805
  "learning_rate": 3.0012435547336736e-05,
806
+ "loss": 0.097,
807
  "step": 472
808
  },
809
  {
810
  "epoch": 8.65,
811
  "learning_rate": 2.7068566939831645e-05,
812
+ "loss": 0.0919,
813
  "step": 476
814
  },
815
  {
816
  "epoch": 8.73,
817
  "learning_rate": 2.4268365428344735e-05,
818
+ "loss": 0.0937,
819
  "step": 480
820
  },
821
  {
822
  "epoch": 8.8,
823
  "learning_rate": 2.1613635589349755e-05,
824
+ "loss": 0.1222,
825
  "step": 484
826
  },
827
  {
828
  "epoch": 8.87,
829
  "learning_rate": 1.9106088250797264e-05,
830
+ "loss": 0.0907,
831
  "step": 488
832
  },
833
  {
834
  "epoch": 8.95,
835
  "learning_rate": 1.674733938957873e-05,
836
+ "loss": 0.0969,
837
  "step": 492
838
  },
839
  {
840
  "epoch": 9.0,
841
+ "pls_score": 55.2,
842
+ "std": 4.022337628792492,
 
 
 
 
843
  "step": 495
844
  },
845
  {
846
  "epoch": 9.0,
847
+ "eval_loss": 3.5086076259613037,
848
+ "eval_runtime": 4.9635,
849
+ "eval_samples_per_second": 4.634,
850
+ "eval_steps_per_second": 1.209,
851
  "step": 495
852
  },
853
  {
854
  "epoch": 9.02,
855
  "learning_rate": 1.4538909090118846e-05,
856
+ "loss": 0.0869,
857
  "step": 496
858
  },
859
  {
860
  "epoch": 9.09,
861
  "learning_rate": 1.2482220564763668e-05,
862
+ "loss": 0.1033,
863
  "step": 500
864
  },
865
  {
866
  "epoch": 9.16,
867
  "learning_rate": 1.0578599236598707e-05,
868
+ "loss": 0.105,
869
  "step": 504
870
  },
871
  {
872
  "epoch": 9.24,
873
  "learning_rate": 8.829271885286095e-06,
874
+ "loss": 0.0989,
875
  "step": 508
876
  },
877
  {
878
  "epoch": 9.31,
879
  "learning_rate": 7.235365856472442e-06,
880
+ "loss": 0.0901,
881
  "step": 512
882
  },
883
  {
884
  "epoch": 9.38,
885
  "learning_rate": 5.797908335276214e-06,
886
+ "loss": 0.0912,
887
  "step": 516
888
  },
889
  {
890
  "epoch": 9.45,
891
  "learning_rate": 4.517825684323323e-06,
892
+ "loss": 0.0857,
893
  "step": 520
894
  },
895
  {
896
  "epoch": 9.53,
897
  "learning_rate": 3.3959428467570664e-06,
898
+ "loss": 0.112,
899
  "step": 524
900
  },
901
  {
902
  "epoch": 9.6,
903
  "learning_rate": 2.4329828146074094e-06,
904
+ "loss": 0.0845,
905
  "step": 528
906
  },
907
  {
908
  "epoch": 9.67,
909
  "learning_rate": 1.6295661628624448e-06,
910
+ "loss": 0.0966,
911
  "step": 532
912
  },
913
  {
914
  "epoch": 9.75,
915
  "learning_rate": 9.862106495415469e-07,
916
+ "loss": 0.0973,
917
  "step": 536
918
  },
919
  {
920
  "epoch": 9.82,
921
  "learning_rate": 5.033308820289185e-07,
922
+ "loss": 0.0894,
923
  "step": 540
924
  },
925
  {
926
  "epoch": 9.89,
927
  "learning_rate": 1.8123804988159908e-07,
928
+ "loss": 0.0867,
929
  "step": 544
930
  },
931
  {
932
  "epoch": 9.96,
933
  "learning_rate": 2.0139724285161975e-08,
934
+ "loss": 0.1768,
935
  "step": 548
936
  },
937
+ {
938
+ "epoch": 10.0,
939
+ "pls_score": 57.95918367346939,
940
+ "std": 4.172255118366954,
941
+ "step": 550
942
+ },
943
+ {
944
+ "epoch": 10.0,
945
+ "eval_loss": 3.5129005908966064,
946
+ "eval_runtime": 4.9489,
947
+ "eval_samples_per_second": 4.648,
948
+ "eval_steps_per_second": 1.212,
949
+ "step": 550
950
+ },
951
  {
952
  "epoch": 10.0,
953
  "step": 550,
954
+ "total_flos": 1.8757245462528e+16,
955
+ "train_loss": 0.1272620278596878,
956
+ "train_runtime": 22126.0761,
957
+ "train_samples_per_second": 0.098,
958
+ "train_steps_per_second": 0.025
959
  }
960
  ],
961
  "logging_steps": 4,
 
963
  "num_input_tokens_seen": 0,
964
  "num_train_epochs": 10,
965
  "save_steps": 55,
966
+ "total_flos": 1.8757245462528e+16,
967
  "train_batch_size": 4,
968
  "trial_name": null,
969
  "trial_params": null