train_cb_1757340242 / trainer_state.json
rbelanec's picture
End of training
9002e63 verified
{
"best_global_step": 116,
"best_metric": 0.07305306196212769,
"best_model_checkpoint": "saves_stability/lora/llama-3-8b-instruct/train_cb_1757340242/checkpoint-116",
"epoch": 10.0,
"eval_steps": 29,
"global_step": 570,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.08771929824561403,
"grad_norm": 8.72421932220459,
"learning_rate": 3.5087719298245615e-06,
"loss": 1.2201,
"num_input_tokens_seen": 3040,
"step": 5
},
{
"epoch": 0.17543859649122806,
"grad_norm": 8.720124244689941,
"learning_rate": 7.894736842105263e-06,
"loss": 1.0176,
"num_input_tokens_seen": 5984,
"step": 10
},
{
"epoch": 0.2631578947368421,
"grad_norm": 7.666252136230469,
"learning_rate": 1.2280701754385964e-05,
"loss": 0.8606,
"num_input_tokens_seen": 9952,
"step": 15
},
{
"epoch": 0.3508771929824561,
"grad_norm": 8.64617919921875,
"learning_rate": 1.6666666666666667e-05,
"loss": 0.9614,
"num_input_tokens_seen": 13120,
"step": 20
},
{
"epoch": 0.43859649122807015,
"grad_norm": 5.245202541351318,
"learning_rate": 2.105263157894737e-05,
"loss": 0.3126,
"num_input_tokens_seen": 16128,
"step": 25
},
{
"epoch": 0.5087719298245614,
"eval_loss": 0.2108706682920456,
"eval_runtime": 0.7811,
"eval_samples_per_second": 32.007,
"eval_steps_per_second": 8.962,
"num_input_tokens_seen": 18528,
"step": 29
},
{
"epoch": 0.5263157894736842,
"grad_norm": 2.6338868141174316,
"learning_rate": 2.5438596491228074e-05,
"loss": 0.256,
"num_input_tokens_seen": 18944,
"step": 30
},
{
"epoch": 0.6140350877192983,
"grad_norm": 0.5924693942070007,
"learning_rate": 2.9824561403508772e-05,
"loss": 0.2782,
"num_input_tokens_seen": 21984,
"step": 35
},
{
"epoch": 0.7017543859649122,
"grad_norm": 2.3033366203308105,
"learning_rate": 3.421052631578947e-05,
"loss": 0.1206,
"num_input_tokens_seen": 25632,
"step": 40
},
{
"epoch": 0.7894736842105263,
"grad_norm": 21.193347930908203,
"learning_rate": 3.859649122807018e-05,
"loss": 0.0998,
"num_input_tokens_seen": 28416,
"step": 45
},
{
"epoch": 0.8771929824561403,
"grad_norm": 7.374438285827637,
"learning_rate": 4.298245614035088e-05,
"loss": 0.4008,
"num_input_tokens_seen": 31264,
"step": 50
},
{
"epoch": 0.9649122807017544,
"grad_norm": 2.5202343463897705,
"learning_rate": 4.736842105263158e-05,
"loss": 0.1176,
"num_input_tokens_seen": 34528,
"step": 55
},
{
"epoch": 1.0175438596491229,
"eval_loss": 0.14360439777374268,
"eval_runtime": 0.791,
"eval_samples_per_second": 31.604,
"eval_steps_per_second": 8.849,
"num_input_tokens_seen": 35960,
"step": 58
},
{
"epoch": 1.0526315789473684,
"grad_norm": 0.08761471509933472,
"learning_rate": 4.999812487773597e-05,
"loss": 0.2361,
"num_input_tokens_seen": 37464,
"step": 60
},
{
"epoch": 1.1403508771929824,
"grad_norm": 4.727806091308594,
"learning_rate": 4.997703298253406e-05,
"loss": 0.2431,
"num_input_tokens_seen": 40568,
"step": 65
},
{
"epoch": 1.2280701754385965,
"grad_norm": 2.3422040939331055,
"learning_rate": 4.993252512887069e-05,
"loss": 0.114,
"num_input_tokens_seen": 43576,
"step": 70
},
{
"epoch": 1.3157894736842106,
"grad_norm": 2.101975917816162,
"learning_rate": 4.986464304284091e-05,
"loss": 0.1381,
"num_input_tokens_seen": 46392,
"step": 75
},
{
"epoch": 1.4035087719298245,
"grad_norm": 4.683772563934326,
"learning_rate": 4.977345036387331e-05,
"loss": 0.1311,
"num_input_tokens_seen": 49560,
"step": 80
},
{
"epoch": 1.4912280701754386,
"grad_norm": 0.3844655752182007,
"learning_rate": 4.965903258506806e-05,
"loss": 0.2421,
"num_input_tokens_seen": 52184,
"step": 85
},
{
"epoch": 1.526315789473684,
"eval_loss": 0.14571700990200043,
"eval_runtime": 0.7971,
"eval_samples_per_second": 31.362,
"eval_steps_per_second": 8.781,
"num_input_tokens_seen": 53272,
"step": 87
},
{
"epoch": 1.5789473684210527,
"grad_norm": 0.08086138963699341,
"learning_rate": 4.952149697304716e-05,
"loss": 0.0712,
"num_input_tokens_seen": 55128,
"step": 90
},
{
"epoch": 1.6666666666666665,
"grad_norm": 0.06137493997812271,
"learning_rate": 4.9360972467392056e-05,
"loss": 0.183,
"num_input_tokens_seen": 58456,
"step": 95
},
{
"epoch": 1.7543859649122808,
"grad_norm": 0.08413129299879074,
"learning_rate": 4.917760955976277e-05,
"loss": 0.1028,
"num_input_tokens_seen": 61272,
"step": 100
},
{
"epoch": 1.8421052631578947,
"grad_norm": 5.190820693969727,
"learning_rate": 4.897158015281209e-05,
"loss": 0.0112,
"num_input_tokens_seen": 64952,
"step": 105
},
{
"epoch": 1.9298245614035088,
"grad_norm": 0.12773796916007996,
"learning_rate": 4.874307739902689e-05,
"loss": 0.1314,
"num_input_tokens_seen": 67896,
"step": 110
},
{
"epoch": 2.017543859649123,
"grad_norm": 0.45073047280311584,
"learning_rate": 4.849231551964771e-05,
"loss": 0.0044,
"num_input_tokens_seen": 70592,
"step": 115
},
{
"epoch": 2.0350877192982457,
"eval_loss": 0.07305306196212769,
"eval_runtime": 0.7898,
"eval_samples_per_second": 31.654,
"eval_steps_per_second": 8.863,
"num_input_tokens_seen": 71200,
"step": 116
},
{
"epoch": 2.1052631578947367,
"grad_norm": 0.042579129338264465,
"learning_rate": 4.821952960383649e-05,
"loss": 0.0257,
"num_input_tokens_seen": 73728,
"step": 120
},
{
"epoch": 2.192982456140351,
"grad_norm": 0.30034151673316956,
"learning_rate": 4.7924975388280524e-05,
"loss": 0.0603,
"num_input_tokens_seen": 76992,
"step": 125
},
{
"epoch": 2.280701754385965,
"grad_norm": 0.05230257660150528,
"learning_rate": 4.760892901743944e-05,
"loss": 0.0556,
"num_input_tokens_seen": 79712,
"step": 130
},
{
"epoch": 2.3684210526315788,
"grad_norm": 7.066877365112305,
"learning_rate": 4.727168678465988e-05,
"loss": 0.1283,
"num_input_tokens_seen": 82944,
"step": 135
},
{
"epoch": 2.456140350877193,
"grad_norm": 0.0821225568652153,
"learning_rate": 4.6913564854400595e-05,
"loss": 0.0038,
"num_input_tokens_seen": 85568,
"step": 140
},
{
"epoch": 2.543859649122807,
"grad_norm": 5.154212951660156,
"learning_rate": 4.6534898965828405e-05,
"loss": 0.1398,
"num_input_tokens_seen": 89088,
"step": 145
},
{
"epoch": 2.543859649122807,
"eval_loss": 0.10992852598428726,
"eval_runtime": 0.7925,
"eval_samples_per_second": 31.548,
"eval_steps_per_second": 8.833,
"num_input_tokens_seen": 89088,
"step": 145
},
{
"epoch": 2.6315789473684212,
"grad_norm": 2.385377883911133,
"learning_rate": 4.613604411806285e-05,
"loss": 0.2091,
"num_input_tokens_seen": 92128,
"step": 150
},
{
"epoch": 2.719298245614035,
"grad_norm": 0.6019948124885559,
"learning_rate": 4.5717374237364665e-05,
"loss": 0.0202,
"num_input_tokens_seen": 95520,
"step": 155
},
{
"epoch": 2.807017543859649,
"grad_norm": 4.617323398590088,
"learning_rate": 4.5279281826580056e-05,
"loss": 0.0537,
"num_input_tokens_seen": 98848,
"step": 160
},
{
"epoch": 2.8947368421052633,
"grad_norm": 16.427473068237305,
"learning_rate": 4.482217759716946e-05,
"loss": 0.0923,
"num_input_tokens_seen": 101856,
"step": 165
},
{
"epoch": 2.982456140350877,
"grad_norm": 3.187981128692627,
"learning_rate": 4.434649008416565e-05,
"loss": 0.0836,
"num_input_tokens_seen": 105632,
"step": 170
},
{
"epoch": 3.0526315789473686,
"eval_loss": 0.08877496421337128,
"eval_runtime": 0.7925,
"eval_samples_per_second": 31.547,
"eval_steps_per_second": 8.833,
"num_input_tokens_seen": 107504,
"step": 174
},
{
"epoch": 3.0701754385964914,
"grad_norm": 0.08316608518362045,
"learning_rate": 4.385266524442241e-05,
"loss": 0.0088,
"num_input_tokens_seen": 108208,
"step": 175
},
{
"epoch": 3.1578947368421053,
"grad_norm": 7.218474864959717,
"learning_rate": 4.334116603853007e-05,
"loss": 0.0362,
"num_input_tokens_seen": 111280,
"step": 180
},
{
"epoch": 3.245614035087719,
"grad_norm": 0.0030809298623353243,
"learning_rate": 4.2812471996790206e-05,
"loss": 0.0019,
"num_input_tokens_seen": 114512,
"step": 185
},
{
"epoch": 3.3333333333333335,
"grad_norm": 0.6352164149284363,
"learning_rate": 4.226707876965611e-05,
"loss": 0.0015,
"num_input_tokens_seen": 117616,
"step": 190
},
{
"epoch": 3.4210526315789473,
"grad_norm": 0.005556692369282246,
"learning_rate": 4.1705497663060767e-05,
"loss": 0.0247,
"num_input_tokens_seen": 121456,
"step": 195
},
{
"epoch": 3.5087719298245617,
"grad_norm": 0.032038453966379166,
"learning_rate": 4.1128255159067665e-05,
"loss": 0.0307,
"num_input_tokens_seen": 124816,
"step": 200
},
{
"epoch": 3.56140350877193,
"eval_loss": 0.21466340124607086,
"eval_runtime": 0.7941,
"eval_samples_per_second": 31.481,
"eval_steps_per_second": 8.815,
"num_input_tokens_seen": 126384,
"step": 203
},
{
"epoch": 3.5964912280701755,
"grad_norm": 0.01475175004452467,
"learning_rate": 4.053589242229412e-05,
"loss": 0.0003,
"num_input_tokens_seen": 128048,
"step": 205
},
{
"epoch": 3.6842105263157894,
"grad_norm": 0.16376392543315887,
"learning_rate": 3.9928964792569655e-05,
"loss": 0.1438,
"num_input_tokens_seen": 131184,
"step": 210
},
{
"epoch": 3.7719298245614032,
"grad_norm": 0.0069417511112987995,
"learning_rate": 3.930804126430513e-05,
"loss": 0.0109,
"num_input_tokens_seen": 134480,
"step": 215
},
{
"epoch": 3.8596491228070176,
"grad_norm": 0.01790393516421318,
"learning_rate": 3.867370395306068e-05,
"loss": 0.0006,
"num_input_tokens_seen": 137616,
"step": 220
},
{
"epoch": 3.9473684210526314,
"grad_norm": 10.957784652709961,
"learning_rate": 3.8026547549812665e-05,
"loss": 0.1284,
"num_input_tokens_seen": 140496,
"step": 225
},
{
"epoch": 4.035087719298246,
"grad_norm": 0.23430874943733215,
"learning_rate": 3.736717876343106e-05,
"loss": 0.023,
"num_input_tokens_seen": 142832,
"step": 230
},
{
"epoch": 4.0701754385964914,
"eval_loss": 0.14353026449680328,
"eval_runtime": 0.7916,
"eval_samples_per_second": 31.583,
"eval_steps_per_second": 8.843,
"num_input_tokens_seen": 143952,
"step": 232
},
{
"epoch": 4.12280701754386,
"grad_norm": 0.016404252499341965,
"learning_rate": 3.66962157518902e-05,
"loss": 0.0026,
"num_input_tokens_seen": 145936,
"step": 235
},
{
"epoch": 4.2105263157894735,
"grad_norm": 1.3963979482650757,
"learning_rate": 3.601428754274584e-05,
"loss": 0.0034,
"num_input_tokens_seen": 148496,
"step": 240
},
{
"epoch": 4.298245614035087,
"grad_norm": 0.008972808718681335,
"learning_rate": 3.532203344342212e-05,
"loss": 0.0045,
"num_input_tokens_seen": 151920,
"step": 245
},
{
"epoch": 4.385964912280702,
"grad_norm": 0.9162855744361877,
"learning_rate": 3.4620102441861143e-05,
"loss": 0.0008,
"num_input_tokens_seen": 155760,
"step": 250
},
{
"epoch": 4.473684210526316,
"grad_norm": 0.03207138925790787,
"learning_rate": 3.390915259809696e-05,
"loss": 0.0026,
"num_input_tokens_seen": 158608,
"step": 255
},
{
"epoch": 4.56140350877193,
"grad_norm": 0.03569697216153145,
"learning_rate": 3.318985042732461e-05,
"loss": 0.0002,
"num_input_tokens_seen": 161232,
"step": 260
},
{
"epoch": 4.578947368421053,
"eval_loss": 0.2053043097257614,
"eval_runtime": 0.7953,
"eval_samples_per_second": 31.434,
"eval_steps_per_second": 8.802,
"num_input_tokens_seen": 161840,
"step": 261
},
{
"epoch": 4.649122807017544,
"grad_norm": 0.009985814802348614,
"learning_rate": 3.246287027504237e-05,
"loss": 0.0001,
"num_input_tokens_seen": 164336,
"step": 265
},
{
"epoch": 4.7368421052631575,
"grad_norm": 0.6571689248085022,
"learning_rate": 3.172889368485311e-05,
"loss": 0.0004,
"num_input_tokens_seen": 167568,
"step": 270
},
{
"epoch": 4.824561403508772,
"grad_norm": 0.001695934566669166,
"learning_rate": 3.0988608759517475e-05,
"loss": 0.0304,
"num_input_tokens_seen": 170672,
"step": 275
},
{
"epoch": 4.912280701754386,
"grad_norm": 11.054994583129883,
"learning_rate": 3.0242709515857758e-05,
"loss": 0.0193,
"num_input_tokens_seen": 173648,
"step": 280
},
{
"epoch": 5.0,
"grad_norm": 0.0005033599445596337,
"learning_rate": 2.949189523411747e-05,
"loss": 0.0683,
"num_input_tokens_seen": 176200,
"step": 285
},
{
"epoch": 5.087719298245614,
"grad_norm": 0.039827704429626465,
"learning_rate": 2.8736869802386364e-05,
"loss": 0.0001,
"num_input_tokens_seen": 179816,
"step": 290
},
{
"epoch": 5.087719298245614,
"eval_loss": 0.4343818128108978,
"eval_runtime": 0.803,
"eval_samples_per_second": 31.134,
"eval_steps_per_second": 8.718,
"num_input_tokens_seen": 179816,
"step": 290
},
{
"epoch": 5.175438596491228,
"grad_norm": 0.015610896050930023,
"learning_rate": 2.797834105670559e-05,
"loss": 0.0018,
"num_input_tokens_seen": 182920,
"step": 295
},
{
"epoch": 5.2631578947368425,
"grad_norm": 0.0058965967036783695,
"learning_rate": 2.7217020117471793e-05,
"loss": 0.0123,
"num_input_tokens_seen": 185928,
"step": 300
},
{
"epoch": 5.350877192982456,
"grad_norm": 0.029363330453634262,
"learning_rate": 2.6453620722761896e-05,
"loss": 0.0009,
"num_input_tokens_seen": 188392,
"step": 305
},
{
"epoch": 5.43859649122807,
"grad_norm": 0.26967063546180725,
"learning_rate": 2.5688858559204053e-05,
"loss": 0.0005,
"num_input_tokens_seen": 191624,
"step": 310
},
{
"epoch": 5.526315789473684,
"grad_norm": 0.004388366360217333,
"learning_rate": 2.492345059102164e-05,
"loss": 0.0005,
"num_input_tokens_seen": 194376,
"step": 315
},
{
"epoch": 5.5964912280701755,
"eval_loss": 0.17018349468708038,
"eval_runtime": 0.8086,
"eval_samples_per_second": 30.917,
"eval_steps_per_second": 8.657,
"num_input_tokens_seen": 197416,
"step": 319
},
{
"epoch": 5.614035087719298,
"grad_norm": 0.0021780149545520544,
"learning_rate": 2.4158114387879616e-05,
"loss": 0.0002,
"num_input_tokens_seen": 198088,
"step": 320
},
{
"epoch": 5.701754385964913,
"grad_norm": 0.45139217376708984,
"learning_rate": 2.3393567452163252e-05,
"loss": 0.0004,
"num_input_tokens_seen": 200680,
"step": 325
},
{
"epoch": 5.7894736842105265,
"grad_norm": 0.0945330411195755,
"learning_rate": 2.2630526546319914e-05,
"loss": 0.0002,
"num_input_tokens_seen": 203784,
"step": 330
},
{
"epoch": 5.87719298245614,
"grad_norm": 0.0018616619054228067,
"learning_rate": 2.186970702089457e-05,
"loss": 0.0001,
"num_input_tokens_seen": 207112,
"step": 335
},
{
"epoch": 5.964912280701754,
"grad_norm": 0.013572554104030132,
"learning_rate": 2.111182214388893e-05,
"loss": 0.0001,
"num_input_tokens_seen": 209800,
"step": 340
},
{
"epoch": 6.052631578947368,
"grad_norm": 0.011023592203855515,
"learning_rate": 2.0357582432072957e-05,
"loss": 0.0001,
"num_input_tokens_seen": 212896,
"step": 345
},
{
"epoch": 6.105263157894737,
"eval_loss": 0.20455622673034668,
"eval_runtime": 0.7874,
"eval_samples_per_second": 31.749,
"eval_steps_per_second": 8.89,
"num_input_tokens_seen": 214432,
"step": 348
},
{
"epoch": 6.140350877192983,
"grad_norm": 0.0013332996750250459,
"learning_rate": 1.9607694984875754e-05,
"loss": 0.0001,
"num_input_tokens_seen": 215456,
"step": 350
},
{
"epoch": 6.228070175438597,
"grad_norm": 0.0019229918252676725,
"learning_rate": 1.8862862821480025e-05,
"loss": 0.0002,
"num_input_tokens_seen": 218944,
"step": 355
},
{
"epoch": 6.315789473684211,
"grad_norm": 0.001408565673045814,
"learning_rate": 1.8123784221741964e-05,
"loss": 0.0001,
"num_input_tokens_seen": 221824,
"step": 360
},
{
"epoch": 6.4035087719298245,
"grad_norm": 0.08359543979167938,
"learning_rate": 1.73911520715541e-05,
"loss": 0.0003,
"num_input_tokens_seen": 225568,
"step": 365
},
{
"epoch": 6.491228070175438,
"grad_norm": 0.0011508375173434615,
"learning_rate": 1.666565321326512e-05,
"loss": 0.0001,
"num_input_tokens_seen": 228960,
"step": 370
},
{
"epoch": 6.578947368421053,
"grad_norm": 0.0016500469064339995,
"learning_rate": 1.5947967801765345e-05,
"loss": 0.0001,
"num_input_tokens_seen": 231936,
"step": 375
},
{
"epoch": 6.614035087719298,
"eval_loss": 0.22089533507823944,
"eval_runtime": 0.7913,
"eval_samples_per_second": 31.594,
"eval_steps_per_second": 8.846,
"num_input_tokens_seen": 233280,
"step": 377
},
{
"epoch": 6.666666666666667,
"grad_norm": 0.001594485598616302,
"learning_rate": 1.5238768666841907e-05,
"loss": 0.0001,
"num_input_tokens_seen": 234944,
"step": 380
},
{
"epoch": 6.754385964912281,
"grad_norm": 0.0012622015783563256,
"learning_rate": 1.4538720682400969e-05,
"loss": 0.0,
"num_input_tokens_seen": 238080,
"step": 385
},
{
"epoch": 6.842105263157895,
"grad_norm": 0.015878915786743164,
"learning_rate": 1.3848480143148839e-05,
"loss": 0.0001,
"num_input_tokens_seen": 241120,
"step": 390
},
{
"epoch": 6.9298245614035086,
"grad_norm": 0.0009251445881091058,
"learning_rate": 1.3168694149315796e-05,
"loss": 0.0001,
"num_input_tokens_seen": 244800,
"step": 395
},
{
"epoch": 7.017543859649122,
"grad_norm": 0.0010710373753681779,
"learning_rate": 1.2500000000000006e-05,
"loss": 0.0,
"num_input_tokens_seen": 247408,
"step": 400
},
{
"epoch": 7.105263157894737,
"grad_norm": 0.01872781477868557,
"learning_rate": 1.1843024595699805e-05,
"loss": 0.0001,
"num_input_tokens_seen": 250512,
"step": 405
},
{
"epoch": 7.12280701754386,
"eval_loss": 0.24220417439937592,
"eval_runtime": 0.7975,
"eval_samples_per_second": 31.346,
"eval_steps_per_second": 8.777,
"num_input_tokens_seen": 251120,
"step": 406
},
{
"epoch": 7.192982456140351,
"grad_norm": 0.009808254428207874,
"learning_rate": 1.1198383850594758e-05,
"loss": 0.0001,
"num_input_tokens_seen": 254352,
"step": 410
},
{
"epoch": 7.280701754385965,
"grad_norm": 0.0015773404156789184,
"learning_rate": 1.0566682115126344e-05,
"loss": 0.0,
"num_input_tokens_seen": 257264,
"step": 415
},
{
"epoch": 7.368421052631579,
"grad_norm": 0.0013568034628406167,
"learning_rate": 9.948511609419675e-06,
"loss": 0.0001,
"num_input_tokens_seen": 260048,
"step": 420
},
{
"epoch": 7.456140350877193,
"grad_norm": 0.0014519336400553584,
"learning_rate": 9.344451868077353e-06,
"loss": 0.0001,
"num_input_tokens_seen": 263568,
"step": 425
},
{
"epoch": 7.543859649122807,
"grad_norm": 0.0013535337056964636,
"learning_rate": 8.755069196866014e-06,
"loss": 0.0,
"num_input_tokens_seen": 266640,
"step": 430
},
{
"epoch": 7.631578947368421,
"grad_norm": 0.018807148560881615,
"learning_rate": 8.180916141804906e-06,
"loss": 0.0001,
"num_input_tokens_seen": 270128,
"step": 435
},
{
"epoch": 7.631578947368421,
"eval_loss": 0.23934833705425262,
"eval_runtime": 0.8064,
"eval_samples_per_second": 31.003,
"eval_steps_per_second": 8.681,
"num_input_tokens_seen": 270128,
"step": 435
},
{
"epoch": 7.719298245614035,
"grad_norm": 0.0012790975160896778,
"learning_rate": 7.622530971154199e-06,
"loss": 0.0001,
"num_input_tokens_seen": 273136,
"step": 440
},
{
"epoch": 7.807017543859649,
"grad_norm": 0.0015526728238910437,
"learning_rate": 7.080437170788723e-06,
"loss": 0.0002,
"num_input_tokens_seen": 276464,
"step": 445
},
{
"epoch": 7.894736842105263,
"grad_norm": 0.006720449309796095,
"learning_rate": 6.555142953430158e-06,
"loss": 0.0001,
"num_input_tokens_seen": 278960,
"step": 450
},
{
"epoch": 7.982456140350877,
"grad_norm": 0.0009156151791103184,
"learning_rate": 6.0471407821978135e-06,
"loss": 0.0,
"num_input_tokens_seen": 282512,
"step": 455
},
{
"epoch": 8.070175438596491,
"grad_norm": 0.03618071228265762,
"learning_rate": 5.556906908924655e-06,
"loss": 0.0001,
"num_input_tokens_seen": 285496,
"step": 460
},
{
"epoch": 8.140350877192983,
"eval_loss": 0.24381086230278015,
"eval_runtime": 0.7882,
"eval_samples_per_second": 31.72,
"eval_steps_per_second": 8.882,
"num_input_tokens_seen": 288216,
"step": 464
},
{
"epoch": 8.157894736842104,
"grad_norm": 0.005941447801887989,
"learning_rate": 5.084900927671393e-06,
"loss": 0.0,
"num_input_tokens_seen": 289080,
"step": 465
},
{
"epoch": 8.24561403508772,
"grad_norm": 0.0010738734854385257,
"learning_rate": 4.631565343857239e-06,
"loss": 0.0,
"num_input_tokens_seen": 292120,
"step": 470
},
{
"epoch": 8.333333333333334,
"grad_norm": 0.040972329676151276,
"learning_rate": 4.19732515941125e-06,
"loss": 0.0001,
"num_input_tokens_seen": 295384,
"step": 475
},
{
"epoch": 8.421052631578947,
"grad_norm": 0.001142106018960476,
"learning_rate": 3.7825874743331907e-06,
"loss": 0.0001,
"num_input_tokens_seen": 298712,
"step": 480
},
{
"epoch": 8.508771929824562,
"grad_norm": 0.0014838232891634107,
"learning_rate": 3.3877411050374424e-06,
"loss": 0.0,
"num_input_tokens_seen": 301688,
"step": 485
},
{
"epoch": 8.596491228070175,
"grad_norm": 0.0010899933986365795,
"learning_rate": 3.013156219837776e-06,
"loss": 0.0001,
"num_input_tokens_seen": 304504,
"step": 490
},
{
"epoch": 8.649122807017545,
"eval_loss": 0.2532731890678406,
"eval_runtime": 0.7915,
"eval_samples_per_second": 31.584,
"eval_steps_per_second": 8.843,
"num_input_tokens_seen": 306648,
"step": 493
},
{
"epoch": 8.68421052631579,
"grad_norm": 0.0009048488573171198,
"learning_rate": 2.659183991914696e-06,
"loss": 0.0001,
"num_input_tokens_seen": 308056,
"step": 495
},
{
"epoch": 8.771929824561404,
"grad_norm": 0.002603427739813924,
"learning_rate": 2.326156270090735e-06,
"loss": 0.0,
"num_input_tokens_seen": 310584,
"step": 500
},
{
"epoch": 8.859649122807017,
"grad_norm": 0.001305464655160904,
"learning_rate": 2.0143852677223075e-06,
"loss": 0.0,
"num_input_tokens_seen": 313528,
"step": 505
},
{
"epoch": 8.947368421052632,
"grad_norm": 0.0036463334690779448,
"learning_rate": 1.7241632699998123e-06,
"loss": 0.0002,
"num_input_tokens_seen": 316472,
"step": 510
},
{
"epoch": 9.035087719298245,
"grad_norm": 0.019788112491369247,
"learning_rate": 1.4557623599303903e-06,
"loss": 0.0001,
"num_input_tokens_seen": 318912,
"step": 515
},
{
"epoch": 9.12280701754386,
"grad_norm": 0.0026145647279918194,
"learning_rate": 1.2094341632602064e-06,
"loss": 0.0,
"num_input_tokens_seen": 321888,
"step": 520
},
{
"epoch": 9.157894736842104,
"eval_loss": 0.253935843706131,
"eval_runtime": 0.7922,
"eval_samples_per_second": 31.558,
"eval_steps_per_second": 8.836,
"num_input_tokens_seen": 323296,
"step": 522
},
{
"epoch": 9.210526315789474,
"grad_norm": 0.0016843760386109352,
"learning_rate": 9.85409612575411e-07,
"loss": 0.0001,
"num_input_tokens_seen": 325248,
"step": 525
},
{
"epoch": 9.298245614035087,
"grad_norm": 0.0010300269350409508,
"learning_rate": 7.838987308029427e-07,
"loss": 0.0001,
"num_input_tokens_seen": 328224,
"step": 530
},
{
"epoch": 9.385964912280702,
"grad_norm": 0.001682847854681313,
"learning_rate": 6.050904343141095e-07,
"loss": 0.0,
"num_input_tokens_seen": 330880,
"step": 535
},
{
"epoch": 9.473684210526315,
"grad_norm": 0.0009157847962342203,
"learning_rate": 4.491523558155714e-07,
"loss": 0.0002,
"num_input_tokens_seen": 334176,
"step": 540
},
{
"epoch": 9.56140350877193,
"grad_norm": 0.0018203345825895667,
"learning_rate": 3.162306871937387e-07,
"loss": 0.0,
"num_input_tokens_seen": 337280,
"step": 545
},
{
"epoch": 9.649122807017545,
"grad_norm": 0.0011194853577762842,
"learning_rate": 2.064500424599436e-07,
"loss": 0.0001,
"num_input_tokens_seen": 340480,
"step": 550
},
{
"epoch": 9.666666666666666,
"eval_loss": 0.2470671385526657,
"eval_runtime": 0.7954,
"eval_samples_per_second": 31.432,
"eval_steps_per_second": 8.801,
"num_input_tokens_seen": 340960,
"step": 551
},
{
"epoch": 9.736842105263158,
"grad_norm": 0.017748655751347542,
"learning_rate": 1.1991334092484318e-07,
"loss": 0.0001,
"num_input_tokens_seen": 343328,
"step": 555
},
{
"epoch": 9.824561403508772,
"grad_norm": 0.008531659841537476,
"learning_rate": 5.6701710711626334e-08,
"loss": 0.0001,
"num_input_tokens_seen": 345984,
"step": 560
},
{
"epoch": 9.912280701754385,
"grad_norm": 0.0015779946697875857,
"learning_rate": 1.6874412698408836e-08,
"loss": 0.0001,
"num_input_tokens_seen": 349280,
"step": 565
},
{
"epoch": 10.0,
"grad_norm": 0.0017016318161040545,
"learning_rate": 4.687849611939576e-10,
"loss": 0.0001,
"num_input_tokens_seen": 352296,
"step": 570
},
{
"epoch": 10.0,
"num_input_tokens_seen": 352296,
"step": 570,
"total_flos": 1.5894760020443136e+16,
"train_loss": 0.07623234518717877,
"train_runtime": 183.2479,
"train_samples_per_second": 12.278,
"train_steps_per_second": 3.111
}
],
"logging_steps": 5,
"max_steps": 570,
"num_input_tokens_seen": 352296,
"num_train_epochs": 10,
"save_steps": 29,
"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.5894760020443136e+16,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}