PubMedBert-abstract-cord19-v2 / trainer_state.json
pritamdeka's picture
Initial Commit
843a927
raw
history blame
3.23 kB
{
"best_metric": 1.2424818277359009,
"best_model_checkpoint": "/content/pubmedbert-abstract-cord19/checkpoint-5000",
"epoch": 4.0,
"global_step": 37788,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.53,
"learning_rate": 2.4970000000000003e-05,
"loss": 1.27,
"step": 5000
},
{
"epoch": 0.53,
"eval_accuracy": 0.7236279677533664,
"eval_loss": 1.2424818277359009,
"eval_runtime": 183.46,
"eval_samples_per_second": 44.206,
"eval_steps_per_second": 2.764,
"step": 5000
},
{
"epoch": 1.06,
"learning_rate": 4.9965e-05,
"loss": 1.2634,
"step": 10000
},
{
"epoch": 1.06,
"eval_accuracy": 0.7140570760001546,
"eval_loss": 1.312286138534546,
"eval_runtime": 183.3534,
"eval_samples_per_second": 44.232,
"eval_steps_per_second": 2.765,
"step": 10000
},
{
"epoch": 1.59,
"learning_rate": 4.1024902835756445e-05,
"loss": 1.3041,
"step": 15000
},
{
"epoch": 1.59,
"eval_accuracy": 0.7072068247593172,
"eval_loss": 1.3583375215530396,
"eval_runtime": 183.3479,
"eval_samples_per_second": 44.233,
"eval_steps_per_second": 2.765,
"step": 15000
},
{
"epoch": 2.12,
"learning_rate": 3.2033611630919824e-05,
"loss": 1.3829,
"step": 20000
},
{
"epoch": 2.12,
"eval_accuracy": 0.7120832508972437,
"eval_loss": 1.3589657545089722,
"eval_runtime": 183.3751,
"eval_samples_per_second": 44.226,
"eval_steps_per_second": 2.765,
"step": 20000
},
{
"epoch": 2.65,
"learning_rate": 2.303692241255218e-05,
"loss": 1.3069,
"step": 25000
},
{
"epoch": 2.65,
"eval_accuracy": 0.7153998687867195,
"eval_loss": 1.3505500555038452,
"eval_runtime": 183.3915,
"eval_samples_per_second": 44.222,
"eval_steps_per_second": 2.765,
"step": 25000
},
{
"epoch": 3.18,
"learning_rate": 1.4040233194184541e-05,
"loss": 1.2921,
"step": 30000
},
{
"epoch": 3.18,
"eval_accuracy": 0.7159658967235916,
"eval_loss": 1.3447710275650024,
"eval_runtime": 183.3443,
"eval_samples_per_second": 44.234,
"eval_steps_per_second": 2.765,
"step": 30000
},
{
"epoch": 3.7,
"learning_rate": 5.0471426515042465e-06,
"loss": 1.2731,
"step": 35000
},
{
"epoch": 3.7,
"eval_accuracy": 0.7177905274023574,
"eval_loss": 1.3374903202056885,
"eval_runtime": 183.3182,
"eval_samples_per_second": 44.24,
"eval_steps_per_second": 2.766,
"step": 35000
},
{
"epoch": 4.0,
"step": 37788,
"total_flos": 1.3985984051532e+17,
"train_loss": 1.2965481223991413,
"train_runtime": 37723.3327,
"train_samples_per_second": 16.027,
"train_steps_per_second": 1.002
}
],
"max_steps": 37788,
"num_train_epochs": 4,
"total_flos": 1.3985984051532e+17,
"trial_name": null,
"trial_params": null
}