289 lines
9.6 KiB
JSON
289 lines
9.6 KiB
JSON
|
{
|
||
|
|
"best_global_step": null,
|
||
|
|
"best_metric": null,
|
||
|
|
"best_model_checkpoint": null,
|
||
|
|
"epoch": 1.0,
|
||
|
|
"eval_steps": 50,
|
||
|
|
"global_step": 63,
|
||
|
|
"is_hyper_param_search": false,
|
||
|
|
"is_local_process_zero": true,
|
||
|
|
"is_world_process_zero": true,
|
||
|
|
"log_history": [
|
||
|
|
{
|
||
|
|
"entropy": 2.5553497850894926,
|
||
|
|
"epoch": 0.08,
|
||
|
|
"grad_norm": 357.8926696777344,
|
||
|
|
"learning_rate": 2.8571428571428573e-06,
|
||
|
|
"logits/chosen": 0.41187580737484436,
|
||
|
|
"logits/rejected": 0.43087167594886183,
|
||
|
|
"logps/chosen": -157.8165241241455,
|
||
|
|
"logps/rejected": -150.432616519928,
|
||
|
|
"loss": 0.6890408515930175,
|
||
|
|
"mean_token_accuracy": 0.45499045923352244,
|
||
|
|
"num_tokens": 63280.0,
|
||
|
|
"rewards/accuracies": 0.55625,
|
||
|
|
"rewards/chosen": 0.18442693948745728,
|
||
|
|
"rewards/margins": 0.05980904102325439,
|
||
|
|
"rewards/rejected": 0.12461789846420288,
|
||
|
|
"step": 5
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"entropy": 2.585687577724457,
|
||
|
|
"epoch": 0.16,
|
||
|
|
"grad_norm": 550.5374145507812,
|
||
|
|
"learning_rate": 4.984280524733107e-06,
|
||
|
|
"logits/chosen": 0.23282768258015127,
|
||
|
|
"logits/rejected": 0.41657135452577226,
|
||
|
|
"logps/chosen": -160.46632223129274,
|
||
|
|
"logps/rejected": -155.05524663925172,
|
||
|
|
"loss": 1.0951576232910156,
|
||
|
|
"mean_token_accuracy": 0.4666588671505451,
|
||
|
|
"num_tokens": 128751.0,
|
||
|
|
"rewards/accuracies": 0.46875,
|
||
|
|
"rewards/chosen": 0.04196357727050781,
|
||
|
|
"rewards/margins": -0.13510316610336304,
|
||
|
|
"rewards/rejected": 0.17706674337387085,
|
||
|
|
"step": 10
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"entropy": 2.4887936890125273,
|
||
|
|
"epoch": 0.24,
|
||
|
|
"grad_norm": 544.36474609375,
|
||
|
|
"learning_rate": 4.809698831278217e-06,
|
||
|
|
"logits/chosen": 0.19960332682723014,
|
||
|
|
"logits/rejected": 0.25714549020005323,
|
||
|
|
"logps/chosen": -151.403657245636,
|
||
|
|
"logps/rejected": -163.51016159057616,
|
||
|
|
"loss": 1.1309956550598144,
|
||
|
|
"mean_token_accuracy": 0.4729401096701622,
|
||
|
|
"num_tokens": 194142.0,
|
||
|
|
"rewards/accuracies": 0.45625,
|
||
|
|
"rewards/chosen": -0.6780350089073182,
|
||
|
|
"rewards/margins": 0.11501268744468689,
|
||
|
|
"rewards/rejected": -0.793047696352005,
|
||
|
|
"step": 15
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"entropy": 2.543038326501846,
|
||
|
|
"epoch": 0.32,
|
||
|
|
"grad_norm": 545.773193359375,
|
||
|
|
"learning_rate": 4.454578706170075e-06,
|
||
|
|
"logits/chosen": 0.10757563983449289,
|
||
|
|
"logits/rejected": 0.2191312620805499,
|
||
|
|
"logps/chosen": -134.6160714149475,
|
||
|
|
"logps/rejected": -134.7640838623047,
|
||
|
|
"loss": 1.2686697959899902,
|
||
|
|
"mean_token_accuracy": 0.46390472874045374,
|
||
|
|
"num_tokens": 254871.0,
|
||
|
|
"rewards/accuracies": 0.54375,
|
||
|
|
"rewards/chosen": -1.045345139503479,
|
||
|
|
"rewards/margins": 0.4233867168426514,
|
||
|
|
"rewards/rejected": -1.4687318563461305,
|
||
|
|
"step": 20
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"entropy": 2.4839668542146685,
|
||
|
|
"epoch": 0.4,
|
||
|
|
"grad_norm": 609.0315551757812,
|
||
|
|
"learning_rate": 3.946678240449515e-06,
|
||
|
|
"logits/chosen": 0.23401674829751978,
|
||
|
|
"logits/rejected": 0.22746947777638224,
|
||
|
|
"logps/chosen": -144.78962602615357,
|
||
|
|
"logps/rejected": -136.76443700790406,
|
||
|
|
"loss": 1.4144909858703614,
|
||
|
|
"mean_token_accuracy": 0.45985892340540885,
|
||
|
|
"num_tokens": 317752.0,
|
||
|
|
"rewards/accuracies": 0.5625,
|
||
|
|
"rewards/chosen": -0.8229869723320007,
|
||
|
|
"rewards/margins": 0.27719623446464536,
|
||
|
|
"rewards/rejected": -1.1001832067966462,
|
||
|
|
"step": 25
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"entropy": 2.4549559235572813,
|
||
|
|
"epoch": 0.48,
|
||
|
|
"grad_norm": 685.6503295898438,
|
||
|
|
"learning_rate": 3.3256976548879183e-06,
|
||
|
|
"logits/chosen": 0.21252182170691483,
|
||
|
|
"logits/rejected": 0.27063995156819315,
|
||
|
|
"logps/chosen": -148.57344274520875,
|
||
|
|
"logps/rejected": -146.81006059646606,
|
||
|
|
"loss": 1.6434017181396485,
|
||
|
|
"mean_token_accuracy": 0.454254624247551,
|
||
|
|
"num_tokens": 387452.0,
|
||
|
|
"rewards/accuracies": 0.4625,
|
||
|
|
"rewards/chosen": -0.23523921966552735,
|
||
|
|
"rewards/margins": 0.0028411030769348146,
|
||
|
|
"rewards/rejected": -0.23808032274246216,
|
||
|
|
"step": 30
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"entropy": 2.4848766714334487,
|
||
|
|
"epoch": 0.56,
|
||
|
|
"grad_norm": 572.0255737304688,
|
||
|
|
"learning_rate": 2.6401761180929798e-06,
|
||
|
|
"logits/chosen": 0.1604402064572174,
|
||
|
|
"logits/rejected": 0.23192453424270135,
|
||
|
|
"logps/chosen": -178.91495656967163,
|
||
|
|
"logps/rejected": -171.79455051422119,
|
||
|
|
"loss": 1.2930749893188476,
|
||
|
|
"mean_token_accuracy": 0.4774269789457321,
|
||
|
|
"num_tokens": 454873.0,
|
||
|
|
"rewards/accuracies": 0.54375,
|
||
|
|
"rewards/chosen": 0.2875680088996887,
|
||
|
|
"rewards/margins": 0.7235915064811707,
|
||
|
|
"rewards/rejected": -0.43602349758148196,
|
||
|
|
"step": 35
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"entropy": 2.4956657856702806,
|
||
|
|
"epoch": 0.64,
|
||
|
|
"grad_norm": 581.42724609375,
|
||
|
|
"learning_rate": 1.9436976651092143e-06,
|
||
|
|
"logits/chosen": 0.26337065405985355,
|
||
|
|
"logits/rejected": 0.36300255439029444,
|
||
|
|
"logps/chosen": -157.99055194854736,
|
||
|
|
"logps/rejected": -159.0982545375824,
|
||
|
|
"loss": 1.5171399116516113,
|
||
|
|
"mean_token_accuracy": 0.46507507339119913,
|
||
|
|
"num_tokens": 518527.0,
|
||
|
|
"rewards/accuracies": 0.5,
|
||
|
|
"rewards/chosen": -0.061191481351852414,
|
||
|
|
"rewards/margins": 0.1269888162612915,
|
||
|
|
"rewards/rejected": -0.18818029761314392,
|
||
|
|
"step": 40
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"entropy": 2.434966567158699,
|
||
|
|
"epoch": 0.72,
|
||
|
|
"grad_norm": 432.56536865234375,
|
||
|
|
"learning_rate": 1.2907027822369006e-06,
|
||
|
|
"logits/chosen": 0.268132887364752,
|
||
|
|
"logits/rejected": 0.30894473259910554,
|
||
|
|
"logps/chosen": -154.1576229095459,
|
||
|
|
"logps/rejected": -159.85538330078126,
|
||
|
|
"loss": 1.3509400367736817,
|
||
|
|
"mean_token_accuracy": 0.4740308240056038,
|
||
|
|
"num_tokens": 586496.0,
|
||
|
|
"rewards/accuracies": 0.5375,
|
||
|
|
"rewards/chosen": 0.02395761013031006,
|
||
|
|
"rewards/margins": 0.8281095027923584,
|
||
|
|
"rewards/rejected": -0.8041518926620483,
|
||
|
|
"step": 45
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"entropy": 2.4690693795681,
|
||
|
|
"epoch": 0.8,
|
||
|
|
"grad_norm": 504.9461364746094,
|
||
|
|
"learning_rate": 7.322330470336314e-07,
|
||
|
|
"logits/chosen": 0.27772643533489566,
|
||
|
|
"logits/rejected": 0.36052606081661537,
|
||
|
|
"logps/chosen": -164.17142915725708,
|
||
|
|
"logps/rejected": -163.73542156219483,
|
||
|
|
"loss": 1.4861845970153809,
|
||
|
|
"mean_token_accuracy": 0.4653610520064831,
|
||
|
|
"num_tokens": 659345.0,
|
||
|
|
"rewards/accuracies": 0.55625,
|
||
|
|
"rewards/chosen": -0.0009298324584960938,
|
||
|
|
"rewards/margins": 0.31063854694366455,
|
||
|
|
"rewards/rejected": -0.31156837940216064,
|
||
|
|
"step": 50
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.8,
|
||
|
|
"eval_entropy": 2.536505832195282,
|
||
|
|
"eval_logits/chosen": 0.2893901140152919,
|
||
|
|
"eval_logits/rejected": 0.34482142702810936,
|
||
|
|
"eval_logps/chosen": -161.61729994201661,
|
||
|
|
"eval_logps/rejected": -153.7386490020752,
|
||
|
|
"eval_loss": 1.3951599597930908,
|
||
|
|
"eval_mean_token_accuracy": 0.46117462730407716,
|
||
|
|
"eval_num_tokens": 659345.0,
|
||
|
|
"eval_rewards/accuracies": 0.539,
|
||
|
|
"eval_rewards/chosen": -0.16949110031127929,
|
||
|
|
"eval_rewards/margins": 0.413674391746521,
|
||
|
|
"eval_rewards/rejected": -0.5831654920578003,
|
||
|
|
"eval_runtime": 204.2784,
|
||
|
|
"eval_samples_per_second": 4.895,
|
||
|
|
"eval_steps_per_second": 1.224,
|
||
|
|
"step": 50
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"entropy": 2.5048249810934067,
|
||
|
|
"epoch": 0.88,
|
||
|
|
"grad_norm": 538.51953125,
|
||
|
|
"learning_rate": 3.119414452281158e-07,
|
||
|
|
"logits/chosen": 0.31241473497619054,
|
||
|
|
"logits/rejected": 0.4336116565754852,
|
||
|
|
"logps/chosen": -148.8257155418396,
|
||
|
|
"logps/rejected": -143.5335810661316,
|
||
|
|
"loss": 1.5545385360717774,
|
||
|
|
"mean_token_accuracy": 0.47073024958372117,
|
||
|
|
"num_tokens": 731475.0,
|
||
|
|
"rewards/accuracies": 0.53125,
|
||
|
|
"rewards/chosen": -0.08134242296218872,
|
||
|
|
"rewards/margins": 0.31841338872909547,
|
||
|
|
"rewards/rejected": -0.39975581169128416,
|
||
|
|
"step": 55
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"entropy": 2.4619424283504485,
|
||
|
|
"epoch": 0.96,
|
||
|
|
"grad_norm": 439.4546813964844,
|
||
|
|
"learning_rate": 6.268021954544095e-08,
|
||
|
|
"logits/chosen": 0.26771219837052634,
|
||
|
|
"logits/rejected": 0.3957730213546686,
|
||
|
|
"logps/chosen": -157.09008331298827,
|
||
|
|
"logps/rejected": -164.03927259445192,
|
||
|
|
"loss": 1.2014739990234375,
|
||
|
|
"mean_token_accuracy": 0.47436425387859343,
|
||
|
|
"num_tokens": 797757.0,
|
||
|
|
"rewards/accuracies": 0.525,
|
||
|
|
"rewards/chosen": 0.21749011874198915,
|
||
|
|
"rewards/margins": 0.5742806613445282,
|
||
|
|
"rewards/rejected": -0.35679054260253906,
|
||
|
|
"step": 60
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 1.0,
|
||
|
|
"eval_entropy": 2.5392660574913024,
|
||
|
|
"eval_logits/chosen": 0.2875409764336183,
|
||
|
|
"eval_logits/rejected": 0.34302489192485985,
|
||
|
|
"eval_logps/chosen": -161.68841091918947,
|
||
|
|
"eval_logps/rejected": -153.80732011413573,
|
||
|
|
"eval_loss": 1.3833612203598022,
|
||
|
|
"eval_mean_token_accuracy": 0.46109312522411344,
|
||
|
|
"eval_num_tokens": 829784.0,
|
||
|
|
"eval_rewards/accuracies": 0.543,
|
||
|
|
"eval_rewards/chosen": -0.24060248374938964,
|
||
|
|
"eval_rewards/margins": 0.4112340784072876,
|
||
|
|
"eval_rewards/rejected": -0.6518365621566773,
|
||
|
|
"eval_runtime": 204.0071,
|
||
|
|
"eval_samples_per_second": 4.902,
|
||
|
|
"eval_steps_per_second": 1.225,
|
||
|
|
"step": 63
|
||
|
|
}
|
||
|
|
],
|
||
|
|
"logging_steps": 5,
|
||
|
|
"max_steps": 63,
|
||
|
|
"num_input_tokens_seen": 0,
|
||
|
|
"num_train_epochs": 1,
|
||
|
|
"save_steps": 100,
|
||
|
|
"stateful_callbacks": {
|
||
|
|
"TrainerControl": {
|
||
|
|
"args": {
|
||
|
|
"should_epoch_stop": false,
|
||
|
|
"should_evaluate": false,
|
||
|
|
"should_log": false,
|
||
|
|
"should_save": true,
|
||
|
|
"should_training_stop": true
|
||
|
|
},
|
||
|
|
"attributes": {}
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"total_flos": 3267470732328960.0,
|
||
|
|
"train_batch_size": 4,
|
||
|
|
"trial_name": null,
|
||
|
|
"trial_params": null
|
||
|
|
}
|