| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.0, | |
| "eval_steps": 100, | |
| "global_step": 385, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.0, | |
| "learning_rate": 1.282051282051282e-08, | |
| "logits/chosen": -1.7278180122375488, | |
| "logits/rejected": -1.7377450466156006, | |
| "logps/chosen": -29.553977966308594, | |
| "logps/rejected": -42.813133239746094, | |
| "loss": 0.3086, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "learning_rate": 1.2820512820512818e-07, | |
| "logits/chosen": -1.866453766822815, | |
| "logits/rejected": -1.8707627058029175, | |
| "logps/chosen": -36.98646545410156, | |
| "logps/rejected": -33.6445426940918, | |
| "loss": 0.3098, | |
| "rewards/accuracies": 0.4444444477558136, | |
| "rewards/chosen": 0.018122652545571327, | |
| "rewards/margins": 0.031144678592681885, | |
| "rewards/rejected": -0.013022023253142834, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "learning_rate": 2.5641025641025636e-07, | |
| "logits/chosen": -1.9970738887786865, | |
| "logits/rejected": -1.9997297525405884, | |
| "logps/chosen": -29.651851654052734, | |
| "logps/rejected": -29.048221588134766, | |
| "loss": 0.3674, | |
| "rewards/accuracies": 0.4375, | |
| "rewards/chosen": -0.008690872229635715, | |
| "rewards/margins": -0.02239001728594303, | |
| "rewards/rejected": 0.01369914598762989, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "learning_rate": 3.8461538461538463e-07, | |
| "logits/chosen": -1.919904351234436, | |
| "logits/rejected": -1.9172093868255615, | |
| "logps/chosen": -31.407363891601562, | |
| "logps/rejected": -33.22914505004883, | |
| "loss": 0.3289, | |
| "rewards/accuracies": 0.5874999761581421, | |
| "rewards/chosen": 0.007855755276978016, | |
| "rewards/margins": 0.017470674589276314, | |
| "rewards/rejected": -0.009614917449653149, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "learning_rate": 4.999896948438433e-07, | |
| "logits/chosen": -2.016488552093506, | |
| "logits/rejected": -2.0077414512634277, | |
| "logps/chosen": -32.588050842285156, | |
| "logps/rejected": -32.51618576049805, | |
| "loss": 0.3526, | |
| "rewards/accuracies": 0.4749999940395355, | |
| "rewards/chosen": -0.010120267979800701, | |
| "rewards/margins": -0.006958496756851673, | |
| "rewards/rejected": -0.0031617730855941772, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "learning_rate": 4.987541037542186e-07, | |
| "logits/chosen": -1.8645082712173462, | |
| "logits/rejected": -1.8537276983261108, | |
| "logps/chosen": -33.5635871887207, | |
| "logps/rejected": -35.42169952392578, | |
| "loss": 0.3858, | |
| "rewards/accuracies": 0.38749998807907104, | |
| "rewards/chosen": -0.005497059319168329, | |
| "rewards/margins": -0.03307711333036423, | |
| "rewards/rejected": 0.027580058202147484, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "learning_rate": 4.954691471941118e-07, | |
| "logits/chosen": -1.9458835124969482, | |
| "logits/rejected": -1.9478065967559814, | |
| "logps/chosen": -32.57460021972656, | |
| "logps/rejected": -33.195518493652344, | |
| "loss": 0.3015, | |
| "rewards/accuracies": 0.574999988079071, | |
| "rewards/chosen": 0.02302517369389534, | |
| "rewards/margins": 0.04474987834692001, | |
| "rewards/rejected": -0.021724706515669823, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "learning_rate": 4.901618883413548e-07, | |
| "logits/chosen": -2.0793328285217285, | |
| "logits/rejected": -2.084324836730957, | |
| "logps/chosen": -33.99799728393555, | |
| "logps/rejected": -36.57799530029297, | |
| "loss": 0.3658, | |
| "rewards/accuracies": 0.48750001192092896, | |
| "rewards/chosen": -0.00855917576700449, | |
| "rewards/margins": -0.00035296304849907756, | |
| "rewards/rejected": -0.008206211030483246, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "learning_rate": 4.828760511501322e-07, | |
| "logits/chosen": -1.9421542882919312, | |
| "logits/rejected": -1.9453080892562866, | |
| "logps/chosen": -34.38970184326172, | |
| "logps/rejected": -34.58138656616211, | |
| "loss": 0.3181, | |
| "rewards/accuracies": 0.5874999761581421, | |
| "rewards/chosen": 0.034300822764635086, | |
| "rewards/margins": 0.024381538853049278, | |
| "rewards/rejected": 0.009919276461005211, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "learning_rate": 4.736716601303429e-07, | |
| "logits/chosen": -1.9507806301116943, | |
| "logits/rejected": -1.955293893814087, | |
| "logps/chosen": -32.461936950683594, | |
| "logps/rejected": -32.3816032409668, | |
| "loss": 0.3094, | |
| "rewards/accuracies": 0.574999988079071, | |
| "rewards/chosen": 0.004934861324727535, | |
| "rewards/margins": 0.02535998821258545, | |
| "rewards/rejected": -0.02042512409389019, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "learning_rate": 4.62624545834521e-07, | |
| "logits/chosen": -2.0485286712646484, | |
| "logits/rejected": -2.046539783477783, | |
| "logps/chosen": -32.23391342163086, | |
| "logps/rejected": -31.265544891357422, | |
| "loss": 0.333, | |
| "rewards/accuracies": 0.512499988079071, | |
| "rewards/chosen": 0.006228716112673283, | |
| "rewards/margins": 0.005629505030810833, | |
| "rewards/rejected": 0.0005992107326164842, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "eval_logits/chosen": -2.243820905685425, | |
| "eval_logits/rejected": -2.238938570022583, | |
| "eval_logps/chosen": -34.02050018310547, | |
| "eval_logps/rejected": -37.50347137451172, | |
| "eval_loss": 0.3404548168182373, | |
| "eval_rewards/accuracies": 0.5103820562362671, | |
| "eval_rewards/chosen": 0.012649022974073887, | |
| "eval_rewards/margins": 0.0008147989865392447, | |
| "eval_rewards/rejected": 0.011834223754703999, | |
| "eval_runtime": 146.2563, | |
| "eval_samples_per_second": 2.345, | |
| "eval_steps_per_second": 0.294, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 4.4982572012636904e-07, | |
| "logits/chosen": -2.0055313110351562, | |
| "logits/rejected": -2.00311541557312, | |
| "logps/chosen": -33.23518753051758, | |
| "logps/rejected": -34.03091812133789, | |
| "loss": 0.3233, | |
| "rewards/accuracies": 0.48750001192092896, | |
| "rewards/chosen": 0.007764431647956371, | |
| "rewards/margins": 0.009159245528280735, | |
| "rewards/rejected": -0.0013948131818324327, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "learning_rate": 4.353806263777677e-07, | |
| "logits/chosen": -2.016556978225708, | |
| "logits/rejected": -2.0081825256347656, | |
| "logps/chosen": -32.435211181640625, | |
| "logps/rejected": -32.18501281738281, | |
| "loss": 0.3121, | |
| "rewards/accuracies": 0.5625, | |
| "rewards/chosen": 0.00872363243252039, | |
| "rewards/margins": 0.029877260327339172, | |
| "rewards/rejected": -0.02115362510085106, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.34, | |
| "learning_rate": 4.194082707715275e-07, | |
| "logits/chosen": -2.0461266040802, | |
| "logits/rejected": -2.03806734085083, | |
| "logps/chosen": -30.502710342407227, | |
| "logps/rejected": -32.0546760559082, | |
| "loss": 0.3674, | |
| "rewards/accuracies": 0.4625000059604645, | |
| "rewards/chosen": -0.02429451420903206, | |
| "rewards/margins": -0.02279272861778736, | |
| "rewards/rejected": -0.0015017850091680884, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "learning_rate": 4.020402418666621e-07, | |
| "logits/chosen": -1.9767634868621826, | |
| "logits/rejected": -1.9870494604110718, | |
| "logps/chosen": -31.402328491210938, | |
| "logps/rejected": -32.55817794799805, | |
| "loss": 0.3125, | |
| "rewards/accuracies": 0.550000011920929, | |
| "rewards/chosen": 0.015601096674799919, | |
| "rewards/margins": 0.03657008707523346, | |
| "rewards/rejected": -0.02096899226307869, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "learning_rate": 3.8341962650351185e-07, | |
| "logits/chosen": -1.8908436298370361, | |
| "logits/rejected": -1.8919413089752197, | |
| "logps/chosen": -34.207401275634766, | |
| "logps/rejected": -34.783695220947266, | |
| "loss": 0.3373, | |
| "rewards/accuracies": 0.5375000238418579, | |
| "rewards/chosen": -0.013235817663371563, | |
| "rewards/margins": 0.01055879332125187, | |
| "rewards/rejected": -0.023794615641236305, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "learning_rate": 3.636998309800572e-07, | |
| "logits/chosen": -1.942582368850708, | |
| "logits/rejected": -1.9391053915023804, | |
| "logps/chosen": -36.15548324584961, | |
| "logps/rejected": -32.729759216308594, | |
| "loss": 0.3207, | |
| "rewards/accuracies": 0.5, | |
| "rewards/chosen": 0.014574882574379444, | |
| "rewards/margins": 0.020880455151200294, | |
| "rewards/rejected": -0.006305573042482138, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.44, | |
| "learning_rate": 3.430433172111807e-07, | |
| "logits/chosen": -2.042206287384033, | |
| "logits/rejected": -2.034825086593628, | |
| "logps/chosen": -33.77631378173828, | |
| "logps/rejected": -31.35869789123535, | |
| "loss": 0.3201, | |
| "rewards/accuracies": 0.5625, | |
| "rewards/chosen": 0.019394155591726303, | |
| "rewards/margins": 0.025556519627571106, | |
| "rewards/rejected": -0.00616236450150609, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.47, | |
| "learning_rate": 3.216202642830543e-07, | |
| "logits/chosen": -2.0474154949188232, | |
| "logits/rejected": -2.0526936054229736, | |
| "logps/chosen": -32.521339416503906, | |
| "logps/rejected": -32.497581481933594, | |
| "loss": 0.2999, | |
| "rewards/accuracies": 0.612500011920929, | |
| "rewards/chosen": 0.021103620529174805, | |
| "rewards/margins": 0.032301973551511765, | |
| "rewards/rejected": -0.01119835115969181, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.49, | |
| "learning_rate": 2.9960716642946403e-07, | |
| "logits/chosen": -2.0485098361968994, | |
| "logits/rejected": -2.0457375049591064, | |
| "logps/chosen": -31.49017906188965, | |
| "logps/rejected": -31.345867156982422, | |
| "loss": 0.3315, | |
| "rewards/accuracies": 0.5625, | |
| "rewards/chosen": -0.001104071387089789, | |
| "rewards/margins": 0.023781923577189445, | |
| "rewards/rejected": -0.02488599717617035, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.52, | |
| "learning_rate": 2.771853789806683e-07, | |
| "logits/chosen": -1.9183101654052734, | |
| "logits/rejected": -1.9229921102523804, | |
| "logps/chosen": -31.60622215270996, | |
| "logps/rejected": -32.805049896240234, | |
| "loss": 0.3306, | |
| "rewards/accuracies": 0.5249999761581421, | |
| "rewards/chosen": 5.159676220500842e-05, | |
| "rewards/margins": 0.012757277116179466, | |
| "rewards/rejected": -0.012705676257610321, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.52, | |
| "eval_logits/chosen": -2.2434937953948975, | |
| "eval_logits/rejected": -2.2386233806610107, | |
| "eval_logps/chosen": -34.02851867675781, | |
| "eval_logps/rejected": -37.51131057739258, | |
| "eval_loss": 0.3466452658176422, | |
| "eval_rewards/accuracies": 0.5020764470100403, | |
| "eval_rewards/chosen": 0.005428452976047993, | |
| "eval_rewards/margins": 0.0006523749325424433, | |
| "eval_rewards/rejected": 0.00477607874199748, | |
| "eval_runtime": 145.8217, | |
| "eval_samples_per_second": 2.352, | |
| "eval_steps_per_second": 0.295, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "learning_rate": 2.5453962426402e-07, | |
| "logits/chosen": -2.032118558883667, | |
| "logits/rejected": -2.042782783508301, | |
| "logps/chosen": -31.939167022705078, | |
| "logps/rejected": -33.88642501831055, | |
| "loss": 0.2871, | |
| "rewards/accuracies": 0.6000000238418579, | |
| "rewards/chosen": 0.02892182394862175, | |
| "rewards/margins": 0.050276655703783035, | |
| "rewards/rejected": -0.021354828029870987, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.57, | |
| "learning_rate": 2.318564697655179e-07, | |
| "logits/chosen": -1.925293207168579, | |
| "logits/rejected": -1.9401395320892334, | |
| "logps/chosen": -30.106586456298828, | |
| "logps/rejected": -31.56791114807129, | |
| "loss": 0.3327, | |
| "rewards/accuracies": 0.6000000238418579, | |
| "rewards/chosen": 0.0044566914439201355, | |
| "rewards/margins": 0.013218318112194538, | |
| "rewards/rejected": -0.008761629462242126, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.6, | |
| "learning_rate": 2.093227910899832e-07, | |
| "logits/chosen": -1.9831531047821045, | |
| "logits/rejected": -1.9871160984039307, | |
| "logps/chosen": -33.386024475097656, | |
| "logps/rejected": -31.5727481842041, | |
| "loss": 0.2974, | |
| "rewards/accuracies": 0.550000011920929, | |
| "rewards/chosen": 0.028298119083046913, | |
| "rewards/margins": 0.04939670115709305, | |
| "rewards/rejected": -0.021098580211400986, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.62, | |
| "learning_rate": 1.8712423238279356e-07, | |
| "logits/chosen": -1.9833450317382812, | |
| "logits/rejected": -1.9613698720932007, | |
| "logps/chosen": -34.1508903503418, | |
| "logps/rejected": -34.95817947387695, | |
| "loss": 0.3388, | |
| "rewards/accuracies": 0.5249999761581421, | |
| "rewards/chosen": 0.002569653559476137, | |
| "rewards/margins": 0.00921549927443266, | |
| "rewards/rejected": -0.006645847111940384, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.65, | |
| "learning_rate": 1.654436768970182e-07, | |
| "logits/chosen": -2.0248680114746094, | |
| "logits/rejected": -2.021559238433838, | |
| "logps/chosen": -32.907005310058594, | |
| "logps/rejected": -36.22737121582031, | |
| "loss": 0.3221, | |
| "rewards/accuracies": 0.550000011920929, | |
| "rewards/chosen": 0.016254058107733727, | |
| "rewards/margins": 0.018458303064107895, | |
| "rewards/rejected": -0.002204245189204812, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.68, | |
| "learning_rate": 1.444597403062196e-07, | |
| "logits/chosen": -1.8913663625717163, | |
| "logits/rejected": -1.8889217376708984, | |
| "logps/chosen": -34.18629455566406, | |
| "logps/rejected": -35.490962982177734, | |
| "loss": 0.3601, | |
| "rewards/accuracies": 0.5, | |
| "rewards/chosen": 0.0018952053505927324, | |
| "rewards/margins": -0.010397391393780708, | |
| "rewards/rejected": 0.012292595580220222, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "learning_rate": 1.2434529917578887e-07, | |
| "logits/chosen": -1.8756721019744873, | |
| "logits/rejected": -1.8731330633163452, | |
| "logps/chosen": -34.39625549316406, | |
| "logps/rejected": -31.76236343383789, | |
| "loss": 0.3492, | |
| "rewards/accuracies": 0.4625000059604645, | |
| "rewards/chosen": 0.001211274997331202, | |
| "rewards/margins": -0.006859898567199707, | |
| "rewards/rejected": 0.008071173913776875, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.73, | |
| "learning_rate": 1.0526606671603521e-07, | |
| "logits/chosen": -1.9795773029327393, | |
| "logits/rejected": -1.9689537286758423, | |
| "logps/chosen": -35.320404052734375, | |
| "logps/rejected": -31.86062240600586, | |
| "loss": 0.298, | |
| "rewards/accuracies": 0.574999988079071, | |
| "rewards/chosen": 0.032611697912216187, | |
| "rewards/margins": 0.05027870461344719, | |
| "rewards/rejected": -0.017666997388005257, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "learning_rate": 8.737922755071453e-08, | |
| "logits/chosen": -2.075582981109619, | |
| "logits/rejected": -2.0605711936950684, | |
| "logps/chosen": -30.907567977905273, | |
| "logps/rejected": -32.65256881713867, | |
| "loss": 0.3186, | |
| "rewards/accuracies": 0.512499988079071, | |
| "rewards/chosen": 0.014904007315635681, | |
| "rewards/margins": 0.015564213506877422, | |
| "rewards/rejected": -0.0006602049106732011, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "learning_rate": 7.08321427484816e-08, | |
| "logits/chosen": -1.9468443393707275, | |
| "logits/rejected": -1.944310188293457, | |
| "logps/chosen": -32.90850067138672, | |
| "logps/rejected": -30.801532745361328, | |
| "loss": 0.3327, | |
| "rewards/accuracies": 0.4749999940395355, | |
| "rewards/chosen": 0.016746358945965767, | |
| "rewards/margins": 0.002998964861035347, | |
| "rewards/rejected": 0.01374739408493042, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "eval_logits/chosen": -2.243875741958618, | |
| "eval_logits/rejected": -2.238999366760254, | |
| "eval_logps/chosen": -34.01763916015625, | |
| "eval_logps/rejected": -37.51422882080078, | |
| "eval_loss": 0.32764142751693726, | |
| "eval_rewards/accuracies": 0.5365448594093323, | |
| "eval_rewards/chosen": 0.015219501219689846, | |
| "eval_rewards/margins": 0.01306714303791523, | |
| "eval_rewards/rejected": 0.0021523565519601107, | |
| "eval_runtime": 145.9076, | |
| "eval_samples_per_second": 2.351, | |
| "eval_steps_per_second": 0.295, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.81, | |
| "learning_rate": 5.576113578589034e-08, | |
| "logits/chosen": -1.9285104274749756, | |
| "logits/rejected": -1.925238847732544, | |
| "logps/chosen": -31.57692527770996, | |
| "logps/rejected": -33.744144439697266, | |
| "loss": 0.3143, | |
| "rewards/accuracies": 0.550000011920929, | |
| "rewards/chosen": 0.023783722892403603, | |
| "rewards/margins": 0.025577496737241745, | |
| "rewards/rejected": -0.0017937703523784876, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "learning_rate": 4.229036944380912e-08, | |
| "logits/chosen": -1.980386734008789, | |
| "logits/rejected": -1.9680719375610352, | |
| "logps/chosen": -34.57415771484375, | |
| "logps/rejected": -33.55961227416992, | |
| "loss": 0.31, | |
| "rewards/accuracies": 0.5375000238418579, | |
| "rewards/chosen": 0.01972355879843235, | |
| "rewards/margins": 0.03841744735836983, | |
| "rewards/rejected": -0.018693890422582626, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 0.86, | |
| "learning_rate": 3.053082288996112e-08, | |
| "logits/chosen": -2.015927314758301, | |
| "logits/rejected": -2.014465570449829, | |
| "logps/chosen": -33.47549057006836, | |
| "logps/rejected": -32.47983169555664, | |
| "loss": 0.3454, | |
| "rewards/accuracies": 0.512499988079071, | |
| "rewards/chosen": 0.009270393289625645, | |
| "rewards/margins": 0.009491671808063984, | |
| "rewards/rejected": -0.0002212822437286377, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 0.88, | |
| "learning_rate": 2.05793773749158e-08, | |
| "logits/chosen": -2.102839946746826, | |
| "logits/rejected": -2.0870471000671387, | |
| "logps/chosen": -34.18696212768555, | |
| "logps/rejected": -33.075828552246094, | |
| "loss": 0.4092, | |
| "rewards/accuracies": 0.4124999940395355, | |
| "rewards/chosen": -0.006814646068960428, | |
| "rewards/margins": -0.04479615017771721, | |
| "rewards/rejected": 0.03798150271177292, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 0.91, | |
| "learning_rate": 1.251801807404168e-08, | |
| "logits/chosen": -1.9747005701065063, | |
| "logits/rejected": -1.9737575054168701, | |
| "logps/chosen": -33.231842041015625, | |
| "logps/rejected": -32.461700439453125, | |
| "loss": 0.3207, | |
| "rewards/accuracies": 0.5249999761581421, | |
| "rewards/chosen": 0.04523162543773651, | |
| "rewards/margins": 0.04212886095046997, | |
| "rewards/rejected": 0.003102770773693919, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "learning_rate": 6.41315865106129e-09, | |
| "logits/chosen": -1.9306046962738037, | |
| "logits/rejected": -1.9409637451171875, | |
| "logps/chosen": -32.217220306396484, | |
| "logps/rejected": -35.306907653808594, | |
| "loss": 0.3411, | |
| "rewards/accuracies": 0.4625000059604645, | |
| "rewards/chosen": -0.011275654658675194, | |
| "rewards/margins": 0.005062489304691553, | |
| "rewards/rejected": -0.01633814536035061, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "learning_rate": 2.3150941078050324e-09, | |
| "logits/chosen": -2.0694775581359863, | |
| "logits/rejected": -2.0629212856292725, | |
| "logps/chosen": -33.65714645385742, | |
| "logps/rejected": -29.221160888671875, | |
| "loss": 0.3549, | |
| "rewards/accuracies": 0.4749999940395355, | |
| "rewards/chosen": -0.004660401493310928, | |
| "rewards/margins": -0.009401795454323292, | |
| "rewards/rejected": 0.0047413939610123634, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 0.99, | |
| "learning_rate": 2.575864278703266e-10, | |
| "logits/chosen": -1.9296157360076904, | |
| "logits/rejected": -1.931777000427246, | |
| "logps/chosen": -34.22825241088867, | |
| "logps/rejected": -30.912120819091797, | |
| "loss": 0.2952, | |
| "rewards/accuracies": 0.5375000238418579, | |
| "rewards/chosen": 0.009714365005493164, | |
| "rewards/margins": 0.04525749385356903, | |
| "rewards/rejected": -0.035543132573366165, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "step": 385, | |
| "total_flos": 0.0, | |
| "train_loss": 0.0744757689438857, | |
| "train_runtime": 628.0651, | |
| "train_samples_per_second": 4.902, | |
| "train_steps_per_second": 0.613 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 385, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 100, | |
| "total_flos": 0.0, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |