| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 2.988593155893536, | |
| "eval_steps": 50, | |
| "global_step": 393, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.07604562737642585, | |
| "grad_norm": 10.680026826860546, | |
| "learning_rate": 1.25e-07, | |
| "logits/chosen": -2.563544511795044, | |
| "logits/rejected": -2.4952266216278076, | |
| "logps/chosen": -273.53424072265625, | |
| "logps/rejected": -239.1501922607422, | |
| "loss": 0.6932, | |
| "rewards/accuracies": 0.38749998807907104, | |
| "rewards/chosen": 0.00019869348034262657, | |
| "rewards/margins": -1.28610986394051e-06, | |
| "rewards/rejected": 0.00019997954950667918, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.1520912547528517, | |
| "grad_norm": 8.993807887778436, | |
| "learning_rate": 2.5e-07, | |
| "logits/chosen": -2.580817222595215, | |
| "logits/rejected": -2.462310314178467, | |
| "logps/chosen": -269.5621337890625, | |
| "logps/rejected": -258.239990234375, | |
| "loss": 0.6892, | |
| "rewards/accuracies": 0.7437499761581421, | |
| "rewards/chosen": 0.00650571146979928, | |
| "rewards/margins": 0.008498361334204674, | |
| "rewards/rejected": -0.0019926498644053936, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.22813688212927757, | |
| "grad_norm": 8.92132470459759, | |
| "learning_rate": 3.75e-07, | |
| "logits/chosen": -2.6156678199768066, | |
| "logits/rejected": -2.4716460704803467, | |
| "logps/chosen": -295.3561096191406, | |
| "logps/rejected": -241.26644897460938, | |
| "loss": 0.6709, | |
| "rewards/accuracies": 0.84375, | |
| "rewards/chosen": 0.040296901017427444, | |
| "rewards/margins": 0.05595223233103752, | |
| "rewards/rejected": -0.015655336901545525, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.3041825095057034, | |
| "grad_norm": 9.266525353467978, | |
| "learning_rate": 5e-07, | |
| "logits/chosen": -2.6184380054473877, | |
| "logits/rejected": -2.513917922973633, | |
| "logps/chosen": -254.7106475830078, | |
| "logps/rejected": -234.47219848632812, | |
| "loss": 0.6239, | |
| "rewards/accuracies": 0.8125, | |
| "rewards/chosen": 0.028813377022743225, | |
| "rewards/margins": 0.14417138695716858, | |
| "rewards/rejected": -0.11535801738500595, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.38022813688212925, | |
| "grad_norm": 12.173416216409002, | |
| "learning_rate": 4.990105959637203e-07, | |
| "logits/chosen": -2.546933650970459, | |
| "logits/rejected": -2.4483542442321777, | |
| "logps/chosen": -243.9954376220703, | |
| "logps/rejected": -256.17242431640625, | |
| "loss": 0.5554, | |
| "rewards/accuracies": 0.800000011920929, | |
| "rewards/chosen": -0.08808934688568115, | |
| "rewards/margins": 0.2857440114021301, | |
| "rewards/rejected": -0.3738333582878113, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.38022813688212925, | |
| "eval_logits/chosen": -2.62440824508667, | |
| "eval_logits/rejected": -2.5786266326904297, | |
| "eval_logps/chosen": -276.6687316894531, | |
| "eval_logps/rejected": -290.07366943359375, | |
| "eval_loss": 0.6140323281288147, | |
| "eval_rewards/accuracies": 0.6896551847457886, | |
| "eval_rewards/chosen": -0.16707415878772736, | |
| "eval_rewards/margins": 0.2508907914161682, | |
| "eval_rewards/rejected": -0.4179649353027344, | |
| "eval_runtime": 94.0274, | |
| "eval_samples_per_second": 19.473, | |
| "eval_steps_per_second": 0.308, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.45627376425855515, | |
| "grad_norm": 24.13668429607915, | |
| "learning_rate": 4.960502152176573e-07, | |
| "logits/chosen": -2.1383702754974365, | |
| "logits/rejected": -2.0003161430358887, | |
| "logps/chosen": -272.81854248046875, | |
| "logps/rejected": -323.1360168457031, | |
| "loss": 0.4773, | |
| "rewards/accuracies": 0.8812500238418579, | |
| "rewards/chosen": -0.23167654871940613, | |
| "rewards/margins": 0.8607891798019409, | |
| "rewards/rejected": -1.0924659967422485, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.532319391634981, | |
| "grad_norm": 40.013780890248455, | |
| "learning_rate": 4.911422898630837e-07, | |
| "logits/chosen": -1.2432479858398438, | |
| "logits/rejected": -0.6645946502685547, | |
| "logps/chosen": -334.7701721191406, | |
| "logps/rejected": -392.15496826171875, | |
| "loss": 0.3889, | |
| "rewards/accuracies": 0.856249988079071, | |
| "rewards/chosen": -0.5091000199317932, | |
| "rewards/margins": 1.100501298904419, | |
| "rewards/rejected": -1.6096012592315674, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.6083650190114068, | |
| "grad_norm": 31.64851560561826, | |
| "learning_rate": 4.84325667269244e-07, | |
| "logits/chosen": 0.15650656819343567, | |
| "logits/rejected": 0.751736044883728, | |
| "logps/chosen": -311.51922607421875, | |
| "logps/rejected": -406.87860107421875, | |
| "loss": 0.3708, | |
| "rewards/accuracies": 0.84375, | |
| "rewards/chosen": -0.7787342667579651, | |
| "rewards/margins": 1.0774469375610352, | |
| "rewards/rejected": -1.8561809062957764, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.6844106463878327, | |
| "grad_norm": 39.73186343074897, | |
| "learning_rate": 4.7565430258740336e-07, | |
| "logits/chosen": 0.31803542375564575, | |
| "logits/rejected": 1.7618499994277954, | |
| "logps/chosen": -330.3303527832031, | |
| "logps/rejected": -454.4091796875, | |
| "loss": 0.3342, | |
| "rewards/accuracies": 0.893750011920929, | |
| "rewards/chosen": -0.8064614534378052, | |
| "rewards/margins": 1.6557552814483643, | |
| "rewards/rejected": -2.46221661567688, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.7604562737642585, | |
| "grad_norm": 26.592175518902643, | |
| "learning_rate": 4.6519683168329195e-07, | |
| "logits/chosen": 0.48759278655052185, | |
| "logits/rejected": 2.2304329872131348, | |
| "logps/chosen": -379.5733947753906, | |
| "logps/rejected": -496.21636962890625, | |
| "loss": 0.3147, | |
| "rewards/accuracies": 0.893750011920929, | |
| "rewards/chosen": -0.8934251070022583, | |
| "rewards/margins": 1.828790307044983, | |
| "rewards/rejected": -2.722215414047241, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.7604562737642585, | |
| "eval_logits/chosen": 0.7404455542564392, | |
| "eval_logits/rejected": 1.5949417352676392, | |
| "eval_logps/chosen": -375.37860107421875, | |
| "eval_logps/rejected": -459.43609619140625, | |
| "eval_loss": 0.5617097020149231, | |
| "eval_rewards/accuracies": 0.732758641242981, | |
| "eval_rewards/chosen": -1.1541731357574463, | |
| "eval_rewards/margins": 0.957415759563446, | |
| "eval_rewards/rejected": -2.111588716506958, | |
| "eval_runtime": 93.8291, | |
| "eval_samples_per_second": 19.514, | |
| "eval_steps_per_second": 0.309, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.8365019011406845, | |
| "grad_norm": 33.75286446177569, | |
| "learning_rate": 4.530360278682841e-07, | |
| "logits/chosen": 0.4949573576450348, | |
| "logits/rejected": 1.976785659790039, | |
| "logps/chosen": -353.00445556640625, | |
| "logps/rejected": -476.89678955078125, | |
| "loss": 0.3642, | |
| "rewards/accuracies": 0.8374999761581421, | |
| "rewards/chosen": -0.8960109949111938, | |
| "rewards/margins": 1.611445665359497, | |
| "rewards/rejected": -2.5074567794799805, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.9125475285171103, | |
| "grad_norm": 25.522512035563206, | |
| "learning_rate": 4.3926814672941166e-07, | |
| "logits/chosen": 0.5252203941345215, | |
| "logits/rejected": 2.202810525894165, | |
| "logps/chosen": -359.41876220703125, | |
| "logps/rejected": -494.12969970703125, | |
| "loss": 0.318, | |
| "rewards/accuracies": 0.8687499761581421, | |
| "rewards/chosen": -1.0113165378570557, | |
| "rewards/margins": 1.6037089824676514, | |
| "rewards/rejected": -2.615025520324707, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.9885931558935361, | |
| "grad_norm": 29.14776533666164, | |
| "learning_rate": 4.240021642440332e-07, | |
| "logits/chosen": -0.04642736166715622, | |
| "logits/rejected": 1.8715919256210327, | |
| "logps/chosen": -352.2441101074219, | |
| "logps/rejected": -486.38970947265625, | |
| "loss": 0.3174, | |
| "rewards/accuracies": 0.875, | |
| "rewards/chosen": -0.9902883768081665, | |
| "rewards/margins": 1.7977688312530518, | |
| "rewards/rejected": -2.7880570888519287, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 1.064638783269962, | |
| "grad_norm": 32.26612817404267, | |
| "learning_rate": 4.073589142096592e-07, | |
| "logits/chosen": 0.5508675575256348, | |
| "logits/rejected": 2.800567150115967, | |
| "logps/chosen": -352.32440185546875, | |
| "logps/rejected": -517.09228515625, | |
| "loss": 0.2352, | |
| "rewards/accuracies": 0.90625, | |
| "rewards/chosen": -0.9236076474189758, | |
| "rewards/margins": 2.010570764541626, | |
| "rewards/rejected": -2.934178590774536, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 1.1406844106463878, | |
| "grad_norm": 24.555758805773927, | |
| "learning_rate": 3.8947013181637624e-07, | |
| "logits/chosen": 0.8321820497512817, | |
| "logits/rejected": 3.650494337081909, | |
| "logps/chosen": -331.7710266113281, | |
| "logps/rejected": -498.3985900878906, | |
| "loss": 0.214, | |
| "rewards/accuracies": 0.8999999761581421, | |
| "rewards/chosen": -0.9579178094863892, | |
| "rewards/margins": 2.1326422691345215, | |
| "rewards/rejected": -3.0905601978302, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.1406844106463878, | |
| "eval_logits/chosen": 0.8076741695404053, | |
| "eval_logits/rejected": 2.0487349033355713, | |
| "eval_logps/chosen": -389.57183837890625, | |
| "eval_logps/rejected": -491.0475158691406, | |
| "eval_loss": 0.5559520125389099, | |
| "eval_rewards/accuracies": 0.7456896305084229, | |
| "eval_rewards/chosen": -1.2961053848266602, | |
| "eval_rewards/margins": 1.1315982341766357, | |
| "eval_rewards/rejected": -2.427703619003296, | |
| "eval_runtime": 93.901, | |
| "eval_samples_per_second": 19.499, | |
| "eval_steps_per_second": 0.309, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.2167300380228137, | |
| "grad_norm": 23.812525631547885, | |
| "learning_rate": 3.7047741093221656e-07, | |
| "logits/chosen": 0.6415907740592957, | |
| "logits/rejected": 3.418233871459961, | |
| "logps/chosen": -337.47711181640625, | |
| "logps/rejected": -565.9837036132812, | |
| "loss": 0.1915, | |
| "rewards/accuracies": 0.9312499761581421, | |
| "rewards/chosen": -0.7956029176712036, | |
| "rewards/margins": 2.5107922554016113, | |
| "rewards/rejected": -3.306394577026367, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 1.2927756653992395, | |
| "grad_norm": 22.027445926013243, | |
| "learning_rate": 3.5053108335480205e-07, | |
| "logits/chosen": 0.43230652809143066, | |
| "logits/rejected": 3.4372222423553467, | |
| "logps/chosen": -404.5794982910156, | |
| "logps/rejected": -585.620849609375, | |
| "loss": 0.2138, | |
| "rewards/accuracies": 0.9624999761581421, | |
| "rewards/chosen": -1.1555755138397217, | |
| "rewards/margins": 2.351931571960449, | |
| "rewards/rejected": -3.50750732421875, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 1.3688212927756653, | |
| "grad_norm": 21.8184530010132, | |
| "learning_rate": 3.29789028900245e-07, | |
| "logits/chosen": 0.6227339506149292, | |
| "logits/rejected": 3.597677707672119, | |
| "logps/chosen": -414.3572692871094, | |
| "logps/rejected": -602.4873657226562, | |
| "loss": 0.2101, | |
| "rewards/accuracies": 0.949999988079071, | |
| "rewards/chosen": -1.1512380838394165, | |
| "rewards/margins": 2.4040679931640625, | |
| "rewards/rejected": -3.5553059577941895, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 1.4448669201520912, | |
| "grad_norm": 30.492941117704806, | |
| "learning_rate": 3.084154257477301e-07, | |
| "logits/chosen": 1.2600494623184204, | |
| "logits/rejected": 3.9732604026794434, | |
| "logps/chosen": -392.4749450683594, | |
| "logps/rejected": -593.7326049804688, | |
| "loss": 0.1835, | |
| "rewards/accuracies": 0.893750011920929, | |
| "rewards/chosen": -1.3306286334991455, | |
| "rewards/margins": 2.4590394496917725, | |
| "rewards/rejected": -3.7896676063537598, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 1.5209125475285172, | |
| "grad_norm": 18.187832229568688, | |
| "learning_rate": 2.865794509310888e-07, | |
| "logits/chosen": 1.334993839263916, | |
| "logits/rejected": 4.3924736976623535, | |
| "logps/chosen": -390.69781494140625, | |
| "logps/rejected": -616.49951171875, | |
| "loss": 0.1866, | |
| "rewards/accuracies": 0.9375, | |
| "rewards/chosen": -1.2437045574188232, | |
| "rewards/margins": 2.7270522117614746, | |
| "rewards/rejected": -3.9707565307617188, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.5209125475285172, | |
| "eval_logits/chosen": 0.7928547859191895, | |
| "eval_logits/rejected": 2.0297083854675293, | |
| "eval_logps/chosen": -404.8973388671875, | |
| "eval_logps/rejected": -510.6939697265625, | |
| "eval_loss": 0.5364186763763428, | |
| "eval_rewards/accuracies": 0.7413793206214905, | |
| "eval_rewards/chosen": -1.4493602514266968, | |
| "eval_rewards/margins": 1.1748069524765015, | |
| "eval_rewards/rejected": -2.624166965484619, | |
| "eval_runtime": 93.5108, | |
| "eval_samples_per_second": 19.581, | |
| "eval_steps_per_second": 0.31, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 1.5969581749049429, | |
| "grad_norm": 30.203312935912965, | |
| "learning_rate": 2.644539412632786e-07, | |
| "logits/chosen": 0.7376717329025269, | |
| "logits/rejected": 3.46760630607605, | |
| "logps/chosen": -381.1366882324219, | |
| "logps/rejected": -576.8548583984375, | |
| "loss": 0.2112, | |
| "rewards/accuracies": 0.918749988079071, | |
| "rewards/chosen": -1.2630369663238525, | |
| "rewards/margins": 2.4375810623168945, | |
| "rewards/rejected": -3.700617551803589, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 1.673003802281369, | |
| "grad_norm": 34.60056674030066, | |
| "learning_rate": 2.422140252928601e-07, | |
| "logits/chosen": 1.3862824440002441, | |
| "logits/rejected": 3.9012253284454346, | |
| "logps/chosen": -363.8896484375, | |
| "logps/rejected": -638.3570556640625, | |
| "loss": 0.2026, | |
| "rewards/accuracies": 0.949999988079071, | |
| "rewards/chosen": -1.3880947828292847, | |
| "rewards/margins": 2.823024272918701, | |
| "rewards/rejected": -4.211119174957275, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 1.7490494296577945, | |
| "grad_norm": 17.75951807278943, | |
| "learning_rate": 2.2003573712085455e-07, | |
| "logits/chosen": 0.32914024591445923, | |
| "logits/rejected": 3.113135576248169, | |
| "logps/chosen": -353.28631591796875, | |
| "logps/rejected": -597.5838012695312, | |
| "loss": 0.1802, | |
| "rewards/accuracies": 0.981249988079071, | |
| "rewards/chosen": -1.1312899589538574, | |
| "rewards/margins": 2.7755885124206543, | |
| "rewards/rejected": -3.906878709793091, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 1.8250950570342206, | |
| "grad_norm": 21.2237904198368, | |
| "learning_rate": 1.980946230499431e-07, | |
| "logits/chosen": 0.5761924982070923, | |
| "logits/rejected": 3.6894748210906982, | |
| "logps/chosen": -369.281982421875, | |
| "logps/rejected": -639.9312133789062, | |
| "loss": 0.1863, | |
| "rewards/accuracies": 0.9624999761581421, | |
| "rewards/chosen": -1.4418867826461792, | |
| "rewards/margins": 2.867917776107788, | |
| "rewards/rejected": -4.309804439544678, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 1.9011406844106464, | |
| "grad_norm": 25.460606683183755, | |
| "learning_rate": 1.7656435209470376e-07, | |
| "logits/chosen": 1.0737273693084717, | |
| "logits/rejected": 3.9221127033233643, | |
| "logps/chosen": -389.84417724609375, | |
| "logps/rejected": -633.5592041015625, | |
| "loss": 0.1899, | |
| "rewards/accuracies": 0.9375, | |
| "rewards/chosen": -1.4279615879058838, | |
| "rewards/margins": 2.770998954772949, | |
| "rewards/rejected": -4.198960304260254, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 1.9011406844106464, | |
| "eval_logits/chosen": 1.2524254322052002, | |
| "eval_logits/rejected": 2.571428060531616, | |
| "eval_logps/chosen": -438.79302978515625, | |
| "eval_logps/rejected": -556.13232421875, | |
| "eval_loss": 0.5391203165054321, | |
| "eval_rewards/accuracies": 0.7456896305084229, | |
| "eval_rewards/chosen": -1.7883172035217285, | |
| "eval_rewards/margins": 1.2902345657348633, | |
| "eval_rewards/rejected": -3.078551769256592, | |
| "eval_runtime": 96.6165, | |
| "eval_samples_per_second": 18.951, | |
| "eval_steps_per_second": 0.3, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 1.9771863117870723, | |
| "grad_norm": 28.509116337681686, | |
| "learning_rate": 1.5561534135101884e-07, | |
| "logits/chosen": 1.822516679763794, | |
| "logits/rejected": 4.293866157531738, | |
| "logps/chosen": -417.22503662109375, | |
| "logps/rejected": -664.3339233398438, | |
| "loss": 0.1979, | |
| "rewards/accuracies": 0.9375, | |
| "rewards/chosen": -1.6485140323638916, | |
| "rewards/margins": 2.746192455291748, | |
| "rewards/rejected": -4.394707202911377, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 2.053231939163498, | |
| "grad_norm": 9.761431912302397, | |
| "learning_rate": 1.3541340710517546e-07, | |
| "logits/chosen": 1.401188611984253, | |
| "logits/rejected": 4.360350608825684, | |
| "logps/chosen": -416.37481689453125, | |
| "logps/rejected": -680.0263671875, | |
| "loss": 0.1217, | |
| "rewards/accuracies": 0.9375, | |
| "rewards/chosen": -1.4166698455810547, | |
| "rewards/margins": 2.9713897705078125, | |
| "rewards/rejected": -4.388059616088867, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 2.129277566539924, | |
| "grad_norm": 19.479793936953538, | |
| "learning_rate": 1.1611845235944143e-07, | |
| "logits/chosen": 2.3020176887512207, | |
| "logits/rejected": 5.1446309089660645, | |
| "logps/chosen": -407.40936279296875, | |
| "logps/rejected": -718.0748901367188, | |
| "loss": 0.1082, | |
| "rewards/accuracies": 0.981249988079071, | |
| "rewards/chosen": -1.5115373134613037, | |
| "rewards/margins": 3.4099037647247314, | |
| "rewards/rejected": -4.921441555023193, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 2.20532319391635, | |
| "grad_norm": 12.421604139145975, | |
| "learning_rate": 9.788320116265892e-08, | |
| "logits/chosen": 1.9953467845916748, | |
| "logits/rejected": 5.533808708190918, | |
| "logps/chosen": -459.0770568847656, | |
| "logps/rejected": -714.7064208984375, | |
| "loss": 0.112, | |
| "rewards/accuracies": 0.9624999761581421, | |
| "rewards/chosen": -1.560991883277893, | |
| "rewards/margins": 3.3094687461853027, | |
| "rewards/rejected": -4.8704609870910645, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 2.2813688212927756, | |
| "grad_norm": 17.541719244774725, | |
| "learning_rate": 8.085198976392124e-08, | |
| "logits/chosen": 2.661414384841919, | |
| "logits/rejected": 5.309275150299072, | |
| "logps/chosen": -396.86163330078125, | |
| "logps/rejected": -682.7183837890625, | |
| "loss": 0.1083, | |
| "rewards/accuracies": 0.956250011920929, | |
| "rewards/chosen": -1.661924958229065, | |
| "rewards/margins": 3.159823179244995, | |
| "rewards/rejected": -4.821747779846191, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.2813688212927756, | |
| "eval_logits/chosen": 2.2903313636779785, | |
| "eval_logits/rejected": 3.6920409202575684, | |
| "eval_logps/chosen": -470.052734375, | |
| "eval_logps/rejected": -601.4120483398438, | |
| "eval_loss": 0.558447539806366, | |
| "eval_rewards/accuracies": 0.7629310488700867, | |
| "eval_rewards/chosen": -2.100914239883423, | |
| "eval_rewards/margins": 1.4304344654083252, | |
| "eval_rewards/rejected": -3.531348466873169, | |
| "eval_runtime": 94.198, | |
| "eval_samples_per_second": 19.438, | |
| "eval_steps_per_second": 0.308, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 2.3574144486692017, | |
| "grad_norm": 11.598664016084259, | |
| "learning_rate": 6.515962415763369e-08, | |
| "logits/chosen": 2.566627025604248, | |
| "logits/rejected": 5.178008079528809, | |
| "logps/chosen": -364.78521728515625, | |
| "logps/rejected": -709.1622314453125, | |
| "loss": 0.0941, | |
| "rewards/accuracies": 0.987500011920929, | |
| "rewards/chosen": -1.6015697717666626, | |
| "rewards/margins": 3.526782989501953, | |
| "rewards/rejected": -5.128352165222168, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 2.4334600760456273, | |
| "grad_norm": 22.496239920192775, | |
| "learning_rate": 5.093031306275308e-08, | |
| "logits/chosen": 2.157808303833008, | |
| "logits/rejected": 5.751224040985107, | |
| "logps/chosen": -454.247314453125, | |
| "logps/rejected": -807.82275390625, | |
| "loss": 0.1048, | |
| "rewards/accuracies": 0.987500011920929, | |
| "rewards/chosen": -1.797298789024353, | |
| "rewards/margins": 3.7231411933898926, | |
| "rewards/rejected": -5.520440101623535, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 2.5095057034220534, | |
| "grad_norm": 13.016382898289478, | |
| "learning_rate": 3.827668478192578e-08, | |
| "logits/chosen": 2.29127836227417, | |
| "logits/rejected": 5.964070796966553, | |
| "logps/chosen": -475.0154724121094, | |
| "logps/rejected": -776.1507568359375, | |
| "loss": 0.0961, | |
| "rewards/accuracies": 0.96875, | |
| "rewards/chosen": -1.8038787841796875, | |
| "rewards/margins": 3.5154175758361816, | |
| "rewards/rejected": -5.319295883178711, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 2.585551330798479, | |
| "grad_norm": 18.59300730528273, | |
| "learning_rate": 2.729889572230856e-08, | |
| "logits/chosen": 2.6335761547088623, | |
| "logits/rejected": 6.27906608581543, | |
| "logps/chosen": -456.11676025390625, | |
| "logps/rejected": -846.2113037109375, | |
| "loss": 0.0807, | |
| "rewards/accuracies": 0.987500011920929, | |
| "rewards/chosen": -2.0658538341522217, | |
| "rewards/margins": 4.058461666107178, | |
| "rewards/rejected": -6.1243157386779785, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 2.661596958174905, | |
| "grad_norm": 13.300072255332687, | |
| "learning_rate": 1.8083837634341766e-08, | |
| "logits/chosen": 2.266738176345825, | |
| "logits/rejected": 6.4141387939453125, | |
| "logps/chosen": -500.92901611328125, | |
| "logps/rejected": -874.1981201171875, | |
| "loss": 0.0955, | |
| "rewards/accuracies": 1.0, | |
| "rewards/chosen": -2.200644016265869, | |
| "rewards/margins": 4.188169956207275, | |
| "rewards/rejected": -6.3888139724731445, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 2.661596958174905, | |
| "eval_logits/chosen": 2.532045602798462, | |
| "eval_logits/rejected": 4.256929397583008, | |
| "eval_logps/chosen": -544.331787109375, | |
| "eval_logps/rejected": -697.9909057617188, | |
| "eval_loss": 0.5829063653945923, | |
| "eval_rewards/accuracies": 0.7629310488700867, | |
| "eval_rewards/chosen": -2.8437047004699707, | |
| "eval_rewards/margins": 1.6534322500228882, | |
| "eval_rewards/rejected": -4.49713659286499, | |
| "eval_runtime": 93.5502, | |
| "eval_samples_per_second": 19.572, | |
| "eval_steps_per_second": 0.31, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 2.7376425855513307, | |
| "grad_norm": 14.902459964324052, | |
| "learning_rate": 1.0704449843359498e-08, | |
| "logits/chosen": 2.454040050506592, | |
| "logits/rejected": 6.299553871154785, | |
| "logps/chosen": -483.86669921875, | |
| "logps/rejected": -877.189453125, | |
| "loss": 0.0847, | |
| "rewards/accuracies": 1.0, | |
| "rewards/chosen": -2.1329433917999268, | |
| "rewards/margins": 4.206452369689941, | |
| "rewards/rejected": -6.3393964767456055, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 2.8136882129277567, | |
| "grad_norm": 15.13819700442962, | |
| "learning_rate": 5.2191419178871935e-09, | |
| "logits/chosen": 2.5856270790100098, | |
| "logits/rejected": 6.600296974182129, | |
| "logps/chosen": -461.5941467285156, | |
| "logps/rejected": -831.21240234375, | |
| "loss": 0.0833, | |
| "rewards/accuracies": 0.96875, | |
| "rewards/chosen": -2.348266363143921, | |
| "rewards/margins": 4.047614574432373, | |
| "rewards/rejected": -6.395880699157715, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 2.8897338403041823, | |
| "grad_norm": 12.919733942557965, | |
| "learning_rate": 1.6713313443384724e-09, | |
| "logits/chosen": 2.7355175018310547, | |
| "logits/rejected": 6.480588436126709, | |
| "logps/chosen": -487.62274169921875, | |
| "logps/rejected": -859.5900268554688, | |
| "loss": 0.0859, | |
| "rewards/accuracies": 0.9750000238418579, | |
| "rewards/chosen": -2.41930890083313, | |
| "rewards/margins": 3.938915967941284, | |
| "rewards/rejected": -6.358224391937256, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 2.9657794676806084, | |
| "grad_norm": 10.73491870552627, | |
| "learning_rate": 8.909986752470012e-11, | |
| "logits/chosen": 2.8181276321411133, | |
| "logits/rejected": 6.899574279785156, | |
| "logps/chosen": -519.549560546875, | |
| "logps/rejected": -837.3341674804688, | |
| "loss": 0.0796, | |
| "rewards/accuracies": 0.9624999761581421, | |
| "rewards/chosen": -2.500697612762451, | |
| "rewards/margins": 3.8103880882263184, | |
| "rewards/rejected": -6.3110857009887695, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 2.988593155893536, | |
| "step": 393, | |
| "total_flos": 0.0, | |
| "train_loss": 0.2546291965564699, | |
| "train_runtime": 10353.4149, | |
| "train_samples_per_second": 4.868, | |
| "train_steps_per_second": 0.038 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 393, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 100, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 0.0, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |