|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 1.9954853273137698, |
|
"eval_steps": 500, |
|
"global_step": 442, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.05, |
|
"grad_norm": 3.34375, |
|
"learning_rate": 4.9936877763590664e-06, |
|
"log_odds_chosen": 0.2890412509441376, |
|
"log_odds_ratio": -0.6332851648330688, |
|
"logits/chosen": -3.1468911170959473, |
|
"logits/rejected": -3.138887643814087, |
|
"logps/chosen": -0.8719667196273804, |
|
"logps/rejected": -1.0382100343704224, |
|
"loss": 0.6482, |
|
"nll_loss": 0.5332359075546265, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.17439337074756622, |
|
"rewards/margins": 0.033248625695705414, |
|
"rewards/rejected": -0.20764198899269104, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"grad_norm": 3.046875, |
|
"learning_rate": 4.9747829807701e-06, |
|
"log_odds_chosen": 0.3164082169532776, |
|
"log_odds_ratio": -0.6052478551864624, |
|
"logits/chosen": -3.0846054553985596, |
|
"logits/rejected": -3.0846097469329834, |
|
"logps/chosen": -0.8607593774795532, |
|
"logps/rejected": -1.0491193532943726, |
|
"loss": 0.6131, |
|
"nll_loss": 0.4854271411895752, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.17215189337730408, |
|
"rewards/margins": 0.03767203167080879, |
|
"rewards/rejected": -0.20982392132282257, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"grad_norm": 3.671875, |
|
"learning_rate": 4.943381078271214e-06, |
|
"log_odds_chosen": 0.3783247768878937, |
|
"log_odds_ratio": -0.6129769086837769, |
|
"logits/chosen": -3.084336996078491, |
|
"logits/rejected": -3.0786914825439453, |
|
"logps/chosen": -0.8424398303031921, |
|
"logps/rejected": -1.0533756017684937, |
|
"loss": 0.602, |
|
"nll_loss": 0.4758261740207672, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.16848795115947723, |
|
"rewards/margins": 0.0421871617436409, |
|
"rewards/rejected": -0.21067512035369873, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"grad_norm": 3.0, |
|
"learning_rate": 4.89964064152747e-06, |
|
"log_odds_chosen": 0.3220486640930176, |
|
"log_odds_ratio": -0.6127817630767822, |
|
"logits/chosen": -3.073997974395752, |
|
"logits/rejected": -3.069819688796997, |
|
"logps/chosen": -0.8541064262390137, |
|
"logps/rejected": -1.0315220355987549, |
|
"loss": 0.5993, |
|
"nll_loss": 0.47563648223876953, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.17082129418849945, |
|
"rewards/margins": 0.03548307344317436, |
|
"rewards/rejected": -0.2063043862581253, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"grad_norm": 2.90625, |
|
"learning_rate": 4.84378255007397e-06, |
|
"log_odds_chosen": 0.3029581606388092, |
|
"log_odds_ratio": -0.6211820840835571, |
|
"logits/chosen": -3.0848793983459473, |
|
"logits/rejected": -3.0841445922851562, |
|
"logps/chosen": -0.8334375619888306, |
|
"logps/rejected": -1.0042452812194824, |
|
"loss": 0.5913, |
|
"nll_loss": 0.4705049991607666, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.1666875034570694, |
|
"rewards/margins": 0.03416154906153679, |
|
"rewards/rejected": -0.20084905624389648, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"grad_norm": 3.015625, |
|
"learning_rate": 4.7760888749230414e-06, |
|
"log_odds_chosen": 0.4831446707248688, |
|
"log_odds_ratio": -0.5493755340576172, |
|
"logits/chosen": -3.095670461654663, |
|
"logits/rejected": -3.0957283973693848, |
|
"logps/chosen": -0.8023236989974976, |
|
"logps/rejected": -1.0722805261611938, |
|
"loss": 0.5964, |
|
"nll_loss": 0.4757974147796631, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -0.16046474874019623, |
|
"rewards/margins": 0.05399137735366821, |
|
"rewards/rejected": -0.21445612609386444, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"grad_norm": 3.09375, |
|
"learning_rate": 4.696901454167989e-06, |
|
"log_odds_chosen": 0.3349444270133972, |
|
"log_odds_ratio": -0.6006864309310913, |
|
"logits/chosen": -3.0792784690856934, |
|
"logits/rejected": -3.073970079421997, |
|
"logps/chosen": -0.8161932229995728, |
|
"logps/rejected": -0.9950379133224487, |
|
"loss": 0.6022, |
|
"nll_loss": 0.48054131865501404, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -0.16323864459991455, |
|
"rewards/margins": 0.035768941044807434, |
|
"rewards/rejected": -0.19900760054588318, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"grad_norm": 3.171875, |
|
"learning_rate": 4.6066201667762944e-06, |
|
"log_odds_chosen": 0.46497654914855957, |
|
"log_odds_ratio": -0.5399398803710938, |
|
"logits/chosen": -3.1036829948425293, |
|
"logits/rejected": -3.099396228790283, |
|
"logps/chosen": -0.8102518916130066, |
|
"logps/rejected": -1.0698442459106445, |
|
"loss": 0.5611, |
|
"nll_loss": 0.47170430421829224, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -0.16205039620399475, |
|
"rewards/margins": 0.05191843584179878, |
|
"rewards/rejected": -0.21396884322166443, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"grad_norm": 3.234375, |
|
"learning_rate": 4.505700913289246e-06, |
|
"log_odds_chosen": 0.385485976934433, |
|
"log_odds_ratio": -0.5975229144096375, |
|
"logits/chosen": -3.096364974975586, |
|
"logits/rejected": -3.085583209991455, |
|
"logps/chosen": -0.8479864001274109, |
|
"logps/rejected": -1.076240062713623, |
|
"loss": 0.6009, |
|
"nll_loss": 0.47817081212997437, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -0.1695972979068756, |
|
"rewards/margins": 0.04565075412392616, |
|
"rewards/rejected": -0.21524803340435028, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"grad_norm": 2.890625, |
|
"learning_rate": 4.3946533136249926e-06, |
|
"log_odds_chosen": 0.43778282403945923, |
|
"log_odds_ratio": -0.5804494619369507, |
|
"logits/chosen": -3.0998694896698, |
|
"logits/rejected": -3.0987915992736816, |
|
"logps/chosen": -0.8241409063339233, |
|
"logps/rejected": -1.0574108362197876, |
|
"loss": 0.5988, |
|
"nll_loss": 0.4965757429599762, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -0.16482816636562347, |
|
"rewards/margins": 0.046653974801301956, |
|
"rewards/rejected": -0.21148213744163513, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"grad_norm": 3.203125, |
|
"learning_rate": 4.274038133610629e-06, |
|
"log_odds_chosen": 0.46187520027160645, |
|
"log_odds_ratio": -0.5494967103004456, |
|
"logits/chosen": -3.1161484718322754, |
|
"logits/rejected": -3.114893913269043, |
|
"logps/chosen": -0.7468121647834778, |
|
"logps/rejected": -0.9928558468818665, |
|
"loss": 0.5811, |
|
"nll_loss": 0.4424869120121002, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -0.14936241507530212, |
|
"rewards/margins": 0.04920873045921326, |
|
"rewards/rejected": -0.19857117533683777, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"grad_norm": 3.375, |
|
"learning_rate": 4.1444644532387485e-06, |
|
"log_odds_chosen": 0.4036913812160492, |
|
"log_odds_ratio": -0.6148982048034668, |
|
"logits/chosen": -3.0943081378936768, |
|
"logits/rejected": -3.089369297027588, |
|
"logps/chosen": -0.8185051083564758, |
|
"logps/rejected": -1.0283105373382568, |
|
"loss": 0.5743, |
|
"nll_loss": 0.4471573233604431, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.16370102763175964, |
|
"rewards/margins": 0.04196108505129814, |
|
"rewards/rejected": -0.2056621015071869, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"grad_norm": 3.109375, |
|
"learning_rate": 4.006586590948141e-06, |
|
"log_odds_chosen": 0.38943806290626526, |
|
"log_odds_ratio": -0.5949686169624329, |
|
"logits/chosen": -3.105708360671997, |
|
"logits/rejected": -3.1006293296813965, |
|
"logps/chosen": -0.8150092363357544, |
|
"logps/rejected": -1.0240681171417236, |
|
"loss": 0.5787, |
|
"nll_loss": 0.46157145500183105, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.16300183534622192, |
|
"rewards/margins": 0.041811779141426086, |
|
"rewards/rejected": -0.20481359958648682, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"grad_norm": 3.109375, |
|
"learning_rate": 3.861100799460336e-06, |
|
"log_odds_chosen": 0.5897554159164429, |
|
"log_odds_ratio": -0.542380690574646, |
|
"logits/chosen": -3.100862503051758, |
|
"logits/rejected": -3.0946009159088135, |
|
"logps/chosen": -0.7791546583175659, |
|
"logps/rejected": -1.0703848600387573, |
|
"loss": 0.5747, |
|
"nll_loss": 0.5026167631149292, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -0.15583094954490662, |
|
"rewards/margins": 0.05824603885412216, |
|
"rewards/rejected": -0.214076966047287, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"grad_norm": 3.03125, |
|
"learning_rate": 3.7087417498572946e-06, |
|
"log_odds_chosen": 0.4274469316005707, |
|
"log_odds_ratio": -0.5712376236915588, |
|
"logits/chosen": -3.102602005004883, |
|
"logits/rejected": -3.1062116622924805, |
|
"logps/chosen": -0.8305424451828003, |
|
"logps/rejected": -1.077083945274353, |
|
"loss": 0.5692, |
|
"nll_loss": 0.4742380678653717, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -0.16610850393772125, |
|
"rewards/margins": 0.04930829256772995, |
|
"rewards/rejected": -0.2154167890548706, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.72, |
|
"grad_norm": 3.28125, |
|
"learning_rate": 3.550278821654866e-06, |
|
"log_odds_chosen": 0.34345707297325134, |
|
"log_odds_ratio": -0.6129624247550964, |
|
"logits/chosen": -3.095494508743286, |
|
"logits/rejected": -3.0863471031188965, |
|
"logps/chosen": -0.8034600019454956, |
|
"logps/rejected": -0.9921743273735046, |
|
"loss": 0.587, |
|
"nll_loss": 0.4692765176296234, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -0.1606920212507248, |
|
"rewards/margins": 0.037742845714092255, |
|
"rewards/rejected": -0.19843485951423645, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"grad_norm": 4.28125, |
|
"learning_rate": 3.386512217606339e-06, |
|
"log_odds_chosen": 0.3175373375415802, |
|
"log_odds_ratio": -0.6278958320617676, |
|
"logits/chosen": -3.1244797706604004, |
|
"logits/rejected": -3.1204514503479004, |
|
"logps/chosen": -0.7932506799697876, |
|
"logps/rejected": -0.9527732133865356, |
|
"loss": 0.6102, |
|
"nll_loss": 0.472064733505249, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.15865013003349304, |
|
"rewards/margins": 0.031904518604278564, |
|
"rewards/rejected": -0.1905546486377716, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"grad_norm": 3.765625, |
|
"learning_rate": 3.218268922855452e-06, |
|
"log_odds_chosen": 0.42049679160118103, |
|
"log_odds_ratio": -0.5808514356613159, |
|
"logits/chosen": -3.1028733253479004, |
|
"logits/rejected": -3.0980591773986816, |
|
"logps/chosen": -0.7747179269790649, |
|
"logps/rejected": -0.9938864707946777, |
|
"loss": 0.5825, |
|
"nll_loss": 0.44331270456314087, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.154943585395813, |
|
"rewards/margins": 0.04383372142910957, |
|
"rewards/rejected": -0.19877730309963226, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"grad_norm": 3.328125, |
|
"learning_rate": 3.046398528844248e-06, |
|
"log_odds_chosen": 0.5076274275779724, |
|
"log_odds_ratio": -0.5742210745811462, |
|
"logits/chosen": -3.104665517807007, |
|
"logits/rejected": -3.095426082611084, |
|
"logps/chosen": -0.776441216468811, |
|
"logps/rejected": -1.0134727954864502, |
|
"loss": 0.5712, |
|
"nll_loss": 0.4288255274295807, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -0.15528824925422668, |
|
"rewards/margins": 0.04740630090236664, |
|
"rewards/rejected": -0.20269456505775452, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"grad_norm": 2.953125, |
|
"learning_rate": 2.871768943064129e-06, |
|
"log_odds_chosen": 0.3015575408935547, |
|
"log_odds_ratio": -0.6296088695526123, |
|
"logits/chosen": -3.0799665451049805, |
|
"logits/rejected": -3.07787823677063, |
|
"logps/chosen": -0.8669988512992859, |
|
"logps/rejected": -1.0326287746429443, |
|
"loss": 0.5744, |
|
"nll_loss": 0.47870683670043945, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.17339977622032166, |
|
"rewards/margins": 0.033125996589660645, |
|
"rewards/rejected": -0.2065257579088211, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.95, |
|
"grad_norm": 2.859375, |
|
"learning_rate": 2.695262006314912e-06, |
|
"log_odds_chosen": 0.3043849766254425, |
|
"log_odds_ratio": -0.6296250224113464, |
|
"logits/chosen": -3.0792198181152344, |
|
"logits/rejected": -3.0685715675354004, |
|
"logps/chosen": -0.8420785069465637, |
|
"logps/rejected": -1.0109227895736694, |
|
"loss": 0.5851, |
|
"nll_loss": 0.49167829751968384, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.16841569542884827, |
|
"rewards/margins": 0.03376886993646622, |
|
"rewards/rejected": -0.20218458771705627, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"grad_norm": 3.390625, |
|
"learning_rate": 2.517769039603744e-06, |
|
"log_odds_chosen": 0.47437596321105957, |
|
"log_odds_ratio": -0.5693033933639526, |
|
"logits/chosen": -3.0923125743865967, |
|
"logits/rejected": -3.08634614944458, |
|
"logps/chosen": -0.7851861119270325, |
|
"logps/rejected": -1.0160605907440186, |
|
"loss": 0.6031, |
|
"nll_loss": 0.4828670620918274, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -0.15703722834587097, |
|
"rewards/margins": 0.04617489501833916, |
|
"rewards/rejected": -0.20321211218833923, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 1.04, |
|
"grad_norm": 5.125, |
|
"learning_rate": 2.3401863431710864e-06, |
|
"log_odds_chosen": 0.8126243352890015, |
|
"log_odds_ratio": -0.4562348425388336, |
|
"logits/chosen": -3.0906317234039307, |
|
"logits/rejected": -3.08894681930542, |
|
"logps/chosen": -0.6749258041381836, |
|
"logps/rejected": -1.0676677227020264, |
|
"loss": 0.5063, |
|
"nll_loss": 0.4003027081489563, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -0.13498517870903015, |
|
"rewards/margins": 0.0785483792424202, |
|
"rewards/rejected": -0.21353355050086975, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 1.08, |
|
"grad_norm": 3.34375, |
|
"learning_rate": 2.163410670372652e-06, |
|
"log_odds_chosen": 0.8564330339431763, |
|
"log_odds_ratio": -0.4268684387207031, |
|
"logits/chosen": -3.0698952674865723, |
|
"logits/rejected": -3.070732831954956, |
|
"logps/chosen": -0.7170256972312927, |
|
"logps/rejected": -1.1589481830596924, |
|
"loss": 0.4975, |
|
"nll_loss": 0.45029813051223755, |
|
"rewards/accuracies": 0.862500011920929, |
|
"rewards/chosen": -0.14340513944625854, |
|
"rewards/margins": 0.08838452398777008, |
|
"rewards/rejected": -0.23178966343402863, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 1.13, |
|
"grad_norm": 3.34375, |
|
"learning_rate": 1.9883346992732256e-06, |
|
"log_odds_chosen": 1.0222368240356445, |
|
"log_odds_ratio": -0.36799290776252747, |
|
"logits/chosen": -3.055107593536377, |
|
"logits/rejected": -3.045011281967163, |
|
"logps/chosen": -0.6467097997665405, |
|
"logps/rejected": -1.2033334970474243, |
|
"loss": 0.4936, |
|
"nll_loss": 0.3834058940410614, |
|
"rewards/accuracies": 0.9125000238418579, |
|
"rewards/chosen": -0.1293419599533081, |
|
"rewards/margins": 0.1113247275352478, |
|
"rewards/rejected": -0.24066667258739471, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 1.17, |
|
"grad_norm": 2.921875, |
|
"learning_rate": 1.8158425248197931e-06, |
|
"log_odds_chosen": 0.9676607251167297, |
|
"log_odds_ratio": -0.3842445909976959, |
|
"logits/chosen": -3.078850269317627, |
|
"logits/rejected": -3.0772275924682617, |
|
"logps/chosen": -0.6771007180213928, |
|
"logps/rejected": -1.177585482597351, |
|
"loss": 0.5028, |
|
"nll_loss": 0.44024211168289185, |
|
"rewards/accuracies": 0.887499988079071, |
|
"rewards/chosen": -0.13542014360427856, |
|
"rewards/margins": 0.10009696334600449, |
|
"rewards/rejected": -0.23551711440086365, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 1.22, |
|
"grad_norm": 3.1875, |
|
"learning_rate": 1.6468051943575242e-06, |
|
"log_odds_chosen": 0.9456725120544434, |
|
"log_odds_ratio": -0.39273351430892944, |
|
"logits/chosen": -3.085444450378418, |
|
"logits/rejected": -3.08333420753479, |
|
"logps/chosen": -0.6596750020980835, |
|
"logps/rejected": -1.1495213508605957, |
|
"loss": 0.4758, |
|
"nll_loss": 0.3547888398170471, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -0.1319349855184555, |
|
"rewards/margins": 0.09796925634145737, |
|
"rewards/rejected": -0.22990426421165466, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 1.26, |
|
"grad_norm": 3.328125, |
|
"learning_rate": 1.482076309033254e-06, |
|
"log_odds_chosen": 0.8887540102005005, |
|
"log_odds_ratio": -0.43248239159584045, |
|
"logits/chosen": -3.0775530338287354, |
|
"logits/rejected": -3.0756869316101074, |
|
"logps/chosen": -0.6567140817642212, |
|
"logps/rejected": -1.1050218343734741, |
|
"loss": 0.4872, |
|
"nll_loss": 0.3975124657154083, |
|
"rewards/accuracies": 0.8500000238418579, |
|
"rewards/chosen": -0.1313427984714508, |
|
"rewards/margins": 0.08966155350208282, |
|
"rewards/rejected": -0.22100436687469482, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 1.31, |
|
"grad_norm": 2.96875, |
|
"learning_rate": 1.3224877132984131e-06, |
|
"log_odds_chosen": 1.054574728012085, |
|
"log_odds_ratio": -0.3588215708732605, |
|
"logits/chosen": -3.103698253631592, |
|
"logits/rejected": -3.107422351837158, |
|
"logps/chosen": -0.6697841882705688, |
|
"logps/rejected": -1.2304431200027466, |
|
"loss": 0.4864, |
|
"nll_loss": 0.4196816384792328, |
|
"rewards/accuracies": 0.918749988079071, |
|
"rewards/chosen": -0.13395683467388153, |
|
"rewards/margins": 0.11213177442550659, |
|
"rewards/rejected": -0.24608862400054932, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 1.35, |
|
"grad_norm": 3.109375, |
|
"learning_rate": 1.1688452942784592e-06, |
|
"log_odds_chosen": 1.018751621246338, |
|
"log_odds_ratio": -0.3791121542453766, |
|
"logits/chosen": -3.0898470878601074, |
|
"logits/rejected": -3.0825257301330566, |
|
"logps/chosen": -0.6466778516769409, |
|
"logps/rejected": -1.1818552017211914, |
|
"loss": 0.4914, |
|
"nll_loss": 0.3884134292602539, |
|
"rewards/accuracies": 0.893750011920929, |
|
"rewards/chosen": -0.12933556735515594, |
|
"rewards/margins": 0.10703550279140472, |
|
"rewards/rejected": -0.23637107014656067, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 1.4, |
|
"grad_norm": 2.984375, |
|
"learning_rate": 1.021924912221062e-06, |
|
"log_odds_chosen": 0.8282191157341003, |
|
"log_odds_ratio": -0.42912721633911133, |
|
"logits/chosen": -3.1064085960388184, |
|
"logits/rejected": -3.105886697769165, |
|
"logps/chosen": -0.669070839881897, |
|
"logps/rejected": -1.1031153202056885, |
|
"loss": 0.4969, |
|
"nll_loss": 0.4509221017360687, |
|
"rewards/accuracies": 0.8687499761581421, |
|
"rewards/chosen": -0.13381414115428925, |
|
"rewards/margins": 0.08680891990661621, |
|
"rewards/rejected": -0.22062306106090546, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 1.44, |
|
"grad_norm": 3.515625, |
|
"learning_rate": 8.824684825733865e-07, |
|
"log_odds_chosen": 1.0580097436904907, |
|
"log_odds_ratio": -0.38433724641799927, |
|
"logits/chosen": -3.081042528152466, |
|
"logits/rejected": -3.075192451477051, |
|
"logps/chosen": -0.6307107210159302, |
|
"logps/rejected": -1.1325790882110596, |
|
"loss": 0.493, |
|
"nll_loss": 0.4108586311340332, |
|
"rewards/accuracies": 0.887499988079071, |
|
"rewards/chosen": -0.12614212930202484, |
|
"rewards/margins": 0.10037368535995483, |
|
"rewards/rejected": -0.22651581466197968, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 1.49, |
|
"grad_norm": 3.046875, |
|
"learning_rate": 7.51180229473116e-07, |
|
"log_odds_chosen": 0.9154760241508484, |
|
"log_odds_ratio": -0.415397584438324, |
|
"logits/chosen": -3.0937957763671875, |
|
"logits/rejected": -3.0928263664245605, |
|
"logps/chosen": -0.6775897741317749, |
|
"logps/rejected": -1.1316546201705933, |
|
"loss": 0.5044, |
|
"nll_loss": 0.4353519380092621, |
|
"rewards/accuracies": 0.8500000238418579, |
|
"rewards/chosen": -0.13551795482635498, |
|
"rewards/margins": 0.09081296622753143, |
|
"rewards/rejected": -0.22633090615272522, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 1.53, |
|
"grad_norm": 3.15625, |
|
"learning_rate": 6.28723129572247e-07, |
|
"log_odds_chosen": 1.139094352722168, |
|
"log_odds_ratio": -0.351654052734375, |
|
"logits/chosen": -3.077972888946533, |
|
"logits/rejected": -3.0774223804473877, |
|
"logps/chosen": -0.6138228178024292, |
|
"logps/rejected": -1.158979058265686, |
|
"loss": 0.4728, |
|
"nll_loss": 0.3857228457927704, |
|
"rewards/accuracies": 0.90625, |
|
"rewards/chosen": -0.12276456505060196, |
|
"rewards/margins": 0.10903123766183853, |
|
"rewards/rejected": -0.2317957878112793, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 1.58, |
|
"grad_norm": 3.234375, |
|
"learning_rate": 5.157155641515766e-07, |
|
"log_odds_chosen": 1.0156279802322388, |
|
"log_odds_ratio": -0.3680209219455719, |
|
"logits/chosen": -3.093818426132202, |
|
"logits/rejected": -3.095491409301758, |
|
"logps/chosen": -0.6614006161689758, |
|
"logps/rejected": -1.1964319944381714, |
|
"loss": 0.5024, |
|
"nll_loss": 0.4554772973060608, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -0.13228009641170502, |
|
"rewards/margins": 0.10700628906488419, |
|
"rewards/rejected": -0.2392864227294922, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 1.63, |
|
"grad_norm": 3.109375, |
|
"learning_rate": 4.127281964319446e-07, |
|
"log_odds_chosen": 1.1072419881820679, |
|
"log_odds_ratio": -0.365578830242157, |
|
"logits/chosen": -3.0632922649383545, |
|
"logits/rejected": -3.053469657897949, |
|
"logps/chosen": -0.6406615376472473, |
|
"logps/rejected": -1.1723295450210571, |
|
"loss": 0.4717, |
|
"nll_loss": 0.39158958196640015, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -0.12813231348991394, |
|
"rewards/margins": 0.10633359849452972, |
|
"rewards/rejected": -0.23446591198444366, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 1.67, |
|
"grad_norm": 3.171875, |
|
"learning_rate": 3.202810898511424e-07, |
|
"log_odds_chosen": 1.0118036270141602, |
|
"log_odds_ratio": -0.3721558451652527, |
|
"logits/chosen": -3.0760838985443115, |
|
"logits/rejected": -3.0786654949188232, |
|
"logps/chosen": -0.6295793652534485, |
|
"logps/rejected": -1.1554322242736816, |
|
"loss": 0.4954, |
|
"nll_loss": 0.3821503221988678, |
|
"rewards/accuracies": 0.893750011920929, |
|
"rewards/chosen": -0.12591585516929626, |
|
"rewards/margins": 0.1051705852150917, |
|
"rewards/rejected": -0.23108646273612976, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 1.72, |
|
"grad_norm": 3.015625, |
|
"learning_rate": 2.388410818585263e-07, |
|
"log_odds_chosen": 1.0309429168701172, |
|
"log_odds_ratio": -0.3757808804512024, |
|
"logits/chosen": -3.0944666862487793, |
|
"logits/rejected": -3.102954149246216, |
|
"logps/chosen": -0.6560705304145813, |
|
"logps/rejected": -1.1918388605117798, |
|
"loss": 0.4848, |
|
"nll_loss": 0.4180088937282562, |
|
"rewards/accuracies": 0.893750011920929, |
|
"rewards/chosen": -0.13121411204338074, |
|
"rewards/margins": 0.10715366899967194, |
|
"rewards/rejected": -0.23836776614189148, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 1.76, |
|
"grad_norm": 3.0625, |
|
"learning_rate": 1.6881942648911077e-07, |
|
"log_odds_chosen": 1.045115351676941, |
|
"log_odds_ratio": -0.3773229420185089, |
|
"logits/chosen": -3.099276304244995, |
|
"logits/rejected": -3.0971086025238037, |
|
"logps/chosen": -0.6689329743385315, |
|
"logps/rejected": -1.209533452987671, |
|
"loss": 0.4865, |
|
"nll_loss": 0.4016016125679016, |
|
"rewards/accuracies": 0.887499988079071, |
|
"rewards/chosen": -0.13378658890724182, |
|
"rewards/margins": 0.10812009871006012, |
|
"rewards/rejected": -0.24190667271614075, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 1.81, |
|
"grad_norm": 3.109375, |
|
"learning_rate": 1.1056971762161584e-07, |
|
"log_odds_chosen": 0.8860037922859192, |
|
"log_odds_ratio": -0.4059659540653229, |
|
"logits/chosen": -3.0970611572265625, |
|
"logits/rejected": -3.0827975273132324, |
|
"logps/chosen": -0.6797454357147217, |
|
"logps/rejected": -1.1361665725708008, |
|
"loss": 0.4883, |
|
"nll_loss": 0.4087887406349182, |
|
"rewards/accuracies": 0.90625, |
|
"rewards/chosen": -0.13594909012317657, |
|
"rewards/margins": 0.09128421545028687, |
|
"rewards/rejected": -0.22723329067230225, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 1.85, |
|
"grad_norm": 3.015625, |
|
"learning_rate": 6.438610340755464e-08, |
|
"log_odds_chosen": 0.9491767883300781, |
|
"log_odds_ratio": -0.3955945372581482, |
|
"logits/chosen": -3.095716953277588, |
|
"logits/rejected": -3.0901029109954834, |
|
"logps/chosen": -0.6622263193130493, |
|
"logps/rejected": -1.1627901792526245, |
|
"loss": 0.4715, |
|
"nll_loss": 0.4081706404685974, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -0.13244526088237762, |
|
"rewards/margins": 0.10011278092861176, |
|
"rewards/rejected": -0.23255804181098938, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 1.9, |
|
"grad_norm": 3.328125, |
|
"learning_rate": 3.050180088809973e-08, |
|
"log_odds_chosen": 0.89338219165802, |
|
"log_odds_ratio": -0.41199594736099243, |
|
"logits/chosen": -3.078094482421875, |
|
"logits/rejected": -3.074589252471924, |
|
"logps/chosen": -0.6442387700080872, |
|
"logps/rejected": -1.103659987449646, |
|
"loss": 0.4976, |
|
"nll_loss": 0.40062204003334045, |
|
"rewards/accuracies": 0.862500011920929, |
|
"rewards/chosen": -0.12884774804115295, |
|
"rewards/margins": 0.09188426285982132, |
|
"rewards/rejected": -0.22073200345039368, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 1.94, |
|
"grad_norm": 3.15625, |
|
"learning_rate": 9.087918299586772e-09, |
|
"log_odds_chosen": 0.9813147783279419, |
|
"log_odds_ratio": -0.4028463363647461, |
|
"logits/chosen": -3.079704523086548, |
|
"logits/rejected": -3.0714404582977295, |
|
"logps/chosen": -0.6575345993041992, |
|
"logps/rejected": -1.1326944828033447, |
|
"loss": 0.4832, |
|
"nll_loss": 0.39662784337997437, |
|
"rewards/accuracies": 0.856249988079071, |
|
"rewards/chosen": -0.13150693476200104, |
|
"rewards/margins": 0.0950319841504097, |
|
"rewards/rejected": -0.22653889656066895, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 1.99, |
|
"grad_norm": 4.03125, |
|
"learning_rate": 2.525910147516131e-10, |
|
"log_odds_chosen": 0.966279149055481, |
|
"log_odds_ratio": -0.4019397795200348, |
|
"logits/chosen": -3.0766844749450684, |
|
"logits/rejected": -3.0788207054138184, |
|
"logps/chosen": -0.6589742302894592, |
|
"logps/rejected": -1.1595604419708252, |
|
"loss": 0.48, |
|
"nll_loss": 0.38374659419059753, |
|
"rewards/accuracies": 0.8500000238418579, |
|
"rewards/chosen": -0.13179484009742737, |
|
"rewards/margins": 0.10011725127696991, |
|
"rewards/rejected": -0.23191209137439728, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"step": 442, |
|
"total_flos": 0.0, |
|
"train_loss": 0.5399103337283587, |
|
"train_runtime": 8872.0808, |
|
"train_samples_per_second": 3.191, |
|
"train_steps_per_second": 0.05 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 442, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 2, |
|
"save_steps": 500, |
|
"total_flos": 0.0, |
|
"train_batch_size": 8, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|