bert_uncased_L-2_H-256_A-4_mrpc / trainer_state.json
gokulsrinivasagan's picture
End of training
c4e3810 verified
{
"best_metric": 0.5344022512435913,
"best_model_checkpoint": "bert_uncased_L-2_H-256_A-4_mrpc/checkpoint-90",
"epoch": 11.0,
"eval_steps": 500,
"global_step": 165,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"grad_norm": 0.9149664044380188,
"learning_rate": 4.9e-05,
"loss": 0.619,
"step": 15
},
{
"epoch": 1.0,
"eval_accuracy": 0.6887254901960784,
"eval_combined_score": 0.7516620151710319,
"eval_f1": 0.8145985401459854,
"eval_loss": 0.5955595374107361,
"eval_runtime": 0.1436,
"eval_samples_per_second": 2840.349,
"eval_steps_per_second": 13.923,
"step": 15
},
{
"epoch": 2.0,
"grad_norm": 1.7355875968933105,
"learning_rate": 4.8e-05,
"loss": 0.5893,
"step": 30
},
{
"epoch": 2.0,
"eval_accuracy": 0.7009803921568627,
"eval_combined_score": 0.7594454199590284,
"eval_f1": 0.817910447761194,
"eval_loss": 0.583465039730072,
"eval_runtime": 0.1527,
"eval_samples_per_second": 2672.428,
"eval_steps_per_second": 13.1,
"step": 30
},
{
"epoch": 3.0,
"grad_norm": 2.3674581050872803,
"learning_rate": 4.7e-05,
"loss": 0.5612,
"step": 45
},
{
"epoch": 3.0,
"eval_accuracy": 0.7058823529411765,
"eval_combined_score": 0.7614777618364419,
"eval_f1": 0.8170731707317074,
"eval_loss": 0.559699535369873,
"eval_runtime": 0.1434,
"eval_samples_per_second": 2844.442,
"eval_steps_per_second": 13.943,
"step": 45
},
{
"epoch": 4.0,
"grad_norm": 1.349136471748352,
"learning_rate": 4.600000000000001e-05,
"loss": 0.5397,
"step": 60
},
{
"epoch": 4.0,
"eval_accuracy": 0.7377450980392157,
"eval_combined_score": 0.7848851078893095,
"eval_f1": 0.8320251177394035,
"eval_loss": 0.539802074432373,
"eval_runtime": 0.1477,
"eval_samples_per_second": 2762.511,
"eval_steps_per_second": 13.542,
"step": 60
},
{
"epoch": 5.0,
"grad_norm": 1.9282344579696655,
"learning_rate": 4.5e-05,
"loss": 0.5063,
"step": 75
},
{
"epoch": 5.0,
"eval_accuracy": 0.7426470588235294,
"eval_combined_score": 0.7881222615829215,
"eval_f1": 0.8335974643423137,
"eval_loss": 0.5358061194419861,
"eval_runtime": 0.1435,
"eval_samples_per_second": 2842.916,
"eval_steps_per_second": 13.936,
"step": 75
},
{
"epoch": 6.0,
"grad_norm": 2.279212713241577,
"learning_rate": 4.4000000000000006e-05,
"loss": 0.476,
"step": 90
},
{
"epoch": 6.0,
"eval_accuracy": 0.7475490196078431,
"eval_combined_score": 0.7916373487193921,
"eval_f1": 0.835725677830941,
"eval_loss": 0.5344022512435913,
"eval_runtime": 0.1435,
"eval_samples_per_second": 2842.637,
"eval_steps_per_second": 13.934,
"step": 90
},
{
"epoch": 7.0,
"grad_norm": 2.8690178394317627,
"learning_rate": 4.3e-05,
"loss": 0.4361,
"step": 105
},
{
"epoch": 7.0,
"eval_accuracy": 0.7450980392156863,
"eval_combined_score": 0.7900093370681607,
"eval_f1": 0.834920634920635,
"eval_loss": 0.5514616370201111,
"eval_runtime": 0.1462,
"eval_samples_per_second": 2790.062,
"eval_steps_per_second": 13.677,
"step": 105
},
{
"epoch": 8.0,
"grad_norm": 2.5032386779785156,
"learning_rate": 4.2e-05,
"loss": 0.4014,
"step": 120
},
{
"epoch": 8.0,
"eval_accuracy": 0.75,
"eval_combined_score": 0.7932692307692307,
"eval_f1": 0.8365384615384615,
"eval_loss": 0.5508028864860535,
"eval_runtime": 0.1417,
"eval_samples_per_second": 2878.654,
"eval_steps_per_second": 14.111,
"step": 120
},
{
"epoch": 9.0,
"grad_norm": 2.740001916885376,
"learning_rate": 4.1e-05,
"loss": 0.3684,
"step": 135
},
{
"epoch": 9.0,
"eval_accuracy": 0.7303921568627451,
"eval_combined_score": 0.7778944911297851,
"eval_f1": 0.8253968253968254,
"eval_loss": 0.5901079177856445,
"eval_runtime": 0.142,
"eval_samples_per_second": 2873.896,
"eval_steps_per_second": 14.088,
"step": 135
},
{
"epoch": 10.0,
"grad_norm": 3.6365294456481934,
"learning_rate": 4e-05,
"loss": 0.3396,
"step": 150
},
{
"epoch": 10.0,
"eval_accuracy": 0.7426470588235294,
"eval_combined_score": 0.7851166328600405,
"eval_f1": 0.8275862068965516,
"eval_loss": 0.5755388140678406,
"eval_runtime": 0.1465,
"eval_samples_per_second": 2784.301,
"eval_steps_per_second": 13.649,
"step": 150
},
{
"epoch": 11.0,
"grad_norm": 2.9824297428131104,
"learning_rate": 3.9000000000000006e-05,
"loss": 0.3061,
"step": 165
},
{
"epoch": 11.0,
"eval_accuracy": 0.75,
"eval_combined_score": 0.7908415841584159,
"eval_f1": 0.8316831683168318,
"eval_loss": 0.5942506194114685,
"eval_runtime": 0.1433,
"eval_samples_per_second": 2847.49,
"eval_steps_per_second": 13.958,
"step": 165
},
{
"epoch": 11.0,
"step": 165,
"total_flos": 102031020478464.0,
"train_loss": 0.4675573955882679,
"train_runtime": 21.1858,
"train_samples_per_second": 8656.733,
"train_steps_per_second": 35.401
}
],
"logging_steps": 1,
"max_steps": 750,
"num_input_tokens_seen": 0,
"num_train_epochs": 50,
"save_steps": 500,
"stateful_callbacks": {
"EarlyStoppingCallback": {
"args": {
"early_stopping_patience": 5,
"early_stopping_threshold": 0.0
},
"attributes": {
"early_stopping_patience_counter": 5
}
},
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 102031020478464.0,
"train_batch_size": 256,
"trial_name": null,
"trial_params": null
}