{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.05952380952380952, "eval_steps": 500, "global_step": 1500, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0, "grad_norm": 23.495065689086914, "learning_rate": 4.0000000000000003e-07, "loss": 8.6232, "step": 10 }, { "epoch": 0.0, "grad_norm": 24.87381935119629, "learning_rate": 9e-07, "loss": 8.4514, "step": 20 }, { "epoch": 0.0, "grad_norm": 22.63846778869629, "learning_rate": 1.4000000000000001e-06, "loss": 8.3152, "step": 30 }, { "epoch": 0.0, "grad_norm": 20.65507698059082, "learning_rate": 1.9e-06, "loss": 7.9732, "step": 40 }, { "epoch": 0.0, "grad_norm": 16.147356033325195, "learning_rate": 2.4000000000000003e-06, "loss": 7.6879, "step": 50 }, { "epoch": 0.0, "grad_norm": 14.252687454223633, "learning_rate": 2.9e-06, "loss": 7.4152, "step": 60 }, { "epoch": 0.0, "grad_norm": 12.011382102966309, "learning_rate": 3.4000000000000005e-06, "loss": 7.2472, "step": 70 }, { "epoch": 0.0, "grad_norm": 10.081536293029785, "learning_rate": 3.9e-06, "loss": 7.2433, "step": 80 }, { "epoch": 0.0, "grad_norm": 9.65556526184082, "learning_rate": 4.4e-06, "loss": 7.2298, "step": 90 }, { "epoch": 0.0, "grad_norm": 8.830799102783203, "learning_rate": 4.9000000000000005e-06, "loss": 7.19, "step": 100 }, { "epoch": 0.0, "grad_norm": 9.019427299499512, "learning_rate": 5.4e-06, "loss": 7.2362, "step": 110 }, { "epoch": 0.0, "grad_norm": 9.13768482208252, "learning_rate": 5.9e-06, "loss": 7.2804, "step": 120 }, { "epoch": 0.01, "grad_norm": 9.02840805053711, "learning_rate": 6.4000000000000006e-06, "loss": 7.2345, "step": 130 }, { "epoch": 0.01, "grad_norm": 8.178731918334961, "learning_rate": 6.900000000000001e-06, "loss": 7.3014, "step": 140 }, { "epoch": 0.01, "grad_norm": 8.83882999420166, "learning_rate": 7.4e-06, "loss": 7.3875, "step": 150 }, { "epoch": 0.01, "grad_norm": 8.3687162399292, "learning_rate": 7.9e-06, "loss": 7.4546, "step": 160 }, { "epoch": 0.01, "grad_norm": 8.09121036529541, "learning_rate": 8.400000000000001e-06, "loss": 7.407, "step": 170 }, { "epoch": 0.01, "grad_norm": 12.731450080871582, "learning_rate": 8.9e-06, "loss": 7.4321, "step": 180 }, { "epoch": 0.01, "grad_norm": 7.601162910461426, "learning_rate": 9.4e-06, "loss": 7.4155, "step": 190 }, { "epoch": 0.01, "grad_norm": 8.006213188171387, "learning_rate": 9.900000000000002e-06, "loss": 7.3705, "step": 200 }, { "epoch": 0.01, "grad_norm": 7.79794454574585, "learning_rate": 1.04e-05, "loss": 7.3759, "step": 210 }, { "epoch": 0.01, "grad_norm": 7.281346321105957, "learning_rate": 1.09e-05, "loss": 7.3798, "step": 220 }, { "epoch": 0.01, "grad_norm": 7.605266094207764, "learning_rate": 1.1400000000000001e-05, "loss": 7.3878, "step": 230 }, { "epoch": 0.01, "grad_norm": 7.446412563323975, "learning_rate": 1.19e-05, "loss": 7.3098, "step": 240 }, { "epoch": 0.01, "grad_norm": 7.6324334144592285, "learning_rate": 1.24e-05, "loss": 7.2995, "step": 250 }, { "epoch": 0.01, "grad_norm": 8.330737113952637, "learning_rate": 1.29e-05, "loss": 7.1887, "step": 260 }, { "epoch": 0.01, "grad_norm": 7.283708572387695, "learning_rate": 1.3400000000000002e-05, "loss": 7.1184, "step": 270 }, { "epoch": 0.01, "grad_norm": 6.950069904327393, "learning_rate": 1.3900000000000002e-05, "loss": 6.9876, "step": 280 }, { "epoch": 0.01, "grad_norm": 6.844958782196045, "learning_rate": 1.44e-05, "loss": 6.9319, "step": 290 }, { "epoch": 0.01, "grad_norm": 7.499362468719482, "learning_rate": 1.49e-05, "loss": 6.9253, "step": 300 }, { "epoch": 0.01, "grad_norm": 8.45572280883789, "learning_rate": 1.54e-05, "loss": 6.7999, "step": 310 }, { "epoch": 0.01, "grad_norm": 8.108625411987305, "learning_rate": 1.59e-05, "loss": 6.7667, "step": 320 }, { "epoch": 0.01, "grad_norm": 8.131596565246582, "learning_rate": 1.6400000000000002e-05, "loss": 6.7599, "step": 330 }, { "epoch": 0.01, "grad_norm": 7.2159247398376465, "learning_rate": 1.69e-05, "loss": 6.7379, "step": 340 }, { "epoch": 0.01, "grad_norm": 7.335048675537109, "learning_rate": 1.74e-05, "loss": 6.6553, "step": 350 }, { "epoch": 0.01, "grad_norm": 7.361081600189209, "learning_rate": 1.79e-05, "loss": 6.6905, "step": 360 }, { "epoch": 0.01, "grad_norm": 7.59442138671875, "learning_rate": 1.84e-05, "loss": 6.6356, "step": 370 }, { "epoch": 0.02, "grad_norm": 7.31695032119751, "learning_rate": 1.8900000000000002e-05, "loss": 6.6538, "step": 380 }, { "epoch": 0.02, "grad_norm": 7.541380882263184, "learning_rate": 1.94e-05, "loss": 6.6873, "step": 390 }, { "epoch": 0.02, "grad_norm": 8.056180000305176, "learning_rate": 1.9900000000000003e-05, "loss": 6.6332, "step": 400 }, { "epoch": 0.02, "grad_norm": 6.74859094619751, "learning_rate": 2.04e-05, "loss": 6.5488, "step": 410 }, { "epoch": 0.02, "grad_norm": 7.700525760650635, "learning_rate": 2.09e-05, "loss": 6.611, "step": 420 }, { "epoch": 0.02, "grad_norm": 7.347009181976318, "learning_rate": 2.1400000000000002e-05, "loss": 6.6621, "step": 430 }, { "epoch": 0.02, "grad_norm": 6.607337951660156, "learning_rate": 2.19e-05, "loss": 6.568, "step": 440 }, { "epoch": 0.02, "grad_norm": 6.816825866699219, "learning_rate": 2.2400000000000002e-05, "loss": 6.6293, "step": 450 }, { "epoch": 0.02, "grad_norm": 6.984443664550781, "learning_rate": 2.29e-05, "loss": 6.5667, "step": 460 }, { "epoch": 0.02, "grad_norm": 6.995021820068359, "learning_rate": 2.3400000000000003e-05, "loss": 6.5062, "step": 470 }, { "epoch": 0.02, "grad_norm": 6.876286029815674, "learning_rate": 2.39e-05, "loss": 6.5034, "step": 480 }, { "epoch": 0.02, "grad_norm": 6.953365802764893, "learning_rate": 2.44e-05, "loss": 6.459, "step": 490 }, { "epoch": 0.02, "grad_norm": 7.246520519256592, "learning_rate": 2.4900000000000002e-05, "loss": 6.4359, "step": 500 }, { "epoch": 0.02, "eval_loss": 6.133457660675049, "eval_runtime": 422.8798, "eval_samples_per_second": 50.182, "eval_steps_per_second": 0.393, "step": 500 }, { "epoch": 0.02, "grad_norm": 7.1657819747924805, "learning_rate": 2.54e-05, "loss": 6.4372, "step": 510 }, { "epoch": 0.02, "grad_norm": 7.203813552856445, "learning_rate": 2.5900000000000003e-05, "loss": 6.3684, "step": 520 }, { "epoch": 0.02, "grad_norm": 6.829228401184082, "learning_rate": 2.64e-05, "loss": 6.3275, "step": 530 }, { "epoch": 0.02, "grad_norm": 6.641738414764404, "learning_rate": 2.6900000000000003e-05, "loss": 6.2923, "step": 540 }, { "epoch": 0.02, "grad_norm": 7.824960708618164, "learning_rate": 2.7400000000000002e-05, "loss": 6.3685, "step": 550 }, { "epoch": 0.02, "grad_norm": 6.7754364013671875, "learning_rate": 2.7900000000000004e-05, "loss": 6.3069, "step": 560 }, { "epoch": 0.02, "grad_norm": 7.3608903884887695, "learning_rate": 2.84e-05, "loss": 6.2915, "step": 570 }, { "epoch": 0.02, "grad_norm": 6.551064968109131, "learning_rate": 2.8899999999999998e-05, "loss": 6.1676, "step": 580 }, { "epoch": 0.02, "grad_norm": 6.588657379150391, "learning_rate": 2.94e-05, "loss": 6.2311, "step": 590 }, { "epoch": 0.02, "grad_norm": 6.631275177001953, "learning_rate": 2.9900000000000002e-05, "loss": 6.1824, "step": 600 }, { "epoch": 0.02, "grad_norm": 6.2990288734436035, "learning_rate": 3.04e-05, "loss": 6.1641, "step": 610 }, { "epoch": 0.02, "grad_norm": 6.6439714431762695, "learning_rate": 3.09e-05, "loss": 6.0125, "step": 620 }, { "epoch": 0.03, "grad_norm": 6.402431964874268, "learning_rate": 3.1400000000000004e-05, "loss": 6.1314, "step": 630 }, { "epoch": 0.03, "grad_norm": 5.883937358856201, "learning_rate": 3.19e-05, "loss": 5.9858, "step": 640 }, { "epoch": 0.03, "grad_norm": 5.988973140716553, "learning_rate": 3.24e-05, "loss": 5.9886, "step": 650 }, { "epoch": 0.03, "grad_norm": 6.186301231384277, "learning_rate": 3.29e-05, "loss": 5.9732, "step": 660 }, { "epoch": 0.03, "grad_norm": 6.2982354164123535, "learning_rate": 3.3400000000000005e-05, "loss": 6.0391, "step": 670 }, { "epoch": 0.03, "grad_norm": 6.037730693817139, "learning_rate": 3.3900000000000004e-05, "loss": 6.0547, "step": 680 }, { "epoch": 0.03, "grad_norm": 6.280949592590332, "learning_rate": 3.4399999999999996e-05, "loss": 5.8734, "step": 690 }, { "epoch": 0.03, "grad_norm": 6.861902236938477, "learning_rate": 3.49e-05, "loss": 5.8997, "step": 700 }, { "epoch": 0.03, "grad_norm": 6.4033589363098145, "learning_rate": 3.54e-05, "loss": 5.8111, "step": 710 }, { "epoch": 0.03, "grad_norm": 6.376819133758545, "learning_rate": 3.59e-05, "loss": 5.9027, "step": 720 }, { "epoch": 0.03, "grad_norm": 5.777069568634033, "learning_rate": 3.6400000000000004e-05, "loss": 5.727, "step": 730 }, { "epoch": 0.03, "grad_norm": 6.183162689208984, "learning_rate": 3.69e-05, "loss": 5.733, "step": 740 }, { "epoch": 0.03, "grad_norm": 6.658604621887207, "learning_rate": 3.74e-05, "loss": 5.7018, "step": 750 }, { "epoch": 0.03, "grad_norm": 6.263491153717041, "learning_rate": 3.79e-05, "loss": 5.7315, "step": 760 }, { "epoch": 0.03, "grad_norm": 5.86756706237793, "learning_rate": 3.8400000000000005e-05, "loss": 5.6637, "step": 770 }, { "epoch": 0.03, "grad_norm": 6.122657299041748, "learning_rate": 3.8900000000000004e-05, "loss": 5.6849, "step": 780 }, { "epoch": 0.03, "grad_norm": 5.6024580001831055, "learning_rate": 3.94e-05, "loss": 5.5835, "step": 790 }, { "epoch": 0.03, "grad_norm": 5.619604587554932, "learning_rate": 3.99e-05, "loss": 5.7, "step": 800 }, { "epoch": 0.03, "grad_norm": 5.7342352867126465, "learning_rate": 4.0400000000000006e-05, "loss": 5.4825, "step": 810 }, { "epoch": 0.03, "grad_norm": 6.796237468719482, "learning_rate": 4.09e-05, "loss": 5.6496, "step": 820 }, { "epoch": 0.03, "grad_norm": 5.601135730743408, "learning_rate": 4.14e-05, "loss": 5.3526, "step": 830 }, { "epoch": 0.03, "grad_norm": 5.576679229736328, "learning_rate": 4.19e-05, "loss": 5.3757, "step": 840 }, { "epoch": 0.03, "grad_norm": 5.774380683898926, "learning_rate": 4.24e-05, "loss": 5.5164, "step": 850 }, { "epoch": 0.03, "grad_norm": 6.140038967132568, "learning_rate": 4.29e-05, "loss": 5.3124, "step": 860 }, { "epoch": 0.03, "grad_norm": 5.971868515014648, "learning_rate": 4.3400000000000005e-05, "loss": 5.3387, "step": 870 }, { "epoch": 0.03, "grad_norm": 6.431824207305908, "learning_rate": 4.39e-05, "loss": 5.4363, "step": 880 }, { "epoch": 0.04, "grad_norm": 6.174604892730713, "learning_rate": 4.44e-05, "loss": 5.282, "step": 890 }, { "epoch": 0.04, "grad_norm": 6.313474178314209, "learning_rate": 4.49e-05, "loss": 5.3754, "step": 900 }, { "epoch": 0.04, "grad_norm": 6.066281318664551, "learning_rate": 4.5400000000000006e-05, "loss": 5.3197, "step": 910 }, { "epoch": 0.04, "grad_norm": 7.4433817863464355, "learning_rate": 4.5900000000000004e-05, "loss": 5.3277, "step": 920 }, { "epoch": 0.04, "grad_norm": 5.884599208831787, "learning_rate": 4.64e-05, "loss": 5.0681, "step": 930 }, { "epoch": 0.04, "grad_norm": 6.17655611038208, "learning_rate": 4.69e-05, "loss": 5.1159, "step": 940 }, { "epoch": 0.04, "grad_norm": 7.090456485748291, "learning_rate": 4.74e-05, "loss": 5.2264, "step": 950 }, { "epoch": 0.04, "grad_norm": 6.508240699768066, "learning_rate": 4.79e-05, "loss": 5.2113, "step": 960 }, { "epoch": 0.04, "grad_norm": 6.5788445472717285, "learning_rate": 4.8400000000000004e-05, "loss": 4.9996, "step": 970 }, { "epoch": 0.04, "grad_norm": 5.722221374511719, "learning_rate": 4.89e-05, "loss": 4.9688, "step": 980 }, { "epoch": 0.04, "grad_norm": 5.813933849334717, "learning_rate": 4.94e-05, "loss": 4.8691, "step": 990 }, { "epoch": 0.04, "grad_norm": 6.425748348236084, "learning_rate": 4.99e-05, "loss": 4.9883, "step": 1000 }, { "epoch": 0.04, "eval_loss": 4.425854682922363, "eval_runtime": 422.6592, "eval_samples_per_second": 50.208, "eval_steps_per_second": 0.393, "step": 1000 }, { "epoch": 0.04, "grad_norm": 6.170848369598389, "learning_rate": 4.9999996764541565e-05, "loss": 4.8794, "step": 1010 }, { "epoch": 0.04, "grad_norm": 6.247808933258057, "learning_rate": 4.99999836204931e-05, "loss": 4.9664, "step": 1020 }, { "epoch": 0.04, "grad_norm": 6.27077054977417, "learning_rate": 4.9999960365643774e-05, "loss": 4.7753, "step": 1030 }, { "epoch": 0.04, "grad_norm": 6.284353733062744, "learning_rate": 4.999992700000298e-05, "loss": 4.8909, "step": 1040 }, { "epoch": 0.04, "grad_norm": 5.716318607330322, "learning_rate": 4.999988352358421e-05, "loss": 4.861, "step": 1050 }, { "epoch": 0.04, "grad_norm": 6.448729991912842, "learning_rate": 4.999982993640506e-05, "loss": 4.8776, "step": 1060 }, { "epoch": 0.04, "grad_norm": 5.401464939117432, "learning_rate": 4.9999766238487204e-05, "loss": 4.8507, "step": 1070 }, { "epoch": 0.04, "grad_norm": 6.219631195068359, "learning_rate": 4.999969242985639e-05, "loss": 4.8234, "step": 1080 }, { "epoch": 0.04, "grad_norm": 5.225848197937012, "learning_rate": 4.9999608510542485e-05, "loss": 4.6093, "step": 1090 }, { "epoch": 0.04, "grad_norm": 6.117152690887451, "learning_rate": 4.999951448057941e-05, "loss": 4.786, "step": 1100 }, { "epoch": 0.04, "grad_norm": 5.788968086242676, "learning_rate": 4.999941034000521e-05, "loss": 4.8389, "step": 1110 }, { "epoch": 0.04, "grad_norm": 6.037156581878662, "learning_rate": 4.999929608886198e-05, "loss": 4.7674, "step": 1120 }, { "epoch": 0.04, "grad_norm": 5.567563056945801, "learning_rate": 4.999917172719596e-05, "loss": 4.8247, "step": 1130 }, { "epoch": 0.05, "grad_norm": 5.557987213134766, "learning_rate": 4.999903725505742e-05, "loss": 4.7517, "step": 1140 }, { "epoch": 0.05, "grad_norm": 6.490599632263184, "learning_rate": 4.999889267250075e-05, "loss": 4.6037, "step": 1150 }, { "epoch": 0.05, "grad_norm": 6.000855922698975, "learning_rate": 4.999873797958443e-05, "loss": 4.6077, "step": 1160 }, { "epoch": 0.05, "grad_norm": 5.378462314605713, "learning_rate": 4.999857317637102e-05, "loss": 4.3662, "step": 1170 }, { "epoch": 0.05, "grad_norm": 7.4290924072265625, "learning_rate": 4.999839826292718e-05, "loss": 4.5339, "step": 1180 }, { "epoch": 0.05, "grad_norm": 5.727224826812744, "learning_rate": 4.999821323932363e-05, "loss": 4.5744, "step": 1190 }, { "epoch": 0.05, "grad_norm": 5.893082618713379, "learning_rate": 4.9998018105635214e-05, "loss": 4.5247, "step": 1200 }, { "epoch": 0.05, "grad_norm": 5.759618759155273, "learning_rate": 4.999781286194085e-05, "loss": 4.5744, "step": 1210 }, { "epoch": 0.05, "grad_norm": 5.082376003265381, "learning_rate": 4.999759750832355e-05, "loss": 4.6011, "step": 1220 }, { "epoch": 0.05, "grad_norm": 6.305749416351318, "learning_rate": 4.999737204487039e-05, "loss": 4.4587, "step": 1230 }, { "epoch": 0.05, "grad_norm": 6.710574626922607, "learning_rate": 4.999713647167258e-05, "loss": 4.46, "step": 1240 }, { "epoch": 0.05, "grad_norm": 6.753530502319336, "learning_rate": 4.999689078882537e-05, "loss": 4.5674, "step": 1250 }, { "epoch": 0.05, "grad_norm": 5.949398040771484, "learning_rate": 4.999663499642814e-05, "loss": 4.3999, "step": 1260 }, { "epoch": 0.05, "grad_norm": 6.530541896820068, "learning_rate": 4.999636909458434e-05, "loss": 4.533, "step": 1270 }, { "epoch": 0.05, "grad_norm": 5.9646453857421875, "learning_rate": 4.9996093083401495e-05, "loss": 4.374, "step": 1280 }, { "epoch": 0.05, "grad_norm": 6.346386432647705, "learning_rate": 4.9995806962991246e-05, "loss": 4.4758, "step": 1290 }, { "epoch": 0.05, "grad_norm": 5.797152519226074, "learning_rate": 4.9995510733469295e-05, "loss": 4.6275, "step": 1300 }, { "epoch": 0.05, "grad_norm": 6.426751136779785, "learning_rate": 4.999520439495547e-05, "loss": 4.3561, "step": 1310 }, { "epoch": 0.05, "grad_norm": 5.3839030265808105, "learning_rate": 4.9994887947573646e-05, "loss": 4.3134, "step": 1320 }, { "epoch": 0.05, "grad_norm": 6.01127290725708, "learning_rate": 4.9994594501953555e-05, "loss": 4.4336, "step": 1330 }, { "epoch": 0.05, "grad_norm": 5.228607654571533, "learning_rate": 4.999425884807851e-05, "loss": 4.1543, "step": 1340 }, { "epoch": 0.05, "grad_norm": 5.958486080169678, "learning_rate": 4.999391308571788e-05, "loss": 4.3806, "step": 1350 }, { "epoch": 0.05, "grad_norm": 5.816386699676514, "learning_rate": 4.99935572150115e-05, "loss": 4.3759, "step": 1360 }, { "epoch": 0.05, "grad_norm": 6.345638751983643, "learning_rate": 4.9993191236103295e-05, "loss": 4.5759, "step": 1370 }, { "epoch": 0.05, "grad_norm": 5.738880634307861, "learning_rate": 4.9992815149141294e-05, "loss": 4.3274, "step": 1380 }, { "epoch": 0.06, "grad_norm": 6.694160461425781, "learning_rate": 4.9992428954277573e-05, "loss": 4.376, "step": 1390 }, { "epoch": 0.06, "grad_norm": 5.050902843475342, "learning_rate": 4.999203265166834e-05, "loss": 4.1629, "step": 1400 }, { "epoch": 0.06, "grad_norm": 7.09380578994751, "learning_rate": 4.9991626241473876e-05, "loss": 4.3638, "step": 1410 }, { "epoch": 0.06, "grad_norm": 5.538166046142578, "learning_rate": 4.999120972385853e-05, "loss": 4.1318, "step": 1420 }, { "epoch": 0.06, "grad_norm": 6.052664279937744, "learning_rate": 4.9990783098990765e-05, "loss": 4.1856, "step": 1430 }, { "epoch": 0.06, "grad_norm": 6.707901477813721, "learning_rate": 4.9990346367043114e-05, "loss": 4.3361, "step": 1440 }, { "epoch": 0.06, "grad_norm": 5.3822150230407715, "learning_rate": 4.998989952819223e-05, "loss": 4.301, "step": 1450 }, { "epoch": 0.06, "grad_norm": 5.586170196533203, "learning_rate": 4.998944258261879e-05, "loss": 4.3131, "step": 1460 }, { "epoch": 0.06, "grad_norm": 5.62904691696167, "learning_rate": 4.998897553050763e-05, "loss": 4.3357, "step": 1470 }, { "epoch": 0.06, "grad_norm": 5.669075965881348, "learning_rate": 4.9988498372047633e-05, "loss": 4.2434, "step": 1480 }, { "epoch": 0.06, "grad_norm": 5.776549339294434, "learning_rate": 4.998801110743177e-05, "loss": 4.0441, "step": 1490 }, { "epoch": 0.06, "grad_norm": 5.442574977874756, "learning_rate": 4.9987513736857106e-05, "loss": 4.0931, "step": 1500 }, { "epoch": 0.06, "eval_loss": 3.676598310470581, "eval_runtime": 423.4536, "eval_samples_per_second": 50.114, "eval_steps_per_second": 0.392, "step": 1500 } ], "logging_steps": 10, "max_steps": 50400, "num_input_tokens_seen": 0, "num_train_epochs": 2, "save_steps": 500, "total_flos": 0.0, "train_batch_size": 8, "trial_name": null, "trial_params": null }