| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 3.0, | |
| "eval_steps": 500, | |
| "global_step": 153, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 5.049024000371086, | |
| "learning_rate": 1.8181818181818183e-06, | |
| "loss": 1.854, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "grad_norm": 4.758646674230607, | |
| "learning_rate": 3.6363636363636366e-06, | |
| "loss": 1.8589, | |
| "step": 2 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "grad_norm": 4.997532808044432, | |
| "learning_rate": 5.4545454545454545e-06, | |
| "loss": 1.8066, | |
| "step": 3 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "grad_norm": 4.658976531052129, | |
| "learning_rate": 7.272727272727273e-06, | |
| "loss": 1.8039, | |
| "step": 4 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 3.2333098053485814, | |
| "learning_rate": 9.090909090909091e-06, | |
| "loss": 1.6903, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "grad_norm": 2.672741130516915, | |
| "learning_rate": 1.0909090909090909e-05, | |
| "loss": 1.6346, | |
| "step": 6 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "grad_norm": 1.8361460465829846, | |
| "learning_rate": 1.2727272727272728e-05, | |
| "loss": 1.5495, | |
| "step": 7 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "grad_norm": 1.7045961378249597, | |
| "learning_rate": 1.4545454545454546e-05, | |
| "loss": 1.5097, | |
| "step": 8 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "grad_norm": 1.635826359191631, | |
| "learning_rate": 1.6363636363636366e-05, | |
| "loss": 1.4721, | |
| "step": 9 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 1.578935320707837, | |
| "learning_rate": 1.8181818181818182e-05, | |
| "loss": 1.4229, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "grad_norm": 1.5221044723806934, | |
| "learning_rate": 2e-05, | |
| "loss": 1.3793, | |
| "step": 11 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "grad_norm": 1.4000849902377055, | |
| "learning_rate": 1.999867521457224e-05, | |
| "loss": 1.3113, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 1.575616258826214, | |
| "learning_rate": 1.9994701209300245e-05, | |
| "loss": 1.2887, | |
| "step": 13 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "grad_norm": 1.3392980593157788, | |
| "learning_rate": 1.9988079037124866e-05, | |
| "loss": 1.2279, | |
| "step": 14 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "grad_norm": 1.424201738399285, | |
| "learning_rate": 1.9978810452637544e-05, | |
| "loss": 1.1972, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "grad_norm": 1.4523213122657395, | |
| "learning_rate": 1.9966897911615417e-05, | |
| "loss": 1.1793, | |
| "step": 16 | |
| }, | |
| { | |
| "epoch": 0.33, | |
| "grad_norm": 1.3820011323242407, | |
| "learning_rate": 1.995234457037063e-05, | |
| "loss": 1.1211, | |
| "step": 17 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "grad_norm": 1.2328298047764605, | |
| "learning_rate": 1.9935154284914063e-05, | |
| "loss": 1.0237, | |
| "step": 18 | |
| }, | |
| { | |
| "epoch": 0.37, | |
| "grad_norm": 1.4014760682876144, | |
| "learning_rate": 1.991533160993366e-05, | |
| "loss": 1.084, | |
| "step": 19 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "grad_norm": 1.304071018054968, | |
| "learning_rate": 1.98928817975876e-05, | |
| "loss": 1.0577, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.41, | |
| "grad_norm": 1.2396816193325995, | |
| "learning_rate": 1.9867810796112742e-05, | |
| "loss": 1.0534, | |
| "step": 21 | |
| }, | |
| { | |
| "epoch": 0.43, | |
| "grad_norm": 1.1906790092813893, | |
| "learning_rate": 1.9840125248248564e-05, | |
| "loss": 0.9375, | |
| "step": 22 | |
| }, | |
| { | |
| "epoch": 0.45, | |
| "grad_norm": 1.2125922433431726, | |
| "learning_rate": 1.9809832489477144e-05, | |
| "loss": 0.907, | |
| "step": 23 | |
| }, | |
| { | |
| "epoch": 0.47, | |
| "grad_norm": 1.3509732764627251, | |
| "learning_rate": 1.9776940546079552e-05, | |
| "loss": 0.8994, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 0.49, | |
| "grad_norm": 1.1575946721358368, | |
| "learning_rate": 1.9741458133009258e-05, | |
| "loss": 0.9001, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.51, | |
| "grad_norm": 1.0819155083735588, | |
| "learning_rate": 1.970339465158301e-05, | |
| "loss": 0.8853, | |
| "step": 26 | |
| }, | |
| { | |
| "epoch": 0.53, | |
| "grad_norm": 1.1831054183701755, | |
| "learning_rate": 1.9662760186989914e-05, | |
| "loss": 0.8559, | |
| "step": 27 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "grad_norm": 1.2787427328132572, | |
| "learning_rate": 1.9619565505619288e-05, | |
| "loss": 0.8908, | |
| "step": 28 | |
| }, | |
| { | |
| "epoch": 0.57, | |
| "grad_norm": 1.1960513256280778, | |
| "learning_rate": 1.9573822052208013e-05, | |
| "loss": 0.8505, | |
| "step": 29 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "grad_norm": 1.1882091713537735, | |
| "learning_rate": 1.9525541946808187e-05, | |
| "loss": 0.8252, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.61, | |
| "grad_norm": 1.1957026823405033, | |
| "learning_rate": 1.9474737981575833e-05, | |
| "loss": 0.8294, | |
| "step": 31 | |
| }, | |
| { | |
| "epoch": 0.63, | |
| "grad_norm": 1.1235625302238945, | |
| "learning_rate": 1.942142361738151e-05, | |
| "loss": 0.7843, | |
| "step": 32 | |
| }, | |
| { | |
| "epoch": 0.65, | |
| "grad_norm": 1.2129049745653777, | |
| "learning_rate": 1.936561298024377e-05, | |
| "loss": 0.7732, | |
| "step": 33 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "grad_norm": 1.0546648941443104, | |
| "learning_rate": 1.9307320857586377e-05, | |
| "loss": 0.7081, | |
| "step": 34 | |
| }, | |
| { | |
| "epoch": 0.69, | |
| "grad_norm": 1.099433031052361, | |
| "learning_rate": 1.9246562694320258e-05, | |
| "loss": 0.6877, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.71, | |
| "grad_norm": 1.3153258284792773, | |
| "learning_rate": 1.9183354588751274e-05, | |
| "loss": 0.7302, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 0.73, | |
| "grad_norm": 1.1045347116178383, | |
| "learning_rate": 1.9117713288314864e-05, | |
| "loss": 0.6839, | |
| "step": 37 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "grad_norm": 1.1110989245435192, | |
| "learning_rate": 1.904965618513868e-05, | |
| "loss": 0.6519, | |
| "step": 38 | |
| }, | |
| { | |
| "epoch": 0.76, | |
| "grad_norm": 1.094231744667558, | |
| "learning_rate": 1.8979201311434434e-05, | |
| "loss": 0.6653, | |
| "step": 39 | |
| }, | |
| { | |
| "epoch": 0.78, | |
| "grad_norm": 1.1989962094861866, | |
| "learning_rate": 1.8906367334720125e-05, | |
| "loss": 0.664, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "grad_norm": 2.0062427841484576, | |
| "learning_rate": 1.8831173552873946e-05, | |
| "loss": 0.7041, | |
| "step": 41 | |
| }, | |
| { | |
| "epoch": 0.82, | |
| "grad_norm": 1.230679254842199, | |
| "learning_rate": 1.8753639889021197e-05, | |
| "loss": 0.6849, | |
| "step": 42 | |
| }, | |
| { | |
| "epoch": 0.84, | |
| "grad_norm": 1.1625293133229038, | |
| "learning_rate": 1.8673786886255478e-05, | |
| "loss": 0.5871, | |
| "step": 43 | |
| }, | |
| { | |
| "epoch": 0.86, | |
| "grad_norm": 1.2011252293407368, | |
| "learning_rate": 1.8591635702195672e-05, | |
| "loss": 0.5814, | |
| "step": 44 | |
| }, | |
| { | |
| "epoch": 0.88, | |
| "grad_norm": 1.1548475575271413, | |
| "learning_rate": 1.8507208103380093e-05, | |
| "loss": 0.6028, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.9, | |
| "grad_norm": 1.3290758207520514, | |
| "learning_rate": 1.8420526459499252e-05, | |
| "loss": 0.5177, | |
| "step": 46 | |
| }, | |
| { | |
| "epoch": 0.92, | |
| "grad_norm": 1.3961190552811622, | |
| "learning_rate": 1.8331613737468888e-05, | |
| "loss": 0.5327, | |
| "step": 47 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "grad_norm": 1.1179664937015907, | |
| "learning_rate": 1.8240493495344695e-05, | |
| "loss": 0.5702, | |
| "step": 48 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "grad_norm": 0.991786755708903, | |
| "learning_rate": 1.8147189876080463e-05, | |
| "loss": 0.5585, | |
| "step": 49 | |
| }, | |
| { | |
| "epoch": 0.98, | |
| "grad_norm": 1.0511710425989889, | |
| "learning_rate": 1.8051727601131228e-05, | |
| "loss": 0.4897, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "grad_norm": 0.9871728886473652, | |
| "learning_rate": 1.7954131963903134e-05, | |
| "loss": 0.5135, | |
| "step": 51 | |
| }, | |
| { | |
| "epoch": 1.02, | |
| "grad_norm": 0.8724604948002991, | |
| "learning_rate": 1.785442882305179e-05, | |
| "loss": 0.3347, | |
| "step": 52 | |
| }, | |
| { | |
| "epoch": 1.04, | |
| "grad_norm": 0.8838295943097, | |
| "learning_rate": 1.775264459563081e-05, | |
| "loss": 0.3181, | |
| "step": 53 | |
| }, | |
| { | |
| "epoch": 1.06, | |
| "grad_norm": 0.9931815866970414, | |
| "learning_rate": 1.764880625009245e-05, | |
| "loss": 0.3026, | |
| "step": 54 | |
| }, | |
| { | |
| "epoch": 1.08, | |
| "grad_norm": 1.0345742370865185, | |
| "learning_rate": 1.7542941299142113e-05, | |
| "loss": 0.3461, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 1.1, | |
| "grad_norm": 0.9441524354940148, | |
| "learning_rate": 1.7435077792448666e-05, | |
| "loss": 0.3211, | |
| "step": 56 | |
| }, | |
| { | |
| "epoch": 1.12, | |
| "grad_norm": 0.8125450415231048, | |
| "learning_rate": 1.7325244309212476e-05, | |
| "loss": 0.276, | |
| "step": 57 | |
| }, | |
| { | |
| "epoch": 1.14, | |
| "grad_norm": 0.8616036387498645, | |
| "learning_rate": 1.7213469950593156e-05, | |
| "loss": 0.3256, | |
| "step": 58 | |
| }, | |
| { | |
| "epoch": 1.16, | |
| "grad_norm": 0.8037201705902857, | |
| "learning_rate": 1.709978433199901e-05, | |
| "loss": 0.2875, | |
| "step": 59 | |
| }, | |
| { | |
| "epoch": 1.18, | |
| "grad_norm": 0.923293041412828, | |
| "learning_rate": 1.6984217575240212e-05, | |
| "loss": 0.3468, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 1.2, | |
| "grad_norm": 0.8864744076614339, | |
| "learning_rate": 1.6866800300547814e-05, | |
| "loss": 0.3045, | |
| "step": 61 | |
| }, | |
| { | |
| "epoch": 1.22, | |
| "grad_norm": 0.8688649629779567, | |
| "learning_rate": 1.674756361846071e-05, | |
| "loss": 0.3243, | |
| "step": 62 | |
| }, | |
| { | |
| "epoch": 1.24, | |
| "grad_norm": 0.77439697554644, | |
| "learning_rate": 1.6626539121582687e-05, | |
| "loss": 0.2579, | |
| "step": 63 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "grad_norm": 0.8571490155577813, | |
| "learning_rate": 1.650375887621171e-05, | |
| "loss": 0.332, | |
| "step": 64 | |
| }, | |
| { | |
| "epoch": 1.27, | |
| "grad_norm": 0.922211924907794, | |
| "learning_rate": 1.637925541384375e-05, | |
| "loss": 0.3416, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 1.29, | |
| "grad_norm": 0.8273052620873378, | |
| "learning_rate": 1.6253061722553353e-05, | |
| "loss": 0.3068, | |
| "step": 66 | |
| }, | |
| { | |
| "epoch": 1.31, | |
| "grad_norm": 0.8095932095629248, | |
| "learning_rate": 1.612521123825317e-05, | |
| "loss": 0.3171, | |
| "step": 67 | |
| }, | |
| { | |
| "epoch": 1.33, | |
| "grad_norm": 0.8019342077433093, | |
| "learning_rate": 1.5995737835834905e-05, | |
| "loss": 0.3117, | |
| "step": 68 | |
| }, | |
| { | |
| "epoch": 1.35, | |
| "grad_norm": 0.7916383152619996, | |
| "learning_rate": 1.586467582019392e-05, | |
| "loss": 0.2724, | |
| "step": 69 | |
| }, | |
| { | |
| "epoch": 1.37, | |
| "grad_norm": 0.8004248593606291, | |
| "learning_rate": 1.5732059917139912e-05, | |
| "loss": 0.2895, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 1.39, | |
| "grad_norm": 0.8986302044736728, | |
| "learning_rate": 1.5597925264196048e-05, | |
| "loss": 0.2968, | |
| "step": 71 | |
| }, | |
| { | |
| "epoch": 1.41, | |
| "grad_norm": 0.9101368979823861, | |
| "learning_rate": 1.546230740128904e-05, | |
| "loss": 0.339, | |
| "step": 72 | |
| }, | |
| { | |
| "epoch": 1.43, | |
| "grad_norm": 0.8045707914289225, | |
| "learning_rate": 1.53252422613326e-05, | |
| "loss": 0.2773, | |
| "step": 73 | |
| }, | |
| { | |
| "epoch": 1.45, | |
| "grad_norm": 0.8169621304521563, | |
| "learning_rate": 1.5186766160706738e-05, | |
| "loss": 0.2971, | |
| "step": 74 | |
| }, | |
| { | |
| "epoch": 1.47, | |
| "grad_norm": 0.7851894618359577, | |
| "learning_rate": 1.504691578963549e-05, | |
| "loss": 0.2505, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 1.49, | |
| "grad_norm": 0.7829032571228582, | |
| "learning_rate": 1.4905728202465596e-05, | |
| "loss": 0.2955, | |
| "step": 76 | |
| }, | |
| { | |
| "epoch": 1.51, | |
| "grad_norm": 0.8280131066501262, | |
| "learning_rate": 1.4763240807848667e-05, | |
| "loss": 0.3118, | |
| "step": 77 | |
| }, | |
| { | |
| "epoch": 1.53, | |
| "grad_norm": 0.7904115230357416, | |
| "learning_rate": 1.4619491358829502e-05, | |
| "loss": 0.2637, | |
| "step": 78 | |
| }, | |
| { | |
| "epoch": 1.55, | |
| "grad_norm": 0.8576605948937364, | |
| "learning_rate": 1.4474517942843173e-05, | |
| "loss": 0.3111, | |
| "step": 79 | |
| }, | |
| { | |
| "epoch": 1.57, | |
| "grad_norm": 0.771179138835728, | |
| "learning_rate": 1.4328358971623455e-05, | |
| "loss": 0.2625, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 1.59, | |
| "grad_norm": 0.7163659653314414, | |
| "learning_rate": 1.4181053171025392e-05, | |
| "loss": 0.2548, | |
| "step": 81 | |
| }, | |
| { | |
| "epoch": 1.61, | |
| "grad_norm": 0.7449243778532777, | |
| "learning_rate": 1.4032639570764595e-05, | |
| "loss": 0.2548, | |
| "step": 82 | |
| }, | |
| { | |
| "epoch": 1.63, | |
| "grad_norm": 0.7999629608162984, | |
| "learning_rate": 1.3883157494076048e-05, | |
| "loss": 0.2926, | |
| "step": 83 | |
| }, | |
| { | |
| "epoch": 1.65, | |
| "grad_norm": 0.8217922125483811, | |
| "learning_rate": 1.3732646547295128e-05, | |
| "loss": 0.2938, | |
| "step": 84 | |
| }, | |
| { | |
| "epoch": 1.67, | |
| "grad_norm": 0.8086828652630952, | |
| "learning_rate": 1.358114660936364e-05, | |
| "loss": 0.297, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 1.69, | |
| "grad_norm": 0.8194748734537538, | |
| "learning_rate": 1.34286978212636e-05, | |
| "loss": 0.2856, | |
| "step": 86 | |
| }, | |
| { | |
| "epoch": 1.71, | |
| "grad_norm": 0.7858003470117391, | |
| "learning_rate": 1.32753405753816e-05, | |
| "loss": 0.2543, | |
| "step": 87 | |
| }, | |
| { | |
| "epoch": 1.73, | |
| "grad_norm": 0.7826963768753953, | |
| "learning_rate": 1.3121115504806554e-05, | |
| "loss": 0.2677, | |
| "step": 88 | |
| }, | |
| { | |
| "epoch": 1.75, | |
| "grad_norm": 0.7729735090495947, | |
| "learning_rate": 1.2966063472563686e-05, | |
| "loss": 0.2276, | |
| "step": 89 | |
| }, | |
| { | |
| "epoch": 1.76, | |
| "grad_norm": 0.7157613149688595, | |
| "learning_rate": 1.2810225560787561e-05, | |
| "loss": 0.2236, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 1.78, | |
| "grad_norm": 0.8637010063044507, | |
| "learning_rate": 1.2653643059837109e-05, | |
| "loss": 0.2915, | |
| "step": 91 | |
| }, | |
| { | |
| "epoch": 1.8, | |
| "grad_norm": 0.7449248979464137, | |
| "learning_rate": 1.2496357457355423e-05, | |
| "loss": 0.2476, | |
| "step": 92 | |
| }, | |
| { | |
| "epoch": 1.82, | |
| "grad_norm": 0.7456800901872752, | |
| "learning_rate": 1.2338410427277342e-05, | |
| "loss": 0.276, | |
| "step": 93 | |
| }, | |
| { | |
| "epoch": 1.84, | |
| "grad_norm": 0.6931855319587382, | |
| "learning_rate": 1.2179843818787625e-05, | |
| "loss": 0.2291, | |
| "step": 94 | |
| }, | |
| { | |
| "epoch": 1.86, | |
| "grad_norm": 0.748062571866356, | |
| "learning_rate": 1.202069964523272e-05, | |
| "loss": 0.2484, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 1.88, | |
| "grad_norm": 0.7159809176976305, | |
| "learning_rate": 1.186102007298904e-05, | |
| "loss": 0.264, | |
| "step": 96 | |
| }, | |
| { | |
| "epoch": 1.9, | |
| "grad_norm": 0.6999413576425992, | |
| "learning_rate": 1.1700847410290667e-05, | |
| "loss": 0.2569, | |
| "step": 97 | |
| }, | |
| { | |
| "epoch": 1.92, | |
| "grad_norm": 0.7184171735319789, | |
| "learning_rate": 1.1540224096019495e-05, | |
| "loss": 0.2533, | |
| "step": 98 | |
| }, | |
| { | |
| "epoch": 1.94, | |
| "grad_norm": 0.7498351154279366, | |
| "learning_rate": 1.137919268846074e-05, | |
| "loss": 0.2625, | |
| "step": 99 | |
| }, | |
| { | |
| "epoch": 1.96, | |
| "grad_norm": 0.8195036397088564, | |
| "learning_rate": 1.121779585402684e-05, | |
| "loss": 0.2872, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.98, | |
| "grad_norm": 0.7239230213655573, | |
| "learning_rate": 1.105607635595266e-05, | |
| "loss": 0.2138, | |
| "step": 101 | |
| }, | |
| { | |
| "epoch": 2.0, | |
| "grad_norm": 0.5927611951490782, | |
| "learning_rate": 1.0894077042965084e-05, | |
| "loss": 0.1527, | |
| "step": 102 | |
| }, | |
| { | |
| "epoch": 2.02, | |
| "grad_norm": 0.5812390972468515, | |
| "learning_rate": 1.0731840837929946e-05, | |
| "loss": 0.0943, | |
| "step": 103 | |
| }, | |
| { | |
| "epoch": 2.04, | |
| "grad_norm": 0.6457949647478547, | |
| "learning_rate": 1.0569410726479301e-05, | |
| "loss": 0.1276, | |
| "step": 104 | |
| }, | |
| { | |
| "epoch": 2.06, | |
| "grad_norm": 0.6076599586709107, | |
| "learning_rate": 1.0406829745622085e-05, | |
| "loss": 0.1118, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 2.08, | |
| "grad_norm": 0.6580690196356788, | |
| "learning_rate": 1.0244140972341155e-05, | |
| "loss": 0.1266, | |
| "step": 106 | |
| }, | |
| { | |
| "epoch": 2.1, | |
| "grad_norm": 0.6652391219627729, | |
| "learning_rate": 1.008138751217973e-05, | |
| "loss": 0.1061, | |
| "step": 107 | |
| }, | |
| { | |
| "epoch": 2.12, | |
| "grad_norm": 0.8077871750305834, | |
| "learning_rate": 9.918612487820274e-06, | |
| "loss": 0.1372, | |
| "step": 108 | |
| }, | |
| { | |
| "epoch": 2.14, | |
| "grad_norm": 0.7590982992651446, | |
| "learning_rate": 9.755859027658848e-06, | |
| "loss": 0.1108, | |
| "step": 109 | |
| }, | |
| { | |
| "epoch": 2.16, | |
| "grad_norm": 0.8397462300303704, | |
| "learning_rate": 9.593170254377915e-06, | |
| "loss": 0.1289, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 2.18, | |
| "grad_norm": 0.7003906062775592, | |
| "learning_rate": 9.430589273520704e-06, | |
| "loss": 0.1058, | |
| "step": 111 | |
| }, | |
| { | |
| "epoch": 2.2, | |
| "grad_norm": 0.7088357228372557, | |
| "learning_rate": 9.268159162070058e-06, | |
| "loss": 0.1135, | |
| "step": 112 | |
| }, | |
| { | |
| "epoch": 2.22, | |
| "grad_norm": 0.7000401962183394, | |
| "learning_rate": 9.105922957034921e-06, | |
| "loss": 0.1096, | |
| "step": 113 | |
| }, | |
| { | |
| "epoch": 2.24, | |
| "grad_norm": 0.5830640142754188, | |
| "learning_rate": 8.943923644047343e-06, | |
| "loss": 0.096, | |
| "step": 114 | |
| }, | |
| { | |
| "epoch": 2.25, | |
| "grad_norm": 0.7040673193362394, | |
| "learning_rate": 8.782204145973162e-06, | |
| "loss": 0.126, | |
| "step": 115 | |
| }, | |
| { | |
| "epoch": 2.27, | |
| "grad_norm": 0.5883035825262033, | |
| "learning_rate": 8.620807311539258e-06, | |
| "loss": 0.1046, | |
| "step": 116 | |
| }, | |
| { | |
| "epoch": 2.29, | |
| "grad_norm": 0.5889783156569407, | |
| "learning_rate": 8.45977590398051e-06, | |
| "loss": 0.1099, | |
| "step": 117 | |
| }, | |
| { | |
| "epoch": 2.31, | |
| "grad_norm": 0.5746493638760929, | |
| "learning_rate": 8.299152589709336e-06, | |
| "loss": 0.0998, | |
| "step": 118 | |
| }, | |
| { | |
| "epoch": 2.33, | |
| "grad_norm": 0.5063169905997035, | |
| "learning_rate": 8.138979927010964e-06, | |
| "loss": 0.0829, | |
| "step": 119 | |
| }, | |
| { | |
| "epoch": 2.35, | |
| "grad_norm": 0.561526728081391, | |
| "learning_rate": 7.979300354767282e-06, | |
| "loss": 0.099, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 2.37, | |
| "grad_norm": 0.6140392260229752, | |
| "learning_rate": 7.82015618121238e-06, | |
| "loss": 0.112, | |
| "step": 121 | |
| }, | |
| { | |
| "epoch": 2.39, | |
| "grad_norm": 0.6024756993343563, | |
| "learning_rate": 7.66158957272266e-06, | |
| "loss": 0.0924, | |
| "step": 122 | |
| }, | |
| { | |
| "epoch": 2.41, | |
| "grad_norm": 0.6149831107386897, | |
| "learning_rate": 7.503642542644581e-06, | |
| "loss": 0.1056, | |
| "step": 123 | |
| }, | |
| { | |
| "epoch": 2.43, | |
| "grad_norm": 0.6268831968674875, | |
| "learning_rate": 7.346356940162895e-06, | |
| "loss": 0.1109, | |
| "step": 124 | |
| }, | |
| { | |
| "epoch": 2.45, | |
| "grad_norm": 0.5774013268612427, | |
| "learning_rate": 7.189774439212442e-06, | |
| "loss": 0.095, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 2.47, | |
| "grad_norm": 0.5981672010212989, | |
| "learning_rate": 7.033936527436318e-06, | |
| "loss": 0.1028, | |
| "step": 126 | |
| }, | |
| { | |
| "epoch": 2.49, | |
| "grad_norm": 0.6129772623459513, | |
| "learning_rate": 6.878884495193448e-06, | |
| "loss": 0.0945, | |
| "step": 127 | |
| }, | |
| { | |
| "epoch": 2.51, | |
| "grad_norm": 0.5804441177255439, | |
| "learning_rate": 6.724659424618401e-06, | |
| "loss": 0.099, | |
| "step": 128 | |
| }, | |
| { | |
| "epoch": 2.53, | |
| "grad_norm": 0.6592505296857643, | |
| "learning_rate": 6.571302178736404e-06, | |
| "loss": 0.1119, | |
| "step": 129 | |
| }, | |
| { | |
| "epoch": 2.55, | |
| "grad_norm": 0.5876228001270772, | |
| "learning_rate": 6.418853390636363e-06, | |
| "loss": 0.1066, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 2.57, | |
| "grad_norm": 0.5711848758996622, | |
| "learning_rate": 6.267353452704876e-06, | |
| "loss": 0.0939, | |
| "step": 131 | |
| }, | |
| { | |
| "epoch": 2.59, | |
| "grad_norm": 0.5276470301226159, | |
| "learning_rate": 6.116842505923955e-06, | |
| "loss": 0.086, | |
| "step": 132 | |
| }, | |
| { | |
| "epoch": 2.61, | |
| "grad_norm": 0.5743382647434178, | |
| "learning_rate": 5.967360429235407e-06, | |
| "loss": 0.0976, | |
| "step": 133 | |
| }, | |
| { | |
| "epoch": 2.63, | |
| "grad_norm": 0.5502701680994874, | |
| "learning_rate": 5.8189468289746075e-06, | |
| "loss": 0.0925, | |
| "step": 134 | |
| }, | |
| { | |
| "epoch": 2.65, | |
| "grad_norm": 0.6137369248887328, | |
| "learning_rate": 5.671641028376547e-06, | |
| "loss": 0.1014, | |
| "step": 135 | |
| }, | |
| { | |
| "epoch": 2.67, | |
| "grad_norm": 0.5582863288518123, | |
| "learning_rate": 5.525482057156833e-06, | |
| "loss": 0.0977, | |
| "step": 136 | |
| }, | |
| { | |
| "epoch": 2.69, | |
| "grad_norm": 0.5454192540734187, | |
| "learning_rate": 5.380508641170499e-06, | |
| "loss": 0.0904, | |
| "step": 137 | |
| }, | |
| { | |
| "epoch": 2.71, | |
| "grad_norm": 0.5209533902715616, | |
| "learning_rate": 5.236759192151336e-06, | |
| "loss": 0.0866, | |
| "step": 138 | |
| }, | |
| { | |
| "epoch": 2.73, | |
| "grad_norm": 0.5717122393516525, | |
| "learning_rate": 5.094271797534404e-06, | |
| "loss": 0.0996, | |
| "step": 139 | |
| }, | |
| { | |
| "epoch": 2.75, | |
| "grad_norm": 0.6064773526535152, | |
| "learning_rate": 4.953084210364508e-06, | |
| "loss": 0.104, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 2.76, | |
| "grad_norm": 0.5691234525814087, | |
| "learning_rate": 4.813233839293265e-06, | |
| "loss": 0.1015, | |
| "step": 141 | |
| }, | |
| { | |
| "epoch": 2.78, | |
| "grad_norm": 0.5208562146564856, | |
| "learning_rate": 4.674757738667405e-06, | |
| "loss": 0.0851, | |
| "step": 142 | |
| }, | |
| { | |
| "epoch": 2.8, | |
| "grad_norm": 0.5729654869317482, | |
| "learning_rate": 4.537692598710962e-06, | |
| "loss": 0.0999, | |
| "step": 143 | |
| }, | |
| { | |
| "epoch": 2.82, | |
| "grad_norm": 0.5098181803420644, | |
| "learning_rate": 4.402074735803955e-06, | |
| "loss": 0.0811, | |
| "step": 144 | |
| }, | |
| { | |
| "epoch": 2.84, | |
| "grad_norm": 0.5599902642629923, | |
| "learning_rate": 4.267940082860088e-06, | |
| "loss": 0.0859, | |
| "step": 145 | |
| }, | |
| { | |
| "epoch": 2.86, | |
| "grad_norm": 0.6128082395071871, | |
| "learning_rate": 4.135324179806079e-06, | |
| "loss": 0.1044, | |
| "step": 146 | |
| }, | |
| { | |
| "epoch": 2.88, | |
| "grad_norm": 0.6027252305351385, | |
| "learning_rate": 4.004262164165098e-06, | |
| "loss": 0.101, | |
| "step": 147 | |
| }, | |
| { | |
| "epoch": 2.9, | |
| "grad_norm": 0.6114396307412036, | |
| "learning_rate": 3.874788761746836e-06, | |
| "loss": 0.1008, | |
| "step": 148 | |
| }, | |
| { | |
| "epoch": 2.92, | |
| "grad_norm": 0.5382288139061067, | |
| "learning_rate": 3.74693827744665e-06, | |
| "loss": 0.0846, | |
| "step": 149 | |
| }, | |
| { | |
| "epoch": 2.94, | |
| "grad_norm": 0.680532062269904, | |
| "learning_rate": 3.6207445861562497e-06, | |
| "loss": 0.1155, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 2.96, | |
| "grad_norm": 0.5889312810479405, | |
| "learning_rate": 3.4962411237882945e-06, | |
| "loss": 0.0916, | |
| "step": 151 | |
| }, | |
| { | |
| "epoch": 2.98, | |
| "grad_norm": 0.5564936393621064, | |
| "learning_rate": 3.373460878417315e-06, | |
| "loss": 0.0937, | |
| "step": 152 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "grad_norm": 0.4415683617442384, | |
| "learning_rate": 3.252436381539291e-06, | |
| "loss": 0.0547, | |
| "step": 153 | |
| } | |
| ], | |
| "logging_steps": 1, | |
| "max_steps": 204, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 4, | |
| "save_steps": 500, | |
| "total_flos": 37890624061440.0, | |
| "train_batch_size": 16, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |