| { | |
| "best_metric": 1.387204885482788, | |
| "best_model_checkpoint": "output_dir/checkpoint-7500", | |
| "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": 3.5781, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "learning_rate": 4.8812351543942996e-05, | |
| "loss": 3.0221, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "learning_rate": 4.821852731591449e-05, | |
| "loss": 2.7656, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.48, | |
| "learning_rate": 4.7624703087885984e-05, | |
| "loss": 2.6432, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "learning_rate": 4.7030878859857484e-05, | |
| "loss": 2.5316, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "eval_loss": 2.2591474056243896, | |
| "eval_runtime": 34.7222, | |
| "eval_samples_per_second": 129.226, | |
| "eval_steps_per_second": 8.093, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.71, | |
| "learning_rate": 4.643705463182898e-05, | |
| "loss": 2.4743, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "learning_rate": 4.584323040380048e-05, | |
| "loss": 2.3371, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.95, | |
| "learning_rate": 4.524940617577197e-05, | |
| "loss": 2.3233, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 1.07, | |
| "learning_rate": 4.465558194774347e-05, | |
| "loss": 2.2412, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 1.19, | |
| "learning_rate": 4.406175771971497e-05, | |
| "loss": 2.2197, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.19, | |
| "eval_loss": 1.979953408241272, | |
| "eval_runtime": 34.803, | |
| "eval_samples_per_second": 128.926, | |
| "eval_steps_per_second": 8.074, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.31, | |
| "learning_rate": 4.3467933491686466e-05, | |
| "loss": 2.1312, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 1.43, | |
| "learning_rate": 4.287410926365796e-05, | |
| "loss": 2.1109, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 1.54, | |
| "learning_rate": 4.228028503562946e-05, | |
| "loss": 2.0618, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 1.66, | |
| "learning_rate": 4.168646080760095e-05, | |
| "loss": 2.0827, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 1.78, | |
| "learning_rate": 4.109263657957245e-05, | |
| "loss": 2.0108, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.78, | |
| "eval_loss": 1.830361008644104, | |
| "eval_runtime": 34.8169, | |
| "eval_samples_per_second": 128.874, | |
| "eval_steps_per_second": 8.071, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.9, | |
| "learning_rate": 4.049881235154395e-05, | |
| "loss": 1.9889, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 2.02, | |
| "learning_rate": 3.990498812351544e-05, | |
| "loss": 1.9597, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 2.14, | |
| "learning_rate": 3.9311163895486934e-05, | |
| "loss": 1.9155, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 2.26, | |
| "learning_rate": 3.8717339667458435e-05, | |
| "loss": 1.9138, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 2.38, | |
| "learning_rate": 3.812351543942993e-05, | |
| "loss": 1.855, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 2.38, | |
| "eval_loss": 1.722902536392212, | |
| "eval_runtime": 34.0608, | |
| "eval_samples_per_second": 131.735, | |
| "eval_steps_per_second": 8.25, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 2.49, | |
| "learning_rate": 3.752969121140142e-05, | |
| "loss": 1.8592, | |
| "step": 2100 | |
| }, | |
| { | |
| "epoch": 2.61, | |
| "learning_rate": 3.693586698337292e-05, | |
| "loss": 1.8951, | |
| "step": 2200 | |
| }, | |
| { | |
| "epoch": 2.73, | |
| "learning_rate": 3.6342042755344416e-05, | |
| "loss": 1.8651, | |
| "step": 2300 | |
| }, | |
| { | |
| "epoch": 2.85, | |
| "learning_rate": 3.5748218527315916e-05, | |
| "loss": 1.808, | |
| "step": 2400 | |
| }, | |
| { | |
| "epoch": 2.97, | |
| "learning_rate": 3.515439429928741e-05, | |
| "loss": 1.7933, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 2.97, | |
| "eval_loss": 1.6560965776443481, | |
| "eval_runtime": 34.0703, | |
| "eval_samples_per_second": 131.698, | |
| "eval_steps_per_second": 8.248, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 3.09, | |
| "learning_rate": 3.456057007125891e-05, | |
| "loss": 1.8124, | |
| "step": 2600 | |
| }, | |
| { | |
| "epoch": 3.21, | |
| "learning_rate": 3.396674584323041e-05, | |
| "loss": 1.7598, | |
| "step": 2700 | |
| }, | |
| { | |
| "epoch": 3.33, | |
| "learning_rate": 3.3372921615201904e-05, | |
| "loss": 1.7716, | |
| "step": 2800 | |
| }, | |
| { | |
| "epoch": 3.44, | |
| "learning_rate": 3.27790973871734e-05, | |
| "loss": 1.7557, | |
| "step": 2900 | |
| }, | |
| { | |
| "epoch": 3.56, | |
| "learning_rate": 3.21852731591449e-05, | |
| "loss": 1.7466, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 3.56, | |
| "eval_loss": 1.6100554466247559, | |
| "eval_runtime": 34.1759, | |
| "eval_samples_per_second": 131.291, | |
| "eval_steps_per_second": 8.222, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 3.68, | |
| "learning_rate": 3.159144893111639e-05, | |
| "loss": 1.745, | |
| "step": 3100 | |
| }, | |
| { | |
| "epoch": 3.8, | |
| "learning_rate": 3.0997624703087885e-05, | |
| "loss": 1.729, | |
| "step": 3200 | |
| }, | |
| { | |
| "epoch": 3.92, | |
| "learning_rate": 3.0403800475059386e-05, | |
| "loss": 1.7112, | |
| "step": 3300 | |
| }, | |
| { | |
| "epoch": 4.04, | |
| "learning_rate": 2.980997624703088e-05, | |
| "loss": 1.6804, | |
| "step": 3400 | |
| }, | |
| { | |
| "epoch": 4.16, | |
| "learning_rate": 2.9216152019002373e-05, | |
| "loss": 1.6906, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 4.16, | |
| "eval_loss": 1.5587605237960815, | |
| "eval_runtime": 34.0323, | |
| "eval_samples_per_second": 131.845, | |
| "eval_steps_per_second": 8.257, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 4.28, | |
| "learning_rate": 2.8622327790973873e-05, | |
| "loss": 1.6638, | |
| "step": 3600 | |
| }, | |
| { | |
| "epoch": 4.39, | |
| "learning_rate": 2.8028503562945367e-05, | |
| "loss": 1.6895, | |
| "step": 3700 | |
| }, | |
| { | |
| "epoch": 4.51, | |
| "learning_rate": 2.7434679334916867e-05, | |
| "loss": 1.6983, | |
| "step": 3800 | |
| }, | |
| { | |
| "epoch": 4.63, | |
| "learning_rate": 2.6840855106888364e-05, | |
| "loss": 1.6648, | |
| "step": 3900 | |
| }, | |
| { | |
| "epoch": 4.75, | |
| "learning_rate": 2.6247030878859858e-05, | |
| "loss": 1.6301, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 4.75, | |
| "eval_loss": 1.5263967514038086, | |
| "eval_runtime": 34.0548, | |
| "eval_samples_per_second": 131.758, | |
| "eval_steps_per_second": 8.251, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 4.87, | |
| "learning_rate": 2.5653206650831358e-05, | |
| "loss": 1.6388, | |
| "step": 4100 | |
| }, | |
| { | |
| "epoch": 4.99, | |
| "learning_rate": 2.5059382422802852e-05, | |
| "loss": 1.6336, | |
| "step": 4200 | |
| }, | |
| { | |
| "epoch": 5.11, | |
| "learning_rate": 2.446555819477435e-05, | |
| "loss": 1.6427, | |
| "step": 4300 | |
| }, | |
| { | |
| "epoch": 5.23, | |
| "learning_rate": 2.3871733966745842e-05, | |
| "loss": 1.6219, | |
| "step": 4400 | |
| }, | |
| { | |
| "epoch": 5.34, | |
| "learning_rate": 2.327790973871734e-05, | |
| "loss": 1.6131, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 5.34, | |
| "eval_loss": 1.489254355430603, | |
| "eval_runtime": 34.1712, | |
| "eval_samples_per_second": 131.309, | |
| "eval_steps_per_second": 8.223, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 5.46, | |
| "learning_rate": 2.2684085510688836e-05, | |
| "loss": 1.5886, | |
| "step": 4600 | |
| }, | |
| { | |
| "epoch": 5.58, | |
| "learning_rate": 2.2090261282660333e-05, | |
| "loss": 1.5906, | |
| "step": 4700 | |
| }, | |
| { | |
| "epoch": 5.7, | |
| "learning_rate": 2.149643705463183e-05, | |
| "loss": 1.5717, | |
| "step": 4800 | |
| }, | |
| { | |
| "epoch": 5.82, | |
| "learning_rate": 2.0902612826603327e-05, | |
| "loss": 1.5887, | |
| "step": 4900 | |
| }, | |
| { | |
| "epoch": 5.94, | |
| "learning_rate": 2.0308788598574824e-05, | |
| "loss": 1.5731, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 5.94, | |
| "eval_loss": 1.4625672101974487, | |
| "eval_runtime": 34.0832, | |
| "eval_samples_per_second": 131.649, | |
| "eval_steps_per_second": 8.245, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 6.06, | |
| "learning_rate": 1.9714964370546318e-05, | |
| "loss": 1.5607, | |
| "step": 5100 | |
| }, | |
| { | |
| "epoch": 6.18, | |
| "learning_rate": 1.9121140142517815e-05, | |
| "loss": 1.5905, | |
| "step": 5200 | |
| }, | |
| { | |
| "epoch": 6.29, | |
| "learning_rate": 1.8527315914489312e-05, | |
| "loss": 1.6074, | |
| "step": 5300 | |
| }, | |
| { | |
| "epoch": 6.41, | |
| "learning_rate": 1.793349168646081e-05, | |
| "loss": 1.5401, | |
| "step": 5400 | |
| }, | |
| { | |
| "epoch": 6.53, | |
| "learning_rate": 1.7339667458432306e-05, | |
| "loss": 1.5248, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 6.53, | |
| "eval_loss": 1.4445180892944336, | |
| "eval_runtime": 33.993, | |
| "eval_samples_per_second": 131.998, | |
| "eval_steps_per_second": 8.266, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 6.65, | |
| "learning_rate": 1.6745843230403803e-05, | |
| "loss": 1.5611, | |
| "step": 5600 | |
| }, | |
| { | |
| "epoch": 6.77, | |
| "learning_rate": 1.61520190023753e-05, | |
| "loss": 1.5312, | |
| "step": 5700 | |
| }, | |
| { | |
| "epoch": 6.89, | |
| "learning_rate": 1.5558194774346793e-05, | |
| "loss": 1.5318, | |
| "step": 5800 | |
| }, | |
| { | |
| "epoch": 7.01, | |
| "learning_rate": 1.496437054631829e-05, | |
| "loss": 1.5366, | |
| "step": 5900 | |
| }, | |
| { | |
| "epoch": 7.13, | |
| "learning_rate": 1.4370546318289787e-05, | |
| "loss": 1.509, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 7.13, | |
| "eval_loss": 1.4273977279663086, | |
| "eval_runtime": 33.9883, | |
| "eval_samples_per_second": 132.016, | |
| "eval_steps_per_second": 8.268, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 7.24, | |
| "learning_rate": 1.3776722090261283e-05, | |
| "loss": 1.5425, | |
| "step": 6100 | |
| }, | |
| { | |
| "epoch": 7.36, | |
| "learning_rate": 1.318289786223278e-05, | |
| "loss": 1.5253, | |
| "step": 6200 | |
| }, | |
| { | |
| "epoch": 7.48, | |
| "learning_rate": 1.2589073634204277e-05, | |
| "loss": 1.5502, | |
| "step": 6300 | |
| }, | |
| { | |
| "epoch": 7.6, | |
| "learning_rate": 1.1995249406175772e-05, | |
| "loss": 1.5011, | |
| "step": 6400 | |
| }, | |
| { | |
| "epoch": 7.72, | |
| "learning_rate": 1.1401425178147269e-05, | |
| "loss": 1.5083, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 7.72, | |
| "eval_loss": 1.410913348197937, | |
| "eval_runtime": 33.9769, | |
| "eval_samples_per_second": 132.06, | |
| "eval_steps_per_second": 8.27, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 7.84, | |
| "learning_rate": 1.0807600950118766e-05, | |
| "loss": 1.5303, | |
| "step": 6600 | |
| }, | |
| { | |
| "epoch": 7.96, | |
| "learning_rate": 1.0213776722090261e-05, | |
| "loss": 1.4877, | |
| "step": 6700 | |
| }, | |
| { | |
| "epoch": 8.08, | |
| "learning_rate": 9.619952494061758e-06, | |
| "loss": 1.5082, | |
| "step": 6800 | |
| }, | |
| { | |
| "epoch": 8.19, | |
| "learning_rate": 9.026128266033253e-06, | |
| "loss": 1.4751, | |
| "step": 6900 | |
| }, | |
| { | |
| "epoch": 8.31, | |
| "learning_rate": 8.432304038004752e-06, | |
| "loss": 1.5277, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 8.31, | |
| "eval_loss": 1.4051495790481567, | |
| "eval_runtime": 33.9346, | |
| "eval_samples_per_second": 132.225, | |
| "eval_steps_per_second": 8.281, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 8.43, | |
| "learning_rate": 7.838479809976247e-06, | |
| "loss": 1.5207, | |
| "step": 7100 | |
| }, | |
| { | |
| "epoch": 8.55, | |
| "learning_rate": 7.244655581947744e-06, | |
| "loss": 1.4948, | |
| "step": 7200 | |
| }, | |
| { | |
| "epoch": 8.67, | |
| "learning_rate": 6.6508313539192404e-06, | |
| "loss": 1.4897, | |
| "step": 7300 | |
| }, | |
| { | |
| "epoch": 8.79, | |
| "learning_rate": 6.0570071258907366e-06, | |
| "loss": 1.4801, | |
| "step": 7400 | |
| }, | |
| { | |
| "epoch": 8.91, | |
| "learning_rate": 5.4631828978622335e-06, | |
| "loss": 1.4884, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 8.91, | |
| "eval_loss": 1.387204885482788, | |
| "eval_runtime": 33.9831, | |
| "eval_samples_per_second": 132.036, | |
| "eval_steps_per_second": 8.269, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 9.03, | |
| "learning_rate": 4.86935866983373e-06, | |
| "loss": 1.5037, | |
| "step": 7600 | |
| }, | |
| { | |
| "epoch": 9.14, | |
| "learning_rate": 4.275534441805226e-06, | |
| "loss": 1.4884, | |
| "step": 7700 | |
| }, | |
| { | |
| "epoch": 9.26, | |
| "learning_rate": 3.681710213776722e-06, | |
| "loss": 1.4968, | |
| "step": 7800 | |
| }, | |
| { | |
| "epoch": 9.38, | |
| "learning_rate": 3.0878859857482185e-06, | |
| "loss": 1.4762, | |
| "step": 7900 | |
| }, | |
| { | |
| "epoch": 9.5, | |
| "learning_rate": 2.494061757719715e-06, | |
| "loss": 1.496, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 9.5, | |
| "eval_loss": 1.3912383317947388, | |
| "eval_runtime": 33.9905, | |
| "eval_samples_per_second": 132.007, | |
| "eval_steps_per_second": 8.267, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 9.62, | |
| "learning_rate": 1.9002375296912116e-06, | |
| "loss": 1.4703, | |
| "step": 8100 | |
| }, | |
| { | |
| "epoch": 9.74, | |
| "learning_rate": 1.306413301662708e-06, | |
| "loss": 1.4722, | |
| "step": 8200 | |
| }, | |
| { | |
| "epoch": 9.86, | |
| "learning_rate": 7.125890736342043e-07, | |
| "loss": 1.4392, | |
| "step": 8300 | |
| }, | |
| { | |
| "epoch": 9.98, | |
| "learning_rate": 1.1876484560570073e-07, | |
| "loss": 1.4692, | |
| "step": 8400 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "step": 8420, | |
| "total_flos": 1.78429300862976e+16, | |
| "train_loss": 1.7687041851233982, | |
| "train_runtime": 3488.0926, | |
| "train_samples_per_second": 38.588, | |
| "train_steps_per_second": 2.414 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_loss": 1.3867350816726685, | |
| "eval_runtime": 34.1096, | |
| "eval_samples_per_second": 131.547, | |
| "eval_steps_per_second": 8.238, | |
| "step": 8420 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_loss": 1.3883991241455078, | |
| "eval_runtime": 34.15, | |
| "eval_samples_per_second": 131.391, | |
| "eval_steps_per_second": 8.228, | |
| "step": 8420 | |
| } | |
| ], | |
| "logging_steps": 100, | |
| "max_steps": 8420, | |
| "num_train_epochs": 10, | |
| "save_steps": 500, | |
| "total_flos": 1.78429300862976e+16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |