| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.4618937644341801, | |
| "eval_steps": 500, | |
| "global_step": 1000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.004618937644341801, | |
| "grad_norm": 11.009225845336914, | |
| "learning_rate": 5.773672055427253e-07, | |
| "loss": 2.4977, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.009237875288683603, | |
| "grad_norm": 9.019889831542969, | |
| "learning_rate": 1.1547344110854505e-06, | |
| "loss": 2.7655, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.013856812933025405, | |
| "grad_norm": 11.212933540344238, | |
| "learning_rate": 1.7321016166281756e-06, | |
| "loss": 2.7241, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.018475750577367205, | |
| "grad_norm": 9.982237815856934, | |
| "learning_rate": 2.309468822170901e-06, | |
| "loss": 2.8357, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.023094688221709007, | |
| "grad_norm": 16.251842498779297, | |
| "learning_rate": 2.886836027713626e-06, | |
| "loss": 2.1159, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.02771362586605081, | |
| "grad_norm": 7.950500011444092, | |
| "learning_rate": 3.464203233256351e-06, | |
| "loss": 1.498, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.03233256351039261, | |
| "grad_norm": 4.816328048706055, | |
| "learning_rate": 4.041570438799077e-06, | |
| "loss": 0.9835, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.03695150115473441, | |
| "grad_norm": 9.529770851135254, | |
| "learning_rate": 4.618937644341802e-06, | |
| "loss": 1.0009, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.04157043879907621, | |
| "grad_norm": 12.515681266784668, | |
| "learning_rate": 5.196304849884527e-06, | |
| "loss": 1.0264, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.046189376443418015, | |
| "grad_norm": 5.393396854400635, | |
| "learning_rate": 5.773672055427252e-06, | |
| "loss": 0.8218, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.050808314087759814, | |
| "grad_norm": 2.777273416519165, | |
| "learning_rate": 6.351039260969978e-06, | |
| "loss": 0.8719, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.05542725173210162, | |
| "grad_norm": 3.8990070819854736, | |
| "learning_rate": 6.928406466512702e-06, | |
| "loss": 0.8592, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.06004618937644342, | |
| "grad_norm": 4.342905521392822, | |
| "learning_rate": 7.505773672055427e-06, | |
| "loss": 0.7982, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.06466512702078522, | |
| "grad_norm": 2.4967517852783203, | |
| "learning_rate": 8.083140877598153e-06, | |
| "loss": 0.7203, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.06928406466512702, | |
| "grad_norm": 5.491476058959961, | |
| "learning_rate": 8.660508083140878e-06, | |
| "loss": 0.7647, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.07390300230946882, | |
| "grad_norm": 4.13068962097168, | |
| "learning_rate": 9.237875288683604e-06, | |
| "loss": 0.7378, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.07852193995381063, | |
| "grad_norm": 3.9301445484161377, | |
| "learning_rate": 9.815242494226329e-06, | |
| "loss": 0.7355, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.08314087759815242, | |
| "grad_norm": 2.891854763031006, | |
| "learning_rate": 1.0392609699769053e-05, | |
| "loss": 0.6911, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.08775981524249422, | |
| "grad_norm": 5.377829074859619, | |
| "learning_rate": 1.0969976905311778e-05, | |
| "loss": 0.7972, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.09237875288683603, | |
| "grad_norm": 3.1744544506073, | |
| "learning_rate": 1.1547344110854504e-05, | |
| "loss": 0.6792, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.09699769053117784, | |
| "grad_norm": 4.408601760864258, | |
| "learning_rate": 1.2124711316397229e-05, | |
| "loss": 0.7378, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.10161662817551963, | |
| "grad_norm": 3.3633627891540527, | |
| "learning_rate": 1.2702078521939955e-05, | |
| "loss": 0.7131, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.10623556581986143, | |
| "grad_norm": 5.786050319671631, | |
| "learning_rate": 1.3279445727482678e-05, | |
| "loss": 0.6806, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.11085450346420324, | |
| "grad_norm": 6.865295886993408, | |
| "learning_rate": 1.3856812933025404e-05, | |
| "loss": 0.7117, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.11547344110854503, | |
| "grad_norm": 3.714210033416748, | |
| "learning_rate": 1.4434180138568129e-05, | |
| "loss": 0.6354, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.12009237875288684, | |
| "grad_norm": 5.501813888549805, | |
| "learning_rate": 1.5011547344110854e-05, | |
| "loss": 0.6266, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.12471131639722864, | |
| "grad_norm": 6.622759819030762, | |
| "learning_rate": 1.558891454965358e-05, | |
| "loss": 0.6894, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.12933025404157045, | |
| "grad_norm": 3.647956609725952, | |
| "learning_rate": 1.6166281755196306e-05, | |
| "loss": 0.66, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.13394919168591224, | |
| "grad_norm": 4.103866100311279, | |
| "learning_rate": 1.674364896073903e-05, | |
| "loss": 0.665, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.13856812933025403, | |
| "grad_norm": 3.84169340133667, | |
| "learning_rate": 1.7321016166281756e-05, | |
| "loss": 0.6676, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.14318706697459585, | |
| "grad_norm": 8.042781829833984, | |
| "learning_rate": 1.789838337182448e-05, | |
| "loss": 0.6236, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 0.14780600461893764, | |
| "grad_norm": 4.35403299331665, | |
| "learning_rate": 1.8475750577367208e-05, | |
| "loss": 0.6283, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 0.15242494226327943, | |
| "grad_norm": 3.7114272117614746, | |
| "learning_rate": 1.9053117782909933e-05, | |
| "loss": 0.6472, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 0.15704387990762125, | |
| "grad_norm": 2.5825653076171875, | |
| "learning_rate": 1.9630484988452657e-05, | |
| "loss": 0.6139, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 0.16166281755196305, | |
| "grad_norm": 4.71831750869751, | |
| "learning_rate": 2.0207852193995382e-05, | |
| "loss": 0.6019, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.16628175519630484, | |
| "grad_norm": 7.264115333557129, | |
| "learning_rate": 2.0785219399538107e-05, | |
| "loss": 0.6484, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 0.17090069284064666, | |
| "grad_norm": 5.484978199005127, | |
| "learning_rate": 2.1362586605080835e-05, | |
| "loss": 0.6423, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 0.17551963048498845, | |
| "grad_norm": 4.004003524780273, | |
| "learning_rate": 2.1939953810623556e-05, | |
| "loss": 0.6167, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 0.18013856812933027, | |
| "grad_norm": 3.858074426651001, | |
| "learning_rate": 2.251732101616628e-05, | |
| "loss": 0.6029, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 0.18475750577367206, | |
| "grad_norm": 4.84448766708374, | |
| "learning_rate": 2.309468822170901e-05, | |
| "loss": 0.5511, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.18937644341801385, | |
| "grad_norm": 3.419329881668091, | |
| "learning_rate": 2.3672055427251733e-05, | |
| "loss": 0.5919, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 0.19399538106235567, | |
| "grad_norm": 2.367598295211792, | |
| "learning_rate": 2.4249422632794458e-05, | |
| "loss": 0.5407, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 0.19861431870669746, | |
| "grad_norm": 4.016935348510742, | |
| "learning_rate": 2.4826789838337182e-05, | |
| "loss": 0.5829, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 0.20323325635103925, | |
| "grad_norm": 4.788145065307617, | |
| "learning_rate": 2.540415704387991e-05, | |
| "loss": 0.593, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 0.20785219399538107, | |
| "grad_norm": 3.618800401687622, | |
| "learning_rate": 2.5981524249422635e-05, | |
| "loss": 0.56, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 0.21247113163972287, | |
| "grad_norm": 3.6879987716674805, | |
| "learning_rate": 2.6558891454965356e-05, | |
| "loss": 0.5448, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 0.21709006928406466, | |
| "grad_norm": 3.6736364364624023, | |
| "learning_rate": 2.7136258660508084e-05, | |
| "loss": 0.5707, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 0.22170900692840648, | |
| "grad_norm": 3.954604387283325, | |
| "learning_rate": 2.771362586605081e-05, | |
| "loss": 0.5891, | |
| "step": 480 | |
| }, | |
| { | |
| "epoch": 0.22632794457274827, | |
| "grad_norm": 6.050353050231934, | |
| "learning_rate": 2.8290993071593537e-05, | |
| "loss": 0.6231, | |
| "step": 490 | |
| }, | |
| { | |
| "epoch": 0.23094688221709006, | |
| "grad_norm": 3.728318214416504, | |
| "learning_rate": 2.8868360277136258e-05, | |
| "loss": 0.6294, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.23556581986143188, | |
| "grad_norm": 5.047093391418457, | |
| "learning_rate": 2.9445727482678986e-05, | |
| "loss": 0.5795, | |
| "step": 510 | |
| }, | |
| { | |
| "epoch": 0.24018475750577367, | |
| "grad_norm": 4.392754077911377, | |
| "learning_rate": 3.0023094688221707e-05, | |
| "loss": 0.5634, | |
| "step": 520 | |
| }, | |
| { | |
| "epoch": 0.24480369515011546, | |
| "grad_norm": 3.634922504425049, | |
| "learning_rate": 3.060046189376444e-05, | |
| "loss": 0.5372, | |
| "step": 530 | |
| }, | |
| { | |
| "epoch": 0.24942263279445728, | |
| "grad_norm": 3.2436106204986572, | |
| "learning_rate": 3.117782909930716e-05, | |
| "loss": 0.5914, | |
| "step": 540 | |
| }, | |
| { | |
| "epoch": 0.2540415704387991, | |
| "grad_norm": 4.555751323699951, | |
| "learning_rate": 3.175519630484989e-05, | |
| "loss": 0.5767, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 0.2586605080831409, | |
| "grad_norm": 3.359252452850342, | |
| "learning_rate": 3.233256351039261e-05, | |
| "loss": 0.5569, | |
| "step": 560 | |
| }, | |
| { | |
| "epoch": 0.2632794457274827, | |
| "grad_norm": 3.362272262573242, | |
| "learning_rate": 3.290993071593534e-05, | |
| "loss": 0.6248, | |
| "step": 570 | |
| }, | |
| { | |
| "epoch": 0.2678983833718245, | |
| "grad_norm": 3.1555299758911133, | |
| "learning_rate": 3.348729792147806e-05, | |
| "loss": 0.5629, | |
| "step": 580 | |
| }, | |
| { | |
| "epoch": 0.27251732101616627, | |
| "grad_norm": 4.786733627319336, | |
| "learning_rate": 3.4064665127020787e-05, | |
| "loss": 0.5942, | |
| "step": 590 | |
| }, | |
| { | |
| "epoch": 0.27713625866050806, | |
| "grad_norm": 4.290501117706299, | |
| "learning_rate": 3.464203233256351e-05, | |
| "loss": 0.5694, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.2817551963048499, | |
| "grad_norm": 2.2575786113739014, | |
| "learning_rate": 3.5219399538106236e-05, | |
| "loss": 0.5459, | |
| "step": 610 | |
| }, | |
| { | |
| "epoch": 0.2863741339491917, | |
| "grad_norm": 4.7077484130859375, | |
| "learning_rate": 3.579676674364896e-05, | |
| "loss": 0.5686, | |
| "step": 620 | |
| }, | |
| { | |
| "epoch": 0.2909930715935335, | |
| "grad_norm": 3.487034559249878, | |
| "learning_rate": 3.6374133949191685e-05, | |
| "loss": 0.5185, | |
| "step": 630 | |
| }, | |
| { | |
| "epoch": 0.2956120092378753, | |
| "grad_norm": 4.230603218078613, | |
| "learning_rate": 3.6951501154734416e-05, | |
| "loss": 0.5548, | |
| "step": 640 | |
| }, | |
| { | |
| "epoch": 0.3002309468822171, | |
| "grad_norm": 3.5367519855499268, | |
| "learning_rate": 3.7528868360277134e-05, | |
| "loss": 0.5658, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 0.30484988452655887, | |
| "grad_norm": 3.3856465816497803, | |
| "learning_rate": 3.8106235565819866e-05, | |
| "loss": 0.5459, | |
| "step": 660 | |
| }, | |
| { | |
| "epoch": 0.3094688221709007, | |
| "grad_norm": 3.4272944927215576, | |
| "learning_rate": 3.868360277136259e-05, | |
| "loss": 0.5823, | |
| "step": 670 | |
| }, | |
| { | |
| "epoch": 0.3140877598152425, | |
| "grad_norm": 2.7848410606384277, | |
| "learning_rate": 3.9260969976905315e-05, | |
| "loss": 0.5481, | |
| "step": 680 | |
| }, | |
| { | |
| "epoch": 0.3187066974595843, | |
| "grad_norm": 2.8584043979644775, | |
| "learning_rate": 3.983833718244804e-05, | |
| "loss": 0.4907, | |
| "step": 690 | |
| }, | |
| { | |
| "epoch": 0.3233256351039261, | |
| "grad_norm": 3.9872729778289795, | |
| "learning_rate": 4.0415704387990764e-05, | |
| "loss": 0.5703, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.3279445727482679, | |
| "grad_norm": 4.052183151245117, | |
| "learning_rate": 4.099307159353349e-05, | |
| "loss": 0.6229, | |
| "step": 710 | |
| }, | |
| { | |
| "epoch": 0.3325635103926097, | |
| "grad_norm": 3.6171059608459473, | |
| "learning_rate": 4.1570438799076213e-05, | |
| "loss": 0.5022, | |
| "step": 720 | |
| }, | |
| { | |
| "epoch": 0.3371824480369515, | |
| "grad_norm": 3.4360239505767822, | |
| "learning_rate": 4.214780600461894e-05, | |
| "loss": 0.5642, | |
| "step": 730 | |
| }, | |
| { | |
| "epoch": 0.3418013856812933, | |
| "grad_norm": 3.328563928604126, | |
| "learning_rate": 4.272517321016167e-05, | |
| "loss": 0.543, | |
| "step": 740 | |
| }, | |
| { | |
| "epoch": 0.3464203233256351, | |
| "grad_norm": 2.494722843170166, | |
| "learning_rate": 4.330254041570439e-05, | |
| "loss": 0.5056, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 0.3510392609699769, | |
| "grad_norm": 2.9856996536254883, | |
| "learning_rate": 4.387990762124711e-05, | |
| "loss": 0.5215, | |
| "step": 760 | |
| }, | |
| { | |
| "epoch": 0.3556581986143187, | |
| "grad_norm": 3.6510608196258545, | |
| "learning_rate": 4.445727482678984e-05, | |
| "loss": 0.5418, | |
| "step": 770 | |
| }, | |
| { | |
| "epoch": 0.36027713625866054, | |
| "grad_norm": 2.837355136871338, | |
| "learning_rate": 4.503464203233256e-05, | |
| "loss": 0.5197, | |
| "step": 780 | |
| }, | |
| { | |
| "epoch": 0.3648960739030023, | |
| "grad_norm": 2.4272849559783936, | |
| "learning_rate": 4.561200923787529e-05, | |
| "loss": 0.5151, | |
| "step": 790 | |
| }, | |
| { | |
| "epoch": 0.3695150115473441, | |
| "grad_norm": 2.2198147773742676, | |
| "learning_rate": 4.618937644341802e-05, | |
| "loss": 0.5037, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 0.3741339491916859, | |
| "grad_norm": 3.082343816757202, | |
| "learning_rate": 4.676674364896074e-05, | |
| "loss": 0.5411, | |
| "step": 810 | |
| }, | |
| { | |
| "epoch": 0.3787528868360277, | |
| "grad_norm": 2.739429235458374, | |
| "learning_rate": 4.7344110854503466e-05, | |
| "loss": 0.5034, | |
| "step": 820 | |
| }, | |
| { | |
| "epoch": 0.3833718244803695, | |
| "grad_norm": 2.8211052417755127, | |
| "learning_rate": 4.792147806004619e-05, | |
| "loss": 0.5077, | |
| "step": 830 | |
| }, | |
| { | |
| "epoch": 0.38799076212471134, | |
| "grad_norm": 4.462055683135986, | |
| "learning_rate": 4.8498845265588916e-05, | |
| "loss": 0.4883, | |
| "step": 840 | |
| }, | |
| { | |
| "epoch": 0.39260969976905313, | |
| "grad_norm": 2.8090898990631104, | |
| "learning_rate": 4.907621247113165e-05, | |
| "loss": 0.4918, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 0.3972286374133949, | |
| "grad_norm": 2.787473440170288, | |
| "learning_rate": 4.9653579676674365e-05, | |
| "loss": 0.4736, | |
| "step": 860 | |
| }, | |
| { | |
| "epoch": 0.4018475750577367, | |
| "grad_norm": 3.198763132095337, | |
| "learning_rate": 4.999996750557592e-05, | |
| "loss": 0.5243, | |
| "step": 870 | |
| }, | |
| { | |
| "epoch": 0.4064665127020785, | |
| "grad_norm": 3.8778421878814697, | |
| "learning_rate": 4.9999601944275146e-05, | |
| "loss": 0.5206, | |
| "step": 880 | |
| }, | |
| { | |
| "epoch": 0.4110854503464203, | |
| "grad_norm": 4.326559066772461, | |
| "learning_rate": 4.999883020960263e-05, | |
| "loss": 0.5998, | |
| "step": 890 | |
| }, | |
| { | |
| "epoch": 0.41570438799076215, | |
| "grad_norm": 2.1009411811828613, | |
| "learning_rate": 4.99976523140969e-05, | |
| "loss": 0.5097, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 0.42032332563510394, | |
| "grad_norm": 3.0838584899902344, | |
| "learning_rate": 4.999606827689545e-05, | |
| "loss": 0.4699, | |
| "step": 910 | |
| }, | |
| { | |
| "epoch": 0.42494226327944573, | |
| "grad_norm": 3.067430019378662, | |
| "learning_rate": 4.999407812373445e-05, | |
| "loss": 0.5223, | |
| "step": 920 | |
| }, | |
| { | |
| "epoch": 0.4295612009237875, | |
| "grad_norm": 2.580418109893799, | |
| "learning_rate": 4.9991681886948297e-05, | |
| "loss": 0.5396, | |
| "step": 930 | |
| }, | |
| { | |
| "epoch": 0.4341801385681293, | |
| "grad_norm": 2.748032808303833, | |
| "learning_rate": 4.9988879605469106e-05, | |
| "loss": 0.4965, | |
| "step": 940 | |
| }, | |
| { | |
| "epoch": 0.4387990762124711, | |
| "grad_norm": 2.841571569442749, | |
| "learning_rate": 4.99856713248261e-05, | |
| "loss": 0.472, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 0.44341801385681295, | |
| "grad_norm": 1.8557236194610596, | |
| "learning_rate": 4.9982057097144826e-05, | |
| "loss": 0.4878, | |
| "step": 960 | |
| }, | |
| { | |
| "epoch": 0.44803695150115475, | |
| "grad_norm": 5.707876205444336, | |
| "learning_rate": 4.9978036981146344e-05, | |
| "loss": 0.4883, | |
| "step": 970 | |
| }, | |
| { | |
| "epoch": 0.45265588914549654, | |
| "grad_norm": 3.867081880569458, | |
| "learning_rate": 4.9973611042146254e-05, | |
| "loss": 0.4929, | |
| "step": 980 | |
| }, | |
| { | |
| "epoch": 0.45727482678983833, | |
| "grad_norm": 2.291074275970459, | |
| "learning_rate": 4.996877935205365e-05, | |
| "loss": 0.5046, | |
| "step": 990 | |
| }, | |
| { | |
| "epoch": 0.4618937644341801, | |
| "grad_norm": 2.332185745239258, | |
| "learning_rate": 4.996354198936993e-05, | |
| "loss": 0.4539, | |
| "step": 1000 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 8660, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 4, | |
| "save_steps": 500, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 2.714316759199908e+17, | |
| "train_batch_size": 1, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |