diff --git "a/multi_student_3/checkpoint-9000/trainer_state.json" "b/multi_student_3/checkpoint-9000/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/multi_student_3/checkpoint-9000/trainer_state.json"
@@ -0,0 +1,6334 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 6.993392926544889,
+  "eval_steps": 500,
+  "global_step": 9000,
+  "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
+    }
+  ],
+  "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": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 663822063894528.0,
+  "train_batch_size": 128,
+  "trial_name": null,
+  "trial_params": null
+}