| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 3.0, | |
| "eval_steps": 500, | |
| "global_step": 1665, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.018026137899954935, | |
| "grad_norm": 21.723926544189453, | |
| "learning_rate": 4.981949458483755e-05, | |
| "loss": 0.8104, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.03605227579990987, | |
| "grad_norm": 55.4254035949707, | |
| "learning_rate": 4.963898916967509e-05, | |
| "loss": 0.5178, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.054078413699864804, | |
| "grad_norm": 41.4888801574707, | |
| "learning_rate": 4.945848375451264e-05, | |
| "loss": 0.5092, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.07210455159981974, | |
| "grad_norm": 25.77608299255371, | |
| "learning_rate": 4.927797833935018e-05, | |
| "loss": 0.5581, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.09013068949977468, | |
| "grad_norm": 61.5268669128418, | |
| "learning_rate": 4.909747292418773e-05, | |
| "loss": 0.5132, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.10815682739972961, | |
| "grad_norm": 18.218244552612305, | |
| "learning_rate": 4.891696750902527e-05, | |
| "loss": 0.4614, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.12618296529968454, | |
| "grad_norm": 30.51982879638672, | |
| "learning_rate": 4.873646209386282e-05, | |
| "loss": 0.4648, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.14420910319963948, | |
| "grad_norm": 20.707260131835938, | |
| "learning_rate": 4.855595667870036e-05, | |
| "loss": 0.4317, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.16223524109959442, | |
| "grad_norm": 21.699357986450195, | |
| "learning_rate": 4.837545126353791e-05, | |
| "loss": 0.4226, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.18026137899954936, | |
| "grad_norm": 51.80731201171875, | |
| "learning_rate": 4.819494584837546e-05, | |
| "loss": 0.5002, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.19828751689950427, | |
| "grad_norm": 44.694942474365234, | |
| "learning_rate": 4.8014440433213e-05, | |
| "loss": 0.4933, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.21631365479945922, | |
| "grad_norm": 22.704530715942383, | |
| "learning_rate": 4.783393501805055e-05, | |
| "loss": 0.4399, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.23433979269941416, | |
| "grad_norm": 26.04544448852539, | |
| "learning_rate": 4.765342960288809e-05, | |
| "loss": 0.438, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.25236593059936907, | |
| "grad_norm": 14.642335891723633, | |
| "learning_rate": 4.747292418772563e-05, | |
| "loss": 0.4591, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.270392068499324, | |
| "grad_norm": 16.13884162902832, | |
| "learning_rate": 4.7292418772563177e-05, | |
| "loss": 0.4546, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.28841820639927895, | |
| "grad_norm": 29.33772087097168, | |
| "learning_rate": 4.711191335740072e-05, | |
| "loss": 0.4665, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.3064443442992339, | |
| "grad_norm": 15.900433540344238, | |
| "learning_rate": 4.693140794223827e-05, | |
| "loss": 0.4703, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.32447048219918884, | |
| "grad_norm": 17.246356964111328, | |
| "learning_rate": 4.675090252707581e-05, | |
| "loss": 0.426, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.3424966200991438, | |
| "grad_norm": 25.457477569580078, | |
| "learning_rate": 4.657039711191336e-05, | |
| "loss": 0.4711, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.3605227579990987, | |
| "grad_norm": 25.43963050842285, | |
| "learning_rate": 4.63898916967509e-05, | |
| "loss": 0.4547, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.3785488958990536, | |
| "grad_norm": 32.49702072143555, | |
| "learning_rate": 4.620938628158845e-05, | |
| "loss": 0.4603, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.39657503379900855, | |
| "grad_norm": 52.78520965576172, | |
| "learning_rate": 4.602888086642599e-05, | |
| "loss": 0.4471, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.4146011716989635, | |
| "grad_norm": 16.70362663269043, | |
| "learning_rate": 4.584837545126354e-05, | |
| "loss": 0.4393, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.43262730959891843, | |
| "grad_norm": 23.55246353149414, | |
| "learning_rate": 4.566787003610109e-05, | |
| "loss": 0.4256, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.45065344749887337, | |
| "grad_norm": 18.868804931640625, | |
| "learning_rate": 4.548736462093863e-05, | |
| "loss": 0.4324, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.4686795853988283, | |
| "grad_norm": 19.074827194213867, | |
| "learning_rate": 4.530685920577618e-05, | |
| "loss": 0.4419, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.48670572329878325, | |
| "grad_norm": 29.232803344726562, | |
| "learning_rate": 4.5126353790613716e-05, | |
| "loss": 0.4179, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.5047318611987381, | |
| "grad_norm": 17.780059814453125, | |
| "learning_rate": 4.494584837545127e-05, | |
| "loss": 0.3917, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.5227579990986931, | |
| "grad_norm": 13.99399471282959, | |
| "learning_rate": 4.4765342960288806e-05, | |
| "loss": 0.4321, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.540784136998648, | |
| "grad_norm": 14.700942039489746, | |
| "learning_rate": 4.458483754512636e-05, | |
| "loss": 0.4163, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.558810274898603, | |
| "grad_norm": 17.43704605102539, | |
| "learning_rate": 4.44043321299639e-05, | |
| "loss": 0.4268, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 0.5768364127985579, | |
| "grad_norm": 28.500301361083984, | |
| "learning_rate": 4.422382671480145e-05, | |
| "loss": 0.4229, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 0.5948625506985128, | |
| "grad_norm": 36.14088439941406, | |
| "learning_rate": 4.404332129963899e-05, | |
| "loss": 0.4559, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 0.6128886885984678, | |
| "grad_norm": 15.279230117797852, | |
| "learning_rate": 4.386281588447654e-05, | |
| "loss": 0.4067, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 0.6309148264984227, | |
| "grad_norm": 21.569740295410156, | |
| "learning_rate": 4.368231046931408e-05, | |
| "loss": 0.4298, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.6489409643983777, | |
| "grad_norm": 20.630115509033203, | |
| "learning_rate": 4.350180505415163e-05, | |
| "loss": 0.4173, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 0.6669671022983326, | |
| "grad_norm": 15.515020370483398, | |
| "learning_rate": 4.332129963898917e-05, | |
| "loss": 0.4063, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 0.6849932401982876, | |
| "grad_norm": 16.33180046081543, | |
| "learning_rate": 4.314079422382672e-05, | |
| "loss": 0.3651, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 0.7030193780982424, | |
| "grad_norm": 16.77664566040039, | |
| "learning_rate": 4.296028880866426e-05, | |
| "loss": 0.3879, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 0.7210455159981974, | |
| "grad_norm": 11.113824844360352, | |
| "learning_rate": 4.277978339350181e-05, | |
| "loss": 0.4205, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.7390716538981523, | |
| "grad_norm": 17.835678100585938, | |
| "learning_rate": 4.259927797833935e-05, | |
| "loss": 0.4466, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 0.7570977917981072, | |
| "grad_norm": 24.48379135131836, | |
| "learning_rate": 4.24187725631769e-05, | |
| "loss": 0.4084, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 0.7751239296980622, | |
| "grad_norm": 29.183530807495117, | |
| "learning_rate": 4.223826714801444e-05, | |
| "loss": 0.3588, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 0.7931500675980171, | |
| "grad_norm": 16.398386001586914, | |
| "learning_rate": 4.205776173285199e-05, | |
| "loss": 0.4594, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 0.8111762054979721, | |
| "grad_norm": 24.11660385131836, | |
| "learning_rate": 4.187725631768953e-05, | |
| "loss": 0.376, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 0.829202343397927, | |
| "grad_norm": 21.480924606323242, | |
| "learning_rate": 4.169675090252708e-05, | |
| "loss": 0.4035, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 0.847228481297882, | |
| "grad_norm": 11.91422176361084, | |
| "learning_rate": 4.151624548736462e-05, | |
| "loss": 0.4074, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 0.8652546191978369, | |
| "grad_norm": 17.17136001586914, | |
| "learning_rate": 4.1335740072202167e-05, | |
| "loss": 0.4259, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 0.8832807570977917, | |
| "grad_norm": 15.024781227111816, | |
| "learning_rate": 4.115523465703972e-05, | |
| "loss": 0.413, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 0.9013068949977467, | |
| "grad_norm": 15.877848625183105, | |
| "learning_rate": 4.0974729241877256e-05, | |
| "loss": 0.4013, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.9193330328977016, | |
| "grad_norm": 20.711393356323242, | |
| "learning_rate": 4.079422382671481e-05, | |
| "loss": 0.3875, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 0.9373591707976566, | |
| "grad_norm": 14.360701560974121, | |
| "learning_rate": 4.0613718411552346e-05, | |
| "loss": 0.4025, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 0.9553853086976115, | |
| "grad_norm": 16.94271469116211, | |
| "learning_rate": 4.043321299638989e-05, | |
| "loss": 0.3971, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 0.9734114465975665, | |
| "grad_norm": 23.218050003051758, | |
| "learning_rate": 4.0252707581227436e-05, | |
| "loss": 0.4305, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 0.9914375844975214, | |
| "grad_norm": 25.439455032348633, | |
| "learning_rate": 4.007220216606498e-05, | |
| "loss": 0.3969, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "eval_f1": 0.8410362694300518, | |
| "eval_loss": 0.36654186248779297, | |
| "eval_precision": 0.7994483845547675, | |
| "eval_recall": 0.8871884564932225, | |
| "eval_runtime": 77.9824, | |
| "eval_samples_per_second": 59.924, | |
| "eval_steps_per_second": 7.502, | |
| "step": 555 | |
| }, | |
| { | |
| "epoch": 1.0090130689499774, | |
| "grad_norm": 23.230247497558594, | |
| "learning_rate": 3.989169675090253e-05, | |
| "loss": 0.3393, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 1.0270392068499323, | |
| "grad_norm": 20.747711181640625, | |
| "learning_rate": 3.971119133574007e-05, | |
| "loss": 0.2951, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 1.0450653447498874, | |
| "grad_norm": 16.3277645111084, | |
| "learning_rate": 3.953068592057762e-05, | |
| "loss": 0.3178, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 1.0630914826498423, | |
| "grad_norm": 27.63875389099121, | |
| "learning_rate": 3.935018050541516e-05, | |
| "loss": 0.3043, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 1.0811176205497972, | |
| "grad_norm": 24.972593307495117, | |
| "learning_rate": 3.916967509025271e-05, | |
| "loss": 0.3521, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 1.099143758449752, | |
| "grad_norm": 22.260204315185547, | |
| "learning_rate": 3.898916967509025e-05, | |
| "loss": 0.2859, | |
| "step": 610 | |
| }, | |
| { | |
| "epoch": 1.117169896349707, | |
| "grad_norm": 18.97205924987793, | |
| "learning_rate": 3.88086642599278e-05, | |
| "loss": 0.3159, | |
| "step": 620 | |
| }, | |
| { | |
| "epoch": 1.135196034249662, | |
| "grad_norm": 25.5858097076416, | |
| "learning_rate": 3.862815884476535e-05, | |
| "loss": 0.2881, | |
| "step": 630 | |
| }, | |
| { | |
| "epoch": 1.153222172149617, | |
| "grad_norm": 16.255144119262695, | |
| "learning_rate": 3.844765342960289e-05, | |
| "loss": 0.356, | |
| "step": 640 | |
| }, | |
| { | |
| "epoch": 1.1712483100495719, | |
| "grad_norm": 16.83954429626465, | |
| "learning_rate": 3.826714801444044e-05, | |
| "loss": 0.2792, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 1.1892744479495267, | |
| "grad_norm": 13.091219902038574, | |
| "learning_rate": 3.8086642599277976e-05, | |
| "loss": 0.3232, | |
| "step": 660 | |
| }, | |
| { | |
| "epoch": 1.2073005858494819, | |
| "grad_norm": 17.03887367248535, | |
| "learning_rate": 3.790613718411553e-05, | |
| "loss": 0.3081, | |
| "step": 670 | |
| }, | |
| { | |
| "epoch": 1.2253267237494367, | |
| "grad_norm": 24.6894588470459, | |
| "learning_rate": 3.7725631768953066e-05, | |
| "loss": 0.29, | |
| "step": 680 | |
| }, | |
| { | |
| "epoch": 1.2433528616493916, | |
| "grad_norm": 23.55462646484375, | |
| "learning_rate": 3.754512635379062e-05, | |
| "loss": 0.3232, | |
| "step": 690 | |
| }, | |
| { | |
| "epoch": 1.2613789995493465, | |
| "grad_norm": 44.3718147277832, | |
| "learning_rate": 3.7364620938628155e-05, | |
| "loss": 0.2878, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 1.2794051374493014, | |
| "grad_norm": 21.339521408081055, | |
| "learning_rate": 3.718411552346571e-05, | |
| "loss": 0.2625, | |
| "step": 710 | |
| }, | |
| { | |
| "epoch": 1.2974312753492563, | |
| "grad_norm": 22.27944564819336, | |
| "learning_rate": 3.700361010830325e-05, | |
| "loss": 0.3261, | |
| "step": 720 | |
| }, | |
| { | |
| "epoch": 1.3154574132492114, | |
| "grad_norm": 20.857757568359375, | |
| "learning_rate": 3.68231046931408e-05, | |
| "loss": 0.3194, | |
| "step": 730 | |
| }, | |
| { | |
| "epoch": 1.3334835511491663, | |
| "grad_norm": 15.615569114685059, | |
| "learning_rate": 3.664259927797834e-05, | |
| "loss": 0.3099, | |
| "step": 740 | |
| }, | |
| { | |
| "epoch": 1.3515096890491212, | |
| "grad_norm": 13.210846900939941, | |
| "learning_rate": 3.646209386281589e-05, | |
| "loss": 0.2961, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 1.3695358269490763, | |
| "grad_norm": 20.196102142333984, | |
| "learning_rate": 3.628158844765343e-05, | |
| "loss": 0.3099, | |
| "step": 760 | |
| }, | |
| { | |
| "epoch": 1.3875619648490312, | |
| "grad_norm": 14.043251991271973, | |
| "learning_rate": 3.610108303249098e-05, | |
| "loss": 0.3548, | |
| "step": 770 | |
| }, | |
| { | |
| "epoch": 1.405588102748986, | |
| "grad_norm": 31.82477569580078, | |
| "learning_rate": 3.592057761732852e-05, | |
| "loss": 0.3255, | |
| "step": 780 | |
| }, | |
| { | |
| "epoch": 1.423614240648941, | |
| "grad_norm": 19.933557510375977, | |
| "learning_rate": 3.574007220216607e-05, | |
| "loss": 0.3404, | |
| "step": 790 | |
| }, | |
| { | |
| "epoch": 1.4416403785488958, | |
| "grad_norm": 27.846755981445312, | |
| "learning_rate": 3.555956678700361e-05, | |
| "loss": 0.3054, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 1.4596665164488507, | |
| "grad_norm": 18.607879638671875, | |
| "learning_rate": 3.537906137184116e-05, | |
| "loss": 0.2802, | |
| "step": 810 | |
| }, | |
| { | |
| "epoch": 1.4776926543488058, | |
| "grad_norm": 29.801652908325195, | |
| "learning_rate": 3.51985559566787e-05, | |
| "loss": 0.2975, | |
| "step": 820 | |
| }, | |
| { | |
| "epoch": 1.4957187922487607, | |
| "grad_norm": 21.80506134033203, | |
| "learning_rate": 3.5018050541516247e-05, | |
| "loss": 0.3251, | |
| "step": 830 | |
| }, | |
| { | |
| "epoch": 1.5137449301487156, | |
| "grad_norm": 12.826874732971191, | |
| "learning_rate": 3.483754512635379e-05, | |
| "loss": 0.2943, | |
| "step": 840 | |
| }, | |
| { | |
| "epoch": 1.5317710680486707, | |
| "grad_norm": 23.97177505493164, | |
| "learning_rate": 3.4657039711191336e-05, | |
| "loss": 0.3221, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 1.5497972059486256, | |
| "grad_norm": 20.326284408569336, | |
| "learning_rate": 3.447653429602888e-05, | |
| "loss": 0.3276, | |
| "step": 860 | |
| }, | |
| { | |
| "epoch": 1.5678233438485805, | |
| "grad_norm": 19.27255630493164, | |
| "learning_rate": 3.4296028880866426e-05, | |
| "loss": 0.3338, | |
| "step": 870 | |
| }, | |
| { | |
| "epoch": 1.5858494817485354, | |
| "grad_norm": 17.299917221069336, | |
| "learning_rate": 3.411552346570397e-05, | |
| "loss": 0.3177, | |
| "step": 880 | |
| }, | |
| { | |
| "epoch": 1.6038756196484902, | |
| "grad_norm": 15.277181625366211, | |
| "learning_rate": 3.3935018050541516e-05, | |
| "loss": 0.2773, | |
| "step": 890 | |
| }, | |
| { | |
| "epoch": 1.6219017575484451, | |
| "grad_norm": 27.945833206176758, | |
| "learning_rate": 3.375451263537907e-05, | |
| "loss": 0.3417, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 1.6399278954484002, | |
| "grad_norm": 25.78679847717285, | |
| "learning_rate": 3.3574007220216606e-05, | |
| "loss": 0.2959, | |
| "step": 910 | |
| }, | |
| { | |
| "epoch": 1.6579540333483551, | |
| "grad_norm": 17.608028411865234, | |
| "learning_rate": 3.339350180505416e-05, | |
| "loss": 0.273, | |
| "step": 920 | |
| }, | |
| { | |
| "epoch": 1.67598017124831, | |
| "grad_norm": 17.937299728393555, | |
| "learning_rate": 3.3212996389891696e-05, | |
| "loss": 0.3121, | |
| "step": 930 | |
| }, | |
| { | |
| "epoch": 1.694006309148265, | |
| "grad_norm": 22.207386016845703, | |
| "learning_rate": 3.303249097472924e-05, | |
| "loss": 0.3389, | |
| "step": 940 | |
| }, | |
| { | |
| "epoch": 1.71203244704822, | |
| "grad_norm": 40.351959228515625, | |
| "learning_rate": 3.2851985559566786e-05, | |
| "loss": 0.3138, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 1.7300585849481749, | |
| "grad_norm": 23.845033645629883, | |
| "learning_rate": 3.267148014440433e-05, | |
| "loss": 0.2865, | |
| "step": 960 | |
| }, | |
| { | |
| "epoch": 1.7480847228481298, | |
| "grad_norm": 18.095985412597656, | |
| "learning_rate": 3.249097472924188e-05, | |
| "loss": 0.294, | |
| "step": 970 | |
| }, | |
| { | |
| "epoch": 1.7661108607480847, | |
| "grad_norm": 25.775617599487305, | |
| "learning_rate": 3.231046931407942e-05, | |
| "loss": 0.2785, | |
| "step": 980 | |
| }, | |
| { | |
| "epoch": 1.7841369986480395, | |
| "grad_norm": 20.0789737701416, | |
| "learning_rate": 3.212996389891697e-05, | |
| "loss": 0.3533, | |
| "step": 990 | |
| }, | |
| { | |
| "epoch": 1.8021631365479944, | |
| "grad_norm": 26.862163543701172, | |
| "learning_rate": 3.194945848375451e-05, | |
| "loss": 0.2957, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 1.8201892744479495, | |
| "grad_norm": 24.20423698425293, | |
| "learning_rate": 3.176895306859206e-05, | |
| "loss": 0.3195, | |
| "step": 1010 | |
| }, | |
| { | |
| "epoch": 1.8382154123479044, | |
| "grad_norm": 25.36762809753418, | |
| "learning_rate": 3.15884476534296e-05, | |
| "loss": 0.2823, | |
| "step": 1020 | |
| }, | |
| { | |
| "epoch": 1.8562415502478595, | |
| "grad_norm": 37.54848098754883, | |
| "learning_rate": 3.140794223826715e-05, | |
| "loss": 0.3557, | |
| "step": 1030 | |
| }, | |
| { | |
| "epoch": 1.8742676881478144, | |
| "grad_norm": 24.26515769958496, | |
| "learning_rate": 3.12274368231047e-05, | |
| "loss": 0.3194, | |
| "step": 1040 | |
| }, | |
| { | |
| "epoch": 1.8922938260477693, | |
| "grad_norm": 22.1879940032959, | |
| "learning_rate": 3.104693140794224e-05, | |
| "loss": 0.2912, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 1.9103199639477242, | |
| "grad_norm": 21.703449249267578, | |
| "learning_rate": 3.086642599277979e-05, | |
| "loss": 0.2635, | |
| "step": 1060 | |
| }, | |
| { | |
| "epoch": 1.928346101847679, | |
| "grad_norm": 33.39324951171875, | |
| "learning_rate": 3.0685920577617325e-05, | |
| "loss": 0.2943, | |
| "step": 1070 | |
| }, | |
| { | |
| "epoch": 1.946372239747634, | |
| "grad_norm": 21.398008346557617, | |
| "learning_rate": 3.0505415162454877e-05, | |
| "loss": 0.353, | |
| "step": 1080 | |
| }, | |
| { | |
| "epoch": 1.9643983776475888, | |
| "grad_norm": 17.463472366333008, | |
| "learning_rate": 3.032490974729242e-05, | |
| "loss": 0.2934, | |
| "step": 1090 | |
| }, | |
| { | |
| "epoch": 1.982424515547544, | |
| "grad_norm": 32.96004104614258, | |
| "learning_rate": 3.0144404332129967e-05, | |
| "loss": 0.3206, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "grad_norm": 22.864667892456055, | |
| "learning_rate": 2.996389891696751e-05, | |
| "loss": 0.2765, | |
| "step": 1110 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "eval_f1": 0.8415584415584415, | |
| "eval_loss": 0.3972894847393036, | |
| "eval_precision": 0.7748344370860927, | |
| "eval_recall": 0.9208570179274158, | |
| "eval_runtime": 67.5759, | |
| "eval_samples_per_second": 69.152, | |
| "eval_steps_per_second": 8.657, | |
| "step": 1110 | |
| }, | |
| { | |
| "epoch": 2.018026137899955, | |
| "grad_norm": 25.821582794189453, | |
| "learning_rate": 2.9783393501805057e-05, | |
| "loss": 0.1921, | |
| "step": 1120 | |
| }, | |
| { | |
| "epoch": 2.0360522757999098, | |
| "grad_norm": 29.275466918945312, | |
| "learning_rate": 2.9602888086642598e-05, | |
| "loss": 0.1621, | |
| "step": 1130 | |
| }, | |
| { | |
| "epoch": 2.0540784136998647, | |
| "grad_norm": 26.00608253479004, | |
| "learning_rate": 2.9422382671480147e-05, | |
| "loss": 0.1792, | |
| "step": 1140 | |
| }, | |
| { | |
| "epoch": 2.0721045515998195, | |
| "grad_norm": 44.50676345825195, | |
| "learning_rate": 2.924187725631769e-05, | |
| "loss": 0.1525, | |
| "step": 1150 | |
| }, | |
| { | |
| "epoch": 2.090130689499775, | |
| "grad_norm": 35.6057014465332, | |
| "learning_rate": 2.906137184115524e-05, | |
| "loss": 0.1869, | |
| "step": 1160 | |
| }, | |
| { | |
| "epoch": 2.1081568273997298, | |
| "grad_norm": 15.054738998413086, | |
| "learning_rate": 2.888086642599278e-05, | |
| "loss": 0.1549, | |
| "step": 1170 | |
| }, | |
| { | |
| "epoch": 2.1261829652996846, | |
| "grad_norm": 31.124298095703125, | |
| "learning_rate": 2.870036101083033e-05, | |
| "loss": 0.1712, | |
| "step": 1180 | |
| }, | |
| { | |
| "epoch": 2.1442091031996395, | |
| "grad_norm": 31.94968032836914, | |
| "learning_rate": 2.851985559566787e-05, | |
| "loss": 0.1562, | |
| "step": 1190 | |
| }, | |
| { | |
| "epoch": 2.1622352410995944, | |
| "grad_norm": 27.812400817871094, | |
| "learning_rate": 2.8339350180505413e-05, | |
| "loss": 0.1615, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 2.1802613789995493, | |
| "grad_norm": 20.641502380371094, | |
| "learning_rate": 2.815884476534296e-05, | |
| "loss": 0.1744, | |
| "step": 1210 | |
| }, | |
| { | |
| "epoch": 2.198287516899504, | |
| "grad_norm": 22.02704429626465, | |
| "learning_rate": 2.7978339350180506e-05, | |
| "loss": 0.1794, | |
| "step": 1220 | |
| }, | |
| { | |
| "epoch": 2.216313654799459, | |
| "grad_norm": 23.43486213684082, | |
| "learning_rate": 2.779783393501805e-05, | |
| "loss": 0.1821, | |
| "step": 1230 | |
| }, | |
| { | |
| "epoch": 2.234339792699414, | |
| "grad_norm": 11.662972450256348, | |
| "learning_rate": 2.7617328519855596e-05, | |
| "loss": 0.1842, | |
| "step": 1240 | |
| }, | |
| { | |
| "epoch": 2.2523659305993693, | |
| "grad_norm": 55.82278823852539, | |
| "learning_rate": 2.7436823104693144e-05, | |
| "loss": 0.1587, | |
| "step": 1250 | |
| }, | |
| { | |
| "epoch": 2.270392068499324, | |
| "grad_norm": 52.459190368652344, | |
| "learning_rate": 2.7256317689530686e-05, | |
| "loss": 0.1755, | |
| "step": 1260 | |
| }, | |
| { | |
| "epoch": 2.288418206399279, | |
| "grad_norm": 11.363420486450195, | |
| "learning_rate": 2.7075812274368234e-05, | |
| "loss": 0.1507, | |
| "step": 1270 | |
| }, | |
| { | |
| "epoch": 2.306444344299234, | |
| "grad_norm": 19.20196533203125, | |
| "learning_rate": 2.6895306859205776e-05, | |
| "loss": 0.1538, | |
| "step": 1280 | |
| }, | |
| { | |
| "epoch": 2.324470482199189, | |
| "grad_norm": 24.035587310791016, | |
| "learning_rate": 2.6714801444043324e-05, | |
| "loss": 0.1604, | |
| "step": 1290 | |
| }, | |
| { | |
| "epoch": 2.3424966200991437, | |
| "grad_norm": 25.760459899902344, | |
| "learning_rate": 2.6534296028880866e-05, | |
| "loss": 0.179, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 2.3605227579990986, | |
| "grad_norm": 38.57444381713867, | |
| "learning_rate": 2.6353790613718414e-05, | |
| "loss": 0.1668, | |
| "step": 1310 | |
| }, | |
| { | |
| "epoch": 2.3785488958990535, | |
| "grad_norm": 26.489524841308594, | |
| "learning_rate": 2.617328519855596e-05, | |
| "loss": 0.1531, | |
| "step": 1320 | |
| }, | |
| { | |
| "epoch": 2.3965750337990084, | |
| "grad_norm": 30.741003036499023, | |
| "learning_rate": 2.59927797833935e-05, | |
| "loss": 0.201, | |
| "step": 1330 | |
| }, | |
| { | |
| "epoch": 2.4146011716989637, | |
| "grad_norm": 13.293583869934082, | |
| "learning_rate": 2.581227436823105e-05, | |
| "loss": 0.1413, | |
| "step": 1340 | |
| }, | |
| { | |
| "epoch": 2.4326273095989186, | |
| "grad_norm": 38.17484664916992, | |
| "learning_rate": 2.563176895306859e-05, | |
| "loss": 0.1479, | |
| "step": 1350 | |
| }, | |
| { | |
| "epoch": 2.4506534474988735, | |
| "grad_norm": 26.58913230895996, | |
| "learning_rate": 2.545126353790614e-05, | |
| "loss": 0.162, | |
| "step": 1360 | |
| }, | |
| { | |
| "epoch": 2.4686795853988284, | |
| "grad_norm": 18.360992431640625, | |
| "learning_rate": 2.527075812274368e-05, | |
| "loss": 0.1938, | |
| "step": 1370 | |
| }, | |
| { | |
| "epoch": 2.4867057232987833, | |
| "grad_norm": 31.35187339782715, | |
| "learning_rate": 2.509025270758123e-05, | |
| "loss": 0.1895, | |
| "step": 1380 | |
| }, | |
| { | |
| "epoch": 2.504731861198738, | |
| "grad_norm": 23.93364715576172, | |
| "learning_rate": 2.4909747292418774e-05, | |
| "loss": 0.1983, | |
| "step": 1390 | |
| }, | |
| { | |
| "epoch": 2.522757999098693, | |
| "grad_norm": 48.125240325927734, | |
| "learning_rate": 2.472924187725632e-05, | |
| "loss": 0.2107, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 2.540784136998648, | |
| "grad_norm": 16.627723693847656, | |
| "learning_rate": 2.4548736462093864e-05, | |
| "loss": 0.1798, | |
| "step": 1410 | |
| }, | |
| { | |
| "epoch": 2.558810274898603, | |
| "grad_norm": 33.984134674072266, | |
| "learning_rate": 2.436823104693141e-05, | |
| "loss": 0.1903, | |
| "step": 1420 | |
| }, | |
| { | |
| "epoch": 2.576836412798558, | |
| "grad_norm": 41.849796295166016, | |
| "learning_rate": 2.4187725631768953e-05, | |
| "loss": 0.1883, | |
| "step": 1430 | |
| }, | |
| { | |
| "epoch": 2.5948625506985126, | |
| "grad_norm": 35.38132095336914, | |
| "learning_rate": 2.40072202166065e-05, | |
| "loss": 0.218, | |
| "step": 1440 | |
| }, | |
| { | |
| "epoch": 2.612888688598468, | |
| "grad_norm": 45.718360900878906, | |
| "learning_rate": 2.3826714801444043e-05, | |
| "loss": 0.1857, | |
| "step": 1450 | |
| }, | |
| { | |
| "epoch": 2.630914826498423, | |
| "grad_norm": 21.124282836914062, | |
| "learning_rate": 2.3646209386281588e-05, | |
| "loss": 0.1966, | |
| "step": 1460 | |
| }, | |
| { | |
| "epoch": 2.6489409643983777, | |
| "grad_norm": 51.59513854980469, | |
| "learning_rate": 2.3465703971119137e-05, | |
| "loss": 0.2052, | |
| "step": 1470 | |
| }, | |
| { | |
| "epoch": 2.6669671022983326, | |
| "grad_norm": 27.74148941040039, | |
| "learning_rate": 2.328519855595668e-05, | |
| "loss": 0.1972, | |
| "step": 1480 | |
| }, | |
| { | |
| "epoch": 2.6849932401982874, | |
| "grad_norm": 23.677188873291016, | |
| "learning_rate": 2.3104693140794227e-05, | |
| "loss": 0.1564, | |
| "step": 1490 | |
| }, | |
| { | |
| "epoch": 2.7030193780982423, | |
| "grad_norm": 37.59199523925781, | |
| "learning_rate": 2.292418772563177e-05, | |
| "loss": 0.1556, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 2.721045515998197, | |
| "grad_norm": 39.62023162841797, | |
| "learning_rate": 2.2743682310469316e-05, | |
| "loss": 0.181, | |
| "step": 1510 | |
| }, | |
| { | |
| "epoch": 2.7390716538981525, | |
| "grad_norm": 41.49861526489258, | |
| "learning_rate": 2.2563176895306858e-05, | |
| "loss": 0.1858, | |
| "step": 1520 | |
| }, | |
| { | |
| "epoch": 2.757097791798107, | |
| "grad_norm": 24.64577293395996, | |
| "learning_rate": 2.2382671480144403e-05, | |
| "loss": 0.1665, | |
| "step": 1530 | |
| }, | |
| { | |
| "epoch": 2.7751239296980623, | |
| "grad_norm": 19.250690460205078, | |
| "learning_rate": 2.220216606498195e-05, | |
| "loss": 0.1582, | |
| "step": 1540 | |
| }, | |
| { | |
| "epoch": 2.793150067598017, | |
| "grad_norm": 32.70537185668945, | |
| "learning_rate": 2.2021660649819496e-05, | |
| "loss": 0.1774, | |
| "step": 1550 | |
| }, | |
| { | |
| "epoch": 2.811176205497972, | |
| "grad_norm": 27.5251522064209, | |
| "learning_rate": 2.184115523465704e-05, | |
| "loss": 0.1877, | |
| "step": 1560 | |
| }, | |
| { | |
| "epoch": 2.829202343397927, | |
| "grad_norm": 34.118892669677734, | |
| "learning_rate": 2.1660649819494586e-05, | |
| "loss": 0.1763, | |
| "step": 1570 | |
| }, | |
| { | |
| "epoch": 2.847228481297882, | |
| "grad_norm": 23.93291473388672, | |
| "learning_rate": 2.148014440433213e-05, | |
| "loss": 0.1448, | |
| "step": 1580 | |
| }, | |
| { | |
| "epoch": 2.8652546191978367, | |
| "grad_norm": 35.88968276977539, | |
| "learning_rate": 2.1299638989169676e-05, | |
| "loss": 0.2251, | |
| "step": 1590 | |
| }, | |
| { | |
| "epoch": 2.8832807570977916, | |
| "grad_norm": 28.2044620513916, | |
| "learning_rate": 2.111913357400722e-05, | |
| "loss": 0.1503, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 2.901306894997747, | |
| "grad_norm": 36.51847457885742, | |
| "learning_rate": 2.0938628158844766e-05, | |
| "loss": 0.1471, | |
| "step": 1610 | |
| }, | |
| { | |
| "epoch": 2.9193330328977014, | |
| "grad_norm": 35.991207122802734, | |
| "learning_rate": 2.075812274368231e-05, | |
| "loss": 0.1822, | |
| "step": 1620 | |
| }, | |
| { | |
| "epoch": 2.9373591707976567, | |
| "grad_norm": 47.02556610107422, | |
| "learning_rate": 2.057761732851986e-05, | |
| "loss": 0.1727, | |
| "step": 1630 | |
| }, | |
| { | |
| "epoch": 2.9553853086976116, | |
| "grad_norm": 22.891300201416016, | |
| "learning_rate": 2.0397111913357404e-05, | |
| "loss": 0.1627, | |
| "step": 1640 | |
| }, | |
| { | |
| "epoch": 2.9734114465975665, | |
| "grad_norm": 47.76613998413086, | |
| "learning_rate": 2.0216606498194946e-05, | |
| "loss": 0.171, | |
| "step": 1650 | |
| }, | |
| { | |
| "epoch": 2.9914375844975214, | |
| "grad_norm": 16.816410064697266, | |
| "learning_rate": 2.003610108303249e-05, | |
| "loss": 0.1314, | |
| "step": 1660 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "eval_f1": 0.8327495621716288, | |
| "eval_loss": 0.5104337930679321, | |
| "eval_precision": 0.8338448049101271, | |
| "eval_recall": 0.8316571928290337, | |
| "eval_runtime": 66.4346, | |
| "eval_samples_per_second": 70.34, | |
| "eval_steps_per_second": 8.806, | |
| "step": 1665 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 2770, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 5, | |
| "save_steps": 500, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 5.787771384265114e+16, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |