DunnBC22's picture
Training in progress, epoch 5
5705b05
raw
history blame
2.54 kB
{
"best_metric": 0.836962977801342,
"best_model_checkpoint": "distilbert-base-uncased-US_Airline_Twitter_Sentiment_Analysis/checkpoint-460",
"epoch": 5.0,
"global_step": 460,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.01,
"learning_rate": 1.997282608695652e-05,
"loss": 1.0978,
"step": 1
},
{
"epoch": 1.0,
"learning_rate": 1.7500000000000002e-05,
"loss": 0.8015,
"step": 92
},
{
"epoch": 1.0,
"eval_accuracy": 0.7952054794520548,
"eval_f1_score": 0.8018452790126995,
"eval_loss": 0.5483363270759583,
"eval_runtime": 87.3239,
"eval_samples_per_second": 16.719,
"eval_steps_per_second": 0.137,
"step": 92
},
{
"epoch": 2.0,
"learning_rate": 1.5000000000000002e-05,
"loss": 0.4795,
"step": 184
},
{
"epoch": 2.0,
"eval_accuracy": 0.8232876712328767,
"eval_f1_score": 0.8266497180049651,
"eval_loss": 0.4993366301059723,
"eval_runtime": 86.9513,
"eval_samples_per_second": 16.791,
"eval_steps_per_second": 0.138,
"step": 184
},
{
"epoch": 3.0,
"learning_rate": 1.25e-05,
"loss": 0.3995,
"step": 276
},
{
"epoch": 3.0,
"eval_accuracy": 0.8205479452054795,
"eval_f1_score": 0.8159502183368431,
"eval_loss": 0.588766872882843,
"eval_runtime": 86.0412,
"eval_samples_per_second": 16.969,
"eval_steps_per_second": 0.139,
"step": 276
},
{
"epoch": 4.0,
"learning_rate": 1e-05,
"loss": 0.339,
"step": 368
},
{
"epoch": 4.0,
"eval_accuracy": 0.834931506849315,
"eval_f1_score": 0.8350190343859925,
"eval_loss": 0.4934794306755066,
"eval_runtime": 86.1523,
"eval_samples_per_second": 16.947,
"eval_steps_per_second": 0.139,
"step": 368
},
{
"epoch": 5.0,
"learning_rate": 7.500000000000001e-06,
"loss": 0.2857,
"step": 460
},
{
"epoch": 5.0,
"eval_accuracy": 0.8335616438356165,
"eval_f1_score": 0.836962977801342,
"eval_loss": 0.5100029110908508,
"eval_runtime": 86.4788,
"eval_samples_per_second": 16.883,
"eval_steps_per_second": 0.139,
"step": 460
}
],
"max_steps": 736,
"num_train_epochs": 8,
"total_flos": 452576211558444.0,
"trial_name": null,
"trial_params": null
}