layout-xlm-base-finetuned-with-DocLayNet-base-at-paragraphlevel-ml512
/
checkpoint-6000
/trainer_state.json
{ | |
"best_metric": 0.7797986932721173, | |
"best_model_checkpoint": "DocLayNet/layout-xlm-base-finetuned-DocLayNet-base_paragraphs_ml512-v6/checkpoint-6000", | |
"epoch": 3.1982942430703627, | |
"global_step": 6000, | |
"is_hyper_param_search": false, | |
"is_local_process_zero": true, | |
"is_world_process_zero": true, | |
"log_history": [ | |
{ | |
"epoch": 0.11, | |
"eval_accuracy": 0.8841533429944805, | |
"eval_f1": 0.10664094791953704, | |
"eval_loss": 0.4427741765975952, | |
"eval_precision": 0.11536168521462639, | |
"eval_recall": 0.09914602903501281, | |
"eval_runtime": 348.7843, | |
"eval_samples_per_second": 4.607, | |
"eval_steps_per_second": 0.29, | |
"step": 200 | |
}, | |
{ | |
"epoch": 0.21, | |
"eval_accuracy": 0.9242655812394248, | |
"eval_f1": 0.44395873404441333, | |
"eval_loss": 0.30403900146484375, | |
"eval_precision": 0.4547734193086154, | |
"eval_recall": 0.4336464560204953, | |
"eval_runtime": 350.1211, | |
"eval_samples_per_second": 4.59, | |
"eval_steps_per_second": 0.288, | |
"step": 400 | |
}, | |
{ | |
"epoch": 0.27, | |
"learning_rate": 1.3209054593874834e-05, | |
"loss": 0.7241, | |
"step": 500 | |
}, | |
{ | |
"epoch": 0.32, | |
"eval_accuracy": 0.9359434268564053, | |
"eval_f1": 0.5543897040741045, | |
"eval_loss": 0.22647875547409058, | |
"eval_precision": 0.5330233291298865, | |
"eval_recall": 0.5775405636208369, | |
"eval_runtime": 342.3688, | |
"eval_samples_per_second": 4.694, | |
"eval_steps_per_second": 0.295, | |
"step": 600 | |
}, | |
{ | |
"epoch": 0.43, | |
"eval_accuracy": 0.947861481323102, | |
"eval_f1": 0.6015281512784394, | |
"eval_loss": 0.2140066921710968, | |
"eval_precision": 0.6013484680378937, | |
"eval_recall": 0.6017079419299743, | |
"eval_runtime": 342.7133, | |
"eval_samples_per_second": 4.689, | |
"eval_steps_per_second": 0.295, | |
"step": 800 | |
}, | |
{ | |
"epoch": 0.53, | |
"learning_rate": 1.927439656448986e-05, | |
"loss": 0.2343, | |
"step": 1000 | |
}, | |
{ | |
"epoch": 0.53, | |
"eval_accuracy": 0.9402091136821377, | |
"eval_f1": 0.6132131856006622, | |
"eval_loss": 0.285193532705307, | |
"eval_precision": 0.6641768748132656, | |
"eval_recall": 0.5695132365499573, | |
"eval_runtime": 341.4863, | |
"eval_samples_per_second": 4.706, | |
"eval_steps_per_second": 0.296, | |
"step": 1000 | |
}, | |
{ | |
"epoch": 0.64, | |
"eval_accuracy": 0.9539986854504707, | |
"eval_f1": 0.6604133950171894, | |
"eval_loss": 0.16944007575511932, | |
"eval_precision": 0.6564846848367226, | |
"eval_recall": 0.6643894107600341, | |
"eval_runtime": 342.4627, | |
"eval_samples_per_second": 4.692, | |
"eval_steps_per_second": 0.295, | |
"step": 1200 | |
}, | |
{ | |
"epoch": 0.75, | |
"eval_accuracy": 0.9354234142916347, | |
"eval_f1": 0.6198065842234727, | |
"eval_loss": 0.230770543217659, | |
"eval_precision": 0.5118828964212166, | |
"eval_recall": 0.785397096498719, | |
"eval_runtime": 332.1227, | |
"eval_samples_per_second": 4.839, | |
"eval_steps_per_second": 0.304, | |
"step": 1400 | |
}, | |
{ | |
"epoch": 0.8, | |
"learning_rate": 1.7793573226714055e-05, | |
"loss": 0.1913, | |
"step": 1500 | |
}, | |
{ | |
"epoch": 0.85, | |
"eval_accuracy": 0.9593994250825553, | |
"eval_f1": 0.6589563286454478, | |
"eval_loss": 0.1601293832063675, | |
"eval_precision": 0.7189582071471835, | |
"eval_recall": 0.6081981212638771, | |
"eval_runtime": 327.08, | |
"eval_samples_per_second": 4.913, | |
"eval_steps_per_second": 0.309, | |
"step": 1600 | |
}, | |
{ | |
"epoch": 0.96, | |
"eval_accuracy": 0.9540725958657681, | |
"eval_f1": 0.6596589238459277, | |
"eval_loss": 0.16707618534564972, | |
"eval_precision": 0.5790424570912376, | |
"eval_recall": 0.7663535439795047, | |
"eval_runtime": 324.9043, | |
"eval_samples_per_second": 4.946, | |
"eval_steps_per_second": 0.311, | |
"step": 1800 | |
}, | |
{ | |
"epoch": 1.07, | |
"learning_rate": 1.6315711535613804e-05, | |
"loss": 0.1346, | |
"step": 2000 | |
}, | |
{ | |
"epoch": 1.07, | |
"eval_accuracy": 0.9611943923112051, | |
"eval_f1": 0.6985545993817363, | |
"eval_loss": 0.15802155435085297, | |
"eval_precision": 0.6837585868498528, | |
"eval_recall": 0.71400512382579, | |
"eval_runtime": 326.3436, | |
"eval_samples_per_second": 4.924, | |
"eval_steps_per_second": 0.309, | |
"step": 2000 | |
}, | |
{ | |
"epoch": 1.17, | |
"eval_accuracy": 0.9597082650321906, | |
"eval_f1": 0.6896523517382412, | |
"eval_loss": 0.14230071008205414, | |
"eval_precision": 0.661773940345369, | |
"eval_recall": 0.7199829205807002, | |
"eval_runtime": 352.3832, | |
"eval_samples_per_second": 4.56, | |
"eval_steps_per_second": 0.287, | |
"step": 2200 | |
}, | |
{ | |
"epoch": 1.28, | |
"eval_accuracy": 0.9662862919936542, | |
"eval_f1": 0.6980116745713244, | |
"eval_loss": 0.15801101922988892, | |
"eval_precision": 0.7489724016441573, | |
"eval_recall": 0.6535439795046969, | |
"eval_runtime": 346.3949, | |
"eval_samples_per_second": 4.639, | |
"eval_steps_per_second": 0.292, | |
"step": 2400 | |
}, | |
{ | |
"epoch": 1.33, | |
"learning_rate": 1.483784984451355e-05, | |
"loss": 0.098, | |
"step": 2500 | |
}, | |
{ | |
"epoch": 1.39, | |
"eval_accuracy": 0.9615850616492053, | |
"eval_f1": 0.6800371073905552, | |
"eval_loss": 0.1394117921590805, | |
"eval_precision": 0.7044019401482566, | |
"eval_recall": 0.6573014517506405, | |
"eval_runtime": 340.6651, | |
"eval_samples_per_second": 4.717, | |
"eval_steps_per_second": 0.296, | |
"step": 2600 | |
}, | |
{ | |
"epoch": 1.49, | |
"eval_accuracy": 0.9685537579486693, | |
"eval_f1": 0.725116373203805, | |
"eval_loss": 0.1756211668252945, | |
"eval_precision": 0.6892651019622932, | |
"eval_recall": 0.7649017933390265, | |
"eval_runtime": 336.4339, | |
"eval_samples_per_second": 4.777, | |
"eval_steps_per_second": 0.3, | |
"step": 2800 | |
}, | |
{ | |
"epoch": 1.6, | |
"learning_rate": 1.33599881534133e-05, | |
"loss": 0.0999, | |
"step": 3000 | |
}, | |
{ | |
"epoch": 1.6, | |
"eval_accuracy": 0.9636387153313958, | |
"eval_f1": 0.69845832244578, | |
"eval_loss": 0.154231458902359, | |
"eval_precision": 0.7126733025239957, | |
"eval_recall": 0.684799316823228, | |
"eval_runtime": 338.3198, | |
"eval_samples_per_second": 4.75, | |
"eval_steps_per_second": 0.299, | |
"step": 3000 | |
}, | |
{ | |
"epoch": 1.71, | |
"eval_accuracy": 0.9670465134081412, | |
"eval_f1": 0.7096949152542371, | |
"eval_loss": 0.11871206015348434, | |
"eval_precision": 0.7538166106577052, | |
"eval_recall": 0.6704526046114432, | |
"eval_runtime": 330.1098, | |
"eval_samples_per_second": 4.868, | |
"eval_steps_per_second": 0.306, | |
"step": 3200 | |
}, | |
{ | |
"epoch": 1.81, | |
"eval_accuracy": 0.9585151397566763, | |
"eval_f1": 0.7426724326213252, | |
"eval_loss": 0.17930778861045837, | |
"eval_precision": 0.7601716891710633, | |
"eval_recall": 0.7259607173356106, | |
"eval_runtime": 334.0583, | |
"eval_samples_per_second": 4.811, | |
"eval_steps_per_second": 0.302, | |
"step": 3400 | |
}, | |
{ | |
"epoch": 1.87, | |
"learning_rate": 1.1879164815637496e-05, | |
"loss": 0.0972, | |
"step": 3500 | |
}, | |
{ | |
"epoch": 1.92, | |
"eval_accuracy": 0.9621024345562867, | |
"eval_f1": 0.7188762229602368, | |
"eval_loss": 0.18358349800109863, | |
"eval_precision": 0.7575671585319712, | |
"eval_recall": 0.6839453458582408, | |
"eval_runtime": 323.7426, | |
"eval_samples_per_second": 4.964, | |
"eval_steps_per_second": 0.312, | |
"step": 3600 | |
}, | |
{ | |
"epoch": 2.03, | |
"eval_accuracy": 0.9641745658423015, | |
"eval_f1": 0.7188527825286147, | |
"eval_loss": 0.1464972347021103, | |
"eval_precision": 0.7388443162354638, | |
"eval_recall": 0.6999146029035013, | |
"eval_runtime": 324.6992, | |
"eval_samples_per_second": 4.949, | |
"eval_steps_per_second": 0.311, | |
"step": 3800 | |
}, | |
{ | |
"epoch": 2.13, | |
"learning_rate": 1.0398341477861691e-05, | |
"loss": 0.0662, | |
"step": 4000 | |
}, | |
{ | |
"epoch": 2.13, | |
"eval_accuracy": 0.9690632118826831, | |
"eval_f1": 0.7450159228722244, | |
"eval_loss": 0.1409263014793396, | |
"eval_precision": 0.7615267992508695, | |
"eval_recall": 0.7292058070025619, | |
"eval_runtime": 314.9474, | |
"eval_samples_per_second": 5.102, | |
"eval_steps_per_second": 0.321, | |
"step": 4000 | |
}, | |
{ | |
"epoch": 2.24, | |
"eval_accuracy": 0.9615428271261782, | |
"eval_f1": 0.7431866723622383, | |
"eval_loss": 0.17204762995243073, | |
"eval_precision": 0.7435042735042735, | |
"eval_recall": 0.7428693424423569, | |
"eval_runtime": 307.7666, | |
"eval_samples_per_second": 5.221, | |
"eval_steps_per_second": 0.328, | |
"step": 4200 | |
}, | |
{ | |
"epoch": 2.35, | |
"eval_accuracy": 0.9666637630432086, | |
"eval_f1": 0.7338054481470216, | |
"eval_loss": 0.14404301345348358, | |
"eval_precision": 0.7468824621915627, | |
"eval_recall": 0.7211784799316823, | |
"eval_runtime": 310.6556, | |
"eval_samples_per_second": 5.173, | |
"eval_steps_per_second": 0.325, | |
"step": 4400 | |
}, | |
{ | |
"epoch": 2.4, | |
"learning_rate": 8.917518140085888e-06, | |
"loss": 0.0581, | |
"step": 4500 | |
}, | |
{ | |
"epoch": 2.45, | |
"eval_accuracy": 0.9657451621673702, | |
"eval_f1": 0.7135284422468708, | |
"eval_loss": 0.19275733828544617, | |
"eval_precision": 0.7457863860694665, | |
"eval_recall": 0.6839453458582408, | |
"eval_runtime": 303.8599, | |
"eval_samples_per_second": 5.289, | |
"eval_steps_per_second": 0.332, | |
"step": 4600 | |
}, | |
{ | |
"epoch": 2.56, | |
"eval_accuracy": 0.969187275794075, | |
"eval_f1": 0.7378380713730234, | |
"eval_loss": 0.16446392238140106, | |
"eval_precision": 0.7466771598461001, | |
"eval_recall": 0.7292058070025619, | |
"eval_runtime": 304.6717, | |
"eval_samples_per_second": 5.275, | |
"eval_steps_per_second": 0.332, | |
"step": 4800 | |
}, | |
{ | |
"epoch": 2.67, | |
"learning_rate": 7.436694802310085e-06, | |
"loss": 0.0538, | |
"step": 5000 | |
}, | |
{ | |
"epoch": 2.67, | |
"eval_accuracy": 0.9655762240752619, | |
"eval_f1": 0.7619294158253516, | |
"eval_loss": 0.15173019468784332, | |
"eval_precision": 0.7699686083013603, | |
"eval_recall": 0.7540563620836892, | |
"eval_runtime": 301.8453, | |
"eval_samples_per_second": 5.324, | |
"eval_steps_per_second": 0.335, | |
"step": 5000 | |
}, | |
{ | |
"epoch": 2.77, | |
"eval_accuracy": 0.9683900991719394, | |
"eval_f1": 0.7727910873601738, | |
"eval_loss": 0.16763265430927277, | |
"eval_precision": 0.8226786230835985, | |
"eval_recall": 0.7286080273270709, | |
"eval_runtime": 303.0486, | |
"eval_samples_per_second": 5.303, | |
"eval_steps_per_second": 0.333, | |
"step": 5200 | |
}, | |
{ | |
"epoch": 2.88, | |
"eval_accuracy": 0.9724974065363203, | |
"eval_f1": 0.7607919220424181, | |
"eval_loss": 0.12770096957683563, | |
"eval_precision": 0.7864891968274227, | |
"eval_recall": 0.7367207514944492, | |
"eval_runtime": 298.6206, | |
"eval_samples_per_second": 5.381, | |
"eval_steps_per_second": 0.338, | |
"step": 5400 | |
}, | |
{ | |
"epoch": 2.93, | |
"learning_rate": 5.955871464534282e-06, | |
"loss": 0.0432, | |
"step": 5500 | |
}, | |
{ | |
"epoch": 2.99, | |
"eval_accuracy": 0.9692585465516832, | |
"eval_f1": 0.7784317119612255, | |
"eval_loss": 0.15319885313510895, | |
"eval_precision": 0.78908580452711, | |
"eval_recall": 0.7680614859094791, | |
"eval_runtime": 299.3727, | |
"eval_samples_per_second": 5.368, | |
"eval_steps_per_second": 0.337, | |
"step": 5600 | |
}, | |
{ | |
"epoch": 3.09, | |
"eval_accuracy": 0.9691767171633182, | |
"eval_f1": 0.7783425414364641, | |
"eval_loss": 0.1701451987028122, | |
"eval_precision": 0.8066880439761795, | |
"eval_recall": 0.7519214346712212, | |
"eval_runtime": 330.9586, | |
"eval_samples_per_second": 4.856, | |
"eval_steps_per_second": 0.305, | |
"step": 5800 | |
}, | |
{ | |
"epoch": 3.2, | |
"learning_rate": 4.475048126758478e-06, | |
"loss": 0.0272, | |
"step": 6000 | |
}, | |
{ | |
"epoch": 3.2, | |
"eval_accuracy": 0.973239150346983, | |
"eval_f1": 0.7797986932721173, | |
"eval_loss": 0.11586667597293854, | |
"eval_precision": 0.8071650520928533, | |
"eval_recall": 0.7542271562766866, | |
"eval_runtime": 326.8152, | |
"eval_samples_per_second": 4.917, | |
"eval_steps_per_second": 0.309, | |
"step": 6000 | |
} | |
], | |
"max_steps": 7504, | |
"num_train_epochs": 4, | |
"total_flos": 2.5913706725376e+16, | |
"trial_name": null, | |
"trial_params": null | |
} | |