train_copa_1757340229 / trainer_state.json
rbelanec's picture
End of training
0fabd20 verified
{
"best_global_step": 45,
"best_metric": 0.09552183002233505,
"best_model_checkpoint": "saves_stability/p-tuning/llama-3-8b-instruct/train_copa_1757340229/checkpoint-45",
"epoch": 10.0,
"eval_steps": 45,
"global_step": 900,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.05555555555555555,
"grad_norm": 286.81683349609375,
"learning_rate": 2.2222222222222225e-06,
"loss": 3.7346,
"num_input_tokens_seen": 1568,
"step": 5
},
{
"epoch": 0.1111111111111111,
"grad_norm": 11.651448249816895,
"learning_rate": 5e-06,
"loss": 0.8153,
"num_input_tokens_seen": 3104,
"step": 10
},
{
"epoch": 0.16666666666666666,
"grad_norm": 18.36787223815918,
"learning_rate": 7.777777777777777e-06,
"loss": 0.2738,
"num_input_tokens_seen": 4704,
"step": 15
},
{
"epoch": 0.2222222222222222,
"grad_norm": 10.665700912475586,
"learning_rate": 1.0555555555555555e-05,
"loss": 0.1179,
"num_input_tokens_seen": 6304,
"step": 20
},
{
"epoch": 0.2777777777777778,
"grad_norm": 309.3634338378906,
"learning_rate": 1.3333333333333333e-05,
"loss": 0.9347,
"num_input_tokens_seen": 7808,
"step": 25
},
{
"epoch": 0.3333333333333333,
"grad_norm": 5.189738750457764,
"learning_rate": 1.6111111111111115e-05,
"loss": 1.4511,
"num_input_tokens_seen": 9344,
"step": 30
},
{
"epoch": 0.3888888888888889,
"grad_norm": 1.809667706489563,
"learning_rate": 1.888888888888889e-05,
"loss": 0.1261,
"num_input_tokens_seen": 10912,
"step": 35
},
{
"epoch": 0.4444444444444444,
"grad_norm": 20.138389587402344,
"learning_rate": 2.1666666666666667e-05,
"loss": 0.1587,
"num_input_tokens_seen": 12512,
"step": 40
},
{
"epoch": 0.5,
"grad_norm": 10.385217666625977,
"learning_rate": 2.4444444444444445e-05,
"loss": 0.0899,
"num_input_tokens_seen": 14016,
"step": 45
},
{
"epoch": 0.5,
"eval_loss": 0.09552183002233505,
"eval_runtime": 0.9022,
"eval_samples_per_second": 44.334,
"eval_steps_per_second": 11.084,
"num_input_tokens_seen": 14016,
"step": 45
},
{
"epoch": 0.5555555555555556,
"grad_norm": 0.6066714525222778,
"learning_rate": 2.7222222222222223e-05,
"loss": 0.0126,
"num_input_tokens_seen": 15552,
"step": 50
},
{
"epoch": 0.6111111111111112,
"grad_norm": 338.5174560546875,
"learning_rate": 3e-05,
"loss": 0.2108,
"num_input_tokens_seen": 17120,
"step": 55
},
{
"epoch": 0.6666666666666666,
"grad_norm": 20.13188362121582,
"learning_rate": 3.277777777777778e-05,
"loss": 0.5152,
"num_input_tokens_seen": 18720,
"step": 60
},
{
"epoch": 0.7222222222222222,
"grad_norm": 24.089767456054688,
"learning_rate": 3.555555555555556e-05,
"loss": 0.1552,
"num_input_tokens_seen": 20288,
"step": 65
},
{
"epoch": 0.7777777777777778,
"grad_norm": 6.525434970855713,
"learning_rate": 3.8333333333333334e-05,
"loss": 0.0261,
"num_input_tokens_seen": 21856,
"step": 70
},
{
"epoch": 0.8333333333333334,
"grad_norm": 20.758010864257812,
"learning_rate": 4.111111111111111e-05,
"loss": 0.175,
"num_input_tokens_seen": 23456,
"step": 75
},
{
"epoch": 0.8888888888888888,
"grad_norm": 14.998849868774414,
"learning_rate": 4.388888888888889e-05,
"loss": 0.8421,
"num_input_tokens_seen": 24992,
"step": 80
},
{
"epoch": 0.9444444444444444,
"grad_norm": 0.2764253318309784,
"learning_rate": 4.666666666666667e-05,
"loss": 0.1216,
"num_input_tokens_seen": 26528,
"step": 85
},
{
"epoch": 1.0,
"grad_norm": 5.038161277770996,
"learning_rate": 4.9444444444444446e-05,
"loss": 0.0561,
"num_input_tokens_seen": 28096,
"step": 90
},
{
"epoch": 1.0,
"eval_loss": 0.11698226630687714,
"eval_runtime": 0.9055,
"eval_samples_per_second": 44.173,
"eval_steps_per_second": 11.043,
"num_input_tokens_seen": 28096,
"step": 90
},
{
"epoch": 1.0555555555555556,
"grad_norm": 0.3352654278278351,
"learning_rate": 4.9996991493233693e-05,
"loss": 0.0441,
"num_input_tokens_seen": 29632,
"step": 95
},
{
"epoch": 1.1111111111111112,
"grad_norm": 1.5694634914398193,
"learning_rate": 4.99847706754774e-05,
"loss": 0.0354,
"num_input_tokens_seen": 31200,
"step": 100
},
{
"epoch": 1.1666666666666667,
"grad_norm": 0.6956053376197815,
"learning_rate": 4.9963154107272295e-05,
"loss": 0.0172,
"num_input_tokens_seen": 32768,
"step": 105
},
{
"epoch": 1.2222222222222223,
"grad_norm": 0.03370476886630058,
"learning_rate": 4.993214991772563e-05,
"loss": 0.2122,
"num_input_tokens_seen": 34336,
"step": 110
},
{
"epoch": 1.2777777777777777,
"grad_norm": 8.2134428024292,
"learning_rate": 4.989176976624511e-05,
"loss": 0.0418,
"num_input_tokens_seen": 35872,
"step": 115
},
{
"epoch": 1.3333333333333333,
"grad_norm": 0.2531067132949829,
"learning_rate": 4.9842028838154285e-05,
"loss": 0.0308,
"num_input_tokens_seen": 37440,
"step": 120
},
{
"epoch": 1.3888888888888888,
"grad_norm": 0.12402282655239105,
"learning_rate": 4.978294583898196e-05,
"loss": 0.0014,
"num_input_tokens_seen": 38976,
"step": 125
},
{
"epoch": 1.4444444444444444,
"grad_norm": 9.632392883300781,
"learning_rate": 4.971454298742779e-05,
"loss": 0.2373,
"num_input_tokens_seen": 40544,
"step": 130
},
{
"epoch": 1.5,
"grad_norm": 0.13377882540225983,
"learning_rate": 4.963684600700679e-05,
"loss": 0.064,
"num_input_tokens_seen": 42144,
"step": 135
},
{
"epoch": 1.5,
"eval_loss": 0.10020444542169571,
"eval_runtime": 0.9068,
"eval_samples_per_second": 44.11,
"eval_steps_per_second": 11.028,
"num_input_tokens_seen": 42144,
"step": 135
},
{
"epoch": 1.5555555555555556,
"grad_norm": 0.45847994089126587,
"learning_rate": 4.9549884116375714e-05,
"loss": 0.2137,
"num_input_tokens_seen": 43744,
"step": 140
},
{
"epoch": 1.6111111111111112,
"grad_norm": 1.7336419820785522,
"learning_rate": 4.9453690018345144e-05,
"loss": 0.2228,
"num_input_tokens_seen": 45280,
"step": 145
},
{
"epoch": 1.6666666666666665,
"grad_norm": 0.2694900631904602,
"learning_rate": 4.934829988758131e-05,
"loss": 0.2064,
"num_input_tokens_seen": 46880,
"step": 150
},
{
"epoch": 1.7222222222222223,
"grad_norm": 2.060546398162842,
"learning_rate": 4.923375335700223e-05,
"loss": 0.1129,
"num_input_tokens_seen": 48480,
"step": 155
},
{
"epoch": 1.7777777777777777,
"grad_norm": 0.08969313651323318,
"learning_rate": 4.9110093502873476e-05,
"loss": 0.0211,
"num_input_tokens_seen": 50080,
"step": 160
},
{
"epoch": 1.8333333333333335,
"grad_norm": 8.815054893493652,
"learning_rate": 4.897736682860885e-05,
"loss": 0.0805,
"num_input_tokens_seen": 51648,
"step": 165
},
{
"epoch": 1.8888888888888888,
"grad_norm": 1.0820326805114746,
"learning_rate": 4.883562324728241e-05,
"loss": 0.006,
"num_input_tokens_seen": 53216,
"step": 170
},
{
"epoch": 1.9444444444444444,
"grad_norm": 2.855459451675415,
"learning_rate": 4.868491606285823e-05,
"loss": 0.0836,
"num_input_tokens_seen": 54752,
"step": 175
},
{
"epoch": 2.0,
"grad_norm": 3.110377073287964,
"learning_rate": 4.8525301950144894e-05,
"loss": 0.2623,
"num_input_tokens_seen": 56352,
"step": 180
},
{
"epoch": 2.0,
"eval_loss": 0.11590667068958282,
"eval_runtime": 0.9068,
"eval_samples_per_second": 44.111,
"eval_steps_per_second": 11.028,
"num_input_tokens_seen": 56352,
"step": 180
},
{
"epoch": 2.0555555555555554,
"grad_norm": 0.006817061919718981,
"learning_rate": 4.835684093348244e-05,
"loss": 0.0512,
"num_input_tokens_seen": 57888,
"step": 185
},
{
"epoch": 2.111111111111111,
"grad_norm": 0.02554362453520298,
"learning_rate": 4.817959636416969e-05,
"loss": 0.0021,
"num_input_tokens_seen": 59488,
"step": 190
},
{
"epoch": 2.1666666666666665,
"grad_norm": 0.022813625633716583,
"learning_rate": 4.7993634896640394e-05,
"loss": 0.0761,
"num_input_tokens_seen": 61024,
"step": 195
},
{
"epoch": 2.2222222222222223,
"grad_norm": 6.663330078125,
"learning_rate": 4.779902646339722e-05,
"loss": 0.0738,
"num_input_tokens_seen": 62528,
"step": 200
},
{
"epoch": 2.2777777777777777,
"grad_norm": 3.1534013748168945,
"learning_rate": 4.759584424871302e-05,
"loss": 0.0114,
"num_input_tokens_seen": 64064,
"step": 205
},
{
"epoch": 2.3333333333333335,
"grad_norm": 3.0600011348724365,
"learning_rate": 4.7384164661109176e-05,
"loss": 0.1938,
"num_input_tokens_seen": 65696,
"step": 210
},
{
"epoch": 2.388888888888889,
"grad_norm": 1.7588708400726318,
"learning_rate": 4.7164067304621536e-05,
"loss": 0.1694,
"num_input_tokens_seen": 67232,
"step": 215
},
{
"epoch": 2.4444444444444446,
"grad_norm": 6.040250301361084,
"learning_rate": 4.693563494886455e-05,
"loss": 0.1881,
"num_input_tokens_seen": 68832,
"step": 220
},
{
"epoch": 2.5,
"grad_norm": 5.0182271003723145,
"learning_rate": 4.669895349790502e-05,
"loss": 0.0829,
"num_input_tokens_seen": 70432,
"step": 225
},
{
"epoch": 2.5,
"eval_loss": 0.12768635153770447,
"eval_runtime": 0.9111,
"eval_samples_per_second": 43.902,
"eval_steps_per_second": 10.976,
"num_input_tokens_seen": 70432,
"step": 225
},
{
"epoch": 2.5555555555555554,
"grad_norm": 0.31099933385849,
"learning_rate": 4.645411195795709e-05,
"loss": 0.0129,
"num_input_tokens_seen": 72032,
"step": 230
},
{
"epoch": 2.611111111111111,
"grad_norm": 0.03881705179810524,
"learning_rate": 4.620120240391065e-05,
"loss": 0.0251,
"num_input_tokens_seen": 73632,
"step": 235
},
{
"epoch": 2.6666666666666665,
"grad_norm": 0.08282572776079178,
"learning_rate": 4.5940319944705736e-05,
"loss": 0.0013,
"num_input_tokens_seen": 75168,
"step": 240
},
{
"epoch": 2.7222222222222223,
"grad_norm": 7.691892623901367,
"learning_rate": 4.567156268756594e-05,
"loss": 0.0583,
"num_input_tokens_seen": 76768,
"step": 245
},
{
"epoch": 2.7777777777777777,
"grad_norm": 0.023660924285650253,
"learning_rate": 4.539503170110431e-05,
"loss": 0.0017,
"num_input_tokens_seen": 78336,
"step": 250
},
{
"epoch": 2.8333333333333335,
"grad_norm": 0.010544403456151485,
"learning_rate": 4.5110830977315556e-05,
"loss": 0.0358,
"num_input_tokens_seen": 79872,
"step": 255
},
{
"epoch": 2.888888888888889,
"grad_norm": 11.006378173828125,
"learning_rate": 4.4819067392468944e-05,
"loss": 0.0652,
"num_input_tokens_seen": 81440,
"step": 260
},
{
"epoch": 2.9444444444444446,
"grad_norm": 0.12277956306934357,
"learning_rate": 4.4519850666916484e-05,
"loss": 0.1543,
"num_input_tokens_seen": 83008,
"step": 265
},
{
"epoch": 3.0,
"grad_norm": 0.08844027668237686,
"learning_rate": 4.4213293323831585e-05,
"loss": 0.0578,
"num_input_tokens_seen": 84544,
"step": 270
},
{
"epoch": 3.0,
"eval_loss": 0.11280348151922226,
"eval_runtime": 1.3051,
"eval_samples_per_second": 30.649,
"eval_steps_per_second": 7.662,
"num_input_tokens_seen": 84544,
"step": 270
},
{
"epoch": 3.0555555555555554,
"grad_norm": 0.346179336309433,
"learning_rate": 4.38995106468937e-05,
"loss": 0.0042,
"num_input_tokens_seen": 86112,
"step": 275
},
{
"epoch": 3.111111111111111,
"grad_norm": 0.06369267404079437,
"learning_rate": 4.357862063693486e-05,
"loss": 0.0824,
"num_input_tokens_seen": 87680,
"step": 280
},
{
"epoch": 3.1666666666666665,
"grad_norm": 0.07292094081640244,
"learning_rate": 4.325074396756437e-05,
"loss": 0.002,
"num_input_tokens_seen": 89216,
"step": 285
},
{
"epoch": 3.2222222222222223,
"grad_norm": 0.005305001512169838,
"learning_rate": 4.2916003939788403e-05,
"loss": 0.0049,
"num_input_tokens_seen": 90816,
"step": 290
},
{
"epoch": 3.2777777777777777,
"grad_norm": 0.11449064314365387,
"learning_rate": 4.257452643564155e-05,
"loss": 0.091,
"num_input_tokens_seen": 92448,
"step": 295
},
{
"epoch": 3.3333333333333335,
"grad_norm": 0.009809437207877636,
"learning_rate": 4.22264398708477e-05,
"loss": 0.0019,
"num_input_tokens_seen": 93984,
"step": 300
},
{
"epoch": 3.388888888888889,
"grad_norm": 0.01593100279569626,
"learning_rate": 4.1871875146528195e-05,
"loss": 0.0307,
"num_input_tokens_seen": 95584,
"step": 305
},
{
"epoch": 3.4444444444444446,
"grad_norm": 0.017739348113536835,
"learning_rate": 4.1510965599975196e-05,
"loss": 0.005,
"num_input_tokens_seen": 97088,
"step": 310
},
{
"epoch": 3.5,
"grad_norm": 0.005050200968980789,
"learning_rate": 4.114384695450906e-05,
"loss": 0.0002,
"num_input_tokens_seen": 98688,
"step": 315
},
{
"epoch": 3.5,
"eval_loss": 0.11504854261875153,
"eval_runtime": 0.9108,
"eval_samples_per_second": 43.916,
"eval_steps_per_second": 10.979,
"num_input_tokens_seen": 98688,
"step": 315
},
{
"epoch": 3.5555555555555554,
"grad_norm": 0.9385918378829956,
"learning_rate": 4.077065726843828e-05,
"loss": 0.0378,
"num_input_tokens_seen": 100288,
"step": 320
},
{
"epoch": 3.611111111111111,
"grad_norm": 0.006187118124216795,
"learning_rate": 4.039153688314145e-05,
"loss": 0.0027,
"num_input_tokens_seen": 101824,
"step": 325
},
{
"epoch": 3.6666666666666665,
"grad_norm": 0.006210720632225275,
"learning_rate": 4.000662837029062e-05,
"loss": 0.001,
"num_input_tokens_seen": 103392,
"step": 330
},
{
"epoch": 3.7222222222222223,
"grad_norm": 0.032786380499601364,
"learning_rate": 3.961607647823583e-05,
"loss": 0.008,
"num_input_tokens_seen": 104928,
"step": 335
},
{
"epoch": 3.7777777777777777,
"grad_norm": 0.3668730556964874,
"learning_rate": 3.9220028077571295e-05,
"loss": 0.1409,
"num_input_tokens_seen": 106560,
"step": 340
},
{
"epoch": 3.8333333333333335,
"grad_norm": 0.012007039040327072,
"learning_rate": 3.881863210590332e-05,
"loss": 0.0014,
"num_input_tokens_seen": 108096,
"step": 345
},
{
"epoch": 3.888888888888889,
"grad_norm": 0.029211360961198807,
"learning_rate": 3.841203951184095e-05,
"loss": 0.0751,
"num_input_tokens_seen": 109664,
"step": 350
},
{
"epoch": 3.9444444444444446,
"grad_norm": 0.0241960808634758,
"learning_rate": 3.8000403198230387e-05,
"loss": 0.1969,
"num_input_tokens_seen": 111232,
"step": 355
},
{
"epoch": 4.0,
"grad_norm": 0.07501935958862305,
"learning_rate": 3.75838779646545e-05,
"loss": 0.0039,
"num_input_tokens_seen": 112800,
"step": 360
},
{
"epoch": 4.0,
"eval_loss": 0.14795640110969543,
"eval_runtime": 0.9132,
"eval_samples_per_second": 43.804,
"eval_steps_per_second": 10.951,
"num_input_tokens_seen": 112800,
"step": 360
},
{
"epoch": 4.055555555555555,
"grad_norm": 0.07753298431634903,
"learning_rate": 3.7162620449219e-05,
"loss": 0.0621,
"num_input_tokens_seen": 114400,
"step": 365
},
{
"epoch": 4.111111111111111,
"grad_norm": 1.9199938774108887,
"learning_rate": 3.673678906964727e-05,
"loss": 0.0033,
"num_input_tokens_seen": 115936,
"step": 370
},
{
"epoch": 4.166666666666667,
"grad_norm": 0.12986178696155548,
"learning_rate": 3.630654396370594e-05,
"loss": 0.0199,
"num_input_tokens_seen": 117472,
"step": 375
},
{
"epoch": 4.222222222222222,
"grad_norm": 0.07004016637802124,
"learning_rate": 3.5872046928983626e-05,
"loss": 0.0155,
"num_input_tokens_seen": 119040,
"step": 380
},
{
"epoch": 4.277777777777778,
"grad_norm": 0.10150422155857086,
"learning_rate": 3.543346136204545e-05,
"loss": 0.0173,
"num_input_tokens_seen": 120640,
"step": 385
},
{
"epoch": 4.333333333333333,
"grad_norm": 0.4992082417011261,
"learning_rate": 3.499095219698631e-05,
"loss": 0.0066,
"num_input_tokens_seen": 122176,
"step": 390
},
{
"epoch": 4.388888888888889,
"grad_norm": 0.10953623801469803,
"learning_rate": 3.454468584340588e-05,
"loss": 0.0046,
"num_input_tokens_seen": 123744,
"step": 395
},
{
"epoch": 4.444444444444445,
"grad_norm": 0.0021757897920906544,
"learning_rate": 3.409483012382879e-05,
"loss": 0.022,
"num_input_tokens_seen": 125280,
"step": 400
},
{
"epoch": 4.5,
"grad_norm": 0.25139936804771423,
"learning_rate": 3.364155421059342e-05,
"loss": 0.0225,
"num_input_tokens_seen": 126816,
"step": 405
},
{
"epoch": 4.5,
"eval_loss": 0.16026058793067932,
"eval_runtime": 1.0209,
"eval_samples_per_second": 39.182,
"eval_steps_per_second": 9.796,
"num_input_tokens_seen": 126816,
"step": 405
},
{
"epoch": 4.555555555555555,
"grad_norm": 0.01223969180136919,
"learning_rate": 3.318502856223311e-05,
"loss": 0.0002,
"num_input_tokens_seen": 128352,
"step": 410
},
{
"epoch": 4.611111111111111,
"grad_norm": 0.0031944545917212963,
"learning_rate": 3.272542485937369e-05,
"loss": 0.005,
"num_input_tokens_seen": 129920,
"step": 415
},
{
"epoch": 4.666666666666667,
"grad_norm": 0.17521123588085175,
"learning_rate": 3.2262915940171376e-05,
"loss": 0.0003,
"num_input_tokens_seen": 131424,
"step": 420
},
{
"epoch": 4.722222222222222,
"grad_norm": 0.0066048940643668175,
"learning_rate": 3.1797675735315455e-05,
"loss": 0.0002,
"num_input_tokens_seen": 132960,
"step": 425
},
{
"epoch": 4.777777777777778,
"grad_norm": 0.002535363659262657,
"learning_rate": 3.132987920262005e-05,
"loss": 0.0043,
"num_input_tokens_seen": 134528,
"step": 430
},
{
"epoch": 4.833333333333333,
"grad_norm": 0.0035507865250110626,
"learning_rate": 3.085970226122962e-05,
"loss": 0.0003,
"num_input_tokens_seen": 136096,
"step": 435
},
{
"epoch": 4.888888888888889,
"grad_norm": 0.0028070162516087294,
"learning_rate": 3.0387321725463e-05,
"loss": 0.0012,
"num_input_tokens_seen": 137728,
"step": 440
},
{
"epoch": 4.944444444444445,
"grad_norm": 1.8034868240356445,
"learning_rate": 2.9912915238320754e-05,
"loss": 0.003,
"num_input_tokens_seen": 139264,
"step": 445
},
{
"epoch": 5.0,
"grad_norm": 0.005396835505962372,
"learning_rate": 2.9436661204680882e-05,
"loss": 0.0257,
"num_input_tokens_seen": 140800,
"step": 450
},
{
"epoch": 5.0,
"eval_loss": 0.2160557061433792,
"eval_runtime": 0.9094,
"eval_samples_per_second": 43.984,
"eval_steps_per_second": 10.996,
"num_input_tokens_seen": 140800,
"step": 450
},
{
"epoch": 5.055555555555555,
"grad_norm": 0.004269185476005077,
"learning_rate": 2.8958738724208072e-05,
"loss": 0.0003,
"num_input_tokens_seen": 142400,
"step": 455
},
{
"epoch": 5.111111111111111,
"grad_norm": 0.41225236654281616,
"learning_rate": 2.8479327524001636e-05,
"loss": 0.0007,
"num_input_tokens_seen": 143872,
"step": 460
},
{
"epoch": 5.166666666666667,
"grad_norm": 0.0035078164655715227,
"learning_rate": 2.7998607891007495e-05,
"loss": 0.0005,
"num_input_tokens_seen": 145440,
"step": 465
},
{
"epoch": 5.222222222222222,
"grad_norm": 0.002311155665665865,
"learning_rate": 2.7516760604219617e-05,
"loss": 0.0002,
"num_input_tokens_seen": 147040,
"step": 470
},
{
"epoch": 5.277777777777778,
"grad_norm": 0.0020269257947802544,
"learning_rate": 2.7033966866696457e-05,
"loss": 0.0773,
"num_input_tokens_seen": 148576,
"step": 475
},
{
"epoch": 5.333333333333333,
"grad_norm": 0.007861685007810593,
"learning_rate": 2.6550408237417885e-05,
"loss": 0.0053,
"num_input_tokens_seen": 150112,
"step": 480
},
{
"epoch": 5.388888888888889,
"grad_norm": 0.005717500578612089,
"learning_rate": 2.6066266563008267e-05,
"loss": 0.0002,
"num_input_tokens_seen": 151712,
"step": 485
},
{
"epoch": 5.444444444444445,
"grad_norm": 0.013064628466963768,
"learning_rate": 2.5581723909351406e-05,
"loss": 0.0001,
"num_input_tokens_seen": 153280,
"step": 490
},
{
"epoch": 5.5,
"grad_norm": 0.0037898269947618246,
"learning_rate": 2.5096962493123012e-05,
"loss": 0.0001,
"num_input_tokens_seen": 154848,
"step": 495
},
{
"epoch": 5.5,
"eval_loss": 0.2598717212677002,
"eval_runtime": 0.9079,
"eval_samples_per_second": 44.057,
"eval_steps_per_second": 11.014,
"num_input_tokens_seen": 154848,
"step": 495
},
{
"epoch": 5.555555555555555,
"grad_norm": 0.002790295286104083,
"learning_rate": 2.461216461326642e-05,
"loss": 0.0005,
"num_input_tokens_seen": 156352,
"step": 500
},
{
"epoch": 5.611111111111111,
"grad_norm": 0.02554924227297306,
"learning_rate": 2.4127512582437485e-05,
"loss": 0.0071,
"num_input_tokens_seen": 157920,
"step": 505
},
{
"epoch": 5.666666666666667,
"grad_norm": 0.01036996953189373,
"learning_rate": 2.364318865844416e-05,
"loss": 0.0002,
"num_input_tokens_seen": 159488,
"step": 510
},
{
"epoch": 5.722222222222222,
"grad_norm": 0.00804724358022213,
"learning_rate": 2.3159374975706884e-05,
"loss": 0.0001,
"num_input_tokens_seen": 161056,
"step": 515
},
{
"epoch": 5.777777777777778,
"grad_norm": 0.0011961529962718487,
"learning_rate": 2.2676253476765196e-05,
"loss": 0.0876,
"num_input_tokens_seen": 162624,
"step": 520
},
{
"epoch": 5.833333333333333,
"grad_norm": 2.99995493888855,
"learning_rate": 2.2194005843856636e-05,
"loss": 0.008,
"num_input_tokens_seen": 164128,
"step": 525
},
{
"epoch": 5.888888888888889,
"grad_norm": 0.034230273216962814,
"learning_rate": 2.1712813430593436e-05,
"loss": 0.0001,
"num_input_tokens_seen": 165664,
"step": 530
},
{
"epoch": 5.944444444444445,
"grad_norm": 0.005278330761939287,
"learning_rate": 2.1232857193762924e-05,
"loss": 0.0002,
"num_input_tokens_seen": 167200,
"step": 535
},
{
"epoch": 6.0,
"grad_norm": 1.0005654096603394,
"learning_rate": 2.0754317625276983e-05,
"loss": 0.0015,
"num_input_tokens_seen": 168736,
"step": 540
},
{
"epoch": 6.0,
"eval_loss": 0.19720563292503357,
"eval_runtime": 1.2014,
"eval_samples_per_second": 33.295,
"eval_steps_per_second": 8.324,
"num_input_tokens_seen": 168736,
"step": 540
},
{
"epoch": 6.055555555555555,
"grad_norm": 0.0017703189514577389,
"learning_rate": 2.02773746842965e-05,
"loss": 0.0001,
"num_input_tokens_seen": 170304,
"step": 545
},
{
"epoch": 6.111111111111111,
"grad_norm": 0.001825918909162283,
"learning_rate": 1.980220772955602e-05,
"loss": 0.0001,
"num_input_tokens_seen": 171840,
"step": 550
},
{
"epoch": 6.166666666666667,
"grad_norm": 0.0015316482167690992,
"learning_rate": 1.932899545191433e-05,
"loss": 0.0001,
"num_input_tokens_seen": 173408,
"step": 555
},
{
"epoch": 6.222222222222222,
"grad_norm": 0.0012983162887394428,
"learning_rate": 1.8857915807156092e-05,
"loss": 0.0001,
"num_input_tokens_seen": 174976,
"step": 560
},
{
"epoch": 6.277777777777778,
"grad_norm": 0.0012986973160877824,
"learning_rate": 1.838914594906995e-05,
"loss": 0.0002,
"num_input_tokens_seen": 176544,
"step": 565
},
{
"epoch": 6.333333333333333,
"grad_norm": 0.0011983534786850214,
"learning_rate": 1.792286216282824e-05,
"loss": 0.0001,
"num_input_tokens_seen": 178112,
"step": 570
},
{
"epoch": 6.388888888888889,
"grad_norm": 0.0012220325879752636,
"learning_rate": 1.7459239798693364e-05,
"loss": 0.0001,
"num_input_tokens_seen": 179584,
"step": 575
},
{
"epoch": 6.444444444444445,
"grad_norm": 0.0017407550476491451,
"learning_rate": 1.699845320607571e-05,
"loss": 0.0,
"num_input_tokens_seen": 181152,
"step": 580
},
{
"epoch": 6.5,
"grad_norm": 0.0314888134598732,
"learning_rate": 1.6540675667967974e-05,
"loss": 0.0001,
"num_input_tokens_seen": 182688,
"step": 585
},
{
"epoch": 6.5,
"eval_loss": 0.23569929599761963,
"eval_runtime": 0.9121,
"eval_samples_per_second": 43.856,
"eval_steps_per_second": 10.964,
"num_input_tokens_seen": 182688,
"step": 585
},
{
"epoch": 6.555555555555555,
"grad_norm": 0.0012551225954666734,
"learning_rate": 1.60860793357805e-05,
"loss": 0.0001,
"num_input_tokens_seen": 184224,
"step": 590
},
{
"epoch": 6.611111111111111,
"grad_norm": 0.01309239026159048,
"learning_rate": 1.56348351646022e-05,
"loss": 0.0,
"num_input_tokens_seen": 185792,
"step": 595
},
{
"epoch": 6.666666666666667,
"grad_norm": 0.0008750059641897678,
"learning_rate": 1.5187112848911323e-05,
"loss": 0.0001,
"num_input_tokens_seen": 187392,
"step": 600
},
{
"epoch": 6.722222222222222,
"grad_norm": 0.0012049698270857334,
"learning_rate": 1.47430807587603e-05,
"loss": 0.0152,
"num_input_tokens_seen": 188992,
"step": 605
},
{
"epoch": 6.777777777777778,
"grad_norm": 0.0015398699324578047,
"learning_rate": 1.430290587645865e-05,
"loss": 0.0004,
"num_input_tokens_seen": 190560,
"step": 610
},
{
"epoch": 6.833333333333333,
"grad_norm": 0.004634140990674496,
"learning_rate": 1.3866753733777765e-05,
"loss": 0.0002,
"num_input_tokens_seen": 192128,
"step": 615
},
{
"epoch": 6.888888888888889,
"grad_norm": 0.0009558245656080544,
"learning_rate": 1.343478834970121e-05,
"loss": 0.0001,
"num_input_tokens_seen": 193664,
"step": 620
},
{
"epoch": 6.944444444444445,
"grad_norm": 0.004232191946357489,
"learning_rate": 1.3007172168743854e-05,
"loss": 0.0,
"num_input_tokens_seen": 195296,
"step": 625
},
{
"epoch": 7.0,
"grad_norm": 0.0013792107347398996,
"learning_rate": 1.2584065999863102e-05,
"loss": 0.0,
"num_input_tokens_seen": 196896,
"step": 630
},
{
"epoch": 7.0,
"eval_loss": 0.23841562867164612,
"eval_runtime": 0.9078,
"eval_samples_per_second": 44.063,
"eval_steps_per_second": 11.016,
"num_input_tokens_seen": 196896,
"step": 630
},
{
"epoch": 7.055555555555555,
"grad_norm": 0.001491329981945455,
"learning_rate": 1.2165628955985314e-05,
"loss": 0.0001,
"num_input_tokens_seen": 198528,
"step": 635
},
{
"epoch": 7.111111111111111,
"grad_norm": 0.003210023744031787,
"learning_rate": 1.175201839416988e-05,
"loss": 0.0001,
"num_input_tokens_seen": 200096,
"step": 640
},
{
"epoch": 7.166666666666667,
"grad_norm": 0.0011346667306497693,
"learning_rate": 1.1343389856433658e-05,
"loss": 0.0001,
"num_input_tokens_seen": 201696,
"step": 645
},
{
"epoch": 7.222222222222222,
"grad_norm": 0.002059970749542117,
"learning_rate": 1.0939897011258001e-05,
"loss": 0.0001,
"num_input_tokens_seen": 203232,
"step": 650
},
{
"epoch": 7.277777777777778,
"grad_norm": 0.0009237073827534914,
"learning_rate": 1.0541691595800337e-05,
"loss": 0.0,
"num_input_tokens_seen": 204768,
"step": 655
},
{
"epoch": 7.333333333333333,
"grad_norm": 0.10238360613584518,
"learning_rate": 1.0148923358832022e-05,
"loss": 0.0001,
"num_input_tokens_seen": 206368,
"step": 660
},
{
"epoch": 7.388888888888889,
"grad_norm": 0.0007826680084690452,
"learning_rate": 9.761740004423927e-06,
"loss": 0.0,
"num_input_tokens_seen": 207904,
"step": 665
},
{
"epoch": 7.444444444444445,
"grad_norm": 0.0016665104776620865,
"learning_rate": 9.380287136401e-06,
"loss": 0.0,
"num_input_tokens_seen": 209472,
"step": 670
},
{
"epoch": 7.5,
"grad_norm": 0.002118965145200491,
"learning_rate": 9.00470820358663e-06,
"loss": 0.0,
"num_input_tokens_seen": 211072,
"step": 675
},
{
"epoch": 7.5,
"eval_loss": 0.25775742530822754,
"eval_runtime": 0.9105,
"eval_samples_per_second": 43.931,
"eval_steps_per_second": 10.983,
"num_input_tokens_seen": 211072,
"step": 675
},
{
"epoch": 7.555555555555555,
"grad_norm": 0.0010499365162104368,
"learning_rate": 8.635144445857406e-06,
"loss": 0.0,
"num_input_tokens_seen": 212608,
"step": 680
},
{
"epoch": 7.611111111111111,
"grad_norm": 0.001249464345164597,
"learning_rate": 8.271734841028553e-06,
"loss": 0.0001,
"num_input_tokens_seen": 214176,
"step": 685
},
{
"epoch": 7.666666666666667,
"grad_norm": 0.0022609757725149393,
"learning_rate": 7.914616052590071e-06,
"loss": 0.0,
"num_input_tokens_seen": 215744,
"step": 690
},
{
"epoch": 7.722222222222222,
"grad_norm": 0.0012265975819900632,
"learning_rate": 7.563922378313218e-06,
"loss": 0.0,
"num_input_tokens_seen": 217248,
"step": 695
},
{
"epoch": 7.777777777777778,
"grad_norm": 0.0009276677155867219,
"learning_rate": 7.219785699746573e-06,
"loss": 0.0,
"num_input_tokens_seen": 218816,
"step": 700
},
{
"epoch": 7.833333333333333,
"grad_norm": 0.0009464971371926367,
"learning_rate": 6.882335432620779e-06,
"loss": 0.0,
"num_input_tokens_seen": 220288,
"step": 705
},
{
"epoch": 7.888888888888889,
"grad_norm": 0.0009623964433558285,
"learning_rate": 6.55169847818059e-06,
"loss": 0.0001,
"num_input_tokens_seen": 221856,
"step": 710
},
{
"epoch": 7.944444444444445,
"grad_norm": 0.0021013396326452494,
"learning_rate": 6.22799917546252e-06,
"loss": 0.0,
"num_input_tokens_seen": 223424,
"step": 715
},
{
"epoch": 8.0,
"grad_norm": 0.0012038721470162272,
"learning_rate": 5.9113592545359945e-06,
"loss": 0.0,
"num_input_tokens_seen": 225056,
"step": 720
},
{
"epoch": 8.0,
"eval_loss": 0.2483683079481125,
"eval_runtime": 0.9096,
"eval_samples_per_second": 43.974,
"eval_steps_per_second": 10.994,
"num_input_tokens_seen": 225056,
"step": 720
},
{
"epoch": 8.055555555555555,
"grad_norm": 0.003333949949592352,
"learning_rate": 5.601897790725643e-06,
"loss": 0.0001,
"num_input_tokens_seen": 226656,
"step": 725
},
{
"epoch": 8.11111111111111,
"grad_norm": 0.0008439634111709893,
"learning_rate": 5.299731159831953e-06,
"loss": 0.0,
"num_input_tokens_seen": 228160,
"step": 730
},
{
"epoch": 8.166666666666666,
"grad_norm": 0.00669031823053956,
"learning_rate": 5.004972994367102e-06,
"loss": 0.0,
"num_input_tokens_seen": 229728,
"step": 735
},
{
"epoch": 8.222222222222221,
"grad_norm": 0.02617042511701584,
"learning_rate": 4.7177341408224e-06,
"loss": 0.0001,
"num_input_tokens_seen": 231328,
"step": 740
},
{
"epoch": 8.277777777777779,
"grad_norm": 0.0006735489587299526,
"learning_rate": 4.438122617983443e-06,
"loss": 0.0,
"num_input_tokens_seen": 232832,
"step": 745
},
{
"epoch": 8.333333333333334,
"grad_norm": 0.0006344356224872172,
"learning_rate": 4.166243576308712e-06,
"loss": 0.0,
"num_input_tokens_seen": 234432,
"step": 750
},
{
"epoch": 8.38888888888889,
"grad_norm": 0.0006052847602404654,
"learning_rate": 3.9021992583867325e-06,
"loss": 0.0,
"num_input_tokens_seen": 235968,
"step": 755
},
{
"epoch": 8.444444444444445,
"grad_norm": 0.0009797350503504276,
"learning_rate": 3.6460889604868626e-06,
"loss": 0.0,
"num_input_tokens_seen": 237568,
"step": 760
},
{
"epoch": 8.5,
"grad_norm": 0.0129812341183424,
"learning_rate": 3.398008995217988e-06,
"loss": 0.0,
"num_input_tokens_seen": 239168,
"step": 765
},
{
"epoch": 8.5,
"eval_loss": 0.26260504126548767,
"eval_runtime": 0.9128,
"eval_samples_per_second": 43.822,
"eval_steps_per_second": 10.956,
"num_input_tokens_seen": 239168,
"step": 765
},
{
"epoch": 8.555555555555555,
"grad_norm": 0.0011858679354190826,
"learning_rate": 3.158052655309332e-06,
"loss": 0.0,
"num_input_tokens_seen": 240800,
"step": 770
},
{
"epoch": 8.61111111111111,
"grad_norm": 0.0008492617052979767,
"learning_rate": 2.9263101785268254e-06,
"loss": 0.0,
"num_input_tokens_seen": 242368,
"step": 775
},
{
"epoch": 8.666666666666666,
"grad_norm": 0.0008660260355100036,
"learning_rate": 2.7028687137384267e-06,
"loss": 0.0001,
"num_input_tokens_seen": 243872,
"step": 780
},
{
"epoch": 8.722222222222221,
"grad_norm": 0.0008407417917624116,
"learning_rate": 2.487812288140945e-06,
"loss": 0.0,
"num_input_tokens_seen": 245472,
"step": 785
},
{
"epoch": 8.777777777777779,
"grad_norm": 0.001819881028495729,
"learning_rate": 2.281221775660894e-06,
"loss": 0.0,
"num_input_tokens_seen": 247072,
"step": 790
},
{
"epoch": 8.833333333333334,
"grad_norm": 0.0009300917736254632,
"learning_rate": 2.0831748665410765e-06,
"loss": 0.0,
"num_input_tokens_seen": 248640,
"step": 795
},
{
"epoch": 8.88888888888889,
"grad_norm": 0.000822664937004447,
"learning_rate": 1.893746038124497e-06,
"loss": 0.0,
"num_input_tokens_seen": 250208,
"step": 800
},
{
"epoch": 8.944444444444445,
"grad_norm": 0.0009573014685884118,
"learning_rate": 1.713006526846439e-06,
"loss": 0.0001,
"num_input_tokens_seen": 251744,
"step": 805
},
{
"epoch": 9.0,
"grad_norm": 0.0008078826940618455,
"learning_rate": 1.541024301445404e-06,
"loss": 0.0,
"num_input_tokens_seen": 253312,
"step": 810
},
{
"epoch": 9.0,
"eval_loss": 0.25753912329673767,
"eval_runtime": 0.9042,
"eval_samples_per_second": 44.237,
"eval_steps_per_second": 11.059,
"num_input_tokens_seen": 253312,
"step": 810
},
{
"epoch": 9.055555555555555,
"grad_norm": 0.0016976039623841643,
"learning_rate": 1.3778640374027985e-06,
"loss": 0.0001,
"num_input_tokens_seen": 254944,
"step": 815
},
{
"epoch": 9.11111111111111,
"grad_norm": 0.0009876543190330267,
"learning_rate": 1.2235870926211619e-06,
"loss": 0.0,
"num_input_tokens_seen": 256512,
"step": 820
},
{
"epoch": 9.166666666666666,
"grad_norm": 0.0006386281456798315,
"learning_rate": 1.0782514843499653e-06,
"loss": 0.0,
"num_input_tokens_seen": 258016,
"step": 825
},
{
"epoch": 9.222222222222221,
"grad_norm": 0.0006937114521861076,
"learning_rate": 9.419118673676924e-07,
"loss": 0.0,
"num_input_tokens_seen": 259552,
"step": 830
},
{
"epoch": 9.277777777777779,
"grad_norm": 0.0008850015583448112,
"learning_rate": 8.146195134284052e-07,
"loss": 0.0,
"num_input_tokens_seen": 261152,
"step": 835
},
{
"epoch": 9.333333333333334,
"grad_norm": 0.0009804527508094907,
"learning_rate": 6.964222919805391e-07,
"loss": 0.0,
"num_input_tokens_seen": 262720,
"step": 840
},
{
"epoch": 9.38888888888889,
"grad_norm": 0.0011179933790117502,
"learning_rate": 5.87364652165176e-07,
"loss": 0.0,
"num_input_tokens_seen": 264224,
"step": 845
},
{
"epoch": 9.444444444444445,
"grad_norm": 0.0026051115710288286,
"learning_rate": 4.874876061005173e-07,
"loss": 0.0,
"num_input_tokens_seen": 265792,
"step": 850
},
{
"epoch": 9.5,
"grad_norm": 0.002146265935152769,
"learning_rate": 3.9682871345891883e-07,
"loss": 0.0,
"num_input_tokens_seen": 267392,
"step": 855
},
{
"epoch": 9.5,
"eval_loss": 0.2586374878883362,
"eval_runtime": 0.9135,
"eval_samples_per_second": 43.786,
"eval_steps_per_second": 10.946,
"num_input_tokens_seen": 267392,
"step": 855
},
{
"epoch": 9.555555555555555,
"grad_norm": 0.0067813582718372345,
"learning_rate": 3.1542206734221924e-07,
"loss": 0.0,
"num_input_tokens_seen": 268896,
"step": 860
},
{
"epoch": 9.61111111111111,
"grad_norm": 0.0008614276885055006,
"learning_rate": 2.4329828146074095e-07,
"loss": 0.0,
"num_input_tokens_seen": 270432,
"step": 865
},
{
"epoch": 9.666666666666666,
"grad_norm": 0.010032759048044682,
"learning_rate": 1.8048447862070718e-07,
"loss": 0.0,
"num_input_tokens_seen": 271968,
"step": 870
},
{
"epoch": 9.722222222222221,
"grad_norm": 0.0012930516386404634,
"learning_rate": 1.2700428052447033e-07,
"loss": 0.0,
"num_input_tokens_seen": 273504,
"step": 875
},
{
"epoch": 9.777777777777779,
"grad_norm": 0.0014648187207058072,
"learning_rate": 8.28777988873486e-08,
"loss": 0.0,
"num_input_tokens_seen": 275104,
"step": 880
},
{
"epoch": 9.833333333333334,
"grad_norm": 0.001766075729392469,
"learning_rate": 4.8121627874450625e-08,
"loss": 0.0,
"num_input_tokens_seen": 276672,
"step": 885
},
{
"epoch": 9.88888888888889,
"grad_norm": 0.021048838272690773,
"learning_rate": 2.2748837860270267e-08,
"loss": 0.0001,
"num_input_tokens_seen": 278272,
"step": 890
},
{
"epoch": 9.944444444444445,
"grad_norm": 0.004338031634688377,
"learning_rate": 6.768970513457151e-09,
"loss": 0.0,
"num_input_tokens_seen": 279808,
"step": 895
},
{
"epoch": 10.0,
"grad_norm": 0.001522857928648591,
"learning_rate": 1.8803520859811406e-10,
"loss": 0.0001,
"num_input_tokens_seen": 281408,
"step": 900
},
{
"epoch": 10.0,
"eval_loss": 0.25754714012145996,
"eval_runtime": 0.9209,
"eval_samples_per_second": 43.434,
"eval_steps_per_second": 10.858,
"num_input_tokens_seen": 281408,
"step": 900
},
{
"epoch": 10.0,
"num_input_tokens_seen": 281408,
"step": 900,
"total_flos": 1.2671675052589056e+16,
"train_loss": 0.07824999911183517,
"train_runtime": 214.2274,
"train_samples_per_second": 16.805,
"train_steps_per_second": 4.201
}
],
"logging_steps": 5,
"max_steps": 900,
"num_input_tokens_seen": 281408,
"num_train_epochs": 10,
"save_steps": 45,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 1.2671675052589056e+16,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}