diff --git "a/protT5-xl/cls/trainer_state.json" "b/protT5-xl/cls/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/protT5-xl/cls/trainer_state.json"
@@ -0,0 +1,19453 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 2.9780769269465655,
+  "eval_steps": 100,
+  "global_step": 27750,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "grad_norm": 1.136018991470337,
+      "learning_rate": 2e-05,
+      "loss": 9.0135,
+      "step": 1
+    },
+    {
+      "epoch": 0.0,
+      "grad_norm": 1.1227364540100098,
+      "learning_rate": 2e-05,
+      "loss": 8.9064,
+      "step": 10
+    },
+    {
+      "epoch": 0.0,
+      "grad_norm": 1.0535037517547607,
+      "learning_rate": 2e-05,
+      "loss": 8.8687,
+      "step": 20
+    },
+    {
+      "epoch": 0.0,
+      "grad_norm": 1.1087745428085327,
+      "learning_rate": 2e-05,
+      "loss": 8.8735,
+      "step": 30
+    },
+    {
+      "epoch": 0.0,
+      "grad_norm": 1.071560263633728,
+      "learning_rate": 2e-05,
+      "loss": 8.8029,
+      "step": 40
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 1.0942052602767944,
+      "learning_rate": 2e-05,
+      "loss": 8.8298,
+      "step": 50
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 1.0125452280044556,
+      "learning_rate": 2e-05,
+      "loss": 8.8414,
+      "step": 60
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 1.0418239831924438,
+      "learning_rate": 2e-05,
+      "loss": 8.8539,
+      "step": 70
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.9567419290542603,
+      "learning_rate": 2e-05,
+      "loss": 8.8778,
+      "step": 80
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 1.1631693840026855,
+      "learning_rate": 2e-05,
+      "loss": 8.8898,
+      "step": 90
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 1.0369843244552612,
+      "learning_rate": 2e-05,
+      "loss": 8.9118,
+      "step": 100
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 1.0023720264434814,
+      "learning_rate": 2e-05,
+      "loss": 8.9138,
+      "step": 110
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 1.0158734321594238,
+      "learning_rate": 2e-05,
+      "loss": 8.8885,
+      "step": 120
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 1.0896579027175903,
+      "learning_rate": 2e-05,
+      "loss": 8.9376,
+      "step": 130
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 1.1101820468902588,
+      "learning_rate": 2e-05,
+      "loss": 8.8095,
+      "step": 140
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 1.1257922649383545,
+      "learning_rate": 2e-05,
+      "loss": 8.8255,
+      "step": 150
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 1.0258537530899048,
+      "learning_rate": 2e-05,
+      "loss": 8.8541,
+      "step": 160
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 1.013587236404419,
+      "learning_rate": 2e-05,
+      "loss": 8.8277,
+      "step": 170
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.9561657905578613,
+      "learning_rate": 2e-05,
+      "loss": 8.9145,
+      "step": 180
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 1.0177621841430664,
+      "learning_rate": 2e-05,
+      "loss": 8.8461,
+      "step": 190
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 1.0207089185714722,
+      "learning_rate": 2e-05,
+      "loss": 8.8517,
+      "step": 200
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.9225640296936035,
+      "learning_rate": 2e-05,
+      "loss": 8.8535,
+      "step": 210
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 1.093534231185913,
+      "learning_rate": 2e-05,
+      "loss": 8.8394,
+      "step": 220
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 1.159427523612976,
+      "learning_rate": 2e-05,
+      "loss": 8.8724,
+      "step": 230
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 1.0807198286056519,
+      "learning_rate": 2e-05,
+      "loss": 8.8235,
+      "step": 240
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.9698489904403687,
+      "learning_rate": 2e-05,
+      "loss": 8.8356,
+      "step": 250
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.9817768931388855,
+      "learning_rate": 2e-05,
+      "loss": 8.8376,
+      "step": 260
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.9796839952468872,
+      "learning_rate": 2e-05,
+      "loss": 8.8173,
+      "step": 270
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 1.111617088317871,
+      "learning_rate": 2e-05,
+      "loss": 8.8214,
+      "step": 280
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 1.1058907508850098,
+      "learning_rate": 2e-05,
+      "loss": 8.8017,
+      "step": 290
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 1.0654045343399048,
+      "learning_rate": 2e-05,
+      "loss": 8.9059,
+      "step": 300
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 1.0694764852523804,
+      "learning_rate": 2e-05,
+      "loss": 8.8323,
+      "step": 310
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.8366920351982117,
+      "learning_rate": 2e-05,
+      "loss": 8.7871,
+      "step": 320
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.9657266736030579,
+      "learning_rate": 2e-05,
+      "loss": 8.8233,
+      "step": 330
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.9996088743209839,
+      "learning_rate": 2e-05,
+      "loss": 8.8576,
+      "step": 340
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 1.144029974937439,
+      "learning_rate": 2e-05,
+      "loss": 8.814,
+      "step": 350
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.9891992807388306,
+      "learning_rate": 2e-05,
+      "loss": 8.8079,
+      "step": 360
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 1.0989946126937866,
+      "learning_rate": 2e-05,
+      "loss": 8.8286,
+      "step": 370
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 1.1845835447311401,
+      "learning_rate": 2e-05,
+      "loss": 8.7808,
+      "step": 380
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.9469441175460815,
+      "learning_rate": 2e-05,
+      "loss": 8.8763,
+      "step": 390
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 1.0189638137817383,
+      "learning_rate": 2e-05,
+      "loss": 8.7818,
+      "step": 400
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 1.025667428970337,
+      "learning_rate": 2e-05,
+      "loss": 8.8173,
+      "step": 410
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 1.0881075859069824,
+      "learning_rate": 2e-05,
+      "loss": 8.7967,
+      "step": 420
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 1.0212202072143555,
+      "learning_rate": 2e-05,
+      "loss": 8.822,
+      "step": 430
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 1.1483595371246338,
+      "learning_rate": 2e-05,
+      "loss": 8.7831,
+      "step": 440
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 1.0635871887207031,
+      "learning_rate": 2e-05,
+      "loss": 8.7962,
+      "step": 450
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.9455224871635437,
+      "learning_rate": 2e-05,
+      "loss": 8.7993,
+      "step": 460
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 1.0709508657455444,
+      "learning_rate": 2e-05,
+      "loss": 8.7879,
+      "step": 470
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 1.347190499305725,
+      "learning_rate": 2e-05,
+      "loss": 8.7962,
+      "step": 480
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 1.7278716564178467,
+      "learning_rate": 2e-05,
+      "loss": 8.7436,
+      "step": 490
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 1.226672887802124,
+      "learning_rate": 2e-05,
+      "loss": 8.6874,
+      "step": 500
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 1.1453171968460083,
+      "learning_rate": 2e-05,
+      "loss": 8.6818,
+      "step": 510
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 1.0818852186203003,
+      "learning_rate": 2e-05,
+      "loss": 8.7125,
+      "step": 520
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.9080603718757629,
+      "learning_rate": 2e-05,
+      "loss": 8.6988,
+      "step": 530
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.9160961508750916,
+      "learning_rate": 2e-05,
+      "loss": 8.714,
+      "step": 540
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.8492730259895325,
+      "learning_rate": 2e-05,
+      "loss": 8.7106,
+      "step": 550
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.8488112688064575,
+      "learning_rate": 2e-05,
+      "loss": 8.6911,
+      "step": 560
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.7244778275489807,
+      "learning_rate": 2e-05,
+      "loss": 8.7017,
+      "step": 570
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.8408392071723938,
+      "learning_rate": 2e-05,
+      "loss": 8.7013,
+      "step": 580
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.7833953499794006,
+      "learning_rate": 2e-05,
+      "loss": 8.6981,
+      "step": 590
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.7584900259971619,
+      "learning_rate": 2e-05,
+      "loss": 8.7093,
+      "step": 600
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.7876583337783813,
+      "learning_rate": 2e-05,
+      "loss": 8.6756,
+      "step": 610
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.7830280065536499,
+      "learning_rate": 2e-05,
+      "loss": 8.6869,
+      "step": 620
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.7406884431838989,
+      "learning_rate": 2e-05,
+      "loss": 8.7008,
+      "step": 630
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.7786482572555542,
+      "learning_rate": 2e-05,
+      "loss": 8.6805,
+      "step": 640
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.719681978225708,
+      "learning_rate": 2e-05,
+      "loss": 8.7015,
+      "step": 650
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.7766366004943848,
+      "learning_rate": 2e-05,
+      "loss": 8.6827,
+      "step": 660
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.7706497311592102,
+      "learning_rate": 2e-05,
+      "loss": 8.6885,
+      "step": 670
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.7037889957427979,
+      "learning_rate": 2e-05,
+      "loss": 8.6774,
+      "step": 680
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.7186391949653625,
+      "learning_rate": 2e-05,
+      "loss": 8.7127,
+      "step": 690
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.7355700731277466,
+      "learning_rate": 2e-05,
+      "loss": 8.6795,
+      "step": 700
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.700326681137085,
+      "learning_rate": 2e-05,
+      "loss": 8.6922,
+      "step": 710
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.8040889501571655,
+      "learning_rate": 2e-05,
+      "loss": 8.6719,
+      "step": 720
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.788592517375946,
+      "learning_rate": 2e-05,
+      "loss": 8.6802,
+      "step": 730
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.761727511882782,
+      "learning_rate": 2e-05,
+      "loss": 8.6678,
+      "step": 740
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.7505279183387756,
+      "learning_rate": 2e-05,
+      "loss": 8.6624,
+      "step": 750
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.783634603023529,
+      "learning_rate": 2e-05,
+      "loss": 8.6616,
+      "step": 760
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.9118529558181763,
+      "learning_rate": 2e-05,
+      "loss": 8.6787,
+      "step": 770
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.804297924041748,
+      "learning_rate": 2e-05,
+      "loss": 8.71,
+      "step": 780
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.7445526123046875,
+      "learning_rate": 2e-05,
+      "loss": 8.6899,
+      "step": 790
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.6968070268630981,
+      "learning_rate": 2e-05,
+      "loss": 8.6828,
+      "step": 800
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.6915277242660522,
+      "learning_rate": 2e-05,
+      "loss": 8.6759,
+      "step": 810
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.7348951697349548,
+      "learning_rate": 2e-05,
+      "loss": 8.7039,
+      "step": 820
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.7289301753044128,
+      "learning_rate": 2e-05,
+      "loss": 8.6708,
+      "step": 830
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.8944723010063171,
+      "learning_rate": 2e-05,
+      "loss": 8.6669,
+      "step": 840
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.8326644897460938,
+      "learning_rate": 2e-05,
+      "loss": 8.6934,
+      "step": 850
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.7058615684509277,
+      "learning_rate": 2e-05,
+      "loss": 8.6698,
+      "step": 860
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.7978131771087646,
+      "learning_rate": 2e-05,
+      "loss": 8.658,
+      "step": 870
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.879593551158905,
+      "learning_rate": 2e-05,
+      "loss": 8.6714,
+      "step": 880
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.7924543023109436,
+      "learning_rate": 2e-05,
+      "loss": 8.6909,
+      "step": 890
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.9171891808509827,
+      "learning_rate": 2e-05,
+      "loss": 8.6639,
+      "step": 900
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.9398960471153259,
+      "learning_rate": 2e-05,
+      "loss": 8.6629,
+      "step": 910
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.8466478586196899,
+      "learning_rate": 2e-05,
+      "loss": 8.6507,
+      "step": 920
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.9456720352172852,
+      "learning_rate": 2e-05,
+      "loss": 8.6528,
+      "step": 930
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.9696431159973145,
+      "learning_rate": 2e-05,
+      "loss": 8.6733,
+      "step": 940
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.9205406308174133,
+      "learning_rate": 2e-05,
+      "loss": 8.6327,
+      "step": 950
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 1.1785640716552734,
+      "learning_rate": 2e-05,
+      "loss": 8.6539,
+      "step": 960
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 1.0798403024673462,
+      "learning_rate": 2e-05,
+      "loss": 8.6376,
+      "step": 970
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 1.0533510446548462,
+      "learning_rate": 2e-05,
+      "loss": 8.656,
+      "step": 980
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 2.1338257789611816,
+      "learning_rate": 2e-05,
+      "loss": 8.657,
+      "step": 990
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 1.0870320796966553,
+      "learning_rate": 2e-05,
+      "loss": 8.6389,
+      "step": 1000
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.9737922549247742,
+      "learning_rate": 2e-05,
+      "loss": 8.6461,
+      "step": 1010
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 1.1025416851043701,
+      "learning_rate": 2e-05,
+      "loss": 8.6578,
+      "step": 1020
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 1.1098887920379639,
+      "learning_rate": 2e-05,
+      "loss": 8.6572,
+      "step": 1030
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 1.2702275514602661,
+      "learning_rate": 2e-05,
+      "loss": 8.6574,
+      "step": 1040
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 1.348967432975769,
+      "learning_rate": 2e-05,
+      "loss": 8.6312,
+      "step": 1050
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 1.1451374292373657,
+      "learning_rate": 2e-05,
+      "loss": 8.6465,
+      "step": 1060
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 1.2983516454696655,
+      "learning_rate": 2e-05,
+      "loss": 8.6246,
+      "step": 1070
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 1.433199405670166,
+      "learning_rate": 2e-05,
+      "loss": 8.629,
+      "step": 1080
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 1.293949842453003,
+      "learning_rate": 2e-05,
+      "loss": 8.6396,
+      "step": 1090
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 1.3439069986343384,
+      "learning_rate": 2e-05,
+      "loss": 8.6298,
+      "step": 1100
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 1.724899411201477,
+      "learning_rate": 2e-05,
+      "loss": 8.6446,
+      "step": 1110
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 1.308885097503662,
+      "learning_rate": 2e-05,
+      "loss": 8.6069,
+      "step": 1120
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 1.4800187349319458,
+      "learning_rate": 2e-05,
+      "loss": 8.6004,
+      "step": 1130
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 1.5355836153030396,
+      "learning_rate": 2e-05,
+      "loss": 8.6054,
+      "step": 1140
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 1.6553165912628174,
+      "learning_rate": 2e-05,
+      "loss": 8.6223,
+      "step": 1150
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 1.7274365425109863,
+      "learning_rate": 2e-05,
+      "loss": 8.6071,
+      "step": 1160
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 1.5869295597076416,
+      "learning_rate": 2e-05,
+      "loss": 8.6436,
+      "step": 1170
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 1.7578085660934448,
+      "learning_rate": 2e-05,
+      "loss": 8.6444,
+      "step": 1180
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 1.6309038400650024,
+      "learning_rate": 2e-05,
+      "loss": 8.6073,
+      "step": 1190
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 1.432953953742981,
+      "learning_rate": 2e-05,
+      "loss": 8.6033,
+      "step": 1200
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 1.6652449369430542,
+      "learning_rate": 2e-05,
+      "loss": 8.6333,
+      "step": 1210
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 1.4232771396636963,
+      "learning_rate": 2e-05,
+      "loss": 8.6317,
+      "step": 1220
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 1.5350998640060425,
+      "learning_rate": 2e-05,
+      "loss": 8.5981,
+      "step": 1230
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 1.745444655418396,
+      "learning_rate": 2e-05,
+      "loss": 8.5722,
+      "step": 1240
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 1.822112798690796,
+      "learning_rate": 2e-05,
+      "loss": 8.5626,
+      "step": 1250
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 1.6986939907073975,
+      "learning_rate": 2e-05,
+      "loss": 8.6084,
+      "step": 1260
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 1.8796597719192505,
+      "learning_rate": 2e-05,
+      "loss": 8.5963,
+      "step": 1270
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 2.2829110622406006,
+      "learning_rate": 2e-05,
+      "loss": 8.5664,
+      "step": 1280
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 1.9445936679840088,
+      "learning_rate": 2e-05,
+      "loss": 8.569,
+      "step": 1290
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 2.08772611618042,
+      "learning_rate": 2e-05,
+      "loss": 8.5576,
+      "step": 1300
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 1.9294919967651367,
+      "learning_rate": 2e-05,
+      "loss": 8.5666,
+      "step": 1310
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 2.047457456588745,
+      "learning_rate": 2e-05,
+      "loss": 8.5522,
+      "step": 1320
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 2.21968412399292,
+      "learning_rate": 2e-05,
+      "loss": 8.5841,
+      "step": 1330
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 2.196680784225464,
+      "learning_rate": 2e-05,
+      "loss": 8.5964,
+      "step": 1340
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 2.5114238262176514,
+      "learning_rate": 2e-05,
+      "loss": 8.528,
+      "step": 1350
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 2.3431286811828613,
+      "learning_rate": 2e-05,
+      "loss": 8.5671,
+      "step": 1360
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 2.395749807357788,
+      "learning_rate": 2e-05,
+      "loss": 8.5312,
+      "step": 1370
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 2.301731586456299,
+      "learning_rate": 2e-05,
+      "loss": 8.546,
+      "step": 1380
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 2.391402244567871,
+      "learning_rate": 2e-05,
+      "loss": 8.5312,
+      "step": 1390
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 2.8734259605407715,
+      "learning_rate": 2e-05,
+      "loss": 8.571,
+      "step": 1400
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 3.005164861679077,
+      "learning_rate": 2e-05,
+      "loss": 8.5753,
+      "step": 1410
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 2.664741039276123,
+      "learning_rate": 2e-05,
+      "loss": 8.5518,
+      "step": 1420
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 2.7693045139312744,
+      "learning_rate": 2e-05,
+      "loss": 8.561,
+      "step": 1430
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 4.607264995574951,
+      "learning_rate": 2e-05,
+      "loss": 8.5225,
+      "step": 1440
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 2.626310110092163,
+      "learning_rate": 2e-05,
+      "loss": 8.5276,
+      "step": 1450
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 2.8724071979522705,
+      "learning_rate": 2e-05,
+      "loss": 8.5512,
+      "step": 1460
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 2.796130657196045,
+      "learning_rate": 2e-05,
+      "loss": 8.5355,
+      "step": 1470
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 2.9112415313720703,
+      "learning_rate": 2e-05,
+      "loss": 8.5762,
+      "step": 1480
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 2.7517759799957275,
+      "learning_rate": 2e-05,
+      "loss": 8.5049,
+      "step": 1490
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 3.045830726623535,
+      "learning_rate": 2e-05,
+      "loss": 8.4853,
+      "step": 1500
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 2.7766242027282715,
+      "learning_rate": 2e-05,
+      "loss": 8.436,
+      "step": 1510
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 3.078204870223999,
+      "learning_rate": 2e-05,
+      "loss": 8.5458,
+      "step": 1520
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 3.1019699573516846,
+      "learning_rate": 2e-05,
+      "loss": 8.4897,
+      "step": 1530
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 3.0855705738067627,
+      "learning_rate": 2e-05,
+      "loss": 8.4909,
+      "step": 1540
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 3.281068801879883,
+      "learning_rate": 2e-05,
+      "loss": 8.4713,
+      "step": 1550
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 3.6366384029388428,
+      "learning_rate": 2e-05,
+      "loss": 8.5097,
+      "step": 1560
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 2.557335615158081,
+      "learning_rate": 2e-05,
+      "loss": 8.4919,
+      "step": 1570
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 3.8144819736480713,
+      "learning_rate": 2e-05,
+      "loss": 8.4716,
+      "step": 1580
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 3.34610652923584,
+      "learning_rate": 2e-05,
+      "loss": 8.5151,
+      "step": 1590
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 4.035268306732178,
+      "learning_rate": 2e-05,
+      "loss": 8.497,
+      "step": 1600
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 3.0076894760131836,
+      "learning_rate": 2e-05,
+      "loss": 8.4698,
+      "step": 1610
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 3.495839834213257,
+      "learning_rate": 2e-05,
+      "loss": 8.4671,
+      "step": 1620
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 3.6816112995147705,
+      "learning_rate": 2e-05,
+      "loss": 8.4196,
+      "step": 1630
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 3.4001786708831787,
+      "learning_rate": 2e-05,
+      "loss": 8.4845,
+      "step": 1640
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 6.0509796142578125,
+      "learning_rate": 2e-05,
+      "loss": 8.4746,
+      "step": 1650
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 3.88356876373291,
+      "learning_rate": 2e-05,
+      "loss": 8.482,
+      "step": 1660
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 3.796940326690674,
+      "learning_rate": 2e-05,
+      "loss": 8.474,
+      "step": 1670
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 3.8238089084625244,
+      "learning_rate": 2e-05,
+      "loss": 8.4961,
+      "step": 1680
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 3.561783790588379,
+      "learning_rate": 2e-05,
+      "loss": 8.4295,
+      "step": 1690
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 4.091838836669922,
+      "learning_rate": 2e-05,
+      "loss": 8.4619,
+      "step": 1700
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 3.5598275661468506,
+      "learning_rate": 2e-05,
+      "loss": 8.4828,
+      "step": 1710
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 3.771270751953125,
+      "learning_rate": 2e-05,
+      "loss": 8.4257,
+      "step": 1720
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 4.134157180786133,
+      "learning_rate": 2e-05,
+      "loss": 8.4453,
+      "step": 1730
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 3.934246063232422,
+      "learning_rate": 2e-05,
+      "loss": 8.4283,
+      "step": 1740
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 4.219895362854004,
+      "learning_rate": 2e-05,
+      "loss": 8.4008,
+      "step": 1750
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 3.7355568408966064,
+      "learning_rate": 2e-05,
+      "loss": 8.4099,
+      "step": 1760
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 4.669015407562256,
+      "learning_rate": 2e-05,
+      "loss": 8.4224,
+      "step": 1770
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 3.5804080963134766,
+      "learning_rate": 2e-05,
+      "loss": 8.4457,
+      "step": 1780
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 4.3701252937316895,
+      "learning_rate": 2e-05,
+      "loss": 8.4046,
+      "step": 1790
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 4.461766242980957,
+      "learning_rate": 2e-05,
+      "loss": 8.3569,
+      "step": 1800
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 3.9159204959869385,
+      "learning_rate": 2e-05,
+      "loss": 8.4344,
+      "step": 1810
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 4.945244789123535,
+      "learning_rate": 2e-05,
+      "loss": 8.3794,
+      "step": 1820
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 4.186527729034424,
+      "learning_rate": 2e-05,
+      "loss": 8.4015,
+      "step": 1830
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 4.171858310699463,
+      "learning_rate": 2e-05,
+      "loss": 8.3895,
+      "step": 1840
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 4.725074768066406,
+      "learning_rate": 2e-05,
+      "loss": 8.4572,
+      "step": 1850
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 4.5031232833862305,
+      "learning_rate": 2e-05,
+      "loss": 8.3882,
+      "step": 1860
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 4.348865985870361,
+      "learning_rate": 2e-05,
+      "loss": 8.3875,
+      "step": 1870
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 5.031768798828125,
+      "learning_rate": 2e-05,
+      "loss": 8.3908,
+      "step": 1880
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 4.828806400299072,
+      "learning_rate": 2e-05,
+      "loss": 8.3963,
+      "step": 1890
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 5.013701915740967,
+      "learning_rate": 2e-05,
+      "loss": 8.4011,
+      "step": 1900
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 4.196092128753662,
+      "learning_rate": 2e-05,
+      "loss": 8.3408,
+      "step": 1910
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 4.565566539764404,
+      "learning_rate": 2e-05,
+      "loss": 8.377,
+      "step": 1920
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 4.385209560394287,
+      "learning_rate": 2e-05,
+      "loss": 8.3827,
+      "step": 1930
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 4.397568702697754,
+      "learning_rate": 2e-05,
+      "loss": 8.3643,
+      "step": 1940
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 5.016266345977783,
+      "learning_rate": 2e-05,
+      "loss": 8.4147,
+      "step": 1950
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 5.0581560134887695,
+      "learning_rate": 2e-05,
+      "loss": 8.4038,
+      "step": 1960
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 4.776040077209473,
+      "learning_rate": 2e-05,
+      "loss": 8.3231,
+      "step": 1970
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 4.691780090332031,
+      "learning_rate": 2e-05,
+      "loss": 8.392,
+      "step": 1980
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 5.166812896728516,
+      "learning_rate": 2e-05,
+      "loss": 8.3811,
+      "step": 1990
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 4.771829128265381,
+      "learning_rate": 2e-05,
+      "loss": 8.438,
+      "step": 2000
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 5.228240013122559,
+      "learning_rate": 2e-05,
+      "loss": 8.32,
+      "step": 2010
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 5.145392417907715,
+      "learning_rate": 2e-05,
+      "loss": 8.3434,
+      "step": 2020
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 5.7475128173828125,
+      "learning_rate": 2e-05,
+      "loss": 8.3574,
+      "step": 2030
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 6.590136528015137,
+      "learning_rate": 2e-05,
+      "loss": 8.4262,
+      "step": 2040
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 5.059377670288086,
+      "learning_rate": 2e-05,
+      "loss": 8.3681,
+      "step": 2050
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 5.0697431564331055,
+      "learning_rate": 2e-05,
+      "loss": 8.3111,
+      "step": 2060
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 5.3987603187561035,
+      "learning_rate": 2e-05,
+      "loss": 8.2845,
+      "step": 2070
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 5.548408031463623,
+      "learning_rate": 2e-05,
+      "loss": 8.32,
+      "step": 2080
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 5.673028469085693,
+      "learning_rate": 2e-05,
+      "loss": 8.3013,
+      "step": 2090
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 5.980769634246826,
+      "learning_rate": 2e-05,
+      "loss": 8.3804,
+      "step": 2100
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 6.499025344848633,
+      "learning_rate": 2e-05,
+      "loss": 8.3212,
+      "step": 2110
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 5.225012302398682,
+      "learning_rate": 2e-05,
+      "loss": 8.3142,
+      "step": 2120
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 5.8247880935668945,
+      "learning_rate": 2e-05,
+      "loss": 8.32,
+      "step": 2130
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 5.744804859161377,
+      "learning_rate": 2e-05,
+      "loss": 8.2623,
+      "step": 2140
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 6.2565460205078125,
+      "learning_rate": 2e-05,
+      "loss": 8.3126,
+      "step": 2150
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 5.727328777313232,
+      "learning_rate": 2e-05,
+      "loss": 8.2696,
+      "step": 2160
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 5.600546836853027,
+      "learning_rate": 2e-05,
+      "loss": 8.3122,
+      "step": 2170
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 4.877242565155029,
+      "learning_rate": 2e-05,
+      "loss": 8.3546,
+      "step": 2180
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 4.916385173797607,
+      "learning_rate": 2e-05,
+      "loss": 8.2456,
+      "step": 2190
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 5.928600788116455,
+      "learning_rate": 2e-05,
+      "loss": 8.3788,
+      "step": 2200
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 5.893412113189697,
+      "learning_rate": 2e-05,
+      "loss": 8.2796,
+      "step": 2210
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 5.70504093170166,
+      "learning_rate": 2e-05,
+      "loss": 8.3153,
+      "step": 2220
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 5.469809055328369,
+      "learning_rate": 2e-05,
+      "loss": 8.2944,
+      "step": 2230
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 4.875389099121094,
+      "learning_rate": 2e-05,
+      "loss": 8.32,
+      "step": 2240
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 6.230598449707031,
+      "learning_rate": 2e-05,
+      "loss": 8.2594,
+      "step": 2250
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 5.352538108825684,
+      "learning_rate": 2e-05,
+      "loss": 8.289,
+      "step": 2260
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 6.170636177062988,
+      "learning_rate": 2e-05,
+      "loss": 8.2375,
+      "step": 2270
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 5.64437198638916,
+      "learning_rate": 2e-05,
+      "loss": 8.3219,
+      "step": 2280
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 5.307595252990723,
+      "learning_rate": 2e-05,
+      "loss": 8.259,
+      "step": 2290
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 7.112575531005859,
+      "learning_rate": 2e-05,
+      "loss": 8.2954,
+      "step": 2300
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 5.717131614685059,
+      "learning_rate": 2e-05,
+      "loss": 8.2867,
+      "step": 2310
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 6.140155792236328,
+      "learning_rate": 2e-05,
+      "loss": 8.3464,
+      "step": 2320
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 6.6766676902771,
+      "learning_rate": 2e-05,
+      "loss": 8.26,
+      "step": 2330
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 6.319295406341553,
+      "learning_rate": 2e-05,
+      "loss": 8.3517,
+      "step": 2340
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 5.670939922332764,
+      "learning_rate": 2e-05,
+      "loss": 8.2351,
+      "step": 2350
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 6.272426128387451,
+      "learning_rate": 2e-05,
+      "loss": 8.259,
+      "step": 2360
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 6.085712432861328,
+      "learning_rate": 2e-05,
+      "loss": 8.2685,
+      "step": 2370
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 6.603766441345215,
+      "learning_rate": 2e-05,
+      "loss": 8.2987,
+      "step": 2380
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 6.202737331390381,
+      "learning_rate": 2e-05,
+      "loss": 8.2242,
+      "step": 2390
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 5.555804252624512,
+      "learning_rate": 2e-05,
+      "loss": 8.3002,
+      "step": 2400
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 6.579737186431885,
+      "learning_rate": 2e-05,
+      "loss": 8.2731,
+      "step": 2410
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 6.484910488128662,
+      "learning_rate": 2e-05,
+      "loss": 8.2018,
+      "step": 2420
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 6.738310813903809,
+      "learning_rate": 2e-05,
+      "loss": 8.3376,
+      "step": 2430
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 6.857169151306152,
+      "learning_rate": 2e-05,
+      "loss": 8.247,
+      "step": 2440
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 6.32554817199707,
+      "learning_rate": 2e-05,
+      "loss": 8.2371,
+      "step": 2450
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 7.523972511291504,
+      "learning_rate": 2e-05,
+      "loss": 8.2244,
+      "step": 2460
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 7.119365692138672,
+      "learning_rate": 2e-05,
+      "loss": 8.2792,
+      "step": 2470
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 5.997178554534912,
+      "learning_rate": 2e-05,
+      "loss": 8.2552,
+      "step": 2480
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 6.755461692810059,
+      "learning_rate": 2e-05,
+      "loss": 8.2107,
+      "step": 2490
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 6.772032260894775,
+      "learning_rate": 2e-05,
+      "loss": 8.2653,
+      "step": 2500
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 6.8230743408203125,
+      "learning_rate": 2e-05,
+      "loss": 8.2384,
+      "step": 2510
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 7.548639297485352,
+      "learning_rate": 2e-05,
+      "loss": 8.3176,
+      "step": 2520
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 6.754918098449707,
+      "learning_rate": 2e-05,
+      "loss": 8.2187,
+      "step": 2530
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 6.019648551940918,
+      "learning_rate": 2e-05,
+      "loss": 8.2169,
+      "step": 2540
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 6.9078826904296875,
+      "learning_rate": 2e-05,
+      "loss": 8.2428,
+      "step": 2550
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 7.788626194000244,
+      "learning_rate": 2e-05,
+      "loss": 8.2701,
+      "step": 2560
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 7.226101875305176,
+      "learning_rate": 2e-05,
+      "loss": 8.2713,
+      "step": 2570
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 7.735256671905518,
+      "learning_rate": 2e-05,
+      "loss": 8.2804,
+      "step": 2580
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 7.1968207359313965,
+      "learning_rate": 2e-05,
+      "loss": 8.1624,
+      "step": 2590
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 7.534297466278076,
+      "learning_rate": 2e-05,
+      "loss": 8.2226,
+      "step": 2600
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 7.119368076324463,
+      "learning_rate": 2e-05,
+      "loss": 8.2076,
+      "step": 2610
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 7.478950023651123,
+      "learning_rate": 2e-05,
+      "loss": 8.187,
+      "step": 2620
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 7.206047534942627,
+      "learning_rate": 2e-05,
+      "loss": 8.2793,
+      "step": 2630
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 7.570435047149658,
+      "learning_rate": 2e-05,
+      "loss": 8.1833,
+      "step": 2640
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 7.585190296173096,
+      "learning_rate": 2e-05,
+      "loss": 8.2172,
+      "step": 2650
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 6.506697654724121,
+      "learning_rate": 2e-05,
+      "loss": 8.2377,
+      "step": 2660
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 6.698332786560059,
+      "learning_rate": 2e-05,
+      "loss": 8.1993,
+      "step": 2670
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 7.1197614669799805,
+      "learning_rate": 2e-05,
+      "loss": 8.2362,
+      "step": 2680
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 7.6868815422058105,
+      "learning_rate": 2e-05,
+      "loss": 8.2545,
+      "step": 2690
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 6.924311637878418,
+      "learning_rate": 2e-05,
+      "loss": 8.2546,
+      "step": 2700
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 7.5660200119018555,
+      "learning_rate": 2e-05,
+      "loss": 8.229,
+      "step": 2710
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 6.53151798248291,
+      "learning_rate": 2e-05,
+      "loss": 8.2206,
+      "step": 2720
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 8.109471321105957,
+      "learning_rate": 2e-05,
+      "loss": 8.1522,
+      "step": 2730
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 6.929211139678955,
+      "learning_rate": 2e-05,
+      "loss": 8.2357,
+      "step": 2740
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 13.111730575561523,
+      "learning_rate": 2e-05,
+      "loss": 8.2102,
+      "step": 2750
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 7.416685104370117,
+      "learning_rate": 2e-05,
+      "loss": 8.1316,
+      "step": 2760
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 7.254404067993164,
+      "learning_rate": 2e-05,
+      "loss": 8.2509,
+      "step": 2770
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 7.444552898406982,
+      "learning_rate": 2e-05,
+      "loss": 8.2459,
+      "step": 2780
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 8.362957954406738,
+      "learning_rate": 2e-05,
+      "loss": 8.206,
+      "step": 2790
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 7.651193618774414,
+      "learning_rate": 2e-05,
+      "loss": 8.0888,
+      "step": 2800
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 8.819253921508789,
+      "learning_rate": 2e-05,
+      "loss": 8.1295,
+      "step": 2810
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 8.072683334350586,
+      "learning_rate": 2e-05,
+      "loss": 8.2398,
+      "step": 2820
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 6.802022457122803,
+      "learning_rate": 2e-05,
+      "loss": 8.2294,
+      "step": 2830
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 7.837174892425537,
+      "learning_rate": 2e-05,
+      "loss": 8.1206,
+      "step": 2840
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 8.092991828918457,
+      "learning_rate": 2e-05,
+      "loss": 8.195,
+      "step": 2850
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 8.1737699508667,
+      "learning_rate": 2e-05,
+      "loss": 8.2092,
+      "step": 2860
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 8.332792282104492,
+      "learning_rate": 2e-05,
+      "loss": 8.1275,
+      "step": 2870
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 8.164402961730957,
+      "learning_rate": 2e-05,
+      "loss": 8.2281,
+      "step": 2880
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 8.91853141784668,
+      "learning_rate": 2e-05,
+      "loss": 8.2366,
+      "step": 2890
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 8.273011207580566,
+      "learning_rate": 2e-05,
+      "loss": 8.187,
+      "step": 2900
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 9.031209945678711,
+      "learning_rate": 2e-05,
+      "loss": 8.1091,
+      "step": 2910
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 8.436809539794922,
+      "learning_rate": 2e-05,
+      "loss": 8.1667,
+      "step": 2920
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 9.466180801391602,
+      "learning_rate": 2e-05,
+      "loss": 8.1035,
+      "step": 2930
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 9.363399505615234,
+      "learning_rate": 2e-05,
+      "loss": 8.1313,
+      "step": 2940
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 8.18155574798584,
+      "learning_rate": 2e-05,
+      "loss": 8.1461,
+      "step": 2950
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 7.7825493812561035,
+      "learning_rate": 2e-05,
+      "loss": 8.157,
+      "step": 2960
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 10.097177505493164,
+      "learning_rate": 2e-05,
+      "loss": 8.2136,
+      "step": 2970
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 8.945893287658691,
+      "learning_rate": 2e-05,
+      "loss": 8.2288,
+      "step": 2980
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 7.5212297439575195,
+      "learning_rate": 2e-05,
+      "loss": 8.2055,
+      "step": 2990
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 8.03127384185791,
+      "learning_rate": 2e-05,
+      "loss": 8.157,
+      "step": 3000
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 7.418972969055176,
+      "learning_rate": 2e-05,
+      "loss": 8.2125,
+      "step": 3010
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 7.739259719848633,
+      "learning_rate": 2e-05,
+      "loss": 8.1299,
+      "step": 3020
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 9.243000984191895,
+      "learning_rate": 2e-05,
+      "loss": 8.1828,
+      "step": 3030
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 9.442794799804688,
+      "learning_rate": 2e-05,
+      "loss": 8.1276,
+      "step": 3040
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 7.8897294998168945,
+      "learning_rate": 2e-05,
+      "loss": 8.1823,
+      "step": 3050
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 8.448980331420898,
+      "learning_rate": 2e-05,
+      "loss": 8.1804,
+      "step": 3060
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 7.404087066650391,
+      "learning_rate": 2e-05,
+      "loss": 8.129,
+      "step": 3070
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 8.939870834350586,
+      "learning_rate": 2e-05,
+      "loss": 8.1669,
+      "step": 3080
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 8.8375244140625,
+      "learning_rate": 2e-05,
+      "loss": 8.1444,
+      "step": 3090
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 8.917304039001465,
+      "learning_rate": 2e-05,
+      "loss": 8.1399,
+      "step": 3100
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 7.979014873504639,
+      "learning_rate": 2e-05,
+      "loss": 8.1577,
+      "step": 3110
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 7.688670635223389,
+      "learning_rate": 2e-05,
+      "loss": 8.1443,
+      "step": 3120
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 8.688896179199219,
+      "learning_rate": 2e-05,
+      "loss": 8.1667,
+      "step": 3130
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 8.550789833068848,
+      "learning_rate": 2e-05,
+      "loss": 8.147,
+      "step": 3140
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 8.966559410095215,
+      "learning_rate": 2e-05,
+      "loss": 8.1567,
+      "step": 3150
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 8.216341018676758,
+      "learning_rate": 2e-05,
+      "loss": 7.9819,
+      "step": 3160
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 8.326035499572754,
+      "learning_rate": 2e-05,
+      "loss": 8.0558,
+      "step": 3170
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 9.354870796203613,
+      "learning_rate": 2e-05,
+      "loss": 8.1656,
+      "step": 3180
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 9.584141731262207,
+      "learning_rate": 2e-05,
+      "loss": 8.1,
+      "step": 3190
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 9.282711029052734,
+      "learning_rate": 2e-05,
+      "loss": 8.0608,
+      "step": 3200
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 8.27320671081543,
+      "learning_rate": 2e-05,
+      "loss": 8.1636,
+      "step": 3210
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 9.02051830291748,
+      "learning_rate": 2e-05,
+      "loss": 8.099,
+      "step": 3220
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 9.247579574584961,
+      "learning_rate": 2e-05,
+      "loss": 8.1235,
+      "step": 3230
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 11.166683197021484,
+      "learning_rate": 2e-05,
+      "loss": 8.0758,
+      "step": 3240
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 9.137251853942871,
+      "learning_rate": 2e-05,
+      "loss": 8.1371,
+      "step": 3250
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 9.355645179748535,
+      "learning_rate": 2e-05,
+      "loss": 7.9876,
+      "step": 3260
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 9.31998348236084,
+      "learning_rate": 2e-05,
+      "loss": 8.0867,
+      "step": 3270
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 9.193875312805176,
+      "learning_rate": 2e-05,
+      "loss": 8.1544,
+      "step": 3280
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 8.579877853393555,
+      "learning_rate": 2e-05,
+      "loss": 8.0881,
+      "step": 3290
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 11.129776954650879,
+      "learning_rate": 2e-05,
+      "loss": 8.0891,
+      "step": 3300
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 8.864716529846191,
+      "learning_rate": 2e-05,
+      "loss": 8.0441,
+      "step": 3310
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 9.308782577514648,
+      "learning_rate": 2e-05,
+      "loss": 7.9785,
+      "step": 3320
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 9.680919647216797,
+      "learning_rate": 2e-05,
+      "loss": 8.0948,
+      "step": 3330
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 8.62364387512207,
+      "learning_rate": 2e-05,
+      "loss": 8.1355,
+      "step": 3340
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 8.852781295776367,
+      "learning_rate": 2e-05,
+      "loss": 8.0703,
+      "step": 3350
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 10.227861404418945,
+      "learning_rate": 2e-05,
+      "loss": 8.0436,
+      "step": 3360
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 8.079804420471191,
+      "learning_rate": 2e-05,
+      "loss": 8.054,
+      "step": 3370
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 10.006095886230469,
+      "learning_rate": 2e-05,
+      "loss": 8.0519,
+      "step": 3380
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 10.324209213256836,
+      "learning_rate": 2e-05,
+      "loss": 8.1139,
+      "step": 3390
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 9.23745346069336,
+      "learning_rate": 2e-05,
+      "loss": 8.1708,
+      "step": 3400
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 9.457587242126465,
+      "learning_rate": 2e-05,
+      "loss": 8.1425,
+      "step": 3410
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 9.576904296875,
+      "learning_rate": 2e-05,
+      "loss": 8.0679,
+      "step": 3420
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 11.345622062683105,
+      "learning_rate": 2e-05,
+      "loss": 8.0637,
+      "step": 3430
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 9.576415061950684,
+      "learning_rate": 2e-05,
+      "loss": 8.0438,
+      "step": 3440
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 9.51587963104248,
+      "learning_rate": 2e-05,
+      "loss": 8.0507,
+      "step": 3450
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 9.820635795593262,
+      "learning_rate": 2e-05,
+      "loss": 8.0606,
+      "step": 3460
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 9.417253494262695,
+      "learning_rate": 2e-05,
+      "loss": 8.1759,
+      "step": 3470
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 9.284456253051758,
+      "learning_rate": 2e-05,
+      "loss": 8.0534,
+      "step": 3480
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 8.338359832763672,
+      "learning_rate": 2e-05,
+      "loss": 8.0779,
+      "step": 3490
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 9.401165008544922,
+      "learning_rate": 2e-05,
+      "loss": 8.0186,
+      "step": 3500
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 10.181764602661133,
+      "learning_rate": 2e-05,
+      "loss": 8.0635,
+      "step": 3510
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 9.012606620788574,
+      "learning_rate": 2e-05,
+      "loss": 8.0679,
+      "step": 3520
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 9.654341697692871,
+      "learning_rate": 2e-05,
+      "loss": 8.0289,
+      "step": 3530
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 9.150238037109375,
+      "learning_rate": 2e-05,
+      "loss": 8.0275,
+      "step": 3540
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 10.103556632995605,
+      "learning_rate": 2e-05,
+      "loss": 8.0991,
+      "step": 3550
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 8.847561836242676,
+      "learning_rate": 2e-05,
+      "loss": 7.9913,
+      "step": 3560
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 9.694567680358887,
+      "learning_rate": 2e-05,
+      "loss": 7.9898,
+      "step": 3570
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 9.53827953338623,
+      "learning_rate": 2e-05,
+      "loss": 8.0335,
+      "step": 3580
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 9.19649600982666,
+      "learning_rate": 2e-05,
+      "loss": 8.0523,
+      "step": 3590
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 8.776641845703125,
+      "learning_rate": 2e-05,
+      "loss": 7.9891,
+      "step": 3600
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 11.14356517791748,
+      "learning_rate": 2e-05,
+      "loss": 8.0279,
+      "step": 3610
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 9.697918891906738,
+      "learning_rate": 2e-05,
+      "loss": 8.1107,
+      "step": 3620
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 8.67265510559082,
+      "learning_rate": 2e-05,
+      "loss": 8.0801,
+      "step": 3630
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 10.395180702209473,
+      "learning_rate": 2e-05,
+      "loss": 8.0173,
+      "step": 3640
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 10.186140060424805,
+      "learning_rate": 2e-05,
+      "loss": 8.0012,
+      "step": 3650
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 11.245129585266113,
+      "learning_rate": 2e-05,
+      "loss": 8.0237,
+      "step": 3660
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 10.15030288696289,
+      "learning_rate": 2e-05,
+      "loss": 8.0261,
+      "step": 3670
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 10.449320793151855,
+      "learning_rate": 2e-05,
+      "loss": 7.9202,
+      "step": 3680
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 9.802896499633789,
+      "learning_rate": 2e-05,
+      "loss": 8.0213,
+      "step": 3690
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 10.889958381652832,
+      "learning_rate": 2e-05,
+      "loss": 8.0289,
+      "step": 3700
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 9.30747127532959,
+      "learning_rate": 2e-05,
+      "loss": 8.0528,
+      "step": 3710
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 8.981389999389648,
+      "learning_rate": 2e-05,
+      "loss": 8.0091,
+      "step": 3720
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 9.483953475952148,
+      "learning_rate": 2e-05,
+      "loss": 7.8909,
+      "step": 3730
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 9.39050579071045,
+      "learning_rate": 2e-05,
+      "loss": 8.0115,
+      "step": 3740
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 9.001296043395996,
+      "learning_rate": 2e-05,
+      "loss": 8.0501,
+      "step": 3750
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 9.78409194946289,
+      "learning_rate": 2e-05,
+      "loss": 8.069,
+      "step": 3760
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 9.297168731689453,
+      "learning_rate": 2e-05,
+      "loss": 8.0491,
+      "step": 3770
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 8.856414794921875,
+      "learning_rate": 2e-05,
+      "loss": 8.0541,
+      "step": 3780
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 9.554786682128906,
+      "learning_rate": 2e-05,
+      "loss": 8.0133,
+      "step": 3790
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 9.665678977966309,
+      "learning_rate": 2e-05,
+      "loss": 8.0117,
+      "step": 3800
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 10.269054412841797,
+      "learning_rate": 2e-05,
+      "loss": 7.9861,
+      "step": 3810
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 9.229450225830078,
+      "learning_rate": 2e-05,
+      "loss": 7.9407,
+      "step": 3820
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 9.85071086883545,
+      "learning_rate": 2e-05,
+      "loss": 7.9525,
+      "step": 3830
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 10.244184494018555,
+      "learning_rate": 2e-05,
+      "loss": 7.9838,
+      "step": 3840
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 10.48344612121582,
+      "learning_rate": 2e-05,
+      "loss": 7.987,
+      "step": 3850
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 11.154979705810547,
+      "learning_rate": 2e-05,
+      "loss": 7.9471,
+      "step": 3860
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 12.503634452819824,
+      "learning_rate": 2e-05,
+      "loss": 8.0579,
+      "step": 3870
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 9.788127899169922,
+      "learning_rate": 2e-05,
+      "loss": 7.9867,
+      "step": 3880
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 11.778328895568848,
+      "learning_rate": 2e-05,
+      "loss": 7.9438,
+      "step": 3890
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 10.351473808288574,
+      "learning_rate": 2e-05,
+      "loss": 7.934,
+      "step": 3900
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 9.388279914855957,
+      "learning_rate": 2e-05,
+      "loss": 7.8476,
+      "step": 3910
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 10.567912101745605,
+      "learning_rate": 2e-05,
+      "loss": 7.9815,
+      "step": 3920
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 9.967890739440918,
+      "learning_rate": 2e-05,
+      "loss": 8.0708,
+      "step": 3930
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 9.973456382751465,
+      "learning_rate": 2e-05,
+      "loss": 8.0513,
+      "step": 3940
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 10.04819393157959,
+      "learning_rate": 2e-05,
+      "loss": 8.0632,
+      "step": 3950
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 9.168389320373535,
+      "learning_rate": 2e-05,
+      "loss": 7.969,
+      "step": 3960
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 9.483097076416016,
+      "learning_rate": 2e-05,
+      "loss": 7.9711,
+      "step": 3970
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 9.73969554901123,
+      "learning_rate": 2e-05,
+      "loss": 8.023,
+      "step": 3980
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 10.169995307922363,
+      "learning_rate": 2e-05,
+      "loss": 7.947,
+      "step": 3990
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 9.4255952835083,
+      "learning_rate": 2e-05,
+      "loss": 7.9002,
+      "step": 4000
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 12.37176513671875,
+      "learning_rate": 2e-05,
+      "loss": 7.8886,
+      "step": 4010
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 10.53352165222168,
+      "learning_rate": 2e-05,
+      "loss": 8.0114,
+      "step": 4020
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 10.623584747314453,
+      "learning_rate": 2e-05,
+      "loss": 7.9997,
+      "step": 4030
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 10.543665885925293,
+      "learning_rate": 2e-05,
+      "loss": 7.9844,
+      "step": 4040
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 9.694969177246094,
+      "learning_rate": 2e-05,
+      "loss": 8.0585,
+      "step": 4050
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 10.558191299438477,
+      "learning_rate": 2e-05,
+      "loss": 7.8797,
+      "step": 4060
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 10.467811584472656,
+      "learning_rate": 2e-05,
+      "loss": 7.9119,
+      "step": 4070
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 9.917263984680176,
+      "learning_rate": 2e-05,
+      "loss": 7.9201,
+      "step": 4080
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 9.848325729370117,
+      "learning_rate": 2e-05,
+      "loss": 7.9108,
+      "step": 4090
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 10.110299110412598,
+      "learning_rate": 2e-05,
+      "loss": 7.9027,
+      "step": 4100
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 11.31668758392334,
+      "learning_rate": 2e-05,
+      "loss": 7.9183,
+      "step": 4110
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 11.457937240600586,
+      "learning_rate": 2e-05,
+      "loss": 7.9389,
+      "step": 4120
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 11.301414489746094,
+      "learning_rate": 2e-05,
+      "loss": 8.0034,
+      "step": 4130
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 12.37903881072998,
+      "learning_rate": 2e-05,
+      "loss": 8.0493,
+      "step": 4140
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 11.516155242919922,
+      "learning_rate": 2e-05,
+      "loss": 7.9628,
+      "step": 4150
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 11.090786933898926,
+      "learning_rate": 2e-05,
+      "loss": 7.9776,
+      "step": 4160
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 10.444341659545898,
+      "learning_rate": 2e-05,
+      "loss": 7.9261,
+      "step": 4170
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 11.582113265991211,
+      "learning_rate": 2e-05,
+      "loss": 7.9792,
+      "step": 4180
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 10.500890731811523,
+      "learning_rate": 2e-05,
+      "loss": 7.9973,
+      "step": 4190
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 9.741511344909668,
+      "learning_rate": 2e-05,
+      "loss": 7.964,
+      "step": 4200
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 10.729896545410156,
+      "learning_rate": 2e-05,
+      "loss": 7.9272,
+      "step": 4210
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 10.87218952178955,
+      "learning_rate": 2e-05,
+      "loss": 7.992,
+      "step": 4220
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 11.525153160095215,
+      "learning_rate": 2e-05,
+      "loss": 7.9667,
+      "step": 4230
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 11.26878547668457,
+      "learning_rate": 2e-05,
+      "loss": 7.8476,
+      "step": 4240
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 10.181687355041504,
+      "learning_rate": 2e-05,
+      "loss": 7.9559,
+      "step": 4250
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 10.845573425292969,
+      "learning_rate": 2e-05,
+      "loss": 7.9213,
+      "step": 4260
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 10.797900199890137,
+      "learning_rate": 2e-05,
+      "loss": 7.8455,
+      "step": 4270
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 13.006621360778809,
+      "learning_rate": 2e-05,
+      "loss": 7.8958,
+      "step": 4280
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 10.487809181213379,
+      "learning_rate": 2e-05,
+      "loss": 7.9173,
+      "step": 4290
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 10.153270721435547,
+      "learning_rate": 2e-05,
+      "loss": 7.9152,
+      "step": 4300
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 9.259800910949707,
+      "learning_rate": 2e-05,
+      "loss": 7.946,
+      "step": 4310
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 12.682476997375488,
+      "learning_rate": 2e-05,
+      "loss": 7.7964,
+      "step": 4320
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 12.45101261138916,
+      "learning_rate": 2e-05,
+      "loss": 7.8686,
+      "step": 4330
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 11.018670082092285,
+      "learning_rate": 2e-05,
+      "loss": 7.9374,
+      "step": 4340
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 9.952048301696777,
+      "learning_rate": 2e-05,
+      "loss": 7.9823,
+      "step": 4350
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 10.442728996276855,
+      "learning_rate": 2e-05,
+      "loss": 7.9187,
+      "step": 4360
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 11.25718879699707,
+      "learning_rate": 2e-05,
+      "loss": 7.8376,
+      "step": 4370
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 10.24433422088623,
+      "learning_rate": 2e-05,
+      "loss": 7.8765,
+      "step": 4380
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 10.578080177307129,
+      "learning_rate": 2e-05,
+      "loss": 7.9199,
+      "step": 4390
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 12.144083023071289,
+      "learning_rate": 2e-05,
+      "loss": 8.0144,
+      "step": 4400
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 11.248555183410645,
+      "learning_rate": 2e-05,
+      "loss": 7.8882,
+      "step": 4410
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 11.468417167663574,
+      "learning_rate": 2e-05,
+      "loss": 7.8958,
+      "step": 4420
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 10.33659553527832,
+      "learning_rate": 2e-05,
+      "loss": 7.8773,
+      "step": 4430
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 10.965078353881836,
+      "learning_rate": 2e-05,
+      "loss": 7.9466,
+      "step": 4440
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 9.387639045715332,
+      "learning_rate": 2e-05,
+      "loss": 7.8408,
+      "step": 4450
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 11.628375053405762,
+      "learning_rate": 2e-05,
+      "loss": 7.9698,
+      "step": 4460
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 12.198318481445312,
+      "learning_rate": 2e-05,
+      "loss": 7.8986,
+      "step": 4470
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 10.824650764465332,
+      "learning_rate": 2e-05,
+      "loss": 7.9679,
+      "step": 4480
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 11.820637702941895,
+      "learning_rate": 2e-05,
+      "loss": 7.7974,
+      "step": 4490
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 10.52723503112793,
+      "learning_rate": 2e-05,
+      "loss": 7.9733,
+      "step": 4500
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 10.386611938476562,
+      "learning_rate": 2e-05,
+      "loss": 7.9027,
+      "step": 4510
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 11.348063468933105,
+      "learning_rate": 2e-05,
+      "loss": 7.9171,
+      "step": 4520
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 11.450540542602539,
+      "learning_rate": 2e-05,
+      "loss": 7.8686,
+      "step": 4530
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 11.179993629455566,
+      "learning_rate": 2e-05,
+      "loss": 7.8701,
+      "step": 4540
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 11.810144424438477,
+      "learning_rate": 2e-05,
+      "loss": 7.905,
+      "step": 4550
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 11.196237564086914,
+      "learning_rate": 2e-05,
+      "loss": 7.9519,
+      "step": 4560
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 11.698023796081543,
+      "learning_rate": 2e-05,
+      "loss": 7.8401,
+      "step": 4570
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 12.389296531677246,
+      "learning_rate": 2e-05,
+      "loss": 7.8635,
+      "step": 4580
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 12.083377838134766,
+      "learning_rate": 2e-05,
+      "loss": 7.8977,
+      "step": 4590
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 12.124412536621094,
+      "learning_rate": 2e-05,
+      "loss": 7.9365,
+      "step": 4600
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 11.31529426574707,
+      "learning_rate": 2e-05,
+      "loss": 7.8923,
+      "step": 4610
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 12.892043113708496,
+      "learning_rate": 2e-05,
+      "loss": 7.8972,
+      "step": 4620
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 11.472381591796875,
+      "learning_rate": 2e-05,
+      "loss": 7.8723,
+      "step": 4630
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 12.507353782653809,
+      "learning_rate": 2e-05,
+      "loss": 7.8369,
+      "step": 4640
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 12.927081108093262,
+      "learning_rate": 2e-05,
+      "loss": 7.8243,
+      "step": 4650
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 12.60131549835205,
+      "learning_rate": 2e-05,
+      "loss": 7.9038,
+      "step": 4660
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 11.823628425598145,
+      "learning_rate": 2e-05,
+      "loss": 7.9244,
+      "step": 4670
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 11.845699310302734,
+      "learning_rate": 2e-05,
+      "loss": 7.8348,
+      "step": 4680
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 10.824399948120117,
+      "learning_rate": 2e-05,
+      "loss": 7.7431,
+      "step": 4690
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 13.317840576171875,
+      "learning_rate": 2e-05,
+      "loss": 7.836,
+      "step": 4700
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 11.34177017211914,
+      "learning_rate": 2e-05,
+      "loss": 7.9134,
+      "step": 4710
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 10.560473442077637,
+      "learning_rate": 2e-05,
+      "loss": 7.8829,
+      "step": 4720
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 12.243988037109375,
+      "learning_rate": 2e-05,
+      "loss": 7.9266,
+      "step": 4730
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 12.940305709838867,
+      "learning_rate": 2e-05,
+      "loss": 7.9087,
+      "step": 4740
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 11.258865356445312,
+      "learning_rate": 2e-05,
+      "loss": 7.8854,
+      "step": 4750
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 12.567364692687988,
+      "learning_rate": 2e-05,
+      "loss": 7.8561,
+      "step": 4760
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 11.7863187789917,
+      "learning_rate": 2e-05,
+      "loss": 7.814,
+      "step": 4770
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 13.81584644317627,
+      "learning_rate": 2e-05,
+      "loss": 7.9494,
+      "step": 4780
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 11.639108657836914,
+      "learning_rate": 2e-05,
+      "loss": 7.8561,
+      "step": 4790
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 11.763618469238281,
+      "learning_rate": 2e-05,
+      "loss": 7.7795,
+      "step": 4800
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 11.98811149597168,
+      "learning_rate": 2e-05,
+      "loss": 7.8992,
+      "step": 4810
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 11.929391860961914,
+      "learning_rate": 2e-05,
+      "loss": 7.7392,
+      "step": 4820
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 12.053762435913086,
+      "learning_rate": 2e-05,
+      "loss": 7.8673,
+      "step": 4830
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 12.402192115783691,
+      "learning_rate": 2e-05,
+      "loss": 7.7126,
+      "step": 4840
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 13.394449234008789,
+      "learning_rate": 2e-05,
+      "loss": 7.867,
+      "step": 4850
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 12.262216567993164,
+      "learning_rate": 2e-05,
+      "loss": 7.9043,
+      "step": 4860
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 13.228407859802246,
+      "learning_rate": 2e-05,
+      "loss": 7.8631,
+      "step": 4870
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 13.424823760986328,
+      "learning_rate": 2e-05,
+      "loss": 7.886,
+      "step": 4880
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 11.338706970214844,
+      "learning_rate": 2e-05,
+      "loss": 7.8099,
+      "step": 4890
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 10.792954444885254,
+      "learning_rate": 2e-05,
+      "loss": 7.8354,
+      "step": 4900
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 12.56781005859375,
+      "learning_rate": 2e-05,
+      "loss": 7.8084,
+      "step": 4910
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 11.307384490966797,
+      "learning_rate": 2e-05,
+      "loss": 7.808,
+      "step": 4920
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 11.481977462768555,
+      "learning_rate": 2e-05,
+      "loss": 7.8056,
+      "step": 4930
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 11.77100944519043,
+      "learning_rate": 2e-05,
+      "loss": 7.7072,
+      "step": 4940
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 12.401727676391602,
+      "learning_rate": 2e-05,
+      "loss": 7.7386,
+      "step": 4950
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 12.1062650680542,
+      "learning_rate": 2e-05,
+      "loss": 7.8765,
+      "step": 4960
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 11.662871360778809,
+      "learning_rate": 2e-05,
+      "loss": 7.8104,
+      "step": 4970
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 10.319502830505371,
+      "learning_rate": 2e-05,
+      "loss": 7.811,
+      "step": 4980
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 14.375482559204102,
+      "learning_rate": 2e-05,
+      "loss": 7.8092,
+      "step": 4990
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 11.772231101989746,
+      "learning_rate": 2e-05,
+      "loss": 7.8678,
+      "step": 5000
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 11.580702781677246,
+      "learning_rate": 2e-05,
+      "loss": 7.8491,
+      "step": 5010
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 12.376486778259277,
+      "learning_rate": 2e-05,
+      "loss": 7.7666,
+      "step": 5020
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 12.969429969787598,
+      "learning_rate": 2e-05,
+      "loss": 7.9628,
+      "step": 5030
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 11.849020957946777,
+      "learning_rate": 2e-05,
+      "loss": 7.7645,
+      "step": 5040
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 11.740395545959473,
+      "learning_rate": 2e-05,
+      "loss": 7.8281,
+      "step": 5050
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 11.01830005645752,
+      "learning_rate": 2e-05,
+      "loss": 7.7651,
+      "step": 5060
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 11.940682411193848,
+      "learning_rate": 2e-05,
+      "loss": 7.874,
+      "step": 5070
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 11.571117401123047,
+      "learning_rate": 2e-05,
+      "loss": 7.7252,
+      "step": 5080
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 11.25271987915039,
+      "learning_rate": 2e-05,
+      "loss": 7.7538,
+      "step": 5090
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 11.6790132522583,
+      "learning_rate": 2e-05,
+      "loss": 7.7931,
+      "step": 5100
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 12.91955280303955,
+      "learning_rate": 2e-05,
+      "loss": 7.7827,
+      "step": 5110
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 11.968914985656738,
+      "learning_rate": 2e-05,
+      "loss": 7.8374,
+      "step": 5120
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 13.723814010620117,
+      "learning_rate": 2e-05,
+      "loss": 7.7657,
+      "step": 5130
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 11.105441093444824,
+      "learning_rate": 2e-05,
+      "loss": 7.715,
+      "step": 5140
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 10.934996604919434,
+      "learning_rate": 2e-05,
+      "loss": 7.9029,
+      "step": 5150
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 12.546372413635254,
+      "learning_rate": 2e-05,
+      "loss": 7.8078,
+      "step": 5160
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 11.79869270324707,
+      "learning_rate": 2e-05,
+      "loss": 7.8074,
+      "step": 5170
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 10.983095169067383,
+      "learning_rate": 2e-05,
+      "loss": 7.7544,
+      "step": 5180
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 11.589228630065918,
+      "learning_rate": 2e-05,
+      "loss": 7.7862,
+      "step": 5190
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 11.117733001708984,
+      "learning_rate": 2e-05,
+      "loss": 7.6902,
+      "step": 5200
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 12.679126739501953,
+      "learning_rate": 2e-05,
+      "loss": 7.8266,
+      "step": 5210
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 12.305403709411621,
+      "learning_rate": 2e-05,
+      "loss": 7.7561,
+      "step": 5220
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 13.092717170715332,
+      "learning_rate": 2e-05,
+      "loss": 7.7515,
+      "step": 5230
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 13.391594886779785,
+      "learning_rate": 2e-05,
+      "loss": 7.6795,
+      "step": 5240
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 11.946535110473633,
+      "learning_rate": 2e-05,
+      "loss": 7.8545,
+      "step": 5250
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 11.089138984680176,
+      "learning_rate": 2e-05,
+      "loss": 7.8476,
+      "step": 5260
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 14.20877456665039,
+      "learning_rate": 2e-05,
+      "loss": 7.7826,
+      "step": 5270
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 11.574477195739746,
+      "learning_rate": 2e-05,
+      "loss": 7.7184,
+      "step": 5280
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 12.851311683654785,
+      "learning_rate": 2e-05,
+      "loss": 7.8299,
+      "step": 5290
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 12.378896713256836,
+      "learning_rate": 2e-05,
+      "loss": 7.795,
+      "step": 5300
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 12.522514343261719,
+      "learning_rate": 2e-05,
+      "loss": 7.718,
+      "step": 5310
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 13.493570327758789,
+      "learning_rate": 2e-05,
+      "loss": 7.7958,
+      "step": 5320
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 12.857158660888672,
+      "learning_rate": 2e-05,
+      "loss": 7.7224,
+      "step": 5330
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 12.470840454101562,
+      "learning_rate": 2e-05,
+      "loss": 7.8529,
+      "step": 5340
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 12.814496994018555,
+      "learning_rate": 2e-05,
+      "loss": 7.8089,
+      "step": 5350
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 11.507871627807617,
+      "learning_rate": 2e-05,
+      "loss": 7.6937,
+      "step": 5360
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 11.652280807495117,
+      "learning_rate": 2e-05,
+      "loss": 7.7474,
+      "step": 5370
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 12.437298774719238,
+      "learning_rate": 2e-05,
+      "loss": 7.8588,
+      "step": 5380
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 11.579487800598145,
+      "learning_rate": 2e-05,
+      "loss": 7.7317,
+      "step": 5390
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 13.867403984069824,
+      "learning_rate": 2e-05,
+      "loss": 7.7823,
+      "step": 5400
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 12.129871368408203,
+      "learning_rate": 2e-05,
+      "loss": 7.7584,
+      "step": 5410
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 12.282061576843262,
+      "learning_rate": 2e-05,
+      "loss": 7.7109,
+      "step": 5420
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 11.934682846069336,
+      "learning_rate": 2e-05,
+      "loss": 7.7228,
+      "step": 5430
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 12.423234939575195,
+      "learning_rate": 2e-05,
+      "loss": 7.7178,
+      "step": 5440
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 11.44439697265625,
+      "learning_rate": 2e-05,
+      "loss": 7.7274,
+      "step": 5450
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 12.633554458618164,
+      "learning_rate": 2e-05,
+      "loss": 7.833,
+      "step": 5460
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 11.462240219116211,
+      "learning_rate": 2e-05,
+      "loss": 7.7612,
+      "step": 5470
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 11.739327430725098,
+      "learning_rate": 2e-05,
+      "loss": 7.7043,
+      "step": 5480
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 14.724932670593262,
+      "learning_rate": 2e-05,
+      "loss": 7.8413,
+      "step": 5490
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 12.084114074707031,
+      "learning_rate": 2e-05,
+      "loss": 7.7399,
+      "step": 5500
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 13.651264190673828,
+      "learning_rate": 2e-05,
+      "loss": 7.6787,
+      "step": 5510
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 13.092822074890137,
+      "learning_rate": 2e-05,
+      "loss": 7.7688,
+      "step": 5520
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 12.528792381286621,
+      "learning_rate": 2e-05,
+      "loss": 7.7531,
+      "step": 5530
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 11.917206764221191,
+      "learning_rate": 2e-05,
+      "loss": 7.6801,
+      "step": 5540
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 12.68786907196045,
+      "learning_rate": 2e-05,
+      "loss": 7.7833,
+      "step": 5550
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 14.289521217346191,
+      "learning_rate": 2e-05,
+      "loss": 7.7304,
+      "step": 5560
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 14.458538055419922,
+      "learning_rate": 2e-05,
+      "loss": 7.7492,
+      "step": 5570
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 13.892851829528809,
+      "learning_rate": 2e-05,
+      "loss": 7.7708,
+      "step": 5580
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 10.881427764892578,
+      "learning_rate": 2e-05,
+      "loss": 7.7356,
+      "step": 5590
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 12.215607643127441,
+      "learning_rate": 2e-05,
+      "loss": 7.7422,
+      "step": 5600
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 11.983213424682617,
+      "learning_rate": 2e-05,
+      "loss": 7.7991,
+      "step": 5610
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 12.86385440826416,
+      "learning_rate": 2e-05,
+      "loss": 7.7741,
+      "step": 5620
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 13.395235061645508,
+      "learning_rate": 2e-05,
+      "loss": 7.6965,
+      "step": 5630
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 13.407181739807129,
+      "learning_rate": 2e-05,
+      "loss": 7.6858,
+      "step": 5640
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 14.736617088317871,
+      "learning_rate": 2e-05,
+      "loss": 7.6883,
+      "step": 5650
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 12.862197875976562,
+      "learning_rate": 2e-05,
+      "loss": 7.6392,
+      "step": 5660
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 11.421501159667969,
+      "learning_rate": 2e-05,
+      "loss": 7.7546,
+      "step": 5670
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 12.803354263305664,
+      "learning_rate": 2e-05,
+      "loss": 7.6882,
+      "step": 5680
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 11.781133651733398,
+      "learning_rate": 2e-05,
+      "loss": 7.6355,
+      "step": 5690
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 12.948162078857422,
+      "learning_rate": 2e-05,
+      "loss": 7.7195,
+      "step": 5700
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 12.40371322631836,
+      "learning_rate": 2e-05,
+      "loss": 7.7991,
+      "step": 5710
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 12.687602043151855,
+      "learning_rate": 2e-05,
+      "loss": 7.8181,
+      "step": 5720
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 12.892142295837402,
+      "learning_rate": 2e-05,
+      "loss": 7.6451,
+      "step": 5730
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 11.78799819946289,
+      "learning_rate": 2e-05,
+      "loss": 7.7268,
+      "step": 5740
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 13.14787483215332,
+      "learning_rate": 2e-05,
+      "loss": 7.7392,
+      "step": 5750
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 10.668608665466309,
+      "learning_rate": 2e-05,
+      "loss": 7.6119,
+      "step": 5760
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 11.593132019042969,
+      "learning_rate": 2e-05,
+      "loss": 7.7505,
+      "step": 5770
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 12.012129783630371,
+      "learning_rate": 2e-05,
+      "loss": 7.6458,
+      "step": 5780
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 13.61383056640625,
+      "learning_rate": 2e-05,
+      "loss": 7.6075,
+      "step": 5790
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 13.427445411682129,
+      "learning_rate": 2e-05,
+      "loss": 7.545,
+      "step": 5800
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 11.738509178161621,
+      "learning_rate": 2e-05,
+      "loss": 7.7265,
+      "step": 5810
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 11.858267784118652,
+      "learning_rate": 2e-05,
+      "loss": 7.5798,
+      "step": 5820
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 12.053930282592773,
+      "learning_rate": 2e-05,
+      "loss": 7.6853,
+      "step": 5830
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 13.20464038848877,
+      "learning_rate": 2e-05,
+      "loss": 7.6413,
+      "step": 5840
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 13.153386116027832,
+      "learning_rate": 2e-05,
+      "loss": 7.5968,
+      "step": 5850
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 11.358457565307617,
+      "learning_rate": 2e-05,
+      "loss": 7.6922,
+      "step": 5860
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 12.747382164001465,
+      "learning_rate": 2e-05,
+      "loss": 7.6579,
+      "step": 5870
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 10.794285774230957,
+      "learning_rate": 2e-05,
+      "loss": 7.7513,
+      "step": 5880
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 11.909981727600098,
+      "learning_rate": 2e-05,
+      "loss": 7.691,
+      "step": 5890
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 12.693133354187012,
+      "learning_rate": 2e-05,
+      "loss": 7.6933,
+      "step": 5900
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 13.190779685974121,
+      "learning_rate": 2e-05,
+      "loss": 7.5918,
+      "step": 5910
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 12.421589851379395,
+      "learning_rate": 2e-05,
+      "loss": 7.7407,
+      "step": 5920
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 11.49426555633545,
+      "learning_rate": 2e-05,
+      "loss": 7.725,
+      "step": 5930
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 14.024295806884766,
+      "learning_rate": 2e-05,
+      "loss": 7.6448,
+      "step": 5940
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 12.889427185058594,
+      "learning_rate": 2e-05,
+      "loss": 7.6898,
+      "step": 5950
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 13.766159057617188,
+      "learning_rate": 2e-05,
+      "loss": 7.7653,
+      "step": 5960
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 14.475167274475098,
+      "learning_rate": 2e-05,
+      "loss": 7.5573,
+      "step": 5970
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 13.757892608642578,
+      "learning_rate": 2e-05,
+      "loss": 7.5462,
+      "step": 5980
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 12.061361312866211,
+      "learning_rate": 2e-05,
+      "loss": 7.8055,
+      "step": 5990
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 13.636422157287598,
+      "learning_rate": 2e-05,
+      "loss": 7.6214,
+      "step": 6000
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 14.150060653686523,
+      "learning_rate": 2e-05,
+      "loss": 7.6065,
+      "step": 6010
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 12.258604049682617,
+      "learning_rate": 2e-05,
+      "loss": 7.5144,
+      "step": 6020
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 12.606104850769043,
+      "learning_rate": 2e-05,
+      "loss": 7.6093,
+      "step": 6030
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 13.50027847290039,
+      "learning_rate": 2e-05,
+      "loss": 7.6883,
+      "step": 6040
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 13.978836059570312,
+      "learning_rate": 2e-05,
+      "loss": 7.6374,
+      "step": 6050
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 13.105231285095215,
+      "learning_rate": 2e-05,
+      "loss": 7.6436,
+      "step": 6060
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 12.367439270019531,
+      "learning_rate": 2e-05,
+      "loss": 7.6322,
+      "step": 6070
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 14.727394104003906,
+      "learning_rate": 2e-05,
+      "loss": 7.6855,
+      "step": 6080
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 13.642365455627441,
+      "learning_rate": 2e-05,
+      "loss": 7.6753,
+      "step": 6090
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 12.201065063476562,
+      "learning_rate": 2e-05,
+      "loss": 7.5322,
+      "step": 6100
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 14.81038761138916,
+      "learning_rate": 2e-05,
+      "loss": 7.7141,
+      "step": 6110
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 12.707942962646484,
+      "learning_rate": 2e-05,
+      "loss": 7.698,
+      "step": 6120
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 13.137734413146973,
+      "learning_rate": 2e-05,
+      "loss": 7.6428,
+      "step": 6130
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 13.739848136901855,
+      "learning_rate": 2e-05,
+      "loss": 7.6363,
+      "step": 6140
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 12.056344985961914,
+      "learning_rate": 2e-05,
+      "loss": 7.7061,
+      "step": 6150
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 12.79162883758545,
+      "learning_rate": 2e-05,
+      "loss": 7.5253,
+      "step": 6160
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 11.619795799255371,
+      "learning_rate": 2e-05,
+      "loss": 7.7321,
+      "step": 6170
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 13.802376747131348,
+      "learning_rate": 2e-05,
+      "loss": 7.6508,
+      "step": 6180
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 13.521082878112793,
+      "learning_rate": 2e-05,
+      "loss": 7.7344,
+      "step": 6190
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 12.086933135986328,
+      "learning_rate": 2e-05,
+      "loss": 7.7302,
+      "step": 6200
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 12.61682415008545,
+      "learning_rate": 2e-05,
+      "loss": 7.5631,
+      "step": 6210
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 11.745560646057129,
+      "learning_rate": 2e-05,
+      "loss": 7.6811,
+      "step": 6220
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 14.0714111328125,
+      "learning_rate": 2e-05,
+      "loss": 7.596,
+      "step": 6230
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 15.994614601135254,
+      "learning_rate": 2e-05,
+      "loss": 7.6223,
+      "step": 6240
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 15.441205978393555,
+      "learning_rate": 2e-05,
+      "loss": 7.6425,
+      "step": 6250
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 14.514402389526367,
+      "learning_rate": 2e-05,
+      "loss": 7.7376,
+      "step": 6260
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 12.288745880126953,
+      "learning_rate": 2e-05,
+      "loss": 7.5163,
+      "step": 6270
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 12.284796714782715,
+      "learning_rate": 2e-05,
+      "loss": 7.6676,
+      "step": 6280
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 15.070308685302734,
+      "learning_rate": 2e-05,
+      "loss": 7.6102,
+      "step": 6290
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 12.163321495056152,
+      "learning_rate": 2e-05,
+      "loss": 7.6352,
+      "step": 6300
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 11.574402809143066,
+      "learning_rate": 2e-05,
+      "loss": 7.6139,
+      "step": 6310
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 16.586841583251953,
+      "learning_rate": 2e-05,
+      "loss": 7.7446,
+      "step": 6320
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 14.0075044631958,
+      "learning_rate": 2e-05,
+      "loss": 7.5986,
+      "step": 6330
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 12.41870403289795,
+      "learning_rate": 2e-05,
+      "loss": 7.6609,
+      "step": 6340
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 11.521857261657715,
+      "learning_rate": 2e-05,
+      "loss": 7.6029,
+      "step": 6350
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 13.122491836547852,
+      "learning_rate": 2e-05,
+      "loss": 7.6668,
+      "step": 6360
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 12.871292114257812,
+      "learning_rate": 2e-05,
+      "loss": 7.5329,
+      "step": 6370
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 11.035774230957031,
+      "learning_rate": 2e-05,
+      "loss": 7.5447,
+      "step": 6380
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 14.130277633666992,
+      "learning_rate": 2e-05,
+      "loss": 7.6739,
+      "step": 6390
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 13.307592391967773,
+      "learning_rate": 2e-05,
+      "loss": 7.6254,
+      "step": 6400
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 13.237303733825684,
+      "learning_rate": 2e-05,
+      "loss": 7.5589,
+      "step": 6410
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 12.791316986083984,
+      "learning_rate": 2e-05,
+      "loss": 7.6076,
+      "step": 6420
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 12.028976440429688,
+      "learning_rate": 2e-05,
+      "loss": 7.7252,
+      "step": 6430
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 13.669393539428711,
+      "learning_rate": 2e-05,
+      "loss": 7.6529,
+      "step": 6440
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 12.609371185302734,
+      "learning_rate": 2e-05,
+      "loss": 7.5711,
+      "step": 6450
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 13.343849182128906,
+      "learning_rate": 2e-05,
+      "loss": 7.5705,
+      "step": 6460
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 11.671335220336914,
+      "learning_rate": 2e-05,
+      "loss": 7.6177,
+      "step": 6470
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 12.821616172790527,
+      "learning_rate": 2e-05,
+      "loss": 7.5159,
+      "step": 6480
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 13.23855972290039,
+      "learning_rate": 2e-05,
+      "loss": 7.6207,
+      "step": 6490
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 13.155604362487793,
+      "learning_rate": 2e-05,
+      "loss": 7.7152,
+      "step": 6500
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 13.549511909484863,
+      "learning_rate": 2e-05,
+      "loss": 7.5282,
+      "step": 6510
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 13.697501182556152,
+      "learning_rate": 2e-05,
+      "loss": 7.6439,
+      "step": 6520
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 13.089825630187988,
+      "learning_rate": 2e-05,
+      "loss": 7.691,
+      "step": 6530
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 11.67117977142334,
+      "learning_rate": 2e-05,
+      "loss": 7.6133,
+      "step": 6540
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 12.503938674926758,
+      "learning_rate": 2e-05,
+      "loss": 7.6035,
+      "step": 6550
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 13.610810279846191,
+      "learning_rate": 2e-05,
+      "loss": 7.6091,
+      "step": 6560
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 15.0921630859375,
+      "learning_rate": 2e-05,
+      "loss": 7.4853,
+      "step": 6570
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 24.06075668334961,
+      "learning_rate": 2e-05,
+      "loss": 7.602,
+      "step": 6580
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 13.069271087646484,
+      "learning_rate": 2e-05,
+      "loss": 7.4784,
+      "step": 6590
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 12.580757141113281,
+      "learning_rate": 2e-05,
+      "loss": 7.5774,
+      "step": 6600
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 14.417656898498535,
+      "learning_rate": 2e-05,
+      "loss": 7.6231,
+      "step": 6610
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 14.776442527770996,
+      "learning_rate": 2e-05,
+      "loss": 7.6178,
+      "step": 6620
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 13.63784122467041,
+      "learning_rate": 2e-05,
+      "loss": 7.6662,
+      "step": 6630
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 13.281730651855469,
+      "learning_rate": 2e-05,
+      "loss": 7.4478,
+      "step": 6640
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 14.626810073852539,
+      "learning_rate": 2e-05,
+      "loss": 7.5529,
+      "step": 6650
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 12.669391632080078,
+      "learning_rate": 2e-05,
+      "loss": 7.5721,
+      "step": 6660
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 13.264552116394043,
+      "learning_rate": 2e-05,
+      "loss": 7.5117,
+      "step": 6670
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 13.644275665283203,
+      "learning_rate": 2e-05,
+      "loss": 7.5442,
+      "step": 6680
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 12.299631118774414,
+      "learning_rate": 2e-05,
+      "loss": 7.5635,
+      "step": 6690
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 12.347768783569336,
+      "learning_rate": 2e-05,
+      "loss": 7.5485,
+      "step": 6700
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 12.435714721679688,
+      "learning_rate": 2e-05,
+      "loss": 7.5513,
+      "step": 6710
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 11.771161079406738,
+      "learning_rate": 2e-05,
+      "loss": 7.488,
+      "step": 6720
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 13.051584243774414,
+      "learning_rate": 2e-05,
+      "loss": 7.5964,
+      "step": 6730
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 14.269665718078613,
+      "learning_rate": 2e-05,
+      "loss": 7.5144,
+      "step": 6740
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 13.062386512756348,
+      "learning_rate": 2e-05,
+      "loss": 7.4314,
+      "step": 6750
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 14.155104637145996,
+      "learning_rate": 2e-05,
+      "loss": 7.5015,
+      "step": 6760
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 12.917915344238281,
+      "learning_rate": 2e-05,
+      "loss": 7.4759,
+      "step": 6770
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 14.08447265625,
+      "learning_rate": 2e-05,
+      "loss": 7.5126,
+      "step": 6780
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 14.374971389770508,
+      "learning_rate": 2e-05,
+      "loss": 7.6565,
+      "step": 6790
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 12.398031234741211,
+      "learning_rate": 2e-05,
+      "loss": 7.5984,
+      "step": 6800
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 14.600011825561523,
+      "learning_rate": 2e-05,
+      "loss": 7.8005,
+      "step": 6810
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 13.158529281616211,
+      "learning_rate": 2e-05,
+      "loss": 7.4831,
+      "step": 6820
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 12.538118362426758,
+      "learning_rate": 2e-05,
+      "loss": 7.492,
+      "step": 6830
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 13.80060863494873,
+      "learning_rate": 2e-05,
+      "loss": 7.5616,
+      "step": 6840
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 12.7617826461792,
+      "learning_rate": 2e-05,
+      "loss": 7.5462,
+      "step": 6850
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 13.496440887451172,
+      "learning_rate": 2e-05,
+      "loss": 7.6323,
+      "step": 6860
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 15.512273788452148,
+      "learning_rate": 2e-05,
+      "loss": 7.6673,
+      "step": 6870
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 12.945910453796387,
+      "learning_rate": 2e-05,
+      "loss": 7.617,
+      "step": 6880
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 14.490726470947266,
+      "learning_rate": 2e-05,
+      "loss": 7.5075,
+      "step": 6890
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 14.290457725524902,
+      "learning_rate": 2e-05,
+      "loss": 7.6535,
+      "step": 6900
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 13.886164665222168,
+      "learning_rate": 2e-05,
+      "loss": 7.5951,
+      "step": 6910
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 13.921050071716309,
+      "learning_rate": 2e-05,
+      "loss": 7.6179,
+      "step": 6920
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 13.68610668182373,
+      "learning_rate": 2e-05,
+      "loss": 7.5707,
+      "step": 6930
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 13.03588581085205,
+      "learning_rate": 2e-05,
+      "loss": 7.4473,
+      "step": 6940
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 12.563400268554688,
+      "learning_rate": 2e-05,
+      "loss": 7.5615,
+      "step": 6950
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 12.349410057067871,
+      "learning_rate": 2e-05,
+      "loss": 7.5659,
+      "step": 6960
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 12.014132499694824,
+      "learning_rate": 2e-05,
+      "loss": 7.5601,
+      "step": 6970
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 17.268798828125,
+      "learning_rate": 2e-05,
+      "loss": 7.4141,
+      "step": 6980
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 13.139918327331543,
+      "learning_rate": 2e-05,
+      "loss": 7.4966,
+      "step": 6990
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 13.890971183776855,
+      "learning_rate": 2e-05,
+      "loss": 7.4616,
+      "step": 7000
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 14.168231010437012,
+      "learning_rate": 2e-05,
+      "loss": 7.4725,
+      "step": 7010
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 13.274535179138184,
+      "learning_rate": 2e-05,
+      "loss": 7.5551,
+      "step": 7020
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 14.029889106750488,
+      "learning_rate": 2e-05,
+      "loss": 7.4012,
+      "step": 7030
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 14.23734188079834,
+      "learning_rate": 2e-05,
+      "loss": 7.5281,
+      "step": 7040
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 12.383326530456543,
+      "learning_rate": 2e-05,
+      "loss": 7.5442,
+      "step": 7050
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 12.809000015258789,
+      "learning_rate": 2e-05,
+      "loss": 7.5413,
+      "step": 7060
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 16.28673553466797,
+      "learning_rate": 2e-05,
+      "loss": 7.528,
+      "step": 7070
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 12.65638542175293,
+      "learning_rate": 2e-05,
+      "loss": 7.4358,
+      "step": 7080
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 12.994460105895996,
+      "learning_rate": 2e-05,
+      "loss": 7.5908,
+      "step": 7090
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 14.29158878326416,
+      "learning_rate": 2e-05,
+      "loss": 7.5413,
+      "step": 7100
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 14.240143775939941,
+      "learning_rate": 2e-05,
+      "loss": 7.5555,
+      "step": 7110
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 12.428028106689453,
+      "learning_rate": 2e-05,
+      "loss": 7.4771,
+      "step": 7120
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 12.828241348266602,
+      "learning_rate": 2e-05,
+      "loss": 7.3765,
+      "step": 7130
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 13.394918441772461,
+      "learning_rate": 2e-05,
+      "loss": 7.4571,
+      "step": 7140
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 12.844101905822754,
+      "learning_rate": 2e-05,
+      "loss": 7.5225,
+      "step": 7150
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 12.868141174316406,
+      "learning_rate": 2e-05,
+      "loss": 7.4247,
+      "step": 7160
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 12.855489730834961,
+      "learning_rate": 2e-05,
+      "loss": 7.5269,
+      "step": 7170
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 11.858540534973145,
+      "learning_rate": 2e-05,
+      "loss": 7.3854,
+      "step": 7180
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 14.341757774353027,
+      "learning_rate": 2e-05,
+      "loss": 7.5081,
+      "step": 7190
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 13.02086353302002,
+      "learning_rate": 2e-05,
+      "loss": 7.4882,
+      "step": 7200
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 11.801441192626953,
+      "learning_rate": 2e-05,
+      "loss": 7.3642,
+      "step": 7210
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 13.310677528381348,
+      "learning_rate": 2e-05,
+      "loss": 7.5,
+      "step": 7220
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 12.495135307312012,
+      "learning_rate": 2e-05,
+      "loss": 7.5118,
+      "step": 7230
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 14.30810832977295,
+      "learning_rate": 2e-05,
+      "loss": 7.5492,
+      "step": 7240
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 12.292363166809082,
+      "learning_rate": 2e-05,
+      "loss": 7.4785,
+      "step": 7250
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 14.976513862609863,
+      "learning_rate": 2e-05,
+      "loss": 7.4816,
+      "step": 7260
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 14.179727554321289,
+      "learning_rate": 2e-05,
+      "loss": 7.5389,
+      "step": 7270
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 14.563206672668457,
+      "learning_rate": 2e-05,
+      "loss": 7.5654,
+      "step": 7280
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 15.764737129211426,
+      "learning_rate": 2e-05,
+      "loss": 7.4662,
+      "step": 7290
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 13.340702056884766,
+      "learning_rate": 2e-05,
+      "loss": 7.4842,
+      "step": 7300
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 13.077180862426758,
+      "learning_rate": 2e-05,
+      "loss": 7.4497,
+      "step": 7310
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 13.05450439453125,
+      "learning_rate": 2e-05,
+      "loss": 7.5267,
+      "step": 7320
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 12.377302169799805,
+      "learning_rate": 2e-05,
+      "loss": 7.3455,
+      "step": 7330
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 12.927446365356445,
+      "learning_rate": 2e-05,
+      "loss": 7.3704,
+      "step": 7340
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 12.839237213134766,
+      "learning_rate": 2e-05,
+      "loss": 7.4656,
+      "step": 7350
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 13.63042163848877,
+      "learning_rate": 2e-05,
+      "loss": 7.5184,
+      "step": 7360
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 13.776434898376465,
+      "learning_rate": 2e-05,
+      "loss": 7.5183,
+      "step": 7370
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 14.940461158752441,
+      "learning_rate": 2e-05,
+      "loss": 7.3559,
+      "step": 7380
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 14.326674461364746,
+      "learning_rate": 2e-05,
+      "loss": 7.4069,
+      "step": 7390
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 14.799200057983398,
+      "learning_rate": 2e-05,
+      "loss": 7.5054,
+      "step": 7400
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 13.770232200622559,
+      "learning_rate": 2e-05,
+      "loss": 7.4284,
+      "step": 7410
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 12.693459510803223,
+      "learning_rate": 2e-05,
+      "loss": 7.5346,
+      "step": 7420
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 13.208507537841797,
+      "learning_rate": 2e-05,
+      "loss": 7.4268,
+      "step": 7430
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 12.999556541442871,
+      "learning_rate": 2e-05,
+      "loss": 7.459,
+      "step": 7440
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 14.830228805541992,
+      "learning_rate": 2e-05,
+      "loss": 7.4693,
+      "step": 7450
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 14.03957748413086,
+      "learning_rate": 2e-05,
+      "loss": 7.4358,
+      "step": 7460
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 14.627981185913086,
+      "learning_rate": 2e-05,
+      "loss": 7.4786,
+      "step": 7470
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 13.621996879577637,
+      "learning_rate": 2e-05,
+      "loss": 7.393,
+      "step": 7480
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 12.983625411987305,
+      "learning_rate": 2e-05,
+      "loss": 7.4349,
+      "step": 7490
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 13.211760520935059,
+      "learning_rate": 2e-05,
+      "loss": 7.4481,
+      "step": 7500
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 11.73476505279541,
+      "learning_rate": 2e-05,
+      "loss": 7.3221,
+      "step": 7510
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 12.624098777770996,
+      "learning_rate": 2e-05,
+      "loss": 7.4635,
+      "step": 7520
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 13.645166397094727,
+      "learning_rate": 2e-05,
+      "loss": 7.4375,
+      "step": 7530
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 14.92273235321045,
+      "learning_rate": 2e-05,
+      "loss": 7.4818,
+      "step": 7540
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 13.742351531982422,
+      "learning_rate": 2e-05,
+      "loss": 7.5449,
+      "step": 7550
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 14.764723777770996,
+      "learning_rate": 2e-05,
+      "loss": 7.4358,
+      "step": 7560
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 13.81969165802002,
+      "learning_rate": 2e-05,
+      "loss": 7.5809,
+      "step": 7570
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 13.539517402648926,
+      "learning_rate": 2e-05,
+      "loss": 7.5862,
+      "step": 7580
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 11.803396224975586,
+      "learning_rate": 2e-05,
+      "loss": 7.5771,
+      "step": 7590
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 12.753072738647461,
+      "learning_rate": 2e-05,
+      "loss": 7.5,
+      "step": 7600
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 12.407511711120605,
+      "learning_rate": 2e-05,
+      "loss": 7.3288,
+      "step": 7610
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 15.371244430541992,
+      "learning_rate": 2e-05,
+      "loss": 7.5215,
+      "step": 7620
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 13.076952934265137,
+      "learning_rate": 2e-05,
+      "loss": 7.3726,
+      "step": 7630
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 14.617666244506836,
+      "learning_rate": 2e-05,
+      "loss": 7.4606,
+      "step": 7640
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 13.455782890319824,
+      "learning_rate": 2e-05,
+      "loss": 7.439,
+      "step": 7650
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 13.863014221191406,
+      "learning_rate": 2e-05,
+      "loss": 7.3237,
+      "step": 7660
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 14.569392204284668,
+      "learning_rate": 2e-05,
+      "loss": 7.5177,
+      "step": 7670
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 15.902585983276367,
+      "learning_rate": 2e-05,
+      "loss": 7.4238,
+      "step": 7680
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 13.035500526428223,
+      "learning_rate": 2e-05,
+      "loss": 7.5244,
+      "step": 7690
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 13.047713279724121,
+      "learning_rate": 2e-05,
+      "loss": 7.4139,
+      "step": 7700
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 12.996772766113281,
+      "learning_rate": 2e-05,
+      "loss": 7.5037,
+      "step": 7710
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 12.771913528442383,
+      "learning_rate": 2e-05,
+      "loss": 7.4143,
+      "step": 7720
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 12.52166748046875,
+      "learning_rate": 2e-05,
+      "loss": 7.4476,
+      "step": 7730
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 13.958206176757812,
+      "learning_rate": 2e-05,
+      "loss": 7.3754,
+      "step": 7740
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 12.911038398742676,
+      "learning_rate": 2e-05,
+      "loss": 7.5346,
+      "step": 7750
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 14.906147003173828,
+      "learning_rate": 2e-05,
+      "loss": 7.5676,
+      "step": 7760
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 14.6663818359375,
+      "learning_rate": 2e-05,
+      "loss": 7.4197,
+      "step": 7770
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 13.564197540283203,
+      "learning_rate": 2e-05,
+      "loss": 7.4049,
+      "step": 7780
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 14.497262954711914,
+      "learning_rate": 2e-05,
+      "loss": 7.4398,
+      "step": 7790
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 13.776274681091309,
+      "learning_rate": 2e-05,
+      "loss": 7.3667,
+      "step": 7800
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 12.940631866455078,
+      "learning_rate": 2e-05,
+      "loss": 7.3757,
+      "step": 7810
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 13.792564392089844,
+      "learning_rate": 2e-05,
+      "loss": 7.4213,
+      "step": 7820
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 14.88924503326416,
+      "learning_rate": 2e-05,
+      "loss": 7.4126,
+      "step": 7830
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 13.505547523498535,
+      "learning_rate": 2e-05,
+      "loss": 7.3781,
+      "step": 7840
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 13.678604125976562,
+      "learning_rate": 2e-05,
+      "loss": 7.3134,
+      "step": 7850
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 13.167811393737793,
+      "learning_rate": 2e-05,
+      "loss": 7.3562,
+      "step": 7860
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 12.239130020141602,
+      "learning_rate": 2e-05,
+      "loss": 7.3488,
+      "step": 7870
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 15.091113090515137,
+      "learning_rate": 2e-05,
+      "loss": 7.3989,
+      "step": 7880
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 13.62412166595459,
+      "learning_rate": 2e-05,
+      "loss": 7.4648,
+      "step": 7890
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 13.304557800292969,
+      "learning_rate": 2e-05,
+      "loss": 7.4043,
+      "step": 7900
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 13.760061264038086,
+      "learning_rate": 2e-05,
+      "loss": 7.2145,
+      "step": 7910
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 15.031874656677246,
+      "learning_rate": 2e-05,
+      "loss": 7.4044,
+      "step": 7920
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 14.076127052307129,
+      "learning_rate": 2e-05,
+      "loss": 7.4113,
+      "step": 7930
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 14.220410346984863,
+      "learning_rate": 2e-05,
+      "loss": 7.4377,
+      "step": 7940
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 13.492103576660156,
+      "learning_rate": 2e-05,
+      "loss": 7.2467,
+      "step": 7950
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 13.282194137573242,
+      "learning_rate": 2e-05,
+      "loss": 7.4319,
+      "step": 7960
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 13.472664833068848,
+      "learning_rate": 2e-05,
+      "loss": 7.3398,
+      "step": 7970
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 12.505082130432129,
+      "learning_rate": 2e-05,
+      "loss": 7.4133,
+      "step": 7980
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 13.477477073669434,
+      "learning_rate": 2e-05,
+      "loss": 7.4264,
+      "step": 7990
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 14.167861938476562,
+      "learning_rate": 2e-05,
+      "loss": 7.3354,
+      "step": 8000
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 13.848207473754883,
+      "learning_rate": 2e-05,
+      "loss": 7.3764,
+      "step": 8010
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 14.328885078430176,
+      "learning_rate": 2e-05,
+      "loss": 7.5825,
+      "step": 8020
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 14.757925987243652,
+      "learning_rate": 2e-05,
+      "loss": 7.4146,
+      "step": 8030
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 12.706913948059082,
+      "learning_rate": 2e-05,
+      "loss": 7.2122,
+      "step": 8040
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 13.134736061096191,
+      "learning_rate": 2e-05,
+      "loss": 7.3669,
+      "step": 8050
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 13.385063171386719,
+      "learning_rate": 2e-05,
+      "loss": 7.3147,
+      "step": 8060
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 15.942737579345703,
+      "learning_rate": 2e-05,
+      "loss": 7.3808,
+      "step": 8070
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 12.40909481048584,
+      "learning_rate": 2e-05,
+      "loss": 7.2547,
+      "step": 8080
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 14.604430198669434,
+      "learning_rate": 2e-05,
+      "loss": 7.4981,
+      "step": 8090
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 12.278432846069336,
+      "learning_rate": 2e-05,
+      "loss": 7.4168,
+      "step": 8100
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 14.4340238571167,
+      "learning_rate": 2e-05,
+      "loss": 7.2392,
+      "step": 8110
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 13.629953384399414,
+      "learning_rate": 2e-05,
+      "loss": 7.1938,
+      "step": 8120
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 12.709035873413086,
+      "learning_rate": 2e-05,
+      "loss": 7.3626,
+      "step": 8130
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 13.00856876373291,
+      "learning_rate": 2e-05,
+      "loss": 7.2785,
+      "step": 8140
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 15.521533966064453,
+      "learning_rate": 2e-05,
+      "loss": 7.429,
+      "step": 8150
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 14.078548431396484,
+      "learning_rate": 2e-05,
+      "loss": 7.4413,
+      "step": 8160
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 14.085650444030762,
+      "learning_rate": 2e-05,
+      "loss": 7.5131,
+      "step": 8170
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 13.876462936401367,
+      "learning_rate": 2e-05,
+      "loss": 7.4281,
+      "step": 8180
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 16.023845672607422,
+      "learning_rate": 2e-05,
+      "loss": 7.5266,
+      "step": 8190
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 15.046711921691895,
+      "learning_rate": 2e-05,
+      "loss": 7.4026,
+      "step": 8200
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 14.296401977539062,
+      "learning_rate": 2e-05,
+      "loss": 7.314,
+      "step": 8210
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 14.766364097595215,
+      "learning_rate": 2e-05,
+      "loss": 7.4446,
+      "step": 8220
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 13.495145797729492,
+      "learning_rate": 2e-05,
+      "loss": 7.4522,
+      "step": 8230
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 13.117164611816406,
+      "learning_rate": 2e-05,
+      "loss": 7.1438,
+      "step": 8240
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 13.154569625854492,
+      "learning_rate": 2e-05,
+      "loss": 7.3664,
+      "step": 8250
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 12.642875671386719,
+      "learning_rate": 2e-05,
+      "loss": 7.355,
+      "step": 8260
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 13.021675109863281,
+      "learning_rate": 2e-05,
+      "loss": 7.3503,
+      "step": 8270
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 13.33282470703125,
+      "learning_rate": 2e-05,
+      "loss": 7.3221,
+      "step": 8280
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 12.603179931640625,
+      "learning_rate": 2e-05,
+      "loss": 7.2909,
+      "step": 8290
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 14.890718460083008,
+      "learning_rate": 2e-05,
+      "loss": 7.2152,
+      "step": 8300
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 13.281075477600098,
+      "learning_rate": 2e-05,
+      "loss": 7.3991,
+      "step": 8310
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 13.560399055480957,
+      "learning_rate": 2e-05,
+      "loss": 7.2459,
+      "step": 8320
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 13.2423095703125,
+      "learning_rate": 2e-05,
+      "loss": 7.2792,
+      "step": 8330
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 15.665305137634277,
+      "learning_rate": 2e-05,
+      "loss": 7.4538,
+      "step": 8340
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 14.071041107177734,
+      "learning_rate": 2e-05,
+      "loss": 7.3747,
+      "step": 8350
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 11.5364990234375,
+      "learning_rate": 2e-05,
+      "loss": 7.2958,
+      "step": 8360
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 14.473661422729492,
+      "learning_rate": 2e-05,
+      "loss": 7.4251,
+      "step": 8370
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 13.102906227111816,
+      "learning_rate": 2e-05,
+      "loss": 7.3323,
+      "step": 8380
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 15.513550758361816,
+      "learning_rate": 2e-05,
+      "loss": 7.4031,
+      "step": 8390
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 13.744675636291504,
+      "learning_rate": 2e-05,
+      "loss": 7.2792,
+      "step": 8400
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 12.821061134338379,
+      "learning_rate": 2e-05,
+      "loss": 7.2553,
+      "step": 8410
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 13.858121871948242,
+      "learning_rate": 2e-05,
+      "loss": 7.355,
+      "step": 8420
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 13.039094924926758,
+      "learning_rate": 2e-05,
+      "loss": 7.1756,
+      "step": 8430
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 14.7315034866333,
+      "learning_rate": 2e-05,
+      "loss": 7.3312,
+      "step": 8440
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 14.9122314453125,
+      "learning_rate": 2e-05,
+      "loss": 7.3931,
+      "step": 8450
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 13.226001739501953,
+      "learning_rate": 2e-05,
+      "loss": 7.2189,
+      "step": 8460
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 14.39482307434082,
+      "learning_rate": 2e-05,
+      "loss": 7.1974,
+      "step": 8470
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 14.118139266967773,
+      "learning_rate": 2e-05,
+      "loss": 7.3022,
+      "step": 8480
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 15.14077377319336,
+      "learning_rate": 2e-05,
+      "loss": 7.2694,
+      "step": 8490
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 14.233264923095703,
+      "learning_rate": 2e-05,
+      "loss": 7.4207,
+      "step": 8500
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 13.734562873840332,
+      "learning_rate": 2e-05,
+      "loss": 7.3134,
+      "step": 8510
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 13.860827445983887,
+      "learning_rate": 2e-05,
+      "loss": 7.1741,
+      "step": 8520
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 14.363543510437012,
+      "learning_rate": 2e-05,
+      "loss": 7.4492,
+      "step": 8530
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 14.604022979736328,
+      "learning_rate": 2e-05,
+      "loss": 7.4138,
+      "step": 8540
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 15.0270357131958,
+      "learning_rate": 2e-05,
+      "loss": 7.2271,
+      "step": 8550
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 13.6764497756958,
+      "learning_rate": 2e-05,
+      "loss": 7.3031,
+      "step": 8560
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 14.23155403137207,
+      "learning_rate": 2e-05,
+      "loss": 7.2749,
+      "step": 8570
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 14.137436866760254,
+      "learning_rate": 2e-05,
+      "loss": 7.403,
+      "step": 8580
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 13.979974746704102,
+      "learning_rate": 2e-05,
+      "loss": 7.1409,
+      "step": 8590
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 16.712656021118164,
+      "learning_rate": 2e-05,
+      "loss": 7.3323,
+      "step": 8600
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 12.402235984802246,
+      "learning_rate": 2e-05,
+      "loss": 7.2525,
+      "step": 8610
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 14.992640495300293,
+      "learning_rate": 2e-05,
+      "loss": 7.2808,
+      "step": 8620
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 13.278471946716309,
+      "learning_rate": 2e-05,
+      "loss": 7.2311,
+      "step": 8630
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 14.911993026733398,
+      "learning_rate": 2e-05,
+      "loss": 7.3961,
+      "step": 8640
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 14.609057426452637,
+      "learning_rate": 2e-05,
+      "loss": 7.2808,
+      "step": 8650
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 14.923026084899902,
+      "learning_rate": 2e-05,
+      "loss": 7.2456,
+      "step": 8660
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 15.779901504516602,
+      "learning_rate": 2e-05,
+      "loss": 7.4038,
+      "step": 8670
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 13.297290802001953,
+      "learning_rate": 2e-05,
+      "loss": 7.3738,
+      "step": 8680
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 13.297189712524414,
+      "learning_rate": 2e-05,
+      "loss": 7.2709,
+      "step": 8690
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 13.342057228088379,
+      "learning_rate": 2e-05,
+      "loss": 7.2581,
+      "step": 8700
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 13.540092468261719,
+      "learning_rate": 2e-05,
+      "loss": 7.4574,
+      "step": 8710
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 15.104598045349121,
+      "learning_rate": 2e-05,
+      "loss": 7.2432,
+      "step": 8720
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 13.733980178833008,
+      "learning_rate": 2e-05,
+      "loss": 7.3227,
+      "step": 8730
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 13.771142959594727,
+      "learning_rate": 2e-05,
+      "loss": 7.2731,
+      "step": 8740
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 14.466370582580566,
+      "learning_rate": 2e-05,
+      "loss": 7.4363,
+      "step": 8750
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 14.393912315368652,
+      "learning_rate": 2e-05,
+      "loss": 7.3483,
+      "step": 8760
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 13.25371265411377,
+      "learning_rate": 2e-05,
+      "loss": 7.4486,
+      "step": 8770
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 15.148337364196777,
+      "learning_rate": 2e-05,
+      "loss": 7.4591,
+      "step": 8780
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 14.952364921569824,
+      "learning_rate": 2e-05,
+      "loss": 7.3735,
+      "step": 8790
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 14.581990242004395,
+      "learning_rate": 2e-05,
+      "loss": 7.1917,
+      "step": 8800
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 14.454474449157715,
+      "learning_rate": 2e-05,
+      "loss": 7.2739,
+      "step": 8810
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 12.220720291137695,
+      "learning_rate": 2e-05,
+      "loss": 7.1352,
+      "step": 8820
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 13.840889930725098,
+      "learning_rate": 2e-05,
+      "loss": 7.1518,
+      "step": 8830
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 13.629042625427246,
+      "learning_rate": 2e-05,
+      "loss": 7.3426,
+      "step": 8840
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 13.80989933013916,
+      "learning_rate": 2e-05,
+      "loss": 7.123,
+      "step": 8850
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 13.11877155303955,
+      "learning_rate": 2e-05,
+      "loss": 7.355,
+      "step": 8860
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 14.269912719726562,
+      "learning_rate": 2e-05,
+      "loss": 7.3688,
+      "step": 8870
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 15.53414249420166,
+      "learning_rate": 2e-05,
+      "loss": 7.2993,
+      "step": 8880
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 13.83571720123291,
+      "learning_rate": 2e-05,
+      "loss": 7.2616,
+      "step": 8890
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 14.263980865478516,
+      "learning_rate": 2e-05,
+      "loss": 7.282,
+      "step": 8900
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 16.473045349121094,
+      "learning_rate": 2e-05,
+      "loss": 7.3862,
+      "step": 8910
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 13.581949234008789,
+      "learning_rate": 2e-05,
+      "loss": 7.527,
+      "step": 8920
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 15.920560836791992,
+      "learning_rate": 2e-05,
+      "loss": 7.3198,
+      "step": 8930
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 12.954968452453613,
+      "learning_rate": 2e-05,
+      "loss": 7.2541,
+      "step": 8940
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 14.22775936126709,
+      "learning_rate": 2e-05,
+      "loss": 7.2439,
+      "step": 8950
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 13.780340194702148,
+      "learning_rate": 2e-05,
+      "loss": 7.1994,
+      "step": 8960
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 13.404021263122559,
+      "learning_rate": 2e-05,
+      "loss": 7.1816,
+      "step": 8970
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 14.422134399414062,
+      "learning_rate": 2e-05,
+      "loss": 7.2719,
+      "step": 8980
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 14.367630958557129,
+      "learning_rate": 2e-05,
+      "loss": 7.2786,
+      "step": 8990
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 14.694226264953613,
+      "learning_rate": 2e-05,
+      "loss": 7.2843,
+      "step": 9000
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 15.477653503417969,
+      "learning_rate": 2e-05,
+      "loss": 7.2732,
+      "step": 9010
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 13.62797737121582,
+      "learning_rate": 2e-05,
+      "loss": 7.1441,
+      "step": 9020
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 15.645630836486816,
+      "learning_rate": 2e-05,
+      "loss": 7.2342,
+      "step": 9030
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 15.049473762512207,
+      "learning_rate": 2e-05,
+      "loss": 7.2413,
+      "step": 9040
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 13.369446754455566,
+      "learning_rate": 2e-05,
+      "loss": 7.2658,
+      "step": 9050
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 13.821754455566406,
+      "learning_rate": 2e-05,
+      "loss": 7.2725,
+      "step": 9060
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 15.101029396057129,
+      "learning_rate": 2e-05,
+      "loss": 7.3177,
+      "step": 9070
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 16.214996337890625,
+      "learning_rate": 2e-05,
+      "loss": 7.2465,
+      "step": 9080
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 14.33379077911377,
+      "learning_rate": 2e-05,
+      "loss": 7.1242,
+      "step": 9090
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 14.77404499053955,
+      "learning_rate": 2e-05,
+      "loss": 7.2581,
+      "step": 9100
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 14.745279312133789,
+      "learning_rate": 2e-05,
+      "loss": 7.3988,
+      "step": 9110
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 14.430118560791016,
+      "learning_rate": 2e-05,
+      "loss": 7.3423,
+      "step": 9120
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 14.131326675415039,
+      "learning_rate": 2e-05,
+      "loss": 7.2911,
+      "step": 9130
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 13.840328216552734,
+      "learning_rate": 2e-05,
+      "loss": 7.2111,
+      "step": 9140
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 15.260168075561523,
+      "learning_rate": 2e-05,
+      "loss": 7.2334,
+      "step": 9150
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 14.34538745880127,
+      "learning_rate": 2e-05,
+      "loss": 7.209,
+      "step": 9160
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 13.442482948303223,
+      "learning_rate": 2e-05,
+      "loss": 7.254,
+      "step": 9170
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 14.495719909667969,
+      "learning_rate": 2e-05,
+      "loss": 7.1489,
+      "step": 9180
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 14.710139274597168,
+      "learning_rate": 2e-05,
+      "loss": 7.4206,
+      "step": 9190
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 13.333538055419922,
+      "learning_rate": 2e-05,
+      "loss": 7.2544,
+      "step": 9200
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 14.066051483154297,
+      "learning_rate": 2e-05,
+      "loss": 6.9927,
+      "step": 9210
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 15.206609725952148,
+      "learning_rate": 2e-05,
+      "loss": 7.1829,
+      "step": 9220
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 14.185057640075684,
+      "learning_rate": 2e-05,
+      "loss": 7.213,
+      "step": 9230
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 13.9644775390625,
+      "learning_rate": 2e-05,
+      "loss": 7.389,
+      "step": 9240
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 14.814619064331055,
+      "learning_rate": 2e-05,
+      "loss": 7.2852,
+      "step": 9250
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 14.526883125305176,
+      "learning_rate": 2e-05,
+      "loss": 7.363,
+      "step": 9260
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 12.601645469665527,
+      "learning_rate": 2e-05,
+      "loss": 7.2127,
+      "step": 9270
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 13.036627769470215,
+      "learning_rate": 2e-05,
+      "loss": 7.2525,
+      "step": 9280
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 11.950976371765137,
+      "learning_rate": 2e-05,
+      "loss": 7.1921,
+      "step": 9290
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 14.716158866882324,
+      "learning_rate": 2e-05,
+      "loss": 7.2488,
+      "step": 9300
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 14.15019702911377,
+      "learning_rate": 2e-05,
+      "loss": 7.2514,
+      "step": 9310
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 13.270162582397461,
+      "learning_rate": 2e-05,
+      "loss": 7.2037,
+      "step": 9320
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 14.31289005279541,
+      "learning_rate": 2e-05,
+      "loss": 7.2848,
+      "step": 9330
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 16.287261962890625,
+      "learning_rate": 2e-05,
+      "loss": 7.1395,
+      "step": 9340
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 16.439550399780273,
+      "learning_rate": 2e-05,
+      "loss": 7.3133,
+      "step": 9350
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 13.011539459228516,
+      "learning_rate": 2e-05,
+      "loss": 7.1952,
+      "step": 9360
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 12.626483917236328,
+      "learning_rate": 2e-05,
+      "loss": 7.0183,
+      "step": 9370
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 15.876221656799316,
+      "learning_rate": 2e-05,
+      "loss": 7.2612,
+      "step": 9380
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 14.129226684570312,
+      "learning_rate": 2e-05,
+      "loss": 7.1869,
+      "step": 9390
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 14.763731002807617,
+      "learning_rate": 2e-05,
+      "loss": 7.2349,
+      "step": 9400
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 15.852397918701172,
+      "learning_rate": 2e-05,
+      "loss": 7.2579,
+      "step": 9410
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 16.266401290893555,
+      "learning_rate": 2e-05,
+      "loss": 7.1627,
+      "step": 9420
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 13.075157165527344,
+      "learning_rate": 2e-05,
+      "loss": 7.293,
+      "step": 9430
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 14.038750648498535,
+      "learning_rate": 2e-05,
+      "loss": 7.1767,
+      "step": 9440
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 16.25745391845703,
+      "learning_rate": 2e-05,
+      "loss": 7.2434,
+      "step": 9450
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 14.916772842407227,
+      "learning_rate": 2e-05,
+      "loss": 7.1311,
+      "step": 9460
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 14.690471649169922,
+      "learning_rate": 2e-05,
+      "loss": 7.2043,
+      "step": 9470
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 16.873512268066406,
+      "learning_rate": 2e-05,
+      "loss": 7.1767,
+      "step": 9480
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 14.680343627929688,
+      "learning_rate": 2e-05,
+      "loss": 7.0155,
+      "step": 9490
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 14.314539909362793,
+      "learning_rate": 2e-05,
+      "loss": 7.1677,
+      "step": 9500
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 14.74007511138916,
+      "learning_rate": 2e-05,
+      "loss": 7.225,
+      "step": 9510
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 14.37302017211914,
+      "learning_rate": 2e-05,
+      "loss": 7.3669,
+      "step": 9520
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 15.46247386932373,
+      "learning_rate": 2e-05,
+      "loss": 7.2948,
+      "step": 9530
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 14.08154010772705,
+      "learning_rate": 2e-05,
+      "loss": 7.1503,
+      "step": 9540
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 15.056352615356445,
+      "learning_rate": 2e-05,
+      "loss": 7.2257,
+      "step": 9550
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 13.64146614074707,
+      "learning_rate": 2e-05,
+      "loss": 7.2584,
+      "step": 9560
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 15.102680206298828,
+      "learning_rate": 2e-05,
+      "loss": 7.1548,
+      "step": 9570
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 13.658087730407715,
+      "learning_rate": 2e-05,
+      "loss": 7.0557,
+      "step": 9580
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 12.820012092590332,
+      "learning_rate": 2e-05,
+      "loss": 7.1369,
+      "step": 9590
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 15.444236755371094,
+      "learning_rate": 2e-05,
+      "loss": 7.1988,
+      "step": 9600
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 14.140412330627441,
+      "learning_rate": 2e-05,
+      "loss": 7.1541,
+      "step": 9610
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 13.730337142944336,
+      "learning_rate": 2e-05,
+      "loss": 7.2285,
+      "step": 9620
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 14.727079391479492,
+      "learning_rate": 2e-05,
+      "loss": 7.1747,
+      "step": 9630
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 14.011664390563965,
+      "learning_rate": 2e-05,
+      "loss": 7.1918,
+      "step": 9640
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 13.813437461853027,
+      "learning_rate": 2e-05,
+      "loss": 7.2218,
+      "step": 9650
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 13.742414474487305,
+      "learning_rate": 2e-05,
+      "loss": 7.1548,
+      "step": 9660
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 15.848419189453125,
+      "learning_rate": 2e-05,
+      "loss": 7.1055,
+      "step": 9670
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 16.324819564819336,
+      "learning_rate": 2e-05,
+      "loss": 7.1707,
+      "step": 9680
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 13.762989044189453,
+      "learning_rate": 2e-05,
+      "loss": 7.0618,
+      "step": 9690
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 14.229901313781738,
+      "learning_rate": 2e-05,
+      "loss": 7.0334,
+      "step": 9700
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 14.125149726867676,
+      "learning_rate": 2e-05,
+      "loss": 7.0846,
+      "step": 9710
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 15.20899772644043,
+      "learning_rate": 2e-05,
+      "loss": 7.2806,
+      "step": 9720
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 14.617901802062988,
+      "learning_rate": 2e-05,
+      "loss": 7.2284,
+      "step": 9730
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 14.899394989013672,
+      "learning_rate": 2e-05,
+      "loss": 7.2158,
+      "step": 9740
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 14.098103523254395,
+      "learning_rate": 2e-05,
+      "loss": 7.1235,
+      "step": 9750
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 15.658750534057617,
+      "learning_rate": 2e-05,
+      "loss": 7.2769,
+      "step": 9760
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 13.667555809020996,
+      "learning_rate": 2e-05,
+      "loss": 7.1281,
+      "step": 9770
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 13.4503755569458,
+      "learning_rate": 2e-05,
+      "loss": 7.1962,
+      "step": 9780
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 13.403559684753418,
+      "learning_rate": 2e-05,
+      "loss": 7.1575,
+      "step": 9790
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 14.735502243041992,
+      "learning_rate": 2e-05,
+      "loss": 7.052,
+      "step": 9800
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 15.967671394348145,
+      "learning_rate": 2e-05,
+      "loss": 7.102,
+      "step": 9810
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 12.678009033203125,
+      "learning_rate": 2e-05,
+      "loss": 7.0514,
+      "step": 9820
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 15.478487014770508,
+      "learning_rate": 2e-05,
+      "loss": 7.2716,
+      "step": 9830
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 15.451637268066406,
+      "learning_rate": 2e-05,
+      "loss": 7.2591,
+      "step": 9840
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 14.206971168518066,
+      "learning_rate": 2e-05,
+      "loss": 7.1123,
+      "step": 9850
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 13.925130844116211,
+      "learning_rate": 2e-05,
+      "loss": 7.195,
+      "step": 9860
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 14.823182106018066,
+      "learning_rate": 2e-05,
+      "loss": 7.2746,
+      "step": 9870
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 15.28787612915039,
+      "learning_rate": 2e-05,
+      "loss": 7.2418,
+      "step": 9880
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 14.715753555297852,
+      "learning_rate": 2e-05,
+      "loss": 7.1597,
+      "step": 9890
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 15.041627883911133,
+      "learning_rate": 2e-05,
+      "loss": 7.1147,
+      "step": 9900
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 14.809428215026855,
+      "learning_rate": 2e-05,
+      "loss": 7.1718,
+      "step": 9910
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 14.57148551940918,
+      "learning_rate": 2e-05,
+      "loss": 7.241,
+      "step": 9920
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 13.665536880493164,
+      "learning_rate": 2e-05,
+      "loss": 7.2257,
+      "step": 9930
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 13.304644584655762,
+      "learning_rate": 2e-05,
+      "loss": 7.0582,
+      "step": 9940
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 15.122304916381836,
+      "learning_rate": 2e-05,
+      "loss": 7.1202,
+      "step": 9950
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 15.708093643188477,
+      "learning_rate": 2e-05,
+      "loss": 7.1674,
+      "step": 9960
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 13.809382438659668,
+      "learning_rate": 2e-05,
+      "loss": 7.1311,
+      "step": 9970
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 14.36973762512207,
+      "learning_rate": 2e-05,
+      "loss": 7.1884,
+      "step": 9980
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 14.45726490020752,
+      "learning_rate": 2e-05,
+      "loss": 6.99,
+      "step": 9990
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 14.253656387329102,
+      "learning_rate": 2e-05,
+      "loss": 7.2002,
+      "step": 10000
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 15.209531784057617,
+      "learning_rate": 2e-05,
+      "loss": 7.0931,
+      "step": 10010
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 14.692426681518555,
+      "learning_rate": 2e-05,
+      "loss": 7.1067,
+      "step": 10020
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 15.321356773376465,
+      "learning_rate": 2e-05,
+      "loss": 7.1725,
+      "step": 10030
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 15.052484512329102,
+      "learning_rate": 2e-05,
+      "loss": 7.2853,
+      "step": 10040
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 15.900465965270996,
+      "learning_rate": 2e-05,
+      "loss": 7.0967,
+      "step": 10050
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 14.1983642578125,
+      "learning_rate": 2e-05,
+      "loss": 7.2274,
+      "step": 10060
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 13.799038887023926,
+      "learning_rate": 2e-05,
+      "loss": 7.0582,
+      "step": 10070
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 13.682351112365723,
+      "learning_rate": 2e-05,
+      "loss": 6.9834,
+      "step": 10080
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 16.271533966064453,
+      "learning_rate": 2e-05,
+      "loss": 7.2204,
+      "step": 10090
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 13.445121765136719,
+      "learning_rate": 2e-05,
+      "loss": 6.8777,
+      "step": 10100
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 15.068289756774902,
+      "learning_rate": 2e-05,
+      "loss": 7.1331,
+      "step": 10110
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 13.262035369873047,
+      "learning_rate": 2e-05,
+      "loss": 7.1075,
+      "step": 10120
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 14.457252502441406,
+      "learning_rate": 2e-05,
+      "loss": 7.0949,
+      "step": 10130
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 15.220915794372559,
+      "learning_rate": 2e-05,
+      "loss": 7.1662,
+      "step": 10140
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 14.160903930664062,
+      "learning_rate": 2e-05,
+      "loss": 7.0725,
+      "step": 10150
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 14.885871887207031,
+      "learning_rate": 2e-05,
+      "loss": 7.114,
+      "step": 10160
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 17.431228637695312,
+      "learning_rate": 2e-05,
+      "loss": 6.9578,
+      "step": 10170
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 13.35297679901123,
+      "learning_rate": 2e-05,
+      "loss": 7.1717,
+      "step": 10180
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 14.317094802856445,
+      "learning_rate": 2e-05,
+      "loss": 7.1785,
+      "step": 10190
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 13.305404663085938,
+      "learning_rate": 2e-05,
+      "loss": 7.1076,
+      "step": 10200
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 13.791671752929688,
+      "learning_rate": 2e-05,
+      "loss": 7.1017,
+      "step": 10210
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 14.001263618469238,
+      "learning_rate": 2e-05,
+      "loss": 7.127,
+      "step": 10220
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 17.6578426361084,
+      "learning_rate": 2e-05,
+      "loss": 7.2282,
+      "step": 10230
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 16.051616668701172,
+      "learning_rate": 2e-05,
+      "loss": 7.3023,
+      "step": 10240
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 14.411608695983887,
+      "learning_rate": 2e-05,
+      "loss": 7.0761,
+      "step": 10250
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 13.683361053466797,
+      "learning_rate": 2e-05,
+      "loss": 7.0778,
+      "step": 10260
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 16.05841636657715,
+      "learning_rate": 2e-05,
+      "loss": 7.1004,
+      "step": 10270
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 15.773926734924316,
+      "learning_rate": 2e-05,
+      "loss": 7.0444,
+      "step": 10280
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 13.373499870300293,
+      "learning_rate": 2e-05,
+      "loss": 6.8166,
+      "step": 10290
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 15.623120307922363,
+      "learning_rate": 2e-05,
+      "loss": 7.1087,
+      "step": 10300
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 13.843756675720215,
+      "learning_rate": 2e-05,
+      "loss": 6.9776,
+      "step": 10310
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 15.322917938232422,
+      "learning_rate": 2e-05,
+      "loss": 7.2731,
+      "step": 10320
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 14.351551055908203,
+      "learning_rate": 2e-05,
+      "loss": 7.1794,
+      "step": 10330
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 13.059287071228027,
+      "learning_rate": 2e-05,
+      "loss": 6.9283,
+      "step": 10340
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 14.376687049865723,
+      "learning_rate": 2e-05,
+      "loss": 7.11,
+      "step": 10350
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 16.910690307617188,
+      "learning_rate": 2e-05,
+      "loss": 7.1882,
+      "step": 10360
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 14.397239685058594,
+      "learning_rate": 2e-05,
+      "loss": 7.0278,
+      "step": 10370
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 14.211201667785645,
+      "learning_rate": 2e-05,
+      "loss": 7.131,
+      "step": 10380
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 13.709233283996582,
+      "learning_rate": 2e-05,
+      "loss": 7.1,
+      "step": 10390
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 14.833255767822266,
+      "learning_rate": 2e-05,
+      "loss": 7.1809,
+      "step": 10400
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 13.253549575805664,
+      "learning_rate": 2e-05,
+      "loss": 7.1713,
+      "step": 10410
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 16.45342445373535,
+      "learning_rate": 2e-05,
+      "loss": 7.1969,
+      "step": 10420
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 14.965041160583496,
+      "learning_rate": 2e-05,
+      "loss": 7.0542,
+      "step": 10430
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 14.657515525817871,
+      "learning_rate": 2e-05,
+      "loss": 7.1816,
+      "step": 10440
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 17.24167251586914,
+      "learning_rate": 2e-05,
+      "loss": 7.0011,
+      "step": 10450
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 14.447762489318848,
+      "learning_rate": 2e-05,
+      "loss": 6.9754,
+      "step": 10460
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 14.063021659851074,
+      "learning_rate": 2e-05,
+      "loss": 7.0122,
+      "step": 10470
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 14.59058666229248,
+      "learning_rate": 2e-05,
+      "loss": 7.0037,
+      "step": 10480
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 17.353010177612305,
+      "learning_rate": 2e-05,
+      "loss": 7.1145,
+      "step": 10490
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 13.999716758728027,
+      "learning_rate": 2e-05,
+      "loss": 6.9757,
+      "step": 10500
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 16.959611892700195,
+      "learning_rate": 2e-05,
+      "loss": 6.9572,
+      "step": 10510
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 13.240046501159668,
+      "learning_rate": 2e-05,
+      "loss": 6.9513,
+      "step": 10520
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 13.166268348693848,
+      "learning_rate": 2e-05,
+      "loss": 6.9942,
+      "step": 10530
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 13.161182403564453,
+      "learning_rate": 2e-05,
+      "loss": 7.0683,
+      "step": 10540
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 15.460906982421875,
+      "learning_rate": 2e-05,
+      "loss": 7.0425,
+      "step": 10550
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 12.809261322021484,
+      "learning_rate": 2e-05,
+      "loss": 6.9554,
+      "step": 10560
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 15.04209041595459,
+      "learning_rate": 2e-05,
+      "loss": 7.1921,
+      "step": 10570
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 14.61618709564209,
+      "learning_rate": 2e-05,
+      "loss": 7.0486,
+      "step": 10580
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 13.282368659973145,
+      "learning_rate": 2e-05,
+      "loss": 7.005,
+      "step": 10590
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 14.702110290527344,
+      "learning_rate": 2e-05,
+      "loss": 6.9927,
+      "step": 10600
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 14.720727920532227,
+      "learning_rate": 2e-05,
+      "loss": 7.0138,
+      "step": 10610
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 14.89784049987793,
+      "learning_rate": 2e-05,
+      "loss": 7.0469,
+      "step": 10620
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 17.568721771240234,
+      "learning_rate": 2e-05,
+      "loss": 6.9354,
+      "step": 10630
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 16.76685905456543,
+      "learning_rate": 2e-05,
+      "loss": 7.0193,
+      "step": 10640
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 14.571706771850586,
+      "learning_rate": 2e-05,
+      "loss": 6.962,
+      "step": 10650
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 14.616537094116211,
+      "learning_rate": 2e-05,
+      "loss": 6.9553,
+      "step": 10660
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 15.909575462341309,
+      "learning_rate": 2e-05,
+      "loss": 6.9223,
+      "step": 10670
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 16.72798728942871,
+      "learning_rate": 2e-05,
+      "loss": 6.9661,
+      "step": 10680
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 14.47726058959961,
+      "learning_rate": 2e-05,
+      "loss": 6.9752,
+      "step": 10690
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 12.774328231811523,
+      "learning_rate": 2e-05,
+      "loss": 6.9621,
+      "step": 10700
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 15.479938507080078,
+      "learning_rate": 2e-05,
+      "loss": 7.2177,
+      "step": 10710
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 13.66833782196045,
+      "learning_rate": 2e-05,
+      "loss": 7.1643,
+      "step": 10720
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 13.47901439666748,
+      "learning_rate": 2e-05,
+      "loss": 7.0873,
+      "step": 10730
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 15.435907363891602,
+      "learning_rate": 2e-05,
+      "loss": 6.9927,
+      "step": 10740
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 13.699430465698242,
+      "learning_rate": 2e-05,
+      "loss": 7.0351,
+      "step": 10750
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 15.520047187805176,
+      "learning_rate": 2e-05,
+      "loss": 6.9714,
+      "step": 10760
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 15.034466743469238,
+      "learning_rate": 2e-05,
+      "loss": 7.2242,
+      "step": 10770
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 16.278736114501953,
+      "learning_rate": 2e-05,
+      "loss": 7.0873,
+      "step": 10780
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 17.27407455444336,
+      "learning_rate": 2e-05,
+      "loss": 7.1495,
+      "step": 10790
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 16.735618591308594,
+      "learning_rate": 2e-05,
+      "loss": 6.8887,
+      "step": 10800
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 14.617740631103516,
+      "learning_rate": 2e-05,
+      "loss": 6.931,
+      "step": 10810
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 15.052382469177246,
+      "learning_rate": 2e-05,
+      "loss": 7.0619,
+      "step": 10820
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 14.139616966247559,
+      "learning_rate": 2e-05,
+      "loss": 6.9084,
+      "step": 10830
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 13.337227821350098,
+      "learning_rate": 2e-05,
+      "loss": 7.0175,
+      "step": 10840
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 16.25478172302246,
+      "learning_rate": 2e-05,
+      "loss": 7.0116,
+      "step": 10850
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 15.871460914611816,
+      "learning_rate": 2e-05,
+      "loss": 6.9619,
+      "step": 10860
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 14.824347496032715,
+      "learning_rate": 2e-05,
+      "loss": 6.9763,
+      "step": 10870
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 14.589641571044922,
+      "learning_rate": 2e-05,
+      "loss": 7.0436,
+      "step": 10880
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 13.422178268432617,
+      "learning_rate": 2e-05,
+      "loss": 6.9,
+      "step": 10890
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 13.11312198638916,
+      "learning_rate": 2e-05,
+      "loss": 7.0883,
+      "step": 10900
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 14.24538803100586,
+      "learning_rate": 2e-05,
+      "loss": 6.8091,
+      "step": 10910
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 14.958952903747559,
+      "learning_rate": 2e-05,
+      "loss": 6.9424,
+      "step": 10920
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 14.55583667755127,
+      "learning_rate": 2e-05,
+      "loss": 7.0455,
+      "step": 10930
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 15.322731018066406,
+      "learning_rate": 2e-05,
+      "loss": 6.9517,
+      "step": 10940
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 13.94517993927002,
+      "learning_rate": 2e-05,
+      "loss": 7.1261,
+      "step": 10950
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 13.752155303955078,
+      "learning_rate": 2e-05,
+      "loss": 6.7571,
+      "step": 10960
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 15.281647682189941,
+      "learning_rate": 2e-05,
+      "loss": 7.0626,
+      "step": 10970
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 15.322835922241211,
+      "learning_rate": 2e-05,
+      "loss": 7.1818,
+      "step": 10980
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 13.11985969543457,
+      "learning_rate": 2e-05,
+      "loss": 7.1471,
+      "step": 10990
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 16.183074951171875,
+      "learning_rate": 2e-05,
+      "loss": 6.8988,
+      "step": 11000
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 14.592451095581055,
+      "learning_rate": 2e-05,
+      "loss": 6.8718,
+      "step": 11010
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 13.284855842590332,
+      "learning_rate": 2e-05,
+      "loss": 7.0189,
+      "step": 11020
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 14.399516105651855,
+      "learning_rate": 2e-05,
+      "loss": 6.9765,
+      "step": 11030
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 13.141850471496582,
+      "learning_rate": 2e-05,
+      "loss": 6.9499,
+      "step": 11040
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 14.262334823608398,
+      "learning_rate": 2e-05,
+      "loss": 6.8391,
+      "step": 11050
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 14.662503242492676,
+      "learning_rate": 2e-05,
+      "loss": 7.044,
+      "step": 11060
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 16.206960678100586,
+      "learning_rate": 2e-05,
+      "loss": 6.9539,
+      "step": 11070
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 15.762690544128418,
+      "learning_rate": 2e-05,
+      "loss": 6.9445,
+      "step": 11080
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 16.584928512573242,
+      "learning_rate": 2e-05,
+      "loss": 7.0006,
+      "step": 11090
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 14.847896575927734,
+      "learning_rate": 2e-05,
+      "loss": 6.9489,
+      "step": 11100
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 15.938279151916504,
+      "learning_rate": 2e-05,
+      "loss": 6.9821,
+      "step": 11110
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 13.682432174682617,
+      "learning_rate": 2e-05,
+      "loss": 6.8394,
+      "step": 11120
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 13.38754940032959,
+      "learning_rate": 2e-05,
+      "loss": 6.9907,
+      "step": 11130
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 12.975614547729492,
+      "learning_rate": 2e-05,
+      "loss": 7.0663,
+      "step": 11140
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 16.868425369262695,
+      "learning_rate": 2e-05,
+      "loss": 7.1078,
+      "step": 11150
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 15.341955184936523,
+      "learning_rate": 2e-05,
+      "loss": 6.6717,
+      "step": 11160
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 15.533119201660156,
+      "learning_rate": 2e-05,
+      "loss": 7.071,
+      "step": 11170
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 14.350326538085938,
+      "learning_rate": 2e-05,
+      "loss": 6.9514,
+      "step": 11180
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 14.948601722717285,
+      "learning_rate": 2e-05,
+      "loss": 6.9775,
+      "step": 11190
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 16.777021408081055,
+      "learning_rate": 2e-05,
+      "loss": 7.0809,
+      "step": 11200
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 13.495430946350098,
+      "learning_rate": 2e-05,
+      "loss": 7.1081,
+      "step": 11210
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 14.736259460449219,
+      "learning_rate": 2e-05,
+      "loss": 6.829,
+      "step": 11220
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 15.131936073303223,
+      "learning_rate": 2e-05,
+      "loss": 7.0527,
+      "step": 11230
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 14.989665985107422,
+      "learning_rate": 2e-05,
+      "loss": 7.0044,
+      "step": 11240
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 14.380468368530273,
+      "learning_rate": 2e-05,
+      "loss": 6.8567,
+      "step": 11250
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 14.907323837280273,
+      "learning_rate": 2e-05,
+      "loss": 7.059,
+      "step": 11260
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 14.06887435913086,
+      "learning_rate": 2e-05,
+      "loss": 6.9637,
+      "step": 11270
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 14.379768371582031,
+      "learning_rate": 2e-05,
+      "loss": 6.8882,
+      "step": 11280
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 12.601763725280762,
+      "learning_rate": 2e-05,
+      "loss": 6.934,
+      "step": 11290
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 14.22280216217041,
+      "learning_rate": 2e-05,
+      "loss": 6.9432,
+      "step": 11300
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 13.595396995544434,
+      "learning_rate": 2e-05,
+      "loss": 7.0407,
+      "step": 11310
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 16.092321395874023,
+      "learning_rate": 2e-05,
+      "loss": 6.9874,
+      "step": 11320
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 13.295510292053223,
+      "learning_rate": 2e-05,
+      "loss": 7.1539,
+      "step": 11330
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 14.64439868927002,
+      "learning_rate": 2e-05,
+      "loss": 7.0602,
+      "step": 11340
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 13.792032241821289,
+      "learning_rate": 2e-05,
+      "loss": 7.0413,
+      "step": 11350
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 16.125415802001953,
+      "learning_rate": 2e-05,
+      "loss": 6.9962,
+      "step": 11360
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 17.33074378967285,
+      "learning_rate": 2e-05,
+      "loss": 6.9098,
+      "step": 11370
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 13.874446868896484,
+      "learning_rate": 2e-05,
+      "loss": 6.9042,
+      "step": 11380
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 15.442166328430176,
+      "learning_rate": 2e-05,
+      "loss": 6.8695,
+      "step": 11390
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 13.640735626220703,
+      "learning_rate": 2e-05,
+      "loss": 6.7693,
+      "step": 11400
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 13.442179679870605,
+      "learning_rate": 2e-05,
+      "loss": 7.0266,
+      "step": 11410
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 14.3765287399292,
+      "learning_rate": 2e-05,
+      "loss": 6.7023,
+      "step": 11420
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 14.431687355041504,
+      "learning_rate": 2e-05,
+      "loss": 7.1396,
+      "step": 11430
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 13.380819320678711,
+      "learning_rate": 2e-05,
+      "loss": 6.8443,
+      "step": 11440
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 15.129609107971191,
+      "learning_rate": 2e-05,
+      "loss": 6.895,
+      "step": 11450
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 13.047367095947266,
+      "learning_rate": 2e-05,
+      "loss": 6.7494,
+      "step": 11460
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 15.060064315795898,
+      "learning_rate": 2e-05,
+      "loss": 6.7929,
+      "step": 11470
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 14.912361145019531,
+      "learning_rate": 2e-05,
+      "loss": 7.0019,
+      "step": 11480
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 14.848867416381836,
+      "learning_rate": 2e-05,
+      "loss": 6.9764,
+      "step": 11490
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 12.468055725097656,
+      "learning_rate": 2e-05,
+      "loss": 7.0432,
+      "step": 11500
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 14.945636749267578,
+      "learning_rate": 2e-05,
+      "loss": 6.8918,
+      "step": 11510
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 14.663524627685547,
+      "learning_rate": 2e-05,
+      "loss": 6.9476,
+      "step": 11520
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 14.093116760253906,
+      "learning_rate": 2e-05,
+      "loss": 7.0069,
+      "step": 11530
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 14.418761253356934,
+      "learning_rate": 2e-05,
+      "loss": 7.0597,
+      "step": 11540
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 14.965607643127441,
+      "learning_rate": 2e-05,
+      "loss": 7.0054,
+      "step": 11550
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 14.131927490234375,
+      "learning_rate": 2e-05,
+      "loss": 6.9689,
+      "step": 11560
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 14.277298927307129,
+      "learning_rate": 2e-05,
+      "loss": 6.8783,
+      "step": 11570
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 14.084067344665527,
+      "learning_rate": 2e-05,
+      "loss": 7.0112,
+      "step": 11580
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 14.963968276977539,
+      "learning_rate": 2e-05,
+      "loss": 7.0636,
+      "step": 11590
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 13.143362998962402,
+      "learning_rate": 2e-05,
+      "loss": 6.8615,
+      "step": 11600
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 15.547945976257324,
+      "learning_rate": 2e-05,
+      "loss": 6.9915,
+      "step": 11610
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 13.54202938079834,
+      "learning_rate": 2e-05,
+      "loss": 7.0606,
+      "step": 11620
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 13.626235961914062,
+      "learning_rate": 2e-05,
+      "loss": 6.852,
+      "step": 11630
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 15.054424285888672,
+      "learning_rate": 2e-05,
+      "loss": 6.8823,
+      "step": 11640
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 15.337044715881348,
+      "learning_rate": 2e-05,
+      "loss": 7.232,
+      "step": 11650
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 14.341991424560547,
+      "learning_rate": 2e-05,
+      "loss": 7.0072,
+      "step": 11660
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 15.256158828735352,
+      "learning_rate": 2e-05,
+      "loss": 7.0084,
+      "step": 11670
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 15.8220796585083,
+      "learning_rate": 2e-05,
+      "loss": 6.9268,
+      "step": 11680
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 14.312150001525879,
+      "learning_rate": 2e-05,
+      "loss": 6.9233,
+      "step": 11690
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 15.248773574829102,
+      "learning_rate": 2e-05,
+      "loss": 6.8638,
+      "step": 11700
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 13.790478706359863,
+      "learning_rate": 2e-05,
+      "loss": 6.7751,
+      "step": 11710
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 15.23730182647705,
+      "learning_rate": 2e-05,
+      "loss": 6.7782,
+      "step": 11720
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 15.317927360534668,
+      "learning_rate": 2e-05,
+      "loss": 6.9445,
+      "step": 11730
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 15.392999649047852,
+      "learning_rate": 2e-05,
+      "loss": 6.9533,
+      "step": 11740
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 14.66800594329834,
+      "learning_rate": 2e-05,
+      "loss": 6.7955,
+      "step": 11750
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 16.125532150268555,
+      "learning_rate": 2e-05,
+      "loss": 7.0254,
+      "step": 11760
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 13.661246299743652,
+      "learning_rate": 2e-05,
+      "loss": 6.9936,
+      "step": 11770
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 14.031455993652344,
+      "learning_rate": 2e-05,
+      "loss": 6.8832,
+      "step": 11780
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 15.530049324035645,
+      "learning_rate": 2e-05,
+      "loss": 6.871,
+      "step": 11790
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 14.123373031616211,
+      "learning_rate": 2e-05,
+      "loss": 7.1159,
+      "step": 11800
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 13.690377235412598,
+      "learning_rate": 2e-05,
+      "loss": 6.9907,
+      "step": 11810
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 13.960504531860352,
+      "learning_rate": 2e-05,
+      "loss": 6.9728,
+      "step": 11820
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 14.501957893371582,
+      "learning_rate": 2e-05,
+      "loss": 6.7951,
+      "step": 11830
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 13.442206382751465,
+      "learning_rate": 2e-05,
+      "loss": 6.8878,
+      "step": 11840
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 14.55359172821045,
+      "learning_rate": 2e-05,
+      "loss": 6.9198,
+      "step": 11850
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 13.578266143798828,
+      "learning_rate": 2e-05,
+      "loss": 6.7497,
+      "step": 11860
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 14.727328300476074,
+      "learning_rate": 2e-05,
+      "loss": 6.7109,
+      "step": 11870
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 14.183723449707031,
+      "learning_rate": 2e-05,
+      "loss": 6.9691,
+      "step": 11880
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 15.056022644042969,
+      "learning_rate": 2e-05,
+      "loss": 6.7466,
+      "step": 11890
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 14.903724670410156,
+      "learning_rate": 2e-05,
+      "loss": 6.926,
+      "step": 11900
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 16.0764217376709,
+      "learning_rate": 2e-05,
+      "loss": 6.9742,
+      "step": 11910
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 15.002456665039062,
+      "learning_rate": 2e-05,
+      "loss": 6.9699,
+      "step": 11920
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 15.099604606628418,
+      "learning_rate": 2e-05,
+      "loss": 6.9004,
+      "step": 11930
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 15.348109245300293,
+      "learning_rate": 2e-05,
+      "loss": 6.9219,
+      "step": 11940
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 16.074190139770508,
+      "learning_rate": 2e-05,
+      "loss": 6.8932,
+      "step": 11950
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 14.809697151184082,
+      "learning_rate": 2e-05,
+      "loss": 6.94,
+      "step": 11960
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 15.425007820129395,
+      "learning_rate": 2e-05,
+      "loss": 7.1163,
+      "step": 11970
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 16.460338592529297,
+      "learning_rate": 2e-05,
+      "loss": 7.0966,
+      "step": 11980
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 16.579219818115234,
+      "learning_rate": 2e-05,
+      "loss": 6.9686,
+      "step": 11990
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 14.590304374694824,
+      "learning_rate": 2e-05,
+      "loss": 6.8136,
+      "step": 12000
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 13.27401065826416,
+      "learning_rate": 2e-05,
+      "loss": 6.8299,
+      "step": 12010
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 13.152297019958496,
+      "learning_rate": 2e-05,
+      "loss": 6.7478,
+      "step": 12020
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 14.569437980651855,
+      "learning_rate": 2e-05,
+      "loss": 6.7659,
+      "step": 12030
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 15.887906074523926,
+      "learning_rate": 2e-05,
+      "loss": 6.8634,
+      "step": 12040
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 15.12689208984375,
+      "learning_rate": 2e-05,
+      "loss": 6.7506,
+      "step": 12050
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 16.169830322265625,
+      "learning_rate": 2e-05,
+      "loss": 6.9034,
+      "step": 12060
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 15.456263542175293,
+      "learning_rate": 2e-05,
+      "loss": 6.8155,
+      "step": 12070
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 14.329488754272461,
+      "learning_rate": 2e-05,
+      "loss": 6.9811,
+      "step": 12080
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 14.51008415222168,
+      "learning_rate": 2e-05,
+      "loss": 6.883,
+      "step": 12090
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 14.532035827636719,
+      "learning_rate": 2e-05,
+      "loss": 6.8428,
+      "step": 12100
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 15.2678804397583,
+      "learning_rate": 2e-05,
+      "loss": 6.696,
+      "step": 12110
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 14.919146537780762,
+      "learning_rate": 2e-05,
+      "loss": 6.78,
+      "step": 12120
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 15.526893615722656,
+      "learning_rate": 2e-05,
+      "loss": 6.8573,
+      "step": 12130
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 14.489819526672363,
+      "learning_rate": 2e-05,
+      "loss": 6.8623,
+      "step": 12140
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 16.344440460205078,
+      "learning_rate": 2e-05,
+      "loss": 7.0979,
+      "step": 12150
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 13.971983909606934,
+      "learning_rate": 2e-05,
+      "loss": 6.9484,
+      "step": 12160
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 15.119500160217285,
+      "learning_rate": 2e-05,
+      "loss": 6.8804,
+      "step": 12170
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 16.677227020263672,
+      "learning_rate": 2e-05,
+      "loss": 6.9562,
+      "step": 12180
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 14.595513343811035,
+      "learning_rate": 2e-05,
+      "loss": 6.8553,
+      "step": 12190
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 15.386113166809082,
+      "learning_rate": 2e-05,
+      "loss": 6.9369,
+      "step": 12200
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 14.474902153015137,
+      "learning_rate": 2e-05,
+      "loss": 6.8746,
+      "step": 12210
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 14.144264221191406,
+      "learning_rate": 2e-05,
+      "loss": 6.63,
+      "step": 12220
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 14.181920051574707,
+      "learning_rate": 2e-05,
+      "loss": 6.9011,
+      "step": 12230
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 14.714789390563965,
+      "learning_rate": 2e-05,
+      "loss": 6.8649,
+      "step": 12240
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 14.505697250366211,
+      "learning_rate": 2e-05,
+      "loss": 6.8386,
+      "step": 12250
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 14.523388862609863,
+      "learning_rate": 2e-05,
+      "loss": 6.811,
+      "step": 12260
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 15.059090614318848,
+      "learning_rate": 2e-05,
+      "loss": 6.9968,
+      "step": 12270
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 15.855351448059082,
+      "learning_rate": 2e-05,
+      "loss": 6.9189,
+      "step": 12280
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 14.302210807800293,
+      "learning_rate": 2e-05,
+      "loss": 7.1279,
+      "step": 12290
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 13.499333381652832,
+      "learning_rate": 2e-05,
+      "loss": 6.9013,
+      "step": 12300
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 13.675992965698242,
+      "learning_rate": 2e-05,
+      "loss": 7.0112,
+      "step": 12310
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 14.09472370147705,
+      "learning_rate": 2e-05,
+      "loss": 6.8119,
+      "step": 12320
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 15.910489082336426,
+      "learning_rate": 2e-05,
+      "loss": 6.8071,
+      "step": 12330
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 13.741456985473633,
+      "learning_rate": 2e-05,
+      "loss": 6.8012,
+      "step": 12340
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 16.015424728393555,
+      "learning_rate": 2e-05,
+      "loss": 6.8551,
+      "step": 12350
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 16.0078182220459,
+      "learning_rate": 2e-05,
+      "loss": 6.9247,
+      "step": 12360
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 14.197285652160645,
+      "learning_rate": 2e-05,
+      "loss": 6.696,
+      "step": 12370
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 15.12751293182373,
+      "learning_rate": 2e-05,
+      "loss": 6.7859,
+      "step": 12380
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 13.424810409545898,
+      "learning_rate": 2e-05,
+      "loss": 6.9675,
+      "step": 12390
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 15.481910705566406,
+      "learning_rate": 2e-05,
+      "loss": 6.9555,
+      "step": 12400
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 13.489352226257324,
+      "learning_rate": 2e-05,
+      "loss": 6.871,
+      "step": 12410
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 14.45278549194336,
+      "learning_rate": 2e-05,
+      "loss": 6.9495,
+      "step": 12420
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 15.372004508972168,
+      "learning_rate": 2e-05,
+      "loss": 6.962,
+      "step": 12430
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 15.522083282470703,
+      "learning_rate": 2e-05,
+      "loss": 6.7689,
+      "step": 12440
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 14.067648887634277,
+      "learning_rate": 2e-05,
+      "loss": 6.8884,
+      "step": 12450
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 14.975571632385254,
+      "learning_rate": 2e-05,
+      "loss": 6.7107,
+      "step": 12460
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 15.06131362915039,
+      "learning_rate": 2e-05,
+      "loss": 6.8117,
+      "step": 12470
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 13.310413360595703,
+      "learning_rate": 2e-05,
+      "loss": 6.7297,
+      "step": 12480
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 15.830842018127441,
+      "learning_rate": 2e-05,
+      "loss": 6.7086,
+      "step": 12490
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 16.957712173461914,
+      "learning_rate": 2e-05,
+      "loss": 7.008,
+      "step": 12500
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 14.18000316619873,
+      "learning_rate": 2e-05,
+      "loss": 6.7716,
+      "step": 12510
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 14.618433952331543,
+      "learning_rate": 2e-05,
+      "loss": 6.8729,
+      "step": 12520
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 13.470077514648438,
+      "learning_rate": 2e-05,
+      "loss": 6.7258,
+      "step": 12530
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 14.02498722076416,
+      "learning_rate": 2e-05,
+      "loss": 6.8613,
+      "step": 12540
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 21.067899703979492,
+      "learning_rate": 2e-05,
+      "loss": 6.8176,
+      "step": 12550
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 14.9075288772583,
+      "learning_rate": 2e-05,
+      "loss": 6.6843,
+      "step": 12560
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 14.410500526428223,
+      "learning_rate": 2e-05,
+      "loss": 6.7648,
+      "step": 12570
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 20.212970733642578,
+      "learning_rate": 2e-05,
+      "loss": 6.8692,
+      "step": 12580
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 15.506143569946289,
+      "learning_rate": 2e-05,
+      "loss": 6.822,
+      "step": 12590
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 14.99155330657959,
+      "learning_rate": 2e-05,
+      "loss": 6.8148,
+      "step": 12600
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 15.59750747680664,
+      "learning_rate": 2e-05,
+      "loss": 6.7161,
+      "step": 12610
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 15.968754768371582,
+      "learning_rate": 2e-05,
+      "loss": 6.8607,
+      "step": 12620
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 15.50302791595459,
+      "learning_rate": 2e-05,
+      "loss": 6.664,
+      "step": 12630
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 14.804305076599121,
+      "learning_rate": 2e-05,
+      "loss": 6.9051,
+      "step": 12640
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 14.873908042907715,
+      "learning_rate": 2e-05,
+      "loss": 6.8801,
+      "step": 12650
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 15.092792510986328,
+      "learning_rate": 2e-05,
+      "loss": 6.8402,
+      "step": 12660
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 15.65123176574707,
+      "learning_rate": 2e-05,
+      "loss": 6.8576,
+      "step": 12670
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 14.559158325195312,
+      "learning_rate": 2e-05,
+      "loss": 6.7845,
+      "step": 12680
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 15.860212326049805,
+      "learning_rate": 2e-05,
+      "loss": 6.8202,
+      "step": 12690
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 16.368013381958008,
+      "learning_rate": 2e-05,
+      "loss": 6.9262,
+      "step": 12700
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 13.347464561462402,
+      "learning_rate": 2e-05,
+      "loss": 6.687,
+      "step": 12710
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 16.429759979248047,
+      "learning_rate": 2e-05,
+      "loss": 6.9619,
+      "step": 12720
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 15.15931224822998,
+      "learning_rate": 2e-05,
+      "loss": 6.8911,
+      "step": 12730
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 16.038387298583984,
+      "learning_rate": 2e-05,
+      "loss": 6.9293,
+      "step": 12740
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 15.691514015197754,
+      "learning_rate": 2e-05,
+      "loss": 6.9226,
+      "step": 12750
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 15.749049186706543,
+      "learning_rate": 2e-05,
+      "loss": 6.7533,
+      "step": 12760
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 15.956521987915039,
+      "learning_rate": 2e-05,
+      "loss": 6.8364,
+      "step": 12770
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 15.003963470458984,
+      "learning_rate": 2e-05,
+      "loss": 6.8786,
+      "step": 12780
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 13.921293258666992,
+      "learning_rate": 2e-05,
+      "loss": 6.9078,
+      "step": 12790
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 14.75953197479248,
+      "learning_rate": 2e-05,
+      "loss": 6.6984,
+      "step": 12800
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 15.881766319274902,
+      "learning_rate": 2e-05,
+      "loss": 6.7317,
+      "step": 12810
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 14.956071853637695,
+      "learning_rate": 2e-05,
+      "loss": 6.7134,
+      "step": 12820
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 15.718018531799316,
+      "learning_rate": 2e-05,
+      "loss": 6.7428,
+      "step": 12830
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 14.024212837219238,
+      "learning_rate": 2e-05,
+      "loss": 6.9431,
+      "step": 12840
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 14.824289321899414,
+      "learning_rate": 2e-05,
+      "loss": 6.8736,
+      "step": 12850
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 14.627041816711426,
+      "learning_rate": 2e-05,
+      "loss": 6.8033,
+      "step": 12860
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 16.278949737548828,
+      "learning_rate": 2e-05,
+      "loss": 6.7255,
+      "step": 12870
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 13.992329597473145,
+      "learning_rate": 2e-05,
+      "loss": 6.8231,
+      "step": 12880
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 15.390740394592285,
+      "learning_rate": 2e-05,
+      "loss": 6.788,
+      "step": 12890
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 13.907502174377441,
+      "learning_rate": 2e-05,
+      "loss": 6.7807,
+      "step": 12900
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 13.133862495422363,
+      "learning_rate": 2e-05,
+      "loss": 6.685,
+      "step": 12910
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 14.521788597106934,
+      "learning_rate": 2e-05,
+      "loss": 6.7424,
+      "step": 12920
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 14.643852233886719,
+      "learning_rate": 2e-05,
+      "loss": 6.7617,
+      "step": 12930
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 13.222611427307129,
+      "learning_rate": 2e-05,
+      "loss": 6.775,
+      "step": 12940
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 15.258116722106934,
+      "learning_rate": 2e-05,
+      "loss": 6.8252,
+      "step": 12950
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 15.171924591064453,
+      "learning_rate": 2e-05,
+      "loss": 6.6531,
+      "step": 12960
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 13.643464088439941,
+      "learning_rate": 2e-05,
+      "loss": 6.6724,
+      "step": 12970
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 14.608119010925293,
+      "learning_rate": 2e-05,
+      "loss": 6.8638,
+      "step": 12980
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 14.682918548583984,
+      "learning_rate": 2e-05,
+      "loss": 6.909,
+      "step": 12990
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 15.799134254455566,
+      "learning_rate": 2e-05,
+      "loss": 6.8826,
+      "step": 13000
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 14.44350814819336,
+      "learning_rate": 2e-05,
+      "loss": 6.9495,
+      "step": 13010
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 14.864227294921875,
+      "learning_rate": 2e-05,
+      "loss": 6.8504,
+      "step": 13020
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 16.32910919189453,
+      "learning_rate": 2e-05,
+      "loss": 6.6813,
+      "step": 13030
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 16.749719619750977,
+      "learning_rate": 2e-05,
+      "loss": 6.8561,
+      "step": 13040
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 15.532760620117188,
+      "learning_rate": 2e-05,
+      "loss": 6.6817,
+      "step": 13050
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 12.771021842956543,
+      "learning_rate": 2e-05,
+      "loss": 6.6342,
+      "step": 13060
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 16.961483001708984,
+      "learning_rate": 2e-05,
+      "loss": 6.8178,
+      "step": 13070
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 13.684576988220215,
+      "learning_rate": 2e-05,
+      "loss": 6.7374,
+      "step": 13080
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 15.65865421295166,
+      "learning_rate": 2e-05,
+      "loss": 6.8634,
+      "step": 13090
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 15.712656021118164,
+      "learning_rate": 2e-05,
+      "loss": 6.9288,
+      "step": 13100
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 15.840964317321777,
+      "learning_rate": 2e-05,
+      "loss": 6.8272,
+      "step": 13110
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 14.39294147491455,
+      "learning_rate": 2e-05,
+      "loss": 6.6816,
+      "step": 13120
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 15.575393676757812,
+      "learning_rate": 2e-05,
+      "loss": 6.8424,
+      "step": 13130
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 15.867019653320312,
+      "learning_rate": 2e-05,
+      "loss": 6.8392,
+      "step": 13140
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 14.367290496826172,
+      "learning_rate": 2e-05,
+      "loss": 6.6919,
+      "step": 13150
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 15.707910537719727,
+      "learning_rate": 2e-05,
+      "loss": 6.7056,
+      "step": 13160
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 16.238046646118164,
+      "learning_rate": 2e-05,
+      "loss": 6.5361,
+      "step": 13170
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 13.90558910369873,
+      "learning_rate": 2e-05,
+      "loss": 6.8854,
+      "step": 13180
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 16.39263153076172,
+      "learning_rate": 2e-05,
+      "loss": 6.9173,
+      "step": 13190
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 15.538141250610352,
+      "learning_rate": 2e-05,
+      "loss": 6.6391,
+      "step": 13200
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 15.193815231323242,
+      "learning_rate": 2e-05,
+      "loss": 6.4824,
+      "step": 13210
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 13.259400367736816,
+      "learning_rate": 2e-05,
+      "loss": 6.7217,
+      "step": 13220
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 14.997381210327148,
+      "learning_rate": 2e-05,
+      "loss": 6.8352,
+      "step": 13230
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 14.977153778076172,
+      "learning_rate": 2e-05,
+      "loss": 6.7553,
+      "step": 13240
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 17.14776039123535,
+      "learning_rate": 2e-05,
+      "loss": 6.6459,
+      "step": 13250
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 14.988301277160645,
+      "learning_rate": 2e-05,
+      "loss": 6.6819,
+      "step": 13260
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 16.204256057739258,
+      "learning_rate": 2e-05,
+      "loss": 6.5963,
+      "step": 13270
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 14.076333999633789,
+      "learning_rate": 2e-05,
+      "loss": 6.7001,
+      "step": 13280
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 13.345596313476562,
+      "learning_rate": 2e-05,
+      "loss": 6.4728,
+      "step": 13290
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 14.718652725219727,
+      "learning_rate": 2e-05,
+      "loss": 6.6714,
+      "step": 13300
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 14.753920555114746,
+      "learning_rate": 2e-05,
+      "loss": 6.6913,
+      "step": 13310
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 13.566704750061035,
+      "learning_rate": 2e-05,
+      "loss": 6.6705,
+      "step": 13320
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 14.26477336883545,
+      "learning_rate": 2e-05,
+      "loss": 6.7835,
+      "step": 13330
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 13.181867599487305,
+      "learning_rate": 2e-05,
+      "loss": 6.9109,
+      "step": 13340
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 15.467188835144043,
+      "learning_rate": 2e-05,
+      "loss": 6.7735,
+      "step": 13350
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 14.674585342407227,
+      "learning_rate": 2e-05,
+      "loss": 6.7335,
+      "step": 13360
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 14.689886093139648,
+      "learning_rate": 2e-05,
+      "loss": 6.6491,
+      "step": 13370
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 15.599201202392578,
+      "learning_rate": 2e-05,
+      "loss": 6.775,
+      "step": 13380
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 15.787370681762695,
+      "learning_rate": 2e-05,
+      "loss": 6.6746,
+      "step": 13390
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 16.097930908203125,
+      "learning_rate": 2e-05,
+      "loss": 6.5494,
+      "step": 13400
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 16.510597229003906,
+      "learning_rate": 2e-05,
+      "loss": 6.7613,
+      "step": 13410
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 14.370491981506348,
+      "learning_rate": 2e-05,
+      "loss": 6.738,
+      "step": 13420
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 15.698972702026367,
+      "learning_rate": 2e-05,
+      "loss": 6.6512,
+      "step": 13430
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 13.620295524597168,
+      "learning_rate": 2e-05,
+      "loss": 7.0173,
+      "step": 13440
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 13.458703994750977,
+      "learning_rate": 2e-05,
+      "loss": 6.6519,
+      "step": 13450
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 14.593814849853516,
+      "learning_rate": 2e-05,
+      "loss": 6.6689,
+      "step": 13460
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 16.001279830932617,
+      "learning_rate": 2e-05,
+      "loss": 6.7697,
+      "step": 13470
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 17.85947608947754,
+      "learning_rate": 2e-05,
+      "loss": 6.6305,
+      "step": 13480
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 14.950535774230957,
+      "learning_rate": 2e-05,
+      "loss": 6.7137,
+      "step": 13490
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 14.8195161819458,
+      "learning_rate": 2e-05,
+      "loss": 6.7757,
+      "step": 13500
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 16.471384048461914,
+      "learning_rate": 2e-05,
+      "loss": 6.6652,
+      "step": 13510
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 16.042421340942383,
+      "learning_rate": 2e-05,
+      "loss": 6.6116,
+      "step": 13520
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 13.899086952209473,
+      "learning_rate": 2e-05,
+      "loss": 6.8608,
+      "step": 13530
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 16.219690322875977,
+      "learning_rate": 2e-05,
+      "loss": 6.8093,
+      "step": 13540
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 15.49570369720459,
+      "learning_rate": 2e-05,
+      "loss": 6.6456,
+      "step": 13550
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 14.554037094116211,
+      "learning_rate": 2e-05,
+      "loss": 6.6964,
+      "step": 13560
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 14.855463027954102,
+      "learning_rate": 2e-05,
+      "loss": 6.6779,
+      "step": 13570
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 15.173818588256836,
+      "learning_rate": 2e-05,
+      "loss": 6.8206,
+      "step": 13580
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 16.090442657470703,
+      "learning_rate": 2e-05,
+      "loss": 6.725,
+      "step": 13590
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 14.434093475341797,
+      "learning_rate": 2e-05,
+      "loss": 6.6458,
+      "step": 13600
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 14.370340347290039,
+      "learning_rate": 2e-05,
+      "loss": 7.0154,
+      "step": 13610
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 15.422374725341797,
+      "learning_rate": 2e-05,
+      "loss": 6.562,
+      "step": 13620
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 14.857234001159668,
+      "learning_rate": 2e-05,
+      "loss": 6.7318,
+      "step": 13630
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 14.608033180236816,
+      "learning_rate": 2e-05,
+      "loss": 6.7896,
+      "step": 13640
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 14.441346168518066,
+      "learning_rate": 2e-05,
+      "loss": 6.9408,
+      "step": 13650
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 14.724777221679688,
+      "learning_rate": 2e-05,
+      "loss": 6.6247,
+      "step": 13660
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 14.789780616760254,
+      "learning_rate": 2e-05,
+      "loss": 6.806,
+      "step": 13670
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 15.368186950683594,
+      "learning_rate": 2e-05,
+      "loss": 6.6835,
+      "step": 13680
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 15.841795921325684,
+      "learning_rate": 2e-05,
+      "loss": 6.6376,
+      "step": 13690
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 14.394085884094238,
+      "learning_rate": 2e-05,
+      "loss": 6.7565,
+      "step": 13700
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 15.615621566772461,
+      "learning_rate": 2e-05,
+      "loss": 6.7426,
+      "step": 13710
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 17.118175506591797,
+      "learning_rate": 2e-05,
+      "loss": 6.6848,
+      "step": 13720
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 14.675713539123535,
+      "learning_rate": 2e-05,
+      "loss": 6.6946,
+      "step": 13730
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 14.872252464294434,
+      "learning_rate": 2e-05,
+      "loss": 6.5875,
+      "step": 13740
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 15.56899356842041,
+      "learning_rate": 2e-05,
+      "loss": 6.6656,
+      "step": 13750
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 14.299012184143066,
+      "learning_rate": 2e-05,
+      "loss": 6.804,
+      "step": 13760
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 16.08584976196289,
+      "learning_rate": 2e-05,
+      "loss": 6.8331,
+      "step": 13770
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 15.828598976135254,
+      "learning_rate": 2e-05,
+      "loss": 6.5777,
+      "step": 13780
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 13.40284538269043,
+      "learning_rate": 2e-05,
+      "loss": 6.8622,
+      "step": 13790
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 14.53720474243164,
+      "learning_rate": 2e-05,
+      "loss": 6.8176,
+      "step": 13800
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 16.334604263305664,
+      "learning_rate": 2e-05,
+      "loss": 6.6526,
+      "step": 13810
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 14.665732383728027,
+      "learning_rate": 2e-05,
+      "loss": 6.7554,
+      "step": 13820
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 14.748302459716797,
+      "learning_rate": 2e-05,
+      "loss": 6.5043,
+      "step": 13830
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 14.525221824645996,
+      "learning_rate": 2e-05,
+      "loss": 6.4015,
+      "step": 13840
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 15.270462036132812,
+      "learning_rate": 2e-05,
+      "loss": 6.6658,
+      "step": 13850
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 14.441527366638184,
+      "learning_rate": 2e-05,
+      "loss": 6.777,
+      "step": 13860
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 14.572378158569336,
+      "learning_rate": 2e-05,
+      "loss": 6.5661,
+      "step": 13870
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 17.600065231323242,
+      "learning_rate": 2e-05,
+      "loss": 6.7747,
+      "step": 13880
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 14.855683326721191,
+      "learning_rate": 2e-05,
+      "loss": 6.359,
+      "step": 13890
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 14.742898941040039,
+      "learning_rate": 2e-05,
+      "loss": 6.6549,
+      "step": 13900
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 15.247746467590332,
+      "learning_rate": 2e-05,
+      "loss": 6.4532,
+      "step": 13910
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 13.629854202270508,
+      "learning_rate": 2e-05,
+      "loss": 6.5834,
+      "step": 13920
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 14.170391082763672,
+      "learning_rate": 2e-05,
+      "loss": 6.6758,
+      "step": 13930
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 15.60438060760498,
+      "learning_rate": 2e-05,
+      "loss": 6.6405,
+      "step": 13940
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 15.954154014587402,
+      "learning_rate": 2e-05,
+      "loss": 6.4359,
+      "step": 13950
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 15.658191680908203,
+      "learning_rate": 2e-05,
+      "loss": 6.688,
+      "step": 13960
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 15.468618392944336,
+      "learning_rate": 2e-05,
+      "loss": 6.49,
+      "step": 13970
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 14.310837745666504,
+      "learning_rate": 2e-05,
+      "loss": 6.9283,
+      "step": 13980
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 14.935101509094238,
+      "learning_rate": 2e-05,
+      "loss": 6.5512,
+      "step": 13990
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 14.687759399414062,
+      "learning_rate": 2e-05,
+      "loss": 6.5333,
+      "step": 14000
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 14.713521957397461,
+      "learning_rate": 2e-05,
+      "loss": 6.6968,
+      "step": 14010
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 14.432047843933105,
+      "learning_rate": 2e-05,
+      "loss": 6.708,
+      "step": 14020
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 14.903938293457031,
+      "learning_rate": 2e-05,
+      "loss": 6.841,
+      "step": 14030
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 16.81429100036621,
+      "learning_rate": 2e-05,
+      "loss": 6.5048,
+      "step": 14040
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 17.86644172668457,
+      "learning_rate": 2e-05,
+      "loss": 6.6385,
+      "step": 14050
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 14.738943099975586,
+      "learning_rate": 2e-05,
+      "loss": 6.7044,
+      "step": 14060
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 14.312878608703613,
+      "learning_rate": 2e-05,
+      "loss": 6.7809,
+      "step": 14070
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 14.45374870300293,
+      "learning_rate": 2e-05,
+      "loss": 6.3254,
+      "step": 14080
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 14.215295791625977,
+      "learning_rate": 2e-05,
+      "loss": 6.3796,
+      "step": 14090
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 15.804530143737793,
+      "learning_rate": 2e-05,
+      "loss": 6.8231,
+      "step": 14100
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 14.36791706085205,
+      "learning_rate": 2e-05,
+      "loss": 6.8425,
+      "step": 14110
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 13.84842586517334,
+      "learning_rate": 2e-05,
+      "loss": 6.6719,
+      "step": 14120
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 15.357117652893066,
+      "learning_rate": 2e-05,
+      "loss": 6.7039,
+      "step": 14130
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 13.66261100769043,
+      "learning_rate": 2e-05,
+      "loss": 6.7584,
+      "step": 14140
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 16.76607322692871,
+      "learning_rate": 2e-05,
+      "loss": 6.7451,
+      "step": 14150
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 14.813446044921875,
+      "learning_rate": 2e-05,
+      "loss": 6.7916,
+      "step": 14160
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 14.233673095703125,
+      "learning_rate": 2e-05,
+      "loss": 6.6622,
+      "step": 14170
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 16.24745750427246,
+      "learning_rate": 2e-05,
+      "loss": 6.5519,
+      "step": 14180
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 16.477802276611328,
+      "learning_rate": 2e-05,
+      "loss": 6.7033,
+      "step": 14190
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 15.421667098999023,
+      "learning_rate": 2e-05,
+      "loss": 6.5432,
+      "step": 14200
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 15.834403038024902,
+      "learning_rate": 2e-05,
+      "loss": 6.9277,
+      "step": 14210
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 15.007030487060547,
+      "learning_rate": 2e-05,
+      "loss": 6.838,
+      "step": 14220
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 14.370731353759766,
+      "learning_rate": 2e-05,
+      "loss": 6.8294,
+      "step": 14230
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 14.737106323242188,
+      "learning_rate": 2e-05,
+      "loss": 6.7022,
+      "step": 14240
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 15.299291610717773,
+      "learning_rate": 2e-05,
+      "loss": 6.5546,
+      "step": 14250
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 15.349090576171875,
+      "learning_rate": 2e-05,
+      "loss": 6.523,
+      "step": 14260
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 15.036453247070312,
+      "learning_rate": 2e-05,
+      "loss": 6.8426,
+      "step": 14270
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 13.927254676818848,
+      "learning_rate": 2e-05,
+      "loss": 6.7322,
+      "step": 14280
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 16.602388381958008,
+      "learning_rate": 2e-05,
+      "loss": 6.4404,
+      "step": 14290
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 15.294507026672363,
+      "learning_rate": 2e-05,
+      "loss": 6.6335,
+      "step": 14300
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 14.126951217651367,
+      "learning_rate": 2e-05,
+      "loss": 6.5957,
+      "step": 14310
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 14.299042701721191,
+      "learning_rate": 2e-05,
+      "loss": 6.5496,
+      "step": 14320
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 13.677967071533203,
+      "learning_rate": 2e-05,
+      "loss": 6.5795,
+      "step": 14330
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 14.801259994506836,
+      "learning_rate": 2e-05,
+      "loss": 6.5435,
+      "step": 14340
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 16.320741653442383,
+      "learning_rate": 2e-05,
+      "loss": 6.4941,
+      "step": 14350
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 14.996045112609863,
+      "learning_rate": 2e-05,
+      "loss": 6.7401,
+      "step": 14360
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 15.045677185058594,
+      "learning_rate": 2e-05,
+      "loss": 6.8478,
+      "step": 14370
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 15.467405319213867,
+      "learning_rate": 2e-05,
+      "loss": 6.5008,
+      "step": 14380
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 14.314493179321289,
+      "learning_rate": 2e-05,
+      "loss": 6.4677,
+      "step": 14390
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 17.77299690246582,
+      "learning_rate": 2e-05,
+      "loss": 6.5207,
+      "step": 14400
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 15.464516639709473,
+      "learning_rate": 2e-05,
+      "loss": 6.7919,
+      "step": 14410
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 16.159595489501953,
+      "learning_rate": 2e-05,
+      "loss": 6.6623,
+      "step": 14420
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 14.659346580505371,
+      "learning_rate": 2e-05,
+      "loss": 6.6264,
+      "step": 14430
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 14.202614784240723,
+      "learning_rate": 2e-05,
+      "loss": 6.3346,
+      "step": 14440
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 14.236883163452148,
+      "learning_rate": 2e-05,
+      "loss": 6.7717,
+      "step": 14450
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 15.198843955993652,
+      "learning_rate": 2e-05,
+      "loss": 6.713,
+      "step": 14460
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 15.088488578796387,
+      "learning_rate": 2e-05,
+      "loss": 6.427,
+      "step": 14470
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 15.992968559265137,
+      "learning_rate": 2e-05,
+      "loss": 6.6145,
+      "step": 14480
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 14.647175788879395,
+      "learning_rate": 2e-05,
+      "loss": 6.7512,
+      "step": 14490
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 14.480786323547363,
+      "learning_rate": 2e-05,
+      "loss": 6.5385,
+      "step": 14500
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 15.541913032531738,
+      "learning_rate": 2e-05,
+      "loss": 6.5668,
+      "step": 14510
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 13.710472106933594,
+      "learning_rate": 2e-05,
+      "loss": 6.681,
+      "step": 14520
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 14.57370662689209,
+      "learning_rate": 2e-05,
+      "loss": 6.6456,
+      "step": 14530
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 15.920002937316895,
+      "learning_rate": 2e-05,
+      "loss": 6.5133,
+      "step": 14540
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 15.882360458374023,
+      "learning_rate": 2e-05,
+      "loss": 6.7896,
+      "step": 14550
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 16.04817008972168,
+      "learning_rate": 2e-05,
+      "loss": 6.6603,
+      "step": 14560
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 15.284319877624512,
+      "learning_rate": 2e-05,
+      "loss": 6.5695,
+      "step": 14570
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 14.631726264953613,
+      "learning_rate": 2e-05,
+      "loss": 6.5862,
+      "step": 14580
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 14.455802917480469,
+      "learning_rate": 2e-05,
+      "loss": 6.6057,
+      "step": 14590
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 18.092912673950195,
+      "learning_rate": 2e-05,
+      "loss": 6.724,
+      "step": 14600
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 12.785186767578125,
+      "learning_rate": 2e-05,
+      "loss": 6.6084,
+      "step": 14610
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 14.566864967346191,
+      "learning_rate": 2e-05,
+      "loss": 6.7087,
+      "step": 14620
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 14.472478866577148,
+      "learning_rate": 2e-05,
+      "loss": 6.5392,
+      "step": 14630
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 14.56663990020752,
+      "learning_rate": 2e-05,
+      "loss": 6.6232,
+      "step": 14640
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 14.698761940002441,
+      "learning_rate": 2e-05,
+      "loss": 6.6496,
+      "step": 14650
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 15.579837799072266,
+      "learning_rate": 2e-05,
+      "loss": 6.6196,
+      "step": 14660
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 15.02534294128418,
+      "learning_rate": 2e-05,
+      "loss": 6.624,
+      "step": 14670
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 15.129195213317871,
+      "learning_rate": 2e-05,
+      "loss": 6.3557,
+      "step": 14680
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 16.721607208251953,
+      "learning_rate": 2e-05,
+      "loss": 6.4398,
+      "step": 14690
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 15.278815269470215,
+      "learning_rate": 2e-05,
+      "loss": 6.4738,
+      "step": 14700
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 16.346054077148438,
+      "learning_rate": 2e-05,
+      "loss": 6.6816,
+      "step": 14710
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 14.81169605255127,
+      "learning_rate": 2e-05,
+      "loss": 6.5351,
+      "step": 14720
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 16.356847763061523,
+      "learning_rate": 2e-05,
+      "loss": 6.5258,
+      "step": 14730
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 16.129968643188477,
+      "learning_rate": 2e-05,
+      "loss": 6.6026,
+      "step": 14740
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 15.857667922973633,
+      "learning_rate": 2e-05,
+      "loss": 6.4946,
+      "step": 14750
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 15.911877632141113,
+      "learning_rate": 2e-05,
+      "loss": 6.6474,
+      "step": 14760
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 15.480733871459961,
+      "learning_rate": 2e-05,
+      "loss": 6.7118,
+      "step": 14770
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 15.487770080566406,
+      "learning_rate": 2e-05,
+      "loss": 6.7074,
+      "step": 14780
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 15.876402854919434,
+      "learning_rate": 2e-05,
+      "loss": 6.3819,
+      "step": 14790
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 16.24234962463379,
+      "learning_rate": 2e-05,
+      "loss": 6.4404,
+      "step": 14800
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 15.897218704223633,
+      "learning_rate": 2e-05,
+      "loss": 6.6319,
+      "step": 14810
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 15.84553050994873,
+      "learning_rate": 2e-05,
+      "loss": 6.6451,
+      "step": 14820
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 15.583945274353027,
+      "learning_rate": 2e-05,
+      "loss": 6.7459,
+      "step": 14830
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 15.094155311584473,
+      "learning_rate": 2e-05,
+      "loss": 6.4914,
+      "step": 14840
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 16.74850082397461,
+      "learning_rate": 2e-05,
+      "loss": 6.7261,
+      "step": 14850
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 16.20612335205078,
+      "learning_rate": 2e-05,
+      "loss": 6.4544,
+      "step": 14860
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 15.429441452026367,
+      "learning_rate": 2e-05,
+      "loss": 6.4862,
+      "step": 14870
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 16.521364212036133,
+      "learning_rate": 2e-05,
+      "loss": 6.3568,
+      "step": 14880
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 16.331472396850586,
+      "learning_rate": 2e-05,
+      "loss": 6.5702,
+      "step": 14890
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 15.611903190612793,
+      "learning_rate": 2e-05,
+      "loss": 6.5884,
+      "step": 14900
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 17.04602813720703,
+      "learning_rate": 2e-05,
+      "loss": 6.4233,
+      "step": 14910
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 17.158998489379883,
+      "learning_rate": 2e-05,
+      "loss": 6.7816,
+      "step": 14920
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 14.79043197631836,
+      "learning_rate": 2e-05,
+      "loss": 6.647,
+      "step": 14930
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 16.092388153076172,
+      "learning_rate": 2e-05,
+      "loss": 6.5711,
+      "step": 14940
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 14.712019920349121,
+      "learning_rate": 2e-05,
+      "loss": 6.6477,
+      "step": 14950
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 15.271794319152832,
+      "learning_rate": 2e-05,
+      "loss": 6.5325,
+      "step": 14960
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 13.900286674499512,
+      "learning_rate": 2e-05,
+      "loss": 6.3178,
+      "step": 14970
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 15.506702423095703,
+      "learning_rate": 2e-05,
+      "loss": 6.517,
+      "step": 14980
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 15.487544059753418,
+      "learning_rate": 2e-05,
+      "loss": 6.6178,
+      "step": 14990
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 14.402244567871094,
+      "learning_rate": 2e-05,
+      "loss": 6.3698,
+      "step": 15000
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 15.296663284301758,
+      "learning_rate": 2e-05,
+      "loss": 6.6834,
+      "step": 15010
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 15.020416259765625,
+      "learning_rate": 2e-05,
+      "loss": 6.6198,
+      "step": 15020
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 13.360940933227539,
+      "learning_rate": 2e-05,
+      "loss": 6.3988,
+      "step": 15030
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 16.097654342651367,
+      "learning_rate": 2e-05,
+      "loss": 6.6633,
+      "step": 15040
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 15.560067176818848,
+      "learning_rate": 2e-05,
+      "loss": 6.4201,
+      "step": 15050
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 15.162254333496094,
+      "learning_rate": 2e-05,
+      "loss": 6.6268,
+      "step": 15060
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 17.082534790039062,
+      "learning_rate": 2e-05,
+      "loss": 6.6624,
+      "step": 15070
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 15.690873146057129,
+      "learning_rate": 2e-05,
+      "loss": 6.4963,
+      "step": 15080
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 14.872403144836426,
+      "learning_rate": 2e-05,
+      "loss": 6.5746,
+      "step": 15090
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 15.98568058013916,
+      "learning_rate": 2e-05,
+      "loss": 6.6036,
+      "step": 15100
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 15.859519958496094,
+      "learning_rate": 2e-05,
+      "loss": 6.3925,
+      "step": 15110
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 15.172274589538574,
+      "learning_rate": 2e-05,
+      "loss": 6.7225,
+      "step": 15120
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 16.468400955200195,
+      "learning_rate": 2e-05,
+      "loss": 6.5953,
+      "step": 15130
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 15.711435317993164,
+      "learning_rate": 2e-05,
+      "loss": 6.6065,
+      "step": 15140
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 16.142654418945312,
+      "learning_rate": 2e-05,
+      "loss": 6.684,
+      "step": 15150
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 15.803149223327637,
+      "learning_rate": 2e-05,
+      "loss": 6.5515,
+      "step": 15160
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 18.80848503112793,
+      "learning_rate": 2e-05,
+      "loss": 6.5933,
+      "step": 15170
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 17.83698081970215,
+      "learning_rate": 2e-05,
+      "loss": 6.4374,
+      "step": 15180
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 14.776803016662598,
+      "learning_rate": 2e-05,
+      "loss": 6.518,
+      "step": 15190
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 14.723366737365723,
+      "learning_rate": 2e-05,
+      "loss": 6.4181,
+      "step": 15200
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 15.753849983215332,
+      "learning_rate": 2e-05,
+      "loss": 6.4429,
+      "step": 15210
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 15.203802108764648,
+      "learning_rate": 2e-05,
+      "loss": 6.7441,
+      "step": 15220
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 14.311327934265137,
+      "learning_rate": 2e-05,
+      "loss": 6.523,
+      "step": 15230
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 16.481412887573242,
+      "learning_rate": 2e-05,
+      "loss": 6.4969,
+      "step": 15240
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 13.90329360961914,
+      "learning_rate": 2e-05,
+      "loss": 6.6403,
+      "step": 15250
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 13.513293266296387,
+      "learning_rate": 2e-05,
+      "loss": 6.5265,
+      "step": 15260
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 15.07665729522705,
+      "learning_rate": 2e-05,
+      "loss": 6.4938,
+      "step": 15270
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 14.79079818725586,
+      "learning_rate": 2e-05,
+      "loss": 6.6404,
+      "step": 15280
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 17.01940155029297,
+      "learning_rate": 2e-05,
+      "loss": 6.3836,
+      "step": 15290
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 17.019996643066406,
+      "learning_rate": 2e-05,
+      "loss": 6.4928,
+      "step": 15300
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 14.108888626098633,
+      "learning_rate": 2e-05,
+      "loss": 6.516,
+      "step": 15310
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 13.100480079650879,
+      "learning_rate": 2e-05,
+      "loss": 6.4863,
+      "step": 15320
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 16.810903549194336,
+      "learning_rate": 2e-05,
+      "loss": 6.3276,
+      "step": 15330
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 15.083131790161133,
+      "learning_rate": 2e-05,
+      "loss": 6.5097,
+      "step": 15340
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 17.349939346313477,
+      "learning_rate": 2e-05,
+      "loss": 6.8096,
+      "step": 15350
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 15.86758804321289,
+      "learning_rate": 2e-05,
+      "loss": 6.3667,
+      "step": 15360
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 14.937186241149902,
+      "learning_rate": 2e-05,
+      "loss": 6.4983,
+      "step": 15370
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 15.384864807128906,
+      "learning_rate": 2e-05,
+      "loss": 6.6233,
+      "step": 15380
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 15.40036678314209,
+      "learning_rate": 2e-05,
+      "loss": 6.5654,
+      "step": 15390
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 14.189545631408691,
+      "learning_rate": 2e-05,
+      "loss": 6.6544,
+      "step": 15400
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 15.211139678955078,
+      "learning_rate": 2e-05,
+      "loss": 6.4939,
+      "step": 15410
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 14.49018383026123,
+      "learning_rate": 2e-05,
+      "loss": 6.6113,
+      "step": 15420
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 15.555649757385254,
+      "learning_rate": 2e-05,
+      "loss": 6.4018,
+      "step": 15430
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 14.646556854248047,
+      "learning_rate": 2e-05,
+      "loss": 6.4167,
+      "step": 15440
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 15.22208309173584,
+      "learning_rate": 2e-05,
+      "loss": 6.4551,
+      "step": 15450
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 19.484954833984375,
+      "learning_rate": 2e-05,
+      "loss": 6.6714,
+      "step": 15460
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 18.042903900146484,
+      "learning_rate": 2e-05,
+      "loss": 6.6256,
+      "step": 15470
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 14.223432540893555,
+      "learning_rate": 2e-05,
+      "loss": 6.6015,
+      "step": 15480
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 16.706811904907227,
+      "learning_rate": 2e-05,
+      "loss": 6.4923,
+      "step": 15490
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 15.608891487121582,
+      "learning_rate": 2e-05,
+      "loss": 6.7369,
+      "step": 15500
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 16.472776412963867,
+      "learning_rate": 2e-05,
+      "loss": 6.5006,
+      "step": 15510
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 14.596131324768066,
+      "learning_rate": 2e-05,
+      "loss": 6.5842,
+      "step": 15520
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 16.04163932800293,
+      "learning_rate": 2e-05,
+      "loss": 6.3871,
+      "step": 15530
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 15.579840660095215,
+      "learning_rate": 2e-05,
+      "loss": 6.4479,
+      "step": 15540
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 14.929306030273438,
+      "learning_rate": 2e-05,
+      "loss": 6.6375,
+      "step": 15550
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 15.60108757019043,
+      "learning_rate": 2e-05,
+      "loss": 6.6131,
+      "step": 15560
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 14.621530532836914,
+      "learning_rate": 2e-05,
+      "loss": 6.5859,
+      "step": 15570
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 15.65206241607666,
+      "learning_rate": 2e-05,
+      "loss": 6.4341,
+      "step": 15580
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 16.292713165283203,
+      "learning_rate": 2e-05,
+      "loss": 6.4954,
+      "step": 15590
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 15.040979385375977,
+      "learning_rate": 2e-05,
+      "loss": 6.3735,
+      "step": 15600
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 15.18702507019043,
+      "learning_rate": 2e-05,
+      "loss": 6.5236,
+      "step": 15610
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 14.99183177947998,
+      "learning_rate": 2e-05,
+      "loss": 6.3869,
+      "step": 15620
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 15.134499549865723,
+      "learning_rate": 2e-05,
+      "loss": 6.5019,
+      "step": 15630
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 15.355353355407715,
+      "learning_rate": 2e-05,
+      "loss": 6.5282,
+      "step": 15640
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 16.89632225036621,
+      "learning_rate": 2e-05,
+      "loss": 6.6216,
+      "step": 15650
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 15.715384483337402,
+      "learning_rate": 2e-05,
+      "loss": 6.6431,
+      "step": 15660
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 15.018484115600586,
+      "learning_rate": 2e-05,
+      "loss": 6.5329,
+      "step": 15670
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 14.363258361816406,
+      "learning_rate": 2e-05,
+      "loss": 6.5374,
+      "step": 15680
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 15.5237398147583,
+      "learning_rate": 2e-05,
+      "loss": 6.5557,
+      "step": 15690
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 14.866918563842773,
+      "learning_rate": 2e-05,
+      "loss": 6.8029,
+      "step": 15700
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 13.848626136779785,
+      "learning_rate": 2e-05,
+      "loss": 6.4091,
+      "step": 15710
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 15.833540916442871,
+      "learning_rate": 2e-05,
+      "loss": 6.5661,
+      "step": 15720
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 14.479293823242188,
+      "learning_rate": 2e-05,
+      "loss": 6.2653,
+      "step": 15730
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 15.29383373260498,
+      "learning_rate": 2e-05,
+      "loss": 6.4425,
+      "step": 15740
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 14.912490844726562,
+      "learning_rate": 2e-05,
+      "loss": 6.4661,
+      "step": 15750
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 15.302436828613281,
+      "learning_rate": 2e-05,
+      "loss": 6.4616,
+      "step": 15760
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 14.232975959777832,
+      "learning_rate": 2e-05,
+      "loss": 6.5444,
+      "step": 15770
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 14.85550594329834,
+      "learning_rate": 2e-05,
+      "loss": 6.2942,
+      "step": 15780
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 16.86231231689453,
+      "learning_rate": 2e-05,
+      "loss": 6.5574,
+      "step": 15790
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 13.633233070373535,
+      "learning_rate": 2e-05,
+      "loss": 6.6495,
+      "step": 15800
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 15.849940299987793,
+      "learning_rate": 2e-05,
+      "loss": 6.3241,
+      "step": 15810
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 15.80882740020752,
+      "learning_rate": 2e-05,
+      "loss": 6.4183,
+      "step": 15820
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 16.56382942199707,
+      "learning_rate": 2e-05,
+      "loss": 6.5182,
+      "step": 15830
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 15.270042419433594,
+      "learning_rate": 2e-05,
+      "loss": 6.3367,
+      "step": 15840
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 15.830792427062988,
+      "learning_rate": 2e-05,
+      "loss": 6.3804,
+      "step": 15850
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 13.50974178314209,
+      "learning_rate": 2e-05,
+      "loss": 6.4015,
+      "step": 15860
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 14.135320663452148,
+      "learning_rate": 2e-05,
+      "loss": 6.4237,
+      "step": 15870
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 14.497892379760742,
+      "learning_rate": 2e-05,
+      "loss": 6.626,
+      "step": 15880
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 14.920828819274902,
+      "learning_rate": 2e-05,
+      "loss": 6.4398,
+      "step": 15890
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 15.668168067932129,
+      "learning_rate": 2e-05,
+      "loss": 6.6895,
+      "step": 15900
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 14.712526321411133,
+      "learning_rate": 2e-05,
+      "loss": 6.4982,
+      "step": 15910
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 14.169650077819824,
+      "learning_rate": 2e-05,
+      "loss": 6.4782,
+      "step": 15920
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 15.709796905517578,
+      "learning_rate": 2e-05,
+      "loss": 6.5842,
+      "step": 15930
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 16.67365264892578,
+      "learning_rate": 2e-05,
+      "loss": 6.3936,
+      "step": 15940
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 14.30532455444336,
+      "learning_rate": 2e-05,
+      "loss": 6.407,
+      "step": 15950
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 14.759259223937988,
+      "learning_rate": 2e-05,
+      "loss": 6.5726,
+      "step": 15960
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 14.976035118103027,
+      "learning_rate": 2e-05,
+      "loss": 6.4241,
+      "step": 15970
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 14.01376724243164,
+      "learning_rate": 2e-05,
+      "loss": 6.6032,
+      "step": 15980
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 15.709385871887207,
+      "learning_rate": 2e-05,
+      "loss": 6.3727,
+      "step": 15990
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 14.588010787963867,
+      "learning_rate": 2e-05,
+      "loss": 6.5436,
+      "step": 16000
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 15.100138664245605,
+      "learning_rate": 2e-05,
+      "loss": 6.6195,
+      "step": 16010
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 14.950286865234375,
+      "learning_rate": 2e-05,
+      "loss": 6.6003,
+      "step": 16020
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 17.17409324645996,
+      "learning_rate": 2e-05,
+      "loss": 6.378,
+      "step": 16030
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 16.047353744506836,
+      "learning_rate": 2e-05,
+      "loss": 6.3816,
+      "step": 16040
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 15.27676010131836,
+      "learning_rate": 2e-05,
+      "loss": 6.3565,
+      "step": 16050
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 15.59419059753418,
+      "learning_rate": 2e-05,
+      "loss": 6.5325,
+      "step": 16060
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 15.91185474395752,
+      "learning_rate": 2e-05,
+      "loss": 6.6204,
+      "step": 16070
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 16.623432159423828,
+      "learning_rate": 2e-05,
+      "loss": 6.3098,
+      "step": 16080
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 15.676054954528809,
+      "learning_rate": 2e-05,
+      "loss": 6.65,
+      "step": 16090
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 14.239545822143555,
+      "learning_rate": 2e-05,
+      "loss": 6.4156,
+      "step": 16100
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 15.070923805236816,
+      "learning_rate": 2e-05,
+      "loss": 6.5807,
+      "step": 16110
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 15.43294620513916,
+      "learning_rate": 2e-05,
+      "loss": 6.6593,
+      "step": 16120
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 14.782135009765625,
+      "learning_rate": 2e-05,
+      "loss": 6.5138,
+      "step": 16130
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 15.034358978271484,
+      "learning_rate": 2e-05,
+      "loss": 6.6642,
+      "step": 16140
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 13.86994457244873,
+      "learning_rate": 2e-05,
+      "loss": 6.4027,
+      "step": 16150
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 15.070655822753906,
+      "learning_rate": 2e-05,
+      "loss": 6.5765,
+      "step": 16160
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 15.837564468383789,
+      "learning_rate": 2e-05,
+      "loss": 6.7411,
+      "step": 16170
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 17.149192810058594,
+      "learning_rate": 2e-05,
+      "loss": 6.364,
+      "step": 16180
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 16.211307525634766,
+      "learning_rate": 2e-05,
+      "loss": 6.7104,
+      "step": 16190
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 14.281487464904785,
+      "learning_rate": 2e-05,
+      "loss": 6.4213,
+      "step": 16200
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 14.621048927307129,
+      "learning_rate": 2e-05,
+      "loss": 6.5507,
+      "step": 16210
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 15.403440475463867,
+      "learning_rate": 2e-05,
+      "loss": 6.2547,
+      "step": 16220
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 16.846567153930664,
+      "learning_rate": 2e-05,
+      "loss": 6.2475,
+      "step": 16230
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 15.933964729309082,
+      "learning_rate": 2e-05,
+      "loss": 6.316,
+      "step": 16240
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 15.933347702026367,
+      "learning_rate": 2e-05,
+      "loss": 6.5947,
+      "step": 16250
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 15.379637718200684,
+      "learning_rate": 2e-05,
+      "loss": 6.5035,
+      "step": 16260
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 14.970175743103027,
+      "learning_rate": 2e-05,
+      "loss": 6.2216,
+      "step": 16270
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 15.534355163574219,
+      "learning_rate": 2e-05,
+      "loss": 6.2637,
+      "step": 16280
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 15.250776290893555,
+      "learning_rate": 2e-05,
+      "loss": 6.5953,
+      "step": 16290
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 15.492959022521973,
+      "learning_rate": 2e-05,
+      "loss": 6.4721,
+      "step": 16300
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 13.91783332824707,
+      "learning_rate": 2e-05,
+      "loss": 6.455,
+      "step": 16310
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 15.803215026855469,
+      "learning_rate": 2e-05,
+      "loss": 6.578,
+      "step": 16320
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 14.390982627868652,
+      "learning_rate": 2e-05,
+      "loss": 6.2843,
+      "step": 16330
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 15.16487979888916,
+      "learning_rate": 2e-05,
+      "loss": 6.3023,
+      "step": 16340
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 15.763325691223145,
+      "learning_rate": 2e-05,
+      "loss": 6.5185,
+      "step": 16350
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 14.605792045593262,
+      "learning_rate": 2e-05,
+      "loss": 6.5899,
+      "step": 16360
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 14.928783416748047,
+      "learning_rate": 2e-05,
+      "loss": 6.5643,
+      "step": 16370
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 14.614912986755371,
+      "learning_rate": 2e-05,
+      "loss": 6.3854,
+      "step": 16380
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 16.401464462280273,
+      "learning_rate": 2e-05,
+      "loss": 6.5272,
+      "step": 16390
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 15.788320541381836,
+      "learning_rate": 2e-05,
+      "loss": 6.1818,
+      "step": 16400
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 16.77754783630371,
+      "learning_rate": 2e-05,
+      "loss": 6.3456,
+      "step": 16410
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 16.16180992126465,
+      "learning_rate": 2e-05,
+      "loss": 6.4311,
+      "step": 16420
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 16.215192794799805,
+      "learning_rate": 2e-05,
+      "loss": 6.3395,
+      "step": 16430
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 15.59289836883545,
+      "learning_rate": 2e-05,
+      "loss": 6.2643,
+      "step": 16440
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 16.414077758789062,
+      "learning_rate": 2e-05,
+      "loss": 6.3895,
+      "step": 16450
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 17.55326271057129,
+      "learning_rate": 2e-05,
+      "loss": 6.4448,
+      "step": 16460
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 15.335395812988281,
+      "learning_rate": 2e-05,
+      "loss": 6.3698,
+      "step": 16470
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 15.532212257385254,
+      "learning_rate": 2e-05,
+      "loss": 6.6018,
+      "step": 16480
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 16.02492332458496,
+      "learning_rate": 2e-05,
+      "loss": 6.3343,
+      "step": 16490
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 13.808239936828613,
+      "learning_rate": 2e-05,
+      "loss": 6.2185,
+      "step": 16500
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 15.412820816040039,
+      "learning_rate": 2e-05,
+      "loss": 6.4264,
+      "step": 16510
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 16.071989059448242,
+      "learning_rate": 2e-05,
+      "loss": 6.4237,
+      "step": 16520
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 16.42438316345215,
+      "learning_rate": 2e-05,
+      "loss": 6.4635,
+      "step": 16530
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 15.231734275817871,
+      "learning_rate": 2e-05,
+      "loss": 6.3772,
+      "step": 16540
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 16.48702049255371,
+      "learning_rate": 2e-05,
+      "loss": 6.5059,
+      "step": 16550
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 16.046371459960938,
+      "learning_rate": 2e-05,
+      "loss": 6.5054,
+      "step": 16560
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 15.428272247314453,
+      "learning_rate": 2e-05,
+      "loss": 6.2002,
+      "step": 16570
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 15.427823066711426,
+      "learning_rate": 2e-05,
+      "loss": 6.5236,
+      "step": 16580
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 17.034761428833008,
+      "learning_rate": 2e-05,
+      "loss": 6.57,
+      "step": 16590
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 15.14773941040039,
+      "learning_rate": 2e-05,
+      "loss": 6.4119,
+      "step": 16600
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 15.145598411560059,
+      "learning_rate": 2e-05,
+      "loss": 6.3194,
+      "step": 16610
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 16.085742950439453,
+      "learning_rate": 2e-05,
+      "loss": 6.3297,
+      "step": 16620
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 14.04321575164795,
+      "learning_rate": 2e-05,
+      "loss": 6.3645,
+      "step": 16630
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 15.112314224243164,
+      "learning_rate": 2e-05,
+      "loss": 6.4765,
+      "step": 16640
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 15.185831069946289,
+      "learning_rate": 2e-05,
+      "loss": 6.367,
+      "step": 16650
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 16.96576690673828,
+      "learning_rate": 2e-05,
+      "loss": 6.4536,
+      "step": 16660
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 14.061229705810547,
+      "learning_rate": 2e-05,
+      "loss": 6.4912,
+      "step": 16670
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 16.532245635986328,
+      "learning_rate": 2e-05,
+      "loss": 6.3902,
+      "step": 16680
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 15.878189086914062,
+      "learning_rate": 2e-05,
+      "loss": 6.6136,
+      "step": 16690
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 15.063907623291016,
+      "learning_rate": 2e-05,
+      "loss": 6.3866,
+      "step": 16700
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 16.407148361206055,
+      "learning_rate": 2e-05,
+      "loss": 6.2772,
+      "step": 16710
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 14.829337120056152,
+      "learning_rate": 2e-05,
+      "loss": 6.4445,
+      "step": 16720
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 17.683502197265625,
+      "learning_rate": 2e-05,
+      "loss": 6.2246,
+      "step": 16730
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 16.88100814819336,
+      "learning_rate": 2e-05,
+      "loss": 6.371,
+      "step": 16740
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 16.445804595947266,
+      "learning_rate": 2e-05,
+      "loss": 6.6432,
+      "step": 16750
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 15.33750057220459,
+      "learning_rate": 2e-05,
+      "loss": 6.3729,
+      "step": 16760
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 15.61681079864502,
+      "learning_rate": 2e-05,
+      "loss": 6.4183,
+      "step": 16770
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 16.720821380615234,
+      "learning_rate": 2e-05,
+      "loss": 6.4322,
+      "step": 16780
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 16.895566940307617,
+      "learning_rate": 2e-05,
+      "loss": 6.5033,
+      "step": 16790
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 16.051645278930664,
+      "learning_rate": 2e-05,
+      "loss": 6.402,
+      "step": 16800
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 15.376147270202637,
+      "learning_rate": 2e-05,
+      "loss": 6.4331,
+      "step": 16810
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 14.922941207885742,
+      "learning_rate": 2e-05,
+      "loss": 6.4638,
+      "step": 16820
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 14.51933765411377,
+      "learning_rate": 2e-05,
+      "loss": 6.5636,
+      "step": 16830
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 15.90237045288086,
+      "learning_rate": 2e-05,
+      "loss": 6.3235,
+      "step": 16840
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 14.201658248901367,
+      "learning_rate": 2e-05,
+      "loss": 6.4703,
+      "step": 16850
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 17.214933395385742,
+      "learning_rate": 2e-05,
+      "loss": 6.1164,
+      "step": 16860
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 15.45417308807373,
+      "learning_rate": 2e-05,
+      "loss": 6.5375,
+      "step": 16870
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 15.57089614868164,
+      "learning_rate": 2e-05,
+      "loss": 6.5076,
+      "step": 16880
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 14.282402992248535,
+      "learning_rate": 2e-05,
+      "loss": 6.5648,
+      "step": 16890
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 16.020814895629883,
+      "learning_rate": 2e-05,
+      "loss": 6.3186,
+      "step": 16900
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 15.972450256347656,
+      "learning_rate": 2e-05,
+      "loss": 6.4534,
+      "step": 16910
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 15.032903671264648,
+      "learning_rate": 2e-05,
+      "loss": 6.4298,
+      "step": 16920
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 15.59566879272461,
+      "learning_rate": 2e-05,
+      "loss": 6.3784,
+      "step": 16930
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 13.764512062072754,
+      "learning_rate": 2e-05,
+      "loss": 6.4685,
+      "step": 16940
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 17.923255920410156,
+      "learning_rate": 2e-05,
+      "loss": 6.4123,
+      "step": 16950
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 16.73320198059082,
+      "learning_rate": 2e-05,
+      "loss": 6.3553,
+      "step": 16960
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 14.744147300720215,
+      "learning_rate": 2e-05,
+      "loss": 6.4843,
+      "step": 16970
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 15.430245399475098,
+      "learning_rate": 2e-05,
+      "loss": 6.5294,
+      "step": 16980
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 15.154815673828125,
+      "learning_rate": 2e-05,
+      "loss": 6.3755,
+      "step": 16990
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 15.524189949035645,
+      "learning_rate": 2e-05,
+      "loss": 6.3872,
+      "step": 17000
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 17.528409957885742,
+      "learning_rate": 2e-05,
+      "loss": 6.2884,
+      "step": 17010
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 15.907470703125,
+      "learning_rate": 2e-05,
+      "loss": 6.3414,
+      "step": 17020
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 15.841470718383789,
+      "learning_rate": 2e-05,
+      "loss": 6.4827,
+      "step": 17030
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 15.527191162109375,
+      "learning_rate": 2e-05,
+      "loss": 6.583,
+      "step": 17040
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 14.948493957519531,
+      "learning_rate": 2e-05,
+      "loss": 6.5091,
+      "step": 17050
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 15.477849006652832,
+      "learning_rate": 2e-05,
+      "loss": 6.4073,
+      "step": 17060
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 14.027935981750488,
+      "learning_rate": 2e-05,
+      "loss": 6.4253,
+      "step": 17070
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 15.432393074035645,
+      "learning_rate": 2e-05,
+      "loss": 6.6746,
+      "step": 17080
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 14.713491439819336,
+      "learning_rate": 2e-05,
+      "loss": 6.2242,
+      "step": 17090
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 14.431781768798828,
+      "learning_rate": 2e-05,
+      "loss": 6.4744,
+      "step": 17100
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 15.861937522888184,
+      "learning_rate": 2e-05,
+      "loss": 6.6046,
+      "step": 17110
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 14.785232543945312,
+      "learning_rate": 2e-05,
+      "loss": 6.4316,
+      "step": 17120
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 14.362011909484863,
+      "learning_rate": 2e-05,
+      "loss": 6.5845,
+      "step": 17130
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 15.790543556213379,
+      "learning_rate": 2e-05,
+      "loss": 6.208,
+      "step": 17140
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 16.220972061157227,
+      "learning_rate": 2e-05,
+      "loss": 6.3257,
+      "step": 17150
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 16.348140716552734,
+      "learning_rate": 2e-05,
+      "loss": 6.7068,
+      "step": 17160
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 14.53695297241211,
+      "learning_rate": 2e-05,
+      "loss": 6.3767,
+      "step": 17170
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 15.674139022827148,
+      "learning_rate": 2e-05,
+      "loss": 6.3758,
+      "step": 17180
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 15.621737480163574,
+      "learning_rate": 2e-05,
+      "loss": 6.4031,
+      "step": 17190
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 15.915032386779785,
+      "learning_rate": 2e-05,
+      "loss": 6.3065,
+      "step": 17200
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 14.112237930297852,
+      "learning_rate": 2e-05,
+      "loss": 6.448,
+      "step": 17210
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 14.610252380371094,
+      "learning_rate": 2e-05,
+      "loss": 6.374,
+      "step": 17220
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 16.43100357055664,
+      "learning_rate": 2e-05,
+      "loss": 6.3906,
+      "step": 17230
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 16.207136154174805,
+      "learning_rate": 2e-05,
+      "loss": 6.1154,
+      "step": 17240
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 16.097253799438477,
+      "learning_rate": 2e-05,
+      "loss": 6.2636,
+      "step": 17250
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 14.790066719055176,
+      "learning_rate": 2e-05,
+      "loss": 6.1698,
+      "step": 17260
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 17.181846618652344,
+      "learning_rate": 2e-05,
+      "loss": 6.247,
+      "step": 17270
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 14.363494873046875,
+      "learning_rate": 2e-05,
+      "loss": 6.3573,
+      "step": 17280
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 16.31484031677246,
+      "learning_rate": 2e-05,
+      "loss": 6.3242,
+      "step": 17290
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 16.39850425720215,
+      "learning_rate": 2e-05,
+      "loss": 6.4113,
+      "step": 17300
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 15.576443672180176,
+      "learning_rate": 2e-05,
+      "loss": 6.2096,
+      "step": 17310
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 14.851789474487305,
+      "learning_rate": 2e-05,
+      "loss": 6.2807,
+      "step": 17320
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 15.378833770751953,
+      "learning_rate": 2e-05,
+      "loss": 6.1079,
+      "step": 17330
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 15.224509239196777,
+      "learning_rate": 2e-05,
+      "loss": 6.3602,
+      "step": 17340
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 15.412925720214844,
+      "learning_rate": 2e-05,
+      "loss": 6.3439,
+      "step": 17350
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 15.330805778503418,
+      "learning_rate": 2e-05,
+      "loss": 6.2882,
+      "step": 17360
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 15.383980751037598,
+      "learning_rate": 2e-05,
+      "loss": 6.3224,
+      "step": 17370
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 15.20047378540039,
+      "learning_rate": 2e-05,
+      "loss": 6.3545,
+      "step": 17380
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 14.811988830566406,
+      "learning_rate": 2e-05,
+      "loss": 6.2847,
+      "step": 17390
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 15.751958847045898,
+      "learning_rate": 2e-05,
+      "loss": 6.3572,
+      "step": 17400
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 15.148112297058105,
+      "learning_rate": 2e-05,
+      "loss": 6.1849,
+      "step": 17410
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 15.617837905883789,
+      "learning_rate": 2e-05,
+      "loss": 6.5101,
+      "step": 17420
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 16.74441909790039,
+      "learning_rate": 2e-05,
+      "loss": 6.3028,
+      "step": 17430
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 15.869138717651367,
+      "learning_rate": 2e-05,
+      "loss": 6.1736,
+      "step": 17440
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 14.541826248168945,
+      "learning_rate": 2e-05,
+      "loss": 6.5135,
+      "step": 17450
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 13.694757461547852,
+      "learning_rate": 2e-05,
+      "loss": 6.3508,
+      "step": 17460
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 16.05943489074707,
+      "learning_rate": 2e-05,
+      "loss": 6.4004,
+      "step": 17470
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 16.171180725097656,
+      "learning_rate": 2e-05,
+      "loss": 6.1743,
+      "step": 17480
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 15.65196704864502,
+      "learning_rate": 2e-05,
+      "loss": 6.4541,
+      "step": 17490
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 15.453271865844727,
+      "learning_rate": 2e-05,
+      "loss": 6.2176,
+      "step": 17500
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 16.472261428833008,
+      "learning_rate": 2e-05,
+      "loss": 6.5269,
+      "step": 17510
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 15.546710968017578,
+      "learning_rate": 2e-05,
+      "loss": 6.4769,
+      "step": 17520
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 15.737564086914062,
+      "learning_rate": 2e-05,
+      "loss": 6.1915,
+      "step": 17530
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 16.295520782470703,
+      "learning_rate": 2e-05,
+      "loss": 6.1528,
+      "step": 17540
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 17.075054168701172,
+      "learning_rate": 2e-05,
+      "loss": 6.4266,
+      "step": 17550
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 16.729127883911133,
+      "learning_rate": 2e-05,
+      "loss": 6.3183,
+      "step": 17560
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 16.02288055419922,
+      "learning_rate": 2e-05,
+      "loss": 6.455,
+      "step": 17570
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 15.7424898147583,
+      "learning_rate": 2e-05,
+      "loss": 6.212,
+      "step": 17580
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 14.914790153503418,
+      "learning_rate": 2e-05,
+      "loss": 6.1827,
+      "step": 17590
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 15.75505256652832,
+      "learning_rate": 2e-05,
+      "loss": 6.2545,
+      "step": 17600
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 15.695699691772461,
+      "learning_rate": 2e-05,
+      "loss": 6.1703,
+      "step": 17610
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 16.04456901550293,
+      "learning_rate": 2e-05,
+      "loss": 6.3854,
+      "step": 17620
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 15.872148513793945,
+      "learning_rate": 2e-05,
+      "loss": 6.3394,
+      "step": 17630
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 14.977651596069336,
+      "learning_rate": 2e-05,
+      "loss": 6.1262,
+      "step": 17640
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 15.181034088134766,
+      "learning_rate": 2e-05,
+      "loss": 6.3101,
+      "step": 17650
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 15.670106887817383,
+      "learning_rate": 2e-05,
+      "loss": 6.3055,
+      "step": 17660
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 16.3615665435791,
+      "learning_rate": 2e-05,
+      "loss": 6.4221,
+      "step": 17670
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 15.825201988220215,
+      "learning_rate": 2e-05,
+      "loss": 6.3916,
+      "step": 17680
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 14.1323823928833,
+      "learning_rate": 2e-05,
+      "loss": 6.4108,
+      "step": 17690
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 15.443329811096191,
+      "learning_rate": 2e-05,
+      "loss": 6.4489,
+      "step": 17700
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 17.505895614624023,
+      "learning_rate": 2e-05,
+      "loss": 6.1396,
+      "step": 17710
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 14.360444068908691,
+      "learning_rate": 2e-05,
+      "loss": 6.1325,
+      "step": 17720
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 14.146031379699707,
+      "learning_rate": 2e-05,
+      "loss": 6.3175,
+      "step": 17730
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 16.414203643798828,
+      "learning_rate": 2e-05,
+      "loss": 6.2563,
+      "step": 17740
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 14.6528959274292,
+      "learning_rate": 2e-05,
+      "loss": 6.2208,
+      "step": 17750
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 14.82217025756836,
+      "learning_rate": 2e-05,
+      "loss": 6.1185,
+      "step": 17760
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 18.766132354736328,
+      "learning_rate": 2e-05,
+      "loss": 6.2115,
+      "step": 17770
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 14.955482482910156,
+      "learning_rate": 2e-05,
+      "loss": 6.1853,
+      "step": 17780
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 16.865880966186523,
+      "learning_rate": 2e-05,
+      "loss": 6.2739,
+      "step": 17790
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 16.20660400390625,
+      "learning_rate": 2e-05,
+      "loss": 6.5087,
+      "step": 17800
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 16.103879928588867,
+      "learning_rate": 2e-05,
+      "loss": 6.3825,
+      "step": 17810
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 15.601266860961914,
+      "learning_rate": 2e-05,
+      "loss": 6.5025,
+      "step": 17820
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 13.353988647460938,
+      "learning_rate": 2e-05,
+      "loss": 6.1621,
+      "step": 17830
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 15.227518081665039,
+      "learning_rate": 2e-05,
+      "loss": 6.2911,
+      "step": 17840
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 15.4844331741333,
+      "learning_rate": 2e-05,
+      "loss": 6.5624,
+      "step": 17850
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 13.58100414276123,
+      "learning_rate": 2e-05,
+      "loss": 6.0261,
+      "step": 17860
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 16.552165985107422,
+      "learning_rate": 2e-05,
+      "loss": 6.1274,
+      "step": 17870
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 15.954316139221191,
+      "learning_rate": 2e-05,
+      "loss": 6.4822,
+      "step": 17880
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 15.806802749633789,
+      "learning_rate": 2e-05,
+      "loss": 6.3296,
+      "step": 17890
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 16.282268524169922,
+      "learning_rate": 2e-05,
+      "loss": 6.2438,
+      "step": 17900
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 14.029036521911621,
+      "learning_rate": 2e-05,
+      "loss": 6.171,
+      "step": 17910
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 16.254344940185547,
+      "learning_rate": 2e-05,
+      "loss": 6.2773,
+      "step": 17920
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 16.644939422607422,
+      "learning_rate": 2e-05,
+      "loss": 6.4085,
+      "step": 17930
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 14.737886428833008,
+      "learning_rate": 2e-05,
+      "loss": 6.3779,
+      "step": 17940
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 15.40264892578125,
+      "learning_rate": 2e-05,
+      "loss": 6.2238,
+      "step": 17950
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 15.45809555053711,
+      "learning_rate": 2e-05,
+      "loss": 6.2731,
+      "step": 17960
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 14.831330299377441,
+      "learning_rate": 2e-05,
+      "loss": 6.2754,
+      "step": 17970
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 15.763202667236328,
+      "learning_rate": 2e-05,
+      "loss": 6.3801,
+      "step": 17980
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 17.454090118408203,
+      "learning_rate": 2e-05,
+      "loss": 6.2817,
+      "step": 17990
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 15.10185718536377,
+      "learning_rate": 2e-05,
+      "loss": 6.2394,
+      "step": 18000
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 15.768720626831055,
+      "learning_rate": 2e-05,
+      "loss": 6.04,
+      "step": 18010
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 15.462987899780273,
+      "learning_rate": 2e-05,
+      "loss": 6.39,
+      "step": 18020
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 16.94598960876465,
+      "learning_rate": 2e-05,
+      "loss": 6.3897,
+      "step": 18030
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 17.118303298950195,
+      "learning_rate": 2e-05,
+      "loss": 6.2868,
+      "step": 18040
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 17.321847915649414,
+      "learning_rate": 2e-05,
+      "loss": 6.3052,
+      "step": 18050
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 15.140386581420898,
+      "learning_rate": 2e-05,
+      "loss": 6.3368,
+      "step": 18060
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 16.900842666625977,
+      "learning_rate": 2e-05,
+      "loss": 6.3568,
+      "step": 18070
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 16.495912551879883,
+      "learning_rate": 2e-05,
+      "loss": 6.2167,
+      "step": 18080
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 16.369455337524414,
+      "learning_rate": 2e-05,
+      "loss": 6.2903,
+      "step": 18090
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 15.903060913085938,
+      "learning_rate": 2e-05,
+      "loss": 6.1567,
+      "step": 18100
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 15.655290603637695,
+      "learning_rate": 2e-05,
+      "loss": 6.2405,
+      "step": 18110
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 14.273642539978027,
+      "learning_rate": 2e-05,
+      "loss": 6.2077,
+      "step": 18120
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 22.1506290435791,
+      "learning_rate": 2e-05,
+      "loss": 6.3718,
+      "step": 18130
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 15.582695960998535,
+      "learning_rate": 2e-05,
+      "loss": 6.1459,
+      "step": 18140
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 14.475653648376465,
+      "learning_rate": 2e-05,
+      "loss": 6.0962,
+      "step": 18150
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 14.587137222290039,
+      "learning_rate": 2e-05,
+      "loss": 6.3205,
+      "step": 18160
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 14.008703231811523,
+      "learning_rate": 2e-05,
+      "loss": 6.4416,
+      "step": 18170
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 14.803190231323242,
+      "learning_rate": 2e-05,
+      "loss": 6.2533,
+      "step": 18180
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 16.931217193603516,
+      "learning_rate": 2e-05,
+      "loss": 6.3646,
+      "step": 18190
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 15.44107723236084,
+      "learning_rate": 2e-05,
+      "loss": 6.3597,
+      "step": 18200
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 14.536946296691895,
+      "learning_rate": 2e-05,
+      "loss": 6.1444,
+      "step": 18210
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 16.937063217163086,
+      "learning_rate": 2e-05,
+      "loss": 6.1361,
+      "step": 18220
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 16.07759666442871,
+      "learning_rate": 2e-05,
+      "loss": 6.2773,
+      "step": 18230
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 18.095205307006836,
+      "learning_rate": 2e-05,
+      "loss": 6.1689,
+      "step": 18240
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 17.034482955932617,
+      "learning_rate": 2e-05,
+      "loss": 6.3041,
+      "step": 18250
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 16.474294662475586,
+      "learning_rate": 2e-05,
+      "loss": 6.6822,
+      "step": 18260
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 13.549620628356934,
+      "learning_rate": 2e-05,
+      "loss": 6.5143,
+      "step": 18270
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 15.335073471069336,
+      "learning_rate": 2e-05,
+      "loss": 6.22,
+      "step": 18280
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 15.05056095123291,
+      "learning_rate": 2e-05,
+      "loss": 6.1926,
+      "step": 18290
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 16.813846588134766,
+      "learning_rate": 2e-05,
+      "loss": 6.0921,
+      "step": 18300
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 15.718114852905273,
+      "learning_rate": 2e-05,
+      "loss": 6.1658,
+      "step": 18310
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 15.986509323120117,
+      "learning_rate": 2e-05,
+      "loss": 6.4874,
+      "step": 18320
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 16.196514129638672,
+      "learning_rate": 2e-05,
+      "loss": 6.3124,
+      "step": 18330
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 17.50399398803711,
+      "learning_rate": 2e-05,
+      "loss": 6.0365,
+      "step": 18340
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 15.265746116638184,
+      "learning_rate": 2e-05,
+      "loss": 6.2429,
+      "step": 18350
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 15.341573715209961,
+      "learning_rate": 2e-05,
+      "loss": 6.2164,
+      "step": 18360
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 15.688251495361328,
+      "learning_rate": 2e-05,
+      "loss": 6.2718,
+      "step": 18370
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 16.826013565063477,
+      "learning_rate": 2e-05,
+      "loss": 6.2233,
+      "step": 18380
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 17.19365692138672,
+      "learning_rate": 2e-05,
+      "loss": 6.3999,
+      "step": 18390
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 14.397275924682617,
+      "learning_rate": 2e-05,
+      "loss": 6.2494,
+      "step": 18400
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 15.839926719665527,
+      "learning_rate": 2e-05,
+      "loss": 6.2276,
+      "step": 18410
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 15.966907501220703,
+      "learning_rate": 2e-05,
+      "loss": 6.1722,
+      "step": 18420
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 15.419812202453613,
+      "learning_rate": 2e-05,
+      "loss": 6.3667,
+      "step": 18430
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 17.960243225097656,
+      "learning_rate": 2e-05,
+      "loss": 6.1161,
+      "step": 18440
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 17.00394630432129,
+      "learning_rate": 2e-05,
+      "loss": 6.0787,
+      "step": 18450
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 15.028756141662598,
+      "learning_rate": 2e-05,
+      "loss": 6.1299,
+      "step": 18460
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 16.4580078125,
+      "learning_rate": 2e-05,
+      "loss": 6.2162,
+      "step": 18470
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 15.266437530517578,
+      "learning_rate": 2e-05,
+      "loss": 6.1735,
+      "step": 18480
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 14.986762046813965,
+      "learning_rate": 2e-05,
+      "loss": 5.8828,
+      "step": 18490
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 16.2413330078125,
+      "learning_rate": 2e-05,
+      "loss": 6.342,
+      "step": 18500
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 15.510326385498047,
+      "learning_rate": 2e-05,
+      "loss": 6.0892,
+      "step": 18510
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 14.807727813720703,
+      "learning_rate": 2e-05,
+      "loss": 6.5027,
+      "step": 18520
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 16.764667510986328,
+      "learning_rate": 2e-05,
+      "loss": 6.1809,
+      "step": 18530
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 14.886832237243652,
+      "learning_rate": 2e-05,
+      "loss": 6.422,
+      "step": 18540
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 16.083106994628906,
+      "learning_rate": 2e-05,
+      "loss": 6.1705,
+      "step": 18550
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 15.904338836669922,
+      "learning_rate": 2e-05,
+      "loss": 6.2574,
+      "step": 18560
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 16.794803619384766,
+      "learning_rate": 2e-05,
+      "loss": 6.2639,
+      "step": 18570
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 16.16462516784668,
+      "learning_rate": 2e-05,
+      "loss": 6.143,
+      "step": 18580
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 14.623804092407227,
+      "learning_rate": 2e-05,
+      "loss": 6.0229,
+      "step": 18590
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 16.243864059448242,
+      "learning_rate": 2e-05,
+      "loss": 6.4045,
+      "step": 18600
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 15.087252616882324,
+      "learning_rate": 2e-05,
+      "loss": 6.4187,
+      "step": 18610
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 15.045507431030273,
+      "learning_rate": 2e-05,
+      "loss": 6.2327,
+      "step": 18620
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 16.611663818359375,
+      "learning_rate": 2e-05,
+      "loss": 6.5491,
+      "step": 18630
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 14.559623718261719,
+      "learning_rate": 2e-05,
+      "loss": 6.0034,
+      "step": 18640
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 14.97830581665039,
+      "learning_rate": 2e-05,
+      "loss": 6.1751,
+      "step": 18650
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 15.34595012664795,
+      "learning_rate": 2e-05,
+      "loss": 6.1722,
+      "step": 18660
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 19.92973518371582,
+      "learning_rate": 2e-05,
+      "loss": 6.6027,
+      "step": 18670
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 15.257964134216309,
+      "learning_rate": 2e-05,
+      "loss": 6.0853,
+      "step": 18680
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 15.206181526184082,
+      "learning_rate": 2e-05,
+      "loss": 5.9515,
+      "step": 18690
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 16.421276092529297,
+      "learning_rate": 2e-05,
+      "loss": 6.2044,
+      "step": 18700
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 13.733359336853027,
+      "learning_rate": 2e-05,
+      "loss": 6.1803,
+      "step": 18710
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 14.752503395080566,
+      "learning_rate": 2e-05,
+      "loss": 6.0538,
+      "step": 18720
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 15.216063499450684,
+      "learning_rate": 2e-05,
+      "loss": 6.2469,
+      "step": 18730
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 15.779196739196777,
+      "learning_rate": 2e-05,
+      "loss": 6.052,
+      "step": 18740
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 14.515335083007812,
+      "learning_rate": 2e-05,
+      "loss": 6.0535,
+      "step": 18750
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 15.444271087646484,
+      "learning_rate": 2e-05,
+      "loss": 6.149,
+      "step": 18760
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 16.57893943786621,
+      "learning_rate": 2e-05,
+      "loss": 6.3572,
+      "step": 18770
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 15.839405059814453,
+      "learning_rate": 2e-05,
+      "loss": 6.2138,
+      "step": 18780
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 16.1677188873291,
+      "learning_rate": 2e-05,
+      "loss": 6.214,
+      "step": 18790
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 14.890835762023926,
+      "learning_rate": 2e-05,
+      "loss": 6.26,
+      "step": 18800
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 15.936962127685547,
+      "learning_rate": 2e-05,
+      "loss": 6.176,
+      "step": 18810
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 17.090944290161133,
+      "learning_rate": 2e-05,
+      "loss": 6.2463,
+      "step": 18820
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 16.930789947509766,
+      "learning_rate": 2e-05,
+      "loss": 6.2716,
+      "step": 18830
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 15.275690078735352,
+      "learning_rate": 2e-05,
+      "loss": 6.0659,
+      "step": 18840
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 17.437715530395508,
+      "learning_rate": 2e-05,
+      "loss": 6.2374,
+      "step": 18850
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 14.972350120544434,
+      "learning_rate": 2e-05,
+      "loss": 6.1792,
+      "step": 18860
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 16.9863338470459,
+      "learning_rate": 2e-05,
+      "loss": 6.2091,
+      "step": 18870
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 15.763738632202148,
+      "learning_rate": 2e-05,
+      "loss": 6.284,
+      "step": 18880
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 17.083955764770508,
+      "learning_rate": 2e-05,
+      "loss": 6.0468,
+      "step": 18890
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 15.04819393157959,
+      "learning_rate": 2e-05,
+      "loss": 6.0508,
+      "step": 18900
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 15.513570785522461,
+      "learning_rate": 2e-05,
+      "loss": 6.1,
+      "step": 18910
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 14.337035179138184,
+      "learning_rate": 2e-05,
+      "loss": 6.2206,
+      "step": 18920
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 15.504044532775879,
+      "learning_rate": 2e-05,
+      "loss": 6.2167,
+      "step": 18930
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 16.772335052490234,
+      "learning_rate": 2e-05,
+      "loss": 6.1324,
+      "step": 18940
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 15.534796714782715,
+      "learning_rate": 2e-05,
+      "loss": 6.2517,
+      "step": 18950
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 15.162538528442383,
+      "learning_rate": 2e-05,
+      "loss": 6.4671,
+      "step": 18960
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 14.35653305053711,
+      "learning_rate": 2e-05,
+      "loss": 6.085,
+      "step": 18970
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 16.467140197753906,
+      "learning_rate": 2e-05,
+      "loss": 6.0649,
+      "step": 18980
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 15.817630767822266,
+      "learning_rate": 2e-05,
+      "loss": 6.1263,
+      "step": 18990
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 16.56173324584961,
+      "learning_rate": 2e-05,
+      "loss": 6.1252,
+      "step": 19000
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 19.00716781616211,
+      "learning_rate": 2e-05,
+      "loss": 6.1405,
+      "step": 19010
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 15.938605308532715,
+      "learning_rate": 2e-05,
+      "loss": 5.9821,
+      "step": 19020
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 18.251745223999023,
+      "learning_rate": 2e-05,
+      "loss": 6.4292,
+      "step": 19030
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 16.25697135925293,
+      "learning_rate": 2e-05,
+      "loss": 6.2765,
+      "step": 19040
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 16.630950927734375,
+      "learning_rate": 2e-05,
+      "loss": 6.2554,
+      "step": 19050
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 15.682162284851074,
+      "learning_rate": 2e-05,
+      "loss": 6.3229,
+      "step": 19060
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 14.89696979522705,
+      "learning_rate": 2e-05,
+      "loss": 6.2721,
+      "step": 19070
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 13.963735580444336,
+      "learning_rate": 2e-05,
+      "loss": 6.3747,
+      "step": 19080
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 15.614315032958984,
+      "learning_rate": 2e-05,
+      "loss": 6.363,
+      "step": 19090
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 15.260639190673828,
+      "learning_rate": 2e-05,
+      "loss": 6.3291,
+      "step": 19100
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 15.79055404663086,
+      "learning_rate": 2e-05,
+      "loss": 6.1541,
+      "step": 19110
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 16.482818603515625,
+      "learning_rate": 2e-05,
+      "loss": 6.3342,
+      "step": 19120
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 15.904244422912598,
+      "learning_rate": 2e-05,
+      "loss": 6.1941,
+      "step": 19130
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 16.00639533996582,
+      "learning_rate": 2e-05,
+      "loss": 6.016,
+      "step": 19140
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 14.305451393127441,
+      "learning_rate": 2e-05,
+      "loss": 6.0176,
+      "step": 19150
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 15.954229354858398,
+      "learning_rate": 2e-05,
+      "loss": 6.0271,
+      "step": 19160
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 15.551297187805176,
+      "learning_rate": 2e-05,
+      "loss": 6.144,
+      "step": 19170
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 16.436447143554688,
+      "learning_rate": 2e-05,
+      "loss": 6.4536,
+      "step": 19180
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 16.605831146240234,
+      "learning_rate": 2e-05,
+      "loss": 6.2235,
+      "step": 19190
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 15.751364707946777,
+      "learning_rate": 2e-05,
+      "loss": 6.2085,
+      "step": 19200
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 15.618922233581543,
+      "learning_rate": 2e-05,
+      "loss": 6.1201,
+      "step": 19210
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 17.50379180908203,
+      "learning_rate": 2e-05,
+      "loss": 5.9254,
+      "step": 19220
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 13.701366424560547,
+      "learning_rate": 2e-05,
+      "loss": 6.384,
+      "step": 19230
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 16.033836364746094,
+      "learning_rate": 2e-05,
+      "loss": 6.1229,
+      "step": 19240
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 16.395526885986328,
+      "learning_rate": 2e-05,
+      "loss": 6.442,
+      "step": 19250
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 16.095134735107422,
+      "learning_rate": 2e-05,
+      "loss": 6.2393,
+      "step": 19260
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 14.26583480834961,
+      "learning_rate": 2e-05,
+      "loss": 6.1817,
+      "step": 19270
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 16.7813777923584,
+      "learning_rate": 2e-05,
+      "loss": 6.1752,
+      "step": 19280
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 16.28823471069336,
+      "learning_rate": 2e-05,
+      "loss": 6.0152,
+      "step": 19290
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 14.35616683959961,
+      "learning_rate": 2e-05,
+      "loss": 6.131,
+      "step": 19300
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 14.425654411315918,
+      "learning_rate": 2e-05,
+      "loss": 5.9855,
+      "step": 19310
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 14.78880786895752,
+      "learning_rate": 2e-05,
+      "loss": 6.1342,
+      "step": 19320
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 15.537471771240234,
+      "learning_rate": 2e-05,
+      "loss": 6.4523,
+      "step": 19330
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 14.942159652709961,
+      "learning_rate": 2e-05,
+      "loss": 6.3222,
+      "step": 19340
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 16.288259506225586,
+      "learning_rate": 2e-05,
+      "loss": 6.1027,
+      "step": 19350
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 13.213309288024902,
+      "learning_rate": 2e-05,
+      "loss": 5.8228,
+      "step": 19360
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 17.383317947387695,
+      "learning_rate": 2e-05,
+      "loss": 5.929,
+      "step": 19370
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 15.846534729003906,
+      "learning_rate": 2e-05,
+      "loss": 6.1113,
+      "step": 19380
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 17.370426177978516,
+      "learning_rate": 2e-05,
+      "loss": 6.2562,
+      "step": 19390
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 16.86368179321289,
+      "learning_rate": 2e-05,
+      "loss": 6.2217,
+      "step": 19400
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 14.378195762634277,
+      "learning_rate": 2e-05,
+      "loss": 6.1339,
+      "step": 19410
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 16.49493980407715,
+      "learning_rate": 2e-05,
+      "loss": 6.2527,
+      "step": 19420
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 15.70553970336914,
+      "learning_rate": 2e-05,
+      "loss": 6.1987,
+      "step": 19430
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 15.951569557189941,
+      "learning_rate": 2e-05,
+      "loss": 6.0105,
+      "step": 19440
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 15.885408401489258,
+      "learning_rate": 2e-05,
+      "loss": 6.264,
+      "step": 19450
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 17.4884090423584,
+      "learning_rate": 2e-05,
+      "loss": 6.1321,
+      "step": 19460
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 15.07073974609375,
+      "learning_rate": 2e-05,
+      "loss": 6.3975,
+      "step": 19470
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 16.054773330688477,
+      "learning_rate": 2e-05,
+      "loss": 6.0824,
+      "step": 19480
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 15.820854187011719,
+      "learning_rate": 2e-05,
+      "loss": 6.2078,
+      "step": 19490
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 15.058493614196777,
+      "learning_rate": 2e-05,
+      "loss": 5.9512,
+      "step": 19500
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 15.133792877197266,
+      "learning_rate": 2e-05,
+      "loss": 6.1448,
+      "step": 19510
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 18.522838592529297,
+      "learning_rate": 2e-05,
+      "loss": 6.053,
+      "step": 19520
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 17.351898193359375,
+      "learning_rate": 2e-05,
+      "loss": 6.1033,
+      "step": 19530
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 16.317811965942383,
+      "learning_rate": 2e-05,
+      "loss": 6.2578,
+      "step": 19540
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 16.672340393066406,
+      "learning_rate": 2e-05,
+      "loss": 6.348,
+      "step": 19550
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 16.26076889038086,
+      "learning_rate": 2e-05,
+      "loss": 6.2305,
+      "step": 19560
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 14.018013000488281,
+      "learning_rate": 2e-05,
+      "loss": 6.1956,
+      "step": 19570
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 15.856391906738281,
+      "learning_rate": 2e-05,
+      "loss": 6.248,
+      "step": 19580
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 14.423810958862305,
+      "learning_rate": 2e-05,
+      "loss": 6.0384,
+      "step": 19590
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 16.43643569946289,
+      "learning_rate": 2e-05,
+      "loss": 6.2401,
+      "step": 19600
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 15.939482688903809,
+      "learning_rate": 2e-05,
+      "loss": 5.962,
+      "step": 19610
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 16.011877059936523,
+      "learning_rate": 2e-05,
+      "loss": 6.1571,
+      "step": 19620
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 16.33401107788086,
+      "learning_rate": 2e-05,
+      "loss": 6.2319,
+      "step": 19630
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 16.382476806640625,
+      "learning_rate": 2e-05,
+      "loss": 6.2658,
+      "step": 19640
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 16.394439697265625,
+      "learning_rate": 2e-05,
+      "loss": 6.2197,
+      "step": 19650
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 17.995256423950195,
+      "learning_rate": 2e-05,
+      "loss": 6.196,
+      "step": 19660
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 16.111345291137695,
+      "learning_rate": 2e-05,
+      "loss": 6.2215,
+      "step": 19670
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 14.951397895812988,
+      "learning_rate": 2e-05,
+      "loss": 6.1726,
+      "step": 19680
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 15.498074531555176,
+      "learning_rate": 2e-05,
+      "loss": 5.889,
+      "step": 19690
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 15.584684371948242,
+      "learning_rate": 2e-05,
+      "loss": 6.1877,
+      "step": 19700
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 16.766324996948242,
+      "learning_rate": 2e-05,
+      "loss": 6.1133,
+      "step": 19710
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 15.294193267822266,
+      "learning_rate": 2e-05,
+      "loss": 6.0956,
+      "step": 19720
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 14.753824234008789,
+      "learning_rate": 2e-05,
+      "loss": 6.1695,
+      "step": 19730
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 16.538864135742188,
+      "learning_rate": 2e-05,
+      "loss": 6.1917,
+      "step": 19740
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 17.47024154663086,
+      "learning_rate": 2e-05,
+      "loss": 6.2126,
+      "step": 19750
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 18.143266677856445,
+      "learning_rate": 2e-05,
+      "loss": 6.2023,
+      "step": 19760
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 16.22307014465332,
+      "learning_rate": 2e-05,
+      "loss": 6.0643,
+      "step": 19770
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 13.70028305053711,
+      "learning_rate": 2e-05,
+      "loss": 6.2267,
+      "step": 19780
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 14.707581520080566,
+      "learning_rate": 2e-05,
+      "loss": 6.3374,
+      "step": 19790
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 16.11334228515625,
+      "learning_rate": 2e-05,
+      "loss": 6.3045,
+      "step": 19800
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 15.95144271850586,
+      "learning_rate": 2e-05,
+      "loss": 5.9112,
+      "step": 19810
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 16.505090713500977,
+      "learning_rate": 2e-05,
+      "loss": 6.2221,
+      "step": 19820
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 17.96413803100586,
+      "learning_rate": 2e-05,
+      "loss": 5.8948,
+      "step": 19830
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 17.1361083984375,
+      "learning_rate": 2e-05,
+      "loss": 6.3453,
+      "step": 19840
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 16.134185791015625,
+      "learning_rate": 2e-05,
+      "loss": 6.1164,
+      "step": 19850
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 16.040380477905273,
+      "learning_rate": 2e-05,
+      "loss": 6.1684,
+      "step": 19860
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 13.387770652770996,
+      "learning_rate": 2e-05,
+      "loss": 5.8154,
+      "step": 19870
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 16.5738525390625,
+      "learning_rate": 2e-05,
+      "loss": 5.9344,
+      "step": 19880
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 17.08277130126953,
+      "learning_rate": 2e-05,
+      "loss": 6.2705,
+      "step": 19890
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 16.234926223754883,
+      "learning_rate": 2e-05,
+      "loss": 6.2724,
+      "step": 19900
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 16.746789932250977,
+      "learning_rate": 2e-05,
+      "loss": 6.0371,
+      "step": 19910
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 15.765117645263672,
+      "learning_rate": 2e-05,
+      "loss": 6.1986,
+      "step": 19920
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 16.663724899291992,
+      "learning_rate": 2e-05,
+      "loss": 6.3417,
+      "step": 19930
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 16.99330711364746,
+      "learning_rate": 2e-05,
+      "loss": 6.1477,
+      "step": 19940
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 16.304035186767578,
+      "learning_rate": 2e-05,
+      "loss": 6.2111,
+      "step": 19950
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 15.789288520812988,
+      "learning_rate": 2e-05,
+      "loss": 6.1334,
+      "step": 19960
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 17.210771560668945,
+      "learning_rate": 2e-05,
+      "loss": 5.8805,
+      "step": 19970
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 15.058265686035156,
+      "learning_rate": 2e-05,
+      "loss": 6.1397,
+      "step": 19980
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 16.128198623657227,
+      "learning_rate": 2e-05,
+      "loss": 6.2468,
+      "step": 19990
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 16.79635238647461,
+      "learning_rate": 2e-05,
+      "loss": 6.2141,
+      "step": 20000
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 14.58762264251709,
+      "learning_rate": 2e-05,
+      "loss": 6.1514,
+      "step": 20010
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 15.515220642089844,
+      "learning_rate": 2e-05,
+      "loss": 5.7887,
+      "step": 20020
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 16.22983169555664,
+      "learning_rate": 2e-05,
+      "loss": 6.1164,
+      "step": 20030
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 15.598810195922852,
+      "learning_rate": 2e-05,
+      "loss": 6.37,
+      "step": 20040
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 15.490750312805176,
+      "learning_rate": 2e-05,
+      "loss": 5.9501,
+      "step": 20050
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 13.613740921020508,
+      "learning_rate": 2e-05,
+      "loss": 5.8921,
+      "step": 20060
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 14.59473991394043,
+      "learning_rate": 2e-05,
+      "loss": 6.2356,
+      "step": 20070
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 13.546895980834961,
+      "learning_rate": 2e-05,
+      "loss": 6.2297,
+      "step": 20080
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 15.24441909790039,
+      "learning_rate": 2e-05,
+      "loss": 6.097,
+      "step": 20090
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 16.222192764282227,
+      "learning_rate": 2e-05,
+      "loss": 6.1909,
+      "step": 20100
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 15.674488067626953,
+      "learning_rate": 2e-05,
+      "loss": 6.2166,
+      "step": 20110
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 14.907276153564453,
+      "learning_rate": 2e-05,
+      "loss": 6.0576,
+      "step": 20120
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 16.088756561279297,
+      "learning_rate": 2e-05,
+      "loss": 5.8771,
+      "step": 20130
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 16.42784881591797,
+      "learning_rate": 2e-05,
+      "loss": 5.9388,
+      "step": 20140
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 15.592179298400879,
+      "learning_rate": 2e-05,
+      "loss": 6.2017,
+      "step": 20150
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 16.660579681396484,
+      "learning_rate": 2e-05,
+      "loss": 6.3627,
+      "step": 20160
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 15.94429874420166,
+      "learning_rate": 2e-05,
+      "loss": 6.275,
+      "step": 20170
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 14.291220664978027,
+      "learning_rate": 2e-05,
+      "loss": 6.2084,
+      "step": 20180
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 15.49881649017334,
+      "learning_rate": 2e-05,
+      "loss": 6.3254,
+      "step": 20190
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 15.949112892150879,
+      "learning_rate": 2e-05,
+      "loss": 6.1438,
+      "step": 20200
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 18.238405227661133,
+      "learning_rate": 2e-05,
+      "loss": 6.1501,
+      "step": 20210
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 15.672358512878418,
+      "learning_rate": 2e-05,
+      "loss": 6.2348,
+      "step": 20220
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 15.7500581741333,
+      "learning_rate": 2e-05,
+      "loss": 5.9869,
+      "step": 20230
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 16.402990341186523,
+      "learning_rate": 2e-05,
+      "loss": 6.1552,
+      "step": 20240
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 14.907893180847168,
+      "learning_rate": 2e-05,
+      "loss": 6.1026,
+      "step": 20250
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 16.58329963684082,
+      "learning_rate": 2e-05,
+      "loss": 6.0013,
+      "step": 20260
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 14.750893592834473,
+      "learning_rate": 2e-05,
+      "loss": 5.9433,
+      "step": 20270
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 15.245884895324707,
+      "learning_rate": 2e-05,
+      "loss": 6.1248,
+      "step": 20280
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 17.056673049926758,
+      "learning_rate": 2e-05,
+      "loss": 6.326,
+      "step": 20290
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 15.576537132263184,
+      "learning_rate": 2e-05,
+      "loss": 6.2575,
+      "step": 20300
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 18.190460205078125,
+      "learning_rate": 2e-05,
+      "loss": 5.8895,
+      "step": 20310
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 14.541739463806152,
+      "learning_rate": 2e-05,
+      "loss": 6.0007,
+      "step": 20320
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 15.682243347167969,
+      "learning_rate": 2e-05,
+      "loss": 6.0927,
+      "step": 20330
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 14.83790111541748,
+      "learning_rate": 2e-05,
+      "loss": 6.1423,
+      "step": 20340
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 14.937458992004395,
+      "learning_rate": 2e-05,
+      "loss": 6.3052,
+      "step": 20350
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 14.976640701293945,
+      "learning_rate": 2e-05,
+      "loss": 6.1138,
+      "step": 20360
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 15.172950744628906,
+      "learning_rate": 2e-05,
+      "loss": 6.0196,
+      "step": 20370
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 15.581155776977539,
+      "learning_rate": 2e-05,
+      "loss": 6.2059,
+      "step": 20380
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 18.57327651977539,
+      "learning_rate": 2e-05,
+      "loss": 6.1457,
+      "step": 20390
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 14.579609870910645,
+      "learning_rate": 2e-05,
+      "loss": 6.0725,
+      "step": 20400
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 15.213146209716797,
+      "learning_rate": 2e-05,
+      "loss": 6.1563,
+      "step": 20410
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 17.054174423217773,
+      "learning_rate": 2e-05,
+      "loss": 6.0991,
+      "step": 20420
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 15.089062690734863,
+      "learning_rate": 2e-05,
+      "loss": 6.0704,
+      "step": 20430
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 14.901512145996094,
+      "learning_rate": 2e-05,
+      "loss": 5.8669,
+      "step": 20440
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 15.764300346374512,
+      "learning_rate": 2e-05,
+      "loss": 5.9484,
+      "step": 20450
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 15.675581932067871,
+      "learning_rate": 2e-05,
+      "loss": 6.1701,
+      "step": 20460
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 14.555734634399414,
+      "learning_rate": 2e-05,
+      "loss": 6.0215,
+      "step": 20470
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 17.252225875854492,
+      "learning_rate": 2e-05,
+      "loss": 5.9447,
+      "step": 20480
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 16.63077735900879,
+      "learning_rate": 2e-05,
+      "loss": 6.0538,
+      "step": 20490
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 17.482250213623047,
+      "learning_rate": 2e-05,
+      "loss": 6.3014,
+      "step": 20500
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 13.60798168182373,
+      "learning_rate": 2e-05,
+      "loss": 6.0289,
+      "step": 20510
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 16.521886825561523,
+      "learning_rate": 2e-05,
+      "loss": 6.0168,
+      "step": 20520
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 15.851801872253418,
+      "learning_rate": 2e-05,
+      "loss": 6.0672,
+      "step": 20530
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 17.34090805053711,
+      "learning_rate": 2e-05,
+      "loss": 6.0141,
+      "step": 20540
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 16.600021362304688,
+      "learning_rate": 2e-05,
+      "loss": 6.0783,
+      "step": 20550
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 16.136327743530273,
+      "learning_rate": 2e-05,
+      "loss": 6.0333,
+      "step": 20560
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 15.702614784240723,
+      "learning_rate": 2e-05,
+      "loss": 5.9803,
+      "step": 20570
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 14.833301544189453,
+      "learning_rate": 2e-05,
+      "loss": 5.8479,
+      "step": 20580
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 15.544854164123535,
+      "learning_rate": 2e-05,
+      "loss": 5.8486,
+      "step": 20590
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 16.066429138183594,
+      "learning_rate": 2e-05,
+      "loss": 6.288,
+      "step": 20600
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 15.285183906555176,
+      "learning_rate": 2e-05,
+      "loss": 5.981,
+      "step": 20610
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 15.225436210632324,
+      "learning_rate": 2e-05,
+      "loss": 6.1592,
+      "step": 20620
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 15.740286827087402,
+      "learning_rate": 2e-05,
+      "loss": 6.1198,
+      "step": 20630
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 16.04486656188965,
+      "learning_rate": 2e-05,
+      "loss": 6.0522,
+      "step": 20640
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 16.445690155029297,
+      "learning_rate": 2e-05,
+      "loss": 5.8589,
+      "step": 20650
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 15.795793533325195,
+      "learning_rate": 2e-05,
+      "loss": 5.9324,
+      "step": 20660
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 16.927061080932617,
+      "learning_rate": 2e-05,
+      "loss": 6.0502,
+      "step": 20670
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 17.04458236694336,
+      "learning_rate": 2e-05,
+      "loss": 6.059,
+      "step": 20680
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 14.20008373260498,
+      "learning_rate": 2e-05,
+      "loss": 6.0598,
+      "step": 20690
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 16.812257766723633,
+      "learning_rate": 2e-05,
+      "loss": 5.9603,
+      "step": 20700
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 17.934356689453125,
+      "learning_rate": 2e-05,
+      "loss": 6.2948,
+      "step": 20710
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 15.612955093383789,
+      "learning_rate": 2e-05,
+      "loss": 6.0663,
+      "step": 20720
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 16.09332847595215,
+      "learning_rate": 2e-05,
+      "loss": 6.2831,
+      "step": 20730
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 17.62060546875,
+      "learning_rate": 2e-05,
+      "loss": 5.917,
+      "step": 20740
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 16.85783576965332,
+      "learning_rate": 2e-05,
+      "loss": 6.152,
+      "step": 20750
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 16.36855697631836,
+      "learning_rate": 2e-05,
+      "loss": 5.9757,
+      "step": 20760
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 13.798477172851562,
+      "learning_rate": 2e-05,
+      "loss": 5.9191,
+      "step": 20770
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 15.113873481750488,
+      "learning_rate": 2e-05,
+      "loss": 5.8484,
+      "step": 20780
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 16.43352699279785,
+      "learning_rate": 2e-05,
+      "loss": 5.9393,
+      "step": 20790
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 14.506684303283691,
+      "learning_rate": 2e-05,
+      "loss": 6.1054,
+      "step": 20800
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 15.057573318481445,
+      "learning_rate": 2e-05,
+      "loss": 6.0911,
+      "step": 20810
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 15.717312812805176,
+      "learning_rate": 2e-05,
+      "loss": 6.1586,
+      "step": 20820
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 17.096811294555664,
+      "learning_rate": 2e-05,
+      "loss": 6.1646,
+      "step": 20830
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 15.519171714782715,
+      "learning_rate": 2e-05,
+      "loss": 6.0424,
+      "step": 20840
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 14.659160614013672,
+      "learning_rate": 2e-05,
+      "loss": 5.9176,
+      "step": 20850
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 14.62610912322998,
+      "learning_rate": 2e-05,
+      "loss": 6.0773,
+      "step": 20860
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 17.110504150390625,
+      "learning_rate": 2e-05,
+      "loss": 5.9958,
+      "step": 20870
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 15.251090049743652,
+      "learning_rate": 2e-05,
+      "loss": 6.3424,
+      "step": 20880
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 14.901779174804688,
+      "learning_rate": 2e-05,
+      "loss": 6.011,
+      "step": 20890
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 15.343659400939941,
+      "learning_rate": 2e-05,
+      "loss": 5.9869,
+      "step": 20900
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 16.10903549194336,
+      "learning_rate": 2e-05,
+      "loss": 6.0764,
+      "step": 20910
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 15.390097618103027,
+      "learning_rate": 2e-05,
+      "loss": 5.9954,
+      "step": 20920
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 15.101361274719238,
+      "learning_rate": 2e-05,
+      "loss": 5.8054,
+      "step": 20930
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 14.882206916809082,
+      "learning_rate": 2e-05,
+      "loss": 5.8599,
+      "step": 20940
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 16.6712703704834,
+      "learning_rate": 2e-05,
+      "loss": 6.0957,
+      "step": 20950
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 16.481653213500977,
+      "learning_rate": 2e-05,
+      "loss": 6.2772,
+      "step": 20960
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 17.251741409301758,
+      "learning_rate": 2e-05,
+      "loss": 5.9835,
+      "step": 20970
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 15.838088989257812,
+      "learning_rate": 2e-05,
+      "loss": 6.1054,
+      "step": 20980
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 14.798274040222168,
+      "learning_rate": 2e-05,
+      "loss": 5.8109,
+      "step": 20990
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 15.026103019714355,
+      "learning_rate": 2e-05,
+      "loss": 6.0928,
+      "step": 21000
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 14.457690238952637,
+      "learning_rate": 2e-05,
+      "loss": 5.9477,
+      "step": 21010
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 16.039663314819336,
+      "learning_rate": 2e-05,
+      "loss": 6.1222,
+      "step": 21020
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 18.6109619140625,
+      "learning_rate": 2e-05,
+      "loss": 6.0227,
+      "step": 21030
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 14.848153114318848,
+      "learning_rate": 2e-05,
+      "loss": 5.8944,
+      "step": 21040
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 16.604995727539062,
+      "learning_rate": 2e-05,
+      "loss": 5.8379,
+      "step": 21050
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 15.922149658203125,
+      "learning_rate": 2e-05,
+      "loss": 6.2513,
+      "step": 21060
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 15.86734390258789,
+      "learning_rate": 2e-05,
+      "loss": 6.0073,
+      "step": 21070
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 15.630167007446289,
+      "learning_rate": 2e-05,
+      "loss": 5.9132,
+      "step": 21080
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 16.436235427856445,
+      "learning_rate": 2e-05,
+      "loss": 6.0263,
+      "step": 21090
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 16.4499568939209,
+      "learning_rate": 2e-05,
+      "loss": 5.9837,
+      "step": 21100
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 15.900408744812012,
+      "learning_rate": 2e-05,
+      "loss": 6.1237,
+      "step": 21110
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 15.801793098449707,
+      "learning_rate": 2e-05,
+      "loss": 6.0165,
+      "step": 21120
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 13.600173950195312,
+      "learning_rate": 2e-05,
+      "loss": 6.2286,
+      "step": 21130
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 15.719366073608398,
+      "learning_rate": 2e-05,
+      "loss": 5.9123,
+      "step": 21140
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 16.68903923034668,
+      "learning_rate": 2e-05,
+      "loss": 6.0155,
+      "step": 21150
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 15.163426399230957,
+      "learning_rate": 2e-05,
+      "loss": 6.1407,
+      "step": 21160
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 19.102336883544922,
+      "learning_rate": 2e-05,
+      "loss": 6.1053,
+      "step": 21170
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 16.753459930419922,
+      "learning_rate": 2e-05,
+      "loss": 5.851,
+      "step": 21180
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 15.009908676147461,
+      "learning_rate": 2e-05,
+      "loss": 5.9845,
+      "step": 21190
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 16.687349319458008,
+      "learning_rate": 2e-05,
+      "loss": 6.0516,
+      "step": 21200
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 15.938105583190918,
+      "learning_rate": 2e-05,
+      "loss": 6.2341,
+      "step": 21210
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 14.490304946899414,
+      "learning_rate": 2e-05,
+      "loss": 6.1826,
+      "step": 21220
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 14.760560989379883,
+      "learning_rate": 2e-05,
+      "loss": 5.7411,
+      "step": 21230
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 14.535091400146484,
+      "learning_rate": 2e-05,
+      "loss": 5.8927,
+      "step": 21240
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 18.148662567138672,
+      "learning_rate": 2e-05,
+      "loss": 6.1126,
+      "step": 21250
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 15.735257148742676,
+      "learning_rate": 2e-05,
+      "loss": 5.7662,
+      "step": 21260
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 15.159509658813477,
+      "learning_rate": 2e-05,
+      "loss": 6.1072,
+      "step": 21270
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 17.040603637695312,
+      "learning_rate": 2e-05,
+      "loss": 5.8416,
+      "step": 21280
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 16.638072967529297,
+      "learning_rate": 2e-05,
+      "loss": 6.0629,
+      "step": 21290
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 15.780477523803711,
+      "learning_rate": 2e-05,
+      "loss": 5.957,
+      "step": 21300
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 15.635530471801758,
+      "learning_rate": 2e-05,
+      "loss": 6.0068,
+      "step": 21310
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 16.039073944091797,
+      "learning_rate": 2e-05,
+      "loss": 5.9505,
+      "step": 21320
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 26.885337829589844,
+      "learning_rate": 2e-05,
+      "loss": 6.1031,
+      "step": 21330
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 16.22350311279297,
+      "learning_rate": 2e-05,
+      "loss": 5.9723,
+      "step": 21340
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 15.848748207092285,
+      "learning_rate": 2e-05,
+      "loss": 6.1534,
+      "step": 21350
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 16.02888298034668,
+      "learning_rate": 2e-05,
+      "loss": 6.0811,
+      "step": 21360
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 16.150224685668945,
+      "learning_rate": 2e-05,
+      "loss": 6.1478,
+      "step": 21370
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 15.481626510620117,
+      "learning_rate": 2e-05,
+      "loss": 5.8647,
+      "step": 21380
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 14.588208198547363,
+      "learning_rate": 2e-05,
+      "loss": 5.9558,
+      "step": 21390
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 14.888575553894043,
+      "learning_rate": 2e-05,
+      "loss": 5.9574,
+      "step": 21400
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 17.654321670532227,
+      "learning_rate": 2e-05,
+      "loss": 6.1994,
+      "step": 21410
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 15.009186744689941,
+      "learning_rate": 2e-05,
+      "loss": 6.1099,
+      "step": 21420
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 15.957594871520996,
+      "learning_rate": 2e-05,
+      "loss": 6.0183,
+      "step": 21430
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 14.947836875915527,
+      "learning_rate": 2e-05,
+      "loss": 6.0352,
+      "step": 21440
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 16.035573959350586,
+      "learning_rate": 2e-05,
+      "loss": 5.9326,
+      "step": 21450
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 15.496720314025879,
+      "learning_rate": 2e-05,
+      "loss": 6.1212,
+      "step": 21460
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 16.30972671508789,
+      "learning_rate": 2e-05,
+      "loss": 6.2501,
+      "step": 21470
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 16.70850372314453,
+      "learning_rate": 2e-05,
+      "loss": 6.1085,
+      "step": 21480
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 16.23195457458496,
+      "learning_rate": 2e-05,
+      "loss": 5.9329,
+      "step": 21490
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 16.572872161865234,
+      "learning_rate": 2e-05,
+      "loss": 5.7873,
+      "step": 21500
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 14.93887996673584,
+      "learning_rate": 2e-05,
+      "loss": 5.8798,
+      "step": 21510
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 15.859906196594238,
+      "learning_rate": 2e-05,
+      "loss": 6.0444,
+      "step": 21520
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 14.99279499053955,
+      "learning_rate": 2e-05,
+      "loss": 5.9812,
+      "step": 21530
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 15.940704345703125,
+      "learning_rate": 2e-05,
+      "loss": 6.2158,
+      "step": 21540
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 16.8408260345459,
+      "learning_rate": 2e-05,
+      "loss": 5.9628,
+      "step": 21550
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 16.98685646057129,
+      "learning_rate": 2e-05,
+      "loss": 6.1116,
+      "step": 21560
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 15.081361770629883,
+      "learning_rate": 2e-05,
+      "loss": 6.0507,
+      "step": 21570
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 15.788143157958984,
+      "learning_rate": 2e-05,
+      "loss": 6.108,
+      "step": 21580
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 14.384382247924805,
+      "learning_rate": 2e-05,
+      "loss": 6.2997,
+      "step": 21590
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 15.993515014648438,
+      "learning_rate": 2e-05,
+      "loss": 6.0685,
+      "step": 21600
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 14.66801929473877,
+      "learning_rate": 2e-05,
+      "loss": 6.3238,
+      "step": 21610
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 18.058162689208984,
+      "learning_rate": 2e-05,
+      "loss": 5.8402,
+      "step": 21620
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 14.088196754455566,
+      "learning_rate": 2e-05,
+      "loss": 5.9288,
+      "step": 21630
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 15.671133041381836,
+      "learning_rate": 2e-05,
+      "loss": 5.9852,
+      "step": 21640
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 17.295377731323242,
+      "learning_rate": 2e-05,
+      "loss": 6.1281,
+      "step": 21650
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 16.14383316040039,
+      "learning_rate": 2e-05,
+      "loss": 6.2045,
+      "step": 21660
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 16.0428524017334,
+      "learning_rate": 2e-05,
+      "loss": 6.0105,
+      "step": 21670
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 14.207866668701172,
+      "learning_rate": 2e-05,
+      "loss": 5.9152,
+      "step": 21680
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 16.214824676513672,
+      "learning_rate": 2e-05,
+      "loss": 5.8885,
+      "step": 21690
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 16.08735466003418,
+      "learning_rate": 2e-05,
+      "loss": 6.0795,
+      "step": 21700
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 15.02086067199707,
+      "learning_rate": 2e-05,
+      "loss": 5.8437,
+      "step": 21710
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 15.730320930480957,
+      "learning_rate": 2e-05,
+      "loss": 6.1402,
+      "step": 21720
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 14.780223846435547,
+      "learning_rate": 2e-05,
+      "loss": 6.2958,
+      "step": 21730
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 17.009601593017578,
+      "learning_rate": 2e-05,
+      "loss": 5.9891,
+      "step": 21740
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 18.053491592407227,
+      "learning_rate": 2e-05,
+      "loss": 5.9891,
+      "step": 21750
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 15.103466033935547,
+      "learning_rate": 2e-05,
+      "loss": 6.0527,
+      "step": 21760
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 15.67296314239502,
+      "learning_rate": 2e-05,
+      "loss": 5.8002,
+      "step": 21770
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 15.969305038452148,
+      "learning_rate": 2e-05,
+      "loss": 5.7225,
+      "step": 21780
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 17.97157859802246,
+      "learning_rate": 2e-05,
+      "loss": 6.107,
+      "step": 21790
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 17.733184814453125,
+      "learning_rate": 2e-05,
+      "loss": 5.8402,
+      "step": 21800
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 15.489290237426758,
+      "learning_rate": 2e-05,
+      "loss": 6.0899,
+      "step": 21810
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 16.42489242553711,
+      "learning_rate": 2e-05,
+      "loss": 5.8971,
+      "step": 21820
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 15.886832237243652,
+      "learning_rate": 2e-05,
+      "loss": 5.6697,
+      "step": 21830
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 17.695240020751953,
+      "learning_rate": 2e-05,
+      "loss": 6.163,
+      "step": 21840
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 16.74485206604004,
+      "learning_rate": 2e-05,
+      "loss": 5.9129,
+      "step": 21850
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 14.630234718322754,
+      "learning_rate": 2e-05,
+      "loss": 6.0831,
+      "step": 21860
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 16.816448211669922,
+      "learning_rate": 2e-05,
+      "loss": 6.0462,
+      "step": 21870
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 16.80683708190918,
+      "learning_rate": 2e-05,
+      "loss": 6.0934,
+      "step": 21880
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 15.185300827026367,
+      "learning_rate": 2e-05,
+      "loss": 6.2703,
+      "step": 21890
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 16.640108108520508,
+      "learning_rate": 2e-05,
+      "loss": 6.0115,
+      "step": 21900
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 17.792709350585938,
+      "learning_rate": 2e-05,
+      "loss": 6.0079,
+      "step": 21910
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 17.515514373779297,
+      "learning_rate": 2e-05,
+      "loss": 6.0301,
+      "step": 21920
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 16.264209747314453,
+      "learning_rate": 2e-05,
+      "loss": 6.1674,
+      "step": 21930
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 15.695699691772461,
+      "learning_rate": 2e-05,
+      "loss": 5.8738,
+      "step": 21940
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 15.15998363494873,
+      "learning_rate": 2e-05,
+      "loss": 5.8846,
+      "step": 21950
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 14.823592185974121,
+      "learning_rate": 2e-05,
+      "loss": 6.1471,
+      "step": 21960
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 16.55012321472168,
+      "learning_rate": 2e-05,
+      "loss": 5.8184,
+      "step": 21970
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 16.383251190185547,
+      "learning_rate": 2e-05,
+      "loss": 5.9675,
+      "step": 21980
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 16.247394561767578,
+      "learning_rate": 2e-05,
+      "loss": 5.9584,
+      "step": 21990
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 18.399621963500977,
+      "learning_rate": 2e-05,
+      "loss": 6.112,
+      "step": 22000
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 15.990718841552734,
+      "learning_rate": 2e-05,
+      "loss": 5.9894,
+      "step": 22010
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 16.615047454833984,
+      "learning_rate": 2e-05,
+      "loss": 5.9724,
+      "step": 22020
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 15.347589492797852,
+      "learning_rate": 2e-05,
+      "loss": 6.1234,
+      "step": 22030
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 15.600197792053223,
+      "learning_rate": 2e-05,
+      "loss": 6.1977,
+      "step": 22040
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 14.967818260192871,
+      "learning_rate": 2e-05,
+      "loss": 5.9468,
+      "step": 22050
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 15.880413055419922,
+      "learning_rate": 2e-05,
+      "loss": 5.7976,
+      "step": 22060
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 15.764298439025879,
+      "learning_rate": 2e-05,
+      "loss": 6.0659,
+      "step": 22070
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 15.594691276550293,
+      "learning_rate": 2e-05,
+      "loss": 5.9625,
+      "step": 22080
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 16.96647071838379,
+      "learning_rate": 2e-05,
+      "loss": 6.1744,
+      "step": 22090
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 15.315774917602539,
+      "learning_rate": 2e-05,
+      "loss": 5.699,
+      "step": 22100
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 16.72232437133789,
+      "learning_rate": 2e-05,
+      "loss": 6.0941,
+      "step": 22110
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 16.62149429321289,
+      "learning_rate": 2e-05,
+      "loss": 5.8558,
+      "step": 22120
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 14.292546272277832,
+      "learning_rate": 2e-05,
+      "loss": 5.6964,
+      "step": 22130
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 14.3430757522583,
+      "learning_rate": 2e-05,
+      "loss": 5.9318,
+      "step": 22140
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 15.596691131591797,
+      "learning_rate": 2e-05,
+      "loss": 6.0782,
+      "step": 22150
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 15.39555549621582,
+      "learning_rate": 2e-05,
+      "loss": 5.8048,
+      "step": 22160
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 16.417001724243164,
+      "learning_rate": 2e-05,
+      "loss": 6.2731,
+      "step": 22170
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 15.738948822021484,
+      "learning_rate": 2e-05,
+      "loss": 5.7573,
+      "step": 22180
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 16.077648162841797,
+      "learning_rate": 2e-05,
+      "loss": 6.0037,
+      "step": 22190
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 16.537689208984375,
+      "learning_rate": 2e-05,
+      "loss": 5.819,
+      "step": 22200
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 16.07887840270996,
+      "learning_rate": 2e-05,
+      "loss": 6.0363,
+      "step": 22210
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 16.73328399658203,
+      "learning_rate": 2e-05,
+      "loss": 6.1343,
+      "step": 22220
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 17.323213577270508,
+      "learning_rate": 2e-05,
+      "loss": 6.041,
+      "step": 22230
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 15.926288604736328,
+      "learning_rate": 2e-05,
+      "loss": 5.9421,
+      "step": 22240
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 15.669966697692871,
+      "learning_rate": 2e-05,
+      "loss": 5.9401,
+      "step": 22250
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 16.048368453979492,
+      "learning_rate": 2e-05,
+      "loss": 5.9065,
+      "step": 22260
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 15.856804847717285,
+      "learning_rate": 2e-05,
+      "loss": 6.0628,
+      "step": 22270
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 16.07671356201172,
+      "learning_rate": 2e-05,
+      "loss": 6.0955,
+      "step": 22280
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 15.093032836914062,
+      "learning_rate": 2e-05,
+      "loss": 6.0667,
+      "step": 22290
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 14.692880630493164,
+      "learning_rate": 2e-05,
+      "loss": 6.0794,
+      "step": 22300
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 17.59325408935547,
+      "learning_rate": 2e-05,
+      "loss": 5.9763,
+      "step": 22310
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 17.39862823486328,
+      "learning_rate": 2e-05,
+      "loss": 5.8622,
+      "step": 22320
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 15.576937675476074,
+      "learning_rate": 2e-05,
+      "loss": 5.9171,
+      "step": 22330
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 15.019445419311523,
+      "learning_rate": 2e-05,
+      "loss": 6.0803,
+      "step": 22340
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 16.190893173217773,
+      "learning_rate": 2e-05,
+      "loss": 6.1766,
+      "step": 22350
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 15.752138137817383,
+      "learning_rate": 2e-05,
+      "loss": 5.7726,
+      "step": 22360
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 17.470699310302734,
+      "learning_rate": 2e-05,
+      "loss": 5.7465,
+      "step": 22370
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 14.1793794631958,
+      "learning_rate": 2e-05,
+      "loss": 5.886,
+      "step": 22380
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 16.836509704589844,
+      "learning_rate": 2e-05,
+      "loss": 5.7444,
+      "step": 22390
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 15.922734260559082,
+      "learning_rate": 2e-05,
+      "loss": 6.1531,
+      "step": 22400
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 14.848880767822266,
+      "learning_rate": 2e-05,
+      "loss": 5.5351,
+      "step": 22410
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 16.19536018371582,
+      "learning_rate": 2e-05,
+      "loss": 5.5992,
+      "step": 22420
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 17.248205184936523,
+      "learning_rate": 2e-05,
+      "loss": 6.2033,
+      "step": 22430
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 14.332897186279297,
+      "learning_rate": 2e-05,
+      "loss": 5.9125,
+      "step": 22440
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 16.063270568847656,
+      "learning_rate": 2e-05,
+      "loss": 5.9799,
+      "step": 22450
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 15.621318817138672,
+      "learning_rate": 2e-05,
+      "loss": 5.7804,
+      "step": 22460
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 16.557645797729492,
+      "learning_rate": 2e-05,
+      "loss": 5.9708,
+      "step": 22470
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 16.485492706298828,
+      "learning_rate": 2e-05,
+      "loss": 5.752,
+      "step": 22480
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 16.69672393798828,
+      "learning_rate": 2e-05,
+      "loss": 6.1009,
+      "step": 22490
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 17.587303161621094,
+      "learning_rate": 2e-05,
+      "loss": 5.7805,
+      "step": 22500
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 16.100189208984375,
+      "learning_rate": 2e-05,
+      "loss": 5.9459,
+      "step": 22510
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 15.66550064086914,
+      "learning_rate": 2e-05,
+      "loss": 6.2067,
+      "step": 22520
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 14.558318138122559,
+      "learning_rate": 2e-05,
+      "loss": 5.852,
+      "step": 22530
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 16.168954849243164,
+      "learning_rate": 2e-05,
+      "loss": 5.719,
+      "step": 22540
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 16.606121063232422,
+      "learning_rate": 2e-05,
+      "loss": 5.865,
+      "step": 22550
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 15.323657989501953,
+      "learning_rate": 2e-05,
+      "loss": 6.1244,
+      "step": 22560
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 16.737001419067383,
+      "learning_rate": 2e-05,
+      "loss": 5.9475,
+      "step": 22570
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 15.0458402633667,
+      "learning_rate": 2e-05,
+      "loss": 5.9156,
+      "step": 22580
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 17.2247257232666,
+      "learning_rate": 2e-05,
+      "loss": 5.7716,
+      "step": 22590
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 17.221843719482422,
+      "learning_rate": 2e-05,
+      "loss": 6.152,
+      "step": 22600
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 16.7856502532959,
+      "learning_rate": 2e-05,
+      "loss": 6.0701,
+      "step": 22610
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 15.579530715942383,
+      "learning_rate": 2e-05,
+      "loss": 6.0961,
+      "step": 22620
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 17.262393951416016,
+      "learning_rate": 2e-05,
+      "loss": 5.8312,
+      "step": 22630
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 14.492802619934082,
+      "learning_rate": 2e-05,
+      "loss": 5.9309,
+      "step": 22640
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 15.708468437194824,
+      "learning_rate": 2e-05,
+      "loss": 6.0649,
+      "step": 22650
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 15.8517484664917,
+      "learning_rate": 2e-05,
+      "loss": 5.9138,
+      "step": 22660
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 17.672718048095703,
+      "learning_rate": 2e-05,
+      "loss": 6.1091,
+      "step": 22670
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 15.508344650268555,
+      "learning_rate": 2e-05,
+      "loss": 5.8604,
+      "step": 22680
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 14.629253387451172,
+      "learning_rate": 2e-05,
+      "loss": 6.0081,
+      "step": 22690
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 15.532172203063965,
+      "learning_rate": 2e-05,
+      "loss": 6.0822,
+      "step": 22700
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 16.356786727905273,
+      "learning_rate": 2e-05,
+      "loss": 6.0923,
+      "step": 22710
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 15.010872840881348,
+      "learning_rate": 2e-05,
+      "loss": 5.759,
+      "step": 22720
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 14.297571182250977,
+      "learning_rate": 2e-05,
+      "loss": 6.3003,
+      "step": 22730
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 15.935193061828613,
+      "learning_rate": 2e-05,
+      "loss": 6.0051,
+      "step": 22740
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 16.249073028564453,
+      "learning_rate": 2e-05,
+      "loss": 5.8173,
+      "step": 22750
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 17.147113800048828,
+      "learning_rate": 2e-05,
+      "loss": 5.721,
+      "step": 22760
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 15.208268165588379,
+      "learning_rate": 2e-05,
+      "loss": 5.9953,
+      "step": 22770
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 14.361098289489746,
+      "learning_rate": 2e-05,
+      "loss": 5.9226,
+      "step": 22780
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 16.279396057128906,
+      "learning_rate": 2e-05,
+      "loss": 5.8862,
+      "step": 22790
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 16.07816505432129,
+      "learning_rate": 2e-05,
+      "loss": 5.9593,
+      "step": 22800
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 16.041234970092773,
+      "learning_rate": 2e-05,
+      "loss": 6.1842,
+      "step": 22810
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 16.178274154663086,
+      "learning_rate": 2e-05,
+      "loss": 5.9468,
+      "step": 22820
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 17.758909225463867,
+      "learning_rate": 2e-05,
+      "loss": 5.8045,
+      "step": 22830
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 17.93916130065918,
+      "learning_rate": 2e-05,
+      "loss": 5.9675,
+      "step": 22840
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 16.234161376953125,
+      "learning_rate": 2e-05,
+      "loss": 5.9683,
+      "step": 22850
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 15.511011123657227,
+      "learning_rate": 2e-05,
+      "loss": 5.8891,
+      "step": 22860
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 16.243730545043945,
+      "learning_rate": 2e-05,
+      "loss": 5.8654,
+      "step": 22870
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 16.842227935791016,
+      "learning_rate": 2e-05,
+      "loss": 6.2377,
+      "step": 22880
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 15.644248962402344,
+      "learning_rate": 2e-05,
+      "loss": 5.8579,
+      "step": 22890
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 16.392683029174805,
+      "learning_rate": 2e-05,
+      "loss": 5.6046,
+      "step": 22900
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 15.755634307861328,
+      "learning_rate": 2e-05,
+      "loss": 6.1784,
+      "step": 22910
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 17.084186553955078,
+      "learning_rate": 2e-05,
+      "loss": 6.0646,
+      "step": 22920
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 17.315298080444336,
+      "learning_rate": 2e-05,
+      "loss": 6.2075,
+      "step": 22930
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 17.303939819335938,
+      "learning_rate": 2e-05,
+      "loss": 6.0571,
+      "step": 22940
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 16.53485107421875,
+      "learning_rate": 2e-05,
+      "loss": 5.9865,
+      "step": 22950
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 15.829534530639648,
+      "learning_rate": 2e-05,
+      "loss": 5.695,
+      "step": 22960
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 16.13177490234375,
+      "learning_rate": 2e-05,
+      "loss": 6.0377,
+      "step": 22970
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 15.766841888427734,
+      "learning_rate": 2e-05,
+      "loss": 5.7622,
+      "step": 22980
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 16.915325164794922,
+      "learning_rate": 2e-05,
+      "loss": 6.0513,
+      "step": 22990
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 16.983915328979492,
+      "learning_rate": 2e-05,
+      "loss": 6.0007,
+      "step": 23000
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 17.271686553955078,
+      "learning_rate": 2e-05,
+      "loss": 5.919,
+      "step": 23010
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 14.48534870147705,
+      "learning_rate": 2e-05,
+      "loss": 6.0966,
+      "step": 23020
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 15.595926284790039,
+      "learning_rate": 2e-05,
+      "loss": 6.0469,
+      "step": 23030
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 17.663854598999023,
+      "learning_rate": 2e-05,
+      "loss": 5.9609,
+      "step": 23040
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 16.339954376220703,
+      "learning_rate": 2e-05,
+      "loss": 5.7824,
+      "step": 23050
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 15.448039054870605,
+      "learning_rate": 2e-05,
+      "loss": 5.9192,
+      "step": 23060
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 15.88376522064209,
+      "learning_rate": 2e-05,
+      "loss": 5.8576,
+      "step": 23070
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 15.140755653381348,
+      "learning_rate": 2e-05,
+      "loss": 6.1675,
+      "step": 23080
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 15.856246948242188,
+      "learning_rate": 2e-05,
+      "loss": 5.9099,
+      "step": 23090
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 16.509510040283203,
+      "learning_rate": 2e-05,
+      "loss": 5.7557,
+      "step": 23100
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 15.390368461608887,
+      "learning_rate": 2e-05,
+      "loss": 6.0001,
+      "step": 23110
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 17.112686157226562,
+      "learning_rate": 2e-05,
+      "loss": 6.2757,
+      "step": 23120
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 16.95307159423828,
+      "learning_rate": 2e-05,
+      "loss": 5.9658,
+      "step": 23130
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 14.35330581665039,
+      "learning_rate": 2e-05,
+      "loss": 5.7188,
+      "step": 23140
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 16.183284759521484,
+      "learning_rate": 2e-05,
+      "loss": 5.8412,
+      "step": 23150
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 18.742292404174805,
+      "learning_rate": 2e-05,
+      "loss": 5.6818,
+      "step": 23160
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 14.994415283203125,
+      "learning_rate": 2e-05,
+      "loss": 5.5986,
+      "step": 23170
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 16.182723999023438,
+      "learning_rate": 2e-05,
+      "loss": 5.9208,
+      "step": 23180
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 15.51449966430664,
+      "learning_rate": 2e-05,
+      "loss": 5.9744,
+      "step": 23190
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 15.008615493774414,
+      "learning_rate": 2e-05,
+      "loss": 5.7276,
+      "step": 23200
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 16.40753173828125,
+      "learning_rate": 2e-05,
+      "loss": 5.8685,
+      "step": 23210
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 17.134878158569336,
+      "learning_rate": 2e-05,
+      "loss": 5.7674,
+      "step": 23220
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 15.119187355041504,
+      "learning_rate": 2e-05,
+      "loss": 5.9073,
+      "step": 23230
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 15.935954093933105,
+      "learning_rate": 2e-05,
+      "loss": 5.9868,
+      "step": 23240
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 15.12692642211914,
+      "learning_rate": 2e-05,
+      "loss": 5.6547,
+      "step": 23250
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 17.21491813659668,
+      "learning_rate": 2e-05,
+      "loss": 5.9771,
+      "step": 23260
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 16.18280029296875,
+      "learning_rate": 2e-05,
+      "loss": 5.7742,
+      "step": 23270
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 16.87044334411621,
+      "learning_rate": 2e-05,
+      "loss": 5.6443,
+      "step": 23280
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 16.92005729675293,
+      "learning_rate": 2e-05,
+      "loss": 6.0617,
+      "step": 23290
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 16.993654251098633,
+      "learning_rate": 2e-05,
+      "loss": 5.7793,
+      "step": 23300
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 16.391769409179688,
+      "learning_rate": 2e-05,
+      "loss": 5.932,
+      "step": 23310
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 15.346217155456543,
+      "learning_rate": 2e-05,
+      "loss": 5.7483,
+      "step": 23320
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 17.41927719116211,
+      "learning_rate": 2e-05,
+      "loss": 6.0447,
+      "step": 23330
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 17.53887939453125,
+      "learning_rate": 2e-05,
+      "loss": 6.0553,
+      "step": 23340
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 17.590618133544922,
+      "learning_rate": 2e-05,
+      "loss": 5.8831,
+      "step": 23350
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 16.191547393798828,
+      "learning_rate": 2e-05,
+      "loss": 5.6237,
+      "step": 23360
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 16.90030288696289,
+      "learning_rate": 2e-05,
+      "loss": 5.9013,
+      "step": 23370
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 16.85875129699707,
+      "learning_rate": 2e-05,
+      "loss": 5.932,
+      "step": 23380
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 16.552988052368164,
+      "learning_rate": 2e-05,
+      "loss": 5.8929,
+      "step": 23390
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 15.895064353942871,
+      "learning_rate": 2e-05,
+      "loss": 5.9536,
+      "step": 23400
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 14.447576522827148,
+      "learning_rate": 2e-05,
+      "loss": 5.8057,
+      "step": 23410
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 14.692564010620117,
+      "learning_rate": 2e-05,
+      "loss": 5.8067,
+      "step": 23420
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 16.23661994934082,
+      "learning_rate": 2e-05,
+      "loss": 5.5999,
+      "step": 23430
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 16.46430778503418,
+      "learning_rate": 2e-05,
+      "loss": 5.9485,
+      "step": 23440
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 15.251350402832031,
+      "learning_rate": 2e-05,
+      "loss": 5.975,
+      "step": 23450
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 16.783340454101562,
+      "learning_rate": 2e-05,
+      "loss": 5.9614,
+      "step": 23460
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 14.977662086486816,
+      "learning_rate": 2e-05,
+      "loss": 5.8499,
+      "step": 23470
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 14.507057189941406,
+      "learning_rate": 2e-05,
+      "loss": 5.9081,
+      "step": 23480
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 16.259113311767578,
+      "learning_rate": 2e-05,
+      "loss": 5.6951,
+      "step": 23490
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 15.51435661315918,
+      "learning_rate": 2e-05,
+      "loss": 6.1005,
+      "step": 23500
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 16.439939498901367,
+      "learning_rate": 2e-05,
+      "loss": 6.1307,
+      "step": 23510
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 15.476346015930176,
+      "learning_rate": 2e-05,
+      "loss": 5.6232,
+      "step": 23520
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 16.978178024291992,
+      "learning_rate": 2e-05,
+      "loss": 5.63,
+      "step": 23530
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 14.646075248718262,
+      "learning_rate": 2e-05,
+      "loss": 5.8968,
+      "step": 23540
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 17.15751075744629,
+      "learning_rate": 2e-05,
+      "loss": 5.357,
+      "step": 23550
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 16.20404052734375,
+      "learning_rate": 2e-05,
+      "loss": 5.7273,
+      "step": 23560
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 15.56240463256836,
+      "learning_rate": 2e-05,
+      "loss": 5.9862,
+      "step": 23570
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 17.24268913269043,
+      "learning_rate": 2e-05,
+      "loss": 5.731,
+      "step": 23580
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 14.058793067932129,
+      "learning_rate": 2e-05,
+      "loss": 5.827,
+      "step": 23590
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 15.944547653198242,
+      "learning_rate": 2e-05,
+      "loss": 6.0925,
+      "step": 23600
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 15.318071365356445,
+      "learning_rate": 2e-05,
+      "loss": 5.9442,
+      "step": 23610
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 16.43045425415039,
+      "learning_rate": 2e-05,
+      "loss": 5.8568,
+      "step": 23620
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 16.721046447753906,
+      "learning_rate": 2e-05,
+      "loss": 5.5675,
+      "step": 23630
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 15.569696426391602,
+      "learning_rate": 2e-05,
+      "loss": 5.806,
+      "step": 23640
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 14.664253234863281,
+      "learning_rate": 2e-05,
+      "loss": 5.7178,
+      "step": 23650
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 17.399791717529297,
+      "learning_rate": 2e-05,
+      "loss": 6.2514,
+      "step": 23660
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 15.765493392944336,
+      "learning_rate": 2e-05,
+      "loss": 5.9119,
+      "step": 23670
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 15.43872356414795,
+      "learning_rate": 2e-05,
+      "loss": 6.0214,
+      "step": 23680
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 15.837587356567383,
+      "learning_rate": 2e-05,
+      "loss": 6.066,
+      "step": 23690
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 15.538782119750977,
+      "learning_rate": 2e-05,
+      "loss": 5.8151,
+      "step": 23700
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 16.423364639282227,
+      "learning_rate": 2e-05,
+      "loss": 5.9998,
+      "step": 23710
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 15.99908447265625,
+      "learning_rate": 2e-05,
+      "loss": 5.9045,
+      "step": 23720
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 15.32610034942627,
+      "learning_rate": 2e-05,
+      "loss": 5.8894,
+      "step": 23730
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 14.51338005065918,
+      "learning_rate": 2e-05,
+      "loss": 5.9032,
+      "step": 23740
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 16.486581802368164,
+      "learning_rate": 2e-05,
+      "loss": 5.9467,
+      "step": 23750
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 15.58364486694336,
+      "learning_rate": 2e-05,
+      "loss": 5.9635,
+      "step": 23760
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 15.3807373046875,
+      "learning_rate": 2e-05,
+      "loss": 5.6249,
+      "step": 23770
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 15.12619400024414,
+      "learning_rate": 2e-05,
+      "loss": 5.9139,
+      "step": 23780
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 16.20196533203125,
+      "learning_rate": 2e-05,
+      "loss": 5.8173,
+      "step": 23790
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 17.645219802856445,
+      "learning_rate": 2e-05,
+      "loss": 5.8701,
+      "step": 23800
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 15.970287322998047,
+      "learning_rate": 2e-05,
+      "loss": 5.8831,
+      "step": 23810
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 16.419965744018555,
+      "learning_rate": 2e-05,
+      "loss": 5.624,
+      "step": 23820
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 17.4334659576416,
+      "learning_rate": 2e-05,
+      "loss": 5.8153,
+      "step": 23830
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 16.434776306152344,
+      "learning_rate": 2e-05,
+      "loss": 5.7391,
+      "step": 23840
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 15.775887489318848,
+      "learning_rate": 2e-05,
+      "loss": 5.7899,
+      "step": 23850
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 20.391220092773438,
+      "learning_rate": 2e-05,
+      "loss": 5.8494,
+      "step": 23860
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 24.018335342407227,
+      "learning_rate": 2e-05,
+      "loss": 5.7279,
+      "step": 23870
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 16.94266700744629,
+      "learning_rate": 2e-05,
+      "loss": 6.0297,
+      "step": 23880
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 16.202308654785156,
+      "learning_rate": 2e-05,
+      "loss": 5.7897,
+      "step": 23890
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 16.283056259155273,
+      "learning_rate": 2e-05,
+      "loss": 5.8746,
+      "step": 23900
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 14.959268569946289,
+      "learning_rate": 2e-05,
+      "loss": 6.0964,
+      "step": 23910
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 14.592902183532715,
+      "learning_rate": 2e-05,
+      "loss": 5.6875,
+      "step": 23920
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 16.526203155517578,
+      "learning_rate": 2e-05,
+      "loss": 5.8071,
+      "step": 23930
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 17.779054641723633,
+      "learning_rate": 2e-05,
+      "loss": 5.7784,
+      "step": 23940
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 15.865787506103516,
+      "learning_rate": 2e-05,
+      "loss": 5.818,
+      "step": 23950
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 16.83468246459961,
+      "learning_rate": 2e-05,
+      "loss": 5.8067,
+      "step": 23960
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 15.343466758728027,
+      "learning_rate": 2e-05,
+      "loss": 5.9262,
+      "step": 23970
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 15.965609550476074,
+      "learning_rate": 2e-05,
+      "loss": 5.8257,
+      "step": 23980
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 15.557772636413574,
+      "learning_rate": 2e-05,
+      "loss": 6.0448,
+      "step": 23990
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 17.555505752563477,
+      "learning_rate": 2e-05,
+      "loss": 6.0692,
+      "step": 24000
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 16.236831665039062,
+      "learning_rate": 2e-05,
+      "loss": 6.0557,
+      "step": 24010
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 14.768006324768066,
+      "learning_rate": 2e-05,
+      "loss": 5.9068,
+      "step": 24020
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 17.6851749420166,
+      "learning_rate": 2e-05,
+      "loss": 5.6998,
+      "step": 24030
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 14.881142616271973,
+      "learning_rate": 2e-05,
+      "loss": 5.9102,
+      "step": 24040
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 16.097015380859375,
+      "learning_rate": 2e-05,
+      "loss": 5.9735,
+      "step": 24050
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 17.531219482421875,
+      "learning_rate": 2e-05,
+      "loss": 5.9219,
+      "step": 24060
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 14.363456726074219,
+      "learning_rate": 2e-05,
+      "loss": 5.9042,
+      "step": 24070
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 14.756726264953613,
+      "learning_rate": 2e-05,
+      "loss": 5.9292,
+      "step": 24080
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 14.762608528137207,
+      "learning_rate": 2e-05,
+      "loss": 6.0699,
+      "step": 24090
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 17.657384872436523,
+      "learning_rate": 2e-05,
+      "loss": 5.8923,
+      "step": 24100
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 15.775351524353027,
+      "learning_rate": 2e-05,
+      "loss": 5.6385,
+      "step": 24110
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 13.486385345458984,
+      "learning_rate": 2e-05,
+      "loss": 5.9084,
+      "step": 24120
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 15.180458068847656,
+      "learning_rate": 2e-05,
+      "loss": 5.7424,
+      "step": 24130
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 18.465007781982422,
+      "learning_rate": 2e-05,
+      "loss": 6.1277,
+      "step": 24140
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 16.17591667175293,
+      "learning_rate": 2e-05,
+      "loss": 5.7753,
+      "step": 24150
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 17.3056583404541,
+      "learning_rate": 2e-05,
+      "loss": 5.9058,
+      "step": 24160
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 15.726470947265625,
+      "learning_rate": 2e-05,
+      "loss": 5.522,
+      "step": 24170
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 14.906989097595215,
+      "learning_rate": 2e-05,
+      "loss": 5.8218,
+      "step": 24180
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 15.7610502243042,
+      "learning_rate": 2e-05,
+      "loss": 5.6704,
+      "step": 24190
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 15.817811012268066,
+      "learning_rate": 2e-05,
+      "loss": 5.8127,
+      "step": 24200
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 15.941366195678711,
+      "learning_rate": 2e-05,
+      "loss": 5.5995,
+      "step": 24210
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 17.324338912963867,
+      "learning_rate": 2e-05,
+      "loss": 5.9312,
+      "step": 24220
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 17.62506866455078,
+      "learning_rate": 2e-05,
+      "loss": 5.9359,
+      "step": 24230
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 16.616737365722656,
+      "learning_rate": 2e-05,
+      "loss": 5.9077,
+      "step": 24240
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 15.162822723388672,
+      "learning_rate": 2e-05,
+      "loss": 5.772,
+      "step": 24250
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 16.619131088256836,
+      "learning_rate": 2e-05,
+      "loss": 6.032,
+      "step": 24260
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 17.362077713012695,
+      "learning_rate": 2e-05,
+      "loss": 5.9027,
+      "step": 24270
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 16.118974685668945,
+      "learning_rate": 2e-05,
+      "loss": 6.0995,
+      "step": 24280
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 18.738792419433594,
+      "learning_rate": 2e-05,
+      "loss": 5.9475,
+      "step": 24290
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 18.96558952331543,
+      "learning_rate": 2e-05,
+      "loss": 5.9512,
+      "step": 24300
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 16.650415420532227,
+      "learning_rate": 2e-05,
+      "loss": 5.6779,
+      "step": 24310
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 14.423113822937012,
+      "learning_rate": 2e-05,
+      "loss": 5.5939,
+      "step": 24320
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 17.39287757873535,
+      "learning_rate": 2e-05,
+      "loss": 5.8156,
+      "step": 24330
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 15.209733009338379,
+      "learning_rate": 2e-05,
+      "loss": 5.853,
+      "step": 24340
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 15.405449867248535,
+      "learning_rate": 2e-05,
+      "loss": 5.9643,
+      "step": 24350
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 16.8925724029541,
+      "learning_rate": 2e-05,
+      "loss": 6.0209,
+      "step": 24360
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 14.916582107543945,
+      "learning_rate": 2e-05,
+      "loss": 5.7797,
+      "step": 24370
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 17.24296760559082,
+      "learning_rate": 2e-05,
+      "loss": 5.7331,
+      "step": 24380
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 18.335350036621094,
+      "learning_rate": 2e-05,
+      "loss": 5.6478,
+      "step": 24390
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 15.678045272827148,
+      "learning_rate": 2e-05,
+      "loss": 5.7107,
+      "step": 24400
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 16.617494583129883,
+      "learning_rate": 2e-05,
+      "loss": 5.8429,
+      "step": 24410
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 18.74945831298828,
+      "learning_rate": 2e-05,
+      "loss": 5.9894,
+      "step": 24420
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 14.843206405639648,
+      "learning_rate": 2e-05,
+      "loss": 5.9832,
+      "step": 24430
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 17.222200393676758,
+      "learning_rate": 2e-05,
+      "loss": 5.7858,
+      "step": 24440
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 14.908010482788086,
+      "learning_rate": 2e-05,
+      "loss": 5.7714,
+      "step": 24450
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 16.42299461364746,
+      "learning_rate": 2e-05,
+      "loss": 5.7583,
+      "step": 24460
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 15.423325538635254,
+      "learning_rate": 2e-05,
+      "loss": 5.831,
+      "step": 24470
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 14.972670555114746,
+      "learning_rate": 2e-05,
+      "loss": 5.6998,
+      "step": 24480
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 15.199299812316895,
+      "learning_rate": 2e-05,
+      "loss": 5.8869,
+      "step": 24490
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 15.697855949401855,
+      "learning_rate": 2e-05,
+      "loss": 5.809,
+      "step": 24500
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 14.672245025634766,
+      "learning_rate": 2e-05,
+      "loss": 5.9572,
+      "step": 24510
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 16.380538940429688,
+      "learning_rate": 2e-05,
+      "loss": 5.7588,
+      "step": 24520
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 16.63641929626465,
+      "learning_rate": 2e-05,
+      "loss": 5.9947,
+      "step": 24530
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 16.542823791503906,
+      "learning_rate": 2e-05,
+      "loss": 5.7472,
+      "step": 24540
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 15.72082233428955,
+      "learning_rate": 2e-05,
+      "loss": 5.9061,
+      "step": 24550
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 15.476910591125488,
+      "learning_rate": 2e-05,
+      "loss": 5.7087,
+      "step": 24560
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 16.510122299194336,
+      "learning_rate": 2e-05,
+      "loss": 5.6726,
+      "step": 24570
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 16.81976318359375,
+      "learning_rate": 2e-05,
+      "loss": 5.8398,
+      "step": 24580
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 18.274322509765625,
+      "learning_rate": 2e-05,
+      "loss": 5.891,
+      "step": 24590
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 14.313516616821289,
+      "learning_rate": 2e-05,
+      "loss": 5.8543,
+      "step": 24600
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 14.905831336975098,
+      "learning_rate": 2e-05,
+      "loss": 5.801,
+      "step": 24610
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 17.23286247253418,
+      "learning_rate": 2e-05,
+      "loss": 5.7486,
+      "step": 24620
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 15.083555221557617,
+      "learning_rate": 2e-05,
+      "loss": 5.8804,
+      "step": 24630
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 18.018537521362305,
+      "learning_rate": 2e-05,
+      "loss": 5.7986,
+      "step": 24640
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 16.256303787231445,
+      "learning_rate": 2e-05,
+      "loss": 5.6684,
+      "step": 24650
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 16.113426208496094,
+      "learning_rate": 2e-05,
+      "loss": 5.629,
+      "step": 24660
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 16.2322940826416,
+      "learning_rate": 2e-05,
+      "loss": 5.9883,
+      "step": 24670
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 16.988157272338867,
+      "learning_rate": 2e-05,
+      "loss": 5.4553,
+      "step": 24680
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 15.921769142150879,
+      "learning_rate": 2e-05,
+      "loss": 5.8703,
+      "step": 24690
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 14.416786193847656,
+      "learning_rate": 2e-05,
+      "loss": 5.5931,
+      "step": 24700
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 15.97974967956543,
+      "learning_rate": 2e-05,
+      "loss": 5.9207,
+      "step": 24710
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 15.915593147277832,
+      "learning_rate": 2e-05,
+      "loss": 5.846,
+      "step": 24720
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 17.017038345336914,
+      "learning_rate": 2e-05,
+      "loss": 5.7417,
+      "step": 24730
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 15.684623718261719,
+      "learning_rate": 2e-05,
+      "loss": 5.7882,
+      "step": 24740
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 14.164400100708008,
+      "learning_rate": 2e-05,
+      "loss": 5.9196,
+      "step": 24750
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 16.802135467529297,
+      "learning_rate": 2e-05,
+      "loss": 5.7521,
+      "step": 24760
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 17.3746280670166,
+      "learning_rate": 2e-05,
+      "loss": 5.5641,
+      "step": 24770
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 16.448354721069336,
+      "learning_rate": 2e-05,
+      "loss": 5.7704,
+      "step": 24780
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 16.59912872314453,
+      "learning_rate": 2e-05,
+      "loss": 6.0257,
+      "step": 24790
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 15.816500663757324,
+      "learning_rate": 2e-05,
+      "loss": 5.8678,
+      "step": 24800
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 15.087115287780762,
+      "learning_rate": 2e-05,
+      "loss": 5.8259,
+      "step": 24810
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 16.201330184936523,
+      "learning_rate": 2e-05,
+      "loss": 5.8483,
+      "step": 24820
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 16.512357711791992,
+      "learning_rate": 2e-05,
+      "loss": 5.7748,
+      "step": 24830
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 17.906585693359375,
+      "learning_rate": 2e-05,
+      "loss": 5.6658,
+      "step": 24840
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 14.537487030029297,
+      "learning_rate": 2e-05,
+      "loss": 5.7394,
+      "step": 24850
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 16.1270809173584,
+      "learning_rate": 2e-05,
+      "loss": 5.8789,
+      "step": 24860
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 14.299910545349121,
+      "learning_rate": 2e-05,
+      "loss": 5.561,
+      "step": 24870
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 16.740692138671875,
+      "learning_rate": 2e-05,
+      "loss": 5.8098,
+      "step": 24880
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 16.311614990234375,
+      "learning_rate": 2e-05,
+      "loss": 5.8084,
+      "step": 24890
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 17.536638259887695,
+      "learning_rate": 2e-05,
+      "loss": 5.9496,
+      "step": 24900
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 14.662753105163574,
+      "learning_rate": 2e-05,
+      "loss": 5.7367,
+      "step": 24910
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 15.499448776245117,
+      "learning_rate": 2e-05,
+      "loss": 6.0189,
+      "step": 24920
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 14.599725723266602,
+      "learning_rate": 2e-05,
+      "loss": 5.5743,
+      "step": 24930
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 15.419778823852539,
+      "learning_rate": 2e-05,
+      "loss": 5.9004,
+      "step": 24940
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 16.654111862182617,
+      "learning_rate": 2e-05,
+      "loss": 5.7276,
+      "step": 24950
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 16.40028190612793,
+      "learning_rate": 2e-05,
+      "loss": 5.8952,
+      "step": 24960
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 15.274345397949219,
+      "learning_rate": 2e-05,
+      "loss": 5.8255,
+      "step": 24970
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 15.068805694580078,
+      "learning_rate": 2e-05,
+      "loss": 6.0149,
+      "step": 24980
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 17.160022735595703,
+      "learning_rate": 2e-05,
+      "loss": 5.6044,
+      "step": 24990
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 15.511263847351074,
+      "learning_rate": 2e-05,
+      "loss": 5.6182,
+      "step": 25000
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 16.46701431274414,
+      "learning_rate": 2e-05,
+      "loss": 5.7943,
+      "step": 25010
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 15.69792366027832,
+      "learning_rate": 2e-05,
+      "loss": 5.5312,
+      "step": 25020
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 15.721455574035645,
+      "learning_rate": 2e-05,
+      "loss": 5.7401,
+      "step": 25030
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 15.705838203430176,
+      "learning_rate": 2e-05,
+      "loss": 5.7016,
+      "step": 25040
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 16.106401443481445,
+      "learning_rate": 2e-05,
+      "loss": 5.8613,
+      "step": 25050
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 16.677732467651367,
+      "learning_rate": 2e-05,
+      "loss": 5.6798,
+      "step": 25060
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 15.733518600463867,
+      "learning_rate": 2e-05,
+      "loss": 5.8406,
+      "step": 25070
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 15.068275451660156,
+      "learning_rate": 2e-05,
+      "loss": 5.5952,
+      "step": 25080
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 15.713346481323242,
+      "learning_rate": 2e-05,
+      "loss": 5.7377,
+      "step": 25090
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 14.945537567138672,
+      "learning_rate": 2e-05,
+      "loss": 5.6655,
+      "step": 25100
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 16.120216369628906,
+      "learning_rate": 2e-05,
+      "loss": 5.765,
+      "step": 25110
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 15.900423049926758,
+      "learning_rate": 2e-05,
+      "loss": 5.6838,
+      "step": 25120
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 17.00295639038086,
+      "learning_rate": 2e-05,
+      "loss": 5.7784,
+      "step": 25130
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 15.897491455078125,
+      "learning_rate": 2e-05,
+      "loss": 5.7721,
+      "step": 25140
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 17.89361000061035,
+      "learning_rate": 2e-05,
+      "loss": 5.7274,
+      "step": 25150
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 16.236894607543945,
+      "learning_rate": 2e-05,
+      "loss": 5.8382,
+      "step": 25160
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 15.558456420898438,
+      "learning_rate": 2e-05,
+      "loss": 5.4951,
+      "step": 25170
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 15.969861030578613,
+      "learning_rate": 2e-05,
+      "loss": 5.5119,
+      "step": 25180
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 15.978837966918945,
+      "learning_rate": 2e-05,
+      "loss": 5.5385,
+      "step": 25190
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 14.669622421264648,
+      "learning_rate": 2e-05,
+      "loss": 5.6154,
+      "step": 25200
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 14.956947326660156,
+      "learning_rate": 2e-05,
+      "loss": 5.7309,
+      "step": 25210
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 15.7445707321167,
+      "learning_rate": 2e-05,
+      "loss": 5.7821,
+      "step": 25220
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 15.522754669189453,
+      "learning_rate": 2e-05,
+      "loss": 5.7851,
+      "step": 25230
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 17.24550437927246,
+      "learning_rate": 2e-05,
+      "loss": 5.721,
+      "step": 25240
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 15.240259170532227,
+      "learning_rate": 2e-05,
+      "loss": 5.8187,
+      "step": 25250
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 16.006881713867188,
+      "learning_rate": 2e-05,
+      "loss": 5.8837,
+      "step": 25260
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 15.05899715423584,
+      "learning_rate": 2e-05,
+      "loss": 5.9326,
+      "step": 25270
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 15.865565299987793,
+      "learning_rate": 2e-05,
+      "loss": 6.0281,
+      "step": 25280
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 15.843626022338867,
+      "learning_rate": 2e-05,
+      "loss": 5.733,
+      "step": 25290
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 17.049072265625,
+      "learning_rate": 2e-05,
+      "loss": 6.0472,
+      "step": 25300
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 15.333700180053711,
+      "learning_rate": 2e-05,
+      "loss": 5.7053,
+      "step": 25310
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 16.99417495727539,
+      "learning_rate": 2e-05,
+      "loss": 5.9128,
+      "step": 25320
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 16.667007446289062,
+      "learning_rate": 2e-05,
+      "loss": 5.7651,
+      "step": 25330
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 14.765226364135742,
+      "learning_rate": 2e-05,
+      "loss": 5.8461,
+      "step": 25340
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 15.42281723022461,
+      "learning_rate": 2e-05,
+      "loss": 6.0357,
+      "step": 25350
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 16.676807403564453,
+      "learning_rate": 2e-05,
+      "loss": 5.6485,
+      "step": 25360
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 17.126523971557617,
+      "learning_rate": 2e-05,
+      "loss": 5.8768,
+      "step": 25370
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 15.461708068847656,
+      "learning_rate": 2e-05,
+      "loss": 5.8054,
+      "step": 25380
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 14.587892532348633,
+      "learning_rate": 2e-05,
+      "loss": 5.6208,
+      "step": 25390
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 17.27574920654297,
+      "learning_rate": 2e-05,
+      "loss": 5.74,
+      "step": 25400
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 15.643343925476074,
+      "learning_rate": 2e-05,
+      "loss": 5.5169,
+      "step": 25410
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 17.594369888305664,
+      "learning_rate": 2e-05,
+      "loss": 5.7314,
+      "step": 25420
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 16.52899932861328,
+      "learning_rate": 2e-05,
+      "loss": 5.8402,
+      "step": 25430
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 16.300851821899414,
+      "learning_rate": 2e-05,
+      "loss": 5.8038,
+      "step": 25440
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 18.514503479003906,
+      "learning_rate": 2e-05,
+      "loss": 5.8622,
+      "step": 25450
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 16.273151397705078,
+      "learning_rate": 2e-05,
+      "loss": 5.7991,
+      "step": 25460
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 17.4202938079834,
+      "learning_rate": 2e-05,
+      "loss": 5.3434,
+      "step": 25470
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 17.14916229248047,
+      "learning_rate": 2e-05,
+      "loss": 5.666,
+      "step": 25480
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 16.169645309448242,
+      "learning_rate": 2e-05,
+      "loss": 5.7823,
+      "step": 25490
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 16.323467254638672,
+      "learning_rate": 2e-05,
+      "loss": 5.5865,
+      "step": 25500
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 16.21868324279785,
+      "learning_rate": 2e-05,
+      "loss": 5.8327,
+      "step": 25510
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 15.727302551269531,
+      "learning_rate": 2e-05,
+      "loss": 5.9197,
+      "step": 25520
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 17.286617279052734,
+      "learning_rate": 2e-05,
+      "loss": 5.7687,
+      "step": 25530
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 16.882314682006836,
+      "learning_rate": 2e-05,
+      "loss": 5.6891,
+      "step": 25540
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 14.606783866882324,
+      "learning_rate": 2e-05,
+      "loss": 5.6859,
+      "step": 25550
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 15.758453369140625,
+      "learning_rate": 2e-05,
+      "loss": 5.5914,
+      "step": 25560
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 17.612075805664062,
+      "learning_rate": 2e-05,
+      "loss": 5.9641,
+      "step": 25570
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 14.464558601379395,
+      "learning_rate": 2e-05,
+      "loss": 5.9091,
+      "step": 25580
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 15.749378204345703,
+      "learning_rate": 2e-05,
+      "loss": 5.8327,
+      "step": 25590
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 16.643171310424805,
+      "learning_rate": 2e-05,
+      "loss": 5.759,
+      "step": 25600
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 16.390592575073242,
+      "learning_rate": 2e-05,
+      "loss": 5.8857,
+      "step": 25610
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 14.712851524353027,
+      "learning_rate": 2e-05,
+      "loss": 5.8663,
+      "step": 25620
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 15.966470718383789,
+      "learning_rate": 2e-05,
+      "loss": 5.5477,
+      "step": 25630
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 15.399444580078125,
+      "learning_rate": 2e-05,
+      "loss": 5.8039,
+      "step": 25640
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 16.31035804748535,
+      "learning_rate": 2e-05,
+      "loss": 5.8022,
+      "step": 25650
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 16.163101196289062,
+      "learning_rate": 2e-05,
+      "loss": 5.5927,
+      "step": 25660
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 15.400129318237305,
+      "learning_rate": 2e-05,
+      "loss": 5.692,
+      "step": 25670
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 15.726388931274414,
+      "learning_rate": 2e-05,
+      "loss": 5.7215,
+      "step": 25680
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 17.966033935546875,
+      "learning_rate": 2e-05,
+      "loss": 5.4991,
+      "step": 25690
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 15.896224021911621,
+      "learning_rate": 2e-05,
+      "loss": 5.5037,
+      "step": 25700
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 14.737597465515137,
+      "learning_rate": 2e-05,
+      "loss": 5.9165,
+      "step": 25710
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 14.52827262878418,
+      "learning_rate": 2e-05,
+      "loss": 5.5849,
+      "step": 25720
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 16.82767105102539,
+      "learning_rate": 2e-05,
+      "loss": 5.8835,
+      "step": 25730
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 13.74605655670166,
+      "learning_rate": 2e-05,
+      "loss": 5.4492,
+      "step": 25740
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 16.880386352539062,
+      "learning_rate": 2e-05,
+      "loss": 5.7009,
+      "step": 25750
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 15.247879028320312,
+      "learning_rate": 2e-05,
+      "loss": 5.8442,
+      "step": 25760
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 15.468958854675293,
+      "learning_rate": 2e-05,
+      "loss": 5.8029,
+      "step": 25770
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 15.573247909545898,
+      "learning_rate": 2e-05,
+      "loss": 5.7202,
+      "step": 25780
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 15.274388313293457,
+      "learning_rate": 2e-05,
+      "loss": 5.5101,
+      "step": 25790
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 15.769818305969238,
+      "learning_rate": 2e-05,
+      "loss": 5.7384,
+      "step": 25800
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 15.36095905303955,
+      "learning_rate": 2e-05,
+      "loss": 5.6266,
+      "step": 25810
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 14.464593887329102,
+      "learning_rate": 2e-05,
+      "loss": 5.5258,
+      "step": 25820
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 15.717238426208496,
+      "learning_rate": 2e-05,
+      "loss": 5.7294,
+      "step": 25830
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 14.975890159606934,
+      "learning_rate": 2e-05,
+      "loss": 5.5959,
+      "step": 25840
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 14.79320240020752,
+      "learning_rate": 2e-05,
+      "loss": 5.8387,
+      "step": 25850
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 15.511950492858887,
+      "learning_rate": 2e-05,
+      "loss": 5.8039,
+      "step": 25860
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 16.623260498046875,
+      "learning_rate": 2e-05,
+      "loss": 5.6871,
+      "step": 25870
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 15.790567398071289,
+      "learning_rate": 2e-05,
+      "loss": 5.603,
+      "step": 25880
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 15.4313383102417,
+      "learning_rate": 2e-05,
+      "loss": 5.5307,
+      "step": 25890
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 15.267356872558594,
+      "learning_rate": 2e-05,
+      "loss": 5.7875,
+      "step": 25900
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 14.71072769165039,
+      "learning_rate": 2e-05,
+      "loss": 5.6275,
+      "step": 25910
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 15.879714965820312,
+      "learning_rate": 2e-05,
+      "loss": 5.7999,
+      "step": 25920
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 15.001500129699707,
+      "learning_rate": 2e-05,
+      "loss": 5.3483,
+      "step": 25930
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 16.124126434326172,
+      "learning_rate": 2e-05,
+      "loss": 5.6552,
+      "step": 25940
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 17.20475959777832,
+      "learning_rate": 2e-05,
+      "loss": 5.8361,
+      "step": 25950
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 15.981983184814453,
+      "learning_rate": 2e-05,
+      "loss": 5.7011,
+      "step": 25960
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 15.476761817932129,
+      "learning_rate": 2e-05,
+      "loss": 5.6807,
+      "step": 25970
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 17.380216598510742,
+      "learning_rate": 2e-05,
+      "loss": 5.6798,
+      "step": 25980
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 16.254600524902344,
+      "learning_rate": 2e-05,
+      "loss": 5.5096,
+      "step": 25990
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 15.652602195739746,
+      "learning_rate": 2e-05,
+      "loss": 5.7336,
+      "step": 26000
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 15.32880973815918,
+      "learning_rate": 2e-05,
+      "loss": 5.7407,
+      "step": 26010
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 16.319358825683594,
+      "learning_rate": 2e-05,
+      "loss": 5.9859,
+      "step": 26020
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 16.007287979125977,
+      "learning_rate": 2e-05,
+      "loss": 5.5027,
+      "step": 26030
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 15.803464889526367,
+      "learning_rate": 2e-05,
+      "loss": 5.7623,
+      "step": 26040
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 15.572185516357422,
+      "learning_rate": 2e-05,
+      "loss": 5.5515,
+      "step": 26050
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 14.01984977722168,
+      "learning_rate": 2e-05,
+      "loss": 5.8891,
+      "step": 26060
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 16.125944137573242,
+      "learning_rate": 2e-05,
+      "loss": 5.7782,
+      "step": 26070
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 16.69143295288086,
+      "learning_rate": 2e-05,
+      "loss": 5.9178,
+      "step": 26080
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 15.138254165649414,
+      "learning_rate": 2e-05,
+      "loss": 5.526,
+      "step": 26090
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 14.738810539245605,
+      "learning_rate": 2e-05,
+      "loss": 5.4695,
+      "step": 26100
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 19.75293731689453,
+      "learning_rate": 2e-05,
+      "loss": 6.0241,
+      "step": 26110
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 14.781432151794434,
+      "learning_rate": 2e-05,
+      "loss": 5.7495,
+      "step": 26120
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 16.098356246948242,
+      "learning_rate": 2e-05,
+      "loss": 5.8073,
+      "step": 26130
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 16.20099639892578,
+      "learning_rate": 2e-05,
+      "loss": 5.72,
+      "step": 26140
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 14.798359870910645,
+      "learning_rate": 2e-05,
+      "loss": 5.4974,
+      "step": 26150
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 15.930750846862793,
+      "learning_rate": 2e-05,
+      "loss": 5.7292,
+      "step": 26160
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 16.80059242248535,
+      "learning_rate": 2e-05,
+      "loss": 5.6918,
+      "step": 26170
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 17.32004737854004,
+      "learning_rate": 2e-05,
+      "loss": 5.6914,
+      "step": 26180
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 16.07136344909668,
+      "learning_rate": 2e-05,
+      "loss": 5.7464,
+      "step": 26190
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 16.196937561035156,
+      "learning_rate": 2e-05,
+      "loss": 5.9597,
+      "step": 26200
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 17.099912643432617,
+      "learning_rate": 2e-05,
+      "loss": 5.6417,
+      "step": 26210
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 16.121061325073242,
+      "learning_rate": 2e-05,
+      "loss": 5.6625,
+      "step": 26220
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 16.890958786010742,
+      "learning_rate": 2e-05,
+      "loss": 5.7491,
+      "step": 26230
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 16.67205238342285,
+      "learning_rate": 2e-05,
+      "loss": 5.8196,
+      "step": 26240
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 16.85543441772461,
+      "learning_rate": 2e-05,
+      "loss": 5.4102,
+      "step": 26250
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 16.360942840576172,
+      "learning_rate": 2e-05,
+      "loss": 5.782,
+      "step": 26260
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 16.83590316772461,
+      "learning_rate": 2e-05,
+      "loss": 5.56,
+      "step": 26270
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 15.954924583435059,
+      "learning_rate": 2e-05,
+      "loss": 5.7358,
+      "step": 26280
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 16.25735092163086,
+      "learning_rate": 2e-05,
+      "loss": 5.6209,
+      "step": 26290
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 16.26198387145996,
+      "learning_rate": 2e-05,
+      "loss": 5.8932,
+      "step": 26300
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 15.394820213317871,
+      "learning_rate": 2e-05,
+      "loss": 5.6191,
+      "step": 26310
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 16.611589431762695,
+      "learning_rate": 2e-05,
+      "loss": 5.3074,
+      "step": 26320
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 15.172592163085938,
+      "learning_rate": 2e-05,
+      "loss": 5.8936,
+      "step": 26330
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 16.855127334594727,
+      "learning_rate": 2e-05,
+      "loss": 5.6221,
+      "step": 26340
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 15.028707504272461,
+      "learning_rate": 2e-05,
+      "loss": 5.6173,
+      "step": 26350
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 16.14339828491211,
+      "learning_rate": 2e-05,
+      "loss": 5.5267,
+      "step": 26360
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 14.806946754455566,
+      "learning_rate": 2e-05,
+      "loss": 5.8098,
+      "step": 26370
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 15.981405258178711,
+      "learning_rate": 2e-05,
+      "loss": 5.6367,
+      "step": 26380
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 17.596420288085938,
+      "learning_rate": 2e-05,
+      "loss": 5.6819,
+      "step": 26390
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 14.369641304016113,
+      "learning_rate": 2e-05,
+      "loss": 5.8217,
+      "step": 26400
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 16.989980697631836,
+      "learning_rate": 2e-05,
+      "loss": 5.395,
+      "step": 26410
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 17.253414154052734,
+      "learning_rate": 2e-05,
+      "loss": 5.4988,
+      "step": 26420
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 17.532106399536133,
+      "learning_rate": 2e-05,
+      "loss": 5.4459,
+      "step": 26430
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 16.816238403320312,
+      "learning_rate": 2e-05,
+      "loss": 5.8774,
+      "step": 26440
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 14.351988792419434,
+      "learning_rate": 2e-05,
+      "loss": 5.5888,
+      "step": 26450
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 16.696409225463867,
+      "learning_rate": 2e-05,
+      "loss": 5.6674,
+      "step": 26460
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 18.53239631652832,
+      "learning_rate": 2e-05,
+      "loss": 5.6723,
+      "step": 26470
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 14.938704490661621,
+      "learning_rate": 2e-05,
+      "loss": 5.531,
+      "step": 26480
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 16.196699142456055,
+      "learning_rate": 2e-05,
+      "loss": 5.6394,
+      "step": 26490
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 16.457908630371094,
+      "learning_rate": 2e-05,
+      "loss": 5.9445,
+      "step": 26500
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 15.244229316711426,
+      "learning_rate": 2e-05,
+      "loss": 5.7331,
+      "step": 26510
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 17.97573471069336,
+      "learning_rate": 2e-05,
+      "loss": 5.5793,
+      "step": 26520
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 15.274157524108887,
+      "learning_rate": 2e-05,
+      "loss": 5.7014,
+      "step": 26530
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 14.917441368103027,
+      "learning_rate": 2e-05,
+      "loss": 5.5433,
+      "step": 26540
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 14.618415832519531,
+      "learning_rate": 2e-05,
+      "loss": 5.7087,
+      "step": 26550
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 16.334800720214844,
+      "learning_rate": 2e-05,
+      "loss": 5.6136,
+      "step": 26560
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 16.425737380981445,
+      "learning_rate": 2e-05,
+      "loss": 5.7339,
+      "step": 26570
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 15.530668258666992,
+      "learning_rate": 2e-05,
+      "loss": 5.6145,
+      "step": 26580
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 18.347843170166016,
+      "learning_rate": 2e-05,
+      "loss": 5.8547,
+      "step": 26590
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 17.02280616760254,
+      "learning_rate": 2e-05,
+      "loss": 5.7053,
+      "step": 26600
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 17.459739685058594,
+      "learning_rate": 2e-05,
+      "loss": 5.8642,
+      "step": 26610
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 16.418739318847656,
+      "learning_rate": 2e-05,
+      "loss": 5.7878,
+      "step": 26620
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 15.8900728225708,
+      "learning_rate": 2e-05,
+      "loss": 5.6723,
+      "step": 26630
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 16.524166107177734,
+      "learning_rate": 2e-05,
+      "loss": 5.7608,
+      "step": 26640
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 19.23190689086914,
+      "learning_rate": 2e-05,
+      "loss": 5.8141,
+      "step": 26650
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 15.994050979614258,
+      "learning_rate": 2e-05,
+      "loss": 5.6672,
+      "step": 26660
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 15.726591110229492,
+      "learning_rate": 2e-05,
+      "loss": 5.673,
+      "step": 26670
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 18.450918197631836,
+      "learning_rate": 2e-05,
+      "loss": 5.7636,
+      "step": 26680
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 15.770854949951172,
+      "learning_rate": 2e-05,
+      "loss": 5.6478,
+      "step": 26690
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 16.53892707824707,
+      "learning_rate": 2e-05,
+      "loss": 5.8401,
+      "step": 26700
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 17.09707260131836,
+      "learning_rate": 2e-05,
+      "loss": 5.5616,
+      "step": 26710
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 16.387651443481445,
+      "learning_rate": 2e-05,
+      "loss": 5.3431,
+      "step": 26720
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 16.243350982666016,
+      "learning_rate": 2e-05,
+      "loss": 5.455,
+      "step": 26730
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 12.491715431213379,
+      "learning_rate": 2e-05,
+      "loss": 5.6621,
+      "step": 26740
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 14.740032196044922,
+      "learning_rate": 2e-05,
+      "loss": 5.5579,
+      "step": 26750
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 15.94528579711914,
+      "learning_rate": 2e-05,
+      "loss": 5.4712,
+      "step": 26760
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 16.09248924255371,
+      "learning_rate": 2e-05,
+      "loss": 5.7125,
+      "step": 26770
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 15.642207145690918,
+      "learning_rate": 2e-05,
+      "loss": 5.6087,
+      "step": 26780
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 17.739221572875977,
+      "learning_rate": 2e-05,
+      "loss": 5.4875,
+      "step": 26790
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 16.15724754333496,
+      "learning_rate": 2e-05,
+      "loss": 5.3209,
+      "step": 26800
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 16.126203536987305,
+      "learning_rate": 2e-05,
+      "loss": 5.7727,
+      "step": 26810
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 16.05876922607422,
+      "learning_rate": 2e-05,
+      "loss": 5.9296,
+      "step": 26820
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 16.255931854248047,
+      "learning_rate": 2e-05,
+      "loss": 5.8004,
+      "step": 26830
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 19.711685180664062,
+      "learning_rate": 2e-05,
+      "loss": 5.9945,
+      "step": 26840
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 16.699848175048828,
+      "learning_rate": 2e-05,
+      "loss": 5.7169,
+      "step": 26850
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 15.808348655700684,
+      "learning_rate": 2e-05,
+      "loss": 5.8994,
+      "step": 26860
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 14.194148063659668,
+      "learning_rate": 2e-05,
+      "loss": 5.5089,
+      "step": 26870
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 16.801607131958008,
+      "learning_rate": 2e-05,
+      "loss": 5.6244,
+      "step": 26880
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 15.313393592834473,
+      "learning_rate": 2e-05,
+      "loss": 5.7143,
+      "step": 26890
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 16.026620864868164,
+      "learning_rate": 2e-05,
+      "loss": 5.6011,
+      "step": 26900
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 16.50848388671875,
+      "learning_rate": 2e-05,
+      "loss": 5.7047,
+      "step": 26910
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 14.484375953674316,
+      "learning_rate": 2e-05,
+      "loss": 5.6685,
+      "step": 26920
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 16.282012939453125,
+      "learning_rate": 2e-05,
+      "loss": 5.8909,
+      "step": 26930
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 15.523811340332031,
+      "learning_rate": 2e-05,
+      "loss": 5.7258,
+      "step": 26940
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 16.436328887939453,
+      "learning_rate": 2e-05,
+      "loss": 5.3889,
+      "step": 26950
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 15.875506401062012,
+      "learning_rate": 2e-05,
+      "loss": 5.6708,
+      "step": 26960
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 16.477476119995117,
+      "learning_rate": 2e-05,
+      "loss": 5.3784,
+      "step": 26970
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 16.012998580932617,
+      "learning_rate": 2e-05,
+      "loss": 5.7197,
+      "step": 26980
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 15.219725608825684,
+      "learning_rate": 2e-05,
+      "loss": 5.7195,
+      "step": 26990
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 15.960883140563965,
+      "learning_rate": 2e-05,
+      "loss": 5.5544,
+      "step": 27000
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 16.716726303100586,
+      "learning_rate": 2e-05,
+      "loss": 5.4269,
+      "step": 27010
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 15.050186157226562,
+      "learning_rate": 2e-05,
+      "loss": 5.6537,
+      "step": 27020
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 15.081988334655762,
+      "learning_rate": 2e-05,
+      "loss": 5.583,
+      "step": 27030
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 15.405526161193848,
+      "learning_rate": 2e-05,
+      "loss": 5.7795,
+      "step": 27040
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 17.402793884277344,
+      "learning_rate": 2e-05,
+      "loss": 5.5669,
+      "step": 27050
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 16.659448623657227,
+      "learning_rate": 2e-05,
+      "loss": 5.6651,
+      "step": 27060
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 15.937525749206543,
+      "learning_rate": 2e-05,
+      "loss": 5.4914,
+      "step": 27070
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 16.455055236816406,
+      "learning_rate": 2e-05,
+      "loss": 5.5552,
+      "step": 27080
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 15.439491271972656,
+      "learning_rate": 2e-05,
+      "loss": 5.6301,
+      "step": 27090
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 16.051513671875,
+      "learning_rate": 2e-05,
+      "loss": 5.7075,
+      "step": 27100
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 15.433682441711426,
+      "learning_rate": 2e-05,
+      "loss": 5.5908,
+      "step": 27110
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 15.26805591583252,
+      "learning_rate": 2e-05,
+      "loss": 5.6377,
+      "step": 27120
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 15.808613777160645,
+      "learning_rate": 2e-05,
+      "loss": 5.6671,
+      "step": 27130
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 14.77077579498291,
+      "learning_rate": 2e-05,
+      "loss": 5.5437,
+      "step": 27140
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 18.234817504882812,
+      "learning_rate": 2e-05,
+      "loss": 5.7647,
+      "step": 27150
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 16.812824249267578,
+      "learning_rate": 2e-05,
+      "loss": 5.4503,
+      "step": 27160
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 16.10135269165039,
+      "learning_rate": 2e-05,
+      "loss": 5.8165,
+      "step": 27170
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 17.25412940979004,
+      "learning_rate": 2e-05,
+      "loss": 5.5088,
+      "step": 27180
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 15.580052375793457,
+      "learning_rate": 2e-05,
+      "loss": 5.7868,
+      "step": 27190
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 16.24864959716797,
+      "learning_rate": 2e-05,
+      "loss": 5.7187,
+      "step": 27200
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 16.742263793945312,
+      "learning_rate": 2e-05,
+      "loss": 5.6591,
+      "step": 27210
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 17.206092834472656,
+      "learning_rate": 2e-05,
+      "loss": 5.819,
+      "step": 27220
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 14.3692045211792,
+      "learning_rate": 2e-05,
+      "loss": 5.6759,
+      "step": 27230
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 15.728034019470215,
+      "learning_rate": 2e-05,
+      "loss": 5.8209,
+      "step": 27240
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 16.60927391052246,
+      "learning_rate": 2e-05,
+      "loss": 5.7156,
+      "step": 27250
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 16.52739906311035,
+      "learning_rate": 2e-05,
+      "loss": 5.5087,
+      "step": 27260
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 18.361658096313477,
+      "learning_rate": 2e-05,
+      "loss": 5.5928,
+      "step": 27270
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 16.228290557861328,
+      "learning_rate": 2e-05,
+      "loss": 5.766,
+      "step": 27280
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 18.484601974487305,
+      "learning_rate": 2e-05,
+      "loss": 5.8884,
+      "step": 27290
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 16.127214431762695,
+      "learning_rate": 2e-05,
+      "loss": 6.0763,
+      "step": 27300
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 15.91859245300293,
+      "learning_rate": 2e-05,
+      "loss": 5.8614,
+      "step": 27310
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 16.410600662231445,
+      "learning_rate": 2e-05,
+      "loss": 5.8976,
+      "step": 27320
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 15.058786392211914,
+      "learning_rate": 2e-05,
+      "loss": 5.5918,
+      "step": 27330
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 16.1148624420166,
+      "learning_rate": 2e-05,
+      "loss": 5.6728,
+      "step": 27340
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 13.488886833190918,
+      "learning_rate": 2e-05,
+      "loss": 5.8541,
+      "step": 27350
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 15.440864562988281,
+      "learning_rate": 2e-05,
+      "loss": 5.4708,
+      "step": 27360
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 15.56913948059082,
+      "learning_rate": 2e-05,
+      "loss": 5.7346,
+      "step": 27370
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 15.27827262878418,
+      "learning_rate": 2e-05,
+      "loss": 5.8792,
+      "step": 27380
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 17.355253219604492,
+      "learning_rate": 2e-05,
+      "loss": 5.3835,
+      "step": 27390
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 14.280370712280273,
+      "learning_rate": 2e-05,
+      "loss": 5.5611,
+      "step": 27400
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 17.702489852905273,
+      "learning_rate": 2e-05,
+      "loss": 5.2979,
+      "step": 27410
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 17.274635314941406,
+      "learning_rate": 2e-05,
+      "loss": 5.7541,
+      "step": 27420
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 14.85482120513916,
+      "learning_rate": 2e-05,
+      "loss": 5.6902,
+      "step": 27430
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 16.722490310668945,
+      "learning_rate": 2e-05,
+      "loss": 5.4068,
+      "step": 27440
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 14.801114082336426,
+      "learning_rate": 2e-05,
+      "loss": 5.5945,
+      "step": 27450
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 16.884883880615234,
+      "learning_rate": 2e-05,
+      "loss": 5.6464,
+      "step": 27460
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 17.573558807373047,
+      "learning_rate": 2e-05,
+      "loss": 5.8447,
+      "step": 27470
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 14.895045280456543,
+      "learning_rate": 2e-05,
+      "loss": 5.6985,
+      "step": 27480
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 16.41156005859375,
+      "learning_rate": 2e-05,
+      "loss": 5.54,
+      "step": 27490
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 15.96730899810791,
+      "learning_rate": 2e-05,
+      "loss": 5.7718,
+      "step": 27500
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 14.95307731628418,
+      "learning_rate": 2e-05,
+      "loss": 5.8408,
+      "step": 27510
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 16.519298553466797,
+      "learning_rate": 2e-05,
+      "loss": 5.5358,
+      "step": 27520
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 15.79047679901123,
+      "learning_rate": 2e-05,
+      "loss": 5.813,
+      "step": 27530
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 17.301408767700195,
+      "learning_rate": 2e-05,
+      "loss": 5.3249,
+      "step": 27540
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 16.360219955444336,
+      "learning_rate": 2e-05,
+      "loss": 5.436,
+      "step": 27550
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 16.5490665435791,
+      "learning_rate": 2e-05,
+      "loss": 5.4321,
+      "step": 27560
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 17.331926345825195,
+      "learning_rate": 2e-05,
+      "loss": 5.6459,
+      "step": 27570
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 17.119943618774414,
+      "learning_rate": 2e-05,
+      "loss": 5.7686,
+      "step": 27580
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 15.548226356506348,
+      "learning_rate": 2e-05,
+      "loss": 5.4934,
+      "step": 27590
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 16.086181640625,
+      "learning_rate": 2e-05,
+      "loss": 5.8154,
+      "step": 27600
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 16.196950912475586,
+      "learning_rate": 2e-05,
+      "loss": 5.7304,
+      "step": 27610
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 17.754539489746094,
+      "learning_rate": 2e-05,
+      "loss": 5.9199,
+      "step": 27620
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 14.885168075561523,
+      "learning_rate": 2e-05,
+      "loss": 5.6223,
+      "step": 27630
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 19.083765029907227,
+      "learning_rate": 2e-05,
+      "loss": 5.2275,
+      "step": 27640
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 14.798792839050293,
+      "learning_rate": 2e-05,
+      "loss": 5.5352,
+      "step": 27650
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 15.985157012939453,
+      "learning_rate": 2e-05,
+      "loss": 5.8815,
+      "step": 27660
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 17.68824577331543,
+      "learning_rate": 2e-05,
+      "loss": 5.7011,
+      "step": 27670
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 16.31987953186035,
+      "learning_rate": 2e-05,
+      "loss": 5.8286,
+      "step": 27680
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 17.36715316772461,
+      "learning_rate": 2e-05,
+      "loss": 5.785,
+      "step": 27690
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 17.248641967773438,
+      "learning_rate": 2e-05,
+      "loss": 5.6306,
+      "step": 27700
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 16.549015045166016,
+      "learning_rate": 2e-05,
+      "loss": 5.6187,
+      "step": 27710
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 16.897403717041016,
+      "learning_rate": 2e-05,
+      "loss": 5.6912,
+      "step": 27720
+    },
+    {
+      "epoch": 2.98,
+      "grad_norm": 15.757067680358887,
+      "learning_rate": 2e-05,
+      "loss": 5.431,
+      "step": 27730
+    },
+    {
+      "epoch": 2.98,
+      "grad_norm": 17.133358001708984,
+      "learning_rate": 2e-05,
+      "loss": 5.735,
+      "step": 27740
+    },
+    {
+      "epoch": 2.98,
+      "grad_norm": 14.59324836730957,
+      "learning_rate": 2e-05,
+      "loss": 5.4553,
+      "step": 27750
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 27954,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 3,
+  "save_steps": 250,
+  "total_flos": 7.740992572563456e+18,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}