| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.03968253968253968, | |
| "eval_steps": 500, | |
| "global_step": 1000, | |
| "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 | |
| } | |
| ], | |
| "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 | |
| } | |