| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.9988554780980127, | |
| "eval_steps": 150, | |
| "global_step": 600, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.01, | |
| "learning_rate": 5e-06, | |
| "loss": 8.3066, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "learning_rate": 1e-05, | |
| "loss": 7.9763, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "learning_rate": 1.5e-05, | |
| "loss": 7.2999, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "learning_rate": 2e-05, | |
| "loss": 6.3145, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "learning_rate": 2.5e-05, | |
| "loss": 4.8549, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "learning_rate": 3e-05, | |
| "loss": 3.7485, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "learning_rate": 3.5e-05, | |
| "loss": 3.0173, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "learning_rate": 4e-05, | |
| "loss": 2.6364, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "learning_rate": 4.5e-05, | |
| "loss": 2.4726, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "learning_rate": 5e-05, | |
| "loss": 2.3895, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "learning_rate": 5.500000000000001e-05, | |
| "loss": 2.2992, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "learning_rate": 6e-05, | |
| "loss": 2.291, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "learning_rate": 6.500000000000001e-05, | |
| "loss": 2.2597, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 7e-05, | |
| "loss": 2.2245, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 7.500000000000001e-05, | |
| "loss": 2.2076, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "learning_rate": 8e-05, | |
| "loss": 2.2132, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "learning_rate": 8.5e-05, | |
| "loss": 2.1925, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "learning_rate": 9e-05, | |
| "loss": 2.1868, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "learning_rate": 9.5e-05, | |
| "loss": 2.1916, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "learning_rate": 0.0001, | |
| "loss": 2.1774, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "learning_rate": 9.900000000000001e-05, | |
| "loss": 2.183, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "learning_rate": 9.8e-05, | |
| "loss": 2.1712, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "learning_rate": 9.7e-05, | |
| "loss": 2.1713, | |
| "step": 115 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "learning_rate": 9.6e-05, | |
| "loss": 2.1731, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "learning_rate": 9.5e-05, | |
| "loss": 2.1565, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "learning_rate": 9.4e-05, | |
| "loss": 2.1575, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "learning_rate": 9.300000000000001e-05, | |
| "loss": 2.1414, | |
| "step": 135 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "learning_rate": 9.200000000000001e-05, | |
| "loss": 2.1384, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "learning_rate": 9.1e-05, | |
| "loss": 2.1487, | |
| "step": 145 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "learning_rate": 9e-05, | |
| "loss": 2.1283, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "eval_accuracy": 0.6017748783939872, | |
| "eval_loss": 2.089164972305298, | |
| "eval_runtime": 58.1804, | |
| "eval_samples_per_second": 5.156, | |
| "eval_steps_per_second": 1.289, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "learning_rate": 8.900000000000001e-05, | |
| "loss": 2.1224, | |
| "step": 155 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "learning_rate": 8.800000000000001e-05, | |
| "loss": 2.144, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "learning_rate": 8.7e-05, | |
| "loss": 2.1293, | |
| "step": 165 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "learning_rate": 8.6e-05, | |
| "loss": 2.1417, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 8.5e-05, | |
| "loss": 2.1282, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 0.3, | |
| "learning_rate": 8.4e-05, | |
| "loss": 2.1008, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "learning_rate": 8.3e-05, | |
| "loss": 2.1218, | |
| "step": 185 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "learning_rate": 8.2e-05, | |
| "loss": 2.1292, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "learning_rate": 8.1e-05, | |
| "loss": 2.114, | |
| "step": 195 | |
| }, | |
| { | |
| "epoch": 0.33, | |
| "learning_rate": 8e-05, | |
| "loss": 2.1135, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.34, | |
| "learning_rate": 7.900000000000001e-05, | |
| "loss": 2.118, | |
| "step": 205 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "learning_rate": 7.800000000000001e-05, | |
| "loss": 2.1231, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "learning_rate": 7.7e-05, | |
| "loss": 2.0705, | |
| "step": 215 | |
| }, | |
| { | |
| "epoch": 0.37, | |
| "learning_rate": 7.6e-05, | |
| "loss": 2.11, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.37, | |
| "learning_rate": 7.500000000000001e-05, | |
| "loss": 2.0866, | |
| "step": 225 | |
| }, | |
| { | |
| "epoch": 0.38, | |
| "learning_rate": 7.4e-05, | |
| "loss": 2.1115, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "learning_rate": 7.3e-05, | |
| "loss": 2.1069, | |
| "step": 235 | |
| }, | |
| { | |
| "epoch": 0.4, | |
| "learning_rate": 7.2e-05, | |
| "loss": 2.1083, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.41, | |
| "learning_rate": 7.1e-05, | |
| "loss": 2.1014, | |
| "step": 245 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "learning_rate": 7e-05, | |
| "loss": 2.1029, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "learning_rate": 6.9e-05, | |
| "loss": 2.0846, | |
| "step": 255 | |
| }, | |
| { | |
| "epoch": 0.43, | |
| "learning_rate": 6.800000000000001e-05, | |
| "loss": 2.1059, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.44, | |
| "learning_rate": 6.7e-05, | |
| "loss": 2.0875, | |
| "step": 265 | |
| }, | |
| { | |
| "epoch": 0.45, | |
| "learning_rate": 6.6e-05, | |
| "loss": 2.0973, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.46, | |
| "learning_rate": 6.500000000000001e-05, | |
| "loss": 2.0932, | |
| "step": 275 | |
| }, | |
| { | |
| "epoch": 0.47, | |
| "learning_rate": 6.400000000000001e-05, | |
| "loss": 2.0721, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.47, | |
| "learning_rate": 6.3e-05, | |
| "loss": 2.0892, | |
| "step": 285 | |
| }, | |
| { | |
| "epoch": 0.48, | |
| "learning_rate": 6.2e-05, | |
| "loss": 2.1002, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.49, | |
| "learning_rate": 6.1e-05, | |
| "loss": 2.0804, | |
| "step": 295 | |
| }, | |
| { | |
| "epoch": 0.5, | |
| "learning_rate": 6e-05, | |
| "loss": 2.0999, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.5, | |
| "eval_accuracy": 0.6084190234842455, | |
| "eval_loss": 2.038738965988159, | |
| "eval_runtime": 68.8578, | |
| "eval_samples_per_second": 4.357, | |
| "eval_steps_per_second": 1.089, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.51, | |
| "learning_rate": 5.9e-05, | |
| "loss": 2.0905, | |
| "step": 305 | |
| }, | |
| { | |
| "epoch": 0.52, | |
| "learning_rate": 5.8e-05, | |
| "loss": 2.046, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 0.52, | |
| "learning_rate": 5.6999999999999996e-05, | |
| "loss": 2.0794, | |
| "step": 315 | |
| }, | |
| { | |
| "epoch": 0.53, | |
| "learning_rate": 5.6000000000000006e-05, | |
| "loss": 2.0714, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 0.54, | |
| "learning_rate": 5.500000000000001e-05, | |
| "loss": 2.0711, | |
| "step": 325 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "learning_rate": 5.4000000000000005e-05, | |
| "loss": 2.0525, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 0.56, | |
| "learning_rate": 5.300000000000001e-05, | |
| "loss": 2.0689, | |
| "step": 335 | |
| }, | |
| { | |
| "epoch": 0.57, | |
| "learning_rate": 5.2000000000000004e-05, | |
| "loss": 2.0936, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 0.57, | |
| "learning_rate": 5.1000000000000006e-05, | |
| "loss": 2.062, | |
| "step": 345 | |
| }, | |
| { | |
| "epoch": 0.58, | |
| "learning_rate": 5e-05, | |
| "loss": 2.0621, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "learning_rate": 4.9e-05, | |
| "loss": 2.0662, | |
| "step": 355 | |
| }, | |
| { | |
| "epoch": 0.6, | |
| "learning_rate": 4.8e-05, | |
| "loss": 2.0779, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 0.61, | |
| "learning_rate": 4.7e-05, | |
| "loss": 2.0773, | |
| "step": 365 | |
| }, | |
| { | |
| "epoch": 0.62, | |
| "learning_rate": 4.600000000000001e-05, | |
| "loss": 2.0407, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 0.62, | |
| "learning_rate": 4.5e-05, | |
| "loss": 2.0603, | |
| "step": 375 | |
| }, | |
| { | |
| "epoch": 0.63, | |
| "learning_rate": 4.4000000000000006e-05, | |
| "loss": 2.0556, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 0.64, | |
| "learning_rate": 4.3e-05, | |
| "loss": 2.0742, | |
| "step": 385 | |
| }, | |
| { | |
| "epoch": 0.65, | |
| "learning_rate": 4.2e-05, | |
| "loss": 2.0444, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 0.66, | |
| "learning_rate": 4.1e-05, | |
| "loss": 2.0409, | |
| "step": 395 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "learning_rate": 4e-05, | |
| "loss": 2.0539, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "learning_rate": 3.9000000000000006e-05, | |
| "loss": 2.0696, | |
| "step": 405 | |
| }, | |
| { | |
| "epoch": 0.68, | |
| "learning_rate": 3.8e-05, | |
| "loss": 2.0614, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 0.69, | |
| "learning_rate": 3.7e-05, | |
| "loss": 2.0604, | |
| "step": 415 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "learning_rate": 3.6e-05, | |
| "loss": 2.0608, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 0.71, | |
| "learning_rate": 3.5e-05, | |
| "loss": 2.0412, | |
| "step": 425 | |
| }, | |
| { | |
| "epoch": 0.72, | |
| "learning_rate": 3.4000000000000007e-05, | |
| "loss": 2.0558, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 0.72, | |
| "learning_rate": 3.3e-05, | |
| "loss": 2.0338, | |
| "step": 435 | |
| }, | |
| { | |
| "epoch": 0.73, | |
| "learning_rate": 3.2000000000000005e-05, | |
| "loss": 2.0322, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 0.74, | |
| "learning_rate": 3.1e-05, | |
| "loss": 2.0511, | |
| "step": 445 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "learning_rate": 3e-05, | |
| "loss": 2.0595, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "eval_accuracy": 0.6142522292024847, | |
| "eval_loss": 1.9970886707305908, | |
| "eval_runtime": 61.0863, | |
| "eval_samples_per_second": 4.911, | |
| "eval_steps_per_second": 1.228, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 0.76, | |
| "learning_rate": 2.9e-05, | |
| "loss": 2.032, | |
| "step": 455 | |
| }, | |
| { | |
| "epoch": 0.77, | |
| "learning_rate": 2.8000000000000003e-05, | |
| "loss": 2.0421, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 0.77, | |
| "learning_rate": 2.7000000000000002e-05, | |
| "loss": 2.0255, | |
| "step": 465 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "learning_rate": 2.6000000000000002e-05, | |
| "loss": 2.0259, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 0.79, | |
| "learning_rate": 2.5e-05, | |
| "loss": 2.0361, | |
| "step": 475 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "learning_rate": 2.4e-05, | |
| "loss": 2.0584, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 0.81, | |
| "learning_rate": 2.3000000000000003e-05, | |
| "loss": 2.0589, | |
| "step": 485 | |
| }, | |
| { | |
| "epoch": 0.82, | |
| "learning_rate": 2.2000000000000003e-05, | |
| "loss": 2.0555, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 0.82, | |
| "learning_rate": 2.1e-05, | |
| "loss": 2.0413, | |
| "step": 495 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "learning_rate": 2e-05, | |
| "loss": 2.0097, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.84, | |
| "learning_rate": 1.9e-05, | |
| "loss": 2.0466, | |
| "step": 505 | |
| }, | |
| { | |
| "epoch": 0.85, | |
| "learning_rate": 1.8e-05, | |
| "loss": 2.0572, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 0.86, | |
| "learning_rate": 1.7000000000000003e-05, | |
| "loss": 2.0376, | |
| "step": 515 | |
| }, | |
| { | |
| "epoch": 0.87, | |
| "learning_rate": 1.6000000000000003e-05, | |
| "loss": 2.0202, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 0.87, | |
| "learning_rate": 1.5e-05, | |
| "loss": 2.0643, | |
| "step": 525 | |
| }, | |
| { | |
| "epoch": 0.88, | |
| "learning_rate": 1.4000000000000001e-05, | |
| "loss": 2.0201, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 0.89, | |
| "learning_rate": 1.3000000000000001e-05, | |
| "loss": 2.0356, | |
| "step": 535 | |
| }, | |
| { | |
| "epoch": 0.9, | |
| "learning_rate": 1.2e-05, | |
| "loss": 2.0255, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 0.91, | |
| "learning_rate": 1.1000000000000001e-05, | |
| "loss": 2.0374, | |
| "step": 545 | |
| }, | |
| { | |
| "epoch": 0.92, | |
| "learning_rate": 1e-05, | |
| "loss": 2.0385, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 0.92, | |
| "learning_rate": 9e-06, | |
| "loss": 2.024, | |
| "step": 555 | |
| }, | |
| { | |
| "epoch": 0.93, | |
| "learning_rate": 8.000000000000001e-06, | |
| "loss": 2.0144, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "learning_rate": 7.000000000000001e-06, | |
| "loss": 2.04, | |
| "step": 565 | |
| }, | |
| { | |
| "epoch": 0.95, | |
| "learning_rate": 6e-06, | |
| "loss": 2.0372, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "learning_rate": 5e-06, | |
| "loss": 2.0122, | |
| "step": 575 | |
| }, | |
| { | |
| "epoch": 0.97, | |
| "learning_rate": 4.000000000000001e-06, | |
| "loss": 2.0182, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 0.97, | |
| "learning_rate": 3e-06, | |
| "loss": 2.0323, | |
| "step": 585 | |
| }, | |
| { | |
| "epoch": 0.98, | |
| "learning_rate": 2.0000000000000003e-06, | |
| "loss": 2.0298, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 0.99, | |
| "learning_rate": 1.0000000000000002e-06, | |
| "loss": 2.0212, | |
| "step": 595 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "learning_rate": 0.0, | |
| "loss": 2.0481, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.6152243190218628, | |
| "eval_loss": 1.9892692565917969, | |
| "eval_runtime": 64.3793, | |
| "eval_samples_per_second": 4.66, | |
| "eval_steps_per_second": 1.165, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "step": 600, | |
| "total_flos": 5.41006975991808e+16, | |
| "train_loss": 2.32403786500295, | |
| "train_runtime": 14945.8713, | |
| "train_samples_per_second": 2.572, | |
| "train_steps_per_second": 0.04 | |
| } | |
| ], | |
| "logging_steps": 5, | |
| "max_steps": 600, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 100, | |
| "total_flos": 5.41006975991808e+16, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |