train_cb_1756729050 / trainer_state.json
rbelanec's picture
End of training
a6eaa29 verified
{
"best_global_step": 285,
"best_metric": 0.20361463725566864,
"best_model_checkpoint": "saves/prefix-tuning/llama-3-8b-instruct/train_cb_1756729050/checkpoint-285",
"epoch": 10.0,
"eval_steps": 57,
"global_step": 1130,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.04424778761061947,
"grad_norm": 124.327880859375,
"learning_rate": 1.7699115044247788e-06,
"loss": 9.9026,
"num_input_tokens_seen": 1296,
"step": 5
},
{
"epoch": 0.08849557522123894,
"grad_norm": 98.0537338256836,
"learning_rate": 3.982300884955752e-06,
"loss": 8.6995,
"num_input_tokens_seen": 2944,
"step": 10
},
{
"epoch": 0.13274336283185842,
"grad_norm": 77.6874008178711,
"learning_rate": 6.194690265486726e-06,
"loss": 6.9874,
"num_input_tokens_seen": 4544,
"step": 15
},
{
"epoch": 0.17699115044247787,
"grad_norm": 72.0156021118164,
"learning_rate": 8.407079646017701e-06,
"loss": 4.7381,
"num_input_tokens_seen": 6048,
"step": 20
},
{
"epoch": 0.22123893805309736,
"grad_norm": 73.16333770751953,
"learning_rate": 1.0619469026548673e-05,
"loss": 2.6354,
"num_input_tokens_seen": 7296,
"step": 25
},
{
"epoch": 0.26548672566371684,
"grad_norm": 59.97528839111328,
"learning_rate": 1.2831858407079647e-05,
"loss": 1.1568,
"num_input_tokens_seen": 8944,
"step": 30
},
{
"epoch": 0.30973451327433627,
"grad_norm": 14.040083885192871,
"learning_rate": 1.504424778761062e-05,
"loss": 0.3202,
"num_input_tokens_seen": 10848,
"step": 35
},
{
"epoch": 0.35398230088495575,
"grad_norm": 59.06977081298828,
"learning_rate": 1.7256637168141594e-05,
"loss": 0.4024,
"num_input_tokens_seen": 12416,
"step": 40
},
{
"epoch": 0.39823008849557523,
"grad_norm": 58.10889434814453,
"learning_rate": 1.946902654867257e-05,
"loss": 0.2926,
"num_input_tokens_seen": 14064,
"step": 45
},
{
"epoch": 0.4424778761061947,
"grad_norm": 7.289990425109863,
"learning_rate": 2.1681415929203542e-05,
"loss": 1.0618,
"num_input_tokens_seen": 15504,
"step": 50
},
{
"epoch": 0.48672566371681414,
"grad_norm": 39.314292907714844,
"learning_rate": 2.3893805309734516e-05,
"loss": 0.6315,
"num_input_tokens_seen": 16608,
"step": 55
},
{
"epoch": 0.504424778761062,
"eval_loss": 0.4109819531440735,
"eval_runtime": 0.6839,
"eval_samples_per_second": 36.555,
"eval_steps_per_second": 19.008,
"num_input_tokens_seen": 17136,
"step": 57
},
{
"epoch": 0.5309734513274337,
"grad_norm": 14.896934509277344,
"learning_rate": 2.610619469026549e-05,
"loss": 0.578,
"num_input_tokens_seen": 17904,
"step": 60
},
{
"epoch": 0.5752212389380531,
"grad_norm": 7.11203145980835,
"learning_rate": 2.831858407079646e-05,
"loss": 0.2188,
"num_input_tokens_seen": 19440,
"step": 65
},
{
"epoch": 0.6194690265486725,
"grad_norm": 31.49278450012207,
"learning_rate": 3.0530973451327434e-05,
"loss": 0.4324,
"num_input_tokens_seen": 21216,
"step": 70
},
{
"epoch": 0.6637168141592921,
"grad_norm": 1.894446849822998,
"learning_rate": 3.274336283185841e-05,
"loss": 0.1241,
"num_input_tokens_seen": 22400,
"step": 75
},
{
"epoch": 0.7079646017699115,
"grad_norm": 19.167335510253906,
"learning_rate": 3.495575221238938e-05,
"loss": 0.7159,
"num_input_tokens_seen": 24016,
"step": 80
},
{
"epoch": 0.7522123893805309,
"grad_norm": 22.42424774169922,
"learning_rate": 3.716814159292036e-05,
"loss": 0.3208,
"num_input_tokens_seen": 25360,
"step": 85
},
{
"epoch": 0.7964601769911505,
"grad_norm": 67.5926513671875,
"learning_rate": 3.938053097345133e-05,
"loss": 0.8685,
"num_input_tokens_seen": 26528,
"step": 90
},
{
"epoch": 0.8407079646017699,
"grad_norm": 41.6654052734375,
"learning_rate": 4.15929203539823e-05,
"loss": 1.3102,
"num_input_tokens_seen": 27728,
"step": 95
},
{
"epoch": 0.8849557522123894,
"grad_norm": 6.129133224487305,
"learning_rate": 4.380530973451328e-05,
"loss": 0.3212,
"num_input_tokens_seen": 29024,
"step": 100
},
{
"epoch": 0.9292035398230089,
"grad_norm": 7.041464805603027,
"learning_rate": 4.601769911504425e-05,
"loss": 0.2257,
"num_input_tokens_seen": 30128,
"step": 105
},
{
"epoch": 0.9734513274336283,
"grad_norm": 6.312572479248047,
"learning_rate": 4.823008849557522e-05,
"loss": 0.4932,
"num_input_tokens_seen": 31312,
"step": 110
},
{
"epoch": 1.008849557522124,
"eval_loss": 0.890156626701355,
"eval_runtime": 0.6901,
"eval_samples_per_second": 36.229,
"eval_steps_per_second": 18.839,
"num_input_tokens_seen": 32376,
"step": 114
},
{
"epoch": 1.0176991150442478,
"grad_norm": 0.2683965861797333,
"learning_rate": 4.9999880720033856e-05,
"loss": 0.2382,
"num_input_tokens_seen": 32664,
"step": 115
},
{
"epoch": 1.0619469026548674,
"grad_norm": 5.7456583976745605,
"learning_rate": 4.999570604073014e-05,
"loss": 0.6186,
"num_input_tokens_seen": 33992,
"step": 120
},
{
"epoch": 1.1061946902654867,
"grad_norm": 2.7945072650909424,
"learning_rate": 4.998556850128691e-05,
"loss": 0.2452,
"num_input_tokens_seen": 35528,
"step": 125
},
{
"epoch": 1.1504424778761062,
"grad_norm": 19.44849967956543,
"learning_rate": 4.996947052006874e-05,
"loss": 0.8597,
"num_input_tokens_seen": 36984,
"step": 130
},
{
"epoch": 1.1946902654867257,
"grad_norm": 12.182269096374512,
"learning_rate": 4.9947415937335635e-05,
"loss": 0.5896,
"num_input_tokens_seen": 38152,
"step": 135
},
{
"epoch": 1.238938053097345,
"grad_norm": 4.649380207061768,
"learning_rate": 4.9919410014326955e-05,
"loss": 0.2736,
"num_input_tokens_seen": 39528,
"step": 140
},
{
"epoch": 1.2831858407079646,
"grad_norm": 9.028091430664062,
"learning_rate": 4.98854594320063e-05,
"loss": 0.436,
"num_input_tokens_seen": 41288,
"step": 145
},
{
"epoch": 1.3274336283185841,
"grad_norm": 5.05112886428833,
"learning_rate": 4.98455722894677e-05,
"loss": 0.245,
"num_input_tokens_seen": 42744,
"step": 150
},
{
"epoch": 1.3716814159292037,
"grad_norm": 3.1134116649627686,
"learning_rate": 4.979975810200359e-05,
"loss": 0.601,
"num_input_tokens_seen": 43976,
"step": 155
},
{
"epoch": 1.415929203539823,
"grad_norm": 1.1430481672286987,
"learning_rate": 4.974802779883484e-05,
"loss": 0.4877,
"num_input_tokens_seen": 45576,
"step": 160
},
{
"epoch": 1.4601769911504425,
"grad_norm": 1.3173317909240723,
"learning_rate": 4.969039372050356e-05,
"loss": 0.1783,
"num_input_tokens_seen": 47208,
"step": 165
},
{
"epoch": 1.504424778761062,
"grad_norm": 0.2009192407131195,
"learning_rate": 4.9626869615929175e-05,
"loss": 0.3213,
"num_input_tokens_seen": 48488,
"step": 170
},
{
"epoch": 1.5132743362831858,
"eval_loss": 0.814801812171936,
"eval_runtime": 0.6887,
"eval_samples_per_second": 36.302,
"eval_steps_per_second": 18.877,
"num_input_tokens_seen": 48728,
"step": 171
},
{
"epoch": 1.5486725663716814,
"grad_norm": 3.0707428455352783,
"learning_rate": 4.9557470639128554e-05,
"loss": 0.9809,
"num_input_tokens_seen": 49736,
"step": 175
},
{
"epoch": 1.592920353982301,
"grad_norm": 2.4363770484924316,
"learning_rate": 4.9482213345600936e-05,
"loss": 0.3842,
"num_input_tokens_seen": 50920,
"step": 180
},
{
"epoch": 1.6371681415929205,
"grad_norm": 0.2922735810279846,
"learning_rate": 4.940111568837854e-05,
"loss": 0.3056,
"num_input_tokens_seen": 52296,
"step": 185
},
{
"epoch": 1.6814159292035398,
"grad_norm": 4.165451526641846,
"learning_rate": 4.931419701374377e-05,
"loss": 0.4018,
"num_input_tokens_seen": 54120,
"step": 190
},
{
"epoch": 1.7256637168141593,
"grad_norm": 1.9844036102294922,
"learning_rate": 4.922147805661402e-05,
"loss": 0.2221,
"num_input_tokens_seen": 55640,
"step": 195
},
{
"epoch": 1.7699115044247788,
"grad_norm": 4.524892330169678,
"learning_rate": 4.91229809355953e-05,
"loss": 0.1992,
"num_input_tokens_seen": 56728,
"step": 200
},
{
"epoch": 1.8141592920353982,
"grad_norm": 2.205657720565796,
"learning_rate": 4.901872914770569e-05,
"loss": 0.1971,
"num_input_tokens_seen": 58344,
"step": 205
},
{
"epoch": 1.8584070796460177,
"grad_norm": 3.2507164478302,
"learning_rate": 4.8908747562769995e-05,
"loss": 0.445,
"num_input_tokens_seen": 59512,
"step": 210
},
{
"epoch": 1.9026548672566372,
"grad_norm": 3.625652313232422,
"learning_rate": 4.8793062417486976e-05,
"loss": 0.4323,
"num_input_tokens_seen": 60968,
"step": 215
},
{
"epoch": 1.9469026548672566,
"grad_norm": 10.442010879516602,
"learning_rate": 4.867170130917034e-05,
"loss": 0.4836,
"num_input_tokens_seen": 62152,
"step": 220
},
{
"epoch": 1.991150442477876,
"grad_norm": 1.7526259422302246,
"learning_rate": 4.8544693189165324e-05,
"loss": 0.429,
"num_input_tokens_seen": 63432,
"step": 225
},
{
"epoch": 2.017699115044248,
"eval_loss": 0.24925395846366882,
"eval_runtime": 0.6896,
"eval_samples_per_second": 36.255,
"eval_steps_per_second": 18.852,
"num_input_tokens_seen": 64040,
"step": 228
},
{
"epoch": 2.0353982300884956,
"grad_norm": 1.434633493423462,
"learning_rate": 4.841206835594222e-05,
"loss": 0.1314,
"num_input_tokens_seen": 64520,
"step": 230
},
{
"epoch": 2.079646017699115,
"grad_norm": 0.7020671367645264,
"learning_rate": 4.8273858447868424e-05,
"loss": 0.2019,
"num_input_tokens_seen": 66104,
"step": 235
},
{
"epoch": 2.1238938053097347,
"grad_norm": 0.884209394454956,
"learning_rate": 4.813009643566101e-05,
"loss": 0.1156,
"num_input_tokens_seen": 67464,
"step": 240
},
{
"epoch": 2.168141592920354,
"grad_norm": 0.8240254521369934,
"learning_rate": 4.798081661452135e-05,
"loss": 0.2676,
"num_input_tokens_seen": 68824,
"step": 245
},
{
"epoch": 2.2123893805309733,
"grad_norm": 0.5540083646774292,
"learning_rate": 4.7826054595953815e-05,
"loss": 0.1612,
"num_input_tokens_seen": 70040,
"step": 250
},
{
"epoch": 2.256637168141593,
"grad_norm": 5.181052207946777,
"learning_rate": 4.766584729927049e-05,
"loss": 0.4297,
"num_input_tokens_seen": 71784,
"step": 255
},
{
"epoch": 2.3008849557522124,
"grad_norm": 2.5868945121765137,
"learning_rate": 4.750023294278381e-05,
"loss": 0.2929,
"num_input_tokens_seen": 73096,
"step": 260
},
{
"epoch": 2.3451327433628317,
"grad_norm": 5.291065216064453,
"learning_rate": 4.732925103468944e-05,
"loss": 0.3534,
"num_input_tokens_seen": 74328,
"step": 265
},
{
"epoch": 2.3893805309734515,
"grad_norm": 5.41591215133667,
"learning_rate": 4.715294236364135e-05,
"loss": 0.3324,
"num_input_tokens_seen": 75656,
"step": 270
},
{
"epoch": 2.433628318584071,
"grad_norm": 4.985315322875977,
"learning_rate": 4.6971348989021446e-05,
"loss": 0.419,
"num_input_tokens_seen": 77016,
"step": 275
},
{
"epoch": 2.47787610619469,
"grad_norm": 6.609219551086426,
"learning_rate": 4.6784514230906146e-05,
"loss": 0.4543,
"num_input_tokens_seen": 78376,
"step": 280
},
{
"epoch": 2.52212389380531,
"grad_norm": 1.5750737190246582,
"learning_rate": 4.659248265973205e-05,
"loss": 0.1579,
"num_input_tokens_seen": 79784,
"step": 285
},
{
"epoch": 2.52212389380531,
"eval_loss": 0.20361463725566864,
"eval_runtime": 0.6928,
"eval_samples_per_second": 36.086,
"eval_steps_per_second": 18.765,
"num_input_tokens_seen": 79784,
"step": 285
},
{
"epoch": 2.566371681415929,
"grad_norm": 1.2069061994552612,
"learning_rate": 4.639530008566349e-05,
"loss": 0.2658,
"num_input_tokens_seen": 81192,
"step": 290
},
{
"epoch": 2.6106194690265485,
"grad_norm": 1.4829492568969727,
"learning_rate": 4.6193013547664255e-05,
"loss": 0.1033,
"num_input_tokens_seen": 82552,
"step": 295
},
{
"epoch": 2.6548672566371683,
"grad_norm": 0.8109806776046753,
"learning_rate": 4.5985671302276165e-05,
"loss": 0.3427,
"num_input_tokens_seen": 84280,
"step": 300
},
{
"epoch": 2.6991150442477876,
"grad_norm": 3.9050145149230957,
"learning_rate": 4.577332281210727e-05,
"loss": 0.1133,
"num_input_tokens_seen": 85752,
"step": 305
},
{
"epoch": 2.7433628318584073,
"grad_norm": 0.07267215102910995,
"learning_rate": 4.55560187340322e-05,
"loss": 0.0774,
"num_input_tokens_seen": 87448,
"step": 310
},
{
"epoch": 2.7876106194690267,
"grad_norm": 6.648308753967285,
"learning_rate": 4.533381090710776e-05,
"loss": 0.3355,
"num_input_tokens_seen": 89080,
"step": 315
},
{
"epoch": 2.831858407079646,
"grad_norm": 10.083158493041992,
"learning_rate": 4.5106752340206435e-05,
"loss": 0.1969,
"num_input_tokens_seen": 90456,
"step": 320
},
{
"epoch": 2.8761061946902657,
"grad_norm": 0.3898797631263733,
"learning_rate": 4.4874897199370805e-05,
"loss": 0.5118,
"num_input_tokens_seen": 91624,
"step": 325
},
{
"epoch": 2.920353982300885,
"grad_norm": 0.1865801066160202,
"learning_rate": 4.463830079489196e-05,
"loss": 0.1496,
"num_input_tokens_seen": 92792,
"step": 330
},
{
"epoch": 2.9646017699115044,
"grad_norm": 6.828113079071045,
"learning_rate": 4.4397019568114987e-05,
"loss": 0.197,
"num_input_tokens_seen": 94280,
"step": 335
},
{
"epoch": 3.0088495575221237,
"grad_norm": 0.16399507224559784,
"learning_rate": 4.415111107797445e-05,
"loss": 0.0141,
"num_input_tokens_seen": 95752,
"step": 340
},
{
"epoch": 3.0265486725663715,
"eval_loss": 0.29375046491622925,
"eval_runtime": 0.6907,
"eval_samples_per_second": 36.196,
"eval_steps_per_second": 18.822,
"num_input_tokens_seen": 96200,
"step": 342
},
{
"epoch": 3.0530973451327434,
"grad_norm": 0.037823472172021866,
"learning_rate": 4.390063398726356e-05,
"loss": 0.3267,
"num_input_tokens_seen": 97000,
"step": 345
},
{
"epoch": 3.0973451327433628,
"grad_norm": 0.03760886937379837,
"learning_rate": 4.3645648048639734e-05,
"loss": 0.1783,
"num_input_tokens_seen": 98248,
"step": 350
},
{
"epoch": 3.1415929203539825,
"grad_norm": 0.14455364644527435,
"learning_rate": 4.338621409037031e-05,
"loss": 0.1624,
"num_input_tokens_seen": 99800,
"step": 355
},
{
"epoch": 3.185840707964602,
"grad_norm": 0.8414601683616638,
"learning_rate": 4.312239400182166e-05,
"loss": 0.1213,
"num_input_tokens_seen": 101080,
"step": 360
},
{
"epoch": 3.230088495575221,
"grad_norm": 0.21965432167053223,
"learning_rate": 4.285425071869511e-05,
"loss": 0.1963,
"num_input_tokens_seen": 102488,
"step": 365
},
{
"epoch": 3.274336283185841,
"grad_norm": 0.43919920921325684,
"learning_rate": 4.258184820801331e-05,
"loss": 0.0989,
"num_input_tokens_seen": 103912,
"step": 370
},
{
"epoch": 3.3185840707964602,
"grad_norm": 0.45976346731185913,
"learning_rate": 4.230525145286057e-05,
"loss": 0.0376,
"num_input_tokens_seen": 105400,
"step": 375
},
{
"epoch": 3.3628318584070795,
"grad_norm": 0.06820009648799896,
"learning_rate": 4.2024526436880785e-05,
"loss": 0.0357,
"num_input_tokens_seen": 106920,
"step": 380
},
{
"epoch": 3.4070796460176993,
"grad_norm": 11.349090576171875,
"learning_rate": 4.173974012853665e-05,
"loss": 0.4318,
"num_input_tokens_seen": 108280,
"step": 385
},
{
"epoch": 3.4513274336283186,
"grad_norm": 11.474207878112793,
"learning_rate": 4.1450960465134025e-05,
"loss": 0.2942,
"num_input_tokens_seen": 110024,
"step": 390
},
{
"epoch": 3.495575221238938,
"grad_norm": 0.09786193072795868,
"learning_rate": 4.1158256336615096e-05,
"loss": 0.1094,
"num_input_tokens_seen": 111304,
"step": 395
},
{
"epoch": 3.5309734513274336,
"eval_loss": 0.23912635445594788,
"eval_runtime": 0.6904,
"eval_samples_per_second": 36.211,
"eval_steps_per_second": 18.83,
"num_input_tokens_seen": 112440,
"step": 399
},
{
"epoch": 3.5398230088495577,
"grad_norm": 6.929598808288574,
"learning_rate": 4.086169756912431e-05,
"loss": 0.3946,
"num_input_tokens_seen": 112696,
"step": 400
},
{
"epoch": 3.584070796460177,
"grad_norm": 1.80954110622406,
"learning_rate": 4.056135490835098e-05,
"loss": 0.1046,
"num_input_tokens_seen": 114248,
"step": 405
},
{
"epoch": 3.6283185840707963,
"grad_norm": 1.144965648651123,
"learning_rate": 4.025730000265251e-05,
"loss": 0.2252,
"num_input_tokens_seen": 115400,
"step": 410
},
{
"epoch": 3.672566371681416,
"grad_norm": 4.598496913909912,
"learning_rate": 3.994960538596232e-05,
"loss": 0.2358,
"num_input_tokens_seen": 116808,
"step": 415
},
{
"epoch": 3.7168141592920354,
"grad_norm": 1.339223861694336,
"learning_rate": 3.963834446048644e-05,
"loss": 0.6255,
"num_input_tokens_seen": 118056,
"step": 420
},
{
"epoch": 3.7610619469026547,
"grad_norm": 0.20089839398860931,
"learning_rate": 3.9323591479193074e-05,
"loss": 0.1883,
"num_input_tokens_seen": 119512,
"step": 425
},
{
"epoch": 3.8053097345132745,
"grad_norm": 10.758119583129883,
"learning_rate": 3.90054215280991e-05,
"loss": 0.1571,
"num_input_tokens_seen": 120872,
"step": 430
},
{
"epoch": 3.849557522123894,
"grad_norm": 3.9737775325775146,
"learning_rate": 3.8683910508357934e-05,
"loss": 0.1996,
"num_input_tokens_seen": 122376,
"step": 435
},
{
"epoch": 3.893805309734513,
"grad_norm": 0.47557440400123596,
"learning_rate": 3.83591351181529e-05,
"loss": 0.1971,
"num_input_tokens_seen": 123544,
"step": 440
},
{
"epoch": 3.938053097345133,
"grad_norm": 1.0647215843200684,
"learning_rate": 3.803117283440044e-05,
"loss": 0.2362,
"num_input_tokens_seen": 125720,
"step": 445
},
{
"epoch": 3.982300884955752,
"grad_norm": 1.551072120666504,
"learning_rate": 3.7700101894267615e-05,
"loss": 0.3458,
"num_input_tokens_seen": 126952,
"step": 450
},
{
"epoch": 4.0265486725663715,
"grad_norm": 0.5870348811149597,
"learning_rate": 3.736600127650817e-05,
"loss": 0.2226,
"num_input_tokens_seen": 128488,
"step": 455
},
{
"epoch": 4.035398230088496,
"eval_loss": 0.21712732315063477,
"eval_runtime": 0.6947,
"eval_samples_per_second": 35.985,
"eval_steps_per_second": 18.712,
"num_input_tokens_seen": 128712,
"step": 456
},
{
"epoch": 4.070796460176991,
"grad_norm": 1.6020293235778809,
"learning_rate": 3.702895068262177e-05,
"loss": 0.0339,
"num_input_tokens_seen": 129880,
"step": 460
},
{
"epoch": 4.115044247787611,
"grad_norm": 3.542776584625244,
"learning_rate": 3.66890305178407e-05,
"loss": 0.1494,
"num_input_tokens_seen": 131176,
"step": 465
},
{
"epoch": 4.15929203539823,
"grad_norm": 0.09453494846820831,
"learning_rate": 3.63463218719489e-05,
"loss": 0.1752,
"num_input_tokens_seen": 132376,
"step": 470
},
{
"epoch": 4.20353982300885,
"grad_norm": 0.05962882563471794,
"learning_rate": 3.600090649993741e-05,
"loss": 0.2276,
"num_input_tokens_seen": 133608,
"step": 475
},
{
"epoch": 4.247787610619469,
"grad_norm": 0.5571654438972473,
"learning_rate": 3.565286680250138e-05,
"loss": 0.1814,
"num_input_tokens_seen": 135080,
"step": 480
},
{
"epoch": 4.292035398230088,
"grad_norm": 2.1021406650543213,
"learning_rate": 3.530228580638291e-05,
"loss": 0.0218,
"num_input_tokens_seen": 136264,
"step": 485
},
{
"epoch": 4.336283185840708,
"grad_norm": 0.1647375077009201,
"learning_rate": 3.494924714456454e-05,
"loss": 0.1887,
"num_input_tokens_seen": 137480,
"step": 490
},
{
"epoch": 4.380530973451328,
"grad_norm": 12.375977516174316,
"learning_rate": 3.459383503631823e-05,
"loss": 0.3622,
"num_input_tokens_seen": 138536,
"step": 495
},
{
"epoch": 4.424778761061947,
"grad_norm": 0.7150776982307434,
"learning_rate": 3.423613426711431e-05,
"loss": 0.1113,
"num_input_tokens_seen": 140360,
"step": 500
},
{
"epoch": 4.469026548672566,
"grad_norm": 0.1175035908818245,
"learning_rate": 3.3876230168395525e-05,
"loss": 0.1433,
"num_input_tokens_seen": 141784,
"step": 505
},
{
"epoch": 4.513274336283186,
"grad_norm": 3.413067579269409,
"learning_rate": 3.3514208597220705e-05,
"loss": 0.0679,
"num_input_tokens_seen": 143192,
"step": 510
},
{
"epoch": 4.539823008849558,
"eval_loss": 0.3177097737789154,
"eval_runtime": 0.6885,
"eval_samples_per_second": 36.313,
"eval_steps_per_second": 18.883,
"num_input_tokens_seen": 143944,
"step": 513
},
{
"epoch": 4.557522123893805,
"grad_norm": 4.12099027633667,
"learning_rate": 3.315015591578314e-05,
"loss": 0.0438,
"num_input_tokens_seen": 144440,
"step": 515
},
{
"epoch": 4.601769911504425,
"grad_norm": 0.3228464126586914,
"learning_rate": 3.278415897080839e-05,
"loss": 0.037,
"num_input_tokens_seen": 145768,
"step": 520
},
{
"epoch": 4.646017699115045,
"grad_norm": 0.06757593154907227,
"learning_rate": 3.2416305072836557e-05,
"loss": 0.1863,
"num_input_tokens_seen": 147128,
"step": 525
},
{
"epoch": 4.6902654867256635,
"grad_norm": 0.14144816994667053,
"learning_rate": 3.204668197539385e-05,
"loss": 0.0026,
"num_input_tokens_seen": 148184,
"step": 530
},
{
"epoch": 4.734513274336283,
"grad_norm": 26.914226531982422,
"learning_rate": 3.167537785405854e-05,
"loss": 0.2017,
"num_input_tokens_seen": 149880,
"step": 535
},
{
"epoch": 4.778761061946903,
"grad_norm": 0.06391548365354538,
"learning_rate": 3.13024812854262e-05,
"loss": 0.3003,
"num_input_tokens_seen": 151528,
"step": 540
},
{
"epoch": 4.823008849557522,
"grad_norm": 0.18799445033073425,
"learning_rate": 3.092808122597924e-05,
"loss": 0.2774,
"num_input_tokens_seen": 153080,
"step": 545
},
{
"epoch": 4.867256637168142,
"grad_norm": 4.300349712371826,
"learning_rate": 3.0552266990865934e-05,
"loss": 0.1622,
"num_input_tokens_seen": 154952,
"step": 550
},
{
"epoch": 4.911504424778761,
"grad_norm": 0.1420593559741974,
"learning_rate": 3.017512823259373e-05,
"loss": 0.0403,
"num_input_tokens_seen": 156136,
"step": 555
},
{
"epoch": 4.95575221238938,
"grad_norm": 0.020832767710089684,
"learning_rate": 2.9796754919642228e-05,
"loss": 0.0034,
"num_input_tokens_seen": 157560,
"step": 560
},
{
"epoch": 5.0,
"grad_norm": 19.973548889160156,
"learning_rate": 2.941723731500068e-05,
"loss": 0.1692,
"num_input_tokens_seen": 158768,
"step": 565
},
{
"epoch": 5.04424778761062,
"grad_norm": 0.10568743944168091,
"learning_rate": 2.9036665954635266e-05,
"loss": 0.286,
"num_input_tokens_seen": 160016,
"step": 570
},
{
"epoch": 5.04424778761062,
"eval_loss": 0.26774680614471436,
"eval_runtime": 0.6897,
"eval_samples_per_second": 36.245,
"eval_steps_per_second": 18.848,
"num_input_tokens_seen": 160016,
"step": 570
},
{
"epoch": 5.088495575221239,
"grad_norm": 0.019847629591822624,
"learning_rate": 2.8655131625891274e-05,
"loss": 0.1958,
"num_input_tokens_seen": 161360,
"step": 575
},
{
"epoch": 5.132743362831858,
"grad_norm": 0.04539639875292778,
"learning_rate": 2.8272725345835278e-05,
"loss": 0.024,
"num_input_tokens_seen": 163104,
"step": 580
},
{
"epoch": 5.176991150442478,
"grad_norm": 0.7734869122505188,
"learning_rate": 2.7889538339542527e-05,
"loss": 0.0948,
"num_input_tokens_seen": 164368,
"step": 585
},
{
"epoch": 5.221238938053097,
"grad_norm": 0.4986306428909302,
"learning_rate": 2.7505662018334698e-05,
"loss": 0.0052,
"num_input_tokens_seen": 165936,
"step": 590
},
{
"epoch": 5.265486725663717,
"grad_norm": 2.053908348083496,
"learning_rate": 2.712118795797329e-05,
"loss": 0.1832,
"num_input_tokens_seen": 167360,
"step": 595
},
{
"epoch": 5.3097345132743365,
"grad_norm": 8.486614227294922,
"learning_rate": 2.6736207876813646e-05,
"loss": 0.1174,
"num_input_tokens_seen": 168944,
"step": 600
},
{
"epoch": 5.353982300884955,
"grad_norm": 0.1965610682964325,
"learning_rate": 2.635081361392513e-05,
"loss": 0.0581,
"num_input_tokens_seen": 170672,
"step": 605
},
{
"epoch": 5.398230088495575,
"grad_norm": 0.14119859039783478,
"learning_rate": 2.596509710718235e-05,
"loss": 0.0386,
"num_input_tokens_seen": 172016,
"step": 610
},
{
"epoch": 5.442477876106195,
"grad_norm": 0.8027634024620056,
"learning_rate": 2.5579150371332956e-05,
"loss": 0.0684,
"num_input_tokens_seen": 173184,
"step": 615
},
{
"epoch": 5.486725663716814,
"grad_norm": 0.2336353361606598,
"learning_rate": 2.5193065476046955e-05,
"loss": 0.0022,
"num_input_tokens_seen": 174352,
"step": 620
},
{
"epoch": 5.530973451327434,
"grad_norm": 0.20318181812763214,
"learning_rate": 2.480693452395305e-05,
"loss": 0.0158,
"num_input_tokens_seen": 175968,
"step": 625
},
{
"epoch": 5.548672566371682,
"eval_loss": 0.36652490496635437,
"eval_runtime": 0.6926,
"eval_samples_per_second": 36.097,
"eval_steps_per_second": 18.771,
"num_input_tokens_seen": 176688,
"step": 627
},
{
"epoch": 5.575221238938053,
"grad_norm": 2.126171588897705,
"learning_rate": 2.4420849628667046e-05,
"loss": 0.1081,
"num_input_tokens_seen": 177408,
"step": 630
},
{
"epoch": 5.619469026548672,
"grad_norm": 0.00936133973300457,
"learning_rate": 2.4034902892817648e-05,
"loss": 0.1238,
"num_input_tokens_seen": 178608,
"step": 635
},
{
"epoch": 5.663716814159292,
"grad_norm": 0.017639046534895897,
"learning_rate": 2.3649186386074872e-05,
"loss": 0.0125,
"num_input_tokens_seen": 179824,
"step": 640
},
{
"epoch": 5.707964601769912,
"grad_norm": 0.0590643435716629,
"learning_rate": 2.3263792123186353e-05,
"loss": 0.1419,
"num_input_tokens_seen": 181088,
"step": 645
},
{
"epoch": 5.752212389380531,
"grad_norm": 17.686687469482422,
"learning_rate": 2.2878812042026713e-05,
"loss": 0.1366,
"num_input_tokens_seen": 182816,
"step": 650
},
{
"epoch": 5.79646017699115,
"grad_norm": 2.9926984310150146,
"learning_rate": 2.2494337981665304e-05,
"loss": 0.1155,
"num_input_tokens_seen": 184224,
"step": 655
},
{
"epoch": 5.84070796460177,
"grad_norm": 0.08732306212186813,
"learning_rate": 2.2110461660457482e-05,
"loss": 0.1308,
"num_input_tokens_seen": 185776,
"step": 660
},
{
"epoch": 5.88495575221239,
"grad_norm": 0.05544602870941162,
"learning_rate": 2.1727274654164728e-05,
"loss": 0.0374,
"num_input_tokens_seen": 187184,
"step": 665
},
{
"epoch": 5.929203539823009,
"grad_norm": 0.07477652281522751,
"learning_rate": 2.1344868374108732e-05,
"loss": 0.0458,
"num_input_tokens_seen": 188528,
"step": 670
},
{
"epoch": 5.9734513274336285,
"grad_norm": 12.648833274841309,
"learning_rate": 2.096333404536474e-05,
"loss": 0.1143,
"num_input_tokens_seen": 189680,
"step": 675
},
{
"epoch": 6.017699115044247,
"grad_norm": 10.302857398986816,
"learning_rate": 2.0582762684999327e-05,
"loss": 0.027,
"num_input_tokens_seen": 191008,
"step": 680
},
{
"epoch": 6.053097345132743,
"eval_loss": 0.2993113398551941,
"eval_runtime": 0.6925,
"eval_samples_per_second": 36.099,
"eval_steps_per_second": 18.771,
"num_input_tokens_seen": 192272,
"step": 684
},
{
"epoch": 6.061946902654867,
"grad_norm": 7.7002339363098145,
"learning_rate": 2.0203245080357775e-05,
"loss": 0.0254,
"num_input_tokens_seen": 192512,
"step": 685
},
{
"epoch": 6.106194690265487,
"grad_norm": 15.887398719787598,
"learning_rate": 1.982487176740627e-05,
"loss": 0.0627,
"num_input_tokens_seen": 193904,
"step": 690
},
{
"epoch": 6.150442477876107,
"grad_norm": 0.022938942536711693,
"learning_rate": 1.944773300913407e-05,
"loss": 0.1792,
"num_input_tokens_seen": 195616,
"step": 695
},
{
"epoch": 6.1946902654867255,
"grad_norm": 0.017901377752423286,
"learning_rate": 1.907191877402076e-05,
"loss": 0.0006,
"num_input_tokens_seen": 196832,
"step": 700
},
{
"epoch": 6.238938053097345,
"grad_norm": 0.10689708590507507,
"learning_rate": 1.8697518714573807e-05,
"loss": 0.0377,
"num_input_tokens_seen": 198592,
"step": 705
},
{
"epoch": 6.283185840707965,
"grad_norm": 0.5050591826438904,
"learning_rate": 1.832462214594146e-05,
"loss": 0.2142,
"num_input_tokens_seen": 199664,
"step": 710
},
{
"epoch": 6.327433628318584,
"grad_norm": 0.05065830796957016,
"learning_rate": 1.7953318024606157e-05,
"loss": 0.0059,
"num_input_tokens_seen": 201456,
"step": 715
},
{
"epoch": 6.371681415929204,
"grad_norm": 2.396775245666504,
"learning_rate": 1.7583694927163453e-05,
"loss": 0.0098,
"num_input_tokens_seen": 202752,
"step": 720
},
{
"epoch": 6.415929203539823,
"grad_norm": 0.011908108368515968,
"learning_rate": 1.7215841029191616e-05,
"loss": 0.0026,
"num_input_tokens_seen": 204336,
"step": 725
},
{
"epoch": 6.460176991150442,
"grad_norm": 5.227853298187256,
"learning_rate": 1.6849844084216874e-05,
"loss": 0.0056,
"num_input_tokens_seen": 205488,
"step": 730
},
{
"epoch": 6.504424778761062,
"grad_norm": 0.00683935359120369,
"learning_rate": 1.648579140277931e-05,
"loss": 0.0017,
"num_input_tokens_seen": 207008,
"step": 735
},
{
"epoch": 6.548672566371682,
"grad_norm": 0.029682686552405357,
"learning_rate": 1.612376983160449e-05,
"loss": 0.0012,
"num_input_tokens_seen": 208512,
"step": 740
},
{
"epoch": 6.557522123893805,
"eval_loss": 0.32990938425064087,
"eval_runtime": 0.6905,
"eval_samples_per_second": 36.204,
"eval_steps_per_second": 18.826,
"num_input_tokens_seen": 208944,
"step": 741
},
{
"epoch": 6.592920353982301,
"grad_norm": 0.06845102459192276,
"learning_rate": 1.576386573288569e-05,
"loss": 0.0184,
"num_input_tokens_seen": 209888,
"step": 745
},
{
"epoch": 6.6371681415929205,
"grad_norm": 0.012435441836714745,
"learning_rate": 1.5406164963681774e-05,
"loss": 0.0081,
"num_input_tokens_seen": 211008,
"step": 750
},
{
"epoch": 6.68141592920354,
"grad_norm": 0.006153010297566652,
"learning_rate": 1.5050752855435457e-05,
"loss": 0.0157,
"num_input_tokens_seen": 212240,
"step": 755
},
{
"epoch": 6.725663716814159,
"grad_norm": 7.11568546295166,
"learning_rate": 1.4697714193617097e-05,
"loss": 0.0158,
"num_input_tokens_seen": 213456,
"step": 760
},
{
"epoch": 6.769911504424779,
"grad_norm": 0.006165863014757633,
"learning_rate": 1.4347133197498619e-05,
"loss": 0.0016,
"num_input_tokens_seen": 214768,
"step": 765
},
{
"epoch": 6.814159292035399,
"grad_norm": 0.005989910569041967,
"learning_rate": 1.3999093500062587e-05,
"loss": 0.0008,
"num_input_tokens_seen": 216000,
"step": 770
},
{
"epoch": 6.8584070796460175,
"grad_norm": 0.0031818023417145014,
"learning_rate": 1.3653678128051106e-05,
"loss": 0.123,
"num_input_tokens_seen": 217520,
"step": 775
},
{
"epoch": 6.902654867256637,
"grad_norm": 0.28696584701538086,
"learning_rate": 1.3310969482159297e-05,
"loss": 0.0133,
"num_input_tokens_seen": 219040,
"step": 780
},
{
"epoch": 6.946902654867257,
"grad_norm": 0.6769438982009888,
"learning_rate": 1.2971049317378237e-05,
"loss": 0.0026,
"num_input_tokens_seen": 220688,
"step": 785
},
{
"epoch": 6.991150442477876,
"grad_norm": 0.004186061676591635,
"learning_rate": 1.263399872349183e-05,
"loss": 0.0002,
"num_input_tokens_seen": 222304,
"step": 790
},
{
"epoch": 7.035398230088496,
"grad_norm": 0.006434621289372444,
"learning_rate": 1.2299898105732385e-05,
"loss": 0.0001,
"num_input_tokens_seen": 223584,
"step": 795
},
{
"epoch": 7.061946902654867,
"eval_loss": 0.26334142684936523,
"eval_runtime": 0.6922,
"eval_samples_per_second": 36.118,
"eval_steps_per_second": 18.782,
"num_input_tokens_seen": 224288,
"step": 798
},
{
"epoch": 7.079646017699115,
"grad_norm": 0.004887331277132034,
"learning_rate": 1.1968827165599564e-05,
"loss": 0.0055,
"num_input_tokens_seen": 224816,
"step": 800
},
{
"epoch": 7.123893805309734,
"grad_norm": 0.006834018509835005,
"learning_rate": 1.1640864881847105e-05,
"loss": 0.0004,
"num_input_tokens_seen": 226192,
"step": 805
},
{
"epoch": 7.168141592920354,
"grad_norm": 0.015779342502355576,
"learning_rate": 1.1316089491642076e-05,
"loss": 0.0001,
"num_input_tokens_seen": 227440,
"step": 810
},
{
"epoch": 7.212389380530974,
"grad_norm": 0.0020266680512577295,
"learning_rate": 1.099457847190091e-05,
"loss": 0.0001,
"num_input_tokens_seen": 228928,
"step": 815
},
{
"epoch": 7.256637168141593,
"grad_norm": 0.014176726341247559,
"learning_rate": 1.067640852080693e-05,
"loss": 0.0002,
"num_input_tokens_seen": 230416,
"step": 820
},
{
"epoch": 7.300884955752212,
"grad_norm": 0.0020503615960478783,
"learning_rate": 1.0361655539513565e-05,
"loss": 0.0007,
"num_input_tokens_seen": 231536,
"step": 825
},
{
"epoch": 7.345132743362832,
"grad_norm": 0.004866134840995073,
"learning_rate": 1.0050394614037687e-05,
"loss": 0.1925,
"num_input_tokens_seen": 233072,
"step": 830
},
{
"epoch": 7.389380530973451,
"grad_norm": 0.004155035596340895,
"learning_rate": 9.742699997347498e-06,
"loss": 0.0002,
"num_input_tokens_seen": 234416,
"step": 835
},
{
"epoch": 7.433628318584071,
"grad_norm": 0.0061325933784246445,
"learning_rate": 9.438645091649029e-06,
"loss": 0.0121,
"num_input_tokens_seen": 235776,
"step": 840
},
{
"epoch": 7.477876106194691,
"grad_norm": 0.022556915879249573,
"learning_rate": 9.138302430875695e-06,
"loss": 0.0001,
"num_input_tokens_seen": 237248,
"step": 845
},
{
"epoch": 7.522123893805309,
"grad_norm": 0.5717719793319702,
"learning_rate": 8.841743663384913e-06,
"loss": 0.0008,
"num_input_tokens_seen": 238544,
"step": 850
},
{
"epoch": 7.566371681415929,
"grad_norm": 0.031122738495469093,
"learning_rate": 8.549039534865979e-06,
"loss": 0.0017,
"num_input_tokens_seen": 239840,
"step": 855
},
{
"epoch": 7.566371681415929,
"eval_loss": 0.26843762397766113,
"eval_runtime": 0.6962,
"eval_samples_per_second": 35.91,
"eval_steps_per_second": 18.673,
"num_input_tokens_seen": 239840,
"step": 855
},
{
"epoch": 7.610619469026549,
"grad_norm": 0.003987740259617567,
"learning_rate": 8.260259871463352e-06,
"loss": 0.0056,
"num_input_tokens_seen": 241440,
"step": 860
},
{
"epoch": 7.654867256637168,
"grad_norm": 0.013421153649687767,
"learning_rate": 7.975473563119216e-06,
"loss": 0.0002,
"num_input_tokens_seen": 243216,
"step": 865
},
{
"epoch": 7.699115044247788,
"grad_norm": 0.5913791060447693,
"learning_rate": 7.69474854713943e-06,
"loss": 0.0009,
"num_input_tokens_seen": 244400,
"step": 870
},
{
"epoch": 7.743362831858407,
"grad_norm": 0.00446932390332222,
"learning_rate": 7.4181517919866914e-06,
"loss": 0.0005,
"num_input_tokens_seen": 245520,
"step": 875
},
{
"epoch": 7.787610619469026,
"grad_norm": 0.0035711266100406647,
"learning_rate": 7.1457492813049e-06,
"loss": 0.0002,
"num_input_tokens_seen": 246688,
"step": 880
},
{
"epoch": 7.831858407079646,
"grad_norm": 0.053317610174417496,
"learning_rate": 6.877605998178344e-06,
"loss": 0.0003,
"num_input_tokens_seen": 248320,
"step": 885
},
{
"epoch": 7.876106194690266,
"grad_norm": 0.05435475334525108,
"learning_rate": 6.613785909629686e-06,
"loss": 0.0005,
"num_input_tokens_seen": 249616,
"step": 890
},
{
"epoch": 7.920353982300885,
"grad_norm": 0.00680264737457037,
"learning_rate": 6.354351951360268e-06,
"loss": 0.0001,
"num_input_tokens_seen": 251264,
"step": 895
},
{
"epoch": 7.964601769911504,
"grad_norm": 0.006205928511917591,
"learning_rate": 6.099366012736438e-06,
"loss": 0.0004,
"num_input_tokens_seen": 252768,
"step": 900
},
{
"epoch": 8.008849557522124,
"grad_norm": 0.004214154090732336,
"learning_rate": 5.848888922025553e-06,
"loss": 0.0001,
"num_input_tokens_seen": 254096,
"step": 905
},
{
"epoch": 8.053097345132743,
"grad_norm": 0.002537103835493326,
"learning_rate": 5.602980431885024e-06,
"loss": 0.0001,
"num_input_tokens_seen": 255408,
"step": 910
},
{
"epoch": 8.070796460176991,
"eval_loss": 0.2845822274684906,
"eval_runtime": 0.6949,
"eval_samples_per_second": 35.977,
"eval_steps_per_second": 18.708,
"num_input_tokens_seen": 255984,
"step": 912
},
{
"epoch": 8.097345132743364,
"grad_norm": 0.003600750118494034,
"learning_rate": 5.361699205108042e-06,
"loss": 0.0002,
"num_input_tokens_seen": 257008,
"step": 915
},
{
"epoch": 8.141592920353983,
"grad_norm": 0.009577029384672642,
"learning_rate": 5.125102800629203e-06,
"loss": 0.0006,
"num_input_tokens_seen": 258176,
"step": 920
},
{
"epoch": 8.185840707964601,
"grad_norm": 0.0025581864174455404,
"learning_rate": 4.893247659793568e-06,
"loss": 0.0001,
"num_input_tokens_seen": 259552,
"step": 925
},
{
"epoch": 8.230088495575222,
"grad_norm": 0.02438344433903694,
"learning_rate": 4.666189092892245e-06,
"loss": 0.0001,
"num_input_tokens_seen": 261104,
"step": 930
},
{
"epoch": 8.274336283185841,
"grad_norm": 0.0018822092097252607,
"learning_rate": 4.443981265967806e-06,
"loss": 0.0001,
"num_input_tokens_seen": 262496,
"step": 935
},
{
"epoch": 8.31858407079646,
"grad_norm": 0.003804213600233197,
"learning_rate": 4.226677187892739e-06,
"loss": 0.0001,
"num_input_tokens_seen": 263744,
"step": 940
},
{
"epoch": 8.36283185840708,
"grad_norm": 0.0037322845309972763,
"learning_rate": 4.014328697723835e-06,
"loss": 0.0001,
"num_input_tokens_seen": 265024,
"step": 945
},
{
"epoch": 8.4070796460177,
"grad_norm": 0.0029338463209569454,
"learning_rate": 3.806986452335748e-06,
"loss": 0.044,
"num_input_tokens_seen": 266320,
"step": 950
},
{
"epoch": 8.451327433628318,
"grad_norm": 0.010909894481301308,
"learning_rate": 3.604699914336515e-06,
"loss": 0.0002,
"num_input_tokens_seen": 267648,
"step": 955
},
{
"epoch": 8.495575221238939,
"grad_norm": 0.0041561429388821125,
"learning_rate": 3.4075173402679574e-06,
"loss": 0.0001,
"num_input_tokens_seen": 269264,
"step": 960
},
{
"epoch": 8.539823008849558,
"grad_norm": 0.003678722772747278,
"learning_rate": 3.215485769093862e-06,
"loss": 0.0008,
"num_input_tokens_seen": 270816,
"step": 965
},
{
"epoch": 8.575221238938052,
"eval_loss": 0.28000906109809875,
"eval_runtime": 0.6906,
"eval_samples_per_second": 36.2,
"eval_steps_per_second": 18.824,
"num_input_tokens_seen": 272064,
"step": 969
},
{
"epoch": 8.584070796460177,
"grad_norm": 0.018815837800502777,
"learning_rate": 3.028651010978556e-06,
"loss": 0.0002,
"num_input_tokens_seen": 272304,
"step": 970
},
{
"epoch": 8.628318584070797,
"grad_norm": 0.04938015341758728,
"learning_rate": 2.8470576363586633e-06,
"loss": 0.0002,
"num_input_tokens_seen": 273824,
"step": 975
},
{
"epoch": 8.672566371681416,
"grad_norm": 0.016595328226685524,
"learning_rate": 2.67074896531056e-06,
"loss": 0.0002,
"num_input_tokens_seen": 275024,
"step": 980
},
{
"epoch": 8.716814159292035,
"grad_norm": 0.005576372612267733,
"learning_rate": 2.499767057216193e-06,
"loss": 0.0002,
"num_input_tokens_seen": 276256,
"step": 985
},
{
"epoch": 8.761061946902656,
"grad_norm": 0.04221276193857193,
"learning_rate": 2.334152700729511e-06,
"loss": 0.0001,
"num_input_tokens_seen": 277536,
"step": 990
},
{
"epoch": 8.805309734513274,
"grad_norm": 0.01613134704530239,
"learning_rate": 2.1739454040461794e-06,
"loss": 0.0001,
"num_input_tokens_seen": 279104,
"step": 995
},
{
"epoch": 8.849557522123893,
"grad_norm": 0.00385455833747983,
"learning_rate": 2.019183385478654e-06,
"loss": 0.0001,
"num_input_tokens_seen": 280480,
"step": 1000
},
{
"epoch": 8.893805309734514,
"grad_norm": 0.024490782991051674,
"learning_rate": 1.8699035643389928e-06,
"loss": 0.0003,
"num_input_tokens_seen": 281664,
"step": 1005
},
{
"epoch": 8.938053097345133,
"grad_norm": 0.0034661281388252974,
"learning_rate": 1.7261415521315798e-06,
"loss": 0.0002,
"num_input_tokens_seen": 283248,
"step": 1010
},
{
"epoch": 8.982300884955752,
"grad_norm": 0.024979224428534508,
"learning_rate": 1.587931644057783e-06,
"loss": 0.0003,
"num_input_tokens_seen": 284400,
"step": 1015
},
{
"epoch": 9.026548672566372,
"grad_norm": 0.005210904870182276,
"learning_rate": 1.455306810834678e-06,
"loss": 0.0002,
"num_input_tokens_seen": 285880,
"step": 1020
},
{
"epoch": 9.070796460176991,
"grad_norm": 0.003795287339016795,
"learning_rate": 1.3282986908296713e-06,
"loss": 0.0003,
"num_input_tokens_seen": 287480,
"step": 1025
},
{
"epoch": 9.079646017699115,
"eval_loss": 0.27309226989746094,
"eval_runtime": 0.6951,
"eval_samples_per_second": 35.968,
"eval_steps_per_second": 18.703,
"num_input_tokens_seen": 287928,
"step": 1026
},
{
"epoch": 9.11504424778761,
"grad_norm": 0.006229293067008257,
"learning_rate": 1.2069375825130292e-06,
"loss": 0.0003,
"num_input_tokens_seen": 289144,
"step": 1030
},
{
"epoch": 9.15929203539823,
"grad_norm": 0.0026166548486799,
"learning_rate": 1.091252437230003e-06,
"loss": 0.0001,
"num_input_tokens_seen": 290504,
"step": 1035
},
{
"epoch": 9.20353982300885,
"grad_norm": 0.00397622212767601,
"learning_rate": 9.812708522943164e-07,
"loss": 0.0002,
"num_input_tokens_seen": 291784,
"step": 1040
},
{
"epoch": 9.247787610619469,
"grad_norm": 0.0022089998237788677,
"learning_rate": 8.770190644047077e-07,
"loss": 0.0001,
"num_input_tokens_seen": 292904,
"step": 1045
},
{
"epoch": 9.29203539823009,
"grad_norm": 0.0031590466387569904,
"learning_rate": 7.785219433859847e-07,
"loss": 0.0002,
"num_input_tokens_seen": 294360,
"step": 1050
},
{
"epoch": 9.336283185840708,
"grad_norm": 0.010320561937987804,
"learning_rate": 6.858029862562343e-07,
"loss": 0.0001,
"num_input_tokens_seen": 295992,
"step": 1055
},
{
"epoch": 9.380530973451327,
"grad_norm": 0.016771700233221054,
"learning_rate": 5.988843116214616e-07,
"loss": 0.0003,
"num_input_tokens_seen": 297272,
"step": 1060
},
{
"epoch": 9.424778761061948,
"grad_norm": 0.005522893276065588,
"learning_rate": 5.177866543990689e-07,
"loss": 0.0001,
"num_input_tokens_seen": 298360,
"step": 1065
},
{
"epoch": 9.469026548672566,
"grad_norm": 0.00910731591284275,
"learning_rate": 4.425293608714548e-07,
"loss": 0.0002,
"num_input_tokens_seen": 299656,
"step": 1070
},
{
"epoch": 9.513274336283185,
"grad_norm": 0.016494672745466232,
"learning_rate": 3.7313038407083003e-07,
"loss": 0.0001,
"num_input_tokens_seen": 301432,
"step": 1075
},
{
"epoch": 9.557522123893806,
"grad_norm": 0.005664309021085501,
"learning_rate": 3.0960627949644107e-07,
"loss": 0.0001,
"num_input_tokens_seen": 302952,
"step": 1080
},
{
"epoch": 9.584070796460177,
"eval_loss": 0.2795742452144623,
"eval_runtime": 0.6922,
"eval_samples_per_second": 36.119,
"eval_steps_per_second": 18.782,
"num_input_tokens_seen": 303800,
"step": 1083
},
{
"epoch": 9.601769911504425,
"grad_norm": 0.03134797140955925,
"learning_rate": 2.5197220116515987e-07,
"loss": 0.0002,
"num_input_tokens_seen": 304296,
"step": 1085
},
{
"epoch": 9.646017699115044,
"grad_norm": 0.0026311087422072887,
"learning_rate": 2.0024189799641435e-07,
"loss": 0.0001,
"num_input_tokens_seen": 305784,
"step": 1090
},
{
"epoch": 9.690265486725664,
"grad_norm": 0.003348633646965027,
"learning_rate": 1.5442771053230665e-07,
"loss": 0.0001,
"num_input_tokens_seen": 307592,
"step": 1095
},
{
"epoch": 9.734513274336283,
"grad_norm": 0.005454523954540491,
"learning_rate": 1.1454056799370672e-07,
"loss": 0.0001,
"num_input_tokens_seen": 308760,
"step": 1100
},
{
"epoch": 9.778761061946902,
"grad_norm": 0.01998157612979412,
"learning_rate": 8.058998567304633e-08,
"loss": 0.0002,
"num_input_tokens_seen": 310040,
"step": 1105
},
{
"epoch": 9.823008849557523,
"grad_norm": 0.003131413133814931,
"learning_rate": 5.2584062664368105e-08,
"loss": 0.0001,
"num_input_tokens_seen": 311752,
"step": 1110
},
{
"epoch": 9.867256637168142,
"grad_norm": 0.004119914025068283,
"learning_rate": 3.0529479931265535e-08,
"loss": 0.0001,
"num_input_tokens_seen": 313128,
"step": 1115
},
{
"epoch": 9.91150442477876,
"grad_norm": 0.003136214567348361,
"learning_rate": 1.443149871309113e-08,
"loss": 0.0002,
"num_input_tokens_seen": 314568,
"step": 1120
},
{
"epoch": 9.955752212389381,
"grad_norm": 0.012359565123915672,
"learning_rate": 4.293959269863201e-09,
"loss": 0.0001,
"num_input_tokens_seen": 315800,
"step": 1125
},
{
"epoch": 10.0,
"grad_norm": 0.0031107976101338863,
"learning_rate": 1.192799661514643e-10,
"loss": 0.016,
"num_input_tokens_seen": 316840,
"step": 1130
},
{
"epoch": 10.0,
"num_input_tokens_seen": 316840,
"step": 1130,
"total_flos": 1.426716199845888e+16,
"train_loss": 0.30517576300256793,
"train_runtime": 144.7241,
"train_samples_per_second": 15.547,
"train_steps_per_second": 7.808
}
],
"logging_steps": 5,
"max_steps": 1130,
"num_input_tokens_seen": 316840,
"num_train_epochs": 10,
"save_steps": 57,
"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.426716199845888e+16,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}