240 lines
7.8 KiB
JSON
240 lines
7.8 KiB
JSON
{
|
|
"best_metric": null,
|
|
"best_model_checkpoint": null,
|
|
"epoch": 1.971563981042654,
|
|
"eval_steps": 500,
|
|
"global_step": 104,
|
|
"is_hyper_param_search": false,
|
|
"is_local_process_zero": true,
|
|
"is_world_process_zero": true,
|
|
"log_history": [
|
|
{
|
|
"epoch": 0.018957345971563982,
|
|
"grad_norm": 18.54137556493684,
|
|
"learning_rate": 4.545454545454545e-08,
|
|
"logits/chosen": -1.3836915493011475,
|
|
"logits/rejected": -1.3647675514221191,
|
|
"logps/chosen": -369.8086242675781,
|
|
"logps/rejected": -456.2351989746094,
|
|
"loss": 0.6931,
|
|
"rewards/accuracies": 0.0,
|
|
"rewards/chosen": 0.0,
|
|
"rewards/margins": 0.0,
|
|
"rewards/rejected": 0.0,
|
|
"step": 1
|
|
},
|
|
{
|
|
"epoch": 0.1895734597156398,
|
|
"grad_norm": 18.357440019150488,
|
|
"learning_rate": 4.545454545454545e-07,
|
|
"logits/chosen": -1.3435577154159546,
|
|
"logits/rejected": -1.3215419054031372,
|
|
"logps/chosen": -312.6362609863281,
|
|
"logps/rejected": -380.2178039550781,
|
|
"loss": 0.6925,
|
|
"rewards/accuracies": 0.4722222089767456,
|
|
"rewards/chosen": 0.0017008990980684757,
|
|
"rewards/margins": 0.003187247784808278,
|
|
"rewards/rejected": -0.0014863491524010897,
|
|
"step": 10
|
|
},
|
|
{
|
|
"epoch": 0.3791469194312796,
|
|
"grad_norm": 16.59445996406656,
|
|
"learning_rate": 4.885348141000122e-07,
|
|
"logits/chosen": -1.3366063833236694,
|
|
"logits/rejected": -1.3214499950408936,
|
|
"logps/chosen": -345.36834716796875,
|
|
"logps/rejected": -392.05963134765625,
|
|
"loss": 0.6751,
|
|
"rewards/accuracies": 0.699999988079071,
|
|
"rewards/chosen": 0.056390948593616486,
|
|
"rewards/margins": 0.037608057260513306,
|
|
"rewards/rejected": 0.01878289319574833,
|
|
"step": 20
|
|
},
|
|
{
|
|
"epoch": 0.5687203791469194,
|
|
"grad_norm": 17.22094713854238,
|
|
"learning_rate": 4.5025027361734613e-07,
|
|
"logits/chosen": -1.3375577926635742,
|
|
"logits/rejected": -1.3127400875091553,
|
|
"logps/chosen": -305.53546142578125,
|
|
"logps/rejected": -315.6367492675781,
|
|
"loss": 0.6308,
|
|
"rewards/accuracies": 0.699999988079071,
|
|
"rewards/chosen": 0.1503826230764389,
|
|
"rewards/margins": 0.11851520836353302,
|
|
"rewards/rejected": 0.03186742216348648,
|
|
"step": 30
|
|
},
|
|
{
|
|
"epoch": 0.7582938388625592,
|
|
"grad_norm": 14.166324855532773,
|
|
"learning_rate": 3.893311157806091e-07,
|
|
"logits/chosen": -1.3806421756744385,
|
|
"logits/rejected": -1.3631458282470703,
|
|
"logps/chosen": -326.44732666015625,
|
|
"logps/rejected": -320.19818115234375,
|
|
"loss": 0.599,
|
|
"rewards/accuracies": 0.737500011920929,
|
|
"rewards/chosen": 0.34801143407821655,
|
|
"rewards/margins": 0.3300670087337494,
|
|
"rewards/rejected": 0.01794440858066082,
|
|
"step": 40
|
|
},
|
|
{
|
|
"epoch": 0.9478672985781991,
|
|
"grad_norm": 15.139763404001993,
|
|
"learning_rate": 3.126631330646801e-07,
|
|
"logits/chosen": -1.362728476524353,
|
|
"logits/rejected": -1.3629658222198486,
|
|
"logps/chosen": -323.0518493652344,
|
|
"logps/rejected": -405.2288513183594,
|
|
"loss": 0.5532,
|
|
"rewards/accuracies": 0.7250000238418579,
|
|
"rewards/chosen": 0.43860048055648804,
|
|
"rewards/margins": 0.3733087182044983,
|
|
"rewards/rejected": 0.06529171764850616,
|
|
"step": 50
|
|
},
|
|
{
|
|
"epoch": 0.985781990521327,
|
|
"eval_logits/chosen": -1.3316012620925903,
|
|
"eval_logits/rejected": -1.3205280303955078,
|
|
"eval_logps/chosen": -276.7043151855469,
|
|
"eval_logps/rejected": -322.6759948730469,
|
|
"eval_loss": 0.5591163039207458,
|
|
"eval_rewards/accuracies": 0.7604166865348816,
|
|
"eval_rewards/chosen": 0.4980466663837433,
|
|
"eval_rewards/margins": 0.42884430289268494,
|
|
"eval_rewards/rejected": 0.06920242309570312,
|
|
"eval_runtime": 102.7929,
|
|
"eval_samples_per_second": 7.296,
|
|
"eval_steps_per_second": 0.117,
|
|
"step": 52
|
|
},
|
|
{
|
|
"epoch": 1.1374407582938388,
|
|
"grad_norm": 14.364642887348316,
|
|
"learning_rate": 2.2891223348923882e-07,
|
|
"logits/chosen": -1.401728630065918,
|
|
"logits/rejected": -1.378286361694336,
|
|
"logps/chosen": -322.6551208496094,
|
|
"logps/rejected": -355.2884826660156,
|
|
"loss": 0.5136,
|
|
"rewards/accuracies": 0.737500011920929,
|
|
"rewards/chosen": 0.5107188820838928,
|
|
"rewards/margins": 0.5089418292045593,
|
|
"rewards/rejected": 0.0017769678961485624,
|
|
"step": 60
|
|
},
|
|
{
|
|
"epoch": 1.3270142180094786,
|
|
"grad_norm": 16.536036390235253,
|
|
"learning_rate": 1.4754491880085317e-07,
|
|
"logits/chosen": -1.3928285837173462,
|
|
"logits/rejected": -1.377036690711975,
|
|
"logps/chosen": -287.08209228515625,
|
|
"logps/rejected": -335.96124267578125,
|
|
"loss": 0.5006,
|
|
"rewards/accuracies": 0.7749999761581421,
|
|
"rewards/chosen": 0.5614706873893738,
|
|
"rewards/margins": 0.6524538397789001,
|
|
"rewards/rejected": -0.09098317474126816,
|
|
"step": 70
|
|
},
|
|
{
|
|
"epoch": 1.5165876777251186,
|
|
"grad_norm": 13.006389392455016,
|
|
"learning_rate": 7.775827023107834e-08,
|
|
"logits/chosen": -1.3728402853012085,
|
|
"logits/rejected": -1.3621151447296143,
|
|
"logps/chosen": -333.00079345703125,
|
|
"logps/rejected": -385.9339599609375,
|
|
"loss": 0.4643,
|
|
"rewards/accuracies": 0.831250011920929,
|
|
"rewards/chosen": 0.5941373705863953,
|
|
"rewards/margins": 0.8581901788711548,
|
|
"rewards/rejected": -0.26405268907546997,
|
|
"step": 80
|
|
},
|
|
{
|
|
"epoch": 1.7061611374407581,
|
|
"grad_norm": 14.37088192786904,
|
|
"learning_rate": 2.7440387297912122e-08,
|
|
"logits/chosen": -1.39938223361969,
|
|
"logits/rejected": -1.3864850997924805,
|
|
"logps/chosen": -325.26055908203125,
|
|
"logps/rejected": -378.0798645019531,
|
|
"loss": 0.4576,
|
|
"rewards/accuracies": 0.831250011920929,
|
|
"rewards/chosen": 0.4705162048339844,
|
|
"rewards/margins": 0.7767011523246765,
|
|
"rewards/rejected": -0.30618494749069214,
|
|
"step": 90
|
|
},
|
|
{
|
|
"epoch": 1.8957345971563981,
|
|
"grad_norm": 14.30291539625333,
|
|
"learning_rate": 2.27878296044029e-09,
|
|
"logits/chosen": -1.3900349140167236,
|
|
"logits/rejected": -1.3774442672729492,
|
|
"logps/chosen": -304.0644836425781,
|
|
"logps/rejected": -361.84539794921875,
|
|
"loss": 0.4562,
|
|
"rewards/accuracies": 0.8187500238418579,
|
|
"rewards/chosen": 0.5220262408256531,
|
|
"rewards/margins": 0.7719391584396362,
|
|
"rewards/rejected": -0.24991300702095032,
|
|
"step": 100
|
|
},
|
|
{
|
|
"epoch": 1.971563981042654,
|
|
"eval_logits/chosen": -1.3361939191818237,
|
|
"eval_logits/rejected": -1.3248072862625122,
|
|
"eval_logps/chosen": -277.1591491699219,
|
|
"eval_logps/rejected": -326.42169189453125,
|
|
"eval_loss": 0.5286681056022644,
|
|
"eval_rewards/accuracies": 0.7708333134651184,
|
|
"eval_rewards/chosen": 0.47530409693717957,
|
|
"eval_rewards/margins": 0.5933875441551208,
|
|
"eval_rewards/rejected": -0.11808355897665024,
|
|
"eval_runtime": 94.773,
|
|
"eval_samples_per_second": 7.914,
|
|
"eval_steps_per_second": 0.127,
|
|
"step": 104
|
|
},
|
|
{
|
|
"epoch": 1.971563981042654,
|
|
"step": 104,
|
|
"total_flos": 0.0,
|
|
"train_loss": 0.5530139460013463,
|
|
"train_runtime": 7803.0424,
|
|
"train_samples_per_second": 1.73,
|
|
"train_steps_per_second": 0.013
|
|
}
|
|
],
|
|
"logging_steps": 10,
|
|
"max_steps": 104,
|
|
"num_input_tokens_seen": 0,
|
|
"num_train_epochs": 2,
|
|
"save_steps": 500,
|
|
"stateful_callbacks": {
|
|
"TrainerControl": {
|
|
"args": {
|
|
"should_epoch_stop": false,
|
|
"should_evaluate": false,
|
|
"should_log": false,
|
|
"should_save": true,
|
|
"should_training_stop": true
|
|
},
|
|
"attributes": {}
|
|
}
|
|
},
|
|
"total_flos": 0.0,
|
|
"train_batch_size": 4,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|