{ "best_metric": 1.4312853813171387, "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.2196, "step": 100 }, { "epoch": 0.24, "learning_rate": 4.8812351543942996e-05, "loss": 2.1188, "step": 200 }, { "epoch": 0.36, "learning_rate": 4.821852731591449e-05, "loss": 2.0304, "step": 300 }, { "epoch": 0.48, "learning_rate": 4.7624703087885984e-05, "loss": 2.0009, "step": 400 }, { "epoch": 0.59, "learning_rate": 4.7030878859857484e-05, "loss": 1.9328, "step": 500 }, { "epoch": 0.59, "eval_loss": 1.9144912958145142, "eval_runtime": 50.3099, "eval_samples_per_second": 89.187, "eval_steps_per_second": 5.585, "step": 500 }, { "epoch": 0.71, "learning_rate": 4.643705463182898e-05, "loss": 1.9414, "step": 600 }, { "epoch": 0.83, "learning_rate": 4.584323040380048e-05, "loss": 1.8604, "step": 700 }, { "epoch": 0.95, "learning_rate": 4.524940617577197e-05, "loss": 1.8739, "step": 800 }, { "epoch": 1.07, "learning_rate": 4.465558194774347e-05, "loss": 1.8085, "step": 900 }, { "epoch": 1.19, "learning_rate": 4.406175771971497e-05, "loss": 1.863, "step": 1000 }, { "epoch": 1.19, "eval_loss": 1.7881555557250977, "eval_runtime": 50.4256, "eval_samples_per_second": 88.983, "eval_steps_per_second": 5.573, "step": 1000 }, { "epoch": 1.31, "learning_rate": 4.3467933491686466e-05, "loss": 1.7697, "step": 1100 }, { "epoch": 1.43, "learning_rate": 4.287410926365796e-05, "loss": 1.7683, "step": 1200 }, { "epoch": 1.54, "learning_rate": 4.228028503562946e-05, "loss": 1.7315, "step": 1300 }, { "epoch": 1.66, "learning_rate": 4.168646080760095e-05, "loss": 1.7758, "step": 1400 }, { "epoch": 1.78, "learning_rate": 4.109263657957245e-05, "loss": 1.7407, "step": 1500 }, { "epoch": 1.78, "eval_loss": 1.7240532636642456, "eval_runtime": 49.8395, "eval_samples_per_second": 90.029, "eval_steps_per_second": 5.638, "step": 1500 }, { "epoch": 1.9, "learning_rate": 4.049881235154395e-05, "loss": 1.7151, "step": 1600 }, { "epoch": 2.02, "learning_rate": 3.990498812351544e-05, "loss": 1.7013, "step": 1700 }, { "epoch": 2.14, "learning_rate": 3.9311163895486934e-05, "loss": 1.6831, "step": 1800 }, { "epoch": 2.26, "learning_rate": 3.8717339667458435e-05, "loss": 1.705, "step": 1900 }, { "epoch": 2.38, "learning_rate": 3.812351543942993e-05, "loss": 1.648, "step": 2000 }, { "epoch": 2.38, "eval_loss": 1.665128469467163, "eval_runtime": 49.4586, "eval_samples_per_second": 90.722, "eval_steps_per_second": 5.682, "step": 2000 }, { "epoch": 2.49, "learning_rate": 3.752969121140142e-05, "loss": 1.6486, "step": 2100 }, { "epoch": 2.61, "learning_rate": 3.693586698337292e-05, "loss": 1.685, "step": 2200 }, { "epoch": 2.73, "learning_rate": 3.6342042755344416e-05, "loss": 1.6529, "step": 2300 }, { "epoch": 2.85, "learning_rate": 3.5748218527315916e-05, "loss": 1.6458, "step": 2400 }, { "epoch": 2.97, "learning_rate": 3.515439429928741e-05, "loss": 1.606, "step": 2500 }, { "epoch": 2.97, "eval_loss": 1.6101973056793213, "eval_runtime": 49.0631, "eval_samples_per_second": 91.454, "eval_steps_per_second": 5.727, "step": 2500 }, { "epoch": 3.09, "learning_rate": 3.456057007125891e-05, "loss": 1.6211, "step": 2600 }, { "epoch": 3.21, "learning_rate": 3.396674584323041e-05, "loss": 1.5867, "step": 2700 }, { "epoch": 3.33, "learning_rate": 3.3372921615201904e-05, "loss": 1.6011, "step": 2800 }, { "epoch": 3.44, "learning_rate": 3.27790973871734e-05, "loss": 1.6018, "step": 2900 }, { "epoch": 3.56, "learning_rate": 3.21852731591449e-05, "loss": 1.5833, "step": 3000 }, { "epoch": 3.56, "eval_loss": 1.5911996364593506, "eval_runtime": 49.4655, "eval_samples_per_second": 90.71, "eval_steps_per_second": 5.681, "step": 3000 }, { "epoch": 3.68, "learning_rate": 3.159144893111639e-05, "loss": 1.6086, "step": 3100 }, { "epoch": 3.8, "learning_rate": 3.0997624703087885e-05, "loss": 1.5814, "step": 3200 }, { "epoch": 3.92, "learning_rate": 3.0403800475059386e-05, "loss": 1.5749, "step": 3300 }, { "epoch": 4.04, "learning_rate": 2.980997624703088e-05, "loss": 1.5372, "step": 3400 }, { "epoch": 4.16, "learning_rate": 2.9216152019002373e-05, "loss": 1.5663, "step": 3500 }, { "epoch": 4.16, "eval_loss": 1.5642139911651611, "eval_runtime": 50.1277, "eval_samples_per_second": 89.511, "eval_steps_per_second": 5.606, "step": 3500 }, { "epoch": 4.28, "learning_rate": 2.8622327790973873e-05, "loss": 1.5571, "step": 3600 }, { "epoch": 4.39, "learning_rate": 2.8028503562945367e-05, "loss": 1.5498, "step": 3700 }, { "epoch": 4.51, "learning_rate": 2.7434679334916867e-05, "loss": 1.5647, "step": 3800 }, { "epoch": 4.63, "learning_rate": 2.6840855106888364e-05, "loss": 1.5563, "step": 3900 }, { "epoch": 4.75, "learning_rate": 2.6247030878859858e-05, "loss": 1.5104, "step": 4000 }, { "epoch": 4.75, "eval_loss": 1.5389657020568848, "eval_runtime": 49.7746, "eval_samples_per_second": 90.146, "eval_steps_per_second": 5.645, "step": 4000 }, { "epoch": 4.87, "learning_rate": 2.5653206650831358e-05, "loss": 1.5108, "step": 4100 }, { "epoch": 4.99, "learning_rate": 2.5059382422802852e-05, "loss": 1.5179, "step": 4200 }, { "epoch": 5.11, "learning_rate": 2.446555819477435e-05, "loss": 1.5328, "step": 4300 }, { "epoch": 5.23, "learning_rate": 2.3871733966745842e-05, "loss": 1.5241, "step": 4400 }, { "epoch": 5.34, "learning_rate": 2.327790973871734e-05, "loss": 1.5252, "step": 4500 }, { "epoch": 5.34, "eval_loss": 1.5197077989578247, "eval_runtime": 49.1921, "eval_samples_per_second": 91.214, "eval_steps_per_second": 5.712, "step": 4500 }, { "epoch": 5.46, "learning_rate": 2.2684085510688836e-05, "loss": 1.4934, "step": 4600 }, { "epoch": 5.58, "learning_rate": 2.2090261282660333e-05, "loss": 1.4894, "step": 4700 }, { "epoch": 5.7, "learning_rate": 2.149643705463183e-05, "loss": 1.482, "step": 4800 }, { "epoch": 5.82, "learning_rate": 2.0902612826603327e-05, "loss": 1.5119, "step": 4900 }, { "epoch": 5.94, "learning_rate": 2.0308788598574824e-05, "loss": 1.4676, "step": 5000 }, { "epoch": 5.94, "eval_loss": 1.4950082302093506, "eval_runtime": 49.4722, "eval_samples_per_second": 90.697, "eval_steps_per_second": 5.68, "step": 5000 }, { "epoch": 6.06, "learning_rate": 1.9714964370546318e-05, "loss": 1.4807, "step": 5100 }, { "epoch": 6.18, "learning_rate": 1.9121140142517815e-05, "loss": 1.5018, "step": 5200 }, { "epoch": 6.29, "learning_rate": 1.8527315914489312e-05, "loss": 1.5298, "step": 5300 }, { "epoch": 6.41, "learning_rate": 1.793349168646081e-05, "loss": 1.4542, "step": 5400 }, { "epoch": 6.53, "learning_rate": 1.7339667458432306e-05, "loss": 1.4502, "step": 5500 }, { "epoch": 6.53, "eval_loss": 1.4766093492507935, "eval_runtime": 49.593, "eval_samples_per_second": 90.476, "eval_steps_per_second": 5.666, "step": 5500 }, { "epoch": 6.65, "learning_rate": 1.6745843230403803e-05, "loss": 1.4733, "step": 5600 }, { "epoch": 6.77, "learning_rate": 1.61520190023753e-05, "loss": 1.4759, "step": 5700 }, { "epoch": 6.89, "learning_rate": 1.5558194774346793e-05, "loss": 1.4482, "step": 5800 }, { "epoch": 7.01, "learning_rate": 1.496437054631829e-05, "loss": 1.4793, "step": 5900 }, { "epoch": 7.13, "learning_rate": 1.4370546318289787e-05, "loss": 1.4336, "step": 6000 }, { "epoch": 7.13, "eval_loss": 1.469414472579956, "eval_runtime": 49.7391, "eval_samples_per_second": 90.211, "eval_steps_per_second": 5.649, "step": 6000 }, { "epoch": 7.24, "learning_rate": 1.3776722090261283e-05, "loss": 1.4613, "step": 6100 }, { "epoch": 7.36, "learning_rate": 1.318289786223278e-05, "loss": 1.4433, "step": 6200 }, { "epoch": 7.48, "learning_rate": 1.2589073634204277e-05, "loss": 1.4808, "step": 6300 }, { "epoch": 7.6, "learning_rate": 1.1995249406175772e-05, "loss": 1.4319, "step": 6400 }, { "epoch": 7.72, "learning_rate": 1.1401425178147269e-05, "loss": 1.4355, "step": 6500 }, { "epoch": 7.72, "eval_loss": 1.4527209997177124, "eval_runtime": 50.5118, "eval_samples_per_second": 88.831, "eval_steps_per_second": 5.563, "step": 6500 }, { "epoch": 7.84, "learning_rate": 1.0807600950118766e-05, "loss": 1.4442, "step": 6600 }, { "epoch": 7.96, "learning_rate": 1.0213776722090261e-05, "loss": 1.4364, "step": 6700 }, { "epoch": 8.08, "learning_rate": 9.619952494061758e-06, "loss": 1.4462, "step": 6800 }, { "epoch": 8.19, "learning_rate": 9.026128266033253e-06, "loss": 1.4133, "step": 6900 }, { "epoch": 8.31, "learning_rate": 8.432304038004752e-06, "loss": 1.457, "step": 7000 }, { "epoch": 8.31, "eval_loss": 1.4403332471847534, "eval_runtime": 50.2785, "eval_samples_per_second": 89.243, "eval_steps_per_second": 5.589, "step": 7000 }, { "epoch": 8.43, "learning_rate": 7.838479809976247e-06, "loss": 1.4475, "step": 7100 }, { "epoch": 8.55, "learning_rate": 7.244655581947744e-06, "loss": 1.4356, "step": 7200 }, { "epoch": 8.67, "learning_rate": 6.6508313539192404e-06, "loss": 1.4198, "step": 7300 }, { "epoch": 8.79, "learning_rate": 6.0570071258907366e-06, "loss": 1.3991, "step": 7400 }, { "epoch": 8.91, "learning_rate": 5.4631828978622335e-06, "loss": 1.4219, "step": 7500 }, { "epoch": 8.91, "eval_loss": 1.4380402565002441, "eval_runtime": 50.3243, "eval_samples_per_second": 89.162, "eval_steps_per_second": 5.584, "step": 7500 }, { "epoch": 9.03, "learning_rate": 4.86935866983373e-06, "loss": 1.4489, "step": 7600 }, { "epoch": 9.14, "learning_rate": 4.275534441805226e-06, "loss": 1.4326, "step": 7700 }, { "epoch": 9.26, "learning_rate": 3.681710213776722e-06, "loss": 1.4397, "step": 7800 }, { "epoch": 9.38, "learning_rate": 3.0878859857482185e-06, "loss": 1.4247, "step": 7900 }, { "epoch": 9.5, "learning_rate": 2.494061757719715e-06, "loss": 1.4503, "step": 8000 }, { "epoch": 9.5, "eval_loss": 1.4312853813171387, "eval_runtime": 50.224, "eval_samples_per_second": 89.34, "eval_steps_per_second": 5.595, "step": 8000 }, { "epoch": 9.62, "learning_rate": 1.9002375296912116e-06, "loss": 1.4, "step": 8100 }, { "epoch": 9.74, "learning_rate": 1.306413301662708e-06, "loss": 1.4061, "step": 8200 }, { "epoch": 9.86, "learning_rate": 7.125890736342043e-07, "loss": 1.3747, "step": 8300 }, { "epoch": 9.98, "learning_rate": 1.1876484560570073e-07, "loss": 1.4183, "step": 8400 }, { "epoch": 10.0, "step": 8420, "total_flos": 3025468799385600.0, "train_loss": 1.581510691518172, "train_runtime": 4792.5391, "train_samples_per_second": 28.085, "train_steps_per_second": 1.757 }, { "epoch": 10.0, "eval_loss": 1.4350265264511108, "eval_runtime": 49.5388, "eval_samples_per_second": 90.575, "eval_steps_per_second": 5.672, "step": 8420 }, { "epoch": 10.0, "eval_loss": 1.4306435585021973, "eval_runtime": 49.7063, "eval_samples_per_second": 90.27, "eval_steps_per_second": 5.653, "step": 8420 } ], "logging_steps": 100, "max_steps": 8420, "num_train_epochs": 10, "save_steps": 500, "total_flos": 3025468799385600.0, "trial_name": null, "trial_params": null }