{ "best_metric": 0.5987413950993845, "best_model_checkpoint": "bart-base-lora/checkpoint-1607", "epoch": 9.996889580093313, "eval_steps": 500, "global_step": 1607, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 1.0, "eval_accuracy": 0.611154144074361, "eval_f1_macro": 0.28705545756004525, "eval_f1_micro": 0.611154144074361, "eval_f1_weighted": 0.5574695078229231, "eval_loss": 1.3204560279846191, "eval_macro_fpr": 0.046390243375546196, "eval_macro_sensitivity": 0.302378554842108, "eval_macro_specificity": 0.9691815961629444, "eval_precision": 0.5321554731456403, "eval_precision_macro": 0.2887439395071009, "eval_recall": 0.611154144074361, "eval_recall_macro": 0.302378554842108, "eval_runtime": 45.1674, "eval_samples_per_second": 28.583, "eval_steps_per_second": 3.587, "eval_weighted_fpr": 0.04347073086248701, "eval_weighted_sensitivity": 0.611154144074361, "eval_weighted_specificity": 0.9265697983698051, "step": 160 }, { "epoch": 2.0, "eval_accuracy": 0.6994577846630519, "eval_f1_macro": 0.3947617401663212, "eval_f1_micro": 0.6994577846630519, "eval_f1_weighted": 0.6807748226338182, "eval_loss": 0.887535810470581, "eval_macro_fpr": 0.03062685085400977, "eval_macro_sensitivity": 0.4253981018601158, "eval_macro_specificity": 0.977359054380045, "eval_precision": 0.6728416067370143, "eval_precision_macro": 0.3822291983692299, "eval_recall": 0.6994577846630519, "eval_recall_macro": 0.4253981018601158, "eval_runtime": 44.0045, "eval_samples_per_second": 29.338, "eval_steps_per_second": 3.681, "eval_weighted_fpr": 0.02977743668457406, "eval_weighted_sensitivity": 0.6994577846630519, "eval_weighted_specificity": 0.9609280310376243, "step": 321 }, { "epoch": 3.0, "eval_accuracy": 0.7064291247095275, "eval_f1_macro": 0.39687975393644837, "eval_f1_micro": 0.7064291247095275, "eval_f1_weighted": 0.6751887743731854, "eval_loss": 0.842707633972168, "eval_macro_fpr": 0.029525759570689643, "eval_macro_sensitivity": 0.44418263304043937, "eval_macro_specificity": 0.9780349023882488, "eval_precision": 0.6951917464608133, "eval_precision_macro": 0.4131313886422618, "eval_recall": 0.7064291247095275, "eval_recall_macro": 0.44418263304043937, "eval_runtime": 44.2756, "eval_samples_per_second": 29.158, "eval_steps_per_second": 3.659, "eval_weighted_fpr": 0.02882786947592607, "eval_weighted_sensitivity": 0.7064291247095275, "eval_weighted_specificity": 0.9640944111142056, "step": 482 }, { "epoch": 3.11, "learning_rate": 3.958333333333333e-05, "loss": 1.2895, "step": 500 }, { "epoch": 4.0, "eval_accuracy": 0.7273431448489543, "eval_f1_macro": 0.42839565617146974, "eval_f1_micro": 0.7273431448489543, "eval_f1_weighted": 0.7166656051521775, "eval_loss": 0.7718679904937744, "eval_macro_fpr": 0.026373610346420835, "eval_macro_sensitivity": 0.4597947061276887, "eval_macro_specificity": 0.9797578098165848, "eval_precision": 0.7132141916978711, "eval_precision_macro": 0.41981312370683244, "eval_recall": 0.7273431448489543, "eval_recall_macro": 0.4597947061276887, "eval_runtime": 44.0471, "eval_samples_per_second": 29.31, "eval_steps_per_second": 3.678, "eval_weighted_fpr": 0.026077937472218107, "eval_weighted_sensitivity": 0.7273431448489543, "eval_weighted_specificity": 0.9690240023998159, "step": 643 }, { "epoch": 5.0, "eval_accuracy": 0.750580945003873, "eval_f1_macro": 0.5071625387211971, "eval_f1_micro": 0.750580945003873, "eval_f1_weighted": 0.7368350875727435, "eval_loss": 0.7387559413909912, "eval_macro_fpr": 0.02390193870216418, "eval_macro_sensitivity": 0.5165232292614671, "eval_macro_specificity": 0.9813510652812688, "eval_precision": 0.739991301984922, "eval_precision_macro": 0.5733174750951799, "eval_recall": 0.750580945003873, "eval_recall_macro": 0.5165232292614671, "eval_runtime": 43.5191, "eval_samples_per_second": 29.665, "eval_steps_per_second": 3.723, "eval_weighted_fpr": 0.023185483870967742, "eval_weighted_sensitivity": 0.750580945003873, "eval_weighted_specificity": 0.9696850342151593, "step": 803 }, { "epoch": 6.0, "eval_accuracy": 0.7443841982958946, "eval_f1_macro": 0.508766573370445, "eval_f1_micro": 0.7443841982958946, "eval_f1_weighted": 0.726841801268282, "eval_loss": 0.7526118755340576, "eval_macro_fpr": 0.024741953499129225, "eval_macro_sensitivity": 0.5229557335240369, "eval_macro_specificity": 0.980899976318631, "eval_precision": 0.7337183019873953, "eval_precision_macro": 0.5703412270657953, "eval_recall": 0.7443841982958946, "eval_recall_macro": 0.5229557335240369, "eval_runtime": 43.7116, "eval_samples_per_second": 29.534, "eval_steps_per_second": 3.706, "eval_weighted_fpr": 0.023940800928612886, "eval_weighted_sensitivity": 0.7443841982958946, "eval_weighted_specificity": 0.9691154464835722, "step": 964 }, { "epoch": 6.22, "learning_rate": 2.916666666666667e-05, "loss": 0.7332, "step": 1000 }, { "epoch": 7.0, "eval_accuracy": 0.7552285050348567, "eval_f1_macro": 0.5608894910772386, "eval_f1_micro": 0.7552285050348567, "eval_f1_weighted": 0.7460771491633669, "eval_loss": 0.7082085609436035, "eval_macro_fpr": 0.02329269548734161, "eval_macro_sensitivity": 0.5727565701904757, "eval_macro_specificity": 0.9817624646436656, "eval_precision": 0.7436383222975608, "eval_precision_macro": 0.5665418243254761, "eval_recall": 0.7552285050348567, "eval_recall_macro": 0.5727565701904757, "eval_runtime": 43.8592, "eval_samples_per_second": 29.435, "eval_steps_per_second": 3.694, "eval_weighted_fpr": 0.02262637834741515, "eval_weighted_sensitivity": 0.7552285050348567, "eval_weighted_specificity": 0.9712084646201266, "step": 1125 }, { "epoch": 8.0, "eval_accuracy": 0.7583268783888458, "eval_f1_macro": 0.5755845408528306, "eval_f1_micro": 0.7583268783888458, "eval_f1_weighted": 0.7502605827076951, "eval_loss": 0.7160602807998657, "eval_macro_fpr": 0.022834923601538912, "eval_macro_sensitivity": 0.5974764193374731, "eval_macro_specificity": 0.9820281006961107, "eval_precision": 0.7489019370925815, "eval_precision_macro": 0.5641453776326693, "eval_recall": 0.7583268783888458, "eval_recall_macro": 0.5974764193374731, "eval_runtime": 47.4544, "eval_samples_per_second": 27.205, "eval_steps_per_second": 3.414, "eval_weighted_fpr": 0.022257098016835498, "eval_weighted_sensitivity": 0.7583268783888458, "eval_weighted_specificity": 0.9720946320528147, "step": 1286 }, { "epoch": 9.0, "eval_accuracy": 0.7776917118512781, "eval_f1_macro": 0.5874833340927881, "eval_f1_micro": 0.7776917118512781, "eval_f1_weighted": 0.7653015782806978, "eval_loss": 0.683149516582489, "eval_macro_fpr": 0.020831516213897273, "eval_macro_sensitivity": 0.606896137249408, "eval_macro_specificity": 0.9832816294136357, "eval_precision": 0.7586915140412427, "eval_precision_macro": 0.5780981033617572, "eval_recall": 0.7776917118512781, "eval_recall_macro": 0.606896137249408, "eval_runtime": 50.31, "eval_samples_per_second": 25.661, "eval_steps_per_second": 3.22, "eval_weighted_fpr": 0.02000976085895559, "eval_weighted_sensitivity": 0.7776917118512781, "eval_weighted_specificity": 0.9715327293532582, "step": 1446 }, { "epoch": 9.33, "learning_rate": 1.8750000000000002e-05, "loss": 0.6167, "step": 1500 }, { "epoch": 10.0, "eval_accuracy": 0.7862122385747483, "eval_f1_macro": 0.5987413950993845, "eval_f1_micro": 0.7862122385747482, "eval_f1_weighted": 0.7753590541094936, "eval_loss": 0.6683156490325928, "eval_macro_fpr": 0.019819293314469463, "eval_macro_sensitivity": 0.617364154098103, "eval_macro_specificity": 0.9839313497952282, "eval_precision": 0.7714272913975577, "eval_precision_macro": 0.5916700769971839, "eval_recall": 0.7862122385747483, "eval_recall_macro": 0.617364154098103, "eval_runtime": 46.3094, "eval_samples_per_second": 27.878, "eval_steps_per_second": 3.498, "eval_weighted_fpr": 0.019052878641446915, "eval_weighted_sensitivity": 0.7862122385747483, "eval_weighted_specificity": 0.972758008353676, "step": 1607 } ], "logging_steps": 500, "max_steps": 2400, "num_train_epochs": 15, "save_steps": 500, "total_flos": 1.627706135881728e+16, "trial_name": null, "trial_params": null }