{ "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, "grad_norm": 65.5, "learning_rate": 1.282051282051282e-07, "logits/chosen": 88.18099975585938, "logits/rejected": 88.25153350830078, "logps/chosen": -29.073104858398438, "logps/rejected": -26.25731658935547, "loss": 1.0, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 1 }, { "epoch": 0.03, "grad_norm": 54.5, "learning_rate": 1.282051282051282e-06, "logits/chosen": 81.0613784790039, "logits/rejected": 80.76800537109375, "logps/chosen": -34.20709228515625, "logps/rejected": -33.1387939453125, "loss": 0.8925, "rewards/accuracies": 0.5416666865348816, "rewards/chosen": 0.017316989600658417, "rewards/margins": 0.10751060396432877, "rewards/rejected": -0.09019361436367035, "step": 10 }, { "epoch": 0.05, "grad_norm": 56.0, "learning_rate": 2.564102564102564e-06, "logits/chosen": 80.64421081542969, "logits/rejected": 80.53112030029297, "logps/chosen": -33.46975326538086, "logps/rejected": -30.782833099365234, "loss": 0.8999, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": 0.11007847636938095, "rewards/margins": 0.11275172233581543, "rewards/rejected": -0.002673250390216708, "step": 20 }, { "epoch": 0.08, "grad_norm": 60.75, "learning_rate": 3.846153846153847e-06, "logits/chosen": 82.45377349853516, "logits/rejected": 82.48570251464844, "logps/chosen": -33.79710006713867, "logps/rejected": -31.156320571899414, "loss": 1.0372, "rewards/accuracies": 0.42500001192092896, "rewards/chosen": 0.10785794258117676, "rewards/margins": -0.025897562503814697, "rewards/rejected": 0.13375550508499146, "step": 30 }, { "epoch": 0.1, "grad_norm": 47.0, "learning_rate": 4.999896948438434e-06, "logits/chosen": 80.95109558105469, "logits/rejected": 80.94834899902344, "logps/chosen": -32.803199768066406, "logps/rejected": -33.15715026855469, "loss": 0.9441, "rewards/accuracies": 0.5874999761581421, "rewards/chosen": 0.1854299008846283, "rewards/margins": 0.07974345237016678, "rewards/rejected": 0.10568644851446152, "step": 40 }, { "epoch": 0.13, "grad_norm": 43.25, "learning_rate": 4.987541037542187e-06, "logits/chosen": 78.6640625, "logits/rejected": 78.6778335571289, "logps/chosen": -30.60561180114746, "logps/rejected": -30.74114418029785, "loss": 0.9478, "rewards/accuracies": 0.512499988079071, "rewards/chosen": 0.29981669783592224, "rewards/margins": 0.1337459534406662, "rewards/rejected": 0.16607072949409485, "step": 50 }, { "epoch": 0.16, "grad_norm": 50.5, "learning_rate": 4.954691471941119e-06, "logits/chosen": 83.31804656982422, "logits/rejected": 83.37259674072266, "logps/chosen": -30.898645401000977, "logps/rejected": -29.465993881225586, "loss": 0.9771, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": 0.13825830817222595, "rewards/margins": 0.07599552720785141, "rewards/rejected": 0.06226278096437454, "step": 60 }, { "epoch": 0.18, "grad_norm": 62.5, "learning_rate": 4.901618883413549e-06, "logits/chosen": 83.98554229736328, "logits/rejected": 84.01966094970703, "logps/chosen": -30.47078514099121, "logps/rejected": -33.03969192504883, "loss": 0.9425, "rewards/accuracies": 0.5874999761581421, "rewards/chosen": 0.0791207104921341, "rewards/margins": 0.08011289685964584, "rewards/rejected": -0.0009921938180923462, "step": 70 }, { "epoch": 0.21, "grad_norm": 55.5, "learning_rate": 4.828760511501322e-06, "logits/chosen": 81.63408660888672, "logits/rejected": 81.62284088134766, "logps/chosen": -31.412799835205078, "logps/rejected": -30.981252670288086, "loss": 0.8729, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": 0.046071670949459076, "rewards/margins": 0.1578732579946518, "rewards/rejected": -0.11180160194635391, "step": 80 }, { "epoch": 0.23, "grad_norm": 67.0, "learning_rate": 4.7367166013034295e-06, "logits/chosen": 78.38566589355469, "logits/rejected": 78.3572998046875, "logps/chosen": -32.54473114013672, "logps/rejected": -31.261743545532227, "loss": 0.8672, "rewards/accuracies": 0.550000011920929, "rewards/chosen": 0.04672273248434067, "rewards/margins": 0.2018592804670334, "rewards/rejected": -0.15513655543327332, "step": 90 }, { "epoch": 0.26, "grad_norm": 59.25, "learning_rate": 4.626245458345211e-06, "logits/chosen": 83.54986572265625, "logits/rejected": 83.57047271728516, "logps/chosen": -34.15292739868164, "logps/rejected": -31.826343536376953, "loss": 0.9172, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": 0.07603093981742859, "rewards/margins": 0.11392843723297119, "rewards/rejected": -0.0378975048661232, "step": 100 }, { "epoch": 0.26, "eval_logits/chosen": 98.79605865478516, "eval_logits/rejected": 98.78559875488281, "eval_logps/chosen": -32.45810317993164, "eval_logps/rejected": -36.06624221801758, "eval_loss": 0.9737041592597961, "eval_rewards/accuracies": 0.5103820562362671, "eval_rewards/chosen": -0.0074609918519854546, "eval_rewards/margins": 0.042396578937768936, "eval_rewards/rejected": -0.049857571721076965, "eval_runtime": 104.0533, "eval_samples_per_second": 3.296, "eval_steps_per_second": 0.413, "step": 100 }, { "epoch": 0.29, "grad_norm": 68.5, "learning_rate": 4.498257201263691e-06, "logits/chosen": 83.77371978759766, "logits/rejected": 83.6593017578125, "logps/chosen": -32.28753662109375, "logps/rejected": -32.78463363647461, "loss": 0.7046, "rewards/accuracies": 0.7124999761581421, "rewards/chosen": 0.3676328957080841, "rewards/margins": 0.43200382590293884, "rewards/rejected": -0.06437097489833832, "step": 110 }, { "epoch": 0.31, "grad_norm": 67.5, "learning_rate": 4.353806263777678e-06, "logits/chosen": 83.89173889160156, "logits/rejected": 84.00154113769531, "logps/chosen": -28.253698348999023, "logps/rejected": -35.52687072753906, "loss": 0.7808, "rewards/accuracies": 0.6499999761581421, "rewards/chosen": 0.33794277906417847, "rewards/margins": 0.37180477380752563, "rewards/rejected": -0.03386203199625015, "step": 120 }, { "epoch": 0.34, "grad_norm": 47.5, "learning_rate": 4.1940827077152755e-06, "logits/chosen": 81.15254974365234, "logits/rejected": 81.17030334472656, "logps/chosen": -30.407424926757812, "logps/rejected": -32.09453201293945, "loss": 0.8069, "rewards/accuracies": 0.637499988079071, "rewards/chosen": 0.24996523559093475, "rewards/margins": 0.3314167857170105, "rewards/rejected": -0.08145156502723694, "step": 130 }, { "epoch": 0.36, "grad_norm": 44.5, "learning_rate": 4.0204024186666215e-06, "logits/chosen": 82.38260650634766, "logits/rejected": 82.38970184326172, "logps/chosen": -27.169906616210938, "logps/rejected": -33.05832290649414, "loss": 0.7356, "rewards/accuracies": 0.675000011920929, "rewards/chosen": 0.13870806992053986, "rewards/margins": 0.4561801850795746, "rewards/rejected": -0.3174721598625183, "step": 140 }, { "epoch": 0.39, "grad_norm": 50.5, "learning_rate": 3.834196265035119e-06, "logits/chosen": 80.86981201171875, "logits/rejected": 80.84159851074219, "logps/chosen": -28.839962005615234, "logps/rejected": -33.071983337402344, "loss": 0.628, "rewards/accuracies": 0.737500011920929, "rewards/chosen": 0.2745725214481354, "rewards/margins": 0.5273140668869019, "rewards/rejected": -0.2527415156364441, "step": 150 }, { "epoch": 0.42, "grad_norm": 63.5, "learning_rate": 3.636998309800573e-06, "logits/chosen": 82.75099182128906, "logits/rejected": 82.76429748535156, "logps/chosen": -33.66413116455078, "logps/rejected": -30.465320587158203, "loss": 0.7241, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": 0.2363094538450241, "rewards/margins": 0.4926998019218445, "rewards/rejected": -0.2563903033733368, "step": 160 }, { "epoch": 0.44, "grad_norm": 59.0, "learning_rate": 3.4304331721118078e-06, "logits/chosen": 83.50555419921875, "logits/rejected": 83.4540786743164, "logps/chosen": -30.95868492126465, "logps/rejected": -32.49937057495117, "loss": 0.758, "rewards/accuracies": 0.637499988079071, "rewards/chosen": 0.15030263364315033, "rewards/margins": 0.38995999097824097, "rewards/rejected": -0.23965740203857422, "step": 170 }, { "epoch": 0.47, "grad_norm": 35.25, "learning_rate": 3.2162026428305436e-06, "logits/chosen": 81.02674865722656, "logits/rejected": 81.00919342041016, "logps/chosen": -30.531005859375, "logps/rejected": -31.705917358398438, "loss": 0.6438, "rewards/accuracies": 0.737500011920929, "rewards/chosen": 0.3330736756324768, "rewards/margins": 0.6074270606040955, "rewards/rejected": -0.27435341477394104, "step": 180 }, { "epoch": 0.49, "grad_norm": 37.0, "learning_rate": 2.996071664294641e-06, "logits/chosen": 82.6941909790039, "logits/rejected": 82.67329406738281, "logps/chosen": -30.421396255493164, "logps/rejected": -30.745235443115234, "loss": 0.8436, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": 0.172510027885437, "rewards/margins": 0.3111144006252289, "rewards/rejected": -0.13860435783863068, "step": 190 }, { "epoch": 0.52, "grad_norm": 35.0, "learning_rate": 2.7718537898066833e-06, "logits/chosen": 78.1072006225586, "logits/rejected": 78.06266021728516, "logps/chosen": -33.60405731201172, "logps/rejected": -32.724769592285156, "loss": 0.6396, "rewards/accuracies": 0.737500011920929, "rewards/chosen": 0.5743416547775269, "rewards/margins": 0.690025269985199, "rewards/rejected": -0.11568355560302734, "step": 200 }, { "epoch": 0.52, "eval_logits/chosen": 98.68441009521484, "eval_logits/rejected": 98.66287994384766, "eval_logps/chosen": -32.568355560302734, "eval_logps/rejected": -36.311798095703125, "eval_loss": 0.9231801629066467, "eval_rewards/accuracies": 0.5834717750549316, "eval_rewards/chosen": -0.06259084492921829, "eval_rewards/margins": 0.11004550755023956, "eval_rewards/rejected": -0.17263635993003845, "eval_runtime": 103.7953, "eval_samples_per_second": 3.305, "eval_steps_per_second": 0.414, "step": 200 }, { "epoch": 0.55, "grad_norm": 71.5, "learning_rate": 2.5453962426402006e-06, "logits/chosen": 80.72740173339844, "logits/rejected": 80.64077758789062, "logps/chosen": -33.17305374145508, "logps/rejected": -35.227195739746094, "loss": 0.6879, "rewards/accuracies": 0.75, "rewards/chosen": 0.3863331079483032, "rewards/margins": 0.5068855881690979, "rewards/rejected": -0.12055248022079468, "step": 210 }, { "epoch": 0.57, "grad_norm": 53.5, "learning_rate": 2.3185646976551794e-06, "logits/chosen": 82.8826675415039, "logits/rejected": 82.97186279296875, "logps/chosen": -31.021907806396484, "logps/rejected": -31.140172958374023, "loss": 0.6275, "rewards/accuracies": 0.75, "rewards/chosen": 0.46352285146713257, "rewards/margins": 0.6691335439682007, "rewards/rejected": -0.20561063289642334, "step": 220 }, { "epoch": 0.6, "grad_norm": 52.5, "learning_rate": 2.0932279108998323e-06, "logits/chosen": 80.10808563232422, "logits/rejected": 80.16153717041016, "logps/chosen": -32.17018508911133, "logps/rejected": -34.35881423950195, "loss": 0.7509, "rewards/accuracies": 0.6499999761581421, "rewards/chosen": 0.31778058409690857, "rewards/margins": 0.49111372232437134, "rewards/rejected": -0.17333316802978516, "step": 230 }, { "epoch": 0.62, "grad_norm": 65.0, "learning_rate": 1.8712423238279358e-06, "logits/chosen": 82.51493835449219, "logits/rejected": 82.7935562133789, "logps/chosen": -30.67529296875, "logps/rejected": -31.787023544311523, "loss": 0.5481, "rewards/accuracies": 0.737500011920929, "rewards/chosen": 0.5063848495483398, "rewards/margins": 0.6338765621185303, "rewards/rejected": -0.12749171257019043, "step": 240 }, { "epoch": 0.65, "grad_norm": 58.75, "learning_rate": 1.6544367689701824e-06, "logits/chosen": 81.21392059326172, "logits/rejected": 81.27059173583984, "logps/chosen": -26.883764266967773, "logps/rejected": -30.1818790435791, "loss": 0.7593, "rewards/accuracies": 0.612500011920929, "rewards/chosen": 0.3560645580291748, "rewards/margins": 0.4966358244419098, "rewards/rejected": -0.1405712366104126, "step": 250 }, { "epoch": 0.68, "grad_norm": 37.5, "learning_rate": 1.4445974030621963e-06, "logits/chosen": 78.52751159667969, "logits/rejected": 78.65741729736328, "logps/chosen": -30.363637924194336, "logps/rejected": -36.511695861816406, "loss": 0.5428, "rewards/accuracies": 0.7875000238418579, "rewards/chosen": 0.5986787676811218, "rewards/margins": 0.8037135004997253, "rewards/rejected": -0.20503473281860352, "step": 260 }, { "epoch": 0.7, "grad_norm": 35.0, "learning_rate": 1.243452991757889e-06, "logits/chosen": 77.82562255859375, "logits/rejected": 77.8570785522461, "logps/chosen": -30.939916610717773, "logps/rejected": -31.843109130859375, "loss": 0.5855, "rewards/accuracies": 0.7749999761581421, "rewards/chosen": 0.4195052981376648, "rewards/margins": 0.5883487462997437, "rewards/rejected": -0.1688433587551117, "step": 270 }, { "epoch": 0.73, "grad_norm": 64.5, "learning_rate": 1.0526606671603523e-06, "logits/chosen": 80.51162719726562, "logits/rejected": 80.30082702636719, "logps/chosen": -30.975082397460938, "logps/rejected": -29.75954818725586, "loss": 0.7416, "rewards/accuracies": 0.7124999761581421, "rewards/chosen": 0.417405366897583, "rewards/margins": 0.4636818468570709, "rewards/rejected": -0.04627644270658493, "step": 280 }, { "epoch": 0.75, "grad_norm": 42.5, "learning_rate": 8.737922755071455e-07, "logits/chosen": 80.64032745361328, "logits/rejected": 80.54534149169922, "logps/chosen": -33.00518035888672, "logps/rejected": -32.465091705322266, "loss": 0.5643, "rewards/accuracies": 0.7875000238418579, "rewards/chosen": 0.5495337247848511, "rewards/margins": 0.7736457586288452, "rewards/rejected": -0.22411206364631653, "step": 290 }, { "epoch": 0.78, "grad_norm": 46.75, "learning_rate": 7.08321427484816e-07, "logits/chosen": 76.24769592285156, "logits/rejected": 76.3338851928711, "logps/chosen": -32.11992645263672, "logps/rejected": -29.312702178955078, "loss": 0.6566, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": 0.642888605594635, "rewards/margins": 0.748228907585144, "rewards/rejected": -0.1053403839468956, "step": 300 }, { "epoch": 0.78, "eval_logits/chosen": 98.71778106689453, "eval_logits/rejected": 98.69977569580078, "eval_logps/chosen": -32.561683654785156, "eval_logps/rejected": -36.26484298706055, "eval_loss": 0.9383498430252075, "eval_rewards/accuracies": 0.5456810593605042, "eval_rewards/chosen": -0.05925469473004341, "eval_rewards/margins": 0.08990433067083359, "eval_rewards/rejected": -0.1491590440273285, "eval_runtime": 103.9955, "eval_samples_per_second": 3.298, "eval_steps_per_second": 0.413, "step": 300 }, { "epoch": 0.81, "grad_norm": 44.25, "learning_rate": 5.576113578589035e-07, "logits/chosen": 83.27408599853516, "logits/rejected": 83.30269622802734, "logps/chosen": -30.081134796142578, "logps/rejected": -32.3824348449707, "loss": 0.6944, "rewards/accuracies": 0.699999988079071, "rewards/chosen": 0.39120084047317505, "rewards/margins": 0.48253631591796875, "rewards/rejected": -0.0913354903459549, "step": 310 }, { "epoch": 0.83, "grad_norm": 30.375, "learning_rate": 4.229036944380913e-07, "logits/chosen": 80.83680725097656, "logits/rejected": 80.83592224121094, "logps/chosen": -30.626052856445312, "logps/rejected": -29.309789657592773, "loss": 0.5574, "rewards/accuracies": 0.762499988079071, "rewards/chosen": 0.48625460267066956, "rewards/margins": 0.6814028024673462, "rewards/rejected": -0.19514815509319305, "step": 320 }, { "epoch": 0.86, "grad_norm": 37.0, "learning_rate": 3.053082288996112e-07, "logits/chosen": 78.04208374023438, "logits/rejected": 78.09200286865234, "logps/chosen": -29.09530258178711, "logps/rejected": -32.94050979614258, "loss": 0.5255, "rewards/accuracies": 0.7875000238418579, "rewards/chosen": 0.6286521553993225, "rewards/margins": 0.7814761400222778, "rewards/rejected": -0.15282386541366577, "step": 330 }, { "epoch": 0.88, "grad_norm": 54.75, "learning_rate": 2.0579377374915805e-07, "logits/chosen": 82.37565612792969, "logits/rejected": 82.40386962890625, "logps/chosen": -32.312721252441406, "logps/rejected": -33.788272857666016, "loss": 0.7038, "rewards/accuracies": 0.699999988079071, "rewards/chosen": 0.4497138559818268, "rewards/margins": 0.6595038175582886, "rewards/rejected": -0.2097899168729782, "step": 340 }, { "epoch": 0.91, "grad_norm": 30.0, "learning_rate": 1.2518018074041684e-07, "logits/chosen": 81.36256408691406, "logits/rejected": 81.37139892578125, "logps/chosen": -32.320640563964844, "logps/rejected": -33.50237274169922, "loss": 0.5293, "rewards/accuracies": 0.762499988079071, "rewards/chosen": 0.6615498065948486, "rewards/margins": 0.866836428642273, "rewards/rejected": -0.20528674125671387, "step": 350 }, { "epoch": 0.94, "grad_norm": 44.25, "learning_rate": 6.41315865106129e-08, "logits/chosen": 82.83390045166016, "logits/rejected": 82.8671875, "logps/chosen": -28.39254379272461, "logps/rejected": -31.90033531188965, "loss": 0.6161, "rewards/accuracies": 0.737500011920929, "rewards/chosen": 0.579832911491394, "rewards/margins": 0.7050749659538269, "rewards/rejected": -0.12524208426475525, "step": 360 }, { "epoch": 0.96, "grad_norm": 70.0, "learning_rate": 2.3150941078050325e-08, "logits/chosen": 82.28783416748047, "logits/rejected": 82.31135559082031, "logps/chosen": -31.76995849609375, "logps/rejected": -35.398277282714844, "loss": 0.6445, "rewards/accuracies": 0.7124999761581421, "rewards/chosen": 0.553102970123291, "rewards/margins": 0.6975550651550293, "rewards/rejected": -0.14445209503173828, "step": 370 }, { "epoch": 0.99, "grad_norm": 56.5, "learning_rate": 2.575864278703266e-09, "logits/chosen": 76.20292663574219, "logits/rejected": 76.08332824707031, "logps/chosen": -29.707965850830078, "logps/rejected": -28.27734375, "loss": 0.7106, "rewards/accuracies": 0.6625000238418579, "rewards/chosen": 0.43407946825027466, "rewards/margins": 0.492464154958725, "rewards/rejected": -0.05838475376367569, "step": 380 }, { "epoch": 1.0, "step": 385, "total_flos": 0.0, "train_loss": 0.735349414874981, "train_runtime": 2550.735, "train_samples_per_second": 1.207, "train_steps_per_second": 0.151 } ], "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 }