{
  "best_global_step": 5452,
  "best_metric": 2.662151575088501,
  "best_model_checkpoint": "/common/home/users/m/mq.nguyen.2023/scratchDirectory/finetune/train_nbtpj_summ_gpt2_inst_cnnxsumsam/checkpoint-5452",
  "epoch": 1.0,
  "eval_steps": 500,
  "global_step": 5452,
  "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
    }
  ],
  "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": 1.70773805085696e+17,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}