| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.9984, | |
| "eval_steps": 100, | |
| "global_step": 312, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.0, | |
| "learning_rate": 1.5625e-08, | |
| "logits/chosen": -3.15263032913208, | |
| "logits/rejected": -3.167269468307495, | |
| "logps/chosen": -410.4931945800781, | |
| "logps/rejected": -418.35675048828125, | |
| "loss": 0.6931, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "learning_rate": 1.5624999999999999e-07, | |
| "logits/chosen": -3.1613829135894775, | |
| "logits/rejected": -3.1396090984344482, | |
| "logps/chosen": -395.172607421875, | |
| "logps/rejected": -445.29937744140625, | |
| "loss": 0.6929, | |
| "rewards/accuracies": 0.4791666567325592, | |
| "rewards/chosen": 0.0008217308204621077, | |
| "rewards/margins": 0.0008943447028286755, | |
| "rewards/rejected": -7.261387509061024e-05, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "learning_rate": 3.1249999999999997e-07, | |
| "logits/chosen": -3.176825761795044, | |
| "logits/rejected": -3.1581790447235107, | |
| "logps/chosen": -407.73675537109375, | |
| "logps/rejected": -476.7193298339844, | |
| "loss": 0.6907, | |
| "rewards/accuracies": 0.606249988079071, | |
| "rewards/chosen": -0.006854791194200516, | |
| "rewards/margins": 0.0052978708408772945, | |
| "rewards/rejected": -0.012152662500739098, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "learning_rate": 4.6874999999999996e-07, | |
| "logits/chosen": -3.1266303062438965, | |
| "logits/rejected": -3.104175567626953, | |
| "logps/chosen": -379.49029541015625, | |
| "logps/rejected": -453.5487365722656, | |
| "loss": 0.6765, | |
| "rewards/accuracies": 0.71875, | |
| "rewards/chosen": -0.059107840061187744, | |
| "rewards/margins": 0.03545823693275452, | |
| "rewards/rejected": -0.09456606954336166, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "learning_rate": 4.989935734988097e-07, | |
| "logits/chosen": -3.2789883613586426, | |
| "logits/rejected": -3.2879974842071533, | |
| "logps/chosen": -407.4563903808594, | |
| "logps/rejected": -472.73638916015625, | |
| "loss": 0.649, | |
| "rewards/accuracies": 0.675000011920929, | |
| "rewards/chosen": -0.3158263564109802, | |
| "rewards/margins": 0.08948425948619843, | |
| "rewards/rejected": -0.40531063079833984, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "learning_rate": 4.949188496058089e-07, | |
| "logits/chosen": -3.3798394203186035, | |
| "logits/rejected": -3.365309238433838, | |
| "logps/chosen": -461.53118896484375, | |
| "logps/rejected": -543.4600830078125, | |
| "loss": 0.6241, | |
| "rewards/accuracies": 0.737500011920929, | |
| "rewards/chosen": -0.7056063413619995, | |
| "rewards/margins": 0.2066904753446579, | |
| "rewards/rejected": -0.9122966527938843, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "learning_rate": 4.877641290737883e-07, | |
| "logits/chosen": -3.391582489013672, | |
| "logits/rejected": -3.3562042713165283, | |
| "logps/chosen": -509.7789611816406, | |
| "logps/rejected": -615.9661254882812, | |
| "loss": 0.5795, | |
| "rewards/accuracies": 0.6812499761581421, | |
| "rewards/chosen": -1.1751911640167236, | |
| "rewards/margins": 0.28747081756591797, | |
| "rewards/rejected": -1.4626619815826416, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "learning_rate": 4.776193866647039e-07, | |
| "logits/chosen": -3.2612807750701904, | |
| "logits/rejected": -3.2683074474334717, | |
| "logps/chosen": -575.8006591796875, | |
| "logps/rejected": -681.9743041992188, | |
| "loss": 0.5936, | |
| "rewards/accuracies": 0.6812499761581421, | |
| "rewards/chosen": -1.586014747619629, | |
| "rewards/margins": 0.3645917773246765, | |
| "rewards/rejected": -1.9506065845489502, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "learning_rate": 4.646121984004665e-07, | |
| "logits/chosen": -3.149127960205078, | |
| "logits/rejected": -3.1329290866851807, | |
| "logps/chosen": -516.2041625976562, | |
| "logps/rejected": -626.0014038085938, | |
| "loss": 0.5641, | |
| "rewards/accuracies": 0.737500011920929, | |
| "rewards/chosen": -1.2160618305206299, | |
| "rewards/margins": 0.4495382308959961, | |
| "rewards/rejected": -1.6655998229980469, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 4.489061372204452e-07, | |
| "logits/chosen": -3.1035568714141846, | |
| "logits/rejected": -3.1130125522613525, | |
| "logps/chosen": -568.3084106445312, | |
| "logps/rejected": -665.2039184570312, | |
| "loss": 0.5624, | |
| "rewards/accuracies": 0.7124999761581421, | |
| "rewards/chosen": -1.4589087963104248, | |
| "rewards/margins": 0.38649308681488037, | |
| "rewards/rejected": -1.8454017639160156, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "learning_rate": 4.3069871595684787e-07, | |
| "logits/chosen": -3.124558210372925, | |
| "logits/rejected": -3.1174521446228027, | |
| "logps/chosen": -525.4788818359375, | |
| "logps/rejected": -628.5894775390625, | |
| "loss": 0.5687, | |
| "rewards/accuracies": 0.78125, | |
| "rewards/chosen": -1.3210557699203491, | |
| "rewards/margins": 0.48030009865760803, | |
| "rewards/rejected": -1.8013557195663452, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "eval_logits/chosen": -3.141606092453003, | |
| "eval_logits/rejected": -3.1296393871307373, | |
| "eval_logps/chosen": -548.8582763671875, | |
| "eval_logps/rejected": -571.9306030273438, | |
| "eval_loss": 0.7266306281089783, | |
| "eval_rewards/accuracies": 0.53515625, | |
| "eval_rewards/chosen": -1.4744807481765747, | |
| "eval_rewards/margins": 0.06807918846607208, | |
| "eval_rewards/rejected": -1.54256010055542, | |
| "eval_runtime": 126.328, | |
| "eval_samples_per_second": 7.916, | |
| "eval_steps_per_second": 0.253, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "learning_rate": 4.10218903496256e-07, | |
| "logits/chosen": -3.023545742034912, | |
| "logits/rejected": -3.021019220352173, | |
| "logps/chosen": -570.5040283203125, | |
| "logps/rejected": -650.7171630859375, | |
| "loss": 0.5714, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": -1.4776172637939453, | |
| "rewards/margins": 0.3569051921367645, | |
| "rewards/rejected": -1.8345226049423218, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.38, | |
| "learning_rate": 3.877242453630256e-07, | |
| "logits/chosen": -2.912734031677246, | |
| "logits/rejected": -2.893033266067505, | |
| "logps/chosen": -547.8372802734375, | |
| "logps/rejected": -656.3632202148438, | |
| "loss": 0.557, | |
| "rewards/accuracies": 0.768750011920929, | |
| "rewards/chosen": -1.3632628917694092, | |
| "rewards/margins": 0.49201005697250366, | |
| "rewards/rejected": -1.8552730083465576, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "learning_rate": 3.634976249348867e-07, | |
| "logits/chosen": -2.885218620300293, | |
| "logits/rejected": -2.895531415939331, | |
| "logps/chosen": -534.7477416992188, | |
| "logps/rejected": -645.1211547851562, | |
| "loss": 0.5572, | |
| "rewards/accuracies": 0.7562500238418579, | |
| "rewards/chosen": -1.410983681678772, | |
| "rewards/margins": 0.4554961621761322, | |
| "rewards/rejected": -1.8664801120758057, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.45, | |
| "learning_rate": 3.378437060203357e-07, | |
| "logits/chosen": -2.7777891159057617, | |
| "logits/rejected": -2.754775047302246, | |
| "logps/chosen": -589.5593872070312, | |
| "logps/rejected": -722.3430786132812, | |
| "loss": 0.5561, | |
| "rewards/accuracies": 0.762499988079071, | |
| "rewards/chosen": -1.5671552419662476, | |
| "rewards/margins": 0.4903177320957184, | |
| "rewards/rejected": -2.0574729442596436, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.48, | |
| "learning_rate": 3.110851015344735e-07, | |
| "logits/chosen": -2.9534642696380615, | |
| "logits/rejected": -2.938105344772339, | |
| "logps/chosen": -572.8906860351562, | |
| "logps/rejected": -684.935302734375, | |
| "loss": 0.5776, | |
| "rewards/accuracies": 0.6937500238418579, | |
| "rewards/chosen": -1.72029709815979, | |
| "rewards/margins": 0.4341273307800293, | |
| "rewards/rejected": -2.1544244289398193, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.51, | |
| "learning_rate": 2.8355831645441387e-07, | |
| "logits/chosen": -2.9231104850769043, | |
| "logits/rejected": -2.9094436168670654, | |
| "logps/chosen": -572.6766967773438, | |
| "logps/rejected": -667.0675048828125, | |
| "loss": 0.5856, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": -1.6497344970703125, | |
| "rewards/margins": 0.4040594696998596, | |
| "rewards/rejected": -2.0537939071655273, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.54, | |
| "learning_rate": 2.5560951607395126e-07, | |
| "logits/chosen": -3.066063165664673, | |
| "logits/rejected": -3.0717461109161377, | |
| "logps/chosen": -541.1444091796875, | |
| "logps/rejected": -630.5333862304688, | |
| "loss": 0.5867, | |
| "rewards/accuracies": 0.7437499761581421, | |
| "rewards/chosen": -1.3789122104644775, | |
| "rewards/margins": 0.38145893812179565, | |
| "rewards/rejected": -1.760371208190918, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.58, | |
| "learning_rate": 2.2759017277414164e-07, | |
| "logits/chosen": -3.058684825897217, | |
| "logits/rejected": -3.044579029083252, | |
| "logps/chosen": -521.42431640625, | |
| "logps/rejected": -601.6608276367188, | |
| "loss": 0.5895, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": -1.1656461954116821, | |
| "rewards/margins": 0.37542057037353516, | |
| "rewards/rejected": -1.5410667657852173, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.61, | |
| "learning_rate": 1.998526460541818e-07, | |
| "logits/chosen": -3.0458412170410156, | |
| "logits/rejected": -3.019465446472168, | |
| "logps/chosen": -549.1454467773438, | |
| "logps/rejected": -631.4022216796875, | |
| "loss": 0.5923, | |
| "rewards/accuracies": 0.643750011920929, | |
| "rewards/chosen": -1.2936315536499023, | |
| "rewards/margins": 0.31920185685157776, | |
| "rewards/rejected": -1.6128332614898682, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.64, | |
| "learning_rate": 1.7274575140626315e-07, | |
| "logits/chosen": -3.0824837684631348, | |
| "logits/rejected": -3.063490390777588, | |
| "logps/chosen": -519.8092651367188, | |
| "logps/rejected": -580.396484375, | |
| "loss": 0.5901, | |
| "rewards/accuracies": 0.65625, | |
| "rewards/chosen": -1.2900115251541138, | |
| "rewards/margins": 0.2603384256362915, | |
| "rewards/rejected": -1.5503495931625366, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.64, | |
| "eval_logits/chosen": -3.1155648231506348, | |
| "eval_logits/rejected": -3.1042416095733643, | |
| "eval_logps/chosen": -542.4268798828125, | |
| "eval_logps/rejected": -566.1749267578125, | |
| "eval_loss": 0.7176706790924072, | |
| "eval_rewards/accuracies": 0.546875, | |
| "eval_rewards/chosen": -1.4101661443710327, | |
| "eval_rewards/margins": 0.07483740895986557, | |
| "eval_rewards/rejected": -1.4850035905838013, | |
| "eval_runtime": 126.7957, | |
| "eval_samples_per_second": 7.887, | |
| "eval_steps_per_second": 0.252, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "learning_rate": 1.4661037375836987e-07, | |
| "logits/chosen": -3.0473384857177734, | |
| "logits/rejected": -3.034618377685547, | |
| "logps/chosen": -558.7247924804688, | |
| "logps/rejected": -649.9381103515625, | |
| "loss": 0.5889, | |
| "rewards/accuracies": 0.6812499761581421, | |
| "rewards/chosen": -1.4881097078323364, | |
| "rewards/margins": 0.40473729372024536, | |
| "rewards/rejected": -1.8928470611572266, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "learning_rate": 1.2177518064852348e-07, | |
| "logits/chosen": -3.0184223651885986, | |
| "logits/rejected": -3.0191338062286377, | |
| "logps/chosen": -550.5185546875, | |
| "logps/rejected": -641.130859375, | |
| "loss": 0.5914, | |
| "rewards/accuracies": 0.7124999761581421, | |
| "rewards/chosen": -1.4187710285186768, | |
| "rewards/margins": 0.4211079478263855, | |
| "rewards/rejected": -1.839879035949707, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.74, | |
| "learning_rate": 9.855248903979505e-08, | |
| "logits/chosen": -3.091451406478882, | |
| "logits/rejected": -3.0760319232940674, | |
| "logps/chosen": -546.5443115234375, | |
| "logps/rejected": -596.8983764648438, | |
| "loss": 0.6215, | |
| "rewards/accuracies": 0.6625000238418579, | |
| "rewards/chosen": -1.3793073892593384, | |
| "rewards/margins": 0.22927220165729523, | |
| "rewards/rejected": -1.6085796356201172, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.77, | |
| "learning_rate": 7.723433775328384e-08, | |
| "logits/chosen": -3.0834617614746094, | |
| "logits/rejected": -3.0846691131591797, | |
| "logps/chosen": -537.6251831054688, | |
| "logps/rejected": -614.9900512695312, | |
| "loss": 0.6082, | |
| "rewards/accuracies": 0.6812499761581421, | |
| "rewards/chosen": -1.3861608505249023, | |
| "rewards/margins": 0.38685861229896545, | |
| "rewards/rejected": -1.7730194330215454, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "learning_rate": 5.808881491049722e-08, | |
| "logits/chosen": -3.067990779876709, | |
| "logits/rejected": -3.0540151596069336, | |
| "logps/chosen": -529.740234375, | |
| "logps/rejected": -618.4736938476562, | |
| "loss": 0.6129, | |
| "rewards/accuracies": 0.706250011920929, | |
| "rewards/chosen": -1.354187250137329, | |
| "rewards/margins": 0.431583970785141, | |
| "rewards/rejected": -1.785771369934082, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "learning_rate": 4.1356686569674335e-08, | |
| "logits/chosen": -3.0292751789093018, | |
| "logits/rejected": -3.0266265869140625, | |
| "logps/chosen": -606.5236206054688, | |
| "logps/rejected": -676.3375854492188, | |
| "loss": 0.6213, | |
| "rewards/accuracies": 0.668749988079071, | |
| "rewards/chosen": -1.5674151182174683, | |
| "rewards/margins": 0.31385400891304016, | |
| "rewards/rejected": -1.8812694549560547, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.86, | |
| "learning_rate": 2.724836895290805e-08, | |
| "logits/chosen": -3.087231159210205, | |
| "logits/rejected": -3.0835347175598145, | |
| "logps/chosen": -523.4954223632812, | |
| "logps/rejected": -572.1464233398438, | |
| "loss": 0.6095, | |
| "rewards/accuracies": 0.668749988079071, | |
| "rewards/chosen": -1.3111555576324463, | |
| "rewards/margins": 0.2772643268108368, | |
| "rewards/rejected": -1.588419795036316, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.9, | |
| "learning_rate": 1.5941282340065697e-08, | |
| "logits/chosen": -3.0525717735290527, | |
| "logits/rejected": -3.0471031665802, | |
| "logps/chosen": -573.8348388671875, | |
| "logps/rejected": -643.480224609375, | |
| "loss": 0.6231, | |
| "rewards/accuracies": 0.668749988079071, | |
| "rewards/chosen": -1.4671975374221802, | |
| "rewards/margins": 0.28212133049964905, | |
| "rewards/rejected": -1.7493188381195068, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.93, | |
| "learning_rate": 7.577619905828281e-09, | |
| "logits/chosen": -3.091033458709717, | |
| "logits/rejected": -3.096426248550415, | |
| "logps/chosen": -537.3826293945312, | |
| "logps/rejected": -589.7948608398438, | |
| "loss": 0.6389, | |
| "rewards/accuracies": 0.637499988079071, | |
| "rewards/chosen": -1.4276825189590454, | |
| "rewards/margins": 0.31839436292648315, | |
| "rewards/rejected": -1.7460769414901733, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "learning_rate": 2.2625595580163247e-09, | |
| "logits/chosen": -3.074458360671997, | |
| "logits/rejected": -3.0356478691101074, | |
| "logps/chosen": -585.68505859375, | |
| "logps/rejected": -669.7423095703125, | |
| "loss": 0.6278, | |
| "rewards/accuracies": 0.6875, | |
| "rewards/chosen": -1.4561302661895752, | |
| "rewards/margins": 0.38690048456192017, | |
| "rewards/rejected": -1.8430306911468506, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "eval_logits/chosen": -3.164099931716919, | |
| "eval_logits/rejected": -3.1529414653778076, | |
| "eval_logps/chosen": -543.5320434570312, | |
| "eval_logps/rejected": -567.779052734375, | |
| "eval_loss": 0.7142006754875183, | |
| "eval_rewards/accuracies": 0.5546875, | |
| "eval_rewards/chosen": -1.4212186336517334, | |
| "eval_rewards/margins": 0.07982616126537323, | |
| "eval_rewards/rejected": -1.5010446310043335, | |
| "eval_runtime": 125.3843, | |
| "eval_samples_per_second": 7.975, | |
| "eval_steps_per_second": 0.255, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.99, | |
| "learning_rate": 6.294126437336733e-11, | |
| "logits/chosen": -3.101105213165283, | |
| "logits/rejected": -3.063551664352417, | |
| "logps/chosen": -558.6819458007812, | |
| "logps/rejected": -620.3117065429688, | |
| "loss": 0.6361, | |
| "rewards/accuracies": 0.637499988079071, | |
| "rewards/chosen": -1.3710179328918457, | |
| "rewards/margins": 0.27891039848327637, | |
| "rewards/rejected": -1.649928331375122, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "step": 312, | |
| "total_flos": 0.0, | |
| "train_loss": 0.6047722502396657, | |
| "train_runtime": 5332.3567, | |
| "train_samples_per_second": 3.75, | |
| "train_steps_per_second": 0.059 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 312, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 100, | |
| "total_flos": 0.0, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |