|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 1.8896447467876039, |
|
"eval_steps": 500, |
|
"global_step": 15000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 0.000975, |
|
"loss": 1.0238, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"eval_loss": 0.8929525017738342, |
|
"eval_runtime": 4.0089, |
|
"eval_samples_per_second": 74.833, |
|
"eval_steps_per_second": 37.417, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 0.00095, |
|
"loss": 0.9971, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"eval_loss": 0.8510345220565796, |
|
"eval_runtime": 4.0089, |
|
"eval_samples_per_second": 74.833, |
|
"eval_steps_per_second": 37.417, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 0.000925, |
|
"loss": 0.9522, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"eval_loss": 0.8565580248832703, |
|
"eval_runtime": 4.0278, |
|
"eval_samples_per_second": 74.483, |
|
"eval_steps_per_second": 37.241, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 0.0009000000000000001, |
|
"loss": 0.9143, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"eval_loss": 0.8810315132141113, |
|
"eval_runtime": 4.0097, |
|
"eval_samples_per_second": 74.819, |
|
"eval_steps_per_second": 37.409, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 0.000875, |
|
"loss": 0.9151, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"eval_loss": 0.8461073040962219, |
|
"eval_runtime": 4.0108, |
|
"eval_samples_per_second": 74.798, |
|
"eval_steps_per_second": 37.399, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 0.00085, |
|
"loss": 0.9263, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"eval_loss": 0.8521466255187988, |
|
"eval_runtime": 4.0059, |
|
"eval_samples_per_second": 74.89, |
|
"eval_steps_per_second": 37.445, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 0.000825, |
|
"loss": 0.9102, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"eval_loss": 0.8521981835365295, |
|
"eval_runtime": 4.0413, |
|
"eval_samples_per_second": 74.234, |
|
"eval_steps_per_second": 37.117, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 0.0008, |
|
"loss": 0.8866, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"eval_loss": 0.8259674906730652, |
|
"eval_runtime": 4.0316, |
|
"eval_samples_per_second": 74.412, |
|
"eval_steps_per_second": 37.206, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 0.0007750000000000001, |
|
"loss": 0.911, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"eval_loss": 0.8008900284767151, |
|
"eval_runtime": 4.0224, |
|
"eval_samples_per_second": 74.583, |
|
"eval_steps_per_second": 37.292, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 0.00075, |
|
"loss": 0.8536, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"eval_loss": 0.808663547039032, |
|
"eval_runtime": 5.0615, |
|
"eval_samples_per_second": 59.271, |
|
"eval_steps_per_second": 29.635, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 0.000725, |
|
"loss": 0.8277, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"eval_loss": 0.7903586626052856, |
|
"eval_runtime": 4.0081, |
|
"eval_samples_per_second": 74.849, |
|
"eval_steps_per_second": 37.425, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 0.0007, |
|
"loss": 0.8413, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"eval_loss": 0.8013813495635986, |
|
"eval_runtime": 3.994, |
|
"eval_samples_per_second": 75.113, |
|
"eval_steps_per_second": 37.556, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 0.000675, |
|
"loss": 0.8491, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"eval_loss": 0.7867017984390259, |
|
"eval_runtime": 4.0072, |
|
"eval_samples_per_second": 74.864, |
|
"eval_steps_per_second": 37.432, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"learning_rate": 0.0006500000000000001, |
|
"loss": 0.8077, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"eval_loss": 0.759165346622467, |
|
"eval_runtime": 4.0166, |
|
"eval_samples_per_second": 74.691, |
|
"eval_steps_per_second": 37.345, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 0.000625, |
|
"loss": 0.796, |
|
"step": 7500 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"eval_loss": 0.7527943849563599, |
|
"eval_runtime": 3.998, |
|
"eval_samples_per_second": 75.037, |
|
"eval_steps_per_second": 37.518, |
|
"step": 7500 |
|
}, |
|
{ |
|
"epoch": 1.01, |
|
"learning_rate": 0.0006, |
|
"loss": 0.7952, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 1.01, |
|
"eval_loss": 0.7407393455505371, |
|
"eval_runtime": 3.9421, |
|
"eval_samples_per_second": 76.102, |
|
"eval_steps_per_second": 38.051, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 1.07, |
|
"learning_rate": 0.000575, |
|
"loss": 0.5761, |
|
"step": 8500 |
|
}, |
|
{ |
|
"epoch": 1.07, |
|
"eval_loss": 0.7574229836463928, |
|
"eval_runtime": 3.9855, |
|
"eval_samples_per_second": 75.273, |
|
"eval_steps_per_second": 37.636, |
|
"step": 8500 |
|
}, |
|
{ |
|
"epoch": 1.13, |
|
"learning_rate": 0.00055, |
|
"loss": 0.6084, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 1.13, |
|
"eval_loss": 0.7349967956542969, |
|
"eval_runtime": 3.995, |
|
"eval_samples_per_second": 75.093, |
|
"eval_steps_per_second": 37.547, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 1.2, |
|
"learning_rate": 0.0005250000000000001, |
|
"loss": 0.5863, |
|
"step": 9500 |
|
}, |
|
{ |
|
"epoch": 1.2, |
|
"eval_loss": 0.7407773733139038, |
|
"eval_runtime": 3.9874, |
|
"eval_samples_per_second": 75.238, |
|
"eval_steps_per_second": 37.619, |
|
"step": 9500 |
|
}, |
|
{ |
|
"epoch": 1.26, |
|
"learning_rate": 0.0005, |
|
"loss": 0.5835, |
|
"step": 10000 |
|
}, |
|
{ |
|
"epoch": 1.26, |
|
"eval_loss": 0.719527006149292, |
|
"eval_runtime": 3.9471, |
|
"eval_samples_per_second": 76.006, |
|
"eval_steps_per_second": 38.003, |
|
"step": 10000 |
|
}, |
|
{ |
|
"epoch": 1.32, |
|
"learning_rate": 0.000475, |
|
"loss": 0.5751, |
|
"step": 10500 |
|
}, |
|
{ |
|
"epoch": 1.32, |
|
"eval_loss": 0.7423205971717834, |
|
"eval_runtime": 3.9978, |
|
"eval_samples_per_second": 75.041, |
|
"eval_steps_per_second": 37.52, |
|
"step": 10500 |
|
}, |
|
{ |
|
"epoch": 1.39, |
|
"learning_rate": 0.00045000000000000004, |
|
"loss": 0.5746, |
|
"step": 11000 |
|
}, |
|
{ |
|
"epoch": 1.39, |
|
"eval_loss": 0.7284040451049805, |
|
"eval_runtime": 3.9657, |
|
"eval_samples_per_second": 75.649, |
|
"eval_steps_per_second": 37.824, |
|
"step": 11000 |
|
}, |
|
{ |
|
"epoch": 1.45, |
|
"learning_rate": 0.000425, |
|
"loss": 0.5847, |
|
"step": 11500 |
|
}, |
|
{ |
|
"epoch": 1.45, |
|
"eval_loss": 0.7247716188430786, |
|
"eval_runtime": 3.962, |
|
"eval_samples_per_second": 75.719, |
|
"eval_steps_per_second": 37.86, |
|
"step": 11500 |
|
}, |
|
{ |
|
"epoch": 1.51, |
|
"learning_rate": 0.0004, |
|
"loss": 0.5759, |
|
"step": 12000 |
|
}, |
|
{ |
|
"epoch": 1.51, |
|
"eval_loss": 0.7151039838790894, |
|
"eval_runtime": 3.9922, |
|
"eval_samples_per_second": 75.147, |
|
"eval_steps_per_second": 37.573, |
|
"step": 12000 |
|
}, |
|
{ |
|
"epoch": 1.57, |
|
"learning_rate": 0.000375, |
|
"loss": 0.5922, |
|
"step": 12500 |
|
}, |
|
{ |
|
"epoch": 1.57, |
|
"eval_loss": 0.7031030058860779, |
|
"eval_runtime": 4.0386, |
|
"eval_samples_per_second": 74.282, |
|
"eval_steps_per_second": 37.141, |
|
"step": 12500 |
|
}, |
|
{ |
|
"epoch": 1.64, |
|
"learning_rate": 0.00035, |
|
"loss": 0.5678, |
|
"step": 13000 |
|
}, |
|
{ |
|
"epoch": 1.64, |
|
"eval_loss": 0.710035502910614, |
|
"eval_runtime": 4.0808, |
|
"eval_samples_per_second": 73.515, |
|
"eval_steps_per_second": 36.758, |
|
"step": 13000 |
|
}, |
|
{ |
|
"epoch": 1.7, |
|
"learning_rate": 0.00032500000000000004, |
|
"loss": 0.5627, |
|
"step": 13500 |
|
}, |
|
{ |
|
"epoch": 1.7, |
|
"eval_loss": 0.7093074321746826, |
|
"eval_runtime": 4.0326, |
|
"eval_samples_per_second": 74.393, |
|
"eval_steps_per_second": 37.197, |
|
"step": 13500 |
|
}, |
|
{ |
|
"epoch": 1.76, |
|
"learning_rate": 0.0003, |
|
"loss": 0.5568, |
|
"step": 14000 |
|
}, |
|
{ |
|
"epoch": 1.76, |
|
"eval_loss": 0.6942310333251953, |
|
"eval_runtime": 4.0875, |
|
"eval_samples_per_second": 73.395, |
|
"eval_steps_per_second": 36.697, |
|
"step": 14000 |
|
}, |
|
{ |
|
"epoch": 1.83, |
|
"learning_rate": 0.000275, |
|
"loss": 0.5589, |
|
"step": 14500 |
|
}, |
|
{ |
|
"epoch": 1.83, |
|
"eval_loss": 0.6947352290153503, |
|
"eval_runtime": 3.9856, |
|
"eval_samples_per_second": 75.27, |
|
"eval_steps_per_second": 37.635, |
|
"step": 14500 |
|
}, |
|
{ |
|
"epoch": 1.89, |
|
"learning_rate": 0.00025, |
|
"loss": 0.5675, |
|
"step": 15000 |
|
}, |
|
{ |
|
"epoch": 1.89, |
|
"eval_loss": 0.6902616620063782, |
|
"eval_runtime": 4.0066, |
|
"eval_samples_per_second": 74.876, |
|
"eval_steps_per_second": 37.438, |
|
"step": 15000 |
|
} |
|
], |
|
"logging_steps": 500, |
|
"max_steps": 20000, |
|
"num_train_epochs": 3, |
|
"save_steps": 5000, |
|
"total_flos": 2.66437480937472e+16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|