diff --git "a/trainer_state.json" "b/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/trainer_state.json"
@@ -0,0 +1,8988 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 3.0,
+  "eval_steps": 500,
+  "global_step": 25638,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "grad_norm": 11.189342496234838,
+      "learning_rate": 4.2000000000000006e-07,
+      "loss": 0.6334,
+      "step": 20
+    },
+    {
+      "epoch": 0.0,
+      "grad_norm": 12.102946252189938,
+      "learning_rate": 6.515789473684211e-07,
+      "loss": 0.6334,
+      "step": 40
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 9.23868714422811,
+      "learning_rate": 8.831578947368421e-07,
+      "loss": 0.5919,
+      "step": 60
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 10.163427827153294,
+      "learning_rate": 1.1147368421052632e-06,
+      "loss": 0.591,
+      "step": 80
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 7.703301789458378,
+      "learning_rate": 1.3463157894736842e-06,
+      "loss": 0.5663,
+      "step": 100
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 10.292384648986614,
+      "learning_rate": 1.5778947368421053e-06,
+      "loss": 0.5649,
+      "step": 120
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 9.622475032798782,
+      "learning_rate": 1.8094736842105263e-06,
+      "loss": 0.5215,
+      "step": 140
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 12.80974448742081,
+      "learning_rate": 2.0410526315789475e-06,
+      "loss": 0.5166,
+      "step": 160
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 7.2494201428270495,
+      "learning_rate": 2.2726315789473687e-06,
+      "loss": 0.4708,
+      "step": 180
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 7.525527996410342,
+      "learning_rate": 2.5042105263157895e-06,
+      "loss": 0.4809,
+      "step": 200
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 9.553244905277726,
+      "learning_rate": 2.735789473684211e-06,
+      "loss": 0.4149,
+      "step": 220
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 10.08934392514418,
+      "learning_rate": 2.967368421052632e-06,
+      "loss": 0.4709,
+      "step": 240
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 8.77781615537256,
+      "learning_rate": 3.198947368421053e-06,
+      "loss": 0.4706,
+      "step": 260
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 8.842231539252408,
+      "learning_rate": 3.430526315789474e-06,
+      "loss": 0.4033,
+      "step": 280
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 10.007575408247385,
+      "learning_rate": 3.662105263157895e-06,
+      "loss": 0.4411,
+      "step": 300
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 8.652379461692668,
+      "learning_rate": 3.893684210526316e-06,
+      "loss": 0.4011,
+      "step": 320
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 9.03023367578369,
+      "learning_rate": 4.125263157894737e-06,
+      "loss": 0.4252,
+      "step": 340
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 8.141636176102654,
+      "learning_rate": 4.356842105263159e-06,
+      "loss": 0.4364,
+      "step": 360
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 8.361423248626743,
+      "learning_rate": 4.58842105263158e-06,
+      "loss": 0.4193,
+      "step": 380
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 9.373028764113052,
+      "learning_rate": 4.8200000000000004e-06,
+      "loss": 0.4078,
+      "step": 400
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 8.799844515033747,
+      "learning_rate": 5.051578947368422e-06,
+      "loss": 0.4291,
+      "step": 420
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 8.642174028335415,
+      "learning_rate": 5.283157894736842e-06,
+      "loss": 0.382,
+      "step": 440
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 9.451813409785364,
+      "learning_rate": 5.514736842105264e-06,
+      "loss": 0.4089,
+      "step": 460
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 11.329024075313082,
+      "learning_rate": 5.7463157894736845e-06,
+      "loss": 0.4015,
+      "step": 480
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 7.271736970582471,
+      "learning_rate": 5.977894736842106e-06,
+      "loss": 0.4028,
+      "step": 500
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 10.856360235187129,
+      "learning_rate": 6.209473684210528e-06,
+      "loss": 0.3814,
+      "step": 520
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 10.40735773829138,
+      "learning_rate": 6.441052631578948e-06,
+      "loss": 0.3868,
+      "step": 540
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 7.176143531188922,
+      "learning_rate": 6.672631578947369e-06,
+      "loss": 0.4089,
+      "step": 560
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 9.628282656024224,
+      "learning_rate": 6.90421052631579e-06,
+      "loss": 0.3771,
+      "step": 580
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 11.199839686200068,
+      "learning_rate": 7.135789473684212e-06,
+      "loss": 0.383,
+      "step": 600
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 8.042449860326105,
+      "learning_rate": 7.367368421052632e-06,
+      "loss": 0.3875,
+      "step": 620
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 6.256777636127039,
+      "learning_rate": 7.598947368421053e-06,
+      "loss": 0.399,
+      "step": 640
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 12.140400652487944,
+      "learning_rate": 7.830526315789473e-06,
+      "loss": 0.399,
+      "step": 660
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 9.31740323216306,
+      "learning_rate": 8.062105263157896e-06,
+      "loss": 0.383,
+      "step": 680
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 11.518507949844242,
+      "learning_rate": 8.293684210526317e-06,
+      "loss": 0.3673,
+      "step": 700
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 12.23485079993539,
+      "learning_rate": 8.525263157894737e-06,
+      "loss": 0.4105,
+      "step": 720
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 7.506176757455166,
+      "learning_rate": 8.756842105263158e-06,
+      "loss": 0.425,
+      "step": 740
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 28.3327677352401,
+      "learning_rate": 8.988421052631579e-06,
+      "loss": 0.3874,
+      "step": 760
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 10.006030362522525,
+      "learning_rate": 9.220000000000002e-06,
+      "loss": 0.4055,
+      "step": 780
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 10.010446813711976,
+      "learning_rate": 9.451578947368422e-06,
+      "loss": 0.3599,
+      "step": 800
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 7.043688000572858,
+      "learning_rate": 9.683157894736843e-06,
+      "loss": 0.3624,
+      "step": 820
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 9.607021831412599,
+      "learning_rate": 9.914736842105264e-06,
+      "loss": 0.3952,
+      "step": 840
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 9.147395412600936,
+      "learning_rate": 1.0146315789473686e-05,
+      "loss": 0.3819,
+      "step": 860
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 7.993375419567414,
+      "learning_rate": 1.0377894736842107e-05,
+      "loss": 0.3678,
+      "step": 880
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 7.52609874523302,
+      "learning_rate": 1.0609473684210528e-05,
+      "loss": 0.375,
+      "step": 900
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 9.52738539095041,
+      "learning_rate": 1.084105263157895e-05,
+      "loss": 0.3681,
+      "step": 920
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 11.370588516129013,
+      "learning_rate": 1.107263157894737e-05,
+      "loss": 0.3911,
+      "step": 940
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 9.391054368953709,
+      "learning_rate": 1.130421052631579e-05,
+      "loss": 0.329,
+      "step": 960
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 7.997549873904863,
+      "learning_rate": 1.1535789473684211e-05,
+      "loss": 0.359,
+      "step": 980
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 10.755576173466588,
+      "learning_rate": 1.1767368421052634e-05,
+      "loss": 0.393,
+      "step": 1000
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 6.458117606549104,
+      "learning_rate": 1.1998947368421054e-05,
+      "loss": 0.363,
+      "step": 1020
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 9.937055469862932,
+      "learning_rate": 1.2230526315789475e-05,
+      "loss": 0.3766,
+      "step": 1040
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 7.485816768913991,
+      "learning_rate": 1.2462105263157898e-05,
+      "loss": 0.3845,
+      "step": 1060
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 8.01333127284645,
+      "learning_rate": 1.2693684210526319e-05,
+      "loss": 0.3617,
+      "step": 1080
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 8.911089094498925,
+      "learning_rate": 1.292526315789474e-05,
+      "loss": 0.3599,
+      "step": 1100
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 8.32079434857686,
+      "learning_rate": 1.3156842105263158e-05,
+      "loss": 0.3722,
+      "step": 1120
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 9.806231592184165,
+      "learning_rate": 1.3388421052631581e-05,
+      "loss": 0.3755,
+      "step": 1140
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 8.798076540805445,
+      "learning_rate": 1.3620000000000002e-05,
+      "loss": 0.3616,
+      "step": 1160
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 8.343613755617508,
+      "learning_rate": 1.3851578947368423e-05,
+      "loss": 0.3631,
+      "step": 1180
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 8.35814690526204,
+      "learning_rate": 1.4083157894736843e-05,
+      "loss": 0.3638,
+      "step": 1200
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 7.706542458286787,
+      "learning_rate": 1.4314736842105266e-05,
+      "loss": 0.375,
+      "step": 1220
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 10.89212756963322,
+      "learning_rate": 1.4546315789473687e-05,
+      "loss": 0.4049,
+      "step": 1240
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 11.46198790559678,
+      "learning_rate": 1.4777894736842107e-05,
+      "loss": 0.332,
+      "step": 1260
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 8.365156167551179,
+      "learning_rate": 1.500947368421053e-05,
+      "loss": 0.3586,
+      "step": 1280
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 7.841036992292662,
+      "learning_rate": 1.5241052631578949e-05,
+      "loss": 0.368,
+      "step": 1300
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 9.84275168548854,
+      "learning_rate": 1.547263157894737e-05,
+      "loss": 0.3767,
+      "step": 1320
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 10.506254240871392,
+      "learning_rate": 1.5704210526315792e-05,
+      "loss": 0.378,
+      "step": 1340
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 7.844581104033147,
+      "learning_rate": 1.5935789473684213e-05,
+      "loss": 0.3453,
+      "step": 1360
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 8.193758138016792,
+      "learning_rate": 1.6167368421052634e-05,
+      "loss": 0.3752,
+      "step": 1380
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 7.475620318034102,
+      "learning_rate": 1.6398947368421055e-05,
+      "loss": 0.3642,
+      "step": 1400
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 7.4964850772334355,
+      "learning_rate": 1.6630526315789475e-05,
+      "loss": 0.3377,
+      "step": 1420
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 7.903753436810423,
+      "learning_rate": 1.6862105263157896e-05,
+      "loss": 0.3586,
+      "step": 1440
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 10.49594728277505,
+      "learning_rate": 1.7093684210526317e-05,
+      "loss": 0.3468,
+      "step": 1460
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 8.027713457118994,
+      "learning_rate": 1.7325263157894738e-05,
+      "loss": 0.3491,
+      "step": 1480
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 7.2748681125287415,
+      "learning_rate": 1.755684210526316e-05,
+      "loss": 0.3581,
+      "step": 1500
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 6.077326940397276,
+      "learning_rate": 1.7788421052631583e-05,
+      "loss": 0.3356,
+      "step": 1520
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 6.724980645701413,
+      "learning_rate": 1.802e-05,
+      "loss": 0.3656,
+      "step": 1540
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 8.592315221136559,
+      "learning_rate": 1.825157894736842e-05,
+      "loss": 0.3467,
+      "step": 1560
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 9.036253074638084,
+      "learning_rate": 1.8483157894736845e-05,
+      "loss": 0.3414,
+      "step": 1580
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 11.349080832480995,
+      "learning_rate": 1.8714736842105266e-05,
+      "loss": 0.3655,
+      "step": 1600
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 9.361475025905081,
+      "learning_rate": 1.8946315789473687e-05,
+      "loss": 0.3564,
+      "step": 1620
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 6.53643855564322,
+      "learning_rate": 1.9177894736842108e-05,
+      "loss": 0.3714,
+      "step": 1640
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 6.65699298275954,
+      "learning_rate": 1.940947368421053e-05,
+      "loss": 0.3546,
+      "step": 1660
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 9.856127510930383,
+      "learning_rate": 1.964105263157895e-05,
+      "loss": 0.3308,
+      "step": 1680
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 8.532146476622113,
+      "learning_rate": 1.987263157894737e-05,
+      "loss": 0.366,
+      "step": 1700
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 8.327666937449752,
+      "learning_rate": 1.999999532098205e-05,
+      "loss": 0.3247,
+      "step": 1720
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 10.876035443701728,
+      "learning_rate": 1.99999578888647e-05,
+      "loss": 0.3543,
+      "step": 1740
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 8.263711805253918,
+      "learning_rate": 1.999988302477013e-05,
+      "loss": 0.3664,
+      "step": 1760
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 7.575419338189409,
+      "learning_rate": 1.9999770728978607e-05,
+      "loss": 0.345,
+      "step": 1780
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 7.814279338388738,
+      "learning_rate": 1.9999621001910508e-05,
+      "loss": 0.3573,
+      "step": 1800
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 13.040107479471628,
+      "learning_rate": 1.999943384412636e-05,
+      "loss": 0.3344,
+      "step": 1820
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 7.535441012066771,
+      "learning_rate": 1.99992092563268e-05,
+      "loss": 0.3552,
+      "step": 1840
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 8.059491683038381,
+      "learning_rate": 1.9998947239352593e-05,
+      "loss": 0.3442,
+      "step": 1860
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 6.700609139607897,
+      "learning_rate": 1.999864779418462e-05,
+      "loss": 0.3584,
+      "step": 1880
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 6.831136707088909,
+      "learning_rate": 1.999831092194389e-05,
+      "loss": 0.3341,
+      "step": 1900
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 10.30262237054047,
+      "learning_rate": 1.9997936623891498e-05,
+      "loss": 0.3704,
+      "step": 1920
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 6.703691792033561,
+      "learning_rate": 1.9997524901428677e-05,
+      "loss": 0.3769,
+      "step": 1940
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 7.251026212477321,
+      "learning_rate": 1.9997075756096734e-05,
+      "loss": 0.3234,
+      "step": 1960
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 6.14471190830255,
+      "learning_rate": 1.9996589189577094e-05,
+      "loss": 0.3443,
+      "step": 1980
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 7.254885638733633,
+      "learning_rate": 1.999606520369125e-05,
+      "loss": 0.3178,
+      "step": 2000
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 7.330977870771951,
+      "learning_rate": 1.9995503800400805e-05,
+      "loss": 0.3309,
+      "step": 2020
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 7.291895603037368,
+      "learning_rate": 1.9994904981807407e-05,
+      "loss": 0.3372,
+      "step": 2040
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 12.772894555548028,
+      "learning_rate": 1.9994268750152793e-05,
+      "loss": 0.3747,
+      "step": 2060
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 9.497788021161657,
+      "learning_rate": 1.999359510781875e-05,
+      "loss": 0.3529,
+      "step": 2080
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 6.581748417513793,
+      "learning_rate": 1.9992884057327118e-05,
+      "loss": 0.3395,
+      "step": 2100
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 9.523030924078585,
+      "learning_rate": 1.9992135601339776e-05,
+      "loss": 0.3469,
+      "step": 2120
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 17.640537636796516,
+      "learning_rate": 1.999134974265864e-05,
+      "loss": 0.3365,
+      "step": 2140
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 5.4385511763432985,
+      "learning_rate": 1.999052648422563e-05,
+      "loss": 0.3845,
+      "step": 2160
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 7.585921156892153,
+      "learning_rate": 1.9989665829122704e-05,
+      "loss": 0.3464,
+      "step": 2180
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 12.163408304702841,
+      "learning_rate": 1.9988767780571778e-05,
+      "loss": 0.3127,
+      "step": 2200
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 8.246994395835456,
+      "learning_rate": 1.9987832341934793e-05,
+      "loss": 0.3424,
+      "step": 2220
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 9.628684787640838,
+      "learning_rate": 1.9986859516713636e-05,
+      "loss": 0.3472,
+      "step": 2240
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 10.912326639203211,
+      "learning_rate": 1.9985849308550167e-05,
+      "loss": 0.3326,
+      "step": 2260
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 6.419410251137725,
+      "learning_rate": 1.9984801721226188e-05,
+      "loss": 0.3297,
+      "step": 2280
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 7.798212276898864,
+      "learning_rate": 1.998371675866344e-05,
+      "loss": 0.3389,
+      "step": 2300
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 7.811483576458649,
+      "learning_rate": 1.9982594424923566e-05,
+      "loss": 0.332,
+      "step": 2320
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 6.967736687994363,
+      "learning_rate": 1.9981434724208126e-05,
+      "loss": 0.3099,
+      "step": 2340
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 9.635115819134949,
+      "learning_rate": 1.9980237660858565e-05,
+      "loss": 0.3285,
+      "step": 2360
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 6.720069081541748,
+      "learning_rate": 1.997900323935619e-05,
+      "loss": 0.343,
+      "step": 2380
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 10.300889386045979,
+      "learning_rate": 1.9977731464322162e-05,
+      "loss": 0.3667,
+      "step": 2400
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 7.888854177455323,
+      "learning_rate": 1.9976422340517495e-05,
+      "loss": 0.3397,
+      "step": 2420
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 9.643051064776678,
+      "learning_rate": 1.9975075872843e-05,
+      "loss": 0.3371,
+      "step": 2440
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 6.166262467342929,
+      "learning_rate": 1.9973692066339298e-05,
+      "loss": 0.3424,
+      "step": 2460
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 7.520601841401043,
+      "learning_rate": 1.9972270926186794e-05,
+      "loss": 0.3083,
+      "step": 2480
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 6.095472200047568,
+      "learning_rate": 1.9970812457705642e-05,
+      "loss": 0.3512,
+      "step": 2500
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 7.259492119293066,
+      "learning_rate": 1.9969316666355757e-05,
+      "loss": 0.3309,
+      "step": 2520
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 8.355978245335697,
+      "learning_rate": 1.996778355773676e-05,
+      "loss": 0.3296,
+      "step": 2540
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 9.612502269967832,
+      "learning_rate": 1.996621313758798e-05,
+      "loss": 0.3391,
+      "step": 2560
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 8.148285817739712,
+      "learning_rate": 1.9964605411788424e-05,
+      "loss": 0.3416,
+      "step": 2580
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 7.645206131186397,
+      "learning_rate": 1.9962960386356748e-05,
+      "loss": 0.352,
+      "step": 2600
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 5.406075006199623,
+      "learning_rate": 1.9961278067451258e-05,
+      "loss": 0.3194,
+      "step": 2620
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 7.23674984010716,
+      "learning_rate": 1.995955846136986e-05,
+      "loss": 0.3216,
+      "step": 2640
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 8.044071872202302,
+      "learning_rate": 1.9957801574550044e-05,
+      "loss": 0.321,
+      "step": 2660
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 7.2624583573058485,
+      "learning_rate": 1.9956007413568874e-05,
+      "loss": 0.3352,
+      "step": 2680
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 9.277277549105909,
+      "learning_rate": 1.9954175985142955e-05,
+      "loss": 0.3391,
+      "step": 2700
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 10.975275864316648,
+      "learning_rate": 1.9952307296128395e-05,
+      "loss": 0.3298,
+      "step": 2720
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 11.416959735607907,
+      "learning_rate": 1.9950401353520795e-05,
+      "loss": 0.3382,
+      "step": 2740
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 7.384993942351236,
+      "learning_rate": 1.994845816445522e-05,
+      "loss": 0.3422,
+      "step": 2760
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 6.775839902047113,
+      "learning_rate": 1.9946477736206167e-05,
+      "loss": 0.3694,
+      "step": 2780
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 8.246414330131564,
+      "learning_rate": 1.9944460076187546e-05,
+      "loss": 0.3458,
+      "step": 2800
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 7.152660285783071,
+      "learning_rate": 1.9942405191952646e-05,
+      "loss": 0.3426,
+      "step": 2820
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 6.9408668390055315,
+      "learning_rate": 1.9940313091194096e-05,
+      "loss": 0.3097,
+      "step": 2840
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 5.797513942002627,
+      "learning_rate": 1.9938183781743873e-05,
+      "loss": 0.3102,
+      "step": 2860
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 7.361935180345544,
+      "learning_rate": 1.9936017271573223e-05,
+      "loss": 0.3102,
+      "step": 2880
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 7.4796257480282,
+      "learning_rate": 1.9933813568792678e-05,
+      "loss": 0.3256,
+      "step": 2900
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 5.500192812227557,
+      "learning_rate": 1.9931572681651984e-05,
+      "loss": 0.3107,
+      "step": 2920
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 13.84776864376109,
+      "learning_rate": 1.9929294618540103e-05,
+      "loss": 0.3062,
+      "step": 2940
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 7.937410128828084,
+      "learning_rate": 1.992697938798516e-05,
+      "loss": 0.3293,
+      "step": 2960
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 5.447364190742796,
+      "learning_rate": 1.9924626998654433e-05,
+      "loss": 0.3318,
+      "step": 2980
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 7.825239369201605,
+      "learning_rate": 1.9922237459354287e-05,
+      "loss": 0.3238,
+      "step": 3000
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 6.466926354506395,
+      "learning_rate": 1.9919810779030177e-05,
+      "loss": 0.3136,
+      "step": 3020
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 10.115420483963225,
+      "learning_rate": 1.9917346966766596e-05,
+      "loss": 0.3329,
+      "step": 3040
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 6.08100630613984,
+      "learning_rate": 1.991484603178704e-05,
+      "loss": 0.2999,
+      "step": 3060
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 6.153474908414458,
+      "learning_rate": 1.991230798345398e-05,
+      "loss": 0.3344,
+      "step": 3080
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 6.956842076140817,
+      "learning_rate": 1.990973283126882e-05,
+      "loss": 0.3356,
+      "step": 3100
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 7.977473013654998,
+      "learning_rate": 1.990712058487187e-05,
+      "loss": 0.2984,
+      "step": 3120
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 7.472067890046691,
+      "learning_rate": 1.9904471254042304e-05,
+      "loss": 0.308,
+      "step": 3140
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 6.98090482864238,
+      "learning_rate": 1.9901784848698125e-05,
+      "loss": 0.3036,
+      "step": 3160
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 7.229553593404789,
+      "learning_rate": 1.989906137889613e-05,
+      "loss": 0.3027,
+      "step": 3180
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 5.9700766482137295,
+      "learning_rate": 1.9896300854831862e-05,
+      "loss": 0.3329,
+      "step": 3200
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 7.864131148182889,
+      "learning_rate": 1.989350328683959e-05,
+      "loss": 0.308,
+      "step": 3220
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 20.04577205534678,
+      "learning_rate": 1.989066868539225e-05,
+      "loss": 0.3217,
+      "step": 3240
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 8.46751929768339,
+      "learning_rate": 1.9887797061101424e-05,
+      "loss": 0.3036,
+      "step": 3260
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 7.667111535917501,
+      "learning_rate": 1.9884888424717296e-05,
+      "loss": 0.3084,
+      "step": 3280
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 6.615069450077278,
+      "learning_rate": 1.9881942787128594e-05,
+      "loss": 0.3177,
+      "step": 3300
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 9.324876177990959,
+      "learning_rate": 1.9878960159362572e-05,
+      "loss": 0.3496,
+      "step": 3320
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 7.1979554134343635,
+      "learning_rate": 1.987594055258496e-05,
+      "loss": 0.3167,
+      "step": 3340
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 29.076306154493622,
+      "learning_rate": 1.987288397809993e-05,
+      "loss": 0.34,
+      "step": 3360
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 7.861933206527291,
+      "learning_rate": 1.986979044735002e-05,
+      "loss": 0.3424,
+      "step": 3380
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 5.7878723156599445,
+      "learning_rate": 1.9866659971916146e-05,
+      "loss": 0.3157,
+      "step": 3400
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 8.470531768859948,
+      "learning_rate": 1.9863492563517513e-05,
+      "loss": 0.3201,
+      "step": 3420
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 8.104846079245535,
+      "learning_rate": 1.98602882340116e-05,
+      "loss": 0.322,
+      "step": 3440
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 6.180289951670451,
+      "learning_rate": 1.9857046995394084e-05,
+      "loss": 0.2991,
+      "step": 3460
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 4.962160167060266,
+      "learning_rate": 1.985376885979884e-05,
+      "loss": 0.3097,
+      "step": 3480
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 6.813966925636774,
+      "learning_rate": 1.9850453839497844e-05,
+      "loss": 0.3294,
+      "step": 3500
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 5.902763508299491,
+      "learning_rate": 1.9847101946901172e-05,
+      "loss": 0.2967,
+      "step": 3520
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 7.359798475906595,
+      "learning_rate": 1.984371319455693e-05,
+      "loss": 0.3268,
+      "step": 3540
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 7.377881796357637,
+      "learning_rate": 1.9840287595151205e-05,
+      "loss": 0.3251,
+      "step": 3560
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 7.261601866242624,
+      "learning_rate": 1.9836825161508032e-05,
+      "loss": 0.284,
+      "step": 3580
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 6.123609423918951,
+      "learning_rate": 1.983332590658934e-05,
+      "loss": 0.3137,
+      "step": 3600
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 7.205290164862431,
+      "learning_rate": 1.9829789843494885e-05,
+      "loss": 0.3,
+      "step": 3620
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 6.0471563544848035,
+      "learning_rate": 1.9826216985462245e-05,
+      "loss": 0.2991,
+      "step": 3640
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 7.436589545934819,
+      "learning_rate": 1.9822607345866715e-05,
+      "loss": 0.3299,
+      "step": 3660
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 8.240853065109379,
+      "learning_rate": 1.9818960938221308e-05,
+      "loss": 0.2886,
+      "step": 3680
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 5.451034188795461,
+      "learning_rate": 1.9815277776176663e-05,
+      "loss": 0.3138,
+      "step": 3700
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 7.874196329434049,
+      "learning_rate": 1.9811557873521027e-05,
+      "loss": 0.2946,
+      "step": 3720
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 6.547528436867746,
+      "learning_rate": 1.9807801244180174e-05,
+      "loss": 0.2978,
+      "step": 3740
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 5.816214082411462,
+      "learning_rate": 1.980400790221739e-05,
+      "loss": 0.3067,
+      "step": 3760
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 6.245697982302123,
+      "learning_rate": 1.9800177861833372e-05,
+      "loss": 0.3191,
+      "step": 3780
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 7.932045331416635,
+      "learning_rate": 1.9796311137366212e-05,
+      "loss": 0.3439,
+      "step": 3800
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 6.088006858952151,
+      "learning_rate": 1.9792407743291343e-05,
+      "loss": 0.33,
+      "step": 3820
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 5.770262468473867,
+      "learning_rate": 1.9788467694221453e-05,
+      "loss": 0.2897,
+      "step": 3840
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 7.978864884873927,
+      "learning_rate": 1.9784491004906468e-05,
+      "loss": 0.3118,
+      "step": 3860
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 6.631368778757732,
+      "learning_rate": 1.9780477690233476e-05,
+      "loss": 0.3165,
+      "step": 3880
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 6.541290304773085,
+      "learning_rate": 1.9776427765226664e-05,
+      "loss": 0.3379,
+      "step": 3900
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 6.42886929503141,
+      "learning_rate": 1.9772341245047285e-05,
+      "loss": 0.3136,
+      "step": 3920
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 9.004651033459043,
+      "learning_rate": 1.9768218144993593e-05,
+      "loss": 0.311,
+      "step": 3940
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 6.241942743461084,
+      "learning_rate": 1.976405848050077e-05,
+      "loss": 0.3224,
+      "step": 3960
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 7.030346892956473,
+      "learning_rate": 1.975986226714089e-05,
+      "loss": 0.3513,
+      "step": 3980
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 6.984184655106913,
+      "learning_rate": 1.9755629520622835e-05,
+      "loss": 0.3031,
+      "step": 4000
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 3.6594620639728253,
+      "learning_rate": 1.9751360256792274e-05,
+      "loss": 0.2819,
+      "step": 4020
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 3.9453364305431418,
+      "learning_rate": 1.9747054491631567e-05,
+      "loss": 0.2806,
+      "step": 4040
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 3.8004852838340475,
+      "learning_rate": 1.9742712241259726e-05,
+      "loss": 0.2712,
+      "step": 4060
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 4.671864793440983,
+      "learning_rate": 1.9738333521932348e-05,
+      "loss": 0.264,
+      "step": 4080
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 3.9158215253486452,
+      "learning_rate": 1.9733918350041546e-05,
+      "loss": 0.2466,
+      "step": 4100
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 3.232730170815344,
+      "learning_rate": 1.972946674211591e-05,
+      "loss": 0.28,
+      "step": 4120
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 5.26102706338969,
+      "learning_rate": 1.972497871482043e-05,
+      "loss": 0.2793,
+      "step": 4140
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 6.036896745386087,
+      "learning_rate": 1.972045428495642e-05,
+      "loss": 0.2551,
+      "step": 4160
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 3.4899809349872823,
+      "learning_rate": 1.971589346946149e-05,
+      "loss": 0.2746,
+      "step": 4180
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 4.003065603442425,
+      "learning_rate": 1.9711296285409443e-05,
+      "loss": 0.2639,
+      "step": 4200
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 2.8477299266017626,
+      "learning_rate": 1.9706662750010255e-05,
+      "loss": 0.2689,
+      "step": 4220
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 3.5491058459283837,
+      "learning_rate": 1.970199288060996e-05,
+      "loss": 0.2642,
+      "step": 4240
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 3.658049963633355,
+      "learning_rate": 1.969728669469063e-05,
+      "loss": 0.2699,
+      "step": 4260
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 3.7093939532444127,
+      "learning_rate": 1.9692544209870287e-05,
+      "loss": 0.2697,
+      "step": 4280
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 3.0342042036143444,
+      "learning_rate": 1.968776544390283e-05,
+      "loss": 0.2654,
+      "step": 4300
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 3.8174199112021174,
+      "learning_rate": 1.9682950414678004e-05,
+      "loss": 0.2525,
+      "step": 4320
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 5.35961771156863,
+      "learning_rate": 1.9678099140221284e-05,
+      "loss": 0.2464,
+      "step": 4340
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 5.063501894675061,
+      "learning_rate": 1.9673211638693844e-05,
+      "loss": 0.2788,
+      "step": 4360
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 3.8581072620927723,
+      "learning_rate": 1.9668287928392478e-05,
+      "loss": 0.2743,
+      "step": 4380
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 3.386524586463125,
+      "learning_rate": 1.9663328027749523e-05,
+      "loss": 0.2547,
+      "step": 4400
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 4.049063661443696,
+      "learning_rate": 1.9658331955332807e-05,
+      "loss": 0.2787,
+      "step": 4420
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 5.12761026294579,
+      "learning_rate": 1.9653299729845564e-05,
+      "loss": 0.2577,
+      "step": 4440
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 3.808933278135121,
+      "learning_rate": 1.9648231370126375e-05,
+      "loss": 0.2654,
+      "step": 4460
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 4.402659150192465,
+      "learning_rate": 1.9643126895149087e-05,
+      "loss": 0.2603,
+      "step": 4480
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 8.492295307503369,
+      "learning_rate": 1.963798632402275e-05,
+      "loss": 0.2691,
+      "step": 4500
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 3.628014593555578,
+      "learning_rate": 1.9632809675991544e-05,
+      "loss": 0.2711,
+      "step": 4520
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 5.207256154885955,
+      "learning_rate": 1.9627596970434708e-05,
+      "loss": 0.2604,
+      "step": 4540
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 3.892539047603965,
+      "learning_rate": 1.962234822686646e-05,
+      "loss": 0.2693,
+      "step": 4560
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 4.005334872796194,
+      "learning_rate": 1.961706346493594e-05,
+      "loss": 0.2774,
+      "step": 4580
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 4.456881895914548,
+      "learning_rate": 1.9611742704427106e-05,
+      "loss": 0.2531,
+      "step": 4600
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 3.925512807597538,
+      "learning_rate": 1.9606385965258697e-05,
+      "loss": 0.2507,
+      "step": 4620
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 4.401211069070843,
+      "learning_rate": 1.9600993267484147e-05,
+      "loss": 0.262,
+      "step": 4640
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 5.3437799820979075,
+      "learning_rate": 1.9595564631291478e-05,
+      "loss": 0.2516,
+      "step": 4660
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 4.553539308066039,
+      "learning_rate": 1.9590100077003278e-05,
+      "loss": 0.2603,
+      "step": 4680
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 3.169670353545694,
+      "learning_rate": 1.958459962507658e-05,
+      "loss": 0.2875,
+      "step": 4700
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 3.572973282405317,
+      "learning_rate": 1.9579063296102812e-05,
+      "loss": 0.2671,
+      "step": 4720
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 3.59782258517998,
+      "learning_rate": 1.957349111080771e-05,
+      "loss": 0.2837,
+      "step": 4740
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 5.199297439358037,
+      "learning_rate": 1.956788309005123e-05,
+      "loss": 0.2753,
+      "step": 4760
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 4.993257936146954,
+      "learning_rate": 1.95622392548275e-05,
+      "loss": 0.2714,
+      "step": 4780
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 3.7288744650364642,
+      "learning_rate": 1.955655962626471e-05,
+      "loss": 0.2609,
+      "step": 4800
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 5.377784540016259,
+      "learning_rate": 1.9550844225625043e-05,
+      "loss": 0.2605,
+      "step": 4820
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 3.1237811954279353,
+      "learning_rate": 1.954509307430461e-05,
+      "loss": 0.2569,
+      "step": 4840
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 4.101705029145583,
+      "learning_rate": 1.953930619383335e-05,
+      "loss": 0.2719,
+      "step": 4860
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 2.734092828632754,
+      "learning_rate": 1.953348360587496e-05,
+      "loss": 0.2503,
+      "step": 4880
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 3.4516150854343604,
+      "learning_rate": 1.9527625332226804e-05,
+      "loss": 0.2588,
+      "step": 4900
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 3.0561153265178085,
+      "learning_rate": 1.952173139481985e-05,
+      "loss": 0.2779,
+      "step": 4920
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 3.5096519082422986,
+      "learning_rate": 1.9515801815718566e-05,
+      "loss": 0.2567,
+      "step": 4940
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 3.1690240083159993,
+      "learning_rate": 1.9509836617120856e-05,
+      "loss": 0.2466,
+      "step": 4960
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 3.2970371296868484,
+      "learning_rate": 1.950383582135796e-05,
+      "loss": 0.2762,
+      "step": 4980
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 4.326207366177782,
+      "learning_rate": 1.9497799450894387e-05,
+      "loss": 0.2681,
+      "step": 5000
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 3.977255410491104,
+      "learning_rate": 1.949172752832782e-05,
+      "loss": 0.235,
+      "step": 5020
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 3.5634772315616665,
+      "learning_rate": 1.9485620076389036e-05,
+      "loss": 0.254,
+      "step": 5040
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 2.9667663371741377,
+      "learning_rate": 1.9479477117941815e-05,
+      "loss": 0.2582,
+      "step": 5060
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 3.628580495546253,
+      "learning_rate": 1.9473298675982863e-05,
+      "loss": 0.2526,
+      "step": 5080
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 7.393689026504584,
+      "learning_rate": 1.946708477364172e-05,
+      "loss": 0.2587,
+      "step": 5100
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 3.1975448011133407,
+      "learning_rate": 1.9460835434180677e-05,
+      "loss": 0.2679,
+      "step": 5120
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 3.231683615655966,
+      "learning_rate": 1.9454550680994684e-05,
+      "loss": 0.2589,
+      "step": 5140
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 2.956736460495959,
+      "learning_rate": 1.9448230537611266e-05,
+      "loss": 0.2594,
+      "step": 5160
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 3.326543297808526,
+      "learning_rate": 1.9441875027690435e-05,
+      "loss": 0.2556,
+      "step": 5180
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 3.635188025395725,
+      "learning_rate": 1.94354841750246e-05,
+      "loss": 0.2391,
+      "step": 5200
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 4.934642109517512,
+      "learning_rate": 1.942905800353848e-05,
+      "loss": 0.2576,
+      "step": 5220
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 3.8685357103793767,
+      "learning_rate": 1.9422596537289016e-05,
+      "loss": 0.2735,
+      "step": 5240
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 4.387694986472,
+      "learning_rate": 1.941609980046527e-05,
+      "loss": 0.2584,
+      "step": 5260
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 2.8046325848231977,
+      "learning_rate": 1.9409567817388347e-05,
+      "loss": 0.2527,
+      "step": 5280
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 3.4012361074638466,
+      "learning_rate": 1.94030006125113e-05,
+      "loss": 0.2566,
+      "step": 5300
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 5.450089536158873,
+      "learning_rate": 1.9396398210419038e-05,
+      "loss": 0.2667,
+      "step": 5320
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 3.7967573233998677,
+      "learning_rate": 1.938976063582824e-05,
+      "loss": 0.2536,
+      "step": 5340
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 3.0487888683110866,
+      "learning_rate": 1.938308791358723e-05,
+      "loss": 0.2673,
+      "step": 5360
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 3.2736066169116285,
+      "learning_rate": 1.9376380068675956e-05,
+      "loss": 0.2592,
+      "step": 5380
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 9.727952335441667,
+      "learning_rate": 1.93696371262058e-05,
+      "loss": 0.273,
+      "step": 5400
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 3.3824664284674686,
+      "learning_rate": 1.936285911141958e-05,
+      "loss": 0.2787,
+      "step": 5420
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 6.879080376487243,
+      "learning_rate": 1.935604604969138e-05,
+      "loss": 0.2698,
+      "step": 5440
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 4.579168728014292,
+      "learning_rate": 1.934919796652649e-05,
+      "loss": 0.2589,
+      "step": 5460
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 4.6776324362587784,
+      "learning_rate": 1.9342314887561323e-05,
+      "loss": 0.2544,
+      "step": 5480
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 3.693997622998789,
+      "learning_rate": 1.9335396838563282e-05,
+      "loss": 0.2446,
+      "step": 5500
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 5.06873813162095,
+      "learning_rate": 1.932844384543069e-05,
+      "loss": 0.2618,
+      "step": 5520
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 6.816411566527297,
+      "learning_rate": 1.9321455934192693e-05,
+      "loss": 0.2614,
+      "step": 5540
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 3.5475232040366853,
+      "learning_rate": 1.9314433131009146e-05,
+      "loss": 0.2942,
+      "step": 5560
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 5.287550198652632,
+      "learning_rate": 1.9307375462170527e-05,
+      "loss": 0.2482,
+      "step": 5580
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 11.389854038766874,
+      "learning_rate": 1.9300282954097847e-05,
+      "loss": 0.245,
+      "step": 5600
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 4.178435806663215,
+      "learning_rate": 1.9293155633342534e-05,
+      "loss": 0.2534,
+      "step": 5620
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 3.1710922703595035,
+      "learning_rate": 1.928599352658633e-05,
+      "loss": 0.2622,
+      "step": 5640
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 2.6946627953356996,
+      "learning_rate": 1.927879666064122e-05,
+      "loss": 0.2626,
+      "step": 5660
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 4.8581812333670955,
+      "learning_rate": 1.92715650624493e-05,
+      "loss": 0.2346,
+      "step": 5680
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 3.092927765677092,
+      "learning_rate": 1.926429875908269e-05,
+      "loss": 0.2585,
+      "step": 5700
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 4.029661826813332,
+      "learning_rate": 1.9256997777743437e-05,
+      "loss": 0.2467,
+      "step": 5720
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 4.642691104141709,
+      "learning_rate": 1.924966214576341e-05,
+      "loss": 0.2681,
+      "step": 5740
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 7.315047122076067,
+      "learning_rate": 1.9242291890604186e-05,
+      "loss": 0.2508,
+      "step": 5760
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 3.540549397482083,
+      "learning_rate": 1.923488703985696e-05,
+      "loss": 0.2592,
+      "step": 5780
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 3.475018859373003,
+      "learning_rate": 1.9227447621242458e-05,
+      "loss": 0.2422,
+      "step": 5800
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 3.8676644079356355,
+      "learning_rate": 1.921997366261078e-05,
+      "loss": 0.2603,
+      "step": 5820
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 3.5885966628483,
+      "learning_rate": 1.9212465191941346e-05,
+      "loss": 0.2542,
+      "step": 5840
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 3.6495890137770886,
+      "learning_rate": 1.920492223734279e-05,
+      "loss": 0.2374,
+      "step": 5860
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 4.323488251305847,
+      "learning_rate": 1.9197344827052813e-05,
+      "loss": 0.2687,
+      "step": 5880
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 5.051497760408599,
+      "learning_rate": 1.9189732989438123e-05,
+      "loss": 0.2678,
+      "step": 5900
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 3.5559342765441495,
+      "learning_rate": 1.9182086752994295e-05,
+      "loss": 0.2726,
+      "step": 5920
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 3.029184640641301,
+      "learning_rate": 1.9174406146345694e-05,
+      "loss": 0.246,
+      "step": 5940
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 3.4792605785250994,
+      "learning_rate": 1.9166691198245343e-05,
+      "loss": 0.2843,
+      "step": 5960
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 3.0247490770922,
+      "learning_rate": 1.915894193757483e-05,
+      "loss": 0.2528,
+      "step": 5980
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 3.7727567007247855,
+      "learning_rate": 1.915115839334419e-05,
+      "loss": 0.2442,
+      "step": 6000
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 2.31881357193246,
+      "learning_rate": 1.9143340594691806e-05,
+      "loss": 0.2561,
+      "step": 6020
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 3.138315746753083,
+      "learning_rate": 1.913548857088429e-05,
+      "loss": 0.2636,
+      "step": 6040
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 8.010145906221986,
+      "learning_rate": 1.9127602351316393e-05,
+      "loss": 0.2876,
+      "step": 6060
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 3.867596475572881,
+      "learning_rate": 1.9119681965510858e-05,
+      "loss": 0.2659,
+      "step": 6080
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 3.3587166074477777,
+      "learning_rate": 1.9111727443118357e-05,
+      "loss": 0.2634,
+      "step": 6100
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 2.89895288044486,
+      "learning_rate": 1.9103738813917336e-05,
+      "loss": 0.2533,
+      "step": 6120
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 4.037752097559083,
+      "learning_rate": 1.909571610781393e-05,
+      "loss": 0.2485,
+      "step": 6140
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 3.763608653829743,
+      "learning_rate": 1.908765935484185e-05,
+      "loss": 0.259,
+      "step": 6160
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 3.161229351032073,
+      "learning_rate": 1.9079568585162248e-05,
+      "loss": 0.2508,
+      "step": 6180
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 2.725004696623228,
+      "learning_rate": 1.9071443829063644e-05,
+      "loss": 0.2524,
+      "step": 6200
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 3.7116897623627985,
+      "learning_rate": 1.906328511696177e-05,
+      "loss": 0.2598,
+      "step": 6220
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 2.9004110110815544,
+      "learning_rate": 1.9055092479399474e-05,
+      "loss": 0.2553,
+      "step": 6240
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 4.311030925472568,
+      "learning_rate": 1.9046865947046627e-05,
+      "loss": 0.2678,
+      "step": 6260
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 3.5377476420175977,
+      "learning_rate": 1.903860555069997e-05,
+      "loss": 0.2379,
+      "step": 6280
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 3.58934533910836,
+      "learning_rate": 1.9030311321283022e-05,
+      "loss": 0.2608,
+      "step": 6300
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 3.4612535457241624,
+      "learning_rate": 1.9021983289845956e-05,
+      "loss": 0.246,
+      "step": 6320
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 3.9763337015158258,
+      "learning_rate": 1.9013621487565498e-05,
+      "loss": 0.238,
+      "step": 6340
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 3.05488683337997,
+      "learning_rate": 1.9005225945744782e-05,
+      "loss": 0.2472,
+      "step": 6360
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 3.2350295137123313,
+      "learning_rate": 1.8996796695813256e-05,
+      "loss": 0.261,
+      "step": 6380
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 3.2500906711548123,
+      "learning_rate": 1.8988333769326564e-05,
+      "loss": 0.2641,
+      "step": 6400
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 2.815087548041305,
+      "learning_rate": 1.8979837197966415e-05,
+      "loss": 0.2583,
+      "step": 6420
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 3.7397965536014257,
+      "learning_rate": 1.8971307013540464e-05,
+      "loss": 0.2485,
+      "step": 6440
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 2.7592373806083423,
+      "learning_rate": 1.8962743247982205e-05,
+      "loss": 0.2641,
+      "step": 6460
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 2.818080452105616,
+      "learning_rate": 1.895414593335085e-05,
+      "loss": 0.2554,
+      "step": 6480
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 3.627385505536174,
+      "learning_rate": 1.8945515101831203e-05,
+      "loss": 0.2693,
+      "step": 6500
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 3.255509767944512,
+      "learning_rate": 1.893685078573354e-05,
+      "loss": 0.2459,
+      "step": 6520
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 4.135182845698409,
+      "learning_rate": 1.8928153017493488e-05,
+      "loss": 0.2499,
+      "step": 6540
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 5.892816677202276,
+      "learning_rate": 1.8919421829671904e-05,
+      "loss": 0.2786,
+      "step": 6560
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 3.479892875732629,
+      "learning_rate": 1.891065725495476e-05,
+      "loss": 0.265,
+      "step": 6580
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 3.8343809741781216,
+      "learning_rate": 1.890185932615301e-05,
+      "loss": 0.2527,
+      "step": 6600
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 3.1643033159708107,
+      "learning_rate": 1.8893028076202472e-05,
+      "loss": 0.2741,
+      "step": 6620
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 4.477352585232187,
+      "learning_rate": 1.8884163538163708e-05,
+      "loss": 0.2525,
+      "step": 6640
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 3.529760400999529,
+      "learning_rate": 1.8875265745221894e-05,
+      "loss": 0.2643,
+      "step": 6660
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 2.3110189076619525,
+      "learning_rate": 1.8866334730686703e-05,
+      "loss": 0.2721,
+      "step": 6680
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 3.685226774238388,
+      "learning_rate": 1.8857370527992165e-05,
+      "loss": 0.2549,
+      "step": 6700
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 3.8074891978039864,
+      "learning_rate": 1.8848373170696567e-05,
+      "loss": 0.2418,
+      "step": 6720
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 7.507877790185021,
+      "learning_rate": 1.8839342692482302e-05,
+      "loss": 0.2451,
+      "step": 6740
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 2.7359733051983186,
+      "learning_rate": 1.8830279127155764e-05,
+      "loss": 0.2484,
+      "step": 6760
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 2.4635290638854657,
+      "learning_rate": 1.88211825086472e-05,
+      "loss": 0.2407,
+      "step": 6780
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 3.040130192327906,
+      "learning_rate": 1.8812052871010605e-05,
+      "loss": 0.2544,
+      "step": 6800
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 5.610778632978326,
+      "learning_rate": 1.8802890248423574e-05,
+      "loss": 0.2397,
+      "step": 6820
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 4.156547822535594,
+      "learning_rate": 1.8793694675187197e-05,
+      "loss": 0.2635,
+      "step": 6840
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 4.057314806159278,
+      "learning_rate": 1.8784466185725902e-05,
+      "loss": 0.2598,
+      "step": 6860
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 8.314874066232434,
+      "learning_rate": 1.877520481458736e-05,
+      "loss": 0.2706,
+      "step": 6880
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 3.7767818313058195,
+      "learning_rate": 1.8765910596442316e-05,
+      "loss": 0.2483,
+      "step": 6900
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 3.53905010431742,
+      "learning_rate": 1.8756583566084502e-05,
+      "loss": 0.2683,
+      "step": 6920
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 3.2053331160944625,
+      "learning_rate": 1.8747223758430467e-05,
+      "loss": 0.2567,
+      "step": 6940
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 3.655977418308093,
+      "learning_rate": 1.8737831208519478e-05,
+      "loss": 0.2433,
+      "step": 6960
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 3.334646935711793,
+      "learning_rate": 1.8728405951513376e-05,
+      "loss": 0.2442,
+      "step": 6980
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 3.3678493336120505,
+      "learning_rate": 1.8718948022696427e-05,
+      "loss": 0.2634,
+      "step": 7000
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 2.9941614875638956,
+      "learning_rate": 1.8709457457475224e-05,
+      "loss": 0.2719,
+      "step": 7020
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 3.3212764283242495,
+      "learning_rate": 1.8699934291378534e-05,
+      "loss": 0.2616,
+      "step": 7040
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 3.7449113015848963,
+      "learning_rate": 1.8690378560057165e-05,
+      "loss": 0.251,
+      "step": 7060
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 2.588981565454773,
+      "learning_rate": 1.868079029928383e-05,
+      "loss": 0.2345,
+      "step": 7080
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 4.188807496871825,
+      "learning_rate": 1.8671169544953033e-05,
+      "loss": 0.2537,
+      "step": 7100
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 4.66518430781191,
+      "learning_rate": 1.866151633308091e-05,
+      "loss": 0.2397,
+      "step": 7120
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 4.197676860816234,
+      "learning_rate": 1.8651830699805108e-05,
+      "loss": 0.2446,
+      "step": 7140
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 3.6434672643458543,
+      "learning_rate": 1.8642112681384645e-05,
+      "loss": 0.2402,
+      "step": 7160
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 3.950387723218738,
+      "learning_rate": 1.8632362314199778e-05,
+      "loss": 0.2536,
+      "step": 7180
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 9.658211616844094,
+      "learning_rate": 1.8622579634751864e-05,
+      "loss": 0.2541,
+      "step": 7200
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 3.799337836345952,
+      "learning_rate": 1.8612764679663222e-05,
+      "loss": 0.2586,
+      "step": 7220
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 3.737906856782671,
+      "learning_rate": 1.8602917485677e-05,
+      "loss": 0.2539,
+      "step": 7240
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 3.2570157254536016,
+      "learning_rate": 1.859303808965703e-05,
+      "loss": 0.2533,
+      "step": 7260
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 2.8736525985585226,
+      "learning_rate": 1.85831265285877e-05,
+      "loss": 0.2532,
+      "step": 7280
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 3.5447389329516503,
+      "learning_rate": 1.857318283957381e-05,
+      "loss": 0.2387,
+      "step": 7300
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 2.7468496831393403,
+      "learning_rate": 1.8563207059840436e-05,
+      "loss": 0.2455,
+      "step": 7320
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 3.483830293921577,
+      "learning_rate": 1.8553199226732784e-05,
+      "loss": 0.2456,
+      "step": 7340
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 3.3132796899404635,
+      "learning_rate": 1.8543159377716054e-05,
+      "loss": 0.2564,
+      "step": 7360
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 3.3795508326649575,
+      "learning_rate": 1.8533087550375302e-05,
+      "loss": 0.2527,
+      "step": 7380
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 4.001941448174156,
+      "learning_rate": 1.8522983782415306e-05,
+      "loss": 0.2433,
+      "step": 7400
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 3.4205382361927117,
+      "learning_rate": 1.85128481116604e-05,
+      "loss": 0.2538,
+      "step": 7420
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 3.896330422632185,
+      "learning_rate": 1.8502680576054365e-05,
+      "loss": 0.2478,
+      "step": 7440
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 5.020228094868969,
+      "learning_rate": 1.849248121366026e-05,
+      "loss": 0.2427,
+      "step": 7460
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 3.574276782956127,
+      "learning_rate": 1.8482250062660293e-05,
+      "loss": 0.2389,
+      "step": 7480
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 3.5309579103911184,
+      "learning_rate": 1.847198716135568e-05,
+      "loss": 0.2567,
+      "step": 7500
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 3.1760414458004,
+      "learning_rate": 1.846169254816649e-05,
+      "loss": 0.2291,
+      "step": 7520
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 4.214334621561301,
+      "learning_rate": 1.8451366261631514e-05,
+      "loss": 0.2624,
+      "step": 7540
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 2.977221959069712,
+      "learning_rate": 1.8441008340408115e-05,
+      "loss": 0.2425,
+      "step": 7560
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 3.78383188925913,
+      "learning_rate": 1.8430618823272076e-05,
+      "loss": 0.2437,
+      "step": 7580
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 3.2432934448936224,
+      "learning_rate": 1.842019774911747e-05,
+      "loss": 0.2628,
+      "step": 7600
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 3.05382414419798,
+      "learning_rate": 1.8409745156956503e-05,
+      "loss": 0.2332,
+      "step": 7620
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 3.098744070302099,
+      "learning_rate": 1.839926108591937e-05,
+      "loss": 0.2359,
+      "step": 7640
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 3.427098715766002,
+      "learning_rate": 1.8388745575254117e-05,
+      "loss": 0.2429,
+      "step": 7660
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 5.630711008711337,
+      "learning_rate": 1.8378198664326474e-05,
+      "loss": 0.2416,
+      "step": 7680
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 3.1144723181358622,
+      "learning_rate": 1.836762039261974e-05,
+      "loss": 0.2406,
+      "step": 7700
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 4.070301722184627,
+      "learning_rate": 1.8357010799734588e-05,
+      "loss": 0.234,
+      "step": 7720
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 3.255428182680105,
+      "learning_rate": 1.834636992538898e-05,
+      "loss": 0.2365,
+      "step": 7740
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 3.4823775104086576,
+      "learning_rate": 1.8335697809417952e-05,
+      "loss": 0.2376,
+      "step": 7760
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 5.528269163918348,
+      "learning_rate": 1.8324994491773507e-05,
+      "loss": 0.2679,
+      "step": 7780
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 2.7221100312640467,
+      "learning_rate": 1.8314260012524462e-05,
+      "loss": 0.2544,
+      "step": 7800
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 3.224515621271091,
+      "learning_rate": 1.8303494411856278e-05,
+      "loss": 0.2586,
+      "step": 7820
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 2.798714485709818,
+      "learning_rate": 1.829269773007092e-05,
+      "loss": 0.2277,
+      "step": 7840
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 3.890849401904209,
+      "learning_rate": 1.8281870007586735e-05,
+      "loss": 0.2565,
+      "step": 7860
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 3.701194549732654,
+      "learning_rate": 1.8271011284938233e-05,
+      "loss": 0.2307,
+      "step": 7880
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 3.9057568048025875,
+      "learning_rate": 1.8260121602776003e-05,
+      "loss": 0.2464,
+      "step": 7900
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 3.397196786583742,
+      "learning_rate": 1.8249201001866533e-05,
+      "loss": 0.25,
+      "step": 7920
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 3.6136942859367496,
+      "learning_rate": 1.8238249523092046e-05,
+      "loss": 0.2398,
+      "step": 7940
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 3.6009293204428765,
+      "learning_rate": 1.8227267207450363e-05,
+      "loss": 0.2356,
+      "step": 7960
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 3.1462270393488585,
+      "learning_rate": 1.8216254096054747e-05,
+      "loss": 0.2292,
+      "step": 7980
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 3.453766085807792,
+      "learning_rate": 1.8205210230133756e-05,
+      "loss": 0.2532,
+      "step": 8000
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 4.718419953986993,
+      "learning_rate": 1.8194135651031063e-05,
+      "loss": 0.2379,
+      "step": 8020
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 4.074122309800583,
+      "learning_rate": 1.8183030400205335e-05,
+      "loss": 0.247,
+      "step": 8040
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 3.597629476607082,
+      "learning_rate": 1.817189451923005e-05,
+      "loss": 0.2563,
+      "step": 8060
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 3.2550443969136436,
+      "learning_rate": 1.8160728049793357e-05,
+      "loss": 0.2425,
+      "step": 8080
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 2.6027515890006505,
+      "learning_rate": 1.8149531033697913e-05,
+      "loss": 0.2329,
+      "step": 8100
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 3.705349797095471,
+      "learning_rate": 1.8138303512860737e-05,
+      "loss": 0.2521,
+      "step": 8120
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 3.510055266811937,
+      "learning_rate": 1.812704552931303e-05,
+      "loss": 0.2413,
+      "step": 8140
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 3.2863076604207997,
+      "learning_rate": 1.8115757125200054e-05,
+      "loss": 0.2471,
+      "step": 8160
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 3.174018473859266,
+      "learning_rate": 1.810443834278094e-05,
+      "loss": 0.2429,
+      "step": 8180
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 4.78684161965386,
+      "learning_rate": 1.8093089224428533e-05,
+      "loss": 0.21,
+      "step": 8200
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 3.778736256263061,
+      "learning_rate": 1.8081709812629265e-05,
+      "loss": 0.2438,
+      "step": 8220
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 2.011964890319697,
+      "learning_rate": 1.807030014998296e-05,
+      "loss": 0.2235,
+      "step": 8240
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 2.469231474076661,
+      "learning_rate": 1.8058860279202698e-05,
+      "loss": 0.2361,
+      "step": 8260
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 4.194473608952758,
+      "learning_rate": 1.804739024311463e-05,
+      "loss": 0.2485,
+      "step": 8280
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 3.648802255045255,
+      "learning_rate": 1.8035890084657854e-05,
+      "loss": 0.2448,
+      "step": 8300
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 2.8256525011924443,
+      "learning_rate": 1.8024359846884213e-05,
+      "loss": 0.2688,
+      "step": 8320
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 4.409589867929299,
+      "learning_rate": 1.8012799572958173e-05,
+      "loss": 0.2405,
+      "step": 8340
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 5.024619525530542,
+      "learning_rate": 1.8001209306156634e-05,
+      "loss": 0.2284,
+      "step": 8360
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 3.5087106485202026,
+      "learning_rate": 1.7989589089868774e-05,
+      "loss": 0.2298,
+      "step": 8380
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 4.40165070000877,
+      "learning_rate": 1.7977938967595897e-05,
+      "loss": 0.2507,
+      "step": 8400
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 3.9660729217673705,
+      "learning_rate": 1.7966258982951255e-05,
+      "loss": 0.2459,
+      "step": 8420
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 4.665706172372416,
+      "learning_rate": 1.7954549179659903e-05,
+      "loss": 0.2458,
+      "step": 8440
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 3.2908069759526333,
+      "learning_rate": 1.794280960155851e-05,
+      "loss": 0.2251,
+      "step": 8460
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 6.7504752662905725,
+      "learning_rate": 1.7931040292595227e-05,
+      "loss": 0.254,
+      "step": 8480
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 2.2047073417435787,
+      "learning_rate": 1.791924129682949e-05,
+      "loss": 0.2456,
+      "step": 8500
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 3.9048911210632236,
+      "learning_rate": 1.7907412658431877e-05,
+      "loss": 0.2307,
+      "step": 8520
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 3.33290909610624,
+      "learning_rate": 1.7895554421683938e-05,
+      "loss": 0.2514,
+      "step": 8540
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 3.7515500997507605,
+      "learning_rate": 1.7883666630978024e-05,
+      "loss": 0.2189,
+      "step": 8560
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 2.495653761409699,
+      "learning_rate": 1.7871749330817114e-05,
+      "loss": 0.216,
+      "step": 8580
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 3.1012959017387556,
+      "learning_rate": 1.785980256581468e-05,
+      "loss": 0.2183,
+      "step": 8600
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 2.7851575340492745,
+      "learning_rate": 1.7847826380694482e-05,
+      "loss": 0.2086,
+      "step": 8620
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 5.673400169424141,
+      "learning_rate": 1.783582082029042e-05,
+      "loss": 0.2216,
+      "step": 8640
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 3.9347827557672113,
+      "learning_rate": 1.7823785929546363e-05,
+      "loss": 0.2351,
+      "step": 8660
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 4.263640336790709,
+      "learning_rate": 1.781172175351599e-05,
+      "loss": 0.2237,
+      "step": 8680
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 2.8181611623874034,
+      "learning_rate": 1.779962833736259e-05,
+      "loss": 0.2078,
+      "step": 8700
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 3.826255497546828,
+      "learning_rate": 1.7787505726358937e-05,
+      "loss": 0.2224,
+      "step": 8720
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 3.2239814391820603,
+      "learning_rate": 1.777535396588709e-05,
+      "loss": 0.2196,
+      "step": 8740
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 3.4517898394902753,
+      "learning_rate": 1.7763173101438232e-05,
+      "loss": 0.2056,
+      "step": 8760
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 4.604513417070624,
+      "learning_rate": 1.7750963178612498e-05,
+      "loss": 0.2434,
+      "step": 8780
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 3.835720949606093,
+      "learning_rate": 1.7738724243118807e-05,
+      "loss": 0.2282,
+      "step": 8800
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 3.437975625165332,
+      "learning_rate": 1.7726456340774684e-05,
+      "loss": 0.25,
+      "step": 8820
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 3.362299298390976,
+      "learning_rate": 1.771415951750611e-05,
+      "loss": 0.2301,
+      "step": 8840
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 3.469988077896261,
+      "learning_rate": 1.7701833819347317e-05,
+      "loss": 0.2268,
+      "step": 8860
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 3.3257530965726834,
+      "learning_rate": 1.7689479292440635e-05,
+      "loss": 0.2362,
+      "step": 8880
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 4.357144130197379,
+      "learning_rate": 1.767709598303633e-05,
+      "loss": 0.2179,
+      "step": 8900
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 2.833662313173985,
+      "learning_rate": 1.7664683937492394e-05,
+      "loss": 0.2082,
+      "step": 8920
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 3.6248674039596347,
+      "learning_rate": 1.7652243202274427e-05,
+      "loss": 0.2399,
+      "step": 8940
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 3.374713885577454,
+      "learning_rate": 1.7639773823955405e-05,
+      "loss": 0.2303,
+      "step": 8960
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 4.623143210009699,
+      "learning_rate": 1.7627275849215543e-05,
+      "loss": 0.2197,
+      "step": 8980
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 2.7733450860758713,
+      "learning_rate": 1.7614749324842106e-05,
+      "loss": 0.2419,
+      "step": 9000
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 4.136130142481253,
+      "learning_rate": 1.7602194297729238e-05,
+      "loss": 0.2315,
+      "step": 9020
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 2.5713300459015476,
+      "learning_rate": 1.758961081487779e-05,
+      "loss": 0.2285,
+      "step": 9040
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 2.90938161486602,
+      "learning_rate": 1.7576998923395128e-05,
+      "loss": 0.2357,
+      "step": 9060
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 3.2699691862001896,
+      "learning_rate": 1.756435867049498e-05,
+      "loss": 0.2161,
+      "step": 9080
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 3.353066842604925,
+      "learning_rate": 1.7551690103497248e-05,
+      "loss": 0.2379,
+      "step": 9100
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 3.1865972100619815,
+      "learning_rate": 1.7538993269827817e-05,
+      "loss": 0.2324,
+      "step": 9120
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 4.115179219937697,
+      "learning_rate": 1.7526268217018404e-05,
+      "loss": 0.2165,
+      "step": 9140
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 3.0706453021663553,
+      "learning_rate": 1.751351499270636e-05,
+      "loss": 0.2128,
+      "step": 9160
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 3.0307089725880774,
+      "learning_rate": 1.7500733644634503e-05,
+      "loss": 0.2153,
+      "step": 9180
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 3.0414096427563435,
+      "learning_rate": 1.748792422065093e-05,
+      "loss": 0.2159,
+      "step": 9200
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 3.581867909009255,
+      "learning_rate": 1.7475086768708844e-05,
+      "loss": 0.2108,
+      "step": 9220
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 3.3902710057413707,
+      "learning_rate": 1.7462221336866373e-05,
+      "loss": 0.2171,
+      "step": 9240
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 4.186442227351193,
+      "learning_rate": 1.74493279732864e-05,
+      "loss": 0.213,
+      "step": 9260
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 2.111492196482593,
+      "learning_rate": 1.743640672623636e-05,
+      "loss": 0.231,
+      "step": 9280
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 3.3151227536495744,
+      "learning_rate": 1.7423457644088067e-05,
+      "loss": 0.2255,
+      "step": 9300
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 2.979480504575997,
+      "learning_rate": 1.741048077531756e-05,
+      "loss": 0.2193,
+      "step": 9320
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 8.144688622580304,
+      "learning_rate": 1.7397476168504877e-05,
+      "loss": 0.2213,
+      "step": 9340
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 2.3573272104543395,
+      "learning_rate": 1.738444387233392e-05,
+      "loss": 0.23,
+      "step": 9360
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 5.626318713423072,
+      "learning_rate": 1.737138393559222e-05,
+      "loss": 0.2231,
+      "step": 9380
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 3.659177993436024,
+      "learning_rate": 1.7358296407170803e-05,
+      "loss": 0.2201,
+      "step": 9400
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 4.1924943889109505,
+      "learning_rate": 1.734518133606398e-05,
+      "loss": 0.2211,
+      "step": 9420
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 3.1828991465696626,
+      "learning_rate": 1.7332038771369178e-05,
+      "loss": 0.2178,
+      "step": 9440
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 3.210581516107296,
+      "learning_rate": 1.7318868762286735e-05,
+      "loss": 0.2252,
+      "step": 9460
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 3.1129128729970272,
+      "learning_rate": 1.7305671358119743e-05,
+      "loss": 0.2218,
+      "step": 9480
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 3.041995792784782,
+      "learning_rate": 1.7292446608273838e-05,
+      "loss": 0.2089,
+      "step": 9500
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 4.604556704379658,
+      "learning_rate": 1.7279194562257034e-05,
+      "loss": 0.2104,
+      "step": 9520
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 2.4749100524278944,
+      "learning_rate": 1.7265915269679533e-05,
+      "loss": 0.2077,
+      "step": 9540
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 3.256783082638953,
+      "learning_rate": 1.7252608780253527e-05,
+      "loss": 0.2222,
+      "step": 9560
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 3.250990203251271,
+      "learning_rate": 1.723927514379304e-05,
+      "loss": 0.238,
+      "step": 9580
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 4.17869256557804,
+      "learning_rate": 1.7225914410213695e-05,
+      "loss": 0.2264,
+      "step": 9600
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 3.001794596192333,
+      "learning_rate": 1.7212526629532585e-05,
+      "loss": 0.2156,
+      "step": 9620
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 4.51471973043563,
+      "learning_rate": 1.7199111851868038e-05,
+      "loss": 0.2212,
+      "step": 9640
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 3.343622223025876,
+      "learning_rate": 1.7185670127439453e-05,
+      "loss": 0.2292,
+      "step": 9660
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 5.351871498264467,
+      "learning_rate": 1.7172201506567103e-05,
+      "loss": 0.233,
+      "step": 9680
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 2.480488358067184,
+      "learning_rate": 1.7158706039671962e-05,
+      "loss": 0.2253,
+      "step": 9700
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 9.126953333979776,
+      "learning_rate": 1.7145183777275484e-05,
+      "loss": 0.2216,
+      "step": 9720
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 3.0992364496831994,
+      "learning_rate": 1.7131634769999458e-05,
+      "loss": 0.2339,
+      "step": 9740
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 3.536084536918467,
+      "learning_rate": 1.7118059068565775e-05,
+      "loss": 0.2244,
+      "step": 9760
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 2.48499093727593,
+      "learning_rate": 1.710445672379627e-05,
+      "loss": 0.221,
+      "step": 9780
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 3.7310914483561097,
+      "learning_rate": 1.7090827786612515e-05,
+      "loss": 0.2356,
+      "step": 9800
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 3.0512655853035975,
+      "learning_rate": 1.7077172308035627e-05,
+      "loss": 0.2157,
+      "step": 9820
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 2.6139447976363868,
+      "learning_rate": 1.70634903391861e-05,
+      "loss": 0.2251,
+      "step": 9840
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 2.7800692505753135,
+      "learning_rate": 1.7049781931283576e-05,
+      "loss": 0.2385,
+      "step": 9860
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 3.8229638009279614,
+      "learning_rate": 1.7036047135646695e-05,
+      "loss": 0.2163,
+      "step": 9880
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 3.4670030473465205,
+      "learning_rate": 1.7022286003692863e-05,
+      "loss": 0.2172,
+      "step": 9900
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 3.1067639606886908,
+      "learning_rate": 1.700849858693809e-05,
+      "loss": 0.228,
+      "step": 9920
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 2.775258672531786,
+      "learning_rate": 1.6994684936996782e-05,
+      "loss": 0.2174,
+      "step": 9940
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 3.0648920487627045,
+      "learning_rate": 1.698084510558155e-05,
+      "loss": 0.2148,
+      "step": 9960
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 3.2600668007495064,
+      "learning_rate": 1.696697914450302e-05,
+      "loss": 0.2133,
+      "step": 9980
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 5.51464299005287,
+      "learning_rate": 1.6953087105669627e-05,
+      "loss": 0.2297,
+      "step": 10000
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 3.872665563483291,
+      "learning_rate": 1.693916904108745e-05,
+      "loss": 0.2277,
+      "step": 10020
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 3.0527595230989917,
+      "learning_rate": 1.692522500285998e-05,
+      "loss": 0.2084,
+      "step": 10040
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 3.009517988727613,
+      "learning_rate": 1.6911255043187952e-05,
+      "loss": 0.2105,
+      "step": 10060
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 4.461421265567992,
+      "learning_rate": 1.6897259214369137e-05,
+      "loss": 0.2156,
+      "step": 10080
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 3.2651682265027757,
+      "learning_rate": 1.6883237568798148e-05,
+      "loss": 0.2223,
+      "step": 10100
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 4.560070279232337,
+      "learning_rate": 1.6869190158966248e-05,
+      "loss": 0.2032,
+      "step": 10120
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 2.17671049536392,
+      "learning_rate": 1.6855117037461147e-05,
+      "loss": 0.2086,
+      "step": 10140
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 2.835463807449662,
+      "learning_rate": 1.6841018256966817e-05,
+      "loss": 0.2285,
+      "step": 10160
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 2.9456662236305045,
+      "learning_rate": 1.682689387026328e-05,
+      "loss": 0.2288,
+      "step": 10180
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 2.6149578560812046,
+      "learning_rate": 1.681274393022641e-05,
+      "loss": 0.2196,
+      "step": 10200
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 2.9578551501440025,
+      "learning_rate": 1.6798568489827764e-05,
+      "loss": 0.2233,
+      "step": 10220
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 3.191017752898972,
+      "learning_rate": 1.6784367602134337e-05,
+      "loss": 0.2093,
+      "step": 10240
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 3.353192695439166,
+      "learning_rate": 1.6770141320308405e-05,
+      "loss": 0.233,
+      "step": 10260
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 3.8431182590288397,
+      "learning_rate": 1.6755889697607306e-05,
+      "loss": 0.2206,
+      "step": 10280
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 3.479325951377326,
+      "learning_rate": 1.6741612787383236e-05,
+      "loss": 0.2205,
+      "step": 10300
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 2.5880779155399316,
+      "learning_rate": 1.6727310643083066e-05,
+      "loss": 0.2103,
+      "step": 10320
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 7.419706714519934,
+      "learning_rate": 1.671298331824813e-05,
+      "loss": 0.207,
+      "step": 10340
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 4.622811908173908,
+      "learning_rate": 1.6698630866514022e-05,
+      "loss": 0.2205,
+      "step": 10360
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 4.038977026265583,
+      "learning_rate": 1.6684253341610414e-05,
+      "loss": 0.2046,
+      "step": 10380
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 3.2506861329229912,
+      "learning_rate": 1.6669850797360825e-05,
+      "loss": 0.2161,
+      "step": 10400
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 3.502316934183813,
+      "learning_rate": 1.665542328768245e-05,
+      "loss": 0.2202,
+      "step": 10420
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 2.774170254901442,
+      "learning_rate": 1.6640970866585937e-05,
+      "loss": 0.2313,
+      "step": 10440
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 3.3567765943400345,
+      "learning_rate": 1.6626493588175194e-05,
+      "loss": 0.2118,
+      "step": 10460
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 3.147468966936314,
+      "learning_rate": 1.6611991506647186e-05,
+      "loss": 0.2015,
+      "step": 10480
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 2.794186200047937,
+      "learning_rate": 1.659746467629173e-05,
+      "loss": 0.2267,
+      "step": 10500
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 3.208243100437357,
+      "learning_rate": 1.6582913151491286e-05,
+      "loss": 0.2272,
+      "step": 10520
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 2.8524752057641996,
+      "learning_rate": 1.6568336986720773e-05,
+      "loss": 0.2169,
+      "step": 10540
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 3.3114099688426513,
+      "learning_rate": 1.6553736236547336e-05,
+      "loss": 0.2161,
+      "step": 10560
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 3.763983657663605,
+      "learning_rate": 1.653911095563017e-05,
+      "loss": 0.2252,
+      "step": 10580
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 3.266499087518214,
+      "learning_rate": 1.6524461198720303e-05,
+      "loss": 0.1967,
+      "step": 10600
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 3.9946972745895017,
+      "learning_rate": 1.650978702066038e-05,
+      "loss": 0.2174,
+      "step": 10620
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 2.8854543268016055,
+      "learning_rate": 1.649508847638448e-05,
+      "loss": 0.2136,
+      "step": 10640
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 3.5610681969299653,
+      "learning_rate": 1.648036562091789e-05,
+      "loss": 0.225,
+      "step": 10660
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 8.713288377423561,
+      "learning_rate": 1.6465618509376923e-05,
+      "loss": 0.2172,
+      "step": 10680
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 3.2387951119923932,
+      "learning_rate": 1.6450847196968676e-05,
+      "loss": 0.2131,
+      "step": 10700
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 3.8639660913863234,
+      "learning_rate": 1.643605173899086e-05,
+      "loss": 0.2126,
+      "step": 10720
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 4.013150770232071,
+      "learning_rate": 1.642123219083156e-05,
+      "loss": 0.2253,
+      "step": 10740
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 3.448212871140601,
+      "learning_rate": 1.640638860796908e-05,
+      "loss": 0.2122,
+      "step": 10760
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 4.008865307397649,
+      "learning_rate": 1.6391521045971654e-05,
+      "loss": 0.2198,
+      "step": 10780
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 2.668310542854165,
+      "learning_rate": 1.6376629560497318e-05,
+      "loss": 0.2303,
+      "step": 10800
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 4.253939653866208,
+      "learning_rate": 1.636171420729365e-05,
+      "loss": 0.2165,
+      "step": 10820
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 3.662446989645766,
+      "learning_rate": 1.634677504219759e-05,
+      "loss": 0.2093,
+      "step": 10840
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 3.7845535349608115,
+      "learning_rate": 1.6331812121135216e-05,
+      "loss": 0.1997,
+      "step": 10860
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 4.21364980338036,
+      "learning_rate": 1.6316825500121528e-05,
+      "loss": 0.2186,
+      "step": 10880
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 3.4198924553007153,
+      "learning_rate": 1.630181523526027e-05,
+      "loss": 0.2303,
+      "step": 10900
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 3.569247097261353,
+      "learning_rate": 1.6286781382743678e-05,
+      "loss": 0.2084,
+      "step": 10920
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 3.2057387689879966,
+      "learning_rate": 1.6271723998852305e-05,
+      "loss": 0.2223,
+      "step": 10940
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 3.438084084566052,
+      "learning_rate": 1.6256643139954787e-05,
+      "loss": 0.2087,
+      "step": 10960
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 3.1629151324387075,
+      "learning_rate": 1.6241538862507647e-05,
+      "loss": 0.2225,
+      "step": 10980
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 3.1696025055249097,
+      "learning_rate": 1.622641122305507e-05,
+      "loss": 0.2315,
+      "step": 11000
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 3.053371916892243,
+      "learning_rate": 1.621126027822871e-05,
+      "loss": 0.2116,
+      "step": 11020
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 2.9543774083504766,
+      "learning_rate": 1.619608608474745e-05,
+      "loss": 0.2254,
+      "step": 11040
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 3.3693833415742453,
+      "learning_rate": 1.618088869941723e-05,
+      "loss": 0.2169,
+      "step": 11060
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 3.9995183654737376,
+      "learning_rate": 1.6165668179130784e-05,
+      "loss": 0.2365,
+      "step": 11080
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 13.880528903525311,
+      "learning_rate": 1.6150424580867476e-05,
+      "loss": 0.2333,
+      "step": 11100
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 3.1836707357447778,
+      "learning_rate": 1.613515796169305e-05,
+      "loss": 0.2237,
+      "step": 11120
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 3.8509352600789506,
+      "learning_rate": 1.6119868378759436e-05,
+      "loss": 0.2053,
+      "step": 11140
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 3.2583341112322906,
+      "learning_rate": 1.610455588930453e-05,
+      "loss": 0.2139,
+      "step": 11160
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 5.386703878469319,
+      "learning_rate": 1.6089220550651985e-05,
+      "loss": 0.2104,
+      "step": 11180
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 3.1002668081338114,
+      "learning_rate": 1.607386242021099e-05,
+      "loss": 0.2048,
+      "step": 11200
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 4.67566499111419,
+      "learning_rate": 1.605848155547605e-05,
+      "loss": 0.2182,
+      "step": 11220
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 1.8252132735330482,
+      "learning_rate": 1.6043078014026786e-05,
+      "loss": 0.2211,
+      "step": 11240
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 3.5104873755754635,
+      "learning_rate": 1.6027651853527706e-05,
+      "loss": 0.2082,
+      "step": 11260
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 2.596015767829866,
+      "learning_rate": 1.6012203131728e-05,
+      "loss": 0.2053,
+      "step": 11280
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 3.9320961337285247,
+      "learning_rate": 1.5996731906461314e-05,
+      "loss": 0.2124,
+      "step": 11300
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 3.239895886290778,
+      "learning_rate": 1.598123823564554e-05,
+      "loss": 0.2126,
+      "step": 11320
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 7.536836064136331,
+      "learning_rate": 1.5965722177282595e-05,
+      "loss": 0.2199,
+      "step": 11340
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 5.562084911160559,
+      "learning_rate": 1.5950183789458206e-05,
+      "loss": 0.2246,
+      "step": 11360
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 2.742122377671127,
+      "learning_rate": 1.5934623130341695e-05,
+      "loss": 0.2178,
+      "step": 11380
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 4.143294024804004,
+      "learning_rate": 1.5919040258185753e-05,
+      "loss": 0.2189,
+      "step": 11400
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 3.6948127005341975,
+      "learning_rate": 1.5903435231326233e-05,
+      "loss": 0.2311,
+      "step": 11420
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 2.8456773200400005,
+      "learning_rate": 1.5887808108181922e-05,
+      "loss": 0.2047,
+      "step": 11440
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 4.524975240228682,
+      "learning_rate": 1.5872158947254328e-05,
+      "loss": 0.216,
+      "step": 11460
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 3.22887435738582,
+      "learning_rate": 1.585648780712746e-05,
+      "loss": 0.2139,
+      "step": 11480
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 2.40402306888702,
+      "learning_rate": 1.5840794746467615e-05,
+      "loss": 0.2066,
+      "step": 11500
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 3.4971814386926967,
+      "learning_rate": 1.5825079824023132e-05,
+      "loss": 0.2131,
+      "step": 11520
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 3.210588645083174,
+      "learning_rate": 1.5809343098624212e-05,
+      "loss": 0.2114,
+      "step": 11540
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 2.632446897574305,
+      "learning_rate": 1.5793584629182667e-05,
+      "loss": 0.2151,
+      "step": 11560
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 4.5614133612079755,
+      "learning_rate": 1.577780447469171e-05,
+      "loss": 0.2151,
+      "step": 11580
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 2.394202553621421,
+      "learning_rate": 1.5762002694225735e-05,
+      "loss": 0.2079,
+      "step": 11600
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 2.6251453177782604,
+      "learning_rate": 1.5746179346940102e-05,
+      "loss": 0.2159,
+      "step": 11620
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 3.3437649735445256,
+      "learning_rate": 1.5730334492070893e-05,
+      "loss": 0.22,
+      "step": 11640
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 2.63771302867986,
+      "learning_rate": 1.5714468188934716e-05,
+      "loss": 0.2136,
+      "step": 11660
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 2.655161017296502,
+      "learning_rate": 1.5698580496928475e-05,
+      "loss": 0.2067,
+      "step": 11680
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 2.948812575507089,
+      "learning_rate": 1.5682671475529137e-05,
+      "loss": 0.2008,
+      "step": 11700
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 3.469779471829569,
+      "learning_rate": 1.566674118429352e-05,
+      "loss": 0.2228,
+      "step": 11720
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 3.2002593829411747,
+      "learning_rate": 1.5650789682858073e-05,
+      "loss": 0.2149,
+      "step": 11740
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 3.0178239938264864,
+      "learning_rate": 1.563481703093864e-05,
+      "loss": 0.2173,
+      "step": 11760
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 3.359858420431831,
+      "learning_rate": 1.5618823288330245e-05,
+      "loss": 0.2109,
+      "step": 11780
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 4.244464186067146,
+      "learning_rate": 1.5602808514906874e-05,
+      "loss": 0.2172,
+      "step": 11800
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 3.02678280463322,
+      "learning_rate": 1.5586772770621236e-05,
+      "loss": 0.2055,
+      "step": 11820
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 3.2941320498135602,
+      "learning_rate": 1.5570716115504544e-05,
+      "loss": 0.2212,
+      "step": 11840
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 3.2434629574232474,
+      "learning_rate": 1.5554638609666298e-05,
+      "loss": 0.2183,
+      "step": 11860
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 2.7694471424497342,
+      "learning_rate": 1.5538540313294058e-05,
+      "loss": 0.2254,
+      "step": 11880
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 3.5269104616476983,
+      "learning_rate": 1.5522421286653208e-05,
+      "loss": 0.2201,
+      "step": 11900
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 3.8299784007845363,
+      "learning_rate": 1.550628159008674e-05,
+      "loss": 0.2133,
+      "step": 11920
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 5.026447159875988,
+      "learning_rate": 1.5490121284015026e-05,
+      "loss": 0.2177,
+      "step": 11940
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 3.55665309344499,
+      "learning_rate": 1.5473940428935594e-05,
+      "loss": 0.2269,
+      "step": 11960
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 3.8149873717812732,
+      "learning_rate": 1.545773908542289e-05,
+      "loss": 0.2135,
+      "step": 11980
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 4.453465498906573,
+      "learning_rate": 1.5441517314128075e-05,
+      "loss": 0.2066,
+      "step": 12000
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 3.4558864961105167,
+      "learning_rate": 1.5425275175778765e-05,
+      "loss": 0.2221,
+      "step": 12020
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 3.0249571615921726,
+      "learning_rate": 1.5409012731178844e-05,
+      "loss": 0.2176,
+      "step": 12040
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 3.378968625237132,
+      "learning_rate": 1.539273004120819e-05,
+      "loss": 0.2129,
+      "step": 12060
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 3.599643022004301,
+      "learning_rate": 1.5376427166822493e-05,
+      "loss": 0.2005,
+      "step": 12080
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 4.369469771352546,
+      "learning_rate": 1.5360104169052983e-05,
+      "loss": 0.2072,
+      "step": 12100
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 2.9804449117964618,
+      "learning_rate": 1.534376110900625e-05,
+      "loss": 0.2235,
+      "step": 12120
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 3.477739662012998,
+      "learning_rate": 1.532739804786397e-05,
+      "loss": 0.2044,
+      "step": 12140
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 4.365666269531915,
+      "learning_rate": 1.5311015046882695e-05,
+      "loss": 0.2068,
+      "step": 12160
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 3.097981651118829,
+      "learning_rate": 1.5294612167393633e-05,
+      "loss": 0.2034,
+      "step": 12180
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 3.1761710102538485,
+      "learning_rate": 1.5278189470802402e-05,
+      "loss": 0.2148,
+      "step": 12200
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 6.183507999444912,
+      "learning_rate": 1.526174701858881e-05,
+      "loss": 0.2252,
+      "step": 12220
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 3.794441982049424,
+      "learning_rate": 1.5245284872306627e-05,
+      "loss": 0.2194,
+      "step": 12240
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 3.061187462874714,
+      "learning_rate": 1.5228803093583333e-05,
+      "loss": 0.2255,
+      "step": 12260
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 3.26228124827874,
+      "learning_rate": 1.5212301744119922e-05,
+      "loss": 0.2297,
+      "step": 12280
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 2.8942060403720355,
+      "learning_rate": 1.5195780885690641e-05,
+      "loss": 0.2013,
+      "step": 12300
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 2.6600186708698534,
+      "learning_rate": 1.5179240580142782e-05,
+      "loss": 0.2047,
+      "step": 12320
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 3.969215305715658,
+      "learning_rate": 1.5162680889396424e-05,
+      "loss": 0.2173,
+      "step": 12340
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 2.6829571142305904,
+      "learning_rate": 1.5146101875444226e-05,
+      "loss": 0.2023,
+      "step": 12360
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 3.6657811020233684,
+      "learning_rate": 1.5129503600351186e-05,
+      "loss": 0.2147,
+      "step": 12380
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 3.90125498508867,
+      "learning_rate": 1.5112886126254405e-05,
+      "loss": 0.2137,
+      "step": 12400
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 3.147105349280156,
+      "learning_rate": 1.5096249515362853e-05,
+      "loss": 0.2118,
+      "step": 12420
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 2.7729387721425645,
+      "learning_rate": 1.5079593829957144e-05,
+      "loss": 0.2149,
+      "step": 12440
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 5.342410092487611,
+      "learning_rate": 1.5062919132389305e-05,
+      "loss": 0.2169,
+      "step": 12460
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 3.9559478933684638,
+      "learning_rate": 1.5046225485082526e-05,
+      "loss": 0.2205,
+      "step": 12480
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 2.726490183748406,
+      "learning_rate": 1.5029512950530943e-05,
+      "loss": 0.224,
+      "step": 12500
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 2.990487116043663,
+      "learning_rate": 1.5012781591299399e-05,
+      "loss": 0.2287,
+      "step": 12520
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 3.8406463746608215,
+      "learning_rate": 1.4996031470023205e-05,
+      "loss": 0.2198,
+      "step": 12540
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 3.29662607708646,
+      "learning_rate": 1.4979262649407914e-05,
+      "loss": 0.2198,
+      "step": 12560
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 2.7040482195837834,
+      "learning_rate": 1.496247519222908e-05,
+      "loss": 0.2079,
+      "step": 12580
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 2.6175719021573918,
+      "learning_rate": 1.4945669161332021e-05,
+      "loss": 0.2198,
+      "step": 12600
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 2.9074738909532285,
+      "learning_rate": 1.4928844619631588e-05,
+      "loss": 0.2084,
+      "step": 12620
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 3.1468612473612563,
+      "learning_rate": 1.4912001630111936e-05,
+      "loss": 0.2013,
+      "step": 12640
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 3.779361095424968,
+      "learning_rate": 1.4895140255826273e-05,
+      "loss": 0.1918,
+      "step": 12660
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 2.696019176777178,
+      "learning_rate": 1.4878260559896634e-05,
+      "loss": 0.2126,
+      "step": 12680
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 3.7345031971654103,
+      "learning_rate": 1.4861362605513643e-05,
+      "loss": 0.2212,
+      "step": 12700
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 4.465836435468654,
+      "learning_rate": 1.4844446455936277e-05,
+      "loss": 0.2211,
+      "step": 12720
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 2.9078487797789645,
+      "learning_rate": 1.4827512174491631e-05,
+      "loss": 0.2267,
+      "step": 12740
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 3.296252006498326,
+      "learning_rate": 1.4810559824574675e-05,
+      "loss": 0.2216,
+      "step": 12760
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 3.227367236112649,
+      "learning_rate": 1.4793589469648017e-05,
+      "loss": 0.2233,
+      "step": 12780
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 3.4962503238332774,
+      "learning_rate": 1.4776601173241671e-05,
+      "loss": 0.2269,
+      "step": 12800
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 2.878724712305482,
+      "learning_rate": 1.4759594998952821e-05,
+      "loss": 0.2119,
+      "step": 12820
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 3.682489644312493,
+      "learning_rate": 1.4742571010445572e-05,
+      "loss": 0.2011,
+      "step": 12840
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 5.266174287756561,
+      "learning_rate": 1.472552927145072e-05,
+      "loss": 0.2258,
+      "step": 12860
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 4.045996375421248,
+      "learning_rate": 1.4708469845765512e-05,
+      "loss": 0.1922,
+      "step": 12880
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 3.266689729284762,
+      "learning_rate": 1.4691392797253405e-05,
+      "loss": 0.2075,
+      "step": 12900
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 2.242467882655721,
+      "learning_rate": 1.4674298189843837e-05,
+      "loss": 0.212,
+      "step": 12920
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 2.807279276540352,
+      "learning_rate": 1.4657186087531962e-05,
+      "loss": 0.1924,
+      "step": 12940
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 6.9144252531074875,
+      "learning_rate": 1.4640056554378448e-05,
+      "loss": 0.224,
+      "step": 12960
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 3.5667764527961645,
+      "learning_rate": 1.4622909654509204e-05,
+      "loss": 0.2175,
+      "step": 12980
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 2.431697492432426,
+      "learning_rate": 1.4605745452115155e-05,
+      "loss": 0.2172,
+      "step": 13000
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 3.336010970081616,
+      "learning_rate": 1.4588564011451998e-05,
+      "loss": 0.2219,
+      "step": 13020
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 2.9985996792871656,
+      "learning_rate": 1.4571365396839969e-05,
+      "loss": 0.2226,
+      "step": 13040
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 2.7817912914440677,
+      "learning_rate": 1.4554149672663591e-05,
+      "loss": 0.2166,
+      "step": 13060
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 3.5654978937601007,
+      "learning_rate": 1.4536916903371443e-05,
+      "loss": 0.209,
+      "step": 13080
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 3.140872774741023,
+      "learning_rate": 1.4519667153475908e-05,
+      "loss": 0.2059,
+      "step": 13100
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 3.7735270929856304,
+      "learning_rate": 1.4502400487552937e-05,
+      "loss": 0.2078,
+      "step": 13120
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 3.3704008625810826,
+      "learning_rate": 1.4485116970241814e-05,
+      "loss": 0.2288,
+      "step": 13140
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 3.058405104235901,
+      "learning_rate": 1.4467816666244904e-05,
+      "loss": 0.2145,
+      "step": 13160
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 3.2628541717630295,
+      "learning_rate": 1.4450499640327418e-05,
+      "loss": 0.2096,
+      "step": 13180
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 2.8358576319353475,
+      "learning_rate": 1.443316595731716e-05,
+      "loss": 0.2162,
+      "step": 13200
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 2.130034709343158,
+      "learning_rate": 1.4415815682104295e-05,
+      "loss": 0.2038,
+      "step": 13220
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 3.679976984449645,
+      "learning_rate": 1.4398448879641102e-05,
+      "loss": 0.216,
+      "step": 13240
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 3.5668380826834647,
+      "learning_rate": 1.4381065614941733e-05,
+      "loss": 0.2262,
+      "step": 13260
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 3.0243831604612765,
+      "learning_rate": 1.4363665953081969e-05,
+      "loss": 0.201,
+      "step": 13280
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 3.2212853677485933,
+      "learning_rate": 1.4346249959198965e-05,
+      "loss": 0.2163,
+      "step": 13300
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 2.596122759445438,
+      "learning_rate": 1.4328817698491031e-05,
+      "loss": 0.22,
+      "step": 13320
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 3.0666724605781996,
+      "learning_rate": 1.4311369236217367e-05,
+      "loss": 0.2139,
+      "step": 13340
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 3.657252769223528,
+      "learning_rate": 1.4293904637697817e-05,
+      "loss": 0.215,
+      "step": 13360
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 3.4144546667640383,
+      "learning_rate": 1.4276423968312645e-05,
+      "loss": 0.227,
+      "step": 13380
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 3.2413250982644675,
+      "learning_rate": 1.4258927293502273e-05,
+      "loss": 0.2077,
+      "step": 13400
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 3.5063106320828488,
+      "learning_rate": 1.4241414678767039e-05,
+      "loss": 0.2047,
+      "step": 13420
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 2.1504143515238514,
+      "learning_rate": 1.4223886189666951e-05,
+      "loss": 0.1982,
+      "step": 13440
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 2.9525594825793005,
+      "learning_rate": 1.4206341891821456e-05,
+      "loss": 0.2066,
+      "step": 13460
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 5.213240221920547,
+      "learning_rate": 1.4188781850909165e-05,
+      "loss": 0.2246,
+      "step": 13480
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 4.305790200898134,
+      "learning_rate": 1.417120613266764e-05,
+      "loss": 0.1994,
+      "step": 13500
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 3.370920187857142,
+      "learning_rate": 1.4153614802893125e-05,
+      "loss": 0.2085,
+      "step": 13520
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 3.425048053710641,
+      "learning_rate": 1.4136007927440307e-05,
+      "loss": 0.207,
+      "step": 13540
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 3.24691919878633,
+      "learning_rate": 1.4118385572222072e-05,
+      "loss": 0.2194,
+      "step": 13560
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 2.8908834805780064,
+      "learning_rate": 1.4100747803209252e-05,
+      "loss": 0.2024,
+      "step": 13580
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 3.5087368773081473,
+      "learning_rate": 1.4083094686430387e-05,
+      "loss": 0.2102,
+      "step": 13600
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 2.5718331983652485,
+      "learning_rate": 1.4065426287971473e-05,
+      "loss": 0.2205,
+      "step": 13620
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 3.0529680412811375,
+      "learning_rate": 1.4047742673975706e-05,
+      "loss": 0.214,
+      "step": 13640
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 3.6927881631261505,
+      "learning_rate": 1.4030043910643254e-05,
+      "loss": 0.2124,
+      "step": 13660
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 3.3041534353009743,
+      "learning_rate": 1.401233006423099e-05,
+      "loss": 0.2033,
+      "step": 13680
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 3.539906112805195,
+      "learning_rate": 1.3994601201052253e-05,
+      "loss": 0.2022,
+      "step": 13700
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 3.981704115906401,
+      "learning_rate": 1.39768573874766e-05,
+      "loss": 0.213,
+      "step": 13720
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 2.5831456167458375,
+      "learning_rate": 1.3959098689929555e-05,
+      "loss": 0.1958,
+      "step": 13740
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 2.4960100759889725,
+      "learning_rate": 1.3941325174892371e-05,
+      "loss": 0.2243,
+      "step": 13760
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 2.9399908340726424,
+      "learning_rate": 1.392353690890175e-05,
+      "loss": 0.2052,
+      "step": 13780
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 3.161745178851194,
+      "learning_rate": 1.3905733958549632e-05,
+      "loss": 0.1905,
+      "step": 13800
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 3.6763150256287305,
+      "learning_rate": 1.3887916390482927e-05,
+      "loss": 0.2097,
+      "step": 13820
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 2.589478710103983,
+      "learning_rate": 1.3870084271403269e-05,
+      "loss": 0.2108,
+      "step": 13840
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 2.9786347312251205,
+      "learning_rate": 1.3852237668066763e-05,
+      "loss": 0.2021,
+      "step": 13860
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 2.918896140906093,
+      "learning_rate": 1.3834376647283729e-05,
+      "loss": 0.2184,
+      "step": 13880
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 3.4250636463795074,
+      "learning_rate": 1.3816501275918475e-05,
+      "loss": 0.1989,
+      "step": 13900
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 2.7399597682922923,
+      "learning_rate": 1.3798611620889017e-05,
+      "loss": 0.2089,
+      "step": 13920
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 3.678704067595457,
+      "learning_rate": 1.378070774916686e-05,
+      "loss": 0.2192,
+      "step": 13940
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 3.1242276572434133,
+      "learning_rate": 1.3762789727776708e-05,
+      "loss": 0.2157,
+      "step": 13960
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 2.885737395564972,
+      "learning_rate": 1.3744857623796255e-05,
+      "loss": 0.2133,
+      "step": 13980
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 2.998690955346801,
+      "learning_rate": 1.3726911504355908e-05,
+      "loss": 0.2129,
+      "step": 14000
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 4.013414301345931,
+      "learning_rate": 1.3708951436638542e-05,
+      "loss": 0.2031,
+      "step": 14020
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 3.6060060573956503,
+      "learning_rate": 1.3690977487879247e-05,
+      "loss": 0.2149,
+      "step": 14040
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 2.8295749719869105,
+      "learning_rate": 1.3672989725365076e-05,
+      "loss": 0.2113,
+      "step": 14060
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 4.634259826093287,
+      "learning_rate": 1.3654988216434802e-05,
+      "loss": 0.2013,
+      "step": 14080
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 4.019301812951925,
+      "learning_rate": 1.3636973028478655e-05,
+      "loss": 0.2079,
+      "step": 14100
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 3.3314202699929387,
+      "learning_rate": 1.3618944228938072e-05,
+      "loss": 0.1968,
+      "step": 14120
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 4.011000289395802,
+      "learning_rate": 1.3600901885305446e-05,
+      "loss": 0.2075,
+      "step": 14140
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 3.1307277830268183,
+      "learning_rate": 1.358284606512388e-05,
+      "loss": 0.2004,
+      "step": 14160
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 2.6402421481891047,
+      "learning_rate": 1.3564776835986921e-05,
+      "loss": 0.2102,
+      "step": 14180
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 3.1118142226597745,
+      "learning_rate": 1.3546694265538316e-05,
+      "loss": 0.2124,
+      "step": 14200
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 3.4864022554268614,
+      "learning_rate": 1.3528598421471755e-05,
+      "loss": 0.2067,
+      "step": 14220
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 2.8051441214080457,
+      "learning_rate": 1.3510489371530615e-05,
+      "loss": 0.2163,
+      "step": 14240
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 3.1532788786376638,
+      "learning_rate": 1.3492367183507726e-05,
+      "loss": 0.203,
+      "step": 14260
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 3.6383798904923412,
+      "learning_rate": 1.3474231925245087e-05,
+      "loss": 0.2117,
+      "step": 14280
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 3.2573585834385925,
+      "learning_rate": 1.3456083664633626e-05,
+      "loss": 0.2097,
+      "step": 14300
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 4.797438398942751,
+      "learning_rate": 1.3437922469612951e-05,
+      "loss": 0.2116,
+      "step": 14320
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 2.5155576140103646,
+      "learning_rate": 1.3419748408171094e-05,
+      "loss": 0.2225,
+      "step": 14340
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 5.78637039662545,
+      "learning_rate": 1.3401561548344253e-05,
+      "loss": 0.2074,
+      "step": 14360
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 5.171051974083877,
+      "learning_rate": 1.3383361958216531e-05,
+      "loss": 0.1966,
+      "step": 14380
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 3.6388434115246278,
+      "learning_rate": 1.3365149705919693e-05,
+      "loss": 0.2202,
+      "step": 14400
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 3.6368767430520426,
+      "learning_rate": 1.3346924859632908e-05,
+      "loss": 0.2156,
+      "step": 14420
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 2.6252892652713293,
+      "learning_rate": 1.3328687487582486e-05,
+      "loss": 0.2047,
+      "step": 14440
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 3.2310005313042853,
+      "learning_rate": 1.3310437658041631e-05,
+      "loss": 0.1995,
+      "step": 14460
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 2.591667902252132,
+      "learning_rate": 1.3292175439330182e-05,
+      "loss": 0.1936,
+      "step": 14480
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 6.920226165515457,
+      "learning_rate": 1.3273900899814365e-05,
+      "loss": 0.2008,
+      "step": 14500
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 3.051444827700318,
+      "learning_rate": 1.3255614107906521e-05,
+      "loss": 0.1976,
+      "step": 14520
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 2.9835362079482435,
+      "learning_rate": 1.3237315132064863e-05,
+      "loss": 0.2148,
+      "step": 14540
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 2.8495541541728024,
+      "learning_rate": 1.321900404079321e-05,
+      "loss": 0.1977,
+      "step": 14560
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 2.7688725257095443,
+      "learning_rate": 1.3200680902640744e-05,
+      "loss": 0.2178,
+      "step": 14580
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 2.7073866045339576,
+      "learning_rate": 1.318234578620174e-05,
+      "loss": 0.2146,
+      "step": 14600
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 3.161990242988256,
+      "learning_rate": 1.3163998760115325e-05,
+      "loss": 0.2134,
+      "step": 14620
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 2.7928417256953812,
+      "learning_rate": 1.314563989306519e-05,
+      "loss": 0.2065,
+      "step": 14640
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 2.356147630324507,
+      "learning_rate": 1.3127269253779376e-05,
+      "loss": 0.2105,
+      "step": 14660
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 4.346477722398746,
+      "learning_rate": 1.310888691102998e-05,
+      "loss": 0.2134,
+      "step": 14680
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 2.7518253337355763,
+      "learning_rate": 1.3090492933632926e-05,
+      "loss": 0.2107,
+      "step": 14700
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 3.241061463203708,
+      "learning_rate": 1.307208739044767e-05,
+      "loss": 0.2094,
+      "step": 14720
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 2.666615356507985,
+      "learning_rate": 1.3053670350376993e-05,
+      "loss": 0.1989,
+      "step": 14740
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 3.0536397322406663,
+      "learning_rate": 1.3035241882366692e-05,
+      "loss": 0.2156,
+      "step": 14760
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 2.3893756469119007,
+      "learning_rate": 1.3016802055405363e-05,
+      "loss": 0.2091,
+      "step": 14780
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 3.252027759325605,
+      "learning_rate": 1.2998350938524114e-05,
+      "loss": 0.1993,
+      "step": 14800
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 3.9445454371124105,
+      "learning_rate": 1.2979888600796322e-05,
+      "loss": 0.2021,
+      "step": 14820
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 3.811107068357804,
+      "learning_rate": 1.2961415111337374e-05,
+      "loss": 0.2257,
+      "step": 14840
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 2.437893371140398,
+      "learning_rate": 1.2942930539304397e-05,
+      "loss": 0.1892,
+      "step": 14860
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 2.5823821090570105,
+      "learning_rate": 1.2924434953896014e-05,
+      "loss": 0.1954,
+      "step": 14880
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 2.817062999748064,
+      "learning_rate": 1.2905928424352068e-05,
+      "loss": 0.2016,
+      "step": 14900
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 5.869940669265181,
+      "learning_rate": 1.288741101995338e-05,
+      "loss": 0.211,
+      "step": 14920
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 3.047154496896096,
+      "learning_rate": 1.286888281002148e-05,
+      "loss": 0.1963,
+      "step": 14940
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 3.7639733962315756,
+      "learning_rate": 1.2850343863918358e-05,
+      "loss": 0.2057,
+      "step": 14960
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 2.7118107601240107,
+      "learning_rate": 1.2831794251046173e-05,
+      "loss": 0.2064,
+      "step": 14980
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 4.293885901020362,
+      "learning_rate": 1.281323404084704e-05,
+      "loss": 0.189,
+      "step": 15000
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 3.136084771606493,
+      "learning_rate": 1.2794663302802732e-05,
+      "loss": 0.222,
+      "step": 15020
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 2.33967520945844,
+      "learning_rate": 1.2776082106434442e-05,
+      "loss": 0.21,
+      "step": 15040
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 3.058105585694254,
+      "learning_rate": 1.2757490521302506e-05,
+      "loss": 0.2066,
+      "step": 15060
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 3.783134432254202,
+      "learning_rate": 1.2738888617006158e-05,
+      "loss": 0.204,
+      "step": 15080
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 3.3088612273817577,
+      "learning_rate": 1.2720276463183264e-05,
+      "loss": 0.2088,
+      "step": 15100
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 2.5164740415038063,
+      "learning_rate": 1.2701654129510053e-05,
+      "loss": 0.2083,
+      "step": 15120
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 2.680468920777465,
+      "learning_rate": 1.2683021685700867e-05,
+      "loss": 0.2085,
+      "step": 15140
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 2.7932332800591206,
+      "learning_rate": 1.26643792015079e-05,
+      "loss": 0.218,
+      "step": 15160
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 4.372169232777801,
+      "learning_rate": 1.2645726746720925e-05,
+      "loss": 0.1995,
+      "step": 15180
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 3.1330033622307245,
+      "learning_rate": 1.2627064391167045e-05,
+      "loss": 0.2057,
+      "step": 15200
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 3.6755304848957446,
+      "learning_rate": 1.2608392204710436e-05,
+      "loss": 0.2154,
+      "step": 15220
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 2.54686994493603,
+      "learning_rate": 1.2589710257252058e-05,
+      "loss": 0.2056,
+      "step": 15240
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 3.1076386937733926,
+      "learning_rate": 1.2571018618729422e-05,
+      "loss": 0.2006,
+      "step": 15260
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 4.9484787578205855,
+      "learning_rate": 1.2552317359116326e-05,
+      "loss": 0.2089,
+      "step": 15280
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 2.4893284006088634,
+      "learning_rate": 1.2533606548422577e-05,
+      "loss": 0.2024,
+      "step": 15300
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 3.1084832681147594,
+      "learning_rate": 1.2514886256693732e-05,
+      "loss": 0.2106,
+      "step": 15320
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 3.0389123403917804,
+      "learning_rate": 1.249615655401085e-05,
+      "loss": 0.2096,
+      "step": 15340
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 4.392113049940431,
+      "learning_rate": 1.2477417510490223e-05,
+      "loss": 0.21,
+      "step": 15360
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 2.4290921871050077,
+      "learning_rate": 1.245866919628311e-05,
+      "loss": 0.2101,
+      "step": 15380
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 2.53729167515282,
+      "learning_rate": 1.2439911681575459e-05,
+      "loss": 0.2085,
+      "step": 15400
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.955267140537957,
+      "learning_rate": 1.2421145036587681e-05,
+      "loss": 0.1946,
+      "step": 15420
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 2.3147872747972924,
+      "learning_rate": 1.2402369331574363e-05,
+      "loss": 0.2133,
+      "step": 15440
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 2.6069168116002013,
+      "learning_rate": 1.2383584636824002e-05,
+      "loss": 0.2151,
+      "step": 15460
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 3.6384307404668057,
+      "learning_rate": 1.2364791022658756e-05,
+      "loss": 0.2055,
+      "step": 15480
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 2.7467900528862796,
+      "learning_rate": 1.2345988559434163e-05,
+      "loss": 0.1968,
+      "step": 15500
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 3.712915796699261,
+      "learning_rate": 1.2327177317538903e-05,
+      "loss": 0.2022,
+      "step": 15520
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 3.0273517830034553,
+      "learning_rate": 1.2308357367394515e-05,
+      "loss": 0.2059,
+      "step": 15540
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 2.9422286911490256,
+      "learning_rate": 1.2289528779455129e-05,
+      "loss": 0.2109,
+      "step": 15560
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 2.7461702382273097,
+      "learning_rate": 1.227069162420722e-05,
+      "loss": 0.2008,
+      "step": 15580
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 2.9259269484541868,
+      "learning_rate": 1.2251845972169336e-05,
+      "loss": 0.2087,
+      "step": 15600
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 2.351163202901331,
+      "learning_rate": 1.2232991893891826e-05,
+      "loss": 0.2162,
+      "step": 15620
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 3.3864085617483486,
+      "learning_rate": 1.2214129459956595e-05,
+      "loss": 0.2109,
+      "step": 15640
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 3.1532912786097556,
+      "learning_rate": 1.2195258740976816e-05,
+      "loss": 0.1927,
+      "step": 15660
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 2.32467980179587,
+      "learning_rate": 1.2176379807596684e-05,
+      "loss": 0.2083,
+      "step": 15680
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 3.692322632410928,
+      "learning_rate": 1.2157492730491142e-05,
+      "loss": 0.1952,
+      "step": 15700
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 3.5592767958117704,
+      "learning_rate": 1.2138597580365635e-05,
+      "loss": 0.2259,
+      "step": 15720
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 3.312387140618671,
+      "learning_rate": 1.21196944279558e-05,
+      "loss": 0.1996,
+      "step": 15740
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 2.4404373077236974,
+      "learning_rate": 1.2100783344027257e-05,
+      "loss": 0.2061,
+      "step": 15760
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 3.0929231405689337,
+      "learning_rate": 1.2081864399375306e-05,
+      "loss": 0.1989,
+      "step": 15780
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 3.6918547181282237,
+      "learning_rate": 1.2062937664824692e-05,
+      "loss": 0.2039,
+      "step": 15800
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 25.439570612132734,
+      "learning_rate": 1.2044003211229292e-05,
+      "loss": 0.2023,
+      "step": 15820
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 3.238254008621471,
+      "learning_rate": 1.2025061109471903e-05,
+      "loss": 0.195,
+      "step": 15840
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 2.717237819317576,
+      "learning_rate": 1.2006111430463945e-05,
+      "loss": 0.2159,
+      "step": 15860
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 3.3138924317200207,
+      "learning_rate": 1.198715424514521e-05,
+      "loss": 0.2037,
+      "step": 15880
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 2.477738254765999,
+      "learning_rate": 1.1968189624483583e-05,
+      "loss": 0.2081,
+      "step": 15900
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 5.595275329960663,
+      "learning_rate": 1.194921763947479e-05,
+      "loss": 0.2048,
+      "step": 15920
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 6.622925343547304,
+      "learning_rate": 1.1930238361142116e-05,
+      "loss": 0.2149,
+      "step": 15940
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 3.576100405484642,
+      "learning_rate": 1.1911251860536169e-05,
+      "loss": 0.2103,
+      "step": 15960
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 2.5035188705559004,
+      "learning_rate": 1.1892258208734573e-05,
+      "loss": 0.2108,
+      "step": 15980
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 2.6938599141436494,
+      "learning_rate": 1.1873257476841734e-05,
+      "loss": 0.2029,
+      "step": 16000
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 2.7076420610940914,
+      "learning_rate": 1.1854249735988559e-05,
+      "loss": 0.203,
+      "step": 16020
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 2.53155657006711,
+      "learning_rate": 1.18352350573322e-05,
+      "loss": 0.2073,
+      "step": 16040
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 2.8351358589034037,
+      "learning_rate": 1.1816213512055778e-05,
+      "loss": 0.1972,
+      "step": 16060
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 1.994239617859996,
+      "learning_rate": 1.1797185171368113e-05,
+      "loss": 0.2157,
+      "step": 16080
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 8.626140574352826,
+      "learning_rate": 1.1778150106503472e-05,
+      "loss": 0.2219,
+      "step": 16100
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 3.0368386887010326,
+      "learning_rate": 1.1759108388721294e-05,
+      "loss": 0.212,
+      "step": 16120
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 3.720694025784477,
+      "learning_rate": 1.1740060089305923e-05,
+      "loss": 0.2188,
+      "step": 16140
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 4.140749601972492,
+      "learning_rate": 1.1721005279566335e-05,
+      "loss": 0.2205,
+      "step": 16160
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 3.696009886666488,
+      "learning_rate": 1.170194403083589e-05,
+      "loss": 0.202,
+      "step": 16180
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 2.913336533021661,
+      "learning_rate": 1.1682876414472041e-05,
+      "loss": 0.2122,
+      "step": 16200
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 3.7716338149283706,
+      "learning_rate": 1.1663802501856097e-05,
+      "loss": 0.2256,
+      "step": 16220
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 4.6668357591421215,
+      "learning_rate": 1.1644722364392916e-05,
+      "loss": 0.1985,
+      "step": 16240
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 2.246832949864308,
+      "learning_rate": 1.1625636073510669e-05,
+      "loss": 0.2221,
+      "step": 16260
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 3.962744426709658,
+      "learning_rate": 1.1606543700660566e-05,
+      "loss": 0.2083,
+      "step": 16280
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 2.801003078896173,
+      "learning_rate": 1.1587445317316579e-05,
+      "loss": 0.2005,
+      "step": 16300
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 2.1091323147793117,
+      "learning_rate": 1.1568340994975185e-05,
+      "loss": 0.2045,
+      "step": 16320
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 3.198361161909217,
+      "learning_rate": 1.1549230805155091e-05,
+      "loss": 0.206,
+      "step": 16340
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 10.406894789872744,
+      "learning_rate": 1.1530114819396976e-05,
+      "loss": 0.2123,
+      "step": 16360
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 3.2619393462547737,
+      "learning_rate": 1.1510993109263206e-05,
+      "loss": 0.2041,
+      "step": 16380
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 3.409171994718834,
+      "learning_rate": 1.149186574633759e-05,
+      "loss": 0.2056,
+      "step": 16400
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 2.5059832499949737,
+      "learning_rate": 1.1472732802225081e-05,
+      "loss": 0.2034,
+      "step": 16420
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 3.267940886449616,
+      "learning_rate": 1.1453594348551547e-05,
+      "loss": 0.2012,
+      "step": 16440
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 3.2066882155626875,
+      "learning_rate": 1.1434450456963462e-05,
+      "loss": 0.1971,
+      "step": 16460
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 3.1243325092793883,
+      "learning_rate": 1.1415301199127671e-05,
+      "loss": 0.2041,
+      "step": 16480
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 2.6001065965955448,
+      "learning_rate": 1.1396146646731104e-05,
+      "loss": 0.1996,
+      "step": 16500
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 2.424714226942306,
+      "learning_rate": 1.1376986871480505e-05,
+      "loss": 0.1946,
+      "step": 16520
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 2.4162929070497,
+      "learning_rate": 1.1357821945102177e-05,
+      "loss": 0.1907,
+      "step": 16540
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 3.868349706003454,
+      "learning_rate": 1.133865193934171e-05,
+      "loss": 0.2202,
+      "step": 16560
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 5.899437583367488,
+      "learning_rate": 1.1319476925963702e-05,
+      "loss": 0.203,
+      "step": 16580
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 4.061783477977705,
+      "learning_rate": 1.1300296976751496e-05,
+      "loss": 0.2078,
+      "step": 16600
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 3.279344277780531,
+      "learning_rate": 1.1281112163506923e-05,
+      "loss": 0.2004,
+      "step": 16620
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 3.012593064512875,
+      "learning_rate": 1.1261922558050012e-05,
+      "loss": 0.1992,
+      "step": 16640
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 3.108910147265754,
+      "learning_rate": 1.1242728232218743e-05,
+      "loss": 0.2006,
+      "step": 16660
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 7.0034021556776525,
+      "learning_rate": 1.122352925786875e-05,
+      "loss": 0.2147,
+      "step": 16680
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 1.8028644419788564,
+      "learning_rate": 1.1204325706873091e-05,
+      "loss": 0.2024,
+      "step": 16700
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 2.515029836822745,
+      "learning_rate": 1.1185117651121942e-05,
+      "loss": 0.2128,
+      "step": 16720
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 5.397378511249806,
+      "learning_rate": 1.1165905162522347e-05,
+      "loss": 0.2158,
+      "step": 16740
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 3.52178009365664,
+      "learning_rate": 1.1146688312997946e-05,
+      "loss": 0.2038,
+      "step": 16760
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 4.3841535451132305,
+      "learning_rate": 1.1127467174488697e-05,
+      "loss": 0.2101,
+      "step": 16780
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 3.869824799113014,
+      "learning_rate": 1.1108241818950627e-05,
+      "loss": 0.1925,
+      "step": 16800
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 4.329393772389082,
+      "learning_rate": 1.1089012318355544e-05,
+      "loss": 0.2014,
+      "step": 16820
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 3.781154410675153,
+      "learning_rate": 1.1069778744690771e-05,
+      "loss": 0.1978,
+      "step": 16840
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 2.96129756942421,
+      "learning_rate": 1.1050541169958878e-05,
+      "loss": 0.1964,
+      "step": 16860
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 3.730945241165853,
+      "learning_rate": 1.1031299666177414e-05,
+      "loss": 0.1997,
+      "step": 16880
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 3.6034480060136094,
+      "learning_rate": 1.1012054305378647e-05,
+      "loss": 0.2064,
+      "step": 16900
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 3.2852161959917856,
+      "learning_rate": 1.0992805159609274e-05,
+      "loss": 0.2013,
+      "step": 16920
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 3.2808070292511253,
+      "learning_rate": 1.0973552300930157e-05,
+      "loss": 0.2114,
+      "step": 16940
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 2.8603265565226517,
+      "learning_rate": 1.0954295801416068e-05,
+      "loss": 0.2045,
+      "step": 16960
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 4.057061413690102,
+      "learning_rate": 1.0935035733155402e-05,
+      "loss": 0.1918,
+      "step": 16980
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 3.9395052935238755,
+      "learning_rate": 1.0915772168249923e-05,
+      "loss": 0.213,
+      "step": 17000
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 9.969773195187368,
+      "learning_rate": 1.0896505178814471e-05,
+      "loss": 0.2098,
+      "step": 17020
+    },
+    {
+      "epoch": 1.99,
+      "grad_norm": 3.0741477095309926,
+      "learning_rate": 1.0877234836976714e-05,
+      "loss": 0.2178,
+      "step": 17040
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 3.0763547812686323,
+      "learning_rate": 1.0857961214876875e-05,
+      "loss": 0.2009,
+      "step": 17060
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 2.581952300141561,
+      "learning_rate": 1.0838684384667446e-05,
+      "loss": 0.2033,
+      "step": 17080
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 3.187089482219754,
+      "learning_rate": 1.0819404418512938e-05,
+      "loss": 0.1985,
+      "step": 17100
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 2.6884529730734985,
+      "learning_rate": 1.0800121388589594e-05,
+      "loss": 0.1817,
+      "step": 17120
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 2.647278478646676,
+      "learning_rate": 1.0780835367085132e-05,
+      "loss": 0.1963,
+      "step": 17140
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 4.435409913327422,
+      "learning_rate": 1.0761546426198472e-05,
+      "loss": 0.1834,
+      "step": 17160
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 2.906528582603181,
+      "learning_rate": 1.0742254638139455e-05,
+      "loss": 0.1775,
+      "step": 17180
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 2.0907428839367546,
+      "learning_rate": 1.0722960075128581e-05,
+      "loss": 0.1772,
+      "step": 17200
+    },
+    {
+      "epoch": 2.01,
+      "grad_norm": 2.4500955718726187,
+      "learning_rate": 1.070366280939675e-05,
+      "loss": 0.1839,
+      "step": 17220
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 2.55696802104814,
+      "learning_rate": 1.0684362913184958e-05,
+      "loss": 0.1875,
+      "step": 17240
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 3.7661813665264834,
+      "learning_rate": 1.0665060458744079e-05,
+      "loss": 0.1876,
+      "step": 17260
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 3.3501159818698785,
+      "learning_rate": 1.0645755518334535e-05,
+      "loss": 0.1872,
+      "step": 17280
+    },
+    {
+      "epoch": 2.02,
+      "grad_norm": 4.545667769892908,
+      "learning_rate": 1.0626448164226067e-05,
+      "loss": 0.1732,
+      "step": 17300
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 2.5282252571210164,
+      "learning_rate": 1.0607138468697453e-05,
+      "loss": 0.1885,
+      "step": 17320
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 3.466764276915403,
+      "learning_rate": 1.058782650403624e-05,
+      "loss": 0.1836,
+      "step": 17340
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 4.565230297283624,
+      "learning_rate": 1.0568512342538455e-05,
+      "loss": 0.1808,
+      "step": 17360
+    },
+    {
+      "epoch": 2.03,
+      "grad_norm": 2.702805693184825,
+      "learning_rate": 1.0549196056508364e-05,
+      "loss": 0.1799,
+      "step": 17380
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 3.6330853759662816,
+      "learning_rate": 1.0529877718258179e-05,
+      "loss": 0.1832,
+      "step": 17400
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 3.075596889467387,
+      "learning_rate": 1.0510557400107796e-05,
+      "loss": 0.1901,
+      "step": 17420
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 4.726698086898242,
+      "learning_rate": 1.0491235174384524e-05,
+      "loss": 0.1959,
+      "step": 17440
+    },
+    {
+      "epoch": 2.04,
+      "grad_norm": 4.278041333313194,
+      "learning_rate": 1.0471911113422813e-05,
+      "loss": 0.1851,
+      "step": 17460
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 3.492295368073298,
+      "learning_rate": 1.0452585289563979e-05,
+      "loss": 0.1888,
+      "step": 17480
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 2.54721890801404,
+      "learning_rate": 1.0433257775155945e-05,
+      "loss": 0.188,
+      "step": 17500
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 3.690291043514027,
+      "learning_rate": 1.0413928642552955e-05,
+      "loss": 0.1848,
+      "step": 17520
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 2.420677528904685,
+      "learning_rate": 1.0394597964115322e-05,
+      "loss": 0.1812,
+      "step": 17540
+    },
+    {
+      "epoch": 2.05,
+      "grad_norm": 3.833408368799868,
+      "learning_rate": 1.0375265812209132e-05,
+      "loss": 0.1873,
+      "step": 17560
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 2.594813668795708,
+      "learning_rate": 1.0355932259205996e-05,
+      "loss": 0.1869,
+      "step": 17580
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 2.7847592791858102,
+      "learning_rate": 1.0336597377482773e-05,
+      "loss": 0.1969,
+      "step": 17600
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 3.2040901032730926,
+      "learning_rate": 1.0317261239421283e-05,
+      "loss": 0.1831,
+      "step": 17620
+    },
+    {
+      "epoch": 2.06,
+      "grad_norm": 3.7875189399875167,
+      "learning_rate": 1.0297923917408064e-05,
+      "loss": 0.1832,
+      "step": 17640
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 2.5234037698197986,
+      "learning_rate": 1.0278585483834073e-05,
+      "loss": 0.1634,
+      "step": 17660
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 2.9933831039856162,
+      "learning_rate": 1.0259246011094443e-05,
+      "loss": 0.177,
+      "step": 17680
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 3.610660208566244,
+      "learning_rate": 1.0239905571588182e-05,
+      "loss": 0.1933,
+      "step": 17700
+    },
+    {
+      "epoch": 2.07,
+      "grad_norm": 2.552193547159119,
+      "learning_rate": 1.0220564237717925e-05,
+      "loss": 0.1851,
+      "step": 17720
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 2.3741283825079127,
+      "learning_rate": 1.0201222081889656e-05,
+      "loss": 0.1894,
+      "step": 17740
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 3.0761058400650367,
+      "learning_rate": 1.0181879176512441e-05,
+      "loss": 0.1848,
+      "step": 17760
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 2.8149565247181823,
+      "learning_rate": 1.0162535593998133e-05,
+      "loss": 0.1832,
+      "step": 17780
+    },
+    {
+      "epoch": 2.08,
+      "grad_norm": 3.3079895404618025,
+      "learning_rate": 1.0143191406761144e-05,
+      "loss": 0.2008,
+      "step": 17800
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 3.4780019459030735,
+      "learning_rate": 1.0123846687218129e-05,
+      "loss": 0.177,
+      "step": 17820
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 2.956742911346678,
+      "learning_rate": 1.010450150778775e-05,
+      "loss": 0.1829,
+      "step": 17840
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 4.258286867927119,
+      "learning_rate": 1.0085155940890386e-05,
+      "loss": 0.1738,
+      "step": 17860
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 3.661775245173711,
+      "learning_rate": 1.0065810058947862e-05,
+      "loss": 0.1764,
+      "step": 17880
+    },
+    {
+      "epoch": 2.09,
+      "grad_norm": 3.0954483456868456,
+      "learning_rate": 1.0046463934383193e-05,
+      "loss": 0.1961,
+      "step": 17900
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 3.6220892697165,
+      "learning_rate": 1.0027117639620287e-05,
+      "loss": 0.197,
+      "step": 17920
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 2.184218752270426,
+      "learning_rate": 1.0007771247083709e-05,
+      "loss": 0.1813,
+      "step": 17940
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 6.033162974838077,
+      "learning_rate": 9.988424829198365e-06,
+      "loss": 0.201,
+      "step": 17960
+    },
+    {
+      "epoch": 2.1,
+      "grad_norm": 3.198884119737886,
+      "learning_rate": 9.969078458389281e-06,
+      "loss": 0.1868,
+      "step": 17980
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 2.1035316062893963,
+      "learning_rate": 9.949732207081292e-06,
+      "loss": 0.1803,
+      "step": 18000
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 2.7102647105925195,
+      "learning_rate": 9.930386147698793e-06,
+      "loss": 0.1879,
+      "step": 18020
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 2.797729106040711,
+      "learning_rate": 9.911040352665449e-06,
+      "loss": 0.1763,
+      "step": 18040
+    },
+    {
+      "epoch": 2.11,
+      "grad_norm": 2.5070535337242945,
+      "learning_rate": 9.891694894403947e-06,
+      "loss": 0.1761,
+      "step": 18060
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 2.7653928325208987,
+      "learning_rate": 9.872349845335711e-06,
+      "loss": 0.1874,
+      "step": 18080
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 2.4731031258102307,
+      "learning_rate": 9.853005277880631e-06,
+      "loss": 0.1695,
+      "step": 18100
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 4.932234068564112,
+      "learning_rate": 9.833661264456793e-06,
+      "loss": 0.1788,
+      "step": 18120
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 2.7969357787619646,
+      "learning_rate": 9.814317877480212e-06,
+      "loss": 0.1864,
+      "step": 18140
+    },
+    {
+      "epoch": 2.12,
+      "grad_norm": 6.958001987387912,
+      "learning_rate": 9.794975189364554e-06,
+      "loss": 0.1906,
+      "step": 18160
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 13.874898274366679,
+      "learning_rate": 9.775633272520874e-06,
+      "loss": 0.1887,
+      "step": 18180
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 2.4563842309668944,
+      "learning_rate": 9.756292199357337e-06,
+      "loss": 0.1718,
+      "step": 18200
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 3.701500516205849,
+      "learning_rate": 9.736952042278947e-06,
+      "loss": 0.2008,
+      "step": 18220
+    },
+    {
+      "epoch": 2.13,
+      "grad_norm": 3.8540761301215185,
+      "learning_rate": 9.717612873687285e-06,
+      "loss": 0.1944,
+      "step": 18240
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 2.5734610245537937,
+      "learning_rate": 9.698274765980221e-06,
+      "loss": 0.1953,
+      "step": 18260
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 3.6988039197017955,
+      "learning_rate": 9.678937791551665e-06,
+      "loss": 0.1782,
+      "step": 18280
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 3.5093985117721687,
+      "learning_rate": 9.659602022791278e-06,
+      "loss": 0.1737,
+      "step": 18300
+    },
+    {
+      "epoch": 2.14,
+      "grad_norm": 2.806102002835496,
+      "learning_rate": 9.640267532084208e-06,
+      "loss": 0.1947,
+      "step": 18320
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 4.637926543739354,
+      "learning_rate": 9.620934391810822e-06,
+      "loss": 0.1896,
+      "step": 18340
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 4.20279507955276,
+      "learning_rate": 9.601602674346428e-06,
+      "loss": 0.1808,
+      "step": 18360
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 3.93610696522427,
+      "learning_rate": 9.582272452061003e-06,
+      "loss": 0.1786,
+      "step": 18380
+    },
+    {
+      "epoch": 2.15,
+      "grad_norm": 2.5104348800652727,
+      "learning_rate": 9.562943797318939e-06,
+      "loss": 0.1715,
+      "step": 18400
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 5.142293270927721,
+      "learning_rate": 9.543616782478752e-06,
+      "loss": 0.1771,
+      "step": 18420
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 2.7896695371322764,
+      "learning_rate": 9.52429147989282e-06,
+      "loss": 0.181,
+      "step": 18440
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 2.8005597440629466,
+      "learning_rate": 9.50496796190711e-06,
+      "loss": 0.1936,
+      "step": 18460
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 3.2682102319338617,
+      "learning_rate": 9.485646300860904e-06,
+      "loss": 0.1943,
+      "step": 18480
+    },
+    {
+      "epoch": 2.16,
+      "grad_norm": 3.1744273773216425,
+      "learning_rate": 9.466326569086545e-06,
+      "loss": 0.1826,
+      "step": 18500
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 3.1673830292863894,
+      "learning_rate": 9.447008838909144e-06,
+      "loss": 0.1749,
+      "step": 18520
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 2.2288241494638825,
+      "learning_rate": 9.42769318264632e-06,
+      "loss": 0.1826,
+      "step": 18540
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 51.583019118001666,
+      "learning_rate": 9.408379672607929e-06,
+      "loss": 0.1862,
+      "step": 18560
+    },
+    {
+      "epoch": 2.17,
+      "grad_norm": 3.645865274925182,
+      "learning_rate": 9.389068381095792e-06,
+      "loss": 0.1833,
+      "step": 18580
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 3.6838881050372,
+      "learning_rate": 9.369759380403422e-06,
+      "loss": 0.1863,
+      "step": 18600
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 2.20917195165408,
+      "learning_rate": 9.35045274281577e-06,
+      "loss": 0.1701,
+      "step": 18620
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 3.5141625647505466,
+      "learning_rate": 9.331148540608918e-06,
+      "loss": 0.1827,
+      "step": 18640
+    },
+    {
+      "epoch": 2.18,
+      "grad_norm": 3.633798116884379,
+      "learning_rate": 9.311846846049849e-06,
+      "loss": 0.1821,
+      "step": 18660
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 2.9164096900721406,
+      "learning_rate": 9.29254773139615e-06,
+      "loss": 0.1884,
+      "step": 18680
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 2.5084531922287368,
+      "learning_rate": 9.273251268895755e-06,
+      "loss": 0.17,
+      "step": 18700
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 3.6314194500201795,
+      "learning_rate": 9.253957530786663e-06,
+      "loss": 0.1865,
+      "step": 18720
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 5.532515903593115,
+      "learning_rate": 9.234666589296678e-06,
+      "loss": 0.1784,
+      "step": 18740
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 2.285776324792612,
+      "learning_rate": 9.215378516643135e-06,
+      "loss": 0.1966,
+      "step": 18760
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 1.6912381482889705,
+      "learning_rate": 9.19609338503263e-06,
+      "loss": 0.1684,
+      "step": 18780
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 4.691665019728991,
+      "learning_rate": 9.176811266660744e-06,
+      "loss": 0.1989,
+      "step": 18800
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 3.5520723553410516,
+      "learning_rate": 9.157532233711778e-06,
+      "loss": 0.1919,
+      "step": 18820
+    },
+    {
+      "epoch": 2.2,
+      "grad_norm": 2.354445404385382,
+      "learning_rate": 9.138256358358495e-06,
+      "loss": 0.1646,
+      "step": 18840
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 2.3996097048660343,
+      "learning_rate": 9.11898371276182e-06,
+      "loss": 0.1848,
+      "step": 18860
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 3.179901640708416,
+      "learning_rate": 9.099714369070602e-06,
+      "loss": 0.1848,
+      "step": 18880
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 4.23191486133106,
+      "learning_rate": 9.080448399421314e-06,
+      "loss": 0.1988,
+      "step": 18900
+    },
+    {
+      "epoch": 2.21,
+      "grad_norm": 2.6002312997490864,
+      "learning_rate": 9.061185875937807e-06,
+      "loss": 0.174,
+      "step": 18920
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 2.203729533047983,
+      "learning_rate": 9.041926870731034e-06,
+      "loss": 0.1899,
+      "step": 18940
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 2.4547031453081436,
+      "learning_rate": 9.022671455898773e-06,
+      "loss": 0.1906,
+      "step": 18960
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 3.0866563949523904,
+      "learning_rate": 9.003419703525355e-06,
+      "loss": 0.1809,
+      "step": 18980
+    },
+    {
+      "epoch": 2.22,
+      "grad_norm": 2.2096591664201033,
+      "learning_rate": 8.98417168568141e-06,
+      "loss": 0.1886,
+      "step": 19000
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 2.06152413140976,
+      "learning_rate": 8.96492747442358e-06,
+      "loss": 0.1755,
+      "step": 19020
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 2.3654089563201643,
+      "learning_rate": 8.945687141794268e-06,
+      "loss": 0.1846,
+      "step": 19040
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 3.656689609280586,
+      "learning_rate": 8.926450759821344e-06,
+      "loss": 0.1778,
+      "step": 19060
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 3.573948851020321,
+      "learning_rate": 8.907218400517894e-06,
+      "loss": 0.1856,
+      "step": 19080
+    },
+    {
+      "epoch": 2.23,
+      "grad_norm": 3.059058610806695,
+      "learning_rate": 8.887990135881947e-06,
+      "loss": 0.1871,
+      "step": 19100
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 2.9163171467938107,
+      "learning_rate": 8.868766037896199e-06,
+      "loss": 0.1759,
+      "step": 19120
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 2.690429067178921,
+      "learning_rate": 8.84954617852775e-06,
+      "loss": 0.1983,
+      "step": 19140
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 3.48897409766989,
+      "learning_rate": 8.830330629727828e-06,
+      "loss": 0.1824,
+      "step": 19160
+    },
+    {
+      "epoch": 2.24,
+      "grad_norm": 2.505307095828609,
+      "learning_rate": 8.811119463431534e-06,
+      "loss": 0.1861,
+      "step": 19180
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 2.860339392951664,
+      "learning_rate": 8.791912751557554e-06,
+      "loss": 0.1875,
+      "step": 19200
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 2.3013803320616724,
+      "learning_rate": 8.772710566007902e-06,
+      "loss": 0.1734,
+      "step": 19220
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 3.269140660006422,
+      "learning_rate": 8.753512978667644e-06,
+      "loss": 0.1717,
+      "step": 19240
+    },
+    {
+      "epoch": 2.25,
+      "grad_norm": 2.5218023429151177,
+      "learning_rate": 8.734320061404637e-06,
+      "loss": 0.194,
+      "step": 19260
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 3.1714206614450915,
+      "learning_rate": 8.71513188606925e-06,
+      "loss": 0.1797,
+      "step": 19280
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 2.008888164295546,
+      "learning_rate": 8.69594852449411e-06,
+      "loss": 0.1928,
+      "step": 19300
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 3.074172062016426,
+      "learning_rate": 8.676770048493806e-06,
+      "loss": 0.1933,
+      "step": 19320
+    },
+    {
+      "epoch": 2.26,
+      "grad_norm": 3.0341910026543117,
+      "learning_rate": 8.657596529864656e-06,
+      "loss": 0.1905,
+      "step": 19340
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 2.6160519086313183,
+      "learning_rate": 8.638428040384407e-06,
+      "loss": 0.1963,
+      "step": 19360
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 7.1675419235999245,
+      "learning_rate": 8.619264651811987e-06,
+      "loss": 0.2065,
+      "step": 19380
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 5.44898895544822,
+      "learning_rate": 8.60010643588722e-06,
+      "loss": 0.1876,
+      "step": 19400
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 2.248782358238912,
+      "learning_rate": 8.580953464330571e-06,
+      "loss": 0.1908,
+      "step": 19420
+    },
+    {
+      "epoch": 2.27,
+      "grad_norm": 2.6269853441042317,
+      "learning_rate": 8.561805808842873e-06,
+      "loss": 0.1737,
+      "step": 19440
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 2.654278461092079,
+      "learning_rate": 8.542663541105058e-06,
+      "loss": 0.1735,
+      "step": 19460
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 3.5278579489207997,
+      "learning_rate": 8.523526732777881e-06,
+      "loss": 0.1867,
+      "step": 19480
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 2.5779515063546503,
+      "learning_rate": 8.504395455501673e-06,
+      "loss": 0.1661,
+      "step": 19500
+    },
+    {
+      "epoch": 2.28,
+      "grad_norm": 3.514009367259985,
+      "learning_rate": 8.485269780896044e-06,
+      "loss": 0.1862,
+      "step": 19520
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 2.5847053473530592,
+      "learning_rate": 8.466149780559642e-06,
+      "loss": 0.1879,
+      "step": 19540
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 2.9447271843079865,
+      "learning_rate": 8.447035526069868e-06,
+      "loss": 0.1923,
+      "step": 19560
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 1.9904986834010254,
+      "learning_rate": 8.427927088982608e-06,
+      "loss": 0.1758,
+      "step": 19580
+    },
+    {
+      "epoch": 2.29,
+      "grad_norm": 3.0672202032272557,
+      "learning_rate": 8.408824540831977e-06,
+      "loss": 0.1809,
+      "step": 19600
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 2.2937829620737027,
+      "learning_rate": 8.389727953130044e-06,
+      "loss": 0.1839,
+      "step": 19620
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 3.797253750720813,
+      "learning_rate": 8.370637397366563e-06,
+      "loss": 0.1839,
+      "step": 19640
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 2.796604164692134,
+      "learning_rate": 8.351552945008701e-06,
+      "loss": 0.1743,
+      "step": 19660
+    },
+    {
+      "epoch": 2.3,
+      "grad_norm": 4.4643765903487695,
+      "learning_rate": 8.332474667500786e-06,
+      "loss": 0.1803,
+      "step": 19680
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 3.6583758311891637,
+      "learning_rate": 8.313402636264024e-06,
+      "loss": 0.186,
+      "step": 19700
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 3.7600154645074215,
+      "learning_rate": 8.294336922696247e-06,
+      "loss": 0.173,
+      "step": 19720
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 2.943679478280475,
+      "learning_rate": 8.275277598171615e-06,
+      "loss": 0.1901,
+      "step": 19740
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 2.521023283388264,
+      "learning_rate": 8.256224734040397e-06,
+      "loss": 0.1853,
+      "step": 19760
+    },
+    {
+      "epoch": 2.31,
+      "grad_norm": 9.148411230347637,
+      "learning_rate": 8.237178401628654e-06,
+      "loss": 0.1889,
+      "step": 19780
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 2.3073382118186014,
+      "learning_rate": 8.218138672238008e-06,
+      "loss": 0.1904,
+      "step": 19800
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 2.4455256262578975,
+      "learning_rate": 8.19910561714536e-06,
+      "loss": 0.1942,
+      "step": 19820
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 3.1936382953495905,
+      "learning_rate": 8.180079307602623e-06,
+      "loss": 0.1732,
+      "step": 19840
+    },
+    {
+      "epoch": 2.32,
+      "grad_norm": 2.5890410547283422,
+      "learning_rate": 8.161059814836456e-06,
+      "loss": 0.1905,
+      "step": 19860
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 2.3357113119483905,
+      "learning_rate": 8.142047210048e-06,
+      "loss": 0.1868,
+      "step": 19880
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 3.8518130702595963,
+      "learning_rate": 8.123041564412619e-06,
+      "loss": 0.1772,
+      "step": 19900
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 3.2440906500139315,
+      "learning_rate": 8.104042949079605e-06,
+      "loss": 0.1894,
+      "step": 19920
+    },
+    {
+      "epoch": 2.33,
+      "grad_norm": 2.8690166407648023,
+      "learning_rate": 8.085051435171953e-06,
+      "loss": 0.1972,
+      "step": 19940
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 5.11949923182371,
+      "learning_rate": 8.06606709378606e-06,
+      "loss": 0.1646,
+      "step": 19960
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 3.6058833420132292,
+      "learning_rate": 8.04708999599148e-06,
+      "loss": 0.1686,
+      "step": 19980
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 3.190984429599093,
+      "learning_rate": 8.028120212830638e-06,
+      "loss": 0.196,
+      "step": 20000
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 3.5551243775576666,
+      "learning_rate": 8.009157815318589e-06,
+      "loss": 0.1698,
+      "step": 20020
+    },
+    {
+      "epoch": 2.34,
+      "grad_norm": 3.102970447513887,
+      "learning_rate": 7.99020287444273e-06,
+      "loss": 0.2009,
+      "step": 20040
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 3.059829014258245,
+      "learning_rate": 7.971255461162554e-06,
+      "loss": 0.1845,
+      "step": 20060
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 2.030503708119072,
+      "learning_rate": 7.952315646409361e-06,
+      "loss": 0.1703,
+      "step": 20080
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 3.7340316295377285,
+      "learning_rate": 7.933383501086012e-06,
+      "loss": 0.1806,
+      "step": 20100
+    },
+    {
+      "epoch": 2.35,
+      "grad_norm": 2.381515800479722,
+      "learning_rate": 7.914459096066656e-06,
+      "loss": 0.1764,
+      "step": 20120
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 3.213566578542309,
+      "learning_rate": 7.895542502196467e-06,
+      "loss": 0.1943,
+      "step": 20140
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 3.5176444124185204,
+      "learning_rate": 7.876633790291378e-06,
+      "loss": 0.1798,
+      "step": 20160
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 2.418359580060765,
+      "learning_rate": 7.85773303113781e-06,
+      "loss": 0.1846,
+      "step": 20180
+    },
+    {
+      "epoch": 2.36,
+      "grad_norm": 3.620888713476074,
+      "learning_rate": 7.838840295492415e-06,
+      "loss": 0.1838,
+      "step": 20200
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 2.332237145106214,
+      "learning_rate": 7.81995565408181e-06,
+      "loss": 0.1828,
+      "step": 20220
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 3.043301095307801,
+      "learning_rate": 7.80107917760231e-06,
+      "loss": 0.1766,
+      "step": 20240
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 3.2505637926978066,
+      "learning_rate": 7.782210936719661e-06,
+      "loss": 0.1846,
+      "step": 20260
+    },
+    {
+      "epoch": 2.37,
+      "grad_norm": 2.9027355643269837,
+      "learning_rate": 7.763351002068778e-06,
+      "loss": 0.1754,
+      "step": 20280
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 3.373330869133493,
+      "learning_rate": 7.744499444253488e-06,
+      "loss": 0.1889,
+      "step": 20300
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 2.562535634607174,
+      "learning_rate": 7.725656333846251e-06,
+      "loss": 0.1905,
+      "step": 20320
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 3.269212276527265,
+      "learning_rate": 7.706821741387906e-06,
+      "loss": 0.1735,
+      "step": 20340
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 2.694612185401067,
+      "learning_rate": 7.687995737387405e-06,
+      "loss": 0.1814,
+      "step": 20360
+    },
+    {
+      "epoch": 2.38,
+      "grad_norm": 2.550536910734683,
+      "learning_rate": 7.669178392321549e-06,
+      "loss": 0.1951,
+      "step": 20380
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 3.7192454288549706,
+      "learning_rate": 7.65036977663472e-06,
+      "loss": 0.1811,
+      "step": 20400
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 4.583412424040906,
+      "learning_rate": 7.631569960738625e-06,
+      "loss": 0.1912,
+      "step": 20420
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 3.6102709958551844,
+      "learning_rate": 7.612779015012024e-06,
+      "loss": 0.1857,
+      "step": 20440
+    },
+    {
+      "epoch": 2.39,
+      "grad_norm": 2.132609263367914,
+      "learning_rate": 7.593997009800476e-06,
+      "loss": 0.1852,
+      "step": 20460
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 3.4880324528085604,
+      "learning_rate": 7.575224015416064e-06,
+      "loss": 0.1715,
+      "step": 20480
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 1.8270056794678053,
+      "learning_rate": 7.5564601021371455e-06,
+      "loss": 0.1859,
+      "step": 20500
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 3.2661614228423366,
+      "learning_rate": 7.537705340208075e-06,
+      "loss": 0.1763,
+      "step": 20520
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 2.652309411884222,
+      "learning_rate": 7.518959799838952e-06,
+      "loss": 0.1848,
+      "step": 20540
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 2.840447321036168,
+      "learning_rate": 7.5002235512053515e-06,
+      "loss": 0.1885,
+      "step": 20560
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 2.8852660946629722,
+      "learning_rate": 7.481496664448073e-06,
+      "loss": 0.1823,
+      "step": 20580
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 3.287773695311469,
+      "learning_rate": 7.462779209672856e-06,
+      "loss": 0.1617,
+      "step": 20600
+    },
+    {
+      "epoch": 2.41,
+      "grad_norm": 3.3685486912072653,
+      "learning_rate": 7.444071256950138e-06,
+      "loss": 0.1796,
+      "step": 20620
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 4.499428818798512,
+      "learning_rate": 7.425372876314783e-06,
+      "loss": 0.2029,
+      "step": 20640
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 3.1348393987132894,
+      "learning_rate": 7.406684137765827e-06,
+      "loss": 0.1828,
+      "step": 20660
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 2.3227046875533435,
+      "learning_rate": 7.388005111266197e-06,
+      "loss": 0.1909,
+      "step": 20680
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 3.142369938528309,
+      "learning_rate": 7.369335866742471e-06,
+      "loss": 0.1769,
+      "step": 20700
+    },
+    {
+      "epoch": 2.42,
+      "grad_norm": 2.4074339492943193,
+      "learning_rate": 7.350676474084608e-06,
+      "loss": 0.1833,
+      "step": 20720
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 2.4803162054779415,
+      "learning_rate": 7.3320270031456855e-06,
+      "loss": 0.1754,
+      "step": 20740
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 1.998861465638246,
+      "learning_rate": 7.31338752374163e-06,
+      "loss": 0.1832,
+      "step": 20760
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 3.1286831322602557,
+      "learning_rate": 7.294758105650972e-06,
+      "loss": 0.1896,
+      "step": 20780
+    },
+    {
+      "epoch": 2.43,
+      "grad_norm": 2.5799109544099244,
+      "learning_rate": 7.276138818614579e-06,
+      "loss": 0.1748,
+      "step": 20800
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 3.3287032077321066,
+      "learning_rate": 7.257529732335385e-06,
+      "loss": 0.1987,
+      "step": 20820
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 6.680428732630699,
+      "learning_rate": 7.238930916478139e-06,
+      "loss": 0.1866,
+      "step": 20840
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 2.5614379915099614,
+      "learning_rate": 7.2203424406691416e-06,
+      "loss": 0.1848,
+      "step": 20860
+    },
+    {
+      "epoch": 2.44,
+      "grad_norm": 3.3339886656936364,
+      "learning_rate": 7.201764374495986e-06,
+      "loss": 0.1811,
+      "step": 20880
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 3.0393813095952913,
+      "learning_rate": 7.1831967875072935e-06,
+      "loss": 0.1604,
+      "step": 20900
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 1.7498486998411649,
+      "learning_rate": 7.1646397492124605e-06,
+      "loss": 0.1858,
+      "step": 20920
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 3.448053147917569,
+      "learning_rate": 7.146093329081384e-06,
+      "loss": 0.1735,
+      "step": 20940
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 3.7318359426110197,
+      "learning_rate": 7.127557596544221e-06,
+      "loss": 0.1803,
+      "step": 20960
+    },
+    {
+      "epoch": 2.45,
+      "grad_norm": 2.294152755094677,
+      "learning_rate": 7.109032620991115e-06,
+      "loss": 0.1812,
+      "step": 20980
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 2.695852120602769,
+      "learning_rate": 7.0905184717719374e-06,
+      "loss": 0.1698,
+      "step": 21000
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 1.7682884854314216,
+      "learning_rate": 7.072015218196033e-06,
+      "loss": 0.1794,
+      "step": 21020
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 2.8223377680550117,
+      "learning_rate": 7.053522929531956e-06,
+      "loss": 0.1879,
+      "step": 21040
+    },
+    {
+      "epoch": 2.46,
+      "grad_norm": 2.3633329023281004,
+      "learning_rate": 7.035041675007215e-06,
+      "loss": 0.1774,
+      "step": 21060
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 2.5439977014459267,
+      "learning_rate": 7.016571523808011e-06,
+      "loss": 0.1801,
+      "step": 21080
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 3.017965722730571,
+      "learning_rate": 6.998112545078976e-06,
+      "loss": 0.1835,
+      "step": 21100
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 3.2340560977012465,
+      "learning_rate": 6.9796648079229154e-06,
+      "loss": 0.183,
+      "step": 21120
+    },
+    {
+      "epoch": 2.47,
+      "grad_norm": 2.7076792196060855,
+      "learning_rate": 6.961228381400557e-06,
+      "loss": 0.17,
+      "step": 21140
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 2.5294049921088604,
+      "learning_rate": 6.9428033345302836e-06,
+      "loss": 0.1748,
+      "step": 21160
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 1.9755482842123588,
+      "learning_rate": 6.924389736287876e-06,
+      "loss": 0.1797,
+      "step": 21180
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 2.6170283482684837,
+      "learning_rate": 6.905987655606256e-06,
+      "loss": 0.1742,
+      "step": 21200
+    },
+    {
+      "epoch": 2.48,
+      "grad_norm": 2.706014158619269,
+      "learning_rate": 6.887597161375228e-06,
+      "loss": 0.1881,
+      "step": 21220
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 2.4950356785549106,
+      "learning_rate": 6.869218322441225e-06,
+      "loss": 0.1644,
+      "step": 21240
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 4.077341289413537,
+      "learning_rate": 6.850851207607047e-06,
+      "loss": 0.1901,
+      "step": 21260
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 2.5754376826477765,
+      "learning_rate": 6.832495885631597e-06,
+      "loss": 0.1783,
+      "step": 21280
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 6.354918922129214,
+      "learning_rate": 6.814152425229636e-06,
+      "loss": 0.1678,
+      "step": 21300
+    },
+    {
+      "epoch": 2.49,
+      "grad_norm": 2.394268674906887,
+      "learning_rate": 6.795820895071523e-06,
+      "loss": 0.1782,
+      "step": 21320
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 3.186718863720809,
+      "learning_rate": 6.777501363782952e-06,
+      "loss": 0.175,
+      "step": 21340
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 2.5749398641138184,
+      "learning_rate": 6.759193899944691e-06,
+      "loss": 0.1827,
+      "step": 21360
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 2.8466404907249983,
+      "learning_rate": 6.740898572092345e-06,
+      "loss": 0.1846,
+      "step": 21380
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 1.8342194005424137,
+      "learning_rate": 6.722615448716077e-06,
+      "loss": 0.1707,
+      "step": 21400
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 3.129190153464715,
+      "learning_rate": 6.704344598260372e-06,
+      "loss": 0.1676,
+      "step": 21420
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 3.723062274962248,
+      "learning_rate": 6.6860860891237524e-06,
+      "loss": 0.1897,
+      "step": 21440
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 5.087016329631688,
+      "learning_rate": 6.667839989658562e-06,
+      "loss": 0.1752,
+      "step": 21460
+    },
+    {
+      "epoch": 2.51,
+      "grad_norm": 3.2186600755889,
+      "learning_rate": 6.649606368170671e-06,
+      "loss": 0.1899,
+      "step": 21480
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 3.005904268619833,
+      "learning_rate": 6.631385292919246e-06,
+      "loss": 0.1844,
+      "step": 21500
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 2.6779566031964586,
+      "learning_rate": 6.613176832116484e-06,
+      "loss": 0.1836,
+      "step": 21520
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 3.702777315066449,
+      "learning_rate": 6.594981053927354e-06,
+      "loss": 0.1896,
+      "step": 21540
+    },
+    {
+      "epoch": 2.52,
+      "grad_norm": 3.289749907290013,
+      "learning_rate": 6.576798026469352e-06,
+      "loss": 0.1784,
+      "step": 21560
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 2.1363438027971995,
+      "learning_rate": 6.558627817812236e-06,
+      "loss": 0.1841,
+      "step": 21580
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 2.8492415506482476,
+      "learning_rate": 6.540470495977783e-06,
+      "loss": 0.1745,
+      "step": 21600
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 2.4044134884900337,
+      "learning_rate": 6.522326128939518e-06,
+      "loss": 0.1825,
+      "step": 21620
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 2.280363668416859,
+      "learning_rate": 6.504194784622475e-06,
+      "loss": 0.1967,
+      "step": 21640
+    },
+    {
+      "epoch": 2.53,
+      "grad_norm": 5.67258820896823,
+      "learning_rate": 6.486076530902932e-06,
+      "loss": 0.1881,
+      "step": 21660
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 2.4280569609507388,
+      "learning_rate": 6.467971435608168e-06,
+      "loss": 0.18,
+      "step": 21680
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 2.7465414493529705,
+      "learning_rate": 6.449879566516194e-06,
+      "loss": 0.1893,
+      "step": 21700
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 3.7485576398899934,
+      "learning_rate": 6.4318009913555115e-06,
+      "loss": 0.1684,
+      "step": 21720
+    },
+    {
+      "epoch": 2.54,
+      "grad_norm": 2.1872776463442554,
+      "learning_rate": 6.413735777804855e-06,
+      "loss": 0.1695,
+      "step": 21740
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 2.6147402124846324,
+      "learning_rate": 6.395683993492941e-06,
+      "loss": 0.1918,
+      "step": 21760
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 3.2226855467418334,
+      "learning_rate": 6.377645705998208e-06,
+      "loss": 0.1685,
+      "step": 21780
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 4.304393945228268,
+      "learning_rate": 6.359620982848567e-06,
+      "loss": 0.1824,
+      "step": 21800
+    },
+    {
+      "epoch": 2.55,
+      "grad_norm": 2.460673581458399,
+      "learning_rate": 6.341609891521153e-06,
+      "loss": 0.1804,
+      "step": 21820
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 2.4334232766585817,
+      "learning_rate": 6.3236124994420746e-06,
+      "loss": 0.1856,
+      "step": 21840
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 12.823001484263724,
+      "learning_rate": 6.305628873986145e-06,
+      "loss": 0.1741,
+      "step": 21860
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 3.3642127400826882,
+      "learning_rate": 6.2876590824766445e-06,
+      "loss": 0.181,
+      "step": 21880
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 4.637360423164878,
+      "learning_rate": 6.269703192185069e-06,
+      "loss": 0.1872,
+      "step": 21900
+    },
+    {
+      "epoch": 2.56,
+      "grad_norm": 3.0708541586457465,
+      "learning_rate": 6.251761270330871e-06,
+      "loss": 0.1802,
+      "step": 21920
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 3.0814400264184236,
+      "learning_rate": 6.233833384081214e-06,
+      "loss": 0.1793,
+      "step": 21940
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 2.43342072926522,
+      "learning_rate": 6.215919600550707e-06,
+      "loss": 0.1758,
+      "step": 21960
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 2.881630756102204,
+      "learning_rate": 6.198019986801178e-06,
+      "loss": 0.1757,
+      "step": 21980
+    },
+    {
+      "epoch": 2.57,
+      "grad_norm": 2.211538603792218,
+      "learning_rate": 6.1801346098414036e-06,
+      "loss": 0.1783,
+      "step": 22000
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 8.512703770734728,
+      "learning_rate": 6.162263536626864e-06,
+      "loss": 0.1805,
+      "step": 22020
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 2.2659396249079604,
+      "learning_rate": 6.144406834059488e-06,
+      "loss": 0.1768,
+      "step": 22040
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 2.9857345599922196,
+      "learning_rate": 6.126564568987412e-06,
+      "loss": 0.1814,
+      "step": 22060
+    },
+    {
+      "epoch": 2.58,
+      "grad_norm": 2.3473919864135215,
+      "learning_rate": 6.108736808204723e-06,
+      "loss": 0.1655,
+      "step": 22080
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 1.7312928039547535,
+      "learning_rate": 6.090923618451211e-06,
+      "loss": 0.1737,
+      "step": 22100
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 2.807589503939605,
+      "learning_rate": 6.073125066412114e-06,
+      "loss": 0.1844,
+      "step": 22120
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 3.4617900960703105,
+      "learning_rate": 6.055341218717877e-06,
+      "loss": 0.1907,
+      "step": 22140
+    },
+    {
+      "epoch": 2.59,
+      "grad_norm": 3.4841808042594486,
+      "learning_rate": 6.037572141943895e-06,
+      "loss": 0.1787,
+      "step": 22160
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 2.6092051376516756,
+      "learning_rate": 6.019817902610269e-06,
+      "loss": 0.1656,
+      "step": 22180
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 3.136799318411884,
+      "learning_rate": 6.002078567181554e-06,
+      "loss": 0.1732,
+      "step": 22200
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 3.4455409936543546,
+      "learning_rate": 5.984354202066508e-06,
+      "loss": 0.1815,
+      "step": 22220
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 2.612732332110929,
+      "learning_rate": 5.966644873617851e-06,
+      "loss": 0.1779,
+      "step": 22240
+    },
+    {
+      "epoch": 2.6,
+      "grad_norm": 2.3915440407218425,
+      "learning_rate": 5.948950648132007e-06,
+      "loss": 0.1795,
+      "step": 22260
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 4.31280781849793,
+      "learning_rate": 5.931271591848869e-06,
+      "loss": 0.1631,
+      "step": 22280
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 4.414478599718549,
+      "learning_rate": 5.913607770951527e-06,
+      "loss": 0.1811,
+      "step": 22300
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 8.180075970509124,
+      "learning_rate": 5.895959251566058e-06,
+      "loss": 0.1801,
+      "step": 22320
+    },
+    {
+      "epoch": 2.61,
+      "grad_norm": 2.0782483286580717,
+      "learning_rate": 5.878326099761238e-06,
+      "loss": 0.1739,
+      "step": 22340
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 2.5269629826915856,
+      "learning_rate": 5.860708381548322e-06,
+      "loss": 0.181,
+      "step": 22360
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 2.5519186991832656,
+      "learning_rate": 5.843106162880785e-06,
+      "loss": 0.1834,
+      "step": 22380
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 3.435157671423404,
+      "learning_rate": 5.825519509654079e-06,
+      "loss": 0.1763,
+      "step": 22400
+    },
+    {
+      "epoch": 2.62,
+      "grad_norm": 2.6878877404420196,
+      "learning_rate": 5.807948487705384e-06,
+      "loss": 0.1941,
+      "step": 22420
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 2.5448246291018366,
+      "learning_rate": 5.790393162813365e-06,
+      "loss": 0.1815,
+      "step": 22440
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 4.743917083813302,
+      "learning_rate": 5.772853600697924e-06,
+      "loss": 0.1831,
+      "step": 22460
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 2.379670720282324,
+      "learning_rate": 5.755329867019952e-06,
+      "loss": 0.1811,
+      "step": 22480
+    },
+    {
+      "epoch": 2.63,
+      "grad_norm": 2.610666195409938,
+      "learning_rate": 5.737822027381088e-06,
+      "loss": 0.1743,
+      "step": 22500
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 3.0100376370126853,
+      "learning_rate": 5.720330147323458e-06,
+      "loss": 0.1902,
+      "step": 22520
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 2.99934268744031,
+      "learning_rate": 5.702854292329461e-06,
+      "loss": 0.1736,
+      "step": 22540
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 3.5108678209005477,
+      "learning_rate": 5.685394527821494e-06,
+      "loss": 0.1882,
+      "step": 22560
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 6.166234628475323,
+      "learning_rate": 5.66795091916172e-06,
+      "loss": 0.1776,
+      "step": 22580
+    },
+    {
+      "epoch": 2.64,
+      "grad_norm": 2.265912267954542,
+      "learning_rate": 5.650523531651817e-06,
+      "loss": 0.1775,
+      "step": 22600
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 2.875442305003155,
+      "learning_rate": 5.63311243053274e-06,
+      "loss": 0.182,
+      "step": 22620
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 4.23319985387031,
+      "learning_rate": 5.615717680984476e-06,
+      "loss": 0.1862,
+      "step": 22640
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 2.813515458637544,
+      "learning_rate": 5.598339348125805e-06,
+      "loss": 0.1754,
+      "step": 22660
+    },
+    {
+      "epoch": 2.65,
+      "grad_norm": 2.465967179258659,
+      "learning_rate": 5.580977497014037e-06,
+      "loss": 0.1752,
+      "step": 22680
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 3.0256927516690193,
+      "learning_rate": 5.563632192644785e-06,
+      "loss": 0.1836,
+      "step": 22700
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 7.287674455861031,
+      "learning_rate": 5.546303499951722e-06,
+      "loss": 0.1891,
+      "step": 22720
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 2.3533914240438594,
+      "learning_rate": 5.528991483806333e-06,
+      "loss": 0.1849,
+      "step": 22740
+    },
+    {
+      "epoch": 2.66,
+      "grad_norm": 3.6142213990161087,
+      "learning_rate": 5.511696209017668e-06,
+      "loss": 0.1799,
+      "step": 22760
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 4.110370703915694,
+      "learning_rate": 5.494417740332109e-06,
+      "loss": 0.1884,
+      "step": 22780
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 3.28314953271809,
+      "learning_rate": 5.477156142433121e-06,
+      "loss": 0.1792,
+      "step": 22800
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 3.4918275515348984,
+      "learning_rate": 5.459911479941013e-06,
+      "loss": 0.1678,
+      "step": 22820
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 3.5246556918764216,
+      "learning_rate": 5.442683817412696e-06,
+      "loss": 0.1834,
+      "step": 22840
+    },
+    {
+      "epoch": 2.67,
+      "grad_norm": 3.9386810469092604,
+      "learning_rate": 5.4254732193414325e-06,
+      "loss": 0.1855,
+      "step": 22860
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 2.7961797436144646,
+      "learning_rate": 5.408279750156611e-06,
+      "loss": 0.177,
+      "step": 22880
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 1.8885662755469237,
+      "learning_rate": 5.3911034742234935e-06,
+      "loss": 0.1877,
+      "step": 22900
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 3.5211002015468824,
+      "learning_rate": 5.3739444558429785e-06,
+      "loss": 0.173,
+      "step": 22920
+    },
+    {
+      "epoch": 2.68,
+      "grad_norm": 2.889315251733768,
+      "learning_rate": 5.356802759251359e-06,
+      "loss": 0.1721,
+      "step": 22940
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 2.6526195174893497,
+      "learning_rate": 5.33967844862008e-06,
+      "loss": 0.1891,
+      "step": 22960
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 2.3210056695240207,
+      "learning_rate": 5.322571588055504e-06,
+      "loss": 0.1779,
+      "step": 22980
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 2.93332602832347,
+      "learning_rate": 5.305482241598673e-06,
+      "loss": 0.1816,
+      "step": 23000
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 2.3996137785091047,
+      "learning_rate": 5.288410473225043e-06,
+      "loss": 0.1767,
+      "step": 23020
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 3.156745343004959,
+      "learning_rate": 5.271356346844285e-06,
+      "loss": 0.1745,
+      "step": 23040
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 2.7810676600659145,
+      "learning_rate": 5.254319926300022e-06,
+      "loss": 0.207,
+      "step": 23060
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 3.7710824255400843,
+      "learning_rate": 5.237301275369584e-06,
+      "loss": 0.1686,
+      "step": 23080
+    },
+    {
+      "epoch": 2.7,
+      "grad_norm": 2.4391670395260356,
+      "learning_rate": 5.220300457763794e-06,
+      "loss": 0.1783,
+      "step": 23100
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 2.4162439656268617,
+      "learning_rate": 5.203317537126702e-06,
+      "loss": 0.1804,
+      "step": 23120
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 2.584051729623389,
+      "learning_rate": 5.186352577035364e-06,
+      "loss": 0.1782,
+      "step": 23140
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 4.071893718029455,
+      "learning_rate": 5.169405640999598e-06,
+      "loss": 0.1774,
+      "step": 23160
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 3.8461652161356787,
+      "learning_rate": 5.152476792461754e-06,
+      "loss": 0.1788,
+      "step": 23180
+    },
+    {
+      "epoch": 2.71,
+      "grad_norm": 2.2006858146864485,
+      "learning_rate": 5.135566094796451e-06,
+      "loss": 0.1821,
+      "step": 23200
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 3.4525315823446716,
+      "learning_rate": 5.11867361131038e-06,
+      "loss": 0.176,
+      "step": 23220
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 3.2240358670987,
+      "learning_rate": 5.1017994052420315e-06,
+      "loss": 0.1821,
+      "step": 23240
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 3.174379459324645,
+      "learning_rate": 5.084943539761484e-06,
+      "loss": 0.1742,
+      "step": 23260
+    },
+    {
+      "epoch": 2.72,
+      "grad_norm": 2.002869335885304,
+      "learning_rate": 5.068106077970147e-06,
+      "loss": 0.1714,
+      "step": 23280
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 3.039475753654841,
+      "learning_rate": 5.05128708290054e-06,
+      "loss": 0.1862,
+      "step": 23300
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 11.415182049603324,
+      "learning_rate": 5.034486617516047e-06,
+      "loss": 0.1864,
+      "step": 23320
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 2.215650946245953,
+      "learning_rate": 5.01770474471069e-06,
+      "loss": 0.1707,
+      "step": 23340
+    },
+    {
+      "epoch": 2.73,
+      "grad_norm": 1.5861810205971256,
+      "learning_rate": 5.000941527308877e-06,
+      "loss": 0.1723,
+      "step": 23360
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 2.736141048682862,
+      "learning_rate": 4.984197028065187e-06,
+      "loss": 0.1817,
+      "step": 23380
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 2.785186296597876,
+      "learning_rate": 4.967471309664126e-06,
+      "loss": 0.1764,
+      "step": 23400
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 2.885900151662176,
+      "learning_rate": 4.950764434719889e-06,
+      "loss": 0.1828,
+      "step": 23420
+    },
+    {
+      "epoch": 2.74,
+      "grad_norm": 3.137029763048628,
+      "learning_rate": 4.934076465776131e-06,
+      "loss": 0.1809,
+      "step": 23440
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 2.923199358588338,
+      "learning_rate": 4.917407465305728e-06,
+      "loss": 0.1674,
+      "step": 23460
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 2.964303953866439,
+      "learning_rate": 4.90075749571055e-06,
+      "loss": 0.1773,
+      "step": 23480
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 4.009182404005084,
+      "learning_rate": 4.884126619321215e-06,
+      "loss": 0.1728,
+      "step": 23500
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 2.9239608744705476,
+      "learning_rate": 4.867514898396881e-06,
+      "loss": 0.1854,
+      "step": 23520
+    },
+    {
+      "epoch": 2.75,
+      "grad_norm": 3.0183117218620064,
+      "learning_rate": 4.8509223951249725e-06,
+      "loss": 0.1688,
+      "step": 23540
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 2.9178365450200583,
+      "learning_rate": 4.8343491716209906e-06,
+      "loss": 0.1804,
+      "step": 23560
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 1.9094178814607006,
+      "learning_rate": 4.81779528992825e-06,
+      "loss": 0.1607,
+      "step": 23580
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 4.365735954125327,
+      "learning_rate": 4.801260812017662e-06,
+      "loss": 0.1806,
+      "step": 23600
+    },
+    {
+      "epoch": 2.76,
+      "grad_norm": 2.428095255980094,
+      "learning_rate": 4.7847457997874975e-06,
+      "loss": 0.1755,
+      "step": 23620
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 1.5540702783116471,
+      "learning_rate": 4.768250315063154e-06,
+      "loss": 0.174,
+      "step": 23640
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 2.549882669180469,
+      "learning_rate": 4.751774419596929e-06,
+      "loss": 0.1722,
+      "step": 23660
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 3.5723809525533547,
+      "learning_rate": 4.735318175067789e-06,
+      "loss": 0.1929,
+      "step": 23680
+    },
+    {
+      "epoch": 2.77,
+      "grad_norm": 3.2517252524328635,
+      "learning_rate": 4.7188816430811215e-06,
+      "loss": 0.1901,
+      "step": 23700
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 3.250214496283374,
+      "learning_rate": 4.702464885168532e-06,
+      "loss": 0.1726,
+      "step": 23720
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 3.0741278551325983,
+      "learning_rate": 4.686067962787599e-06,
+      "loss": 0.1789,
+      "step": 23740
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 3.4015668119894724,
+      "learning_rate": 4.669690937321637e-06,
+      "loss": 0.1731,
+      "step": 23760
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 2.162316454556547,
+      "learning_rate": 4.653333870079482e-06,
+      "loss": 0.1727,
+      "step": 23780
+    },
+    {
+      "epoch": 2.78,
+      "grad_norm": 2.954847529781422,
+      "learning_rate": 4.636996822295253e-06,
+      "loss": 0.1926,
+      "step": 23800
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 3.218453402881352,
+      "learning_rate": 4.620679855128122e-06,
+      "loss": 0.1752,
+      "step": 23820
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 2.3305631837756633,
+      "learning_rate": 4.604383029662088e-06,
+      "loss": 0.1681,
+      "step": 23840
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 3.2650324087593714,
+      "learning_rate": 4.588106406905756e-06,
+      "loss": 0.167,
+      "step": 23860
+    },
+    {
+      "epoch": 2.79,
+      "grad_norm": 2.7496586501069826,
+      "learning_rate": 4.5718500477920825e-06,
+      "loss": 0.1646,
+      "step": 23880
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 3.211021403680614,
+      "learning_rate": 4.555614013178179e-06,
+      "loss": 0.1704,
+      "step": 23900
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 2.902403059465085,
+      "learning_rate": 4.5393983638450714e-06,
+      "loss": 0.1814,
+      "step": 23920
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 2.460405952329076,
+      "learning_rate": 4.5232031604974605e-06,
+      "loss": 0.1753,
+      "step": 23940
+    },
+    {
+      "epoch": 2.8,
+      "grad_norm": 3.1027053533889952,
+      "learning_rate": 4.507028463763521e-06,
+      "loss": 0.1821,
+      "step": 23960
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 2.4814554970152076,
+      "learning_rate": 4.490874334194647e-06,
+      "loss": 0.1754,
+      "step": 23980
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 3.0816354630415317,
+      "learning_rate": 4.474740832265242e-06,
+      "loss": 0.1647,
+      "step": 24000
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 4.258299409515962,
+      "learning_rate": 4.45862801837249e-06,
+      "loss": 0.1651,
+      "step": 24020
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 2.5109760616168644,
+      "learning_rate": 4.442535952836121e-06,
+      "loss": 0.1737,
+      "step": 24040
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 3.670190952540861,
+      "learning_rate": 4.426464695898196e-06,
+      "loss": 0.1815,
+      "step": 24060
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 2.246500684943945,
+      "learning_rate": 4.41041430772288e-06,
+      "loss": 0.1778,
+      "step": 24080
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 2.8719645522541355,
+      "learning_rate": 4.394384848396209e-06,
+      "loss": 0.1729,
+      "step": 24100
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 2.5285651956593784,
+      "learning_rate": 4.378376377925873e-06,
+      "loss": 0.1812,
+      "step": 24120
+    },
+    {
+      "epoch": 2.82,
+      "grad_norm": 1.589113859923903,
+      "learning_rate": 4.362388956240989e-06,
+      "loss": 0.1765,
+      "step": 24140
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 3.212563652862593,
+      "learning_rate": 4.346422643191872e-06,
+      "loss": 0.1861,
+      "step": 24160
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 1.9036122817924004,
+      "learning_rate": 4.330477498549818e-06,
+      "loss": 0.1757,
+      "step": 24180
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 2.773805921557294,
+      "learning_rate": 4.314553582006882e-06,
+      "loss": 0.1795,
+      "step": 24200
+    },
+    {
+      "epoch": 2.83,
+      "grad_norm": 4.356117261837307,
+      "learning_rate": 4.298650953175635e-06,
+      "loss": 0.1817,
+      "step": 24220
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 2.629481169141941,
+      "learning_rate": 4.282769671588969e-06,
+      "loss": 0.1777,
+      "step": 24240
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 3.3316093437094025,
+      "learning_rate": 4.2669097966998586e-06,
+      "loss": 0.1684,
+      "step": 24260
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 3.671906945921538,
+      "learning_rate": 4.251071387881136e-06,
+      "loss": 0.1828,
+      "step": 24280
+    },
+    {
+      "epoch": 2.84,
+      "grad_norm": 2.67347838717775,
+      "learning_rate": 4.235254504425278e-06,
+      "loss": 0.1708,
+      "step": 24300
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 2.862267370641254,
+      "learning_rate": 4.2194592055441765e-06,
+      "loss": 0.1706,
+      "step": 24320
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 2.953771449788878,
+      "learning_rate": 4.203685550368921e-06,
+      "loss": 0.1683,
+      "step": 24340
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 2.5105236764052354,
+      "learning_rate": 4.187933597949578e-06,
+      "loss": 0.186,
+      "step": 24360
+    },
+    {
+      "epoch": 2.85,
+      "grad_norm": 2.218245972886692,
+      "learning_rate": 4.172203407254954e-06,
+      "loss": 0.1627,
+      "step": 24380
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 3.517470625569687,
+      "learning_rate": 4.1564950371724115e-06,
+      "loss": 0.1745,
+      "step": 24400
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 2.3898984515591137,
+      "learning_rate": 4.140808546507611e-06,
+      "loss": 0.1757,
+      "step": 24420
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 3.7119551698756417,
+      "learning_rate": 4.125143993984306e-06,
+      "loss": 0.171,
+      "step": 24440
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 1.8304933384323714,
+      "learning_rate": 4.109501438244126e-06,
+      "loss": 0.1804,
+      "step": 24460
+    },
+    {
+      "epoch": 2.86,
+      "grad_norm": 2.967840597504324,
+      "learning_rate": 4.093880937846353e-06,
+      "loss": 0.182,
+      "step": 24480
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 3.1239513926711426,
+      "learning_rate": 4.078282551267702e-06,
+      "loss": 0.1856,
+      "step": 24500
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 3.283381033484747,
+      "learning_rate": 4.062706336902103e-06,
+      "loss": 0.1816,
+      "step": 24520
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 2.6231464926634684,
+      "learning_rate": 4.047152353060487e-06,
+      "loss": 0.1738,
+      "step": 24540
+    },
+    {
+      "epoch": 2.87,
+      "grad_norm": 3.169329895288349,
+      "learning_rate": 4.031620657970553e-06,
+      "loss": 0.1751,
+      "step": 24560
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 3.054745100410017,
+      "learning_rate": 4.0161113097765655e-06,
+      "loss": 0.1731,
+      "step": 24580
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 2.1135692576460454,
+      "learning_rate": 4.000624366539136e-06,
+      "loss": 0.1724,
+      "step": 24600
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 1.9962103225500736,
+      "learning_rate": 3.985159886234995e-06,
+      "loss": 0.1721,
+      "step": 24620
+    },
+    {
+      "epoch": 2.88,
+      "grad_norm": 3.3958365473507763,
+      "learning_rate": 3.969717926756783e-06,
+      "loss": 0.1764,
+      "step": 24640
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 2.0383790723400077,
+      "learning_rate": 3.954298545912833e-06,
+      "loss": 0.1777,
+      "step": 24660
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 2.7379240709118675,
+      "learning_rate": 3.93890180142695e-06,
+      "loss": 0.1809,
+      "step": 24680
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 2.0472564341865227,
+      "learning_rate": 3.923527750938205e-06,
+      "loss": 0.1754,
+      "step": 24700
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 2.8384107833691874,
+      "learning_rate": 3.908176452000697e-06,
+      "loss": 0.1731,
+      "step": 24720
+    },
+    {
+      "epoch": 2.89,
+      "grad_norm": 2.6243764954263122,
+      "learning_rate": 3.892847962083369e-06,
+      "loss": 0.1791,
+      "step": 24740
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 2.7421433317534447,
+      "learning_rate": 3.877542338569767e-06,
+      "loss": 0.1749,
+      "step": 24760
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 3.215031225548206,
+      "learning_rate": 3.862259638757837e-06,
+      "loss": 0.1749,
+      "step": 24780
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 2.5967047533401417,
+      "learning_rate": 3.846999919859708e-06,
+      "loss": 0.1749,
+      "step": 24800
+    },
+    {
+      "epoch": 2.9,
+      "grad_norm": 3.203524383209205,
+      "learning_rate": 3.831763239001478e-06,
+      "loss": 0.1704,
+      "step": 24820
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 4.166137407385502,
+      "learning_rate": 3.816549653223004e-06,
+      "loss": 0.1847,
+      "step": 24840
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 2.4957440389496837,
+      "learning_rate": 3.801359219477679e-06,
+      "loss": 0.174,
+      "step": 24860
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 2.552841966458733,
+      "learning_rate": 3.7861919946322273e-06,
+      "loss": 0.1823,
+      "step": 24880
+    },
+    {
+      "epoch": 2.91,
+      "grad_norm": 3.4426880959489736,
+      "learning_rate": 3.7710480354664838e-06,
+      "loss": 0.1678,
+      "step": 24900
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 2.4191332648960073,
+      "learning_rate": 3.7559273986731896e-06,
+      "loss": 0.1774,
+      "step": 24920
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 3.6865799612802497,
+      "learning_rate": 3.7408301408577813e-06,
+      "loss": 0.1681,
+      "step": 24940
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 2.135317941259829,
+      "learning_rate": 3.7257563185381663e-06,
+      "loss": 0.1681,
+      "step": 24960
+    },
+    {
+      "epoch": 2.92,
+      "grad_norm": 3.460522766752715,
+      "learning_rate": 3.7107059881445233e-06,
+      "loss": 0.1847,
+      "step": 24980
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 2.320078520103796,
+      "learning_rate": 3.695679206019088e-06,
+      "loss": 0.1775,
+      "step": 25000
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 4.046108335242948,
+      "learning_rate": 3.6806760284159377e-06,
+      "loss": 0.1775,
+      "step": 25020
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 2.7317959202771216,
+      "learning_rate": 3.665696511500789e-06,
+      "loss": 0.1801,
+      "step": 25040
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 3.2533030231293756,
+      "learning_rate": 3.6507407113507725e-06,
+      "loss": 0.169,
+      "step": 25060
+    },
+    {
+      "epoch": 2.93,
+      "grad_norm": 3.2224973327294224,
+      "learning_rate": 3.635808683954245e-06,
+      "loss": 0.18,
+      "step": 25080
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 1.6534753986947448,
+      "learning_rate": 3.6209004852105604e-06,
+      "loss": 0.1709,
+      "step": 25100
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 2.6494173021082297,
+      "learning_rate": 3.6060161709298714e-06,
+      "loss": 0.1736,
+      "step": 25120
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 3.1748671676603415,
+      "learning_rate": 3.5911557968329165e-06,
+      "loss": 0.1733,
+      "step": 25140
+    },
+    {
+      "epoch": 2.94,
+      "grad_norm": 3.125506855868197,
+      "learning_rate": 3.5763194185508114e-06,
+      "loss": 0.1747,
+      "step": 25160
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 3.1171335010105934,
+      "learning_rate": 3.5615070916248402e-06,
+      "loss": 0.1897,
+      "step": 25180
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 2.136778222185654,
+      "learning_rate": 3.5467188715062528e-06,
+      "loss": 0.1711,
+      "step": 25200
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 3.0959637360728443,
+      "learning_rate": 3.5319548135560493e-06,
+      "loss": 0.1781,
+      "step": 25220
+    },
+    {
+      "epoch": 2.95,
+      "grad_norm": 2.5024311642320893,
+      "learning_rate": 3.5172149730447746e-06,
+      "loss": 0.1812,
+      "step": 25240
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 4.167284093251525,
+      "learning_rate": 3.5024994051523154e-06,
+      "loss": 0.1808,
+      "step": 25260
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 2.348151384844568,
+      "learning_rate": 3.4878081649676994e-06,
+      "loss": 0.1718,
+      "step": 25280
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 3.29314656529847,
+      "learning_rate": 3.4731413074888713e-06,
+      "loss": 0.1703,
+      "step": 25300
+    },
+    {
+      "epoch": 2.96,
+      "grad_norm": 2.343550304806459,
+      "learning_rate": 3.4584988876225e-06,
+      "loss": 0.1674,
+      "step": 25320
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 6.679659231025708,
+      "learning_rate": 3.443880960183772e-06,
+      "loss": 0.1747,
+      "step": 25340
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 2.41908162132978,
+      "learning_rate": 3.4292875798961815e-06,
+      "loss": 0.1787,
+      "step": 25360
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 2.3049198356805327,
+      "learning_rate": 3.414718801391333e-06,
+      "loss": 0.1854,
+      "step": 25380
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 2.3512446293219433,
+      "learning_rate": 3.400174679208722e-06,
+      "loss": 0.1708,
+      "step": 25400
+    },
+    {
+      "epoch": 2.97,
+      "grad_norm": 3.478845425392513,
+      "learning_rate": 3.3856552677955525e-06,
+      "loss": 0.17,
+      "step": 25420
+    },
+    {
+      "epoch": 2.98,
+      "grad_norm": 2.268869221394817,
+      "learning_rate": 3.3711606215065136e-06,
+      "loss": 0.1798,
+      "step": 25440
+    },
+    {
+      "epoch": 2.98,
+      "grad_norm": 4.871263294588487,
+      "learning_rate": 3.356690794603589e-06,
+      "loss": 0.1749,
+      "step": 25460
+    },
+    {
+      "epoch": 2.98,
+      "grad_norm": 2.3442633511697126,
+      "learning_rate": 3.342245841255847e-06,
+      "loss": 0.1828,
+      "step": 25480
+    },
+    {
+      "epoch": 2.98,
+      "grad_norm": 3.5581851728863607,
+      "learning_rate": 3.3278258155392384e-06,
+      "loss": 0.1833,
+      "step": 25500
+    },
+    {
+      "epoch": 2.99,
+      "grad_norm": 2.573504752674091,
+      "learning_rate": 3.3134307714363954e-06,
+      "loss": 0.1587,
+      "step": 25520
+    },
+    {
+      "epoch": 2.99,
+      "grad_norm": 2.330468758184399,
+      "learning_rate": 3.2990607628364324e-06,
+      "loss": 0.1722,
+      "step": 25540
+    },
+    {
+      "epoch": 2.99,
+      "grad_norm": 3.2204463895739854,
+      "learning_rate": 3.2847158435347417e-06,
+      "loss": 0.1916,
+      "step": 25560
+    },
+    {
+      "epoch": 2.99,
+      "grad_norm": 2.557634339742982,
+      "learning_rate": 3.270396067232781e-06,
+      "loss": 0.1808,
+      "step": 25580
+    },
+    {
+      "epoch": 3.0,
+      "grad_norm": 3.6599295595821943,
+      "learning_rate": 3.2561014875378948e-06,
+      "loss": 0.1777,
+      "step": 25600
+    },
+    {
+      "epoch": 3.0,
+      "grad_norm": 2.50785271443005,
+      "learning_rate": 3.2418321579630986e-06,
+      "loss": 0.1799,
+      "step": 25620
+    }
+  ],
+  "logging_steps": 20,
+  "max_steps": 34184,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 4,
+  "save_steps": 500,
+  "total_flos": 0.0,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}