PubMedBert-abstract-cord19 / trainer_state.json
pritamdeka's picture
Updated Commit
f62d69e
raw history blame
No virus
7.19 kB
{
"best_metric": 1.3029065132141113,
"best_model_checkpoint": "/content/pubmedbert-abstract-cord19/checkpoint-100000",
"epoch": 3.0,
"global_step": 102084,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.15,
"learning_rate": 2.496e-05,
"loss": 1.3774,
"step": 5000
},
{
"epoch": 0.15,
"eval_loss": 1.3211560249328613,
"eval_runtime": 169.3169,
"eval_samples_per_second": 86.258,
"eval_steps_per_second": 10.785,
"step": 5000
},
{
"epoch": 0.29,
"learning_rate": 4.9945000000000004e-05,
"loss": 1.3937,
"step": 10000
},
{
"epoch": 0.29,
"eval_loss": 1.4059454202651978,
"eval_runtime": 169.3069,
"eval_samples_per_second": 86.263,
"eval_steps_per_second": 10.785,
"step": 10000
},
{
"epoch": 0.44,
"learning_rate": 4.7291603318709e-05,
"loss": 1.6812,
"step": 15000
},
{
"epoch": 0.44,
"eval_loss": 1.6174367666244507,
"eval_runtime": 168.9751,
"eval_samples_per_second": 86.433,
"eval_steps_per_second": 10.806,
"step": 15000
},
{
"epoch": 0.59,
"learning_rate": 4.457723382998132e-05,
"loss": 1.4712,
"step": 20000
},
{
"epoch": 0.59,
"eval_loss": 1.4382678270339966,
"eval_runtime": 168.9508,
"eval_samples_per_second": 86.445,
"eval_steps_per_second": 10.808,
"step": 20000
},
{
"epoch": 0.73,
"learning_rate": 4.186340732374788e-05,
"loss": 1.4293,
"step": 25000
},
{
"epoch": 0.73,
"eval_loss": 1.4356446266174316,
"eval_runtime": 168.8828,
"eval_samples_per_second": 86.48,
"eval_steps_per_second": 10.812,
"step": 25000
},
{
"epoch": 0.88,
"learning_rate": 3.914849485252596e-05,
"loss": 1.4155,
"step": 30000
},
{
"epoch": 0.88,
"eval_loss": 1.4283361434936523,
"eval_runtime": 169.0831,
"eval_samples_per_second": 86.378,
"eval_steps_per_second": 10.799,
"step": 30000
},
{
"epoch": 1.03,
"learning_rate": 3.6435754311281005e-05,
"loss": 1.3963,
"step": 35000
},
{
"epoch": 1.03,
"eval_loss": 1.4134665727615356,
"eval_runtime": 169.088,
"eval_samples_per_second": 86.375,
"eval_steps_per_second": 10.799,
"step": 35000
},
{
"epoch": 1.18,
"learning_rate": 3.372138482255332e-05,
"loss": 1.3718,
"step": 40000
},
{
"epoch": 1.18,
"eval_loss": 1.3948187828063965,
"eval_runtime": 169.0271,
"eval_samples_per_second": 86.406,
"eval_steps_per_second": 10.803,
"step": 40000
},
{
"epoch": 1.32,
"learning_rate": 3.1006472351331396e-05,
"loss": 1.369,
"step": 45000
},
{
"epoch": 1.32,
"eval_loss": 1.3961154222488403,
"eval_runtime": 169.0308,
"eval_samples_per_second": 86.404,
"eval_steps_per_second": 10.803,
"step": 45000
},
{
"epoch": 1.47,
"learning_rate": 2.8292645845097955e-05,
"loss": 1.354,
"step": 50000
},
{
"epoch": 1.47,
"eval_loss": 1.378829836845398,
"eval_runtime": 169.1643,
"eval_samples_per_second": 86.336,
"eval_steps_per_second": 10.794,
"step": 50000
},
{
"epoch": 1.62,
"learning_rate": 2.5579362321358764e-05,
"loss": 1.3399,
"step": 55000
},
{
"epoch": 1.62,
"eval_loss": 1.3865987062454224,
"eval_runtime": 169.0688,
"eval_samples_per_second": 86.385,
"eval_steps_per_second": 10.8,
"step": 55000
},
{
"epoch": 1.76,
"learning_rate": 2.2866078797619567e-05,
"loss": 1.3289,
"step": 60000
},
{
"epoch": 1.76,
"eval_loss": 1.362976312637329,
"eval_runtime": 169.1136,
"eval_samples_per_second": 86.362,
"eval_steps_per_second": 10.797,
"step": 60000
},
{
"epoch": 1.91,
"learning_rate": 2.0151709308891882e-05,
"loss": 1.3155,
"step": 65000
},
{
"epoch": 1.91,
"eval_loss": 1.3609150648117065,
"eval_runtime": 169.0556,
"eval_samples_per_second": 86.392,
"eval_steps_per_second": 10.801,
"step": 65000
},
{
"epoch": 2.06,
"learning_rate": 1.743788280265844e-05,
"loss": 1.2976,
"step": 70000
},
{
"epoch": 2.06,
"eval_loss": 1.3489034175872803,
"eval_runtime": 169.1114,
"eval_samples_per_second": 86.363,
"eval_steps_per_second": 10.798,
"step": 70000
},
{
"epoch": 2.2,
"learning_rate": 1.4724599278919249e-05,
"loss": 1.2783,
"step": 75000
},
{
"epoch": 2.2,
"eval_loss": 1.3333380222320557,
"eval_runtime": 169.0576,
"eval_samples_per_second": 86.391,
"eval_steps_per_second": 10.801,
"step": 75000
},
{
"epoch": 2.35,
"learning_rate": 1.2009686807697321e-05,
"loss": 1.2696,
"step": 80000
},
{
"epoch": 2.35,
"eval_loss": 1.3259785175323486,
"eval_runtime": 168.9798,
"eval_samples_per_second": 86.43,
"eval_steps_per_second": 10.806,
"step": 80000
},
{
"epoch": 2.5,
"learning_rate": 9.29586030146388e-06,
"loss": 1.2607,
"step": 85000
},
{
"epoch": 2.5,
"eval_loss": 1.3232313394546509,
"eval_runtime": 168.7484,
"eval_samples_per_second": 86.549,
"eval_steps_per_second": 10.821,
"step": 85000
},
{
"epoch": 2.64,
"learning_rate": 6.582033795230443e-06,
"loss": 1.2547,
"step": 90000
},
{
"epoch": 2.64,
"eval_loss": 1.3033902645111084,
"eval_runtime": 168.7446,
"eval_samples_per_second": 86.551,
"eval_steps_per_second": 10.821,
"step": 90000
},
{
"epoch": 2.79,
"learning_rate": 3.867664306502758e-06,
"loss": 1.2495,
"step": 95000
},
{
"epoch": 2.79,
"eval_loss": 1.3035175800323486,
"eval_runtime": 168.7573,
"eval_samples_per_second": 86.544,
"eval_steps_per_second": 10.82,
"step": 95000
},
{
"epoch": 2.94,
"learning_rate": 1.1538378002693194e-06,
"loss": 1.2404,
"step": 100000
},
{
"epoch": 2.94,
"eval_loss": 1.3029065132141113,
"eval_runtime": 168.8016,
"eval_samples_per_second": 86.522,
"eval_steps_per_second": 10.817,
"step": 100000
},
{
"epoch": 3.0,
"step": 102084,
"total_flos": 1.04954332801725e+17,
"train_loss": 1.35220570928797,
"train_runtime": 32542.0864,
"train_samples_per_second": 25.095,
"train_steps_per_second": 3.137
}
],
"max_steps": 102084,
"num_train_epochs": 3,
"total_flos": 1.04954332801725e+17,
"trial_name": null,
"trial_params": null
}