{ "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": 37.4633897932904, "learning_rate": 2.1276595744680853e-06, "logits/chosen": -2.397902250289917, "logits/rejected": -2.391808271408081, "logps/chosen": -0.5661157369613647, "logps/rejected": -0.5547963380813599, "loss": 1.5468, "rewards/accuracies": 0.4375, "rewards/chosen": -0.5661157369613647, "rewards/margins": -0.011319375596940517, "rewards/rejected": -0.5547963380813599, "step": 5 }, { "epoch": 0.021433355659745478, "grad_norm": 18.848503699490514, "learning_rate": 4.255319148936171e-06, "logits/chosen": -2.374025344848633, "logits/rejected": -2.345165729522705, "logps/chosen": -0.5381986498832703, "logps/rejected": -0.5637714266777039, "loss": 1.5538, "rewards/accuracies": 0.518750011920929, "rewards/chosen": -0.5381986498832703, "rewards/margins": 0.025572726503014565, "rewards/rejected": -0.5637714266777039, "step": 10 }, { "epoch": 0.032150033489618215, "grad_norm": 18.865749434279863, "learning_rate": 6.382978723404256e-06, "logits/chosen": -2.228337526321411, "logits/rejected": -2.2436676025390625, "logps/chosen": -0.6078538298606873, "logps/rejected": -0.6489665508270264, "loss": 1.5486, "rewards/accuracies": 0.550000011920929, "rewards/chosen": -0.6078538298606873, "rewards/margins": 0.04111265391111374, "rewards/rejected": -0.6489665508270264, "step": 15 }, { "epoch": 0.042866711319490956, "grad_norm": 11.695416903513719, "learning_rate": 8.510638297872341e-06, "logits/chosen": -2.3822503089904785, "logits/rejected": -2.399451732635498, "logps/chosen": -0.6396132111549377, "logps/rejected": -0.6679492592811584, "loss": 1.5283, "rewards/accuracies": 0.574999988079071, "rewards/chosen": -0.6396132111549377, "rewards/margins": 0.028336137533187866, "rewards/rejected": -0.6679492592811584, "step": 20 }, { "epoch": 0.0535833891493637, "grad_norm": 8.657694670203071, "learning_rate": 1.0638297872340426e-05, "logits/chosen": -2.6283469200134277, "logits/rejected": -2.6289353370666504, "logps/chosen": -0.6763823628425598, "logps/rejected": -0.7394057512283325, "loss": 1.5522, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -0.6763823628425598, "rewards/margins": 0.0630234032869339, "rewards/rejected": -0.7394057512283325, "step": 25 }, { "epoch": 0.06430006697923643, "grad_norm": 11.701639197469074, "learning_rate": 1.2765957446808513e-05, "logits/chosen": -2.7873454093933105, "logits/rejected": -2.815898895263672, "logps/chosen": -0.8007046580314636, "logps/rejected": -0.8376566767692566, "loss": 1.5826, "rewards/accuracies": 0.5062500238418579, "rewards/chosen": -0.8007046580314636, "rewards/margins": 0.03695206344127655, "rewards/rejected": -0.8376566767692566, "step": 30 }, { "epoch": 0.07501674480910918, "grad_norm": 85.64689375367887, "learning_rate": 1.4893617021276596e-05, "logits/chosen": -2.5493829250335693, "logits/rejected": -2.5297470092773438, "logps/chosen": -0.9185592532157898, "logps/rejected": -1.0066051483154297, "loss": 1.6494, "rewards/accuracies": 0.543749988079071, "rewards/chosen": -0.9185592532157898, "rewards/margins": 0.08804582059383392, "rewards/rejected": -1.0066051483154297, "step": 35 }, { "epoch": 0.08573342263898191, "grad_norm": 12.618515943070657, "learning_rate": 1.7021276595744682e-05, "logits/chosen": -2.7925362586975098, "logits/rejected": -2.7926783561706543, "logps/chosen": -0.95704585313797, "logps/rejected": -0.9239652752876282, "loss": 1.6286, "rewards/accuracies": 0.4437499940395355, "rewards/chosen": -0.95704585313797, "rewards/margins": -0.033080630004405975, "rewards/rejected": -0.9239652752876282, "step": 40 }, { "epoch": 0.09645010046885466, "grad_norm": 7.437533737509709, "learning_rate": 1.914893617021277e-05, "logits/chosen": -2.668349027633667, "logits/rejected": -2.6576356887817383, "logps/chosen": -0.9402163624763489, "logps/rejected": -0.9749400019645691, "loss": 1.5995, "rewards/accuracies": 0.5562499761581421, "rewards/chosen": -0.9402163624763489, "rewards/margins": 0.034723617136478424, "rewards/rejected": -0.9749400019645691, "step": 45 }, { "epoch": 0.1071667782987274, "grad_norm": 8.099072281553982, "learning_rate": 1.999747031816659e-05, "logits/chosen": -2.796558380126953, "logits/rejected": -2.7897918224334717, "logps/chosen": -0.8397698402404785, "logps/rejected": -0.8385842442512512, "loss": 1.566, "rewards/accuracies": 0.512499988079071, "rewards/chosen": -0.8397698402404785, "rewards/margins": -0.0011855497723445296, "rewards/rejected": -0.8385842442512512, "step": 50 }, { "epoch": 0.11788345612860013, "grad_norm": 13.021189847208735, "learning_rate": 1.9982015786040484e-05, "logits/chosen": -2.577963352203369, "logits/rejected": -2.5813965797424316, "logps/chosen": -0.9185325503349304, "logps/rejected": -0.9634258151054382, "loss": 1.587, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": -0.9185325503349304, "rewards/margins": 0.044893164187669754, "rewards/rejected": -0.9634258151054382, "step": 55 }, { "epoch": 0.12860013395847286, "grad_norm": 29.275115325498533, "learning_rate": 1.9952533792092286e-05, "logits/chosen": -2.7369465827941895, "logits/rejected": -2.746910810470581, "logps/chosen": -0.8204312324523926, "logps/rejected": -0.871277928352356, "loss": 1.6101, "rewards/accuracies": 0.5687500238418579, "rewards/chosen": -0.8204312324523926, "rewards/margins": 0.05084659531712532, "rewards/rejected": -0.871277928352356, "step": 60 }, { "epoch": 0.13931681178834562, "grad_norm": 9.013295050879966, "learning_rate": 1.9909065766585522e-05, "logits/chosen": -2.476846694946289, "logits/rejected": -2.4859910011291504, "logps/chosen": -0.8948932886123657, "logps/rejected": -0.9379276037216187, "loss": 1.6324, "rewards/accuracies": 0.5687500238418579, "rewards/chosen": -0.8948932886123657, "rewards/margins": 0.043034326285123825, "rewards/rejected": -0.9379276037216187, "step": 65 }, { "epoch": 0.15003348961821836, "grad_norm": 8.308683516584036, "learning_rate": 1.9851672793982618e-05, "logits/chosen": -2.443450450897217, "logits/rejected": -2.465125799179077, "logps/chosen": -0.957764744758606, "logps/rejected": -0.9942213296890259, "loss": 1.6029, "rewards/accuracies": 0.550000011920929, "rewards/chosen": -0.957764744758606, "rewards/margins": 0.03645653277635574, "rewards/rejected": -0.9942213296890259, "step": 70 }, { "epoch": 0.1607501674480911, "grad_norm": 7.915844945947729, "learning_rate": 1.978043552710454e-05, "logits/chosen": -2.4849705696105957, "logits/rejected": -2.4826462268829346, "logps/chosen": -0.9050397872924805, "logps/rejected": -0.9400200843811035, "loss": 1.6052, "rewards/accuracies": 0.53125, "rewards/chosen": -0.9050397872924805, "rewards/margins": 0.03498036414384842, "rewards/rejected": -0.9400200843811035, "step": 75 }, { "epoch": 0.17146684527796383, "grad_norm": 304.0629881690128, "learning_rate": 1.9695454073791517e-05, "logits/chosen": -2.500096559524536, "logits/rejected": -2.5168652534484863, "logps/chosen": -0.906775951385498, "logps/rejected": -1.02707040309906, "loss": 1.82, "rewards/accuracies": 0.6312500238418579, "rewards/chosen": -0.906775951385498, "rewards/margins": 0.12029434740543365, "rewards/rejected": -1.02707040309906, "step": 80 }, { "epoch": 0.18218352310783656, "grad_norm": 10.960148765645817, "learning_rate": 1.9596847856224143e-05, "logits/chosen": -2.257906436920166, "logits/rejected": -2.265631675720215, "logps/chosen": -1.1177845001220703, "logps/rejected": -1.1891874074935913, "loss": 1.6996, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -1.1177845001220703, "rewards/margins": 0.07140282541513443, "rewards/rejected": -1.1891874074935913, "step": 85 }, { "epoch": 0.19290020093770932, "grad_norm": 10.383486663685009, "learning_rate": 1.9484755443102573e-05, "logits/chosen": -2.215013027191162, "logits/rejected": -2.2123208045959473, "logps/chosen": -1.035982370376587, "logps/rejected": -1.0804064273834229, "loss": 1.6576, "rewards/accuracies": 0.543749988079071, "rewards/chosen": -1.035982370376587, "rewards/margins": 0.044424109160900116, "rewards/rejected": -1.0804064273834229, "step": 90 }, { "epoch": 0.20361687876758205, "grad_norm": 9.444434646488055, "learning_rate": 1.9359334354919587e-05, "logits/chosen": -2.5854439735412598, "logits/rejected": -2.604400157928467, "logps/chosen": -1.0645908117294312, "logps/rejected": -1.1030633449554443, "loss": 1.6255, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -1.0645908117294312, "rewards/margins": 0.03847253695130348, "rewards/rejected": -1.1030633449554443, "step": 95 }, { "epoch": 0.2143335565974548, "grad_norm": 10.008429713827578, "learning_rate": 1.9220760842601245e-05, "logits/chosen": -2.41093373298645, "logits/rejected": -2.416090488433838, "logps/chosen": -1.0662959814071655, "logps/rejected": -1.0942766666412354, "loss": 1.6366, "rewards/accuracies": 0.518750011920929, "rewards/chosen": -1.0662959814071655, "rewards/margins": 0.027980733662843704, "rewards/rejected": -1.0942766666412354, "step": 100 }, { "epoch": 0.22505023442732752, "grad_norm": 13.793372325395232, "learning_rate": 1.9069229639826113e-05, "logits/chosen": -2.593897581100464, "logits/rejected": -2.592832326889038, "logps/chosen": -1.0412575006484985, "logps/rejected": -1.1209934949874878, "loss": 1.6663, "rewards/accuracies": 0.5625, "rewards/chosen": -1.0412575006484985, "rewards/margins": 0.0797361508011818, "rewards/rejected": -1.1209934949874878, "step": 105 }, { "epoch": 0.23576691225720026, "grad_norm": 25.707371820916816, "learning_rate": 1.8904953689371223e-05, "logits/chosen": -2.1585533618927, "logits/rejected": -2.143022060394287, "logps/chosen": -1.0903128385543823, "logps/rejected": -1.1769222021102905, "loss": 1.6708, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -1.0903128385543823, "rewards/margins": 0.08660950511693954, "rewards/rejected": -1.1769222021102905, "step": 110 }, { "epoch": 0.24648359008707302, "grad_norm": 15.08396540818952, "learning_rate": 1.8728163843869215e-05, "logits/chosen": -2.3684897422790527, "logits/rejected": -2.3717474937438965, "logps/chosen": -1.1836369037628174, "logps/rejected": -1.1988976001739502, "loss": 1.8024, "rewards/accuracies": 0.518750011920929, "rewards/chosen": -1.1836369037628174, "rewards/margins": 0.015260684303939342, "rewards/rejected": -1.1988976001739502, "step": 115 }, { "epoch": 0.2572002679169457, "grad_norm": 10.565734009587494, "learning_rate": 1.853910854139727e-05, "logits/chosen": -2.122260570526123, "logits/rejected": -2.138091802597046, "logps/chosen": -1.133177399635315, "logps/rejected": -1.1549627780914307, "loss": 1.7056, "rewards/accuracies": 0.543749988079071, "rewards/chosen": -1.133177399635315, "rewards/margins": 0.02178535796701908, "rewards/rejected": -1.1549627780914307, "step": 120 }, { "epoch": 0.2679169457468185, "grad_norm": 7.929268982931603, "learning_rate": 1.833805345635369e-05, "logits/chosen": -2.467066526412964, "logits/rejected": -2.470454692840576, "logps/chosen": -1.0294818878173828, "logps/rejected": -1.0639444589614868, "loss": 1.6829, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": -1.0294818878173828, "rewards/margins": 0.03446248173713684, "rewards/rejected": -1.0639444589614868, "step": 125 }, { "epoch": 0.27863362357669125, "grad_norm": 9.40947063841382, "learning_rate": 1.8125281126112678e-05, "logits/chosen": -2.5180554389953613, "logits/rejected": -2.5261032581329346, "logps/chosen": -1.044611930847168, "logps/rejected": -1.1444398164749146, "loss": 1.6283, "rewards/accuracies": 0.5625, "rewards/chosen": -1.044611930847168, "rewards/margins": 0.09982797503471375, "rewards/rejected": -1.1444398164749146, "step": 130 }, { "epoch": 0.289350301406564, "grad_norm": 7.797174649236775, "learning_rate": 1.790109055398212e-05, "logits/chosen": -2.5548009872436523, "logits/rejected": -2.5629782676696777, "logps/chosen": -0.9959312677383423, "logps/rejected": -1.0395903587341309, "loss": 1.6327, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -0.9959312677383423, "rewards/margins": 0.04365905374288559, "rewards/rejected": -1.0395903587341309, "step": 135 }, { "epoch": 0.3000669792364367, "grad_norm": 8.65910654888172, "learning_rate": 1.766579678902212e-05, "logits/chosen": -2.452951431274414, "logits/rejected": -2.459916114807129, "logps/chosen": -0.9893513917922974, "logps/rejected": -1.0323150157928467, "loss": 1.6058, "rewards/accuracies": 0.5625, "rewards/chosen": -0.9893513917922974, "rewards/margins": 0.042963750660419464, "rewards/rejected": -1.0323150157928467, "step": 140 }, { "epoch": 0.31078365706630945, "grad_norm": 8.475342441948424, "learning_rate": 1.741973048331496e-05, "logits/chosen": -2.608076810836792, "logits/rejected": -2.592602491378784, "logps/chosen": -0.959572434425354, "logps/rejected": -1.034380316734314, "loss": 1.5962, "rewards/accuracies": 0.5562499761581421, "rewards/chosen": -0.959572434425354, "rewards/margins": 0.07480800151824951, "rewards/rejected": -1.034380316734314, "step": 145 }, { "epoch": 0.3215003348961822, "grad_norm": 13.2413080829705, "learning_rate": 1.7163237427308477e-05, "logits/chosen": -2.6177635192871094, "logits/rejected": -2.6168532371520996, "logps/chosen": -1.004525899887085, "logps/rejected": -1.0910793542861938, "loss": 1.6239, "rewards/accuracies": 0.625, "rewards/chosen": -1.004525899887085, "rewards/margins": 0.08655332773923874, "rewards/rejected": -1.0910793542861938, "step": 150 }, { "epoch": 0.3322170127260549, "grad_norm": 8.151298227822702, "learning_rate": 1.689667806388594e-05, "logits/chosen": -2.7449164390563965, "logits/rejected": -2.7318592071533203, "logps/chosen": -0.9720630645751953, "logps/rejected": -1.047321081161499, "loss": 1.5985, "rewards/accuracies": 0.612500011920929, "rewards/chosen": -0.9720630645751953, "rewards/margins": 0.0752580463886261, "rewards/rejected": -1.047321081161499, "step": 155 }, { "epoch": 0.34293369055592765, "grad_norm": 7.660153117545113, "learning_rate": 1.6620426981845233e-05, "logits/chosen": -2.5920422077178955, "logits/rejected": -2.5721192359924316, "logps/chosen": -0.9702916145324707, "logps/rejected": -1.0599411725997925, "loss": 1.6107, "rewards/accuracies": 0.5562499761581421, "rewards/chosen": -0.9702916145324707, "rewards/margins": 0.08964961767196655, "rewards/rejected": -1.0599411725997925, "step": 160 }, { "epoch": 0.3536503683858004, "grad_norm": 469.42247386416926, "learning_rate": 1.633487238949915e-05, "logits/chosen": -2.528369426727295, "logits/rejected": -2.5209622383117676, "logps/chosen": -0.9299852252006531, "logps/rejected": -0.9966603517532349, "loss": 1.6799, "rewards/accuracies": 0.643750011920929, "rewards/chosen": -0.9299852252006531, "rewards/margins": 0.06667519360780716, "rewards/rejected": -0.9966603517532349, "step": 165 }, { "epoch": 0.3643670462156731, "grad_norm": 13.123351141683822, "learning_rate": 1.6040415569136586e-05, "logits/chosen": -2.3123598098754883, "logits/rejected": -2.3209071159362793, "logps/chosen": -1.2778890132904053, "logps/rejected": -1.3249436616897583, "loss": 1.7418, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": -1.2778890132904053, "rewards/margins": 0.04705450311303139, "rewards/rejected": -1.3249436616897583, "step": 170 }, { "epoch": 0.3750837240455459, "grad_norm": 7.9321661768835865, "learning_rate": 1.5737470313111134e-05, "logits/chosen": -2.1733322143554688, "logits/rejected": -2.188865900039673, "logps/chosen": -1.1108664274215698, "logps/rejected": -1.1735206842422485, "loss": 1.6869, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": -1.1108664274215698, "rewards/margins": 0.06265412271022797, "rewards/rejected": -1.1735206842422485, "step": 175 }, { "epoch": 0.38580040187541864, "grad_norm": 8.557996815044758, "learning_rate": 1.5426462342349736e-05, "logits/chosen": -2.3044323921203613, "logits/rejected": -2.3116490840911865, "logps/chosen": -1.1034280061721802, "logps/rejected": -1.1603246927261353, "loss": 1.6567, "rewards/accuracies": 0.518750011920929, "rewards/chosen": -1.1034280061721802, "rewards/margins": 0.05689678341150284, "rewards/rejected": -1.1603246927261353, "step": 180 }, { "epoch": 0.3965170797052914, "grad_norm": 10.818770194479542, "learning_rate": 1.5107828708098327e-05, "logits/chosen": -2.3942196369171143, "logits/rejected": -2.4129745960235596, "logps/chosen": -1.0460340976715088, "logps/rejected": -1.105322241783142, "loss": 1.6177, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -1.0460340976715088, "rewards/margins": 0.059288132935762405, "rewards/rejected": -1.105322241783142, "step": 185 }, { "epoch": 0.4072337575351641, "grad_norm": 15.512656922129596, "learning_rate": 1.4782017177745322e-05, "logits/chosen": -2.4357337951660156, "logits/rejected": -2.458134889602661, "logps/chosen": -1.0308730602264404, "logps/rejected": -1.2367624044418335, "loss": 1.6917, "rewards/accuracies": 0.65625, "rewards/chosen": -1.0308730602264404, "rewards/margins": 0.20588929951190948, "rewards/rejected": -1.2367624044418335, "step": 190 }, { "epoch": 0.41795043536503684, "grad_norm": 10.838461128892558, "learning_rate": 1.444948560558601e-05, "logits/chosen": -2.4340341091156006, "logits/rejected": -2.4456231594085693, "logps/chosen": -1.2891473770141602, "logps/rejected": -1.3041666746139526, "loss": 1.7583, "rewards/accuracies": 0.5062500238418579, "rewards/chosen": -1.2891473770141602, "rewards/margins": 0.015019327402114868, "rewards/rejected": -1.3041666746139526, "step": 195 }, { "epoch": 0.4286671131949096, "grad_norm": 8.28412245267914, "learning_rate": 1.4110701289412045e-05, "logits/chosen": -2.515929937362671, "logits/rejected": -2.5307044982910156, "logps/chosen": -1.1040905714035034, "logps/rejected": -1.1781259775161743, "loss": 1.6734, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": -1.1040905714035034, "rewards/margins": 0.07403533160686493, "rewards/rejected": -1.1781259775161743, "step": 200 }, { "epoch": 0.4393837910247823, "grad_norm": 9.96354155003577, "learning_rate": 1.3766140313830279e-05, "logits/chosen": -2.531589984893799, "logits/rejected": -2.520904302597046, "logps/chosen": -1.0145483016967773, "logps/rejected": -1.0664831399917603, "loss": 1.6419, "rewards/accuracies": 0.5874999761581421, "rewards/chosen": -1.0145483016967773, "rewards/margins": 0.05193489044904709, "rewards/rejected": -1.0664831399917603, "step": 205 }, { "epoch": 0.45010046885465504, "grad_norm": 7.434219003968733, "learning_rate": 1.341628688123369e-05, "logits/chosen": -2.517782211303711, "logits/rejected": -2.5262608528137207, "logps/chosen": -1.0025173425674438, "logps/rejected": -1.0794353485107422, "loss": 1.606, "rewards/accuracies": 0.5874999761581421, "rewards/chosen": -1.0025173425674438, "rewards/margins": 0.07691796869039536, "rewards/rejected": -1.0794353485107422, "step": 210 }, { "epoch": 0.4608171466845278, "grad_norm": 7.337174471827815, "learning_rate": 1.3061632631364613e-05, "logits/chosen": -2.5459511280059814, "logits/rejected": -2.546247959136963, "logps/chosen": -0.9931095242500305, "logps/rejected": -1.094226598739624, "loss": 1.594, "rewards/accuracies": 0.5687500238418579, "rewards/chosen": -0.9931095242500305, "rewards/margins": 0.10111699998378754, "rewards/rejected": -1.094226598739624, "step": 215 }, { "epoch": 0.4715338245144005, "grad_norm": 8.701123471362035, "learning_rate": 1.2702675950426477e-05, "logits/chosen": -2.7032644748687744, "logits/rejected": -2.685612440109253, "logps/chosen": -0.9561999440193176, "logps/rejected": -1.0159614086151123, "loss": 1.63, "rewards/accuracies": 0.543749988079071, "rewards/chosen": -0.9561999440193176, "rewards/margins": 0.05976147577166557, "rewards/rejected": -1.0159614086151123, "step": 220 }, { "epoch": 0.4822505023442733, "grad_norm": 10.855107466518529, "learning_rate": 1.2339921270714873e-05, "logits/chosen": -2.7168991565704346, "logits/rejected": -2.742966651916504, "logps/chosen": -0.9448912739753723, "logps/rejected": -1.0768978595733643, "loss": 1.5893, "rewards/accuracies": 0.59375, "rewards/chosen": -0.9448912739753723, "rewards/margins": 0.1320066601037979, "rewards/rejected": -1.0768978595733643, "step": 225 }, { "epoch": 0.49296718017414604, "grad_norm": 7.921384206654596, "learning_rate": 1.1973878361752263e-05, "logits/chosen": -2.7181243896484375, "logits/rejected": -2.7184972763061523, "logps/chosen": -0.9301810264587402, "logps/rejected": -1.000950813293457, "loss": 1.5781, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": -0.9301810264587402, "rewards/margins": 0.07076982408761978, "rewards/rejected": -1.000950813293457, "step": 230 }, { "epoch": 0.5036838580040187, "grad_norm": 8.000512303019988, "learning_rate": 1.160506161392239e-05, "logits/chosen": -2.7740559577941895, "logits/rejected": -2.767812967300415, "logps/chosen": -0.906406044960022, "logps/rejected": -1.006753921508789, "loss": 1.5802, "rewards/accuracies": 0.59375, "rewards/chosen": -0.906406044960022, "rewards/margins": 0.10034787654876709, "rewards/rejected": -1.006753921508789, "step": 235 }, { "epoch": 0.5144005358338914, "grad_norm": 9.398458173467265, "learning_rate": 1.1233989315611132e-05, "logits/chosen": -2.792450428009033, "logits/rejected": -2.7874622344970703, "logps/chosen": -0.9394813776016235, "logps/rejected": -1.022139549255371, "loss": 1.587, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -0.9394813776016235, "rewards/margins": 0.08265805244445801, "rewards/rejected": -1.022139549255371, "step": 240 }, { "epoch": 0.5251172136637642, "grad_norm": 10.738263842024553, "learning_rate": 1.0861182924869584e-05, "logits/chosen": -2.721609115600586, "logits/rejected": -2.712308406829834, "logps/chosen": -0.9511491656303406, "logps/rejected": -1.0187407732009888, "loss": 1.5773, "rewards/accuracies": 0.59375, "rewards/chosen": -0.9511491656303406, "rewards/margins": 0.06759153306484222, "rewards/rejected": -1.0187407732009888, "step": 245 }, { "epoch": 0.535833891493637, "grad_norm": 8.609658733354994, "learning_rate": 1.0487166336622903e-05, "logits/chosen": -2.714517116546631, "logits/rejected": -2.7152135372161865, "logps/chosen": -0.9705566167831421, "logps/rejected": -1.0084294080734253, "loss": 1.5971, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -0.9705566167831421, "rewards/margins": 0.03787278011441231, "rewards/rejected": -1.0084294080734253, "step": 250 }, { "epoch": 0.5465505693235098, "grad_norm": 8.407135051661365, "learning_rate": 1.0112465146454681e-05, "logits/chosen": -2.790414333343506, "logits/rejected": -2.799837112426758, "logps/chosen": -0.8985916376113892, "logps/rejected": -0.9925104379653931, "loss": 1.5696, "rewards/accuracies": 0.6312500238418579, "rewards/chosen": -0.8985916376113892, "rewards/margins": 0.09391861408948898, "rewards/rejected": -0.9925104379653931, "step": 255 }, { "epoch": 0.5572672471533825, "grad_norm": 8.166210902327252, "learning_rate": 9.737605912001413e-06, "logits/chosen": -2.7387256622314453, "logits/rejected": -2.7400269508361816, "logps/chosen": -0.9064778089523315, "logps/rejected": -1.0061310529708862, "loss": 1.5824, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -0.9064778089523315, "rewards/margins": 0.09965313225984573, "rewards/rejected": -1.0061310529708862, "step": 260 }, { "epoch": 0.5679839249832552, "grad_norm": 7.963553030396463, "learning_rate": 9.363115412995038e-06, "logits/chosen": -2.733085870742798, "logits/rejected": -2.7209997177124023, "logps/chosen": -0.9366081357002258, "logps/rejected": -1.074143648147583, "loss": 1.5601, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -0.9366081357002258, "rewards/margins": 0.13753557205200195, "rewards/rejected": -1.074143648147583, "step": 265 }, { "epoch": 0.578700602813128, "grad_norm": 8.394463914962087, "learning_rate": 8.989519910993357e-06, "logits/chosen": -2.741532802581787, "logits/rejected": -2.7389678955078125, "logps/chosen": -0.8945878744125366, "logps/rejected": -0.9697369337081909, "loss": 1.5652, "rewards/accuracies": 0.574999988079071, "rewards/chosen": -0.8945878744125366, "rewards/margins": 0.07514907419681549, "rewards/rejected": -0.9697369337081909, "step": 270 }, { "epoch": 0.5894172806430007, "grad_norm": 7.2839753266908875, "learning_rate": 8.617344409838631e-06, "logits/chosen": -2.837791919708252, "logits/rejected": -2.8426520824432373, "logps/chosen": -0.8537940979003906, "logps/rejected": -0.9409109354019165, "loss": 1.5685, "rewards/accuracies": 0.543749988079071, "rewards/chosen": -0.8537940979003906, "rewards/margins": 0.08711685240268707, "rewards/rejected": -0.9409109354019165, "step": 275 }, { "epoch": 0.6001339584728734, "grad_norm": 7.994187957031587, "learning_rate": 8.247111917883634e-06, "logits/chosen": -2.6431148052215576, "logits/rejected": -2.644808530807495, "logps/chosen": -0.8992069959640503, "logps/rejected": -1.0336244106292725, "loss": 1.5658, "rewards/accuracies": 0.59375, "rewards/chosen": -0.8992069959640503, "rewards/margins": 0.1344173550605774, "rewards/rejected": -1.0336244106292725, "step": 280 }, { "epoch": 0.6108506363027462, "grad_norm": 10.3254576486645, "learning_rate": 7.879342713021875e-06, "logits/chosen": -2.745375156402588, "logits/rejected": -2.744263172149658, "logps/chosen": -0.9114044308662415, "logps/rejected": -0.9392461776733398, "loss": 1.5753, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": -0.9114044308662415, "rewards/margins": 0.027841681614518166, "rewards/rejected": -0.9392461776733398, "step": 285 }, { "epoch": 0.6215673141326189, "grad_norm": 14.256488881900221, "learning_rate": 7.514553611554909e-06, "logits/chosen": -2.9054572582244873, "logits/rejected": -2.915325880050659, "logps/chosen": -0.8290088772773743, "logps/rejected": -0.9459854364395142, "loss": 1.5614, "rewards/accuracies": 0.543749988079071, "rewards/chosen": -0.8290088772773743, "rewards/margins": 0.11697645485401154, "rewards/rejected": -0.9459854364395142, "step": 290 }, { "epoch": 0.6322839919624916, "grad_norm": 8.049072856223585, "learning_rate": 7.15325724192409e-06, "logits/chosen": -2.861290454864502, "logits/rejected": -2.8629579544067383, "logps/chosen": -0.8957312703132629, "logps/rejected": -1.0799039602279663, "loss": 1.5436, "rewards/accuracies": 0.6312500238418579, "rewards/chosen": -0.8957312703132629, "rewards/margins": 0.1841726452112198, "rewards/rejected": -1.0799039602279663, "step": 295 }, { "epoch": 0.6430006697923644, "grad_norm": 8.353215812366958, "learning_rate": 6.79596132432742e-06, "logits/chosen": -2.8521814346313477, "logits/rejected": -2.8686671257019043, "logps/chosen": -0.8559802770614624, "logps/rejected": -0.9652966260910034, "loss": 1.5273, "rewards/accuracies": 0.6312500238418579, "rewards/chosen": -0.8559802770614624, "rewards/margins": 0.10931621491909027, "rewards/rejected": -0.9652966260910034, "step": 300 }, { "epoch": 0.6537173476222371, "grad_norm": 8.558746762469774, "learning_rate": 6.443167957233864e-06, "logits/chosen": -2.8060519695281982, "logits/rejected": -2.793569326400757, "logps/chosen": -0.8676208257675171, "logps/rejected": -0.9558470845222473, "loss": 1.5333, "rewards/accuracies": 0.637499988079071, "rewards/chosen": -0.8676208257675171, "rewards/margins": 0.08822615444660187, "rewards/rejected": -0.9558470845222473, "step": 305 }, { "epoch": 0.6644340254521098, "grad_norm": 8.944155497280653, "learning_rate": 6.095372911797673e-06, "logits/chosen": -2.8632490634918213, "logits/rejected": -2.859429359436035, "logps/chosen": -0.8489475250244141, "logps/rejected": -0.9311763644218445, "loss": 1.5397, "rewards/accuracies": 0.625, "rewards/chosen": -0.8489475250244141, "rewards/margins": 0.08222876489162445, "rewards/rejected": -0.9311763644218445, "step": 310 }, { "epoch": 0.6751507032819826, "grad_norm": 7.822389173887281, "learning_rate": 5.753064935164348e-06, "logits/chosen": -2.7866902351379395, "logits/rejected": -2.792357921600342, "logps/chosen": -0.84296715259552, "logps/rejected": -0.959184467792511, "loss": 1.5381, "rewards/accuracies": 0.6625000238418579, "rewards/chosen": -0.84296715259552, "rewards/margins": 0.11621735244989395, "rewards/rejected": -0.959184467792511, "step": 315 }, { "epoch": 0.6858673811118553, "grad_norm": 9.32294397022148, "learning_rate": 5.4167250636472426e-06, "logits/chosen": -2.7923617362976074, "logits/rejected": -2.7717807292938232, "logps/chosen": -0.8463284373283386, "logps/rejected": -0.9291221499443054, "loss": 1.5458, "rewards/accuracies": 0.625, "rewards/chosen": -0.8463284373283386, "rewards/margins": 0.08279372751712799, "rewards/rejected": -0.9291221499443054, "step": 320 }, { "epoch": 0.696584058941728, "grad_norm": 8.808118982464391, "learning_rate": 5.08682594674002e-06, "logits/chosen": -2.7373509407043457, "logits/rejected": -2.7527077198028564, "logps/chosen": -0.8056324124336243, "logps/rejected": -0.957785964012146, "loss": 1.5196, "rewards/accuracies": 0.643750011920929, "rewards/chosen": -0.8056324124336243, "rewards/margins": 0.15215365588665009, "rewards/rejected": -0.957785964012146, "step": 325 }, { "epoch": 0.7073007367716008, "grad_norm": 8.169267761293227, "learning_rate": 4.763831182914848e-06, "logits/chosen": -2.7620842456817627, "logits/rejected": -2.77044415473938, "logps/chosen": -0.8720673322677612, "logps/rejected": -0.9474166631698608, "loss": 1.5373, "rewards/accuracies": 0.5562499761581421, "rewards/chosen": -0.8720673322677612, "rewards/margins": 0.07534926384687424, "rewards/rejected": -0.9474166631698608, "step": 330 }, { "epoch": 0.7180174146014735, "grad_norm": 8.7345666789259, "learning_rate": 4.448194668139777e-06, "logits/chosen": -2.794130325317383, "logits/rejected": -2.7945706844329834, "logps/chosen": -0.8377755284309387, "logps/rejected": -0.9672821760177612, "loss": 1.5233, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -0.8377755284309387, "rewards/margins": 0.12950672209262848, "rewards/rejected": -0.9672821760177612, "step": 335 }, { "epoch": 0.7287340924313462, "grad_norm": 8.389766148627231, "learning_rate": 4.1403599580307795e-06, "logits/chosen": -2.714247226715088, "logits/rejected": -2.7236764430999756, "logps/chosen": -0.8354495167732239, "logps/rejected": -0.9634197354316711, "loss": 1.5335, "rewards/accuracies": 0.6812499761581421, "rewards/chosen": -0.8354495167732239, "rewards/margins": 0.1279701441526413, "rewards/rejected": -0.9634197354316711, "step": 340 }, { "epoch": 0.739450770261219, "grad_norm": 9.147040621292957, "learning_rate": 3.840759644534828e-06, "logits/chosen": -2.790487766265869, "logits/rejected": -2.8023035526275635, "logps/chosen": -0.8659473657608032, "logps/rejected": -0.9856260418891907, "loss": 1.5348, "rewards/accuracies": 0.606249988079071, "rewards/chosen": -0.8659473657608032, "rewards/margins": 0.11967878043651581, "rewards/rejected": -0.9856260418891907, "step": 345 }, { "epoch": 0.7501674480910918, "grad_norm": 8.599069402922064, "learning_rate": 3.549814748019906e-06, "logits/chosen": -2.815507411956787, "logits/rejected": -2.824169158935547, "logps/chosen": -0.808633029460907, "logps/rejected": -0.9542997479438782, "loss": 1.529, "rewards/accuracies": 0.6187499761581421, "rewards/chosen": -0.808633029460907, "rewards/margins": 0.14566662907600403, "rewards/rejected": -0.9542997479438782, "step": 350 }, { "epoch": 0.7608841259209645, "grad_norm": 7.982319756803084, "learning_rate": 3.2679341256262653e-06, "logits/chosen": -2.8472940921783447, "logits/rejected": -2.8238396644592285, "logps/chosen": -0.9044294357299805, "logps/rejected": -0.9391825795173645, "loss": 1.5418, "rewards/accuracies": 0.5687500238418579, "rewards/chosen": -0.9044294357299805, "rewards/margins": 0.03475295007228851, "rewards/rejected": -0.9391825795173645, "step": 355 }, { "epoch": 0.7716008037508373, "grad_norm": 11.32319755235924, "learning_rate": 2.995513896710327e-06, "logits/chosen": -2.8410215377807617, "logits/rejected": -2.83413028717041, "logps/chosen": -0.7960796356201172, "logps/rejected": -0.9083698391914368, "loss": 1.5306, "rewards/accuracies": 0.668749988079071, "rewards/chosen": -0.7960796356201172, "rewards/margins": 0.1122901439666748, "rewards/rejected": -0.9083698391914368, "step": 360 }, { "epoch": 0.78231748158071, "grad_norm": 7.8205965789155325, "learning_rate": 2.7329368861886862e-06, "logits/chosen": -2.8469526767730713, "logits/rejected": -2.8535444736480713, "logps/chosen": -0.8426831960678101, "logps/rejected": -0.9201619029045105, "loss": 1.5523, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -0.8426831960678101, "rewards/margins": 0.07747860997915268, "rewards/rejected": -0.9201619029045105, "step": 365 }, { "epoch": 0.7930341594105828, "grad_norm": 11.791063151570222, "learning_rate": 2.4805720865643943e-06, "logits/chosen": -2.8243472576141357, "logits/rejected": -2.8304176330566406, "logps/chosen": -0.8739852905273438, "logps/rejected": -0.9622358083724976, "loss": 1.5445, "rewards/accuracies": 0.543749988079071, "rewards/chosen": -0.8739852905273438, "rewards/margins": 0.08825065940618515, "rewards/rejected": -0.9622358083724976, "step": 370 }, { "epoch": 0.8037508372404555, "grad_norm": 10.592008554970972, "learning_rate": 2.2387741393916117e-06, "logits/chosen": -2.7737154960632324, "logits/rejected": -2.7636284828186035, "logps/chosen": -0.9277487993240356, "logps/rejected": -1.0097343921661377, "loss": 1.5368, "rewards/accuracies": 0.59375, "rewards/chosen": -0.9277487993240356, "rewards/margins": 0.08198549598455429, "rewards/rejected": -1.0097343921661377, "step": 375 }, { "epoch": 0.8144675150703282, "grad_norm": 8.911271573494423, "learning_rate": 2.0078828369072323e-06, "logits/chosen": -2.7896554470062256, "logits/rejected": -2.770059823989868, "logps/chosen": -0.8297151327133179, "logps/rejected": -0.9309379458427429, "loss": 1.531, "rewards/accuracies": 0.5625, "rewards/chosen": -0.8297151327133179, "rewards/margins": 0.10122283548116684, "rewards/rejected": -0.9309379458427429, "step": 380 }, { "epoch": 0.825184192900201, "grad_norm": 10.364577364972547, "learning_rate": 1.7882226445298656e-06, "logits/chosen": -2.740999460220337, "logits/rejected": -2.733097791671753, "logps/chosen": -0.8539925813674927, "logps/rejected": -0.9543533325195312, "loss": 1.5174, "rewards/accuracies": 0.637499988079071, "rewards/chosen": -0.8539925813674927, "rewards/margins": 0.10036077350378036, "rewards/rejected": -0.9543533325195312, "step": 385 }, { "epoch": 0.8359008707300737, "grad_norm": 10.862441952775951, "learning_rate": 1.580102244897217e-06, "logits/chosen": -2.7668397426605225, "logits/rejected": -2.7470200061798096, "logps/chosen": -0.8713840246200562, "logps/rejected": -0.940260112285614, "loss": 1.5632, "rewards/accuracies": 0.550000011920929, "rewards/chosen": -0.8713840246200562, "rewards/margins": 0.0688760057091713, "rewards/rejected": -0.940260112285614, "step": 390 }, { "epoch": 0.8466175485599464, "grad_norm": 7.998296949983092, "learning_rate": 1.3838141040825536e-06, "logits/chosen": -2.85998272895813, "logits/rejected": -2.8610522747039795, "logps/chosen": -0.8187233805656433, "logps/rejected": -0.8954393267631531, "loss": 1.5253, "rewards/accuracies": 0.606249988079071, "rewards/chosen": -0.8187233805656433, "rewards/margins": 0.0767158791422844, "rewards/rejected": -0.8954393267631531, "step": 395 }, { "epoch": 0.8573342263898192, "grad_norm": 10.009174586154066, "learning_rate": 1.1996340605999058e-06, "logits/chosen": -2.842839479446411, "logits/rejected": -2.819075107574463, "logps/chosen": -0.7899017333984375, "logps/rejected": -0.9768079519271851, "loss": 1.5218, "rewards/accuracies": 0.6812499761581421, "rewards/chosen": -0.7899017333984375, "rewards/margins": 0.18690629303455353, "rewards/rejected": -0.9768079519271851, "step": 400 }, { "epoch": 0.8573342263898192, "eval_logits/chosen": -2.840418815612793, "eval_logits/rejected": -2.8387703895568848, "eval_logps/chosen": -0.8695030808448792, "eval_logps/rejected": -0.9116575121879578, "eval_loss": 1.5846024751663208, "eval_rewards/accuracies": 0.5558510422706604, "eval_rewards/chosen": -0.8695030808448792, "eval_rewards/margins": 0.0421544648706913, "eval_rewards/rejected": -0.9116575121879578, "eval_runtime": 432.5597, "eval_samples_per_second": 6.922, "eval_steps_per_second": 0.435, "step": 400 }, { "epoch": 0.8680509042196919, "grad_norm": 8.778213042419416, "learning_rate": 1.0278209377755099e-06, "logits/chosen": -2.829361915588379, "logits/rejected": -2.8452582359313965, "logps/chosen": -0.8199126124382019, "logps/rejected": -0.9467391967773438, "loss": 1.5156, "rewards/accuracies": 0.65625, "rewards/chosen": -0.8199126124382019, "rewards/margins": 0.12682652473449707, "rewards/rejected": -0.9467391967773438, "step": 405 }, { "epoch": 0.8787675820495646, "grad_norm": 8.779487119016105, "learning_rate": 8.68616180030275e-07, "logits/chosen": -2.865104913711548, "logits/rejected": -2.867384672164917, "logps/chosen": -0.8341816067695618, "logps/rejected": -0.8998939394950867, "loss": 1.5114, "rewards/accuracies": 0.5687500238418579, "rewards/chosen": -0.8341816067695618, "rewards/margins": 0.06571228057146072, "rewards/rejected": -0.8998939394950867, "step": 410 }, { "epoch": 0.8894842598794374, "grad_norm": 8.92068989780619, "learning_rate": 7.22243513584342e-07, "logits/chosen": -2.77596378326416, "logits/rejected": -2.7669265270233154, "logps/chosen": -0.8306072354316711, "logps/rejected": -0.9651764035224915, "loss": 1.5034, "rewards/accuracies": 0.637499988079071, "rewards/chosen": -0.8306072354316711, "rewards/margins": 0.13456913828849792, "rewards/rejected": -0.9651764035224915, "step": 415 }, { "epoch": 0.9002009377093101, "grad_norm": 9.219885917035732, "learning_rate": 5.889086320605575e-07, "logits/chosen": -2.7921142578125, "logits/rejected": -2.7921335697174072, "logps/chosen": -0.7846695184707642, "logps/rejected": -0.8718813061714172, "loss": 1.5351, "rewards/accuracies": 0.53125, "rewards/chosen": -0.7846695184707642, "rewards/margins": 0.08721177279949188, "rewards/rejected": -0.8718813061714172, "step": 420 }, { "epoch": 0.9109176155391828, "grad_norm": 9.448964270757202, "learning_rate": 4.6879890742869584e-07, "logits/chosen": -2.7913196086883545, "logits/rejected": -2.789302110671997, "logps/chosen": -0.7865716218948364, "logps/rejected": -0.9528288841247559, "loss": 1.5108, "rewards/accuracies": 0.7124999761581421, "rewards/chosen": -0.7865716218948364, "rewards/margins": 0.16625720262527466, "rewards/rejected": -0.9528288841247559, "step": 425 }, { "epoch": 0.9216342933690556, "grad_norm": 10.369844004542797, "learning_rate": 3.6208312669659405e-07, "logits/chosen": -2.817171812057495, "logits/rejected": -2.8241400718688965, "logps/chosen": -0.8919852375984192, "logps/rejected": -1.037209153175354, "loss": 1.5259, "rewards/accuracies": 0.6187499761581421, "rewards/chosen": -0.8919852375984192, "rewards/margins": 0.14522376656532288, "rewards/rejected": -1.037209153175354, "step": 430 }, { "epoch": 0.9323509711989283, "grad_norm": 12.055758939525129, "learning_rate": 2.689112547182593e-07, "logits/chosen": -2.890130043029785, "logits/rejected": -2.882509708404541, "logps/chosen": -0.7903082370758057, "logps/rejected": -0.9467129707336426, "loss": 1.5203, "rewards/accuracies": 0.643750011920929, "rewards/chosen": -0.7903082370758057, "rewards/margins": 0.15640470385551453, "rewards/rejected": -0.9467129707336426, "step": 435 }, { "epoch": 0.943067649028801, "grad_norm": 9.50477372459136, "learning_rate": 1.8941422345223448e-07, "logits/chosen": -2.7733447551727295, "logits/rejected": -2.77803635597229, "logps/chosen": -0.7930269241333008, "logps/rejected": -0.9639067649841309, "loss": 1.4896, "rewards/accuracies": 0.6625000238418579, "rewards/chosen": -0.7930269241333008, "rewards/margins": 0.17087996006011963, "rewards/rejected": -0.9639067649841309, "step": 440 }, { "epoch": 0.9537843268586738, "grad_norm": 10.00023586076567, "learning_rate": 1.2370374796639383e-07, "logits/chosen": -2.794691324234009, "logits/rejected": -2.799956798553467, "logps/chosen": -0.8402734994888306, "logps/rejected": -0.9617452621459961, "loss": 1.5232, "rewards/accuracies": 0.6312500238418579, "rewards/chosen": -0.8402734994888306, "rewards/margins": 0.12147176265716553, "rewards/rejected": -0.9617452621459961, "step": 445 }, { "epoch": 0.9645010046885466, "grad_norm": 9.634395288787493, "learning_rate": 7.18721694477209e-08, "logits/chosen": -2.8351221084594727, "logits/rejected": -2.833106756210327, "logps/chosen": -0.8035479784011841, "logps/rejected": -0.9219916462898254, "loss": 1.5027, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -0.8035479784011841, "rewards/margins": 0.11844360828399658, "rewards/rejected": -0.9219916462898254, "step": 450 }, { "epoch": 0.9752176825184193, "grad_norm": 9.629343705982949, "learning_rate": 3.399232543769504e-08, "logits/chosen": -2.807852268218994, "logits/rejected": -2.815563678741455, "logps/chosen": -0.8062378168106079, "logps/rejected": -0.9637410044670105, "loss": 1.5073, "rewards/accuracies": 0.606249988079071, "rewards/chosen": -0.8062378168106079, "rewards/margins": 0.15750321745872498, "rewards/rejected": -0.9637410044670105, "step": 455 }, { "epoch": 0.9859343603482921, "grad_norm": 10.887896034397038, "learning_rate": 1.011744747562271e-08, "logits/chosen": -2.826111316680908, "logits/rejected": -2.830981731414795, "logps/chosen": -0.8098823428153992, "logps/rejected": -0.9950210452079773, "loss": 1.5206, "rewards/accuracies": 0.612500011920929, "rewards/chosen": -0.8098823428153992, "rewards/margins": 0.1851385533809662, "rewards/rejected": -0.9950210452079773, "step": 460 }, { "epoch": 0.9966510381781648, "grad_norm": 10.914112703731178, "learning_rate": 2.810862937696612e-10, "logits/chosen": -2.813309907913208, "logits/rejected": -2.8205251693725586, "logps/chosen": -0.8651328086853027, "logps/rejected": -0.9807795286178589, "loss": 1.5128, "rewards/accuracies": 0.637499988079071, "rewards/chosen": -0.8651328086853027, "rewards/margins": 0.11564669758081436, "rewards/rejected": -0.9807795286178589, "step": 465 }, { "epoch": 0.9987943737441393, "step": 466, "total_flos": 0.0, "train_loss": 1.5861326811651304, "train_runtime": 19371.3426, "train_samples_per_second": 3.083, "train_steps_per_second": 0.024 } ], "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 }