File size: 3,965 Bytes
8eeab76 0347211 8eeab76 0347211 8eeab76 0347211 8eeab76 0347211 8eeab76 0347211 8eeab76 0347211 8eeab76 0347211 8eeab76 0347211 8eeab76 0347211 8eeab76 4fb0ffe 0347211 8eeab76 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 3.0,
"global_step": 3000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.5,
"learning_rate": 1.6673333333333335e-05,
"loss": 0.4114,
"step": 500
},
{
"epoch": 1.0,
"learning_rate": 1.3340000000000001e-05,
"loss": 0.3333,
"step": 1000
},
{
"epoch": 1.0,
"eval_accuracy": 0.86925,
"eval_classification_report": " precision recall f1-score support\n0 0.902408 0.927723 0.914890 3030.00000\n1 0.752542 0.686598 0.718059 970.00000\naccuracy 0.869250 0.869250 0.869250 0.86925\nmacro avg 0.827475 0.807160 0.816475 4000.00000\nweighted avg 0.866065 0.869250 0.867159 4000.00000",
"eval_confusion_matrix": "[[2811 219]\n [ 304 666]]",
"eval_confusion_matrix_norm": "[[0.92772277 0.07227723]\n [0.31340206 0.68659794]]",
"eval_f1": 0.7180592991913748,
"eval_f1_macro": 0.8164747268943042,
"eval_f1_weighted": 0.8671586721613128,
"eval_loss": 0.3063889145851135,
"eval_runtime": 9.6316,
"eval_samples_per_second": 415.3,
"eval_steps_per_second": 25.956,
"step": 1000
},
{
"epoch": 1.5,
"learning_rate": 1.0013333333333335e-05,
"loss": 0.2455,
"step": 1500
},
{
"epoch": 2.0,
"learning_rate": 6.6866666666666665e-06,
"loss": 0.2271,
"step": 2000
},
{
"epoch": 2.0,
"eval_accuracy": 0.871,
"eval_classification_report": " precision recall f1-score support\n0 0.913487 0.916502 0.914992 3030.000\n1 0.736458 0.728866 0.732642 970.000\naccuracy 0.871000 0.871000 0.871000 0.871\nmacro avg 0.824973 0.822684 0.823817 4000.000\nweighted avg 0.870557 0.871000 0.870772 4000.000",
"eval_confusion_matrix": "[[2777 253]\n [ 263 707]]",
"eval_confusion_matrix_norm": "[[0.91650165 0.08349835]\n [0.27113402 0.72886598]]",
"eval_f1": 0.732642487046632,
"eval_f1_macro": 0.8238171249071711,
"eval_f1_weighted": 0.8707720634053486,
"eval_loss": 0.3905148506164551,
"eval_runtime": 9.6371,
"eval_samples_per_second": 415.062,
"eval_steps_per_second": 25.941,
"step": 2000
},
{
"epoch": 2.5,
"learning_rate": 3.3533333333333336e-06,
"loss": 0.1517,
"step": 2500
},
{
"epoch": 3.0,
"learning_rate": 2e-08,
"loss": 0.1435,
"step": 3000
},
{
"epoch": 3.0,
"eval_accuracy": 0.87925,
"eval_classification_report": " precision recall f1-score support\n0 0.908042 0.935314 0.921476 3030.00000\n1 0.777019 0.704124 0.738778 970.00000\naccuracy 0.879250 0.879250 0.879250 0.87925\nmacro avg 0.842531 0.819719 0.830127 4000.00000\nweighted avg 0.876269 0.879250 0.877172 4000.00000",
"eval_confusion_matrix": "[[2834 196]\n [ 287 683]]",
"eval_confusion_matrix_norm": "[[0.93531353 0.06468647]\n [0.29587629 0.70412371]]",
"eval_f1": 0.7387777176852353,
"eval_f1_macro": 0.8301269502098749,
"eval_f1_weighted": 0.8771718049600645,
"eval_loss": 0.4823172092437744,
"eval_runtime": 9.6475,
"eval_samples_per_second": 414.617,
"eval_steps_per_second": 25.914,
"step": 3000
},
{
"epoch": 3.0,
"step": 3000,
"total_flos": 1.262933065728e+16,
"train_loss": 0.2520793151855469,
"train_runtime": 430.4281,
"train_samples_per_second": 111.517,
"train_steps_per_second": 6.97
}
],
"max_steps": 3000,
"num_train_epochs": 3,
"total_flos": 1.262933065728e+16,
"trial_name": null,
"trial_params": null
}
|