|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 2.5195263290501386, |
|
"global_step": 10000, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 9.5e-05, |
|
"loss": 0.8107, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"eval_loss": 0.6702454090118408, |
|
"eval_runtime": 4.5337, |
|
"eval_samples_per_second": 66.17, |
|
"eval_steps_per_second": 33.085, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 9e-05, |
|
"loss": 0.717, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"eval_loss": 0.6662179827690125, |
|
"eval_runtime": 5.109, |
|
"eval_samples_per_second": 58.719, |
|
"eval_steps_per_second": 29.36, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 8.5e-05, |
|
"loss": 0.7212, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"eval_loss": 0.6485440135002136, |
|
"eval_runtime": 4.6475, |
|
"eval_samples_per_second": 64.551, |
|
"eval_steps_per_second": 32.276, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 8e-05, |
|
"loss": 0.7094, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"eval_loss": 0.6365593671798706, |
|
"eval_runtime": 4.9661, |
|
"eval_samples_per_second": 60.41, |
|
"eval_steps_per_second": 30.205, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 7.500000000000001e-05, |
|
"loss": 0.7081, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"eval_loss": 0.6411292552947998, |
|
"eval_runtime": 5.0449, |
|
"eval_samples_per_second": 59.466, |
|
"eval_steps_per_second": 29.733, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 7e-05, |
|
"loss": 0.6803, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"eval_loss": 0.629453718662262, |
|
"eval_runtime": 5.1787, |
|
"eval_samples_per_second": 57.929, |
|
"eval_steps_per_second": 28.965, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"learning_rate": 6.500000000000001e-05, |
|
"loss": 0.6886, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"eval_loss": 0.6171028017997742, |
|
"eval_runtime": 6.0926, |
|
"eval_samples_per_second": 49.24, |
|
"eval_steps_per_second": 24.62, |
|
"step": 3500 |
|
}, |
|
{ |
|
"epoch": 1.01, |
|
"learning_rate": 6e-05, |
|
"loss": 0.673, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 1.01, |
|
"eval_loss": 0.6199834942817688, |
|
"eval_runtime": 4.5562, |
|
"eval_samples_per_second": 65.844, |
|
"eval_steps_per_second": 32.922, |
|
"step": 4000 |
|
}, |
|
{ |
|
"epoch": 1.13, |
|
"learning_rate": 5.500000000000001e-05, |
|
"loss": 0.6123, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 1.13, |
|
"eval_loss": 0.615871250629425, |
|
"eval_runtime": 4.3914, |
|
"eval_samples_per_second": 68.315, |
|
"eval_steps_per_second": 34.158, |
|
"step": 4500 |
|
}, |
|
{ |
|
"epoch": 1.26, |
|
"learning_rate": 5e-05, |
|
"loss": 0.6036, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 1.26, |
|
"eval_loss": 0.6081268191337585, |
|
"eval_runtime": 4.202, |
|
"eval_samples_per_second": 71.395, |
|
"eval_steps_per_second": 35.697, |
|
"step": 5000 |
|
}, |
|
{ |
|
"epoch": 1.39, |
|
"learning_rate": 4.5e-05, |
|
"loss": 0.5854, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 1.39, |
|
"eval_loss": 0.610887885093689, |
|
"eval_runtime": 4.5399, |
|
"eval_samples_per_second": 66.08, |
|
"eval_steps_per_second": 33.04, |
|
"step": 5500 |
|
}, |
|
{ |
|
"epoch": 1.51, |
|
"learning_rate": 4e-05, |
|
"loss": 0.5936, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 1.51, |
|
"eval_loss": 0.6057543158531189, |
|
"eval_runtime": 5.4963, |
|
"eval_samples_per_second": 54.582, |
|
"eval_steps_per_second": 27.291, |
|
"step": 6000 |
|
}, |
|
{ |
|
"epoch": 1.64, |
|
"learning_rate": 3.5e-05, |
|
"loss": 0.5917, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 1.64, |
|
"eval_loss": 0.6038711071014404, |
|
"eval_runtime": 4.4545, |
|
"eval_samples_per_second": 67.348, |
|
"eval_steps_per_second": 33.674, |
|
"step": 6500 |
|
}, |
|
{ |
|
"epoch": 1.76, |
|
"learning_rate": 3e-05, |
|
"loss": 0.5848, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 1.76, |
|
"eval_loss": 0.6017142534255981, |
|
"eval_runtime": 4.4106, |
|
"eval_samples_per_second": 68.018, |
|
"eval_steps_per_second": 34.009, |
|
"step": 7000 |
|
}, |
|
{ |
|
"epoch": 1.89, |
|
"learning_rate": 2.5e-05, |
|
"loss": 0.5875, |
|
"step": 7500 |
|
}, |
|
{ |
|
"epoch": 1.89, |
|
"eval_loss": 0.6017369627952576, |
|
"eval_runtime": 4.4904, |
|
"eval_samples_per_second": 66.809, |
|
"eval_steps_per_second": 33.404, |
|
"step": 7500 |
|
}, |
|
{ |
|
"epoch": 2.02, |
|
"learning_rate": 2e-05, |
|
"loss": 0.5975, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 2.02, |
|
"eval_loss": 0.5987575054168701, |
|
"eval_runtime": 4.1962, |
|
"eval_samples_per_second": 71.493, |
|
"eval_steps_per_second": 35.747, |
|
"step": 8000 |
|
}, |
|
{ |
|
"epoch": 2.14, |
|
"learning_rate": 1.5e-05, |
|
"loss": 0.5551, |
|
"step": 8500 |
|
}, |
|
{ |
|
"epoch": 2.14, |
|
"eval_loss": 0.6018974781036377, |
|
"eval_runtime": 4.1728, |
|
"eval_samples_per_second": 71.894, |
|
"eval_steps_per_second": 35.947, |
|
"step": 8500 |
|
}, |
|
{ |
|
"epoch": 2.27, |
|
"learning_rate": 1e-05, |
|
"loss": 0.5338, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 2.27, |
|
"eval_loss": 0.600771963596344, |
|
"eval_runtime": 4.5111, |
|
"eval_samples_per_second": 66.502, |
|
"eval_steps_per_second": 33.251, |
|
"step": 9000 |
|
}, |
|
{ |
|
"epoch": 2.39, |
|
"learning_rate": 5e-06, |
|
"loss": 0.5648, |
|
"step": 9500 |
|
}, |
|
{ |
|
"epoch": 2.39, |
|
"eval_loss": 0.599320650100708, |
|
"eval_runtime": 4.3392, |
|
"eval_samples_per_second": 69.138, |
|
"eval_steps_per_second": 34.569, |
|
"step": 9500 |
|
}, |
|
{ |
|
"epoch": 2.52, |
|
"learning_rate": 0.0, |
|
"loss": 0.5328, |
|
"step": 10000 |
|
}, |
|
{ |
|
"epoch": 2.52, |
|
"eval_loss": 0.6002569794654846, |
|
"eval_runtime": 5.4729, |
|
"eval_samples_per_second": 54.816, |
|
"eval_steps_per_second": 27.408, |
|
"step": 10000 |
|
} |
|
], |
|
"max_steps": 10000, |
|
"num_train_epochs": 3, |
|
"total_flos": 3.963545541632102e+16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|