| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 2.994991652754591, | |
| "eval_steps": 500, | |
| "global_step": 897, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.00333889816360601, | |
| "grad_norm": 15.957291352883223, | |
| "learning_rate": 2.2222222222222224e-07, | |
| "loss": 1.659, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.01669449081803005, | |
| "grad_norm": 20.82554498626078, | |
| "learning_rate": 1.111111111111111e-06, | |
| "loss": 1.6865, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.0333889816360601, | |
| "grad_norm": 86.6993488709915, | |
| "learning_rate": 2.222222222222222e-06, | |
| "loss": 1.6197, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.05008347245409015, | |
| "grad_norm": 14.254925959501568, | |
| "learning_rate": 3.3333333333333333e-06, | |
| "loss": 1.5486, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.0667779632721202, | |
| "grad_norm": 13.342089213956086, | |
| "learning_rate": 4.444444444444444e-06, | |
| "loss": 1.4139, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.08347245409015025, | |
| "grad_norm": 4.153939027731178, | |
| "learning_rate": 5.555555555555557e-06, | |
| "loss": 1.3362, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.1001669449081803, | |
| "grad_norm": 3.6584133983100555, | |
| "learning_rate": 6.666666666666667e-06, | |
| "loss": 1.2915, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.11686143572621036, | |
| "grad_norm": 1.7065592509714724, | |
| "learning_rate": 7.77777777777778e-06, | |
| "loss": 1.2528, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.1335559265442404, | |
| "grad_norm": 1.2120527319735446, | |
| "learning_rate": 8.888888888888888e-06, | |
| "loss": 1.1931, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.15025041736227046, | |
| "grad_norm": 0.9282324284674316, | |
| "learning_rate": 1e-05, | |
| "loss": 1.2201, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.1669449081803005, | |
| "grad_norm": 1.0951455236906156, | |
| "learning_rate": 1.1111111111111113e-05, | |
| "loss": 1.1159, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.18363939899833054, | |
| "grad_norm": 1.2930034758923397, | |
| "learning_rate": 1.2222222222222224e-05, | |
| "loss": 1.1869, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.2003338898163606, | |
| "grad_norm": 1.0326836881864259, | |
| "learning_rate": 1.3333333333333333e-05, | |
| "loss": 1.1471, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.21702838063439064, | |
| "grad_norm": 1.3171962712550565, | |
| "learning_rate": 1.4444444444444446e-05, | |
| "loss": 1.1128, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.2337228714524207, | |
| "grad_norm": 1.0884778593805544, | |
| "learning_rate": 1.555555555555556e-05, | |
| "loss": 1.1264, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.25041736227045075, | |
| "grad_norm": 0.9392194413455617, | |
| "learning_rate": 1.6666666666666667e-05, | |
| "loss": 1.0988, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.2671118530884808, | |
| "grad_norm": 1.9552115109721053, | |
| "learning_rate": 1.7777777777777777e-05, | |
| "loss": 1.1021, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.2838063439065108, | |
| "grad_norm": 3.0360726960980946, | |
| "learning_rate": 1.888888888888889e-05, | |
| "loss": 1.0973, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 0.3005008347245409, | |
| "grad_norm": 4.983883252794476, | |
| "learning_rate": 2e-05, | |
| "loss": 1.1114, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.31719532554257096, | |
| "grad_norm": 2.3180149380481923, | |
| "learning_rate": 1.9998105699049984e-05, | |
| "loss": 1.0802, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 0.333889816360601, | |
| "grad_norm": 2.319946130765325, | |
| "learning_rate": 1.9992423513875158e-05, | |
| "loss": 1.0662, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.35058430717863104, | |
| "grad_norm": 3.3442318487800233, | |
| "learning_rate": 1.9982955597229275e-05, | |
| "loss": 1.0806, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 0.3672787979966611, | |
| "grad_norm": 2.142182950971819, | |
| "learning_rate": 1.9969705536129033e-05, | |
| "loss": 1.0649, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.38397328881469117, | |
| "grad_norm": 0.6856937250318401, | |
| "learning_rate": 1.9952678350495104e-05, | |
| "loss": 1.0447, | |
| "step": 115 | |
| }, | |
| { | |
| "epoch": 0.4006677796327212, | |
| "grad_norm": 0.7829500151676976, | |
| "learning_rate": 1.9931880491250263e-05, | |
| "loss": 1.0597, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.41736227045075125, | |
| "grad_norm": 0.7638616840316438, | |
| "learning_rate": 1.990731983787542e-05, | |
| "loss": 1.0531, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 0.4340567612687813, | |
| "grad_norm": 2.040636856622385, | |
| "learning_rate": 1.987900569542438e-05, | |
| "loss": 1.0477, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.4507512520868113, | |
| "grad_norm": 2.3786229682426097, | |
| "learning_rate": 1.9846948790998532e-05, | |
| "loss": 1.0269, | |
| "step": 135 | |
| }, | |
| { | |
| "epoch": 0.4674457429048414, | |
| "grad_norm": 2.1337859327827924, | |
| "learning_rate": 1.9811161269682776e-05, | |
| "loss": 1.0433, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.48414023372287146, | |
| "grad_norm": 1.1867114123037232, | |
| "learning_rate": 1.9771656689944238e-05, | |
| "loss": 1.0553, | |
| "step": 145 | |
| }, | |
| { | |
| "epoch": 0.5008347245409015, | |
| "grad_norm": 0.7895227440440761, | |
| "learning_rate": 1.9728450018495506e-05, | |
| "loss": 1.0324, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.5175292153589316, | |
| "grad_norm": 1.0628554871553173, | |
| "learning_rate": 1.968155762462433e-05, | |
| "loss": 1.0328, | |
| "step": 155 | |
| }, | |
| { | |
| "epoch": 0.5342237061769616, | |
| "grad_norm": 0.6995453797111978, | |
| "learning_rate": 1.9630997273991964e-05, | |
| "loss": 1.0464, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.5509181969949917, | |
| "grad_norm": 0.8920498995880328, | |
| "learning_rate": 1.9576788121902457e-05, | |
| "loss": 1.0387, | |
| "step": 165 | |
| }, | |
| { | |
| "epoch": 0.5676126878130217, | |
| "grad_norm": 0.5794791237069282, | |
| "learning_rate": 1.951895070604547e-05, | |
| "loss": 1.0344, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.5843071786310517, | |
| "grad_norm": 0.812865125617091, | |
| "learning_rate": 1.9457506938715357e-05, | |
| "loss": 1.0334, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 0.6010016694490818, | |
| "grad_norm": 0.6221536202120724, | |
| "learning_rate": 1.9392480098509488e-05, | |
| "loss": 1.0174, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.6176961602671118, | |
| "grad_norm": 1.183734025560333, | |
| "learning_rate": 1.93238948215089e-05, | |
| "loss": 1.0419, | |
| "step": 185 | |
| }, | |
| { | |
| "epoch": 0.6343906510851419, | |
| "grad_norm": 0.9014792857328605, | |
| "learning_rate": 1.9251777091944665e-05, | |
| "loss": 1.0379, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.6510851419031719, | |
| "grad_norm": 0.6277333718665531, | |
| "learning_rate": 1.9176154232353513e-05, | |
| "loss": 1.048, | |
| "step": 195 | |
| }, | |
| { | |
| "epoch": 0.667779632721202, | |
| "grad_norm": 0.7852268352013753, | |
| "learning_rate": 1.9097054893226395e-05, | |
| "loss": 1.0036, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.6844741235392321, | |
| "grad_norm": 0.7644282091895362, | |
| "learning_rate": 1.9014509042153964e-05, | |
| "loss": 1.0185, | |
| "step": 205 | |
| }, | |
| { | |
| "epoch": 0.7011686143572621, | |
| "grad_norm": 1.155642192053512, | |
| "learning_rate": 1.8928547952473037e-05, | |
| "loss": 1.031, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.7178631051752922, | |
| "grad_norm": 0.8244257339118122, | |
| "learning_rate": 1.8839204191418386e-05, | |
| "loss": 0.9993, | |
| "step": 215 | |
| }, | |
| { | |
| "epoch": 0.7345575959933222, | |
| "grad_norm": 1.277466050359365, | |
| "learning_rate": 1.8746511607784298e-05, | |
| "loss": 1.0109, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.7512520868113522, | |
| "grad_norm": 1.5066052003680952, | |
| "learning_rate": 1.865050531910062e-05, | |
| "loss": 1.023, | |
| "step": 225 | |
| }, | |
| { | |
| "epoch": 0.7679465776293823, | |
| "grad_norm": 0.9113854901116754, | |
| "learning_rate": 1.855122169832813e-05, | |
| "loss": 1.0242, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.7846410684474123, | |
| "grad_norm": 1.2402339775346432, | |
| "learning_rate": 1.844869836007825e-05, | |
| "loss": 1.0286, | |
| "step": 235 | |
| }, | |
| { | |
| "epoch": 0.8013355592654424, | |
| "grad_norm": 0.6049851988239257, | |
| "learning_rate": 1.8342974146362397e-05, | |
| "loss": 1.0186, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.8180300500834724, | |
| "grad_norm": 1.3277263188043547, | |
| "learning_rate": 1.8234089111876256e-05, | |
| "loss": 0.9915, | |
| "step": 245 | |
| }, | |
| { | |
| "epoch": 0.8347245409015025, | |
| "grad_norm": 1.1833581547921075, | |
| "learning_rate": 1.8122084508824692e-05, | |
| "loss": 0.9774, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.8514190317195326, | |
| "grad_norm": 0.7893966735725576, | |
| "learning_rate": 1.80070027712929e-05, | |
| "loss": 0.9796, | |
| "step": 255 | |
| }, | |
| { | |
| "epoch": 0.8681135225375626, | |
| "grad_norm": 0.6376249393560345, | |
| "learning_rate": 1.7888887499169816e-05, | |
| "loss": 1.0042, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.8848080133555927, | |
| "grad_norm": 0.5190419036120189, | |
| "learning_rate": 1.7767783441629883e-05, | |
| "loss": 1.0154, | |
| "step": 265 | |
| }, | |
| { | |
| "epoch": 0.9015025041736227, | |
| "grad_norm": 1.1818796423028393, | |
| "learning_rate": 1.7643736480179353e-05, | |
| "loss": 1.0011, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.9181969949916527, | |
| "grad_norm": 1.1796026630847036, | |
| "learning_rate": 1.7516793611273614e-05, | |
| "loss": 1.028, | |
| "step": 275 | |
| }, | |
| { | |
| "epoch": 0.9348914858096828, | |
| "grad_norm": 1.439445951878622, | |
| "learning_rate": 1.7387002928512093e-05, | |
| "loss": 0.9978, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.9515859766277128, | |
| "grad_norm": 0.8419035609125706, | |
| "learning_rate": 1.725441360441752e-05, | |
| "loss": 0.9684, | |
| "step": 285 | |
| }, | |
| { | |
| "epoch": 0.9682804674457429, | |
| "grad_norm": 0.5556776261573866, | |
| "learning_rate": 1.711907587180642e-05, | |
| "loss": 1.0063, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.9849749582637729, | |
| "grad_norm": 0.5859264166409325, | |
| "learning_rate": 1.698104100475788e-05, | |
| "loss": 0.9699, | |
| "step": 295 | |
| }, | |
| { | |
| "epoch": 0.998330550918197, | |
| "eval_loss": 1.0447537899017334, | |
| "eval_runtime": 8.3378, | |
| "eval_samples_per_second": 229.437, | |
| "eval_steps_per_second": 7.196, | |
| "step": 299 | |
| }, | |
| { | |
| "epoch": 1.001669449081803, | |
| "grad_norm": 0.684116955981944, | |
| "learning_rate": 1.684036129918786e-05, | |
| "loss": 0.994, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 1.018363939899833, | |
| "grad_norm": 0.636525659695419, | |
| "learning_rate": 1.6697090053036344e-05, | |
| "loss": 0.9265, | |
| "step": 305 | |
| }, | |
| { | |
| "epoch": 1.0350584307178632, | |
| "grad_norm": 0.6445403739585581, | |
| "learning_rate": 1.6551281546074863e-05, | |
| "loss": 0.9375, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 1.0517529215358932, | |
| "grad_norm": 0.5637049176444268, | |
| "learning_rate": 1.6402991019342073e-05, | |
| "loss": 0.8965, | |
| "step": 315 | |
| }, | |
| { | |
| "epoch": 1.0684474123539232, | |
| "grad_norm": 1.4378688957966754, | |
| "learning_rate": 1.625227465421511e-05, | |
| "loss": 0.9485, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 1.0851419031719534, | |
| "grad_norm": 0.9003525847557393, | |
| "learning_rate": 1.60991895511247e-05, | |
| "loss": 0.9224, | |
| "step": 325 | |
| }, | |
| { | |
| "epoch": 1.1018363939899833, | |
| "grad_norm": 1.3901582427227526, | |
| "learning_rate": 1.5943793707922086e-05, | |
| "loss": 0.8996, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 1.1185308848080133, | |
| "grad_norm": 0.6502172371002246, | |
| "learning_rate": 1.5786145997905952e-05, | |
| "loss": 0.8986, | |
| "step": 335 | |
| }, | |
| { | |
| "epoch": 1.1352253756260433, | |
| "grad_norm": 0.7409388125111871, | |
| "learning_rate": 1.5626306147517665e-05, | |
| "loss": 0.9041, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 1.1519198664440735, | |
| "grad_norm": 0.7699617044708045, | |
| "learning_rate": 1.5464334713713312e-05, | |
| "loss": 0.9105, | |
| "step": 345 | |
| }, | |
| { | |
| "epoch": 1.1686143572621035, | |
| "grad_norm": 0.482173835799333, | |
| "learning_rate": 1.5300293061021084e-05, | |
| "loss": 0.9284, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 1.1853088480801335, | |
| "grad_norm": 0.6661684757723031, | |
| "learning_rate": 1.5134243338292686e-05, | |
| "loss": 0.9138, | |
| "step": 355 | |
| }, | |
| { | |
| "epoch": 1.2020033388981637, | |
| "grad_norm": 0.6181201510230905, | |
| "learning_rate": 1.4966248455157622e-05, | |
| "loss": 0.9133, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 1.2186978297161937, | |
| "grad_norm": 0.6932256442185771, | |
| "learning_rate": 1.4796372058189235e-05, | |
| "loss": 0.8791, | |
| "step": 365 | |
| }, | |
| { | |
| "epoch": 1.2353923205342237, | |
| "grad_norm": 0.8467977250421432, | |
| "learning_rate": 1.4624678506791556e-05, | |
| "loss": 0.9125, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 1.2520868113522536, | |
| "grad_norm": 0.5242049477951446, | |
| "learning_rate": 1.445123284881609e-05, | |
| "loss": 0.88, | |
| "step": 375 | |
| }, | |
| { | |
| "epoch": 1.2687813021702838, | |
| "grad_norm": 1.1383635594624812, | |
| "learning_rate": 1.4276100795917777e-05, | |
| "loss": 0.906, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 1.2854757929883138, | |
| "grad_norm": 0.7300479274157736, | |
| "learning_rate": 1.409934869865945e-05, | |
| "loss": 0.9329, | |
| "step": 385 | |
| }, | |
| { | |
| "epoch": 1.302170283806344, | |
| "grad_norm": 0.6241250631319742, | |
| "learning_rate": 1.392104352137426e-05, | |
| "loss": 0.91, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 1.318864774624374, | |
| "grad_norm": 0.5314621726495606, | |
| "learning_rate": 1.3741252816795552e-05, | |
| "loss": 0.9082, | |
| "step": 395 | |
| }, | |
| { | |
| "epoch": 1.335559265442404, | |
| "grad_norm": 0.5158076915249926, | |
| "learning_rate": 1.3560044700463824e-05, | |
| "loss": 0.9071, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 1.352253756260434, | |
| "grad_norm": 0.6440319283841749, | |
| "learning_rate": 1.3377487824920459e-05, | |
| "loss": 0.9222, | |
| "step": 405 | |
| }, | |
| { | |
| "epoch": 1.3689482470784642, | |
| "grad_norm": 0.6783380289367208, | |
| "learning_rate": 1.3193651353698012e-05, | |
| "loss": 0.9047, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 1.3856427378964942, | |
| "grad_norm": 0.5351014013897331, | |
| "learning_rate": 1.30086049351169e-05, | |
| "loss": 0.9222, | |
| "step": 415 | |
| }, | |
| { | |
| "epoch": 1.4023372287145242, | |
| "grad_norm": 0.6007683835319969, | |
| "learning_rate": 1.2822418675898428e-05, | |
| "loss": 0.9233, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 1.4190317195325544, | |
| "grad_norm": 0.6163929600226291, | |
| "learning_rate": 1.2635163114604131e-05, | |
| "loss": 0.9055, | |
| "step": 425 | |
| }, | |
| { | |
| "epoch": 1.4357262103505843, | |
| "grad_norm": 0.49834894870770663, | |
| "learning_rate": 1.2446909194911552e-05, | |
| "loss": 0.9218, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 1.4524207011686143, | |
| "grad_norm": 0.5385307548828243, | |
| "learning_rate": 1.2257728238736468e-05, | |
| "loss": 0.892, | |
| "step": 435 | |
| }, | |
| { | |
| "epoch": 1.4691151919866443, | |
| "grad_norm": 0.5517156095572195, | |
| "learning_rate": 1.2067691919211879e-05, | |
| "loss": 0.8929, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 1.4858096828046745, | |
| "grad_norm": 0.7153685327652367, | |
| "learning_rate": 1.1876872233533909e-05, | |
| "loss": 0.923, | |
| "step": 445 | |
| }, | |
| { | |
| "epoch": 1.5025041736227045, | |
| "grad_norm": 0.7676764132245686, | |
| "learning_rate": 1.1685341475684935e-05, | |
| "loss": 0.9045, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 1.5191986644407347, | |
| "grad_norm": 0.5222423769515294, | |
| "learning_rate": 1.1493172209044259e-05, | |
| "loss": 0.8796, | |
| "step": 455 | |
| }, | |
| { | |
| "epoch": 1.5358931552587647, | |
| "grad_norm": 0.6572944121034952, | |
| "learning_rate": 1.1300437238896758e-05, | |
| "loss": 0.8864, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 1.5525876460767947, | |
| "grad_norm": 0.5696483456027143, | |
| "learning_rate": 1.1107209584849845e-05, | |
| "loss": 0.9003, | |
| "step": 465 | |
| }, | |
| { | |
| "epoch": 1.5692821368948247, | |
| "grad_norm": 0.6111254138799543, | |
| "learning_rate": 1.0913562453169241e-05, | |
| "loss": 0.9148, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 1.5859766277128546, | |
| "grad_norm": 0.5948615420574401, | |
| "learning_rate": 1.0719569209044047e-05, | |
| "loss": 0.886, | |
| "step": 475 | |
| }, | |
| { | |
| "epoch": 1.6026711185308848, | |
| "grad_norm": 0.5708947368895286, | |
| "learning_rate": 1.0525303348791599e-05, | |
| "loss": 0.9145, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 1.6193656093489148, | |
| "grad_norm": 0.5425313255069921, | |
| "learning_rate": 1.0330838472012617e-05, | |
| "loss": 0.9145, | |
| "step": 485 | |
| }, | |
| { | |
| "epoch": 1.636060100166945, | |
| "grad_norm": 0.5410656168164641, | |
| "learning_rate": 1.0136248253707267e-05, | |
| "loss": 0.9009, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 1.652754590984975, | |
| "grad_norm": 0.6530203724837228, | |
| "learning_rate": 9.94160641636263e-06, | |
| "loss": 0.8822, | |
| "step": 495 | |
| }, | |
| { | |
| "epoch": 1.669449081803005, | |
| "grad_norm": 0.4919274312787223, | |
| "learning_rate": 9.74698670202218e-06, | |
| "loss": 0.9086, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 1.686143572621035, | |
| "grad_norm": 0.5238976053781923, | |
| "learning_rate": 9.552462844347883e-06, | |
| "loss": 0.9035, | |
| "step": 505 | |
| }, | |
| { | |
| "epoch": 1.702838063439065, | |
| "grad_norm": 0.523434772533601, | |
| "learning_rate": 9.358108540685406e-06, | |
| "loss": 0.8863, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 1.7195325542570952, | |
| "grad_norm": 0.5475333278586838, | |
| "learning_rate": 9.163997424143167e-06, | |
| "loss": 0.8931, | |
| "step": 515 | |
| }, | |
| { | |
| "epoch": 1.7362270450751254, | |
| "grad_norm": 0.5167332021656158, | |
| "learning_rate": 8.970203035695662e-06, | |
| "loss": 0.8994, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 1.7529215358931554, | |
| "grad_norm": 0.5326685781863387, | |
| "learning_rate": 8.776798796321715e-06, | |
| "loss": 0.8926, | |
| "step": 525 | |
| }, | |
| { | |
| "epoch": 1.7696160267111853, | |
| "grad_norm": 0.5863957550550153, | |
| "learning_rate": 8.583857979188203e-06, | |
| "loss": 0.8838, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 1.7863105175292153, | |
| "grad_norm": 0.628416420944202, | |
| "learning_rate": 8.391453681889772e-06, | |
| "loss": 0.9048, | |
| "step": 535 | |
| }, | |
| { | |
| "epoch": 1.8030050083472453, | |
| "grad_norm": 0.5843287954680072, | |
| "learning_rate": 8.199658798755048e-06, | |
| "loss": 0.9497, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 1.8196994991652755, | |
| "grad_norm": 0.45288053121542915, | |
| "learning_rate": 8.008545993229897e-06, | |
| "loss": 0.8966, | |
| "step": 545 | |
| }, | |
| { | |
| "epoch": 1.8363939899833055, | |
| "grad_norm": 0.5590037561540859, | |
| "learning_rate": 7.818187670348133e-06, | |
| "loss": 0.894, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 1.8530884808013357, | |
| "grad_norm": 0.4833538496147019, | |
| "learning_rate": 7.628655949300133e-06, | |
| "loss": 0.9073, | |
| "step": 555 | |
| }, | |
| { | |
| "epoch": 1.8697829716193657, | |
| "grad_norm": 0.5876142408674571, | |
| "learning_rate": 7.440022636109742e-06, | |
| "loss": 0.9323, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 1.8864774624373957, | |
| "grad_norm": 0.5308705244707532, | |
| "learning_rate": 7.2523591964298345e-06, | |
| "loss": 0.8729, | |
| "step": 565 | |
| }, | |
| { | |
| "epoch": 1.9031719532554257, | |
| "grad_norm": 0.5518206549230582, | |
| "learning_rate": 7.065736728466832e-06, | |
| "loss": 0.8941, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 1.9198664440734556, | |
| "grad_norm": 0.5663164888037011, | |
| "learning_rate": 6.880225936044402e-06, | |
| "loss": 0.8969, | |
| "step": 575 | |
| }, | |
| { | |
| "epoch": 1.9365609348914858, | |
| "grad_norm": 0.7716989331300791, | |
| "learning_rate": 6.695897101816606e-06, | |
| "loss": 0.889, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 1.9532554257095158, | |
| "grad_norm": 0.4854407441914746, | |
| "learning_rate": 6.512820060640608e-06, | |
| "loss": 0.9096, | |
| "step": 585 | |
| }, | |
| { | |
| "epoch": 1.969949916527546, | |
| "grad_norm": 0.5964300371902113, | |
| "learning_rate": 6.331064173119008e-06, | |
| "loss": 0.9019, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 1.986644407345576, | |
| "grad_norm": 0.6395862090435611, | |
| "learning_rate": 6.150698299321889e-06, | |
| "loss": 0.8939, | |
| "step": 595 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_loss": 1.0374501943588257, | |
| "eval_runtime": 8.2489, | |
| "eval_samples_per_second": 231.909, | |
| "eval_steps_per_second": 7.274, | |
| "step": 599 | |
| }, | |
| { | |
| "epoch": 2.003338898163606, | |
| "grad_norm": 1.1426098029417433, | |
| "learning_rate": 5.971790772698467e-06, | |
| "loss": 0.8679, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 2.020033388981636, | |
| "grad_norm": 0.5981322500928512, | |
| "learning_rate": 5.794409374188272e-06, | |
| "loss": 0.8143, | |
| "step": 605 | |
| }, | |
| { | |
| "epoch": 2.036727879799666, | |
| "grad_norm": 2.0693984258723956, | |
| "learning_rate": 5.61862130654165e-06, | |
| "loss": 0.8096, | |
| "step": 610 | |
| }, | |
| { | |
| "epoch": 2.053422370617696, | |
| "grad_norm": 0.5390665268645869, | |
| "learning_rate": 5.444493168859304e-06, | |
| "loss": 0.7971, | |
| "step": 615 | |
| }, | |
| { | |
| "epoch": 2.0701168614357264, | |
| "grad_norm": 0.5206821850461111, | |
| "learning_rate": 5.272090931360564e-06, | |
| "loss": 0.8068, | |
| "step": 620 | |
| }, | |
| { | |
| "epoch": 2.0868113522537564, | |
| "grad_norm": 0.4396188787285125, | |
| "learning_rate": 5.101479910389888e-06, | |
| "loss": 0.8178, | |
| "step": 625 | |
| }, | |
| { | |
| "epoch": 2.1035058430717863, | |
| "grad_norm": 0.48916345255937493, | |
| "learning_rate": 4.932724743671089e-06, | |
| "loss": 0.8474, | |
| "step": 630 | |
| }, | |
| { | |
| "epoch": 2.1202003338898163, | |
| "grad_norm": 0.4681540498427145, | |
| "learning_rate": 4.765889365818708e-06, | |
| "loss": 0.8171, | |
| "step": 635 | |
| }, | |
| { | |
| "epoch": 2.1368948247078463, | |
| "grad_norm": 0.46734523192627164, | |
| "learning_rate": 4.601036984115684e-06, | |
| "loss": 0.8417, | |
| "step": 640 | |
| }, | |
| { | |
| "epoch": 2.1535893155258763, | |
| "grad_norm": 0.4848107575518711, | |
| "learning_rate": 4.438230054566678e-06, | |
| "loss": 0.7979, | |
| "step": 645 | |
| }, | |
| { | |
| "epoch": 2.1702838063439067, | |
| "grad_norm": 0.5080508354482104, | |
| "learning_rate": 4.277530258235955e-06, | |
| "loss": 0.8318, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 2.1869782971619367, | |
| "grad_norm": 0.4773573845484071, | |
| "learning_rate": 4.118998477878879e-06, | |
| "loss": 0.8369, | |
| "step": 655 | |
| }, | |
| { | |
| "epoch": 2.2036727879799667, | |
| "grad_norm": 0.48182311371778996, | |
| "learning_rate": 3.96269477487588e-06, | |
| "loss": 0.858, | |
| "step": 660 | |
| }, | |
| { | |
| "epoch": 2.2203672787979967, | |
| "grad_norm": 0.465215050189049, | |
| "learning_rate": 3.8086783664775827e-06, | |
| "loss": 0.7989, | |
| "step": 665 | |
| }, | |
| { | |
| "epoch": 2.2370617696160267, | |
| "grad_norm": 0.46077400437834526, | |
| "learning_rate": 3.657007603369728e-06, | |
| "loss": 0.8292, | |
| "step": 670 | |
| }, | |
| { | |
| "epoch": 2.2537562604340566, | |
| "grad_norm": 0.4782306803989219, | |
| "learning_rate": 3.5077399475664474e-06, | |
| "loss": 0.8197, | |
| "step": 675 | |
| }, | |
| { | |
| "epoch": 2.2704507512520866, | |
| "grad_norm": 0.4829460892597705, | |
| "learning_rate": 3.360931950640185e-06, | |
| "loss": 0.8481, | |
| "step": 680 | |
| }, | |
| { | |
| "epoch": 2.287145242070117, | |
| "grad_norm": 0.44292881547818397, | |
| "learning_rate": 3.2166392322965423e-06, | |
| "loss": 0.8306, | |
| "step": 685 | |
| }, | |
| { | |
| "epoch": 2.303839732888147, | |
| "grad_norm": 0.48932957112501885, | |
| "learning_rate": 3.074916459302211e-06, | |
| "loss": 0.805, | |
| "step": 690 | |
| }, | |
| { | |
| "epoch": 2.320534223706177, | |
| "grad_norm": 0.4990000411624427, | |
| "learning_rate": 2.935817324773893e-06, | |
| "loss": 0.8189, | |
| "step": 695 | |
| }, | |
| { | |
| "epoch": 2.337228714524207, | |
| "grad_norm": 0.45360783259155707, | |
| "learning_rate": 2.799394527836129e-06, | |
| "loss": 0.8281, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 2.353923205342237, | |
| "grad_norm": 0.5557178814320654, | |
| "learning_rate": 2.665699753655684e-06, | |
| "loss": 0.8177, | |
| "step": 705 | |
| }, | |
| { | |
| "epoch": 2.370617696160267, | |
| "grad_norm": 0.6076004327005405, | |
| "learning_rate": 2.5347836538601113e-06, | |
| "loss": 0.8234, | |
| "step": 710 | |
| }, | |
| { | |
| "epoch": 2.3873121869782974, | |
| "grad_norm": 0.6588302547775924, | |
| "learning_rate": 2.406695827347848e-06, | |
| "loss": 0.8301, | |
| "step": 715 | |
| }, | |
| { | |
| "epoch": 2.4040066777963274, | |
| "grad_norm": 1.066202010754452, | |
| "learning_rate": 2.281484801497186e-06, | |
| "loss": 0.8218, | |
| "step": 720 | |
| }, | |
| { | |
| "epoch": 2.4207011686143574, | |
| "grad_norm": 0.4552773887453401, | |
| "learning_rate": 2.1591980137811684e-06, | |
| "loss": 0.8349, | |
| "step": 725 | |
| }, | |
| { | |
| "epoch": 2.4373956594323873, | |
| "grad_norm": 0.4798048974347353, | |
| "learning_rate": 2.0398817937954275e-06, | |
| "loss": 0.8291, | |
| "step": 730 | |
| }, | |
| { | |
| "epoch": 2.4540901502504173, | |
| "grad_norm": 0.49645874059938744, | |
| "learning_rate": 1.923581345705736e-06, | |
| "loss": 0.7979, | |
| "step": 735 | |
| }, | |
| { | |
| "epoch": 2.4707846410684473, | |
| "grad_norm": 0.48114179285266284, | |
| "learning_rate": 1.8103407311219523e-06, | |
| "loss": 0.8202, | |
| "step": 740 | |
| }, | |
| { | |
| "epoch": 2.4874791318864773, | |
| "grad_norm": 0.5067062797156703, | |
| "learning_rate": 1.7002028524048354e-06, | |
| "loss": 0.818, | |
| "step": 745 | |
| }, | |
| { | |
| "epoch": 2.5041736227045073, | |
| "grad_norm": 0.44105556646795846, | |
| "learning_rate": 1.5932094364120453e-06, | |
| "loss": 0.8341, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 2.5208681135225377, | |
| "grad_norm": 0.4877712957131641, | |
| "learning_rate": 1.489401018689488e-06, | |
| "loss": 0.82, | |
| "step": 755 | |
| }, | |
| { | |
| "epoch": 2.5375626043405677, | |
| "grad_norm": 0.46677823682519043, | |
| "learning_rate": 1.3888169281140284e-06, | |
| "loss": 0.8254, | |
| "step": 760 | |
| }, | |
| { | |
| "epoch": 2.5542570951585977, | |
| "grad_norm": 0.5467343146310263, | |
| "learning_rate": 1.291495271993337e-06, | |
| "loss": 0.8187, | |
| "step": 765 | |
| }, | |
| { | |
| "epoch": 2.5709515859766277, | |
| "grad_norm": 0.45191390111853974, | |
| "learning_rate": 1.1974729216285386e-06, | |
| "loss": 0.8284, | |
| "step": 770 | |
| }, | |
| { | |
| "epoch": 2.5876460767946576, | |
| "grad_norm": 0.4560656029244313, | |
| "learning_rate": 1.1067854983451575e-06, | |
| "loss": 0.8085, | |
| "step": 775 | |
| }, | |
| { | |
| "epoch": 2.604340567612688, | |
| "grad_norm": 0.4496802766749698, | |
| "learning_rate": 1.0194673599976134e-06, | |
| "loss": 0.8156, | |
| "step": 780 | |
| }, | |
| { | |
| "epoch": 2.621035058430718, | |
| "grad_norm": 0.4541891811425995, | |
| "learning_rate": 9.355515879523858e-07, | |
| "loss": 0.796, | |
| "step": 785 | |
| }, | |
| { | |
| "epoch": 2.637729549248748, | |
| "grad_norm": 0.4842807068320624, | |
| "learning_rate": 8.550699745548196e-07, | |
| "loss": 0.8402, | |
| "step": 790 | |
| }, | |
| { | |
| "epoch": 2.654424040066778, | |
| "grad_norm": 0.4825989478965968, | |
| "learning_rate": 7.780530110842566e-07, | |
| "loss": 0.8352, | |
| "step": 795 | |
| }, | |
| { | |
| "epoch": 2.671118530884808, | |
| "grad_norm": 0.431656777738719, | |
| "learning_rate": 7.045298762021125e-07, | |
| "loss": 0.8087, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 2.687813021702838, | |
| "grad_norm": 0.4425844519927156, | |
| "learning_rate": 6.345284248972383e-07, | |
| "loss": 0.8176, | |
| "step": 805 | |
| }, | |
| { | |
| "epoch": 2.704507512520868, | |
| "grad_norm": 0.43974863745670073, | |
| "learning_rate": 5.680751779327742e-07, | |
| "loss": 0.8198, | |
| "step": 810 | |
| }, | |
| { | |
| "epoch": 2.721202003338898, | |
| "grad_norm": 0.47030660258496065, | |
| "learning_rate": 5.05195311798491e-07, | |
| "loss": 0.8056, | |
| "step": 815 | |
| }, | |
| { | |
| "epoch": 2.7378964941569284, | |
| "grad_norm": 0.4413617160159506, | |
| "learning_rate": 4.4591264917242195e-07, | |
| "loss": 0.8061, | |
| "step": 820 | |
| }, | |
| { | |
| "epoch": 2.7545909849749584, | |
| "grad_norm": 0.45369770612418003, | |
| "learning_rate": 3.9024964989539227e-07, | |
| "loss": 0.8194, | |
| "step": 825 | |
| }, | |
| { | |
| "epoch": 2.7712854757929883, | |
| "grad_norm": 0.4491419061194818, | |
| "learning_rate": 3.3822740246188477e-07, | |
| "loss": 0.823, | |
| "step": 830 | |
| }, | |
| { | |
| "epoch": 2.7879799666110183, | |
| "grad_norm": 0.46281617581131723, | |
| "learning_rate": 2.8986561603044694e-07, | |
| "loss": 0.8369, | |
| "step": 835 | |
| }, | |
| { | |
| "epoch": 2.8046744574290483, | |
| "grad_norm": 0.4550289027905284, | |
| "learning_rate": 2.4518261295667255e-07, | |
| "loss": 0.8142, | |
| "step": 840 | |
| }, | |
| { | |
| "epoch": 2.8213689482470787, | |
| "grad_norm": 0.509529012047999, | |
| "learning_rate": 2.0419532185159796e-07, | |
| "loss": 0.827, | |
| "step": 845 | |
| }, | |
| { | |
| "epoch": 2.8380634390651087, | |
| "grad_norm": 0.49947769295928235, | |
| "learning_rate": 1.6691927116812002e-07, | |
| "loss": 0.8107, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 2.8547579298831387, | |
| "grad_norm": 0.4348773123366404, | |
| "learning_rate": 1.3336858331787993e-07, | |
| "loss": 0.8105, | |
| "step": 855 | |
| }, | |
| { | |
| "epoch": 2.8714524207011687, | |
| "grad_norm": 0.5120538311262153, | |
| "learning_rate": 1.0355596932085432e-07, | |
| "loss": 0.8407, | |
| "step": 860 | |
| }, | |
| { | |
| "epoch": 2.8881469115191987, | |
| "grad_norm": 0.4475268573366682, | |
| "learning_rate": 7.749272398964613e-08, | |
| "loss": 0.7998, | |
| "step": 865 | |
| }, | |
| { | |
| "epoch": 2.9048414023372287, | |
| "grad_norm": 0.5489973312590801, | |
| "learning_rate": 5.518872165033329e-08, | |
| "loss": 0.7822, | |
| "step": 870 | |
| }, | |
| { | |
| "epoch": 2.9215358931552586, | |
| "grad_norm": 0.4824018196136428, | |
| "learning_rate": 3.6652412401478875e-08, | |
| "loss": 0.8033, | |
| "step": 875 | |
| }, | |
| { | |
| "epoch": 2.9382303839732886, | |
| "grad_norm": 0.6043350861738036, | |
| "learning_rate": 2.1890818912728706e-08, | |
| "loss": 0.819, | |
| "step": 880 | |
| }, | |
| { | |
| "epoch": 2.9549248747913186, | |
| "grad_norm": 0.47649337249685186, | |
| "learning_rate": 1.0909533764194013e-08, | |
| "loss": 0.8066, | |
| "step": 885 | |
| }, | |
| { | |
| "epoch": 2.971619365609349, | |
| "grad_norm": 0.4549028191201416, | |
| "learning_rate": 3.7127173276563234e-09, | |
| "loss": 0.8075, | |
| "step": 890 | |
| }, | |
| { | |
| "epoch": 2.988313856427379, | |
| "grad_norm": 0.4521942548940854, | |
| "learning_rate": 3.0309619035495675e-10, | |
| "loss": 0.8191, | |
| "step": 895 | |
| }, | |
| { | |
| "epoch": 2.994991652754591, | |
| "eval_loss": 1.0529426336288452, | |
| "eval_runtime": 8.2524, | |
| "eval_samples_per_second": 231.812, | |
| "eval_steps_per_second": 7.271, | |
| "step": 897 | |
| }, | |
| { | |
| "epoch": 2.994991652754591, | |
| "step": 897, | |
| "total_flos": 106879455854592.0, | |
| "train_loss": 0.9405513189841008, | |
| "train_runtime": 1810.6389, | |
| "train_samples_per_second": 63.501, | |
| "train_steps_per_second": 0.495 | |
| } | |
| ], | |
| "logging_steps": 5, | |
| "max_steps": 897, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 500, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": false, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 106879455854592.0, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |