{ "best_metric": 72.54565014579936, "best_model_checkpoint": "/data/jcanete/all_results/mlqa/albeto_xlarge/epochs_4_bs_16_lr_5e-6/checkpoint-14400", "epoch": 3.999951250426559, "global_step": 20512, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.06, "eval_exact_match": 32.6, "eval_f1": 56.15794028928574, "step": 300 }, { "epoch": 0.1, "learning_rate": 4.8798264430577225e-06, "loss": 3.0063, "step": 500 }, { "epoch": 0.12, "eval_exact_match": 38.2, "eval_f1": 62.74933530055881, "step": 600 }, { "epoch": 0.18, "eval_exact_match": 38.4, "eval_f1": 64.44399563896586, "step": 900 }, { "epoch": 0.19, "learning_rate": 4.757946567862715e-06, "loss": 2.1048, "step": 1000 }, { "epoch": 0.23, "eval_exact_match": 39.6, "eval_f1": 65.51476428813997, "step": 1200 }, { "epoch": 0.29, "learning_rate": 4.6363104524180965e-06, "loss": 1.9106, "step": 1500 }, { "epoch": 0.29, "eval_exact_match": 39.2, "eval_f1": 66.56103827722897, "step": 1500 }, { "epoch": 0.35, "eval_exact_match": 40.2, "eval_f1": 66.52847493412486, "step": 1800 }, { "epoch": 0.39, "learning_rate": 4.51443057722309e-06, "loss": 1.8845, "step": 2000 }, { "epoch": 0.41, "eval_exact_match": 43.4, "eval_f1": 68.06136327215253, "step": 2100 }, { "epoch": 0.47, "eval_exact_match": 45.4, "eval_f1": 68.94645961954767, "step": 2400 }, { "epoch": 0.49, "learning_rate": 4.392550702028082e-06, "loss": 1.8377, "step": 2500 }, { "epoch": 0.53, "eval_exact_match": 44.6, "eval_f1": 69.74983503448324, "step": 2700 }, { "epoch": 0.58, "learning_rate": 4.270670826833074e-06, "loss": 1.8444, "step": 3000 }, { "epoch": 0.58, "eval_exact_match": 44.8, "eval_f1": 68.72422391765423, "step": 3000 }, { "epoch": 0.64, "eval_exact_match": 44.2, "eval_f1": 68.80247260343292, "step": 3300 }, { "epoch": 0.68, "learning_rate": 4.1487909516380656e-06, "loss": 1.7901, "step": 3500 }, { "epoch": 0.7, "eval_exact_match": 45.6, "eval_f1": 70.11442299691804, "step": 3600 }, { "epoch": 0.76, "eval_exact_match": 45.4, "eval_f1": 71.84540250423026, "step": 3900 }, { "epoch": 0.78, "learning_rate": 4.026911076443058e-06, "loss": 1.7667, "step": 4000 }, { "epoch": 0.82, "eval_exact_match": 45.0, "eval_f1": 71.82625857885415, "step": 4200 }, { "epoch": 0.88, "learning_rate": 3.90503120124805e-06, "loss": 1.7574, "step": 4500 }, { "epoch": 0.88, "eval_exact_match": 46.2, "eval_f1": 71.38925591495055, "step": 4500 }, { "epoch": 0.94, "eval_exact_match": 47.0, "eval_f1": 70.80568844306376, "step": 4800 }, { "epoch": 0.97, "learning_rate": 3.7831513260530424e-06, "loss": 1.7351, "step": 5000 }, { "epoch": 0.99, "eval_exact_match": 46.4, "eval_f1": 69.96654581975262, "step": 5100 }, { "epoch": 1.05, "eval_exact_match": 46.8, "eval_f1": 71.40276694078021, "step": 5400 }, { "epoch": 1.07, "learning_rate": 3.661271450858034e-06, "loss": 1.5758, "step": 5500 }, { "epoch": 1.11, "eval_exact_match": 45.4, "eval_f1": 70.95444550622851, "step": 5700 }, { "epoch": 1.17, "learning_rate": 3.539391575663027e-06, "loss": 1.5533, "step": 6000 }, { "epoch": 1.17, "eval_exact_match": 45.8, "eval_f1": 71.16796131798688, "step": 6000 }, { "epoch": 1.23, "eval_exact_match": 45.4, "eval_f1": 71.17415998507276, "step": 6300 }, { "epoch": 1.27, "learning_rate": 3.417511700468019e-06, "loss": 1.5236, "step": 6500 }, { "epoch": 1.29, "eval_exact_match": 47.2, "eval_f1": 71.96064560948882, "step": 6600 }, { "epoch": 1.35, "eval_exact_match": 46.8, "eval_f1": 71.57893095238637, "step": 6900 }, { "epoch": 1.37, "learning_rate": 3.2956318252730114e-06, "loss": 1.5082, "step": 7000 }, { "epoch": 1.4, "eval_exact_match": 46.2, "eval_f1": 71.0417637265508, "step": 7200 }, { "epoch": 1.46, "learning_rate": 3.1739957098283936e-06, "loss": 1.5279, "step": 7500 }, { "epoch": 1.46, "eval_exact_match": 46.8, "eval_f1": 71.61420929224126, "step": 7500 }, { "epoch": 1.52, "eval_exact_match": 47.4, "eval_f1": 71.4861929271134, "step": 7800 }, { "epoch": 1.56, "learning_rate": 3.052115834633386e-06, "loss": 1.5222, "step": 8000 }, { "epoch": 1.58, "eval_exact_match": 45.2, "eval_f1": 71.80594062526184, "step": 8100 }, { "epoch": 1.64, "eval_exact_match": 47.2, "eval_f1": 71.47797864113711, "step": 8400 }, { "epoch": 1.66, "learning_rate": 2.9302359594383777e-06, "loss": 1.5115, "step": 8500 }, { "epoch": 1.7, "eval_exact_match": 46.4, "eval_f1": 71.83295074421588, "step": 8700 }, { "epoch": 1.76, "learning_rate": 2.80859984399376e-06, "loss": 1.5468, "step": 9000 }, { "epoch": 1.76, "eval_exact_match": 45.0, "eval_f1": 71.84617673539766, "step": 9000 }, { "epoch": 1.81, "eval_exact_match": 47.0, "eval_f1": 72.05451513566037, "step": 9300 }, { "epoch": 1.85, "learning_rate": 2.686719968798752e-06, "loss": 1.5399, "step": 9500 }, { "epoch": 1.87, "eval_exact_match": 44.8, "eval_f1": 71.05231817141241, "step": 9600 }, { "epoch": 1.93, "eval_exact_match": 44.8, "eval_f1": 70.76686017808167, "step": 9900 }, { "epoch": 1.95, "learning_rate": 2.5650838533541343e-06, "loss": 1.507, "step": 10000 }, { "epoch": 1.99, "eval_exact_match": 45.0, "eval_f1": 71.0282044654407, "step": 10200 }, { "epoch": 2.05, "learning_rate": 2.4432039781591266e-06, "loss": 1.4224, "step": 10500 }, { "epoch": 2.05, "eval_exact_match": 45.0, "eval_f1": 71.18815762834795, "step": 10500 }, { "epoch": 2.11, "eval_exact_match": 45.6, "eval_f1": 71.5028784465516, "step": 10800 }, { "epoch": 2.15, "learning_rate": 2.321324102964119e-06, "loss": 1.3113, "step": 11000 }, { "epoch": 2.16, "eval_exact_match": 44.4, "eval_f1": 71.15086056380413, "step": 11100 }, { "epoch": 2.22, "eval_exact_match": 45.8, "eval_f1": 71.54431190031305, "step": 11400 }, { "epoch": 2.24, "learning_rate": 2.1994442277691107e-06, "loss": 1.3522, "step": 11500 }, { "epoch": 2.28, "eval_exact_match": 45.0, "eval_f1": 71.81209765647928, "step": 11700 }, { "epoch": 2.34, "learning_rate": 2.0775643525741034e-06, "loss": 1.3144, "step": 12000 }, { "epoch": 2.34, "eval_exact_match": 43.6, "eval_f1": 71.12699620880493, "step": 12000 }, { "epoch": 2.4, "eval_exact_match": 44.2, "eval_f1": 71.41750598254089, "step": 12300 }, { "epoch": 2.44, "learning_rate": 1.955684477379095e-06, "loss": 1.2981, "step": 12500 }, { "epoch": 2.46, "eval_exact_match": 47.2, "eval_f1": 71.45377213612431, "step": 12600 }, { "epoch": 2.52, "eval_exact_match": 45.6, "eval_f1": 70.77094339280019, "step": 12900 }, { "epoch": 2.54, "learning_rate": 1.8340483619344776e-06, "loss": 1.3352, "step": 13000 }, { "epoch": 2.57, "eval_exact_match": 45.2, "eval_f1": 71.05521519892982, "step": 13200 }, { "epoch": 2.63, "learning_rate": 1.7121684867394696e-06, "loss": 1.2933, "step": 13500 }, { "epoch": 2.63, "eval_exact_match": 44.0, "eval_f1": 70.46603727609677, "step": 13500 }, { "epoch": 2.69, "eval_exact_match": 46.2, "eval_f1": 72.22404409889462, "step": 13800 }, { "epoch": 2.73, "learning_rate": 1.590288611544462e-06, "loss": 1.3227, "step": 14000 }, { "epoch": 2.75, "eval_exact_match": 44.8, "eval_f1": 71.47399920055251, "step": 14100 }, { "epoch": 2.81, "eval_exact_match": 47.0, "eval_f1": 72.54565014579936, "step": 14400 }, { "epoch": 2.83, "learning_rate": 1.468408736349454e-06, "loss": 1.3303, "step": 14500 }, { "epoch": 2.87, "eval_exact_match": 46.0, "eval_f1": 71.42400233167065, "step": 14700 }, { "epoch": 2.93, "learning_rate": 1.3467726209048363e-06, "loss": 1.3055, "step": 15000 }, { "epoch": 2.93, "eval_exact_match": 45.6, "eval_f1": 71.51695078489598, "step": 15000 }, { "epoch": 2.98, "eval_exact_match": 46.6, "eval_f1": 71.50435043692885, "step": 15300 }, { "epoch": 3.02, "learning_rate": 1.2248927457098284e-06, "loss": 1.2986, "step": 15500 }, { "epoch": 3.04, "eval_exact_match": 45.8, "eval_f1": 71.06499193256445, "step": 15600 }, { "epoch": 3.1, "eval_exact_match": 45.6, "eval_f1": 70.9580439024025, "step": 15900 }, { "epoch": 3.12, "learning_rate": 1.1032566302652108e-06, "loss": 1.1564, "step": 16000 }, { "epoch": 3.16, "eval_exact_match": 45.4, "eval_f1": 70.85970253602206, "step": 16200 }, { "epoch": 3.22, "learning_rate": 9.813767550702028e-07, "loss": 1.1861, "step": 16500 }, { "epoch": 3.22, "eval_exact_match": 44.8, "eval_f1": 70.73497821095955, "step": 16500 }, { "epoch": 3.28, "eval_exact_match": 44.4, "eval_f1": 70.1719318341078, "step": 16800 }, { "epoch": 3.32, "learning_rate": 8.594968798751951e-07, "loss": 1.177, "step": 17000 }, { "epoch": 3.33, "eval_exact_match": 44.6, "eval_f1": 71.00348481939916, "step": 17100 }, { "epoch": 3.39, "eval_exact_match": 44.8, "eval_f1": 71.04230855179065, "step": 17400 }, { "epoch": 3.41, "learning_rate": 7.376170046801873e-07, "loss": 1.1581, "step": 17500 }, { "epoch": 3.45, "eval_exact_match": 45.0, "eval_f1": 71.32642226951928, "step": 17700 }, { "epoch": 3.51, "learning_rate": 6.157371294851794e-07, "loss": 1.1398, "step": 18000 }, { "epoch": 3.51, "eval_exact_match": 44.8, "eval_f1": 70.45204643207842, "step": 18000 }, { "epoch": 3.57, "eval_exact_match": 44.4, "eval_f1": 70.37255165848242, "step": 18300 }, { "epoch": 3.61, "learning_rate": 4.938572542901717e-07, "loss": 1.1478, "step": 18500 }, { "epoch": 3.63, "eval_exact_match": 44.0, "eval_f1": 70.48540517137832, "step": 18600 }, { "epoch": 3.69, "eval_exact_match": 44.6, "eval_f1": 70.22760956321906, "step": 18900 }, { "epoch": 3.71, "learning_rate": 3.7197737909516383e-07, "loss": 1.1574, "step": 19000 }, { "epoch": 3.74, "eval_exact_match": 45.0, "eval_f1": 70.64118099179048, "step": 19200 }, { "epoch": 3.8, "learning_rate": 2.5009750390015604e-07, "loss": 1.1667, "step": 19500 }, { "epoch": 3.8, "eval_exact_match": 43.4, "eval_f1": 70.15239653063759, "step": 19500 }, { "epoch": 3.86, "eval_exact_match": 44.4, "eval_f1": 70.2883601771379, "step": 19800 }, { "epoch": 3.9, "learning_rate": 1.2846138845553822e-07, "loss": 1.1602, "step": 20000 }, { "epoch": 3.92, "eval_exact_match": 43.8, "eval_f1": 70.33719627550457, "step": 20100 }, { "epoch": 3.98, "eval_exact_match": 43.8, "eval_f1": 70.33111568942401, "step": 20400 }, { "epoch": 4.0, "learning_rate": 6.581513260530422e-09, "loss": 1.1577, "step": 20500 }, { "epoch": 4.0, "step": 20512, "total_flos": 2.7267161088888864e+16, "train_loss": 1.4912065643230206, "train_runtime": 23906.2096, "train_samples_per_second": 13.729, "train_steps_per_second": 0.858 } ], "max_steps": 20512, "num_train_epochs": 4, "total_flos": 2.7267161088888864e+16, "trial_name": null, "trial_params": null }