|
{
|
|
"best_metric": 0.06587601453065872,
|
|
"best_model_checkpoint": "test-ner/checkpoint-2636",
|
|
"epoch": 4.0,
|
|
"global_step": 5272,
|
|
"is_hyper_param_search": false,
|
|
"is_local_process_zero": true,
|
|
"is_world_process_zero": true,
|
|
"log_history": [
|
|
{
|
|
"epoch": 0.38,
|
|
"learning_rate": 9.241274658573596e-06,
|
|
"loss": 0.2967,
|
|
"step": 500
|
|
},
|
|
{
|
|
"epoch": 0.76,
|
|
"learning_rate": 8.482549317147192e-06,
|
|
"loss": 0.0903,
|
|
"step": 1000
|
|
},
|
|
{
|
|
"epoch": 1.0,
|
|
"eval_accuracy": 0.9800172512938471,
|
|
"eval_f1": 0.8510216941218831,
|
|
"eval_loss": 0.07860440760850906,
|
|
"eval_precision": 0.8489583333333334,
|
|
"eval_recall": 0.8530951091860675,
|
|
"eval_runtime": 11.6301,
|
|
"eval_samples_per_second": 249.008,
|
|
"step": 1318
|
|
},
|
|
{
|
|
"epoch": 1.14,
|
|
"learning_rate": 7.72382397572079e-06,
|
|
"loss": 0.0695,
|
|
"step": 1500
|
|
},
|
|
{
|
|
"epoch": 1.52,
|
|
"learning_rate": 6.965098634294386e-06,
|
|
"loss": 0.0487,
|
|
"step": 2000
|
|
},
|
|
{
|
|
"epoch": 1.9,
|
|
"learning_rate": 6.206373292867982e-06,
|
|
"loss": 0.0521,
|
|
"step": 2500
|
|
},
|
|
{
|
|
"epoch": 2.0,
|
|
"eval_accuracy": 0.9845816561242093,
|
|
"eval_f1": 0.8887689955939214,
|
|
"eval_loss": 0.06587601453065872,
|
|
"eval_precision": 0.885663082437276,
|
|
"eval_recall": 0.8918967695361848,
|
|
"eval_runtime": 11.691,
|
|
"eval_samples_per_second": 247.711,
|
|
"step": 2636
|
|
},
|
|
{
|
|
"epoch": 2.28,
|
|
"learning_rate": 5.447647951441578e-06,
|
|
"loss": 0.0388,
|
|
"step": 3000
|
|
},
|
|
{
|
|
"epoch": 2.66,
|
|
"learning_rate": 4.688922610015174e-06,
|
|
"loss": 0.0344,
|
|
"step": 3500
|
|
},
|
|
{
|
|
"epoch": 3.0,
|
|
"eval_accuracy": 0.9849949683726279,
|
|
"eval_f1": 0.8925949424453912,
|
|
"eval_loss": 0.06827788800001144,
|
|
"eval_precision": 0.8965768390386016,
|
|
"eval_recall": 0.8886482584371053,
|
|
"eval_runtime": 11.7404,
|
|
"eval_samples_per_second": 246.67,
|
|
"step": 3954
|
|
},
|
|
{
|
|
"epoch": 3.03,
|
|
"learning_rate": 3.930197268588772e-06,
|
|
"loss": 0.0297,
|
|
"step": 4000
|
|
},
|
|
{
|
|
"epoch": 3.41,
|
|
"learning_rate": 3.1714719271623673e-06,
|
|
"loss": 0.0255,
|
|
"step": 4500
|
|
},
|
|
{
|
|
"epoch": 3.79,
|
|
"learning_rate": 2.4127465857359637e-06,
|
|
"loss": 0.0235,
|
|
"step": 5000
|
|
},
|
|
{
|
|
"epoch": 4.0,
|
|
"eval_accuracy": 0.9865943070730305,
|
|
"eval_f1": 0.904637078550122,
|
|
"eval_loss": 0.06835205852985382,
|
|
"eval_precision": 0.90619340818544,
|
|
"eval_recall": 0.9030860855441256,
|
|
"eval_runtime": 11.7952,
|
|
"eval_samples_per_second": 245.524,
|
|
"step": 5272
|
|
}
|
|
],
|
|
"max_steps": 6590,
|
|
"num_train_epochs": 5,
|
|
"total_flos": 1452761017373610.0,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|
|
|