bert-ajgt / trainer_state.json
mofawzy's picture
add model for ajgt dataset
017450e
{
"best_metric": 0.16990411281585693,
"best_model_checkpoint": "BERT-AJGT/checkpoint-30",
"epoch": 6.0,
"global_step": 36,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"learning_rate": 1.9e-05,
"loss": 0.5804,
"step": 6
},
{
"epoch": 1.0,
"eval_accuracy": 0.8703703703703703,
"eval_f1": 0.864516129032258,
"eval_loss": 0.436138391494751,
"eval_precision": 0.9054054054054054,
"eval_recall": 0.8271604938271605,
"eval_runtime": 0.0502,
"eval_samples_per_second": 3228.319,
"eval_steps_per_second": 19.928,
"step": 6
},
{
"epoch": 2.0,
"learning_rate": 1.8e-05,
"loss": 0.3676,
"step": 12
},
{
"epoch": 2.0,
"eval_accuracy": 0.9135802469135802,
"eval_f1": 0.9135802469135802,
"eval_loss": 0.279801607131958,
"eval_precision": 0.9135802469135802,
"eval_recall": 0.9135802469135802,
"eval_runtime": 0.0493,
"eval_samples_per_second": 3288.154,
"eval_steps_per_second": 20.297,
"step": 12
},
{
"epoch": 3.0,
"learning_rate": 1.7e-05,
"loss": 0.2252,
"step": 18
},
{
"epoch": 3.0,
"eval_accuracy": 0.9320987654320988,
"eval_f1": 0.9325153374233128,
"eval_loss": 0.20178434252738953,
"eval_precision": 0.926829268292683,
"eval_recall": 0.9382716049382716,
"eval_runtime": 0.0509,
"eval_samples_per_second": 3184.457,
"eval_steps_per_second": 19.657,
"step": 18
},
{
"epoch": 4.0,
"learning_rate": 1.6000000000000003e-05,
"loss": 0.138,
"step": 24
},
{
"epoch": 4.0,
"eval_accuracy": 0.9382716049382716,
"eval_f1": 0.9390243902439024,
"eval_loss": 0.17006997764110565,
"eval_precision": 0.927710843373494,
"eval_recall": 0.9506172839506173,
"eval_runtime": 0.0494,
"eval_samples_per_second": 3277.401,
"eval_steps_per_second": 20.231,
"step": 24
},
{
"epoch": 5.0,
"learning_rate": 1.5000000000000002e-05,
"loss": 0.0975,
"step": 30
},
{
"epoch": 5.0,
"eval_accuracy": 0.9444444444444444,
"eval_f1": 0.9454545454545454,
"eval_loss": 0.16990411281585693,
"eval_precision": 0.9285714285714286,
"eval_recall": 0.9629629629629629,
"eval_runtime": 0.0525,
"eval_samples_per_second": 3084.733,
"eval_steps_per_second": 19.042,
"step": 30
},
{
"epoch": 6.0,
"learning_rate": 1.4e-05,
"loss": 0.0679,
"step": 36
},
{
"epoch": 6.0,
"eval_accuracy": 0.9567901234567902,
"eval_f1": 0.9575757575757574,
"eval_loss": 0.17372706532478333,
"eval_precision": 0.9404761904761905,
"eval_recall": 0.9753086419753086,
"eval_runtime": 0.0501,
"eval_samples_per_second": 3232.281,
"eval_steps_per_second": 19.952,
"step": 36
}
],
"max_steps": 120,
"num_train_epochs": 20,
"total_flos": 420675008084580.0,
"trial_name": null,
"trial_params": null
}