| { | |
| "best_global_step": null, | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.18066847335140018, | |
| "eval_steps": 500, | |
| "global_step": 100, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.0018066847335140017, | |
| "grad_norm": 2.8337757951637657, | |
| "learning_rate": 0.0, | |
| "loss": 0.8251, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.0036133694670280035, | |
| "grad_norm": 2.8815146894381005, | |
| "learning_rate": 1.7857142857142858e-07, | |
| "loss": 0.8284, | |
| "step": 2 | |
| }, | |
| { | |
| "epoch": 0.005420054200542005, | |
| "grad_norm": 2.888366333138549, | |
| "learning_rate": 3.5714285714285716e-07, | |
| "loss": 0.8432, | |
| "step": 3 | |
| }, | |
| { | |
| "epoch": 0.007226738934056007, | |
| "grad_norm": 2.8554157896663708, | |
| "learning_rate": 5.357142857142857e-07, | |
| "loss": 0.8398, | |
| "step": 4 | |
| }, | |
| { | |
| "epoch": 0.009033423667570008, | |
| "grad_norm": 2.7996302090063954, | |
| "learning_rate": 7.142857142857143e-07, | |
| "loss": 0.8303, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.01084010840108401, | |
| "grad_norm": 2.758540000006854, | |
| "learning_rate": 8.928571428571429e-07, | |
| "loss": 0.8306, | |
| "step": 6 | |
| }, | |
| { | |
| "epoch": 0.012646793134598013, | |
| "grad_norm": 2.7440712027687453, | |
| "learning_rate": 1.0714285714285714e-06, | |
| "loss": 0.8129, | |
| "step": 7 | |
| }, | |
| { | |
| "epoch": 0.014453477868112014, | |
| "grad_norm": 2.789670841276354, | |
| "learning_rate": 1.25e-06, | |
| "loss": 0.8217, | |
| "step": 8 | |
| }, | |
| { | |
| "epoch": 0.016260162601626018, | |
| "grad_norm": 2.637271873271262, | |
| "learning_rate": 1.4285714285714286e-06, | |
| "loss": 0.8192, | |
| "step": 9 | |
| }, | |
| { | |
| "epoch": 0.018066847335140017, | |
| "grad_norm": 2.6274284830887247, | |
| "learning_rate": 1.6071428571428574e-06, | |
| "loss": 0.8146, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.01987353206865402, | |
| "grad_norm": 2.2614743320388992, | |
| "learning_rate": 1.7857142857142859e-06, | |
| "loss": 0.8022, | |
| "step": 11 | |
| }, | |
| { | |
| "epoch": 0.02168021680216802, | |
| "grad_norm": 2.1218891806221882, | |
| "learning_rate": 1.9642857142857144e-06, | |
| "loss": 0.8131, | |
| "step": 12 | |
| }, | |
| { | |
| "epoch": 0.023486901535682024, | |
| "grad_norm": 2.1362019297809636, | |
| "learning_rate": 2.1428571428571427e-06, | |
| "loss": 0.8047, | |
| "step": 13 | |
| }, | |
| { | |
| "epoch": 0.025293586269196026, | |
| "grad_norm": 2.043181941163786, | |
| "learning_rate": 2.321428571428572e-06, | |
| "loss": 0.8006, | |
| "step": 14 | |
| }, | |
| { | |
| "epoch": 0.02710027100271003, | |
| "grad_norm": 1.4654153096579285, | |
| "learning_rate": 2.5e-06, | |
| "loss": 0.7863, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.028906955736224028, | |
| "grad_norm": 1.424950643183691, | |
| "learning_rate": 2.6785714285714285e-06, | |
| "loss": 0.7794, | |
| "step": 16 | |
| }, | |
| { | |
| "epoch": 0.03071364046973803, | |
| "grad_norm": 1.371800886806853, | |
| "learning_rate": 2.8571428571428573e-06, | |
| "loss": 0.7871, | |
| "step": 17 | |
| }, | |
| { | |
| "epoch": 0.032520325203252036, | |
| "grad_norm": 1.2910993566284936, | |
| "learning_rate": 3.0357142857142856e-06, | |
| "loss": 0.7792, | |
| "step": 18 | |
| }, | |
| { | |
| "epoch": 0.03432700993676603, | |
| "grad_norm": 1.2099658145135412, | |
| "learning_rate": 3.2142857142857147e-06, | |
| "loss": 0.7728, | |
| "step": 19 | |
| }, | |
| { | |
| "epoch": 0.036133694670280034, | |
| "grad_norm": 1.0479474035735314, | |
| "learning_rate": 3.3928571428571435e-06, | |
| "loss": 0.743, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.037940379403794036, | |
| "grad_norm": 1.112276534792924, | |
| "learning_rate": 3.5714285714285718e-06, | |
| "loss": 0.7403, | |
| "step": 21 | |
| }, | |
| { | |
| "epoch": 0.03974706413730804, | |
| "grad_norm": 1.3035077079579298, | |
| "learning_rate": 3.7500000000000005e-06, | |
| "loss": 0.7296, | |
| "step": 22 | |
| }, | |
| { | |
| "epoch": 0.04155374887082204, | |
| "grad_norm": 1.3695661288387202, | |
| "learning_rate": 3.928571428571429e-06, | |
| "loss": 0.7302, | |
| "step": 23 | |
| }, | |
| { | |
| "epoch": 0.04336043360433604, | |
| "grad_norm": 1.2631605637070566, | |
| "learning_rate": 4.107142857142857e-06, | |
| "loss": 0.7359, | |
| "step": 24 | |
| }, | |
| { | |
| "epoch": 0.045167118337850046, | |
| "grad_norm": 0.9577414796019008, | |
| "learning_rate": 4.2857142857142855e-06, | |
| "loss": 0.713, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.04697380307136405, | |
| "grad_norm": 0.7548293526053346, | |
| "learning_rate": 4.464285714285715e-06, | |
| "loss": 0.7275, | |
| "step": 26 | |
| }, | |
| { | |
| "epoch": 0.04878048780487805, | |
| "grad_norm": 0.6708116039946823, | |
| "learning_rate": 4.642857142857144e-06, | |
| "loss": 0.7021, | |
| "step": 27 | |
| }, | |
| { | |
| "epoch": 0.05058717253839205, | |
| "grad_norm": 0.7960055478682195, | |
| "learning_rate": 4.821428571428572e-06, | |
| "loss": 0.6998, | |
| "step": 28 | |
| }, | |
| { | |
| "epoch": 0.052393857271906055, | |
| "grad_norm": 0.8790784871024999, | |
| "learning_rate": 5e-06, | |
| "loss": 0.7086, | |
| "step": 29 | |
| }, | |
| { | |
| "epoch": 0.05420054200542006, | |
| "grad_norm": 0.9579979234005123, | |
| "learning_rate": 4.999955240022903e-06, | |
| "loss": 0.7065, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.05600722673893405, | |
| "grad_norm": 1.003933265230055, | |
| "learning_rate": 4.999820961694372e-06, | |
| "loss": 0.6937, | |
| "step": 31 | |
| }, | |
| { | |
| "epoch": 0.057813911472448055, | |
| "grad_norm": 0.6710805038763018, | |
| "learning_rate": 4.999597169822646e-06, | |
| "loss": 0.6812, | |
| "step": 32 | |
| }, | |
| { | |
| "epoch": 0.05962059620596206, | |
| "grad_norm": 0.5001839964623817, | |
| "learning_rate": 4.999283872421259e-06, | |
| "loss": 0.6911, | |
| "step": 33 | |
| }, | |
| { | |
| "epoch": 0.06142728093947606, | |
| "grad_norm": 0.5681434362593133, | |
| "learning_rate": 4.998881080708759e-06, | |
| "loss": 0.6766, | |
| "step": 34 | |
| }, | |
| { | |
| "epoch": 0.06323396567299007, | |
| "grad_norm": 0.5752663504650961, | |
| "learning_rate": 4.998388809108304e-06, | |
| "loss": 0.6816, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.06504065040650407, | |
| "grad_norm": 0.4990120714898939, | |
| "learning_rate": 4.997807075247147e-06, | |
| "loss": 0.6754, | |
| "step": 36 | |
| }, | |
| { | |
| "epoch": 0.06684733514001806, | |
| "grad_norm": 0.4974519304946596, | |
| "learning_rate": 4.997135899956002e-06, | |
| "loss": 0.6782, | |
| "step": 37 | |
| }, | |
| { | |
| "epoch": 0.06865401987353206, | |
| "grad_norm": 0.4349874019011085, | |
| "learning_rate": 4.996375307268303e-06, | |
| "loss": 0.6709, | |
| "step": 38 | |
| }, | |
| { | |
| "epoch": 0.07046070460704607, | |
| "grad_norm": 0.48312199739205014, | |
| "learning_rate": 4.995525324419338e-06, | |
| "loss": 0.6707, | |
| "step": 39 | |
| }, | |
| { | |
| "epoch": 0.07226738934056007, | |
| "grad_norm": 0.3760455090466998, | |
| "learning_rate": 4.994585981845278e-06, | |
| "loss": 0.6592, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.07407407407407407, | |
| "grad_norm": 0.362711899175025, | |
| "learning_rate": 4.993557313182086e-06, | |
| "loss": 0.6666, | |
| "step": 41 | |
| }, | |
| { | |
| "epoch": 0.07588075880758807, | |
| "grad_norm": 0.3527864788229306, | |
| "learning_rate": 4.992439355264308e-06, | |
| "loss": 0.656, | |
| "step": 42 | |
| }, | |
| { | |
| "epoch": 0.07768744354110207, | |
| "grad_norm": 0.37039474310454246, | |
| "learning_rate": 4.9912321481237616e-06, | |
| "loss": 0.6567, | |
| "step": 43 | |
| }, | |
| { | |
| "epoch": 0.07949412827461608, | |
| "grad_norm": 0.3381704985127657, | |
| "learning_rate": 4.989935734988098e-06, | |
| "loss": 0.6492, | |
| "step": 44 | |
| }, | |
| { | |
| "epoch": 0.08130081300813008, | |
| "grad_norm": 0.3368799245864057, | |
| "learning_rate": 4.988550162279255e-06, | |
| "loss": 0.6558, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.08310749774164408, | |
| "grad_norm": 0.33217201284358455, | |
| "learning_rate": 4.9870754796117956e-06, | |
| "loss": 0.658, | |
| "step": 46 | |
| }, | |
| { | |
| "epoch": 0.08491418247515808, | |
| "grad_norm": 0.3203999210132069, | |
| "learning_rate": 4.985511739791129e-06, | |
| "loss": 0.6596, | |
| "step": 47 | |
| }, | |
| { | |
| "epoch": 0.08672086720867209, | |
| "grad_norm": 0.3108831280349791, | |
| "learning_rate": 4.983858998811622e-06, | |
| "loss": 0.6465, | |
| "step": 48 | |
| }, | |
| { | |
| "epoch": 0.08852755194218609, | |
| "grad_norm": 0.28944557461245024, | |
| "learning_rate": 4.982117315854594e-06, | |
| "loss": 0.651, | |
| "step": 49 | |
| }, | |
| { | |
| "epoch": 0.09033423667570009, | |
| "grad_norm": 0.2724500694923891, | |
| "learning_rate": 4.980286753286196e-06, | |
| "loss": 0.6493, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.0921409214092141, | |
| "grad_norm": 0.29233908216498683, | |
| "learning_rate": 4.978367376655177e-06, | |
| "loss": 0.6391, | |
| "step": 51 | |
| }, | |
| { | |
| "epoch": 0.0939476061427281, | |
| "grad_norm": 0.29050327837368234, | |
| "learning_rate": 4.976359254690543e-06, | |
| "loss": 0.6577, | |
| "step": 52 | |
| }, | |
| { | |
| "epoch": 0.0957542908762421, | |
| "grad_norm": 0.26277310012199956, | |
| "learning_rate": 4.974262459299088e-06, | |
| "loss": 0.641, | |
| "step": 53 | |
| }, | |
| { | |
| "epoch": 0.0975609756097561, | |
| "grad_norm": 0.2694317681957786, | |
| "learning_rate": 4.9720770655628216e-06, | |
| "loss": 0.6322, | |
| "step": 54 | |
| }, | |
| { | |
| "epoch": 0.0993676603432701, | |
| "grad_norm": 0.27643434259676186, | |
| "learning_rate": 4.969803151736285e-06, | |
| "loss": 0.6446, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.1011743450767841, | |
| "grad_norm": 0.27938135134555137, | |
| "learning_rate": 4.967440799243739e-06, | |
| "loss": 0.6505, | |
| "step": 56 | |
| }, | |
| { | |
| "epoch": 0.10298102981029811, | |
| "grad_norm": 0.2723254558437683, | |
| "learning_rate": 4.964990092676263e-06, | |
| "loss": 0.6358, | |
| "step": 57 | |
| }, | |
| { | |
| "epoch": 0.10478771454381211, | |
| "grad_norm": 0.2476211836425744, | |
| "learning_rate": 4.962451119788709e-06, | |
| "loss": 0.6435, | |
| "step": 58 | |
| }, | |
| { | |
| "epoch": 0.10659439927732611, | |
| "grad_norm": 0.2385367032602417, | |
| "learning_rate": 4.959823971496575e-06, | |
| "loss": 0.6289, | |
| "step": 59 | |
| }, | |
| { | |
| "epoch": 0.10840108401084012, | |
| "grad_norm": 0.24895994998658083, | |
| "learning_rate": 4.957108741872736e-06, | |
| "loss": 0.6299, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.1102077687443541, | |
| "grad_norm": 0.28129654057596637, | |
| "learning_rate": 4.954305528144085e-06, | |
| "loss": 0.6339, | |
| "step": 61 | |
| }, | |
| { | |
| "epoch": 0.1120144534778681, | |
| "grad_norm": 0.25373425749505557, | |
| "learning_rate": 4.9514144306880506e-06, | |
| "loss": 0.6393, | |
| "step": 62 | |
| }, | |
| { | |
| "epoch": 0.11382113821138211, | |
| "grad_norm": 0.2583299024885213, | |
| "learning_rate": 4.948435553028994e-06, | |
| "loss": 0.6383, | |
| "step": 63 | |
| }, | |
| { | |
| "epoch": 0.11562782294489611, | |
| "grad_norm": 0.24423529044993955, | |
| "learning_rate": 4.9453690018345144e-06, | |
| "loss": 0.6359, | |
| "step": 64 | |
| }, | |
| { | |
| "epoch": 0.11743450767841011, | |
| "grad_norm": 0.24605558051142096, | |
| "learning_rate": 4.942214886911619e-06, | |
| "loss": 0.631, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.11924119241192412, | |
| "grad_norm": 0.2548336350045181, | |
| "learning_rate": 4.938973321202799e-06, | |
| "loss": 0.6323, | |
| "step": 66 | |
| }, | |
| { | |
| "epoch": 0.12104787714543812, | |
| "grad_norm": 0.32845268505362657, | |
| "learning_rate": 4.935644420781978e-06, | |
| "loss": 0.6229, | |
| "step": 67 | |
| }, | |
| { | |
| "epoch": 0.12285456187895212, | |
| "grad_norm": 0.25203625601717533, | |
| "learning_rate": 4.932228304850363e-06, | |
| "loss": 0.6386, | |
| "step": 68 | |
| }, | |
| { | |
| "epoch": 0.12466124661246612, | |
| "grad_norm": 0.2467239132093227, | |
| "learning_rate": 4.9287250957321685e-06, | |
| "loss": 0.6401, | |
| "step": 69 | |
| }, | |
| { | |
| "epoch": 0.12646793134598014, | |
| "grad_norm": 0.28444822854929847, | |
| "learning_rate": 4.925134918870245e-06, | |
| "loss": 0.646, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.12827461607949414, | |
| "grad_norm": 0.23105312295557145, | |
| "learning_rate": 4.921457902821578e-06, | |
| "loss": 0.6164, | |
| "step": 71 | |
| }, | |
| { | |
| "epoch": 0.13008130081300814, | |
| "grad_norm": 0.23048405537510025, | |
| "learning_rate": 4.917694179252692e-06, | |
| "loss": 0.6342, | |
| "step": 72 | |
| }, | |
| { | |
| "epoch": 0.13188798554652212, | |
| "grad_norm": 0.24150602333720247, | |
| "learning_rate": 4.9138438829349296e-06, | |
| "loss": 0.6332, | |
| "step": 73 | |
| }, | |
| { | |
| "epoch": 0.13369467028003612, | |
| "grad_norm": 0.2462886765781574, | |
| "learning_rate": 4.909907151739634e-06, | |
| "loss": 0.6246, | |
| "step": 74 | |
| }, | |
| { | |
| "epoch": 0.13550135501355012, | |
| "grad_norm": 0.2821908512766841, | |
| "learning_rate": 4.9058841266332005e-06, | |
| "loss": 0.6227, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 0.13730803974706413, | |
| "grad_norm": 0.25172603846639535, | |
| "learning_rate": 4.901774951672041e-06, | |
| "loss": 0.6181, | |
| "step": 76 | |
| }, | |
| { | |
| "epoch": 0.13911472448057813, | |
| "grad_norm": 0.23478283786140364, | |
| "learning_rate": 4.897579773997415e-06, | |
| "loss": 0.6123, | |
| "step": 77 | |
| }, | |
| { | |
| "epoch": 0.14092140921409213, | |
| "grad_norm": 0.24524464756556244, | |
| "learning_rate": 4.893298743830168e-06, | |
| "loss": 0.625, | |
| "step": 78 | |
| }, | |
| { | |
| "epoch": 0.14272809394760613, | |
| "grad_norm": 0.2557156779275681, | |
| "learning_rate": 4.8889320144653525e-06, | |
| "loss": 0.6347, | |
| "step": 79 | |
| }, | |
| { | |
| "epoch": 0.14453477868112014, | |
| "grad_norm": 0.2501383477274707, | |
| "learning_rate": 4.884479742266731e-06, | |
| "loss": 0.6435, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.14634146341463414, | |
| "grad_norm": 0.2777905831842133, | |
| "learning_rate": 4.879942086661185e-06, | |
| "loss": 0.6249, | |
| "step": 81 | |
| }, | |
| { | |
| "epoch": 0.14814814814814814, | |
| "grad_norm": 0.2311051834333071, | |
| "learning_rate": 4.875319210133004e-06, | |
| "loss": 0.6186, | |
| "step": 82 | |
| }, | |
| { | |
| "epoch": 0.14995483288166214, | |
| "grad_norm": 0.24510662627866567, | |
| "learning_rate": 4.870611278218066e-06, | |
| "loss": 0.6301, | |
| "step": 83 | |
| }, | |
| { | |
| "epoch": 0.15176151761517614, | |
| "grad_norm": 0.2457506431539522, | |
| "learning_rate": 4.865818459497911e-06, | |
| "loss": 0.6221, | |
| "step": 84 | |
| }, | |
| { | |
| "epoch": 0.15356820234869015, | |
| "grad_norm": 0.23750506338642635, | |
| "learning_rate": 4.860940925593703e-06, | |
| "loss": 0.6238, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 0.15537488708220415, | |
| "grad_norm": 0.24583995522041222, | |
| "learning_rate": 4.855978851160088e-06, | |
| "loss": 0.6245, | |
| "step": 86 | |
| }, | |
| { | |
| "epoch": 0.15718157181571815, | |
| "grad_norm": 0.24412234277510458, | |
| "learning_rate": 4.850932413878934e-06, | |
| "loss": 0.6266, | |
| "step": 87 | |
| }, | |
| { | |
| "epoch": 0.15898825654923215, | |
| "grad_norm": 0.23692543465028104, | |
| "learning_rate": 4.845801794452978e-06, | |
| "loss": 0.6307, | |
| "step": 88 | |
| }, | |
| { | |
| "epoch": 0.16079494128274616, | |
| "grad_norm": 0.24055583410087153, | |
| "learning_rate": 4.8405871765993435e-06, | |
| "loss": 0.6258, | |
| "step": 89 | |
| }, | |
| { | |
| "epoch": 0.16260162601626016, | |
| "grad_norm": 0.25682681695753845, | |
| "learning_rate": 4.8352887470429726e-06, | |
| "loss": 0.6193, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.16440831074977416, | |
| "grad_norm": 0.2451987238773762, | |
| "learning_rate": 4.829906695509934e-06, | |
| "loss": 0.6175, | |
| "step": 91 | |
| }, | |
| { | |
| "epoch": 0.16621499548328816, | |
| "grad_norm": 0.33308910526145086, | |
| "learning_rate": 4.824441214720629e-06, | |
| "loss": 0.6207, | |
| "step": 92 | |
| }, | |
| { | |
| "epoch": 0.16802168021680217, | |
| "grad_norm": 0.2599636527935745, | |
| "learning_rate": 4.8188925003828945e-06, | |
| "loss": 0.6275, | |
| "step": 93 | |
| }, | |
| { | |
| "epoch": 0.16982836495031617, | |
| "grad_norm": 0.24262017713570547, | |
| "learning_rate": 4.813260751184992e-06, | |
| "loss": 0.6255, | |
| "step": 94 | |
| }, | |
| { | |
| "epoch": 0.17163504968383017, | |
| "grad_norm": 0.24155086390799096, | |
| "learning_rate": 4.8075461687884935e-06, | |
| "loss": 0.6184, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 0.17344173441734417, | |
| "grad_norm": 0.2399226609650259, | |
| "learning_rate": 4.801748957821061e-06, | |
| "loss": 0.6164, | |
| "step": 96 | |
| }, | |
| { | |
| "epoch": 0.17524841915085818, | |
| "grad_norm": 0.22903596103655702, | |
| "learning_rate": 4.795869325869117e-06, | |
| "loss": 0.6229, | |
| "step": 97 | |
| }, | |
| { | |
| "epoch": 0.17705510388437218, | |
| "grad_norm": 0.2307755782529314, | |
| "learning_rate": 4.7899074834704165e-06, | |
| "loss": 0.6173, | |
| "step": 98 | |
| }, | |
| { | |
| "epoch": 0.17886178861788618, | |
| "grad_norm": 0.24823755288051938, | |
| "learning_rate": 4.783863644106502e-06, | |
| "loss": 0.6292, | |
| "step": 99 | |
| }, | |
| { | |
| "epoch": 0.18066847335140018, | |
| "grad_norm": 0.21658543875130712, | |
| "learning_rate": 4.777738024195065e-06, | |
| "loss": 0.6032, | |
| "step": 100 | |
| } | |
| ], | |
| "logging_steps": 1, | |
| "max_steps": 553, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 100, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 115440686923776.0, | |
| "train_batch_size": 2, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |