{
  "best_global_step": 16356,
  "best_metric": 2.342665433883667,
  "best_model_checkpoint": "/common/home/users/m/mq.nguyen.2023/scratchDirectory/finetune/train_nbtpj_summ_gpt2_inst_cnnxsumsam/checkpoint-16356",
  "epoch": 3.0,
  "eval_steps": 500,
  "global_step": 16356,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.009172208209126348,
      "grad_norm": 7.051512241363525,
      "learning_rate": 9.800000000000001e-06,
      "loss": 6.4476,
      "num_tokens": 3070741.0,
      "step": 50
    },
    {
      "epoch": 0.018344416418252695,
      "grad_norm": 6.664697170257568,
      "learning_rate": 1.98e-05,
      "loss": 5.4332,
      "num_tokens": 6150092.0,
      "step": 100
    },
    {
      "epoch": 0.02751662462737904,
      "grad_norm": 5.514298439025879,
      "learning_rate": 1.9999839379410334e-05,
      "loss": 4.9984,
      "num_tokens": 9221796.0,
      "step": 150
    },
    {
      "epoch": 0.03668883283650539,
      "grad_norm": 4.882246494293213,
      "learning_rate": 1.9999344344268732e-05,
      "loss": 4.7556,
      "num_tokens": 12294380.0,
      "step": 200
    },
    {
      "epoch": 0.04586104104563173,
      "grad_norm": 4.8477373123168945,
      "learning_rate": 1.9998514844235922e-05,
      "loss": 4.6238,
      "num_tokens": 15366995.0,
      "step": 250
    },
    {
      "epoch": 0.05503324925475808,
      "grad_norm": 4.187108516693115,
      "learning_rate": 1.9997350907057584e-05,
      "loss": 4.5485,
      "num_tokens": 18447066.0,
      "step": 300
    },
    {
      "epoch": 0.06420545746388442,
      "grad_norm": 5.372704982757568,
      "learning_rate": 1.9995852571665885e-05,
      "loss": 4.4819,
      "num_tokens": 21511375.0,
      "step": 350
    },
    {
      "epoch": 0.07337766567301078,
      "grad_norm": 5.349086761474609,
      "learning_rate": 1.9994019888178163e-05,
      "loss": 4.3881,
      "num_tokens": 24591962.0,
      "step": 400
    },
    {
      "epoch": 0.08254987388213712,
      "grad_norm": 4.023826599121094,
      "learning_rate": 1.9991852917895272e-05,
      "loss": 4.3618,
      "num_tokens": 27676055.0,
      "step": 450
    },
    {
      "epoch": 0.09172208209126347,
      "grad_norm": 3.742589235305786,
      "learning_rate": 1.9989351733299514e-05,
      "loss": 4.3175,
      "num_tokens": 30745102.0,
      "step": 500
    },
    {
      "epoch": 0.10089429030038982,
      "grad_norm": 3.394568681716919,
      "learning_rate": 1.9986516418052213e-05,
      "loss": 4.2946,
      "num_tokens": 33797572.0,
      "step": 550
    },
    {
      "epoch": 0.11006649850951616,
      "grad_norm": 3.7485358715057373,
      "learning_rate": 1.998334706699093e-05,
      "loss": 4.2073,
      "num_tokens": 36862581.0,
      "step": 600
    },
    {
      "epoch": 0.1192387067186425,
      "grad_norm": 3.286533832550049,
      "learning_rate": 1.997984378612628e-05,
      "loss": 4.1903,
      "num_tokens": 39940075.0,
      "step": 650
    },
    {
      "epoch": 0.12841091492776885,
      "grad_norm": 3.757556676864624,
      "learning_rate": 1.997600669263839e-05,
      "loss": 4.1797,
      "num_tokens": 43022248.0,
      "step": 700
    },
    {
      "epoch": 0.13758312313689522,
      "grad_norm": 3.648007392883301,
      "learning_rate": 1.9971835914872985e-05,
      "loss": 4.1506,
      "num_tokens": 46088933.0,
      "step": 750
    },
    {
      "epoch": 0.14675533134602156,
      "grad_norm": 3.151127815246582,
      "learning_rate": 1.996733159233708e-05,
      "loss": 4.0749,
      "num_tokens": 49170683.0,
      "step": 800
    },
    {
      "epoch": 0.1559275395551479,
      "grad_norm": 3.370467185974121,
      "learning_rate": 1.9962493875694335e-05,
      "loss": 4.0687,
      "num_tokens": 52242629.0,
      "step": 850
    },
    {
      "epoch": 0.16509974776427425,
      "grad_norm": 3.516242027282715,
      "learning_rate": 1.995732292675999e-05,
      "loss": 3.9537,
      "num_tokens": 55316954.0,
      "step": 900
    },
    {
      "epoch": 0.1742719559734006,
      "grad_norm": 3.9187300205230713,
      "learning_rate": 1.9951818918495474e-05,
      "loss": 3.9209,
      "num_tokens": 58384220.0,
      "step": 950
    },
    {
      "epoch": 0.18344416418252693,
      "grad_norm": 3.647482395172119,
      "learning_rate": 1.9945982035002613e-05,
      "loss": 3.8353,
      "num_tokens": 61463865.0,
      "step": 1000
    },
    {
      "epoch": 0.1926163723916533,
      "grad_norm": 4.165699005126953,
      "learning_rate": 1.9939812471517463e-05,
      "loss": 3.8031,
      "num_tokens": 64528502.0,
      "step": 1050
    },
    {
      "epoch": 0.20178858060077964,
      "grad_norm": 3.750986099243164,
      "learning_rate": 1.99333104344038e-05,
      "loss": 3.723,
      "num_tokens": 67601444.0,
      "step": 1100
    },
    {
      "epoch": 0.21096078880990599,
      "grad_norm": 4.958056926727295,
      "learning_rate": 1.9926476141146186e-05,
      "loss": 3.6597,
      "num_tokens": 70661859.0,
      "step": 1150
    },
    {
      "epoch": 0.22013299701903233,
      "grad_norm": 3.76973295211792,
      "learning_rate": 1.9919309820342735e-05,
      "loss": 3.5977,
      "num_tokens": 73730542.0,
      "step": 1200
    },
    {
      "epoch": 0.22930520522815867,
      "grad_norm": 4.332479953765869,
      "learning_rate": 1.991181171169743e-05,
      "loss": 3.5533,
      "num_tokens": 76796322.0,
      "step": 1250
    },
    {
      "epoch": 0.238477413437285,
      "grad_norm": 3.7759413719177246,
      "learning_rate": 1.9903982066012126e-05,
      "loss": 3.5245,
      "num_tokens": 79858262.0,
      "step": 1300
    },
    {
      "epoch": 0.24764962164641138,
      "grad_norm": 3.8706881999969482,
      "learning_rate": 1.9895821145178148e-05,
      "loss": 3.4821,
      "num_tokens": 82926316.0,
      "step": 1350
    },
    {
      "epoch": 0.2568218298555377,
      "grad_norm": 3.8657612800598145,
      "learning_rate": 1.9887329222167548e-05,
      "loss": 3.3993,
      "num_tokens": 85997475.0,
      "step": 1400
    },
    {
      "epoch": 0.2659940380646641,
      "grad_norm": 3.64652156829834,
      "learning_rate": 1.9878506581023957e-05,
      "loss": 3.4047,
      "num_tokens": 89061497.0,
      "step": 1450
    },
    {
      "epoch": 0.27516624627379044,
      "grad_norm": 3.8434200286865234,
      "learning_rate": 1.98693535168531e-05,
      "loss": 3.3561,
      "num_tokens": 92130920.0,
      "step": 1500
    },
    {
      "epoch": 0.2843384544829168,
      "grad_norm": 3.7041072845458984,
      "learning_rate": 1.9859870335812904e-05,
      "loss": 3.2907,
      "num_tokens": 95201616.0,
      "step": 1550
    },
    {
      "epoch": 0.2935106626920431,
      "grad_norm": 3.672961711883545,
      "learning_rate": 1.9850057355103283e-05,
      "loss": 3.2824,
      "num_tokens": 98262928.0,
      "step": 1600
    },
    {
      "epoch": 0.30268287090116947,
      "grad_norm": 4.109488487243652,
      "learning_rate": 1.9839914902955514e-05,
      "loss": 3.2584,
      "num_tokens": 101331029.0,
      "step": 1650
    },
    {
      "epoch": 0.3118550791102958,
      "grad_norm": 3.8466403484344482,
      "learning_rate": 1.982944331862125e-05,
      "loss": 3.2185,
      "num_tokens": 104400096.0,
      "step": 1700
    },
    {
      "epoch": 0.32102728731942215,
      "grad_norm": 3.791473150253296,
      "learning_rate": 1.9818642952361188e-05,
      "loss": 3.2011,
      "num_tokens": 107470267.0,
      "step": 1750
    },
    {
      "epoch": 0.3301994955285485,
      "grad_norm": 4.251014232635498,
      "learning_rate": 1.9807514165433342e-05,
      "loss": 3.1644,
      "num_tokens": 110539470.0,
      "step": 1800
    },
    {
      "epoch": 0.33937170373767483,
      "grad_norm": 3.785540819168091,
      "learning_rate": 1.979605733008098e-05,
      "loss": 3.129,
      "num_tokens": 113605849.0,
      "step": 1850
    },
    {
      "epoch": 0.3485439119468012,
      "grad_norm": 3.8080990314483643,
      "learning_rate": 1.978427282952014e-05,
      "loss": 3.1142,
      "num_tokens": 116669302.0,
      "step": 1900
    },
    {
      "epoch": 0.3577161201559275,
      "grad_norm": 4.565866470336914,
      "learning_rate": 1.977216105792684e-05,
      "loss": 3.0942,
      "num_tokens": 119736059.0,
      "step": 1950
    },
    {
      "epoch": 0.36688832836505386,
      "grad_norm": 3.6725170612335205,
      "learning_rate": 1.9759722420423875e-05,
      "loss": 3.0913,
      "num_tokens": 122795575.0,
      "step": 2000
    },
    {
      "epoch": 0.37606053657418026,
      "grad_norm": 4.128929138183594,
      "learning_rate": 1.974695733306728e-05,
      "loss": 3.0232,
      "num_tokens": 125860925.0,
      "step": 2050
    },
    {
      "epoch": 0.3852327447833066,
      "grad_norm": 4.008982181549072,
      "learning_rate": 1.9733866222832404e-05,
      "loss": 3.0281,
      "num_tokens": 128939549.0,
      "step": 2100
    },
    {
      "epoch": 0.39440495299243294,
      "grad_norm": 3.8258960247039795,
      "learning_rate": 1.972044952759963e-05,
      "loss": 3.005,
      "num_tokens": 132002665.0,
      "step": 2150
    },
    {
      "epoch": 0.4035771612015593,
      "grad_norm": 3.722099781036377,
      "learning_rate": 1.970670769613974e-05,
      "loss": 2.9717,
      "num_tokens": 135066326.0,
      "step": 2200
    },
    {
      "epoch": 0.41274936941068563,
      "grad_norm": 3.6987600326538086,
      "learning_rate": 1.9692641188098886e-05,
      "loss": 2.9571,
      "num_tokens": 138128630.0,
      "step": 2250
    },
    {
      "epoch": 0.42192157761981197,
      "grad_norm": 6.789268493652344,
      "learning_rate": 1.9678250473983223e-05,
      "loss": 2.9399,
      "num_tokens": 141192291.0,
      "step": 2300
    },
    {
      "epoch": 0.4310937858289383,
      "grad_norm": 4.319190979003906,
      "learning_rate": 1.966353603514317e-05,
      "loss": 2.9201,
      "num_tokens": 144272511.0,
      "step": 2350
    },
    {
      "epoch": 0.44026599403806466,
      "grad_norm": 4.209902763366699,
      "learning_rate": 1.9648498363757325e-05,
      "loss": 2.924,
      "num_tokens": 147343206.0,
      "step": 2400
    },
    {
      "epoch": 0.449438202247191,
      "grad_norm": 3.9288434982299805,
      "learning_rate": 1.9633137962815973e-05,
      "loss": 2.8886,
      "num_tokens": 150412293.0,
      "step": 2450
    },
    {
      "epoch": 0.45861041045631734,
      "grad_norm": 4.768758296966553,
      "learning_rate": 1.9617455346104283e-05,
      "loss": 2.8774,
      "num_tokens": 153481791.0,
      "step": 2500
    },
    {
      "epoch": 0.4677826186654437,
      "grad_norm": 4.089212894439697,
      "learning_rate": 1.9601451038185117e-05,
      "loss": 2.8393,
      "num_tokens": 156551627.0,
      "step": 2550
    },
    {
      "epoch": 0.47695482687457,
      "grad_norm": 3.6109139919281006,
      "learning_rate": 1.958512557438149e-05,
      "loss": 2.8697,
      "num_tokens": 159628649.0,
      "step": 2600
    },
    {
      "epoch": 0.4861270350836964,
      "grad_norm": 4.69704008102417,
      "learning_rate": 1.9568479500758648e-05,
      "loss": 2.8156,
      "num_tokens": 162692475.0,
      "step": 2650
    },
    {
      "epoch": 0.49529924329282277,
      "grad_norm": 3.726196527481079,
      "learning_rate": 1.9551513374105823e-05,
      "loss": 2.8183,
      "num_tokens": 165764216.0,
      "step": 2700
    },
    {
      "epoch": 0.504471451501949,
      "grad_norm": 3.8877694606781006,
      "learning_rate": 1.9534227761917585e-05,
      "loss": 2.8098,
      "num_tokens": 168832733.0,
      "step": 2750
    },
    {
      "epoch": 0.5136436597110754,
      "grad_norm": 4.2488112449646,
      "learning_rate": 1.9516623242374888e-05,
      "loss": 2.8193,
      "num_tokens": 171887021.0,
      "step": 2800
    },
    {
      "epoch": 0.5228158679202017,
      "grad_norm": 3.4136576652526855,
      "learning_rate": 1.949870040432571e-05,
      "loss": 2.7806,
      "num_tokens": 174960619.0,
      "step": 2850
    },
    {
      "epoch": 0.5319880761293282,
      "grad_norm": 4.282463550567627,
      "learning_rate": 1.948045984726537e-05,
      "loss": 2.779,
      "num_tokens": 178028517.0,
      "step": 2900
    },
    {
      "epoch": 0.5411602843384545,
      "grad_norm": 4.703866958618164,
      "learning_rate": 1.9461902181316455e-05,
      "loss": 2.761,
      "num_tokens": 181092191.0,
      "step": 2950
    },
    {
      "epoch": 0.5503324925475809,
      "grad_norm": 3.5504965782165527,
      "learning_rate": 1.9443028027208443e-05,
      "loss": 2.7438,
      "num_tokens": 184160542.0,
      "step": 3000
    },
    {
      "epoch": 0.5595047007567072,
      "grad_norm": 4.609411239624023,
      "learning_rate": 1.9423838016256913e-05,
      "loss": 2.7322,
      "num_tokens": 187232548.0,
      "step": 3050
    },
    {
      "epoch": 0.5686769089658336,
      "grad_norm": 3.5375072956085205,
      "learning_rate": 1.9404332790342446e-05,
      "loss": 2.7147,
      "num_tokens": 190297910.0,
      "step": 3100
    },
    {
      "epoch": 0.5778491171749599,
      "grad_norm": 3.5921809673309326,
      "learning_rate": 1.9384513001889144e-05,
      "loss": 2.7105,
      "num_tokens": 193364684.0,
      "step": 3150
    },
    {
      "epoch": 0.5870213253840862,
      "grad_norm": 3.7573354244232178,
      "learning_rate": 1.936437931384281e-05,
      "loss": 2.7081,
      "num_tokens": 196431776.0,
      "step": 3200
    },
    {
      "epoch": 0.5961935335932126,
      "grad_norm": 4.44867467880249,
      "learning_rate": 1.934393239964878e-05,
      "loss": 2.6803,
      "num_tokens": 199500117.0,
      "step": 3250
    },
    {
      "epoch": 0.6053657418023389,
      "grad_norm": 3.4919068813323975,
      "learning_rate": 1.9323172943229386e-05,
      "loss": 2.6674,
      "num_tokens": 202569017.0,
      "step": 3300
    },
    {
      "epoch": 0.6145379500114653,
      "grad_norm": 3.851284980773926,
      "learning_rate": 1.930210163896109e-05,
      "loss": 2.6882,
      "num_tokens": 205635983.0,
      "step": 3350
    },
    {
      "epoch": 0.6237101582205916,
      "grad_norm": 5.8265910148620605,
      "learning_rate": 1.9280719191651252e-05,
      "loss": 2.648,
      "num_tokens": 208697081.0,
      "step": 3400
    },
    {
      "epoch": 0.632882366429718,
      "grad_norm": 3.4978857040405273,
      "learning_rate": 1.925902631651456e-05,
      "loss": 2.6435,
      "num_tokens": 211770956.0,
      "step": 3450
    },
    {
      "epoch": 0.6420545746388443,
      "grad_norm": 3.321568250656128,
      "learning_rate": 1.9237023739149094e-05,
      "loss": 2.6556,
      "num_tokens": 214830999.0,
      "step": 3500
    },
    {
      "epoch": 0.6512267828479706,
      "grad_norm": 3.848578691482544,
      "learning_rate": 1.9214712195512074e-05,
      "loss": 2.6334,
      "num_tokens": 217899025.0,
      "step": 3550
    },
    {
      "epoch": 0.660398991057097,
      "grad_norm": 3.511373996734619,
      "learning_rate": 1.919209243189524e-05,
      "loss": 2.6138,
      "num_tokens": 220972192.0,
      "step": 3600
    },
    {
      "epoch": 0.6695711992662233,
      "grad_norm": 3.4985859394073486,
      "learning_rate": 1.9169165204899866e-05,
      "loss": 2.6251,
      "num_tokens": 224041762.0,
      "step": 3650
    },
    {
      "epoch": 0.6787434074753497,
      "grad_norm": 5.905818462371826,
      "learning_rate": 1.9145931281411485e-05,
      "loss": 2.6089,
      "num_tokens": 227110870.0,
      "step": 3700
    },
    {
      "epoch": 0.687915615684476,
      "grad_norm": 8.281390190124512,
      "learning_rate": 1.9122391438574226e-05,
      "loss": 2.6084,
      "num_tokens": 230193112.0,
      "step": 3750
    },
    {
      "epoch": 0.6970878238936024,
      "grad_norm": 3.555816888809204,
      "learning_rate": 1.9098546463764802e-05,
      "loss": 2.6074,
      "num_tokens": 233261587.0,
      "step": 3800
    },
    {
      "epoch": 0.7062600321027287,
      "grad_norm": 4.379006862640381,
      "learning_rate": 1.9074397154566203e-05,
      "loss": 2.5825,
      "num_tokens": 236318086.0,
      "step": 3850
    },
    {
      "epoch": 0.715432240311855,
      "grad_norm": 4.370838642120361,
      "learning_rate": 1.9049944318740998e-05,
      "loss": 2.6007,
      "num_tokens": 239390375.0,
      "step": 3900
    },
    {
      "epoch": 0.7246044485209814,
      "grad_norm": 3.347520112991333,
      "learning_rate": 1.902518877420432e-05,
      "loss": 2.5861,
      "num_tokens": 242454603.0,
      "step": 3950
    },
    {
      "epoch": 0.7337766567301077,
      "grad_norm": 3.8170406818389893,
      "learning_rate": 1.900013134899651e-05,
      "loss": 2.5956,
      "num_tokens": 245520349.0,
      "step": 4000
    },
    {
      "epoch": 0.7429488649392341,
      "grad_norm": 4.084133625030518,
      "learning_rate": 1.8974772881255423e-05,
      "loss": 2.5744,
      "num_tokens": 248591742.0,
      "step": 4050
    },
    {
      "epoch": 0.7521210731483605,
      "grad_norm": 4.162419319152832,
      "learning_rate": 1.8949114219188386e-05,
      "loss": 2.5557,
      "num_tokens": 251668641.0,
      "step": 4100
    },
    {
      "epoch": 0.7612932813574869,
      "grad_norm": 4.139531135559082,
      "learning_rate": 1.8923156221043834e-05,
      "loss": 2.5529,
      "num_tokens": 254744782.0,
      "step": 4150
    },
    {
      "epoch": 0.7704654895666132,
      "grad_norm": 3.5236408710479736,
      "learning_rate": 1.8896899755082594e-05,
      "loss": 2.5342,
      "num_tokens": 257825042.0,
      "step": 4200
    },
    {
      "epoch": 0.7796376977757395,
      "grad_norm": 7.408937931060791,
      "learning_rate": 1.8870345699548856e-05,
      "loss": 2.5367,
      "num_tokens": 260890737.0,
      "step": 4250
    },
    {
      "epoch": 0.7888099059848659,
      "grad_norm": 4.408638000488281,
      "learning_rate": 1.8843494942640782e-05,
      "loss": 2.5397,
      "num_tokens": 263956000.0,
      "step": 4300
    },
    {
      "epoch": 0.7979821141939922,
      "grad_norm": 3.64819073677063,
      "learning_rate": 1.8816348382480807e-05,
      "loss": 2.5326,
      "num_tokens": 267024229.0,
      "step": 4350
    },
    {
      "epoch": 0.8071543224031186,
      "grad_norm": 3.3967065811157227,
      "learning_rate": 1.8788906927085596e-05,
      "loss": 2.5291,
      "num_tokens": 270093879.0,
      "step": 4400
    },
    {
      "epoch": 0.8163265306122449,
      "grad_norm": 3.771216869354248,
      "learning_rate": 1.876117149433567e-05,
      "loss": 2.5167,
      "num_tokens": 273153385.0,
      "step": 4450
    },
    {
      "epoch": 0.8254987388213713,
      "grad_norm": 3.552034854888916,
      "learning_rate": 1.87331430119447e-05,
      "loss": 2.516,
      "num_tokens": 276228897.0,
      "step": 4500
    },
    {
      "epoch": 0.8346709470304976,
      "grad_norm": 4.362802982330322,
      "learning_rate": 1.870482241742849e-05,
      "loss": 2.5083,
      "num_tokens": 279299108.0,
      "step": 4550
    },
    {
      "epoch": 0.8438431552396239,
      "grad_norm": 3.8707432746887207,
      "learning_rate": 1.867621065807361e-05,
      "loss": 2.4962,
      "num_tokens": 282367606.0,
      "step": 4600
    },
    {
      "epoch": 0.8530153634487503,
      "grad_norm": 3.7560534477233887,
      "learning_rate": 1.8647308690905695e-05,
      "loss": 2.5022,
      "num_tokens": 285431292.0,
      "step": 4650
    },
    {
      "epoch": 0.8621875716578766,
      "grad_norm": 4.073310375213623,
      "learning_rate": 1.8618117482657473e-05,
      "loss": 2.5052,
      "num_tokens": 288509576.0,
      "step": 4700
    },
    {
      "epoch": 0.871359779867003,
      "grad_norm": 4.2512288093566895,
      "learning_rate": 1.858863800973638e-05,
      "loss": 2.4922,
      "num_tokens": 291577926.0,
      "step": 4750
    },
    {
      "epoch": 0.8805319880761293,
      "grad_norm": 3.6681947708129883,
      "learning_rate": 1.8558871258191943e-05,
      "loss": 2.4687,
      "num_tokens": 294652257.0,
      "step": 4800
    },
    {
      "epoch": 0.8897041962852557,
      "grad_norm": 3.7940540313720703,
      "learning_rate": 1.8528818223682777e-05,
      "loss": 2.4622,
      "num_tokens": 297724675.0,
      "step": 4850
    },
    {
      "epoch": 0.898876404494382,
      "grad_norm": 4.411775588989258,
      "learning_rate": 1.849847991144328e-05,
      "loss": 2.472,
      "num_tokens": 300795065.0,
      "step": 4900
    },
    {
      "epoch": 0.9080486127035083,
      "grad_norm": 3.331857442855835,
      "learning_rate": 1.8467857336250015e-05,
      "loss": 2.4734,
      "num_tokens": 303858894.0,
      "step": 4950
    },
    {
      "epoch": 0.9172208209126347,
      "grad_norm": 5.33769416809082,
      "learning_rate": 1.8436951522387766e-05,
      "loss": 2.4534,
      "num_tokens": 306912692.0,
      "step": 5000
    },
    {
      "epoch": 0.926393029121761,
      "grad_norm": 3.5148868560791016,
      "learning_rate": 1.8405763503615285e-05,
      "loss": 2.4871,
      "num_tokens": 310002551.0,
      "step": 5050
    },
    {
      "epoch": 0.9355652373308874,
      "grad_norm": 3.6607162952423096,
      "learning_rate": 1.8374294323130697e-05,
      "loss": 2.4801,
      "num_tokens": 313070094.0,
      "step": 5100
    },
    {
      "epoch": 0.9447374455400137,
      "grad_norm": 6.60593318939209,
      "learning_rate": 1.8342545033536622e-05,
      "loss": 2.431,
      "num_tokens": 316134272.0,
      "step": 5150
    },
    {
      "epoch": 0.95390965374914,
      "grad_norm": 3.5577642917633057,
      "learning_rate": 1.8310516696804953e-05,
      "loss": 2.425,
      "num_tokens": 319208656.0,
      "step": 5200
    },
    {
      "epoch": 0.9630818619582665,
      "grad_norm": 3.443085193634033,
      "learning_rate": 1.8278210384241353e-05,
      "loss": 2.419,
      "num_tokens": 322277785.0,
      "step": 5250
    },
    {
      "epoch": 0.9722540701673928,
      "grad_norm": 4.775812149047852,
      "learning_rate": 1.8245627176449395e-05,
      "loss": 2.4398,
      "num_tokens": 325343977.0,
      "step": 5300
    },
    {
      "epoch": 0.9814262783765192,
      "grad_norm": 3.54435133934021,
      "learning_rate": 1.8212768163294453e-05,
      "loss": 2.4288,
      "num_tokens": 328414604.0,
      "step": 5350
    },
    {
      "epoch": 0.9905984865856455,
      "grad_norm": 3.6476333141326904,
      "learning_rate": 1.8179634443867207e-05,
      "loss": 2.4228,
      "num_tokens": 331477653.0,
      "step": 5400
    },
    {
      "epoch": 0.9997706947947719,
      "grad_norm": 4.548381805419922,
      "learning_rate": 1.814622712644691e-05,
      "loss": 2.4154,
      "num_tokens": 334553911.0,
      "step": 5450
    },
    {
      "epoch": 1.0,
      "eval_loss": 2.662151575088501,
      "eval_num_tokens": 334630136.0,
      "eval_runtime": 5.3343,
      "eval_samples_per_second": 90.922,
      "eval_steps_per_second": 5.811,
      "step": 5452
    },
    {
      "epoch": 1.0088053198807614,
      "grad_norm": 3.1079516410827637,
      "learning_rate": 1.8112547328464308e-05,
      "loss": 2.4206,
      "num_tokens": 337566175.0,
      "step": 5500
    },
    {
      "epoch": 1.0179775280898877,
      "grad_norm": 3.530143976211548,
      "learning_rate": 1.8078596176464257e-05,
      "loss": 2.426,
      "num_tokens": 340636573.0,
      "step": 5550
    },
    {
      "epoch": 1.027149736299014,
      "grad_norm": 3.131134033203125,
      "learning_rate": 1.8044374806068057e-05,
      "loss": 2.3843,
      "num_tokens": 343696926.0,
      "step": 5600
    },
    {
      "epoch": 1.0363219445081404,
      "grad_norm": 3.275998592376709,
      "learning_rate": 1.8009884361935445e-05,
      "loss": 2.4091,
      "num_tokens": 346772520.0,
      "step": 5650
    },
    {
      "epoch": 1.0454941527172668,
      "grad_norm": 3.3414154052734375,
      "learning_rate": 1.7975125997726328e-05,
      "loss": 2.3759,
      "num_tokens": 349844605.0,
      "step": 5700
    },
    {
      "epoch": 1.054666360926393,
      "grad_norm": 4.579761981964111,
      "learning_rate": 1.7940100876062184e-05,
      "loss": 2.3762,
      "num_tokens": 352917933.0,
      "step": 5750
    },
    {
      "epoch": 1.0638385691355194,
      "grad_norm": 3.126434326171875,
      "learning_rate": 1.7904810168487175e-05,
      "loss": 2.371,
      "num_tokens": 355987563.0,
      "step": 5800
    },
    {
      "epoch": 1.0730107773446458,
      "grad_norm": 3.765179395675659,
      "learning_rate": 1.7869255055428964e-05,
      "loss": 2.369,
      "num_tokens": 359052419.0,
      "step": 5850
    },
    {
      "epoch": 1.0821829855537721,
      "grad_norm": 3.1499340534210205,
      "learning_rate": 1.783343672615923e-05,
      "loss": 2.3784,
      "num_tokens": 362111794.0,
      "step": 5900
    },
    {
      "epoch": 1.0913551937628985,
      "grad_norm": 3.7670187950134277,
      "learning_rate": 1.7797356378753886e-05,
      "loss": 2.362,
      "num_tokens": 365184289.0,
      "step": 5950
    },
    {
      "epoch": 1.1005274019720248,
      "grad_norm": 4.070444583892822,
      "learning_rate": 1.7761015220053006e-05,
      "loss": 2.3939,
      "num_tokens": 368258725.0,
      "step": 6000
    },
    {
      "epoch": 1.1096996101811512,
      "grad_norm": 3.125969886779785,
      "learning_rate": 1.772441446562046e-05,
      "loss": 2.3923,
      "num_tokens": 371333945.0,
      "step": 6050
    },
    {
      "epoch": 1.1188718183902775,
      "grad_norm": 3.1043412685394287,
      "learning_rate": 1.7687555339703245e-05,
      "loss": 2.3639,
      "num_tokens": 374400787.0,
      "step": 6100
    },
    {
      "epoch": 1.1280440265994038,
      "grad_norm": 3.3613622188568115,
      "learning_rate": 1.7650439075190552e-05,
      "loss": 2.3593,
      "num_tokens": 377470930.0,
      "step": 6150
    },
    {
      "epoch": 1.1372162348085302,
      "grad_norm": 3.4898386001586914,
      "learning_rate": 1.7613066913572525e-05,
      "loss": 2.398,
      "num_tokens": 380540191.0,
      "step": 6200
    },
    {
      "epoch": 1.1463884430176565,
      "grad_norm": 2.7355456352233887,
      "learning_rate": 1.7575440104898712e-05,
      "loss": 2.37,
      "num_tokens": 383603448.0,
      "step": 6250
    },
    {
      "epoch": 1.1555606512267829,
      "grad_norm": 3.0368947982788086,
      "learning_rate": 1.7537559907736282e-05,
      "loss": 2.347,
      "num_tokens": 386679512.0,
      "step": 6300
    },
    {
      "epoch": 1.1647328594359092,
      "grad_norm": 3.0240418910980225,
      "learning_rate": 1.7499427589127913e-05,
      "loss": 2.3726,
      "num_tokens": 389735095.0,
      "step": 6350
    },
    {
      "epoch": 1.1739050676450355,
      "grad_norm": 3.9326765537261963,
      "learning_rate": 1.7461044424549408e-05,
      "loss": 2.3671,
      "num_tokens": 392794314.0,
      "step": 6400
    },
    {
      "epoch": 1.183077275854162,
      "grad_norm": 2.9769821166992188,
      "learning_rate": 1.7422411697867045e-05,
      "loss": 2.3217,
      "num_tokens": 395861368.0,
      "step": 6450
    },
    {
      "epoch": 1.1922494840632882,
      "grad_norm": 3.362730026245117,
      "learning_rate": 1.738353070129462e-05,
      "loss": 2.3555,
      "num_tokens": 398934759.0,
      "step": 6500
    },
    {
      "epoch": 1.2014216922724146,
      "grad_norm": 3.829068899154663,
      "learning_rate": 1.7344402735350235e-05,
      "loss": 2.3512,
      "num_tokens": 401999347.0,
      "step": 6550
    },
    {
      "epoch": 1.210593900481541,
      "grad_norm": 2.987025737762451,
      "learning_rate": 1.7305029108812777e-05,
      "loss": 2.3352,
      "num_tokens": 405069670.0,
      "step": 6600
    },
    {
      "epoch": 1.2197661086906673,
      "grad_norm": 3.466670274734497,
      "learning_rate": 1.7265411138678178e-05,
      "loss": 2.3183,
      "num_tokens": 408145682.0,
      "step": 6650
    },
    {
      "epoch": 1.2289383168997936,
      "grad_norm": 3.421823263168335,
      "learning_rate": 1.722555015011533e-05,
      "loss": 2.342,
      "num_tokens": 411222733.0,
      "step": 6700
    },
    {
      "epoch": 1.23811052510892,
      "grad_norm": 4.740370750427246,
      "learning_rate": 1.7185447476421764e-05,
      "loss": 2.33,
      "num_tokens": 414293206.0,
      "step": 6750
    },
    {
      "epoch": 1.2472827333180463,
      "grad_norm": 3.6490085124969482,
      "learning_rate": 1.7145104458979074e-05,
      "loss": 2.3191,
      "num_tokens": 417348351.0,
      "step": 6800
    },
    {
      "epoch": 1.2564549415271726,
      "grad_norm": 2.6619176864624023,
      "learning_rate": 1.7104522447208028e-05,
      "loss": 2.3287,
      "num_tokens": 420420095.0,
      "step": 6850
    },
    {
      "epoch": 1.265627149736299,
      "grad_norm": 3.2309610843658447,
      "learning_rate": 1.706370279852344e-05,
      "loss": 2.3252,
      "num_tokens": 423491946.0,
      "step": 6900
    },
    {
      "epoch": 1.2747993579454253,
      "grad_norm": 3.0654044151306152,
      "learning_rate": 1.702264687828877e-05,
      "loss": 2.3279,
      "num_tokens": 426558795.0,
      "step": 6950
    },
    {
      "epoch": 1.2839715661545517,
      "grad_norm": 3.152662992477417,
      "learning_rate": 1.6981356059770442e-05,
      "loss": 2.326,
      "num_tokens": 429622896.0,
      "step": 7000
    },
    {
      "epoch": 1.293143774363678,
      "grad_norm": 2.96634578704834,
      "learning_rate": 1.6939831724091927e-05,
      "loss": 2.3234,
      "num_tokens": 432687860.0,
      "step": 7050
    },
    {
      "epoch": 1.3023159825728043,
      "grad_norm": 5.338396072387695,
      "learning_rate": 1.6898075260187527e-05,
      "loss": 2.3119,
      "num_tokens": 435752672.0,
      "step": 7100
    },
    {
      "epoch": 1.3114881907819307,
      "grad_norm": 3.5085184574127197,
      "learning_rate": 1.6856088064755936e-05,
      "loss": 2.3226,
      "num_tokens": 438820637.0,
      "step": 7150
    },
    {
      "epoch": 1.320660398991057,
      "grad_norm": 2.9175689220428467,
      "learning_rate": 1.6813871542213514e-05,
      "loss": 2.2943,
      "num_tokens": 441888429.0,
      "step": 7200
    },
    {
      "epoch": 1.3298326072001834,
      "grad_norm": 4.301224708557129,
      "learning_rate": 1.6771427104647303e-05,
      "loss": 2.3095,
      "num_tokens": 444961887.0,
      "step": 7250
    },
    {
      "epoch": 1.3390048154093097,
      "grad_norm": 3.6543874740600586,
      "learning_rate": 1.6728756171767807e-05,
      "loss": 2.3122,
      "num_tokens": 448044553.0,
      "step": 7300
    },
    {
      "epoch": 1.3481770236184363,
      "grad_norm": 2.869036912918091,
      "learning_rate": 1.66858601708615e-05,
      "loss": 2.2963,
      "num_tokens": 451118849.0,
      "step": 7350
    },
    {
      "epoch": 1.3573492318275626,
      "grad_norm": 2.6811025142669678,
      "learning_rate": 1.6642740536743086e-05,
      "loss": 2.2987,
      "num_tokens": 454182648.0,
      "step": 7400
    },
    {
      "epoch": 1.366521440036689,
      "grad_norm": 3.101973295211792,
      "learning_rate": 1.65993987117075e-05,
      "loss": 2.2987,
      "num_tokens": 457255403.0,
      "step": 7450
    },
    {
      "epoch": 1.3756936482458153,
      "grad_norm": 5.224907875061035,
      "learning_rate": 1.6555836145481673e-05,
      "loss": 2.2874,
      "num_tokens": 460321588.0,
      "step": 7500
    },
    {
      "epoch": 1.3848658564549416,
      "grad_norm": 3.0296120643615723,
      "learning_rate": 1.6512054295176042e-05,
      "loss": 2.2832,
      "num_tokens": 463395678.0,
      "step": 7550
    },
    {
      "epoch": 1.394038064664068,
      "grad_norm": 3.4018895626068115,
      "learning_rate": 1.64680546252358e-05,
      "loss": 2.2759,
      "num_tokens": 466458283.0,
      "step": 7600
    },
    {
      "epoch": 1.4032102728731943,
      "grad_norm": 4.157562255859375,
      "learning_rate": 1.642383860739193e-05,
      "loss": 2.3066,
      "num_tokens": 469533607.0,
      "step": 7650
    },
    {
      "epoch": 1.4123824810823207,
      "grad_norm": 3.008226156234741,
      "learning_rate": 1.637940772061196e-05,
      "loss": 2.2832,
      "num_tokens": 472604436.0,
      "step": 7700
    },
    {
      "epoch": 1.421554689291447,
      "grad_norm": 3.0457003116607666,
      "learning_rate": 1.6334763451050496e-05,
      "loss": 2.2693,
      "num_tokens": 475671029.0,
      "step": 7750
    },
    {
      "epoch": 1.4307268975005734,
      "grad_norm": 2.6443099975585938,
      "learning_rate": 1.628990729199953e-05,
      "loss": 2.2919,
      "num_tokens": 478727676.0,
      "step": 7800
    },
    {
      "epoch": 1.4398991057096997,
      "grad_norm": 4.169058322906494,
      "learning_rate": 1.6244840743838463e-05,
      "loss": 2.271,
      "num_tokens": 481796442.0,
      "step": 7850
    },
    {
      "epoch": 1.449071313918826,
      "grad_norm": 4.345733642578125,
      "learning_rate": 1.619956531398395e-05,
      "loss": 2.2957,
      "num_tokens": 484867456.0,
      "step": 7900
    },
    {
      "epoch": 1.4582435221279524,
      "grad_norm": 4.380826950073242,
      "learning_rate": 1.615408251683945e-05,
      "loss": 2.2822,
      "num_tokens": 487941363.0,
      "step": 7950
    },
    {
      "epoch": 1.4674157303370787,
      "grad_norm": 3.4112560749053955,
      "learning_rate": 1.6108393873744593e-05,
      "loss": 2.287,
      "num_tokens": 491013621.0,
      "step": 8000
    },
    {
      "epoch": 1.476587938546205,
      "grad_norm": 4.045591354370117,
      "learning_rate": 1.606250091292428e-05,
      "loss": 2.2905,
      "num_tokens": 494090566.0,
      "step": 8050
    },
    {
      "epoch": 1.4857601467553314,
      "grad_norm": 3.067234754562378,
      "learning_rate": 1.6016405169437582e-05,
      "loss": 2.2676,
      "num_tokens": 497162524.0,
      "step": 8100
    },
    {
      "epoch": 1.4949323549644578,
      "grad_norm": 3.4898037910461426,
      "learning_rate": 1.5970108185126362e-05,
      "loss": 2.2657,
      "num_tokens": 500225621.0,
      "step": 8150
    },
    {
      "epoch": 1.504104563173584,
      "grad_norm": 3.147956609725952,
      "learning_rate": 1.592361150856374e-05,
      "loss": 2.2511,
      "num_tokens": 503293194.0,
      "step": 8200
    },
    {
      "epoch": 1.5132767713827104,
      "grad_norm": 2.8965494632720947,
      "learning_rate": 1.5876916695002277e-05,
      "loss": 2.2704,
      "num_tokens": 506366343.0,
      "step": 8250
    },
    {
      "epoch": 1.5224489795918368,
      "grad_norm": 4.127537250518799,
      "learning_rate": 1.5830025306321945e-05,
      "loss": 2.2745,
      "num_tokens": 509436830.0,
      "step": 8300
    },
    {
      "epoch": 1.5316211878009631,
      "grad_norm": 3.495630979537964,
      "learning_rate": 1.578293891097791e-05,
      "loss": 2.2761,
      "num_tokens": 512509535.0,
      "step": 8350
    },
    {
      "epoch": 1.5407933960100895,
      "grad_norm": 2.908043622970581,
      "learning_rate": 1.573565908394804e-05,
      "loss": 2.2706,
      "num_tokens": 515566764.0,
      "step": 8400
    },
    {
      "epoch": 1.5499656042192158,
      "grad_norm": 3.0703983306884766,
      "learning_rate": 1.568818740668024e-05,
      "loss": 2.2665,
      "num_tokens": 518634607.0,
      "step": 8450
    },
    {
      "epoch": 1.5591378124283422,
      "grad_norm": 3.2771713733673096,
      "learning_rate": 1.564052546703957e-05,
      "loss": 2.2468,
      "num_tokens": 521690844.0,
      "step": 8500
    },
    {
      "epoch": 1.5683100206374685,
      "grad_norm": 3.3331332206726074,
      "learning_rate": 1.5592674859255093e-05,
      "loss": 2.2691,
      "num_tokens": 524759416.0,
      "step": 8550
    },
    {
      "epoch": 1.5774822288465948,
      "grad_norm": 3.091536283493042,
      "learning_rate": 1.554463718386659e-05,
      "loss": 2.2526,
      "num_tokens": 527834639.0,
      "step": 8600
    },
    {
      "epoch": 1.5866544370557212,
      "grad_norm": 2.8725991249084473,
      "learning_rate": 1.5496414047670985e-05,
      "loss": 2.2763,
      "num_tokens": 530900064.0,
      "step": 8650
    },
    {
      "epoch": 1.5958266452648475,
      "grad_norm": 5.627836227416992,
      "learning_rate": 1.5448007063668645e-05,
      "loss": 2.262,
      "num_tokens": 533979089.0,
      "step": 8700
    },
    {
      "epoch": 1.6049988534739739,
      "grad_norm": 2.99426007270813,
      "learning_rate": 1.5399417851009394e-05,
      "loss": 2.243,
      "num_tokens": 537049251.0,
      "step": 8750
    },
    {
      "epoch": 1.6141710616831002,
      "grad_norm": 2.950291395187378,
      "learning_rate": 1.5350648034938364e-05,
      "loss": 2.2536,
      "num_tokens": 540115638.0,
      "step": 8800
    },
    {
      "epoch": 1.6233432698922265,
      "grad_norm": 4.661383152008057,
      "learning_rate": 1.5301699246741626e-05,
      "loss": 2.2568,
      "num_tokens": 543188518.0,
      "step": 8850
    },
    {
      "epoch": 1.632515478101353,
      "grad_norm": 3.3682801723480225,
      "learning_rate": 1.5252573123691654e-05,
      "loss": 2.269,
      "num_tokens": 546259101.0,
      "step": 8900
    },
    {
      "epoch": 1.6416876863104792,
      "grad_norm": 3.7050347328186035,
      "learning_rate": 1.520327130899252e-05,
      "loss": 2.2242,
      "num_tokens": 549332246.0,
      "step": 8950
    },
    {
      "epoch": 1.6508598945196056,
      "grad_norm": 2.5407910346984863,
      "learning_rate": 1.5153795451724953e-05,
      "loss": 2.2399,
      "num_tokens": 552401113.0,
      "step": 9000
    },
    {
      "epoch": 1.660032102728732,
      "grad_norm": 2.9334683418273926,
      "learning_rate": 1.5104147206791182e-05,
      "loss": 2.23,
      "num_tokens": 555478577.0,
      "step": 9050
    },
    {
      "epoch": 1.6692043109378583,
      "grad_norm": 3.646679401397705,
      "learning_rate": 1.5054328234859572e-05,
      "loss": 2.2431,
      "num_tokens": 558544082.0,
      "step": 9100
    },
    {
      "epoch": 1.6783765191469846,
      "grad_norm": 3.0259110927581787,
      "learning_rate": 1.5004340202309089e-05,
      "loss": 2.2374,
      "num_tokens": 561616790.0,
      "step": 9150
    },
    {
      "epoch": 1.687548727356111,
      "grad_norm": 5.128026485443115,
      "learning_rate": 1.495418478117354e-05,
      "loss": 2.2372,
      "num_tokens": 564687010.0,
      "step": 9200
    },
    {
      "epoch": 1.6967209355652373,
      "grad_norm": 2.761215925216675,
      "learning_rate": 1.4903863649085668e-05,
      "loss": 2.2316,
      "num_tokens": 567756890.0,
      "step": 9250
    },
    {
      "epoch": 1.7058931437743636,
      "grad_norm": 3.4462618827819824,
      "learning_rate": 1.4853378489221033e-05,
      "loss": 2.2451,
      "num_tokens": 570823485.0,
      "step": 9300
    },
    {
      "epoch": 1.71506535198349,
      "grad_norm": 3.8544795513153076,
      "learning_rate": 1.4802730990241702e-05,
      "loss": 2.2442,
      "num_tokens": 573890500.0,
      "step": 9350
    },
    {
      "epoch": 1.7242375601926163,
      "grad_norm": 3.1527886390686035,
      "learning_rate": 1.475192284623977e-05,
      "loss": 2.2257,
      "num_tokens": 576956808.0,
      "step": 9400
    },
    {
      "epoch": 1.7334097684017427,
      "grad_norm": 3.3589236736297607,
      "learning_rate": 1.4700955756680704e-05,
      "loss": 2.2256,
      "num_tokens": 580028975.0,
      "step": 9450
    },
    {
      "epoch": 1.742581976610869,
      "grad_norm": 2.782637119293213,
      "learning_rate": 1.464983142634648e-05,
      "loss": 2.2284,
      "num_tokens": 583101191.0,
      "step": 9500
    },
    {
      "epoch": 1.7517541848199953,
      "grad_norm": 2.791273832321167,
      "learning_rate": 1.4598551565278578e-05,
      "loss": 2.2466,
      "num_tokens": 586177194.0,
      "step": 9550
    },
    {
      "epoch": 1.7609263930291217,
      "grad_norm": 3.3659918308258057,
      "learning_rate": 1.4547117888720776e-05,
      "loss": 2.2151,
      "num_tokens": 589248537.0,
      "step": 9600
    },
    {
      "epoch": 1.770098601238248,
      "grad_norm": 3.0505056381225586,
      "learning_rate": 1.4495532117061772e-05,
      "loss": 2.2645,
      "num_tokens": 592318226.0,
      "step": 9650
    },
    {
      "epoch": 1.7792708094473744,
      "grad_norm": 3.632218360900879,
      "learning_rate": 1.4443795975777652e-05,
      "loss": 2.2267,
      "num_tokens": 595386393.0,
      "step": 9700
    },
    {
      "epoch": 1.7884430176565007,
      "grad_norm": 3.192918300628662,
      "learning_rate": 1.4391911195374164e-05,
      "loss": 2.204,
      "num_tokens": 598456081.0,
      "step": 9750
    },
    {
      "epoch": 1.797615225865627,
      "grad_norm": 2.828756809234619,
      "learning_rate": 1.4339879511328837e-05,
      "loss": 2.2332,
      "num_tokens": 601526298.0,
      "step": 9800
    },
    {
      "epoch": 1.8067874340747534,
      "grad_norm": 2.9059629440307617,
      "learning_rate": 1.4287702664032939e-05,
      "loss": 2.2152,
      "num_tokens": 604587712.0,
      "step": 9850
    },
    {
      "epoch": 1.8159596422838797,
      "grad_norm": 3.254810094833374,
      "learning_rate": 1.4235382398733247e-05,
      "loss": 2.21,
      "num_tokens": 607658810.0,
      "step": 9900
    },
    {
      "epoch": 1.825131850493006,
      "grad_norm": 2.805896043777466,
      "learning_rate": 1.4182920465473692e-05,
      "loss": 2.216,
      "num_tokens": 610716018.0,
      "step": 9950
    },
    {
      "epoch": 1.8343040587021324,
      "grad_norm": 3.190450429916382,
      "learning_rate": 1.4130318619036808e-05,
      "loss": 2.2301,
      "num_tokens": 613780917.0,
      "step": 10000
    },
    {
      "epoch": 1.8434762669112588,
      "grad_norm": 2.745699644088745,
      "learning_rate": 1.407757861888504e-05,
      "loss": 2.2432,
      "num_tokens": 616850698.0,
      "step": 10050
    },
    {
      "epoch": 1.852648475120385,
      "grad_norm": 3.0942554473876953,
      "learning_rate": 1.4024702229101885e-05,
      "loss": 2.2301,
      "num_tokens": 619916687.0,
      "step": 10100
    },
    {
      "epoch": 1.8618206833295114,
      "grad_norm": 2.85457444190979,
      "learning_rate": 1.3971691218332912e-05,
      "loss": 2.205,
      "num_tokens": 622985743.0,
      "step": 10150
    },
    {
      "epoch": 1.8709928915386378,
      "grad_norm": 3.447561025619507,
      "learning_rate": 1.3918547359726569e-05,
      "loss": 2.2299,
      "num_tokens": 626061264.0,
      "step": 10200
    },
    {
      "epoch": 1.8801650997477641,
      "grad_norm": 3.3711204528808594,
      "learning_rate": 1.3865272430874892e-05,
      "loss": 2.2207,
      "num_tokens": 629127449.0,
      "step": 10250
    },
    {
      "epoch": 1.8893373079568905,
      "grad_norm": 3.6021645069122314,
      "learning_rate": 1.3811868213754043e-05,
      "loss": 2.2439,
      "num_tokens": 632195091.0,
      "step": 10300
    },
    {
      "epoch": 1.8985095161660168,
      "grad_norm": 2.6557374000549316,
      "learning_rate": 1.3758336494664717e-05,
      "loss": 2.2255,
      "num_tokens": 635272904.0,
      "step": 10350
    },
    {
      "epoch": 1.9076817243751432,
      "grad_norm": 2.8971426486968994,
      "learning_rate": 1.3704679064172367e-05,
      "loss": 2.1996,
      "num_tokens": 638348466.0,
      "step": 10400
    },
    {
      "epoch": 1.9168539325842695,
      "grad_norm": 2.906564950942993,
      "learning_rate": 1.3650897717047336e-05,
      "loss": 2.2005,
      "num_tokens": 641418105.0,
      "step": 10450
    },
    {
      "epoch": 1.9260261407933958,
      "grad_norm": 3.114825487136841,
      "learning_rate": 1.359699425220481e-05,
      "loss": 2.2054,
      "num_tokens": 644495331.0,
      "step": 10500
    },
    {
      "epoch": 1.9351983490025224,
      "grad_norm": 2.6857142448425293,
      "learning_rate": 1.3542970472644656e-05,
      "loss": 2.2049,
      "num_tokens": 647574515.0,
      "step": 10550
    },
    {
      "epoch": 1.9443705572116488,
      "grad_norm": 3.0475053787231445,
      "learning_rate": 1.348882818539111e-05,
      "loss": 2.2099,
      "num_tokens": 650651609.0,
      "step": 10600
    },
    {
      "epoch": 1.953542765420775,
      "grad_norm": 2.635979413986206,
      "learning_rate": 1.3434569201432326e-05,
      "loss": 2.1955,
      "num_tokens": 653727158.0,
      "step": 10650
    },
    {
      "epoch": 1.9627149736299014,
      "grad_norm": 2.658064365386963,
      "learning_rate": 1.3380195335659822e-05,
      "loss": 2.215,
      "num_tokens": 656801882.0,
      "step": 10700
    },
    {
      "epoch": 1.9718871818390278,
      "grad_norm": 2.7926363945007324,
      "learning_rate": 1.3325708406807753e-05,
      "loss": 2.1944,
      "num_tokens": 659872536.0,
      "step": 10750
    },
    {
      "epoch": 1.9810593900481541,
      "grad_norm": 2.6653809547424316,
      "learning_rate": 1.3271110237392083e-05,
      "loss": 2.2009,
      "num_tokens": 662940974.0,
      "step": 10800
    },
    {
      "epoch": 1.9902315982572805,
      "grad_norm": 2.3410346508026123,
      "learning_rate": 1.3216402653649631e-05,
      "loss": 2.1746,
      "num_tokens": 665999433.0,
      "step": 10850
    },
    {
      "epoch": 1.9994038064664068,
      "grad_norm": 2.8723654747009277,
      "learning_rate": 1.3161587485476978e-05,
      "loss": 2.225,
      "num_tokens": 669062836.0,
      "step": 10900
    },
    {
      "epoch": 2.0,
      "eval_loss": 2.4245221614837646,
      "eval_num_tokens": 669261246.0,
      "eval_runtime": 5.3377,
      "eval_samples_per_second": 90.863,
      "eval_steps_per_second": 5.808,
      "step": 10904
    },
    {
      "epoch": 2.0084384315523964,
      "grad_norm": 2.9814906120300293,
      "learning_rate": 1.3106666566369266e-05,
      "loss": 2.1629,
      "num_tokens": 672079827.0,
      "step": 10950
    },
    {
      "epoch": 2.0176106397615228,
      "grad_norm": 2.8018527030944824,
      "learning_rate": 1.3051641733358869e-05,
      "loss": 2.179,
      "num_tokens": 675156433.0,
      "step": 11000
    },
    {
      "epoch": 2.026782847970649,
      "grad_norm": 2.977572202682495,
      "learning_rate": 1.2996514826953931e-05,
      "loss": 2.1796,
      "num_tokens": 678221060.0,
      "step": 11050
    },
    {
      "epoch": 2.0359550561797755,
      "grad_norm": 3.5244157314300537,
      "learning_rate": 1.2941287691076833e-05,
      "loss": 2.1746,
      "num_tokens": 681279496.0,
      "step": 11100
    },
    {
      "epoch": 2.045127264388902,
      "grad_norm": 4.0321431159973145,
      "learning_rate": 1.288596217300249e-05,
      "loss": 2.1692,
      "num_tokens": 684357409.0,
      "step": 11150
    },
    {
      "epoch": 2.054299472598028,
      "grad_norm": 3.5743963718414307,
      "learning_rate": 1.2830540123296568e-05,
      "loss": 2.1903,
      "num_tokens": 687419819.0,
      "step": 11200
    },
    {
      "epoch": 2.0634716808071545,
      "grad_norm": 2.714646339416504,
      "learning_rate": 1.27750233957536e-05,
      "loss": 2.1826,
      "num_tokens": 690478740.0,
      "step": 11250
    },
    {
      "epoch": 2.072643889016281,
      "grad_norm": 3.1788077354431152,
      "learning_rate": 1.2719413847334963e-05,
      "loss": 2.1772,
      "num_tokens": 693539914.0,
      "step": 11300
    },
    {
      "epoch": 2.081816097225407,
      "grad_norm": 2.851320266723633,
      "learning_rate": 1.2663713338106765e-05,
      "loss": 2.1902,
      "num_tokens": 696609574.0,
      "step": 11350
    },
    {
      "epoch": 2.0909883054345335,
      "grad_norm": 3.043736219406128,
      "learning_rate": 1.2607923731177642e-05,
      "loss": 2.164,
      "num_tokens": 699671518.0,
      "step": 11400
    },
    {
      "epoch": 2.10016051364366,
      "grad_norm": 2.4879658222198486,
      "learning_rate": 1.2552046892636427e-05,
      "loss": 2.1851,
      "num_tokens": 702735604.0,
      "step": 11450
    },
    {
      "epoch": 2.109332721852786,
      "grad_norm": 3.014916181564331,
      "learning_rate": 1.2496084691489731e-05,
      "loss": 2.167,
      "num_tokens": 705808446.0,
      "step": 11500
    },
    {
      "epoch": 2.1185049300619125,
      "grad_norm": 5.272231101989746,
      "learning_rate": 1.2440038999599438e-05,
      "loss": 2.1686,
      "num_tokens": 708874044.0,
      "step": 11550
    },
    {
      "epoch": 2.127677138271039,
      "grad_norm": 2.8708434104919434,
      "learning_rate": 1.2383911691620083e-05,
      "loss": 2.1651,
      "num_tokens": 711930253.0,
      "step": 11600
    },
    {
      "epoch": 2.1368493464801652,
      "grad_norm": 2.5448362827301025,
      "learning_rate": 1.2327704644936156e-05,
      "loss": 2.1723,
      "num_tokens": 714992067.0,
      "step": 11650
    },
    {
      "epoch": 2.1460215546892916,
      "grad_norm": 2.435708999633789,
      "learning_rate": 1.22714197395993e-05,
      "loss": 2.1682,
      "num_tokens": 718068361.0,
      "step": 11700
    },
    {
      "epoch": 2.155193762898418,
      "grad_norm": 2.605455160140991,
      "learning_rate": 1.2215058858265423e-05,
      "loss": 2.178,
      "num_tokens": 721130428.0,
      "step": 11750
    },
    {
      "epoch": 2.1643659711075443,
      "grad_norm": 3.2399065494537354,
      "learning_rate": 1.2158623886131733e-05,
      "loss": 2.1682,
      "num_tokens": 724200490.0,
      "step": 11800
    },
    {
      "epoch": 2.1735381793166706,
      "grad_norm": 2.644134998321533,
      "learning_rate": 1.2102116710873681e-05,
      "loss": 2.1561,
      "num_tokens": 727279410.0,
      "step": 11850
    },
    {
      "epoch": 2.182710387525797,
      "grad_norm": 2.811669111251831,
      "learning_rate": 1.2045539222581812e-05,
      "loss": 2.168,
      "num_tokens": 730340441.0,
      "step": 11900
    },
    {
      "epoch": 2.1918825957349233,
      "grad_norm": 3.6238574981689453,
      "learning_rate": 1.1988893313698556e-05,
      "loss": 2.1496,
      "num_tokens": 733410874.0,
      "step": 11950
    },
    {
      "epoch": 2.2010548039440496,
      "grad_norm": 2.5450801849365234,
      "learning_rate": 1.1932180878954913e-05,
      "loss": 2.1789,
      "num_tokens": 736478072.0,
      "step": 12000
    },
    {
      "epoch": 2.210227012153176,
      "grad_norm": 3.001086711883545,
      "learning_rate": 1.1875403815307092e-05,
      "loss": 2.1639,
      "num_tokens": 739547375.0,
      "step": 12050
    },
    {
      "epoch": 2.2193992203623023,
      "grad_norm": 2.8402042388916016,
      "learning_rate": 1.1818564021873051e-05,
      "loss": 2.1697,
      "num_tokens": 742621016.0,
      "step": 12100
    },
    {
      "epoch": 2.2285714285714286,
      "grad_norm": 3.5109944343566895,
      "learning_rate": 1.1761663399868972e-05,
      "loss": 2.1792,
      "num_tokens": 745682838.0,
      "step": 12150
    },
    {
      "epoch": 2.237743636780555,
      "grad_norm": 3.4116134643554688,
      "learning_rate": 1.1704703852545685e-05,
      "loss": 2.1695,
      "num_tokens": 748753387.0,
      "step": 12200
    },
    {
      "epoch": 2.2469158449896813,
      "grad_norm": 3.222461700439453,
      "learning_rate": 1.1647687285124985e-05,
      "loss": 2.1513,
      "num_tokens": 751827240.0,
      "step": 12250
    },
    {
      "epoch": 2.2560880531988077,
      "grad_norm": 2.8149147033691406,
      "learning_rate": 1.1590615604735913e-05,
      "loss": 2.1506,
      "num_tokens": 754897622.0,
      "step": 12300
    },
    {
      "epoch": 2.265260261407934,
      "grad_norm": 3.038508176803589,
      "learning_rate": 1.1533490720350972e-05,
      "loss": 2.1518,
      "num_tokens": 757971210.0,
      "step": 12350
    },
    {
      "epoch": 2.2744324696170604,
      "grad_norm": 2.754978656768799,
      "learning_rate": 1.1476314542722267e-05,
      "loss": 2.1507,
      "num_tokens": 761038532.0,
      "step": 12400
    },
    {
      "epoch": 2.2836046778261867,
      "grad_norm": 2.7081894874572754,
      "learning_rate": 1.1419088984317589e-05,
      "loss": 2.1854,
      "num_tokens": 764119294.0,
      "step": 12450
    },
    {
      "epoch": 2.292776886035313,
      "grad_norm": 3.2845664024353027,
      "learning_rate": 1.1361815959256456e-05,
      "loss": 2.1522,
      "num_tokens": 767188411.0,
      "step": 12500
    },
    {
      "epoch": 2.3019490942444394,
      "grad_norm": 3.1656436920166016,
      "learning_rate": 1.1304497383246083e-05,
      "loss": 2.162,
      "num_tokens": 770249106.0,
      "step": 12550
    },
    {
      "epoch": 2.3111213024535657,
      "grad_norm": 3.843442440032959,
      "learning_rate": 1.1247135173517303e-05,
      "loss": 2.1501,
      "num_tokens": 773312866.0,
      "step": 12600
    },
    {
      "epoch": 2.320293510662692,
      "grad_norm": 2.4421207904815674,
      "learning_rate": 1.1189731248760438e-05,
      "loss": 2.1549,
      "num_tokens": 776389614.0,
      "step": 12650
    },
    {
      "epoch": 2.3294657188718184,
      "grad_norm": 2.5875680446624756,
      "learning_rate": 1.1132287529061127e-05,
      "loss": 2.1563,
      "num_tokens": 779462718.0,
      "step": 12700
    },
    {
      "epoch": 2.3386379270809448,
      "grad_norm": 3.222823143005371,
      "learning_rate": 1.1074805935836096e-05,
      "loss": 2.1493,
      "num_tokens": 782526270.0,
      "step": 12750
    },
    {
      "epoch": 2.347810135290071,
      "grad_norm": 2.859107494354248,
      "learning_rate": 1.1017288391768885e-05,
      "loss": 2.1694,
      "num_tokens": 785606459.0,
      "step": 12800
    },
    {
      "epoch": 2.3569823434991974,
      "grad_norm": 3.297732353210449,
      "learning_rate": 1.095973682074555e-05,
      "loss": 2.1502,
      "num_tokens": 788671501.0,
      "step": 12850
    },
    {
      "epoch": 2.366154551708324,
      "grad_norm": 2.5810225009918213,
      "learning_rate": 1.09021531477903e-05,
      "loss": 2.168,
      "num_tokens": 791747077.0,
      "step": 12900
    },
    {
      "epoch": 2.37532675991745,
      "grad_norm": 2.870887517929077,
      "learning_rate": 1.0844539299001112e-05,
      "loss": 2.1531,
      "num_tokens": 794816268.0,
      "step": 12950
    },
    {
      "epoch": 2.3844989681265765,
      "grad_norm": 3.0016744136810303,
      "learning_rate": 1.0786897201485303e-05,
      "loss": 2.1369,
      "num_tokens": 797885120.0,
      "step": 13000
    },
    {
      "epoch": 2.393671176335703,
      "grad_norm": 3.464268684387207,
      "learning_rate": 1.0729228783295072e-05,
      "loss": 2.146,
      "num_tokens": 800951727.0,
      "step": 13050
    },
    {
      "epoch": 2.402843384544829,
      "grad_norm": 2.76597261428833,
      "learning_rate": 1.0671535973363019e-05,
      "loss": 2.1607,
      "num_tokens": 804020260.0,
      "step": 13100
    },
    {
      "epoch": 2.4120155927539555,
      "grad_norm": 2.9096412658691406,
      "learning_rate": 1.06138207014376e-05,
      "loss": 2.141,
      "num_tokens": 807091595.0,
      "step": 13150
    },
    {
      "epoch": 2.421187800963082,
      "grad_norm": 3.2226877212524414,
      "learning_rate": 1.0556084898018615e-05,
      "loss": 2.1548,
      "num_tokens": 810167283.0,
      "step": 13200
    },
    {
      "epoch": 2.430360009172208,
      "grad_norm": 2.603027820587158,
      "learning_rate": 1.0498330494292596e-05,
      "loss": 2.149,
      "num_tokens": 813226356.0,
      "step": 13250
    },
    {
      "epoch": 2.4395322173813345,
      "grad_norm": 3.0920653343200684,
      "learning_rate": 1.0440559422068241e-05,
      "loss": 2.1495,
      "num_tokens": 816295716.0,
      "step": 13300
    },
    {
      "epoch": 2.448704425590461,
      "grad_norm": 2.6723990440368652,
      "learning_rate": 1.0382773613711787e-05,
      "loss": 2.1556,
      "num_tokens": 819371405.0,
      "step": 13350
    },
    {
      "epoch": 2.457876633799587,
      "grad_norm": 2.6446688175201416,
      "learning_rate": 1.032497500208237e-05,
      "loss": 2.1454,
      "num_tokens": 822442259.0,
      "step": 13400
    },
    {
      "epoch": 2.4670488420087135,
      "grad_norm": 2.5830724239349365,
      "learning_rate": 1.0267165520467385e-05,
      "loss": 2.1227,
      "num_tokens": 825509067.0,
      "step": 13450
    },
    {
      "epoch": 2.47622105021784,
      "grad_norm": 2.625577688217163,
      "learning_rate": 1.020934710251781e-05,
      "loss": 2.1478,
      "num_tokens": 828588777.0,
      "step": 13500
    },
    {
      "epoch": 2.4853932584269662,
      "grad_norm": 2.7063612937927246,
      "learning_rate": 1.0151521682183534e-05,
      "loss": 2.1322,
      "num_tokens": 831656407.0,
      "step": 13550
    },
    {
      "epoch": 2.4945654666360926,
      "grad_norm": 2.5114853382110596,
      "learning_rate": 1.0093691193648666e-05,
      "loss": 2.1542,
      "num_tokens": 834726053.0,
      "step": 13600
    },
    {
      "epoch": 2.503737674845219,
      "grad_norm": 2.416917324066162,
      "learning_rate": 1.0035857571266832e-05,
      "loss": 2.1343,
      "num_tokens": 837801784.0,
      "step": 13650
    },
    {
      "epoch": 2.5129098830543453,
      "grad_norm": 2.9593300819396973,
      "learning_rate": 9.978022749496495e-06,
      "loss": 2.151,
      "num_tokens": 840873527.0,
      "step": 13700
    },
    {
      "epoch": 2.5220820912634716,
      "grad_norm": 2.8286054134368896,
      "learning_rate": 9.920188662836223e-06,
      "loss": 2.1556,
      "num_tokens": 843955836.0,
      "step": 13750
    },
    {
      "epoch": 2.531254299472598,
      "grad_norm": 2.9257073402404785,
      "learning_rate": 9.862357245760006e-06,
      "loss": 2.1223,
      "num_tokens": 847025588.0,
      "step": 13800
    },
    {
      "epoch": 2.5404265076817243,
      "grad_norm": 3.1148195266723633,
      "learning_rate": 9.804530432652533e-06,
      "loss": 2.1481,
      "num_tokens": 850085106.0,
      "step": 13850
    },
    {
      "epoch": 2.5495987158908506,
      "grad_norm": 3.747142791748047,
      "learning_rate": 9.7467101577445e-06,
      "loss": 2.1562,
      "num_tokens": 853174410.0,
      "step": 13900
    },
    {
      "epoch": 2.558770924099977,
      "grad_norm": 2.368061065673828,
      "learning_rate": 9.688898355047905e-06,
      "loss": 2.1477,
      "num_tokens": 856251300.0,
      "step": 13950
    },
    {
      "epoch": 2.5679431323091033,
      "grad_norm": 2.6596877574920654,
      "learning_rate": 9.631096958291361e-06,
      "loss": 2.1328,
      "num_tokens": 859334390.0,
      "step": 14000
    },
    {
      "epoch": 2.5771153405182297,
      "grad_norm": 2.4528298377990723,
      "learning_rate": 9.573307900855428e-06,
      "loss": 2.1233,
      "num_tokens": 862423963.0,
      "step": 14050
    },
    {
      "epoch": 2.586287548727356,
      "grad_norm": 2.7410616874694824,
      "learning_rate": 9.515533115707915e-06,
      "loss": 2.1542,
      "num_tokens": 865499353.0,
      "step": 14100
    },
    {
      "epoch": 2.5954597569364823,
      "grad_norm": 2.893670082092285,
      "learning_rate": 9.457774535339251e-06,
      "loss": 2.1284,
      "num_tokens": 868572178.0,
      "step": 14150
    },
    {
      "epoch": 2.6046319651456087,
      "grad_norm": 2.4961345195770264,
      "learning_rate": 9.40003409169784e-06,
      "loss": 2.1387,
      "num_tokens": 871636138.0,
      "step": 14200
    },
    {
      "epoch": 2.613804173354735,
      "grad_norm": 2.655097723007202,
      "learning_rate": 9.342313716125427e-06,
      "loss": 2.1272,
      "num_tokens": 874711460.0,
      "step": 14250
    },
    {
      "epoch": 2.6229763815638614,
      "grad_norm": 2.478729724884033,
      "learning_rate": 9.284615339292515e-06,
      "loss": 2.1134,
      "num_tokens": 877775972.0,
      "step": 14300
    },
    {
      "epoch": 2.6321485897729877,
      "grad_norm": 2.3555896282196045,
      "learning_rate": 9.22694089113377e-06,
      "loss": 2.1121,
      "num_tokens": 880850816.0,
      "step": 14350
    },
    {
      "epoch": 2.641320797982114,
      "grad_norm": 3.220005512237549,
      "learning_rate": 9.16929230078348e-06,
      "loss": 2.1338,
      "num_tokens": 883915833.0,
      "step": 14400
    },
    {
      "epoch": 2.6504930061912404,
      "grad_norm": 2.8448081016540527,
      "learning_rate": 9.111671496511027e-06,
      "loss": 2.1302,
      "num_tokens": 886986053.0,
      "step": 14450
    },
    {
      "epoch": 2.6596652144003667,
      "grad_norm": 2.555471181869507,
      "learning_rate": 9.05408040565638e-06,
      "loss": 2.1248,
      "num_tokens": 890052402.0,
      "step": 14500
    },
    {
      "epoch": 2.668837422609493,
      "grad_norm": 2.9007201194763184,
      "learning_rate": 8.996520954565632e-06,
      "loss": 2.1339,
      "num_tokens": 893109082.0,
      "step": 14550
    },
    {
      "epoch": 2.6780096308186194,
      "grad_norm": 4.271463394165039,
      "learning_rate": 8.938995068526573e-06,
      "loss": 2.1229,
      "num_tokens": 896176023.0,
      "step": 14600
    },
    {
      "epoch": 2.6871818390277458,
      "grad_norm": 2.6032025814056396,
      "learning_rate": 8.881504671704282e-06,
      "loss": 2.1437,
      "num_tokens": 899239231.0,
      "step": 14650
    },
    {
      "epoch": 2.6963540472368726,
      "grad_norm": 2.927783489227295,
      "learning_rate": 8.824051687076774e-06,
      "loss": 2.1421,
      "num_tokens": 902305139.0,
      "step": 14700
    },
    {
      "epoch": 2.7055262554459985,
      "grad_norm": 2.9317362308502197,
      "learning_rate": 8.76663803637067e-06,
      "loss": 2.1258,
      "num_tokens": 905384894.0,
      "step": 14750
    },
    {
      "epoch": 2.7146984636551252,
      "grad_norm": 2.6969382762908936,
      "learning_rate": 8.709265639996932e-06,
      "loss": 2.132,
      "num_tokens": 908453576.0,
      "step": 14800
    },
    {
      "epoch": 2.723870671864251,
      "grad_norm": 2.784421443939209,
      "learning_rate": 8.651936416986608e-06,
      "loss": 2.1265,
      "num_tokens": 911520406.0,
      "step": 14850
    },
    {
      "epoch": 2.733042880073378,
      "grad_norm": 2.754910469055176,
      "learning_rate": 8.59465228492666e-06,
      "loss": 2.1287,
      "num_tokens": 914588413.0,
      "step": 14900
    },
    {
      "epoch": 2.742215088282504,
      "grad_norm": 2.538515329360962,
      "learning_rate": 8.53741515989582e-06,
      "loss": 2.1435,
      "num_tokens": 917659649.0,
      "step": 14950
    },
    {
      "epoch": 2.7513872964916306,
      "grad_norm": 3.178069591522217,
      "learning_rate": 8.480226956400491e-06,
      "loss": 2.1234,
      "num_tokens": 920724902.0,
      "step": 15000
    },
    {
      "epoch": 2.7605595047007565,
      "grad_norm": 3.3320059776306152,
      "learning_rate": 8.423089587310718e-06,
      "loss": 2.1176,
      "num_tokens": 923803197.0,
      "step": 15050
    },
    {
      "epoch": 2.7697317129098833,
      "grad_norm": 2.3997714519500732,
      "learning_rate": 8.366004963796202e-06,
      "loss": 2.1208,
      "num_tokens": 926880331.0,
      "step": 15100
    },
    {
      "epoch": 2.778903921119009,
      "grad_norm": 2.872494697570801,
      "learning_rate": 8.308974995262375e-06,
      "loss": 2.1405,
      "num_tokens": 929947114.0,
      "step": 15150
    },
    {
      "epoch": 2.788076129328136,
      "grad_norm": 2.511892557144165,
      "learning_rate": 8.25200158928653e-06,
      "loss": 2.1139,
      "num_tokens": 933010582.0,
      "step": 15200
    },
    {
      "epoch": 2.797248337537262,
      "grad_norm": 2.8765101432800293,
      "learning_rate": 8.195086651554013e-06,
      "loss": 2.1321,
      "num_tokens": 936084714.0,
      "step": 15250
    },
    {
      "epoch": 2.8064205457463887,
      "grad_norm": 2.4140682220458984,
      "learning_rate": 8.138232085794494e-06,
      "loss": 2.1322,
      "num_tokens": 939135927.0,
      "step": 15300
    },
    {
      "epoch": 2.8155927539555146,
      "grad_norm": 3.0238280296325684,
      "learning_rate": 8.081439793718271e-06,
      "loss": 2.1255,
      "num_tokens": 942203988.0,
      "step": 15350
    },
    {
      "epoch": 2.8247649621646413,
      "grad_norm": 3.9417409896850586,
      "learning_rate": 8.024711674952678e-06,
      "loss": 2.1234,
      "num_tokens": 945271149.0,
      "step": 15400
    },
    {
      "epoch": 2.8339371703737672,
      "grad_norm": 2.6532320976257324,
      "learning_rate": 7.968049626978527e-06,
      "loss": 2.1159,
      "num_tokens": 948330185.0,
      "step": 15450
    },
    {
      "epoch": 2.843109378582894,
      "grad_norm": 2.8720295429229736,
      "learning_rate": 7.911455545066658e-06,
      "loss": 2.1192,
      "num_tokens": 951405046.0,
      "step": 15500
    },
    {
      "epoch": 2.85228158679202,
      "grad_norm": 2.9304111003875732,
      "learning_rate": 7.854931322214525e-06,
      "loss": 2.1237,
      "num_tokens": 954478327.0,
      "step": 15550
    },
    {
      "epoch": 2.8614537950011467,
      "grad_norm": 2.6521966457366943,
      "learning_rate": 7.798478849082902e-06,
      "loss": 2.122,
      "num_tokens": 957538224.0,
      "step": 15600
    },
    {
      "epoch": 2.8706260032102726,
      "grad_norm": 2.3696980476379395,
      "learning_rate": 7.742100013932618e-06,
      "loss": 2.1414,
      "num_tokens": 960610742.0,
      "step": 15650
    },
    {
      "epoch": 2.8797982114193994,
      "grad_norm": 2.6325161457061768,
      "learning_rate": 7.685796702561416e-06,
      "loss": 2.1172,
      "num_tokens": 963681234.0,
      "step": 15700
    },
    {
      "epoch": 2.8889704196285253,
      "grad_norm": 4.786980152130127,
      "learning_rate": 7.629570798240861e-06,
      "loss": 2.1083,
      "num_tokens": 966763405.0,
      "step": 15750
    },
    {
      "epoch": 2.898142627837652,
      "grad_norm": 2.6920909881591797,
      "learning_rate": 7.573424181653363e-06,
      "loss": 2.1262,
      "num_tokens": 969828117.0,
      "step": 15800
    },
    {
      "epoch": 2.9073148360467784,
      "grad_norm": 2.511882781982422,
      "learning_rate": 7.51735873082925e-06,
      "loss": 2.1066,
      "num_tokens": 972905740.0,
      "step": 15850
    },
    {
      "epoch": 2.9164870442559048,
      "grad_norm": 2.9187281131744385,
      "learning_rate": 7.461376321083975e-06,
      "loss": 2.1116,
      "num_tokens": 975971572.0,
      "step": 15900
    },
    {
      "epoch": 2.925659252465031,
      "grad_norm": 3.1962969303131104,
      "learning_rate": 7.40547882495537e-06,
      "loss": 2.1431,
      "num_tokens": 979041359.0,
      "step": 15950
    },
    {
      "epoch": 2.9348314606741575,
      "grad_norm": 2.8684613704681396,
      "learning_rate": 7.34966811214102e-06,
      "loss": 2.1283,
      "num_tokens": 982097803.0,
      "step": 16000
    },
    {
      "epoch": 2.944003668883284,
      "grad_norm": 3.554492950439453,
      "learning_rate": 7.293946049435718e-06,
      "loss": 2.1225,
      "num_tokens": 985160591.0,
      "step": 16050
    },
    {
      "epoch": 2.95317587709241,
      "grad_norm": 2.750422239303589,
      "learning_rate": 7.238314500669033e-06,
      "loss": 2.1216,
      "num_tokens": 988236506.0,
      "step": 16100
    },
    {
      "epoch": 2.9623480853015365,
      "grad_norm": 3.027853012084961,
      "learning_rate": 7.182775326642961e-06,
      "loss": 2.1338,
      "num_tokens": 991298224.0,
      "step": 16150
    },
    {
      "epoch": 2.971520293510663,
      "grad_norm": 2.8112590312957764,
      "learning_rate": 7.127330385069682e-06,
      "loss": 2.1197,
      "num_tokens": 994361927.0,
      "step": 16200
    },
    {
      "epoch": 2.980692501719789,
      "grad_norm": 2.5668013095855713,
      "learning_rate": 7.071981530509424e-06,
      "loss": 2.0874,
      "num_tokens": 997437861.0,
      "step": 16250
    },
    {
      "epoch": 2.9898647099289155,
      "grad_norm": 2.909646987915039,
      "learning_rate": 7.01673061430844e-06,
      "loss": 2.1306,
      "num_tokens": 1000499652.0,
      "step": 16300
    },
    {
      "epoch": 2.999036918138042,
      "grad_norm": 2.9382026195526123,
      "learning_rate": 6.961579484537065e-06,
      "loss": 2.1142,
      "num_tokens": 1003569038.0,
      "step": 16350
    },
    {
      "epoch": 3.0,
      "eval_loss": 2.342665433883667,
      "eval_num_tokens": 1003891312.0,
      "eval_runtime": 5.3817,
      "eval_samples_per_second": 90.12,
      "eval_steps_per_second": 5.76,
      "step": 16356
    }
  ],
  "logging_steps": 50,
  "max_steps": 27260,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "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": 5.12321876600832e+17,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}