xlm-roberta-base-conll2002-dutch / trainer_state.json
wpnbos's picture
First model commit
01819a3
raw history blame
No virus
2.91 kB
{
"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
}