| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.27986162397481246, | |
| "eval_steps": 500, | |
| "global_step": 900, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 0.47774845361709595, | |
| "learning_rate": 4.999970160815579e-05, | |
| "loss": 2.0765, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 0.6051416397094727, | |
| "learning_rate": 4.999880643974619e-05, | |
| "loss": 2.2297, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.0, | |
| "grad_norm": 0.6161717772483826, | |
| "learning_rate": 4.9997314516140056e-05, | |
| "loss": 2.1103, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.4686434268951416, | |
| "learning_rate": 4.999522587295162e-05, | |
| "loss": 2.0057, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.8412289023399353, | |
| "learning_rate": 4.999254056003963e-05, | |
| "loss": 2.1778, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.5333625078201294, | |
| "learning_rate": 4.99892586415061e-05, | |
| "loss": 2.2399, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.821148157119751, | |
| "learning_rate": 4.9985380195694856e-05, | |
| "loss": 2.3215, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.8403909206390381, | |
| "learning_rate": 4.998090531518962e-05, | |
| "loss": 1.8295, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.01, | |
| "grad_norm": 0.6633398532867432, | |
| "learning_rate": 4.9975834106811834e-05, | |
| "loss": 2.0195, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.6386868357658386, | |
| "learning_rate": 4.997016669161806e-05, | |
| "loss": 2.1257, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.7762248516082764, | |
| "learning_rate": 4.996390320489715e-05, | |
| "loss": 2.057, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.3192856311798096, | |
| "learning_rate": 4.9957043796166966e-05, | |
| "loss": 2.0753, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.9797518849372864, | |
| "learning_rate": 4.994958862917083e-05, | |
| "loss": 1.9736, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.000693440437317, | |
| "learning_rate": 4.994153788187363e-05, | |
| "loss": 2.1572, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 0.6852813959121704, | |
| "learning_rate": 4.993289174645757e-05, | |
| "loss": 2.1491, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "grad_norm": 1.0075691938400269, | |
| "learning_rate": 4.992365042931752e-05, | |
| "loss": 1.945, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "grad_norm": 1.1973133087158203, | |
| "learning_rate": 4.991381415105619e-05, | |
| "loss": 2.0811, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "grad_norm": 0.9927239418029785, | |
| "learning_rate": 4.990338314647881e-05, | |
| "loss": 1.961, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "grad_norm": 0.9499759674072266, | |
| "learning_rate": 4.98923576645875e-05, | |
| "loss": 2.0653, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "grad_norm": 0.7233040928840637, | |
| "learning_rate": 4.9880737968575365e-05, | |
| "loss": 1.9999, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "grad_norm": 1.55235755443573, | |
| "learning_rate": 4.986852433582022e-05, | |
| "loss": 2.2258, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 0.03, | |
| "grad_norm": 0.9007890820503235, | |
| "learning_rate": 4.985571705787793e-05, | |
| "loss": 2.1034, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "grad_norm": 0.6774860620498657, | |
| "learning_rate": 4.9842316440475475e-05, | |
| "loss": 2.1753, | |
| "step": 115 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "grad_norm": 0.7676737308502197, | |
| "learning_rate": 4.9828322803503665e-05, | |
| "loss": 2.1384, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "grad_norm": 0.9624544978141785, | |
| "learning_rate": 4.981373648100946e-05, | |
| "loss": 2.0521, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "grad_norm": 0.9315722584724426, | |
| "learning_rate": 4.979855782118802e-05, | |
| "loss": 1.9256, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "grad_norm": 0.9035864472389221, | |
| "learning_rate": 4.978278718637443e-05, | |
| "loss": 2.0882, | |
| "step": 135 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "grad_norm": 0.7997236251831055, | |
| "learning_rate": 4.9766424953035e-05, | |
| "loss": 2.0724, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 1.0692921876907349, | |
| "learning_rate": 4.974947151175826e-05, | |
| "loss": 2.1329, | |
| "step": 145 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 0.9506180286407471, | |
| "learning_rate": 4.973192726724572e-05, | |
| "loss": 2.082, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 0.8647387027740479, | |
| "learning_rate": 4.9713792638302145e-05, | |
| "loss": 2.0366, | |
| "step": 155 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 1.105302095413208, | |
| "learning_rate": 4.969506805782555e-05, | |
| "loss": 2.1481, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 0.7593303918838501, | |
| "learning_rate": 4.967575397279689e-05, | |
| "loss": 2.032, | |
| "step": 165 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 0.7521979808807373, | |
| "learning_rate": 4.965585084426943e-05, | |
| "loss": 2.0379, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.05, | |
| "grad_norm": 0.947120726108551, | |
| "learning_rate": 4.9635359147357655e-05, | |
| "loss": 2.1444, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "grad_norm": 1.2184454202651978, | |
| "learning_rate": 4.961427937122598e-05, | |
| "loss": 1.9164, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "grad_norm": 1.221663475036621, | |
| "learning_rate": 4.959261201907707e-05, | |
| "loss": 2.0084, | |
| "step": 185 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "grad_norm": 1.0457361936569214, | |
| "learning_rate": 4.957035760813982e-05, | |
| "loss": 2.2032, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "grad_norm": 0.8834909200668335, | |
| "learning_rate": 4.954751666965701e-05, | |
| "loss": 2.2101, | |
| "step": 195 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "grad_norm": 0.791902482509613, | |
| "learning_rate": 4.9524089748872615e-05, | |
| "loss": 2.0472, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "grad_norm": 1.2905739545822144, | |
| "learning_rate": 4.9500077405018807e-05, | |
| "loss": 2.0987, | |
| "step": 205 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "grad_norm": 0.8612006306648254, | |
| "learning_rate": 4.9475480211302583e-05, | |
| "loss": 2.1765, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "grad_norm": 1.3128459453582764, | |
| "learning_rate": 4.945029875489212e-05, | |
| "loss": 1.9926, | |
| "step": 215 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "grad_norm": 0.9610918164253235, | |
| "learning_rate": 4.94245336369027e-05, | |
| "loss": 2.0124, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "grad_norm": 0.873160183429718, | |
| "learning_rate": 4.939818547238241e-05, | |
| "loss": 2.2229, | |
| "step": 225 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "grad_norm": 1.5535285472869873, | |
| "learning_rate": 4.9371254890297446e-05, | |
| "loss": 2.2013, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "grad_norm": 1.1951836347579956, | |
| "learning_rate": 4.93437425335171e-05, | |
| "loss": 2.014, | |
| "step": 235 | |
| }, | |
| { | |
| "epoch": 0.07, | |
| "grad_norm": 0.7874170541763306, | |
| "learning_rate": 4.9315649058798384e-05, | |
| "loss": 2.1701, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "grad_norm": 1.3503323793411255, | |
| "learning_rate": 4.928697513677042e-05, | |
| "loss": 2.1681, | |
| "step": 245 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "grad_norm": 1.3091179132461548, | |
| "learning_rate": 4.925772145191834e-05, | |
| "loss": 2.1224, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "grad_norm": 1.4428555965423584, | |
| "learning_rate": 4.9227888702567044e-05, | |
| "loss": 2.0512, | |
| "step": 255 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "grad_norm": 0.8234395980834961, | |
| "learning_rate": 4.9197477600864446e-05, | |
| "loss": 2.1067, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "grad_norm": 1.9094969034194946, | |
| "learning_rate": 4.9166488872764526e-05, | |
| "loss": 1.8884, | |
| "step": 265 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "grad_norm": 1.0074087381362915, | |
| "learning_rate": 4.913492325800999e-05, | |
| "loss": 1.9345, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "grad_norm": 1.0867297649383545, | |
| "learning_rate": 4.910278151011458e-05, | |
| "loss": 2.1928, | |
| "step": 275 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "grad_norm": 0.6842357516288757, | |
| "learning_rate": 4.907006439634516e-05, | |
| "loss": 2.0407, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "grad_norm": 0.8409023284912109, | |
| "learning_rate": 4.903677269770329e-05, | |
| "loss": 2.2344, | |
| "step": 285 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "grad_norm": 0.8119503259658813, | |
| "learning_rate": 4.900290720890671e-05, | |
| "loss": 2.1296, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "grad_norm": 0.9938147068023682, | |
| "learning_rate": 4.8968468738370244e-05, | |
| "loss": 2.152, | |
| "step": 295 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "grad_norm": 0.9865244030952454, | |
| "learning_rate": 4.8933458108186606e-05, | |
| "loss": 1.9623, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.09, | |
| "grad_norm": 1.3944802284240723, | |
| "learning_rate": 4.889787615410672e-05, | |
| "loss": 1.915, | |
| "step": 305 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 1.3749767541885376, | |
| "learning_rate": 4.886172372551977e-05, | |
| "loss": 1.9934, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 0.9024938941001892, | |
| "learning_rate": 4.882500168543294e-05, | |
| "loss": 2.1541, | |
| "step": 315 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 1.1978263854980469, | |
| "learning_rate": 4.878771091045082e-05, | |
| "loss": 2.1688, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 0.8360010981559753, | |
| "learning_rate": 4.874985229075446e-05, | |
| "loss": 2.1387, | |
| "step": 325 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 0.7683364152908325, | |
| "learning_rate": 4.871142673008012e-05, | |
| "loss": 2.0215, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "grad_norm": 1.4230670928955078, | |
| "learning_rate": 4.867243514569772e-05, | |
| "loss": 1.9491, | |
| "step": 335 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "grad_norm": 0.8198773860931396, | |
| "learning_rate": 4.863287846838891e-05, | |
| "loss": 2.0151, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "grad_norm": 1.467207908630371, | |
| "learning_rate": 4.85927576424249e-05, | |
| "loss": 1.8906, | |
| "step": 345 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "grad_norm": 0.9537095427513123, | |
| "learning_rate": 4.855207362554385e-05, | |
| "loss": 2.1844, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "grad_norm": 1.0757155418395996, | |
| "learning_rate": 4.851082738892809e-05, | |
| "loss": 2.048, | |
| "step": 355 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "grad_norm": 1.6884938478469849, | |
| "learning_rate": 4.8469019917180846e-05, | |
| "loss": 1.9537, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 0.11, | |
| "grad_norm": 1.4680182933807373, | |
| "learning_rate": 4.8426652208302814e-05, | |
| "loss": 1.9731, | |
| "step": 365 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "grad_norm": 1.1778632402420044, | |
| "learning_rate": 4.83837252736683e-05, | |
| "loss": 2.1395, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "grad_norm": 1.2865056991577148, | |
| "learning_rate": 4.834024013800108e-05, | |
| "loss": 2.0016, | |
| "step": 375 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "grad_norm": 1.055177092552185, | |
| "learning_rate": 4.8296197839349944e-05, | |
| "loss": 1.9632, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "grad_norm": 1.0041871070861816, | |
| "learning_rate": 4.825159942906389e-05, | |
| "loss": 2.3302, | |
| "step": 385 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "grad_norm": 1.0026438236236572, | |
| "learning_rate": 4.820644597176709e-05, | |
| "loss": 2.1517, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "grad_norm": 1.3532180786132812, | |
| "learning_rate": 4.81607385453334e-05, | |
| "loss": 2.1229, | |
| "step": 395 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "grad_norm": 0.7670988440513611, | |
| "learning_rate": 4.81144782408607e-05, | |
| "loss": 2.1382, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "grad_norm": 1.0405700206756592, | |
| "learning_rate": 4.8067666162644774e-05, | |
| "loss": 1.9614, | |
| "step": 405 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "grad_norm": 1.2252662181854248, | |
| "learning_rate": 4.802030342815304e-05, | |
| "loss": 2.1399, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "grad_norm": 1.237946629524231, | |
| "learning_rate": 4.7972391167997754e-05, | |
| "loss": 1.9034, | |
| "step": 415 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "grad_norm": 0.8064705729484558, | |
| "learning_rate": 4.7923930525909156e-05, | |
| "loss": 2.0075, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "grad_norm": 0.8717565536499023, | |
| "learning_rate": 4.7874922658708065e-05, | |
| "loss": 2.0105, | |
| "step": 425 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "grad_norm": 1.6693098545074463, | |
| "learning_rate": 4.782536873627832e-05, | |
| "loss": 2.0242, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "grad_norm": 0.82447350025177, | |
| "learning_rate": 4.777526994153882e-05, | |
| "loss": 2.0267, | |
| "step": 435 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "grad_norm": 0.9926588535308838, | |
| "learning_rate": 4.7724627470415307e-05, | |
| "loss": 1.9119, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "grad_norm": 1.0924450159072876, | |
| "learning_rate": 4.7673442531811796e-05, | |
| "loss": 2.2653, | |
| "step": 445 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "grad_norm": 1.1592103242874146, | |
| "learning_rate": 4.762171634758177e-05, | |
| "loss": 2.0017, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "grad_norm": 0.9172110557556152, | |
| "learning_rate": 4.7569450152498927e-05, | |
| "loss": 2.1408, | |
| "step": 455 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "grad_norm": 1.1897525787353516, | |
| "learning_rate": 4.751664519422778e-05, | |
| "loss": 2.0935, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 0.14, | |
| "grad_norm": 0.8793094158172607, | |
| "learning_rate": 4.746330273329386e-05, | |
| "loss": 2.1142, | |
| "step": 465 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "grad_norm": 1.4337489604949951, | |
| "learning_rate": 4.740942404305356e-05, | |
| "loss": 2.1289, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "grad_norm": 1.0251764059066772, | |
| "learning_rate": 4.735501040966383e-05, | |
| "loss": 1.9741, | |
| "step": 475 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "grad_norm": 1.2659822702407837, | |
| "learning_rate": 4.730006313205143e-05, | |
| "loss": 2.088, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "grad_norm": 0.8884140849113464, | |
| "learning_rate": 4.724458352188192e-05, | |
| "loss": 2.2079, | |
| "step": 485 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "grad_norm": 1.1937768459320068, | |
| "learning_rate": 4.718857290352835e-05, | |
| "loss": 2.048, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "grad_norm": 0.9741552472114563, | |
| "learning_rate": 4.713203261403966e-05, | |
| "loss": 2.2569, | |
| "step": 495 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "grad_norm": 0.7996780872344971, | |
| "learning_rate": 4.707496400310874e-05, | |
| "loss": 1.9574, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "grad_norm": 1.8182051181793213, | |
| "learning_rate": 4.701736843304025e-05, | |
| "loss": 2.0951, | |
| "step": 505 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "grad_norm": 1.507320761680603, | |
| "learning_rate": 4.695924727871805e-05, | |
| "loss": 2.0253, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "grad_norm": 0.759121835231781, | |
| "learning_rate": 4.690060192757242e-05, | |
| "loss": 2.0602, | |
| "step": 515 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "grad_norm": 1.5943195819854736, | |
| "learning_rate": 4.684143377954691e-05, | |
| "loss": 2.0386, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "grad_norm": 0.8568710088729858, | |
| "learning_rate": 4.6781744247064955e-05, | |
| "loss": 2.073, | |
| "step": 525 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "grad_norm": 1.3352620601654053, | |
| "learning_rate": 4.6721534754996125e-05, | |
| "loss": 2.1443, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "grad_norm": 1.3417474031448364, | |
| "learning_rate": 4.666080674062213e-05, | |
| "loss": 2.0288, | |
| "step": 535 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "grad_norm": 1.5334464311599731, | |
| "learning_rate": 4.659956165360251e-05, | |
| "loss": 2.0609, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "grad_norm": 0.9658721089363098, | |
| "learning_rate": 4.6537800955940005e-05, | |
| "loss": 1.9539, | |
| "step": 545 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "grad_norm": 1.9197947978973389, | |
| "learning_rate": 4.647552612194572e-05, | |
| "loss": 2.149, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "grad_norm": 0.8512137532234192, | |
| "learning_rate": 4.641273863820383e-05, | |
| "loss": 1.9722, | |
| "step": 555 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "grad_norm": 1.827289342880249, | |
| "learning_rate": 4.634944000353622e-05, | |
| "loss": 2.0729, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "grad_norm": 1.088416337966919, | |
| "learning_rate": 4.628563172896655e-05, | |
| "loss": 1.9507, | |
| "step": 565 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "grad_norm": 1.3566908836364746, | |
| "learning_rate": 4.6221315337684353e-05, | |
| "loss": 2.1643, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "grad_norm": 1.3541293144226074, | |
| "learning_rate": 4.615649236500854e-05, | |
| "loss": 2.1839, | |
| "step": 575 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "grad_norm": 0.991269588470459, | |
| "learning_rate": 4.609116435835083e-05, | |
| "loss": 2.0976, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "grad_norm": 1.0280535221099854, | |
| "learning_rate": 4.602533287717877e-05, | |
| "loss": 2.1474, | |
| "step": 585 | |
| }, | |
| { | |
| "epoch": 0.18, | |
| "grad_norm": 1.013123631477356, | |
| "learning_rate": 4.5958999492978524e-05, | |
| "loss": 2.1873, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "grad_norm": 1.1753040552139282, | |
| "learning_rate": 4.589216578921737e-05, | |
| "loss": 2.1744, | |
| "step": 595 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "grad_norm": 1.1839090585708618, | |
| "learning_rate": 4.582483336130586e-05, | |
| "loss": 1.9982, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "grad_norm": 1.0724798440933228, | |
| "learning_rate": 4.575700381655979e-05, | |
| "loss": 2.1234, | |
| "step": 605 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "grad_norm": 2.009913682937622, | |
| "learning_rate": 4.5688678774161796e-05, | |
| "loss": 1.9478, | |
| "step": 610 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "grad_norm": 0.9897060394287109, | |
| "learning_rate": 4.561985986512271e-05, | |
| "loss": 1.8268, | |
| "step": 615 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "grad_norm": 0.8881808519363403, | |
| "learning_rate": 4.555054873224263e-05, | |
| "loss": 1.9887, | |
| "step": 620 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "grad_norm": 1.155900001525879, | |
| "learning_rate": 4.54807470300717e-05, | |
| "loss": 2.0777, | |
| "step": 625 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 0.8782421350479126, | |
| "learning_rate": 4.5410456424870596e-05, | |
| "loss": 2.0566, | |
| "step": 630 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 1.3324674367904663, | |
| "learning_rate": 4.5339678594570795e-05, | |
| "loss": 2.047, | |
| "step": 635 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 1.9805939197540283, | |
| "learning_rate": 4.526841522873449e-05, | |
| "loss": 1.962, | |
| "step": 640 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 1.4999943971633911, | |
| "learning_rate": 4.519666802851422e-05, | |
| "loss": 2.0972, | |
| "step": 645 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 1.4504961967468262, | |
| "learning_rate": 4.5124438706612376e-05, | |
| "loss": 2.0041, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "grad_norm": 0.9078169465065002, | |
| "learning_rate": 4.505172898724018e-05, | |
| "loss": 2.1229, | |
| "step": 655 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "grad_norm": 1.1635804176330566, | |
| "learning_rate": 4.497854060607662e-05, | |
| "loss": 2.0195, | |
| "step": 660 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "grad_norm": 1.46576726436615, | |
| "learning_rate": 4.490487531022699e-05, | |
| "loss": 2.0745, | |
| "step": 665 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "grad_norm": 1.2094652652740479, | |
| "learning_rate": 4.4830734858181145e-05, | |
| "loss": 2.1068, | |
| "step": 670 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "grad_norm": 1.4738895893096924, | |
| "learning_rate": 4.47561210197716e-05, | |
| "loss": 1.8088, | |
| "step": 675 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "grad_norm": 1.23384690284729, | |
| "learning_rate": 4.4681035576131215e-05, | |
| "loss": 2.0995, | |
| "step": 680 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "grad_norm": 0.8332946300506592, | |
| "learning_rate": 4.46054803196507e-05, | |
| "loss": 2.0541, | |
| "step": 685 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "grad_norm": 0.9207485318183899, | |
| "learning_rate": 4.452945705393586e-05, | |
| "loss": 2.166, | |
| "step": 690 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "grad_norm": 1.292945146560669, | |
| "learning_rate": 4.445296759376449e-05, | |
| "loss": 2.0784, | |
| "step": 695 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "grad_norm": 0.9874763488769531, | |
| "learning_rate": 4.437601376504307e-05, | |
| "loss": 2.2087, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "grad_norm": 0.9427415132522583, | |
| "learning_rate": 4.4298597404763186e-05, | |
| "loss": 2.1199, | |
| "step": 705 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "grad_norm": 1.7369529008865356, | |
| "learning_rate": 4.422072036095768e-05, | |
| "loss": 2.0355, | |
| "step": 710 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "grad_norm": 1.2423696517944336, | |
| "learning_rate": 4.414238449265654e-05, | |
| "loss": 2.0011, | |
| "step": 715 | |
| }, | |
| { | |
| "epoch": 0.22, | |
| "grad_norm": 1.2304831743240356, | |
| "learning_rate": 4.406359166984249e-05, | |
| "loss": 2.0368, | |
| "step": 720 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "grad_norm": 0.9090413451194763, | |
| "learning_rate": 4.39843437734064e-05, | |
| "loss": 1.9983, | |
| "step": 725 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "grad_norm": 1.2729507684707642, | |
| "learning_rate": 4.390464269510233e-05, | |
| "loss": 2.021, | |
| "step": 730 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "grad_norm": 1.3009227514266968, | |
| "learning_rate": 4.382449033750244e-05, | |
| "loss": 1.9743, | |
| "step": 735 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "grad_norm": 1.5456056594848633, | |
| "learning_rate": 4.37438886139515e-05, | |
| "loss": 2.0689, | |
| "step": 740 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "grad_norm": 1.3235007524490356, | |
| "learning_rate": 4.3662839448521264e-05, | |
| "loss": 2.0838, | |
| "step": 745 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "grad_norm": 2.2074007987976074, | |
| "learning_rate": 4.358134477596454e-05, | |
| "loss": 2.0835, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "grad_norm": 1.403738021850586, | |
| "learning_rate": 4.3499406541668966e-05, | |
| "loss": 2.0916, | |
| "step": 755 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "grad_norm": 1.0940325260162354, | |
| "learning_rate": 4.3417026701610616e-05, | |
| "loss": 1.972, | |
| "step": 760 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "grad_norm": 1.666353702545166, | |
| "learning_rate": 4.3334207222307275e-05, | |
| "loss": 1.927, | |
| "step": 765 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "grad_norm": 1.0777515172958374, | |
| "learning_rate": 4.325095008077154e-05, | |
| "loss": 2.1192, | |
| "step": 770 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "grad_norm": 1.7218186855316162, | |
| "learning_rate": 4.316725726446353e-05, | |
| "loss": 2.0774, | |
| "step": 775 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "grad_norm": 1.356753945350647, | |
| "learning_rate": 4.3083130771243586e-05, | |
| "loss": 2.0847, | |
| "step": 780 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "grad_norm": 0.9967429637908936, | |
| "learning_rate": 4.299857260932445e-05, | |
| "loss": 2.0485, | |
| "step": 785 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 1.6216442584991455, | |
| "learning_rate": 4.2913584797223397e-05, | |
| "loss": 2.1008, | |
| "step": 790 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 1.2556742429733276, | |
| "learning_rate": 4.2828169363714016e-05, | |
| "loss": 1.9209, | |
| "step": 795 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 1.1800439357757568, | |
| "learning_rate": 4.274232834777782e-05, | |
| "loss": 1.9722, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 1.1313499212265015, | |
| "learning_rate": 4.2656063798555515e-05, | |
| "loss": 1.9176, | |
| "step": 805 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 1.137534737586975, | |
| "learning_rate": 4.256937777529815e-05, | |
| "loss": 1.9929, | |
| "step": 810 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 1.0575093030929565, | |
| "learning_rate": 4.2482272347317906e-05, | |
| "loss": 2.166, | |
| "step": 815 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "grad_norm": 1.5939594507217407, | |
| "learning_rate": 4.2394749593938733e-05, | |
| "loss": 2.1334, | |
| "step": 820 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "grad_norm": 1.1045507192611694, | |
| "learning_rate": 4.230681160444669e-05, | |
| "loss": 2.0853, | |
| "step": 825 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "grad_norm": 1.3480136394500732, | |
| "learning_rate": 4.221846047804009e-05, | |
| "loss": 2.1802, | |
| "step": 830 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "grad_norm": 1.1822657585144043, | |
| "learning_rate": 4.2129698323779366e-05, | |
| "loss": 2.0739, | |
| "step": 835 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "grad_norm": 1.1771117448806763, | |
| "learning_rate": 4.204052726053676e-05, | |
| "loss": 2.0238, | |
| "step": 840 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "grad_norm": 1.4757814407348633, | |
| "learning_rate": 4.195094941694571e-05, | |
| "loss": 2.1557, | |
| "step": 845 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "grad_norm": 0.9095075726509094, | |
| "learning_rate": 4.1860966931350054e-05, | |
| "loss": 2.1666, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "grad_norm": 1.1039543151855469, | |
| "learning_rate": 4.1770581951752976e-05, | |
| "loss": 2.105, | |
| "step": 855 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "grad_norm": 0.8517205119132996, | |
| "learning_rate": 4.1679796635765735e-05, | |
| "loss": 1.9656, | |
| "step": 860 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "grad_norm": 1.239492654800415, | |
| "learning_rate": 4.158861315055617e-05, | |
| "loss": 2.0166, | |
| "step": 865 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "grad_norm": 1.1358321905136108, | |
| "learning_rate": 4.1497033672796924e-05, | |
| "loss": 2.0076, | |
| "step": 870 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "grad_norm": 1.6215249300003052, | |
| "learning_rate": 4.140506038861356e-05, | |
| "loss": 2.1594, | |
| "step": 875 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "grad_norm": 1.0528080463409424, | |
| "learning_rate": 4.131269549353229e-05, | |
| "loss": 2.1416, | |
| "step": 880 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "grad_norm": 0.8976901769638062, | |
| "learning_rate": 4.1219941192427644e-05, | |
| "loss": 2.1242, | |
| "step": 885 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "grad_norm": 1.263594388961792, | |
| "learning_rate": 4.112679969946977e-05, | |
| "loss": 2.02, | |
| "step": 890 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "grad_norm": 1.4173017740249634, | |
| "learning_rate": 4.103327323807162e-05, | |
| "loss": 2.0438, | |
| "step": 895 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "grad_norm": 1.876170039176941, | |
| "learning_rate": 4.093936404083585e-05, | |
| "loss": 1.9806, | |
| "step": 900 | |
| } | |
| ], | |
| "logging_steps": 5, | |
| "max_steps": 3215, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 100, | |
| "total_flos": 1.210101614051328e+16, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |