| { | |
| "best_metric": 1.0948013067245483, | |
| "best_model_checkpoint": "output_dir/checkpoint-8000", | |
| "epoch": 10.0, | |
| "eval_steps": 500, | |
| "global_step": 8420, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 4.9406175771971496e-05, | |
| "loss": 2.9665, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "learning_rate": 4.8812351543942996e-05, | |
| "loss": 2.3245, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "learning_rate": 4.821852731591449e-05, | |
| "loss": 2.1171, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.48, | |
| "learning_rate": 4.7624703087885984e-05, | |
| "loss": 2.0118, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "learning_rate": 4.7030878859857484e-05, | |
| "loss": 1.9308, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "eval_loss": 1.7528491020202637, | |
| "eval_runtime": 55.8537, | |
| "eval_samples_per_second": 80.335, | |
| "eval_steps_per_second": 5.031, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.71, | |
| "learning_rate": 4.643705463182898e-05, | |
| "loss": 1.8934, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "learning_rate": 4.584323040380048e-05, | |
| "loss": 1.7962, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.95, | |
| "learning_rate": 4.524940617577197e-05, | |
| "loss": 1.7785, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 1.07, | |
| "learning_rate": 4.465558194774347e-05, | |
| "loss": 1.7253, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 1.19, | |
| "learning_rate": 4.406175771971497e-05, | |
| "loss": 1.7079, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.19, | |
| "eval_loss": 1.5516979694366455, | |
| "eval_runtime": 56.1057, | |
| "eval_samples_per_second": 79.974, | |
| "eval_steps_per_second": 5.008, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.31, | |
| "learning_rate": 4.3467933491686466e-05, | |
| "loss": 1.629, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 1.43, | |
| "learning_rate": 4.287410926365796e-05, | |
| "loss": 1.6273, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 1.54, | |
| "learning_rate": 4.228028503562946e-05, | |
| "loss": 1.5842, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 1.66, | |
| "learning_rate": 4.168646080760095e-05, | |
| "loss": 1.6089, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 1.78, | |
| "learning_rate": 4.109263657957245e-05, | |
| "loss": 1.5614, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.78, | |
| "eval_loss": 1.4558075666427612, | |
| "eval_runtime": 56.2297, | |
| "eval_samples_per_second": 79.798, | |
| "eval_steps_per_second": 4.997, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.9, | |
| "learning_rate": 4.049881235154395e-05, | |
| "loss": 1.5462, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 2.02, | |
| "learning_rate": 3.990498812351544e-05, | |
| "loss": 1.5253, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 2.14, | |
| "learning_rate": 3.9311163895486934e-05, | |
| "loss": 1.4815, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 2.26, | |
| "learning_rate": 3.8717339667458435e-05, | |
| "loss": 1.5012, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 2.38, | |
| "learning_rate": 3.812351543942993e-05, | |
| "loss": 1.4384, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 2.38, | |
| "eval_loss": 1.3825178146362305, | |
| "eval_runtime": 56.3145, | |
| "eval_samples_per_second": 79.678, | |
| "eval_steps_per_second": 4.99, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 2.49, | |
| "learning_rate": 3.752969121140142e-05, | |
| "loss": 1.4401, | |
| "step": 2100 | |
| }, | |
| { | |
| "epoch": 2.61, | |
| "learning_rate": 3.693586698337292e-05, | |
| "loss": 1.4834, | |
| "step": 2200 | |
| }, | |
| { | |
| "epoch": 2.73, | |
| "learning_rate": 3.6342042755344416e-05, | |
| "loss": 1.4534, | |
| "step": 2300 | |
| }, | |
| { | |
| "epoch": 2.85, | |
| "learning_rate": 3.5748218527315916e-05, | |
| "loss": 1.4139, | |
| "step": 2400 | |
| }, | |
| { | |
| "epoch": 2.97, | |
| "learning_rate": 3.515439429928741e-05, | |
| "loss": 1.3972, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 2.97, | |
| "eval_loss": 1.3105206489562988, | |
| "eval_runtime": 56.1399, | |
| "eval_samples_per_second": 79.925, | |
| "eval_steps_per_second": 5.005, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 3.09, | |
| "learning_rate": 3.456057007125891e-05, | |
| "loss": 1.4154, | |
| "step": 2600 | |
| }, | |
| { | |
| "epoch": 3.21, | |
| "learning_rate": 3.396674584323041e-05, | |
| "loss": 1.3644, | |
| "step": 2700 | |
| }, | |
| { | |
| "epoch": 3.33, | |
| "learning_rate": 3.3372921615201904e-05, | |
| "loss": 1.3769, | |
| "step": 2800 | |
| }, | |
| { | |
| "epoch": 3.44, | |
| "learning_rate": 3.27790973871734e-05, | |
| "loss": 1.3772, | |
| "step": 2900 | |
| }, | |
| { | |
| "epoch": 3.56, | |
| "learning_rate": 3.21852731591449e-05, | |
| "loss": 1.3532, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 3.56, | |
| "eval_loss": 1.2871623039245605, | |
| "eval_runtime": 56.4013, | |
| "eval_samples_per_second": 79.555, | |
| "eval_steps_per_second": 4.982, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 3.68, | |
| "learning_rate": 3.159144893111639e-05, | |
| "loss": 1.3614, | |
| "step": 3100 | |
| }, | |
| { | |
| "epoch": 3.8, | |
| "learning_rate": 3.0997624703087885e-05, | |
| "loss": 1.3531, | |
| "step": 3200 | |
| }, | |
| { | |
| "epoch": 3.92, | |
| "learning_rate": 3.0403800475059386e-05, | |
| "loss": 1.3365, | |
| "step": 3300 | |
| }, | |
| { | |
| "epoch": 4.04, | |
| "learning_rate": 2.980997624703088e-05, | |
| "loss": 1.3075, | |
| "step": 3400 | |
| }, | |
| { | |
| "epoch": 4.16, | |
| "learning_rate": 2.9216152019002373e-05, | |
| "loss": 1.3114, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 4.16, | |
| "eval_loss": 1.2454777956008911, | |
| "eval_runtime": 56.1952, | |
| "eval_samples_per_second": 79.847, | |
| "eval_steps_per_second": 5.0, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 4.28, | |
| "learning_rate": 2.8622327790973873e-05, | |
| "loss": 1.2952, | |
| "step": 3600 | |
| }, | |
| { | |
| "epoch": 4.39, | |
| "learning_rate": 2.8028503562945367e-05, | |
| "loss": 1.3133, | |
| "step": 3700 | |
| }, | |
| { | |
| "epoch": 4.51, | |
| "learning_rate": 2.7434679334916867e-05, | |
| "loss": 1.3225, | |
| "step": 3800 | |
| }, | |
| { | |
| "epoch": 4.63, | |
| "learning_rate": 2.6840855106888364e-05, | |
| "loss": 1.2958, | |
| "step": 3900 | |
| }, | |
| { | |
| "epoch": 4.75, | |
| "learning_rate": 2.6247030878859858e-05, | |
| "loss": 1.269, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 4.75, | |
| "eval_loss": 1.2113255262374878, | |
| "eval_runtime": 57.2608, | |
| "eval_samples_per_second": 78.361, | |
| "eval_steps_per_second": 4.907, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 4.87, | |
| "learning_rate": 2.5653206650831358e-05, | |
| "loss": 1.2693, | |
| "step": 4100 | |
| }, | |
| { | |
| "epoch": 4.99, | |
| "learning_rate": 2.5059382422802852e-05, | |
| "loss": 1.2646, | |
| "step": 4200 | |
| }, | |
| { | |
| "epoch": 5.11, | |
| "learning_rate": 2.446555819477435e-05, | |
| "loss": 1.2702, | |
| "step": 4300 | |
| }, | |
| { | |
| "epoch": 5.23, | |
| "learning_rate": 2.3871733966745842e-05, | |
| "loss": 1.2568, | |
| "step": 4400 | |
| }, | |
| { | |
| "epoch": 5.34, | |
| "learning_rate": 2.327790973871734e-05, | |
| "loss": 1.2525, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 5.34, | |
| "eval_loss": 1.1863404512405396, | |
| "eval_runtime": 56.357, | |
| "eval_samples_per_second": 79.617, | |
| "eval_steps_per_second": 4.986, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 5.46, | |
| "learning_rate": 2.2684085510688836e-05, | |
| "loss": 1.2343, | |
| "step": 4600 | |
| }, | |
| { | |
| "epoch": 5.58, | |
| "learning_rate": 2.2090261282660333e-05, | |
| "loss": 1.2289, | |
| "step": 4700 | |
| }, | |
| { | |
| "epoch": 5.7, | |
| "learning_rate": 2.149643705463183e-05, | |
| "loss": 1.2091, | |
| "step": 4800 | |
| }, | |
| { | |
| "epoch": 5.82, | |
| "learning_rate": 2.0902612826603327e-05, | |
| "loss": 1.2231, | |
| "step": 4900 | |
| }, | |
| { | |
| "epoch": 5.94, | |
| "learning_rate": 2.0308788598574824e-05, | |
| "loss": 1.2171, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 5.94, | |
| "eval_loss": 1.1596688032150269, | |
| "eval_runtime": 56.2736, | |
| "eval_samples_per_second": 79.735, | |
| "eval_steps_per_second": 4.993, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 6.06, | |
| "learning_rate": 1.9714964370546318e-05, | |
| "loss": 1.1989, | |
| "step": 5100 | |
| }, | |
| { | |
| "epoch": 6.18, | |
| "learning_rate": 1.9121140142517815e-05, | |
| "loss": 1.2272, | |
| "step": 5200 | |
| }, | |
| { | |
| "epoch": 6.29, | |
| "learning_rate": 1.8527315914489312e-05, | |
| "loss": 1.2426, | |
| "step": 5300 | |
| }, | |
| { | |
| "epoch": 6.41, | |
| "learning_rate": 1.793349168646081e-05, | |
| "loss": 1.1808, | |
| "step": 5400 | |
| }, | |
| { | |
| "epoch": 6.53, | |
| "learning_rate": 1.7339667458432306e-05, | |
| "loss": 1.1669, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 6.53, | |
| "eval_loss": 1.141239047050476, | |
| "eval_runtime": 56.1754, | |
| "eval_samples_per_second": 79.875, | |
| "eval_steps_per_second": 5.002, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 6.65, | |
| "learning_rate": 1.6745843230403803e-05, | |
| "loss": 1.2044, | |
| "step": 5600 | |
| }, | |
| { | |
| "epoch": 6.77, | |
| "learning_rate": 1.61520190023753e-05, | |
| "loss": 1.1781, | |
| "step": 5700 | |
| }, | |
| { | |
| "epoch": 6.89, | |
| "learning_rate": 1.5558194774346793e-05, | |
| "loss": 1.1757, | |
| "step": 5800 | |
| }, | |
| { | |
| "epoch": 7.01, | |
| "learning_rate": 1.496437054631829e-05, | |
| "loss": 1.1777, | |
| "step": 5900 | |
| }, | |
| { | |
| "epoch": 7.13, | |
| "learning_rate": 1.4370546318289787e-05, | |
| "loss": 1.1529, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 7.13, | |
| "eval_loss": 1.1352198123931885, | |
| "eval_runtime": 56.8461, | |
| "eval_samples_per_second": 78.932, | |
| "eval_steps_per_second": 4.943, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 7.24, | |
| "learning_rate": 1.3776722090261283e-05, | |
| "loss": 1.1854, | |
| "step": 6100 | |
| }, | |
| { | |
| "epoch": 7.36, | |
| "learning_rate": 1.318289786223278e-05, | |
| "loss": 1.1736, | |
| "step": 6200 | |
| }, | |
| { | |
| "epoch": 7.48, | |
| "learning_rate": 1.2589073634204277e-05, | |
| "loss": 1.1932, | |
| "step": 6300 | |
| }, | |
| { | |
| "epoch": 7.6, | |
| "learning_rate": 1.1995249406175772e-05, | |
| "loss": 1.1431, | |
| "step": 6400 | |
| }, | |
| { | |
| "epoch": 7.72, | |
| "learning_rate": 1.1401425178147269e-05, | |
| "loss": 1.1532, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 7.72, | |
| "eval_loss": 1.1115353107452393, | |
| "eval_runtime": 56.4012, | |
| "eval_samples_per_second": 79.555, | |
| "eval_steps_per_second": 4.982, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 7.84, | |
| "learning_rate": 1.0807600950118766e-05, | |
| "loss": 1.1762, | |
| "step": 6600 | |
| }, | |
| { | |
| "epoch": 7.96, | |
| "learning_rate": 1.0213776722090261e-05, | |
| "loss": 1.1327, | |
| "step": 6700 | |
| }, | |
| { | |
| "epoch": 8.08, | |
| "learning_rate": 9.619952494061758e-06, | |
| "loss": 1.1492, | |
| "step": 6800 | |
| }, | |
| { | |
| "epoch": 8.19, | |
| "learning_rate": 9.026128266033253e-06, | |
| "loss": 1.1214, | |
| "step": 6900 | |
| }, | |
| { | |
| "epoch": 8.31, | |
| "learning_rate": 8.432304038004752e-06, | |
| "loss": 1.1684, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 8.31, | |
| "eval_loss": 1.1077940464019775, | |
| "eval_runtime": 56.2764, | |
| "eval_samples_per_second": 79.731, | |
| "eval_steps_per_second": 4.993, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 8.43, | |
| "learning_rate": 7.838479809976247e-06, | |
| "loss": 1.154, | |
| "step": 7100 | |
| }, | |
| { | |
| "epoch": 8.55, | |
| "learning_rate": 7.244655581947744e-06, | |
| "loss": 1.1439, | |
| "step": 7200 | |
| }, | |
| { | |
| "epoch": 8.67, | |
| "learning_rate": 6.6508313539192404e-06, | |
| "loss": 1.1398, | |
| "step": 7300 | |
| }, | |
| { | |
| "epoch": 8.79, | |
| "learning_rate": 6.0570071258907366e-06, | |
| "loss": 1.117, | |
| "step": 7400 | |
| }, | |
| { | |
| "epoch": 8.91, | |
| "learning_rate": 5.4631828978622335e-06, | |
| "loss": 1.1286, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 8.91, | |
| "eval_loss": 1.0950287580490112, | |
| "eval_runtime": 56.0134, | |
| "eval_samples_per_second": 80.106, | |
| "eval_steps_per_second": 5.017, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 9.03, | |
| "learning_rate": 4.86935866983373e-06, | |
| "loss": 1.1485, | |
| "step": 7600 | |
| }, | |
| { | |
| "epoch": 9.14, | |
| "learning_rate": 4.275534441805226e-06, | |
| "loss": 1.1273, | |
| "step": 7700 | |
| }, | |
| { | |
| "epoch": 9.26, | |
| "learning_rate": 3.681710213776722e-06, | |
| "loss": 1.1412, | |
| "step": 7800 | |
| }, | |
| { | |
| "epoch": 9.38, | |
| "learning_rate": 3.0878859857482185e-06, | |
| "loss": 1.1206, | |
| "step": 7900 | |
| }, | |
| { | |
| "epoch": 9.5, | |
| "learning_rate": 2.494061757719715e-06, | |
| "loss": 1.1416, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 9.5, | |
| "eval_loss": 1.0948013067245483, | |
| "eval_runtime": 56.75, | |
| "eval_samples_per_second": 79.066, | |
| "eval_steps_per_second": 4.952, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 9.62, | |
| "learning_rate": 1.9002375296912116e-06, | |
| "loss": 1.1101, | |
| "step": 8100 | |
| }, | |
| { | |
| "epoch": 9.74, | |
| "learning_rate": 1.306413301662708e-06, | |
| "loss": 1.1181, | |
| "step": 8200 | |
| }, | |
| { | |
| "epoch": 9.86, | |
| "learning_rate": 7.125890736342043e-07, | |
| "loss": 1.0831, | |
| "step": 8300 | |
| }, | |
| { | |
| "epoch": 9.98, | |
| "learning_rate": 1.1876484560570073e-07, | |
| "loss": 1.1164, | |
| "step": 8400 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "step": 8420, | |
| "total_flos": 3.5427566278656e+16, | |
| "train_loss": 1.365048002347244, | |
| "train_runtime": 5770.015, | |
| "train_samples_per_second": 23.327, | |
| "train_steps_per_second": 1.459 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_loss": 1.0815315246582031, | |
| "eval_runtime": 57.2783, | |
| "eval_samples_per_second": 78.337, | |
| "eval_steps_per_second": 4.906, | |
| "step": 8420 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_loss": 1.0882647037506104, | |
| "eval_runtime": 56.9424, | |
| "eval_samples_per_second": 78.799, | |
| "eval_steps_per_second": 4.935, | |
| "step": 8420 | |
| } | |
| ], | |
| "logging_steps": 100, | |
| "max_steps": 8420, | |
| "num_train_epochs": 10, | |
| "save_steps": 500, | |
| "total_flos": 3.5427566278656e+16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |