| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.9974025974025974, | |
| "eval_steps": 500, | |
| "global_step": 216, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 176.64862060546875, | |
| "learning_rate": 9.999998677875842e-06, | |
| "loss": 11.3275, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 130.30368041992188, | |
| "learning_rate": 9.999994711504062e-06, | |
| "loss": 6.8023, | |
| "step": 2 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 103.31938171386719, | |
| "learning_rate": 9.99998810088676e-06, | |
| "loss": 4.4755, | |
| "step": 3 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 74.88383483886719, | |
| "learning_rate": 9.999978846027432e-06, | |
| "loss": 2.9368, | |
| "step": 4 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 42.227046966552734, | |
| "learning_rate": 9.999966946930972e-06, | |
| "loss": 1.2319, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "grad_norm": 12.707175254821777, | |
| "learning_rate": 9.999952403603674e-06, | |
| "loss": 0.8505, | |
| "step": 6 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "grad_norm": 15.292159080505371, | |
| "learning_rate": 9.999935216053227e-06, | |
| "loss": 0.9937, | |
| "step": 7 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "grad_norm": 39.702308654785156, | |
| "learning_rate": 9.999915384288723e-06, | |
| "loss": 1.2296, | |
| "step": 8 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "grad_norm": 45.62909698486328, | |
| "learning_rate": 9.999892908320647e-06, | |
| "loss": 1.6743, | |
| "step": 9 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 44.297847747802734, | |
| "learning_rate": 9.999867788160888e-06, | |
| "loss": 1.5054, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 36.63005828857422, | |
| "learning_rate": 9.99984002382273e-06, | |
| "loss": 1.3223, | |
| "step": 11 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "grad_norm": 19.54110336303711, | |
| "learning_rate": 9.999809615320857e-06, | |
| "loss": 0.8807, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "grad_norm": 21.871042251586914, | |
| "learning_rate": 9.999776562671349e-06, | |
| "loss": 1.1686, | |
| "step": 13 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "grad_norm": 8.16606616973877, | |
| "learning_rate": 9.999740865891686e-06, | |
| "loss": 0.8842, | |
| "step": 14 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "grad_norm": 6.233722686767578, | |
| "learning_rate": 9.99970252500075e-06, | |
| "loss": 0.6474, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "grad_norm": 16.253034591674805, | |
| "learning_rate": 9.999661540018812e-06, | |
| "loss": 0.8965, | |
| "step": 16 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "grad_norm": 12.703817367553711, | |
| "learning_rate": 9.99961791096755e-06, | |
| "loss": 0.8819, | |
| "step": 17 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "grad_norm": 14.576752662658691, | |
| "learning_rate": 9.999571637870035e-06, | |
| "loss": 0.8724, | |
| "step": 18 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "grad_norm": 15.657022476196289, | |
| "learning_rate": 9.999522720750743e-06, | |
| "loss": 0.7275, | |
| "step": 19 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "grad_norm": 7.936721324920654, | |
| "learning_rate": 9.999471159635538e-06, | |
| "loss": 0.9085, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 12.384129524230957, | |
| "learning_rate": 9.999416954551693e-06, | |
| "loss": 0.5257, | |
| "step": 21 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 7.639270782470703, | |
| "learning_rate": 9.999360105527871e-06, | |
| "loss": 0.765, | |
| "step": 22 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "grad_norm": 8.124340057373047, | |
| "learning_rate": 9.999300612594139e-06, | |
| "loss": 0.6521, | |
| "step": 23 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "grad_norm": 15.96132755279541, | |
| "learning_rate": 9.999238475781957e-06, | |
| "loss": 0.8599, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "grad_norm": 6.499070167541504, | |
| "learning_rate": 9.999173695124188e-06, | |
| "loss": 0.5235, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "grad_norm": 13.837294578552246, | |
| "learning_rate": 9.999106270655093e-06, | |
| "loss": 0.7003, | |
| "step": 26 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "grad_norm": 8.020848274230957, | |
| "learning_rate": 9.999036202410324e-06, | |
| "loss": 0.7127, | |
| "step": 27 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "grad_norm": 7.126096248626709, | |
| "learning_rate": 9.998963490426943e-06, | |
| "loss": 0.6233, | |
| "step": 28 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "grad_norm": 9.910087585449219, | |
| "learning_rate": 9.998888134743398e-06, | |
| "loss": 0.6812, | |
| "step": 29 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "grad_norm": 17.2391414642334, | |
| "learning_rate": 9.998810135399545e-06, | |
| "loss": 0.6481, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "grad_norm": 18.45667266845703, | |
| "learning_rate": 9.998729492436635e-06, | |
| "loss": 0.6672, | |
| "step": 31 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "grad_norm": 22.11907196044922, | |
| "learning_rate": 9.99864620589731e-06, | |
| "loss": 0.4814, | |
| "step": 32 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "grad_norm": 11.804239273071289, | |
| "learning_rate": 9.99856027582562e-06, | |
| "loss": 0.6971, | |
| "step": 33 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "grad_norm": 6.253346920013428, | |
| "learning_rate": 9.998471702267007e-06, | |
| "loss": 0.4941, | |
| "step": 34 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "grad_norm": 9.442488670349121, | |
| "learning_rate": 9.998380485268317e-06, | |
| "loss": 0.5258, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "grad_norm": 10.5572509765625, | |
| "learning_rate": 9.998286624877786e-06, | |
| "loss": 0.5023, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "grad_norm": 11.808622360229492, | |
| "learning_rate": 9.998190121145056e-06, | |
| "loss": 0.5166, | |
| "step": 37 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "grad_norm": 5.973758697509766, | |
| "learning_rate": 9.99809097412116e-06, | |
| "loss": 0.4458, | |
| "step": 38 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "grad_norm": 7.20625114440918, | |
| "learning_rate": 9.997989183858531e-06, | |
| "loss": 0.4228, | |
| "step": 39 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "grad_norm": 13.362131118774414, | |
| "learning_rate": 9.997884750411004e-06, | |
| "loss": 0.681, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "grad_norm": 9.089489936828613, | |
| "learning_rate": 9.997777673833807e-06, | |
| "loss": 0.4588, | |
| "step": 41 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "grad_norm": 6.178775787353516, | |
| "learning_rate": 9.997667954183566e-06, | |
| "loss": 0.483, | |
| "step": 42 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 8.83432674407959, | |
| "learning_rate": 9.997555591518307e-06, | |
| "loss": 0.4187, | |
| "step": 43 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 14.531988143920898, | |
| "learning_rate": 9.997440585897455e-06, | |
| "loss": 0.4184, | |
| "step": 44 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "grad_norm": 7.901028633117676, | |
| "learning_rate": 9.997322937381829e-06, | |
| "loss": 0.7002, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "grad_norm": 6.49567985534668, | |
| "learning_rate": 9.997202646033649e-06, | |
| "loss": 0.4165, | |
| "step": 46 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "grad_norm": 32.977596282958984, | |
| "learning_rate": 9.997079711916527e-06, | |
| "loss": 0.512, | |
| "step": 47 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "grad_norm": 7.730500221252441, | |
| "learning_rate": 9.99695413509548e-06, | |
| "loss": 0.5751, | |
| "step": 48 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "grad_norm": 7.134888172149658, | |
| "learning_rate": 9.996825915636918e-06, | |
| "loss": 0.5055, | |
| "step": 49 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "grad_norm": 11.032304763793945, | |
| "learning_rate": 9.996695053608651e-06, | |
| "loss": 0.5691, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "grad_norm": 28.639476776123047, | |
| "learning_rate": 9.996561549079886e-06, | |
| "loss": 0.4184, | |
| "step": 51 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "grad_norm": 8.934990882873535, | |
| "learning_rate": 9.996425402121224e-06, | |
| "loss": 0.4129, | |
| "step": 52 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "grad_norm": 8.567286491394043, | |
| "learning_rate": 9.996286612804666e-06, | |
| "loss": 0.5257, | |
| "step": 53 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 12.532737731933594, | |
| "learning_rate": 9.996145181203616e-06, | |
| "loss": 0.4802, | |
| "step": 54 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 7.696206092834473, | |
| "learning_rate": 9.996001107392864e-06, | |
| "loss": 0.5477, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "grad_norm": 7.3067426681518555, | |
| "learning_rate": 9.995854391448607e-06, | |
| "loss": 0.4384, | |
| "step": 56 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "grad_norm": 7.36694860458374, | |
| "learning_rate": 9.995705033448435e-06, | |
| "loss": 0.483, | |
| "step": 57 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "grad_norm": 16.650466918945312, | |
| "learning_rate": 9.995553033471335e-06, | |
| "loss": 0.5578, | |
| "step": 58 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "grad_norm": 11.836296081542969, | |
| "learning_rate": 9.995398391597693e-06, | |
| "loss": 0.4419, | |
| "step": 59 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "grad_norm": 6.363118648529053, | |
| "learning_rate": 9.99524110790929e-06, | |
| "loss": 0.424, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "grad_norm": 7.7522101402282715, | |
| "learning_rate": 9.995081182489306e-06, | |
| "loss": 0.4357, | |
| "step": 61 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "grad_norm": 7.991522312164307, | |
| "learning_rate": 9.99491861542232e-06, | |
| "loss": 0.437, | |
| "step": 62 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "grad_norm": 14.417398452758789, | |
| "learning_rate": 9.994753406794303e-06, | |
| "loss": 0.6347, | |
| "step": 63 | |
| }, | |
| { | |
| "epoch": 0.3, | |
| "grad_norm": 4.412823677062988, | |
| "learning_rate": 9.994585556692624e-06, | |
| "loss": 0.2691, | |
| "step": 64 | |
| }, | |
| { | |
| "epoch": 0.3, | |
| "grad_norm": 14.479527473449707, | |
| "learning_rate": 9.994415065206053e-06, | |
| "loss": 0.4691, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.3, | |
| "grad_norm": 7.585235595703125, | |
| "learning_rate": 9.994241932424755e-06, | |
| "loss": 0.4724, | |
| "step": 66 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "grad_norm": 5.731332302093506, | |
| "learning_rate": 9.99406615844029e-06, | |
| "loss": 0.5277, | |
| "step": 67 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "grad_norm": 6.436354637145996, | |
| "learning_rate": 9.993887743345613e-06, | |
| "loss": 0.4632, | |
| "step": 68 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "grad_norm": 5.3371992111206055, | |
| "learning_rate": 9.993706687235085e-06, | |
| "loss": 0.3623, | |
| "step": 69 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "grad_norm": 10.071105003356934, | |
| "learning_rate": 9.993522990204453e-06, | |
| "loss": 0.4256, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.33, | |
| "grad_norm": 12.001503944396973, | |
| "learning_rate": 9.993336652350867e-06, | |
| "loss": 0.5697, | |
| "step": 71 | |
| }, | |
| { | |
| "epoch": 0.33, | |
| "grad_norm": 10.96662712097168, | |
| "learning_rate": 9.993147673772869e-06, | |
| "loss": 0.3635, | |
| "step": 72 | |
| }, | |
| { | |
| "epoch": 0.34, | |
| "grad_norm": 8.96445083618164, | |
| "learning_rate": 9.992956054570405e-06, | |
| "loss": 0.3633, | |
| "step": 73 | |
| }, | |
| { | |
| "epoch": 0.34, | |
| "grad_norm": 10.597489356994629, | |
| "learning_rate": 9.99276179484481e-06, | |
| "loss": 0.5179, | |
| "step": 74 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "grad_norm": 37.88457107543945, | |
| "learning_rate": 9.992564894698816e-06, | |
| "loss": 0.3815, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "grad_norm": 10.796125411987305, | |
| "learning_rate": 9.992365354236557e-06, | |
| "loss": 0.3985, | |
| "step": 76 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "grad_norm": 6.9689860343933105, | |
| "learning_rate": 9.992163173563559e-06, | |
| "loss": 0.3025, | |
| "step": 77 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "grad_norm": 9.331859588623047, | |
| "learning_rate": 9.991958352786744e-06, | |
| "loss": 0.3846, | |
| "step": 78 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "grad_norm": 8.02157211303711, | |
| "learning_rate": 9.991750892014433e-06, | |
| "loss": 0.421, | |
| "step": 79 | |
| }, | |
| { | |
| "epoch": 0.37, | |
| "grad_norm": 18.640714645385742, | |
| "learning_rate": 9.991540791356342e-06, | |
| "loss": 0.6442, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.37, | |
| "grad_norm": 7.119135856628418, | |
| "learning_rate": 9.99132805092358e-06, | |
| "loss": 0.4399, | |
| "step": 81 | |
| }, | |
| { | |
| "epoch": 0.38, | |
| "grad_norm": 9.806747436523438, | |
| "learning_rate": 9.991112670828657e-06, | |
| "loss": 0.3378, | |
| "step": 82 | |
| }, | |
| { | |
| "epoch": 0.38, | |
| "grad_norm": 7.447767734527588, | |
| "learning_rate": 9.990894651185475e-06, | |
| "loss": 0.4278, | |
| "step": 83 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "grad_norm": 28.66916847229004, | |
| "learning_rate": 9.990673992109335e-06, | |
| "loss": 0.4325, | |
| "step": 84 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "grad_norm": 13.08967399597168, | |
| "learning_rate": 9.990450693716933e-06, | |
| "loss": 0.3821, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 0.4, | |
| "grad_norm": 11.561846733093262, | |
| "learning_rate": 9.99022475612636e-06, | |
| "loss": 0.4719, | |
| "step": 86 | |
| }, | |
| { | |
| "epoch": 0.4, | |
| "grad_norm": 11.134137153625488, | |
| "learning_rate": 9.9899961794571e-06, | |
| "loss": 0.4789, | |
| "step": 87 | |
| }, | |
| { | |
| "epoch": 0.41, | |
| "grad_norm": 5.945651531219482, | |
| "learning_rate": 9.989764963830038e-06, | |
| "loss": 0.5293, | |
| "step": 88 | |
| }, | |
| { | |
| "epoch": 0.41, | |
| "grad_norm": 6.415275573730469, | |
| "learning_rate": 9.989531109367454e-06, | |
| "loss": 0.489, | |
| "step": 89 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "grad_norm": 14.867259979248047, | |
| "learning_rate": 9.989294616193018e-06, | |
| "loss": 0.7109, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "grad_norm": 18.615076065063477, | |
| "learning_rate": 9.989055484431803e-06, | |
| "loss": 0.521, | |
| "step": 91 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "grad_norm": 16.754840850830078, | |
| "learning_rate": 9.988813714210272e-06, | |
| "loss": 0.4094, | |
| "step": 92 | |
| }, | |
| { | |
| "epoch": 0.43, | |
| "grad_norm": 6.629899024963379, | |
| "learning_rate": 9.988569305656286e-06, | |
| "loss": 0.4783, | |
| "step": 93 | |
| }, | |
| { | |
| "epoch": 0.43, | |
| "grad_norm": 8.339197158813477, | |
| "learning_rate": 9.988322258899099e-06, | |
| "loss": 0.4288, | |
| "step": 94 | |
| }, | |
| { | |
| "epoch": 0.44, | |
| "grad_norm": 8.170295715332031, | |
| "learning_rate": 9.988072574069363e-06, | |
| "loss": 0.4073, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 0.44, | |
| "grad_norm": 9.301414489746094, | |
| "learning_rate": 9.987820251299121e-06, | |
| "loss": 0.5135, | |
| "step": 96 | |
| }, | |
| { | |
| "epoch": 0.45, | |
| "grad_norm": 24.828275680541992, | |
| "learning_rate": 9.98756529072182e-06, | |
| "loss": 0.5708, | |
| "step": 97 | |
| }, | |
| { | |
| "epoch": 0.45, | |
| "grad_norm": 8.960503578186035, | |
| "learning_rate": 9.987307692472288e-06, | |
| "loss": 0.4161, | |
| "step": 98 | |
| }, | |
| { | |
| "epoch": 0.46, | |
| "grad_norm": 8.696434020996094, | |
| "learning_rate": 9.98704745668676e-06, | |
| "loss": 0.4182, | |
| "step": 99 | |
| }, | |
| { | |
| "epoch": 0.46, | |
| "grad_norm": 9.047277450561523, | |
| "learning_rate": 9.986784583502863e-06, | |
| "loss": 0.3782, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.47, | |
| "grad_norm": 15.280583381652832, | |
| "learning_rate": 9.986519073059613e-06, | |
| "loss": 0.5169, | |
| "step": 101 | |
| }, | |
| { | |
| "epoch": 0.47, | |
| "grad_norm": 11.528756141662598, | |
| "learning_rate": 9.986250925497429e-06, | |
| "loss": 0.549, | |
| "step": 102 | |
| }, | |
| { | |
| "epoch": 0.48, | |
| "grad_norm": 7.328488349914551, | |
| "learning_rate": 9.98598014095812e-06, | |
| "loss": 0.4786, | |
| "step": 103 | |
| }, | |
| { | |
| "epoch": 0.48, | |
| "grad_norm": 7.289353847503662, | |
| "learning_rate": 9.985706719584888e-06, | |
| "loss": 0.3513, | |
| "step": 104 | |
| }, | |
| { | |
| "epoch": 0.48, | |
| "grad_norm": 10.800015449523926, | |
| "learning_rate": 9.985430661522333e-06, | |
| "loss": 0.4872, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 0.49, | |
| "grad_norm": 13.80408000946045, | |
| "learning_rate": 9.985151966916451e-06, | |
| "loss": 0.389, | |
| "step": 106 | |
| }, | |
| { | |
| "epoch": 0.49, | |
| "grad_norm": 13.14911937713623, | |
| "learning_rate": 9.984870635914625e-06, | |
| "loss": 0.4488, | |
| "step": 107 | |
| }, | |
| { | |
| "epoch": 0.5, | |
| "grad_norm": 8.940996170043945, | |
| "learning_rate": 9.984586668665641e-06, | |
| "loss": 0.389, | |
| "step": 108 | |
| }, | |
| { | |
| "epoch": 0.5, | |
| "grad_norm": 7.7654924392700195, | |
| "learning_rate": 9.984300065319673e-06, | |
| "loss": 0.3816, | |
| "step": 109 | |
| }, | |
| { | |
| "epoch": 0.51, | |
| "grad_norm": 6.716012954711914, | |
| "learning_rate": 9.984010826028289e-06, | |
| "loss": 0.3724, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.51, | |
| "grad_norm": 16.961824417114258, | |
| "learning_rate": 9.983718950944457e-06, | |
| "loss": 0.5116, | |
| "step": 111 | |
| }, | |
| { | |
| "epoch": 0.52, | |
| "grad_norm": 8.058645248413086, | |
| "learning_rate": 9.98342444022253e-06, | |
| "loss": 0.4868, | |
| "step": 112 | |
| }, | |
| { | |
| "epoch": 0.52, | |
| "grad_norm": 7.157467365264893, | |
| "learning_rate": 9.983127294018267e-06, | |
| "loss": 0.3658, | |
| "step": 113 | |
| }, | |
| { | |
| "epoch": 0.53, | |
| "grad_norm": 6.30387544631958, | |
| "learning_rate": 9.982827512488809e-06, | |
| "loss": 0.422, | |
| "step": 114 | |
| }, | |
| { | |
| "epoch": 0.53, | |
| "grad_norm": 5.9396748542785645, | |
| "learning_rate": 9.982525095792694e-06, | |
| "loss": 0.3137, | |
| "step": 115 | |
| }, | |
| { | |
| "epoch": 0.54, | |
| "grad_norm": 9.931533813476562, | |
| "learning_rate": 9.98222004408986e-06, | |
| "loss": 0.3283, | |
| "step": 116 | |
| }, | |
| { | |
| "epoch": 0.54, | |
| "grad_norm": 8.188447952270508, | |
| "learning_rate": 9.981912357541628e-06, | |
| "loss": 0.3709, | |
| "step": 117 | |
| }, | |
| { | |
| "epoch": 0.54, | |
| "grad_norm": 7.596179485321045, | |
| "learning_rate": 9.981602036310722e-06, | |
| "loss": 0.4277, | |
| "step": 118 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "grad_norm": 7.1218767166137695, | |
| "learning_rate": 9.981289080561253e-06, | |
| "loss": 0.338, | |
| "step": 119 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "grad_norm": 12.419051170349121, | |
| "learning_rate": 9.980973490458728e-06, | |
| "loss": 0.3293, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.56, | |
| "grad_norm": 11.043360710144043, | |
| "learning_rate": 9.980655266170049e-06, | |
| "loss": 0.4129, | |
| "step": 121 | |
| }, | |
| { | |
| "epoch": 0.56, | |
| "grad_norm": 20.372882843017578, | |
| "learning_rate": 9.980334407863506e-06, | |
| "loss": 0.4247, | |
| "step": 122 | |
| }, | |
| { | |
| "epoch": 0.57, | |
| "grad_norm": 7.637713432312012, | |
| "learning_rate": 9.980010915708785e-06, | |
| "loss": 0.3427, | |
| "step": 123 | |
| }, | |
| { | |
| "epoch": 0.57, | |
| "grad_norm": 8.155780792236328, | |
| "learning_rate": 9.979684789876965e-06, | |
| "loss": 0.3592, | |
| "step": 124 | |
| }, | |
| { | |
| "epoch": 0.58, | |
| "grad_norm": 12.149063110351562, | |
| "learning_rate": 9.97935603054052e-06, | |
| "loss": 0.3915, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 0.58, | |
| "grad_norm": 18.71912956237793, | |
| "learning_rate": 9.979024637873309e-06, | |
| "loss": 0.5225, | |
| "step": 126 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "grad_norm": 11.921587944030762, | |
| "learning_rate": 9.978690612050594e-06, | |
| "loss": 0.3912, | |
| "step": 127 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "grad_norm": 11.384403228759766, | |
| "learning_rate": 9.978353953249023e-06, | |
| "loss": 0.4185, | |
| "step": 128 | |
| }, | |
| { | |
| "epoch": 0.6, | |
| "grad_norm": 15.501189231872559, | |
| "learning_rate": 9.978014661646637e-06, | |
| "loss": 0.2584, | |
| "step": 129 | |
| }, | |
| { | |
| "epoch": 0.6, | |
| "grad_norm": 6.86824893951416, | |
| "learning_rate": 9.97767273742287e-06, | |
| "loss": 0.2435, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.6, | |
| "grad_norm": 10.46956729888916, | |
| "learning_rate": 9.97732818075855e-06, | |
| "loss": 0.2893, | |
| "step": 131 | |
| }, | |
| { | |
| "epoch": 0.61, | |
| "grad_norm": 8.8429594039917, | |
| "learning_rate": 9.976980991835896e-06, | |
| "loss": 0.3392, | |
| "step": 132 | |
| }, | |
| { | |
| "epoch": 0.61, | |
| "grad_norm": 10.744973182678223, | |
| "learning_rate": 9.976631170838516e-06, | |
| "loss": 0.381, | |
| "step": 133 | |
| }, | |
| { | |
| "epoch": 0.62, | |
| "grad_norm": 11.55877685546875, | |
| "learning_rate": 9.976278717951414e-06, | |
| "loss": 0.5731, | |
| "step": 134 | |
| }, | |
| { | |
| "epoch": 0.62, | |
| "grad_norm": 5.640347480773926, | |
| "learning_rate": 9.975923633360985e-06, | |
| "loss": 0.3342, | |
| "step": 135 | |
| }, | |
| { | |
| "epoch": 0.63, | |
| "grad_norm": 5.5956926345825195, | |
| "learning_rate": 9.975565917255017e-06, | |
| "loss": 0.257, | |
| "step": 136 | |
| }, | |
| { | |
| "epoch": 0.63, | |
| "grad_norm": 9.520169258117676, | |
| "learning_rate": 9.975205569822683e-06, | |
| "loss": 0.4152, | |
| "step": 137 | |
| }, | |
| { | |
| "epoch": 0.64, | |
| "grad_norm": 12.1378755569458, | |
| "learning_rate": 9.974842591254559e-06, | |
| "loss": 0.398, | |
| "step": 138 | |
| }, | |
| { | |
| "epoch": 0.64, | |
| "grad_norm": 16.691791534423828, | |
| "learning_rate": 9.974476981742602e-06, | |
| "loss": 0.4328, | |
| "step": 139 | |
| }, | |
| { | |
| "epoch": 0.65, | |
| "grad_norm": 6.700257778167725, | |
| "learning_rate": 9.974108741480167e-06, | |
| "loss": 0.301, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.65, | |
| "grad_norm": 7.227187156677246, | |
| "learning_rate": 9.973737870661995e-06, | |
| "loss": 0.3648, | |
| "step": 141 | |
| }, | |
| { | |
| "epoch": 0.66, | |
| "grad_norm": 7.791788578033447, | |
| "learning_rate": 9.973364369484222e-06, | |
| "loss": 0.4057, | |
| "step": 142 | |
| }, | |
| { | |
| "epoch": 0.66, | |
| "grad_norm": 12.814080238342285, | |
| "learning_rate": 9.972988238144373e-06, | |
| "loss": 0.3357, | |
| "step": 143 | |
| }, | |
| { | |
| "epoch": 0.66, | |
| "grad_norm": 6.2968316078186035, | |
| "learning_rate": 9.972609476841368e-06, | |
| "loss": 0.3597, | |
| "step": 144 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "grad_norm": 8.382381439208984, | |
| "learning_rate": 9.972228085775512e-06, | |
| "loss": 0.276, | |
| "step": 145 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "grad_norm": 13.944960594177246, | |
| "learning_rate": 9.971844065148504e-06, | |
| "loss": 0.327, | |
| "step": 146 | |
| }, | |
| { | |
| "epoch": 0.68, | |
| "grad_norm": 18.251310348510742, | |
| "learning_rate": 9.971457415163435e-06, | |
| "loss": 0.5018, | |
| "step": 147 | |
| }, | |
| { | |
| "epoch": 0.68, | |
| "grad_norm": 16.452730178833008, | |
| "learning_rate": 9.971068136024781e-06, | |
| "loss": 0.5458, | |
| "step": 148 | |
| }, | |
| { | |
| "epoch": 0.69, | |
| "grad_norm": 6.843854904174805, | |
| "learning_rate": 9.970676227938416e-06, | |
| "loss": 0.3638, | |
| "step": 149 | |
| }, | |
| { | |
| "epoch": 0.69, | |
| "grad_norm": 7.829722881317139, | |
| "learning_rate": 9.970281691111598e-06, | |
| "loss": 0.2724, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "grad_norm": 10.796073913574219, | |
| "learning_rate": 9.96988452575298e-06, | |
| "loss": 0.2902, | |
| "step": 151 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "grad_norm": 8.34335994720459, | |
| "learning_rate": 9.9694847320726e-06, | |
| "loss": 0.3067, | |
| "step": 152 | |
| }, | |
| { | |
| "epoch": 0.71, | |
| "grad_norm": 6.335442066192627, | |
| "learning_rate": 9.96908231028189e-06, | |
| "loss": 0.3904, | |
| "step": 153 | |
| }, | |
| { | |
| "epoch": 0.71, | |
| "grad_norm": 7.972692489624023, | |
| "learning_rate": 9.968677260593673e-06, | |
| "loss": 0.3659, | |
| "step": 154 | |
| }, | |
| { | |
| "epoch": 0.72, | |
| "grad_norm": 7.458805561065674, | |
| "learning_rate": 9.968269583222155e-06, | |
| "loss": 0.5201, | |
| "step": 155 | |
| }, | |
| { | |
| "epoch": 0.72, | |
| "grad_norm": 8.518513679504395, | |
| "learning_rate": 9.967859278382939e-06, | |
| "loss": 0.3055, | |
| "step": 156 | |
| }, | |
| { | |
| "epoch": 0.72, | |
| "grad_norm": 3.929286479949951, | |
| "learning_rate": 9.967446346293013e-06, | |
| "loss": 0.2564, | |
| "step": 157 | |
| }, | |
| { | |
| "epoch": 0.73, | |
| "grad_norm": 8.628119468688965, | |
| "learning_rate": 9.967030787170757e-06, | |
| "loss": 0.2978, | |
| "step": 158 | |
| }, | |
| { | |
| "epoch": 0.73, | |
| "grad_norm": 7.673137187957764, | |
| "learning_rate": 9.96661260123594e-06, | |
| "loss": 0.4875, | |
| "step": 159 | |
| }, | |
| { | |
| "epoch": 0.74, | |
| "grad_norm": 5.047264099121094, | |
| "learning_rate": 9.966191788709716e-06, | |
| "loss": 0.3381, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.74, | |
| "grad_norm": 4.241339206695557, | |
| "learning_rate": 9.965768349814635e-06, | |
| "loss": 0.2596, | |
| "step": 161 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "grad_norm": 8.526716232299805, | |
| "learning_rate": 9.965342284774633e-06, | |
| "loss": 0.2951, | |
| "step": 162 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "grad_norm": 8.26424503326416, | |
| "learning_rate": 9.964913593815032e-06, | |
| "loss": 0.4233, | |
| "step": 163 | |
| }, | |
| { | |
| "epoch": 0.76, | |
| "grad_norm": 9.364177703857422, | |
| "learning_rate": 9.964482277162547e-06, | |
| "loss": 0.4115, | |
| "step": 164 | |
| }, | |
| { | |
| "epoch": 0.76, | |
| "grad_norm": 6.392032146453857, | |
| "learning_rate": 9.964048335045276e-06, | |
| "loss": 0.2874, | |
| "step": 165 | |
| }, | |
| { | |
| "epoch": 0.77, | |
| "grad_norm": 7.744924545288086, | |
| "learning_rate": 9.963611767692714e-06, | |
| "loss": 0.3342, | |
| "step": 166 | |
| }, | |
| { | |
| "epoch": 0.77, | |
| "grad_norm": 7.376719951629639, | |
| "learning_rate": 9.963172575335735e-06, | |
| "loss": 0.3293, | |
| "step": 167 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "grad_norm": 10.256063461303711, | |
| "learning_rate": 9.962730758206612e-06, | |
| "loss": 0.3503, | |
| "step": 168 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "grad_norm": 4.126972198486328, | |
| "learning_rate": 9.962286316538992e-06, | |
| "loss": 0.1529, | |
| "step": 169 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "grad_norm": 7.471617698669434, | |
| "learning_rate": 9.961839250567925e-06, | |
| "loss": 0.3363, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.79, | |
| "grad_norm": 6.28734827041626, | |
| "learning_rate": 9.961389560529835e-06, | |
| "loss": 0.3863, | |
| "step": 171 | |
| }, | |
| { | |
| "epoch": 0.79, | |
| "grad_norm": 11.83276653289795, | |
| "learning_rate": 9.960937246662546e-06, | |
| "loss": 0.3687, | |
| "step": 172 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "grad_norm": 7.140760898590088, | |
| "learning_rate": 9.960482309205262e-06, | |
| "loss": 0.3402, | |
| "step": 173 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "grad_norm": 10.465855598449707, | |
| "learning_rate": 9.960024748398576e-06, | |
| "loss": 0.3292, | |
| "step": 174 | |
| }, | |
| { | |
| "epoch": 0.81, | |
| "grad_norm": 12.055079460144043, | |
| "learning_rate": 9.959564564484469e-06, | |
| "loss": 0.3464, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 0.81, | |
| "grad_norm": 10.145856857299805, | |
| "learning_rate": 9.959101757706308e-06, | |
| "loss": 0.2928, | |
| "step": 176 | |
| }, | |
| { | |
| "epoch": 0.82, | |
| "grad_norm": 13.143282890319824, | |
| "learning_rate": 9.958636328308852e-06, | |
| "loss": 0.3878, | |
| "step": 177 | |
| }, | |
| { | |
| "epoch": 0.82, | |
| "grad_norm": 8.187761306762695, | |
| "learning_rate": 9.95816827653824e-06, | |
| "loss": 0.4443, | |
| "step": 178 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "grad_norm": 8.577767372131348, | |
| "learning_rate": 9.957697602642002e-06, | |
| "loss": 0.5333, | |
| "step": 179 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "grad_norm": 6.963403224945068, | |
| "learning_rate": 9.957224306869053e-06, | |
| "loss": 0.3212, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.84, | |
| "grad_norm": 5.820532321929932, | |
| "learning_rate": 9.956748389469697e-06, | |
| "loss": 0.4086, | |
| "step": 181 | |
| }, | |
| { | |
| "epoch": 0.84, | |
| "grad_norm": 11.523386001586914, | |
| "learning_rate": 9.95626985069562e-06, | |
| "loss": 0.391, | |
| "step": 182 | |
| }, | |
| { | |
| "epoch": 0.85, | |
| "grad_norm": 13.058838844299316, | |
| "learning_rate": 9.9557886907999e-06, | |
| "loss": 0.438, | |
| "step": 183 | |
| }, | |
| { | |
| "epoch": 0.85, | |
| "grad_norm": 6.208215236663818, | |
| "learning_rate": 9.955304910036993e-06, | |
| "loss": 0.4455, | |
| "step": 184 | |
| }, | |
| { | |
| "epoch": 0.85, | |
| "grad_norm": 7.638975620269775, | |
| "learning_rate": 9.954818508662754e-06, | |
| "loss": 0.5551, | |
| "step": 185 | |
| }, | |
| { | |
| "epoch": 0.86, | |
| "grad_norm": 7.3545966148376465, | |
| "learning_rate": 9.954329486934411e-06, | |
| "loss": 0.3912, | |
| "step": 186 | |
| }, | |
| { | |
| "epoch": 0.86, | |
| "grad_norm": 6.2744140625, | |
| "learning_rate": 9.953837845110585e-06, | |
| "loss": 0.4615, | |
| "step": 187 | |
| }, | |
| { | |
| "epoch": 0.87, | |
| "grad_norm": 5.212122440338135, | |
| "learning_rate": 9.95334358345128e-06, | |
| "loss": 0.3151, | |
| "step": 188 | |
| }, | |
| { | |
| "epoch": 0.87, | |
| "grad_norm": 11.323850631713867, | |
| "learning_rate": 9.952846702217886e-06, | |
| "loss": 0.4894, | |
| "step": 189 | |
| }, | |
| { | |
| "epoch": 0.88, | |
| "grad_norm": 10.906036376953125, | |
| "learning_rate": 9.952347201673181e-06, | |
| "loss": 0.4583, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.88, | |
| "grad_norm": 18.138147354125977, | |
| "learning_rate": 9.95184508208132e-06, | |
| "loss": 0.5014, | |
| "step": 191 | |
| }, | |
| { | |
| "epoch": 0.89, | |
| "grad_norm": 9.336864471435547, | |
| "learning_rate": 9.951340343707852e-06, | |
| "loss": 0.4421, | |
| "step": 192 | |
| }, | |
| { | |
| "epoch": 0.89, | |
| "grad_norm": 6.270864963531494, | |
| "learning_rate": 9.95083298681971e-06, | |
| "loss": 0.2916, | |
| "step": 193 | |
| }, | |
| { | |
| "epoch": 0.9, | |
| "grad_norm": 10.805733680725098, | |
| "learning_rate": 9.950323011685206e-06, | |
| "loss": 0.4854, | |
| "step": 194 | |
| }, | |
| { | |
| "epoch": 0.9, | |
| "grad_norm": 6.747642517089844, | |
| "learning_rate": 9.94981041857404e-06, | |
| "loss": 0.2705, | |
| "step": 195 | |
| }, | |
| { | |
| "epoch": 0.91, | |
| "grad_norm": 6.886074542999268, | |
| "learning_rate": 9.9492952077573e-06, | |
| "loss": 0.2599, | |
| "step": 196 | |
| }, | |
| { | |
| "epoch": 0.91, | |
| "grad_norm": 18.723520278930664, | |
| "learning_rate": 9.948777379507453e-06, | |
| "loss": 0.5256, | |
| "step": 197 | |
| }, | |
| { | |
| "epoch": 0.91, | |
| "grad_norm": 7.4744873046875, | |
| "learning_rate": 9.948256934098353e-06, | |
| "loss": 0.3145, | |
| "step": 198 | |
| }, | |
| { | |
| "epoch": 0.92, | |
| "grad_norm": 6.738812446594238, | |
| "learning_rate": 9.947733871805235e-06, | |
| "loss": 0.324, | |
| "step": 199 | |
| }, | |
| { | |
| "epoch": 0.92, | |
| "grad_norm": 7.052412986755371, | |
| "learning_rate": 9.947208192904722e-06, | |
| "loss": 0.4534, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.93, | |
| "grad_norm": 4.8173508644104, | |
| "learning_rate": 9.946679897674823e-06, | |
| "loss": 0.372, | |
| "step": 201 | |
| }, | |
| { | |
| "epoch": 0.93, | |
| "grad_norm": 7.492809772491455, | |
| "learning_rate": 9.94614898639492e-06, | |
| "loss": 0.5313, | |
| "step": 202 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "grad_norm": 7.617343425750732, | |
| "learning_rate": 9.945615459345789e-06, | |
| "loss": 0.4958, | |
| "step": 203 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "grad_norm": 9.660953521728516, | |
| "learning_rate": 9.945079316809585e-06, | |
| "loss": 0.3639, | |
| "step": 204 | |
| }, | |
| { | |
| "epoch": 0.95, | |
| "grad_norm": 8.472088813781738, | |
| "learning_rate": 9.944540559069847e-06, | |
| "loss": 0.3534, | |
| "step": 205 | |
| }, | |
| { | |
| "epoch": 0.95, | |
| "grad_norm": 10.049768447875977, | |
| "learning_rate": 9.943999186411496e-06, | |
| "loss": 0.4788, | |
| "step": 206 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "grad_norm": 6.580361366271973, | |
| "learning_rate": 9.943455199120836e-06, | |
| "loss": 0.3854, | |
| "step": 207 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "grad_norm": 5.525043487548828, | |
| "learning_rate": 9.942908597485558e-06, | |
| "loss": 0.3237, | |
| "step": 208 | |
| }, | |
| { | |
| "epoch": 0.97, | |
| "grad_norm": 8.51529598236084, | |
| "learning_rate": 9.94235938179473e-06, | |
| "loss": 0.2758, | |
| "step": 209 | |
| }, | |
| { | |
| "epoch": 0.97, | |
| "grad_norm": 11.293745994567871, | |
| "learning_rate": 9.941807552338805e-06, | |
| "loss": 0.319, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.97, | |
| "grad_norm": 7.503701686859131, | |
| "learning_rate": 9.941253109409615e-06, | |
| "loss": 0.4216, | |
| "step": 211 | |
| }, | |
| { | |
| "epoch": 0.98, | |
| "grad_norm": 8.137164115905762, | |
| "learning_rate": 9.94069605330038e-06, | |
| "loss": 0.391, | |
| "step": 212 | |
| }, | |
| { | |
| "epoch": 0.98, | |
| "grad_norm": 5.768956661224365, | |
| "learning_rate": 9.940136384305699e-06, | |
| "loss": 0.3896, | |
| "step": 213 | |
| }, | |
| { | |
| "epoch": 0.99, | |
| "grad_norm": 6.318339824676514, | |
| "learning_rate": 9.939574102721552e-06, | |
| "loss": 0.3058, | |
| "step": 214 | |
| }, | |
| { | |
| "epoch": 0.99, | |
| "grad_norm": 5.358760356903076, | |
| "learning_rate": 9.939009208845301e-06, | |
| "loss": 0.3874, | |
| "step": 215 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "grad_norm": 7.182950973510742, | |
| "learning_rate": 9.938441702975689e-06, | |
| "loss": 0.3189, | |
| "step": 216 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_loss": 0.3517548441886902, | |
| "eval_mse": 0.35175487080665485, | |
| "eval_runtime": 13.1487, | |
| "eval_samples_per_second": 55.519, | |
| "eval_steps_per_second": 27.759, | |
| "step": 216 | |
| } | |
| ], | |
| "logging_steps": 1.0, | |
| "max_steps": 4320, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 20, | |
| "save_steps": 500, | |
| "total_flos": 1.7659591890305024e+16, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |