Mistral-7B-v0.3-spin-10k / trainer_state.json
AmberYifan's picture
Model save
8d5a7cd verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9984,
"eval_steps": 200,
"global_step": 312,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0032,
"grad_norm": 309.07493134536526,
"learning_rate": 1.5625e-08,
"logits/generated": -3.1874351501464844,
"logits/real": -2.811344623565674,
"logps/generated": -277.39678955078125,
"logps/real": -164.29153442382812,
"loss": 0.8248,
"rewards/accuracies": 0.0,
"rewards/generated": 0.0,
"rewards/margins": 0.0,
"rewards/real": 0.0,
"step": 1
},
{
"epoch": 0.032,
"grad_norm": 13.846248517181778,
"learning_rate": 1.5624999999999999e-07,
"logits/generated": -2.9546539783477783,
"logits/real": -2.5018956661224365,
"logps/generated": -249.66824340820312,
"logps/real": -122.42621612548828,
"loss": 0.5265,
"rewards/accuracies": 0.8888888955116272,
"rewards/generated": -0.8262240886688232,
"rewards/margins": 1.2902787923812866,
"rewards/real": 0.46405479311943054,
"step": 10
},
{
"epoch": 0.064,
"grad_norm": 1.537832573373159,
"learning_rate": 3.1249999999999997e-07,
"logits/generated": -3.0161662101745605,
"logits/real": -2.225759267807007,
"logps/generated": -312.00897216796875,
"logps/real": -104.78621673583984,
"loss": 0.1131,
"rewards/accuracies": 1.0,
"rewards/generated": -5.391150951385498,
"rewards/margins": 8.365974426269531,
"rewards/real": 2.9748237133026123,
"step": 20
},
{
"epoch": 0.096,
"grad_norm": 1.5948197394392936,
"learning_rate": 4.6874999999999996e-07,
"logits/generated": -2.8325116634368896,
"logits/real": -2.309696674346924,
"logps/generated": -343.28424072265625,
"logps/real": -100.86705780029297,
"loss": 0.1004,
"rewards/accuracies": 1.0,
"rewards/generated": -8.481008529663086,
"rewards/margins": 12.336121559143066,
"rewards/real": 3.855113983154297,
"step": 30
},
{
"epoch": 0.128,
"grad_norm": 1.360990457967047,
"learning_rate": 4.857142857142857e-07,
"logits/generated": -2.8159852027893066,
"logits/real": -2.518409013748169,
"logps/generated": -355.71441650390625,
"logps/real": -102.14837646484375,
"loss": 0.1018,
"rewards/accuracies": 1.0,
"rewards/generated": -10.088699340820312,
"rewards/margins": 14.381967544555664,
"rewards/real": 4.293266773223877,
"step": 40
},
{
"epoch": 0.16,
"grad_norm": 1.1637353343229784,
"learning_rate": 4.6785714285714283e-07,
"logits/generated": -2.810716152191162,
"logits/real": -2.397970199584961,
"logps/generated": -361.4827575683594,
"logps/real": -93.07586669921875,
"loss": 0.0866,
"rewards/accuracies": 1.0,
"rewards/generated": -10.693376541137695,
"rewards/margins": 14.997743606567383,
"rewards/real": 4.304368019104004,
"step": 50
},
{
"epoch": 0.192,
"grad_norm": 1.5949083844151817,
"learning_rate": 4.5e-07,
"logits/generated": -2.5569677352905273,
"logits/real": -2.209700584411621,
"logps/generated": -356.9215393066406,
"logps/real": -90.63230895996094,
"loss": 0.0871,
"rewards/accuracies": 1.0,
"rewards/generated": -11.640242576599121,
"rewards/margins": 16.65045928955078,
"rewards/real": 5.0102152824401855,
"step": 60
},
{
"epoch": 0.224,
"grad_norm": 1.1997502879904265,
"learning_rate": 4.3214285714285713e-07,
"logits/generated": -2.778857946395874,
"logits/real": -2.4418845176696777,
"logps/generated": -363.03106689453125,
"logps/real": -83.89906311035156,
"loss": 0.0889,
"rewards/accuracies": 1.0,
"rewards/generated": -11.029804229736328,
"rewards/margins": 16.131534576416016,
"rewards/real": 5.1017303466796875,
"step": 70
},
{
"epoch": 0.256,
"grad_norm": 1.2541389286949767,
"learning_rate": 4.142857142857143e-07,
"logits/generated": -2.748169422149658,
"logits/real": -2.537468910217285,
"logps/generated": -352.0079040527344,
"logps/real": -90.74658966064453,
"loss": 0.0847,
"rewards/accuracies": 1.0,
"rewards/generated": -10.95399284362793,
"rewards/margins": 16.11164093017578,
"rewards/real": 5.157646179199219,
"step": 80
},
{
"epoch": 0.288,
"grad_norm": 1.3716409313987092,
"learning_rate": 3.9642857142857137e-07,
"logits/generated": -2.7057158946990967,
"logits/real": -2.4590487480163574,
"logps/generated": -369.472412109375,
"logps/real": -93.29542541503906,
"loss": 0.0867,
"rewards/accuracies": 1.0,
"rewards/generated": -11.155221939086914,
"rewards/margins": 17.421768188476562,
"rewards/real": 6.266547203063965,
"step": 90
},
{
"epoch": 0.32,
"grad_norm": 1.2776829690193363,
"learning_rate": 3.785714285714285e-07,
"logits/generated": -2.7385950088500977,
"logits/real": -2.558948040008545,
"logps/generated": -360.800048828125,
"logps/real": -94.03065490722656,
"loss": 0.0812,
"rewards/accuracies": 1.0,
"rewards/generated": -11.058977127075195,
"rewards/margins": 17.12522315979004,
"rewards/real": 6.066246032714844,
"step": 100
},
{
"epoch": 0.352,
"grad_norm": 1.1502347571032556,
"learning_rate": 3.607142857142857e-07,
"logits/generated": -2.7538702487945557,
"logits/real": -2.577524185180664,
"logps/generated": -369.98388671875,
"logps/real": -81.54554748535156,
"loss": 0.0768,
"rewards/accuracies": 1.0,
"rewards/generated": -12.037736892700195,
"rewards/margins": 18.702112197875977,
"rewards/real": 6.664376735687256,
"step": 110
},
{
"epoch": 0.384,
"grad_norm": 1.3343308865576284,
"learning_rate": 3.4285714285714286e-07,
"logits/generated": -2.6860907077789307,
"logits/real": -2.498666763305664,
"logps/generated": -367.09027099609375,
"logps/real": -77.16963195800781,
"loss": 0.0813,
"rewards/accuracies": 1.0,
"rewards/generated": -12.206511497497559,
"rewards/margins": 18.975597381591797,
"rewards/real": 6.7690839767456055,
"step": 120
},
{
"epoch": 0.416,
"grad_norm": 1.3770646126862427,
"learning_rate": 3.25e-07,
"logits/generated": -2.624868869781494,
"logits/real": -2.4215009212493896,
"logps/generated": -371.2015380859375,
"logps/real": -70.61559295654297,
"loss": 0.0775,
"rewards/accuracies": 1.0,
"rewards/generated": -12.158758163452148,
"rewards/margins": 18.808048248291016,
"rewards/real": 6.649291038513184,
"step": 130
},
{
"epoch": 0.448,
"grad_norm": 1.4072742854376892,
"learning_rate": 3.0714285714285716e-07,
"logits/generated": -2.632169246673584,
"logits/real": -2.5124592781066895,
"logps/generated": -378.91949462890625,
"logps/real": -85.79364013671875,
"loss": 0.0783,
"rewards/accuracies": 1.0,
"rewards/generated": -12.807360649108887,
"rewards/margins": 20.574337005615234,
"rewards/real": 7.7669782638549805,
"step": 140
},
{
"epoch": 0.48,
"grad_norm": 1.6731854026570883,
"learning_rate": 2.892857142857143e-07,
"logits/generated": -2.7312309741973877,
"logits/real": -2.513092517852783,
"logps/generated": -376.5494689941406,
"logps/real": -86.8825912475586,
"loss": 0.0719,
"rewards/accuracies": 1.0,
"rewards/generated": -12.506660461425781,
"rewards/margins": 19.993375778198242,
"rewards/real": 7.486716270446777,
"step": 150
},
{
"epoch": 0.512,
"grad_norm": 1.7208597186189485,
"learning_rate": 2.714285714285714e-07,
"logits/generated": -2.6538147926330566,
"logits/real": -2.4443583488464355,
"logps/generated": -379.8941955566406,
"logps/real": -63.8643798828125,
"loss": 0.0664,
"rewards/accuracies": 1.0,
"rewards/generated": -12.761404991149902,
"rewards/margins": 21.195444107055664,
"rewards/real": 8.434039115905762,
"step": 160
},
{
"epoch": 0.544,
"grad_norm": 3.296728367897804,
"learning_rate": 2.5357142857142855e-07,
"logits/generated": -2.679011821746826,
"logits/real": -2.4271175861358643,
"logps/generated": -368.987060546875,
"logps/real": -65.5553207397461,
"loss": 0.0683,
"rewards/accuracies": 1.0,
"rewards/generated": -12.584733963012695,
"rewards/margins": 19.998342514038086,
"rewards/real": 7.413609504699707,
"step": 170
},
{
"epoch": 0.576,
"grad_norm": 1.509303126217559,
"learning_rate": 2.357142857142857e-07,
"logits/generated": -2.706902503967285,
"logits/real": -2.339202404022217,
"logps/generated": -393.6845703125,
"logps/real": -57.10161590576172,
"loss": 0.0613,
"rewards/accuracies": 1.0,
"rewards/generated": -13.54321575164795,
"rewards/margins": 20.973468780517578,
"rewards/real": 7.430250644683838,
"step": 180
},
{
"epoch": 0.608,
"grad_norm": 1.6226015420004847,
"learning_rate": 2.1785714285714284e-07,
"logits/generated": -2.6822266578674316,
"logits/real": -2.390279769897461,
"logps/generated": -391.8895263671875,
"logps/real": -64.2778549194336,
"loss": 0.0593,
"rewards/accuracies": 1.0,
"rewards/generated": -13.559152603149414,
"rewards/margins": 21.35408592224121,
"rewards/real": 7.794934272766113,
"step": 190
},
{
"epoch": 0.64,
"grad_norm": 1.4621480589976257,
"learning_rate": 2e-07,
"logits/generated": -2.64497447013855,
"logits/real": -2.3843231201171875,
"logps/generated": -378.09307861328125,
"logps/real": -53.13884353637695,
"loss": 0.0622,
"rewards/accuracies": 1.0,
"rewards/generated": -12.936541557312012,
"rewards/margins": 21.156665802001953,
"rewards/real": 8.220122337341309,
"step": 200
},
{
"epoch": 0.64,
"eval_logits/generated": -2.709409236907959,
"eval_logits/real": -2.410205602645874,
"eval_logps/generated": -384.0584411621094,
"eval_logps/real": -105.40039825439453,
"eval_loss": 0.10775644332170486,
"eval_rewards/accuracies": 1.0,
"eval_rewards/generated": -13.081828117370605,
"eval_rewards/margins": 16.210613250732422,
"eval_rewards/real": 3.1287851333618164,
"eval_runtime": 31.4401,
"eval_samples_per_second": 6.361,
"eval_steps_per_second": 0.413,
"step": 200
},
{
"epoch": 0.672,
"grad_norm": 1.6581201639125833,
"learning_rate": 1.8214285714285714e-07,
"logits/generated": -2.6259570121765137,
"logits/real": -2.3245978355407715,
"logps/generated": -380.79241943359375,
"logps/real": -50.675270080566406,
"loss": 0.0563,
"rewards/accuracies": 1.0,
"rewards/generated": -13.35838508605957,
"rewards/margins": 21.13888168334961,
"rewards/real": 7.780497074127197,
"step": 210
},
{
"epoch": 0.704,
"grad_norm": 2.0377232989636846,
"learning_rate": 1.6428571428571429e-07,
"logits/generated": -2.6223349571228027,
"logits/real": -2.4116692543029785,
"logps/generated": -388.216064453125,
"logps/real": -75.10206604003906,
"loss": 0.0601,
"rewards/accuracies": 1.0,
"rewards/generated": -13.288862228393555,
"rewards/margins": 22.80645751953125,
"rewards/real": 9.51759147644043,
"step": 220
},
{
"epoch": 0.736,
"grad_norm": 1.4117008274451168,
"learning_rate": 1.4642857142857143e-07,
"logits/generated": -2.6506168842315674,
"logits/real": -2.356983184814453,
"logps/generated": -402.24298095703125,
"logps/real": -52.900665283203125,
"loss": 0.051,
"rewards/accuracies": 1.0,
"rewards/generated": -14.113009452819824,
"rewards/margins": 22.862171173095703,
"rewards/real": 8.749161720275879,
"step": 230
},
{
"epoch": 0.768,
"grad_norm": 1.5408677183120254,
"learning_rate": 1.2857142857142855e-07,
"logits/generated": -2.663072109222412,
"logits/real": -2.295074939727783,
"logps/generated": -392.62335205078125,
"logps/real": -48.768280029296875,
"loss": 0.0565,
"rewards/accuracies": 1.0,
"rewards/generated": -13.465108871459961,
"rewards/margins": 21.69220542907715,
"rewards/real": 8.227096557617188,
"step": 240
},
{
"epoch": 0.8,
"grad_norm": 1.6617128645150359,
"learning_rate": 1.107142857142857e-07,
"logits/generated": -2.659288167953491,
"logits/real": -2.284973621368408,
"logps/generated": -391.0361022949219,
"logps/real": -49.54872131347656,
"loss": 0.0525,
"rewards/accuracies": 1.0,
"rewards/generated": -13.513246536254883,
"rewards/margins": 21.818363189697266,
"rewards/real": 8.30511474609375,
"step": 250
},
{
"epoch": 0.832,
"grad_norm": 1.9545108954634698,
"learning_rate": 9.285714285714286e-08,
"logits/generated": -2.5850462913513184,
"logits/real": -2.2890560626983643,
"logps/generated": -377.42095947265625,
"logps/real": -50.45993423461914,
"loss": 0.0541,
"rewards/accuracies": 1.0,
"rewards/generated": -13.420709609985352,
"rewards/margins": 21.94915008544922,
"rewards/real": 8.528444290161133,
"step": 260
},
{
"epoch": 0.864,
"grad_norm": 1.663160142834365,
"learning_rate": 7.5e-08,
"logits/generated": -2.639012098312378,
"logits/real": -2.241528272628784,
"logps/generated": -385.27972412109375,
"logps/real": -44.987178802490234,
"loss": 0.0519,
"rewards/accuracies": 1.0,
"rewards/generated": -13.790578842163086,
"rewards/margins": 21.756423950195312,
"rewards/real": 7.965843200683594,
"step": 270
},
{
"epoch": 0.896,
"grad_norm": 1.7475888656947995,
"learning_rate": 5.714285714285714e-08,
"logits/generated": -2.634594440460205,
"logits/real": -2.309357166290283,
"logps/generated": -383.92266845703125,
"logps/real": -47.04390335083008,
"loss": 0.0523,
"rewards/accuracies": 1.0,
"rewards/generated": -13.899473190307617,
"rewards/margins": 22.34453582763672,
"rewards/real": 8.445062637329102,
"step": 280
},
{
"epoch": 0.928,
"grad_norm": 1.8843842875992995,
"learning_rate": 3.9285714285714285e-08,
"logits/generated": -2.6036441326141357,
"logits/real": -2.2659499645233154,
"logps/generated": -395.3347473144531,
"logps/real": -55.319923400878906,
"loss": 0.0537,
"rewards/accuracies": 1.0,
"rewards/generated": -14.550788879394531,
"rewards/margins": 24.063692092895508,
"rewards/real": 9.512903213500977,
"step": 290
},
{
"epoch": 0.96,
"grad_norm": 1.7336216562094997,
"learning_rate": 2.142857142857143e-08,
"logits/generated": -2.5489935874938965,
"logits/real": -2.3403241634368896,
"logps/generated": -381.0558166503906,
"logps/real": -60.32090377807617,
"loss": 0.0515,
"rewards/accuracies": 1.0,
"rewards/generated": -13.906036376953125,
"rewards/margins": 23.397653579711914,
"rewards/real": 9.491620063781738,
"step": 300
},
{
"epoch": 0.992,
"grad_norm": 1.7332047882804558,
"learning_rate": 3.571428571428571e-09,
"logits/generated": -2.6001510620117188,
"logits/real": -2.3211655616760254,
"logps/generated": -384.77154541015625,
"logps/real": -53.601829528808594,
"loss": 0.0525,
"rewards/accuracies": 1.0,
"rewards/generated": -13.838152885437012,
"rewards/margins": 23.025869369506836,
"rewards/real": 9.187714576721191,
"step": 310
},
{
"epoch": 0.9984,
"step": 312,
"total_flos": 0.0,
"train_loss": 0.086211400011029,
"train_runtime": 3213.2155,
"train_samples_per_second": 3.111,
"train_steps_per_second": 0.097
}
],
"logging_steps": 10,
"max_steps": 312,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 200,
"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
}