{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.9987943737441393, "eval_steps": 400, "global_step": 466, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.010716677829872739, "grad_norm": 8.776682238147446, "learning_rate": 5.3191489361702123e-08, "logits/chosen": -2.8905415534973145, "logits/rejected": -2.8877615928649902, "logps/chosen": -0.9742914438247681, "logps/rejected": -0.9831219911575317, "loss": 0.0001, "rewards/accuracies": 0.4749999940395355, "rewards/chosen": -0.9742914438247681, "rewards/margins": 0.008830604143440723, "rewards/rejected": -0.9831219911575317, "step": 5 }, { "epoch": 0.021433355659745478, "grad_norm": 9.810248541108757, "learning_rate": 1.0638297872340425e-07, "logits/chosen": -2.9070327281951904, "logits/rejected": -2.8929152488708496, "logps/chosen": -0.978168785572052, "logps/rejected": -1.001371145248413, "loss": -0.0, "rewards/accuracies": 0.512499988079071, "rewards/chosen": -0.978168785572052, "rewards/margins": 0.023202406242489815, "rewards/rejected": -1.001371145248413, "step": 10 }, { "epoch": 0.032150033489618215, "grad_norm": 7.0995253496741135, "learning_rate": 1.5957446808510638e-07, "logits/chosen": -2.895216703414917, "logits/rejected": -2.9180421829223633, "logps/chosen": -0.9596306085586548, "logps/rejected": -0.9847423434257507, "loss": -0.0001, "rewards/accuracies": 0.518750011920929, "rewards/chosen": -0.9596306085586548, "rewards/margins": 0.02511175535619259, "rewards/rejected": -0.9847423434257507, "step": 15 }, { "epoch": 0.042866711319490956, "grad_norm": 7.785343608601427, "learning_rate": 2.127659574468085e-07, "logits/chosen": -2.902580976486206, "logits/rejected": -2.9009933471679688, "logps/chosen": -0.9715825319290161, "logps/rejected": -0.9759886860847473, "loss": -0.0, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": -0.9715825319290161, "rewards/margins": 0.00440616998821497, "rewards/rejected": -0.9759886860847473, "step": 20 }, { "epoch": 0.0535833891493637, "grad_norm": 6.677386701450202, "learning_rate": 2.659574468085106e-07, "logits/chosen": -2.8314852714538574, "logits/rejected": -2.8363277912139893, "logps/chosen": -0.9519669413566589, "logps/rejected": -1.0002533197402954, "loss": -0.0003, "rewards/accuracies": 0.5562499761581421, "rewards/chosen": -0.9519669413566589, "rewards/margins": 0.048286281526088715, "rewards/rejected": -1.0002533197402954, "step": 25 }, { "epoch": 0.06430006697923643, "grad_norm": 8.4552870442767, "learning_rate": 3.1914893617021275e-07, "logits/chosen": -2.895077705383301, "logits/rejected": -2.9221832752227783, "logps/chosen": -0.983117938041687, "logps/rejected": -0.9829673767089844, "loss": -0.0002, "rewards/accuracies": 0.46875, "rewards/chosen": -0.983117938041687, "rewards/margins": -0.00015044640167616308, "rewards/rejected": -0.9829673767089844, "step": 30 }, { "epoch": 0.07501674480910918, "grad_norm": 7.60248558460589, "learning_rate": 3.7234042553191484e-07, "logits/chosen": -2.9559242725372314, "logits/rejected": -2.9249794483184814, "logps/chosen": -0.95380699634552, "logps/rejected": -0.9984177350997925, "loss": -0.0004, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -0.95380699634552, "rewards/margins": 0.04461073502898216, "rewards/rejected": -0.9984177350997925, "step": 35 }, { "epoch": 0.08573342263898191, "grad_norm": 8.131750300550957, "learning_rate": 4.25531914893617e-07, "logits/chosen": -2.8956754207611084, "logits/rejected": -2.8939049243927, "logps/chosen": -0.9987373352050781, "logps/rejected": -0.9930660128593445, "loss": -0.0015, "rewards/accuracies": 0.48124998807907104, "rewards/chosen": -0.9987373352050781, "rewards/margins": -0.005671312101185322, "rewards/rejected": -0.9930660128593445, "step": 40 }, { "epoch": 0.09645010046885466, "grad_norm": 7.488747662199227, "learning_rate": 4.787234042553192e-07, "logits/chosen": -2.8674330711364746, "logits/rejected": -2.856853485107422, "logps/chosen": -0.9764472246170044, "logps/rejected": -0.9902997016906738, "loss": -0.001, "rewards/accuracies": 0.512499988079071, "rewards/chosen": -0.9764472246170044, "rewards/margins": 0.013852563686668873, "rewards/rejected": -0.9902997016906738, "step": 45 }, { "epoch": 0.1071667782987274, "grad_norm": 7.447612263721868, "learning_rate": 4.999367579541646e-07, "logits/chosen": -2.875795364379883, "logits/rejected": -2.8871512413024902, "logps/chosen": -0.9369581937789917, "logps/rejected": -0.9167054891586304, "loss": -0.0027, "rewards/accuracies": 0.4937500059604645, "rewards/chosen": -0.9369581937789917, "rewards/margins": -0.020252685993909836, "rewards/rejected": -0.9167054891586304, "step": 50 }, { "epoch": 0.11788345612860013, "grad_norm": 8.432358893020524, "learning_rate": 4.99550394651012e-07, "logits/chosen": -2.915107011795044, "logits/rejected": -2.925875186920166, "logps/chosen": -0.9737237095832825, "logps/rejected": -1.0020536184310913, "loss": -0.0022, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": -0.9737237095832825, "rewards/margins": 0.028329964727163315, "rewards/rejected": -1.0020536184310913, "step": 55 }, { "epoch": 0.12860013395847286, "grad_norm": 9.97321343058875, "learning_rate": 4.988133448023071e-07, "logits/chosen": -2.8224921226501465, "logits/rejected": -2.8447964191436768, "logps/chosen": -0.9149552583694458, "logps/rejected": -0.9350795745849609, "loss": -0.0041, "rewards/accuracies": 0.5, "rewards/chosen": -0.9149552583694458, "rewards/margins": 0.02012430503964424, "rewards/rejected": -0.9350795745849609, "step": 60 }, { "epoch": 0.13931681178834562, "grad_norm": 8.412802955896868, "learning_rate": 4.977266441646379e-07, "logits/chosen": -2.900336503982544, "logits/rejected": -2.9115452766418457, "logps/chosen": -0.8882263898849487, "logps/rejected": -0.9012020230293274, "loss": -0.0057, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": -0.8882263898849487, "rewards/margins": 0.012975621037185192, "rewards/rejected": -0.9012020230293274, "step": 65 }, { "epoch": 0.15003348961821836, "grad_norm": 8.21023142941744, "learning_rate": 4.962918198495654e-07, "logits/chosen": -2.816331386566162, "logits/rejected": -2.839690685272217, "logps/chosen": -0.9164741635322571, "logps/rejected": -0.9583846926689148, "loss": -0.007, "rewards/accuracies": 0.550000011920929, "rewards/chosen": -0.9164741635322571, "rewards/margins": 0.04191048443317413, "rewards/rejected": -0.9583846926689148, "step": 70 }, { "epoch": 0.1607501674480911, "grad_norm": 8.354728288045035, "learning_rate": 4.945108881776135e-07, "logits/chosen": -2.8184382915496826, "logits/rejected": -2.81858229637146, "logps/chosen": -0.8914631009101868, "logps/rejected": -0.9294856190681458, "loss": -0.0066, "rewards/accuracies": 0.5562499761581421, "rewards/chosen": -0.8914631009101868, "rewards/margins": 0.038022562861442566, "rewards/rejected": -0.9294856190681458, "step": 75 }, { "epoch": 0.17146684527796383, "grad_norm": 9.546162491226012, "learning_rate": 4.923863518447879e-07, "logits/chosen": -2.840766191482544, "logits/rejected": -2.8635830879211426, "logps/chosen": -0.887871265411377, "logps/rejected": -0.958912193775177, "loss": -0.006, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -0.887871265411377, "rewards/margins": 0.07104082405567169, "rewards/rejected": -0.958912193775177, "step": 80 }, { "epoch": 0.18218352310783656, "grad_norm": 9.354318155678731, "learning_rate": 4.899211964056035e-07, "logits/chosen": -2.854689359664917, "logits/rejected": -2.866008758544922, "logps/chosen": -0.9082525968551636, "logps/rejected": -0.9574602842330933, "loss": -0.0132, "rewards/accuracies": 0.59375, "rewards/chosen": -0.9082525968551636, "rewards/margins": 0.04920773208141327, "rewards/rejected": -0.9574602842330933, "step": 85 }, { "epoch": 0.19290020093770932, "grad_norm": 9.38237658589797, "learning_rate": 4.871188860775643e-07, "logits/chosen": -2.830294132232666, "logits/rejected": -2.8332040309906006, "logps/chosen": -0.9233630895614624, "logps/rejected": -0.9633795022964478, "loss": -0.0142, "rewards/accuracies": 0.53125, "rewards/chosen": -0.9233630895614624, "rewards/margins": 0.040016382932662964, "rewards/rejected": -0.9633795022964478, "step": 90 }, { "epoch": 0.20361687876758205, "grad_norm": 8.979685614931006, "learning_rate": 4.839833588729897e-07, "logits/chosen": -2.8387644290924072, "logits/rejected": -2.8495821952819824, "logps/chosen": -0.9509484171867371, "logps/rejected": -0.974162220954895, "loss": -0.0085, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": -0.9509484171867371, "rewards/margins": 0.02321377769112587, "rewards/rejected": -0.974162220954895, "step": 95 }, { "epoch": 0.2143335565974548, "grad_norm": 9.427728619757332, "learning_rate": 4.80519021065031e-07, "logits/chosen": -2.860654354095459, "logits/rejected": -2.8680408000946045, "logps/chosen": -0.9869228601455688, "logps/rejected": -1.0385963916778564, "loss": -0.015, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -0.9869228601455688, "rewards/margins": 0.05167357996106148, "rewards/rejected": -1.0385963916778564, "step": 100 }, { "epoch": 0.22505023442732752, "grad_norm": 8.191485445947501, "learning_rate": 4.767307409956528e-07, "logits/chosen": -2.859048843383789, "logits/rejected": -2.8625106811523438, "logps/chosen": -0.8909996151924133, "logps/rejected": -0.9282458424568176, "loss": -0.0154, "rewards/accuracies": 0.5687500238418579, "rewards/chosen": -0.8909996151924133, "rewards/margins": 0.03724620118737221, "rewards/rejected": -0.9282458424568176, "step": 105 }, { "epoch": 0.23576691225720026, "grad_norm": 9.657244741544229, "learning_rate": 4.726238422342805e-07, "logits/chosen": -2.8523192405700684, "logits/rejected": -2.84688138961792, "logps/chosen": -0.9256642460823059, "logps/rejected": -0.9944097399711609, "loss": -0.0136, "rewards/accuracies": 0.606249988079071, "rewards/chosen": -0.9256642460823059, "rewards/margins": 0.06874561309814453, "rewards/rejected": -0.9944097399711609, "step": 110 }, { "epoch": 0.24648359008707302, "grad_norm": 11.45037121550009, "learning_rate": 4.6820409609673027e-07, "logits/chosen": -2.895726203918457, "logits/rejected": -2.900205612182617, "logps/chosen": -0.9234002828598022, "logps/rejected": -0.938769519329071, "loss": -0.0216, "rewards/accuracies": 0.518750011920929, "rewards/chosen": -0.9234002828598022, "rewards/margins": 0.015369323082268238, "rewards/rejected": -0.938769519329071, "step": 115 }, { "epoch": 0.2572002679169457, "grad_norm": 9.77719736840442, "learning_rate": 4.6347771353493173e-07, "logits/chosen": -2.9136910438537598, "logits/rejected": -2.930593252182007, "logps/chosen": -0.9225813150405884, "logps/rejected": -0.917222797870636, "loss": -0.0168, "rewards/accuracies": 0.5, "rewards/chosen": -0.9225813150405884, "rewards/margins": -0.005358507391065359, "rewards/rejected": -0.917222797870636, "step": 120 }, { "epoch": 0.2679169457468185, "grad_norm": 10.417139695198527, "learning_rate": 4.5845133640884217e-07, "logits/chosen": -2.898656129837036, "logits/rejected": -2.8996481895446777, "logps/chosen": -0.9208585619926453, "logps/rejected": -0.9479470252990723, "loss": -0.0193, "rewards/accuracies": 0.543749988079071, "rewards/chosen": -0.9208585619926453, "rewards/margins": 0.0270883496850729, "rewards/rejected": -0.9479470252990723, "step": 125 }, { "epoch": 0.27863362357669125, "grad_norm": 11.697763554027677, "learning_rate": 4.531320281528169e-07, "logits/chosen": -2.9303739070892334, "logits/rejected": -2.9241251945495605, "logps/chosen": -0.9110285639762878, "logps/rejected": -0.9726667404174805, "loss": -0.0227, "rewards/accuracies": 0.543749988079071, "rewards/chosen": -0.9110285639762878, "rewards/margins": 0.06163806468248367, "rewards/rejected": -0.9726667404174805, "step": 130 }, { "epoch": 0.289350301406564, "grad_norm": 11.144902490144164, "learning_rate": 4.4752726384955293e-07, "logits/chosen": -2.9404830932617188, "logits/rejected": -2.947343111038208, "logps/chosen": -0.9363792538642883, "logps/rejected": -0.9744508862495422, "loss": -0.0195, "rewards/accuracies": 0.5625, "rewards/chosen": -0.9363792538642883, "rewards/margins": 0.03807152435183525, "rewards/rejected": -0.9744508862495422, "step": 135 }, { "epoch": 0.3000669792364367, "grad_norm": 11.921950508886521, "learning_rate": 4.4164491972555294e-07, "logits/chosen": -2.8976681232452393, "logits/rejected": -2.898315906524658, "logps/chosen": -0.9219269752502441, "logps/rejected": -0.9640507698059082, "loss": -0.0259, "rewards/accuracies": 0.5625, "rewards/chosen": -0.9219269752502441, "rewards/margins": 0.04212382063269615, "rewards/rejected": -0.9640507698059082, "step": 140 }, { "epoch": 0.31078365706630945, "grad_norm": 13.471098814591679, "learning_rate": 4.3549326208287393e-07, "logits/chosen": -2.9619617462158203, "logits/rejected": -2.9274888038635254, "logps/chosen": -0.9526675343513489, "logps/rejected": -1.007359266281128, "loss": -0.0355, "rewards/accuracies": 0.5625, "rewards/chosen": -0.9526675343513489, "rewards/margins": 0.0546916201710701, "rewards/rejected": -1.007359266281128, "step": 145 }, { "epoch": 0.3215003348961822, "grad_norm": 12.766870563789471, "learning_rate": 4.2908093568271187e-07, "logits/chosen": -2.9606540203094482, "logits/rejected": -2.9726486206054688, "logps/chosen": -0.9194633364677429, "logps/rejected": -0.9709534645080566, "loss": -0.0371, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -0.9194633364677429, "rewards/margins": 0.05149019509553909, "rewards/rejected": -0.9709534645080566, "step": 150 }, { "epoch": 0.3322170127260549, "grad_norm": 12.28868938106036, "learning_rate": 4.2241695159714843e-07, "logits/chosen": -3.043233633041382, "logits/rejected": -3.0254199504852295, "logps/chosen": -0.9320791959762573, "logps/rejected": -0.9708281755447388, "loss": -0.0274, "rewards/accuracies": 0.5625, "rewards/chosen": -0.9320791959762573, "rewards/margins": 0.038748957216739655, "rewards/rejected": -0.9708281755447388, "step": 155 }, { "epoch": 0.34293369055592765, "grad_norm": 16.212616603609877, "learning_rate": 4.1551067454613074e-07, "logits/chosen": -2.943720579147339, "logits/rejected": -2.928537607192993, "logps/chosen": -0.921017050743103, "logps/rejected": -0.9882138967514038, "loss": -0.0359, "rewards/accuracies": 0.574999988079071, "rewards/chosen": -0.921017050743103, "rewards/margins": 0.06719694286584854, "rewards/rejected": -0.9882138967514038, "step": 160 }, { "epoch": 0.3536503683858004, "grad_norm": 15.792664568932544, "learning_rate": 4.0837180973747876e-07, "logits/chosen": -3.0298409461975098, "logits/rejected": -3.0069289207458496, "logps/chosen": -0.9174503087997437, "logps/rejected": -0.9726656079292297, "loss": -0.0342, "rewards/accuracies": 0.5687500238418579, "rewards/chosen": -0.9174503087997437, "rewards/margins": 0.05521535873413086, "rewards/rejected": -0.9726656079292297, "step": 165 }, { "epoch": 0.3643670462156731, "grad_norm": 10.787104513853434, "learning_rate": 4.0101038922841455e-07, "logits/chosen": -2.968522548675537, "logits/rejected": -2.9739623069763184, "logps/chosen": -0.9771224856376648, "logps/rejected": -1.0373847484588623, "loss": -0.0466, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": -0.9771224856376648, "rewards/margins": 0.06026231497526169, "rewards/rejected": -1.0373847484588623, "step": 170 }, { "epoch": 0.3750837240455459, "grad_norm": 13.17830711446757, "learning_rate": 3.934367578277783e-07, "logits/chosen": -2.9837470054626465, "logits/rejected": -2.9880218505859375, "logps/chosen": -1.000985026359558, "logps/rejected": -1.0419981479644775, "loss": -0.0381, "rewards/accuracies": 0.612500011920929, "rewards/chosen": -1.000985026359558, "rewards/margins": 0.041013117879629135, "rewards/rejected": -1.0419981479644775, "step": 175 }, { "epoch": 0.38580040187541864, "grad_norm": 14.72612824066128, "learning_rate": 3.856615585587434e-07, "logits/chosen": -2.9933667182922363, "logits/rejected": -3.0047719478607178, "logps/chosen": -1.0653940439224243, "logps/rejected": -1.1087322235107422, "loss": -0.0319, "rewards/accuracies": 0.518750011920929, "rewards/chosen": -1.0653940439224243, "rewards/margins": 0.043338216841220856, "rewards/rejected": -1.1087322235107422, "step": 180 }, { "epoch": 0.3965170797052914, "grad_norm": 13.397319728202142, "learning_rate": 3.776957177024581e-07, "logits/chosen": -2.9812378883361816, "logits/rejected": -2.9986259937286377, "logps/chosen": -1.0316323041915894, "logps/rejected": -1.0616919994354248, "loss": -0.033, "rewards/accuracies": 0.574999988079071, "rewards/chosen": -1.0316323041915894, "rewards/margins": 0.030059661716222763, "rewards/rejected": -1.0616919994354248, "step": 185 }, { "epoch": 0.4072337575351641, "grad_norm": 14.059564629221503, "learning_rate": 3.69550429443633e-07, "logits/chosen": -2.8960227966308594, "logits/rejected": -2.913637161254883, "logps/chosen": -1.012019157409668, "logps/rejected": -1.215237021446228, "loss": -0.0447, "rewards/accuracies": 0.65625, "rewards/chosen": -1.012019157409668, "rewards/margins": 0.20321793854236603, "rewards/rejected": -1.215237021446228, "step": 190 }, { "epoch": 0.41795043536503684, "grad_norm": 14.952805148520074, "learning_rate": 3.612371401396502e-07, "logits/chosen": -3.007540225982666, "logits/rejected": -2.9880564212799072, "logps/chosen": -1.0366661548614502, "logps/rejected": -1.0943790674209595, "loss": -0.0368, "rewards/accuracies": 0.5625, "rewards/chosen": -1.0366661548614502, "rewards/margins": 0.05771297961473465, "rewards/rejected": -1.0943790674209595, "step": 195 }, { "epoch": 0.4286671131949096, "grad_norm": 14.170450035952996, "learning_rate": 3.527675322353011e-07, "logits/chosen": -2.922602415084839, "logits/rejected": -2.9457390308380127, "logps/chosen": -0.970561146736145, "logps/rejected": -1.0033313035964966, "loss": -0.0473, "rewards/accuracies": 0.518750011920929, "rewards/chosen": -0.970561146736145, "rewards/margins": 0.032770149409770966, "rewards/rejected": -1.0033313035964966, "step": 200 }, { "epoch": 0.4393837910247823, "grad_norm": 15.488305590600266, "learning_rate": 3.4415350784575693e-07, "logits/chosen": -3.004404067993164, "logits/rejected": -2.9882898330688477, "logps/chosen": -0.9613070487976074, "logps/rejected": -1.0151447057724, "loss": -0.044, "rewards/accuracies": 0.612500011920929, "rewards/chosen": -0.9613070487976074, "rewards/margins": 0.053837649524211884, "rewards/rejected": -1.0151447057724, "step": 205 }, { "epoch": 0.45010046885465504, "grad_norm": 13.543647630514888, "learning_rate": 3.3540717203084224e-07, "logits/chosen": -2.956808090209961, "logits/rejected": -2.9622817039489746, "logps/chosen": -0.9170505404472351, "logps/rejected": -0.9884069561958313, "loss": -0.0514, "rewards/accuracies": 0.606249988079071, "rewards/chosen": -0.9170505404472351, "rewards/margins": 0.07135634124279022, "rewards/rejected": -0.9884069561958313, "step": 210 }, { "epoch": 0.4608171466845278, "grad_norm": 14.183913951293327, "learning_rate": 3.265408157841153e-07, "logits/chosen": -2.9880080223083496, "logits/rejected": -2.986445903778076, "logps/chosen": -0.9918737411499023, "logps/rejected": -1.0286133289337158, "loss": -0.0446, "rewards/accuracies": 0.550000011920929, "rewards/chosen": -0.9918737411499023, "rewards/margins": 0.036739740520715714, "rewards/rejected": -1.0286133289337158, "step": 215 }, { "epoch": 0.4715338245144005, "grad_norm": 16.687524624276737, "learning_rate": 3.1756689876066187e-07, "logits/chosen": -2.940664291381836, "logits/rejected": -2.940556049346924, "logps/chosen": -0.9756967425346375, "logps/rejected": -1.0494619607925415, "loss": -0.0496, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": -0.9756967425346375, "rewards/margins": 0.07376515865325928, "rewards/rejected": -1.0494619607925415, "step": 220 }, { "epoch": 0.4822505023442733, "grad_norm": 15.106449736541158, "learning_rate": 3.0849803176787184e-07, "logits/chosen": -2.978764057159424, "logits/rejected": -3.02057147026062, "logps/chosen": -0.9675830602645874, "logps/rejected": -1.0782067775726318, "loss": -0.0528, "rewards/accuracies": 0.606249988079071, "rewards/chosen": -0.9675830602645874, "rewards/margins": 0.11062371730804443, "rewards/rejected": -1.0782067775726318, "step": 225 }, { "epoch": 0.49296718017414604, "grad_norm": 19.732642999044504, "learning_rate": 2.993469590438065e-07, "logits/chosen": -2.91894793510437, "logits/rejected": -2.92021107673645, "logps/chosen": -0.9777423143386841, "logps/rejected": -1.0478038787841797, "loss": -0.0487, "rewards/accuracies": 0.606249988079071, "rewards/chosen": -0.9777423143386841, "rewards/margins": 0.07006163895130157, "rewards/rejected": -1.0478038787841797, "step": 230 }, { "epoch": 0.5036838580040187, "grad_norm": 15.470655040763342, "learning_rate": 2.901265403480597e-07, "logits/chosen": -2.868373394012451, "logits/rejected": -2.8822197914123535, "logps/chosen": -0.9746626615524292, "logps/rejected": -1.109392523765564, "loss": -0.0532, "rewards/accuracies": 0.574999988079071, "rewards/chosen": -0.9746626615524292, "rewards/margins": 0.13472989201545715, "rewards/rejected": -1.109392523765564, "step": 235 }, { "epoch": 0.5144005358338914, "grad_norm": 18.117444556863273, "learning_rate": 2.808497328902782e-07, "logits/chosen": -2.829807758331299, "logits/rejected": -2.837395191192627, "logps/chosen": -1.0404951572418213, "logps/rejected": -1.1007030010223389, "loss": -0.0453, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -1.0404951572418213, "rewards/margins": 0.06020789220929146, "rewards/rejected": -1.1007030010223389, "step": 240 }, { "epoch": 0.5251172136637642, "grad_norm": 18.628678925066545, "learning_rate": 2.715295731217396e-07, "logits/chosen": -2.81119966506958, "logits/rejected": -2.815497636795044, "logps/chosen": -1.0369569063186646, "logps/rejected": -1.112683892250061, "loss": -0.0596, "rewards/accuracies": 0.5562499761581421, "rewards/chosen": -1.0369569063186646, "rewards/margins": 0.07572712004184723, "rewards/rejected": -1.112683892250061, "step": 245 }, { "epoch": 0.535833891493637, "grad_norm": 17.866763466807367, "learning_rate": 2.621791584155725e-07, "logits/chosen": -2.83900785446167, "logits/rejected": -2.859978199005127, "logps/chosen": -1.1023390293121338, "logps/rejected": -1.1515963077545166, "loss": -0.0617, "rewards/accuracies": 0.6187499761581421, "rewards/chosen": -1.1023390293121338, "rewards/margins": 0.04925721138715744, "rewards/rejected": -1.1515963077545166, "step": 250 }, { "epoch": 0.5465505693235098, "grad_norm": 17.478329638335467, "learning_rate": 2.52811628661367e-07, "logits/chosen": -2.8983101844787598, "logits/rejected": -2.908754348754883, "logps/chosen": -1.0288032293319702, "logps/rejected": -1.134108066558838, "loss": -0.0631, "rewards/accuracies": 0.543749988079071, "rewards/chosen": -1.0288032293319702, "rewards/margins": 0.10530481487512589, "rewards/rejected": -1.134108066558838, "step": 255 }, { "epoch": 0.5572672471533825, "grad_norm": 15.049601923886751, "learning_rate": 2.434401478000353e-07, "logits/chosen": -2.8707144260406494, "logits/rejected": -2.876131057739258, "logps/chosen": -1.0516908168792725, "logps/rejected": -1.1409704685211182, "loss": -0.0405, "rewards/accuracies": 0.5, "rewards/chosen": -1.0516908168792725, "rewards/margins": 0.08927960693836212, "rewards/rejected": -1.1409704685211182, "step": 260 }, { "epoch": 0.5679839249832552, "grad_norm": 14.714810625194913, "learning_rate": 2.340778853248759e-07, "logits/chosen": -2.86578369140625, "logits/rejected": -2.8567006587982178, "logps/chosen": -1.0564695596694946, "logps/rejected": -1.199512004852295, "loss": -0.0695, "rewards/accuracies": 0.6312500238418579, "rewards/chosen": -1.0564695596694946, "rewards/margins": 0.143042653799057, "rewards/rejected": -1.199512004852295, "step": 265 }, { "epoch": 0.578700602813128, "grad_norm": 18.053080062016626, "learning_rate": 2.247379977748339e-07, "logits/chosen": -2.9022586345672607, "logits/rejected": -2.9201228618621826, "logps/chosen": -1.0403499603271484, "logps/rejected": -1.1265124082565308, "loss": -0.0673, "rewards/accuracies": 0.606249988079071, "rewards/chosen": -1.0403499603271484, "rewards/margins": 0.08616244792938232, "rewards/rejected": -1.1265124082565308, "step": 270 }, { "epoch": 0.5894172806430007, "grad_norm": 14.15566071309189, "learning_rate": 2.1543361024596573e-07, "logits/chosen": -2.9802463054656982, "logits/rejected": -2.988959550857544, "logps/chosen": -0.9800962209701538, "logps/rejected": -1.0803120136260986, "loss": -0.0525, "rewards/accuracies": 0.59375, "rewards/chosen": -0.9800962209701538, "rewards/margins": 0.1002158373594284, "rewards/rejected": -1.0803120136260986, "step": 275 }, { "epoch": 0.6001339584728734, "grad_norm": 17.56629547418189, "learning_rate": 2.0617779794709082e-07, "logits/chosen": -2.8620638847351074, "logits/rejected": -2.8629186153411865, "logps/chosen": -1.103073000907898, "logps/rejected": -1.196608304977417, "loss": -0.0632, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -1.103073000907898, "rewards/margins": 0.09353531897068024, "rewards/rejected": -1.196608304977417, "step": 280 }, { "epoch": 0.6108506363027462, "grad_norm": 15.46761530330997, "learning_rate": 1.9698356782554687e-07, "logits/chosen": -2.9556548595428467, "logits/rejected": -2.9529995918273926, "logps/chosen": -1.109263300895691, "logps/rejected": -1.1731650829315186, "loss": -0.0564, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": -1.109263300895691, "rewards/margins": 0.06390184909105301, "rewards/rejected": -1.1731650829315186, "step": 285 }, { "epoch": 0.6215673141326189, "grad_norm": 16.8037567850014, "learning_rate": 1.878638402888727e-07, "logits/chosen": -2.9940178394317627, "logits/rejected": -2.9970099925994873, "logps/chosen": -0.9795197248458862, "logps/rejected": -1.0418453216552734, "loss": -0.0668, "rewards/accuracies": 0.5687500238418579, "rewards/chosen": -0.9795197248458862, "rewards/margins": 0.06232555955648422, "rewards/rejected": -1.0418453216552734, "step": 290 }, { "epoch": 0.6322839919624916, "grad_norm": 16.60747046570722, "learning_rate": 1.7883143104810223e-07, "logits/chosen": -2.9415085315704346, "logits/rejected": -2.9544739723205566, "logps/chosen": -1.069349765777588, "logps/rejected": -1.2494323253631592, "loss": -0.0739, "rewards/accuracies": 0.625, "rewards/chosen": -1.069349765777588, "rewards/margins": 0.1800827533006668, "rewards/rejected": -1.2494323253631592, "step": 295 }, { "epoch": 0.6430006697923644, "grad_norm": 15.422934564351497, "learning_rate": 1.6989903310818548e-07, "logits/chosen": -3.0317275524139404, "logits/rejected": -3.049278736114502, "logps/chosen": -1.0510863065719604, "logps/rejected": -1.1395753622055054, "loss": -0.0517, "rewards/accuracies": 0.5874999761581421, "rewards/chosen": -1.0510863065719604, "rewards/margins": 0.08848897367715836, "rewards/rejected": -1.1395753622055054, "step": 300 }, { "epoch": 0.6537173476222371, "grad_norm": 17.56071364411128, "learning_rate": 1.610791989308466e-07, "logits/chosen": -2.9625232219696045, "logits/rejected": -2.9480719566345215, "logps/chosen": -1.0810691118240356, "logps/rejected": -1.1474703550338745, "loss": -0.0711, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -1.0810691118240356, "rewards/margins": 0.06640125066041946, "rewards/rejected": -1.1474703550338745, "step": 305 }, { "epoch": 0.6644340254521098, "grad_norm": 19.400175535223255, "learning_rate": 1.523843227949418e-07, "logits/chosen": -3.0158638954162598, "logits/rejected": -3.014029026031494, "logps/chosen": -1.054434061050415, "logps/rejected": -1.1792142391204834, "loss": -0.0711, "rewards/accuracies": 0.625, "rewards/chosen": -1.054434061050415, "rewards/margins": 0.12478007376194, "rewards/rejected": -1.1792142391204834, "step": 310 }, { "epoch": 0.6751507032819826, "grad_norm": 18.063066961906372, "learning_rate": 1.4382662337910867e-07, "logits/chosen": -2.9173102378845215, "logits/rejected": -2.9165103435516357, "logps/chosen": -1.1207153797149658, "logps/rejected": -1.2241191864013672, "loss": -0.0582, "rewards/accuracies": 0.6312500238418579, "rewards/chosen": -1.1207153797149658, "rewards/margins": 0.10340368747711182, "rewards/rejected": -1.2241191864013672, "step": 315 }, { "epoch": 0.6858673811118553, "grad_norm": 17.656998709658712, "learning_rate": 1.3541812659118104e-07, "logits/chosen": -2.924173593521118, "logits/rejected": -2.901674747467041, "logps/chosen": -1.1284511089324951, "logps/rejected": -1.2347378730773926, "loss": -0.0605, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -1.1284511089324951, "rewards/margins": 0.10628668963909149, "rewards/rejected": -1.2347378730773926, "step": 320 }, { "epoch": 0.696584058941728, "grad_norm": 17.674353062779463, "learning_rate": 1.2717064866850047e-07, "logits/chosen": -2.9069221019744873, "logits/rejected": -2.9266703128814697, "logps/chosen": -1.0635308027267456, "logps/rejected": -1.2443645000457764, "loss": -0.076, "rewards/accuracies": 0.637499988079071, "rewards/chosen": -1.0635308027267456, "rewards/margins": 0.1808338761329651, "rewards/rejected": -1.2443645000457764, "step": 325 }, { "epoch": 0.7073007367716008, "grad_norm": 16.692003983984833, "learning_rate": 1.1909577957287117e-07, "logits/chosen": -2.9482550621032715, "logits/rejected": -2.9383444786071777, "logps/chosen": -1.1115998029708862, "logps/rejected": -1.269691824913025, "loss": -0.0875, "rewards/accuracies": 0.606249988079071, "rewards/chosen": -1.1115998029708862, "rewards/margins": 0.15809203684329987, "rewards/rejected": -1.269691824913025, "step": 330 }, { "epoch": 0.7180174146014735, "grad_norm": 16.832132437998734, "learning_rate": 1.1120486670349441e-07, "logits/chosen": -3.0022974014282227, "logits/rejected": -3.0084285736083984, "logps/chosen": -1.1076433658599854, "logps/rejected": -1.2158708572387695, "loss": -0.0631, "rewards/accuracies": 0.6187499761581421, "rewards/chosen": -1.1076433658599854, "rewards/margins": 0.10822737216949463, "rewards/rejected": -1.2158708572387695, "step": 335 }, { "epoch": 0.7287340924313462, "grad_norm": 16.746834458412387, "learning_rate": 1.0350899895076948e-07, "logits/chosen": -2.960242509841919, "logits/rejected": -2.9590137004852295, "logps/chosen": -1.1618969440460205, "logps/rejected": -1.3054715394973755, "loss": -0.063, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -1.1618969440460205, "rewards/margins": 0.14357443153858185, "rewards/rejected": -1.3054715394973755, "step": 340 }, { "epoch": 0.739450770261219, "grad_norm": 16.472011449839883, "learning_rate": 9.601899111337069e-08, "logits/chosen": -2.9834091663360596, "logits/rejected": -2.9975457191467285, "logps/chosen": -1.126527190208435, "logps/rejected": -1.3144327402114868, "loss": -0.0777, "rewards/accuracies": 0.543749988079071, "rewards/chosen": -1.126527190208435, "rewards/margins": 0.18790560960769653, "rewards/rejected": -1.3144327402114868, "step": 345 }, { "epoch": 0.7501674480910918, "grad_norm": 17.215572295903478, "learning_rate": 8.874536870049765e-08, "logits/chosen": -2.969423294067383, "logits/rejected": -2.969230890274048, "logps/chosen": -1.127826452255249, "logps/rejected": -1.271470069885254, "loss": -0.0752, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -1.127826452255249, "rewards/margins": 0.1436435580253601, "rewards/rejected": -1.271470069885254, "step": 350 }, { "epoch": 0.7608841259209645, "grad_norm": 19.033752258043222, "learning_rate": 8.169835314065663e-08, "logits/chosen": -2.9632205963134766, "logits/rejected": -2.9548847675323486, "logps/chosen": -1.1820392608642578, "logps/rejected": -1.2698676586151123, "loss": -0.0668, "rewards/accuracies": 0.518750011920929, "rewards/chosen": -1.1820392608642578, "rewards/margins": 0.08782847970724106, "rewards/rejected": -1.2698676586151123, "step": 355 }, { "epoch": 0.7716008037508373, "grad_norm": 17.064795493163807, "learning_rate": 7.488784741775816e-08, "logits/chosen": -2.976466178894043, "logits/rejected": -2.9713473320007324, "logps/chosen": -1.054131269454956, "logps/rejected": -1.1566834449768066, "loss": -0.0663, "rewards/accuracies": 0.5625, "rewards/chosen": -1.054131269454956, "rewards/margins": 0.10255217552185059, "rewards/rejected": -1.1566834449768066, "step": 360 }, { "epoch": 0.78231748158071, "grad_norm": 17.868156711335107, "learning_rate": 6.832342215471715e-08, "logits/chosen": -2.98659086227417, "logits/rejected": -2.986912488937378, "logps/chosen": -1.1025737524032593, "logps/rejected": -1.242219090461731, "loss": -0.0823, "rewards/accuracies": 0.612500011920929, "rewards/chosen": -1.1025737524032593, "rewards/margins": 0.13964524865150452, "rewards/rejected": -1.242219090461731, "step": 365 }, { "epoch": 0.7930341594105828, "grad_norm": 24.714287844140323, "learning_rate": 6.201430216410985e-08, "logits/chosen": -3.0034446716308594, "logits/rejected": -3.001354217529297, "logps/chosen": -1.1147871017456055, "logps/rejected": -1.200984239578247, "loss": -0.081, "rewards/accuracies": 0.5625, "rewards/chosen": -1.1147871017456055, "rewards/margins": 0.08619704097509384, "rewards/rejected": -1.200984239578247, "step": 370 }, { "epoch": 0.8037508372404555, "grad_norm": 20.89166421346418, "learning_rate": 5.596935348479029e-08, "logits/chosen": -2.9388105869293213, "logits/rejected": -2.948521137237549, "logps/chosen": -1.2442572116851807, "logps/rejected": -1.393157958984375, "loss": -0.0861, "rewards/accuracies": 0.5625, "rewards/chosen": -1.2442572116851807, "rewards/margins": 0.1489005833864212, "rewards/rejected": -1.393157958984375, "step": 375 }, { "epoch": 0.8144675150703282, "grad_norm": 22.406198534031674, "learning_rate": 5.01970709226808e-08, "logits/chosen": -3.028179883956909, "logits/rejected": -3.009589672088623, "logps/chosen": -1.0915172100067139, "logps/rejected": -1.243607997894287, "loss": -0.0837, "rewards/accuracies": 0.637499988079071, "rewards/chosen": -1.0915172100067139, "rewards/margins": 0.1520906686782837, "rewards/rejected": -1.243607997894287, "step": 380 }, { "epoch": 0.825184192900201, "grad_norm": 24.898962036749563, "learning_rate": 4.4705566113246635e-08, "logits/chosen": -2.9715943336486816, "logits/rejected": -2.949638605117798, "logps/chosen": -1.228461503982544, "logps/rejected": -1.3865272998809814, "loss": -0.0897, "rewards/accuracies": 0.612500011920929, "rewards/chosen": -1.228461503982544, "rewards/margins": 0.15806564688682556, "rewards/rejected": -1.3865272998809814, "step": 385 }, { "epoch": 0.8359008707300737, "grad_norm": 19.449984589836117, "learning_rate": 3.950255612243042e-08, "logits/chosen": -2.951904058456421, "logits/rejected": -2.928208827972412, "logps/chosen": -1.2001017332077026, "logps/rejected": -1.3440535068511963, "loss": -0.0724, "rewards/accuracies": 0.612500011920929, "rewards/chosen": -1.2001017332077026, "rewards/margins": 0.1439516693353653, "rewards/rejected": -1.3440535068511963, "step": 390 }, { "epoch": 0.8466175485599464, "grad_norm": 19.857222600639766, "learning_rate": 3.459535260206384e-08, "logits/chosen": -2.9977169036865234, "logits/rejected": -3.020946979522705, "logps/chosen": -1.096103549003601, "logps/rejected": -1.21526300907135, "loss": -0.0898, "rewards/accuracies": 0.6312500238418579, "rewards/chosen": -1.096103549003601, "rewards/margins": 0.11915931850671768, "rewards/rejected": -1.21526300907135, "step": 395 }, { "epoch": 0.8573342263898192, "grad_norm": 21.73532568075866, "learning_rate": 2.999085151499764e-08, "logits/chosen": -2.9534425735473633, "logits/rejected": -2.9359939098358154, "logps/chosen": -1.0908204317092896, "logps/rejected": -1.356081485748291, "loss": -0.0905, "rewards/accuracies": 0.737500011920929, "rewards/chosen": -1.0908204317092896, "rewards/margins": 0.2652612626552582, "rewards/rejected": -1.356081485748291, "step": 400 }, { "epoch": 0.8573342263898192, "eval_logits/chosen": -2.8999884128570557, "eval_logits/rejected": -2.892754316329956, "eval_logps/chosen": -1.1346372365951538, "eval_logps/rejected": -1.2466005086898804, "eval_loss": -0.07080772519111633, "eval_rewards/accuracies": 0.5757978558540344, "eval_rewards/chosen": -1.1346372365951538, "eval_rewards/margins": 0.11196334660053253, "eval_rewards/rejected": -1.2466005086898804, "eval_runtime": 432.4273, "eval_samples_per_second": 6.924, "eval_steps_per_second": 0.435, "step": 400 }, { "epoch": 0.8680509042196919, "grad_norm": 19.440021908588918, "learning_rate": 2.5695523444387747e-08, "logits/chosen": -2.9388387203216553, "logits/rejected": -2.9628615379333496, "logps/chosen": -1.111882209777832, "logps/rejected": -1.2351758480072021, "loss": -0.09, "rewards/accuracies": 0.625, "rewards/chosen": -1.111882209777832, "rewards/margins": 0.12329357862472534, "rewards/rejected": -1.2351758480072021, "step": 405 }, { "epoch": 0.8787675820495646, "grad_norm": 19.454852119565878, "learning_rate": 2.1715404500756875e-08, "logits/chosen": -2.98136305809021, "logits/rejected": -3.0059518814086914, "logps/chosen": -1.090047836303711, "logps/rejected": -1.203460693359375, "loss": -0.0823, "rewards/accuracies": 0.5874999761581421, "rewards/chosen": -1.090047836303711, "rewards/margins": 0.11341279745101929, "rewards/rejected": -1.203460693359375, "step": 410 }, { "epoch": 0.8894842598794374, "grad_norm": 27.119138813457198, "learning_rate": 1.8056087839608546e-08, "logits/chosen": -2.9424166679382324, "logits/rejected": -2.9254748821258545, "logps/chosen": -1.1260377168655396, "logps/rejected": -1.3184921741485596, "loss": -0.1086, "rewards/accuracies": 0.6875, "rewards/chosen": -1.1260377168655396, "rewards/margins": 0.19245454668998718, "rewards/rejected": -1.3184921741485596, "step": 415 }, { "epoch": 0.9002009377093101, "grad_norm": 19.3060602023721, "learning_rate": 1.4722715801513935e-08, "logits/chosen": -2.9605276584625244, "logits/rejected": -2.967475414276123, "logps/chosen": -1.0344493389129639, "logps/rejected": -1.0916310548782349, "loss": -0.0802, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": -1.0344493389129639, "rewards/margins": 0.057181768119335175, "rewards/rejected": -1.0916310548782349, "step": 420 }, { "epoch": 0.9109176155391828, "grad_norm": 20.048601077954377, "learning_rate": 1.1719972685717395e-08, "logits/chosen": -2.970942258834839, "logits/rejected": -2.9757351875305176, "logps/chosen": -1.0653189420700073, "logps/rejected": -1.3184258937835693, "loss": -0.1043, "rewards/accuracies": 0.65625, "rewards/chosen": -1.0653189420700073, "rewards/margins": 0.2531070411205292, "rewards/rejected": -1.3184258937835693, "step": 425 }, { "epoch": 0.9216342933690556, "grad_norm": 21.66418470338373, "learning_rate": 9.05207816741485e-09, "logits/chosen": -2.939943313598633, "logits/rejected": -2.9273343086242676, "logps/chosen": -1.2503459453582764, "logps/rejected": -1.4009029865264893, "loss": -0.1013, "rewards/accuracies": 0.59375, "rewards/chosen": -1.2503459453582764, "rewards/margins": 0.15055687725543976, "rewards/rejected": -1.4009029865264893, "step": 430 }, { "epoch": 0.9323509711989283, "grad_norm": 21.774320932145653, "learning_rate": 6.722781367956481e-09, "logits/chosen": -3.0610954761505127, "logits/rejected": -3.0484843254089355, "logps/chosen": -1.0789310932159424, "logps/rejected": -1.2742266654968262, "loss": -0.1009, "rewards/accuracies": 0.59375, "rewards/chosen": -1.0789310932159424, "rewards/margins": 0.19529545307159424, "rewards/rejected": -1.2742266654968262, "step": 435 }, { "epoch": 0.943067649028801, "grad_norm": 19.319233646112725, "learning_rate": 4.735355586305861e-09, "logits/chosen": -2.94116473197937, "logits/rejected": -2.949489116668701, "logps/chosen": -1.1311179399490356, "logps/rejected": -1.3173315525054932, "loss": -0.0993, "rewards/accuracies": 0.6499999761581421, "rewards/chosen": -1.1311179399490356, "rewards/margins": 0.1862136423587799, "rewards/rejected": -1.3173315525054932, "step": 440 }, { "epoch": 0.9537843268586738, "grad_norm": 19.556600822562356, "learning_rate": 3.0925936991598454e-09, "logits/chosen": -2.9415977001190186, "logits/rejected": -2.944352388381958, "logps/chosen": -1.1981102228164673, "logps/rejected": -1.329925775527954, "loss": -0.0746, "rewards/accuracies": 0.6187499761581421, "rewards/chosen": -1.1981102228164673, "rewards/margins": 0.1318153738975525, "rewards/rejected": -1.329925775527954, "step": 445 }, { "epoch": 0.9645010046885466, "grad_norm": 24.43813858171124, "learning_rate": 1.7968042361930225e-09, "logits/chosen": -2.958519697189331, "logits/rejected": -2.9468445777893066, "logps/chosen": -1.1451342105865479, "logps/rejected": -1.265005111694336, "loss": -0.0894, "rewards/accuracies": 0.5687500238418579, "rewards/chosen": -1.1451342105865479, "rewards/margins": 0.11987074464559555, "rewards/rejected": -1.265005111694336, "step": 450 }, { "epoch": 0.9752176825184193, "grad_norm": 23.495311684580507, "learning_rate": 8.498081359423759e-10, "logits/chosen": -2.9030869007110596, "logits/rejected": -2.932225465774536, "logps/chosen": -1.1186686754226685, "logps/rejected": -1.335756540298462, "loss": -0.1008, "rewards/accuracies": 0.606249988079071, "rewards/chosen": -1.1186686754226685, "rewards/margins": 0.21708789467811584, "rewards/rejected": -1.335756540298462, "step": 455 }, { "epoch": 0.9859343603482921, "grad_norm": 20.358552883416746, "learning_rate": 2.5293618689056774e-10, "logits/chosen": -2.9370243549346924, "logits/rejected": -2.953622341156006, "logps/chosen": -1.1718332767486572, "logps/rejected": -1.4025189876556396, "loss": -0.0805, "rewards/accuracies": 0.606249988079071, "rewards/chosen": -1.1718332767486572, "rewards/margins": 0.230685755610466, "rewards/rejected": -1.4025189876556396, "step": 460 }, { "epoch": 0.9966510381781648, "grad_norm": 27.657325299971138, "learning_rate": 7.02715734424153e-12, "logits/chosen": -2.9429261684417725, "logits/rejected": -2.951002359390259, "logps/chosen": -1.2534146308898926, "logps/rejected": -1.3702634572982788, "loss": -0.0688, "rewards/accuracies": 0.6187499761581421, "rewards/chosen": -1.2534146308898926, "rewards/margins": 0.11684894561767578, "rewards/rejected": -1.3702634572982788, "step": 465 }, { "epoch": 0.9987943737441393, "step": 466, "total_flos": 0.0, "train_loss": 0.0, "train_runtime": 4.3106, "train_samples_per_second": 13854.197, "train_steps_per_second": 108.105 } ], "logging_steps": 5, "max_steps": 466, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 1000000, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": true }, "attributes": {} } }, "total_flos": 0.0, "train_batch_size": 2, "trial_name": null, "trial_params": null }