{ "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": 52.25, "learning_rate": 1.282051282051282e-07, "logits/chosen": 88.18099975585938, "logits/rejected": 88.25153350830078, "logps/chosen": -29.073104858398438, "logps/rejected": -26.25731658935547, "loss": 0.6931, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 1 }, { "epoch": 0.03, "grad_norm": 40.75, "learning_rate": 1.282051282051282e-06, "logits/chosen": 81.06388092041016, "logits/rejected": 80.76617431640625, "logps/chosen": -34.315635681152344, "logps/rejected": -33.083072662353516, "loss": 0.7277, "rewards/accuracies": 0.4305555522441864, "rewards/chosen": -0.05912955477833748, "rewards/margins": 0.04060414060950279, "rewards/rejected": -0.09973368048667908, "step": 10 }, { "epoch": 0.05, "grad_norm": 36.25, "learning_rate": 2.564102564102564e-06, "logits/chosen": 80.6710433959961, "logits/rejected": 80.55816650390625, "logps/chosen": -33.6015625, "logps/rejected": -30.793548583984375, "loss": 0.724, "rewards/accuracies": 0.4749999940395355, "rewards/chosen": 0.07067807018756866, "rewards/margins": 0.08352533727884293, "rewards/rejected": -0.012847280129790306, "step": 20 }, { "epoch": 0.08, "grad_norm": 48.0, "learning_rate": 3.846153846153847e-06, "logits/chosen": 82.53105163574219, "logits/rejected": 82.56202697753906, "logps/chosen": -33.846534729003906, "logps/rejected": -31.217113494873047, "loss": 0.823, "rewards/accuracies": 0.44999998807907104, "rewards/chosen": 0.13303251564502716, "rewards/margins": -0.03234311193227768, "rewards/rejected": 0.16537563502788544, "step": 30 }, { "epoch": 0.1, "grad_norm": 49.0, "learning_rate": 4.999896948438434e-06, "logits/chosen": 81.03451538085938, "logits/rejected": 81.03263092041016, "logps/chosen": -32.87987518310547, "logps/rejected": -33.08030700683594, "loss": 0.7898, "rewards/accuracies": 0.5, "rewards/chosen": 0.2353484332561493, "rewards/margins": 0.004774211905896664, "rewards/rejected": 0.23057425022125244, "step": 40 }, { "epoch": 0.13, "grad_norm": 37.0, "learning_rate": 4.987541037542187e-06, "logits/chosen": 78.62713623046875, "logits/rejected": 78.64068603515625, "logps/chosen": -30.693645477294922, "logps/rejected": -30.871618270874023, "loss": 0.7362, "rewards/accuracies": 0.550000011920929, "rewards/chosen": 0.40928253531455994, "rewards/margins": 0.24794897437095642, "rewards/rejected": 0.1613335758447647, "step": 50 }, { "epoch": 0.16, "grad_norm": 35.75, "learning_rate": 4.954691471941119e-06, "logits/chosen": 83.1414566040039, "logits/rejected": 83.20195007324219, "logps/chosen": -30.973648071289062, "logps/rejected": -29.62088394165039, "loss": 0.7214, "rewards/accuracies": 0.5, "rewards/chosen": 0.16120928525924683, "rewards/margins": 0.18549844622612, "rewards/rejected": -0.02428916096687317, "step": 60 }, { "epoch": 0.18, "grad_norm": 72.5, "learning_rate": 4.901618883413549e-06, "logits/chosen": 83.78003692626953, "logits/rejected": 83.8111343383789, "logps/chosen": -30.537906646728516, "logps/rejected": -33.126197814941406, "loss": 0.7291, "rewards/accuracies": 0.612500011920929, "rewards/chosen": 0.07289580255746841, "rewards/margins": 0.14368507266044617, "rewards/rejected": -0.07078926265239716, "step": 70 }, { "epoch": 0.21, "grad_norm": 46.25, "learning_rate": 4.828760511501322e-06, "logits/chosen": 81.44255065917969, "logits/rejected": 81.430908203125, "logps/chosen": -31.439010620117188, "logps/rejected": -30.995372772216797, "loss": 0.7039, "rewards/accuracies": 0.48750001192092896, "rewards/chosen": 0.0527455098927021, "rewards/margins": 0.24292418360710144, "rewards/rejected": -0.19017870724201202, "step": 80 }, { "epoch": 0.23, "grad_norm": 61.75, "learning_rate": 4.7367166013034295e-06, "logits/chosen": 78.15830993652344, "logits/rejected": 78.127197265625, "logps/chosen": -32.702423095703125, "logps/rejected": -31.252826690673828, "loss": 0.7773, "rewards/accuracies": 0.5625, "rewards/chosen": -0.0513942614197731, "rewards/margins": 0.18969133496284485, "rewards/rejected": -0.24108560383319855, "step": 90 }, { "epoch": 0.26, "grad_norm": 39.0, "learning_rate": 4.626245458345211e-06, "logits/chosen": 83.44832611083984, "logits/rejected": 83.47621154785156, "logps/chosen": -34.1397819519043, "logps/rejected": -31.86690330505371, "loss": 0.7017, "rewards/accuracies": 0.574999988079071, "rewards/chosen": 0.1321648806333542, "rewards/margins": 0.225250244140625, "rewards/rejected": -0.09308534860610962, "step": 100 }, { "epoch": 0.26, "eval_logits/chosen": 98.7764892578125, "eval_logits/rejected": 98.76370239257812, "eval_logps/chosen": -32.49420166015625, "eval_logps/rejected": -36.03456497192383, "eval_loss": 0.8030149340629578, "eval_rewards/accuracies": 0.5137043595314026, "eval_rewards/chosen": -0.04081834852695465, "eval_rewards/margins": 0.013612199574708939, "eval_rewards/rejected": -0.05443055182695389, "eval_runtime": 104.1164, "eval_samples_per_second": 3.294, "eval_steps_per_second": 0.413, "step": 100 }, { "epoch": 0.29, "grad_norm": 48.25, "learning_rate": 4.498257201263691e-06, "logits/chosen": 83.67167663574219, "logits/rejected": 83.56147766113281, "logps/chosen": -32.48063659667969, "logps/rejected": -32.763511657714844, "loss": 0.6258, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": 0.43373650312423706, "rewards/margins": 0.5198318362236023, "rewards/rejected": -0.08609537780284882, "step": 110 }, { "epoch": 0.31, "grad_norm": 53.75, "learning_rate": 4.353806263777678e-06, "logits/chosen": 83.80461120605469, "logits/rejected": 83.91490173339844, "logps/chosen": -28.273183822631836, "logps/rejected": -35.56282043457031, "loss": 0.5439, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": 0.5251210927963257, "rewards/margins": 0.608059287071228, "rewards/rejected": -0.08293820172548294, "step": 120 }, { "epoch": 0.34, "grad_norm": 28.75, "learning_rate": 4.1940827077152755e-06, "logits/chosen": 81.06788635253906, "logits/rejected": 81.09136962890625, "logps/chosen": -30.509033203125, "logps/rejected": -32.07707595825195, "loss": 0.6608, "rewards/accuracies": 0.5874999761581421, "rewards/chosen": 0.3186565339565277, "rewards/margins": 0.4350167214870453, "rewards/rejected": -0.11636020243167877, "step": 130 }, { "epoch": 0.36, "grad_norm": 24.875, "learning_rate": 4.0204024186666215e-06, "logits/chosen": 82.33757019042969, "logits/rejected": 82.34171295166016, "logps/chosen": -26.971057891845703, "logps/rejected": -32.84387969970703, "loss": 0.5554, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": 0.38101473450660706, "rewards/margins": 0.7174161076545715, "rewards/rejected": -0.33640143275260925, "step": 140 }, { "epoch": 0.39, "grad_norm": 30.125, "learning_rate": 3.834196265035119e-06, "logits/chosen": 80.85295104980469, "logits/rejected": 80.82408142089844, "logps/chosen": -28.91988754272461, "logps/rejected": -33.04113006591797, "loss": 0.5578, "rewards/accuracies": 0.699999988079071, "rewards/chosen": 0.375379741191864, "rewards/margins": 0.7550806403160095, "rewards/rejected": -0.37970098853111267, "step": 150 }, { "epoch": 0.42, "grad_norm": 60.5, "learning_rate": 3.636998309800573e-06, "logits/chosen": 82.69292449951172, "logits/rejected": 82.70225524902344, "logps/chosen": -33.54145050048828, "logps/rejected": -30.415197372436523, "loss": 0.6193, "rewards/accuracies": 0.699999988079071, "rewards/chosen": 0.4762394428253174, "rewards/margins": 0.8463689684867859, "rewards/rejected": -0.37012940645217896, "step": 160 }, { "epoch": 0.44, "grad_norm": 30.375, "learning_rate": 3.4304331721118078e-06, "logits/chosen": 83.42604064941406, "logits/rejected": 83.37831115722656, "logps/chosen": -30.8799991607666, "logps/rejected": -32.62963104248047, "loss": 0.5673, "rewards/accuracies": 0.6625000238418579, "rewards/chosen": 0.3034316599369049, "rewards/margins": 0.7910871505737305, "rewards/rejected": -0.4876554012298584, "step": 170 }, { "epoch": 0.47, "grad_norm": 32.0, "learning_rate": 3.2162026428305436e-06, "logits/chosen": 80.95600891113281, "logits/rejected": 80.93622589111328, "logps/chosen": -30.429906845092773, "logps/rejected": -31.591838836669922, "loss": 0.4992, "rewards/accuracies": 0.737500011920929, "rewards/chosen": 0.6137940287590027, "rewards/margins": 0.9614995718002319, "rewards/rejected": -0.34770551323890686, "step": 180 }, { "epoch": 0.49, "grad_norm": 17.25, "learning_rate": 2.996071664294641e-06, "logits/chosen": 82.7037124633789, "logits/rejected": 82.69169616699219, "logps/chosen": -30.465667724609375, "logps/rejected": -30.747249603271484, "loss": 0.7493, "rewards/accuracies": 0.574999988079071, "rewards/chosen": 0.24060054123401642, "rewards/margins": 0.463981568813324, "rewards/rejected": -0.22338099777698517, "step": 190 }, { "epoch": 0.52, "grad_norm": 21.5, "learning_rate": 2.7718537898066833e-06, "logits/chosen": 78.2643814086914, "logits/rejected": 78.21286010742188, "logps/chosen": -33.79939270019531, "logps/rejected": -32.67485046386719, "loss": 0.6236, "rewards/accuracies": 0.675000011920929, "rewards/chosen": 0.7626792192459106, "rewards/margins": 0.9078337550163269, "rewards/rejected": -0.14515459537506104, "step": 200 }, { "epoch": 0.52, "eval_logits/chosen": 98.72710418701172, "eval_logits/rejected": 98.70391082763672, "eval_logps/chosen": -32.6581916809082, "eval_logps/rejected": -36.33292770385742, "eval_loss": 0.7719493508338928, "eval_rewards/accuracies": 0.5215947031974792, "eval_rewards/chosen": -0.17201441526412964, "eval_rewards/margins": 0.12110573798418045, "eval_rewards/rejected": -0.2931201457977295, "eval_runtime": 103.8817, "eval_samples_per_second": 3.302, "eval_steps_per_second": 0.414, "step": 200 }, { "epoch": 0.55, "grad_norm": 78.5, "learning_rate": 2.5453962426402006e-06, "logits/chosen": 80.8266830444336, "logits/rejected": 80.73197174072266, "logps/chosen": -33.31422805786133, "logps/rejected": -35.28189468383789, "loss": 0.6059, "rewards/accuracies": 0.6625000238418579, "rewards/chosen": 0.5051938891410828, "rewards/margins": 0.7418341636657715, "rewards/rejected": -0.23664025962352753, "step": 210 }, { "epoch": 0.57, "grad_norm": 21.0, "learning_rate": 2.3185646976551794e-06, "logits/chosen": 83.00508117675781, "logits/rejected": 83.09242248535156, "logps/chosen": -31.03920555114746, "logps/rejected": -30.981882095336914, "loss": 0.5212, "rewards/accuracies": 0.800000011920929, "rewards/chosen": 0.7277995347976685, "rewards/margins": 0.930141270160675, "rewards/rejected": -0.20234176516532898, "step": 220 }, { "epoch": 0.6, "grad_norm": 49.25, "learning_rate": 2.0932279108998323e-06, "logits/chosen": 80.0581283569336, "logits/rejected": 80.11595153808594, "logps/chosen": -32.08351516723633, "logps/rejected": -34.33815002441406, "loss": 0.594, "rewards/accuracies": 0.637499988079071, "rewards/chosen": 0.5777846574783325, "rewards/margins": 0.8385850787162781, "rewards/rejected": -0.26080039143562317, "step": 230 }, { "epoch": 0.62, "grad_norm": 28.0, "learning_rate": 1.8712423238279358e-06, "logits/chosen": 82.56427001953125, "logits/rejected": 82.85057830810547, "logps/chosen": -30.730464935302734, "logps/rejected": -31.813983917236328, "loss": 0.4701, "rewards/accuracies": 0.75, "rewards/chosen": 0.7660826444625854, "rewards/margins": 0.9916342496871948, "rewards/rejected": -0.22555160522460938, "step": 240 }, { "epoch": 0.65, "grad_norm": 50.5, "learning_rate": 1.6544367689701824e-06, "logits/chosen": 81.2629623413086, "logits/rejected": 81.32804870605469, "logps/chosen": -26.856103897094727, "logps/rejected": -30.107614517211914, "loss": 0.6072, "rewards/accuracies": 0.6499999761581421, "rewards/chosen": 0.5918346047401428, "rewards/margins": 0.7573351263999939, "rewards/rejected": -0.1655004471540451, "step": 250 }, { "epoch": 0.68, "grad_norm": 36.25, "learning_rate": 1.4445974030621963e-06, "logits/chosen": 78.54646301269531, "logits/rejected": 78.68824768066406, "logps/chosen": -30.280277252197266, "logps/rejected": -36.395103454589844, "loss": 0.4323, "rewards/accuracies": 0.800000011920929, "rewards/chosen": 1.0245726108551025, "rewards/margins": 1.259353756904602, "rewards/rejected": -0.23478105664253235, "step": 260 }, { "epoch": 0.7, "grad_norm": 22.625, "learning_rate": 1.243452991757889e-06, "logits/chosen": 77.9116439819336, "logits/rejected": 77.93907165527344, "logps/chosen": -30.94207763671875, "logps/rejected": -31.825298309326172, "loss": 0.5298, "rewards/accuracies": 0.7124999761581421, "rewards/chosen": 0.6694773435592651, "rewards/margins": 0.9253811836242676, "rewards/rejected": -0.2559038996696472, "step": 270 }, { "epoch": 0.73, "grad_norm": 48.0, "learning_rate": 1.0526606671603523e-06, "logits/chosen": 80.6185531616211, "logits/rejected": 80.40409851074219, "logps/chosen": -31.1258487701416, "logps/rejected": -29.83230972290039, "loss": 0.6589, "rewards/accuracies": 0.6625000238418579, "rewards/chosen": 0.5472350716590881, "rewards/margins": 0.6794873476028442, "rewards/rejected": -0.1322522908449173, "step": 280 }, { "epoch": 0.75, "grad_norm": 17.25, "learning_rate": 8.737922755071455e-07, "logits/chosen": 80.76725006103516, "logits/rejected": 80.67757415771484, "logps/chosen": -33.04716491699219, "logps/rejected": -32.507728576660156, "loss": 0.4727, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": 0.8456665277481079, "rewards/margins": 1.238355040550232, "rewards/rejected": -0.39268869161605835, "step": 290 }, { "epoch": 0.78, "grad_norm": 44.75, "learning_rate": 7.08321427484816e-07, "logits/chosen": 76.41008758544922, "logits/rejected": 76.48558044433594, "logps/chosen": -32.2567253112793, "logps/rejected": -29.211252212524414, "loss": 0.5655, "rewards/accuracies": 0.75, "rewards/chosen": 0.9191843271255493, "rewards/margins": 1.0065667629241943, "rewards/rejected": -0.08738242089748383, "step": 300 }, { "epoch": 0.78, "eval_logits/chosen": 98.77224731445312, "eval_logits/rejected": 98.75171661376953, "eval_logps/chosen": -32.550052642822266, "eval_logps/rejected": -36.224205017089844, "eval_loss": 0.7744476795196533, "eval_rewards/accuracies": 0.5427741408348083, "eval_rewards/chosen": -0.08550228923559189, "eval_rewards/margins": 0.1206393763422966, "eval_rewards/rejected": -0.2061416506767273, "eval_runtime": 104.0332, "eval_samples_per_second": 3.297, "eval_steps_per_second": 0.413, "step": 300 }, { "epoch": 0.81, "grad_norm": 31.875, "learning_rate": 5.576113578589035e-07, "logits/chosen": 83.43260192871094, "logits/rejected": 83.46700286865234, "logps/chosen": -30.02802085876465, "logps/rejected": -32.44657897949219, "loss": 0.5905, "rewards/accuracies": 0.699999988079071, "rewards/chosen": 0.6684112548828125, "rewards/margins": 0.8658668398857117, "rewards/rejected": -0.19745555520057678, "step": 310 }, { "epoch": 0.83, "grad_norm": 24.875, "learning_rate": 4.229036944380913e-07, "logits/chosen": 80.97013092041016, "logits/rejected": 80.96881103515625, "logps/chosen": -30.49709701538086, "logps/rejected": -29.160675048828125, "loss": 0.4662, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": 0.8811699748039246, "rewards/margins": 1.0741162300109863, "rewards/rejected": -0.19294631481170654, "step": 320 }, { "epoch": 0.86, "grad_norm": 19.625, "learning_rate": 3.053082288996112e-07, "logits/chosen": 78.14985656738281, "logits/rejected": 78.19586181640625, "logps/chosen": -29.189855575561523, "logps/rejected": -32.966094970703125, "loss": 0.4527, "rewards/accuracies": 0.7749999761581421, "rewards/chosen": 0.9302013516426086, "rewards/margins": 1.195185661315918, "rewards/rejected": -0.26498422026634216, "step": 330 }, { "epoch": 0.88, "grad_norm": 61.5, "learning_rate": 2.0579377374915805e-07, "logits/chosen": 82.46419525146484, "logits/rejected": 82.48377990722656, "logps/chosen": -32.1870002746582, "logps/rejected": -33.69792175292969, "loss": 0.5537, "rewards/accuracies": 0.737500011920929, "rewards/chosen": 0.8201173543930054, "rewards/margins": 1.0834996700286865, "rewards/rejected": -0.26338234543800354, "step": 340 }, { "epoch": 0.91, "grad_norm": 16.75, "learning_rate": 1.2518018074041684e-07, "logits/chosen": 81.44327545166016, "logits/rejected": 81.4502944946289, "logps/chosen": -32.606056213378906, "logps/rejected": -33.298622131347656, "loss": 0.5676, "rewards/accuracies": 0.6875, "rewards/chosen": 0.8301448822021484, "rewards/margins": 0.9956042170524597, "rewards/rejected": -0.16545933485031128, "step": 350 }, { "epoch": 0.94, "grad_norm": 33.0, "learning_rate": 6.41315865106129e-08, "logits/chosen": 82.95188903808594, "logits/rejected": 82.9841079711914, "logps/chosen": -28.371601104736328, "logps/rejected": -31.830394744873047, "loss": 0.4872, "rewards/accuracies": 0.7124999761581421, "rewards/chosen": 0.9444905519485474, "rewards/margins": 1.0889289379119873, "rewards/rejected": -0.14443838596343994, "step": 360 }, { "epoch": 0.96, "grad_norm": 34.25, "learning_rate": 2.3150941078050325e-08, "logits/chosen": 82.37725067138672, "logits/rejected": 82.40049743652344, "logps/chosen": -31.749374389648438, "logps/rejected": -35.2952766418457, "loss": 0.508, "rewards/accuracies": 0.7124999761581421, "rewards/chosen": 0.9014309644699097, "rewards/margins": 1.0501554012298584, "rewards/rejected": -0.14872442185878754, "step": 370 }, { "epoch": 0.99, "grad_norm": 38.0, "learning_rate": 2.575864278703266e-09, "logits/chosen": 76.2959976196289, "logits/rejected": 76.17599487304688, "logps/chosen": -29.714282989501953, "logps/rejected": -28.307018280029297, "loss": 0.57, "rewards/accuracies": 0.6625000238418579, "rewards/chosen": 0.6894746422767639, "rewards/margins": 0.806629478931427, "rewards/rejected": -0.11715485900640488, "step": 380 }, { "epoch": 1.0, "step": 385, "total_flos": 0.0, "train_loss": 0.6050039254225694, "train_runtime": 2556.17, "train_samples_per_second": 1.205, "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 }