File size: 3,960 Bytes
887c4ac 8fa6cc5 887c4ac 8fa6cc5 887c4ac 8fa6cc5 887c4ac 8fa6cc5 887c4ac |
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.6686666666666667e-05,
"loss": 0.4418,
"step": 500
},
{
"epoch": 1.0,
"learning_rate": 1.3353333333333333e-05,
"loss": 0.3253,
"step": 1000
},
{
"epoch": 1.0,
"eval_accuracy": 0.878,
"eval_classification_report": " precision recall f1-score support\n0 0.901961 0.941254 0.921189 3030.000\n1 0.787589 0.680412 0.730088 970.000\naccuracy 0.878000 0.878000 0.878000 0.878\nmacro avg 0.844775 0.810833 0.825639 4000.000\nweighted avg 0.874226 0.878000 0.874847 4000.000",
"eval_confusion_matrix": "[[2852 178]\n [ 310 660]]",
"eval_confusion_matrix_norm": "[[0.94125413 0.05874587]\n [0.31958763 0.68041237]]",
"eval_f1": 0.7300884955752213,
"eval_f1_macro": 0.8256385630330887,
"eval_f1_weighted": 0.8748468477738904,
"eval_loss": 0.3010723888874054,
"eval_runtime": 9.6596,
"eval_samples_per_second": 414.097,
"eval_steps_per_second": 25.881,
"step": 1000
},
{
"epoch": 1.5,
"learning_rate": 1.002e-05,
"loss": 0.2684,
"step": 1500
},
{
"epoch": 2.0,
"learning_rate": 6.6866666666666665e-06,
"loss": 0.2439,
"step": 2000
},
{
"epoch": 2.0,
"eval_accuracy": 0.8865,
"eval_classification_report": " precision recall f1-score support\n0 0.914414 0.937954 0.926035 3030.0000\n1 0.789238 0.725773 0.756176 970.0000\naccuracy 0.886500 0.886500 0.886500 0.8865\nmacro avg 0.851826 0.831863 0.841105 4000.0000\nweighted avg 0.884059 0.886500 0.884844 4000.0000",
"eval_confusion_matrix": "[[2842 188]\n [ 266 704]]",
"eval_confusion_matrix_norm": "[[0.9379538 0.0620462]\n [0.2742268 0.7257732]]",
"eval_f1": 0.7561761546723953,
"eval_f1_macro": 0.84110534680508,
"eval_f1_weighted": 0.8848438807534127,
"eval_loss": 0.3121766746044159,
"eval_runtime": 9.6642,
"eval_samples_per_second": 413.897,
"eval_steps_per_second": 25.869,
"step": 2000
},
{
"epoch": 2.5,
"learning_rate": 3.3600000000000004e-06,
"loss": 0.2023,
"step": 2500
},
{
"epoch": 3.0,
"learning_rate": 2.6666666666666667e-08,
"loss": 0.1962,
"step": 3000
},
{
"epoch": 3.0,
"eval_accuracy": 0.88075,
"eval_classification_report": " precision recall f1-score support\n0 0.920593 0.922112 0.921352 3030.00000\n1 0.755440 0.751546 0.753488 970.00000\naccuracy 0.880750 0.880750 0.880750 0.88075\nmacro avg 0.838017 0.836829 0.837420 4000.00000\nweighted avg 0.880544 0.880750 0.880645 4000.00000",
"eval_confusion_matrix": "[[2794 236]\n [ 241 729]]",
"eval_confusion_matrix_norm": "[[0.92211221 0.07788779]\n [0.24845361 0.75154639]]",
"eval_f1": 0.7534883720930233,
"eval_f1_macro": 0.8374201959393394,
"eval_f1_weighted": 0.8806450852201921,
"eval_loss": 0.3457934856414795,
"eval_runtime": 9.6651,
"eval_samples_per_second": 413.86,
"eval_steps_per_second": 25.866,
"step": 3000
},
{
"epoch": 3.0,
"step": 3000,
"total_flos": 1.262933065728e+16,
"train_loss": 0.27966182454427085,
"train_runtime": 430.7509,
"train_samples_per_second": 111.433,
"train_steps_per_second": 6.965
}
],
"max_steps": 3000,
"num_train_epochs": 3,
"total_flos": 1.262933065728e+16,
"trial_name": null,
"trial_params": null
}
|