{ "best_metric": null, "best_model_checkpoint": null, "epoch": 2.0, "eval_steps": 500, "global_step": 286, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.01, "learning_rate": 1.7241379310344825e-08, "logits/chosen": -2.46767258644104, "logits/rejected": -2.4595911502838135, "logps/chosen": -22.810455322265625, "logps/rejected": -22.677112579345703, "loss": 0.6931, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 1 }, { "epoch": 0.07, "learning_rate": 1.7241379310344828e-07, "logits/chosen": -2.815382719039917, "logits/rejected": -2.7576191425323486, "logps/chosen": -19.62689971923828, "logps/rejected": -11.512568473815918, "loss": 0.6817, "rewards/accuracies": 0.6944444179534912, "rewards/chosen": -0.002836173167452216, "rewards/margins": 0.020304910838603973, "rewards/rejected": -0.023141082376241684, "step": 10 }, { "epoch": 0.14, "learning_rate": 3.4482758620689656e-07, "logits/chosen": -2.760253429412842, "logits/rejected": -2.776731014251709, "logps/chosen": -18.500263214111328, "logps/rejected": -24.879409790039062, "loss": 0.3663, "rewards/accuracies": 0.925000011920929, "rewards/chosen": -0.05979099124670029, "rewards/margins": 1.2453944683074951, "rewards/rejected": -1.3051855564117432, "step": 20 }, { "epoch": 0.21, "learning_rate": 4.980544747081712e-07, "logits/chosen": -2.8509573936462402, "logits/rejected": -2.8971199989318848, "logps/chosen": -20.137048721313477, "logps/rejected": -61.92778778076172, "loss": 0.1236, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": -0.16065621376037598, "rewards/margins": 4.86615514755249, "rewards/rejected": -5.026811599731445, "step": 30 }, { "epoch": 0.28, "learning_rate": 4.785992217898833e-07, "logits/chosen": -2.792508602142334, "logits/rejected": -2.8558077812194824, "logps/chosen": -23.098939895629883, "logps/rejected": -88.87196350097656, "loss": 0.0921, "rewards/accuracies": 0.987500011920929, "rewards/chosen": -0.44465622305870056, "rewards/margins": 7.166872978210449, "rewards/rejected": -7.611530303955078, "step": 40 }, { "epoch": 0.35, "learning_rate": 4.591439688715953e-07, "logits/chosen": -2.8872122764587402, "logits/rejected": -2.8914027214050293, "logps/chosen": -21.817670822143555, "logps/rejected": -82.24397277832031, "loss": 0.0947, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": -0.6026766896247864, "rewards/margins": 6.256636619567871, "rewards/rejected": -6.85931396484375, "step": 50 }, { "epoch": 0.42, "learning_rate": 4.3968871595330735e-07, "logits/chosen": -2.8102235794067383, "logits/rejected": -2.867000102996826, "logps/chosen": -22.098957061767578, "logps/rejected": -106.98995208740234, "loss": 0.0684, "rewards/accuracies": 1.0, "rewards/chosen": -0.6044292449951172, "rewards/margins": 8.731573104858398, "rewards/rejected": -9.336004257202148, "step": 60 }, { "epoch": 0.49, "learning_rate": 4.202334630350194e-07, "logits/chosen": -2.838426113128662, "logits/rejected": -2.8441996574401855, "logps/chosen": -32.25341033935547, "logps/rejected": -112.56993103027344, "loss": 0.0467, "rewards/accuracies": 1.0, "rewards/chosen": -0.9181193113327026, "rewards/margins": 8.931884765625, "rewards/rejected": -9.850004196166992, "step": 70 }, { "epoch": 0.56, "learning_rate": 4.007782101167315e-07, "logits/chosen": -2.7943294048309326, "logits/rejected": -2.81146240234375, "logps/chosen": -28.41353416442871, "logps/rejected": -115.3572769165039, "loss": 0.0531, "rewards/accuracies": 0.987500011920929, "rewards/chosen": -0.8534424901008606, "rewards/margins": 9.425840377807617, "rewards/rejected": -10.27928352355957, "step": 80 }, { "epoch": 0.63, "learning_rate": 3.813229571984435e-07, "logits/chosen": -2.6642768383026123, "logits/rejected": -2.695777416229248, "logps/chosen": -30.20281410217285, "logps/rejected": -115.2649917602539, "loss": 0.0493, "rewards/accuracies": 0.987500011920929, "rewards/chosen": -0.9560843706130981, "rewards/margins": 9.298072814941406, "rewards/rejected": -10.254158020019531, "step": 90 }, { "epoch": 0.7, "learning_rate": 3.618677042801556e-07, "logits/chosen": -2.61277437210083, "logits/rejected": -2.7204833030700684, "logps/chosen": -33.576595306396484, "logps/rejected": -150.16848754882812, "loss": 0.0391, "rewards/accuracies": 0.987500011920929, "rewards/chosen": -1.3517389297485352, "rewards/margins": 12.079813003540039, "rewards/rejected": -13.431551933288574, "step": 100 }, { "epoch": 0.77, "learning_rate": 3.4241245136186767e-07, "logits/chosen": -2.7867584228515625, "logits/rejected": -2.7667887210845947, "logps/chosen": -24.368350982666016, "logps/rejected": -107.03987884521484, "loss": 0.0492, "rewards/accuracies": 0.987500011920929, "rewards/chosen": -0.7897183895111084, "rewards/margins": 8.651564598083496, "rewards/rejected": -9.441283226013184, "step": 110 }, { "epoch": 0.84, "learning_rate": 3.2295719844357975e-07, "logits/chosen": -2.6104767322540283, "logits/rejected": -2.675987482070923, "logps/chosen": -33.704322814941406, "logps/rejected": -135.02169799804688, "loss": 0.023, "rewards/accuracies": 0.987500011920929, "rewards/chosen": -1.1443722248077393, "rewards/margins": 11.034332275390625, "rewards/rejected": -12.178705215454102, "step": 120 }, { "epoch": 0.91, "learning_rate": 3.0350194552529177e-07, "logits/chosen": -2.568190574645996, "logits/rejected": -2.6318023204803467, "logps/chosen": -30.85686683654785, "logps/rejected": -167.8513946533203, "loss": 0.0389, "rewards/accuracies": 0.987500011920929, "rewards/chosen": -1.3852249383926392, "rewards/margins": 14.069249153137207, "rewards/rejected": -15.454473495483398, "step": 130 }, { "epoch": 0.98, "learning_rate": 2.8404669260700384e-07, "logits/chosen": -2.5317270755767822, "logits/rejected": -2.593646764755249, "logps/chosen": -33.05012512207031, "logps/rejected": -199.79788208007812, "loss": 0.0281, "rewards/accuracies": 1.0, "rewards/chosen": -1.1634151935577393, "rewards/margins": 17.322010040283203, "rewards/rejected": -18.485424041748047, "step": 140 }, { "epoch": 1.05, "learning_rate": 2.645914396887159e-07, "logits/chosen": -2.6736645698547363, "logits/rejected": -2.65024995803833, "logps/chosen": -32.11931610107422, "logps/rejected": -172.3567352294922, "loss": 0.011, "rewards/accuracies": 1.0, "rewards/chosen": -1.398026704788208, "rewards/margins": 14.626693725585938, "rewards/rejected": -16.02471923828125, "step": 150 }, { "epoch": 1.12, "learning_rate": 2.45136186770428e-07, "logits/chosen": -2.6248397827148438, "logits/rejected": -2.6503357887268066, "logps/chosen": -35.05925369262695, "logps/rejected": -219.28378295898438, "loss": 0.0095, "rewards/accuracies": 1.0, "rewards/chosen": -1.4844539165496826, "rewards/margins": 19.174663543701172, "rewards/rejected": -20.659120559692383, "step": 160 }, { "epoch": 1.19, "learning_rate": 2.2568093385214007e-07, "logits/chosen": -2.6490864753723145, "logits/rejected": -2.6252989768981934, "logps/chosen": -31.650588989257812, "logps/rejected": -194.4810333251953, "loss": 0.0158, "rewards/accuracies": 1.0, "rewards/chosen": -0.9476087689399719, "rewards/margins": 17.060314178466797, "rewards/rejected": -18.007923126220703, "step": 170 }, { "epoch": 1.26, "learning_rate": 2.0622568093385212e-07, "logits/chosen": -2.7093279361724854, "logits/rejected": -2.638579845428467, "logps/chosen": -32.82453155517578, "logps/rejected": -185.39230346679688, "loss": 0.0104, "rewards/accuracies": 1.0, "rewards/chosen": -1.2247194051742554, "rewards/margins": 15.979866027832031, "rewards/rejected": -17.204586029052734, "step": 180 }, { "epoch": 1.33, "learning_rate": 1.867704280155642e-07, "logits/chosen": -2.6887717247009277, "logits/rejected": -2.6667675971984863, "logps/chosen": -34.072509765625, "logps/rejected": -194.00869750976562, "loss": 0.0099, "rewards/accuracies": 1.0, "rewards/chosen": -1.21720290184021, "rewards/margins": 16.749324798583984, "rewards/rejected": -17.96652603149414, "step": 190 }, { "epoch": 1.4, "learning_rate": 1.6731517509727624e-07, "logits/chosen": -2.6063685417175293, "logits/rejected": -2.5614001750946045, "logps/chosen": -41.58473587036133, "logps/rejected": -219.73104858398438, "loss": 0.0121, "rewards/accuracies": 0.987500011920929, "rewards/chosen": -1.829288125038147, "rewards/margins": 18.672809600830078, "rewards/rejected": -20.502098083496094, "step": 200 }, { "epoch": 1.47, "learning_rate": 1.4785992217898832e-07, "logits/chosen": -2.7379167079925537, "logits/rejected": -2.674070119857788, "logps/chosen": -35.14494323730469, "logps/rejected": -182.973388671875, "loss": 0.0093, "rewards/accuracies": 1.0, "rewards/chosen": -1.3816758394241333, "rewards/margins": 15.531547546386719, "rewards/rejected": -16.913225173950195, "step": 210 }, { "epoch": 1.54, "learning_rate": 1.2840466926070036e-07, "logits/chosen": -2.7067456245422363, "logits/rejected": -2.632633686065674, "logps/chosen": -34.483577728271484, "logps/rejected": -222.17990112304688, "loss": 0.0054, "rewards/accuracies": 1.0, "rewards/chosen": -0.9836966395378113, "rewards/margins": 19.769222259521484, "rewards/rejected": -20.752918243408203, "step": 220 }, { "epoch": 1.61, "learning_rate": 1.0894941634241245e-07, "logits/chosen": -2.608846664428711, "logits/rejected": -2.604660749435425, "logps/chosen": -32.35722732543945, "logps/rejected": -246.7973175048828, "loss": 0.0076, "rewards/accuracies": 1.0, "rewards/chosen": -1.2495921850204468, "rewards/margins": 22.08835220336914, "rewards/rejected": -23.33794403076172, "step": 230 }, { "epoch": 1.68, "learning_rate": 8.949416342412451e-08, "logits/chosen": -2.607797622680664, "logits/rejected": -2.566305637359619, "logps/chosen": -27.041118621826172, "logps/rejected": -209.3827667236328, "loss": 0.0122, "rewards/accuracies": 1.0, "rewards/chosen": -0.9218173027038574, "rewards/margins": 18.870208740234375, "rewards/rejected": -19.79202651977539, "step": 240 }, { "epoch": 1.75, "learning_rate": 7.003891050583658e-08, "logits/chosen": -2.518440008163452, "logits/rejected": -2.5331220626831055, "logps/chosen": -31.538745880126953, "logps/rejected": -247.35238647460938, "loss": 0.004, "rewards/accuracies": 1.0, "rewards/chosen": -1.2074544429779053, "rewards/margins": 22.2752742767334, "rewards/rejected": -23.48272705078125, "step": 250 }, { "epoch": 1.82, "learning_rate": 5.058365758754864e-08, "logits/chosen": -2.5735764503479004, "logits/rejected": -2.52424693107605, "logps/chosen": -28.21148681640625, "logps/rejected": -196.92958068847656, "loss": 0.0018, "rewards/accuracies": 1.0, "rewards/chosen": -1.0159205198287964, "rewards/margins": 17.290904998779297, "rewards/rejected": -18.306825637817383, "step": 260 }, { "epoch": 1.89, "learning_rate": 3.11284046692607e-08, "logits/chosen": -2.643744945526123, "logits/rejected": -2.626131772994995, "logps/chosen": -29.778423309326172, "logps/rejected": -234.2376251220703, "loss": 0.0074, "rewards/accuracies": 1.0, "rewards/chosen": -1.0768438577651978, "rewards/margins": 20.66169548034668, "rewards/rejected": -21.738544464111328, "step": 270 }, { "epoch": 1.96, "learning_rate": 1.1673151750972762e-08, "logits/chosen": -2.467341899871826, "logits/rejected": -2.524524211883545, "logps/chosen": -29.820302963256836, "logps/rejected": -301.8487243652344, "loss": 0.0019, "rewards/accuracies": 1.0, "rewards/chosen": -0.9357167482376099, "rewards/margins": 27.8673152923584, "rewards/rejected": -28.80303382873535, "step": 280 }, { "epoch": 2.0, "step": 286, "total_flos": 0.0, "train_loss": 0.06552632141962543, "train_runtime": 3687.7187, "train_samples_per_second": 2.478, "train_steps_per_second": 0.078 } ], "logging_steps": 10, "max_steps": 286, "num_input_tokens_seen": 0, "num_train_epochs": 2, "save_steps": 500, "total_flos": 0.0, "train_batch_size": 8, "trial_name": null, "trial_params": null }