| { | |
| "best_metric": 3.903717041015625, | |
| "best_model_checkpoint": "./models/gpt_test/checkpoint-10000", | |
| "epoch": 5.0, | |
| "global_step": 13240, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.38, | |
| "learning_rate": 2.2658610271903327e-05, | |
| "loss": 3.3245, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.38, | |
| "eval_loss": 4.017566680908203, | |
| "eval_runtime": 6.7548, | |
| "eval_samples_per_second": 63.51, | |
| "eval_steps_per_second": 3.997, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.76, | |
| "learning_rate": 2.8298086606243708e-05, | |
| "loss": 3.1222, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.76, | |
| "eval_loss": 3.984508514404297, | |
| "eval_runtime": 6.7536, | |
| "eval_samples_per_second": 63.522, | |
| "eval_steps_per_second": 3.998, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 1.13, | |
| "learning_rate": 2.5780463242698894e-05, | |
| "loss": 2.9992, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 1.13, | |
| "eval_loss": 3.9634785652160645, | |
| "eval_runtime": 6.7624, | |
| "eval_samples_per_second": 63.439, | |
| "eval_steps_per_second": 3.993, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 1.51, | |
| "learning_rate": 2.3262839879154077e-05, | |
| "loss": 2.8843, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 1.51, | |
| "eval_loss": 3.9377200603485107, | |
| "eval_runtime": 6.7434, | |
| "eval_samples_per_second": 63.618, | |
| "eval_steps_per_second": 4.004, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 1.89, | |
| "learning_rate": 2.0745216515609266e-05, | |
| "loss": 2.882, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 1.89, | |
| "eval_loss": 3.926788091659546, | |
| "eval_runtime": 6.7619, | |
| "eval_samples_per_second": 63.443, | |
| "eval_steps_per_second": 3.993, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 2.27, | |
| "learning_rate": 1.822759315206445e-05, | |
| "loss": 2.7411, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 2.27, | |
| "eval_loss": 3.9208364486694336, | |
| "eval_runtime": 6.7546, | |
| "eval_samples_per_second": 63.512, | |
| "eval_steps_per_second": 3.997, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 2.64, | |
| "learning_rate": 1.570996978851964e-05, | |
| "loss": 2.7204, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 2.64, | |
| "eval_loss": 3.916015148162842, | |
| "eval_runtime": 6.7585, | |
| "eval_samples_per_second": 63.476, | |
| "eval_steps_per_second": 3.995, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 3.02, | |
| "learning_rate": 1.3192346424974824e-05, | |
| "loss": 2.7106, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 3.02, | |
| "eval_loss": 3.9170773029327393, | |
| "eval_runtime": 6.749, | |
| "eval_samples_per_second": 63.565, | |
| "eval_steps_per_second": 4.001, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 3.4, | |
| "learning_rate": 1.067472306143001e-05, | |
| "loss": 2.5857, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 3.4, | |
| "eval_loss": 3.9161624908447266, | |
| "eval_runtime": 6.7583, | |
| "eval_samples_per_second": 63.477, | |
| "eval_steps_per_second": 3.995, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 3.78, | |
| "learning_rate": 8.157099697885196e-06, | |
| "loss": 2.5863, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 3.78, | |
| "eval_loss": 3.903717041015625, | |
| "eval_runtime": 6.7566, | |
| "eval_samples_per_second": 63.493, | |
| "eval_steps_per_second": 3.996, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 4.15, | |
| "learning_rate": 5.6394763343403835e-06, | |
| "loss": 2.5674, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 4.15, | |
| "eval_loss": 3.9134981632232666, | |
| "eval_runtime": 6.7557, | |
| "eval_samples_per_second": 63.502, | |
| "eval_steps_per_second": 3.997, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 4.53, | |
| "learning_rate": 3.121852970795569e-06, | |
| "loss": 2.4901, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 4.53, | |
| "eval_loss": 3.9125208854675293, | |
| "eval_runtime": 6.7457, | |
| "eval_samples_per_second": 63.596, | |
| "eval_steps_per_second": 4.003, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 4.91, | |
| "learning_rate": 6.042296072507553e-07, | |
| "loss": 2.505, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 4.91, | |
| "eval_loss": 3.9078235626220703, | |
| "eval_runtime": 6.747, | |
| "eval_samples_per_second": 63.584, | |
| "eval_steps_per_second": 4.002, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "step": 13240, | |
| "total_flos": 5.535079759872e+16, | |
| "train_loss": 2.77278194830857, | |
| "train_runtime": 10839.8418, | |
| "train_samples_per_second": 19.542, | |
| "train_steps_per_second": 1.221 | |
| } | |
| ], | |
| "max_steps": 13240, | |
| "num_train_epochs": 5, | |
| "total_flos": 5.535079759872e+16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |