{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.9980806142034548, "eval_steps": 1000000, "global_step": 390, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0, "grad_norm": 1292.2103490533002, "learning_rate": 1.282051282051282e-08, "logits/chosen": -2.5583817958831787, "logits/rejected": -2.4487552642822266, "logps/chosen": -258.1644592285156, "logps/rejected": -216.25729370117188, "loss": 0.6964, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 1 }, { "epoch": 0.03, "grad_norm": 1261.1198733310573, "learning_rate": 1.2820512820512818e-07, "logits/chosen": -2.6061007976531982, "logits/rejected": -2.553147315979004, "logps/chosen": -267.5506591796875, "logps/rejected": -217.63583374023438, "loss": 0.7045, "rewards/accuracies": 0.4375, "rewards/chosen": 0.012439604848623276, "rewards/margins": 0.01009546872228384, "rewards/rejected": 0.002344133099541068, "step": 10 }, { "epoch": 0.05, "grad_norm": 853.2707411908161, "learning_rate": 2.5641025641025636e-07, "logits/chosen": -2.629751443862915, "logits/rejected": -2.5669989585876465, "logps/chosen": -260.5412292480469, "logps/rejected": -207.0039825439453, "loss": 0.5192, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": 0.6866833567619324, "rewards/margins": 0.6415502429008484, "rewards/rejected": 0.045133065432310104, "step": 20 }, { "epoch": 0.08, "grad_norm": 1142.6320871817904, "learning_rate": 3.8461538461538463e-07, "logits/chosen": -2.6441783905029297, "logits/rejected": -2.5700392723083496, "logps/chosen": -251.12313842773438, "logps/rejected": -198.34071350097656, "loss": 0.3383, "rewards/accuracies": 0.856249988079071, "rewards/chosen": 3.621673583984375, "rewards/margins": 3.1466643810272217, "rewards/rejected": 0.47500935196876526, "step": 30 }, { "epoch": 0.1, "grad_norm": 498.4795422948633, "learning_rate": 4.99989986344963e-07, "logits/chosen": -2.6392924785614014, "logits/rejected": -2.561156749725342, "logps/chosen": -243.8309783935547, "logps/rejected": -192.75845336914062, "loss": 0.3207, "rewards/accuracies": 0.8500000238418579, "rewards/chosen": 5.866530418395996, "rewards/margins": 5.031473636627197, "rewards/rejected": 0.8350569009780884, "step": 40 }, { "epoch": 0.13, "grad_norm": 687.258084457276, "learning_rate": 4.987893180827479e-07, "logits/chosen": -2.6534440517425537, "logits/rejected": -2.5842742919921875, "logps/chosen": -258.9037170410156, "logps/rejected": -203.41665649414062, "loss": 0.3666, "rewards/accuracies": 0.8687499761581421, "rewards/chosen": 8.400110244750977, "rewards/margins": 6.8410444259643555, "rewards/rejected": 1.559065818786621, "step": 50 }, { "epoch": 0.15, "grad_norm": 759.1516730607283, "learning_rate": 4.955969343539162e-07, "logits/chosen": -2.616115093231201, "logits/rejected": -2.5453152656555176, "logps/chosen": -262.9014892578125, "logps/rejected": -209.2264404296875, "loss": 0.3916, "rewards/accuracies": 0.862500011920929, "rewards/chosen": 5.333600044250488, "rewards/margins": 6.3847455978393555, "rewards/rejected": -1.0511456727981567, "step": 60 }, { "epoch": 0.18, "grad_norm": 560.0502144214485, "learning_rate": 4.90438392204474e-07, "logits/chosen": -2.590355634689331, "logits/rejected": -2.5212607383728027, "logps/chosen": -292.03955078125, "logps/rejected": -227.56900024414062, "loss": 0.366, "rewards/accuracies": 0.918749988079071, "rewards/chosen": 5.28397798538208, "rewards/margins": 7.424477577209473, "rewards/rejected": -2.1404995918273926, "step": 70 }, { "epoch": 0.2, "grad_norm": 1037.6404299504059, "learning_rate": 4.83354989019146e-07, "logits/chosen": -2.5414998531341553, "logits/rejected": -2.468327283859253, "logps/chosen": -260.0003356933594, "logps/rejected": -203.8599090576172, "loss": 0.3627, "rewards/accuracies": 0.875, "rewards/chosen": 6.802065372467041, "rewards/margins": 7.507330894470215, "rewards/rejected": -0.705264687538147, "step": 80 }, { "epoch": 0.23, "grad_norm": 1095.0164975833522, "learning_rate": 4.7440343190975353e-07, "logits/chosen": -2.5701346397399902, "logits/rejected": -2.513249397277832, "logps/chosen": -257.8319396972656, "logps/rejected": -217.47695922851562, "loss": 0.3529, "rewards/accuracies": 0.84375, "rewards/chosen": 3.5220108032226562, "rewards/margins": 6.205338478088379, "rewards/rejected": -2.683328151702881, "step": 90 }, { "epoch": 0.26, "grad_norm": 493.8561719761038, "learning_rate": 4.6365538373900506e-07, "logits/chosen": -2.624537944793701, "logits/rejected": -2.5495262145996094, "logps/chosen": -237.0422821044922, "logps/rejected": -201.1056671142578, "loss": 0.6296, "rewards/accuracies": 0.8687499761581421, "rewards/chosen": 4.613960266113281, "rewards/margins": 6.805362701416016, "rewards/rejected": -2.191401958465576, "step": 100 }, { "epoch": 0.28, "grad_norm": 783.3339585792174, "learning_rate": 4.5119688941406386e-07, "logits/chosen": -2.6185028553009033, "logits/rejected": -2.5373358726501465, "logps/chosen": -258.2229919433594, "logps/rejected": -209.95010375976562, "loss": 0.5085, "rewards/accuracies": 0.862500011920929, "rewards/chosen": 6.706292629241943, "rewards/margins": 8.07982349395752, "rewards/rejected": -1.3735301494598389, "step": 110 }, { "epoch": 0.31, "grad_norm": 1013.1425815966796, "learning_rate": 4.3712768704277524e-07, "logits/chosen": -2.5965781211853027, "logits/rejected": -2.5263915061950684, "logps/chosen": -263.1861572265625, "logps/rejected": -208.8050079345703, "loss": 0.4699, "rewards/accuracies": 0.8812500238418579, "rewards/chosen": 5.6486382484436035, "rewards/margins": 7.843411445617676, "rewards/rejected": -2.194772720336914, "step": 120 }, { "epoch": 0.33, "grad_norm": 659.1716861853262, "learning_rate": 4.2156040946718343e-07, "logits/chosen": -2.571542263031006, "logits/rejected": -2.502825975418091, "logps/chosen": -252.36941528320312, "logps/rejected": -197.37109375, "loss": 0.4289, "rewards/accuracies": 0.8374999761581421, "rewards/chosen": 4.440427303314209, "rewards/margins": 7.714503288269043, "rewards/rejected": -3.274075746536255, "step": 130 }, { "epoch": 0.36, "grad_norm": 706.862930333198, "learning_rate": 4.046196825665637e-07, "logits/chosen": -2.5883898735046387, "logits/rejected": -2.5185062885284424, "logps/chosen": -271.88494873046875, "logps/rejected": -218.22323608398438, "loss": 0.4943, "rewards/accuracies": 0.84375, "rewards/chosen": 2.9622278213500977, "rewards/margins": 7.161763668060303, "rewards/rejected": -4.199534893035889, "step": 140 }, { "epoch": 0.38, "grad_norm": 715.990249793818, "learning_rate": 3.864411275486261e-07, "logits/chosen": -2.5699923038482666, "logits/rejected": -2.502592086791992, "logps/chosen": -264.418212890625, "logps/rejected": -213.1693115234375, "loss": 0.5111, "rewards/accuracies": 0.875, "rewards/chosen": 5.556182384490967, "rewards/margins": 8.052742004394531, "rewards/rejected": -2.4965591430664062, "step": 150 }, { "epoch": 0.41, "grad_norm": 710.6520782907305, "learning_rate": 3.671702752161759e-07, "logits/chosen": -2.559586763381958, "logits/rejected": -2.4897525310516357, "logps/chosen": -245.1999969482422, "logps/rejected": -198.19442749023438, "loss": 0.4696, "rewards/accuracies": 0.875, "rewards/chosen": 3.364131212234497, "rewards/margins": 8.019769668579102, "rewards/rejected": -4.655638694763184, "step": 160 }, { "epoch": 0.44, "grad_norm": 865.8215089865342, "learning_rate": 3.4696140090121375e-07, "logits/chosen": -2.57694935798645, "logits/rejected": -2.5136027336120605, "logps/chosen": -266.91412353515625, "logps/rejected": -211.6914825439453, "loss": 0.397, "rewards/accuracies": 0.862500011920929, "rewards/chosen": 3.399867296218872, "rewards/margins": 8.182429313659668, "rewards/rejected": -4.782561302185059, "step": 170 }, { "epoch": 0.46, "grad_norm": 658.5054936178984, "learning_rate": 3.259762893935617e-07, "logits/chosen": -2.6607277393341064, "logits/rejected": -2.581068515777588, "logps/chosen": -239.2282257080078, "logps/rejected": -188.63772583007812, "loss": 0.4691, "rewards/accuracies": 0.8687499761581421, "rewards/chosen": 2.5246458053588867, "rewards/margins": 7.373200416564941, "rewards/rejected": -4.848554611206055, "step": 180 }, { "epoch": 0.49, "grad_norm": 746.921617266345, "learning_rate": 3.0438293975154184e-07, "logits/chosen": -2.6273646354675293, "logits/rejected": -2.5594921112060547, "logps/chosen": -263.73443603515625, "logps/rejected": -207.8045654296875, "loss": 0.4399, "rewards/accuracies": 0.8687499761581421, "rewards/chosen": 0.9775351285934448, "rewards/margins": 8.604721069335938, "rewards/rejected": -7.6271867752075195, "step": 190 }, { "epoch": 0.51, "grad_norm": 859.0725144089504, "learning_rate": 2.823542203635138e-07, "logits/chosen": -2.6617209911346436, "logits/rejected": -2.5823440551757812, "logps/chosen": -278.6440734863281, "logps/rejected": -222.3423309326172, "loss": 0.493, "rewards/accuracies": 0.887499988079071, "rewards/chosen": 1.6911423206329346, "rewards/margins": 9.542765617370605, "rewards/rejected": -7.851624488830566, "step": 200 }, { "epoch": 0.54, "grad_norm": 1073.7778122145994, "learning_rate": 2.600664850273538e-07, "logits/chosen": -2.6473941802978516, "logits/rejected": -2.581458568572998, "logps/chosen": -270.93621826171875, "logps/rejected": -214.9030303955078, "loss": 0.7378, "rewards/accuracies": 0.8374999761581421, "rewards/chosen": 0.10583686828613281, "rewards/margins": 7.7938642501831055, "rewards/rejected": -7.688027858734131, "step": 210 }, { "epoch": 0.56, "grad_norm": 783.9779969138841, "learning_rate": 2.3769816112703045e-07, "logits/chosen": -2.6640350818634033, "logits/rejected": -2.606898307800293, "logps/chosen": -260.6977233886719, "logps/rejected": -216.708740234375, "loss": 0.5298, "rewards/accuracies": 0.8687499761581421, "rewards/chosen": 0.881191611289978, "rewards/margins": 7.128912448883057, "rewards/rejected": -6.247720241546631, "step": 220 }, { "epoch": 0.59, "grad_norm": 656.1996441950056, "learning_rate": 2.1542832120881677e-07, "logits/chosen": -2.6948089599609375, "logits/rejected": -2.615889072418213, "logps/chosen": -270.77557373046875, "logps/rejected": -219.38320922851562, "loss": 0.4547, "rewards/accuracies": 0.856249988079071, "rewards/chosen": 2.3647713661193848, "rewards/margins": 8.186816215515137, "rewards/rejected": -5.82204532623291, "step": 230 }, { "epoch": 0.61, "grad_norm": 1015.3762946599339, "learning_rate": 1.934352493925695e-07, "logits/chosen": -2.6751513481140137, "logits/rejected": -2.6343884468078613, "logps/chosen": -266.14263916015625, "logps/rejected": -223.5273895263672, "loss": 0.429, "rewards/accuracies": 0.8999999761581421, "rewards/chosen": 1.4297797679901123, "rewards/margins": 10.390259742736816, "rewards/rejected": -8.960479736328125, "step": 240 }, { "epoch": 0.64, "grad_norm": 586.0366917195352, "learning_rate": 1.7189501409486059e-07, "logits/chosen": -2.6815237998962402, "logits/rejected": -2.6186068058013916, "logps/chosen": -271.32818603515625, "logps/rejected": -225.2965545654297, "loss": 0.4243, "rewards/accuracies": 0.8374999761581421, "rewards/chosen": 0.4054955542087555, "rewards/margins": 8.429037094116211, "rewards/rejected": -8.023542404174805, "step": 250 }, { "epoch": 0.67, "grad_norm": 567.2324686905487, "learning_rate": 1.5098005849021078e-07, "logits/chosen": -2.6706247329711914, "logits/rejected": -2.6183619499206543, "logps/chosen": -265.3566589355469, "logps/rejected": -211.55789184570312, "loss": 0.4026, "rewards/accuracies": 0.856249988079071, "rewards/chosen": 0.452970027923584, "rewards/margins": 7.969753265380859, "rewards/rejected": -7.516783237457275, "step": 260 }, { "epoch": 0.69, "grad_norm": 651.6922692823155, "learning_rate": 1.30857819994673e-07, "logits/chosen": -2.648026943206787, "logits/rejected": -2.5697438716888428, "logps/chosen": -276.09661865234375, "logps/rejected": -233.24966430664062, "loss": 0.4615, "rewards/accuracies": 0.875, "rewards/chosen": 0.4385630190372467, "rewards/margins": 11.782003402709961, "rewards/rejected": -11.343441009521484, "step": 270 }, { "epoch": 0.72, "grad_norm": 492.898374979646, "learning_rate": 1.116893898236716e-07, "logits/chosen": -2.6764636039733887, "logits/rejected": -2.6275370121002197, "logps/chosen": -273.55120849609375, "logps/rejected": -222.28256225585938, "loss": 0.431, "rewards/accuracies": 0.8687499761581421, "rewards/chosen": 0.1396121382713318, "rewards/margins": 8.543681144714355, "rewards/rejected": -8.404068946838379, "step": 280 }, { "epoch": 0.74, "grad_norm": 601.4691313440442, "learning_rate": 9.362822335518062e-08, "logits/chosen": -2.639714241027832, "logits/rejected": -2.6006019115448, "logps/chosen": -271.51104736328125, "logps/rejected": -219.78646850585938, "loss": 0.3785, "rewards/accuracies": 0.887499988079071, "rewards/chosen": 0.39154329895973206, "rewards/margins": 8.193072319030762, "rewards/rejected": -7.8015289306640625, "step": 290 }, { "epoch": 0.77, "grad_norm": 876.6568683843423, "learning_rate": 7.681891162260015e-08, "logits/chosen": -2.6577625274658203, "logits/rejected": -2.610130786895752, "logps/chosen": -278.3086853027344, "logps/rejected": -223.9765625, "loss": 0.4175, "rewards/accuracies": 0.862500011920929, "rewards/chosen": 0.23941349983215332, "rewards/margins": 8.298933029174805, "rewards/rejected": -8.059518814086914, "step": 300 }, { "epoch": 0.79, "grad_norm": 491.2362210232499, "learning_rate": 6.139602377230247e-08, "logits/chosen": -2.623286724090576, "logits/rejected": -2.5633485317230225, "logps/chosen": -281.3511047363281, "logps/rejected": -218.5331268310547, "loss": 0.4291, "rewards/accuracies": 0.875, "rewards/chosen": 0.8305476903915405, "rewards/margins": 8.879867553710938, "rewards/rejected": -8.049318313598633, "step": 310 }, { "epoch": 0.82, "grad_norm": 762.3598954431114, "learning_rate": 4.748302975270837e-08, "logits/chosen": -2.6497387886047363, "logits/rejected": -2.610168933868408, "logps/chosen": -264.21588134765625, "logps/rejected": -206.8611602783203, "loss": 0.4476, "rewards/accuracies": 0.875, "rewards/chosen": 0.6263083219528198, "rewards/margins": 7.853513240814209, "rewards/rejected": -7.227204322814941, "step": 320 }, { "epoch": 0.84, "grad_norm": 668.8344256472935, "learning_rate": 3.5191311859445795e-08, "logits/chosen": -2.6693060398101807, "logits/rejected": -2.6197760105133057, "logps/chosen": -268.38543701171875, "logps/rejected": -220.74081420898438, "loss": 0.4113, "rewards/accuracies": 0.90625, "rewards/chosen": 1.1819716691970825, "rewards/margins": 8.392292976379395, "rewards/rejected": -7.210320949554443, "step": 330 }, { "epoch": 0.87, "grad_norm": 600.9418350776771, "learning_rate": 2.4619273049795996e-08, "logits/chosen": -2.653104066848755, "logits/rejected": -2.6034810543060303, "logps/chosen": -263.5737609863281, "logps/rejected": -213.79544067382812, "loss": 0.3933, "rewards/accuracies": 0.9125000238418579, "rewards/chosen": 1.1136972904205322, "rewards/margins": 9.145872116088867, "rewards/rejected": -8.032175064086914, "step": 340 }, { "epoch": 0.9, "grad_norm": 448.6960620183454, "learning_rate": 1.5851549164932115e-08, "logits/chosen": -2.663357734680176, "logits/rejected": -2.6194939613342285, "logps/chosen": -272.70233154296875, "logps/rejected": -229.30697631835938, "loss": 0.4249, "rewards/accuracies": 0.887499988079071, "rewards/chosen": 1.3936760425567627, "rewards/margins": 8.432661056518555, "rewards/rejected": -7.0389862060546875, "step": 350 }, { "epoch": 0.92, "grad_norm": 606.8337189375926, "learning_rate": 8.958331366609423e-09, "logits/chosen": -2.666874885559082, "logits/rejected": -2.6079227924346924, "logps/chosen": -278.23712158203125, "logps/rejected": -222.1784210205078, "loss": 0.4777, "rewards/accuracies": 0.893750011920929, "rewards/chosen": 0.8818317651748657, "rewards/margins": 8.486412048339844, "rewards/rejected": -7.604579925537109, "step": 360 }, { "epoch": 0.95, "grad_norm": 565.1329564450134, "learning_rate": 3.994804212627461e-09, "logits/chosen": -2.626528739929199, "logits/rejected": -2.5960590839385986, "logps/chosen": -277.90789794921875, "logps/rejected": -232.6277313232422, "loss": 0.4164, "rewards/accuracies": 0.84375, "rewards/chosen": 1.226670503616333, "rewards/margins": 8.475992202758789, "rewards/rejected": -7.249321937561035, "step": 370 }, { "epoch": 0.97, "grad_norm": 494.78405010061283, "learning_rate": 1.0007038696262516e-09, "logits/chosen": -2.672760486602783, "logits/rejected": -2.6364920139312744, "logps/chosen": -266.59918212890625, "logps/rejected": -233.4165496826172, "loss": 0.4064, "rewards/accuracies": 0.862500011920929, "rewards/chosen": 1.6075836420059204, "rewards/margins": 8.459333419799805, "rewards/rejected": -6.851749420166016, "step": 380 }, { "epoch": 1.0, "grad_norm": 917.0969156966863, "learning_rate": 0.0, "logits/chosen": -2.6849241256713867, "logits/rejected": -2.630876302719116, "logps/chosen": -254.27664184570312, "logps/rejected": -213.14102172851562, "loss": 0.4983, "rewards/accuracies": 0.862500011920929, "rewards/chosen": 0.6460372805595398, "rewards/margins": 8.144643783569336, "rewards/rejected": -7.498606204986572, "step": 390 }, { "epoch": 1.0, "step": 390, "total_flos": 0.0, "train_loss": 0.4513903107398596, "train_runtime": 5854.463, "train_samples_per_second": 8.54, "train_steps_per_second": 0.067 } ], "logging_steps": 10, "max_steps": 390, "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 }