| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.3888888888888888, | |
| "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.06944444444444445, | |
| "grad_norm": 34.922038802469906, | |
| "learning_rate": 5e-07, | |
| "logits/chosen": -2.745856761932373, | |
| "logits/rejected": -2.7519428730010986, | |
| "logps/chosen": -158.59893798828125, | |
| "logps/rejected": -160.2094268798828, | |
| "loss": 0.6939, | |
| "rewards/accuracies": 0.33125001192092896, | |
| "rewards/chosen": 0.000581090513151139, | |
| "rewards/margins": -0.00047404784709215164, | |
| "rewards/rejected": 0.0010551378363743424, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.1388888888888889, | |
| "grad_norm": 36.413747880379944, | |
| "learning_rate": 1e-06, | |
| "logits/chosen": -2.736849546432495, | |
| "logits/rejected": -2.7453856468200684, | |
| "logps/chosen": -174.9836883544922, | |
| "logps/rejected": -171.3789825439453, | |
| "loss": 0.6883, | |
| "rewards/accuracies": 0.5249999761581421, | |
| "rewards/chosen": 0.022891324013471603, | |
| "rewards/margins": 0.0072159310802817345, | |
| "rewards/rejected": 0.015675392001867294, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.20833333333333334, | |
| "grad_norm": 37.39831623643963, | |
| "learning_rate": 9.985471028179154e-07, | |
| "logits/chosen": -2.6100494861602783, | |
| "logits/rejected": -2.613814115524292, | |
| "logps/chosen": -173.74456787109375, | |
| "logps/rejected": -171.9447784423828, | |
| "loss": 0.6733, | |
| "rewards/accuracies": 0.6187499761581421, | |
| "rewards/chosen": 0.21788981556892395, | |
| "rewards/margins": 0.09981251507997513, | |
| "rewards/rejected": 0.11807730048894882, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.2777777777777778, | |
| "grad_norm": 32.21436012894337, | |
| "learning_rate": 9.94196854912548e-07, | |
| "logits/chosen": -2.52358341217041, | |
| "logits/rejected": -2.5229320526123047, | |
| "logps/chosen": -155.1029815673828, | |
| "logps/rejected": -159.9917755126953, | |
| "loss": 0.653, | |
| "rewards/accuracies": 0.6625000238418579, | |
| "rewards/chosen": 0.3672500252723694, | |
| "rewards/margins": 0.20777833461761475, | |
| "rewards/rejected": 0.15947169065475464, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.3472222222222222, | |
| "grad_norm": 32.0223445235898, | |
| "learning_rate": 9.869745381355905e-07, | |
| "logits/chosen": -2.4307353496551514, | |
| "logits/rejected": -2.429652452468872, | |
| "logps/chosen": -159.20545959472656, | |
| "logps/rejected": -158.57797241210938, | |
| "loss": 0.6327, | |
| "rewards/accuracies": 0.6312500238418579, | |
| "rewards/chosen": 0.3310449421405792, | |
| "rewards/margins": 0.22794027626514435, | |
| "rewards/rejected": 0.10310468822717667, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.4166666666666667, | |
| "grad_norm": 36.79676539267814, | |
| "learning_rate": 9.769221256218162e-07, | |
| "logits/chosen": -2.4173130989074707, | |
| "logits/rejected": -2.3855977058410645, | |
| "logps/chosen": -165.49618530273438, | |
| "logps/rejected": -164.41111755371094, | |
| "loss": 0.6264, | |
| "rewards/accuracies": 0.637499988079071, | |
| "rewards/chosen": -0.17511680722236633, | |
| "rewards/margins": 0.2607855200767517, | |
| "rewards/rejected": -0.43590235710144043, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.4861111111111111, | |
| "grad_norm": 32.497757396344305, | |
| "learning_rate": 9.64098037858483e-07, | |
| "logits/chosen": -2.482489824295044, | |
| "logits/rejected": -2.4674153327941895, | |
| "logps/chosen": -162.7029571533203, | |
| "logps/rejected": -164.48619079589844, | |
| "loss": 0.5914, | |
| "rewards/accuracies": 0.6499999761581421, | |
| "rewards/chosen": -0.3191075921058655, | |
| "rewards/margins": 0.362353652715683, | |
| "rewards/rejected": -0.6814612150192261, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.5555555555555556, | |
| "grad_norm": 34.40764043054464, | |
| "learning_rate": 9.485768031694871e-07, | |
| "logits/chosen": -2.518897771835327, | |
| "logits/rejected": -2.5143680572509766, | |
| "logps/chosen": -167.1140899658203, | |
| "logps/rejected": -168.67987060546875, | |
| "loss": 0.58, | |
| "rewards/accuracies": 0.6812499761581421, | |
| "rewards/chosen": -0.037984561175107956, | |
| "rewards/margins": 0.48854589462280273, | |
| "rewards/rejected": -0.5265304446220398, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.625, | |
| "grad_norm": 33.97876407643923, | |
| "learning_rate": 9.304486245873971e-07, | |
| "logits/chosen": -2.542407989501953, | |
| "logits/rejected": -2.510068416595459, | |
| "logps/chosen": -163.48536682128906, | |
| "logps/rejected": -164.54147338867188, | |
| "loss": 0.5815, | |
| "rewards/accuracies": 0.6812499761581421, | |
| "rewards/chosen": -0.062438905239105225, | |
| "rewards/margins": 0.5621680021286011, | |
| "rewards/rejected": -0.6246069073677063, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.6944444444444444, | |
| "grad_norm": 32.726531127211416, | |
| "learning_rate": 9.098188556305262e-07, | |
| "logits/chosen": -2.513942241668701, | |
| "logits/rejected": -2.51139235496521, | |
| "logps/chosen": -165.90362548828125, | |
| "logps/rejected": -170.9777069091797, | |
| "loss": 0.5342, | |
| "rewards/accuracies": 0.7250000238418579, | |
| "rewards/chosen": -0.2061791867017746, | |
| "rewards/margins": 0.6521804928779602, | |
| "rewards/rejected": -0.8583596348762512, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.6944444444444444, | |
| "eval_logits/chosen": -2.4995272159576416, | |
| "eval_logits/rejected": -2.4678475856781006, | |
| "eval_logps/chosen": -159.87913513183594, | |
| "eval_logps/rejected": -170.17739868164062, | |
| "eval_loss": 0.5209046602249146, | |
| "eval_rewards/accuracies": 0.71484375, | |
| "eval_rewards/chosen": -0.20708096027374268, | |
| "eval_rewards/margins": 0.7420069575309753, | |
| "eval_rewards/rejected": -0.9490878582000732, | |
| "eval_runtime": 129.0562, | |
| "eval_samples_per_second": 15.83, | |
| "eval_steps_per_second": 0.248, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.7638888888888888, | |
| "grad_norm": 30.615472770103583, | |
| "learning_rate": 8.868073880316123e-07, | |
| "logits/chosen": -2.500612497329712, | |
| "logits/rejected": -2.520470142364502, | |
| "logps/chosen": -177.12451171875, | |
| "logps/rejected": -183.56619262695312, | |
| "loss": 0.5577, | |
| "rewards/accuracies": 0.737500011920929, | |
| "rewards/chosen": -0.32517191767692566, | |
| "rewards/margins": 0.8004968762397766, | |
| "rewards/rejected": -1.1256687641143799, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.8333333333333334, | |
| "grad_norm": 32.6842144140191, | |
| "learning_rate": 8.615479549763755e-07, | |
| "logits/chosen": -2.5329596996307373, | |
| "logits/rejected": -2.5310966968536377, | |
| "logps/chosen": -183.05056762695312, | |
| "logps/rejected": -193.45396423339844, | |
| "loss": 0.5335, | |
| "rewards/accuracies": 0.7749999761581421, | |
| "rewards/chosen": -0.4021238386631012, | |
| "rewards/margins": 0.8954024314880371, | |
| "rewards/rejected": -1.297526240348816, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.9027777777777778, | |
| "grad_norm": 31.818374577978716, | |
| "learning_rate": 8.341873539012443e-07, | |
| "logits/chosen": -2.518287420272827, | |
| "logits/rejected": -2.5078444480895996, | |
| "logps/chosen": -168.2073974609375, | |
| "logps/rejected": -179.64515686035156, | |
| "loss": 0.5272, | |
| "rewards/accuracies": 0.675000011920929, | |
| "rewards/chosen": -0.6193049550056458, | |
| "rewards/margins": 0.8720922470092773, | |
| "rewards/rejected": -1.4913971424102783, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.9722222222222222, | |
| "grad_norm": 30.74282933667216, | |
| "learning_rate": 8.048845933670271e-07, | |
| "logits/chosen": -2.474501371383667, | |
| "logits/rejected": -2.486248731613159, | |
| "logps/chosen": -170.53952026367188, | |
| "logps/rejected": -183.97750854492188, | |
| "loss": 0.516, | |
| "rewards/accuracies": 0.668749988079071, | |
| "rewards/chosen": -1.0543142557144165, | |
| "rewards/margins": 0.9860897064208984, | |
| "rewards/rejected": -2.0404040813446045, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 1.0416666666666667, | |
| "grad_norm": 20.857882794392346, | |
| "learning_rate": 7.738099689665539e-07, | |
| "logits/chosen": -2.4805214405059814, | |
| "logits/rejected": -2.4716594219207764, | |
| "logps/chosen": -179.7224884033203, | |
| "logps/rejected": -194.9944610595703, | |
| "loss": 0.3668, | |
| "rewards/accuracies": 0.800000011920929, | |
| "rewards/chosen": -0.5534920692443848, | |
| "rewards/margins": 1.5888177156448364, | |
| "rewards/rejected": -2.1423099040985107, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 1.1111111111111112, | |
| "grad_norm": 19.198293152522673, | |
| "learning_rate": 7.41144073636728e-07, | |
| "logits/chosen": -2.4745230674743652, | |
| "logits/rejected": -2.449141025543213, | |
| "logps/chosen": -166.55409240722656, | |
| "logps/rejected": -190.5934295654297, | |
| "loss": 0.2405, | |
| "rewards/accuracies": 0.9312499761581421, | |
| "rewards/chosen": 0.21927456557750702, | |
| "rewards/margins": 2.280102014541626, | |
| "rewards/rejected": -2.0608274936676025, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 1.1805555555555556, | |
| "grad_norm": 22.513577769114043, | |
| "learning_rate": 7.070767481266492e-07, | |
| "logits/chosen": -2.457094192504883, | |
| "logits/rejected": -2.434199571609497, | |
| "logps/chosen": -168.0836944580078, | |
| "logps/rejected": -188.27374267578125, | |
| "loss": 0.2396, | |
| "rewards/accuracies": 0.925000011920929, | |
| "rewards/chosen": -0.004831351339817047, | |
| "rewards/margins": 2.4242899417877197, | |
| "rewards/rejected": -2.429121494293213, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "grad_norm": 18.850259072813216, | |
| "learning_rate": 6.718059777212565e-07, | |
| "logits/chosen": -2.4347877502441406, | |
| "logits/rejected": -2.432413101196289, | |
| "logps/chosen": -154.73831176757812, | |
| "logps/rejected": -174.75621032714844, | |
| "loss": 0.2034, | |
| "rewards/accuracies": 0.9375, | |
| "rewards/chosen": -0.1518968939781189, | |
| "rewards/margins": 2.676344394683838, | |
| "rewards/rejected": -2.8282413482666016, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 1.3194444444444444, | |
| "grad_norm": 19.10543784815799, | |
| "learning_rate": 6.355367416322778e-07, | |
| "logits/chosen": -2.433638572692871, | |
| "logits/rejected": -2.4281392097473145, | |
| "logps/chosen": -173.74961853027344, | |
| "logps/rejected": -198.87208557128906, | |
| "loss": 0.2048, | |
| "rewards/accuracies": 0.8812500238418579, | |
| "rewards/chosen": -0.30015555024147034, | |
| "rewards/margins": 2.652449369430542, | |
| "rewards/rejected": -2.9526050090789795, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 1.3888888888888888, | |
| "grad_norm": 19.991814546360683, | |
| "learning_rate": 5.984798217433531e-07, | |
| "logits/chosen": -2.434572696685791, | |
| "logits/rejected": -2.417861223220825, | |
| "logps/chosen": -158.5247802734375, | |
| "logps/rejected": -190.9485626220703, | |
| "loss": 0.217, | |
| "rewards/accuracies": 0.90625, | |
| "rewards/chosen": -0.37773507833480835, | |
| "rewards/margins": 2.8572323322296143, | |
| "rewards/rejected": -3.2349674701690674, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.3888888888888888, | |
| "eval_logits/chosen": -2.4349896907806396, | |
| "eval_logits/rejected": -2.40444278717041, | |
| "eval_logps/chosen": -169.47793579101562, | |
| "eval_logps/rejected": -184.55125427246094, | |
| "eval_loss": 0.5166749954223633, | |
| "eval_rewards/accuracies": 0.70703125, | |
| "eval_rewards/chosen": -1.1669622659683228, | |
| "eval_rewards/margins": 1.2195117473602295, | |
| "eval_rewards/rejected": -2.386474132537842, | |
| "eval_runtime": 128.5113, | |
| "eval_samples_per_second": 15.897, | |
| "eval_steps_per_second": 0.249, | |
| "step": 100 | |
| } | |
| ], | |
| "logging_steps": 5, | |
| "max_steps": 216, | |
| "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 | |
| } | |