Instructions to use dzungpham/graphcodebert-code-classification with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use dzungpham/graphcodebert-code-classification with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("dzungpham/graphcodebert-code-classification", dtype="auto") - Notebooks
- Google Colab
- Kaggle
| { | |
| "best_global_step": 1200, | |
| "best_metric": 0.6993136399748257, | |
| "best_model_checkpoint": "./taskA-codebert-base/checkpoint-1200", | |
| "epoch": 0.15359017022910534, | |
| "eval_steps": 200, | |
| "global_step": 1200, | |
| "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 | |
| }, | |
| { | |
| "epoch": 0.1036733649046461, | |
| "grad_norm": 1.0268975496292114, | |
| "learning_rate": 3.452101557499467e-06, | |
| "loss": 0.6602, | |
| "step": 810 | |
| }, | |
| { | |
| "epoch": 0.10495328298988865, | |
| "grad_norm": 2.6206603050231934, | |
| "learning_rate": 3.494772775762748e-06, | |
| "loss": 0.6705, | |
| "step": 820 | |
| }, | |
| { | |
| "epoch": 0.10623320107513119, | |
| "grad_norm": 1.035157561302185, | |
| "learning_rate": 3.5374439940260297e-06, | |
| "loss": 0.6643, | |
| "step": 830 | |
| }, | |
| { | |
| "epoch": 0.10751311916037373, | |
| "grad_norm": 1.736412763595581, | |
| "learning_rate": 3.5801152122893113e-06, | |
| "loss": 0.6612, | |
| "step": 840 | |
| }, | |
| { | |
| "epoch": 0.10879303724561629, | |
| "grad_norm": 0.6906964778900146, | |
| "learning_rate": 3.622786430552593e-06, | |
| "loss": 0.6576, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 0.11007295533085883, | |
| "grad_norm": 0.8148729801177979, | |
| "learning_rate": 3.665457648815874e-06, | |
| "loss": 0.6561, | |
| "step": 860 | |
| }, | |
| { | |
| "epoch": 0.11135287341610137, | |
| "grad_norm": 1.0217723846435547, | |
| "learning_rate": 3.7081288670791556e-06, | |
| "loss": 0.6581, | |
| "step": 870 | |
| }, | |
| { | |
| "epoch": 0.11263279150134392, | |
| "grad_norm": 1.2743115425109863, | |
| "learning_rate": 3.750800085342437e-06, | |
| "loss": 0.6606, | |
| "step": 880 | |
| }, | |
| { | |
| "epoch": 0.11391270958658646, | |
| "grad_norm": 0.6405361294746399, | |
| "learning_rate": 3.793471303605718e-06, | |
| "loss": 0.652, | |
| "step": 890 | |
| }, | |
| { | |
| "epoch": 0.115192627671829, | |
| "grad_norm": 0.8357161283493042, | |
| "learning_rate": 3.836142521869e-06, | |
| "loss": 0.6558, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 0.11647254575707154, | |
| "grad_norm": 1.3505173921585083, | |
| "learning_rate": 3.8788137401322815e-06, | |
| "loss": 0.6576, | |
| "step": 910 | |
| }, | |
| { | |
| "epoch": 0.1177524638423141, | |
| "grad_norm": 1.2253319025039673, | |
| "learning_rate": 3.921484958395563e-06, | |
| "loss": 0.6548, | |
| "step": 920 | |
| }, | |
| { | |
| "epoch": 0.11903238192755664, | |
| "grad_norm": 0.6269568800926208, | |
| "learning_rate": 3.964156176658844e-06, | |
| "loss": 0.6562, | |
| "step": 930 | |
| }, | |
| { | |
| "epoch": 0.12031230001279918, | |
| "grad_norm": 0.9413424134254456, | |
| "learning_rate": 4.006827394922125e-06, | |
| "loss": 0.6461, | |
| "step": 940 | |
| }, | |
| { | |
| "epoch": 0.12159221809804173, | |
| "grad_norm": 1.0903513431549072, | |
| "learning_rate": 4.049498613185407e-06, | |
| "loss": 0.6457, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 0.12287213618328427, | |
| "grad_norm": 2.0023152828216553, | |
| "learning_rate": 4.092169831448688e-06, | |
| "loss": 0.6458, | |
| "step": 960 | |
| }, | |
| { | |
| "epoch": 0.12415205426852681, | |
| "grad_norm": 0.9638109803199768, | |
| "learning_rate": 4.134841049711969e-06, | |
| "loss": 0.6486, | |
| "step": 970 | |
| }, | |
| { | |
| "epoch": 0.12543197235376935, | |
| "grad_norm": 0.9894546270370483, | |
| "learning_rate": 4.177512267975251e-06, | |
| "loss": 0.6453, | |
| "step": 980 | |
| }, | |
| { | |
| "epoch": 0.1267118904390119, | |
| "grad_norm": 1.2289057970046997, | |
| "learning_rate": 4.220183486238532e-06, | |
| "loss": 0.6605, | |
| "step": 990 | |
| }, | |
| { | |
| "epoch": 0.12799180852425446, | |
| "grad_norm": 1.119125247001648, | |
| "learning_rate": 4.262854704501814e-06, | |
| "loss": 0.6564, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.12799180852425446, | |
| "eval_accuracy": 0.69068, | |
| "eval_loss": 0.6396247148513794, | |
| "eval_macro_f1": 0.689588440856387, | |
| "eval_precision": 0.6900076024604327, | |
| "eval_recall": 0.6894114140312133, | |
| "eval_runtime": 1634.4321, | |
| "eval_samples_per_second": 61.183, | |
| "eval_steps_per_second": 0.478, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.12927172660949698, | |
| "grad_norm": 2.4629580974578857, | |
| "learning_rate": 4.3055259227650956e-06, | |
| "loss": 0.6485, | |
| "step": 1010 | |
| }, | |
| { | |
| "epoch": 0.13055164469473954, | |
| "grad_norm": 0.7082359790802002, | |
| "learning_rate": 4.348197141028377e-06, | |
| "loss": 0.6491, | |
| "step": 1020 | |
| }, | |
| { | |
| "epoch": 0.1318315627799821, | |
| "grad_norm": 1.4061448574066162, | |
| "learning_rate": 4.390868359291658e-06, | |
| "loss": 0.6439, | |
| "step": 1030 | |
| }, | |
| { | |
| "epoch": 0.13311148086522462, | |
| "grad_norm": 0.8420212864875793, | |
| "learning_rate": 4.4335395775549394e-06, | |
| "loss": 0.6422, | |
| "step": 1040 | |
| }, | |
| { | |
| "epoch": 0.13439139895046717, | |
| "grad_norm": 1.0872454643249512, | |
| "learning_rate": 4.476210795818221e-06, | |
| "loss": 0.6364, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 0.13567131703570973, | |
| "grad_norm": 1.1760668754577637, | |
| "learning_rate": 4.518882014081503e-06, | |
| "loss": 0.6487, | |
| "step": 1060 | |
| }, | |
| { | |
| "epoch": 0.13695123512095225, | |
| "grad_norm": 1.1484756469726562, | |
| "learning_rate": 4.561553232344783e-06, | |
| "loss": 0.6451, | |
| "step": 1070 | |
| }, | |
| { | |
| "epoch": 0.1382311532061948, | |
| "grad_norm": 0.7577368021011353, | |
| "learning_rate": 4.604224450608065e-06, | |
| "loss": 0.6419, | |
| "step": 1080 | |
| }, | |
| { | |
| "epoch": 0.13951107129143736, | |
| "grad_norm": 0.6963039636611938, | |
| "learning_rate": 4.6468956688713465e-06, | |
| "loss": 0.634, | |
| "step": 1090 | |
| }, | |
| { | |
| "epoch": 0.1407909893766799, | |
| "grad_norm": 0.7238577008247375, | |
| "learning_rate": 4.689566887134628e-06, | |
| "loss": 0.6466, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 0.14207090746192244, | |
| "grad_norm": 0.6751101016998291, | |
| "learning_rate": 4.73223810539791e-06, | |
| "loss": 0.6377, | |
| "step": 1110 | |
| }, | |
| { | |
| "epoch": 0.143350825547165, | |
| "grad_norm": 0.7622250318527222, | |
| "learning_rate": 4.774909323661191e-06, | |
| "loss": 0.6318, | |
| "step": 1120 | |
| }, | |
| { | |
| "epoch": 0.14463074363240752, | |
| "grad_norm": 1.3731235265731812, | |
| "learning_rate": 4.817580541924473e-06, | |
| "loss": 0.6464, | |
| "step": 1130 | |
| }, | |
| { | |
| "epoch": 0.14591066171765008, | |
| "grad_norm": 0.8598525524139404, | |
| "learning_rate": 4.8602517601877535e-06, | |
| "loss": 0.6268, | |
| "step": 1140 | |
| }, | |
| { | |
| "epoch": 0.1471905798028926, | |
| "grad_norm": 1.1462016105651855, | |
| "learning_rate": 4.902922978451035e-06, | |
| "loss": 0.6357, | |
| "step": 1150 | |
| }, | |
| { | |
| "epoch": 0.14847049788813516, | |
| "grad_norm": 1.2314528226852417, | |
| "learning_rate": 4.945594196714317e-06, | |
| "loss": 0.6407, | |
| "step": 1160 | |
| }, | |
| { | |
| "epoch": 0.1497504159733777, | |
| "grad_norm": 1.0393744707107544, | |
| "learning_rate": 4.9882654149775974e-06, | |
| "loss": 0.6425, | |
| "step": 1170 | |
| }, | |
| { | |
| "epoch": 0.15103033405862024, | |
| "grad_norm": 0.9097604751586914, | |
| "learning_rate": 5.030936633240879e-06, | |
| "loss": 0.6275, | |
| "step": 1180 | |
| }, | |
| { | |
| "epoch": 0.1523102521438628, | |
| "grad_norm": 0.8274818062782288, | |
| "learning_rate": 5.073607851504161e-06, | |
| "loss": 0.6369, | |
| "step": 1190 | |
| }, | |
| { | |
| "epoch": 0.15359017022910534, | |
| "grad_norm": 0.8929425477981567, | |
| "learning_rate": 5.116279069767442e-06, | |
| "loss": 0.6382, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 0.15359017022910534, | |
| "eval_accuracy": 0.70003, | |
| "eval_loss": 0.6226751804351807, | |
| "eval_macro_f1": 0.6993136399748257, | |
| "eval_precision": 0.6993716582660007, | |
| "eval_recall": 0.6992659459810584, | |
| "eval_runtime": 1635.7462, | |
| "eval_samples_per_second": 61.134, | |
| "eval_steps_per_second": 0.478, | |
| "step": 1200 | |
| } | |
| ], | |
| "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": 2.0206929051648e+16, | |
| "train_batch_size": 64, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |