|
{ |
|
"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.3306902726749321, |
|
"train_runtime": 3256.1932, |
|
"train_samples_per_second": 0.946, |
|
"train_steps_per_second": 0.118 |
|
} |
|
], |
|
"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 |
|
} |
|
|