255 lines
8.1 KiB
JSON
255 lines
8.1 KiB
JSON
{
|
|
"best_metric": null,
|
|
"best_model_checkpoint": null,
|
|
"epoch": 0.9968652037617555,
|
|
"eval_steps": 500,
|
|
"global_step": 159,
|
|
"is_hyper_param_search": false,
|
|
"is_local_process_zero": true,
|
|
"is_world_process_zero": true,
|
|
"log_history": [
|
|
{
|
|
"epoch": 0.01,
|
|
"learning_rate": 3.125e-08,
|
|
"logits/chosen": -2.689429759979248,
|
|
"logits/rejected": -2.605907440185547,
|
|
"logps/chosen": -143.16458129882812,
|
|
"logps/rejected": -148.07847595214844,
|
|
"loss": 0.6931,
|
|
"rewards/accuracies": 0.0,
|
|
"rewards/chosen": 0.0,
|
|
"rewards/margins": 0.0,
|
|
"rewards/rejected": 0.0,
|
|
"step": 1
|
|
},
|
|
{
|
|
"epoch": 0.06,
|
|
"learning_rate": 3.1249999999999997e-07,
|
|
"logits/chosen": -2.69895076751709,
|
|
"logits/rejected": -2.6911985874176025,
|
|
"logps/chosen": -234.76565551757812,
|
|
"logps/rejected": -260.6521911621094,
|
|
"loss": 0.6922,
|
|
"rewards/accuracies": 0.4166666567325592,
|
|
"rewards/chosen": -0.010081589221954346,
|
|
"rewards/margins": -0.0022414023987948895,
|
|
"rewards/rejected": -0.007840188220143318,
|
|
"step": 10
|
|
},
|
|
{
|
|
"epoch": 0.13,
|
|
"learning_rate": 4.990353313429303e-07,
|
|
"logits/chosen": -2.6826164722442627,
|
|
"logits/rejected": -2.676023006439209,
|
|
"logps/chosen": -270.6432189941406,
|
|
"logps/rejected": -290.40869140625,
|
|
"loss": 0.6794,
|
|
"rewards/accuracies": 0.5062500238418579,
|
|
"rewards/chosen": -0.2684400975704193,
|
|
"rewards/margins": 0.04983901605010033,
|
|
"rewards/rejected": -0.31827908754348755,
|
|
"step": 20
|
|
},
|
|
{
|
|
"epoch": 0.19,
|
|
"learning_rate": 4.882681251368548e-07,
|
|
"logits/chosen": -2.6072068214416504,
|
|
"logits/rejected": -2.588265895843506,
|
|
"logps/chosen": -228.98974609375,
|
|
"logps/rejected": -268.8215637207031,
|
|
"loss": 0.6895,
|
|
"rewards/accuracies": 0.5625,
|
|
"rewards/chosen": -0.09740053117275238,
|
|
"rewards/margins": 0.05857750028371811,
|
|
"rewards/rejected": -0.1559780389070511,
|
|
"step": 30
|
|
},
|
|
{
|
|
"epoch": 0.25,
|
|
"learning_rate": 4.6604720940421207e-07,
|
|
"logits/chosen": -2.5079424381256104,
|
|
"logits/rejected": -2.485220432281494,
|
|
"logps/chosen": -233.1949462890625,
|
|
"logps/rejected": -272.9608459472656,
|
|
"loss": 0.6737,
|
|
"rewards/accuracies": 0.5874999761581421,
|
|
"rewards/chosen": -0.1403357833623886,
|
|
"rewards/margins": 0.14330314099788666,
|
|
"rewards/rejected": -0.28363892436027527,
|
|
"step": 40
|
|
},
|
|
{
|
|
"epoch": 0.31,
|
|
"learning_rate": 4.3344075855595097e-07,
|
|
"logits/chosen": -2.430572986602783,
|
|
"logits/rejected": -2.4246697425842285,
|
|
"logps/chosen": -258.1377868652344,
|
|
"logps/rejected": -303.17877197265625,
|
|
"loss": 0.6657,
|
|
"rewards/accuracies": 0.5687500238418579,
|
|
"rewards/chosen": -0.4008946418762207,
|
|
"rewards/margins": 0.11444135755300522,
|
|
"rewards/rejected": -0.5153360366821289,
|
|
"step": 50
|
|
},
|
|
{
|
|
"epoch": 0.38,
|
|
"learning_rate": 3.920161866827889e-07,
|
|
"logits/chosen": -2.3226020336151123,
|
|
"logits/rejected": -2.313552141189575,
|
|
"logps/chosen": -289.2204284667969,
|
|
"logps/rejected": -296.24609375,
|
|
"loss": 0.6682,
|
|
"rewards/accuracies": 0.59375,
|
|
"rewards/chosen": -0.5691416263580322,
|
|
"rewards/margins": 0.10885188728570938,
|
|
"rewards/rejected": -0.6779934167861938,
|
|
"step": 60
|
|
},
|
|
{
|
|
"epoch": 0.44,
|
|
"learning_rate": 3.4376480090239047e-07,
|
|
"logits/chosen": -2.2763919830322266,
|
|
"logits/rejected": -2.2820637226104736,
|
|
"logps/chosen": -253.41159057617188,
|
|
"logps/rejected": -296.48333740234375,
|
|
"loss": 0.667,
|
|
"rewards/accuracies": 0.574999988079071,
|
|
"rewards/chosen": -0.4127040505409241,
|
|
"rewards/margins": 0.14596061408519745,
|
|
"rewards/rejected": -0.5586646199226379,
|
|
"step": 70
|
|
},
|
|
{
|
|
"epoch": 0.5,
|
|
"learning_rate": 2.910060778827554e-07,
|
|
"logits/chosen": -2.2447080612182617,
|
|
"logits/rejected": -2.2723259925842285,
|
|
"logps/chosen": -234.634521484375,
|
|
"logps/rejected": -309.20513916015625,
|
|
"loss": 0.653,
|
|
"rewards/accuracies": 0.581250011920929,
|
|
"rewards/chosen": -0.40885549783706665,
|
|
"rewards/margins": 0.18530671298503876,
|
|
"rewards/rejected": -0.5941622257232666,
|
|
"step": 80
|
|
},
|
|
{
|
|
"epoch": 0.56,
|
|
"learning_rate": 2.3627616503391812e-07,
|
|
"logits/chosen": -2.330038070678711,
|
|
"logits/rejected": -2.3056416511535645,
|
|
"logps/chosen": -243.3378448486328,
|
|
"logps/rejected": -270.39666748046875,
|
|
"loss": 0.6571,
|
|
"rewards/accuracies": 0.5375000238418579,
|
|
"rewards/chosen": -0.4189552664756775,
|
|
"rewards/margins": 0.1503242701292038,
|
|
"rewards/rejected": -0.5692795515060425,
|
|
"step": 90
|
|
},
|
|
{
|
|
"epoch": 0.63,
|
|
"learning_rate": 1.8220596619089573e-07,
|
|
"logits/chosen": -2.235563278198242,
|
|
"logits/rejected": -2.2629618644714355,
|
|
"logps/chosen": -269.2986145019531,
|
|
"logps/rejected": -299.86138916015625,
|
|
"loss": 0.6631,
|
|
"rewards/accuracies": 0.65625,
|
|
"rewards/chosen": -0.46052223443984985,
|
|
"rewards/margins": 0.18169686198234558,
|
|
"rewards/rejected": -0.6422190070152283,
|
|
"step": 100
|
|
},
|
|
{
|
|
"epoch": 0.69,
|
|
"learning_rate": 1.3139467229135998e-07,
|
|
"logits/chosen": -2.3022847175598145,
|
|
"logits/rejected": -2.3016762733459473,
|
|
"logps/chosen": -234.77597045898438,
|
|
"logps/rejected": -316.5335998535156,
|
|
"loss": 0.6379,
|
|
"rewards/accuracies": 0.643750011920929,
|
|
"rewards/chosen": -0.4068765640258789,
|
|
"rewards/margins": 0.2481352835893631,
|
|
"rewards/rejected": -0.6550118923187256,
|
|
"step": 110
|
|
},
|
|
{
|
|
"epoch": 0.75,
|
|
"learning_rate": 8.628481651367875e-08,
|
|
"logits/chosen": -2.2812705039978027,
|
|
"logits/rejected": -2.2640042304992676,
|
|
"logps/chosen": -265.8632507324219,
|
|
"logps/rejected": -339.9316101074219,
|
|
"loss": 0.6563,
|
|
"rewards/accuracies": 0.6187499761581421,
|
|
"rewards/chosen": -0.49150800704956055,
|
|
"rewards/margins": 0.2535227835178375,
|
|
"rewards/rejected": -0.7450307607650757,
|
|
"step": 120
|
|
},
|
|
{
|
|
"epoch": 0.82,
|
|
"learning_rate": 4.904486005914027e-08,
|
|
"logits/chosen": -2.3139283657073975,
|
|
"logits/rejected": -2.327219247817993,
|
|
"logps/chosen": -259.82452392578125,
|
|
"logps/rejected": -339.85772705078125,
|
|
"loss": 0.6411,
|
|
"rewards/accuracies": 0.637499988079071,
|
|
"rewards/chosen": -0.4681639075279236,
|
|
"rewards/margins": 0.2101978361606598,
|
|
"rewards/rejected": -0.6783617734909058,
|
|
"step": 130
|
|
},
|
|
{
|
|
"epoch": 0.88,
|
|
"learning_rate": 2.1464952759020856e-08,
|
|
"logits/chosen": -2.3203117847442627,
|
|
"logits/rejected": -2.3070220947265625,
|
|
"logps/chosen": -270.1856994628906,
|
|
"logps/rejected": -336.08758544921875,
|
|
"loss": 0.6446,
|
|
"rewards/accuracies": 0.65625,
|
|
"rewards/chosen": -0.48976272344589233,
|
|
"rewards/margins": 0.19642502069473267,
|
|
"rewards/rejected": -0.686187744140625,
|
|
"step": 140
|
|
},
|
|
{
|
|
"epoch": 0.94,
|
|
"learning_rate": 4.8708793644441086e-09,
|
|
"logits/chosen": -2.2997617721557617,
|
|
"logits/rejected": -2.2623441219329834,
|
|
"logps/chosen": -261.05316162109375,
|
|
"logps/rejected": -276.8531188964844,
|
|
"loss": 0.6467,
|
|
"rewards/accuracies": 0.5625,
|
|
"rewards/chosen": -0.49157172441482544,
|
|
"rewards/margins": 0.11737655103206635,
|
|
"rewards/rejected": -0.6089483499526978,
|
|
"step": 150
|
|
},
|
|
{
|
|
"epoch": 1.0,
|
|
"step": 159,
|
|
"total_flos": 0.0,
|
|
"train_loss": 0.6627466093819097,
|
|
"train_runtime": 2639.006,
|
|
"train_samples_per_second": 7.722,
|
|
"train_steps_per_second": 0.06
|
|
}
|
|
],
|
|
"logging_steps": 10,
|
|
"max_steps": 159,
|
|
"num_input_tokens_seen": 0,
|
|
"num_train_epochs": 1,
|
|
"save_steps": 100,
|
|
"total_flos": 0.0,
|
|
"train_batch_size": 8,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|