File size: 22,727 Bytes
22385c4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.9984301412872841,
  "eval_steps": 100,
  "global_step": 477,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0,
      "learning_rate": 1.0416666666666666e-08,
      "logits/chosen": -1.375,
      "logits/rejected": -1.21875,
      "logps/chosen": -404.0,
      "logps/rejected": -374.0,
      "loss": 0.6914,
      "rewards/accuracies": 0.0,
      "rewards/chosen": 0.0,
      "rewards/margins": 0.0,
      "rewards/rejected": 0.0,
      "step": 1
    },
    {
      "epoch": 0.02,
      "learning_rate": 1.0416666666666667e-07,
      "logits/chosen": -1.1875,
      "logits/rejected": -1.15625,
      "logps/chosen": -394.0,
      "logps/rejected": -334.0,
      "loss": 0.6944,
      "rewards/accuracies": 0.34375,
      "rewards/chosen": -0.0024261474609375,
      "rewards/margins": -0.004364013671875,
      "rewards/rejected": 0.001922607421875,
      "step": 10
    },
    {
      "epoch": 0.04,
      "learning_rate": 2.0833333333333333e-07,
      "logits/chosen": -1.2109375,
      "logits/rejected": -1.171875,
      "logps/chosen": -412.0,
      "logps/rejected": -342.0,
      "loss": 0.6931,
      "rewards/accuracies": 0.47187501192092896,
      "rewards/chosen": 0.000743865966796875,
      "rewards/margins": 0.003692626953125,
      "rewards/rejected": -0.0029449462890625,
      "step": 20
    },
    {
      "epoch": 0.06,
      "learning_rate": 3.1249999999999997e-07,
      "logits/chosen": -1.2109375,
      "logits/rejected": -1.2265625,
      "logps/chosen": -432.0,
      "logps/rejected": -372.0,
      "loss": 0.6923,
      "rewards/accuracies": 0.45625001192092896,
      "rewards/chosen": 0.002197265625,
      "rewards/margins": 0.003204345703125,
      "rewards/rejected": -0.0009918212890625,
      "step": 30
    },
    {
      "epoch": 0.08,
      "learning_rate": 4.1666666666666667e-07,
      "logits/chosen": -1.2421875,
      "logits/rejected": -1.21875,
      "logps/chosen": -444.0,
      "logps/rejected": -368.0,
      "loss": 0.6933,
      "rewards/accuracies": 0.46875,
      "rewards/chosen": -0.0019073486328125,
      "rewards/margins": 9.357929229736328e-06,
      "rewards/rejected": -0.00191497802734375,
      "step": 40
    },
    {
      "epoch": 0.1,
      "learning_rate": 4.999731868769026e-07,
      "logits/chosen": -1.1953125,
      "logits/rejected": -1.15625,
      "logps/chosen": -432.0,
      "logps/rejected": -384.0,
      "loss": 0.6918,
      "rewards/accuracies": 0.4906249940395355,
      "rewards/chosen": 0.000812530517578125,
      "rewards/margins": 0.00439453125,
      "rewards/rejected": -0.0035858154296875,
      "step": 50
    },
    {
      "epoch": 0.13,
      "learning_rate": 4.990353313429303e-07,
      "logits/chosen": -1.1484375,
      "logits/rejected": -1.078125,
      "logps/chosen": -400.0,
      "logps/rejected": -308.0,
      "loss": 0.6915,
      "rewards/accuracies": 0.484375,
      "rewards/chosen": 0.0038299560546875,
      "rewards/margins": 0.00457763671875,
      "rewards/rejected": -0.000759124755859375,
      "step": 60
    },
    {
      "epoch": 0.15,
      "learning_rate": 4.967625656594781e-07,
      "logits/chosen": -1.203125,
      "logits/rejected": -1.203125,
      "logps/chosen": -406.0,
      "logps/rejected": -336.0,
      "loss": 0.6907,
      "rewards/accuracies": 0.4625000059604645,
      "rewards/chosen": -0.00010776519775390625,
      "rewards/margins": 0.002593994140625,
      "rewards/rejected": -0.002685546875,
      "step": 70
    },
    {
      "epoch": 0.17,
      "learning_rate": 4.93167072587771e-07,
      "logits/chosen": -1.2265625,
      "logits/rejected": -1.2265625,
      "logps/chosen": -416.0,
      "logps/rejected": -394.0,
      "loss": 0.6907,
      "rewards/accuracies": 0.45625001192092896,
      "rewards/chosen": -0.000720977783203125,
      "rewards/margins": 0.00144195556640625,
      "rewards/rejected": -0.002166748046875,
      "step": 80
    },
    {
      "epoch": 0.19,
      "learning_rate": 4.882681251368548e-07,
      "logits/chosen": -1.171875,
      "logits/rejected": -1.1796875,
      "logps/chosen": -382.0,
      "logps/rejected": -324.0,
      "loss": 0.6871,
      "rewards/accuracies": 0.546875,
      "rewards/chosen": 0.0035858154296875,
      "rewards/margins": 0.01318359375,
      "rewards/rejected": -0.0096435546875,
      "step": 90
    },
    {
      "epoch": 0.21,
      "learning_rate": 4.820919832540181e-07,
      "logits/chosen": -1.28125,
      "logits/rejected": -1.171875,
      "logps/chosen": -420.0,
      "logps/rejected": -372.0,
      "loss": 0.6867,
      "rewards/accuracies": 0.5625,
      "rewards/chosen": -0.00103759765625,
      "rewards/margins": 0.015625,
      "rewards/rejected": -0.0167236328125,
      "step": 100
    },
    {
      "epoch": 0.21,
      "eval_logits/chosen": -1.2421875,
      "eval_logits/rejected": -1.1953125,
      "eval_logps/chosen": -424.0,
      "eval_logps/rejected": -368.0,
      "eval_loss": 0.6852812767028809,
      "eval_rewards/accuracies": 0.5555555820465088,
      "eval_rewards/chosen": -0.005645751953125,
      "eval_rewards/margins": 0.0216064453125,
      "eval_rewards/rejected": -0.027099609375,
      "eval_runtime": 51.3914,
      "eval_samples_per_second": 38.917,
      "eval_steps_per_second": 1.226,
      "step": 100
    },
    {
      "epoch": 0.23,
      "learning_rate": 4.7467175306295647e-07,
      "logits/chosen": -1.234375,
      "logits/rejected": -1.21875,
      "logps/chosen": -448.0,
      "logps/rejected": -366.0,
      "loss": 0.6836,
      "rewards/accuracies": 0.5625,
      "rewards/chosen": -0.0045166015625,
      "rewards/margins": 0.0247802734375,
      "rewards/rejected": -0.029296875,
      "step": 110
    },
    {
      "epoch": 0.25,
      "learning_rate": 4.6604720940421207e-07,
      "logits/chosen": -1.203125,
      "logits/rejected": -1.140625,
      "logps/chosen": -410.0,
      "logps/rejected": -332.0,
      "loss": 0.6802,
      "rewards/accuracies": 0.578125,
      "rewards/chosen": -0.0155029296875,
      "rewards/margins": 0.025390625,
      "rewards/rejected": -0.040771484375,
      "step": 120
    },
    {
      "epoch": 0.27,
      "learning_rate": 4.5626458262912735e-07,
      "logits/chosen": -1.265625,
      "logits/rejected": -1.2265625,
      "logps/chosen": -404.0,
      "logps/rejected": -380.0,
      "loss": 0.6806,
      "rewards/accuracies": 0.5718749761581421,
      "rewards/chosen": -0.04638671875,
      "rewards/margins": 0.0257568359375,
      "rewards/rejected": -0.072265625,
      "step": 130
    },
    {
      "epoch": 0.29,
      "learning_rate": 4.453763107901675e-07,
      "logits/chosen": -1.265625,
      "logits/rejected": -1.2578125,
      "logps/chosen": -454.0,
      "logps/rejected": -388.0,
      "loss": 0.6776,
      "rewards/accuracies": 0.574999988079071,
      "rewards/chosen": -0.0654296875,
      "rewards/margins": 0.037841796875,
      "rewards/rejected": -0.10302734375,
      "step": 140
    },
    {
      "epoch": 0.31,
      "learning_rate": 4.3344075855595097e-07,
      "logits/chosen": -1.2421875,
      "logits/rejected": -1.234375,
      "logps/chosen": -420.0,
      "logps/rejected": -356.0,
      "loss": 0.6716,
      "rewards/accuracies": 0.609375,
      "rewards/chosen": -0.09423828125,
      "rewards/margins": 0.049560546875,
      "rewards/rejected": -0.1435546875,
      "step": 150
    },
    {
      "epoch": 0.33,
      "learning_rate": 4.2052190435769554e-07,
      "logits/chosen": -1.28125,
      "logits/rejected": -1.234375,
      "logps/chosen": -422.0,
      "logps/rejected": -388.0,
      "loss": 0.6672,
      "rewards/accuracies": 0.6000000238418579,
      "rewards/chosen": -0.12353515625,
      "rewards/margins": 0.061767578125,
      "rewards/rejected": -0.185546875,
      "step": 160
    },
    {
      "epoch": 0.36,
      "learning_rate": 4.0668899744407567e-07,
      "logits/chosen": -1.28125,
      "logits/rejected": -1.2265625,
      "logps/chosen": -390.0,
      "logps/rejected": -336.0,
      "loss": 0.674,
      "rewards/accuracies": 0.59375,
      "rewards/chosen": -0.1630859375,
      "rewards/margins": 0.037353515625,
      "rewards/rejected": -0.2001953125,
      "step": 170
    },
    {
      "epoch": 0.38,
      "learning_rate": 3.920161866827889e-07,
      "logits/chosen": -1.3125,
      "logits/rejected": -1.296875,
      "logps/chosen": -426.0,
      "logps/rejected": -382.0,
      "loss": 0.6659,
      "rewards/accuracies": 0.574999988079071,
      "rewards/chosen": -0.181640625,
      "rewards/margins": 0.06298828125,
      "rewards/rejected": -0.244140625,
      "step": 180
    },
    {
      "epoch": 0.4,
      "learning_rate": 3.765821230985757e-07,
      "logits/chosen": -1.3203125,
      "logits/rejected": -1.265625,
      "logps/chosen": -410.0,
      "logps/rejected": -380.0,
      "loss": 0.6594,
      "rewards/accuracies": 0.659375011920929,
      "rewards/chosen": -0.19140625,
      "rewards/margins": 0.09814453125,
      "rewards/rejected": -0.2890625,
      "step": 190
    },
    {
      "epoch": 0.42,
      "learning_rate": 3.604695382782159e-07,
      "logits/chosen": -1.3671875,
      "logits/rejected": -1.265625,
      "logps/chosen": -438.0,
      "logps/rejected": -374.0,
      "loss": 0.6632,
      "rewards/accuracies": 0.621874988079071,
      "rewards/chosen": -0.21875,
      "rewards/margins": 0.07421875,
      "rewards/rejected": -0.29296875,
      "step": 200
    },
    {
      "epoch": 0.42,
      "eval_logits/chosen": -1.3671875,
      "eval_logits/rejected": -1.3203125,
      "eval_logps/chosen": -446.0,
      "eval_logps/rejected": -396.0,
      "eval_loss": 0.6625468730926514,
      "eval_rewards/accuracies": 0.625,
      "eval_rewards/chosen": -0.228515625,
      "eval_rewards/margins": 0.09130859375,
      "eval_rewards/rejected": -0.3203125,
      "eval_runtime": 52.3612,
      "eval_samples_per_second": 38.196,
      "eval_steps_per_second": 1.203,
      "step": 200
    },
    {
      "epoch": 0.44,
      "learning_rate": 3.4376480090239047e-07,
      "logits/chosen": -1.4296875,
      "logits/rejected": -1.359375,
      "logps/chosen": -442.0,
      "logps/rejected": -388.0,
      "loss": 0.6604,
      "rewards/accuracies": 0.609375,
      "rewards/chosen": -0.2421875,
      "rewards/margins": 0.0888671875,
      "rewards/rejected": -0.33203125,
      "step": 210
    },
    {
      "epoch": 0.46,
      "learning_rate": 3.265574537815398e-07,
      "logits/chosen": -1.296875,
      "logits/rejected": -1.28125,
      "logps/chosen": -406.0,
      "logps/rejected": -374.0,
      "loss": 0.6588,
      "rewards/accuracies": 0.590624988079071,
      "rewards/chosen": -0.271484375,
      "rewards/margins": 0.0673828125,
      "rewards/rejected": -0.33984375,
      "step": 220
    },
    {
      "epoch": 0.48,
      "learning_rate": 3.0893973387735683e-07,
      "logits/chosen": -1.34375,
      "logits/rejected": -1.296875,
      "logps/chosen": -406.0,
      "logps/rejected": -372.0,
      "loss": 0.6516,
      "rewards/accuracies": 0.6000000238418579,
      "rewards/chosen": -0.26171875,
      "rewards/margins": 0.1064453125,
      "rewards/rejected": -0.369140625,
      "step": 230
    },
    {
      "epoch": 0.5,
      "learning_rate": 2.910060778827554e-07,
      "logits/chosen": -1.3515625,
      "logits/rejected": -1.328125,
      "logps/chosen": -420.0,
      "logps/rejected": -368.0,
      "loss": 0.6635,
      "rewards/accuracies": 0.609375,
      "rewards/chosen": -0.302734375,
      "rewards/margins": 0.09619140625,
      "rewards/rejected": -0.400390625,
      "step": 240
    },
    {
      "epoch": 0.52,
      "learning_rate": 2.7285261601056697e-07,
      "logits/chosen": -1.3828125,
      "logits/rejected": -1.328125,
      "logps/chosen": -426.0,
      "logps/rejected": -384.0,
      "loss": 0.6546,
      "rewards/accuracies": 0.550000011920929,
      "rewards/chosen": -0.30859375,
      "rewards/margins": 0.10693359375,
      "rewards/rejected": -0.416015625,
      "step": 250
    },
    {
      "epoch": 0.54,
      "learning_rate": 2.5457665670441937e-07,
      "logits/chosen": -1.3515625,
      "logits/rejected": -1.2890625,
      "logps/chosen": -432.0,
      "logps/rejected": -384.0,
      "loss": 0.6521,
      "rewards/accuracies": 0.637499988079071,
      "rewards/chosen": -0.337890625,
      "rewards/margins": 0.142578125,
      "rewards/rejected": -0.48046875,
      "step": 260
    },
    {
      "epoch": 0.57,
      "learning_rate": 2.3627616503391812e-07,
      "logits/chosen": -1.3828125,
      "logits/rejected": -1.390625,
      "logps/chosen": -454.0,
      "logps/rejected": -426.0,
      "loss": 0.6522,
      "rewards/accuracies": 0.609375,
      "rewards/chosen": -0.34375,
      "rewards/margins": 0.11181640625,
      "rewards/rejected": -0.45703125,
      "step": 270
    },
    {
      "epoch": 0.59,
      "learning_rate": 2.1804923757009882e-07,
      "logits/chosen": -1.296875,
      "logits/rejected": -1.3046875,
      "logps/chosen": -426.0,
      "logps/rejected": -372.0,
      "loss": 0.6545,
      "rewards/accuracies": 0.606249988079071,
      "rewards/chosen": -0.35546875,
      "rewards/margins": 0.134765625,
      "rewards/rejected": -0.48828125,
      "step": 280
    },
    {
      "epoch": 0.61,
      "learning_rate": 1.9999357655598891e-07,
      "logits/chosen": -1.3515625,
      "logits/rejected": -1.3671875,
      "logps/chosen": -436.0,
      "logps/rejected": -388.0,
      "loss": 0.6546,
      "rewards/accuracies": 0.6000000238418579,
      "rewards/chosen": -0.37890625,
      "rewards/margins": 0.109375,
      "rewards/rejected": -0.48828125,
      "step": 290
    },
    {
      "epoch": 0.63,
      "learning_rate": 1.8220596619089573e-07,
      "logits/chosen": -1.375,
      "logits/rejected": -1.3046875,
      "logps/chosen": -474.0,
      "logps/rejected": -442.0,
      "loss": 0.6493,
      "rewards/accuracies": 0.6312500238418579,
      "rewards/chosen": -0.33984375,
      "rewards/margins": 0.12109375,
      "rewards/rejected": -0.458984375,
      "step": 300
    },
    {
      "epoch": 0.63,
      "eval_logits/chosen": -1.3984375,
      "eval_logits/rejected": -1.3515625,
      "eval_logps/chosen": -460.0,
      "eval_logps/rejected": -412.0,
      "eval_loss": 0.653249979019165,
      "eval_rewards/accuracies": 0.625,
      "eval_rewards/chosen": -0.35546875,
      "eval_rewards/margins": 0.134765625,
      "eval_rewards/rejected": -0.490234375,
      "eval_runtime": 52.3802,
      "eval_samples_per_second": 38.182,
      "eval_steps_per_second": 1.203,
      "step": 300
    },
    {
      "epoch": 0.65,
      "learning_rate": 1.647817538357072e-07,
      "logits/chosen": -1.390625,
      "logits/rejected": -1.3203125,
      "logps/chosen": -432.0,
      "logps/rejected": -366.0,
      "loss": 0.6467,
      "rewards/accuracies": 0.637499988079071,
      "rewards/chosen": -0.353515625,
      "rewards/margins": 0.125,
      "rewards/rejected": -0.478515625,
      "step": 310
    },
    {
      "epoch": 0.67,
      "learning_rate": 1.478143389201113e-07,
      "logits/chosen": -1.296875,
      "logits/rejected": -1.328125,
      "logps/chosen": -416.0,
      "logps/rejected": -386.0,
      "loss": 0.6499,
      "rewards/accuracies": 0.6343749761581421,
      "rewards/chosen": -0.3515625,
      "rewards/margins": 0.11865234375,
      "rewards/rejected": -0.47265625,
      "step": 320
    },
    {
      "epoch": 0.69,
      "learning_rate": 1.3139467229135998e-07,
      "logits/chosen": -1.4296875,
      "logits/rejected": -1.390625,
      "logps/chosen": -476.0,
      "logps/rejected": -422.0,
      "loss": 0.6409,
      "rewards/accuracies": 0.6343749761581421,
      "rewards/chosen": -0.373046875,
      "rewards/margins": 0.1669921875,
      "rewards/rejected": -0.5390625,
      "step": 330
    },
    {
      "epoch": 0.71,
      "learning_rate": 1.1561076868822755e-07,
      "logits/chosen": -1.3671875,
      "logits/rejected": -1.328125,
      "logps/chosen": -428.0,
      "logps/rejected": -404.0,
      "loss": 0.6505,
      "rewards/accuracies": 0.59375,
      "rewards/chosen": -0.384765625,
      "rewards/margins": 0.10400390625,
      "rewards/rejected": -0.48828125,
      "step": 340
    },
    {
      "epoch": 0.73,
      "learning_rate": 1.0054723495346482e-07,
      "logits/chosen": -1.359375,
      "logits/rejected": -1.2890625,
      "logps/chosen": -420.0,
      "logps/rejected": -388.0,
      "loss": 0.6417,
      "rewards/accuracies": 0.6312500238418579,
      "rewards/chosen": -0.37890625,
      "rewards/margins": 0.1455078125,
      "rewards/rejected": -0.5234375,
      "step": 350
    },
    {
      "epoch": 0.75,
      "learning_rate": 8.628481651367875e-08,
      "logits/chosen": -1.3828125,
      "logits/rejected": -1.3515625,
      "logps/chosen": -476.0,
      "logps/rejected": -400.0,
      "loss": 0.6398,
      "rewards/accuracies": 0.6312500238418579,
      "rewards/chosen": -0.37890625,
      "rewards/margins": 0.1396484375,
      "rewards/rejected": -0.51953125,
      "step": 360
    },
    {
      "epoch": 0.77,
      "learning_rate": 7.289996455765748e-08,
      "logits/chosen": -1.421875,
      "logits/rejected": -1.3984375,
      "logps/chosen": -492.0,
      "logps/rejected": -404.0,
      "loss": 0.6472,
      "rewards/accuracies": 0.625,
      "rewards/chosen": -0.376953125,
      "rewards/margins": 0.138671875,
      "rewards/rejected": -0.515625,
      "step": 370
    },
    {
      "epoch": 0.8,
      "learning_rate": 6.046442623320145e-08,
      "logits/chosen": -1.4140625,
      "logits/rejected": -1.3984375,
      "logps/chosen": -448.0,
      "logps/rejected": -416.0,
      "loss": 0.6386,
      "rewards/accuracies": 0.640625,
      "rewards/chosen": -0.39453125,
      "rewards/margins": 0.126953125,
      "rewards/rejected": -0.51953125,
      "step": 380
    },
    {
      "epoch": 0.82,
      "learning_rate": 4.904486005914027e-08,
      "logits/chosen": -1.3515625,
      "logits/rejected": -1.3515625,
      "logps/chosen": -442.0,
      "logps/rejected": -416.0,
      "loss": 0.6437,
      "rewards/accuracies": 0.6312500238418579,
      "rewards/chosen": -0.39453125,
      "rewards/margins": 0.126953125,
      "rewards/rejected": -0.5234375,
      "step": 390
    },
    {
      "epoch": 0.84,
      "learning_rate": 3.8702478614051345e-08,
      "logits/chosen": -1.375,
      "logits/rejected": -1.34375,
      "logps/chosen": -492.0,
      "logps/rejected": -426.0,
      "loss": 0.6462,
      "rewards/accuracies": 0.5625,
      "rewards/chosen": -0.392578125,
      "rewards/margins": 0.1318359375,
      "rewards/rejected": -0.5234375,
      "step": 400
    },
    {
      "epoch": 0.84,
      "eval_logits/chosen": -1.421875,
      "eval_logits/rejected": -1.375,
      "eval_logps/chosen": -464.0,
      "eval_logps/rejected": -420.0,
      "eval_loss": 0.6495468616485596,
      "eval_rewards/accuracies": 0.6388888955116272,
      "eval_rewards/chosen": -0.400390625,
      "eval_rewards/margins": 0.1552734375,
      "eval_rewards/rejected": -0.5546875,
      "eval_runtime": 51.9691,
      "eval_samples_per_second": 38.484,
      "eval_steps_per_second": 1.212,
      "step": 400
    },
    {
      "epoch": 0.86,
      "learning_rate": 2.9492720416985e-08,
      "logits/chosen": -1.328125,
      "logits/rejected": -1.3046875,
      "logps/chosen": -428.0,
      "logps/rejected": -374.0,
      "loss": 0.651,
      "rewards/accuracies": 0.6343749761581421,
      "rewards/chosen": -0.400390625,
      "rewards/margins": 0.130859375,
      "rewards/rejected": -0.53125,
      "step": 410
    },
    {
      "epoch": 0.88,
      "learning_rate": 2.1464952759020856e-08,
      "logits/chosen": -1.40625,
      "logits/rejected": -1.375,
      "logps/chosen": -432.0,
      "logps/rejected": -408.0,
      "loss": 0.6437,
      "rewards/accuracies": 0.6187499761581421,
      "rewards/chosen": -0.396484375,
      "rewards/margins": 0.12109375,
      "rewards/rejected": -0.51953125,
      "step": 420
    },
    {
      "epoch": 0.9,
      "learning_rate": 1.4662207078575684e-08,
      "logits/chosen": -1.375,
      "logits/rejected": -1.359375,
      "logps/chosen": -458.0,
      "logps/rejected": -404.0,
      "loss": 0.6478,
      "rewards/accuracies": 0.643750011920929,
      "rewards/chosen": -0.380859375,
      "rewards/margins": 0.181640625,
      "rewards/rejected": -0.5625,
      "step": 430
    },
    {
      "epoch": 0.92,
      "learning_rate": 9.12094829893642e-09,
      "logits/chosen": -1.421875,
      "logits/rejected": -1.390625,
      "logps/chosen": -420.0,
      "logps/rejected": -392.0,
      "loss": 0.6468,
      "rewards/accuracies": 0.625,
      "rewards/chosen": -0.390625,
      "rewards/margins": 0.138671875,
      "rewards/rejected": -0.52734375,
      "step": 440
    },
    {
      "epoch": 0.94,
      "learning_rate": 4.8708793644441086e-09,
      "logits/chosen": -1.4296875,
      "logits/rejected": -1.359375,
      "logps/chosen": -452.0,
      "logps/rejected": -426.0,
      "loss": 0.6587,
      "rewards/accuracies": 0.5843750238418579,
      "rewards/chosen": -0.40234375,
      "rewards/margins": 0.08447265625,
      "rewards/rejected": -0.48828125,
      "step": 450
    },
    {
      "epoch": 0.96,
      "learning_rate": 1.9347820230782295e-09,
      "logits/chosen": -1.359375,
      "logits/rejected": -1.3359375,
      "logps/chosen": -426.0,
      "logps/rejected": -390.0,
      "loss": 0.6394,
      "rewards/accuracies": 0.65625,
      "rewards/chosen": -0.40234375,
      "rewards/margins": 0.140625,
      "rewards/rejected": -0.54296875,
      "step": 460
    },
    {
      "epoch": 0.98,
      "learning_rate": 3.2839470889836627e-10,
      "logits/chosen": -1.421875,
      "logits/rejected": -1.375,
      "logps/chosen": -470.0,
      "logps/rejected": -422.0,
      "loss": 0.6303,
      "rewards/accuracies": 0.690625011920929,
      "rewards/chosen": -0.36328125,
      "rewards/margins": 0.19921875,
      "rewards/rejected": -0.55859375,
      "step": 470
    },
    {
      "epoch": 1.0,
      "step": 477,
      "total_flos": 0.0,
      "train_loss": 0.6625334221861897,
      "train_runtime": 3684.7873,
      "train_samples_per_second": 16.591,
      "train_steps_per_second": 0.129
    }
  ],
  "logging_steps": 10,
  "max_steps": 477,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 100,
  "total_flos": 0.0,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}