| { |
| "best_metric": 0.7736489492761546, |
| "best_model_checkpoint": "bart-base/checkpoint-4501", |
| "epoch": 11.0, |
| "eval_steps": 500, |
| "global_step": 7073, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.78, |
| "learning_rate": 4.871176775531364e-05, |
| "loss": 1.2028, |
| "step": 500 |
| }, |
| { |
| "epoch": 1.0, |
| "eval_accuracy": 0.7598760650658405, |
| "eval_f1_macro": 0.6063644681337579, |
| "eval_f1_micro": 0.7598760650658405, |
| "eval_f1_weighted": 0.7489266744747745, |
| "eval_loss": 0.8429557681083679, |
| "eval_macro_fpr": 0.023226157303183535, |
| "eval_macro_sensitivity": 0.6366904609623596, |
| "eval_macro_specificity": 0.981694425945952, |
| "eval_precision": 0.7600880594489849, |
| "eval_precision_macro": 0.6003620126022547, |
| "eval_recall": 0.7598760650658405, |
| "eval_recall_macro": 0.6366904609623596, |
| "eval_runtime": 43.5934, |
| "eval_samples_per_second": 29.615, |
| "eval_steps_per_second": 3.716, |
| "eval_weighted_fpr": 0.02207348333808032, |
| "eval_weighted_sensitivity": 0.7598760650658405, |
| "eval_weighted_specificity": 0.9655403241234398, |
| "step": 643 |
| }, |
| { |
| "epoch": 1.56, |
| "learning_rate": 4.741575946086055e-05, |
| "loss": 0.715, |
| "step": 1000 |
| }, |
| { |
| "epoch": 2.0, |
| "eval_accuracy": 0.7931835786212239, |
| "eval_f1_macro": 0.7213629796675549, |
| "eval_f1_micro": 0.793183578621224, |
| "eval_f1_weighted": 0.7853477061503886, |
| "eval_loss": 0.7331996560096741, |
| "eval_macro_fpr": 0.018982515099395962, |
| "eval_macro_sensitivity": 0.7320748785168393, |
| "eval_macro_specificity": 0.9845140063777027, |
| "eval_precision": 0.8019650926098448, |
| "eval_precision_macro": 0.7386000214414028, |
| "eval_recall": 0.7931835786212239, |
| "eval_recall_macro": 0.7320748785168393, |
| "eval_runtime": 77.2393, |
| "eval_samples_per_second": 16.714, |
| "eval_steps_per_second": 2.097, |
| "eval_weighted_fpr": 0.018283914264192288, |
| "eval_weighted_sensitivity": 0.7931835786212239, |
| "eval_weighted_specificity": 0.9745265170443169, |
| "step": 1286 |
| }, |
| { |
| "epoch": 2.33, |
| "learning_rate": 4.611975116640747e-05, |
| "loss": 0.578, |
| "step": 1500 |
| }, |
| { |
| "epoch": 3.0, |
| "eval_accuracy": 0.7939581719597212, |
| "eval_f1_macro": 0.6998313050068915, |
| "eval_f1_micro": 0.7939581719597211, |
| "eval_f1_weighted": 0.79006368267302, |
| "eval_loss": 0.8045273423194885, |
| "eval_macro_fpr": 0.018545878882531782, |
| "eval_macro_sensitivity": 0.7068834234702027, |
| "eval_macro_specificity": 0.9847671550097226, |
| "eval_precision": 0.8075005963568469, |
| "eval_precision_macro": 0.7230632440844518, |
| "eval_recall": 0.7939581719597212, |
| "eval_recall_macro": 0.7068834234702027, |
| "eval_runtime": 86.4099, |
| "eval_samples_per_second": 14.94, |
| "eval_steps_per_second": 1.875, |
| "eval_weighted_fpr": 0.018199233716475097, |
| "eval_weighted_sensitivity": 0.7939581719597212, |
| "eval_weighted_specificity": 0.977549153186118, |
| "step": 1929 |
| }, |
| { |
| "epoch": 3.11, |
| "learning_rate": 4.482633488854329e-05, |
| "loss": 0.481, |
| "step": 2000 |
| }, |
| { |
| "epoch": 3.89, |
| "learning_rate": 4.3530326594090203e-05, |
| "loss": 0.3938, |
| "step": 2500 |
| }, |
| { |
| "epoch": 4.0, |
| "eval_accuracy": 0.8156467854376452, |
| "eval_f1_macro": 0.7369315728111729, |
| "eval_f1_micro": 0.8156467854376452, |
| "eval_f1_weighted": 0.8105168035589256, |
| "eval_loss": 0.8291025161743164, |
| "eval_macro_fpr": 0.016938074954137138, |
| "eval_macro_sensitivity": 0.7217605983180461, |
| "eval_macro_specificity": 0.9857842035419169, |
| "eval_precision": 0.8171220006778526, |
| "eval_precision_macro": 0.7937034642937142, |
| "eval_recall": 0.8156467854376452, |
| "eval_recall_macro": 0.7217605983180461, |
| "eval_runtime": 84.4366, |
| "eval_samples_per_second": 15.29, |
| "eval_steps_per_second": 1.919, |
| "eval_weighted_fpr": 0.01588785046728972, |
| "eval_weighted_sensitivity": 0.8156467854376452, |
| "eval_weighted_specificity": 0.9711162676911109, |
| "step": 2572 |
| }, |
| { |
| "epoch": 4.67, |
| "learning_rate": 4.2236910316226025e-05, |
| "loss": 0.3238, |
| "step": 3000 |
| }, |
| { |
| "epoch": 5.0, |
| "eval_accuracy": 0.7939581719597212, |
| "eval_f1_macro": 0.7360579650857649, |
| "eval_f1_micro": 0.7939581719597211, |
| "eval_f1_weighted": 0.7995200606905476, |
| "eval_loss": 0.8888681530952454, |
| "eval_macro_fpr": 0.018767728188278533, |
| "eval_macro_sensitivity": 0.7515467286591376, |
| "eval_macro_specificity": 0.9846803211687324, |
| "eval_precision": 0.8146370555192632, |
| "eval_precision_macro": 0.7464262358644205, |
| "eval_recall": 0.7939581719597212, |
| "eval_recall_macro": 0.7515467286591376, |
| "eval_runtime": 82.179, |
| "eval_samples_per_second": 15.71, |
| "eval_steps_per_second": 1.971, |
| "eval_weighted_fpr": 0.018199233716475097, |
| "eval_weighted_sensitivity": 0.7939581719597212, |
| "eval_weighted_specificity": 0.9762466455712672, |
| "step": 3215 |
| }, |
| { |
| "epoch": 5.44, |
| "learning_rate": 4.094090202177294e-05, |
| "loss": 0.246, |
| "step": 3500 |
| }, |
| { |
| "epoch": 6.0, |
| "eval_accuracy": 0.7955073586367157, |
| "eval_f1_macro": 0.7362210043644567, |
| "eval_f1_micro": 0.7955073586367158, |
| "eval_f1_weighted": 0.7945977979999089, |
| "eval_loss": 1.1628626585006714, |
| "eval_macro_fpr": 0.01863187379355594, |
| "eval_macro_sensitivity": 0.7599986341032186, |
| "eval_macro_specificity": 0.9846932142915057, |
| "eval_precision": 0.8066758597712774, |
| "eval_precision_macro": 0.7483163388808308, |
| "eval_recall": 0.7955073586367157, |
| "eval_recall_macro": 0.7599986341032186, |
| "eval_runtime": 82.1621, |
| "eval_samples_per_second": 15.713, |
| "eval_steps_per_second": 1.972, |
| "eval_weighted_fpr": 0.018030323726266905, |
| "eval_weighted_sensitivity": 0.7955073586367157, |
| "eval_weighted_specificity": 0.9748908557358684, |
| "step": 3858 |
| }, |
| { |
| "epoch": 6.22, |
| "learning_rate": 3.964489372731986e-05, |
| "loss": 0.228, |
| "step": 4000 |
| }, |
| { |
| "epoch": 7.0, |
| "learning_rate": 3.834888543286677e-05, |
| "loss": 0.1791, |
| "step": 4500 |
| }, |
| { |
| "epoch": 7.0, |
| "eval_accuracy": 0.8179705654531371, |
| "eval_f1_macro": 0.7736489492761546, |
| "eval_f1_micro": 0.817970565453137, |
| "eval_f1_weighted": 0.8146592022169725, |
| "eval_loss": 1.1354283094406128, |
| "eval_macro_fpr": 0.016454731140779742, |
| "eval_macro_sensitivity": 0.7696557438486226, |
| "eval_macro_specificity": 0.9861772746118799, |
| "eval_precision": 0.8150792023791334, |
| "eval_precision_macro": 0.7832166669456337, |
| "eval_recall": 0.8179705654531371, |
| "eval_recall_macro": 0.7696557438486226, |
| "eval_runtime": 85.4548, |
| "eval_samples_per_second": 15.107, |
| "eval_steps_per_second": 1.896, |
| "eval_weighted_fpr": 0.01564684732671949, |
| "eval_weighted_sensitivity": 0.8179705654531371, |
| "eval_weighted_specificity": 0.9746885537250626, |
| "step": 4501 |
| }, |
| { |
| "epoch": 7.78, |
| "learning_rate": 3.705546915500259e-05, |
| "loss": 0.1305, |
| "step": 5000 |
| }, |
| { |
| "epoch": 8.0, |
| "eval_accuracy": 0.8109992254066615, |
| "eval_f1_macro": 0.7368975448573482, |
| "eval_f1_micro": 0.8109992254066615, |
| "eval_f1_weighted": 0.808822910649843, |
| "eval_loss": 1.282464861869812, |
| "eval_macro_fpr": 0.016943107063520774, |
| "eval_macro_sensitivity": 0.7488577570944955, |
| "eval_macro_specificity": 0.9858309939135591, |
| "eval_precision": 0.8147632227505013, |
| "eval_precision_macro": 0.7422132710401315, |
| "eval_recall": 0.8109992254066615, |
| "eval_recall_macro": 0.7488577570944955, |
| "eval_runtime": 87.6049, |
| "eval_samples_per_second": 14.737, |
| "eval_steps_per_second": 1.849, |
| "eval_weighted_fpr": 0.016373641121997046, |
| "eval_weighted_sensitivity": 0.8109992254066615, |
| "eval_weighted_specificity": 0.9764656832967246, |
| "step": 5144 |
| }, |
| { |
| "epoch": 8.55, |
| "learning_rate": 3.575946086054951e-05, |
| "loss": 0.0924, |
| "step": 5500 |
| }, |
| { |
| "epoch": 9.0, |
| "eval_accuracy": 0.8040278853601859, |
| "eval_f1_macro": 0.7559721746710123, |
| "eval_f1_micro": 0.8040278853601859, |
| "eval_f1_weighted": 0.80145679310201, |
| "eval_loss": 1.421708583831787, |
| "eval_macro_fpr": 0.017802568540250287, |
| "eval_macro_sensitivity": 0.7809236600744539, |
| "eval_macro_specificity": 0.9853492039682624, |
| "eval_precision": 0.8113877598541575, |
| "eval_precision_macro": 0.7464644897074477, |
| "eval_recall": 0.8040278853601859, |
| "eval_recall_macro": 0.7809236600744539, |
| "eval_runtime": 89.5999, |
| "eval_samples_per_second": 14.408, |
| "eval_steps_per_second": 1.808, |
| "eval_weighted_fpr": 0.017111937774771727, |
| "eval_weighted_sensitivity": 0.8040278853601859, |
| "eval_weighted_specificity": 0.9762101741637506, |
| "step": 5787 |
| }, |
| { |
| "epoch": 9.33, |
| "learning_rate": 3.446345256609643e-05, |
| "loss": 0.0953, |
| "step": 6000 |
| }, |
| { |
| "epoch": 10.0, |
| "eval_accuracy": 0.8024786986831913, |
| "eval_f1_macro": 0.7661426626929498, |
| "eval_f1_micro": 0.8024786986831914, |
| "eval_f1_weighted": 0.8001228885030943, |
| "eval_loss": 1.5552372932434082, |
| "eval_macro_fpr": 0.018250858387380567, |
| "eval_macro_sensitivity": 0.7822186507003115, |
| "eval_macro_specificity": 0.9849087494473102, |
| "eval_precision": 0.8055922813865947, |
| "eval_precision_macro": 0.7702182213697708, |
| "eval_recall": 0.8024786986831913, |
| "eval_recall_macro": 0.7822186507003115, |
| "eval_runtime": 85.2534, |
| "eval_samples_per_second": 15.143, |
| "eval_steps_per_second": 1.9, |
| "eval_weighted_fpr": 0.017277593332881633, |
| "eval_weighted_sensitivity": 0.8024786986831913, |
| "eval_weighted_specificity": 0.9711525430264618, |
| "step": 6430 |
| }, |
| { |
| "epoch": 10.11, |
| "learning_rate": 3.316744427164334e-05, |
| "loss": 0.0668, |
| "step": 6500 |
| }, |
| { |
| "epoch": 10.89, |
| "learning_rate": 3.1871435977190256e-05, |
| "loss": 0.0617, |
| "step": 7000 |
| }, |
| { |
| "epoch": 11.0, |
| "eval_accuracy": 0.8040278853601859, |
| "eval_f1_macro": 0.7422420279485343, |
| "eval_f1_micro": 0.8040278853601859, |
| "eval_f1_weighted": 0.8069943234338898, |
| "eval_loss": 1.5223894119262695, |
| "eval_macro_fpr": 0.017602619829831197, |
| "eval_macro_sensitivity": 0.7512050029477755, |
| "eval_macro_specificity": 0.9853474796083734, |
| "eval_precision": 0.8144253833246468, |
| "eval_precision_macro": 0.745702567690635, |
| "eval_recall": 0.8040278853601859, |
| "eval_recall_macro": 0.7512050029477755, |
| "eval_runtime": 85.7004, |
| "eval_samples_per_second": 15.064, |
| "eval_steps_per_second": 1.89, |
| "eval_weighted_fpr": 0.017111937774771727, |
| "eval_weighted_sensitivity": 0.8040278853601859, |
| "eval_weighted_specificity": 0.9761843087654142, |
| "step": 7073 |
| } |
| ], |
| "logging_steps": 500, |
| "max_steps": 19290, |
| "num_train_epochs": 30, |
| "save_steps": 500, |
| "total_flos": 1.7335085847754752e+16, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
