{ "best_metric": null, "best_model_checkpoint": null, "epoch": 1.0, "eval_steps": 100, "global_step": 328, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0, "grad_norm": 5.561485449698178, "learning_rate": 1.5151515151515152e-07, "logits/chosen": -0.4639463722705841, "logits/rejected": -0.17450889945030212, "logps/chosen": -612.9666748046875, "logps/rejected": -1227.4912109375, "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": 5.576753539954733, "learning_rate": 1.5151515151515152e-06, "logits/chosen": -0.41454941034317017, "logits/rejected": -0.577505350112915, "logps/chosen": -869.0648803710938, "logps/rejected": -1276.6817626953125, "loss": 0.6913, "rewards/accuracies": 0.5, "rewards/chosen": -0.0039122202433645725, "rewards/margins": 0.003580428659915924, "rewards/rejected": -0.007492648903280497, "step": 10 }, { "epoch": 0.06, "grad_norm": 6.198667632843589, "learning_rate": 3.0303030303030305e-06, "logits/chosen": -0.3819029927253723, "logits/rejected": -0.46113020181655884, "logps/chosen": -997.2161865234375, "logps/rejected": -1214.39990234375, "loss": 0.6612, "rewards/accuracies": 0.643750011920929, "rewards/chosen": -0.15533407032489777, "rewards/margins": 0.047475699335336685, "rewards/rejected": -0.20280976593494415, "step": 20 }, { "epoch": 0.09, "grad_norm": 5.889725033116788, "learning_rate": 4.5454545454545455e-06, "logits/chosen": -0.24505889415740967, "logits/rejected": -0.24203690886497498, "logps/chosen": -911.0994873046875, "logps/rejected": -1363.7496337890625, "loss": 0.541, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": -0.39497110247612, "rewards/margins": 0.5585557222366333, "rewards/rejected": -0.9535268545150757, "step": 30 }, { "epoch": 0.12, "grad_norm": 4.086637897535325, "learning_rate": 4.9930567839810125e-06, "logits/chosen": -0.17689356207847595, "logits/rejected": -0.16114920377731323, "logps/chosen": -1011.1546630859375, "logps/rejected": -1480.3541259765625, "loss": 0.4484, "rewards/accuracies": 0.7875000238418579, "rewards/chosen": -0.3060458302497864, "rewards/margins": 1.0700631141662598, "rewards/rejected": -1.3761091232299805, "step": 40 }, { "epoch": 0.15, "grad_norm": 4.522119109661391, "learning_rate": 4.959142005221991e-06, "logits/chosen": -0.11198903620243073, "logits/rejected": -0.15591135621070862, "logps/chosen": -899.4259033203125, "logps/rejected": -1485.4820556640625, "loss": 0.4169, "rewards/accuracies": 0.800000011920929, "rewards/chosen": -0.2616763412952423, "rewards/margins": 1.9235137701034546, "rewards/rejected": -2.185190439224243, "step": 50 }, { "epoch": 0.18, "grad_norm": 5.555413111169151, "learning_rate": 4.897364164920515e-06, "logits/chosen": -0.1413438618183136, "logits/rejected": -0.248577281832695, "logps/chosen": -856.3390502929688, "logps/rejected": -1524.873291015625, "loss": 0.3649, "rewards/accuracies": 0.824999988079071, "rewards/chosen": -0.2767874300479889, "rewards/margins": 1.9685882329940796, "rewards/rejected": -2.245375871658325, "step": 60 }, { "epoch": 0.21, "grad_norm": 5.814409469094451, "learning_rate": 4.808423230692374e-06, "logits/chosen": -0.23499484360218048, "logits/rejected": -0.3045286536216736, "logps/chosen": -930.0250854492188, "logps/rejected": -1563.339599609375, "loss": 0.3102, "rewards/accuracies": 0.856249988079071, "rewards/chosen": -0.22926001250743866, "rewards/margins": 2.3029136657714844, "rewards/rejected": -2.5321736335754395, "step": 70 }, { "epoch": 0.24, "grad_norm": 7.080973924268642, "learning_rate": 4.693326938861367e-06, "logits/chosen": -0.19219891726970673, "logits/rejected": -0.29335862398147583, "logps/chosen": -1067.5966796875, "logps/rejected": -1646.7294921875, "loss": 0.3024, "rewards/accuracies": 0.8812500238418579, "rewards/chosen": -0.3816714882850647, "rewards/margins": 2.590569019317627, "rewards/rejected": -2.972240447998047, "step": 80 }, { "epoch": 0.27, "grad_norm": 4.198144592019097, "learning_rate": 4.553379376404085e-06, "logits/chosen": -0.16756731271743774, "logits/rejected": -0.2766783535480499, "logps/chosen": -901.4991455078125, "logps/rejected": -1794.3072509765625, "loss": 0.249, "rewards/accuracies": 0.925000011920929, "rewards/chosen": -0.32676273584365845, "rewards/margins": 4.70559549331665, "rewards/rejected": -5.032357215881348, "step": 90 }, { "epoch": 0.3, "grad_norm": 3.514239110297717, "learning_rate": 4.3901662051233755e-06, "logits/chosen": -0.13113151490688324, "logits/rejected": -0.23850134015083313, "logps/chosen": -943.93701171875, "logps/rejected": -1710.283203125, "loss": 0.2635, "rewards/accuracies": 0.90625, "rewards/chosen": -0.5341259837150574, "rewards/margins": 4.174313545227051, "rewards/rejected": -4.708439826965332, "step": 100 }, { "epoch": 0.3, "eval_logits/chosen": -0.3091484606266022, "eval_logits/rejected": -0.10848761349916458, "eval_logps/chosen": -380.92041015625, "eval_logps/rejected": -1646.8138427734375, "eval_loss": 0.199021577835083, "eval_rewards/accuracies": 0.9166666865348816, "eval_rewards/chosen": -0.5114283561706543, "eval_rewards/margins": 9.306483268737793, "eval_rewards/rejected": -9.817912101745605, "eval_runtime": 29.898, "eval_samples_per_second": 6.422, "eval_steps_per_second": 0.201, "step": 100 }, { "epoch": 0.34, "grad_norm": 2.7779512658548504, "learning_rate": 4.205536695466524e-06, "logits/chosen": -0.13202831149101257, "logits/rejected": -0.31658726930618286, "logps/chosen": -881.8616333007812, "logps/rejected": -1843.311767578125, "loss": 0.2265, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": -0.4379568099975586, "rewards/margins": 4.332350254058838, "rewards/rejected": -4.7703070640563965, "step": 110 }, { "epoch": 0.37, "grad_norm": 3.058731694797242, "learning_rate": 4.001582773552153e-06, "logits/chosen": -0.2267748862504959, "logits/rejected": -0.37614327669143677, "logps/chosen": -1021.9287109375, "logps/rejected": -1790.741455078125, "loss": 0.2448, "rewards/accuracies": 0.925000011920929, "rewards/chosen": -0.7997490167617798, "rewards/margins": 3.615135908126831, "rewards/rejected": -4.4148850440979, "step": 120 }, { "epoch": 0.4, "grad_norm": 3.5646553088179957, "learning_rate": 3.7806153188114027e-06, "logits/chosen": -0.14952275156974792, "logits/rejected": -0.26807302236557007, "logps/chosen": -988.8695068359375, "logps/rejected": -1687.877197265625, "loss": 0.199, "rewards/accuracies": 0.956250011920929, "rewards/chosen": -0.5245489478111267, "rewards/margins": 2.954119920730591, "rewards/rejected": -3.4786689281463623, "step": 130 }, { "epoch": 0.43, "grad_norm": 4.994600163477232, "learning_rate": 3.5451379808006014e-06, "logits/chosen": -0.1331457495689392, "logits/rejected": -0.23687326908111572, "logps/chosen": -885.8282470703125, "logps/rejected": -1685.4974365234375, "loss": 0.2011, "rewards/accuracies": 0.918749988079071, "rewards/chosen": -0.609821617603302, "rewards/margins": 3.710085391998291, "rewards/rejected": -4.319907188415527, "step": 140 }, { "epoch": 0.46, "grad_norm": 4.000160476158621, "learning_rate": 3.2978188118513814e-06, "logits/chosen": -0.13606548309326172, "logits/rejected": -0.32709023356437683, "logps/chosen": -931.5257568359375, "logps/rejected": -1801.125, "loss": 0.1791, "rewards/accuracies": 0.96875, "rewards/chosen": -0.4420163631439209, "rewards/margins": 4.814242362976074, "rewards/rejected": -5.256258964538574, "step": 150 }, { "epoch": 0.49, "grad_norm": 5.561015522393931, "learning_rate": 3.041460036971664e-06, "logits/chosen": -0.20122027397155762, "logits/rejected": -0.37438589334487915, "logps/chosen": -960.2258911132812, "logps/rejected": -1840.1077880859375, "loss": 0.1646, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": -0.548378050327301, "rewards/margins": 5.22580623626709, "rewards/rejected": -5.774184703826904, "step": 160 }, { "epoch": 0.52, "grad_norm": 3.0160283224172506, "learning_rate": 2.7789663035166035e-06, "logits/chosen": -0.2793598771095276, "logits/rejected": -0.4746064245700836, "logps/chosen": -1049.3095703125, "logps/rejected": -1993.3609619140625, "loss": 0.1263, "rewards/accuracies": 0.96875, "rewards/chosen": -0.5465644598007202, "rewards/margins": 6.312196731567383, "rewards/rejected": -6.858761787414551, "step": 170 }, { "epoch": 0.55, "grad_norm": 2.93594175440836, "learning_rate": 2.513311770373421e-06, "logits/chosen": -0.31422796845436096, "logits/rejected": -0.5171582102775574, "logps/chosen": -960.77978515625, "logps/rejected": -1946.6195068359375, "loss": 0.1157, "rewards/accuracies": 0.949999988079071, "rewards/chosen": -0.6941060423851013, "rewards/margins": 5.883716583251953, "rewards/rejected": -6.577822685241699, "step": 180 }, { "epoch": 0.58, "grad_norm": 3.989156252845766, "learning_rate": 2.247506409552795e-06, "logits/chosen": -0.3567090332508087, "logits/rejected": -0.5727603435516357, "logps/chosen": -1111.8785400390625, "logps/rejected": -2083.59716796875, "loss": 0.1418, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": -0.5756088495254517, "rewards/margins": 6.701145172119141, "rewards/rejected": -7.276753902435303, "step": 190 }, { "epoch": 0.61, "grad_norm": 3.220493331131803, "learning_rate": 1.9845619020032552e-06, "logits/chosen": -0.4283437132835388, "logits/rejected": -0.6103153228759766, "logps/chosen": -977.1068115234375, "logps/rejected": -1989.1517333984375, "loss": 0.1415, "rewards/accuracies": 0.956250011920929, "rewards/chosen": -0.26823538541793823, "rewards/margins": 6.212519645690918, "rewards/rejected": -6.480755805969238, "step": 200 }, { "epoch": 0.61, "eval_logits/chosen": -0.48242926597595215, "eval_logits/rejected": -0.42804303765296936, "eval_logps/chosen": -420.5835876464844, "eval_logps/rejected": -1771.5301513671875, "eval_loss": 0.15017390251159668, "eval_rewards/accuracies": 0.9166666865348816, "eval_rewards/chosen": -0.9080603122711182, "eval_rewards/margins": 10.157015800476074, "eval_rewards/rejected": -11.065074920654297, "eval_runtime": 29.9491, "eval_samples_per_second": 6.411, "eval_steps_per_second": 0.2, "step": 200 }, { "epoch": 0.64, "grad_norm": 5.03992800632795, "learning_rate": 1.7274575140626318e-06, "logits/chosen": -0.40144267678260803, "logits/rejected": -0.6470441818237305, "logps/chosen": -992.0281372070312, "logps/rejected": -1956.958984375, "loss": 0.1581, "rewards/accuracies": 0.9312499761581421, "rewards/chosen": -0.683698832988739, "rewards/margins": 6.963919162750244, "rewards/rejected": -7.647618293762207, "step": 210 }, { "epoch": 0.67, "grad_norm": 2.5930618000995724, "learning_rate": 1.4791063411799938e-06, "logits/chosen": -0.4398053288459778, "logits/rejected": -0.6155021786689758, "logps/chosen": -901.8489990234375, "logps/rejected": -1997.6116943359375, "loss": 0.1258, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": -0.4012775421142578, "rewards/margins": 6.776458740234375, "rewards/rejected": -7.177737236022949, "step": 220 }, { "epoch": 0.7, "grad_norm": 3.126671458993684, "learning_rate": 1.2423223013801946e-06, "logits/chosen": -0.4161323606967926, "logits/rejected": -0.6325975656509399, "logps/chosen": -993.7923583984375, "logps/rejected": -2073.91064453125, "loss": 0.1237, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": -0.23834288120269775, "rewards/margins": 6.739261627197266, "rewards/rejected": -6.977604866027832, "step": 230 }, { "epoch": 0.73, "grad_norm": 3.053249955458482, "learning_rate": 1.019788252448267e-06, "logits/chosen": -0.41894397139549255, "logits/rejected": -0.5673054456710815, "logps/chosen": -946.7174682617188, "logps/rejected": -2171.60009765625, "loss": 0.0947, "rewards/accuracies": 0.981249988079071, "rewards/chosen": -0.34920141100883484, "rewards/margins": 8.334190368652344, "rewards/rejected": -8.683391571044922, "step": 240 }, { "epoch": 0.76, "grad_norm": 3.2492894915267074, "learning_rate": 8.140255940787059e-07, "logits/chosen": -0.4345846176147461, "logits/rejected": -0.5576716661453247, "logps/chosen": -990.5559692382812, "logps/rejected": -2134.9990234375, "loss": 0.1116, "rewards/accuracies": 0.956250011920929, "rewards/chosen": -0.7247036695480347, "rewards/margins": 7.789909362792969, "rewards/rejected": -8.514613151550293, "step": 250 }, { "epoch": 0.79, "grad_norm": 2.064615334983041, "learning_rate": 6.273656994094232e-07, "logits/chosen": -0.4094814658164978, "logits/rejected": -0.6032238006591797, "logps/chosen": -973.67041015625, "logps/rejected": -2180.19677734375, "loss": 0.1092, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": -0.7021085023880005, "rewards/margins": 8.40003776550293, "rewards/rejected": -9.102147102355957, "step": 260 }, { "epoch": 0.82, "grad_norm": 2.2441288839699647, "learning_rate": 4.619234996325314e-07, "logits/chosen": -0.4224594235420227, "logits/rejected": -0.5946496725082397, "logps/chosen": -1018.1749267578125, "logps/rejected": -2218.16552734375, "loss": 0.0905, "rewards/accuracies": 0.981249988079071, "rewards/chosen": -0.5338882803916931, "rewards/margins": 8.705224990844727, "rewards/rejected": -9.23911190032959, "step": 270 }, { "epoch": 0.85, "grad_norm": 4.0256681132412035, "learning_rate": 3.195735209788528e-07, "logits/chosen": -0.37011057138442993, "logits/rejected": -0.6081669330596924, "logps/chosen": -967.2252197265625, "logps/rejected": -2145.68798828125, "loss": 0.1005, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": -0.7551916837692261, "rewards/margins": 7.838353157043457, "rewards/rejected": -8.593544960021973, "step": 280 }, { "epoch": 0.88, "grad_norm": 4.259685670607253, "learning_rate": 2.019286455866981e-07, "logits/chosen": -0.41176638007164, "logits/rejected": -0.5948176980018616, "logps/chosen": -1099.498779296875, "logps/rejected": -2072.706298828125, "loss": 0.1126, "rewards/accuracies": 0.956250011920929, "rewards/chosen": -0.7403249740600586, "rewards/margins": 6.6156768798828125, "rewards/rejected": -7.356001853942871, "step": 290 }, { "epoch": 0.91, "grad_norm": 2.837717904545823, "learning_rate": 1.1032183690276754e-07, "logits/chosen": -0.38978689908981323, "logits/rejected": -0.554754376411438, "logps/chosen": -1006.9293212890625, "logps/rejected": -2237.94970703125, "loss": 0.0892, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": -0.7637746334075928, "rewards/margins": 8.33720874786377, "rewards/rejected": -9.100981712341309, "step": 300 }, { "epoch": 0.91, "eval_logits/chosen": -0.48870301246643066, "eval_logits/rejected": -0.40482982993125916, "eval_logps/chosen": -455.8128662109375, "eval_logps/rejected": -1987.8828125, "eval_loss": 0.14211897552013397, "eval_rewards/accuracies": 0.9166666865348816, "eval_rewards/chosen": -1.2603524923324585, "eval_rewards/margins": 11.968250274658203, "eval_rewards/rejected": -13.228604316711426, "eval_runtime": 29.6183, "eval_samples_per_second": 6.482, "eval_steps_per_second": 0.203, "step": 300 }, { "epoch": 0.95, "grad_norm": 2.6436300030745867, "learning_rate": 4.579103667367385e-08, "logits/chosen": -0.4472607672214508, "logits/rejected": -0.6000027656555176, "logps/chosen": -1035.2442626953125, "logps/rejected": -2228.332275390625, "loss": 0.1212, "rewards/accuracies": 0.96875, "rewards/chosen": -0.6586896777153015, "rewards/margins": 9.017664909362793, "rewards/rejected": -9.676355361938477, "step": 310 }, { "epoch": 0.98, "grad_norm": 4.053048154617744, "learning_rate": 9.067404651211808e-09, "logits/chosen": -0.42106541991233826, "logits/rejected": -0.5793359279632568, "logps/chosen": -992.0040283203125, "logps/rejected": -2126.542724609375, "loss": 0.0966, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": -0.767730176448822, "rewards/margins": 8.566682815551758, "rewards/rejected": -9.334413528442383, "step": 320 }, { "epoch": 1.0, "step": 328, "total_flos": 0.0, "train_loss": 0.22599055091055428, "train_runtime": 7017.6011, "train_samples_per_second": 2.987, "train_steps_per_second": 0.047 } ], "logging_steps": 10, "max_steps": 328, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 100, "total_flos": 0.0, "train_batch_size": 8, "trial_name": null, "trial_params": null }