| { | |
| "best_global_step": 2000, | |
| "best_metric": 1.0576004802313843, | |
| "best_model_checkpoint": "./SALAMA_NEWMED7/checkpoint-2000", | |
| "epoch": 0.7990012484394506, | |
| "eval_steps": 2000, | |
| "global_step": 2000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.003995006242197253, | |
| "grad_norm": 1.3738492727279663, | |
| "learning_rate": 1.8e-07, | |
| "loss": 0.0168, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.007990012484394507, | |
| "grad_norm": 0.9679616093635559, | |
| "learning_rate": 3.8e-07, | |
| "loss": 0.0139, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.01198501872659176, | |
| "grad_norm": 2.5446205139160156, | |
| "learning_rate": 5.800000000000001e-07, | |
| "loss": 0.0127, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.015980024968789013, | |
| "grad_norm": 1.0809508562088013, | |
| "learning_rate": 7.8e-07, | |
| "loss": 0.0083, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.019975031210986267, | |
| "grad_norm": 0.6600648164749146, | |
| "learning_rate": 9.800000000000001e-07, | |
| "loss": 0.0093, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.02397003745318352, | |
| "grad_norm": 2.1248271465301514, | |
| "learning_rate": 1.1800000000000001e-06, | |
| "loss": 0.0126, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.027965043695380776, | |
| "grad_norm": 1.3413838148117065, | |
| "learning_rate": 1.3800000000000001e-06, | |
| "loss": 0.0155, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.031960049937578026, | |
| "grad_norm": 1.3642956018447876, | |
| "learning_rate": 1.5800000000000001e-06, | |
| "loss": 0.0161, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.035955056179775284, | |
| "grad_norm": 0.761688232421875, | |
| "learning_rate": 1.7800000000000001e-06, | |
| "loss": 0.009, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.039950062421972535, | |
| "grad_norm": 1.131451964378357, | |
| "learning_rate": 1.98e-06, | |
| "loss": 0.0128, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.043945068664169785, | |
| "grad_norm": 0.8744670152664185, | |
| "learning_rate": 2.1800000000000003e-06, | |
| "loss": 0.0147, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.04794007490636704, | |
| "grad_norm": 0.42453721165657043, | |
| "learning_rate": 2.38e-06, | |
| "loss": 0.0094, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.051935081148564294, | |
| "grad_norm": 1.0128003358840942, | |
| "learning_rate": 2.5800000000000003e-06, | |
| "loss": 0.0157, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.05593008739076155, | |
| "grad_norm": 0.7458897829055786, | |
| "learning_rate": 2.7800000000000005e-06, | |
| "loss": 0.0111, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.0599250936329588, | |
| "grad_norm": 0.5461632013320923, | |
| "learning_rate": 2.9800000000000003e-06, | |
| "loss": 0.0144, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.06392009987515605, | |
| "grad_norm": 0.7912344932556152, | |
| "learning_rate": 3.1800000000000005e-06, | |
| "loss": 0.0179, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.06791510611735331, | |
| "grad_norm": 1.0206142663955688, | |
| "learning_rate": 3.3800000000000007e-06, | |
| "loss": 0.0173, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.07191011235955057, | |
| "grad_norm": 1.172113299369812, | |
| "learning_rate": 3.58e-06, | |
| "loss": 0.0075, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.07590511860174781, | |
| "grad_norm": 1.5526639223098755, | |
| "learning_rate": 3.7800000000000002e-06, | |
| "loss": 0.0123, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.07990012484394507, | |
| "grad_norm": 1.708471655845642, | |
| "learning_rate": 3.980000000000001e-06, | |
| "loss": 0.0158, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.08389513108614233, | |
| "grad_norm": 1.0267976522445679, | |
| "learning_rate": 4.18e-06, | |
| "loss": 0.0139, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.08789013732833957, | |
| "grad_norm": 1.3162474632263184, | |
| "learning_rate": 4.38e-06, | |
| "loss": 0.011, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.09188514357053683, | |
| "grad_norm": 1.0626009702682495, | |
| "learning_rate": 4.58e-06, | |
| "loss": 0.0121, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.09588014981273409, | |
| "grad_norm": 1.7782543897628784, | |
| "learning_rate": 4.78e-06, | |
| "loss": 0.0137, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.09987515605493133, | |
| "grad_norm": 1.5565178394317627, | |
| "learning_rate": 4.980000000000001e-06, | |
| "loss": 0.0216, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.10387016229712859, | |
| "grad_norm": 2.0086753368377686, | |
| "learning_rate": 5.18e-06, | |
| "loss": 0.0153, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.10786516853932585, | |
| "grad_norm": 2.0375168323516846, | |
| "learning_rate": 5.380000000000001e-06, | |
| "loss": 0.0162, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.1118601747815231, | |
| "grad_norm": 1.385829210281372, | |
| "learning_rate": 5.580000000000001e-06, | |
| "loss": 0.0136, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.11585518102372035, | |
| "grad_norm": 1.4296222925186157, | |
| "learning_rate": 5.78e-06, | |
| "loss": 0.0198, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.1198501872659176, | |
| "grad_norm": 0.8523766398429871, | |
| "learning_rate": 5.98e-06, | |
| "loss": 0.014, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.12384519350811486, | |
| "grad_norm": 0.915155291557312, | |
| "learning_rate": 6.18e-06, | |
| "loss": 0.0193, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 0.1278401997503121, | |
| "grad_norm": 2.0550453662872314, | |
| "learning_rate": 6.380000000000001e-06, | |
| "loss": 0.0148, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 0.13183520599250936, | |
| "grad_norm": 1.2964582443237305, | |
| "learning_rate": 6.5800000000000005e-06, | |
| "loss": 0.0178, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 0.13583021223470662, | |
| "grad_norm": 0.7987473011016846, | |
| "learning_rate": 6.780000000000001e-06, | |
| "loss": 0.0132, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 0.13982521847690388, | |
| "grad_norm": 0.7583047151565552, | |
| "learning_rate": 6.98e-06, | |
| "loss": 0.0154, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.14382022471910114, | |
| "grad_norm": 0.9069733619689941, | |
| "learning_rate": 7.180000000000001e-06, | |
| "loss": 0.015, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 0.14781523096129837, | |
| "grad_norm": 2.4443578720092773, | |
| "learning_rate": 7.3800000000000005e-06, | |
| "loss": 0.0158, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 0.15181023720349562, | |
| "grad_norm": 2.005502700805664, | |
| "learning_rate": 7.58e-06, | |
| "loss": 0.0172, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 0.15580524344569288, | |
| "grad_norm": 0.6428951621055603, | |
| "learning_rate": 7.78e-06, | |
| "loss": 0.0125, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 0.15980024968789014, | |
| "grad_norm": 1.6475672721862793, | |
| "learning_rate": 7.980000000000002e-06, | |
| "loss": 0.0251, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.1637952559300874, | |
| "grad_norm": 0.7950750589370728, | |
| "learning_rate": 8.18e-06, | |
| "loss": 0.0114, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 0.16779026217228465, | |
| "grad_norm": 1.9027169942855835, | |
| "learning_rate": 8.380000000000001e-06, | |
| "loss": 0.0185, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 0.17178526841448188, | |
| "grad_norm": 1.4045602083206177, | |
| "learning_rate": 8.580000000000001e-06, | |
| "loss": 0.0202, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 0.17578027465667914, | |
| "grad_norm": 1.5001755952835083, | |
| "learning_rate": 8.78e-06, | |
| "loss": 0.0157, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 0.1797752808988764, | |
| "grad_norm": 1.9885104894638062, | |
| "learning_rate": 8.98e-06, | |
| "loss": 0.0198, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 0.18377028714107366, | |
| "grad_norm": 0.8559455871582031, | |
| "learning_rate": 9.180000000000002e-06, | |
| "loss": 0.0167, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 0.18776529338327091, | |
| "grad_norm": 1.4045313596725464, | |
| "learning_rate": 9.38e-06, | |
| "loss": 0.0117, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 0.19176029962546817, | |
| "grad_norm": 1.5280107259750366, | |
| "learning_rate": 9.58e-06, | |
| "loss": 0.016, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 0.19575530586766543, | |
| "grad_norm": 2.3487863540649414, | |
| "learning_rate": 9.780000000000001e-06, | |
| "loss": 0.0201, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 0.19975031210986266, | |
| "grad_norm": 1.2929073572158813, | |
| "learning_rate": 9.980000000000001e-06, | |
| "loss": 0.02, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.20374531835205992, | |
| "grad_norm": 2.014071226119995, | |
| "learning_rate": 9.980035492457854e-06, | |
| "loss": 0.0242, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 0.20774032459425718, | |
| "grad_norm": 1.8505971431732178, | |
| "learning_rate": 9.957852706299913e-06, | |
| "loss": 0.0144, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 0.21173533083645443, | |
| "grad_norm": 2.4067506790161133, | |
| "learning_rate": 9.935669920141971e-06, | |
| "loss": 0.0164, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 0.2157303370786517, | |
| "grad_norm": 1.4055867195129395, | |
| "learning_rate": 9.91348713398403e-06, | |
| "loss": 0.0159, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 0.21972534332084895, | |
| "grad_norm": 1.937062382698059, | |
| "learning_rate": 9.891304347826088e-06, | |
| "loss": 0.0187, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 0.2237203495630462, | |
| "grad_norm": 2.5414481163024902, | |
| "learning_rate": 9.869121561668146e-06, | |
| "loss": 0.0212, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 0.22771535580524344, | |
| "grad_norm": 3.127715826034546, | |
| "learning_rate": 9.846938775510205e-06, | |
| "loss": 0.0341, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 0.2317103620474407, | |
| "grad_norm": 2.196579933166504, | |
| "learning_rate": 9.824755989352263e-06, | |
| "loss": 0.0367, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 0.23570536828963795, | |
| "grad_norm": 2.0753867626190186, | |
| "learning_rate": 9.802573203194321e-06, | |
| "loss": 0.0203, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 0.2397003745318352, | |
| "grad_norm": 2.2783772945404053, | |
| "learning_rate": 9.78039041703638e-06, | |
| "loss": 0.0233, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.24369538077403247, | |
| "grad_norm": 1.8765735626220703, | |
| "learning_rate": 9.758207630878438e-06, | |
| "loss": 0.0301, | |
| "step": 610 | |
| }, | |
| { | |
| "epoch": 0.24769038701622972, | |
| "grad_norm": 2.1278908252716064, | |
| "learning_rate": 9.736024844720497e-06, | |
| "loss": 0.0194, | |
| "step": 620 | |
| }, | |
| { | |
| "epoch": 0.251685393258427, | |
| "grad_norm": 1.0078867673873901, | |
| "learning_rate": 9.713842058562555e-06, | |
| "loss": 0.019, | |
| "step": 630 | |
| }, | |
| { | |
| "epoch": 0.2556803995006242, | |
| "grad_norm": 1.9328465461730957, | |
| "learning_rate": 9.691659272404614e-06, | |
| "loss": 0.0263, | |
| "step": 640 | |
| }, | |
| { | |
| "epoch": 0.2596754057428215, | |
| "grad_norm": 1.519873857498169, | |
| "learning_rate": 9.669476486246674e-06, | |
| "loss": 0.0245, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 0.2636704119850187, | |
| "grad_norm": 2.022794485092163, | |
| "learning_rate": 9.647293700088732e-06, | |
| "loss": 0.0276, | |
| "step": 660 | |
| }, | |
| { | |
| "epoch": 0.26766541822721596, | |
| "grad_norm": 1.337441086769104, | |
| "learning_rate": 9.62511091393079e-06, | |
| "loss": 0.0269, | |
| "step": 670 | |
| }, | |
| { | |
| "epoch": 0.27166042446941324, | |
| "grad_norm": 1.5506941080093384, | |
| "learning_rate": 9.602928127772849e-06, | |
| "loss": 0.0266, | |
| "step": 680 | |
| }, | |
| { | |
| "epoch": 0.27565543071161047, | |
| "grad_norm": 1.447799801826477, | |
| "learning_rate": 9.580745341614907e-06, | |
| "loss": 0.0134, | |
| "step": 690 | |
| }, | |
| { | |
| "epoch": 0.27965043695380776, | |
| "grad_norm": 1.189295768737793, | |
| "learning_rate": 9.558562555456966e-06, | |
| "loss": 0.027, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.283645443196005, | |
| "grad_norm": 1.873569369316101, | |
| "learning_rate": 9.536379769299024e-06, | |
| "loss": 0.0276, | |
| "step": 710 | |
| }, | |
| { | |
| "epoch": 0.2876404494382023, | |
| "grad_norm": 1.3089381456375122, | |
| "learning_rate": 9.514196983141083e-06, | |
| "loss": 0.0169, | |
| "step": 720 | |
| }, | |
| { | |
| "epoch": 0.2916354556803995, | |
| "grad_norm": 1.7334715127944946, | |
| "learning_rate": 9.492014196983141e-06, | |
| "loss": 0.0227, | |
| "step": 730 | |
| }, | |
| { | |
| "epoch": 0.29563046192259673, | |
| "grad_norm": 3.0353081226348877, | |
| "learning_rate": 9.469831410825201e-06, | |
| "loss": 0.0237, | |
| "step": 740 | |
| }, | |
| { | |
| "epoch": 0.299625468164794, | |
| "grad_norm": 2.1285839080810547, | |
| "learning_rate": 9.44764862466726e-06, | |
| "loss": 0.0248, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 0.30362047440699125, | |
| "grad_norm": 3.155933380126953, | |
| "learning_rate": 9.425465838509318e-06, | |
| "loss": 0.0228, | |
| "step": 760 | |
| }, | |
| { | |
| "epoch": 0.30761548064918853, | |
| "grad_norm": 2.1449387073516846, | |
| "learning_rate": 9.403283052351377e-06, | |
| "loss": 0.0246, | |
| "step": 770 | |
| }, | |
| { | |
| "epoch": 0.31161048689138576, | |
| "grad_norm": 2.538461685180664, | |
| "learning_rate": 9.381100266193435e-06, | |
| "loss": 0.032, | |
| "step": 780 | |
| }, | |
| { | |
| "epoch": 0.31560549313358305, | |
| "grad_norm": 1.5362180471420288, | |
| "learning_rate": 9.358917480035493e-06, | |
| "loss": 0.0208, | |
| "step": 790 | |
| }, | |
| { | |
| "epoch": 0.3196004993757803, | |
| "grad_norm": 1.6396677494049072, | |
| "learning_rate": 9.336734693877552e-06, | |
| "loss": 0.0262, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 0.3235955056179775, | |
| "grad_norm": 2.5148026943206787, | |
| "learning_rate": 9.31455190771961e-06, | |
| "loss": 0.0235, | |
| "step": 810 | |
| }, | |
| { | |
| "epoch": 0.3275905118601748, | |
| "grad_norm": 1.766427755355835, | |
| "learning_rate": 9.292369121561669e-06, | |
| "loss": 0.0217, | |
| "step": 820 | |
| }, | |
| { | |
| "epoch": 0.331585518102372, | |
| "grad_norm": 1.905369520187378, | |
| "learning_rate": 9.270186335403727e-06, | |
| "loss": 0.0271, | |
| "step": 830 | |
| }, | |
| { | |
| "epoch": 0.3355805243445693, | |
| "grad_norm": 2.183584690093994, | |
| "learning_rate": 9.248003549245787e-06, | |
| "loss": 0.0298, | |
| "step": 840 | |
| }, | |
| { | |
| "epoch": 0.33957553058676654, | |
| "grad_norm": 1.4525411128997803, | |
| "learning_rate": 9.225820763087846e-06, | |
| "loss": 0.019, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 0.34357053682896377, | |
| "grad_norm": 2.1158807277679443, | |
| "learning_rate": 9.203637976929904e-06, | |
| "loss": 0.0213, | |
| "step": 860 | |
| }, | |
| { | |
| "epoch": 0.34756554307116105, | |
| "grad_norm": 1.1012670993804932, | |
| "learning_rate": 9.181455190771963e-06, | |
| "loss": 0.034, | |
| "step": 870 | |
| }, | |
| { | |
| "epoch": 0.3515605493133583, | |
| "grad_norm": 1.8962483406066895, | |
| "learning_rate": 9.159272404614021e-06, | |
| "loss": 0.02, | |
| "step": 880 | |
| }, | |
| { | |
| "epoch": 0.35555555555555557, | |
| "grad_norm": 1.8815522193908691, | |
| "learning_rate": 9.13708961845608e-06, | |
| "loss": 0.0274, | |
| "step": 890 | |
| }, | |
| { | |
| "epoch": 0.3595505617977528, | |
| "grad_norm": 2.9523744583129883, | |
| "learning_rate": 9.114906832298138e-06, | |
| "loss": 0.0267, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 0.3635455680399501, | |
| "grad_norm": 1.1391562223434448, | |
| "learning_rate": 9.092724046140196e-06, | |
| "loss": 0.0199, | |
| "step": 910 | |
| }, | |
| { | |
| "epoch": 0.3675405742821473, | |
| "grad_norm": 1.0723015069961548, | |
| "learning_rate": 9.070541259982255e-06, | |
| "loss": 0.0197, | |
| "step": 920 | |
| }, | |
| { | |
| "epoch": 0.37153558052434454, | |
| "grad_norm": 1.778450608253479, | |
| "learning_rate": 9.048358473824313e-06, | |
| "loss": 0.0175, | |
| "step": 930 | |
| }, | |
| { | |
| "epoch": 0.37553058676654183, | |
| "grad_norm": 1.4739247560501099, | |
| "learning_rate": 9.026175687666371e-06, | |
| "loss": 0.023, | |
| "step": 940 | |
| }, | |
| { | |
| "epoch": 0.37952559300873906, | |
| "grad_norm": 2.744870185852051, | |
| "learning_rate": 9.00399290150843e-06, | |
| "loss": 0.0278, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 0.38352059925093634, | |
| "grad_norm": 1.8237658739089966, | |
| "learning_rate": 8.981810115350488e-06, | |
| "loss": 0.016, | |
| "step": 960 | |
| }, | |
| { | |
| "epoch": 0.3875156054931336, | |
| "grad_norm": 1.86380136013031, | |
| "learning_rate": 8.959627329192547e-06, | |
| "loss": 0.0206, | |
| "step": 970 | |
| }, | |
| { | |
| "epoch": 0.39151061173533086, | |
| "grad_norm": 1.941651701927185, | |
| "learning_rate": 8.937444543034605e-06, | |
| "loss": 0.02, | |
| "step": 980 | |
| }, | |
| { | |
| "epoch": 0.3955056179775281, | |
| "grad_norm": 1.7950303554534912, | |
| "learning_rate": 8.915261756876664e-06, | |
| "loss": 0.0198, | |
| "step": 990 | |
| }, | |
| { | |
| "epoch": 0.3995006242197253, | |
| "grad_norm": 1.2369314432144165, | |
| "learning_rate": 8.893078970718722e-06, | |
| "loss": 0.0213, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.4034956304619226, | |
| "grad_norm": 1.7613532543182373, | |
| "learning_rate": 8.87089618456078e-06, | |
| "loss": 0.0225, | |
| "step": 1010 | |
| }, | |
| { | |
| "epoch": 0.40749063670411984, | |
| "grad_norm": 1.2709797620773315, | |
| "learning_rate": 8.84871339840284e-06, | |
| "loss": 0.0204, | |
| "step": 1020 | |
| }, | |
| { | |
| "epoch": 0.4114856429463171, | |
| "grad_norm": 2.4285876750946045, | |
| "learning_rate": 8.826530612244899e-06, | |
| "loss": 0.0256, | |
| "step": 1030 | |
| }, | |
| { | |
| "epoch": 0.41548064918851435, | |
| "grad_norm": 1.5237114429473877, | |
| "learning_rate": 8.804347826086957e-06, | |
| "loss": 0.0228, | |
| "step": 1040 | |
| }, | |
| { | |
| "epoch": 0.41947565543071164, | |
| "grad_norm": 1.1931017637252808, | |
| "learning_rate": 8.782165039929016e-06, | |
| "loss": 0.0155, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 0.42347066167290887, | |
| "grad_norm": 2.397512912750244, | |
| "learning_rate": 8.759982253771074e-06, | |
| "loss": 0.0347, | |
| "step": 1060 | |
| }, | |
| { | |
| "epoch": 0.4274656679151061, | |
| "grad_norm": 1.4002119302749634, | |
| "learning_rate": 8.737799467613133e-06, | |
| "loss": 0.0248, | |
| "step": 1070 | |
| }, | |
| { | |
| "epoch": 0.4314606741573034, | |
| "grad_norm": 2.0373523235321045, | |
| "learning_rate": 8.715616681455191e-06, | |
| "loss": 0.0182, | |
| "step": 1080 | |
| }, | |
| { | |
| "epoch": 0.4354556803995006, | |
| "grad_norm": 1.43647301197052, | |
| "learning_rate": 8.69343389529725e-06, | |
| "loss": 0.0277, | |
| "step": 1090 | |
| }, | |
| { | |
| "epoch": 0.4394506866416979, | |
| "grad_norm": 1.7748357057571411, | |
| "learning_rate": 8.671251109139308e-06, | |
| "loss": 0.0257, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 0.4434456928838951, | |
| "grad_norm": 1.1857247352600098, | |
| "learning_rate": 8.649068322981368e-06, | |
| "loss": 0.021, | |
| "step": 1110 | |
| }, | |
| { | |
| "epoch": 0.4474406991260924, | |
| "grad_norm": 2.5260214805603027, | |
| "learning_rate": 8.626885536823427e-06, | |
| "loss": 0.026, | |
| "step": 1120 | |
| }, | |
| { | |
| "epoch": 0.45143570536828964, | |
| "grad_norm": 1.3535419702529907, | |
| "learning_rate": 8.604702750665485e-06, | |
| "loss": 0.021, | |
| "step": 1130 | |
| }, | |
| { | |
| "epoch": 0.45543071161048687, | |
| "grad_norm": 1.5999841690063477, | |
| "learning_rate": 8.582519964507543e-06, | |
| "loss": 0.0283, | |
| "step": 1140 | |
| }, | |
| { | |
| "epoch": 0.45942571785268416, | |
| "grad_norm": 1.9351871013641357, | |
| "learning_rate": 8.560337178349602e-06, | |
| "loss": 0.0261, | |
| "step": 1150 | |
| }, | |
| { | |
| "epoch": 0.4634207240948814, | |
| "grad_norm": 1.9496780633926392, | |
| "learning_rate": 8.53815439219166e-06, | |
| "loss": 0.0265, | |
| "step": 1160 | |
| }, | |
| { | |
| "epoch": 0.46741573033707867, | |
| "grad_norm": 1.3758857250213623, | |
| "learning_rate": 8.515971606033719e-06, | |
| "loss": 0.0215, | |
| "step": 1170 | |
| }, | |
| { | |
| "epoch": 0.4714107365792759, | |
| "grad_norm": 2.382258415222168, | |
| "learning_rate": 8.493788819875777e-06, | |
| "loss": 0.0202, | |
| "step": 1180 | |
| }, | |
| { | |
| "epoch": 0.47540574282147313, | |
| "grad_norm": 1.7476609945297241, | |
| "learning_rate": 8.471606033717836e-06, | |
| "loss": 0.0221, | |
| "step": 1190 | |
| }, | |
| { | |
| "epoch": 0.4794007490636704, | |
| "grad_norm": 2.400864601135254, | |
| "learning_rate": 8.449423247559894e-06, | |
| "loss": 0.0227, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 0.48339575530586765, | |
| "grad_norm": 2.5008318424224854, | |
| "learning_rate": 8.427240461401952e-06, | |
| "loss": 0.026, | |
| "step": 1210 | |
| }, | |
| { | |
| "epoch": 0.48739076154806493, | |
| "grad_norm": 1.983939528465271, | |
| "learning_rate": 8.40505767524401e-06, | |
| "loss": 0.0211, | |
| "step": 1220 | |
| }, | |
| { | |
| "epoch": 0.49138576779026216, | |
| "grad_norm": 1.262866497039795, | |
| "learning_rate": 8.38287488908607e-06, | |
| "loss": 0.0214, | |
| "step": 1230 | |
| }, | |
| { | |
| "epoch": 0.49538077403245945, | |
| "grad_norm": 1.4889227151870728, | |
| "learning_rate": 8.360692102928128e-06, | |
| "loss": 0.0211, | |
| "step": 1240 | |
| }, | |
| { | |
| "epoch": 0.4993757802746567, | |
| "grad_norm": 1.6180564165115356, | |
| "learning_rate": 8.338509316770186e-06, | |
| "loss": 0.0218, | |
| "step": 1250 | |
| }, | |
| { | |
| "epoch": 0.503370786516854, | |
| "grad_norm": 1.6044875383377075, | |
| "learning_rate": 8.316326530612246e-06, | |
| "loss": 0.027, | |
| "step": 1260 | |
| }, | |
| { | |
| "epoch": 0.5073657927590511, | |
| "grad_norm": 2.0169568061828613, | |
| "learning_rate": 8.294143744454305e-06, | |
| "loss": 0.0233, | |
| "step": 1270 | |
| }, | |
| { | |
| "epoch": 0.5113607990012484, | |
| "grad_norm": 2.908473253250122, | |
| "learning_rate": 8.271960958296363e-06, | |
| "loss": 0.0301, | |
| "step": 1280 | |
| }, | |
| { | |
| "epoch": 0.5153558052434457, | |
| "grad_norm": 2.558861255645752, | |
| "learning_rate": 8.249778172138421e-06, | |
| "loss": 0.0231, | |
| "step": 1290 | |
| }, | |
| { | |
| "epoch": 0.519350811485643, | |
| "grad_norm": 0.6813347935676575, | |
| "learning_rate": 8.22759538598048e-06, | |
| "loss": 0.039, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 0.5233458177278402, | |
| "grad_norm": 1.937474250793457, | |
| "learning_rate": 8.205412599822538e-06, | |
| "loss": 0.026, | |
| "step": 1310 | |
| }, | |
| { | |
| "epoch": 0.5273408239700375, | |
| "grad_norm": 0.9743942022323608, | |
| "learning_rate": 8.183229813664597e-06, | |
| "loss": 0.024, | |
| "step": 1320 | |
| }, | |
| { | |
| "epoch": 0.5313358302122347, | |
| "grad_norm": 1.7376044988632202, | |
| "learning_rate": 8.161047027506655e-06, | |
| "loss": 0.0216, | |
| "step": 1330 | |
| }, | |
| { | |
| "epoch": 0.5353308364544319, | |
| "grad_norm": 1.795914888381958, | |
| "learning_rate": 8.138864241348714e-06, | |
| "loss": 0.0235, | |
| "step": 1340 | |
| }, | |
| { | |
| "epoch": 0.5393258426966292, | |
| "grad_norm": 2.0701029300689697, | |
| "learning_rate": 8.116681455190772e-06, | |
| "loss": 0.0233, | |
| "step": 1350 | |
| }, | |
| { | |
| "epoch": 0.5433208489388265, | |
| "grad_norm": 2.0427629947662354, | |
| "learning_rate": 8.09449866903283e-06, | |
| "loss": 0.025, | |
| "step": 1360 | |
| }, | |
| { | |
| "epoch": 0.5473158551810238, | |
| "grad_norm": 1.0753390789031982, | |
| "learning_rate": 8.072315882874889e-06, | |
| "loss": 0.0287, | |
| "step": 1370 | |
| }, | |
| { | |
| "epoch": 0.5513108614232209, | |
| "grad_norm": 1.8762608766555786, | |
| "learning_rate": 8.050133096716947e-06, | |
| "loss": 0.0166, | |
| "step": 1380 | |
| }, | |
| { | |
| "epoch": 0.5553058676654182, | |
| "grad_norm": 2.1662561893463135, | |
| "learning_rate": 8.027950310559007e-06, | |
| "loss": 0.0278, | |
| "step": 1390 | |
| }, | |
| { | |
| "epoch": 0.5593008739076155, | |
| "grad_norm": 2.839298963546753, | |
| "learning_rate": 8.005767524401066e-06, | |
| "loss": 0.0204, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 0.5632958801498127, | |
| "grad_norm": 2.9252588748931885, | |
| "learning_rate": 7.983584738243124e-06, | |
| "loss": 0.0246, | |
| "step": 1410 | |
| }, | |
| { | |
| "epoch": 0.56729088639201, | |
| "grad_norm": 7.009346008300781, | |
| "learning_rate": 7.961401952085183e-06, | |
| "loss": 0.0304, | |
| "step": 1420 | |
| }, | |
| { | |
| "epoch": 0.5712858926342073, | |
| "grad_norm": 1.4174095392227173, | |
| "learning_rate": 7.939219165927241e-06, | |
| "loss": 0.0205, | |
| "step": 1430 | |
| }, | |
| { | |
| "epoch": 0.5752808988764045, | |
| "grad_norm": 1.833414912223816, | |
| "learning_rate": 7.9170363797693e-06, | |
| "loss": 0.0148, | |
| "step": 1440 | |
| }, | |
| { | |
| "epoch": 0.5792759051186017, | |
| "grad_norm": 2.492591381072998, | |
| "learning_rate": 7.894853593611358e-06, | |
| "loss": 0.0204, | |
| "step": 1450 | |
| }, | |
| { | |
| "epoch": 0.583270911360799, | |
| "grad_norm": 0.9227661490440369, | |
| "learning_rate": 7.872670807453416e-06, | |
| "loss": 0.0281, | |
| "step": 1460 | |
| }, | |
| { | |
| "epoch": 0.5872659176029963, | |
| "grad_norm": 2.0791642665863037, | |
| "learning_rate": 7.850488021295475e-06, | |
| "loss": 0.0178, | |
| "step": 1470 | |
| }, | |
| { | |
| "epoch": 0.5912609238451935, | |
| "grad_norm": 2.88958477973938, | |
| "learning_rate": 7.828305235137535e-06, | |
| "loss": 0.0218, | |
| "step": 1480 | |
| }, | |
| { | |
| "epoch": 0.5952559300873907, | |
| "grad_norm": 2.2784717082977295, | |
| "learning_rate": 7.806122448979593e-06, | |
| "loss": 0.021, | |
| "step": 1490 | |
| }, | |
| { | |
| "epoch": 0.599250936329588, | |
| "grad_norm": 2.2620465755462646, | |
| "learning_rate": 7.783939662821652e-06, | |
| "loss": 0.0251, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.6032459425717853, | |
| "grad_norm": 3.146375894546509, | |
| "learning_rate": 7.76175687666371e-06, | |
| "loss": 0.0176, | |
| "step": 1510 | |
| }, | |
| { | |
| "epoch": 0.6072409488139825, | |
| "grad_norm": 1.6521497964859009, | |
| "learning_rate": 7.739574090505769e-06, | |
| "loss": 0.0159, | |
| "step": 1520 | |
| }, | |
| { | |
| "epoch": 0.6112359550561798, | |
| "grad_norm": 1.9786664247512817, | |
| "learning_rate": 7.717391304347827e-06, | |
| "loss": 0.0175, | |
| "step": 1530 | |
| }, | |
| { | |
| "epoch": 0.6152309612983771, | |
| "grad_norm": 1.5835528373718262, | |
| "learning_rate": 7.695208518189886e-06, | |
| "loss": 0.0205, | |
| "step": 1540 | |
| }, | |
| { | |
| "epoch": 0.6192259675405742, | |
| "grad_norm": 2.4139864444732666, | |
| "learning_rate": 7.673025732031944e-06, | |
| "loss": 0.0185, | |
| "step": 1550 | |
| }, | |
| { | |
| "epoch": 0.6232209737827715, | |
| "grad_norm": 2.727465867996216, | |
| "learning_rate": 7.650842945874002e-06, | |
| "loss": 0.0195, | |
| "step": 1560 | |
| }, | |
| { | |
| "epoch": 0.6272159800249688, | |
| "grad_norm": 1.3791508674621582, | |
| "learning_rate": 7.628660159716062e-06, | |
| "loss": 0.0223, | |
| "step": 1570 | |
| }, | |
| { | |
| "epoch": 0.6312109862671661, | |
| "grad_norm": 2.2841720581054688, | |
| "learning_rate": 7.60647737355812e-06, | |
| "loss": 0.0222, | |
| "step": 1580 | |
| }, | |
| { | |
| "epoch": 0.6352059925093633, | |
| "grad_norm": 1.3033839464187622, | |
| "learning_rate": 7.5842945874001785e-06, | |
| "loss": 0.0211, | |
| "step": 1590 | |
| }, | |
| { | |
| "epoch": 0.6392009987515606, | |
| "grad_norm": 2.1349937915802, | |
| "learning_rate": 7.562111801242237e-06, | |
| "loss": 0.0343, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 0.6431960049937578, | |
| "grad_norm": 1.826735019683838, | |
| "learning_rate": 7.539929015084295e-06, | |
| "loss": 0.0226, | |
| "step": 1610 | |
| }, | |
| { | |
| "epoch": 0.647191011235955, | |
| "grad_norm": 1.3957531452178955, | |
| "learning_rate": 7.517746228926354e-06, | |
| "loss": 0.0156, | |
| "step": 1620 | |
| }, | |
| { | |
| "epoch": 0.6511860174781523, | |
| "grad_norm": 1.722582221031189, | |
| "learning_rate": 7.495563442768412e-06, | |
| "loss": 0.0312, | |
| "step": 1630 | |
| }, | |
| { | |
| "epoch": 0.6551810237203496, | |
| "grad_norm": 2.3157331943511963, | |
| "learning_rate": 7.473380656610471e-06, | |
| "loss": 0.021, | |
| "step": 1640 | |
| }, | |
| { | |
| "epoch": 0.6591760299625468, | |
| "grad_norm": 1.8118211030960083, | |
| "learning_rate": 7.451197870452529e-06, | |
| "loss": 0.0226, | |
| "step": 1650 | |
| }, | |
| { | |
| "epoch": 0.663171036204744, | |
| "grad_norm": 1.969317078590393, | |
| "learning_rate": 7.429015084294588e-06, | |
| "loss": 0.0262, | |
| "step": 1660 | |
| }, | |
| { | |
| "epoch": 0.6671660424469413, | |
| "grad_norm": 1.152690052986145, | |
| "learning_rate": 7.406832298136647e-06, | |
| "loss": 0.0251, | |
| "step": 1670 | |
| }, | |
| { | |
| "epoch": 0.6711610486891386, | |
| "grad_norm": 1.7638611793518066, | |
| "learning_rate": 7.384649511978705e-06, | |
| "loss": 0.0154, | |
| "step": 1680 | |
| }, | |
| { | |
| "epoch": 0.6751560549313358, | |
| "grad_norm": 2.218303680419922, | |
| "learning_rate": 7.362466725820764e-06, | |
| "loss": 0.0204, | |
| "step": 1690 | |
| }, | |
| { | |
| "epoch": 0.6791510611735331, | |
| "grad_norm": 15.661515235900879, | |
| "learning_rate": 7.340283939662822e-06, | |
| "loss": 0.2211, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 0.6831460674157304, | |
| "grad_norm": 2.014061689376831, | |
| "learning_rate": 7.3181011535048805e-06, | |
| "loss": 0.0835, | |
| "step": 1710 | |
| }, | |
| { | |
| "epoch": 0.6871410736579275, | |
| "grad_norm": 1.8309181928634644, | |
| "learning_rate": 7.295918367346939e-06, | |
| "loss": 0.0191, | |
| "step": 1720 | |
| }, | |
| { | |
| "epoch": 0.6911360799001248, | |
| "grad_norm": 1.6535840034484863, | |
| "learning_rate": 7.273735581188997e-06, | |
| "loss": 0.0227, | |
| "step": 1730 | |
| }, | |
| { | |
| "epoch": 0.6951310861423221, | |
| "grad_norm": 1.5450973510742188, | |
| "learning_rate": 7.251552795031056e-06, | |
| "loss": 0.0194, | |
| "step": 1740 | |
| }, | |
| { | |
| "epoch": 0.6991260923845194, | |
| "grad_norm": 1.069799542427063, | |
| "learning_rate": 7.229370008873116e-06, | |
| "loss": 0.0213, | |
| "step": 1750 | |
| }, | |
| { | |
| "epoch": 0.7031210986267166, | |
| "grad_norm": 1.9177722930908203, | |
| "learning_rate": 7.207187222715174e-06, | |
| "loss": 0.0224, | |
| "step": 1760 | |
| }, | |
| { | |
| "epoch": 0.7071161048689139, | |
| "grad_norm": 1.9900991916656494, | |
| "learning_rate": 7.185004436557233e-06, | |
| "loss": 0.0209, | |
| "step": 1770 | |
| }, | |
| { | |
| "epoch": 0.7111111111111111, | |
| "grad_norm": 1.564157247543335, | |
| "learning_rate": 7.162821650399291e-06, | |
| "loss": 0.021, | |
| "step": 1780 | |
| }, | |
| { | |
| "epoch": 0.7151061173533083, | |
| "grad_norm": 1.640061616897583, | |
| "learning_rate": 7.1406388642413496e-06, | |
| "loss": 0.0166, | |
| "step": 1790 | |
| }, | |
| { | |
| "epoch": 0.7191011235955056, | |
| "grad_norm": 2.0153284072875977, | |
| "learning_rate": 7.118456078083408e-06, | |
| "loss": 0.0224, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 0.7230961298377029, | |
| "grad_norm": 1.827107548713684, | |
| "learning_rate": 7.0962732919254664e-06, | |
| "loss": 0.0152, | |
| "step": 1810 | |
| }, | |
| { | |
| "epoch": 0.7270911360799002, | |
| "grad_norm": 1.6065489053726196, | |
| "learning_rate": 7.074090505767525e-06, | |
| "loss": 0.0209, | |
| "step": 1820 | |
| }, | |
| { | |
| "epoch": 0.7310861423220973, | |
| "grad_norm": 1.9071681499481201, | |
| "learning_rate": 7.051907719609583e-06, | |
| "loss": 0.0258, | |
| "step": 1830 | |
| }, | |
| { | |
| "epoch": 0.7350811485642946, | |
| "grad_norm": 1.726271629333496, | |
| "learning_rate": 7.0297249334516426e-06, | |
| "loss": 0.0236, | |
| "step": 1840 | |
| }, | |
| { | |
| "epoch": 0.7390761548064919, | |
| "grad_norm": 1.4094561338424683, | |
| "learning_rate": 7.007542147293701e-06, | |
| "loss": 0.027, | |
| "step": 1850 | |
| }, | |
| { | |
| "epoch": 0.7430711610486891, | |
| "grad_norm": 1.450642466545105, | |
| "learning_rate": 6.985359361135759e-06, | |
| "loss": 0.0148, | |
| "step": 1860 | |
| }, | |
| { | |
| "epoch": 0.7470661672908864, | |
| "grad_norm": 1.8943090438842773, | |
| "learning_rate": 6.963176574977818e-06, | |
| "loss": 0.0204, | |
| "step": 1870 | |
| }, | |
| { | |
| "epoch": 0.7510611735330837, | |
| "grad_norm": 1.98420250415802, | |
| "learning_rate": 6.940993788819876e-06, | |
| "loss": 0.0166, | |
| "step": 1880 | |
| }, | |
| { | |
| "epoch": 0.755056179775281, | |
| "grad_norm": 1.9308773279190063, | |
| "learning_rate": 6.918811002661935e-06, | |
| "loss": 0.0186, | |
| "step": 1890 | |
| }, | |
| { | |
| "epoch": 0.7590511860174781, | |
| "grad_norm": 2.016893148422241, | |
| "learning_rate": 6.896628216503993e-06, | |
| "loss": 0.0165, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 0.7630461922596754, | |
| "grad_norm": 3.5878658294677734, | |
| "learning_rate": 6.8744454303460515e-06, | |
| "loss": 0.023, | |
| "step": 1910 | |
| }, | |
| { | |
| "epoch": 0.7670411985018727, | |
| "grad_norm": 1.3550212383270264, | |
| "learning_rate": 6.85226264418811e-06, | |
| "loss": 0.0215, | |
| "step": 1920 | |
| }, | |
| { | |
| "epoch": 0.7710362047440699, | |
| "grad_norm": 1.6893417835235596, | |
| "learning_rate": 6.830079858030168e-06, | |
| "loss": 0.0286, | |
| "step": 1930 | |
| }, | |
| { | |
| "epoch": 0.7750312109862671, | |
| "grad_norm": 1.8765745162963867, | |
| "learning_rate": 6.8078970718722285e-06, | |
| "loss": 0.0198, | |
| "step": 1940 | |
| }, | |
| { | |
| "epoch": 0.7790262172284644, | |
| "grad_norm": 2.7229654788970947, | |
| "learning_rate": 6.785714285714287e-06, | |
| "loss": 0.0234, | |
| "step": 1950 | |
| }, | |
| { | |
| "epoch": 0.7830212234706617, | |
| "grad_norm": 2.0672197341918945, | |
| "learning_rate": 6.763531499556345e-06, | |
| "loss": 0.0156, | |
| "step": 1960 | |
| }, | |
| { | |
| "epoch": 0.7870162297128589, | |
| "grad_norm": 0.8524231314659119, | |
| "learning_rate": 6.741348713398404e-06, | |
| "loss": 0.0231, | |
| "step": 1970 | |
| }, | |
| { | |
| "epoch": 0.7910112359550562, | |
| "grad_norm": 1.3600095510482788, | |
| "learning_rate": 6.719165927240462e-06, | |
| "loss": 0.0203, | |
| "step": 1980 | |
| }, | |
| { | |
| "epoch": 0.7950062421972535, | |
| "grad_norm": 1.834830641746521, | |
| "learning_rate": 6.696983141082521e-06, | |
| "loss": 0.0227, | |
| "step": 1990 | |
| }, | |
| { | |
| "epoch": 0.7990012484394506, | |
| "grad_norm": 3.3071742057800293, | |
| "learning_rate": 6.674800354924579e-06, | |
| "loss": 0.0212, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.7990012484394506, | |
| "eval_loss": 0.011697466485202312, | |
| "eval_runtime": 14533.1869, | |
| "eval_samples_per_second": 1.378, | |
| "eval_steps_per_second": 0.172, | |
| "eval_wer": 1.0576004802313843, | |
| "step": 2000 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 5008, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 2, | |
| "save_steps": 2000, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 6.531871408128e+19, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |