| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 3.0, | |
| "eval_steps": 50, | |
| "global_step": 201, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.07462686567164178, | |
| "grad_norm": 67.58726313106416, | |
| "learning_rate": 5e-07, | |
| "logits/chosen": -2.728022575378418, | |
| "logits/rejected": -2.724689245223999, | |
| "logps/chosen": -256.9372253417969, | |
| "logps/rejected": -205.8960418701172, | |
| "loss": 0.6906, | |
| "rewards/accuracies": 0.3499999940395355, | |
| "rewards/chosen": 0.012728104367852211, | |
| "rewards/margins": 0.00967978872358799, | |
| "rewards/rejected": 0.003048315178602934, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.14925373134328357, | |
| "grad_norm": 46.21946007010114, | |
| "learning_rate": 1e-06, | |
| "logits/chosen": -2.7133026123046875, | |
| "logits/rejected": -2.6811420917510986, | |
| "logps/chosen": -248.5566864013672, | |
| "logps/rejected": -197.87265014648438, | |
| "loss": 0.648, | |
| "rewards/accuracies": 0.6625000238418579, | |
| "rewards/chosen": 0.4624863564968109, | |
| "rewards/margins": 0.19924615323543549, | |
| "rewards/rejected": 0.26324018836021423, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.22388059701492538, | |
| "grad_norm": 39.90578699736001, | |
| "learning_rate": 9.983100718730718e-07, | |
| "logits/chosen": -2.5118980407714844, | |
| "logits/rejected": -2.4951248168945312, | |
| "logps/chosen": -241.9331512451172, | |
| "logps/rejected": -202.6668701171875, | |
| "loss": 0.6322, | |
| "rewards/accuracies": 0.65625, | |
| "rewards/chosen": 1.4060888290405273, | |
| "rewards/margins": 0.5313084721565247, | |
| "rewards/rejected": 0.8747804760932922, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.29850746268656714, | |
| "grad_norm": 62.96556196720542, | |
| "learning_rate": 9.932517109205849e-07, | |
| "logits/chosen": -2.3295137882232666, | |
| "logits/rejected": -2.323544502258301, | |
| "logps/chosen": -236.51119995117188, | |
| "logps/rejected": -223.9521026611328, | |
| "loss": 0.6902, | |
| "rewards/accuracies": 0.6187499761581421, | |
| "rewards/chosen": 1.4421606063842773, | |
| "rewards/margins": 0.5772963762283325, | |
| "rewards/rejected": 0.8648644685745239, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.373134328358209, | |
| "grad_norm": 52.70054919387278, | |
| "learning_rate": 9.848591102083375e-07, | |
| "logits/chosen": -2.291351795196533, | |
| "logits/rejected": -2.298973560333252, | |
| "logps/chosen": -248.5207061767578, | |
| "logps/rejected": -212.27572631835938, | |
| "loss": 0.6318, | |
| "rewards/accuracies": 0.6812499761581421, | |
| "rewards/chosen": 1.2592790126800537, | |
| "rewards/margins": 0.7029441595077515, | |
| "rewards/rejected": 0.5563349723815918, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.44776119402985076, | |
| "grad_norm": 39.25239538787194, | |
| "learning_rate": 9.731890013043367e-07, | |
| "logits/chosen": -2.3153462409973145, | |
| "logits/rejected": -2.2776923179626465, | |
| "logps/chosen": -230.8772430419922, | |
| "logps/rejected": -191.97027587890625, | |
| "loss": 0.6011, | |
| "rewards/accuracies": 0.699999988079071, | |
| "rewards/chosen": 1.068457841873169, | |
| "rewards/margins": 0.6923708319664001, | |
| "rewards/rejected": 0.3760870099067688, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.5223880597014925, | |
| "grad_norm": 42.62760194044431, | |
| "learning_rate": 9.583202707897073e-07, | |
| "logits/chosen": -2.3093903064727783, | |
| "logits/rejected": -2.291133403778076, | |
| "logps/chosen": -219.6804656982422, | |
| "logps/rejected": -220.82974243164062, | |
| "loss": 0.6313, | |
| "rewards/accuracies": 0.668749988079071, | |
| "rewards/chosen": 0.7500138282775879, | |
| "rewards/margins": 0.756812572479248, | |
| "rewards/rejected": -0.006798672489821911, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.5970149253731343, | |
| "grad_norm": 40.93212888100826, | |
| "learning_rate": 9.403534270080829e-07, | |
| "logits/chosen": -2.398266315460205, | |
| "logits/rejected": -2.383267641067505, | |
| "logps/chosen": -220.89498901367188, | |
| "logps/rejected": -231.33663940429688, | |
| "loss": 0.6053, | |
| "rewards/accuracies": 0.625, | |
| "rewards/chosen": 0.6151864528656006, | |
| "rewards/margins": 0.6149100065231323, | |
| "rewards/rejected": 0.00027652084827423096, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.6716417910447762, | |
| "grad_norm": 44.441524435463975, | |
| "learning_rate": 9.19409920658098e-07, | |
| "logits/chosen": -2.38905668258667, | |
| "logits/rejected": -2.418109893798828, | |
| "logps/chosen": -243.25039672851562, | |
| "logps/rejected": -227.03500366210938, | |
| "loss": 0.6158, | |
| "rewards/accuracies": 0.7250000238418579, | |
| "rewards/chosen": 0.7249364256858826, | |
| "rewards/margins": 0.903680682182312, | |
| "rewards/rejected": -0.17874428629875183, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.746268656716418, | |
| "grad_norm": 36.109012224798654, | |
| "learning_rate": 8.956313238215823e-07, | |
| "logits/chosen": -2.4329724311828613, | |
| "logits/rejected": -2.393233060836792, | |
| "logps/chosen": -235.9673614501953, | |
| "logps/rejected": -207.4291229248047, | |
| "loss": 0.5788, | |
| "rewards/accuracies": 0.7250000238418579, | |
| "rewards/chosen": 0.7963231801986694, | |
| "rewards/margins": 1.1296331882476807, | |
| "rewards/rejected": -0.3333100378513336, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.746268656716418, | |
| "eval_logits/chosen": -2.3947904109954834, | |
| "eval_logits/rejected": -2.381561756134033, | |
| "eval_logps/chosen": -248.11183166503906, | |
| "eval_logps/rejected": -209.1226348876953, | |
| "eval_loss": 0.5886067152023315, | |
| "eval_rewards/accuracies": 0.7666666507720947, | |
| "eval_rewards/chosen": 1.3111679553985596, | |
| "eval_rewards/margins": 1.323790431022644, | |
| "eval_rewards/rejected": -0.01262261439114809, | |
| "eval_runtime": 126.2751, | |
| "eval_samples_per_second": 15.047, | |
| "eval_steps_per_second": 0.238, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.8208955223880597, | |
| "grad_norm": 37.42293105255099, | |
| "learning_rate": 8.691783729769873e-07, | |
| "logits/chosen": -2.3496580123901367, | |
| "logits/rejected": -2.370349407196045, | |
| "logps/chosen": -237.4831085205078, | |
| "logps/rejected": -204.3516082763672, | |
| "loss": 0.5805, | |
| "rewards/accuracies": 0.699999988079071, | |
| "rewards/chosen": 0.9665477871894836, | |
| "rewards/margins": 1.0714900493621826, | |
| "rewards/rejected": -0.10494247823953629, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.8955223880597015, | |
| "grad_norm": 36.18527140538969, | |
| "learning_rate": 8.402298824670029e-07, | |
| "logits/chosen": -2.3262667655944824, | |
| "logits/rejected": -2.3069987297058105, | |
| "logps/chosen": -252.2436065673828, | |
| "logps/rejected": -218.7754364013672, | |
| "loss": 0.5423, | |
| "rewards/accuracies": 0.7437499761581421, | |
| "rewards/chosen": 1.2422854900360107, | |
| "rewards/margins": 1.209867238998413, | |
| "rewards/rejected": 0.032418303191661835, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.9701492537313433, | |
| "grad_norm": 33.097833441355846, | |
| "learning_rate": 8.089815357650089e-07, | |
| "logits/chosen": -2.33039927482605, | |
| "logits/rejected": -2.325122833251953, | |
| "logps/chosen": -254.16928100585938, | |
| "logps/rejected": -207.00668334960938, | |
| "loss": 0.5462, | |
| "rewards/accuracies": 0.737500011920929, | |
| "rewards/chosen": 1.058384656906128, | |
| "rewards/margins": 1.3959627151489258, | |
| "rewards/rejected": -0.33757805824279785, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 1.044776119402985, | |
| "grad_norm": 20.023976264712793, | |
| "learning_rate": 7.756445627110522e-07, | |
| "logits/chosen": -2.3418185710906982, | |
| "logits/rejected": -2.3507180213928223, | |
| "logps/chosen": -239.88333129882812, | |
| "logps/rejected": -218.7294464111328, | |
| "loss": 0.3251, | |
| "rewards/accuracies": 0.856249988079071, | |
| "rewards/chosen": 1.4150011539459229, | |
| "rewards/margins": 1.9995105266571045, | |
| "rewards/rejected": -0.584509551525116, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 1.1194029850746268, | |
| "grad_norm": 23.675520442711395, | |
| "learning_rate": 7.404443116588547e-07, | |
| "logits/chosen": -2.398703098297119, | |
| "logits/rejected": -2.3791286945343018, | |
| "logps/chosen": -230.90005493164062, | |
| "logps/rejected": -201.66864013671875, | |
| "loss": 0.221, | |
| "rewards/accuracies": 0.9375, | |
| "rewards/chosen": 1.7823137044906616, | |
| "rewards/margins": 2.5341625213623047, | |
| "rewards/rejected": -0.7518486380577087, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 1.1940298507462686, | |
| "grad_norm": 15.225016299890036, | |
| "learning_rate": 7.036187261857288e-07, | |
| "logits/chosen": -2.3974556922912598, | |
| "logits/rejected": -2.388159990310669, | |
| "logps/chosen": -242.7141571044922, | |
| "logps/rejected": -225.48001098632812, | |
| "loss": 0.2114, | |
| "rewards/accuracies": 0.9125000238418579, | |
| "rewards/chosen": 2.029745578765869, | |
| "rewards/margins": 3.112726926803589, | |
| "rewards/rejected": -1.0829811096191406, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 1.2686567164179103, | |
| "grad_norm": 20.47443225957059, | |
| "learning_rate": 6.654167366624008e-07, | |
| "logits/chosen": -2.3772902488708496, | |
| "logits/rejected": -2.351728916168213, | |
| "logps/chosen": -233.95236206054688, | |
| "logps/rejected": -230.03994750976562, | |
| "loss": 0.2168, | |
| "rewards/accuracies": 0.925000011920929, | |
| "rewards/chosen": 1.4827768802642822, | |
| "rewards/margins": 2.783128499984741, | |
| "rewards/rejected": -1.3003517389297485, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 1.3432835820895521, | |
| "grad_norm": 23.797530563309465, | |
| "learning_rate": 6.260965775552713e-07, | |
| "logits/chosen": -2.3302738666534424, | |
| "logits/rejected": -2.3255810737609863, | |
| "logps/chosen": -242.00732421875, | |
| "logps/rejected": -240.259521484375, | |
| "loss": 0.2087, | |
| "rewards/accuracies": 0.9437500238418579, | |
| "rewards/chosen": 1.8802661895751953, | |
| "rewards/margins": 3.69026517868042, | |
| "rewards/rejected": -1.8099991083145142, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 1.417910447761194, | |
| "grad_norm": 22.2827308670365, | |
| "learning_rate": 5.859240418356614e-07, | |
| "logits/chosen": -2.346041679382324, | |
| "logits/rejected": -2.3258674144744873, | |
| "logps/chosen": -241.3457794189453, | |
| "logps/rejected": -232.9997100830078, | |
| "loss": 0.2454, | |
| "rewards/accuracies": 0.9312499761581421, | |
| "rewards/chosen": 2.422487497329712, | |
| "rewards/margins": 3.9014511108398438, | |
| "rewards/rejected": -1.4789636135101318, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 1.4925373134328357, | |
| "grad_norm": 23.32738252611807, | |
| "learning_rate": 5.451706842957421e-07, | |
| "logits/chosen": -2.3272500038146973, | |
| "logits/rejected": -2.290189266204834, | |
| "logps/chosen": -237.8894805908203, | |
| "logps/rejected": -224.0364227294922, | |
| "loss": 0.2623, | |
| "rewards/accuracies": 0.956250011920929, | |
| "rewards/chosen": 2.44073748588562, | |
| "rewards/margins": 3.8086400032043457, | |
| "rewards/rejected": -1.3679022789001465, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.4925373134328357, | |
| "eval_logits/chosen": -2.272735118865967, | |
| "eval_logits/rejected": -2.24358868598938, | |
| "eval_logps/chosen": -245.39662170410156, | |
| "eval_logps/rejected": -213.9952850341797, | |
| "eval_loss": 0.6056690216064453, | |
| "eval_rewards/accuracies": 0.8041666746139526, | |
| "eval_rewards/chosen": 1.5826876163482666, | |
| "eval_rewards/margins": 2.0825743675231934, | |
| "eval_rewards/rejected": -0.49988678097724915, | |
| "eval_runtime": 126.0708, | |
| "eval_samples_per_second": 15.071, | |
| "eval_steps_per_second": 0.238, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.5671641791044775, | |
| "grad_norm": 18.955563276330228, | |
| "learning_rate": 5.041119859162068e-07, | |
| "logits/chosen": -2.266010046005249, | |
| "logits/rejected": -2.224538564682007, | |
| "logps/chosen": -229.598876953125, | |
| "logps/rejected": -229.48312377929688, | |
| "loss": 0.2328, | |
| "rewards/accuracies": 0.9312499761581421, | |
| "rewards/chosen": 2.0126357078552246, | |
| "rewards/margins": 3.625946044921875, | |
| "rewards/rejected": -1.6133100986480713, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 1.6417910447761193, | |
| "grad_norm": 22.19988117931717, | |
| "learning_rate": 4.630254916940423e-07, | |
| "logits/chosen": -2.241581678390503, | |
| "logits/rejected": -2.228926658630371, | |
| "logps/chosen": -234.0880889892578, | |
| "logps/rejected": -211.94546508789062, | |
| "loss": 0.2614, | |
| "rewards/accuracies": 0.887499988079071, | |
| "rewards/chosen": 1.8189502954483032, | |
| "rewards/margins": 3.2742087841033936, | |
| "rewards/rejected": -1.4552589654922485, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 1.716417910447761, | |
| "grad_norm": 25.211183388734273, | |
| "learning_rate": 4.2218893451814e-07, | |
| "logits/chosen": -2.240575075149536, | |
| "logits/rejected": -2.2353932857513428, | |
| "logps/chosen": -234.106201171875, | |
| "logps/rejected": -225.87057495117188, | |
| "loss": 0.2607, | |
| "rewards/accuracies": 0.9125000238418579, | |
| "rewards/chosen": 1.9395558834075928, | |
| "rewards/margins": 3.2138640880584717, | |
| "rewards/rejected": -1.2743080854415894, | |
| "step": 115 | |
| }, | |
| { | |
| "epoch": 1.7910447761194028, | |
| "grad_norm": 20.408403893097674, | |
| "learning_rate": 3.8187835777481375e-07, | |
| "logits/chosen": -2.2678146362304688, | |
| "logits/rejected": -2.248746395111084, | |
| "logps/chosen": -234.76974487304688, | |
| "logps/rejected": -229.4392852783203, | |
| "loss": 0.2434, | |
| "rewards/accuracies": 0.9312499761581421, | |
| "rewards/chosen": 2.1745102405548096, | |
| "rewards/margins": 3.79044771194458, | |
| "rewards/rejected": -1.6159374713897705, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 1.8656716417910446, | |
| "grad_norm": 23.108934441307838, | |
| "learning_rate": 3.423662493738687e-07, | |
| "logits/chosen": -2.2544753551483154, | |
| "logits/rejected": -2.226003408432007, | |
| "logps/chosen": -215.0713348388672, | |
| "logps/rejected": -198.3508758544922, | |
| "loss": 0.2838, | |
| "rewards/accuracies": 0.8687499761581421, | |
| "rewards/chosen": 1.776116967201233, | |
| "rewards/margins": 3.0843865871429443, | |
| "rewards/rejected": -1.3082692623138428, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 1.9402985074626866, | |
| "grad_norm": 24.748752156744665, | |
| "learning_rate": 3.039196998086687e-07, | |
| "logits/chosen": -2.273759365081787, | |
| "logits/rejected": -2.2541890144348145, | |
| "logps/chosen": -239.5052947998047, | |
| "logps/rejected": -248.71157836914062, | |
| "loss": 0.2544, | |
| "rewards/accuracies": 0.893750011920929, | |
| "rewards/chosen": 2.4912400245666504, | |
| "rewards/margins": 3.995276927947998, | |
| "rewards/rejected": -1.5040366649627686, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 2.014925373134328, | |
| "grad_norm": 17.84439016806703, | |
| "learning_rate": 2.667985967011878e-07, | |
| "logits/chosen": -2.287829875946045, | |
| "logits/rejected": -2.2854392528533936, | |
| "logps/chosen": -243.9482879638672, | |
| "logps/rejected": -236.39785766601562, | |
| "loss": 0.2205, | |
| "rewards/accuracies": 0.918749988079071, | |
| "rewards/chosen": 2.2010834217071533, | |
| "rewards/margins": 3.6561388969421387, | |
| "rewards/rejected": -1.455055832862854, | |
| "step": 135 | |
| }, | |
| { | |
| "epoch": 2.08955223880597, | |
| "grad_norm": 17.681759766337574, | |
| "learning_rate": 2.3125386803640183e-07, | |
| "logits/chosen": -2.303140163421631, | |
| "logits/rejected": -2.2794697284698486, | |
| "logps/chosen": -229.8805389404297, | |
| "logps/rejected": -261.2996520996094, | |
| "loss": 0.1265, | |
| "rewards/accuracies": 0.949999988079071, | |
| "rewards/chosen": 2.40771222114563, | |
| "rewards/margins": 4.214150428771973, | |
| "rewards/rejected": -1.8064378499984741, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 2.1641791044776117, | |
| "grad_norm": 14.355965133474355, | |
| "learning_rate": 1.9752578596124952e-07, | |
| "logits/chosen": -2.2925033569335938, | |
| "logits/rejected": -2.2765016555786133, | |
| "logps/chosen": -239.452392578125, | |
| "logps/rejected": -258.3964538574219, | |
| "loss": 0.1418, | |
| "rewards/accuracies": 0.9624999761581421, | |
| "rewards/chosen": 2.2769477367401123, | |
| "rewards/margins": 4.572505474090576, | |
| "rewards/rejected": -2.295557975769043, | |
| "step": 145 | |
| }, | |
| { | |
| "epoch": 2.2388059701492535, | |
| "grad_norm": 13.075018639785867, | |
| "learning_rate": 1.6584234261399532e-07, | |
| "logits/chosen": -2.2472779750823975, | |
| "logits/rejected": -2.2457058429718018, | |
| "logps/chosen": -215.5850372314453, | |
| "logps/rejected": -245.4586181640625, | |
| "loss": 0.1176, | |
| "rewards/accuracies": 0.9312499761581421, | |
| "rewards/chosen": 2.197263479232788, | |
| "rewards/margins": 4.124143123626709, | |
| "rewards/rejected": -1.9268791675567627, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 2.2388059701492535, | |
| "eval_logits/chosen": -2.2484564781188965, | |
| "eval_logits/rejected": -2.223518133163452, | |
| "eval_logps/chosen": -246.53546142578125, | |
| "eval_logps/rejected": -216.13336181640625, | |
| "eval_loss": 0.5922633409500122, | |
| "eval_rewards/accuracies": 0.7875000238418579, | |
| "eval_rewards/chosen": 1.4688013792037964, | |
| "eval_rewards/margins": 2.1824986934661865, | |
| "eval_rewards/rejected": -0.7136973738670349, | |
| "eval_runtime": 125.9912, | |
| "eval_samples_per_second": 15.08, | |
| "eval_steps_per_second": 0.238, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 2.3134328358208958, | |
| "grad_norm": 13.851661957735093, | |
| "learning_rate": 1.3641770896292082e-07, | |
| "logits/chosen": -2.2474873065948486, | |
| "logits/rejected": -2.247633457183838, | |
| "logps/chosen": -244.7652587890625, | |
| "logps/rejected": -235.2009735107422, | |
| "loss": 0.1112, | |
| "rewards/accuracies": 0.9750000238418579, | |
| "rewards/chosen": 2.8528218269348145, | |
| "rewards/margins": 4.657607555389404, | |
| "rewards/rejected": -1.804785966873169, | |
| "step": 155 | |
| }, | |
| { | |
| "epoch": 2.388059701492537, | |
| "grad_norm": 21.198096536107492, | |
| "learning_rate": 1.0945078707215221e-07, | |
| "logits/chosen": -2.260723829269409, | |
| "logits/rejected": -2.250204563140869, | |
| "logps/chosen": -234.06942749023438, | |
| "logps/rejected": -261.3254699707031, | |
| "loss": 0.1333, | |
| "rewards/accuracies": 0.956250011920929, | |
| "rewards/chosen": 2.2776079177856445, | |
| "rewards/margins": 4.555299282073975, | |
| "rewards/rejected": -2.27769136428833, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 2.4626865671641793, | |
| "grad_norm": 17.709308743546742, | |
| "learning_rate": 8.512386558088919e-08, | |
| "logits/chosen": -2.2542097568511963, | |
| "logits/rejected": -2.2354049682617188, | |
| "logps/chosen": -227.0293426513672, | |
| "logps/rejected": -220.221435546875, | |
| "loss": 0.1199, | |
| "rewards/accuracies": 0.9624999761581421, | |
| "rewards/chosen": 2.320430278778076, | |
| "rewards/margins": 4.206162452697754, | |
| "rewards/rejected": -1.8857324123382568, | |
| "step": 165 | |
| }, | |
| { | |
| "epoch": 2.5373134328358207, | |
| "grad_norm": 10.709332675693139, | |
| "learning_rate": 6.360138748461013e-08, | |
| "logits/chosen": -2.2639975547790527, | |
| "logits/rejected": -2.234525203704834, | |
| "logps/chosen": -229.2438507080078, | |
| "logps/rejected": -223.31326293945312, | |
| "loss": 0.1403, | |
| "rewards/accuracies": 0.9750000238418579, | |
| "rewards/chosen": 2.291680335998535, | |
| "rewards/margins": 4.302089691162109, | |
| "rewards/rejected": -2.010409116744995, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 2.611940298507463, | |
| "grad_norm": 15.275450815772352, | |
| "learning_rate": 4.5028838547699346e-08, | |
| "logits/chosen": -2.238157272338867, | |
| "logits/rejected": -2.2448458671569824, | |
| "logps/chosen": -223.23828125, | |
| "logps/rejected": -251.35513305664062, | |
| "loss": 0.1258, | |
| "rewards/accuracies": 0.949999988079071, | |
| "rewards/chosen": 2.0068905353546143, | |
| "rewards/margins": 4.182096481323242, | |
| "rewards/rejected": -2.175205945968628, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 2.6865671641791042, | |
| "grad_norm": 10.670496827369625, | |
| "learning_rate": 2.9531763861505964e-08, | |
| "logits/chosen": -2.246856689453125, | |
| "logits/rejected": -2.2432849407196045, | |
| "logps/chosen": -230.0186309814453, | |
| "logps/rejected": -235.7653045654297, | |
| "loss": 0.1521, | |
| "rewards/accuracies": 0.956250011920929, | |
| "rewards/chosen": 2.0715701580047607, | |
| "rewards/margins": 4.315110683441162, | |
| "rewards/rejected": -2.2435405254364014, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 2.7611940298507465, | |
| "grad_norm": 13.296818618991479, | |
| "learning_rate": 1.7214919195619125e-08, | |
| "logits/chosen": -2.2300615310668945, | |
| "logits/rejected": -2.224147319793701, | |
| "logps/chosen": -233.1608428955078, | |
| "logps/rejected": -218.90585327148438, | |
| "loss": 0.1109, | |
| "rewards/accuracies": 0.956250011920929, | |
| "rewards/chosen": 2.1774356365203857, | |
| "rewards/margins": 4.205111503601074, | |
| "rewards/rejected": -2.0276761054992676, | |
| "step": 185 | |
| }, | |
| { | |
| "epoch": 2.835820895522388, | |
| "grad_norm": 17.367538516485407, | |
| "learning_rate": 8.161562878982398e-09, | |
| "logits/chosen": -2.2490463256835938, | |
| "logits/rejected": -2.2461249828338623, | |
| "logps/chosen": -229.1264190673828, | |
| "logps/rejected": -234.72836303710938, | |
| "loss": 0.1378, | |
| "rewards/accuracies": 0.956250011920929, | |
| "rewards/chosen": 2.0733642578125, | |
| "rewards/margins": 4.36142635345459, | |
| "rewards/rejected": -2.288062334060669, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 2.91044776119403, | |
| "grad_norm": 19.281670591260532, | |
| "learning_rate": 2.432892997526026e-09, | |
| "logits/chosen": -2.247451066970825, | |
| "logits/rejected": -2.228789806365967, | |
| "logps/chosen": -236.5612030029297, | |
| "logps/rejected": -241.4217987060547, | |
| "loss": 0.1481, | |
| "rewards/accuracies": 0.949999988079071, | |
| "rewards/chosen": 2.031038761138916, | |
| "rewards/margins": 4.2105302810668945, | |
| "rewards/rejected": -2.1794912815093994, | |
| "step": 195 | |
| }, | |
| { | |
| "epoch": 2.9850746268656714, | |
| "grad_norm": 16.314964119933162, | |
| "learning_rate": 6.763371270035457e-11, | |
| "logits/chosen": -2.2567026615142822, | |
| "logits/rejected": -2.2431235313415527, | |
| "logps/chosen": -244.52407836914062, | |
| "logps/rejected": -221.75595092773438, | |
| "loss": 0.1387, | |
| "rewards/accuracies": 0.949999988079071, | |
| "rewards/chosen": 2.3067245483398438, | |
| "rewards/margins": 4.139992713928223, | |
| "rewards/rejected": -1.8332679271697998, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 2.9850746268656714, | |
| "eval_logits/chosen": -2.243741035461426, | |
| "eval_logits/rejected": -2.2175543308258057, | |
| "eval_logps/chosen": -248.4080047607422, | |
| "eval_logps/rejected": -218.735107421875, | |
| "eval_loss": 0.6027006506919861, | |
| "eval_rewards/accuracies": 0.7875000238418579, | |
| "eval_rewards/chosen": 1.2815468311309814, | |
| "eval_rewards/margins": 2.2554140090942383, | |
| "eval_rewards/rejected": -0.9738672375679016, | |
| "eval_runtime": 125.8747, | |
| "eval_samples_per_second": 15.094, | |
| "eval_steps_per_second": 0.238, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "step": 201, | |
| "total_flos": 2369906314051584.0, | |
| "train_loss": 0.32727647339230154, | |
| "train_runtime": 7356.5112, | |
| "train_samples_per_second": 6.971, | |
| "train_steps_per_second": 0.027 | |
| } | |
| ], | |
| "logging_steps": 5, | |
| "max_steps": 201, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 100, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 2369906314051584.0, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |