{ "best_global_step": 800, "best_metric": 0.6514270557181998, "best_model_checkpoint": "./taskA-codebert-base/checkpoint-800", "epoch": 0.10239344681940356, "eval_steps": 200, "global_step": 800, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0012799180852425445, "grad_norm": 3.6954128742218018, "learning_rate": 3.840409643695328e-08, "loss": 0.7763, "step": 10 }, { "epoch": 0.002559836170485089, "grad_norm": 3.3660645484924316, "learning_rate": 8.10753147002347e-08, "loss": 0.7763, "step": 20 }, { "epoch": 0.0038397542557276334, "grad_norm": 4.491067886352539, "learning_rate": 1.2374653296351612e-07, "loss": 0.7889, "step": 30 }, { "epoch": 0.005119672340970178, "grad_norm": 1.5920521020889282, "learning_rate": 1.6641775122679754e-07, "loss": 0.7888, "step": 40 }, { "epoch": 0.006399590426212722, "grad_norm": 4.410898208618164, "learning_rate": 2.0908896949007894e-07, "loss": 0.7692, "step": 50 }, { "epoch": 0.007679508511455267, "grad_norm": 2.5669264793395996, "learning_rate": 2.517601877533604e-07, "loss": 0.7735, "step": 60 }, { "epoch": 0.008959426596697812, "grad_norm": 6.190986156463623, "learning_rate": 2.944314060166418e-07, "loss": 0.7679, "step": 70 }, { "epoch": 0.010239344681940356, "grad_norm": 5.736413478851318, "learning_rate": 3.371026242799232e-07, "loss": 0.744, "step": 80 }, { "epoch": 0.011519262767182901, "grad_norm": 4.278866767883301, "learning_rate": 3.7977384254320464e-07, "loss": 0.7542, "step": 90 }, { "epoch": 0.012799180852425445, "grad_norm": 5.565455436706543, "learning_rate": 4.22445060806486e-07, "loss": 0.752, "step": 100 }, { "epoch": 0.01407909893766799, "grad_norm": 4.031781196594238, "learning_rate": 4.651162790697675e-07, "loss": 0.7614, "step": 110 }, { "epoch": 0.015359017022910534, "grad_norm": 2.484126329421997, "learning_rate": 5.077874973330489e-07, "loss": 0.7664, "step": 120 }, { "epoch": 0.016638935108153077, "grad_norm": 2.7525155544281006, "learning_rate": 5.504587155963304e-07, "loss": 0.7521, "step": 130 }, { "epoch": 0.017918853193395624, "grad_norm": 1.6374385356903076, "learning_rate": 5.931299338596117e-07, "loss": 0.7436, "step": 140 }, { "epoch": 0.019198771278638168, "grad_norm": 2.370774269104004, "learning_rate": 6.358011521228932e-07, "loss": 0.7429, "step": 150 }, { "epoch": 0.02047868936388071, "grad_norm": 2.8948566913604736, "learning_rate": 6.784723703861745e-07, "loss": 0.7333, "step": 160 }, { "epoch": 0.021758607449123255, "grad_norm": 2.7481186389923096, "learning_rate": 7.21143588649456e-07, "loss": 0.7294, "step": 170 }, { "epoch": 0.023038525534365802, "grad_norm": 2.0159404277801514, "learning_rate": 7.638148069127374e-07, "loss": 0.731, "step": 180 }, { "epoch": 0.024318443619608346, "grad_norm": 3.6898131370544434, "learning_rate": 8.064860251760189e-07, "loss": 0.7153, "step": 190 }, { "epoch": 0.02559836170485089, "grad_norm": 3.2842342853546143, "learning_rate": 8.491572434393003e-07, "loss": 0.7245, "step": 200 }, { "epoch": 0.02559836170485089, "eval_accuracy": 0.47695, "eval_loss": 0.7148504853248596, "eval_macro_f1": 0.3229290091066048, "eval_precision": 0.238475, "eval_recall": 0.5, "eval_runtime": 1641.324, "eval_samples_per_second": 60.926, "eval_steps_per_second": 0.476, "step": 200 }, { "epoch": 0.026878279790093433, "grad_norm": 2.294558048248291, "learning_rate": 8.918284617025817e-07, "loss": 0.718, "step": 210 }, { "epoch": 0.02815819787533598, "grad_norm": 2.9467175006866455, "learning_rate": 9.344996799658632e-07, "loss": 0.7076, "step": 220 }, { "epoch": 0.029438115960578524, "grad_norm": 2.6889193058013916, "learning_rate": 9.771708982291445e-07, "loss": 0.7019, "step": 230 }, { "epoch": 0.030718034045821067, "grad_norm": 0.7491384744644165, "learning_rate": 1.0198421164924258e-06, "loss": 0.7046, "step": 240 }, { "epoch": 0.031997952131063614, "grad_norm": 2.0363433361053467, "learning_rate": 1.0625133347557074e-06, "loss": 0.7085, "step": 250 }, { "epoch": 0.033277870216306155, "grad_norm": 1.1431337594985962, "learning_rate": 1.1051845530189888e-06, "loss": 0.7008, "step": 260 }, { "epoch": 0.0345577883015487, "grad_norm": 3.7735233306884766, "learning_rate": 1.1478557712822702e-06, "loss": 0.7015, "step": 270 }, { "epoch": 0.03583770638679125, "grad_norm": 0.7889328002929688, "learning_rate": 1.1905269895455517e-06, "loss": 0.699, "step": 280 }, { "epoch": 0.03711762447203379, "grad_norm": 0.8796226382255554, "learning_rate": 1.233198207808833e-06, "loss": 0.6922, "step": 290 }, { "epoch": 0.038397542557276336, "grad_norm": 0.7966652512550354, "learning_rate": 1.2758694260721145e-06, "loss": 0.6949, "step": 300 }, { "epoch": 0.039677460642518876, "grad_norm": 0.7029069066047668, "learning_rate": 1.318540644335396e-06, "loss": 0.695, "step": 310 }, { "epoch": 0.04095737872776142, "grad_norm": 2.04327392578125, "learning_rate": 1.3612118625986772e-06, "loss": 0.6944, "step": 320 }, { "epoch": 0.04223729681300397, "grad_norm": 0.918025553226471, "learning_rate": 1.4038830808619588e-06, "loss": 0.6863, "step": 330 }, { "epoch": 0.04351721489824651, "grad_norm": 0.9528706073760986, "learning_rate": 1.4465542991252401e-06, "loss": 0.6907, "step": 340 }, { "epoch": 0.04479713298348906, "grad_norm": 1.071359634399414, "learning_rate": 1.4892255173885215e-06, "loss": 0.6873, "step": 350 }, { "epoch": 0.046077051068731605, "grad_norm": 1.9875625371932983, "learning_rate": 1.531896735651803e-06, "loss": 0.6942, "step": 360 }, { "epoch": 0.047356969153974145, "grad_norm": 2.0222439765930176, "learning_rate": 1.5745679539150842e-06, "loss": 0.6898, "step": 370 }, { "epoch": 0.04863688723921669, "grad_norm": 1.7827857732772827, "learning_rate": 1.6172391721783658e-06, "loss": 0.6909, "step": 380 }, { "epoch": 0.04991680532445923, "grad_norm": 1.065674901008606, "learning_rate": 1.6599103904416472e-06, "loss": 0.6877, "step": 390 }, { "epoch": 0.05119672340970178, "grad_norm": 0.8346193432807922, "learning_rate": 1.7025816087049288e-06, "loss": 0.6889, "step": 400 }, { "epoch": 0.05119672340970178, "eval_accuracy": 0.54043, "eval_loss": 0.6827595233917236, "eval_macro_f1": 0.41833840347598905, "eval_precision": 0.5839205994464474, "eval_recall": 0.5204152767503025, "eval_runtime": 1634.6244, "eval_samples_per_second": 61.176, "eval_steps_per_second": 0.478, "step": 400 }, { "epoch": 0.052476641494944326, "grad_norm": 0.8074935674667358, "learning_rate": 1.7452528269682101e-06, "loss": 0.684, "step": 410 }, { "epoch": 0.053756559580186866, "grad_norm": 0.8965069651603699, "learning_rate": 1.7879240452314913e-06, "loss": 0.6922, "step": 420 }, { "epoch": 0.05503647766542941, "grad_norm": 2.597215414047241, "learning_rate": 1.8305952634947729e-06, "loss": 0.6909, "step": 430 }, { "epoch": 0.05631639575067196, "grad_norm": 0.8233373165130615, "learning_rate": 1.8732664817580542e-06, "loss": 0.6831, "step": 440 }, { "epoch": 0.0575963138359145, "grad_norm": 1.0624489784240723, "learning_rate": 1.915937700021336e-06, "loss": 0.6817, "step": 450 }, { "epoch": 0.05887623192115705, "grad_norm": 2.0623559951782227, "learning_rate": 1.958608918284617e-06, "loss": 0.6819, "step": 460 }, { "epoch": 0.06015615000639959, "grad_norm": 0.912433922290802, "learning_rate": 2.0012801365478988e-06, "loss": 0.686, "step": 470 }, { "epoch": 0.061436068091642135, "grad_norm": 0.6926271319389343, "learning_rate": 2.04395135481118e-06, "loss": 0.6834, "step": 480 }, { "epoch": 0.06271598617688467, "grad_norm": 1.498634934425354, "learning_rate": 2.0866225730744615e-06, "loss": 0.6805, "step": 490 }, { "epoch": 0.06399590426212723, "grad_norm": 1.1818156242370605, "learning_rate": 2.129293791337743e-06, "loss": 0.6805, "step": 500 }, { "epoch": 0.06527582234736977, "grad_norm": 1.4476051330566406, "learning_rate": 2.1719650096010242e-06, "loss": 0.6773, "step": 510 }, { "epoch": 0.06655574043261231, "grad_norm": 1.1143743991851807, "learning_rate": 2.214636227864306e-06, "loss": 0.6801, "step": 520 }, { "epoch": 0.06783565851785486, "grad_norm": 1.7030247449874878, "learning_rate": 2.257307446127587e-06, "loss": 0.6794, "step": 530 }, { "epoch": 0.0691155766030974, "grad_norm": 0.8592290282249451, "learning_rate": 2.2999786643908685e-06, "loss": 0.6736, "step": 540 }, { "epoch": 0.07039549468833994, "grad_norm": 0.8231940865516663, "learning_rate": 2.34264988265415e-06, "loss": 0.67, "step": 550 }, { "epoch": 0.0716754127735825, "grad_norm": 0.88897705078125, "learning_rate": 2.3853211009174317e-06, "loss": 0.685, "step": 560 }, { "epoch": 0.07295533085882504, "grad_norm": 1.6297541856765747, "learning_rate": 2.427992319180713e-06, "loss": 0.6797, "step": 570 }, { "epoch": 0.07423524894406758, "grad_norm": 2.9297447204589844, "learning_rate": 2.470663537443994e-06, "loss": 0.6766, "step": 580 }, { "epoch": 0.07551516702931012, "grad_norm": 0.9277703762054443, "learning_rate": 2.5133347557072756e-06, "loss": 0.6824, "step": 590 }, { "epoch": 0.07679508511455267, "grad_norm": 0.5883333683013916, "learning_rate": 2.556005973970557e-06, "loss": 0.6818, "step": 600 }, { "epoch": 0.07679508511455267, "eval_accuracy": 0.63909, "eval_loss": 0.6700075268745422, "eval_macro_f1": 0.621848143053056, "eval_precision": 0.6524313071088819, "eval_recall": 0.6305863955136095, "eval_runtime": 1633.9732, "eval_samples_per_second": 61.201, "eval_steps_per_second": 0.479, "step": 600 }, { "epoch": 0.07807500319979521, "grad_norm": 0.7149448394775391, "learning_rate": 2.5986771922338383e-06, "loss": 0.6746, "step": 610 }, { "epoch": 0.07935492128503775, "grad_norm": 2.0167856216430664, "learning_rate": 2.64134841049712e-06, "loss": 0.682, "step": 620 }, { "epoch": 0.0806348393702803, "grad_norm": 1.303001880645752, "learning_rate": 2.6840196287604015e-06, "loss": 0.6735, "step": 630 }, { "epoch": 0.08191475745552285, "grad_norm": 2.0516321659088135, "learning_rate": 2.7266908470236826e-06, "loss": 0.6705, "step": 640 }, { "epoch": 0.08319467554076539, "grad_norm": 2.242105484008789, "learning_rate": 2.7693620652869642e-06, "loss": 0.6798, "step": 650 }, { "epoch": 0.08447459362600794, "grad_norm": 1.1836538314819336, "learning_rate": 2.812033283550246e-06, "loss": 0.6684, "step": 660 }, { "epoch": 0.08575451171125048, "grad_norm": 0.8640874624252319, "learning_rate": 2.8547045018135274e-06, "loss": 0.6783, "step": 670 }, { "epoch": 0.08703442979649302, "grad_norm": 0.6901177763938904, "learning_rate": 2.897375720076808e-06, "loss": 0.6724, "step": 680 }, { "epoch": 0.08831434788173557, "grad_norm": 1.3443949222564697, "learning_rate": 2.9400469383400897e-06, "loss": 0.67, "step": 690 }, { "epoch": 0.08959426596697811, "grad_norm": 0.7775189280509949, "learning_rate": 2.9827181566033713e-06, "loss": 0.6733, "step": 700 }, { "epoch": 0.09087418405222066, "grad_norm": 1.16916823387146, "learning_rate": 3.025389374866653e-06, "loss": 0.6713, "step": 710 }, { "epoch": 0.09215410213746321, "grad_norm": 1.707152009010315, "learning_rate": 3.068060593129934e-06, "loss": 0.6659, "step": 720 }, { "epoch": 0.09343402022270575, "grad_norm": 0.9780622124671936, "learning_rate": 3.1107318113932156e-06, "loss": 0.6639, "step": 730 }, { "epoch": 0.09471393830794829, "grad_norm": 2.3687374591827393, "learning_rate": 3.153403029656497e-06, "loss": 0.6666, "step": 740 }, { "epoch": 0.09599385639319083, "grad_norm": 0.8506460189819336, "learning_rate": 3.1960742479197783e-06, "loss": 0.6778, "step": 750 }, { "epoch": 0.09727377447843338, "grad_norm": 1.0102107524871826, "learning_rate": 3.23874546618306e-06, "loss": 0.6635, "step": 760 }, { "epoch": 0.09855369256367592, "grad_norm": 1.1480194330215454, "learning_rate": 3.2814166844463415e-06, "loss": 0.6709, "step": 770 }, { "epoch": 0.09983361064891846, "grad_norm": 1.3119179010391235, "learning_rate": 3.324087902709623e-06, "loss": 0.6564, "step": 780 }, { "epoch": 0.10111352873416102, "grad_norm": 0.681377649307251, "learning_rate": 3.3667591209729038e-06, "loss": 0.6656, "step": 790 }, { "epoch": 0.10239344681940356, "grad_norm": 1.022257685661316, "learning_rate": 3.4094303392361854e-06, "loss": 0.6709, "step": 800 }, { "epoch": 0.10239344681940356, "eval_accuracy": 0.66247, "eval_loss": 0.6552017331123352, "eval_macro_f1": 0.6514270557181998, "eval_precision": 0.6718270655389242, "eval_recall": 0.6556580720913894, "eval_runtime": 1633.5478, "eval_samples_per_second": 61.216, "eval_steps_per_second": 0.479, "step": 800 } ], "logging_steps": 10, "max_steps": 23439, "num_input_tokens_seen": 0, "num_train_epochs": 3, "save_steps": 200, "stateful_callbacks": { "EarlyStoppingCallback": { "args": { "early_stopping_patience": 3, "early_stopping_threshold": 0.0 }, "attributes": { "early_stopping_patience_counter": 0 } }, "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 1.3471286034432e+16, "train_batch_size": 64, "trial_name": null, "trial_params": null }