diff --git "a/checkpoint-3596/trainer_state.json" "b/checkpoint-3596/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/checkpoint-3596/trainer_state.json"
@@ -0,0 +1,25201 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 1.9834538375973305,
+  "eval_steps": 500,
+  "global_step": 3596,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "grad_norm": 0.02996581234037876,
+      "learning_rate": 2e-05,
+      "loss": 1.2729,
+      "step": 1
+    },
+    {
+      "epoch": 0.0,
+      "grad_norm": 0.029612887650728226,
+      "learning_rate": 4e-05,
+      "loss": 1.208,
+      "step": 2
+    },
+    {
+      "epoch": 0.0,
+      "grad_norm": 0.027333809062838554,
+      "learning_rate": 6e-05,
+      "loss": 1.0525,
+      "step": 3
+    },
+    {
+      "epoch": 0.0,
+      "grad_norm": 0.02706489898264408,
+      "learning_rate": 8e-05,
+      "loss": 1.021,
+      "step": 4
+    },
+    {
+      "epoch": 0.0,
+      "grad_norm": 0.033770810812711716,
+      "learning_rate": 0.0001,
+      "loss": 1.223,
+      "step": 5
+    },
+    {
+      "epoch": 0.0,
+      "grad_norm": 0.03023155778646469,
+      "learning_rate": 0.00012,
+      "loss": 1.0973,
+      "step": 6
+    },
+    {
+      "epoch": 0.0,
+      "grad_norm": 0.034239206463098526,
+      "learning_rate": 0.00014,
+      "loss": 1.1578,
+      "step": 7
+    },
+    {
+      "epoch": 0.0,
+      "grad_norm": 0.037418968975543976,
+      "learning_rate": 0.00016,
+      "loss": 1.0797,
+      "step": 8
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.03651932254433632,
+      "learning_rate": 0.00018,
+      "loss": 1.0795,
+      "step": 9
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.03634285926818848,
+      "learning_rate": 0.0002,
+      "loss": 0.9999,
+      "step": 10
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.03892536088824272,
+      "learning_rate": 0.00019999996162493256,
+      "loss": 1.0103,
+      "step": 11
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.0294953566044569,
+      "learning_rate": 0.00019999984649975977,
+      "loss": 1.0729,
+      "step": 12
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.04408486932516098,
+      "learning_rate": 0.00019999965462456993,
+      "loss": 1.0439,
+      "step": 13
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.03400202468037605,
+      "learning_rate": 0.00019999938599951034,
+      "loss": 0.8412,
+      "step": 14
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.034877073019742966,
+      "learning_rate": 0.00019999904062478714,
+      "loss": 0.9438,
+      "step": 15
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.0340421162545681,
+      "learning_rate": 0.0001999986185006654,
+      "loss": 1.015,
+      "step": 16
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.03420884534716606,
+      "learning_rate": 0.00019999811962746915,
+      "loss": 0.9792,
+      "step": 17
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.03177829086780548,
+      "learning_rate": 0.00019999754400558124,
+      "loss": 0.8856,
+      "step": 18
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.03189516067504883,
+      "learning_rate": 0.00019999689163544346,
+      "loss": 0.8841,
+      "step": 19
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.025729015469551086,
+      "learning_rate": 0.00019999616251755651,
+      "loss": 0.9524,
+      "step": 20
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.02780766598880291,
+      "learning_rate": 0.00019999535665248002,
+      "loss": 0.8998,
+      "step": 21
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.023537974804639816,
+      "learning_rate": 0.00019999447404083244,
+      "loss": 0.9479,
+      "step": 22
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.023690495640039444,
+      "learning_rate": 0.0001999935146832912,
+      "loss": 0.9138,
+      "step": 23
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.023732604458928108,
+      "learning_rate": 0.00019999247858059257,
+      "loss": 1.036,
+      "step": 24
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.02902216836810112,
+      "learning_rate": 0.00019999136573353184,
+      "loss": 0.9619,
+      "step": 25
+    },
+    {
+      "epoch": 0.01,
+      "grad_norm": 0.024132205173373222,
+      "learning_rate": 0.0001999901761429631,
+      "loss": 0.9338,
+      "step": 26
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.02214036136865616,
+      "learning_rate": 0.0001999889098097993,
+      "loss": 0.8229,
+      "step": 27
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.023357391357421875,
+      "learning_rate": 0.0001999875667350124,
+      "loss": 0.8983,
+      "step": 28
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.022214027121663094,
+      "learning_rate": 0.00019998614691963323,
+      "loss": 0.7894,
+      "step": 29
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.021391017362475395,
+      "learning_rate": 0.00019998465036475148,
+      "loss": 0.9419,
+      "step": 30
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.02627565711736679,
+      "learning_rate": 0.0001999830770715157,
+      "loss": 0.9285,
+      "step": 31
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.022590599954128265,
+      "learning_rate": 0.0001999814270411335,
+      "loss": 0.9401,
+      "step": 32
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.024948442354798317,
+      "learning_rate": 0.00019997970027487122,
+      "loss": 0.9069,
+      "step": 33
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.02103595621883869,
+      "learning_rate": 0.00019997789677405416,
+      "loss": 0.9892,
+      "step": 34
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.02005797065794468,
+      "learning_rate": 0.0001999760165400665,
+      "loss": 0.9577,
+      "step": 35
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.021683409810066223,
+      "learning_rate": 0.00019997405957435135,
+      "loss": 0.9099,
+      "step": 36
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.020573126152157784,
+      "learning_rate": 0.00019997202587841066,
+      "loss": 0.8264,
+      "step": 37
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.023178325966000557,
+      "learning_rate": 0.00019996991545380533,
+      "loss": 0.8658,
+      "step": 38
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.01991512067615986,
+      "learning_rate": 0.00019996772830215505,
+      "loss": 0.9239,
+      "step": 39
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.02087959088385105,
+      "learning_rate": 0.00019996546442513854,
+      "loss": 0.8532,
+      "step": 40
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.01916460692882538,
+      "learning_rate": 0.00019996312382449327,
+      "loss": 1.0405,
+      "step": 41
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.023638034239411354,
+      "learning_rate": 0.00019996070650201567,
+      "loss": 0.9984,
+      "step": 42
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.023328714072704315,
+      "learning_rate": 0.000199958212459561,
+      "loss": 0.8696,
+      "step": 43
+    },
+    {
+      "epoch": 0.02,
+      "grad_norm": 0.021662859246134758,
+      "learning_rate": 0.00019995564169904354,
+      "loss": 0.8519,
+      "step": 44
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.024608176201581955,
+      "learning_rate": 0.0001999529942224363,
+      "loss": 0.8428,
+      "step": 45
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.020750664174556732,
+      "learning_rate": 0.00019995027003177118,
+      "loss": 0.8653,
+      "step": 46
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.021747557446360588,
+      "learning_rate": 0.00019994746912913906,
+      "loss": 0.8973,
+      "step": 47
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.02252582646906376,
+      "learning_rate": 0.00019994459151668957,
+      "loss": 0.8664,
+      "step": 48
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.0229500625282526,
+      "learning_rate": 0.00019994163719663137,
+      "loss": 0.8613,
+      "step": 49
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.021505488082766533,
+      "learning_rate": 0.00019993860617123184,
+      "loss": 0.812,
+      "step": 50
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.020829355344176292,
+      "learning_rate": 0.0001999354984428173,
+      "loss": 0.9139,
+      "step": 51
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.024170633405447006,
+      "learning_rate": 0.000199932314013773,
+      "loss": 0.937,
+      "step": 52
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.021097104996442795,
+      "learning_rate": 0.0001999290528865429,
+      "loss": 0.9093,
+      "step": 53
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.023191392421722412,
+      "learning_rate": 0.00019992571506363,
+      "loss": 0.9728,
+      "step": 54
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.023909488692879677,
+      "learning_rate": 0.000199922300547596,
+      "loss": 0.8279,
+      "step": 55
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.024314479902386665,
+      "learning_rate": 0.0001999188093410616,
+      "loss": 0.8118,
+      "step": 56
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.022049203515052795,
+      "learning_rate": 0.00019991524144670635,
+      "loss": 0.8637,
+      "step": 57
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.030246440321207047,
+      "learning_rate": 0.0001999115968672685,
+      "loss": 0.8515,
+      "step": 58
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.021488850936293602,
+      "learning_rate": 0.00019990787560554538,
+      "loss": 0.844,
+      "step": 59
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.022307060658931732,
+      "learning_rate": 0.00019990407766439297,
+      "loss": 0.7771,
+      "step": 60
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.027180947363376617,
+      "learning_rate": 0.00019990020304672628,
+      "loss": 0.9951,
+      "step": 61
+    },
+    {
+      "epoch": 0.03,
+      "grad_norm": 0.02594638243317604,
+      "learning_rate": 0.000199896251755519,
+      "loss": 0.8643,
+      "step": 62
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.025260182097554207,
+      "learning_rate": 0.00019989222379380384,
+      "loss": 0.9858,
+      "step": 63
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.029186906293034554,
+      "learning_rate": 0.0001998881191646722,
+      "loss": 0.7514,
+      "step": 64
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.02333957888185978,
+      "learning_rate": 0.00019988393787127441,
+      "loss": 0.7914,
+      "step": 65
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.0235314778983593,
+      "learning_rate": 0.00019987967991681965,
+      "loss": 0.8198,
+      "step": 66
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.023193303495645523,
+      "learning_rate": 0.00019987534530457583,
+      "loss": 0.8976,
+      "step": 67
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.026709623634815216,
+      "learning_rate": 0.00019987093403786986,
+      "loss": 0.8833,
+      "step": 68
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.024577545002102852,
+      "learning_rate": 0.00019986644612008728,
+      "loss": 0.8259,
+      "step": 69
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.027963118627667427,
+      "learning_rate": 0.0001998618815546727,
+      "loss": 0.8767,
+      "step": 70
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.02565196342766285,
+      "learning_rate": 0.00019985724034512936,
+      "loss": 0.7295,
+      "step": 71
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.026185255497694016,
+      "learning_rate": 0.0001998525224950194,
+      "loss": 0.876,
+      "step": 72
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.023345299065113068,
+      "learning_rate": 0.00019984772800796377,
+      "loss": 0.997,
+      "step": 73
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.024012910202145576,
+      "learning_rate": 0.00019984285688764226,
+      "loss": 0.9337,
+      "step": 74
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.02843514457345009,
+      "learning_rate": 0.00019983790913779347,
+      "loss": 0.8949,
+      "step": 75
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.022462787106633186,
+      "learning_rate": 0.0001998328847622148,
+      "loss": 0.9018,
+      "step": 76
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.024008071050047874,
+      "learning_rate": 0.00019982778376476245,
+      "loss": 0.8376,
+      "step": 77
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.023886198177933693,
+      "learning_rate": 0.00019982260614935144,
+      "loss": 0.8118,
+      "step": 78
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.025386439636349678,
+      "learning_rate": 0.00019981735191995563,
+      "loss": 0.9161,
+      "step": 79
+    },
+    {
+      "epoch": 0.04,
+      "grad_norm": 0.02613622136414051,
+      "learning_rate": 0.0001998120210806076,
+      "loss": 0.9008,
+      "step": 80
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.02593972161412239,
+      "learning_rate": 0.00019980661363539883,
+      "loss": 0.9567,
+      "step": 81
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.02825133316218853,
+      "learning_rate": 0.0001998011295884795,
+      "loss": 0.8386,
+      "step": 82
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.025737276300787926,
+      "learning_rate": 0.00019979556894405862,
+      "loss": 0.7957,
+      "step": 83
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.023754647001624107,
+      "learning_rate": 0.00019978993170640403,
+      "loss": 0.8167,
+      "step": 84
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.02431335486471653,
+      "learning_rate": 0.00019978421787984228,
+      "loss": 0.7568,
+      "step": 85
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.024608569219708443,
+      "learning_rate": 0.00019977842746875878,
+      "loss": 0.7992,
+      "step": 86
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.024010393768548965,
+      "learning_rate": 0.00019977256047759765,
+      "loss": 0.9094,
+      "step": 87
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.023806700482964516,
+      "learning_rate": 0.00019976661691086182,
+      "loss": 0.8677,
+      "step": 88
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.02781900390982628,
+      "learning_rate": 0.00019976059677311297,
+      "loss": 0.7736,
+      "step": 89
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.027234690263867378,
+      "learning_rate": 0.0001997545000689716,
+      "loss": 0.789,
+      "step": 90
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.028865503147244453,
+      "learning_rate": 0.00019974832680311688,
+      "loss": 0.9041,
+      "step": 91
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.05750074237585068,
+      "learning_rate": 0.00019974207698028685,
+      "loss": 0.8886,
+      "step": 92
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.025579992681741714,
+      "learning_rate": 0.00019973575060527825,
+      "loss": 0.835,
+      "step": 93
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.02479216828942299,
+      "learning_rate": 0.00019972934768294659,
+      "loss": 0.765,
+      "step": 94
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.024221569299697876,
+      "learning_rate": 0.0001997228682182061,
+      "loss": 0.7935,
+      "step": 95
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.025168055668473244,
+      "learning_rate": 0.00019971631221602976,
+      "loss": 0.7801,
+      "step": 96
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.027467774227261543,
+      "learning_rate": 0.00019970967968144937,
+      "loss": 0.8287,
+      "step": 97
+    },
+    {
+      "epoch": 0.05,
+      "grad_norm": 0.023378336802124977,
+      "learning_rate": 0.00019970297061955533,
+      "loss": 0.7486,
+      "step": 98
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.025852292776107788,
+      "learning_rate": 0.00019969618503549693,
+      "loss": 0.8939,
+      "step": 99
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.02506246417760849,
+      "learning_rate": 0.00019968932293448207,
+      "loss": 0.8112,
+      "step": 100
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.026856260374188423,
+      "learning_rate": 0.00019968238432177744,
+      "loss": 0.8782,
+      "step": 101
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.028643622994422913,
+      "learning_rate": 0.0001996753692027084,
+      "loss": 0.7884,
+      "step": 102
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.029276320710778236,
+      "learning_rate": 0.00019966827758265913,
+      "loss": 0.8926,
+      "step": 103
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.027838734909892082,
+      "learning_rate": 0.00019966110946707244,
+      "loss": 0.87,
+      "step": 104
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.026706190779805183,
+      "learning_rate": 0.0001996538648614498,
+      "loss": 0.9119,
+      "step": 105
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.024687886238098145,
+      "learning_rate": 0.00019964654377135153,
+      "loss": 0.7647,
+      "step": 106
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.02482466958463192,
+      "learning_rate": 0.00019963914620239656,
+      "loss": 0.894,
+      "step": 107
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.024776702746748924,
+      "learning_rate": 0.0001996316721602625,
+      "loss": 0.7515,
+      "step": 108
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.02663383074104786,
+      "learning_rate": 0.00019962412165068573,
+      "loss": 0.8499,
+      "step": 109
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.027590934187173843,
+      "learning_rate": 0.00019961649467946125,
+      "loss": 0.7446,
+      "step": 110
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.025346562266349792,
+      "learning_rate": 0.0001996087912524428,
+      "loss": 0.8308,
+      "step": 111
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.02552192285656929,
+      "learning_rate": 0.0001996010113755427,
+      "loss": 0.8552,
+      "step": 112
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.024720516055822372,
+      "learning_rate": 0.0001995931550547321,
+      "loss": 0.8648,
+      "step": 113
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.02904409170150757,
+      "learning_rate": 0.0001995852222960407,
+      "loss": 0.8826,
+      "step": 114
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.027031444013118744,
+      "learning_rate": 0.00019957721310555693,
+      "loss": 0.8748,
+      "step": 115
+    },
+    {
+      "epoch": 0.06,
+      "grad_norm": 0.025683356449007988,
+      "learning_rate": 0.0001995691274894278,
+      "loss": 0.8441,
+      "step": 116
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.03195862099528313,
+      "learning_rate": 0.00019956096545385905,
+      "loss": 0.7829,
+      "step": 117
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.025551646947860718,
+      "learning_rate": 0.00019955272700511507,
+      "loss": 0.7934,
+      "step": 118
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.028811989352107048,
+      "learning_rate": 0.0001995444121495189,
+      "loss": 0.8001,
+      "step": 119
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.030996061861515045,
+      "learning_rate": 0.00019953602089345217,
+      "loss": 0.7128,
+      "step": 120
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.026600060984492302,
+      "learning_rate": 0.00019952755324335514,
+      "loss": 0.7414,
+      "step": 121
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.025535378605127335,
+      "learning_rate": 0.00019951900920572684,
+      "loss": 0.8381,
+      "step": 122
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.035039834678173065,
+      "learning_rate": 0.00019951038878712475,
+      "loss": 0.9607,
+      "step": 123
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.03230242058634758,
+      "learning_rate": 0.00019950169199416513,
+      "loss": 0.876,
+      "step": 124
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.031107056885957718,
+      "learning_rate": 0.0001994929188335227,
+      "loss": 0.8444,
+      "step": 125
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.031076829880475998,
+      "learning_rate": 0.0001994840693119309,
+      "loss": 0.8577,
+      "step": 126
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.028398986905813217,
+      "learning_rate": 0.0001994751434361818,
+      "loss": 0.7828,
+      "step": 127
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.027821028605103493,
+      "learning_rate": 0.00019946614121312594,
+      "loss": 0.8243,
+      "step": 128
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.030113285407423973,
+      "learning_rate": 0.0001994570626496726,
+      "loss": 0.7923,
+      "step": 129
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.028168270364403725,
+      "learning_rate": 0.00019944790775278955,
+      "loss": 0.722,
+      "step": 130
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.030299954116344452,
+      "learning_rate": 0.0001994386765295032,
+      "loss": 0.8462,
+      "step": 131
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.029045602306723595,
+      "learning_rate": 0.00019942936898689854,
+      "loss": 0.7916,
+      "step": 132
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.02803182788193226,
+      "learning_rate": 0.0001994199851321191,
+      "loss": 0.7949,
+      "step": 133
+    },
+    {
+      "epoch": 0.07,
+      "grad_norm": 0.029968315735459328,
+      "learning_rate": 0.00019941052497236703,
+      "loss": 0.9672,
+      "step": 134
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.027893947437405586,
+      "learning_rate": 0.00019940098851490298,
+      "loss": 0.8284,
+      "step": 135
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.030356327071785927,
+      "learning_rate": 0.0001993913757670462,
+      "loss": 0.8001,
+      "step": 136
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.02701139822602272,
+      "learning_rate": 0.00019938168673617454,
+      "loss": 0.8137,
+      "step": 137
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.032546404749155045,
+      "learning_rate": 0.00019937192142972427,
+      "loss": 0.7451,
+      "step": 138
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.02752624824643135,
+      "learning_rate": 0.0001993620798551903,
+      "loss": 0.9893,
+      "step": 139
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.02512490376830101,
+      "learning_rate": 0.0001993521620201261,
+      "loss": 0.9108,
+      "step": 140
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.029267175123095512,
+      "learning_rate": 0.00019934216793214354,
+      "loss": 0.8274,
+      "step": 141
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.02983001247048378,
+      "learning_rate": 0.00019933209759891317,
+      "loss": 0.8394,
+      "step": 142
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.031017007306218147,
+      "learning_rate": 0.0001993219510281639,
+      "loss": 0.8455,
+      "step": 143
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.02880048379302025,
+      "learning_rate": 0.00019931172822768335,
+      "loss": 0.8257,
+      "step": 144
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.03203625977039337,
+      "learning_rate": 0.00019930142920531743,
+      "loss": 0.8693,
+      "step": 145
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.027382776141166687,
+      "learning_rate": 0.0001992910539689707,
+      "loss": 0.8478,
+      "step": 146
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.03054049052298069,
+      "learning_rate": 0.00019928060252660617,
+      "loss": 0.9593,
+      "step": 147
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.026648158207535744,
+      "learning_rate": 0.00019927007488624535,
+      "loss": 0.8915,
+      "step": 148
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.03315131738781929,
+      "learning_rate": 0.00019925947105596816,
+      "loss": 0.9008,
+      "step": 149
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.027361949905753136,
+      "learning_rate": 0.00019924879104391309,
+      "loss": 0.8705,
+      "step": 150
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.029892416670918465,
+      "learning_rate": 0.0001992380348582771,
+      "loss": 0.9083,
+      "step": 151
+    },
+    {
+      "epoch": 0.08,
+      "grad_norm": 0.027648137882351875,
+      "learning_rate": 0.00019922720250731553,
+      "loss": 0.7381,
+      "step": 152
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.02656717598438263,
+      "learning_rate": 0.00019921629399934223,
+      "loss": 0.9495,
+      "step": 153
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.027288291603326797,
+      "learning_rate": 0.0001992053093427295,
+      "loss": 0.8505,
+      "step": 154
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.032024212181568146,
+      "learning_rate": 0.0001991942485459081,
+      "loss": 0.8288,
+      "step": 155
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.027418840676546097,
+      "learning_rate": 0.00019918311161736717,
+      "loss": 0.8137,
+      "step": 156
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.03174857795238495,
+      "learning_rate": 0.0001991718985656543,
+      "loss": 0.8754,
+      "step": 157
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.02887050248682499,
+      "learning_rate": 0.0001991606093993756,
+      "loss": 0.8497,
+      "step": 158
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.03064458630979061,
+      "learning_rate": 0.00019914924412719546,
+      "loss": 0.9331,
+      "step": 159
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.02989426627755165,
+      "learning_rate": 0.00019913780275783674,
+      "loss": 0.9075,
+      "step": 160
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.03449084982275963,
+      "learning_rate": 0.00019912628530008075,
+      "loss": 0.8531,
+      "step": 161
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.03213748335838318,
+      "learning_rate": 0.00019911469176276712,
+      "loss": 0.7411,
+      "step": 162
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.031524017453193665,
+      "learning_rate": 0.0001991030221547939,
+      "loss": 0.8003,
+      "step": 163
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.028880394995212555,
+      "learning_rate": 0.00019909127648511755,
+      "loss": 0.86,
+      "step": 164
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.0295130405575037,
+      "learning_rate": 0.0001990794547627529,
+      "loss": 0.802,
+      "step": 165
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.032397400587797165,
+      "learning_rate": 0.00019906755699677312,
+      "loss": 0.8574,
+      "step": 166
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.030780531466007233,
+      "learning_rate": 0.00019905558319630978,
+      "loss": 0.8748,
+      "step": 167
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.026546768844127655,
+      "learning_rate": 0.0001990435333705527,
+      "loss": 0.8099,
+      "step": 168
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.03083784319460392,
+      "learning_rate": 0.00019903140752875026,
+      "loss": 0.7867,
+      "step": 169
+    },
+    {
+      "epoch": 0.09,
+      "grad_norm": 0.02955833077430725,
+      "learning_rate": 0.000199019205680209,
+      "loss": 0.8806,
+      "step": 170
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.029685625806450844,
+      "learning_rate": 0.00019900692783429385,
+      "loss": 0.9009,
+      "step": 171
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.03194108605384827,
+      "learning_rate": 0.00019899457400042808,
+      "loss": 0.7654,
+      "step": 172
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.032116640359163284,
+      "learning_rate": 0.0001989821441880933,
+      "loss": 1.0055,
+      "step": 173
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.03061555325984955,
+      "learning_rate": 0.00019896963840682937,
+      "loss": 0.782,
+      "step": 174
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.030177190899848938,
+      "learning_rate": 0.0001989570566662345,
+      "loss": 0.7403,
+      "step": 175
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.031266387552022934,
+      "learning_rate": 0.00019894439897596522,
+      "loss": 0.8553,
+      "step": 176
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.032219842076301575,
+      "learning_rate": 0.0001989316653457363,
+      "loss": 0.7479,
+      "step": 177
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.029450630769133568,
+      "learning_rate": 0.0001989188557853208,
+      "loss": 0.7373,
+      "step": 178
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.03027605637907982,
+      "learning_rate": 0.00019890597030455015,
+      "loss": 0.7844,
+      "step": 179
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.03511941432952881,
+      "learning_rate": 0.00019889300891331392,
+      "loss": 0.8679,
+      "step": 180
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.031290799379348755,
+      "learning_rate": 0.00019887997162155996,
+      "loss": 0.7283,
+      "step": 181
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.02689247764647007,
+      "learning_rate": 0.0001988668584392945,
+      "loss": 0.7628,
+      "step": 182
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.030273688957095146,
+      "learning_rate": 0.0001988536693765818,
+      "loss": 0.7985,
+      "step": 183
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.030390895903110504,
+      "learning_rate": 0.00019884040444354461,
+      "loss": 0.8086,
+      "step": 184
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.029167182743549347,
+      "learning_rate": 0.00019882706365036373,
+      "loss": 0.7081,
+      "step": 185
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.03029320389032364,
+      "learning_rate": 0.00019881364700727823,
+      "loss": 0.6738,
+      "step": 186
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.028749890625476837,
+      "learning_rate": 0.00019880015452458544,
+      "loss": 0.812,
+      "step": 187
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 0.030206618830561638,
+      "learning_rate": 0.0001987865862126408,
+      "loss": 0.8358,
+      "step": 188
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.030770782381296158,
+      "learning_rate": 0.00019877294208185803,
+      "loss": 0.7326,
+      "step": 189
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.03188487887382507,
+      "learning_rate": 0.00019875922214270903,
+      "loss": 0.817,
+      "step": 190
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.031357571482658386,
+      "learning_rate": 0.00019874542640572384,
+      "loss": 0.7981,
+      "step": 191
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.06510663032531738,
+      "learning_rate": 0.00019873155488149078,
+      "loss": 0.8416,
+      "step": 192
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.030021656304597855,
+      "learning_rate": 0.0001987176075806562,
+      "loss": 0.7412,
+      "step": 193
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.033732689917087555,
+      "learning_rate": 0.00019870358451392467,
+      "loss": 1.0041,
+      "step": 194
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.02999632991850376,
+      "learning_rate": 0.00019868948569205897,
+      "loss": 0.7975,
+      "step": 195
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.02894209697842598,
+      "learning_rate": 0.0001986753111258799,
+      "loss": 0.7465,
+      "step": 196
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.033549994230270386,
+      "learning_rate": 0.0001986610608262665,
+      "loss": 0.8114,
+      "step": 197
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.03032383695244789,
+      "learning_rate": 0.00019864673480415589,
+      "loss": 0.7155,
+      "step": 198
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.03509935364127159,
+      "learning_rate": 0.00019863233307054325,
+      "loss": 0.8527,
+      "step": 199
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.03238251060247421,
+      "learning_rate": 0.00019861785563648202,
+      "loss": 0.7574,
+      "step": 200
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.03257328271865845,
+      "learning_rate": 0.00019860330251308362,
+      "loss": 0.9374,
+      "step": 201
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.03191230073571205,
+      "learning_rate": 0.00019858867371151754,
+      "loss": 0.8127,
+      "step": 202
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.030960606411099434,
+      "learning_rate": 0.0001985739692430115,
+      "loss": 0.7888,
+      "step": 203
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.03744737431406975,
+      "learning_rate": 0.0001985591891188511,
+      "loss": 0.8152,
+      "step": 204
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.03747240826487541,
+      "learning_rate": 0.0001985443333503802,
+      "loss": 0.7457,
+      "step": 205
+    },
+    {
+      "epoch": 0.11,
+      "grad_norm": 0.030524997040629387,
+      "learning_rate": 0.00019852940194900053,
+      "loss": 0.8093,
+      "step": 206
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.032599907368421555,
+      "learning_rate": 0.00019851439492617204,
+      "loss": 0.7419,
+      "step": 207
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.0342191606760025,
+      "learning_rate": 0.00019849931229341258,
+      "loss": 0.7183,
+      "step": 208
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.03359980136156082,
+      "learning_rate": 0.00019848415406229812,
+      "loss": 0.8392,
+      "step": 209
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.03139995038509369,
+      "learning_rate": 0.00019846892024446265,
+      "loss": 0.7657,
+      "step": 210
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.029806002974510193,
+      "learning_rate": 0.00019845361085159806,
+      "loss": 0.6918,
+      "step": 211
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.03195658698678017,
+      "learning_rate": 0.00019843822589545441,
+      "loss": 0.8361,
+      "step": 212
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.03134671226143837,
+      "learning_rate": 0.00019842276538783966,
+      "loss": 0.9247,
+      "step": 213
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.03182549774646759,
+      "learning_rate": 0.00019840722934061974,
+      "loss": 0.8478,
+      "step": 214
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.035036999732255936,
+      "learning_rate": 0.00019839161776571864,
+      "loss": 0.7699,
+      "step": 215
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.030924562364816666,
+      "learning_rate": 0.00019837593067511823,
+      "loss": 0.8543,
+      "step": 216
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.03198299929499626,
+      "learning_rate": 0.00019836016808085836,
+      "loss": 0.761,
+      "step": 217
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.03266490250825882,
+      "learning_rate": 0.00019834432999503684,
+      "loss": 0.7841,
+      "step": 218
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.03606751561164856,
+      "learning_rate": 0.00019832841642980945,
+      "loss": 0.8398,
+      "step": 219
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.03128010407090187,
+      "learning_rate": 0.00019831242739738985,
+      "loss": 0.7795,
+      "step": 220
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.031938567757606506,
+      "learning_rate": 0.00019829636291004968,
+      "loss": 0.7383,
+      "step": 221
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.03146064281463623,
+      "learning_rate": 0.00019828022298011842,
+      "loss": 0.8463,
+      "step": 222
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.03626205027103424,
+      "learning_rate": 0.00019826400761998353,
+      "loss": 0.9204,
+      "step": 223
+    },
+    {
+      "epoch": 0.12,
+      "grad_norm": 0.03110690787434578,
+      "learning_rate": 0.00019824771684209025,
+      "loss": 0.7676,
+      "step": 224
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.0522528775036335,
+      "learning_rate": 0.00019823135065894185,
+      "loss": 0.8221,
+      "step": 225
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.031512025743722916,
+      "learning_rate": 0.00019821490908309932,
+      "loss": 0.745,
+      "step": 226
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.03326869383454323,
+      "learning_rate": 0.00019819839212718168,
+      "loss": 0.8885,
+      "step": 227
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.03730284795165062,
+      "learning_rate": 0.00019818179980386563,
+      "loss": 0.7942,
+      "step": 228
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.03505445271730423,
+      "learning_rate": 0.00019816513212588585,
+      "loss": 0.7352,
+      "step": 229
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.03741452097892761,
+      "learning_rate": 0.00019814838910603481,
+      "loss": 0.9211,
+      "step": 230
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.03283902630209923,
+      "learning_rate": 0.00019813157075716277,
+      "loss": 0.9182,
+      "step": 231
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.0364568717777729,
+      "learning_rate": 0.0001981146770921779,
+      "loss": 0.9211,
+      "step": 232
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.03377554938197136,
+      "learning_rate": 0.000198097708124046,
+      "loss": 0.8209,
+      "step": 233
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.030919229611754417,
+      "learning_rate": 0.00019808066386579085,
+      "loss": 0.7564,
+      "step": 234
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.032216716557741165,
+      "learning_rate": 0.00019806354433049393,
+      "loss": 0.8913,
+      "step": 235
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.038493942469358444,
+      "learning_rate": 0.00019804634953129454,
+      "loss": 0.701,
+      "step": 236
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.047857046127319336,
+      "learning_rate": 0.00019802907948138963,
+      "loss": 1.0259,
+      "step": 237
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.0336415059864521,
+      "learning_rate": 0.00019801173419403405,
+      "loss": 0.9172,
+      "step": 238
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.030060969293117523,
+      "learning_rate": 0.00019799431368254034,
+      "loss": 0.8397,
+      "step": 239
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.043135106563568115,
+      "learning_rate": 0.0001979768179602787,
+      "loss": 0.9176,
+      "step": 240
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.03419271484017372,
+      "learning_rate": 0.00019795924704067721,
+      "loss": 0.9116,
+      "step": 241
+    },
+    {
+      "epoch": 0.13,
+      "grad_norm": 0.03154682740569115,
+      "learning_rate": 0.0001979416009372215,
+      "loss": 0.7786,
+      "step": 242
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.03246486559510231,
+      "learning_rate": 0.000197923879663455,
+      "loss": 0.7744,
+      "step": 243
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.03503431752324104,
+      "learning_rate": 0.0001979060832329788,
+      "loss": 0.7365,
+      "step": 244
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.03330378606915474,
+      "learning_rate": 0.00019788821165945172,
+      "loss": 0.6759,
+      "step": 245
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.030394921079277992,
+      "learning_rate": 0.00019787026495659022,
+      "loss": 0.7804,
+      "step": 246
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.032463330775499344,
+      "learning_rate": 0.00019785224313816836,
+      "loss": 0.8431,
+      "step": 247
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.032576996833086014,
+      "learning_rate": 0.00019783414621801798,
+      "loss": 0.787,
+      "step": 248
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.03479291498661041,
+      "learning_rate": 0.00019781597421002843,
+      "loss": 0.8343,
+      "step": 249
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.03418460860848427,
+      "learning_rate": 0.00019779772712814677,
+      "loss": 0.8228,
+      "step": 250
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.03314866125583649,
+      "learning_rate": 0.0001977794049863777,
+      "loss": 0.8703,
+      "step": 251
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.031179893761873245,
+      "learning_rate": 0.00019776100779878345,
+      "loss": 0.6981,
+      "step": 252
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.03774740546941757,
+      "learning_rate": 0.00019774253557948386,
+      "loss": 0.5688,
+      "step": 253
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.033188916742801666,
+      "learning_rate": 0.00019772398834265643,
+      "loss": 0.7994,
+      "step": 254
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.03498241677880287,
+      "learning_rate": 0.00019770536610253618,
+      "loss": 0.7059,
+      "step": 255
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.035760391503572464,
+      "learning_rate": 0.00019768666887341567,
+      "loss": 0.8468,
+      "step": 256
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.0330347940325737,
+      "learning_rate": 0.0001976678966696451,
+      "loss": 0.7864,
+      "step": 257
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.036673109978437424,
+      "learning_rate": 0.00019764904950563214,
+      "loss": 0.867,
+      "step": 258
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.03514648601412773,
+      "learning_rate": 0.00019763012739584205,
+      "loss": 0.9021,
+      "step": 259
+    },
+    {
+      "epoch": 0.14,
+      "grad_norm": 0.03706090897321701,
+      "learning_rate": 0.00019761113035479747,
+      "loss": 0.7132,
+      "step": 260
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.033022210001945496,
+      "learning_rate": 0.00019759205839707877,
+      "loss": 0.8326,
+      "step": 261
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.03423427790403366,
+      "learning_rate": 0.00019757291153732362,
+      "loss": 0.9028,
+      "step": 262
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.03209824860095978,
+      "learning_rate": 0.00019755368979022732,
+      "loss": 0.7991,
+      "step": 263
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.03312317654490471,
+      "learning_rate": 0.00019753439317054256,
+      "loss": 0.7846,
+      "step": 264
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.03310060128569603,
+      "learning_rate": 0.00019751502169307954,
+      "loss": 0.7369,
+      "step": 265
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.03352689370512962,
+      "learning_rate": 0.00019749557537270584,
+      "loss": 0.8118,
+      "step": 266
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.037001077085733414,
+      "learning_rate": 0.00019747605422434662,
+      "loss": 0.7978,
+      "step": 267
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.03153040632605553,
+      "learning_rate": 0.00019745645826298432,
+      "loss": 0.6908,
+      "step": 268
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.029728813096880913,
+      "learning_rate": 0.00019743678750365888,
+      "loss": 0.7491,
+      "step": 269
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.04103512689471245,
+      "learning_rate": 0.00019741704196146766,
+      "loss": 0.8505,
+      "step": 270
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.036623742431402206,
+      "learning_rate": 0.00019739722165156538,
+      "loss": 0.8811,
+      "step": 271
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.031018711626529694,
+      "learning_rate": 0.00019737732658916412,
+      "loss": 0.7914,
+      "step": 272
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.0346776582300663,
+      "learning_rate": 0.00019735735678953344,
+      "loss": 0.8786,
+      "step": 273
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.035631489008665085,
+      "learning_rate": 0.00019733731226800015,
+      "loss": 0.7063,
+      "step": 274
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.034868933260440826,
+      "learning_rate": 0.0001973171930399484,
+      "loss": 0.7211,
+      "step": 275
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.030991340056061745,
+      "learning_rate": 0.00019729699912081979,
+      "loss": 0.7908,
+      "step": 276
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.034880708903074265,
+      "learning_rate": 0.00019727673052611315,
+      "loss": 0.6827,
+      "step": 277
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 0.03172425180673599,
+      "learning_rate": 0.00019725638727138468,
+      "loss": 0.7884,
+      "step": 278
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.035629045218229294,
+      "learning_rate": 0.00019723596937224781,
+      "loss": 0.8101,
+      "step": 279
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.03561673313379288,
+      "learning_rate": 0.00019721547684437333,
+      "loss": 0.8164,
+      "step": 280
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.0325719490647316,
+      "learning_rate": 0.0001971949097034893,
+      "loss": 0.7356,
+      "step": 281
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.04635859653353691,
+      "learning_rate": 0.00019717426796538102,
+      "loss": 0.9928,
+      "step": 282
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.0328851044178009,
+      "learning_rate": 0.00019715355164589106,
+      "loss": 0.7742,
+      "step": 283
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.032772939652204514,
+      "learning_rate": 0.00019713276076091917,
+      "loss": 0.7487,
+      "step": 284
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.03649919852614403,
+      "learning_rate": 0.00019711189532642243,
+      "loss": 0.7322,
+      "step": 285
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.03704332187771797,
+      "learning_rate": 0.0001970909553584151,
+      "loss": 0.7951,
+      "step": 286
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.03600141033530235,
+      "learning_rate": 0.00019706994087296859,
+      "loss": 0.8053,
+      "step": 287
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.03415602818131447,
+      "learning_rate": 0.0001970488518862116,
+      "loss": 0.8079,
+      "step": 288
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.032455701380968094,
+      "learning_rate": 0.0001970276884143299,
+      "loss": 0.8268,
+      "step": 289
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.03594019636511803,
+      "learning_rate": 0.00019700645047356652,
+      "loss": 0.7773,
+      "step": 290
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.03449614718556404,
+      "learning_rate": 0.0001969851380802216,
+      "loss": 0.8421,
+      "step": 291
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.035057246685028076,
+      "learning_rate": 0.0001969637512506524,
+      "loss": 0.7891,
+      "step": 292
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.03100276179611683,
+      "learning_rate": 0.00019694229000127337,
+      "loss": 0.7188,
+      "step": 293
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.035093460232019424,
+      "learning_rate": 0.00019692075434855606,
+      "loss": 0.6879,
+      "step": 294
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.03476954251527786,
+      "learning_rate": 0.0001968991443090291,
+      "loss": 0.8428,
+      "step": 295
+    },
+    {
+      "epoch": 0.16,
+      "grad_norm": 0.03413707762956619,
+      "learning_rate": 0.00019687745989927823,
+      "loss": 0.8469,
+      "step": 296
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.03546985611319542,
+      "learning_rate": 0.00019685570113594624,
+      "loss": 0.8675,
+      "step": 297
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.036925192922353745,
+      "learning_rate": 0.00019683386803573302,
+      "loss": 0.7033,
+      "step": 298
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.03350600227713585,
+      "learning_rate": 0.00019681196061539552,
+      "loss": 0.7405,
+      "step": 299
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.03388972207903862,
+      "learning_rate": 0.00019678997889174772,
+      "loss": 0.7672,
+      "step": 300
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.03225654736161232,
+      "learning_rate": 0.00019676792288166056,
+      "loss": 0.8161,
+      "step": 301
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.035303112119436264,
+      "learning_rate": 0.00019674579260206215,
+      "loss": 0.8378,
+      "step": 302
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.03430440276861191,
+      "learning_rate": 0.00019672358806993744,
+      "loss": 0.8755,
+      "step": 303
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.03425530344247818,
+      "learning_rate": 0.00019670130930232849,
+      "loss": 0.8579,
+      "step": 304
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.03408996760845184,
+      "learning_rate": 0.00019667895631633427,
+      "loss": 0.7241,
+      "step": 305
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.034297868609428406,
+      "learning_rate": 0.00019665652912911067,
+      "loss": 0.7586,
+      "step": 306
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.03555282577872276,
+      "learning_rate": 0.00019663402775787066,
+      "loss": 0.7913,
+      "step": 307
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.046342890709638596,
+      "learning_rate": 0.000196611452219884,
+      "loss": 0.756,
+      "step": 308
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.035010941326618195,
+      "learning_rate": 0.00019658880253247752,
+      "loss": 0.7984,
+      "step": 309
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.034292541444301605,
+      "learning_rate": 0.00019656607871303486,
+      "loss": 0.7529,
+      "step": 310
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.034864772111177444,
+      "learning_rate": 0.00019654328077899655,
+      "loss": 0.7659,
+      "step": 311
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.03819848969578743,
+      "learning_rate": 0.0001965204087478601,
+      "loss": 0.8601,
+      "step": 312
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.03443516790866852,
+      "learning_rate": 0.00019649746263717974,
+      "loss": 0.755,
+      "step": 313
+    },
+    {
+      "epoch": 0.17,
+      "grad_norm": 0.039128776639699936,
+      "learning_rate": 0.00019647444246456672,
+      "loss": 0.8283,
+      "step": 314
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03328540921211243,
+      "learning_rate": 0.000196451348247689,
+      "loss": 0.8621,
+      "step": 315
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03752969205379486,
+      "learning_rate": 0.00019642818000427145,
+      "loss": 0.6716,
+      "step": 316
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.033460091799497604,
+      "learning_rate": 0.00019640493775209574,
+      "loss": 0.644,
+      "step": 317
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03952079266309738,
+      "learning_rate": 0.00019638162150900027,
+      "loss": 0.8444,
+      "step": 318
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03375870734453201,
+      "learning_rate": 0.00019635823129288038,
+      "loss": 0.7582,
+      "step": 319
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.0385657362639904,
+      "learning_rate": 0.00019633476712168803,
+      "loss": 0.6819,
+      "step": 320
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03305663540959358,
+      "learning_rate": 0.000196311229013432,
+      "loss": 0.769,
+      "step": 321
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.033982716500759125,
+      "learning_rate": 0.00019628761698617782,
+      "loss": 0.8634,
+      "step": 322
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.035749975591897964,
+      "learning_rate": 0.00019626393105804776,
+      "loss": 0.7829,
+      "step": 323
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03398462384939194,
+      "learning_rate": 0.00019624017124722086,
+      "loss": 0.803,
+      "step": 324
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03670307993888855,
+      "learning_rate": 0.0001962163375719327,
+      "loss": 0.9054,
+      "step": 325
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.034984368830919266,
+      "learning_rate": 0.00019619243005047576,
+      "loss": 0.7222,
+      "step": 326
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03537537157535553,
+      "learning_rate": 0.00019616844870119904,
+      "loss": 0.8563,
+      "step": 327
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03322187438607216,
+      "learning_rate": 0.00019614439354250824,
+      "loss": 0.8273,
+      "step": 328
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03316264599561691,
+      "learning_rate": 0.00019612026459286578,
+      "loss": 0.6768,
+      "step": 329
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03384651243686676,
+      "learning_rate": 0.00019609606187079065,
+      "loss": 0.7345,
+      "step": 330
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03392638638615608,
+      "learning_rate": 0.0001960717853948584,
+      "loss": 0.7833,
+      "step": 331
+    },
+    {
+      "epoch": 0.18,
+      "grad_norm": 0.03634137287735939,
+      "learning_rate": 0.00019604743518370133,
+      "loss": 0.8709,
+      "step": 332
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.03778916969895363,
+      "learning_rate": 0.00019602301125600828,
+      "loss": 0.8933,
+      "step": 333
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.04155290871858597,
+      "learning_rate": 0.0001959985136305246,
+      "loss": 0.9487,
+      "step": 334
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.03602902963757515,
+      "learning_rate": 0.00019597394232605223,
+      "loss": 0.8372,
+      "step": 335
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.033098701387643814,
+      "learning_rate": 0.00019594929736144976,
+      "loss": 0.8942,
+      "step": 336
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.04318083077669144,
+      "learning_rate": 0.00019592457875563214,
+      "loss": 0.7353,
+      "step": 337
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.035677891224622726,
+      "learning_rate": 0.000195899786527571,
+      "loss": 0.7229,
+      "step": 338
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.03783939778804779,
+      "learning_rate": 0.0001958749206962944,
+      "loss": 0.7931,
+      "step": 339
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.03574792295694351,
+      "learning_rate": 0.00019584998128088684,
+      "loss": 0.8139,
+      "step": 340
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.042883213609457016,
+      "learning_rate": 0.00019582496830048947,
+      "loss": 0.8044,
+      "step": 341
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.040097806602716446,
+      "learning_rate": 0.00019579988177429968,
+      "loss": 0.7571,
+      "step": 342
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.033648643642663956,
+      "learning_rate": 0.00019577472172157144,
+      "loss": 0.771,
+      "step": 343
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.04123647138476372,
+      "learning_rate": 0.00019574948816161513,
+      "loss": 0.8319,
+      "step": 344
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.036461152136325836,
+      "learning_rate": 0.00019572418111379758,
+      "loss": 0.8421,
+      "step": 345
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.034517571330070496,
+      "learning_rate": 0.00019569880059754194,
+      "loss": 0.9179,
+      "step": 346
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.03678009659051895,
+      "learning_rate": 0.00019567334663232776,
+      "loss": 0.7756,
+      "step": 347
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.038212236016988754,
+      "learning_rate": 0.00019564781923769108,
+      "loss": 0.7422,
+      "step": 348
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.039440006017684937,
+      "learning_rate": 0.00019562221843322415,
+      "loss": 0.8191,
+      "step": 349
+    },
+    {
+      "epoch": 0.19,
+      "grad_norm": 0.0428134948015213,
+      "learning_rate": 0.0001955965442385756,
+      "loss": 0.8473,
+      "step": 350
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.03534878045320511,
+      "learning_rate": 0.0001955707966734505,
+      "loss": 0.7492,
+      "step": 351
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.03833247721195221,
+      "learning_rate": 0.00019554497575761004,
+      "loss": 0.7775,
+      "step": 352
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.03430182486772537,
+      "learning_rate": 0.00019551908151087187,
+      "loss": 0.8519,
+      "step": 353
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.03104781173169613,
+      "learning_rate": 0.00019549311395310984,
+      "loss": 0.7292,
+      "step": 354
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.033952996134757996,
+      "learning_rate": 0.00019546707310425407,
+      "loss": 0.7592,
+      "step": 355
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.03525162115693092,
+      "learning_rate": 0.00019544095898429097,
+      "loss": 0.7484,
+      "step": 356
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.035235751420259476,
+      "learning_rate": 0.00019541477161326318,
+      "loss": 0.7206,
+      "step": 357
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.038712091743946075,
+      "learning_rate": 0.00019538851101126947,
+      "loss": 0.8494,
+      "step": 358
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.034723225980997086,
+      "learning_rate": 0.00019536217719846497,
+      "loss": 0.784,
+      "step": 359
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.03219115734100342,
+      "learning_rate": 0.0001953357701950609,
+      "loss": 0.7373,
+      "step": 360
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.037317853420972824,
+      "learning_rate": 0.00019530929002132462,
+      "loss": 0.7768,
+      "step": 361
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.033880557864904404,
+      "learning_rate": 0.00019528273669757972,
+      "loss": 0.8359,
+      "step": 362
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.037638477981090546,
+      "learning_rate": 0.00019525611024420595,
+      "loss": 0.7786,
+      "step": 363
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.03666766360402107,
+      "learning_rate": 0.0001952294106816391,
+      "loss": 0.7666,
+      "step": 364
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.03592545911669731,
+      "learning_rate": 0.00019520263803037116,
+      "loss": 0.8049,
+      "step": 365
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.03477128967642784,
+      "learning_rate": 0.00019517579231095018,
+      "loss": 0.8243,
+      "step": 366
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.03351607918739319,
+      "learning_rate": 0.00019514887354398023,
+      "loss": 0.6382,
+      "step": 367
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 0.03531970828771591,
+      "learning_rate": 0.00019512188175012157,
+      "loss": 0.7363,
+      "step": 368
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.03541957587003708,
+      "learning_rate": 0.00019509481695009035,
+      "loss": 0.7136,
+      "step": 369
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.03538355231285095,
+      "learning_rate": 0.00019506767916465894,
+      "loss": 0.8702,
+      "step": 370
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.036403317004442215,
+      "learning_rate": 0.00019504046841465556,
+      "loss": 0.773,
+      "step": 371
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.050964437425136566,
+      "learning_rate": 0.0001950131847209645,
+      "loss": 0.65,
+      "step": 372
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.033455755561590195,
+      "learning_rate": 0.0001949858281045261,
+      "loss": 0.8465,
+      "step": 373
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.038496553897857666,
+      "learning_rate": 0.00019495839858633648,
+      "loss": 0.9064,
+      "step": 374
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.046744123101234436,
+      "learning_rate": 0.00019493089618744796,
+      "loss": 0.7656,
+      "step": 375
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.03650742024183273,
+      "learning_rate": 0.00019490332092896858,
+      "loss": 0.7547,
+      "step": 376
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.03537287563085556,
+      "learning_rate": 0.00019487567283206242,
+      "loss": 0.7773,
+      "step": 377
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.03856958448886871,
+      "learning_rate": 0.00019484795191794944,
+      "loss": 0.8212,
+      "step": 378
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.03926542028784752,
+      "learning_rate": 0.0001948201582079055,
+      "loss": 0.7704,
+      "step": 379
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.03413626179099083,
+      "learning_rate": 0.00019479229172326222,
+      "loss": 0.8243,
+      "step": 380
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.036304738372564316,
+      "learning_rate": 0.00019476435248540726,
+      "loss": 0.7403,
+      "step": 381
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.035184238106012344,
+      "learning_rate": 0.00019473634051578396,
+      "loss": 0.7825,
+      "step": 382
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.0506867952644825,
+      "learning_rate": 0.00019470825583589157,
+      "loss": 0.951,
+      "step": 383
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.034371040761470795,
+      "learning_rate": 0.00019468009846728513,
+      "loss": 0.6246,
+      "step": 384
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.036034394055604935,
+      "learning_rate": 0.00019465186843157546,
+      "loss": 0.8194,
+      "step": 385
+    },
+    {
+      "epoch": 0.21,
+      "grad_norm": 0.03933184966444969,
+      "learning_rate": 0.00019462356575042914,
+      "loss": 0.6906,
+      "step": 386
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03522591292858124,
+      "learning_rate": 0.00019459519044556846,
+      "loss": 0.6198,
+      "step": 387
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.033830877393484116,
+      "learning_rate": 0.00019456674253877162,
+      "loss": 0.7374,
+      "step": 388
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03900926932692528,
+      "learning_rate": 0.00019453822205187232,
+      "loss": 0.7763,
+      "step": 389
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.035312309861183167,
+      "learning_rate": 0.00019450962900676016,
+      "loss": 0.6672,
+      "step": 390
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.034244999289512634,
+      "learning_rate": 0.0001944809634253803,
+      "loss": 0.6852,
+      "step": 391
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03672676905989647,
+      "learning_rate": 0.00019445222532973356,
+      "loss": 0.7416,
+      "step": 392
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03532637655735016,
+      "learning_rate": 0.00019442341474187658,
+      "loss": 0.7647,
+      "step": 393
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03439967334270477,
+      "learning_rate": 0.00019439453168392144,
+      "loss": 0.7458,
+      "step": 394
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03463077172636986,
+      "learning_rate": 0.00019436557617803595,
+      "loss": 0.8125,
+      "step": 395
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03627005219459534,
+      "learning_rate": 0.00019433654824644347,
+      "loss": 0.6992,
+      "step": 396
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03494995832443237,
+      "learning_rate": 0.00019430744791142305,
+      "loss": 0.7078,
+      "step": 397
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03355133906006813,
+      "learning_rate": 0.00019427827519530918,
+      "loss": 0.6952,
+      "step": 398
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03713174909353256,
+      "learning_rate": 0.00019424903012049195,
+      "loss": 0.7782,
+      "step": 399
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03741822391748428,
+      "learning_rate": 0.00019421971270941701,
+      "loss": 0.7951,
+      "step": 400
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03539593145251274,
+      "learning_rate": 0.00019419032298458554,
+      "loss": 0.6305,
+      "step": 401
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03932331129908562,
+      "learning_rate": 0.00019416086096855418,
+      "loss": 0.6978,
+      "step": 402
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.03865070268511772,
+      "learning_rate": 0.00019413132668393503,
+      "loss": 0.7058,
+      "step": 403
+    },
+    {
+      "epoch": 0.22,
+      "grad_norm": 0.036367420107126236,
+      "learning_rate": 0.00019410172015339576,
+      "loss": 0.7991,
+      "step": 404
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.04757410287857056,
+      "learning_rate": 0.00019407204139965936,
+      "loss": 0.7277,
+      "step": 405
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.04415794834494591,
+      "learning_rate": 0.00019404229044550433,
+      "loss": 0.7263,
+      "step": 406
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.030960189178586006,
+      "learning_rate": 0.0001940124673137646,
+      "loss": 0.7514,
+      "step": 407
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.0355895459651947,
+      "learning_rate": 0.0001939825720273294,
+      "loss": 0.9352,
+      "step": 408
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.034076202660799026,
+      "learning_rate": 0.00019395260460914346,
+      "loss": 0.7354,
+      "step": 409
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.03768068179488182,
+      "learning_rate": 0.00019392256508220682,
+      "loss": 0.797,
+      "step": 410
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.03416946902871132,
+      "learning_rate": 0.00019389245346957478,
+      "loss": 0.8355,
+      "step": 411
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.036176249384880066,
+      "learning_rate": 0.00019386226979435813,
+      "loss": 0.775,
+      "step": 412
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.036431025713682175,
+      "learning_rate": 0.00019383201407972286,
+      "loss": 0.6291,
+      "step": 413
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.03727782517671585,
+      "learning_rate": 0.00019380168634889025,
+      "loss": 0.8582,
+      "step": 414
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.03564276918768883,
+      "learning_rate": 0.00019377128662513687,
+      "loss": 0.7397,
+      "step": 415
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.037083700299263,
+      "learning_rate": 0.00019374081493179457,
+      "loss": 0.7173,
+      "step": 416
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.040047336369752884,
+      "learning_rate": 0.00019371027129225042,
+      "loss": 0.6917,
+      "step": 417
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.038617394864559174,
+      "learning_rate": 0.00019367965572994667,
+      "loss": 0.6494,
+      "step": 418
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.03738728538155556,
+      "learning_rate": 0.00019364896826838083,
+      "loss": 0.8023,
+      "step": 419
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.03948422148823738,
+      "learning_rate": 0.00019361820893110557,
+      "loss": 0.8126,
+      "step": 420
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.035054076462984085,
+      "learning_rate": 0.0001935873777417287,
+      "loss": 0.7377,
+      "step": 421
+    },
+    {
+      "epoch": 0.23,
+      "grad_norm": 0.04039939120411873,
+      "learning_rate": 0.00019355647472391328,
+      "loss": 0.7804,
+      "step": 422
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.034298498183488846,
+      "learning_rate": 0.00019352549990137732,
+      "loss": 0.8092,
+      "step": 423
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.035063210874795914,
+      "learning_rate": 0.00019349445329789404,
+      "loss": 0.7162,
+      "step": 424
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.03656880930066109,
+      "learning_rate": 0.00019346333493729182,
+      "loss": 0.7648,
+      "step": 425
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.03690245375037193,
+      "learning_rate": 0.000193432144843454,
+      "loss": 0.7194,
+      "step": 426
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.04461353272199631,
+      "learning_rate": 0.00019340088304031905,
+      "loss": 0.6901,
+      "step": 427
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.03801785781979561,
+      "learning_rate": 0.0001933695495518804,
+      "loss": 0.7416,
+      "step": 428
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.03857959061861038,
+      "learning_rate": 0.00019333814440218656,
+      "loss": 0.7467,
+      "step": 429
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.03680335357785225,
+      "learning_rate": 0.00019330666761534104,
+      "loss": 0.7313,
+      "step": 430
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.03958728164434433,
+      "learning_rate": 0.00019327511921550232,
+      "loss": 0.7054,
+      "step": 431
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.03355373814702034,
+      "learning_rate": 0.0001932434992268838,
+      "loss": 0.7193,
+      "step": 432
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.035141825675964355,
+      "learning_rate": 0.00019321180767375393,
+      "loss": 0.7841,
+      "step": 433
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.037220679223537445,
+      "learning_rate": 0.00019318004458043596,
+      "loss": 0.7195,
+      "step": 434
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.040347468107938766,
+      "learning_rate": 0.00019314820997130814,
+      "loss": 0.7725,
+      "step": 435
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.03546297922730446,
+      "learning_rate": 0.00019311630387080356,
+      "loss": 0.7774,
+      "step": 436
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.03774712607264519,
+      "learning_rate": 0.0001930843263034102,
+      "loss": 0.842,
+      "step": 437
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.0372396744787693,
+      "learning_rate": 0.0001930522772936709,
+      "loss": 0.8061,
+      "step": 438
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.03598784655332565,
+      "learning_rate": 0.00019302015686618326,
+      "loss": 0.8289,
+      "step": 439
+    },
+    {
+      "epoch": 0.24,
+      "grad_norm": 0.038949351757764816,
+      "learning_rate": 0.00019298796504559982,
+      "loss": 0.9173,
+      "step": 440
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.03886549547314644,
+      "learning_rate": 0.00019295570185662783,
+      "loss": 0.7436,
+      "step": 441
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.03816141560673714,
+      "learning_rate": 0.0001929233673240293,
+      "loss": 0.7178,
+      "step": 442
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.03811522200703621,
+      "learning_rate": 0.00019289096147262106,
+      "loss": 0.7304,
+      "step": 443
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.03597824648022652,
+      "learning_rate": 0.00019285848432727465,
+      "loss": 0.9174,
+      "step": 444
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.036536138504743576,
+      "learning_rate": 0.0001928259359129163,
+      "loss": 0.6808,
+      "step": 445
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.03995755687355995,
+      "learning_rate": 0.00019279331625452696,
+      "loss": 0.8131,
+      "step": 446
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.03879891335964203,
+      "learning_rate": 0.00019276062537714226,
+      "loss": 0.7044,
+      "step": 447
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.04052872955799103,
+      "learning_rate": 0.00019272786330585252,
+      "loss": 0.7703,
+      "step": 448
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.047087304294109344,
+      "learning_rate": 0.00019269503006580266,
+      "loss": 0.764,
+      "step": 449
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.035790085792541504,
+      "learning_rate": 0.0001926621256821922,
+      "loss": 0.72,
+      "step": 450
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.03301222249865532,
+      "learning_rate": 0.00019262915018027538,
+      "loss": 0.8394,
+      "step": 451
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.0383736714720726,
+      "learning_rate": 0.00019259610358536087,
+      "loss": 0.8389,
+      "step": 452
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.0351652093231678,
+      "learning_rate": 0.00019256298592281202,
+      "loss": 0.7535,
+      "step": 453
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.03904608264565468,
+      "learning_rate": 0.0001925297972180466,
+      "loss": 0.7091,
+      "step": 454
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.035092540085315704,
+      "learning_rate": 0.00019249653749653711,
+      "loss": 0.6403,
+      "step": 455
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.03878278285264969,
+      "learning_rate": 0.00019246320678381035,
+      "loss": 0.761,
+      "step": 456
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.055852390825748444,
+      "learning_rate": 0.0001924298051054477,
+      "loss": 1.0766,
+      "step": 457
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 0.038434069603681564,
+      "learning_rate": 0.000192396332487085,
+      "loss": 0.7273,
+      "step": 458
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.04319579154253006,
+      "learning_rate": 0.0001923627889544125,
+      "loss": 0.7543,
+      "step": 459
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.03686597943305969,
+      "learning_rate": 0.00019232917453317495,
+      "loss": 0.7785,
+      "step": 460
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.03705907613039017,
+      "learning_rate": 0.00019229548924917146,
+      "loss": 0.8554,
+      "step": 461
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.043490804731845856,
+      "learning_rate": 0.00019226173312825553,
+      "loss": 0.7574,
+      "step": 462
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.0411020964384079,
+      "learning_rate": 0.00019222790619633499,
+      "loss": 0.7173,
+      "step": 463
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.04355505481362343,
+      "learning_rate": 0.0001921940084793721,
+      "loss": 0.7586,
+      "step": 464
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.03533725440502167,
+      "learning_rate": 0.0001921600400033834,
+      "loss": 0.7315,
+      "step": 465
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.037127457559108734,
+      "learning_rate": 0.0001921260007944397,
+      "loss": 0.6851,
+      "step": 466
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.04237948730587959,
+      "learning_rate": 0.00019209189087866614,
+      "loss": 0.7746,
+      "step": 467
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.03694011643528938,
+      "learning_rate": 0.0001920577102822422,
+      "loss": 0.8222,
+      "step": 468
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.03798561915755272,
+      "learning_rate": 0.00019202345903140147,
+      "loss": 0.7533,
+      "step": 469
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.03925656899809837,
+      "learning_rate": 0.00019198913715243182,
+      "loss": 0.7317,
+      "step": 470
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.03404255583882332,
+      "learning_rate": 0.00019195474467167544,
+      "loss": 0.7474,
+      "step": 471
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.03907306119799614,
+      "learning_rate": 0.00019192028161552847,
+      "loss": 0.8662,
+      "step": 472
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.040038101375103,
+      "learning_rate": 0.0001918857480104414,
+      "loss": 0.811,
+      "step": 473
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.04068785905838013,
+      "learning_rate": 0.00019185114388291886,
+      "loss": 0.7831,
+      "step": 474
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.03962669521570206,
+      "learning_rate": 0.00019181646925951954,
+      "loss": 0.7659,
+      "step": 475
+    },
+    {
+      "epoch": 0.26,
+      "grad_norm": 0.036200929433107376,
+      "learning_rate": 0.00019178172416685628,
+      "loss": 0.8251,
+      "step": 476
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.03592352196574211,
+      "learning_rate": 0.00019174690863159593,
+      "loss": 0.7206,
+      "step": 477
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.03576627001166344,
+      "learning_rate": 0.0001917120226804595,
+      "loss": 0.8105,
+      "step": 478
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.034843627363443375,
+      "learning_rate": 0.00019167706634022199,
+      "loss": 0.8635,
+      "step": 479
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.040766939520835876,
+      "learning_rate": 0.00019164203963771244,
+      "loss": 0.6993,
+      "step": 480
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.04198118671774864,
+      "learning_rate": 0.00019160694259981388,
+      "loss": 0.8341,
+      "step": 481
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.03662072494626045,
+      "learning_rate": 0.0001915717752534634,
+      "loss": 0.794,
+      "step": 482
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.0427091047167778,
+      "learning_rate": 0.0001915365376256519,
+      "loss": 0.7304,
+      "step": 483
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.03634573519229889,
+      "learning_rate": 0.00019150122974342435,
+      "loss": 0.7365,
+      "step": 484
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.03949028626084328,
+      "learning_rate": 0.0001914658516338796,
+      "loss": 0.8231,
+      "step": 485
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.04077625274658203,
+      "learning_rate": 0.00019143040332417037,
+      "loss": 0.66,
+      "step": 486
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.03831211104989052,
+      "learning_rate": 0.0001913948848415033,
+      "loss": 0.8029,
+      "step": 487
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.042038679122924805,
+      "learning_rate": 0.0001913592962131389,
+      "loss": 0.807,
+      "step": 488
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.040266357362270355,
+      "learning_rate": 0.00019132363746639147,
+      "loss": 0.6887,
+      "step": 489
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.0406985804438591,
+      "learning_rate": 0.0001912879086286291,
+      "loss": 0.7928,
+      "step": 490
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.03616916015744209,
+      "learning_rate": 0.00019125210972727378,
+      "loss": 0.697,
+      "step": 491
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.0417737253010273,
+      "learning_rate": 0.00019121624078980123,
+      "loss": 0.8405,
+      "step": 492
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.04053768515586853,
+      "learning_rate": 0.00019118030184374086,
+      "loss": 0.7304,
+      "step": 493
+    },
+    {
+      "epoch": 0.27,
+      "grad_norm": 0.037591926753520966,
+      "learning_rate": 0.00019114429291667583,
+      "loss": 0.797,
+      "step": 494
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.03530837222933769,
+      "learning_rate": 0.00019110821403624316,
+      "loss": 0.759,
+      "step": 495
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.035193149000406265,
+      "learning_rate": 0.0001910720652301333,
+      "loss": 0.7179,
+      "step": 496
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.03709745407104492,
+      "learning_rate": 0.0001910358465260906,
+      "loss": 0.8888,
+      "step": 497
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.036165978759527206,
+      "learning_rate": 0.00019099955795191296,
+      "loss": 0.8409,
+      "step": 498
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.03585299476981163,
+      "learning_rate": 0.00019096319953545185,
+      "loss": 0.6962,
+      "step": 499
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.03879237174987793,
+      "learning_rate": 0.00019092677130461245,
+      "loss": 0.7037,
+      "step": 500
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.040262021124362946,
+      "learning_rate": 0.00019089027328735348,
+      "loss": 0.8989,
+      "step": 501
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.09765855222940445,
+      "learning_rate": 0.0001908537055116872,
+      "loss": 0.8365,
+      "step": 502
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.04247608780860901,
+      "learning_rate": 0.00019081706800567943,
+      "loss": 0.8854,
+      "step": 503
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.04151815176010132,
+      "learning_rate": 0.00019078036079744947,
+      "loss": 0.7774,
+      "step": 504
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.04607046768069267,
+      "learning_rate": 0.00019074358391517023,
+      "loss": 0.7117,
+      "step": 505
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.047151170670986176,
+      "learning_rate": 0.00019070673738706798,
+      "loss": 0.8029,
+      "step": 506
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.0413680225610733,
+      "learning_rate": 0.00019066982124142244,
+      "loss": 0.6586,
+      "step": 507
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.039332833141088486,
+      "learning_rate": 0.00019063283550656689,
+      "loss": 0.6051,
+      "step": 508
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.03690175712108612,
+      "learning_rate": 0.0001905957802108878,
+      "loss": 0.6932,
+      "step": 509
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.04148999601602554,
+      "learning_rate": 0.0001905586553828253,
+      "loss": 0.7663,
+      "step": 510
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.03818729892373085,
+      "learning_rate": 0.00019052146105087267,
+      "loss": 0.7886,
+      "step": 511
+    },
+    {
+      "epoch": 0.28,
+      "grad_norm": 0.03793442249298096,
+      "learning_rate": 0.0001904841972435766,
+      "loss": 0.7099,
+      "step": 512
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.04220060631632805,
+      "learning_rate": 0.00019044686398953715,
+      "loss": 0.7333,
+      "step": 513
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.03924279659986496,
+      "learning_rate": 0.00019040946131740764,
+      "loss": 0.7619,
+      "step": 514
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.03894231095910072,
+      "learning_rate": 0.00019037198925589465,
+      "loss": 0.8214,
+      "step": 515
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.038151923567056656,
+      "learning_rate": 0.00019033444783375804,
+      "loss": 0.8511,
+      "step": 516
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.04167793318629265,
+      "learning_rate": 0.0001902968370798109,
+      "loss": 0.7582,
+      "step": 517
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.039893049746751785,
+      "learning_rate": 0.00019025915702291956,
+      "loss": 0.8991,
+      "step": 518
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.03804799169301987,
+      "learning_rate": 0.00019022140769200352,
+      "loss": 0.7959,
+      "step": 519
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.03987705335021019,
+      "learning_rate": 0.00019018358911603538,
+      "loss": 0.7415,
+      "step": 520
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.036575764417648315,
+      "learning_rate": 0.000190145701324041,
+      "loss": 0.6772,
+      "step": 521
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.03801872953772545,
+      "learning_rate": 0.00019010774434509933,
+      "loss": 0.7397,
+      "step": 522
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.03547314926981926,
+      "learning_rate": 0.0001900697182083423,
+      "loss": 0.6802,
+      "step": 523
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.03981153294444084,
+      "learning_rate": 0.00019003162294295515,
+      "loss": 0.8714,
+      "step": 524
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.035997990518808365,
+      "learning_rate": 0.000189993458578176,
+      "loss": 0.6773,
+      "step": 525
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.04118787497282028,
+      "learning_rate": 0.00018995522514329602,
+      "loss": 0.7247,
+      "step": 526
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.04163433611392975,
+      "learning_rate": 0.00018991692266765947,
+      "loss": 0.7977,
+      "step": 527
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.03958054631948471,
+      "learning_rate": 0.0001898785511806635,
+      "loss": 0.7665,
+      "step": 528
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.040298108011484146,
+      "learning_rate": 0.0001898401107117583,
+      "loss": 0.8497,
+      "step": 529
+    },
+    {
+      "epoch": 0.29,
+      "grad_norm": 0.039216410368680954,
+      "learning_rate": 0.00018980160129044698,
+      "loss": 0.8379,
+      "step": 530
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.037412021309137344,
+      "learning_rate": 0.0001897630229462856,
+      "loss": 0.8241,
+      "step": 531
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.042515527456998825,
+      "learning_rate": 0.0001897243757088831,
+      "loss": 0.7151,
+      "step": 532
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.038337383419275284,
+      "learning_rate": 0.0001896856596079012,
+      "loss": 0.7465,
+      "step": 533
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.03905492648482323,
+      "learning_rate": 0.00018964687467305465,
+      "loss": 0.6749,
+      "step": 534
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.038022320717573166,
+      "learning_rate": 0.0001896080209341109,
+      "loss": 0.6692,
+      "step": 535
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.0380224771797657,
+      "learning_rate": 0.00018956909842089023,
+      "loss": 0.866,
+      "step": 536
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.03461451828479767,
+      "learning_rate": 0.00018953010716326577,
+      "loss": 0.7494,
+      "step": 537
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.04129749909043312,
+      "learning_rate": 0.00018949104719116332,
+      "loss": 0.783,
+      "step": 538
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.0383567251265049,
+      "learning_rate": 0.00018945191853456152,
+      "loss": 0.7299,
+      "step": 539
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.038509126752614975,
+      "learning_rate": 0.0001894127212234916,
+      "loss": 0.8074,
+      "step": 540
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.0398080088198185,
+      "learning_rate": 0.00018937345528803755,
+      "loss": 0.7684,
+      "step": 541
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.04023388773202896,
+      "learning_rate": 0.0001893341207583361,
+      "loss": 0.8234,
+      "step": 542
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.0396123044192791,
+      "learning_rate": 0.00018929471766457648,
+      "loss": 0.8573,
+      "step": 543
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.03965805470943451,
+      "learning_rate": 0.00018925524603700063,
+      "loss": 0.8654,
+      "step": 544
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.03725765272974968,
+      "learning_rate": 0.00018921570590590316,
+      "loss": 0.7674,
+      "step": 545
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.042921282351017,
+      "learning_rate": 0.00018917609730163105,
+      "loss": 0.7293,
+      "step": 546
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.04281310364603996,
+      "learning_rate": 0.00018913642025458405,
+      "loss": 0.8283,
+      "step": 547
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 0.03923151642084122,
+      "learning_rate": 0.00018909667479521426,
+      "loss": 0.8305,
+      "step": 548
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.035802267491817474,
+      "learning_rate": 0.00018905686095402647,
+      "loss": 0.6796,
+      "step": 549
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.04895975813269615,
+      "learning_rate": 0.00018901697876157777,
+      "loss": 0.8961,
+      "step": 550
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.039671383798122406,
+      "learning_rate": 0.00018897702824847786,
+      "loss": 0.7407,
+      "step": 551
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.038863249123096466,
+      "learning_rate": 0.00018893700944538883,
+      "loss": 0.7412,
+      "step": 552
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.03964029625058174,
+      "learning_rate": 0.00018889692238302508,
+      "loss": 0.7407,
+      "step": 553
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.03963325545191765,
+      "learning_rate": 0.00018885676709215355,
+      "loss": 0.671,
+      "step": 554
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.04223211482167244,
+      "learning_rate": 0.00018881654360359344,
+      "loss": 0.9177,
+      "step": 555
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.04407535493373871,
+      "learning_rate": 0.00018877625194821637,
+      "loss": 0.8018,
+      "step": 556
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.045421354472637177,
+      "learning_rate": 0.00018873589215694623,
+      "loss": 0.7992,
+      "step": 557
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.03759824484586716,
+      "learning_rate": 0.00018869546426075919,
+      "loss": 0.6757,
+      "step": 558
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.03787359595298767,
+      "learning_rate": 0.00018865496829068373,
+      "loss": 0.7533,
+      "step": 559
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.04168887808918953,
+      "learning_rate": 0.0001886144042778006,
+      "loss": 0.8529,
+      "step": 560
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.03604380786418915,
+      "learning_rate": 0.00018857377225324265,
+      "loss": 0.6959,
+      "step": 561
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.039811234921216965,
+      "learning_rate": 0.00018853307224819506,
+      "loss": 0.6923,
+      "step": 562
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.039177920669317245,
+      "learning_rate": 0.00018849230429389516,
+      "loss": 0.8089,
+      "step": 563
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.03779724985361099,
+      "learning_rate": 0.0001884514684216324,
+      "loss": 0.6745,
+      "step": 564
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.03678756207227707,
+      "learning_rate": 0.00018841056466274836,
+      "loss": 0.6761,
+      "step": 565
+    },
+    {
+      "epoch": 0.31,
+      "grad_norm": 0.03539562225341797,
+      "learning_rate": 0.0001883695930486367,
+      "loss": 0.7111,
+      "step": 566
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.03760172799229622,
+      "learning_rate": 0.00018832855361074322,
+      "loss": 0.8614,
+      "step": 567
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.03915516659617424,
+      "learning_rate": 0.00018828744638056574,
+      "loss": 0.8167,
+      "step": 568
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.039351411163806915,
+      "learning_rate": 0.00018824627138965414,
+      "loss": 0.7736,
+      "step": 569
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.03440462425351143,
+      "learning_rate": 0.00018820502866961022,
+      "loss": 0.6989,
+      "step": 570
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.04510714113712311,
+      "learning_rate": 0.00018816371825208789,
+      "loss": 0.7669,
+      "step": 571
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.038387760519981384,
+      "learning_rate": 0.0001881223401687929,
+      "loss": 0.7742,
+      "step": 572
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.04283255338668823,
+      "learning_rate": 0.00018808089445148302,
+      "loss": 0.8729,
+      "step": 573
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.03683461621403694,
+      "learning_rate": 0.00018803938113196787,
+      "loss": 0.7528,
+      "step": 574
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.03634823486208916,
+      "learning_rate": 0.00018799780024210898,
+      "loss": 0.7812,
+      "step": 575
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.04318126663565636,
+      "learning_rate": 0.00018795615181381976,
+      "loss": 0.8744,
+      "step": 576
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.04043075069785118,
+      "learning_rate": 0.00018791443587906542,
+      "loss": 0.719,
+      "step": 577
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.03615717217326164,
+      "learning_rate": 0.00018787265246986298,
+      "loss": 0.6402,
+      "step": 578
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.0358918234705925,
+      "learning_rate": 0.0001878308016182813,
+      "loss": 0.7614,
+      "step": 579
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.03841692954301834,
+      "learning_rate": 0.0001877888833564409,
+      "loss": 0.6891,
+      "step": 580
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.04361288622021675,
+      "learning_rate": 0.00018774689771651422,
+      "loss": 0.866,
+      "step": 581
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.04058700054883957,
+      "learning_rate": 0.0001877048447307252,
+      "loss": 0.7412,
+      "step": 582
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.04075286164879799,
+      "learning_rate": 0.00018766272443134955,
+      "loss": 0.8213,
+      "step": 583
+    },
+    {
+      "epoch": 0.32,
+      "grad_norm": 0.03992673009634018,
+      "learning_rate": 0.00018762053685071473,
+      "loss": 0.7053,
+      "step": 584
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.03777340427041054,
+      "learning_rate": 0.00018757828202119973,
+      "loss": 0.656,
+      "step": 585
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.04056774452328682,
+      "learning_rate": 0.00018753595997523514,
+      "loss": 0.8583,
+      "step": 586
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.03643214702606201,
+      "learning_rate": 0.00018749357074530326,
+      "loss": 0.7368,
+      "step": 587
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.040921106934547424,
+      "learning_rate": 0.00018745111436393788,
+      "loss": 0.7502,
+      "step": 588
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.037460386753082275,
+      "learning_rate": 0.00018740859086372429,
+      "loss": 0.7211,
+      "step": 589
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.04132199287414551,
+      "learning_rate": 0.00018736600027729932,
+      "loss": 0.8679,
+      "step": 590
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.038056958466768265,
+      "learning_rate": 0.00018732334263735136,
+      "loss": 0.7041,
+      "step": 591
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.036732178181409836,
+      "learning_rate": 0.0001872806179766202,
+      "loss": 0.8356,
+      "step": 592
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.04578113928437233,
+      "learning_rate": 0.00018723782632789701,
+      "loss": 0.7287,
+      "step": 593
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.04152052104473114,
+      "learning_rate": 0.0001871949677240245,
+      "loss": 0.7445,
+      "step": 594
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.05432068929076195,
+      "learning_rate": 0.00018715204219789668,
+      "loss": 0.6701,
+      "step": 595
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.040836673229932785,
+      "learning_rate": 0.00018710904978245894,
+      "loss": 0.7228,
+      "step": 596
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.04091890901327133,
+      "learning_rate": 0.00018706599051070808,
+      "loss": 0.7615,
+      "step": 597
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.039113231003284454,
+      "learning_rate": 0.00018702286441569206,
+      "loss": 0.8016,
+      "step": 598
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.03947889804840088,
+      "learning_rate": 0.00018697967153051028,
+      "loss": 0.8327,
+      "step": 599
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.03761962801218033,
+      "learning_rate": 0.0001869364118883133,
+      "loss": 0.7843,
+      "step": 600
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.035844236612319946,
+      "learning_rate": 0.00018689308552230296,
+      "loss": 0.7678,
+      "step": 601
+    },
+    {
+      "epoch": 0.33,
+      "grad_norm": 0.04012085869908333,
+      "learning_rate": 0.00018684969246573235,
+      "loss": 0.8195,
+      "step": 602
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.04106171056628227,
+      "learning_rate": 0.00018680623275190564,
+      "loss": 0.8068,
+      "step": 603
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.03810250759124756,
+      "learning_rate": 0.00018676270641417822,
+      "loss": 0.6382,
+      "step": 604
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.04322788119316101,
+      "learning_rate": 0.00018671911348595667,
+      "loss": 0.7498,
+      "step": 605
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.04026930034160614,
+      "learning_rate": 0.00018667545400069858,
+      "loss": 0.7506,
+      "step": 606
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.040973201394081116,
+      "learning_rate": 0.00018663172799191264,
+      "loss": 0.7408,
+      "step": 607
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.03879684954881668,
+      "learning_rate": 0.00018658793549315868,
+      "loss": 0.805,
+      "step": 608
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.038516003638505936,
+      "learning_rate": 0.00018654407653804746,
+      "loss": 0.6352,
+      "step": 609
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.041547078639268875,
+      "learning_rate": 0.0001865001511602408,
+      "loss": 0.7504,
+      "step": 610
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.03565856069326401,
+      "learning_rate": 0.0001864561593934515,
+      "loss": 0.6952,
+      "step": 611
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.043429479002952576,
+      "learning_rate": 0.00018641210127144327,
+      "loss": 0.7584,
+      "step": 612
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.043735940009355545,
+      "learning_rate": 0.00018636797682803082,
+      "loss": 0.6309,
+      "step": 613
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.03792436793446541,
+      "learning_rate": 0.00018632378609707967,
+      "loss": 0.6673,
+      "step": 614
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.03878699988126755,
+      "learning_rate": 0.0001862795291125063,
+      "loss": 0.6326,
+      "step": 615
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.04062122479081154,
+      "learning_rate": 0.00018623520590827799,
+      "loss": 0.6807,
+      "step": 616
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.046398624777793884,
+      "learning_rate": 0.00018619081651841286,
+      "loss": 0.6643,
+      "step": 617
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.04108712822198868,
+      "learning_rate": 0.00018614636097697985,
+      "loss": 0.7254,
+      "step": 618
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.0392933115363121,
+      "learning_rate": 0.00018610183931809862,
+      "loss": 0.7912,
+      "step": 619
+    },
+    {
+      "epoch": 0.34,
+      "grad_norm": 0.03968581184744835,
+      "learning_rate": 0.00018605725157593958,
+      "loss": 0.7038,
+      "step": 620
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.03825043514370918,
+      "learning_rate": 0.00018601259778472395,
+      "loss": 0.6573,
+      "step": 621
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.04096253588795662,
+      "learning_rate": 0.00018596787797872354,
+      "loss": 0.7289,
+      "step": 622
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.03817110136151314,
+      "learning_rate": 0.00018592309219226083,
+      "loss": 0.7687,
+      "step": 623
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.0371723398566246,
+      "learning_rate": 0.00018587824045970903,
+      "loss": 0.631,
+      "step": 624
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.04173294082283974,
+      "learning_rate": 0.0001858333228154919,
+      "loss": 0.7503,
+      "step": 625
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.03969898074865341,
+      "learning_rate": 0.0001857883392940837,
+      "loss": 0.7999,
+      "step": 626
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.034285057336091995,
+      "learning_rate": 0.00018574328993000946,
+      "loss": 0.742,
+      "step": 627
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.03918811306357384,
+      "learning_rate": 0.0001856981747578446,
+      "loss": 0.647,
+      "step": 628
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.039601150900125504,
+      "learning_rate": 0.00018565299381221505,
+      "loss": 0.6368,
+      "step": 629
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.04142673686146736,
+      "learning_rate": 0.0001856077471277972,
+      "loss": 0.8081,
+      "step": 630
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.04136984422802925,
+      "learning_rate": 0.00018556243473931801,
+      "loss": 0.7583,
+      "step": 631
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.046918828040361404,
+      "learning_rate": 0.0001855170566815548,
+      "loss": 0.8415,
+      "step": 632
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.03764290362596512,
+      "learning_rate": 0.0001854716129893353,
+      "loss": 0.6575,
+      "step": 633
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.04044364020228386,
+      "learning_rate": 0.00018542610369753755,
+      "loss": 0.6595,
+      "step": 634
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.04079195857048035,
+      "learning_rate": 0.00018538052884109005,
+      "loss": 0.6936,
+      "step": 635
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.04137120023369789,
+      "learning_rate": 0.00018533488845497148,
+      "loss": 0.7419,
+      "step": 636
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.0409332811832428,
+      "learning_rate": 0.000185289182574211,
+      "loss": 0.7343,
+      "step": 637
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 0.03855755180120468,
+      "learning_rate": 0.0001852434112338879,
+      "loss": 0.7341,
+      "step": 638
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.04691479727625847,
+      "learning_rate": 0.0001851975744691317,
+      "loss": 0.6813,
+      "step": 639
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.04192007705569267,
+      "learning_rate": 0.00018515167231512224,
+      "loss": 0.8011,
+      "step": 640
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.04521464928984642,
+      "learning_rate": 0.00018510570480708942,
+      "loss": 0.7823,
+      "step": 641
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.035000029951334,
+      "learning_rate": 0.0001850596719803134,
+      "loss": 0.5621,
+      "step": 642
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.04110797494649887,
+      "learning_rate": 0.00018501357387012447,
+      "loss": 0.6851,
+      "step": 643
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.04273466393351555,
+      "learning_rate": 0.00018496741051190297,
+      "loss": 0.7967,
+      "step": 644
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.04418378323316574,
+      "learning_rate": 0.0001849211819410793,
+      "loss": 0.7684,
+      "step": 645
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.03610142320394516,
+      "learning_rate": 0.00018487488819313402,
+      "loss": 0.6515,
+      "step": 646
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.041490521281957626,
+      "learning_rate": 0.0001848285293035976,
+      "loss": 0.8379,
+      "step": 647
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.041976895183324814,
+      "learning_rate": 0.0001847821053080505,
+      "loss": 0.7772,
+      "step": 648
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.041484154760837555,
+      "learning_rate": 0.0001847356162421233,
+      "loss": 0.7797,
+      "step": 649
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.04266678914427757,
+      "learning_rate": 0.00018468906214149638,
+      "loss": 0.681,
+      "step": 650
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.041691817343235016,
+      "learning_rate": 0.00018464244304190004,
+      "loss": 0.8475,
+      "step": 651
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.039543136954307556,
+      "learning_rate": 0.00018459575897911455,
+      "loss": 0.5914,
+      "step": 652
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.035682205110788345,
+      "learning_rate": 0.00018454900998896996,
+      "loss": 0.6921,
+      "step": 653
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.04010608792304993,
+      "learning_rate": 0.0001845021961073462,
+      "loss": 0.7686,
+      "step": 654
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.04509714990854263,
+      "learning_rate": 0.00018445531737017298,
+      "loss": 0.7965,
+      "step": 655
+    },
+    {
+      "epoch": 0.36,
+      "grad_norm": 0.04406216740608215,
+      "learning_rate": 0.0001844083738134298,
+      "loss": 0.8243,
+      "step": 656
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.03819778561592102,
+      "learning_rate": 0.00018436136547314593,
+      "loss": 0.6683,
+      "step": 657
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.04046725109219551,
+      "learning_rate": 0.00018431429238540027,
+      "loss": 0.8624,
+      "step": 658
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.04344424977898598,
+      "learning_rate": 0.00018426715458632153,
+      "loss": 0.7828,
+      "step": 659
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.040390148758888245,
+      "learning_rate": 0.00018421995211208802,
+      "loss": 0.7757,
+      "step": 660
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.03947126492857933,
+      "learning_rate": 0.00018417268499892767,
+      "loss": 0.6989,
+      "step": 661
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.04135662689805031,
+      "learning_rate": 0.00018412535328311814,
+      "loss": 0.847,
+      "step": 662
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.049522459506988525,
+      "learning_rate": 0.00018407795700098648,
+      "loss": 0.7971,
+      "step": 663
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.040498074144124985,
+      "learning_rate": 0.0001840304961889095,
+      "loss": 0.7314,
+      "step": 664
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.05009998753666878,
+      "learning_rate": 0.00018398297088331332,
+      "loss": 0.7435,
+      "step": 665
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.03603378310799599,
+      "learning_rate": 0.0001839353811206738,
+      "loss": 0.7178,
+      "step": 666
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.037989791482686996,
+      "learning_rate": 0.00018388772693751602,
+      "loss": 0.8039,
+      "step": 667
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.03886855021119118,
+      "learning_rate": 0.00018384000837041477,
+      "loss": 0.7001,
+      "step": 668
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.04233938828110695,
+      "learning_rate": 0.00018379222545599402,
+      "loss": 0.7929,
+      "step": 669
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.04034002870321274,
+      "learning_rate": 0.00018374437823092724,
+      "loss": 0.6706,
+      "step": 670
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.04359966889023781,
+      "learning_rate": 0.00018369646673193724,
+      "loss": 0.825,
+      "step": 671
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.04203416034579277,
+      "learning_rate": 0.0001836484909957962,
+      "loss": 0.8887,
+      "step": 672
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.04705547168850899,
+      "learning_rate": 0.0001836004510593255,
+      "loss": 0.7003,
+      "step": 673
+    },
+    {
+      "epoch": 0.37,
+      "grad_norm": 0.040125973522663116,
+      "learning_rate": 0.00018355234695939586,
+      "loss": 0.7106,
+      "step": 674
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.040442850440740585,
+      "learning_rate": 0.0001835041787329273,
+      "loss": 0.6736,
+      "step": 675
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.04361211508512497,
+      "learning_rate": 0.00018345594641688893,
+      "loss": 0.7427,
+      "step": 676
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.039316337555646896,
+      "learning_rate": 0.0001834076500482992,
+      "loss": 0.7529,
+      "step": 677
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.041209638118743896,
+      "learning_rate": 0.00018335928966422557,
+      "loss": 0.7345,
+      "step": 678
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.04208279401063919,
+      "learning_rate": 0.0001833108653017847,
+      "loss": 0.7544,
+      "step": 679
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.042629461735486984,
+      "learning_rate": 0.0001832623769981424,
+      "loss": 0.6676,
+      "step": 680
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.038116682320833206,
+      "learning_rate": 0.00018321382479051347,
+      "loss": 0.705,
+      "step": 681
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.03879912942647934,
+      "learning_rate": 0.0001831652087161618,
+      "loss": 0.7405,
+      "step": 682
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.04380761831998825,
+      "learning_rate": 0.00018311652881240032,
+      "loss": 0.6791,
+      "step": 683
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.03818592429161072,
+      "learning_rate": 0.00018306778511659085,
+      "loss": 0.6545,
+      "step": 684
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.0418897345662117,
+      "learning_rate": 0.00018301897766614435,
+      "loss": 0.815,
+      "step": 685
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.04104520007967949,
+      "learning_rate": 0.00018297010649852052,
+      "loss": 0.6821,
+      "step": 686
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.04211374744772911,
+      "learning_rate": 0.0001829211716512281,
+      "loss": 0.778,
+      "step": 687
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.045832112431526184,
+      "learning_rate": 0.00018287217316182458,
+      "loss": 0.8436,
+      "step": 688
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.042089615017175674,
+      "learning_rate": 0.00018282311106791645,
+      "loss": 0.6899,
+      "step": 689
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.039306432008743286,
+      "learning_rate": 0.0001827739854071589,
+      "loss": 0.6268,
+      "step": 690
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.04297361150383949,
+      "learning_rate": 0.00018272479621725589,
+      "loss": 0.7461,
+      "step": 691
+    },
+    {
+      "epoch": 0.38,
+      "grad_norm": 0.04032345116138458,
+      "learning_rate": 0.00018267554353596025,
+      "loss": 0.797,
+      "step": 692
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.04208484664559364,
+      "learning_rate": 0.0001826262274010735,
+      "loss": 0.8043,
+      "step": 693
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.039476584643125534,
+      "learning_rate": 0.00018257684785044577,
+      "loss": 0.6953,
+      "step": 694
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.040461137890815735,
+      "learning_rate": 0.00018252740492197598,
+      "loss": 0.754,
+      "step": 695
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.04078752547502518,
+      "learning_rate": 0.00018247789865361166,
+      "loss": 0.6886,
+      "step": 696
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.04459698870778084,
+      "learning_rate": 0.00018242832908334886,
+      "loss": 0.7355,
+      "step": 697
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.039952464401721954,
+      "learning_rate": 0.00018237869624923236,
+      "loss": 0.7305,
+      "step": 698
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.037101615220308304,
+      "learning_rate": 0.0001823290001893554,
+      "loss": 0.7426,
+      "step": 699
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.050815049558877945,
+      "learning_rate": 0.0001822792409418598,
+      "loss": 0.6909,
+      "step": 700
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.04514652490615845,
+      "learning_rate": 0.0001822294185449358,
+      "loss": 0.8794,
+      "step": 701
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.04079204425215721,
+      "learning_rate": 0.00018217953303682223,
+      "loss": 0.6969,
+      "step": 702
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.03969675302505493,
+      "learning_rate": 0.0001821295844558062,
+      "loss": 0.6827,
+      "step": 703
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.04383305460214615,
+      "learning_rate": 0.0001820795728402234,
+      "loss": 0.8434,
+      "step": 704
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.039204008877277374,
+      "learning_rate": 0.00018202949822845773,
+      "loss": 0.6717,
+      "step": 705
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.03731399402022362,
+      "learning_rate": 0.00018197936065894157,
+      "loss": 0.5858,
+      "step": 706
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.04733480513095856,
+      "learning_rate": 0.00018192916017015557,
+      "loss": 0.7383,
+      "step": 707
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.04360615089535713,
+      "learning_rate": 0.00018187889680062864,
+      "loss": 0.716,
+      "step": 708
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.040373023599386215,
+      "learning_rate": 0.00018182857058893803,
+      "loss": 0.6859,
+      "step": 709
+    },
+    {
+      "epoch": 0.39,
+      "grad_norm": 0.041170086711645126,
+      "learning_rate": 0.00018177818157370915,
+      "loss": 0.6335,
+      "step": 710
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.039953090250492096,
+      "learning_rate": 0.00018172772979361567,
+      "loss": 0.6232,
+      "step": 711
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.04438021779060364,
+      "learning_rate": 0.0001816772152873793,
+      "loss": 0.7427,
+      "step": 712
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.037649158388376236,
+      "learning_rate": 0.00018162663809377012,
+      "loss": 0.6732,
+      "step": 713
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.04502052068710327,
+      "learning_rate": 0.0001815759982516061,
+      "loss": 0.8635,
+      "step": 714
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.04553660750389099,
+      "learning_rate": 0.00018152529579975343,
+      "loss": 0.7883,
+      "step": 715
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.038996774703264236,
+      "learning_rate": 0.00018147453077712634,
+      "loss": 0.7818,
+      "step": 716
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.03996030241250992,
+      "learning_rate": 0.000181423703222687,
+      "loss": 0.7777,
+      "step": 717
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.04063892737030983,
+      "learning_rate": 0.0001813728131754456,
+      "loss": 0.7201,
+      "step": 718
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.04045387730002403,
+      "learning_rate": 0.00018132186067446043,
+      "loss": 0.8018,
+      "step": 719
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.042122699320316315,
+      "learning_rate": 0.0001812708457588375,
+      "loss": 0.7921,
+      "step": 720
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.04201977327466011,
+      "learning_rate": 0.00018121976846773084,
+      "loss": 0.7049,
+      "step": 721
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.03838672116398811,
+      "learning_rate": 0.00018116862884034243,
+      "loss": 0.6836,
+      "step": 722
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.044829823076725006,
+      "learning_rate": 0.0001811174269159219,
+      "loss": 0.765,
+      "step": 723
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.04038670286536217,
+      "learning_rate": 0.00018106616273376683,
+      "loss": 0.7528,
+      "step": 724
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.041412707418203354,
+      "learning_rate": 0.00018101483633322255,
+      "loss": 0.8253,
+      "step": 725
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.04251568764448166,
+      "learning_rate": 0.00018096344775368214,
+      "loss": 0.7483,
+      "step": 726
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.043436773121356964,
+      "learning_rate": 0.0001809119970345864,
+      "loss": 0.7989,
+      "step": 727
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 0.04096989333629608,
+      "learning_rate": 0.00018086048421542383,
+      "loss": 0.7947,
+      "step": 728
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.04080723971128464,
+      "learning_rate": 0.0001808089093357306,
+      "loss": 0.7878,
+      "step": 729
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.043036088347435,
+      "learning_rate": 0.0001807572724350905,
+      "loss": 0.7725,
+      "step": 730
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.040440741926431656,
+      "learning_rate": 0.0001807055735531349,
+      "loss": 0.688,
+      "step": 731
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.044196490198373795,
+      "learning_rate": 0.00018065381272954278,
+      "loss": 0.8786,
+      "step": 732
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.042028769850730896,
+      "learning_rate": 0.00018060199000404062,
+      "loss": 0.8395,
+      "step": 733
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.04810195043683052,
+      "learning_rate": 0.00018055010541640244,
+      "loss": 0.7793,
+      "step": 734
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.03826632723212242,
+      "learning_rate": 0.0001804981590064498,
+      "loss": 0.6893,
+      "step": 735
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.04325885698199272,
+      "learning_rate": 0.00018044615081405153,
+      "loss": 0.8006,
+      "step": 736
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.039932068437337875,
+      "learning_rate": 0.00018039408087912402,
+      "loss": 0.6664,
+      "step": 737
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.04278779402375221,
+      "learning_rate": 0.00018034194924163103,
+      "loss": 0.7397,
+      "step": 738
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.03901227191090584,
+      "learning_rate": 0.0001802897559415837,
+      "loss": 0.5662,
+      "step": 739
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.043960485607385635,
+      "learning_rate": 0.0001802375010190404,
+      "loss": 0.7809,
+      "step": 740
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.04186437651515007,
+      "learning_rate": 0.0001801851845141069,
+      "loss": 0.6976,
+      "step": 741
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.041978247463703156,
+      "learning_rate": 0.00018013280646693612,
+      "loss": 0.7146,
+      "step": 742
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.04281838610768318,
+      "learning_rate": 0.00018008036691772835,
+      "loss": 0.7627,
+      "step": 743
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.0473211370408535,
+      "learning_rate": 0.00018002786590673098,
+      "loss": 0.8042,
+      "step": 744
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.04652298986911774,
+      "learning_rate": 0.0001799753034742386,
+      "loss": 0.731,
+      "step": 745
+    },
+    {
+      "epoch": 0.41,
+      "grad_norm": 0.039468780159950256,
+      "learning_rate": 0.000179922679660593,
+      "loss": 0.7135,
+      "step": 746
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.04733370244503021,
+      "learning_rate": 0.00017986999450618295,
+      "loss": 0.8023,
+      "step": 747
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.04231337085366249,
+      "learning_rate": 0.00017981724805144443,
+      "loss": 0.7528,
+      "step": 748
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.04062468931078911,
+      "learning_rate": 0.00017976444033686043,
+      "loss": 0.7331,
+      "step": 749
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.04496033489704132,
+      "learning_rate": 0.0001797115714029609,
+      "loss": 0.6968,
+      "step": 750
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.04302246496081352,
+      "learning_rate": 0.00017965864129032284,
+      "loss": 0.8159,
+      "step": 751
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.04262755811214447,
+      "learning_rate": 0.00017960565003957018,
+      "loss": 0.7148,
+      "step": 752
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.040617603808641434,
+      "learning_rate": 0.00017955259769137375,
+      "loss": 0.7302,
+      "step": 753
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.04194014146924019,
+      "learning_rate": 0.00017949948428645134,
+      "loss": 0.6785,
+      "step": 754
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.04514269158244133,
+      "learning_rate": 0.00017944630986556752,
+      "loss": 0.748,
+      "step": 755
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.042905841022729874,
+      "learning_rate": 0.0001793930744695338,
+      "loss": 0.7357,
+      "step": 756
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.04359228163957596,
+      "learning_rate": 0.00017933977813920833,
+      "loss": 0.7426,
+      "step": 757
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.040619876235723495,
+      "learning_rate": 0.00017928642091549613,
+      "loss": 0.7278,
+      "step": 758
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.039674315601587296,
+      "learning_rate": 0.000179233002839349,
+      "loss": 0.7459,
+      "step": 759
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.039900798350572586,
+      "learning_rate": 0.00017917952395176537,
+      "loss": 0.6963,
+      "step": 760
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.039163414388895035,
+      "learning_rate": 0.0001791259842937903,
+      "loss": 0.652,
+      "step": 761
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.04467619210481644,
+      "learning_rate": 0.0001790723839065156,
+      "loss": 0.585,
+      "step": 762
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.03953075036406517,
+      "learning_rate": 0.00017901872283107966,
+      "loss": 0.7537,
+      "step": 763
+    },
+    {
+      "epoch": 0.42,
+      "grad_norm": 0.0408446304500103,
+      "learning_rate": 0.0001789650011086674,
+      "loss": 0.7833,
+      "step": 764
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.05530129000544548,
+      "learning_rate": 0.00017891121878051025,
+      "loss": 0.8796,
+      "step": 765
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.03987415134906769,
+      "learning_rate": 0.00017885737588788634,
+      "loss": 0.6762,
+      "step": 766
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.042246054857969284,
+      "learning_rate": 0.00017880347247212005,
+      "loss": 0.7387,
+      "step": 767
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.04334084317088127,
+      "learning_rate": 0.00017874950857458242,
+      "loss": 0.7526,
+      "step": 768
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.04815899208188057,
+      "learning_rate": 0.00017869548423669077,
+      "loss": 0.8278,
+      "step": 769
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.05985521152615547,
+      "learning_rate": 0.00017864139949990885,
+      "loss": 0.6726,
+      "step": 770
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.038774825632572174,
+      "learning_rate": 0.00017858725440574676,
+      "loss": 0.6549,
+      "step": 771
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.09850037842988968,
+      "learning_rate": 0.00017853304899576093,
+      "loss": 1.1263,
+      "step": 772
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.0418662466108799,
+      "learning_rate": 0.0001784787833115541,
+      "loss": 0.6866,
+      "step": 773
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.04924052208662033,
+      "learning_rate": 0.00017842445739477532,
+      "loss": 0.7591,
+      "step": 774
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.04036710783839226,
+      "learning_rate": 0.0001783700712871197,
+      "loss": 0.7177,
+      "step": 775
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.04530615732073784,
+      "learning_rate": 0.00017831562503032867,
+      "loss": 0.8464,
+      "step": 776
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.04415806382894516,
+      "learning_rate": 0.00017826111866618986,
+      "loss": 0.6993,
+      "step": 777
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.04042017087340355,
+      "learning_rate": 0.00017820655223653692,
+      "loss": 0.6305,
+      "step": 778
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.04573072865605354,
+      "learning_rate": 0.0001781519257832497,
+      "loss": 0.8541,
+      "step": 779
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.04222429171204567,
+      "learning_rate": 0.00017809723934825405,
+      "loss": 0.6579,
+      "step": 780
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.04141012579202652,
+      "learning_rate": 0.0001780424929735219,
+      "loss": 0.7556,
+      "step": 781
+    },
+    {
+      "epoch": 0.43,
+      "grad_norm": 0.049307554960250854,
+      "learning_rate": 0.00017798768670107114,
+      "loss": 0.6392,
+      "step": 782
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.0450025275349617,
+      "learning_rate": 0.0001779328205729657,
+      "loss": 0.7718,
+      "step": 783
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.043787937611341476,
+      "learning_rate": 0.00017787789463131538,
+      "loss": 0.7086,
+      "step": 784
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.03968477621674538,
+      "learning_rate": 0.0001778229089182759,
+      "loss": 0.6744,
+      "step": 785
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.044153615832328796,
+      "learning_rate": 0.00017776786347604892,
+      "loss": 0.7169,
+      "step": 786
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.04364906996488571,
+      "learning_rate": 0.00017771275834688183,
+      "loss": 0.7977,
+      "step": 787
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.04209036007523537,
+      "learning_rate": 0.00017765759357306796,
+      "loss": 0.6939,
+      "step": 788
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.04396096616983414,
+      "learning_rate": 0.00017760236919694627,
+      "loss": 0.7431,
+      "step": 789
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.039003144949674606,
+      "learning_rate": 0.00017754708526090157,
+      "loss": 0.6491,
+      "step": 790
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.04044824466109276,
+      "learning_rate": 0.00017749174180736442,
+      "loss": 0.6875,
+      "step": 791
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.04009506106376648,
+      "learning_rate": 0.0001774363388788109,
+      "loss": 0.6727,
+      "step": 792
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.044321801513433456,
+      "learning_rate": 0.00017738087651776287,
+      "loss": 0.6705,
+      "step": 793
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.047294605523347855,
+      "learning_rate": 0.00017732535476678777,
+      "loss": 0.8299,
+      "step": 794
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.04393976926803589,
+      "learning_rate": 0.00017726977366849865,
+      "loss": 0.8299,
+      "step": 795
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.038830842822790146,
+      "learning_rate": 0.000177214133265554,
+      "loss": 0.768,
+      "step": 796
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.04319118335843086,
+      "learning_rate": 0.00017715843360065797,
+      "loss": 0.83,
+      "step": 797
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.04209451377391815,
+      "learning_rate": 0.00017710267471656016,
+      "loss": 0.6847,
+      "step": 798
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.04092387482523918,
+      "learning_rate": 0.00017704685665605546,
+      "loss": 0.7254,
+      "step": 799
+    },
+    {
+      "epoch": 0.44,
+      "grad_norm": 0.041155993938446045,
+      "learning_rate": 0.00017699097946198444,
+      "loss": 0.6125,
+      "step": 800
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.03681296482682228,
+      "learning_rate": 0.00017693504317723284,
+      "loss": 0.6512,
+      "step": 801
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.04595888406038284,
+      "learning_rate": 0.00017687904784473188,
+      "loss": 0.7719,
+      "step": 802
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.04086902737617493,
+      "learning_rate": 0.00017682299350745803,
+      "loss": 0.6339,
+      "step": 803
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.04581453278660774,
+      "learning_rate": 0.00017676688020843307,
+      "loss": 0.6704,
+      "step": 804
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.041530586779117584,
+      "learning_rate": 0.00017671070799072402,
+      "loss": 0.79,
+      "step": 805
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.04776669666171074,
+      "learning_rate": 0.0001766544768974432,
+      "loss": 0.782,
+      "step": 806
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.041267361491918564,
+      "learning_rate": 0.00017659818697174796,
+      "loss": 0.6647,
+      "step": 807
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.04318636655807495,
+      "learning_rate": 0.00017654183825684092,
+      "loss": 0.7395,
+      "step": 808
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.04291536659002304,
+      "learning_rate": 0.00017648543079596982,
+      "loss": 0.7857,
+      "step": 809
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.04914848133921623,
+      "learning_rate": 0.00017642896463242747,
+      "loss": 0.8949,
+      "step": 810
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.042548660188913345,
+      "learning_rate": 0.00017637243980955168,
+      "loss": 0.6978,
+      "step": 811
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.044454239308834076,
+      "learning_rate": 0.00017631585637072536,
+      "loss": 0.7645,
+      "step": 812
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.044261105358600616,
+      "learning_rate": 0.00017625921435937637,
+      "loss": 0.704,
+      "step": 813
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.03984127193689346,
+      "learning_rate": 0.00017620251381897752,
+      "loss": 0.6803,
+      "step": 814
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.04928235337138176,
+      "learning_rate": 0.00017614575479304662,
+      "loss": 0.6964,
+      "step": 815
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.03964222967624664,
+      "learning_rate": 0.00017608893732514616,
+      "loss": 0.7508,
+      "step": 816
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.04107435792684555,
+      "learning_rate": 0.00017603206145888373,
+      "loss": 0.6977,
+      "step": 817
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 0.03979944810271263,
+      "learning_rate": 0.00017597512723791162,
+      "loss": 0.6823,
+      "step": 818
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04214751720428467,
+      "learning_rate": 0.00017591813470592692,
+      "loss": 0.756,
+      "step": 819
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04152442887425423,
+      "learning_rate": 0.00017586108390667144,
+      "loss": 0.6977,
+      "step": 820
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04409961402416229,
+      "learning_rate": 0.00017580397488393176,
+      "loss": 0.7103,
+      "step": 821
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04341358318924904,
+      "learning_rate": 0.00017574680768153917,
+      "loss": 0.7553,
+      "step": 822
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.040573328733444214,
+      "learning_rate": 0.00017568958234336952,
+      "loss": 0.6273,
+      "step": 823
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04173550009727478,
+      "learning_rate": 0.00017563229891334338,
+      "loss": 0.6156,
+      "step": 824
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.043724823743104935,
+      "learning_rate": 0.00017557495743542585,
+      "loss": 0.681,
+      "step": 825
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04628165438771248,
+      "learning_rate": 0.00017551755795362656,
+      "loss": 0.707,
+      "step": 826
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.0463298037648201,
+      "learning_rate": 0.00017546010051199971,
+      "loss": 0.7532,
+      "step": 827
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04120299220085144,
+      "learning_rate": 0.00017540258515464396,
+      "loss": 0.6755,
+      "step": 828
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04657519608736038,
+      "learning_rate": 0.00017534501192570246,
+      "loss": 0.794,
+      "step": 829
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04502912238240242,
+      "learning_rate": 0.0001752873808693627,
+      "loss": 0.7433,
+      "step": 830
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.05056634172797203,
+      "learning_rate": 0.00017522969202985657,
+      "loss": 0.8462,
+      "step": 831
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04288472607731819,
+      "learning_rate": 0.00017517194545146037,
+      "loss": 0.6889,
+      "step": 832
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.044761136174201965,
+      "learning_rate": 0.0001751141411784947,
+      "loss": 0.6657,
+      "step": 833
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04813416302204132,
+      "learning_rate": 0.00017505627925532442,
+      "loss": 0.7993,
+      "step": 834
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04681932181119919,
+      "learning_rate": 0.00017499835972635856,
+      "loss": 0.8888,
+      "step": 835
+    },
+    {
+      "epoch": 0.46,
+      "grad_norm": 0.04293239116668701,
+      "learning_rate": 0.0001749403826360505,
+      "loss": 0.7787,
+      "step": 836
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04279159754514694,
+      "learning_rate": 0.0001748823480288977,
+      "loss": 0.6838,
+      "step": 837
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04303467273712158,
+      "learning_rate": 0.00017482425594944184,
+      "loss": 0.7993,
+      "step": 838
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.044777967035770416,
+      "learning_rate": 0.0001747661064422686,
+      "loss": 0.8355,
+      "step": 839
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04495497792959213,
+      "learning_rate": 0.00017470789955200788,
+      "loss": 0.8078,
+      "step": 840
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.05037945136427879,
+      "learning_rate": 0.00017464963532333352,
+      "loss": 0.6933,
+      "step": 841
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04452716186642647,
+      "learning_rate": 0.0001745913138009634,
+      "loss": 0.7993,
+      "step": 842
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.042573876678943634,
+      "learning_rate": 0.0001745329350296593,
+      "loss": 0.6218,
+      "step": 843
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04359611123800278,
+      "learning_rate": 0.00017447449905422713,
+      "loss": 0.7138,
+      "step": 844
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.03893902152776718,
+      "learning_rate": 0.00017441600591951647,
+      "loss": 0.5625,
+      "step": 845
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04326329007744789,
+      "learning_rate": 0.00017435745567042095,
+      "loss": 0.808,
+      "step": 846
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04221475496888161,
+      "learning_rate": 0.0001742988483518779,
+      "loss": 0.6852,
+      "step": 847
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04148104414343834,
+      "learning_rate": 0.0001742401840088686,
+      "loss": 0.7226,
+      "step": 848
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.043071821331977844,
+      "learning_rate": 0.00017418146268641794,
+      "loss": 0.7685,
+      "step": 849
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04110527038574219,
+      "learning_rate": 0.00017412268442959465,
+      "loss": 0.6696,
+      "step": 850
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04123936966061592,
+      "learning_rate": 0.00017406384928351113,
+      "loss": 0.6247,
+      "step": 851
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04614235460758209,
+      "learning_rate": 0.00017400495729332337,
+      "loss": 0.7239,
+      "step": 852
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04393448308110237,
+      "learning_rate": 0.00017394600850423114,
+      "loss": 0.6983,
+      "step": 853
+    },
+    {
+      "epoch": 0.47,
+      "grad_norm": 0.04823063313961029,
+      "learning_rate": 0.00017388700296147765,
+      "loss": 0.8802,
+      "step": 854
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.04590068385004997,
+      "learning_rate": 0.00017382794071034975,
+      "loss": 0.764,
+      "step": 855
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.043450817465782166,
+      "learning_rate": 0.00017376882179617783,
+      "loss": 0.6518,
+      "step": 856
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.0406193882226944,
+      "learning_rate": 0.00017370964626433567,
+      "loss": 0.6845,
+      "step": 857
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.0440056286752224,
+      "learning_rate": 0.00017365041416024065,
+      "loss": 0.7249,
+      "step": 858
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.044533830136060715,
+      "learning_rate": 0.00017359112552935347,
+      "loss": 0.7045,
+      "step": 859
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.04595007747411728,
+      "learning_rate": 0.00017353178041717814,
+      "loss": 0.7224,
+      "step": 860
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.04445382580161095,
+      "learning_rate": 0.00017347237886926225,
+      "loss": 0.7391,
+      "step": 861
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.044705115258693695,
+      "learning_rate": 0.0001734129209311965,
+      "loss": 0.8049,
+      "step": 862
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.03983869031071663,
+      "learning_rate": 0.00017335340664861493,
+      "loss": 0.6245,
+      "step": 863
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.03903631865978241,
+      "learning_rate": 0.00017329383606719481,
+      "loss": 0.6075,
+      "step": 864
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.0470118448138237,
+      "learning_rate": 0.00017323420923265673,
+      "loss": 0.7978,
+      "step": 865
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.046046625822782516,
+      "learning_rate": 0.00017317452619076428,
+      "loss": 0.7536,
+      "step": 866
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.044642165303230286,
+      "learning_rate": 0.00017311478698732433,
+      "loss": 0.6868,
+      "step": 867
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.043679915368556976,
+      "learning_rate": 0.0001730549916681868,
+      "loss": 0.6715,
+      "step": 868
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.04374048486351967,
+      "learning_rate": 0.0001729951402792446,
+      "loss": 0.7951,
+      "step": 869
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.04199199378490448,
+      "learning_rate": 0.00017293523286643386,
+      "loss": 0.6943,
+      "step": 870
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.045180678367614746,
+      "learning_rate": 0.00017287526947573354,
+      "loss": 0.7037,
+      "step": 871
+    },
+    {
+      "epoch": 0.48,
+      "grad_norm": 0.04861563816666603,
+      "learning_rate": 0.0001728152501531656,
+      "loss": 0.808,
+      "step": 872
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.06491879373788834,
+      "learning_rate": 0.000172755174944795,
+      "loss": 0.7232,
+      "step": 873
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.056284647434949875,
+      "learning_rate": 0.0001726950438967295,
+      "loss": 0.7938,
+      "step": 874
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.04053090140223503,
+      "learning_rate": 0.00017263485705511984,
+      "loss": 0.67,
+      "step": 875
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.04802017658948898,
+      "learning_rate": 0.00017257461446615942,
+      "loss": 0.8086,
+      "step": 876
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.055473774671554565,
+      "learning_rate": 0.00017251431617608452,
+      "loss": 0.6176,
+      "step": 877
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.04549332708120346,
+      "learning_rate": 0.0001724539622311742,
+      "loss": 0.6832,
+      "step": 878
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.040823038667440414,
+      "learning_rate": 0.00017239355267775018,
+      "loss": 0.6617,
+      "step": 879
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.045911453664302826,
+      "learning_rate": 0.00017233308756217682,
+      "loss": 0.7098,
+      "step": 880
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.044026948511600494,
+      "learning_rate": 0.00017227256693086123,
+      "loss": 0.7281,
+      "step": 881
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.04080135002732277,
+      "learning_rate": 0.00017221199083025307,
+      "loss": 0.6427,
+      "step": 882
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.04171103984117508,
+      "learning_rate": 0.0001721513593068446,
+      "loss": 0.77,
+      "step": 883
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.04672902449965477,
+      "learning_rate": 0.00017209067240717057,
+      "loss": 0.6622,
+      "step": 884
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.047331150621175766,
+      "learning_rate": 0.00017202993017780823,
+      "loss": 0.6539,
+      "step": 885
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.041491780430078506,
+      "learning_rate": 0.0001719691326653774,
+      "loss": 0.693,
+      "step": 886
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.04602917283773422,
+      "learning_rate": 0.00017190827991654014,
+      "loss": 0.6925,
+      "step": 887
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.04347170516848564,
+      "learning_rate": 0.00017184737197800115,
+      "loss": 0.6765,
+      "step": 888
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.04298747330904007,
+      "learning_rate": 0.00017178640889650723,
+      "loss": 0.6898,
+      "step": 889
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 0.04869624972343445,
+      "learning_rate": 0.0001717253907188477,
+      "loss": 0.8587,
+      "step": 890
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.04220546409487724,
+      "learning_rate": 0.0001716643174918541,
+      "loss": 0.6139,
+      "step": 891
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.04683152586221695,
+      "learning_rate": 0.00017160318926240015,
+      "loss": 0.7845,
+      "step": 892
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.04245654493570328,
+      "learning_rate": 0.0001715420060774019,
+      "loss": 0.6194,
+      "step": 893
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.051294971257448196,
+      "learning_rate": 0.00017148076798381755,
+      "loss": 0.7762,
+      "step": 894
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.05587690323591232,
+      "learning_rate": 0.00017141947502864738,
+      "loss": 0.7732,
+      "step": 895
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.04731575399637222,
+      "learning_rate": 0.00017135812725893381,
+      "loss": 0.7229,
+      "step": 896
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.04307183623313904,
+      "learning_rate": 0.00017129672472176134,
+      "loss": 0.6533,
+      "step": 897
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.044035330414772034,
+      "learning_rate": 0.00017123526746425652,
+      "loss": 0.7754,
+      "step": 898
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.04161735251545906,
+      "learning_rate": 0.00017117375553358786,
+      "loss": 0.6989,
+      "step": 899
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.0438537523150444,
+      "learning_rate": 0.00017111218897696587,
+      "loss": 0.742,
+      "step": 900
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.04731033742427826,
+      "learning_rate": 0.00017105056784164294,
+      "loss": 0.6689,
+      "step": 901
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.04555543512105942,
+      "learning_rate": 0.00017098889217491338,
+      "loss": 0.7309,
+      "step": 902
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.041630081832408905,
+      "learning_rate": 0.00017092716202411336,
+      "loss": 0.6845,
+      "step": 903
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.045560967177152634,
+      "learning_rate": 0.00017086537743662086,
+      "loss": 0.6913,
+      "step": 904
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.03997902199625969,
+      "learning_rate": 0.00017080353845985559,
+      "loss": 0.6016,
+      "step": 905
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.04500821977853775,
+      "learning_rate": 0.0001707416451412791,
+      "loss": 0.7355,
+      "step": 906
+    },
+    {
+      "epoch": 0.5,
+      "grad_norm": 0.04442552849650383,
+      "learning_rate": 0.00017067969752839458,
+      "loss": 0.6525,
+      "step": 907
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.04527292773127556,
+      "learning_rate": 0.0001706176956687469,
+      "loss": 0.6991,
+      "step": 908
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.04470342397689819,
+      "learning_rate": 0.00017055563960992256,
+      "loss": 0.7352,
+      "step": 909
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.04706728830933571,
+      "learning_rate": 0.00017049352939954967,
+      "loss": 0.7081,
+      "step": 910
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.049026861786842346,
+      "learning_rate": 0.00017043136508529793,
+      "loss": 0.833,
+      "step": 911
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.04529300704598427,
+      "learning_rate": 0.00017036914671487852,
+      "loss": 0.6624,
+      "step": 912
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.05243083834648132,
+      "learning_rate": 0.0001703068743360441,
+      "loss": 0.9235,
+      "step": 913
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.050981760025024414,
+      "learning_rate": 0.00017024454799658884,
+      "loss": 0.866,
+      "step": 914
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.045431166887283325,
+      "learning_rate": 0.00017018216774434828,
+      "loss": 0.7997,
+      "step": 915
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.049921710044145584,
+      "learning_rate": 0.00017011973362719932,
+      "loss": 0.6015,
+      "step": 916
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.04454491659998894,
+      "learning_rate": 0.00017005724569306026,
+      "loss": 0.6789,
+      "step": 917
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.04739166796207428,
+      "learning_rate": 0.0001699947039898907,
+      "loss": 0.7727,
+      "step": 918
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.04357178509235382,
+      "learning_rate": 0.0001699321085656914,
+      "loss": 0.71,
+      "step": 919
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.04351083189249039,
+      "learning_rate": 0.00016986945946850446,
+      "loss": 0.7437,
+      "step": 920
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.04492335394024849,
+      "learning_rate": 0.00016980675674641322,
+      "loss": 0.7705,
+      "step": 921
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.04446355253458023,
+      "learning_rate": 0.000169744000447542,
+      "loss": 0.6758,
+      "step": 922
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.04474789649248123,
+      "learning_rate": 0.00016968119062005642,
+      "loss": 0.665,
+      "step": 923
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.03871838003396988,
+      "learning_rate": 0.00016961832731216307,
+      "loss": 0.644,
+      "step": 924
+    },
+    {
+      "epoch": 0.51,
+      "grad_norm": 0.04366254433989525,
+      "learning_rate": 0.00016955541057210965,
+      "loss": 0.8131,
+      "step": 925
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.04740358516573906,
+      "learning_rate": 0.0001694924404481848,
+      "loss": 0.6964,
+      "step": 926
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.04698627069592476,
+      "learning_rate": 0.00016942941698871818,
+      "loss": 0.6736,
+      "step": 927
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.0437115915119648,
+      "learning_rate": 0.00016936634024208047,
+      "loss": 0.7068,
+      "step": 928
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.04518444836139679,
+      "learning_rate": 0.00016930321025668306,
+      "loss": 0.7357,
+      "step": 929
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.0465158186852932,
+      "learning_rate": 0.00016924002708097833,
+      "loss": 0.7876,
+      "step": 930
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.04457319527864456,
+      "learning_rate": 0.00016917679076345943,
+      "loss": 0.7335,
+      "step": 931
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.05104387551546097,
+      "learning_rate": 0.00016911350135266035,
+      "loss": 0.7055,
+      "step": 932
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.04391132667660713,
+      "learning_rate": 0.0001690501588971558,
+      "loss": 0.7218,
+      "step": 933
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.042982205748558044,
+      "learning_rate": 0.00016898676344556118,
+      "loss": 0.7202,
+      "step": 934
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.042211346328258514,
+      "learning_rate": 0.00016892331504653259,
+      "loss": 0.708,
+      "step": 935
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.043890226632356644,
+      "learning_rate": 0.00016885981374876677,
+      "loss": 0.7524,
+      "step": 936
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.044012345373630524,
+      "learning_rate": 0.00016879625960100104,
+      "loss": 0.7103,
+      "step": 937
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.0452919527888298,
+      "learning_rate": 0.0001687326526520133,
+      "loss": 0.7012,
+      "step": 938
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.04428261145949364,
+      "learning_rate": 0.00016866899295062197,
+      "loss": 0.6329,
+      "step": 939
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.043744850903749466,
+      "learning_rate": 0.00016860528054568597,
+      "loss": 0.7682,
+      "step": 940
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.0442328006029129,
+      "learning_rate": 0.00016854151548610462,
+      "loss": 0.6931,
+      "step": 941
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.044094718992710114,
+      "learning_rate": 0.00016847769782081772,
+      "loss": 0.7495,
+      "step": 942
+    },
+    {
+      "epoch": 0.52,
+      "grad_norm": 0.041264671832323074,
+      "learning_rate": 0.00016841382759880542,
+      "loss": 0.6035,
+      "step": 943
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.03848943114280701,
+      "learning_rate": 0.00016834990486908817,
+      "loss": 0.6,
+      "step": 944
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.0429924838244915,
+      "learning_rate": 0.00016828592968072678,
+      "loss": 0.7149,
+      "step": 945
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.037183403968811035,
+      "learning_rate": 0.00016822190208282226,
+      "loss": 0.6253,
+      "step": 946
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.04193463176488876,
+      "learning_rate": 0.00016815782212451592,
+      "loss": 0.7332,
+      "step": 947
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.04574836045503616,
+      "learning_rate": 0.00016809368985498918,
+      "loss": 0.676,
+      "step": 948
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.044316504150629044,
+      "learning_rate": 0.0001680295053234637,
+      "loss": 0.6696,
+      "step": 949
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.04649119824171066,
+      "learning_rate": 0.00016796526857920112,
+      "loss": 0.6853,
+      "step": 950
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.041028618812561035,
+      "learning_rate": 0.00016790097967150325,
+      "loss": 0.5797,
+      "step": 951
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.045178357511758804,
+      "learning_rate": 0.00016783663864971193,
+      "loss": 0.733,
+      "step": 952
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.044117413461208344,
+      "learning_rate": 0.00016777224556320896,
+      "loss": 0.6571,
+      "step": 953
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.04647046700119972,
+      "learning_rate": 0.00016770780046141616,
+      "loss": 0.7674,
+      "step": 954
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.04585673660039902,
+      "learning_rate": 0.0001676433033937952,
+      "loss": 0.7153,
+      "step": 955
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.04684153199195862,
+      "learning_rate": 0.00016757875440984768,
+      "loss": 0.712,
+      "step": 956
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.05014558881521225,
+      "learning_rate": 0.000167514153559115,
+      "loss": 0.7872,
+      "step": 957
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.04870294779539108,
+      "learning_rate": 0.00016744950089117846,
+      "loss": 0.7779,
+      "step": 958
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.047236453741788864,
+      "learning_rate": 0.00016738479645565902,
+      "loss": 0.6254,
+      "step": 959
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.04619355872273445,
+      "learning_rate": 0.00016732004030221745,
+      "loss": 0.7061,
+      "step": 960
+    },
+    {
+      "epoch": 0.53,
+      "grad_norm": 0.04691663756966591,
+      "learning_rate": 0.00016725523248055415,
+      "loss": 0.7212,
+      "step": 961
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.046586498618125916,
+      "learning_rate": 0.00016719037304040922,
+      "loss": 0.7438,
+      "step": 962
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.047182347625494,
+      "learning_rate": 0.00016712546203156243,
+      "loss": 0.7781,
+      "step": 963
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.04682426154613495,
+      "learning_rate": 0.000167060499503833,
+      "loss": 0.7651,
+      "step": 964
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.04150580242276192,
+      "learning_rate": 0.00016699548550707974,
+      "loss": 0.6261,
+      "step": 965
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.047207970172166824,
+      "learning_rate": 0.00016693042009120106,
+      "loss": 0.7482,
+      "step": 966
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.03972497582435608,
+      "learning_rate": 0.0001668653033061347,
+      "loss": 0.6568,
+      "step": 967
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.05301308631896973,
+      "learning_rate": 0.00016680013520185786,
+      "loss": 0.8632,
+      "step": 968
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.044552650302648544,
+      "learning_rate": 0.0001667349158283872,
+      "loss": 0.7905,
+      "step": 969
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.0502367801964283,
+      "learning_rate": 0.0001666696452357787,
+      "loss": 0.694,
+      "step": 970
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.047164347022771835,
+      "learning_rate": 0.00016660432347412752,
+      "loss": 0.6664,
+      "step": 971
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.04584898054599762,
+      "learning_rate": 0.00016653895059356828,
+      "loss": 0.6614,
+      "step": 972
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.04450858384370804,
+      "learning_rate": 0.00016647352664427473,
+      "loss": 0.7293,
+      "step": 973
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.045772675424814224,
+      "learning_rate": 0.00016640805167645985,
+      "loss": 0.6825,
+      "step": 974
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.04510961472988129,
+      "learning_rate": 0.00016634252574037575,
+      "loss": 0.628,
+      "step": 975
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.0424661748111248,
+      "learning_rate": 0.00016627694888631377,
+      "loss": 0.7242,
+      "step": 976
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.04516896605491638,
+      "learning_rate": 0.00016621132116460404,
+      "loss": 0.728,
+      "step": 977
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.04178481549024582,
+      "learning_rate": 0.00016614564262561608,
+      "loss": 0.6749,
+      "step": 978
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 0.04447667673230171,
+      "learning_rate": 0.00016607991331975822,
+      "loss": 0.7377,
+      "step": 979
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.04504280164837837,
+      "learning_rate": 0.0001660141332974778,
+      "loss": 0.6817,
+      "step": 980
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.0446137972176075,
+      "learning_rate": 0.00016594830260926102,
+      "loss": 0.7593,
+      "step": 981
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.04577165096998215,
+      "learning_rate": 0.0001658824213056331,
+      "loss": 0.6594,
+      "step": 982
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.04543627053499222,
+      "learning_rate": 0.00016581648943715796,
+      "loss": 0.7227,
+      "step": 983
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.051089901477098465,
+      "learning_rate": 0.0001657505070544384,
+      "loss": 0.6931,
+      "step": 984
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.04433685168623924,
+      "learning_rate": 0.00016568447420811612,
+      "loss": 0.7223,
+      "step": 985
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.04630895331501961,
+      "learning_rate": 0.00016561839094887123,
+      "loss": 0.8788,
+      "step": 986
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.04735618084669113,
+      "learning_rate": 0.00016555225732742281,
+      "loss": 0.6527,
+      "step": 987
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.04732111841440201,
+      "learning_rate": 0.00016548607339452853,
+      "loss": 0.7427,
+      "step": 988
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.04409315809607506,
+      "learning_rate": 0.0001654198392009846,
+      "loss": 0.7473,
+      "step": 989
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.044037654995918274,
+      "learning_rate": 0.00016535355479762586,
+      "loss": 0.735,
+      "step": 990
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.043528955429792404,
+      "learning_rate": 0.00016528722023532574,
+      "loss": 0.6415,
+      "step": 991
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.04991381987929344,
+      "learning_rate": 0.00016522083556499596,
+      "loss": 0.7989,
+      "step": 992
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.044087547808885574,
+      "learning_rate": 0.000165154400837587,
+      "loss": 0.7178,
+      "step": 993
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.04327988997101784,
+      "learning_rate": 0.00016508791610408754,
+      "loss": 0.6599,
+      "step": 994
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.04584032669663429,
+      "learning_rate": 0.0001650213814155247,
+      "loss": 0.6482,
+      "step": 995
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.05077075958251953,
+      "learning_rate": 0.00016495479682296395,
+      "loss": 0.727,
+      "step": 996
+    },
+    {
+      "epoch": 0.55,
+      "grad_norm": 0.04770886152982712,
+      "learning_rate": 0.00016488816237750906,
+      "loss": 0.6737,
+      "step": 997
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.04274902865290642,
+      "learning_rate": 0.00016482147813030203,
+      "loss": 0.6277,
+      "step": 998
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.048387110233306885,
+      "learning_rate": 0.00016475474413252315,
+      "loss": 0.7836,
+      "step": 999
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.04784883186221123,
+      "learning_rate": 0.0001646879604353908,
+      "loss": 0.8291,
+      "step": 1000
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.047025181353092194,
+      "learning_rate": 0.00016462112709016163,
+      "loss": 0.7344,
+      "step": 1001
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.042724139988422394,
+      "learning_rate": 0.00016455424414813026,
+      "loss": 0.607,
+      "step": 1002
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.04707411676645279,
+      "learning_rate": 0.00016448731166062946,
+      "loss": 0.7879,
+      "step": 1003
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.048031605780124664,
+      "learning_rate": 0.00016442032967903,
+      "loss": 0.7665,
+      "step": 1004
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.0522460862994194,
+      "learning_rate": 0.00016435329825474065,
+      "loss": 0.8697,
+      "step": 1005
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.043403007090091705,
+      "learning_rate": 0.00016428621743920814,
+      "loss": 0.6733,
+      "step": 1006
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.04464132711291313,
+      "learning_rate": 0.00016421908728391703,
+      "loss": 0.6495,
+      "step": 1007
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.04746426269412041,
+      "learning_rate": 0.00016415190784038983,
+      "loss": 0.6364,
+      "step": 1008
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.04561330005526543,
+      "learning_rate": 0.00016408467916018688,
+      "loss": 0.7492,
+      "step": 1009
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.046681590378284454,
+      "learning_rate": 0.00016401740129490624,
+      "loss": 0.6807,
+      "step": 1010
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.04807223007082939,
+      "learning_rate": 0.00016395007429618382,
+      "loss": 0.7732,
+      "step": 1011
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.04678153246641159,
+      "learning_rate": 0.00016388269821569312,
+      "loss": 0.753,
+      "step": 1012
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.0424916073679924,
+      "learning_rate": 0.0001638152731051454,
+      "loss": 0.6378,
+      "step": 1013
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.04331868886947632,
+      "learning_rate": 0.00016374779901628952,
+      "loss": 0.766,
+      "step": 1014
+    },
+    {
+      "epoch": 0.56,
+      "grad_norm": 0.043369825929403305,
+      "learning_rate": 0.00016368027600091194,
+      "loss": 0.6492,
+      "step": 1015
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.04750450327992439,
+      "learning_rate": 0.00016361270411083666,
+      "loss": 0.7226,
+      "step": 1016
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.04926105588674545,
+      "learning_rate": 0.0001635450833979252,
+      "loss": 0.7402,
+      "step": 1017
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.04888901859521866,
+      "learning_rate": 0.00016347741391407655,
+      "loss": 0.7061,
+      "step": 1018
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.046233415603637695,
+      "learning_rate": 0.00016340969571122712,
+      "loss": 0.7367,
+      "step": 1019
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.0504348948597908,
+      "learning_rate": 0.00016334192884135074,
+      "loss": 0.7785,
+      "step": 1020
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.04318249598145485,
+      "learning_rate": 0.00016327411335645853,
+      "loss": 0.6495,
+      "step": 1021
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.04340077564120293,
+      "learning_rate": 0.00016320624930859904,
+      "loss": 0.6685,
+      "step": 1022
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.04627622663974762,
+      "learning_rate": 0.00016313833674985796,
+      "loss": 0.6857,
+      "step": 1023
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.04838955029845238,
+      "learning_rate": 0.0001630703757323583,
+      "loss": 0.7336,
+      "step": 1024
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.04140018671751022,
+      "learning_rate": 0.00016300236630826024,
+      "loss": 0.6872,
+      "step": 1025
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.04307800903916359,
+      "learning_rate": 0.000162934308529761,
+      "loss": 0.689,
+      "step": 1026
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.046013642102479935,
+      "learning_rate": 0.00016286620244909518,
+      "loss": 0.676,
+      "step": 1027
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.045836132019758224,
+      "learning_rate": 0.0001627980481185342,
+      "loss": 0.7321,
+      "step": 1028
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.046595752239227295,
+      "learning_rate": 0.00016272984559038655,
+      "loss": 0.7788,
+      "step": 1029
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.041627172380685806,
+      "learning_rate": 0.0001626615949169979,
+      "loss": 0.5976,
+      "step": 1030
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.056944046169519424,
+      "learning_rate": 0.00016259329615075062,
+      "loss": 0.937,
+      "step": 1031
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.0483165942132473,
+      "learning_rate": 0.00016252494934406415,
+      "loss": 0.6425,
+      "step": 1032
+    },
+    {
+      "epoch": 0.57,
+      "grad_norm": 0.05059583857655525,
+      "learning_rate": 0.00016245655454939474,
+      "loss": 0.7802,
+      "step": 1033
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.04494575038552284,
+      "learning_rate": 0.0001623881118192355,
+      "loss": 0.709,
+      "step": 1034
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.04515180364251137,
+      "learning_rate": 0.00016231962120611635,
+      "loss": 0.7219,
+      "step": 1035
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.051081229001283646,
+      "learning_rate": 0.00016225108276260385,
+      "loss": 0.7955,
+      "step": 1036
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.04739641770720482,
+      "learning_rate": 0.00016218249654130138,
+      "loss": 0.7017,
+      "step": 1037
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.048389632254838943,
+      "learning_rate": 0.000162113862594849,
+      "loss": 0.7864,
+      "step": 1038
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.04880169779062271,
+      "learning_rate": 0.0001620451809759233,
+      "loss": 0.7688,
+      "step": 1039
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.044919900596141815,
+      "learning_rate": 0.00016197645173723758,
+      "loss": 0.754,
+      "step": 1040
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.04826303571462631,
+      "learning_rate": 0.00016190767493154154,
+      "loss": 0.7931,
+      "step": 1041
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.042201928794384,
+      "learning_rate": 0.0001618388506116215,
+      "loss": 0.603,
+      "step": 1042
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.051390066742897034,
+      "learning_rate": 0.00016176997883030026,
+      "loss": 0.7412,
+      "step": 1043
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.045629408210515976,
+      "learning_rate": 0.00016170105964043695,
+      "loss": 0.7082,
+      "step": 1044
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.04842764884233475,
+      "learning_rate": 0.0001616320930949272,
+      "loss": 0.7796,
+      "step": 1045
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.04266642406582832,
+      "learning_rate": 0.00016156307924670288,
+      "loss": 0.6521,
+      "step": 1046
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.04461183398962021,
+      "learning_rate": 0.00016149401814873223,
+      "loss": 0.7277,
+      "step": 1047
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.044011376798152924,
+      "learning_rate": 0.0001614249098540197,
+      "loss": 0.7393,
+      "step": 1048
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.04548850655555725,
+      "learning_rate": 0.0001613557544156061,
+      "loss": 0.6724,
+      "step": 1049
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.04711510241031647,
+      "learning_rate": 0.00016128655188656817,
+      "loss": 0.75,
+      "step": 1050
+    },
+    {
+      "epoch": 0.58,
+      "grad_norm": 0.04443375766277313,
+      "learning_rate": 0.00016121730232001905,
+      "loss": 0.692,
+      "step": 1051
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.047497253865003586,
+      "learning_rate": 0.00016114800576910788,
+      "loss": 0.6364,
+      "step": 1052
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.04716882482171059,
+      "learning_rate": 0.00016107866228701981,
+      "loss": 0.8157,
+      "step": 1053
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.049585748463869095,
+      "learning_rate": 0.0001610092719269761,
+      "loss": 0.7222,
+      "step": 1054
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.04663019999861717,
+      "learning_rate": 0.0001609398347422339,
+      "loss": 0.674,
+      "step": 1055
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.04515184462070465,
+      "learning_rate": 0.00016087035078608637,
+      "loss": 0.641,
+      "step": 1056
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.04762982949614525,
+      "learning_rate": 0.0001608008201118625,
+      "loss": 0.6312,
+      "step": 1057
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.04451216757297516,
+      "learning_rate": 0.00016073124277292728,
+      "loss": 0.7158,
+      "step": 1058
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.0480036623775959,
+      "learning_rate": 0.0001606616188226813,
+      "loss": 0.7029,
+      "step": 1059
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.04552299529314041,
+      "learning_rate": 0.00016059194831456107,
+      "loss": 0.7185,
+      "step": 1060
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.04811175912618637,
+      "learning_rate": 0.0001605222313020388,
+      "loss": 0.5746,
+      "step": 1061
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.045087020844221115,
+      "learning_rate": 0.0001604524678386224,
+      "loss": 0.7196,
+      "step": 1062
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.05363542214035988,
+      "learning_rate": 0.00016038265797785542,
+      "loss": 0.7144,
+      "step": 1063
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.05288343131542206,
+      "learning_rate": 0.00016031280177331706,
+      "loss": 0.7173,
+      "step": 1064
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.046650927513837814,
+      "learning_rate": 0.00016024289927862195,
+      "loss": 0.789,
+      "step": 1065
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.048903193324804306,
+      "learning_rate": 0.00016017295054742046,
+      "loss": 0.7106,
+      "step": 1066
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.04834052920341492,
+      "learning_rate": 0.00016010295563339825,
+      "loss": 0.6996,
+      "step": 1067
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.05408628657460213,
+      "learning_rate": 0.00016003291459027653,
+      "loss": 0.8056,
+      "step": 1068
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 0.04541575536131859,
+      "learning_rate": 0.00015996282747181197,
+      "loss": 0.6443,
+      "step": 1069
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.04552549123764038,
+      "learning_rate": 0.00015989269433179645,
+      "loss": 0.69,
+      "step": 1070
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.0493139810860157,
+      "learning_rate": 0.00015982251522405727,
+      "loss": 0.7327,
+      "step": 1071
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.049078866839408875,
+      "learning_rate": 0.000159752290202457,
+      "loss": 0.8271,
+      "step": 1072
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.0488913394510746,
+      "learning_rate": 0.0001596820193208934,
+      "loss": 0.7298,
+      "step": 1073
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.04880421981215477,
+      "learning_rate": 0.00015961170263329948,
+      "loss": 0.6863,
+      "step": 1074
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.04644745960831642,
+      "learning_rate": 0.00015954134019364346,
+      "loss": 0.6754,
+      "step": 1075
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.05141911655664444,
+      "learning_rate": 0.00015947093205592855,
+      "loss": 0.7986,
+      "step": 1076
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.052950434386730194,
+      "learning_rate": 0.00015940047827419303,
+      "loss": 0.8126,
+      "step": 1077
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.04494583606719971,
+      "learning_rate": 0.0001593299789025104,
+      "loss": 0.724,
+      "step": 1078
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.051292311400175095,
+      "learning_rate": 0.00015925943399498898,
+      "loss": 0.7201,
+      "step": 1079
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.049635812640190125,
+      "learning_rate": 0.000159188843605772,
+      "loss": 0.6293,
+      "step": 1080
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.08753520250320435,
+      "learning_rate": 0.00015911820778903777,
+      "loss": 0.7986,
+      "step": 1081
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.05015747994184494,
+      "learning_rate": 0.00015904752659899935,
+      "loss": 0.7774,
+      "step": 1082
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.05062435567378998,
+      "learning_rate": 0.00015897680008990467,
+      "loss": 0.7423,
+      "step": 1083
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.04625916853547096,
+      "learning_rate": 0.00015890602831603634,
+      "loss": 0.7179,
+      "step": 1084
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.06423801183700562,
+      "learning_rate": 0.00015883521133171186,
+      "loss": 0.6282,
+      "step": 1085
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.048834312707185745,
+      "learning_rate": 0.00015876434919128335,
+      "loss": 0.7035,
+      "step": 1086
+    },
+    {
+      "epoch": 0.6,
+      "grad_norm": 0.049968525767326355,
+      "learning_rate": 0.0001586934419491376,
+      "loss": 0.777,
+      "step": 1087
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.04475943744182587,
+      "learning_rate": 0.00015862248965969604,
+      "loss": 0.6788,
+      "step": 1088
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.046110913157463074,
+      "learning_rate": 0.0001585514923774146,
+      "loss": 0.7237,
+      "step": 1089
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.047238875180482864,
+      "learning_rate": 0.0001584804501567838,
+      "loss": 0.7412,
+      "step": 1090
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.04489819332957268,
+      "learning_rate": 0.00015840936305232869,
+      "loss": 0.6327,
+      "step": 1091
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.04712117463350296,
+      "learning_rate": 0.00015833823111860863,
+      "loss": 0.7175,
+      "step": 1092
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.05276874080300331,
+      "learning_rate": 0.0001582670544102175,
+      "loss": 0.626,
+      "step": 1093
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.047401316463947296,
+      "learning_rate": 0.0001581958329817836,
+      "loss": 0.7426,
+      "step": 1094
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.049675095826387405,
+      "learning_rate": 0.0001581245668879694,
+      "loss": 0.7332,
+      "step": 1095
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.048146385699510574,
+      "learning_rate": 0.00015805325618347172,
+      "loss": 0.7209,
+      "step": 1096
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.04951384291052818,
+      "learning_rate": 0.00015798190092302164,
+      "loss": 0.6746,
+      "step": 1097
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.0463559590280056,
+      "learning_rate": 0.0001579105011613844,
+      "loss": 0.6808,
+      "step": 1098
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.04673503711819649,
+      "learning_rate": 0.00015783905695335946,
+      "loss": 0.6946,
+      "step": 1099
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.04709373414516449,
+      "learning_rate": 0.0001577675683537803,
+      "loss": 0.691,
+      "step": 1100
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.05507972836494446,
+      "learning_rate": 0.00015769603541751455,
+      "loss": 0.7342,
+      "step": 1101
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.043717097491025925,
+      "learning_rate": 0.00015762445819946384,
+      "loss": 0.6183,
+      "step": 1102
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.04856884479522705,
+      "learning_rate": 0.0001575528367545637,
+      "loss": 0.7489,
+      "step": 1103
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.04887085407972336,
+      "learning_rate": 0.0001574811711377838,
+      "loss": 0.7202,
+      "step": 1104
+    },
+    {
+      "epoch": 0.61,
+      "grad_norm": 0.05098757520318031,
+      "learning_rate": 0.00015740946140412753,
+      "loss": 0.7171,
+      "step": 1105
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.05076475441455841,
+      "learning_rate": 0.0001573377076086322,
+      "loss": 0.6581,
+      "step": 1106
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.05398886650800705,
+      "learning_rate": 0.00015726590980636896,
+      "loss": 0.8532,
+      "step": 1107
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.05237511545419693,
+      "learning_rate": 0.00015719406805244276,
+      "loss": 0.6418,
+      "step": 1108
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.05127400532364845,
+      "learning_rate": 0.00015712218240199218,
+      "loss": 0.75,
+      "step": 1109
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.051004305481910706,
+      "learning_rate": 0.0001570502529101896,
+      "loss": 0.6545,
+      "step": 1110
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.05179214105010033,
+      "learning_rate": 0.000156978279632241,
+      "loss": 0.6854,
+      "step": 1111
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.04017964377999306,
+      "learning_rate": 0.0001569062626233859,
+      "loss": 0.5733,
+      "step": 1112
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.05255338177084923,
+      "learning_rate": 0.00015683420193889753,
+      "loss": 0.9239,
+      "step": 1113
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.04868584871292114,
+      "learning_rate": 0.00015676209763408253,
+      "loss": 0.7038,
+      "step": 1114
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.04898803308606148,
+      "learning_rate": 0.0001566899497642811,
+      "loss": 0.6677,
+      "step": 1115
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.044663190841674805,
+      "learning_rate": 0.00015661775838486674,
+      "loss": 0.741,
+      "step": 1116
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.048298414796590805,
+      "learning_rate": 0.00015654552355124648,
+      "loss": 0.7292,
+      "step": 1117
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.04956801235675812,
+      "learning_rate": 0.00015647324531886065,
+      "loss": 0.7475,
+      "step": 1118
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.051773227751255035,
+      "learning_rate": 0.0001564009237431829,
+      "loss": 0.751,
+      "step": 1119
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.04530036076903343,
+      "learning_rate": 0.00015632855887972008,
+      "loss": 0.7123,
+      "step": 1120
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.05154275894165039,
+      "learning_rate": 0.00015625615078401244,
+      "loss": 0.8531,
+      "step": 1121
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.04365009814500809,
+      "learning_rate": 0.00015618369951163317,
+      "loss": 0.6508,
+      "step": 1122
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 0.04828101769089699,
+      "learning_rate": 0.00015611120511818877,
+      "loss": 0.6855,
+      "step": 1123
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.04952272027730942,
+      "learning_rate": 0.00015603866765931875,
+      "loss": 0.6806,
+      "step": 1124
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.0446164608001709,
+      "learning_rate": 0.00015596608719069578,
+      "loss": 0.6364,
+      "step": 1125
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.05198860913515091,
+      "learning_rate": 0.00015589346376802544,
+      "loss": 0.6415,
+      "step": 1126
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.04873516410589218,
+      "learning_rate": 0.00015582079744704626,
+      "loss": 0.6668,
+      "step": 1127
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.04862043261528015,
+      "learning_rate": 0.00015574808828352978,
+      "loss": 0.6277,
+      "step": 1128
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.050693485885858536,
+      "learning_rate": 0.00015567533633328035,
+      "loss": 0.6917,
+      "step": 1129
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.048907842487096786,
+      "learning_rate": 0.00015560254165213522,
+      "loss": 0.7091,
+      "step": 1130
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.04532686248421669,
+      "learning_rate": 0.00015552970429596438,
+      "loss": 0.622,
+      "step": 1131
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.043171364814043045,
+      "learning_rate": 0.00015545682432067067,
+      "loss": 0.6719,
+      "step": 1132
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.05676614120602608,
+      "learning_rate": 0.00015538390178218946,
+      "loss": 0.8366,
+      "step": 1133
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.045390404760837555,
+      "learning_rate": 0.00015531093673648897,
+      "loss": 0.6162,
+      "step": 1134
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.05167704075574875,
+      "learning_rate": 0.00015523792923957,
+      "loss": 0.7946,
+      "step": 1135
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.04950125515460968,
+      "learning_rate": 0.00015516487934746578,
+      "loss": 0.6541,
+      "step": 1136
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.0524483397603035,
+      "learning_rate": 0.00015509178711624233,
+      "loss": 0.7793,
+      "step": 1137
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.046446293592453,
+      "learning_rate": 0.00015501865260199795,
+      "loss": 0.6489,
+      "step": 1138
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.0487704761326313,
+      "learning_rate": 0.00015494547586086352,
+      "loss": 0.6154,
+      "step": 1139
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.04823431745171547,
+      "learning_rate": 0.00015487225694900224,
+      "loss": 0.6899,
+      "step": 1140
+    },
+    {
+      "epoch": 0.63,
+      "grad_norm": 0.04303416609764099,
+      "learning_rate": 0.00015479899592260983,
+      "loss": 0.5816,
+      "step": 1141
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.048107780516147614,
+      "learning_rate": 0.0001547256928379141,
+      "loss": 0.6998,
+      "step": 1142
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.04937562346458435,
+      "learning_rate": 0.0001546523477511754,
+      "loss": 0.7552,
+      "step": 1143
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.05084756389260292,
+      "learning_rate": 0.00015457896071868602,
+      "loss": 0.6517,
+      "step": 1144
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.04428539052605629,
+      "learning_rate": 0.00015450553179677075,
+      "loss": 0.6143,
+      "step": 1145
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.050209447741508484,
+      "learning_rate": 0.0001544320610417863,
+      "loss": 0.6805,
+      "step": 1146
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.04966219514608383,
+      "learning_rate": 0.0001543585485101216,
+      "loss": 0.7239,
+      "step": 1147
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.0482448972761631,
+      "learning_rate": 0.00015428499425819764,
+      "loss": 0.7492,
+      "step": 1148
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.05310830846428871,
+      "learning_rate": 0.00015421139834246737,
+      "loss": 0.7678,
+      "step": 1149
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.04781487584114075,
+      "learning_rate": 0.0001541377608194158,
+      "loss": 0.7321,
+      "step": 1150
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.050008464604616165,
+      "learning_rate": 0.00015406408174555976,
+      "loss": 0.7224,
+      "step": 1151
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.047798726707696915,
+      "learning_rate": 0.00015399036117744812,
+      "loss": 0.6562,
+      "step": 1152
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.04253503307700157,
+      "learning_rate": 0.00015391659917166143,
+      "loss": 0.613,
+      "step": 1153
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.04759254679083824,
+      "learning_rate": 0.00015384279578481221,
+      "loss": 0.6502,
+      "step": 1154
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.044068843126297,
+      "learning_rate": 0.00015376895107354464,
+      "loss": 0.6639,
+      "step": 1155
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.04848353564739227,
+      "learning_rate": 0.00015369506509453458,
+      "loss": 0.6952,
+      "step": 1156
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.046723756939172745,
+      "learning_rate": 0.00015362113790448967,
+      "loss": 0.5937,
+      "step": 1157
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.050520990043878555,
+      "learning_rate": 0.0001535471695601491,
+      "loss": 0.7107,
+      "step": 1158
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 0.04594268649816513,
+      "learning_rate": 0.00015347316011828373,
+      "loss": 0.6756,
+      "step": 1159
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.05000408738851547,
+      "learning_rate": 0.00015339910963569584,
+      "loss": 0.6676,
+      "step": 1160
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.05357789620757103,
+      "learning_rate": 0.00015332501816921928,
+      "loss": 0.7249,
+      "step": 1161
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.051022421568632126,
+      "learning_rate": 0.00015325088577571939,
+      "loss": 0.7145,
+      "step": 1162
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.04610150307416916,
+      "learning_rate": 0.00015317671251209285,
+      "loss": 0.6579,
+      "step": 1163
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.04873840510845184,
+      "learning_rate": 0.00015310249843526776,
+      "loss": 0.7044,
+      "step": 1164
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.046522945165634155,
+      "learning_rate": 0.00015302824360220353,
+      "loss": 0.6772,
+      "step": 1165
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.046785563230514526,
+      "learning_rate": 0.0001529539480698908,
+      "loss": 0.6593,
+      "step": 1166
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.04677551984786987,
+      "learning_rate": 0.00015287961189535155,
+      "loss": 0.7643,
+      "step": 1167
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.057435374706983566,
+      "learning_rate": 0.00015280523513563885,
+      "loss": 0.7578,
+      "step": 1168
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.048298779875040054,
+      "learning_rate": 0.000152730817847837,
+      "loss": 0.7108,
+      "step": 1169
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.04588484391570091,
+      "learning_rate": 0.00015265636008906133,
+      "loss": 0.6706,
+      "step": 1170
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.05275345966219902,
+      "learning_rate": 0.00015258186191645829,
+      "loss": 0.7065,
+      "step": 1171
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.045389268547296524,
+      "learning_rate": 0.00015250732338720533,
+      "loss": 0.6161,
+      "step": 1172
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.051542263478040695,
+      "learning_rate": 0.00015243274455851085,
+      "loss": 0.6771,
+      "step": 1173
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.04666198417544365,
+      "learning_rate": 0.00015235812548761426,
+      "loss": 0.6068,
+      "step": 1174
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.04389272257685661,
+      "learning_rate": 0.00015228346623178573,
+      "loss": 0.6682,
+      "step": 1175
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.044190727174282074,
+      "learning_rate": 0.00015220876684832638,
+      "loss": 0.6145,
+      "step": 1176
+    },
+    {
+      "epoch": 0.65,
+      "grad_norm": 0.049365028738975525,
+      "learning_rate": 0.0001521340273945681,
+      "loss": 0.7827,
+      "step": 1177
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.04868901148438454,
+      "learning_rate": 0.00015205924792787345,
+      "loss": 0.6776,
+      "step": 1178
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.04864593222737312,
+      "learning_rate": 0.00015198442850563584,
+      "loss": 0.6541,
+      "step": 1179
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.046926818788051605,
+      "learning_rate": 0.00015190956918527924,
+      "loss": 0.6965,
+      "step": 1180
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.044857464730739594,
+      "learning_rate": 0.0001518346700242583,
+      "loss": 0.7031,
+      "step": 1181
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.05430439114570618,
+      "learning_rate": 0.0001517597310800582,
+      "loss": 0.7288,
+      "step": 1182
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.05577121302485466,
+      "learning_rate": 0.00015168475241019474,
+      "loss": 0.8773,
+      "step": 1183
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.046056222170591354,
+      "learning_rate": 0.0001516097340722141,
+      "loss": 0.6732,
+      "step": 1184
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.05171097069978714,
+      "learning_rate": 0.00015153467612369297,
+      "loss": 0.7971,
+      "step": 1185
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.04264886677265167,
+      "learning_rate": 0.00015145957862223845,
+      "loss": 0.6446,
+      "step": 1186
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.04378258436918259,
+      "learning_rate": 0.0001513844416254879,
+      "loss": 0.6927,
+      "step": 1187
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.04480038583278656,
+      "learning_rate": 0.00015130926519110914,
+      "loss": 0.6879,
+      "step": 1188
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.04745401442050934,
+      "learning_rate": 0.00015123404937680016,
+      "loss": 0.6483,
+      "step": 1189
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.050725579261779785,
+      "learning_rate": 0.0001511587942402892,
+      "loss": 0.668,
+      "step": 1190
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.046093638986349106,
+      "learning_rate": 0.00015108349983933465,
+      "loss": 0.6211,
+      "step": 1191
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.0595737025141716,
+      "learning_rate": 0.00015100816623172514,
+      "loss": 0.7197,
+      "step": 1192
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.055872704833745956,
+      "learning_rate": 0.0001509327934752792,
+      "loss": 0.7615,
+      "step": 1193
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.05438286438584328,
+      "learning_rate": 0.00015085738162784565,
+      "loss": 0.7135,
+      "step": 1194
+    },
+    {
+      "epoch": 0.66,
+      "grad_norm": 0.048681970685720444,
+      "learning_rate": 0.0001507819307473031,
+      "loss": 0.7368,
+      "step": 1195
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.0487043671309948,
+      "learning_rate": 0.00015070644089156026,
+      "loss": 0.6612,
+      "step": 1196
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.0529305525124073,
+      "learning_rate": 0.0001506309121185556,
+      "loss": 0.6877,
+      "step": 1197
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.05354293808341026,
+      "learning_rate": 0.00015055534448625766,
+      "loss": 0.7075,
+      "step": 1198
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.04844878613948822,
+      "learning_rate": 0.00015047973805266466,
+      "loss": 0.693,
+      "step": 1199
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.05151020362973213,
+      "learning_rate": 0.0001504040928758046,
+      "loss": 0.6552,
+      "step": 1200
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.055405277758836746,
+      "learning_rate": 0.00015032840901373532,
+      "loss": 0.7242,
+      "step": 1201
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.052574001252651215,
+      "learning_rate": 0.00015025268652454421,
+      "loss": 0.7383,
+      "step": 1202
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.05063820630311966,
+      "learning_rate": 0.0001501769254663485,
+      "loss": 0.6245,
+      "step": 1203
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.04701600968837738,
+      "learning_rate": 0.00015010112589729482,
+      "loss": 0.6344,
+      "step": 1204
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.04830322787165642,
+      "learning_rate": 0.00015002528787555944,
+      "loss": 0.6309,
+      "step": 1205
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.04776900261640549,
+      "learning_rate": 0.00014994941145934814,
+      "loss": 0.7883,
+      "step": 1206
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.048602622002363205,
+      "learning_rate": 0.00014987349670689623,
+      "loss": 0.7106,
+      "step": 1207
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.05431029573082924,
+      "learning_rate": 0.00014979754367646834,
+      "loss": 0.7456,
+      "step": 1208
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.04521273076534271,
+      "learning_rate": 0.00014972155242635852,
+      "loss": 0.5661,
+      "step": 1209
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.04839641973376274,
+      "learning_rate": 0.0001496455230148902,
+      "loss": 0.6362,
+      "step": 1210
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.05457577481865883,
+      "learning_rate": 0.00014956945550041595,
+      "loss": 0.792,
+      "step": 1211
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.051022760570049286,
+      "learning_rate": 0.0001494933499413178,
+      "loss": 0.7257,
+      "step": 1212
+    },
+    {
+      "epoch": 0.67,
+      "grad_norm": 0.05146949738264084,
+      "learning_rate": 0.00014941720639600686,
+      "loss": 0.7664,
+      "step": 1213
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.04937746375799179,
+      "learning_rate": 0.00014934102492292336,
+      "loss": 0.7638,
+      "step": 1214
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.05288033187389374,
+      "learning_rate": 0.0001492648055805367,
+      "loss": 0.7008,
+      "step": 1215
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.04645581543445587,
+      "learning_rate": 0.00014918854842734533,
+      "loss": 0.659,
+      "step": 1216
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.049302008002996445,
+      "learning_rate": 0.0001491122535218767,
+      "loss": 0.6522,
+      "step": 1217
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.047872643917798996,
+      "learning_rate": 0.00014903592092268728,
+      "loss": 0.7293,
+      "step": 1218
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.046734388917684555,
+      "learning_rate": 0.00014895955068836242,
+      "loss": 0.5885,
+      "step": 1219
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.05098286643624306,
+      "learning_rate": 0.0001488831428775164,
+      "loss": 0.6703,
+      "step": 1220
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.05204523354768753,
+      "learning_rate": 0.00014880669754879227,
+      "loss": 0.7022,
+      "step": 1221
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.049951422959566116,
+      "learning_rate": 0.00014873021476086203,
+      "loss": 0.6648,
+      "step": 1222
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.04843631759285927,
+      "learning_rate": 0.00014865369457242621,
+      "loss": 0.6847,
+      "step": 1223
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.05291084572672844,
+      "learning_rate": 0.00014857713704221422,
+      "loss": 0.8186,
+      "step": 1224
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.05095936357975006,
+      "learning_rate": 0.00014850054222898402,
+      "loss": 0.705,
+      "step": 1225
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.05311046168208122,
+      "learning_rate": 0.00014842391019152226,
+      "loss": 0.6515,
+      "step": 1226
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.05315253511071205,
+      "learning_rate": 0.00014834724098864417,
+      "loss": 0.6898,
+      "step": 1227
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.048906102776527405,
+      "learning_rate": 0.0001482705346791934,
+      "loss": 0.7955,
+      "step": 1228
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.0493389368057251,
+      "learning_rate": 0.00014819379132204218,
+      "loss": 0.7394,
+      "step": 1229
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.04702428728342056,
+      "learning_rate": 0.00014811701097609112,
+      "loss": 0.5547,
+      "step": 1230
+    },
+    {
+      "epoch": 0.68,
+      "grad_norm": 0.04860011860728264,
+      "learning_rate": 0.00014804019370026926,
+      "loss": 0.7455,
+      "step": 1231
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.05252429470419884,
+      "learning_rate": 0.00014796333955353395,
+      "loss": 0.7699,
+      "step": 1232
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.04439985752105713,
+      "learning_rate": 0.0001478864485948709,
+      "loss": 0.6318,
+      "step": 1233
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.051365215331315994,
+      "learning_rate": 0.00014780952088329396,
+      "loss": 0.748,
+      "step": 1234
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.050039131194353104,
+      "learning_rate": 0.00014773255647784525,
+      "loss": 0.6782,
+      "step": 1235
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.04989631101489067,
+      "learning_rate": 0.0001476555554375951,
+      "loss": 0.6101,
+      "step": 1236
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.05142849683761597,
+      "learning_rate": 0.0001475785178216419,
+      "loss": 0.6938,
+      "step": 1237
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.04728643596172333,
+      "learning_rate": 0.00014750144368911209,
+      "loss": 0.6917,
+      "step": 1238
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.0472208671271801,
+      "learning_rate": 0.0001474243330991602,
+      "loss": 0.6225,
+      "step": 1239
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.05022706091403961,
+      "learning_rate": 0.00014734718611096877,
+      "loss": 0.6543,
+      "step": 1240
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.05018246918916702,
+      "learning_rate": 0.00014727000278374808,
+      "loss": 0.7208,
+      "step": 1241
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.04962945356965065,
+      "learning_rate": 0.00014719278317673655,
+      "loss": 0.6386,
+      "step": 1242
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.04697633534669876,
+      "learning_rate": 0.0001471155273492003,
+      "loss": 0.6764,
+      "step": 1243
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.05113474652171135,
+      "learning_rate": 0.00014703823536043323,
+      "loss": 0.7663,
+      "step": 1244
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.05091071128845215,
+      "learning_rate": 0.00014696090726975714,
+      "loss": 0.7107,
+      "step": 1245
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.05539362505078316,
+      "learning_rate": 0.0001468835431365214,
+      "loss": 0.6792,
+      "step": 1246
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.05495639145374298,
+      "learning_rate": 0.0001468061430201031,
+      "loss": 0.7616,
+      "step": 1247
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.05079134553670883,
+      "learning_rate": 0.00014672870697990685,
+      "loss": 0.6765,
+      "step": 1248
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 0.048591259866952896,
+      "learning_rate": 0.00014665123507536505,
+      "loss": 0.6775,
+      "step": 1249
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.04820895567536354,
+      "learning_rate": 0.00014657372736593739,
+      "loss": 0.6892,
+      "step": 1250
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.0469762422144413,
+      "learning_rate": 0.00014649618391111117,
+      "loss": 0.6423,
+      "step": 1251
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.048589132726192474,
+      "learning_rate": 0.00014641860477040112,
+      "loss": 0.6854,
+      "step": 1252
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.05065524950623512,
+      "learning_rate": 0.0001463409900033493,
+      "loss": 0.6663,
+      "step": 1253
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.049449969083070755,
+      "learning_rate": 0.0001462633396695252,
+      "loss": 0.6662,
+      "step": 1254
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.052549589425325394,
+      "learning_rate": 0.00014618565382852548,
+      "loss": 0.7664,
+      "step": 1255
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.05117465555667877,
+      "learning_rate": 0.00014610793253997423,
+      "loss": 0.7495,
+      "step": 1256
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.054655950516462326,
+      "learning_rate": 0.0001460301758635225,
+      "loss": 0.6196,
+      "step": 1257
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.049078892916440964,
+      "learning_rate": 0.0001459523838588488,
+      "loss": 0.5712,
+      "step": 1258
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.05013487488031387,
+      "learning_rate": 0.00014587455658565847,
+      "loss": 0.6546,
+      "step": 1259
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.05131218209862709,
+      "learning_rate": 0.00014579669410368413,
+      "loss": 0.6762,
+      "step": 1260
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.05478007718920708,
+      "learning_rate": 0.00014571879647268528,
+      "loss": 0.6899,
+      "step": 1261
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.04590092971920967,
+      "learning_rate": 0.00014564086375244855,
+      "loss": 0.6682,
+      "step": 1262
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.055836863815784454,
+      "learning_rate": 0.00014556289600278733,
+      "loss": 0.8131,
+      "step": 1263
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.05465158447623253,
+      "learning_rate": 0.00014548489328354195,
+      "loss": 0.7749,
+      "step": 1264
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.04707031324505806,
+      "learning_rate": 0.00014540685565457968,
+      "loss": 0.5979,
+      "step": 1265
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.048351485282182693,
+      "learning_rate": 0.00014532878317579444,
+      "loss": 0.7093,
+      "step": 1266
+    },
+    {
+      "epoch": 0.7,
+      "grad_norm": 0.05322815105319023,
+      "learning_rate": 0.000145250675907107,
+      "loss": 0.7281,
+      "step": 1267
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.04762796312570572,
+      "learning_rate": 0.0001451725339084648,
+      "loss": 0.6659,
+      "step": 1268
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.047569163143634796,
+      "learning_rate": 0.0001450943572398419,
+      "loss": 0.6894,
+      "step": 1269
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.04456879943609238,
+      "learning_rate": 0.00014501614596123898,
+      "loss": 0.6958,
+      "step": 1270
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.050228893756866455,
+      "learning_rate": 0.00014493790013268338,
+      "loss": 0.7011,
+      "step": 1271
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.056763775646686554,
+      "learning_rate": 0.0001448596198142288,
+      "loss": 0.8265,
+      "step": 1272
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.04859040677547455,
+      "learning_rate": 0.00014478130506595556,
+      "loss": 0.7295,
+      "step": 1273
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.054411161690950394,
+      "learning_rate": 0.0001447029559479703,
+      "loss": 0.7045,
+      "step": 1274
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.047824714332818985,
+      "learning_rate": 0.00014462457252040607,
+      "loss": 0.7665,
+      "step": 1275
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.04883917048573494,
+      "learning_rate": 0.00014454615484342222,
+      "loss": 0.7283,
+      "step": 1276
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.04813641682267189,
+      "learning_rate": 0.00014446770297720448,
+      "loss": 0.688,
+      "step": 1277
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.049928583204746246,
+      "learning_rate": 0.00014438921698196476,
+      "loss": 0.7347,
+      "step": 1278
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.05147488787770271,
+      "learning_rate": 0.00014431069691794114,
+      "loss": 0.7492,
+      "step": 1279
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.04984632879495621,
+      "learning_rate": 0.0001442321428453979,
+      "loss": 0.744,
+      "step": 1280
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.05544523149728775,
+      "learning_rate": 0.00014415355482462534,
+      "loss": 0.7598,
+      "step": 1281
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.05009017884731293,
+      "learning_rate": 0.00014407493291593995,
+      "loss": 0.6616,
+      "step": 1282
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.050635501742362976,
+      "learning_rate": 0.00014399627717968412,
+      "loss": 0.6241,
+      "step": 1283
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.054587721824645996,
+      "learning_rate": 0.0001439175876762262,
+      "loss": 0.6985,
+      "step": 1284
+    },
+    {
+      "epoch": 0.71,
+      "grad_norm": 0.05211169272661209,
+      "learning_rate": 0.00014383886446596057,
+      "loss": 0.6875,
+      "step": 1285
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.05252880975604057,
+      "learning_rate": 0.00014376010760930728,
+      "loss": 0.6947,
+      "step": 1286
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.04912309721112251,
+      "learning_rate": 0.00014368131716671246,
+      "loss": 0.6376,
+      "step": 1287
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.05693266540765762,
+      "learning_rate": 0.00014360249319864775,
+      "loss": 0.7267,
+      "step": 1288
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.05367673560976982,
+      "learning_rate": 0.00014352363576561073,
+      "loss": 0.714,
+      "step": 1289
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.04969675466418266,
+      "learning_rate": 0.00014344474492812461,
+      "loss": 0.6023,
+      "step": 1290
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.04907987639307976,
+      "learning_rate": 0.00014336582074673813,
+      "loss": 0.7009,
+      "step": 1291
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.04774461314082146,
+      "learning_rate": 0.00014328686328202582,
+      "loss": 0.6393,
+      "step": 1292
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.047509051859378815,
+      "learning_rate": 0.00014320787259458753,
+      "loss": 0.6444,
+      "step": 1293
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.057917118072509766,
+      "learning_rate": 0.00014312884874504876,
+      "loss": 0.8577,
+      "step": 1294
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.04852381348609924,
+      "learning_rate": 0.0001430497917940604,
+      "loss": 0.6916,
+      "step": 1295
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.04885304719209671,
+      "learning_rate": 0.00014297070180229883,
+      "loss": 0.7174,
+      "step": 1296
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.052427150309085846,
+      "learning_rate": 0.00014289157883046568,
+      "loss": 0.8375,
+      "step": 1297
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.04951222613453865,
+      "learning_rate": 0.0001428124229392879,
+      "loss": 0.7188,
+      "step": 1298
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.049055252224206924,
+      "learning_rate": 0.0001427332341895178,
+      "loss": 0.6977,
+      "step": 1299
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.047418877482414246,
+      "learning_rate": 0.00014265401264193286,
+      "loss": 0.5577,
+      "step": 1300
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.051448144018650055,
+      "learning_rate": 0.00014257475835733572,
+      "loss": 0.6462,
+      "step": 1301
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.053073249757289886,
+      "learning_rate": 0.00014249547139655409,
+      "loss": 0.6129,
+      "step": 1302
+    },
+    {
+      "epoch": 0.72,
+      "grad_norm": 0.05240590125322342,
+      "learning_rate": 0.00014241615182044088,
+      "loss": 0.7354,
+      "step": 1303
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.04916330799460411,
+      "learning_rate": 0.00014233679968987393,
+      "loss": 0.6193,
+      "step": 1304
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.046741846948862076,
+      "learning_rate": 0.00014225741506575617,
+      "loss": 0.5996,
+      "step": 1305
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.050647541880607605,
+      "learning_rate": 0.00014217799800901533,
+      "loss": 0.6619,
+      "step": 1306
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.052585843950510025,
+      "learning_rate": 0.00014209854858060415,
+      "loss": 0.6119,
+      "step": 1307
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.05018146336078644,
+      "learning_rate": 0.0001420190668415002,
+      "loss": 0.6795,
+      "step": 1308
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.052429914474487305,
+      "learning_rate": 0.00014193955285270575,
+      "loss": 0.6865,
+      "step": 1309
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.051422689110040665,
+      "learning_rate": 0.00014186000667524795,
+      "loss": 0.7339,
+      "step": 1310
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.05736821889877319,
+      "learning_rate": 0.00014178042837017856,
+      "loss": 0.7109,
+      "step": 1311
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.04986963048577309,
+      "learning_rate": 0.0001417008179985741,
+      "loss": 0.6964,
+      "step": 1312
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.05136452242732048,
+      "learning_rate": 0.00014162117562153558,
+      "loss": 0.5924,
+      "step": 1313
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.05401051789522171,
+      "learning_rate": 0.00014154150130018866,
+      "loss": 0.756,
+      "step": 1314
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.05107508972287178,
+      "learning_rate": 0.00014146179509568344,
+      "loss": 0.6913,
+      "step": 1315
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.048386212438344955,
+      "learning_rate": 0.0001413820570691946,
+      "loss": 0.6138,
+      "step": 1316
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.056987110525369644,
+      "learning_rate": 0.00014130228728192118,
+      "loss": 0.6351,
+      "step": 1317
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.04835304245352745,
+      "learning_rate": 0.00014122248579508657,
+      "loss": 0.6808,
+      "step": 1318
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.0571272112429142,
+      "learning_rate": 0.00014114265266993846,
+      "loss": 0.6881,
+      "step": 1319
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.058113086968660355,
+      "learning_rate": 0.00014106278796774903,
+      "loss": 0.7849,
+      "step": 1320
+    },
+    {
+      "epoch": 0.73,
+      "grad_norm": 0.05638326704502106,
+      "learning_rate": 0.00014098289174981443,
+      "loss": 0.6941,
+      "step": 1321
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.05447527393698692,
+      "learning_rate": 0.00014090296407745517,
+      "loss": 0.7279,
+      "step": 1322
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.05510277301073074,
+      "learning_rate": 0.0001408230050120158,
+      "loss": 0.7933,
+      "step": 1323
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.059467945247888565,
+      "learning_rate": 0.00014074301461486504,
+      "loss": 0.7845,
+      "step": 1324
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.060325901955366135,
+      "learning_rate": 0.00014066299294739567,
+      "loss": 0.7254,
+      "step": 1325
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.05181081220507622,
+      "learning_rate": 0.00014058294007102432,
+      "loss": 0.6383,
+      "step": 1326
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.050908952951431274,
+      "learning_rate": 0.00014050285604719177,
+      "loss": 0.7441,
+      "step": 1327
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.049533553421497345,
+      "learning_rate": 0.0001404227409373626,
+      "loss": 0.6214,
+      "step": 1328
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.05213542655110359,
+      "learning_rate": 0.00014034259480302523,
+      "loss": 0.6518,
+      "step": 1329
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.05297151952981949,
+      "learning_rate": 0.00014026241770569197,
+      "loss": 0.7337,
+      "step": 1330
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.04704327508807182,
+      "learning_rate": 0.00014018220970689883,
+      "loss": 0.619,
+      "step": 1331
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.054115235805511475,
+      "learning_rate": 0.00014010197086820552,
+      "loss": 0.7122,
+      "step": 1332
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.050988852977752686,
+      "learning_rate": 0.00014002170125119553,
+      "loss": 0.7545,
+      "step": 1333
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.049325957894325256,
+      "learning_rate": 0.00013994140091747587,
+      "loss": 0.6834,
+      "step": 1334
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.06573436409235,
+      "learning_rate": 0.00013986106992867713,
+      "loss": 0.7652,
+      "step": 1335
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.050615664571523666,
+      "learning_rate": 0.00013978070834645348,
+      "loss": 0.648,
+      "step": 1336
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.050100456923246384,
+      "learning_rate": 0.0001397003162324825,
+      "loss": 0.6443,
+      "step": 1337
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.04777170345187187,
+      "learning_rate": 0.00013961989364846532,
+      "loss": 0.6743,
+      "step": 1338
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 0.05937036871910095,
+      "learning_rate": 0.00013953944065612633,
+      "loss": 0.7196,
+      "step": 1339
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.05465259775519371,
+      "learning_rate": 0.0001394589573172133,
+      "loss": 0.7038,
+      "step": 1340
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.04839110001921654,
+      "learning_rate": 0.00013937844369349734,
+      "loss": 0.7035,
+      "step": 1341
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.05537423491477966,
+      "learning_rate": 0.00013929789984677278,
+      "loss": 0.7386,
+      "step": 1342
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.051877472549676895,
+      "learning_rate": 0.00013921732583885705,
+      "loss": 0.6811,
+      "step": 1343
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.052817557007074356,
+      "learning_rate": 0.00013913672173159088,
+      "loss": 0.6852,
+      "step": 1344
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.0512346476316452,
+      "learning_rate": 0.000139056087586838,
+      "loss": 0.6145,
+      "step": 1345
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.05505014955997467,
+      "learning_rate": 0.00013897542346648524,
+      "loss": 0.8609,
+      "step": 1346
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.0500144399702549,
+      "learning_rate": 0.00013889472943244243,
+      "loss": 0.6782,
+      "step": 1347
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.0514250211417675,
+      "learning_rate": 0.0001388140055466423,
+      "loss": 0.7126,
+      "step": 1348
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.049233101308345795,
+      "learning_rate": 0.00013873325187104056,
+      "loss": 0.6516,
+      "step": 1349
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.04899629205465317,
+      "learning_rate": 0.00013865246846761581,
+      "loss": 0.617,
+      "step": 1350
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.05683201178908348,
+      "learning_rate": 0.0001385716553983694,
+      "loss": 0.7033,
+      "step": 1351
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.05018038675189018,
+      "learning_rate": 0.00013849081272532544,
+      "loss": 0.6809,
+      "step": 1352
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.05704934149980545,
+      "learning_rate": 0.00013840994051053085,
+      "loss": 0.8067,
+      "step": 1353
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.050851497799158096,
+      "learning_rate": 0.00013832903881605508,
+      "loss": 0.6693,
+      "step": 1354
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.051365941762924194,
+      "learning_rate": 0.00013824810770399036,
+      "loss": 0.6739,
+      "step": 1355
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.058710698038339615,
+      "learning_rate": 0.0001381671472364514,
+      "loss": 0.6647,
+      "step": 1356
+    },
+    {
+      "epoch": 0.75,
+      "grad_norm": 0.05513182282447815,
+      "learning_rate": 0.00013808615747557549,
+      "loss": 0.6949,
+      "step": 1357
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.05616655945777893,
+      "learning_rate": 0.0001380051384835223,
+      "loss": 0.7774,
+      "step": 1358
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.05610072240233421,
+      "learning_rate": 0.0001379240903224741,
+      "loss": 0.7263,
+      "step": 1359
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.05135612562298775,
+      "learning_rate": 0.00013784301305463549,
+      "loss": 0.6889,
+      "step": 1360
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.05268790200352669,
+      "learning_rate": 0.00013776190674223327,
+      "loss": 0.7119,
+      "step": 1361
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.04713091999292374,
+      "learning_rate": 0.00013768077144751673,
+      "loss": 0.6209,
+      "step": 1362
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.049119122326374054,
+      "learning_rate": 0.00013759960723275732,
+      "loss": 0.673,
+      "step": 1363
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.05238724872469902,
+      "learning_rate": 0.00013751841416024865,
+      "loss": 0.7073,
+      "step": 1364
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.04365584999322891,
+      "learning_rate": 0.0001374371922923065,
+      "loss": 0.5371,
+      "step": 1365
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.05708995461463928,
+      "learning_rate": 0.0001373559416912688,
+      "loss": 0.6894,
+      "step": 1366
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.04669976979494095,
+      "learning_rate": 0.00013727466241949545,
+      "loss": 0.7083,
+      "step": 1367
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.05093741416931152,
+      "learning_rate": 0.00013719335453936846,
+      "loss": 0.6603,
+      "step": 1368
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.04979169741272926,
+      "learning_rate": 0.0001371120181132917,
+      "loss": 0.6269,
+      "step": 1369
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.05657560005784035,
+      "learning_rate": 0.000137030653203691,
+      "loss": 0.8572,
+      "step": 1370
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.05758770555257797,
+      "learning_rate": 0.00013694925987301404,
+      "loss": 0.7561,
+      "step": 1371
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.048717837780714035,
+      "learning_rate": 0.00013686783818373028,
+      "loss": 0.7044,
+      "step": 1372
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.049861840903759,
+      "learning_rate": 0.00013678638819833103,
+      "loss": 0.6199,
+      "step": 1373
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.050144508481025696,
+      "learning_rate": 0.00013670490997932922,
+      "loss": 0.6802,
+      "step": 1374
+    },
+    {
+      "epoch": 0.76,
+      "grad_norm": 0.05109990015625954,
+      "learning_rate": 0.0001366234035892595,
+      "loss": 0.6872,
+      "step": 1375
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.05122920870780945,
+      "learning_rate": 0.00013654186909067817,
+      "loss": 0.6246,
+      "step": 1376
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.05392398312687874,
+      "learning_rate": 0.00013646030654616302,
+      "loss": 0.6679,
+      "step": 1377
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.051943257451057434,
+      "learning_rate": 0.0001363787160183134,
+      "loss": 0.6608,
+      "step": 1378
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.05808079242706299,
+      "learning_rate": 0.00013629709756975023,
+      "loss": 0.6256,
+      "step": 1379
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.05690327659249306,
+      "learning_rate": 0.0001362154512631157,
+      "loss": 0.6313,
+      "step": 1380
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.0537363737821579,
+      "learning_rate": 0.0001361337771610735,
+      "loss": 0.6827,
+      "step": 1381
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.05171459913253784,
+      "learning_rate": 0.00013605207532630864,
+      "loss": 0.6683,
+      "step": 1382
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.051463544368743896,
+      "learning_rate": 0.00013597034582152733,
+      "loss": 0.6894,
+      "step": 1383
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.049201156944036484,
+      "learning_rate": 0.0001358885887094571,
+      "loss": 0.6486,
+      "step": 1384
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.052608225494623184,
+      "learning_rate": 0.00013580680405284664,
+      "loss": 0.7302,
+      "step": 1385
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.05633961036801338,
+      "learning_rate": 0.00013572499191446578,
+      "loss": 0.7057,
+      "step": 1386
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.05480135232210159,
+      "learning_rate": 0.00013564315235710546,
+      "loss": 0.6837,
+      "step": 1387
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.04823005944490433,
+      "learning_rate": 0.00013556128544357763,
+      "loss": 0.6007,
+      "step": 1388
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.05478844791650772,
+      "learning_rate": 0.0001354793912367153,
+      "loss": 0.6724,
+      "step": 1389
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.04951076954603195,
+      "learning_rate": 0.00013539746979937233,
+      "loss": 0.6648,
+      "step": 1390
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.08351165801286697,
+      "learning_rate": 0.00013531552119442356,
+      "loss": 0.7277,
+      "step": 1391
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.050557348877191544,
+      "learning_rate": 0.00013523354548476468,
+      "loss": 0.7675,
+      "step": 1392
+    },
+    {
+      "epoch": 0.77,
+      "grad_norm": 0.04991862177848816,
+      "learning_rate": 0.0001351515427333121,
+      "loss": 0.666,
+      "step": 1393
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.04630931839346886,
+      "learning_rate": 0.0001350695130030031,
+      "loss": 0.6061,
+      "step": 1394
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.05627691000699997,
+      "learning_rate": 0.00013498745635679557,
+      "loss": 0.7015,
+      "step": 1395
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.05329517647624016,
+      "learning_rate": 0.00013490537285766808,
+      "loss": 0.6464,
+      "step": 1396
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.059979844838380814,
+      "learning_rate": 0.00013482326256861988,
+      "loss": 0.6884,
+      "step": 1397
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.04928671196103096,
+      "learning_rate": 0.0001347411255526707,
+      "loss": 0.6286,
+      "step": 1398
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.05504937097430229,
+      "learning_rate": 0.00013465896187286083,
+      "loss": 0.7726,
+      "step": 1399
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.04796033352613449,
+      "learning_rate": 0.00013457677159225097,
+      "loss": 0.5906,
+      "step": 1400
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.05083645507693291,
+      "learning_rate": 0.0001344945547739223,
+      "loss": 0.6439,
+      "step": 1401
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.05347789451479912,
+      "learning_rate": 0.0001344123114809763,
+      "loss": 0.6212,
+      "step": 1402
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.04871118441224098,
+      "learning_rate": 0.00013433004177653486,
+      "loss": 0.6443,
+      "step": 1403
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.05615173652768135,
+      "learning_rate": 0.00013424774572374005,
+      "loss": 0.7706,
+      "step": 1404
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.04990394786000252,
+      "learning_rate": 0.00013416542338575424,
+      "loss": 0.5935,
+      "step": 1405
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.05090782046318054,
+      "learning_rate": 0.0001340830748257599,
+      "loss": 0.6519,
+      "step": 1406
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.05769623816013336,
+      "learning_rate": 0.00013400070010695966,
+      "loss": 0.6945,
+      "step": 1407
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.05073278769850731,
+      "learning_rate": 0.00013391829929257625,
+      "loss": 0.6583,
+      "step": 1408
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.05385829880833626,
+      "learning_rate": 0.0001338358724458524,
+      "loss": 0.6754,
+      "step": 1409
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.06437458097934723,
+      "learning_rate": 0.0001337534196300508,
+      "loss": 0.7668,
+      "step": 1410
+    },
+    {
+      "epoch": 0.78,
+      "grad_norm": 0.0532778799533844,
+      "learning_rate": 0.0001336709409084542,
+      "loss": 0.6855,
+      "step": 1411
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.06863218545913696,
+      "learning_rate": 0.00013358843634436496,
+      "loss": 0.7613,
+      "step": 1412
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.05395227298140526,
+      "learning_rate": 0.00013350590600110556,
+      "loss": 0.6725,
+      "step": 1413
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.0562172457575798,
+      "learning_rate": 0.00013342334994201815,
+      "loss": 0.7959,
+      "step": 1414
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.052872899919748306,
+      "learning_rate": 0.00013334076823046456,
+      "loss": 0.666,
+      "step": 1415
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.05251738429069519,
+      "learning_rate": 0.0001332581609298264,
+      "loss": 0.686,
+      "step": 1416
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.05102743208408356,
+      "learning_rate": 0.00013317552810350488,
+      "loss": 0.5487,
+      "step": 1417
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.056528471410274506,
+      "learning_rate": 0.00013309286981492085,
+      "loss": 0.6795,
+      "step": 1418
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.05416551232337952,
+      "learning_rate": 0.00013301018612751458,
+      "loss": 0.6314,
+      "step": 1419
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.052541207522153854,
+      "learning_rate": 0.00013292747710474593,
+      "loss": 0.6579,
+      "step": 1420
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.05116291716694832,
+      "learning_rate": 0.0001328447428100942,
+      "loss": 0.6836,
+      "step": 1421
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.05055323615670204,
+      "learning_rate": 0.0001327619833070581,
+      "loss": 0.6109,
+      "step": 1422
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.06023389473557472,
+      "learning_rate": 0.00013267919865915564,
+      "loss": 0.7022,
+      "step": 1423
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.050478942692279816,
+      "learning_rate": 0.00013259638892992412,
+      "loss": 0.5096,
+      "step": 1424
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.050790175795555115,
+      "learning_rate": 0.00013251355418292019,
+      "loss": 0.7475,
+      "step": 1425
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.05030534043908119,
+      "learning_rate": 0.00013243069448171953,
+      "loss": 0.6937,
+      "step": 1426
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.04698512703180313,
+      "learning_rate": 0.00013234780988991712,
+      "loss": 0.5889,
+      "step": 1427
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.05166896432638168,
+      "learning_rate": 0.00013226490047112702,
+      "loss": 0.7394,
+      "step": 1428
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 0.05660312995314598,
+      "learning_rate": 0.00013218196628898233,
+      "loss": 0.6686,
+      "step": 1429
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.04474378004670143,
+      "learning_rate": 0.00013209900740713507,
+      "loss": 0.6026,
+      "step": 1430
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.06732626259326935,
+      "learning_rate": 0.00013201602388925637,
+      "loss": 0.8321,
+      "step": 1431
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.05723815783858299,
+      "learning_rate": 0.00013193301579903616,
+      "loss": 0.6508,
+      "step": 1432
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.05704987049102783,
+      "learning_rate": 0.00013184998320018326,
+      "loss": 0.7649,
+      "step": 1433
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.05435892567038536,
+      "learning_rate": 0.00013176692615642533,
+      "loss": 0.7658,
+      "step": 1434
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.04823015630245209,
+      "learning_rate": 0.0001316838447315087,
+      "loss": 0.6798,
+      "step": 1435
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.05567566677927971,
+      "learning_rate": 0.00013160073898919853,
+      "loss": 0.6289,
+      "step": 1436
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.059023939073085785,
+      "learning_rate": 0.0001315176089932786,
+      "loss": 0.7173,
+      "step": 1437
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.052320726215839386,
+      "learning_rate": 0.00013143445480755123,
+      "loss": 0.6257,
+      "step": 1438
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.05305778235197067,
+      "learning_rate": 0.00013135127649583744,
+      "loss": 0.6709,
+      "step": 1439
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.054861389100551605,
+      "learning_rate": 0.00013126807412197665,
+      "loss": 0.6239,
+      "step": 1440
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.04839559271931648,
+      "learning_rate": 0.0001311848477498268,
+      "loss": 0.584,
+      "step": 1441
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.04709853231906891,
+      "learning_rate": 0.00013110159744326427,
+      "loss": 0.5997,
+      "step": 1442
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.05461667478084564,
+      "learning_rate": 0.00013101832326618376,
+      "loss": 0.6488,
+      "step": 1443
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.05399639531970024,
+      "learning_rate": 0.0001309350252824983,
+      "loss": 0.6913,
+      "step": 1444
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.05999092012643814,
+      "learning_rate": 0.00013085170355613926,
+      "loss": 0.6265,
+      "step": 1445
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.05753492936491966,
+      "learning_rate": 0.0001307683581510561,
+      "loss": 0.7015,
+      "step": 1446
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 0.054452694952487946,
+      "learning_rate": 0.00013068498913121657,
+      "loss": 0.6592,
+      "step": 1447
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05475104600191116,
+      "learning_rate": 0.00013060159656060654,
+      "loss": 0.6994,
+      "step": 1448
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05358335003256798,
+      "learning_rate": 0.00013051818050322986,
+      "loss": 0.6967,
+      "step": 1449
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05165765434503555,
+      "learning_rate": 0.0001304347410231085,
+      "loss": 0.681,
+      "step": 1450
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05551374331116676,
+      "learning_rate": 0.0001303512781842824,
+      "loss": 0.565,
+      "step": 1451
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05121142789721489,
+      "learning_rate": 0.00013026779205080932,
+      "loss": 0.6739,
+      "step": 1452
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05909129977226257,
+      "learning_rate": 0.00013018428268676504,
+      "loss": 0.8198,
+      "step": 1453
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.048282720148563385,
+      "learning_rate": 0.0001301007501562431,
+      "loss": 0.636,
+      "step": 1454
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05285617336630821,
+      "learning_rate": 0.00013001719452335485,
+      "loss": 0.623,
+      "step": 1455
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05915940925478935,
+      "learning_rate": 0.00012993361585222928,
+      "loss": 0.7059,
+      "step": 1456
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.0531185120344162,
+      "learning_rate": 0.00012985001420701318,
+      "loss": 0.6381,
+      "step": 1457
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05316080525517464,
+      "learning_rate": 0.00012976638965187095,
+      "loss": 0.6969,
+      "step": 1458
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.052480001002550125,
+      "learning_rate": 0.00012968274225098452,
+      "loss": 0.6696,
+      "step": 1459
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05531112849712372,
+      "learning_rate": 0.00012959907206855343,
+      "loss": 0.7064,
+      "step": 1460
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.055633753538131714,
+      "learning_rate": 0.00012951537916879458,
+      "loss": 0.6559,
+      "step": 1461
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.052477676421403885,
+      "learning_rate": 0.00012943166361594242,
+      "loss": 0.6901,
+      "step": 1462
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05346304923295975,
+      "learning_rate": 0.00012934792547424873,
+      "loss": 0.6428,
+      "step": 1463
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05244500935077667,
+      "learning_rate": 0.0001292641648079827,
+      "loss": 0.6615,
+      "step": 1464
+    },
+    {
+      "epoch": 0.81,
+      "grad_norm": 0.05374164506793022,
+      "learning_rate": 0.00012918038168143066,
+      "loss": 0.711,
+      "step": 1465
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05638580396771431,
+      "learning_rate": 0.00012909657615889638,
+      "loss": 0.731,
+      "step": 1466
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05788502097129822,
+      "learning_rate": 0.00012901274830470064,
+      "loss": 0.7525,
+      "step": 1467
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05340276658535004,
+      "learning_rate": 0.0001289288981831815,
+      "loss": 0.6505,
+      "step": 1468
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05570358410477638,
+      "learning_rate": 0.00012884502585869395,
+      "loss": 0.6989,
+      "step": 1469
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05580015480518341,
+      "learning_rate": 0.00012876113139561018,
+      "loss": 0.7683,
+      "step": 1470
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05318528413772583,
+      "learning_rate": 0.0001286772148583193,
+      "loss": 0.6406,
+      "step": 1471
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05146130174398422,
+      "learning_rate": 0.0001285932763112273,
+      "loss": 0.65,
+      "step": 1472
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05019865557551384,
+      "learning_rate": 0.00012850931581875723,
+      "loss": 0.6883,
+      "step": 1473
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.06139828637242317,
+      "learning_rate": 0.00012842533344534877,
+      "loss": 0.6625,
+      "step": 1474
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.055851712822914124,
+      "learning_rate": 0.0001283413292554586,
+      "loss": 0.6335,
+      "step": 1475
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05272622033953667,
+      "learning_rate": 0.00012825730331355995,
+      "loss": 0.6381,
+      "step": 1476
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.053820542991161346,
+      "learning_rate": 0.00012817325568414297,
+      "loss": 0.6784,
+      "step": 1477
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05233492702245712,
+      "learning_rate": 0.00012808918643171424,
+      "loss": 0.6492,
+      "step": 1478
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05471348017454147,
+      "learning_rate": 0.00012800509562079705,
+      "loss": 0.6332,
+      "step": 1479
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.051646675914525986,
+      "learning_rate": 0.0001279209833159312,
+      "loss": 0.6563,
+      "step": 1480
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05445919558405876,
+      "learning_rate": 0.00012783684958167304,
+      "loss": 0.7054,
+      "step": 1481
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.053671520203351974,
+      "learning_rate": 0.00012775269448259526,
+      "loss": 0.6493,
+      "step": 1482
+    },
+    {
+      "epoch": 0.82,
+      "grad_norm": 0.05467415601015091,
+      "learning_rate": 0.00012766851808328707,
+      "loss": 0.6223,
+      "step": 1483
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.05039871484041214,
+      "learning_rate": 0.00012758432044835392,
+      "loss": 0.5863,
+      "step": 1484
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.048955757170915604,
+      "learning_rate": 0.00012750010164241764,
+      "loss": 0.5957,
+      "step": 1485
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.06016591191291809,
+      "learning_rate": 0.00012741586173011625,
+      "loss": 0.7174,
+      "step": 1486
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.05054686218500137,
+      "learning_rate": 0.00012733160077610403,
+      "loss": 0.5833,
+      "step": 1487
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.06371674686670303,
+      "learning_rate": 0.00012724731884505134,
+      "loss": 0.6756,
+      "step": 1488
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.04819337651133537,
+      "learning_rate": 0.0001271630160016447,
+      "loss": 0.6621,
+      "step": 1489
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.053725238889455795,
+      "learning_rate": 0.00012707869231058666,
+      "loss": 0.6667,
+      "step": 1490
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.04935624450445175,
+      "learning_rate": 0.00012699434783659577,
+      "loss": 0.6873,
+      "step": 1491
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.05342969298362732,
+      "learning_rate": 0.00012690998264440652,
+      "loss": 0.7205,
+      "step": 1492
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.04896765202283859,
+      "learning_rate": 0.0001268255967987693,
+      "loss": 0.5972,
+      "step": 1493
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.05099526047706604,
+      "learning_rate": 0.00012674119036445037,
+      "loss": 0.6677,
+      "step": 1494
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.0643346831202507,
+      "learning_rate": 0.0001266567634062317,
+      "loss": 0.7039,
+      "step": 1495
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.06624721735715866,
+      "learning_rate": 0.00012657231598891126,
+      "loss": 0.7797,
+      "step": 1496
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.05854468047618866,
+      "learning_rate": 0.00012648784817730242,
+      "loss": 0.7198,
+      "step": 1497
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.05921991169452667,
+      "learning_rate": 0.00012640336003623444,
+      "loss": 0.7499,
+      "step": 1498
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.055420514196157455,
+      "learning_rate": 0.000126318851630552,
+      "loss": 0.5993,
+      "step": 1499
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.06067932769656181,
+      "learning_rate": 0.00012623432302511544,
+      "loss": 0.8006,
+      "step": 1500
+    },
+    {
+      "epoch": 0.83,
+      "grad_norm": 0.05395696684718132,
+      "learning_rate": 0.0001261497742848006,
+      "loss": 0.654,
+      "step": 1501
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.05506988614797592,
+      "learning_rate": 0.0001260652054744987,
+      "loss": 0.6884,
+      "step": 1502
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.0567973367869854,
+      "learning_rate": 0.0001259806166591165,
+      "loss": 0.685,
+      "step": 1503
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.05710934102535248,
+      "learning_rate": 0.00012589600790357592,
+      "loss": 0.6786,
+      "step": 1504
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.0530974306166172,
+      "learning_rate": 0.00012581137927281439,
+      "loss": 0.6175,
+      "step": 1505
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.05763082951307297,
+      "learning_rate": 0.0001257267308317845,
+      "loss": 0.5977,
+      "step": 1506
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.056460849940776825,
+      "learning_rate": 0.00012564206264545396,
+      "loss": 0.706,
+      "step": 1507
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.05578916147351265,
+      "learning_rate": 0.00012555737477880577,
+      "loss": 0.736,
+      "step": 1508
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.05610078200697899,
+      "learning_rate": 0.00012547266729683797,
+      "loss": 0.6614,
+      "step": 1509
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.05859774723649025,
+      "learning_rate": 0.00012538794026456366,
+      "loss": 0.7192,
+      "step": 1510
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.05389096215367317,
+      "learning_rate": 0.00012530319374701098,
+      "loss": 0.6748,
+      "step": 1511
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.05462174117565155,
+      "learning_rate": 0.000125218427809223,
+      "loss": 0.6765,
+      "step": 1512
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.047790851444005966,
+      "learning_rate": 0.00012513364251625766,
+      "loss": 0.6126,
+      "step": 1513
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.05302601680159569,
+      "learning_rate": 0.0001250488379331878,
+      "loss": 0.6352,
+      "step": 1514
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.05396215245127678,
+      "learning_rate": 0.00012496401412510102,
+      "loss": 0.6203,
+      "step": 1515
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.051495350897312164,
+      "learning_rate": 0.00012487917115709975,
+      "loss": 0.7411,
+      "step": 1516
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.05668788030743599,
+      "learning_rate": 0.00012479430909430108,
+      "loss": 0.7574,
+      "step": 1517
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.056466199457645416,
+      "learning_rate": 0.00012470942800183675,
+      "loss": 0.6395,
+      "step": 1518
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 0.061602648347616196,
+      "learning_rate": 0.0001246245279448531,
+      "loss": 0.7101,
+      "step": 1519
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.05588141828775406,
+      "learning_rate": 0.00012453960898851108,
+      "loss": 0.7231,
+      "step": 1520
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.05210031941533089,
+      "learning_rate": 0.00012445467119798605,
+      "loss": 0.7094,
+      "step": 1521
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.05762970820069313,
+      "learning_rate": 0.00012436971463846788,
+      "loss": 0.6623,
+      "step": 1522
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.05559748038649559,
+      "learning_rate": 0.0001242847393751609,
+      "loss": 0.6677,
+      "step": 1523
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.05742710828781128,
+      "learning_rate": 0.00012419974547328366,
+      "loss": 0.7276,
+      "step": 1524
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.053973183035850525,
+      "learning_rate": 0.00012411473299806918,
+      "loss": 0.7039,
+      "step": 1525
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.056114938110113144,
+      "learning_rate": 0.00012402970201476457,
+      "loss": 0.6718,
+      "step": 1526
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.06136467307806015,
+      "learning_rate": 0.0001239446525886313,
+      "loss": 0.7666,
+      "step": 1527
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.057720523327589035,
+      "learning_rate": 0.00012385958478494487,
+      "loss": 0.6303,
+      "step": 1528
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.054988570511341095,
+      "learning_rate": 0.00012377449866899493,
+      "loss": 0.6074,
+      "step": 1529
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.05281984433531761,
+      "learning_rate": 0.00012368939430608522,
+      "loss": 0.6453,
+      "step": 1530
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.06022608280181885,
+      "learning_rate": 0.00012360427176153342,
+      "loss": 0.7284,
+      "step": 1531
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.05347907170653343,
+      "learning_rate": 0.00012351913110067122,
+      "loss": 0.5784,
+      "step": 1532
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.0521782748401165,
+      "learning_rate": 0.0001234339723888442,
+      "loss": 0.6836,
+      "step": 1533
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.05518525093793869,
+      "learning_rate": 0.00012334879569141172,
+      "loss": 0.6714,
+      "step": 1534
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.0590486079454422,
+      "learning_rate": 0.00012326360107374712,
+      "loss": 0.7145,
+      "step": 1535
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.05373707413673401,
+      "learning_rate": 0.0001231783886012373,
+      "loss": 0.6897,
+      "step": 1536
+    },
+    {
+      "epoch": 0.85,
+      "grad_norm": 0.05092402547597885,
+      "learning_rate": 0.00012309315833928302,
+      "loss": 0.6245,
+      "step": 1537
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.05266068875789642,
+      "learning_rate": 0.00012300791035329853,
+      "loss": 0.5606,
+      "step": 1538
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.05256986245512962,
+      "learning_rate": 0.00012292264470871182,
+      "loss": 0.6055,
+      "step": 1539
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.05765622481703758,
+      "learning_rate": 0.0001228373614709644,
+      "loss": 0.7045,
+      "step": 1540
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.05402863398194313,
+      "learning_rate": 0.00012275206070551126,
+      "loss": 0.7477,
+      "step": 1541
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.049794167280197144,
+      "learning_rate": 0.00012266674247782085,
+      "loss": 0.6824,
+      "step": 1542
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.052845150232315063,
+      "learning_rate": 0.000122581406853375,
+      "loss": 0.6614,
+      "step": 1543
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.05592498928308487,
+      "learning_rate": 0.00012249605389766895,
+      "loss": 0.6355,
+      "step": 1544
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.05580741912126541,
+      "learning_rate": 0.0001224106836762112,
+      "loss": 0.557,
+      "step": 1545
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.05675602704286575,
+      "learning_rate": 0.00012232529625452352,
+      "loss": 0.6461,
+      "step": 1546
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.05478064343333244,
+      "learning_rate": 0.00012223989169814087,
+      "loss": 0.6754,
+      "step": 1547
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.05122774466872215,
+      "learning_rate": 0.00012215447007261134,
+      "loss": 0.6126,
+      "step": 1548
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.05637194588780403,
+      "learning_rate": 0.00012206903144349615,
+      "loss": 0.599,
+      "step": 1549
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.06174594536423683,
+      "learning_rate": 0.00012198357587636957,
+      "loss": 0.7692,
+      "step": 1550
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.0552188940346241,
+      "learning_rate": 0.00012189810343681889,
+      "loss": 0.5888,
+      "step": 1551
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.06255478411912918,
+      "learning_rate": 0.00012181261419044428,
+      "loss": 0.6878,
+      "step": 1552
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.049522750079631805,
+      "learning_rate": 0.00012172710820285885,
+      "loss": 0.645,
+      "step": 1553
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.05213455855846405,
+      "learning_rate": 0.00012164158553968856,
+      "loss": 0.5865,
+      "step": 1554
+    },
+    {
+      "epoch": 0.86,
+      "grad_norm": 0.05377736687660217,
+      "learning_rate": 0.00012155604626657222,
+      "loss": 0.6835,
+      "step": 1555
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.06013650447130203,
+      "learning_rate": 0.0001214704904491613,
+      "loss": 0.6399,
+      "step": 1556
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.057262714952230453,
+      "learning_rate": 0.00012138491815312001,
+      "loss": 0.68,
+      "step": 1557
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.05029933899641037,
+      "learning_rate": 0.00012129932944412518,
+      "loss": 0.6265,
+      "step": 1558
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.05597386881709099,
+      "learning_rate": 0.00012121372438786631,
+      "loss": 0.7077,
+      "step": 1559
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.05315025895833969,
+      "learning_rate": 0.00012112810305004535,
+      "loss": 0.6948,
+      "step": 1560
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.056836340576410294,
+      "learning_rate": 0.00012104246549637683,
+      "loss": 0.6079,
+      "step": 1561
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.05192887783050537,
+      "learning_rate": 0.00012095681179258765,
+      "loss": 0.6264,
+      "step": 1562
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.05339023098349571,
+      "learning_rate": 0.00012087114200441714,
+      "loss": 0.6584,
+      "step": 1563
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.054986581206321716,
+      "learning_rate": 0.00012078545619761703,
+      "loss": 0.5941,
+      "step": 1564
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.05665560066699982,
+      "learning_rate": 0.00012069975443795126,
+      "loss": 0.6785,
+      "step": 1565
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.055501628667116165,
+      "learning_rate": 0.00012061403679119603,
+      "loss": 0.7018,
+      "step": 1566
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.058468155562877655,
+      "learning_rate": 0.00012052830332313978,
+      "loss": 0.76,
+      "step": 1567
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.055983930826187134,
+      "learning_rate": 0.00012044255409958305,
+      "loss": 0.7121,
+      "step": 1568
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.056311774998903275,
+      "learning_rate": 0.00012035678918633848,
+      "loss": 0.6893,
+      "step": 1569
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.05754856392741203,
+      "learning_rate": 0.00012027100864923076,
+      "loss": 0.7448,
+      "step": 1570
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.05376417189836502,
+      "learning_rate": 0.00012018521255409656,
+      "loss": 0.667,
+      "step": 1571
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.0655338391661644,
+      "learning_rate": 0.00012009940096678452,
+      "loss": 0.7666,
+      "step": 1572
+    },
+    {
+      "epoch": 0.87,
+      "grad_norm": 0.055434565991163254,
+      "learning_rate": 0.00012001357395315511,
+      "loss": 0.5904,
+      "step": 1573
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.053381387144327164,
+      "learning_rate": 0.00011992773157908073,
+      "loss": 0.6542,
+      "step": 1574
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.05259239673614502,
+      "learning_rate": 0.00011984187391044548,
+      "loss": 0.6848,
+      "step": 1575
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.05607330799102783,
+      "learning_rate": 0.00011975600101314525,
+      "loss": 0.7138,
+      "step": 1576
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.051935825496912,
+      "learning_rate": 0.00011967011295308761,
+      "loss": 0.6423,
+      "step": 1577
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.058591440320014954,
+      "learning_rate": 0.00011958420979619176,
+      "loss": 0.6529,
+      "step": 1578
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.06154406815767288,
+      "learning_rate": 0.00011949829160838844,
+      "loss": 0.7658,
+      "step": 1579
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.05416911095380783,
+      "learning_rate": 0.00011941235845562006,
+      "loss": 0.6634,
+      "step": 1580
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.056073009967803955,
+      "learning_rate": 0.00011932641040384038,
+      "loss": 0.575,
+      "step": 1581
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.06217704340815544,
+      "learning_rate": 0.00011924044751901466,
+      "loss": 0.7539,
+      "step": 1582
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.05384403467178345,
+      "learning_rate": 0.00011915446986711953,
+      "loss": 0.566,
+      "step": 1583
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.05363078787922859,
+      "learning_rate": 0.00011906847751414291,
+      "loss": 0.6722,
+      "step": 1584
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.05821472778916359,
+      "learning_rate": 0.00011898247052608414,
+      "loss": 0.685,
+      "step": 1585
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.06892435252666473,
+      "learning_rate": 0.00011889644896895362,
+      "loss": 0.6482,
+      "step": 1586
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.06398369371891022,
+      "learning_rate": 0.00011881041290877303,
+      "loss": 0.675,
+      "step": 1587
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.05386562645435333,
+      "learning_rate": 0.00011872436241157518,
+      "loss": 0.6116,
+      "step": 1588
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.05238373950123787,
+      "learning_rate": 0.00011863829754340395,
+      "loss": 0.628,
+      "step": 1589
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.05979369580745697,
+      "learning_rate": 0.00011855221837031418,
+      "loss": 0.723,
+      "step": 1590
+    },
+    {
+      "epoch": 0.88,
+      "grad_norm": 0.06569157540798187,
+      "learning_rate": 0.00011846612495837182,
+      "loss": 0.7057,
+      "step": 1591
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.057322900742292404,
+      "learning_rate": 0.00011838001737365365,
+      "loss": 0.6529,
+      "step": 1592
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.06029359623789787,
+      "learning_rate": 0.00011829389568224734,
+      "loss": 0.7157,
+      "step": 1593
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.05852045491337776,
+      "learning_rate": 0.00011820775995025147,
+      "loss": 0.6289,
+      "step": 1594
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.05572306364774704,
+      "learning_rate": 0.00011812161024377526,
+      "loss": 0.6662,
+      "step": 1595
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.05891454592347145,
+      "learning_rate": 0.00011803544662893875,
+      "loss": 0.662,
+      "step": 1596
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.05507583171129227,
+      "learning_rate": 0.00011794926917187264,
+      "loss": 0.572,
+      "step": 1597
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.05523572862148285,
+      "learning_rate": 0.00011786307793871824,
+      "loss": 0.6808,
+      "step": 1598
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.06197218969464302,
+      "learning_rate": 0.00011777687299562743,
+      "loss": 0.6696,
+      "step": 1599
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.05204472318291664,
+      "learning_rate": 0.00011769065440876263,
+      "loss": 0.6031,
+      "step": 1600
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.05904833972454071,
+      "learning_rate": 0.0001176044222442967,
+      "loss": 0.6351,
+      "step": 1601
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.06034965440630913,
+      "learning_rate": 0.00011751817656841298,
+      "loss": 0.7582,
+      "step": 1602
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.05305292084813118,
+      "learning_rate": 0.0001174319174473051,
+      "loss": 0.6891,
+      "step": 1603
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.052683088928461075,
+      "learning_rate": 0.00011734564494717711,
+      "loss": 0.6354,
+      "step": 1604
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.059987373650074005,
+      "learning_rate": 0.0001172593591342432,
+      "loss": 0.6069,
+      "step": 1605
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.05370206758379936,
+      "learning_rate": 0.0001171730600747279,
+      "loss": 0.6376,
+      "step": 1606
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.05009806901216507,
+      "learning_rate": 0.00011708674783486583,
+      "loss": 0.537,
+      "step": 1607
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.05144254118204117,
+      "learning_rate": 0.00011700042248090176,
+      "loss": 0.6138,
+      "step": 1608
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 0.05199851468205452,
+      "learning_rate": 0.0001169140840790905,
+      "loss": 0.6005,
+      "step": 1609
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.06040224805474281,
+      "learning_rate": 0.00011682773269569693,
+      "loss": 0.6571,
+      "step": 1610
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.05754357576370239,
+      "learning_rate": 0.00011674136839699581,
+      "loss": 0.7045,
+      "step": 1611
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.05448243021965027,
+      "learning_rate": 0.00011665499124927184,
+      "loss": 0.642,
+      "step": 1612
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.0632404014468193,
+      "learning_rate": 0.00011656860131881966,
+      "loss": 0.7868,
+      "step": 1613
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.054406315088272095,
+      "learning_rate": 0.00011648219867194362,
+      "loss": 0.6224,
+      "step": 1614
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.059626247733831406,
+      "learning_rate": 0.00011639578337495787,
+      "loss": 0.6865,
+      "step": 1615
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.05667116865515709,
+      "learning_rate": 0.00011630935549418627,
+      "loss": 0.5988,
+      "step": 1616
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.0557304285466671,
+      "learning_rate": 0.00011622291509596234,
+      "loss": 0.6422,
+      "step": 1617
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.0590471550822258,
+      "learning_rate": 0.00011613646224662921,
+      "loss": 0.777,
+      "step": 1618
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.06502439081668854,
+      "learning_rate": 0.00011604999701253953,
+      "loss": 0.7052,
+      "step": 1619
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.05545181408524513,
+      "learning_rate": 0.00011596351946005552,
+      "loss": 0.6561,
+      "step": 1620
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.05329308658838272,
+      "learning_rate": 0.00011587702965554878,
+      "loss": 0.5068,
+      "step": 1621
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.05559957027435303,
+      "learning_rate": 0.00011579052766540039,
+      "loss": 0.6479,
+      "step": 1622
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.061965446919202805,
+      "learning_rate": 0.00011570401355600071,
+      "loss": 0.7318,
+      "step": 1623
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.061798419803380966,
+      "learning_rate": 0.00011561748739374945,
+      "loss": 0.7131,
+      "step": 1624
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.05673898011445999,
+      "learning_rate": 0.00011553094924505557,
+      "loss": 0.7431,
+      "step": 1625
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.055540382862091064,
+      "learning_rate": 0.00011544439917633718,
+      "loss": 0.6852,
+      "step": 1626
+    },
+    {
+      "epoch": 0.9,
+      "grad_norm": 0.05674376338720322,
+      "learning_rate": 0.00011535783725402163,
+      "loss": 0.7056,
+      "step": 1627
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.057587672024965286,
+      "learning_rate": 0.00011527126354454525,
+      "loss": 0.6101,
+      "step": 1628
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.06215091049671173,
+      "learning_rate": 0.00011518467811435352,
+      "loss": 0.742,
+      "step": 1629
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.054263584315776825,
+      "learning_rate": 0.00011509808102990085,
+      "loss": 0.6079,
+      "step": 1630
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.05362584441900253,
+      "learning_rate": 0.00011501147235765063,
+      "loss": 0.5511,
+      "step": 1631
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.05936874449253082,
+      "learning_rate": 0.00011492485216407513,
+      "loss": 0.6716,
+      "step": 1632
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.05744662880897522,
+      "learning_rate": 0.00011483822051565549,
+      "loss": 0.6429,
+      "step": 1633
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.06137154996395111,
+      "learning_rate": 0.0001147515774788816,
+      "loss": 0.6882,
+      "step": 1634
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.06001037731766701,
+      "learning_rate": 0.0001146649231202521,
+      "loss": 0.655,
+      "step": 1635
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.054453328251838684,
+      "learning_rate": 0.0001145782575062743,
+      "loss": 0.6446,
+      "step": 1636
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.056741863489151,
+      "learning_rate": 0.00011449158070346424,
+      "loss": 0.6569,
+      "step": 1637
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.05473573878407478,
+      "learning_rate": 0.00011440489277834645,
+      "loss": 0.6475,
+      "step": 1638
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.060271989554166794,
+      "learning_rate": 0.00011431819379745401,
+      "loss": 0.7187,
+      "step": 1639
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.05573005974292755,
+      "learning_rate": 0.00011423148382732853,
+      "loss": 0.639,
+      "step": 1640
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.05600711703300476,
+      "learning_rate": 0.00011414476293452001,
+      "loss": 0.6168,
+      "step": 1641
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.053048767149448395,
+      "learning_rate": 0.0001140580311855869,
+      "loss": 0.5581,
+      "step": 1642
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.05543733760714531,
+      "learning_rate": 0.00011397128864709586,
+      "loss": 0.6627,
+      "step": 1643
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.05584166571497917,
+      "learning_rate": 0.00011388453538562195,
+      "loss": 0.6437,
+      "step": 1644
+    },
+    {
+      "epoch": 0.91,
+      "grad_norm": 0.06031232327222824,
+      "learning_rate": 0.00011379777146774844,
+      "loss": 0.5947,
+      "step": 1645
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.055425215512514114,
+      "learning_rate": 0.0001137109969600667,
+      "loss": 0.6921,
+      "step": 1646
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.05608774349093437,
+      "learning_rate": 0.00011362421192917631,
+      "loss": 0.7744,
+      "step": 1647
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.06419660151004791,
+      "learning_rate": 0.00011353741644168487,
+      "loss": 0.7103,
+      "step": 1648
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.060391318053007126,
+      "learning_rate": 0.0001134506105642081,
+      "loss": 0.7168,
+      "step": 1649
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.05262453481554985,
+      "learning_rate": 0.00011336379436336955,
+      "loss": 0.5733,
+      "step": 1650
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.0592212975025177,
+      "learning_rate": 0.00011327696790580083,
+      "loss": 0.6959,
+      "step": 1651
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.061455368995666504,
+      "learning_rate": 0.00011319013125814131,
+      "loss": 0.661,
+      "step": 1652
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.05837690457701683,
+      "learning_rate": 0.00011310328448703829,
+      "loss": 0.6521,
+      "step": 1653
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.055113084614276886,
+      "learning_rate": 0.00011301642765914673,
+      "loss": 0.7173,
+      "step": 1654
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.06535536795854568,
+      "learning_rate": 0.00011292956084112943,
+      "loss": 0.6889,
+      "step": 1655
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.056155428290367126,
+      "learning_rate": 0.00011284268409965673,
+      "loss": 0.6287,
+      "step": 1656
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.05229957774281502,
+      "learning_rate": 0.00011275579750140666,
+      "loss": 0.617,
+      "step": 1657
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.05848075821995735,
+      "learning_rate": 0.00011266890111306484,
+      "loss": 0.6121,
+      "step": 1658
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.06387878954410553,
+      "learning_rate": 0.00011258199500132429,
+      "loss": 0.6307,
+      "step": 1659
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.05076488479971886,
+      "learning_rate": 0.00011249507923288562,
+      "loss": 0.6157,
+      "step": 1660
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.053368885070085526,
+      "learning_rate": 0.0001124081538744568,
+      "loss": 0.675,
+      "step": 1661
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.06017875671386719,
+      "learning_rate": 0.00011232121899275314,
+      "loss": 0.6557,
+      "step": 1662
+    },
+    {
+      "epoch": 0.92,
+      "grad_norm": 0.06376302242279053,
+      "learning_rate": 0.00011223427465449729,
+      "loss": 0.6831,
+      "step": 1663
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.06142215430736542,
+      "learning_rate": 0.00011214732092641916,
+      "loss": 0.6506,
+      "step": 1664
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.0578768253326416,
+      "learning_rate": 0.00011206035787525585,
+      "loss": 0.7647,
+      "step": 1665
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.05308634787797928,
+      "learning_rate": 0.00011197338556775156,
+      "loss": 0.6295,
+      "step": 1666
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.050338249653577805,
+      "learning_rate": 0.00011188640407065776,
+      "loss": 0.5454,
+      "step": 1667
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.05560674890875816,
+      "learning_rate": 0.00011179941345073278,
+      "loss": 0.7058,
+      "step": 1668
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.054467298090457916,
+      "learning_rate": 0.00011171241377474207,
+      "loss": 0.5801,
+      "step": 1669
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.05152300372719765,
+      "learning_rate": 0.00011162540510945799,
+      "loss": 0.5644,
+      "step": 1670
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.05688504874706268,
+      "learning_rate": 0.0001115383875216598,
+      "loss": 0.6589,
+      "step": 1671
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.05523272603750229,
+      "learning_rate": 0.00011145136107813363,
+      "loss": 0.6098,
+      "step": 1672
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.058240048587322235,
+      "learning_rate": 0.0001113643258456724,
+      "loss": 0.7046,
+      "step": 1673
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.054752644151449203,
+      "learning_rate": 0.00011127728189107576,
+      "loss": 0.7045,
+      "step": 1674
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.0511183924973011,
+      "learning_rate": 0.00011119022928115007,
+      "loss": 0.6525,
+      "step": 1675
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.05391978099942207,
+      "learning_rate": 0.00011110316808270831,
+      "loss": 0.6324,
+      "step": 1676
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.05610180273652077,
+      "learning_rate": 0.00011101609836257008,
+      "loss": 0.6382,
+      "step": 1677
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.05987925082445145,
+      "learning_rate": 0.00011092902018756151,
+      "loss": 0.7006,
+      "step": 1678
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.06315992027521133,
+      "learning_rate": 0.0001108419336245152,
+      "loss": 0.7349,
+      "step": 1679
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.05764647200703621,
+      "learning_rate": 0.0001107548387402702,
+      "loss": 0.6374,
+      "step": 1680
+    },
+    {
+      "epoch": 0.93,
+      "grad_norm": 0.05375726893544197,
+      "learning_rate": 0.00011066773560167196,
+      "loss": 0.6368,
+      "step": 1681
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.05238979682326317,
+      "learning_rate": 0.00011058062427557229,
+      "loss": 0.5693,
+      "step": 1682
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.055107586085796356,
+      "learning_rate": 0.00011049350482882919,
+      "loss": 0.6024,
+      "step": 1683
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.0552542470395565,
+      "learning_rate": 0.00011040637732830701,
+      "loss": 0.5507,
+      "step": 1684
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.06206995248794556,
+      "learning_rate": 0.00011031924184087618,
+      "loss": 0.7039,
+      "step": 1685
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.05662161484360695,
+      "learning_rate": 0.00011023209843341333,
+      "loss": 0.6329,
+      "step": 1686
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.060033902525901794,
+      "learning_rate": 0.00011014494717280115,
+      "loss": 0.6859,
+      "step": 1687
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.05562319979071617,
+      "learning_rate": 0.00011005778812592832,
+      "loss": 0.6442,
+      "step": 1688
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.05592164769768715,
+      "learning_rate": 0.00010997062135968956,
+      "loss": 0.6388,
+      "step": 1689
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.05650092288851738,
+      "learning_rate": 0.00010988344694098545,
+      "loss": 0.6502,
+      "step": 1690
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.06203974410891533,
+      "learning_rate": 0.00010979626493672245,
+      "loss": 0.7274,
+      "step": 1691
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.052279599010944366,
+      "learning_rate": 0.00010970907541381295,
+      "loss": 0.5828,
+      "step": 1692
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.06009140610694885,
+      "learning_rate": 0.00010962187843917497,
+      "loss": 0.6759,
+      "step": 1693
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.05527809262275696,
+      "learning_rate": 0.0001095346740797323,
+      "loss": 0.6657,
+      "step": 1694
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.0608220137655735,
+      "learning_rate": 0.00010944746240241444,
+      "loss": 0.6928,
+      "step": 1695
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.05583418905735016,
+      "learning_rate": 0.00010936024347415643,
+      "loss": 0.5839,
+      "step": 1696
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.05640941858291626,
+      "learning_rate": 0.00010927301736189893,
+      "loss": 0.661,
+      "step": 1697
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.0547964982688427,
+      "learning_rate": 0.0001091857841325881,
+      "loss": 0.562,
+      "step": 1698
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.05608777329325676,
+      "learning_rate": 0.00010909854385317557,
+      "loss": 0.5974,
+      "step": 1699
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.05370324105024338,
+      "learning_rate": 0.00010901129659061837,
+      "loss": 0.6899,
+      "step": 1700
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.06166364252567291,
+      "learning_rate": 0.00010892404241187886,
+      "loss": 0.6538,
+      "step": 1701
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.0633421465754509,
+      "learning_rate": 0.00010883678138392477,
+      "loss": 0.785,
+      "step": 1702
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.06182889640331268,
+      "learning_rate": 0.00010874951357372906,
+      "loss": 0.6403,
+      "step": 1703
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.06533250212669373,
+      "learning_rate": 0.0001086622390482699,
+      "loss": 0.7551,
+      "step": 1704
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.055042069405317307,
+      "learning_rate": 0.00010857495787453058,
+      "loss": 0.6083,
+      "step": 1705
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.06447125971317291,
+      "learning_rate": 0.00010848767011949952,
+      "loss": 0.615,
+      "step": 1706
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.056245360523462296,
+      "learning_rate": 0.00010840037585017022,
+      "loss": 0.6705,
+      "step": 1707
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.05670240893959999,
+      "learning_rate": 0.00010831307513354112,
+      "loss": 0.6491,
+      "step": 1708
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.05644702538847923,
+      "learning_rate": 0.00010822576803661564,
+      "loss": 0.6462,
+      "step": 1709
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.05394309014081955,
+      "learning_rate": 0.00010813845462640206,
+      "loss": 0.6643,
+      "step": 1710
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.057082321494817734,
+      "learning_rate": 0.00010805113496991364,
+      "loss": 0.7565,
+      "step": 1711
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.06196596845984459,
+      "learning_rate": 0.00010796380913416823,
+      "loss": 0.6655,
+      "step": 1712
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.06284771114587784,
+      "learning_rate": 0.0001078764771861886,
+      "loss": 0.647,
+      "step": 1713
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.06070135906338692,
+      "learning_rate": 0.0001077891391930021,
+      "loss": 0.6143,
+      "step": 1714
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.058664754033088684,
+      "learning_rate": 0.00010770179522164079,
+      "loss": 0.6618,
+      "step": 1715
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.0531286895275116,
+      "learning_rate": 0.00010761444533914125,
+      "loss": 0.6304,
+      "step": 1716
+    },
+    {
+      "epoch": 0.95,
+      "grad_norm": 0.053345970809459686,
+      "learning_rate": 0.0001075270896125446,
+      "loss": 0.6148,
+      "step": 1717
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.05614785850048065,
+      "learning_rate": 0.00010743972810889655,
+      "loss": 0.5945,
+      "step": 1718
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.05940824747085571,
+      "learning_rate": 0.00010735236089524716,
+      "loss": 0.6681,
+      "step": 1719
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.0561952069401741,
+      "learning_rate": 0.00010726498803865088,
+      "loss": 0.6925,
+      "step": 1720
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.053867727518081665,
+      "learning_rate": 0.00010717760960616643,
+      "loss": 0.5798,
+      "step": 1721
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.06327812373638153,
+      "learning_rate": 0.00010709022566485698,
+      "loss": 0.7174,
+      "step": 1722
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.06408220529556274,
+      "learning_rate": 0.00010700283628178975,
+      "loss": 0.7552,
+      "step": 1723
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.06066514551639557,
+      "learning_rate": 0.00010691544152403623,
+      "loss": 0.72,
+      "step": 1724
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.06779921799898148,
+      "learning_rate": 0.00010682804145867204,
+      "loss": 0.7299,
+      "step": 1725
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.060735609382390976,
+      "learning_rate": 0.0001067406361527768,
+      "loss": 0.6431,
+      "step": 1726
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.05965111032128334,
+      "learning_rate": 0.00010665322567343423,
+      "loss": 0.6426,
+      "step": 1727
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.05794420465826988,
+      "learning_rate": 0.00010656581008773198,
+      "loss": 0.6485,
+      "step": 1728
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.06031234934926033,
+      "learning_rate": 0.00010647838946276165,
+      "loss": 0.6548,
+      "step": 1729
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.056517090648412704,
+      "learning_rate": 0.00010639096386561864,
+      "loss": 0.6871,
+      "step": 1730
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.06209394708275795,
+      "learning_rate": 0.00010630353336340226,
+      "loss": 0.6645,
+      "step": 1731
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.059892479330301285,
+      "learning_rate": 0.00010621609802321555,
+      "loss": 0.6261,
+      "step": 1732
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.06408750265836716,
+      "learning_rate": 0.0001061286579121652,
+      "loss": 0.7269,
+      "step": 1733
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.057373423129320145,
+      "learning_rate": 0.00010604121309736164,
+      "loss": 0.6173,
+      "step": 1734
+    },
+    {
+      "epoch": 0.96,
+      "grad_norm": 0.055811841040849686,
+      "learning_rate": 0.00010595376364591889,
+      "loss": 0.6444,
+      "step": 1735
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.05903814360499382,
+      "learning_rate": 0.00010586630962495452,
+      "loss": 0.7005,
+      "step": 1736
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.06342557072639465,
+      "learning_rate": 0.00010577885110158958,
+      "loss": 0.6226,
+      "step": 1737
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.06033981963992119,
+      "learning_rate": 0.00010569138814294864,
+      "loss": 0.6606,
+      "step": 1738
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.0658353865146637,
+      "learning_rate": 0.00010560392081615962,
+      "loss": 0.6379,
+      "step": 1739
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.05433519929647446,
+      "learning_rate": 0.00010551644918835381,
+      "loss": 0.5832,
+      "step": 1740
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.058659233152866364,
+      "learning_rate": 0.00010542897332666581,
+      "loss": 0.6252,
+      "step": 1741
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.06791430711746216,
+      "learning_rate": 0.00010534149329823349,
+      "loss": 0.7116,
+      "step": 1742
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.06629349291324615,
+      "learning_rate": 0.00010525400917019784,
+      "loss": 0.9018,
+      "step": 1743
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.05723525211215019,
+      "learning_rate": 0.00010516652100970308,
+      "loss": 0.6521,
+      "step": 1744
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.0574793666601181,
+      "learning_rate": 0.00010507902888389647,
+      "loss": 0.65,
+      "step": 1745
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.05418836697936058,
+      "learning_rate": 0.00010499153285992833,
+      "loss": 0.5964,
+      "step": 1746
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.05051202327013016,
+      "learning_rate": 0.00010490403300495201,
+      "loss": 0.6072,
+      "step": 1747
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.05381901189684868,
+      "learning_rate": 0.00010481652938612374,
+      "loss": 0.5846,
+      "step": 1748
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.055063627660274506,
+      "learning_rate": 0.00010472902207060265,
+      "loss": 0.5894,
+      "step": 1749
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.061404235661029816,
+      "learning_rate": 0.00010464151112555077,
+      "loss": 0.6717,
+      "step": 1750
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.05025755986571312,
+      "learning_rate": 0.00010455399661813283,
+      "loss": 0.5984,
+      "step": 1751
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.0560632087290287,
+      "learning_rate": 0.00010446647861551633,
+      "loss": 0.5581,
+      "step": 1752
+    },
+    {
+      "epoch": 0.97,
+      "grad_norm": 0.06142037361860275,
+      "learning_rate": 0.0001043789571848715,
+      "loss": 0.5679,
+      "step": 1753
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.056344058364629745,
+      "learning_rate": 0.00010429143239337112,
+      "loss": 0.6612,
+      "step": 1754
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.06066104397177696,
+      "learning_rate": 0.00010420390430819058,
+      "loss": 0.7477,
+      "step": 1755
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.05510355904698372,
+      "learning_rate": 0.00010411637299650783,
+      "loss": 0.6163,
+      "step": 1756
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.057294245809316635,
+      "learning_rate": 0.00010402883852550325,
+      "loss": 0.6468,
+      "step": 1757
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.056174419820308685,
+      "learning_rate": 0.00010394130096235966,
+      "loss": 0.6213,
+      "step": 1758
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.06194084510207176,
+      "learning_rate": 0.00010385376037426226,
+      "loss": 0.7077,
+      "step": 1759
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.06177836284041405,
+      "learning_rate": 0.00010376621682839857,
+      "loss": 0.6536,
+      "step": 1760
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.059714220464229584,
+      "learning_rate": 0.00010367867039195842,
+      "loss": 0.5912,
+      "step": 1761
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.052113138139247894,
+      "learning_rate": 0.00010359112113213376,
+      "loss": 0.5401,
+      "step": 1762
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.06121697649359703,
+      "learning_rate": 0.0001035035691161188,
+      "loss": 0.6862,
+      "step": 1763
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.05976003035902977,
+      "learning_rate": 0.00010341601441110983,
+      "loss": 0.58,
+      "step": 1764
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.05741668865084648,
+      "learning_rate": 0.00010332845708430519,
+      "loss": 0.5233,
+      "step": 1765
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.06236950680613518,
+      "learning_rate": 0.00010324089720290521,
+      "loss": 0.7257,
+      "step": 1766
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.06206139549612999,
+      "learning_rate": 0.00010315333483411232,
+      "loss": 0.6985,
+      "step": 1767
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.05406549200415611,
+      "learning_rate": 0.00010306577004513065,
+      "loss": 0.5677,
+      "step": 1768
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.06077892333269119,
+      "learning_rate": 0.0001029782029031663,
+      "loss": 0.6745,
+      "step": 1769
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.06338007748126984,
+      "learning_rate": 0.00010289063347542726,
+      "loss": 0.6612,
+      "step": 1770
+    },
+    {
+      "epoch": 0.98,
+      "grad_norm": 0.060978084802627563,
+      "learning_rate": 0.00010280306182912313,
+      "loss": 0.7625,
+      "step": 1771
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.0552746020257473,
+      "learning_rate": 0.00010271548803146526,
+      "loss": 0.5827,
+      "step": 1772
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.05428915098309517,
+      "learning_rate": 0.00010262791214966668,
+      "loss": 0.6786,
+      "step": 1773
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.057721514254808426,
+      "learning_rate": 0.00010254033425094197,
+      "loss": 0.5914,
+      "step": 1774
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.05720565468072891,
+      "learning_rate": 0.00010245275440250728,
+      "loss": 0.6013,
+      "step": 1775
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.05914613604545593,
+      "learning_rate": 0.00010236517267158028,
+      "loss": 0.6439,
+      "step": 1776
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.057543400675058365,
+      "learning_rate": 0.00010227758912538008,
+      "loss": 0.7292,
+      "step": 1777
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.056838035583496094,
+      "learning_rate": 0.00010219000383112713,
+      "loss": 0.6506,
+      "step": 1778
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.05806044489145279,
+      "learning_rate": 0.00010210241685604331,
+      "loss": 0.6114,
+      "step": 1779
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.0533684603869915,
+      "learning_rate": 0.00010201482826735172,
+      "loss": 0.5946,
+      "step": 1780
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.05922561138868332,
+      "learning_rate": 0.00010192723813227672,
+      "loss": 0.6626,
+      "step": 1781
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.055819686502218246,
+      "learning_rate": 0.00010183964651804382,
+      "loss": 0.6422,
+      "step": 1782
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.050924383103847504,
+      "learning_rate": 0.00010175205349187977,
+      "loss": 0.5344,
+      "step": 1783
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.06296874582767487,
+      "learning_rate": 0.00010166445912101231,
+      "loss": 0.6451,
+      "step": 1784
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.05765317752957344,
+      "learning_rate": 0.00010157686347267021,
+      "loss": 0.6403,
+      "step": 1785
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.060042284429073334,
+      "learning_rate": 0.00010148926661408327,
+      "loss": 0.6919,
+      "step": 1786
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.06098796799778938,
+      "learning_rate": 0.0001014016686124822,
+      "loss": 0.6371,
+      "step": 1787
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.05685941502451897,
+      "learning_rate": 0.00010131406953509857,
+      "loss": 0.6399,
+      "step": 1788
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 0.06483474373817444,
+      "learning_rate": 0.00010122646944916483,
+      "loss": 0.6915,
+      "step": 1789
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.06274085491895676,
+      "learning_rate": 0.00010113886842191408,
+      "loss": 0.6212,
+      "step": 1790
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.06120248883962631,
+      "learning_rate": 0.00010105126652058032,
+      "loss": 0.6859,
+      "step": 1791
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.06124896928668022,
+      "learning_rate": 0.00010096366381239808,
+      "loss": 0.5996,
+      "step": 1792
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.05330495536327362,
+      "learning_rate": 0.00010087606036460257,
+      "loss": 0.7189,
+      "step": 1793
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.06091773882508278,
+      "learning_rate": 0.00010078845624442954,
+      "loss": 0.6692,
+      "step": 1794
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.057791825383901596,
+      "learning_rate": 0.0001007008515191153,
+      "loss": 0.6403,
+      "step": 1795
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.06233106181025505,
+      "learning_rate": 0.00010061324625589657,
+      "loss": 0.6166,
+      "step": 1796
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.06440749019384384,
+      "learning_rate": 0.0001005256405220105,
+      "loss": 0.7032,
+      "step": 1797
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.060089919716119766,
+      "learning_rate": 0.00010043803438469461,
+      "loss": 0.6484,
+      "step": 1798
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.05943863466382027,
+      "learning_rate": 0.00010035042791118674,
+      "loss": 0.6422,
+      "step": 1799
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.07573998719453812,
+      "learning_rate": 0.00010026282116872499,
+      "loss": 0.6176,
+      "step": 1800
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.061618976294994354,
+      "learning_rate": 0.00010017521422454763,
+      "loss": 0.7475,
+      "step": 1801
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.05466758459806442,
+      "learning_rate": 0.00010008760714589311,
+      "loss": 0.6769,
+      "step": 1802
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.05463040620088577,
+      "learning_rate": 0.0001,
+      "loss": 0.6142,
+      "step": 1803
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.05533997714519501,
+      "learning_rate": 9.991239285410691e-05,
+      "loss": 0.6714,
+      "step": 1804
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.0693361908197403,
+      "learning_rate": 9.98247857754524e-05,
+      "loss": 0.6588,
+      "step": 1805
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.061774842441082,
+      "learning_rate": 9.973717883127504e-05,
+      "loss": 0.7012,
+      "step": 1806
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.057546455413103104,
+      "learning_rate": 9.964957208881329e-05,
+      "loss": 0.6242,
+      "step": 1807
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.059542592614889145,
+      "learning_rate": 9.956196561530541e-05,
+      "loss": 0.5466,
+      "step": 1808
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.07098076492547989,
+      "learning_rate": 9.947435947798953e-05,
+      "loss": 0.7131,
+      "step": 1809
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.05665259063243866,
+      "learning_rate": 9.938675374410347e-05,
+      "loss": 0.6429,
+      "step": 1810
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06300745159387589,
+      "learning_rate": 9.929914848088473e-05,
+      "loss": 0.6545,
+      "step": 1811
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.05833979323506355,
+      "learning_rate": 9.921154375557047e-05,
+      "loss": 0.5777,
+      "step": 1812
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06335113197565079,
+      "learning_rate": 9.912393963539745e-05,
+      "loss": 0.7244,
+      "step": 1813
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06383019685745239,
+      "learning_rate": 9.903633618760195e-05,
+      "loss": 0.6863,
+      "step": 1814
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.0583442859351635,
+      "learning_rate": 9.894873347941971e-05,
+      "loss": 0.6324,
+      "step": 1815
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06413304060697556,
+      "learning_rate": 9.886113157808594e-05,
+      "loss": 0.6864,
+      "step": 1816
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.05635803937911987,
+      "learning_rate": 9.87735305508352e-05,
+      "loss": 0.6943,
+      "step": 1817
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.061192139983177185,
+      "learning_rate": 9.868593046490144e-05,
+      "loss": 0.6384,
+      "step": 1818
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.060894276946783066,
+      "learning_rate": 9.859833138751783e-05,
+      "loss": 0.6649,
+      "step": 1819
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.066097192466259,
+      "learning_rate": 9.851073338591675e-05,
+      "loss": 0.7619,
+      "step": 1820
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06046619266271591,
+      "learning_rate": 9.842313652732982e-05,
+      "loss": 0.7299,
+      "step": 1821
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06305580586194992,
+      "learning_rate": 9.833554087898773e-05,
+      "loss": 0.6843,
+      "step": 1822
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.05506165698170662,
+      "learning_rate": 9.824794650812026e-05,
+      "loss": 0.5665,
+      "step": 1823
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06139134615659714,
+      "learning_rate": 9.81603534819562e-05,
+      "loss": 0.7178,
+      "step": 1824
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.05725986137986183,
+      "learning_rate": 9.807276186772333e-05,
+      "loss": 0.6013,
+      "step": 1825
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.060024797916412354,
+      "learning_rate": 9.798517173264833e-05,
+      "loss": 0.6464,
+      "step": 1826
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.06363005191087723,
+      "learning_rate": 9.789758314395672e-05,
+      "loss": 0.6539,
+      "step": 1827
+    },
+    {
+      "epoch": 1.02,
+      "eval_loss": 0.6588459610939026,
+      "eval_runtime": 196.7902,
+      "eval_samples_per_second": 28.391,
+      "eval_steps_per_second": 14.198,
+      "step": 1827
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.11711059510707855,
+      "learning_rate": 9.780999616887288e-05,
+      "loss": 0.7841,
+      "step": 1828
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.06279384344816208,
+      "learning_rate": 9.772241087461997e-05,
+      "loss": 0.6586,
+      "step": 1829
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.06254998594522476,
+      "learning_rate": 9.763482732841976e-05,
+      "loss": 0.6021,
+      "step": 1830
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.06098167970776558,
+      "learning_rate": 9.754724559749276e-05,
+      "loss": 0.6118,
+      "step": 1831
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.06855977326631546,
+      "learning_rate": 9.74596657490581e-05,
+      "loss": 0.6888,
+      "step": 1832
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.051197465509176254,
+      "learning_rate": 9.737208785033337e-05,
+      "loss": 0.5669,
+      "step": 1833
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.06223751977086067,
+      "learning_rate": 9.728451196853477e-05,
+      "loss": 0.6522,
+      "step": 1834
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.05435439944267273,
+      "learning_rate": 9.71969381708769e-05,
+      "loss": 0.534,
+      "step": 1835
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.05871767923235893,
+      "learning_rate": 9.710936652457276e-05,
+      "loss": 0.5852,
+      "step": 1836
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06656838208436966,
+      "learning_rate": 9.70217970968337e-05,
+      "loss": 0.6139,
+      "step": 1837
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06376973539590836,
+      "learning_rate": 9.693422995486939e-05,
+      "loss": 0.608,
+      "step": 1838
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.058163829147815704,
+      "learning_rate": 9.684666516588772e-05,
+      "loss": 0.528,
+      "step": 1839
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.062393542379140854,
+      "learning_rate": 9.675910279709477e-05,
+      "loss": 0.6297,
+      "step": 1840
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06609448045492172,
+      "learning_rate": 9.667154291569482e-05,
+      "loss": 0.6068,
+      "step": 1841
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06638655811548233,
+      "learning_rate": 9.658398558889018e-05,
+      "loss": 0.6549,
+      "step": 1842
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06767024099826813,
+      "learning_rate": 9.649643088388119e-05,
+      "loss": 0.5385,
+      "step": 1843
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06396374851465225,
+      "learning_rate": 9.640887886786624e-05,
+      "loss": 0.7188,
+      "step": 1844
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06499580293893814,
+      "learning_rate": 9.632132960804159e-05,
+      "loss": 0.6873,
+      "step": 1845
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06490854173898697,
+      "learning_rate": 9.623378317160142e-05,
+      "loss": 0.6862,
+      "step": 1846
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.053400713950395584,
+      "learning_rate": 9.614623962573776e-05,
+      "loss": 0.4934,
+      "step": 1847
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.060843680053949356,
+      "learning_rate": 9.605869903764036e-05,
+      "loss": 0.5619,
+      "step": 1848
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.05479586124420166,
+      "learning_rate": 9.597116147449676e-05,
+      "loss": 0.5344,
+      "step": 1849
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06369245052337646,
+      "learning_rate": 9.588362700349218e-05,
+      "loss": 0.5114,
+      "step": 1850
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.05581709370017052,
+      "learning_rate": 9.579609569180942e-05,
+      "loss": 0.5555,
+      "step": 1851
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.059186048805713654,
+      "learning_rate": 9.570856760662889e-05,
+      "loss": 0.6179,
+      "step": 1852
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06566520035266876,
+      "learning_rate": 9.562104281512852e-05,
+      "loss": 0.6656,
+      "step": 1853
+    },
+    {
+      "epoch": 1.01,
+      "grad_norm": 0.06835220754146576,
+      "learning_rate": 9.553352138448366e-05,
+      "loss": 0.7285,
+      "step": 1854
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.06603655219078064,
+      "learning_rate": 9.54460033818672e-05,
+      "loss": 0.6859,
+      "step": 1855
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.0615200474858284,
+      "learning_rate": 9.535848887444925e-05,
+      "loss": 0.6289,
+      "step": 1856
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.06494131684303284,
+      "learning_rate": 9.527097792939737e-05,
+      "loss": 0.6473,
+      "step": 1857
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.06510718911886215,
+      "learning_rate": 9.518347061387628e-05,
+      "loss": 0.6555,
+      "step": 1858
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.06615065038204193,
+      "learning_rate": 9.509596699504801e-05,
+      "loss": 0.5815,
+      "step": 1859
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.06684267520904541,
+      "learning_rate": 9.500846714007168e-05,
+      "loss": 0.6201,
+      "step": 1860
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.06057841703295708,
+      "learning_rate": 9.492097111610357e-05,
+      "loss": 0.6083,
+      "step": 1861
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.06085360795259476,
+      "learning_rate": 9.483347899029695e-05,
+      "loss": 0.5041,
+      "step": 1862
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.06703361123800278,
+      "learning_rate": 9.474599082980217e-05,
+      "loss": 0.6671,
+      "step": 1863
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.0637744590640068,
+      "learning_rate": 9.465850670176654e-05,
+      "loss": 0.6451,
+      "step": 1864
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.06312189251184464,
+      "learning_rate": 9.45710266733342e-05,
+      "loss": 0.6395,
+      "step": 1865
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.0770537480711937,
+      "learning_rate": 9.448355081164621e-05,
+      "loss": 0.6448,
+      "step": 1866
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.061500776559114456,
+      "learning_rate": 9.439607918384039e-05,
+      "loss": 0.6372,
+      "step": 1867
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.059113096445798874,
+      "learning_rate": 9.430861185705137e-05,
+      "loss": 0.5822,
+      "step": 1868
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.0576748363673687,
+      "learning_rate": 9.422114889841044e-05,
+      "loss": 0.5552,
+      "step": 1869
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.06234674155712128,
+      "learning_rate": 9.413369037504552e-05,
+      "loss": 0.6586,
+      "step": 1870
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.06132230535149574,
+      "learning_rate": 9.404623635408112e-05,
+      "loss": 0.6556,
+      "step": 1871
+    },
+    {
+      "epoch": 1.02,
+      "grad_norm": 0.07447368651628494,
+      "learning_rate": 9.395878690263837e-05,
+      "loss": 0.6087,
+      "step": 1872
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.05781631916761398,
+      "learning_rate": 9.387134208783482e-05,
+      "loss": 0.5581,
+      "step": 1873
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.06479276716709137,
+      "learning_rate": 9.378390197678449e-05,
+      "loss": 0.6272,
+      "step": 1874
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.05904553085565567,
+      "learning_rate": 9.369646663659775e-05,
+      "loss": 0.534,
+      "step": 1875
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.0636134073138237,
+      "learning_rate": 9.360903613438138e-05,
+      "loss": 0.6396,
+      "step": 1876
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.06313899904489517,
+      "learning_rate": 9.352161053723838e-05,
+      "loss": 0.6119,
+      "step": 1877
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.0698072761297226,
+      "learning_rate": 9.343418991226803e-05,
+      "loss": 0.6818,
+      "step": 1878
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.05524126812815666,
+      "learning_rate": 9.33467743265658e-05,
+      "loss": 0.506,
+      "step": 1879
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.06468210369348526,
+      "learning_rate": 9.325936384722321e-05,
+      "loss": 0.6361,
+      "step": 1880
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.0695754885673523,
+      "learning_rate": 9.317195854132798e-05,
+      "loss": 0.6385,
+      "step": 1881
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.06824535876512527,
+      "learning_rate": 9.308455847596378e-05,
+      "loss": 0.6619,
+      "step": 1882
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.06896868348121643,
+      "learning_rate": 9.299716371821027e-05,
+      "loss": 0.6911,
+      "step": 1883
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.05827944353222847,
+      "learning_rate": 9.290977433514306e-05,
+      "loss": 0.5489,
+      "step": 1884
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.0637725293636322,
+      "learning_rate": 9.282239039383358e-05,
+      "loss": 0.7276,
+      "step": 1885
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.05886625126004219,
+      "learning_rate": 9.273501196134915e-05,
+      "loss": 0.569,
+      "step": 1886
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.07639192044734955,
+      "learning_rate": 9.264763910475285e-05,
+      "loss": 0.7335,
+      "step": 1887
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.05936943367123604,
+      "learning_rate": 9.256027189110346e-05,
+      "loss": 0.5261,
+      "step": 1888
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.06819011270999908,
+      "learning_rate": 9.247291038745542e-05,
+      "loss": 0.5912,
+      "step": 1889
+    },
+    {
+      "epoch": 1.03,
+      "grad_norm": 0.05976708605885506,
+      "learning_rate": 9.23855546608588e-05,
+      "loss": 0.5962,
+      "step": 1890
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.06304595619440079,
+      "learning_rate": 9.229820477835927e-05,
+      "loss": 0.6114,
+      "step": 1891
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.06259685754776001,
+      "learning_rate": 9.221086080699793e-05,
+      "loss": 0.5958,
+      "step": 1892
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.06620876491069794,
+      "learning_rate": 9.212352281381143e-05,
+      "loss": 0.6885,
+      "step": 1893
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.058716122061014175,
+      "learning_rate": 9.203619086583179e-05,
+      "loss": 0.5823,
+      "step": 1894
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.0609891451895237,
+      "learning_rate": 9.19488650300864e-05,
+      "loss": 0.5641,
+      "step": 1895
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.06413526087999344,
+      "learning_rate": 9.186154537359796e-05,
+      "loss": 0.5867,
+      "step": 1896
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.0639336034655571,
+      "learning_rate": 9.177423196338442e-05,
+      "loss": 0.6663,
+      "step": 1897
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.06244528293609619,
+      "learning_rate": 9.168692486645893e-05,
+      "loss": 0.5846,
+      "step": 1898
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.05837123841047287,
+      "learning_rate": 9.159962414982983e-05,
+      "loss": 0.5787,
+      "step": 1899
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.0634046420454979,
+      "learning_rate": 9.151232988050051e-05,
+      "loss": 0.5736,
+      "step": 1900
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.05949348211288452,
+      "learning_rate": 9.142504212546947e-05,
+      "loss": 0.5429,
+      "step": 1901
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.05976530909538269,
+      "learning_rate": 9.133776095173015e-05,
+      "loss": 0.5851,
+      "step": 1902
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.0690242350101471,
+      "learning_rate": 9.125048642627098e-05,
+      "loss": 0.6296,
+      "step": 1903
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.06474726647138596,
+      "learning_rate": 9.116321861607524e-05,
+      "loss": 0.6606,
+      "step": 1904
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.0658322274684906,
+      "learning_rate": 9.107595758812117e-05,
+      "loss": 0.5752,
+      "step": 1905
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.059737034142017365,
+      "learning_rate": 9.098870340938168e-05,
+      "loss": 0.589,
+      "step": 1906
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.05496500805020332,
+      "learning_rate": 9.090145614682444e-05,
+      "loss": 0.583,
+      "step": 1907
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 0.0656789019703865,
+      "learning_rate": 9.081421586741189e-05,
+      "loss": 0.6171,
+      "step": 1908
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.06451639533042908,
+      "learning_rate": 9.072698263810107e-05,
+      "loss": 0.6105,
+      "step": 1909
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.061076030135154724,
+      "learning_rate": 9.063975652584358e-05,
+      "loss": 0.5201,
+      "step": 1910
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.06417499482631683,
+      "learning_rate": 9.055253759758557e-05,
+      "loss": 0.567,
+      "step": 1911
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.06392619013786316,
+      "learning_rate": 9.04653259202677e-05,
+      "loss": 0.6125,
+      "step": 1912
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.06343825906515121,
+      "learning_rate": 9.037812156082504e-05,
+      "loss": 0.6414,
+      "step": 1913
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.06423365324735641,
+      "learning_rate": 9.029092458618705e-05,
+      "loss": 0.5424,
+      "step": 1914
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.065882109105587,
+      "learning_rate": 9.020373506327754e-05,
+      "loss": 0.5651,
+      "step": 1915
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.07035372406244278,
+      "learning_rate": 9.011655305901458e-05,
+      "loss": 0.7719,
+      "step": 1916
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.0645388588309288,
+      "learning_rate": 9.002937864031047e-05,
+      "loss": 0.5106,
+      "step": 1917
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.06212789565324783,
+      "learning_rate": 8.994221187407167e-05,
+      "loss": 0.5512,
+      "step": 1918
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.06453739106655121,
+      "learning_rate": 8.985505282719885e-05,
+      "loss": 0.5985,
+      "step": 1919
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.06423369795084,
+      "learning_rate": 8.976790156658665e-05,
+      "loss": 0.6009,
+      "step": 1920
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.06537653505802155,
+      "learning_rate": 8.968075815912381e-05,
+      "loss": 0.5746,
+      "step": 1921
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.06721478700637817,
+      "learning_rate": 8.9593622671693e-05,
+      "loss": 0.5686,
+      "step": 1922
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.0701824277639389,
+      "learning_rate": 8.950649517117081e-05,
+      "loss": 0.7034,
+      "step": 1923
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.058217499405145645,
+      "learning_rate": 8.941937572442773e-05,
+      "loss": 0.5781,
+      "step": 1924
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.0669047012925148,
+      "learning_rate": 8.933226439832805e-05,
+      "loss": 0.6353,
+      "step": 1925
+    },
+    {
+      "epoch": 1.05,
+      "grad_norm": 0.06103359907865524,
+      "learning_rate": 8.924516125972984e-05,
+      "loss": 0.5691,
+      "step": 1926
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06625822931528091,
+      "learning_rate": 8.915806637548482e-05,
+      "loss": 0.6668,
+      "step": 1927
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06848111748695374,
+      "learning_rate": 8.907097981243851e-05,
+      "loss": 0.6594,
+      "step": 1928
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06156426668167114,
+      "learning_rate": 8.898390163742993e-05,
+      "loss": 0.6222,
+      "step": 1929
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06666103005409241,
+      "learning_rate": 8.88968319172917e-05,
+      "loss": 0.6358,
+      "step": 1930
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.0592927522957325,
+      "learning_rate": 8.880977071884994e-05,
+      "loss": 0.5957,
+      "step": 1931
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.07227488607168198,
+      "learning_rate": 8.872271810892425e-05,
+      "loss": 0.5982,
+      "step": 1932
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06739647686481476,
+      "learning_rate": 8.863567415432763e-05,
+      "loss": 0.5339,
+      "step": 1933
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06235653534531593,
+      "learning_rate": 8.85486389218664e-05,
+      "loss": 0.636,
+      "step": 1934
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06076808646321297,
+      "learning_rate": 8.846161247834024e-05,
+      "loss": 0.6074,
+      "step": 1935
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06882358342409134,
+      "learning_rate": 8.837459489054204e-05,
+      "loss": 0.5976,
+      "step": 1936
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.058474499732255936,
+      "learning_rate": 8.828758622525797e-05,
+      "loss": 0.614,
+      "step": 1937
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06190056726336479,
+      "learning_rate": 8.820058654926725e-05,
+      "loss": 0.5597,
+      "step": 1938
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.05922474339604378,
+      "learning_rate": 8.811359592934228e-05,
+      "loss": 0.5483,
+      "step": 1939
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06522875279188156,
+      "learning_rate": 8.802661443224845e-05,
+      "loss": 0.7029,
+      "step": 1940
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06892760097980499,
+      "learning_rate": 8.793964212474419e-05,
+      "loss": 0.6374,
+      "step": 1941
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06366990506649017,
+      "learning_rate": 8.785267907358085e-05,
+      "loss": 0.6161,
+      "step": 1942
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.06452450156211853,
+      "learning_rate": 8.776572534550272e-05,
+      "loss": 0.5453,
+      "step": 1943
+    },
+    {
+      "epoch": 1.06,
+      "grad_norm": 0.0678330734372139,
+      "learning_rate": 8.76787810072469e-05,
+      "loss": 0.6144,
+      "step": 1944
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06669484078884125,
+      "learning_rate": 8.759184612554321e-05,
+      "loss": 0.5738,
+      "step": 1945
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06842266023159027,
+      "learning_rate": 8.750492076711439e-05,
+      "loss": 0.6512,
+      "step": 1946
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06860072910785675,
+      "learning_rate": 8.741800499867573e-05,
+      "loss": 0.6068,
+      "step": 1947
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06875675916671753,
+      "learning_rate": 8.73310988869352e-05,
+      "loss": 0.5997,
+      "step": 1948
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06354532390832901,
+      "learning_rate": 8.724420249859335e-05,
+      "loss": 0.5348,
+      "step": 1949
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06942533701658249,
+      "learning_rate": 8.71573159003433e-05,
+      "loss": 0.6694,
+      "step": 1950
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06216884031891823,
+      "learning_rate": 8.70704391588706e-05,
+      "loss": 0.5807,
+      "step": 1951
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06261352449655533,
+      "learning_rate": 8.698357234085328e-05,
+      "loss": 0.5809,
+      "step": 1952
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06839074939489365,
+      "learning_rate": 8.689671551296175e-05,
+      "loss": 0.5921,
+      "step": 1953
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06489168107509613,
+      "learning_rate": 8.680986874185873e-05,
+      "loss": 0.5952,
+      "step": 1954
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.05821004509925842,
+      "learning_rate": 8.67230320941992e-05,
+      "loss": 0.5973,
+      "step": 1955
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.059854816645383835,
+      "learning_rate": 8.663620563663046e-05,
+      "loss": 0.5361,
+      "step": 1956
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.05962301418185234,
+      "learning_rate": 8.654938943579194e-05,
+      "loss": 0.5073,
+      "step": 1957
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06811553239822388,
+      "learning_rate": 8.646258355831514e-05,
+      "loss": 0.58,
+      "step": 1958
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06248083338141441,
+      "learning_rate": 8.637578807082373e-05,
+      "loss": 0.6003,
+      "step": 1959
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06940522789955139,
+      "learning_rate": 8.628900303993335e-05,
+      "loss": 0.7196,
+      "step": 1960
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.07490598410367966,
+      "learning_rate": 8.620222853225161e-05,
+      "loss": 0.591,
+      "step": 1961
+    },
+    {
+      "epoch": 1.07,
+      "grad_norm": 0.06339261680841446,
+      "learning_rate": 8.611546461437808e-05,
+      "loss": 0.5554,
+      "step": 1962
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.06448446959257126,
+      "learning_rate": 8.602871135290418e-05,
+      "loss": 0.6441,
+      "step": 1963
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.06204627826809883,
+      "learning_rate": 8.594196881441314e-05,
+      "loss": 0.6758,
+      "step": 1964
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.06171583756804466,
+      "learning_rate": 8.585523706548001e-05,
+      "loss": 0.6001,
+      "step": 1965
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.07351014763116837,
+      "learning_rate": 8.57685161726715e-05,
+      "loss": 0.6279,
+      "step": 1966
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.07400185614824295,
+      "learning_rate": 8.568180620254603e-05,
+      "loss": 0.7228,
+      "step": 1967
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.07116332650184631,
+      "learning_rate": 8.55951072216536e-05,
+      "loss": 0.63,
+      "step": 1968
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.06596235930919647,
+      "learning_rate": 8.550841929653579e-05,
+      "loss": 0.5578,
+      "step": 1969
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.06503760814666748,
+      "learning_rate": 8.542174249372573e-05,
+      "loss": 0.5068,
+      "step": 1970
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.06710238754749298,
+      "learning_rate": 8.533507687974795e-05,
+      "loss": 0.6624,
+      "step": 1971
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.0639219582080841,
+      "learning_rate": 8.524842252111844e-05,
+      "loss": 0.6277,
+      "step": 1972
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.06799468398094177,
+      "learning_rate": 8.516177948434452e-05,
+      "loss": 0.5988,
+      "step": 1973
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.06464463472366333,
+      "learning_rate": 8.507514783592488e-05,
+      "loss": 0.4894,
+      "step": 1974
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.0756271556019783,
+      "learning_rate": 8.498852764234939e-05,
+      "loss": 0.6645,
+      "step": 1975
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.06705022603273392,
+      "learning_rate": 8.490191897009916e-05,
+      "loss": 0.543,
+      "step": 1976
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.06154650077223778,
+      "learning_rate": 8.481532188564648e-05,
+      "loss": 0.5,
+      "step": 1977
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.06594829261302948,
+      "learning_rate": 8.472873645545474e-05,
+      "loss": 0.5656,
+      "step": 1978
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.07296450436115265,
+      "learning_rate": 8.464216274597838e-05,
+      "loss": 0.7295,
+      "step": 1979
+    },
+    {
+      "epoch": 1.08,
+      "grad_norm": 0.06797216832637787,
+      "learning_rate": 8.45556008236628e-05,
+      "loss": 0.5848,
+      "step": 1980
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.06360683590173721,
+      "learning_rate": 8.446905075494443e-05,
+      "loss": 0.6403,
+      "step": 1981
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.06966099143028259,
+      "learning_rate": 8.438251260625056e-05,
+      "loss": 0.6367,
+      "step": 1982
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.06762167811393738,
+      "learning_rate": 8.42959864439993e-05,
+      "loss": 0.6355,
+      "step": 1983
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.07219623774290085,
+      "learning_rate": 8.420947233459962e-05,
+      "loss": 0.6437,
+      "step": 1984
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.0665636733174324,
+      "learning_rate": 8.412297034445123e-05,
+      "loss": 0.6083,
+      "step": 1985
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.06295057386159897,
+      "learning_rate": 8.403648053994447e-05,
+      "loss": 0.5336,
+      "step": 1986
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.05705847963690758,
+      "learning_rate": 8.395000298746045e-05,
+      "loss": 0.5142,
+      "step": 1987
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.0647684782743454,
+      "learning_rate": 8.38635377533708e-05,
+      "loss": 0.5619,
+      "step": 1988
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.07010988146066666,
+      "learning_rate": 8.377708490403767e-05,
+      "loss": 0.642,
+      "step": 1989
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.06674306839704514,
+      "learning_rate": 8.369064450581373e-05,
+      "loss": 0.6161,
+      "step": 1990
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.0655483826994896,
+      "learning_rate": 8.360421662504214e-05,
+      "loss": 0.6301,
+      "step": 1991
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.07575639337301254,
+      "learning_rate": 8.35178013280564e-05,
+      "loss": 0.5834,
+      "step": 1992
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.07528828829526901,
+      "learning_rate": 8.343139868118036e-05,
+      "loss": 0.7007,
+      "step": 1993
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.06885155290365219,
+      "learning_rate": 8.334500875072818e-05,
+      "loss": 0.5922,
+      "step": 1994
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.0729137510061264,
+      "learning_rate": 8.325863160300423e-05,
+      "loss": 0.5965,
+      "step": 1995
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.06822966039180756,
+      "learning_rate": 8.317226730430309e-05,
+      "loss": 0.5858,
+      "step": 1996
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.06264076381921768,
+      "learning_rate": 8.30859159209095e-05,
+      "loss": 0.5204,
+      "step": 1997
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 0.0732724666595459,
+      "learning_rate": 8.299957751909825e-05,
+      "loss": 0.6712,
+      "step": 1998
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.07096925377845764,
+      "learning_rate": 8.291325216513419e-05,
+      "loss": 0.6521,
+      "step": 1999
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.06218922510743141,
+      "learning_rate": 8.282693992527213e-05,
+      "loss": 0.574,
+      "step": 2000
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.07510274648666382,
+      "learning_rate": 8.274064086575681e-05,
+      "loss": 0.6368,
+      "step": 2001
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.06338323652744293,
+      "learning_rate": 8.265435505282293e-05,
+      "loss": 0.5627,
+      "step": 2002
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.06242498382925987,
+      "learning_rate": 8.256808255269492e-05,
+      "loss": 0.6282,
+      "step": 2003
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.07350806891918182,
+      "learning_rate": 8.248182343158706e-05,
+      "loss": 0.627,
+      "step": 2004
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.07304222136735916,
+      "learning_rate": 8.23955777557033e-05,
+      "loss": 0.668,
+      "step": 2005
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.06777157634496689,
+      "learning_rate": 8.230934559123739e-05,
+      "loss": 0.5669,
+      "step": 2006
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.06221771612763405,
+      "learning_rate": 8.22231270043726e-05,
+      "loss": 0.5628,
+      "step": 2007
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.06386017054319382,
+      "learning_rate": 8.213692206128178e-05,
+      "loss": 0.576,
+      "step": 2008
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.06762038916349411,
+      "learning_rate": 8.205073082812737e-05,
+      "loss": 0.536,
+      "step": 2009
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.06616533547639847,
+      "learning_rate": 8.196455337106127e-05,
+      "loss": 0.5657,
+      "step": 2010
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.06463898718357086,
+      "learning_rate": 8.187838975622478e-05,
+      "loss": 0.5349,
+      "step": 2011
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.06426985561847687,
+      "learning_rate": 8.179224004974857e-05,
+      "loss": 0.5844,
+      "step": 2012
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.07058148086071014,
+      "learning_rate": 8.170610431775267e-05,
+      "loss": 0.6695,
+      "step": 2013
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.07236622273921967,
+      "learning_rate": 8.161998262634637e-05,
+      "loss": 0.6522,
+      "step": 2014
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.06184220314025879,
+      "learning_rate": 8.15338750416282e-05,
+      "loss": 0.5503,
+      "step": 2015
+    },
+    {
+      "epoch": 1.1,
+      "grad_norm": 0.07017958909273148,
+      "learning_rate": 8.144778162968584e-05,
+      "loss": 0.6168,
+      "step": 2016
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.0607057586312294,
+      "learning_rate": 8.136170245659609e-05,
+      "loss": 0.5648,
+      "step": 2017
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06027553975582123,
+      "learning_rate": 8.127563758842483e-05,
+      "loss": 0.5837,
+      "step": 2018
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06839162111282349,
+      "learning_rate": 8.118958709122698e-05,
+      "loss": 0.7024,
+      "step": 2019
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.0698401927947998,
+      "learning_rate": 8.110355103104641e-05,
+      "loss": 0.629,
+      "step": 2020
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06260319799184799,
+      "learning_rate": 8.101752947391588e-05,
+      "loss": 0.5965,
+      "step": 2021
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06395679712295532,
+      "learning_rate": 8.09315224858571e-05,
+      "loss": 0.526,
+      "step": 2022
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06215297058224678,
+      "learning_rate": 8.084553013288048e-05,
+      "loss": 0.5134,
+      "step": 2023
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06511794030666351,
+      "learning_rate": 8.075955248098536e-05,
+      "loss": 0.5383,
+      "step": 2024
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06743456423282623,
+      "learning_rate": 8.067358959615963e-05,
+      "loss": 0.6318,
+      "step": 2025
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.07108740508556366,
+      "learning_rate": 8.058764154437996e-05,
+      "loss": 0.6597,
+      "step": 2026
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06761852651834488,
+      "learning_rate": 8.050170839161157e-05,
+      "loss": 0.6172,
+      "step": 2027
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06814343482255936,
+      "learning_rate": 8.04157902038083e-05,
+      "loss": 0.5837,
+      "step": 2028
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06850943714380264,
+      "learning_rate": 8.032988704691243e-05,
+      "loss": 0.5799,
+      "step": 2029
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06714814901351929,
+      "learning_rate": 8.02439989868548e-05,
+      "loss": 0.5786,
+      "step": 2030
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06138110160827637,
+      "learning_rate": 8.015812608955457e-05,
+      "loss": 0.5518,
+      "step": 2031
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06873062998056412,
+      "learning_rate": 8.007226842091931e-05,
+      "loss": 0.5336,
+      "step": 2032
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.06783683598041534,
+      "learning_rate": 7.998642604684491e-05,
+      "loss": 0.61,
+      "step": 2033
+    },
+    {
+      "epoch": 1.11,
+      "grad_norm": 0.0697760209441185,
+      "learning_rate": 7.990059903321553e-05,
+      "loss": 0.603,
+      "step": 2034
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.062311042100191116,
+      "learning_rate": 7.981478744590348e-05,
+      "loss": 0.5093,
+      "step": 2035
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.07325945049524307,
+      "learning_rate": 7.972899135076929e-05,
+      "loss": 0.6595,
+      "step": 2036
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.06448288261890411,
+      "learning_rate": 7.964321081366157e-05,
+      "loss": 0.5403,
+      "step": 2037
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.06438078731298447,
+      "learning_rate": 7.955744590041701e-05,
+      "loss": 0.57,
+      "step": 2038
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.06576252728700638,
+      "learning_rate": 7.947169667686027e-05,
+      "loss": 0.6023,
+      "step": 2039
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.07431968301534653,
+      "learning_rate": 7.938596320880402e-05,
+      "loss": 0.6057,
+      "step": 2040
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.07968160510063171,
+      "learning_rate": 7.93002455620488e-05,
+      "loss": 0.6573,
+      "step": 2041
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.06240157037973404,
+      "learning_rate": 7.9214543802383e-05,
+      "loss": 0.5845,
+      "step": 2042
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.06589362770318985,
+      "learning_rate": 7.912885799558288e-05,
+      "loss": 0.5057,
+      "step": 2043
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.06928303092718124,
+      "learning_rate": 7.904318820741239e-05,
+      "loss": 0.5991,
+      "step": 2044
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.06346700340509415,
+      "learning_rate": 7.89575345036232e-05,
+      "loss": 0.5788,
+      "step": 2045
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.06340914219617844,
+      "learning_rate": 7.887189694995466e-05,
+      "loss": 0.544,
+      "step": 2046
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.06660696119070053,
+      "learning_rate": 7.878627561213369e-05,
+      "loss": 0.6379,
+      "step": 2047
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.0683973878622055,
+      "learning_rate": 7.870067055587481e-05,
+      "loss": 0.5923,
+      "step": 2048
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.06100190058350563,
+      "learning_rate": 7.861508184688e-05,
+      "loss": 0.6059,
+      "step": 2049
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.059445932507514954,
+      "learning_rate": 7.85295095508387e-05,
+      "loss": 0.5476,
+      "step": 2050
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.07229001820087433,
+      "learning_rate": 7.844395373342779e-05,
+      "loss": 0.6259,
+      "step": 2051
+    },
+    {
+      "epoch": 1.12,
+      "grad_norm": 0.06739497929811478,
+      "learning_rate": 7.835841446031143e-05,
+      "loss": 0.5855,
+      "step": 2052
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.06135958433151245,
+      "learning_rate": 7.827289179714118e-05,
+      "loss": 0.5265,
+      "step": 2053
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.07160507887601852,
+      "learning_rate": 7.818738580955576e-05,
+      "loss": 0.6323,
+      "step": 2054
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.060510244220495224,
+      "learning_rate": 7.810189656318112e-05,
+      "loss": 0.4678,
+      "step": 2055
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.06444878876209259,
+      "learning_rate": 7.801642412363041e-05,
+      "loss": 0.6195,
+      "step": 2056
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.06845416873693466,
+      "learning_rate": 7.793096855650385e-05,
+      "loss": 0.6249,
+      "step": 2057
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.06821136176586151,
+      "learning_rate": 7.784552992738867e-05,
+      "loss": 0.6733,
+      "step": 2058
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.07368767261505127,
+      "learning_rate": 7.776010830185914e-05,
+      "loss": 0.5954,
+      "step": 2059
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.06696832925081253,
+      "learning_rate": 7.767470374547647e-05,
+      "loss": 0.661,
+      "step": 2060
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.07673228532075882,
+      "learning_rate": 7.75893163237888e-05,
+      "loss": 0.6957,
+      "step": 2061
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.06611277163028717,
+      "learning_rate": 7.750394610233106e-05,
+      "loss": 0.5931,
+      "step": 2062
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.06206559017300606,
+      "learning_rate": 7.741859314662502e-05,
+      "loss": 0.5773,
+      "step": 2063
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.06988830119371414,
+      "learning_rate": 7.733325752217917e-05,
+      "loss": 0.6255,
+      "step": 2064
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.07555034756660461,
+      "learning_rate": 7.724793929448875e-05,
+      "loss": 0.6309,
+      "step": 2065
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.06582862883806229,
+      "learning_rate": 7.716263852903561e-05,
+      "loss": 0.5482,
+      "step": 2066
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.06785890460014343,
+      "learning_rate": 7.707735529128819e-05,
+      "loss": 0.562,
+      "step": 2067
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.062424518167972565,
+      "learning_rate": 7.699208964670149e-05,
+      "loss": 0.5801,
+      "step": 2068
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.07258402556180954,
+      "learning_rate": 7.690684166071702e-05,
+      "loss": 0.6196,
+      "step": 2069
+    },
+    {
+      "epoch": 1.13,
+      "grad_norm": 0.06637036055326462,
+      "learning_rate": 7.68216113987627e-05,
+      "loss": 0.5658,
+      "step": 2070
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.06919743120670319,
+      "learning_rate": 7.673639892625289e-05,
+      "loss": 0.5619,
+      "step": 2071
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.06857898831367493,
+      "learning_rate": 7.665120430858829e-05,
+      "loss": 0.7035,
+      "step": 2072
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.0656585618853569,
+      "learning_rate": 7.656602761115583e-05,
+      "loss": 0.6331,
+      "step": 2073
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.0701858177781105,
+      "learning_rate": 7.648086889932879e-05,
+      "loss": 0.6627,
+      "step": 2074
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.06948113441467285,
+      "learning_rate": 7.63957282384666e-05,
+      "loss": 0.6194,
+      "step": 2075
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.07097858935594559,
+      "learning_rate": 7.631060569391482e-05,
+      "loss": 0.6377,
+      "step": 2076
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.0637185126543045,
+      "learning_rate": 7.62255013310051e-05,
+      "loss": 0.5771,
+      "step": 2077
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.0651545375585556,
+      "learning_rate": 7.614041521505517e-05,
+      "loss": 0.6193,
+      "step": 2078
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.07133258879184723,
+      "learning_rate": 7.605534741136873e-05,
+      "loss": 0.5815,
+      "step": 2079
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.0651950016617775,
+      "learning_rate": 7.597029798523545e-05,
+      "loss": 0.5649,
+      "step": 2080
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.06561518460512161,
+      "learning_rate": 7.588526700193086e-05,
+      "loss": 0.5826,
+      "step": 2081
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.06808426976203918,
+      "learning_rate": 7.580025452671637e-05,
+      "loss": 0.6098,
+      "step": 2082
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.0729101300239563,
+      "learning_rate": 7.571526062483912e-05,
+      "loss": 0.6483,
+      "step": 2083
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.07908526808023453,
+      "learning_rate": 7.563028536153213e-05,
+      "loss": 0.6949,
+      "step": 2084
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.06607040762901306,
+      "learning_rate": 7.554532880201399e-05,
+      "loss": 0.5709,
+      "step": 2085
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.06513672322034836,
+      "learning_rate": 7.546039101148895e-05,
+      "loss": 0.5823,
+      "step": 2086
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.0633048489689827,
+      "learning_rate": 7.53754720551469e-05,
+      "loss": 0.554,
+      "step": 2087
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 0.07791243493556976,
+      "learning_rate": 7.529057199816326e-05,
+      "loss": 0.6958,
+      "step": 2088
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.07528932392597198,
+      "learning_rate": 7.520569090569893e-05,
+      "loss": 0.643,
+      "step": 2089
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.06492157280445099,
+      "learning_rate": 7.512082884290026e-05,
+      "loss": 0.575,
+      "step": 2090
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.07389669865369797,
+      "learning_rate": 7.5035985874899e-05,
+      "loss": 0.6689,
+      "step": 2091
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.0760781541466713,
+      "learning_rate": 7.495116206681224e-05,
+      "loss": 0.6634,
+      "step": 2092
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.06599968671798706,
+      "learning_rate": 7.486635748374237e-05,
+      "loss": 0.5395,
+      "step": 2093
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.07587437331676483,
+      "learning_rate": 7.478157219077703e-05,
+      "loss": 0.6457,
+      "step": 2094
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.0665985643863678,
+      "learning_rate": 7.469680625298903e-05,
+      "loss": 0.5487,
+      "step": 2095
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.06659691780805588,
+      "learning_rate": 7.461205973543636e-05,
+      "loss": 0.4887,
+      "step": 2096
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.06467236578464508,
+      "learning_rate": 7.452733270316209e-05,
+      "loss": 0.5376,
+      "step": 2097
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.07357700169086456,
+      "learning_rate": 7.444262522119428e-05,
+      "loss": 0.6996,
+      "step": 2098
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.06766042858362198,
+      "learning_rate": 7.435793735454611e-05,
+      "loss": 0.5223,
+      "step": 2099
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.07042094320058823,
+      "learning_rate": 7.427326916821557e-05,
+      "loss": 0.5802,
+      "step": 2100
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.0585549958050251,
+      "learning_rate": 7.418862072718562e-05,
+      "loss": 0.5306,
+      "step": 2101
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.06087832525372505,
+      "learning_rate": 7.410399209642409e-05,
+      "loss": 0.5706,
+      "step": 2102
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.06485297530889511,
+      "learning_rate": 7.401938334088356e-05,
+      "loss": 0.5307,
+      "step": 2103
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.07619531452655792,
+      "learning_rate": 7.393479452550133e-05,
+      "loss": 0.5977,
+      "step": 2104
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.06514798104763031,
+      "learning_rate": 7.385022571519947e-05,
+      "loss": 0.5945,
+      "step": 2105
+    },
+    {
+      "epoch": 1.15,
+      "grad_norm": 0.06858575344085693,
+      "learning_rate": 7.376567697488461e-05,
+      "loss": 0.6253,
+      "step": 2106
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.06849953532218933,
+      "learning_rate": 7.368114836944805e-05,
+      "loss": 0.5536,
+      "step": 2107
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.0654776319861412,
+      "learning_rate": 7.35966399637656e-05,
+      "loss": 0.5465,
+      "step": 2108
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.06670403480529785,
+      "learning_rate": 7.35121518226976e-05,
+      "loss": 0.5944,
+      "step": 2109
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.06398824602365494,
+      "learning_rate": 7.342768401108876e-05,
+      "loss": 0.5662,
+      "step": 2110
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.06844029575586319,
+      "learning_rate": 7.334323659376829e-05,
+      "loss": 0.5791,
+      "step": 2111
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.06193400174379349,
+      "learning_rate": 7.325880963554969e-05,
+      "loss": 0.4949,
+      "step": 2112
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.061342112720012665,
+      "learning_rate": 7.317440320123075e-05,
+      "loss": 0.5043,
+      "step": 2113
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.06053828075528145,
+      "learning_rate": 7.309001735559349e-05,
+      "loss": 0.5127,
+      "step": 2114
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.07439390569925308,
+      "learning_rate": 7.300565216340422e-05,
+      "loss": 0.5662,
+      "step": 2115
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.07639770209789276,
+      "learning_rate": 7.292130768941332e-05,
+      "loss": 0.5874,
+      "step": 2116
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.06395827233791351,
+      "learning_rate": 7.283698399835528e-05,
+      "loss": 0.5762,
+      "step": 2117
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.07043325901031494,
+      "learning_rate": 7.275268115494866e-05,
+      "loss": 0.6473,
+      "step": 2118
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.07612812519073486,
+      "learning_rate": 7.266839922389598e-05,
+      "loss": 0.6374,
+      "step": 2119
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.06510752439498901,
+      "learning_rate": 7.258413826988376e-05,
+      "loss": 0.5649,
+      "step": 2120
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.071219302713871,
+      "learning_rate": 7.249989835758239e-05,
+      "loss": 0.6823,
+      "step": 2121
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.07150042057037354,
+      "learning_rate": 7.24156795516461e-05,
+      "loss": 0.6678,
+      "step": 2122
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.07282208651304245,
+      "learning_rate": 7.233148191671293e-05,
+      "loss": 0.6312,
+      "step": 2123
+    },
+    {
+      "epoch": 1.16,
+      "grad_norm": 0.06840363144874573,
+      "learning_rate": 7.224730551740472e-05,
+      "loss": 0.5773,
+      "step": 2124
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.07560744881629944,
+      "learning_rate": 7.216315041832696e-05,
+      "loss": 0.5611,
+      "step": 2125
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.0657825618982315,
+      "learning_rate": 7.207901668406878e-05,
+      "loss": 0.5698,
+      "step": 2126
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.07382172346115112,
+      "learning_rate": 7.199490437920294e-05,
+      "loss": 0.6428,
+      "step": 2127
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.07186535745859146,
+      "learning_rate": 7.191081356828575e-05,
+      "loss": 0.6556,
+      "step": 2128
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.0643572062253952,
+      "learning_rate": 7.182674431585704e-05,
+      "loss": 0.5022,
+      "step": 2129
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.06504008919000626,
+      "learning_rate": 7.174269668644004e-05,
+      "loss": 0.5544,
+      "step": 2130
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.06553123146295547,
+      "learning_rate": 7.165867074454145e-05,
+      "loss": 0.5749,
+      "step": 2131
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.07009954005479813,
+      "learning_rate": 7.157466655465125e-05,
+      "loss": 0.5924,
+      "step": 2132
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.058126140385866165,
+      "learning_rate": 7.149068418124281e-05,
+      "loss": 0.505,
+      "step": 2133
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.06979527324438095,
+      "learning_rate": 7.14067236887727e-05,
+      "loss": 0.5521,
+      "step": 2134
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.07132134586572647,
+      "learning_rate": 7.132278514168073e-05,
+      "loss": 0.6489,
+      "step": 2135
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.07470317929983139,
+      "learning_rate": 7.123886860438984e-05,
+      "loss": 0.6272,
+      "step": 2136
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.0786786749958992,
+      "learning_rate": 7.115497414130606e-05,
+      "loss": 0.6775,
+      "step": 2137
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.07232896238565445,
+      "learning_rate": 7.107110181681852e-05,
+      "loss": 0.6154,
+      "step": 2138
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.06931086629629135,
+      "learning_rate": 7.098725169529936e-05,
+      "loss": 0.6516,
+      "step": 2139
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.06935492902994156,
+      "learning_rate": 7.090342384110365e-05,
+      "loss": 0.6855,
+      "step": 2140
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.08043521642684937,
+      "learning_rate": 7.081961831856936e-05,
+      "loss": 0.6556,
+      "step": 2141
+    },
+    {
+      "epoch": 1.17,
+      "grad_norm": 0.06161235272884369,
+      "learning_rate": 7.073583519201734e-05,
+      "loss": 0.5758,
+      "step": 2142
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.09204084426164627,
+      "learning_rate": 7.06520745257513e-05,
+      "loss": 0.6648,
+      "step": 2143
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.06960479170084,
+      "learning_rate": 7.056833638405762e-05,
+      "loss": 0.6028,
+      "step": 2144
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.07587411999702454,
+      "learning_rate": 7.048462083120547e-05,
+      "loss": 0.702,
+      "step": 2145
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.07557187974452972,
+      "learning_rate": 7.04009279314466e-05,
+      "loss": 0.6113,
+      "step": 2146
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.06679632514715195,
+      "learning_rate": 7.031725774901547e-05,
+      "loss": 0.6169,
+      "step": 2147
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.07555227726697922,
+      "learning_rate": 7.023361034812906e-05,
+      "loss": 0.6971,
+      "step": 2148
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.06371418386697769,
+      "learning_rate": 7.014998579298683e-05,
+      "loss": 0.4516,
+      "step": 2149
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.06644103676080704,
+      "learning_rate": 7.006638414777075e-05,
+      "loss": 0.5879,
+      "step": 2150
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.07008033245801926,
+      "learning_rate": 6.99828054766452e-05,
+      "loss": 0.5955,
+      "step": 2151
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.07703608274459839,
+      "learning_rate": 6.989924984375692e-05,
+      "loss": 0.6454,
+      "step": 2152
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.07317256182432175,
+      "learning_rate": 6.981571731323497e-05,
+      "loss": 0.673,
+      "step": 2153
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.06549296528100967,
+      "learning_rate": 6.97322079491907e-05,
+      "loss": 0.5843,
+      "step": 2154
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.06609703600406647,
+      "learning_rate": 6.964872181571764e-05,
+      "loss": 0.5097,
+      "step": 2155
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.06859977543354034,
+      "learning_rate": 6.956525897689152e-05,
+      "loss": 0.5882,
+      "step": 2156
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.07044606655836105,
+      "learning_rate": 6.948181949677015e-05,
+      "loss": 0.6284,
+      "step": 2157
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.06426123529672623,
+      "learning_rate": 6.93984034393935e-05,
+      "loss": 0.5967,
+      "step": 2158
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.06794054806232452,
+      "learning_rate": 6.931501086878345e-05,
+      "loss": 0.6094,
+      "step": 2159
+    },
+    {
+      "epoch": 1.18,
+      "grad_norm": 0.06806287169456482,
+      "learning_rate": 6.923164184894391e-05,
+      "loss": 0.6135,
+      "step": 2160
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.07279844582080841,
+      "learning_rate": 6.914829644386077e-05,
+      "loss": 0.6009,
+      "step": 2161
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.07700279355049133,
+      "learning_rate": 6.906497471750171e-05,
+      "loss": 0.6697,
+      "step": 2162
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.0661427304148674,
+      "learning_rate": 6.898167673381627e-05,
+      "loss": 0.5603,
+      "step": 2163
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.08361556380987167,
+      "learning_rate": 6.889840255673577e-05,
+      "loss": 0.7131,
+      "step": 2164
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.07225959002971649,
+      "learning_rate": 6.881515225017323e-05,
+      "loss": 0.6105,
+      "step": 2165
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.06336675584316254,
+      "learning_rate": 6.87319258780234e-05,
+      "loss": 0.5211,
+      "step": 2166
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.06372500211000443,
+      "learning_rate": 6.86487235041626e-05,
+      "loss": 0.5295,
+      "step": 2167
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.07312193512916565,
+      "learning_rate": 6.85655451924488e-05,
+      "loss": 0.625,
+      "step": 2168
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.06828183680772781,
+      "learning_rate": 6.848239100672145e-05,
+      "loss": 0.6262,
+      "step": 2169
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.06808409094810486,
+      "learning_rate": 6.839926101080148e-05,
+      "loss": 0.6716,
+      "step": 2170
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.06251493841409683,
+      "learning_rate": 6.831615526849133e-05,
+      "loss": 0.5642,
+      "step": 2171
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.06784352660179138,
+      "learning_rate": 6.823307384357471e-05,
+      "loss": 0.5771,
+      "step": 2172
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.07426799833774567,
+      "learning_rate": 6.815001679981678e-05,
+      "loss": 0.5582,
+      "step": 2173
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.06320761144161224,
+      "learning_rate": 6.806698420096387e-05,
+      "loss": 0.6032,
+      "step": 2174
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.06084798276424408,
+      "learning_rate": 6.798397611074367e-05,
+      "loss": 0.5677,
+      "step": 2175
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.07241813093423843,
+      "learning_rate": 6.790099259286497e-05,
+      "loss": 0.6071,
+      "step": 2176
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.0648108497262001,
+      "learning_rate": 6.781803371101774e-05,
+      "loss": 0.5046,
+      "step": 2177
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 0.0848846361041069,
+      "learning_rate": 6.773509952887302e-05,
+      "loss": 0.6551,
+      "step": 2178
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.06519858539104462,
+      "learning_rate": 6.76521901100829e-05,
+      "loss": 0.5732,
+      "step": 2179
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.06119070202112198,
+      "learning_rate": 6.756930551828052e-05,
+      "loss": 0.5153,
+      "step": 2180
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.0737023800611496,
+      "learning_rate": 6.748644581707988e-05,
+      "loss": 0.6208,
+      "step": 2181
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.07560352236032486,
+      "learning_rate": 6.74036110700759e-05,
+      "loss": 0.6223,
+      "step": 2182
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.07263612747192383,
+      "learning_rate": 6.732080134084435e-05,
+      "loss": 0.6347,
+      "step": 2183
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.06753487884998322,
+      "learning_rate": 6.723801669294189e-05,
+      "loss": 0.5513,
+      "step": 2184
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.07577301561832428,
+      "learning_rate": 6.715525718990578e-05,
+      "loss": 0.7817,
+      "step": 2185
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.06963416188955307,
+      "learning_rate": 6.707252289525407e-05,
+      "loss": 0.5903,
+      "step": 2186
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.07357131689786911,
+      "learning_rate": 6.698981387248544e-05,
+      "loss": 0.6524,
+      "step": 2187
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.0715678483247757,
+      "learning_rate": 6.690713018507918e-05,
+      "loss": 0.6235,
+      "step": 2188
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.07252723723649979,
+      "learning_rate": 6.68244718964951e-05,
+      "loss": 0.537,
+      "step": 2189
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.06699419021606445,
+      "learning_rate": 6.67418390701736e-05,
+      "loss": 0.6012,
+      "step": 2190
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.07284466177225113,
+      "learning_rate": 6.665923176953546e-05,
+      "loss": 0.6235,
+      "step": 2191
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.06402984261512756,
+      "learning_rate": 6.657665005798186e-05,
+      "loss": 0.5386,
+      "step": 2192
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.07105006277561188,
+      "learning_rate": 6.649409399889443e-05,
+      "loss": 0.5789,
+      "step": 2193
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.07561016082763672,
+      "learning_rate": 6.641156365563504e-05,
+      "loss": 0.5687,
+      "step": 2194
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.06907906383275986,
+      "learning_rate": 6.632905909154583e-05,
+      "loss": 0.6156,
+      "step": 2195
+    },
+    {
+      "epoch": 1.2,
+      "grad_norm": 0.06653522700071335,
+      "learning_rate": 6.624658036994918e-05,
+      "loss": 0.6062,
+      "step": 2196
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.0678929015994072,
+      "learning_rate": 6.616412755414761e-05,
+      "loss": 0.6009,
+      "step": 2197
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.062168996781110764,
+      "learning_rate": 6.608170070742377e-05,
+      "loss": 0.576,
+      "step": 2198
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.06343018263578415,
+      "learning_rate": 6.599929989304035e-05,
+      "loss": 0.5534,
+      "step": 2199
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.06593502312898636,
+      "learning_rate": 6.591692517424013e-05,
+      "loss": 0.5599,
+      "step": 2200
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.0697464793920517,
+      "learning_rate": 6.583457661424576e-05,
+      "loss": 0.6572,
+      "step": 2201
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.06897808611392975,
+      "learning_rate": 6.575225427625995e-05,
+      "loss": 0.6817,
+      "step": 2202
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.07382930815219879,
+      "learning_rate": 6.566995822346516e-05,
+      "loss": 0.62,
+      "step": 2203
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.06902208924293518,
+      "learning_rate": 6.55876885190237e-05,
+      "loss": 0.578,
+      "step": 2204
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.06347808241844177,
+      "learning_rate": 6.550544522607773e-05,
+      "loss": 0.5382,
+      "step": 2205
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.07611478865146637,
+      "learning_rate": 6.542322840774905e-05,
+      "loss": 0.657,
+      "step": 2206
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.06822540611028671,
+      "learning_rate": 6.534103812713919e-05,
+      "loss": 0.5694,
+      "step": 2207
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.08152306824922562,
+      "learning_rate": 6.525887444732932e-05,
+      "loss": 0.5875,
+      "step": 2208
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.06699424982070923,
+      "learning_rate": 6.517673743138015e-05,
+      "loss": 0.55,
+      "step": 2209
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.06906050443649292,
+      "learning_rate": 6.509462714233195e-05,
+      "loss": 0.5825,
+      "step": 2210
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.06700971722602844,
+      "learning_rate": 6.501254364320446e-05,
+      "loss": 0.579,
+      "step": 2211
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.07808460295200348,
+      "learning_rate": 6.493048699699693e-05,
+      "loss": 0.6068,
+      "step": 2212
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.0653437152504921,
+      "learning_rate": 6.48484572666879e-05,
+      "loss": 0.512,
+      "step": 2213
+    },
+    {
+      "epoch": 1.21,
+      "grad_norm": 0.06242841109633446,
+      "learning_rate": 6.476645451523535e-05,
+      "loss": 0.4806,
+      "step": 2214
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.06796973198652267,
+      "learning_rate": 6.468447880557644e-05,
+      "loss": 0.6215,
+      "step": 2215
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.07036803662776947,
+      "learning_rate": 6.460253020062768e-05,
+      "loss": 0.6199,
+      "step": 2216
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.06817260384559631,
+      "learning_rate": 6.452060876328474e-05,
+      "loss": 0.5752,
+      "step": 2217
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.07045266032218933,
+      "learning_rate": 6.443871455642238e-05,
+      "loss": 0.583,
+      "step": 2218
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.06571897864341736,
+      "learning_rate": 6.435684764289457e-05,
+      "loss": 0.5529,
+      "step": 2219
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.06711366027593613,
+      "learning_rate": 6.427500808553424e-05,
+      "loss": 0.574,
+      "step": 2220
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.07405182719230652,
+      "learning_rate": 6.419319594715339e-05,
+      "loss": 0.5532,
+      "step": 2221
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.07331159710884094,
+      "learning_rate": 6.411141129054292e-05,
+      "loss": 0.5487,
+      "step": 2222
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.07121601700782776,
+      "learning_rate": 6.402965417847268e-05,
+      "loss": 0.5828,
+      "step": 2223
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.0719437301158905,
+      "learning_rate": 6.394792467369138e-05,
+      "loss": 0.6311,
+      "step": 2224
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.06829216331243515,
+      "learning_rate": 6.38662228389265e-05,
+      "loss": 0.6195,
+      "step": 2225
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.06692846864461899,
+      "learning_rate": 6.378454873688431e-05,
+      "loss": 0.5517,
+      "step": 2226
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.07758577913045883,
+      "learning_rate": 6.370290243024978e-05,
+      "loss": 0.6157,
+      "step": 2227
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.07376914471387863,
+      "learning_rate": 6.36212839816866e-05,
+      "loss": 0.6016,
+      "step": 2228
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.06754311174154282,
+      "learning_rate": 6.353969345383701e-05,
+      "loss": 0.6271,
+      "step": 2229
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.07988160848617554,
+      "learning_rate": 6.345813090932186e-05,
+      "loss": 0.5742,
+      "step": 2230
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.06557673960924149,
+      "learning_rate": 6.337659641074052e-05,
+      "loss": 0.5015,
+      "step": 2231
+    },
+    {
+      "epoch": 1.22,
+      "grad_norm": 0.0661676675081253,
+      "learning_rate": 6.32950900206708e-05,
+      "loss": 0.5408,
+      "step": 2232
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.06480776518583298,
+      "learning_rate": 6.3213611801669e-05,
+      "loss": 0.4707,
+      "step": 2233
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.0753173679113388,
+      "learning_rate": 6.313216181626974e-05,
+      "loss": 0.611,
+      "step": 2234
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.0757727399468422,
+      "learning_rate": 6.3050740126986e-05,
+      "loss": 0.5423,
+      "step": 2235
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.07315197587013245,
+      "learning_rate": 6.296934679630904e-05,
+      "loss": 0.5457,
+      "step": 2236
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.08855360001325607,
+      "learning_rate": 6.288798188670833e-05,
+      "loss": 0.7,
+      "step": 2237
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.06280341744422913,
+      "learning_rate": 6.280664546063157e-05,
+      "loss": 0.4838,
+      "step": 2238
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.06725641340017319,
+      "learning_rate": 6.272533758050457e-05,
+      "loss": 0.5553,
+      "step": 2239
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.07229411602020264,
+      "learning_rate": 6.264405830873125e-05,
+      "loss": 0.6519,
+      "step": 2240
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.06996262073516846,
+      "learning_rate": 6.256280770769354e-05,
+      "loss": 0.6112,
+      "step": 2241
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.0687815472483635,
+      "learning_rate": 6.248158583975141e-05,
+      "loss": 0.5465,
+      "step": 2242
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.07132161408662796,
+      "learning_rate": 6.240039276724272e-05,
+      "loss": 0.6383,
+      "step": 2243
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.07156707346439362,
+      "learning_rate": 6.231922855248329e-05,
+      "loss": 0.5916,
+      "step": 2244
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.06640041619539261,
+      "learning_rate": 6.223809325776677e-05,
+      "loss": 0.5626,
+      "step": 2245
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.06485997885465622,
+      "learning_rate": 6.215698694536456e-05,
+      "loss": 0.4876,
+      "step": 2246
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.07079490274190903,
+      "learning_rate": 6.20759096775259e-05,
+      "loss": 0.6112,
+      "step": 2247
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.06052457168698311,
+      "learning_rate": 6.199486151647773e-05,
+      "loss": 0.6048,
+      "step": 2248
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.06677879393100739,
+      "learning_rate": 6.191384252442458e-05,
+      "loss": 0.6207,
+      "step": 2249
+    },
+    {
+      "epoch": 1.23,
+      "grad_norm": 0.06265201419591904,
+      "learning_rate": 6.183285276354865e-05,
+      "loss": 0.5206,
+      "step": 2250
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.07499994337558746,
+      "learning_rate": 6.175189229600969e-05,
+      "loss": 0.6053,
+      "step": 2251
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.07424361258745193,
+      "learning_rate": 6.16709611839449e-05,
+      "loss": 0.6065,
+      "step": 2252
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.07379718124866486,
+      "learning_rate": 6.159005948946916e-05,
+      "loss": 0.5934,
+      "step": 2253
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.07687895745038986,
+      "learning_rate": 6.150918727467455e-05,
+      "loss": 0.6532,
+      "step": 2254
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.07283301651477814,
+      "learning_rate": 6.14283446016306e-05,
+      "loss": 0.5334,
+      "step": 2255
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.0724155604839325,
+      "learning_rate": 6.134753153238418e-05,
+      "loss": 0.5745,
+      "step": 2256
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.07714327424764633,
+      "learning_rate": 6.126674812895944e-05,
+      "loss": 0.6511,
+      "step": 2257
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.0738411694765091,
+      "learning_rate": 6.118599445335773e-05,
+      "loss": 0.5277,
+      "step": 2258
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.0703483298420906,
+      "learning_rate": 6.110527056755762e-05,
+      "loss": 0.5723,
+      "step": 2259
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.070271797478199,
+      "learning_rate": 6.102457653351479e-05,
+      "loss": 0.6019,
+      "step": 2260
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.08007260411977768,
+      "learning_rate": 6.0943912413161994e-05,
+      "loss": 0.4893,
+      "step": 2261
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.06550659239292145,
+      "learning_rate": 6.086327826840912e-05,
+      "loss": 0.5876,
+      "step": 2262
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.08582957834005356,
+      "learning_rate": 6.078267416114295e-05,
+      "loss": 0.7156,
+      "step": 2263
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.07212957739830017,
+      "learning_rate": 6.070210015322724e-05,
+      "loss": 0.5369,
+      "step": 2264
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.0679006278514862,
+      "learning_rate": 6.062155630650265e-05,
+      "loss": 0.5725,
+      "step": 2265
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.08907914161682129,
+      "learning_rate": 6.054104268278669e-05,
+      "loss": 0.6409,
+      "step": 2266
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.07050459831953049,
+      "learning_rate": 6.046055934387368e-05,
+      "loss": 0.5286,
+      "step": 2267
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 0.07432929426431656,
+      "learning_rate": 6.038010635153469e-05,
+      "loss": 0.6366,
+      "step": 2268
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.07176590710878372,
+      "learning_rate": 6.02996837675175e-05,
+      "loss": 0.6173,
+      "step": 2269
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.07120334357023239,
+      "learning_rate": 6.0219291653546536e-05,
+      "loss": 0.5726,
+      "step": 2270
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.07870449125766754,
+      "learning_rate": 6.0138930071322874e-05,
+      "loss": 0.5968,
+      "step": 2271
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.06898409128189087,
+      "learning_rate": 6.005859908252415e-05,
+      "loss": 0.6193,
+      "step": 2272
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.060963038355112076,
+      "learning_rate": 5.997829874880447e-05,
+      "loss": 0.5096,
+      "step": 2273
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.0753115639090538,
+      "learning_rate": 5.9898029131794485e-05,
+      "loss": 0.5989,
+      "step": 2274
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.06675612926483154,
+      "learning_rate": 5.9817790293101204e-05,
+      "loss": 0.5225,
+      "step": 2275
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.07885386049747467,
+      "learning_rate": 5.973758229430806e-05,
+      "loss": 0.5717,
+      "step": 2276
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.06998579949140549,
+      "learning_rate": 5.965740519697479e-05,
+      "loss": 0.5932,
+      "step": 2277
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.07550673931837082,
+      "learning_rate": 5.957725906263743e-05,
+      "loss": 0.7,
+      "step": 2278
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.06955604255199432,
+      "learning_rate": 5.9497143952808234e-05,
+      "loss": 0.5823,
+      "step": 2279
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.07412168383598328,
+      "learning_rate": 5.9417059928975686e-05,
+      "loss": 0.5776,
+      "step": 2280
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.07235770672559738,
+      "learning_rate": 5.933700705260437e-05,
+      "loss": 0.5945,
+      "step": 2281
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.08059383928775787,
+      "learning_rate": 5.9256985385134955e-05,
+      "loss": 0.6109,
+      "step": 2282
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.06218232214450836,
+      "learning_rate": 5.917699498798421e-05,
+      "loss": 0.5188,
+      "step": 2283
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.07667962461709976,
+      "learning_rate": 5.909703592254485e-05,
+      "loss": 0.6085,
+      "step": 2284
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.07095900923013687,
+      "learning_rate": 5.9017108250185584e-05,
+      "loss": 0.5872,
+      "step": 2285
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.07091958820819855,
+      "learning_rate": 5.8937212032251e-05,
+      "loss": 0.6289,
+      "step": 2286
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.06871485710144043,
+      "learning_rate": 5.885734733006154e-05,
+      "loss": 0.5561,
+      "step": 2287
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.06925121694803238,
+      "learning_rate": 5.877751420491346e-05,
+      "loss": 0.6291,
+      "step": 2288
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.07990697771310806,
+      "learning_rate": 5.869771271807885e-05,
+      "loss": 0.636,
+      "step": 2289
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.07092578709125519,
+      "learning_rate": 5.86179429308054e-05,
+      "loss": 0.615,
+      "step": 2290
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.07381773740053177,
+      "learning_rate": 5.8538204904316565e-05,
+      "loss": 0.6482,
+      "step": 2291
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.0735030248761177,
+      "learning_rate": 5.845849869981137e-05,
+      "loss": 0.5469,
+      "step": 2292
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.07174662500619888,
+      "learning_rate": 5.8378824378464434e-05,
+      "loss": 0.6268,
+      "step": 2293
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.06630636006593704,
+      "learning_rate": 5.82991820014259e-05,
+      "loss": 0.6448,
+      "step": 2294
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.07417547702789307,
+      "learning_rate": 5.821957162982143e-05,
+      "loss": 0.6635,
+      "step": 2295
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.07115903496742249,
+      "learning_rate": 5.813999332475206e-05,
+      "loss": 0.525,
+      "step": 2296
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.07035580277442932,
+      "learning_rate": 5.8060447147294285e-05,
+      "loss": 0.6121,
+      "step": 2297
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.06738772243261337,
+      "learning_rate": 5.798093315849984e-05,
+      "loss": 0.5151,
+      "step": 2298
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.06736040860414505,
+      "learning_rate": 5.790145141939588e-05,
+      "loss": 0.5185,
+      "step": 2299
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.07977905124425888,
+      "learning_rate": 5.7822001990984685e-05,
+      "loss": 0.6448,
+      "step": 2300
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.06976289302110672,
+      "learning_rate": 5.7742584934243894e-05,
+      "loss": 0.597,
+      "step": 2301
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.07432825863361359,
+      "learning_rate": 5.7663200310126084e-05,
+      "loss": 0.5596,
+      "step": 2302
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.06685048341751099,
+      "learning_rate": 5.7583848179559154e-05,
+      "loss": 0.617,
+      "step": 2303
+    },
+    {
+      "epoch": 1.26,
+      "grad_norm": 0.07461295276880264,
+      "learning_rate": 5.750452860344595e-05,
+      "loss": 0.5618,
+      "step": 2304
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.0821809321641922,
+      "learning_rate": 5.742524164266432e-05,
+      "loss": 0.746,
+      "step": 2305
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.07233615964651108,
+      "learning_rate": 5.734598735806718e-05,
+      "loss": 0.6114,
+      "step": 2306
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.07483091950416565,
+      "learning_rate": 5.7266765810482205e-05,
+      "loss": 0.5878,
+      "step": 2307
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.06762590259313583,
+      "learning_rate": 5.718757706071214e-05,
+      "loss": 0.5829,
+      "step": 2308
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.0718020498752594,
+      "learning_rate": 5.710842116953438e-05,
+      "loss": 0.5813,
+      "step": 2309
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.06753186881542206,
+      "learning_rate": 5.702929819770123e-05,
+      "loss": 0.5491,
+      "step": 2310
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.07231669872999191,
+      "learning_rate": 5.6950208205939626e-05,
+      "loss": 0.6531,
+      "step": 2311
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.07114548236131668,
+      "learning_rate": 5.687115125495127e-05,
+      "loss": 0.6119,
+      "step": 2312
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.08065725862979889,
+      "learning_rate": 5.679212740541253e-05,
+      "loss": 0.6962,
+      "step": 2313
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.067630834877491,
+      "learning_rate": 5.6713136717974226e-05,
+      "loss": 0.5528,
+      "step": 2314
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.0677843689918518,
+      "learning_rate": 5.6634179253261885e-05,
+      "loss": 0.6157,
+      "step": 2315
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.07071098685264587,
+      "learning_rate": 5.6555255071875415e-05,
+      "loss": 0.5523,
+      "step": 2316
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.07934273034334183,
+      "learning_rate": 5.647636423438929e-05,
+      "loss": 0.5646,
+      "step": 2317
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.08034515380859375,
+      "learning_rate": 5.639750680135227e-05,
+      "loss": 0.6589,
+      "step": 2318
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.07613985240459442,
+      "learning_rate": 5.6318682833287626e-05,
+      "loss": 0.5973,
+      "step": 2319
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.07653201371431351,
+      "learning_rate": 5.623989239069275e-05,
+      "loss": 0.6002,
+      "step": 2320
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.0726369246840477,
+      "learning_rate": 5.6161135534039476e-05,
+      "loss": 0.665,
+      "step": 2321
+    },
+    {
+      "epoch": 1.27,
+      "grad_norm": 0.06761615723371506,
+      "learning_rate": 5.60824123237738e-05,
+      "loss": 0.5183,
+      "step": 2322
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.0639297217130661,
+      "learning_rate": 5.6003722820315916e-05,
+      "loss": 0.5161,
+      "step": 2323
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.06320559233427048,
+      "learning_rate": 5.592506708406006e-05,
+      "loss": 0.5844,
+      "step": 2324
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.07683044672012329,
+      "learning_rate": 5.584644517537464e-05,
+      "loss": 0.6143,
+      "step": 2325
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.06600732356309891,
+      "learning_rate": 5.5767857154602135e-05,
+      "loss": 0.5548,
+      "step": 2326
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.07150949537754059,
+      "learning_rate": 5.568930308205886e-05,
+      "loss": 0.562,
+      "step": 2327
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.06904012709856033,
+      "learning_rate": 5.561078301803526e-05,
+      "loss": 0.5105,
+      "step": 2328
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.0722079873085022,
+      "learning_rate": 5.553229702279551e-05,
+      "loss": 0.6219,
+      "step": 2329
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.07431759685277939,
+      "learning_rate": 5.545384515657779e-05,
+      "loss": 0.6837,
+      "step": 2330
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.07866154611110687,
+      "learning_rate": 5.537542747959394e-05,
+      "loss": 0.5907,
+      "step": 2331
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.07409841567277908,
+      "learning_rate": 5.529704405202972e-05,
+      "loss": 0.5776,
+      "step": 2332
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.07749959826469421,
+      "learning_rate": 5.521869493404444e-05,
+      "loss": 0.5411,
+      "step": 2333
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.07008834928274155,
+      "learning_rate": 5.5140380185771154e-05,
+      "loss": 0.5368,
+      "step": 2334
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.06928420811891556,
+      "learning_rate": 5.506209986731662e-05,
+      "loss": 0.5241,
+      "step": 2335
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.07040610909461975,
+      "learning_rate": 5.4983854038760995e-05,
+      "loss": 0.6249,
+      "step": 2336
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.06945261359214783,
+      "learning_rate": 5.4905642760158124e-05,
+      "loss": 0.6035,
+      "step": 2337
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.06809067726135254,
+      "learning_rate": 5.4827466091535215e-05,
+      "loss": 0.6002,
+      "step": 2338
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.07596398890018463,
+      "learning_rate": 5.474932409289302e-05,
+      "loss": 0.5523,
+      "step": 2339
+    },
+    {
+      "epoch": 1.28,
+      "grad_norm": 0.07113833725452423,
+      "learning_rate": 5.4671216824205575e-05,
+      "loss": 0.5829,
+      "step": 2340
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.07487820088863373,
+      "learning_rate": 5.4593144345420366e-05,
+      "loss": 0.659,
+      "step": 2341
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.07420715689659119,
+      "learning_rate": 5.451510671645807e-05,
+      "loss": 0.54,
+      "step": 2342
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.07427990436553955,
+      "learning_rate": 5.443710399721269e-05,
+      "loss": 0.6448,
+      "step": 2343
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.06675884127616882,
+      "learning_rate": 5.435913624755148e-05,
+      "loss": 0.4935,
+      "step": 2344
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.07301631569862366,
+      "learning_rate": 5.4281203527314696e-05,
+      "loss": 0.6389,
+      "step": 2345
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.06832318007946014,
+      "learning_rate": 5.42033058963159e-05,
+      "loss": 0.5719,
+      "step": 2346
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.08214893192052841,
+      "learning_rate": 5.4125443414341534e-05,
+      "loss": 0.6581,
+      "step": 2347
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.07999354600906372,
+      "learning_rate": 5.4047616141151255e-05,
+      "loss": 0.6648,
+      "step": 2348
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.06501716375350952,
+      "learning_rate": 5.39698241364775e-05,
+      "loss": 0.5674,
+      "step": 2349
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.06726286560297012,
+      "learning_rate": 5.389206746002584e-05,
+      "loss": 0.5106,
+      "step": 2350
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.08208262920379639,
+      "learning_rate": 5.381434617147452e-05,
+      "loss": 0.5974,
+      "step": 2351
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.07486006617546082,
+      "learning_rate": 5.37366603304748e-05,
+      "loss": 0.6769,
+      "step": 2352
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.075009286403656,
+      "learning_rate": 5.36590099966507e-05,
+      "loss": 0.6224,
+      "step": 2353
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.06720884144306183,
+      "learning_rate": 5.3581395229598887e-05,
+      "loss": 0.5017,
+      "step": 2354
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.0757581815123558,
+      "learning_rate": 5.350381608888885e-05,
+      "loss": 0.6254,
+      "step": 2355
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.06871607899665833,
+      "learning_rate": 5.3426272634062624e-05,
+      "loss": 0.5213,
+      "step": 2356
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.07220682501792908,
+      "learning_rate": 5.3348764924634983e-05,
+      "loss": 0.6204,
+      "step": 2357
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 0.09039363265037537,
+      "learning_rate": 5.3271293020093146e-05,
+      "loss": 0.6442,
+      "step": 2358
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.07300581783056259,
+      "learning_rate": 5.319385697989696e-05,
+      "loss": 0.6176,
+      "step": 2359
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.060775429010391235,
+      "learning_rate": 5.311645686347861e-05,
+      "loss": 0.482,
+      "step": 2360
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.06812864542007446,
+      "learning_rate": 5.3039092730242875e-05,
+      "loss": 0.5296,
+      "step": 2361
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.07054027915000916,
+      "learning_rate": 5.296176463956677e-05,
+      "loss": 0.5841,
+      "step": 2362
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.07183098793029785,
+      "learning_rate": 5.288447265079972e-05,
+      "loss": 0.5124,
+      "step": 2363
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.09338720142841339,
+      "learning_rate": 5.2807216823263484e-05,
+      "loss": 0.5999,
+      "step": 2364
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.07272778451442719,
+      "learning_rate": 5.2729997216251926e-05,
+      "loss": 0.6407,
+      "step": 2365
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.06850509345531464,
+      "learning_rate": 5.2652813889031294e-05,
+      "loss": 0.5894,
+      "step": 2366
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.0760633572936058,
+      "learning_rate": 5.257566690083979e-05,
+      "loss": 0.6144,
+      "step": 2367
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.07636122405529022,
+      "learning_rate": 5.249855631088794e-05,
+      "loss": 0.5995,
+      "step": 2368
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.06955023854970932,
+      "learning_rate": 5.2421482178358125e-05,
+      "loss": 0.6299,
+      "step": 2369
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.07433290779590607,
+      "learning_rate": 5.234444456240495e-05,
+      "loss": 0.6417,
+      "step": 2370
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.06652615964412689,
+      "learning_rate": 5.226744352215478e-05,
+      "loss": 0.5181,
+      "step": 2371
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.06942848861217499,
+      "learning_rate": 5.2190479116706073e-05,
+      "loss": 0.5557,
+      "step": 2372
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.06768972426652908,
+      "learning_rate": 5.2113551405129145e-05,
+      "loss": 0.5716,
+      "step": 2373
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.06542663276195526,
+      "learning_rate": 5.2036660446466045e-05,
+      "loss": 0.5762,
+      "step": 2374
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.07299809157848358,
+      "learning_rate": 5.1959806299730774e-05,
+      "loss": 0.6196,
+      "step": 2375
+    },
+    {
+      "epoch": 1.3,
+      "grad_norm": 0.0744137391448021,
+      "learning_rate": 5.1882989023908915e-05,
+      "loss": 0.6276,
+      "step": 2376
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.07640177011489868,
+      "learning_rate": 5.180620867795788e-05,
+      "loss": 0.539,
+      "step": 2377
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.0750659853219986,
+      "learning_rate": 5.1729465320806645e-05,
+      "loss": 0.5995,
+      "step": 2378
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.08161938935518265,
+      "learning_rate": 5.16527590113559e-05,
+      "loss": 0.6323,
+      "step": 2379
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.07055678963661194,
+      "learning_rate": 5.157608980847777e-05,
+      "loss": 0.5966,
+      "step": 2380
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.07690481096506119,
+      "learning_rate": 5.1499457771016e-05,
+      "loss": 0.7723,
+      "step": 2381
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.07799238711595535,
+      "learning_rate": 5.1422862957785834e-05,
+      "loss": 0.6716,
+      "step": 2382
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.07273693382740021,
+      "learning_rate": 5.1346305427573816e-05,
+      "loss": 0.5985,
+      "step": 2383
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.07442136853933334,
+      "learning_rate": 5.1269785239138015e-05,
+      "loss": 0.5927,
+      "step": 2384
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.06980384886264801,
+      "learning_rate": 5.1193302451207724e-05,
+      "loss": 0.5611,
+      "step": 2385
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.0675831288099289,
+      "learning_rate": 5.111685712248364e-05,
+      "loss": 0.5339,
+      "step": 2386
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.06704030185937881,
+      "learning_rate": 5.1040449311637605e-05,
+      "loss": 0.6157,
+      "step": 2387
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.07446156442165375,
+      "learning_rate": 5.0964079077312774e-05,
+      "loss": 0.546,
+      "step": 2388
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.07804377377033234,
+      "learning_rate": 5.0887746478123336e-05,
+      "loss": 0.6954,
+      "step": 2389
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.07373851537704468,
+      "learning_rate": 5.08114515726547e-05,
+      "loss": 0.5971,
+      "step": 2390
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.07430601865053177,
+      "learning_rate": 5.0735194419463304e-05,
+      "loss": 0.609,
+      "step": 2391
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.08051323890686035,
+      "learning_rate": 5.0658975077076667e-05,
+      "loss": 0.5799,
+      "step": 2392
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.07049138844013214,
+      "learning_rate": 5.058279360399314e-05,
+      "loss": 0.6887,
+      "step": 2393
+    },
+    {
+      "epoch": 1.31,
+      "grad_norm": 0.07234661281108856,
+      "learning_rate": 5.050665005868216e-05,
+      "loss": 0.551,
+      "step": 2394
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.06585332751274109,
+      "learning_rate": 5.043054449958404e-05,
+      "loss": 0.516,
+      "step": 2395
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.07594798505306244,
+      "learning_rate": 5.035447698510982e-05,
+      "loss": 0.6143,
+      "step": 2396
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.07606874406337738,
+      "learning_rate": 5.0278447573641495e-05,
+      "loss": 0.598,
+      "step": 2397
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.07295016199350357,
+      "learning_rate": 5.0202456323531667e-05,
+      "loss": 0.633,
+      "step": 2398
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.0775969848036766,
+      "learning_rate": 5.012650329310379e-05,
+      "loss": 0.6548,
+      "step": 2399
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.07147696614265442,
+      "learning_rate": 5.005058854065185e-05,
+      "loss": 0.5813,
+      "step": 2400
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.07123620808124542,
+      "learning_rate": 4.997471212444059e-05,
+      "loss": 0.6711,
+      "step": 2401
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.07753394544124603,
+      "learning_rate": 4.9898874102705204e-05,
+      "loss": 0.5952,
+      "step": 2402
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.07657833397388458,
+      "learning_rate": 4.9823074533651495e-05,
+      "loss": 0.5465,
+      "step": 2403
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.0730203315615654,
+      "learning_rate": 4.974731347545578e-05,
+      "loss": 0.5301,
+      "step": 2404
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.08199814707040787,
+      "learning_rate": 4.967159098626469e-05,
+      "loss": 0.6718,
+      "step": 2405
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.07243099808692932,
+      "learning_rate": 4.959590712419543e-05,
+      "loss": 0.62,
+      "step": 2406
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.06932693719863892,
+      "learning_rate": 4.9520261947335364e-05,
+      "loss": 0.5187,
+      "step": 2407
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.06874432414770126,
+      "learning_rate": 4.944465551374238e-05,
+      "loss": 0.5736,
+      "step": 2408
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.06884592771530151,
+      "learning_rate": 4.936908788144441e-05,
+      "loss": 0.5958,
+      "step": 2409
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.07854578644037247,
+      "learning_rate": 4.92935591084398e-05,
+      "loss": 0.6262,
+      "step": 2410
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.08149847388267517,
+      "learning_rate": 4.921806925269691e-05,
+      "loss": 0.6254,
+      "step": 2411
+    },
+    {
+      "epoch": 1.32,
+      "grad_norm": 0.07334032654762268,
+      "learning_rate": 4.914261837215435e-05,
+      "loss": 0.5486,
+      "step": 2412
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.0973329022526741,
+      "learning_rate": 4.90672065247208e-05,
+      "loss": 0.7698,
+      "step": 2413
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.06986068189144135,
+      "learning_rate": 4.899183376827487e-05,
+      "loss": 0.5833,
+      "step": 2414
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.07007471472024918,
+      "learning_rate": 4.891650016066536e-05,
+      "loss": 0.6134,
+      "step": 2415
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.07884490489959717,
+      "learning_rate": 4.884120575971082e-05,
+      "loss": 0.5585,
+      "step": 2416
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.07099177688360214,
+      "learning_rate": 4.876595062319987e-05,
+      "loss": 0.668,
+      "step": 2417
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.08233867585659027,
+      "learning_rate": 4.869073480889087e-05,
+      "loss": 0.5372,
+      "step": 2418
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.06042160093784332,
+      "learning_rate": 4.861555837451213e-05,
+      "loss": 0.5287,
+      "step": 2419
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.07258826494216919,
+      "learning_rate": 4.854042137776158e-05,
+      "loss": 0.5865,
+      "step": 2420
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.08246668428182602,
+      "learning_rate": 4.8465323876307024e-05,
+      "loss": 0.5417,
+      "step": 2421
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.07026161253452301,
+      "learning_rate": 4.8390265927785905e-05,
+      "loss": 0.5653,
+      "step": 2422
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.07703935354948044,
+      "learning_rate": 4.831524758980526e-05,
+      "loss": 0.6307,
+      "step": 2423
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.07275483757257462,
+      "learning_rate": 4.82402689199418e-05,
+      "loss": 0.5763,
+      "step": 2424
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.07985451072454453,
+      "learning_rate": 4.8165329975741715e-05,
+      "loss": 0.5512,
+      "step": 2425
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.06975404918193817,
+      "learning_rate": 4.80904308147208e-05,
+      "loss": 0.5738,
+      "step": 2426
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.08306935429573059,
+      "learning_rate": 4.801557149436419e-05,
+      "loss": 0.6353,
+      "step": 2427
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.06815838813781738,
+      "learning_rate": 4.794075207212659e-05,
+      "loss": 0.4735,
+      "step": 2428
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.07782094925642014,
+      "learning_rate": 4.786597260543194e-05,
+      "loss": 0.6385,
+      "step": 2429
+    },
+    {
+      "epoch": 1.33,
+      "grad_norm": 0.07282759994268417,
+      "learning_rate": 4.779123315167362e-05,
+      "loss": 0.4963,
+      "step": 2430
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.08053456246852875,
+      "learning_rate": 4.771653376821429e-05,
+      "loss": 0.6452,
+      "step": 2431
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.07558408379554749,
+      "learning_rate": 4.7641874512385745e-05,
+      "loss": 0.6229,
+      "step": 2432
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.07459430396556854,
+      "learning_rate": 4.7567255441489155e-05,
+      "loss": 0.5877,
+      "step": 2433
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.07865982502698898,
+      "learning_rate": 4.749267661279469e-05,
+      "loss": 0.5827,
+      "step": 2434
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.0782080590724945,
+      "learning_rate": 4.741813808354175e-05,
+      "loss": 0.5833,
+      "step": 2435
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.06451378017663956,
+      "learning_rate": 4.7343639910938695e-05,
+      "loss": 0.5107,
+      "step": 2436
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.06435713171958923,
+      "learning_rate": 4.726918215216305e-05,
+      "loss": 0.4768,
+      "step": 2437
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.06687245517969131,
+      "learning_rate": 4.7194764864361174e-05,
+      "loss": 0.533,
+      "step": 2438
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.0725407525897026,
+      "learning_rate": 4.712038810464847e-05,
+      "loss": 0.5419,
+      "step": 2439
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.06240203604102135,
+      "learning_rate": 4.704605193010922e-05,
+      "loss": 0.56,
+      "step": 2440
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.08678898215293884,
+      "learning_rate": 4.6971756397796504e-05,
+      "loss": 0.6051,
+      "step": 2441
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.07972444593906403,
+      "learning_rate": 4.689750156473228e-05,
+      "loss": 0.6196,
+      "step": 2442
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.08531578630208969,
+      "learning_rate": 4.6823287487907173e-05,
+      "loss": 0.6209,
+      "step": 2443
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.06963545829057693,
+      "learning_rate": 4.6749114224280664e-05,
+      "loss": 0.6311,
+      "step": 2444
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.08195082098245621,
+      "learning_rate": 4.667498183078076e-05,
+      "loss": 0.7213,
+      "step": 2445
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.07406240701675415,
+      "learning_rate": 4.660089036430424e-05,
+      "loss": 0.6393,
+      "step": 2446
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.08004584908485413,
+      "learning_rate": 4.652683988171632e-05,
+      "loss": 0.5596,
+      "step": 2447
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 0.0669763907790184,
+      "learning_rate": 4.645283043985095e-05,
+      "loss": 0.5218,
+      "step": 2448
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.08354940265417099,
+      "learning_rate": 4.637886209551038e-05,
+      "loss": 0.583,
+      "step": 2449
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.06934605538845062,
+      "learning_rate": 4.6304934905465445e-05,
+      "loss": 0.528,
+      "step": 2450
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.06799960881471634,
+      "learning_rate": 4.6231048926455415e-05,
+      "loss": 0.573,
+      "step": 2451
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.0804983377456665,
+      "learning_rate": 4.61572042151878e-05,
+      "loss": 0.6451,
+      "step": 2452
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.07285530120134354,
+      "learning_rate": 4.60834008283386e-05,
+      "loss": 0.5765,
+      "step": 2453
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.07094667106866837,
+      "learning_rate": 4.600963882255192e-05,
+      "loss": 0.5697,
+      "step": 2454
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.0664157122373581,
+      "learning_rate": 4.593591825444028e-05,
+      "loss": 0.6048,
+      "step": 2455
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.06729783862829208,
+      "learning_rate": 4.586223918058424e-05,
+      "loss": 0.594,
+      "step": 2456
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.07497063279151917,
+      "learning_rate": 4.578860165753268e-05,
+      "loss": 0.6829,
+      "step": 2457
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.08435964584350586,
+      "learning_rate": 4.57150057418024e-05,
+      "loss": 0.6255,
+      "step": 2458
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.07012388855218887,
+      "learning_rate": 4.5641451489878414e-05,
+      "loss": 0.5059,
+      "step": 2459
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.06958837062120438,
+      "learning_rate": 4.5567938958213704e-05,
+      "loss": 0.5567,
+      "step": 2460
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.07721869647502899,
+      "learning_rate": 4.549446820322929e-05,
+      "loss": 0.591,
+      "step": 2461
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.06777055561542511,
+      "learning_rate": 4.5421039281313985e-05,
+      "loss": 0.506,
+      "step": 2462
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.07227466255426407,
+      "learning_rate": 4.5347652248824624e-05,
+      "loss": 0.6017,
+      "step": 2463
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.07450714707374573,
+      "learning_rate": 4.5274307162085894e-05,
+      "loss": 0.5657,
+      "step": 2464
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.07071810960769653,
+      "learning_rate": 4.520100407739016e-05,
+      "loss": 0.5606,
+      "step": 2465
+    },
+    {
+      "epoch": 1.35,
+      "grad_norm": 0.07373818010091782,
+      "learning_rate": 4.512774305099775e-05,
+      "loss": 0.5771,
+      "step": 2466
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.0895610824227333,
+      "learning_rate": 4.505452413913649e-05,
+      "loss": 0.7296,
+      "step": 2467
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.0758524090051651,
+      "learning_rate": 4.498134739800207e-05,
+      "loss": 0.587,
+      "step": 2468
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.08151973783969879,
+      "learning_rate": 4.490821288375769e-05,
+      "loss": 0.5925,
+      "step": 2469
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.07796274870634079,
+      "learning_rate": 4.4835120652534235e-05,
+      "loss": 0.6123,
+      "step": 2470
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.07752220332622528,
+      "learning_rate": 4.476207076043003e-05,
+      "loss": 0.619,
+      "step": 2471
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.06401935964822769,
+      "learning_rate": 4.4689063263511e-05,
+      "loss": 0.5283,
+      "step": 2472
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.07612209767103195,
+      "learning_rate": 4.461609821781054e-05,
+      "loss": 0.5274,
+      "step": 2473
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.07507737725973129,
+      "learning_rate": 4.4543175679329344e-05,
+      "loss": 0.6352,
+      "step": 2474
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.0714292898774147,
+      "learning_rate": 4.447029570403561e-05,
+      "loss": 0.5376,
+      "step": 2475
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.08474813401699066,
+      "learning_rate": 4.4397458347864785e-05,
+      "loss": 0.658,
+      "step": 2476
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.08169824630022049,
+      "learning_rate": 4.432466366671968e-05,
+      "loss": 0.6571,
+      "step": 2477
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.0662381574511528,
+      "learning_rate": 4.425191171647025e-05,
+      "loss": 0.6264,
+      "step": 2478
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.07380590587854385,
+      "learning_rate": 4.417920255295379e-05,
+      "loss": 0.5699,
+      "step": 2479
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.08449859172105789,
+      "learning_rate": 4.4106536231974596e-05,
+      "loss": 0.5647,
+      "step": 2480
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.07241953909397125,
+      "learning_rate": 4.4033912809304214e-05,
+      "loss": 0.6136,
+      "step": 2481
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.07251517474651337,
+      "learning_rate": 4.396133234068126e-05,
+      "loss": 0.5484,
+      "step": 2482
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.08115177601575851,
+      "learning_rate": 4.388879488181125e-05,
+      "loss": 0.6351,
+      "step": 2483
+    },
+    {
+      "epoch": 1.36,
+      "grad_norm": 0.07590151578187943,
+      "learning_rate": 4.381630048836687e-05,
+      "loss": 0.5785,
+      "step": 2484
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.07244231551885605,
+      "learning_rate": 4.3743849215987595e-05,
+      "loss": 0.6151,
+      "step": 2485
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.07568778842687607,
+      "learning_rate": 4.3671441120279934e-05,
+      "loss": 0.5097,
+      "step": 2486
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.07233958691358566,
+      "learning_rate": 4.3599076256817125e-05,
+      "loss": 0.5486,
+      "step": 2487
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.08074472099542618,
+      "learning_rate": 4.3526754681139395e-05,
+      "loss": 0.626,
+      "step": 2488
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.06744801998138428,
+      "learning_rate": 4.3454476448753546e-05,
+      "loss": 0.5095,
+      "step": 2489
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.07199655473232269,
+      "learning_rate": 4.338224161513327e-05,
+      "loss": 0.5503,
+      "step": 2490
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.08650276809930801,
+      "learning_rate": 4.331005023571895e-05,
+      "loss": 0.6253,
+      "step": 2491
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.08837499469518661,
+      "learning_rate": 4.323790236591746e-05,
+      "loss": 0.5481,
+      "step": 2492
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.0681748315691948,
+      "learning_rate": 4.316579806110249e-05,
+      "loss": 0.5562,
+      "step": 2493
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.06990314275026321,
+      "learning_rate": 4.309373737661411e-05,
+      "loss": 0.5825,
+      "step": 2494
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.07371961325407028,
+      "learning_rate": 4.3021720367759056e-05,
+      "loss": 0.5226,
+      "step": 2495
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.07199500501155853,
+      "learning_rate": 4.294974708981041e-05,
+      "loss": 0.5349,
+      "step": 2496
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.08209903538227081,
+      "learning_rate": 4.287781759800784e-05,
+      "loss": 0.6039,
+      "step": 2497
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.07699441909790039,
+      "learning_rate": 4.280593194755727e-05,
+      "loss": 0.6163,
+      "step": 2498
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.06460884213447571,
+      "learning_rate": 4.273409019363103e-05,
+      "loss": 0.5296,
+      "step": 2499
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.07461009919643402,
+      "learning_rate": 4.266229239136783e-05,
+      "loss": 0.5694,
+      "step": 2500
+    },
+    {
+      "epoch": 1.37,
+      "grad_norm": 0.07356124371290207,
+      "learning_rate": 4.2590538595872495e-05,
+      "loss": 0.5508,
+      "step": 2501
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.07879485189914703,
+      "learning_rate": 4.251882886221623e-05,
+      "loss": 0.6915,
+      "step": 2502
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.07226796448230743,
+      "learning_rate": 4.2447163245436295e-05,
+      "loss": 0.5294,
+      "step": 2503
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.07331380248069763,
+      "learning_rate": 4.237554180053621e-05,
+      "loss": 0.6423,
+      "step": 2504
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.07386188954114914,
+      "learning_rate": 4.230396458248546e-05,
+      "loss": 0.627,
+      "step": 2505
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.06912330538034439,
+      "learning_rate": 4.223243164621973e-05,
+      "loss": 0.5624,
+      "step": 2506
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.06901656836271286,
+      "learning_rate": 4.216094304664056e-05,
+      "loss": 0.4814,
+      "step": 2507
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.06887569278478622,
+      "learning_rate": 4.208949883861559e-05,
+      "loss": 0.5264,
+      "step": 2508
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.07022027671337128,
+      "learning_rate": 4.20180990769784e-05,
+      "loss": 0.5159,
+      "step": 2509
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.06848796457052231,
+      "learning_rate": 4.194674381652831e-05,
+      "loss": 0.4915,
+      "step": 2510
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.07609882205724716,
+      "learning_rate": 4.187543311203066e-05,
+      "loss": 0.5331,
+      "step": 2511
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.06382356584072113,
+      "learning_rate": 4.180416701821643e-05,
+      "loss": 0.5464,
+      "step": 2512
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.0681033730506897,
+      "learning_rate": 4.173294558978253e-05,
+      "loss": 0.5528,
+      "step": 2513
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.07078997790813446,
+      "learning_rate": 4.1661768881391416e-05,
+      "loss": 0.5908,
+      "step": 2514
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.06293249875307083,
+      "learning_rate": 4.159063694767138e-05,
+      "loss": 0.5,
+      "step": 2515
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.08230625092983246,
+      "learning_rate": 4.151954984321622e-05,
+      "loss": 0.6461,
+      "step": 2516
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.06888657808303833,
+      "learning_rate": 4.1448507622585405e-05,
+      "loss": 0.5592,
+      "step": 2517
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.08879472315311432,
+      "learning_rate": 4.137751034030399e-05,
+      "loss": 0.6627,
+      "step": 2518
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 0.07641344517469406,
+      "learning_rate": 4.1306558050862384e-05,
+      "loss": 0.5746,
+      "step": 2519
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.08045165240764618,
+      "learning_rate": 4.1235650808716665e-05,
+      "loss": 0.6647,
+      "step": 2520
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.08117100596427917,
+      "learning_rate": 4.1164788668288155e-05,
+      "loss": 0.5934,
+      "step": 2521
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.07026588171720505,
+      "learning_rate": 4.1093971683963706e-05,
+      "loss": 0.5756,
+      "step": 2522
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.07604368031024933,
+      "learning_rate": 4.102319991009539e-05,
+      "loss": 0.5721,
+      "step": 2523
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.07874717563390732,
+      "learning_rate": 4.095247340100069e-05,
+      "loss": 0.5319,
+      "step": 2524
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.07756995409727097,
+      "learning_rate": 4.088179221096225e-05,
+      "loss": 0.5374,
+      "step": 2525
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.0727904736995697,
+      "learning_rate": 4.0811156394228046e-05,
+      "loss": 0.5539,
+      "step": 2526
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.07348831743001938,
+      "learning_rate": 4.074056600501107e-05,
+      "loss": 0.5754,
+      "step": 2527
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.08963430672883987,
+      "learning_rate": 4.067002109748961e-05,
+      "loss": 0.701,
+      "step": 2528
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.07955507934093475,
+      "learning_rate": 4.059952172580694e-05,
+      "loss": 0.63,
+      "step": 2529
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.06813513487577438,
+      "learning_rate": 4.0529067944071455e-05,
+      "loss": 0.5808,
+      "step": 2530
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.07727110385894775,
+      "learning_rate": 4.045865980635655e-05,
+      "loss": 0.6367,
+      "step": 2531
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.0674891322851181,
+      "learning_rate": 4.038829736670049e-05,
+      "loss": 0.5446,
+      "step": 2532
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.07553569972515106,
+      "learning_rate": 4.0317980679106613e-05,
+      "loss": 0.594,
+      "step": 2533
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.0750473290681839,
+      "learning_rate": 4.024770979754301e-05,
+      "loss": 0.5466,
+      "step": 2534
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.08206024020910263,
+      "learning_rate": 4.0177484775942755e-05,
+      "loss": 0.6497,
+      "step": 2535
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.06640418618917465,
+      "learning_rate": 4.010730566820355e-05,
+      "loss": 0.5366,
+      "step": 2536
+    },
+    {
+      "epoch": 1.39,
+      "grad_norm": 0.07963666319847107,
+      "learning_rate": 4.003717252818805e-05,
+      "loss": 0.6514,
+      "step": 2537
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.07602626830339432,
+      "learning_rate": 3.996708540972345e-05,
+      "loss": 0.5406,
+      "step": 2538
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.08422016352415085,
+      "learning_rate": 3.989704436660178e-05,
+      "loss": 0.6306,
+      "step": 2539
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.080567367374897,
+      "learning_rate": 3.982704945257957e-05,
+      "loss": 0.5657,
+      "step": 2540
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.07916505634784698,
+      "learning_rate": 3.975710072137805e-05,
+      "loss": 0.6178,
+      "step": 2541
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.08165088295936584,
+      "learning_rate": 3.968719822668299e-05,
+      "loss": 0.683,
+      "step": 2542
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.07973385602235794,
+      "learning_rate": 3.961734202214458e-05,
+      "loss": 0.6307,
+      "step": 2543
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.08140292018651962,
+      "learning_rate": 3.954753216137762e-05,
+      "loss": 0.5833,
+      "step": 2544
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.06821579486131668,
+      "learning_rate": 3.94777686979612e-05,
+      "loss": 0.4946,
+      "step": 2545
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.07947936654090881,
+      "learning_rate": 3.9408051685438975e-05,
+      "loss": 0.6955,
+      "step": 2546
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.0787663385272026,
+      "learning_rate": 3.933838117731873e-05,
+      "loss": 0.5568,
+      "step": 2547
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.0801193043589592,
+      "learning_rate": 3.9268757227072763e-05,
+      "loss": 0.5793,
+      "step": 2548
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.07403931766748428,
+      "learning_rate": 3.919917988813748e-05,
+      "loss": 0.5898,
+      "step": 2549
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.0729985162615776,
+      "learning_rate": 3.912964921391363e-05,
+      "loss": 0.5412,
+      "step": 2550
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.07448332756757736,
+      "learning_rate": 3.906016525776611e-05,
+      "loss": 0.5064,
+      "step": 2551
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.07552726566791534,
+      "learning_rate": 3.8990728073023906e-05,
+      "loss": 0.5618,
+      "step": 2552
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.07500220835208893,
+      "learning_rate": 3.89213377129802e-05,
+      "loss": 0.6134,
+      "step": 2553
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.07273518294095993,
+      "learning_rate": 3.885199423089212e-05,
+      "loss": 0.5164,
+      "step": 2554
+    },
+    {
+      "epoch": 1.4,
+      "grad_norm": 0.0652259886264801,
+      "learning_rate": 3.878269767998096e-05,
+      "loss": 0.541,
+      "step": 2555
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.07713053375482559,
+      "learning_rate": 3.871344811343184e-05,
+      "loss": 0.6068,
+      "step": 2556
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.08411411195993423,
+      "learning_rate": 3.8644245584393965e-05,
+      "loss": 0.6864,
+      "step": 2557
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.06960765272378922,
+      "learning_rate": 3.857509014598031e-05,
+      "loss": 0.4659,
+      "step": 2558
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.07402585446834564,
+      "learning_rate": 3.850598185126778e-05,
+      "loss": 0.53,
+      "step": 2559
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.08061912655830383,
+      "learning_rate": 3.843692075329714e-05,
+      "loss": 0.5711,
+      "step": 2560
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.07501520961523056,
+      "learning_rate": 3.8367906905072816e-05,
+      "loss": 0.5578,
+      "step": 2561
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.09120304882526398,
+      "learning_rate": 3.829894035956306e-05,
+      "loss": 0.6549,
+      "step": 2562
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.08057157695293427,
+      "learning_rate": 3.823002116969976e-05,
+      "loss": 0.4961,
+      "step": 2563
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.07594940811395645,
+      "learning_rate": 3.816114938837853e-05,
+      "loss": 0.5762,
+      "step": 2564
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.07835210114717484,
+      "learning_rate": 3.8092325068458486e-05,
+      "loss": 0.6856,
+      "step": 2565
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.07431218773126602,
+      "learning_rate": 3.802354826276248e-05,
+      "loss": 0.6213,
+      "step": 2566
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.06993934512138367,
+      "learning_rate": 3.795481902407672e-05,
+      "loss": 0.5887,
+      "step": 2567
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.07355695962905884,
+      "learning_rate": 3.7886137405151e-05,
+      "loss": 0.5146,
+      "step": 2568
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.08074449002742767,
+      "learning_rate": 3.7817503458698634e-05,
+      "loss": 0.6928,
+      "step": 2569
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.08331965655088425,
+      "learning_rate": 3.774891723739616e-05,
+      "loss": 0.6044,
+      "step": 2570
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.07481932640075684,
+      "learning_rate": 3.7680378793883696e-05,
+      "loss": 0.5267,
+      "step": 2571
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.08036099374294281,
+      "learning_rate": 3.76118881807645e-05,
+      "loss": 0.6552,
+      "step": 2572
+    },
+    {
+      "epoch": 1.41,
+      "grad_norm": 0.0766538679599762,
+      "learning_rate": 3.7543445450605285e-05,
+      "loss": 0.5748,
+      "step": 2573
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.08161585032939911,
+      "learning_rate": 3.747505065593586e-05,
+      "loss": 0.5854,
+      "step": 2574
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.070757195353508,
+      "learning_rate": 3.740670384924941e-05,
+      "loss": 0.4891,
+      "step": 2575
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.07526080310344696,
+      "learning_rate": 3.733840508300213e-05,
+      "loss": 0.6242,
+      "step": 2576
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.07664346694946289,
+      "learning_rate": 3.727015440961343e-05,
+      "loss": 0.6276,
+      "step": 2577
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.07076764851808548,
+      "learning_rate": 3.7201951881465846e-05,
+      "loss": 0.6141,
+      "step": 2578
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.08236127346754074,
+      "learning_rate": 3.7133797550904834e-05,
+      "loss": 0.484,
+      "step": 2579
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.061127692461013794,
+      "learning_rate": 3.7065691470239016e-05,
+      "loss": 0.4091,
+      "step": 2580
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.07331135123968124,
+      "learning_rate": 3.699763369173982e-05,
+      "loss": 0.5704,
+      "step": 2581
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.0723845437169075,
+      "learning_rate": 3.692962426764175e-05,
+      "loss": 0.6224,
+      "step": 2582
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.07726027071475983,
+      "learning_rate": 3.686166325014206e-05,
+      "loss": 0.4605,
+      "step": 2583
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.08010595291852951,
+      "learning_rate": 3.6793750691400994e-05,
+      "loss": 0.57,
+      "step": 2584
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.08348097652196884,
+      "learning_rate": 3.672588664354148e-05,
+      "loss": 0.6409,
+      "step": 2585
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.08310791850090027,
+      "learning_rate": 3.665807115864928e-05,
+      "loss": 0.7339,
+      "step": 2586
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.07286600768566132,
+      "learning_rate": 3.659030428877292e-05,
+      "loss": 0.5772,
+      "step": 2587
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.0777403935790062,
+      "learning_rate": 3.652258608592347e-05,
+      "loss": 0.6053,
+      "step": 2588
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.07888954877853394,
+      "learning_rate": 3.645491660207484e-05,
+      "loss": 0.6204,
+      "step": 2589
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.07620932906866074,
+      "learning_rate": 3.6387295889163366e-05,
+      "loss": 0.6508,
+      "step": 2590
+    },
+    {
+      "epoch": 1.42,
+      "grad_norm": 0.08210571855306625,
+      "learning_rate": 3.631972399908811e-05,
+      "loss": 0.623,
+      "step": 2591
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.06440088897943497,
+      "learning_rate": 3.6252200983710514e-05,
+      "loss": 0.4804,
+      "step": 2592
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.07377233356237411,
+      "learning_rate": 3.6184726894854656e-05,
+      "loss": 0.5823,
+      "step": 2593
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.08177551627159119,
+      "learning_rate": 3.611730178430692e-05,
+      "loss": 0.6104,
+      "step": 2594
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.0679471418261528,
+      "learning_rate": 3.6049925703816214e-05,
+      "loss": 0.5188,
+      "step": 2595
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.06674007326364517,
+      "learning_rate": 3.5982598705093784e-05,
+      "loss": 0.654,
+      "step": 2596
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.07756850123405457,
+      "learning_rate": 3.591532083981315e-05,
+      "loss": 0.6562,
+      "step": 2597
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.07346350699663162,
+      "learning_rate": 3.584809215961017e-05,
+      "loss": 0.5255,
+      "step": 2598
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.06543095409870148,
+      "learning_rate": 3.578091271608297e-05,
+      "loss": 0.5301,
+      "step": 2599
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.07410463690757751,
+      "learning_rate": 3.5713782560791886e-05,
+      "loss": 0.6076,
+      "step": 2600
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.07580257207155228,
+      "learning_rate": 3.564670174525934e-05,
+      "loss": 0.5918,
+      "step": 2601
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.07186981290578842,
+      "learning_rate": 3.557967032097e-05,
+      "loss": 0.5289,
+      "step": 2602
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.08301349729299545,
+      "learning_rate": 3.551268833937054e-05,
+      "loss": 0.5828,
+      "step": 2603
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.06555919349193573,
+      "learning_rate": 3.544575585186976e-05,
+      "loss": 0.5391,
+      "step": 2604
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.0664808601140976,
+      "learning_rate": 3.537887290983838e-05,
+      "loss": 0.5634,
+      "step": 2605
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.06681336462497711,
+      "learning_rate": 3.53120395646092e-05,
+      "loss": 0.5548,
+      "step": 2606
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.07828280329704285,
+      "learning_rate": 3.5245255867476856e-05,
+      "loss": 0.6027,
+      "step": 2607
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.07380471378564835,
+      "learning_rate": 3.5178521869697956e-05,
+      "loss": 0.5826,
+      "step": 2608
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 0.08058056235313416,
+      "learning_rate": 3.511183762249095e-05,
+      "loss": 0.6185,
+      "step": 2609
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.06906460970640182,
+      "learning_rate": 3.5045203177036035e-05,
+      "loss": 0.5639,
+      "step": 2610
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.07293884456157684,
+      "learning_rate": 3.497861858447531e-05,
+      "loss": 0.5673,
+      "step": 2611
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.07099581509828568,
+      "learning_rate": 3.491208389591245e-05,
+      "loss": 0.4846,
+      "step": 2612
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.07652019709348679,
+      "learning_rate": 3.484559916241301e-05,
+      "loss": 0.6081,
+      "step": 2613
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.06962558627128601,
+      "learning_rate": 3.477916443500403e-05,
+      "loss": 0.5261,
+      "step": 2614
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.06621481478214264,
+      "learning_rate": 3.471277976467432e-05,
+      "loss": 0.5502,
+      "step": 2615
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.06833726167678833,
+      "learning_rate": 3.4646445202374146e-05,
+      "loss": 0.5267,
+      "step": 2616
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.07530024647712708,
+      "learning_rate": 3.458016079901544e-05,
+      "loss": 0.5676,
+      "step": 2617
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.07977502793073654,
+      "learning_rate": 3.45139266054715e-05,
+      "loss": 0.6458,
+      "step": 2618
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.07798527181148529,
+      "learning_rate": 3.444774267257719e-05,
+      "loss": 0.5013,
+      "step": 2619
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.06748402863740921,
+      "learning_rate": 3.438160905112881e-05,
+      "loss": 0.5423,
+      "step": 2620
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.07368659228086472,
+      "learning_rate": 3.4315525791883915e-05,
+      "loss": 0.6145,
+      "step": 2621
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.07269573956727982,
+      "learning_rate": 3.424949294556159e-05,
+      "loss": 0.5452,
+      "step": 2622
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.07823427021503448,
+      "learning_rate": 3.418351056284206e-05,
+      "loss": 0.6056,
+      "step": 2623
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.08030471205711365,
+      "learning_rate": 3.411757869436695e-05,
+      "loss": 0.597,
+      "step": 2624
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.08266133069992065,
+      "learning_rate": 3.405169739073899e-05,
+      "loss": 0.6015,
+      "step": 2625
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.07893429696559906,
+      "learning_rate": 3.398586670252225e-05,
+      "loss": 0.5495,
+      "step": 2626
+    },
+    {
+      "epoch": 1.44,
+      "grad_norm": 0.07488974928855896,
+      "learning_rate": 3.3920086680241795e-05,
+      "loss": 0.6026,
+      "step": 2627
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.0660768523812294,
+      "learning_rate": 3.38543573743839e-05,
+      "loss": 0.4573,
+      "step": 2628
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.0831848755478859,
+      "learning_rate": 3.378867883539597e-05,
+      "loss": 0.6324,
+      "step": 2629
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.07275530695915222,
+      "learning_rate": 3.3723051113686263e-05,
+      "loss": 0.5845,
+      "step": 2630
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.08007675409317017,
+      "learning_rate": 3.365747425962424e-05,
+      "loss": 0.6177,
+      "step": 2631
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.0778718814253807,
+      "learning_rate": 3.359194832354014e-05,
+      "loss": 0.5392,
+      "step": 2632
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.07712651044130325,
+      "learning_rate": 3.3526473355725294e-05,
+      "loss": 0.6194,
+      "step": 2633
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.071379154920578,
+      "learning_rate": 3.346104940643174e-05,
+      "loss": 0.6409,
+      "step": 2634
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.06798027455806732,
+      "learning_rate": 3.339567652587252e-05,
+      "loss": 0.601,
+      "step": 2635
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.07837600260972977,
+      "learning_rate": 3.333035476422134e-05,
+      "loss": 0.6524,
+      "step": 2636
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.0715259313583374,
+      "learning_rate": 3.326508417161278e-05,
+      "loss": 0.5168,
+      "step": 2637
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.07512487471103668,
+      "learning_rate": 3.3199864798142146e-05,
+      "loss": 0.578,
+      "step": 2638
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.07843317836523056,
+      "learning_rate": 3.3134696693865316e-05,
+      "loss": 0.583,
+      "step": 2639
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.0695166289806366,
+      "learning_rate": 3.3069579908798964e-05,
+      "loss": 0.5844,
+      "step": 2640
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.08100777119398117,
+      "learning_rate": 3.3004514492920257e-05,
+      "loss": 0.6408,
+      "step": 2641
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.06774459779262543,
+      "learning_rate": 3.2939500496167044e-05,
+      "loss": 0.5583,
+      "step": 2642
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.08072764426469803,
+      "learning_rate": 3.287453796843759e-05,
+      "loss": 0.6122,
+      "step": 2643
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.08036383241415024,
+      "learning_rate": 3.280962695959079e-05,
+      "loss": 0.5712,
+      "step": 2644
+    },
+    {
+      "epoch": 1.45,
+      "grad_norm": 0.06367004662752151,
+      "learning_rate": 3.274476751944587e-05,
+      "loss": 0.524,
+      "step": 2645
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.07899057865142822,
+      "learning_rate": 3.267995969778257e-05,
+      "loss": 0.5748,
+      "step": 2646
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.07749677449464798,
+      "learning_rate": 3.2615203544341e-05,
+      "loss": 0.5404,
+      "step": 2647
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.0736050009727478,
+      "learning_rate": 3.255049910882155e-05,
+      "loss": 0.5814,
+      "step": 2648
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.07629131525754929,
+      "learning_rate": 3.248584644088501e-05,
+      "loss": 0.6311,
+      "step": 2649
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.07851318269968033,
+      "learning_rate": 3.242124559015234e-05,
+      "loss": 0.6012,
+      "step": 2650
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.06685943901538849,
+      "learning_rate": 3.235669660620483e-05,
+      "loss": 0.4975,
+      "step": 2651
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.08383215963840485,
+      "learning_rate": 3.229219953858386e-05,
+      "loss": 0.6611,
+      "step": 2652
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.08177929371595383,
+      "learning_rate": 3.222775443679107e-05,
+      "loss": 0.4994,
+      "step": 2653
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.07951292395591736,
+      "learning_rate": 3.21633613502881e-05,
+      "loss": 0.6144,
+      "step": 2654
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.07102277129888535,
+      "learning_rate": 3.2099020328496765e-05,
+      "loss": 0.5143,
+      "step": 2655
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.08011867851018906,
+      "learning_rate": 3.2034731420798926e-05,
+      "loss": 0.6795,
+      "step": 2656
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.0753115639090538,
+      "learning_rate": 3.197049467653633e-05,
+      "loss": 0.634,
+      "step": 2657
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.06923001259565353,
+      "learning_rate": 3.1906310145010845e-05,
+      "loss": 0.4819,
+      "step": 2658
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.07761702686548233,
+      "learning_rate": 3.1842177875484095e-05,
+      "loss": 0.5315,
+      "step": 2659
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.07101301848888397,
+      "learning_rate": 3.177809791717778e-05,
+      "loss": 0.4987,
+      "step": 2660
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.08525367081165314,
+      "learning_rate": 3.171407031927325e-05,
+      "loss": 0.5819,
+      "step": 2661
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.07306365668773651,
+      "learning_rate": 3.165009513091187e-05,
+      "loss": 0.516,
+      "step": 2662
+    },
+    {
+      "epoch": 1.46,
+      "grad_norm": 0.06797949224710464,
+      "learning_rate": 3.158617240119461e-05,
+      "loss": 0.5117,
+      "step": 2663
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.07655341923236847,
+      "learning_rate": 3.152230217918229e-05,
+      "loss": 0.5793,
+      "step": 2664
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.07946762442588806,
+      "learning_rate": 3.145848451389542e-05,
+      "loss": 0.6222,
+      "step": 2665
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.08384986221790314,
+      "learning_rate": 3.139471945431406e-05,
+      "loss": 0.6602,
+      "step": 2666
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.07310443371534348,
+      "learning_rate": 3.133100704937804e-05,
+      "loss": 0.5332,
+      "step": 2667
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.07994786649942398,
+      "learning_rate": 3.126734734798669e-05,
+      "loss": 0.596,
+      "step": 2668
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.073966383934021,
+      "learning_rate": 3.120374039899897e-05,
+      "loss": 0.6057,
+      "step": 2669
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.08127514272928238,
+      "learning_rate": 3.114018625123323e-05,
+      "loss": 0.5887,
+      "step": 2670
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.08686904609203339,
+      "learning_rate": 3.107668495346743e-05,
+      "loss": 0.6865,
+      "step": 2671
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.07904399186372757,
+      "learning_rate": 3.101323655443882e-05,
+      "loss": 0.5394,
+      "step": 2672
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.08745576441287994,
+      "learning_rate": 3.0949841102844225e-05,
+      "loss": 0.5725,
+      "step": 2673
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.06875587999820709,
+      "learning_rate": 3.088649864733965e-05,
+      "loss": 0.5781,
+      "step": 2674
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.07002407312393188,
+      "learning_rate": 3.0823209236540596e-05,
+      "loss": 0.4903,
+      "step": 2675
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.06887030601501465,
+      "learning_rate": 3.0759972919021695e-05,
+      "loss": 0.5208,
+      "step": 2676
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.07510718703269958,
+      "learning_rate": 3.0696789743316945e-05,
+      "loss": 0.5329,
+      "step": 2677
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.07978718727827072,
+      "learning_rate": 3.0633659757919544e-05,
+      "loss": 0.608,
+      "step": 2678
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.08159972727298737,
+      "learning_rate": 3.057058301128178e-05,
+      "loss": 0.6215,
+      "step": 2679
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.06593898683786392,
+      "learning_rate": 3.0507559551815223e-05,
+      "loss": 0.4854,
+      "step": 2680
+    },
+    {
+      "epoch": 1.47,
+      "grad_norm": 0.0734141394495964,
+      "learning_rate": 3.044458942789037e-05,
+      "loss": 0.5405,
+      "step": 2681
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.07597970217466354,
+      "learning_rate": 3.0381672687836948e-05,
+      "loss": 0.5526,
+      "step": 2682
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.08808930218219757,
+      "learning_rate": 3.031880937994359e-05,
+      "loss": 0.596,
+      "step": 2683
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.07469739764928818,
+      "learning_rate": 3.0255999552458026e-05,
+      "loss": 0.5118,
+      "step": 2684
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.07475581020116806,
+      "learning_rate": 3.0193243253586813e-05,
+      "loss": 0.6382,
+      "step": 2685
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.07039845734834671,
+      "learning_rate": 3.0130540531495534e-05,
+      "loss": 0.5234,
+      "step": 2686
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.07057277858257294,
+      "learning_rate": 3.0067891434308638e-05,
+      "loss": 0.4961,
+      "step": 2687
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.06783699244260788,
+      "learning_rate": 3.000529601010934e-05,
+      "loss": 0.4642,
+      "step": 2688
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.08259709924459457,
+      "learning_rate": 2.9942754306939758e-05,
+      "loss": 0.6044,
+      "step": 2689
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.06971608847379684,
+      "learning_rate": 2.988026637280069e-05,
+      "loss": 0.5563,
+      "step": 2690
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.07627476751804352,
+      "learning_rate": 2.9817832255651757e-05,
+      "loss": 0.5481,
+      "step": 2691
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.07380031794309616,
+      "learning_rate": 2.9755452003411166e-05,
+      "loss": 0.5532,
+      "step": 2692
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.08517714589834213,
+      "learning_rate": 2.9693125663955924e-05,
+      "loss": 0.6814,
+      "step": 2693
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.07390910387039185,
+      "learning_rate": 2.9630853285121508e-05,
+      "loss": 0.593,
+      "step": 2694
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.080739825963974,
+      "learning_rate": 2.9568634914702077e-05,
+      "loss": 0.5196,
+      "step": 2695
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.0803290456533432,
+      "learning_rate": 2.950647060045034e-05,
+      "loss": 0.6102,
+      "step": 2696
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.07901672273874283,
+      "learning_rate": 2.9444360390077452e-05,
+      "loss": 0.6337,
+      "step": 2697
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.07600972801446915,
+      "learning_rate": 2.938230433125314e-05,
+      "loss": 0.6069,
+      "step": 2698
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 0.08403575420379639,
+      "learning_rate": 2.932030247160543e-05,
+      "loss": 0.695,
+      "step": 2699
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07824236154556274,
+      "learning_rate": 2.9258354858720926e-05,
+      "loss": 0.6478,
+      "step": 2700
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07876936346292496,
+      "learning_rate": 2.9196461540144414e-05,
+      "loss": 0.6292,
+      "step": 2701
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07499687373638153,
+      "learning_rate": 2.9134622563379188e-05,
+      "loss": 0.5043,
+      "step": 2702
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07794086635112762,
+      "learning_rate": 2.9072837975886657e-05,
+      "loss": 0.5335,
+      "step": 2703
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.0772542953491211,
+      "learning_rate": 2.901110782508665e-05,
+      "loss": 0.4854,
+      "step": 2704
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07389385998249054,
+      "learning_rate": 2.894943215835708e-05,
+      "loss": 0.6506,
+      "step": 2705
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.08659908920526505,
+      "learning_rate": 2.8887811023034138e-05,
+      "loss": 0.6311,
+      "step": 2706
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07332868874073029,
+      "learning_rate": 2.8826244466412156e-05,
+      "loss": 0.5668,
+      "step": 2707
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07572486996650696,
+      "learning_rate": 2.8764732535743487e-05,
+      "loss": 0.5522,
+      "step": 2708
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07586922496557236,
+      "learning_rate": 2.8703275278238683e-05,
+      "loss": 0.5782,
+      "step": 2709
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07996062189340591,
+      "learning_rate": 2.8641872741066213e-05,
+      "loss": 0.6577,
+      "step": 2710
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07385245710611343,
+      "learning_rate": 2.858052497135265e-05,
+      "loss": 0.5567,
+      "step": 2711
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07866589725017548,
+      "learning_rate": 2.8519232016182463e-05,
+      "loss": 0.6275,
+      "step": 2712
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07945741713047028,
+      "learning_rate": 2.8457993922598103e-05,
+      "loss": 0.546,
+      "step": 2713
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.07978387176990509,
+      "learning_rate": 2.8396810737599854e-05,
+      "loss": 0.5909,
+      "step": 2714
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.08178815245628357,
+      "learning_rate": 2.8335682508145922e-05,
+      "loss": 0.5666,
+      "step": 2715
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.0759677067399025,
+      "learning_rate": 2.827460928115232e-05,
+      "loss": 0.6056,
+      "step": 2716
+    },
+    {
+      "epoch": 1.49,
+      "grad_norm": 0.0705474466085434,
+      "learning_rate": 2.821359110349279e-05,
+      "loss": 0.5868,
+      "step": 2717
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.07681458443403244,
+      "learning_rate": 2.8152628021998905e-05,
+      "loss": 0.5932,
+      "step": 2718
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.07275570183992386,
+      "learning_rate": 2.809172008345986e-05,
+      "loss": 0.5254,
+      "step": 2719
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.07109987735748291,
+      "learning_rate": 2.8030867334622655e-05,
+      "loss": 0.5581,
+      "step": 2720
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.07160382717847824,
+      "learning_rate": 2.797006982219178e-05,
+      "loss": 0.5916,
+      "step": 2721
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.08834366500377655,
+      "learning_rate": 2.790932759282947e-05,
+      "loss": 0.6798,
+      "step": 2722
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.07338585704565048,
+      "learning_rate": 2.7848640693155415e-05,
+      "loss": 0.5371,
+      "step": 2723
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.0876719132065773,
+      "learning_rate": 2.778800916974692e-05,
+      "loss": 0.6038,
+      "step": 2724
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.06971060484647751,
+      "learning_rate": 2.7727433069138785e-05,
+      "loss": 0.5224,
+      "step": 2725
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.08299615979194641,
+      "learning_rate": 2.76669124378232e-05,
+      "loss": 0.5817,
+      "step": 2726
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.07271869480609894,
+      "learning_rate": 2.7606447322249872e-05,
+      "loss": 0.5423,
+      "step": 2727
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.08478042483329773,
+      "learning_rate": 2.7546037768825827e-05,
+      "loss": 0.594,
+      "step": 2728
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.08502127230167389,
+      "learning_rate": 2.7485683823915507e-05,
+      "loss": 0.6399,
+      "step": 2729
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.0918736606836319,
+      "learning_rate": 2.742538553384061e-05,
+      "loss": 0.5878,
+      "step": 2730
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.07426369935274124,
+      "learning_rate": 2.7365142944880206e-05,
+      "loss": 0.5661,
+      "step": 2731
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.07882078737020493,
+      "learning_rate": 2.7304956103270508e-05,
+      "loss": 0.5248,
+      "step": 2732
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.06941094249486923,
+      "learning_rate": 2.7244825055205015e-05,
+      "loss": 0.5908,
+      "step": 2733
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.07542286813259125,
+      "learning_rate": 2.71847498468344e-05,
+      "loss": 0.6083,
+      "step": 2734
+    },
+    {
+      "epoch": 1.5,
+      "grad_norm": 0.09603607654571533,
+      "learning_rate": 2.7124730524266496e-05,
+      "loss": 0.6575,
+      "step": 2735
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.0671052411198616,
+      "learning_rate": 2.706476713356615e-05,
+      "loss": 0.4713,
+      "step": 2736
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.08022913336753845,
+      "learning_rate": 2.7004859720755372e-05,
+      "loss": 0.6117,
+      "step": 2737
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.08039288222789764,
+      "learning_rate": 2.6945008331813226e-05,
+      "loss": 0.5949,
+      "step": 2738
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.07160980254411697,
+      "learning_rate": 2.688521301267565e-05,
+      "loss": 0.6303,
+      "step": 2739
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.07343259453773499,
+      "learning_rate": 2.6825473809235713e-05,
+      "loss": 0.5956,
+      "step": 2740
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.07557319104671478,
+      "learning_rate": 2.6765790767343267e-05,
+      "loss": 0.5702,
+      "step": 2741
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.07988648861646652,
+      "learning_rate": 2.6706163932805195e-05,
+      "loss": 0.5878,
+      "step": 2742
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.0786256268620491,
+      "learning_rate": 2.6646593351385097e-05,
+      "loss": 0.5026,
+      "step": 2743
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.08495064079761505,
+      "learning_rate": 2.6587079068803545e-05,
+      "loss": 0.635,
+      "step": 2744
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.08269646018743515,
+      "learning_rate": 2.6527621130737768e-05,
+      "loss": 0.6067,
+      "step": 2745
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.08189453929662704,
+      "learning_rate": 2.646821958282184e-05,
+      "loss": 0.5704,
+      "step": 2746
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.06907159090042114,
+      "learning_rate": 2.6408874470646572e-05,
+      "loss": 0.5026,
+      "step": 2747
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.07225730270147324,
+      "learning_rate": 2.6349585839759348e-05,
+      "loss": 0.5752,
+      "step": 2748
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.07018494606018066,
+      "learning_rate": 2.629035373566433e-05,
+      "loss": 0.5393,
+      "step": 2749
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.08471867442131042,
+      "learning_rate": 2.6231178203822182e-05,
+      "loss": 0.5768,
+      "step": 2750
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.07804112136363983,
+      "learning_rate": 2.6172059289650263e-05,
+      "loss": 0.6254,
+      "step": 2751
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.06657396256923676,
+      "learning_rate": 2.6112997038522368e-05,
+      "loss": 0.523,
+      "step": 2752
+    },
+    {
+      "epoch": 1.51,
+      "grad_norm": 0.075431227684021,
+      "learning_rate": 2.6053991495768903e-05,
+      "loss": 0.5552,
+      "step": 2753
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.07600487768650055,
+      "learning_rate": 2.5995042706676643e-05,
+      "loss": 0.5392,
+      "step": 2754
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.07607197016477585,
+      "learning_rate": 2.5936150716488894e-05,
+      "loss": 0.5937,
+      "step": 2755
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.07737214118242264,
+      "learning_rate": 2.5877315570405368e-05,
+      "loss": 0.5638,
+      "step": 2756
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.08522753417491913,
+      "learning_rate": 2.5818537313582058e-05,
+      "loss": 0.5747,
+      "step": 2757
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.07582385092973709,
+      "learning_rate": 2.575981599113142e-05,
+      "loss": 0.5801,
+      "step": 2758
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.09301973879337311,
+      "learning_rate": 2.570115164812209e-05,
+      "loss": 0.597,
+      "step": 2759
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.07498759031295776,
+      "learning_rate": 2.5642544329579088e-05,
+      "loss": 0.5826,
+      "step": 2760
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.07844389975070953,
+      "learning_rate": 2.558399408048354e-05,
+      "loss": 0.5778,
+      "step": 2761
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.07821419835090637,
+      "learning_rate": 2.5525500945772918e-05,
+      "loss": 0.6254,
+      "step": 2762
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.07561523467302322,
+      "learning_rate": 2.5467064970340704e-05,
+      "loss": 0.508,
+      "step": 2763
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.0730084627866745,
+      "learning_rate": 2.5408686199036623e-05,
+      "loss": 0.5472,
+      "step": 2764
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.07727015018463135,
+      "learning_rate": 2.5350364676666505e-05,
+      "loss": 0.6141,
+      "step": 2765
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.06693361699581146,
+      "learning_rate": 2.529210044799213e-05,
+      "loss": 0.546,
+      "step": 2766
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.07514087855815887,
+      "learning_rate": 2.5233893557731412e-05,
+      "loss": 0.4996,
+      "step": 2767
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.08636999875307083,
+      "learning_rate": 2.517574405055819e-05,
+      "loss": 0.6738,
+      "step": 2768
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.09145054966211319,
+      "learning_rate": 2.511765197110233e-05,
+      "loss": 0.5968,
+      "step": 2769
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.07830934226512909,
+      "learning_rate": 2.5059617363949518e-05,
+      "loss": 0.5855,
+      "step": 2770
+    },
+    {
+      "epoch": 1.52,
+      "grad_norm": 0.06930552423000336,
+      "learning_rate": 2.500164027364147e-05,
+      "loss": 0.552,
+      "step": 2771
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.08295781910419464,
+      "learning_rate": 2.4943720744675603e-05,
+      "loss": 0.6518,
+      "step": 2772
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.07150809466838837,
+      "learning_rate": 2.4885858821505272e-05,
+      "loss": 0.5272,
+      "step": 2773
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.07201548665761948,
+      "learning_rate": 2.4828054548539615e-05,
+      "loss": 0.5088,
+      "step": 2774
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.07628802210092545,
+      "learning_rate": 2.4770307970143424e-05,
+      "loss": 0.5727,
+      "step": 2775
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.07237250357866287,
+      "learning_rate": 2.471261913063734e-05,
+      "loss": 0.48,
+      "step": 2776
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.08393524587154388,
+      "learning_rate": 2.4654988074297557e-05,
+      "loss": 0.5324,
+      "step": 2777
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.0864768996834755,
+      "learning_rate": 2.459741484535606e-05,
+      "loss": 0.634,
+      "step": 2778
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.07512786239385605,
+      "learning_rate": 2.4539899488000305e-05,
+      "loss": 0.5888,
+      "step": 2779
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.07718805223703384,
+      "learning_rate": 2.4482442046373478e-05,
+      "loss": 0.6082,
+      "step": 2780
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.08047004789113998,
+      "learning_rate": 2.4425042564574184e-05,
+      "loss": 0.5864,
+      "step": 2781
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.07809320092201233,
+      "learning_rate": 2.4367701086656624e-05,
+      "loss": 0.6021,
+      "step": 2782
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.07543138414621353,
+      "learning_rate": 2.431041765663049e-05,
+      "loss": 0.5307,
+      "step": 2783
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.07181257754564285,
+      "learning_rate": 2.425319231846085e-05,
+      "loss": 0.5957,
+      "step": 2784
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.0678059458732605,
+      "learning_rate": 2.4196025116068256e-05,
+      "loss": 0.513,
+      "step": 2785
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.08409433811903,
+      "learning_rate": 2.4138916093328578e-05,
+      "loss": 0.6004,
+      "step": 2786
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.0672868862748146,
+      "learning_rate": 2.4081865294073124e-05,
+      "loss": 0.4609,
+      "step": 2787
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.07355806231498718,
+      "learning_rate": 2.402487276208839e-05,
+      "loss": 0.6233,
+      "step": 2788
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 0.06832050532102585,
+      "learning_rate": 2.3967938541116297e-05,
+      "loss": 0.5685,
+      "step": 2789
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.07053340971469879,
+      "learning_rate": 2.3911062674853858e-05,
+      "loss": 0.5607,
+      "step": 2790
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.08480045199394226,
+      "learning_rate": 2.3854245206953452e-05,
+      "loss": 0.5808,
+      "step": 2791
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.073263980448246,
+      "learning_rate": 2.37974861810225e-05,
+      "loss": 0.5948,
+      "step": 2792
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.08538652211427689,
+      "learning_rate": 2.3740785640623643e-05,
+      "loss": 0.5938,
+      "step": 2793
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.07441440224647522,
+      "learning_rate": 2.368414362927468e-05,
+      "loss": 0.6418,
+      "step": 2794
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.07826292514801025,
+      "learning_rate": 2.362756019044835e-05,
+      "loss": 0.6989,
+      "step": 2795
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.07607203722000122,
+      "learning_rate": 2.357103536757258e-05,
+      "loss": 0.5775,
+      "step": 2796
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.0835345908999443,
+      "learning_rate": 2.35145692040302e-05,
+      "loss": 0.4798,
+      "step": 2797
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.06911890208721161,
+      "learning_rate": 2.3458161743159124e-05,
+      "loss": 0.5362,
+      "step": 2798
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.07400982826948166,
+      "learning_rate": 2.3401813028252085e-05,
+      "loss": 0.569,
+      "step": 2799
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.08338724821805954,
+      "learning_rate": 2.3345523102556867e-05,
+      "loss": 0.5607,
+      "step": 2800
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.07647686451673508,
+      "learning_rate": 2.3289292009276e-05,
+      "loss": 0.668,
+      "step": 2801
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.08820470422506332,
+      "learning_rate": 2.3233119791566948e-05,
+      "loss": 0.641,
+      "step": 2802
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.0804409384727478,
+      "learning_rate": 2.3177006492541976e-05,
+      "loss": 0.6515,
+      "step": 2803
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.060682669281959534,
+      "learning_rate": 2.312095215526814e-05,
+      "loss": 0.3945,
+      "step": 2804
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.09143857657909393,
+      "learning_rate": 2.3064956822767157e-05,
+      "loss": 0.6366,
+      "step": 2805
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.07404692471027374,
+      "learning_rate": 2.300902053801556e-05,
+      "loss": 0.5778,
+      "step": 2806
+    },
+    {
+      "epoch": 1.54,
+      "grad_norm": 0.07459226250648499,
+      "learning_rate": 2.2953143343944528e-05,
+      "loss": 0.5378,
+      "step": 2807
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.08721205592155457,
+      "learning_rate": 2.2897325283439864e-05,
+      "loss": 0.6642,
+      "step": 2808
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.07177091389894485,
+      "learning_rate": 2.284156639934203e-05,
+      "loss": 0.5285,
+      "step": 2809
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.08235837519168854,
+      "learning_rate": 2.2785866734445994e-05,
+      "loss": 0.6906,
+      "step": 2810
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.0843455046415329,
+      "learning_rate": 2.2730226331501393e-05,
+      "loss": 0.6085,
+      "step": 2811
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.07604355365037918,
+      "learning_rate": 2.2674645233212234e-05,
+      "loss": 0.5811,
+      "step": 2812
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.08346116542816162,
+      "learning_rate": 2.261912348223717e-05,
+      "loss": 0.586,
+      "step": 2813
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.08567336946725845,
+      "learning_rate": 2.256366112118913e-05,
+      "loss": 0.6379,
+      "step": 2814
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.0706150084733963,
+      "learning_rate": 2.2508258192635612e-05,
+      "loss": 0.4955,
+      "step": 2815
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.07677091658115387,
+      "learning_rate": 2.245291473909844e-05,
+      "loss": 0.5504,
+      "step": 2816
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.0727921649813652,
+      "learning_rate": 2.239763080305375e-05,
+      "loss": 0.6073,
+      "step": 2817
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.0712139904499054,
+      "learning_rate": 2.2342406426932084e-05,
+      "loss": 0.5859,
+      "step": 2818
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.07416416704654694,
+      "learning_rate": 2.2287241653118172e-05,
+      "loss": 0.4962,
+      "step": 2819
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.07774526625871658,
+      "learning_rate": 2.2232136523951107e-05,
+      "loss": 0.5622,
+      "step": 2820
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.06952360272407532,
+      "learning_rate": 2.21770910817241e-05,
+      "loss": 0.4522,
+      "step": 2821
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.07877884060144424,
+      "learning_rate": 2.2122105368684643e-05,
+      "loss": 0.6359,
+      "step": 2822
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.08205553144216537,
+      "learning_rate": 2.2067179427034314e-05,
+      "loss": 0.5587,
+      "step": 2823
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.07097803801298141,
+      "learning_rate": 2.2012313298928855e-05,
+      "loss": 0.5439,
+      "step": 2824
+    },
+    {
+      "epoch": 1.55,
+      "grad_norm": 0.08037848025560379,
+      "learning_rate": 2.1957507026478118e-05,
+      "loss": 0.6708,
+      "step": 2825
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.07214182615280151,
+      "learning_rate": 2.1902760651745958e-05,
+      "loss": 0.5717,
+      "step": 2826
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.08701501786708832,
+      "learning_rate": 2.1848074216750324e-05,
+      "loss": 0.5975,
+      "step": 2827
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.07122653722763062,
+      "learning_rate": 2.1793447763463093e-05,
+      "loss": 0.5434,
+      "step": 2828
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.08395606279373169,
+      "learning_rate": 2.173888133381018e-05,
+      "loss": 0.6117,
+      "step": 2829
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.08226127922534943,
+      "learning_rate": 2.168437496967134e-05,
+      "loss": 0.5687,
+      "step": 2830
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.07219874113798141,
+      "learning_rate": 2.1629928712880344e-05,
+      "loss": 0.5374,
+      "step": 2831
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.0728921890258789,
+      "learning_rate": 2.1575542605224707e-05,
+      "loss": 0.5228,
+      "step": 2832
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.08040345460176468,
+      "learning_rate": 2.152121668844588e-05,
+      "loss": 0.5647,
+      "step": 2833
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.08091174066066742,
+      "learning_rate": 2.1466951004239077e-05,
+      "loss": 0.4601,
+      "step": 2834
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.06813590973615646,
+      "learning_rate": 2.141274559425326e-05,
+      "loss": 0.4936,
+      "step": 2835
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.07925499230623245,
+      "learning_rate": 2.1358600500091185e-05,
+      "loss": 0.529,
+      "step": 2836
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.08047482371330261,
+      "learning_rate": 2.1304515763309253e-05,
+      "loss": 0.4992,
+      "step": 2837
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.08449546247720718,
+      "learning_rate": 2.1250491425417607e-05,
+      "loss": 0.6248,
+      "step": 2838
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.08615768700838089,
+      "learning_rate": 2.1196527527879952e-05,
+      "loss": 0.5692,
+      "step": 2839
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.0787179172039032,
+      "learning_rate": 2.1142624112113707e-05,
+      "loss": 0.5379,
+      "step": 2840
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.07938480377197266,
+      "learning_rate": 2.1088781219489762e-05,
+      "loss": 0.5346,
+      "step": 2841
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.07051555812358856,
+      "learning_rate": 2.1034998891332637e-05,
+      "loss": 0.4809,
+      "step": 2842
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 0.08885900676250458,
+      "learning_rate": 2.0981277168920364e-05,
+      "loss": 0.6303,
+      "step": 2843
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.0760311633348465,
+      "learning_rate": 2.0927616093484394e-05,
+      "loss": 0.5412,
+      "step": 2844
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.0825723186135292,
+      "learning_rate": 2.0874015706209716e-05,
+      "loss": 0.6254,
+      "step": 2845
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.08421153575181961,
+      "learning_rate": 2.082047604823465e-05,
+      "loss": 0.6521,
+      "step": 2846
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.07435724139213562,
+      "learning_rate": 2.0766997160651015e-05,
+      "loss": 0.6036,
+      "step": 2847
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.0800776332616806,
+      "learning_rate": 2.0713579084503876e-05,
+      "loss": 0.5592,
+      "step": 2848
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.07602731883525848,
+      "learning_rate": 2.0660221860791717e-05,
+      "loss": 0.5496,
+      "step": 2849
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.08413247764110565,
+      "learning_rate": 2.0606925530466248e-05,
+      "loss": 0.5143,
+      "step": 2850
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.07869786769151688,
+      "learning_rate": 2.055369013443248e-05,
+      "loss": 0.5678,
+      "step": 2851
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.07760642468929291,
+      "learning_rate": 2.0500515713548685e-05,
+      "loss": 0.4704,
+      "step": 2852
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.07946940511465073,
+      "learning_rate": 2.0447402308626262e-05,
+      "loss": 0.5505,
+      "step": 2853
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.08587105572223663,
+      "learning_rate": 2.039434996042986e-05,
+      "loss": 0.6598,
+      "step": 2854
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.06604398787021637,
+      "learning_rate": 2.0341358709677173e-05,
+      "loss": 0.4961,
+      "step": 2855
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.06659547239542007,
+      "learning_rate": 2.0288428597039122e-05,
+      "loss": 0.5459,
+      "step": 2856
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.07560455799102783,
+      "learning_rate": 2.023555966313958e-05,
+      "loss": 0.5326,
+      "step": 2857
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.07178471237421036,
+      "learning_rate": 2.0182751948555577e-05,
+      "loss": 0.5295,
+      "step": 2858
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.0831179991364479,
+      "learning_rate": 2.013000549381706e-05,
+      "loss": 0.5768,
+      "step": 2859
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.08885984122753143,
+      "learning_rate": 2.0077320339407023e-05,
+      "loss": 0.637,
+      "step": 2860
+    },
+    {
+      "epoch": 1.57,
+      "grad_norm": 0.0871300920844078,
+      "learning_rate": 2.002469652576141e-05,
+      "loss": 0.6254,
+      "step": 2861
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.07685824483633041,
+      "learning_rate": 1.9972134093269035e-05,
+      "loss": 0.5274,
+      "step": 2862
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.08027210086584091,
+      "learning_rate": 1.9919633082271682e-05,
+      "loss": 0.5917,
+      "step": 2863
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.0697130337357521,
+      "learning_rate": 1.9867193533063898e-05,
+      "loss": 0.5018,
+      "step": 2864
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.08178181946277618,
+      "learning_rate": 1.9814815485893145e-05,
+      "loss": 0.5617,
+      "step": 2865
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.06824837625026703,
+      "learning_rate": 1.976249898095962e-05,
+      "loss": 0.5842,
+      "step": 2866
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.06705306470394135,
+      "learning_rate": 1.971024405841634e-05,
+      "loss": 0.4984,
+      "step": 2867
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.07768554240465164,
+      "learning_rate": 1.9658050758368975e-05,
+      "loss": 0.5888,
+      "step": 2868
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.08375062048435211,
+      "learning_rate": 1.9605919120876016e-05,
+      "loss": 0.5891,
+      "step": 2869
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.07355199009180069,
+      "learning_rate": 1.9553849185948512e-05,
+      "loss": 0.533,
+      "step": 2870
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.08470550924539566,
+      "learning_rate": 1.9501840993550236e-05,
+      "loss": 0.6016,
+      "step": 2871
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.07025481760501862,
+      "learning_rate": 1.9449894583597537e-05,
+      "loss": 0.5826,
+      "step": 2872
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.07526448369026184,
+      "learning_rate": 1.9398009995959365e-05,
+      "loss": 0.5368,
+      "step": 2873
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.07720872014760971,
+      "learning_rate": 1.934618727045724e-05,
+      "loss": 0.654,
+      "step": 2874
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.07709453254938126,
+      "learning_rate": 1.92944264468651e-05,
+      "loss": 0.5871,
+      "step": 2875
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.08008085936307907,
+      "learning_rate": 1.9242727564909524e-05,
+      "loss": 0.579,
+      "step": 2876
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.08657196909189224,
+      "learning_rate": 1.9191090664269396e-05,
+      "loss": 0.6261,
+      "step": 2877
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.0801343247294426,
+      "learning_rate": 1.913951578457619e-05,
+      "loss": 0.6157,
+      "step": 2878
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 0.07138290256261826,
+      "learning_rate": 1.908800296541361e-05,
+      "loss": 0.4559,
+      "step": 2879
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.07896512746810913,
+      "learning_rate": 1.9036552246317895e-05,
+      "loss": 0.5572,
+      "step": 2880
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.08345163613557816,
+      "learning_rate": 1.8985163666777473e-05,
+      "loss": 0.626,
+      "step": 2881
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.07745735347270966,
+      "learning_rate": 1.8933837266233212e-05,
+      "loss": 0.5568,
+      "step": 2882
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.07590015232563019,
+      "learning_rate": 1.8882573084078124e-05,
+      "loss": 0.4918,
+      "step": 2883
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.07770450413227081,
+      "learning_rate": 1.8831371159657584e-05,
+      "loss": 0.5597,
+      "step": 2884
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.08126380294561386,
+      "learning_rate": 1.8780231532269153e-05,
+      "loss": 0.6236,
+      "step": 2885
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.07325980067253113,
+      "learning_rate": 1.8729154241162505e-05,
+      "loss": 0.5563,
+      "step": 2886
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.07390178740024567,
+      "learning_rate": 1.86781393255396e-05,
+      "loss": 0.5878,
+      "step": 2887
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.09135127812623978,
+      "learning_rate": 1.862718682455439e-05,
+      "loss": 0.6212,
+      "step": 2888
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.08383052796125412,
+      "learning_rate": 1.8576296777313028e-05,
+      "loss": 0.591,
+      "step": 2889
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.07796967774629593,
+      "learning_rate": 1.852546922287367e-05,
+      "loss": 0.531,
+      "step": 2890
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.08362308144569397,
+      "learning_rate": 1.8474704200246573e-05,
+      "loss": 0.6157,
+      "step": 2891
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.07821770757436752,
+      "learning_rate": 1.8424001748393905e-05,
+      "loss": 0.6096,
+      "step": 2892
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.07985088974237442,
+      "learning_rate": 1.837336190622989e-05,
+      "loss": 0.6307,
+      "step": 2893
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.0782153308391571,
+      "learning_rate": 1.832278471262071e-05,
+      "loss": 0.5395,
+      "step": 2894
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.07065875083208084,
+      "learning_rate": 1.8272270206384368e-05,
+      "loss": 0.4953,
+      "step": 2895
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.07387331128120422,
+      "learning_rate": 1.822181842629087e-05,
+      "loss": 0.5472,
+      "step": 2896
+    },
+    {
+      "epoch": 1.59,
+      "grad_norm": 0.08914386481046677,
+      "learning_rate": 1.817142941106198e-05,
+      "loss": 0.6485,
+      "step": 2897
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.08163358271121979,
+      "learning_rate": 1.812110319937138e-05,
+      "loss": 0.5835,
+      "step": 2898
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.08932827413082123,
+      "learning_rate": 1.8070839829844453e-05,
+      "loss": 0.6668,
+      "step": 2899
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.08409236371517181,
+      "learning_rate": 1.8020639341058465e-05,
+      "loss": 0.6596,
+      "step": 2900
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.07313519716262817,
+      "learning_rate": 1.797050177154229e-05,
+      "loss": 0.5545,
+      "step": 2901
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.0781264528632164,
+      "learning_rate": 1.792042715977662e-05,
+      "loss": 0.5564,
+      "step": 2902
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.07732515037059784,
+      "learning_rate": 1.787041554419381e-05,
+      "loss": 0.4992,
+      "step": 2903
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.07877063006162643,
+      "learning_rate": 1.7820466963177783e-05,
+      "loss": 0.5458,
+      "step": 2904
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.07867057621479034,
+      "learning_rate": 1.7770581455064206e-05,
+      "loss": 0.5732,
+      "step": 2905
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.07103104889392853,
+      "learning_rate": 1.7720759058140212e-05,
+      "loss": 0.5122,
+      "step": 2906
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.07974914461374283,
+      "learning_rate": 1.7670999810644616e-05,
+      "loss": 0.5454,
+      "step": 2907
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.0706530511379242,
+      "learning_rate": 1.7621303750767648e-05,
+      "loss": 0.4803,
+      "step": 2908
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.07859217375516891,
+      "learning_rate": 1.7571670916651162e-05,
+      "loss": 0.5667,
+      "step": 2909
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.08591758459806442,
+      "learning_rate": 1.7522101346388376e-05,
+      "loss": 0.6084,
+      "step": 2910
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.08627095073461533,
+      "learning_rate": 1.7472595078024012e-05,
+      "loss": 0.6266,
+      "step": 2911
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.07653766870498657,
+      "learning_rate": 1.7423152149554233e-05,
+      "loss": 0.5526,
+      "step": 2912
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.08037185668945312,
+      "learning_rate": 1.7373772598926507e-05,
+      "loss": 0.5548,
+      "step": 2913
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.07113780826330185,
+      "learning_rate": 1.7324456464039752e-05,
+      "loss": 0.5352,
+      "step": 2914
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.07585661113262177,
+      "learning_rate": 1.727520378274412e-05,
+      "loss": 0.5925,
+      "step": 2915
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.07407297194004059,
+      "learning_rate": 1.7226014592841144e-05,
+      "loss": 0.5098,
+      "step": 2916
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.07840467989444733,
+      "learning_rate": 1.7176888932083568e-05,
+      "loss": 0.5716,
+      "step": 2917
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.08341764658689499,
+      "learning_rate": 1.7127826838175442e-05,
+      "loss": 0.6606,
+      "step": 2918
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.08113296329975128,
+      "learning_rate": 1.7078828348771935e-05,
+      "loss": 0.621,
+      "step": 2919
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.07104890048503876,
+      "learning_rate": 1.702989350147948e-05,
+      "loss": 0.485,
+      "step": 2920
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.07283183932304382,
+      "learning_rate": 1.6981022333855667e-05,
+      "loss": 0.551,
+      "step": 2921
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.07866816967725754,
+      "learning_rate": 1.6932214883409135e-05,
+      "loss": 0.6187,
+      "step": 2922
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.07849008589982986,
+      "learning_rate": 1.688347118759972e-05,
+      "loss": 0.6274,
+      "step": 2923
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.07671231031417847,
+      "learning_rate": 1.6834791283838215e-05,
+      "loss": 0.5543,
+      "step": 2924
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.07396913319826126,
+      "learning_rate": 1.6786175209486566e-05,
+      "loss": 0.5548,
+      "step": 2925
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.07642372697591782,
+      "learning_rate": 1.6737623001857618e-05,
+      "loss": 0.687,
+      "step": 2926
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.06577810645103455,
+      "learning_rate": 1.6689134698215325e-05,
+      "loss": 0.4493,
+      "step": 2927
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.08086547255516052,
+      "learning_rate": 1.6640710335774457e-05,
+      "loss": 0.5445,
+      "step": 2928
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.07448156177997589,
+      "learning_rate": 1.6592349951700813e-05,
+      "loss": 0.5462,
+      "step": 2929
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.07236510515213013,
+      "learning_rate": 1.6544053583111075e-05,
+      "loss": 0.5746,
+      "step": 2930
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.08082794398069382,
+      "learning_rate": 1.649582126707272e-05,
+      "loss": 0.5823,
+      "step": 2931
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.07060989737510681,
+      "learning_rate": 1.644765304060416e-05,
+      "loss": 0.5668,
+      "step": 2932
+    },
+    {
+      "epoch": 1.61,
+      "grad_norm": 0.09212745726108551,
+      "learning_rate": 1.6399548940674536e-05,
+      "loss": 0.6745,
+      "step": 2933
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.07585174590349197,
+      "learning_rate": 1.6351509004203857e-05,
+      "loss": 0.5716,
+      "step": 2934
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.07427546381950378,
+      "learning_rate": 1.6303533268062778e-05,
+      "loss": 0.5363,
+      "step": 2935
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.0801323726773262,
+      "learning_rate": 1.6255621769072805e-05,
+      "loss": 0.621,
+      "step": 2936
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.07136417180299759,
+      "learning_rate": 1.6207774544006016e-05,
+      "loss": 0.5717,
+      "step": 2937
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.08079267293214798,
+      "learning_rate": 1.6159991629585248e-05,
+      "loss": 0.5589,
+      "step": 2938
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.07775907218456268,
+      "learning_rate": 1.6112273062483985e-05,
+      "loss": 0.509,
+      "step": 2939
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.07815612852573395,
+      "learning_rate": 1.6064618879326242e-05,
+      "loss": 0.6139,
+      "step": 2940
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.07643534988164902,
+      "learning_rate": 1.6017029116686688e-05,
+      "loss": 0.703,
+      "step": 2941
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.07673238962888718,
+      "learning_rate": 1.5969503811090525e-05,
+      "loss": 0.5453,
+      "step": 2942
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.07304929941892624,
+      "learning_rate": 1.592204299901353e-05,
+      "loss": 0.5945,
+      "step": 2943
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.06565451622009277,
+      "learning_rate": 1.587464671688187e-05,
+      "loss": 0.5119,
+      "step": 2944
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.06776011735200882,
+      "learning_rate": 1.5827315001072318e-05,
+      "loss": 0.4697,
+      "step": 2945
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.07718344032764435,
+      "learning_rate": 1.5780047887911987e-05,
+      "loss": 0.5887,
+      "step": 2946
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.08132103085517883,
+      "learning_rate": 1.5732845413678477e-05,
+      "loss": 0.6108,
+      "step": 2947
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.07473303377628326,
+      "learning_rate": 1.568570761459972e-05,
+      "loss": 0.5892,
+      "step": 2948
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.08634821325540543,
+      "learning_rate": 1.563863452685409e-05,
+      "loss": 0.5048,
+      "step": 2949
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.08100888878107071,
+      "learning_rate": 1.5591626186570183e-05,
+      "loss": 0.5892,
+      "step": 2950
+    },
+    {
+      "epoch": 1.62,
+      "grad_norm": 0.07569902390241623,
+      "learning_rate": 1.5544682629827002e-05,
+      "loss": 0.5609,
+      "step": 2951
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.08422853797674179,
+      "learning_rate": 1.5497803892653807e-05,
+      "loss": 0.5317,
+      "step": 2952
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.0842873826622963,
+      "learning_rate": 1.5450990011030044e-05,
+      "loss": 0.5676,
+      "step": 2953
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.08060505241155624,
+      "learning_rate": 1.540424102088548e-05,
+      "loss": 0.5739,
+      "step": 2954
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.0765761137008667,
+      "learning_rate": 1.5357556958099973e-05,
+      "loss": 0.5434,
+      "step": 2955
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.07481849938631058,
+      "learning_rate": 1.531093785850366e-05,
+      "loss": 0.5044,
+      "step": 2956
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.08106032758951187,
+      "learning_rate": 1.526438375787671e-05,
+      "loss": 0.6152,
+      "step": 2957
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.08591877669095993,
+      "learning_rate": 1.521789469194952e-05,
+      "loss": 0.5657,
+      "step": 2958
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.07527689635753632,
+      "learning_rate": 1.5171470696402446e-05,
+      "loss": 0.6218,
+      "step": 2959
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.07529833912849426,
+      "learning_rate": 1.512511180686601e-05,
+      "loss": 0.589,
+      "step": 2960
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.07544808834791183,
+      "learning_rate": 1.5078818058920696e-05,
+      "loss": 0.5448,
+      "step": 2961
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.07565164566040039,
+      "learning_rate": 1.5032589488097038e-05,
+      "loss": 0.5945,
+      "step": 2962
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.0887230783700943,
+      "learning_rate": 1.498642612987553e-05,
+      "loss": 0.6773,
+      "step": 2963
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.0782867819070816,
+      "learning_rate": 1.4940328019686589e-05,
+      "loss": 0.5615,
+      "step": 2964
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.07043831050395966,
+      "learning_rate": 1.48942951929106e-05,
+      "loss": 0.5101,
+      "step": 2965
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.06438145786523819,
+      "learning_rate": 1.4848327684877794e-05,
+      "loss": 0.4303,
+      "step": 2966
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.06960533559322357,
+      "learning_rate": 1.4802425530868324e-05,
+      "loss": 0.4657,
+      "step": 2967
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.07519534230232239,
+      "learning_rate": 1.4756588766112133e-05,
+      "loss": 0.5149,
+      "step": 2968
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 0.07465105503797531,
+      "learning_rate": 1.4710817425789014e-05,
+      "loss": 0.5502,
+      "step": 2969
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.08811614662408829,
+      "learning_rate": 1.466511154502852e-05,
+      "loss": 0.6154,
+      "step": 2970
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.0730571523308754,
+      "learning_rate": 1.461947115890997e-05,
+      "loss": 0.545,
+      "step": 2971
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.07383215427398682,
+      "learning_rate": 1.4573896302462464e-05,
+      "loss": 0.5388,
+      "step": 2972
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.07530464231967926,
+      "learning_rate": 1.4528387010664701e-05,
+      "loss": 0.5037,
+      "step": 2973
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.06940647214651108,
+      "learning_rate": 1.4482943318445197e-05,
+      "loss": 0.6025,
+      "step": 2974
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.07640254497528076,
+      "learning_rate": 1.4437565260681974e-05,
+      "loss": 0.5243,
+      "step": 2975
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.07426881045103073,
+      "learning_rate": 1.4392252872202816e-05,
+      "loss": 0.5138,
+      "step": 2976
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.08832768350839615,
+      "learning_rate": 1.4347006187784995e-05,
+      "loss": 0.6679,
+      "step": 2977
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.08714038133621216,
+      "learning_rate": 1.4301825242155441e-05,
+      "loss": 0.5764,
+      "step": 2978
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.07089292258024216,
+      "learning_rate": 1.4256710069990542e-05,
+      "loss": 0.62,
+      "step": 2979
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.07467392086982727,
+      "learning_rate": 1.4211660705916285e-05,
+      "loss": 0.5115,
+      "step": 2980
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.08002919703722,
+      "learning_rate": 1.4166677184508136e-05,
+      "loss": 0.6898,
+      "step": 2981
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.07483605295419693,
+      "learning_rate": 1.4121759540290969e-05,
+      "loss": 0.5999,
+      "step": 2982
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.08835967630147934,
+      "learning_rate": 1.4076907807739182e-05,
+      "loss": 0.5828,
+      "step": 2983
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.07726197689771652,
+      "learning_rate": 1.4032122021276484e-05,
+      "loss": 0.6079,
+      "step": 2984
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.07789423316717148,
+      "learning_rate": 1.398740221527608e-05,
+      "loss": 0.5703,
+      "step": 2985
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.09219390153884888,
+      "learning_rate": 1.3942748424060425e-05,
+      "loss": 0.807,
+      "step": 2986
+    },
+    {
+      "epoch": 1.64,
+      "grad_norm": 0.07657124847173691,
+      "learning_rate": 1.3898160681901429e-05,
+      "loss": 0.5378,
+      "step": 2987
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.07728318125009537,
+      "learning_rate": 1.3853639023020181e-05,
+      "loss": 0.6365,
+      "step": 2988
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.08017481863498688,
+      "learning_rate": 1.3809183481587152e-05,
+      "loss": 0.5895,
+      "step": 2989
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.09209847450256348,
+      "learning_rate": 1.3764794091722033e-05,
+      "loss": 0.6123,
+      "step": 2990
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.07465118914842606,
+      "learning_rate": 1.3720470887493719e-05,
+      "loss": 0.5438,
+      "step": 2991
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.08009976893663406,
+      "learning_rate": 1.3676213902920354e-05,
+      "loss": 0.5507,
+      "step": 2992
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.07504164427518845,
+      "learning_rate": 1.36320231719692e-05,
+      "loss": 0.5589,
+      "step": 2993
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.0721011683344841,
+      "learning_rate": 1.3587898728556747e-05,
+      "loss": 0.4541,
+      "step": 2994
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.07778453081846237,
+      "learning_rate": 1.354384060654852e-05,
+      "loss": 0.5965,
+      "step": 2995
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.08092589676380157,
+      "learning_rate": 1.3499848839759222e-05,
+      "loss": 0.5814,
+      "step": 2996
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.078467458486557,
+      "learning_rate": 1.3455923461952558e-05,
+      "loss": 0.5764,
+      "step": 2997
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.07226195186376572,
+      "learning_rate": 1.3412064506841327e-05,
+      "loss": 0.5042,
+      "step": 2998
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.07678015530109406,
+      "learning_rate": 1.336827200808738e-05,
+      "loss": 0.5554,
+      "step": 2999
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.07457426935434341,
+      "learning_rate": 1.3324545999301452e-05,
+      "loss": 0.6258,
+      "step": 3000
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.0747811570763588,
+      "learning_rate": 1.3280886514043367e-05,
+      "loss": 0.5506,
+      "step": 3001
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.07037195563316345,
+      "learning_rate": 1.3237293585821786e-05,
+      "loss": 0.4983,
+      "step": 3002
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.06816434115171432,
+      "learning_rate": 1.3193767248094402e-05,
+      "loss": 0.4848,
+      "step": 3003
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.07628194242715836,
+      "learning_rate": 1.315030753426768e-05,
+      "loss": 0.5544,
+      "step": 3004
+    },
+    {
+      "epoch": 1.65,
+      "grad_norm": 0.08764028549194336,
+      "learning_rate": 1.3106914477697063e-05,
+      "loss": 0.5445,
+      "step": 3005
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.08041933923959732,
+      "learning_rate": 1.306358811168673e-05,
+      "loss": 0.5387,
+      "step": 3006
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.07397464662790298,
+      "learning_rate": 1.302032846948975e-05,
+      "loss": 0.4767,
+      "step": 3007
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.06808964163064957,
+      "learning_rate": 1.2977135584307964e-05,
+      "loss": 0.5131,
+      "step": 3008
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.07995598763227463,
+      "learning_rate": 1.2934009489291955e-05,
+      "loss": 0.605,
+      "step": 3009
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.0879918783903122,
+      "learning_rate": 1.2890950217541053e-05,
+      "loss": 0.5728,
+      "step": 3010
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.07948015630245209,
+      "learning_rate": 1.2847957802103317e-05,
+      "loss": 0.5891,
+      "step": 3011
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.07937068492174149,
+      "learning_rate": 1.2805032275975514e-05,
+      "loss": 0.5653,
+      "step": 3012
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.0827297791838646,
+      "learning_rate": 1.2762173672102996e-05,
+      "loss": 0.549,
+      "step": 3013
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.08054407685995102,
+      "learning_rate": 1.2719382023379834e-05,
+      "loss": 0.553,
+      "step": 3014
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.06850113719701767,
+      "learning_rate": 1.2676657362648636e-05,
+      "loss": 0.5766,
+      "step": 3015
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.08243125677108765,
+      "learning_rate": 1.2633999722700684e-05,
+      "loss": 0.6332,
+      "step": 3016
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.08675704896450043,
+      "learning_rate": 1.2591409136275723e-05,
+      "loss": 0.6286,
+      "step": 3017
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.07562907040119171,
+      "learning_rate": 1.2548885636062146e-05,
+      "loss": 0.5745,
+      "step": 3018
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.07691862434148788,
+      "learning_rate": 1.250642925469674e-05,
+      "loss": 0.7418,
+      "step": 3019
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.08036176860332489,
+      "learning_rate": 1.2464040024764855e-05,
+      "loss": 0.5572,
+      "step": 3020
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.06832167506217957,
+      "learning_rate": 1.2421717978800306e-05,
+      "loss": 0.56,
+      "step": 3021
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.07241854816675186,
+      "learning_rate": 1.2379463149285286e-05,
+      "loss": 0.5392,
+      "step": 3022
+    },
+    {
+      "epoch": 1.66,
+      "grad_norm": 0.07803373783826828,
+      "learning_rate": 1.2337275568650464e-05,
+      "loss": 0.5652,
+      "step": 3023
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.07604583352804184,
+      "learning_rate": 1.2295155269274827e-05,
+      "loss": 0.5596,
+      "step": 3024
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.07595550268888474,
+      "learning_rate": 1.2253102283485807e-05,
+      "loss": 0.6225,
+      "step": 3025
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.08902128040790558,
+      "learning_rate": 1.2211116643559083e-05,
+      "loss": 0.712,
+      "step": 3026
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.07837363332509995,
+      "learning_rate": 1.2169198381718726e-05,
+      "loss": 0.5764,
+      "step": 3027
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.0752459168434143,
+      "learning_rate": 1.2127347530137034e-05,
+      "loss": 0.5559,
+      "step": 3028
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.08141879737377167,
+      "learning_rate": 1.20855641209346e-05,
+      "loss": 0.5588,
+      "step": 3029
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.08007825911045074,
+      "learning_rate": 1.2043848186180262e-05,
+      "loss": 0.5518,
+      "step": 3030
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.07503150403499603,
+      "learning_rate": 1.2002199757891031e-05,
+      "loss": 0.5589,
+      "step": 3031
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.08165138959884644,
+      "learning_rate": 1.1960618868032159e-05,
+      "loss": 0.6055,
+      "step": 3032
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.07894934713840485,
+      "learning_rate": 1.1919105548516996e-05,
+      "loss": 0.5359,
+      "step": 3033
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.07462869584560394,
+      "learning_rate": 1.1877659831207122e-05,
+      "loss": 0.5296,
+      "step": 3034
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.08617985993623734,
+      "learning_rate": 1.1836281747912125e-05,
+      "loss": 0.5807,
+      "step": 3035
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.07207958400249481,
+      "learning_rate": 1.1794971330389792e-05,
+      "loss": 0.4785,
+      "step": 3036
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.08790726959705353,
+      "learning_rate": 1.1753728610345883e-05,
+      "loss": 0.6675,
+      "step": 3037
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.08325272798538208,
+      "learning_rate": 1.1712553619434252e-05,
+      "loss": 0.6261,
+      "step": 3038
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.07597482204437256,
+      "learning_rate": 1.167144638925679e-05,
+      "loss": 0.5318,
+      "step": 3039
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.08048231899738312,
+      "learning_rate": 1.1630406951363314e-05,
+      "loss": 0.5865,
+      "step": 3040
+    },
+    {
+      "epoch": 1.67,
+      "grad_norm": 0.08627559244632721,
+      "learning_rate": 1.1589435337251676e-05,
+      "loss": 0.6283,
+      "step": 3041
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.07512358576059341,
+      "learning_rate": 1.1548531578367604e-05,
+      "loss": 0.5793,
+      "step": 3042
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.08154304325580597,
+      "learning_rate": 1.1507695706104849e-05,
+      "loss": 0.5347,
+      "step": 3043
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.0808219239115715,
+      "learning_rate": 1.1466927751804935e-05,
+      "loss": 0.6504,
+      "step": 3044
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.07881506532430649,
+      "learning_rate": 1.1426227746757378e-05,
+      "loss": 0.5969,
+      "step": 3045
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.0822736844420433,
+      "learning_rate": 1.1385595722199438e-05,
+      "loss": 0.5548,
+      "step": 3046
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.07585962116718292,
+      "learning_rate": 1.134503170931629e-05,
+      "loss": 0.5943,
+      "step": 3047
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.08921745419502258,
+      "learning_rate": 1.130453573924083e-05,
+      "loss": 0.6943,
+      "step": 3048
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.08134070783853531,
+      "learning_rate": 1.1264107843053783e-05,
+      "loss": 0.5871,
+      "step": 3049
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.09125948697328568,
+      "learning_rate": 1.1223748051783644e-05,
+      "loss": 0.5844,
+      "step": 3050
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.07791607826948166,
+      "learning_rate": 1.1183456396406556e-05,
+      "loss": 0.6624,
+      "step": 3051
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.07797951251268387,
+      "learning_rate": 1.1143232907846478e-05,
+      "loss": 0.5632,
+      "step": 3052
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.07615441083908081,
+      "learning_rate": 1.1103077616974932e-05,
+      "loss": 0.5388,
+      "step": 3053
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.08791297674179077,
+      "learning_rate": 1.1062990554611207e-05,
+      "loss": 0.5462,
+      "step": 3054
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.0754203200340271,
+      "learning_rate": 1.1022971751522127e-05,
+      "loss": 0.5189,
+      "step": 3055
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.07922713458538055,
+      "learning_rate": 1.0983021238422242e-05,
+      "loss": 0.6262,
+      "step": 3056
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.08247672766447067,
+      "learning_rate": 1.0943139045973549e-05,
+      "loss": 0.6003,
+      "step": 3057
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.08441609889268875,
+      "learning_rate": 1.0903325204785742e-05,
+      "loss": 0.624,
+      "step": 3058
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 0.0722791850566864,
+      "learning_rate": 1.0863579745415997e-05,
+      "loss": 0.5629,
+      "step": 3059
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.08199836313724518,
+      "learning_rate": 1.082390269836896e-05,
+      "loss": 0.5925,
+      "step": 3060
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.06521271169185638,
+      "learning_rate": 1.0784294094096881e-05,
+      "loss": 0.5251,
+      "step": 3061
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.07767980545759201,
+      "learning_rate": 1.0744753962999355e-05,
+      "loss": 0.5666,
+      "step": 3062
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.07202478498220444,
+      "learning_rate": 1.0705282335423539e-05,
+      "loss": 0.4518,
+      "step": 3063
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.07885854691267014,
+      "learning_rate": 1.0665879241663922e-05,
+      "loss": 0.5496,
+      "step": 3064
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.07686977833509445,
+      "learning_rate": 1.062654471196246e-05,
+      "loss": 0.4803,
+      "step": 3065
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.08224175125360489,
+      "learning_rate": 1.0587278776508424e-05,
+      "loss": 0.5551,
+      "step": 3066
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.08237634599208832,
+      "learning_rate": 1.0548081465438497e-05,
+      "loss": 0.595,
+      "step": 3067
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.07160717993974686,
+      "learning_rate": 1.050895280883668e-05,
+      "loss": 0.5952,
+      "step": 3068
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.08529774099588394,
+      "learning_rate": 1.046989283673424e-05,
+      "loss": 0.5683,
+      "step": 3069
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.0765925943851471,
+      "learning_rate": 1.0430901579109787e-05,
+      "loss": 0.5274,
+      "step": 3070
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.07462650537490845,
+      "learning_rate": 1.0391979065889135e-05,
+      "loss": 0.5388,
+      "step": 3071
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.07412373274564743,
+      "learning_rate": 1.035312532694539e-05,
+      "loss": 0.5452,
+      "step": 3072
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.08629931509494781,
+      "learning_rate": 1.031434039209883e-05,
+      "loss": 0.6178,
+      "step": 3073
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.08814838528633118,
+      "learning_rate": 1.0275624291116958e-05,
+      "loss": 0.5785,
+      "step": 3074
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.07783772051334381,
+      "learning_rate": 1.0236977053714414e-05,
+      "loss": 0.5858,
+      "step": 3075
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.07740164548158646,
+      "learning_rate": 1.0198398709553025e-05,
+      "loss": 0.5905,
+      "step": 3076
+    },
+    {
+      "epoch": 1.69,
+      "grad_norm": 0.07513068616390228,
+      "learning_rate": 1.0159889288241731e-05,
+      "loss": 0.5925,
+      "step": 3077
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.07727821171283722,
+      "learning_rate": 1.0121448819336532e-05,
+      "loss": 0.4903,
+      "step": 3078
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.07482651621103287,
+      "learning_rate": 1.0083077332340562e-05,
+      "loss": 0.6038,
+      "step": 3079
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.08040370792150497,
+      "learning_rate": 1.0044774856703976e-05,
+      "loss": 0.5698,
+      "step": 3080
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.07068987935781479,
+      "learning_rate": 1.0006541421824012e-05,
+      "loss": 0.5326,
+      "step": 3081
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.08372973650693893,
+      "learning_rate": 9.968377057044831e-06,
+      "loss": 0.6151,
+      "step": 3082
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.07655537128448486,
+      "learning_rate": 9.93028179165768e-06,
+      "loss": 0.5554,
+      "step": 3083
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.07460279017686844,
+      "learning_rate": 9.892255654900695e-06,
+      "loss": 0.5039,
+      "step": 3084
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.07027482986450195,
+      "learning_rate": 9.854298675959007e-06,
+      "loss": 0.548,
+      "step": 3085
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.09165271371603012,
+      "learning_rate": 9.816410883964623e-06,
+      "loss": 0.575,
+      "step": 3086
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.08436719328165054,
+      "learning_rate": 9.778592307996504e-06,
+      "loss": 0.6398,
+      "step": 3087
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.07201489061117172,
+      "learning_rate": 9.740842977080433e-06,
+      "loss": 0.5213,
+      "step": 3088
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.0690862387418747,
+      "learning_rate": 9.703162920189079e-06,
+      "loss": 0.4581,
+      "step": 3089
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.0757426768541336,
+      "learning_rate": 9.665552166241964e-06,
+      "loss": 0.5442,
+      "step": 3090
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.07909826934337616,
+      "learning_rate": 9.628010744105353e-06,
+      "loss": 0.5935,
+      "step": 3091
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.07645463198423386,
+      "learning_rate": 9.590538682592376e-06,
+      "loss": 0.5586,
+      "step": 3092
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.07282956689596176,
+      "learning_rate": 9.55313601046285e-06,
+      "loss": 0.5714,
+      "step": 3093
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.06841188669204712,
+      "learning_rate": 9.515802756423409e-06,
+      "loss": 0.5364,
+      "step": 3094
+    },
+    {
+      "epoch": 1.7,
+      "grad_norm": 0.08608322590589523,
+      "learning_rate": 9.478538949127346e-06,
+      "loss": 0.5332,
+      "step": 3095
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.08561312407255173,
+      "learning_rate": 9.441344617174718e-06,
+      "loss": 0.6552,
+      "step": 3096
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07228785008192062,
+      "learning_rate": 9.40421978911219e-06,
+      "loss": 0.5415,
+      "step": 3097
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07194254547357559,
+      "learning_rate": 9.367164493433133e-06,
+      "loss": 0.5126,
+      "step": 3098
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07450117915868759,
+      "learning_rate": 9.330178758577568e-06,
+      "loss": 0.5843,
+      "step": 3099
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07577648013830185,
+      "learning_rate": 9.29326261293203e-06,
+      "loss": 0.5851,
+      "step": 3100
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07993178069591522,
+      "learning_rate": 9.256416084829778e-06,
+      "loss": 0.527,
+      "step": 3101
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07520022988319397,
+      "learning_rate": 9.219639202550523e-06,
+      "loss": 0.5325,
+      "step": 3102
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.08247778564691544,
+      "learning_rate": 9.18293199432061e-06,
+      "loss": 0.533,
+      "step": 3103
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07813543826341629,
+      "learning_rate": 9.146294488312823e-06,
+      "loss": 0.5314,
+      "step": 3104
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.075026735663414,
+      "learning_rate": 9.109726712646548e-06,
+      "loss": 0.5245,
+      "step": 3105
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07687216252088547,
+      "learning_rate": 9.07322869538756e-06,
+      "loss": 0.5349,
+      "step": 3106
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07420308142900467,
+      "learning_rate": 9.036800464548157e-06,
+      "loss": 0.5751,
+      "step": 3107
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07741032540798187,
+      "learning_rate": 9.000442048087076e-06,
+      "loss": 0.5099,
+      "step": 3108
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.08252329379320145,
+      "learning_rate": 8.964153473909397e-06,
+      "loss": 0.5713,
+      "step": 3109
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07514884322881699,
+      "learning_rate": 8.927934769866719e-06,
+      "loss": 0.5836,
+      "step": 3110
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07817213237285614,
+      "learning_rate": 8.891785963756872e-06,
+      "loss": 0.66,
+      "step": 3111
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.0779666155576706,
+      "learning_rate": 8.855707083324183e-06,
+      "loss": 0.6034,
+      "step": 3112
+    },
+    {
+      "epoch": 1.71,
+      "grad_norm": 0.07593736797571182,
+      "learning_rate": 8.819698156259182e-06,
+      "loss": 0.5346,
+      "step": 3113
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.08052858710289001,
+      "learning_rate": 8.783759210198805e-06,
+      "loss": 0.5325,
+      "step": 3114
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.08734001964330673,
+      "learning_rate": 8.747890272726222e-06,
+      "loss": 0.6829,
+      "step": 3115
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.08773903548717499,
+      "learning_rate": 8.712091371370912e-06,
+      "loss": 0.5986,
+      "step": 3116
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.07741262018680573,
+      "learning_rate": 8.676362533608573e-06,
+      "loss": 0.474,
+      "step": 3117
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.08860704302787781,
+      "learning_rate": 8.640703786861116e-06,
+      "loss": 0.5966,
+      "step": 3118
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.0868702232837677,
+      "learning_rate": 8.605115158496713e-06,
+      "loss": 0.566,
+      "step": 3119
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.07514653354883194,
+      "learning_rate": 8.56959667582965e-06,
+      "loss": 0.5436,
+      "step": 3120
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.06962697952985764,
+      "learning_rate": 8.534148366120432e-06,
+      "loss": 0.4152,
+      "step": 3121
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.0840853601694107,
+      "learning_rate": 8.498770256575662e-06,
+      "loss": 0.642,
+      "step": 3122
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.07120905071496964,
+      "learning_rate": 8.46346237434813e-06,
+      "loss": 0.5534,
+      "step": 3123
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.0809093713760376,
+      "learning_rate": 8.428224746536627e-06,
+      "loss": 0.6058,
+      "step": 3124
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.07151578366756439,
+      "learning_rate": 8.39305740018611e-06,
+      "loss": 0.5478,
+      "step": 3125
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.07958785444498062,
+      "learning_rate": 8.357960362287587e-06,
+      "loss": 0.5557,
+      "step": 3126
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.08158773183822632,
+      "learning_rate": 8.322933659778032e-06,
+      "loss": 0.5649,
+      "step": 3127
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.08477183431386948,
+      "learning_rate": 8.287977319540541e-06,
+      "loss": 0.5418,
+      "step": 3128
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.08409066498279572,
+      "learning_rate": 8.253091368404098e-06,
+      "loss": 0.5837,
+      "step": 3129
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.07809256762266159,
+      "learning_rate": 8.218275833143752e-06,
+      "loss": 0.5635,
+      "step": 3130
+    },
+    {
+      "epoch": 1.72,
+      "grad_norm": 0.07307461649179459,
+      "learning_rate": 8.18353074048046e-06,
+      "loss": 0.583,
+      "step": 3131
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.09218110144138336,
+      "learning_rate": 8.14885611708115e-06,
+      "loss": 0.5929,
+      "step": 3132
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.07997137308120728,
+      "learning_rate": 8.114251989558596e-06,
+      "loss": 0.5021,
+      "step": 3133
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.07108749449253082,
+      "learning_rate": 8.079718384471557e-06,
+      "loss": 0.5638,
+      "step": 3134
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.08197928965091705,
+      "learning_rate": 8.045255328324596e-06,
+      "loss": 0.58,
+      "step": 3135
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.08596963435411453,
+      "learning_rate": 8.010862847568168e-06,
+      "loss": 0.5852,
+      "step": 3136
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.07822061330080032,
+      "learning_rate": 7.976540968598555e-06,
+      "loss": 0.5436,
+      "step": 3137
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.07026013731956482,
+      "learning_rate": 7.942289717757812e-06,
+      "loss": 0.5479,
+      "step": 3138
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.07042836397886276,
+      "learning_rate": 7.908109121333873e-06,
+      "loss": 0.4859,
+      "step": 3139
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.073862724006176,
+      "learning_rate": 7.873999205560334e-06,
+      "loss": 0.6089,
+      "step": 3140
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.07902369648218155,
+      "learning_rate": 7.839959996616652e-06,
+      "loss": 0.6239,
+      "step": 3141
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.07941651344299316,
+      "learning_rate": 7.80599152062792e-06,
+      "loss": 0.5479,
+      "step": 3142
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.07300753891468048,
+      "learning_rate": 7.772093803665037e-06,
+      "loss": 0.5183,
+      "step": 3143
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.08596652001142502,
+      "learning_rate": 7.738266871744494e-06,
+      "loss": 0.6281,
+      "step": 3144
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.07829903066158295,
+      "learning_rate": 7.704510750828542e-06,
+      "loss": 0.5724,
+      "step": 3145
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.08107545226812363,
+      "learning_rate": 7.67082546682506e-06,
+      "loss": 0.5444,
+      "step": 3146
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.07941924035549164,
+      "learning_rate": 7.637211045587512e-06,
+      "loss": 0.6107,
+      "step": 3147
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.08565372228622437,
+      "learning_rate": 7.603667512915025e-06,
+      "loss": 0.5424,
+      "step": 3148
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 0.061931777745485306,
+      "learning_rate": 7.570194894552307e-06,
+      "loss": 0.4901,
+      "step": 3149
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.07151300460100174,
+      "learning_rate": 7.536793216189675e-06,
+      "loss": 0.5057,
+      "step": 3150
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.22302797436714172,
+      "learning_rate": 7.5034625034628995e-06,
+      "loss": 0.6756,
+      "step": 3151
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.0780290961265564,
+      "learning_rate": 7.470202781953395e-06,
+      "loss": 0.675,
+      "step": 3152
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.07317093014717102,
+      "learning_rate": 7.437014077188009e-06,
+      "loss": 0.586,
+      "step": 3153
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.08226374536752701,
+      "learning_rate": 7.403896414639144e-06,
+      "loss": 0.5778,
+      "step": 3154
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.07802766561508179,
+      "learning_rate": 7.370849819724634e-06,
+      "loss": 0.6207,
+      "step": 3155
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.07580267637968063,
+      "learning_rate": 7.337874317807802e-06,
+      "loss": 0.5755,
+      "step": 3156
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.08161929994821548,
+      "learning_rate": 7.304969934197359e-06,
+      "loss": 0.5729,
+      "step": 3157
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.07730533182621002,
+      "learning_rate": 7.27213669414748e-06,
+      "loss": 0.5366,
+      "step": 3158
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.07133332639932632,
+      "learning_rate": 7.239374622857742e-06,
+      "loss": 0.6159,
+      "step": 3159
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.08000887930393219,
+      "learning_rate": 7.206683745473053e-06,
+      "loss": 0.5803,
+      "step": 3160
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.07099801301956177,
+      "learning_rate": 7.17406408708372e-06,
+      "loss": 0.5313,
+      "step": 3161
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.07111170887947083,
+      "learning_rate": 7.141515672725363e-06,
+      "loss": 0.4635,
+      "step": 3162
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.0841594710946083,
+      "learning_rate": 7.109038527378942e-06,
+      "loss": 0.5765,
+      "step": 3163
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.07840998470783234,
+      "learning_rate": 7.076632675970707e-06,
+      "loss": 0.5434,
+      "step": 3164
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.07809460163116455,
+      "learning_rate": 7.044298143372197e-06,
+      "loss": 0.5752,
+      "step": 3165
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.0699603483080864,
+      "learning_rate": 7.012034954400193e-06,
+      "loss": 0.4794,
+      "step": 3166
+    },
+    {
+      "epoch": 1.74,
+      "grad_norm": 0.07665000855922699,
+      "learning_rate": 6.979843133816743e-06,
+      "loss": 0.5516,
+      "step": 3167
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07922467589378357,
+      "learning_rate": 6.9477227063291405e-06,
+      "loss": 0.6229,
+      "step": 3168
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07732383906841278,
+      "learning_rate": 6.9156736965898085e-06,
+      "loss": 0.6062,
+      "step": 3169
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.0833093523979187,
+      "learning_rate": 6.88369612919646e-06,
+      "loss": 0.6237,
+      "step": 3170
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07883663475513458,
+      "learning_rate": 6.8517900286918735e-06,
+      "loss": 0.5874,
+      "step": 3171
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.08125128597021103,
+      "learning_rate": 6.81995541956405e-06,
+      "loss": 0.6341,
+      "step": 3172
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07269998639822006,
+      "learning_rate": 6.788192326246079e-06,
+      "loss": 0.5798,
+      "step": 3173
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07897648215293884,
+      "learning_rate": 6.756500773116203e-06,
+      "loss": 0.6475,
+      "step": 3174
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.08471954613924026,
+      "learning_rate": 6.724880784497689e-06,
+      "loss": 0.5701,
+      "step": 3175
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07630597054958344,
+      "learning_rate": 6.6933323846589566e-06,
+      "loss": 0.5155,
+      "step": 3176
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07588117569684982,
+      "learning_rate": 6.661855597813449e-06,
+      "loss": 0.6294,
+      "step": 3177
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.0732317864894867,
+      "learning_rate": 6.630450448119618e-06,
+      "loss": 0.5349,
+      "step": 3178
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07867201417684555,
+      "learning_rate": 6.599116959680973e-06,
+      "loss": 0.5625,
+      "step": 3179
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07942135632038116,
+      "learning_rate": 6.567855156545999e-06,
+      "loss": 0.5349,
+      "step": 3180
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07216234505176544,
+      "learning_rate": 6.536665062708192e-06,
+      "loss": 0.4608,
+      "step": 3181
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07037379592657089,
+      "learning_rate": 6.505546702105958e-06,
+      "loss": 0.5011,
+      "step": 3182
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.08123207092285156,
+      "learning_rate": 6.4745000986227155e-06,
+      "loss": 0.636,
+      "step": 3183
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07815258204936981,
+      "learning_rate": 6.443525276086748e-06,
+      "loss": 0.5994,
+      "step": 3184
+    },
+    {
+      "epoch": 1.75,
+      "grad_norm": 0.07790284603834152,
+      "learning_rate": 6.412622258271284e-06,
+      "loss": 0.5094,
+      "step": 3185
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.08153413236141205,
+      "learning_rate": 6.381791068894438e-06,
+      "loss": 0.54,
+      "step": 3186
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.07703083008527756,
+      "learning_rate": 6.3510317316191725e-06,
+      "loss": 0.5901,
+      "step": 3187
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.08073649555444717,
+      "learning_rate": 6.320344270053357e-06,
+      "loss": 0.542,
+      "step": 3188
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.07454603165388107,
+      "learning_rate": 6.289728707749609e-06,
+      "loss": 0.5095,
+      "step": 3189
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.07908552139997482,
+      "learning_rate": 6.2591850682054535e-06,
+      "loss": 0.5629,
+      "step": 3190
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.07613983005285263,
+      "learning_rate": 6.228713374863137e-06,
+      "loss": 0.6605,
+      "step": 3191
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.08534809201955795,
+      "learning_rate": 6.198313651109777e-06,
+      "loss": 0.5604,
+      "step": 3192
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.07495297491550446,
+      "learning_rate": 6.167985920277153e-06,
+      "loss": 0.5866,
+      "step": 3193
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.07653788477182388,
+      "learning_rate": 6.137730205641856e-06,
+      "loss": 0.6514,
+      "step": 3194
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.08158232271671295,
+      "learning_rate": 6.107546530425212e-06,
+      "loss": 0.6196,
+      "step": 3195
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.07958748936653137,
+      "learning_rate": 6.0774349177932014e-06,
+      "loss": 0.5411,
+      "step": 3196
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.07684215903282166,
+      "learning_rate": 6.047395390856547e-06,
+      "loss": 0.5609,
+      "step": 3197
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.07646127045154572,
+      "learning_rate": 6.017427972670608e-06,
+      "loss": 0.5212,
+      "step": 3198
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.07235292345285416,
+      "learning_rate": 5.987532686235431e-06,
+      "loss": 0.5087,
+      "step": 3199
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.07906308770179749,
+      "learning_rate": 5.957709554495683e-06,
+      "loss": 0.4798,
+      "step": 3200
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.07810641825199127,
+      "learning_rate": 5.927958600340666e-06,
+      "loss": 0.5738,
+      "step": 3201
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.08186232298612595,
+      "learning_rate": 5.89827984660426e-06,
+      "loss": 0.5817,
+      "step": 3202
+    },
+    {
+      "epoch": 1.76,
+      "grad_norm": 0.0750940591096878,
+      "learning_rate": 5.868673316064965e-06,
+      "loss": 0.5311,
+      "step": 3203
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.07697701454162598,
+      "learning_rate": 5.8391390314458395e-06,
+      "loss": 0.5718,
+      "step": 3204
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.07784340530633926,
+      "learning_rate": 5.809677015414461e-06,
+      "loss": 0.48,
+      "step": 3205
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.07141119986772537,
+      "learning_rate": 5.780287290582997e-06,
+      "loss": 0.5174,
+      "step": 3206
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.08472003042697906,
+      "learning_rate": 5.7509698795080726e-06,
+      "loss": 0.6321,
+      "step": 3207
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.08304649591445923,
+      "learning_rate": 5.721724804690853e-06,
+      "loss": 0.5654,
+      "step": 3208
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.06623322516679764,
+      "learning_rate": 5.6925520885769745e-06,
+      "loss": 0.5074,
+      "step": 3209
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.07251337170600891,
+      "learning_rate": 5.663451753556537e-06,
+      "loss": 0.5376,
+      "step": 3210
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.08150351792573929,
+      "learning_rate": 5.634423821964074e-06,
+      "loss": 0.5113,
+      "step": 3211
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.07414893060922623,
+      "learning_rate": 5.605468316078588e-06,
+      "loss": 0.6034,
+      "step": 3212
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.07834034413099289,
+      "learning_rate": 5.5765852581234455e-06,
+      "loss": 0.5592,
+      "step": 3213
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.09270931035280228,
+      "learning_rate": 5.547774670266426e-06,
+      "loss": 0.5344,
+      "step": 3214
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.07382823526859283,
+      "learning_rate": 5.519036574619729e-06,
+      "loss": 0.5711,
+      "step": 3215
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.07459236681461334,
+      "learning_rate": 5.490370993239846e-06,
+      "loss": 0.5897,
+      "step": 3216
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.09094478189945221,
+      "learning_rate": 5.4617779481276665e-06,
+      "loss": 0.5864,
+      "step": 3217
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.07464843988418579,
+      "learning_rate": 5.43325746122838e-06,
+      "loss": 0.5389,
+      "step": 3218
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.08413833379745483,
+      "learning_rate": 5.404809554431534e-06,
+      "loss": 0.5677,
+      "step": 3219
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.07826337218284607,
+      "learning_rate": 5.376434249570883e-06,
+      "loss": 0.5155,
+      "step": 3220
+    },
+    {
+      "epoch": 1.77,
+      "grad_norm": 0.07135774940252304,
+      "learning_rate": 5.348131568424563e-06,
+      "loss": 0.5348,
+      "step": 3221
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.07859191298484802,
+      "learning_rate": 5.319901532714877e-06,
+      "loss": 0.4778,
+      "step": 3222
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.07505518943071365,
+      "learning_rate": 5.2917441641084365e-06,
+      "loss": 0.6227,
+      "step": 3223
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.09244594722986221,
+      "learning_rate": 5.2636594842160545e-06,
+      "loss": 0.6427,
+      "step": 3224
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.08588739484548569,
+      "learning_rate": 5.235647514592779e-06,
+      "loss": 0.646,
+      "step": 3225
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.08701588958501816,
+      "learning_rate": 5.207708276737789e-06,
+      "loss": 0.5213,
+      "step": 3226
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.07898075133562088,
+      "learning_rate": 5.1798417920945305e-06,
+      "loss": 0.547,
+      "step": 3227
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.0855628252029419,
+      "learning_rate": 5.152048082050564e-06,
+      "loss": 0.5389,
+      "step": 3228
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.07486437261104584,
+      "learning_rate": 5.124327167937571e-06,
+      "loss": 0.5612,
+      "step": 3229
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.0741388276219368,
+      "learning_rate": 5.096679071031429e-06,
+      "loss": 0.4953,
+      "step": 3230
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.08409564197063446,
+      "learning_rate": 5.069103812552045e-06,
+      "loss": 0.5383,
+      "step": 3231
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.07952644675970078,
+      "learning_rate": 5.041601413663511e-06,
+      "loss": 0.4607,
+      "step": 3232
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.07196789979934692,
+      "learning_rate": 5.014171895473929e-06,
+      "loss": 0.5096,
+      "step": 3233
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.06814949959516525,
+      "learning_rate": 4.986815279035506e-06,
+      "loss": 0.4978,
+      "step": 3234
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.06847819685935974,
+      "learning_rate": 4.959531585344457e-06,
+      "loss": 0.528,
+      "step": 3235
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.08616593480110168,
+      "learning_rate": 4.932320835341075e-06,
+      "loss": 0.5765,
+      "step": 3236
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.0824628472328186,
+      "learning_rate": 4.9051830499096565e-06,
+      "loss": 0.5438,
+      "step": 3237
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.07347706705331802,
+      "learning_rate": 4.878118249878461e-06,
+      "loss": 0.5409,
+      "step": 3238
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 0.08699614554643631,
+      "learning_rate": 4.851126456019784e-06,
+      "loss": 0.6104,
+      "step": 3239
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.08858591318130493,
+      "learning_rate": 4.8242076890498315e-06,
+      "loss": 0.6004,
+      "step": 3240
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.08170289546251297,
+      "learning_rate": 4.7973619696288445e-06,
+      "loss": 0.6143,
+      "step": 3241
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.08804669231176376,
+      "learning_rate": 4.770589318360896e-06,
+      "loss": 0.6162,
+      "step": 3242
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.07593541592359543,
+      "learning_rate": 4.743889755794062e-06,
+      "loss": 0.5595,
+      "step": 3243
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.08401387929916382,
+      "learning_rate": 4.717263302420283e-06,
+      "loss": 0.5798,
+      "step": 3244
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.0785154476761818,
+      "learning_rate": 4.690709978675401e-06,
+      "loss": 0.5527,
+      "step": 3245
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.07624941319227219,
+      "learning_rate": 4.664229804939135e-06,
+      "loss": 0.5307,
+      "step": 3246
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.08080951869487762,
+      "learning_rate": 4.63782280153503e-06,
+      "loss": 0.5225,
+      "step": 3247
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.07695840299129486,
+      "learning_rate": 4.611488988730528e-06,
+      "loss": 0.5664,
+      "step": 3248
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.07765883207321167,
+      "learning_rate": 4.58522838673684e-06,
+      "loss": 0.4966,
+      "step": 3249
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.08841906487941742,
+      "learning_rate": 4.559041015709042e-06,
+      "loss": 0.6194,
+      "step": 3250
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.0755036398768425,
+      "learning_rate": 4.532926895745937e-06,
+      "loss": 0.5943,
+      "step": 3251
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.07947423309087753,
+      "learning_rate": 4.5068860468901905e-06,
+      "loss": 0.5331,
+      "step": 3252
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.08045419305562973,
+      "learning_rate": 4.4809184891281495e-06,
+      "loss": 0.6534,
+      "step": 3253
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.07684405148029327,
+      "learning_rate": 4.455024242389972e-06,
+      "loss": 0.502,
+      "step": 3254
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.07834628224372864,
+      "learning_rate": 4.429203326549525e-06,
+      "loss": 0.5875,
+      "step": 3255
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.08634544163942337,
+      "learning_rate": 4.403455761424391e-06,
+      "loss": 0.5447,
+      "step": 3256
+    },
+    {
+      "epoch": 1.79,
+      "grad_norm": 0.08276887238025665,
+      "learning_rate": 4.377781566775874e-06,
+      "loss": 0.6446,
+      "step": 3257
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.07752779871225357,
+      "learning_rate": 4.352180762308933e-06,
+      "loss": 0.6086,
+      "step": 3258
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.08547976613044739,
+      "learning_rate": 4.32665336767224e-06,
+      "loss": 0.5989,
+      "step": 3259
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.08702553063631058,
+      "learning_rate": 4.3011994024580785e-06,
+      "loss": 0.6573,
+      "step": 3260
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.08295542746782303,
+      "learning_rate": 4.2758188862024425e-06,
+      "loss": 0.538,
+      "step": 3261
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.07643800973892212,
+      "learning_rate": 4.250511838384863e-06,
+      "loss": 0.5697,
+      "step": 3262
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.07573480159044266,
+      "learning_rate": 4.225278278428568e-06,
+      "loss": 0.5794,
+      "step": 3263
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.07962439954280853,
+      "learning_rate": 4.200118225700345e-06,
+      "loss": 0.6272,
+      "step": 3264
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.07286319136619568,
+      "learning_rate": 4.1750316995105455e-06,
+      "loss": 0.5384,
+      "step": 3265
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.07239627093076706,
+      "learning_rate": 4.1500187191131466e-06,
+      "loss": 0.5269,
+      "step": 3266
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.07041292637586594,
+      "learning_rate": 4.1250793037056145e-06,
+      "loss": 0.5506,
+      "step": 3267
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.0758143737912178,
+      "learning_rate": 4.100213472429015e-06,
+      "loss": 0.5682,
+      "step": 3268
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.08220957219600677,
+      "learning_rate": 4.075421244367861e-06,
+      "loss": 0.6196,
+      "step": 3269
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.07229150831699371,
+      "learning_rate": 4.050702638550275e-06,
+      "loss": 0.5414,
+      "step": 3270
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.08285313099622726,
+      "learning_rate": 4.026057673947769e-06,
+      "loss": 0.5856,
+      "step": 3271
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.08436094224452972,
+      "learning_rate": 4.001486369475416e-06,
+      "loss": 0.6143,
+      "step": 3272
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.08027667552232742,
+      "learning_rate": 3.976988743991739e-06,
+      "loss": 0.5629,
+      "step": 3273
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.07529149204492569,
+      "learning_rate": 3.952564816298665e-06,
+      "loss": 0.503,
+      "step": 3274
+    },
+    {
+      "epoch": 1.8,
+      "grad_norm": 0.0745483860373497,
+      "learning_rate": 3.9282146051416115e-06,
+      "loss": 0.5342,
+      "step": 3275
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.0774783343076706,
+      "learning_rate": 3.903938129209395e-06,
+      "loss": 0.5474,
+      "step": 3276
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.08215305954217911,
+      "learning_rate": 3.879735407134244e-06,
+      "loss": 0.5585,
+      "step": 3277
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.08368073403835297,
+      "learning_rate": 3.855606457491767e-06,
+      "loss": 0.6391,
+      "step": 3278
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.07771727442741394,
+      "learning_rate": 3.831551298800995e-06,
+      "loss": 0.5553,
+      "step": 3279
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.08193734288215637,
+      "learning_rate": 3.807569949524259e-06,
+      "loss": 0.5824,
+      "step": 3280
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.0646643117070198,
+      "learning_rate": 3.7836624280672916e-06,
+      "loss": 0.5098,
+      "step": 3281
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.0763091966509819,
+      "learning_rate": 3.75982875277916e-06,
+      "loss": 0.5041,
+      "step": 3282
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.08347593247890472,
+      "learning_rate": 3.736068941952231e-06,
+      "loss": 0.6356,
+      "step": 3283
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.07270161807537079,
+      "learning_rate": 3.712383013822196e-06,
+      "loss": 0.5294,
+      "step": 3284
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.07627815008163452,
+      "learning_rate": 3.6887709865680353e-06,
+      "loss": 0.4897,
+      "step": 3285
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.07596608996391296,
+      "learning_rate": 3.665232878311997e-06,
+      "loss": 0.5376,
+      "step": 3286
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.09310761094093323,
+      "learning_rate": 3.6417687071196304e-06,
+      "loss": 0.6365,
+      "step": 3287
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.062461577355861664,
+      "learning_rate": 3.6183784909997187e-06,
+      "loss": 0.475,
+      "step": 3288
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.07654763758182526,
+      "learning_rate": 3.5950622479042683e-06,
+      "loss": 0.4855,
+      "step": 3289
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.07452768087387085,
+      "learning_rate": 3.571819995728554e-06,
+      "loss": 0.5316,
+      "step": 3290
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.07540355622768402,
+      "learning_rate": 3.5486517523110054e-06,
+      "loss": 0.5403,
+      "step": 3291
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.09252862632274628,
+      "learning_rate": 3.5255575354332993e-06,
+      "loss": 0.6733,
+      "step": 3292
+    },
+    {
+      "epoch": 1.81,
+      "grad_norm": 0.08389545232057571,
+      "learning_rate": 3.5025373628202685e-06,
+      "loss": 0.5625,
+      "step": 3293
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.0911962017416954,
+      "learning_rate": 3.4795912521399245e-06,
+      "loss": 0.6122,
+      "step": 3294
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.07891615480184555,
+      "learning_rate": 3.456719221003457e-06,
+      "loss": 0.6536,
+      "step": 3295
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.07303906232118607,
+      "learning_rate": 3.4339212869651582e-06,
+      "loss": 0.5028,
+      "step": 3296
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.09300211817026138,
+      "learning_rate": 3.4111974675224976e-06,
+      "loss": 0.6058,
+      "step": 3297
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.07399295270442963,
+      "learning_rate": 3.388547780116003e-06,
+      "loss": 0.5205,
+      "step": 3298
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.07781734317541122,
+      "learning_rate": 3.3659722421293783e-06,
+      "loss": 0.5615,
+      "step": 3299
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.07778247445821762,
+      "learning_rate": 3.3434708708893425e-06,
+      "loss": 0.6124,
+      "step": 3300
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.080428346991539,
+      "learning_rate": 3.32104368366577e-06,
+      "loss": 0.5574,
+      "step": 3301
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.08679382503032684,
+      "learning_rate": 3.2986906976715136e-06,
+      "loss": 0.5437,
+      "step": 3302
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.07860066741704941,
+      "learning_rate": 3.276411930062551e-06,
+      "loss": 0.5408,
+      "step": 3303
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.0861663743853569,
+      "learning_rate": 3.2542073979378473e-06,
+      "loss": 0.6513,
+      "step": 3304
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.07389487326145172,
+      "learning_rate": 3.232077118339416e-06,
+      "loss": 0.5688,
+      "step": 3305
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.09026335924863815,
+      "learning_rate": 3.2100211082523037e-06,
+      "loss": 0.664,
+      "step": 3306
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.0709616020321846,
+      "learning_rate": 3.18803938460448e-06,
+      "loss": 0.4946,
+      "step": 3307
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.07391082495450974,
+      "learning_rate": 3.166131964266983e-06,
+      "loss": 0.5226,
+      "step": 3308
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.0852513313293457,
+      "learning_rate": 3.144298864053774e-06,
+      "loss": 0.6249,
+      "step": 3309
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.0745469480752945,
+      "learning_rate": 3.1225401007217936e-06,
+      "loss": 0.5212,
+      "step": 3310
+    },
+    {
+      "epoch": 1.82,
+      "grad_norm": 0.0812850072979927,
+      "learning_rate": 3.1008556909709054e-06,
+      "loss": 0.676,
+      "step": 3311
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.06535523384809494,
+      "learning_rate": 3.0792456514439518e-06,
+      "loss": 0.4669,
+      "step": 3312
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.07964853197336197,
+      "learning_rate": 3.0577099987266323e-06,
+      "loss": 0.5554,
+      "step": 3313
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.08415551483631134,
+      "learning_rate": 3.0362487493476142e-06,
+      "loss": 0.5914,
+      "step": 3314
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.08151557296514511,
+      "learning_rate": 3.014861919778433e-06,
+      "loss": 0.6269,
+      "step": 3315
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.08608482778072357,
+      "learning_rate": 2.9935495264334925e-06,
+      "loss": 0.5265,
+      "step": 3316
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.08237703889608383,
+      "learning_rate": 2.9723115856701177e-06,
+      "loss": 0.5901,
+      "step": 3317
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.07674803584814072,
+      "learning_rate": 2.9511481137884157e-06,
+      "loss": 0.5535,
+      "step": 3318
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.0770275890827179,
+      "learning_rate": 2.930059127031415e-06,
+      "loss": 0.5352,
+      "step": 3319
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.08275414258241653,
+      "learning_rate": 2.9090446415849017e-06,
+      "loss": 0.5957,
+      "step": 3320
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.07877998799085617,
+      "learning_rate": 2.8881046735775742e-06,
+      "loss": 0.5577,
+      "step": 3321
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.08630228042602539,
+      "learning_rate": 2.8672392390808323e-06,
+      "loss": 0.5732,
+      "step": 3322
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.07677796483039856,
+      "learning_rate": 2.846448354108955e-06,
+      "loss": 0.5038,
+      "step": 3323
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.07356994599103928,
+      "learning_rate": 2.8257320346189776e-06,
+      "loss": 0.588,
+      "step": 3324
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.07670172303915024,
+      "learning_rate": 2.8050902965106927e-06,
+      "loss": 0.5243,
+      "step": 3325
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.07875744998455048,
+      "learning_rate": 2.7845231556266617e-06,
+      "loss": 0.5418,
+      "step": 3326
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.07886647433042526,
+      "learning_rate": 2.764030627752201e-06,
+      "loss": 0.6001,
+      "step": 3327
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.07417932897806168,
+      "learning_rate": 2.7436127286153524e-06,
+      "loss": 0.5922,
+      "step": 3328
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 0.08960950374603271,
+      "learning_rate": 2.723269473886858e-06,
+      "loss": 0.721,
+      "step": 3329
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.089479461312294,
+      "learning_rate": 2.7030008791802284e-06,
+      "loss": 0.5886,
+      "step": 3330
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.0802077129483223,
+      "learning_rate": 2.6828069600516205e-06,
+      "loss": 0.5378,
+      "step": 3331
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.06853347271680832,
+      "learning_rate": 2.66268773199988e-06,
+      "loss": 0.478,
+      "step": 3332
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.0751049593091011,
+      "learning_rate": 2.6426432104665777e-06,
+      "loss": 0.5244,
+      "step": 3333
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.07996946573257446,
+      "learning_rate": 2.6226734108358743e-06,
+      "loss": 0.5685,
+      "step": 3334
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.06483065336942673,
+      "learning_rate": 2.6027783484346424e-06,
+      "loss": 0.4836,
+      "step": 3335
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.08791980147361755,
+      "learning_rate": 2.582958038532357e-06,
+      "loss": 0.6577,
+      "step": 3336
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.09538288414478302,
+      "learning_rate": 2.5632124963411386e-06,
+      "loss": 0.6429,
+      "step": 3337
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.08513102680444717,
+      "learning_rate": 2.543541737015709e-06,
+      "loss": 0.5725,
+      "step": 3338
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.08426357060670853,
+      "learning_rate": 2.5239457756534133e-06,
+      "loss": 0.5865,
+      "step": 3339
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.086363285779953,
+      "learning_rate": 2.5044246272941663e-06,
+      "loss": 0.5069,
+      "step": 3340
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.09307163953781128,
+      "learning_rate": 2.4849783069204824e-06,
+      "loss": 0.6304,
+      "step": 3341
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.07800079882144928,
+      "learning_rate": 2.465606829457445e-06,
+      "loss": 0.6352,
+      "step": 3342
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.07679854333400726,
+      "learning_rate": 2.446310209772684e-06,
+      "loss": 0.5483,
+      "step": 3343
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.07857970893383026,
+      "learning_rate": 2.4270884626763858e-06,
+      "loss": 0.57,
+      "step": 3344
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.07641496509313583,
+      "learning_rate": 2.407941602921249e-06,
+      "loss": 0.5747,
+      "step": 3345
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.07856767624616623,
+      "learning_rate": 2.3888696452025404e-06,
+      "loss": 0.5282,
+      "step": 3346
+    },
+    {
+      "epoch": 1.84,
+      "grad_norm": 0.07555583864450455,
+      "learning_rate": 2.3698726041579853e-06,
+      "loss": 0.5985,
+      "step": 3347
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.07063550502061844,
+      "learning_rate": 2.350950494367865e-06,
+      "loss": 0.5288,
+      "step": 3348
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.07329586893320084,
+      "learning_rate": 2.3321033303548955e-06,
+      "loss": 0.5528,
+      "step": 3349
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.09025662392377853,
+      "learning_rate": 2.313331126584328e-06,
+      "loss": 0.6068,
+      "step": 3350
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.09756608307361603,
+      "learning_rate": 2.2946338974638493e-06,
+      "loss": 0.6122,
+      "step": 3351
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.07663445174694061,
+      "learning_rate": 2.2760116573435906e-06,
+      "loss": 0.5624,
+      "step": 3352
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.0778941959142685,
+      "learning_rate": 2.257464420516164e-06,
+      "loss": 0.5596,
+      "step": 3353
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.07972193509340286,
+      "learning_rate": 2.2389922012165944e-06,
+      "loss": 0.4985,
+      "step": 3354
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.08464938402175903,
+      "learning_rate": 2.2205950136223176e-06,
+      "loss": 0.5566,
+      "step": 3355
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.07752576470375061,
+      "learning_rate": 2.202272871853228e-06,
+      "loss": 0.5314,
+      "step": 3356
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.08926267921924591,
+      "learning_rate": 2.1840257899715887e-06,
+      "loss": 0.6661,
+      "step": 3357
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.08092442899942398,
+      "learning_rate": 2.1658537819820414e-06,
+      "loss": 0.6168,
+      "step": 3358
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.08687537908554077,
+      "learning_rate": 2.1477568618316404e-06,
+      "loss": 0.6626,
+      "step": 3359
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.06949488818645477,
+      "learning_rate": 2.1297350434097974e-06,
+      "loss": 0.5432,
+      "step": 3360
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.09435413777828217,
+      "learning_rate": 2.1117883405482708e-06,
+      "loss": 0.6576,
+      "step": 3361
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.08354892581701279,
+      "learning_rate": 2.0939167670211978e-06,
+      "loss": 0.6777,
+      "step": 3362
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.0721585676074028,
+      "learning_rate": 2.0761203365450064e-06,
+      "loss": 0.4559,
+      "step": 3363
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.07381545752286911,
+      "learning_rate": 2.0583990627785154e-06,
+      "loss": 0.6137,
+      "step": 3364
+    },
+    {
+      "epoch": 1.85,
+      "grad_norm": 0.08608017861843109,
+      "learning_rate": 2.0407529593228116e-06,
+      "loss": 0.5615,
+      "step": 3365
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.07255735248327255,
+      "learning_rate": 2.023182039721294e-06,
+      "loss": 0.5131,
+      "step": 3366
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.0802227184176445,
+      "learning_rate": 2.0056863174596762e-06,
+      "loss": 0.6041,
+      "step": 3367
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.08381889015436172,
+      "learning_rate": 1.9882658059659496e-06,
+      "loss": 0.5783,
+      "step": 3368
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.08564453572034836,
+      "learning_rate": 1.970920518610375e-06,
+      "loss": 0.5746,
+      "step": 3369
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.07094965130090714,
+      "learning_rate": 1.9536504687054924e-06,
+      "loss": 0.5272,
+      "step": 3370
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.07661442458629608,
+      "learning_rate": 1.9364556695060764e-06,
+      "loss": 0.5754,
+      "step": 3371
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.07895109057426453,
+      "learning_rate": 1.91933613420916e-06,
+      "loss": 0.6123,
+      "step": 3372
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.07868780195713043,
+      "learning_rate": 1.9022918759540209e-06,
+      "loss": 0.5591,
+      "step": 3373
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.0766960084438324,
+      "learning_rate": 1.88532290782214e-06,
+      "loss": 0.5008,
+      "step": 3374
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.08017481863498688,
+      "learning_rate": 1.868429242837233e-06,
+      "loss": 0.5747,
+      "step": 3375
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.08352438360452652,
+      "learning_rate": 1.8516108939651945e-06,
+      "loss": 0.6316,
+      "step": 3376
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.08620418608188629,
+      "learning_rate": 1.8348678741141546e-06,
+      "loss": 0.6034,
+      "step": 3377
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.0783904418349266,
+      "learning_rate": 1.8182001961343787e-06,
+      "loss": 0.5761,
+      "step": 3378
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.08407583832740784,
+      "learning_rate": 1.801607872818356e-06,
+      "loss": 0.6141,
+      "step": 3379
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.08542238920927048,
+      "learning_rate": 1.785090916900678e-06,
+      "loss": 0.6253,
+      "step": 3380
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.07493787258863449,
+      "learning_rate": 1.7686493410581707e-06,
+      "loss": 0.5326,
+      "step": 3381
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.07105356454849243,
+      "learning_rate": 1.752283157909762e-06,
+      "loss": 0.463,
+      "step": 3382
+    },
+    {
+      "epoch": 1.86,
+      "grad_norm": 0.09019310772418976,
+      "learning_rate": 1.7359923800164935e-06,
+      "loss": 0.589,
+      "step": 3383
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.08535916358232498,
+      "learning_rate": 1.7197770198815743e-06,
+      "loss": 0.6479,
+      "step": 3384
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.08877049386501312,
+      "learning_rate": 1.7036370899503162e-06,
+      "loss": 0.5914,
+      "step": 3385
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.07365193217992783,
+      "learning_rate": 1.6875726026101435e-06,
+      "loss": 0.562,
+      "step": 3386
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.08467639982700348,
+      "learning_rate": 1.6715835701905603e-06,
+      "loss": 0.5883,
+      "step": 3387
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.08218105882406235,
+      "learning_rate": 1.6556700049631835e-06,
+      "loss": 0.6002,
+      "step": 3388
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.07743723690509796,
+      "learning_rate": 1.6398319191416656e-06,
+      "loss": 0.5423,
+      "step": 3389
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.07386884838342667,
+      "learning_rate": 1.624069324881805e-06,
+      "loss": 0.5223,
+      "step": 3390
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.07357634603977203,
+      "learning_rate": 1.6083822342813692e-06,
+      "loss": 0.478,
+      "step": 3391
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.07314004749059677,
+      "learning_rate": 1.5927706593802494e-06,
+      "loss": 0.5009,
+      "step": 3392
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.06364717334508896,
+      "learning_rate": 1.5772346121603498e-06,
+      "loss": 0.4972,
+      "step": 3393
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.08853484690189362,
+      "learning_rate": 1.561774104545588e-06,
+      "loss": 0.6418,
+      "step": 3394
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.07860343158245087,
+      "learning_rate": 1.54638914840195e-06,
+      "loss": 0.5295,
+      "step": 3395
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.06886250525712967,
+      "learning_rate": 1.531079755537379e-06,
+      "loss": 0.4841,
+      "step": 3396
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.07765613496303558,
+      "learning_rate": 1.5158459377018873e-06,
+      "loss": 0.5708,
+      "step": 3397
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.084499292075634,
+      "learning_rate": 1.5006877065874336e-06,
+      "loss": 0.6062,
+      "step": 3398
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.0805201455950737,
+      "learning_rate": 1.4856050738279892e-06,
+      "loss": 0.5724,
+      "step": 3399
+    },
+    {
+      "epoch": 1.87,
+      "grad_norm": 0.08347020298242569,
+      "learning_rate": 1.4705980509994833e-06,
+      "loss": 0.6772,
+      "step": 3400
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.08040057867765427,
+      "learning_rate": 1.4556666496198245e-06,
+      "loss": 0.599,
+      "step": 3401
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.0760309025645256,
+      "learning_rate": 1.4408108811489018e-06,
+      "loss": 0.489,
+      "step": 3402
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.070058174431324,
+      "learning_rate": 1.4260307569885168e-06,
+      "loss": 0.5206,
+      "step": 3403
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.07758305966854095,
+      "learning_rate": 1.411326288482462e-06,
+      "loss": 0.5079,
+      "step": 3404
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.08807816356420517,
+      "learning_rate": 1.3966974869163985e-06,
+      "loss": 0.557,
+      "step": 3405
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.08464004099369049,
+      "learning_rate": 1.3821443635179898e-06,
+      "loss": 0.5922,
+      "step": 3406
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.06907836347818375,
+      "learning_rate": 1.3676669294567456e-06,
+      "loss": 0.5663,
+      "step": 3407
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.07782316207885742,
+      "learning_rate": 1.3532651958441444e-06,
+      "loss": 0.5586,
+      "step": 3408
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.07036988437175751,
+      "learning_rate": 1.3389391737335112e-06,
+      "loss": 0.5858,
+      "step": 3409
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.08822328597307205,
+      "learning_rate": 1.3246888741201058e-06,
+      "loss": 0.6201,
+      "step": 3410
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.07636477798223495,
+      "learning_rate": 1.3105143079410465e-06,
+      "loss": 0.545,
+      "step": 3411
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.08318522572517395,
+      "learning_rate": 1.2964154860753197e-06,
+      "loss": 0.5728,
+      "step": 3412
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.07751517742872238,
+      "learning_rate": 1.2823924193438142e-06,
+      "loss": 0.6117,
+      "step": 3413
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.07766176015138626,
+      "learning_rate": 1.2684451185092317e-06,
+      "loss": 0.5847,
+      "step": 3414
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.07549408078193665,
+      "learning_rate": 1.2545735942761539e-06,
+      "loss": 0.5441,
+      "step": 3415
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.08800029009580612,
+      "learning_rate": 1.2407778572909868e-06,
+      "loss": 0.663,
+      "step": 3416
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.06972342729568481,
+      "learning_rate": 1.227057918141994e-06,
+      "loss": 0.5834,
+      "step": 3417
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 0.07410095632076263,
+      "learning_rate": 1.2134137873592299e-06,
+      "loss": 0.6273,
+      "step": 3418
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.08256086707115173,
+      "learning_rate": 1.199845475414585e-06,
+      "loss": 0.5854,
+      "step": 3419
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.08870168775320053,
+      "learning_rate": 1.1863529927217732e-06,
+      "loss": 0.7093,
+      "step": 3420
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.08100131154060364,
+      "learning_rate": 1.1729363496362777e-06,
+      "loss": 0.6304,
+      "step": 3421
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.07548246532678604,
+      "learning_rate": 1.159595556455395e-06,
+      "loss": 0.4772,
+      "step": 3422
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.09052509814500809,
+      "learning_rate": 1.1463306234181903e-06,
+      "loss": 0.5927,
+      "step": 3423
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.07959606498479843,
+      "learning_rate": 1.1331415607055306e-06,
+      "loss": 0.5433,
+      "step": 3424
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.08933278918266296,
+      "learning_rate": 1.1200283784400411e-06,
+      "loss": 0.5826,
+      "step": 3425
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.07218793034553528,
+      "learning_rate": 1.1069910866861155e-06,
+      "loss": 0.4826,
+      "step": 3426
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.0752687007188797,
+      "learning_rate": 1.0940296954498607e-06,
+      "loss": 0.5799,
+      "step": 3427
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.08115094900131226,
+      "learning_rate": 1.0811442146791972e-06,
+      "loss": 0.5199,
+      "step": 3428
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.08585885167121887,
+      "learning_rate": 1.068334654263714e-06,
+      "loss": 0.603,
+      "step": 3429
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.07445257157087326,
+      "learning_rate": 1.0556010240348025e-06,
+      "loss": 0.555,
+      "step": 3430
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.08283374458551407,
+      "learning_rate": 1.0429433337655115e-06,
+      "loss": 0.5513,
+      "step": 3431
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.0685088112950325,
+      "learning_rate": 1.030361593170648e-06,
+      "loss": 0.4646,
+      "step": 3432
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.08084689825773239,
+      "learning_rate": 1.0178558119067315e-06,
+      "loss": 0.5691,
+      "step": 3433
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.07466350495815277,
+      "learning_rate": 1.0054259995719295e-06,
+      "loss": 0.5365,
+      "step": 3434
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.0814463198184967,
+      "learning_rate": 9.930721657061659e-07,
+      "loss": 0.5889,
+      "step": 3435
+    },
+    {
+      "epoch": 1.89,
+      "grad_norm": 0.07925049960613251,
+      "learning_rate": 9.80794319791012e-07,
+      "loss": 0.5265,
+      "step": 3436
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.07562460750341415,
+      "learning_rate": 9.68592471249752e-07,
+      "loss": 0.6428,
+      "step": 3437
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.07313285022974014,
+      "learning_rate": 9.564666294472946e-07,
+      "loss": 0.5749,
+      "step": 3438
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.0887812003493309,
+      "learning_rate": 9.444168036902513e-07,
+      "loss": 0.6197,
+      "step": 3439
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.07397332042455673,
+      "learning_rate": 9.324430032268794e-07,
+      "loss": 0.5552,
+      "step": 3440
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.07077188044786453,
+      "learning_rate": 9.205452372470835e-07,
+      "loss": 0.4652,
+      "step": 3441
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.07836198806762695,
+      "learning_rate": 9.087235148824368e-07,
+      "loss": 0.5961,
+      "step": 3442
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.07381219416856766,
+      "learning_rate": 8.969778452060928e-07,
+      "loss": 0.5567,
+      "step": 3443
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.07355515658855438,
+      "learning_rate": 8.853082372328847e-07,
+      "loss": 0.5673,
+      "step": 3444
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.0868539810180664,
+      "learning_rate": 8.737146999192592e-07,
+      "loss": 0.5933,
+      "step": 3445
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.08018617331981659,
+      "learning_rate": 8.621972421632651e-07,
+      "loss": 0.5437,
+      "step": 3446
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.08229728788137436,
+      "learning_rate": 8.507558728045539e-07,
+      "loss": 0.4987,
+      "step": 3447
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.0776493176817894,
+      "learning_rate": 8.393906006244234e-07,
+      "loss": 0.5068,
+      "step": 3448
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.08558415621519089,
+      "learning_rate": 8.281014343457072e-07,
+      "loss": 0.5341,
+      "step": 3449
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.08440893143415451,
+      "learning_rate": 8.168883826328633e-07,
+      "loss": 0.6416,
+      "step": 3450
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.08656568825244904,
+      "learning_rate": 8.057514540919297e-07,
+      "loss": 0.5967,
+      "step": 3451
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.08011357486248016,
+      "learning_rate": 7.946906572705137e-07,
+      "loss": 0.663,
+      "step": 3452
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.07203607261180878,
+      "learning_rate": 7.837060006577801e-07,
+      "loss": 0.5436,
+      "step": 3453
+    },
+    {
+      "epoch": 1.9,
+      "grad_norm": 0.07224507629871368,
+      "learning_rate": 7.727974926844849e-07,
+      "loss": 0.5395,
+      "step": 3454
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.07532103359699249,
+      "learning_rate": 7.61965141722909e-07,
+      "loss": 0.6166,
+      "step": 3455
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.078483946621418,
+      "learning_rate": 7.512089560869018e-07,
+      "loss": 0.5581,
+      "step": 3456
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.07536689192056656,
+      "learning_rate": 7.405289440318486e-07,
+      "loss": 0.526,
+      "step": 3457
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.08381485939025879,
+      "learning_rate": 7.299251137546814e-07,
+      "loss": 0.561,
+      "step": 3458
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.09058713912963867,
+      "learning_rate": 7.193974733938346e-07,
+      "loss": 0.6357,
+      "step": 3459
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.07965592294931412,
+      "learning_rate": 7.089460310293117e-07,
+      "loss": 0.5686,
+      "step": 3460
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.07699765264987946,
+      "learning_rate": 6.985707946825848e-07,
+      "loss": 0.5332,
+      "step": 3461
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.07802991569042206,
+      "learning_rate": 6.882717723166731e-07,
+      "loss": 0.6359,
+      "step": 3462
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.0804867297410965,
+      "learning_rate": 6.780489718360983e-07,
+      "loss": 0.6162,
+      "step": 3463
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.0790751576423645,
+      "learning_rate": 6.679024010868618e-07,
+      "loss": 0.5907,
+      "step": 3464
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.07376383990049362,
+      "learning_rate": 6.578320678564676e-07,
+      "loss": 0.6181,
+      "step": 3465
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.07800883054733276,
+      "learning_rate": 6.47837979873922e-07,
+      "loss": 0.5485,
+      "step": 3466
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.08249462395906448,
+      "learning_rate": 6.379201448097005e-07,
+      "loss": 0.6131,
+      "step": 3467
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.08678142726421356,
+      "learning_rate": 6.280785702757475e-07,
+      "loss": 0.5681,
+      "step": 3468
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.07358631491661072,
+      "learning_rate": 6.183132638254763e-07,
+      "loss": 0.4937,
+      "step": 3469
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.08811584860086441,
+      "learning_rate": 6.086242329537917e-07,
+      "loss": 0.6157,
+      "step": 3470
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.07266715168952942,
+      "learning_rate": 5.99011485097023e-07,
+      "loss": 0.5419,
+      "step": 3471
+    },
+    {
+      "epoch": 1.91,
+      "grad_norm": 0.0865073874592781,
+      "learning_rate": 5.894750276329796e-07,
+      "loss": 0.6344,
+      "step": 3472
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.08059767633676529,
+      "learning_rate": 5.800148678808958e-07,
+      "loss": 0.5713,
+      "step": 3473
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.07227861881256104,
+      "learning_rate": 5.706310131014636e-07,
+      "loss": 0.5453,
+      "step": 3474
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.0802195817232132,
+      "learning_rate": 5.613234704967996e-07,
+      "loss": 0.5387,
+      "step": 3475
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.0846519023180008,
+      "learning_rate": 5.520922472104561e-07,
+      "loss": 0.6752,
+      "step": 3476
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.08230707049369812,
+      "learning_rate": 5.429373503274215e-07,
+      "loss": 0.571,
+      "step": 3477
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.07529239356517792,
+      "learning_rate": 5.338587868740641e-07,
+      "loss": 0.5601,
+      "step": 3478
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.08313122391700745,
+      "learning_rate": 5.248565638182101e-07,
+      "loss": 0.5917,
+      "step": 3479
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.08534807711839676,
+      "learning_rate": 5.159306880690884e-07,
+      "loss": 0.5682,
+      "step": 3480
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.08856728672981262,
+      "learning_rate": 5.070811664773079e-07,
+      "loss": 0.635,
+      "step": 3481
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.0904325619339943,
+      "learning_rate": 4.98308005834891e-07,
+      "loss": 0.5966,
+      "step": 3482
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.09154371917247772,
+      "learning_rate": 4.896112128752406e-07,
+      "loss": 0.5869,
+      "step": 3483
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.0799337774515152,
+      "learning_rate": 4.809907942731729e-07,
+      "loss": 0.5264,
+      "step": 3484
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.09307362139225006,
+      "learning_rate": 4.72446756644862e-07,
+      "loss": 0.6198,
+      "step": 3485
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.08572401106357574,
+      "learning_rate": 4.639791065478738e-07,
+      "loss": 0.5863,
+      "step": 3486
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.08228857815265656,
+      "learning_rate": 4.555878504811317e-07,
+      "loss": 0.5623,
+      "step": 3487
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.08371774852275848,
+      "learning_rate": 4.472729948849397e-07,
+      "loss": 0.5971,
+      "step": 3488
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.09152055531740189,
+      "learning_rate": 4.390345461409706e-07,
+      "loss": 0.594,
+      "step": 3489
+    },
+    {
+      "epoch": 1.92,
+      "grad_norm": 0.07677818089723587,
+      "learning_rate": 4.308725105722333e-07,
+      "loss": 0.551,
+      "step": 3490
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.06396432965993881,
+      "learning_rate": 4.227868944431057e-07,
+      "loss": 0.4516,
+      "step": 3491
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.07630179822444916,
+      "learning_rate": 4.147777039593015e-07,
+      "loss": 0.5061,
+      "step": 3492
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.07213296741247177,
+      "learning_rate": 4.068449452679035e-07,
+      "loss": 0.4783,
+      "step": 3493
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.07577905803918839,
+      "learning_rate": 3.9898862445729714e-07,
+      "loss": 0.5182,
+      "step": 3494
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.07999127358198166,
+      "learning_rate": 3.912087475572257e-07,
+      "loss": 0.5724,
+      "step": 3495
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.10361068695783615,
+      "learning_rate": 3.835053205387573e-07,
+      "loss": 0.6712,
+      "step": 3496
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.08972208946943283,
+      "learning_rate": 3.758783493142737e-07,
+      "loss": 0.563,
+      "step": 3497
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.07700824737548828,
+      "learning_rate": 3.683278397374923e-07,
+      "loss": 0.5676,
+      "step": 3498
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.07342316210269928,
+      "learning_rate": 3.608537976034554e-07,
+      "loss": 0.5639,
+      "step": 3499
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.08734098076820374,
+      "learning_rate": 3.534562286484633e-07,
+      "loss": 0.6276,
+      "step": 3500
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.07715500891208649,
+      "learning_rate": 3.4613513855018543e-07,
+      "loss": 0.6094,
+      "step": 3501
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.08446547389030457,
+      "learning_rate": 3.3889053292757155e-07,
+      "loss": 0.5558,
+      "step": 3502
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.07040911167860031,
+      "learning_rate": 3.3172241734085176e-07,
+      "loss": 0.5575,
+      "step": 3503
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.07131405174732208,
+      "learning_rate": 3.2463079729158076e-07,
+      "loss": 0.4731,
+      "step": 3504
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.07866710424423218,
+      "learning_rate": 3.1761567822257146e-07,
+      "loss": 0.6311,
+      "step": 3505
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.07367514818906784,
+      "learning_rate": 3.106770655179392e-07,
+      "loss": 0.5383,
+      "step": 3506
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.08176425844430923,
+      "learning_rate": 3.0381496450309075e-07,
+      "loss": 0.5508,
+      "step": 3507
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 0.07364267110824585,
+      "learning_rate": 2.9702938044468e-07,
+      "loss": 0.4938,
+      "step": 3508
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.0771787241101265,
+      "learning_rate": 2.903203185506631e-07,
+      "loss": 0.5832,
+      "step": 3509
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.07928218692541122,
+      "learning_rate": 2.836877839702545e-07,
+      "loss": 0.6251,
+      "step": 3510
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.08749215304851532,
+      "learning_rate": 2.7713178179392673e-07,
+      "loss": 0.5893,
+      "step": 3511
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.08024594187736511,
+      "learning_rate": 2.706523170534325e-07,
+      "loss": 0.5203,
+      "step": 3512
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.08996916562318802,
+      "learning_rate": 2.6424939472176056e-07,
+      "loss": 0.6149,
+      "step": 3513
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.08545570820569992,
+      "learning_rate": 2.579230197131577e-07,
+      "loss": 0.5568,
+      "step": 3514
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.07621780782938004,
+      "learning_rate": 2.5167319688312876e-07,
+      "loss": 0.4913,
+      "step": 3515
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.08227244019508362,
+      "learning_rate": 2.454999310284256e-07,
+      "loss": 0.5735,
+      "step": 3516
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.08408751338720322,
+      "learning_rate": 2.394032268870472e-07,
+      "loss": 0.6155,
+      "step": 3517
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.07970957458019257,
+      "learning_rate": 2.3338308913819496e-07,
+      "loss": 0.5079,
+      "step": 3518
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.06704852730035782,
+      "learning_rate": 2.2743952240236176e-07,
+      "loss": 0.4832,
+      "step": 3519
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.07459887862205505,
+      "learning_rate": 2.2157253124122092e-07,
+      "loss": 0.5476,
+      "step": 3520
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.07807392627000809,
+      "learning_rate": 2.157821201577148e-07,
+      "loss": 0.6195,
+      "step": 3521
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.07530328631401062,
+      "learning_rate": 2.1006829359597747e-07,
+      "loss": 0.5083,
+      "step": 3522
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.0794893279671669,
+      "learning_rate": 2.0443105594137868e-07,
+      "loss": 0.5903,
+      "step": 3523
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.06803248077630997,
+      "learning_rate": 1.988704115205242e-07,
+      "loss": 0.4399,
+      "step": 3524
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.07905720919370651,
+      "learning_rate": 1.9338636460118908e-07,
+      "loss": 0.5813,
+      "step": 3525
+    },
+    {
+      "epoch": 1.94,
+      "grad_norm": 0.0881994217634201,
+      "learning_rate": 1.8797891939240643e-07,
+      "loss": 0.5493,
+      "step": 3526
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.07461023330688477,
+      "learning_rate": 1.8264808004438972e-07,
+      "loss": 0.4342,
+      "step": 3527
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.07800307869911194,
+      "learning_rate": 1.7739385064856618e-07,
+      "loss": 0.5938,
+      "step": 3528
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.07744286209344864,
+      "learning_rate": 1.7221623523756557e-07,
+      "loss": 0.5503,
+      "step": 3529
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.07589315623044968,
+      "learning_rate": 1.6711523778520921e-07,
+      "loss": 0.5541,
+      "step": 3530
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.07493191957473755,
+      "learning_rate": 1.62090862206532e-07,
+      "loss": 0.5086,
+      "step": 3531
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.08878099918365479,
+      "learning_rate": 1.5714311235773825e-07,
+      "loss": 0.6694,
+      "step": 3532
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.07499578595161438,
+      "learning_rate": 1.5227199203624586e-07,
+      "loss": 0.5938,
+      "step": 3533
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.07987038791179657,
+      "learning_rate": 1.4747750498061986e-07,
+      "loss": 0.541,
+      "step": 3534
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.08032140135765076,
+      "learning_rate": 1.4275965487066112e-07,
+      "loss": 0.5959,
+      "step": 3535
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.06550218909978867,
+      "learning_rate": 1.3811844532731766e-07,
+      "loss": 0.5164,
+      "step": 3536
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.07750283181667328,
+      "learning_rate": 1.3355387991271783e-07,
+      "loss": 0.5726,
+      "step": 3537
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.0810721144080162,
+      "learning_rate": 1.290659621301704e-07,
+      "loss": 0.6531,
+      "step": 3538
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.07887770980596542,
+      "learning_rate": 1.2465469542417563e-07,
+      "loss": 0.5742,
+      "step": 3539
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.08414478600025177,
+      "learning_rate": 1.203200831803808e-07,
+      "loss": 0.5464,
+      "step": 3540
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.0833955630660057,
+      "learning_rate": 1.1606212872559141e-07,
+      "loss": 0.639,
+      "step": 3541
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.08545012772083282,
+      "learning_rate": 1.1188083532780447e-07,
+      "loss": 0.5234,
+      "step": 3542
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.07801700383424759,
+      "learning_rate": 1.0777620619616402e-07,
+      "loss": 0.5809,
+      "step": 3543
+    },
+    {
+      "epoch": 1.95,
+      "grad_norm": 0.08191072195768356,
+      "learning_rate": 1.0374824448099451e-07,
+      "loss": 0.5946,
+      "step": 3544
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.08125972747802734,
+      "learning_rate": 9.979695327373417e-08,
+      "loss": 0.5225,
+      "step": 3545
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.08104407042264938,
+      "learning_rate": 9.59223356070349e-08,
+      "loss": 0.576,
+      "step": 3546
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.0694657638669014,
+      "learning_rate": 9.212439445464017e-08,
+      "loss": 0.5447,
+      "step": 3547
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.08874676376581192,
+      "learning_rate": 8.840313273149603e-08,
+      "loss": 0.5643,
+      "step": 3548
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.08475618064403534,
+      "learning_rate": 8.475855329367343e-08,
+      "loss": 0.5496,
+      "step": 3549
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.08129516243934631,
+      "learning_rate": 8.119065893839039e-08,
+      "loss": 0.6609,
+      "step": 3550
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.08017243444919586,
+      "learning_rate": 7.76994524040009e-08,
+      "loss": 0.5847,
+      "step": 3551
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.07957109808921814,
+      "learning_rate": 7.428493637002821e-08,
+      "loss": 0.6086,
+      "step": 3552
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.0790736973285675,
+      "learning_rate": 7.094711345710936e-08,
+      "loss": 0.6083,
+      "step": 3553
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.07656752318143845,
+      "learning_rate": 6.768598622701738e-08,
+      "loss": 0.5596,
+      "step": 3554
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.07584652304649353,
+      "learning_rate": 6.450155718268347e-08,
+      "loss": 0.5284,
+      "step": 3555
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.08076095581054688,
+      "learning_rate": 6.139382876816368e-08,
+      "loss": 0.5233,
+      "step": 3556
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.07191152125597,
+      "learning_rate": 5.8362803368638974e-08,
+      "loss": 0.5918,
+      "step": 3557
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.08384308964014053,
+      "learning_rate": 5.5408483310426253e-08,
+      "loss": 0.5641,
+      "step": 3558
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.08836299180984497,
+      "learning_rate": 5.2530870860956204e-08,
+      "loss": 0.6588,
+      "step": 3559
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.08138474076986313,
+      "learning_rate": 4.97299682288288e-08,
+      "loss": 0.6033,
+      "step": 3560
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.07241957634687424,
+      "learning_rate": 4.7005777563724483e-08,
+      "loss": 0.5177,
+      "step": 3561
+    },
+    {
+      "epoch": 1.96,
+      "grad_norm": 0.0745629146695137,
+      "learning_rate": 4.4358300956459654e-08,
+      "loss": 0.5968,
+      "step": 3562
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.07611432671546936,
+      "learning_rate": 4.178754043898669e-08,
+      "loss": 0.5377,
+      "step": 3563
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.07590966671705246,
+      "learning_rate": 3.929349798434956e-08,
+      "loss": 0.4752,
+      "step": 3564
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.07255428284406662,
+      "learning_rate": 3.6876175506750376e-08,
+      "loss": 0.4553,
+      "step": 3565
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.08332665264606476,
+      "learning_rate": 3.453557486148284e-08,
+      "loss": 0.5895,
+      "step": 3566
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.07977409660816193,
+      "learning_rate": 3.2271697844954426e-08,
+      "loss": 0.5473,
+      "step": 3567
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.08140242099761963,
+      "learning_rate": 3.008454619469747e-08,
+      "loss": 0.6037,
+      "step": 3568
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.08464977890253067,
+      "learning_rate": 2.797412158934698e-08,
+      "loss": 0.5727,
+      "step": 3569
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.0898507609963417,
+      "learning_rate": 2.5940425648662836e-08,
+      "loss": 0.6351,
+      "step": 3570
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.07864800095558167,
+      "learning_rate": 2.398345993350759e-08,
+      "loss": 0.6619,
+      "step": 3571
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.08140621334314346,
+      "learning_rate": 2.2103225945857563e-08,
+      "loss": 0.5289,
+      "step": 3572
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.07230167835950851,
+      "learning_rate": 2.029972512880285e-08,
+      "loss": 0.4946,
+      "step": 3573
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.08280620723962784,
+      "learning_rate": 1.8572958866514e-08,
+      "loss": 0.4832,
+      "step": 3574
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.08428557217121124,
+      "learning_rate": 1.6922928484297552e-08,
+      "loss": 0.6934,
+      "step": 3575
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.07942302525043488,
+      "learning_rate": 1.5349635248551596e-08,
+      "loss": 0.5662,
+      "step": 3576
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.08093717694282532,
+      "learning_rate": 1.3853080366788008e-08,
+      "loss": 0.5828,
+      "step": 3577
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.08332273364067078,
+      "learning_rate": 1.2433264987599113e-08,
+      "loss": 0.5815,
+      "step": 3578
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.0855347216129303,
+      "learning_rate": 1.1090190200713225e-08,
+      "loss": 0.6216,
+      "step": 3579
+    },
+    {
+      "epoch": 1.97,
+      "grad_norm": 0.08472579717636108,
+      "learning_rate": 9.823857036928008e-09,
+      "loss": 0.563,
+      "step": 3580
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.08015488088130951,
+      "learning_rate": 8.634266468154905e-09,
+      "loss": 0.562,
+      "step": 3581
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.06776360422372818,
+      "learning_rate": 7.521419407419128e-09,
+      "loss": 0.4865,
+      "step": 3582
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.08664952963590622,
+      "learning_rate": 6.48531670882635e-09,
+      "loss": 0.5859,
+      "step": 3583
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.07996014505624771,
+      "learning_rate": 5.5259591675849155e-09,
+      "loss": 0.5584,
+      "step": 3584
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.08663001656532288,
+      "learning_rate": 4.643347520005836e-09,
+      "loss": 0.6266,
+      "step": 3585
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.08204808831214905,
+      "learning_rate": 3.8374824434916915e-09,
+      "loss": 0.5886,
+      "step": 3586
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.08297493308782578,
+      "learning_rate": 3.1083645565477272e-09,
+      "loss": 0.5517,
+      "step": 3587
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.07561254501342773,
+      "learning_rate": 2.4559944187707572e-09,
+      "loss": 0.528,
+      "step": 3588
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.08274608105421066,
+      "learning_rate": 1.880372530860264e-09,
+      "loss": 0.5366,
+      "step": 3589
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.07005135715007782,
+      "learning_rate": 1.3814993345961924e-09,
+      "loss": 0.5466,
+      "step": 3590
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.07817287743091583,
+      "learning_rate": 9.593752128833622e-10,
+      "loss": 0.4973,
+      "step": 3591
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.0750674456357956,
+      "learning_rate": 6.140004896737495e-10,
+      "loss": 0.5421,
+      "step": 3592
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.077664814889431,
+      "learning_rate": 3.4537543006640804e-10,
+      "loss": 0.5494,
+      "step": 3593
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.08603768795728683,
+      "learning_rate": 1.5350024022975362e-10,
+      "loss": 0.5667,
+      "step": 3594
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.08145732432603836,
+      "learning_rate": 3.837506742376817e-11,
+      "loss": 0.5577,
+      "step": 3595
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 0.09160507470369339,
+      "learning_rate": 0.0,
+      "loss": 0.5693,
+      "step": 3596
+    }
+  ],
+  "logging_steps": 1,
+  "max_steps": 3596,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2,
+  "save_steps": 1798,
+  "total_flos": 6.436865157585961e+17,
+  "train_batch_size": 2,
+  "trial_name": null,
+  "trial_params": null
+}