{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 1.0,
  "eval_steps": 500,
  "global_step": 4768,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0020973154362416107,
      "grad_norm": 432.3580017089844,
      "learning_rate": 7.500000000000001e-05,
      "loss": 2560.12109375,
      "step": 10
    },
    {
      "epoch": 0.0041946308724832215,
      "grad_norm": 248.64732360839844,
      "learning_rate": 0.00015833333333333332,
      "loss": 2143.7953125,
      "step": 20
    },
    {
      "epoch": 0.006291946308724832,
      "grad_norm": 132.24789428710938,
      "learning_rate": 0.00024166666666666667,
      "loss": 1870.5203125,
      "step": 30
    },
    {
      "epoch": 0.008389261744966443,
      "grad_norm": 167.39337158203125,
      "learning_rate": 0.00032500000000000004,
      "loss": 1728.409765625,
      "step": 40
    },
    {
      "epoch": 0.010486577181208054,
      "grad_norm": 139.81005859375,
      "learning_rate": 0.00039999996012889783,
      "loss": 1605.034765625,
      "step": 50
    },
    {
      "epoch": 0.012583892617449664,
      "grad_norm": 165.83798217773438,
      "learning_rate": 0.0003999951756180062,
      "loss": 1494.68759765625,
      "step": 60
    },
    {
      "epoch": 0.014681208053691275,
      "grad_norm": 107.8211669921875,
      "learning_rate": 0.00039998241712954386,
      "loss": 1407.76708984375,
      "step": 70
    },
    {
      "epoch": 0.016778523489932886,
      "grad_norm": 106.40216827392578,
      "learning_rate": 0.00039996168522872554,
      "loss": 1347.30556640625,
      "step": 80
    },
    {
      "epoch": 0.018875838926174497,
      "grad_norm": 107.21435546875,
      "learning_rate": 0.00039993298083399625,
      "loss": 1307.0609375,
      "step": 90
    },
    {
      "epoch": 0.02097315436241611,
      "grad_norm": 82.80896759033203,
      "learning_rate": 0.0003998963052169914,
      "loss": 1265.7685546875,
      "step": 100
    },
    {
      "epoch": 0.02307046979865772,
      "grad_norm": 112.75000762939453,
      "learning_rate": 0.0003998516600024797,
      "loss": 1228.41376953125,
      "step": 110
    },
    {
      "epoch": 0.025167785234899327,
      "grad_norm": 109.4422836303711,
      "learning_rate": 0.00039979904716829157,
      "loss": 1204.2453125,
      "step": 120
    },
    {
      "epoch": 0.02726510067114094,
      "grad_norm": 134.94847106933594,
      "learning_rate": 0.0003997384690452316,
      "loss": 1183.50576171875,
      "step": 130
    },
    {
      "epoch": 0.02936241610738255,
      "grad_norm": 68.75386047363281,
      "learning_rate": 0.0003996699283169749,
      "loss": 1160.85048828125,
      "step": 140
    },
    {
      "epoch": 0.031459731543624164,
      "grad_norm": 128.99111938476562,
      "learning_rate": 0.0003995934280199486,
      "loss": 1137.86796875,
      "step": 150
    },
    {
      "epoch": 0.03355704697986577,
      "grad_norm": 96.78877258300781,
      "learning_rate": 0.00039950897154319715,
      "loss": 1123.6431640625,
      "step": 160
    },
    {
      "epoch": 0.03565436241610738,
      "grad_norm": 99.28411865234375,
      "learning_rate": 0.0003994165626282323,
      "loss": 1115.12412109375,
      "step": 170
    },
    {
      "epoch": 0.037751677852348994,
      "grad_norm": 89.26158905029297,
      "learning_rate": 0.0003993162053688671,
      "loss": 1099.6396484375,
      "step": 180
    },
    {
      "epoch": 0.0398489932885906,
      "grad_norm": 86.16292572021484,
      "learning_rate": 0.0003992079042110349,
      "loss": 1081.5064453125,
      "step": 190
    },
    {
      "epoch": 0.04194630872483222,
      "grad_norm": 117.42514038085938,
      "learning_rate": 0.0003990916639525919,
      "loss": 1074.983984375,
      "step": 200
    },
    {
      "epoch": 0.044043624161073824,
      "grad_norm": 78.53323364257812,
      "learning_rate": 0.0003989674897431053,
      "loss": 1065.10068359375,
      "step": 210
    },
    {
      "epoch": 0.04614093959731544,
      "grad_norm": 101.44924926757812,
      "learning_rate": 0.00039883538708362454,
      "loss": 1046.9310546875,
      "step": 220
    },
    {
      "epoch": 0.04823825503355705,
      "grad_norm": 82.18006134033203,
      "learning_rate": 0.0003986953618264376,
      "loss": 1035.0302734375,
      "step": 230
    },
    {
      "epoch": 0.050335570469798654,
      "grad_norm": 105.84237670898438,
      "learning_rate": 0.0003985474201748124,
      "loss": 1032.61591796875,
      "step": 240
    },
    {
      "epoch": 0.05243288590604027,
      "grad_norm": 73.59429168701172,
      "learning_rate": 0.00039839156868272125,
      "loss": 1025.34921875,
      "step": 250
    },
    {
      "epoch": 0.05453020134228188,
      "grad_norm": 89.02239227294922,
      "learning_rate": 0.0003982278142545509,
      "loss": 1014.68466796875,
      "step": 260
    },
    {
      "epoch": 0.05662751677852349,
      "grad_norm": 82.29888916015625,
      "learning_rate": 0.00039805616414479654,
      "loss": 1002.25478515625,
      "step": 270
    },
    {
      "epoch": 0.0587248322147651,
      "grad_norm": 62.16850662231445,
      "learning_rate": 0.00039787662595774047,
      "loss": 1002.7884765625,
      "step": 280
    },
    {
      "epoch": 0.060822147651006714,
      "grad_norm": 79.44703674316406,
      "learning_rate": 0.00039768920764711517,
      "loss": 998.8552734375,
      "step": 290
    },
    {
      "epoch": 0.06291946308724833,
      "grad_norm": 64.72073364257812,
      "learning_rate": 0.00039749391751575094,
      "loss": 985.01357421875,
      "step": 300
    },
    {
      "epoch": 0.06501677852348993,
      "grad_norm": 68.88458251953125,
      "learning_rate": 0.00039729076421520827,
      "loss": 974.8654296875,
      "step": 310
    },
    {
      "epoch": 0.06711409395973154,
      "grad_norm": 67.44086456298828,
      "learning_rate": 0.0003970797567453941,
      "loss": 971.33193359375,
      "step": 320
    },
    {
      "epoch": 0.06921140939597316,
      "grad_norm": 78.49050903320312,
      "learning_rate": 0.0003968609044541637,
      "loss": 971.046875,
      "step": 330
    },
    {
      "epoch": 0.07130872483221476,
      "grad_norm": 65.71349334716797,
      "learning_rate": 0.0003966342170369062,
      "loss": 961.2708984375,
      "step": 340
    },
    {
      "epoch": 0.07340604026845637,
      "grad_norm": 72.90547943115234,
      "learning_rate": 0.000396399704536115,
      "loss": 953.9611328125,
      "step": 350
    },
    {
      "epoch": 0.07550335570469799,
      "grad_norm": 59.775489807128906,
      "learning_rate": 0.0003961573773409431,
      "loss": 950.950390625,
      "step": 360
    },
    {
      "epoch": 0.0776006711409396,
      "grad_norm": 69.97833251953125,
      "learning_rate": 0.00039590724618674295,
      "loss": 951.301953125,
      "step": 370
    },
    {
      "epoch": 0.0796979865771812,
      "grad_norm": 57.928466796875,
      "learning_rate": 0.00039564932215459036,
      "loss": 938.84296875,
      "step": 380
    },
    {
      "epoch": 0.08179530201342282,
      "grad_norm": 63.261505126953125,
      "learning_rate": 0.0003953836166707941,
      "loss": 933.84169921875,
      "step": 390
    },
    {
      "epoch": 0.08389261744966443,
      "grad_norm": 54.047210693359375,
      "learning_rate": 0.00039511014150638945,
      "loss": 926.5919921875,
      "step": 400
    },
    {
      "epoch": 0.08598993288590603,
      "grad_norm": 78.32682037353516,
      "learning_rate": 0.00039482890877661685,
      "loss": 925.966015625,
      "step": 410
    },
    {
      "epoch": 0.08808724832214765,
      "grad_norm": 47.812320709228516,
      "learning_rate": 0.0003945399309403852,
      "loss": 919.83203125,
      "step": 420
    },
    {
      "epoch": 0.09018456375838926,
      "grad_norm": 54.73347473144531,
      "learning_rate": 0.00039424322079971956,
      "loss": 913.32080078125,
      "step": 430
    },
    {
      "epoch": 0.09228187919463088,
      "grad_norm": 50.86581039428711,
      "learning_rate": 0.00039393879149919467,
      "loss": 908.5560546875,
      "step": 440
    },
    {
      "epoch": 0.09437919463087248,
      "grad_norm": 59.764652252197266,
      "learning_rate": 0.0003936266565253521,
      "loss": 913.7609375,
      "step": 450
    },
    {
      "epoch": 0.0964765100671141,
      "grad_norm": 56.112552642822266,
      "learning_rate": 0.00039330682970610276,
      "loss": 905.3310546875,
      "step": 460
    },
    {
      "epoch": 0.09857382550335571,
      "grad_norm": 56.61573028564453,
      "learning_rate": 0.00039297932521011465,
      "loss": 899.1501953125,
      "step": 470
    },
    {
      "epoch": 0.10067114093959731,
      "grad_norm": 58.438987731933594,
      "learning_rate": 0.0003926441575461852,
      "loss": 897.4177734375,
      "step": 480
    },
    {
      "epoch": 0.10276845637583892,
      "grad_norm": 57.88336944580078,
      "learning_rate": 0.0003923013415625978,
      "loss": 899.0765625,
      "step": 490
    },
    {
      "epoch": 0.10486577181208054,
      "grad_norm": 53.40110397338867,
      "learning_rate": 0.00039195089244646516,
      "loss": 891.66181640625,
      "step": 500
    },
    {
      "epoch": 0.10696308724832215,
      "grad_norm": 52.10205078125,
      "learning_rate": 0.00039159282572305543,
      "loss": 884.9646484375,
      "step": 510
    },
    {
      "epoch": 0.10906040268456375,
      "grad_norm": 56.75968933105469,
      "learning_rate": 0.00039122715725510506,
      "loss": 882.880078125,
      "step": 520
    },
    {
      "epoch": 0.11115771812080537,
      "grad_norm": 46.76315689086914,
      "learning_rate": 0.00039085390324211563,
      "loss": 886.416015625,
      "step": 530
    },
    {
      "epoch": 0.11325503355704698,
      "grad_norm": 61.728546142578125,
      "learning_rate": 0.0003904730802196368,
      "loss": 881.79140625,
      "step": 540
    },
    {
      "epoch": 0.11535234899328858,
      "grad_norm": 49.4840087890625,
      "learning_rate": 0.00039008470505853306,
      "loss": 874.3978515625,
      "step": 550
    },
    {
      "epoch": 0.1174496644295302,
      "grad_norm": 52.824913024902344,
      "learning_rate": 0.00038968879496423687,
      "loss": 870.12294921875,
      "step": 560
    },
    {
      "epoch": 0.11954697986577181,
      "grad_norm": 51.651973724365234,
      "learning_rate": 0.000389285367475986,
      "loss": 871.175,
      "step": 570
    },
    {
      "epoch": 0.12164429530201343,
      "grad_norm": 55.0900764465332,
      "learning_rate": 0.0003888744404660472,
      "loss": 868.1658203125,
      "step": 580
    },
    {
      "epoch": 0.12374161073825503,
      "grad_norm": 46.64694595336914,
      "learning_rate": 0.00038845603213892344,
      "loss": 859.77978515625,
      "step": 590
    },
    {
      "epoch": 0.12583892617449666,
      "grad_norm": 48.35624313354492,
      "learning_rate": 0.0003880301610305485,
      "loss": 861.89365234375,
      "step": 600
    },
    {
      "epoch": 0.12793624161073824,
      "grad_norm": 48.568016052246094,
      "learning_rate": 0.00038759684600746496,
      "loss": 864.978125,
      "step": 610
    },
    {
      "epoch": 0.13003355704697986,
      "grad_norm": 46.272010803222656,
      "learning_rate": 0.00038715610626598884,
      "loss": 856.43916015625,
      "step": 620
    },
    {
      "epoch": 0.13213087248322147,
      "grad_norm": 49.437721252441406,
      "learning_rate": 0.00038670796133135916,
      "loss": 853.705859375,
      "step": 630
    },
    {
      "epoch": 0.1342281879194631,
      "grad_norm": 51.41856002807617,
      "learning_rate": 0.0003862524310568727,
      "loss": 850.9275390625,
      "step": 640
    },
    {
      "epoch": 0.1363255033557047,
      "grad_norm": 50.722129821777344,
      "learning_rate": 0.0003857895356230049,
      "loss": 856.30986328125,
      "step": 650
    },
    {
      "epoch": 0.13842281879194632,
      "grad_norm": 51.90019989013672,
      "learning_rate": 0.00038531929553651525,
      "loss": 850.203125,
      "step": 660
    },
    {
      "epoch": 0.14052013422818793,
      "grad_norm": 49.493263244628906,
      "learning_rate": 0.0003848417316295394,
      "loss": 842.3240234375,
      "step": 670
    },
    {
      "epoch": 0.14261744966442952,
      "grad_norm": 43.617515563964844,
      "learning_rate": 0.00038435686505866595,
      "loss": 835.8236328125,
      "step": 680
    },
    {
      "epoch": 0.14471476510067113,
      "grad_norm": 40.38718032836914,
      "learning_rate": 0.0003838647173039994,
      "loss": 846.9615234375,
      "step": 690
    },
    {
      "epoch": 0.14681208053691275,
      "grad_norm": 39.99060821533203,
      "learning_rate": 0.00038336531016820805,
      "loss": 837.17314453125,
      "step": 700
    },
    {
      "epoch": 0.14890939597315436,
      "grad_norm": 45.416568756103516,
      "learning_rate": 0.000382858665775559,
      "loss": 836.13984375,
      "step": 710
    },
    {
      "epoch": 0.15100671140939598,
      "grad_norm": 41.88158416748047,
      "learning_rate": 0.0003823448065709373,
      "loss": 830.47421875,
      "step": 720
    },
    {
      "epoch": 0.1531040268456376,
      "grad_norm": 48.792327880859375,
      "learning_rate": 0.0003818237553188516,
      "loss": 838.256640625,
      "step": 730
    },
    {
      "epoch": 0.1552013422818792,
      "grad_norm": 47.30402755737305,
      "learning_rate": 0.00038129553510242626,
      "loss": 832.5138671875,
      "step": 740
    },
    {
      "epoch": 0.1572986577181208,
      "grad_norm": 46.606544494628906,
      "learning_rate": 0.00038076016932237823,
      "loss": 817.96171875,
      "step": 750
    },
    {
      "epoch": 0.1593959731543624,
      "grad_norm": 42.48833465576172,
      "learning_rate": 0.0003802176816959805,
      "loss": 822.3994140625,
      "step": 760
    },
    {
      "epoch": 0.16149328859060402,
      "grad_norm": 49.13645553588867,
      "learning_rate": 0.00037966809625601146,
      "loss": 826.9443359375,
      "step": 770
    },
    {
      "epoch": 0.16359060402684564,
      "grad_norm": 48.193199157714844,
      "learning_rate": 0.0003791114373496902,
      "loss": 821.3197265625,
      "step": 780
    },
    {
      "epoch": 0.16568791946308725,
      "grad_norm": 49.94052505493164,
      "learning_rate": 0.00037854772963759786,
      "loss": 814.4888671875,
      "step": 790
    },
    {
      "epoch": 0.16778523489932887,
      "grad_norm": 42.485595703125,
      "learning_rate": 0.0003779769980925852,
      "loss": 811.553369140625,
      "step": 800
    },
    {
      "epoch": 0.16988255033557048,
      "grad_norm": 44.533050537109375,
      "learning_rate": 0.0003773992679986661,
      "loss": 828.06875,
      "step": 810
    },
    {
      "epoch": 0.17197986577181207,
      "grad_norm": 41.512874603271484,
      "learning_rate": 0.00037681456494989785,
      "loss": 813.058349609375,
      "step": 820
    },
    {
      "epoch": 0.17407718120805368,
      "grad_norm": 42.67609405517578,
      "learning_rate": 0.00037622291484924676,
      "loss": 808.00771484375,
      "step": 830
    },
    {
      "epoch": 0.1761744966442953,
      "grad_norm": 45.48208999633789,
      "learning_rate": 0.00037562434390744123,
      "loss": 810.965771484375,
      "step": 840
    },
    {
      "epoch": 0.1782718120805369,
      "grad_norm": 43.91945266723633,
      "learning_rate": 0.00037501887864180994,
      "loss": 816.663232421875,
      "step": 850
    },
    {
      "epoch": 0.18036912751677853,
      "grad_norm": 49.31304168701172,
      "learning_rate": 0.00037440654587510774,
      "loss": 807.310546875,
      "step": 860
    },
    {
      "epoch": 0.18246644295302014,
      "grad_norm": 36.273887634277344,
      "learning_rate": 0.00037378737273432696,
      "loss": 798.728369140625,
      "step": 870
    },
    {
      "epoch": 0.18456375838926176,
      "grad_norm": 44.13105392456055,
      "learning_rate": 0.0003731613866494956,
      "loss": 805.8744140625,
      "step": 880
    },
    {
      "epoch": 0.18666107382550334,
      "grad_norm": 45.374900817871094,
      "learning_rate": 0.00037252861535246273,
      "loss": 806.17021484375,
      "step": 890
    },
    {
      "epoch": 0.18875838926174496,
      "grad_norm": 41.498138427734375,
      "learning_rate": 0.0003718890868756691,
      "loss": 798.01435546875,
      "step": 900
    },
    {
      "epoch": 0.19085570469798657,
      "grad_norm": 44.652252197265625,
      "learning_rate": 0.00037124282955090603,
      "loss": 797.1451171875,
      "step": 910
    },
    {
      "epoch": 0.1929530201342282,
      "grad_norm": 41.43980026245117,
      "learning_rate": 0.0003705898720080597,
      "loss": 803.592626953125,
      "step": 920
    },
    {
      "epoch": 0.1950503355704698,
      "grad_norm": 44.510902404785156,
      "learning_rate": 0.0003699302431738433,
      "loss": 797.47109375,
      "step": 930
    },
    {
      "epoch": 0.19714765100671142,
      "grad_norm": 37.388824462890625,
      "learning_rate": 0.0003692639722705152,
      "loss": 790.10703125,
      "step": 940
    },
    {
      "epoch": 0.19924496644295303,
      "grad_norm": 36.10209655761719,
      "learning_rate": 0.00036859108881458436,
      "loss": 789.671240234375,
      "step": 950
    },
    {
      "epoch": 0.20134228187919462,
      "grad_norm": 36.71262741088867,
      "learning_rate": 0.0003679116226155031,
      "loss": 797.35458984375,
      "step": 960
    },
    {
      "epoch": 0.20343959731543623,
      "grad_norm": 43.58747482299805,
      "learning_rate": 0.0003672256037743462,
      "loss": 787.1970703125,
      "step": 970
    },
    {
      "epoch": 0.20553691275167785,
      "grad_norm": 38.86661148071289,
      "learning_rate": 0.0003665330626824772,
      "loss": 785.28095703125,
      "step": 980
    },
    {
      "epoch": 0.20763422818791946,
      "grad_norm": 37.13184356689453,
      "learning_rate": 0.00036583403002020274,
      "loss": 784.462646484375,
      "step": 990
    },
    {
      "epoch": 0.20973154362416108,
      "grad_norm": 40.659400939941406,
      "learning_rate": 0.00036512853675541246,
      "loss": 791.35859375,
      "step": 1000
    },
    {
      "epoch": 0.2118288590604027,
      "grad_norm": 44.29231643676758,
      "learning_rate": 0.00036441661414220785,
      "loss": 784.15751953125,
      "step": 1010
    },
    {
      "epoch": 0.2139261744966443,
      "grad_norm": 40.54804611206055,
      "learning_rate": 0.0003636982937195172,
      "loss": 782.91201171875,
      "step": 1020
    },
    {
      "epoch": 0.2160234899328859,
      "grad_norm": 37.70088577270508,
      "learning_rate": 0.0003629736073096987,
      "loss": 782.441162109375,
      "step": 1030
    },
    {
      "epoch": 0.2181208053691275,
      "grad_norm": 36.9510612487793,
      "learning_rate": 0.0003622425870171304,
      "loss": 786.77060546875,
      "step": 1040
    },
    {
      "epoch": 0.22021812080536912,
      "grad_norm": 38.41440963745117,
      "learning_rate": 0.0003615052652267882,
      "loss": 778.93291015625,
      "step": 1050
    },
    {
      "epoch": 0.22231543624161074,
      "grad_norm": 43.160770416259766,
      "learning_rate": 0.000360761674602811,
      "loss": 775.032421875,
      "step": 1060
    },
    {
      "epoch": 0.22441275167785235,
      "grad_norm": 38.03440856933594,
      "learning_rate": 0.0003600118480870538,
      "loss": 770.278173828125,
      "step": 1070
    },
    {
      "epoch": 0.22651006711409397,
      "grad_norm": 38.17487716674805,
      "learning_rate": 0.0003592558188976279,
      "loss": 786.5306640625,
      "step": 1080
    },
    {
      "epoch": 0.22860738255033558,
      "grad_norm": 39.2180061340332,
      "learning_rate": 0.00035849362052743004,
      "loss": 780.42021484375,
      "step": 1090
    },
    {
      "epoch": 0.23070469798657717,
      "grad_norm": 36.615989685058594,
      "learning_rate": 0.00035772528674265803,
      "loss": 773.95361328125,
      "step": 1100
    },
    {
      "epoch": 0.23280201342281878,
      "grad_norm": 37.757266998291016,
      "learning_rate": 0.000356950851581315,
      "loss": 767.45908203125,
      "step": 1110
    },
    {
      "epoch": 0.2348993288590604,
      "grad_norm": 34.517276763916016,
      "learning_rate": 0.0003561703493517017,
      "loss": 778.86240234375,
      "step": 1120
    },
    {
      "epoch": 0.236996644295302,
      "grad_norm": 35.07598114013672,
      "learning_rate": 0.00035538381463089645,
      "loss": 772.373095703125,
      "step": 1130
    },
    {
      "epoch": 0.23909395973154363,
      "grad_norm": 36.75849151611328,
      "learning_rate": 0.0003545912822632232,
      "loss": 766.216015625,
      "step": 1140
    },
    {
      "epoch": 0.24119127516778524,
      "grad_norm": 35.139984130859375,
      "learning_rate": 0.0003537927873587081,
      "loss": 763.338525390625,
      "step": 1150
    },
    {
      "epoch": 0.24328859060402686,
      "grad_norm": 36.147621154785156,
      "learning_rate": 0.000352988365291524,
      "loss": 773.221142578125,
      "step": 1160
    },
    {
      "epoch": 0.24538590604026847,
      "grad_norm": 40.629539489746094,
      "learning_rate": 0.0003521780516984234,
      "loss": 766.941748046875,
      "step": 1170
    },
    {
      "epoch": 0.24748322147651006,
      "grad_norm": 33.118160247802734,
      "learning_rate": 0.00035136188247715967,
      "loss": 761.17802734375,
      "step": 1180
    },
    {
      "epoch": 0.24958053691275167,
      "grad_norm": 33.310420989990234,
      "learning_rate": 0.0003505398937848967,
      "loss": 761.124462890625,
      "step": 1190
    },
    {
      "epoch": 0.2516778523489933,
      "grad_norm": 34.028202056884766,
      "learning_rate": 0.0003497121220366071,
      "loss": 771.833837890625,
      "step": 1200
    },
    {
      "epoch": 0.2537751677852349,
      "grad_norm": 36.89501953125,
      "learning_rate": 0.0003488786039034591,
      "loss": 759.81103515625,
      "step": 1210
    },
    {
      "epoch": 0.2558724832214765,
      "grad_norm": 33.90177536010742,
      "learning_rate": 0.0003480393763111917,
      "loss": 759.73125,
      "step": 1220
    },
    {
      "epoch": 0.25796979865771813,
      "grad_norm": 37.61647415161133,
      "learning_rate": 0.00034719447643847927,
      "loss": 756.5494140625,
      "step": 1230
    },
    {
      "epoch": 0.2600671140939597,
      "grad_norm": 32.71785354614258,
      "learning_rate": 0.00034634394171528403,
      "loss": 767.09501953125,
      "step": 1240
    },
    {
      "epoch": 0.26216442953020136,
      "grad_norm": 35.68928909301758,
      "learning_rate": 0.00034548780982119806,
      "loss": 755.03046875,
      "step": 1250
    },
    {
      "epoch": 0.26426174496644295,
      "grad_norm": 37.2677116394043,
      "learning_rate": 0.00034462611868377424,
      "loss": 755.941796875,
      "step": 1260
    },
    {
      "epoch": 0.2663590604026846,
      "grad_norm": 35.99335861206055,
      "learning_rate": 0.0003437589064768457,
      "loss": 753.703564453125,
      "step": 1270
    },
    {
      "epoch": 0.2684563758389262,
      "grad_norm": 34.95077896118164,
      "learning_rate": 0.0003428862116188348,
      "loss": 764.32802734375,
      "step": 1280
    },
    {
      "epoch": 0.27055369127516776,
      "grad_norm": 32.43818664550781,
      "learning_rate": 0.0003420080727710512,
      "loss": 756.809033203125,
      "step": 1290
    },
    {
      "epoch": 0.2726510067114094,
      "grad_norm": 33.54920959472656,
      "learning_rate": 0.0003411245288359792,
      "loss": 751.679833984375,
      "step": 1300
    },
    {
      "epoch": 0.274748322147651,
      "grad_norm": 33.81038284301758,
      "learning_rate": 0.00034023561895555395,
      "loss": 754.639111328125,
      "step": 1310
    },
    {
      "epoch": 0.27684563758389263,
      "grad_norm": 36.19585418701172,
      "learning_rate": 0.000339341382509428,
      "loss": 763.320654296875,
      "step": 1320
    },
    {
      "epoch": 0.2789429530201342,
      "grad_norm": 32.98769760131836,
      "learning_rate": 0.00033844185911322623,
      "loss": 753.488671875,
      "step": 1330
    },
    {
      "epoch": 0.28104026845637586,
      "grad_norm": 33.56504821777344,
      "learning_rate": 0.000337537088616791,
      "loss": 747.947998046875,
      "step": 1340
    },
    {
      "epoch": 0.28313758389261745,
      "grad_norm": 32.269737243652344,
      "learning_rate": 0.00033662711110241697,
      "loss": 750.63203125,
      "step": 1350
    },
    {
      "epoch": 0.28523489932885904,
      "grad_norm": 33.046749114990234,
      "learning_rate": 0.000335711966883075,
      "loss": 759.793017578125,
      "step": 1360
    },
    {
      "epoch": 0.2873322147651007,
      "grad_norm": 36.37275695800781,
      "learning_rate": 0.0003347916965006266,
      "loss": 744.22412109375,
      "step": 1370
    },
    {
      "epoch": 0.28942953020134227,
      "grad_norm": 32.7099494934082,
      "learning_rate": 0.00033386634072402776,
      "loss": 746.197607421875,
      "step": 1380
    },
    {
      "epoch": 0.2915268456375839,
      "grad_norm": 31.205947875976562,
      "learning_rate": 0.0003329359405475228,
      "loss": 748.474365234375,
      "step": 1390
    },
    {
      "epoch": 0.2936241610738255,
      "grad_norm": 28.828580856323242,
      "learning_rate": 0.00033200053718882835,
      "loss": 751.84599609375,
      "step": 1400
    },
    {
      "epoch": 0.29572147651006714,
      "grad_norm": 34.22489547729492,
      "learning_rate": 0.0003310601720873073,
      "loss": 746.10986328125,
      "step": 1410
    },
    {
      "epoch": 0.2978187919463087,
      "grad_norm": 29.78771209716797,
      "learning_rate": 0.00033011488690213296,
      "loss": 741.93037109375,
      "step": 1420
    },
    {
      "epoch": 0.2999161073825503,
      "grad_norm": 39.27846145629883,
      "learning_rate": 0.0003291647235104438,
      "loss": 743.660400390625,
      "step": 1430
    },
    {
      "epoch": 0.30201342281879195,
      "grad_norm": 31.457706451416016,
      "learning_rate": 0.00032820972400548766,
      "loss": 754.897265625,
      "step": 1440
    },
    {
      "epoch": 0.30411073825503354,
      "grad_norm": 32.80243682861328,
      "learning_rate": 0.00032724993069475764,
      "loss": 742.11484375,
      "step": 1450
    },
    {
      "epoch": 0.3062080536912752,
      "grad_norm": 34.35562515258789,
      "learning_rate": 0.00032628538609811745,
      "loss": 742.219775390625,
      "step": 1460
    },
    {
      "epoch": 0.30830536912751677,
      "grad_norm": 33.573577880859375,
      "learning_rate": 0.0003253161329459178,
      "loss": 745.37138671875,
      "step": 1470
    },
    {
      "epoch": 0.3104026845637584,
      "grad_norm": 32.058101654052734,
      "learning_rate": 0.00032434221417710333,
      "loss": 753.84130859375,
      "step": 1480
    },
    {
      "epoch": 0.3125,
      "grad_norm": 31.893978118896484,
      "learning_rate": 0.0003233636729373106,
      "loss": 741.5162109375,
      "step": 1490
    },
    {
      "epoch": 0.3145973154362416,
      "grad_norm": 36.885005950927734,
      "learning_rate": 0.00032238055257695645,
      "loss": 734.73974609375,
      "step": 1500
    },
    {
      "epoch": 0.31669463087248323,
      "grad_norm": 42.988834381103516,
      "learning_rate": 0.0003213928966493177,
      "loss": 739.119580078125,
      "step": 1510
    },
    {
      "epoch": 0.3187919463087248,
      "grad_norm": 29.642290115356445,
      "learning_rate": 0.00032040074890860156,
      "loss": 752.603564453125,
      "step": 1520
    },
    {
      "epoch": 0.32088926174496646,
      "grad_norm": 29.82461929321289,
      "learning_rate": 0.00031940415330800737,
      "loss": 734.8591796875,
      "step": 1530
    },
    {
      "epoch": 0.32298657718120805,
      "grad_norm": 34.89297866821289,
      "learning_rate": 0.0003184031539977794,
      "loss": 732.84501953125,
      "step": 1540
    },
    {
      "epoch": 0.3250838926174497,
      "grad_norm": 34.86556625366211,
      "learning_rate": 0.0003173977953232508,
      "loss": 736.87841796875,
      "step": 1550
    },
    {
      "epoch": 0.3271812080536913,
      "grad_norm": 32.03319549560547,
      "learning_rate": 0.0003163881218228794,
      "loss": 742.657666015625,
      "step": 1560
    },
    {
      "epoch": 0.32927852348993286,
      "grad_norm": 30.916311264038086,
      "learning_rate": 0.0003153741782262743,
      "loss": 734.43837890625,
      "step": 1570
    },
    {
      "epoch": 0.3313758389261745,
      "grad_norm": 31.965553283691406,
      "learning_rate": 0.0003143560094522143,
      "loss": 730.24296875,
      "step": 1580
    },
    {
      "epoch": 0.3334731543624161,
      "grad_norm": 31.810258865356445,
      "learning_rate": 0.0003133336606066583,
      "loss": 732.60625,
      "step": 1590
    },
    {
      "epoch": 0.33557046979865773,
      "grad_norm": 30.140993118286133,
      "learning_rate": 0.0003123071769807465,
      "loss": 744.71767578125,
      "step": 1600
    },
    {
      "epoch": 0.3376677852348993,
      "grad_norm": 60.086910247802734,
      "learning_rate": 0.00031127660404879463,
      "loss": 730.52451171875,
      "step": 1610
    },
    {
      "epoch": 0.33976510067114096,
      "grad_norm": 29.318464279174805,
      "learning_rate": 0.00031024198746627867,
      "loss": 727.399609375,
      "step": 1620
    },
    {
      "epoch": 0.34186241610738255,
      "grad_norm": 32.13727951049805,
      "learning_rate": 0.00030920337306781284,
      "loss": 729.554833984375,
      "step": 1630
    },
    {
      "epoch": 0.34395973154362414,
      "grad_norm": 30.829389572143555,
      "learning_rate": 0.00030816080686511875,
      "loss": 743.65224609375,
      "step": 1640
    },
    {
      "epoch": 0.3460570469798658,
      "grad_norm": 28.18938446044922,
      "learning_rate": 0.0003071143350449872,
      "loss": 729.828125,
      "step": 1650
    },
    {
      "epoch": 0.34815436241610737,
      "grad_norm": 27.202878952026367,
      "learning_rate": 0.00030606400396723186,
      "loss": 724.96044921875,
      "step": 1660
    },
    {
      "epoch": 0.350251677852349,
      "grad_norm": 39.66158676147461,
      "learning_rate": 0.0003050098601626359,
      "loss": 727.02880859375,
      "step": 1670
    },
    {
      "epoch": 0.3523489932885906,
      "grad_norm": 28.056537628173828,
      "learning_rate": 0.00030395195033089005,
      "loss": 737.3568359375,
      "step": 1680
    },
    {
      "epoch": 0.35444630872483224,
      "grad_norm": 28.82770347595215,
      "learning_rate": 0.0003028903213385242,
      "loss": 732.960888671875,
      "step": 1690
    },
    {
      "epoch": 0.3565436241610738,
      "grad_norm": 30.616226196289062,
      "learning_rate": 0.0003018250202168308,
      "loss": 725.4568359375,
      "step": 1700
    },
    {
      "epoch": 0.3586409395973154,
      "grad_norm": 27.57636070251465,
      "learning_rate": 0.00030075609415978203,
      "loss": 718.759716796875,
      "step": 1710
    },
    {
      "epoch": 0.36073825503355705,
      "grad_norm": 31.625394821166992,
      "learning_rate": 0.00029968359052193785,
      "loss": 725.88857421875,
      "step": 1720
    },
    {
      "epoch": 0.36283557046979864,
      "grad_norm": 27.597814559936523,
      "learning_rate": 0.00029860755681634953,
      "loss": 730.74892578125,
      "step": 1730
    },
    {
      "epoch": 0.3649328859060403,
      "grad_norm": 28.66240119934082,
      "learning_rate": 0.0002975280407124536,
      "loss": 722.49345703125,
      "step": 1740
    },
    {
      "epoch": 0.36703020134228187,
      "grad_norm": 27.074068069458008,
      "learning_rate": 0.00029644509003396075,
      "loss": 734.28369140625,
      "step": 1750
    },
    {
      "epoch": 0.3691275167785235,
      "grad_norm": 29.8409366607666,
      "learning_rate": 0.00029535875275673706,
      "loss": 738.629541015625,
      "step": 1760
    },
    {
      "epoch": 0.3712248322147651,
      "grad_norm": 28.747722625732422,
      "learning_rate": 0.0002942690770066783,
      "loss": 736.925048828125,
      "step": 1770
    },
    {
      "epoch": 0.3733221476510067,
      "grad_norm": 26.50994110107422,
      "learning_rate": 0.00029317611105757836,
      "loss": 733.42626953125,
      "step": 1780
    },
    {
      "epoch": 0.37541946308724833,
      "grad_norm": 27.646202087402344,
      "learning_rate": 0.00029207990332899016,
      "loss": 733.86328125,
      "step": 1790
    },
    {
      "epoch": 0.3775167785234899,
      "grad_norm": 28.608095169067383,
      "learning_rate": 0.00029098050238408137,
      "loss": 734.38486328125,
      "step": 1800
    },
    {
      "epoch": 0.37961409395973156,
      "grad_norm": 27.358755111694336,
      "learning_rate": 0.0002898779569274819,
      "loss": 733.97880859375,
      "step": 1810
    },
    {
      "epoch": 0.38171140939597314,
      "grad_norm": 29.266138076782227,
      "learning_rate": 0.00028877231580312743,
      "loss": 731.997412109375,
      "step": 1820
    },
    {
      "epoch": 0.3838087248322148,
      "grad_norm": 28.123027801513672,
      "learning_rate": 0.0002876636279920948,
      "loss": 732.208837890625,
      "step": 1830
    },
    {
      "epoch": 0.3859060402684564,
      "grad_norm": 28.35926628112793,
      "learning_rate": 0.0002865519426104322,
      "loss": 730.1966796875,
      "step": 1840
    },
    {
      "epoch": 0.38800335570469796,
      "grad_norm": 26.62183952331543,
      "learning_rate": 0.00028543730890698353,
      "loss": 730.421484375,
      "step": 1850
    },
    {
      "epoch": 0.3901006711409396,
      "grad_norm": 26.039772033691406,
      "learning_rate": 0.00028431977626120657,
      "loss": 731.6849609375,
      "step": 1860
    },
    {
      "epoch": 0.3921979865771812,
      "grad_norm": 28.390792846679688,
      "learning_rate": 0.00028319939418098493,
      "loss": 728.848291015625,
      "step": 1870
    },
    {
      "epoch": 0.39429530201342283,
      "grad_norm": 28.321779251098633,
      "learning_rate": 0.0002820762123004356,
      "loss": 729.0255859375,
      "step": 1880
    },
    {
      "epoch": 0.3963926174496644,
      "grad_norm": 28.969951629638672,
      "learning_rate": 0.0002809502803777095,
      "loss": 729.58837890625,
      "step": 1890
    },
    {
      "epoch": 0.39848993288590606,
      "grad_norm": 25.39800453186035,
      "learning_rate": 0.00027982164829278754,
      "loss": 727.68134765625,
      "step": 1900
    },
    {
      "epoch": 0.40058724832214765,
      "grad_norm": 29.222623825073242,
      "learning_rate": 0.00027869036604527043,
      "loss": 729.85634765625,
      "step": 1910
    },
    {
      "epoch": 0.40268456375838924,
      "grad_norm": 30.955928802490234,
      "learning_rate": 0.000277556483752164,
      "loss": 727.31318359375,
      "step": 1920
    },
    {
      "epoch": 0.4047818791946309,
      "grad_norm": 29.15259552001953,
      "learning_rate": 0.0002764200516456589,
      "loss": 728.60517578125,
      "step": 1930
    },
    {
      "epoch": 0.40687919463087246,
      "grad_norm": 28.483592987060547,
      "learning_rate": 0.00027528112007090525,
      "loss": 729.50048828125,
      "step": 1940
    },
    {
      "epoch": 0.4089765100671141,
      "grad_norm": 27.479665756225586,
      "learning_rate": 0.0002741397394837821,
      "loss": 725.64970703125,
      "step": 1950
    },
    {
      "epoch": 0.4110738255033557,
      "grad_norm": 31.93459701538086,
      "learning_rate": 0.0002729959604486626,
      "loss": 728.76865234375,
      "step": 1960
    },
    {
      "epoch": 0.41317114093959734,
      "grad_norm": 29.415254592895508,
      "learning_rate": 0.0002718498336361734,
      "loss": 727.822216796875,
      "step": 1970
    },
    {
      "epoch": 0.4152684563758389,
      "grad_norm": 25.998945236206055,
      "learning_rate": 0.0002707014098209505,
      "loss": 725.624560546875,
      "step": 1980
    },
    {
      "epoch": 0.4173657718120805,
      "grad_norm": 33.45818328857422,
      "learning_rate": 0.00026955073987938933,
      "loss": 730.52978515625,
      "step": 1990
    },
    {
      "epoch": 0.41946308724832215,
      "grad_norm": 28.501672744750977,
      "learning_rate": 0.0002683978747873912,
      "loss": 725.5873046875,
      "step": 2000
    },
    {
      "epoch": 0.42156040268456374,
      "grad_norm": 28.55805206298828,
      "learning_rate": 0.0002672428656181049,
      "loss": 723.09990234375,
      "step": 2010
    },
    {
      "epoch": 0.4236577181208054,
      "grad_norm": 27.068178176879883,
      "learning_rate": 0.0002660857635396641,
      "loss": 723.821875,
      "step": 2020
    },
    {
      "epoch": 0.42575503355704697,
      "grad_norm": 27.90006446838379,
      "learning_rate": 0.0002649266198129206,
      "loss": 718.319140625,
      "step": 2030
    },
    {
      "epoch": 0.4278523489932886,
      "grad_norm": 28.625322341918945,
      "learning_rate": 0.00026376548578917343,
      "loss": 724.609619140625,
      "step": 2040
    },
    {
      "epoch": 0.4299496644295302,
      "grad_norm": 31.34563636779785,
      "learning_rate": 0.0002626024129078938,
      "loss": 719.7814453125,
      "step": 2050
    },
    {
      "epoch": 0.4320469798657718,
      "grad_norm": 26.983905792236328,
      "learning_rate": 0.00026143745269444643,
      "loss": 722.23017578125,
      "step": 2060
    },
    {
      "epoch": 0.4341442953020134,
      "grad_norm": 29.680858612060547,
      "learning_rate": 0.0002602706567578069,
      "loss": 720.07890625,
      "step": 2070
    },
    {
      "epoch": 0.436241610738255,
      "grad_norm": 31.25922966003418,
      "learning_rate": 0.000259102076788275,
      "loss": 720.61064453125,
      "step": 2080
    },
    {
      "epoch": 0.43833892617449666,
      "grad_norm": 25.962329864501953,
      "learning_rate": 0.00025793176455518547,
      "loss": 724.55810546875,
      "step": 2090
    },
    {
      "epoch": 0.44043624161073824,
      "grad_norm": 27.091432571411133,
      "learning_rate": 0.000256759771904614,
      "loss": 721.320458984375,
      "step": 2100
    },
    {
      "epoch": 0.4425335570469799,
      "grad_norm": 33.23907470703125,
      "learning_rate": 0.00025558615075708057,
      "loss": 719.5658203125,
      "step": 2110
    },
    {
      "epoch": 0.4446308724832215,
      "grad_norm": 27.312124252319336,
      "learning_rate": 0.0002544109531052492,
      "loss": 723.623046875,
      "step": 2120
    },
    {
      "epoch": 0.44672818791946306,
      "grad_norm": 25.877601623535156,
      "learning_rate": 0.00025323423101162484,
      "loss": 723.46025390625,
      "step": 2130
    },
    {
      "epoch": 0.4488255033557047,
      "grad_norm": 26.769596099853516,
      "learning_rate": 0.00025205603660624675,
      "loss": 718.48427734375,
      "step": 2140
    },
    {
      "epoch": 0.4509228187919463,
      "grad_norm": 25.944448471069336,
      "learning_rate": 0.00025087642208437936,
      "loss": 716.782421875,
      "step": 2150
    },
    {
      "epoch": 0.45302013422818793,
      "grad_norm": 28.341115951538086,
      "learning_rate": 0.00024969543970419956,
      "loss": 720.827490234375,
      "step": 2160
    },
    {
      "epoch": 0.4551174496644295,
      "grad_norm": 26.407798767089844,
      "learning_rate": 0.00024851314178448184,
      "loss": 720.140234375,
      "step": 2170
    },
    {
      "epoch": 0.45721476510067116,
      "grad_norm": 28.054893493652344,
      "learning_rate": 0.0002473295807022806,
      "loss": 714.789990234375,
      "step": 2180
    },
    {
      "epoch": 0.45931208053691275,
      "grad_norm": 26.11944007873535,
      "learning_rate": 0.00024614480889060956,
      "loss": 719.75595703125,
      "step": 2190
    },
    {
      "epoch": 0.46140939597315433,
      "grad_norm": 26.539321899414062,
      "learning_rate": 0.0002449588788361192,
      "loss": 716.24951171875,
      "step": 2200
    },
    {
      "epoch": 0.463506711409396,
      "grad_norm": 27.907739639282227,
      "learning_rate": 0.0002437718430767712,
      "loss": 718.15361328125,
      "step": 2210
    },
    {
      "epoch": 0.46560402684563756,
      "grad_norm": 26.146020889282227,
      "learning_rate": 0.00024258375419951124,
      "loss": 718.4966796875,
      "step": 2220
    },
    {
      "epoch": 0.4677013422818792,
      "grad_norm": 29.019397735595703,
      "learning_rate": 0.00024139466483793926,
      "loss": 718.3244140625,
      "step": 2230
    },
    {
      "epoch": 0.4697986577181208,
      "grad_norm": 27.466388702392578,
      "learning_rate": 0.00024020462766997764,
      "loss": 717.709375,
      "step": 2240
    },
    {
      "epoch": 0.47189597315436244,
      "grad_norm": 28.666629791259766,
      "learning_rate": 0.0002390136954155377,
      "loss": 720.43193359375,
      "step": 2250
    },
    {
      "epoch": 0.473993288590604,
      "grad_norm": 34.103336334228516,
      "learning_rate": 0.00023782192083418395,
      "loss": 718.31767578125,
      "step": 2260
    },
    {
      "epoch": 0.47609060402684567,
      "grad_norm": 29.75569725036621,
      "learning_rate": 0.00023662935672279684,
      "loss": 715.5525390625,
      "step": 2270
    },
    {
      "epoch": 0.47818791946308725,
      "grad_norm": 26.467947006225586,
      "learning_rate": 0.0002354360559132341,
      "loss": 718.21328125,
      "step": 2280
    },
    {
      "epoch": 0.48028523489932884,
      "grad_norm": 27.470417022705078,
      "learning_rate": 0.0002342420712699895,
      "loss": 713.359033203125,
      "step": 2290
    },
    {
      "epoch": 0.4823825503355705,
      "grad_norm": 28.60648536682129,
      "learning_rate": 0.0002330474556878519,
      "loss": 715.951904296875,
      "step": 2300
    },
    {
      "epoch": 0.48447986577181207,
      "grad_norm": 29.705326080322266,
      "learning_rate": 0.00023185226208956107,
      "loss": 716.36962890625,
      "step": 2310
    },
    {
      "epoch": 0.4865771812080537,
      "grad_norm": 26.422521591186523,
      "learning_rate": 0.00023065654342346363,
      "loss": 717.31083984375,
      "step": 2320
    },
    {
      "epoch": 0.4886744966442953,
      "grad_norm": 49.67610168457031,
      "learning_rate": 0.0002294603526611672,
      "loss": 717.445703125,
      "step": 2330
    },
    {
      "epoch": 0.49077181208053694,
      "grad_norm": 27.15092658996582,
      "learning_rate": 0.0002282637427951939,
      "loss": 717.614111328125,
      "step": 2340
    },
    {
      "epoch": 0.4928691275167785,
      "grad_norm": 26.76052474975586,
      "learning_rate": 0.00022706676683663239,
      "loss": 716.904833984375,
      "step": 2350
    },
    {
      "epoch": 0.4949664429530201,
      "grad_norm": 27.0493106842041,
      "learning_rate": 0.00022586947781278976,
      "loss": 715.82685546875,
      "step": 2360
    },
    {
      "epoch": 0.49706375838926176,
      "grad_norm": 27.63730812072754,
      "learning_rate": 0.00022467192876484223,
      "loss": 713.4669921875,
      "step": 2370
    },
    {
      "epoch": 0.49916107382550334,
      "grad_norm": 26.431467056274414,
      "learning_rate": 0.00022347417274548524,
      "loss": 716.6283203125,
      "step": 2380
    },
    {
      "epoch": 0.5012583892617449,
      "grad_norm": 26.240686416625977,
      "learning_rate": 0.00022227626281658353,
      "loss": 716.108740234375,
      "step": 2390
    },
    {
      "epoch": 0.5033557046979866,
      "grad_norm": 27.485626220703125,
      "learning_rate": 0.0002210782520468199,
      "loss": 713.7689453125,
      "step": 2400
    },
    {
      "epoch": 0.5054530201342282,
      "grad_norm": 28.32665252685547,
      "learning_rate": 0.00021988019350934469,
      "loss": 716.45576171875,
      "step": 2410
    },
    {
      "epoch": 0.5075503355704698,
      "grad_norm": 29.576181411743164,
      "learning_rate": 0.00021868214027942435,
      "loss": 712.18935546875,
      "step": 2420
    },
    {
      "epoch": 0.5096476510067114,
      "grad_norm": 27.404142379760742,
      "learning_rate": 0.00021748414543209014,
      "loss": 709.567724609375,
      "step": 2430
    },
    {
      "epoch": 0.511744966442953,
      "grad_norm": 25.870176315307617,
      "learning_rate": 0.000216286262039787,
      "loss": 712.60595703125,
      "step": 2440
    },
    {
      "epoch": 0.5138422818791947,
      "grad_norm": 25.011234283447266,
      "learning_rate": 0.00021508854317002244,
      "loss": 712.865966796875,
      "step": 2450
    },
    {
      "epoch": 0.5159395973154363,
      "grad_norm": 26.89911460876465,
      "learning_rate": 0.00021389104188301498,
      "loss": 713.5578125,
      "step": 2460
    },
    {
      "epoch": 0.5180369127516778,
      "grad_norm": 25.816125869750977,
      "learning_rate": 0.0002126938112293445,
      "loss": 714.549267578125,
      "step": 2470
    },
    {
      "epoch": 0.5201342281879194,
      "grad_norm": 24.598859786987305,
      "learning_rate": 0.00021149690424760115,
      "loss": 710.9453125,
      "step": 2480
    },
    {
      "epoch": 0.522231543624161,
      "grad_norm": 25.585561752319336,
      "learning_rate": 0.00021030037396203648,
      "loss": 710.580859375,
      "step": 2490
    },
    {
      "epoch": 0.5243288590604027,
      "grad_norm": 26.241941452026367,
      "learning_rate": 0.00020910427338021354,
      "loss": 710.328271484375,
      "step": 2500
    },
    {
      "epoch": 0.5264261744966443,
      "grad_norm": 24.599651336669922,
      "learning_rate": 0.0002079086554906594,
      "loss": 712.6861328125,
      "step": 2510
    },
    {
      "epoch": 0.5285234899328859,
      "grad_norm": 26.42645835876465,
      "learning_rate": 0.00020671357326051712,
      "loss": 712.0703125,
      "step": 2520
    },
    {
      "epoch": 0.5306208053691275,
      "grad_norm": 28.313579559326172,
      "learning_rate": 0.00020551907963319957,
      "loss": 710.93173828125,
      "step": 2530
    },
    {
      "epoch": 0.5327181208053692,
      "grad_norm": 24.56991958618164,
      "learning_rate": 0.00020432522752604398,
      "loss": 711.9142578125,
      "step": 2540
    },
    {
      "epoch": 0.5348154362416108,
      "grad_norm": 25.90009117126465,
      "learning_rate": 0.00020313206982796743,
      "loss": 710.29892578125,
      "step": 2550
    },
    {
      "epoch": 0.5369127516778524,
      "grad_norm": 26.455686569213867,
      "learning_rate": 0.00020193965939712397,
      "loss": 704.16845703125,
      "step": 2560
    },
    {
      "epoch": 0.5390100671140939,
      "grad_norm": 24.865222930908203,
      "learning_rate": 0.000200748049058563,
      "loss": 706.30869140625,
      "step": 2570
    },
    {
      "epoch": 0.5411073825503355,
      "grad_norm": 24.55437469482422,
      "learning_rate": 0.0001995572916018888,
      "loss": 705.83759765625,
      "step": 2580
    },
    {
      "epoch": 0.5432046979865772,
      "grad_norm": 25.504764556884766,
      "learning_rate": 0.00019836743977892223,
      "loss": 708.3240234375,
      "step": 2590
    },
    {
      "epoch": 0.5453020134228188,
      "grad_norm": 25.164077758789062,
      "learning_rate": 0.0001971785463013636,
      "loss": 705.59912109375,
      "step": 2600
    },
    {
      "epoch": 0.5473993288590604,
      "grad_norm": 23.877357482910156,
      "learning_rate": 0.00019599066383845743,
      "loss": 708.88505859375,
      "step": 2610
    },
    {
      "epoch": 0.549496644295302,
      "grad_norm": 27.058509826660156,
      "learning_rate": 0.00019480384501465925,
      "loss": 707.0541015625,
      "step": 2620
    },
    {
      "epoch": 0.5515939597315436,
      "grad_norm": 29.5660343170166,
      "learning_rate": 0.0001936181424073043,
      "loss": 709.3982421875,
      "step": 2630
    },
    {
      "epoch": 0.5536912751677853,
      "grad_norm": 25.05129051208496,
      "learning_rate": 0.00019243360854427808,
      "loss": 708.7525390625,
      "step": 2640
    },
    {
      "epoch": 0.5557885906040269,
      "grad_norm": 26.034975051879883,
      "learning_rate": 0.0001912502959016897,
      "loss": 705.56171875,
      "step": 2650
    },
    {
      "epoch": 0.5578859060402684,
      "grad_norm": 25.737016677856445,
      "learning_rate": 0.00019006825690154683,
      "loss": 708.69892578125,
      "step": 2660
    },
    {
      "epoch": 0.55998322147651,
      "grad_norm": 26.19977569580078,
      "learning_rate": 0.00018888754390943325,
      "loss": 706.80400390625,
      "step": 2670
    },
    {
      "epoch": 0.5620805369127517,
      "grad_norm": 25.73967933654785,
      "learning_rate": 0.0001877082092321895,
      "loss": 703.82470703125,
      "step": 2680
    },
    {
      "epoch": 0.5641778523489933,
      "grad_norm": 24.554115295410156,
      "learning_rate": 0.00018653030511559497,
      "loss": 701.961083984375,
      "step": 2690
    },
    {
      "epoch": 0.5662751677852349,
      "grad_norm": 27.607160568237305,
      "learning_rate": 0.0001853538837420539,
      "loss": 702.956298828125,
      "step": 2700
    },
    {
      "epoch": 0.5683724832214765,
      "grad_norm": 27.145713806152344,
      "learning_rate": 0.00018417899722828341,
      "loss": 701.333984375,
      "step": 2710
    },
    {
      "epoch": 0.5704697986577181,
      "grad_norm": 26.37285041809082,
      "learning_rate": 0.0001830056976230047,
      "loss": 700.68173828125,
      "step": 2720
    },
    {
      "epoch": 0.5725671140939598,
      "grad_norm": 28.513526916503906,
      "learning_rate": 0.00018183403690463708,
      "loss": 704.793017578125,
      "step": 2730
    },
    {
      "epoch": 0.5746644295302014,
      "grad_norm": 27.879859924316406,
      "learning_rate": 0.00018066406697899576,
      "loss": 704.948876953125,
      "step": 2740
    },
    {
      "epoch": 0.576761744966443,
      "grad_norm": 26.13296890258789,
      "learning_rate": 0.00017949583967699168,
      "loss": 705.763671875,
      "step": 2750
    },
    {
      "epoch": 0.5788590604026845,
      "grad_norm": 24.964950561523438,
      "learning_rate": 0.0001783294067523359,
      "loss": 701.885009765625,
      "step": 2760
    },
    {
      "epoch": 0.5809563758389261,
      "grad_norm": 24.55069351196289,
      "learning_rate": 0.00017716481987924674,
      "loss": 703.38232421875,
      "step": 2770
    },
    {
      "epoch": 0.5830536912751678,
      "grad_norm": 23.53226661682129,
      "learning_rate": 0.00017600213065016012,
      "loss": 700.89462890625,
      "step": 2780
    },
    {
      "epoch": 0.5851510067114094,
      "grad_norm": 24.06133270263672,
      "learning_rate": 0.00017484139057344472,
      "loss": 701.2154296875,
      "step": 2790
    },
    {
      "epoch": 0.587248322147651,
      "grad_norm": 25.854036331176758,
      "learning_rate": 0.00017368265107111944,
      "loss": 702.04521484375,
      "step": 2800
    },
    {
      "epoch": 0.5893456375838926,
      "grad_norm": 24.032381057739258,
      "learning_rate": 0.00017252596347657574,
      "loss": 699.448583984375,
      "step": 2810
    },
    {
      "epoch": 0.5914429530201343,
      "grad_norm": 24.54527473449707,
      "learning_rate": 0.00017137137903230323,
      "loss": 693.13505859375,
      "step": 2820
    },
    {
      "epoch": 0.5935402684563759,
      "grad_norm": 25.05814552307129,
      "learning_rate": 0.00017021894888761998,
      "loss": 696.93486328125,
      "step": 2830
    },
    {
      "epoch": 0.5956375838926175,
      "grad_norm": 23.6625919342041,
      "learning_rate": 0.00016906872409640607,
      "loss": 697.78701171875,
      "step": 2840
    },
    {
      "epoch": 0.597734899328859,
      "grad_norm": 23.811920166015625,
      "learning_rate": 0.00016792075561484216,
      "loss": 698.3015625,
      "step": 2850
    },
    {
      "epoch": 0.5998322147651006,
      "grad_norm": 23.187089920043945,
      "learning_rate": 0.00016677509429915198,
      "loss": 699.289208984375,
      "step": 2860
    },
    {
      "epoch": 0.6019295302013423,
      "grad_norm": 24.96914291381836,
      "learning_rate": 0.0001656317909033494,
      "loss": 699.05029296875,
      "step": 2870
    },
    {
      "epoch": 0.6040268456375839,
      "grad_norm": 22.78306770324707,
      "learning_rate": 0.0001644908960769899,
      "loss": 699.95283203125,
      "step": 2880
    },
    {
      "epoch": 0.6061241610738255,
      "grad_norm": 25.343849182128906,
      "learning_rate": 0.00016335246036292675,
      "loss": 696.755615234375,
      "step": 2890
    },
    {
      "epoch": 0.6082214765100671,
      "grad_norm": 26.005495071411133,
      "learning_rate": 0.00016221653419507207,
      "loss": 699.58486328125,
      "step": 2900
    },
    {
      "epoch": 0.6103187919463087,
      "grad_norm": 24.401639938354492,
      "learning_rate": 0.00016108316789616213,
      "loss": 696.3732421875,
      "step": 2910
    },
    {
      "epoch": 0.6124161073825504,
      "grad_norm": 24.31859588623047,
      "learning_rate": 0.00015995241167552856,
      "loss": 700.176953125,
      "step": 2920
    },
    {
      "epoch": 0.614513422818792,
      "grad_norm": 31.25603675842285,
      "learning_rate": 0.00015882431562687366,
      "loss": 698.004443359375,
      "step": 2930
    },
    {
      "epoch": 0.6166107382550335,
      "grad_norm": 25.79340934753418,
      "learning_rate": 0.00015769892972605125,
      "loss": 698.315625,
      "step": 2940
    },
    {
      "epoch": 0.6187080536912751,
      "grad_norm": 23.582733154296875,
      "learning_rate": 0.0001565763038288528,
      "loss": 691.946630859375,
      "step": 2950
    },
    {
      "epoch": 0.6208053691275168,
      "grad_norm": 26.264375686645508,
      "learning_rate": 0.0001554564876687986,
      "loss": 699.2384765625,
      "step": 2960
    },
    {
      "epoch": 0.6229026845637584,
      "grad_norm": 26.286468505859375,
      "learning_rate": 0.0001543395308549346,
      "loss": 697.033544921875,
      "step": 2970
    },
    {
      "epoch": 0.625,
      "grad_norm": 25.956165313720703,
      "learning_rate": 0.00015322548286963476,
      "loss": 698.5783203125,
      "step": 2980
    },
    {
      "epoch": 0.6270973154362416,
      "grad_norm": 26.558116912841797,
      "learning_rate": 0.0001521143930664087,
      "loss": 695.806298828125,
      "step": 2990
    },
    {
      "epoch": 0.6291946308724832,
      "grad_norm": 24.454757690429688,
      "learning_rate": 0.00015100631066771562,
      "loss": 698.87841796875,
      "step": 3000
    },
    {
      "epoch": 0.6312919463087249,
      "grad_norm": 24.38653564453125,
      "learning_rate": 0.00014990128476278337,
      "loss": 696.64560546875,
      "step": 3010
    },
    {
      "epoch": 0.6333892617449665,
      "grad_norm": 25.045244216918945,
      "learning_rate": 0.00014879936430543385,
      "loss": 697.28154296875,
      "step": 3020
    },
    {
      "epoch": 0.635486577181208,
      "grad_norm": 24.258338928222656,
      "learning_rate": 0.00014770059811191447,
      "loss": 700.28935546875,
      "step": 3030
    },
    {
      "epoch": 0.6375838926174496,
      "grad_norm": 24.587146759033203,
      "learning_rate": 0.00014660503485873535,
      "loss": 698.47548828125,
      "step": 3040
    },
    {
      "epoch": 0.6396812080536913,
      "grad_norm": 24.621126174926758,
      "learning_rate": 0.0001455127230805128,
      "loss": 697.98115234375,
      "step": 3050
    },
    {
      "epoch": 0.6417785234899329,
      "grad_norm": 23.170021057128906,
      "learning_rate": 0.0001444237111678195,
      "loss": 693.545849609375,
      "step": 3060
    },
    {
      "epoch": 0.6438758389261745,
      "grad_norm": 24.980670928955078,
      "learning_rate": 0.00014333804736504056,
      "loss": 694.336376953125,
      "step": 3070
    },
    {
      "epoch": 0.6459731543624161,
      "grad_norm": 27.34921646118164,
      "learning_rate": 0.00014225577976823627,
      "loss": 699.40654296875,
      "step": 3080
    },
    {
      "epoch": 0.6480704697986577,
      "grad_norm": 25.38248062133789,
      "learning_rate": 0.00014117695632301139,
      "loss": 696.404150390625,
      "step": 3090
    },
    {
      "epoch": 0.6501677852348994,
      "grad_norm": 25.941797256469727,
      "learning_rate": 0.00014010162482239088,
      "loss": 696.0484375,
      "step": 3100
    },
    {
      "epoch": 0.652265100671141,
      "grad_norm": 25.004669189453125,
      "learning_rate": 0.00013902983290470317,
      "loss": 696.5728515625,
      "step": 3110
    },
    {
      "epoch": 0.6543624161073825,
      "grad_norm": 23.577775955200195,
      "learning_rate": 0.0001379616280514692,
      "loss": 696.34404296875,
      "step": 3120
    },
    {
      "epoch": 0.6564597315436241,
      "grad_norm": 24.280529022216797,
      "learning_rate": 0.0001368970575852993,
      "loss": 697.48203125,
      "step": 3130
    },
    {
      "epoch": 0.6585570469798657,
      "grad_norm": 23.28042984008789,
      "learning_rate": 0.00013583616866779652,
      "loss": 690.66455078125,
      "step": 3140
    },
    {
      "epoch": 0.6606543624161074,
      "grad_norm": 24.422304153442383,
      "learning_rate": 0.00013477900829746725,
      "loss": 694.853955078125,
      "step": 3150
    },
    {
      "epoch": 0.662751677852349,
      "grad_norm": 22.98871421813965,
      "learning_rate": 0.00013372562330763943,
      "loss": 696.26904296875,
      "step": 3160
    },
    {
      "epoch": 0.6648489932885906,
      "grad_norm": 25.702192306518555,
      "learning_rate": 0.00013267606036438762,
      "loss": 693.992333984375,
      "step": 3170
    },
    {
      "epoch": 0.6669463087248322,
      "grad_norm": 23.169281005859375,
      "learning_rate": 0.00013163036596446564,
      "loss": 691.390869140625,
      "step": 3180
    },
    {
      "epoch": 0.6690436241610739,
      "grad_norm": 24.02280044555664,
      "learning_rate": 0.00013058858643324675,
      "loss": 692.553173828125,
      "step": 3190
    },
    {
      "epoch": 0.6711409395973155,
      "grad_norm": 23.723628997802734,
      "learning_rate": 0.00012955076792267134,
      "loss": 687.7876953125,
      "step": 3200
    },
    {
      "epoch": 0.673238255033557,
      "grad_norm": 23.922508239746094,
      "learning_rate": 0.00012851695640920228,
      "loss": 691.37255859375,
      "step": 3210
    },
    {
      "epoch": 0.6753355704697986,
      "grad_norm": 23.9777889251709,
      "learning_rate": 0.00012748719769178836,
      "loss": 689.18642578125,
      "step": 3220
    },
    {
      "epoch": 0.6774328859060402,
      "grad_norm": 23.559240341186523,
      "learning_rate": 0.00012646153738983518,
      "loss": 688.249755859375,
      "step": 3230
    },
    {
      "epoch": 0.6795302013422819,
      "grad_norm": 25.408052444458008,
      "learning_rate": 0.0001254400209411841,
      "loss": 691.2591796875,
      "step": 3240
    },
    {
      "epoch": 0.6816275167785235,
      "grad_norm": 22.419090270996094,
      "learning_rate": 0.00012442269360009943,
      "loss": 692.08369140625,
      "step": 3250
    },
    {
      "epoch": 0.6837248322147651,
      "grad_norm": 25.38911247253418,
      "learning_rate": 0.00012340960043526354,
      "loss": 693.0171875,
      "step": 3260
    },
    {
      "epoch": 0.6858221476510067,
      "grad_norm": 26.608829498291016,
      "learning_rate": 0.0001224007863277804,
      "loss": 691.488037109375,
      "step": 3270
    },
    {
      "epoch": 0.6879194630872483,
      "grad_norm": 23.240951538085938,
      "learning_rate": 0.00012139629596918692,
      "loss": 691.84345703125,
      "step": 3280
    },
    {
      "epoch": 0.69001677852349,
      "grad_norm": 23.333946228027344,
      "learning_rate": 0.00012039617385947362,
      "loss": 693.92431640625,
      "step": 3290
    },
    {
      "epoch": 0.6921140939597316,
      "grad_norm": 26.162805557250977,
      "learning_rate": 0.0001194004643051128,
      "loss": 690.355908203125,
      "step": 3300
    },
    {
      "epoch": 0.6942114093959731,
      "grad_norm": 21.26649284362793,
      "learning_rate": 0.00011840921141709596,
      "loss": 691.67705078125,
      "step": 3310
    },
    {
      "epoch": 0.6963087248322147,
      "grad_norm": 23.431072235107422,
      "learning_rate": 0.00011742245910897946,
      "loss": 691.683349609375,
      "step": 3320
    },
    {
      "epoch": 0.6984060402684564,
      "grad_norm": 23.77557945251465,
      "learning_rate": 0.00011644025109493925,
      "loss": 687.243115234375,
      "step": 3330
    },
    {
      "epoch": 0.700503355704698,
      "grad_norm": 23.87569236755371,
      "learning_rate": 0.0001154626308878342,
      "loss": 687.0201171875,
      "step": 3340
    },
    {
      "epoch": 0.7026006711409396,
      "grad_norm": 21.87002944946289,
      "learning_rate": 0.00011448964179727846,
      "loss": 687.4349609375,
      "step": 3350
    },
    {
      "epoch": 0.7046979865771812,
      "grad_norm": 23.286794662475586,
      "learning_rate": 0.00011352132692772275,
      "loss": 690.5765625,
      "step": 3360
    },
    {
      "epoch": 0.7067953020134228,
      "grad_norm": 27.445226669311523,
      "learning_rate": 0.0001125577291765449,
      "loss": 691.41484375,
      "step": 3370
    },
    {
      "epoch": 0.7088926174496645,
      "grad_norm": 23.718297958374023,
      "learning_rate": 0.00011159889123214947,
      "loss": 691.42841796875,
      "step": 3380
    },
    {
      "epoch": 0.7109899328859061,
      "grad_norm": 24.48626708984375,
      "learning_rate": 0.00011064485557207618,
      "loss": 688.381884765625,
      "step": 3390
    },
    {
      "epoch": 0.7130872483221476,
      "grad_norm": 25.892141342163086,
      "learning_rate": 0.00010969566446111879,
      "loss": 690.50966796875,
      "step": 3400
    },
    {
      "epoch": 0.7151845637583892,
      "grad_norm": 23.631595611572266,
      "learning_rate": 0.00010875135994945229,
      "loss": 690.73642578125,
      "step": 3410
    },
    {
      "epoch": 0.7172818791946308,
      "grad_norm": 25.584020614624023,
      "learning_rate": 0.00010781198387077007,
      "loss": 691.406640625,
      "step": 3420
    },
    {
      "epoch": 0.7193791946308725,
      "grad_norm": 25.226858139038086,
      "learning_rate": 0.00010687757784043082,
      "loss": 689.0376953125,
      "step": 3430
    },
    {
      "epoch": 0.7214765100671141,
      "grad_norm": 23.374839782714844,
      "learning_rate": 0.00010594818325361471,
      "loss": 688.791650390625,
      "step": 3440
    },
    {
      "epoch": 0.7235738255033557,
      "grad_norm": 24.697153091430664,
      "learning_rate": 0.00010502384128348954,
      "loss": 689.23720703125,
      "step": 3450
    },
    {
      "epoch": 0.7256711409395973,
      "grad_norm": 25.77028465270996,
      "learning_rate": 0.00010410459287938703,
      "loss": 687.74521484375,
      "step": 3460
    },
    {
      "epoch": 0.727768456375839,
      "grad_norm": 25.273399353027344,
      "learning_rate": 0.00010319047876498844,
      "loss": 688.24140625,
      "step": 3470
    },
    {
      "epoch": 0.7298657718120806,
      "grad_norm": 25.190956115722656,
      "learning_rate": 0.00010228153943652041,
      "loss": 684.262451171875,
      "step": 3480
    },
    {
      "epoch": 0.7319630872483222,
      "grad_norm": 24.785213470458984,
      "learning_rate": 0.0001013778151609613,
      "loss": 685.20732421875,
      "step": 3490
    },
    {
      "epoch": 0.7340604026845637,
      "grad_norm": 24.36125946044922,
      "learning_rate": 0.00010047934597425688,
      "loss": 685.501708984375,
      "step": 3500
    },
    {
      "epoch": 0.7361577181208053,
      "grad_norm": 26.047834396362305,
      "learning_rate": 9.958617167954706e-05,
      "loss": 683.8818359375,
      "step": 3510
    },
    {
      "epoch": 0.738255033557047,
      "grad_norm": 22.43846893310547,
      "learning_rate": 9.869833184540213e-05,
      "loss": 687.33955078125,
      "step": 3520
    },
    {
      "epoch": 0.7403523489932886,
      "grad_norm": 22.825130462646484,
      "learning_rate": 9.781586580407048e-05,
      "loss": 685.4853515625,
      "step": 3530
    },
    {
      "epoch": 0.7424496644295302,
      "grad_norm": 23.325157165527344,
      "learning_rate": 9.693881264973552e-05,
      "loss": 685.995556640625,
      "step": 3540
    },
    {
      "epoch": 0.7445469798657718,
      "grad_norm": 23.20819664001465,
      "learning_rate": 9.606721123678415e-05,
      "loss": 685.19609375,
      "step": 3550
    },
    {
      "epoch": 0.7466442953020134,
      "grad_norm": 24.189987182617188,
      "learning_rate": 9.520110017808516e-05,
      "loss": 683.45908203125,
      "step": 3560
    },
    {
      "epoch": 0.7487416107382551,
      "grad_norm": 26.871835708618164,
      "learning_rate": 9.434051784327894e-05,
      "loss": 685.9505859375,
      "step": 3570
    },
    {
      "epoch": 0.7508389261744967,
      "grad_norm": 22.42962646484375,
      "learning_rate": 9.348550235707753e-05,
      "loss": 686.2205078125,
      "step": 3580
    },
    {
      "epoch": 0.7529362416107382,
      "grad_norm": 22.479677200317383,
      "learning_rate": 9.263609159757563e-05,
      "loss": 684.18955078125,
      "step": 3590
    },
    {
      "epoch": 0.7550335570469798,
      "grad_norm": 22.954744338989258,
      "learning_rate": 9.179232319457263e-05,
      "loss": 686.7052734375,
      "step": 3600
    },
    {
      "epoch": 0.7571308724832215,
      "grad_norm": 23.088876724243164,
      "learning_rate": 9.095423452790545e-05,
      "loss": 685.3046875,
      "step": 3610
    },
    {
      "epoch": 0.7592281879194631,
      "grad_norm": 24.69822120666504,
      "learning_rate": 9.012186272579281e-05,
      "loss": 685.684521484375,
      "step": 3620
    },
    {
      "epoch": 0.7613255033557047,
      "grad_norm": 25.845073699951172,
      "learning_rate": 8.929524466319001e-05,
      "loss": 683.671142578125,
      "step": 3630
    },
    {
      "epoch": 0.7634228187919463,
      "grad_norm": 23.543195724487305,
      "learning_rate": 8.84744169601558e-05,
      "loss": 684.632177734375,
      "step": 3640
    },
    {
      "epoch": 0.7655201342281879,
      "grad_norm": 22.957509994506836,
      "learning_rate": 8.765941598022974e-05,
      "loss": 685.3962890625,
      "step": 3650
    },
    {
      "epoch": 0.7676174496644296,
      "grad_norm": 22.979705810546875,
      "learning_rate": 8.685027782882137e-05,
      "loss": 683.3501953125,
      "step": 3660
    },
    {
      "epoch": 0.7697147651006712,
      "grad_norm": 22.801042556762695,
      "learning_rate": 8.604703835161078e-05,
      "loss": 680.969775390625,
      "step": 3670
    },
    {
      "epoch": 0.7718120805369127,
      "grad_norm": 23.970422744750977,
      "learning_rate": 8.524973313296044e-05,
      "loss": 683.797705078125,
      "step": 3680
    },
    {
      "epoch": 0.7739093959731543,
      "grad_norm": 23.845319747924805,
      "learning_rate": 8.445839749433879e-05,
      "loss": 680.916796875,
      "step": 3690
    },
    {
      "epoch": 0.7760067114093959,
      "grad_norm": 23.90414047241211,
      "learning_rate": 8.36730664927557e-05,
      "loss": 684.574951171875,
      "step": 3700
    },
    {
      "epoch": 0.7781040268456376,
      "grad_norm": 22.141538619995117,
      "learning_rate": 8.289377491920906e-05,
      "loss": 686.685107421875,
      "step": 3710
    },
    {
      "epoch": 0.7802013422818792,
      "grad_norm": 22.511852264404297,
      "learning_rate": 8.212055729714395e-05,
      "loss": 682.07216796875,
      "step": 3720
    },
    {
      "epoch": 0.7822986577181208,
      "grad_norm": 23.860305786132812,
      "learning_rate": 8.135344788092268e-05,
      "loss": 683.2880859375,
      "step": 3730
    },
    {
      "epoch": 0.7843959731543624,
      "grad_norm": 24.159772872924805,
      "learning_rate": 8.059248065430754e-05,
      "loss": 683.333544921875,
      "step": 3740
    },
    {
      "epoch": 0.7864932885906041,
      "grad_norm": 25.201622009277344,
      "learning_rate": 7.983768932895552e-05,
      "loss": 684.48095703125,
      "step": 3750
    },
    {
      "epoch": 0.7885906040268457,
      "grad_norm": 23.27935791015625,
      "learning_rate": 7.908910734292443e-05,
      "loss": 682.99365234375,
      "step": 3760
    },
    {
      "epoch": 0.7906879194630873,
      "grad_norm": 23.803667068481445,
      "learning_rate": 7.834676785919184e-05,
      "loss": 683.983349609375,
      "step": 3770
    },
    {
      "epoch": 0.7927852348993288,
      "grad_norm": 22.286733627319336,
      "learning_rate": 7.761070376418576e-05,
      "loss": 683.07724609375,
      "step": 3780
    },
    {
      "epoch": 0.7948825503355704,
      "grad_norm": 23.109832763671875,
      "learning_rate": 7.688094766632796e-05,
      "loss": 683.54560546875,
      "step": 3790
    },
    {
      "epoch": 0.7969798657718121,
      "grad_norm": 22.52189826965332,
      "learning_rate": 7.615753189458901e-05,
      "loss": 680.865185546875,
      "step": 3800
    },
    {
      "epoch": 0.7990771812080537,
      "grad_norm": 24.4259090423584,
      "learning_rate": 7.544048849705643e-05,
      "loss": 681.9951171875,
      "step": 3810
    },
    {
      "epoch": 0.8011744966442953,
      "grad_norm": 24.631175994873047,
      "learning_rate": 7.47298492395148e-05,
      "loss": 682.766259765625,
      "step": 3820
    },
    {
      "epoch": 0.8032718120805369,
      "grad_norm": 24.409805297851562,
      "learning_rate": 7.402564560403849e-05,
      "loss": 680.79716796875,
      "step": 3830
    },
    {
      "epoch": 0.8053691275167785,
      "grad_norm": 23.451454162597656,
      "learning_rate": 7.332790878759689e-05,
      "loss": 682.815185546875,
      "step": 3840
    },
    {
      "epoch": 0.8074664429530202,
      "grad_norm": 24.23875617980957,
      "learning_rate": 7.263666970067252e-05,
      "loss": 682.263330078125,
      "step": 3850
    },
    {
      "epoch": 0.8095637583892618,
      "grad_norm": 24.88316535949707,
      "learning_rate": 7.19519589658916e-05,
      "loss": 680.236279296875,
      "step": 3860
    },
    {
      "epoch": 0.8116610738255033,
      "grad_norm": 24.26546859741211,
      "learning_rate": 7.127380691666728e-05,
      "loss": 682.1150390625,
      "step": 3870
    },
    {
      "epoch": 0.8137583892617449,
      "grad_norm": 24.102313995361328,
      "learning_rate": 7.060224359585608e-05,
      "loss": 680.230615234375,
      "step": 3880
    },
    {
      "epoch": 0.8158557046979866,
      "grad_norm": 23.298391342163086,
      "learning_rate": 6.993729875442679e-05,
      "loss": 683.053955078125,
      "step": 3890
    },
    {
      "epoch": 0.8179530201342282,
      "grad_norm": 22.2108097076416,
      "learning_rate": 6.927900185014261e-05,
      "loss": 679.271044921875,
      "step": 3900
    },
    {
      "epoch": 0.8200503355704698,
      "grad_norm": 25.65964126586914,
      "learning_rate": 6.862738204625598e-05,
      "loss": 681.87509765625,
      "step": 3910
    },
    {
      "epoch": 0.8221476510067114,
      "grad_norm": 24.050338745117188,
      "learning_rate": 6.798246821021675e-05,
      "loss": 679.88876953125,
      "step": 3920
    },
    {
      "epoch": 0.824244966442953,
      "grad_norm": 24.42888832092285,
      "learning_rate": 6.734428891239319e-05,
      "loss": 680.12578125,
      "step": 3930
    },
    {
      "epoch": 0.8263422818791947,
      "grad_norm": 24.935672760009766,
      "learning_rate": 6.671287242480648e-05,
      "loss": 681.04013671875,
      "step": 3940
    },
    {
      "epoch": 0.8284395973154363,
      "grad_norm": 25.129545211791992,
      "learning_rate": 6.608824671987803e-05,
      "loss": 678.59296875,
      "step": 3950
    },
    {
      "epoch": 0.8305369127516778,
      "grad_norm": 24.83428192138672,
      "learning_rate": 6.547043946919043e-05,
      "loss": 682.446533203125,
      "step": 3960
    },
    {
      "epoch": 0.8326342281879194,
      "grad_norm": 24.628469467163086,
      "learning_rate": 6.485947804226143e-05,
      "loss": 677.285546875,
      "step": 3970
    },
    {
      "epoch": 0.834731543624161,
      "grad_norm": 24.317861557006836,
      "learning_rate": 6.425538950533153e-05,
      "loss": 679.77578125,
      "step": 3980
    },
    {
      "epoch": 0.8368288590604027,
      "grad_norm": 23.234214782714844,
      "learning_rate": 6.36582006201649e-05,
      "loss": 680.6521484375,
      "step": 3990
    },
    {
      "epoch": 0.8389261744966443,
      "grad_norm": 22.52564811706543,
      "learning_rate": 6.306793784286387e-05,
      "loss": 677.32109375,
      "step": 4000
    },
    {
      "epoch": 0.8410234899328859,
      "grad_norm": 23.42584800720215,
      "learning_rate": 6.248462732269678e-05,
      "loss": 681.26416015625,
      "step": 4010
    },
    {
      "epoch": 0.8431208053691275,
      "grad_norm": 23.126731872558594,
      "learning_rate": 6.190829490093956e-05,
      "loss": 677.49208984375,
      "step": 4020
    },
    {
      "epoch": 0.8452181208053692,
      "grad_norm": 23.167861938476562,
      "learning_rate": 6.133896610973102e-05,
      "loss": 680.8375,
      "step": 4030
    },
    {
      "epoch": 0.8473154362416108,
      "grad_norm": 22.692258834838867,
      "learning_rate": 6.0776666170941634e-05,
      "loss": 679.8037109375,
      "step": 4040
    },
    {
      "epoch": 0.8494127516778524,
      "grad_norm": 22.7047061920166,
      "learning_rate": 6.022141999505628e-05,
      "loss": 681.933935546875,
      "step": 4050
    },
    {
      "epoch": 0.8515100671140939,
      "grad_norm": 23.45292091369629,
      "learning_rate": 5.967325218007066e-05,
      "loss": 680.2009765625,
      "step": 4060
    },
    {
      "epoch": 0.8536073825503355,
      "grad_norm": 24.43787956237793,
      "learning_rate": 5.913218701040155e-05,
      "loss": 676.1564453125,
      "step": 4070
    },
    {
      "epoch": 0.8557046979865772,
      "grad_norm": 24.374242782592773,
      "learning_rate": 5.859824845581097e-05,
      "loss": 678.980322265625,
      "step": 4080
    },
    {
      "epoch": 0.8578020134228188,
      "grad_norm": 23.4588565826416,
      "learning_rate": 5.807146017034434e-05,
      "loss": 678.86953125,
      "step": 4090
    },
    {
      "epoch": 0.8598993288590604,
      "grad_norm": 23.756441116333008,
      "learning_rate": 5.755184549128258e-05,
      "loss": 681.618994140625,
      "step": 4100
    },
    {
      "epoch": 0.861996644295302,
      "grad_norm": 25.192712783813477,
      "learning_rate": 5.703942743810811e-05,
      "loss": 677.488037109375,
      "step": 4110
    },
    {
      "epoch": 0.8640939597315436,
      "grad_norm": 24.152353286743164,
      "learning_rate": 5.653422871148536e-05,
      "loss": 679.816748046875,
      "step": 4120
    },
    {
      "epoch": 0.8661912751677853,
      "grad_norm": 22.807497024536133,
      "learning_rate": 5.6036271692254784e-05,
      "loss": 678.42666015625,
      "step": 4130
    },
    {
      "epoch": 0.8682885906040269,
      "grad_norm": 25.389739990234375,
      "learning_rate": 5.554557844044157e-05,
      "loss": 679.6646484375,
      "step": 4140
    },
    {
      "epoch": 0.8703859060402684,
      "grad_norm": 22.66519546508789,
      "learning_rate": 5.506217069427819e-05,
      "loss": 676.6759765625,
      "step": 4150
    },
    {
      "epoch": 0.87248322147651,
      "grad_norm": 24.590576171875,
      "learning_rate": 5.458606986924159e-05,
      "loss": 678.8537109375,
      "step": 4160
    },
    {
      "epoch": 0.8745805369127517,
      "grad_norm": 22.72840118408203,
      "learning_rate": 5.4117297057104274e-05,
      "loss": 677.28154296875,
      "step": 4170
    },
    {
      "epoch": 0.8766778523489933,
      "grad_norm": 22.46537208557129,
      "learning_rate": 5.365587302500005e-05,
      "loss": 675.40986328125,
      "step": 4180
    },
    {
      "epoch": 0.8787751677852349,
      "grad_norm": 29.71303939819336,
      "learning_rate": 5.320181821450388e-05,
      "loss": 677.40400390625,
      "step": 4190
    },
    {
      "epoch": 0.8808724832214765,
      "grad_norm": 23.0802059173584,
      "learning_rate": 5.2755152740726445e-05,
      "loss": 674.81826171875,
      "step": 4200
    },
    {
      "epoch": 0.8829697986577181,
      "grad_norm": 23.870975494384766,
      "learning_rate": 5.231589639142296e-05,
      "loss": 675.855322265625,
      "step": 4210
    },
    {
      "epoch": 0.8850671140939598,
      "grad_norm": 24.489511489868164,
      "learning_rate": 5.1884068626116434e-05,
      "loss": 676.7513671875,
      "step": 4220
    },
    {
      "epoch": 0.8871644295302014,
      "grad_norm": 22.859840393066406,
      "learning_rate": 5.145968857523584e-05,
      "loss": 675.190576171875,
      "step": 4230
    },
    {
      "epoch": 0.889261744966443,
      "grad_norm": 23.360322952270508,
      "learning_rate": 5.1042775039268465e-05,
      "loss": 672.90810546875,
      "step": 4240
    },
    {
      "epoch": 0.8913590604026845,
      "grad_norm": 24.995880126953125,
      "learning_rate": 5.0633346487926995e-05,
      "loss": 676.45029296875,
      "step": 4250
    },
    {
      "epoch": 0.8934563758389261,
      "grad_norm": 23.80856704711914,
      "learning_rate": 5.0231421059331404e-05,
      "loss": 673.52822265625,
      "step": 4260
    },
    {
      "epoch": 0.8955536912751678,
      "grad_norm": 24.263673782348633,
      "learning_rate": 4.9837016559205346e-05,
      "loss": 674.06865234375,
      "step": 4270
    },
    {
      "epoch": 0.8976510067114094,
      "grad_norm": 23.509172439575195,
      "learning_rate": 4.9450150460087276e-05,
      "loss": 675.976171875,
      "step": 4280
    },
    {
      "epoch": 0.899748322147651,
      "grad_norm": 24.816654205322266,
      "learning_rate": 4.907083990055655e-05,
      "loss": 673.865185546875,
      "step": 4290
    },
    {
      "epoch": 0.9018456375838926,
      "grad_norm": 23.49567222595215,
      "learning_rate": 4.8699101684474055e-05,
      "loss": 676.16005859375,
      "step": 4300
    },
    {
      "epoch": 0.9039429530201343,
      "grad_norm": 25.57634925842285,
      "learning_rate": 4.8334952280237875e-05,
      "loss": 673.1826171875,
      "step": 4310
    },
    {
      "epoch": 0.9060402684563759,
      "grad_norm": 24.98488998413086,
      "learning_rate": 4.797840782005362e-05,
      "loss": 676.1115234375,
      "step": 4320
    },
    {
      "epoch": 0.9081375838926175,
      "grad_norm": 24.250463485717773,
      "learning_rate": 4.762948409921976e-05,
      "loss": 674.84365234375,
      "step": 4330
    },
    {
      "epoch": 0.910234899328859,
      "grad_norm": 25.994915008544922,
      "learning_rate": 4.7288196575427945e-05,
      "loss": 674.15810546875,
      "step": 4340
    },
    {
      "epoch": 0.9123322147651006,
      "grad_norm": 22.44687271118164,
      "learning_rate": 4.69545603680782e-05,
      "loss": 673.69814453125,
      "step": 4350
    },
    {
      "epoch": 0.9144295302013423,
      "grad_norm": 24.059288024902344,
      "learning_rate": 4.6628590257609074e-05,
      "loss": 675.2294921875,
      "step": 4360
    },
    {
      "epoch": 0.9165268456375839,
      "grad_norm": 23.866180419921875,
      "learning_rate": 4.63103006848429e-05,
      "loss": 674.218896484375,
      "step": 4370
    },
    {
      "epoch": 0.9186241610738255,
      "grad_norm": 23.916122436523438,
      "learning_rate": 4.599970575034604e-05,
      "loss": 674.363037109375,
      "step": 4380
    },
    {
      "epoch": 0.9207214765100671,
      "grad_norm": 26.1231632232666,
      "learning_rate": 4.569681921380412e-05,
      "loss": 675.52900390625,
      "step": 4390
    },
    {
      "epoch": 0.9228187919463087,
      "grad_norm": 24.31917953491211,
      "learning_rate": 4.540165449341264e-05,
      "loss": 671.547607421875,
      "step": 4400
    },
    {
      "epoch": 0.9249161073825504,
      "grad_norm": 25.087600708007812,
      "learning_rate": 4.511422466528239e-05,
      "loss": 673.746875,
      "step": 4410
    },
    {
      "epoch": 0.927013422818792,
      "grad_norm": 25.387001037597656,
      "learning_rate": 4.4834542462860253e-05,
      "loss": 672.59580078125,
      "step": 4420
    },
    {
      "epoch": 0.9291107382550335,
      "grad_norm": 25.317588806152344,
      "learning_rate": 4.4562620276364964e-05,
      "loss": 670.037890625,
      "step": 4430
    },
    {
      "epoch": 0.9312080536912751,
      "grad_norm": 26.259300231933594,
      "learning_rate": 4.4298470152238445e-05,
      "loss": 672.92578125,
      "step": 4440
    },
    {
      "epoch": 0.9333053691275168,
      "grad_norm": 26.846012115478516,
      "learning_rate": 4.404210379261189e-05,
      "loss": 672.447265625,
      "step": 4450
    },
    {
      "epoch": 0.9354026845637584,
      "grad_norm": 23.90787124633789,
      "learning_rate": 4.3793532554787466e-05,
      "loss": 672.944970703125,
      "step": 4460
    },
    {
      "epoch": 0.9375,
      "grad_norm": 28.882965087890625,
      "learning_rate": 4.355276745073519e-05,
      "loss": 671.073095703125,
      "step": 4470
    },
    {
      "epoch": 0.9395973154362416,
      "grad_norm": 26.22090721130371,
      "learning_rate": 4.3319819146605076e-05,
      "loss": 674.5400390625,
      "step": 4480
    },
    {
      "epoch": 0.9416946308724832,
      "grad_norm": 24.984975814819336,
      "learning_rate": 4.309469796225452e-05,
      "loss": 675.30908203125,
      "step": 4490
    },
    {
      "epoch": 0.9437919463087249,
      "grad_norm": 23.35356903076172,
      "learning_rate": 4.2877413870791265e-05,
      "loss": 675.83837890625,
      "step": 4500
    },
    {
      "epoch": 0.9458892617449665,
      "grad_norm": 26.29161834716797,
      "learning_rate": 4.2667976498131484e-05,
      "loss": 673.48466796875,
      "step": 4510
    },
    {
      "epoch": 0.947986577181208,
      "grad_norm": 24.633699417114258,
      "learning_rate": 4.2466395122573355e-05,
      "loss": 674.9208984375,
      "step": 4520
    },
    {
      "epoch": 0.9500838926174496,
      "grad_norm": 23.10916519165039,
      "learning_rate": 4.227267867438603e-05,
      "loss": 676.23984375,
      "step": 4530
    },
    {
      "epoch": 0.9521812080536913,
      "grad_norm": 26.628690719604492,
      "learning_rate": 4.208683573541408e-05,
      "loss": 678.2919921875,
      "step": 4540
    },
    {
      "epoch": 0.9542785234899329,
      "grad_norm": 25.501096725463867,
      "learning_rate": 4.1908874538697176e-05,
      "loss": 674.9896484375,
      "step": 4550
    },
    {
      "epoch": 0.9563758389261745,
      "grad_norm": 26.29096794128418,
      "learning_rate": 4.1738802968105485e-05,
      "loss": 675.5177734375,
      "step": 4560
    },
    {
      "epoch": 0.9584731543624161,
      "grad_norm": 25.067407608032227,
      "learning_rate": 4.1576628557990294e-05,
      "loss": 674.130517578125,
      "step": 4570
    },
    {
      "epoch": 0.9605704697986577,
      "grad_norm": 25.92313003540039,
      "learning_rate": 4.142235849285035e-05,
      "loss": 675.11962890625,
      "step": 4580
    },
    {
      "epoch": 0.9626677852348994,
      "grad_norm": 26.59023666381836,
      "learning_rate": 4.127599960701349e-05,
      "loss": 676.64970703125,
      "step": 4590
    },
    {
      "epoch": 0.964765100671141,
      "grad_norm": 25.86935806274414,
      "learning_rate": 4.113755838433387e-05,
      "loss": 673.15,
      "step": 4600
    },
    {
      "epoch": 0.9668624161073825,
      "grad_norm": 26.771474838256836,
      "learning_rate": 4.1007040957904835e-05,
      "loss": 673.42724609375,
      "step": 4610
    },
    {
      "epoch": 0.9689597315436241,
      "grad_norm": 23.894729614257812,
      "learning_rate": 4.088445310978703e-05,
      "loss": 671.387646484375,
      "step": 4620
    },
    {
      "epoch": 0.9710570469798657,
      "grad_norm": 25.638530731201172,
      "learning_rate": 4.076980027075245e-05,
      "loss": 670.821240234375,
      "step": 4630
    },
    {
      "epoch": 0.9731543624161074,
      "grad_norm": 29.335660934448242,
      "learning_rate": 4.066308752004368e-05,
      "loss": 672.8646484375,
      "step": 4640
    },
    {
      "epoch": 0.975251677852349,
      "grad_norm": 24.27260971069336,
      "learning_rate": 4.056431958514902e-05,
      "loss": 672.34208984375,
      "step": 4650
    },
    {
      "epoch": 0.9773489932885906,
      "grad_norm": 25.83704948425293,
      "learning_rate": 4.047350084159293e-05,
      "loss": 672.367578125,
      "step": 4660
    },
    {
      "epoch": 0.9794463087248322,
      "grad_norm": 24.82935905456543,
      "learning_rate": 4.039063531274225e-05,
      "loss": 671.880322265625,
      "step": 4670
    },
    {
      "epoch": 0.9815436241610739,
      "grad_norm": 25.728792190551758,
      "learning_rate": 4.0315726669628004e-05,
      "loss": 671.256298828125,
      "step": 4680
    },
    {
      "epoch": 0.9836409395973155,
      "grad_norm": 26.020658493041992,
      "learning_rate": 4.0248778230782685e-05,
      "loss": 672.35029296875,
      "step": 4690
    },
    {
      "epoch": 0.985738255033557,
      "grad_norm": 26.47356414794922,
      "learning_rate": 4.018979296209327e-05,
      "loss": 671.268408203125,
      "step": 4700
    },
    {
      "epoch": 0.9878355704697986,
      "grad_norm": 25.695480346679688,
      "learning_rate": 4.013877347666988e-05,
      "loss": 669.023486328125,
      "step": 4710
    },
    {
      "epoch": 0.9899328859060402,
      "grad_norm": 28.546607971191406,
      "learning_rate": 4.009572203472991e-05,
      "loss": 675.0337890625,
      "step": 4720
    },
    {
      "epoch": 0.9920302013422819,
      "grad_norm": 24.35601043701172,
      "learning_rate": 4.0060640543498e-05,
      "loss": 675.898876953125,
      "step": 4730
    },
    {
      "epoch": 0.9941275167785235,
      "grad_norm": 25.573619842529297,
      "learning_rate": 4.003353055712148e-05,
      "loss": 674.478271484375,
      "step": 4740
    },
    {
      "epoch": 0.9962248322147651,
      "grad_norm": 24.39256477355957,
      "learning_rate": 4.0014393276601606e-05,
      "loss": 675.296240234375,
      "step": 4750
    },
    {
      "epoch": 0.9983221476510067,
      "grad_norm": 25.135915756225586,
      "learning_rate": 4.000322954974018e-05,
      "loss": 676.800146484375,
      "step": 4760
    },
    {
      "epoch": 1.0,
      "step": 4768,
      "total_flos": 2.1166212224737346e+19,
      "train_loss": 764.8368647018535,
      "train_runtime": 201110.1372,
      "train_samples_per_second": 12.139,
      "train_steps_per_second": 0.024
    }
  ],
  "logging_steps": 10,
  "max_steps": 4768,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 9223372036854775807,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 2.1166212224737346e+19,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}