{ "best_global_step": null, "best_metric": null, "best_model_checkpoint": null, "epoch": 0.1430316755564826, "eval_steps": 300, "global_step": 150, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0009535445037098841, "grad_norm": 9.156327247619629, "learning_rate": 3.7735849056603773e-06, "logits/chosen": -0.6435090899467468, "logits/rejected": -0.6501147150993347, "logps/chosen": -12.8789701461792, "logps/rejected": -12.67190170288086, "loss": 1.3995760679244995, "memory(GiB)": 46.35, "nll_loss": 0.7064292430877686, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 1, "train_speed(iter/s)": 0.005503 }, { "epoch": 0.0019070890074197682, "grad_norm": 3.750561237335205, "learning_rate": 7.547169811320755e-06, "logits/chosen": -0.5811247229576111, "logits/rejected": -0.5784763693809509, "logps/chosen": -11.550155639648438, "logps/rejected": -10.463687896728516, "loss": 1.1846946477890015, "memory(GiB)": 46.35, "nll_loss": 0.4915475845336914, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 2, "train_speed(iter/s)": 0.005627 }, { "epoch": 0.002860633511129652, "grad_norm": 9.728832244873047, "learning_rate": 1.1320754716981132e-05, "logits/chosen": -0.6211333274841309, "logits/rejected": -0.6250367760658264, "logps/chosen": -13.265661239624023, "logps/rejected": -17.363025665283203, "loss": 1.4022446870803833, "memory(GiB)": 46.35, "nll_loss": 0.7087438106536865, "rewards/accuracies": 0.46875, "rewards/chosen": 0.0019078836776316166, "rewards/margins": -0.0005585900507867336, "rewards/rejected": 0.002466473262757063, "step": 3, "train_speed(iter/s)": 0.005589 }, { "epoch": 0.0038141780148395364, "grad_norm": 4.456181049346924, "learning_rate": 1.509433962264151e-05, "logits/chosen": -0.6295092105865479, "logits/rejected": -0.6310409307479858, "logps/chosen": -20.259197235107422, "logps/rejected": -15.924739837646484, "loss": 1.4261250495910645, "memory(GiB)": 46.35, "nll_loss": 0.7309765815734863, "rewards/accuracies": 0.4375, "rewards/chosen": -0.003216111334040761, "rewards/margins": -0.003784709610044956, "rewards/rejected": 0.0005685988580808043, "step": 4, "train_speed(iter/s)": 0.005576 }, { "epoch": 0.004767722518549421, "grad_norm": 5.137070178985596, "learning_rate": 1.8867924528301888e-05, "logits/chosen": -0.6352660059928894, "logits/rejected": -0.6393146514892578, "logps/chosen": -11.320538520812988, "logps/rejected": -18.213350296020508, "loss": 1.3472224473953247, "memory(GiB)": 46.35, "nll_loss": 0.659798800945282, "rewards/accuracies": 0.5, "rewards/chosen": 0.009231260977685452, "rewards/margins": 0.011925339698791504, "rewards/rejected": -0.0026940787211060524, "step": 5, "train_speed(iter/s)": 0.005563 }, { "epoch": 0.005721267022259304, "grad_norm": 4.738046169281006, "learning_rate": 2.2641509433962265e-05, "logits/chosen": -0.621586799621582, "logits/rejected": -0.6225624680519104, "logps/chosen": -13.36612319946289, "logps/rejected": -12.092449188232422, "loss": 1.2847044467926025, "memory(GiB)": 46.35, "nll_loss": 0.593052089214325, "rewards/accuracies": 0.46875, "rewards/chosen": 0.001901240088045597, "rewards/margins": 0.0050404006615281105, "rewards/rejected": -0.0031391603406518698, "step": 6, "train_speed(iter/s)": 0.005557 }, { "epoch": 0.0066748115259691885, "grad_norm": 4.152891159057617, "learning_rate": 2.641509433962264e-05, "logits/chosen": -0.6192730665206909, "logits/rejected": -0.6235961318016052, "logps/chosen": -12.630643844604492, "logps/rejected": -15.917936325073242, "loss": 1.2920165061950684, "memory(GiB)": 46.35, "nll_loss": 0.5905699133872986, "rewards/accuracies": 0.4375, "rewards/chosen": 0.016993187367916107, "rewards/margins": -0.008502829819917679, "rewards/rejected": 0.025496022775769234, "step": 7, "train_speed(iter/s)": 0.005538 }, { "epoch": 0.007628356029679073, "grad_norm": 4.160374641418457, "learning_rate": 3.018867924528302e-05, "logits/chosen": -0.6182655692100525, "logits/rejected": -0.6189552545547485, "logps/chosen": -10.698919296264648, "logps/rejected": -14.148299217224121, "loss": 1.0840333700180054, "memory(GiB)": 46.35, "nll_loss": 0.3833664357662201, "rewards/accuracies": 0.5, "rewards/chosen": -0.006827078759670258, "rewards/margins": 0.02289848029613495, "rewards/rejected": -0.0297255702316761, "step": 8, "train_speed(iter/s)": 0.005529 }, { "epoch": 0.008581900533388956, "grad_norm": 7.751194953918457, "learning_rate": 3.39622641509434e-05, "logits/chosen": -0.5885738134384155, "logits/rejected": -0.5893712043762207, "logps/chosen": -14.677619934082031, "logps/rejected": -19.348114013671875, "loss": 1.4759101867675781, "memory(GiB)": 46.35, "nll_loss": 0.7508934140205383, "rewards/accuracies": 0.53125, "rewards/chosen": -0.06757626682519913, "rewards/margins": -0.03109188750386238, "rewards/rejected": -0.036484383046627045, "step": 9, "train_speed(iter/s)": 0.005546 }, { "epoch": 0.009535445037098841, "grad_norm": 5.425357818603516, "learning_rate": 3.7735849056603776e-05, "logits/chosen": -0.6194073557853699, "logits/rejected": -0.6175965666770935, "logps/chosen": -16.864004135131836, "logps/rejected": -13.315808296203613, "loss": 1.264159083366394, "memory(GiB)": 46.35, "nll_loss": 0.5726930499076843, "rewards/accuracies": 0.5, "rewards/chosen": -0.03342372551560402, "rewards/margins": 0.05060018226504326, "rewards/rejected": -0.08402392268180847, "step": 10, "train_speed(iter/s)": 0.005559 }, { "epoch": 0.010488989540808725, "grad_norm": 4.181962013244629, "learning_rate": 4.150943396226415e-05, "logits/chosen": -0.6295148730278015, "logits/rejected": -0.6311144232749939, "logps/chosen": -17.622665405273438, "logps/rejected": -22.323240280151367, "loss": 1.1960029602050781, "memory(GiB)": 46.35, "nll_loss": 0.4701622426509857, "rewards/accuracies": 0.46875, "rewards/chosen": -0.05372938886284828, "rewards/margins": -0.0011130962520837784, "rewards/rejected": -0.05261629447340965, "step": 11, "train_speed(iter/s)": 0.00555 }, { "epoch": 0.011442534044518608, "grad_norm": 3.3646111488342285, "learning_rate": 4.528301886792453e-05, "logits/chosen": -0.6567115783691406, "logits/rejected": -0.6585782170295715, "logps/chosen": -11.29936408996582, "logps/rejected": -18.29203987121582, "loss": 1.1115467548370361, "memory(GiB)": 46.35, "nll_loss": 0.4559721350669861, "rewards/accuracies": 0.625, "rewards/chosen": 0.08929000794887543, "rewards/margins": 0.12125876545906067, "rewards/rejected": -0.03196876123547554, "step": 12, "train_speed(iter/s)": 0.005551 }, { "epoch": 0.012396078548228493, "grad_norm": 3.79887318611145, "learning_rate": 4.9056603773584906e-05, "logits/chosen": -0.5698614716529846, "logits/rejected": -0.5705659985542297, "logps/chosen": -16.54460906982422, "logps/rejected": -16.52007484436035, "loss": 1.2756099700927734, "memory(GiB)": 46.35, "nll_loss": 0.6095594167709351, "rewards/accuracies": 0.53125, "rewards/chosen": 0.08876940608024597, "rewards/margins": 0.0944020226597786, "rewards/rejected": -0.005632609128952026, "step": 13, "train_speed(iter/s)": 0.005543 }, { "epoch": 0.013349623051938377, "grad_norm": 3.3852715492248535, "learning_rate": 5.283018867924528e-05, "logits/chosen": -0.5822104811668396, "logits/rejected": -0.5813897848129272, "logps/chosen": -15.234231948852539, "logps/rejected": -15.407049179077148, "loss": 1.2322038412094116, "memory(GiB)": 46.35, "nll_loss": 0.6042034029960632, "rewards/accuracies": 0.6875, "rewards/chosen": 0.1384495347738266, "rewards/margins": 0.16144631803035736, "rewards/rejected": -0.022996775805950165, "step": 14, "train_speed(iter/s)": 0.005541 }, { "epoch": 0.014303167555648262, "grad_norm": 2.7666451930999756, "learning_rate": 5.660377358490566e-05, "logits/chosen": -0.5926915407180786, "logits/rejected": -0.5929235816001892, "logps/chosen": -14.370649337768555, "logps/rejected": -14.672292709350586, "loss": 1.268752098083496, "memory(GiB)": 46.35, "nll_loss": 0.5840609669685364, "rewards/accuracies": 0.53125, "rewards/chosen": 0.13987205922603607, "rewards/margins": 0.04084888845682144, "rewards/rejected": 0.09902316331863403, "step": 15, "train_speed(iter/s)": 0.005545 }, { "epoch": 0.015256712059358146, "grad_norm": 3.996248960494995, "learning_rate": 6.037735849056604e-05, "logits/chosen": -0.6161852478981018, "logits/rejected": -0.6143717169761658, "logps/chosen": -17.775665283203125, "logps/rejected": -16.12677001953125, "loss": 1.3597179651260376, "memory(GiB)": 46.35, "nll_loss": 0.7031959295272827, "rewards/accuracies": 0.59375, "rewards/chosen": 0.14894011616706848, "rewards/margins": 0.10105747729539871, "rewards/rejected": 0.047882646322250366, "step": 16, "train_speed(iter/s)": 0.005542 }, { "epoch": 0.01621025656306803, "grad_norm": 3.343315601348877, "learning_rate": 6.415094339622641e-05, "logits/chosen": -0.5822757482528687, "logits/rejected": -0.5849615931510925, "logps/chosen": -11.210283279418945, "logps/rejected": -17.080455780029297, "loss": 1.1165493726730347, "memory(GiB)": 46.35, "nll_loss": 0.4471588432788849, "rewards/accuracies": 0.5625, "rewards/chosen": 0.13627491891384125, "rewards/margins": 0.07742398977279663, "rewards/rejected": 0.058850936591625214, "step": 17, "train_speed(iter/s)": 0.005542 }, { "epoch": 0.017163801066777912, "grad_norm": 2.5265791416168213, "learning_rate": 6.79245283018868e-05, "logits/chosen": -0.5711070895195007, "logits/rejected": -0.5730242729187012, "logps/chosen": -14.197075843811035, "logps/rejected": -15.04891586303711, "loss": 1.1029906272888184, "memory(GiB)": 46.35, "nll_loss": 0.45255714654922485, "rewards/accuracies": 0.53125, "rewards/chosen": 0.1637343168258667, "rewards/margins": 0.1597798466682434, "rewards/rejected": 0.0039544664323329926, "step": 18, "train_speed(iter/s)": 0.005535 }, { "epoch": 0.0181173455704878, "grad_norm": 4.558610916137695, "learning_rate": 7.169811320754717e-05, "logits/chosen": -0.572235643863678, "logits/rejected": -0.5743283629417419, "logps/chosen": -9.268120765686035, "logps/rejected": -12.118350982666016, "loss": 1.1668801307678223, "memory(GiB)": 46.35, "nll_loss": 0.5335167646408081, "rewards/accuracies": 0.6875, "rewards/chosen": 0.19031290709972382, "rewards/margins": 0.15231794118881226, "rewards/rejected": 0.03799496963620186, "step": 19, "train_speed(iter/s)": 0.005519 }, { "epoch": 0.019070890074197683, "grad_norm": 3.7134947776794434, "learning_rate": 7.547169811320755e-05, "logits/chosen": -0.585778534412384, "logits/rejected": -0.5890673398971558, "logps/chosen": -8.275850296020508, "logps/rejected": -16.348182678222656, "loss": 1.0512793064117432, "memory(GiB)": 46.35, "nll_loss": 0.39501887559890747, "rewards/accuracies": 0.5625, "rewards/chosen": 0.1688304841518402, "rewards/margins": 0.12996332347393036, "rewards/rejected": 0.03886716440320015, "step": 20, "train_speed(iter/s)": 0.005525 }, { "epoch": 0.020024434577907566, "grad_norm": 3.5586955547332764, "learning_rate": 7.924528301886794e-05, "logits/chosen": -0.5529674291610718, "logits/rejected": -0.5549667477607727, "logps/chosen": -14.883947372436523, "logps/rejected": -19.51848793029785, "loss": 1.2289440631866455, "memory(GiB)": 46.35, "nll_loss": 0.5555365085601807, "rewards/accuracies": 0.625, "rewards/chosen": 0.06489060074090958, "rewards/margins": 0.08113954961299896, "rewards/rejected": -0.016248945146799088, "step": 21, "train_speed(iter/s)": 0.005523 }, { "epoch": 0.02097797908161745, "grad_norm": 2.970125913619995, "learning_rate": 8.30188679245283e-05, "logits/chosen": -0.5630563497543335, "logits/rejected": -0.5648879408836365, "logps/chosen": -9.709285736083984, "logps/rejected": -14.605964660644531, "loss": 1.0496439933776855, "memory(GiB)": 46.35, "nll_loss": 0.4030095934867859, "rewards/accuracies": 0.59375, "rewards/chosen": 0.13459333777427673, "rewards/margins": 0.13257721066474915, "rewards/rejected": 0.0020161233842372894, "step": 22, "train_speed(iter/s)": 0.005527 }, { "epoch": 0.021931523585327333, "grad_norm": 3.7449333667755127, "learning_rate": 8.679245283018869e-05, "logits/chosen": -0.5388307571411133, "logits/rejected": -0.5379776954650879, "logps/chosen": -20.206819534301758, "logps/rejected": -10.312554359436035, "loss": 1.3675460815429688, "memory(GiB)": 46.35, "nll_loss": 0.7346486449241638, "rewards/accuracies": 0.71875, "rewards/chosen": 0.22365330159664154, "rewards/margins": 0.19076848030090332, "rewards/rejected": 0.032884806394577026, "step": 23, "train_speed(iter/s)": 0.005527 }, { "epoch": 0.022885068089037217, "grad_norm": 4.3556928634643555, "learning_rate": 9.056603773584906e-05, "logits/chosen": -0.5620133876800537, "logits/rejected": -0.5651191473007202, "logps/chosen": -13.681132316589355, "logps/rejected": -17.737977981567383, "loss": 1.427290439605713, "memory(GiB)": 46.35, "nll_loss": 0.7979669570922852, "rewards/accuracies": 0.6875, "rewards/chosen": 0.20859572291374207, "rewards/margins": 0.19208168983459473, "rewards/rejected": 0.01651405170559883, "step": 24, "train_speed(iter/s)": 0.005528 }, { "epoch": 0.023838612592747103, "grad_norm": 2.59773850440979, "learning_rate": 9.433962264150944e-05, "logits/chosen": -0.5109421014785767, "logits/rejected": -0.5130017995834351, "logps/chosen": -11.422687530517578, "logps/rejected": -19.77487564086914, "loss": 1.0190178155899048, "memory(GiB)": 46.35, "nll_loss": 0.3792160749435425, "rewards/accuracies": 0.625, "rewards/chosen": 0.17286184430122375, "rewards/margins": 0.1656832993030548, "rewards/rejected": 0.007178528234362602, "step": 25, "train_speed(iter/s)": 0.005528 }, { "epoch": 0.024792157096456987, "grad_norm": 3.2465970516204834, "learning_rate": 9.811320754716981e-05, "logits/chosen": -0.4895856976509094, "logits/rejected": -0.490243524312973, "logps/chosen": -8.71623420715332, "logps/rejected": -7.8305277824401855, "loss": 1.0329375267028809, "memory(GiB)": 46.35, "nll_loss": 0.3276180028915405, "rewards/accuracies": 0.4375, "rewards/chosen": 0.13548454642295837, "rewards/margins": 0.014407174661755562, "rewards/rejected": 0.12107735872268677, "step": 26, "train_speed(iter/s)": 0.005523 }, { "epoch": 0.02574570160016687, "grad_norm": 2.9056291580200195, "learning_rate": 0.0001018867924528302, "logits/chosen": -0.5098599195480347, "logits/rejected": -0.5105721950531006, "logps/chosen": -17.795873641967773, "logps/rejected": -13.080977439880371, "loss": 1.1328891515731812, "memory(GiB)": 46.35, "nll_loss": 0.4805879294872284, "rewards/accuracies": 0.6875, "rewards/chosen": 0.18526741862297058, "rewards/margins": 0.11028175055980682, "rewards/rejected": 0.07498566806316376, "step": 27, "train_speed(iter/s)": 0.005528 }, { "epoch": 0.026699246103876754, "grad_norm": 3.9831905364990234, "learning_rate": 0.00010566037735849057, "logits/chosen": -0.5249112844467163, "logits/rejected": -0.5256497859954834, "logps/chosen": -9.17713737487793, "logps/rejected": -16.791439056396484, "loss": 1.0309749841690063, "memory(GiB)": 46.35, "nll_loss": 0.49379563331604004, "rewards/accuracies": 0.71875, "rewards/chosen": 0.26130685210227966, "rewards/margins": 0.4484291970729828, "rewards/rejected": -0.1871223747730255, "step": 28, "train_speed(iter/s)": 0.00553 }, { "epoch": 0.027652790607586637, "grad_norm": 3.447692394256592, "learning_rate": 0.00010943396226415095, "logits/chosen": -0.46926626563072205, "logits/rejected": -0.474046528339386, "logps/chosen": -10.564407348632812, "logps/rejected": -13.679093360900879, "loss": 1.0266515016555786, "memory(GiB)": 46.35, "nll_loss": 0.4402708113193512, "rewards/accuracies": 0.8125, "rewards/chosen": 0.25625136494636536, "rewards/margins": 0.27123552560806274, "rewards/rejected": -0.014984178356826305, "step": 29, "train_speed(iter/s)": 0.005529 }, { "epoch": 0.028606335111296524, "grad_norm": 2.3513646125793457, "learning_rate": 0.00011320754716981132, "logits/chosen": -0.4841839671134949, "logits/rejected": -0.4823201894760132, "logps/chosen": -16.52553367614746, "logps/rejected": -12.728946685791016, "loss": 1.051925539970398, "memory(GiB)": 46.35, "nll_loss": 0.5132753252983093, "rewards/accuracies": 0.875, "rewards/chosen": 0.28733503818511963, "rewards/margins": 0.3806324601173401, "rewards/rejected": -0.09329745918512344, "step": 30, "train_speed(iter/s)": 0.00553 }, { "epoch": 0.029559879615006408, "grad_norm": 2.8920440673828125, "learning_rate": 0.0001169811320754717, "logits/chosen": -0.4617501497268677, "logits/rejected": -0.4629114866256714, "logps/chosen": -9.827648162841797, "logps/rejected": -11.787566184997559, "loss": 1.0702807903289795, "memory(GiB)": 46.35, "nll_loss": 0.4221873879432678, "rewards/accuracies": 0.5625, "rewards/chosen": 0.12683041393756866, "rewards/margins": 0.20381762087345123, "rewards/rejected": -0.07698723673820496, "step": 31, "train_speed(iter/s)": 0.005529 }, { "epoch": 0.03051342411871629, "grad_norm": 3.150108575820923, "learning_rate": 0.00012075471698113207, "logits/chosen": -0.46168380975723267, "logits/rejected": -0.45957043766975403, "logps/chosen": -14.481122970581055, "logps/rejected": -12.509255409240723, "loss": 1.1068114042282104, "memory(GiB)": 46.35, "nll_loss": 0.4239238202571869, "rewards/accuracies": 0.53125, "rewards/chosen": 0.07766711711883545, "rewards/margins": 0.11046308279037476, "rewards/rejected": -0.03279596567153931, "step": 32, "train_speed(iter/s)": 0.005524 }, { "epoch": 0.03146696862242618, "grad_norm": 4.185500621795654, "learning_rate": 0.00012452830188679244, "logits/chosen": -0.4712987542152405, "logits/rejected": -0.47331178188323975, "logps/chosen": -13.179000854492188, "logps/rejected": -19.023120880126953, "loss": 1.3572198152542114, "memory(GiB)": 46.35, "nll_loss": 0.7039042711257935, "rewards/accuracies": 0.59375, "rewards/chosen": 0.17201806604862213, "rewards/margins": 0.16012105345726013, "rewards/rejected": 0.011897009797394276, "step": 33, "train_speed(iter/s)": 0.005529 }, { "epoch": 0.03242051312613606, "grad_norm": 3.005161762237549, "learning_rate": 0.00012830188679245283, "logits/chosen": -0.4754801392555237, "logits/rejected": -0.4769393503665924, "logps/chosen": -11.273967742919922, "logps/rejected": -16.496191024780273, "loss": 1.0072283744812012, "memory(GiB)": 46.35, "nll_loss": 0.41234296560287476, "rewards/accuracies": 0.75, "rewards/chosen": 0.22565320134162903, "rewards/margins": 0.27443331480026245, "rewards/rejected": -0.04878011345863342, "step": 34, "train_speed(iter/s)": 0.005526 }, { "epoch": 0.033374057629845945, "grad_norm": 5.103848457336426, "learning_rate": 0.0001320754716981132, "logits/chosen": -0.44018203020095825, "logits/rejected": -0.44089382886886597, "logps/chosen": -15.832134246826172, "logps/rejected": -19.550304412841797, "loss": 1.2559186220169067, "memory(GiB)": 46.35, "nll_loss": 0.5793285965919495, "rewards/accuracies": 0.53125, "rewards/chosen": 0.09596908837556839, "rewards/margins": 0.11671226471662521, "rewards/rejected": -0.020743174478411674, "step": 35, "train_speed(iter/s)": 0.005527 }, { "epoch": 0.034327602133555825, "grad_norm": 4.321141719818115, "learning_rate": 0.0001358490566037736, "logits/chosen": -0.49340876936912537, "logits/rejected": -0.49693170189857483, "logps/chosen": -17.06662940979004, "logps/rejected": -20.88056755065918, "loss": 1.2329245805740356, "memory(GiB)": 46.35, "nll_loss": 0.6244450807571411, "rewards/accuracies": 0.65625, "rewards/chosen": 0.2872624695301056, "rewards/margins": 0.23183389008045197, "rewards/rejected": 0.05542857572436333, "step": 36, "train_speed(iter/s)": 0.00553 }, { "epoch": 0.03528114663726571, "grad_norm": 3.112792491912842, "learning_rate": 0.00013962264150943395, "logits/chosen": -0.4647415578365326, "logits/rejected": -0.4648529291152954, "logps/chosen": -12.979642868041992, "logps/rejected": -13.105558395385742, "loss": 1.1428803205490112, "memory(GiB)": 46.35, "nll_loss": 0.48155656456947327, "rewards/accuracies": 0.53125, "rewards/chosen": 0.19289334118366241, "rewards/margins": 0.169374018907547, "rewards/rejected": 0.023519329726696014, "step": 37, "train_speed(iter/s)": 0.005535 }, { "epoch": 0.0362346911409756, "grad_norm": 4.363394260406494, "learning_rate": 0.00014339622641509434, "logits/chosen": -0.47051119804382324, "logits/rejected": -0.4702276289463043, "logps/chosen": -11.824732780456543, "logps/rejected": -18.102703094482422, "loss": 0.9977712631225586, "memory(GiB)": 46.35, "nll_loss": 0.48080000281333923, "rewards/accuracies": 0.78125, "rewards/chosen": 0.35463425517082214, "rewards/margins": 0.5215649008750916, "rewards/rejected": -0.166930690407753, "step": 38, "train_speed(iter/s)": 0.005537 }, { "epoch": 0.03718823564468548, "grad_norm": 2.3291094303131104, "learning_rate": 0.00014716981132075472, "logits/chosen": -0.4471542239189148, "logits/rejected": -0.44849908351898193, "logps/chosen": -10.541170120239258, "logps/rejected": -13.821127891540527, "loss": 1.0517562627792358, "memory(GiB)": 46.35, "nll_loss": 0.4245188236236572, "rewards/accuracies": 0.5625, "rewards/chosen": 0.17497971653938293, "rewards/margins": 0.2404995709657669, "rewards/rejected": -0.06551986932754517, "step": 39, "train_speed(iter/s)": 0.005533 }, { "epoch": 0.038141780148395366, "grad_norm": 2.5439131259918213, "learning_rate": 0.0001509433962264151, "logits/chosen": -0.43407440185546875, "logits/rejected": -0.4330942928791046, "logps/chosen": -12.606916427612305, "logps/rejected": -8.916302680969238, "loss": 1.1393225193023682, "memory(GiB)": 46.35, "nll_loss": 0.500526487827301, "rewards/accuracies": 0.625, "rewards/chosen": 0.15790946781635284, "rewards/margins": 0.28719791769981384, "rewards/rejected": -0.129288449883461, "step": 40, "train_speed(iter/s)": 0.005533 }, { "epoch": 0.039095324652105246, "grad_norm": 2.752281904220581, "learning_rate": 0.0001547169811320755, "logits/chosen": -0.4207141399383545, "logits/rejected": -0.42105770111083984, "logps/chosen": -9.193354606628418, "logps/rejected": -15.80713939666748, "loss": 1.0188286304473877, "memory(GiB)": 46.35, "nll_loss": 0.41979092359542847, "rewards/accuracies": 0.65625, "rewards/chosen": 0.11082945019006729, "rewards/margins": 0.3942355513572693, "rewards/rejected": -0.2834060788154602, "step": 41, "train_speed(iter/s)": 0.005532 }, { "epoch": 0.04004886915581513, "grad_norm": 2.4327123165130615, "learning_rate": 0.00015849056603773587, "logits/chosen": -0.445608913898468, "logits/rejected": -0.4462995231151581, "logps/chosen": -8.890420913696289, "logps/rejected": -9.104397773742676, "loss": 1.0348408222198486, "memory(GiB)": 46.35, "nll_loss": 0.4377897083759308, "rewards/accuracies": 0.71875, "rewards/chosen": 0.1634826809167862, "rewards/margins": 0.28891459107398987, "rewards/rejected": -0.12543192505836487, "step": 42, "train_speed(iter/s)": 0.005537 }, { "epoch": 0.04100241365952501, "grad_norm": 2.331282377243042, "learning_rate": 0.00016226415094339625, "logits/chosen": -0.4530089497566223, "logits/rejected": -0.4580504298210144, "logps/chosen": -7.02070426940918, "logps/rejected": -22.712093353271484, "loss": 0.876542329788208, "memory(GiB)": 46.35, "nll_loss": 0.3533342182636261, "rewards/accuracies": 0.78125, "rewards/chosen": 0.29712867736816406, "rewards/margins": 0.5244494676589966, "rewards/rejected": -0.22732074558734894, "step": 43, "train_speed(iter/s)": 0.005538 }, { "epoch": 0.0419559581632349, "grad_norm": 2.865417003631592, "learning_rate": 0.0001660377358490566, "logits/chosen": -0.4700409770011902, "logits/rejected": -0.4735233187675476, "logps/chosen": -12.126070022583008, "logps/rejected": -24.352678298950195, "loss": 1.157395362854004, "memory(GiB)": 46.35, "nll_loss": 0.5185694694519043, "rewards/accuracies": 0.65625, "rewards/chosen": 0.1286190301179886, "rewards/margins": 0.2031334638595581, "rewards/rejected": -0.07451442629098892, "step": 44, "train_speed(iter/s)": 0.005538 }, { "epoch": 0.042909502666944786, "grad_norm": 2.9107770919799805, "learning_rate": 0.000169811320754717, "logits/chosen": -0.45035916566848755, "logits/rejected": -0.4494967758655548, "logps/chosen": -17.021671295166016, "logps/rejected": -16.504793167114258, "loss": 1.2039690017700195, "memory(GiB)": 46.35, "nll_loss": 0.5990548729896545, "rewards/accuracies": 0.6875, "rewards/chosen": 0.22049224376678467, "rewards/margins": 0.25519850850105286, "rewards/rejected": -0.03470626473426819, "step": 45, "train_speed(iter/s)": 0.005534 }, { "epoch": 0.043863047170654666, "grad_norm": 2.5569493770599365, "learning_rate": 0.00017358490566037738, "logits/chosen": -0.4576927423477173, "logits/rejected": -0.4589836597442627, "logps/chosen": -15.28840160369873, "logps/rejected": -21.630834579467773, "loss": 1.0875338315963745, "memory(GiB)": 46.35, "nll_loss": 0.5480765104293823, "rewards/accuracies": 0.71875, "rewards/chosen": 0.2394683063030243, "rewards/margins": 0.43596386909484863, "rewards/rejected": -0.19649554789066315, "step": 46, "train_speed(iter/s)": 0.005534 }, { "epoch": 0.04481659167436455, "grad_norm": 2.6082468032836914, "learning_rate": 0.00017735849056603776, "logits/chosen": -0.44933271408081055, "logits/rejected": -0.4535468816757202, "logps/chosen": -13.067358016967773, "logps/rejected": -25.21561622619629, "loss": 0.9420796632766724, "memory(GiB)": 46.35, "nll_loss": 0.44500476121902466, "rewards/accuracies": 0.8125, "rewards/chosen": 0.3027465343475342, "rewards/margins": 0.5848984122276306, "rewards/rejected": -0.28215184807777405, "step": 47, "train_speed(iter/s)": 0.005534 }, { "epoch": 0.04577013617807443, "grad_norm": 2.8552000522613525, "learning_rate": 0.00018113207547169812, "logits/chosen": -0.44021791219711304, "logits/rejected": -0.4409497380256653, "logps/chosen": -13.639575958251953, "logps/rejected": -19.479185104370117, "loss": 1.0777249336242676, "memory(GiB)": 46.35, "nll_loss": 0.5401443243026733, "rewards/accuracies": 0.78125, "rewards/chosen": 0.2517797350883484, "rewards/margins": 0.4618395268917084, "rewards/rejected": -0.21005980670452118, "step": 48, "train_speed(iter/s)": 0.00553 }, { "epoch": 0.04672368068178432, "grad_norm": 2.530268430709839, "learning_rate": 0.0001849056603773585, "logits/chosen": -0.47215527296066284, "logits/rejected": -0.47504734992980957, "logps/chosen": -11.48988151550293, "logps/rejected": -36.38770294189453, "loss": 0.9700489640235901, "memory(GiB)": 46.35, "nll_loss": 0.46057599782943726, "rewards/accuracies": 0.8125, "rewards/chosen": 0.21084292232990265, "rewards/margins": 0.533352792263031, "rewards/rejected": -0.32250988483428955, "step": 49, "train_speed(iter/s)": 0.005532 }, { "epoch": 0.04767722518549421, "grad_norm": 2.7035655975341797, "learning_rate": 0.00018867924528301889, "logits/chosen": -0.4868597388267517, "logits/rejected": -0.4888378977775574, "logps/chosen": -13.504924774169922, "logps/rejected": -21.10767936706543, "loss": 1.0414435863494873, "memory(GiB)": 46.35, "nll_loss": 0.5301634669303894, "rewards/accuracies": 0.78125, "rewards/chosen": 0.18318726122379303, "rewards/margins": 0.619285523891449, "rewards/rejected": -0.43609827756881714, "step": 50, "train_speed(iter/s)": 0.005531 }, { "epoch": 0.04863076968920409, "grad_norm": 3.0783157348632812, "learning_rate": 0.00019245283018867927, "logits/chosen": -0.488979697227478, "logits/rejected": -0.4920251965522766, "logps/chosen": -6.389867305755615, "logps/rejected": -23.881044387817383, "loss": 0.7357016801834106, "memory(GiB)": 46.35, "nll_loss": 0.2519811689853668, "rewards/accuracies": 0.8125, "rewards/chosen": 0.22045277059078217, "rewards/margins": 0.6349221467971802, "rewards/rejected": -0.4144693613052368, "step": 51, "train_speed(iter/s)": 0.00552 }, { "epoch": 0.049584314192913974, "grad_norm": 2.936868906021118, "learning_rate": 0.00019622641509433963, "logits/chosen": -0.49481600522994995, "logits/rejected": -0.4982725977897644, "logps/chosen": -7.566017150878906, "logps/rejected": -28.77280044555664, "loss": 0.8173349499702454, "memory(GiB)": 46.35, "nll_loss": 0.2955441176891327, "rewards/accuracies": 0.8125, "rewards/chosen": 0.12177374958992004, "rewards/margins": 0.6433938145637512, "rewards/rejected": -0.5216200947761536, "step": 52, "train_speed(iter/s)": 0.005518 }, { "epoch": 0.050537858696623854, "grad_norm": 4.539004802703857, "learning_rate": 0.0002, "logits/chosen": -0.5527479648590088, "logits/rejected": -0.5551312565803528, "logps/chosen": -15.817987442016602, "logps/rejected": -22.29538345336914, "loss": 1.1207798719406128, "memory(GiB)": 46.35, "nll_loss": 0.5884667038917542, "rewards/accuracies": 0.75, "rewards/chosen": 0.18014809489250183, "rewards/margins": 0.6673915982246399, "rewards/rejected": -0.4872434735298157, "step": 53, "train_speed(iter/s)": 0.005519 }, { "epoch": 0.05149140320033374, "grad_norm": 4.143117904663086, "learning_rate": 0.00019999950154813253, "logits/chosen": -0.5354641675949097, "logits/rejected": -0.5355905294418335, "logps/chosen": -12.528202056884766, "logps/rejected": -18.404970169067383, "loss": 1.0412046909332275, "memory(GiB)": 46.35, "nll_loss": 0.5650326609611511, "rewards/accuracies": 0.8125, "rewards/chosen": 0.17625540494918823, "rewards/margins": 0.8302103877067566, "rewards/rejected": -0.6539549827575684, "step": 54, "train_speed(iter/s)": 0.005522 }, { "epoch": 0.05244494770404363, "grad_norm": 3.329041004180908, "learning_rate": 0.00019999800619749922, "logits/chosen": -0.5789484977722168, "logits/rejected": -0.5782615542411804, "logps/chosen": -14.283157348632812, "logps/rejected": -24.138996124267578, "loss": 1.0249700546264648, "memory(GiB)": 46.35, "nll_loss": 0.5574489831924438, "rewards/accuracies": 0.78125, "rewards/chosen": 0.08275524526834488, "rewards/margins": 0.9348181486129761, "rewards/rejected": -0.8520628809928894, "step": 55, "train_speed(iter/s)": 0.005525 }, { "epoch": 0.05339849220775351, "grad_norm": 13.451016426086426, "learning_rate": 0.00019999551396300723, "logits/chosen": -0.6104876399040222, "logits/rejected": -0.6144217848777771, "logps/chosen": -11.782422065734863, "logps/rejected": -26.71551513671875, "loss": 1.1726720333099365, "memory(GiB)": 46.35, "nll_loss": 0.6291560530662537, "rewards/accuracies": 0.75, "rewards/chosen": -0.0043334998190402985, "rewards/margins": 0.6636021137237549, "rewards/rejected": -0.6679355502128601, "step": 56, "train_speed(iter/s)": 0.005523 }, { "epoch": 0.054352036711463395, "grad_norm": 7.804044246673584, "learning_rate": 0.00019999202486950177, "logits/chosen": -0.5772112011909485, "logits/rejected": -0.5761610865592957, "logps/chosen": -23.908967971801758, "logps/rejected": -30.196250915527344, "loss": 1.3278322219848633, "memory(GiB)": 46.35, "nll_loss": 0.7636411786079407, "rewards/accuracies": 0.65625, "rewards/chosen": 0.15834051370620728, "rewards/margins": 0.4628582298755646, "rewards/rejected": -0.3045177161693573, "step": 57, "train_speed(iter/s)": 0.005522 }, { "epoch": 0.055305581215173275, "grad_norm": 7.5992913246154785, "learning_rate": 0.00019998753895176575, "logits/chosen": -0.6457486748695374, "logits/rejected": -0.6466608047485352, "logps/chosen": -10.734827041625977, "logps/rejected": -32.13155746459961, "loss": 1.0030261278152466, "memory(GiB)": 46.35, "nll_loss": 0.5355030298233032, "rewards/accuracies": 0.8125, "rewards/chosen": -0.01070893183350563, "rewards/margins": 0.7878642082214355, "rewards/rejected": -0.7985730171203613, "step": 58, "train_speed(iter/s)": 0.005524 }, { "epoch": 0.05625912571888316, "grad_norm": 5.3810601234436035, "learning_rate": 0.00019998205625451943, "logits/chosen": -0.6111299991607666, "logits/rejected": -0.6088098287582397, "logps/chosen": -16.970266342163086, "logps/rejected": -19.622106552124023, "loss": 0.9686940908432007, "memory(GiB)": 46.35, "nll_loss": 0.5118071436882019, "rewards/accuracies": 0.78125, "rewards/chosen": 0.15675117075443268, "rewards/margins": 0.8334945440292358, "rewards/rejected": -0.6767433881759644, "step": 59, "train_speed(iter/s)": 0.005525 }, { "epoch": 0.05721267022259305, "grad_norm": 3.480093002319336, "learning_rate": 0.00019997557683242004, "logits/chosen": -0.5985093712806702, "logits/rejected": -0.6007335186004639, "logps/chosen": -14.120916366577148, "logps/rejected": -24.104907989501953, "loss": 0.9003526568412781, "memory(GiB)": 46.35, "nll_loss": 0.466304212808609, "rewards/accuracies": 0.75, "rewards/chosen": 0.13193026185035706, "rewards/margins": 1.0070805549621582, "rewards/rejected": -0.8751502633094788, "step": 60, "train_speed(iter/s)": 0.005522 }, { "epoch": 0.05816621472630293, "grad_norm": 3.3445444107055664, "learning_rate": 0.00019996810075006117, "logits/chosen": -0.6064149141311646, "logits/rejected": -0.6075825691223145, "logps/chosen": -10.57797622680664, "logps/rejected": -21.275867462158203, "loss": 0.9624521732330322, "memory(GiB)": 46.35, "nll_loss": 0.41281017661094666, "rewards/accuracies": 0.6875, "rewards/chosen": 0.06739869713783264, "rewards/margins": 0.5023229122161865, "rewards/rejected": -0.43492427468299866, "step": 61, "train_speed(iter/s)": 0.005519 }, { "epoch": 0.059119759230012815, "grad_norm": 5.101805210113525, "learning_rate": 0.00019995962808197216, "logits/chosen": -0.6019335985183716, "logits/rejected": -0.6061650514602661, "logps/chosen": -12.45846939086914, "logps/rejected": -22.627965927124023, "loss": 1.1021240949630737, "memory(GiB)": 46.35, "nll_loss": 0.5154472589492798, "rewards/accuracies": 0.6875, "rewards/chosen": -0.0749291479587555, "rewards/margins": 0.5026906728744507, "rewards/rejected": -0.5776197910308838, "step": 62, "train_speed(iter/s)": 0.005522 }, { "epoch": 0.060073303733722695, "grad_norm": 3.7086844444274902, "learning_rate": 0.0001999501589126174, "logits/chosen": -0.598800539970398, "logits/rejected": -0.5973880887031555, "logps/chosen": -16.170019149780273, "logps/rejected": -24.100147247314453, "loss": 0.9728513956069946, "memory(GiB)": 46.35, "nll_loss": 0.4919710159301758, "rewards/accuracies": 0.78125, "rewards/chosen": 0.14402300119400024, "rewards/margins": 0.7554571628570557, "rewards/rejected": -0.6114341616630554, "step": 63, "train_speed(iter/s)": 0.005522 }, { "epoch": 0.06102684823743258, "grad_norm": 6.82169246673584, "learning_rate": 0.00019993969333639532, "logits/chosen": -0.5916394591331482, "logits/rejected": -0.5948237180709839, "logps/chosen": -6.504903793334961, "logps/rejected": -23.65030860900879, "loss": 0.8641103506088257, "memory(GiB)": 46.35, "nll_loss": 0.3815235197544098, "rewards/accuracies": 0.84375, "rewards/chosen": 0.06883495301008224, "rewards/margins": 0.6778679490089417, "rewards/rejected": -0.6090329885482788, "step": 64, "train_speed(iter/s)": 0.005524 }, { "epoch": 0.06198039274114246, "grad_norm": 6.41129207611084, "learning_rate": 0.0001999282314576377, "logits/chosen": -0.6117686033248901, "logits/rejected": -0.6133745312690735, "logps/chosen": -7.471805572509766, "logps/rejected": -19.006999969482422, "loss": 0.8599905371665955, "memory(GiB)": 46.35, "nll_loss": 0.3879605829715729, "rewards/accuracies": 0.84375, "rewards/chosen": 0.15403376519680023, "rewards/margins": 0.6219788193702698, "rewards/rejected": -0.4679450988769531, "step": 65, "train_speed(iter/s)": 0.005525 }, { "epoch": 0.06293393724485236, "grad_norm": 3.6126596927642822, "learning_rate": 0.00019991577339060842, "logits/chosen": -0.6491844654083252, "logits/rejected": -0.6502828001976013, "logps/chosen": -9.777487754821777, "logps/rejected": -19.9858455657959, "loss": 0.9987360835075378, "memory(GiB)": 46.35, "nll_loss": 0.4526561498641968, "rewards/accuracies": 0.8125, "rewards/chosen": 0.14287549257278442, "rewards/margins": 0.528179407119751, "rewards/rejected": -0.38530388474464417, "step": 66, "train_speed(iter/s)": 0.005524 }, { "epoch": 0.06388748174856224, "grad_norm": 3.5133907794952393, "learning_rate": 0.0001999023192595024, "logits/chosen": -0.6023973822593689, "logits/rejected": -0.6058337092399597, "logps/chosen": -11.287931442260742, "logps/rejected": -23.294925689697266, "loss": 0.949726939201355, "memory(GiB)": 46.35, "nll_loss": 0.49871301651000977, "rewards/accuracies": 0.8125, "rewards/chosen": 0.25335025787353516, "rewards/margins": 0.7444449663162231, "rewards/rejected": -0.491094708442688, "step": 67, "train_speed(iter/s)": 0.005525 }, { "epoch": 0.06484102625227212, "grad_norm": 4.242381572723389, "learning_rate": 0.00019988786919844436, "logits/chosen": -0.577968955039978, "logits/rejected": -0.584266185760498, "logps/chosen": -7.122470378875732, "logps/rejected": -23.440967559814453, "loss": 0.825219988822937, "memory(GiB)": 46.35, "nll_loss": 0.3752996325492859, "rewards/accuracies": 0.78125, "rewards/chosen": 0.28059279918670654, "rewards/margins": 0.8312418460845947, "rewards/rejected": -0.5506490468978882, "step": 68, "train_speed(iter/s)": 0.005524 }, { "epoch": 0.065794570755982, "grad_norm": 4.247474670410156, "learning_rate": 0.00019987242335148757, "logits/chosen": -0.6062893271446228, "logits/rejected": -0.6092509031295776, "logps/chosen": -9.965551376342773, "logps/rejected": -16.18874740600586, "loss": 1.0371503829956055, "memory(GiB)": 46.35, "nll_loss": 0.4371563792228699, "rewards/accuracies": 0.625, "rewards/chosen": 0.08624501526355743, "rewards/margins": 0.4600314795970917, "rewards/rejected": -0.37378647923469543, "step": 69, "train_speed(iter/s)": 0.005525 }, { "epoch": 0.06674811525969189, "grad_norm": 3.118610143661499, "learning_rate": 0.0001998559818726122, "logits/chosen": -0.651878297328949, "logits/rejected": -0.6511794328689575, "logps/chosen": -17.028488159179688, "logps/rejected": -18.37329864501953, "loss": 1.1015472412109375, "memory(GiB)": 46.35, "nll_loss": 0.5589602589607239, "rewards/accuracies": 0.6875, "rewards/chosen": 0.17374631762504578, "rewards/margins": 0.5566114187240601, "rewards/rejected": -0.3828650414943695, "step": 70, "train_speed(iter/s)": 0.005526 }, { "epoch": 0.06770165976340177, "grad_norm": 3.3582799434661865, "learning_rate": 0.00019983854492572394, "logits/chosen": -0.615595281124115, "logits/rejected": -0.6146353483200073, "logps/chosen": -12.519451141357422, "logps/rejected": -18.15428924560547, "loss": 0.9180172085762024, "memory(GiB)": 46.35, "nll_loss": 0.3709501624107361, "rewards/accuracies": 0.71875, "rewards/chosen": 0.09964233636856079, "rewards/margins": 0.6540214419364929, "rewards/rejected": -0.5543791651725769, "step": 71, "train_speed(iter/s)": 0.005525 }, { "epoch": 0.06865520426711165, "grad_norm": 2.883702516555786, "learning_rate": 0.00019982011268465243, "logits/chosen": -0.5868794918060303, "logits/rejected": -0.5905233025550842, "logps/chosen": -14.138279914855957, "logps/rejected": -27.953182220458984, "loss": 1.0747126340866089, "memory(GiB)": 46.35, "nll_loss": 0.46605974435806274, "rewards/accuracies": 0.6875, "rewards/chosen": 0.03510252386331558, "rewards/margins": 0.6189652681350708, "rewards/rejected": -0.5838626623153687, "step": 72, "train_speed(iter/s)": 0.005526 }, { "epoch": 0.06960874877082154, "grad_norm": 2.9610211849212646, "learning_rate": 0.00019980068533314934, "logits/chosen": -0.6018360257148743, "logits/rejected": -0.6016470789909363, "logps/chosen": -14.77442455291748, "logps/rejected": -16.293825149536133, "loss": 1.1167826652526855, "memory(GiB)": 46.35, "nll_loss": 0.5493913888931274, "rewards/accuracies": 0.78125, "rewards/chosen": 0.20179376006126404, "rewards/margins": 0.5395517945289612, "rewards/rejected": -0.3377581536769867, "step": 73, "train_speed(iter/s)": 0.005526 }, { "epoch": 0.07056229327453142, "grad_norm": 3.1142945289611816, "learning_rate": 0.00019978026306488668, "logits/chosen": -0.5901756286621094, "logits/rejected": -0.5861356258392334, "logps/chosen": -12.717107772827148, "logps/rejected": -19.3599796295166, "loss": 0.9646521210670471, "memory(GiB)": 46.35, "nll_loss": 0.42209792137145996, "rewards/accuracies": 0.65625, "rewards/chosen": 0.19932276010513306, "rewards/margins": 0.5936742424964905, "rewards/rejected": -0.3943515419960022, "step": 74, "train_speed(iter/s)": 0.005525 }, { "epoch": 0.0715158377782413, "grad_norm": 2.8381545543670654, "learning_rate": 0.00019975884608345476, "logits/chosen": -0.5791196227073669, "logits/rejected": -0.5793704390525818, "logps/chosen": -11.382536888122559, "logps/rejected": -17.99052619934082, "loss": 1.0251520872116089, "memory(GiB)": 46.35, "nll_loss": 0.5087648630142212, "rewards/accuracies": 0.8125, "rewards/chosen": 0.225914865732193, "rewards/margins": 0.544399619102478, "rewards/rejected": -0.3184847831726074, "step": 75, "train_speed(iter/s)": 0.005523 }, { "epoch": 0.0724693822819512, "grad_norm": 2.720726251602173, "learning_rate": 0.0001997364346023603, "logits/chosen": -0.5865166187286377, "logits/rejected": -0.5875246524810791, "logps/chosen": -14.62844467163086, "logps/rejected": -26.72977638244629, "loss": 0.9625909924507141, "memory(GiB)": 46.35, "nll_loss": 0.5162026882171631, "rewards/accuracies": 0.8125, "rewards/chosen": 0.2656521499156952, "rewards/margins": 0.8063244819641113, "rewards/rejected": -0.5406723022460938, "step": 76, "train_speed(iter/s)": 0.005522 }, { "epoch": 0.07342292678566108, "grad_norm": 7.292218208312988, "learning_rate": 0.0001997130288450242, "logits/chosen": -0.5742957592010498, "logits/rejected": -0.5747869610786438, "logps/chosen": -15.201194763183594, "logps/rejected": -21.86297607421875, "loss": 1.2068696022033691, "memory(GiB)": 46.35, "nll_loss": 0.7135031819343567, "rewards/accuracies": 0.75, "rewards/chosen": 0.23197884857654572, "rewards/margins": 0.7994663119316101, "rewards/rejected": -0.5674874782562256, "step": 77, "train_speed(iter/s)": 0.005522 }, { "epoch": 0.07437647128937096, "grad_norm": 3.270038366317749, "learning_rate": 0.00019968862904477935, "logits/chosen": -0.5858986377716064, "logits/rejected": -0.588924765586853, "logps/chosen": -9.959943771362305, "logps/rejected": -30.939224243164062, "loss": 0.8423591256141663, "memory(GiB)": 46.35, "nll_loss": 0.42530590295791626, "rewards/accuracies": 0.84375, "rewards/chosen": 0.16566459834575653, "rewards/margins": 1.2258960008621216, "rewards/rejected": -1.060231328010559, "step": 78, "train_speed(iter/s)": 0.005518 }, { "epoch": 0.07533001579308084, "grad_norm": 4.3024821281433105, "learning_rate": 0.00019966323544486818, "logits/chosen": -0.6145219802856445, "logits/rejected": -0.6170623898506165, "logps/chosen": -15.870341300964355, "logps/rejected": -31.39044952392578, "loss": 0.8412056565284729, "memory(GiB)": 46.35, "nll_loss": 0.4725872874259949, "rewards/accuracies": 0.875, "rewards/chosen": 0.3251492381095886, "rewards/margins": 1.2906752824783325, "rewards/rejected": -0.9655259251594543, "step": 79, "train_speed(iter/s)": 0.005516 }, { "epoch": 0.07628356029679073, "grad_norm": 5.5750732421875, "learning_rate": 0.00019963684829844052, "logits/chosen": -0.6042959094047546, "logits/rejected": -0.6049582362174988, "logps/chosen": -13.382628440856934, "logps/rejected": -21.927295684814453, "loss": 0.993065357208252, "memory(GiB)": 46.35, "nll_loss": 0.49278005957603455, "rewards/accuracies": 0.71875, "rewards/chosen": 0.10083693265914917, "rewards/margins": 0.7778795957565308, "rewards/rejected": -0.6770426630973816, "step": 80, "train_speed(iter/s)": 0.005517 }, { "epoch": 0.07723710480050061, "grad_norm": 7.816519260406494, "learning_rate": 0.0001996094678685508, "logits/chosen": -0.5581300258636475, "logits/rejected": -0.5570173263549805, "logps/chosen": -16.26945686340332, "logps/rejected": -28.130666732788086, "loss": 1.1555508375167847, "memory(GiB)": 46.35, "nll_loss": 0.5590921640396118, "rewards/accuracies": 0.65625, "rewards/chosen": -0.22734299302101135, "rewards/margins": 0.652872622013092, "rewards/rejected": -0.8802156448364258, "step": 81, "train_speed(iter/s)": 0.005518 }, { "epoch": 0.07819064930421049, "grad_norm": 4.555399417877197, "learning_rate": 0.00019958109442815553, "logits/chosen": -0.562245786190033, "logits/rejected": -0.5638425946235657, "logps/chosen": -17.669172286987305, "logps/rejected": -31.455379486083984, "loss": 1.0409901142120361, "memory(GiB)": 46.35, "nll_loss": 0.5925040245056152, "rewards/accuracies": 0.75, "rewards/chosen": -0.18633463978767395, "rewards/margins": 1.1558425426483154, "rewards/rejected": -1.3421772718429565, "step": 82, "train_speed(iter/s)": 0.005517 }, { "epoch": 0.07914419380792038, "grad_norm": 4.788554668426514, "learning_rate": 0.00019955172826011062, "logits/chosen": -0.53816819190979, "logits/rejected": -0.539014458656311, "logps/chosen": -13.409234046936035, "logps/rejected": -24.46169090270996, "loss": 1.0504765510559082, "memory(GiB)": 46.35, "nll_loss": 0.5079448819160461, "rewards/accuracies": 0.78125, "rewards/chosen": -0.030599184334278107, "rewards/margins": 0.6909205317497253, "rewards/rejected": -0.7215197086334229, "step": 83, "train_speed(iter/s)": 0.005518 }, { "epoch": 0.08009773831163026, "grad_norm": 2.9610979557037354, "learning_rate": 0.00019952136965716846, "logits/chosen": -0.5609460473060608, "logits/rejected": -0.5607710480690002, "logps/chosen": -15.43799114227295, "logps/rejected": -20.685916900634766, "loss": 0.9568982720375061, "memory(GiB)": 46.35, "nll_loss": 0.49247080087661743, "rewards/accuracies": 0.8125, "rewards/chosen": 0.2509080469608307, "rewards/margins": 0.9044961333274841, "rewards/rejected": -0.653588056564331, "step": 84, "train_speed(iter/s)": 0.005516 }, { "epoch": 0.08105128281534014, "grad_norm": 2.7049379348754883, "learning_rate": 0.00019949001892197515, "logits/chosen": -0.5431854128837585, "logits/rejected": -0.5459212064743042, "logps/chosen": -11.692042350769043, "logps/rejected": -25.70047378540039, "loss": 0.9204345941543579, "memory(GiB)": 46.35, "nll_loss": 0.5127474069595337, "rewards/accuracies": 0.84375, "rewards/chosen": 0.16273778676986694, "rewards/margins": 1.052358865737915, "rewards/rejected": -0.8896210193634033, "step": 85, "train_speed(iter/s)": 0.005518 }, { "epoch": 0.08200482731905002, "grad_norm": 2.835888624191284, "learning_rate": 0.00019945767636706728, "logits/chosen": -0.5524388551712036, "logits/rejected": -0.5518356561660767, "logps/chosen": -9.604469299316406, "logps/rejected": -18.570831298828125, "loss": 0.7963011860847473, "memory(GiB)": 46.35, "nll_loss": 0.35101598501205444, "rewards/accuracies": 0.78125, "rewards/chosen": 0.20881126821041107, "rewards/margins": 0.7979729771614075, "rewards/rejected": -0.58916175365448, "step": 86, "train_speed(iter/s)": 0.005516 }, { "epoch": 0.08295837182275992, "grad_norm": 4.458975315093994, "learning_rate": 0.00019942434231486902, "logits/chosen": -0.5035311579704285, "logits/rejected": -0.5041376352310181, "logps/chosen": -12.906057357788086, "logps/rejected": -21.71013641357422, "loss": 0.9929911494255066, "memory(GiB)": 46.35, "nll_loss": 0.5292322039604187, "rewards/accuracies": 0.78125, "rewards/chosen": 0.23509997129440308, "rewards/margins": 0.7336819171905518, "rewards/rejected": -0.49858203530311584, "step": 87, "train_speed(iter/s)": 0.005515 }, { "epoch": 0.0839119163264698, "grad_norm": 5.752819061279297, "learning_rate": 0.0001993900170976888, "logits/chosen": -0.5372301340103149, "logits/rejected": -0.5387623310089111, "logps/chosen": -12.41868782043457, "logps/rejected": -23.16730499267578, "loss": 0.9672287106513977, "memory(GiB)": 46.35, "nll_loss": 0.4671844244003296, "rewards/accuracies": 0.71875, "rewards/chosen": 0.22129932045936584, "rewards/margins": 0.8583503365516663, "rewards/rejected": -0.637050986289978, "step": 88, "train_speed(iter/s)": 0.005516 }, { "epoch": 0.08486546083017968, "grad_norm": 3.0927765369415283, "learning_rate": 0.00019935470105771598, "logits/chosen": -0.5962197780609131, "logits/rejected": -0.5928555130958557, "logps/chosen": -15.36233139038086, "logps/rejected": -16.967832565307617, "loss": 1.0067566633224487, "memory(GiB)": 46.35, "nll_loss": 0.5986248850822449, "rewards/accuracies": 0.84375, "rewards/chosen": 0.20106852054595947, "rewards/margins": 0.8472704291343689, "rewards/rejected": -0.6462019681930542, "step": 89, "train_speed(iter/s)": 0.005517 }, { "epoch": 0.08581900533388957, "grad_norm": 4.051328659057617, "learning_rate": 0.00019931839454701743, "logits/chosen": -0.5495755672454834, "logits/rejected": -0.54953533411026, "logps/chosen": -9.681739807128906, "logps/rejected": -17.38517951965332, "loss": 0.9176310896873474, "memory(GiB)": 46.35, "nll_loss": 0.4568890631198883, "rewards/accuracies": 0.8125, "rewards/chosen": 0.18029510974884033, "rewards/margins": 0.7410764694213867, "rewards/rejected": -0.5607813596725464, "step": 90, "train_speed(iter/s)": 0.005516 }, { "epoch": 0.08677254983759945, "grad_norm": 2.651430368423462, "learning_rate": 0.00019928109792753418, "logits/chosen": -0.5709578990936279, "logits/rejected": -0.5723389983177185, "logps/chosen": -12.097832679748535, "logps/rejected": -25.470287322998047, "loss": 0.8452543020248413, "memory(GiB)": 46.35, "nll_loss": 0.4146232008934021, "rewards/accuracies": 0.78125, "rewards/chosen": 0.2571614384651184, "rewards/margins": 0.9069947600364685, "rewards/rejected": -0.6498332619667053, "step": 91, "train_speed(iter/s)": 0.005515 }, { "epoch": 0.08772609434130933, "grad_norm": 3.9205963611602783, "learning_rate": 0.0001992428115710776, "logits/chosen": -0.5633900761604309, "logits/rejected": -0.5629602074623108, "logps/chosen": -15.341146469116211, "logps/rejected": -22.676523208618164, "loss": 1.0960146188735962, "memory(GiB)": 46.35, "nll_loss": 0.6228011846542358, "rewards/accuracies": 0.71875, "rewards/chosen": 0.2320929765701294, "rewards/margins": 0.802864670753479, "rewards/rejected": -0.5707716941833496, "step": 92, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.08867963884501923, "grad_norm": 2.4844846725463867, "learning_rate": 0.00019920353585932578, "logits/chosen": -0.5643666982650757, "logits/rejected": -0.5697072148323059, "logps/chosen": -7.112787246704102, "logps/rejected": -29.775066375732422, "loss": 0.7670912742614746, "memory(GiB)": 46.35, "nll_loss": 0.3726654350757599, "rewards/accuracies": 0.84375, "rewards/chosen": 0.11930885910987854, "rewards/margins": 1.221017837524414, "rewards/rejected": -1.101709008216858, "step": 93, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.0896331833487291, "grad_norm": 3.3409717082977295, "learning_rate": 0.00019916327118381982, "logits/chosen": -0.5507174730300903, "logits/rejected": -0.5550180673599243, "logps/chosen": -6.935158729553223, "logps/rejected": -26.466045379638672, "loss": 0.8178773522377014, "memory(GiB)": 46.35, "nll_loss": 0.3691931962966919, "rewards/accuracies": 0.84375, "rewards/chosen": -0.014170083217322826, "rewards/margins": 0.906183123588562, "rewards/rejected": -0.9203532338142395, "step": 94, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.09058672785243899, "grad_norm": 3.8606507778167725, "learning_rate": 0.0001991220179459597, "logits/chosen": -0.5692034959793091, "logits/rejected": -0.5723109841346741, "logps/chosen": -9.636994361877441, "logps/rejected": -28.736644744873047, "loss": 0.7489418387413025, "memory(GiB)": 46.35, "nll_loss": 0.38759085536003113, "rewards/accuracies": 0.8125, "rewards/chosen": 0.1789592206478119, "rewards/margins": 1.2420873641967773, "rewards/rejected": -1.0631282329559326, "step": 95, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.09154027235614887, "grad_norm": 4.69579553604126, "learning_rate": 0.00019907977655700054, "logits/chosen": -0.5835409760475159, "logits/rejected": -0.5857175588607788, "logps/chosen": -8.839982032775879, "logps/rejected": -26.350032806396484, "loss": 0.8248448371887207, "memory(GiB)": 46.35, "nll_loss": 0.4012177586555481, "rewards/accuracies": 0.8125, "rewards/chosen": 0.14620812237262726, "rewards/margins": 1.2952075004577637, "rewards/rejected": -1.1489993333816528, "step": 96, "train_speed(iter/s)": 0.005512 }, { "epoch": 0.09249381685985876, "grad_norm": 3.4665541648864746, "learning_rate": 0.00019903654743804833, "logits/chosen": -0.6227341294288635, "logits/rejected": -0.6252981424331665, "logps/chosen": -11.844514846801758, "logps/rejected": -29.124967575073242, "loss": 0.8466665744781494, "memory(GiB)": 46.35, "nll_loss": 0.3924159109592438, "rewards/accuracies": 0.6875, "rewards/chosen": -0.03820016235113144, "rewards/margins": 1.3871092796325684, "rewards/rejected": -1.425309419631958, "step": 97, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.09344736136356864, "grad_norm": 13.193551063537598, "learning_rate": 0.00019899233102005573, "logits/chosen": -0.5675249695777893, "logits/rejected": -0.571427583694458, "logps/chosen": -11.932504653930664, "logps/rejected": -29.481807708740234, "loss": 1.3480544090270996, "memory(GiB)": 46.35, "nll_loss": 0.8751646280288696, "rewards/accuracies": 0.71875, "rewards/chosen": -0.04504670575261116, "rewards/margins": 1.2038631439208984, "rewards/rejected": -1.248909831047058, "step": 98, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.09440090586727852, "grad_norm": 13.958789825439453, "learning_rate": 0.00019894712774381787, "logits/chosen": -0.5426734685897827, "logits/rejected": -0.5455094575881958, "logps/chosen": -14.97989273071289, "logps/rejected": -29.963254928588867, "loss": 0.9666093587875366, "memory(GiB)": 46.35, "nll_loss": 0.5926772356033325, "rewards/accuracies": 0.875, "rewards/chosen": -0.12036357820034027, "rewards/margins": 1.3575993776321411, "rewards/rejected": -1.4779632091522217, "step": 99, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.09535445037098841, "grad_norm": 3.8765738010406494, "learning_rate": 0.00019890093805996793, "logits/chosen": -0.5699120163917542, "logits/rejected": -0.5716601014137268, "logps/chosen": -18.304710388183594, "logps/rejected": -37.15324783325195, "loss": 1.0460050106048584, "memory(GiB)": 46.35, "nll_loss": 0.6231843829154968, "rewards/accuracies": 0.78125, "rewards/chosen": -0.23249366879463196, "rewards/margins": 1.3045672178268433, "rewards/rejected": -1.5370607376098633, "step": 100, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.0963079948746983, "grad_norm": 3.2104878425598145, "learning_rate": 0.00019885376242897258, "logits/chosen": -0.5706620812416077, "logits/rejected": -0.5722963213920593, "logps/chosen": -14.833114624023438, "logps/rejected": -30.274829864501953, "loss": 1.1376781463623047, "memory(GiB)": 46.35, "nll_loss": 0.653939425945282, "rewards/accuracies": 0.78125, "rewards/chosen": 0.14056257903575897, "rewards/margins": 1.1054493188858032, "rewards/rejected": -0.9648867845535278, "step": 101, "train_speed(iter/s)": 0.005506 }, { "epoch": 0.09726153937840817, "grad_norm": 2.326752185821533, "learning_rate": 0.00019880560132112742, "logits/chosen": -0.5489917993545532, "logits/rejected": -0.5558469891548157, "logps/chosen": -10.081119537353516, "logps/rejected": -31.342273712158203, "loss": 0.9489866495132446, "memory(GiB)": 46.35, "nll_loss": 0.5341838002204895, "rewards/accuracies": 0.78125, "rewards/chosen": 0.08325084298849106, "rewards/margins": 1.2551041841506958, "rewards/rejected": -1.1718534231185913, "step": 102, "train_speed(iter/s)": 0.005507 }, { "epoch": 0.09821508388211805, "grad_norm": 2.702747344970703, "learning_rate": 0.0001987564552165524, "logits/chosen": -0.5335314869880676, "logits/rejected": -0.5375723838806152, "logps/chosen": -6.767879962921143, "logps/rejected": -26.971349716186523, "loss": 0.7666488885879517, "memory(GiB)": 46.35, "nll_loss": 0.3699600100517273, "rewards/accuracies": 0.90625, "rewards/chosen": 0.2037293165922165, "rewards/margins": 1.2131260633468628, "rewards/rejected": -1.0093967914581299, "step": 103, "train_speed(iter/s)": 0.005507 }, { "epoch": 0.09916862838582795, "grad_norm": 3.298539638519287, "learning_rate": 0.0001987063246051868, "logits/chosen": -0.5586405992507935, "logits/rejected": -0.5575414299964905, "logps/chosen": -10.250675201416016, "logps/rejected": -19.98495864868164, "loss": 0.9064714312553406, "memory(GiB)": 46.35, "nll_loss": 0.38871869444847107, "rewards/accuracies": 0.71875, "rewards/chosen": 0.07911191880702972, "rewards/margins": 0.7202980518341064, "rewards/rejected": -0.6411861181259155, "step": 104, "train_speed(iter/s)": 0.005508 }, { "epoch": 0.10012217288953783, "grad_norm": 3.7112255096435547, "learning_rate": 0.00019865520998678458, "logits/chosen": -0.5400758385658264, "logits/rejected": -0.5405173301696777, "logps/chosen": -16.44561004638672, "logps/rejected": -24.691865921020508, "loss": 1.1246851682662964, "memory(GiB)": 46.35, "nll_loss": 0.6655758023262024, "rewards/accuracies": 0.84375, "rewards/chosen": 0.11685307323932648, "rewards/margins": 0.8305923938751221, "rewards/rejected": -0.713739275932312, "step": 105, "train_speed(iter/s)": 0.005507 }, { "epoch": 0.10107571739324771, "grad_norm": 6.40952205657959, "learning_rate": 0.0001986031118709093, "logits/chosen": -0.548525333404541, "logits/rejected": -0.5509157180786133, "logps/chosen": -11.731019020080566, "logps/rejected": -22.46083641052246, "loss": 1.37913179397583, "memory(GiB)": 46.35, "nll_loss": 0.7638791799545288, "rewards/accuracies": 0.78125, "rewards/chosen": -0.029659561812877655, "rewards/margins": 0.5513206124305725, "rewards/rejected": -0.5809801816940308, "step": 106, "train_speed(iter/s)": 0.005505 }, { "epoch": 0.1020292618969576, "grad_norm": 5.338200092315674, "learning_rate": 0.00019855003077692897, "logits/chosen": -0.5491806268692017, "logits/rejected": -0.5485914945602417, "logps/chosen": -10.374083518981934, "logps/rejected": -22.91812515258789, "loss": 0.9197418689727783, "memory(GiB)": 46.35, "nll_loss": 0.42541149258613586, "rewards/accuracies": 0.78125, "rewards/chosen": 0.23835767805576324, "rewards/margins": 0.8718117475509644, "rewards/rejected": -0.6334540843963623, "step": 107, "train_speed(iter/s)": 0.005505 }, { "epoch": 0.10298280640066748, "grad_norm": 4.72066593170166, "learning_rate": 0.00019849596723401107, "logits/chosen": -0.5481430888175964, "logits/rejected": -0.552554190158844, "logps/chosen": -14.082160949707031, "logps/rejected": -25.61027717590332, "loss": 1.2021526098251343, "memory(GiB)": 46.35, "nll_loss": 0.6988036036491394, "rewards/accuracies": 0.8125, "rewards/chosen": 0.09246861189603806, "rewards/margins": 0.8137391805648804, "rewards/rejected": -0.7212705612182617, "step": 108, "train_speed(iter/s)": 0.005507 }, { "epoch": 0.10393635090437736, "grad_norm": 2.7633321285247803, "learning_rate": 0.00019844092178111702, "logits/chosen": -0.508384108543396, "logits/rejected": -0.5140159130096436, "logps/chosen": -6.925086975097656, "logps/rejected": -24.077953338623047, "loss": 0.8022588491439819, "memory(GiB)": 46.35, "nll_loss": 0.41396474838256836, "rewards/accuracies": 0.96875, "rewards/chosen": 0.16561280190944672, "rewards/margins": 1.0861924886703491, "rewards/rejected": -0.9205797910690308, "step": 109, "train_speed(iter/s)": 0.005509 }, { "epoch": 0.10488989540808726, "grad_norm": 2.4014365673065186, "learning_rate": 0.00019838489496699704, "logits/chosen": -0.5266930460929871, "logits/rejected": -0.5241900682449341, "logps/chosen": -15.143444061279297, "logps/rejected": -18.07321548461914, "loss": 0.8957412242889404, "memory(GiB)": 46.35, "nll_loss": 0.5194482803344727, "rewards/accuracies": 0.875, "rewards/chosen": 0.33800825476646423, "rewards/margins": 1.0479741096496582, "rewards/rejected": -0.7099659442901611, "step": 110, "train_speed(iter/s)": 0.005508 }, { "epoch": 0.10584343991179714, "grad_norm": 3.5473034381866455, "learning_rate": 0.0001983278873501845, "logits/chosen": -0.5651346445083618, "logits/rejected": -0.5652989149093628, "logps/chosen": -13.63066291809082, "logps/rejected": -21.72479820251465, "loss": 1.070034146308899, "memory(GiB)": 46.35, "nll_loss": 0.5245512127876282, "rewards/accuracies": 0.6875, "rewards/chosen": 0.1012478768825531, "rewards/margins": 0.5533490180969238, "rewards/rejected": -0.4521011710166931, "step": 111, "train_speed(iter/s)": 0.005508 }, { "epoch": 0.10679698441550702, "grad_norm": 2.844266653060913, "learning_rate": 0.00019826989949899048, "logits/chosen": -0.5871210694313049, "logits/rejected": -0.5863486528396606, "logps/chosen": -10.202323913574219, "logps/rejected": -17.968069076538086, "loss": 0.9712742567062378, "memory(GiB)": 46.35, "nll_loss": 0.4458525776863098, "rewards/accuracies": 0.65625, "rewards/chosen": 0.31141582131385803, "rewards/margins": 0.6074751615524292, "rewards/rejected": -0.29605934023857117, "step": 112, "train_speed(iter/s)": 0.005507 }, { "epoch": 0.1077505289192169, "grad_norm": 3.325552225112915, "learning_rate": 0.00019821093199149804, "logits/chosen": -0.5867654085159302, "logits/rejected": -0.590225338935852, "logps/chosen": -14.67109489440918, "logps/rejected": -31.97191619873047, "loss": 0.8784574270248413, "memory(GiB)": 46.35, "nll_loss": 0.5371198058128357, "rewards/accuracies": 0.90625, "rewards/chosen": 0.31997063755989075, "rewards/margins": 1.1458133459091187, "rewards/rejected": -0.8258427381515503, "step": 113, "train_speed(iter/s)": 0.005509 }, { "epoch": 0.10870407342292679, "grad_norm": 2.5996289253234863, "learning_rate": 0.00019815098541555646, "logits/chosen": -0.6402957439422607, "logits/rejected": -0.6378954648971558, "logps/chosen": -12.175540924072266, "logps/rejected": -20.870878219604492, "loss": 0.9693875908851624, "memory(GiB)": 46.35, "nll_loss": 0.5651832818984985, "rewards/accuracies": 0.875, "rewards/chosen": 0.3929046392440796, "rewards/margins": 1.0991772413253784, "rewards/rejected": -0.7062726020812988, "step": 114, "train_speed(iter/s)": 0.005509 }, { "epoch": 0.10965761792663667, "grad_norm": 2.5734198093414307, "learning_rate": 0.00019809006036877538, "logits/chosen": -0.6097349524497986, "logits/rejected": -0.6098358035087585, "logps/chosen": -11.490409851074219, "logps/rejected": -15.817056655883789, "loss": 0.9781900644302368, "memory(GiB)": 46.35, "nll_loss": 0.502380907535553, "rewards/accuracies": 0.8125, "rewards/chosen": 0.2869223654270172, "rewards/margins": 0.795809268951416, "rewards/rejected": -0.5088868737220764, "step": 115, "train_speed(iter/s)": 0.00551 }, { "epoch": 0.11061116243034655, "grad_norm": 2.6200523376464844, "learning_rate": 0.00019802815745851885, "logits/chosen": -0.6539290547370911, "logits/rejected": -0.6565437912940979, "logps/chosen": -13.780235290527344, "logps/rejected": -29.117767333984375, "loss": 0.8663976192474365, "memory(GiB)": 46.35, "nll_loss": 0.406424343585968, "rewards/accuracies": 0.71875, "rewards/chosen": 0.24194149672985077, "rewards/margins": 1.0130045413970947, "rewards/rejected": -0.7710630893707275, "step": 116, "train_speed(iter/s)": 0.005509 }, { "epoch": 0.11156470693405644, "grad_norm": 2.858734130859375, "learning_rate": 0.00019796527730189936, "logits/chosen": -0.6368309259414673, "logits/rejected": -0.6411804556846619, "logps/chosen": -12.20506477355957, "logps/rejected": -22.50167465209961, "loss": 0.9522262811660767, "memory(GiB)": 46.35, "nll_loss": 0.5067539215087891, "rewards/accuracies": 0.8125, "rewards/chosen": 0.35891181230545044, "rewards/margins": 0.9037349224090576, "rewards/rejected": -0.5448231101036072, "step": 117, "train_speed(iter/s)": 0.005508 }, { "epoch": 0.11251825143776632, "grad_norm": 3.5055670738220215, "learning_rate": 0.0001979014205257715, "logits/chosen": -0.6525743007659912, "logits/rejected": -0.6540606021881104, "logps/chosen": -15.675086975097656, "logps/rejected": -24.1904239654541, "loss": 1.0509792566299438, "memory(GiB)": 46.35, "nll_loss": 0.5997871160507202, "rewards/accuracies": 0.8125, "rewards/chosen": 0.07162542641162872, "rewards/margins": 1.1154685020446777, "rewards/rejected": -1.0438430309295654, "step": 118, "train_speed(iter/s)": 0.005507 }, { "epoch": 0.1134717959414762, "grad_norm": 3.984374761581421, "learning_rate": 0.0001978365877667258, "logits/chosen": -0.6372765302658081, "logits/rejected": -0.6411250233650208, "logps/chosen": -10.17730712890625, "logps/rejected": -22.074499130249023, "loss": 1.0653222799301147, "memory(GiB)": 46.35, "nll_loss": 0.6034587025642395, "rewards/accuracies": 0.78125, "rewards/chosen": 0.198074609041214, "rewards/margins": 1.0864827632904053, "rewards/rejected": -0.8884081244468689, "step": 119, "train_speed(iter/s)": 0.005508 }, { "epoch": 0.1144253404451861, "grad_norm": 3.182776689529419, "learning_rate": 0.00019777077967108255, "logits/chosen": -0.6417007446289062, "logits/rejected": -0.6392112970352173, "logps/chosen": -19.301258087158203, "logps/rejected": -18.364166259765625, "loss": 1.108437180519104, "memory(GiB)": 46.35, "nll_loss": 0.5851361751556396, "rewards/accuracies": 0.71875, "rewards/chosen": 0.22438707947731018, "rewards/margins": 0.5671754479408264, "rewards/rejected": -0.3427883982658386, "step": 120, "train_speed(iter/s)": 0.005508 }, { "epoch": 0.11537888494889598, "grad_norm": 3.537256956100464, "learning_rate": 0.00019770399689488506, "logits/chosen": -0.6460685729980469, "logits/rejected": -0.6467642784118652, "logps/chosen": -13.258935928344727, "logps/rejected": -27.513343811035156, "loss": 0.9472922682762146, "memory(GiB)": 46.35, "nll_loss": 0.5081149935722351, "rewards/accuracies": 0.75, "rewards/chosen": 0.23222729563713074, "rewards/margins": 1.276731014251709, "rewards/rejected": -1.0445036888122559, "step": 121, "train_speed(iter/s)": 0.005508 }, { "epoch": 0.11633242945260586, "grad_norm": 3.096590280532837, "learning_rate": 0.00019763624010389334, "logits/chosen": -0.6088275909423828, "logits/rejected": -0.6121100783348083, "logps/chosen": -13.26574420928955, "logps/rejected": -19.00273323059082, "loss": 0.9677691459655762, "memory(GiB)": 46.35, "nll_loss": 0.507895290851593, "rewards/accuracies": 0.8125, "rewards/chosen": 0.22311073541641235, "rewards/margins": 0.7144712209701538, "rewards/rejected": -0.49136048555374146, "step": 122, "train_speed(iter/s)": 0.005509 }, { "epoch": 0.11728597395631574, "grad_norm": 2.810216188430786, "learning_rate": 0.0001975675099735774, "logits/chosen": -0.6021947264671326, "logits/rejected": -0.6030389070510864, "logps/chosen": -8.321534156799316, "logps/rejected": -17.207504272460938, "loss": 0.8276495933532715, "memory(GiB)": 46.35, "nll_loss": 0.3906245231628418, "rewards/accuracies": 0.8125, "rewards/chosen": 0.21257750689983368, "rewards/margins": 0.894737958908081, "rewards/rejected": -0.682160496711731, "step": 123, "train_speed(iter/s)": 0.005509 }, { "epoch": 0.11823951846002563, "grad_norm": 2.6153383255004883, "learning_rate": 0.0001974978071891104, "logits/chosen": -0.6311705112457275, "logits/rejected": -0.6307083964347839, "logps/chosen": -10.833940505981445, "logps/rejected": -22.41739273071289, "loss": 0.8693855404853821, "memory(GiB)": 46.35, "nll_loss": 0.4562944173812866, "rewards/accuracies": 0.875, "rewards/chosen": 0.15080919861793518, "rewards/margins": 1.2266969680786133, "rewards/rejected": -1.0758877992630005, "step": 124, "train_speed(iter/s)": 0.00551 }, { "epoch": 0.11919306296373551, "grad_norm": 3.493828535079956, "learning_rate": 0.00019742713244536204, "logits/chosen": -0.6434034109115601, "logits/rejected": -0.6431565284729004, "logps/chosen": -18.75763702392578, "logps/rejected": -25.821609497070312, "loss": 1.0967546701431274, "memory(GiB)": 46.35, "nll_loss": 0.5587143898010254, "rewards/accuracies": 0.8125, "rewards/chosen": 0.0902448296546936, "rewards/margins": 0.9879235029220581, "rewards/rejected": -0.8976786732673645, "step": 125, "train_speed(iter/s)": 0.00551 }, { "epoch": 0.12014660746744539, "grad_norm": 2.5747878551483154, "learning_rate": 0.00019735548644689147, "logits/chosen": -0.6225179433822632, "logits/rejected": -0.6244460940361023, "logps/chosen": -15.244443893432617, "logps/rejected": -17.300045013427734, "loss": 1.0364809036254883, "memory(GiB)": 46.35, "nll_loss": 0.5735288858413696, "rewards/accuracies": 0.78125, "rewards/chosen": 0.416117787361145, "rewards/margins": 0.7515360713005066, "rewards/rejected": -0.3354182839393616, "step": 126, "train_speed(iter/s)": 0.00551 }, { "epoch": 0.12110015197115528, "grad_norm": 3.198167562484741, "learning_rate": 0.00019728286990794037, "logits/chosen": -0.6006501317024231, "logits/rejected": -0.6046801805496216, "logps/chosen": -14.4312162399292, "logps/rejected": -23.69890785217285, "loss": 1.0553488731384277, "memory(GiB)": 46.35, "nll_loss": 0.5887905955314636, "rewards/accuracies": 0.84375, "rewards/chosen": 0.162513867020607, "rewards/margins": 0.9815673232078552, "rewards/rejected": -0.8190534114837646, "step": 127, "train_speed(iter/s)": 0.00551 }, { "epoch": 0.12205369647486516, "grad_norm": 3.224283456802368, "learning_rate": 0.00019720928355242568, "logits/chosen": -0.5655272006988525, "logits/rejected": -0.5726292133331299, "logps/chosen": -13.809501647949219, "logps/rejected": -30.130277633666992, "loss": 0.8311694264411926, "memory(GiB)": 46.35, "nll_loss": 0.3935176134109497, "rewards/accuracies": 0.84375, "rewards/chosen": 0.027349811047315598, "rewards/margins": 1.382154941558838, "rewards/rejected": -1.3548052310943604, "step": 128, "train_speed(iter/s)": 0.00551 }, { "epoch": 0.12300724097857504, "grad_norm": 2.575568199157715, "learning_rate": 0.00019713472811393258, "logits/chosen": -0.5765466094017029, "logits/rejected": -0.5786842107772827, "logps/chosen": -16.621131896972656, "logps/rejected": -26.19173240661621, "loss": 0.9337475895881653, "memory(GiB)": 46.35, "nll_loss": 0.5804612636566162, "rewards/accuracies": 0.875, "rewards/chosen": 0.37209561467170715, "rewards/margins": 1.4012634754180908, "rewards/rejected": -1.0291677713394165, "step": 129, "train_speed(iter/s)": 0.005512 }, { "epoch": 0.12396078548228492, "grad_norm": 3.0303828716278076, "learning_rate": 0.00019705920433570694, "logits/chosen": -0.5693878531455994, "logits/rejected": -0.5738221406936646, "logps/chosen": -7.376468658447266, "logps/rejected": -25.63204574584961, "loss": 0.9063543081283569, "memory(GiB)": 46.35, "nll_loss": 0.5075121521949768, "rewards/accuracies": 0.875, "rewards/chosen": 0.23270781338214874, "rewards/margins": 1.1151927709579468, "rewards/rejected": -0.8824849128723145, "step": 130, "train_speed(iter/s)": 0.005511 }, { "epoch": 0.12491432998599482, "grad_norm": 4.375239849090576, "learning_rate": 0.0001969827129706482, "logits/chosen": -0.5413441061973572, "logits/rejected": -0.5445973873138428, "logps/chosen": -9.677428245544434, "logps/rejected": -23.95619010925293, "loss": 0.982276439666748, "memory(GiB)": 46.35, "nll_loss": 0.5150867104530334, "rewards/accuracies": 0.71875, "rewards/chosen": 0.23964405059814453, "rewards/margins": 0.8764944076538086, "rewards/rejected": -0.6368504762649536, "step": 131, "train_speed(iter/s)": 0.005511 }, { "epoch": 0.1258678744897047, "grad_norm": 2.136566162109375, "learning_rate": 0.0001969052547813016, "logits/chosen": -0.5252913236618042, "logits/rejected": -0.5288749933242798, "logps/chosen": -8.76069450378418, "logps/rejected": -24.643917083740234, "loss": 0.6882707476615906, "memory(GiB)": 46.35, "nll_loss": 0.3399735689163208, "rewards/accuracies": 0.84375, "rewards/chosen": 0.3247303366661072, "rewards/margins": 1.320786952972412, "rewards/rejected": -0.9960566759109497, "step": 132, "train_speed(iter/s)": 0.005512 }, { "epoch": 0.1268214189934146, "grad_norm": 2.7579689025878906, "learning_rate": 0.00019682683053985072, "logits/chosen": -0.4897473454475403, "logits/rejected": -0.49097952246665955, "logps/chosen": -14.222724914550781, "logps/rejected": -26.548240661621094, "loss": 0.8395786881446838, "memory(GiB)": 46.35, "nll_loss": 0.49962103366851807, "rewards/accuracies": 0.875, "rewards/chosen": 0.3120686113834381, "rewards/margins": 1.3334513902664185, "rewards/rejected": -1.0213829278945923, "step": 133, "train_speed(iter/s)": 0.005511 }, { "epoch": 0.12777496349712447, "grad_norm": 4.233667850494385, "learning_rate": 0.00019674744102810978, "logits/chosen": -0.49640408158302307, "logits/rejected": -0.49791407585144043, "logps/chosen": -11.952919006347656, "logps/rejected": -20.992504119873047, "loss": 0.9811766147613525, "memory(GiB)": 46.35, "nll_loss": 0.6130494475364685, "rewards/accuracies": 0.90625, "rewards/chosen": 0.26505932211875916, "rewards/margins": 1.113063097000122, "rewards/rejected": -0.8480038046836853, "step": 134, "train_speed(iter/s)": 0.005511 }, { "epoch": 0.12872850800083435, "grad_norm": 4.032670974731445, "learning_rate": 0.00019666708703751576, "logits/chosen": -0.5555613040924072, "logits/rejected": -0.5562028288841248, "logps/chosen": -12.461511611938477, "logps/rejected": -29.6381893157959, "loss": 0.9406546354293823, "memory(GiB)": 46.35, "nll_loss": 0.5542497634887695, "rewards/accuracies": 0.8125, "rewards/chosen": 0.14334315061569214, "rewards/margins": 1.2391164302825928, "rewards/rejected": -1.0957733392715454, "step": 135, "train_speed(iter/s)": 0.005511 }, { "epoch": 0.12968205250454423, "grad_norm": 2.8314244747161865, "learning_rate": 0.00019658576936912057, "logits/chosen": -0.5166746973991394, "logits/rejected": -0.5175175666809082, "logps/chosen": -11.453944206237793, "logps/rejected": -32.48845672607422, "loss": 0.741032600402832, "memory(GiB)": 46.35, "nll_loss": 0.39420661330223083, "rewards/accuracies": 0.84375, "rewards/chosen": 0.2297428399324417, "rewards/margins": 1.5928231477737427, "rewards/rejected": -1.3630802631378174, "step": 136, "train_speed(iter/s)": 0.005511 }, { "epoch": 0.1306355970082541, "grad_norm": 3.786283016204834, "learning_rate": 0.00019650348883358315, "logits/chosen": -0.5285561084747314, "logits/rejected": -0.529099702835083, "logps/chosen": -14.34618091583252, "logps/rejected": -24.930580139160156, "loss": 1.0361278057098389, "memory(GiB)": 46.35, "nll_loss": 0.598253607749939, "rewards/accuracies": 0.8125, "rewards/chosen": 0.2112821340560913, "rewards/margins": 1.2288541793823242, "rewards/rejected": -1.017572045326233, "step": 137, "train_speed(iter/s)": 0.005512 }, { "epoch": 0.131589141511964, "grad_norm": 3.4468750953674316, "learning_rate": 0.00019642024625116117, "logits/chosen": -0.5355381369590759, "logits/rejected": -0.5351199507713318, "logps/chosen": -14.2181396484375, "logps/rejected": -21.662395477294922, "loss": 0.9722803831100464, "memory(GiB)": 46.35, "nll_loss": 0.5212066769599915, "rewards/accuracies": 0.8125, "rewards/chosen": 0.12246156483888626, "rewards/margins": 1.046801209449768, "rewards/rejected": -0.9243396520614624, "step": 138, "train_speed(iter/s)": 0.005512 }, { "epoch": 0.1325426860156739, "grad_norm": 4.629439830780029, "learning_rate": 0.0001963360424517031, "logits/chosen": -0.5707206726074219, "logits/rejected": -0.5748513340950012, "logps/chosen": -17.805238723754883, "logps/rejected": -29.64284896850586, "loss": 1.254711389541626, "memory(GiB)": 46.35, "nll_loss": 0.743948221206665, "rewards/accuracies": 0.78125, "rewards/chosen": 0.11810468882322311, "rewards/margins": 0.9775583744049072, "rewards/rejected": -0.8594536185264587, "step": 139, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.13349623051938378, "grad_norm": 4.580717086791992, "learning_rate": 0.0001962508782746397, "logits/chosen": -0.580274224281311, "logits/rejected": -0.581068217754364, "logps/chosen": -16.117170333862305, "logps/rejected": -23.330665588378906, "loss": 0.9780559539794922, "memory(GiB)": 46.35, "nll_loss": 0.48741835355758667, "rewards/accuracies": 0.8125, "rewards/chosen": 0.07310809940099716, "rewards/margins": 0.9340574741363525, "rewards/rejected": -0.8609493374824524, "step": 140, "train_speed(iter/s)": 0.005514 }, { "epoch": 0.13444977502309366, "grad_norm": 3.511350631713867, "learning_rate": 0.0001961647545689759, "logits/chosen": -0.5712682604789734, "logits/rejected": -0.5709258317947388, "logps/chosen": -14.216093063354492, "logps/rejected": -23.430875778198242, "loss": 1.023219108581543, "memory(GiB)": 46.35, "nll_loss": 0.5690780878067017, "rewards/accuracies": 0.84375, "rewards/chosen": 0.3575439751148224, "rewards/margins": 0.9377648830413818, "rewards/rejected": -0.5802209973335266, "step": 141, "train_speed(iter/s)": 0.005514 }, { "epoch": 0.13540331952680354, "grad_norm": 2.3146886825561523, "learning_rate": 0.0001960776721932821, "logits/chosen": -0.5722199082374573, "logits/rejected": -0.5763861536979675, "logps/chosen": -7.839071273803711, "logps/rejected": -18.26845932006836, "loss": 0.810951828956604, "memory(GiB)": 46.35, "nll_loss": 0.3562580943107605, "rewards/accuracies": 0.8125, "rewards/chosen": 0.24190367758274078, "rewards/margins": 0.711327850818634, "rewards/rejected": -0.4694240987300873, "step": 142, "train_speed(iter/s)": 0.005514 }, { "epoch": 0.13635686403051342, "grad_norm": 3.1744766235351562, "learning_rate": 0.00019598963201568573, "logits/chosen": -0.5400611758232117, "logits/rejected": -0.5411737561225891, "logps/chosen": -13.320467948913574, "logps/rejected": -24.537696838378906, "loss": 1.0167641639709473, "memory(GiB)": 46.35, "nll_loss": 0.47375380992889404, "rewards/accuracies": 0.6875, "rewards/chosen": 0.18297314643859863, "rewards/margins": 0.6672495007514954, "rewards/rejected": -0.48427632451057434, "step": 143, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.1373104085342233, "grad_norm": 2.385725259780884, "learning_rate": 0.0001959006349138626, "logits/chosen": -0.5638188123703003, "logits/rejected": -0.5665206909179688, "logps/chosen": -12.801736831665039, "logps/rejected": -24.967641830444336, "loss": 0.9001826047897339, "memory(GiB)": 46.35, "nll_loss": 0.45819178223609924, "rewards/accuracies": 0.75, "rewards/chosen": 0.24550366401672363, "rewards/margins": 0.9985946416854858, "rewards/rejected": -0.7530909776687622, "step": 144, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.13826395303793318, "grad_norm": 4.4898362159729, "learning_rate": 0.00019581068177502826, "logits/chosen": -0.5841139554977417, "logits/rejected": -0.5885876417160034, "logps/chosen": -8.781448364257812, "logps/rejected": -21.54075813293457, "loss": 0.9311922192573547, "memory(GiB)": 46.35, "nll_loss": 0.46065330505371094, "rewards/accuracies": 0.78125, "rewards/chosen": 0.2900657057762146, "rewards/margins": 0.7273497581481934, "rewards/rejected": -0.43728405237197876, "step": 145, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.1392174975416431, "grad_norm": 2.794843912124634, "learning_rate": 0.00019571977349592878, "logits/chosen": -0.5932512879371643, "logits/rejected": -0.5940472483634949, "logps/chosen": -9.732755661010742, "logps/rejected": -20.826379776000977, "loss": 0.945945680141449, "memory(GiB)": 46.35, "nll_loss": 0.47954535484313965, "rewards/accuracies": 0.78125, "rewards/chosen": 0.2869046926498413, "rewards/margins": 0.9880754947662354, "rewards/rejected": -0.7011707425117493, "step": 146, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.14017104204535297, "grad_norm": 2.540703058242798, "learning_rate": 0.00019562791098283225, "logits/chosen": -0.6081781983375549, "logits/rejected": -0.614071786403656, "logps/chosen": -12.527294158935547, "logps/rejected": -29.05521011352539, "loss": 0.9200064539909363, "memory(GiB)": 46.35, "nll_loss": 0.4640219807624817, "rewards/accuracies": 0.78125, "rewards/chosen": 0.25594115257263184, "rewards/margins": 0.8903652429580688, "rewards/rejected": -0.634424090385437, "step": 147, "train_speed(iter/s)": 0.005512 }, { "epoch": 0.14112458654906285, "grad_norm": 2.9323384761810303, "learning_rate": 0.0001955350951515195, "logits/chosen": -0.6128370761871338, "logits/rejected": -0.6138598322868347, "logps/chosen": -12.749382019042969, "logps/rejected": -25.719005584716797, "loss": 0.9213395118713379, "memory(GiB)": 46.35, "nll_loss": 0.43978482484817505, "rewards/accuracies": 0.6875, "rewards/chosen": 0.32896676659584045, "rewards/margins": 1.2176496982574463, "rewards/rejected": -0.8886829018592834, "step": 148, "train_speed(iter/s)": 0.005513 }, { "epoch": 0.14207813105277273, "grad_norm": 4.304955959320068, "learning_rate": 0.00019544132692727497, "logits/chosen": -0.6183297634124756, "logits/rejected": -0.6181198358535767, "logps/chosen": -13.527120590209961, "logps/rejected": -16.62553596496582, "loss": 1.1245884895324707, "memory(GiB)": 46.35, "nll_loss": 0.5154122114181519, "rewards/accuracies": 0.65625, "rewards/chosen": 0.12139306217432022, "rewards/margins": 0.5662888884544373, "rewards/rejected": -0.44489583373069763, "step": 149, "train_speed(iter/s)": 0.005512 }, { "epoch": 0.1430316755564826, "grad_norm": 3.4540951251983643, "learning_rate": 0.00019534660724487764, "logits/chosen": -0.6565526127815247, "logits/rejected": -0.6572035551071167, "logps/chosen": -8.482949256896973, "logps/rejected": -25.549943923950195, "loss": 0.6943767070770264, "memory(GiB)": 46.35, "nll_loss": 0.3159071207046509, "rewards/accuracies": 0.75, "rewards/chosen": 0.13372917473316193, "rewards/margins": 1.538865327835083, "rewards/rejected": -1.4051361083984375, "step": 150, "train_speed(iter/s)": 0.005513 } ], "logging_steps": 1, "max_steps": 1048, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 50, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 1.0844345149874504e+18, "train_batch_size": 1, "trial_name": null, "trial_params": null }