| { | |
| "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": 60.46895766728989, | |
| "learning_rate": 5e-07, | |
| "logits/chosen": -2.737339735031128, | |
| "logits/rejected": -2.717646837234497, | |
| "logps/chosen": -269.46966552734375, | |
| "logps/rejected": -218.519287109375, | |
| "loss": 0.6897, | |
| "rewards/accuracies": 0.33125001192092896, | |
| "rewards/chosen": 0.016166506335139275, | |
| "rewards/margins": 0.005314634181559086, | |
| "rewards/rejected": 0.010851874016225338, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.14925373134328357, | |
| "grad_norm": 47.55077346828281, | |
| "learning_rate": 1e-06, | |
| "logits/chosen": -2.707444190979004, | |
| "logits/rejected": -2.6988959312438965, | |
| "logps/chosen": -259.7918395996094, | |
| "logps/rejected": -198.8335723876953, | |
| "loss": 0.6419, | |
| "rewards/accuracies": 0.668749988079071, | |
| "rewards/chosen": 0.5464529991149902, | |
| "rewards/margins": 0.22305770218372345, | |
| "rewards/rejected": 0.323395311832428, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.22388059701492538, | |
| "grad_norm": 49.794624138997065, | |
| "learning_rate": 9.983100718730718e-07, | |
| "logits/chosen": -2.5312893390655518, | |
| "logits/rejected": -2.528350353240967, | |
| "logps/chosen": -250.9792022705078, | |
| "logps/rejected": -215.8843231201172, | |
| "loss": 0.6325, | |
| "rewards/accuracies": 0.668749988079071, | |
| "rewards/chosen": 1.2436325550079346, | |
| "rewards/margins": 0.6625381708145142, | |
| "rewards/rejected": 0.5810943841934204, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.29850746268656714, | |
| "grad_norm": 45.89015113492354, | |
| "learning_rate": 9.932517109205849e-07, | |
| "logits/chosen": -2.345860719680786, | |
| "logits/rejected": -2.330219030380249, | |
| "logps/chosen": -261.6305236816406, | |
| "logps/rejected": -205.9855499267578, | |
| "loss": 0.5741, | |
| "rewards/accuracies": 0.731249988079071, | |
| "rewards/chosen": 1.5929079055786133, | |
| "rewards/margins": 0.922720730304718, | |
| "rewards/rejected": 0.6701873540878296, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.373134328358209, | |
| "grad_norm": 63.53673896186562, | |
| "learning_rate": 9.848591102083375e-07, | |
| "logits/chosen": -2.2251970767974854, | |
| "logits/rejected": -2.210585832595825, | |
| "logps/chosen": -256.94158935546875, | |
| "logps/rejected": -198.90554809570312, | |
| "loss": 0.5846, | |
| "rewards/accuracies": 0.699999988079071, | |
| "rewards/chosen": 1.4611073732376099, | |
| "rewards/margins": 0.925146758556366, | |
| "rewards/rejected": 0.5359605550765991, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.44776119402985076, | |
| "grad_norm": 40.6998829965568, | |
| "learning_rate": 9.731890013043367e-07, | |
| "logits/chosen": -2.203108310699463, | |
| "logits/rejected": -2.188232898712158, | |
| "logps/chosen": -220.20474243164062, | |
| "logps/rejected": -174.1934356689453, | |
| "loss": 0.5755, | |
| "rewards/accuracies": 0.731249988079071, | |
| "rewards/chosen": 1.1598186492919922, | |
| "rewards/margins": 1.0439460277557373, | |
| "rewards/rejected": 0.11587252467870712, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.5223880597014925, | |
| "grad_norm": 39.86530018922445, | |
| "learning_rate": 9.583202707897073e-07, | |
| "logits/chosen": -2.159541606903076, | |
| "logits/rejected": -2.146280288696289, | |
| "logps/chosen": -271.1134033203125, | |
| "logps/rejected": -222.50732421875, | |
| "loss": 0.5581, | |
| "rewards/accuracies": 0.78125, | |
| "rewards/chosen": 1.6402053833007812, | |
| "rewards/margins": 1.4621944427490234, | |
| "rewards/rejected": 0.1780107319355011, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.5970149253731343, | |
| "grad_norm": 47.764041993597665, | |
| "learning_rate": 9.403534270080829e-07, | |
| "logits/chosen": -2.097519874572754, | |
| "logits/rejected": -2.0528969764709473, | |
| "logps/chosen": -246.8362274169922, | |
| "logps/rejected": -208.1300048828125, | |
| "loss": 0.5329, | |
| "rewards/accuracies": 0.668749988079071, | |
| "rewards/chosen": 1.0587269067764282, | |
| "rewards/margins": 1.0248701572418213, | |
| "rewards/rejected": 0.033856604248285294, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.6716417910447762, | |
| "grad_norm": 33.33992129726437, | |
| "learning_rate": 9.19409920658098e-07, | |
| "logits/chosen": -2.1117985248565674, | |
| "logits/rejected": -2.0575757026672363, | |
| "logps/chosen": -238.47335815429688, | |
| "logps/rejected": -232.6268310546875, | |
| "loss": 0.5597, | |
| "rewards/accuracies": 0.706250011920929, | |
| "rewards/chosen": 1.020668387413025, | |
| "rewards/margins": 1.3137915134429932, | |
| "rewards/rejected": -0.2931232154369354, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.746268656716418, | |
| "grad_norm": 36.75599018498795, | |
| "learning_rate": 8.956313238215823e-07, | |
| "logits/chosen": -2.192502021789551, | |
| "logits/rejected": -2.1830239295959473, | |
| "logps/chosen": -246.92733764648438, | |
| "logps/rejected": -219.84817504882812, | |
| "loss": 0.5242, | |
| "rewards/accuracies": 0.7437499761581421, | |
| "rewards/chosen": 0.8840469121932983, | |
| "rewards/margins": 1.3445603847503662, | |
| "rewards/rejected": -0.4605133533477783, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.746268656716418, | |
| "eval_logits/chosen": -2.2182116508483887, | |
| "eval_logits/rejected": -2.195751190185547, | |
| "eval_logps/chosen": -256.5732727050781, | |
| "eval_logps/rejected": -219.44544982910156, | |
| "eval_loss": 0.5641670823097229, | |
| "eval_rewards/accuracies": 0.7583333253860474, | |
| "eval_rewards/chosen": 0.8116810321807861, | |
| "eval_rewards/margins": 1.5396502017974854, | |
| "eval_rewards/rejected": -0.7279691696166992, | |
| "eval_runtime": 125.8473, | |
| "eval_samples_per_second": 15.098, | |
| "eval_steps_per_second": 0.238, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.8208955223880597, | |
| "grad_norm": 33.66480221016764, | |
| "learning_rate": 8.691783729769873e-07, | |
| "logits/chosen": -2.252897024154663, | |
| "logits/rejected": -2.210509777069092, | |
| "logps/chosen": -273.36102294921875, | |
| "logps/rejected": -232.4573974609375, | |
| "loss": 0.5359, | |
| "rewards/accuracies": 0.793749988079071, | |
| "rewards/chosen": 1.0710394382476807, | |
| "rewards/margins": 1.8447446823120117, | |
| "rewards/rejected": -0.7737051248550415, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.8955223880597015, | |
| "grad_norm": 42.22753576523594, | |
| "learning_rate": 8.402298824670029e-07, | |
| "logits/chosen": -2.2703890800476074, | |
| "logits/rejected": -2.252363443374634, | |
| "logps/chosen": -274.8329162597656, | |
| "logps/rejected": -254.45663452148438, | |
| "loss": 0.5811, | |
| "rewards/accuracies": 0.699999988079071, | |
| "rewards/chosen": 0.9601489901542664, | |
| "rewards/margins": 1.3747169971466064, | |
| "rewards/rejected": -0.4145679473876953, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.9701492537313433, | |
| "grad_norm": 38.82616914526827, | |
| "learning_rate": 8.089815357650089e-07, | |
| "logits/chosen": -2.237731456756592, | |
| "logits/rejected": -2.245671510696411, | |
| "logps/chosen": -264.57989501953125, | |
| "logps/rejected": -241.2982940673828, | |
| "loss": 0.5166, | |
| "rewards/accuracies": 0.7562500238418579, | |
| "rewards/chosen": 1.0662319660186768, | |
| "rewards/margins": 1.8653961420059204, | |
| "rewards/rejected": -0.799164354801178, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 1.044776119402985, | |
| "grad_norm": 21.744667598719886, | |
| "learning_rate": 7.756445627110522e-07, | |
| "logits/chosen": -2.2960398197174072, | |
| "logits/rejected": -2.2724010944366455, | |
| "logps/chosen": -264.32720947265625, | |
| "logps/rejected": -239.6345977783203, | |
| "loss": 0.346, | |
| "rewards/accuracies": 0.856249988079071, | |
| "rewards/chosen": 2.158931255340576, | |
| "rewards/margins": 2.908942222595215, | |
| "rewards/rejected": -0.7500110864639282, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 1.1194029850746268, | |
| "grad_norm": 19.678947962162013, | |
| "learning_rate": 7.404443116588547e-07, | |
| "logits/chosen": -2.396239995956421, | |
| "logits/rejected": -2.362546443939209, | |
| "logps/chosen": -246.3358154296875, | |
| "logps/rejected": -195.57357788085938, | |
| "loss": 0.2283, | |
| "rewards/accuracies": 0.918749988079071, | |
| "rewards/chosen": 2.2872486114501953, | |
| "rewards/margins": 2.971306562423706, | |
| "rewards/rejected": -0.6840583086013794, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 1.1940298507462686, | |
| "grad_norm": 20.2993371431914, | |
| "learning_rate": 7.036187261857288e-07, | |
| "logits/chosen": -2.4099860191345215, | |
| "logits/rejected": -2.384648084640503, | |
| "logps/chosen": -260.5028381347656, | |
| "logps/rejected": -223.2133331298828, | |
| "loss": 0.2312, | |
| "rewards/accuracies": 0.9125000238418579, | |
| "rewards/chosen": 2.834780693054199, | |
| "rewards/margins": 3.5776984691619873, | |
| "rewards/rejected": -0.7429176568984985, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 1.2686567164179103, | |
| "grad_norm": 25.4573275034557, | |
| "learning_rate": 6.654167366624008e-07, | |
| "logits/chosen": -2.36556339263916, | |
| "logits/rejected": -2.349255084991455, | |
| "logps/chosen": -231.6350860595703, | |
| "logps/rejected": -221.1610870361328, | |
| "loss": 0.2307, | |
| "rewards/accuracies": 0.9437500238418579, | |
| "rewards/chosen": 1.9625335931777954, | |
| "rewards/margins": 3.2484028339385986, | |
| "rewards/rejected": -1.2858693599700928, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 1.3432835820895521, | |
| "grad_norm": 29.79365948091661, | |
| "learning_rate": 6.260965775552713e-07, | |
| "logits/chosen": -2.374537944793701, | |
| "logits/rejected": -2.348567485809326, | |
| "logps/chosen": -247.57614135742188, | |
| "logps/rejected": -238.37899780273438, | |
| "loss": 0.2523, | |
| "rewards/accuracies": 0.925000011920929, | |
| "rewards/chosen": 1.934003233909607, | |
| "rewards/margins": 3.8235995769500732, | |
| "rewards/rejected": -1.8895965814590454, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 1.417910447761194, | |
| "grad_norm": 22.109779497676236, | |
| "learning_rate": 5.859240418356614e-07, | |
| "logits/chosen": -2.3896210193634033, | |
| "logits/rejected": -2.3736777305603027, | |
| "logps/chosen": -257.51898193359375, | |
| "logps/rejected": -228.7509765625, | |
| "loss": 0.2453, | |
| "rewards/accuracies": 0.9125000238418579, | |
| "rewards/chosen": 2.010754108428955, | |
| "rewards/margins": 3.4373581409454346, | |
| "rewards/rejected": -1.4266036748886108, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 1.4925373134328357, | |
| "grad_norm": 18.59821752381129, | |
| "learning_rate": 5.451706842957421e-07, | |
| "logits/chosen": -2.3828063011169434, | |
| "logits/rejected": -2.3802871704101562, | |
| "logps/chosen": -238.89547729492188, | |
| "logps/rejected": -224.2349395751953, | |
| "loss": 0.2523, | |
| "rewards/accuracies": 0.918749988079071, | |
| "rewards/chosen": 2.049046039581299, | |
| "rewards/margins": 3.4148106575012207, | |
| "rewards/rejected": -1.3657643795013428, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.4925373134328357, | |
| "eval_logits/chosen": -2.3702290058135986, | |
| "eval_logits/rejected": -2.352924108505249, | |
| "eval_logps/chosen": -250.75942993164062, | |
| "eval_logps/rejected": -224.2054901123047, | |
| "eval_loss": 0.5878031253814697, | |
| "eval_rewards/accuracies": 0.8041666746139526, | |
| "eval_rewards/chosen": 1.393066644668579, | |
| "eval_rewards/margins": 2.597040891647339, | |
| "eval_rewards/rejected": -1.2039742469787598, | |
| "eval_runtime": 125.6186, | |
| "eval_samples_per_second": 15.125, | |
| "eval_steps_per_second": 0.239, | |
| "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 | |
| } | |