ziansu's picture
Training in progress, step 150, checkpoint
6456528 verified
raw
history blame
10.7 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.12310217480508823,
"eval_steps": 50,
"global_step": 150,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.008206811653672548,
"grad_norm": 0.06318386644124985,
"learning_rate": 4.999451708687114e-06,
"logits/chosen": -2.1367907524108887,
"logits/rejected": -2.4948182106018066,
"logps/chosen": -0.291498601436615,
"logps/rejected": -0.3196522295475006,
"loss": 7.5728,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.4372479021549225,
"rewards/margins": 0.04223042353987694,
"rewards/rejected": -0.47947829961776733,
"step": 10
},
{
"epoch": 0.016413623307345096,
"grad_norm": 0.07310314476490021,
"learning_rate": 4.997807075247147e-06,
"logits/chosen": -2.1456007957458496,
"logits/rejected": -2.4455342292785645,
"logps/chosen": -0.26213544607162476,
"logps/rejected": -0.32332050800323486,
"loss": 7.5298,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.3932031989097595,
"rewards/margins": 0.09177760779857635,
"rewards/rejected": -0.4849807620048523,
"step": 20
},
{
"epoch": 0.024620434961017644,
"grad_norm": 0.05936102196574211,
"learning_rate": 4.9950668210706795e-06,
"logits/chosen": -2.0765950679779053,
"logits/rejected": -2.485799789428711,
"logps/chosen": -0.26631081104278564,
"logps/rejected": -0.32647624611854553,
"loss": 7.5208,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.39946624636650085,
"rewards/margins": 0.09024813771247864,
"rewards/rejected": -0.4897143840789795,
"step": 30
},
{
"epoch": 0.03282724661469019,
"grad_norm": 0.08499134331941605,
"learning_rate": 4.9912321481237616e-06,
"logits/chosen": -2.0753884315490723,
"logits/rejected": -2.441580295562744,
"logps/chosen": -0.2749950885772705,
"logps/rejected": -0.30180150270462036,
"loss": 7.4229,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.41249266266822815,
"rewards/margins": 0.04020959883928299,
"rewards/rejected": -0.45270222425460815,
"step": 40
},
{
"epoch": 0.04103405826836274,
"grad_norm": 0.07681389898061752,
"learning_rate": 4.986304738420684e-06,
"logits/chosen": -2.145660877227783,
"logits/rejected": -2.465946912765503,
"logps/chosen": -0.24909739196300507,
"logps/rejected": -0.2796121835708618,
"loss": 7.4811,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.373646080493927,
"rewards/margins": 0.045772187411785126,
"rewards/rejected": -0.41941824555397034,
"step": 50
},
{
"epoch": 0.04103405826836274,
"eval_logits/chosen": -2.012000799179077,
"eval_logits/rejected": -2.5381252765655518,
"eval_logps/chosen": -0.24157460033893585,
"eval_logps/rejected": -0.2957758605480194,
"eval_loss": 0.9317650198936462,
"eval_rewards/accuracies": 0.5252525210380554,
"eval_rewards/chosen": -0.3623619079589844,
"eval_rewards/margins": 0.08130191266536713,
"eval_rewards/rejected": -0.4436637759208679,
"eval_runtime": 26.0809,
"eval_samples_per_second": 30.214,
"eval_steps_per_second": 3.796,
"step": 50
},
{
"epoch": 0.04924086992203529,
"grad_norm": 0.06638535112142563,
"learning_rate": 4.980286753286196e-06,
"logits/chosen": -2.145846128463745,
"logits/rejected": -2.4077115058898926,
"logps/chosen": -0.22265203297138214,
"logps/rejected": -0.30774614214897156,
"loss": 7.4605,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.3339780271053314,
"rewards/margins": 0.1276412308216095,
"rewards/rejected": -0.4616192877292633,
"step": 60
},
{
"epoch": 0.057447681575707836,
"grad_norm": 0.057281140238046646,
"learning_rate": 4.973180832407471e-06,
"logits/chosen": -2.0021350383758545,
"logits/rejected": -2.4299912452697754,
"logps/chosen": -0.23488977551460266,
"logps/rejected": -0.33270469307899475,
"loss": 7.4257,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.3523346781730652,
"rewards/margins": 0.14672236144542694,
"rewards/rejected": -0.4990570545196533,
"step": 70
},
{
"epoch": 0.06565449322938038,
"grad_norm": 0.07725922018289566,
"learning_rate": 4.964990092676263e-06,
"logits/chosen": -2.117995023727417,
"logits/rejected": -2.359265089035034,
"logps/chosen": -0.21598832309246063,
"logps/rejected": -0.300583153963089,
"loss": 7.4384,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.32398244738578796,
"rewards/margins": 0.12689228355884552,
"rewards/rejected": -0.4508747458457947,
"step": 80
},
{
"epoch": 0.07386130488305294,
"grad_norm": 0.0598183274269104,
"learning_rate": 4.9557181268217225e-06,
"logits/chosen": -2.282627582550049,
"logits/rejected": -2.441333532333374,
"logps/chosen": -0.23655852675437927,
"logps/rejected": -0.3246815800666809,
"loss": 7.4584,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.3548378050327301,
"rewards/margins": 0.13218457996845245,
"rewards/rejected": -0.48702239990234375,
"step": 90
},
{
"epoch": 0.08206811653672548,
"grad_norm": 0.058213479816913605,
"learning_rate": 4.9453690018345144e-06,
"logits/chosen": -2.1114468574523926,
"logits/rejected": -2.5035691261291504,
"logps/chosen": -0.23073866963386536,
"logps/rejected": -0.29445192217826843,
"loss": 7.4116,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.34610801935195923,
"rewards/margins": 0.09556989371776581,
"rewards/rejected": -0.44167789816856384,
"step": 100
},
{
"epoch": 0.08206811653672548,
"eval_logits/chosen": -2.0183491706848145,
"eval_logits/rejected": -2.5400593280792236,
"eval_logps/chosen": -0.20393377542495728,
"eval_logps/rejected": -0.2818409502506256,
"eval_loss": 0.9129964113235474,
"eval_rewards/accuracies": 0.5656565427780151,
"eval_rewards/chosen": -0.3059006631374359,
"eval_rewards/margins": 0.11686072498559952,
"eval_rewards/rejected": -0.4227614104747772,
"eval_runtime": 26.0825,
"eval_samples_per_second": 30.212,
"eval_steps_per_second": 3.796,
"step": 100
},
{
"epoch": 0.09027492819039803,
"grad_norm": 0.06249881908297539,
"learning_rate": 4.933947257182901e-06,
"logits/chosen": -2.1324548721313477,
"logits/rejected": -2.434319019317627,
"logps/chosen": -0.22180762887001038,
"logps/rejected": -0.28862181305885315,
"loss": 7.3604,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.332711398601532,
"rewards/margins": 0.10022131353616714,
"rewards/rejected": -0.43293270468711853,
"step": 110
},
{
"epoch": 0.09848173984407058,
"grad_norm": 0.061758093535900116,
"learning_rate": 4.921457902821578e-06,
"logits/chosen": -2.0597169399261475,
"logits/rejected": -2.4386391639709473,
"logps/chosen": -0.22720107436180115,
"logps/rejected": -0.303659051656723,
"loss": 7.3624,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.3408016264438629,
"rewards/margins": 0.1146869882941246,
"rewards/rejected": -0.4554885923862457,
"step": 120
},
{
"epoch": 0.10668855149774313,
"grad_norm": 0.08368540555238724,
"learning_rate": 4.907906416994146e-06,
"logits/chosen": -2.0944437980651855,
"logits/rejected": -2.4157254695892334,
"logps/chosen": -0.19590887427330017,
"logps/rejected": -0.3365771770477295,
"loss": 7.3464,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.29386329650878906,
"rewards/margins": 0.2110024392604828,
"rewards/rejected": -0.5048657655715942,
"step": 130
},
{
"epoch": 0.11489536315141567,
"grad_norm": 0.060954928398132324,
"learning_rate": 4.893298743830168e-06,
"logits/chosen": -2.1551766395568848,
"logits/rejected": -2.5695576667785645,
"logps/chosen": -0.19875812530517578,
"logps/rejected": -0.2967599928379059,
"loss": 7.3179,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.29813718795776367,
"rewards/margins": 0.14700281620025635,
"rewards/rejected": -0.4451400339603424,
"step": 140
},
{
"epoch": 0.12310217480508823,
"grad_norm": 0.05665091797709465,
"learning_rate": 4.8776412907378845e-06,
"logits/chosen": -2.1410365104675293,
"logits/rejected": -2.4798667430877686,
"logps/chosen": -0.19316771626472473,
"logps/rejected": -0.2972142696380615,
"loss": 7.2384,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.2897515296936035,
"rewards/margins": 0.15606984496116638,
"rewards/rejected": -0.4458213746547699,
"step": 150
},
{
"epoch": 0.12310217480508823,
"eval_logits/chosen": -2.0654072761535645,
"eval_logits/rejected": -2.596571207046509,
"eval_logps/chosen": -0.17970335483551025,
"eval_logps/rejected": -0.2767573893070221,
"eval_loss": 0.8982937335968018,
"eval_rewards/accuracies": 0.5858585834503174,
"eval_rewards/chosen": -0.269555002450943,
"eval_rewards/margins": 0.14558106660842896,
"eval_rewards/rejected": -0.41513609886169434,
"eval_runtime": 26.0741,
"eval_samples_per_second": 30.222,
"eval_steps_per_second": 3.797,
"step": 150
}
],
"logging_steps": 10,
"max_steps": 1500,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"save_steps": 50,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 5.5967069864617574e+17,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}