|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 1.2597631645250693, |
|
"eval_steps": 500, |
|
"global_step": 10000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 9.5e-05, |
|
"loss": 1.6085, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"eval_loss": 1.3112831115722656, |
|
"eval_runtime": 3.3067, |
|
"eval_samples_per_second": 30.241, |
|
"eval_steps_per_second": 3.931, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 9e-05, |
|
"loss": 1.5215, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"eval_loss": 1.5054197311401367, |
|
"eval_runtime": 3.1798, |
|
"eval_samples_per_second": 31.449, |
|
"eval_steps_per_second": 4.088, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 8.5e-05, |
|
"loss": 1.4697, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"eval_loss": 1.32753324508667, |
|
"eval_runtime": 3.3247, |
|
"eval_samples_per_second": 30.078, |
|
"eval_steps_per_second": 3.91, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 8e-05, |
|
"loss": 1.4948, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"eval_loss": 1.4151370525360107, |
|
"eval_runtime": 3.174, |
|
"eval_samples_per_second": 31.506, |
|
"eval_steps_per_second": 4.096, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 7.500000000000001e-05, |
|
"loss": 1.4787, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"eval_loss": 1.1171963214874268, |
|
"eval_runtime": 3.1742, |
|
"eval_samples_per_second": 31.504, |
|
"eval_steps_per_second": 4.096, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 7e-05, |
|
"loss": 1.4196, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"eval_loss": 1.495429277420044, |
|
"eval_runtime": 3.1924, |
|
"eval_samples_per_second": 31.325, |
|
"eval_steps_per_second": 4.072, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 6.500000000000001e-05, |
|
"loss": 1.4509, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"eval_loss": 1.3065623044967651, |
|
"eval_runtime": 3.1668, |
|
"eval_samples_per_second": 31.577, |
|
"eval_steps_per_second": 4.105, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 6e-05, |
|
"loss": 1.4333, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"eval_loss": 1.4525870084762573, |
|
"eval_runtime": 3.1817, |
|
"eval_samples_per_second": 31.43, |
|
"eval_steps_per_second": 4.086, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 5.500000000000001e-05, |
|
"loss": 1.4404, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"eval_loss": 1.3187968730926514, |
|
"eval_runtime": 3.191, |
|
"eval_samples_per_second": 31.338, |
|
"eval_steps_per_second": 4.074, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 5e-05, |
|
"loss": 1.4408, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"eval_loss": 1.1972750425338745, |
|
"eval_runtime": 3.1767, |
|
"eval_samples_per_second": 31.479, |
|
"eval_steps_per_second": 4.092, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 4.5e-05, |
|
"loss": 1.4296, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"eval_loss": 1.3503276109695435, |
|
"eval_runtime": 3.1871, |
|
"eval_samples_per_second": 31.376, |
|
"eval_steps_per_second": 4.079, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 4e-05, |
|
"loss": 1.3828, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"eval_loss": 1.1146811246871948, |
|
"eval_runtime": 3.1707, |
|
"eval_samples_per_second": 31.538, |
|
"eval_steps_per_second": 4.1, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 3.5e-05, |
|
"loss": 1.482, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"eval_loss": 1.4396498203277588, |
|
"eval_runtime": 3.2045, |
|
"eval_samples_per_second": 31.206, |
|
"eval_steps_per_second": 4.057, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"learning_rate": 3e-05, |
|
"loss": 1.4022, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"eval_loss": 1.2612046003341675, |
|
"eval_runtime": 3.187, |
|
"eval_samples_per_second": 31.378, |
|
"eval_steps_per_second": 4.079, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 2.5e-05, |
|
"loss": 1.4278, |
|
"step": 7500 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"eval_loss": 1.3243725299835205, |
|
"eval_runtime": 3.2028, |
|
"eval_samples_per_second": 31.223, |
|
"eval_steps_per_second": 4.059, |
|
"step": 7500 |
|
}, |
|
{ |
|
"epoch": 1.01, |
|
"learning_rate": 2e-05, |
|
"loss": 1.3767, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 1.01, |
|
"eval_loss": 1.3397793769836426, |
|
"eval_runtime": 3.2017, |
|
"eval_samples_per_second": 31.234, |
|
"eval_steps_per_second": 4.06, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 1.07, |
|
"learning_rate": 1.5e-05, |
|
"loss": 1.346, |
|
"step": 8500 |
|
}, |
|
{ |
|
"epoch": 1.07, |
|
"eval_loss": 1.278018593788147, |
|
"eval_runtime": 3.373, |
|
"eval_samples_per_second": 29.647, |
|
"eval_steps_per_second": 3.854, |
|
"step": 8500 |
|
}, |
|
{ |
|
"epoch": 1.13, |
|
"learning_rate": 1e-05, |
|
"loss": 1.3513, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 1.13, |
|
"eval_loss": 1.3960318565368652, |
|
"eval_runtime": 3.2292, |
|
"eval_samples_per_second": 30.967, |
|
"eval_steps_per_second": 4.026, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 1.2, |
|
"learning_rate": 5e-06, |
|
"loss": 1.3398, |
|
"step": 9500 |
|
}, |
|
{ |
|
"epoch": 1.2, |
|
"eval_loss": 1.2023816108703613, |
|
"eval_runtime": 3.2042, |
|
"eval_samples_per_second": 31.209, |
|
"eval_steps_per_second": 4.057, |
|
"step": 9500 |
|
}, |
|
{ |
|
"epoch": 1.26, |
|
"learning_rate": 0.0, |
|
"loss": 1.3608, |
|
"step": 10000 |
|
}, |
|
{ |
|
"epoch": 1.26, |
|
"eval_loss": 1.2703720331192017, |
|
"eval_runtime": 3.2009, |
|
"eval_samples_per_second": 31.241, |
|
"eval_steps_per_second": 4.061, |
|
"step": 10000 |
|
} |
|
], |
|
"logging_steps": 500, |
|
"max_steps": 10000, |
|
"num_train_epochs": 2, |
|
"save_steps": 5000, |
|
"total_flos": 1.3694633480945664e+17, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|