| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.0, | |
| "eval_steps": 100, | |
| "global_step": 385, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.0, | |
| "learning_rate": 1.282051282051282e-07, | |
| "logits/chosen": -1.7278180122375488, | |
| "logits/rejected": -1.7377450466156006, | |
| "logps/chosen": -29.553977966308594, | |
| "logps/rejected": -42.813133239746094, | |
| "loss": 0.3906, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "learning_rate": 1.282051282051282e-06, | |
| "logits/chosen": -1.8662821054458618, | |
| "logits/rejected": -1.8706018924713135, | |
| "logps/chosen": -36.98260498046875, | |
| "logps/rejected": -33.66376876831055, | |
| "loss": 0.3501, | |
| "rewards/accuracies": 0.5694444179534912, | |
| "rewards/chosen": 0.01919599249958992, | |
| "rewards/margins": 0.04614981636404991, | |
| "rewards/rejected": -0.02695382386445999, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "learning_rate": 2.564102564102564e-06, | |
| "logits/chosen": -1.9981708526611328, | |
| "logits/rejected": -2.000824451446533, | |
| "logps/chosen": -29.6453914642334, | |
| "logps/rejected": -29.056461334228516, | |
| "loss": 0.4314, | |
| "rewards/accuracies": 0.48750001192092896, | |
| "rewards/chosen": -0.0025594178587198257, | |
| "rewards/margins": -0.008148794062435627, | |
| "rewards/rejected": 0.005589376203715801, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "learning_rate": 3.846153846153847e-06, | |
| "logits/chosen": -1.9205211400985718, | |
| "logits/rejected": -1.917824149131775, | |
| "logps/chosen": -31.402332305908203, | |
| "logps/rejected": -33.20569610595703, | |
| "loss": 0.4366, | |
| "rewards/accuracies": 0.574999988079071, | |
| "rewards/chosen": 0.011007689870893955, | |
| "rewards/margins": 0.0007972270250320435, | |
| "rewards/rejected": 0.010210464708507061, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "learning_rate": 4.999896948438434e-06, | |
| "logits/chosen": -2.0170958042144775, | |
| "logits/rejected": -2.0083703994750977, | |
| "logps/chosen": -32.547943115234375, | |
| "logps/rejected": -32.49810791015625, | |
| "loss": 0.4286, | |
| "rewards/accuracies": 0.5375000238418579, | |
| "rewards/chosen": 0.023093106225132942, | |
| "rewards/margins": 0.011441526934504509, | |
| "rewards/rejected": 0.011651577427983284, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "learning_rate": 4.987541037542187e-06, | |
| "logits/chosen": -1.8613898754119873, | |
| "logits/rejected": -1.8506320714950562, | |
| "logps/chosen": -33.56399154663086, | |
| "logps/rejected": -35.46310043334961, | |
| "loss": 0.4498, | |
| "rewards/accuracies": 0.4625000059604645, | |
| "rewards/chosen": -0.005208463408052921, | |
| "rewards/margins": 0.0033974028192460537, | |
| "rewards/rejected": -0.008605867624282837, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "learning_rate": 4.954691471941119e-06, | |
| "logits/chosen": -1.9389715194702148, | |
| "logits/rejected": -1.9409143924713135, | |
| "logps/chosen": -32.57710266113281, | |
| "logps/rejected": -33.21430206298828, | |
| "loss": 0.3914, | |
| "rewards/accuracies": 0.612500011920929, | |
| "rewards/chosen": 0.018466468900442123, | |
| "rewards/margins": 0.05280427262187004, | |
| "rewards/rejected": -0.03433779999613762, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "learning_rate": 4.901618883413549e-06, | |
| "logits/chosen": -2.0723679065704346, | |
| "logits/rejected": -2.0773472785949707, | |
| "logps/chosen": -33.946128845214844, | |
| "logps/rejected": -36.60929489135742, | |
| "loss": 0.4672, | |
| "rewards/accuracies": 0.5, | |
| "rewards/chosen": 0.033884815871715546, | |
| "rewards/margins": 0.0662197396159172, | |
| "rewards/rejected": -0.03233493119478226, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "learning_rate": 4.828760511501322e-06, | |
| "logits/chosen": -1.9366763830184937, | |
| "logits/rejected": -1.9398372173309326, | |
| "logps/chosen": -34.32460403442383, | |
| "logps/rejected": -34.59550094604492, | |
| "loss": 0.3932, | |
| "rewards/accuracies": 0.5625, | |
| "rewards/chosen": 0.0825641006231308, | |
| "rewards/margins": 0.08503931760787964, | |
| "rewards/rejected": -0.002475212560966611, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "learning_rate": 4.7367166013034295e-06, | |
| "logits/chosen": -1.9469963312149048, | |
| "logits/rejected": -1.9515202045440674, | |
| "logps/chosen": -32.401084899902344, | |
| "logps/rejected": -32.32483673095703, | |
| "loss": 0.4402, | |
| "rewards/accuracies": 0.574999988079071, | |
| "rewards/chosen": 0.05306895822286606, | |
| "rewards/margins": 0.02581069990992546, | |
| "rewards/rejected": 0.027258265763521194, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "learning_rate": 4.626245458345211e-06, | |
| "logits/chosen": -2.0446877479553223, | |
| "logits/rejected": -2.0427000522613525, | |
| "logps/chosen": -32.168331146240234, | |
| "logps/rejected": -31.263940811157227, | |
| "loss": 0.3817, | |
| "rewards/accuracies": 0.5375000238418579, | |
| "rewards/chosen": 0.05800473690032959, | |
| "rewards/margins": 0.056188035756349564, | |
| "rewards/rejected": 0.0018166989320889115, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "eval_logits/chosen": -2.2377800941467285, | |
| "eval_logits/rejected": -2.2329328060150146, | |
| "eval_logps/chosen": -33.97116470336914, | |
| "eval_logps/rejected": -37.47990036010742, | |
| "eval_loss": 0.42868927121162415, | |
| "eval_rewards/accuracies": 0.5577242374420166, | |
| "eval_rewards/chosen": 0.05071057379245758, | |
| "eval_rewards/margins": 0.021336428821086884, | |
| "eval_rewards/rejected": 0.029374146834015846, | |
| "eval_runtime": 146.0732, | |
| "eval_samples_per_second": 2.348, | |
| "eval_steps_per_second": 0.294, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 4.498257201263691e-06, | |
| "logits/chosen": -1.9993603229522705, | |
| "logits/rejected": -1.996971845626831, | |
| "logps/chosen": -33.112144470214844, | |
| "logps/rejected": -33.986610412597656, | |
| "loss": 0.5297, | |
| "rewards/accuracies": 0.574999988079071, | |
| "rewards/chosen": 0.105336032807827, | |
| "rewards/margins": 0.07112538814544678, | |
| "rewards/rejected": 0.03421063348650932, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "learning_rate": 4.353806263777678e-06, | |
| "logits/chosen": -2.009766101837158, | |
| "logits/rejected": -2.0014333724975586, | |
| "logps/chosen": -32.351280212402344, | |
| "logps/rejected": -32.099830627441406, | |
| "loss": 0.5288, | |
| "rewards/accuracies": 0.550000011920929, | |
| "rewards/chosen": 0.07490243762731552, | |
| "rewards/margins": 0.025563379749655724, | |
| "rewards/rejected": 0.04933905601501465, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.34, | |
| "learning_rate": 4.1940827077152755e-06, | |
| "logits/chosen": -2.0372567176818848, | |
| "logits/rejected": -2.029240369796753, | |
| "logps/chosen": -30.34758949279785, | |
| "logps/rejected": -32.03752899169922, | |
| "loss": 0.4171, | |
| "rewards/accuracies": 0.550000011920929, | |
| "rewards/chosen": 0.10250232368707657, | |
| "rewards/margins": 0.09011942148208618, | |
| "rewards/rejected": 0.01238289289176464, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "learning_rate": 4.0204024186666215e-06, | |
| "logits/chosen": -1.966684341430664, | |
| "logits/rejected": -1.9769500494003296, | |
| "logps/chosen": -31.219501495361328, | |
| "logps/rejected": -32.558082580566406, | |
| "loss": 0.4241, | |
| "rewards/accuracies": 0.675000011920929, | |
| "rewards/chosen": 0.1601306051015854, | |
| "rewards/margins": 0.178691104054451, | |
| "rewards/rejected": -0.018560513854026794, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "learning_rate": 3.834196265035119e-06, | |
| "logits/chosen": -1.8799800872802734, | |
| "logits/rejected": -1.8811286687850952, | |
| "logps/chosen": -34.00492858886719, | |
| "logps/rejected": -34.78579330444336, | |
| "loss": 0.3742, | |
| "rewards/accuracies": 0.625, | |
| "rewards/chosen": 0.15021029114723206, | |
| "rewards/margins": 0.17304366827011108, | |
| "rewards/rejected": -0.02283337712287903, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "learning_rate": 3.636998309800573e-06, | |
| "logits/chosen": -1.932443618774414, | |
| "logits/rejected": -1.9290469884872437, | |
| "logps/chosen": -36.061256408691406, | |
| "logps/rejected": -32.7225456237793, | |
| "loss": 0.377, | |
| "rewards/accuracies": 0.625, | |
| "rewards/chosen": 0.0883389413356781, | |
| "rewards/margins": 0.0881747230887413, | |
| "rewards/rejected": 0.00016421750478912145, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.44, | |
| "learning_rate": 3.4304331721118078e-06, | |
| "logits/chosen": -2.0334668159484863, | |
| "logits/rejected": -2.026076078414917, | |
| "logps/chosen": -33.546730041503906, | |
| "logps/rejected": -31.355152130126953, | |
| "loss": 0.3457, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": 0.2009061872959137, | |
| "rewards/margins": 0.2035447061061859, | |
| "rewards/rejected": -0.0026385621167719364, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.47, | |
| "learning_rate": 3.2162026428305436e-06, | |
| "logits/chosen": -2.0394930839538574, | |
| "logits/rejected": -2.044752359390259, | |
| "logps/chosen": -32.33013153076172, | |
| "logps/rejected": -32.470909118652344, | |
| "loss": 0.2919, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": 0.17172439396381378, | |
| "rewards/margins": 0.1603413224220276, | |
| "rewards/rejected": 0.01138305850327015, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.49, | |
| "learning_rate": 2.996071664294641e-06, | |
| "logits/chosen": -2.0403592586517334, | |
| "logits/rejected": -2.0375704765319824, | |
| "logps/chosen": -31.314708709716797, | |
| "logps/rejected": -31.29607582092285, | |
| "loss": 0.4191, | |
| "rewards/accuracies": 0.612500011920929, | |
| "rewards/chosen": 0.13939552009105682, | |
| "rewards/margins": 0.12168798595666885, | |
| "rewards/rejected": 0.017707547172904015, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.52, | |
| "learning_rate": 2.7718537898066833e-06, | |
| "logits/chosen": -1.9107002019882202, | |
| "logits/rejected": -1.915356993675232, | |
| "logps/chosen": -31.39776611328125, | |
| "logps/rejected": -32.7841682434082, | |
| "loss": 0.3991, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": 0.1668148785829544, | |
| "rewards/margins": 0.16140125691890717, | |
| "rewards/rejected": 0.005413623061031103, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.52, | |
| "eval_logits/chosen": -2.234880208969116, | |
| "eval_logits/rejected": -2.2300188541412354, | |
| "eval_logps/chosen": -34.03581619262695, | |
| "eval_logps/rejected": -37.54121017456055, | |
| "eval_loss": 0.4536176025867462, | |
| "eval_rewards/accuracies": 0.5373754501342773, | |
| "eval_rewards/chosen": -0.00101160176564008, | |
| "eval_rewards/margins": 0.01865854486823082, | |
| "eval_rewards/rejected": -0.019670147448778152, | |
| "eval_runtime": 145.4825, | |
| "eval_samples_per_second": 2.358, | |
| "eval_steps_per_second": 0.296, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "learning_rate": 2.5453962426402006e-06, | |
| "logits/chosen": -2.022460460662842, | |
| "logits/rejected": -2.03314471244812, | |
| "logps/chosen": -31.764179229736328, | |
| "logps/rejected": -33.924598693847656, | |
| "loss": 0.2827, | |
| "rewards/accuracies": 0.75, | |
| "rewards/chosen": 0.1656998097896576, | |
| "rewards/margins": 0.21521887183189392, | |
| "rewards/rejected": -0.049519073218107224, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.57, | |
| "learning_rate": 2.3185646976551794e-06, | |
| "logits/chosen": -1.9155023097991943, | |
| "logits/rejected": -1.9303117990493774, | |
| "logps/chosen": -29.97748374938965, | |
| "logps/rejected": -31.563989639282227, | |
| "loss": 0.3625, | |
| "rewards/accuracies": 0.6499999761581421, | |
| "rewards/chosen": 0.10724345594644547, | |
| "rewards/margins": 0.11189230531454086, | |
| "rewards/rejected": -0.004648865200579166, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.6, | |
| "learning_rate": 2.0932279108998323e-06, | |
| "logits/chosen": -1.9727697372436523, | |
| "logits/rejected": -1.9767513275146484, | |
| "logps/chosen": -33.23460006713867, | |
| "logps/rejected": -31.581985473632812, | |
| "loss": 0.3234, | |
| "rewards/accuracies": 0.7250000238418579, | |
| "rewards/chosen": 0.1462947130203247, | |
| "rewards/margins": 0.17244157195091248, | |
| "rewards/rejected": -0.026146870106458664, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.62, | |
| "learning_rate": 1.8712423238279358e-06, | |
| "logits/chosen": -1.971836805343628, | |
| "logits/rejected": -1.9498882293701172, | |
| "logps/chosen": -33.9459228515625, | |
| "logps/rejected": -35.02843475341797, | |
| "loss": 0.3771, | |
| "rewards/accuracies": 0.699999988079071, | |
| "rewards/chosen": 0.1662607342004776, | |
| "rewards/margins": 0.2283717393875122, | |
| "rewards/rejected": -0.0621110200881958, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.65, | |
| "learning_rate": 1.6544367689701824e-06, | |
| "logits/chosen": -2.0139522552490234, | |
| "logits/rejected": -2.010632038116455, | |
| "logps/chosen": -32.76514434814453, | |
| "logps/rejected": -36.231590270996094, | |
| "loss": 0.3601, | |
| "rewards/accuracies": 0.637499988079071, | |
| "rewards/chosen": 0.1279398649930954, | |
| "rewards/margins": 0.1332733929157257, | |
| "rewards/rejected": -0.005333536770194769, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.68, | |
| "learning_rate": 1.4445974030621963e-06, | |
| "logits/chosen": -1.8817164897918701, | |
| "logits/rejected": -1.8792623281478882, | |
| "logps/chosen": -34.03675842285156, | |
| "logps/rejected": -35.52077102661133, | |
| "loss": 0.3873, | |
| "rewards/accuracies": 0.699999988079071, | |
| "rewards/chosen": 0.12131254374980927, | |
| "rewards/margins": 0.13423141837120056, | |
| "rewards/rejected": -0.012918862514197826, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "learning_rate": 1.243452991757889e-06, | |
| "logits/chosen": -1.8678665161132812, | |
| "logits/rejected": -1.8653860092163086, | |
| "logps/chosen": -34.24772262573242, | |
| "logps/rejected": -31.75213050842285, | |
| "loss": 0.3738, | |
| "rewards/accuracies": 0.6499999761581421, | |
| "rewards/chosen": 0.11989933252334595, | |
| "rewards/margins": 0.1045403853058815, | |
| "rewards/rejected": 0.015358942560851574, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.73, | |
| "learning_rate": 1.0526606671603523e-06, | |
| "logits/chosen": -1.9715077877044678, | |
| "logits/rejected": -1.9609073400497437, | |
| "logps/chosen": -35.06243133544922, | |
| "logps/rejected": -31.837047576904297, | |
| "loss": 0.2652, | |
| "rewards/accuracies": 0.7749999761581421, | |
| "rewards/chosen": 0.2353697121143341, | |
| "rewards/margins": 0.2322143018245697, | |
| "rewards/rejected": 0.003155359299853444, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "learning_rate": 8.737922755071455e-07, | |
| "logits/chosen": -2.0671467781066895, | |
| "logits/rejected": -2.052150011062622, | |
| "logps/chosen": -30.7277774810791, | |
| "logps/rejected": -32.63950729370117, | |
| "loss": 0.3926, | |
| "rewards/accuracies": 0.625, | |
| "rewards/chosen": 0.15707775950431824, | |
| "rewards/margins": 0.14721640944480896, | |
| "rewards/rejected": 0.009861335158348083, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "learning_rate": 7.08321427484816e-07, | |
| "logits/chosen": -1.9383538961410522, | |
| "logits/rejected": -1.935782790184021, | |
| "logps/chosen": -32.601158142089844, | |
| "logps/rejected": -30.8317813873291, | |
| "loss": 0.2816, | |
| "rewards/accuracies": 0.75, | |
| "rewards/chosen": 0.26075875759124756, | |
| "rewards/margins": 0.2727365493774414, | |
| "rewards/rejected": -0.011977789923548698, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "eval_logits/chosen": -2.2351884841918945, | |
| "eval_logits/rejected": -2.230333089828491, | |
| "eval_logps/chosen": -34.01213836669922, | |
| "eval_logps/rejected": -37.532142639160156, | |
| "eval_loss": 0.4463596045970917, | |
| "eval_rewards/accuracies": 0.5427741408348083, | |
| "eval_rewards/chosen": 0.017930733039975166, | |
| "eval_rewards/margins": 0.03035038523375988, | |
| "eval_rewards/rejected": -0.012419654987752438, | |
| "eval_runtime": 145.7115, | |
| "eval_samples_per_second": 2.354, | |
| "eval_steps_per_second": 0.295, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.81, | |
| "learning_rate": 5.576113578589035e-07, | |
| "logits/chosen": -1.9204727411270142, | |
| "logits/rejected": -1.9172074794769287, | |
| "logps/chosen": -31.360204696655273, | |
| "logps/rejected": -33.759212493896484, | |
| "loss": 0.3011, | |
| "rewards/accuracies": 0.75, | |
| "rewards/chosen": 0.19451384246349335, | |
| "rewards/margins": 0.20816302299499512, | |
| "rewards/rejected": -0.01364919263869524, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "learning_rate": 4.229036944380913e-07, | |
| "logits/chosen": -1.9719558954238892, | |
| "logits/rejected": -1.959651231765747, | |
| "logps/chosen": -34.44158935546875, | |
| "logps/rejected": -33.596771240234375, | |
| "loss": 0.2835, | |
| "rewards/accuracies": 0.675000011920929, | |
| "rewards/chosen": 0.12358621507883072, | |
| "rewards/margins": 0.1699295938014984, | |
| "rewards/rejected": -0.0463433675467968, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 0.86, | |
| "learning_rate": 3.053082288996112e-07, | |
| "logits/chosen": -2.0068559646606445, | |
| "logits/rejected": -2.00539231300354, | |
| "logps/chosen": -33.331153869628906, | |
| "logps/rejected": -32.49522018432617, | |
| "loss": 0.3707, | |
| "rewards/accuracies": 0.6499999761581421, | |
| "rewards/chosen": 0.12371524423360825, | |
| "rewards/margins": 0.13622507452964783, | |
| "rewards/rejected": -0.012509837746620178, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 0.88, | |
| "learning_rate": 2.0579377374915805e-07, | |
| "logits/chosen": -2.0939385890960693, | |
| "logits/rejected": -2.0781521797180176, | |
| "logps/chosen": -33.88405227661133, | |
| "logps/rejected": -33.090179443359375, | |
| "loss": 0.36, | |
| "rewards/accuracies": 0.699999988079071, | |
| "rewards/chosen": 0.23627004027366638, | |
| "rewards/margins": 0.21398480236530304, | |
| "rewards/rejected": 0.02228522300720215, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 0.91, | |
| "learning_rate": 1.2518018074041684e-07, | |
| "logits/chosen": -1.9664535522460938, | |
| "logits/rejected": -1.9655389785766602, | |
| "logps/chosen": -32.96236038208008, | |
| "logps/rejected": -32.478477478027344, | |
| "loss": 0.3, | |
| "rewards/accuracies": 0.737500011920929, | |
| "rewards/chosen": 0.2557886242866516, | |
| "rewards/margins": 0.2664529085159302, | |
| "rewards/rejected": -0.0106642572209239, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "learning_rate": 6.41315865106129e-08, | |
| "logits/chosen": -1.921805739402771, | |
| "logits/rejected": -1.9321292638778687, | |
| "logps/chosen": -32.022247314453125, | |
| "logps/rejected": -35.31328582763672, | |
| "loss": 0.3671, | |
| "rewards/accuracies": 0.7124999761581421, | |
| "rewards/chosen": 0.14595167338848114, | |
| "rewards/margins": 0.16557307541370392, | |
| "rewards/rejected": -0.01962139829993248, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "learning_rate": 2.3150941078050325e-08, | |
| "logits/chosen": -2.060645580291748, | |
| "logits/rejected": -2.0541434288024902, | |
| "logps/chosen": -33.4498405456543, | |
| "logps/rejected": -29.211456298828125, | |
| "loss": 0.3226, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": 0.16169998049736023, | |
| "rewards/margins": 0.1497238427400589, | |
| "rewards/rejected": 0.011976108886301517, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 0.99, | |
| "learning_rate": 2.575864278703266e-09, | |
| "logits/chosen": -1.920902967453003, | |
| "logits/rejected": -1.9230811595916748, | |
| "logps/chosen": -33.97188186645508, | |
| "logps/rejected": -30.862768173217773, | |
| "loss": 0.3117, | |
| "rewards/accuracies": 0.7124999761581421, | |
| "rewards/chosen": 0.21373608708381653, | |
| "rewards/margins": 0.20584869384765625, | |
| "rewards/rejected": 0.00788736529648304, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "step": 385, | |
| "total_flos": 0.0, | |
| "train_loss": 0.3756082092012678, | |
| "train_runtime": 3251.4223, | |
| "train_samples_per_second": 0.947, | |
| "train_steps_per_second": 0.118 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 385, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 100, | |
| "total_flos": 0.0, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |