{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 8.0,
  "eval_steps": 500,
  "global_step": 10296,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.00777302759424796,
      "grad_norm": 19.018803349985753,
      "learning_rate": 8.737864077669904e-07,
      "loss": 8.2473,
      "step": 10
    },
    {
      "epoch": 0.01554605518849592,
      "grad_norm": 17.967208179671328,
      "learning_rate": 1.8446601941747572e-06,
      "loss": 8.1304,
      "step": 20
    },
    {
      "epoch": 0.023319082782743878,
      "grad_norm": 9.21576906708124,
      "learning_rate": 2.8155339805825245e-06,
      "loss": 7.4913,
      "step": 30
    },
    {
      "epoch": 0.03109211037699184,
      "grad_norm": 4.8422984662740465,
      "learning_rate": 3.7864077669902915e-06,
      "loss": 6.7578,
      "step": 40
    },
    {
      "epoch": 0.0388651379712398,
      "grad_norm": 3.0883237528757603,
      "learning_rate": 4.7572815533980585e-06,
      "loss": 6.2789,
      "step": 50
    },
    {
      "epoch": 0.046638165565487756,
      "grad_norm": 3.0641013238554837,
      "learning_rate": 5.728155339805825e-06,
      "loss": 5.9381,
      "step": 60
    },
    {
      "epoch": 0.054411193159735714,
      "grad_norm": 3.6989178160715803,
      "learning_rate": 6.699029126213592e-06,
      "loss": 5.547,
      "step": 70
    },
    {
      "epoch": 0.06218422075398368,
      "grad_norm": 3.8926133061231023,
      "learning_rate": 7.66990291262136e-06,
      "loss": 4.9449,
      "step": 80
    },
    {
      "epoch": 0.06995724834823164,
      "grad_norm": 6.005297193380389,
      "learning_rate": 8.640776699029127e-06,
      "loss": 4.2741,
      "step": 90
    },
    {
      "epoch": 0.0777302759424796,
      "grad_norm": 4.634097942418493,
      "learning_rate": 9.611650485436892e-06,
      "loss": 3.692,
      "step": 100
    },
    {
      "epoch": 0.08550330353672755,
      "grad_norm": 3.9161281136802915,
      "learning_rate": 1.0582524271844662e-05,
      "loss": 3.3021,
      "step": 110
    },
    {
      "epoch": 0.09327633113097551,
      "grad_norm": 2.668188197386523,
      "learning_rate": 1.1553398058252427e-05,
      "loss": 3.0699,
      "step": 120
    },
    {
      "epoch": 0.10104935872522347,
      "grad_norm": 5.67542909747789,
      "learning_rate": 1.2524271844660197e-05,
      "loss": 2.9232,
      "step": 130
    },
    {
      "epoch": 0.10882238631947143,
      "grad_norm": 8.358891491079559,
      "learning_rate": 1.3495145631067962e-05,
      "loss": 2.8142,
      "step": 140
    },
    {
      "epoch": 0.11659541391371939,
      "grad_norm": 7.521330978238392,
      "learning_rate": 1.446601941747573e-05,
      "loss": 2.736,
      "step": 150
    },
    {
      "epoch": 0.12436844150796736,
      "grad_norm": 16.178722459983547,
      "learning_rate": 1.5436893203883496e-05,
      "loss": 2.6651,
      "step": 160
    },
    {
      "epoch": 0.1321414691022153,
      "grad_norm": 8.335965189770683,
      "learning_rate": 1.6407766990291263e-05,
      "loss": 2.6021,
      "step": 170
    },
    {
      "epoch": 0.13991449669646328,
      "grad_norm": 8.022474097021126,
      "learning_rate": 1.737864077669903e-05,
      "loss": 2.5458,
      "step": 180
    },
    {
      "epoch": 0.14768752429071122,
      "grad_norm": 11.647238033711355,
      "learning_rate": 1.8349514563106798e-05,
      "loss": 2.4889,
      "step": 190
    },
    {
      "epoch": 0.1554605518849592,
      "grad_norm": 11.910883080289608,
      "learning_rate": 1.9320388349514565e-05,
      "loss": 2.4326,
      "step": 200
    },
    {
      "epoch": 0.16323357947920716,
      "grad_norm": 11.410643498860912,
      "learning_rate": 2.0291262135922333e-05,
      "loss": 2.3697,
      "step": 210
    },
    {
      "epoch": 0.1710066070734551,
      "grad_norm": 11.108752154756692,
      "learning_rate": 2.1262135922330097e-05,
      "loss": 2.309,
      "step": 220
    },
    {
      "epoch": 0.17877963466770308,
      "grad_norm": 19.998718279627422,
      "learning_rate": 2.2233009708737864e-05,
      "loss": 2.241,
      "step": 230
    },
    {
      "epoch": 0.18655266226195102,
      "grad_norm": 11.758179967806427,
      "learning_rate": 2.3203883495145632e-05,
      "loss": 2.1862,
      "step": 240
    },
    {
      "epoch": 0.194325689856199,
      "grad_norm": 13.686852090688575,
      "learning_rate": 2.41747572815534e-05,
      "loss": 2.1354,
      "step": 250
    },
    {
      "epoch": 0.20209871745044694,
      "grad_norm": 7.097958211758118,
      "learning_rate": 2.5145631067961167e-05,
      "loss": 2.0252,
      "step": 260
    },
    {
      "epoch": 0.2098717450446949,
      "grad_norm": 12.553623894402433,
      "learning_rate": 2.6116504854368934e-05,
      "loss": 1.92,
      "step": 270
    },
    {
      "epoch": 0.21764477263894286,
      "grad_norm": 7.033016298250838,
      "learning_rate": 2.7087378640776702e-05,
      "loss": 1.832,
      "step": 280
    },
    {
      "epoch": 0.22541780023319083,
      "grad_norm": 8.05454776441266,
      "learning_rate": 2.8058252427184466e-05,
      "loss": 1.7505,
      "step": 290
    },
    {
      "epoch": 0.23319082782743877,
      "grad_norm": 6.32342513670527,
      "learning_rate": 2.9029126213592233e-05,
      "loss": 1.6596,
      "step": 300
    },
    {
      "epoch": 0.24096385542168675,
      "grad_norm": 6.0607588085936435,
      "learning_rate": 3e-05,
      "loss": 1.5836,
      "step": 310
    },
    {
      "epoch": 0.24873688301593472,
      "grad_norm": 7.205879956336063,
      "learning_rate": 3.097087378640777e-05,
      "loss": 1.5148,
      "step": 320
    },
    {
      "epoch": 0.2565099106101827,
      "grad_norm": 5.233158820350977,
      "learning_rate": 3.194174757281553e-05,
      "loss": 1.4655,
      "step": 330
    },
    {
      "epoch": 0.2642829382044306,
      "grad_norm": 6.5668167184777495,
      "learning_rate": 3.29126213592233e-05,
      "loss": 1.3996,
      "step": 340
    },
    {
      "epoch": 0.2720559657986786,
      "grad_norm": 5.866172867997477,
      "learning_rate": 3.3883495145631074e-05,
      "loss": 1.3546,
      "step": 350
    },
    {
      "epoch": 0.27982899339292655,
      "grad_norm": 7.078627326366601,
      "learning_rate": 3.485436893203884e-05,
      "loss": 1.2957,
      "step": 360
    },
    {
      "epoch": 0.2876020209871745,
      "grad_norm": 7.013906915714377,
      "learning_rate": 3.58252427184466e-05,
      "loss": 1.265,
      "step": 370
    },
    {
      "epoch": 0.29537504858142244,
      "grad_norm": 5.721220928268079,
      "learning_rate": 3.679611650485437e-05,
      "loss": 1.2163,
      "step": 380
    },
    {
      "epoch": 0.3031480761756704,
      "grad_norm": 5.571852622479908,
      "learning_rate": 3.776699029126214e-05,
      "loss": 1.163,
      "step": 390
    },
    {
      "epoch": 0.3109211037699184,
      "grad_norm": 5.4843565963743295,
      "learning_rate": 3.873786407766991e-05,
      "loss": 1.1277,
      "step": 400
    },
    {
      "epoch": 0.31869413136416636,
      "grad_norm": 7.069970365890535,
      "learning_rate": 3.970873786407767e-05,
      "loss": 1.1094,
      "step": 410
    },
    {
      "epoch": 0.3264671589584143,
      "grad_norm": 5.425993491783754,
      "learning_rate": 4.0679611650485436e-05,
      "loss": 1.069,
      "step": 420
    },
    {
      "epoch": 0.33424018655266224,
      "grad_norm": 5.03054433074538,
      "learning_rate": 4.165048543689321e-05,
      "loss": 1.0382,
      "step": 430
    },
    {
      "epoch": 0.3420132141469102,
      "grad_norm": 7.103802226586121,
      "learning_rate": 4.262135922330097e-05,
      "loss": 0.9988,
      "step": 440
    },
    {
      "epoch": 0.3497862417411582,
      "grad_norm": 5.180871741408765,
      "learning_rate": 4.3592233009708735e-05,
      "loss": 0.979,
      "step": 450
    },
    {
      "epoch": 0.35755926933540616,
      "grad_norm": 5.168730423055745,
      "learning_rate": 4.4563106796116506e-05,
      "loss": 0.9442,
      "step": 460
    },
    {
      "epoch": 0.3653322969296541,
      "grad_norm": 4.509772926215358,
      "learning_rate": 4.553398058252427e-05,
      "loss": 0.9304,
      "step": 470
    },
    {
      "epoch": 0.37310532452390205,
      "grad_norm": 5.497064414551351,
      "learning_rate": 4.650485436893204e-05,
      "loss": 0.8971,
      "step": 480
    },
    {
      "epoch": 0.38087835211815,
      "grad_norm": 4.2878368849547375,
      "learning_rate": 4.747572815533981e-05,
      "loss": 0.8754,
      "step": 490
    },
    {
      "epoch": 0.388651379712398,
      "grad_norm": 5.023778493195039,
      "learning_rate": 4.8446601941747576e-05,
      "loss": 0.8535,
      "step": 500
    },
    {
      "epoch": 0.39642440730664597,
      "grad_norm": 4.85429288760384,
      "learning_rate": 4.941747572815535e-05,
      "loss": 0.8176,
      "step": 510
    },
    {
      "epoch": 0.4041974349008939,
      "grad_norm": 4.78285229843812,
      "learning_rate": 5.0388349514563104e-05,
      "loss": 0.8114,
      "step": 520
    },
    {
      "epoch": 0.41197046249514185,
      "grad_norm": 5.101751542601933,
      "learning_rate": 5.135922330097087e-05,
      "loss": 0.7752,
      "step": 530
    },
    {
      "epoch": 0.4197434900893898,
      "grad_norm": 5.448054773232898,
      "learning_rate": 5.233009708737864e-05,
      "loss": 0.755,
      "step": 540
    },
    {
      "epoch": 0.4275165176836378,
      "grad_norm": 4.141934961389859,
      "learning_rate": 5.3300970873786417e-05,
      "loss": 0.7273,
      "step": 550
    },
    {
      "epoch": 0.4352895452778857,
      "grad_norm": 4.514745713617805,
      "learning_rate": 5.427184466019418e-05,
      "loss": 0.7143,
      "step": 560
    },
    {
      "epoch": 0.4430625728721337,
      "grad_norm": 3.852074915498448,
      "learning_rate": 5.524271844660195e-05,
      "loss": 0.6954,
      "step": 570
    },
    {
      "epoch": 0.45083560046638166,
      "grad_norm": 4.549795432019635,
      "learning_rate": 5.6213592233009716e-05,
      "loss": 0.6695,
      "step": 580
    },
    {
      "epoch": 0.45860862806062963,
      "grad_norm": 3.920314510813715,
      "learning_rate": 5.718446601941748e-05,
      "loss": 0.6503,
      "step": 590
    },
    {
      "epoch": 0.46638165565487755,
      "grad_norm": 5.518859711149864,
      "learning_rate": 5.815533980582525e-05,
      "loss": 0.6249,
      "step": 600
    },
    {
      "epoch": 0.4741546832491255,
      "grad_norm": 4.876638883083892,
      "learning_rate": 5.9126213592233015e-05,
      "loss": 0.6064,
      "step": 610
    },
    {
      "epoch": 0.4819277108433735,
      "grad_norm": 5.2941697252063955,
      "learning_rate": 6.009708737864078e-05,
      "loss": 0.5911,
      "step": 620
    },
    {
      "epoch": 0.48970073843762146,
      "grad_norm": 4.245229081646686,
      "learning_rate": 6.106796116504854e-05,
      "loss": 0.5704,
      "step": 630
    },
    {
      "epoch": 0.49747376603186944,
      "grad_norm": 4.127680025681991,
      "learning_rate": 6.203883495145631e-05,
      "loss": 0.5589,
      "step": 640
    },
    {
      "epoch": 0.5052467936261174,
      "grad_norm": 3.5437231540377616,
      "learning_rate": 6.300970873786408e-05,
      "loss": 0.5378,
      "step": 650
    },
    {
      "epoch": 0.5130198212203654,
      "grad_norm": 4.180991087743002,
      "learning_rate": 6.398058252427184e-05,
      "loss": 0.5056,
      "step": 660
    },
    {
      "epoch": 0.5207928488146133,
      "grad_norm": 3.7905222943864034,
      "learning_rate": 6.495145631067961e-05,
      "loss": 0.4944,
      "step": 670
    },
    {
      "epoch": 0.5285658764088612,
      "grad_norm": 2.829726998778342,
      "learning_rate": 6.592233009708738e-05,
      "loss": 0.4706,
      "step": 680
    },
    {
      "epoch": 0.5363389040031092,
      "grad_norm": 2.9270824536877456,
      "learning_rate": 6.689320388349515e-05,
      "loss": 0.4539,
      "step": 690
    },
    {
      "epoch": 0.5441119315973572,
      "grad_norm": 2.742603316305724,
      "learning_rate": 6.786407766990291e-05,
      "loss": 0.4376,
      "step": 700
    },
    {
      "epoch": 0.5518849591916052,
      "grad_norm": 2.896538805797215,
      "learning_rate": 6.883495145631068e-05,
      "loss": 0.4113,
      "step": 710
    },
    {
      "epoch": 0.5596579867858531,
      "grad_norm": 3.4815145675652013,
      "learning_rate": 6.980582524271845e-05,
      "loss": 0.4005,
      "step": 720
    },
    {
      "epoch": 0.567431014380101,
      "grad_norm": 2.270621987351258,
      "learning_rate": 7.077669902912621e-05,
      "loss": 0.3886,
      "step": 730
    },
    {
      "epoch": 0.575204041974349,
      "grad_norm": 2.7187166457028344,
      "learning_rate": 7.174757281553398e-05,
      "loss": 0.3765,
      "step": 740
    },
    {
      "epoch": 0.582977069568597,
      "grad_norm": 3.184495607766821,
      "learning_rate": 7.271844660194175e-05,
      "loss": 0.3514,
      "step": 750
    },
    {
      "epoch": 0.5907500971628449,
      "grad_norm": 2.379059028542611,
      "learning_rate": 7.368932038834951e-05,
      "loss": 0.341,
      "step": 760
    },
    {
      "epoch": 0.5985231247570929,
      "grad_norm": 2.4310101684060705,
      "learning_rate": 7.466019417475728e-05,
      "loss": 0.3263,
      "step": 770
    },
    {
      "epoch": 0.6062961523513408,
      "grad_norm": 2.5710567898956005,
      "learning_rate": 7.563106796116505e-05,
      "loss": 0.3117,
      "step": 780
    },
    {
      "epoch": 0.6140691799455888,
      "grad_norm": 2.416082850923419,
      "learning_rate": 7.660194174757281e-05,
      "loss": 0.2944,
      "step": 790
    },
    {
      "epoch": 0.6218422075398368,
      "grad_norm": 2.109447352556029,
      "learning_rate": 7.757281553398059e-05,
      "loss": 0.2832,
      "step": 800
    },
    {
      "epoch": 0.6296152351340847,
      "grad_norm": 2.2994104123752988,
      "learning_rate": 7.854368932038836e-05,
      "loss": 0.2726,
      "step": 810
    },
    {
      "epoch": 0.6373882627283327,
      "grad_norm": 2.9267817958414524,
      "learning_rate": 7.951456310679612e-05,
      "loss": 0.2606,
      "step": 820
    },
    {
      "epoch": 0.6451612903225806,
      "grad_norm": 2.1699995560860197,
      "learning_rate": 8.048543689320389e-05,
      "loss": 0.2471,
      "step": 830
    },
    {
      "epoch": 0.6529343179168287,
      "grad_norm": 1.680681697729957,
      "learning_rate": 8.145631067961166e-05,
      "loss": 0.2354,
      "step": 840
    },
    {
      "epoch": 0.6607073455110766,
      "grad_norm": 1.88393966323043,
      "learning_rate": 8.242718446601942e-05,
      "loss": 0.2263,
      "step": 850
    },
    {
      "epoch": 0.6684803731053245,
      "grad_norm": 2.7385999995385957,
      "learning_rate": 8.339805825242719e-05,
      "loss": 0.2109,
      "step": 860
    },
    {
      "epoch": 0.6762534006995725,
      "grad_norm": 1.71125390353767,
      "learning_rate": 8.436893203883496e-05,
      "loss": 0.2034,
      "step": 870
    },
    {
      "epoch": 0.6840264282938204,
      "grad_norm": 1.8856456915119455,
      "learning_rate": 8.533980582524272e-05,
      "loss": 0.1928,
      "step": 880
    },
    {
      "epoch": 0.6917994558880683,
      "grad_norm": 1.54640226468132,
      "learning_rate": 8.631067961165049e-05,
      "loss": 0.1797,
      "step": 890
    },
    {
      "epoch": 0.6995724834823164,
      "grad_norm": 1.3309710354478357,
      "learning_rate": 8.728155339805826e-05,
      "loss": 0.1712,
      "step": 900
    },
    {
      "epoch": 0.7073455110765643,
      "grad_norm": 1.2897601826602498,
      "learning_rate": 8.825242718446602e-05,
      "loss": 0.1624,
      "step": 910
    },
    {
      "epoch": 0.7151185386708123,
      "grad_norm": 1.1022252566548159,
      "learning_rate": 8.922330097087379e-05,
      "loss": 0.1509,
      "step": 920
    },
    {
      "epoch": 0.7228915662650602,
      "grad_norm": 1.4797407171136665,
      "learning_rate": 9.019417475728156e-05,
      "loss": 0.1495,
      "step": 930
    },
    {
      "epoch": 0.7306645938593082,
      "grad_norm": 1.4708443604696282,
      "learning_rate": 9.116504854368933e-05,
      "loss": 0.1407,
      "step": 940
    },
    {
      "epoch": 0.7384376214535562,
      "grad_norm": 1.2062811965627402,
      "learning_rate": 9.213592233009709e-05,
      "loss": 0.1409,
      "step": 950
    },
    {
      "epoch": 0.7462106490478041,
      "grad_norm": 0.991340571431122,
      "learning_rate": 9.310679611650486e-05,
      "loss": 0.1267,
      "step": 960
    },
    {
      "epoch": 0.7539836766420521,
      "grad_norm": 1.0799976482859923,
      "learning_rate": 9.407766990291263e-05,
      "loss": 0.1294,
      "step": 970
    },
    {
      "epoch": 0.7617567042363,
      "grad_norm": 1.0055612260520772,
      "learning_rate": 9.504854368932039e-05,
      "loss": 0.1167,
      "step": 980
    },
    {
      "epoch": 0.769529731830548,
      "grad_norm": 0.99809772380525,
      "learning_rate": 9.601941747572816e-05,
      "loss": 0.1104,
      "step": 990
    },
    {
      "epoch": 0.777302759424796,
      "grad_norm": 0.8341966391057081,
      "learning_rate": 9.699029126213593e-05,
      "loss": 0.104,
      "step": 1000
    },
    {
      "epoch": 0.7850757870190439,
      "grad_norm": 0.7440675821604489,
      "learning_rate": 9.796116504854369e-05,
      "loss": 0.1,
      "step": 1010
    },
    {
      "epoch": 0.7928488146132919,
      "grad_norm": 1.0089217362175902,
      "learning_rate": 9.893203883495146e-05,
      "loss": 0.0926,
      "step": 1020
    },
    {
      "epoch": 0.8006218422075398,
      "grad_norm": 0.669475479205569,
      "learning_rate": 9.990291262135923e-05,
      "loss": 0.0855,
      "step": 1030
    },
    {
      "epoch": 0.8083948698017878,
      "grad_norm": 0.7508700947786369,
      "learning_rate": 9.999976722311928e-05,
      "loss": 0.0824,
      "step": 1040
    },
    {
      "epoch": 0.8161678973960358,
      "grad_norm": 0.5833904781662174,
      "learning_rate": 9.999896256507963e-05,
      "loss": 0.0794,
      "step": 1050
    },
    {
      "epoch": 0.8239409249902837,
      "grad_norm": 0.6285661168460489,
      "learning_rate": 9.99975831613399e-05,
      "loss": 0.0737,
      "step": 1060
    },
    {
      "epoch": 0.8317139525845316,
      "grad_norm": 0.574291945207083,
      "learning_rate": 9.99956290277565e-05,
      "loss": 0.0689,
      "step": 1070
    },
    {
      "epoch": 0.8394869801787797,
      "grad_norm": 0.4876511188851532,
      "learning_rate": 9.999310018679253e-05,
      "loss": 0.0659,
      "step": 1080
    },
    {
      "epoch": 0.8472600077730276,
      "grad_norm": 0.553792085705227,
      "learning_rate": 9.998999666751738e-05,
      "loss": 0.064,
      "step": 1090
    },
    {
      "epoch": 0.8550330353672756,
      "grad_norm": 0.4302541179456994,
      "learning_rate": 9.998631850560647e-05,
      "loss": 0.0612,
      "step": 1100
    },
    {
      "epoch": 0.8628060629615235,
      "grad_norm": 0.47710435973221127,
      "learning_rate": 9.998206574334089e-05,
      "loss": 0.0572,
      "step": 1110
    },
    {
      "epoch": 0.8705790905557714,
      "grad_norm": 0.6437234961571133,
      "learning_rate": 9.997723842960671e-05,
      "loss": 0.0556,
      "step": 1120
    },
    {
      "epoch": 0.8783521181500195,
      "grad_norm": 0.44297654610119586,
      "learning_rate": 9.997183661989468e-05,
      "loss": 0.0551,
      "step": 1130
    },
    {
      "epoch": 0.8861251457442674,
      "grad_norm": 0.40789308058272417,
      "learning_rate": 9.996586037629944e-05,
      "loss": 0.052,
      "step": 1140
    },
    {
      "epoch": 0.8938981733385154,
      "grad_norm": 0.412683747214122,
      "learning_rate": 9.99593097675188e-05,
      "loss": 0.0503,
      "step": 1150
    },
    {
      "epoch": 0.9016712009327633,
      "grad_norm": 0.4042307583726218,
      "learning_rate": 9.995218486885302e-05,
      "loss": 0.0491,
      "step": 1160
    },
    {
      "epoch": 0.9094442285270112,
      "grad_norm": 0.3917357425568075,
      "learning_rate": 9.994448576220388e-05,
      "loss": 0.0472,
      "step": 1170
    },
    {
      "epoch": 0.9172172561212593,
      "grad_norm": 0.42847580686284464,
      "learning_rate": 9.99362125360738e-05,
      "loss": 0.0461,
      "step": 1180
    },
    {
      "epoch": 0.9249902837155072,
      "grad_norm": 0.36275912435050334,
      "learning_rate": 9.992736528556474e-05,
      "loss": 0.045,
      "step": 1190
    },
    {
      "epoch": 0.9327633113097551,
      "grad_norm": 0.29922320469821373,
      "learning_rate": 9.991794411237725e-05,
      "loss": 0.0429,
      "step": 1200
    },
    {
      "epoch": 0.9405363389040031,
      "grad_norm": 0.3914552553052167,
      "learning_rate": 9.990794912480913e-05,
      "loss": 0.0426,
      "step": 1210
    },
    {
      "epoch": 0.948309366498251,
      "grad_norm": 0.38509264588619113,
      "learning_rate": 9.989738043775428e-05,
      "loss": 0.0423,
      "step": 1220
    },
    {
      "epoch": 0.9560823940924991,
      "grad_norm": 0.3259823714473278,
      "learning_rate": 9.988623817270137e-05,
      "loss": 0.0406,
      "step": 1230
    },
    {
      "epoch": 0.963855421686747,
      "grad_norm": 0.31452752346688945,
      "learning_rate": 9.987452245773243e-05,
      "loss": 0.0398,
      "step": 1240
    },
    {
      "epoch": 0.9716284492809949,
      "grad_norm": 0.338353347661472,
      "learning_rate": 9.986223342752141e-05,
      "loss": 0.0386,
      "step": 1250
    },
    {
      "epoch": 0.9794014768752429,
      "grad_norm": 0.3002478510776732,
      "learning_rate": 9.984937122333256e-05,
      "loss": 0.0382,
      "step": 1260
    },
    {
      "epoch": 0.9871745044694908,
      "grad_norm": 0.40670865355319846,
      "learning_rate": 9.983593599301891e-05,
      "loss": 0.037,
      "step": 1270
    },
    {
      "epoch": 0.9949475320637389,
      "grad_norm": 0.6429283417769656,
      "learning_rate": 9.982192789102047e-05,
      "loss": 0.0373,
      "step": 1280
    },
    {
      "epoch": 1.0023319082782745,
      "grad_norm": 0.2801176127856459,
      "learning_rate": 9.980734707836248e-05,
      "loss": 0.0354,
      "step": 1290
    },
    {
      "epoch": 1.0101049358725223,
      "grad_norm": 0.3724656663427553,
      "learning_rate": 9.979219372265365e-05,
      "loss": 0.0344,
      "step": 1300
    },
    {
      "epoch": 1.0178779634667703,
      "grad_norm": 0.2573842211275922,
      "learning_rate": 9.97764679980841e-05,
      "loss": 0.0333,
      "step": 1310
    },
    {
      "epoch": 1.0256509910610183,
      "grad_norm": 0.24713329711260812,
      "learning_rate": 9.97601700854234e-05,
      "loss": 0.0336,
      "step": 1320
    },
    {
      "epoch": 1.0334240186552661,
      "grad_norm": 0.23958650404178872,
      "learning_rate": 9.974330017201859e-05,
      "loss": 0.0328,
      "step": 1330
    },
    {
      "epoch": 1.0411970462495141,
      "grad_norm": 0.22856953483452255,
      "learning_rate": 9.972585845179188e-05,
      "loss": 0.0316,
      "step": 1340
    },
    {
      "epoch": 1.0489700738437622,
      "grad_norm": 0.25194563553211635,
      "learning_rate": 9.970784512523848e-05,
      "loss": 0.0311,
      "step": 1350
    },
    {
      "epoch": 1.0567431014380102,
      "grad_norm": 0.23842502807724422,
      "learning_rate": 9.968926039942435e-05,
      "loss": 0.0306,
      "step": 1360
    },
    {
      "epoch": 1.064516129032258,
      "grad_norm": 0.2701262980707877,
      "learning_rate": 9.967010448798375e-05,
      "loss": 0.0303,
      "step": 1370
    },
    {
      "epoch": 1.072289156626506,
      "grad_norm": 0.25970806872919033,
      "learning_rate": 9.965037761111681e-05,
      "loss": 0.0295,
      "step": 1380
    },
    {
      "epoch": 1.080062184220754,
      "grad_norm": 0.24331306995976767,
      "learning_rate": 9.963007999558696e-05,
      "loss": 0.0291,
      "step": 1390
    },
    {
      "epoch": 1.0878352118150019,
      "grad_norm": 0.24100978412084842,
      "learning_rate": 9.96092118747184e-05,
      "loss": 0.0287,
      "step": 1400
    },
    {
      "epoch": 1.09560823940925,
      "grad_norm": 0.2184207676934967,
      "learning_rate": 9.958777348839336e-05,
      "loss": 0.0277,
      "step": 1410
    },
    {
      "epoch": 1.103381267003498,
      "grad_norm": 0.21996045383271537,
      "learning_rate": 9.956576508304938e-05,
      "loss": 0.0272,
      "step": 1420
    },
    {
      "epoch": 1.1111542945977457,
      "grad_norm": 0.21628224837569832,
      "learning_rate": 9.954318691167641e-05,
      "loss": 0.0273,
      "step": 1430
    },
    {
      "epoch": 1.1189273221919938,
      "grad_norm": 0.22298883167746006,
      "learning_rate": 9.952003923381399e-05,
      "loss": 0.0264,
      "step": 1440
    },
    {
      "epoch": 1.1267003497862418,
      "grad_norm": 0.2246150118171153,
      "learning_rate": 9.949632231554819e-05,
      "loss": 0.0265,
      "step": 1450
    },
    {
      "epoch": 1.1344733773804898,
      "grad_norm": 0.23186180017701422,
      "learning_rate": 9.947203642950862e-05,
      "loss": 0.0262,
      "step": 1460
    },
    {
      "epoch": 1.1422464049747376,
      "grad_norm": 0.24648849817769855,
      "learning_rate": 9.944718185486523e-05,
      "loss": 0.0256,
      "step": 1470
    },
    {
      "epoch": 1.1500194325689856,
      "grad_norm": 0.30413187225809407,
      "learning_rate": 9.942175887732517e-05,
      "loss": 0.025,
      "step": 1480
    },
    {
      "epoch": 1.1577924601632337,
      "grad_norm": 0.23138441255979614,
      "learning_rate": 9.93957677891294e-05,
      "loss": 0.0251,
      "step": 1490
    },
    {
      "epoch": 1.1655654877574815,
      "grad_norm": 0.24336635292704567,
      "learning_rate": 9.93692088890495e-05,
      "loss": 0.0244,
      "step": 1500
    },
    {
      "epoch": 1.1733385153517295,
      "grad_norm": 0.21185931343408212,
      "learning_rate": 9.934208248238403e-05,
      "loss": 0.0243,
      "step": 1510
    },
    {
      "epoch": 1.1811115429459775,
      "grad_norm": 0.19638440276738353,
      "learning_rate": 9.93143888809552e-05,
      "loss": 0.0239,
      "step": 1520
    },
    {
      "epoch": 1.1888845705402253,
      "grad_norm": 0.21756730427462398,
      "learning_rate": 9.928612840310514e-05,
      "loss": 0.0232,
      "step": 1530
    },
    {
      "epoch": 1.1966575981344734,
      "grad_norm": 0.21305245786411112,
      "learning_rate": 9.925730137369242e-05,
      "loss": 0.023,
      "step": 1540
    },
    {
      "epoch": 1.2044306257287214,
      "grad_norm": 0.22828714542453502,
      "learning_rate": 9.922790812408808e-05,
      "loss": 0.0226,
      "step": 1550
    },
    {
      "epoch": 1.2122036533229692,
      "grad_norm": 0.19930203197693303,
      "learning_rate": 9.919794899217203e-05,
      "loss": 0.0224,
      "step": 1560
    },
    {
      "epoch": 1.2199766809172172,
      "grad_norm": 0.20433748074735467,
      "learning_rate": 9.916742432232909e-05,
      "loss": 0.0222,
      "step": 1570
    },
    {
      "epoch": 1.2277497085114653,
      "grad_norm": 0.20184151603700423,
      "learning_rate": 9.913633446544493e-05,
      "loss": 0.0215,
      "step": 1580
    },
    {
      "epoch": 1.2355227361057133,
      "grad_norm": 0.2081529716929725,
      "learning_rate": 9.910467977890225e-05,
      "loss": 0.0216,
      "step": 1590
    },
    {
      "epoch": 1.243295763699961,
      "grad_norm": 0.27882658958977674,
      "learning_rate": 9.907246062657648e-05,
      "loss": 0.0234,
      "step": 1600
    },
    {
      "epoch": 1.2510687912942091,
      "grad_norm": 0.30986570505669037,
      "learning_rate": 9.903967737883168e-05,
      "loss": 0.0248,
      "step": 1610
    },
    {
      "epoch": 1.258841818888457,
      "grad_norm": 0.2205524563203657,
      "learning_rate": 9.900633041251631e-05,
      "loss": 0.0222,
      "step": 1620
    },
    {
      "epoch": 1.266614846482705,
      "grad_norm": 0.19645809273796086,
      "learning_rate": 9.897242011095883e-05,
      "loss": 0.0208,
      "step": 1630
    },
    {
      "epoch": 1.274387874076953,
      "grad_norm": 0.20220041423279006,
      "learning_rate": 9.893794686396329e-05,
      "loss": 0.0203,
      "step": 1640
    },
    {
      "epoch": 1.282160901671201,
      "grad_norm": 0.19701852522226898,
      "learning_rate": 9.890291106780495e-05,
      "loss": 0.0199,
      "step": 1650
    },
    {
      "epoch": 1.289933929265449,
      "grad_norm": 0.19055264078653414,
      "learning_rate": 9.886731312522563e-05,
      "loss": 0.0196,
      "step": 1660
    },
    {
      "epoch": 1.2977069568596968,
      "grad_norm": 0.17498611562411595,
      "learning_rate": 9.883115344542911e-05,
      "loss": 0.0189,
      "step": 1670
    },
    {
      "epoch": 1.3054799844539449,
      "grad_norm": 0.18664040460319856,
      "learning_rate": 9.879443244407643e-05,
      "loss": 0.019,
      "step": 1680
    },
    {
      "epoch": 1.3132530120481927,
      "grad_norm": 0.186899833087349,
      "learning_rate": 9.875715054328108e-05,
      "loss": 0.0186,
      "step": 1690
    },
    {
      "epoch": 1.3210260396424407,
      "grad_norm": 0.18960588899249967,
      "learning_rate": 9.871930817160422e-05,
      "loss": 0.018,
      "step": 1700
    },
    {
      "epoch": 1.3287990672366887,
      "grad_norm": 0.2246998119833756,
      "learning_rate": 9.868090576404966e-05,
      "loss": 0.0185,
      "step": 1710
    },
    {
      "epoch": 1.3365720948309368,
      "grad_norm": 0.18612335330028418,
      "learning_rate": 9.864194376205895e-05,
      "loss": 0.0181,
      "step": 1720
    },
    {
      "epoch": 1.3443451224251846,
      "grad_norm": 0.19504144084562025,
      "learning_rate": 9.860242261350623e-05,
      "loss": 0.0174,
      "step": 1730
    },
    {
      "epoch": 1.3521181500194326,
      "grad_norm": 0.1940460381388096,
      "learning_rate": 9.856234277269313e-05,
      "loss": 0.0173,
      "step": 1740
    },
    {
      "epoch": 1.3598911776136804,
      "grad_norm": 0.1882305169943444,
      "learning_rate": 9.852170470034356e-05,
      "loss": 0.0173,
      "step": 1750
    },
    {
      "epoch": 1.3676642052079284,
      "grad_norm": 0.1844609631309131,
      "learning_rate": 9.848050886359835e-05,
      "loss": 0.0171,
      "step": 1760
    },
    {
      "epoch": 1.3754372328021764,
      "grad_norm": 0.16755276537647784,
      "learning_rate": 9.843875573600989e-05,
      "loss": 0.0169,
      "step": 1770
    },
    {
      "epoch": 1.3832102603964245,
      "grad_norm": 0.16696100318285242,
      "learning_rate": 9.839644579753677e-05,
      "loss": 0.0164,
      "step": 1780
    },
    {
      "epoch": 1.3909832879906725,
      "grad_norm": 0.1818918991872044,
      "learning_rate": 9.835357953453821e-05,
      "loss": 0.0163,
      "step": 1790
    },
    {
      "epoch": 1.3987563155849203,
      "grad_norm": 0.18963889299339048,
      "learning_rate": 9.831015743976843e-05,
      "loss": 0.0158,
      "step": 1800
    },
    {
      "epoch": 1.4065293431791683,
      "grad_norm": 0.1634818998306774,
      "learning_rate": 9.8266180012371e-05,
      "loss": 0.0154,
      "step": 1810
    },
    {
      "epoch": 1.4143023707734161,
      "grad_norm": 0.197090617938791,
      "learning_rate": 9.82216477578732e-05,
      "loss": 0.0152,
      "step": 1820
    },
    {
      "epoch": 1.4220753983676642,
      "grad_norm": 0.15850580732892314,
      "learning_rate": 9.817656118818006e-05,
      "loss": 0.0148,
      "step": 1830
    },
    {
      "epoch": 1.4298484259619122,
      "grad_norm": 0.15186976089409565,
      "learning_rate": 9.813092082156858e-05,
      "loss": 0.0146,
      "step": 1840
    },
    {
      "epoch": 1.4376214535561602,
      "grad_norm": 0.17352561780495993,
      "learning_rate": 9.808472718268174e-05,
      "loss": 0.0146,
      "step": 1850
    },
    {
      "epoch": 1.445394481150408,
      "grad_norm": 0.16575574275890378,
      "learning_rate": 9.803798080252245e-05,
      "loss": 0.014,
      "step": 1860
    },
    {
      "epoch": 1.453167508744656,
      "grad_norm": 0.16700742476995556,
      "learning_rate": 9.79906822184475e-05,
      "loss": 0.0139,
      "step": 1870
    },
    {
      "epoch": 1.460940536338904,
      "grad_norm": 0.15966930327908196,
      "learning_rate": 9.79428319741613e-05,
      "loss": 0.0136,
      "step": 1880
    },
    {
      "epoch": 1.4687135639331519,
      "grad_norm": 0.16653090767039122,
      "learning_rate": 9.789443061970973e-05,
      "loss": 0.0134,
      "step": 1890
    },
    {
      "epoch": 1.4764865915274,
      "grad_norm": 0.190626948029195,
      "learning_rate": 9.784547871147374e-05,
      "loss": 0.0129,
      "step": 1900
    },
    {
      "epoch": 1.484259619121648,
      "grad_norm": 0.1763435524084487,
      "learning_rate": 9.779597681216297e-05,
      "loss": 0.0127,
      "step": 1910
    },
    {
      "epoch": 1.492032646715896,
      "grad_norm": 0.1672345466206139,
      "learning_rate": 9.774592549080932e-05,
      "loss": 0.0126,
      "step": 1920
    },
    {
      "epoch": 1.4998056743101438,
      "grad_norm": 0.15678075113388967,
      "learning_rate": 9.769532532276034e-05,
      "loss": 0.0123,
      "step": 1930
    },
    {
      "epoch": 1.5075787019043918,
      "grad_norm": 0.14761977546971236,
      "learning_rate": 9.764417688967268e-05,
      "loss": 0.0118,
      "step": 1940
    },
    {
      "epoch": 1.5153517294986396,
      "grad_norm": 0.17063805371316892,
      "learning_rate": 9.75924807795054e-05,
      "loss": 0.0118,
      "step": 1950
    },
    {
      "epoch": 1.5231247570928876,
      "grad_norm": 0.14571436924477155,
      "learning_rate": 9.754023758651314e-05,
      "loss": 0.0115,
      "step": 1960
    },
    {
      "epoch": 1.5308977846871357,
      "grad_norm": 0.15118080081237256,
      "learning_rate": 9.74874479112394e-05,
      "loss": 0.0112,
      "step": 1970
    },
    {
      "epoch": 1.5386708122813837,
      "grad_norm": 0.1524883133010242,
      "learning_rate": 9.743411236050953e-05,
      "loss": 0.0112,
      "step": 1980
    },
    {
      "epoch": 1.5464438398756317,
      "grad_norm": 0.15389453717185902,
      "learning_rate": 9.738023154742383e-05,
      "loss": 0.0114,
      "step": 1990
    },
    {
      "epoch": 1.5542168674698795,
      "grad_norm": 0.14981753056593858,
      "learning_rate": 9.732580609135047e-05,
      "loss": 0.0108,
      "step": 2000
    },
    {
      "epoch": 1.5619898950641273,
      "grad_norm": 1.9198802022850896,
      "learning_rate": 9.727083661791836e-05,
      "loss": 0.0165,
      "step": 2010
    },
    {
      "epoch": 1.5697629226583754,
      "grad_norm": 0.26855012872950995,
      "learning_rate": 9.721532375900997e-05,
      "loss": 0.0144,
      "step": 2020
    },
    {
      "epoch": 1.5775359502526234,
      "grad_norm": 0.2071283206940206,
      "learning_rate": 9.71592681527541e-05,
      "loss": 0.0117,
      "step": 2030
    },
    {
      "epoch": 1.5853089778468714,
      "grad_norm": 0.14770534677344455,
      "learning_rate": 9.710267044351846e-05,
      "loss": 0.0109,
      "step": 2040
    },
    {
      "epoch": 1.5930820054411194,
      "grad_norm": 0.14368540945251085,
      "learning_rate": 9.704553128190239e-05,
      "loss": 0.0104,
      "step": 2050
    },
    {
      "epoch": 1.6008550330353672,
      "grad_norm": 0.13894193794675433,
      "learning_rate": 9.698785132472926e-05,
      "loss": 0.01,
      "step": 2060
    },
    {
      "epoch": 1.6086280606296153,
      "grad_norm": 0.13441313900735385,
      "learning_rate": 9.692963123503894e-05,
      "loss": 0.0094,
      "step": 2070
    },
    {
      "epoch": 1.616401088223863,
      "grad_norm": 0.12841760115701012,
      "learning_rate": 9.687087168208026e-05,
      "loss": 0.0094,
      "step": 2080
    },
    {
      "epoch": 1.624174115818111,
      "grad_norm": 0.1378414358921271,
      "learning_rate": 9.681157334130326e-05,
      "loss": 0.0092,
      "step": 2090
    },
    {
      "epoch": 1.6319471434123591,
      "grad_norm": 0.13875742382402598,
      "learning_rate": 9.67517368943514e-05,
      "loss": 0.0093,
      "step": 2100
    },
    {
      "epoch": 1.6397201710066072,
      "grad_norm": 0.12150978897988295,
      "learning_rate": 9.669136302905374e-05,
      "loss": 0.0094,
      "step": 2110
    },
    {
      "epoch": 1.6474931986008552,
      "grad_norm": 0.13386154431399386,
      "learning_rate": 9.663045243941711e-05,
      "loss": 0.009,
      "step": 2120
    },
    {
      "epoch": 1.655266226195103,
      "grad_norm": 0.14353227081750533,
      "learning_rate": 9.656900582561801e-05,
      "loss": 0.009,
      "step": 2130
    },
    {
      "epoch": 1.6630392537893508,
      "grad_norm": 0.13602583881387148,
      "learning_rate": 9.650702389399461e-05,
      "loss": 0.0088,
      "step": 2140
    },
    {
      "epoch": 1.6708122813835988,
      "grad_norm": 0.13201111610838254,
      "learning_rate": 9.644450735703872e-05,
      "loss": 0.0087,
      "step": 2150
    },
    {
      "epoch": 1.6785853089778469,
      "grad_norm": 0.11982302853617144,
      "learning_rate": 9.638145693338743e-05,
      "loss": 0.0087,
      "step": 2160
    },
    {
      "epoch": 1.6863583365720949,
      "grad_norm": 0.11312867242927008,
      "learning_rate": 9.631787334781497e-05,
      "loss": 0.0087,
      "step": 2170
    },
    {
      "epoch": 1.694131364166343,
      "grad_norm": 0.11767848177751597,
      "learning_rate": 9.625375733122439e-05,
      "loss": 0.0085,
      "step": 2180
    },
    {
      "epoch": 1.7019043917605907,
      "grad_norm": 0.11041145208342112,
      "learning_rate": 9.618910962063903e-05,
      "loss": 0.0082,
      "step": 2190
    },
    {
      "epoch": 1.7096774193548387,
      "grad_norm": 0.10799469734884444,
      "learning_rate": 9.612393095919423e-05,
      "loss": 0.0078,
      "step": 2200
    },
    {
      "epoch": 1.7174504469490866,
      "grad_norm": 0.11477500796146126,
      "learning_rate": 9.605822209612862e-05,
      "loss": 0.0078,
      "step": 2210
    },
    {
      "epoch": 1.7252234745433346,
      "grad_norm": 0.17423934102739813,
      "learning_rate": 9.599198378677559e-05,
      "loss": 0.0079,
      "step": 2220
    },
    {
      "epoch": 1.7329965021375826,
      "grad_norm": 0.11632697553641431,
      "learning_rate": 9.592521679255464e-05,
      "loss": 0.008,
      "step": 2230
    },
    {
      "epoch": 1.7407695297318306,
      "grad_norm": 0.12288031599865465,
      "learning_rate": 9.585792188096252e-05,
      "loss": 0.008,
      "step": 2240
    },
    {
      "epoch": 1.7485425573260787,
      "grad_norm": 0.15984027115213267,
      "learning_rate": 9.579009982556452e-05,
      "loss": 0.0079,
      "step": 2250
    },
    {
      "epoch": 1.7563155849203265,
      "grad_norm": 0.11306629622107851,
      "learning_rate": 9.572175140598553e-05,
      "loss": 0.0076,
      "step": 2260
    },
    {
      "epoch": 1.7640886125145743,
      "grad_norm": 0.1278172766126942,
      "learning_rate": 9.565287740790101e-05,
      "loss": 0.0077,
      "step": 2270
    },
    {
      "epoch": 1.7718616401088223,
      "grad_norm": 0.1251067207191811,
      "learning_rate": 9.55834786230281e-05,
      "loss": 0.0074,
      "step": 2280
    },
    {
      "epoch": 1.7796346677030703,
      "grad_norm": 0.12072123910224757,
      "learning_rate": 9.551355584911641e-05,
      "loss": 0.0077,
      "step": 2290
    },
    {
      "epoch": 1.7874076952973184,
      "grad_norm": 0.13420830427564445,
      "learning_rate": 9.544310988993889e-05,
      "loss": 0.0076,
      "step": 2300
    },
    {
      "epoch": 1.7951807228915664,
      "grad_norm": 0.11275927835211168,
      "learning_rate": 9.53721415552826e-05,
      "loss": 0.0072,
      "step": 2310
    },
    {
      "epoch": 1.8029537504858142,
      "grad_norm": 0.11663410063710995,
      "learning_rate": 9.530065166093936e-05,
      "loss": 0.007,
      "step": 2320
    },
    {
      "epoch": 1.8107267780800622,
      "grad_norm": 0.11515297587501505,
      "learning_rate": 9.52286410286964e-05,
      "loss": 0.007,
      "step": 2330
    },
    {
      "epoch": 1.81849980567431,
      "grad_norm": 0.12034539299056037,
      "learning_rate": 9.515611048632692e-05,
      "loss": 0.0071,
      "step": 2340
    },
    {
      "epoch": 1.826272833268558,
      "grad_norm": 0.13525219378529196,
      "learning_rate": 9.508306086758059e-05,
      "loss": 0.0069,
      "step": 2350
    },
    {
      "epoch": 1.834045860862806,
      "grad_norm": 0.11355987672855966,
      "learning_rate": 9.500949301217388e-05,
      "loss": 0.0066,
      "step": 2360
    },
    {
      "epoch": 1.841818888457054,
      "grad_norm": 0.10553177998244642,
      "learning_rate": 9.493540776578053e-05,
      "loss": 0.0067,
      "step": 2370
    },
    {
      "epoch": 1.8495919160513021,
      "grad_norm": 0.10733467006153838,
      "learning_rate": 9.486080598002176e-05,
      "loss": 0.0066,
      "step": 2380
    },
    {
      "epoch": 1.85736494364555,
      "grad_norm": 0.11088682968161348,
      "learning_rate": 9.478568851245643e-05,
      "loss": 0.0064,
      "step": 2390
    },
    {
      "epoch": 1.8651379712397977,
      "grad_norm": 0.11788349609407499,
      "learning_rate": 9.471005622657134e-05,
      "loss": 0.0064,
      "step": 2400
    },
    {
      "epoch": 1.8729109988340458,
      "grad_norm": 0.1382091673100836,
      "learning_rate": 9.463390999177113e-05,
      "loss": 0.0066,
      "step": 2410
    },
    {
      "epoch": 1.8806840264282938,
      "grad_norm": 0.11080598206299531,
      "learning_rate": 9.455725068336837e-05,
      "loss": 0.0067,
      "step": 2420
    },
    {
      "epoch": 1.8884570540225418,
      "grad_norm": 0.11345029521361591,
      "learning_rate": 9.44800791825735e-05,
      "loss": 0.0063,
      "step": 2430
    },
    {
      "epoch": 1.8962300816167899,
      "grad_norm": 0.12000127996735861,
      "learning_rate": 9.440239637648473e-05,
      "loss": 0.0064,
      "step": 2440
    },
    {
      "epoch": 1.9040031092110377,
      "grad_norm": 0.11742977407222017,
      "learning_rate": 9.432420315807771e-05,
      "loss": 0.0063,
      "step": 2450
    },
    {
      "epoch": 1.9117761368052857,
      "grad_norm": 0.10552545531273427,
      "learning_rate": 9.424550042619544e-05,
      "loss": 0.0062,
      "step": 2460
    },
    {
      "epoch": 1.9195491643995335,
      "grad_norm": 0.0996572096889176,
      "learning_rate": 9.416628908553784e-05,
      "loss": 0.0059,
      "step": 2470
    },
    {
      "epoch": 1.9273221919937815,
      "grad_norm": 0.09626416174818198,
      "learning_rate": 9.408657004665131e-05,
      "loss": 0.0058,
      "step": 2480
    },
    {
      "epoch": 1.9350952195880295,
      "grad_norm": 0.09783061431201936,
      "learning_rate": 9.400634422591838e-05,
      "loss": 0.0058,
      "step": 2490
    },
    {
      "epoch": 1.9428682471822776,
      "grad_norm": 0.11102429789832584,
      "learning_rate": 9.392561254554713e-05,
      "loss": 0.0059,
      "step": 2500
    },
    {
      "epoch": 1.9506412747765256,
      "grad_norm": 0.10011364069857738,
      "learning_rate": 9.384437593356048e-05,
      "loss": 0.0059,
      "step": 2510
    },
    {
      "epoch": 1.9584143023707734,
      "grad_norm": 0.10584225350181801,
      "learning_rate": 9.37626353237857e-05,
      "loss": 0.0061,
      "step": 2520
    },
    {
      "epoch": 1.9661873299650212,
      "grad_norm": 0.10444281925647403,
      "learning_rate": 9.368039165584359e-05,
      "loss": 0.0061,
      "step": 2530
    },
    {
      "epoch": 1.9739603575592692,
      "grad_norm": 0.10604191503186843,
      "learning_rate": 9.359764587513761e-05,
      "loss": 0.0057,
      "step": 2540
    },
    {
      "epoch": 1.9817333851535173,
      "grad_norm": 0.09211496249761421,
      "learning_rate": 9.351439893284317e-05,
      "loss": 0.0056,
      "step": 2550
    },
    {
      "epoch": 1.9895064127477653,
      "grad_norm": 0.14501544032188865,
      "learning_rate": 9.343065178589654e-05,
      "loss": 0.0056,
      "step": 2560
    },
    {
      "epoch": 1.9972794403420133,
      "grad_norm": 0.11456289818413888,
      "learning_rate": 9.334640539698396e-05,
      "loss": 0.0058,
      "step": 2570
    },
    {
      "epoch": 2.004663816556549,
      "grad_norm": 0.08068921187905599,
      "learning_rate": 9.326166073453051e-05,
      "loss": 0.0052,
      "step": 2580
    },
    {
      "epoch": 2.012436844150797,
      "grad_norm": 0.10032295518621465,
      "learning_rate": 9.317641877268901e-05,
      "loss": 0.0045,
      "step": 2590
    },
    {
      "epoch": 2.0202098717450445,
      "grad_norm": 0.0906748520799375,
      "learning_rate": 9.309068049132882e-05,
      "loss": 0.0044,
      "step": 2600
    },
    {
      "epoch": 2.0279828993392925,
      "grad_norm": 0.07982904566291607,
      "learning_rate": 9.300444687602457e-05,
      "loss": 0.0041,
      "step": 2610
    },
    {
      "epoch": 2.0357559269335406,
      "grad_norm": 0.1253437061299033,
      "learning_rate": 9.291771891804478e-05,
      "loss": 0.004,
      "step": 2620
    },
    {
      "epoch": 2.0435289545277886,
      "grad_norm": 0.09180533228491213,
      "learning_rate": 9.283049761434059e-05,
      "loss": 0.0041,
      "step": 2630
    },
    {
      "epoch": 2.0513019821220366,
      "grad_norm": 0.09238927797437002,
      "learning_rate": 9.27427839675342e-05,
      "loss": 0.004,
      "step": 2640
    },
    {
      "epoch": 2.0590750097162847,
      "grad_norm": 0.08085544322712232,
      "learning_rate": 9.26545789859073e-05,
      "loss": 0.004,
      "step": 2650
    },
    {
      "epoch": 2.0668480373105322,
      "grad_norm": 0.08278933444675111,
      "learning_rate": 9.256588368338968e-05,
      "loss": 0.0039,
      "step": 2660
    },
    {
      "epoch": 2.0746210649047803,
      "grad_norm": 0.0821491544443607,
      "learning_rate": 9.24766990795473e-05,
      "loss": 0.004,
      "step": 2670
    },
    {
      "epoch": 2.0823940924990283,
      "grad_norm": 0.0820536970017854,
      "learning_rate": 9.238702619957083e-05,
      "loss": 0.0039,
      "step": 2680
    },
    {
      "epoch": 2.0901671200932763,
      "grad_norm": 0.07783393489314461,
      "learning_rate": 9.229686607426368e-05,
      "loss": 0.0038,
      "step": 2690
    },
    {
      "epoch": 2.0979401476875243,
      "grad_norm": 0.08972295811517289,
      "learning_rate": 9.220621974003025e-05,
      "loss": 0.0038,
      "step": 2700
    },
    {
      "epoch": 2.1057131752817724,
      "grad_norm": 0.10850037565865495,
      "learning_rate": 9.211508823886398e-05,
      "loss": 0.0038,
      "step": 2710
    },
    {
      "epoch": 2.1134862028760204,
      "grad_norm": 0.1165050619921317,
      "learning_rate": 9.202347261833536e-05,
      "loss": 0.004,
      "step": 2720
    },
    {
      "epoch": 2.121259230470268,
      "grad_norm": 0.09613835587740172,
      "learning_rate": 9.193137393157994e-05,
      "loss": 0.004,
      "step": 2730
    },
    {
      "epoch": 2.129032258064516,
      "grad_norm": 0.08222637851192974,
      "learning_rate": 9.183879323728613e-05,
      "loss": 0.0041,
      "step": 2740
    },
    {
      "epoch": 2.136805285658764,
      "grad_norm": 0.09364974494864589,
      "learning_rate": 9.174573159968319e-05,
      "loss": 0.0041,
      "step": 2750
    },
    {
      "epoch": 2.144578313253012,
      "grad_norm": 0.09079410215789208,
      "learning_rate": 9.165219008852879e-05,
      "loss": 0.0037,
      "step": 2760
    },
    {
      "epoch": 2.15235134084726,
      "grad_norm": 0.08693641953192356,
      "learning_rate": 9.155816977909688e-05,
      "loss": 0.0038,
      "step": 2770
    },
    {
      "epoch": 2.160124368441508,
      "grad_norm": 0.08407058222727168,
      "learning_rate": 9.146367175216527e-05,
      "loss": 0.0038,
      "step": 2780
    },
    {
      "epoch": 2.167897396035756,
      "grad_norm": 0.08864727152361798,
      "learning_rate": 9.136869709400318e-05,
      "loss": 0.0037,
      "step": 2790
    },
    {
      "epoch": 2.1756704236300037,
      "grad_norm": 0.08585633309380411,
      "learning_rate": 9.127324689635879e-05,
      "loss": 0.0038,
      "step": 2800
    },
    {
      "epoch": 2.1834434512242518,
      "grad_norm": 0.09079038707160886,
      "learning_rate": 9.117732225644668e-05,
      "loss": 0.0037,
      "step": 2810
    },
    {
      "epoch": 2.1912164788185,
      "grad_norm": 0.08945120074693964,
      "learning_rate": 9.108092427693521e-05,
      "loss": 0.0039,
      "step": 2820
    },
    {
      "epoch": 2.198989506412748,
      "grad_norm": 0.09181352080010166,
      "learning_rate": 9.098405406593387e-05,
      "loss": 0.0038,
      "step": 2830
    },
    {
      "epoch": 2.206762534006996,
      "grad_norm": 0.08585652120551285,
      "learning_rate": 9.088671273698049e-05,
      "loss": 0.0038,
      "step": 2840
    },
    {
      "epoch": 2.214535561601244,
      "grad_norm": 0.08716662155761687,
      "learning_rate": 9.078890140902851e-05,
      "loss": 0.0037,
      "step": 2850
    },
    {
      "epoch": 2.2223085891954915,
      "grad_norm": 0.08274857092352111,
      "learning_rate": 9.069062120643406e-05,
      "loss": 0.0036,
      "step": 2860
    },
    {
      "epoch": 2.2300816167897395,
      "grad_norm": 0.08054936545881469,
      "learning_rate": 9.059187325894305e-05,
      "loss": 0.0037,
      "step": 2870
    },
    {
      "epoch": 2.2378546443839875,
      "grad_norm": 0.08928250602229335,
      "learning_rate": 9.04926587016782e-05,
      "loss": 0.0037,
      "step": 2880
    },
    {
      "epoch": 2.2456276719782355,
      "grad_norm": 0.14580475649103322,
      "learning_rate": 9.039297867512598e-05,
      "loss": 0.0039,
      "step": 2890
    },
    {
      "epoch": 2.2534006995724836,
      "grad_norm": 0.09684044113116728,
      "learning_rate": 9.02928343251235e-05,
      "loss": 0.0036,
      "step": 2900
    },
    {
      "epoch": 2.2611737271667316,
      "grad_norm": 4.610227312049106,
      "learning_rate": 9.019222680284531e-05,
      "loss": 0.0177,
      "step": 2910
    },
    {
      "epoch": 2.2689467547609796,
      "grad_norm": 3.9441098766180716,
      "learning_rate": 9.009115726479028e-05,
      "loss": 0.0845,
      "step": 2920
    },
    {
      "epoch": 2.276719782355227,
      "grad_norm": 0.30362597840942473,
      "learning_rate": 8.998962687276809e-05,
      "loss": 0.0336,
      "step": 2930
    },
    {
      "epoch": 2.2844928099494752,
      "grad_norm": 0.17872746957322608,
      "learning_rate": 8.988763679388614e-05,
      "loss": 0.0106,
      "step": 2940
    },
    {
      "epoch": 2.2922658375437233,
      "grad_norm": 0.11891620122269164,
      "learning_rate": 8.97851882005359e-05,
      "loss": 0.0069,
      "step": 2950
    },
    {
      "epoch": 2.3000388651379713,
      "grad_norm": 0.11931811737114671,
      "learning_rate": 8.968228227037961e-05,
      "loss": 0.0054,
      "step": 2960
    },
    {
      "epoch": 2.3078118927322193,
      "grad_norm": 0.11297205800033525,
      "learning_rate": 8.957892018633662e-05,
      "loss": 0.005,
      "step": 2970
    },
    {
      "epoch": 2.3155849203264673,
      "grad_norm": 0.08607374863983304,
      "learning_rate": 8.947510313656983e-05,
      "loss": 0.0046,
      "step": 2980
    },
    {
      "epoch": 2.323357947920715,
      "grad_norm": 0.08646791211129752,
      "learning_rate": 8.93708323144721e-05,
      "loss": 0.0043,
      "step": 2990
    },
    {
      "epoch": 2.331130975514963,
      "grad_norm": 0.09633865197527709,
      "learning_rate": 8.926610891865239e-05,
      "loss": 0.0042,
      "step": 3000
    },
    {
      "epoch": 2.338904003109211,
      "grad_norm": 0.07797950909850374,
      "learning_rate": 8.916093415292212e-05,
      "loss": 0.004,
      "step": 3010
    },
    {
      "epoch": 2.346677030703459,
      "grad_norm": 0.08803265774925108,
      "learning_rate": 8.905530922628129e-05,
      "loss": 0.0039,
      "step": 3020
    },
    {
      "epoch": 2.354450058297707,
      "grad_norm": 0.08578035815680632,
      "learning_rate": 8.894923535290452e-05,
      "loss": 0.0038,
      "step": 3030
    },
    {
      "epoch": 2.362223085891955,
      "grad_norm": 0.0723027097191447,
      "learning_rate": 8.884271375212713e-05,
      "loss": 0.0037,
      "step": 3040
    },
    {
      "epoch": 2.369996113486203,
      "grad_norm": 0.07933890872564231,
      "learning_rate": 8.873574564843123e-05,
      "loss": 0.0037,
      "step": 3050
    },
    {
      "epoch": 2.3777691410804507,
      "grad_norm": 0.08309597358808056,
      "learning_rate": 8.862833227143143e-05,
      "loss": 0.0037,
      "step": 3060
    },
    {
      "epoch": 2.3855421686746987,
      "grad_norm": 0.0796136928178535,
      "learning_rate": 8.852047485586093e-05,
      "loss": 0.0034,
      "step": 3070
    },
    {
      "epoch": 2.3933151962689467,
      "grad_norm": 0.0906501106792225,
      "learning_rate": 8.841217464155714e-05,
      "loss": 0.0036,
      "step": 3080
    },
    {
      "epoch": 2.4010882238631948,
      "grad_norm": 0.09232238343629308,
      "learning_rate": 8.830343287344753e-05,
      "loss": 0.0036,
      "step": 3090
    },
    {
      "epoch": 2.408861251457443,
      "grad_norm": 0.07677160549185906,
      "learning_rate": 8.819425080153533e-05,
      "loss": 0.0036,
      "step": 3100
    },
    {
      "epoch": 2.416634279051691,
      "grad_norm": 0.07965459624459517,
      "learning_rate": 8.808462968088512e-05,
      "loss": 0.0036,
      "step": 3110
    },
    {
      "epoch": 2.4244073066459384,
      "grad_norm": 0.08152518289735053,
      "learning_rate": 8.797457077160836e-05,
      "loss": 0.0036,
      "step": 3120
    },
    {
      "epoch": 2.4321803342401864,
      "grad_norm": 0.07840524642090889,
      "learning_rate": 8.786407533884902e-05,
      "loss": 0.0035,
      "step": 3130
    },
    {
      "epoch": 2.4399533618344345,
      "grad_norm": 0.0753906754915079,
      "learning_rate": 8.775314465276896e-05,
      "loss": 0.0034,
      "step": 3140
    },
    {
      "epoch": 2.4477263894286825,
      "grad_norm": 0.07553857495501508,
      "learning_rate": 8.764177998853334e-05,
      "loss": 0.0034,
      "step": 3150
    },
    {
      "epoch": 2.4554994170229305,
      "grad_norm": 0.07989895676116511,
      "learning_rate": 8.752998262629593e-05,
      "loss": 0.0034,
      "step": 3160
    },
    {
      "epoch": 2.4632724446171785,
      "grad_norm": 0.08604538052038023,
      "learning_rate": 8.741775385118447e-05,
      "loss": 0.0035,
      "step": 3170
    },
    {
      "epoch": 2.4710454722114266,
      "grad_norm": 0.07301243329178161,
      "learning_rate": 8.730509495328587e-05,
      "loss": 0.0034,
      "step": 3180
    },
    {
      "epoch": 2.478818499805674,
      "grad_norm": 0.07831526315789164,
      "learning_rate": 8.71920072276313e-05,
      "loss": 0.0034,
      "step": 3190
    },
    {
      "epoch": 2.486591527399922,
      "grad_norm": 0.10194734979674533,
      "learning_rate": 8.707849197418147e-05,
      "loss": 0.0034,
      "step": 3200
    },
    {
      "epoch": 2.49436455499417,
      "grad_norm": 0.07574798451139898,
      "learning_rate": 8.696455049781147e-05,
      "loss": 0.0035,
      "step": 3210
    },
    {
      "epoch": 2.5021375825884182,
      "grad_norm": 0.07681960051751799,
      "learning_rate": 8.6850184108296e-05,
      "loss": 0.0033,
      "step": 3220
    },
    {
      "epoch": 2.5099106101826663,
      "grad_norm": 0.07836534928409306,
      "learning_rate": 8.673539412029413e-05,
      "loss": 0.0033,
      "step": 3230
    },
    {
      "epoch": 2.517683637776914,
      "grad_norm": 0.08548600777071955,
      "learning_rate": 8.662018185333425e-05,
      "loss": 0.0034,
      "step": 3240
    },
    {
      "epoch": 2.525456665371162,
      "grad_norm": 0.0873567409210325,
      "learning_rate": 8.650454863179899e-05,
      "loss": 0.0032,
      "step": 3250
    },
    {
      "epoch": 2.53322969296541,
      "grad_norm": 0.08119732193506358,
      "learning_rate": 8.638849578490983e-05,
      "loss": 0.0032,
      "step": 3260
    },
    {
      "epoch": 2.541002720559658,
      "grad_norm": 0.07579199273478096,
      "learning_rate": 8.627202464671196e-05,
      "loss": 0.0031,
      "step": 3270
    },
    {
      "epoch": 2.548775748153906,
      "grad_norm": 0.07826696555904925,
      "learning_rate": 8.615513655605888e-05,
      "loss": 0.0031,
      "step": 3280
    },
    {
      "epoch": 2.556548775748154,
      "grad_norm": 0.07982815545083491,
      "learning_rate": 8.6037832856597e-05,
      "loss": 0.0031,
      "step": 3290
    },
    {
      "epoch": 2.564321803342402,
      "grad_norm": 0.08481539873916785,
      "learning_rate": 8.592011489675025e-05,
      "loss": 0.0031,
      "step": 3300
    },
    {
      "epoch": 2.57209483093665,
      "grad_norm": 0.07916097799005735,
      "learning_rate": 8.580198402970456e-05,
      "loss": 0.003,
      "step": 3310
    },
    {
      "epoch": 2.579867858530898,
      "grad_norm": 0.07473474197283404,
      "learning_rate": 8.56834416133922e-05,
      "loss": 0.003,
      "step": 3320
    },
    {
      "epoch": 2.5876408861251456,
      "grad_norm": 0.09130974953799981,
      "learning_rate": 8.556448901047638e-05,
      "loss": 0.0031,
      "step": 3330
    },
    {
      "epoch": 2.5954139137193937,
      "grad_norm": 0.1047401105445018,
      "learning_rate": 8.544512758833537e-05,
      "loss": 0.003,
      "step": 3340
    },
    {
      "epoch": 2.6031869413136417,
      "grad_norm": 0.06673230506522639,
      "learning_rate": 8.532535871904695e-05,
      "loss": 0.003,
      "step": 3350
    },
    {
      "epoch": 2.6109599689078897,
      "grad_norm": 0.07876904924601609,
      "learning_rate": 8.520518377937253e-05,
      "loss": 0.0031,
      "step": 3360
    },
    {
      "epoch": 2.6187329965021373,
      "grad_norm": 0.07009255329786931,
      "learning_rate": 8.508460415074137e-05,
      "loss": 0.0029,
      "step": 3370
    },
    {
      "epoch": 2.6265060240963853,
      "grad_norm": 0.07519024159981848,
      "learning_rate": 8.49636212192347e-05,
      "loss": 0.003,
      "step": 3380
    },
    {
      "epoch": 2.6342790516906334,
      "grad_norm": 0.08094145785537127,
      "learning_rate": 8.484223637556974e-05,
      "loss": 0.0031,
      "step": 3390
    },
    {
      "epoch": 2.6420520792848814,
      "grad_norm": 0.07736222204990759,
      "learning_rate": 8.472045101508382e-05,
      "loss": 0.003,
      "step": 3400
    },
    {
      "epoch": 2.6498251068791294,
      "grad_norm": 0.0776215180930073,
      "learning_rate": 8.459826653771821e-05,
      "loss": 0.003,
      "step": 3410
    },
    {
      "epoch": 2.6575981344733774,
      "grad_norm": 0.06742420975546093,
      "learning_rate": 8.44756843480021e-05,
      "loss": 0.0028,
      "step": 3420
    },
    {
      "epoch": 2.6653711620676255,
      "grad_norm": 0.0698399904847748,
      "learning_rate": 8.435270585503647e-05,
      "loss": 0.0029,
      "step": 3430
    },
    {
      "epoch": 2.6731441896618735,
      "grad_norm": 0.07591621006160705,
      "learning_rate": 8.422933247247786e-05,
      "loss": 0.0029,
      "step": 3440
    },
    {
      "epoch": 2.6809172172561215,
      "grad_norm": 0.06930363699655918,
      "learning_rate": 8.410556561852212e-05,
      "loss": 0.0029,
      "step": 3450
    },
    {
      "epoch": 2.688690244850369,
      "grad_norm": 0.07067953717607145,
      "learning_rate": 8.39814067158881e-05,
      "loss": 0.0028,
      "step": 3460
    },
    {
      "epoch": 2.696463272444617,
      "grad_norm": 0.077677315745107,
      "learning_rate": 8.385685719180132e-05,
      "loss": 0.0029,
      "step": 3470
    },
    {
      "epoch": 2.704236300038865,
      "grad_norm": 0.07570167649828695,
      "learning_rate": 8.373191847797754e-05,
      "loss": 0.0028,
      "step": 3480
    },
    {
      "epoch": 2.712009327633113,
      "grad_norm": 0.06955310603007175,
      "learning_rate": 8.360659201060639e-05,
      "loss": 0.0029,
      "step": 3490
    },
    {
      "epoch": 2.719782355227361,
      "grad_norm": 0.0659051677019647,
      "learning_rate": 8.348087923033465e-05,
      "loss": 0.0029,
      "step": 3500
    },
    {
      "epoch": 2.727555382821609,
      "grad_norm": 0.07181363122453262,
      "learning_rate": 8.335478158224996e-05,
      "loss": 0.0027,
      "step": 3510
    },
    {
      "epoch": 2.735328410415857,
      "grad_norm": 0.06585918912737933,
      "learning_rate": 8.3228300515864e-05,
      "loss": 0.0027,
      "step": 3520
    },
    {
      "epoch": 2.743101438010105,
      "grad_norm": 0.08664632265940041,
      "learning_rate": 8.310143748509588e-05,
      "loss": 0.0029,
      "step": 3530
    },
    {
      "epoch": 2.750874465604353,
      "grad_norm": 0.0773656650662347,
      "learning_rate": 8.297419394825554e-05,
      "loss": 0.0028,
      "step": 3540
    },
    {
      "epoch": 2.758647493198601,
      "grad_norm": 0.07477928292429042,
      "learning_rate": 8.28465713680268e-05,
      "loss": 0.0029,
      "step": 3550
    },
    {
      "epoch": 2.766420520792849,
      "grad_norm": 0.06614018293609181,
      "learning_rate": 8.271857121145073e-05,
      "loss": 0.0028,
      "step": 3560
    },
    {
      "epoch": 2.774193548387097,
      "grad_norm": 0.07953850815353782,
      "learning_rate": 8.25901949499086e-05,
      "loss": 0.0028,
      "step": 3570
    },
    {
      "epoch": 2.781966575981345,
      "grad_norm": 0.06857404401820201,
      "learning_rate": 8.246144405910517e-05,
      "loss": 0.0028,
      "step": 3580
    },
    {
      "epoch": 2.7897396035755926,
      "grad_norm": 0.06706667623803336,
      "learning_rate": 8.233232001905151e-05,
      "loss": 0.0029,
      "step": 3590
    },
    {
      "epoch": 2.7975126311698406,
      "grad_norm": 0.06548223623701806,
      "learning_rate": 8.220282431404818e-05,
      "loss": 0.0027,
      "step": 3600
    },
    {
      "epoch": 2.8052856587640886,
      "grad_norm": 0.06281526839867438,
      "learning_rate": 8.207295843266807e-05,
      "loss": 0.0025,
      "step": 3610
    },
    {
      "epoch": 2.8130586863583367,
      "grad_norm": 0.06453609241898647,
      "learning_rate": 8.194272386773928e-05,
      "loss": 0.0027,
      "step": 3620
    },
    {
      "epoch": 2.8208317139525847,
      "grad_norm": 0.06696000141114256,
      "learning_rate": 8.181212211632799e-05,
      "loss": 0.0027,
      "step": 3630
    },
    {
      "epoch": 2.8286047415468323,
      "grad_norm": 0.07223279206852921,
      "learning_rate": 8.168115467972127e-05,
      "loss": 0.0027,
      "step": 3640
    },
    {
      "epoch": 2.8363777691410803,
      "grad_norm": 0.0591131066140708,
      "learning_rate": 8.154982306340976e-05,
      "loss": 0.0025,
      "step": 3650
    },
    {
      "epoch": 2.8441507967353283,
      "grad_norm": 0.07344251027891184,
      "learning_rate": 8.141812877707043e-05,
      "loss": 0.0026,
      "step": 3660
    },
    {
      "epoch": 2.8519238243295764,
      "grad_norm": 0.061294159799181576,
      "learning_rate": 8.128607333454916e-05,
      "loss": 0.0025,
      "step": 3670
    },
    {
      "epoch": 2.8596968519238244,
      "grad_norm": 0.07538040581676492,
      "learning_rate": 8.115365825384343e-05,
      "loss": 0.0027,
      "step": 3680
    },
    {
      "epoch": 2.8674698795180724,
      "grad_norm": 0.068469569338319,
      "learning_rate": 8.10208850570848e-05,
      "loss": 0.0027,
      "step": 3690
    },
    {
      "epoch": 2.8752429071123204,
      "grad_norm": 0.07861201842013094,
      "learning_rate": 8.088775527052138e-05,
      "loss": 0.0026,
      "step": 3700
    },
    {
      "epoch": 2.8830159347065685,
      "grad_norm": 0.07032168038481322,
      "learning_rate": 8.075427042450039e-05,
      "loss": 0.0026,
      "step": 3710
    },
    {
      "epoch": 2.890788962300816,
      "grad_norm": 0.0727619064752268,
      "learning_rate": 8.062043205345048e-05,
      "loss": 0.0026,
      "step": 3720
    },
    {
      "epoch": 2.898561989895064,
      "grad_norm": 0.06690511970065317,
      "learning_rate": 8.048624169586416e-05,
      "loss": 0.0025,
      "step": 3730
    },
    {
      "epoch": 2.906335017489312,
      "grad_norm": 0.06902204913355058,
      "learning_rate": 8.035170089428002e-05,
      "loss": 0.0027,
      "step": 3740
    },
    {
      "epoch": 2.91410804508356,
      "grad_norm": 0.06444050726503392,
      "learning_rate": 8.021681119526514e-05,
      "loss": 0.0025,
      "step": 3750
    },
    {
      "epoch": 2.921881072677808,
      "grad_norm": 0.0703027939730794,
      "learning_rate": 8.008157414939716e-05,
      "loss": 0.0025,
      "step": 3760
    },
    {
      "epoch": 2.9296541002720558,
      "grad_norm": 0.0918381465147948,
      "learning_rate": 7.994599131124653e-05,
      "loss": 0.0024,
      "step": 3770
    },
    {
      "epoch": 2.9374271278663038,
      "grad_norm": 0.08102559952546114,
      "learning_rate": 7.981006423935867e-05,
      "loss": 0.0026,
      "step": 3780
    },
    {
      "epoch": 2.945200155460552,
      "grad_norm": 0.09770131109668051,
      "learning_rate": 7.967379449623601e-05,
      "loss": 0.0027,
      "step": 3790
    },
    {
      "epoch": 2.9529731830548,
      "grad_norm": 0.08951762311198015,
      "learning_rate": 7.953718364832002e-05,
      "loss": 0.0028,
      "step": 3800
    },
    {
      "epoch": 2.960746210649048,
      "grad_norm": 0.0816846540545239,
      "learning_rate": 7.940023326597323e-05,
      "loss": 0.0024,
      "step": 3810
    },
    {
      "epoch": 2.968519238243296,
      "grad_norm": 0.09871637060621355,
      "learning_rate": 7.926294492346116e-05,
      "loss": 0.0025,
      "step": 3820
    },
    {
      "epoch": 2.976292265837544,
      "grad_norm": 0.08781538442295887,
      "learning_rate": 7.912532019893429e-05,
      "loss": 0.0024,
      "step": 3830
    },
    {
      "epoch": 2.984065293431792,
      "grad_norm": 0.06767178490942136,
      "learning_rate": 7.898736067440975e-05,
      "loss": 0.0025,
      "step": 3840
    },
    {
      "epoch": 2.9918383210260395,
      "grad_norm": 0.0745712201927625,
      "learning_rate": 7.884906793575337e-05,
      "loss": 0.0025,
      "step": 3850
    },
    {
      "epoch": 2.9996113486202876,
      "grad_norm": 0.07397282294720663,
      "learning_rate": 7.871044357266124e-05,
      "loss": 0.0026,
      "step": 3860
    },
    {
      "epoch": 3.006995724834823,
      "grad_norm": 0.06317771768056155,
      "learning_rate": 7.85714891786416e-05,
      "loss": 0.002,
      "step": 3870
    },
    {
      "epoch": 3.014768752429071,
      "grad_norm": 0.05292609157896675,
      "learning_rate": 7.843220635099633e-05,
      "loss": 0.0016,
      "step": 3880
    },
    {
      "epoch": 3.022541780023319,
      "grad_norm": 0.04882267190819284,
      "learning_rate": 7.829259669080284e-05,
      "loss": 0.0015,
      "step": 3890
    },
    {
      "epoch": 3.030314807617567,
      "grad_norm": 0.07045644307155315,
      "learning_rate": 7.815266180289542e-05,
      "loss": 0.0015,
      "step": 3900
    },
    {
      "epoch": 3.038087835211815,
      "grad_norm": 0.055379078896731576,
      "learning_rate": 7.8012403295847e-05,
      "loss": 0.0014,
      "step": 3910
    },
    {
      "epoch": 3.045860862806063,
      "grad_norm": 0.057709864957742905,
      "learning_rate": 7.787182278195047e-05,
      "loss": 0.0014,
      "step": 3920
    },
    {
      "epoch": 3.053633890400311,
      "grad_norm": 0.05040524795529855,
      "learning_rate": 7.773092187720029e-05,
      "loss": 0.0014,
      "step": 3930
    },
    {
      "epoch": 3.061406917994559,
      "grad_norm": 0.05035382666579594,
      "learning_rate": 7.758970220127383e-05,
      "loss": 0.0013,
      "step": 3940
    },
    {
      "epoch": 3.069179945588807,
      "grad_norm": 0.05592780408751526,
      "learning_rate": 7.744816537751283e-05,
      "loss": 0.0013,
      "step": 3950
    },
    {
      "epoch": 3.076952973183055,
      "grad_norm": 0.05049381750502883,
      "learning_rate": 7.730631303290466e-05,
      "loss": 0.0012,
      "step": 3960
    },
    {
      "epoch": 3.084726000777303,
      "grad_norm": 0.04277340066544573,
      "learning_rate": 7.716414679806364e-05,
      "loss": 0.0012,
      "step": 3970
    },
    {
      "epoch": 3.0924990283715506,
      "grad_norm": 0.05126338146130596,
      "learning_rate": 7.702166830721234e-05,
      "loss": 0.0012,
      "step": 3980
    },
    {
      "epoch": 3.1002720559657986,
      "grad_norm": 0.055321540703598515,
      "learning_rate": 7.687887919816271e-05,
      "loss": 0.0012,
      "step": 3990
    },
    {
      "epoch": 3.1080450835600466,
      "grad_norm": 0.05448432149053431,
      "learning_rate": 7.673578111229738e-05,
      "loss": 0.0013,
      "step": 4000
    },
    {
      "epoch": 3.1158181111542946,
      "grad_norm": 0.05849173735885185,
      "learning_rate": 7.659237569455064e-05,
      "loss": 0.0013,
      "step": 4010
    },
    {
      "epoch": 3.1235911387485427,
      "grad_norm": 0.0532028841075432,
      "learning_rate": 7.644866459338964e-05,
      "loss": 0.0012,
      "step": 4020
    },
    {
      "epoch": 3.1313641663427907,
      "grad_norm": 0.04800300277794506,
      "learning_rate": 7.630464946079543e-05,
      "loss": 0.0013,
      "step": 4030
    },
    {
      "epoch": 3.1391371939370383,
      "grad_norm": 0.05653617859760966,
      "learning_rate": 7.616033195224391e-05,
      "loss": 0.0013,
      "step": 4040
    },
    {
      "epoch": 3.1469102215312863,
      "grad_norm": 0.06752313338740537,
      "learning_rate": 7.601571372668682e-05,
      "loss": 0.0013,
      "step": 4050
    },
    {
      "epoch": 3.1546832491255343,
      "grad_norm": 0.05046243795487172,
      "learning_rate": 7.587079644653277e-05,
      "loss": 0.0013,
      "step": 4060
    },
    {
      "epoch": 3.1624562767197824,
      "grad_norm": 0.05678047625277804,
      "learning_rate": 7.572558177762797e-05,
      "loss": 0.0013,
      "step": 4070
    },
    {
      "epoch": 3.1702293043140304,
      "grad_norm": 0.05051804020211108,
      "learning_rate": 7.558007138923718e-05,
      "loss": 0.0012,
      "step": 4080
    },
    {
      "epoch": 3.1780023319082784,
      "grad_norm": 0.05610066773526129,
      "learning_rate": 7.543426695402452e-05,
      "loss": 0.0013,
      "step": 4090
    },
    {
      "epoch": 3.1857753595025264,
      "grad_norm": 0.04650111729446382,
      "learning_rate": 7.52881701480342e-05,
      "loss": 0.0012,
      "step": 4100
    },
    {
      "epoch": 3.193548387096774,
      "grad_norm": 0.05002456571280218,
      "learning_rate": 7.514178265067129e-05,
      "loss": 0.0013,
      "step": 4110
    },
    {
      "epoch": 3.201321414691022,
      "grad_norm": 0.054771821749841916,
      "learning_rate": 7.49951061446824e-05,
      "loss": 0.0012,
      "step": 4120
    },
    {
      "epoch": 3.20909444228527,
      "grad_norm": 0.054851003739267906,
      "learning_rate": 7.484814231613633e-05,
      "loss": 0.0012,
      "step": 4130
    },
    {
      "epoch": 3.216867469879518,
      "grad_norm": 0.05095639836386469,
      "learning_rate": 7.470089285440471e-05,
      "loss": 0.0013,
      "step": 4140
    },
    {
      "epoch": 3.224640497473766,
      "grad_norm": 0.051210658981078835,
      "learning_rate": 7.455335945214257e-05,
      "loss": 0.0013,
      "step": 4150
    },
    {
      "epoch": 3.232413525068014,
      "grad_norm": 0.050630687016918294,
      "learning_rate": 7.440554380526884e-05,
      "loss": 0.0012,
      "step": 4160
    },
    {
      "epoch": 3.2401865526622617,
      "grad_norm": 0.05481015847694094,
      "learning_rate": 7.425744761294696e-05,
      "loss": 0.0014,
      "step": 4170
    },
    {
      "epoch": 3.2479595802565098,
      "grad_norm": 0.04746139811284292,
      "learning_rate": 7.41090725775652e-05,
      "loss": 0.0012,
      "step": 4180
    },
    {
      "epoch": 3.255732607850758,
      "grad_norm": 0.0488852581761488,
      "learning_rate": 7.396042040471728e-05,
      "loss": 0.0012,
      "step": 4190
    },
    {
      "epoch": 3.263505635445006,
      "grad_norm": 0.054029849239582,
      "learning_rate": 7.381149280318255e-05,
      "loss": 0.0013,
      "step": 4200
    },
    {
      "epoch": 3.271278663039254,
      "grad_norm": 0.05953354906133648,
      "learning_rate": 7.366229148490654e-05,
      "loss": 0.0013,
      "step": 4210
    },
    {
      "epoch": 3.279051690633502,
      "grad_norm": 0.06131326556461926,
      "learning_rate": 7.351281816498113e-05,
      "loss": 0.0014,
      "step": 4220
    },
    {
      "epoch": 3.2868247182277495,
      "grad_norm": 0.059091602709714926,
      "learning_rate": 7.336307456162492e-05,
      "loss": 0.0014,
      "step": 4230
    },
    {
      "epoch": 3.2945977458219975,
      "grad_norm": 0.07565280363176315,
      "learning_rate": 7.32130623961635e-05,
      "loss": 0.0014,
      "step": 4240
    },
    {
      "epoch": 3.3023707734162455,
      "grad_norm": 0.056892619126486126,
      "learning_rate": 7.306278339300959e-05,
      "loss": 0.0013,
      "step": 4250
    },
    {
      "epoch": 3.3101438010104935,
      "grad_norm": 0.11209680884929493,
      "learning_rate": 7.291223927964321e-05,
      "loss": 0.0015,
      "step": 4260
    },
    {
      "epoch": 3.3179168286047416,
      "grad_norm": 0.053482607975617505,
      "learning_rate": 7.276143178659194e-05,
      "loss": 0.0014,
      "step": 4270
    },
    {
      "epoch": 3.3256898561989896,
      "grad_norm": 0.04564916808200296,
      "learning_rate": 7.261036264741092e-05,
      "loss": 0.0013,
      "step": 4280
    },
    {
      "epoch": 3.3334628837932376,
      "grad_norm": 0.045353190964480296,
      "learning_rate": 7.24590335986629e-05,
      "loss": 0.0013,
      "step": 4290
    },
    {
      "epoch": 3.3412359113874857,
      "grad_norm": 0.05770878613559124,
      "learning_rate": 7.230744637989841e-05,
      "loss": 0.0012,
      "step": 4300
    },
    {
      "epoch": 3.3490089389817332,
      "grad_norm": 0.04582085382704389,
      "learning_rate": 7.215560273363564e-05,
      "loss": 0.0012,
      "step": 4310
    },
    {
      "epoch": 3.3567819665759813,
      "grad_norm": 0.05126717402300159,
      "learning_rate": 7.200350440534048e-05,
      "loss": 0.0013,
      "step": 4320
    },
    {
      "epoch": 3.3645549941702293,
      "grad_norm": 0.05175638481119324,
      "learning_rate": 7.185115314340638e-05,
      "loss": 0.0013,
      "step": 4330
    },
    {
      "epoch": 3.3723280217644773,
      "grad_norm": 0.05311419733020393,
      "learning_rate": 7.169855069913434e-05,
      "loss": 0.0013,
      "step": 4340
    },
    {
      "epoch": 3.3801010493587254,
      "grad_norm": 0.056853567416186954,
      "learning_rate": 7.154569882671273e-05,
      "loss": 0.0013,
      "step": 4350
    },
    {
      "epoch": 3.387874076952973,
      "grad_norm": 0.04541502908826204,
      "learning_rate": 7.139259928319715e-05,
      "loss": 0.0013,
      "step": 4360
    },
    {
      "epoch": 3.395647104547221,
      "grad_norm": 0.050130248796668546,
      "learning_rate": 7.12392538284902e-05,
      "loss": 0.0014,
      "step": 4370
    },
    {
      "epoch": 3.403420132141469,
      "grad_norm": 0.05299964388790246,
      "learning_rate": 7.108566422532123e-05,
      "loss": 0.0013,
      "step": 4380
    },
    {
      "epoch": 3.411193159735717,
      "grad_norm": 0.0508152329921838,
      "learning_rate": 7.093183223922621e-05,
      "loss": 0.0013,
      "step": 4390
    },
    {
      "epoch": 3.418966187329965,
      "grad_norm": 0.057542259788822554,
      "learning_rate": 7.077775963852724e-05,
      "loss": 0.0014,
      "step": 4400
    },
    {
      "epoch": 3.426739214924213,
      "grad_norm": 0.050631581931810374,
      "learning_rate": 7.06234481943124e-05,
      "loss": 0.0014,
      "step": 4410
    },
    {
      "epoch": 3.434512242518461,
      "grad_norm": 0.06953796070789972,
      "learning_rate": 7.046889968041524e-05,
      "loss": 0.0014,
      "step": 4420
    },
    {
      "epoch": 3.442285270112709,
      "grad_norm": 0.05342125151124089,
      "learning_rate": 7.03141158733945e-05,
      "loss": 0.0013,
      "step": 4430
    },
    {
      "epoch": 3.4500582977069567,
      "grad_norm": 0.052119437229514516,
      "learning_rate": 7.015909855251367e-05,
      "loss": 0.0014,
      "step": 4440
    },
    {
      "epoch": 3.4578313253012047,
      "grad_norm": 0.06486726674849891,
      "learning_rate": 7.000384949972047e-05,
      "loss": 0.0014,
      "step": 4450
    },
    {
      "epoch": 3.4656043528954528,
      "grad_norm": 0.06859610570190047,
      "learning_rate": 6.984837049962646e-05,
      "loss": 0.0014,
      "step": 4460
    },
    {
      "epoch": 3.473377380489701,
      "grad_norm": 0.05617796165249675,
      "learning_rate": 6.969266333948646e-05,
      "loss": 0.0014,
      "step": 4470
    },
    {
      "epoch": 3.481150408083949,
      "grad_norm": 0.06717819564707593,
      "learning_rate": 6.953672980917801e-05,
      "loss": 0.0015,
      "step": 4480
    },
    {
      "epoch": 3.488923435678197,
      "grad_norm": 0.0685334676438363,
      "learning_rate": 6.938057170118082e-05,
      "loss": 0.0014,
      "step": 4490
    },
    {
      "epoch": 3.4966964632724444,
      "grad_norm": 0.053206585518083986,
      "learning_rate": 6.922419081055617e-05,
      "loss": 0.0013,
      "step": 4500
    },
    {
      "epoch": 3.5044694908666925,
      "grad_norm": 0.057804877493537,
      "learning_rate": 6.906758893492624e-05,
      "loss": 0.0013,
      "step": 4510
    },
    {
      "epoch": 3.5122425184609405,
      "grad_norm": 0.0752984087415248,
      "learning_rate": 6.891076787445348e-05,
      "loss": 0.0014,
      "step": 4520
    },
    {
      "epoch": 3.5200155460551885,
      "grad_norm": 0.07684456686374479,
      "learning_rate": 6.875372943181989e-05,
      "loss": 0.0014,
      "step": 4530
    },
    {
      "epoch": 3.5277885736494365,
      "grad_norm": 0.05151187127939253,
      "learning_rate": 6.859647541220633e-05,
      "loss": 0.0015,
      "step": 4540
    },
    {
      "epoch": 3.5355616012436846,
      "grad_norm": 0.06120977928590361,
      "learning_rate": 6.843900762327171e-05,
      "loss": 0.0015,
      "step": 4550
    },
    {
      "epoch": 3.5433346288379326,
      "grad_norm": 0.052241865212588494,
      "learning_rate": 6.828132787513232e-05,
      "loss": 0.0015,
      "step": 4560
    },
    {
      "epoch": 3.55110765643218,
      "grad_norm": 0.062291869625660766,
      "learning_rate": 6.812343798034088e-05,
      "loss": 0.0015,
      "step": 4570
    },
    {
      "epoch": 3.558880684026428,
      "grad_norm": 0.06940645079889347,
      "learning_rate": 6.796533975386582e-05,
      "loss": 0.0014,
      "step": 4580
    },
    {
      "epoch": 3.5666537116206762,
      "grad_norm": 0.05299761798149833,
      "learning_rate": 6.780703501307037e-05,
      "loss": 0.0015,
      "step": 4590
    },
    {
      "epoch": 3.5744267392149243,
      "grad_norm": 0.0553631663183678,
      "learning_rate": 6.764852557769168e-05,
      "loss": 0.0014,
      "step": 4600
    },
    {
      "epoch": 3.5821997668091723,
      "grad_norm": 0.05928665736489485,
      "learning_rate": 6.748981326981987e-05,
      "loss": 0.0014,
      "step": 4610
    },
    {
      "epoch": 3.58997279440342,
      "grad_norm": 0.06100690574773009,
      "learning_rate": 6.733089991387711e-05,
      "loss": 0.0013,
      "step": 4620
    },
    {
      "epoch": 3.597745821997668,
      "grad_norm": 0.06554679880311645,
      "learning_rate": 6.717178733659669e-05,
      "loss": 0.0015,
      "step": 4630
    },
    {
      "epoch": 3.605518849591916,
      "grad_norm": 0.04648835107205993,
      "learning_rate": 6.701247736700193e-05,
      "loss": 0.0014,
      "step": 4640
    },
    {
      "epoch": 3.613291877186164,
      "grad_norm": 0.0485433533499338,
      "learning_rate": 6.685297183638524e-05,
      "loss": 0.0013,
      "step": 4650
    },
    {
      "epoch": 3.621064904780412,
      "grad_norm": 0.05273132935003627,
      "learning_rate": 6.669327257828702e-05,
      "loss": 0.0013,
      "step": 4660
    },
    {
      "epoch": 3.62883793237466,
      "grad_norm": 0.050580958804008154,
      "learning_rate": 6.653338142847459e-05,
      "loss": 0.0013,
      "step": 4670
    },
    {
      "epoch": 3.636610959968908,
      "grad_norm": 0.0522871787450838,
      "learning_rate": 6.637330022492112e-05,
      "loss": 0.0013,
      "step": 4680
    },
    {
      "epoch": 3.644383987563156,
      "grad_norm": 0.054308887619061295,
      "learning_rate": 6.621303080778445e-05,
      "loss": 0.0014,
      "step": 4690
    },
    {
      "epoch": 3.6521570151574037,
      "grad_norm": 0.05378027866810176,
      "learning_rate": 6.605257501938596e-05,
      "loss": 0.0014,
      "step": 4700
    },
    {
      "epoch": 3.6599300427516517,
      "grad_norm": 0.05266369589319868,
      "learning_rate": 6.589193470418941e-05,
      "loss": 0.0014,
      "step": 4710
    },
    {
      "epoch": 3.6677030703458997,
      "grad_norm": 0.05465334294471026,
      "learning_rate": 6.573111170877976e-05,
      "loss": 0.0013,
      "step": 4720
    },
    {
      "epoch": 3.6754760979401477,
      "grad_norm": 0.05753126693229123,
      "learning_rate": 6.557010788184182e-05,
      "loss": 0.0015,
      "step": 4730
    },
    {
      "epoch": 3.6832491255343958,
      "grad_norm": 0.0867356792695403,
      "learning_rate": 6.540892507413917e-05,
      "loss": 0.0016,
      "step": 4740
    },
    {
      "epoch": 3.6910221531286433,
      "grad_norm": 0.06764490662130833,
      "learning_rate": 6.524756513849273e-05,
      "loss": 0.0014,
      "step": 4750
    },
    {
      "epoch": 3.6987951807228914,
      "grad_norm": 0.0704576802572128,
      "learning_rate": 6.508602992975963e-05,
      "loss": 0.0016,
      "step": 4760
    },
    {
      "epoch": 3.7065682083171394,
      "grad_norm": 0.0802767664255637,
      "learning_rate": 6.492432130481166e-05,
      "loss": 0.0015,
      "step": 4770
    },
    {
      "epoch": 3.7143412359113874,
      "grad_norm": 0.06395636400017118,
      "learning_rate": 6.476244112251418e-05,
      "loss": 0.0014,
      "step": 4780
    },
    {
      "epoch": 3.7221142635056355,
      "grad_norm": 0.0811016126799815,
      "learning_rate": 6.460039124370455e-05,
      "loss": 0.0015,
      "step": 4790
    },
    {
      "epoch": 3.7298872910998835,
      "grad_norm": 0.07545299601321427,
      "learning_rate": 6.443817353117085e-05,
      "loss": 0.0014,
      "step": 4800
    },
    {
      "epoch": 3.7376603186941315,
      "grad_norm": 0.0573497980247029,
      "learning_rate": 6.42757898496304e-05,
      "loss": 0.0013,
      "step": 4810
    },
    {
      "epoch": 3.7454333462883795,
      "grad_norm": 0.05669253189317607,
      "learning_rate": 6.411324206570843e-05,
      "loss": 0.0014,
      "step": 4820
    },
    {
      "epoch": 3.753206373882627,
      "grad_norm": 0.060651625048542414,
      "learning_rate": 6.395053204791648e-05,
      "loss": 0.0013,
      "step": 4830
    },
    {
      "epoch": 3.760979401476875,
      "grad_norm": 0.06271483916694727,
      "learning_rate": 6.378766166663102e-05,
      "loss": 0.0014,
      "step": 4840
    },
    {
      "epoch": 3.768752429071123,
      "grad_norm": 0.058985216905366365,
      "learning_rate": 6.362463279407192e-05,
      "loss": 0.0013,
      "step": 4850
    },
    {
      "epoch": 3.776525456665371,
      "grad_norm": 0.08016429533408413,
      "learning_rate": 6.346144730428094e-05,
      "loss": 0.0014,
      "step": 4860
    },
    {
      "epoch": 3.7842984842596192,
      "grad_norm": 0.05464432483179166,
      "learning_rate": 6.329810707310017e-05,
      "loss": 0.0014,
      "step": 4870
    },
    {
      "epoch": 3.792071511853867,
      "grad_norm": 0.05302175690868426,
      "learning_rate": 6.313461397815049e-05,
      "loss": 0.0014,
      "step": 4880
    },
    {
      "epoch": 3.799844539448115,
      "grad_norm": 0.051860409972017484,
      "learning_rate": 6.297096989880994e-05,
      "loss": 0.0014,
      "step": 4890
    },
    {
      "epoch": 3.807617567042363,
      "grad_norm": 0.06121655864510598,
      "learning_rate": 6.280717671619219e-05,
      "loss": 0.0013,
      "step": 4900
    },
    {
      "epoch": 3.815390594636611,
      "grad_norm": 0.04964639600777849,
      "learning_rate": 6.264323631312483e-05,
      "loss": 0.0013,
      "step": 4910
    },
    {
      "epoch": 3.823163622230859,
      "grad_norm": 0.05610965209960934,
      "learning_rate": 6.247915057412781e-05,
      "loss": 0.0013,
      "step": 4920
    },
    {
      "epoch": 3.830936649825107,
      "grad_norm": 0.06260573144334115,
      "learning_rate": 6.231492138539174e-05,
      "loss": 0.0013,
      "step": 4930
    },
    {
      "epoch": 3.838709677419355,
      "grad_norm": 0.05298276249108928,
      "learning_rate": 6.21505506347562e-05,
      "loss": 0.0014,
      "step": 4940
    },
    {
      "epoch": 3.846482705013603,
      "grad_norm": 0.058326570847754826,
      "learning_rate": 6.198604021168798e-05,
      "loss": 0.0013,
      "step": 4950
    },
    {
      "epoch": 3.8542557326078506,
      "grad_norm": 0.05260934675853932,
      "learning_rate": 6.182139200725953e-05,
      "loss": 0.0013,
      "step": 4960
    },
    {
      "epoch": 3.8620287602020986,
      "grad_norm": 0.05423996230834983,
      "learning_rate": 6.165660791412704e-05,
      "loss": 0.0013,
      "step": 4970
    },
    {
      "epoch": 3.8698017877963466,
      "grad_norm": 0.051638650500176166,
      "learning_rate": 6.149168982650882e-05,
      "loss": 0.0013,
      "step": 4980
    },
    {
      "epoch": 3.8775748153905947,
      "grad_norm": 0.05511171077247131,
      "learning_rate": 6.132663964016343e-05,
      "loss": 0.0013,
      "step": 4990
    },
    {
      "epoch": 3.8853478429848427,
      "grad_norm": 0.04961157230866775,
      "learning_rate": 6.116145925236795e-05,
      "loss": 0.0013,
      "step": 5000
    },
    {
      "epoch": 3.8931208705790903,
      "grad_norm": 0.05265851472161743,
      "learning_rate": 6.099615056189609e-05,
      "loss": 0.0013,
      "step": 5010
    },
    {
      "epoch": 3.9008938981733383,
      "grad_norm": 0.05197040384153092,
      "learning_rate": 6.083071546899649e-05,
      "loss": 0.0012,
      "step": 5020
    },
    {
      "epoch": 3.9086669257675863,
      "grad_norm": 0.058677791758000766,
      "learning_rate": 6.066515587537077e-05,
      "loss": 0.0014,
      "step": 5030
    },
    {
      "epoch": 3.9164399533618344,
      "grad_norm": 0.05080762076034653,
      "learning_rate": 6.0499473684151695e-05,
      "loss": 0.0013,
      "step": 5040
    },
    {
      "epoch": 3.9242129809560824,
      "grad_norm": 0.0505243429638093,
      "learning_rate": 6.0333670799881346e-05,
      "loss": 0.0012,
      "step": 5050
    },
    {
      "epoch": 3.9319860085503304,
      "grad_norm": 0.04592349968189807,
      "learning_rate": 6.0167749128489136e-05,
      "loss": 0.0012,
      "step": 5060
    },
    {
      "epoch": 3.9397590361445785,
      "grad_norm": 0.05707189919349449,
      "learning_rate": 6.000171057726999e-05,
      "loss": 0.0013,
      "step": 5070
    },
    {
      "epoch": 3.9475320637388265,
      "grad_norm": 0.0526931495556647,
      "learning_rate": 5.983555705486239e-05,
      "loss": 0.0012,
      "step": 5080
    },
    {
      "epoch": 3.9553050913330745,
      "grad_norm": 0.051013582659466365,
      "learning_rate": 5.9669290471226405e-05,
      "loss": 0.0013,
      "step": 5090
    },
    {
      "epoch": 3.963078118927322,
      "grad_norm": 0.052603820387661857,
      "learning_rate": 5.9502912737621764e-05,
      "loss": 0.0012,
      "step": 5100
    },
    {
      "epoch": 3.97085114652157,
      "grad_norm": 0.051720172092351926,
      "learning_rate": 5.9336425766585924e-05,
      "loss": 0.0012,
      "step": 5110
    },
    {
      "epoch": 3.978624174115818,
      "grad_norm": 0.10597495725038626,
      "learning_rate": 5.9169831471911974e-05,
      "loss": 0.0013,
      "step": 5120
    },
    {
      "epoch": 3.986397201710066,
      "grad_norm": 0.04999517365959679,
      "learning_rate": 5.900313176862675e-05,
      "loss": 0.0013,
      "step": 5130
    },
    {
      "epoch": 3.9941702293043138,
      "grad_norm": 0.05994926145731603,
      "learning_rate": 5.883632857296876e-05,
      "loss": 0.0013,
      "step": 5140
    },
    {
      "epoch": 4.00155460551885,
      "grad_norm": 0.06828827893546477,
      "learning_rate": 5.8669423802366177e-05,
      "loss": 0.0012,
      "step": 5150
    },
    {
      "epoch": 4.009327633113098,
      "grad_norm": 0.04581917688238487,
      "learning_rate": 5.850241937541482e-05,
      "loss": 0.0008,
      "step": 5160
    },
    {
      "epoch": 4.017100660707346,
      "grad_norm": 0.04767494608662352,
      "learning_rate": 5.833531721185601e-05,
      "loss": 0.0008,
      "step": 5170
    },
    {
      "epoch": 4.024873688301594,
      "grad_norm": 0.03415064425236542,
      "learning_rate": 5.8168119232554605e-05,
      "loss": 0.0007,
      "step": 5180
    },
    {
      "epoch": 4.032646715895841,
      "grad_norm": 0.03156168690639689,
      "learning_rate": 5.80008273594769e-05,
      "loss": 0.0006,
      "step": 5190
    },
    {
      "epoch": 4.040419743490089,
      "grad_norm": 0.02775459034078891,
      "learning_rate": 5.783344351566849e-05,
      "loss": 0.0006,
      "step": 5200
    },
    {
      "epoch": 4.048192771084337,
      "grad_norm": 0.03752890797711968,
      "learning_rate": 5.7665969625232174e-05,
      "loss": 0.0006,
      "step": 5210
    },
    {
      "epoch": 4.055965798678585,
      "grad_norm": 0.039795978106431484,
      "learning_rate": 5.749840761330587e-05,
      "loss": 0.0006,
      "step": 5220
    },
    {
      "epoch": 4.063738826272833,
      "grad_norm": 0.04591425788703762,
      "learning_rate": 5.733075940604045e-05,
      "loss": 0.0006,
      "step": 5230
    },
    {
      "epoch": 4.071511853867081,
      "grad_norm": 0.03678302094567926,
      "learning_rate": 5.7163026930577636e-05,
      "loss": 0.0006,
      "step": 5240
    },
    {
      "epoch": 4.079284881461329,
      "grad_norm": 0.042733165935266,
      "learning_rate": 5.69952121150278e-05,
      "loss": 0.0006,
      "step": 5250
    },
    {
      "epoch": 4.087057909055577,
      "grad_norm": 0.034514951085523264,
      "learning_rate": 5.682731688844786e-05,
      "loss": 0.0005,
      "step": 5260
    },
    {
      "epoch": 4.094830936649825,
      "grad_norm": 0.03331717312064519,
      "learning_rate": 5.6659343180819036e-05,
      "loss": 0.0005,
      "step": 5270
    },
    {
      "epoch": 4.102603964244073,
      "grad_norm": 0.04058083373382621,
      "learning_rate": 5.649129292302473e-05,
      "loss": 0.0005,
      "step": 5280
    },
    {
      "epoch": 4.110376991838321,
      "grad_norm": 0.034283836253464796,
      "learning_rate": 5.632316804682828e-05,
      "loss": 0.0005,
      "step": 5290
    },
    {
      "epoch": 4.118150019432569,
      "grad_norm": 0.037857136276281164,
      "learning_rate": 5.615497048485078e-05,
      "loss": 0.0005,
      "step": 5300
    },
    {
      "epoch": 4.125923047026817,
      "grad_norm": 0.038838286688120054,
      "learning_rate": 5.598670217054883e-05,
      "loss": 0.0005,
      "step": 5310
    },
    {
      "epoch": 4.1336960746210645,
      "grad_norm": 0.02729201236801898,
      "learning_rate": 5.58183650381924e-05,
      "loss": 0.0005,
      "step": 5320
    },
    {
      "epoch": 4.1414691022153125,
      "grad_norm": 0.04084905595927271,
      "learning_rate": 5.56499610228425e-05,
      "loss": 0.0005,
      "step": 5330
    },
    {
      "epoch": 4.1492421298095605,
      "grad_norm": 0.032422360783001807,
      "learning_rate": 5.5481492060328945e-05,
      "loss": 0.0005,
      "step": 5340
    },
    {
      "epoch": 4.157015157403809,
      "grad_norm": 0.037801558928930515,
      "learning_rate": 5.531296008722816e-05,
      "loss": 0.0004,
      "step": 5350
    },
    {
      "epoch": 4.164788184998057,
      "grad_norm": 0.03619523638310936,
      "learning_rate": 5.5144367040840896e-05,
      "loss": 0.0006,
      "step": 5360
    },
    {
      "epoch": 4.172561212592305,
      "grad_norm": 0.04120623681054301,
      "learning_rate": 5.4975714859169914e-05,
      "loss": 0.0005,
      "step": 5370
    },
    {
      "epoch": 4.180334240186553,
      "grad_norm": 0.029566587403069115,
      "learning_rate": 5.480700548089779e-05,
      "loss": 0.0005,
      "step": 5380
    },
    {
      "epoch": 4.188107267780801,
      "grad_norm": 0.03750169201927612,
      "learning_rate": 5.463824084536453e-05,
      "loss": 0.0005,
      "step": 5390
    },
    {
      "epoch": 4.195880295375049,
      "grad_norm": 0.030745808156288912,
      "learning_rate": 5.446942289254536e-05,
      "loss": 0.0006,
      "step": 5400
    },
    {
      "epoch": 4.203653322969297,
      "grad_norm": 0.0359645720597556,
      "learning_rate": 5.430055356302841e-05,
      "loss": 0.0006,
      "step": 5410
    },
    {
      "epoch": 4.211426350563545,
      "grad_norm": 0.034515918980682146,
      "learning_rate": 5.413163479799236e-05,
      "loss": 0.0006,
      "step": 5420
    },
    {
      "epoch": 4.219199378157793,
      "grad_norm": 0.03877548820046214,
      "learning_rate": 5.396266853918418e-05,
      "loss": 0.0006,
      "step": 5430
    },
    {
      "epoch": 4.226972405752041,
      "grad_norm": 0.04425623388447145,
      "learning_rate": 5.37936567288968e-05,
      "loss": 0.0006,
      "step": 5440
    },
    {
      "epoch": 4.234745433346289,
      "grad_norm": 0.04096699167456297,
      "learning_rate": 5.3624601309946754e-05,
      "loss": 0.0006,
      "step": 5450
    },
    {
      "epoch": 4.242518460940536,
      "grad_norm": 0.040203913710602966,
      "learning_rate": 5.345550422565184e-05,
      "loss": 0.0006,
      "step": 5460
    },
    {
      "epoch": 4.250291488534784,
      "grad_norm": 0.03670112205788334,
      "learning_rate": 5.328636741980887e-05,
      "loss": 0.0005,
      "step": 5470
    },
    {
      "epoch": 4.258064516129032,
      "grad_norm": 0.03631922782433492,
      "learning_rate": 5.311719283667118e-05,
      "loss": 0.0006,
      "step": 5480
    },
    {
      "epoch": 4.26583754372328,
      "grad_norm": 0.08057650609897654,
      "learning_rate": 5.294798242092642e-05,
      "loss": 0.0006,
      "step": 5490
    },
    {
      "epoch": 4.273610571317528,
      "grad_norm": 0.04169293878514899,
      "learning_rate": 5.2778738117674155e-05,
      "loss": 0.0005,
      "step": 5500
    },
    {
      "epoch": 4.281383598911776,
      "grad_norm": 0.04297356425214829,
      "learning_rate": 5.260946187240342e-05,
      "loss": 0.0006,
      "step": 5510
    },
    {
      "epoch": 4.289156626506024,
      "grad_norm": 0.05584211566238541,
      "learning_rate": 5.244015563097049e-05,
      "loss": 0.0007,
      "step": 5520
    },
    {
      "epoch": 4.296929654100272,
      "grad_norm": 0.04991582674756648,
      "learning_rate": 5.227082133957643e-05,
      "loss": 0.0008,
      "step": 5530
    },
    {
      "epoch": 4.30470268169452,
      "grad_norm": 0.04222493800756607,
      "learning_rate": 5.210146094474474e-05,
      "loss": 0.0008,
      "step": 5540
    },
    {
      "epoch": 4.312475709288768,
      "grad_norm": 0.03656442818921343,
      "learning_rate": 5.193207639329901e-05,
      "loss": 0.0008,
      "step": 5550
    },
    {
      "epoch": 4.320248736883016,
      "grad_norm": 0.034567143296557225,
      "learning_rate": 5.176266963234048e-05,
      "loss": 0.0007,
      "step": 5560
    },
    {
      "epoch": 4.328021764477264,
      "grad_norm": 0.03483817210989784,
      "learning_rate": 5.159324260922567e-05,
      "loss": 0.0006,
      "step": 5570
    },
    {
      "epoch": 4.335794792071512,
      "grad_norm": 0.035603164851300176,
      "learning_rate": 5.142379727154409e-05,
      "loss": 0.0006,
      "step": 5580
    },
    {
      "epoch": 4.3435678196657594,
      "grad_norm": 0.028459828702280508,
      "learning_rate": 5.125433556709573e-05,
      "loss": 0.0006,
      "step": 5590
    },
    {
      "epoch": 4.3513408472600075,
      "grad_norm": 0.04113990106440544,
      "learning_rate": 5.108485944386873e-05,
      "loss": 0.0006,
      "step": 5600
    },
    {
      "epoch": 4.3591138748542555,
      "grad_norm": 0.04543144317857235,
      "learning_rate": 5.091537085001695e-05,
      "loss": 0.0005,
      "step": 5610
    },
    {
      "epoch": 4.3668869024485035,
      "grad_norm": 0.04324399977725394,
      "learning_rate": 5.074587173383768e-05,
      "loss": 0.0006,
      "step": 5620
    },
    {
      "epoch": 4.3746599300427516,
      "grad_norm": 0.030022351311664933,
      "learning_rate": 5.057636404374905e-05,
      "loss": 0.0006,
      "step": 5630
    },
    {
      "epoch": 4.382432957637,
      "grad_norm": 0.04051856954025839,
      "learning_rate": 5.040684972826785e-05,
      "loss": 0.0006,
      "step": 5640
    },
    {
      "epoch": 4.390205985231248,
      "grad_norm": 0.03471005917429346,
      "learning_rate": 5.023733073598695e-05,
      "loss": 0.0005,
      "step": 5650
    },
    {
      "epoch": 4.397979012825496,
      "grad_norm": 0.03265174066617206,
      "learning_rate": 5.006780901555307e-05,
      "loss": 0.0005,
      "step": 5660
    },
    {
      "epoch": 4.405752040419744,
      "grad_norm": 0.04507957606154376,
      "learning_rate": 4.989828651564419e-05,
      "loss": 0.0006,
      "step": 5670
    },
    {
      "epoch": 4.413525068013992,
      "grad_norm": 0.033075272882098945,
      "learning_rate": 4.972876518494734e-05,
      "loss": 0.0006,
      "step": 5680
    },
    {
      "epoch": 4.42129809560824,
      "grad_norm": 0.02836050085506267,
      "learning_rate": 4.955924697213605e-05,
      "loss": 0.0006,
      "step": 5690
    },
    {
      "epoch": 4.429071123202488,
      "grad_norm": 0.03852732063598139,
      "learning_rate": 4.938973382584806e-05,
      "loss": 0.0006,
      "step": 5700
    },
    {
      "epoch": 4.436844150796736,
      "grad_norm": 0.042783988393428805,
      "learning_rate": 4.922022769466279e-05,
      "loss": 0.0006,
      "step": 5710
    },
    {
      "epoch": 4.444617178390983,
      "grad_norm": 0.04469622309194219,
      "learning_rate": 4.9050730527079096e-05,
      "loss": 0.0005,
      "step": 5720
    },
    {
      "epoch": 4.452390205985231,
      "grad_norm": 0.04742224629097243,
      "learning_rate": 4.888124427149277e-05,
      "loss": 0.0005,
      "step": 5730
    },
    {
      "epoch": 4.460163233579479,
      "grad_norm": 0.02835405265520966,
      "learning_rate": 4.8711770876174175e-05,
      "loss": 0.0006,
      "step": 5740
    },
    {
      "epoch": 4.467936261173727,
      "grad_norm": 0.04055659419668568,
      "learning_rate": 4.854231228924582e-05,
      "loss": 0.0006,
      "step": 5750
    },
    {
      "epoch": 4.475709288767975,
      "grad_norm": 0.037513767560614523,
      "learning_rate": 4.837287045866001e-05,
      "loss": 0.0006,
      "step": 5760
    },
    {
      "epoch": 4.483482316362223,
      "grad_norm": 0.030948737484918637,
      "learning_rate": 4.8203447332176434e-05,
      "loss": 0.0005,
      "step": 5770
    },
    {
      "epoch": 4.491255343956471,
      "grad_norm": 0.041860398988624016,
      "learning_rate": 4.803404485733977e-05,
      "loss": 0.0006,
      "step": 5780
    },
    {
      "epoch": 4.499028371550719,
      "grad_norm": 0.03928519045542616,
      "learning_rate": 4.78646649814573e-05,
      "loss": 0.0005,
      "step": 5790
    },
    {
      "epoch": 4.506801399144967,
      "grad_norm": 0.026381525778442577,
      "learning_rate": 4.769530965157654e-05,
      "loss": 0.0005,
      "step": 5800
    },
    {
      "epoch": 4.514574426739215,
      "grad_norm": 0.03983527752214128,
      "learning_rate": 4.752598081446282e-05,
      "loss": 0.0005,
      "step": 5810
    },
    {
      "epoch": 4.522347454333463,
      "grad_norm": 0.03822679489740094,
      "learning_rate": 4.735668041657697e-05,
      "loss": 0.0005,
      "step": 5820
    },
    {
      "epoch": 4.530120481927711,
      "grad_norm": 0.027994164547717068,
      "learning_rate": 4.7187410404052865e-05,
      "loss": 0.0005,
      "step": 5830
    },
    {
      "epoch": 4.537893509521959,
      "grad_norm": 0.05281713573969291,
      "learning_rate": 4.70181727226751e-05,
      "loss": 0.0006,
      "step": 5840
    },
    {
      "epoch": 4.545666537116206,
      "grad_norm": 0.03183555510669356,
      "learning_rate": 4.684896931785666e-05,
      "loss": 0.0005,
      "step": 5850
    },
    {
      "epoch": 4.553439564710454,
      "grad_norm": 0.04425960004494883,
      "learning_rate": 4.667980213461647e-05,
      "loss": 0.0005,
      "step": 5860
    },
    {
      "epoch": 4.561212592304702,
      "grad_norm": 0.035447893053189544,
      "learning_rate": 4.651067311755712e-05,
      "loss": 0.0006,
      "step": 5870
    },
    {
      "epoch": 4.5689856198989505,
      "grad_norm": 0.03138729892970769,
      "learning_rate": 4.6341584210842445e-05,
      "loss": 0.0005,
      "step": 5880
    },
    {
      "epoch": 4.5767586474931985,
      "grad_norm": 0.02817622851989118,
      "learning_rate": 4.6172537358175214e-05,
      "loss": 0.0005,
      "step": 5890
    },
    {
      "epoch": 4.5845316750874465,
      "grad_norm": 0.02803642103316403,
      "learning_rate": 4.60035345027748e-05,
      "loss": 0.0006,
      "step": 5900
    },
    {
      "epoch": 4.5923047026816946,
      "grad_norm": 0.039503471899411856,
      "learning_rate": 4.5834577587354764e-05,
      "loss": 0.0006,
      "step": 5910
    },
    {
      "epoch": 4.600077730275943,
      "grad_norm": 0.03209705648101513,
      "learning_rate": 4.566566855410064e-05,
      "loss": 0.0006,
      "step": 5920
    },
    {
      "epoch": 4.607850757870191,
      "grad_norm": 0.036379862319002865,
      "learning_rate": 4.549680934464755e-05,
      "loss": 0.0006,
      "step": 5930
    },
    {
      "epoch": 4.615623785464439,
      "grad_norm": 0.03836163405627623,
      "learning_rate": 4.53280019000578e-05,
      "loss": 0.0006,
      "step": 5940
    },
    {
      "epoch": 4.623396813058687,
      "grad_norm": 0.04298997406601343,
      "learning_rate": 4.515924816079875e-05,
      "loss": 0.0006,
      "step": 5950
    },
    {
      "epoch": 4.631169840652935,
      "grad_norm": 0.03550821491196621,
      "learning_rate": 4.499055006672035e-05,
      "loss": 0.0006,
      "step": 5960
    },
    {
      "epoch": 4.638942868247183,
      "grad_norm": 0.032404204251160834,
      "learning_rate": 4.4821909557032945e-05,
      "loss": 0.0006,
      "step": 5970
    },
    {
      "epoch": 4.64671589584143,
      "grad_norm": 0.040733882859301615,
      "learning_rate": 4.465332857028489e-05,
      "loss": 0.0006,
      "step": 5980
    },
    {
      "epoch": 4.654488923435678,
      "grad_norm": 0.04237952442574177,
      "learning_rate": 4.448480904434037e-05,
      "loss": 0.0005,
      "step": 5990
    },
    {
      "epoch": 4.662261951029926,
      "grad_norm": 0.04459356513079646,
      "learning_rate": 4.4316352916356994e-05,
      "loss": 0.0006,
      "step": 6000
    },
    {
      "epoch": 4.670034978624174,
      "grad_norm": 0.037861533286158544,
      "learning_rate": 4.414796212276368e-05,
      "loss": 0.0005,
      "step": 6010
    },
    {
      "epoch": 4.677808006218422,
      "grad_norm": 0.03957507901230309,
      "learning_rate": 4.3979638599238255e-05,
      "loss": 0.0006,
      "step": 6020
    },
    {
      "epoch": 4.68558103381267,
      "grad_norm": 0.0382952293024233,
      "learning_rate": 4.3811384280685314e-05,
      "loss": 0.0006,
      "step": 6030
    },
    {
      "epoch": 4.693354061406918,
      "grad_norm": 0.04187911129915023,
      "learning_rate": 4.3643201101213924e-05,
      "loss": 0.0006,
      "step": 6040
    },
    {
      "epoch": 4.701127089001166,
      "grad_norm": 0.0370382821229986,
      "learning_rate": 4.347509099411533e-05,
      "loss": 0.0006,
      "step": 6050
    },
    {
      "epoch": 4.708900116595414,
      "grad_norm": 0.037130592559926755,
      "learning_rate": 4.330705589184089e-05,
      "loss": 0.0006,
      "step": 6060
    },
    {
      "epoch": 4.716673144189662,
      "grad_norm": 0.033489041804650066,
      "learning_rate": 4.313909772597972e-05,
      "loss": 0.0006,
      "step": 6070
    },
    {
      "epoch": 4.72444617178391,
      "grad_norm": 0.04023357069607381,
      "learning_rate": 4.297121842723657e-05,
      "loss": 0.0006,
      "step": 6080
    },
    {
      "epoch": 4.732219199378158,
      "grad_norm": 0.038278502632585185,
      "learning_rate": 4.280341992540956e-05,
      "loss": 0.0006,
      "step": 6090
    },
    {
      "epoch": 4.739992226972406,
      "grad_norm": 0.028658105892572495,
      "learning_rate": 4.263570414936807e-05,
      "loss": 0.0006,
      "step": 6100
    },
    {
      "epoch": 4.747765254566653,
      "grad_norm": 0.03769276819690435,
      "learning_rate": 4.246807302703052e-05,
      "loss": 0.0006,
      "step": 6110
    },
    {
      "epoch": 4.755538282160901,
      "grad_norm": 0.02821667325584842,
      "learning_rate": 4.230052848534223e-05,
      "loss": 0.0005,
      "step": 6120
    },
    {
      "epoch": 4.763311309755149,
      "grad_norm": 0.031860499212090505,
      "learning_rate": 4.213307245025326e-05,
      "loss": 0.0005,
      "step": 6130
    },
    {
      "epoch": 4.771084337349397,
      "grad_norm": 0.038768056070783374,
      "learning_rate": 4.196570684669627e-05,
      "loss": 0.0005,
      "step": 6140
    },
    {
      "epoch": 4.778857364943645,
      "grad_norm": 0.032625189900051796,
      "learning_rate": 4.1798433598564404e-05,
      "loss": 0.0005,
      "step": 6150
    },
    {
      "epoch": 4.7866303925378935,
      "grad_norm": 0.028251385265285735,
      "learning_rate": 4.163125462868917e-05,
      "loss": 0.0005,
      "step": 6160
    },
    {
      "epoch": 4.7944034201321415,
      "grad_norm": 0.033430380873123416,
      "learning_rate": 4.146417185881829e-05,
      "loss": 0.0006,
      "step": 6170
    },
    {
      "epoch": 4.8021764477263895,
      "grad_norm": 0.03981308165767607,
      "learning_rate": 4.129718720959372e-05,
      "loss": 0.0005,
      "step": 6180
    },
    {
      "epoch": 4.8099494753206375,
      "grad_norm": 0.03071785984517098,
      "learning_rate": 4.113030260052943e-05,
      "loss": 0.0004,
      "step": 6190
    },
    {
      "epoch": 4.817722502914886,
      "grad_norm": 0.039123776730175724,
      "learning_rate": 4.09635199499895e-05,
      "loss": 0.0005,
      "step": 6200
    },
    {
      "epoch": 4.825495530509134,
      "grad_norm": 0.03126204913854568,
      "learning_rate": 4.07968411751659e-05,
      "loss": 0.0005,
      "step": 6210
    },
    {
      "epoch": 4.833268558103382,
      "grad_norm": 0.036954611691286936,
      "learning_rate": 4.063026819205658e-05,
      "loss": 0.0005,
      "step": 6220
    },
    {
      "epoch": 4.84104158569763,
      "grad_norm": 0.04201937389302552,
      "learning_rate": 4.046380291544338e-05,
      "loss": 0.0005,
      "step": 6230
    },
    {
      "epoch": 4.848814613291877,
      "grad_norm": 0.034308301558943675,
      "learning_rate": 4.029744725887007e-05,
      "loss": 0.0005,
      "step": 6240
    },
    {
      "epoch": 4.856587640886125,
      "grad_norm": 0.03496979511736733,
      "learning_rate": 4.0131203134620255e-05,
      "loss": 0.0005,
      "step": 6250
    },
    {
      "epoch": 4.864360668480373,
      "grad_norm": 0.039032383869847566,
      "learning_rate": 3.996507245369551e-05,
      "loss": 0.0005,
      "step": 6260
    },
    {
      "epoch": 4.872133696074621,
      "grad_norm": 0.04672609065214268,
      "learning_rate": 3.979905712579337e-05,
      "loss": 0.0005,
      "step": 6270
    },
    {
      "epoch": 4.879906723668869,
      "grad_norm": 0.040546725722874294,
      "learning_rate": 3.963315905928532e-05,
      "loss": 0.0005,
      "step": 6280
    },
    {
      "epoch": 4.887679751263117,
      "grad_norm": 0.0378778673072386,
      "learning_rate": 3.946738016119494e-05,
      "loss": 0.0005,
      "step": 6290
    },
    {
      "epoch": 4.895452778857365,
      "grad_norm": 0.03653505654798841,
      "learning_rate": 3.930172233717595e-05,
      "loss": 0.0005,
      "step": 6300
    },
    {
      "epoch": 4.903225806451613,
      "grad_norm": 0.036025865353444486,
      "learning_rate": 3.91361874914903e-05,
      "loss": 0.0005,
      "step": 6310
    },
    {
      "epoch": 4.910998834045861,
      "grad_norm": 0.03351661782015591,
      "learning_rate": 3.8970777526986305e-05,
      "loss": 0.0005,
      "step": 6320
    },
    {
      "epoch": 4.918771861640109,
      "grad_norm": 0.03085402835322917,
      "learning_rate": 3.880549434507671e-05,
      "loss": 0.0005,
      "step": 6330
    },
    {
      "epoch": 4.926544889234357,
      "grad_norm": 0.03839492426475462,
      "learning_rate": 3.8640339845716926e-05,
      "loss": 0.0005,
      "step": 6340
    },
    {
      "epoch": 4.934317916828605,
      "grad_norm": 0.03253878806087454,
      "learning_rate": 3.847531592738309e-05,
      "loss": 0.0005,
      "step": 6350
    },
    {
      "epoch": 4.942090944422853,
      "grad_norm": 0.031363091899245576,
      "learning_rate": 3.8310424487050305e-05,
      "loss": 0.0005,
      "step": 6360
    },
    {
      "epoch": 4.9498639720171,
      "grad_norm": 0.03125931224362856,
      "learning_rate": 3.814566742017084e-05,
      "loss": 0.0005,
      "step": 6370
    },
    {
      "epoch": 4.957636999611348,
      "grad_norm": 0.031779154731528925,
      "learning_rate": 3.798104662065233e-05,
      "loss": 0.0005,
      "step": 6380
    },
    {
      "epoch": 4.965410027205596,
      "grad_norm": 0.040037883580701344,
      "learning_rate": 3.7816563980835915e-05,
      "loss": 0.0006,
      "step": 6390
    },
    {
      "epoch": 4.973183054799844,
      "grad_norm": 0.03485580192900446,
      "learning_rate": 3.765222139147465e-05,
      "loss": 0.0005,
      "step": 6400
    },
    {
      "epoch": 4.980956082394092,
      "grad_norm": 0.02605254313423768,
      "learning_rate": 3.7488020741711684e-05,
      "loss": 0.0005,
      "step": 6410
    },
    {
      "epoch": 4.98872910998834,
      "grad_norm": 0.03922318617471436,
      "learning_rate": 3.73239639190585e-05,
      "loss": 0.0005,
      "step": 6420
    },
    {
      "epoch": 4.996502137582588,
      "grad_norm": 0.042195200425152776,
      "learning_rate": 3.7160052809373326e-05,
      "loss": 0.0005,
      "step": 6430
    },
    {
      "epoch": 5.003886513797124,
      "grad_norm": 0.0435221672114546,
      "learning_rate": 3.699628929683933e-05,
      "loss": 0.0004,
      "step": 6440
    },
    {
      "epoch": 5.011659541391372,
      "grad_norm": 0.01657527742325064,
      "learning_rate": 3.683267526394309e-05,
      "loss": 0.0003,
      "step": 6450
    },
    {
      "epoch": 5.01943256898562,
      "grad_norm": 0.025699647370980343,
      "learning_rate": 3.666921259145286e-05,
      "loss": 0.0003,
      "step": 6460
    },
    {
      "epoch": 5.027205596579868,
      "grad_norm": 0.023869538526672097,
      "learning_rate": 3.6505903158396984e-05,
      "loss": 0.0003,
      "step": 6470
    },
    {
      "epoch": 5.034978624174116,
      "grad_norm": 0.021696416506413144,
      "learning_rate": 3.6342748842042315e-05,
      "loss": 0.0003,
      "step": 6480
    },
    {
      "epoch": 5.042751651768364,
      "grad_norm": 0.02138839902695352,
      "learning_rate": 3.6179751517872584e-05,
      "loss": 0.0002,
      "step": 6490
    },
    {
      "epoch": 5.050524679362612,
      "grad_norm": 0.017711646478733176,
      "learning_rate": 3.6016913059566934e-05,
      "loss": 0.0002,
      "step": 6500
    },
    {
      "epoch": 5.05829770695686,
      "grad_norm": 0.018527570632037076,
      "learning_rate": 3.5854235338978224e-05,
      "loss": 0.0002,
      "step": 6510
    },
    {
      "epoch": 5.066070734551108,
      "grad_norm": 0.018221510040150053,
      "learning_rate": 3.569172022611169e-05,
      "loss": 0.0002,
      "step": 6520
    },
    {
      "epoch": 5.073843762145356,
      "grad_norm": 0.016333026851073535,
      "learning_rate": 3.5529369589103325e-05,
      "loss": 0.0002,
      "step": 6530
    },
    {
      "epoch": 5.081616789739604,
      "grad_norm": 0.020490618812553214,
      "learning_rate": 3.536718529419847e-05,
      "loss": 0.0002,
      "step": 6540
    },
    {
      "epoch": 5.089389817333852,
      "grad_norm": 0.023527533859118526,
      "learning_rate": 3.520516920573034e-05,
      "loss": 0.0002,
      "step": 6550
    },
    {
      "epoch": 5.0971628449281,
      "grad_norm": 0.018702447223960132,
      "learning_rate": 3.5043323186098566e-05,
      "loss": 0.0002,
      "step": 6560
    },
    {
      "epoch": 5.104935872522347,
      "grad_norm": 0.014705563163092392,
      "learning_rate": 3.4881649095747834e-05,
      "loss": 0.0002,
      "step": 6570
    },
    {
      "epoch": 5.112708900116595,
      "grad_norm": 0.024095689249138356,
      "learning_rate": 3.472014879314647e-05,
      "loss": 0.0002,
      "step": 6580
    },
    {
      "epoch": 5.120481927710843,
      "grad_norm": 0.01933876437653428,
      "learning_rate": 3.455882413476509e-05,
      "loss": 0.0002,
      "step": 6590
    },
    {
      "epoch": 5.128254955305091,
      "grad_norm": 0.02632230897239421,
      "learning_rate": 3.439767697505523e-05,
      "loss": 0.0002,
      "step": 6600
    },
    {
      "epoch": 5.136027982899339,
      "grad_norm": 0.0202543121018914,
      "learning_rate": 3.4236709166428084e-05,
      "loss": 0.0002,
      "step": 6610
    },
    {
      "epoch": 5.143801010493587,
      "grad_norm": 0.026753419387336896,
      "learning_rate": 3.407592255923315e-05,
      "loss": 0.0002,
      "step": 6620
    },
    {
      "epoch": 5.151574038087835,
      "grad_norm": 0.02781203015847914,
      "learning_rate": 3.3915319001737e-05,
      "loss": 0.0002,
      "step": 6630
    },
    {
      "epoch": 5.159347065682083,
      "grad_norm": 0.030530224790509916,
      "learning_rate": 3.375490034010202e-05,
      "loss": 0.0002,
      "step": 6640
    },
    {
      "epoch": 5.167120093276331,
      "grad_norm": 0.02288450371118925,
      "learning_rate": 3.3594668418365197e-05,
      "loss": 0.0002,
      "step": 6650
    },
    {
      "epoch": 5.174893120870579,
      "grad_norm": 0.021517896010865727,
      "learning_rate": 3.3434625078416916e-05,
      "loss": 0.0002,
      "step": 6660
    },
    {
      "epoch": 5.182666148464827,
      "grad_norm": 0.023523502439129602,
      "learning_rate": 3.3274772159979755e-05,
      "loss": 0.0002,
      "step": 6670
    },
    {
      "epoch": 5.190439176059075,
      "grad_norm": 0.02688107764784857,
      "learning_rate": 3.311511150058742e-05,
      "loss": 0.0002,
      "step": 6680
    },
    {
      "epoch": 5.198212203653323,
      "grad_norm": 0.02203793061037175,
      "learning_rate": 3.295564493556352e-05,
      "loss": 0.0002,
      "step": 6690
    },
    {
      "epoch": 5.2059852312475705,
      "grad_norm": 0.02352224439669748,
      "learning_rate": 3.279637429800054e-05,
      "loss": 0.0002,
      "step": 6700
    },
    {
      "epoch": 5.2137582588418185,
      "grad_norm": 0.026805358592703707,
      "learning_rate": 3.263730141873876e-05,
      "loss": 0.0002,
      "step": 6710
    },
    {
      "epoch": 5.221531286436067,
      "grad_norm": 0.01249193172360089,
      "learning_rate": 3.2478428126345204e-05,
      "loss": 0.0002,
      "step": 6720
    },
    {
      "epoch": 5.229304314030315,
      "grad_norm": 0.02200380263558912,
      "learning_rate": 3.231975624709255e-05,
      "loss": 0.0002,
      "step": 6730
    },
    {
      "epoch": 5.237077341624563,
      "grad_norm": 0.012099660245927285,
      "learning_rate": 3.2161287604938284e-05,
      "loss": 0.0002,
      "step": 6740
    },
    {
      "epoch": 5.244850369218811,
      "grad_norm": 0.02228157538893601,
      "learning_rate": 3.200302402150361e-05,
      "loss": 0.0002,
      "step": 6750
    },
    {
      "epoch": 5.252623396813059,
      "grad_norm": 0.017294136648550864,
      "learning_rate": 3.184496731605255e-05,
      "loss": 0.0002,
      "step": 6760
    },
    {
      "epoch": 5.260396424407307,
      "grad_norm": 0.017701806669633422,
      "learning_rate": 3.1687119305471044e-05,
      "loss": 0.0002,
      "step": 6770
    },
    {
      "epoch": 5.268169452001555,
      "grad_norm": 0.01607060888847898,
      "learning_rate": 3.152948180424606e-05,
      "loss": 0.0002,
      "step": 6780
    },
    {
      "epoch": 5.275942479595803,
      "grad_norm": 0.026074325285235036,
      "learning_rate": 3.137205662444469e-05,
      "loss": 0.0002,
      "step": 6790
    },
    {
      "epoch": 5.283715507190051,
      "grad_norm": 0.030305490939110877,
      "learning_rate": 3.12148455756934e-05,
      "loss": 0.0002,
      "step": 6800
    },
    {
      "epoch": 5.291488534784299,
      "grad_norm": 0.021364468559777604,
      "learning_rate": 3.1057850465157165e-05,
      "loss": 0.0002,
      "step": 6810
    },
    {
      "epoch": 5.299261562378547,
      "grad_norm": 0.03043435601752709,
      "learning_rate": 3.090107309751874e-05,
      "loss": 0.0002,
      "step": 6820
    },
    {
      "epoch": 5.307034589972794,
      "grad_norm": 0.019796218385760327,
      "learning_rate": 3.074451527495785e-05,
      "loss": 0.0002,
      "step": 6830
    },
    {
      "epoch": 5.314807617567042,
      "grad_norm": 0.01811760620856345,
      "learning_rate": 3.05881787971306e-05,
      "loss": 0.0002,
      "step": 6840
    },
    {
      "epoch": 5.32258064516129,
      "grad_norm": 0.017876293682531665,
      "learning_rate": 3.0432065461148562e-05,
      "loss": 0.0002,
      "step": 6850
    },
    {
      "epoch": 5.330353672755538,
      "grad_norm": 0.020700840254488838,
      "learning_rate": 3.0276177061558398e-05,
      "loss": 0.0002,
      "step": 6860
    },
    {
      "epoch": 5.338126700349786,
      "grad_norm": 0.008647403099733837,
      "learning_rate": 3.0120515390321024e-05,
      "loss": 0.0002,
      "step": 6870
    },
    {
      "epoch": 5.345899727944034,
      "grad_norm": 0.02535643487487269,
      "learning_rate": 2.9965082236791075e-05,
      "loss": 0.0002,
      "step": 6880
    },
    {
      "epoch": 5.353672755538282,
      "grad_norm": 0.019226261794019324,
      "learning_rate": 2.9809879387696382e-05,
      "loss": 0.0002,
      "step": 6890
    },
    {
      "epoch": 5.36144578313253,
      "grad_norm": 0.025829891731141336,
      "learning_rate": 2.9654908627117344e-05,
      "loss": 0.0002,
      "step": 6900
    },
    {
      "epoch": 5.369218810726778,
      "grad_norm": 0.015384295860231387,
      "learning_rate": 2.9500171736466497e-05,
      "loss": 0.0002,
      "step": 6910
    },
    {
      "epoch": 5.376991838321026,
      "grad_norm": 0.024431366522012523,
      "learning_rate": 2.934567049446801e-05,
      "loss": 0.0002,
      "step": 6920
    },
    {
      "epoch": 5.384764865915274,
      "grad_norm": 0.01554894939276322,
      "learning_rate": 2.91914066771372e-05,
      "loss": 0.0002,
      "step": 6930
    },
    {
      "epoch": 5.392537893509522,
      "grad_norm": 0.02388782341896548,
      "learning_rate": 2.9037382057760208e-05,
      "loss": 0.0002,
      "step": 6940
    },
    {
      "epoch": 5.40031092110377,
      "grad_norm": 0.012463665580678317,
      "learning_rate": 2.8883598406873496e-05,
      "loss": 0.0002,
      "step": 6950
    },
    {
      "epoch": 5.4080839486980175,
      "grad_norm": 0.02408095869990002,
      "learning_rate": 2.8730057492243596e-05,
      "loss": 0.0002,
      "step": 6960
    },
    {
      "epoch": 5.4158569762922655,
      "grad_norm": 0.01920432511210979,
      "learning_rate": 2.8576761078846713e-05,
      "loss": 0.0002,
      "step": 6970
    },
    {
      "epoch": 5.4236300038865135,
      "grad_norm": 0.01830892007095771,
      "learning_rate": 2.8423710928848535e-05,
      "loss": 0.0002,
      "step": 6980
    },
    {
      "epoch": 5.4314030314807615,
      "grad_norm": 0.017686611612559185,
      "learning_rate": 2.8270908801583845e-05,
      "loss": 0.0002,
      "step": 6990
    },
    {
      "epoch": 5.43917605907501,
      "grad_norm": 0.0210658153885968,
      "learning_rate": 2.8118356453536444e-05,
      "loss": 0.0002,
      "step": 7000
    },
    {
      "epoch": 5.446949086669258,
      "grad_norm": 0.021931753692395774,
      "learning_rate": 2.79660556383188e-05,
      "loss": 0.0002,
      "step": 7010
    },
    {
      "epoch": 5.454722114263506,
      "grad_norm": 0.021499018501008473,
      "learning_rate": 2.7814008106652012e-05,
      "loss": 0.0002,
      "step": 7020
    },
    {
      "epoch": 5.462495141857754,
      "grad_norm": 0.0168747362936273,
      "learning_rate": 2.7662215606345678e-05,
      "loss": 0.0002,
      "step": 7030
    },
    {
      "epoch": 5.470268169452002,
      "grad_norm": 0.009359646842040159,
      "learning_rate": 2.75106798822777e-05,
      "loss": 0.0002,
      "step": 7040
    },
    {
      "epoch": 5.47804119704625,
      "grad_norm": 0.03309650897710097,
      "learning_rate": 2.735940267637437e-05,
      "loss": 0.0002,
      "step": 7050
    },
    {
      "epoch": 5.485814224640498,
      "grad_norm": 0.01563621784010453,
      "learning_rate": 2.720838572759023e-05,
      "loss": 0.0002,
      "step": 7060
    },
    {
      "epoch": 5.493587252234746,
      "grad_norm": 0.017774938937897788,
      "learning_rate": 2.705763077188813e-05,
      "loss": 0.0001,
      "step": 7070
    },
    {
      "epoch": 5.501360279828994,
      "grad_norm": 0.020360336887821972,
      "learning_rate": 2.6907139542219273e-05,
      "loss": 0.0002,
      "step": 7080
    },
    {
      "epoch": 5.509133307423241,
      "grad_norm": 0.011898631264119943,
      "learning_rate": 2.6756913768503323e-05,
      "loss": 0.0002,
      "step": 7090
    },
    {
      "epoch": 5.516906335017489,
      "grad_norm": 0.02364217964429555,
      "learning_rate": 2.6606955177608438e-05,
      "loss": 0.0002,
      "step": 7100
    },
    {
      "epoch": 5.524679362611737,
      "grad_norm": 0.018077323668190225,
      "learning_rate": 2.645726549333148e-05,
      "loss": 0.0002,
      "step": 7110
    },
    {
      "epoch": 5.532452390205985,
      "grad_norm": 0.017724857691957925,
      "learning_rate": 2.630784643637823e-05,
      "loss": 0.0002,
      "step": 7120
    },
    {
      "epoch": 5.540225417800233,
      "grad_norm": 0.016568436588534154,
      "learning_rate": 2.6158699724343506e-05,
      "loss": 0.0002,
      "step": 7130
    },
    {
      "epoch": 5.547998445394481,
      "grad_norm": 0.01826213795206956,
      "learning_rate": 2.600982707169154e-05,
      "loss": 0.0002,
      "step": 7140
    },
    {
      "epoch": 5.555771472988729,
      "grad_norm": 0.01687117093788665,
      "learning_rate": 2.586123018973614e-05,
      "loss": 0.0001,
      "step": 7150
    },
    {
      "epoch": 5.563544500582977,
      "grad_norm": 0.022543857479375335,
      "learning_rate": 2.5712910786621175e-05,
      "loss": 0.0002,
      "step": 7160
    },
    {
      "epoch": 5.571317528177225,
      "grad_norm": 0.013807162813923518,
      "learning_rate": 2.556487056730078e-05,
      "loss": 0.0002,
      "step": 7170
    },
    {
      "epoch": 5.579090555771473,
      "grad_norm": 0.02180872837058625,
      "learning_rate": 2.5417111233519846e-05,
      "loss": 0.0001,
      "step": 7180
    },
    {
      "epoch": 5.586863583365721,
      "grad_norm": 0.01819012057425971,
      "learning_rate": 2.5269634483794458e-05,
      "loss": 0.0002,
      "step": 7190
    },
    {
      "epoch": 5.594636610959969,
      "grad_norm": 0.009381877368019908,
      "learning_rate": 2.5122442013392333e-05,
      "loss": 0.0001,
      "step": 7200
    },
    {
      "epoch": 5.602409638554217,
      "grad_norm": 0.016509397900761707,
      "learning_rate": 2.4975535514313392e-05,
      "loss": 0.0001,
      "step": 7210
    },
    {
      "epoch": 5.610182666148464,
      "grad_norm": 0.026997335391609513,
      "learning_rate": 2.482891667527021e-05,
      "loss": 0.0001,
      "step": 7220
    },
    {
      "epoch": 5.617955693742712,
      "grad_norm": 0.02283003245119994,
      "learning_rate": 2.4682587181668748e-05,
      "loss": 0.0002,
      "step": 7230
    },
    {
      "epoch": 5.6257287213369604,
      "grad_norm": 0.022705954725808653,
      "learning_rate": 2.4536548715588815e-05,
      "loss": 0.0002,
      "step": 7240
    },
    {
      "epoch": 5.6335017489312085,
      "grad_norm": 0.010299474257551027,
      "learning_rate": 2.439080295576488e-05,
      "loss": 0.0002,
      "step": 7250
    },
    {
      "epoch": 5.6412747765254565,
      "grad_norm": 0.03036140069772998,
      "learning_rate": 2.424535157756667e-05,
      "loss": 0.0002,
      "step": 7260
    },
    {
      "epoch": 5.6490478041197045,
      "grad_norm": 0.01954423321301803,
      "learning_rate": 2.4100196252979938e-05,
      "loss": 0.0002,
      "step": 7270
    },
    {
      "epoch": 5.656820831713953,
      "grad_norm": 0.016784558780200815,
      "learning_rate": 2.3955338650587313e-05,
      "loss": 0.0002,
      "step": 7280
    },
    {
      "epoch": 5.664593859308201,
      "grad_norm": 0.015535557635310981,
      "learning_rate": 2.381078043554901e-05,
      "loss": 0.0001,
      "step": 7290
    },
    {
      "epoch": 5.672366886902449,
      "grad_norm": 0.026046312820072404,
      "learning_rate": 2.3666523269583764e-05,
      "loss": 0.0002,
      "step": 7300
    },
    {
      "epoch": 5.680139914496697,
      "grad_norm": 0.026831352918801052,
      "learning_rate": 2.352256881094967e-05,
      "loss": 0.0002,
      "step": 7310
    },
    {
      "epoch": 5.687912942090945,
      "grad_norm": 0.025880861688391255,
      "learning_rate": 2.3378918714425223e-05,
      "loss": 0.0002,
      "step": 7320
    },
    {
      "epoch": 5.695685969685193,
      "grad_norm": 0.019509071602191973,
      "learning_rate": 2.323557463129016e-05,
      "loss": 0.0001,
      "step": 7330
    },
    {
      "epoch": 5.703458997279441,
      "grad_norm": 0.015762480666792787,
      "learning_rate": 2.3092538209306612e-05,
      "loss": 0.0002,
      "step": 7340
    },
    {
      "epoch": 5.711232024873688,
      "grad_norm": 0.024058697175532318,
      "learning_rate": 2.2949811092700047e-05,
      "loss": 0.0002,
      "step": 7350
    },
    {
      "epoch": 5.719005052467936,
      "grad_norm": 0.02233649819608418,
      "learning_rate": 2.2807394922140417e-05,
      "loss": 0.0001,
      "step": 7360
    },
    {
      "epoch": 5.726778080062184,
      "grad_norm": 0.01727815623468289,
      "learning_rate": 2.2665291334723358e-05,
      "loss": 0.0002,
      "step": 7370
    },
    {
      "epoch": 5.734551107656432,
      "grad_norm": 0.024803814652401755,
      "learning_rate": 2.2523501963951254e-05,
      "loss": 0.0002,
      "step": 7380
    },
    {
      "epoch": 5.74232413525068,
      "grad_norm": 0.02353209152685875,
      "learning_rate": 2.238202843971457e-05,
      "loss": 0.0002,
      "step": 7390
    },
    {
      "epoch": 5.750097162844928,
      "grad_norm": 0.024137451721606546,
      "learning_rate": 2.224087238827303e-05,
      "loss": 0.0002,
      "step": 7400
    },
    {
      "epoch": 5.757870190439176,
      "grad_norm": 0.016644195488546468,
      "learning_rate": 2.210003543223698e-05,
      "loss": 0.0002,
      "step": 7410
    },
    {
      "epoch": 5.765643218033424,
      "grad_norm": 0.02050293736587614,
      "learning_rate": 2.1959519190548678e-05,
      "loss": 0.0002,
      "step": 7420
    },
    {
      "epoch": 5.773416245627672,
      "grad_norm": 0.025110815895328563,
      "learning_rate": 2.1819325278463797e-05,
      "loss": 0.0002,
      "step": 7430
    },
    {
      "epoch": 5.78118927322192,
      "grad_norm": 0.03305569052965538,
      "learning_rate": 2.167945530753272e-05,
      "loss": 0.0002,
      "step": 7440
    },
    {
      "epoch": 5.788962300816168,
      "grad_norm": 0.02203403588654069,
      "learning_rate": 2.1539910885582075e-05,
      "loss": 0.0002,
      "step": 7450
    },
    {
      "epoch": 5.796735328410416,
      "grad_norm": 0.015049480072185014,
      "learning_rate": 2.1400693616696304e-05,
      "loss": 0.0002,
      "step": 7460
    },
    {
      "epoch": 5.804508356004664,
      "grad_norm": 0.00954747311981441,
      "learning_rate": 2.1261805101199123e-05,
      "loss": 0.0002,
      "step": 7470
    },
    {
      "epoch": 5.812281383598911,
      "grad_norm": 0.03030842086619983,
      "learning_rate": 2.112324693563522e-05,
      "loss": 0.0002,
      "step": 7480
    },
    {
      "epoch": 5.820054411193159,
      "grad_norm": 0.020855433610673657,
      "learning_rate": 2.098502071275182e-05,
      "loss": 0.0002,
      "step": 7490
    },
    {
      "epoch": 5.827827438787407,
      "grad_norm": 0.012578647489582854,
      "learning_rate": 2.0847128021480456e-05,
      "loss": 0.0001,
      "step": 7500
    },
    {
      "epoch": 5.835600466381655,
      "grad_norm": 0.01734378952115989,
      "learning_rate": 2.0709570446918636e-05,
      "loss": 0.0001,
      "step": 7510
    },
    {
      "epoch": 5.843373493975903,
      "grad_norm": 0.025733611769194024,
      "learning_rate": 2.0572349570311654e-05,
      "loss": 0.0002,
      "step": 7520
    },
    {
      "epoch": 5.8511465215701515,
      "grad_norm": 0.013574963870773879,
      "learning_rate": 2.043546696903442e-05,
      "loss": 0.0001,
      "step": 7530
    },
    {
      "epoch": 5.8589195491643995,
      "grad_norm": 0.014466066517960154,
      "learning_rate": 2.0298924216573296e-05,
      "loss": 0.0002,
      "step": 7540
    },
    {
      "epoch": 5.8666925767586475,
      "grad_norm": 0.02690419563527895,
      "learning_rate": 2.0162722882508072e-05,
      "loss": 0.0002,
      "step": 7550
    },
    {
      "epoch": 5.8744656043528956,
      "grad_norm": 0.012569768694763188,
      "learning_rate": 2.0026864532493826e-05,
      "loss": 0.0002,
      "step": 7560
    },
    {
      "epoch": 5.882238631947144,
      "grad_norm": 0.01117625827468703,
      "learning_rate": 1.989135072824304e-05,
      "loss": 0.0002,
      "step": 7570
    },
    {
      "epoch": 5.890011659541392,
      "grad_norm": 0.01420653454750281,
      "learning_rate": 1.975618302750754e-05,
      "loss": 0.0002,
      "step": 7580
    },
    {
      "epoch": 5.89778468713564,
      "grad_norm": 0.01664030249310254,
      "learning_rate": 1.962136298406068e-05,
      "loss": 0.0001,
      "step": 7590
    },
    {
      "epoch": 5.905557714729888,
      "grad_norm": 0.010416056552019848,
      "learning_rate": 1.9486892147679415e-05,
      "loss": 0.0001,
      "step": 7600
    },
    {
      "epoch": 5.913330742324135,
      "grad_norm": 0.010236252185840958,
      "learning_rate": 1.9352772064126495e-05,
      "loss": 0.0001,
      "step": 7610
    },
    {
      "epoch": 5.921103769918383,
      "grad_norm": 0.022891362985504354,
      "learning_rate": 1.921900427513277e-05,
      "loss": 0.0001,
      "step": 7620
    },
    {
      "epoch": 5.928876797512631,
      "grad_norm": 0.018247148981555494,
      "learning_rate": 1.9085590318379358e-05,
      "loss": 0.0001,
      "step": 7630
    },
    {
      "epoch": 5.936649825106879,
      "grad_norm": 0.012618202734315895,
      "learning_rate": 1.8952531727480037e-05,
      "loss": 0.0001,
      "step": 7640
    },
    {
      "epoch": 5.944422852701127,
      "grad_norm": 0.008332374313899176,
      "learning_rate": 1.88198300319636e-05,
      "loss": 0.0001,
      "step": 7650
    },
    {
      "epoch": 5.952195880295375,
      "grad_norm": 0.016742534880818557,
      "learning_rate": 1.8687486757256283e-05,
      "loss": 0.0001,
      "step": 7660
    },
    {
      "epoch": 5.959968907889623,
      "grad_norm": 0.012035482209576525,
      "learning_rate": 1.8555503424664207e-05,
      "loss": 0.0001,
      "step": 7670
    },
    {
      "epoch": 5.967741935483871,
      "grad_norm": 0.018437890242042766,
      "learning_rate": 1.842388155135592e-05,
      "loss": 0.0001,
      "step": 7680
    },
    {
      "epoch": 5.975514963078119,
      "grad_norm": 0.01798374959258155,
      "learning_rate": 1.8292622650344926e-05,
      "loss": 0.0001,
      "step": 7690
    },
    {
      "epoch": 5.983287990672367,
      "grad_norm": 0.01256498438647202,
      "learning_rate": 1.816172823047229e-05,
      "loss": 0.0001,
      "step": 7700
    },
    {
      "epoch": 5.991061018266615,
      "grad_norm": 0.01465340353044906,
      "learning_rate": 1.8031199796389364e-05,
      "loss": 0.0001,
      "step": 7710
    },
    {
      "epoch": 5.998834045860863,
      "grad_norm": 0.018440141413721543,
      "learning_rate": 1.7901038848540375e-05,
      "loss": 0.0002,
      "step": 7720
    },
    {
      "epoch": 6.006218422075398,
      "grad_norm": 0.014154448665573296,
      "learning_rate": 1.777124688314529e-05,
      "loss": 0.0001,
      "step": 7730
    },
    {
      "epoch": 6.013991449669646,
      "grad_norm": 0.01630517136248746,
      "learning_rate": 1.7641825392182553e-05,
      "loss": 0.0001,
      "step": 7740
    },
    {
      "epoch": 6.021764477263894,
      "grad_norm": 0.008238287161327415,
      "learning_rate": 1.7512775863371922e-05,
      "loss": 0.0001,
      "step": 7750
    },
    {
      "epoch": 6.029537504858142,
      "grad_norm": 0.010737285243644595,
      "learning_rate": 1.7384099780157405e-05,
      "loss": 0.0001,
      "step": 7760
    },
    {
      "epoch": 6.03731053245239,
      "grad_norm": 0.006110968090325836,
      "learning_rate": 1.7255798621690223e-05,
      "loss": 0.0001,
      "step": 7770
    },
    {
      "epoch": 6.045083560046638,
      "grad_norm": 0.008068188996489705,
      "learning_rate": 1.712787386281176e-05,
      "loss": 0.0001,
      "step": 7780
    },
    {
      "epoch": 6.052856587640886,
      "grad_norm": 0.011762484459678948,
      "learning_rate": 1.7000326974036608e-05,
      "loss": 0.0001,
      "step": 7790
    },
    {
      "epoch": 6.060629615235134,
      "grad_norm": 0.0038725686943943904,
      "learning_rate": 1.687315942153573e-05,
      "loss": 0.0001,
      "step": 7800
    },
    {
      "epoch": 6.0684026428293825,
      "grad_norm": 0.003752871493291431,
      "learning_rate": 1.6746372667119513e-05,
      "loss": 0.0001,
      "step": 7810
    },
    {
      "epoch": 6.07617567042363,
      "grad_norm": 0.012152254910518275,
      "learning_rate": 1.661996816822106e-05,
      "loss": 0.0001,
      "step": 7820
    },
    {
      "epoch": 6.083948698017878,
      "grad_norm": 0.007396604209044004,
      "learning_rate": 1.649394737787934e-05,
      "loss": 0.0001,
      "step": 7830
    },
    {
      "epoch": 6.091721725612126,
      "grad_norm": 0.003773688560183537,
      "learning_rate": 1.6368311744722593e-05,
      "loss": 0.0001,
      "step": 7840
    },
    {
      "epoch": 6.099494753206374,
      "grad_norm": 0.00760880877332909,
      "learning_rate": 1.624306271295156e-05,
      "loss": 0.0001,
      "step": 7850
    },
    {
      "epoch": 6.107267780800622,
      "grad_norm": 0.007019583706987574,
      "learning_rate": 1.6118201722322955e-05,
      "loss": 0.0001,
      "step": 7860
    },
    {
      "epoch": 6.11504080839487,
      "grad_norm": 0.017182162490011563,
      "learning_rate": 1.5993730208132906e-05,
      "loss": 0.0001,
      "step": 7870
    },
    {
      "epoch": 6.122813835989118,
      "grad_norm": 0.009023438251598844,
      "learning_rate": 1.5869649601200474e-05,
      "loss": 0.0001,
      "step": 7880
    },
    {
      "epoch": 6.130586863583366,
      "grad_norm": 0.005789434823568986,
      "learning_rate": 1.5745961327851154e-05,
      "loss": 0.0001,
      "step": 7890
    },
    {
      "epoch": 6.138359891177614,
      "grad_norm": 0.0035168838088950553,
      "learning_rate": 1.562266680990049e-05,
      "loss": 0.0001,
      "step": 7900
    },
    {
      "epoch": 6.146132918771862,
      "grad_norm": 0.005285038482418299,
      "learning_rate": 1.54997674646378e-05,
      "loss": 0.0001,
      "step": 7910
    },
    {
      "epoch": 6.15390594636611,
      "grad_norm": 0.00936827904054838,
      "learning_rate": 1.5377264704809767e-05,
      "loss": 0.0001,
      "step": 7920
    },
    {
      "epoch": 6.161678973960358,
      "grad_norm": 0.010636173324013657,
      "learning_rate": 1.5255159938604341e-05,
      "loss": 0.0001,
      "step": 7930
    },
    {
      "epoch": 6.169452001554606,
      "grad_norm": 0.004301365056986207,
      "learning_rate": 1.5133454569634408e-05,
      "loss": 0.0001,
      "step": 7940
    },
    {
      "epoch": 6.177225029148853,
      "grad_norm": 0.0016892402926352035,
      "learning_rate": 1.5012149996921731e-05,
      "loss": 0.0001,
      "step": 7950
    },
    {
      "epoch": 6.184998056743101,
      "grad_norm": 0.007700944733481,
      "learning_rate": 1.4891247614880904e-05,
      "loss": 0.0001,
      "step": 7960
    },
    {
      "epoch": 6.192771084337349,
      "grad_norm": 0.009622948232765017,
      "learning_rate": 1.4770748813303232e-05,
      "loss": 0.0001,
      "step": 7970
    },
    {
      "epoch": 6.200544111931597,
      "grad_norm": 0.009225724210367993,
      "learning_rate": 1.4650654977340822e-05,
      "loss": 0.0001,
      "step": 7980
    },
    {
      "epoch": 6.208317139525845,
      "grad_norm": 0.003251969174291781,
      "learning_rate": 1.453096748749061e-05,
      "loss": 0.0001,
      "step": 7990
    },
    {
      "epoch": 6.216090167120093,
      "grad_norm": 0.011867588892390629,
      "learning_rate": 1.4411687719578587e-05,
      "loss": 0.0001,
      "step": 8000
    },
    {
      "epoch": 6.223863194714341,
      "grad_norm": 0.006979155829618675,
      "learning_rate": 1.4292817044743834e-05,
      "loss": 0.0001,
      "step": 8010
    },
    {
      "epoch": 6.231636222308589,
      "grad_norm": 0.006004329053956222,
      "learning_rate": 1.417435682942293e-05,
      "loss": 0.0001,
      "step": 8020
    },
    {
      "epoch": 6.239409249902837,
      "grad_norm": 0.002612520030831492,
      "learning_rate": 1.4056308435334076e-05,
      "loss": 0.0001,
      "step": 8030
    },
    {
      "epoch": 6.247182277497085,
      "grad_norm": 0.004605154518579334,
      "learning_rate": 1.39386732194616e-05,
      "loss": 0.0001,
      "step": 8040
    },
    {
      "epoch": 6.254955305091333,
      "grad_norm": 0.008603917840699209,
      "learning_rate": 1.3821452534040236e-05,
      "loss": 0.0001,
      "step": 8050
    },
    {
      "epoch": 6.262728332685581,
      "grad_norm": 0.01092543969237321,
      "learning_rate": 1.3704647726539622e-05,
      "loss": 0.0001,
      "step": 8060
    },
    {
      "epoch": 6.270501360279829,
      "grad_norm": 0.015978506993862916,
      "learning_rate": 1.3588260139648867e-05,
      "loss": 0.0001,
      "step": 8070
    },
    {
      "epoch": 6.2782743878740765,
      "grad_norm": 0.003933668237172168,
      "learning_rate": 1.3472291111261015e-05,
      "loss": 0.0001,
      "step": 8080
    },
    {
      "epoch": 6.286047415468325,
      "grad_norm": 0.002766043387407925,
      "learning_rate": 1.3356741974457737e-05,
      "loss": 0.0001,
      "step": 8090
    },
    {
      "epoch": 6.293820443062573,
      "grad_norm": 0.004084772968868483,
      "learning_rate": 1.3241614057493968e-05,
      "loss": 0.0001,
      "step": 8100
    },
    {
      "epoch": 6.301593470656821,
      "grad_norm": 0.004067241395666171,
      "learning_rate": 1.3126908683782702e-05,
      "loss": 0.0,
      "step": 8110
    },
    {
      "epoch": 6.309366498251069,
      "grad_norm": 0.0032836224715808754,
      "learning_rate": 1.3012627171879672e-05,
      "loss": 0.0001,
      "step": 8120
    },
    {
      "epoch": 6.317139525845317,
      "grad_norm": 0.0021028582135935894,
      "learning_rate": 1.289877083546831e-05,
      "loss": 0.0,
      "step": 8130
    },
    {
      "epoch": 6.324912553439565,
      "grad_norm": 0.008957490957287324,
      "learning_rate": 1.2785340983344567e-05,
      "loss": 0.0001,
      "step": 8140
    },
    {
      "epoch": 6.332685581033813,
      "grad_norm": 0.004074819763986697,
      "learning_rate": 1.2672338919401866e-05,
      "loss": 0.0,
      "step": 8150
    },
    {
      "epoch": 6.340458608628061,
      "grad_norm": 0.0035986009756733845,
      "learning_rate": 1.2559765942616203e-05,
      "loss": 0.0,
      "step": 8160
    },
    {
      "epoch": 6.348231636222309,
      "grad_norm": 0.007318820173072485,
      "learning_rate": 1.244762334703108e-05,
      "loss": 0.0001,
      "step": 8170
    },
    {
      "epoch": 6.356004663816557,
      "grad_norm": 0.0041003130870685455,
      "learning_rate": 1.2335912421742767e-05,
      "loss": 0.0001,
      "step": 8180
    },
    {
      "epoch": 6.363777691410805,
      "grad_norm": 0.005660565441390202,
      "learning_rate": 1.2224634450885365e-05,
      "loss": 0.0001,
      "step": 8190
    },
    {
      "epoch": 6.371550719005053,
      "grad_norm": 0.007133669800279721,
      "learning_rate": 1.2113790713616153e-05,
      "loss": 0.0001,
      "step": 8200
    },
    {
      "epoch": 6.3793237465993,
      "grad_norm": 0.008070698121721053,
      "learning_rate": 1.2003382484100739e-05,
      "loss": 0.0001,
      "step": 8210
    },
    {
      "epoch": 6.387096774193548,
      "grad_norm": 0.0018965359081475795,
      "learning_rate": 1.1893411031498613e-05,
      "loss": 0.0001,
      "step": 8220
    },
    {
      "epoch": 6.394869801787796,
      "grad_norm": 0.009751754720241309,
      "learning_rate": 1.1783877619948369e-05,
      "loss": 0.0001,
      "step": 8230
    },
    {
      "epoch": 6.402642829382044,
      "grad_norm": 0.002028173824220937,
      "learning_rate": 1.1674783508553271e-05,
      "loss": 0.0001,
      "step": 8240
    },
    {
      "epoch": 6.410415856976292,
      "grad_norm": 0.006289133991651987,
      "learning_rate": 1.15661299513668e-05,
      "loss": 0.0001,
      "step": 8250
    },
    {
      "epoch": 6.41818888457054,
      "grad_norm": 0.005308692313120731,
      "learning_rate": 1.1457918197378126e-05,
      "loss": 0.0001,
      "step": 8260
    },
    {
      "epoch": 6.425961912164788,
      "grad_norm": 0.005611865313561971,
      "learning_rate": 1.1350149490497897e-05,
      "loss": 0.0001,
      "step": 8270
    },
    {
      "epoch": 6.433734939759036,
      "grad_norm": 0.004039833269871884,
      "learning_rate": 1.124282506954381e-05,
      "loss": 0.0001,
      "step": 8280
    },
    {
      "epoch": 6.441507967353284,
      "grad_norm": 0.007505209819239008,
      "learning_rate": 1.113594616822648e-05,
      "loss": 0.0,
      "step": 8290
    },
    {
      "epoch": 6.449280994947532,
      "grad_norm": 0.001563481641230725,
      "learning_rate": 1.1029514015135162e-05,
      "loss": 0.0001,
      "step": 8300
    },
    {
      "epoch": 6.45705402254178,
      "grad_norm": 0.002312874435433927,
      "learning_rate": 1.0923529833723689e-05,
      "loss": 0.0,
      "step": 8310
    },
    {
      "epoch": 6.464827050136028,
      "grad_norm": 0.0034526991867424865,
      "learning_rate": 1.0817994842296385e-05,
      "loss": 0.0001,
      "step": 8320
    },
    {
      "epoch": 6.472600077730276,
      "grad_norm": 0.0032560423082492475,
      "learning_rate": 1.0712910253994069e-05,
      "loss": 0.0,
      "step": 8330
    },
    {
      "epoch": 6.4803731053245235,
      "grad_norm": 0.006355142914743098,
      "learning_rate": 1.0608277276780127e-05,
      "loss": 0.0,
      "step": 8340
    },
    {
      "epoch": 6.4881461329187715,
      "grad_norm": 0.005307492445739638,
      "learning_rate": 1.050409711342657e-05,
      "loss": 0.0,
      "step": 8350
    },
    {
      "epoch": 6.4959191605130195,
      "grad_norm": 0.0031667184323832497,
      "learning_rate": 1.0400370961500283e-05,
      "loss": 0.0,
      "step": 8360
    },
    {
      "epoch": 6.503692188107268,
      "grad_norm": 0.0027970490655325245,
      "learning_rate": 1.0297100013349182e-05,
      "loss": 0.0,
      "step": 8370
    },
    {
      "epoch": 6.511465215701516,
      "grad_norm": 0.005480572597677732,
      "learning_rate": 1.0194285456088593e-05,
      "loss": 0.0,
      "step": 8380
    },
    {
      "epoch": 6.519238243295764,
      "grad_norm": 0.0063248411062732995,
      "learning_rate": 1.0091928471587513e-05,
      "loss": 0.0,
      "step": 8390
    },
    {
      "epoch": 6.527011270890012,
      "grad_norm": 0.0040781433873937414,
      "learning_rate": 9.990030236455083e-06,
      "loss": 0.0,
      "step": 8400
    },
    {
      "epoch": 6.53478429848426,
      "grad_norm": 0.003932912475110347,
      "learning_rate": 9.88859192202708e-06,
      "loss": 0.0,
      "step": 8410
    },
    {
      "epoch": 6.542557326078508,
      "grad_norm": 0.001707451855920444,
      "learning_rate": 9.787614694352388e-06,
      "loss": 0.0001,
      "step": 8420
    },
    {
      "epoch": 6.550330353672756,
      "grad_norm": 0.003157999222835672,
      "learning_rate": 9.687099714179632e-06,
      "loss": 0.0001,
      "step": 8430
    },
    {
      "epoch": 6.558103381267004,
      "grad_norm": 0.016342779842721915,
      "learning_rate": 9.587048136943828e-06,
      "loss": 0.0,
      "step": 8440
    },
    {
      "epoch": 6.565876408861252,
      "grad_norm": 0.0045906400045922615,
      "learning_rate": 9.487461112753148e-06,
      "loss": 0.0,
      "step": 8450
    },
    {
      "epoch": 6.573649436455499,
      "grad_norm": 0.003893108928199538,
      "learning_rate": 9.388339786375594e-06,
      "loss": 0.0001,
      "step": 8460
    },
    {
      "epoch": 6.581422464049748,
      "grad_norm": 0.004311636056043432,
      "learning_rate": 9.289685297225952e-06,
      "loss": 0.0001,
      "step": 8470
    },
    {
      "epoch": 6.589195491643995,
      "grad_norm": 0.0032466402334312174,
      "learning_rate": 9.191498779352609e-06,
      "loss": 0.0,
      "step": 8480
    },
    {
      "epoch": 6.596968519238243,
      "grad_norm": 0.0027841865450238526,
      "learning_rate": 9.093781361424558e-06,
      "loss": 0.0001,
      "step": 8490
    },
    {
      "epoch": 6.604741546832491,
      "grad_norm": 0.0011684913430085695,
      "learning_rate": 8.996534166718435e-06,
      "loss": 0.0,
      "step": 8500
    },
    {
      "epoch": 6.612514574426739,
      "grad_norm": 0.01975689513794622,
      "learning_rate": 8.899758313105555e-06,
      "loss": 0.0001,
      "step": 8510
    },
    {
      "epoch": 6.620287602020987,
      "grad_norm": 0.013211843359117672,
      "learning_rate": 8.803454913039139e-06,
      "loss": 0.0,
      "step": 8520
    },
    {
      "epoch": 6.628060629615235,
      "grad_norm": 0.007429743591031423,
      "learning_rate": 8.707625073541448e-06,
      "loss": 0.0,
      "step": 8530
    },
    {
      "epoch": 6.635833657209483,
      "grad_norm": 0.04172674679723699,
      "learning_rate": 8.612269896191095e-06,
      "loss": 0.0,
      "step": 8540
    },
    {
      "epoch": 6.643606684803731,
      "grad_norm": 0.003732737066491989,
      "learning_rate": 8.517390477110381e-06,
      "loss": 0.0,
      "step": 8550
    },
    {
      "epoch": 6.651379712397979,
      "grad_norm": 0.014802518590614539,
      "learning_rate": 8.422987906952724e-06,
      "loss": 0.0001,
      "step": 8560
    },
    {
      "epoch": 6.659152739992227,
      "grad_norm": 0.0014132023539125706,
      "learning_rate": 8.329063270890047e-06,
      "loss": 0.0001,
      "step": 8570
    },
    {
      "epoch": 6.666925767586475,
      "grad_norm": 0.0013745783161368645,
      "learning_rate": 8.235617648600358e-06,
      "loss": 0.0001,
      "step": 8580
    },
    {
      "epoch": 6.674698795180722,
      "grad_norm": 0.014731183812191016,
      "learning_rate": 8.14265211425536e-06,
      "loss": 0.0001,
      "step": 8590
    },
    {
      "epoch": 6.682471822774971,
      "grad_norm": 0.0019968377299820154,
      "learning_rate": 8.050167736508024e-06,
      "loss": 0.0,
      "step": 8600
    },
    {
      "epoch": 6.6902448503692185,
      "grad_norm": 0.008966863100114431,
      "learning_rate": 7.958165578480403e-06,
      "loss": 0.0,
      "step": 8610
    },
    {
      "epoch": 6.6980178779634665,
      "grad_norm": 0.005031042039948286,
      "learning_rate": 7.866646697751306e-06,
      "loss": 0.0,
      "step": 8620
    },
    {
      "epoch": 6.7057909055577145,
      "grad_norm": 0.009316279948875827,
      "learning_rate": 7.775612146344257e-06,
      "loss": 0.0,
      "step": 8630
    },
    {
      "epoch": 6.7135639331519625,
      "grad_norm": 0.0065508539646422816,
      "learning_rate": 7.68506297071529e-06,
      "loss": 0.0001,
      "step": 8640
    },
    {
      "epoch": 6.721336960746211,
      "grad_norm": 0.002793091462219872,
      "learning_rate": 7.595000211740994e-06,
      "loss": 0.0,
      "step": 8650
    },
    {
      "epoch": 6.729109988340459,
      "grad_norm": 0.001690935684424875,
      "learning_rate": 7.505424904706498e-06,
      "loss": 0.0,
      "step": 8660
    },
    {
      "epoch": 6.736883015934707,
      "grad_norm": 0.0026392129261972237,
      "learning_rate": 7.41633807929365e-06,
      "loss": 0.0,
      "step": 8670
    },
    {
      "epoch": 6.744656043528955,
      "grad_norm": 0.0015107039584184377,
      "learning_rate": 7.327740759569079e-06,
      "loss": 0.0,
      "step": 8680
    },
    {
      "epoch": 6.752429071123203,
      "grad_norm": 0.011490478426607262,
      "learning_rate": 7.239633963972475e-06,
      "loss": 0.0,
      "step": 8690
    },
    {
      "epoch": 6.760202098717451,
      "grad_norm": 0.0026040733002839446,
      "learning_rate": 7.152018705304919e-06,
      "loss": 0.0,
      "step": 8700
    },
    {
      "epoch": 6.767975126311699,
      "grad_norm": 0.003190850976401895,
      "learning_rate": 7.06489599071714e-06,
      "loss": 0.0,
      "step": 8710
    },
    {
      "epoch": 6.775748153905946,
      "grad_norm": 0.0018648652816669246,
      "learning_rate": 6.97826682169806e-06,
      "loss": 0.0,
      "step": 8720
    },
    {
      "epoch": 6.783521181500195,
      "grad_norm": 0.01585922533430392,
      "learning_rate": 6.892132194063178e-06,
      "loss": 0.0,
      "step": 8730
    },
    {
      "epoch": 6.791294209094442,
      "grad_norm": 0.002126611266486979,
      "learning_rate": 6.806493097943157e-06,
      "loss": 0.0001,
      "step": 8740
    },
    {
      "epoch": 6.79906723668869,
      "grad_norm": 0.005000504593456693,
      "learning_rate": 6.7213505177724954e-06,
      "loss": 0.0,
      "step": 8750
    },
    {
      "epoch": 6.806840264282938,
      "grad_norm": 0.0012306312131158685,
      "learning_rate": 6.636705432278123e-06,
      "loss": 0.0,
      "step": 8760
    },
    {
      "epoch": 6.814613291877186,
      "grad_norm": 0.00550232667293492,
      "learning_rate": 6.552558814468207e-06,
      "loss": 0.0,
      "step": 8770
    },
    {
      "epoch": 6.822386319471434,
      "grad_norm": 0.0016053685161805658,
      "learning_rate": 6.46891163162095e-06,
      "loss": 0.0,
      "step": 8780
    },
    {
      "epoch": 6.830159347065682,
      "grad_norm": 0.0034012751294817535,
      "learning_rate": 6.385764845273484e-06,
      "loss": 0.0,
      "step": 8790
    },
    {
      "epoch": 6.83793237465993,
      "grad_norm": 0.0029206828232979736,
      "learning_rate": 6.303119411210789e-06,
      "loss": 0.0,
      "step": 8800
    },
    {
      "epoch": 6.845705402254178,
      "grad_norm": 0.0020952973372429834,
      "learning_rate": 6.2209762794547485e-06,
      "loss": 0.0,
      "step": 8810
    },
    {
      "epoch": 6.853478429848426,
      "grad_norm": 0.0008398113591594053,
      "learning_rate": 6.139336394253176e-06,
      "loss": 0.0,
      "step": 8820
    },
    {
      "epoch": 6.861251457442674,
      "grad_norm": 0.0026409770453385242,
      "learning_rate": 6.058200694069016e-06,
      "loss": 0.0,
      "step": 8830
    },
    {
      "epoch": 6.869024485036922,
      "grad_norm": 0.005735042522109382,
      "learning_rate": 5.9775701115695024e-06,
      "loss": 0.0,
      "step": 8840
    },
    {
      "epoch": 6.87679751263117,
      "grad_norm": 0.0031061906194351035,
      "learning_rate": 5.897445573615484e-06,
      "loss": 0.0,
      "step": 8850
    },
    {
      "epoch": 6.884570540225418,
      "grad_norm": 0.005254148047030939,
      "learning_rate": 5.817828001250747e-06,
      "loss": 0.0,
      "step": 8860
    },
    {
      "epoch": 6.892343567819665,
      "grad_norm": 0.004662330435049924,
      "learning_rate": 5.738718309691432e-06,
      "loss": 0.0,
      "step": 8870
    },
    {
      "epoch": 6.900116595413913,
      "grad_norm": 0.007246112307929083,
      "learning_rate": 5.660117408315496e-06,
      "loss": 0.0,
      "step": 8880
    },
    {
      "epoch": 6.9078896230081615,
      "grad_norm": 0.003495408800656057,
      "learning_rate": 5.58202620065229e-06,
      "loss": 0.0,
      "step": 8890
    },
    {
      "epoch": 6.9156626506024095,
      "grad_norm": 0.0015319741981169295,
      "learning_rate": 5.50444558437217e-06,
      "loss": 0.0,
      "step": 8900
    },
    {
      "epoch": 6.9234356781966575,
      "grad_norm": 0.0026193862674153455,
      "learning_rate": 5.427376451276134e-06,
      "loss": 0.0,
      "step": 8910
    },
    {
      "epoch": 6.9312087057909055,
      "grad_norm": 0.0052639172788985856,
      "learning_rate": 5.350819687285635e-06,
      "loss": 0.0,
      "step": 8920
    },
    {
      "epoch": 6.938981733385154,
      "grad_norm": 0.007817778547725699,
      "learning_rate": 5.27477617243235e-06,
      "loss": 0.0,
      "step": 8930
    },
    {
      "epoch": 6.946754760979402,
      "grad_norm": 0.005751662740577324,
      "learning_rate": 5.19924678084806e-06,
      "loss": 0.0001,
      "step": 8940
    },
    {
      "epoch": 6.95452778857365,
      "grad_norm": 0.0039659853654034315,
      "learning_rate": 5.1242323807546675e-06,
      "loss": 0.0,
      "step": 8950
    },
    {
      "epoch": 6.962300816167898,
      "grad_norm": 0.006129164151707489,
      "learning_rate": 5.049733834454123e-06,
      "loss": 0.0,
      "step": 8960
    },
    {
      "epoch": 6.970073843762146,
      "grad_norm": 0.004187805464255788,
      "learning_rate": 4.97575199831859e-06,
      "loss": 0.0,
      "step": 8970
    },
    {
      "epoch": 6.977846871356394,
      "grad_norm": 0.006948302150623346,
      "learning_rate": 4.902287722780558e-06,
      "loss": 0.0,
      "step": 8980
    },
    {
      "epoch": 6.985619898950642,
      "grad_norm": 0.003535808730100431,
      "learning_rate": 4.829341852323071e-06,
      "loss": 0.0,
      "step": 8990
    },
    {
      "epoch": 6.993392926544889,
      "grad_norm": 0.003982561271510016,
      "learning_rate": 4.756915225470027e-06,
      "loss": 0.0,
      "step": 9000
    },
    {
      "epoch": 7.000777302759425,
      "grad_norm": 0.0009609293197381767,
      "learning_rate": 4.68500867477657e-06,
      "loss": 0.0,
      "step": 9010
    },
    {
      "epoch": 7.008550330353673,
      "grad_norm": 0.001321616068003471,
      "learning_rate": 4.613623026819458e-06,
      "loss": 0.0,
      "step": 9020
    },
    {
      "epoch": 7.016323357947921,
      "grad_norm": 0.0010806415038145181,
      "learning_rate": 4.542759102187582e-06,
      "loss": 0.0,
      "step": 9030
    },
    {
      "epoch": 7.024096385542169,
      "grad_norm": 0.0009429836364269017,
      "learning_rate": 4.472417715472593e-06,
      "loss": 0.0,
      "step": 9040
    },
    {
      "epoch": 7.031869413136417,
      "grad_norm": 0.0009939065040656875,
      "learning_rate": 4.402599675259433e-06,
      "loss": 0.0,
      "step": 9050
    },
    {
      "epoch": 7.039642440730665,
      "grad_norm": 0.0011804782677918904,
      "learning_rate": 4.3333057841171475e-06,
      "loss": 0.0,
      "step": 9060
    },
    {
      "epoch": 7.047415468324912,
      "grad_norm": 0.0015427913406341474,
      "learning_rate": 4.264536838589562e-06,
      "loss": 0.0,
      "step": 9070
    },
    {
      "epoch": 7.05518849591916,
      "grad_norm": 0.0008024300071659224,
      "learning_rate": 4.19629362918621e-06,
      "loss": 0.0,
      "step": 9080
    },
    {
      "epoch": 7.062961523513408,
      "grad_norm": 0.0008222894730903661,
      "learning_rate": 4.128576940373185e-06,
      "loss": 0.0,
      "step": 9090
    },
    {
      "epoch": 7.070734551107656,
      "grad_norm": 0.000763975055938749,
      "learning_rate": 4.06138755056415e-06,
      "loss": 0.0,
      "step": 9100
    },
    {
      "epoch": 7.078507578701904,
      "grad_norm": 0.001366761326684155,
      "learning_rate": 3.99472623211139e-06,
      "loss": 0.0,
      "step": 9110
    },
    {
      "epoch": 7.086280606296152,
      "grad_norm": 0.000988952812605041,
      "learning_rate": 3.928593751296916e-06,
      "loss": 0.0,
      "step": 9120
    },
    {
      "epoch": 7.0940536338904,
      "grad_norm": 0.00046864347906276887,
      "learning_rate": 3.862990868323702e-06,
      "loss": 0.0,
      "step": 9130
    },
    {
      "epoch": 7.101826661484648,
      "grad_norm": 0.0008812139530910874,
      "learning_rate": 3.7979183373068716e-06,
      "loss": 0.0,
      "step": 9140
    },
    {
      "epoch": 7.109599689078896,
      "grad_norm": 0.0014316389957675793,
      "learning_rate": 3.7333769062651124e-06,
      "loss": 0.0,
      "step": 9150
    },
    {
      "epoch": 7.117372716673144,
      "grad_norm": 0.0010288852864815841,
      "learning_rate": 3.6693673171120048e-06,
      "loss": 0.0,
      "step": 9160
    },
    {
      "epoch": 7.1251457442673924,
      "grad_norm": 0.0018663118153428742,
      "learning_rate": 3.605890305647547e-06,
      "loss": 0.0,
      "step": 9170
    },
    {
      "epoch": 7.1329187718616405,
      "grad_norm": 0.0008151035343918915,
      "learning_rate": 3.5429466015496625e-06,
      "loss": 0.0,
      "step": 9180
    },
    {
      "epoch": 7.1406917994558885,
      "grad_norm": 0.0007714788308039119,
      "learning_rate": 3.480536928365824e-06,
      "loss": 0.0,
      "step": 9190
    },
    {
      "epoch": 7.148464827050136,
      "grad_norm": 0.0011775048169888774,
      "learning_rate": 3.4186620035047523e-06,
      "loss": 0.0,
      "step": 9200
    },
    {
      "epoch": 7.156237854644384,
      "grad_norm": 0.0007508614916090825,
      "learning_rate": 3.357322538228147e-06,
      "loss": 0.0,
      "step": 9210
    },
    {
      "epoch": 7.164010882238632,
      "grad_norm": 0.0009166415696453144,
      "learning_rate": 3.2965192376425036e-06,
      "loss": 0.0,
      "step": 9220
    },
    {
      "epoch": 7.17178390983288,
      "grad_norm": 0.0008892857558514102,
      "learning_rate": 3.236252800691042e-06,
      "loss": 0.0,
      "step": 9230
    },
    {
      "epoch": 7.179556937427128,
      "grad_norm": 0.0007251113869919698,
      "learning_rate": 3.1765239201456464e-06,
      "loss": 0.0,
      "step": 9240
    },
    {
      "epoch": 7.187329965021376,
      "grad_norm": 0.0007932361018754159,
      "learning_rate": 3.1173332825989053e-06,
      "loss": 0.0,
      "step": 9250
    },
    {
      "epoch": 7.195102992615624,
      "grad_norm": 0.0005987293215761424,
      "learning_rate": 3.0586815684562396e-06,
      "loss": 0.0,
      "step": 9260
    },
    {
      "epoch": 7.202876020209872,
      "grad_norm": 0.005876406174100428,
      "learning_rate": 3.0005694519280313e-06,
      "loss": 0.0,
      "step": 9270
    },
    {
      "epoch": 7.21064904780412,
      "grad_norm": 0.001071700200113881,
      "learning_rate": 2.942997601021924e-06,
      "loss": 0.0,
      "step": 9280
    },
    {
      "epoch": 7.218422075398368,
      "grad_norm": 0.0006788726901146583,
      "learning_rate": 2.885966677535129e-06,
      "loss": 0.0,
      "step": 9290
    },
    {
      "epoch": 7.226195102992616,
      "grad_norm": 0.0015747045362846824,
      "learning_rate": 2.8294773370468096e-06,
      "loss": 0.0,
      "step": 9300
    },
    {
      "epoch": 7.233968130586864,
      "grad_norm": 0.001261900753208969,
      "learning_rate": 2.7735302289105534e-06,
      "loss": 0.0,
      "step": 9310
    },
    {
      "epoch": 7.241741158181112,
      "grad_norm": 0.0006076637650185912,
      "learning_rate": 2.7181259962468895e-06,
      "loss": 0.0,
      "step": 9320
    },
    {
      "epoch": 7.249514185775359,
      "grad_norm": 0.0006071127917645978,
      "learning_rate": 2.6632652759359446e-06,
      "loss": 0.0,
      "step": 9330
    },
    {
      "epoch": 7.257287213369607,
      "grad_norm": 0.0006803286275209111,
      "learning_rate": 2.608948698610031e-06,
      "loss": 0.0,
      "step": 9340
    },
    {
      "epoch": 7.265060240963855,
      "grad_norm": 0.014754699022412117,
      "learning_rate": 2.555176888646521e-06,
      "loss": 0.0,
      "step": 9350
    },
    {
      "epoch": 7.272833268558103,
      "grad_norm": 0.0006467000064079362,
      "learning_rate": 2.5019504641605574e-06,
      "loss": 0.0,
      "step": 9360
    },
    {
      "epoch": 7.280606296152351,
      "grad_norm": 0.003055465673203673,
      "learning_rate": 2.4492700369980027e-06,
      "loss": 0.0,
      "step": 9370
    },
    {
      "epoch": 7.288379323746599,
      "grad_norm": 0.000661621262906463,
      "learning_rate": 2.3971362127284123e-06,
      "loss": 0.0,
      "step": 9380
    },
    {
      "epoch": 7.296152351340847,
      "grad_norm": 0.0007217089188821212,
      "learning_rate": 2.3455495906380355e-06,
      "loss": 0.0,
      "step": 9390
    },
    {
      "epoch": 7.303925378935095,
      "grad_norm": 0.001273093668952842,
      "learning_rate": 2.2945107637229735e-06,
      "loss": 0.0,
      "step": 9400
    },
    {
      "epoch": 7.311698406529343,
      "grad_norm": 0.0006448463302102047,
      "learning_rate": 2.244020318682316e-06,
      "loss": 0.0,
      "step": 9410
    },
    {
      "epoch": 7.319471434123591,
      "grad_norm": 0.0025607485190233866,
      "learning_rate": 2.194078835911434e-06,
      "loss": 0.0,
      "step": 9420
    },
    {
      "epoch": 7.327244461717839,
      "grad_norm": 0.0007104289013335628,
      "learning_rate": 2.14468688949529e-06,
      "loss": 0.0,
      "step": 9430
    },
    {
      "epoch": 7.335017489312087,
      "grad_norm": 0.0014926481663371015,
      "learning_rate": 2.0958450472018386e-06,
      "loss": 0.0,
      "step": 9440
    },
    {
      "epoch": 7.342790516906335,
      "grad_norm": 0.0006249576492496036,
      "learning_rate": 2.0475538704755047e-06,
      "loss": 0.0,
      "step": 9450
    },
    {
      "epoch": 7.350563544500583,
      "grad_norm": 0.0007092574662016617,
      "learning_rate": 1.999813914430726e-06,
      "loss": 0.0,
      "step": 9460
    },
    {
      "epoch": 7.358336572094831,
      "grad_norm": 0.0006863601177385994,
      "learning_rate": 1.952625727845586e-06,
      "loss": 0.0,
      "step": 9470
    },
    {
      "epoch": 7.366109599689079,
      "grad_norm": 0.0005344771815092314,
      "learning_rate": 1.905989853155482e-06,
      "loss": 0.0,
      "step": 9480
    },
    {
      "epoch": 7.373882627283327,
      "grad_norm": 0.0007443702712828649,
      "learning_rate": 1.8599068264469056e-06,
      "loss": 0.0,
      "step": 9490
    },
    {
      "epoch": 7.381655654877575,
      "grad_norm": 0.0007225809923155882,
      "learning_rate": 1.8143771774512763e-06,
      "loss": 0.0,
      "step": 9500
    },
    {
      "epoch": 7.389428682471823,
      "grad_norm": 0.0007316837553084923,
      "learning_rate": 1.7694014295388639e-06,
      "loss": 0.0,
      "step": 9510
    },
    {
      "epoch": 7.397201710066071,
      "grad_norm": 0.0006634138570056594,
      "learning_rate": 1.7249800997127364e-06,
      "loss": 0.0,
      "step": 9520
    },
    {
      "epoch": 7.404974737660319,
      "grad_norm": 0.0006573555275101724,
      "learning_rate": 1.6811136986028541e-06,
      "loss": 0.0,
      "step": 9530
    },
    {
      "epoch": 7.412747765254567,
      "grad_norm": 0.0006325910073364984,
      "learning_rate": 1.637802730460203e-06,
      "loss": 0.0,
      "step": 9540
    },
    {
      "epoch": 7.420520792848815,
      "grad_norm": 0.0007427869870241537,
      "learning_rate": 1.5950476931509528e-06,
      "loss": 0.0,
      "step": 9550
    },
    {
      "epoch": 7.428293820443063,
      "grad_norm": 0.0013388323315585203,
      "learning_rate": 1.5528490781507866e-06,
      "loss": 0.0,
      "step": 9560
    },
    {
      "epoch": 7.436066848037311,
      "grad_norm": 0.000937354008295726,
      "learning_rate": 1.5112073705392083e-06,
      "loss": 0.0,
      "step": 9570
    },
    {
      "epoch": 7.443839875631559,
      "grad_norm": 0.0008368458606253965,
      "learning_rate": 1.47012304899401e-06,
      "loss": 0.0,
      "step": 9580
    },
    {
      "epoch": 7.451612903225806,
      "grad_norm": 0.0007712267436743975,
      "learning_rate": 1.4295965857857252e-06,
      "loss": 0.0,
      "step": 9590
    },
    {
      "epoch": 7.459385930820054,
      "grad_norm": 0.0007003631871530728,
      "learning_rate": 1.3896284467722398e-06,
      "loss": 0.0,
      "step": 9600
    },
    {
      "epoch": 7.467158958414302,
      "grad_norm": 0.000755508016749,
      "learning_rate": 1.350219091393401e-06,
      "loss": 0.0,
      "step": 9610
    },
    {
      "epoch": 7.47493198600855,
      "grad_norm": 0.00063407265478013,
      "learning_rate": 1.3113689726657551e-06,
      "loss": 0.0,
      "step": 9620
    },
    {
      "epoch": 7.482705013602798,
      "grad_norm": 0.0006961670628009819,
      "learning_rate": 1.2730785371773468e-06,
      "loss": 0.0,
      "step": 9630
    },
    {
      "epoch": 7.490478041197046,
      "grad_norm": 0.004755026275026027,
      "learning_rate": 1.2353482250825666e-06,
      "loss": 0.0,
      "step": 9640
    },
    {
      "epoch": 7.498251068791294,
      "grad_norm": 0.0005789546939814981,
      "learning_rate": 1.1981784700971166e-06,
      "loss": 0.0,
      "step": 9650
    },
    {
      "epoch": 7.506024096385542,
      "grad_norm": 0.0007563653056163621,
      "learning_rate": 1.1615696994929926e-06,
      "loss": 0.0,
      "step": 9660
    },
    {
      "epoch": 7.51379712397979,
      "grad_norm": 0.0007690652504465334,
      "learning_rate": 1.1255223340935984e-06,
      "loss": 0.0,
      "step": 9670
    },
    {
      "epoch": 7.521570151574038,
      "grad_norm": 0.0007468296752969306,
      "learning_rate": 1.090036788268889e-06,
      "loss": 0.0,
      "step": 9680
    },
    {
      "epoch": 7.529343179168286,
      "grad_norm": 0.0018290788411334745,
      "learning_rate": 1.0551134699306352e-06,
      "loss": 0.0,
      "step": 9690
    },
    {
      "epoch": 7.537116206762534,
      "grad_norm": 0.0006343546661807168,
      "learning_rate": 1.0207527805277005e-06,
      "loss": 0.0,
      "step": 9700
    },
    {
      "epoch": 7.544889234356782,
      "grad_norm": 0.0006132020872730862,
      "learning_rate": 9.869551150414603e-07,
      "loss": 0.0,
      "step": 9710
    },
    {
      "epoch": 7.5526622619510295,
      "grad_norm": 0.0011455663378770302,
      "learning_rate": 9.537208619812276e-07,
      "loss": 0.0,
      "step": 9720
    },
    {
      "epoch": 7.5604352895452775,
      "grad_norm": 0.0006575603990109085,
      "learning_rate": 9.210504033797973e-07,
      "loss": 0.0,
      "step": 9730
    },
    {
      "epoch": 7.568208317139526,
      "grad_norm": 0.0010450832230964296,
      "learning_rate": 8.889441147890976e-07,
      "loss": 0.0,
      "step": 9740
    },
    {
      "epoch": 7.575981344733774,
      "grad_norm": 0.0009646636645513483,
      "learning_rate": 8.574023652757945e-07,
      "loss": 0.0,
      "step": 9750
    },
    {
      "epoch": 7.583754372328022,
      "grad_norm": 0.0008660750382369493,
      "learning_rate": 8.264255174171176e-07,
      "loss": 0.0,
      "step": 9760
    },
    {
      "epoch": 7.59152739992227,
      "grad_norm": 0.0015961051255042773,
      "learning_rate": 7.960139272966571e-07,
      "loss": 0.0,
      "step": 9770
    },
    {
      "epoch": 7.599300427516518,
      "grad_norm": 0.00251617459850879,
      "learning_rate": 7.661679445002734e-07,
      "loss": 0.0,
      "step": 9780
    },
    {
      "epoch": 7.607073455110766,
      "grad_norm": 0.0006961561365848857,
      "learning_rate": 7.368879121120942e-07,
      "loss": 0.0,
      "step": 9790
    },
    {
      "epoch": 7.614846482705014,
      "grad_norm": 0.0006414356290991828,
      "learning_rate": 7.081741667105623e-07,
      "loss": 0.0,
      "step": 9800
    },
    {
      "epoch": 7.622619510299262,
      "grad_norm": 0.0005591683955217916,
      "learning_rate": 6.800270383645502e-07,
      "loss": 0.0,
      "step": 9810
    },
    {
      "epoch": 7.63039253789351,
      "grad_norm": 0.0011100734317080255,
      "learning_rate": 6.524468506295956e-07,
      "loss": 0.0,
      "step": 9820
    },
    {
      "epoch": 7.638165565487758,
      "grad_norm": 0.0006949260560080673,
      "learning_rate": 6.254339205441718e-07,
      "loss": 0.0,
      "step": 9830
    },
    {
      "epoch": 7.645938593082006,
      "grad_norm": 0.0007122854818339492,
      "learning_rate": 5.989885586260124e-07,
      "loss": 0.0,
      "step": 9840
    },
    {
      "epoch": 7.653711620676253,
      "grad_norm": 0.000910131387406276,
      "learning_rate": 5.731110688686037e-07,
      "loss": 0.0,
      "step": 9850
    },
    {
      "epoch": 7.661484648270501,
      "grad_norm": 0.0008539901633964141,
      "learning_rate": 5.478017487376196e-07,
      "loss": 0.0,
      "step": 9860
    },
    {
      "epoch": 7.669257675864749,
      "grad_norm": 0.000775153585296624,
      "learning_rate": 5.230608891675703e-07,
      "loss": 0.0,
      "step": 9870
    },
    {
      "epoch": 7.677030703458997,
      "grad_norm": 0.001089864655833332,
      "learning_rate": 4.988887745584037e-07,
      "loss": 0.0,
      "step": 9880
    },
    {
      "epoch": 7.684803731053245,
      "grad_norm": 0.0006897662285900077,
      "learning_rate": 4.7528568277226405e-07,
      "loss": 0.0,
      "step": 9890
    },
    {
      "epoch": 7.692576758647493,
      "grad_norm": 0.0007728670382980491,
      "learning_rate": 4.522518851302893e-07,
      "loss": 0.0,
      "step": 9900
    },
    {
      "epoch": 7.700349786241741,
      "grad_norm": 0.0012928967522706268,
      "learning_rate": 4.297876464094963e-07,
      "loss": 0.0,
      "step": 9910
    },
    {
      "epoch": 7.708122813835989,
      "grad_norm": 0.0006740642016424453,
      "learning_rate": 4.0789322483973934e-07,
      "loss": 0.0,
      "step": 9920
    },
    {
      "epoch": 7.715895841430237,
      "grad_norm": 0.0009131941458940286,
      "learning_rate": 3.8656887210072877e-07,
      "loss": 0.0,
      "step": 9930
    },
    {
      "epoch": 7.723668869024485,
      "grad_norm": 0.0007118523685031266,
      "learning_rate": 3.658148333191558e-07,
      "loss": 0.0,
      "step": 9940
    },
    {
      "epoch": 7.731441896618733,
      "grad_norm": 0.0010680468680305678,
      "learning_rate": 3.4563134706585567e-07,
      "loss": 0.0,
      "step": 9950
    },
    {
      "epoch": 7.739214924212981,
      "grad_norm": 0.0007379792860212389,
      "learning_rate": 3.260186453530878e-07,
      "loss": 0.0,
      "step": 9960
    },
    {
      "epoch": 7.746987951807229,
      "grad_norm": 0.0010117437888497545,
      "learning_rate": 3.069769536318434e-07,
      "loss": 0.0,
      "step": 9970
    },
    {
      "epoch": 7.7547609794014765,
      "grad_norm": 0.0006022784332183378,
      "learning_rate": 2.8850649078927537e-07,
      "loss": 0.0,
      "step": 9980
    },
    {
      "epoch": 7.7625340069957245,
      "grad_norm": 0.0009423335079612556,
      "learning_rate": 2.7060746914617797e-07,
      "loss": 0.0,
      "step": 9990
    },
    {
      "epoch": 7.7703070345899725,
      "grad_norm": 0.0007549035863112267,
      "learning_rate": 2.5328009445452214e-07,
      "loss": 0.0,
      "step": 10000
    },
    {
      "epoch": 7.7780800621842205,
      "grad_norm": 0.000705741476973914,
      "learning_rate": 2.3652456589512983e-07,
      "loss": 0.0,
      "step": 10010
    },
    {
      "epoch": 7.785853089778469,
      "grad_norm": 0.0004977161547313549,
      "learning_rate": 2.203410760753588e-07,
      "loss": 0.0,
      "step": 10020
    },
    {
      "epoch": 7.793626117372717,
      "grad_norm": 0.0018148168364492605,
      "learning_rate": 2.0472981102688804e-07,
      "loss": 0.0,
      "step": 10030
    },
    {
      "epoch": 7.801399144966965,
      "grad_norm": 0.0007960327080494255,
      "learning_rate": 1.8969095020359152e-07,
      "loss": 0.0,
      "step": 10040
    },
    {
      "epoch": 7.809172172561213,
      "grad_norm": 0.0011660029117197294,
      "learning_rate": 1.7522466647946767e-07,
      "loss": 0.0,
      "step": 10050
    },
    {
      "epoch": 7.816945200155461,
      "grad_norm": 0.0017712978431274203,
      "learning_rate": 1.6133112614665768e-07,
      "loss": 0.0,
      "step": 10060
    },
    {
      "epoch": 7.824718227749709,
      "grad_norm": 0.0008953199566172221,
      "learning_rate": 1.4801048891352475e-07,
      "loss": 0.0,
      "step": 10070
    },
    {
      "epoch": 7.832491255343957,
      "grad_norm": 0.0005717037797458049,
      "learning_rate": 1.3526290790282226e-07,
      "loss": 0.0,
      "step": 10080
    },
    {
      "epoch": 7.840264282938205,
      "grad_norm": 0.0009013575543165316,
      "learning_rate": 1.230885296499451e-07,
      "loss": 0.0,
      "step": 10090
    },
    {
      "epoch": 7.848037310532453,
      "grad_norm": 0.0009845714931132692,
      "learning_rate": 1.1148749410122006e-07,
      "loss": 0.0,
      "step": 10100
    },
    {
      "epoch": 7.8558103381267,
      "grad_norm": 0.0005365771511423384,
      "learning_rate": 1.0045993461232361e-07,
      "loss": 0.0,
      "step": 10110
    },
    {
      "epoch": 7.863583365720948,
      "grad_norm": 0.0008342912865935574,
      "learning_rate": 9.000597794673327e-08,
      "loss": 0.0,
      "step": 10120
    },
    {
      "epoch": 7.871356393315196,
      "grad_norm": 0.0009793381129950054,
      "learning_rate": 8.012574427427866e-08,
      "loss": 0.0,
      "step": 10130
    },
    {
      "epoch": 7.879129420909444,
      "grad_norm": 0.0004974651436117407,
      "learning_rate": 7.081934716974826e-08,
      "loss": 0.0,
      "step": 10140
    },
    {
      "epoch": 7.886902448503692,
      "grad_norm": 0.0008665030119875182,
      "learning_rate": 6.208689361159592e-08,
      "loss": 0.0,
      "step": 10150
    },
    {
      "epoch": 7.89467547609794,
      "grad_norm": 0.0008040519021531375,
      "learning_rate": 5.39284839807086e-08,
      "loss": 0.0,
      "step": 10160
    },
    {
      "epoch": 7.902448503692188,
      "grad_norm": 0.0008762517121375106,
      "learning_rate": 4.634421205925166e-08,
      "loss": 0.0,
      "step": 10170
    },
    {
      "epoch": 7.910221531286436,
      "grad_norm": 0.0008338424459509029,
      "learning_rate": 3.933416502958642e-08,
      "loss": 0.0,
      "step": 10180
    },
    {
      "epoch": 7.917994558880684,
      "grad_norm": 0.0010816418000464051,
      "learning_rate": 3.289842347327099e-08,
      "loss": 0.0,
      "step": 10190
    },
    {
      "epoch": 7.925767586474932,
      "grad_norm": 0.0006126906101453305,
      "learning_rate": 2.7037061370133177e-08,
      "loss": 0.0,
      "step": 10200
    },
    {
      "epoch": 7.93354061406918,
      "grad_norm": 0.0008430054854899276,
      "learning_rate": 2.175014609743231e-08,
      "loss": 0.0,
      "step": 10210
    },
    {
      "epoch": 7.941313641663428,
      "grad_norm": 0.0010888024494912743,
      "learning_rate": 1.703773842905987e-08,
      "loss": 0.0,
      "step": 10220
    },
    {
      "epoch": 7.949086669257676,
      "grad_norm": 0.0007663843785077151,
      "learning_rate": 1.2899892534862235e-08,
      "loss": 0.0,
      "step": 10230
    },
    {
      "epoch": 7.956859696851923,
      "grad_norm": 0.0013532995925952163,
      "learning_rate": 9.33665598000788e-09,
      "loss": 0.0,
      "step": 10240
    },
    {
      "epoch": 7.964632724446171,
      "grad_norm": 0.001154671434988171,
      "learning_rate": 6.348069724448902e-09,
      "loss": 0.0,
      "step": 10250
    },
    {
      "epoch": 7.9724057520404195,
      "grad_norm": 0.000713262523541586,
      "learning_rate": 3.934168122432524e-09,
      "loss": 0.0,
      "step": 10260
    },
    {
      "epoch": 7.9801787796346675,
      "grad_norm": 0.000596312733843517,
      "learning_rate": 2.094978922129176e-09,
      "loss": 0.0,
      "step": 10270
    },
    {
      "epoch": 7.9879518072289155,
      "grad_norm": 0.0006974444375209213,
      "learning_rate": 8.305232653049722e-10,
      "loss": 0.0,
      "step": 10280
    },
    {
      "epoch": 7.9957248348231635,
      "grad_norm": 0.0007989153159201642,
      "learning_rate": 1.408156870608135e-10,
      "loss": 0.0,
      "step": 10290
    }
  ],
  "logging_steps": 10,
  "max_steps": 10296,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 8,
  "save_steps": 1800,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 759446402236416.0,
  "train_batch_size": 128,
  "trial_name": null,
  "trial_params": null
}