| { | |
| "best_metric": 0.5063937306404114, | |
| "best_model_checkpoint": "../../experiments_checkpoints/MAdAiLab/Qwen/Qwen1.5_1.8B_ledgar/checkpoint-1800", | |
| "epoch": 3.0, | |
| "eval_steps": 100, | |
| "global_step": 2814, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.03, | |
| "grad_norm": 80.4836196899414, | |
| "learning_rate": 4.9555792466240235e-06, | |
| "loss": 7.9887, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 60.927364349365234, | |
| "learning_rate": 4.911158493248046e-06, | |
| "loss": 3.1908, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "grad_norm": 49.68091583251953, | |
| "learning_rate": 4.866737739872069e-06, | |
| "loss": 1.7183, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "grad_norm": 55.176666259765625, | |
| "learning_rate": 4.822316986496091e-06, | |
| "loss": 1.3077, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "eval_accuracy": 0.7277, | |
| "eval_f1_macro": 0.5770831474844406, | |
| "eval_f1_micro": 0.7277, | |
| "eval_loss": 1.0944937467575073, | |
| "eval_runtime": 25.447, | |
| "eval_samples_per_second": 392.973, | |
| "eval_steps_per_second": 6.17, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "grad_norm": 46.64506530761719, | |
| "learning_rate": 4.777896233120114e-06, | |
| "loss": 1.1393, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "grad_norm": 41.11891174316406, | |
| "learning_rate": 4.733475479744136e-06, | |
| "loss": 1.0243, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "grad_norm": 34.20009994506836, | |
| "learning_rate": 4.6890547263681595e-06, | |
| "loss": 0.9005, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "grad_norm": 38.80377197265625, | |
| "learning_rate": 4.644633972992183e-06, | |
| "loss": 0.8627, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "eval_accuracy": 0.7907, | |
| "eval_f1_macro": 0.6657039157603262, | |
| "eval_f1_micro": 0.7907, | |
| "eval_loss": 0.8368468880653381, | |
| "eval_runtime": 25.9484, | |
| "eval_samples_per_second": 385.38, | |
| "eval_steps_per_second": 6.05, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "grad_norm": 34.973506927490234, | |
| "learning_rate": 4.600213219616206e-06, | |
| "loss": 0.7896, | |
| "step": 225 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "grad_norm": 29.98388671875, | |
| "learning_rate": 4.555792466240228e-06, | |
| "loss": 0.8307, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "grad_norm": 27.09973907470703, | |
| "learning_rate": 4.51137171286425e-06, | |
| "loss": 0.7846, | |
| "step": 275 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "grad_norm": 36.151161193847656, | |
| "learning_rate": 4.466950959488273e-06, | |
| "loss": 0.7179, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "eval_accuracy": 0.7971, | |
| "eval_f1_macro": 0.6861778340669753, | |
| "eval_f1_micro": 0.7971, | |
| "eval_loss": 0.7824062705039978, | |
| "eval_runtime": 25.9003, | |
| "eval_samples_per_second": 386.095, | |
| "eval_steps_per_second": 6.062, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "grad_norm": 33.09822463989258, | |
| "learning_rate": 4.422530206112296e-06, | |
| "loss": 0.7133, | |
| "step": 325 | |
| }, | |
| { | |
| "epoch": 0.37, | |
| "grad_norm": 35.52923583984375, | |
| "learning_rate": 4.378109452736319e-06, | |
| "loss": 0.7354, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.4, | |
| "grad_norm": 39.79545211791992, | |
| "learning_rate": 4.333688699360342e-06, | |
| "loss": 0.6619, | |
| "step": 375 | |
| }, | |
| { | |
| "epoch": 0.43, | |
| "grad_norm": 30.13161849975586, | |
| "learning_rate": 4.289267945984365e-06, | |
| "loss": 0.6961, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.43, | |
| "eval_accuracy": 0.8138, | |
| "eval_f1_macro": 0.6992465625213966, | |
| "eval_f1_micro": 0.8138, | |
| "eval_loss": 0.6951531171798706, | |
| "eval_runtime": 25.6082, | |
| "eval_samples_per_second": 390.5, | |
| "eval_steps_per_second": 6.131, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.45, | |
| "grad_norm": 27.575519561767578, | |
| "learning_rate": 4.244847192608387e-06, | |
| "loss": 0.7162, | |
| "step": 425 | |
| }, | |
| { | |
| "epoch": 0.48, | |
| "grad_norm": 35.084754943847656, | |
| "learning_rate": 4.200426439232409e-06, | |
| "loss": 0.7722, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 0.51, | |
| "grad_norm": 28.47511863708496, | |
| "learning_rate": 4.156005685856432e-06, | |
| "loss": 0.6866, | |
| "step": 475 | |
| }, | |
| { | |
| "epoch": 0.53, | |
| "grad_norm": 32.34709548950195, | |
| "learning_rate": 4.1115849324804554e-06, | |
| "loss": 0.745, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.53, | |
| "eval_accuracy": 0.8121, | |
| "eval_f1_macro": 0.7033560293953169, | |
| "eval_f1_micro": 0.8121, | |
| "eval_loss": 0.6718780994415283, | |
| "eval_runtime": 25.9161, | |
| "eval_samples_per_second": 385.86, | |
| "eval_steps_per_second": 6.058, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.56, | |
| "grad_norm": 25.5845890045166, | |
| "learning_rate": 4.067164179104478e-06, | |
| "loss": 0.6535, | |
| "step": 525 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "grad_norm": 22.466503143310547, | |
| "learning_rate": 4.022743425728501e-06, | |
| "loss": 0.5969, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 0.61, | |
| "grad_norm": 27.53134536743164, | |
| "learning_rate": 3.978322672352524e-06, | |
| "loss": 0.5926, | |
| "step": 575 | |
| }, | |
| { | |
| "epoch": 0.64, | |
| "grad_norm": 31.356454849243164, | |
| "learning_rate": 3.933901918976546e-06, | |
| "loss": 0.6505, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.64, | |
| "eval_accuracy": 0.834, | |
| "eval_f1_macro": 0.7469091035082649, | |
| "eval_f1_micro": 0.834, | |
| "eval_loss": 0.6219750046730042, | |
| "eval_runtime": 25.9316, | |
| "eval_samples_per_second": 385.63, | |
| "eval_steps_per_second": 6.054, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "grad_norm": 37.17654800415039, | |
| "learning_rate": 3.889481165600569e-06, | |
| "loss": 0.6171, | |
| "step": 625 | |
| }, | |
| { | |
| "epoch": 0.69, | |
| "grad_norm": 26.71038055419922, | |
| "learning_rate": 3.8450604122245914e-06, | |
| "loss": 0.6218, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 0.72, | |
| "grad_norm": 27.787952423095703, | |
| "learning_rate": 3.8006396588486145e-06, | |
| "loss": 0.6124, | |
| "step": 675 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "grad_norm": 30.405912399291992, | |
| "learning_rate": 3.756218905472637e-06, | |
| "loss": 0.5914, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "eval_accuracy": 0.8362, | |
| "eval_f1_macro": 0.7410957777496914, | |
| "eval_f1_micro": 0.8362, | |
| "eval_loss": 0.6109625101089478, | |
| "eval_runtime": 25.6247, | |
| "eval_samples_per_second": 390.248, | |
| "eval_steps_per_second": 6.127, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.77, | |
| "grad_norm": 30.52012062072754, | |
| "learning_rate": 3.71179815209666e-06, | |
| "loss": 0.5711, | |
| "step": 725 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "grad_norm": 30.88004493713379, | |
| "learning_rate": 3.667377398720683e-06, | |
| "loss": 0.6695, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "grad_norm": 22.504459381103516, | |
| "learning_rate": 3.622956645344705e-06, | |
| "loss": 0.5731, | |
| "step": 775 | |
| }, | |
| { | |
| "epoch": 0.85, | |
| "grad_norm": 21.515512466430664, | |
| "learning_rate": 3.578535891968728e-06, | |
| "loss": 0.5837, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 0.85, | |
| "eval_accuracy": 0.8385, | |
| "eval_f1_macro": 0.7413235492734335, | |
| "eval_f1_micro": 0.8385, | |
| "eval_loss": 0.5766780972480774, | |
| "eval_runtime": 25.6608, | |
| "eval_samples_per_second": 389.7, | |
| "eval_steps_per_second": 6.118, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 0.88, | |
| "grad_norm": 30.9660587310791, | |
| "learning_rate": 3.534115138592751e-06, | |
| "loss": 0.6085, | |
| "step": 825 | |
| }, | |
| { | |
| "epoch": 0.91, | |
| "grad_norm": 18.883647918701172, | |
| "learning_rate": 3.4896943852167736e-06, | |
| "loss": 0.5121, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 0.93, | |
| "grad_norm": 24.548561096191406, | |
| "learning_rate": 3.4452736318407963e-06, | |
| "loss": 0.5621, | |
| "step": 875 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "grad_norm": 29.833791732788086, | |
| "learning_rate": 3.4008528784648194e-06, | |
| "loss": 0.5218, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "eval_accuracy": 0.849, | |
| "eval_f1_macro": 0.7702797685808792, | |
| "eval_f1_micro": 0.849, | |
| "eval_loss": 0.5365203022956848, | |
| "eval_runtime": 25.9091, | |
| "eval_samples_per_second": 385.964, | |
| "eval_steps_per_second": 6.06, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 0.99, | |
| "grad_norm": 27.948928833007812, | |
| "learning_rate": 3.3564321250888416e-06, | |
| "loss": 0.5681, | |
| "step": 925 | |
| }, | |
| { | |
| "epoch": 1.01, | |
| "grad_norm": 19.800880432128906, | |
| "learning_rate": 3.3120113717128643e-06, | |
| "loss": 0.4014, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 1.04, | |
| "grad_norm": 19.333465576171875, | |
| "learning_rate": 3.2675906183368874e-06, | |
| "loss": 0.2795, | |
| "step": 975 | |
| }, | |
| { | |
| "epoch": 1.07, | |
| "grad_norm": 22.315195083618164, | |
| "learning_rate": 3.22316986496091e-06, | |
| "loss": 0.2632, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.07, | |
| "eval_accuracy": 0.8562, | |
| "eval_f1_macro": 0.7683569808757446, | |
| "eval_f1_micro": 0.8562, | |
| "eval_loss": 0.5503664016723633, | |
| "eval_runtime": 25.5198, | |
| "eval_samples_per_second": 391.852, | |
| "eval_steps_per_second": 6.152, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.09, | |
| "grad_norm": 24.819501876831055, | |
| "learning_rate": 3.1787491115849327e-06, | |
| "loss": 0.2532, | |
| "step": 1025 | |
| }, | |
| { | |
| "epoch": 1.12, | |
| "grad_norm": 21.534936904907227, | |
| "learning_rate": 3.1343283582089558e-06, | |
| "loss": 0.2311, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 1.15, | |
| "grad_norm": 24.088809967041016, | |
| "learning_rate": 3.0899076048329785e-06, | |
| "loss": 0.3134, | |
| "step": 1075 | |
| }, | |
| { | |
| "epoch": 1.17, | |
| "grad_norm": 27.605493545532227, | |
| "learning_rate": 3.0454868514570007e-06, | |
| "loss": 0.2607, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 1.17, | |
| "eval_accuracy": 0.8525, | |
| "eval_f1_macro": 0.7656891626030512, | |
| "eval_f1_micro": 0.8525, | |
| "eval_loss": 0.5496523380279541, | |
| "eval_runtime": 25.7081, | |
| "eval_samples_per_second": 388.982, | |
| "eval_steps_per_second": 6.107, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 1.2, | |
| "grad_norm": 22.955158233642578, | |
| "learning_rate": 3.0010660980810234e-06, | |
| "loss": 0.2674, | |
| "step": 1125 | |
| }, | |
| { | |
| "epoch": 1.23, | |
| "grad_norm": 19.089893341064453, | |
| "learning_rate": 2.9566453447050464e-06, | |
| "loss": 0.2074, | |
| "step": 1150 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "grad_norm": 19.285688400268555, | |
| "learning_rate": 2.912224591329069e-06, | |
| "loss": 0.2488, | |
| "step": 1175 | |
| }, | |
| { | |
| "epoch": 1.28, | |
| "grad_norm": 23.45233726501465, | |
| "learning_rate": 2.867803837953092e-06, | |
| "loss": 0.274, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 1.28, | |
| "eval_accuracy": 0.8584, | |
| "eval_f1_macro": 0.7746299057445165, | |
| "eval_f1_micro": 0.8584, | |
| "eval_loss": 0.5439000129699707, | |
| "eval_runtime": 25.9014, | |
| "eval_samples_per_second": 386.079, | |
| "eval_steps_per_second": 6.061, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 1.31, | |
| "grad_norm": 31.231454849243164, | |
| "learning_rate": 2.823383084577115e-06, | |
| "loss": 0.2624, | |
| "step": 1225 | |
| }, | |
| { | |
| "epoch": 1.33, | |
| "grad_norm": 28.1010799407959, | |
| "learning_rate": 2.7789623312011375e-06, | |
| "loss": 0.2992, | |
| "step": 1250 | |
| }, | |
| { | |
| "epoch": 1.36, | |
| "grad_norm": 30.002384185791016, | |
| "learning_rate": 2.7345415778251598e-06, | |
| "loss": 0.2589, | |
| "step": 1275 | |
| }, | |
| { | |
| "epoch": 1.39, | |
| "grad_norm": 23.61323356628418, | |
| "learning_rate": 2.690120824449183e-06, | |
| "loss": 0.2216, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 1.39, | |
| "eval_accuracy": 0.8563, | |
| "eval_f1_macro": 0.7753520513346309, | |
| "eval_f1_micro": 0.8563, | |
| "eval_loss": 0.5687375068664551, | |
| "eval_runtime": 25.9424, | |
| "eval_samples_per_second": 385.47, | |
| "eval_steps_per_second": 6.052, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 1.41, | |
| "grad_norm": 27.56183433532715, | |
| "learning_rate": 2.6457000710732055e-06, | |
| "loss": 0.2845, | |
| "step": 1325 | |
| }, | |
| { | |
| "epoch": 1.44, | |
| "grad_norm": 18.88576316833496, | |
| "learning_rate": 2.601279317697228e-06, | |
| "loss": 0.2685, | |
| "step": 1350 | |
| }, | |
| { | |
| "epoch": 1.47, | |
| "grad_norm": 19.662220001220703, | |
| "learning_rate": 2.5568585643212513e-06, | |
| "loss": 0.2489, | |
| "step": 1375 | |
| }, | |
| { | |
| "epoch": 1.49, | |
| "grad_norm": 22.736656188964844, | |
| "learning_rate": 2.512437810945274e-06, | |
| "loss": 0.2044, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 1.49, | |
| "eval_accuracy": 0.861, | |
| "eval_f1_macro": 0.7820141563614671, | |
| "eval_f1_micro": 0.861, | |
| "eval_loss": 0.5385035276412964, | |
| "eval_runtime": 25.6666, | |
| "eval_samples_per_second": 389.612, | |
| "eval_steps_per_second": 6.117, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 1.52, | |
| "grad_norm": 24.569435119628906, | |
| "learning_rate": 2.4680170575692966e-06, | |
| "loss": 0.2388, | |
| "step": 1425 | |
| }, | |
| { | |
| "epoch": 1.55, | |
| "grad_norm": 17.50179100036621, | |
| "learning_rate": 2.4235963041933193e-06, | |
| "loss": 0.2556, | |
| "step": 1450 | |
| }, | |
| { | |
| "epoch": 1.57, | |
| "grad_norm": 15.387917518615723, | |
| "learning_rate": 2.379175550817342e-06, | |
| "loss": 0.2343, | |
| "step": 1475 | |
| }, | |
| { | |
| "epoch": 1.6, | |
| "grad_norm": 29.757495880126953, | |
| "learning_rate": 2.3347547974413646e-06, | |
| "loss": 0.2508, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.6, | |
| "eval_accuracy": 0.8577, | |
| "eval_f1_macro": 0.7710712973870113, | |
| "eval_f1_micro": 0.8577, | |
| "eval_loss": 0.5657808780670166, | |
| "eval_runtime": 25.9754, | |
| "eval_samples_per_second": 384.98, | |
| "eval_steps_per_second": 6.044, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 1.63, | |
| "grad_norm": 24.104217529296875, | |
| "learning_rate": 2.2903340440653877e-06, | |
| "loss": 0.2647, | |
| "step": 1525 | |
| }, | |
| { | |
| "epoch": 1.65, | |
| "grad_norm": 29.48048973083496, | |
| "learning_rate": 2.24591329068941e-06, | |
| "loss": 0.212, | |
| "step": 1550 | |
| }, | |
| { | |
| "epoch": 1.68, | |
| "grad_norm": 11.834880828857422, | |
| "learning_rate": 2.201492537313433e-06, | |
| "loss": 0.1939, | |
| "step": 1575 | |
| }, | |
| { | |
| "epoch": 1.71, | |
| "grad_norm": 24.24506378173828, | |
| "learning_rate": 2.1570717839374557e-06, | |
| "loss": 0.2513, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 1.71, | |
| "eval_accuracy": 0.8589, | |
| "eval_f1_macro": 0.7871987440671023, | |
| "eval_f1_micro": 0.8589, | |
| "eval_loss": 0.5366827845573425, | |
| "eval_runtime": 25.9643, | |
| "eval_samples_per_second": 385.144, | |
| "eval_steps_per_second": 6.047, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 1.73, | |
| "grad_norm": 23.33180046081543, | |
| "learning_rate": 2.112651030561479e-06, | |
| "loss": 0.2409, | |
| "step": 1625 | |
| }, | |
| { | |
| "epoch": 1.76, | |
| "grad_norm": 18.71114730834961, | |
| "learning_rate": 2.068230277185501e-06, | |
| "loss": 0.224, | |
| "step": 1650 | |
| }, | |
| { | |
| "epoch": 1.79, | |
| "grad_norm": 21.95819854736328, | |
| "learning_rate": 2.023809523809524e-06, | |
| "loss": 0.2223, | |
| "step": 1675 | |
| }, | |
| { | |
| "epoch": 1.81, | |
| "grad_norm": 27.065677642822266, | |
| "learning_rate": 1.979388770433547e-06, | |
| "loss": 0.2787, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 1.81, | |
| "eval_accuracy": 0.8653, | |
| "eval_f1_macro": 0.790261134849528, | |
| "eval_f1_micro": 0.8653, | |
| "eval_loss": 0.5133171677589417, | |
| "eval_runtime": 25.5701, | |
| "eval_samples_per_second": 391.081, | |
| "eval_steps_per_second": 6.14, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 1.84, | |
| "grad_norm": 35.288761138916016, | |
| "learning_rate": 1.9349680170575695e-06, | |
| "loss": 0.2709, | |
| "step": 1725 | |
| }, | |
| { | |
| "epoch": 1.87, | |
| "grad_norm": 21.077306747436523, | |
| "learning_rate": 1.8905472636815921e-06, | |
| "loss": 0.2002, | |
| "step": 1750 | |
| }, | |
| { | |
| "epoch": 1.89, | |
| "grad_norm": 25.394838333129883, | |
| "learning_rate": 1.846126510305615e-06, | |
| "loss": 0.2461, | |
| "step": 1775 | |
| }, | |
| { | |
| "epoch": 1.92, | |
| "grad_norm": 26.597759246826172, | |
| "learning_rate": 1.8017057569296375e-06, | |
| "loss": 0.2357, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 1.92, | |
| "eval_accuracy": 0.8669, | |
| "eval_f1_macro": 0.7902403947168268, | |
| "eval_f1_micro": 0.8669, | |
| "eval_loss": 0.5063937306404114, | |
| "eval_runtime": 25.6031, | |
| "eval_samples_per_second": 390.577, | |
| "eval_steps_per_second": 6.132, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 1.95, | |
| "grad_norm": 18.62090301513672, | |
| "learning_rate": 1.7572850035536603e-06, | |
| "loss": 0.2612, | |
| "step": 1825 | |
| }, | |
| { | |
| "epoch": 1.97, | |
| "grad_norm": 25.897939682006836, | |
| "learning_rate": 1.7128642501776832e-06, | |
| "loss": 0.2243, | |
| "step": 1850 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "grad_norm": 20.556882858276367, | |
| "learning_rate": 1.668443496801706e-06, | |
| "loss": 0.2078, | |
| "step": 1875 | |
| }, | |
| { | |
| "epoch": 2.03, | |
| "grad_norm": 5.211686134338379, | |
| "learning_rate": 1.6240227434257286e-06, | |
| "loss": 0.049, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 2.03, | |
| "eval_accuracy": 0.8719, | |
| "eval_f1_macro": 0.797777536741942, | |
| "eval_f1_micro": 0.8719, | |
| "eval_loss": 0.5344421863555908, | |
| "eval_runtime": 25.9399, | |
| "eval_samples_per_second": 385.506, | |
| "eval_steps_per_second": 6.052, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 2.05, | |
| "grad_norm": 14.868837356567383, | |
| "learning_rate": 1.5796019900497514e-06, | |
| "loss": 0.0483, | |
| "step": 1925 | |
| }, | |
| { | |
| "epoch": 2.08, | |
| "grad_norm": 3.739365577697754, | |
| "learning_rate": 1.5351812366737743e-06, | |
| "loss": 0.0426, | |
| "step": 1950 | |
| }, | |
| { | |
| "epoch": 2.11, | |
| "grad_norm": 3.052903413772583, | |
| "learning_rate": 1.4907604832977968e-06, | |
| "loss": 0.0468, | |
| "step": 1975 | |
| }, | |
| { | |
| "epoch": 2.13, | |
| "grad_norm": 11.233345985412598, | |
| "learning_rate": 1.4463397299218196e-06, | |
| "loss": 0.0298, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 2.13, | |
| "eval_accuracy": 0.8737, | |
| "eval_f1_macro": 0.7992354841882687, | |
| "eval_f1_micro": 0.8737, | |
| "eval_loss": 0.5761749744415283, | |
| "eval_runtime": 25.6811, | |
| "eval_samples_per_second": 389.392, | |
| "eval_steps_per_second": 6.113, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 2.16, | |
| "grad_norm": 9.836750030517578, | |
| "learning_rate": 1.4019189765458423e-06, | |
| "loss": 0.0306, | |
| "step": 2025 | |
| }, | |
| { | |
| "epoch": 2.19, | |
| "grad_norm": 12.054607391357422, | |
| "learning_rate": 1.357498223169865e-06, | |
| "loss": 0.0408, | |
| "step": 2050 | |
| }, | |
| { | |
| "epoch": 2.21, | |
| "grad_norm": 2.877735137939453, | |
| "learning_rate": 1.3130774697938879e-06, | |
| "loss": 0.032, | |
| "step": 2075 | |
| }, | |
| { | |
| "epoch": 2.24, | |
| "grad_norm": 18.573556900024414, | |
| "learning_rate": 1.2686567164179105e-06, | |
| "loss": 0.0427, | |
| "step": 2100 | |
| }, | |
| { | |
| "epoch": 2.24, | |
| "eval_accuracy": 0.8708, | |
| "eval_f1_macro": 0.7976411680340069, | |
| "eval_f1_micro": 0.8708, | |
| "eval_loss": 0.5961406230926514, | |
| "eval_runtime": 25.6941, | |
| "eval_samples_per_second": 389.194, | |
| "eval_steps_per_second": 6.11, | |
| "step": 2100 | |
| }, | |
| { | |
| "epoch": 2.27, | |
| "grad_norm": 11.545409202575684, | |
| "learning_rate": 1.2242359630419332e-06, | |
| "loss": 0.0343, | |
| "step": 2125 | |
| }, | |
| { | |
| "epoch": 2.29, | |
| "grad_norm": 3.3840279579162598, | |
| "learning_rate": 1.179815209665956e-06, | |
| "loss": 0.0237, | |
| "step": 2150 | |
| }, | |
| { | |
| "epoch": 2.32, | |
| "grad_norm": 7.452319145202637, | |
| "learning_rate": 1.1353944562899787e-06, | |
| "loss": 0.042, | |
| "step": 2175 | |
| }, | |
| { | |
| "epoch": 2.35, | |
| "grad_norm": 7.546860694885254, | |
| "learning_rate": 1.0909737029140014e-06, | |
| "loss": 0.036, | |
| "step": 2200 | |
| }, | |
| { | |
| "epoch": 2.35, | |
| "eval_accuracy": 0.8728, | |
| "eval_f1_macro": 0.7987820731312831, | |
| "eval_f1_micro": 0.8728, | |
| "eval_loss": 0.6128308773040771, | |
| "eval_runtime": 25.9603, | |
| "eval_samples_per_second": 385.204, | |
| "eval_steps_per_second": 6.048, | |
| "step": 2200 | |
| }, | |
| { | |
| "epoch": 2.37, | |
| "grad_norm": 0.8773216605186462, | |
| "learning_rate": 1.0465529495380243e-06, | |
| "loss": 0.0264, | |
| "step": 2225 | |
| }, | |
| { | |
| "epoch": 2.4, | |
| "grad_norm": 1.4390593767166138, | |
| "learning_rate": 1.002132196162047e-06, | |
| "loss": 0.0326, | |
| "step": 2250 | |
| }, | |
| { | |
| "epoch": 2.43, | |
| "grad_norm": 3.424440622329712, | |
| "learning_rate": 9.577114427860696e-07, | |
| "loss": 0.0265, | |
| "step": 2275 | |
| }, | |
| { | |
| "epoch": 2.45, | |
| "grad_norm": 13.154258728027344, | |
| "learning_rate": 9.132906894100925e-07, | |
| "loss": 0.0551, | |
| "step": 2300 | |
| }, | |
| { | |
| "epoch": 2.45, | |
| "eval_accuracy": 0.8708, | |
| "eval_f1_macro": 0.7975921884184456, | |
| "eval_f1_micro": 0.8708, | |
| "eval_loss": 0.6165248155593872, | |
| "eval_runtime": 25.655, | |
| "eval_samples_per_second": 389.788, | |
| "eval_steps_per_second": 6.12, | |
| "step": 2300 | |
| }, | |
| { | |
| "epoch": 2.48, | |
| "grad_norm": 8.90126895904541, | |
| "learning_rate": 8.688699360341152e-07, | |
| "loss": 0.0359, | |
| "step": 2325 | |
| }, | |
| { | |
| "epoch": 2.51, | |
| "grad_norm": 10.456062316894531, | |
| "learning_rate": 8.24449182658138e-07, | |
| "loss": 0.0454, | |
| "step": 2350 | |
| }, | |
| { | |
| "epoch": 2.53, | |
| "grad_norm": 13.38987922668457, | |
| "learning_rate": 7.800284292821607e-07, | |
| "loss": 0.0329, | |
| "step": 2375 | |
| }, | |
| { | |
| "epoch": 2.56, | |
| "grad_norm": 6.421198844909668, | |
| "learning_rate": 7.356076759061834e-07, | |
| "loss": 0.0392, | |
| "step": 2400 | |
| }, | |
| { | |
| "epoch": 2.56, | |
| "eval_accuracy": 0.8749, | |
| "eval_f1_macro": 0.8038155628364919, | |
| "eval_f1_micro": 0.8749, | |
| "eval_loss": 0.6023103594779968, | |
| "eval_runtime": 25.6716, | |
| "eval_samples_per_second": 389.535, | |
| "eval_steps_per_second": 6.116, | |
| "step": 2400 | |
| }, | |
| { | |
| "epoch": 2.59, | |
| "grad_norm": 16.32231330871582, | |
| "learning_rate": 6.911869225302062e-07, | |
| "loss": 0.0319, | |
| "step": 2425 | |
| }, | |
| { | |
| "epoch": 2.61, | |
| "grad_norm": 5.884388446807861, | |
| "learning_rate": 6.467661691542289e-07, | |
| "loss": 0.041, | |
| "step": 2450 | |
| }, | |
| { | |
| "epoch": 2.64, | |
| "grad_norm": 17.850648880004883, | |
| "learning_rate": 6.023454157782517e-07, | |
| "loss": 0.036, | |
| "step": 2475 | |
| }, | |
| { | |
| "epoch": 2.67, | |
| "grad_norm": 16.628997802734375, | |
| "learning_rate": 5.579246624022743e-07, | |
| "loss": 0.0364, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 2.67, | |
| "eval_accuracy": 0.8729, | |
| "eval_f1_macro": 0.8001251167012569, | |
| "eval_f1_micro": 0.8729, | |
| "eval_loss": 0.6167578101158142, | |
| "eval_runtime": 25.9524, | |
| "eval_samples_per_second": 385.321, | |
| "eval_steps_per_second": 6.05, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 2.69, | |
| "grad_norm": 4.081849575042725, | |
| "learning_rate": 5.135039090262971e-07, | |
| "loss": 0.0418, | |
| "step": 2525 | |
| }, | |
| { | |
| "epoch": 2.72, | |
| "grad_norm": 8.027618408203125, | |
| "learning_rate": 4.690831556503199e-07, | |
| "loss": 0.0324, | |
| "step": 2550 | |
| }, | |
| { | |
| "epoch": 2.75, | |
| "grad_norm": 9.084144592285156, | |
| "learning_rate": 4.2466240227434256e-07, | |
| "loss": 0.0286, | |
| "step": 2575 | |
| }, | |
| { | |
| "epoch": 2.77, | |
| "grad_norm": 5.414234161376953, | |
| "learning_rate": 3.8024164889836533e-07, | |
| "loss": 0.0416, | |
| "step": 2600 | |
| }, | |
| { | |
| "epoch": 2.77, | |
| "eval_accuracy": 0.8753, | |
| "eval_f1_macro": 0.8048163846871306, | |
| "eval_f1_micro": 0.8753, | |
| "eval_loss": 0.6102917790412903, | |
| "eval_runtime": 25.9892, | |
| "eval_samples_per_second": 384.775, | |
| "eval_steps_per_second": 6.041, | |
| "step": 2600 | |
| }, | |
| { | |
| "epoch": 2.8, | |
| "grad_norm": 7.557031154632568, | |
| "learning_rate": 3.358208955223881e-07, | |
| "loss": 0.0271, | |
| "step": 2625 | |
| }, | |
| { | |
| "epoch": 2.83, | |
| "grad_norm": 5.741875648498535, | |
| "learning_rate": 2.914001421464108e-07, | |
| "loss": 0.0367, | |
| "step": 2650 | |
| }, | |
| { | |
| "epoch": 2.85, | |
| "grad_norm": 12.493782997131348, | |
| "learning_rate": 2.4697938877043354e-07, | |
| "loss": 0.0274, | |
| "step": 2675 | |
| }, | |
| { | |
| "epoch": 2.88, | |
| "grad_norm": 2.7892916202545166, | |
| "learning_rate": 2.0255863539445632e-07, | |
| "loss": 0.0353, | |
| "step": 2700 | |
| }, | |
| { | |
| "epoch": 2.88, | |
| "eval_accuracy": 0.8749, | |
| "eval_f1_macro": 0.8053988442835892, | |
| "eval_f1_micro": 0.8749, | |
| "eval_loss": 0.6117515563964844, | |
| "eval_runtime": 25.6582, | |
| "eval_samples_per_second": 389.74, | |
| "eval_steps_per_second": 6.119, | |
| "step": 2700 | |
| }, | |
| { | |
| "epoch": 2.91, | |
| "grad_norm": 5.350805759429932, | |
| "learning_rate": 1.5813788201847903e-07, | |
| "loss": 0.0261, | |
| "step": 2725 | |
| }, | |
| { | |
| "epoch": 2.93, | |
| "grad_norm": 11.953265190124512, | |
| "learning_rate": 1.1371712864250178e-07, | |
| "loss": 0.0377, | |
| "step": 2750 | |
| }, | |
| { | |
| "epoch": 2.96, | |
| "grad_norm": 13.371731758117676, | |
| "learning_rate": 6.929637526652453e-08, | |
| "loss": 0.0255, | |
| "step": 2775 | |
| }, | |
| { | |
| "epoch": 2.99, | |
| "grad_norm": 15.417765617370605, | |
| "learning_rate": 2.4875621890547265e-08, | |
| "loss": 0.0308, | |
| "step": 2800 | |
| }, | |
| { | |
| "epoch": 2.99, | |
| "eval_accuracy": 0.875, | |
| "eval_f1_macro": 0.805663420581269, | |
| "eval_f1_micro": 0.875, | |
| "eval_loss": 0.611430287361145, | |
| "eval_runtime": 25.9862, | |
| "eval_samples_per_second": 384.819, | |
| "eval_steps_per_second": 6.042, | |
| "step": 2800 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "step": 2814, | |
| "total_flos": 1.6801018930213683e+17, | |
| "train_loss": 0.42635450247932005, | |
| "train_runtime": 3623.2348, | |
| "train_samples_per_second": 49.679, | |
| "train_steps_per_second": 0.777 | |
| } | |
| ], | |
| "logging_steps": 25, | |
| "max_steps": 2814, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 100, | |
| "total_flos": 1.6801018930213683e+17, | |
| "train_batch_size": 32, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |