{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.998766954377312, "eval_steps": 1000, "global_step": 405, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0, "learning_rate": 1.2195121951219512e-08, "logits/chosen": -2.8088459968566895, "logits/rejected": -2.7595884799957275, "logps/chosen": -368.90777587890625, "logps/rejected": -133.10202026367188, "loss": 0.4545, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 1 }, { "epoch": 0.02, "learning_rate": 1.219512195121951e-07, "logits/chosen": -2.83878231048584, "logits/rejected": -2.824958562850952, "logps/chosen": -433.8194580078125, "logps/rejected": -114.66372680664062, "loss": 0.4227, "rewards/accuracies": 0.5416666865348816, "rewards/chosen": 0.0010460919002071023, "rewards/margins": 0.0013696590904146433, "rewards/rejected": -0.000323567190207541, "step": 10 }, { "epoch": 0.05, "learning_rate": 2.439024390243902e-07, "logits/chosen": -2.7982840538024902, "logits/rejected": -2.7652382850646973, "logps/chosen": -436.67694091796875, "logps/rejected": -109.33970642089844, "loss": 0.4254, "rewards/accuracies": 0.7562500238418579, "rewards/chosen": 0.020646633580327034, "rewards/margins": 0.03670011833310127, "rewards/rejected": -0.01605348475277424, "step": 20 }, { "epoch": 0.07, "learning_rate": 3.6585365853658536e-07, "logits/chosen": -2.717103958129883, "logits/rejected": -2.6900384426116943, "logps/chosen": -422.26702880859375, "logps/rejected": -128.1683349609375, "loss": 0.4108, "rewards/accuracies": 0.7749999761581421, "rewards/chosen": 0.07094015926122665, "rewards/margins": 0.20011821389198303, "rewards/rejected": -0.12917804718017578, "step": 30 }, { "epoch": 0.1, "learning_rate": 4.878048780487804e-07, "logits/chosen": -2.590641498565674, "logits/rejected": -2.5721707344055176, "logps/chosen": -396.3973693847656, "logps/rejected": -139.94859313964844, "loss": 0.3881, "rewards/accuracies": 0.762499988079071, "rewards/chosen": 0.022975314408540726, "rewards/margins": 0.4287249445915222, "rewards/rejected": -0.4057496190071106, "step": 40 }, { "epoch": 0.12, "learning_rate": 4.992461696250783e-07, "logits/chosen": -2.42146635055542, "logits/rejected": -2.394202709197998, "logps/chosen": -445.91644287109375, "logps/rejected": -205.5404052734375, "loss": 0.316, "rewards/accuracies": 0.800000011920929, "rewards/chosen": -0.03603144362568855, "rewards/margins": 0.8689195513725281, "rewards/rejected": -0.9049509763717651, "step": 50 }, { "epoch": 0.15, "learning_rate": 4.966461721767899e-07, "logits/chosen": -2.417520046234131, "logits/rejected": -2.3663182258605957, "logps/chosen": -422.27215576171875, "logps/rejected": -255.75912475585938, "loss": 0.2661, "rewards/accuracies": 0.7437499761581421, "rewards/chosen": -0.35185474157333374, "rewards/margins": 0.9545990228652954, "rewards/rejected": -1.3064535856246948, "step": 60 }, { "epoch": 0.17, "learning_rate": 4.922100518015975e-07, "logits/chosen": -2.45034122467041, "logits/rejected": -2.397273540496826, "logps/chosen": -428.19207763671875, "logps/rejected": -294.82501220703125, "loss": 0.198, "rewards/accuracies": 0.7875000238418579, "rewards/chosen": -0.4432826638221741, "rewards/margins": 1.3181250095367432, "rewards/rejected": -1.7614076137542725, "step": 70 }, { "epoch": 0.2, "learning_rate": 4.859708325770919e-07, "logits/chosen": -2.3751111030578613, "logits/rejected": -2.321465015411377, "logps/chosen": -468.4130859375, "logps/rejected": -331.666259765625, "loss": 0.162, "rewards/accuracies": 0.793749988079071, "rewards/chosen": -0.7735603451728821, "rewards/margins": 1.4863694906234741, "rewards/rejected": -2.25993013381958, "step": 80 }, { "epoch": 0.22, "learning_rate": 4.779749614980225e-07, "logits/chosen": -2.381338596343994, "logits/rejected": -2.327340602874756, "logps/chosen": -520.724365234375, "logps/rejected": -380.0218811035156, "loss": 0.1464, "rewards/accuracies": 0.8812500238418579, "rewards/chosen": -0.4466208815574646, "rewards/margins": 2.054797649383545, "rewards/rejected": -2.5014188289642334, "step": 90 }, { "epoch": 0.25, "learning_rate": 4.682819627081427e-07, "logits/chosen": -2.3299832344055176, "logits/rejected": -2.2486767768859863, "logps/chosen": -477.24261474609375, "logps/rejected": -372.49017333984375, "loss": 0.1456, "rewards/accuracies": 0.862500011920929, "rewards/chosen": -0.6445478200912476, "rewards/margins": 1.895777702331543, "rewards/rejected": -2.54032564163208, "step": 100 }, { "epoch": 0.27, "learning_rate": 4.569639943810477e-07, "logits/chosen": -2.3097102642059326, "logits/rejected": -2.226323127746582, "logps/chosen": -495.50469970703125, "logps/rejected": -389.80078125, "loss": 0.1283, "rewards/accuracies": 0.78125, "rewards/chosen": -0.8834150433540344, "rewards/margins": 1.8450326919555664, "rewards/rejected": -2.728447675704956, "step": 110 }, { "epoch": 0.3, "learning_rate": 4.4410531154874543e-07, "logits/chosen": -2.3541078567504883, "logits/rejected": -2.2549960613250732, "logps/chosen": -524.7901000976562, "logps/rejected": -398.75775146484375, "loss": 0.1283, "rewards/accuracies": 0.78125, "rewards/chosen": -0.7295175790786743, "rewards/margins": 1.9527451992034912, "rewards/rejected": -2.682262897491455, "step": 120 }, { "epoch": 0.32, "learning_rate": 4.298016388768561e-07, "logits/chosen": -2.3804497718811035, "logits/rejected": -2.2821872234344482, "logps/chosen": -518.573974609375, "logps/rejected": -398.14306640625, "loss": 0.114, "rewards/accuracies": 0.84375, "rewards/chosen": -0.5337150692939758, "rewards/margins": 2.2159152030944824, "rewards/rejected": -2.7496302127838135, "step": 130 }, { "epoch": 0.35, "learning_rate": 4.1415945805573005e-07, "logits/chosen": -2.309293270111084, "logits/rejected": -2.2271227836608887, "logps/chosen": -486.838623046875, "logps/rejected": -373.0490417480469, "loss": 0.1246, "rewards/accuracies": 0.862500011920929, "rewards/chosen": -0.6586702466011047, "rewards/margins": 1.7459022998809814, "rewards/rejected": -2.4045722484588623, "step": 140 }, { "epoch": 0.37, "learning_rate": 3.972952151123984e-07, "logits/chosen": -2.288892984390259, "logits/rejected": -2.1915061473846436, "logps/chosen": -450.01556396484375, "logps/rejected": -368.2213439941406, "loss": 0.1152, "rewards/accuracies": 0.856249988079071, "rewards/chosen": -0.7529748678207397, "rewards/margins": 1.9199845790863037, "rewards/rejected": -2.672959566116333, "step": 150 }, { "epoch": 0.39, "learning_rate": 3.793344535444142e-07, "logits/chosen": -2.2575857639312744, "logits/rejected": -2.1550350189208984, "logps/chosen": -547.2183837890625, "logps/rejected": -409.57989501953125, "loss": 0.088, "rewards/accuracies": 0.8125, "rewards/chosen": -0.8003584146499634, "rewards/margins": 2.1838386058807373, "rewards/rejected": -2.9841971397399902, "step": 160 }, { "epoch": 0.42, "learning_rate": 3.604108797288461e-07, "logits/chosen": -2.2742323875427246, "logits/rejected": -2.167198419570923, "logps/chosen": -547.2274169921875, "logps/rejected": -456.614501953125, "loss": 0.0776, "rewards/accuracies": 0.862500011920929, "rewards/chosen": -1.0825190544128418, "rewards/margins": 2.3789236545562744, "rewards/rejected": -3.4614429473876953, "step": 170 }, { "epoch": 0.44, "learning_rate": 3.40665367563858e-07, "logits/chosen": -2.2402544021606445, "logits/rejected": -2.1346538066864014, "logps/chosen": -564.0145263671875, "logps/rejected": -489.21160888671875, "loss": 0.0697, "rewards/accuracies": 0.7749999761581421, "rewards/chosen": -1.539156198501587, "rewards/margins": 2.1975486278533936, "rewards/rejected": -3.7367050647735596, "step": 180 }, { "epoch": 0.47, "learning_rate": 3.202449097526798e-07, "logits/chosen": -2.3025131225585938, "logits/rejected": -2.224256992340088, "logps/chosen": -505.39520263671875, "logps/rejected": -423.83026123046875, "loss": 0.0811, "rewards/accuracies": 0.831250011920929, "rewards/chosen": -1.0325360298156738, "rewards/margins": 2.132319927215576, "rewards/rejected": -3.16485595703125, "step": 190 }, { "epoch": 0.49, "learning_rate": 2.993015235369905e-07, "logits/chosen": -2.3023552894592285, "logits/rejected": -2.2043874263763428, "logps/chosen": -525.6875610351562, "logps/rejected": -416.1629333496094, "loss": 0.0979, "rewards/accuracies": 0.856249988079071, "rewards/chosen": -0.8641435503959656, "rewards/margins": 2.118994951248169, "rewards/rejected": -2.9831383228302, "step": 200 }, { "epoch": 0.52, "learning_rate": 2.7799111902582693e-07, "logits/chosen": -2.3067820072174072, "logits/rejected": -2.2110161781311035, "logps/chosen": -492.69927978515625, "logps/rejected": -381.31878662109375, "loss": 0.0872, "rewards/accuracies": 0.793749988079071, "rewards/chosen": -0.9221334457397461, "rewards/margins": 1.870031714439392, "rewards/rejected": -2.7921650409698486, "step": 210 }, { "epoch": 0.54, "learning_rate": 2.564723385445869e-07, "logits/chosen": -2.3406167030334473, "logits/rejected": -2.2510488033294678, "logps/chosen": -520.8443603515625, "logps/rejected": -442.00732421875, "loss": 0.0908, "rewards/accuracies": 0.8062499761581421, "rewards/chosen": -1.0322405099868774, "rewards/margins": 2.144731044769287, "rewards/rejected": -3.176971912384033, "step": 220 }, { "epoch": 0.57, "learning_rate": 2.3490537564442845e-07, "logits/chosen": -2.284823179244995, "logits/rejected": -2.1653401851654053, "logps/chosen": -511.96929931640625, "logps/rejected": -426.46356201171875, "loss": 0.0967, "rewards/accuracies": 0.793749988079071, "rewards/chosen": -1.2002372741699219, "rewards/margins": 1.9990075826644897, "rewards/rejected": -3.199244976043701, "step": 230 }, { "epoch": 0.59, "learning_rate": 2.1345078256378801e-07, "logits/chosen": -2.321927547454834, "logits/rejected": -2.215357780456543, "logps/chosen": -495.8760681152344, "logps/rejected": -439.46282958984375, "loss": 0.0955, "rewards/accuracies": 0.862500011920929, "rewards/chosen": -0.8706371188163757, "rewards/margins": 2.3429722785949707, "rewards/rejected": -3.213609218597412, "step": 240 }, { "epoch": 0.62, "learning_rate": 1.9226827501969865e-07, "logits/chosen": -2.3428966999053955, "logits/rejected": -2.2573530673980713, "logps/chosen": -526.4675903320312, "logps/rejected": -451.949462890625, "loss": 0.096, "rewards/accuracies": 0.8374999761581421, "rewards/chosen": -0.8379364013671875, "rewards/margins": 2.499549627304077, "rewards/rejected": -3.3374857902526855, "step": 250 }, { "epoch": 0.64, "learning_rate": 1.715155432264775e-07, "logits/chosen": -2.3556008338928223, "logits/rejected": -2.2766494750976562, "logps/chosen": -516.3786010742188, "logps/rejected": -430.13916015625, "loss": 0.0857, "rewards/accuracies": 0.875, "rewards/chosen": -0.8434340357780457, "rewards/margins": 2.294442653656006, "rewards/rejected": -3.1378769874572754, "step": 260 }, { "epoch": 0.67, "learning_rate": 1.51347077992983e-07, "logits/chosen": -2.3460044860839844, "logits/rejected": -2.281031370162964, "logps/chosen": -490.55078125, "logps/rejected": -423.6560974121094, "loss": 0.0821, "rewards/accuracies": 0.856249988079071, "rewards/chosen": -0.8685197830200195, "rewards/margins": 2.1445822715759277, "rewards/rejected": -3.0131022930145264, "step": 270 }, { "epoch": 0.69, "learning_rate": 1.3191302063739906e-07, "logits/chosen": -2.2882773876190186, "logits/rejected": -2.218071699142456, "logps/chosen": -500.769287109375, "logps/rejected": -446.246826171875, "loss": 0.0712, "rewards/accuracies": 0.8187500238418579, "rewards/chosen": -1.2157343626022339, "rewards/margins": 2.1158077716827393, "rewards/rejected": -3.3315422534942627, "step": 280 }, { "epoch": 0.72, "learning_rate": 1.1335804528119475e-07, "logits/chosen": -2.3649039268493652, "logits/rejected": -2.252676486968994, "logps/chosen": -540.1212158203125, "logps/rejected": -467.2939453125, "loss": 0.0686, "rewards/accuracies": 0.8687499761581421, "rewards/chosen": -1.0436217784881592, "rewards/margins": 2.6221861839294434, "rewards/rejected": -3.6658082008361816, "step": 290 }, { "epoch": 0.74, "learning_rate": 9.582028184286423e-08, "logits/chosen": -2.243900775909424, "logits/rejected": -2.1746292114257812, "logps/chosen": -503.1402282714844, "logps/rejected": -486.1592712402344, "loss": 0.0686, "rewards/accuracies": 0.84375, "rewards/chosen": -1.4328491687774658, "rewards/margins": 2.2128751277923584, "rewards/rejected": -3.6457245349884033, "step": 300 }, { "epoch": 0.76, "learning_rate": 7.943028774907065e-08, "logits/chosen": -2.2528328895568848, "logits/rejected": -2.170386791229248, "logps/chosen": -501.7100524902344, "logps/rejected": -471.88897705078125, "loss": 0.0689, "rewards/accuracies": 0.8500000238418579, "rewards/chosen": -1.172387719154358, "rewards/margins": 2.3613522052764893, "rewards/rejected": -3.533740282058716, "step": 310 }, { "epoch": 0.79, "learning_rate": 6.431007601814637e-08, "logits/chosen": -2.258288860321045, "logits/rejected": -2.1915061473846436, "logps/chosen": -471.57330322265625, "logps/rejected": -461.84417724609375, "loss": 0.0601, "rewards/accuracies": 0.8187500238418579, "rewards/chosen": -1.4386770725250244, "rewards/margins": 2.1069023609161377, "rewards/rejected": -3.545579433441162, "step": 320 }, { "epoch": 0.81, "learning_rate": 5.0572206951246e-08, "logits/chosen": -2.2368595600128174, "logits/rejected": -2.1402342319488525, "logps/chosen": -522.8599853515625, "logps/rejected": -482.84893798828125, "loss": 0.0626, "rewards/accuracies": 0.78125, "rewards/chosen": -1.5531214475631714, "rewards/margins": 2.2123360633850098, "rewards/rejected": -3.7654571533203125, "step": 330 }, { "epoch": 0.84, "learning_rate": 3.831895019292897e-08, "logits/chosen": -2.308152675628662, "logits/rejected": -2.2120919227600098, "logps/chosen": -565.0369873046875, "logps/rejected": -535.8488159179688, "loss": 0.0642, "rewards/accuracies": 0.831250011920929, "rewards/chosen": -1.306236982345581, "rewards/margins": 2.8749289512634277, "rewards/rejected": -4.181166172027588, "step": 340 }, { "epoch": 0.86, "learning_rate": 2.764152339909756e-08, "logits/chosen": -2.245577573776245, "logits/rejected": -2.1435444355010986, "logps/chosen": -546.0943603515625, "logps/rejected": -454.5082092285156, "loss": 0.0636, "rewards/accuracies": 0.8500000238418579, "rewards/chosen": -1.2082496881484985, "rewards/margins": 2.3495194911956787, "rewards/rejected": -3.5577690601348877, "step": 350 }, { "epoch": 0.89, "learning_rate": 1.861941317991664e-08, "logits/chosen": -2.302865505218506, "logits/rejected": -2.1724164485931396, "logps/chosen": -559.6376953125, "logps/rejected": -483.40771484375, "loss": 0.0675, "rewards/accuracies": 0.875, "rewards/chosen": -1.0334274768829346, "rewards/margins": 2.655003070831299, "rewards/rejected": -3.6884307861328125, "step": 360 }, { "epoch": 0.91, "learning_rate": 1.13197833728636e-08, "logits/chosen": -2.2556536197662354, "logits/rejected": -2.153872013092041, "logps/chosen": -521.9984130859375, "logps/rejected": -505.71673583984375, "loss": 0.06, "rewards/accuracies": 0.8812500238418579, "rewards/chosen": -1.1932189464569092, "rewards/margins": 2.7444043159484863, "rewards/rejected": -3.9376235008239746, "step": 370 }, { "epoch": 0.94, "learning_rate": 5.79697505093521e-09, "logits/chosen": -2.2588906288146973, "logits/rejected": -2.159388303756714, "logps/chosen": -529.9054565429688, "logps/rejected": -461.11700439453125, "loss": 0.0744, "rewards/accuracies": 0.7749999761581421, "rewards/chosen": -1.276084065437317, "rewards/margins": 2.2816543579101562, "rewards/rejected": -3.5577385425567627, "step": 380 }, { "epoch": 0.96, "learning_rate": 2.092101988131256e-09, "logits/chosen": -2.313697099685669, "logits/rejected": -2.171175003051758, "logps/chosen": -565.225830078125, "logps/rejected": -489.6360778808594, "loss": 0.0609, "rewards/accuracies": 0.893750011920929, "rewards/chosen": -1.0208733081817627, "rewards/margins": 2.784264087677002, "rewards/rejected": -3.8051371574401855, "step": 390 }, { "epoch": 0.99, "learning_rate": 2.327445937151673e-10, "logits/chosen": -2.29669189453125, "logits/rejected": -2.1986515522003174, "logps/chosen": -561.0698852539062, "logps/rejected": -510.22021484375, "loss": 0.0666, "rewards/accuracies": 0.875, "rewards/chosen": -1.1245156526565552, "rewards/margins": 2.6927759647369385, "rewards/rejected": -3.817291736602783, "step": 400 }, { "epoch": 1.0, "step": 405, "total_flos": 0.0, "train_loss": 0.13438091388455145, "train_runtime": 3218.6044, "train_samples_per_second": 16.123, "train_steps_per_second": 0.126 } ], "logging_steps": 10, "max_steps": 405, "num_train_epochs": 1, "save_steps": 1000, "total_flos": 0.0, "trial_name": null, "trial_params": null }