diff --git "a/trainer_state.json" "b/trainer_state.json" new file mode 100644--- /dev/null +++ "b/trainer_state.json" @@ -0,0 +1,3048 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 1.0, + "eval_steps": 100, + "global_step": 1937, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0, + "learning_rate": 2.5773195876288657e-09, + "logits/chosen": -2.3950748443603516, + "logits/rejected": -2.440203905105591, + "logps/chosen": -134.41867065429688, + "logps/rejected": -130.99615478515625, + "loss": 0.6931, + "rewards/accuracies": 0.0, + "rewards/chosen": 0.0, + "rewards/margins": 0.0, + "rewards/rejected": 0.0, + "step": 1 + }, + { + "epoch": 0.01, + "learning_rate": 2.5773195876288656e-08, + "logits/chosen": -2.905548095703125, + "logits/rejected": -2.900480031967163, + "logps/chosen": -333.7513427734375, + "logps/rejected": -256.0840148925781, + "loss": 0.693, + "rewards/accuracies": 0.4444444477558136, + "rewards/chosen": 0.0011086573358625174, + "rewards/margins": 0.0019541841465979815, + "rewards/rejected": -0.0008455271599814296, + "step": 10 + }, + { + "epoch": 0.01, + "learning_rate": 5.154639175257731e-08, + "logits/chosen": -2.7942252159118652, + "logits/rejected": -2.5809876918792725, + "logps/chosen": -318.5552673339844, + "logps/rejected": -244.302734375, + "loss": 0.6874, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 0.007939901202917099, + "rewards/margins": 0.012212857604026794, + "rewards/rejected": -0.004272956866770983, + "step": 20 + }, + { + "epoch": 0.02, + "learning_rate": 7.731958762886598e-08, + "logits/chosen": -2.95878267288208, + "logits/rejected": -2.914288282394409, + "logps/chosen": -328.8043518066406, + "logps/rejected": -237.27151489257812, + "loss": 0.6807, + "rewards/accuracies": 0.5, + "rewards/chosen": 0.042353544384241104, + "rewards/margins": 0.013048592023551464, + "rewards/rejected": 0.029304955154657364, + "step": 30 + }, + { + "epoch": 0.02, + "learning_rate": 1.0309278350515462e-07, + "logits/chosen": -2.766305685043335, + "logits/rejected": -2.8168842792510986, + "logps/chosen": -298.6869812011719, + "logps/rejected": -283.5751037597656, + "loss": 0.6611, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 0.053902607411146164, + "rewards/margins": 0.043827421963214874, + "rewards/rejected": 0.010075189173221588, + "step": 40 + }, + { + "epoch": 0.03, + "learning_rate": 1.2886597938144328e-07, + "logits/chosen": -2.940128803253174, + "logits/rejected": -2.9153294563293457, + "logps/chosen": -353.18408203125, + "logps/rejected": -278.1434326171875, + "loss": 0.6488, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 0.10765328258275986, + "rewards/margins": 0.15659113228321075, + "rewards/rejected": -0.0489378497004509, + "step": 50 + }, + { + "epoch": 0.03, + "learning_rate": 1.5463917525773197e-07, + "logits/chosen": -2.9204277992248535, + "logits/rejected": -2.847609281539917, + "logps/chosen": -337.5159912109375, + "logps/rejected": -260.6918029785156, + "loss": 0.6253, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": 0.17934152483940125, + "rewards/margins": 0.402275413274765, + "rewards/rejected": -0.22293388843536377, + "step": 60 + }, + { + "epoch": 0.04, + "learning_rate": 1.804123711340206e-07, + "logits/chosen": -2.9181199073791504, + "logits/rejected": -2.883514881134033, + "logps/chosen": -402.8895568847656, + "logps/rejected": -262.24609375, + "loss": 0.6168, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": 0.14267697930335999, + "rewards/margins": 0.45374640822410583, + "rewards/rejected": -0.3110693693161011, + "step": 70 + }, + { + "epoch": 0.04, + "learning_rate": 2.0618556701030925e-07, + "logits/chosen": -3.0056729316711426, + "logits/rejected": -2.903585433959961, + "logps/chosen": -391.81524658203125, + "logps/rejected": -263.700439453125, + "loss": 0.5578, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.3499735891819, + "rewards/margins": 0.6209262609481812, + "rewards/rejected": -0.970899760723114, + "step": 80 + }, + { + "epoch": 0.05, + "learning_rate": 2.3195876288659794e-07, + "logits/chosen": -2.8997607231140137, + "logits/rejected": -2.787290573120117, + "logps/chosen": -333.6437072753906, + "logps/rejected": -248.12149047851562, + "loss": 0.5714, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.30834370851516724, + "rewards/margins": 0.7723485231399536, + "rewards/rejected": -1.0806924104690552, + "step": 90 + }, + { + "epoch": 0.05, + "learning_rate": 2.5773195876288655e-07, + "logits/chosen": -2.729923725128174, + "logits/rejected": -2.668921947479248, + "logps/chosen": -299.4058837890625, + "logps/rejected": -299.69464111328125, + "loss": 0.5602, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.3236086368560791, + "rewards/margins": 0.48320120573043823, + "rewards/rejected": -0.8068099021911621, + "step": 100 + }, + { + "epoch": 0.05, + "eval_logits/chosen": -2.679702043533325, + "eval_logits/rejected": -2.655423879623413, + "eval_logps/chosen": -293.716064453125, + "eval_logps/rejected": -306.2606506347656, + "eval_loss": 0.5588741302490234, + "eval_rewards/accuracies": 0.71875, + "eval_rewards/chosen": -0.3358975350856781, + "eval_rewards/margins": 0.48091739416122437, + "eval_rewards/rejected": -0.8168148994445801, + "eval_runtime": 59.9141, + "eval_samples_per_second": 16.691, + "eval_steps_per_second": 0.267, + "step": 100 + }, + { + "epoch": 0.06, + "learning_rate": 2.835051546391752e-07, + "logits/chosen": -2.91627836227417, + "logits/rejected": -2.970510482788086, + "logps/chosen": -383.3172302246094, + "logps/rejected": -347.4027404785156, + "loss": 0.5764, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.4358394742012024, + "rewards/margins": 0.6925853490829468, + "rewards/rejected": -1.128424882888794, + "step": 110 + }, + { + "epoch": 0.06, + "learning_rate": 3.0927835051546394e-07, + "logits/chosen": -2.760704517364502, + "logits/rejected": -2.690364360809326, + "logps/chosen": -230.55453491210938, + "logps/rejected": -258.10711669921875, + "loss": 0.5352, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -0.5467896461486816, + "rewards/margins": 0.3793262541294098, + "rewards/rejected": -0.926115870475769, + "step": 120 + }, + { + "epoch": 0.07, + "learning_rate": 3.3505154639175255e-07, + "logits/chosen": -2.9248669147491455, + "logits/rejected": -2.743758201599121, + "logps/chosen": -375.25665283203125, + "logps/rejected": -237.9368438720703, + "loss": 0.5478, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.2538636326789856, + "rewards/margins": 0.8244791030883789, + "rewards/rejected": -1.0783426761627197, + "step": 130 + }, + { + "epoch": 0.07, + "learning_rate": 3.608247422680412e-07, + "logits/chosen": -2.789468765258789, + "logits/rejected": -2.6087424755096436, + "logps/chosen": -341.793212890625, + "logps/rejected": -303.3952941894531, + "loss": 0.531, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.5787044763565063, + "rewards/margins": 0.7447447776794434, + "rewards/rejected": -1.3234491348266602, + "step": 140 + }, + { + "epoch": 0.08, + "learning_rate": 3.865979381443299e-07, + "logits/chosen": -3.055696964263916, + "logits/rejected": -2.8961217403411865, + "logps/chosen": -342.10162353515625, + "logps/rejected": -362.9927978515625, + "loss": 0.5117, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.22076380252838135, + "rewards/margins": 1.0446679592132568, + "rewards/rejected": -1.2654317617416382, + "step": 150 + }, + { + "epoch": 0.08, + "learning_rate": 4.123711340206185e-07, + "logits/chosen": -2.769010066986084, + "logits/rejected": -2.677729368209839, + "logps/chosen": -316.1387939453125, + "logps/rejected": -257.9635009765625, + "loss": 0.5264, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.5055423378944397, + "rewards/margins": 0.9351354837417603, + "rewards/rejected": -1.4406778812408447, + "step": 160 + }, + { + "epoch": 0.09, + "learning_rate": 4.381443298969072e-07, + "logits/chosen": -2.7085089683532715, + "logits/rejected": -2.600250005722046, + "logps/chosen": -276.8943176269531, + "logps/rejected": -258.60809326171875, + "loss": 0.478, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.2844000458717346, + "rewards/margins": 1.338111162185669, + "rewards/rejected": -1.6225111484527588, + "step": 170 + }, + { + "epoch": 0.09, + "learning_rate": 4.639175257731959e-07, + "logits/chosen": -2.8041911125183105, + "logits/rejected": -2.6761889457702637, + "logps/chosen": -272.099365234375, + "logps/rejected": -236.39675903320312, + "loss": 0.4904, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.0498843789100647, + "rewards/margins": 1.0812078714370728, + "rewards/rejected": -1.1310923099517822, + "step": 180 + }, + { + "epoch": 0.1, + "learning_rate": 4.896907216494845e-07, + "logits/chosen": -2.7893078327178955, + "logits/rejected": -2.787668228149414, + "logps/chosen": -315.68365478515625, + "logps/rejected": -278.1927490234375, + "loss": 0.5809, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.05461850017309189, + "rewards/margins": 0.6613677740097046, + "rewards/rejected": -0.7159863114356995, + "step": 190 + }, + { + "epoch": 0.1, + "learning_rate": 4.982788296041308e-07, + "logits/chosen": -2.7293038368225098, + "logits/rejected": -2.7431674003601074, + "logps/chosen": -319.84307861328125, + "logps/rejected": -285.5979309082031, + "loss": 0.4852, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.25821852684020996, + "rewards/margins": 0.9920754432678223, + "rewards/rejected": -1.2502939701080322, + "step": 200 + }, + { + "epoch": 0.1, + "eval_logits/chosen": -2.5956599712371826, + "eval_logits/rejected": -2.576249122619629, + "eval_logps/chosen": -297.6180725097656, + "eval_logps/rejected": -319.912353515625, + "eval_loss": 0.513595700263977, + "eval_rewards/accuracies": 0.8125, + "eval_rewards/chosen": -0.5309990048408508, + "eval_rewards/margins": 0.9684009552001953, + "eval_rewards/rejected": -1.4993999004364014, + "eval_runtime": 58.9562, + "eval_samples_per_second": 16.962, + "eval_steps_per_second": 0.271, + "step": 200 + }, + { + "epoch": 0.11, + "learning_rate": 4.954102122776821e-07, + "logits/chosen": -2.4963221549987793, + "logits/rejected": -2.2522215843200684, + "logps/chosen": -330.4200134277344, + "logps/rejected": -269.993896484375, + "loss": 0.5766, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.39517518877983093, + "rewards/margins": 1.0865150690078735, + "rewards/rejected": -1.4816901683807373, + "step": 210 + }, + { + "epoch": 0.11, + "learning_rate": 4.925415949512335e-07, + "logits/chosen": -2.709627628326416, + "logits/rejected": -2.6324925422668457, + "logps/chosen": -355.61602783203125, + "logps/rejected": -297.36346435546875, + "loss": 0.5109, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.4435829222202301, + "rewards/margins": 0.6400429606437683, + "rewards/rejected": -1.0836259126663208, + "step": 220 + }, + { + "epoch": 0.12, + "learning_rate": 4.896729776247848e-07, + "logits/chosen": -2.7627291679382324, + "logits/rejected": -2.728057384490967, + "logps/chosen": -326.62017822265625, + "logps/rejected": -297.73150634765625, + "loss": 0.4948, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.0938631147146225, + "rewards/margins": 0.8200471997261047, + "rewards/rejected": -0.9139102697372437, + "step": 230 + }, + { + "epoch": 0.12, + "learning_rate": 4.868043602983362e-07, + "logits/chosen": -2.758730173110962, + "logits/rejected": -2.7346835136413574, + "logps/chosen": -399.9264221191406, + "logps/rejected": -287.3780212402344, + "loss": 0.5489, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -0.5322622656822205, + "rewards/margins": 0.36636096239089966, + "rewards/rejected": -0.8986232876777649, + "step": 240 + }, + { + "epoch": 0.13, + "learning_rate": 4.839357429718875e-07, + "logits/chosen": -2.922417163848877, + "logits/rejected": -2.7935752868652344, + "logps/chosen": -267.0278015136719, + "logps/rejected": -201.1547088623047, + "loss": 0.509, + "rewards/accuracies": 0.949999988079071, + "rewards/chosen": 0.024714648723602295, + "rewards/margins": 1.111724615097046, + "rewards/rejected": -1.087010145187378, + "step": 250 + }, + { + "epoch": 0.13, + "learning_rate": 4.810671256454389e-07, + "logits/chosen": -2.9726548194885254, + "logits/rejected": -2.8518033027648926, + "logps/chosen": -345.22869873046875, + "logps/rejected": -239.1555938720703, + "loss": 0.5219, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": 0.062345243990421295, + "rewards/margins": 1.2734425067901611, + "rewards/rejected": -1.211097240447998, + "step": 260 + }, + { + "epoch": 0.14, + "learning_rate": 4.781985083189902e-07, + "logits/chosen": -2.840888500213623, + "logits/rejected": -2.7617764472961426, + "logps/chosen": -343.1612243652344, + "logps/rejected": -278.6701354980469, + "loss": 0.4955, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.15016205608844757, + "rewards/margins": 0.7697200179100037, + "rewards/rejected": -0.9198821187019348, + "step": 270 + }, + { + "epoch": 0.14, + "learning_rate": 4.7532989099254154e-07, + "logits/chosen": -2.9551005363464355, + "logits/rejected": -2.8918609619140625, + "logps/chosen": -391.0706787109375, + "logps/rejected": -303.4861755371094, + "loss": 0.5101, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.2836604118347168, + "rewards/margins": 0.7845603227615356, + "rewards/rejected": -1.068220853805542, + "step": 280 + }, + { + "epoch": 0.15, + "learning_rate": 4.724612736660929e-07, + "logits/chosen": -2.664578437805176, + "logits/rejected": -2.801819086074829, + "logps/chosen": -272.6911315917969, + "logps/rejected": -268.9104919433594, + "loss": 0.4815, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 0.15687711536884308, + "rewards/margins": 0.9703904986381531, + "rewards/rejected": -0.813513457775116, + "step": 290 + }, + { + "epoch": 0.15, + "learning_rate": 4.695926563396443e-07, + "logits/chosen": -2.7159955501556396, + "logits/rejected": -2.7721734046936035, + "logps/chosen": -444.1826171875, + "logps/rejected": -277.5564270019531, + "loss": 0.5212, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": 0.18774226307868958, + "rewards/margins": 1.6489717960357666, + "rewards/rejected": -1.4612294435501099, + "step": 300 + }, + { + "epoch": 0.15, + "eval_logits/chosen": -2.7125377655029297, + "eval_logits/rejected": -2.6865103244781494, + "eval_logps/chosen": -290.3699035644531, + "eval_logps/rejected": -313.44439697265625, + "eval_loss": 0.5167616009712219, + "eval_rewards/accuracies": 0.78125, + "eval_rewards/chosen": -0.1685914546251297, + "eval_rewards/margins": 1.0074106454849243, + "eval_rewards/rejected": -1.1760022640228271, + "eval_runtime": 57.2558, + "eval_samples_per_second": 17.465, + "eval_steps_per_second": 0.279, + "step": 300 + }, + { + "epoch": 0.16, + "learning_rate": 4.6672403901319564e-07, + "logits/chosen": -2.9332168102264404, + "logits/rejected": -2.83400297164917, + "logps/chosen": -351.634033203125, + "logps/rejected": -267.8584899902344, + "loss": 0.5018, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.10320504754781723, + "rewards/margins": 1.208617091178894, + "rewards/rejected": -1.3118221759796143, + "step": 310 + }, + { + "epoch": 0.17, + "learning_rate": 4.63855421686747e-07, + "logits/chosen": -2.86641001701355, + "logits/rejected": -2.9064583778381348, + "logps/chosen": -401.264404296875, + "logps/rejected": -330.1752624511719, + "loss": 0.4492, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.5078498125076294, + "rewards/margins": 1.2765679359436035, + "rewards/rejected": -1.784417748451233, + "step": 320 + }, + { + "epoch": 0.17, + "learning_rate": 4.609868043602983e-07, + "logits/chosen": -2.7677345275878906, + "logits/rejected": -2.808917760848999, + "logps/chosen": -373.0283203125, + "logps/rejected": -278.2051086425781, + "loss": 0.4576, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.5256485939025879, + "rewards/margins": 1.5822279453277588, + "rewards/rejected": -2.1078765392303467, + "step": 330 + }, + { + "epoch": 0.18, + "learning_rate": 4.581181870338497e-07, + "logits/chosen": -2.5862860679626465, + "logits/rejected": -2.6729650497436523, + "logps/chosen": -264.1101989746094, + "logps/rejected": -302.21832275390625, + "loss": 0.5585, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.3558864891529083, + "rewards/margins": 0.7631327509880066, + "rewards/rejected": -1.1190193891525269, + "step": 340 + }, + { + "epoch": 0.18, + "learning_rate": 4.55249569707401e-07, + "logits/chosen": -2.9297871589660645, + "logits/rejected": -2.8774681091308594, + "logps/chosen": -199.7515106201172, + "logps/rejected": -261.4060363769531, + "loss": 0.4773, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.46969500184059143, + "rewards/margins": 0.5962409973144531, + "rewards/rejected": -1.0659358501434326, + "step": 350 + }, + { + "epoch": 0.19, + "learning_rate": 4.5238095238095237e-07, + "logits/chosen": -2.9737091064453125, + "logits/rejected": -3.0415940284729004, + "logps/chosen": -417.215087890625, + "logps/rejected": -402.1128845214844, + "loss": 0.4545, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": 0.06597085297107697, + "rewards/margins": 1.6805864572525024, + "rewards/rejected": -1.6146154403686523, + "step": 360 + }, + { + "epoch": 0.19, + "learning_rate": 4.495123350545037e-07, + "logits/chosen": -2.9340670108795166, + "logits/rejected": -2.8952622413635254, + "logps/chosen": -200.85787963867188, + "logps/rejected": -190.8145751953125, + "loss": 0.462, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.39831048250198364, + "rewards/margins": 1.7207963466644287, + "rewards/rejected": -2.1191065311431885, + "step": 370 + }, + { + "epoch": 0.2, + "learning_rate": 4.46643717728055e-07, + "logits/chosen": -2.7802371978759766, + "logits/rejected": -2.692594051361084, + "logps/chosen": -342.89556884765625, + "logps/rejected": -360.488037109375, + "loss": 0.5404, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.2285180538892746, + "rewards/margins": 0.9076235890388489, + "rewards/rejected": -1.1361417770385742, + "step": 380 + }, + { + "epoch": 0.2, + "learning_rate": 4.437751004016064e-07, + "logits/chosen": -2.8958873748779297, + "logits/rejected": -2.8253509998321533, + "logps/chosen": -338.5411071777344, + "logps/rejected": -302.31536865234375, + "loss": 0.4479, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.5085374712944031, + "rewards/margins": 1.4444196224212646, + "rewards/rejected": -1.9529569149017334, + "step": 390 + }, + { + "epoch": 0.21, + "learning_rate": 4.4090648307515776e-07, + "logits/chosen": -3.072361469268799, + "logits/rejected": -3.0403876304626465, + "logps/chosen": -298.0294189453125, + "logps/rejected": -269.53759765625, + "loss": 0.5496, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.7352712154388428, + "rewards/margins": 0.4947175979614258, + "rewards/rejected": -1.229988694190979, + "step": 400 + }, + { + "epoch": 0.21, + "eval_logits/chosen": -2.8217618465423584, + "eval_logits/rejected": -2.7946977615356445, + "eval_logps/chosen": -290.2325744628906, + "eval_logps/rejected": -324.2634582519531, + "eval_loss": 0.4835220277309418, + "eval_rewards/accuracies": 0.828125, + "eval_rewards/chosen": -0.16172367334365845, + "eval_rewards/margins": 1.5552303791046143, + "eval_rewards/rejected": -1.716953992843628, + "eval_runtime": 58.8529, + "eval_samples_per_second": 16.992, + "eval_steps_per_second": 0.272, + "step": 400 + }, + { + "epoch": 0.21, + "learning_rate": 4.380378657487091e-07, + "logits/chosen": -2.878788948059082, + "logits/rejected": -2.7508370876312256, + "logps/chosen": -361.15850830078125, + "logps/rejected": -244.7436065673828, + "loss": 0.4209, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.3598092496395111, + "rewards/margins": 0.8073941469192505, + "rewards/rejected": -1.1672031879425049, + "step": 410 + }, + { + "epoch": 0.22, + "learning_rate": 4.3516924842226045e-07, + "logits/chosen": -2.940977096557617, + "logits/rejected": -2.811171531677246, + "logps/chosen": -342.15802001953125, + "logps/rejected": -299.4014587402344, + "loss": 0.4994, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -0.5423839092254639, + "rewards/margins": 1.0953930616378784, + "rewards/rejected": -1.6377769708633423, + "step": 420 + }, + { + "epoch": 0.22, + "learning_rate": 4.323006310958118e-07, + "logits/chosen": -2.928069591522217, + "logits/rejected": -2.839228391647339, + "logps/chosen": -246.8507080078125, + "logps/rejected": -235.521728515625, + "loss": 0.4726, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.7062541246414185, + "rewards/margins": 0.9924257397651672, + "rewards/rejected": -1.6986801624298096, + "step": 430 + }, + { + "epoch": 0.23, + "learning_rate": 4.2943201376936315e-07, + "logits/chosen": -2.8603832721710205, + "logits/rejected": -2.7824721336364746, + "logps/chosen": -376.1601257324219, + "logps/rejected": -356.33905029296875, + "loss": 0.5524, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.40359026193618774, + "rewards/margins": 0.40071097016334534, + "rewards/rejected": -0.8043011426925659, + "step": 440 + }, + { + "epoch": 0.23, + "learning_rate": 4.265633964429145e-07, + "logits/chosen": -2.9410994052886963, + "logits/rejected": -2.9663491249084473, + "logps/chosen": -346.1313171386719, + "logps/rejected": -281.6514587402344, + "loss": 0.469, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.5330409407615662, + "rewards/margins": 0.553443968296051, + "rewards/rejected": -1.0864850282669067, + "step": 450 + }, + { + "epoch": 0.24, + "learning_rate": 4.2369477911646584e-07, + "logits/chosen": -2.9534189701080322, + "logits/rejected": -2.9417405128479004, + "logps/chosen": -362.89105224609375, + "logps/rejected": -356.94598388671875, + "loss": 0.546, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.10002334415912628, + "rewards/margins": 1.6824315786361694, + "rewards/rejected": -1.7824550867080688, + "step": 460 + }, + { + "epoch": 0.24, + "learning_rate": 4.208261617900172e-07, + "logits/chosen": -2.8067574501037598, + "logits/rejected": -2.815100908279419, + "logps/chosen": -359.10791015625, + "logps/rejected": -276.3143615722656, + "loss": 0.4473, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.5546383857727051, + "rewards/margins": 0.8130642175674438, + "rewards/rejected": -1.367702603340149, + "step": 470 + }, + { + "epoch": 0.25, + "learning_rate": 4.179575444635686e-07, + "logits/chosen": -3.0283010005950928, + "logits/rejected": -3.0092482566833496, + "logps/chosen": -320.7592468261719, + "logps/rejected": -327.0148010253906, + "loss": 0.4589, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.023906176909804344, + "rewards/margins": 0.516281008720398, + "rewards/rejected": -0.5401870608329773, + "step": 480 + }, + { + "epoch": 0.25, + "learning_rate": 4.150889271371199e-07, + "logits/chosen": -2.9509072303771973, + "logits/rejected": -2.9088292121887207, + "logps/chosen": -359.8896789550781, + "logps/rejected": -298.53167724609375, + "loss": 0.4272, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.9165481328964233, + "rewards/margins": 0.8448396921157837, + "rewards/rejected": -1.761387586593628, + "step": 490 + }, + { + "epoch": 0.26, + "learning_rate": 4.1222030981067123e-07, + "logits/chosen": -3.1632120609283447, + "logits/rejected": -2.959442138671875, + "logps/chosen": -348.33709716796875, + "logps/rejected": -335.841552734375, + "loss": 0.5209, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 0.01871591992676258, + "rewards/margins": 1.6875536441802979, + "rewards/rejected": -1.668837547302246, + "step": 500 + }, + { + "epoch": 0.26, + "eval_logits/chosen": -2.8667073249816895, + "eval_logits/rejected": -2.8388283252716064, + "eval_logps/chosen": -296.5545654296875, + "eval_logps/rejected": -323.13250732421875, + "eval_loss": 0.5053695440292358, + "eval_rewards/accuracies": 0.734375, + "eval_rewards/chosen": -0.4778231978416443, + "eval_rewards/margins": 1.1825826168060303, + "eval_rewards/rejected": -1.6604057550430298, + "eval_runtime": 53.6768, + "eval_samples_per_second": 18.63, + "eval_steps_per_second": 0.298, + "step": 500 + }, + { + "epoch": 0.26, + "learning_rate": 4.093516924842226e-07, + "logits/chosen": -2.5999233722686768, + "logits/rejected": -2.6488568782806396, + "logps/chosen": -279.40753173828125, + "logps/rejected": -288.4382019042969, + "loss": 0.5262, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.03970176726579666, + "rewards/margins": 0.8601824641227722, + "rewards/rejected": -0.8998842239379883, + "step": 510 + }, + { + "epoch": 0.27, + "learning_rate": 4.064830751577739e-07, + "logits/chosen": -3.004120349884033, + "logits/rejected": -2.983384132385254, + "logps/chosen": -223.1768035888672, + "logps/rejected": -233.73312377929688, + "loss": 0.5426, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.22979409992694855, + "rewards/margins": 1.1553027629852295, + "rewards/rejected": -1.3850969076156616, + "step": 520 + }, + { + "epoch": 0.27, + "learning_rate": 4.036144578313253e-07, + "logits/chosen": -3.057387351989746, + "logits/rejected": -2.9704792499542236, + "logps/chosen": -325.57025146484375, + "logps/rejected": -283.2383117675781, + "loss": 0.5058, + "rewards/accuracies": 0.949999988079071, + "rewards/chosen": 0.17162951827049255, + "rewards/margins": 1.4068710803985596, + "rewards/rejected": -1.2352415323257446, + "step": 530 + }, + { + "epoch": 0.28, + "learning_rate": 4.007458405048766e-07, + "logits/chosen": -2.940493583679199, + "logits/rejected": -2.9570419788360596, + "logps/chosen": -227.3856964111328, + "logps/rejected": -289.257080078125, + "loss": 0.5339, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.835346519947052, + "rewards/margins": 0.5295913219451904, + "rewards/rejected": -1.3649379014968872, + "step": 540 + }, + { + "epoch": 0.28, + "learning_rate": 3.9787722317842796e-07, + "logits/chosen": -3.0390350818634033, + "logits/rejected": -3.0303795337677, + "logps/chosen": -389.0104675292969, + "logps/rejected": -340.8678283691406, + "loss": 0.5671, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.15655937790870667, + "rewards/margins": 1.444120168685913, + "rewards/rejected": -1.6006797552108765, + "step": 550 + }, + { + "epoch": 0.29, + "learning_rate": 3.950086058519793e-07, + "logits/chosen": -3.060770034790039, + "logits/rejected": -3.0684587955474854, + "logps/chosen": -366.5193786621094, + "logps/rejected": -359.5283203125, + "loss": 0.501, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.8856328129768372, + "rewards/margins": 1.0001723766326904, + "rewards/rejected": -1.8858048915863037, + "step": 560 + }, + { + "epoch": 0.29, + "learning_rate": 3.9213998852553066e-07, + "logits/chosen": -3.022430896759033, + "logits/rejected": -2.831993341445923, + "logps/chosen": -379.4962463378906, + "logps/rejected": -278.2923889160156, + "loss": 0.4947, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.7191627621650696, + "rewards/margins": 1.4330142736434937, + "rewards/rejected": -2.152177095413208, + "step": 570 + }, + { + "epoch": 0.3, + "learning_rate": 3.8927137119908206e-07, + "logits/chosen": -2.922940731048584, + "logits/rejected": -2.9428744316101074, + "logps/chosen": -361.6253356933594, + "logps/rejected": -278.06280517578125, + "loss": 0.5375, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.573477029800415, + "rewards/margins": 0.7330626249313354, + "rewards/rejected": -1.30653977394104, + "step": 580 + }, + { + "epoch": 0.3, + "learning_rate": 3.864027538726334e-07, + "logits/chosen": -2.9140264987945557, + "logits/rejected": -2.889644145965576, + "logps/chosen": -289.79058837890625, + "logps/rejected": -292.6978759765625, + "loss": 0.4726, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.1533357799053192, + "rewards/margins": 1.4288156032562256, + "rewards/rejected": -1.582151174545288, + "step": 590 + }, + { + "epoch": 0.31, + "learning_rate": 3.835341365461847e-07, + "logits/chosen": -2.869101047515869, + "logits/rejected": -2.88112735748291, + "logps/chosen": -243.238037109375, + "logps/rejected": -313.36260986328125, + "loss": 0.4617, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.5780342221260071, + "rewards/margins": 1.605724573135376, + "rewards/rejected": -2.1837587356567383, + "step": 600 + }, + { + "epoch": 0.31, + "eval_logits/chosen": -2.8521082401275635, + "eval_logits/rejected": -2.823408603668213, + "eval_logps/chosen": -294.47406005859375, + "eval_logps/rejected": -320.2847900390625, + "eval_loss": 0.4909786283969879, + "eval_rewards/accuracies": 0.765625, + "eval_rewards/chosen": -0.373797744512558, + "eval_rewards/margins": 1.1442232131958008, + "eval_rewards/rejected": -1.5180209875106812, + "eval_runtime": 59.0534, + "eval_samples_per_second": 16.934, + "eval_steps_per_second": 0.271, + "step": 600 + }, + { + "epoch": 0.31, + "learning_rate": 3.8066551921973605e-07, + "logits/chosen": -2.9201841354370117, + "logits/rejected": -2.845341205596924, + "logps/chosen": -338.87127685546875, + "logps/rejected": -337.95928955078125, + "loss": 0.5093, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.7563498020172119, + "rewards/margins": 1.1776630878448486, + "rewards/rejected": -1.934012770652771, + "step": 610 + }, + { + "epoch": 0.32, + "learning_rate": 3.7779690189328745e-07, + "logits/chosen": -3.034162759780884, + "logits/rejected": -3.027251720428467, + "logps/chosen": -379.36590576171875, + "logps/rejected": -306.5363464355469, + "loss": 0.5269, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.6596435308456421, + "rewards/margins": 0.7433546185493469, + "rewards/rejected": -1.4029979705810547, + "step": 620 + }, + { + "epoch": 0.33, + "learning_rate": 3.749282845668388e-07, + "logits/chosen": -2.87540864944458, + "logits/rejected": -2.834453821182251, + "logps/chosen": -252.4665069580078, + "logps/rejected": -234.0683135986328, + "loss": 0.4806, + "rewards/accuracies": 0.75, + "rewards/chosen": -1.2540767192840576, + "rewards/margins": 1.1702220439910889, + "rewards/rejected": -2.4242987632751465, + "step": 630 + }, + { + "epoch": 0.33, + "learning_rate": 3.7205966724039014e-07, + "logits/chosen": -2.9034817218780518, + "logits/rejected": -2.8846733570098877, + "logps/chosen": -308.4912109375, + "logps/rejected": -294.0049743652344, + "loss": 0.5071, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.7412096261978149, + "rewards/margins": 1.7391561269760132, + "rewards/rejected": -2.480365753173828, + "step": 640 + }, + { + "epoch": 0.34, + "learning_rate": 3.6919104991394144e-07, + "logits/chosen": -2.931398868560791, + "logits/rejected": -2.982588291168213, + "logps/chosen": -284.8676452636719, + "logps/rejected": -328.3984069824219, + "loss": 0.5295, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.8696074485778809, + "rewards/margins": 1.0539840459823608, + "rewards/rejected": -1.9235913753509521, + "step": 650 + }, + { + "epoch": 0.34, + "learning_rate": 3.663224325874928e-07, + "logits/chosen": -3.033780097961426, + "logits/rejected": -2.9609603881835938, + "logps/chosen": -372.281005859375, + "logps/rejected": -324.37640380859375, + "loss": 0.4882, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.7286888957023621, + "rewards/margins": 1.0642763376235962, + "rewards/rejected": -1.792965292930603, + "step": 660 + }, + { + "epoch": 0.35, + "learning_rate": 3.634538152610442e-07, + "logits/chosen": -2.93717622756958, + "logits/rejected": -2.9276621341705322, + "logps/chosen": -240.49267578125, + "logps/rejected": -235.82901000976562, + "loss": 0.5725, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.5567636489868164, + "rewards/margins": 0.7625681161880493, + "rewards/rejected": -1.3193317651748657, + "step": 670 + }, + { + "epoch": 0.35, + "learning_rate": 3.6058519793459553e-07, + "logits/chosen": -2.834033727645874, + "logits/rejected": -2.778200387954712, + "logps/chosen": -311.96209716796875, + "logps/rejected": -279.34210205078125, + "loss": 0.4695, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.3849760890007019, + "rewards/margins": 1.1402137279510498, + "rewards/rejected": -1.5251898765563965, + "step": 680 + }, + { + "epoch": 0.36, + "learning_rate": 3.577165806081469e-07, + "logits/chosen": -3.0405967235565186, + "logits/rejected": -2.836169481277466, + "logps/chosen": -377.798828125, + "logps/rejected": -225.7850799560547, + "loss": 0.4235, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": 0.16026358306407928, + "rewards/margins": 1.4788005352020264, + "rewards/rejected": -1.3185369968414307, + "step": 690 + }, + { + "epoch": 0.36, + "learning_rate": 3.5484796328169817e-07, + "logits/chosen": -2.914154052734375, + "logits/rejected": -2.837174415588379, + "logps/chosen": -486.6206970214844, + "logps/rejected": -388.7563171386719, + "loss": 0.4452, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 0.3467257618904114, + "rewards/margins": 1.8637710809707642, + "rewards/rejected": -1.517045021057129, + "step": 700 + }, + { + "epoch": 0.36, + "eval_logits/chosen": -2.7652909755706787, + "eval_logits/rejected": -2.7400858402252197, + "eval_logps/chosen": -296.1795959472656, + "eval_logps/rejected": -323.0770263671875, + "eval_loss": 0.48382142186164856, + "eval_rewards/accuracies": 0.703125, + "eval_rewards/chosen": -0.45907530188560486, + "eval_rewards/margins": 1.19855797290802, + "eval_rewards/rejected": -1.6576331853866577, + "eval_runtime": 56.4234, + "eval_samples_per_second": 17.723, + "eval_steps_per_second": 0.284, + "step": 700 + }, + { + "epoch": 0.37, + "learning_rate": 3.519793459552495e-07, + "logits/chosen": -2.81813645362854, + "logits/rejected": -2.891169309616089, + "logps/chosen": -304.12921142578125, + "logps/rejected": -300.01275634765625, + "loss": 0.4796, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -1.1196739673614502, + "rewards/margins": 0.9914043545722961, + "rewards/rejected": -2.1110780239105225, + "step": 710 + }, + { + "epoch": 0.37, + "learning_rate": 3.491107286288009e-07, + "logits/chosen": -2.8922622203826904, + "logits/rejected": -2.8977553844451904, + "logps/chosen": -317.526123046875, + "logps/rejected": -434.422119140625, + "loss": 0.4804, + "rewards/accuracies": 1.0, + "rewards/chosen": -0.36432236433029175, + "rewards/margins": 2.1755287647247314, + "rewards/rejected": -2.539850950241089, + "step": 720 + }, + { + "epoch": 0.38, + "learning_rate": 3.4624211130235227e-07, + "logits/chosen": -2.897742509841919, + "logits/rejected": -2.849132537841797, + "logps/chosen": -359.33770751953125, + "logps/rejected": -375.489990234375, + "loss": 0.4515, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.7771750688552856, + "rewards/margins": 0.7097036838531494, + "rewards/rejected": -1.4868788719177246, + "step": 730 + }, + { + "epoch": 0.38, + "learning_rate": 3.433734939759036e-07, + "logits/chosen": -2.8762059211730957, + "logits/rejected": -2.845869302749634, + "logps/chosen": -367.5733642578125, + "logps/rejected": -284.77886962890625, + "loss": 0.5371, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.7336939573287964, + "rewards/margins": 1.311545968055725, + "rewards/rejected": -2.0452399253845215, + "step": 740 + }, + { + "epoch": 0.39, + "learning_rate": 3.405048766494549e-07, + "logits/chosen": -2.8201470375061035, + "logits/rejected": -2.8354251384735107, + "logps/chosen": -340.94549560546875, + "logps/rejected": -293.8914794921875, + "loss": 0.4984, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.7219395637512207, + "rewards/margins": 1.4890177249908447, + "rewards/rejected": -2.2109572887420654, + "step": 750 + }, + { + "epoch": 0.39, + "learning_rate": 3.376362593230063e-07, + "logits/chosen": -2.7780914306640625, + "logits/rejected": -2.7091243267059326, + "logps/chosen": -291.79864501953125, + "logps/rejected": -293.62762451171875, + "loss": 0.5131, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.4624660611152649, + "rewards/margins": 1.375497579574585, + "rewards/rejected": -1.8379634618759155, + "step": 760 + }, + { + "epoch": 0.4, + "learning_rate": 3.3476764199655765e-07, + "logits/chosen": -2.9393422603607178, + "logits/rejected": -2.913353681564331, + "logps/chosen": -266.66033935546875, + "logps/rejected": -310.0787658691406, + "loss": 0.5009, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.27197906374931335, + "rewards/margins": 1.5637094974517822, + "rewards/rejected": -1.835688591003418, + "step": 770 + }, + { + "epoch": 0.4, + "learning_rate": 3.31899024670109e-07, + "logits/chosen": -2.799229621887207, + "logits/rejected": -2.7572007179260254, + "logps/chosen": -322.958251953125, + "logps/rejected": -250.6745147705078, + "loss": 0.462, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.1813792735338211, + "rewards/margins": 1.8009974956512451, + "rewards/rejected": -1.9823768138885498, + "step": 780 + }, + { + "epoch": 0.41, + "learning_rate": 3.2903040734366035e-07, + "logits/chosen": -2.9396896362304688, + "logits/rejected": -2.8256068229675293, + "logps/chosen": -336.33111572265625, + "logps/rejected": -305.9268493652344, + "loss": 0.4724, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.5670480728149414, + "rewards/margins": 1.4373046159744263, + "rewards/rejected": -2.004352569580078, + "step": 790 + }, + { + "epoch": 0.41, + "learning_rate": 3.261617900172117e-07, + "logits/chosen": -2.8622994422912598, + "logits/rejected": -2.8278136253356934, + "logps/chosen": -284.0804443359375, + "logps/rejected": -313.81939697265625, + "loss": 0.4674, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": 0.13986745476722717, + "rewards/margins": 1.9817402362823486, + "rewards/rejected": -1.8418725728988647, + "step": 800 + }, + { + "epoch": 0.41, + "eval_logits/chosen": -2.6945066452026367, + "eval_logits/rejected": -2.6739954948425293, + "eval_logps/chosen": -298.38177490234375, + "eval_logps/rejected": -327.2416076660156, + "eval_loss": 0.5077354907989502, + "eval_rewards/accuracies": 0.765625, + "eval_rewards/chosen": -0.5691820383071899, + "eval_rewards/margins": 1.2966790199279785, + "eval_rewards/rejected": -1.865861177444458, + "eval_runtime": 55.4449, + "eval_samples_per_second": 18.036, + "eval_steps_per_second": 0.289, + "step": 800 + }, + { + "epoch": 0.42, + "learning_rate": 3.2329317269076304e-07, + "logits/chosen": -2.8082878589630127, + "logits/rejected": -2.783658981323242, + "logps/chosen": -226.2019500732422, + "logps/rejected": -294.66326904296875, + "loss": 0.5152, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.869047999382019, + "rewards/margins": 1.269092321395874, + "rewards/rejected": -2.1381404399871826, + "step": 810 + }, + { + "epoch": 0.42, + "learning_rate": 3.204245553643144e-07, + "logits/chosen": -2.9127354621887207, + "logits/rejected": -2.894219160079956, + "logps/chosen": -230.79513549804688, + "logps/rejected": -305.60113525390625, + "loss": 0.5279, + "rewards/accuracies": 0.3499999940395355, + "rewards/chosen": -0.7637732028961182, + "rewards/margins": -0.17934365570545197, + "rewards/rejected": -0.5844296216964722, + "step": 820 + }, + { + "epoch": 0.43, + "learning_rate": 3.1755593803786574e-07, + "logits/chosen": -2.9059672355651855, + "logits/rejected": -2.9442248344421387, + "logps/chosen": -238.79248046875, + "logps/rejected": -285.3636474609375, + "loss": 0.5044, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.5089541673660278, + "rewards/margins": 1.2097771167755127, + "rewards/rejected": -1.718731164932251, + "step": 830 + }, + { + "epoch": 0.43, + "learning_rate": 3.146873207114171e-07, + "logits/chosen": -2.953200340270996, + "logits/rejected": -2.8728442192077637, + "logps/chosen": -345.51593017578125, + "logps/rejected": -307.09002685546875, + "loss": 0.5092, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -1.007508397102356, + "rewards/margins": 0.5991470813751221, + "rewards/rejected": -1.6066553592681885, + "step": 840 + }, + { + "epoch": 0.44, + "learning_rate": 3.1181870338496843e-07, + "logits/chosen": -2.8444714546203613, + "logits/rejected": -2.791147470474243, + "logps/chosen": -296.8919677734375, + "logps/rejected": -274.53082275390625, + "loss": 0.4805, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.953098475933075, + "rewards/margins": 1.4887616634368896, + "rewards/rejected": -2.4418601989746094, + "step": 850 + }, + { + "epoch": 0.44, + "learning_rate": 3.089500860585198e-07, + "logits/chosen": -2.6704187393188477, + "logits/rejected": -2.6690378189086914, + "logps/chosen": -249.43881225585938, + "logps/rejected": -302.4588928222656, + "loss": 0.5111, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -1.3775177001953125, + "rewards/margins": 1.329897403717041, + "rewards/rejected": -2.7074151039123535, + "step": 860 + }, + { + "epoch": 0.45, + "learning_rate": 3.060814687320711e-07, + "logits/chosen": -2.839586019515991, + "logits/rejected": -2.834083318710327, + "logps/chosen": -304.1976318359375, + "logps/rejected": -256.19378662109375, + "loss": 0.4354, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.47585782408714294, + "rewards/margins": 2.021141529083252, + "rewards/rejected": -2.4969992637634277, + "step": 870 + }, + { + "epoch": 0.45, + "learning_rate": 3.0321285140562247e-07, + "logits/chosen": -2.7889621257781982, + "logits/rejected": -2.7733607292175293, + "logps/chosen": -280.20977783203125, + "logps/rejected": -347.7413635253906, + "loss": 0.4604, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.9127300977706909, + "rewards/margins": 0.9622424244880676, + "rewards/rejected": -1.8749723434448242, + "step": 880 + }, + { + "epoch": 0.46, + "learning_rate": 3.003442340791738e-07, + "logits/chosen": -2.904717206954956, + "logits/rejected": -2.835303783416748, + "logps/chosen": -336.64776611328125, + "logps/rejected": -316.78045654296875, + "loss": 0.5003, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.2770691215991974, + "rewards/margins": 1.1059465408325195, + "rewards/rejected": -1.383015751838684, + "step": 890 + }, + { + "epoch": 0.46, + "learning_rate": 2.974756167527252e-07, + "logits/chosen": -3.022934913635254, + "logits/rejected": -2.956709146499634, + "logps/chosen": -286.189208984375, + "logps/rejected": -264.89776611328125, + "loss": 0.4656, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.7135367393493652, + "rewards/margins": 1.1268901824951172, + "rewards/rejected": -1.8404268026351929, + "step": 900 + }, + { + "epoch": 0.46, + "eval_logits/chosen": -2.8014752864837646, + "eval_logits/rejected": -2.781651258468628, + "eval_logps/chosen": -297.5553283691406, + "eval_logps/rejected": -323.15179443359375, + "eval_loss": 0.4927058219909668, + "eval_rewards/accuracies": 0.765625, + "eval_rewards/chosen": -0.5278611183166504, + "eval_rewards/margins": 1.1335095167160034, + "eval_rewards/rejected": -1.661370873451233, + "eval_runtime": 57.0998, + "eval_samples_per_second": 17.513, + "eval_steps_per_second": 0.28, + "step": 900 + }, + { + "epoch": 0.47, + "learning_rate": 2.946069994262765e-07, + "logits/chosen": -2.978717803955078, + "logits/rejected": -2.8918721675872803, + "logps/chosen": -336.64495849609375, + "logps/rejected": -247.2961883544922, + "loss": 0.4344, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.6280319094657898, + "rewards/margins": 1.4678236246109009, + "rewards/rejected": -2.095855712890625, + "step": 910 + }, + { + "epoch": 0.47, + "learning_rate": 2.9173838209982786e-07, + "logits/chosen": -2.786372423171997, + "logits/rejected": -2.8410544395446777, + "logps/chosen": -298.88360595703125, + "logps/rejected": -302.8692626953125, + "loss": 0.4686, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.6036109924316406, + "rewards/margins": 0.7661906480789185, + "rewards/rejected": -1.369801640510559, + "step": 920 + }, + { + "epoch": 0.48, + "learning_rate": 2.888697647733792e-07, + "logits/chosen": -2.915067195892334, + "logits/rejected": -2.732910633087158, + "logps/chosen": -356.8029479980469, + "logps/rejected": -226.57974243164062, + "loss": 0.4128, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.25983041524887085, + "rewards/margins": 2.1089906692504883, + "rewards/rejected": -2.368821144104004, + "step": 930 + }, + { + "epoch": 0.49, + "learning_rate": 2.8600114744693055e-07, + "logits/chosen": -2.872128486633301, + "logits/rejected": -2.8349528312683105, + "logps/chosen": -276.8409729003906, + "logps/rejected": -234.0821990966797, + "loss": 0.4633, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.33096224069595337, + "rewards/margins": 1.0401618480682373, + "rewards/rejected": -1.371124029159546, + "step": 940 + }, + { + "epoch": 0.49, + "learning_rate": 2.8313253012048195e-07, + "logits/chosen": -2.768357276916504, + "logits/rejected": -2.8650612831115723, + "logps/chosen": -286.5350341796875, + "logps/rejected": -295.7557678222656, + "loss": 0.4694, + "rewards/accuracies": 1.0, + "rewards/chosen": -0.42378902435302734, + "rewards/margins": 2.2293660640716553, + "rewards/rejected": -2.6531550884246826, + "step": 950 + }, + { + "epoch": 0.5, + "learning_rate": 2.802639127940333e-07, + "logits/chosen": -2.8344969749450684, + "logits/rejected": -2.8180437088012695, + "logps/chosen": -312.8253173828125, + "logps/rejected": -299.6221618652344, + "loss": 0.4998, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.2208956480026245, + "rewards/margins": 1.7145334482192993, + "rewards/rejected": -1.9354288578033447, + "step": 960 + }, + { + "epoch": 0.5, + "learning_rate": 2.773952954675846e-07, + "logits/chosen": -2.809140682220459, + "logits/rejected": -2.699197292327881, + "logps/chosen": -369.69854736328125, + "logps/rejected": -293.45477294921875, + "loss": 0.4849, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.28183019161224365, + "rewards/margins": 1.6505457162857056, + "rewards/rejected": -1.9323756694793701, + "step": 970 + }, + { + "epoch": 0.51, + "learning_rate": 2.7452667814113594e-07, + "logits/chosen": -2.8047561645507812, + "logits/rejected": -2.777113199234009, + "logps/chosen": -326.3648376464844, + "logps/rejected": -289.0523986816406, + "loss": 0.4543, + "rewards/accuracies": 0.949999988079071, + "rewards/chosen": -0.42743197083473206, + "rewards/margins": 1.8371082544326782, + "rewards/rejected": -2.264540195465088, + "step": 980 + }, + { + "epoch": 0.51, + "learning_rate": 2.716580608146873e-07, + "logits/chosen": -2.8819570541381836, + "logits/rejected": -2.8494954109191895, + "logps/chosen": -318.66204833984375, + "logps/rejected": -345.1630859375, + "loss": 0.4534, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.4583119750022888, + "rewards/margins": 1.3975725173950195, + "rewards/rejected": -1.8558847904205322, + "step": 990 + }, + { + "epoch": 0.52, + "learning_rate": 2.687894434882387e-07, + "logits/chosen": -2.8820109367370605, + "logits/rejected": -2.9153826236724854, + "logps/chosen": -251.1084747314453, + "logps/rejected": -264.5345153808594, + "loss": 0.4102, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.5280589461326599, + "rewards/margins": 1.0574977397918701, + "rewards/rejected": -1.5855568647384644, + "step": 1000 + }, + { + "epoch": 0.52, + "eval_logits/chosen": -2.745532512664795, + "eval_logits/rejected": -2.7159857749938965, + "eval_logps/chosen": -298.5311279296875, + "eval_logps/rejected": -331.2577819824219, + "eval_loss": 0.4771742522716522, + "eval_rewards/accuracies": 0.765625, + "eval_rewards/chosen": -0.5766510367393494, + "eval_rewards/margins": 1.4900195598602295, + "eval_rewards/rejected": -2.0666706562042236, + "eval_runtime": 60.3269, + "eval_samples_per_second": 16.576, + "eval_steps_per_second": 0.265, + "step": 1000 + }, + { + "epoch": 0.52, + "learning_rate": 2.6592082616179004e-07, + "logits/chosen": -2.753818988800049, + "logits/rejected": -2.6555442810058594, + "logps/chosen": -421.75408935546875, + "logps/rejected": -326.7854919433594, + "loss": 0.4969, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.7776181101799011, + "rewards/margins": 1.5239381790161133, + "rewards/rejected": -2.30155611038208, + "step": 1010 + }, + { + "epoch": 0.53, + "learning_rate": 2.6305220883534133e-07, + "logits/chosen": -2.795933961868286, + "logits/rejected": -2.875976085662842, + "logps/chosen": -359.01318359375, + "logps/rejected": -325.6358642578125, + "loss": 0.4814, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.12685959041118622, + "rewards/margins": 1.6275579929351807, + "rewards/rejected": -1.7544174194335938, + "step": 1020 + }, + { + "epoch": 0.53, + "learning_rate": 2.601835915088927e-07, + "logits/chosen": -2.7979979515075684, + "logits/rejected": -2.7110390663146973, + "logps/chosen": -307.5633239746094, + "logps/rejected": -272.8973388671875, + "loss": 0.4354, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.4669399857521057, + "rewards/margins": 1.9002759456634521, + "rewards/rejected": -2.367216110229492, + "step": 1030 + }, + { + "epoch": 0.54, + "learning_rate": 2.573149741824441e-07, + "logits/chosen": -2.864424467086792, + "logits/rejected": -2.7874035835266113, + "logps/chosen": -301.7801513671875, + "logps/rejected": -366.8116149902344, + "loss": 0.4799, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.38910773396492004, + "rewards/margins": 2.2144553661346436, + "rewards/rejected": -2.6035633087158203, + "step": 1040 + }, + { + "epoch": 0.54, + "learning_rate": 2.544463568559954e-07, + "logits/chosen": -2.8506453037261963, + "logits/rejected": -2.7827541828155518, + "logps/chosen": -332.9029846191406, + "logps/rejected": -288.8146667480469, + "loss": 0.4728, + "rewards/accuracies": 0.75, + "rewards/chosen": -1.1103570461273193, + "rewards/margins": 1.3038640022277832, + "rewards/rejected": -2.4142210483551025, + "step": 1050 + }, + { + "epoch": 0.55, + "learning_rate": 2.5157773952954677e-07, + "logits/chosen": -2.746716022491455, + "logits/rejected": -2.738701105117798, + "logps/chosen": -362.2853698730469, + "logps/rejected": -263.8562316894531, + "loss": 0.5158, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.4159720540046692, + "rewards/margins": 1.3381980657577515, + "rewards/rejected": -1.7541700601577759, + "step": 1060 + }, + { + "epoch": 0.55, + "learning_rate": 2.4870912220309807e-07, + "logits/chosen": -2.7854630947113037, + "logits/rejected": -2.815742015838623, + "logps/chosen": -434.0101623535156, + "logps/rejected": -322.96990966796875, + "loss": 0.5047, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.38989701867103577, + "rewards/margins": 1.555858850479126, + "rewards/rejected": -1.9457557201385498, + "step": 1070 + }, + { + "epoch": 0.56, + "learning_rate": 2.4584050487664947e-07, + "logits/chosen": -2.9198765754699707, + "logits/rejected": -2.8616695404052734, + "logps/chosen": -290.16680908203125, + "logps/rejected": -285.9183349609375, + "loss": 0.4499, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.7213519811630249, + "rewards/margins": 1.0302088260650635, + "rewards/rejected": -1.751560926437378, + "step": 1080 + }, + { + "epoch": 0.56, + "learning_rate": 2.429718875502008e-07, + "logits/chosen": -2.9124810695648193, + "logits/rejected": -2.816594362258911, + "logps/chosen": -285.74139404296875, + "logps/rejected": -237.6147003173828, + "loss": 0.4673, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.599921464920044, + "rewards/margins": 1.2972980737686157, + "rewards/rejected": -1.8972194194793701, + "step": 1090 + }, + { + "epoch": 0.57, + "learning_rate": 2.4010327022375216e-07, + "logits/chosen": -2.8846845626831055, + "logits/rejected": -2.914504289627075, + "logps/chosen": -373.78887939453125, + "logps/rejected": -349.1485900878906, + "loss": 0.4663, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.4652225971221924, + "rewards/margins": 0.824032187461853, + "rewards/rejected": -1.289254903793335, + "step": 1100 + }, + { + "epoch": 0.57, + "eval_logits/chosen": -2.725741386413574, + "eval_logits/rejected": -2.6993935108184814, + "eval_logps/chosen": -303.0741271972656, + "eval_logps/rejected": -331.96044921875, + "eval_loss": 0.4740181565284729, + "eval_rewards/accuracies": 0.765625, + "eval_rewards/chosen": -0.803801417350769, + "eval_rewards/margins": 1.2980027198791504, + "eval_rewards/rejected": -2.10180401802063, + "eval_runtime": 57.8167, + "eval_samples_per_second": 17.296, + "eval_steps_per_second": 0.277, + "step": 1100 + }, + { + "epoch": 0.57, + "learning_rate": 2.3723465289730348e-07, + "logits/chosen": -2.9184155464172363, + "logits/rejected": -2.913644552230835, + "logps/chosen": -325.4616394042969, + "logps/rejected": -290.3975830078125, + "loss": 0.4519, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.7317468523979187, + "rewards/margins": 0.9042754173278809, + "rewards/rejected": -1.6360222101211548, + "step": 1110 + }, + { + "epoch": 0.58, + "learning_rate": 2.3436603557085483e-07, + "logits/chosen": -2.845045328140259, + "logits/rejected": -2.6523425579071045, + "logps/chosen": -370.0478820800781, + "logps/rejected": -286.46112060546875, + "loss": 0.5856, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.8687052726745605, + "rewards/margins": 1.1778913736343384, + "rewards/rejected": -2.0465967655181885, + "step": 1120 + }, + { + "epoch": 0.58, + "learning_rate": 2.314974182444062e-07, + "logits/chosen": -2.8266937732696533, + "logits/rejected": -2.6822288036346436, + "logps/chosen": -350.9194641113281, + "logps/rejected": -341.3634033203125, + "loss": 0.4118, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -1.1032984256744385, + "rewards/margins": 1.7717195749282837, + "rewards/rejected": -2.8750178813934326, + "step": 1130 + }, + { + "epoch": 0.59, + "learning_rate": 2.2862880091795752e-07, + "logits/chosen": -2.71614146232605, + "logits/rejected": -2.7234530448913574, + "logps/chosen": -332.0517272949219, + "logps/rejected": -334.45977783203125, + "loss": 0.5011, + "rewards/accuracies": 0.75, + "rewards/chosen": -1.2248785495758057, + "rewards/margins": 1.2211261987686157, + "rewards/rejected": -2.4460043907165527, + "step": 1140 + }, + { + "epoch": 0.59, + "learning_rate": 2.257601835915089e-07, + "logits/chosen": -2.746645450592041, + "logits/rejected": -2.7202837467193604, + "logps/chosen": -285.22576904296875, + "logps/rejected": -271.9557800292969, + "loss": 0.4743, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.9407691955566406, + "rewards/margins": 1.4142510890960693, + "rewards/rejected": -2.3550198078155518, + "step": 1150 + }, + { + "epoch": 0.6, + "learning_rate": 2.2289156626506022e-07, + "logits/chosen": -2.6213831901550293, + "logits/rejected": -2.5203697681427, + "logps/chosen": -377.27337646484375, + "logps/rejected": -397.376953125, + "loss": 0.4551, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.6566792726516724, + "rewards/margins": 1.3486577272415161, + "rewards/rejected": -2.0053369998931885, + "step": 1160 + }, + { + "epoch": 0.6, + "learning_rate": 2.200229489386116e-07, + "logits/chosen": -2.8519699573516846, + "logits/rejected": -2.9094433784484863, + "logps/chosen": -343.9521789550781, + "logps/rejected": -308.2449645996094, + "loss": 0.4565, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.4127805829048157, + "rewards/margins": 0.8223036527633667, + "rewards/rejected": -1.2350841760635376, + "step": 1170 + }, + { + "epoch": 0.61, + "learning_rate": 2.1715433161216294e-07, + "logits/chosen": -2.8739943504333496, + "logits/rejected": -2.8903346061706543, + "logps/chosen": -301.22967529296875, + "logps/rejected": -251.42855834960938, + "loss": 0.4426, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.35094451904296875, + "rewards/margins": 1.0077975988388062, + "rewards/rejected": -1.3587422370910645, + "step": 1180 + }, + { + "epoch": 0.61, + "learning_rate": 2.1428571428571426e-07, + "logits/chosen": -2.892425060272217, + "logits/rejected": -2.8500430583953857, + "logps/chosen": -328.62652587890625, + "logps/rejected": -307.386474609375, + "loss": 0.4722, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.8684075474739075, + "rewards/margins": 1.0192383527755737, + "rewards/rejected": -1.887645959854126, + "step": 1190 + }, + { + "epoch": 0.62, + "learning_rate": 2.1141709695926563e-07, + "logits/chosen": -2.910890817642212, + "logits/rejected": -2.8874032497406006, + "logps/chosen": -265.87310791015625, + "logps/rejected": -210.3219451904297, + "loss": 0.4737, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.5260100960731506, + "rewards/margins": 0.7490276098251343, + "rewards/rejected": -1.2750377655029297, + "step": 1200 + }, + { + "epoch": 0.62, + "eval_logits/chosen": -2.713484764099121, + "eval_logits/rejected": -2.6841983795166016, + "eval_logps/chosen": -294.5633850097656, + "eval_logps/rejected": -323.9544677734375, + "eval_loss": 0.47160398960113525, + "eval_rewards/accuracies": 0.796875, + "eval_rewards/chosen": -0.3782654404640198, + "eval_rewards/margins": 1.3232390880584717, + "eval_rewards/rejected": -1.7015043497085571, + "eval_runtime": 57.3455, + "eval_samples_per_second": 17.438, + "eval_steps_per_second": 0.279, + "step": 1200 + }, + { + "epoch": 0.62, + "learning_rate": 2.0854847963281698e-07, + "logits/chosen": -2.970987319946289, + "logits/rejected": -2.867969512939453, + "logps/chosen": -368.6965026855469, + "logps/rejected": -353.63604736328125, + "loss": 0.4535, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.4251924455165863, + "rewards/margins": 1.3024253845214844, + "rewards/rejected": -1.727617859840393, + "step": 1210 + }, + { + "epoch": 0.63, + "learning_rate": 2.0567986230636832e-07, + "logits/chosen": -2.964089870452881, + "logits/rejected": -2.933443784713745, + "logps/chosen": -293.6631164550781, + "logps/rejected": -330.43878173828125, + "loss": 0.4637, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.6487616300582886, + "rewards/margins": 1.531836986541748, + "rewards/rejected": -2.180598735809326, + "step": 1220 + }, + { + "epoch": 0.64, + "learning_rate": 2.0281124497991967e-07, + "logits/chosen": -2.9999537467956543, + "logits/rejected": -2.9158272743225098, + "logps/chosen": -342.1163330078125, + "logps/rejected": -396.15350341796875, + "loss": 0.4611, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.16167068481445312, + "rewards/margins": 1.715951919555664, + "rewards/rejected": -1.8776228427886963, + "step": 1230 + }, + { + "epoch": 0.64, + "learning_rate": 1.9994262765347102e-07, + "logits/chosen": -2.9242331981658936, + "logits/rejected": -2.841064929962158, + "logps/chosen": -337.76983642578125, + "logps/rejected": -282.3847351074219, + "loss": 0.4917, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -1.049020528793335, + "rewards/margins": 0.9485428929328918, + "rewards/rejected": -1.9975636005401611, + "step": 1240 + }, + { + "epoch": 0.65, + "learning_rate": 1.9707401032702237e-07, + "logits/chosen": -2.7807137966156006, + "logits/rejected": -2.775256872177124, + "logps/chosen": -282.36151123046875, + "logps/rejected": -339.3646545410156, + "loss": 0.4765, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.4363161623477936, + "rewards/margins": 1.3644092082977295, + "rewards/rejected": -1.8007253408432007, + "step": 1250 + }, + { + "epoch": 0.65, + "learning_rate": 1.942053930005737e-07, + "logits/chosen": -2.850567579269409, + "logits/rejected": -2.8491926193237305, + "logps/chosen": -355.41064453125, + "logps/rejected": -349.8357238769531, + "loss": 0.4739, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.5196908712387085, + "rewards/margins": 1.9353296756744385, + "rewards/rejected": -2.4550204277038574, + "step": 1260 + }, + { + "epoch": 0.66, + "learning_rate": 1.9133677567412506e-07, + "logits/chosen": -2.6965737342834473, + "logits/rejected": -2.7178375720977783, + "logps/chosen": -253.52798461914062, + "logps/rejected": -266.63330078125, + "loss": 0.4285, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.8542221784591675, + "rewards/margins": 1.580542802810669, + "rewards/rejected": -2.434765338897705, + "step": 1270 + }, + { + "epoch": 0.66, + "learning_rate": 1.884681583476764e-07, + "logits/chosen": -2.8671460151672363, + "logits/rejected": -2.8770196437835693, + "logps/chosen": -395.8647155761719, + "logps/rejected": -322.73565673828125, + "loss": 0.5017, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.8471239805221558, + "rewards/margins": 0.8821918368339539, + "rewards/rejected": -1.729315996170044, + "step": 1280 + }, + { + "epoch": 0.67, + "learning_rate": 1.8559954102122778e-07, + "logits/chosen": -2.8296945095062256, + "logits/rejected": -2.8752951622009277, + "logps/chosen": -439.872314453125, + "logps/rejected": -365.8454284667969, + "loss": 0.4554, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.5656521320343018, + "rewards/margins": 1.5323896408081055, + "rewards/rejected": -2.0980417728424072, + "step": 1290 + }, + { + "epoch": 0.67, + "learning_rate": 1.827309236947791e-07, + "logits/chosen": -2.946869134902954, + "logits/rejected": -2.8591227531433105, + "logps/chosen": -316.06512451171875, + "logps/rejected": -311.39892578125, + "loss": 0.4259, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.757400631904602, + "rewards/margins": 1.6462675333023071, + "rewards/rejected": -2.40366792678833, + "step": 1300 + }, + { + "epoch": 0.67, + "eval_logits/chosen": -2.735586166381836, + "eval_logits/rejected": -2.7046470642089844, + "eval_logps/chosen": -299.4761047363281, + "eval_logps/rejected": -329.3312072753906, + "eval_loss": 0.486631840467453, + "eval_rewards/accuracies": 0.78125, + "eval_rewards/chosen": -0.6239006519317627, + "eval_rewards/margins": 1.346441626548767, + "eval_rewards/rejected": -1.9703422784805298, + "eval_runtime": 57.6013, + "eval_samples_per_second": 17.361, + "eval_steps_per_second": 0.278, + "step": 1300 + }, + { + "epoch": 0.68, + "learning_rate": 1.7986230636833047e-07, + "logits/chosen": -2.9283134937286377, + "logits/rejected": -2.797074556350708, + "logps/chosen": -277.39044189453125, + "logps/rejected": -298.46978759765625, + "loss": 0.4063, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -0.8742395639419556, + "rewards/margins": 0.7818448543548584, + "rewards/rejected": -1.656084418296814, + "step": 1310 + }, + { + "epoch": 0.68, + "learning_rate": 1.769936890418818e-07, + "logits/chosen": -2.8751561641693115, + "logits/rejected": -2.8942153453826904, + "logps/chosen": -327.19158935546875, + "logps/rejected": -371.1598205566406, + "loss": 0.4662, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.4851578176021576, + "rewards/margins": 1.8855422735214233, + "rewards/rejected": -2.3707003593444824, + "step": 1320 + }, + { + "epoch": 0.69, + "learning_rate": 1.7412507171543314e-07, + "logits/chosen": -2.869723081588745, + "logits/rejected": -2.791024684906006, + "logps/chosen": -299.9261474609375, + "logps/rejected": -240.87722778320312, + "loss": 0.5043, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -1.252393364906311, + "rewards/margins": 0.8984333872795105, + "rewards/rejected": -2.1508266925811768, + "step": 1330 + }, + { + "epoch": 0.69, + "learning_rate": 1.7125645438898452e-07, + "logits/chosen": -2.8324294090270996, + "logits/rejected": -2.8103926181793213, + "logps/chosen": -360.5182800292969, + "logps/rejected": -295.58953857421875, + "loss": 0.4544, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.9667569994926453, + "rewards/margins": 0.8512316942214966, + "rewards/rejected": -1.8179887533187866, + "step": 1340 + }, + { + "epoch": 0.7, + "learning_rate": 1.6838783706253584e-07, + "logits/chosen": -2.848896026611328, + "logits/rejected": -2.7923531532287598, + "logps/chosen": -372.48175048828125, + "logps/rejected": -274.67169189453125, + "loss": 0.5062, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.6907297372817993, + "rewards/margins": 1.647875189781189, + "rewards/rejected": -2.3386049270629883, + "step": 1350 + }, + { + "epoch": 0.7, + "learning_rate": 1.655192197360872e-07, + "logits/chosen": -3.0092597007751465, + "logits/rejected": -3.0115504264831543, + "logps/chosen": -414.8208923339844, + "logps/rejected": -411.04327392578125, + "loss": 0.5338, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.2658500671386719, + "rewards/margins": 1.0859276056289673, + "rewards/rejected": -1.3517776727676392, + "step": 1360 + }, + { + "epoch": 0.71, + "learning_rate": 1.6265060240963853e-07, + "logits/chosen": -2.947781801223755, + "logits/rejected": -2.9552831649780273, + "logps/chosen": -383.8572082519531, + "logps/rejected": -299.138427734375, + "loss": 0.5202, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.4735872149467468, + "rewards/margins": 1.4772822856903076, + "rewards/rejected": -1.9508693218231201, + "step": 1370 + }, + { + "epoch": 0.71, + "learning_rate": 1.597819850831899e-07, + "logits/chosen": -2.812499523162842, + "logits/rejected": -2.7452359199523926, + "logps/chosen": -339.11981201171875, + "logps/rejected": -291.80035400390625, + "loss": 0.513, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.7339299917221069, + "rewards/margins": 1.1406757831573486, + "rewards/rejected": -1.8746061325073242, + "step": 1380 + }, + { + "epoch": 0.72, + "learning_rate": 1.5691336775674125e-07, + "logits/chosen": -2.7743217945098877, + "logits/rejected": -2.7072505950927734, + "logps/chosen": -318.3100891113281, + "logps/rejected": -287.02581787109375, + "loss": 0.5028, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.30296844244003296, + "rewards/margins": 1.0583657026290894, + "rewards/rejected": -1.3613340854644775, + "step": 1390 + }, + { + "epoch": 0.72, + "learning_rate": 1.5404475043029257e-07, + "logits/chosen": -2.930776834487915, + "logits/rejected": -2.942199468612671, + "logps/chosen": -239.4253387451172, + "logps/rejected": -327.1017150878906, + "loss": 0.4935, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.503673255443573, + "rewards/margins": 0.9641939401626587, + "rewards/rejected": -1.4678672552108765, + "step": 1400 + }, + { + "epoch": 0.72, + "eval_logits/chosen": -2.7443745136260986, + "eval_logits/rejected": -2.71527099609375, + "eval_logps/chosen": -298.24908447265625, + "eval_logps/rejected": -325.1242980957031, + "eval_loss": 0.4747045338153839, + "eval_rewards/accuracies": 0.78125, + "eval_rewards/chosen": -0.5625503063201904, + "eval_rewards/margins": 1.1974470615386963, + "eval_rewards/rejected": -1.7599973678588867, + "eval_runtime": 59.3107, + "eval_samples_per_second": 16.86, + "eval_steps_per_second": 0.27, + "step": 1400 + }, + { + "epoch": 0.73, + "learning_rate": 1.5117613310384395e-07, + "logits/chosen": -2.92527437210083, + "logits/rejected": -3.0139377117156982, + "logps/chosen": -237.60580444335938, + "logps/rejected": -268.05389404296875, + "loss": 0.4175, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.6132401823997498, + "rewards/margins": 1.2935967445373535, + "rewards/rejected": -1.9068371057510376, + "step": 1410 + }, + { + "epoch": 0.73, + "learning_rate": 1.483075157773953e-07, + "logits/chosen": -2.7750067710876465, + "logits/rejected": -2.552576780319214, + "logps/chosen": -268.2431945800781, + "logps/rejected": -310.37359619140625, + "loss": 0.531, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.5963152647018433, + "rewards/margins": 0.873083770275116, + "rewards/rejected": -1.469399094581604, + "step": 1420 + }, + { + "epoch": 0.74, + "learning_rate": 1.4543889845094664e-07, + "logits/chosen": -2.900043249130249, + "logits/rejected": -2.9251770973205566, + "logps/chosen": -294.197998046875, + "logps/rejected": -420.86981201171875, + "loss": 0.5, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.7220600247383118, + "rewards/margins": 1.1596812009811401, + "rewards/rejected": -1.8817412853240967, + "step": 1430 + }, + { + "epoch": 0.74, + "learning_rate": 1.42570281124498e-07, + "logits/chosen": -2.8284263610839844, + "logits/rejected": -2.861384630203247, + "logps/chosen": -330.3293762207031, + "logps/rejected": -352.83013916015625, + "loss": 0.5019, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.5237518548965454, + "rewards/margins": 1.484616756439209, + "rewards/rejected": -2.008368730545044, + "step": 1440 + }, + { + "epoch": 0.75, + "learning_rate": 1.3970166379804933e-07, + "logits/chosen": -2.9035301208496094, + "logits/rejected": -2.877016544342041, + "logps/chosen": -330.01885986328125, + "logps/rejected": -309.932373046875, + "loss": 0.4207, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.6725677847862244, + "rewards/margins": 1.2416926622390747, + "rewards/rejected": -1.9142605066299438, + "step": 1450 + }, + { + "epoch": 0.75, + "learning_rate": 1.3683304647160068e-07, + "logits/chosen": -2.8522450923919678, + "logits/rejected": -2.804635763168335, + "logps/chosen": -315.32598876953125, + "logps/rejected": -319.85528564453125, + "loss": 0.484, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.6710207462310791, + "rewards/margins": 2.302441120147705, + "rewards/rejected": -2.973461627960205, + "step": 1460 + }, + { + "epoch": 0.76, + "learning_rate": 1.3396442914515203e-07, + "logits/chosen": -2.795161008834839, + "logits/rejected": -2.835954189300537, + "logps/chosen": -301.64306640625, + "logps/rejected": -345.019287109375, + "loss": 0.5058, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.31656789779663086, + "rewards/margins": 1.5656077861785889, + "rewards/rejected": -1.8821756839752197, + "step": 1470 + }, + { + "epoch": 0.76, + "learning_rate": 1.3109581181870338e-07, + "logits/chosen": -2.8382556438446045, + "logits/rejected": -2.7719860076904297, + "logps/chosen": -416.19622802734375, + "logps/rejected": -278.07989501953125, + "loss": 0.3959, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.7636359930038452, + "rewards/margins": 2.003495216369629, + "rewards/rejected": -2.7671313285827637, + "step": 1480 + }, + { + "epoch": 0.77, + "learning_rate": 1.2822719449225472e-07, + "logits/chosen": -2.7587904930114746, + "logits/rejected": -2.747542381286621, + "logps/chosen": -255.3273468017578, + "logps/rejected": -239.77218627929688, + "loss": 0.4899, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.0010371461976319551, + "rewards/margins": 2.0150508880615234, + "rewards/rejected": -2.0160880088806152, + "step": 1490 + }, + { + "epoch": 0.77, + "learning_rate": 1.253585771658061e-07, + "logits/chosen": -2.7033538818359375, + "logits/rejected": -2.7172279357910156, + "logps/chosen": -317.5029296875, + "logps/rejected": -332.539794921875, + "loss": 0.4211, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.8115976452827454, + "rewards/margins": 1.1709970235824585, + "rewards/rejected": -1.9825948476791382, + "step": 1500 + }, + { + "epoch": 0.77, + "eval_logits/chosen": -2.7236168384552, + "eval_logits/rejected": -2.6944026947021484, + "eval_logps/chosen": -299.1958923339844, + "eval_logps/rejected": -329.9108581542969, + "eval_loss": 0.4644867181777954, + "eval_rewards/accuracies": 0.765625, + "eval_rewards/chosen": -0.6098894476890564, + "eval_rewards/margins": 1.389434814453125, + "eval_rewards/rejected": -1.9993242025375366, + "eval_runtime": 57.1288, + "eval_samples_per_second": 17.504, + "eval_steps_per_second": 0.28, + "step": 1500 + }, + { + "epoch": 0.78, + "learning_rate": 1.2248995983935742e-07, + "logits/chosen": -2.7994678020477295, + "logits/rejected": -2.593963384628296, + "logps/chosen": -300.65960693359375, + "logps/rejected": -338.9661560058594, + "loss": 0.4655, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.44014763832092285, + "rewards/margins": 2.050450325012207, + "rewards/rejected": -2.49059796333313, + "step": 1510 + }, + { + "epoch": 0.78, + "learning_rate": 1.1962134251290876e-07, + "logits/chosen": -2.8634955883026123, + "logits/rejected": -2.9231014251708984, + "logps/chosen": -350.2730407714844, + "logps/rejected": -394.0359191894531, + "loss": 0.4687, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.7544436454772949, + "rewards/margins": 0.6226231455802917, + "rewards/rejected": -1.377066731452942, + "step": 1520 + }, + { + "epoch": 0.79, + "learning_rate": 1.1675272518646012e-07, + "logits/chosen": -2.714735984802246, + "logits/rejected": -2.6250107288360596, + "logps/chosen": -271.64691162109375, + "logps/rejected": -272.4801330566406, + "loss": 0.3846, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": 0.06765889376401901, + "rewards/margins": 1.8679250478744507, + "rewards/rejected": -1.800265908241272, + "step": 1530 + }, + { + "epoch": 0.8, + "learning_rate": 1.1388410786001147e-07, + "logits/chosen": -2.952043294906616, + "logits/rejected": -2.9354848861694336, + "logps/chosen": -237.10208129882812, + "logps/rejected": -248.58023071289062, + "loss": 0.4791, + "rewards/accuracies": 0.5, + "rewards/chosen": -0.8253666162490845, + "rewards/margins": 0.6920875906944275, + "rewards/rejected": -1.5174543857574463, + "step": 1540 + }, + { + "epoch": 0.8, + "learning_rate": 1.1101549053356282e-07, + "logits/chosen": -2.839200735092163, + "logits/rejected": -2.827392101287842, + "logps/chosen": -347.25390625, + "logps/rejected": -284.7652893066406, + "loss": 0.4351, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.3293851315975189, + "rewards/margins": 2.0348496437072754, + "rewards/rejected": -2.3642349243164062, + "step": 1550 + }, + { + "epoch": 0.81, + "learning_rate": 1.0814687320711418e-07, + "logits/chosen": -2.853760242462158, + "logits/rejected": -2.728318214416504, + "logps/chosen": -224.41683959960938, + "logps/rejected": -228.2812042236328, + "loss": 0.4528, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.3215061128139496, + "rewards/margins": 1.6524702310562134, + "rewards/rejected": -1.9739763736724854, + "step": 1560 + }, + { + "epoch": 0.81, + "learning_rate": 1.0527825588066551e-07, + "logits/chosen": -2.870434284210205, + "logits/rejected": -2.793577194213867, + "logps/chosen": -282.13787841796875, + "logps/rejected": -271.94915771484375, + "loss": 0.4144, + "rewards/accuracies": 0.949999988079071, + "rewards/chosen": -0.5694266557693481, + "rewards/margins": 1.3067474365234375, + "rewards/rejected": -1.8761742115020752, + "step": 1570 + }, + { + "epoch": 0.82, + "learning_rate": 1.0240963855421686e-07, + "logits/chosen": -2.73341703414917, + "logits/rejected": -2.7930026054382324, + "logps/chosen": -214.6037139892578, + "logps/rejected": -285.880859375, + "loss": 0.4287, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.6492952108383179, + "rewards/margins": 1.3862565755844116, + "rewards/rejected": -2.0355517864227295, + "step": 1580 + }, + { + "epoch": 0.82, + "learning_rate": 9.95410212277682e-08, + "logits/chosen": -2.885023593902588, + "logits/rejected": -2.929999351501465, + "logps/chosen": -294.29364013671875, + "logps/rejected": -273.53729248046875, + "loss": 0.4548, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.3889610171318054, + "rewards/margins": 1.3147281408309937, + "rewards/rejected": -1.7036889791488647, + "step": 1590 + }, + { + "epoch": 0.83, + "learning_rate": 9.667240390131957e-08, + "logits/chosen": -2.8551316261291504, + "logits/rejected": -2.913240432739258, + "logps/chosen": -301.7610168457031, + "logps/rejected": -309.9018249511719, + "loss": 0.4931, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.7070005536079407, + "rewards/margins": 1.0328757762908936, + "rewards/rejected": -1.7398761510849, + "step": 1600 + }, + { + "epoch": 0.83, + "eval_logits/chosen": -2.7304749488830566, + "eval_logits/rejected": -2.7005884647369385, + "eval_logps/chosen": -300.5933532714844, + "eval_logps/rejected": -332.0889892578125, + "eval_loss": 0.4683745205402374, + "eval_rewards/accuracies": 0.765625, + "eval_rewards/chosen": -0.6797637343406677, + "eval_rewards/margins": 1.4284672737121582, + "eval_rewards/rejected": -2.1082310676574707, + "eval_runtime": 58.1263, + "eval_samples_per_second": 17.204, + "eval_steps_per_second": 0.275, + "step": 1600 + }, + { + "epoch": 0.83, + "learning_rate": 9.380378657487091e-08, + "logits/chosen": -2.786533832550049, + "logits/rejected": -2.8000881671905518, + "logps/chosen": -252.6224822998047, + "logps/rejected": -307.4824523925781, + "loss": 0.4282, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.869725227355957, + "rewards/margins": 1.490140438079834, + "rewards/rejected": -2.359865665435791, + "step": 1610 + }, + { + "epoch": 0.84, + "learning_rate": 9.093516924842226e-08, + "logits/chosen": -2.9314494132995605, + "logits/rejected": -2.888124942779541, + "logps/chosen": -322.18988037109375, + "logps/rejected": -340.00958251953125, + "loss": 0.4095, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.7641150951385498, + "rewards/margins": 1.8909038305282593, + "rewards/rejected": -2.6550190448760986, + "step": 1620 + }, + { + "epoch": 0.84, + "learning_rate": 8.806655192197361e-08, + "logits/chosen": -2.91792368888855, + "logits/rejected": -2.8502402305603027, + "logps/chosen": -327.46807861328125, + "logps/rejected": -310.23199462890625, + "loss": 0.4625, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.5610017776489258, + "rewards/margins": 1.5278574228286743, + "rewards/rejected": -2.0888590812683105, + "step": 1630 + }, + { + "epoch": 0.85, + "learning_rate": 8.519793459552494e-08, + "logits/chosen": -2.8847482204437256, + "logits/rejected": -2.8378400802612305, + "logps/chosen": -315.1554870605469, + "logps/rejected": -282.12652587890625, + "loss": 0.418, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": -0.6100664138793945, + "rewards/margins": 1.6232362985610962, + "rewards/rejected": -2.233302593231201, + "step": 1640 + }, + { + "epoch": 0.85, + "learning_rate": 8.23293172690763e-08, + "logits/chosen": -2.943779230117798, + "logits/rejected": -2.894235372543335, + "logps/chosen": -380.72894287109375, + "logps/rejected": -279.9917907714844, + "loss": 0.4489, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": 0.3063332736492157, + "rewards/margins": 2.2862744331359863, + "rewards/rejected": -1.9799413681030273, + "step": 1650 + }, + { + "epoch": 0.86, + "learning_rate": 7.946069994262765e-08, + "logits/chosen": -2.738778829574585, + "logits/rejected": -2.805850028991699, + "logps/chosen": -221.29507446289062, + "logps/rejected": -314.97076416015625, + "loss": 0.4268, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.5129801034927368, + "rewards/margins": 1.8285270929336548, + "rewards/rejected": -2.3415069580078125, + "step": 1660 + }, + { + "epoch": 0.86, + "learning_rate": 7.6592082616179e-08, + "logits/chosen": -2.73237681388855, + "logits/rejected": -2.7697629928588867, + "logps/chosen": -295.9866638183594, + "logps/rejected": -282.50189208984375, + "loss": 0.5111, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.5220045447349548, + "rewards/margins": 1.3860810995101929, + "rewards/rejected": -1.908085823059082, + "step": 1670 + }, + { + "epoch": 0.87, + "learning_rate": 7.372346528973034e-08, + "logits/chosen": -2.9239017963409424, + "logits/rejected": -2.8694822788238525, + "logps/chosen": -266.6614074707031, + "logps/rejected": -303.3201599121094, + "loss": 0.5028, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.5870487689971924, + "rewards/margins": 0.9326974749565125, + "rewards/rejected": -1.5197464227676392, + "step": 1680 + }, + { + "epoch": 0.87, + "learning_rate": 7.08548479632817e-08, + "logits/chosen": -2.8852458000183105, + "logits/rejected": -2.9049694538116455, + "logps/chosen": -256.247802734375, + "logps/rejected": -329.19305419921875, + "loss": 0.4537, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.6385833024978638, + "rewards/margins": 1.4285590648651123, + "rewards/rejected": -2.0671424865722656, + "step": 1690 + }, + { + "epoch": 0.88, + "learning_rate": 6.798623063683305e-08, + "logits/chosen": -2.8753390312194824, + "logits/rejected": -2.7723491191864014, + "logps/chosen": -285.18365478515625, + "logps/rejected": -267.1637878417969, + "loss": 0.5029, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.7313726544380188, + "rewards/margins": 1.1651675701141357, + "rewards/rejected": -1.8965400457382202, + "step": 1700 + }, + { + "epoch": 0.88, + "eval_logits/chosen": -2.740252733230591, + "eval_logits/rejected": -2.710822343826294, + "eval_logps/chosen": -297.1233215332031, + "eval_logps/rejected": -327.8266906738281, + "eval_loss": 0.4595467746257782, + "eval_rewards/accuracies": 0.78125, + "eval_rewards/chosen": -0.5062620639801025, + "eval_rewards/margins": 1.3888535499572754, + "eval_rewards/rejected": -1.895115852355957, + "eval_runtime": 55.2023, + "eval_samples_per_second": 18.115, + "eval_steps_per_second": 0.29, + "step": 1700 + }, + { + "epoch": 0.88, + "learning_rate": 6.511761331038438e-08, + "logits/chosen": -2.895554542541504, + "logits/rejected": -2.7735402584075928, + "logps/chosen": -375.2083435058594, + "logps/rejected": -386.50689697265625, + "loss": 0.4635, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.7999193072319031, + "rewards/margins": 1.7455612421035767, + "rewards/rejected": -2.545480728149414, + "step": 1710 + }, + { + "epoch": 0.89, + "learning_rate": 6.224899598393573e-08, + "logits/chosen": -2.897815704345703, + "logits/rejected": -2.833980083465576, + "logps/chosen": -349.2824401855469, + "logps/rejected": -320.26080322265625, + "loss": 0.5059, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.8347634077072144, + "rewards/margins": 1.1928595304489136, + "rewards/rejected": -2.027622938156128, + "step": 1720 + }, + { + "epoch": 0.89, + "learning_rate": 5.9380378657487085e-08, + "logits/chosen": -2.747729539871216, + "logits/rejected": -2.749624490737915, + "logps/chosen": -389.26104736328125, + "logps/rejected": -379.7428283691406, + "loss": 0.4561, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -0.46041926741600037, + "rewards/margins": 1.0110886096954346, + "rewards/rejected": -1.4715079069137573, + "step": 1730 + }, + { + "epoch": 0.9, + "learning_rate": 5.651176133103844e-08, + "logits/chosen": -2.9265472888946533, + "logits/rejected": -2.7790768146514893, + "logps/chosen": -344.9275817871094, + "logps/rejected": -267.4143371582031, + "loss": 0.4472, + "rewards/accuracies": 0.949999988079071, + "rewards/chosen": -0.3255007266998291, + "rewards/margins": 2.0611414909362793, + "rewards/rejected": -2.3866424560546875, + "step": 1740 + }, + { + "epoch": 0.9, + "learning_rate": 5.3643144004589786e-08, + "logits/chosen": -2.8731534481048584, + "logits/rejected": -2.7964863777160645, + "logps/chosen": -293.29681396484375, + "logps/rejected": -305.7369079589844, + "loss": 0.4481, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.3717958629131317, + "rewards/margins": 1.912921667098999, + "rewards/rejected": -2.2847177982330322, + "step": 1750 + }, + { + "epoch": 0.91, + "learning_rate": 5.077452667814113e-08, + "logits/chosen": -2.771488666534424, + "logits/rejected": -2.865757465362549, + "logps/chosen": -309.177978515625, + "logps/rejected": -285.9333801269531, + "loss": 0.4892, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.7683748602867126, + "rewards/margins": 1.0694787502288818, + "rewards/rejected": -1.8378536701202393, + "step": 1760 + }, + { + "epoch": 0.91, + "learning_rate": 4.790590935169248e-08, + "logits/chosen": -2.8640894889831543, + "logits/rejected": -2.8574938774108887, + "logps/chosen": -270.6001281738281, + "logps/rejected": -282.6536865234375, + "loss": 0.4509, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.5418082475662231, + "rewards/margins": 1.5019443035125732, + "rewards/rejected": -2.043752670288086, + "step": 1770 + }, + { + "epoch": 0.92, + "learning_rate": 4.5037292025243834e-08, + "logits/chosen": -2.916822910308838, + "logits/rejected": -2.945218086242676, + "logps/chosen": -330.62493896484375, + "logps/rejected": -340.20599365234375, + "loss": 0.4988, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.980034351348877, + "rewards/margins": 1.0700609683990479, + "rewards/rejected": -2.0500950813293457, + "step": 1780 + }, + { + "epoch": 0.92, + "learning_rate": 4.2168674698795174e-08, + "logits/chosen": -2.9247031211853027, + "logits/rejected": -2.9124653339385986, + "logps/chosen": -289.2491149902344, + "logps/rejected": -264.84283447265625, + "loss": 0.4577, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.3735916018486023, + "rewards/margins": 0.9222102165222168, + "rewards/rejected": -1.2958018779754639, + "step": 1790 + }, + { + "epoch": 0.93, + "learning_rate": 3.930005737234653e-08, + "logits/chosen": -2.9057695865631104, + "logits/rejected": -2.819000005722046, + "logps/chosen": -337.3312683105469, + "logps/rejected": -276.67010498046875, + "loss": 0.4965, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.23481640219688416, + "rewards/margins": 1.076974868774414, + "rewards/rejected": -1.3117913007736206, + "step": 1800 + }, + { + "epoch": 0.93, + "eval_logits/chosen": -2.752323627471924, + "eval_logits/rejected": -2.7225875854492188, + "eval_logps/chosen": -298.1203308105469, + "eval_logps/rejected": -328.0830993652344, + "eval_loss": 0.46130359172821045, + "eval_rewards/accuracies": 0.78125, + "eval_rewards/chosen": -0.5561107397079468, + "eval_rewards/margins": 1.3518264293670654, + "eval_rewards/rejected": -1.9079371690750122, + "eval_runtime": 57.0782, + "eval_samples_per_second": 17.52, + "eval_steps_per_second": 0.28, + "step": 1800 + }, + { + "epoch": 0.93, + "learning_rate": 3.6431440045897875e-08, + "logits/chosen": -2.8899993896484375, + "logits/rejected": -2.886475086212158, + "logps/chosen": -212.6807861328125, + "logps/rejected": -243.3516082763672, + "loss": 0.4966, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.7232971787452698, + "rewards/margins": 1.2528116703033447, + "rewards/rejected": -1.9761087894439697, + "step": 1810 + }, + { + "epoch": 0.94, + "learning_rate": 3.356282271944923e-08, + "logits/chosen": -2.936781406402588, + "logits/rejected": -2.913865566253662, + "logps/chosen": -259.8666076660156, + "logps/rejected": -268.0596008300781, + "loss": 0.4455, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.7119141221046448, + "rewards/margins": 1.462996244430542, + "rewards/rejected": -2.174910306930542, + "step": 1820 + }, + { + "epoch": 0.94, + "learning_rate": 3.0694205393000576e-08, + "logits/chosen": -2.8601667881011963, + "logits/rejected": -2.8055875301361084, + "logps/chosen": -324.9699401855469, + "logps/rejected": -261.33355712890625, + "loss": 0.4484, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.6698704361915588, + "rewards/margins": 1.4717018604278564, + "rewards/rejected": -2.1415722370147705, + "step": 1830 + }, + { + "epoch": 0.95, + "learning_rate": 2.782558806655192e-08, + "logits/chosen": -2.9527735710144043, + "logits/rejected": -2.9381892681121826, + "logps/chosen": -338.42889404296875, + "logps/rejected": -293.7242126464844, + "loss": 0.4437, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.5508059859275818, + "rewards/margins": 0.5796544551849365, + "rewards/rejected": -1.130460500717163, + "step": 1840 + }, + { + "epoch": 0.96, + "learning_rate": 2.4956970740103267e-08, + "logits/chosen": -2.8355095386505127, + "logits/rejected": -2.847282648086548, + "logps/chosen": -303.50482177734375, + "logps/rejected": -281.2980041503906, + "loss": 0.4709, + "rewards/accuracies": 0.949999988079071, + "rewards/chosen": -0.2886872887611389, + "rewards/margins": 2.1929566860198975, + "rewards/rejected": -2.4816439151763916, + "step": 1850 + }, + { + "epoch": 0.96, + "learning_rate": 2.2088353413654617e-08, + "logits/chosen": -2.8765408992767334, + "logits/rejected": -2.8786520957946777, + "logps/chosen": -321.27203369140625, + "logps/rejected": -308.62628173828125, + "loss": 0.4371, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.6084128618240356, + "rewards/margins": 1.1289150714874268, + "rewards/rejected": -1.7373279333114624, + "step": 1860 + }, + { + "epoch": 0.97, + "learning_rate": 1.9219736087205964e-08, + "logits/chosen": -2.8658809661865234, + "logits/rejected": -2.86934232711792, + "logps/chosen": -271.0024719238281, + "logps/rejected": -249.09765625, + "loss": 0.4678, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.7377957105636597, + "rewards/margins": 0.9156107902526855, + "rewards/rejected": -1.6534065008163452, + "step": 1870 + }, + { + "epoch": 0.97, + "learning_rate": 1.6351118760757314e-08, + "logits/chosen": -2.7913639545440674, + "logits/rejected": -2.7344272136688232, + "logps/chosen": -299.87823486328125, + "logps/rejected": -340.60467529296875, + "loss": 0.4679, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.9796409606933594, + "rewards/margins": 1.4006431102752686, + "rewards/rejected": -2.380284070968628, + "step": 1880 + }, + { + "epoch": 0.98, + "learning_rate": 1.3482501434308661e-08, + "logits/chosen": -2.9909162521362305, + "logits/rejected": -2.8788750171661377, + "logps/chosen": -316.1296081542969, + "logps/rejected": -329.0151062011719, + "loss": 0.4584, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.6042786836624146, + "rewards/margins": 1.349906325340271, + "rewards/rejected": -1.954184889793396, + "step": 1890 + }, + { + "epoch": 0.98, + "learning_rate": 1.061388410786001e-08, + "logits/chosen": -2.840143918991089, + "logits/rejected": -2.8358423709869385, + "logps/chosen": -347.96185302734375, + "logps/rejected": -336.83831787109375, + "loss": 0.4337, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.25721627473831177, + "rewards/margins": 1.7318767309188843, + "rewards/rejected": -1.9890931844711304, + "step": 1900 + }, + { + "epoch": 0.98, + "eval_logits/chosen": -2.746859073638916, + "eval_logits/rejected": -2.717475175857544, + "eval_logps/chosen": -297.12957763671875, + "eval_logps/rejected": -327.35992431640625, + "eval_loss": 0.4607921838760376, + "eval_rewards/accuracies": 0.765625, + "eval_rewards/chosen": -0.5065746307373047, + "eval_rewards/margins": 1.365204095840454, + "eval_rewards/rejected": -1.871778964996338, + "eval_runtime": 59.1698, + "eval_samples_per_second": 16.901, + "eval_steps_per_second": 0.27, + "step": 1900 + }, + { + "epoch": 0.99, + "learning_rate": 7.745266781411359e-09, + "logits/chosen": -2.944617986679077, + "logits/rejected": -2.859666347503662, + "logps/chosen": -312.51654052734375, + "logps/rejected": -263.56494140625, + "loss": 0.4129, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -0.29929789900779724, + "rewards/margins": 0.7224828004837036, + "rewards/rejected": -1.0217806100845337, + "step": 1910 + }, + { + "epoch": 0.99, + "learning_rate": 4.8766494549627085e-09, + "logits/chosen": -2.8194212913513184, + "logits/rejected": -2.776564598083496, + "logps/chosen": -363.00396728515625, + "logps/rejected": -299.03338623046875, + "loss": 0.4647, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -0.38584208488464355, + "rewards/margins": 0.8067569732666016, + "rewards/rejected": -1.1925990581512451, + "step": 1920 + }, + { + "epoch": 1.0, + "learning_rate": 2.008032128514056e-09, + "logits/chosen": -2.8939270973205566, + "logits/rejected": -2.778672933578491, + "logps/chosen": -323.1568298339844, + "logps/rejected": -284.89361572265625, + "loss": 0.4376, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": -0.581506609916687, + "rewards/margins": 1.8644784688949585, + "rewards/rejected": -2.4459850788116455, + "step": 1930 + }, + { + "epoch": 1.0, + "step": 1937, + "total_flos": 0.0, + "train_loss": 0.48803097129668166, + "train_runtime": 7971.1784, + "train_samples_per_second": 7.774, + "train_steps_per_second": 0.243 + } + ], + "logging_steps": 10, + "max_steps": 1937, + "num_train_epochs": 1, + "save_steps": 500, + "total_flos": 0.0, + "trial_name": null, + "trial_params": null +}