| { |
| "best_global_step": 366, |
| "best_metric": 0.41911664605140686, |
| "best_model_checkpoint": "training_output/run_20260713_203733/model/checkpoint-366", |
| "epoch": 3.0, |
| "eval_steps": 500, |
| "global_step": 1098, |
| "is_hyper_param_search": false, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.0273224043715847, |
| "grad_norm": 4.927176475524902, |
| "learning_rate": 7.377049180327868e-07, |
| "loss": 0.6899909496307373, |
| "step": 10 |
| }, |
| { |
| "epoch": 0.0546448087431694, |
| "grad_norm": 6.0230512619018555, |
| "learning_rate": 1.557377049180328e-06, |
| "loss": 0.7346426486968994, |
| "step": 20 |
| }, |
| { |
| "epoch": 0.08196721311475409, |
| "grad_norm": 6.2535271644592285, |
| "learning_rate": 2.377049180327869e-06, |
| "loss": 0.6693419456481934, |
| "step": 30 |
| }, |
| { |
| "epoch": 0.1092896174863388, |
| "grad_norm": 10.038297653198242, |
| "learning_rate": 3.1967213114754097e-06, |
| "loss": 0.6888828277587891, |
| "step": 40 |
| }, |
| { |
| "epoch": 0.1366120218579235, |
| "grad_norm": 10.041961669921875, |
| "learning_rate": 4.016393442622951e-06, |
| "loss": 0.623580026626587, |
| "step": 50 |
| }, |
| { |
| "epoch": 0.16393442622950818, |
| "grad_norm": 9.107032775878906, |
| "learning_rate": 4.836065573770492e-06, |
| "loss": 0.692430830001831, |
| "step": 60 |
| }, |
| { |
| "epoch": 0.1912568306010929, |
| "grad_norm": 9.090417861938477, |
| "learning_rate": 5.655737704918032e-06, |
| "loss": 0.6110644817352295, |
| "step": 70 |
| }, |
| { |
| "epoch": 0.2185792349726776, |
| "grad_norm": 7.914913177490234, |
| "learning_rate": 6.4754098360655735e-06, |
| "loss": 0.6268288612365722, |
| "step": 80 |
| }, |
| { |
| "epoch": 0.2459016393442623, |
| "grad_norm": 8.18041706085205, |
| "learning_rate": 7.2950819672131145e-06, |
| "loss": 0.6859075546264648, |
| "step": 90 |
| }, |
| { |
| "epoch": 0.273224043715847, |
| "grad_norm": 5.7157392501831055, |
| "learning_rate": 8.114754098360657e-06, |
| "loss": 0.7717578887939454, |
| "step": 100 |
| }, |
| { |
| "epoch": 0.3005464480874317, |
| "grad_norm": 11.109936714172363, |
| "learning_rate": 8.934426229508197e-06, |
| "loss": 0.6476659774780273, |
| "step": 110 |
| }, |
| { |
| "epoch": 0.32786885245901637, |
| "grad_norm": 10.369423866271973, |
| "learning_rate": 9.754098360655738e-06, |
| "loss": 0.6707509994506836, |
| "step": 120 |
| }, |
| { |
| "epoch": 0.3551912568306011, |
| "grad_norm": 10.730874061584473, |
| "learning_rate": 1.0573770491803279e-05, |
| "loss": 0.6787842750549317, |
| "step": 130 |
| }, |
| { |
| "epoch": 0.3825136612021858, |
| "grad_norm": 8.120079040527344, |
| "learning_rate": 1.139344262295082e-05, |
| "loss": 0.6765446186065673, |
| "step": 140 |
| }, |
| { |
| "epoch": 0.4098360655737705, |
| "grad_norm": 14.429889678955078, |
| "learning_rate": 1.221311475409836e-05, |
| "loss": 0.623370361328125, |
| "step": 150 |
| }, |
| { |
| "epoch": 0.4371584699453552, |
| "grad_norm": 9.193517684936523, |
| "learning_rate": 1.3032786885245902e-05, |
| "loss": 0.6362186908721924, |
| "step": 160 |
| }, |
| { |
| "epoch": 0.4644808743169399, |
| "grad_norm": 8.7772855758667, |
| "learning_rate": 1.3852459016393443e-05, |
| "loss": 0.6641538619995118, |
| "step": 170 |
| }, |
| { |
| "epoch": 0.4918032786885246, |
| "grad_norm": 5.2151780128479, |
| "learning_rate": 1.4672131147540984e-05, |
| "loss": 0.6653701782226562, |
| "step": 180 |
| }, |
| { |
| "epoch": 0.5191256830601093, |
| "grad_norm": 17.197834014892578, |
| "learning_rate": 1.5491803278688525e-05, |
| "loss": 0.6582591533660889, |
| "step": 190 |
| }, |
| { |
| "epoch": 0.546448087431694, |
| "grad_norm": 7.26473331451416, |
| "learning_rate": 1.6311475409836068e-05, |
| "loss": 0.765175724029541, |
| "step": 200 |
| }, |
| { |
| "epoch": 0.5737704918032787, |
| "grad_norm": 5.47290563583374, |
| "learning_rate": 1.7131147540983607e-05, |
| "loss": 0.676247501373291, |
| "step": 210 |
| }, |
| { |
| "epoch": 0.6010928961748634, |
| "grad_norm": 4.55684232711792, |
| "learning_rate": 1.7950819672131146e-05, |
| "loss": 0.6748649597167968, |
| "step": 220 |
| }, |
| { |
| "epoch": 0.6284153005464481, |
| "grad_norm": 9.52070140838623, |
| "learning_rate": 1.877049180327869e-05, |
| "loss": 0.7023877143859864, |
| "step": 230 |
| }, |
| { |
| "epoch": 0.6557377049180327, |
| "grad_norm": 10.813623428344727, |
| "learning_rate": 1.959016393442623e-05, |
| "loss": 0.5075790405273437, |
| "step": 240 |
| }, |
| { |
| "epoch": 0.6830601092896175, |
| "grad_norm": 15.77322769165039, |
| "learning_rate": 2.040983606557377e-05, |
| "loss": 0.5947454929351806, |
| "step": 250 |
| }, |
| { |
| "epoch": 0.7103825136612022, |
| "grad_norm": 14.141736030578613, |
| "learning_rate": 2.122950819672131e-05, |
| "loss": 0.7629669666290283, |
| "step": 260 |
| }, |
| { |
| "epoch": 0.7377049180327869, |
| "grad_norm": 0.5761239528656006, |
| "learning_rate": 2.2049180327868853e-05, |
| "loss": 0.717600679397583, |
| "step": 270 |
| }, |
| { |
| "epoch": 0.7650273224043715, |
| "grad_norm": 13.026739120483398, |
| "learning_rate": 2.2868852459016393e-05, |
| "loss": 1.5919431686401366, |
| "step": 280 |
| }, |
| { |
| "epoch": 0.7923497267759563, |
| "grad_norm": 12.15133285522461, |
| "learning_rate": 2.3688524590163936e-05, |
| "loss": 0.6558570861816406, |
| "step": 290 |
| }, |
| { |
| "epoch": 0.819672131147541, |
| "grad_norm": 20.35292625427246, |
| "learning_rate": 2.4508196721311478e-05, |
| "loss": 0.6991987705230713, |
| "step": 300 |
| }, |
| { |
| "epoch": 0.8469945355191257, |
| "grad_norm": 9.59134292602539, |
| "learning_rate": 2.5327868852459018e-05, |
| "loss": 1.1239334106445313, |
| "step": 310 |
| }, |
| { |
| "epoch": 0.8743169398907104, |
| "grad_norm": 4.265232086181641, |
| "learning_rate": 2.6147540983606557e-05, |
| "loss": 0.4866987705230713, |
| "step": 320 |
| }, |
| { |
| "epoch": 0.9016393442622951, |
| "grad_norm": 9.87460708618164, |
| "learning_rate": 2.69672131147541e-05, |
| "loss": 0.7435293674468995, |
| "step": 330 |
| }, |
| { |
| "epoch": 0.9289617486338798, |
| "grad_norm": 12.707701683044434, |
| "learning_rate": 2.7786885245901642e-05, |
| "loss": 0.9141542434692382, |
| "step": 340 |
| }, |
| { |
| "epoch": 0.9562841530054644, |
| "grad_norm": 2.9138967990875244, |
| "learning_rate": 2.860655737704918e-05, |
| "loss": 0.5319347858428956, |
| "step": 350 |
| }, |
| { |
| "epoch": 0.9836065573770492, |
| "grad_norm": 6.917504787445068, |
| "learning_rate": 2.942622950819672e-05, |
| "loss": 0.5438683509826661, |
| "step": 360 |
| }, |
| { |
| "epoch": 1.0, |
| "eval_accuracy": 0.8492, |
| "eval_confusion_matrix": [ |
| [ |
| 282, |
| 0 |
| ], |
| [ |
| 54, |
| 22 |
| ] |
| ], |
| "eval_f1": 0.449, |
| "eval_loss": 0.41911664605140686, |
| "eval_pr_auc": 0.6187, |
| "eval_precision": 1.0, |
| "eval_recall": 0.2895, |
| "eval_roc_auc": 0.7485, |
| "eval_runtime": 138.7598, |
| "eval_samples_per_second": 2.58, |
| "eval_steps_per_second": 0.649, |
| "step": 366 |
| }, |
| { |
| "epoch": 1.010928961748634, |
| "grad_norm": 6.519931316375732, |
| "learning_rate": 2.9972677595628417e-05, |
| "loss": 0.796769380569458, |
| "step": 370 |
| }, |
| { |
| "epoch": 1.0382513661202186, |
| "grad_norm": 4.874457359313965, |
| "learning_rate": 2.98816029143898e-05, |
| "loss": 0.398636531829834, |
| "step": 380 |
| }, |
| { |
| "epoch": 1.0655737704918034, |
| "grad_norm": 15.488590240478516, |
| "learning_rate": 2.9790528233151184e-05, |
| "loss": 0.7061968803405761, |
| "step": 390 |
| }, |
| { |
| "epoch": 1.092896174863388, |
| "grad_norm": 19.74693489074707, |
| "learning_rate": 2.9699453551912568e-05, |
| "loss": 0.5211454391479492, |
| "step": 400 |
| }, |
| { |
| "epoch": 1.1202185792349726, |
| "grad_norm": 2.513188362121582, |
| "learning_rate": 2.9608378870673954e-05, |
| "loss": 0.7898036003112793, |
| "step": 410 |
| }, |
| { |
| "epoch": 1.1475409836065573, |
| "grad_norm": 25.31527328491211, |
| "learning_rate": 2.9517304189435338e-05, |
| "loss": 0.8195189476013184, |
| "step": 420 |
| }, |
| { |
| "epoch": 1.174863387978142, |
| "grad_norm": 26.276199340820312, |
| "learning_rate": 2.942622950819672e-05, |
| "loss": 0.8042145729064941, |
| "step": 430 |
| }, |
| { |
| "epoch": 1.2021857923497268, |
| "grad_norm": 19.838821411132812, |
| "learning_rate": 2.9335154826958105e-05, |
| "loss": 0.5532180786132812, |
| "step": 440 |
| }, |
| { |
| "epoch": 1.2295081967213115, |
| "grad_norm": 1.5704028606414795, |
| "learning_rate": 2.924408014571949e-05, |
| "loss": 0.46767144203186034, |
| "step": 450 |
| }, |
| { |
| "epoch": 1.2568306010928962, |
| "grad_norm": 0.24576333165168762, |
| "learning_rate": 2.9153005464480875e-05, |
| "loss": 0.7125775337219238, |
| "step": 460 |
| }, |
| { |
| "epoch": 1.2841530054644807, |
| "grad_norm": 2.8051435947418213, |
| "learning_rate": 2.9061930783242258e-05, |
| "loss": 1.4706084251403808, |
| "step": 470 |
| }, |
| { |
| "epoch": 1.3114754098360657, |
| "grad_norm": 2.1515614986419678, |
| "learning_rate": 2.897085610200364e-05, |
| "loss": 0.740934181213379, |
| "step": 480 |
| }, |
| { |
| "epoch": 1.3387978142076502, |
| "grad_norm": 5.335806846618652, |
| "learning_rate": 2.8879781420765028e-05, |
| "loss": 0.5567354679107666, |
| "step": 490 |
| }, |
| { |
| "epoch": 1.366120218579235, |
| "grad_norm": 9.499934196472168, |
| "learning_rate": 2.878870673952641e-05, |
| "loss": 0.5102160930633545, |
| "step": 500 |
| }, |
| { |
| "epoch": 1.3934426229508197, |
| "grad_norm": 0.30213791131973267, |
| "learning_rate": 2.8697632058287795e-05, |
| "loss": 0.8127809524536133, |
| "step": 510 |
| }, |
| { |
| "epoch": 1.4207650273224044, |
| "grad_norm": 17.827165603637695, |
| "learning_rate": 2.860655737704918e-05, |
| "loss": 1.3400749206542968, |
| "step": 520 |
| }, |
| { |
| "epoch": 1.4480874316939891, |
| "grad_norm": 2.58740234375, |
| "learning_rate": 2.8515482695810565e-05, |
| "loss": 0.5003597736358643, |
| "step": 530 |
| }, |
| { |
| "epoch": 1.4754098360655736, |
| "grad_norm": 22.007869720458984, |
| "learning_rate": 2.842440801457195e-05, |
| "loss": 1.6859323501586914, |
| "step": 540 |
| }, |
| { |
| "epoch": 1.5027322404371586, |
| "grad_norm": 14.658230781555176, |
| "learning_rate": 2.8333333333333332e-05, |
| "loss": 0.9195033073425293, |
| "step": 550 |
| }, |
| { |
| "epoch": 1.530054644808743, |
| "grad_norm": 1.4521483182907104, |
| "learning_rate": 2.8242258652094715e-05, |
| "loss": 0.4108741283416748, |
| "step": 560 |
| }, |
| { |
| "epoch": 1.5573770491803278, |
| "grad_norm": 10.377533912658691, |
| "learning_rate": 2.8151183970856102e-05, |
| "loss": 0.655418586730957, |
| "step": 570 |
| }, |
| { |
| "epoch": 1.5846994535519126, |
| "grad_norm": 0.5859184265136719, |
| "learning_rate": 2.8060109289617486e-05, |
| "loss": 1.2600435256958007, |
| "step": 580 |
| }, |
| { |
| "epoch": 1.6120218579234973, |
| "grad_norm": 0.8876230120658875, |
| "learning_rate": 2.796903460837887e-05, |
| "loss": 1.0982278823852538, |
| "step": 590 |
| }, |
| { |
| "epoch": 1.639344262295082, |
| "grad_norm": 13.285502433776855, |
| "learning_rate": 2.7877959927140256e-05, |
| "loss": 0.6685476779937745, |
| "step": 600 |
| }, |
| { |
| "epoch": 1.6666666666666665, |
| "grad_norm": 0.6555774211883545, |
| "learning_rate": 2.7786885245901642e-05, |
| "loss": 0.19812407493591308, |
| "step": 610 |
| }, |
| { |
| "epoch": 1.6939890710382515, |
| "grad_norm": 2.7470834255218506, |
| "learning_rate": 2.7695810564663026e-05, |
| "loss": 1.1334525108337403, |
| "step": 620 |
| }, |
| { |
| "epoch": 1.721311475409836, |
| "grad_norm": 14.42418098449707, |
| "learning_rate": 2.760473588342441e-05, |
| "loss": 0.9498401641845703, |
| "step": 630 |
| }, |
| { |
| "epoch": 1.748633879781421, |
| "grad_norm": 57.35378646850586, |
| "learning_rate": 2.7513661202185793e-05, |
| "loss": 1.2678983688354493, |
| "step": 640 |
| }, |
| { |
| "epoch": 1.7759562841530054, |
| "grad_norm": 0.7189310789108276, |
| "learning_rate": 2.742258652094718e-05, |
| "loss": 0.6106186389923096, |
| "step": 650 |
| }, |
| { |
| "epoch": 1.8032786885245902, |
| "grad_norm": 13.546385765075684, |
| "learning_rate": 2.7331511839708563e-05, |
| "loss": 1.4015119552612305, |
| "step": 660 |
| }, |
| { |
| "epoch": 1.830601092896175, |
| "grad_norm": 8.754522323608398, |
| "learning_rate": 2.7240437158469946e-05, |
| "loss": 0.9790468215942383, |
| "step": 670 |
| }, |
| { |
| "epoch": 1.8579234972677594, |
| "grad_norm": 13.214091300964355, |
| "learning_rate": 2.7149362477231333e-05, |
| "loss": 0.6752419948577881, |
| "step": 680 |
| }, |
| { |
| "epoch": 1.8852459016393444, |
| "grad_norm": 2.506131887435913, |
| "learning_rate": 2.7058287795992716e-05, |
| "loss": 0.2787914752960205, |
| "step": 690 |
| }, |
| { |
| "epoch": 1.9125683060109289, |
| "grad_norm": 0.4683574438095093, |
| "learning_rate": 2.69672131147541e-05, |
| "loss": 0.8134347915649414, |
| "step": 700 |
| }, |
| { |
| "epoch": 1.9398907103825138, |
| "grad_norm": 0.8958243131637573, |
| "learning_rate": 2.6876138433515483e-05, |
| "loss": 0.4489922046661377, |
| "step": 710 |
| }, |
| { |
| "epoch": 1.9672131147540983, |
| "grad_norm": 95.41824340820312, |
| "learning_rate": 2.678506375227687e-05, |
| "loss": 0.9731103897094726, |
| "step": 720 |
| }, |
| { |
| "epoch": 1.994535519125683, |
| "grad_norm": 0.7653670907020569, |
| "learning_rate": 2.6693989071038253e-05, |
| "loss": 1.0658687591552733, |
| "step": 730 |
| }, |
| { |
| "epoch": 2.0, |
| "eval_accuracy": 0.8659, |
| "eval_confusion_matrix": [ |
| [ |
| 281, |
| 1 |
| ], |
| [ |
| 47, |
| 29 |
| ] |
| ], |
| "eval_f1": 0.5472, |
| "eval_loss": 0.45224741101264954, |
| "eval_pr_auc": 0.6674, |
| "eval_precision": 0.9667, |
| "eval_recall": 0.3816, |
| "eval_roc_auc": 0.8086, |
| "eval_runtime": 137.9939, |
| "eval_samples_per_second": 2.594, |
| "eval_steps_per_second": 0.652, |
| "step": 732 |
| }, |
| { |
| "epoch": 2.021857923497268, |
| "grad_norm": 32.09031677246094, |
| "learning_rate": 2.6602914389799637e-05, |
| "loss": 0.7306198120117188, |
| "step": 740 |
| }, |
| { |
| "epoch": 2.0491803278688523, |
| "grad_norm": 51.15188217163086, |
| "learning_rate": 2.651183970856102e-05, |
| "loss": 0.49246606826782224, |
| "step": 750 |
| }, |
| { |
| "epoch": 2.0765027322404372, |
| "grad_norm": 0.5293377041816711, |
| "learning_rate": 2.6420765027322407e-05, |
| "loss": 0.30229816436767576, |
| "step": 760 |
| }, |
| { |
| "epoch": 2.1038251366120218, |
| "grad_norm": 0.5600886344909668, |
| "learning_rate": 2.632969034608379e-05, |
| "loss": 1.428311824798584, |
| "step": 770 |
| }, |
| { |
| "epoch": 2.1311475409836067, |
| "grad_norm": 8.111888885498047, |
| "learning_rate": 2.6238615664845174e-05, |
| "loss": 0.7316344738006592, |
| "step": 780 |
| }, |
| { |
| "epoch": 2.158469945355191, |
| "grad_norm": 9.91689682006836, |
| "learning_rate": 2.6147540983606557e-05, |
| "loss": 0.7728833198547364, |
| "step": 790 |
| }, |
| { |
| "epoch": 2.185792349726776, |
| "grad_norm": 24.687915802001953, |
| "learning_rate": 2.6056466302367944e-05, |
| "loss": 0.6872700691223145, |
| "step": 800 |
| }, |
| { |
| "epoch": 2.2131147540983607, |
| "grad_norm": 15.530792236328125, |
| "learning_rate": 2.5965391621129327e-05, |
| "loss": 0.8627532005310059, |
| "step": 810 |
| }, |
| { |
| "epoch": 2.240437158469945, |
| "grad_norm": 0.7304887175559998, |
| "learning_rate": 2.587431693989071e-05, |
| "loss": 0.5328950405120849, |
| "step": 820 |
| }, |
| { |
| "epoch": 2.26775956284153, |
| "grad_norm": 52.83394241333008, |
| "learning_rate": 2.5783242258652094e-05, |
| "loss": 0.45772466659545896, |
| "step": 830 |
| }, |
| { |
| "epoch": 2.2950819672131146, |
| "grad_norm": 23.795230865478516, |
| "learning_rate": 2.569216757741348e-05, |
| "loss": 0.6016922950744629, |
| "step": 840 |
| }, |
| { |
| "epoch": 2.3224043715846996, |
| "grad_norm": 22.11771011352539, |
| "learning_rate": 2.5601092896174864e-05, |
| "loss": 0.5151076316833496, |
| "step": 850 |
| }, |
| { |
| "epoch": 2.349726775956284, |
| "grad_norm": 0.44267675280570984, |
| "learning_rate": 2.5510018214936247e-05, |
| "loss": 0.9358211517333984, |
| "step": 860 |
| }, |
| { |
| "epoch": 2.3770491803278686, |
| "grad_norm": 0.3188877999782562, |
| "learning_rate": 2.541894353369763e-05, |
| "loss": 0.46982393264770506, |
| "step": 870 |
| }, |
| { |
| "epoch": 2.4043715846994536, |
| "grad_norm": 0.409311980009079, |
| "learning_rate": 2.5327868852459018e-05, |
| "loss": 0.5571242332458496, |
| "step": 880 |
| }, |
| { |
| "epoch": 2.431693989071038, |
| "grad_norm": 0.22160018980503082, |
| "learning_rate": 2.52367941712204e-05, |
| "loss": 0.686149263381958, |
| "step": 890 |
| }, |
| { |
| "epoch": 2.459016393442623, |
| "grad_norm": 56.0511360168457, |
| "learning_rate": 2.5145719489981784e-05, |
| "loss": 0.4332759857177734, |
| "step": 900 |
| }, |
| { |
| "epoch": 2.4863387978142075, |
| "grad_norm": 14.211305618286133, |
| "learning_rate": 2.5054644808743168e-05, |
| "loss": 0.46988306045532224, |
| "step": 910 |
| }, |
| { |
| "epoch": 2.5136612021857925, |
| "grad_norm": 0.394004762172699, |
| "learning_rate": 2.4963570127504555e-05, |
| "loss": 0.3961897611618042, |
| "step": 920 |
| }, |
| { |
| "epoch": 2.540983606557377, |
| "grad_norm": 56.96977233886719, |
| "learning_rate": 2.4872495446265938e-05, |
| "loss": 1.283346939086914, |
| "step": 930 |
| }, |
| { |
| "epoch": 2.5683060109289615, |
| "grad_norm": 24.957056045532227, |
| "learning_rate": 2.478142076502732e-05, |
| "loss": 0.8271417617797852, |
| "step": 940 |
| }, |
| { |
| "epoch": 2.5956284153005464, |
| "grad_norm": 1.057257056236267, |
| "learning_rate": 2.4690346083788705e-05, |
| "loss": 1.5506049156188966, |
| "step": 950 |
| }, |
| { |
| "epoch": 2.6229508196721314, |
| "grad_norm": 2.8221116065979004, |
| "learning_rate": 2.459927140255009e-05, |
| "loss": 0.5170619487762451, |
| "step": 960 |
| }, |
| { |
| "epoch": 2.650273224043716, |
| "grad_norm": 9.166945457458496, |
| "learning_rate": 2.4508196721311478e-05, |
| "loss": 1.0213055610656738, |
| "step": 970 |
| }, |
| { |
| "epoch": 2.6775956284153004, |
| "grad_norm": 13.581059455871582, |
| "learning_rate": 2.441712204007286e-05, |
| "loss": 0.604024600982666, |
| "step": 980 |
| }, |
| { |
| "epoch": 2.7049180327868854, |
| "grad_norm": 0.4070906639099121, |
| "learning_rate": 2.4326047358834245e-05, |
| "loss": 0.7995193481445313, |
| "step": 990 |
| }, |
| { |
| "epoch": 2.73224043715847, |
| "grad_norm": 0.3183801472187042, |
| "learning_rate": 2.4234972677595632e-05, |
| "loss": 0.7718863487243652, |
| "step": 1000 |
| }, |
| { |
| "epoch": 2.7595628415300544, |
| "grad_norm": 15.769966125488281, |
| "learning_rate": 2.4143897996357015e-05, |
| "loss": 0.7727096080780029, |
| "step": 1010 |
| }, |
| { |
| "epoch": 2.7868852459016393, |
| "grad_norm": 13.82215690612793, |
| "learning_rate": 2.40528233151184e-05, |
| "loss": 0.6281228542327881, |
| "step": 1020 |
| }, |
| { |
| "epoch": 2.8142076502732243, |
| "grad_norm": 0.24273018538951874, |
| "learning_rate": 2.3961748633879782e-05, |
| "loss": 0.7842752933502197, |
| "step": 1030 |
| }, |
| { |
| "epoch": 2.841530054644809, |
| "grad_norm": 0.21391047537326813, |
| "learning_rate": 2.387067395264117e-05, |
| "loss": 0.8222448348999023, |
| "step": 1040 |
| }, |
| { |
| "epoch": 2.8688524590163933, |
| "grad_norm": 0.5637283325195312, |
| "learning_rate": 2.3779599271402552e-05, |
| "loss": 1.0477510452270509, |
| "step": 1050 |
| }, |
| { |
| "epoch": 2.8961748633879782, |
| "grad_norm": 0.2845185101032257, |
| "learning_rate": 2.3688524590163936e-05, |
| "loss": 0.6434628009796143, |
| "step": 1060 |
| }, |
| { |
| "epoch": 2.9234972677595628, |
| "grad_norm": 1.1965957880020142, |
| "learning_rate": 2.359744990892532e-05, |
| "loss": 0.47115397453308105, |
| "step": 1070 |
| }, |
| { |
| "epoch": 2.9508196721311473, |
| "grad_norm": 0.9190018773078918, |
| "learning_rate": 2.3506375227686706e-05, |
| "loss": 0.7713288784027099, |
| "step": 1080 |
| }, |
| { |
| "epoch": 2.978142076502732, |
| "grad_norm": 1.0127384662628174, |
| "learning_rate": 2.341530054644809e-05, |
| "loss": 0.6314740657806397, |
| "step": 1090 |
| }, |
| { |
| "epoch": 3.0, |
| "eval_accuracy": 0.8547, |
| "eval_confusion_matrix": [ |
| [ |
| 277, |
| 5 |
| ], |
| [ |
| 47, |
| 29 |
| ] |
| ], |
| "eval_f1": 0.5273, |
| "eval_loss": 0.498869389295578, |
| "eval_pr_auc": 0.6438, |
| "eval_precision": 0.8529, |
| "eval_recall": 0.3816, |
| "eval_roc_auc": 0.7867, |
| "eval_runtime": 137.6925, |
| "eval_samples_per_second": 2.6, |
| "eval_steps_per_second": 0.654, |
| "step": 1098 |
| } |
| ], |
| "logging_steps": 10, |
| "max_steps": 3660, |
| "num_input_tokens_seen": 0, |
| "num_train_epochs": 10, |
| "save_steps": 500, |
| "stateful_callbacks": { |
| "EarlyStoppingCallback": { |
| "args": { |
| "early_stopping_patience": 3, |
| "early_stopping_threshold": 0.0 |
| }, |
| "attributes": { |
| "early_stopping_patience_counter": 2 |
| } |
| }, |
| "TrainerControl": { |
| "args": { |
| "should_epoch_stop": false, |
| "should_evaluate": false, |
| "should_log": false, |
| "should_save": true, |
| "should_training_stop": false |
| }, |
| "attributes": {} |
| } |
| }, |
| "total_flos": 1155583755141120.0, |
| "train_batch_size": 4, |
| "trial_name": null, |
| "trial_params": null |
| } |
|
|
