|
{ |
|
"best_metric": 0.7558827552193487, |
|
"best_model_checkpoint": "./results/coronanet_classi_roberta/checkpoint-3372", |
|
"epoch": 4.0, |
|
"global_step": 3372, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 8.92132907519561e-05, |
|
"loss": 1.7123, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_accuracy_balanced": 0.6744778154481599, |
|
"eval_accuracy_not_balanced": 0.7375055613228533, |
|
"eval_f1_macro": 0.6774596453859705, |
|
"eval_f1_weighted": 0.7344846513012891, |
|
"eval_loss": 0.8972207307815552, |
|
"eval_precision": 0.7429861548781078, |
|
"eval_recall": 0.7375055613228533, |
|
"eval_runtime": 29.1161, |
|
"eval_samples_per_second": 231.59, |
|
"step": 843 |
|
}, |
|
{ |
|
"epoch": 1.19, |
|
"learning_rate": 7.368172899151806e-05, |
|
"loss": 0.8779, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 1.78, |
|
"learning_rate": 5.815016723108002e-05, |
|
"loss": 0.7389, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_accuracy_balanced": 0.744417109661457, |
|
"eval_accuracy_not_balanced": 0.7655346285036334, |
|
"eval_f1_macro": 0.7393973656051112, |
|
"eval_f1_weighted": 0.7626792308967099, |
|
"eval_loss": 0.8118156790733337, |
|
"eval_precision": 0.7652843843532866, |
|
"eval_recall": 0.7655346285036334, |
|
"eval_runtime": 29.1071, |
|
"eval_samples_per_second": 231.661, |
|
"step": 1686 |
|
}, |
|
{ |
|
"epoch": 2.37, |
|
"learning_rate": 4.261860547064198e-05, |
|
"loss": 0.5835, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 2.97, |
|
"learning_rate": 2.708704371020394e-05, |
|
"loss": 0.4919, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"eval_accuracy_balanced": 0.7503794754553327, |
|
"eval_accuracy_not_balanced": 0.773987839240694, |
|
"eval_f1_macro": 0.7470579664348881, |
|
"eval_f1_weighted": 0.7724065863470936, |
|
"eval_loss": 0.7920669913291931, |
|
"eval_precision": 0.7736999040473431, |
|
"eval_recall": 0.773987839240694, |
|
"eval_runtime": 29.1029, |
|
"eval_samples_per_second": 231.695, |
|
"step": 2529 |
|
}, |
|
{ |
|
"epoch": 3.56, |
|
"learning_rate": 1.15554819497659e-05, |
|
"loss": 0.33, |
|
"step": 3000 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"eval_accuracy_balanced": 0.7558827552193487, |
|
"eval_accuracy_not_balanced": 0.7773987839240694, |
|
"eval_f1_macro": 0.7509844357682027, |
|
"eval_f1_weighted": 0.7760847284277537, |
|
"eval_loss": 0.853062629699707, |
|
"eval_precision": 0.7759580640894472, |
|
"eval_recall": 0.7773987839240694, |
|
"eval_runtime": 29.115, |
|
"eval_samples_per_second": 231.599, |
|
"step": 3372 |
|
} |
|
], |
|
"max_steps": 3372, |
|
"num_train_epochs": 4, |
|
"total_flos": 27219994048880640, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|