| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.4925373134328357, | |
| "eval_steps": 50, | |
| "global_step": 100, | |
| "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 | |
| } | |
| ], | |
| "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": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 1178822762299392.0, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |