AmberYifan commited on
Commit
c63ce2e
·
verified ·
1 Parent(s): 1ff94fc

Training in progress, epoch 2, checkpoint

Browse files
last-checkpoint/global_step624/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8faa3c1a56d2e4b6a2127fa947980cca888393c36509b9151211d17f2f9d6509
3
+ size 16060527788
last-checkpoint/global_step624/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50aad7dfe7d4605443882a0c887da8e0d2c5dfba98b06a3c81e13fd7f8eddc16
3
+ size 16060527788
last-checkpoint/global_step624/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f7c1b3c32ca42fcff708a9eea9c0ae8cf0564a876450edb7c6bfad3bba8499f
3
+ size 16060527788
last-checkpoint/global_step624/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd4528261c2a2f9faa4b1d743038e7735468feca6c64efe97b10fd2f8f39d51e
3
+ size 16060527788
last-checkpoint/global_step624/zero_pp_rank_0_mp_rank_00_model_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b5c5d8a21ab06d56887e86b68b03881ecd6b57c2cb4e5a8a5990099c47ea14b
3
+ size 150693
last-checkpoint/global_step624/zero_pp_rank_1_mp_rank_00_model_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a1fdff8e9100b2b4c12402d7b0dec2b2e5f0825a70159357ee69e460e446e58
3
+ size 150693
last-checkpoint/global_step624/zero_pp_rank_2_mp_rank_00_model_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1aae3f8006170d79cd2384ff9600a67776cdd1627ec59f247567c161efcb0f1b
3
+ size 150693
last-checkpoint/global_step624/zero_pp_rank_3_mp_rank_00_model_states.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6352ff4375f786bbb4deb191d8efeebb3bfebc1e09751decef3b1b92ddffd155
3
+ size 150693
last-checkpoint/latest CHANGED
@@ -1 +1 @@
1
- global_step312
 
1
+ global_step624
last-checkpoint/model-00001-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c8f466e2478b0406023b68ea02d6d3997d7bc9762430b2cbeeac9c01a5e57a46
3
  size 4976698672
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbbd032fa9662e528e5b422ba0c466f38e864fd3738692339c79e0e94af61c91
3
  size 4976698672
last-checkpoint/model-00002-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a82ddd433c542f3145189754603e53591801184672a703f3d6d770fc67c4ceba
3
  size 4999802720
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5130f7d7be6def99210631ab039ac4d1dd8e2bfb3b341bbea09774358a48a87d
3
  size 4999802720
last-checkpoint/model-00003-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:226ddfc6b5f35efca888be4141dc8e65de7479e8be595f4e71ef5c25dc6352c2
3
  size 4915916176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:102195ce8a84f213e428d6a9cc1a8f78a76b23711a1c6ebbb34f7ca8d53eead5
3
  size 4915916176
last-checkpoint/model-00004-of-00004.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9832c1be84570095d5a050aa8680f5be583723c1859fe50e6676f4e4e2fc932f
3
  size 1168138808
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b1cdebb604c781670825469e1a2fcdc35d6afdf2695801f942744266e9ad0495
3
  size 1168138808
last-checkpoint/rng_state_0.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f1ce770862b76ceecb3453a5d0b9ef65311eab0f6a60f52997726a89d718de08
3
  size 15024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2307c03867cef25b5028feb9a23f80e784b9af9a615de13ddca560a6a90fb593
3
  size 15024
last-checkpoint/rng_state_1.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2bcae6573442578b752be3e988d4ccb38056b45c31b5c02eb579ec6cebcfa62e
3
  size 15024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50218cfaecdd818354e567b7167c13899e3b42297e7d8f58bd7e732cfa547800
3
  size 15024
last-checkpoint/rng_state_2.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:398ff45072fee2975e88b9e078f915103d75d94bc08753303cf855b915973623
3
  size 15024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9814a66b49861f5495b06dae3be12ddf7185b88e2cae1fb808ca9efd99d5807f
3
  size 15024
last-checkpoint/rng_state_3.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7f4b210bc832d4c648e3bee8c21dca26e5a8b365d6ec90c638062005a052e57b
3
  size 15024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a7be93040a65e0a29975f6c70b94418e1fdf88423a50c58aa572141d3c92fbfc
3
  size 15024
last-checkpoint/scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ff41da86439a5cd2aa8fd5d3c55e2154d1b88c51f3be19137724b98e60eefb4d
3
  size 1064
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eba46a04fca9d4584d0e7af07b2d004fba441dbf422bdb9977bfac0706fb41fe
3
  size 1064
last-checkpoint/trainer_state.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
- "epoch": 1.0,
5
  "eval_steps": 500,
6
- "global_step": 312,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
@@ -503,6 +503,487 @@
503
  "eval_samples_per_second": 13.4,
504
  "eval_steps_per_second": 0.469,
505
  "step": 312
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
506
  }
507
  ],
508
  "logging_steps": 10,
 
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
+ "epoch": 2.0,
5
  "eval_steps": 500,
6
+ "global_step": 624,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
 
503
  "eval_samples_per_second": 13.4,
504
  "eval_steps_per_second": 0.469,
505
  "step": 312
506
+ },
507
+ {
508
+ "epoch": 1.0256410256410255,
509
+ "grad_norm": 16.449029965031666,
510
+ "learning_rate": 3.6579572446555817e-07,
511
+ "logits/chosen": -1.59375,
512
+ "logits/rejected": -1.578125,
513
+ "logps/chosen": -117.0,
514
+ "logps/rejected": -195.0,
515
+ "loss": 0.1531,
516
+ "rewards/accuracies": 0.925000011920929,
517
+ "rewards/chosen": -0.34765625,
518
+ "rewards/margins": 3.1875,
519
+ "rewards/rejected": -3.546875,
520
+ "step": 320
521
+ },
522
+ {
523
+ "epoch": 1.0576923076923077,
524
+ "grad_norm": 6.562166239087373,
525
+ "learning_rate": 3.598574821852731e-07,
526
+ "logits/chosen": -1.59375,
527
+ "logits/rejected": -1.5703125,
528
+ "logps/chosen": -111.0,
529
+ "logps/rejected": -187.0,
530
+ "loss": 0.1452,
531
+ "rewards/accuracies": 0.8999999761581421,
532
+ "rewards/chosen": -0.1923828125,
533
+ "rewards/margins": 3.40625,
534
+ "rewards/rejected": -3.609375,
535
+ "step": 330
536
+ },
537
+ {
538
+ "epoch": 1.0897435897435896,
539
+ "grad_norm": 25.028970432454248,
540
+ "learning_rate": 3.5391923990498813e-07,
541
+ "logits/chosen": -1.6484375,
542
+ "logits/rejected": -1.609375,
543
+ "logps/chosen": -134.0,
544
+ "logps/rejected": -219.0,
545
+ "loss": 0.1345,
546
+ "rewards/accuracies": 0.949999988079071,
547
+ "rewards/chosen": -0.69921875,
548
+ "rewards/margins": 3.609375,
549
+ "rewards/rejected": -4.3125,
550
+ "step": 340
551
+ },
552
+ {
553
+ "epoch": 1.1217948717948718,
554
+ "grad_norm": 15.596444813277682,
555
+ "learning_rate": 3.479809976247031e-07,
556
+ "logits/chosen": -1.65625,
557
+ "logits/rejected": -1.6015625,
558
+ "logps/chosen": -130.0,
559
+ "logps/rejected": -207.0,
560
+ "loss": 0.1092,
561
+ "rewards/accuracies": 0.9624999761581421,
562
+ "rewards/chosen": -0.00677490234375,
563
+ "rewards/margins": 3.828125,
564
+ "rewards/rejected": -3.828125,
565
+ "step": 350
566
+ },
567
+ {
568
+ "epoch": 1.1538461538461537,
569
+ "grad_norm": 22.37742199079234,
570
+ "learning_rate": 3.42042755344418e-07,
571
+ "logits/chosen": -1.5859375,
572
+ "logits/rejected": -1.5703125,
573
+ "logps/chosen": -136.0,
574
+ "logps/rejected": -242.0,
575
+ "loss": 0.0918,
576
+ "rewards/accuracies": 0.9750000238418579,
577
+ "rewards/chosen": -0.640625,
578
+ "rewards/margins": 4.4375,
579
+ "rewards/rejected": -5.09375,
580
+ "step": 360
581
+ },
582
+ {
583
+ "epoch": 1.185897435897436,
584
+ "grad_norm": 27.25128799333949,
585
+ "learning_rate": 3.36104513064133e-07,
586
+ "logits/chosen": -1.609375,
587
+ "logits/rejected": -1.59375,
588
+ "logps/chosen": -154.0,
589
+ "logps/rejected": -221.0,
590
+ "loss": 0.1215,
591
+ "rewards/accuracies": 0.9750000238418579,
592
+ "rewards/chosen": -0.333984375,
593
+ "rewards/margins": 4.46875,
594
+ "rewards/rejected": -4.78125,
595
+ "step": 370
596
+ },
597
+ {
598
+ "epoch": 1.217948717948718,
599
+ "grad_norm": 16.802650370434502,
600
+ "learning_rate": 3.3016627078384796e-07,
601
+ "logits/chosen": -1.625,
602
+ "logits/rejected": -1.515625,
603
+ "logps/chosen": -121.0,
604
+ "logps/rejected": -184.0,
605
+ "loss": 0.1051,
606
+ "rewards/accuracies": 0.949999988079071,
607
+ "rewards/chosen": -0.75390625,
608
+ "rewards/margins": 3.75,
609
+ "rewards/rejected": -4.5,
610
+ "step": 380
611
+ },
612
+ {
613
+ "epoch": 1.25,
614
+ "grad_norm": 18.14599606798184,
615
+ "learning_rate": 3.2422802850356297e-07,
616
+ "logits/chosen": -1.65625,
617
+ "logits/rejected": -1.5625,
618
+ "logps/chosen": -118.0,
619
+ "logps/rejected": -203.0,
620
+ "loss": 0.1691,
621
+ "rewards/accuracies": 0.925000011920929,
622
+ "rewards/chosen": -0.4609375,
623
+ "rewards/margins": 4.4375,
624
+ "rewards/rejected": -4.875,
625
+ "step": 390
626
+ },
627
+ {
628
+ "epoch": 1.282051282051282,
629
+ "grad_norm": 14.341477798166297,
630
+ "learning_rate": 3.182897862232779e-07,
631
+ "logits/chosen": -1.609375,
632
+ "logits/rejected": -1.5546875,
633
+ "logps/chosen": -112.0,
634
+ "logps/rejected": -202.0,
635
+ "loss": 0.1084,
636
+ "rewards/accuracies": 0.9624999761581421,
637
+ "rewards/chosen": -0.2578125,
638
+ "rewards/margins": 4.65625,
639
+ "rewards/rejected": -4.90625,
640
+ "step": 400
641
+ },
642
+ {
643
+ "epoch": 1.314102564102564,
644
+ "grad_norm": 22.538352683910524,
645
+ "learning_rate": 3.1235154394299283e-07,
646
+ "logits/chosen": -1.671875,
647
+ "logits/rejected": -1.625,
648
+ "logps/chosen": -168.0,
649
+ "logps/rejected": -270.0,
650
+ "loss": 0.136,
651
+ "rewards/accuracies": 0.9750000238418579,
652
+ "rewards/chosen": -1.1796875,
653
+ "rewards/margins": 4.71875,
654
+ "rewards/rejected": -5.90625,
655
+ "step": 410
656
+ },
657
+ {
658
+ "epoch": 1.3461538461538463,
659
+ "grad_norm": 7.69063972210861,
660
+ "learning_rate": 3.0641330166270784e-07,
661
+ "logits/chosen": -1.609375,
662
+ "logits/rejected": -1.59375,
663
+ "logps/chosen": -128.0,
664
+ "logps/rejected": -232.0,
665
+ "loss": 0.1123,
666
+ "rewards/accuracies": 0.9750000238418579,
667
+ "rewards/chosen": -1.0859375,
668
+ "rewards/margins": 4.59375,
669
+ "rewards/rejected": -5.6875,
670
+ "step": 420
671
+ },
672
+ {
673
+ "epoch": 1.3782051282051282,
674
+ "grad_norm": 22.81215314331503,
675
+ "learning_rate": 3.004750593824228e-07,
676
+ "logits/chosen": -1.6328125,
677
+ "logits/rejected": -1.5390625,
678
+ "logps/chosen": -122.5,
679
+ "logps/rejected": -209.0,
680
+ "loss": 0.0971,
681
+ "rewards/accuracies": 0.9750000238418579,
682
+ "rewards/chosen": -0.33984375,
683
+ "rewards/margins": 4.65625,
684
+ "rewards/rejected": -5.0,
685
+ "step": 430
686
+ },
687
+ {
688
+ "epoch": 1.4102564102564101,
689
+ "grad_norm": 16.88060293661355,
690
+ "learning_rate": 2.9453681710213776e-07,
691
+ "logits/chosen": -1.5859375,
692
+ "logits/rejected": -1.4765625,
693
+ "logps/chosen": -123.0,
694
+ "logps/rejected": -205.0,
695
+ "loss": 0.11,
696
+ "rewards/accuracies": 0.9750000238418579,
697
+ "rewards/chosen": -0.4296875,
698
+ "rewards/margins": 4.5625,
699
+ "rewards/rejected": -5.0,
700
+ "step": 440
701
+ },
702
+ {
703
+ "epoch": 1.4423076923076923,
704
+ "grad_norm": 14.494261083829612,
705
+ "learning_rate": 2.885985748218527e-07,
706
+ "logits/chosen": -1.546875,
707
+ "logits/rejected": -1.5546875,
708
+ "logps/chosen": -118.5,
709
+ "logps/rejected": -221.0,
710
+ "loss": 0.0896,
711
+ "rewards/accuracies": 0.9750000238418579,
712
+ "rewards/chosen": -1.0859375,
713
+ "rewards/margins": 4.59375,
714
+ "rewards/rejected": -5.6875,
715
+ "step": 450
716
+ },
717
+ {
718
+ "epoch": 1.4743589743589745,
719
+ "grad_norm": 16.655718879118293,
720
+ "learning_rate": 2.8266033254156767e-07,
721
+ "logits/chosen": -1.59375,
722
+ "logits/rejected": -1.5859375,
723
+ "logps/chosen": -142.0,
724
+ "logps/rejected": -256.0,
725
+ "loss": 0.1071,
726
+ "rewards/accuracies": 0.9750000238418579,
727
+ "rewards/chosen": -1.25,
728
+ "rewards/margins": 4.84375,
729
+ "rewards/rejected": -6.09375,
730
+ "step": 460
731
+ },
732
+ {
733
+ "epoch": 1.5064102564102564,
734
+ "grad_norm": 20.51874312804957,
735
+ "learning_rate": 2.7672209026128263e-07,
736
+ "logits/chosen": -1.5625,
737
+ "logits/rejected": -1.53125,
738
+ "logps/chosen": -132.0,
739
+ "logps/rejected": -231.0,
740
+ "loss": 0.1012,
741
+ "rewards/accuracies": 0.9750000238418579,
742
+ "rewards/chosen": -1.015625,
743
+ "rewards/margins": 4.65625,
744
+ "rewards/rejected": -5.6875,
745
+ "step": 470
746
+ },
747
+ {
748
+ "epoch": 1.5384615384615383,
749
+ "grad_norm": 28.03019085733964,
750
+ "learning_rate": 2.7078384798099764e-07,
751
+ "logits/chosen": -1.4453125,
752
+ "logits/rejected": -1.4609375,
753
+ "logps/chosen": -104.5,
754
+ "logps/rejected": -203.0,
755
+ "loss": 0.1074,
756
+ "rewards/accuracies": 0.987500011920929,
757
+ "rewards/chosen": -1.6796875,
758
+ "rewards/margins": 4.59375,
759
+ "rewards/rejected": -6.3125,
760
+ "step": 480
761
+ },
762
+ {
763
+ "epoch": 1.5705128205128205,
764
+ "grad_norm": 14.674518317046845,
765
+ "learning_rate": 2.648456057007126e-07,
766
+ "logits/chosen": -1.5625,
767
+ "logits/rejected": -1.5390625,
768
+ "logps/chosen": -139.0,
769
+ "logps/rejected": -233.0,
770
+ "loss": 0.1003,
771
+ "rewards/accuracies": 0.987500011920929,
772
+ "rewards/chosen": -1.3125,
773
+ "rewards/margins": 4.9375,
774
+ "rewards/rejected": -6.28125,
775
+ "step": 490
776
+ },
777
+ {
778
+ "epoch": 1.6025641025641026,
779
+ "grad_norm": 18.731133870495857,
780
+ "learning_rate": 2.589073634204275e-07,
781
+ "logits/chosen": -1.5703125,
782
+ "logits/rejected": -1.5625,
783
+ "logps/chosen": -144.0,
784
+ "logps/rejected": -251.0,
785
+ "loss": 0.1228,
786
+ "rewards/accuracies": 0.949999988079071,
787
+ "rewards/chosen": -1.109375,
788
+ "rewards/margins": 5.25,
789
+ "rewards/rejected": -6.375,
790
+ "step": 500
791
+ },
792
+ {
793
+ "epoch": 1.6346153846153846,
794
+ "grad_norm": 16.81631331394118,
795
+ "learning_rate": 2.529691211401425e-07,
796
+ "logits/chosen": -1.5625,
797
+ "logits/rejected": -1.5,
798
+ "logps/chosen": -131.0,
799
+ "logps/rejected": -230.0,
800
+ "loss": 0.0819,
801
+ "rewards/accuracies": 0.987500011920929,
802
+ "rewards/chosen": -0.8359375,
803
+ "rewards/margins": 5.375,
804
+ "rewards/rejected": -6.21875,
805
+ "step": 510
806
+ },
807
+ {
808
+ "epoch": 1.6666666666666665,
809
+ "grad_norm": 15.111001669085608,
810
+ "learning_rate": 2.4703087885985747e-07,
811
+ "logits/chosen": -1.546875,
812
+ "logits/rejected": -1.4921875,
813
+ "logps/chosen": -114.0,
814
+ "logps/rejected": -202.0,
815
+ "loss": 0.1091,
816
+ "rewards/accuracies": 0.9750000238418579,
817
+ "rewards/chosen": -0.87109375,
818
+ "rewards/margins": 4.78125,
819
+ "rewards/rejected": -5.625,
820
+ "step": 520
821
+ },
822
+ {
823
+ "epoch": 1.6987179487179487,
824
+ "grad_norm": 26.75975769912583,
825
+ "learning_rate": 2.410926365795724e-07,
826
+ "logits/chosen": -1.5625,
827
+ "logits/rejected": -1.515625,
828
+ "logps/chosen": -140.0,
829
+ "logps/rejected": -221.0,
830
+ "loss": 0.1139,
831
+ "rewards/accuracies": 0.9750000238418579,
832
+ "rewards/chosen": -1.0625,
833
+ "rewards/margins": 4.8125,
834
+ "rewards/rejected": -5.875,
835
+ "step": 530
836
+ },
837
+ {
838
+ "epoch": 1.7307692307692308,
839
+ "grad_norm": 13.759559493772942,
840
+ "learning_rate": 2.351543942992874e-07,
841
+ "logits/chosen": -1.578125,
842
+ "logits/rejected": -1.5,
843
+ "logps/chosen": -143.0,
844
+ "logps/rejected": -229.0,
845
+ "loss": 0.0978,
846
+ "rewards/accuracies": 0.9750000238418579,
847
+ "rewards/chosen": -0.9765625,
848
+ "rewards/margins": 5.09375,
849
+ "rewards/rejected": -6.0625,
850
+ "step": 540
851
+ },
852
+ {
853
+ "epoch": 1.7628205128205128,
854
+ "grad_norm": 18.98900148314797,
855
+ "learning_rate": 2.2921615201900234e-07,
856
+ "logits/chosen": -1.4921875,
857
+ "logits/rejected": -1.5,
858
+ "logps/chosen": -114.0,
859
+ "logps/rejected": -232.0,
860
+ "loss": 0.1103,
861
+ "rewards/accuracies": 0.9624999761581421,
862
+ "rewards/chosen": -0.6953125,
863
+ "rewards/margins": 5.03125,
864
+ "rewards/rejected": -5.71875,
865
+ "step": 550
866
+ },
867
+ {
868
+ "epoch": 1.7948717948717947,
869
+ "grad_norm": 7.478935072745169,
870
+ "learning_rate": 2.2327790973871732e-07,
871
+ "logits/chosen": -1.515625,
872
+ "logits/rejected": -1.4375,
873
+ "logps/chosen": -112.0,
874
+ "logps/rejected": -177.0,
875
+ "loss": 0.1139,
876
+ "rewards/accuracies": 0.9375,
877
+ "rewards/chosen": -0.6640625,
878
+ "rewards/margins": 4.5625,
879
+ "rewards/rejected": -5.21875,
880
+ "step": 560
881
+ },
882
+ {
883
+ "epoch": 1.8269230769230769,
884
+ "grad_norm": 13.728806584543696,
885
+ "learning_rate": 2.173396674584323e-07,
886
+ "logits/chosen": -1.546875,
887
+ "logits/rejected": -1.5546875,
888
+ "logps/chosen": -130.0,
889
+ "logps/rejected": -239.0,
890
+ "loss": 0.1355,
891
+ "rewards/accuracies": 0.949999988079071,
892
+ "rewards/chosen": -0.466796875,
893
+ "rewards/margins": 4.875,
894
+ "rewards/rejected": -5.34375,
895
+ "step": 570
896
+ },
897
+ {
898
+ "epoch": 1.858974358974359,
899
+ "grad_norm": 9.926742756220793,
900
+ "learning_rate": 2.1140142517814726e-07,
901
+ "logits/chosen": -1.609375,
902
+ "logits/rejected": -1.5078125,
903
+ "logps/chosen": -158.0,
904
+ "logps/rejected": -202.0,
905
+ "loss": 0.1118,
906
+ "rewards/accuracies": 0.949999988079071,
907
+ "rewards/chosen": -1.09375,
908
+ "rewards/margins": 4.21875,
909
+ "rewards/rejected": -5.3125,
910
+ "step": 580
911
+ },
912
+ {
913
+ "epoch": 1.891025641025641,
914
+ "grad_norm": 12.431864869622972,
915
+ "learning_rate": 2.0546318289786222e-07,
916
+ "logits/chosen": -1.5546875,
917
+ "logits/rejected": -1.53125,
918
+ "logps/chosen": -122.0,
919
+ "logps/rejected": -202.0,
920
+ "loss": 0.1325,
921
+ "rewards/accuracies": 0.987500011920929,
922
+ "rewards/chosen": -0.61328125,
923
+ "rewards/margins": 4.65625,
924
+ "rewards/rejected": -5.28125,
925
+ "step": 590
926
+ },
927
+ {
928
+ "epoch": 1.9230769230769231,
929
+ "grad_norm": 12.291656293105886,
930
+ "learning_rate": 1.9952494061757718e-07,
931
+ "logits/chosen": -1.5703125,
932
+ "logits/rejected": -1.5625,
933
+ "logps/chosen": -123.0,
934
+ "logps/rejected": -227.0,
935
+ "loss": 0.1085,
936
+ "rewards/accuracies": 0.9624999761581421,
937
+ "rewards/chosen": -0.67578125,
938
+ "rewards/margins": 4.8125,
939
+ "rewards/rejected": -5.5,
940
+ "step": 600
941
+ },
942
+ {
943
+ "epoch": 1.9551282051282053,
944
+ "grad_norm": 6.534649119157792,
945
+ "learning_rate": 1.9358669833729216e-07,
946
+ "logits/chosen": -1.5546875,
947
+ "logits/rejected": -1.484375,
948
+ "logps/chosen": -122.0,
949
+ "logps/rejected": -202.0,
950
+ "loss": 0.0935,
951
+ "rewards/accuracies": 0.949999988079071,
952
+ "rewards/chosen": -0.56640625,
953
+ "rewards/margins": 5.0,
954
+ "rewards/rejected": -5.59375,
955
+ "step": 610
956
+ },
957
+ {
958
+ "epoch": 1.9871794871794872,
959
+ "grad_norm": 10.91488724265466,
960
+ "learning_rate": 1.876484560570071e-07,
961
+ "logits/chosen": -1.5703125,
962
+ "logits/rejected": -1.515625,
963
+ "logps/chosen": -139.0,
964
+ "logps/rejected": -236.0,
965
+ "loss": 0.1071,
966
+ "rewards/accuracies": 0.9624999761581421,
967
+ "rewards/chosen": -1.1953125,
968
+ "rewards/margins": 5.34375,
969
+ "rewards/rejected": -6.53125,
970
+ "step": 620
971
+ },
972
+ {
973
+ "epoch": 2.0,
974
+ "eval_logits/chosen": -1.5,
975
+ "eval_logits/rejected": -1.46875,
976
+ "eval_logps/chosen": -188.0,
977
+ "eval_logps/rejected": -206.0,
978
+ "eval_loss": 0.700976550579071,
979
+ "eval_rewards/accuracies": 0.6428571343421936,
980
+ "eval_rewards/chosen": -3.296875,
981
+ "eval_rewards/margins": 1.3046875,
982
+ "eval_rewards/rejected": -4.59375,
983
+ "eval_runtime": 15.0135,
984
+ "eval_samples_per_second": 13.321,
985
+ "eval_steps_per_second": 0.466,
986
+ "step": 624
987
  }
988
  ],
989
  "logging_steps": 10,