hBERTv1_qnli / trainer_state.json
gokuls's picture
End of training
b1261c8
{
"best_metric": 0.4198276698589325,
"best_model_checkpoint": "hBERTv1_qnli/checkpoint-1230",
"epoch": 8.0,
"global_step": 3280,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"learning_rate": 4.9e-05,
"loss": 0.6667,
"step": 410
},
{
"epoch": 1.0,
"eval_accuracy": 0.6873512721947648,
"eval_loss": 0.5954663157463074,
"eval_runtime": 6.9854,
"eval_samples_per_second": 782.059,
"eval_steps_per_second": 3.149,
"step": 410
},
{
"epoch": 2.0,
"learning_rate": 4.8e-05,
"loss": 0.4998,
"step": 820
},
{
"epoch": 2.0,
"eval_accuracy": 0.794801391177009,
"eval_loss": 0.44860708713531494,
"eval_runtime": 6.9353,
"eval_samples_per_second": 787.714,
"eval_steps_per_second": 3.172,
"step": 820
},
{
"epoch": 3.0,
"learning_rate": 4.7e-05,
"loss": 0.3985,
"step": 1230
},
{
"epoch": 3.0,
"eval_accuracy": 0.8112758557569101,
"eval_loss": 0.4198276698589325,
"eval_runtime": 6.9396,
"eval_samples_per_second": 787.224,
"eval_steps_per_second": 3.17,
"step": 1230
},
{
"epoch": 4.0,
"learning_rate": 4.600000000000001e-05,
"loss": 0.3106,
"step": 1640
},
{
"epoch": 4.0,
"eval_accuracy": 0.7865641588870584,
"eval_loss": 0.48407182097435,
"eval_runtime": 6.9525,
"eval_samples_per_second": 785.765,
"eval_steps_per_second": 3.164,
"step": 1640
},
{
"epoch": 5.0,
"learning_rate": 4.5e-05,
"loss": 0.2286,
"step": 2050
},
{
"epoch": 5.0,
"eval_accuracy": 0.790591250228812,
"eval_loss": 0.5340340733528137,
"eval_runtime": 6.9383,
"eval_samples_per_second": 787.374,
"eval_steps_per_second": 3.171,
"step": 2050
},
{
"epoch": 6.0,
"learning_rate": 4.4000000000000006e-05,
"loss": 0.1662,
"step": 2460
},
{
"epoch": 6.0,
"eval_accuracy": 0.7728354384038074,
"eval_loss": 0.6281628608703613,
"eval_runtime": 6.9508,
"eval_samples_per_second": 785.951,
"eval_steps_per_second": 3.165,
"step": 2460
},
{
"epoch": 7.0,
"learning_rate": 4.3e-05,
"loss": 0.1237,
"step": 2870
},
{
"epoch": 7.0,
"eval_accuracy": 0.7752150832875709,
"eval_loss": 0.6678447723388672,
"eval_runtime": 6.9465,
"eval_samples_per_second": 786.441,
"eval_steps_per_second": 3.167,
"step": 2870
},
{
"epoch": 8.0,
"learning_rate": 4.2e-05,
"loss": 0.0945,
"step": 3280
},
{
"epoch": 8.0,
"eval_accuracy": 0.7752150832875709,
"eval_loss": 0.7668098211288452,
"eval_runtime": 6.9539,
"eval_samples_per_second": 785.607,
"eval_steps_per_second": 3.164,
"step": 3280
},
{
"epoch": 8.0,
"step": 3280,
"total_flos": 1.0584157486002995e+17,
"train_loss": 0.3110694268854653,
"train_runtime": 3314.8577,
"train_samples_per_second": 1579.902,
"train_steps_per_second": 6.184
}
],
"max_steps": 20500,
"num_train_epochs": 50,
"total_flos": 1.0584157486002995e+17,
"trial_name": null,
"trial_params": null
}