{ "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 }