{ "best_metric": 0.26044607162475586, "best_model_checkpoint": "./checkpoint-distilbert/checkpoint-40000", "epoch": 1.7708517797060386, "eval_steps": 1000, "global_step": 40000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.04, "grad_norm": 11.598106384277344, "learning_rate": 1.9873510587163855e-05, "loss": 0.7232, "step": 1000 }, { "epoch": 0.04, "eval_LOC_f1": 0.6290546246963268, "eval_ORG_f1": 0.5215345154455224, "eval_PER_f1": 0.6879167213048609, "eval_loss": 0.5066247582435608, "eval_overall_accuracy": 0.8370397334429099, "eval_overall_f1": 0.6156872111156381, "eval_overall_precision": 0.6129916634028982, "eval_overall_recall": 0.6184065701464736, "eval_runtime": 273.4377, "eval_samples_per_second": 240.274, "eval_steps_per_second": 1.88, "step": 1000 }, { "epoch": 0.09, "grad_norm": 10.799552917480469, "learning_rate": 1.974702117432771e-05, "loss": 0.5027, "step": 2000 }, { "epoch": 0.09, "eval_LOC_f1": 0.7183374430474798, "eval_ORG_f1": 0.5772430933256492, "eval_PER_f1": 0.7319963163818684, "eval_loss": 0.4309352934360504, "eval_overall_accuracy": 0.8618575027245279, "eval_overall_f1": 0.6810770980734424, "eval_overall_precision": 0.6907076628408085, "eval_overall_recall": 0.6717113989783535, "eval_runtime": 510.4367, "eval_samples_per_second": 128.713, "eval_steps_per_second": 1.007, "step": 2000 }, { "epoch": 0.13, "grad_norm": 6.027167797088623, "learning_rate": 1.9620531761491565e-05, "loss": 0.4459, "step": 3000 }, { "epoch": 0.13, "eval_LOC_f1": 0.725799810401126, "eval_ORG_f1": 0.6044215496887744, "eval_PER_f1": 0.7564098621149212, "eval_loss": 0.40737923979759216, "eval_overall_accuracy": 0.867908783401521, "eval_overall_f1": 0.7006346252072851, "eval_overall_precision": 0.6948635032545487, "eval_overall_recall": 0.706502413004826, "eval_runtime": 385.809, "eval_samples_per_second": 170.292, "eval_steps_per_second": 1.332, "step": 3000 }, { "epoch": 0.18, "grad_norm": 3.919013023376465, "learning_rate": 1.949404234865542e-05, "loss": 0.4238, "step": 4000 }, { "epoch": 0.18, "eval_LOC_f1": 0.7514409560614181, "eval_ORG_f1": 0.6140201882946715, "eval_PER_f1": 0.7751705876362154, "eval_loss": 0.3706400394439697, "eval_overall_accuracy": 0.8801371494589654, "eval_overall_f1": 0.7200897664191354, "eval_overall_precision": 0.711405652907357, "eval_overall_recall": 0.7289885135325826, "eval_runtime": 364.1549, "eval_samples_per_second": 180.418, "eval_steps_per_second": 1.411, "step": 4000 }, { "epoch": 0.22, "grad_norm": 1.984628677368164, "learning_rate": 1.9367552935819272e-05, "loss": 0.3889, "step": 5000 }, { "epoch": 0.22, "eval_LOC_f1": 0.747598650693943, "eval_ORG_f1": 0.6493777882131956, "eval_PER_f1": 0.7714610460826736, "eval_loss": 0.3741539716720581, "eval_overall_accuracy": 0.8788552814154443, "eval_overall_f1": 0.7267549997033853, "eval_overall_precision": 0.7189698622570512, "eval_overall_recall": 0.7347105805322722, "eval_runtime": 381.1257, "eval_samples_per_second": 172.384, "eval_steps_per_second": 1.349, "step": 5000 }, { "epoch": 0.27, "grad_norm": 9.616731643676758, "learning_rate": 1.924106352298313e-05, "loss": 0.3741, "step": 6000 }, { "epoch": 0.27, "eval_LOC_f1": 0.7834622535184133, "eval_ORG_f1": 0.6557229099335325, "eval_PER_f1": 0.7861336370240923, "eval_loss": 0.34435510635375977, "eval_overall_accuracy": 0.8897586038506661, "eval_overall_f1": 0.7479976639412649, "eval_overall_precision": 0.7372337801167668, "eval_overall_recall": 0.7590805181610363, "eval_runtime": 322.2713, "eval_samples_per_second": 203.866, "eval_steps_per_second": 1.595, "step": 6000 }, { "epoch": 0.31, "grad_norm": 23.724206924438477, "learning_rate": 1.9114574110146982e-05, "loss": 0.3565, "step": 7000 }, { "epoch": 0.31, "eval_LOC_f1": 0.7660811273670305, "eval_ORG_f1": 0.6701080291970803, "eval_PER_f1": 0.8082118790158281, "eval_loss": 0.3384684920310974, "eval_overall_accuracy": 0.893588575444113, "eval_overall_f1": 0.7513199894189224, "eval_overall_precision": 0.7511583787634086, "eval_overall_recall": 0.751481669630006, "eval_runtime": 380.8076, "eval_samples_per_second": 172.528, "eval_steps_per_second": 1.35, "step": 7000 }, { "epoch": 0.35, "grad_norm": 12.36683177947998, "learning_rate": 1.8988084697310836e-05, "loss": 0.3438, "step": 8000 }, { "epoch": 0.35, "eval_LOC_f1": 0.7931912766836788, "eval_ORG_f1": 0.6588008889092534, "eval_PER_f1": 0.801709206691627, "eval_loss": 0.33696234226226807, "eval_overall_accuracy": 0.8917491469101174, "eval_overall_f1": 0.7552752293577981, "eval_overall_precision": 0.7441655025063686, "eval_overall_recall": 0.7667217001100669, "eval_runtime": 319.19, "eval_samples_per_second": 205.834, "eval_steps_per_second": 1.61, "step": 8000 }, { "epoch": 0.4, "grad_norm": 11.394143104553223, "learning_rate": 1.8861595284474693e-05, "loss": 0.3371, "step": 9000 }, { "epoch": 0.4, "eval_LOC_f1": 0.7949055775142732, "eval_ORG_f1": 0.69052384904555, "eval_PER_f1": 0.8017390414812541, "eval_loss": 0.3417092263698578, "eval_overall_accuracy": 0.8959111238156492, "eval_overall_f1": 0.7671861870625006, "eval_overall_precision": 0.7645921077527909, "eval_overall_recall": 0.7697979284847458, "eval_runtime": 381.3721, "eval_samples_per_second": 172.273, "eval_steps_per_second": 1.348, "step": 9000 }, { "epoch": 0.44, "grad_norm": 3.7753918170928955, "learning_rate": 1.8735105871638546e-05, "loss": 0.3352, "step": 10000 }, { "epoch": 0.44, "eval_LOC_f1": 0.797682178676422, "eval_ORG_f1": 0.6847098624696625, "eval_PER_f1": 0.8165334907549915, "eval_loss": 0.30863526463508606, "eval_overall_accuracy": 0.8984957032855126, "eval_overall_f1": 0.7713626684335284, "eval_overall_precision": 0.7747418167842222, "eval_overall_recall": 0.7680128693590721, "eval_runtime": 353.766, "eval_samples_per_second": 185.716, "eval_steps_per_second": 1.453, "step": 10000 }, { "epoch": 0.49, "grad_norm": 4.4588704109191895, "learning_rate": 1.86086164588024e-05, "loss": 0.3294, "step": 11000 }, { "epoch": 0.49, "eval_LOC_f1": 0.8038860433458362, "eval_ORG_f1": 0.6876538381909779, "eval_PER_f1": 0.8135372012007188, "eval_loss": 0.3306277096271515, "eval_overall_accuracy": 0.8961009474806305, "eval_overall_f1": 0.7722079415779226, "eval_overall_precision": 0.7585587302451574, "eval_overall_recall": 0.7863573504924788, "eval_runtime": 386.1174, "eval_samples_per_second": 170.155, "eval_steps_per_second": 1.331, "step": 11000 }, { "epoch": 0.53, "grad_norm": 0.8787571787834167, "learning_rate": 1.8482127045966253e-05, "loss": 0.3164, "step": 12000 }, { "epoch": 0.53, "eval_LOC_f1": 0.7928661730178796, "eval_ORG_f1": 0.6923009590976729, "eval_PER_f1": 0.8189351776068096, "eval_loss": 0.32749465107917786, "eval_overall_accuracy": 0.8991314264615677, "eval_overall_f1": 0.7738191399005043, "eval_overall_precision": 0.7606578382394035, "eval_overall_recall": 0.7874439082211497, "eval_runtime": 369.9388, "eval_samples_per_second": 177.597, "eval_steps_per_second": 1.389, "step": 12000 }, { "epoch": 0.58, "grad_norm": 7.5162811279296875, "learning_rate": 1.8355637633130106e-05, "loss": 0.3248, "step": 13000 }, { "epoch": 0.58, "eval_LOC_f1": 0.8073569920769398, "eval_ORG_f1": 0.6888664253138371, "eval_PER_f1": 0.813194105996661, "eval_loss": 0.31697988510131836, "eval_overall_accuracy": 0.9009842543130915, "eval_overall_f1": 0.7742117470068326, "eval_overall_precision": 0.7643891570738953, "eval_overall_recall": 0.7842900685801372, "eval_runtime": 387.7862, "eval_samples_per_second": 169.423, "eval_steps_per_second": 1.325, "step": 13000 }, { "epoch": 0.62, "grad_norm": 15.765552520751953, "learning_rate": 1.8229148220293963e-05, "loss": 0.3031, "step": 14000 }, { "epoch": 0.62, "eval_LOC_f1": 0.8060124842017926, "eval_ORG_f1": 0.7039925586720092, "eval_PER_f1": 0.8251789914844223, "eval_loss": 0.3116307258605957, "eval_overall_accuracy": 0.9031675986636414, "eval_overall_f1": 0.7825302431386867, "eval_overall_precision": 0.7787194499951092, "eval_overall_recall": 0.7863785172014789, "eval_runtime": 393.4395, "eval_samples_per_second": 166.989, "eval_steps_per_second": 1.306, "step": 14000 }, { "epoch": 0.66, "grad_norm": 7.4173808097839355, "learning_rate": 1.8102658807457817e-05, "loss": 0.3215, "step": 15000 }, { "epoch": 0.66, "eval_LOC_f1": 0.8202011293152209, "eval_ORG_f1": 0.7058145261059131, "eval_PER_f1": 0.831773061514806, "eval_loss": 0.2920757532119751, "eval_overall_accuracy": 0.9057320791572127, "eval_overall_f1": 0.7904764259051951, "eval_overall_precision": 0.791295187324571, "eval_overall_recall": 0.7896593570964919, "eval_runtime": 358.5756, "eval_samples_per_second": 183.225, "eval_steps_per_second": 1.433, "step": 15000 }, { "epoch": 0.71, "grad_norm": 2.693660259246826, "learning_rate": 1.7976169394621674e-05, "loss": 0.3004, "step": 16000 }, { "epoch": 0.71, "eval_LOC_f1": 0.820160121404321, "eval_ORG_f1": 0.7145462460731136, "eval_PER_f1": 0.8352356233778416, "eval_loss": 0.2958711087703705, "eval_overall_accuracy": 0.9059717780596597, "eval_overall_f1": 0.793620583311305, "eval_overall_precision": 0.7928911988619157, "eval_overall_recall": 0.7943513109248441, "eval_runtime": 381.2154, "eval_samples_per_second": 172.343, "eval_steps_per_second": 1.348, "step": 16000 }, { "epoch": 0.75, "grad_norm": 3.5598666667938232, "learning_rate": 1.7849679981785527e-05, "loss": 0.2963, "step": 17000 }, { "epoch": 0.75, "eval_LOC_f1": 0.8227697858804539, "eval_ORG_f1": 0.7160294907475222, "eval_PER_f1": 0.8368475991649269, "eval_loss": 0.2975331246852875, "eval_overall_accuracy": 0.9079280784187614, "eval_overall_f1": 0.7966393323888228, "eval_overall_precision": 0.7978373328992902, "eval_overall_recall": 0.7954449242231818, "eval_runtime": 335.9278, "eval_samples_per_second": 195.578, "eval_steps_per_second": 1.53, "step": 17000 }, { "epoch": 0.8, "grad_norm": 4.444880485534668, "learning_rate": 1.772319056894938e-05, "loss": 0.3037, "step": 18000 }, { "epoch": 0.8, "eval_LOC_f1": 0.8179091806697487, "eval_ORG_f1": 0.7172501033887172, "eval_PER_f1": 0.8379966394973857, "eval_loss": 0.29457131028175354, "eval_overall_accuracy": 0.9080412282112209, "eval_overall_f1": 0.7957346666573557, "eval_overall_precision": 0.7876546623349693, "eval_overall_recall": 0.8039821635198826, "eval_runtime": 382.8523, "eval_samples_per_second": 171.607, "eval_steps_per_second": 1.343, "step": 18000 }, { "epoch": 0.84, "grad_norm": 3.651078701019287, "learning_rate": 1.7596701156113234e-05, "loss": 0.2893, "step": 19000 }, { "epoch": 0.84, "eval_LOC_f1": 0.8096252541880508, "eval_ORG_f1": 0.722521212121212, "eval_PER_f1": 0.8371796558038832, "eval_loss": 0.30927255749702454, "eval_overall_accuracy": 0.90601644245142, "eval_overall_f1": 0.7939453776936077, "eval_overall_precision": 0.7907640837923529, "eval_overall_recall": 0.797152372082522, "eval_runtime": 321.3057, "eval_samples_per_second": 204.478, "eval_steps_per_second": 1.6, "step": 19000 }, { "epoch": 0.89, "grad_norm": 7.439753532409668, "learning_rate": 1.747021174327709e-05, "loss": 0.2948, "step": 20000 }, { "epoch": 0.89, "eval_LOC_f1": 0.82627274528115, "eval_ORG_f1": 0.714928600160901, "eval_PER_f1": 0.8321251892983341, "eval_loss": 0.2865753769874573, "eval_overall_accuracy": 0.9090692536282374, "eval_overall_f1": 0.7971877602217748, "eval_overall_precision": 0.7939824049383752, "eval_overall_recall": 0.8004191008382017, "eval_runtime": 380.5435, "eval_samples_per_second": 172.648, "eval_steps_per_second": 1.351, "step": 20000 }, { "epoch": 0.93, "grad_norm": 23.34659767150879, "learning_rate": 1.7343722330440944e-05, "loss": 0.2868, "step": 21000 }, { "epoch": 0.93, "eval_LOC_f1": 0.8167083361060147, "eval_ORG_f1": 0.727712241345968, "eval_PER_f1": 0.840251831974404, "eval_loss": 0.29932352900505066, "eval_overall_accuracy": 0.9084059874105967, "eval_overall_f1": 0.798865244242835, "eval_overall_precision": 0.7950827119165264, "eval_overall_recall": 0.8026839387012107, "eval_runtime": 351.3256, "eval_samples_per_second": 187.006, "eval_steps_per_second": 1.463, "step": 21000 }, { "epoch": 0.97, "grad_norm": 10.305853843688965, "learning_rate": 1.7217232917604798e-05, "loss": 0.2977, "step": 22000 }, { "epoch": 0.97, "eval_LOC_f1": 0.8263365865270175, "eval_ORG_f1": 0.7257252830008247, "eval_PER_f1": 0.8407604251839738, "eval_loss": 0.2778206169605255, "eval_overall_accuracy": 0.9105640219391492, "eval_overall_f1": 0.8018860680855222, "eval_overall_precision": 0.7934180537329926, "eval_overall_recall": 0.8105367877402422, "eval_runtime": 394.5686, "eval_samples_per_second": 166.511, "eval_steps_per_second": 1.303, "step": 22000 }, { "epoch": 1.02, "grad_norm": 11.687705993652344, "learning_rate": 1.709074350476865e-05, "loss": 0.2585, "step": 23000 }, { "epoch": 1.02, "eval_LOC_f1": 0.8267493620904808, "eval_ORG_f1": 0.7272141627126203, "eval_PER_f1": 0.8396679597210218, "eval_loss": 0.2966572642326355, "eval_overall_accuracy": 0.9093171410025072, "eval_overall_f1": 0.8032836474772869, "eval_overall_precision": 0.7918766066838047, "eval_overall_recall": 0.8150241300482601, "eval_runtime": 380.5595, "eval_samples_per_second": 172.641, "eval_steps_per_second": 1.351, "step": 23000 }, { "epoch": 1.06, "grad_norm": 3.907621383666992, "learning_rate": 1.6964254091932504e-05, "loss": 0.2362, "step": 24000 }, { "epoch": 1.06, "eval_LOC_f1": 0.8330133267755635, "eval_ORG_f1": 0.7180792210666076, "eval_PER_f1": 0.830729295671386, "eval_loss": 0.3153696358203888, "eval_overall_accuracy": 0.9078134398132433, "eval_overall_f1": 0.7997513906153431, "eval_overall_precision": 0.7873453202980789, "eval_overall_recall": 0.8125546806649169, "eval_runtime": 373.8915, "eval_samples_per_second": 175.719, "eval_steps_per_second": 1.375, "step": 24000 }, { "epoch": 1.11, "grad_norm": 8.853696823120117, "learning_rate": 1.683776467909636e-05, "loss": 0.2223, "step": 25000 }, { "epoch": 1.11, "eval_LOC_f1": 0.827347355217957, "eval_ORG_f1": 0.7285489531647101, "eval_PER_f1": 0.8452979544031824, "eval_loss": 0.29239869117736816, "eval_overall_accuracy": 0.9101635312263651, "eval_overall_f1": 0.8052130812290949, "eval_overall_precision": 0.7913853365400993, "eval_overall_recall": 0.8195326390652782, "eval_runtime": 381.4229, "eval_samples_per_second": 172.25, "eval_steps_per_second": 1.348, "step": 25000 }, { "epoch": 1.15, "grad_norm": 0.670585572719574, "learning_rate": 1.6711275266260215e-05, "loss": 0.2339, "step": 26000 }, { "epoch": 1.15, "eval_LOC_f1": 0.835739538961472, "eval_ORG_f1": 0.7377061141106085, "eval_PER_f1": 0.8477990752245738, "eval_loss": 0.2966776490211487, "eval_overall_accuracy": 0.9133503355784635, "eval_overall_f1": 0.8116372709464219, "eval_overall_precision": 0.8062829412297131, "eval_overall_recall": 0.8170631896819349, "eval_runtime": 348.6318, "eval_samples_per_second": 188.451, "eval_steps_per_second": 1.474, "step": 26000 }, { "epoch": 1.2, "grad_norm": 5.636080741882324, "learning_rate": 1.658478585342407e-05, "loss": 0.2318, "step": 27000 }, { "epoch": 1.2, "eval_LOC_f1": 0.8379575082665046, "eval_ORG_f1": 0.741773869167331, "eval_PER_f1": 0.8468086407796683, "eval_loss": 0.297035276889801, "eval_overall_accuracy": 0.9143232749123089, "eval_overall_f1": 0.8128532056457225, "eval_overall_precision": 0.8019583070120025, "eval_overall_recall": 0.8240482036519629, "eval_runtime": 381.0554, "eval_samples_per_second": 172.416, "eval_steps_per_second": 1.349, "step": 27000 }, { "epoch": 1.24, "grad_norm": 11.75963306427002, "learning_rate": 1.6458296440587925e-05, "loss": 0.2261, "step": 28000 }, { "epoch": 1.24, "eval_LOC_f1": 0.8355374639087424, "eval_ORG_f1": 0.7354031739753795, "eval_PER_f1": 0.8533292168364721, "eval_loss": 0.28783518075942993, "eval_overall_accuracy": 0.914816072034731, "eval_overall_f1": 0.813075185134273, "eval_overall_precision": 0.8123622174798036, "eval_overall_recall": 0.8137894053565885, "eval_runtime": 319.3665, "eval_samples_per_second": 205.72, "eval_steps_per_second": 1.609, "step": 28000 }, { "epoch": 1.28, "grad_norm": 0.3197329342365265, "learning_rate": 1.633180702775178e-05, "loss": 0.2209, "step": 29000 }, { "epoch": 1.28, "eval_LOC_f1": 0.8348409410512713, "eval_ORG_f1": 0.7400960885063363, "eval_PER_f1": 0.852092035253628, "eval_loss": 0.28716719150543213, "eval_overall_accuracy": 0.9143351854167783, "eval_overall_f1": 0.8130391159543484, "eval_overall_precision": 0.8030238383615944, "eval_overall_recall": 0.8233073688369599, "eval_runtime": 381.0776, "eval_samples_per_second": 172.406, "eval_steps_per_second": 1.349, "step": 29000 }, { "epoch": 1.33, "grad_norm": 2.500840663909912, "learning_rate": 1.6205317614915632e-05, "loss": 0.2326, "step": 30000 }, { "epoch": 1.33, "eval_LOC_f1": 0.835323589241317, "eval_ORG_f1": 0.7427079308204821, "eval_PER_f1": 0.8500763485649586, "eval_loss": 0.28030699491500854, "eval_overall_accuracy": 0.914723765625093, "eval_overall_f1": 0.8127800899066888, "eval_overall_precision": 0.8005707559436636, "eval_overall_recall": 0.8253675951796348, "eval_runtime": 325.8915, "eval_samples_per_second": 201.601, "eval_steps_per_second": 1.577, "step": 30000 }, { "epoch": 1.37, "grad_norm": 0.524834156036377, "learning_rate": 1.607882820207949e-05, "loss": 0.2198, "step": 31000 }, { "epoch": 1.37, "eval_LOC_f1": 0.8371461143834315, "eval_ORG_f1": 0.7432512590283674, "eval_PER_f1": 0.8540951115999625, "eval_loss": 0.28697240352630615, "eval_overall_accuracy": 0.9156036541427712, "eval_overall_f1": 0.8147560933208163, "eval_overall_precision": 0.8031779980531677, "eval_overall_recall": 0.8266728755679733, "eval_runtime": 381.127, "eval_samples_per_second": 172.383, "eval_steps_per_second": 1.349, "step": 31000 }, { "epoch": 1.42, "grad_norm": 18.899179458618164, "learning_rate": 1.5952338789243342e-05, "loss": 0.221, "step": 32000 }, { "epoch": 1.42, "eval_LOC_f1": 0.8441860465116279, "eval_ORG_f1": 0.7411915135442319, "eval_PER_f1": 0.8485990299434801, "eval_loss": 0.28265270590782166, "eval_overall_accuracy": 0.9150192950172404, "eval_overall_f1": 0.8153681889236267, "eval_overall_precision": 0.8065649592710203, "eval_overall_recall": 0.8243657042869641, "eval_runtime": 346.0172, "eval_samples_per_second": 189.875, "eval_steps_per_second": 1.485, "step": 32000 }, { "epoch": 1.46, "grad_norm": 9.521773338317871, "learning_rate": 1.5825849376407196e-05, "loss": 0.2319, "step": 33000 }, { "epoch": 1.46, "eval_LOC_f1": 0.8391690707524869, "eval_ORG_f1": 0.7463789877336044, "eval_PER_f1": 0.8572165373824064, "eval_loss": 0.27687233686447144, "eval_overall_accuracy": 0.9169562408065793, "eval_overall_f1": 0.8172862291713824, "eval_overall_precision": 0.8068870201398582, "eval_overall_recall": 0.8279569892473119, "eval_runtime": 385.1918, "eval_samples_per_second": 170.564, "eval_steps_per_second": 1.334, "step": 33000 }, { "epoch": 1.51, "grad_norm": 4.6437883377075195, "learning_rate": 1.569935996357105e-05, "loss": 0.2284, "step": 34000 }, { "epoch": 1.51, "eval_LOC_f1": 0.8413632604389053, "eval_ORG_f1": 0.7468433377658009, "eval_PER_f1": 0.8579245204510465, "eval_loss": 0.2682867646217346, "eval_overall_accuracy": 0.918707829370113, "eval_overall_f1": 0.8187739611430496, "eval_overall_precision": 0.8141108359780979, "eval_overall_recall": 0.823490813648294, "eval_runtime": 389.0432, "eval_samples_per_second": 168.876, "eval_steps_per_second": 1.321, "step": 34000 }, { "epoch": 1.55, "grad_norm": 8.055586814880371, "learning_rate": 1.5572870550734906e-05, "loss": 0.222, "step": 35000 }, { "epoch": 1.55, "eval_LOC_f1": 0.8416752574836452, "eval_ORG_f1": 0.7401676189600267, "eval_PER_f1": 0.8485277320098252, "eval_loss": 0.2882651388645172, "eval_overall_accuracy": 0.9156967049589385, "eval_overall_f1": 0.8135379092525155, "eval_overall_precision": 0.7963817723503612, "eval_overall_recall": 0.8314494962323258, "eval_runtime": 364.9994, "eval_samples_per_second": 180.0, "eval_steps_per_second": 1.408, "step": 35000 }, { "epoch": 1.59, "grad_norm": 2.8561666011810303, "learning_rate": 1.544638113789876e-05, "loss": 0.2272, "step": 36000 }, { "epoch": 1.59, "eval_LOC_f1": 0.843695144769584, "eval_ORG_f1": 0.7492301387506587, "eval_PER_f1": 0.8506844052649669, "eval_loss": 0.2805135250091553, "eval_overall_accuracy": 0.9165698938178527, "eval_overall_f1": 0.8180830193920244, "eval_overall_precision": 0.805384452359629, "eval_overall_recall": 0.8311884401546581, "eval_runtime": 381.2063, "eval_samples_per_second": 172.348, "eval_steps_per_second": 1.348, "step": 36000 }, { "epoch": 1.64, "grad_norm": 3.3037211894989014, "learning_rate": 1.5319891725062616e-05, "loss": 0.2086, "step": 37000 }, { "epoch": 1.64, "eval_LOC_f1": 0.8383752402995496, "eval_ORG_f1": 0.7563655598566996, "eval_PER_f1": 0.8609615664428694, "eval_loss": 0.2780158519744873, "eval_overall_accuracy": 0.9186519988804126, "eval_overall_f1": 0.822004740154319, "eval_overall_precision": 0.8205800667955704, "eval_overall_recall": 0.8234343690909603, "eval_runtime": 332.7363, "eval_samples_per_second": 197.454, "eval_steps_per_second": 1.545, "step": 37000 }, { "epoch": 1.68, "grad_norm": 3.0146262645721436, "learning_rate": 1.5193402312226468e-05, "loss": 0.2283, "step": 38000 }, { "epoch": 1.68, "eval_LOC_f1": 0.8468252837536212, "eval_ORG_f1": 0.7509236309217602, "eval_PER_f1": 0.8619782260977246, "eval_loss": 0.27215901017189026, "eval_overall_accuracy": 0.9187383500378159, "eval_overall_f1": 0.8234198288625827, "eval_overall_precision": 0.8114117405301733, "eval_overall_recall": 0.8357886715773432, "eval_runtime": 305.4831, "eval_samples_per_second": 215.069, "eval_steps_per_second": 1.683, "step": 38000 }, { "epoch": 1.73, "grad_norm": 7.599725246429443, "learning_rate": 1.5066912899390323e-05, "loss": 0.2322, "step": 39000 }, { "epoch": 1.73, "eval_LOC_f1": 0.8500763831144973, "eval_ORG_f1": 0.7572957665354708, "eval_PER_f1": 0.8590397422347182, "eval_loss": 0.27235275506973267, "eval_overall_accuracy": 0.9206611521030973, "eval_overall_f1": 0.8259451471597206, "eval_overall_precision": 0.8237684495694223, "eval_overall_recall": 0.8281333784889792, "eval_runtime": 580.4463, "eval_samples_per_second": 113.189, "eval_steps_per_second": 0.886, "step": 39000 }, { "epoch": 1.77, "grad_norm": 2.075500249862671, "learning_rate": 1.4940423486554176e-05, "loss": 0.2213, "step": 40000 }, { "epoch": 1.77, "eval_LOC_f1": 0.8435102589085544, "eval_ORG_f1": 0.7572054303898555, "eval_PER_f1": 0.8558481099725878, "eval_loss": 0.26044607162475586, "eval_overall_accuracy": 0.9183281820401503, "eval_overall_f1": 0.8229231947635958, "eval_overall_precision": 0.8123006708457055, "eval_overall_recall": 0.833827223210002, "eval_runtime": 503.8704, "eval_samples_per_second": 130.391, "eval_steps_per_second": 1.02, "step": 40000 } ], "logging_steps": 1000, "max_steps": 158116, "num_input_tokens_seen": 0, "num_train_epochs": 7, "save_steps": 500, "total_flos": 5523298719415224.0, "train_batch_size": 8, "trial_name": null, "trial_params": null }