| { |
| "best_global_step": null, |
| "best_metric": null, |
| "best_model_checkpoint": null, |
| "epoch": 2.0, |
| "eval_steps": 500, |
| "global_step": 498, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.02009545340366742, |
| "grad_norm": 101.4629135131836, |
| "learning_rate": 4.0000000000000003e-07, |
| "loss": 2.9256, |
| "step": 5 |
| }, |
| { |
| "epoch": 0.04019090680733484, |
| "grad_norm": 83.51904296875, |
| "learning_rate": 9.000000000000001e-07, |
| "loss": 2.7459, |
| "step": 10 |
| }, |
| { |
| "epoch": 0.06028636021100226, |
| "grad_norm": 25.10388946533203, |
| "learning_rate": 1.4000000000000001e-06, |
| "loss": 1.637, |
| "step": 15 |
| }, |
| { |
| "epoch": 0.08038181361466969, |
| "grad_norm": 13.024300575256348, |
| "learning_rate": 1.9000000000000002e-06, |
| "loss": 1.1459, |
| "step": 20 |
| }, |
| { |
| "epoch": 0.1004772670183371, |
| "grad_norm": 8.315956115722656, |
| "learning_rate": 2.4000000000000003e-06, |
| "loss": 0.7456, |
| "step": 25 |
| }, |
| { |
| "epoch": 0.12057272042200452, |
| "grad_norm": 8.930346488952637, |
| "learning_rate": 2.9e-06, |
| "loss": 0.5717, |
| "step": 30 |
| }, |
| { |
| "epoch": 0.14066817382567195, |
| "grad_norm": 7.356456756591797, |
| "learning_rate": 3.4000000000000005e-06, |
| "loss": 0.414, |
| "step": 35 |
| }, |
| { |
| "epoch": 0.16076362722933937, |
| "grad_norm": 5.648008346557617, |
| "learning_rate": 3.900000000000001e-06, |
| "loss": 0.2577, |
| "step": 40 |
| }, |
| { |
| "epoch": 0.1808590806330068, |
| "grad_norm": 6.329988956451416, |
| "learning_rate": 4.4e-06, |
| "loss": 0.2394, |
| "step": 45 |
| }, |
| { |
| "epoch": 0.2009545340366742, |
| "grad_norm": 5.1738080978393555, |
| "learning_rate": 4.9000000000000005e-06, |
| "loss": 0.1848, |
| "step": 50 |
| }, |
| { |
| "epoch": 0.22104998744034163, |
| "grad_norm": 4.86478853225708, |
| "learning_rate": 4.999016565957633e-06, |
| "loss": 0.2195, |
| "step": 55 |
| }, |
| { |
| "epoch": 0.24114544084400905, |
| "grad_norm": 4.442976474761963, |
| "learning_rate": 4.995022691166621e-06, |
| "loss": 0.1582, |
| "step": 60 |
| }, |
| { |
| "epoch": 0.26124089424767644, |
| "grad_norm": 3.704345941543579, |
| "learning_rate": 4.987961816680493e-06, |
| "loss": 0.183, |
| "step": 65 |
| }, |
| { |
| "epoch": 0.2813363476513439, |
| "grad_norm": 3.772160768508911, |
| "learning_rate": 4.977842622059546e-06, |
| "loss": 0.164, |
| "step": 70 |
| }, |
| { |
| "epoch": 0.30143180105501133, |
| "grad_norm": 3.2532081604003906, |
| "learning_rate": 4.96467754629559e-06, |
| "loss": 0.1781, |
| "step": 75 |
| }, |
| { |
| "epoch": 0.32152725445867875, |
| "grad_norm": 2.893615484237671, |
| "learning_rate": 4.948482772521346e-06, |
| "loss": 0.1288, |
| "step": 80 |
| }, |
| { |
| "epoch": 0.34162270786234616, |
| "grad_norm": 4.654979705810547, |
| "learning_rate": 4.929278208117378e-06, |
| "loss": 0.1444, |
| "step": 85 |
| }, |
| { |
| "epoch": 0.3617181612660136, |
| "grad_norm": 2.8989882469177246, |
| "learning_rate": 4.907087460240999e-06, |
| "loss": 0.1151, |
| "step": 90 |
| }, |
| { |
| "epoch": 0.381813614669681, |
| "grad_norm": 3.7547988891601562, |
| "learning_rate": 4.881937806807241e-06, |
| "loss": 0.1197, |
| "step": 95 |
| }, |
| { |
| "epoch": 0.4019090680733484, |
| "grad_norm": 2.273831605911255, |
| "learning_rate": 4.8538601629575525e-06, |
| "loss": 0.0966, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.42200452147701584, |
| "grad_norm": 2.4642996788024902, |
| "learning_rate": 4.822889043057446e-06, |
| "loss": 0.1088, |
| "step": 105 |
| }, |
| { |
| "epoch": 0.44209997488068326, |
| "grad_norm": 3.29301381111145, |
| "learning_rate": 4.789062518269812e-06, |
| "loss": 0.1163, |
| "step": 110 |
| }, |
| { |
| "epoch": 0.4621954282843507, |
| "grad_norm": 2.45603084564209, |
| "learning_rate": 4.752422169756048e-06, |
| "loss": 0.1151, |
| "step": 115 |
| }, |
| { |
| "epoch": 0.4822908816880181, |
| "grad_norm": 3.199894666671753, |
| "learning_rate": 4.713013037562531e-06, |
| "loss": 0.098, |
| "step": 120 |
| }, |
| { |
| "epoch": 0.5023863350916855, |
| "grad_norm": 1.7128061056137085, |
| "learning_rate": 4.670883565255264e-06, |
| "loss": 0.0968, |
| "step": 125 |
| }, |
| { |
| "epoch": 0.5224817884953529, |
| "grad_norm": 2.645235776901245, |
| "learning_rate": 4.6260855403707625e-06, |
| "loss": 0.1068, |
| "step": 130 |
| }, |
| { |
| "epoch": 0.5425772418990203, |
| "grad_norm": 2.3140530586242676, |
| "learning_rate": 4.578674030756364e-06, |
| "loss": 0.0914, |
| "step": 135 |
| }, |
| { |
| "epoch": 0.5626726953026878, |
| "grad_norm": 1.8588067293167114, |
| "learning_rate": 4.528707316878233e-06, |
| "loss": 0.0995, |
| "step": 140 |
| }, |
| { |
| "epoch": 0.5827681487063552, |
| "grad_norm": 1.9172967672348022, |
| "learning_rate": 4.476246820180259e-06, |
| "loss": 0.08, |
| "step": 145 |
| }, |
| { |
| "epoch": 0.6028636021100227, |
| "grad_norm": 2.9082584381103516, |
| "learning_rate": 4.421357027581921e-06, |
| "loss": 0.1066, |
| "step": 150 |
| }, |
| { |
| "epoch": 0.62295905551369, |
| "grad_norm": 3.6944031715393066, |
| "learning_rate": 4.364105412207914e-06, |
| "loss": 0.1124, |
| "step": 155 |
| }, |
| { |
| "epoch": 0.6430545089173575, |
| "grad_norm": 1.5622600317001343, |
| "learning_rate": 4.304562350447008e-06, |
| "loss": 0.0856, |
| "step": 160 |
| }, |
| { |
| "epoch": 0.6631499623210249, |
| "grad_norm": 2.388239860534668, |
| "learning_rate": 4.242801035442059e-06, |
| "loss": 0.0766, |
| "step": 165 |
| }, |
| { |
| "epoch": 0.6832454157246923, |
| "grad_norm": 1.7182718515396118, |
| "learning_rate": 4.178897387117547e-06, |
| "loss": 0.104, |
| "step": 170 |
| }, |
| { |
| "epoch": 0.7033408691283597, |
| "grad_norm": 2.2367119789123535, |
| "learning_rate": 4.11292995885522e-06, |
| "loss": 0.0737, |
| "step": 175 |
| }, |
| { |
| "epoch": 0.7234363225320272, |
| "grad_norm": 2.007200241088867, |
| "learning_rate": 4.044979840932567e-06, |
| "loss": 0.0893, |
| "step": 180 |
| }, |
| { |
| "epoch": 0.7435317759356945, |
| "grad_norm": 2.6910905838012695, |
| "learning_rate": 3.975130560842821e-06, |
| "loss": 0.0944, |
| "step": 185 |
| }, |
| { |
| "epoch": 0.763627229339362, |
| "grad_norm": 2.042320489883423, |
| "learning_rate": 3.9034679806190204e-06, |
| "loss": 0.0843, |
| "step": 190 |
| }, |
| { |
| "epoch": 0.7837226827430294, |
| "grad_norm": 2.700977087020874, |
| "learning_rate": 3.830080191288342e-06, |
| "loss": 0.0669, |
| "step": 195 |
| }, |
| { |
| "epoch": 0.8038181361466968, |
| "grad_norm": 3.0307390689849854, |
| "learning_rate": 3.755057404586455e-06, |
| "loss": 0.0955, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.8239135895503642, |
| "grad_norm": 1.0048880577087402, |
| "learning_rate": 3.6784918420649952e-06, |
| "loss": 0.0977, |
| "step": 205 |
| }, |
| { |
| "epoch": 0.8440090429540317, |
| "grad_norm": 1.3049492835998535, |
| "learning_rate": 3.600477621728473e-06, |
| "loss": 0.0731, |
| "step": 210 |
| }, |
| { |
| "epoch": 0.864104496357699, |
| "grad_norm": 2.020117998123169, |
| "learning_rate": 3.521110642339991e-06, |
| "loss": 0.0939, |
| "step": 215 |
| }, |
| { |
| "epoch": 0.8841999497613665, |
| "grad_norm": 2.2311348915100098, |
| "learning_rate": 3.4404884655379433e-06, |
| "loss": 0.0747, |
| "step": 220 |
| }, |
| { |
| "epoch": 0.9042954031650339, |
| "grad_norm": 2.1661417484283447, |
| "learning_rate": 3.358710195908653e-06, |
| "loss": 0.0625, |
| "step": 225 |
| }, |
| { |
| "epoch": 0.9243908565687013, |
| "grad_norm": 2.463257074356079, |
| "learning_rate": 3.27587635916233e-06, |
| "loss": 0.1004, |
| "step": 230 |
| }, |
| { |
| "epoch": 0.9444863099723687, |
| "grad_norm": 1.5093741416931152, |
| "learning_rate": 3.1920887785621233e-06, |
| "loss": 0.1034, |
| "step": 235 |
| }, |
| { |
| "epoch": 0.9645817633760362, |
| "grad_norm": 2.1394777297973633, |
| "learning_rate": 3.10745044975816e-06, |
| "loss": 0.0815, |
| "step": 240 |
| }, |
| { |
| "epoch": 0.9846772167797035, |
| "grad_norm": 1.6689902544021606, |
| "learning_rate": 3.022065414180425e-06, |
| "loss": 0.0887, |
| "step": 245 |
| }, |
| { |
| "epoch": 1.0040190906807336, |
| "grad_norm": 1.9693433046340942, |
| "learning_rate": 2.9360386311461276e-06, |
| "loss": 0.0972, |
| "step": 250 |
| }, |
| { |
| "epoch": 1.0241145440844008, |
| "grad_norm": 0.750286340713501, |
| "learning_rate": 2.849475848838749e-06, |
| "loss": 0.0546, |
| "step": 255 |
| }, |
| { |
| "epoch": 1.0442099974880683, |
| "grad_norm": 3.802541732788086, |
| "learning_rate": 2.762483474317387e-06, |
| "loss": 0.0601, |
| "step": 260 |
| }, |
| { |
| "epoch": 1.0643054508917358, |
| "grad_norm": 1.3461735248565674, |
| "learning_rate": 2.6751684427161684e-06, |
| "loss": 0.0596, |
| "step": 265 |
| }, |
| { |
| "epoch": 1.0844009042954033, |
| "grad_norm": 2.324124813079834, |
| "learning_rate": 2.587638085794541e-06, |
| "loss": 0.0577, |
| "step": 270 |
| }, |
| { |
| "epoch": 1.1044963576990705, |
| "grad_norm": 1.4222259521484375, |
| "learning_rate": 2.5e-06, |
| "loss": 0.0527, |
| "step": 275 |
| }, |
| { |
| "epoch": 1.124591811102738, |
| "grad_norm": 0.48392441868782043, |
| "learning_rate": 2.41236191420546e-06, |
| "loss": 0.0592, |
| "step": 280 |
| }, |
| { |
| "epoch": 1.1446872645064055, |
| "grad_norm": 1.6113179922103882, |
| "learning_rate": 2.3248315572838316e-06, |
| "loss": 0.0552, |
| "step": 285 |
| }, |
| { |
| "epoch": 1.164782717910073, |
| "grad_norm": 2.1131763458251953, |
| "learning_rate": 2.237516525682614e-06, |
| "loss": 0.0601, |
| "step": 290 |
| }, |
| { |
| "epoch": 1.1848781713137402, |
| "grad_norm": 1.010076880455017, |
| "learning_rate": 2.1505241511612522e-06, |
| "loss": 0.0601, |
| "step": 295 |
| }, |
| { |
| "epoch": 1.2049736247174077, |
| "grad_norm": 1.5603766441345215, |
| "learning_rate": 2.0639613688538733e-06, |
| "loss": 0.0626, |
| "step": 300 |
| }, |
| { |
| "epoch": 1.2250690781210751, |
| "grad_norm": 0.9277430176734924, |
| "learning_rate": 1.9779345858195757e-06, |
| "loss": 0.049, |
| "step": 305 |
| }, |
| { |
| "epoch": 1.2451645315247426, |
| "grad_norm": 1.768437147140503, |
| "learning_rate": 1.8925495502418407e-06, |
| "loss": 0.055, |
| "step": 310 |
| }, |
| { |
| "epoch": 1.2652599849284099, |
| "grad_norm": 1.4469853639602661, |
| "learning_rate": 1.8079112214378769e-06, |
| "loss": 0.0608, |
| "step": 315 |
| }, |
| { |
| "epoch": 1.2853554383320773, |
| "grad_norm": 1.503761649131775, |
| "learning_rate": 1.7241236408376702e-06, |
| "loss": 0.0526, |
| "step": 320 |
| }, |
| { |
| "epoch": 1.3054508917357448, |
| "grad_norm": 1.1352767944335938, |
| "learning_rate": 1.6412898040913471e-06, |
| "loss": 0.0509, |
| "step": 325 |
| }, |
| { |
| "epoch": 1.3255463451394123, |
| "grad_norm": 2.6707780361175537, |
| "learning_rate": 1.5595115344620575e-06, |
| "loss": 0.0631, |
| "step": 330 |
| }, |
| { |
| "epoch": 1.3456417985430797, |
| "grad_norm": 1.1756007671356201, |
| "learning_rate": 1.47888935766001e-06, |
| "loss": 0.0524, |
| "step": 335 |
| }, |
| { |
| "epoch": 1.365737251946747, |
| "grad_norm": 1.0455981492996216, |
| "learning_rate": 1.399522378271527e-06, |
| "loss": 0.0507, |
| "step": 340 |
| }, |
| { |
| "epoch": 1.3858327053504145, |
| "grad_norm": 0.7072739601135254, |
| "learning_rate": 1.3215081579350058e-06, |
| "loss": 0.0487, |
| "step": 345 |
| }, |
| { |
| "epoch": 1.405928158754082, |
| "grad_norm": 1.145075798034668, |
| "learning_rate": 1.2449425954135452e-06, |
| "loss": 0.0544, |
| "step": 350 |
| }, |
| { |
| "epoch": 1.4260236121577492, |
| "grad_norm": 1.4399057626724243, |
| "learning_rate": 1.169919808711659e-06, |
| "loss": 0.0606, |
| "step": 355 |
| }, |
| { |
| "epoch": 1.4461190655614167, |
| "grad_norm": 1.0342814922332764, |
| "learning_rate": 1.0965320193809808e-06, |
| "loss": 0.0547, |
| "step": 360 |
| }, |
| { |
| "epoch": 1.4662145189650841, |
| "grad_norm": 0.5163133144378662, |
| "learning_rate": 1.02486943915718e-06, |
| "loss": 0.0534, |
| "step": 365 |
| }, |
| { |
| "epoch": 1.4863099723687516, |
| "grad_norm": 1.0042380094528198, |
| "learning_rate": 9.550201590674343e-07, |
| "loss": 0.0541, |
| "step": 370 |
| }, |
| { |
| "epoch": 1.506405425772419, |
| "grad_norm": 1.1944531202316284, |
| "learning_rate": 8.870700411447817e-07, |
| "loss": 0.0501, |
| "step": 375 |
| }, |
| { |
| "epoch": 1.5265008791760866, |
| "grad_norm": 1.3594350814819336, |
| "learning_rate": 8.21102612882454e-07, |
| "loss": 0.0611, |
| "step": 380 |
| }, |
| { |
| "epoch": 1.5465963325797538, |
| "grad_norm": 1.2450895309448242, |
| "learning_rate": 7.57198964557942e-07, |
| "loss": 0.0505, |
| "step": 385 |
| }, |
| { |
| "epoch": 1.5666917859834213, |
| "grad_norm": 1.6389065980911255, |
| "learning_rate": 6.954376495529927e-07, |
| "loss": 0.0473, |
| "step": 390 |
| }, |
| { |
| "epoch": 1.5867872393870885, |
| "grad_norm": 2.077847719192505, |
| "learning_rate": 6.358945877920861e-07, |
| "loss": 0.0519, |
| "step": 395 |
| }, |
| { |
| "epoch": 1.606882692790756, |
| "grad_norm": 1.5446641445159912, |
| "learning_rate": 5.786429724180801e-07, |
| "loss": 0.0491, |
| "step": 400 |
| }, |
| { |
| "epoch": 1.6269781461944235, |
| "grad_norm": 1.4639307260513306, |
| "learning_rate": 5.237531798197415e-07, |
| "loss": 0.0441, |
| "step": 405 |
| }, |
| { |
| "epoch": 1.647073599598091, |
| "grad_norm": 1.4568461179733276, |
| "learning_rate": 4.712926831217685e-07, |
| "loss": 0.054, |
| "step": 410 |
| }, |
| { |
| "epoch": 1.6671690530017584, |
| "grad_norm": 1.6878377199172974, |
| "learning_rate": 4.2132596924363666e-07, |
| "loss": 0.047, |
| "step": 415 |
| }, |
| { |
| "epoch": 1.687264506405426, |
| "grad_norm": 0.9864447116851807, |
| "learning_rate": 3.7391445962923824e-07, |
| "loss": 0.0503, |
| "step": 420 |
| }, |
| { |
| "epoch": 1.7073599598090932, |
| "grad_norm": 0.9351651668548584, |
| "learning_rate": 3.2911643474473647e-07, |
| "loss": 0.0456, |
| "step": 425 |
| }, |
| { |
| "epoch": 1.7274554132127606, |
| "grad_norm": 0.9683827757835388, |
| "learning_rate": 2.869869624374699e-07, |
| "loss": 0.0535, |
| "step": 430 |
| }, |
| { |
| "epoch": 1.747550866616428, |
| "grad_norm": 1.0934531688690186, |
| "learning_rate": 2.4757783024395244e-07, |
| "loss": 0.0481, |
| "step": 435 |
| }, |
| { |
| "epoch": 1.7676463200200954, |
| "grad_norm": 1.5016802549362183, |
| "learning_rate": 2.1093748173018846e-07, |
| "loss": 0.0519, |
| "step": 440 |
| }, |
| { |
| "epoch": 1.7877417734237628, |
| "grad_norm": 1.4125523567199707, |
| "learning_rate": 1.771109569425547e-07, |
| "loss": 0.0523, |
| "step": 445 |
| }, |
| { |
| "epoch": 1.8078372268274303, |
| "grad_norm": 1.1559640169143677, |
| "learning_rate": 1.4613983704244826e-07, |
| "loss": 0.0454, |
| "step": 450 |
| }, |
| { |
| "epoch": 1.8279326802310978, |
| "grad_norm": 1.0382763147354126, |
| "learning_rate": 1.1806219319275918e-07, |
| "loss": 0.046, |
| "step": 455 |
| }, |
| { |
| "epoch": 1.8480281336347653, |
| "grad_norm": 1.4664262533187866, |
| "learning_rate": 9.291253975900138e-08, |
| "loss": 0.0529, |
| "step": 460 |
| }, |
| { |
| "epoch": 1.8681235870384325, |
| "grad_norm": 0.8224850296974182, |
| "learning_rate": 7.072179188262252e-08, |
| "loss": 0.0389, |
| "step": 465 |
| }, |
| { |
| "epoch": 1.8882190404421, |
| "grad_norm": 0.5471680760383606, |
| "learning_rate": 5.151722747865434e-08, |
| "loss": 0.0489, |
| "step": 470 |
| }, |
| { |
| "epoch": 1.9083144938457672, |
| "grad_norm": 0.9190619587898254, |
| "learning_rate": 3.5322453704410286e-08, |
| "loss": 0.0438, |
| "step": 475 |
| }, |
| { |
| "epoch": 1.9284099472494347, |
| "grad_norm": 0.8270437121391296, |
| "learning_rate": 2.215737794045436e-08, |
| "loss": 0.0448, |
| "step": 480 |
| }, |
| { |
| "epoch": 1.9485054006531022, |
| "grad_norm": 0.6661733388900757, |
| "learning_rate": 1.2038183319507957e-08, |
| "loss": 0.0436, |
| "step": 485 |
| }, |
| { |
| "epoch": 1.9686008540567697, |
| "grad_norm": 0.7018892168998718, |
| "learning_rate": 4.977308833379324e-09, |
| "loss": 0.0419, |
| "step": 490 |
| }, |
| { |
| "epoch": 1.9886963074604371, |
| "grad_norm": 0.8652178645133972, |
| "learning_rate": 9.834340423678368e-10, |
| "loss": 0.048, |
| "step": 495 |
| }, |
| { |
| "epoch": 2.0, |
| "step": 498, |
| "total_flos": 1.28494219771102e+18, |
| "train_loss": 0.03479443759803312, |
| "train_runtime": 6147.788, |
| "train_samples_per_second": 5.179, |
| "train_steps_per_second": 0.081 |
| } |
| ], |
| "logging_steps": 5, |
| "max_steps": 498, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 2, |
| "save_steps": 200, |
| "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.28494219771102e+18, |
| "train_batch_size": 4, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
