| { | |
| "best_metric": 0.6764229637898442, | |
| "best_model_checkpoint": "case-analysis-legal-bert-base-uncased/checkpoint-2464", | |
| "epoch": 25.0, | |
| "eval_steps": 500, | |
| "global_step": 5600, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 1.0, | |
| "eval_accuracy": 0.6570155902004454, | |
| "eval_f1_macro": 0.4081263564022185, | |
| "eval_f1_micro": 0.6570155902004454, | |
| "eval_f1_weighted": 0.6166399839312825, | |
| "eval_loss": 1.0114229917526245, | |
| "eval_macro_fpr": 0.19475132363363487, | |
| "eval_macro_sensitivity": 0.40746667217255456, | |
| "eval_macro_specificity": 0.8324392236563901, | |
| "eval_precision": 0.6178816225558943, | |
| "eval_precision_macro": 0.4664000256488353, | |
| "eval_recall": 0.6570155902004454, | |
| "eval_recall_macro": 0.40746667217255456, | |
| "eval_runtime": 7.3199, | |
| "eval_samples_per_second": 61.34, | |
| "eval_steps_per_second": 7.787, | |
| "eval_weighted_fpr": 0.1482194417709336, | |
| "eval_weighted_sensitivity": 0.6570155902004454, | |
| "eval_weighted_specificity": 0.6727413044251147, | |
| "step": 224 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_accuracy": 0.7750556792873051, | |
| "eval_f1_macro": 0.5658663855845003, | |
| "eval_f1_micro": 0.775055679287305, | |
| "eval_f1_weighted": 0.7564492543508543, | |
| "eval_loss": 0.764952540397644, | |
| "eval_macro_fpr": 0.10943701020063132, | |
| "eval_macro_sensitivity": 0.5805958747135218, | |
| "eval_macro_specificity": 0.9039084791647444, | |
| "eval_precision": 0.7425093237919278, | |
| "eval_precision_macro": 0.5566158325717437, | |
| "eval_recall": 0.7750556792873051, | |
| "eval_recall_macro": 0.5805958747135218, | |
| "eval_runtime": 30.7218, | |
| "eval_samples_per_second": 14.615, | |
| "eval_steps_per_second": 1.855, | |
| "eval_weighted_fpr": 0.08820960698689956, | |
| "eval_weighted_sensitivity": 0.7750556792873051, | |
| "eval_weighted_specificity": 0.8405782373716723, | |
| "step": 448 | |
| }, | |
| { | |
| "epoch": 2.232142857142857, | |
| "grad_norm": 9.53757095336914, | |
| "learning_rate": 4.631696428571429e-05, | |
| "loss": 0.7677, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_accuracy": 0.7817371937639198, | |
| "eval_f1_macro": 0.5707248076329534, | |
| "eval_f1_micro": 0.7817371937639198, | |
| "eval_f1_weighted": 0.7678448816993292, | |
| "eval_loss": 0.7341566681861877, | |
| "eval_macro_fpr": 0.10409605667370833, | |
| "eval_macro_sensitivity": 0.5967117805353099, | |
| "eval_macro_specificity": 0.9083046086144851, | |
| "eval_precision": 0.7694906286035585, | |
| "eval_precision_macro": 0.5673619923602318, | |
| "eval_recall": 0.7817371937639198, | |
| "eval_recall_macro": 0.5967117805353099, | |
| "eval_runtime": 32.9882, | |
| "eval_samples_per_second": 13.611, | |
| "eval_steps_per_second": 1.728, | |
| "eval_weighted_fpr": 0.08514335360556038, | |
| "eval_weighted_sensitivity": 0.7817371937639198, | |
| "eval_weighted_specificity": 0.8514812406940205, | |
| "step": 672 | |
| }, | |
| { | |
| "epoch": 4.0, | |
| "eval_accuracy": 0.8173719376391982, | |
| "eval_f1_macro": 0.5903876477694423, | |
| "eval_f1_micro": 0.8173719376391981, | |
| "eval_f1_weighted": 0.7920921547144288, | |
| "eval_loss": 0.7967568635940552, | |
| "eval_macro_fpr": 0.09654289346260755, | |
| "eval_macro_sensitivity": 0.5892796922208687, | |
| "eval_macro_specificity": 0.9135512326349768, | |
| "eval_precision": 0.776630887817645, | |
| "eval_precision_macro": 0.6035747039094435, | |
| "eval_recall": 0.8173719376391982, | |
| "eval_recall_macro": 0.5892796922208687, | |
| "eval_runtime": 31.5737, | |
| "eval_samples_per_second": 14.221, | |
| "eval_steps_per_second": 1.805, | |
| "eval_weighted_fpr": 0.06931530008453085, | |
| "eval_weighted_sensitivity": 0.8173719376391982, | |
| "eval_weighted_specificity": 0.836832992900709, | |
| "step": 896 | |
| }, | |
| { | |
| "epoch": 4.464285714285714, | |
| "grad_norm": 0.4865407943725586, | |
| "learning_rate": 4.260416666666667e-05, | |
| "loss": 0.482, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 5.0, | |
| "eval_accuracy": 0.8084632516703786, | |
| "eval_f1_macro": 0.6140549414742963, | |
| "eval_f1_micro": 0.8084632516703786, | |
| "eval_f1_weighted": 0.7940411816552344, | |
| "eval_loss": 0.8170605301856995, | |
| "eval_macro_fpr": 0.09903328320796416, | |
| "eval_macro_sensitivity": 0.6038384641325818, | |
| "eval_macro_specificity": 0.9107621964110777, | |
| "eval_precision": 0.7852994506060567, | |
| "eval_precision_macro": 0.6365555789178582, | |
| "eval_recall": 0.8084632516703786, | |
| "eval_recall_macro": 0.6038384641325818, | |
| "eval_runtime": 33.691, | |
| "eval_samples_per_second": 13.327, | |
| "eval_steps_per_second": 1.692, | |
| "eval_weighted_fpr": 0.07319148936170213, | |
| "eval_weighted_sensitivity": 0.8084632516703786, | |
| "eval_weighted_specificity": 0.8345855339739323, | |
| "step": 1120 | |
| }, | |
| { | |
| "epoch": 6.0, | |
| "eval_accuracy": 0.8240534521158129, | |
| "eval_f1_macro": 0.6314846316884821, | |
| "eval_f1_micro": 0.8240534521158129, | |
| "eval_f1_weighted": 0.8083544635272152, | |
| "eval_loss": 0.8909738659858704, | |
| "eval_macro_fpr": 0.087529695086195, | |
| "eval_macro_sensitivity": 0.6358500402618049, | |
| "eval_macro_specificity": 0.9211652121242622, | |
| "eval_precision": 0.8028335761470152, | |
| "eval_precision_macro": 0.6659870176540079, | |
| "eval_recall": 0.8240534521158129, | |
| "eval_recall_macro": 0.6358500402618049, | |
| "eval_runtime": 32.5936, | |
| "eval_samples_per_second": 13.776, | |
| "eval_steps_per_second": 1.749, | |
| "eval_weighted_fpr": 0.06644238856181665, | |
| "eval_weighted_sensitivity": 0.8240534521158129, | |
| "eval_weighted_specificity": 0.8606073963812356, | |
| "step": 1344 | |
| }, | |
| { | |
| "epoch": 6.696428571428571, | |
| "grad_norm": 24.31475257873535, | |
| "learning_rate": 3.888392857142857e-05, | |
| "loss": 0.2993, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 7.0, | |
| "eval_accuracy": 0.8040089086859689, | |
| "eval_f1_macro": 0.6741993002516824, | |
| "eval_f1_micro": 0.8040089086859689, | |
| "eval_f1_weighted": 0.8067155865282774, | |
| "eval_loss": 1.0094465017318726, | |
| "eval_macro_fpr": 0.08913550870826559, | |
| "eval_macro_sensitivity": 0.6773721773721774, | |
| "eval_macro_specificity": 0.9217242184731483, | |
| "eval_precision": 0.8108961670502116, | |
| "eval_precision_macro": 0.6741723864688478, | |
| "eval_recall": 0.8040089086859689, | |
| "eval_recall_macro": 0.6773721773721774, | |
| "eval_runtime": 33.2368, | |
| "eval_samples_per_second": 13.509, | |
| "eval_steps_per_second": 1.715, | |
| "eval_weighted_fpr": 0.07514944491887275, | |
| "eval_weighted_sensitivity": 0.8040089086859689, | |
| "eval_weighted_specificity": 0.8828879652066246, | |
| "step": 1568 | |
| }, | |
| { | |
| "epoch": 8.0, | |
| "eval_accuracy": 0.8106904231625836, | |
| "eval_f1_macro": 0.6229381248976396, | |
| "eval_f1_micro": 0.8106904231625834, | |
| "eval_f1_weighted": 0.8016368957819484, | |
| "eval_loss": 1.1504098176956177, | |
| "eval_macro_fpr": 0.089652400577408, | |
| "eval_macro_sensitivity": 0.6330299933241109, | |
| "eval_macro_specificity": 0.9203722429802373, | |
| "eval_precision": 0.7967862579587831, | |
| "eval_precision_macro": 0.6227581659565835, | |
| "eval_recall": 0.8106904231625836, | |
| "eval_recall_macro": 0.6330299933241109, | |
| "eval_runtime": 36.2257, | |
| "eval_samples_per_second": 12.395, | |
| "eval_steps_per_second": 1.573, | |
| "eval_weighted_fpr": 0.07221750212404418, | |
| "eval_weighted_sensitivity": 0.8106904231625836, | |
| "eval_weighted_specificity": 0.8707985487583659, | |
| "step": 1792 | |
| }, | |
| { | |
| "epoch": 8.928571428571429, | |
| "grad_norm": 0.21054323017597198, | |
| "learning_rate": 3.516369047619048e-05, | |
| "loss": 0.1367, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 9.0, | |
| "eval_accuracy": 0.8062360801781737, | |
| "eval_f1_macro": 0.6359771704966781, | |
| "eval_f1_micro": 0.8062360801781738, | |
| "eval_f1_weighted": 0.8044711013673843, | |
| "eval_loss": 1.2533091306686401, | |
| "eval_macro_fpr": 0.09482503678094849, | |
| "eval_macro_sensitivity": 0.6224819165995636, | |
| "eval_macro_specificity": 0.9164236762016751, | |
| "eval_precision": 0.8059392062620114, | |
| "eval_precision_macro": 0.6536979385970869, | |
| "eval_recall": 0.8062360801781737, | |
| "eval_recall_macro": 0.6224819165995636, | |
| "eval_runtime": 30.7582, | |
| "eval_samples_per_second": 14.598, | |
| "eval_steps_per_second": 1.853, | |
| "eval_weighted_fpr": 0.0741687979539642, | |
| "eval_weighted_sensitivity": 0.8062360801781737, | |
| "eval_weighted_specificity": 0.8594586246285271, | |
| "step": 2016 | |
| }, | |
| { | |
| "epoch": 10.0, | |
| "eval_accuracy": 0.8173719376391982, | |
| "eval_f1_macro": 0.6554404393882288, | |
| "eval_f1_micro": 0.8173719376391981, | |
| "eval_f1_weighted": 0.8137262031233367, | |
| "eval_loss": 1.2516238689422607, | |
| "eval_macro_fpr": 0.08727001865560947, | |
| "eval_macro_sensitivity": 0.649889194006841, | |
| "eval_macro_specificity": 0.9218678293458737, | |
| "eval_precision": 0.8106611480963274, | |
| "eval_precision_macro": 0.6620602008894919, | |
| "eval_recall": 0.8173719376391982, | |
| "eval_recall_macro": 0.649889194006841, | |
| "eval_runtime": 34.0534, | |
| "eval_samples_per_second": 13.185, | |
| "eval_steps_per_second": 1.674, | |
| "eval_weighted_fpr": 0.06931530008453085, | |
| "eval_weighted_sensitivity": 0.8173719376391982, | |
| "eval_weighted_specificity": 0.8700993797442969, | |
| "step": 2240 | |
| }, | |
| { | |
| "epoch": 11.0, | |
| "eval_accuracy": 0.8262806236080178, | |
| "eval_f1_macro": 0.6764229637898442, | |
| "eval_f1_micro": 0.8262806236080178, | |
| "eval_f1_weighted": 0.819522907157828, | |
| "eval_loss": 1.3822201490402222, | |
| "eval_macro_fpr": 0.08329902302757192, | |
| "eval_macro_sensitivity": 0.669645007880302, | |
| "eval_macro_specificity": 0.9243474796823474, | |
| "eval_precision": 0.8204585488197453, | |
| "eval_precision_macro": 0.7084622597280825, | |
| "eval_recall": 0.8262806236080178, | |
| "eval_recall_macro": 0.669645007880302, | |
| "eval_runtime": 31.5236, | |
| "eval_samples_per_second": 14.243, | |
| "eval_steps_per_second": 1.808, | |
| "eval_weighted_fpr": 0.0654911838790932, | |
| "eval_weighted_sensitivity": 0.8262806236080178, | |
| "eval_weighted_specificity": 0.8711092951213717, | |
| "step": 2464 | |
| }, | |
| { | |
| "epoch": 11.160714285714286, | |
| "grad_norm": 2.0276401042938232, | |
| "learning_rate": 3.1443452380952384e-05, | |
| "loss": 0.055, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 12.0, | |
| "eval_accuracy": 0.8017817371937639, | |
| "eval_f1_macro": 0.6399377531674577, | |
| "eval_f1_micro": 0.8017817371937639, | |
| "eval_f1_weighted": 0.806834856085664, | |
| "eval_loss": 1.4574010372161865, | |
| "eval_macro_fpr": 0.0883315306539799, | |
| "eval_macro_sensitivity": 0.6443222252045782, | |
| "eval_macro_specificity": 0.9215573710717683, | |
| "eval_precision": 0.8127232508105453, | |
| "eval_precision_macro": 0.6368985338108845, | |
| "eval_recall": 0.8017817371937639, | |
| "eval_recall_macro": 0.6443222252045782, | |
| "eval_runtime": 35.9686, | |
| "eval_samples_per_second": 12.483, | |
| "eval_steps_per_second": 1.585, | |
| "eval_weighted_fpr": 0.07613344739093242, | |
| "eval_weighted_sensitivity": 0.8017817371937639, | |
| "eval_weighted_specificity": 0.884447747093309, | |
| "step": 2688 | |
| }, | |
| { | |
| "epoch": 13.0, | |
| "eval_accuracy": 0.7884187082405345, | |
| "eval_f1_macro": 0.6042489773120808, | |
| "eval_f1_micro": 0.7884187082405345, | |
| "eval_f1_weighted": 0.7831300624317719, | |
| "eval_loss": 1.6634190082550049, | |
| "eval_macro_fpr": 0.10018097159570645, | |
| "eval_macro_sensitivity": 0.6042144130379424, | |
| "eval_macro_specificity": 0.9125690965302461, | |
| "eval_precision": 0.7810481910770134, | |
| "eval_precision_macro": 0.6089900703871292, | |
| "eval_recall": 0.7884187082405345, | |
| "eval_recall_macro": 0.6042144130379424, | |
| "eval_runtime": 29.9493, | |
| "eval_samples_per_second": 14.992, | |
| "eval_steps_per_second": 1.903, | |
| "eval_weighted_fpr": 0.08210890233362143, | |
| "eval_weighted_sensitivity": 0.7884187082405345, | |
| "eval_weighted_specificity": 0.8618576778804495, | |
| "step": 2912 | |
| }, | |
| { | |
| "epoch": 13.392857142857142, | |
| "grad_norm": 0.00395896565169096, | |
| "learning_rate": 2.773809523809524e-05, | |
| "loss": 0.0431, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 14.0, | |
| "eval_accuracy": 0.8285077951002228, | |
| "eval_f1_macro": 0.6381695277501472, | |
| "eval_f1_micro": 0.8285077951002228, | |
| "eval_f1_weighted": 0.8165889625025802, | |
| "eval_loss": 1.5085251331329346, | |
| "eval_macro_fpr": 0.0849606439034177, | |
| "eval_macro_sensitivity": 0.6367413058589528, | |
| "eval_macro_specificity": 0.9229424414178813, | |
| "eval_precision": 0.8077049262727573, | |
| "eval_precision_macro": 0.6475977473768657, | |
| "eval_recall": 0.8285077951002228, | |
| "eval_recall_macro": 0.6367413058589528, | |
| "eval_runtime": 34.3243, | |
| "eval_samples_per_second": 13.081, | |
| "eval_steps_per_second": 1.661, | |
| "eval_weighted_fpr": 0.06454316848281642, | |
| "eval_weighted_sensitivity": 0.8285077951002228, | |
| "eval_weighted_specificity": 0.8632619705713027, | |
| "step": 3136 | |
| }, | |
| { | |
| "epoch": 15.0, | |
| "eval_accuracy": 0.8106904231625836, | |
| "eval_f1_macro": 0.6229362972793631, | |
| "eval_f1_micro": 0.8106904231625834, | |
| "eval_f1_weighted": 0.8013640427817921, | |
| "eval_loss": 1.641113042831421, | |
| "eval_macro_fpr": 0.09140135508148657, | |
| "eval_macro_sensitivity": 0.6261966703143174, | |
| "eval_macro_specificity": 0.9183149213614681, | |
| "eval_precision": 0.7935778228683108, | |
| "eval_precision_macro": 0.6242729514731414, | |
| "eval_recall": 0.8106904231625836, | |
| "eval_recall_macro": 0.6261966703143174, | |
| "eval_runtime": 35.8672, | |
| "eval_samples_per_second": 12.518, | |
| "eval_steps_per_second": 1.589, | |
| "eval_weighted_fpr": 0.07221750212404418, | |
| "eval_weighted_sensitivity": 0.8106904231625836, | |
| "eval_weighted_specificity": 0.8625692622832889, | |
| "step": 3360 | |
| }, | |
| { | |
| "epoch": 15.625, | |
| "grad_norm": 0.09424740076065063, | |
| "learning_rate": 2.4017857142857146e-05, | |
| "loss": 0.0135, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 16.0, | |
| "eval_accuracy": 0.8062360801781737, | |
| "eval_f1_macro": 0.6221390381080233, | |
| "eval_f1_micro": 0.8062360801781738, | |
| "eval_f1_weighted": 0.7988403583111869, | |
| "eval_loss": 1.7482596635818481, | |
| "eval_macro_fpr": 0.09232876528599361, | |
| "eval_macro_sensitivity": 0.6270887962064433, | |
| "eval_macro_specificity": 0.9177392895526978, | |
| "eval_precision": 0.7924673420177055, | |
| "eval_precision_macro": 0.6200610055562886, | |
| "eval_recall": 0.8062360801781737, | |
| "eval_recall_macro": 0.6270887962064433, | |
| "eval_runtime": 34.8393, | |
| "eval_samples_per_second": 12.888, | |
| "eval_steps_per_second": 1.636, | |
| "eval_weighted_fpr": 0.0741687979539642, | |
| "eval_weighted_sensitivity": 0.8062360801781737, | |
| "eval_weighted_specificity": 0.8647210780326176, | |
| "step": 3584 | |
| }, | |
| { | |
| "epoch": 17.0, | |
| "eval_accuracy": 0.7973273942093542, | |
| "eval_f1_macro": 0.6201227671521832, | |
| "eval_f1_micro": 0.7973273942093542, | |
| "eval_f1_weighted": 0.7932689879372665, | |
| "eval_loss": 1.7232707738876343, | |
| "eval_macro_fpr": 0.09423113155723643, | |
| "eval_macro_sensitivity": 0.6263480822304351, | |
| "eval_macro_specificity": 0.9163848671929355, | |
| "eval_precision": 0.789684456744836, | |
| "eval_precision_macro": 0.6147620506535948, | |
| "eval_recall": 0.7973273942093542, | |
| "eval_recall_macro": 0.6263480822304351, | |
| "eval_runtime": 36.3399, | |
| "eval_samples_per_second": 12.356, | |
| "eval_steps_per_second": 1.569, | |
| "eval_weighted_fpr": 0.07811158798283262, | |
| "eval_weighted_sensitivity": 0.7973273942093542, | |
| "eval_weighted_specificity": 0.8682120745623881, | |
| "step": 3808 | |
| }, | |
| { | |
| "epoch": 17.857142857142858, | |
| "grad_norm": 0.004108293913304806, | |
| "learning_rate": 2.029761904761905e-05, | |
| "loss": 0.0066, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 18.0, | |
| "eval_accuracy": 0.8240534521158129, | |
| "eval_f1_macro": 0.639053660901891, | |
| "eval_f1_micro": 0.8240534521158129, | |
| "eval_f1_weighted": 0.8115429423166062, | |
| "eval_loss": 1.6456769704818726, | |
| "eval_macro_fpr": 0.08791315651117829, | |
| "eval_macro_sensitivity": 0.6388214991156167, | |
| "eval_macro_specificity": 0.9190529442926627, | |
| "eval_precision": 0.8041665668829977, | |
| "eval_precision_macro": 0.6515456989247311, | |
| "eval_recall": 0.8240534521158129, | |
| "eval_recall_macro": 0.6388214991156167, | |
| "eval_runtime": 33.8794, | |
| "eval_samples_per_second": 13.253, | |
| "eval_steps_per_second": 1.682, | |
| "eval_weighted_fpr": 0.06644238856181665, | |
| "eval_weighted_sensitivity": 0.8240534521158129, | |
| "eval_weighted_specificity": 0.8521583250548381, | |
| "step": 4032 | |
| }, | |
| { | |
| "epoch": 19.0, | |
| "eval_accuracy": 0.8173719376391982, | |
| "eval_f1_macro": 0.6317989768305606, | |
| "eval_f1_micro": 0.8173719376391981, | |
| "eval_f1_weighted": 0.8060684012173893, | |
| "eval_loss": 1.6613563299179077, | |
| "eval_macro_fpr": 0.08645884522989711, | |
| "eval_macro_sensitivity": 0.6420174949586714, | |
| "eval_macro_specificity": 0.9219250663589891, | |
| "eval_precision": 0.7975558509149212, | |
| "eval_precision_macro": 0.632433998424945, | |
| "eval_recall": 0.8173719376391982, | |
| "eval_recall_macro": 0.6420174949586714, | |
| "eval_runtime": 37.7131, | |
| "eval_samples_per_second": 11.906, | |
| "eval_steps_per_second": 1.511, | |
| "eval_weighted_fpr": 0.06931530008453085, | |
| "eval_weighted_sensitivity": 0.8173719376391982, | |
| "eval_weighted_specificity": 0.8703283277967583, | |
| "step": 4256 | |
| }, | |
| { | |
| "epoch": 20.0, | |
| "eval_accuracy": 0.8129175946547884, | |
| "eval_f1_macro": 0.6462426038179767, | |
| "eval_f1_micro": 0.8129175946547884, | |
| "eval_f1_weighted": 0.8061306345778022, | |
| "eval_loss": 1.6996936798095703, | |
| "eval_macro_fpr": 0.08602473589055493, | |
| "eval_macro_sensitivity": 0.6576163634987164, | |
| "eval_macro_specificity": 0.9222151830917074, | |
| "eval_precision": 0.8023118936030428, | |
| "eval_precision_macro": 0.6435490147884151, | |
| "eval_recall": 0.8129175946547884, | |
| "eval_recall_macro": 0.6576163634987164, | |
| "eval_runtime": 37.8688, | |
| "eval_samples_per_second": 11.857, | |
| "eval_steps_per_second": 1.505, | |
| "eval_weighted_fpr": 0.07124681933842239, | |
| "eval_weighted_sensitivity": 0.8129175946547884, | |
| "eval_weighted_specificity": 0.8759431377120409, | |
| "step": 4480 | |
| }, | |
| { | |
| "epoch": 20.089285714285715, | |
| "grad_norm": 0.002192622981965542, | |
| "learning_rate": 1.6577380952380952e-05, | |
| "loss": 0.0067, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 21.0, | |
| "eval_accuracy": 0.821826280623608, | |
| "eval_f1_macro": 0.6355569862362493, | |
| "eval_f1_micro": 0.821826280623608, | |
| "eval_f1_weighted": 0.8087869169905122, | |
| "eval_loss": 1.6539603471755981, | |
| "eval_macro_fpr": 0.08801189704305487, | |
| "eval_macro_sensitivity": 0.6380050791815497, | |
| "eval_macro_specificity": 0.9194558785340451, | |
| "eval_precision": 0.8000265748396097, | |
| "eval_precision_macro": 0.6473497189944558, | |
| "eval_recall": 0.821826280623608, | |
| "eval_recall_macro": 0.6380050791815497, | |
| "eval_runtime": 34.1541, | |
| "eval_samples_per_second": 13.146, | |
| "eval_steps_per_second": 1.669, | |
| "eval_weighted_fpr": 0.06739679865206402, | |
| "eval_weighted_sensitivity": 0.821826280623608, | |
| "eval_weighted_specificity": 0.8559972335125722, | |
| "step": 4704 | |
| }, | |
| { | |
| "epoch": 22.0, | |
| "eval_accuracy": 0.8084632516703786, | |
| "eval_f1_macro": 0.6281503208041694, | |
| "eval_f1_micro": 0.8084632516703786, | |
| "eval_f1_weighted": 0.8011014788104702, | |
| "eval_loss": 1.7329360246658325, | |
| "eval_macro_fpr": 0.09298094942813835, | |
| "eval_macro_sensitivity": 0.6267171487759723, | |
| "eval_macro_specificity": 0.915807100650446, | |
| "eval_precision": 0.7945028295139653, | |
| "eval_precision_macro": 0.6312759462759463, | |
| "eval_recall": 0.8084632516703786, | |
| "eval_recall_macro": 0.6267171487759723, | |
| "eval_runtime": 35.4666, | |
| "eval_samples_per_second": 12.66, | |
| "eval_steps_per_second": 1.607, | |
| "eval_weighted_fpr": 0.07319148936170213, | |
| "eval_weighted_sensitivity": 0.8084632516703786, | |
| "eval_weighted_specificity": 0.8547651509314054, | |
| "step": 4928 | |
| }, | |
| { | |
| "epoch": 22.321428571428573, | |
| "grad_norm": 0.0013072589645162225, | |
| "learning_rate": 1.2857142857142857e-05, | |
| "loss": 0.0028, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 23.0, | |
| "eval_accuracy": 0.8062360801781737, | |
| "eval_f1_macro": 0.6388959517008297, | |
| "eval_f1_micro": 0.8062360801781738, | |
| "eval_f1_weighted": 0.803223061183841, | |
| "eval_loss": 1.794916033744812, | |
| "eval_macro_fpr": 0.09024980030498875, | |
| "eval_macro_sensitivity": 0.6419443698855464, | |
| "eval_macro_specificity": 0.9192667115733391, | |
| "eval_precision": 0.8004278433937325, | |
| "eval_precision_macro": 0.6365492158020651, | |
| "eval_recall": 0.8062360801781737, | |
| "eval_recall_macro": 0.6419443698855464, | |
| "eval_runtime": 39.8561, | |
| "eval_samples_per_second": 11.266, | |
| "eval_steps_per_second": 1.43, | |
| "eval_weighted_fpr": 0.0741687979539642, | |
| "eval_weighted_sensitivity": 0.8062360801781737, | |
| "eval_weighted_specificity": 0.8708307661151824, | |
| "step": 5152 | |
| }, | |
| { | |
| "epoch": 24.0, | |
| "eval_accuracy": 0.8084632516703786, | |
| "eval_f1_macro": 0.6404728172579244, | |
| "eval_f1_micro": 0.8084632516703786, | |
| "eval_f1_weighted": 0.8054233127781282, | |
| "eval_loss": 1.8085839748382568, | |
| "eval_macro_fpr": 0.08932689177363767, | |
| "eval_macro_sensitivity": 0.6429096208507973, | |
| "eval_macro_specificity": 0.9199871726684399, | |
| "eval_precision": 0.8025958010658311, | |
| "eval_precision_macro": 0.6387293155835355, | |
| "eval_recall": 0.8084632516703786, | |
| "eval_recall_macro": 0.6429096208507973, | |
| "eval_runtime": 48.4278, | |
| "eval_samples_per_second": 9.272, | |
| "eval_steps_per_second": 1.177, | |
| "eval_weighted_fpr": 0.07319148936170213, | |
| "eval_weighted_sensitivity": 0.8084632516703786, | |
| "eval_weighted_specificity": 0.8714854390033809, | |
| "step": 5376 | |
| }, | |
| { | |
| "epoch": 24.553571428571427, | |
| "grad_norm": 0.0007463924121111631, | |
| "learning_rate": 9.136904761904762e-06, | |
| "loss": 0.0001, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 25.0, | |
| "eval_accuracy": 0.8084632516703786, | |
| "eval_f1_macro": 0.628744939271255, | |
| "eval_f1_micro": 0.8084632516703786, | |
| "eval_f1_weighted": 0.8028068286185739, | |
| "eval_loss": 1.832601547241211, | |
| "eval_macro_fpr": 0.09342569864054004, | |
| "eval_macro_sensitivity": 0.6250825883178824, | |
| "eval_macro_specificity": 0.9155494538554513, | |
| "eval_precision": 0.7987920798501041, | |
| "eval_precision_macro": 0.6342881535692682, | |
| "eval_recall": 0.8084632516703786, | |
| "eval_recall_macro": 0.6250825883178824, | |
| "eval_runtime": 34.7107, | |
| "eval_samples_per_second": 12.935, | |
| "eval_steps_per_second": 1.642, | |
| "eval_weighted_fpr": 0.07319148936170213, | |
| "eval_weighted_sensitivity": 0.8084632516703786, | |
| "eval_weighted_specificity": 0.8537345637514266, | |
| "step": 5600 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 6720, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 30, | |
| "save_steps": 500, | |
| "total_flos": 1.17875869483008e+16, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |