diff --git "a/checkpoint-1798/trainer_state.json" "b/checkpoint-1798/trainer_state.json"
deleted file mode 100644--- "a/checkpoint-1798/trainer_state.json"
+++ /dev/null
@@ -1,12607 +0,0 @@
-{
-  "best_metric": null,
-  "best_model_checkpoint": null,
-  "epoch": 1.0,
-  "eval_steps": 500,
-  "global_step": 1798,
-  "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
-    }
-  ],
-  "logging_steps": 1,
-  "max_steps": 3596,
-  "num_input_tokens_seen": 0,
-  "num_train_epochs": 2,
-  "save_steps": 1798,
-  "total_flos": 3.21854445811925e+17,
-  "train_batch_size": 2,
-  "trial_name": null,
-  "trial_params": null
-}