|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 3.5, |
|
"eval_steps": 20, |
|
"global_step": 350, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 1e-05, |
|
"loss": 0.588, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 2e-05, |
|
"loss": 0.5793, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"eval_loss": 0.5849277377128601, |
|
"eval_runtime": 13.5251, |
|
"eval_samples_per_second": 1.848, |
|
"eval_steps_per_second": 0.296, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 3e-05, |
|
"loss": 0.5065, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 4e-05, |
|
"loss": 0.5404, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"eval_loss": 0.5613898634910583, |
|
"eval_runtime": 13.3646, |
|
"eval_samples_per_second": 1.871, |
|
"eval_steps_per_second": 0.299, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 5e-05, |
|
"loss": 0.4523, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 4.9986331433523156e-05, |
|
"loss": 0.5692, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"eval_loss": 0.5513718724250793, |
|
"eval_runtime": 13.3937, |
|
"eval_samples_per_second": 1.867, |
|
"eval_steps_per_second": 0.299, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 4.994534068046937e-05, |
|
"loss": 0.5257, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 4.9877072563625285e-05, |
|
"loss": 0.4937, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"eval_loss": 0.5424299836158752, |
|
"eval_runtime": 13.4443, |
|
"eval_samples_per_second": 1.86, |
|
"eval_steps_per_second": 0.298, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 4.978160173317438e-05, |
|
"loss": 0.5617, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 4.965903258506806e-05, |
|
"loss": 0.3592, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_loss": 0.5260083675384521, |
|
"eval_runtime": 13.4207, |
|
"eval_samples_per_second": 1.863, |
|
"eval_steps_per_second": 0.298, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 1.1, |
|
"learning_rate": 4.9509499146870236e-05, |
|
"loss": 0.4586, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 1.2, |
|
"learning_rate": 4.933316493120015e-05, |
|
"loss": 0.3724, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 1.2, |
|
"eval_loss": 0.5247484445571899, |
|
"eval_runtime": 13.4092, |
|
"eval_samples_per_second": 1.864, |
|
"eval_steps_per_second": 0.298, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 1.3, |
|
"learning_rate": 4.913022275693372e-05, |
|
"loss": 0.3205, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 1.4, |
|
"learning_rate": 4.8900894538358944e-05, |
|
"loss": 0.4962, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 1.4, |
|
"eval_loss": 0.5248944163322449, |
|
"eval_runtime": 13.4511, |
|
"eval_samples_per_second": 1.859, |
|
"eval_steps_per_second": 0.297, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 1.5, |
|
"learning_rate": 4.864543104251587e-05, |
|
"loss": 0.3324, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 1.6, |
|
"learning_rate": 4.8364111614986527e-05, |
|
"loss": 0.4781, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 1.6, |
|
"eval_loss": 0.5312905311584473, |
|
"eval_runtime": 13.4112, |
|
"eval_samples_per_second": 1.864, |
|
"eval_steps_per_second": 0.298, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 1.7, |
|
"learning_rate": 4.805724387443462e-05, |
|
"loss": 0.3429, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 1.8, |
|
"learning_rate": 4.7725163376229064e-05, |
|
"loss": 0.3504, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 1.8, |
|
"eval_loss": 0.5292657017707825, |
|
"eval_runtime": 13.3837, |
|
"eval_samples_per_second": 1.868, |
|
"eval_steps_per_second": 0.299, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 1.9, |
|
"learning_rate": 4.736823324551909e-05, |
|
"loss": 0.4843, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"learning_rate": 4.698684378016222e-05, |
|
"loss": 0.2449, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_loss": 0.5281097292900085, |
|
"eval_runtime": 13.438, |
|
"eval_samples_per_second": 1.86, |
|
"eval_steps_per_second": 0.298, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 2.1, |
|
"learning_rate": 4.6581412023939354e-05, |
|
"loss": 0.3535, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 2.2, |
|
"learning_rate": 4.6152381310523387e-05, |
|
"loss": 0.2264, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 2.2, |
|
"eval_loss": 0.5974144339561462, |
|
"eval_runtime": 13.4417, |
|
"eval_samples_per_second": 1.86, |
|
"eval_steps_per_second": 0.298, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 2.3, |
|
"learning_rate": 4.5700220778700504e-05, |
|
"loss": 0.2303, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 2.4, |
|
"learning_rate": 4.522542485937369e-05, |
|
"loss": 0.3395, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 2.4, |
|
"eval_loss": 0.5777361989021301, |
|
"eval_runtime": 13.414, |
|
"eval_samples_per_second": 1.864, |
|
"eval_steps_per_second": 0.298, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 2.5, |
|
"learning_rate": 4.4728512734909844e-05, |
|
"loss": 0.1651, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 2.6, |
|
"learning_rate": 4.421002777142148e-05, |
|
"loss": 0.3093, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 2.6, |
|
"eval_loss": 0.5854852199554443, |
|
"eval_runtime": 13.4001, |
|
"eval_samples_per_second": 1.866, |
|
"eval_steps_per_second": 0.299, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 2.7, |
|
"learning_rate": 4.367053692460385e-05, |
|
"loss": 0.2512, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 2.8, |
|
"learning_rate": 4.311063011977723e-05, |
|
"loss": 0.2054, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 2.8, |
|
"eval_loss": 0.5999884605407715, |
|
"eval_runtime": 13.4618, |
|
"eval_samples_per_second": 1.857, |
|
"eval_steps_per_second": 0.297, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 2.9, |
|
"learning_rate": 4.2530919606812216e-05, |
|
"loss": 0.3547, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"learning_rate": 4.193203929064353e-05, |
|
"loss": 0.1739, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"eval_loss": 0.5840221047401428, |
|
"eval_runtime": 13.4097, |
|
"eval_samples_per_second": 1.864, |
|
"eval_steps_per_second": 0.298, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 3.1, |
|
"learning_rate": 4.131464403810422e-05, |
|
"loss": 0.1787, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 3.2, |
|
"learning_rate": 4.067940896183843e-05, |
|
"loss": 0.1165, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 3.2, |
|
"eval_loss": 0.6706953644752502, |
|
"eval_runtime": 13.4189, |
|
"eval_samples_per_second": 1.863, |
|
"eval_steps_per_second": 0.298, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 3.3, |
|
"learning_rate": 4.002702868207563e-05, |
|
"loss": 0.1477, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 3.4, |
|
"learning_rate": 3.935821656707359e-05, |
|
"loss": 0.2011, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 3.4, |
|
"eval_loss": 0.6934244632720947, |
|
"eval_runtime": 13.403, |
|
"eval_samples_per_second": 1.865, |
|
"eval_steps_per_second": 0.298, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 3.5, |
|
"learning_rate": 3.867370395306068e-05, |
|
"loss": 0.1066, |
|
"step": 350 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 1000, |
|
"num_train_epochs": 10, |
|
"save_steps": 50, |
|
"total_flos": 4.924285778959565e+16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|