diff --git "a/trainer_state.json" "b/trainer_state.json"
--- "a/trainer_state.json"
+++ "b/trainer_state.json"
@@ -2,7 +2,7 @@
   "best_metric": null,
   "best_model_checkpoint": null,
   "epoch": 5.0,
-  "global_step": 2230,
+  "global_step": 1115,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -10,13430 +10,6722 @@
     {
       "epoch": 0.0,
       "learning_rate": 0.0,
-      "loss": 8.7584,
+      "loss": 8.7788,
       "step": 1
     },
     {
-      "epoch": 0.0,
+      "epoch": 0.01,
       "learning_rate": 0.0,
-      "loss": 8.5833,
+      "loss": 8.857,
       "step": 2
     },
     {
       "epoch": 0.01,
       "learning_rate": 6e-07,
-      "loss": 8.5559,
+      "loss": 8.7704,
       "step": 3
     },
     {
-      "epoch": 0.01,
+      "epoch": 0.02,
       "learning_rate": 1.2e-06,
-      "loss": 8.4758,
+      "loss": 8.6839,
       "step": 4
     },
     {
-      "epoch": 0.01,
+      "epoch": 0.02,
       "learning_rate": 1.8e-06,
-      "loss": 8.6015,
+      "loss": 8.7517,
       "step": 5
     },
     {
-      "epoch": 0.01,
+      "epoch": 0.03,
       "learning_rate": 2.4e-06,
-      "loss": 8.402,
+      "loss": 8.3985,
       "step": 6
     },
     {
-      "epoch": 0.02,
+      "epoch": 0.03,
       "learning_rate": 2.9999999999999997e-06,
-      "loss": 8.3297,
+      "loss": 8.4278,
       "step": 7
     },
     {
-      "epoch": 0.02,
+      "epoch": 0.04,
       "learning_rate": 3.6e-06,
-      "loss": 7.9299,
+      "loss": 8.0949,
       "step": 8
     },
     {
-      "epoch": 0.02,
+      "epoch": 0.04,
       "learning_rate": 4.2e-06,
-      "loss": 7.6735,
+      "loss": 7.8106,
       "step": 9
     },
     {
-      "epoch": 0.02,
+      "epoch": 0.04,
       "learning_rate": 4.8e-06,
-      "loss": 7.5134,
+      "loss": 7.4337,
       "step": 10
     },
     {
-      "epoch": 0.02,
+      "epoch": 0.05,
       "learning_rate": 5.399999999999999e-06,
-      "loss": 7.3091,
+      "loss": 7.22,
       "step": 11
     },
     {
-      "epoch": 0.03,
+      "epoch": 0.05,
       "learning_rate": 5.999999999999999e-06,
-      "loss": 7.1164,
+      "loss": 6.9989,
       "step": 12
     },
     {
-      "epoch": 0.03,
+      "epoch": 0.06,
       "learning_rate": 6.599999999999999e-06,
-      "loss": 6.7007,
+      "loss": 6.8282,
       "step": 13
     },
     {
-      "epoch": 0.03,
+      "epoch": 0.06,
       "learning_rate": 7.2e-06,
-      "loss": 6.6293,
+      "loss": 6.6831,
       "step": 14
     },
     {
-      "epoch": 0.03,
+      "epoch": 0.07,
       "learning_rate": 7.799999999999998e-06,
-      "loss": 6.3926,
+      "loss": 6.4666,
       "step": 15
     },
     {
-      "epoch": 0.04,
+      "epoch": 0.07,
       "learning_rate": 8.4e-06,
-      "loss": 6.3405,
+      "loss": 6.2614,
       "step": 16
     },
     {
-      "epoch": 0.04,
+      "epoch": 0.08,
       "learning_rate": 8.999999999999999e-06,
-      "loss": 6.1208,
+      "loss": 6.1788,
       "step": 17
     },
     {
-      "epoch": 0.04,
+      "epoch": 0.08,
       "learning_rate": 9.6e-06,
-      "loss": 6.0531,
+      "loss": 6.0435,
       "step": 18
     },
     {
-      "epoch": 0.04,
+      "epoch": 0.09,
       "learning_rate": 1.02e-05,
-      "loss": 5.7463,
+      "loss": 5.8492,
       "step": 19
     },
     {
-      "epoch": 0.04,
+      "epoch": 0.09,
       "learning_rate": 1.0799999999999998e-05,
-      "loss": 5.8868,
+      "loss": 5.8448,
       "step": 20
     },
     {
-      "epoch": 0.05,
+      "epoch": 0.09,
       "learning_rate": 1.14e-05,
-      "loss": 5.6696,
+      "loss": 5.6986,
       "step": 21
     },
     {
-      "epoch": 0.05,
+      "epoch": 0.1,
       "learning_rate": 1.1999999999999999e-05,
-      "loss": 5.6137,
+      "loss": 5.673,
       "step": 22
     },
     {
-      "epoch": 0.05,
+      "epoch": 0.1,
       "learning_rate": 1.26e-05,
-      "loss": 5.4833,
+      "loss": 5.4552,
       "step": 23
     },
     {
-      "epoch": 0.05,
+      "epoch": 0.11,
       "learning_rate": 1.3199999999999997e-05,
-      "loss": 5.4312,
+      "loss": 5.4442,
       "step": 24
     },
     {
-      "epoch": 0.06,
+      "epoch": 0.11,
       "learning_rate": 1.3799999999999998e-05,
-      "loss": 5.1508,
+      "loss": 5.2917,
       "step": 25
     },
     {
-      "epoch": 0.06,
+      "epoch": 0.12,
       "learning_rate": 1.44e-05,
-      "loss": 5.4145,
+      "loss": 5.29,
       "step": 26
     },
     {
-      "epoch": 0.06,
+      "epoch": 0.12,
       "learning_rate": 1.4999999999999999e-05,
-      "loss": 5.1659,
+      "loss": 5.1356,
       "step": 27
     },
     {
-      "epoch": 0.06,
+      "epoch": 0.13,
       "learning_rate": 1.5599999999999996e-05,
-      "loss": 5.0136,
+      "loss": 5.1709,
       "step": 28
     },
     {
-      "epoch": 0.07,
+      "epoch": 0.13,
       "learning_rate": 1.6199999999999997e-05,
-      "loss": 4.967,
+      "loss": 4.9962,
       "step": 29
     },
     {
-      "epoch": 0.07,
+      "epoch": 0.13,
       "learning_rate": 1.68e-05,
-      "loss": 5.1574,
+      "loss": 5.1114,
       "step": 30
     },
     {
-      "epoch": 0.07,
+      "epoch": 0.14,
       "learning_rate": 1.74e-05,
-      "loss": 5.1364,
+      "loss": 5.1065,
       "step": 31
     },
     {
-      "epoch": 0.07,
+      "epoch": 0.14,
       "learning_rate": 1.7999999999999997e-05,
-      "loss": 4.9412,
+      "loss": 4.9779,
       "step": 32
     },
     {
-      "epoch": 0.07,
+      "epoch": 0.15,
       "learning_rate": 1.8599999999999998e-05,
-      "loss": 5.039,
+      "loss": 4.9887,
       "step": 33
     },
     {
-      "epoch": 0.08,
+      "epoch": 0.15,
       "learning_rate": 1.92e-05,
-      "loss": 4.9206,
+      "loss": 4.9495,
       "step": 34
     },
     {
-      "epoch": 0.08,
+      "epoch": 0.16,
       "learning_rate": 1.98e-05,
-      "loss": 4.93,
+      "loss": 4.8838,
       "step": 35
     },
     {
-      "epoch": 0.08,
+      "epoch": 0.16,
       "learning_rate": 2.04e-05,
-      "loss": 4.8945,
+      "loss": 4.9061,
       "step": 36
     },
     {
-      "epoch": 0.08,
+      "epoch": 0.17,
       "learning_rate": 2.1e-05,
-      "loss": 5.0179,
+      "loss": 4.9915,
       "step": 37
     },
     {
-      "epoch": 0.09,
+      "epoch": 0.17,
       "learning_rate": 2.1599999999999996e-05,
-      "loss": 4.918,
+      "loss": 4.8265,
       "step": 38
     },
     {
-      "epoch": 0.09,
+      "epoch": 0.17,
       "learning_rate": 2.2199999999999998e-05,
-      "loss": 5.0055,
+      "loss": 4.9207,
       "step": 39
     },
     {
-      "epoch": 0.09,
+      "epoch": 0.18,
       "learning_rate": 2.28e-05,
-      "loss": 4.8467,
+      "loss": 4.9041,
       "step": 40
     },
     {
-      "epoch": 0.09,
+      "epoch": 0.18,
       "learning_rate": 2.34e-05,
-      "loss": 4.9421,
+      "loss": 4.8838,
       "step": 41
     },
     {
-      "epoch": 0.09,
+      "epoch": 0.19,
       "learning_rate": 2.3999999999999997e-05,
-      "loss": 4.9595,
+      "loss": 4.9159,
       "step": 42
     },
     {
-      "epoch": 0.1,
+      "epoch": 0.19,
       "learning_rate": 2.4599999999999998e-05,
-      "loss": 4.6993,
+      "loss": 4.8392,
       "step": 43
     },
     {
-      "epoch": 0.1,
+      "epoch": 0.2,
       "learning_rate": 2.52e-05,
-      "loss": 4.8895,
+      "loss": 4.9044,
       "step": 44
     },
     {
-      "epoch": 0.1,
+      "epoch": 0.2,
       "learning_rate": 2.5799999999999997e-05,
-      "loss": 4.9856,
+      "loss": 4.8615,
       "step": 45
     },
     {
-      "epoch": 0.1,
+      "epoch": 0.21,
       "learning_rate": 2.6399999999999995e-05,
-      "loss": 4.9902,
+      "loss": 4.9846,
       "step": 46
     },
     {
-      "epoch": 0.11,
+      "epoch": 0.21,
       "learning_rate": 2.6999999999999996e-05,
-      "loss": 4.9278,
+      "loss": 4.8641,
       "step": 47
     },
     {
-      "epoch": 0.11,
+      "epoch": 0.22,
       "learning_rate": 2.7599999999999997e-05,
-      "loss": 4.9454,
+      "loss": 4.8026,
       "step": 48
     },
     {
-      "epoch": 0.11,
+      "epoch": 0.22,
       "learning_rate": 2.8199999999999998e-05,
-      "loss": 4.9972,
+      "loss": 4.8971,
       "step": 49
     },
     {
-      "epoch": 0.11,
+      "epoch": 0.22,
       "learning_rate": 2.88e-05,
-      "loss": 4.6526,
+      "loss": 4.6302,
       "step": 50
     },
     {
-      "epoch": 0.11,
+      "epoch": 0.23,
       "learning_rate": 2.94e-05,
-      "loss": 5.8451,
+      "loss": 6.4289,
       "step": 51
     },
     {
-      "epoch": 0.12,
+      "epoch": 0.23,
       "learning_rate": 2.9999999999999997e-05,
-      "loss": 5.7033,
+      "loss": 6.2769,
       "step": 52
     },
     {
-      "epoch": 0.12,
+      "epoch": 0.24,
       "learning_rate": 3.06e-05,
-      "loss": 5.6843,
+      "loss": 5.9356,
       "step": 53
     },
     {
-      "epoch": 0.12,
+      "epoch": 0.24,
       "learning_rate": 3.119999999999999e-05,
-      "loss": 5.247,
+      "loss": 5.4623,
       "step": 54
     },
     {
-      "epoch": 0.12,
+      "epoch": 0.25,
       "learning_rate": 3.1799999999999994e-05,
-      "loss": 4.7961,
+      "loss": 5.1183,
       "step": 55
     },
     {
-      "epoch": 0.13,
+      "epoch": 0.25,
       "learning_rate": 3.2399999999999995e-05,
-      "loss": 4.9426,
+      "loss": 4.866,
       "step": 56
     },
     {
-      "epoch": 0.13,
+      "epoch": 0.26,
       "learning_rate": 3.2999999999999996e-05,
-      "loss": 4.9681,
+      "loss": 4.8544,
       "step": 57
     },
     {
-      "epoch": 0.13,
+      "epoch": 0.26,
       "learning_rate": 3.36e-05,
-      "loss": 4.9621,
+      "loss": 4.8141,
       "step": 58
     },
     {
-      "epoch": 0.13,
+      "epoch": 0.26,
       "learning_rate": 3.42e-05,
-      "loss": 4.9065,
+      "loss": 4.8047,
       "step": 59
     },
     {
-      "epoch": 0.13,
+      "epoch": 0.27,
       "learning_rate": 3.48e-05,
-      "loss": 4.7891,
+      "loss": 4.8444,
       "step": 60
     },
     {
-      "epoch": 0.14,
+      "epoch": 0.27,
       "learning_rate": 3.539999999999999e-05,
-      "loss": 4.8069,
+      "loss": 4.8777,
       "step": 61
     },
     {
-      "epoch": 0.14,
+      "epoch": 0.28,
       "learning_rate": 3.5999999999999994e-05,
-      "loss": 4.8273,
+      "loss": 4.7374,
       "step": 62
     },
     {
-      "epoch": 0.14,
+      "epoch": 0.28,
       "learning_rate": 3.6599999999999995e-05,
-      "loss": 4.7276,
+      "loss": 4.8447,
       "step": 63
     },
     {
-      "epoch": 0.14,
+      "epoch": 0.29,
       "learning_rate": 3.7199999999999996e-05,
-      "loss": 4.7894,
+      "loss": 4.7271,
       "step": 64
     },
     {
-      "epoch": 0.15,
+      "epoch": 0.29,
       "learning_rate": 3.78e-05,
-      "loss": 4.7477,
+      "loss": 4.691,
       "step": 65
     },
     {
-      "epoch": 0.15,
+      "epoch": 0.3,
       "learning_rate": 3.84e-05,
-      "loss": 4.6233,
+      "loss": 4.7024,
       "step": 66
     },
     {
-      "epoch": 0.15,
+      "epoch": 0.3,
       "learning_rate": 3.9e-05,
-      "loss": 4.7465,
+      "loss": 4.6623,
       "step": 67
     },
     {
-      "epoch": 0.15,
+      "epoch": 0.3,
       "learning_rate": 3.96e-05,
-      "loss": 4.7111,
+      "loss": 4.6917,
       "step": 68
     },
     {
-      "epoch": 0.15,
+      "epoch": 0.31,
       "learning_rate": 4.02e-05,
-      "loss": 4.7447,
+      "loss": 4.7466,
       "step": 69
     },
     {
-      "epoch": 0.16,
+      "epoch": 0.31,
       "learning_rate": 4.08e-05,
-      "loss": 4.6527,
+      "loss": 4.6977,
       "step": 70
     },
     {
-      "epoch": 0.16,
+      "epoch": 0.32,
       "learning_rate": 4.14e-05,
-      "loss": 4.7916,
+      "loss": 4.6366,
       "step": 71
     },
     {
-      "epoch": 0.16,
+      "epoch": 0.32,
       "learning_rate": 4.2e-05,
-      "loss": 4.8087,
+      "loss": 4.6913,
       "step": 72
     },
     {
-      "epoch": 0.16,
+      "epoch": 0.33,
       "learning_rate": 4.259999999999999e-05,
-      "loss": 4.6922,
+      "loss": 4.5726,
       "step": 73
     },
     {
-      "epoch": 0.17,
+      "epoch": 0.33,
       "learning_rate": 4.319999999999999e-05,
-      "loss": 4.7028,
+      "loss": 4.6529,
       "step": 74
     },
     {
-      "epoch": 0.17,
+      "epoch": 0.34,
       "learning_rate": 4.3799999999999994e-05,
-      "loss": 4.7345,
+      "loss": 4.7602,
       "step": 75
     },
     {
-      "epoch": 0.17,
+      "epoch": 0.34,
       "learning_rate": 4.4399999999999995e-05,
-      "loss": 4.6529,
+      "loss": 4.6657,
       "step": 76
     },
     {
-      "epoch": 0.17,
+      "epoch": 0.35,
       "learning_rate": 4.4999999999999996e-05,
-      "loss": 4.6925,
+      "loss": 4.5708,
       "step": 77
     },
     {
-      "epoch": 0.17,
+      "epoch": 0.35,
       "learning_rate": 4.56e-05,
-      "loss": 4.6708,
+      "loss": 4.6321,
       "step": 78
     },
     {
-      "epoch": 0.18,
+      "epoch": 0.35,
       "learning_rate": 4.62e-05,
-      "loss": 4.6112,
+      "loss": 4.5891,
       "step": 79
     },
     {
-      "epoch": 0.18,
+      "epoch": 0.36,
       "learning_rate": 4.68e-05,
-      "loss": 4.7209,
+      "loss": 4.695,
       "step": 80
     },
     {
-      "epoch": 0.18,
+      "epoch": 0.36,
       "learning_rate": 4.7399999999999993e-05,
-      "loss": 4.6872,
+      "loss": 4.5544,
       "step": 81
     },
     {
-      "epoch": 0.18,
+      "epoch": 0.37,
       "learning_rate": 4.7999999999999994e-05,
-      "loss": 4.6316,
+      "loss": 4.6184,
       "step": 82
     },
     {
-      "epoch": 0.19,
+      "epoch": 0.37,
       "learning_rate": 4.8599999999999995e-05,
-      "loss": 4.7507,
+      "loss": 4.6035,
       "step": 83
     },
     {
-      "epoch": 0.19,
+      "epoch": 0.38,
       "learning_rate": 4.9199999999999997e-05,
-      "loss": 4.6468,
+      "loss": 4.6037,
       "step": 84
     },
     {
-      "epoch": 0.19,
+      "epoch": 0.38,
       "learning_rate": 4.98e-05,
-      "loss": 4.6845,
+      "loss": 4.6785,
       "step": 85
     },
     {
-      "epoch": 0.19,
+      "epoch": 0.39,
       "learning_rate": 5.04e-05,
-      "loss": 4.6201,
+      "loss": 4.4987,
       "step": 86
     },
     {
-      "epoch": 0.2,
+      "epoch": 0.39,
       "learning_rate": 5.1e-05,
-      "loss": 4.6043,
+      "loss": 4.6057,
       "step": 87
     },
     {
-      "epoch": 0.2,
+      "epoch": 0.39,
       "learning_rate": 5.1599999999999994e-05,
-      "loss": 4.5494,
+      "loss": 4.5589,
       "step": 88
     },
     {
-      "epoch": 0.2,
+      "epoch": 0.4,
       "learning_rate": 5.2199999999999995e-05,
-      "loss": 4.6675,
+      "loss": 4.6228,
       "step": 89
     },
     {
-      "epoch": 0.2,
+      "epoch": 0.4,
       "learning_rate": 5.279999999999999e-05,
-      "loss": 4.7288,
+      "loss": 4.6534,
       "step": 90
     },
     {
-      "epoch": 0.2,
+      "epoch": 0.41,
       "learning_rate": 5.339999999999999e-05,
-      "loss": 4.5647,
+      "loss": 4.6014,
       "step": 91
     },
     {
-      "epoch": 0.21,
+      "epoch": 0.41,
       "learning_rate": 5.399999999999999e-05,
-      "loss": 4.8602,
+      "loss": 4.5941,
       "step": 92
     },
     {
-      "epoch": 0.21,
+      "epoch": 0.42,
       "learning_rate": 5.459999999999999e-05,
-      "loss": 4.7511,
+      "loss": 4.5625,
       "step": 93
     },
     {
-      "epoch": 0.21,
+      "epoch": 0.42,
       "learning_rate": 5.519999999999999e-05,
-      "loss": 4.6587,
+      "loss": 4.6271,
       "step": 94
     },
     {
-      "epoch": 0.21,
+      "epoch": 0.43,
       "learning_rate": 5.5799999999999994e-05,
-      "loss": 4.5921,
+      "loss": 4.6085,
       "step": 95
     },
     {
-      "epoch": 0.22,
+      "epoch": 0.43,
       "learning_rate": 5.6399999999999995e-05,
-      "loss": 4.8161,
+      "loss": 4.5724,
       "step": 96
     },
     {
-      "epoch": 0.22,
+      "epoch": 0.43,
       "learning_rate": 5.6999999999999996e-05,
-      "loss": 4.697,
+      "loss": 4.6734,
       "step": 97
     },
     {
-      "epoch": 0.22,
+      "epoch": 0.44,
       "learning_rate": 5.76e-05,
-      "loss": 4.5279,
+      "loss": 4.7297,
       "step": 98
     },
     {
-      "epoch": 0.22,
+      "epoch": 0.44,
       "learning_rate": 5.82e-05,
-      "loss": 4.7026,
+      "loss": 4.601,
       "step": 99
     },
     {
-      "epoch": 0.22,
+      "epoch": 0.45,
       "learning_rate": 5.88e-05,
-      "loss": 4.3868,
+      "loss": 4.5277,
       "step": 100
     },
     {
-      "epoch": 0.23,
-      "learning_rate": 5.88e-05,
-      "loss": 5.9542,
+      "epoch": 0.45,
+      "learning_rate": 5.94e-05,
+      "loss": 5.9227,
       "step": 101
     },
     {
-      "epoch": 0.23,
-      "learning_rate": 5.94e-05,
-      "loss": 5.8248,
+      "epoch": 0.46,
+      "learning_rate": 5.9999999999999995e-05,
+      "loss": 5.8449,
       "step": 102
     },
     {
-      "epoch": 0.23,
-      "learning_rate": 5.9999999999999995e-05,
-      "loss": 5.5644,
+      "epoch": 0.46,
+      "learning_rate": 6.0599999999999996e-05,
+      "loss": 5.5435,
       "step": 103
     },
     {
-      "epoch": 0.23,
-      "learning_rate": 6.0599999999999996e-05,
-      "loss": 5.3902,
+      "epoch": 0.47,
+      "learning_rate": 6.12e-05,
+      "loss": 5.1179,
       "step": 104
     },
     {
-      "epoch": 0.24,
-      "learning_rate": 6.12e-05,
-      "loss": 4.9764,
+      "epoch": 0.47,
+      "learning_rate": 6.18e-05,
+      "loss": 4.9785,
       "step": 105
     },
     {
-      "epoch": 0.24,
-      "learning_rate": 6.18e-05,
-      "loss": 4.8396,
+      "epoch": 0.48,
+      "learning_rate": 6.239999999999999e-05,
+      "loss": 4.8924,
       "step": 106
     },
     {
-      "epoch": 0.24,
-      "learning_rate": 6.239999999999999e-05,
-      "loss": 4.7774,
+      "epoch": 0.48,
+      "learning_rate": 6.299999999999999e-05,
+      "loss": 4.7039,
       "step": 107
     },
     {
-      "epoch": 0.24,
-      "learning_rate": 6.299999999999999e-05,
-      "loss": 4.7074,
+      "epoch": 0.48,
+      "learning_rate": 6.359999999999999e-05,
+      "loss": 4.7125,
       "step": 108
     },
     {
-      "epoch": 0.24,
-      "learning_rate": 6.359999999999999e-05,
-      "loss": 4.6497,
+      "epoch": 0.49,
+      "learning_rate": 6.419999999999999e-05,
+      "loss": 4.6977,
       "step": 109
     },
     {
-      "epoch": 0.25,
-      "learning_rate": 6.419999999999999e-05,
-      "loss": 4.7453,
+      "epoch": 0.49,
+      "learning_rate": 6.479999999999999e-05,
+      "loss": 4.6824,
       "step": 110
     },
     {
-      "epoch": 0.25,
-      "learning_rate": 6.479999999999999e-05,
-      "loss": 4.7379,
+      "epoch": 0.5,
+      "learning_rate": 6.539999999999999e-05,
+      "loss": 4.6191,
       "step": 111
     },
     {
-      "epoch": 0.25,
-      "learning_rate": 6.539999999999999e-05,
-      "loss": 4.7307,
+      "epoch": 0.5,
+      "learning_rate": 6.599999999999999e-05,
+      "loss": 4.6363,
       "step": 112
     },
     {
-      "epoch": 0.25,
-      "learning_rate": 6.599999999999999e-05,
-      "loss": 4.6858,
+      "epoch": 0.51,
+      "learning_rate": 6.659999999999999e-05,
+      "loss": 4.6369,
       "step": 113
     },
     {
-      "epoch": 0.26,
-      "learning_rate": 6.659999999999999e-05,
-      "loss": 4.5898,
+      "epoch": 0.51,
+      "learning_rate": 6.72e-05,
+      "loss": 4.5246,
       "step": 114
     },
     {
-      "epoch": 0.26,
-      "learning_rate": 6.72e-05,
-      "loss": 4.568,
+      "epoch": 0.52,
+      "learning_rate": 6.78e-05,
+      "loss": 4.5943,
       "step": 115
     },
     {
-      "epoch": 0.26,
-      "learning_rate": 6.78e-05,
-      "loss": 4.685,
+      "epoch": 0.52,
+      "learning_rate": 6.84e-05,
+      "loss": 4.5545,
       "step": 116
     },
     {
-      "epoch": 0.26,
-      "learning_rate": 6.84e-05,
-      "loss": 4.5926,
+      "epoch": 0.52,
+      "learning_rate": 6.9e-05,
+      "loss": 4.5898,
       "step": 117
     },
     {
-      "epoch": 0.26,
-      "learning_rate": 6.9e-05,
-      "loss": 4.691,
+      "epoch": 0.53,
+      "learning_rate": 6.96e-05,
+      "loss": 4.5237,
       "step": 118
     },
     {
-      "epoch": 0.27,
-      "learning_rate": 6.96e-05,
-      "loss": 4.6207,
+      "epoch": 0.53,
+      "learning_rate": 7.02e-05,
+      "loss": 4.5241,
       "step": 119
     },
     {
-      "epoch": 0.27,
-      "learning_rate": 7.02e-05,
-      "loss": 4.6662,
+      "epoch": 0.54,
+      "learning_rate": 7.079999999999999e-05,
+      "loss": 4.5408,
       "step": 120
     },
     {
-      "epoch": 0.27,
-      "learning_rate": 7.079999999999999e-05,
-      "loss": 4.4633,
+      "epoch": 0.54,
+      "learning_rate": 7.139999999999999e-05,
+      "loss": 4.4809,
       "step": 121
     },
     {
-      "epoch": 0.27,
-      "learning_rate": 7.139999999999999e-05,
-      "loss": 4.5675,
+      "epoch": 0.55,
+      "learning_rate": 7.199999999999999e-05,
+      "loss": 4.5278,
       "step": 122
     },
     {
-      "epoch": 0.28,
-      "learning_rate": 7.199999999999999e-05,
-      "loss": 4.5613,
+      "epoch": 0.55,
+      "learning_rate": 7.259999999999999e-05,
+      "loss": 4.4764,
       "step": 123
     },
     {
-      "epoch": 0.28,
-      "learning_rate": 7.259999999999999e-05,
-      "loss": 4.5392,
+      "epoch": 0.56,
+      "learning_rate": 7.319999999999999e-05,
+      "loss": 4.4472,
       "step": 124
     },
     {
-      "epoch": 0.28,
-      "learning_rate": 7.319999999999999e-05,
-      "loss": 4.6394,
+      "epoch": 0.56,
+      "learning_rate": 7.379999999999999e-05,
+      "loss": 4.5971,
       "step": 125
     },
     {
-      "epoch": 0.28,
-      "learning_rate": 7.379999999999999e-05,
-      "loss": 4.6477,
+      "epoch": 0.57,
+      "learning_rate": 7.439999999999999e-05,
+      "loss": 4.4813,
       "step": 126
     },
     {
-      "epoch": 0.28,
-      "learning_rate": 7.439999999999999e-05,
-      "loss": 4.5763,
+      "epoch": 0.57,
+      "learning_rate": 7.5e-05,
+      "loss": 4.5371,
       "step": 127
     },
     {
-      "epoch": 0.29,
-      "learning_rate": 7.5e-05,
-      "loss": 4.5374,
+      "epoch": 0.57,
+      "learning_rate": 7.56e-05,
+      "loss": 4.5538,
       "step": 128
     },
     {
-      "epoch": 0.29,
-      "learning_rate": 7.56e-05,
-      "loss": 4.469,
+      "epoch": 0.58,
+      "learning_rate": 7.62e-05,
+      "loss": 4.5222,
       "step": 129
     },
     {
-      "epoch": 0.29,
-      "learning_rate": 7.62e-05,
-      "loss": 4.6234,
+      "epoch": 0.58,
+      "learning_rate": 7.68e-05,
+      "loss": 4.5567,
       "step": 130
     },
     {
-      "epoch": 0.29,
-      "learning_rate": 7.68e-05,
-      "loss": 4.4815,
+      "epoch": 0.59,
+      "learning_rate": 7.74e-05,
+      "loss": 4.5519,
       "step": 131
     },
     {
-      "epoch": 0.3,
-      "learning_rate": 7.74e-05,
-      "loss": 4.58,
+      "epoch": 0.59,
+      "learning_rate": 7.8e-05,
+      "loss": 4.3829,
       "step": 132
     },
     {
-      "epoch": 0.3,
-      "learning_rate": 7.8e-05,
-      "loss": 4.4586,
+      "epoch": 0.6,
+      "learning_rate": 7.86e-05,
+      "loss": 4.3992,
       "step": 133
     },
     {
-      "epoch": 0.3,
-      "learning_rate": 7.86e-05,
-      "loss": 4.512,
+      "epoch": 0.6,
+      "learning_rate": 7.92e-05,
+      "loss": 4.4999,
       "step": 134
     },
     {
-      "epoch": 0.3,
-      "learning_rate": 7.92e-05,
-      "loss": 4.6734,
+      "epoch": 0.61,
+      "learning_rate": 7.98e-05,
+      "loss": 4.5518,
       "step": 135
     },
     {
-      "epoch": 0.3,
-      "learning_rate": 7.98e-05,
-      "loss": 4.5342,
+      "epoch": 0.61,
+      "learning_rate": 8.04e-05,
+      "loss": 4.48,
       "step": 136
     },
     {
-      "epoch": 0.31,
-      "learning_rate": 8.04e-05,
-      "loss": 4.5749,
+      "epoch": 0.61,
+      "learning_rate": 8.1e-05,
+      "loss": 4.5714,
       "step": 137
     },
     {
-      "epoch": 0.31,
-      "learning_rate": 8.1e-05,
-      "loss": 4.6223,
+      "epoch": 0.62,
+      "learning_rate": 8.16e-05,
+      "loss": 4.4858,
       "step": 138
     },
     {
-      "epoch": 0.31,
-      "learning_rate": 8.16e-05,
-      "loss": 4.637,
+      "epoch": 0.62,
+      "learning_rate": 8.22e-05,
+      "loss": 4.4397,
       "step": 139
     },
     {
-      "epoch": 0.31,
-      "learning_rate": 8.22e-05,
-      "loss": 4.6462,
+      "epoch": 0.63,
+      "learning_rate": 8.28e-05,
+      "loss": 4.5127,
       "step": 140
     },
     {
-      "epoch": 0.32,
-      "learning_rate": 8.28e-05,
-      "loss": 4.6995,
+      "epoch": 0.63,
+      "learning_rate": 8.34e-05,
+      "loss": 4.5398,
       "step": 141
     },
     {
-      "epoch": 0.32,
-      "learning_rate": 8.34e-05,
-      "loss": 4.5716,
+      "epoch": 0.64,
+      "learning_rate": 8.4e-05,
+      "loss": 4.5412,
       "step": 142
     },
     {
-      "epoch": 0.32,
-      "learning_rate": 8.4e-05,
-      "loss": 4.5627,
+      "epoch": 0.64,
+      "learning_rate": 8.459999999999998e-05,
+      "loss": 4.6863,
       "step": 143
     },
     {
-      "epoch": 0.32,
-      "learning_rate": 8.459999999999998e-05,
-      "loss": 4.6207,
+      "epoch": 0.65,
+      "learning_rate": 8.519999999999998e-05,
+      "loss": 4.5778,
       "step": 144
     },
     {
-      "epoch": 0.33,
-      "learning_rate": 8.519999999999998e-05,
-      "loss": 4.5196,
+      "epoch": 0.65,
+      "learning_rate": 8.579999999999998e-05,
+      "loss": 4.4345,
       "step": 145
     },
     {
-      "epoch": 0.33,
-      "learning_rate": 8.579999999999998e-05,
-      "loss": 4.6074,
+      "epoch": 0.65,
+      "learning_rate": 8.639999999999999e-05,
+      "loss": 4.4066,
       "step": 146
     },
     {
-      "epoch": 0.33,
-      "learning_rate": 8.639999999999999e-05,
-      "loss": 4.7489,
+      "epoch": 0.66,
+      "learning_rate": 8.699999999999999e-05,
+      "loss": 4.3866,
       "step": 147
     },
     {
-      "epoch": 0.33,
-      "learning_rate": 8.699999999999999e-05,
-      "loss": 4.7682,
+      "epoch": 0.66,
+      "learning_rate": 8.759999999999999e-05,
+      "loss": 4.6398,
       "step": 148
     },
     {
-      "epoch": 0.33,
-      "learning_rate": 8.759999999999999e-05,
-      "loss": 4.5681,
+      "epoch": 0.67,
+      "learning_rate": 8.819999999999999e-05,
+      "loss": 4.4105,
       "step": 149
     },
     {
-      "epoch": 0.34,
-      "learning_rate": 8.819999999999999e-05,
-      "loss": 4.469,
+      "epoch": 0.67,
+      "learning_rate": 8.879999999999999e-05,
+      "loss": 4.4866,
       "step": 150
     },
     {
-      "epoch": 0.34,
-      "learning_rate": 8.879999999999999e-05,
-      "loss": 6.2583,
+      "epoch": 0.68,
+      "learning_rate": 8.939999999999999e-05,
+      "loss": 6.0924,
       "step": 151
     },
     {
-      "epoch": 0.34,
-      "learning_rate": 8.939999999999999e-05,
-      "loss": 5.6164,
+      "epoch": 0.68,
+      "learning_rate": 8.999999999999999e-05,
+      "loss": 5.8545,
       "step": 152
     },
     {
-      "epoch": 0.34,
-      "learning_rate": 8.999999999999999e-05,
-      "loss": 5.1907,
+      "epoch": 0.69,
+      "learning_rate": 9.059999999999999e-05,
+      "loss": 5.3625,
       "step": 153
     },
     {
-      "epoch": 0.35,
-      "learning_rate": 9.059999999999999e-05,
-      "loss": 4.7733,
+      "epoch": 0.69,
+      "learning_rate": 9.12e-05,
+      "loss": 4.9468,
       "step": 154
     },
     {
-      "epoch": 0.35,
-      "learning_rate": 9.12e-05,
-      "loss": 4.8415,
+      "epoch": 0.7,
+      "learning_rate": 9.18e-05,
+      "loss": 4.8548,
       "step": 155
     },
     {
-      "epoch": 0.35,
-      "learning_rate": 9.18e-05,
-      "loss": 4.6594,
+      "epoch": 0.7,
+      "learning_rate": 9.24e-05,
+      "loss": 4.7858,
       "step": 156
     },
     {
-      "epoch": 0.35,
-      "learning_rate": 9.24e-05,
-      "loss": 4.6,
+      "epoch": 0.7,
+      "learning_rate": 9.3e-05,
+      "loss": 4.6516,
       "step": 157
     },
     {
-      "epoch": 0.35,
-      "learning_rate": 9.3e-05,
-      "loss": 4.659,
+      "epoch": 0.71,
+      "learning_rate": 9.36e-05,
+      "loss": 4.6538,
       "step": 158
     },
     {
-      "epoch": 0.36,
-      "learning_rate": 9.36e-05,
-      "loss": 4.6704,
+      "epoch": 0.71,
+      "learning_rate": 9.419999999999999e-05,
+      "loss": 4.5987,
       "step": 159
     },
     {
-      "epoch": 0.36,
-      "learning_rate": 9.419999999999999e-05,
-      "loss": 4.7041,
+      "epoch": 0.72,
+      "learning_rate": 9.479999999999999e-05,
+      "loss": 4.6631,
       "step": 160
     },
     {
-      "epoch": 0.36,
-      "learning_rate": 9.479999999999999e-05,
-      "loss": 4.7007,
+      "epoch": 0.72,
+      "learning_rate": 9.539999999999999e-05,
+      "loss": 4.4868,
       "step": 161
     },
     {
-      "epoch": 0.36,
-      "learning_rate": 9.539999999999999e-05,
-      "loss": 4.6682,
+      "epoch": 0.73,
+      "learning_rate": 9.599999999999999e-05,
+      "loss": 4.5287,
       "step": 162
     },
     {
-      "epoch": 0.37,
-      "learning_rate": 9.599999999999999e-05,
-      "loss": 4.6197,
+      "epoch": 0.73,
+      "learning_rate": 9.659999999999999e-05,
+      "loss": 4.5528,
       "step": 163
     },
     {
-      "epoch": 0.37,
-      "learning_rate": 9.659999999999999e-05,
-      "loss": 4.62,
+      "epoch": 0.74,
+      "learning_rate": 9.719999999999999e-05,
+      "loss": 4.5033,
       "step": 164
     },
     {
-      "epoch": 0.37,
-      "learning_rate": 9.719999999999999e-05,
-      "loss": 4.4891,
+      "epoch": 0.74,
+      "learning_rate": 9.779999999999999e-05,
+      "loss": 4.5908,
       "step": 165
     },
     {
-      "epoch": 0.37,
-      "learning_rate": 9.779999999999999e-05,
-      "loss": 4.5451,
+      "epoch": 0.74,
+      "learning_rate": 9.839999999999999e-05,
+      "loss": 4.4931,
       "step": 166
     },
     {
-      "epoch": 0.37,
-      "learning_rate": 9.839999999999999e-05,
-      "loss": 4.4452,
+      "epoch": 0.75,
+      "learning_rate": 9.9e-05,
+      "loss": 4.519,
       "step": 167
     },
     {
-      "epoch": 0.38,
-      "learning_rate": 9.9e-05,
-      "loss": 4.5057,
+      "epoch": 0.75,
+      "learning_rate": 9.96e-05,
+      "loss": 4.5015,
       "step": 168
     },
     {
-      "epoch": 0.38,
-      "learning_rate": 9.96e-05,
-      "loss": 4.5834,
+      "epoch": 0.76,
+      "learning_rate": 0.0001002,
+      "loss": 4.3689,
       "step": 169
     },
     {
-      "epoch": 0.38,
-      "learning_rate": 0.0001002,
-      "loss": 4.5699,
+      "epoch": 0.76,
+      "learning_rate": 0.0001008,
+      "loss": 4.5586,
       "step": 170
     },
     {
-      "epoch": 0.38,
-      "learning_rate": 0.0001008,
-      "loss": 4.5678,
+      "epoch": 0.77,
+      "learning_rate": 0.0001014,
+      "loss": 4.4859,
       "step": 171
     },
     {
-      "epoch": 0.39,
-      "learning_rate": 0.0001014,
-      "loss": 4.6897,
+      "epoch": 0.77,
+      "learning_rate": 0.000102,
+      "loss": 4.3494,
       "step": 172
     },
     {
-      "epoch": 0.39,
-      "learning_rate": 0.000102,
-      "loss": 4.4966,
+      "epoch": 0.78,
+      "learning_rate": 0.0001026,
+      "loss": 4.4217,
       "step": 173
     },
     {
-      "epoch": 0.39,
-      "learning_rate": 0.0001026,
-      "loss": 4.6041,
+      "epoch": 0.78,
+      "learning_rate": 0.00010319999999999999,
+      "loss": 4.3855,
       "step": 174
     },
     {
-      "epoch": 0.39,
-      "learning_rate": 0.00010319999999999999,
-      "loss": 4.6913,
+      "epoch": 0.78,
+      "learning_rate": 0.00010379999999999999,
+      "loss": 4.4697,
       "step": 175
     },
     {
-      "epoch": 0.39,
-      "learning_rate": 0.00010379999999999999,
-      "loss": 4.4995,
+      "epoch": 0.79,
+      "learning_rate": 0.00010439999999999999,
+      "loss": 4.4196,
       "step": 176
     },
     {
-      "epoch": 0.4,
-      "learning_rate": 0.00010439999999999999,
-      "loss": 4.4945,
+      "epoch": 0.79,
+      "learning_rate": 0.00010499999999999999,
+      "loss": 4.438,
       "step": 177
     },
     {
-      "epoch": 0.4,
-      "learning_rate": 0.00010499999999999999,
-      "loss": 4.5747,
+      "epoch": 0.8,
+      "learning_rate": 0.00010559999999999998,
+      "loss": 4.3658,
       "step": 178
     },
     {
-      "epoch": 0.4,
-      "learning_rate": 0.00010559999999999998,
-      "loss": 4.631,
+      "epoch": 0.8,
+      "learning_rate": 0.00010619999999999998,
+      "loss": 4.4174,
       "step": 179
     },
     {
-      "epoch": 0.4,
-      "learning_rate": 0.00010619999999999998,
-      "loss": 4.6599,
+      "epoch": 0.81,
+      "learning_rate": 0.00010679999999999998,
+      "loss": 4.4003,
       "step": 180
     },
     {
-      "epoch": 0.41,
-      "learning_rate": 0.00010679999999999998,
-      "loss": 4.5648,
+      "epoch": 0.81,
+      "learning_rate": 0.00010739999999999998,
+      "loss": 4.5487,
       "step": 181
     },
     {
-      "epoch": 0.41,
-      "learning_rate": 0.00010739999999999998,
-      "loss": 4.5608,
+      "epoch": 0.82,
+      "learning_rate": 0.00010799999999999998,
+      "loss": 4.35,
       "step": 182
     },
     {
-      "epoch": 0.41,
-      "learning_rate": 0.00010799999999999998,
-      "loss": 4.5623,
+      "epoch": 0.82,
+      "learning_rate": 0.00010859999999999998,
+      "loss": 4.5236,
       "step": 183
     },
     {
-      "epoch": 0.41,
-      "learning_rate": 0.00010859999999999998,
-      "loss": 4.6001,
+      "epoch": 0.83,
+      "learning_rate": 0.00010919999999999998,
+      "loss": 4.4523,
       "step": 184
     },
     {
-      "epoch": 0.41,
-      "learning_rate": 0.00010919999999999998,
-      "loss": 4.5398,
+      "epoch": 0.83,
+      "learning_rate": 0.00010979999999999999,
+      "loss": 4.4892,
       "step": 185
     },
     {
-      "epoch": 0.42,
-      "learning_rate": 0.00010979999999999999,
-      "loss": 4.4157,
+      "epoch": 0.83,
+      "learning_rate": 0.00011039999999999999,
+      "loss": 4.3919,
       "step": 186
     },
     {
-      "epoch": 0.42,
-      "learning_rate": 0.00011039999999999999,
-      "loss": 4.4919,
+      "epoch": 0.84,
+      "learning_rate": 0.00011099999999999999,
+      "loss": 4.4118,
       "step": 187
     },
     {
-      "epoch": 0.42,
-      "learning_rate": 0.00011099999999999999,
-      "loss": 4.4482,
+      "epoch": 0.84,
+      "learning_rate": 0.00011159999999999999,
+      "loss": 4.5025,
       "step": 188
     },
     {
-      "epoch": 0.42,
-      "learning_rate": 0.00011159999999999999,
-      "loss": 4.5045,
+      "epoch": 0.85,
+      "learning_rate": 0.00011219999999999999,
+      "loss": 4.4032,
       "step": 189
     },
     {
-      "epoch": 0.43,
-      "learning_rate": 0.00011219999999999999,
-      "loss": 4.618,
+      "epoch": 0.85,
+      "learning_rate": 0.00011279999999999999,
+      "loss": 4.4338,
       "step": 190
     },
     {
-      "epoch": 0.43,
-      "learning_rate": 0.00011279999999999999,
-      "loss": 4.5325,
+      "epoch": 0.86,
+      "learning_rate": 0.00011339999999999999,
+      "loss": 4.4694,
       "step": 191
     },
     {
-      "epoch": 0.43,
-      "learning_rate": 0.00011339999999999999,
-      "loss": 4.5477,
+      "epoch": 0.86,
+      "learning_rate": 0.00011399999999999999,
+      "loss": 4.3931,
       "step": 192
     },
     {
-      "epoch": 0.43,
-      "learning_rate": 0.00011399999999999999,
-      "loss": 4.4201,
+      "epoch": 0.87,
+      "learning_rate": 0.0001146,
+      "loss": 4.3904,
       "step": 193
     },
     {
-      "epoch": 0.43,
-      "learning_rate": 0.0001146,
-      "loss": 4.5925,
+      "epoch": 0.87,
+      "learning_rate": 0.0001152,
+      "loss": 4.4158,
       "step": 194
     },
     {
-      "epoch": 0.44,
-      "learning_rate": 0.0001152,
-      "loss": 4.6806,
+      "epoch": 0.87,
+      "learning_rate": 0.0001158,
+      "loss": 4.4491,
       "step": 195
     },
     {
-      "epoch": 0.44,
-      "learning_rate": 0.0001158,
-      "loss": 4.5368,
+      "epoch": 0.88,
+      "learning_rate": 0.0001164,
+      "loss": 4.4745,
       "step": 196
     },
     {
-      "epoch": 0.44,
-      "learning_rate": 0.0001164,
-      "loss": 4.4654,
+      "epoch": 0.88,
+      "learning_rate": 0.000117,
+      "loss": 4.4689,
       "step": 197
     },
     {
-      "epoch": 0.44,
-      "learning_rate": 0.000117,
-      "loss": 4.5484,
+      "epoch": 0.89,
+      "learning_rate": 0.0001176,
+      "loss": 4.4095,
       "step": 198
     },
     {
-      "epoch": 0.45,
-      "learning_rate": 0.0001176,
-      "loss": 4.4713,
+      "epoch": 0.89,
+      "learning_rate": 0.0001182,
+      "loss": 4.461,
       "step": 199
     },
     {
-      "epoch": 0.45,
-      "learning_rate": 0.0001182,
-      "loss": 4.5144,
+      "epoch": 0.9,
+      "learning_rate": 0.0001188,
+      "loss": 4.2483,
       "step": 200
     },
     {
-      "epoch": 0.45,
-      "learning_rate": 0.0001188,
-      "loss": 5.4254,
+      "epoch": 0.9,
+      "learning_rate": 0.0001194,
+      "loss": 5.7607,
       "step": 201
     },
     {
-      "epoch": 0.45,
-      "learning_rate": 0.0001194,
-      "loss": 5.4006,
+      "epoch": 0.91,
+      "learning_rate": 0.00011999999999999999,
+      "loss": 5.7126,
       "step": 202
     },
     {
-      "epoch": 0.46,
-      "learning_rate": 0.00011999999999999999,
-      "loss": 5.2168,
+      "epoch": 0.91,
+      "learning_rate": 0.00012059999999999999,
+      "loss": 5.3124,
       "step": 203
     },
     {
-      "epoch": 0.46,
-      "learning_rate": 0.00012059999999999999,
-      "loss": 4.9626,
+      "epoch": 0.91,
+      "learning_rate": 0.00012119999999999999,
+      "loss": 4.9813,
       "step": 204
     },
     {
-      "epoch": 0.46,
-      "learning_rate": 0.00012119999999999999,
-      "loss": 4.8992,
+      "epoch": 0.92,
+      "learning_rate": 0.00012179999999999999,
+      "loss": 4.7145,
       "step": 205
     },
     {
-      "epoch": 0.46,
-      "learning_rate": 0.00012179999999999999,
-      "loss": 4.811,
+      "epoch": 0.92,
+      "learning_rate": 0.0001224,
+      "loss": 4.7888,
       "step": 206
     },
     {
-      "epoch": 0.46,
-      "learning_rate": 0.0001224,
-      "loss": 4.6223,
+      "epoch": 0.93,
+      "learning_rate": 0.00012299999999999998,
+      "loss": 4.6485,
       "step": 207
     },
     {
-      "epoch": 0.47,
-      "learning_rate": 0.00012299999999999998,
-      "loss": 4.7224,
+      "epoch": 0.93,
+      "learning_rate": 0.0001236,
+      "loss": 4.6296,
       "step": 208
     },
     {
-      "epoch": 0.47,
-      "learning_rate": 0.0001236,
-      "loss": 4.7679,
+      "epoch": 0.94,
+      "learning_rate": 0.00012419999999999998,
+      "loss": 4.4968,
       "step": 209
     },
     {
-      "epoch": 0.47,
-      "learning_rate": 0.00012419999999999998,
-      "loss": 4.6853,
+      "epoch": 0.94,
+      "learning_rate": 0.00012479999999999997,
+      "loss": 4.514,
       "step": 210
     },
     {
-      "epoch": 0.47,
-      "learning_rate": 0.00012479999999999997,
-      "loss": 4.5118,
+      "epoch": 0.95,
+      "learning_rate": 0.00012539999999999999,
+      "loss": 4.5182,
       "step": 211
     },
     {
-      "epoch": 0.48,
-      "learning_rate": 0.00012539999999999999,
-      "loss": 4.6325,
+      "epoch": 0.95,
+      "learning_rate": 0.00012599999999999997,
+      "loss": 4.534,
       "step": 212
     },
     {
-      "epoch": 0.48,
-      "learning_rate": 0.00012599999999999997,
-      "loss": 4.4748,
+      "epoch": 0.96,
+      "learning_rate": 0.0001266,
+      "loss": 4.5318,
       "step": 213
     },
     {
-      "epoch": 0.48,
-      "learning_rate": 0.0001266,
-      "loss": 4.513,
+      "epoch": 0.96,
+      "learning_rate": 0.00012719999999999997,
+      "loss": 4.4489,
       "step": 214
     },
     {
-      "epoch": 0.48,
-      "learning_rate": 0.00012719999999999997,
-      "loss": 4.5576,
+      "epoch": 0.96,
+      "learning_rate": 0.0001278,
+      "loss": 4.3867,
       "step": 215
     },
     {
-      "epoch": 0.48,
-      "learning_rate": 0.0001278,
-      "loss": 4.555,
+      "epoch": 0.97,
+      "learning_rate": 0.00012839999999999998,
+      "loss": 4.5466,
       "step": 216
     },
     {
-      "epoch": 0.49,
-      "learning_rate": 0.00012839999999999998,
-      "loss": 4.5305,
+      "epoch": 0.97,
+      "learning_rate": 0.000129,
+      "loss": 4.3823,
       "step": 217
     },
     {
-      "epoch": 0.49,
-      "learning_rate": 0.000129,
-      "loss": 4.5359,
+      "epoch": 0.98,
+      "learning_rate": 0.00012959999999999998,
+      "loss": 4.3786,
       "step": 218
     },
     {
-      "epoch": 0.49,
-      "learning_rate": 0.00012959999999999998,
-      "loss": 4.4294,
+      "epoch": 0.98,
+      "learning_rate": 0.0001302,
+      "loss": 4.4267,
       "step": 219
     },
     {
-      "epoch": 0.49,
-      "learning_rate": 0.0001302,
-      "loss": 4.5847,
+      "epoch": 0.99,
+      "learning_rate": 0.00013079999999999998,
+      "loss": 4.433,
       "step": 220
     },
     {
-      "epoch": 0.5,
-      "learning_rate": 0.00013079999999999998,
-      "loss": 4.4666,
+      "epoch": 0.99,
+      "learning_rate": 0.0001314,
+      "loss": 4.4993,
       "step": 221
     },
     {
-      "epoch": 0.5,
-      "learning_rate": 0.0001314,
-      "loss": 4.5346,
+      "epoch": 1.0,
+      "learning_rate": 0.00013199999999999998,
+      "loss": 4.3747,
       "step": 222
     },
     {
-      "epoch": 0.5,
-      "learning_rate": 0.00013199999999999998,
-      "loss": 4.4938,
+      "epoch": 1.0,
+      "learning_rate": 0.0001326,
+      "loss": 4.3646,
       "step": 223
     },
     {
-      "epoch": 0.5,
-      "learning_rate": 0.0001326,
-      "loss": 4.4626,
+      "epoch": 1.0,
+      "learning_rate": 0.00013319999999999999,
+      "loss": 5.5865,
       "step": 224
     },
     {
-      "epoch": 0.5,
-      "learning_rate": 0.00013319999999999999,
-      "loss": 4.668,
+      "epoch": 1.01,
+      "learning_rate": 0.0001338,
+      "loss": 5.3673,
       "step": 225
     },
     {
-      "epoch": 0.51,
-      "learning_rate": 0.0001338,
-      "loss": 4.4876,
+      "epoch": 1.01,
+      "learning_rate": 0.0001344,
+      "loss": 5.1522,
       "step": 226
     },
     {
-      "epoch": 0.51,
-      "learning_rate": 0.0001344,
-      "loss": 4.6419,
+      "epoch": 1.02,
+      "learning_rate": 0.000135,
+      "loss": 4.7286,
       "step": 227
     },
     {
-      "epoch": 0.51,
-      "learning_rate": 0.000135,
-      "loss": 4.5957,
+      "epoch": 1.02,
+      "learning_rate": 0.0001356,
+      "loss": 4.6139,
       "step": 228
     },
     {
-      "epoch": 0.51,
-      "learning_rate": 0.0001356,
-      "loss": 4.5052,
+      "epoch": 1.03,
+      "learning_rate": 0.0001362,
+      "loss": 4.5005,
       "step": 229
     },
     {
-      "epoch": 0.52,
-      "learning_rate": 0.0001362,
-      "loss": 4.5292,
+      "epoch": 1.03,
+      "learning_rate": 0.0001368,
+      "loss": 4.4463,
       "step": 230
     },
     {
-      "epoch": 0.52,
-      "learning_rate": 0.0001368,
-      "loss": 4.5961,
+      "epoch": 1.04,
+      "learning_rate": 0.0001374,
+      "loss": 4.4191,
       "step": 231
     },
     {
-      "epoch": 0.52,
-      "learning_rate": 0.0001374,
-      "loss": 4.4206,
+      "epoch": 1.04,
+      "learning_rate": 0.000138,
+      "loss": 4.3203,
       "step": 232
     },
     {
-      "epoch": 0.52,
-      "learning_rate": 0.000138,
-      "loss": 4.4587,
+      "epoch": 1.04,
+      "learning_rate": 0.0001386,
+      "loss": 4.4397,
       "step": 233
     },
     {
-      "epoch": 0.52,
-      "learning_rate": 0.0001386,
-      "loss": 4.51,
+      "epoch": 1.05,
+      "learning_rate": 0.0001392,
+      "loss": 4.3437,
       "step": 234
     },
     {
-      "epoch": 0.53,
-      "learning_rate": 0.0001392,
-      "loss": 4.5874,
+      "epoch": 1.05,
+      "learning_rate": 0.00013979999999999998,
+      "loss": 4.3979,
       "step": 235
     },
     {
-      "epoch": 0.53,
-      "learning_rate": 0.00013979999999999998,
-      "loss": 4.4853,
+      "epoch": 1.06,
+      "learning_rate": 0.0001404,
+      "loss": 4.4071,
       "step": 236
     },
     {
-      "epoch": 0.53,
-      "learning_rate": 0.0001404,
-      "loss": 4.5932,
+      "epoch": 1.06,
+      "learning_rate": 0.00014099999999999998,
+      "loss": 4.3001,
       "step": 237
     },
     {
-      "epoch": 0.53,
-      "learning_rate": 0.00014099999999999998,
-      "loss": 4.4827,
+      "epoch": 1.07,
+      "learning_rate": 0.00014159999999999997,
+      "loss": 4.36,
       "step": 238
     },
     {
-      "epoch": 0.54,
-      "learning_rate": 0.00014159999999999997,
-      "loss": 4.4535,
+      "epoch": 1.07,
+      "learning_rate": 0.0001422,
+      "loss": 4.3493,
       "step": 239
     },
     {
-      "epoch": 0.54,
-      "learning_rate": 0.0001422,
-      "loss": 4.532,
+      "epoch": 1.08,
+      "learning_rate": 0.00014279999999999997,
+      "loss": 4.4577,
       "step": 240
     },
     {
-      "epoch": 0.54,
-      "learning_rate": 0.00014279999999999997,
-      "loss": 4.5145,
+      "epoch": 1.08,
+      "learning_rate": 0.0001434,
+      "loss": 4.2136,
       "step": 241
     },
     {
-      "epoch": 0.54,
-      "learning_rate": 0.0001434,
-      "loss": 4.5681,
+      "epoch": 1.09,
+      "learning_rate": 0.00014399999999999998,
+      "loss": 4.2935,
       "step": 242
     },
     {
-      "epoch": 0.54,
-      "learning_rate": 0.00014399999999999998,
-      "loss": 4.7017,
+      "epoch": 1.09,
+      "learning_rate": 0.0001446,
+      "loss": 4.2193,
       "step": 243
     },
     {
-      "epoch": 0.55,
-      "learning_rate": 0.0001446,
-      "loss": 4.5186,
+      "epoch": 1.09,
+      "learning_rate": 0.00014519999999999998,
+      "loss": 4.3509,
       "step": 244
     },
     {
-      "epoch": 0.55,
-      "learning_rate": 0.00014519999999999998,
-      "loss": 4.5254,
+      "epoch": 1.1,
+      "learning_rate": 0.0001458,
+      "loss": 4.2148,
       "step": 245
     },
     {
-      "epoch": 0.55,
-      "learning_rate": 0.0001458,
-      "loss": 4.4415,
+      "epoch": 1.1,
+      "learning_rate": 0.00014639999999999998,
+      "loss": 4.2056,
       "step": 246
     },
     {
-      "epoch": 0.55,
-      "learning_rate": 0.00014639999999999998,
-      "loss": 4.3858,
+      "epoch": 1.11,
+      "learning_rate": 0.000147,
+      "loss": 4.2163,
       "step": 247
     },
     {
-      "epoch": 0.56,
-      "learning_rate": 0.000147,
-      "loss": 4.6209,
+      "epoch": 1.11,
+      "learning_rate": 0.00014759999999999998,
+      "loss": 4.2646,
       "step": 248
     },
     {
-      "epoch": 0.56,
-      "learning_rate": 0.00014759999999999998,
-      "loss": 4.5217,
+      "epoch": 1.12,
+      "learning_rate": 0.0001482,
+      "loss": 4.1475,
       "step": 249
     },
     {
-      "epoch": 0.56,
-      "learning_rate": 0.0001482,
-      "loss": 4.3601,
+      "epoch": 1.12,
+      "learning_rate": 0.00014879999999999998,
+      "loss": 4.1661,
       "step": 250
     },
     {
-      "epoch": 0.56,
-      "learning_rate": 0.00014879999999999998,
-      "loss": 5.5394,
+      "epoch": 1.13,
+      "learning_rate": 0.0001494,
+      "loss": 4.0668,
       "step": 251
     },
     {
-      "epoch": 0.57,
-      "learning_rate": 0.0001494,
-      "loss": 5.4343,
+      "epoch": 1.13,
+      "learning_rate": 0.00015,
+      "loss": 4.0718,
       "step": 252
     },
     {
-      "epoch": 0.57,
-      "learning_rate": 0.00015,
-      "loss": 5.2117,
+      "epoch": 1.13,
+      "learning_rate": 0.00015059999999999997,
+      "loss": 4.1521,
       "step": 253
     },
     {
-      "epoch": 0.57,
-      "learning_rate": 0.00015059999999999997,
-      "loss": 4.8667,
+      "epoch": 1.14,
+      "learning_rate": 0.0001512,
+      "loss": 4.0864,
       "step": 254
     },
     {
-      "epoch": 0.57,
-      "learning_rate": 0.0001512,
-      "loss": 4.7689,
+      "epoch": 1.14,
+      "learning_rate": 0.00015179999999999998,
+      "loss": 4.0792,
       "step": 255
     },
     {
-      "epoch": 0.57,
-      "learning_rate": 0.00015179999999999998,
-      "loss": 4.7223,
+      "epoch": 1.15,
+      "learning_rate": 0.0001524,
+      "loss": 4.0559,
       "step": 256
     },
     {
-      "epoch": 0.58,
-      "learning_rate": 0.0001524,
-      "loss": 4.6127,
+      "epoch": 1.15,
+      "learning_rate": 0.00015299999999999998,
+      "loss": 4.0964,
       "step": 257
     },
     {
-      "epoch": 0.58,
-      "learning_rate": 0.00015299999999999998,
-      "loss": 4.5517,
+      "epoch": 1.16,
+      "learning_rate": 0.0001536,
+      "loss": 4.1454,
       "step": 258
     },
     {
-      "epoch": 0.58,
-      "learning_rate": 0.0001536,
-      "loss": 4.6104,
+      "epoch": 1.16,
+      "learning_rate": 0.00015419999999999998,
+      "loss": 4.0752,
       "step": 259
     },
     {
-      "epoch": 0.58,
-      "learning_rate": 0.00015419999999999998,
-      "loss": 4.5236,
+      "epoch": 1.17,
+      "learning_rate": 0.0001548,
+      "loss": 4.0471,
       "step": 260
     },
     {
-      "epoch": 0.59,
-      "learning_rate": 0.0001548,
-      "loss": 4.6442,
+      "epoch": 1.17,
+      "learning_rate": 0.00015539999999999998,
+      "loss": 4.1324,
       "step": 261
     },
     {
-      "epoch": 0.59,
-      "learning_rate": 0.00015539999999999998,
-      "loss": 4.5383,
+      "epoch": 1.17,
+      "learning_rate": 0.000156,
+      "loss": 4.0548,
       "step": 262
     },
     {
-      "epoch": 0.59,
-      "learning_rate": 0.000156,
-      "loss": 4.698,
+      "epoch": 1.18,
+      "learning_rate": 0.00015659999999999998,
+      "loss": 3.9025,
       "step": 263
     },
     {
-      "epoch": 0.59,
-      "learning_rate": 0.00015659999999999998,
-      "loss": 4.5208,
+      "epoch": 1.18,
+      "learning_rate": 0.0001572,
+      "loss": 4.1022,
       "step": 264
     },
     {
-      "epoch": 0.59,
-      "learning_rate": 0.0001572,
-      "loss": 4.5523,
+      "epoch": 1.19,
+      "learning_rate": 0.0001578,
+      "loss": 3.984,
       "step": 265
     },
     {
-      "epoch": 0.6,
-      "learning_rate": 0.0001578,
-      "loss": 4.5186,
+      "epoch": 1.19,
+      "learning_rate": 0.0001584,
+      "loss": 4.0924,
       "step": 266
     },
     {
-      "epoch": 0.6,
-      "learning_rate": 0.0001584,
-      "loss": 4.6301,
+      "epoch": 1.2,
+      "learning_rate": 0.000159,
+      "loss": 3.967,
       "step": 267
     },
     {
-      "epoch": 0.6,
-      "learning_rate": 0.000159,
-      "loss": 4.4806,
+      "epoch": 1.2,
+      "learning_rate": 0.0001596,
+      "loss": 4.06,
       "step": 268
     },
     {
-      "epoch": 0.6,
-      "learning_rate": 0.0001596,
-      "loss": 4.5985,
+      "epoch": 1.21,
+      "learning_rate": 0.0001602,
+      "loss": 3.8391,
       "step": 269
     },
     {
-      "epoch": 0.61,
-      "learning_rate": 0.0001602,
-      "loss": 4.5172,
+      "epoch": 1.21,
+      "learning_rate": 0.0001608,
+      "loss": 3.8725,
       "step": 270
     },
     {
-      "epoch": 0.61,
-      "learning_rate": 0.0001608,
-      "loss": 4.4582,
+      "epoch": 1.22,
+      "learning_rate": 0.0001614,
+      "loss": 3.947,
       "step": 271
     },
     {
-      "epoch": 0.61,
-      "learning_rate": 0.0001614,
-      "loss": 4.5442,
+      "epoch": 1.22,
+      "learning_rate": 0.000162,
+      "loss": 3.8932,
       "step": 272
     },
     {
-      "epoch": 0.61,
-      "learning_rate": 0.000162,
-      "loss": 4.3896,
+      "epoch": 1.22,
+      "learning_rate": 0.0001626,
+      "loss": 3.6977,
       "step": 273
     },
     {
-      "epoch": 0.61,
-      "learning_rate": 0.0001626,
-      "loss": 4.4235,
+      "epoch": 1.23,
+      "learning_rate": 0.0001632,
+      "loss": 5.8216,
       "step": 274
     },
     {
-      "epoch": 0.62,
-      "learning_rate": 0.0001632,
-      "loss": 4.5386,
+      "epoch": 1.23,
+      "learning_rate": 0.0001638,
+      "loss": 5.3449,
       "step": 275
     },
     {
-      "epoch": 0.62,
-      "learning_rate": 0.0001638,
-      "loss": 4.4787,
+      "epoch": 1.24,
+      "learning_rate": 0.0001644,
+      "loss": 4.9906,
       "step": 276
     },
     {
-      "epoch": 0.62,
-      "learning_rate": 0.0001644,
-      "loss": 4.4511,
+      "epoch": 1.24,
+      "learning_rate": 0.000165,
+      "loss": 4.7723,
       "step": 277
     },
     {
-      "epoch": 0.62,
-      "learning_rate": 0.000165,
-      "loss": 4.5093,
+      "epoch": 1.25,
+      "learning_rate": 0.0001656,
+      "loss": 4.5841,
       "step": 278
     },
     {
-      "epoch": 0.63,
-      "learning_rate": 0.0001656,
-      "loss": 4.5233,
+      "epoch": 1.25,
+      "learning_rate": 0.0001662,
+      "loss": 4.5655,
       "step": 279
     },
     {
-      "epoch": 0.63,
-      "learning_rate": 0.0001662,
-      "loss": 4.5999,
+      "epoch": 1.26,
+      "learning_rate": 0.0001668,
+      "loss": 4.5075,
       "step": 280
     },
     {
-      "epoch": 0.63,
-      "learning_rate": 0.0001668,
-      "loss": 4.5561,
+      "epoch": 1.26,
+      "learning_rate": 0.0001674,
+      "loss": 4.3671,
       "step": 281
     },
     {
-      "epoch": 0.63,
-      "learning_rate": 0.0001674,
-      "loss": 4.3713,
+      "epoch": 1.26,
+      "learning_rate": 0.000168,
+      "loss": 4.4138,
       "step": 282
     },
     {
-      "epoch": 0.63,
-      "learning_rate": 0.000168,
-      "loss": 4.3426,
+      "epoch": 1.27,
+      "learning_rate": 0.0001686,
+      "loss": 4.3305,
       "step": 283
     },
     {
-      "epoch": 0.64,
-      "learning_rate": 0.0001686,
-      "loss": 4.5017,
+      "epoch": 1.27,
+      "learning_rate": 0.00016919999999999997,
+      "loss": 4.2737,
       "step": 284
     },
     {
-      "epoch": 0.64,
-      "learning_rate": 0.00016919999999999997,
-      "loss": 4.5855,
+      "epoch": 1.28,
+      "learning_rate": 0.00016979999999999998,
+      "loss": 4.3387,
       "step": 285
     },
     {
-      "epoch": 0.64,
-      "learning_rate": 0.00016979999999999998,
-      "loss": 4.5,
+      "epoch": 1.28,
+      "learning_rate": 0.00017039999999999997,
+      "loss": 4.218,
       "step": 286
     },
     {
-      "epoch": 0.64,
-      "learning_rate": 0.00017039999999999997,
-      "loss": 4.5955,
+      "epoch": 1.29,
+      "learning_rate": 0.00017099999999999998,
+      "loss": 4.2599,
       "step": 287
     },
     {
-      "epoch": 0.65,
-      "learning_rate": 0.00017099999999999998,
-      "loss": 4.4935,
+      "epoch": 1.29,
+      "learning_rate": 0.00017159999999999997,
+      "loss": 4.2436,
       "step": 288
     },
     {
-      "epoch": 0.65,
-      "learning_rate": 0.00017159999999999997,
-      "loss": 4.4523,
+      "epoch": 1.3,
+      "learning_rate": 0.00017219999999999998,
+      "loss": 4.1241,
       "step": 289
     },
     {
-      "epoch": 0.65,
-      "learning_rate": 0.00017219999999999998,
-      "loss": 4.4598,
+      "epoch": 1.3,
+      "learning_rate": 0.00017279999999999997,
+      "loss": 4.2546,
       "step": 290
     },
     {
-      "epoch": 0.65,
-      "learning_rate": 0.00017279999999999997,
-      "loss": 4.6806,
+      "epoch": 1.3,
+      "learning_rate": 0.00017339999999999996,
+      "loss": 4.1435,
       "step": 291
     },
     {
-      "epoch": 0.65,
-      "learning_rate": 0.00017339999999999996,
-      "loss": 4.5561,
+      "epoch": 1.31,
+      "learning_rate": 0.00017399999999999997,
+      "loss": 4.0901,
       "step": 292
     },
     {
-      "epoch": 0.66,
-      "learning_rate": 0.00017399999999999997,
-      "loss": 4.5914,
+      "epoch": 1.31,
+      "learning_rate": 0.00017459999999999996,
+      "loss": 4.1774,
       "step": 293
     },
     {
-      "epoch": 0.66,
-      "learning_rate": 0.00017459999999999996,
-      "loss": 4.537,
+      "epoch": 1.32,
+      "learning_rate": 0.00017519999999999998,
+      "loss": 4.1511,
       "step": 294
     },
     {
-      "epoch": 0.66,
-      "learning_rate": 0.00017519999999999998,
-      "loss": 4.4617,
+      "epoch": 1.32,
+      "learning_rate": 0.00017579999999999996,
+      "loss": 4.1222,
       "step": 295
     },
     {
-      "epoch": 0.66,
-      "learning_rate": 0.00017579999999999996,
-      "loss": 4.4864,
+      "epoch": 1.33,
+      "learning_rate": 0.00017639999999999998,
+      "loss": 4.0986,
       "step": 296
     },
     {
-      "epoch": 0.67,
-      "learning_rate": 0.00017639999999999998,
-      "loss": 4.4368,
+      "epoch": 1.33,
+      "learning_rate": 0.00017699999999999997,
+      "loss": 4.0735,
       "step": 297
     },
     {
-      "epoch": 0.67,
-      "learning_rate": 0.00017699999999999997,
-      "loss": 4.6713,
+      "epoch": 1.34,
+      "learning_rate": 0.00017759999999999998,
+      "loss": 4.0722,
       "step": 298
     },
     {
-      "epoch": 0.67,
-      "learning_rate": 0.00017759999999999998,
-      "loss": 4.2836,
+      "epoch": 1.34,
+      "learning_rate": 0.00017819999999999997,
+      "loss": 4.162,
       "step": 299
     },
     {
-      "epoch": 0.67,
-      "learning_rate": 0.00017819999999999997,
-      "loss": 4.4759,
+      "epoch": 1.35,
+      "learning_rate": 0.00017879999999999998,
+      "loss": 4.1414,
       "step": 300
     },
     {
-      "epoch": 0.67,
-      "learning_rate": 0.00017879999999999998,
-      "loss": 5.7464,
+      "epoch": 1.35,
+      "learning_rate": 0.00017939999999999997,
+      "loss": 4.1157,
       "step": 301
     },
     {
-      "epoch": 0.68,
-      "learning_rate": 0.00017939999999999997,
-      "loss": 5.5533,
+      "epoch": 1.35,
+      "learning_rate": 0.00017999999999999998,
+      "loss": 4.0798,
       "step": 302
     },
     {
-      "epoch": 0.68,
-      "learning_rate": 0.00017999999999999998,
-      "loss": 5.2607,
+      "epoch": 1.36,
+      "learning_rate": 0.00018059999999999997,
+      "loss": 4.1804,
       "step": 303
     },
     {
-      "epoch": 0.68,
-      "learning_rate": 0.00018059999999999997,
-      "loss": 5.0842,
+      "epoch": 1.36,
+      "learning_rate": 0.00018119999999999999,
+      "loss": 4.0958,
       "step": 304
     },
     {
-      "epoch": 0.68,
-      "learning_rate": 0.00018119999999999999,
-      "loss": 4.9463,
+      "epoch": 1.37,
+      "learning_rate": 0.00018179999999999997,
+      "loss": 4.0506,
       "step": 305
     },
     {
-      "epoch": 0.69,
-      "learning_rate": 0.00018179999999999997,
-      "loss": 4.7063,
+      "epoch": 1.37,
+      "learning_rate": 0.0001824,
+      "loss": 4.1091,
       "step": 306
     },
     {
-      "epoch": 0.69,
-      "learning_rate": 0.0001824,
-      "loss": 4.6942,
+      "epoch": 1.38,
+      "learning_rate": 0.00018299999999999998,
+      "loss": 4.1218,
       "step": 307
     },
     {
-      "epoch": 0.69,
-      "learning_rate": 0.00018299999999999998,
-      "loss": 4.6681,
+      "epoch": 1.38,
+      "learning_rate": 0.0001836,
+      "loss": 4.1636,
       "step": 308
     },
     {
-      "epoch": 0.69,
-      "learning_rate": 0.0001836,
-      "loss": 4.6439,
+      "epoch": 1.39,
+      "learning_rate": 0.00018419999999999998,
+      "loss": 4.08,
       "step": 309
     },
     {
-      "epoch": 0.7,
-      "learning_rate": 0.00018419999999999998,
-      "loss": 4.7031,
+      "epoch": 1.39,
+      "learning_rate": 0.0001848,
+      "loss": 4.0701,
       "step": 310
     },
     {
-      "epoch": 0.7,
-      "learning_rate": 0.0001848,
-      "loss": 4.4416,
+      "epoch": 1.39,
+      "learning_rate": 0.00018539999999999998,
+      "loss": 3.9545,
       "step": 311
     },
     {
-      "epoch": 0.7,
-      "learning_rate": 0.00018539999999999998,
-      "loss": 4.5832,
+      "epoch": 1.4,
+      "learning_rate": 0.000186,
+      "loss": 3.8738,
       "step": 312
     },
     {
-      "epoch": 0.7,
-      "learning_rate": 0.000186,
-      "loss": 4.5866,
+      "epoch": 1.4,
+      "learning_rate": 0.00018659999999999998,
+      "loss": 4.0669,
       "step": 313
     },
     {
-      "epoch": 0.7,
-      "learning_rate": 0.00018659999999999998,
-      "loss": 4.4809,
+      "epoch": 1.41,
+      "learning_rate": 0.0001872,
+      "loss": 4.0076,
       "step": 314
     },
     {
-      "epoch": 0.71,
-      "learning_rate": 0.0001872,
-      "loss": 4.5242,
+      "epoch": 1.41,
+      "learning_rate": 0.00018779999999999998,
+      "loss": 4.0221,
       "step": 315
     },
     {
-      "epoch": 0.71,
-      "learning_rate": 0.00018779999999999998,
-      "loss": 4.4804,
+      "epoch": 1.42,
+      "learning_rate": 0.00018839999999999997,
+      "loss": 3.9464,
       "step": 316
     },
     {
-      "epoch": 0.71,
-      "learning_rate": 0.00018839999999999997,
-      "loss": 4.4799,
+      "epoch": 1.42,
+      "learning_rate": 0.00018899999999999999,
+      "loss": 3.771,
       "step": 317
     },
     {
-      "epoch": 0.71,
-      "learning_rate": 0.00018899999999999999,
-      "loss": 4.4712,
+      "epoch": 1.43,
+      "learning_rate": 0.00018959999999999997,
+      "loss": 3.8719,
       "step": 318
     },
     {
-      "epoch": 0.72,
-      "learning_rate": 0.00018959999999999997,
-      "loss": 4.4968,
+      "epoch": 1.43,
+      "learning_rate": 0.0001902,
+      "loss": 3.9347,
       "step": 319
     },
     {
-      "epoch": 0.72,
-      "learning_rate": 0.0001902,
-      "loss": 4.5755,
+      "epoch": 1.43,
+      "learning_rate": 0.00019079999999999998,
+      "loss": 3.8062,
       "step": 320
     },
     {
-      "epoch": 0.72,
-      "learning_rate": 0.00019079999999999998,
-      "loss": 4.4656,
+      "epoch": 1.44,
+      "learning_rate": 0.0001914,
+      "loss": 3.7106,
       "step": 321
     },
     {
-      "epoch": 0.72,
-      "learning_rate": 0.0001914,
-      "loss": 4.3633,
+      "epoch": 1.44,
+      "learning_rate": 0.00019199999999999998,
+      "loss": 3.8163,
       "step": 322
     },
     {
-      "epoch": 0.72,
-      "learning_rate": 0.00019199999999999998,
-      "loss": 4.458,
+      "epoch": 1.45,
+      "learning_rate": 0.0001926,
+      "loss": 3.6449,
       "step": 323
     },
     {
-      "epoch": 0.73,
-      "learning_rate": 0.0001926,
-      "loss": 4.3913,
+      "epoch": 1.45,
+      "learning_rate": 0.00019319999999999998,
+      "loss": 5.7573,
       "step": 324
     },
     {
-      "epoch": 0.73,
-      "learning_rate": 0.00019319999999999998,
-      "loss": 4.5151,
+      "epoch": 1.46,
+      "learning_rate": 0.0001938,
+      "loss": 5.6261,
       "step": 325
     },
     {
-      "epoch": 0.73,
-      "learning_rate": 0.0001938,
-      "loss": 4.4925,
+      "epoch": 1.46,
+      "learning_rate": 0.00019439999999999998,
+      "loss": 5.3271,
       "step": 326
     },
     {
-      "epoch": 0.73,
-      "learning_rate": 0.00019439999999999998,
-      "loss": 4.4414,
+      "epoch": 1.47,
+      "learning_rate": 0.000195,
+      "loss": 4.9718,
       "step": 327
     },
     {
-      "epoch": 0.74,
-      "learning_rate": 0.000195,
-      "loss": 4.3735,
+      "epoch": 1.47,
+      "learning_rate": 0.00019559999999999998,
+      "loss": 4.8634,
       "step": 328
     },
     {
-      "epoch": 0.74,
+      "epoch": 1.48,
       "learning_rate": 0.00019559999999999998,
-      "loss": 4.4079,
+      "loss": 5.2309,
       "step": 329
     },
     {
-      "epoch": 0.74,
+      "epoch": 1.48,
       "learning_rate": 0.0001962,
-      "loss": 4.5079,
+      "loss": 5.1864,
       "step": 330
     },
     {
-      "epoch": 0.74,
+      "epoch": 1.48,
       "learning_rate": 0.00019679999999999999,
-      "loss": 4.515,
+      "loss": 7.7546,
       "step": 331
     },
     {
-      "epoch": 0.74,
+      "epoch": 1.49,
       "learning_rate": 0.0001974,
-      "loss": 4.3635,
+      "loss": 7.6626,
       "step": 332
     },
     {
-      "epoch": 0.75,
+      "epoch": 1.49,
       "learning_rate": 0.000198,
-      "loss": 4.5648,
+      "loss": 7.7001,
       "step": 333
     },
     {
-      "epoch": 0.75,
+      "epoch": 1.5,
       "learning_rate": 0.0001986,
-      "loss": 4.4682,
+      "loss": 7.4637,
       "step": 334
     },
     {
-      "epoch": 0.75,
+      "epoch": 1.5,
       "learning_rate": 0.0001992,
-      "loss": 4.5643,
+      "loss": 7.272,
       "step": 335
     },
     {
-      "epoch": 0.75,
+      "epoch": 1.51,
       "learning_rate": 0.0001998,
-      "loss": 4.4503,
+      "loss": 7.0608,
       "step": 336
     },
     {
-      "epoch": 0.76,
+      "epoch": 1.51,
       "learning_rate": 0.0002004,
-      "loss": 4.4666,
+      "loss": 6.811,
       "step": 337
     },
     {
-      "epoch": 0.76,
+      "epoch": 1.52,
       "learning_rate": 0.000201,
-      "loss": 4.4885,
+      "loss": 6.8756,
       "step": 338
     },
     {
-      "epoch": 0.76,
+      "epoch": 1.52,
       "learning_rate": 0.0002016,
-      "loss": 4.5766,
+      "loss": 6.0821,
       "step": 339
     },
     {
-      "epoch": 0.76,
+      "epoch": 1.52,
       "learning_rate": 0.0002022,
-      "loss": 4.6036,
+      "loss": 5.2635,
       "step": 340
     },
     {
-      "epoch": 0.76,
+      "epoch": 1.53,
       "learning_rate": 0.0002028,
-      "loss": 4.4403,
+      "loss": 5.1346,
       "step": 341
     },
     {
-      "epoch": 0.77,
+      "epoch": 1.53,
       "learning_rate": 0.00020339999999999998,
-      "loss": 4.5286,
+      "loss": 4.9469,
       "step": 342
     },
     {
-      "epoch": 0.77,
+      "epoch": 1.54,
       "learning_rate": 0.000204,
-      "loss": 4.4455,
+      "loss": 4.7854,
       "step": 343
     },
     {
-      "epoch": 0.77,
+      "epoch": 1.54,
       "learning_rate": 0.00020459999999999999,
-      "loss": 4.5612,
+      "loss": 4.4968,
       "step": 344
     },
     {
-      "epoch": 0.77,
+      "epoch": 1.55,
       "learning_rate": 0.0002052,
-      "loss": 4.2895,
+      "loss": 4.5434,
       "step": 345
     },
     {
-      "epoch": 0.78,
+      "epoch": 1.55,
       "learning_rate": 0.0002058,
-      "loss": 4.4651,
+      "loss": 4.4593,
       "step": 346
     },
     {
-      "epoch": 0.78,
+      "epoch": 1.56,
       "learning_rate": 0.00020639999999999998,
-      "loss": 4.3408,
+      "loss": 4.4982,
       "step": 347
     },
     {
-      "epoch": 0.78,
+      "epoch": 1.56,
       "learning_rate": 0.00020699999999999996,
-      "loss": 4.1872,
+      "loss": 4.5314,
       "step": 348
     },
     {
-      "epoch": 0.78,
+      "epoch": 1.57,
       "learning_rate": 0.00020759999999999998,
-      "loss": 4.3745,
+      "loss": 4.4604,
       "step": 349
     },
     {
-      "epoch": 0.78,
+      "epoch": 1.57,
       "learning_rate": 0.00020819999999999996,
-      "loss": 4.0009,
+      "loss": 4.3669,
       "step": 350
     },
     {
-      "epoch": 0.79,
+      "epoch": 1.57,
       "learning_rate": 0.00020879999999999998,
-      "loss": 5.993,
+      "loss": 4.2684,
       "step": 351
     },
     {
-      "epoch": 0.79,
+      "epoch": 1.58,
       "learning_rate": 0.00020939999999999997,
-      "loss": 5.8259,
+      "loss": 4.3108,
       "step": 352
     },
     {
-      "epoch": 0.79,
+      "epoch": 1.58,
       "learning_rate": 0.00020999999999999998,
-      "loss": 5.3393,
+      "loss": 4.3765,
       "step": 353
     },
     {
-      "epoch": 0.79,
+      "epoch": 1.59,
       "learning_rate": 0.00021059999999999997,
-      "loss": 4.9743,
+      "loss": 4.5628,
       "step": 354
     },
     {
-      "epoch": 0.8,
+      "epoch": 1.59,
       "learning_rate": 0.00021119999999999996,
-      "loss": 4.9793,
+      "loss": 4.3952,
       "step": 355
     },
     {
-      "epoch": 0.8,
+      "epoch": 1.6,
       "learning_rate": 0.00021179999999999997,
-      "loss": 4.9003,
+      "loss": 4.2404,
       "step": 356
     },
     {
-      "epoch": 0.8,
+      "epoch": 1.6,
       "learning_rate": 0.00021239999999999996,
-      "loss": 4.7375,
+      "loss": 4.1993,
       "step": 357
     },
     {
-      "epoch": 0.8,
+      "epoch": 1.61,
       "learning_rate": 0.00021299999999999997,
-      "loss": 4.7576,
+      "loss": 4.24,
       "step": 358
     },
     {
-      "epoch": 0.8,
+      "epoch": 1.61,
       "learning_rate": 0.00021359999999999996,
-      "loss": 4.6035,
+      "loss": 4.1699,
       "step": 359
     },
     {
-      "epoch": 0.81,
+      "epoch": 1.61,
       "learning_rate": 0.00021419999999999998,
-      "loss": 4.5928,
+      "loss": 4.2247,
       "step": 360
     },
     {
-      "epoch": 0.81,
+      "epoch": 1.62,
       "learning_rate": 0.00021479999999999996,
-      "loss": 4.5001,
+      "loss": 4.2105,
       "step": 361
     },
     {
-      "epoch": 0.81,
+      "epoch": 1.62,
       "learning_rate": 0.00021539999999999998,
-      "loss": 4.5366,
+      "loss": 4.0637,
       "step": 362
     },
     {
-      "epoch": 0.81,
+      "epoch": 1.63,
       "learning_rate": 0.00021599999999999996,
-      "loss": 4.4527,
+      "loss": 4.1486,
       "step": 363
     },
     {
-      "epoch": 0.82,
+      "epoch": 1.63,
       "learning_rate": 0.00021659999999999998,
-      "loss": 4.5041,
+      "loss": 4.0632,
       "step": 364
     },
     {
-      "epoch": 0.82,
+      "epoch": 1.64,
       "learning_rate": 0.00021719999999999997,
-      "loss": 4.6164,
+      "loss": 4.0394,
       "step": 365
     },
     {
-      "epoch": 0.82,
+      "epoch": 1.64,
       "learning_rate": 0.00021779999999999998,
-      "loss": 4.4303,
+      "loss": 4.1295,
       "step": 366
     },
     {
-      "epoch": 0.82,
+      "epoch": 1.65,
       "learning_rate": 0.00021839999999999997,
-      "loss": 4.5242,
+      "loss": 4.1444,
       "step": 367
     },
     {
-      "epoch": 0.83,
+      "epoch": 1.65,
       "learning_rate": 0.00021899999999999998,
-      "loss": 4.5089,
+      "loss": 4.2003,
       "step": 368
     },
     {
-      "epoch": 0.83,
+      "epoch": 1.65,
       "learning_rate": 0.00021959999999999997,
-      "loss": 4.4234,
+      "loss": 4.0887,
       "step": 369
     },
     {
-      "epoch": 0.83,
+      "epoch": 1.66,
       "learning_rate": 0.00022019999999999999,
-      "loss": 4.4205,
+      "loss": 3.9279,
       "step": 370
     },
     {
-      "epoch": 0.83,
+      "epoch": 1.66,
       "learning_rate": 0.00022079999999999997,
-      "loss": 4.3864,
+      "loss": 3.9932,
       "step": 371
     },
     {
-      "epoch": 0.83,
+      "epoch": 1.67,
       "learning_rate": 0.0002214,
-      "loss": 4.4607,
+      "loss": 3.7584,
       "step": 372
     },
     {
-      "epoch": 0.84,
-      "learning_rate": 0.00022199999999999998,
-      "loss": 4.4477,
+      "epoch": 1.67,
+      "learning_rate": 0.0002214,
+      "loss": 3.4602,
       "step": 373
     },
     {
-      "epoch": 0.84,
-      "learning_rate": 0.0002226,
-      "loss": 4.2569,
+      "epoch": 1.68,
+      "learning_rate": 0.00022199999999999998,
+      "loss": 5.1713,
       "step": 374
     },
     {
-      "epoch": 0.84,
-      "learning_rate": 0.00022319999999999998,
-      "loss": 4.4793,
+      "epoch": 1.68,
+      "learning_rate": 0.0002226,
+      "loss": 4.9749,
       "step": 375
     },
     {
-      "epoch": 0.84,
-      "learning_rate": 0.0002238,
-      "loss": 4.3588,
+      "epoch": 1.69,
+      "learning_rate": 0.00022319999999999998,
+      "loss": 4.77,
       "step": 376
     },
     {
-      "epoch": 0.85,
-      "learning_rate": 0.00022439999999999998,
-      "loss": 4.4029,
+      "epoch": 1.69,
+      "learning_rate": 0.0002238,
+      "loss": 4.525,
       "step": 377
     },
     {
-      "epoch": 0.85,
-      "learning_rate": 0.000225,
-      "loss": 4.3932,
+      "epoch": 1.7,
+      "learning_rate": 0.00022439999999999998,
+      "loss": 4.4688,
       "step": 378
     },
     {
-      "epoch": 0.85,
-      "learning_rate": 0.00022559999999999998,
-      "loss": 4.2944,
+      "epoch": 1.7,
+      "learning_rate": 0.000225,
+      "loss": 4.3824,
       "step": 379
     },
     {
-      "epoch": 0.85,
-      "learning_rate": 0.00022619999999999997,
-      "loss": 4.3351,
+      "epoch": 1.7,
+      "learning_rate": 0.00022559999999999998,
+      "loss": 4.4258,
       "step": 380
     },
     {
-      "epoch": 0.85,
-      "learning_rate": 0.00022679999999999998,
-      "loss": 4.4359,
+      "epoch": 1.71,
+      "learning_rate": 0.00022619999999999997,
+      "loss": 4.4924,
       "step": 381
     },
     {
-      "epoch": 0.86,
-      "learning_rate": 0.00022739999999999997,
-      "loss": 4.4558,
+      "epoch": 1.71,
+      "learning_rate": 0.00022679999999999998,
+      "loss": 4.3423,
       "step": 382
     },
     {
-      "epoch": 0.86,
-      "learning_rate": 0.00022799999999999999,
-      "loss": 4.3975,
+      "epoch": 1.72,
+      "learning_rate": 0.00022739999999999997,
+      "loss": 4.439,
       "step": 383
     },
     {
-      "epoch": 0.86,
-      "learning_rate": 0.00022859999999999997,
-      "loss": 4.308,
+      "epoch": 1.72,
+      "learning_rate": 0.00022799999999999999,
+      "loss": 4.2055,
       "step": 384
     },
     {
-      "epoch": 0.86,
-      "learning_rate": 0.0002292,
-      "loss": 4.4021,
+      "epoch": 1.73,
+      "learning_rate": 0.00022859999999999997,
+      "loss": 4.1395,
       "step": 385
     },
     {
-      "epoch": 0.87,
-      "learning_rate": 0.00022979999999999997,
-      "loss": 4.2904,
+      "epoch": 1.73,
+      "learning_rate": 0.0002292,
+      "loss": 4.1919,
       "step": 386
     },
     {
-      "epoch": 0.87,
-      "learning_rate": 0.0002304,
-      "loss": 4.4339,
+      "epoch": 1.74,
+      "learning_rate": 0.00022979999999999997,
+      "loss": 4.1744,
       "step": 387
     },
     {
-      "epoch": 0.87,
-      "learning_rate": 0.00023099999999999998,
-      "loss": 4.4791,
+      "epoch": 1.74,
+      "learning_rate": 0.0002304,
+      "loss": 4.1852,
       "step": 388
     },
     {
-      "epoch": 0.87,
-      "learning_rate": 0.0002316,
-      "loss": 4.3777,
+      "epoch": 1.74,
+      "learning_rate": 0.00023099999999999998,
+      "loss": 4.1599,
       "step": 389
     },
     {
-      "epoch": 0.87,
-      "learning_rate": 0.00023219999999999998,
-      "loss": 4.3603,
+      "epoch": 1.75,
+      "learning_rate": 0.0002316,
+      "loss": 4.1255,
       "step": 390
     },
     {
-      "epoch": 0.88,
-      "learning_rate": 0.0002328,
-      "loss": 4.478,
+      "epoch": 1.75,
+      "learning_rate": 0.00023219999999999998,
+      "loss": 4.1317,
       "step": 391
     },
     {
-      "epoch": 0.88,
-      "learning_rate": 0.00023339999999999998,
-      "loss": 4.2619,
+      "epoch": 1.76,
+      "learning_rate": 0.0002328,
+      "loss": 4.1679,
       "step": 392
     },
     {
-      "epoch": 0.88,
-      "learning_rate": 0.000234,
-      "loss": 4.2223,
+      "epoch": 1.76,
+      "learning_rate": 0.00023339999999999998,
+      "loss": 4.1555,
       "step": 393
     },
     {
-      "epoch": 0.88,
-      "learning_rate": 0.00023459999999999998,
-      "loss": 4.2092,
+      "epoch": 1.77,
+      "learning_rate": 0.000234,
+      "loss": 4.1475,
       "step": 394
     },
     {
-      "epoch": 0.89,
-      "learning_rate": 0.0002352,
-      "loss": 4.3659,
+      "epoch": 1.77,
+      "learning_rate": 0.00023459999999999998,
+      "loss": 4.1145,
       "step": 395
     },
     {
-      "epoch": 0.89,
-      "learning_rate": 0.00023579999999999999,
-      "loss": 4.2694,
+      "epoch": 1.78,
+      "learning_rate": 0.0002352,
+      "loss": 4.0143,
       "step": 396
     },
     {
-      "epoch": 0.89,
-      "learning_rate": 0.0002364,
-      "loss": 4.3929,
+      "epoch": 1.78,
+      "learning_rate": 0.00023579999999999999,
+      "loss": 4.1586,
       "step": 397
     },
     {
-      "epoch": 0.89,
-      "learning_rate": 0.000237,
-      "loss": 4.1923,
+      "epoch": 1.78,
+      "learning_rate": 0.0002364,
+      "loss": 4.0736,
       "step": 398
     },
     {
-      "epoch": 0.89,
-      "learning_rate": 0.0002376,
-      "loss": 4.1854,
+      "epoch": 1.79,
+      "learning_rate": 0.000237,
+      "loss": 4.1268,
       "step": 399
     },
     {
-      "epoch": 0.9,
-      "learning_rate": 0.0002382,
-      "loss": 4.0039,
+      "epoch": 1.79,
+      "learning_rate": 0.0002376,
+      "loss": 4.1166,
       "step": 400
     },
     {
-      "epoch": 0.9,
-      "learning_rate": 0.0002388,
-      "loss": 5.8865,
+      "epoch": 1.8,
+      "learning_rate": 0.0002382,
+      "loss": 4.0384,
       "step": 401
     },
     {
-      "epoch": 0.9,
-      "learning_rate": 0.0002394,
-      "loss": 5.6627,
+      "epoch": 1.8,
+      "learning_rate": 0.0002388,
+      "loss": 3.9957,
       "step": 402
     },
     {
-      "epoch": 0.9,
-      "learning_rate": 0.00023999999999999998,
-      "loss": 5.4797,
+      "epoch": 1.81,
+      "learning_rate": 0.0002394,
+      "loss": 4.0793,
       "step": 403
     },
     {
-      "epoch": 0.91,
-      "learning_rate": 0.0002406,
-      "loss": 5.0418,
+      "epoch": 1.81,
+      "learning_rate": 0.00023999999999999998,
+      "loss": 4.1288,
       "step": 404
     },
     {
-      "epoch": 0.91,
-      "learning_rate": 0.00024119999999999998,
-      "loss": 4.8755,
+      "epoch": 1.82,
+      "learning_rate": 0.0002406,
+      "loss": 3.9214,
       "step": 405
     },
     {
-      "epoch": 0.91,
-      "learning_rate": 0.0002418,
-      "loss": 4.7933,
+      "epoch": 1.82,
+      "learning_rate": 0.00024119999999999998,
+      "loss": 4.0041,
       "step": 406
     },
     {
-      "epoch": 0.91,
-      "learning_rate": 0.00024239999999999998,
-      "loss": 4.7055,
+      "epoch": 1.83,
+      "learning_rate": 0.0002418,
+      "loss": 4.0388,
       "step": 407
     },
     {
-      "epoch": 0.91,
-      "learning_rate": 0.000243,
-      "loss": 4.6617,
+      "epoch": 1.83,
+      "learning_rate": 0.00024239999999999998,
+      "loss": 3.989,
       "step": 408
     },
     {
-      "epoch": 0.92,
-      "learning_rate": 0.00024359999999999999,
-      "loss": 4.5539,
+      "epoch": 1.83,
+      "learning_rate": 0.000243,
+      "loss": 4.0942,
       "step": 409
     },
     {
-      "epoch": 0.92,
-      "learning_rate": 0.00024419999999999997,
-      "loss": 4.492,
+      "epoch": 1.84,
+      "learning_rate": 0.00024359999999999999,
+      "loss": 3.9554,
       "step": 410
     },
     {
-      "epoch": 0.92,
-      "learning_rate": 0.0002448,
-      "loss": 4.6706,
+      "epoch": 1.84,
+      "learning_rate": 0.00024419999999999997,
+      "loss": 4.0312,
       "step": 411
     },
     {
-      "epoch": 0.92,
-      "learning_rate": 0.00024539999999999995,
-      "loss": 4.4853,
+      "epoch": 1.85,
+      "learning_rate": 0.0002448,
+      "loss": 3.9571,
       "step": 412
     },
     {
-      "epoch": 0.93,
-      "learning_rate": 0.00024599999999999996,
-      "loss": 4.4369,
+      "epoch": 1.85,
+      "learning_rate": 0.00024539999999999995,
+      "loss": 4.0576,
       "step": 413
     },
     {
-      "epoch": 0.93,
-      "learning_rate": 0.0002466,
-      "loss": 4.4848,
+      "epoch": 1.86,
+      "learning_rate": 0.00024599999999999996,
+      "loss": 3.9382,
       "step": 414
     },
     {
-      "epoch": 0.93,
-      "learning_rate": 0.0002472,
-      "loss": 4.5455,
+      "epoch": 1.86,
+      "learning_rate": 0.0002466,
+      "loss": 3.8844,
       "step": 415
     },
     {
-      "epoch": 0.93,
-      "learning_rate": 0.00024779999999999995,
-      "loss": 4.4175,
+      "epoch": 1.87,
+      "learning_rate": 0.0002472,
+      "loss": 3.8587,
       "step": 416
     },
     {
-      "epoch": 0.93,
-      "learning_rate": 0.00024839999999999997,
-      "loss": 4.4364,
+      "epoch": 1.87,
+      "learning_rate": 0.00024779999999999995,
+      "loss": 3.8616,
       "step": 417
     },
     {
-      "epoch": 0.94,
-      "learning_rate": 0.000249,
-      "loss": 4.3047,
+      "epoch": 1.87,
+      "learning_rate": 0.00024839999999999997,
+      "loss": 3.8594,
       "step": 418
     },
     {
-      "epoch": 0.94,
-      "learning_rate": 0.00024959999999999994,
-      "loss": 4.4146,
+      "epoch": 1.88,
+      "learning_rate": 0.000249,
+      "loss": 3.698,
       "step": 419
     },
     {
-      "epoch": 0.94,
-      "learning_rate": 0.00025019999999999996,
-      "loss": 4.3201,
+      "epoch": 1.88,
+      "learning_rate": 0.00024959999999999994,
+      "loss": 3.7176,
       "step": 420
     },
     {
-      "epoch": 0.94,
-      "learning_rate": 0.00025079999999999997,
-      "loss": 4.4562,
+      "epoch": 1.89,
+      "learning_rate": 0.00025019999999999996,
+      "loss": 3.5745,
       "step": 421
     },
     {
-      "epoch": 0.95,
-      "learning_rate": 0.0002514,
-      "loss": 4.3312,
+      "epoch": 1.89,
+      "learning_rate": 0.00025079999999999997,
+      "loss": 3.5771,
       "step": 422
     },
     {
-      "epoch": 0.95,
-      "learning_rate": 0.00025199999999999995,
-      "loss": 4.4033,
+      "epoch": 1.9,
+      "learning_rate": 0.0002514,
+      "loss": 3.1711,
       "step": 423
     },
     {
-      "epoch": 0.95,
-      "learning_rate": 0.00025259999999999996,
-      "loss": 4.4837,
+      "epoch": 1.9,
+      "learning_rate": 0.00025199999999999995,
+      "loss": 5.6223,
       "step": 424
     },
     {
-      "epoch": 0.95,
-      "learning_rate": 0.0002532,
-      "loss": 4.3916,
+      "epoch": 1.91,
+      "learning_rate": 0.00025259999999999996,
+      "loss": 5.5827,
       "step": 425
     },
     {
-      "epoch": 0.96,
-      "learning_rate": 0.0002538,
-      "loss": 4.405,
+      "epoch": 1.91,
+      "learning_rate": 0.0002532,
+      "loss": 5.1421,
       "step": 426
     },
     {
-      "epoch": 0.96,
-      "learning_rate": 0.00025439999999999995,
-      "loss": 4.3168,
+      "epoch": 1.91,
+      "learning_rate": 0.0002538,
+      "loss": 4.695,
       "step": 427
     },
     {
-      "epoch": 0.96,
-      "learning_rate": 0.00025499999999999996,
-      "loss": 4.3303,
+      "epoch": 1.92,
+      "learning_rate": 0.00025439999999999995,
+      "loss": 4.5779,
       "step": 428
     },
     {
-      "epoch": 0.96,
-      "learning_rate": 0.0002556,
-      "loss": 4.2059,
+      "epoch": 1.92,
+      "learning_rate": 0.00025499999999999996,
+      "loss": 4.5219,
       "step": 429
     },
     {
-      "epoch": 0.96,
-      "learning_rate": 0.0002562,
-      "loss": 4.3448,
+      "epoch": 1.93,
+      "learning_rate": 0.0002556,
+      "loss": 4.2696,
       "step": 430
     },
     {
-      "epoch": 0.97,
-      "learning_rate": 0.00025679999999999995,
-      "loss": 4.3357,
+      "epoch": 1.93,
+      "learning_rate": 0.0002562,
+      "loss": 4.1901,
       "step": 431
     },
     {
-      "epoch": 0.97,
-      "learning_rate": 0.00025739999999999997,
-      "loss": 4.5314,
+      "epoch": 1.94,
+      "learning_rate": 0.00025679999999999995,
+      "loss": 4.1369,
       "step": 432
     },
     {
-      "epoch": 0.97,
-      "learning_rate": 0.000258,
-      "loss": 4.3478,
+      "epoch": 1.94,
+      "learning_rate": 0.00025739999999999997,
+      "loss": 4.0599,
       "step": 433
     },
     {
-      "epoch": 0.97,
-      "learning_rate": 0.0002586,
-      "loss": 4.2702,
+      "epoch": 1.95,
+      "learning_rate": 0.000258,
+      "loss": 4.1342,
       "step": 434
     },
     {
-      "epoch": 0.98,
-      "learning_rate": 0.00025919999999999996,
-      "loss": 4.3382,
+      "epoch": 1.95,
+      "learning_rate": 0.0002586,
+      "loss": 4.1208,
       "step": 435
     },
     {
-      "epoch": 0.98,
-      "learning_rate": 0.00025979999999999997,
-      "loss": 4.2302,
+      "epoch": 1.96,
+      "learning_rate": 0.00025919999999999996,
+      "loss": 4.0045,
       "step": 436
     },
     {
-      "epoch": 0.98,
-      "learning_rate": 0.0002604,
-      "loss": 4.2964,
+      "epoch": 1.96,
+      "learning_rate": 0.00025979999999999997,
+      "loss": 4.0492,
       "step": 437
     },
     {
-      "epoch": 0.98,
-      "learning_rate": 0.000261,
-      "loss": 4.3367,
+      "epoch": 1.96,
+      "learning_rate": 0.0002604,
+      "loss": 4.0202,
       "step": 438
     },
     {
-      "epoch": 0.98,
-      "learning_rate": 0.00026159999999999996,
-      "loss": 4.3053,
+      "epoch": 1.97,
+      "learning_rate": 0.000261,
+      "loss": 3.9223,
       "step": 439
     },
     {
-      "epoch": 0.99,
-      "learning_rate": 0.0002622,
-      "loss": 4.3341,
+      "epoch": 1.97,
+      "learning_rate": 0.00026159999999999996,
+      "loss": 3.9732,
       "step": 440
     },
     {
-      "epoch": 0.99,
-      "learning_rate": 0.0002628,
-      "loss": 4.2008,
+      "epoch": 1.98,
+      "learning_rate": 0.0002622,
+      "loss": 3.9046,
       "step": 441
     },
     {
-      "epoch": 0.99,
-      "learning_rate": 0.00026339999999999995,
-      "loss": 4.3948,
+      "epoch": 1.98,
+      "learning_rate": 0.0002628,
+      "loss": 3.9394,
       "step": 442
     },
     {
-      "epoch": 0.99,
-      "learning_rate": 0.00026399999999999997,
-      "loss": 4.0869,
+      "epoch": 1.99,
+      "learning_rate": 0.00026339999999999995,
+      "loss": 3.8511,
       "step": 443
     },
     {
-      "epoch": 1.0,
-      "learning_rate": 0.0002646,
-      "loss": 3.9334,
+      "epoch": 1.99,
+      "learning_rate": 0.00026399999999999997,
+      "loss": 3.7997,
       "step": 444
     },
     {
-      "epoch": 1.0,
-      "learning_rate": 0.0002652,
-      "loss": 4.0116,
+      "epoch": 2.0,
+      "learning_rate": 0.0002646,
+      "loss": 3.6102,
       "step": 445
     },
     {
-      "epoch": 1.0,
-      "learning_rate": 0.00026579999999999996,
-      "loss": 3.6733,
+      "epoch": 2.0,
+      "learning_rate": 0.0002652,
+      "loss": 3.2133,
       "step": 446
     },
     {
-      "epoch": 1.0,
-      "learning_rate": 0.00026639999999999997,
-      "loss": 5.8448,
+      "epoch": 2.0,
+      "learning_rate": 0.00026579999999999996,
+      "loss": 5.3176,
       "step": 447
     },
     {
-      "epoch": 1.0,
-      "learning_rate": 0.000267,
-      "loss": 5.7358,
+      "epoch": 2.01,
+      "learning_rate": 0.00026639999999999997,
+      "loss": 5.2469,
       "step": 448
     },
     {
-      "epoch": 1.01,
-      "learning_rate": 0.0002676,
-      "loss": 5.3335,
+      "epoch": 2.01,
+      "learning_rate": 0.000267,
+      "loss": 4.9693,
       "step": 449
     },
     {
-      "epoch": 1.01,
-      "learning_rate": 0.00026819999999999996,
-      "loss": 4.9919,
-      "step": 450
+      "epoch": 2.02,
+      "learning_rate": 0.0002676,
+      "loss": 4.5121,
+      "step": 450
     },
     {
-      "epoch": 1.01,
-      "learning_rate": 0.0002688,
-      "loss": 4.7456,
+      "epoch": 2.02,
+      "learning_rate": 0.00026819999999999996,
+      "loss": 4.5022,
       "step": 451
     },
     {
-      "epoch": 1.01,
-      "learning_rate": 0.0002694,
-      "loss": 4.6112,
+      "epoch": 2.03,
+      "learning_rate": 0.0002688,
+      "loss": 4.2283,
       "step": 452
     },
     {
-      "epoch": 1.02,
-      "learning_rate": 0.00027,
-      "loss": 4.4133,
+      "epoch": 2.03,
+      "learning_rate": 0.0002694,
+      "loss": 4.0816,
       "step": 453
     },
     {
-      "epoch": 1.02,
-      "learning_rate": 0.00027059999999999996,
-      "loss": 4.3851,
+      "epoch": 2.04,
+      "learning_rate": 0.00027,
+      "loss": 4.091,
       "step": 454
     },
     {
-      "epoch": 1.02,
-      "learning_rate": 0.0002712,
-      "loss": 4.2591,
+      "epoch": 2.04,
+      "learning_rate": 0.00027059999999999996,
+      "loss": 4.0024,
       "step": 455
     },
     {
-      "epoch": 1.02,
-      "learning_rate": 0.0002718,
-      "loss": 4.3699,
+      "epoch": 2.04,
+      "learning_rate": 0.0002712,
+      "loss": 3.9794,
       "step": 456
     },
     {
-      "epoch": 1.02,
-      "learning_rate": 0.0002724,
-      "loss": 4.1498,
+      "epoch": 2.05,
+      "learning_rate": 0.0002718,
+      "loss": 3.8077,
       "step": 457
     },
     {
-      "epoch": 1.03,
-      "learning_rate": 0.00027299999999999997,
-      "loss": 4.1313,
+      "epoch": 2.05,
+      "learning_rate": 0.0002724,
+      "loss": 3.86,
       "step": 458
     },
     {
-      "epoch": 1.03,
-      "learning_rate": 0.0002736,
-      "loss": 4.2658,
+      "epoch": 2.06,
+      "learning_rate": 0.00027299999999999997,
+      "loss": 3.6727,
       "step": 459
     },
     {
-      "epoch": 1.03,
-      "learning_rate": 0.0002742,
-      "loss": 4.1512,
+      "epoch": 2.06,
+      "learning_rate": 0.0002736,
+      "loss": 3.7923,
       "step": 460
     },
     {
-      "epoch": 1.03,
-      "learning_rate": 0.0002748,
-      "loss": 4.1147,
+      "epoch": 2.07,
+      "learning_rate": 0.0002742,
+      "loss": 3.7753,
       "step": 461
     },
     {
-      "epoch": 1.04,
-      "learning_rate": 0.00027539999999999997,
-      "loss": 4.0338,
+      "epoch": 2.07,
+      "learning_rate": 0.0002748,
+      "loss": 3.7465,
       "step": 462
     },
     {
-      "epoch": 1.04,
-      "learning_rate": 0.000276,
-      "loss": 4.0853,
+      "epoch": 2.08,
+      "learning_rate": 0.00027539999999999997,
+      "loss": 3.712,
       "step": 463
     },
     {
-      "epoch": 1.04,
-      "learning_rate": 0.0002766,
-      "loss": 4.0146,
+      "epoch": 2.08,
+      "learning_rate": 0.000276,
+      "loss": 3.5746,
       "step": 464
     },
     {
-      "epoch": 1.04,
-      "learning_rate": 0.0002772,
-      "loss": 3.9486,
+      "epoch": 2.09,
+      "learning_rate": 0.0002766,
+      "loss": 3.6256,
       "step": 465
     },
     {
-      "epoch": 1.04,
-      "learning_rate": 0.0002778,
-      "loss": 3.9637,
+      "epoch": 2.09,
+      "learning_rate": 0.0002772,
+      "loss": 3.5238,
       "step": 466
     },
     {
-      "epoch": 1.05,
-      "learning_rate": 0.0002784,
-      "loss": 4.0797,
+      "epoch": 2.09,
+      "learning_rate": 0.0002778,
+      "loss": 3.4411,
       "step": 467
     },
     {
-      "epoch": 1.05,
-      "learning_rate": 0.000279,
-      "loss": 3.9126,
+      "epoch": 2.1,
+      "learning_rate": 0.0002784,
+      "loss": 3.5065,
       "step": 468
     },
     {
-      "epoch": 1.05,
-      "learning_rate": 0.00027959999999999997,
-      "loss": 3.9294,
+      "epoch": 2.1,
+      "learning_rate": 0.000279,
+      "loss": 3.5197,
       "step": 469
     },
     {
-      "epoch": 1.05,
-      "learning_rate": 0.0002802,
-      "loss": 4.0885,
+      "epoch": 2.11,
+      "learning_rate": 0.00027959999999999997,
+      "loss": 3.4927,
       "step": 470
     },
     {
-      "epoch": 1.06,
-      "learning_rate": 0.0002808,
-      "loss": 4.0577,
+      "epoch": 2.11,
+      "learning_rate": 0.0002802,
+      "loss": 3.5846,
       "step": 471
     },
     {
-      "epoch": 1.06,
-      "learning_rate": 0.00028139999999999996,
-      "loss": 3.9314,
+      "epoch": 2.12,
+      "learning_rate": 0.0002808,
+      "loss": 3.4224,
       "step": 472
     },
     {
-      "epoch": 1.06,
-      "learning_rate": 0.00028199999999999997,
-      "loss": 3.9632,
+      "epoch": 2.12,
+      "learning_rate": 0.00028139999999999996,
+      "loss": 3.4171,
       "step": 473
     },
     {
-      "epoch": 1.06,
-      "learning_rate": 0.0002826,
-      "loss": 3.7662,
+      "epoch": 2.13,
+      "learning_rate": 0.00028199999999999997,
+      "loss": 3.3167,
       "step": 474
     },
     {
-      "epoch": 1.07,
-      "learning_rate": 0.00028319999999999994,
-      "loss": 3.7393,
+      "epoch": 2.13,
+      "learning_rate": 0.0002826,
+      "loss": 3.4456,
       "step": 475
     },
     {
-      "epoch": 1.07,
-      "learning_rate": 0.00028379999999999996,
-      "loss": 3.9623,
+      "epoch": 2.13,
+      "learning_rate": 0.00028319999999999994,
+      "loss": 3.4501,
       "step": 476
     },
     {
-      "epoch": 1.07,
-      "learning_rate": 0.0002844,
-      "loss": 3.8472,
+      "epoch": 2.14,
+      "learning_rate": 0.00028379999999999996,
+      "loss": 3.3677,
       "step": 477
     },
     {
-      "epoch": 1.07,
-      "learning_rate": 0.000285,
-      "loss": 3.8946,
+      "epoch": 2.14,
+      "learning_rate": 0.0002844,
+      "loss": 3.2927,
       "step": 478
     },
     {
-      "epoch": 1.07,
-      "learning_rate": 0.00028559999999999995,
-      "loss": 3.9246,
+      "epoch": 2.15,
+      "learning_rate": 0.000285,
+      "loss": 3.3387,
       "step": 479
     },
     {
-      "epoch": 1.08,
-      "learning_rate": 0.00028619999999999996,
-      "loss": 3.9423,
+      "epoch": 2.15,
+      "learning_rate": 0.00028559999999999995,
+      "loss": 3.3384,
       "step": 480
     },
     {
-      "epoch": 1.08,
-      "learning_rate": 0.0002868,
-      "loss": 4.0871,
+      "epoch": 2.16,
+      "learning_rate": 0.00028619999999999996,
+      "loss": 3.2779,
       "step": 481
     },
     {
-      "epoch": 1.08,
-      "learning_rate": 0.00028739999999999994,
-      "loss": 3.7827,
+      "epoch": 2.16,
+      "learning_rate": 0.0002868,
+      "loss": 3.2192,
       "step": 482
     },
     {
-      "epoch": 1.08,
-      "learning_rate": 0.00028799999999999995,
-      "loss": 3.8545,
+      "epoch": 2.17,
+      "learning_rate": 0.00028739999999999994,
+      "loss": 3.2844,
       "step": 483
     },
     {
-      "epoch": 1.09,
-      "learning_rate": 0.00028859999999999997,
-      "loss": 3.8231,
+      "epoch": 2.17,
+      "learning_rate": 0.00028799999999999995,
+      "loss": 3.178,
       "step": 484
     },
     {
-      "epoch": 1.09,
-      "learning_rate": 0.0002892,
-      "loss": 3.8769,
+      "epoch": 2.17,
+      "learning_rate": 0.00028859999999999997,
+      "loss": 3.179,
       "step": 485
     },
     {
-      "epoch": 1.09,
-      "learning_rate": 0.00028979999999999994,
-      "loss": 3.6917,
+      "epoch": 2.18,
+      "learning_rate": 0.0002892,
+      "loss": 3.1362,
       "step": 486
     },
     {
-      "epoch": 1.09,
-      "learning_rate": 0.00029039999999999996,
-      "loss": 3.971,
+      "epoch": 2.18,
+      "learning_rate": 0.00028979999999999994,
+      "loss": 3.1162,
       "step": 487
     },
     {
-      "epoch": 1.09,
-      "learning_rate": 0.00029099999999999997,
-      "loss": 3.7305,
+      "epoch": 2.19,
+      "learning_rate": 0.00029039999999999996,
+      "loss": 3.0807,
       "step": 488
     },
     {
-      "epoch": 1.1,
-      "learning_rate": 0.0002916,
-      "loss": 3.9215,
+      "epoch": 2.19,
+      "learning_rate": 0.00029099999999999997,
+      "loss": 3.0098,
       "step": 489
     },
     {
-      "epoch": 1.1,
-      "learning_rate": 0.00029219999999999995,
-      "loss": 3.6148,
+      "epoch": 2.2,
+      "learning_rate": 0.0002916,
+      "loss": 2.8276,
       "step": 490
     },
     {
-      "epoch": 1.1,
-      "learning_rate": 0.00029279999999999996,
-      "loss": 3.634,
+      "epoch": 2.2,
+      "learning_rate": 0.00029219999999999995,
+      "loss": 2.8082,
       "step": 491
     },
     {
-      "epoch": 1.1,
-      "learning_rate": 0.0002934,
-      "loss": 3.4109,
+      "epoch": 2.21,
+      "learning_rate": 0.00029279999999999996,
+      "loss": 2.7419,
       "step": 492
     },
     {
-      "epoch": 1.11,
-      "learning_rate": 0.000294,
-      "loss": 3.3743,
+      "epoch": 2.21,
+      "learning_rate": 0.0002934,
+      "loss": 2.6181,
       "step": 493
     },
     {
-      "epoch": 1.11,
-      "learning_rate": 0.00029459999999999995,
-      "loss": 3.5371,
+      "epoch": 2.22,
+      "learning_rate": 0.000294,
+      "loss": 2.5094,
       "step": 494
     },
     {
-      "epoch": 1.11,
-      "learning_rate": 0.00029519999999999997,
-      "loss": 3.1645,
+      "epoch": 2.22,
+      "learning_rate": 0.00029459999999999995,
+      "loss": 2.455,
       "step": 495
     },
     {
-      "epoch": 1.11,
-      "learning_rate": 0.0002958,
-      "loss": 2.9923,
+      "epoch": 2.22,
+      "learning_rate": 0.00029519999999999997,
+      "loss": 2.1926,
       "step": 496
     },
     {
-      "epoch": 1.11,
-      "learning_rate": 0.0002964,
-      "loss": 6.0733,
+      "epoch": 2.23,
+      "learning_rate": 0.0002958,
+      "loss": 6.3152,
       "step": 497
     },
     {
-      "epoch": 1.12,
-      "learning_rate": 0.00029699999999999996,
-      "loss": 5.432,
+      "epoch": 2.23,
+      "learning_rate": 0.0002964,
+      "loss": 5.1198,
       "step": 498
     },
     {
-      "epoch": 1.12,
-      "learning_rate": 0.00029759999999999997,
-      "loss": 4.9446,
+      "epoch": 2.24,
+      "learning_rate": 0.00029699999999999996,
+      "loss": 4.7265,
       "step": 499
     },
     {
-      "epoch": 1.12,
-      "learning_rate": 0.0002982,
-      "loss": 4.6521,
+      "epoch": 2.24,
+      "learning_rate": 0.00029759999999999997,
+      "loss": 4.3539,
       "step": 500
     },
     {
-      "epoch": 1.12,
-      "eval_loss": 4.444121360778809,
-      "eval_runtime": 583.2794,
-      "eval_samples_per_second": 4.53,
-      "eval_steps_per_second": 0.567,
-      "eval_wer": 1.7011503371677905,
+      "epoch": 2.24,
+      "eval_loss": 4.050220966339111,
+      "eval_runtime": 594.0081,
+      "eval_samples_per_second": 4.448,
+      "eval_steps_per_second": 0.557,
+      "eval_wer": 1.7867314557715193,
       "step": 500
     },
     {
-      "epoch": 1.12,
-      "learning_rate": 0.0002988,
-      "loss": 4.414,
+      "epoch": 2.25,
+      "learning_rate": 0.0002982,
+      "loss": 4.1344,
       "step": 501
     },
     {
-      "epoch": 1.13,
-      "learning_rate": 0.00029939999999999996,
-      "loss": 4.4024,
+      "epoch": 2.25,
+      "learning_rate": 0.0002988,
+      "loss": 4.2398,
       "step": 502
     },
     {
-      "epoch": 1.13,
-      "learning_rate": 0.0003,
-      "loss": 4.3597,
+      "epoch": 2.26,
+      "learning_rate": 0.00029939999999999996,
+      "loss": 4.0204,
       "step": 503
     },
     {
-      "epoch": 1.13,
-      "learning_rate": 0.00029982658959537567,
-      "loss": 4.1466,
+      "epoch": 2.26,
+      "learning_rate": 0.0003,
+      "loss": 4.032,
       "step": 504
     },
     {
-      "epoch": 1.13,
-      "learning_rate": 0.0002996531791907514,
-      "loss": 4.204,
+      "epoch": 2.26,
+      "learning_rate": 0.0002995121951219512,
+      "loss": 3.9013,
       "step": 505
     },
     {
-      "epoch": 1.13,
-      "learning_rate": 0.00029947976878612716,
-      "loss": 4.296,
+      "epoch": 2.27,
+      "learning_rate": 0.0002990243902439024,
+      "loss": 3.7927,
       "step": 506
     },
     {
-      "epoch": 1.14,
-      "learning_rate": 0.00029930635838150286,
-      "loss": 4.1418,
+      "epoch": 2.27,
+      "learning_rate": 0.0002985365853658536,
+      "loss": 3.7671,
       "step": 507
     },
     {
-      "epoch": 1.14,
-      "learning_rate": 0.0002991329479768786,
-      "loss": 4.3272,
+      "epoch": 2.28,
+      "learning_rate": 0.00029804878048780484,
+      "loss": 3.7971,
       "step": 508
     },
     {
-      "epoch": 1.14,
-      "learning_rate": 0.0002989595375722543,
-      "loss": 4.1266,
+      "epoch": 2.28,
+      "learning_rate": 0.00029756097560975606,
+      "loss": 3.7401,
       "step": 509
     },
     {
-      "epoch": 1.14,
-      "learning_rate": 0.00029878612716763005,
-      "loss": 4.1107,
+      "epoch": 2.29,
+      "learning_rate": 0.0002970731707317073,
+      "loss": 3.5998,
       "step": 510
     },
     {
-      "epoch": 1.15,
-      "learning_rate": 0.00029861271676300574,
-      "loss": 4.0438,
+      "epoch": 2.29,
+      "learning_rate": 0.0002965853658536585,
+      "loss": 3.6632,
       "step": 511
     },
     {
-      "epoch": 1.15,
-      "learning_rate": 0.0002984393063583815,
-      "loss": 4.0974,
+      "epoch": 2.3,
+      "learning_rate": 0.0002960975609756097,
+      "loss": 3.6384,
       "step": 512
     },
     {
-      "epoch": 1.15,
-      "learning_rate": 0.0002982658959537572,
-      "loss": 4.07,
+      "epoch": 2.3,
+      "learning_rate": 0.0002956097560975609,
+      "loss": 3.6145,
       "step": 513
     },
     {
-      "epoch": 1.15,
-      "learning_rate": 0.00029809248554913293,
-      "loss": 3.9892,
+      "epoch": 2.3,
+      "learning_rate": 0.0002951219512195122,
+      "loss": 3.5137,
       "step": 514
     },
     {
-      "epoch": 1.15,
-      "learning_rate": 0.0002979190751445086,
-      "loss": 4.0635,
+      "epoch": 2.31,
+      "learning_rate": 0.00029463414634146336,
+      "loss": 3.5113,
       "step": 515
     },
     {
-      "epoch": 1.16,
-      "learning_rate": 0.00029774566473988437,
-      "loss": 4.0718,
+      "epoch": 2.31,
+      "learning_rate": 0.0002941463414634146,
+      "loss": 3.4298,
       "step": 516
     },
     {
-      "epoch": 1.16,
-      "learning_rate": 0.00029757225433526006,
-      "loss": 3.99,
+      "epoch": 2.32,
+      "learning_rate": 0.00029365853658536585,
+      "loss": 3.3688,
       "step": 517
     },
     {
-      "epoch": 1.16,
-      "learning_rate": 0.0002973988439306358,
-      "loss": 4.0126,
+      "epoch": 2.32,
+      "learning_rate": 0.00029317073170731706,
+      "loss": 3.3792,
       "step": 518
     },
     {
-      "epoch": 1.16,
-      "learning_rate": 0.00029722543352601156,
-      "loss": 4.0171,
+      "epoch": 2.33,
+      "learning_rate": 0.0002926829268292683,
+      "loss": 3.4692,
       "step": 519
     },
     {
-      "epoch": 1.17,
-      "learning_rate": 0.00029705202312138725,
-      "loss": 4.0175,
+      "epoch": 2.33,
+      "learning_rate": 0.0002921951219512195,
+      "loss": 3.3661,
       "step": 520
     },
     {
-      "epoch": 1.17,
-      "learning_rate": 0.00029687861271676295,
-      "loss": 3.9543,
+      "epoch": 2.34,
+      "learning_rate": 0.0002917073170731707,
+      "loss": 3.4411,
       "step": 521
     },
     {
-      "epoch": 1.17,
-      "learning_rate": 0.0002967052023121387,
-      "loss": 3.9275,
+      "epoch": 2.34,
+      "learning_rate": 0.00029121951219512193,
+      "loss": 3.509,
       "step": 522
     },
     {
-      "epoch": 1.17,
-      "learning_rate": 0.00029653179190751444,
-      "loss": 3.97,
+      "epoch": 2.35,
+      "learning_rate": 0.00029073170731707315,
+      "loss": 3.3465,
       "step": 523
     },
     {
-      "epoch": 1.17,
-      "learning_rate": 0.00029635838150289014,
-      "loss": 4.0399,
+      "epoch": 2.35,
+      "learning_rate": 0.00029024390243902437,
+      "loss": 3.2509,
       "step": 524
     },
     {
-      "epoch": 1.18,
-      "learning_rate": 0.00029618497109826583,
-      "loss": 3.8694,
+      "epoch": 2.35,
+      "learning_rate": 0.0002897560975609756,
+      "loss": 3.1902,
       "step": 525
     },
     {
-      "epoch": 1.18,
-      "learning_rate": 0.0002960115606936416,
-      "loss": 3.8722,
+      "epoch": 2.36,
+      "learning_rate": 0.0002892682926829268,
+      "loss": 3.2894,
       "step": 526
     },
     {
-      "epoch": 1.18,
-      "learning_rate": 0.0002958381502890173,
-      "loss": 3.8752,
+      "epoch": 2.36,
+      "learning_rate": 0.000288780487804878,
+      "loss": 3.2102,
       "step": 527
     },
     {
-      "epoch": 1.18,
-      "learning_rate": 0.000295664739884393,
-      "loss": 3.9737,
+      "epoch": 2.37,
+      "learning_rate": 0.00028829268292682923,
+      "loss": 2.9926,
       "step": 528
     },
     {
-      "epoch": 1.19,
-      "learning_rate": 0.00029549132947976877,
-      "loss": 3.7855,
+      "epoch": 2.37,
+      "learning_rate": 0.00028780487804878045,
+      "loss": 3.1141,
       "step": 529
     },
     {
-      "epoch": 1.19,
-      "learning_rate": 0.0002953179190751445,
-      "loss": 3.8993,
+      "epoch": 2.38,
+      "learning_rate": 0.00028731707317073167,
+      "loss": 2.9693,
       "step": 530
     },
     {
-      "epoch": 1.19,
-      "learning_rate": 0.0002951445086705202,
-      "loss": 3.9065,
+      "epoch": 2.38,
+      "learning_rate": 0.0002868292682926829,
+      "loss": 3.0834,
       "step": 531
     },
     {
-      "epoch": 1.19,
-      "learning_rate": 0.0002949710982658959,
-      "loss": 4.0046,
+      "epoch": 2.39,
+      "learning_rate": 0.0002863414634146341,
+      "loss": 3.0293,
       "step": 532
     },
     {
-      "epoch": 1.2,
-      "learning_rate": 0.00029479768786127165,
-      "loss": 3.8726,
+      "epoch": 2.39,
+      "learning_rate": 0.00028585365853658537,
+      "loss": 2.9632,
       "step": 533
     },
     {
-      "epoch": 1.2,
-      "learning_rate": 0.0002946242774566474,
-      "loss": 3.9889,
+      "epoch": 2.39,
+      "learning_rate": 0.00028536585365853654,
+      "loss": 2.8581,
       "step": 534
     },
     {
-      "epoch": 1.2,
-      "learning_rate": 0.0002944508670520231,
-      "loss": 3.8137,
+      "epoch": 2.4,
+      "learning_rate": 0.00028487804878048775,
+      "loss": 2.6841,
       "step": 535
     },
     {
-      "epoch": 1.2,
-      "learning_rate": 0.00029427745664739884,
-      "loss": 3.712,
+      "epoch": 2.4,
+      "learning_rate": 0.000284390243902439,
+      "loss": 2.7088,
       "step": 536
     },
     {
-      "epoch": 1.2,
-      "learning_rate": 0.00029410404624277453,
-      "loss": 3.8098,
+      "epoch": 2.41,
+      "learning_rate": 0.0002839024390243902,
+      "loss": 2.8209,
       "step": 537
     },
     {
-      "epoch": 1.21,
-      "learning_rate": 0.0002939306358381503,
-      "loss": 3.5976,
+      "epoch": 2.41,
+      "learning_rate": 0.00028341463414634146,
+      "loss": 2.5995,
       "step": 538
     },
     {
-      "epoch": 1.21,
-      "learning_rate": 0.00029375722543352597,
-      "loss": 3.6913,
+      "epoch": 2.42,
+      "learning_rate": 0.0002829268292682927,
+      "loss": 2.7186,
       "step": 539
     },
     {
-      "epoch": 1.21,
-      "learning_rate": 0.0002935838150289017,
-      "loss": 3.7072,
+      "epoch": 2.42,
+      "learning_rate": 0.00028243902439024384,
+      "loss": 2.3942,
       "step": 540
     },
     {
-      "epoch": 1.21,
-      "learning_rate": 0.0002934104046242774,
-      "loss": 3.6075,
+      "epoch": 2.43,
+      "learning_rate": 0.0002819512195121951,
+      "loss": 2.4166,
       "step": 541
     },
     {
-      "epoch": 1.22,
-      "learning_rate": 0.00029323699421965316,
-      "loss": 3.5426,
+      "epoch": 2.43,
+      "learning_rate": 0.0002814634146341463,
+      "loss": 2.1934,
       "step": 542
     },
     {
-      "epoch": 1.22,
-      "learning_rate": 0.00029306358381502886,
-      "loss": 3.4171,
+      "epoch": 2.43,
+      "learning_rate": 0.00028097560975609754,
+      "loss": 2.0537,
       "step": 543
     },
     {
-      "epoch": 1.22,
-      "learning_rate": 0.0002928901734104046,
-      "loss": 3.1822,
+      "epoch": 2.44,
+      "learning_rate": 0.00028048780487804876,
+      "loss": 1.8673,
       "step": 544
     },
     {
-      "epoch": 1.22,
-      "learning_rate": 0.0002927167630057803,
-      "loss": 3.3456,
+      "epoch": 2.44,
+      "learning_rate": 0.00028,
+      "loss": 1.7229,
       "step": 545
     },
     {
-      "epoch": 1.22,
-      "learning_rate": 0.00029254335260115604,
-      "loss": 2.927,
+      "epoch": 2.45,
+      "learning_rate": 0.0002795121951219512,
+      "loss": 1.6225,
       "step": 546
     },
     {
-      "epoch": 1.23,
-      "learning_rate": 0.0002923699421965318,
-      "loss": 5.8096,
+      "epoch": 2.45,
+      "learning_rate": 0.0002790243902439024,
+      "loss": 6.0962,
       "step": 547
     },
     {
-      "epoch": 1.23,
-      "learning_rate": 0.0002921965317919075,
-      "loss": 5.1263,
+      "epoch": 2.46,
+      "learning_rate": 0.0002785365853658536,
+      "loss": 5.1113,
       "step": 548
     },
     {
-      "epoch": 1.23,
-      "learning_rate": 0.0002920231213872832,
-      "loss": 4.7905,
+      "epoch": 2.46,
+      "learning_rate": 0.00027804878048780484,
+      "loss": 4.4441,
       "step": 549
     },
     {
-      "epoch": 1.23,
-      "learning_rate": 0.0002918497109826589,
-      "loss": 4.6619,
+      "epoch": 2.47,
+      "learning_rate": 0.00027756097560975606,
+      "loss": 3.93,
       "step": 550
     },
     {
-      "epoch": 1.24,
-      "learning_rate": 0.0002916763005780347,
-      "loss": 4.4531,
+      "epoch": 2.47,
+      "learning_rate": 0.0002770731707317073,
+      "loss": 3.6004,
       "step": 551
     },
     {
-      "epoch": 1.24,
-      "learning_rate": 0.00029150289017341037,
-      "loss": 4.3329,
+      "epoch": 2.48,
+      "learning_rate": 0.00027658536585365855,
+      "loss": 3.4879,
       "step": 552
     },
     {
-      "epoch": 1.24,
-      "learning_rate": 0.0002913294797687861,
-      "loss": 4.4172,
+      "epoch": 2.48,
+      "learning_rate": 0.0002760975609756097,
+      "loss": 3.3519,
       "step": 553
     },
     {
-      "epoch": 1.24,
-      "learning_rate": 0.00029115606936416186,
-      "loss": 4.2514,
+      "epoch": 2.48,
+      "learning_rate": 0.00027560975609756093,
+      "loss": 3.2981,
       "step": 554
     },
     {
-      "epoch": 1.24,
-      "learning_rate": 0.00029098265895953756,
-      "loss": 4.2204,
+      "epoch": 2.49,
+      "learning_rate": 0.0002751219512195122,
+      "loss": 3.1892,
       "step": 555
     },
     {
-      "epoch": 1.25,
-      "learning_rate": 0.00029080924855491325,
-      "loss": 4.0057,
+      "epoch": 2.49,
+      "learning_rate": 0.00027463414634146336,
+      "loss": 3.1885,
       "step": 556
     },
     {
-      "epoch": 1.25,
-      "learning_rate": 0.000290635838150289,
-      "loss": 4.0476,
+      "epoch": 2.5,
+      "learning_rate": 0.00027414634146341463,
+      "loss": 2.9167,
       "step": 557
     },
     {
-      "epoch": 1.25,
-      "learning_rate": 0.00029046242774566475,
-      "loss": 4.0568,
+      "epoch": 2.5,
+      "learning_rate": 0.00027365853658536585,
+      "loss": 2.8781,
       "step": 558
     },
     {
-      "epoch": 1.25,
-      "learning_rate": 0.00029028901734104044,
-      "loss": 4.0171,
+      "epoch": 2.51,
+      "learning_rate": 0.000273170731707317,
+      "loss": 2.8923,
       "step": 559
     },
     {
-      "epoch": 1.26,
-      "learning_rate": 0.00029011560693641613,
-      "loss": 4.0169,
+      "epoch": 2.51,
+      "learning_rate": 0.0002726829268292683,
+      "loss": 2.6902,
       "step": 560
     },
     {
-      "epoch": 1.26,
-      "learning_rate": 0.0002899421965317919,
-      "loss": 4.0429,
+      "epoch": 2.52,
+      "learning_rate": 0.0002721951219512195,
+      "loss": 2.6606,
       "step": 561
     },
     {
-      "epoch": 1.26,
-      "learning_rate": 0.00028976878612716763,
-      "loss": 3.8805,
+      "epoch": 2.52,
+      "learning_rate": 0.0002717073170731707,
+      "loss": 2.5854,
       "step": 562
     },
     {
-      "epoch": 1.26,
-      "learning_rate": 0.0002895953757225433,
-      "loss": 4.0099,
+      "epoch": 2.52,
+      "learning_rate": 0.00027121951219512193,
+      "loss": 2.4889,
       "step": 563
     },
     {
-      "epoch": 1.26,
-      "learning_rate": 0.00028942196531791907,
-      "loss": 3.8666,
+      "epoch": 2.53,
+      "learning_rate": 0.00027073170731707315,
+      "loss": 2.3526,
       "step": 564
     },
     {
-      "epoch": 1.27,
-      "learning_rate": 0.00028924855491329476,
-      "loss": 3.8508,
+      "epoch": 2.53,
+      "learning_rate": 0.00027024390243902437,
+      "loss": 2.3056,
       "step": 565
     },
     {
-      "epoch": 1.27,
-      "learning_rate": 0.0002890751445086705,
-      "loss": 3.8488,
+      "epoch": 2.54,
+      "learning_rate": 0.0002697560975609756,
+      "loss": 2.2212,
       "step": 566
     },
     {
-      "epoch": 1.27,
-      "learning_rate": 0.0002889017341040462,
-      "loss": 3.8655,
+      "epoch": 2.54,
+      "learning_rate": 0.0002692682926829268,
+      "loss": 2.1994,
       "step": 567
     },
     {
-      "epoch": 1.27,
-      "learning_rate": 0.00028872832369942195,
-      "loss": 4.0298,
+      "epoch": 2.55,
+      "learning_rate": 0.000268780487804878,
+      "loss": 2.1812,
       "step": 568
     },
     {
-      "epoch": 1.28,
-      "learning_rate": 0.00028855491329479765,
-      "loss": 3.8001,
+      "epoch": 2.55,
+      "learning_rate": 0.00026829268292682924,
+      "loss": 2.058,
       "step": 569
     },
     {
-      "epoch": 1.28,
-      "learning_rate": 0.0002883815028901734,
-      "loss": 3.8585,
+      "epoch": 2.56,
+      "learning_rate": 0.00026780487804878045,
+      "loss": 1.9894,
       "step": 570
     },
     {
-      "epoch": 1.28,
-      "learning_rate": 0.0002882080924855491,
-      "loss": 3.7756,
+      "epoch": 2.56,
+      "learning_rate": 0.0002673170731707317,
+      "loss": 2.0365,
       "step": 571
     },
     {
-      "epoch": 1.28,
-      "learning_rate": 0.00028803468208092484,
-      "loss": 3.9276,
+      "epoch": 2.57,
+      "learning_rate": 0.0002668292682926829,
+      "loss": 1.9143,
       "step": 572
     },
     {
-      "epoch": 1.28,
-      "learning_rate": 0.00028786127167630053,
-      "loss": 3.8585,
+      "epoch": 2.57,
+      "learning_rate": 0.0002663414634146341,
+      "loss": 1.7815,
       "step": 573
     },
     {
-      "epoch": 1.29,
-      "learning_rate": 0.0002876878612716763,
-      "loss": 3.7575,
+      "epoch": 2.57,
+      "learning_rate": 0.0002658536585365854,
+      "loss": 1.728,
       "step": 574
     },
     {
-      "epoch": 1.29,
-      "learning_rate": 0.000287514450867052,
-      "loss": 3.7872,
+      "epoch": 2.58,
+      "learning_rate": 0.00026536585365853654,
+      "loss": 1.7936,
       "step": 575
     },
     {
-      "epoch": 1.29,
-      "learning_rate": 0.0002873410404624277,
-      "loss": 3.8801,
+      "epoch": 2.58,
+      "learning_rate": 0.0002648780487804878,
+      "loss": 1.7649,
       "step": 576
     },
     {
-      "epoch": 1.29,
-      "learning_rate": 0.0002871676300578034,
-      "loss": 3.9001,
+      "epoch": 2.59,
+      "learning_rate": 0.000264390243902439,
+      "loss": 1.6668,
       "step": 577
     },
     {
-      "epoch": 1.3,
-      "learning_rate": 0.00028699421965317916,
-      "loss": 3.7872,
+      "epoch": 2.59,
+      "learning_rate": 0.0002639024390243902,
+      "loss": 1.5596,
       "step": 578
     },
     {
-      "epoch": 1.3,
-      "learning_rate": 0.0002868208092485549,
-      "loss": 3.7781,
+      "epoch": 2.6,
+      "learning_rate": 0.00026341463414634146,
+      "loss": 1.5518,
       "step": 579
     },
     {
-      "epoch": 1.3,
-      "learning_rate": 0.0002866473988439306,
-      "loss": 3.8367,
+      "epoch": 2.6,
+      "learning_rate": 0.0002629268292682927,
+      "loss": 1.5524,
       "step": 580
     },
     {
-      "epoch": 1.3,
-      "learning_rate": 0.00028647398843930635,
-      "loss": 3.966,
+      "epoch": 2.61,
+      "learning_rate": 0.0002624390243902439,
+      "loss": 1.426,
       "step": 581
     },
     {
-      "epoch": 1.3,
-      "learning_rate": 0.0002863005780346821,
-      "loss": 3.8266,
+      "epoch": 2.61,
+      "learning_rate": 0.0002619512195121951,
+      "loss": 1.4211,
       "step": 582
     },
     {
-      "epoch": 1.31,
-      "learning_rate": 0.0002861271676300578,
-      "loss": 3.8546,
+      "epoch": 2.61,
+      "learning_rate": 0.0002614634146341463,
+      "loss": 1.4552,
       "step": 583
     },
     {
-      "epoch": 1.31,
-      "learning_rate": 0.0002859537572254335,
-      "loss": 3.6754,
+      "epoch": 2.62,
+      "learning_rate": 0.00026097560975609754,
+      "loss": 1.3657,
       "step": 584
     },
     {
-      "epoch": 1.31,
-      "learning_rate": 0.00028578034682080923,
-      "loss": 3.5941,
+      "epoch": 2.62,
+      "learning_rate": 0.00026048780487804876,
+      "loss": 1.3157,
       "step": 585
     },
     {
-      "epoch": 1.31,
-      "learning_rate": 0.000285606936416185,
-      "loss": 3.7293,
+      "epoch": 2.63,
+      "learning_rate": 0.00026,
+      "loss": 1.2056,
       "step": 586
     },
     {
-      "epoch": 1.32,
-      "learning_rate": 0.0002854335260115607,
-      "loss": 3.6958,
+      "epoch": 2.63,
+      "learning_rate": 0.0002595121951219512,
+      "loss": 1.1727,
       "step": 587
     },
     {
-      "epoch": 1.32,
-      "learning_rate": 0.00028526011560693637,
-      "loss": 3.5926,
+      "epoch": 2.64,
+      "learning_rate": 0.0002590243902439024,
+      "loss": 1.1081,
       "step": 588
     },
     {
-      "epoch": 1.32,
-      "learning_rate": 0.0002850867052023121,
-      "loss": 3.3374,
+      "epoch": 2.64,
+      "learning_rate": 0.00025853658536585363,
+      "loss": 1.1073,
       "step": 589
     },
     {
-      "epoch": 1.32,
-      "learning_rate": 0.00028491329479768786,
-      "loss": 3.3626,
+      "epoch": 2.65,
+      "learning_rate": 0.00025804878048780484,
+      "loss": 1.0629,
       "step": 590
     },
     {
-      "epoch": 1.33,
-      "learning_rate": 0.00028473988439306356,
-      "loss": 3.4774,
+      "epoch": 2.65,
+      "learning_rate": 0.00025756097560975606,
+      "loss": 1.0085,
       "step": 591
     },
     {
-      "epoch": 1.33,
-      "learning_rate": 0.0002845664739884393,
-      "loss": 3.2165,
+      "epoch": 2.65,
+      "learning_rate": 0.0002570731707317073,
+      "loss": 0.9956,
       "step": 592
     },
     {
-      "epoch": 1.33,
-      "learning_rate": 0.000284393063583815,
-      "loss": 2.9479,
+      "epoch": 2.66,
+      "learning_rate": 0.0002565853658536585,
+      "loss": 0.9413,
       "step": 593
     },
     {
-      "epoch": 1.33,
-      "learning_rate": 0.00028421965317919075,
-      "loss": 2.984,
+      "epoch": 2.66,
+      "learning_rate": 0.0002560975609756097,
+      "loss": 0.9303,
       "step": 594
     },
     {
-      "epoch": 1.33,
-      "learning_rate": 0.00028404624277456644,
-      "loss": 3.0605,
+      "epoch": 2.67,
+      "learning_rate": 0.000255609756097561,
+      "loss": 0.917,
       "step": 595
     },
     {
-      "epoch": 1.34,
-      "learning_rate": 0.0002838728323699422,
-      "loss": 2.7119,
+      "epoch": 2.67,
+      "learning_rate": 0.0002551219512195122,
+      "loss": 1.081,
       "step": 596
     },
     {
-      "epoch": 1.34,
-      "learning_rate": 0.0002836994219653179,
-      "loss": 5.6219,
+      "epoch": 2.68,
+      "learning_rate": 0.00025463414634146336,
+      "loss": 9.2655,
       "step": 597
     },
     {
-      "epoch": 1.34,
-      "learning_rate": 0.00028352601156069363,
-      "loss": 5.3126,
+      "epoch": 2.68,
+      "learning_rate": 0.00025414634146341463,
+      "loss": 6.3665,
       "step": 598
     },
     {
-      "epoch": 1.34,
-      "learning_rate": 0.0002833526011560693,
-      "loss": 4.943,
+      "epoch": 2.69,
+      "learning_rate": 0.00025365853658536585,
+      "loss": 4.6114,
       "step": 599
     },
     {
-      "epoch": 1.35,
-      "learning_rate": 0.00028317919075144507,
-      "loss": 4.7358,
+      "epoch": 2.69,
+      "learning_rate": 0.00025317073170731707,
+      "loss": 3.6587,
       "step": 600
     },
     {
-      "epoch": 1.35,
-      "learning_rate": 0.00028300578034682076,
-      "loss": 4.3826,
+      "epoch": 2.7,
+      "learning_rate": 0.0002526829268292683,
+      "loss": 3.0331,
       "step": 601
     },
     {
-      "epoch": 1.35,
-      "learning_rate": 0.0002828323699421965,
-      "loss": 4.3389,
+      "epoch": 2.7,
+      "learning_rate": 0.0002521951219512195,
+      "loss": 2.6576,
       "step": 602
     },
     {
-      "epoch": 1.35,
-      "learning_rate": 0.00028265895953757226,
-      "loss": 4.2211,
+      "epoch": 2.7,
+      "learning_rate": 0.0002517073170731707,
+      "loss": 2.4141,
       "step": 603
     },
     {
-      "epoch": 1.35,
-      "learning_rate": 0.00028248554913294795,
-      "loss": 4.1059,
+      "epoch": 2.71,
+      "learning_rate": 0.00025121951219512194,
+      "loss": 2.2096,
       "step": 604
     },
     {
-      "epoch": 1.36,
-      "learning_rate": 0.00028231213872832365,
-      "loss": 4.0504,
+      "epoch": 2.71,
+      "learning_rate": 0.00025073170731707315,
+      "loss": 2.0265,
       "step": 605
     },
     {
-      "epoch": 1.36,
-      "learning_rate": 0.0002821387283236994,
-      "loss": 3.958,
+      "epoch": 2.72,
+      "learning_rate": 0.00025024390243902437,
+      "loss": 1.9003,
       "step": 606
     },
     {
-      "epoch": 1.36,
-      "learning_rate": 0.00028196531791907514,
-      "loss": 4.0532,
+      "epoch": 2.72,
+      "learning_rate": 0.0002497560975609756,
+      "loss": 1.7156,
       "step": 607
     },
     {
-      "epoch": 1.36,
-      "learning_rate": 0.00028179190751445083,
-      "loss": 3.9602,
+      "epoch": 2.73,
+      "learning_rate": 0.0002492682926829268,
+      "loss": 1.6201,
       "step": 608
     },
     {
-      "epoch": 1.37,
-      "learning_rate": 0.0002816184971098266,
-      "loss": 3.979,
+      "epoch": 2.73,
+      "learning_rate": 0.000248780487804878,
+      "loss": 1.4503,
       "step": 609
     },
     {
-      "epoch": 1.37,
-      "learning_rate": 0.0002814450867052023,
-      "loss": 3.8953,
+      "epoch": 2.74,
+      "learning_rate": 0.00024829268292682924,
+      "loss": 1.4161,
       "step": 610
     },
     {
-      "epoch": 1.37,
-      "learning_rate": 0.000281271676300578,
-      "loss": 3.8389,
+      "epoch": 2.74,
+      "learning_rate": 0.00024780487804878045,
+      "loss": 1.2728,
       "step": 611
     },
     {
-      "epoch": 1.37,
-      "learning_rate": 0.0002810982658959537,
-      "loss": 3.8168,
+      "epoch": 2.74,
+      "learning_rate": 0.00024731707317073167,
+      "loss": 1.2627,
       "step": 612
     },
     {
-      "epoch": 1.37,
-      "learning_rate": 0.00028092485549132947,
-      "loss": 3.8634,
+      "epoch": 2.75,
+      "learning_rate": 0.0002468292682926829,
+      "loss": 1.301,
       "step": 613
     },
     {
-      "epoch": 1.38,
-      "learning_rate": 0.00028075144508670516,
-      "loss": 3.8456,
+      "epoch": 2.75,
+      "learning_rate": 0.00024634146341463416,
+      "loss": 1.1987,
       "step": 614
     },
     {
-      "epoch": 1.38,
-      "learning_rate": 0.0002805780346820809,
-      "loss": 3.9215,
+      "epoch": 2.76,
+      "learning_rate": 0.0002458536585365853,
+      "loss": 1.0851,
       "step": 615
     },
     {
-      "epoch": 1.38,
-      "learning_rate": 0.0002804046242774566,
-      "loss": 3.7437,
+      "epoch": 2.76,
+      "learning_rate": 0.00024536585365853654,
+      "loss": 1.207,
       "step": 616
     },
     {
-      "epoch": 1.38,
-      "learning_rate": 0.00028023121387283235,
-      "loss": 3.7972,
+      "epoch": 2.77,
+      "learning_rate": 0.0002448780487804878,
+      "loss": 1.0918,
       "step": 617
     },
     {
-      "epoch": 1.39,
-      "learning_rate": 0.00028005780346820804,
-      "loss": 3.6682,
+      "epoch": 2.77,
+      "learning_rate": 0.00024439024390243897,
+      "loss": 1.1469,
       "step": 618
     },
     {
-      "epoch": 1.39,
-      "learning_rate": 0.0002798843930635838,
-      "loss": 3.7301,
+      "epoch": 2.78,
+      "learning_rate": 0.00024390243902439022,
+      "loss": 0.9882,
       "step": 619
     },
     {
-      "epoch": 1.39,
-      "learning_rate": 0.00027971098265895954,
-      "loss": 3.663,
+      "epoch": 2.78,
+      "learning_rate": 0.00024341463414634146,
+      "loss": 1.0057,
       "step": 620
     },
     {
-      "epoch": 1.39,
-      "learning_rate": 0.00027953757225433523,
-      "loss": 3.7823,
+      "epoch": 2.78,
+      "learning_rate": 0.00024292682926829268,
+      "loss": 0.9869,
       "step": 621
     },
     {
-      "epoch": 1.39,
-      "learning_rate": 0.0002793641618497109,
-      "loss": 3.763,
+      "epoch": 2.79,
+      "learning_rate": 0.00024243902439024387,
+      "loss": 0.9118,
       "step": 622
     },
     {
-      "epoch": 1.4,
-      "learning_rate": 0.00027919075144508667,
-      "loss": 3.696,
+      "epoch": 2.79,
+      "learning_rate": 0.0002419512195121951,
+      "loss": 0.8895,
       "step": 623
     },
     {
-      "epoch": 1.4,
-      "learning_rate": 0.0002790173410404624,
-      "loss": 3.6394,
+      "epoch": 2.8,
+      "learning_rate": 0.00024146341463414633,
+      "loss": 0.7842,
       "step": 624
     },
     {
-      "epoch": 1.4,
-      "learning_rate": 0.0002788439306358381,
-      "loss": 3.6472,
+      "epoch": 2.8,
+      "learning_rate": 0.00024097560975609755,
+      "loss": 0.8535,
       "step": 625
     },
     {
-      "epoch": 1.4,
-      "learning_rate": 0.00027867052023121386,
-      "loss": 3.6325,
+      "epoch": 2.81,
+      "learning_rate": 0.00024048780487804876,
+      "loss": 0.9101,
       "step": 626
     },
     {
-      "epoch": 1.41,
-      "learning_rate": 0.00027849710982658955,
-      "loss": 3.5473,
+      "epoch": 2.81,
+      "learning_rate": 0.00023999999999999998,
+      "loss": 0.9084,
       "step": 627
     },
     {
-      "epoch": 1.41,
-      "learning_rate": 0.0002783236994219653,
-      "loss": 3.6429,
+      "epoch": 2.82,
+      "learning_rate": 0.0002395121951219512,
+      "loss": 0.8409,
       "step": 628
     },
     {
-      "epoch": 1.41,
-      "learning_rate": 0.000278150289017341,
-      "loss": 3.6473,
+      "epoch": 2.82,
+      "learning_rate": 0.0002390243902439024,
+      "loss": 0.8037,
       "step": 629
     },
     {
-      "epoch": 1.41,
-      "learning_rate": 0.00027797687861271674,
-      "loss": 3.5185,
+      "epoch": 2.83,
+      "learning_rate": 0.00023853658536585366,
+      "loss": 0.7994,
       "step": 630
     },
     {
-      "epoch": 1.41,
-      "learning_rate": 0.0002778034682080925,
-      "loss": 3.7047,
+      "epoch": 2.83,
+      "learning_rate": 0.00023804878048780485,
+      "loss": 0.8201,
       "step": 631
     },
     {
-      "epoch": 1.42,
-      "learning_rate": 0.0002776300578034682,
-      "loss": 3.467,
+      "epoch": 2.83,
+      "learning_rate": 0.00023756097560975606,
+      "loss": 0.7229,
       "step": 632
     },
     {
-      "epoch": 1.42,
-      "learning_rate": 0.0002774566473988439,
-      "loss": 3.4672,
+      "epoch": 2.84,
+      "learning_rate": 0.0002370731707317073,
+      "loss": 0.7606,
       "step": 633
     },
     {
-      "epoch": 1.42,
-      "learning_rate": 0.0002772832369942196,
-      "loss": 3.366,
+      "epoch": 2.84,
+      "learning_rate": 0.0002365853658536585,
+      "loss": 0.7244,
       "step": 634
     },
     {
-      "epoch": 1.42,
-      "learning_rate": 0.0002771098265895954,
-      "loss": 3.3636,
+      "epoch": 2.85,
+      "learning_rate": 0.00023609756097560974,
+      "loss": 0.7032,
       "step": 635
     },
     {
-      "epoch": 1.43,
-      "learning_rate": 0.00027693641618497107,
-      "loss": 3.2793,
+      "epoch": 2.85,
+      "learning_rate": 0.00023560975609756096,
+      "loss": 0.6894,
       "step": 636
     },
     {
-      "epoch": 1.43,
-      "learning_rate": 0.0002767630057803468,
-      "loss": 3.5051,
+      "epoch": 2.86,
+      "learning_rate": 0.00023512195121951215,
+      "loss": 0.7164,
       "step": 637
     },
     {
-      "epoch": 1.43,
-      "learning_rate": 0.0002765895953757225,
-      "loss": 3.1726,
+      "epoch": 2.86,
+      "learning_rate": 0.0002346341463414634,
+      "loss": 0.6915,
       "step": 638
     },
     {
-      "epoch": 1.43,
-      "learning_rate": 0.00027641618497109826,
-      "loss": 3.3622,
+      "epoch": 2.87,
+      "learning_rate": 0.0002341463414634146,
+      "loss": 0.7157,
       "step": 639
     },
     {
-      "epoch": 1.43,
-      "learning_rate": 0.00027624277456647395,
-      "loss": 3.1272,
+      "epoch": 2.87,
+      "learning_rate": 0.00023365853658536583,
+      "loss": 0.7308,
       "step": 640
     },
     {
-      "epoch": 1.44,
-      "learning_rate": 0.0002760693641618497,
-      "loss": 2.7958,
+      "epoch": 2.87,
+      "learning_rate": 0.00023317073170731704,
+      "loss": 0.6428,
       "step": 641
     },
     {
-      "epoch": 1.44,
-      "learning_rate": 0.0002758959537572254,
-      "loss": 3.0075,
+      "epoch": 2.88,
+      "learning_rate": 0.0002326829268292683,
+      "loss": 0.6652,
       "step": 642
     },
     {
-      "epoch": 1.44,
-      "learning_rate": 0.00027572254335260114,
-      "loss": 2.8361,
+      "epoch": 2.88,
+      "learning_rate": 0.00023219512195121948,
+      "loss": 0.6669,
       "step": 643
     },
     {
-      "epoch": 1.44,
-      "learning_rate": 0.00027554913294797683,
-      "loss": 2.4531,
+      "epoch": 2.89,
+      "learning_rate": 0.0002317073170731707,
+      "loss": 0.6834,
       "step": 644
     },
     {
-      "epoch": 1.45,
-      "learning_rate": 0.0002753757225433526,
-      "loss": 2.4185,
+      "epoch": 2.89,
+      "learning_rate": 0.00023121951219512194,
+      "loss": 0.703,
       "step": 645
     },
     {
-      "epoch": 1.45,
-      "learning_rate": 0.0002752023121387283,
-      "loss": 2.4573,
+      "epoch": 2.9,
+      "learning_rate": 0.00023073170731707315,
+      "loss": 0.792,
       "step": 646
     },
     {
-      "epoch": 1.45,
-      "learning_rate": 0.000275028901734104,
-      "loss": 5.8762,
+      "epoch": 2.9,
+      "learning_rate": 0.00023024390243902437,
+      "loss": 7.3397,
       "step": 647
     },
     {
-      "epoch": 1.45,
-      "learning_rate": 0.00027485549132947977,
-      "loss": 5.5451,
+      "epoch": 2.91,
+      "learning_rate": 0.0002297560975609756,
+      "loss": 6.4753,
       "step": 648
     },
     {
-      "epoch": 1.46,
-      "learning_rate": 0.00027468208092485546,
-      "loss": 5.1378,
+      "epoch": 2.91,
+      "learning_rate": 0.00022926829268292683,
+      "loss": 4.5254,
       "step": 649
     },
     {
-      "epoch": 1.46,
-      "learning_rate": 0.00027450867052023116,
-      "loss": 4.6691,
+      "epoch": 2.91,
+      "learning_rate": 0.00022878048780487802,
+      "loss": 3.1011,
       "step": 650
     },
     {
-      "epoch": 1.46,
-      "learning_rate": 0.0002743352601156069,
-      "loss": 4.1925,
+      "epoch": 2.92,
+      "learning_rate": 0.00022829268292682924,
+      "loss": 2.1706,
       "step": 651
     },
     {
-      "epoch": 1.46,
-      "learning_rate": 0.00027416184971098265,
-      "loss": 3.9882,
+      "epoch": 2.92,
+      "learning_rate": 0.00022780487804878048,
+      "loss": 1.8403,
       "step": 652
     },
     {
-      "epoch": 1.46,
-      "learning_rate": 0.00027398843930635835,
-      "loss": 3.879,
+      "epoch": 2.93,
+      "learning_rate": 0.00022731707317073167,
+      "loss": 1.7287,
       "step": 653
     },
     {
-      "epoch": 1.47,
-      "learning_rate": 0.0002738150289017341,
-      "loss": 3.895,
+      "epoch": 2.93,
+      "learning_rate": 0.00022682926829268292,
+      "loss": 1.5158,
       "step": 654
     },
     {
-      "epoch": 1.47,
-      "learning_rate": 0.00027364161849710984,
-      "loss": 3.8319,
+      "epoch": 2.94,
+      "learning_rate": 0.00022634146341463413,
+      "loss": 1.2994,
       "step": 655
     },
     {
-      "epoch": 1.47,
-      "learning_rate": 0.00027346820809248554,
-      "loss": 3.7363,
+      "epoch": 2.94,
+      "learning_rate": 0.00022585365853658532,
+      "loss": 1.099,
       "step": 656
     },
     {
-      "epoch": 1.47,
-      "learning_rate": 0.00027329479768786123,
-      "loss": 3.6597,
+      "epoch": 2.95,
+      "learning_rate": 0.00022536585365853657,
+      "loss": 0.9544,
       "step": 657
     },
     {
-      "epoch": 1.48,
-      "learning_rate": 0.000273121387283237,
-      "loss": 3.6444,
+      "epoch": 2.95,
+      "learning_rate": 0.00022487804878048778,
+      "loss": 0.97,
       "step": 658
     },
     {
-      "epoch": 1.48,
-      "learning_rate": 0.0002729479768786127,
-      "loss": 3.5363,
+      "epoch": 2.96,
+      "learning_rate": 0.000224390243902439,
+      "loss": 0.9019,
       "step": 659
     },
     {
-      "epoch": 1.48,
-      "learning_rate": 0.0002727745664739884,
-      "loss": 3.4726,
+      "epoch": 2.96,
+      "learning_rate": 0.00022390243902439022,
+      "loss": 0.873,
       "step": 660
     },
     {
-      "epoch": 1.48,
-      "learning_rate": 0.0002726011560693641,
-      "loss": 3.3294,
+      "epoch": 2.96,
+      "learning_rate": 0.00022341463414634146,
+      "loss": 0.8081,
       "step": 661
     },
     {
-      "epoch": 1.48,
-      "learning_rate": 0.00027242774566473986,
-      "loss": 3.4765,
+      "epoch": 2.97,
+      "learning_rate": 0.00022292682926829265,
+      "loss": 0.8081,
       "step": 662
     },
     {
-      "epoch": 1.49,
-      "learning_rate": 0.0002722543352601156,
-      "loss": 3.3104,
+      "epoch": 2.97,
+      "learning_rate": 0.00022243902439024387,
+      "loss": 0.697,
       "step": 663
     },
     {
-      "epoch": 1.49,
-      "learning_rate": 0.0002720809248554913,
-      "loss": 3.3259,
+      "epoch": 2.98,
+      "learning_rate": 0.0002219512195121951,
+      "loss": 0.663,
       "step": 664
     },
     {
-      "epoch": 1.49,
-      "learning_rate": 0.00027190751445086705,
-      "loss": 3.1058,
+      "epoch": 2.98,
+      "learning_rate": 0.0002214634146341463,
+      "loss": 0.6758,
       "step": 665
     },
     {
-      "epoch": 1.49,
-      "learning_rate": 0.00027173410404624274,
-      "loss": 3.2575,
+      "epoch": 2.99,
+      "learning_rate": 0.00022097560975609755,
+      "loss": 0.6657,
       "step": 666
     },
     {
-      "epoch": 1.5,
-      "learning_rate": 0.0002715606936416185,
-      "loss": 3.0172,
+      "epoch": 2.99,
+      "learning_rate": 0.00022048780487804876,
+      "loss": 0.5622,
       "step": 667
     },
     {
-      "epoch": 1.5,
-      "learning_rate": 0.0002713872832369942,
-      "loss": 3.0809,
+      "epoch": 3.0,
+      "learning_rate": 0.00021999999999999995,
+      "loss": 0.607,
       "step": 668
     },
     {
-      "epoch": 1.5,
-      "learning_rate": 0.00027121387283236993,
-      "loss": 3.1742,
+      "epoch": 3.0,
+      "learning_rate": 0.0002195121951219512,
+      "loss": 0.6879,
       "step": 669
     },
     {
-      "epoch": 1.5,
-      "learning_rate": 0.0002710404624277456,
-      "loss": 3.0458,
+      "epoch": 3.0,
+      "learning_rate": 0.00021902439024390241,
+      "loss": 3.5432,
       "step": 670
     },
     {
-      "epoch": 1.5,
-      "learning_rate": 0.0002708670520231214,
-      "loss": 2.9033,
+      "epoch": 3.01,
+      "learning_rate": 0.00021853658536585366,
+      "loss": 3.2184,
       "step": 671
     },
     {
-      "epoch": 1.51,
-      "learning_rate": 0.00027069364161849707,
-      "loss": 2.9879,
+      "epoch": 3.01,
+      "learning_rate": 0.00021804878048780485,
+      "loss": 1.9385,
       "step": 672
     },
     {
-      "epoch": 1.51,
-      "learning_rate": 0.0002705202312138728,
-      "loss": 2.8564,
+      "epoch": 3.02,
+      "learning_rate": 0.0002175609756097561,
+      "loss": 1.2234,
       "step": 673
     },
     {
-      "epoch": 1.51,
-      "learning_rate": 0.0002703468208092485,
-      "loss": 2.786,
+      "epoch": 3.02,
+      "learning_rate": 0.0002170731707317073,
+      "loss": 1.2986,
       "step": 674
     },
     {
-      "epoch": 1.51,
-      "learning_rate": 0.00027017341040462426,
-      "loss": 2.9336,
+      "epoch": 3.03,
+      "learning_rate": 0.0002165853658536585,
+      "loss": 1.1385,
       "step": 675
     },
     {
-      "epoch": 1.52,
-      "learning_rate": 0.00027,
-      "loss": 2.7786,
+      "epoch": 3.03,
+      "learning_rate": 0.00021609756097560974,
+      "loss": 1.0251,
       "step": 676
     },
     {
-      "epoch": 1.52,
-      "learning_rate": 0.0002698265895953757,
-      "loss": 2.664,
+      "epoch": 3.04,
+      "learning_rate": 0.00021560975609756096,
+      "loss": 0.8948,
       "step": 677
     },
     {
-      "epoch": 1.52,
-      "learning_rate": 0.0002696531791907514,
-      "loss": 2.65,
+      "epoch": 3.04,
+      "learning_rate": 0.00021512195121951218,
+      "loss": 0.6973,
       "step": 678
     },
     {
-      "epoch": 1.52,
-      "learning_rate": 0.00026947976878612714,
-      "loss": 2.6503,
+      "epoch": 3.04,
+      "learning_rate": 0.0002146341463414634,
+      "loss": 0.7551,
       "step": 679
     },
     {
-      "epoch": 1.52,
-      "learning_rate": 0.0002693063583815029,
-      "loss": 2.7218,
+      "epoch": 3.05,
+      "learning_rate": 0.00021414634146341464,
+      "loss": 0.6294,
       "step": 680
     },
     {
-      "epoch": 1.53,
-      "learning_rate": 0.0002691329479768786,
-      "loss": 2.6982,
+      "epoch": 3.05,
+      "learning_rate": 0.00021365853658536583,
+      "loss": 0.6347,
       "step": 681
     },
     {
-      "epoch": 1.53,
-      "learning_rate": 0.00026895953757225433,
-      "loss": 2.5694,
+      "epoch": 3.06,
+      "learning_rate": 0.00021317073170731704,
+      "loss": 0.6635,
       "step": 682
     },
     {
-      "epoch": 1.53,
-      "learning_rate": 0.0002687861271676301,
-      "loss": 2.5128,
+      "epoch": 3.06,
+      "learning_rate": 0.0002126829268292683,
+      "loss": 0.5707,
       "step": 683
     },
     {
-      "epoch": 1.53,
-      "learning_rate": 0.0002687861271676301,
-      "loss": 2.7005,
+      "epoch": 3.07,
+      "learning_rate": 0.00021219512195121948,
+      "loss": 0.5126,
       "step": 684
     },
     {
-      "epoch": 1.54,
-      "learning_rate": 0.00026861271676300577,
-      "loss": 2.325,
+      "epoch": 3.07,
+      "learning_rate": 0.00021170731707317072,
+      "loss": 0.6055,
       "step": 685
     },
     {
-      "epoch": 1.54,
-      "learning_rate": 0.00026843930635838146,
-      "loss": 2.5138,
+      "epoch": 3.08,
+      "learning_rate": 0.00021121951219512194,
+      "loss": 0.5535,
       "step": 686
     },
     {
-      "epoch": 1.54,
-      "learning_rate": 0.0002682658959537572,
-      "loss": 2.348,
+      "epoch": 3.08,
+      "learning_rate": 0.00021073170731707313,
+      "loss": 0.5278,
       "step": 687
     },
     {
-      "epoch": 1.54,
-      "learning_rate": 0.00026809248554913296,
-      "loss": 2.2269,
+      "epoch": 3.09,
+      "learning_rate": 0.00021024390243902437,
+      "loss": 0.5102,
       "step": 688
     },
     {
-      "epoch": 1.54,
-      "learning_rate": 0.00026791907514450865,
-      "loss": 2.2502,
+      "epoch": 3.09,
+      "learning_rate": 0.0002097560975609756,
+      "loss": 0.5113,
       "step": 689
     },
     {
-      "epoch": 1.55,
-      "learning_rate": 0.00026774566473988435,
-      "loss": 1.9571,
+      "epoch": 3.09,
+      "learning_rate": 0.0002092682926829268,
+      "loss": 0.4592,
       "step": 690
     },
     {
-      "epoch": 1.55,
-      "learning_rate": 0.0002675722543352601,
-      "loss": 2.003,
+      "epoch": 3.1,
+      "learning_rate": 0.00020878048780487802,
+      "loss": 0.4637,
       "step": 691
     },
     {
-      "epoch": 1.55,
-      "learning_rate": 0.00026739884393063584,
-      "loss": 1.8658,
+      "epoch": 3.1,
+      "learning_rate": 0.00020829268292682927,
+      "loss": 0.4902,
       "step": 692
     },
     {
-      "epoch": 1.55,
-      "learning_rate": 0.00026722543352601153,
-      "loss": 1.5905,
+      "epoch": 3.11,
+      "learning_rate": 0.00020780487804878046,
+      "loss": 0.4212,
       "step": 693
     },
     {
-      "epoch": 1.56,
-      "learning_rate": 0.0002670520231213873,
-      "loss": 1.7254,
+      "epoch": 3.11,
+      "learning_rate": 0.00020731707317073167,
+      "loss": 0.4117,
       "step": 694
     },
     {
-      "epoch": 1.56,
-      "learning_rate": 0.000266878612716763,
-      "loss": 1.4685,
+      "epoch": 3.12,
+      "learning_rate": 0.00020682926829268292,
+      "loss": 0.4027,
       "step": 695
     },
     {
-      "epoch": 1.56,
-      "learning_rate": 0.0002667052023121387,
-      "loss": 1.1921,
+      "epoch": 3.12,
+      "learning_rate": 0.0002063414634146341,
+      "loss": 0.4618,
       "step": 696
     },
     {
-      "epoch": 1.56,
-      "learning_rate": 0.0002665317919075144,
-      "loss": 8.7366,
+      "epoch": 3.13,
+      "learning_rate": 0.00020585365853658535,
+      "loss": 0.3627,
       "step": 697
     },
     {
-      "epoch": 1.57,
-      "learning_rate": 0.00026635838150289016,
-      "loss": 6.141,
+      "epoch": 3.13,
+      "learning_rate": 0.00020536585365853657,
+      "loss": 0.4113,
       "step": 698
     },
     {
-      "epoch": 1.57,
-      "learning_rate": 0.00026618497109826586,
-      "loss": 4.6943,
+      "epoch": 3.13,
+      "learning_rate": 0.0002048780487804878,
+      "loss": 0.3982,
       "step": 699
     },
     {
-      "epoch": 1.57,
-      "learning_rate": 0.00026601156069364155,
-      "loss": 3.7024,
+      "epoch": 3.14,
+      "learning_rate": 0.000204390243902439,
+      "loss": 0.3762,
       "step": 700
     },
     {
-      "epoch": 1.57,
-      "learning_rate": 0.0002658381502890173,
-      "loss": 3.4388,
+      "epoch": 3.14,
+      "learning_rate": 0.00020390243902439022,
+      "loss": 0.3667,
       "step": 701
     },
     {
-      "epoch": 1.57,
-      "learning_rate": 0.00026566473988439305,
-      "loss": 3.0568,
+      "epoch": 3.15,
+      "learning_rate": 0.00020341463414634146,
+      "loss": 0.3957,
       "step": 702
     },
     {
-      "epoch": 1.58,
-      "learning_rate": 0.00026549132947976874,
-      "loss": 3.0959,
+      "epoch": 3.15,
+      "learning_rate": 0.00020292682926829265,
+      "loss": 0.3483,
       "step": 703
     },
     {
-      "epoch": 1.58,
-      "learning_rate": 0.0002653179190751445,
-      "loss": 2.9189,
+      "epoch": 3.16,
+      "learning_rate": 0.0002024390243902439,
+      "loss": 0.3318,
       "step": 704
     },
     {
-      "epoch": 1.58,
-      "learning_rate": 0.00026514450867052024,
-      "loss": 2.7027,
+      "epoch": 3.16,
+      "learning_rate": 0.00020195121951219511,
+      "loss": 0.4213,
       "step": 705
     },
     {
-      "epoch": 1.58,
-      "learning_rate": 0.00026497109826589593,
-      "loss": 2.7317,
+      "epoch": 3.17,
+      "learning_rate": 0.0002014634146341463,
+      "loss": 0.368,
       "step": 706
     },
     {
-      "epoch": 1.59,
-      "learning_rate": 0.0002647976878612716,
-      "loss": 2.5358,
+      "epoch": 3.17,
+      "learning_rate": 0.00020097560975609755,
+      "loss": 0.3552,
       "step": 707
     },
     {
-      "epoch": 1.59,
-      "learning_rate": 0.00026462427745664737,
-      "loss": 2.3571,
+      "epoch": 3.17,
+      "learning_rate": 0.00020048780487804877,
+      "loss": 0.3594,
       "step": 708
     },
     {
-      "epoch": 1.59,
-      "learning_rate": 0.0002644508670520231,
-      "loss": 2.3926,
+      "epoch": 3.18,
+      "learning_rate": 0.00019999999999999998,
+      "loss": 0.3226,
       "step": 709
     },
     {
-      "epoch": 1.59,
-      "learning_rate": 0.0002642774566473988,
-      "loss": 2.1589,
+      "epoch": 3.18,
+      "learning_rate": 0.0001995121951219512,
+      "loss": 0.3709,
       "step": 710
     },
     {
-      "epoch": 1.59,
-      "learning_rate": 0.00026410404624277456,
-      "loss": 2.0983,
+      "epoch": 3.19,
+      "learning_rate": 0.00019902439024390244,
+      "loss": 0.3987,
       "step": 711
     },
     {
-      "epoch": 1.6,
-      "learning_rate": 0.00026393063583815025,
-      "loss": 1.9613,
+      "epoch": 3.19,
+      "learning_rate": 0.00019853658536585363,
+      "loss": 0.3575,
       "step": 712
     },
     {
-      "epoch": 1.6,
-      "learning_rate": 0.000263757225433526,
-      "loss": 2.0088,
+      "epoch": 3.2,
+      "learning_rate": 0.00019804878048780485,
+      "loss": 0.3317,
       "step": 713
     },
     {
-      "epoch": 1.6,
-      "learning_rate": 0.0002635838150289017,
-      "loss": 2.0606,
+      "epoch": 3.2,
+      "learning_rate": 0.0001975609756097561,
+      "loss": 0.2524,
       "step": 714
     },
     {
-      "epoch": 1.6,
-      "learning_rate": 0.00026341040462427744,
-      "loss": 1.9965,
+      "epoch": 3.21,
+      "learning_rate": 0.00019707317073170728,
+      "loss": 0.3431,
       "step": 715
     },
     {
-      "epoch": 1.61,
-      "learning_rate": 0.00026323699421965314,
-      "loss": 1.8923,
+      "epoch": 3.21,
+      "learning_rate": 0.00019658536585365853,
+      "loss": 0.3492,
       "step": 716
     },
     {
-      "epoch": 1.61,
-      "learning_rate": 0.0002630635838150289,
-      "loss": 1.83,
+      "epoch": 3.22,
+      "learning_rate": 0.00019609756097560974,
+      "loss": 0.2618,
       "step": 717
     },
     {
-      "epoch": 1.61,
-      "learning_rate": 0.0002628901734104046,
-      "loss": 1.8531,
+      "epoch": 3.22,
+      "learning_rate": 0.00019560975609756093,
+      "loss": 0.3719,
       "step": 718
     },
     {
-      "epoch": 1.61,
-      "learning_rate": 0.0002627167630057803,
-      "loss": 1.6699,
+      "epoch": 3.22,
+      "learning_rate": 0.00019512195121951218,
+      "loss": 0.3291,
       "step": 719
     },
     {
-      "epoch": 1.61,
-      "learning_rate": 0.000262543352601156,
-      "loss": 1.6279,
+      "epoch": 3.23,
+      "learning_rate": 0.0001946341463414634,
+      "loss": 3.3472,
       "step": 720
     },
     {
-      "epoch": 1.62,
-      "learning_rate": 0.00026236994219653177,
-      "loss": 1.6581,
+      "epoch": 3.23,
+      "learning_rate": 0.0001941463414634146,
+      "loss": 2.2989,
       "step": 721
     },
     {
-      "epoch": 1.62,
-      "learning_rate": 0.0002621965317919075,
-      "loss": 1.6376,
+      "epoch": 3.24,
+      "learning_rate": 0.00019365853658536583,
+      "loss": 1.2984,
       "step": 722
     },
     {
-      "epoch": 1.62,
-      "learning_rate": 0.0002620231213872832,
-      "loss": 1.6408,
+      "epoch": 3.24,
+      "learning_rate": 0.00019317073170731707,
+      "loss": 0.9822,
       "step": 723
     },
     {
-      "epoch": 1.62,
-      "learning_rate": 0.0002618497109826589,
-      "loss": 1.4039,
+      "epoch": 3.25,
+      "learning_rate": 0.0001926829268292683,
+      "loss": 0.9058,
       "step": 724
     },
     {
-      "epoch": 1.63,
-      "learning_rate": 0.00026167630057803465,
-      "loss": 1.4447,
+      "epoch": 3.25,
+      "learning_rate": 0.00019219512195121948,
+      "loss": 0.8659,
       "step": 725
     },
     {
-      "epoch": 1.63,
-      "learning_rate": 0.0002615028901734104,
-      "loss": 1.655,
+      "epoch": 3.26,
+      "learning_rate": 0.00019170731707317072,
+      "loss": 0.7585,
       "step": 726
     },
     {
-      "epoch": 1.63,
-      "learning_rate": 0.0002613294797687861,
-      "loss": 1.3933,
+      "epoch": 3.26,
+      "learning_rate": 0.00019121951219512194,
+      "loss": 0.7159,
       "step": 727
     },
     {
-      "epoch": 1.63,
-      "learning_rate": 0.00026115606936416184,
-      "loss": 1.3983,
+      "epoch": 3.26,
+      "learning_rate": 0.00019073170731707316,
+      "loss": 0.643,
       "step": 728
     },
     {
-      "epoch": 1.63,
-      "learning_rate": 0.0002609826589595376,
-      "loss": 1.2188,
+      "epoch": 3.27,
+      "learning_rate": 0.00019024390243902437,
+      "loss": 0.5668,
       "step": 729
     },
     {
-      "epoch": 1.64,
-      "learning_rate": 0.0002608092485549133,
-      "loss": 1.3292,
+      "epoch": 3.27,
+      "learning_rate": 0.00018975609756097562,
+      "loss": 0.5185,
       "step": 730
     },
     {
-      "epoch": 1.64,
-      "learning_rate": 0.000260635838150289,
-      "loss": 1.3526,
+      "epoch": 3.28,
+      "learning_rate": 0.0001892682926829268,
+      "loss": 0.5032,
       "step": 731
     },
     {
-      "epoch": 1.64,
-      "learning_rate": 0.0002604624277456647,
-      "loss": 1.3375,
+      "epoch": 3.28,
+      "learning_rate": 0.00018878048780487803,
+      "loss": 0.478,
       "step": 732
     },
     {
-      "epoch": 1.64,
-      "learning_rate": 0.00026028901734104047,
-      "loss": 1.1941,
+      "epoch": 3.29,
+      "learning_rate": 0.00018829268292682927,
+      "loss": 0.4333,
       "step": 733
     },
     {
-      "epoch": 1.65,
-      "learning_rate": 0.00026011560693641616,
-      "loss": 1.2204,
+      "epoch": 3.29,
+      "learning_rate": 0.00018780487804878046,
+      "loss": 0.4642,
       "step": 734
     },
     {
-      "epoch": 1.65,
-      "learning_rate": 0.00025994219653179186,
-      "loss": 1.2544,
+      "epoch": 3.3,
+      "learning_rate": 0.0001873170731707317,
+      "loss": 0.4117,
       "step": 735
     },
     {
-      "epoch": 1.65,
-      "learning_rate": 0.0002597687861271676,
-      "loss": 1.3232,
+      "epoch": 3.3,
+      "learning_rate": 0.00018682926829268292,
+      "loss": 0.4252,
       "step": 736
     },
     {
-      "epoch": 1.65,
-      "learning_rate": 0.00025959537572254335,
-      "loss": 1.1686,
+      "epoch": 3.3,
+      "learning_rate": 0.0001863414634146341,
+      "loss": 0.3947,
       "step": 737
     },
     {
-      "epoch": 1.65,
-      "learning_rate": 0.00025942196531791905,
-      "loss": 1.1723,
+      "epoch": 3.31,
+      "learning_rate": 0.00018585365853658535,
+      "loss": 0.4007,
       "step": 738
     },
     {
-      "epoch": 1.66,
-      "learning_rate": 0.0002592485549132948,
-      "loss": 1.1758,
+      "epoch": 3.31,
+      "learning_rate": 0.00018536585365853657,
+      "loss": 0.4026,
       "step": 739
     },
     {
-      "epoch": 1.66,
-      "learning_rate": 0.0002590751445086705,
-      "loss": 1.1159,
+      "epoch": 3.32,
+      "learning_rate": 0.0001848780487804878,
+      "loss": 0.3837,
       "step": 740
     },
     {
-      "epoch": 1.66,
-      "learning_rate": 0.00025890173410404624,
-      "loss": 1.3023,
+      "epoch": 3.32,
+      "learning_rate": 0.000184390243902439,
+      "loss": 0.3415,
       "step": 741
     },
     {
-      "epoch": 1.66,
-      "learning_rate": 0.00025872832369942193,
-      "loss": 1.1216,
+      "epoch": 3.33,
+      "learning_rate": 0.00018390243902439025,
+      "loss": 0.4264,
       "step": 742
     },
     {
-      "epoch": 1.67,
-      "learning_rate": 0.0002585549132947977,
-      "loss": 1.0869,
+      "epoch": 3.33,
+      "learning_rate": 0.00018341463414634144,
+      "loss": 0.3209,
       "step": 743
     },
     {
-      "epoch": 1.67,
-      "learning_rate": 0.00025838150289017337,
-      "loss": 1.3723,
+      "epoch": 3.34,
+      "learning_rate": 0.00018292682926829266,
+      "loss": 0.3521,
       "step": 744
     },
     {
-      "epoch": 1.67,
-      "learning_rate": 0.0002582080924855491,
-      "loss": 0.9013,
+      "epoch": 3.34,
+      "learning_rate": 0.0001824390243902439,
+      "loss": 0.3439,
       "step": 745
     },
     {
-      "epoch": 1.67,
-      "learning_rate": 0.0002580346820809248,
-      "loss": 1.2461,
+      "epoch": 3.35,
+      "learning_rate": 0.0001819512195121951,
+      "loss": 0.3283,
       "step": 746
     },
     {
-      "epoch": 1.67,
-      "learning_rate": 0.00025786127167630056,
-      "loss": 6.9996,
+      "epoch": 3.35,
+      "learning_rate": 0.00018146341463414633,
+      "loss": 0.3709,
       "step": 747
     },
     {
-      "epoch": 1.68,
-      "learning_rate": 0.00025768786127167625,
-      "loss": 6.0548,
+      "epoch": 3.35,
+      "learning_rate": 0.00018097560975609755,
+      "loss": 0.3629,
       "step": 748
     },
     {
-      "epoch": 1.68,
-      "learning_rate": 0.000257514450867052,
-      "loss": 4.8795,
+      "epoch": 3.36,
+      "learning_rate": 0.00018048780487804877,
+      "loss": 0.3261,
       "step": 749
     },
     {
-      "epoch": 1.68,
-      "learning_rate": 0.00025734104046242775,
-      "loss": 3.2843,
+      "epoch": 3.36,
+      "learning_rate": 0.00017999999999999998,
+      "loss": 0.3054,
       "step": 750
     },
     {
-      "epoch": 1.68,
-      "learning_rate": 0.00025716763005780344,
-      "loss": 2.8835,
+      "epoch": 3.37,
+      "learning_rate": 0.0001795121951219512,
+      "loss": 0.3341,
       "step": 751
     },
     {
-      "epoch": 1.69,
-      "learning_rate": 0.00025699421965317914,
-      "loss": 2.3855,
+      "epoch": 3.37,
+      "learning_rate": 0.00017902439024390245,
+      "loss": 0.3435,
       "step": 752
     },
     {
-      "epoch": 1.69,
-      "learning_rate": 0.0002568208092485549,
-      "loss": 2.2957,
+      "epoch": 3.38,
+      "learning_rate": 0.00017853658536585363,
+      "loss": 0.3293,
       "step": 753
     },
     {
-      "epoch": 1.69,
-      "learning_rate": 0.00025664739884393063,
-      "loss": 2.001,
+      "epoch": 3.38,
+      "learning_rate": 0.00017804878048780485,
+      "loss": 0.3783,
       "step": 754
     },
     {
-      "epoch": 1.69,
-      "learning_rate": 0.0002564739884393063,
-      "loss": 1.824,
+      "epoch": 3.39,
+      "learning_rate": 0.0001775609756097561,
+      "loss": 0.3352,
       "step": 755
     },
     {
-      "epoch": 1.7,
-      "learning_rate": 0.00025630057803468207,
-      "loss": 1.9435,
+      "epoch": 3.39,
+      "learning_rate": 0.00017707317073170729,
+      "loss": 0.275,
       "step": 756
     },
     {
-      "epoch": 1.7,
-      "learning_rate": 0.0002561271676300578,
-      "loss": 1.5901,
+      "epoch": 3.39,
+      "learning_rate": 0.00017658536585365853,
+      "loss": 0.3101,
       "step": 757
     },
     {
-      "epoch": 1.7,
-      "learning_rate": 0.0002559537572254335,
-      "loss": 1.447,
+      "epoch": 3.4,
+      "learning_rate": 0.00017609756097560975,
+      "loss": 0.322,
       "step": 758
     },
     {
-      "epoch": 1.7,
-      "learning_rate": 0.0002557803468208092,
-      "loss": 1.4363,
+      "epoch": 3.4,
+      "learning_rate": 0.00017560975609756094,
+      "loss": 0.2583,
       "step": 759
     },
     {
-      "epoch": 1.7,
-      "learning_rate": 0.00025560693641618496,
-      "loss": 1.3356,
+      "epoch": 3.41,
+      "learning_rate": 0.00017512195121951218,
+      "loss": 0.2986,
       "step": 760
     },
     {
-      "epoch": 1.71,
-      "learning_rate": 0.0002554335260115607,
-      "loss": 1.3838,
+      "epoch": 3.41,
+      "learning_rate": 0.0001746341463414634,
+      "loss": 0.2965,
       "step": 761
     },
     {
-      "epoch": 1.71,
-      "learning_rate": 0.0002552601156069364,
-      "loss": 1.1423,
+      "epoch": 3.42,
+      "learning_rate": 0.00017414634146341461,
+      "loss": 0.3305,
       "step": 762
     },
     {
-      "epoch": 1.71,
-      "learning_rate": 0.0002550867052023121,
-      "loss": 1.2079,
+      "epoch": 3.42,
+      "learning_rate": 0.00017365853658536583,
+      "loss": 0.2637,
       "step": 763
     },
     {
-      "epoch": 1.71,
-      "learning_rate": 0.00025491329479768784,
-      "loss": 1.1538,
+      "epoch": 3.43,
+      "learning_rate": 0.00017317073170731708,
+      "loss": 0.2815,
       "step": 764
     },
     {
-      "epoch": 1.72,
-      "learning_rate": 0.0002547398843930636,
-      "loss": 1.0943,
+      "epoch": 3.43,
+      "learning_rate": 0.00017268292682926826,
+      "loss": 0.3337,
       "step": 765
     },
     {
-      "epoch": 1.72,
-      "learning_rate": 0.0002545664739884393,
-      "loss": 1.1982,
+      "epoch": 3.43,
+      "learning_rate": 0.00017219512195121948,
+      "loss": 0.3071,
       "step": 766
     },
     {
-      "epoch": 1.72,
-      "learning_rate": 0.00025439306358381503,
-      "loss": 1.1676,
+      "epoch": 3.44,
+      "learning_rate": 0.00017170731707317073,
+      "loss": 0.2576,
       "step": 767
     },
     {
-      "epoch": 1.72,
-      "learning_rate": 0.0002542196531791907,
-      "loss": 1.0995,
+      "epoch": 3.44,
+      "learning_rate": 0.00017121951219512192,
+      "loss": 0.2627,
       "step": 768
     },
     {
-      "epoch": 1.72,
-      "learning_rate": 0.00025404624277456647,
-      "loss": 0.9632,
+      "epoch": 3.45,
+      "learning_rate": 0.00017073170731707316,
+      "loss": 0.2915,
       "step": 769
     },
     {
-      "epoch": 1.73,
-      "learning_rate": 0.00025387283236994216,
-      "loss": 1.0703,
+      "epoch": 3.45,
+      "learning_rate": 0.00017024390243902438,
+      "loss": 1.9187,
       "step": 770
     },
     {
-      "epoch": 1.73,
-      "learning_rate": 0.0002536994219653179,
-      "loss": 0.9207,
+      "epoch": 3.46,
+      "learning_rate": 0.00016975609756097557,
+      "loss": 1.4796,
       "step": 771
     },
     {
-      "epoch": 1.73,
-      "learning_rate": 0.0002535260115606936,
-      "loss": 0.9262,
+      "epoch": 3.46,
+      "learning_rate": 0.0001692682926829268,
+      "loss": 0.9913,
       "step": 772
     },
     {
-      "epoch": 1.73,
-      "learning_rate": 0.00025335260115606935,
-      "loss": 0.9645,
+      "epoch": 3.47,
+      "learning_rate": 0.00016878048780487803,
+      "loss": 0.6812,
       "step": 773
     },
     {
-      "epoch": 1.74,
-      "learning_rate": 0.00025317919075144504,
-      "loss": 1.0928,
+      "epoch": 3.47,
+      "learning_rate": 0.00016829268292682927,
+      "loss": 0.6404,
       "step": 774
     },
     {
-      "epoch": 1.74,
-      "learning_rate": 0.0002530057803468208,
-      "loss": 0.9605,
+      "epoch": 3.48,
+      "learning_rate": 0.00016780487804878046,
+      "loss": 0.6294,
       "step": 775
     },
     {
-      "epoch": 1.74,
-      "learning_rate": 0.0002528323699421965,
-      "loss": 1.0233,
+      "epoch": 3.48,
+      "learning_rate": 0.0001673170731707317,
+      "loss": 0.5524,
       "step": 776
     },
     {
-      "epoch": 1.74,
-      "learning_rate": 0.00025265895953757223,
-      "loss": 0.9323,
+      "epoch": 3.48,
+      "learning_rate": 0.00016682926829268292,
+      "loss": 0.4834,
       "step": 777
     },
     {
-      "epoch": 1.74,
-      "learning_rate": 0.000252485549132948,
-      "loss": 0.8583,
+      "epoch": 3.49,
+      "learning_rate": 0.0001663414634146341,
+      "loss": 0.4867,
       "step": 778
     },
     {
-      "epoch": 1.75,
-      "learning_rate": 0.0002523121387283237,
-      "loss": 1.039,
+      "epoch": 3.49,
+      "learning_rate": 0.00016585365853658536,
+      "loss": 0.4394,
       "step": 779
     },
     {
-      "epoch": 1.75,
-      "learning_rate": 0.00025213872832369937,
-      "loss": 0.9294,
+      "epoch": 3.5,
+      "learning_rate": 0.00016536585365853657,
+      "loss": 0.4228,
       "step": 780
     },
     {
-      "epoch": 1.75,
-      "learning_rate": 0.0002519653179190751,
-      "loss": 0.9475,
+      "epoch": 3.5,
+      "learning_rate": 0.0001648780487804878,
+      "loss": 0.3963,
       "step": 781
     },
     {
-      "epoch": 1.75,
-      "learning_rate": 0.00025179190751445086,
-      "loss": 0.9068,
+      "epoch": 3.51,
+      "learning_rate": 0.000164390243902439,
+      "loss": 0.3693,
       "step": 782
     },
     {
-      "epoch": 1.76,
-      "learning_rate": 0.00025161849710982656,
-      "loss": 0.8889,
+      "epoch": 3.51,
+      "learning_rate": 0.00016390243902439025,
+      "loss": 0.3423,
       "step": 783
     },
     {
-      "epoch": 1.76,
-      "learning_rate": 0.0002514450867052023,
-      "loss": 0.8678,
+      "epoch": 3.52,
+      "learning_rate": 0.00016341463414634144,
+      "loss": 0.3681,
       "step": 784
     },
     {
-      "epoch": 1.76,
-      "learning_rate": 0.00025127167630057805,
-      "loss": 0.8449,
+      "epoch": 3.52,
+      "learning_rate": 0.00016292682926829266,
+      "loss": 0.3444,
       "step": 785
     },
     {
-      "epoch": 1.76,
-      "learning_rate": 0.00025109826589595375,
-      "loss": 0.885,
+      "epoch": 3.52,
+      "learning_rate": 0.0001624390243902439,
+      "loss": 0.3625,
       "step": 786
     },
     {
-      "epoch": 1.76,
-      "learning_rate": 0.00025092485549132944,
-      "loss": 0.7919,
+      "epoch": 3.53,
+      "learning_rate": 0.0001619512195121951,
+      "loss": 0.335,
       "step": 787
     },
     {
-      "epoch": 1.77,
-      "learning_rate": 0.0002507514450867052,
-      "loss": 0.9106,
+      "epoch": 3.53,
+      "learning_rate": 0.00016146341463414634,
+      "loss": 0.3001,
       "step": 788
     },
     {
-      "epoch": 1.77,
-      "learning_rate": 0.00025057803468208094,
-      "loss": 0.8597,
+      "epoch": 3.54,
+      "learning_rate": 0.00016097560975609755,
+      "loss": 0.3523,
       "step": 789
     },
     {
-      "epoch": 1.77,
-      "learning_rate": 0.00025040462427745663,
-      "loss": 0.8783,
+      "epoch": 3.54,
+      "learning_rate": 0.00016048780487804874,
+      "loss": 0.3084,
       "step": 790
     },
     {
-      "epoch": 1.77,
-      "learning_rate": 0.0002502312138728323,
-      "loss": 0.897,
+      "epoch": 3.55,
+      "learning_rate": 0.00015999999999999999,
+      "loss": 0.3502,
       "step": 791
     },
     {
-      "epoch": 1.78,
-      "learning_rate": 0.00025005780346820807,
-      "loss": 0.7942,
+      "epoch": 3.55,
+      "learning_rate": 0.0001595121951219512,
+      "loss": 0.3246,
       "step": 792
     },
     {
-      "epoch": 1.78,
-      "learning_rate": 0.00024988439306358376,
-      "loss": 0.8107,
+      "epoch": 3.56,
+      "learning_rate": 0.00015902439024390242,
+      "loss": 0.3061,
       "step": 793
     },
     {
-      "epoch": 1.78,
-      "learning_rate": 0.0002497109826589595,
-      "loss": 0.8347,
+      "epoch": 3.56,
+      "learning_rate": 0.00015853658536585364,
+      "loss": 0.3581,
       "step": 794
     },
     {
-      "epoch": 1.78,
-      "learning_rate": 0.00024953757225433526,
-      "loss": 1.0615,
+      "epoch": 3.57,
+      "learning_rate": 0.00015804878048780488,
+      "loss": 0.3109,
       "step": 795
     },
     {
-      "epoch": 1.78,
-      "learning_rate": 0.00024936416184971095,
-      "loss": 0.7615,
+      "epoch": 3.57,
+      "learning_rate": 0.00015756097560975607,
+      "loss": 0.2814,
       "step": 796
     },
     {
-      "epoch": 1.79,
-      "learning_rate": 0.00024919075144508665,
-      "loss": 6.4464,
+      "epoch": 3.57,
+      "learning_rate": 0.0001570731707317073,
+      "loss": 0.2509,
       "step": 797
     },
     {
-      "epoch": 1.79,
-      "learning_rate": 0.0002490173410404624,
-      "loss": 5.0751,
+      "epoch": 3.58,
+      "learning_rate": 0.00015658536585365853,
+      "loss": 0.3046,
       "step": 798
     },
     {
-      "epoch": 1.79,
-      "learning_rate": 0.00024884393063583814,
-      "loss": 3.7757,
+      "epoch": 3.58,
+      "learning_rate": 0.00015609756097560975,
+      "loss": 0.2995,
       "step": 799
     },
     {
-      "epoch": 1.79,
-      "learning_rate": 0.00024867052023121384,
-      "loss": 2.5764,
+      "epoch": 3.59,
+      "learning_rate": 0.00015560975609756097,
+      "loss": 0.2679,
       "step": 800
     },
     {
-      "epoch": 1.8,
-      "learning_rate": 0.0002484971098265896,
-      "loss": 2.1743,
+      "epoch": 3.59,
+      "learning_rate": 0.00015512195121951218,
+      "loss": 0.3044,
       "step": 801
     },
     {
-      "epoch": 1.8,
-      "learning_rate": 0.00024832369942196533,
-      "loss": 1.9198,
+      "epoch": 3.6,
+      "learning_rate": 0.00015463414634146343,
+      "loss": 0.286,
       "step": 802
     },
     {
-      "epoch": 1.8,
-      "learning_rate": 0.000248150289017341,
-      "loss": 1.7835,
+      "epoch": 3.6,
+      "learning_rate": 0.00015414634146341462,
+      "loss": 0.2883,
       "step": 803
     },
     {
-      "epoch": 1.8,
-      "learning_rate": 0.0002479768786127167,
-      "loss": 1.6071,
+      "epoch": 3.61,
+      "learning_rate": 0.00015365853658536583,
+      "loss": 0.3195,
       "step": 804
     },
     {
-      "epoch": 1.8,
-      "learning_rate": 0.00024780346820809247,
-      "loss": 1.511,
+      "epoch": 3.61,
+      "learning_rate": 0.00015317073170731708,
+      "loss": 0.284,
       "step": 805
     },
     {
-      "epoch": 1.81,
-      "learning_rate": 0.0002476300578034682,
-      "loss": 1.464,
+      "epoch": 3.61,
+      "learning_rate": 0.00015268292682926827,
+      "loss": 0.2843,
       "step": 806
     },
     {
-      "epoch": 1.81,
-      "learning_rate": 0.0002474566473988439,
-      "loss": 1.1126,
+      "epoch": 3.62,
+      "learning_rate": 0.0001521951219512195,
+      "loss": 0.2142,
       "step": 807
     },
     {
-      "epoch": 1.81,
-      "learning_rate": 0.0002472832369942196,
-      "loss": 1.0599,
+      "epoch": 3.62,
+      "learning_rate": 0.00015170731707317073,
+      "loss": 0.2466,
       "step": 808
     },
     {
-      "epoch": 1.81,
-      "learning_rate": 0.00024710982658959535,
-      "loss": 1.0413,
+      "epoch": 3.63,
+      "learning_rate": 0.00015121951219512192,
+      "loss": 0.2987,
       "step": 809
     },
     {
-      "epoch": 1.82,
-      "learning_rate": 0.0002469364161849711,
-      "loss": 1.0306,
+      "epoch": 3.63,
+      "learning_rate": 0.00015073170731707316,
+      "loss": 0.2849,
       "step": 810
     },
     {
-      "epoch": 1.82,
-      "learning_rate": 0.0002467630057803468,
-      "loss": 0.943,
+      "epoch": 3.64,
+      "learning_rate": 0.00015024390243902438,
+      "loss": 0.2537,
       "step": 811
     },
     {
-      "epoch": 1.82,
-      "learning_rate": 0.00024658959537572254,
-      "loss": 0.9814,
+      "epoch": 3.64,
+      "learning_rate": 0.0001497560975609756,
+      "loss": 0.2193,
       "step": 812
     },
     {
-      "epoch": 1.82,
-      "learning_rate": 0.00024641618497109823,
-      "loss": 0.8165,
+      "epoch": 3.65,
+      "learning_rate": 0.0001492682926829268,
+      "loss": 0.2516,
       "step": 813
     },
     {
-      "epoch": 1.83,
-      "learning_rate": 0.000246242774566474,
-      "loss": 0.8491,
+      "epoch": 3.65,
+      "learning_rate": 0.00014878048780487803,
+      "loss": 0.2595,
       "step": 814
     },
     {
-      "epoch": 1.83,
-      "learning_rate": 0.0002460693641618497,
-      "loss": 0.9818,
+      "epoch": 3.65,
+      "learning_rate": 0.00014829268292682925,
+      "loss": 0.2648,
       "step": 815
     },
     {
-      "epoch": 1.83,
-      "learning_rate": 0.0002458959537572254,
-      "loss": 0.8564,
+      "epoch": 3.66,
+      "learning_rate": 0.00014780487804878046,
+      "loss": 0.2563,
       "step": 816
     },
     {
-      "epoch": 1.83,
-      "learning_rate": 0.0002457225433526011,
-      "loss": 0.8471,
+      "epoch": 3.66,
+      "learning_rate": 0.00014731707317073168,
+      "loss": 0.2177,
       "step": 817
     },
     {
-      "epoch": 1.83,
-      "learning_rate": 0.00024554913294797686,
-      "loss": 0.7855,
+      "epoch": 3.67,
+      "learning_rate": 0.00014682926829268292,
+      "loss": 0.2533,
       "step": 818
     },
     {
-      "epoch": 1.84,
-      "learning_rate": 0.00024537572254335256,
-      "loss": 0.7987,
+      "epoch": 3.67,
+      "learning_rate": 0.00014634146341463414,
+      "loss": 0.292,
       "step": 819
     },
     {
-      "epoch": 1.84,
-      "learning_rate": 0.0002452023121387283,
-      "loss": 0.7538,
+      "epoch": 3.68,
+      "learning_rate": 0.00014585365853658536,
+      "loss": 1.3898,
       "step": 820
     },
     {
-      "epoch": 1.84,
-      "learning_rate": 0.000245028901734104,
-      "loss": 0.8812,
+      "epoch": 3.68,
+      "learning_rate": 0.00014536585365853657,
+      "loss": 1.0223,
       "step": 821
     },
     {
-      "epoch": 1.84,
-      "learning_rate": 0.00024485549132947975,
-      "loss": 0.8958,
+      "epoch": 3.69,
+      "learning_rate": 0.0001448780487804878,
+      "loss": 0.6619,
       "step": 822
     },
     {
-      "epoch": 1.85,
-      "learning_rate": 0.0002446820809248555,
-      "loss": 0.7152,
+      "epoch": 3.69,
+      "learning_rate": 0.000144390243902439,
+      "loss": 0.5068,
       "step": 823
     },
     {
-      "epoch": 1.85,
-      "learning_rate": 0.0002445086705202312,
-      "loss": 0.7971,
+      "epoch": 3.7,
+      "learning_rate": 0.00014390243902439023,
+      "loss": 0.473,
       "step": 824
     },
     {
-      "epoch": 1.85,
-      "learning_rate": 0.0002443352601156069,
-      "loss": 0.7203,
+      "epoch": 3.7,
+      "learning_rate": 0.00014341463414634144,
+      "loss": 0.4847,
       "step": 825
     },
     {
-      "epoch": 1.85,
-      "learning_rate": 0.00024416184971098263,
-      "loss": 0.7015,
+      "epoch": 3.7,
+      "learning_rate": 0.00014292682926829269,
+      "loss": 0.4595,
       "step": 826
     },
     {
-      "epoch": 1.85,
-      "learning_rate": 0.00024398843930635838,
-      "loss": 0.9212,
+      "epoch": 3.71,
+      "learning_rate": 0.00014243902439024388,
+      "loss": 0.3926,
       "step": 827
     },
     {
-      "epoch": 1.86,
-      "learning_rate": 0.00024381502890173407,
-      "loss": 0.6807,
+      "epoch": 3.71,
+      "learning_rate": 0.0001419512195121951,
+      "loss": 0.3979,
       "step": 828
     },
     {
-      "epoch": 1.86,
-      "learning_rate": 0.0002436416184971098,
-      "loss": 0.6554,
+      "epoch": 3.72,
+      "learning_rate": 0.00014146341463414634,
+      "loss": 0.3735,
       "step": 829
     },
     {
-      "epoch": 1.86,
-      "learning_rate": 0.00024346820809248554,
-      "loss": 0.7264,
+      "epoch": 3.72,
+      "learning_rate": 0.00014097560975609755,
+      "loss": 0.3384,
       "step": 830
     },
     {
-      "epoch": 1.86,
-      "learning_rate": 0.00024329479768786126,
-      "loss": 0.7385,
+      "epoch": 3.73,
+      "learning_rate": 0.00014048780487804877,
+      "loss": 0.3536,
       "step": 831
     },
     {
-      "epoch": 1.87,
-      "learning_rate": 0.00024312138728323698,
-      "loss": 0.757,
+      "epoch": 3.73,
+      "learning_rate": 0.00014,
+      "loss": 0.3151,
       "step": 832
     },
     {
-      "epoch": 1.87,
-      "learning_rate": 0.00024294797687861267,
-      "loss": 0.687,
+      "epoch": 3.74,
+      "learning_rate": 0.0001395121951219512,
+      "loss": 0.3147,
       "step": 833
     },
     {
-      "epoch": 1.87,
-      "learning_rate": 0.00024277456647398842,
-      "loss": 0.7073,
+      "epoch": 3.74,
+      "learning_rate": 0.00013902439024390242,
+      "loss": 0.3133,
       "step": 834
     },
     {
-      "epoch": 1.87,
-      "learning_rate": 0.00024260115606936414,
-      "loss": 0.674,
+      "epoch": 3.74,
+      "learning_rate": 0.00013853658536585364,
+      "loss": 0.255,
       "step": 835
     },
     {
-      "epoch": 1.87,
-      "learning_rate": 0.00024242774566473986,
-      "loss": 0.6726,
+      "epoch": 3.75,
+      "learning_rate": 0.00013804878048780486,
+      "loss": 0.2868,
       "step": 836
     },
     {
-      "epoch": 1.88,
-      "learning_rate": 0.00024225433526011558,
-      "loss": 0.7894,
+      "epoch": 3.75,
+      "learning_rate": 0.0001375609756097561,
+      "loss": 0.2783,
       "step": 837
     },
     {
-      "epoch": 1.88,
-      "learning_rate": 0.00024208092485549133,
-      "loss": 0.6616,
+      "epoch": 3.76,
+      "learning_rate": 0.00013707317073170732,
+      "loss": 0.2877,
       "step": 838
     },
     {
-      "epoch": 1.88,
-      "learning_rate": 0.00024190751445086702,
-      "loss": 0.7662,
+      "epoch": 3.76,
+      "learning_rate": 0.0001365853658536585,
+      "loss": 0.2775,
       "step": 839
     },
     {
-      "epoch": 1.88,
-      "learning_rate": 0.00024173410404624275,
-      "loss": 0.644,
+      "epoch": 3.77,
+      "learning_rate": 0.00013609756097560975,
+      "loss": 0.2804,
       "step": 840
     },
     {
-      "epoch": 1.89,
-      "learning_rate": 0.00024156069364161847,
-      "loss": 0.6078,
+      "epoch": 3.77,
+      "learning_rate": 0.00013560975609756097,
+      "loss": 0.2744,
       "step": 841
     },
     {
-      "epoch": 1.89,
-      "learning_rate": 0.0002413872832369942,
-      "loss": 0.6854,
+      "epoch": 3.78,
+      "learning_rate": 0.00013512195121951218,
+      "loss": 0.2385,
       "step": 842
     },
     {
-      "epoch": 1.89,
-      "learning_rate": 0.00024121387283236993,
-      "loss": 0.7161,
+      "epoch": 3.78,
+      "learning_rate": 0.0001346341463414634,
+      "loss": 0.2474,
       "step": 843
     },
     {
-      "epoch": 1.89,
-      "learning_rate": 0.00024104046242774563,
-      "loss": 0.7296,
+      "epoch": 3.78,
+      "learning_rate": 0.00013414634146341462,
+      "loss": 0.2529,
       "step": 844
     },
     {
-      "epoch": 1.89,
-      "learning_rate": 0.00024086705202312135,
-      "loss": 0.9602,
+      "epoch": 3.79,
+      "learning_rate": 0.00013365853658536586,
+      "loss": 0.2812,
       "step": 845
     },
     {
-      "epoch": 1.9,
-      "learning_rate": 0.0002406936416184971,
-      "loss": 0.9445,
+      "epoch": 3.79,
+      "learning_rate": 0.00013317073170731705,
+      "loss": 0.2556,
       "step": 846
     },
     {
-      "epoch": 1.9,
-      "learning_rate": 0.00024052023121387282,
-      "loss": 4.7021,
+      "epoch": 3.8,
+      "learning_rate": 0.00013268292682926827,
+      "loss": 0.2324,
       "step": 847
     },
     {
-      "epoch": 1.9,
-      "learning_rate": 0.00024034682080924854,
-      "loss": 3.3921,
+      "epoch": 3.8,
+      "learning_rate": 0.0001321951219512195,
+      "loss": 0.2562,
       "step": 848
     },
     {
-      "epoch": 1.9,
-      "learning_rate": 0.00024017341040462423,
-      "loss": 1.9051,
+      "epoch": 3.81,
+      "learning_rate": 0.00013170731707317073,
+      "loss": 0.2345,
       "step": 849
     },
     {
-      "epoch": 1.91,
-      "learning_rate": 0.00023999999999999998,
-      "loss": 1.4066,
+      "epoch": 3.81,
+      "learning_rate": 0.00013121951219512195,
+      "loss": 0.2582,
       "step": 850
     },
     {
-      "epoch": 1.91,
-      "learning_rate": 0.0002398265895953757,
-      "loss": 1.3678,
+      "epoch": 3.82,
+      "learning_rate": 0.00013073170731707316,
+      "loss": 0.2171,
       "step": 851
     },
     {
-      "epoch": 1.91,
-      "learning_rate": 0.00023965317919075142,
-      "loss": 1.3079,
+      "epoch": 3.82,
+      "learning_rate": 0.00013024390243902438,
+      "loss": 0.2607,
       "step": 852
     },
     {
-      "epoch": 1.91,
-      "learning_rate": 0.00023947976878612714,
-      "loss": 1.0537,
+      "epoch": 3.83,
+      "learning_rate": 0.0001297560975609756,
+      "loss": 0.227,
       "step": 853
     },
     {
-      "epoch": 1.91,
-      "learning_rate": 0.0002393063583815029,
-      "loss": 1.0467,
+      "epoch": 3.83,
+      "learning_rate": 0.00012926829268292681,
+      "loss": 0.2222,
       "step": 854
     },
     {
-      "epoch": 1.92,
-      "learning_rate": 0.0002391329479768786,
-      "loss": 0.9966,
+      "epoch": 3.83,
+      "learning_rate": 0.00012878048780487803,
+      "loss": 0.2358,
       "step": 855
     },
     {
-      "epoch": 1.92,
-      "learning_rate": 0.0002389595375722543,
-      "loss": 0.8728,
+      "epoch": 3.84,
+      "learning_rate": 0.00012829268292682925,
+      "loss": 0.235,
       "step": 856
     },
     {
-      "epoch": 1.92,
-      "learning_rate": 0.00023878612716763002,
-      "loss": 0.8287,
+      "epoch": 3.84,
+      "learning_rate": 0.0001278048780487805,
+      "loss": 0.2213,
       "step": 857
     },
     {
-      "epoch": 1.92,
-      "learning_rate": 0.00023861271676300577,
-      "loss": 0.891,
+      "epoch": 3.85,
+      "learning_rate": 0.00012731707317073168,
+      "loss": 0.2178,
       "step": 858
     },
     {
-      "epoch": 1.93,
-      "learning_rate": 0.0002384393063583815,
-      "loss": 0.6969,
+      "epoch": 3.85,
+      "learning_rate": 0.00012682926829268293,
+      "loss": 0.2105,
       "step": 859
     },
     {
-      "epoch": 1.93,
-      "learning_rate": 0.0002382658959537572,
-      "loss": 0.7537,
+      "epoch": 3.86,
+      "learning_rate": 0.00012634146341463414,
+      "loss": 0.2181,
       "step": 860
     },
     {
-      "epoch": 1.93,
-      "learning_rate": 0.0002380924855491329,
-      "loss": 0.7415,
+      "epoch": 3.86,
+      "learning_rate": 0.00012585365853658536,
+      "loss": 0.2648,
       "step": 861
     },
     {
-      "epoch": 1.93,
-      "learning_rate": 0.00023791907514450865,
-      "loss": 0.6404,
+      "epoch": 3.87,
+      "learning_rate": 0.00012536585365853658,
+      "loss": 0.2673,
       "step": 862
     },
     {
-      "epoch": 1.93,
-      "learning_rate": 0.00023774566473988437,
-      "loss": 0.6738,
+      "epoch": 3.87,
+      "learning_rate": 0.0001248780487804878,
+      "loss": 0.2179,
       "step": 863
     },
     {
-      "epoch": 1.94,
-      "learning_rate": 0.0002375722543352601,
-      "loss": 0.7438,
+      "epoch": 3.87,
+      "learning_rate": 0.000124390243902439,
+      "loss": 0.1762,
       "step": 864
     },
     {
-      "epoch": 1.94,
-      "learning_rate": 0.00023739884393063582,
-      "loss": 0.5957,
+      "epoch": 3.88,
+      "learning_rate": 0.00012390243902439023,
+      "loss": 0.2107,
       "step": 865
     },
     {
-      "epoch": 1.94,
-      "learning_rate": 0.00023722543352601156,
-      "loss": 0.5418,
+      "epoch": 3.88,
+      "learning_rate": 0.00012341463414634144,
+      "loss": 0.2415,
       "step": 866
     },
     {
-      "epoch": 1.94,
-      "learning_rate": 0.00023705202312138726,
-      "loss": 0.6583,
+      "epoch": 3.89,
+      "learning_rate": 0.00012292682926829266,
+      "loss": 0.265,
       "step": 867
     },
     {
-      "epoch": 1.95,
-      "learning_rate": 0.00023687861271676298,
-      "loss": 0.7395,
+      "epoch": 3.89,
+      "learning_rate": 0.0001224390243902439,
+      "loss": 0.2246,
       "step": 868
     },
     {
-      "epoch": 1.95,
-      "learning_rate": 0.0002367052023121387,
-      "loss": 0.6711,
+      "epoch": 3.9,
+      "learning_rate": 0.00012195121951219511,
+      "loss": 0.3635,
       "step": 869
     },
     {
-      "epoch": 1.95,
-      "learning_rate": 0.00023653179190751445,
-      "loss": 0.5824,
+      "epoch": 3.9,
+      "learning_rate": 0.00012146341463414634,
+      "loss": 0.7013,
       "step": 870
     },
     {
-      "epoch": 1.95,
-      "learning_rate": 0.00023635838150289017,
-      "loss": 0.6412,
+      "epoch": 3.91,
+      "learning_rate": 0.00012097560975609756,
+      "loss": 0.536,
       "step": 871
     },
     {
-      "epoch": 1.96,
-      "learning_rate": 0.00023618497109826586,
-      "loss": 0.5811,
+      "epoch": 3.91,
+      "learning_rate": 0.00012048780487804877,
+      "loss": 0.398,
       "step": 872
     },
     {
-      "epoch": 1.96,
-      "learning_rate": 0.00023601156069364158,
-      "loss": 0.611,
+      "epoch": 3.91,
+      "learning_rate": 0.00011999999999999999,
+      "loss": 0.3576,
       "step": 873
     },
     {
-      "epoch": 1.96,
-      "learning_rate": 0.00023583815028901733,
-      "loss": 0.6409,
+      "epoch": 3.92,
+      "learning_rate": 0.0001195121951219512,
+      "loss": 0.3146,
       "step": 874
     },
     {
-      "epoch": 1.96,
-      "learning_rate": 0.00023566473988439305,
-      "loss": 0.6914,
+      "epoch": 3.92,
+      "learning_rate": 0.00011902439024390242,
+      "loss": 0.3146,
       "step": 875
     },
     {
-      "epoch": 1.96,
-      "learning_rate": 0.00023549132947976877,
-      "loss": 0.5173,
+      "epoch": 3.93,
+      "learning_rate": 0.00011853658536585365,
+      "loss": 0.3472,
       "step": 876
     },
     {
-      "epoch": 1.97,
-      "learning_rate": 0.0002353179190751445,
-      "loss": 0.5725,
+      "epoch": 3.93,
+      "learning_rate": 0.00011804878048780487,
+      "loss": 0.2634,
       "step": 877
     },
     {
-      "epoch": 1.97,
-      "learning_rate": 0.00023514450867052024,
-      "loss": 0.5752,
+      "epoch": 3.94,
+      "learning_rate": 0.00011756097560975607,
+      "loss": 0.3482,
       "step": 878
     },
     {
-      "epoch": 1.97,
-      "learning_rate": 0.00023497109826589593,
-      "loss": 0.5661,
+      "epoch": 3.94,
+      "learning_rate": 0.0001170731707317073,
+      "loss": 0.2643,
       "step": 879
     },
     {
-      "epoch": 1.97,
-      "learning_rate": 0.00023479768786127165,
-      "loss": 0.51,
+      "epoch": 3.95,
+      "learning_rate": 0.00011658536585365852,
+      "loss": 0.2699,
       "step": 880
     },
     {
-      "epoch": 1.98,
-      "learning_rate": 0.00023462427745664737,
-      "loss": 0.5564,
+      "epoch": 3.95,
+      "learning_rate": 0.00011609756097560974,
+      "loss": 0.2764,
       "step": 881
     },
     {
-      "epoch": 1.98,
-      "learning_rate": 0.00023445086705202312,
-      "loss": 0.5911,
+      "epoch": 3.96,
+      "learning_rate": 0.00011560975609756097,
+      "loss": 0.2324,
       "step": 882
     },
     {
-      "epoch": 1.98,
-      "learning_rate": 0.00023427745664739884,
-      "loss": 0.5626,
+      "epoch": 3.96,
+      "learning_rate": 0.00011512195121951219,
+      "loss": 0.2683,
       "step": 883
     },
     {
-      "epoch": 1.98,
-      "learning_rate": 0.00023410404624277454,
-      "loss": 0.5812,
+      "epoch": 3.96,
+      "learning_rate": 0.00011463414634146342,
+      "loss": 0.2209,
       "step": 884
     },
     {
-      "epoch": 1.98,
-      "learning_rate": 0.00023393063583815026,
-      "loss": 0.6222,
+      "epoch": 3.97,
+      "learning_rate": 0.00011414634146341462,
+      "loss": 0.2287,
       "step": 885
     },
     {
-      "epoch": 1.99,
-      "learning_rate": 0.000233757225433526,
-      "loss": 0.7041,
+      "epoch": 3.97,
+      "learning_rate": 0.00011365853658536584,
+      "loss": 0.2358,
       "step": 886
     },
     {
-      "epoch": 1.99,
-      "learning_rate": 0.00023358381502890173,
-      "loss": 0.7997,
+      "epoch": 3.98,
+      "learning_rate": 0.00011317073170731707,
+      "loss": 0.2162,
       "step": 887
     },
     {
-      "epoch": 1.99,
-      "learning_rate": 0.00023341040462427745,
-      "loss": 0.6179,
+      "epoch": 3.98,
+      "learning_rate": 0.00011268292682926828,
+      "loss": 0.2329,
       "step": 888
     },
     {
-      "epoch": 1.99,
-      "learning_rate": 0.00023323699421965314,
-      "loss": 0.7034,
+      "epoch": 3.99,
+      "learning_rate": 0.0001121951219512195,
+      "loss": 0.2235,
       "step": 889
     },
     {
-      "epoch": 2.0,
-      "learning_rate": 0.00023306358381502886,
-      "loss": 0.7888,
+      "epoch": 3.99,
+      "learning_rate": 0.00011170731707317073,
+      "loss": 0.2409,
       "step": 890
     },
     {
-      "epoch": 2.0,
-      "learning_rate": 0.0002328901734104046,
-      "loss": 0.7726,
+      "epoch": 4.0,
+      "learning_rate": 0.00011121951219512193,
+      "loss": 0.1879,
       "step": 891
     },
     {
-      "epoch": 2.0,
-      "learning_rate": 0.00023271676300578033,
-      "loss": 0.7381,
+      "epoch": 4.0,
+      "learning_rate": 0.00011073170731707315,
+      "loss": 0.263,
       "step": 892
     },
     {
-      "epoch": 2.0,
-      "learning_rate": 0.00023254335260115605,
-      "loss": 3.1339,
+      "epoch": 4.0,
+      "learning_rate": 0.00011024390243902438,
+      "loss": 0.4579,
       "step": 893
     },
     {
-      "epoch": 2.0,
-      "learning_rate": 0.00023236994219653174,
-      "loss": 1.8651,
+      "epoch": 4.01,
+      "learning_rate": 0.0001097560975609756,
+      "loss": 0.4033,
       "step": 894
     },
     {
-      "epoch": 2.01,
-      "learning_rate": 0.0002321965317919075,
-      "loss": 1.2221,
+      "epoch": 4.01,
+      "learning_rate": 0.00010926829268292683,
+      "loss": 0.313,
       "step": 895
     },
     {
-      "epoch": 2.01,
-      "learning_rate": 0.0002320231213872832,
-      "loss": 0.9329,
+      "epoch": 4.02,
+      "learning_rate": 0.00010878048780487805,
+      "loss": 0.2394,
       "step": 896
     },
     {
-      "epoch": 2.01,
-      "learning_rate": 0.00023184971098265893,
-      "loss": 0.8674,
+      "epoch": 4.02,
+      "learning_rate": 0.00010829268292682925,
+      "loss": 0.2394,
       "step": 897
     },
     {
-      "epoch": 2.01,
-      "learning_rate": 0.00023167630057803465,
-      "loss": 0.8294,
+      "epoch": 4.03,
+      "learning_rate": 0.00010780487804878048,
+      "loss": 0.2715,
       "step": 898
     },
     {
-      "epoch": 2.02,
-      "learning_rate": 0.0002315028901734104,
-      "loss": 0.807,
+      "epoch": 4.03,
+      "learning_rate": 0.0001073170731707317,
+      "loss": 0.24,
       "step": 899
     },
     {
-      "epoch": 2.02,
-      "learning_rate": 0.00023132947976878612,
-      "loss": 0.6646,
+      "epoch": 4.04,
+      "learning_rate": 0.00010682926829268291,
+      "loss": 0.2086,
       "step": 900
     },
     {
-      "epoch": 2.02,
-      "learning_rate": 0.00023115606936416181,
-      "loss": 0.5948,
+      "epoch": 4.04,
+      "learning_rate": 0.00010634146341463414,
+      "loss": 0.2325,
       "step": 901
     },
     {
-      "epoch": 2.02,
-      "learning_rate": 0.00023098265895953754,
-      "loss": 0.7274,
+      "epoch": 4.04,
+      "learning_rate": 0.00010585365853658536,
+      "loss": 0.1837,
       "step": 902
     },
     {
-      "epoch": 2.02,
-      "learning_rate": 0.00023080924855491328,
-      "loss": 0.6304,
+      "epoch": 4.05,
+      "learning_rate": 0.00010536585365853656,
+      "loss": 0.1641,
       "step": 903
     },
     {
-      "epoch": 2.03,
-      "learning_rate": 0.000230635838150289,
-      "loss": 0.606,
+      "epoch": 4.05,
+      "learning_rate": 0.0001048780487804878,
+      "loss": 0.169,
       "step": 904
     },
     {
-      "epoch": 2.03,
-      "learning_rate": 0.00023046242774566472,
-      "loss": 0.5263,
+      "epoch": 4.06,
+      "learning_rate": 0.00010439024390243901,
+      "loss": 0.1444,
       "step": 905
     },
     {
-      "epoch": 2.03,
-      "learning_rate": 0.00023028901734104042,
-      "loss": 0.4577,
+      "epoch": 4.06,
+      "learning_rate": 0.00010390243902439023,
+      "loss": 0.1955,
       "step": 906
     },
     {
-      "epoch": 2.03,
-      "learning_rate": 0.00023011560693641617,
-      "loss": 0.4746,
+      "epoch": 4.07,
+      "learning_rate": 0.00010341463414634146,
+      "loss": 0.1712,
       "step": 907
     },
     {
-      "epoch": 2.04,
-      "learning_rate": 0.0002299421965317919,
-      "loss": 0.5186,
+      "epoch": 4.07,
+      "learning_rate": 0.00010292682926829268,
+      "loss": 0.1723,
       "step": 908
     },
     {
-      "epoch": 2.04,
-      "learning_rate": 0.0002297687861271676,
-      "loss": 0.4732,
+      "epoch": 4.08,
+      "learning_rate": 0.0001024390243902439,
+      "loss": 0.1577,
       "step": 909
     },
     {
-      "epoch": 2.04,
-      "learning_rate": 0.00022959537572254333,
-      "loss": 0.5486,
+      "epoch": 4.08,
+      "learning_rate": 0.00010195121951219511,
+      "loss": 0.1411,
       "step": 910
     },
     {
-      "epoch": 2.04,
-      "learning_rate": 0.00022942196531791908,
-      "loss": 0.5361,
+      "epoch": 4.09,
+      "learning_rate": 0.00010146341463414633,
+      "loss": 0.147,
       "step": 911
     },
     {
-      "epoch": 2.04,
-      "learning_rate": 0.00022924855491329477,
-      "loss": 0.3955,
+      "epoch": 4.09,
+      "learning_rate": 0.00010097560975609756,
+      "loss": 0.1688,
       "step": 912
     },
     {
-      "epoch": 2.05,
-      "learning_rate": 0.0002290751445086705,
-      "loss": 0.4232,
+      "epoch": 4.09,
+      "learning_rate": 0.00010048780487804877,
+      "loss": 0.1425,
       "step": 913
     },
     {
-      "epoch": 2.05,
-      "learning_rate": 0.0002289017341040462,
-      "loss": 0.4127,
+      "epoch": 4.1,
+      "learning_rate": 9.999999999999999e-05,
+      "loss": 0.184,
       "step": 914
     },
     {
-      "epoch": 2.05,
-      "learning_rate": 0.00022872832369942196,
-      "loss": 0.5143,
+      "epoch": 4.1,
+      "learning_rate": 9.951219512195122e-05,
+      "loss": 0.1604,
       "step": 915
     },
     {
-      "epoch": 2.05,
-      "learning_rate": 0.00022855491329479768,
-      "loss": 0.4155,
+      "epoch": 4.11,
+      "learning_rate": 9.902439024390243e-05,
+      "loss": 0.1326,
       "step": 916
     },
     {
-      "epoch": 2.06,
-      "learning_rate": 0.00022838150289017337,
-      "loss": 0.4304,
+      "epoch": 4.11,
+      "learning_rate": 9.853658536585364e-05,
+      "loss": 0.1505,
       "step": 917
     },
     {
-      "epoch": 2.06,
-      "learning_rate": 0.0002282080924855491,
-      "loss": 0.4103,
+      "epoch": 4.12,
+      "learning_rate": 9.804878048780487e-05,
+      "loss": 0.1165,
       "step": 918
     },
     {
-      "epoch": 2.06,
-      "learning_rate": 0.00022803468208092484,
-      "loss": 0.5457,
+      "epoch": 4.12,
+      "learning_rate": 9.756097560975609e-05,
+      "loss": 0.1217,
       "step": 919
     },
     {
-      "epoch": 2.06,
-      "learning_rate": 0.00022786127167630056,
-      "loss": 0.3894,
+      "epoch": 4.13,
+      "learning_rate": 9.70731707317073e-05,
+      "loss": 0.1376,
       "step": 920
     },
     {
-      "epoch": 2.07,
-      "learning_rate": 0.00022768786127167628,
-      "loss": 0.4171,
+      "epoch": 4.13,
+      "learning_rate": 9.658536585365854e-05,
+      "loss": 0.1416,
       "step": 921
     },
     {
-      "epoch": 2.07,
-      "learning_rate": 0.00022751445086705198,
-      "loss": 0.453,
+      "epoch": 4.13,
+      "learning_rate": 9.609756097560974e-05,
+      "loss": 0.1483,
       "step": 922
     },
     {
-      "epoch": 2.07,
-      "learning_rate": 0.00022734104046242772,
-      "loss": 0.4221,
+      "epoch": 4.14,
+      "learning_rate": 9.560975609756097e-05,
+      "loss": 0.1441,
       "step": 923
     },
     {
-      "epoch": 2.07,
-      "learning_rate": 0.00022716763005780344,
-      "loss": 0.3982,
+      "epoch": 4.14,
+      "learning_rate": 9.512195121951219e-05,
+      "loss": 0.1459,
       "step": 924
     },
     {
-      "epoch": 2.07,
-      "learning_rate": 0.00022699421965317917,
-      "loss": 0.4735,
+      "epoch": 4.15,
+      "learning_rate": 9.46341463414634e-05,
+      "loss": 0.138,
       "step": 925
     },
     {
-      "epoch": 2.08,
-      "learning_rate": 0.00022682080924855489,
-      "loss": 0.4474,
+      "epoch": 4.15,
+      "learning_rate": 9.414634146341463e-05,
+      "loss": 0.1196,
       "step": 926
     },
     {
-      "epoch": 2.08,
-      "learning_rate": 0.00022664739884393063,
-      "loss": 0.4077,
+      "epoch": 4.16,
+      "learning_rate": 9.365853658536585e-05,
+      "loss": 0.1039,
       "step": 927
     },
     {
-      "epoch": 2.08,
-      "learning_rate": 0.00022647398843930635,
-      "loss": 0.3909,
+      "epoch": 4.16,
+      "learning_rate": 9.317073170731706e-05,
+      "loss": 0.1461,
       "step": 928
     },
     {
-      "epoch": 2.08,
-      "learning_rate": 0.00022630057803468205,
-      "loss": 0.4458,
+      "epoch": 4.17,
+      "learning_rate": 9.268292682926829e-05,
+      "loss": 0.156,
       "step": 929
     },
     {
-      "epoch": 2.09,
-      "learning_rate": 0.00022612716763005777,
-      "loss": 0.4183,
+      "epoch": 4.17,
+      "learning_rate": 9.21951219512195e-05,
+      "loss": 0.1005,
       "step": 930
     },
     {
-      "epoch": 2.09,
-      "learning_rate": 0.00022595375722543352,
-      "loss": 0.4089,
+      "epoch": 4.17,
+      "learning_rate": 9.170731707317072e-05,
+      "loss": 0.1227,
       "step": 931
     },
     {
-      "epoch": 2.09,
-      "learning_rate": 0.00022578034682080924,
-      "loss": 0.4871,
+      "epoch": 4.18,
+      "learning_rate": 9.121951219512195e-05,
+      "loss": 0.1158,
       "step": 932
     },
     {
-      "epoch": 2.09,
-      "learning_rate": 0.00022560693641618496,
-      "loss": 0.4871,
+      "epoch": 4.18,
+      "learning_rate": 9.073170731707317e-05,
+      "loss": 0.1069,
       "step": 933
     },
     {
-      "epoch": 2.09,
-      "learning_rate": 0.00022543352601156065,
-      "loss": 0.5296,
+      "epoch": 4.19,
+      "learning_rate": 9.024390243902438e-05,
+      "loss": 0.1131,
       "step": 934
     },
     {
-      "epoch": 2.1,
-      "learning_rate": 0.0002252601156069364,
-      "loss": 0.3508,
+      "epoch": 4.19,
+      "learning_rate": 8.97560975609756e-05,
+      "loss": 0.0864,
       "step": 935
     },
     {
-      "epoch": 2.1,
-      "learning_rate": 0.00022508670520231212,
-      "loss": 0.3258,
+      "epoch": 4.2,
+      "learning_rate": 8.926829268292682e-05,
+      "loss": 0.0965,
       "step": 936
     },
     {
-      "epoch": 2.1,
-      "learning_rate": 0.00022491329479768784,
-      "loss": 0.3775,
+      "epoch": 4.2,
+      "learning_rate": 8.878048780487805e-05,
+      "loss": 0.0969,
       "step": 937
     },
     {
-      "epoch": 2.1,
-      "learning_rate": 0.00022473988439306356,
-      "loss": 0.3574,
+      "epoch": 4.21,
+      "learning_rate": 8.829268292682926e-05,
+      "loss": 0.1018,
       "step": 938
     },
     {
-      "epoch": 2.11,
-      "learning_rate": 0.0002245664739884393,
-      "loss": 0.4213,
+      "epoch": 4.21,
+      "learning_rate": 8.780487804878047e-05,
+      "loss": 0.0772,
       "step": 939
     },
     {
-      "epoch": 2.11,
-      "learning_rate": 0.000224393063583815,
-      "loss": 0.3603,
+      "epoch": 4.22,
+      "learning_rate": 8.73170731707317e-05,
+      "loss": 0.0811,
       "step": 940
     },
     {
-      "epoch": 2.11,
-      "learning_rate": 0.00022421965317919072,
-      "loss": 0.4983,
+      "epoch": 4.22,
+      "learning_rate": 8.682926829268292e-05,
+      "loss": 0.108,
       "step": 941
     },
     {
-      "epoch": 2.11,
-      "learning_rate": 0.00022404624277456644,
-      "loss": 0.2657,
+      "epoch": 4.22,
+      "learning_rate": 8.634146341463413e-05,
+      "loss": 0.1689,
       "step": 942
     },
     {
-      "epoch": 2.11,
-      "learning_rate": 0.0002238728323699422,
-      "loss": 1.5798,
+      "epoch": 4.23,
+      "learning_rate": 8.585365853658536e-05,
+      "loss": 0.4178,
       "step": 943
     },
     {
-      "epoch": 2.12,
-      "learning_rate": 0.0002236994219653179,
-      "loss": 1.1552,
+      "epoch": 4.23,
+      "learning_rate": 8.536585365853658e-05,
+      "loss": 0.3293,
       "step": 944
     },
     {
-      "epoch": 2.12,
-      "learning_rate": 0.0002235260115606936,
-      "loss": 0.9436,
+      "epoch": 4.24,
+      "learning_rate": 8.487804878048778e-05,
+      "loss": 0.3048,
       "step": 945
     },
     {
-      "epoch": 2.12,
-      "learning_rate": 0.00022335260115606933,
-      "loss": 0.7038,
+      "epoch": 4.24,
+      "learning_rate": 8.439024390243901e-05,
+      "loss": 0.2403,
       "step": 946
     },
     {
-      "epoch": 2.12,
-      "learning_rate": 0.00022317919075144507,
-      "loss": 0.6321,
+      "epoch": 4.25,
+      "learning_rate": 8.390243902439023e-05,
+      "loss": 0.2498,
       "step": 947
     },
     {
-      "epoch": 2.13,
-      "learning_rate": 0.0002230057803468208,
-      "loss": 0.656,
+      "epoch": 4.25,
+      "learning_rate": 8.341463414634146e-05,
+      "loss": 0.1854,
       "step": 948
     },
     {
-      "epoch": 2.13,
-      "learning_rate": 0.00022283236994219652,
-      "loss": 0.613,
+      "epoch": 4.26,
+      "learning_rate": 8.292682926829268e-05,
+      "loss": 0.1997,
       "step": 949
     },
     {
-      "epoch": 2.13,
-      "learning_rate": 0.00022265895953757224,
-      "loss": 0.4863,
+      "epoch": 4.26,
+      "learning_rate": 8.24390243902439e-05,
+      "loss": 0.2298,
       "step": 950
     },
     {
-      "epoch": 2.13,
-      "learning_rate": 0.00022248554913294798,
-      "loss": 0.6139,
+      "epoch": 4.26,
+      "learning_rate": 8.195121951219513e-05,
+      "loss": 0.1632,
       "step": 951
     },
     {
-      "epoch": 2.13,
-      "learning_rate": 0.00022231213872832368,
-      "loss": 0.5303,
+      "epoch": 4.27,
+      "learning_rate": 8.146341463414633e-05,
+      "loss": 0.1737,
       "step": 952
     },
     {
-      "epoch": 2.14,
-      "learning_rate": 0.0002221387283236994,
-      "loss": 0.4703,
+      "epoch": 4.27,
+      "learning_rate": 8.097560975609755e-05,
+      "loss": 0.1601,
       "step": 953
     },
     {
-      "epoch": 2.14,
-      "learning_rate": 0.00022196531791907512,
-      "loss": 0.5082,
+      "epoch": 4.28,
+      "learning_rate": 8.048780487804878e-05,
+      "loss": 0.1663,
       "step": 954
     },
     {
-      "epoch": 2.14,
-      "learning_rate": 0.00022179190751445087,
-      "loss": 0.3705,
+      "epoch": 4.28,
+      "learning_rate": 7.999999999999999e-05,
+      "loss": 0.187,
       "step": 955
     },
     {
-      "epoch": 2.14,
-      "learning_rate": 0.0002216184971098266,
-      "loss": 0.3861,
+      "epoch": 4.29,
+      "learning_rate": 7.951219512195121e-05,
+      "loss": 0.1445,
       "step": 956
     },
     {
-      "epoch": 2.15,
-      "learning_rate": 0.00022144508670520228,
-      "loss": 0.4343,
+      "epoch": 4.29,
+      "learning_rate": 7.902439024390244e-05,
+      "loss": 0.1499,
       "step": 957
     },
     {
-      "epoch": 2.15,
-      "learning_rate": 0.000221271676300578,
-      "loss": 0.3629,
+      "epoch": 4.3,
+      "learning_rate": 7.853658536585364e-05,
+      "loss": 0.1406,
       "step": 958
     },
     {
-      "epoch": 2.15,
-      "learning_rate": 0.00022109826589595375,
-      "loss": 0.3247,
+      "epoch": 4.3,
+      "learning_rate": 7.804878048780487e-05,
+      "loss": 0.1639,
       "step": 959
     },
     {
-      "epoch": 2.15,
-      "learning_rate": 0.00022092485549132947,
-      "loss": 0.3734,
+      "epoch": 4.3,
+      "learning_rate": 7.756097560975609e-05,
+      "loss": 0.1295,
       "step": 960
     },
     {
-      "epoch": 2.15,
-      "learning_rate": 0.0002207514450867052,
-      "loss": 0.4231,
+      "epoch": 4.31,
+      "learning_rate": 7.707317073170731e-05,
+      "loss": 0.1336,
       "step": 961
     },
     {
-      "epoch": 2.16,
-      "learning_rate": 0.00022057803468208088,
-      "loss": 0.369,
+      "epoch": 4.31,
+      "learning_rate": 7.658536585365854e-05,
+      "loss": 0.1529,
       "step": 962
     },
     {
-      "epoch": 2.16,
-      "learning_rate": 0.00022040462427745663,
-      "loss": 0.3606,
+      "epoch": 4.32,
+      "learning_rate": 7.609756097560976e-05,
+      "loss": 0.1371,
       "step": 963
     },
     {
-      "epoch": 2.16,
-      "learning_rate": 0.00022023121387283235,
-      "loss": 0.4322,
+      "epoch": 4.32,
+      "learning_rate": 7.560975609756096e-05,
+      "loss": 0.1307,
       "step": 964
     },
     {
-      "epoch": 2.16,
-      "learning_rate": 0.00022005780346820807,
-      "loss": 0.3928,
+      "epoch": 4.33,
+      "learning_rate": 7.512195121951219e-05,
+      "loss": 0.1316,
       "step": 965
     },
     {
-      "epoch": 2.17,
-      "learning_rate": 0.0002198843930635838,
-      "loss": 0.4072,
+      "epoch": 4.33,
+      "learning_rate": 7.46341463414634e-05,
+      "loss": 0.1358,
       "step": 966
     },
     {
-      "epoch": 2.17,
-      "learning_rate": 0.00021971098265895954,
-      "loss": 0.3673,
+      "epoch": 4.34,
+      "learning_rate": 7.414634146341462e-05,
+      "loss": 0.1215,
       "step": 967
     },
     {
-      "epoch": 2.17,
-      "learning_rate": 0.00021953757225433524,
-      "loss": 0.3822,
+      "epoch": 4.34,
+      "learning_rate": 7.365853658536584e-05,
+      "loss": 0.1351,
       "step": 968
     },
     {
-      "epoch": 2.17,
-      "learning_rate": 0.00021936416184971096,
-      "loss": 0.3869,
+      "epoch": 4.35,
+      "learning_rate": 7.317073170731707e-05,
+      "loss": 0.121,
       "step": 969
     },
     {
-      "epoch": 2.17,
-      "learning_rate": 0.00021919075144508668,
-      "loss": 0.3596,
+      "epoch": 4.35,
+      "learning_rate": 7.268292682926829e-05,
+      "loss": 0.1392,
       "step": 970
     },
     {
-      "epoch": 2.18,
-      "learning_rate": 0.00021901734104046243,
-      "loss": 0.4073,
+      "epoch": 4.35,
+      "learning_rate": 7.21951219512195e-05,
+      "loss": 0.106,
       "step": 971
     },
     {
-      "epoch": 2.18,
-      "learning_rate": 0.00021884393063583815,
-      "loss": 0.3531,
+      "epoch": 4.36,
+      "learning_rate": 7.170731707317072e-05,
+      "loss": 0.1224,
       "step": 972
     },
     {
-      "epoch": 2.18,
-      "learning_rate": 0.00021867052023121384,
-      "loss": 0.4388,
+      "epoch": 4.36,
+      "learning_rate": 7.121951219512194e-05,
+      "loss": 0.1107,
       "step": 973
     },
     {
-      "epoch": 2.18,
-      "learning_rate": 0.00021849710982658956,
-      "loss": 0.3399,
+      "epoch": 4.37,
+      "learning_rate": 7.073170731707317e-05,
+      "loss": 0.1096,
       "step": 974
     },
     {
-      "epoch": 2.19,
-      "learning_rate": 0.0002183236994219653,
-      "loss": 0.286,
+      "epoch": 4.37,
+      "learning_rate": 7.024390243902439e-05,
+      "loss": 0.1294,
       "step": 975
     },
     {
-      "epoch": 2.19,
-      "learning_rate": 0.00021815028901734103,
-      "loss": 0.3678,
+      "epoch": 4.38,
+      "learning_rate": 6.97560975609756e-05,
+      "loss": 0.1206,
       "step": 976
     },
     {
-      "epoch": 2.19,
-      "learning_rate": 0.00021797687861271675,
-      "loss": 0.402,
+      "epoch": 4.38,
+      "learning_rate": 6.926829268292682e-05,
+      "loss": 0.1198,
       "step": 977
     },
     {
-      "epoch": 2.19,
-      "learning_rate": 0.00021780346820809247,
-      "loss": 0.4259,
+      "epoch": 4.39,
+      "learning_rate": 6.878048780487805e-05,
+      "loss": 0.1112,
       "step": 978
     },
     {
-      "epoch": 2.2,
-      "learning_rate": 0.00021763005780346822,
-      "loss": 0.3776,
+      "epoch": 4.39,
+      "learning_rate": 6.829268292682925e-05,
+      "loss": 0.1085,
       "step": 979
     },
     {
-      "epoch": 2.2,
-      "learning_rate": 0.0002174566473988439,
-      "loss": 0.3641,
+      "epoch": 4.39,
+      "learning_rate": 6.780487804878048e-05,
+      "loss": 0.089,
       "step": 980
     },
     {
-      "epoch": 2.2,
-      "learning_rate": 0.00021728323699421963,
-      "loss": 0.3582,
+      "epoch": 4.4,
+      "learning_rate": 6.73170731707317e-05,
+      "loss": 0.0981,
       "step": 981
     },
     {
-      "epoch": 2.2,
-      "learning_rate": 0.00021710982658959535,
-      "loss": 0.4073,
+      "epoch": 4.4,
+      "learning_rate": 6.682926829268293e-05,
+      "loss": 0.1129,
       "step": 982
     },
     {
-      "epoch": 2.2,
-      "learning_rate": 0.0002169364161849711,
-      "loss": 0.4097,
+      "epoch": 4.41,
+      "learning_rate": 6.634146341463413e-05,
+      "loss": 0.1026,
       "step": 983
     },
     {
-      "epoch": 2.21,
-      "learning_rate": 0.00021676300578034682,
-      "loss": 0.3869,
+      "epoch": 4.41,
+      "learning_rate": 6.585365853658536e-05,
+      "loss": 0.0927,
       "step": 984
     },
     {
-      "epoch": 2.21,
-      "learning_rate": 0.00021658959537572251,
-      "loss": 0.3229,
+      "epoch": 4.42,
+      "learning_rate": 6.536585365853658e-05,
+      "loss": 0.0987,
       "step": 985
     },
     {
-      "epoch": 2.21,
-      "learning_rate": 0.00021641618497109824,
-      "loss": 0.304,
+      "epoch": 4.42,
+      "learning_rate": 6.48780487804878e-05,
+      "loss": 0.1299,
       "step": 986
     },
     {
-      "epoch": 2.21,
-      "learning_rate": 0.00021624277456647396,
-      "loss": 0.3122,
+      "epoch": 4.43,
+      "learning_rate": 6.439024390243902e-05,
+      "loss": 0.0781,
       "step": 987
     },
     {
-      "epoch": 2.22,
-      "learning_rate": 0.0002160693641618497,
-      "loss": 0.3282,
+      "epoch": 4.43,
+      "learning_rate": 6.390243902439025e-05,
+      "loss": 0.1048,
       "step": 988
     },
     {
-      "epoch": 2.22,
-      "learning_rate": 0.00021589595375722542,
-      "loss": 0.3119,
+      "epoch": 4.43,
+      "learning_rate": 6.341463414634146e-05,
+      "loss": 0.0572,
       "step": 989
     },
     {
-      "epoch": 2.22,
-      "learning_rate": 0.00021572254335260112,
-      "loss": 0.4257,
+      "epoch": 4.44,
+      "learning_rate": 6.292682926829268e-05,
+      "loss": 0.1033,
       "step": 990
     },
     {
-      "epoch": 2.22,
-      "learning_rate": 0.00021554913294797684,
-      "loss": 0.4143,
+      "epoch": 4.44,
+      "learning_rate": 6.24390243902439e-05,
+      "loss": 0.1029,
       "step": 991
     },
     {
-      "epoch": 2.22,
-      "learning_rate": 0.00021537572254335259,
-      "loss": 0.4064,
+      "epoch": 4.45,
+      "learning_rate": 6.195121951219511e-05,
+      "loss": 0.1846,
       "step": 992
     },
     {
-      "epoch": 2.23,
-      "learning_rate": 0.0002152023121387283,
-      "loss": 1.685,
+      "epoch": 4.45,
+      "learning_rate": 6.146341463414633e-05,
+      "loss": 0.3495,
       "step": 993
     },
     {
-      "epoch": 2.23,
-      "learning_rate": 0.00021502890173410403,
-      "loss": 1.1262,
+      "epoch": 4.46,
+      "learning_rate": 6.0975609756097554e-05,
+      "loss": 0.3143,
       "step": 994
     },
     {
-      "epoch": 2.23,
-      "learning_rate": 0.00021485549132947972,
-      "loss": 0.8902,
+      "epoch": 4.46,
+      "learning_rate": 6.048780487804878e-05,
+      "loss": 0.2743,
       "step": 995
     },
     {
-      "epoch": 2.23,
-      "learning_rate": 0.00021468208092485547,
-      "loss": 0.7312,
+      "epoch": 4.47,
+      "learning_rate": 5.9999999999999995e-05,
+      "loss": 0.1964,
       "step": 996
     },
     {
-      "epoch": 2.24,
-      "learning_rate": 0.0002145086705202312,
-      "loss": 0.5592,
+      "epoch": 4.47,
+      "learning_rate": 5.951219512195121e-05,
+      "loss": 0.2446,
       "step": 997
     },
     {
-      "epoch": 2.24,
-      "learning_rate": 0.0002143352601156069,
-      "loss": 0.6647,
+      "epoch": 4.48,
+      "learning_rate": 5.9024390243902435e-05,
+      "loss": 0.1735,
       "step": 998
     },
     {
-      "epoch": 2.24,
-      "learning_rate": 0.00021416184971098263,
-      "loss": 0.5828,
+      "epoch": 4.48,
+      "learning_rate": 5.853658536585365e-05,
+      "loss": 0.2045,
       "step": 999
     },
     {
-      "epoch": 2.24,
-      "learning_rate": 0.00021398843930635838,
-      "loss": 0.5486,
+      "epoch": 4.48,
+      "learning_rate": 5.804878048780487e-05,
+      "loss": 0.1399,
       "step": 1000
     },
     {
-      "epoch": 2.24,
-      "eval_loss": 0.5549545288085938,
-      "eval_runtime": 557.4926,
-      "eval_samples_per_second": 4.739,
-      "eval_steps_per_second": 0.594,
-      "eval_wer": 0.17532725109083697,
+      "epoch": 4.48,
+      "eval_loss": 0.36502909660339355,
+      "eval_runtime": 567.0865,
+      "eval_samples_per_second": 4.659,
+      "eval_steps_per_second": 0.584,
+      "eval_wer": 0.11207854026180088,
       "step": 1000
     },
     {
-      "epoch": 2.24,
-      "learning_rate": 0.0002138150289017341,
-      "loss": 0.5099,
+      "epoch": 4.49,
+      "learning_rate": 5.756097560975609e-05,
+      "loss": 0.1683,
       "step": 1001
     },
     {
-      "epoch": 2.25,
-      "learning_rate": 0.0002136416184971098,
-      "loss": 0.4194,
+      "epoch": 4.49,
+      "learning_rate": 5.707317073170731e-05,
+      "loss": 0.1527,
       "step": 1002
     },
     {
-      "epoch": 2.25,
-      "learning_rate": 0.00021346820809248551,
-      "loss": 0.4526,
+      "epoch": 4.5,
+      "learning_rate": 5.6585365853658533e-05,
+      "loss": 0.1248,
       "step": 1003
     },
     {
-      "epoch": 2.25,
-      "learning_rate": 0.00021329479768786126,
-      "loss": 0.499,
+      "epoch": 4.5,
+      "learning_rate": 5.609756097560975e-05,
+      "loss": 0.1299,
       "step": 1004
     },
     {
-      "epoch": 2.25,
-      "learning_rate": 0.00021312138728323698,
-      "loss": 0.4094,
+      "epoch": 4.51,
+      "learning_rate": 5.560975609756097e-05,
+      "loss": 0.1563,
       "step": 1005
     },
     {
-      "epoch": 2.26,
-      "learning_rate": 0.0002129479768786127,
-      "loss": 0.3473,
+      "epoch": 4.51,
+      "learning_rate": 5.512195121951219e-05,
+      "loss": 0.1239,
       "step": 1006
     },
     {
-      "epoch": 2.26,
-      "learning_rate": 0.0002127745664739884,
-      "loss": 0.4647,
+      "epoch": 4.52,
+      "learning_rate": 5.4634146341463415e-05,
+      "loss": 0.1481,
       "step": 1007
     },
     {
-      "epoch": 2.26,
-      "learning_rate": 0.00021260115606936414,
-      "loss": 0.3955,
+      "epoch": 4.52,
+      "learning_rate": 5.4146341463414625e-05,
+      "loss": 0.1316,
       "step": 1008
     },
     {
-      "epoch": 2.26,
-      "learning_rate": 0.00021242774566473987,
-      "loss": 0.4039,
+      "epoch": 4.52,
+      "learning_rate": 5.365853658536585e-05,
+      "loss": 0.1313,
       "step": 1009
     },
     {
-      "epoch": 2.26,
-      "learning_rate": 0.00021225433526011559,
-      "loss": 0.4174,
+      "epoch": 4.53,
+      "learning_rate": 5.317073170731707e-05,
+      "loss": 0.1185,
       "step": 1010
     },
     {
-      "epoch": 2.27,
-      "learning_rate": 0.0002120809248554913,
-      "loss": 0.3858,
+      "epoch": 4.53,
+      "learning_rate": 5.268292682926828e-05,
+      "loss": 0.1288,
       "step": 1011
     },
     {
-      "epoch": 2.27,
-      "learning_rate": 0.00021190751445086705,
-      "loss": 0.4267,
+      "epoch": 4.54,
+      "learning_rate": 5.2195121951219506e-05,
+      "loss": 0.1273,
       "step": 1012
     },
     {
-      "epoch": 2.27,
-      "learning_rate": 0.00021173410404624275,
-      "loss": 0.4297,
+      "epoch": 4.54,
+      "learning_rate": 5.170731707317073e-05,
+      "loss": 0.1418,
       "step": 1013
     },
     {
-      "epoch": 2.27,
-      "learning_rate": 0.00021156069364161847,
-      "loss": 0.4191,
+      "epoch": 4.55,
+      "learning_rate": 5.121951219512195e-05,
+      "loss": 0.1133,
       "step": 1014
     },
     {
-      "epoch": 2.28,
-      "learning_rate": 0.0002113872832369942,
-      "loss": 0.402,
+      "epoch": 4.55,
+      "learning_rate": 5.0731707317073163e-05,
+      "loss": 0.1178,
       "step": 1015
     },
     {
-      "epoch": 2.28,
-      "learning_rate": 0.00021121387283236994,
-      "loss": 0.3399,
+      "epoch": 4.56,
+      "learning_rate": 5.024390243902439e-05,
+      "loss": 0.097,
       "step": 1016
     },
     {
-      "epoch": 2.28,
-      "learning_rate": 0.00021104046242774566,
-      "loss": 0.3722,
+      "epoch": 4.56,
+      "learning_rate": 4.975609756097561e-05,
+      "loss": 0.1185,
       "step": 1017
     },
     {
-      "epoch": 2.28,
-      "learning_rate": 0.00021086705202312135,
-      "loss": 0.3687,
+      "epoch": 4.57,
+      "learning_rate": 4.926829268292682e-05,
+      "loss": 0.1349,
       "step": 1018
     },
     {
-      "epoch": 2.28,
-      "learning_rate": 0.00021069364161849707,
-      "loss": 0.4594,
+      "epoch": 4.57,
+      "learning_rate": 4.8780487804878045e-05,
+      "loss": 0.1152,
       "step": 1019
     },
     {
-      "epoch": 2.29,
-      "learning_rate": 0.00021052023121387282,
-      "loss": 0.3473,
+      "epoch": 4.57,
+      "learning_rate": 4.829268292682927e-05,
+      "loss": 0.1227,
       "step": 1020
     },
     {
-      "epoch": 2.29,
-      "learning_rate": 0.00021034682080924854,
-      "loss": 0.3437,
+      "epoch": 4.58,
+      "learning_rate": 4.7804878048780485e-05,
+      "loss": 0.1017,
       "step": 1021
     },
     {
-      "epoch": 2.29,
-      "learning_rate": 0.00021017341040462426,
-      "loss": 0.3998,
+      "epoch": 4.58,
+      "learning_rate": 4.73170731707317e-05,
+      "loss": 0.0982,
       "step": 1022
     },
     {
-      "epoch": 2.29,
-      "learning_rate": 0.00020999999999999998,
-      "loss": 0.3963,
+      "epoch": 4.59,
+      "learning_rate": 4.6829268292682926e-05,
+      "loss": 0.1289,
       "step": 1023
     },
     {
-      "epoch": 2.3,
-      "learning_rate": 0.00020982658959537573,
-      "loss": 0.3373,
+      "epoch": 4.59,
+      "learning_rate": 4.634146341463414e-05,
+      "loss": 0.0997,
       "step": 1024
     },
     {
-      "epoch": 2.3,
-      "learning_rate": 0.00020965317919075142,
-      "loss": 0.3339,
+      "epoch": 4.6,
+      "learning_rate": 4.585365853658536e-05,
+      "loss": 0.0879,
       "step": 1025
     },
     {
-      "epoch": 2.3,
-      "learning_rate": 0.00020947976878612714,
-      "loss": 0.3122,
+      "epoch": 4.6,
+      "learning_rate": 4.536585365853658e-05,
+      "loss": 0.0836,
       "step": 1026
     },
     {
-      "epoch": 2.3,
-      "learning_rate": 0.00020930635838150286,
-      "loss": 0.3959,
+      "epoch": 4.61,
+      "learning_rate": 4.48780487804878e-05,
+      "loss": 0.1062,
       "step": 1027
     },
     {
-      "epoch": 2.3,
-      "learning_rate": 0.0002091329479768786,
-      "loss": 0.38,
+      "epoch": 4.61,
+      "learning_rate": 4.4390243902439024e-05,
+      "loss": 0.0964,
       "step": 1028
     },
     {
-      "epoch": 2.31,
-      "learning_rate": 0.00020895953757225433,
-      "loss": 0.2978,
+      "epoch": 4.61,
+      "learning_rate": 4.3902439024390234e-05,
+      "loss": 0.0981,
       "step": 1029
     },
     {
-      "epoch": 2.31,
-      "learning_rate": 0.00020878612716763003,
-      "loss": 0.309,
+      "epoch": 4.62,
+      "learning_rate": 4.341463414634146e-05,
+      "loss": 0.068,
       "step": 1030
     },
     {
-      "epoch": 2.31,
-      "learning_rate": 0.00020861271676300575,
-      "loss": 0.3022,
+      "epoch": 4.62,
+      "learning_rate": 4.292682926829268e-05,
+      "loss": 0.1051,
       "step": 1031
     },
     {
-      "epoch": 2.31,
-      "learning_rate": 0.0002084393063583815,
-      "loss": 0.3544,
+      "epoch": 4.63,
+      "learning_rate": 4.243902439024389e-05,
+      "loss": 0.0968,
       "step": 1032
     },
     {
-      "epoch": 2.32,
-      "learning_rate": 0.00020826589595375722,
-      "loss": 0.3822,
+      "epoch": 4.63,
+      "learning_rate": 4.1951219512195115e-05,
+      "loss": 0.1041,
       "step": 1033
     },
     {
-      "epoch": 2.32,
-      "learning_rate": 0.00020809248554913294,
-      "loss": 0.3033,
+      "epoch": 4.64,
+      "learning_rate": 4.146341463414634e-05,
+      "loss": 0.0835,
       "step": 1034
     },
     {
-      "epoch": 2.32,
-      "learning_rate": 0.00020791907514450863,
-      "loss": 0.3179,
+      "epoch": 4.64,
+      "learning_rate": 4.097560975609756e-05,
+      "loss": 0.1127,
       "step": 1035
     },
     {
-      "epoch": 2.32,
-      "learning_rate": 0.00020774566473988438,
-      "loss": 0.3592,
+      "epoch": 4.65,
+      "learning_rate": 4.048780487804877e-05,
+      "loss": 0.1029,
       "step": 1036
     },
     {
-      "epoch": 2.33,
-      "learning_rate": 0.0002075722543352601,
-      "loss": 0.293,
+      "epoch": 4.65,
+      "learning_rate": 3.9999999999999996e-05,
+      "loss": 0.0855,
       "step": 1037
     },
     {
-      "epoch": 2.33,
-      "learning_rate": 0.00020739884393063582,
-      "loss": 0.4728,
+      "epoch": 4.65,
+      "learning_rate": 3.951219512195122e-05,
+      "loss": 0.0733,
       "step": 1038
     },
     {
-      "epoch": 2.33,
-      "learning_rate": 0.00020722543352601154,
-      "loss": 0.3278,
+      "epoch": 4.66,
+      "learning_rate": 3.902439024390244e-05,
+      "loss": 0.0936,
       "step": 1039
     },
     {
-      "epoch": 2.33,
-      "learning_rate": 0.0002070520231213873,
-      "loss": 0.3808,
+      "epoch": 4.66,
+      "learning_rate": 3.8536585365853654e-05,
+      "loss": 0.0868,
       "step": 1040
     },
     {
-      "epoch": 2.33,
-      "learning_rate": 0.00020687861271676298,
-      "loss": 0.3197,
+      "epoch": 4.67,
+      "learning_rate": 3.804878048780488e-05,
+      "loss": 0.094,
       "step": 1041
     },
     {
-      "epoch": 2.34,
-      "learning_rate": 0.0002067052023121387,
-      "loss": 0.3421,
+      "epoch": 4.67,
+      "learning_rate": 3.7560975609756095e-05,
+      "loss": 0.1837,
       "step": 1042
     },
     {
-      "epoch": 2.34,
-      "learning_rate": 0.00020653179190751442,
-      "loss": 0.9862,
+      "epoch": 4.68,
+      "learning_rate": 3.707317073170731e-05,
+      "loss": 0.3259,
       "step": 1043
     },
     {
-      "epoch": 2.34,
-      "learning_rate": 0.00020635838150289017,
-      "loss": 0.9364,
+      "epoch": 4.68,
+      "learning_rate": 3.6585365853658535e-05,
+      "loss": 0.2573,
       "step": 1044
     },
     {
-      "epoch": 2.34,
-      "learning_rate": 0.0002061849710982659,
-      "loss": 0.5846,
+      "epoch": 4.69,
+      "learning_rate": 3.609756097560975e-05,
+      "loss": 0.2468,
       "step": 1045
     },
     {
-      "epoch": 2.35,
-      "learning_rate": 0.00020601156069364158,
-      "loss": 0.5967,
+      "epoch": 4.69,
+      "learning_rate": 3.560975609756097e-05,
+      "loss": 0.2143,
       "step": 1046
     },
     {
-      "epoch": 2.35,
-      "learning_rate": 0.0002058381502890173,
-      "loss": 0.4982,
+      "epoch": 4.7,
+      "learning_rate": 3.512195121951219e-05,
+      "loss": 0.1886,
       "step": 1047
     },
     {
-      "epoch": 2.35,
-      "learning_rate": 0.00020566473988439305,
-      "loss": 0.4853,
+      "epoch": 4.7,
+      "learning_rate": 3.463414634146341e-05,
+      "loss": 0.2147,
       "step": 1048
     },
     {
-      "epoch": 2.35,
-      "learning_rate": 0.00020549132947976877,
-      "loss": 0.4355,
+      "epoch": 4.7,
+      "learning_rate": 3.4146341463414627e-05,
+      "loss": 0.182,
       "step": 1049
     },
     {
-      "epoch": 2.35,
-      "learning_rate": 0.0002053179190751445,
-      "loss": 0.5347,
+      "epoch": 4.71,
+      "learning_rate": 3.365853658536585e-05,
+      "loss": 0.1671,
       "step": 1050
     },
     {
-      "epoch": 2.36,
-      "learning_rate": 0.00020514450867052021,
-      "loss": 0.4829,
+      "epoch": 4.71,
+      "learning_rate": 3.317073170731707e-05,
+      "loss": 0.15,
       "step": 1051
     },
     {
-      "epoch": 2.36,
-      "learning_rate": 0.00020497109826589596,
-      "loss": 0.3672,
+      "epoch": 4.72,
+      "learning_rate": 3.268292682926829e-05,
+      "loss": 0.1564,
       "step": 1052
     },
     {
-      "epoch": 2.36,
-      "learning_rate": 0.00020479768786127166,
-      "loss": 0.4293,
+      "epoch": 4.72,
+      "learning_rate": 3.219512195121951e-05,
+      "loss": 0.1517,
       "step": 1053
     },
     {
-      "epoch": 2.36,
-      "learning_rate": 0.00020462427745664738,
-      "loss": 0.4222,
+      "epoch": 4.73,
+      "learning_rate": 3.170731707317073e-05,
+      "loss": 0.165,
       "step": 1054
     },
     {
-      "epoch": 2.37,
-      "learning_rate": 0.0002044508670520231,
-      "loss": 0.4015,
+      "epoch": 4.73,
+      "learning_rate": 3.121951219512195e-05,
+      "loss": 0.1202,
       "step": 1055
     },
     {
-      "epoch": 2.37,
-      "learning_rate": 0.00020427745664739885,
-      "loss": 0.4734,
+      "epoch": 4.74,
+      "learning_rate": 3.0731707317073165e-05,
+      "loss": 0.1199,
       "step": 1056
     },
     {
-      "epoch": 2.37,
-      "learning_rate": 0.00020410404624277457,
-      "loss": 0.3764,
+      "epoch": 4.74,
+      "learning_rate": 3.024390243902439e-05,
+      "loss": 0.1425,
       "step": 1057
     },
     {
-      "epoch": 2.37,
-      "learning_rate": 0.00020393063583815026,
-      "loss": 0.4224,
+      "epoch": 4.74,
+      "learning_rate": 2.9756097560975606e-05,
+      "loss": 0.1322,
       "step": 1058
     },
     {
-      "epoch": 2.37,
-      "learning_rate": 0.00020375722543352598,
-      "loss": 0.3608,
+      "epoch": 4.75,
+      "learning_rate": 2.9268292682926826e-05,
+      "loss": 0.1356,
       "step": 1059
     },
     {
-      "epoch": 2.38,
-      "learning_rate": 0.00020358381502890173,
-      "loss": 0.3688,
+      "epoch": 4.75,
+      "learning_rate": 2.8780487804878046e-05,
+      "loss": 0.1227,
       "step": 1060
     },
     {
-      "epoch": 2.38,
-      "learning_rate": 0.00020341040462427745,
-      "loss": 0.3761,
+      "epoch": 4.76,
+      "learning_rate": 2.8292682926829267e-05,
+      "loss": 0.1052,
       "step": 1061
     },
     {
-      "epoch": 2.38,
-      "learning_rate": 0.00020323699421965317,
-      "loss": 0.3932,
+      "epoch": 4.76,
+      "learning_rate": 2.7804878048780484e-05,
+      "loss": 0.1255,
       "step": 1062
     },
     {
-      "epoch": 2.38,
-      "learning_rate": 0.00020306358381502886,
-      "loss": 0.4005,
+      "epoch": 4.77,
+      "learning_rate": 2.7317073170731707e-05,
+      "loss": 0.0951,
       "step": 1063
     },
     {
-      "epoch": 2.39,
-      "learning_rate": 0.0002028901734104046,
-      "loss": 0.4029,
+      "epoch": 4.77,
+      "learning_rate": 2.6829268292682924e-05,
+      "loss": 0.1074,
       "step": 1064
     },
     {
-      "epoch": 2.39,
-      "learning_rate": 0.00020271676300578033,
-      "loss": 0.2571,
+      "epoch": 4.78,
+      "learning_rate": 2.634146341463414e-05,
+      "loss": 0.1143,
       "step": 1065
     },
     {
-      "epoch": 2.39,
-      "learning_rate": 0.00020254335260115605,
-      "loss": 0.2915,
+      "epoch": 4.78,
+      "learning_rate": 2.5853658536585365e-05,
+      "loss": 0.1112,
       "step": 1066
     },
     {
-      "epoch": 2.39,
-      "learning_rate": 0.00020236994219653177,
-      "loss": 0.3363,
+      "epoch": 4.78,
+      "learning_rate": 2.5365853658536582e-05,
+      "loss": 0.1014,
       "step": 1067
     },
     {
-      "epoch": 2.39,
-      "learning_rate": 0.00020219653179190752,
-      "loss": 0.3423,
+      "epoch": 4.79,
+      "learning_rate": 2.4878048780487805e-05,
+      "loss": 0.0987,
       "step": 1068
     },
     {
-      "epoch": 2.4,
-      "learning_rate": 0.00020202312138728321,
-      "loss": 0.3071,
+      "epoch": 4.79,
+      "learning_rate": 2.4390243902439022e-05,
+      "loss": 0.0907,
       "step": 1069
     },
     {
-      "epoch": 2.4,
-      "learning_rate": 0.00020184971098265893,
-      "loss": 0.3336,
+      "epoch": 4.8,
+      "learning_rate": 2.3902439024390243e-05,
+      "loss": 0.1075,
       "step": 1070
     },
     {
-      "epoch": 2.4,
-      "learning_rate": 0.00020167630057803466,
-      "loss": 0.3494,
+      "epoch": 4.8,
+      "learning_rate": 2.3414634146341463e-05,
+      "loss": 0.1121,
       "step": 1071
     },
     {
-      "epoch": 2.4,
-      "learning_rate": 0.0002015028901734104,
-      "loss": 0.3916,
+      "epoch": 4.81,
+      "learning_rate": 2.292682926829268e-05,
+      "loss": 0.0914,
       "step": 1072
     },
     {
-      "epoch": 2.41,
-      "learning_rate": 0.00020132947976878612,
-      "loss": 0.3232,
+      "epoch": 4.81,
+      "learning_rate": 2.24390243902439e-05,
+      "loss": 0.1099,
       "step": 1073
     },
     {
-      "epoch": 2.41,
-      "learning_rate": 0.00020115606936416184,
-      "loss": 0.325,
+      "epoch": 4.82,
+      "learning_rate": 2.1951219512195117e-05,
+      "loss": 0.0843,
       "step": 1074
     },
     {
-      "epoch": 2.41,
-      "learning_rate": 0.00020098265895953754,
-      "loss": 0.2942,
+      "epoch": 4.82,
+      "learning_rate": 2.146341463414634e-05,
+      "loss": 0.1039,
       "step": 1075
     },
     {
-      "epoch": 2.41,
-      "learning_rate": 0.00020080924855491329,
-      "loss": 0.3155,
+      "epoch": 4.83,
+      "learning_rate": 2.0975609756097558e-05,
+      "loss": 0.1008,
       "step": 1076
     },
     {
-      "epoch": 2.41,
-      "learning_rate": 0.000200635838150289,
-      "loss": 0.3347,
+      "epoch": 4.83,
+      "learning_rate": 2.048780487804878e-05,
+      "loss": 0.0954,
       "step": 1077
     },
     {
-      "epoch": 2.42,
-      "learning_rate": 0.00020046242774566473,
-      "loss": 0.3422,
+      "epoch": 4.83,
+      "learning_rate": 1.9999999999999998e-05,
+      "loss": 0.0986,
       "step": 1078
     },
     {
-      "epoch": 2.42,
-      "learning_rate": 0.00020028901734104045,
-      "loss": 0.3474,
+      "epoch": 4.84,
+      "learning_rate": 1.951219512195122e-05,
+      "loss": 0.0843,
       "step": 1079
     },
     {
-      "epoch": 2.42,
-      "learning_rate": 0.0002001156069364162,
-      "loss": 0.3095,
+      "epoch": 4.84,
+      "learning_rate": 1.902439024390244e-05,
+      "loss": 0.0825,
       "step": 1080
     },
     {
-      "epoch": 2.42,
-      "learning_rate": 0.0001999421965317919,
-      "loss": 0.2851,
+      "epoch": 4.85,
+      "learning_rate": 1.8536585365853656e-05,
+      "loss": 0.0804,
       "step": 1081
     },
     {
-      "epoch": 2.43,
-      "learning_rate": 0.0001997687861271676,
-      "loss": 0.3231,
+      "epoch": 4.85,
+      "learning_rate": 1.8048780487804876e-05,
+      "loss": 0.0896,
       "step": 1082
     },
     {
-      "epoch": 2.43,
-      "learning_rate": 0.00019959537572254333,
-      "loss": 0.329,
+      "epoch": 4.86,
+      "learning_rate": 1.7560975609756096e-05,
+      "loss": 0.0916,
       "step": 1083
     },
     {
-      "epoch": 2.43,
-      "learning_rate": 0.00019942196531791905,
-      "loss": 0.2926,
+      "epoch": 4.86,
+      "learning_rate": 1.7073170731707313e-05,
+      "loss": 0.0811,
       "step": 1084
     },
     {
-      "epoch": 2.43,
-      "learning_rate": 0.0001992485549132948,
-      "loss": 0.4081,
+      "epoch": 4.87,
+      "learning_rate": 1.6585365853658534e-05,
+      "loss": 0.0858,
       "step": 1085
     },
     {
-      "epoch": 2.43,
-      "learning_rate": 0.0001990751445086705,
-      "loss": 0.3161,
+      "epoch": 4.87,
+      "learning_rate": 1.6097560975609754e-05,
+      "loss": 0.0718,
       "step": 1086
     },
     {
-      "epoch": 2.44,
-      "learning_rate": 0.0001989017341040462,
-      "loss": 0.3599,
+      "epoch": 4.87,
+      "learning_rate": 1.5609756097560974e-05,
+      "loss": 0.0726,
       "step": 1087
     },
     {
-      "epoch": 2.44,
-      "learning_rate": 0.00019872832369942193,
-      "loss": 0.371,
+      "epoch": 4.88,
+      "learning_rate": 1.5121951219512194e-05,
+      "loss": 0.0752,
       "step": 1088
     },
     {
-      "epoch": 2.44,
-      "learning_rate": 0.00019855491329479768,
-      "loss": 0.4117,
+      "epoch": 4.88,
+      "learning_rate": 1.4634146341463413e-05,
+      "loss": 0.1133,
       "step": 1089
     },
     {
-      "epoch": 2.44,
-      "learning_rate": 0.0001983815028901734,
-      "loss": 0.2874,
+      "epoch": 4.89,
+      "learning_rate": 1.4146341463414633e-05,
+      "loss": 0.0928,
       "step": 1090
     },
     {
-      "epoch": 2.45,
-      "learning_rate": 0.0001982080924855491,
-      "loss": 0.2715,
+      "epoch": 4.89,
+      "learning_rate": 1.3658536585365854e-05,
+      "loss": 0.1092,
       "step": 1091
     },
     {
-      "epoch": 2.45,
-      "learning_rate": 0.00019803468208092482,
-      "loss": 0.4361,
+      "epoch": 4.9,
+      "learning_rate": 1.317073170731707e-05,
+      "loss": 0.1081,
       "step": 1092
     },
     {
-      "epoch": 2.45,
-      "learning_rate": 0.00019786127167630056,
-      "loss": 0.9283,
+      "epoch": 4.9,
+      "learning_rate": 1.2682926829268291e-05,
+      "loss": 0.2064,
       "step": 1093
     },
     {
-      "epoch": 2.45,
-      "learning_rate": 0.00019768786127167629,
-      "loss": 0.7911,
+      "epoch": 4.91,
+      "learning_rate": 1.2195121951219511e-05,
+      "loss": 0.1688,
       "step": 1094
     },
     {
-      "epoch": 2.46,
-      "learning_rate": 0.000197514450867052,
-      "loss": 0.6643,
+      "epoch": 4.91,
+      "learning_rate": 1.1707317073170731e-05,
+      "loss": 0.189,
       "step": 1095
     },
     {
-      "epoch": 2.46,
-      "learning_rate": 0.0001973410404624277,
-      "loss": 0.56,
+      "epoch": 4.91,
+      "learning_rate": 1.121951219512195e-05,
+      "loss": 0.1708,
       "step": 1096
     },
     {
-      "epoch": 2.46,
-      "learning_rate": 0.00019716763005780345,
-      "loss": 0.4258,
+      "epoch": 4.92,
+      "learning_rate": 1.073170731707317e-05,
+      "loss": 0.1589,
       "step": 1097
     },
     {
-      "epoch": 2.46,
-      "learning_rate": 0.00019699421965317917,
-      "loss": 0.4349,
+      "epoch": 4.92,
+      "learning_rate": 1.024390243902439e-05,
+      "loss": 0.1198,
       "step": 1098
     },
     {
-      "epoch": 2.46,
-      "learning_rate": 0.0001968208092485549,
-      "loss": 0.4175,
+      "epoch": 4.93,
+      "learning_rate": 9.75609756097561e-06,
+      "loss": 0.1327,
       "step": 1099
     },
     {
-      "epoch": 2.47,
-      "learning_rate": 0.0001966473988439306,
-      "loss": 0.453,
+      "epoch": 4.93,
+      "learning_rate": 9.268292682926828e-06,
+      "loss": 0.1172,
       "step": 1100
     },
     {
-      "epoch": 2.47,
-      "learning_rate": 0.00019647398843930636,
-      "loss": 0.4161,
+      "epoch": 4.94,
+      "learning_rate": 8.780487804878048e-06,
+      "loss": 0.1378,
       "step": 1101
     },
-    {
-      "epoch": 2.47,
-      "learning_rate": 0.00019630057803468208,
-      "loss": 0.4631,
-      "step": 1102
-    },
-    {
-      "epoch": 2.47,
-      "learning_rate": 0.00019612716763005777,
-      "loss": 0.3479,
-      "step": 1103
-    },
-    {
-      "epoch": 2.48,
-      "learning_rate": 0.0001959537572254335,
-      "loss": 0.3392,
-      "step": 1104
-    },
-    {
-      "epoch": 2.48,
-      "learning_rate": 0.00019578034682080924,
-      "loss": 0.3378,
-      "step": 1105
-    },
-    {
-      "epoch": 2.48,
-      "learning_rate": 0.00019560693641618496,
-      "loss": 0.3503,
-      "step": 1106
-    },
-    {
-      "epoch": 2.48,
-      "learning_rate": 0.00019543352601156068,
-      "loss": 0.3082,
-      "step": 1107
-    },
-    {
-      "epoch": 2.48,
-      "learning_rate": 0.00019526011560693637,
-      "loss": 0.3127,
-      "step": 1108
-    },
-    {
-      "epoch": 2.49,
-      "learning_rate": 0.00019508670520231212,
-      "loss": 0.3518,
-      "step": 1109
-    },
-    {
-      "epoch": 2.49,
-      "learning_rate": 0.00019491329479768784,
-      "loss": 0.2584,
-      "step": 1110
-    },
-    {
-      "epoch": 2.49,
-      "learning_rate": 0.00019473988439306356,
-      "loss": 0.2994,
-      "step": 1111
-    },
-    {
-      "epoch": 2.49,
-      "learning_rate": 0.00019456647398843928,
-      "loss": 0.3466,
-      "step": 1112
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 0.00019439306358381503,
-      "loss": 0.3337,
-      "step": 1113
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 0.00019421965317919073,
-      "loss": 0.3306,
-      "step": 1114
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 0.00019404624277456645,
-      "loss": 0.2953,
-      "step": 1115
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 0.00019387283236994217,
-      "loss": 0.3109,
-      "step": 1116
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 0.00019369942196531792,
-      "loss": 0.3141,
-      "step": 1117
-    },
-    {
-      "epoch": 2.51,
-      "learning_rate": 0.00019352601156069364,
-      "loss": 0.2826,
-      "step": 1118
-    },
-    {
-      "epoch": 2.51,
-      "learning_rate": 0.00019335260115606933,
-      "loss": 0.2811,
-      "step": 1119
-    },
-    {
-      "epoch": 2.51,
-      "learning_rate": 0.00019317919075144505,
-      "loss": 0.308,
-      "step": 1120
-    },
-    {
-      "epoch": 2.51,
-      "learning_rate": 0.0001930057803468208,
-      "loss": 0.3758,
-      "step": 1121
-    },
-    {
-      "epoch": 2.52,
-      "learning_rate": 0.00019283236994219652,
-      "loss": 0.3172,
-      "step": 1122
-    },
-    {
-      "epoch": 2.52,
-      "learning_rate": 0.00019265895953757224,
-      "loss": 0.3789,
-      "step": 1123
-    },
-    {
-      "epoch": 2.52,
-      "learning_rate": 0.00019248554913294796,
-      "loss": 0.326,
-      "step": 1124
-    },
-    {
-      "epoch": 2.52,
-      "learning_rate": 0.0001923121387283237,
-      "loss": 0.3814,
-      "step": 1125
-    },
-    {
-      "epoch": 2.52,
-      "learning_rate": 0.0001921387283236994,
-      "loss": 0.3633,
-      "step": 1126
-    },
-    {
-      "epoch": 2.53,
-      "learning_rate": 0.00019196531791907512,
-      "loss": 0.33,
-      "step": 1127
-    },
-    {
-      "epoch": 2.53,
-      "learning_rate": 0.00019179190751445084,
-      "loss": 0.3174,
-      "step": 1128
-    },
-    {
-      "epoch": 2.53,
-      "learning_rate": 0.0001916184971098266,
-      "loss": 0.3249,
-      "step": 1129
-    },
-    {
-      "epoch": 2.53,
-      "learning_rate": 0.0001914450867052023,
-      "loss": 0.2607,
-      "step": 1130
-    },
-    {
-      "epoch": 2.54,
-      "learning_rate": 0.000191271676300578,
-      "loss": 0.405,
-      "step": 1131
-    },
-    {
-      "epoch": 2.54,
-      "learning_rate": 0.00019109826589595373,
-      "loss": 0.3608,
-      "step": 1132
-    },
-    {
-      "epoch": 2.54,
-      "learning_rate": 0.00019092485549132947,
-      "loss": 0.3364,
-      "step": 1133
-    },
-    {
-      "epoch": 2.54,
-      "learning_rate": 0.0001907514450867052,
-      "loss": 0.3467,
-      "step": 1134
-    },
-    {
-      "epoch": 2.54,
-      "learning_rate": 0.00019057803468208091,
-      "loss": 0.2971,
-      "step": 1135
-    },
-    {
-      "epoch": 2.55,
-      "learning_rate": 0.0001904046242774566,
-      "loss": 0.2842,
-      "step": 1136
-    },
-    {
-      "epoch": 2.55,
-      "learning_rate": 0.00019023121387283236,
-      "loss": 0.3429,
-      "step": 1137
-    },
-    {
-      "epoch": 2.55,
-      "learning_rate": 0.00019005780346820808,
-      "loss": 0.3082,
-      "step": 1138
-    },
-    {
-      "epoch": 2.55,
-      "learning_rate": 0.0001898843930635838,
-      "loss": 0.4151,
-      "step": 1139
-    },
-    {
-      "epoch": 2.56,
-      "learning_rate": 0.00018971098265895952,
-      "loss": 0.3784,
-      "step": 1140
-    },
-    {
-      "epoch": 2.56,
-      "learning_rate": 0.00018953757225433527,
-      "loss": 0.2453,
-      "step": 1141
-    },
-    {
-      "epoch": 2.56,
-      "learning_rate": 0.00018936416184971096,
-      "loss": 0.4695,
-      "step": 1142
-    },
-    {
-      "epoch": 2.56,
-      "learning_rate": 0.00018919075144508668,
-      "loss": 0.7567,
-      "step": 1143
-    },
-    {
-      "epoch": 2.57,
-      "learning_rate": 0.0001890173410404624,
-      "loss": 0.6189,
-      "step": 1144
-    },
-    {
-      "epoch": 2.57,
-      "learning_rate": 0.00018884393063583815,
-      "loss": 0.504,
-      "step": 1145
-    },
-    {
-      "epoch": 2.57,
-      "learning_rate": 0.00018867052023121387,
-      "loss": 0.4176,
-      "step": 1146
-    },
-    {
-      "epoch": 2.57,
-      "learning_rate": 0.0001884971098265896,
-      "loss": 0.4394,
-      "step": 1147
-    },
-    {
-      "epoch": 2.57,
-      "learning_rate": 0.00018832369942196528,
-      "loss": 0.3826,
-      "step": 1148
-    },
-    {
-      "epoch": 2.58,
-      "learning_rate": 0.00018815028901734103,
-      "loss": 0.4139,
-      "step": 1149
-    },
-    {
-      "epoch": 2.58,
-      "learning_rate": 0.00018797687861271675,
-      "loss": 0.4348,
-      "step": 1150
-    },
-    {
-      "epoch": 2.58,
-      "learning_rate": 0.00018780346820809247,
-      "loss": 0.4796,
-      "step": 1151
-    },
-    {
-      "epoch": 2.58,
-      "learning_rate": 0.0001876300578034682,
-      "loss": 0.4092,
-      "step": 1152
-    },
-    {
-      "epoch": 2.59,
-      "learning_rate": 0.00018745664739884394,
-      "loss": 0.3293,
-      "step": 1153
-    },
-    {
-      "epoch": 2.59,
-      "learning_rate": 0.00018728323699421963,
-      "loss": 0.3758,
-      "step": 1154
-    },
-    {
-      "epoch": 2.59,
-      "learning_rate": 0.00018710982658959536,
-      "loss": 0.3988,
-      "step": 1155
-    },
-    {
-      "epoch": 2.59,
-      "learning_rate": 0.00018693641618497108,
-      "loss": 0.3302,
-      "step": 1156
-    },
-    {
-      "epoch": 2.59,
-      "learning_rate": 0.00018676300578034682,
-      "loss": 0.4761,
-      "step": 1157
-    },
-    {
-      "epoch": 2.6,
-      "learning_rate": 0.00018658959537572254,
-      "loss": 0.4218,
-      "step": 1158
-    },
-    {
-      "epoch": 2.6,
-      "learning_rate": 0.00018641618497109824,
-      "loss": 0.2673,
-      "step": 1159
-    },
-    {
-      "epoch": 2.6,
-      "learning_rate": 0.00018624277456647396,
-      "loss": 0.3068,
-      "step": 1160
-    },
-    {
-      "epoch": 2.6,
-      "learning_rate": 0.0001860693641618497,
-      "loss": 0.3173,
-      "step": 1161
-    },
-    {
-      "epoch": 2.61,
-      "learning_rate": 0.00018589595375722543,
-      "loss": 0.2941,
-      "step": 1162
-    },
-    {
-      "epoch": 2.61,
-      "learning_rate": 0.00018572254335260115,
-      "loss": 0.3206,
-      "step": 1163
-    },
-    {
-      "epoch": 2.61,
-      "learning_rate": 0.00018554913294797684,
-      "loss": 0.4003,
-      "step": 1164
-    },
-    {
-      "epoch": 2.61,
-      "learning_rate": 0.0001853757225433526,
-      "loss": 0.302,
-      "step": 1165
-    },
-    {
-      "epoch": 2.61,
-      "learning_rate": 0.0001852023121387283,
-      "loss": 0.3248,
-      "step": 1166
-    },
-    {
-      "epoch": 2.62,
-      "learning_rate": 0.00018502890173410403,
-      "loss": 0.334,
-      "step": 1167
-    },
-    {
-      "epoch": 2.62,
-      "learning_rate": 0.00018485549132947975,
-      "loss": 0.2525,
-      "step": 1168
-    },
-    {
-      "epoch": 2.62,
-      "learning_rate": 0.0001846820809248555,
-      "loss": 0.3136,
-      "step": 1169
-    },
-    {
-      "epoch": 2.62,
-      "learning_rate": 0.0001845086705202312,
-      "loss": 0.2709,
-      "step": 1170
-    },
-    {
-      "epoch": 2.63,
-      "learning_rate": 0.0001843352601156069,
-      "loss": 0.2941,
-      "step": 1171
-    },
-    {
-      "epoch": 2.63,
-      "learning_rate": 0.00018416184971098263,
-      "loss": 0.301,
-      "step": 1172
-    },
-    {
-      "epoch": 2.63,
-      "learning_rate": 0.00018398843930635838,
-      "loss": 0.3081,
-      "step": 1173
-    },
-    {
-      "epoch": 2.63,
-      "learning_rate": 0.0001838150289017341,
-      "loss": 0.2646,
-      "step": 1174
-    },
-    {
-      "epoch": 2.63,
-      "learning_rate": 0.00018364161849710982,
-      "loss": 0.3387,
-      "step": 1175
-    },
-    {
-      "epoch": 2.64,
-      "learning_rate": 0.00018346820809248552,
-      "loss": 0.3115,
-      "step": 1176
-    },
-    {
-      "epoch": 2.64,
-      "learning_rate": 0.00018329479768786124,
-      "loss": 0.3095,
-      "step": 1177
-    },
-    {
-      "epoch": 2.64,
-      "learning_rate": 0.00018312138728323698,
-      "loss": 0.2983,
-      "step": 1178
-    },
-    {
-      "epoch": 2.64,
-      "learning_rate": 0.0001829479768786127,
-      "loss": 0.3246,
-      "step": 1179
-    },
-    {
-      "epoch": 2.65,
-      "learning_rate": 0.00018277456647398843,
-      "loss": 0.3024,
-      "step": 1180
-    },
-    {
-      "epoch": 2.65,
-      "learning_rate": 0.00018260115606936412,
-      "loss": 0.2837,
-      "step": 1181
-    },
-    {
-      "epoch": 2.65,
-      "learning_rate": 0.00018242774566473987,
-      "loss": 0.2031,
-      "step": 1182
-    },
-    {
-      "epoch": 2.65,
-      "learning_rate": 0.0001822543352601156,
-      "loss": 0.2731,
-      "step": 1183
-    },
-    {
-      "epoch": 2.65,
-      "learning_rate": 0.0001820809248554913,
-      "loss": 0.4192,
-      "step": 1184
-    },
-    {
-      "epoch": 2.66,
-      "learning_rate": 0.00018190751445086703,
-      "loss": 0.4043,
-      "step": 1185
-    },
-    {
-      "epoch": 2.66,
-      "learning_rate": 0.00018173410404624278,
-      "loss": 0.3666,
-      "step": 1186
-    },
-    {
-      "epoch": 2.66,
-      "learning_rate": 0.00018156069364161847,
-      "loss": 0.3159,
-      "step": 1187
-    },
-    {
-      "epoch": 2.66,
-      "learning_rate": 0.0001813872832369942,
-      "loss": 0.2942,
-      "step": 1188
-    },
-    {
-      "epoch": 2.67,
-      "learning_rate": 0.0001812138728323699,
-      "loss": 0.2372,
-      "step": 1189
-    },
-    {
-      "epoch": 2.67,
-      "learning_rate": 0.00018104046242774566,
-      "loss": 0.3847,
-      "step": 1190
-    },
-    {
-      "epoch": 2.67,
-      "learning_rate": 0.00018086705202312138,
-      "loss": 0.3073,
-      "step": 1191
-    },
-    {
-      "epoch": 2.67,
-      "learning_rate": 0.00018069364161849707,
-      "loss": 0.3426,
-      "step": 1192
-    },
-    {
-      "epoch": 2.67,
-      "learning_rate": 0.0001805202312138728,
-      "loss": 0.8319,
-      "step": 1193
-    },
-    {
-      "epoch": 2.68,
-      "learning_rate": 0.00018034682080924854,
-      "loss": 0.6785,
-      "step": 1194
-    },
-    {
-      "epoch": 2.68,
-      "learning_rate": 0.00018017341040462426,
-      "loss": 0.5124,
-      "step": 1195
-    },
-    {
-      "epoch": 2.68,
-      "learning_rate": 0.00017999999999999998,
-      "loss": 0.4609,
-      "step": 1196
-    },
-    {
-      "epoch": 2.68,
-      "learning_rate": 0.0001798265895953757,
-      "loss": 0.4206,
-      "step": 1197
-    },
-    {
-      "epoch": 2.69,
-      "learning_rate": 0.00017965317919075145,
-      "loss": 0.4402,
-      "step": 1198
-    },
-    {
-      "epoch": 2.69,
-      "learning_rate": 0.00017947976878612715,
-      "loss": 0.3913,
-      "step": 1199
-    },
-    {
-      "epoch": 2.69,
-      "learning_rate": 0.00017930635838150287,
-      "loss": 0.3591,
-      "step": 1200
-    },
-    {
-      "epoch": 2.69,
-      "learning_rate": 0.0001791329479768786,
-      "loss": 0.3962,
-      "step": 1201
-    },
-    {
-      "epoch": 2.7,
-      "learning_rate": 0.00017895953757225434,
-      "loss": 0.3368,
-      "step": 1202
-    },
-    {
-      "epoch": 2.7,
-      "learning_rate": 0.00017878612716763006,
-      "loss": 0.339,
-      "step": 1203
-    },
-    {
-      "epoch": 2.7,
-      "learning_rate": 0.00017861271676300575,
-      "loss": 0.2718,
-      "step": 1204
-    },
-    {
-      "epoch": 2.7,
-      "learning_rate": 0.00017843930635838147,
-      "loss": 0.2501,
-      "step": 1205
-    },
-    {
-      "epoch": 2.7,
-      "learning_rate": 0.00017826589595375722,
-      "loss": 0.3284,
-      "step": 1206
-    },
-    {
-      "epoch": 2.71,
-      "learning_rate": 0.00017809248554913294,
-      "loss": 0.2951,
-      "step": 1207
-    },
-    {
-      "epoch": 2.71,
-      "learning_rate": 0.00017791907514450866,
-      "loss": 0.2785,
-      "step": 1208
-    },
-    {
-      "epoch": 2.71,
-      "learning_rate": 0.00017774566473988435,
-      "loss": 0.3452,
-      "step": 1209
-    },
-    {
-      "epoch": 2.71,
-      "learning_rate": 0.0001775722543352601,
-      "loss": 0.3002,
-      "step": 1210
-    },
-    {
-      "epoch": 2.72,
-      "learning_rate": 0.00017739884393063582,
-      "loss": 0.3082,
-      "step": 1211
-    },
-    {
-      "epoch": 2.72,
-      "learning_rate": 0.00017722543352601154,
-      "loss": 0.2915,
-      "step": 1212
-    },
-    {
-      "epoch": 2.72,
-      "learning_rate": 0.00017705202312138726,
-      "loss": 0.3227,
-      "step": 1213
-    },
-    {
-      "epoch": 2.72,
-      "learning_rate": 0.000176878612716763,
-      "loss": 0.2923,
-      "step": 1214
-    },
-    {
-      "epoch": 2.72,
-      "learning_rate": 0.0001767052023121387,
-      "loss": 0.2174,
-      "step": 1215
-    },
-    {
-      "epoch": 2.73,
-      "learning_rate": 0.00017653179190751442,
-      "loss": 0.3016,
-      "step": 1216
-    },
-    {
-      "epoch": 2.73,
-      "learning_rate": 0.00017635838150289015,
-      "loss": 0.2278,
-      "step": 1217
-    },
-    {
-      "epoch": 2.73,
-      "learning_rate": 0.0001761849710982659,
-      "loss": 0.2899,
-      "step": 1218
-    },
-    {
-      "epoch": 2.73,
-      "learning_rate": 0.00017601156069364161,
-      "loss": 0.2595,
-      "step": 1219
-    },
-    {
-      "epoch": 2.74,
-      "learning_rate": 0.0001758381502890173,
-      "loss": 0.2312,
-      "step": 1220
-    },
-    {
-      "epoch": 2.74,
-      "learning_rate": 0.00017566473988439303,
-      "loss": 0.2796,
-      "step": 1221
-    },
-    {
-      "epoch": 2.74,
-      "learning_rate": 0.00017549132947976878,
-      "loss": 0.2577,
-      "step": 1222
-    },
-    {
-      "epoch": 2.74,
-      "learning_rate": 0.0001753179190751445,
-      "loss": 0.3298,
-      "step": 1223
-    },
-    {
-      "epoch": 2.74,
-      "learning_rate": 0.00017514450867052022,
-      "loss": 0.2904,
-      "step": 1224
-    },
-    {
-      "epoch": 2.75,
-      "learning_rate": 0.00017497109826589594,
-      "loss": 0.2401,
-      "step": 1225
-    },
-    {
-      "epoch": 2.75,
-      "learning_rate": 0.00017479768786127169,
-      "loss": 0.273,
-      "step": 1226
-    },
-    {
-      "epoch": 2.75,
-      "learning_rate": 0.00017462427745664738,
-      "loss": 0.2479,
-      "step": 1227
-    },
-    {
-      "epoch": 2.75,
-      "learning_rate": 0.0001744508670520231,
-      "loss": 0.325,
-      "step": 1228
-    },
-    {
-      "epoch": 2.76,
-      "learning_rate": 0.00017427745664739882,
-      "loss": 0.2803,
-      "step": 1229
-    },
-    {
-      "epoch": 2.76,
-      "learning_rate": 0.00017410404624277457,
-      "loss": 0.2399,
-      "step": 1230
-    },
-    {
-      "epoch": 2.76,
-      "learning_rate": 0.0001739306358381503,
-      "loss": 0.2507,
-      "step": 1231
-    },
-    {
-      "epoch": 2.76,
-      "learning_rate": 0.00017375722543352598,
-      "loss": 0.2888,
-      "step": 1232
-    },
-    {
-      "epoch": 2.76,
-      "learning_rate": 0.0001735838150289017,
-      "loss": 0.2838,
-      "step": 1233
-    },
-    {
-      "epoch": 2.77,
-      "learning_rate": 0.00017341040462427745,
-      "loss": 0.2881,
-      "step": 1234
-    },
-    {
-      "epoch": 2.77,
-      "learning_rate": 0.00017323699421965317,
-      "loss": 0.2587,
-      "step": 1235
-    },
-    {
-      "epoch": 2.77,
-      "learning_rate": 0.0001730635838150289,
-      "loss": 0.2566,
-      "step": 1236
-    },
-    {
-      "epoch": 2.77,
-      "learning_rate": 0.00017289017341040459,
-      "loss": 0.3898,
-      "step": 1237
-    },
-    {
-      "epoch": 2.78,
-      "learning_rate": 0.00017271676300578033,
-      "loss": 0.2496,
-      "step": 1238
-    },
-    {
-      "epoch": 2.78,
-      "learning_rate": 0.00017254335260115605,
-      "loss": 0.2205,
-      "step": 1239
-    },
-    {
-      "epoch": 2.78,
-      "learning_rate": 0.00017236994219653178,
-      "loss": 0.2732,
-      "step": 1240
-    },
-    {
-      "epoch": 2.78,
-      "learning_rate": 0.0001721965317919075,
-      "loss": 0.2794,
-      "step": 1241
-    },
-    {
-      "epoch": 2.78,
-      "learning_rate": 0.00017202312138728324,
-      "loss": 0.3402,
-      "step": 1242
-    },
-    {
-      "epoch": 2.79,
-      "learning_rate": 0.00017184971098265894,
-      "loss": 0.5904,
-      "step": 1243
-    },
-    {
-      "epoch": 2.79,
-      "learning_rate": 0.00017167630057803466,
-      "loss": 0.5435,
-      "step": 1244
-    },
-    {
-      "epoch": 2.79,
-      "learning_rate": 0.00017150289017341038,
-      "loss": 0.4598,
-      "step": 1245
-    },
-    {
-      "epoch": 2.79,
-      "learning_rate": 0.00017132947976878613,
-      "loss": 0.4237,
-      "step": 1246
-    },
-    {
-      "epoch": 2.8,
-      "learning_rate": 0.00017115606936416185,
-      "loss": 0.4052,
-      "step": 1247
-    },
-    {
-      "epoch": 2.8,
-      "learning_rate": 0.00017098265895953757,
-      "loss": 0.3596,
-      "step": 1248
-    },
-    {
-      "epoch": 2.8,
-      "learning_rate": 0.00017080924855491326,
-      "loss": 0.3375,
-      "step": 1249
-    },
-    {
-      "epoch": 2.8,
-      "learning_rate": 0.000170635838150289,
-      "loss": 0.3509,
-      "step": 1250
-    },
-    {
-      "epoch": 2.8,
-      "learning_rate": 0.00017046242774566473,
-      "loss": 0.3337,
-      "step": 1251
-    },
-    {
-      "epoch": 2.81,
-      "learning_rate": 0.00017028901734104045,
-      "loss": 0.4023,
-      "step": 1252
-    },
-    {
-      "epoch": 2.81,
-      "learning_rate": 0.00017011560693641617,
-      "loss": 0.2961,
-      "step": 1253
-    },
-    {
-      "epoch": 2.81,
-      "learning_rate": 0.00016994219653179192,
-      "loss": 0.2642,
-      "step": 1254
-    },
-    {
-      "epoch": 2.81,
-      "learning_rate": 0.0001697687861271676,
-      "loss": 0.3523,
-      "step": 1255
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 0.00016959537572254333,
-      "loss": 0.2547,
-      "step": 1256
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 0.00016942196531791905,
-      "loss": 0.2283,
-      "step": 1257
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 0.0001692485549132948,
-      "loss": 0.2924,
-      "step": 1258
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 0.00016907514450867052,
-      "loss": 0.3059,
-      "step": 1259
-    },
-    {
-      "epoch": 2.83,
-      "learning_rate": 0.00016890173410404622,
-      "loss": 0.2438,
-      "step": 1260
-    },
-    {
-      "epoch": 2.83,
-      "learning_rate": 0.00016872832369942194,
-      "loss": 0.2312,
-      "step": 1261
-    },
-    {
-      "epoch": 2.83,
-      "learning_rate": 0.00016855491329479768,
-      "loss": 0.2383,
-      "step": 1262
-    },
-    {
-      "epoch": 2.83,
-      "learning_rate": 0.0001683815028901734,
-      "loss": 0.2653,
-      "step": 1263
-    },
-    {
-      "epoch": 2.83,
-      "learning_rate": 0.00016820809248554913,
-      "loss": 0.2651,
-      "step": 1264
-    },
-    {
-      "epoch": 2.84,
-      "learning_rate": 0.00016803468208092482,
-      "loss": 0.3351,
-      "step": 1265
-    },
-    {
-      "epoch": 2.84,
-      "learning_rate": 0.00016786127167630057,
-      "loss": 0.2531,
-      "step": 1266
-    },
-    {
-      "epoch": 2.84,
-      "learning_rate": 0.0001676878612716763,
-      "loss": 0.3217,
-      "step": 1267
-    },
-    {
-      "epoch": 2.84,
-      "learning_rate": 0.000167514450867052,
-      "loss": 0.2335,
-      "step": 1268
-    },
-    {
-      "epoch": 2.85,
-      "learning_rate": 0.00016734104046242773,
-      "loss": 0.2898,
-      "step": 1269
-    },
-    {
-      "epoch": 2.85,
-      "learning_rate": 0.00016716763005780348,
-      "loss": 0.2112,
-      "step": 1270
-    },
-    {
-      "epoch": 2.85,
-      "learning_rate": 0.0001669942196531792,
-      "loss": 0.2423,
-      "step": 1271
-    },
-    {
-      "epoch": 2.85,
-      "learning_rate": 0.0001668208092485549,
-      "loss": 0.2758,
-      "step": 1272
-    },
-    {
-      "epoch": 2.85,
-      "learning_rate": 0.0001666473988439306,
-      "loss": 0.2943,
-      "step": 1273
-    },
-    {
-      "epoch": 2.86,
-      "learning_rate": 0.00016647398843930633,
-      "loss": 0.2411,
-      "step": 1274
-    },
-    {
-      "epoch": 2.86,
-      "learning_rate": 0.00016630057803468208,
-      "loss": 0.2418,
-      "step": 1275
-    },
-    {
-      "epoch": 2.86,
-      "learning_rate": 0.0001661271676300578,
-      "loss": 0.2899,
-      "step": 1276
-    },
-    {
-      "epoch": 2.86,
-      "learning_rate": 0.0001659537572254335,
-      "loss": 0.2431,
-      "step": 1277
-    },
-    {
-      "epoch": 2.87,
-      "learning_rate": 0.00016578034682080922,
-      "loss": 0.2374,
-      "step": 1278
-    },
-    {
-      "epoch": 2.87,
-      "learning_rate": 0.00016560693641618496,
-      "loss": 0.3077,
-      "step": 1279
-    },
-    {
-      "epoch": 2.87,
-      "learning_rate": 0.00016543352601156068,
-      "loss": 0.1856,
-      "step": 1280
-    },
-    {
-      "epoch": 2.87,
-      "learning_rate": 0.0001652601156069364,
-      "loss": 0.2465,
-      "step": 1281
-    },
-    {
-      "epoch": 2.87,
-      "learning_rate": 0.0001650867052023121,
-      "loss": 0.301,
-      "step": 1282
-    },
-    {
-      "epoch": 2.88,
-      "learning_rate": 0.00016491329479768785,
-      "loss": 0.2545,
-      "step": 1283
-    },
-    {
-      "epoch": 2.88,
-      "learning_rate": 0.00016473988439306357,
-      "loss": 0.3224,
-      "step": 1284
-    },
-    {
-      "epoch": 2.88,
-      "learning_rate": 0.0001645664739884393,
-      "loss": 0.3098,
-      "step": 1285
-    },
-    {
-      "epoch": 2.88,
-      "learning_rate": 0.000164393063583815,
-      "loss": 0.255,
-      "step": 1286
-    },
-    {
-      "epoch": 2.89,
-      "learning_rate": 0.00016421965317919076,
-      "loss": 0.2056,
-      "step": 1287
-    },
-    {
-      "epoch": 2.89,
-      "learning_rate": 0.00016404624277456645,
-      "loss": 0.3752,
-      "step": 1288
-    },
-    {
-      "epoch": 2.89,
-      "learning_rate": 0.00016387283236994217,
-      "loss": 0.2446,
-      "step": 1289
-    },
-    {
-      "epoch": 2.89,
-      "learning_rate": 0.0001636994219653179,
-      "loss": 0.4176,
-      "step": 1290
-    },
-    {
-      "epoch": 2.89,
-      "learning_rate": 0.00016352601156069364,
-      "loss": 0.2583,
-      "step": 1291
-    },
-    {
-      "epoch": 2.9,
-      "learning_rate": 0.00016335260115606936,
-      "loss": 0.5816,
-      "step": 1292
-    },
-    {
-      "epoch": 2.9,
-      "learning_rate": 0.00016317919075144505,
-      "loss": 0.591,
-      "step": 1293
-    },
-    {
-      "epoch": 2.9,
-      "learning_rate": 0.00016300578034682077,
-      "loss": 0.4989,
-      "step": 1294
-    },
-    {
-      "epoch": 2.9,
-      "learning_rate": 0.00016283236994219652,
-      "loss": 0.4007,
-      "step": 1295
-    },
-    {
-      "epoch": 2.91,
-      "learning_rate": 0.00016265895953757224,
-      "loss": 0.3692,
-      "step": 1296
-    },
-    {
-      "epoch": 2.91,
-      "learning_rate": 0.00016248554913294796,
-      "loss": 0.3135,
-      "step": 1297
-    },
-    {
-      "epoch": 2.91,
-      "learning_rate": 0.00016231213872832368,
-      "loss": 0.3618,
-      "step": 1298
-    },
-    {
-      "epoch": 2.91,
-      "learning_rate": 0.00016213872832369943,
-      "loss": 0.3442,
-      "step": 1299
-    },
-    {
-      "epoch": 2.91,
-      "learning_rate": 0.00016196531791907512,
-      "loss": 0.3581,
-      "step": 1300
-    },
-    {
-      "epoch": 2.92,
-      "learning_rate": 0.00016179190751445085,
-      "loss": 0.3073,
-      "step": 1301
-    },
-    {
-      "epoch": 2.92,
-      "learning_rate": 0.00016161849710982657,
-      "loss": 0.3005,
-      "step": 1302
-    },
-    {
-      "epoch": 2.92,
-      "learning_rate": 0.00016144508670520231,
-      "loss": 0.2618,
-      "step": 1303
-    },
-    {
-      "epoch": 2.92,
-      "learning_rate": 0.00016127167630057803,
-      "loss": 0.3217,
-      "step": 1304
-    },
-    {
-      "epoch": 2.93,
-      "learning_rate": 0.00016109826589595373,
-      "loss": 0.2993,
-      "step": 1305
-    },
-    {
-      "epoch": 2.93,
-      "learning_rate": 0.00016092485549132945,
-      "loss": 0.2586,
-      "step": 1306
-    },
-    {
-      "epoch": 2.93,
-      "learning_rate": 0.0001607514450867052,
-      "loss": 0.2481,
-      "step": 1307
-    },
-    {
-      "epoch": 2.93,
-      "learning_rate": 0.00016057803468208092,
-      "loss": 0.2872,
-      "step": 1308
-    },
-    {
-      "epoch": 2.93,
-      "learning_rate": 0.00016040462427745664,
-      "loss": 0.2772,
-      "step": 1309
-    },
-    {
-      "epoch": 2.94,
-      "learning_rate": 0.00016023121387283233,
-      "loss": 0.2466,
-      "step": 1310
-    },
-    {
-      "epoch": 2.94,
-      "learning_rate": 0.00016005780346820808,
-      "loss": 0.3004,
-      "step": 1311
-    },
-    {
-      "epoch": 2.94,
-      "learning_rate": 0.0001598843930635838,
-      "loss": 0.1622,
-      "step": 1312
-    },
-    {
-      "epoch": 2.94,
-      "learning_rate": 0.00015971098265895952,
-      "loss": 0.2447,
-      "step": 1313
-    },
-    {
-      "epoch": 2.95,
-      "learning_rate": 0.00015953757225433524,
-      "loss": 0.2302,
-      "step": 1314
-    },
-    {
-      "epoch": 2.95,
-      "learning_rate": 0.000159364161849711,
-      "loss": 0.2657,
-      "step": 1315
-    },
-    {
-      "epoch": 2.95,
-      "learning_rate": 0.00015919075144508668,
-      "loss": 0.267,
-      "step": 1316
-    },
-    {
-      "epoch": 2.95,
-      "learning_rate": 0.0001590173410404624,
-      "loss": 0.243,
-      "step": 1317
-    },
-    {
-      "epoch": 2.96,
-      "learning_rate": 0.00015884393063583812,
-      "loss": 0.2366,
-      "step": 1318
-    },
-    {
-      "epoch": 2.96,
-      "learning_rate": 0.00015867052023121387,
-      "loss": 0.2866,
-      "step": 1319
-    },
-    {
-      "epoch": 2.96,
-      "learning_rate": 0.0001584971098265896,
-      "loss": 0.2798,
-      "step": 1320
-    },
-    {
-      "epoch": 2.96,
-      "learning_rate": 0.0001583236994219653,
-      "loss": 0.3047,
-      "step": 1321
-    },
-    {
-      "epoch": 2.96,
-      "learning_rate": 0.000158150289017341,
-      "loss": 0.2196,
-      "step": 1322
-    },
-    {
-      "epoch": 2.97,
-      "learning_rate": 0.00015797687861271675,
-      "loss": 0.2324,
-      "step": 1323
-    },
-    {
-      "epoch": 2.97,
-      "learning_rate": 0.00015780346820809248,
-      "loss": 0.2934,
-      "step": 1324
-    },
-    {
-      "epoch": 2.97,
-      "learning_rate": 0.0001576300578034682,
-      "loss": 0.2647,
-      "step": 1325
-    },
-    {
-      "epoch": 2.97,
-      "learning_rate": 0.00015745664739884392,
-      "loss": 0.245,
-      "step": 1326
-    },
-    {
-      "epoch": 2.98,
-      "learning_rate": 0.00015728323699421966,
-      "loss": 0.2012,
-      "step": 1327
-    },
-    {
-      "epoch": 2.98,
-      "learning_rate": 0.00015710982658959536,
-      "loss": 0.2425,
-      "step": 1328
-    },
-    {
-      "epoch": 2.98,
-      "learning_rate": 0.00015693641618497108,
-      "loss": 0.229,
-      "step": 1329
-    },
-    {
-      "epoch": 2.98,
-      "learning_rate": 0.0001567630057803468,
-      "loss": 0.2368,
-      "step": 1330
-    },
-    {
-      "epoch": 2.98,
-      "learning_rate": 0.00015658959537572255,
-      "loss": 0.2612,
-      "step": 1331
-    },
-    {
-      "epoch": 2.99,
-      "learning_rate": 0.00015641618497109827,
-      "loss": 0.2954,
-      "step": 1332
-    },
-    {
-      "epoch": 2.99,
-      "learning_rate": 0.00015624277456647396,
-      "loss": 0.2647,
-      "step": 1333
-    },
-    {
-      "epoch": 2.99,
-      "learning_rate": 0.00015606936416184968,
-      "loss": 0.2506,
-      "step": 1334
-    },
-    {
-      "epoch": 2.99,
-      "learning_rate": 0.00015589595375722543,
-      "loss": 0.2866,
-      "step": 1335
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 0.00015572254335260115,
-      "loss": 0.2059,
-      "step": 1336
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 0.00015554913294797687,
-      "loss": 0.2379,
-      "step": 1337
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 0.00015537572254335256,
-      "loss": 0.3188,
-      "step": 1338
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 0.0001552023121387283,
-      "loss": 0.3332,
-      "step": 1339
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 0.00015502890173410403,
-      "loss": 0.3164,
-      "step": 1340
-    },
-    {
-      "epoch": 3.01,
-      "learning_rate": 0.00015485549132947975,
-      "loss": 0.3005,
-      "step": 1341
-    },
-    {
-      "epoch": 3.01,
-      "learning_rate": 0.00015468208092485547,
-      "loss": 0.1913,
-      "step": 1342
-    },
-    {
-      "epoch": 3.01,
-      "learning_rate": 0.00015450867052023122,
-      "loss": 0.2485,
-      "step": 1343
-    },
-    {
-      "epoch": 3.01,
-      "learning_rate": 0.00015433526011560692,
-      "loss": 0.2041,
-      "step": 1344
-    },
-    {
-      "epoch": 3.02,
-      "learning_rate": 0.00015416184971098264,
-      "loss": 0.2291,
-      "step": 1345
-    },
-    {
-      "epoch": 3.02,
-      "learning_rate": 0.00015398843930635836,
-      "loss": 0.1931,
-      "step": 1346
-    },
-    {
-      "epoch": 3.02,
-      "learning_rate": 0.0001538150289017341,
-      "loss": 0.1707,
-      "step": 1347
-    },
-    {
-      "epoch": 3.02,
-      "learning_rate": 0.00015364161849710983,
-      "loss": 0.2488,
-      "step": 1348
-    },
-    {
-      "epoch": 3.02,
-      "learning_rate": 0.00015346820809248555,
-      "loss": 0.167,
-      "step": 1349
-    },
-    {
-      "epoch": 3.03,
-      "learning_rate": 0.00015329479768786124,
-      "loss": 0.1798,
-      "step": 1350
-    },
-    {
-      "epoch": 3.03,
-      "learning_rate": 0.000153121387283237,
-      "loss": 0.1798,
-      "step": 1351
-    },
-    {
-      "epoch": 3.03,
-      "learning_rate": 0.0001529479768786127,
-      "loss": 0.1232,
-      "step": 1352
-    },
-    {
-      "epoch": 3.03,
-      "learning_rate": 0.00015277456647398843,
-      "loss": 0.171,
-      "step": 1353
-    },
-    {
-      "epoch": 3.04,
-      "learning_rate": 0.00015260115606936415,
-      "loss": 0.2154,
-      "step": 1354
-    },
-    {
-      "epoch": 3.04,
-      "learning_rate": 0.0001524277456647399,
-      "loss": 0.1341,
-      "step": 1355
-    },
-    {
-      "epoch": 3.04,
-      "learning_rate": 0.0001522543352601156,
-      "loss": 0.1676,
-      "step": 1356
-    },
-    {
-      "epoch": 3.04,
-      "learning_rate": 0.0001520809248554913,
-      "loss": 0.1611,
-      "step": 1357
-    },
-    {
-      "epoch": 3.04,
-      "learning_rate": 0.00015190751445086703,
-      "loss": 0.1838,
-      "step": 1358
-    },
-    {
-      "epoch": 3.05,
-      "learning_rate": 0.00015173410404624278,
-      "loss": 0.164,
-      "step": 1359
-    },
-    {
-      "epoch": 3.05,
-      "learning_rate": 0.0001515606936416185,
-      "loss": 0.1697,
-      "step": 1360
-    },
-    {
-      "epoch": 3.05,
-      "learning_rate": 0.0001513872832369942,
-      "loss": 0.1793,
-      "step": 1361
-    },
-    {
-      "epoch": 3.05,
-      "learning_rate": 0.00015121387283236992,
-      "loss": 0.1421,
-      "step": 1362
-    },
-    {
-      "epoch": 3.06,
-      "learning_rate": 0.00015104046242774566,
-      "loss": 0.1488,
-      "step": 1363
-    },
-    {
-      "epoch": 3.06,
-      "learning_rate": 0.00015086705202312138,
-      "loss": 0.1655,
-      "step": 1364
-    },
-    {
-      "epoch": 3.06,
-      "learning_rate": 0.0001506936416184971,
-      "loss": 0.2056,
-      "step": 1365
-    },
-    {
-      "epoch": 3.06,
-      "learning_rate": 0.0001505202312138728,
-      "loss": 0.121,
-      "step": 1366
-    },
-    {
-      "epoch": 3.07,
-      "learning_rate": 0.00015034682080924855,
-      "loss": 0.1514,
-      "step": 1367
-    },
-    {
-      "epoch": 3.07,
-      "learning_rate": 0.00015017341040462427,
-      "loss": 0.1511,
-      "step": 1368
-    },
-    {
-      "epoch": 3.07,
-      "learning_rate": 0.00015,
-      "loss": 0.1363,
-      "step": 1369
-    },
-    {
-      "epoch": 3.07,
-      "learning_rate": 0.0001498265895953757,
-      "loss": 0.1644,
-      "step": 1370
-    },
-    {
-      "epoch": 3.07,
-      "learning_rate": 0.00014965317919075143,
-      "loss": 0.1106,
-      "step": 1371
-    },
-    {
-      "epoch": 3.08,
-      "learning_rate": 0.00014947976878612715,
-      "loss": 0.1259,
-      "step": 1372
-    },
-    {
-      "epoch": 3.08,
-      "learning_rate": 0.00014930635838150287,
-      "loss": 0.1348,
-      "step": 1373
-    },
-    {
-      "epoch": 3.08,
-      "learning_rate": 0.0001491329479768786,
-      "loss": 0.1144,
-      "step": 1374
-    },
-    {
-      "epoch": 3.08,
-      "learning_rate": 0.0001489595375722543,
-      "loss": 0.1318,
-      "step": 1375
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 0.00014878612716763003,
-      "loss": 0.1212,
-      "step": 1376
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 0.00014861271676300578,
-      "loss": 0.1427,
-      "step": 1377
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 0.00014843930635838147,
-      "loss": 0.1204,
-      "step": 1378
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 0.00014826589595375722,
-      "loss": 0.6003,
-      "step": 1379
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 0.00014809248554913291,
-      "loss": 0.1959,
-      "step": 1380
-    },
-    {
-      "epoch": 3.1,
-      "learning_rate": 0.00014791907514450866,
-      "loss": 0.1231,
-      "step": 1381
-    },
-    {
-      "epoch": 3.1,
-      "learning_rate": 0.00014774566473988438,
-      "loss": 0.1274,
-      "step": 1382
-    },
-    {
-      "epoch": 3.1,
-      "learning_rate": 0.0001475722543352601,
-      "loss": 0.1249,
-      "step": 1383
-    },
-    {
-      "epoch": 3.1,
-      "learning_rate": 0.00014739884393063582,
-      "loss": 0.1267,
-      "step": 1384
-    },
-    {
-      "epoch": 3.11,
-      "learning_rate": 0.00014722543352601154,
-      "loss": 0.1784,
-      "step": 1385
-    },
-    {
-      "epoch": 3.11,
-      "learning_rate": 0.00014705202312138727,
-      "loss": 0.17,
-      "step": 1386
-    },
-    {
-      "epoch": 3.11,
-      "learning_rate": 0.00014687861271676299,
-      "loss": 0.1838,
-      "step": 1387
-    },
-    {
-      "epoch": 3.11,
-      "learning_rate": 0.0001467052023121387,
-      "loss": 0.2029,
-      "step": 1388
-    },
-    {
-      "epoch": 3.11,
-      "learning_rate": 0.00014653179190751443,
-      "loss": 0.3663,
-      "step": 1389
-    },
-    {
-      "epoch": 3.12,
-      "learning_rate": 0.00014635838150289015,
-      "loss": 0.3393,
-      "step": 1390
-    },
-    {
-      "epoch": 3.12,
-      "learning_rate": 0.0001461849710982659,
-      "loss": 0.2693,
-      "step": 1391
-    },
-    {
-      "epoch": 3.12,
-      "learning_rate": 0.0001460115606936416,
-      "loss": 0.2366,
-      "step": 1392
-    },
-    {
-      "epoch": 3.12,
-      "learning_rate": 0.00014583815028901734,
-      "loss": 0.2271,
-      "step": 1393
-    },
-    {
-      "epoch": 3.13,
-      "learning_rate": 0.00014566473988439306,
-      "loss": 0.2333,
-      "step": 1394
-    },
-    {
-      "epoch": 3.13,
-      "learning_rate": 0.00014549132947976878,
-      "loss": 0.2391,
-      "step": 1395
-    },
-    {
-      "epoch": 3.13,
-      "learning_rate": 0.0001453179190751445,
-      "loss": 0.563,
-      "step": 1396
-    },
-    {
-      "epoch": 3.13,
-      "learning_rate": 0.00014514450867052022,
-      "loss": 0.1767,
-      "step": 1397
-    },
-    {
-      "epoch": 3.13,
-      "learning_rate": 0.00014497109826589594,
-      "loss": 0.2116,
-      "step": 1398
-    },
-    {
-      "epoch": 3.14,
-      "learning_rate": 0.00014479768786127166,
-      "loss": 0.1731,
-      "step": 1399
-    },
-    {
-      "epoch": 3.14,
-      "learning_rate": 0.00014462427745664738,
-      "loss": 0.2047,
-      "step": 1400
-    },
-    {
-      "epoch": 3.14,
-      "learning_rate": 0.0001444508670520231,
-      "loss": 0.1805,
-      "step": 1401
-    },
-    {
-      "epoch": 3.14,
-      "learning_rate": 0.00014427745664739882,
-      "loss": 0.1929,
-      "step": 1402
-    },
-    {
-      "epoch": 3.15,
-      "learning_rate": 0.00014410404624277454,
-      "loss": 0.1438,
-      "step": 1403
-    },
-    {
-      "epoch": 3.15,
-      "learning_rate": 0.00014393063583815026,
-      "loss": 0.1483,
-      "step": 1404
-    },
-    {
-      "epoch": 3.15,
-      "learning_rate": 0.000143757225433526,
-      "loss": 0.1981,
-      "step": 1405
-    },
-    {
-      "epoch": 3.15,
-      "learning_rate": 0.0001435838150289017,
-      "loss": 0.1836,
-      "step": 1406
-    },
-    {
-      "epoch": 3.15,
-      "learning_rate": 0.00014341040462427745,
-      "loss": 0.1852,
-      "step": 1407
-    },
-    {
-      "epoch": 3.16,
-      "learning_rate": 0.00014323699421965317,
-      "loss": 0.1698,
-      "step": 1408
-    },
-    {
-      "epoch": 3.16,
-      "learning_rate": 0.0001430635838150289,
-      "loss": 0.1216,
-      "step": 1409
-    },
-    {
-      "epoch": 3.16,
-      "learning_rate": 0.00014289017341040462,
-      "loss": 0.1676,
-      "step": 1410
-    },
-    {
-      "epoch": 3.16,
-      "learning_rate": 0.00014271676300578034,
-      "loss": 0.1809,
-      "step": 1411
-    },
-    {
-      "epoch": 3.17,
-      "learning_rate": 0.00014254335260115606,
-      "loss": 0.1413,
-      "step": 1412
-    },
-    {
-      "epoch": 3.17,
-      "learning_rate": 0.00014236994219653178,
-      "loss": 0.137,
-      "step": 1413
-    },
-    {
-      "epoch": 3.17,
-      "learning_rate": 0.0001421965317919075,
-      "loss": 0.1761,
-      "step": 1414
-    },
-    {
-      "epoch": 3.17,
-      "learning_rate": 0.00014202312138728322,
-      "loss": 0.126,
-      "step": 1415
-    },
-    {
-      "epoch": 3.17,
-      "learning_rate": 0.00014184971098265894,
-      "loss": 0.1518,
-      "step": 1416
-    },
-    {
-      "epoch": 3.18,
-      "learning_rate": 0.00014167630057803466,
-      "loss": 0.1303,
-      "step": 1417
-    },
-    {
-      "epoch": 3.18,
-      "learning_rate": 0.00014150289017341038,
-      "loss": 0.1535,
-      "step": 1418
-    },
-    {
-      "epoch": 3.18,
-      "learning_rate": 0.00014132947976878613,
-      "loss": 0.1286,
-      "step": 1419
-    },
-    {
-      "epoch": 3.18,
-      "learning_rate": 0.00014115606936416182,
-      "loss": 0.1395,
-      "step": 1420
-    },
-    {
-      "epoch": 3.19,
-      "learning_rate": 0.00014098265895953757,
-      "loss": 0.1791,
-      "step": 1421
-    },
-    {
-      "epoch": 3.19,
-      "learning_rate": 0.0001408092485549133,
-      "loss": 0.1708,
-      "step": 1422
-    },
-    {
-      "epoch": 3.19,
-      "learning_rate": 0.000140635838150289,
-      "loss": 0.5027,
-      "step": 1423
-    },
-    {
-      "epoch": 3.19,
-      "learning_rate": 0.00014046242774566473,
-      "loss": 0.1802,
-      "step": 1424
-    },
-    {
-      "epoch": 3.2,
-      "learning_rate": 0.00014028901734104045,
-      "loss": 0.1555,
-      "step": 1425
-    },
-    {
-      "epoch": 3.2,
-      "learning_rate": 0.00014011560693641617,
-      "loss": 0.1257,
-      "step": 1426
-    },
-    {
-      "epoch": 3.2,
-      "learning_rate": 0.0001399421965317919,
-      "loss": 0.1221,
-      "step": 1427
-    },
-    {
-      "epoch": 3.2,
-      "learning_rate": 0.00013976878612716762,
-      "loss": 0.1183,
-      "step": 1428
-    },
-    {
-      "epoch": 3.2,
-      "learning_rate": 0.00013959537572254334,
-      "loss": 0.1121,
-      "step": 1429
-    },
-    {
-      "epoch": 3.21,
-      "learning_rate": 0.00013942196531791906,
-      "loss": 0.1325,
-      "step": 1430
-    },
-    {
-      "epoch": 3.21,
-      "learning_rate": 0.00013924855491329478,
-      "loss": 0.1182,
-      "step": 1431
-    },
-    {
-      "epoch": 3.21,
-      "learning_rate": 0.0001390751445086705,
-      "loss": 0.1615,
-      "step": 1432
-    },
-    {
-      "epoch": 3.21,
-      "learning_rate": 0.00013890173410404625,
-      "loss": 0.1368,
-      "step": 1433
-    },
-    {
-      "epoch": 3.22,
-      "learning_rate": 0.00013872832369942194,
-      "loss": 0.1295,
-      "step": 1434
-    },
-    {
-      "epoch": 3.22,
-      "learning_rate": 0.0001385549132947977,
-      "loss": 0.0937,
-      "step": 1435
-    },
-    {
-      "epoch": 3.22,
-      "learning_rate": 0.0001383815028901734,
-      "loss": 0.113,
-      "step": 1436
-    },
-    {
-      "epoch": 3.22,
-      "learning_rate": 0.00013820809248554913,
-      "loss": 0.1177,
-      "step": 1437
-    },
-    {
-      "epoch": 3.22,
-      "learning_rate": 0.00013803468208092485,
-      "loss": 0.1841,
-      "step": 1438
-    },
-    {
-      "epoch": 3.23,
-      "learning_rate": 0.00013786127167630057,
-      "loss": 0.3102,
-      "step": 1439
-    },
-    {
-      "epoch": 3.23,
-      "learning_rate": 0.0001376878612716763,
-      "loss": 0.3234,
-      "step": 1440
-    },
-    {
-      "epoch": 3.23,
-      "learning_rate": 0.000137514450867052,
-      "loss": 0.3114,
-      "step": 1441
-    },
-    {
-      "epoch": 3.23,
-      "learning_rate": 0.00013734104046242773,
-      "loss": 0.2625,
-      "step": 1442
-    },
-    {
-      "epoch": 3.24,
-      "learning_rate": 0.00013716763005780345,
-      "loss": 0.1896,
-      "step": 1443
-    },
-    {
-      "epoch": 3.24,
-      "learning_rate": 0.00013699421965317917,
-      "loss": 0.1894,
-      "step": 1444
-    },
-    {
-      "epoch": 3.24,
-      "learning_rate": 0.00013682080924855492,
-      "loss": 0.1994,
-      "step": 1445
-    },
-    {
-      "epoch": 3.24,
-      "learning_rate": 0.00013664739884393061,
-      "loss": 0.1791,
-      "step": 1446
-    },
-    {
-      "epoch": 3.24,
-      "learning_rate": 0.00013647398843930636,
-      "loss": 0.213,
-      "step": 1447
-    },
-    {
-      "epoch": 3.25,
-      "learning_rate": 0.00013630057803468206,
-      "loss": 0.1923,
-      "step": 1448
-    },
-    {
-      "epoch": 3.25,
-      "learning_rate": 0.0001361271676300578,
-      "loss": 0.1412,
-      "step": 1449
-    },
-    {
-      "epoch": 3.25,
-      "learning_rate": 0.00013595375722543352,
-      "loss": 0.1965,
-      "step": 1450
-    },
-    {
-      "epoch": 3.25,
-      "learning_rate": 0.00013578034682080925,
-      "loss": 0.1976,
-      "step": 1451
-    },
-    {
-      "epoch": 3.26,
-      "learning_rate": 0.00013560693641618497,
-      "loss": 0.1655,
-      "step": 1452
-    },
-    {
-      "epoch": 3.26,
-      "learning_rate": 0.0001354335260115607,
-      "loss": 0.2195,
-      "step": 1453
-    },
-    {
-      "epoch": 3.26,
-      "learning_rate": 0.0001352601156069364,
-      "loss": 0.1942,
-      "step": 1454
-    },
-    {
-      "epoch": 3.26,
-      "learning_rate": 0.00013508670520231213,
-      "loss": 0.1557,
-      "step": 1455
-    },
-    {
-      "epoch": 3.26,
-      "learning_rate": 0.00013491329479768785,
-      "loss": 0.155,
-      "step": 1456
-    },
-    {
-      "epoch": 3.27,
-      "learning_rate": 0.00013473988439306357,
-      "loss": 0.1443,
-      "step": 1457
-    },
-    {
-      "epoch": 3.27,
-      "learning_rate": 0.0001345664739884393,
-      "loss": 0.1514,
-      "step": 1458
-    },
-    {
-      "epoch": 3.27,
-      "learning_rate": 0.00013439306358381504,
-      "loss": 0.1367,
-      "step": 1459
-    },
-    {
-      "epoch": 3.27,
-      "learning_rate": 0.00013421965317919073,
-      "loss": 0.1459,
-      "step": 1460
-    },
-    {
-      "epoch": 3.28,
-      "learning_rate": 0.00013404624277456648,
-      "loss": 0.1873,
-      "step": 1461
-    },
-    {
-      "epoch": 3.28,
-      "learning_rate": 0.00013387283236994217,
-      "loss": 0.1342,
-      "step": 1462
-    },
-    {
-      "epoch": 3.28,
-      "learning_rate": 0.00013369942196531792,
-      "loss": 0.1421,
-      "step": 1463
-    },
-    {
-      "epoch": 3.28,
-      "learning_rate": 0.00013352601156069364,
-      "loss": 0.1153,
-      "step": 1464
-    },
-    {
-      "epoch": 3.28,
-      "learning_rate": 0.00013335260115606936,
-      "loss": 0.1168,
-      "step": 1465
-    },
-    {
-      "epoch": 3.29,
-      "learning_rate": 0.00013317919075144508,
-      "loss": 0.1794,
-      "step": 1466
-    },
-    {
-      "epoch": 3.29,
-      "learning_rate": 0.00013300578034682078,
-      "loss": 0.2179,
-      "step": 1467
-    },
-    {
-      "epoch": 3.29,
-      "learning_rate": 0.00013283236994219652,
-      "loss": 0.1447,
-      "step": 1468
-    },
-    {
-      "epoch": 3.29,
-      "learning_rate": 0.00013265895953757224,
-      "loss": 0.1838,
-      "step": 1469
-    },
-    {
-      "epoch": 3.3,
-      "learning_rate": 0.00013248554913294797,
-      "loss": 0.1361,
-      "step": 1470
-    },
-    {
-      "epoch": 3.3,
-      "learning_rate": 0.00013231213872832369,
-      "loss": 0.1267,
-      "step": 1471
-    },
-    {
-      "epoch": 3.3,
-      "learning_rate": 0.0001321387283236994,
-      "loss": 0.1564,
-      "step": 1472
-    },
-    {
-      "epoch": 3.3,
-      "learning_rate": 0.00013196531791907513,
-      "loss": 0.1736,
-      "step": 1473
-    },
-    {
-      "epoch": 3.3,
-      "learning_rate": 0.00013179190751445085,
-      "loss": 0.1423,
-      "step": 1474
-    },
-    {
-      "epoch": 3.31,
-      "learning_rate": 0.00013161849710982657,
-      "loss": 0.1301,
-      "step": 1475
-    },
-    {
-      "epoch": 3.31,
-      "learning_rate": 0.0001314450867052023,
-      "loss": 0.1068,
-      "step": 1476
-    },
-    {
-      "epoch": 3.31,
-      "learning_rate": 0.000131271676300578,
-      "loss": 0.1306,
-      "step": 1477
-    },
-    {
-      "epoch": 3.31,
-      "learning_rate": 0.00013109826589595376,
-      "loss": 0.1315,
-      "step": 1478
-    },
-    {
-      "epoch": 3.32,
-      "learning_rate": 0.00013092485549132945,
-      "loss": 0.1319,
-      "step": 1479
-    },
-    {
-      "epoch": 3.32,
-      "learning_rate": 0.0001307514450867052,
-      "loss": 0.1319,
-      "step": 1480
-    },
-    {
-      "epoch": 3.32,
-      "learning_rate": 0.00013057803468208092,
-      "loss": 0.4801,
-      "step": 1481
-    },
-    {
-      "epoch": 3.32,
-      "learning_rate": 0.00013040462427745664,
-      "loss": 0.1113,
-      "step": 1482
-    },
-    {
-      "epoch": 3.33,
-      "learning_rate": 0.00013023121387283236,
-      "loss": 0.1629,
-      "step": 1483
-    },
-    {
-      "epoch": 3.33,
-      "learning_rate": 0.00013005780346820808,
-      "loss": 0.1464,
-      "step": 1484
-    },
-    {
-      "epoch": 3.33,
-      "learning_rate": 0.0001298843930635838,
-      "loss": 0.1699,
-      "step": 1485
-    },
-    {
-      "epoch": 3.33,
-      "learning_rate": 0.00012971098265895952,
-      "loss": 0.1579,
-      "step": 1486
-    },
-    {
-      "epoch": 3.33,
-      "learning_rate": 0.00012953757225433524,
-      "loss": 0.1498,
-      "step": 1487
-    },
-    {
-      "epoch": 3.34,
-      "learning_rate": 0.00012936416184971096,
-      "loss": 0.1559,
-      "step": 1488
-    },
-    {
-      "epoch": 3.34,
-      "learning_rate": 0.00012919075144508669,
-      "loss": 0.4199,
-      "step": 1489
-    },
-    {
-      "epoch": 3.34,
-      "learning_rate": 0.0001290173410404624,
-      "loss": 0.2768,
-      "step": 1490
-    },
-    {
-      "epoch": 3.34,
-      "learning_rate": 0.00012884393063583813,
-      "loss": 0.2245,
-      "step": 1491
-    },
-    {
-      "epoch": 3.35,
-      "learning_rate": 0.00012867052023121387,
-      "loss": 0.2052,
-      "step": 1492
-    },
-    {
-      "epoch": 3.35,
-      "learning_rate": 0.00012849710982658957,
-      "loss": 0.1749,
-      "step": 1493
-    },
-    {
-      "epoch": 3.35,
-      "learning_rate": 0.00012832369942196532,
-      "loss": 0.1611,
-      "step": 1494
-    },
-    {
-      "epoch": 3.35,
-      "learning_rate": 0.00012815028901734104,
-      "loss": 0.2097,
-      "step": 1495
-    },
-    {
-      "epoch": 3.35,
-      "learning_rate": 0.00012797687861271676,
-      "loss": 0.1994,
-      "step": 1496
-    },
-    {
-      "epoch": 3.36,
-      "learning_rate": 0.00012780346820809248,
-      "loss": 0.1656,
-      "step": 1497
-    },
-    {
-      "epoch": 3.36,
-      "learning_rate": 0.0001276300578034682,
-      "loss": 0.1881,
-      "step": 1498
-    },
-    {
-      "epoch": 3.36,
-      "learning_rate": 0.00012745664739884392,
-      "loss": 0.214,
-      "step": 1499
-    },
-    {
-      "epoch": 3.36,
-      "learning_rate": 0.00012728323699421964,
-      "loss": 0.1277,
-      "step": 1500
-    },
-    {
-      "epoch": 3.36,
-      "eval_loss": 0.34972885251045227,
-      "eval_runtime": 555.1742,
-      "eval_samples_per_second": 4.759,
-      "eval_steps_per_second": 0.596,
-      "eval_wer": 0.1040063466878223,
-      "step": 1500
-    },
-    {
-      "epoch": 3.37,
-      "learning_rate": 0.00012710982658959536,
-      "loss": 0.1352,
-      "step": 1501
-    },
-    {
-      "epoch": 3.37,
-      "learning_rate": 0.00012693641618497108,
-      "loss": 0.1341,
-      "step": 1502
-    },
-    {
-      "epoch": 3.37,
-      "learning_rate": 0.0001267630057803468,
-      "loss": 0.1614,
-      "step": 1503
-    },
-    {
-      "epoch": 3.37,
-      "learning_rate": 0.00012658959537572252,
-      "loss": 0.1356,
-      "step": 1504
-    },
-    {
-      "epoch": 3.37,
-      "learning_rate": 0.00012641618497109824,
-      "loss": 0.1486,
-      "step": 1505
-    },
-    {
-      "epoch": 3.38,
-      "learning_rate": 0.000126242774566474,
-      "loss": 0.1557,
-      "step": 1506
-    },
-    {
-      "epoch": 3.38,
-      "learning_rate": 0.00012606936416184968,
-      "loss": 0.1015,
-      "step": 1507
-    },
-    {
-      "epoch": 3.38,
-      "learning_rate": 0.00012589595375722543,
-      "loss": 0.1544,
-      "step": 1508
-    },
-    {
-      "epoch": 3.38,
-      "learning_rate": 0.00012572254335260115,
-      "loss": 0.1357,
-      "step": 1509
-    },
-    {
-      "epoch": 3.39,
-      "learning_rate": 0.00012554913294797687,
-      "loss": 0.1212,
-      "step": 1510
-    },
-    {
-      "epoch": 3.39,
-      "learning_rate": 0.0001253757225433526,
-      "loss": 0.1688,
-      "step": 1511
-    },
-    {
-      "epoch": 3.39,
-      "learning_rate": 0.00012520231213872831,
-      "loss": 0.1234,
-      "step": 1512
-    },
-    {
-      "epoch": 3.39,
-      "learning_rate": 0.00012502890173410404,
-      "loss": 0.1307,
-      "step": 1513
-    },
-    {
-      "epoch": 3.39,
-      "learning_rate": 0.00012485549132947976,
-      "loss": 0.1111,
-      "step": 1514
-    },
-    {
-      "epoch": 3.4,
-      "learning_rate": 0.00012468208092485548,
-      "loss": 0.133,
-      "step": 1515
-    },
-    {
-      "epoch": 3.4,
-      "learning_rate": 0.0001245086705202312,
-      "loss": 0.1339,
-      "step": 1516
-    },
-    {
-      "epoch": 3.4,
-      "learning_rate": 0.00012433526011560692,
-      "loss": 0.1395,
-      "step": 1517
-    },
-    {
-      "epoch": 3.4,
-      "learning_rate": 0.00012416184971098267,
-      "loss": 0.1324,
-      "step": 1518
-    },
-    {
-      "epoch": 3.41,
-      "learning_rate": 0.00012398843930635836,
-      "loss": 0.0869,
-      "step": 1519
-    },
-    {
-      "epoch": 3.41,
-      "learning_rate": 0.0001238150289017341,
-      "loss": 0.1488,
-      "step": 1520
-    },
-    {
-      "epoch": 3.41,
-      "learning_rate": 0.0001236416184971098,
-      "loss": 0.129,
-      "step": 1521
-    },
-    {
-      "epoch": 3.41,
-      "learning_rate": 0.00012346820809248555,
-      "loss": 0.1255,
-      "step": 1522
-    },
-    {
-      "epoch": 3.41,
-      "learning_rate": 0.00012329479768786127,
-      "loss": 0.1568,
-      "step": 1523
-    },
-    {
-      "epoch": 3.42,
-      "learning_rate": 0.000123121387283237,
-      "loss": 0.1269,
-      "step": 1524
-    },
-    {
-      "epoch": 3.42,
-      "learning_rate": 0.0001229479768786127,
-      "loss": 0.0831,
-      "step": 1525
-    },
-    {
-      "epoch": 3.42,
-      "learning_rate": 0.00012277456647398843,
-      "loss": 0.1454,
-      "step": 1526
-    },
-    {
-      "epoch": 3.42,
-      "learning_rate": 0.00012260115606936415,
-      "loss": 0.1484,
-      "step": 1527
-    },
-    {
-      "epoch": 3.43,
-      "learning_rate": 0.00012242774566473987,
-      "loss": 0.1237,
-      "step": 1528
-    },
-    {
-      "epoch": 3.43,
-      "learning_rate": 0.0001222543352601156,
-      "loss": 0.1523,
-      "step": 1529
-    },
-    {
-      "epoch": 3.43,
-      "learning_rate": 0.00012208092485549131,
-      "loss": 0.1167,
-      "step": 1530
-    },
-    {
-      "epoch": 3.43,
-      "learning_rate": 0.00012190751445086703,
-      "loss": 0.1941,
-      "step": 1531
-    },
-    {
-      "epoch": 3.43,
-      "learning_rate": 0.00012173410404624277,
-      "loss": 0.1381,
-      "step": 1532
-    },
-    {
-      "epoch": 3.44,
-      "learning_rate": 0.00012156069364161849,
-      "loss": 0.1792,
-      "step": 1533
-    },
-    {
-      "epoch": 3.44,
-      "learning_rate": 0.00012138728323699421,
-      "loss": 0.0959,
-      "step": 1534
-    },
-    {
-      "epoch": 3.44,
-      "learning_rate": 0.00012121387283236993,
-      "loss": 0.1264,
-      "step": 1535
-    },
-    {
-      "epoch": 3.44,
-      "learning_rate": 0.00012104046242774567,
-      "loss": 0.0949,
-      "step": 1536
-    },
-    {
-      "epoch": 3.45,
-      "learning_rate": 0.00012086705202312137,
-      "loss": 0.1213,
-      "step": 1537
-    },
-    {
-      "epoch": 3.45,
-      "learning_rate": 0.0001206936416184971,
-      "loss": 0.2097,
-      "step": 1538
-    },
-    {
-      "epoch": 3.45,
-      "learning_rate": 0.00012052023121387281,
-      "loss": 0.292,
-      "step": 1539
-    },
-    {
-      "epoch": 3.45,
-      "learning_rate": 0.00012034682080924855,
-      "loss": 0.2704,
-      "step": 1540
-    },
-    {
-      "epoch": 3.46,
-      "learning_rate": 0.00012017341040462427,
-      "loss": 0.2344,
-      "step": 1541
-    },
-    {
-      "epoch": 3.46,
-      "learning_rate": 0.00011999999999999999,
-      "loss": 0.2288,
-      "step": 1542
-    },
-    {
-      "epoch": 3.46,
-      "learning_rate": 0.00011982658959537571,
-      "loss": 0.2087,
-      "step": 1543
-    },
-    {
-      "epoch": 3.46,
-      "learning_rate": 0.00011965317919075144,
-      "loss": 0.2058,
-      "step": 1544
-    },
-    {
-      "epoch": 3.46,
-      "learning_rate": 0.00011947976878612715,
-      "loss": 0.173,
-      "step": 1545
-    },
-    {
-      "epoch": 3.47,
-      "learning_rate": 0.00011930635838150289,
-      "loss": 0.1584,
-      "step": 1546
-    },
-    {
-      "epoch": 3.47,
-      "learning_rate": 0.0001191329479768786,
-      "loss": 0.2082,
-      "step": 1547
-    },
-    {
-      "epoch": 3.47,
-      "learning_rate": 0.00011895953757225433,
-      "loss": 0.1624,
-      "step": 1548
-    },
-    {
-      "epoch": 3.47,
-      "learning_rate": 0.00011878612716763005,
-      "loss": 0.1681,
-      "step": 1549
-    },
-    {
-      "epoch": 3.48,
-      "learning_rate": 0.00011861271676300578,
-      "loss": 0.1637,
-      "step": 1550
-    },
-    {
-      "epoch": 3.48,
-      "learning_rate": 0.00011843930635838149,
-      "loss": 0.1455,
-      "step": 1551
-    },
-    {
-      "epoch": 3.48,
-      "learning_rate": 0.00011826589595375722,
-      "loss": 0.1189,
-      "step": 1552
-    },
-    {
-      "epoch": 3.48,
-      "learning_rate": 0.00011809248554913293,
-      "loss": 0.1575,
-      "step": 1553
-    },
-    {
-      "epoch": 3.48,
-      "learning_rate": 0.00011791907514450866,
-      "loss": 0.1567,
-      "step": 1554
-    },
-    {
-      "epoch": 3.49,
-      "learning_rate": 0.00011774566473988439,
-      "loss": 0.1484,
-      "step": 1555
-    },
-    {
-      "epoch": 3.49,
-      "learning_rate": 0.00011757225433526012,
-      "loss": 0.1359,
-      "step": 1556
-    },
-    {
-      "epoch": 3.49,
-      "learning_rate": 0.00011739884393063583,
-      "loss": 0.1224,
-      "step": 1557
-    },
-    {
-      "epoch": 3.49,
-      "learning_rate": 0.00011722543352601156,
-      "loss": 0.1439,
-      "step": 1558
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 0.00011705202312138727,
-      "loss": 0.1238,
-      "step": 1559
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 0.000116878612716763,
-      "loss": 0.1421,
-      "step": 1560
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 0.00011670520231213872,
-      "loss": 0.1541,
-      "step": 1561
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 0.00011653179190751443,
-      "loss": 0.1132,
-      "step": 1562
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 0.00011635838150289016,
-      "loss": 0.136,
-      "step": 1563
-    },
-    {
-      "epoch": 3.51,
-      "learning_rate": 0.00011618497109826587,
-      "loss": 0.134,
-      "step": 1564
-    },
-    {
-      "epoch": 3.51,
-      "learning_rate": 0.0001160115606936416,
-      "loss": 0.1325,
-      "step": 1565
-    },
-    {
-      "epoch": 3.51,
-      "learning_rate": 0.00011583815028901733,
-      "loss": 0.1259,
-      "step": 1566
-    },
-    {
-      "epoch": 3.51,
-      "learning_rate": 0.00011566473988439306,
-      "loss": 0.16,
-      "step": 1567
-    },
-    {
-      "epoch": 3.52,
-      "learning_rate": 0.00011549132947976877,
-      "loss": 0.1253,
-      "step": 1568
-    },
-    {
-      "epoch": 3.52,
-      "learning_rate": 0.0001153179190751445,
-      "loss": 0.1421,
-      "step": 1569
-    },
-    {
-      "epoch": 3.52,
-      "learning_rate": 0.00011514450867052021,
-      "loss": 0.1387,
-      "step": 1570
-    },
-    {
-      "epoch": 3.52,
-      "learning_rate": 0.00011497109826589594,
-      "loss": 0.1289,
-      "step": 1571
-    },
-    {
-      "epoch": 3.52,
-      "learning_rate": 0.00011479768786127166,
-      "loss": 0.1405,
-      "step": 1572
-    },
-    {
-      "epoch": 3.53,
-      "learning_rate": 0.00011462427745664738,
-      "loss": 0.171,
-      "step": 1573
-    },
-    {
-      "epoch": 3.53,
-      "learning_rate": 0.0001144508670520231,
-      "loss": 0.1136,
-      "step": 1574
-    },
-    {
-      "epoch": 3.53,
-      "learning_rate": 0.00011427745664739884,
-      "loss": 0.1325,
-      "step": 1575
-    },
-    {
-      "epoch": 3.53,
-      "learning_rate": 0.00011410404624277455,
-      "loss": 0.0906,
-      "step": 1576
-    },
-    {
-      "epoch": 3.54,
-      "learning_rate": 0.00011393063583815028,
-      "loss": 0.0978,
-      "step": 1577
-    },
-    {
-      "epoch": 3.54,
-      "learning_rate": 0.00011375722543352599,
-      "loss": 0.1221,
-      "step": 1578
-    },
-    {
-      "epoch": 3.54,
-      "learning_rate": 0.00011358381502890172,
-      "loss": 0.2028,
-      "step": 1579
-    },
-    {
-      "epoch": 3.54,
-      "learning_rate": 0.00011341040462427744,
-      "loss": 0.131,
-      "step": 1580
-    },
-    {
-      "epoch": 3.54,
-      "learning_rate": 0.00011323699421965318,
-      "loss": 0.1039,
-      "step": 1581
-    },
-    {
-      "epoch": 3.55,
-      "learning_rate": 0.00011306358381502888,
-      "loss": 0.1327,
-      "step": 1582
-    },
-    {
-      "epoch": 3.55,
-      "learning_rate": 0.00011289017341040462,
-      "loss": 0.0906,
-      "step": 1583
-    },
-    {
-      "epoch": 3.55,
-      "learning_rate": 0.00011271676300578033,
-      "loss": 0.1549,
-      "step": 1584
-    },
-    {
-      "epoch": 3.55,
-      "learning_rate": 0.00011254335260115606,
-      "loss": 0.1482,
-      "step": 1585
-    },
-    {
-      "epoch": 3.56,
-      "learning_rate": 0.00011236994219653178,
-      "loss": 0.0944,
-      "step": 1586
-    },
-    {
-      "epoch": 3.56,
-      "learning_rate": 0.0001121965317919075,
-      "loss": 0.0893,
-      "step": 1587
-    },
-    {
-      "epoch": 3.56,
-      "learning_rate": 0.00011202312138728322,
-      "loss": 0.1269,
-      "step": 1588
-    },
-    {
-      "epoch": 3.56,
-      "learning_rate": 0.00011184971098265896,
-      "loss": 0.3062,
-      "step": 1589
-    },
-    {
-      "epoch": 3.57,
-      "learning_rate": 0.00011167630057803466,
-      "loss": 0.2444,
-      "step": 1590
-    },
-    {
-      "epoch": 3.57,
-      "learning_rate": 0.0001115028901734104,
-      "loss": 0.2537,
-      "step": 1591
-    },
-    {
-      "epoch": 3.57,
-      "learning_rate": 0.00011132947976878612,
-      "loss": 0.1882,
-      "step": 1592
-    },
-    {
-      "epoch": 3.57,
-      "learning_rate": 0.00011115606936416184,
-      "loss": 0.2053,
-      "step": 1593
-    },
-    {
-      "epoch": 3.57,
-      "learning_rate": 0.00011098265895953756,
-      "loss": 0.192,
-      "step": 1594
-    },
-    {
-      "epoch": 3.58,
-      "learning_rate": 0.0001108092485549133,
-      "loss": 0.1852,
-      "step": 1595
-    },
-    {
-      "epoch": 3.58,
-      "learning_rate": 0.000110635838150289,
-      "loss": 0.1857,
-      "step": 1596
-    },
-    {
-      "epoch": 3.58,
-      "learning_rate": 0.00011046242774566474,
-      "loss": 0.1598,
-      "step": 1597
-    },
-    {
-      "epoch": 3.58,
-      "learning_rate": 0.00011028901734104044,
-      "loss": 0.1605,
-      "step": 1598
-    },
-    {
-      "epoch": 3.59,
-      "learning_rate": 0.00011011560693641618,
-      "loss": 0.1873,
-      "step": 1599
-    },
-    {
-      "epoch": 3.59,
-      "learning_rate": 0.0001099421965317919,
-      "loss": 0.1568,
-      "step": 1600
-    },
-    {
-      "epoch": 3.59,
-      "learning_rate": 0.00010976878612716762,
-      "loss": 0.1084,
-      "step": 1601
-    },
-    {
-      "epoch": 3.59,
-      "learning_rate": 0.00010959537572254334,
-      "loss": 0.1516,
-      "step": 1602
-    },
-    {
-      "epoch": 3.59,
-      "learning_rate": 0.00010942196531791907,
-      "loss": 0.1708,
-      "step": 1603
-    },
-    {
-      "epoch": 3.6,
-      "learning_rate": 0.00010924855491329478,
-      "loss": 0.1408,
-      "step": 1604
-    },
-    {
-      "epoch": 3.6,
-      "learning_rate": 0.00010907514450867051,
-      "loss": 0.1296,
-      "step": 1605
-    },
-    {
-      "epoch": 3.6,
-      "learning_rate": 0.00010890173410404623,
-      "loss": 0.1297,
-      "step": 1606
-    },
-    {
-      "epoch": 3.6,
-      "learning_rate": 0.00010872832369942196,
-      "loss": 0.1442,
-      "step": 1607
-    },
-    {
-      "epoch": 3.61,
-      "learning_rate": 0.00010855491329479768,
-      "loss": 0.1224,
-      "step": 1608
-    },
-    {
-      "epoch": 3.61,
-      "learning_rate": 0.00010838150289017341,
-      "loss": 0.1211,
-      "step": 1609
-    },
-    {
-      "epoch": 3.61,
-      "learning_rate": 0.00010820809248554912,
-      "loss": 0.1337,
-      "step": 1610
-    },
-    {
-      "epoch": 3.61,
-      "learning_rate": 0.00010803468208092485,
-      "loss": 0.1649,
-      "step": 1611
-    },
-    {
-      "epoch": 3.61,
-      "learning_rate": 0.00010786127167630056,
-      "loss": 0.128,
-      "step": 1612
-    },
-    {
-      "epoch": 3.62,
-      "learning_rate": 0.00010768786127167629,
-      "loss": 0.1535,
-      "step": 1613
-    },
-    {
-      "epoch": 3.62,
-      "learning_rate": 0.00010751445086705201,
-      "loss": 0.1381,
-      "step": 1614
-    },
-    {
-      "epoch": 3.62,
-      "learning_rate": 0.00010734104046242773,
-      "loss": 0.1441,
-      "step": 1615
-    },
-    {
-      "epoch": 3.62,
-      "learning_rate": 0.00010716763005780346,
-      "loss": 0.1232,
-      "step": 1616
-    },
-    {
-      "epoch": 3.63,
-      "learning_rate": 0.00010699421965317919,
-      "loss": 0.1301,
-      "step": 1617
-    },
-    {
-      "epoch": 3.63,
-      "learning_rate": 0.0001068208092485549,
-      "loss": 0.1528,
-      "step": 1618
-    },
-    {
-      "epoch": 3.63,
-      "learning_rate": 0.00010664739884393063,
-      "loss": 0.1144,
-      "step": 1619
-    },
-    {
-      "epoch": 3.63,
-      "learning_rate": 0.00010647398843930635,
-      "loss": 0.1606,
-      "step": 1620
-    },
-    {
-      "epoch": 3.63,
-      "learning_rate": 0.00010630057803468207,
-      "loss": 0.4729,
-      "step": 1621
-    },
-    {
-      "epoch": 3.64,
-      "learning_rate": 0.00010612716763005779,
-      "loss": 0.127,
-      "step": 1622
-    },
-    {
-      "epoch": 3.64,
-      "learning_rate": 0.00010595375722543353,
-      "loss": 0.1248,
-      "step": 1623
-    },
-    {
-      "epoch": 3.64,
-      "learning_rate": 0.00010578034682080923,
-      "loss": 0.1386,
-      "step": 1624
-    },
-    {
-      "epoch": 3.64,
-      "learning_rate": 0.00010560693641618497,
-      "loss": 0.1429,
-      "step": 1625
-    },
-    {
-      "epoch": 3.65,
-      "learning_rate": 0.00010543352601156068,
-      "loss": 0.099,
-      "step": 1626
-    },
-    {
-      "epoch": 3.65,
-      "learning_rate": 0.00010526011560693641,
-      "loss": 0.148,
-      "step": 1627
-    },
-    {
-      "epoch": 3.65,
-      "learning_rate": 0.00010508670520231213,
-      "loss": 0.1268,
-      "step": 1628
-    },
-    {
-      "epoch": 3.65,
-      "learning_rate": 0.00010491329479768786,
-      "loss": 0.1344,
-      "step": 1629
-    },
-    {
-      "epoch": 3.65,
-      "learning_rate": 0.00010473988439306357,
-      "loss": 0.1011,
-      "step": 1630
-    },
-    {
-      "epoch": 3.66,
-      "learning_rate": 0.0001045664739884393,
-      "loss": 0.0999,
-      "step": 1631
-    },
-    {
-      "epoch": 3.66,
-      "learning_rate": 0.00010439306358381501,
-      "loss": 0.1456,
-      "step": 1632
-    },
-    {
-      "epoch": 3.66,
-      "learning_rate": 0.00010421965317919075,
-      "loss": 0.1284,
-      "step": 1633
-    },
-    {
-      "epoch": 3.66,
-      "learning_rate": 0.00010404624277456647,
-      "loss": 0.126,
-      "step": 1634
-    },
-    {
-      "epoch": 3.67,
-      "learning_rate": 0.00010387283236994219,
-      "loss": 0.1044,
-      "step": 1635
-    },
-    {
-      "epoch": 3.67,
-      "learning_rate": 0.00010369942196531791,
-      "loss": 0.1651,
-      "step": 1636
-    },
-    {
-      "epoch": 3.67,
-      "learning_rate": 0.00010352601156069364,
-      "loss": 0.1435,
-      "step": 1637
-    },
-    {
-      "epoch": 3.67,
-      "learning_rate": 0.00010335260115606935,
-      "loss": 0.2565,
-      "step": 1638
-    },
-    {
-      "epoch": 3.67,
-      "learning_rate": 0.00010317919075144509,
-      "loss": 0.2916,
-      "step": 1639
-    },
-    {
-      "epoch": 3.68,
-      "learning_rate": 0.00010300578034682079,
-      "loss": 0.1893,
-      "step": 1640
-    },
-    {
-      "epoch": 3.68,
-      "learning_rate": 0.00010283236994219653,
-      "loss": 0.202,
-      "step": 1641
-    },
-    {
-      "epoch": 3.68,
-      "learning_rate": 0.00010265895953757225,
-      "loss": 0.2011,
-      "step": 1642
-    },
-    {
-      "epoch": 3.68,
-      "learning_rate": 0.00010248554913294798,
-      "loss": 0.1397,
-      "step": 1643
-    },
-    {
-      "epoch": 3.69,
-      "learning_rate": 0.00010231213872832369,
-      "loss": 0.1991,
-      "step": 1644
-    },
-    {
-      "epoch": 3.69,
-      "learning_rate": 0.00010213872832369942,
-      "loss": 0.143,
-      "step": 1645
-    },
-    {
-      "epoch": 3.69,
-      "learning_rate": 0.00010196531791907513,
-      "loss": 0.181,
-      "step": 1646
-    },
-    {
-      "epoch": 3.69,
-      "learning_rate": 0.00010179190751445086,
-      "loss": 0.1497,
-      "step": 1647
-    },
-    {
-      "epoch": 3.7,
-      "learning_rate": 0.00010161849710982658,
-      "loss": 0.2158,
-      "step": 1648
-    },
-    {
-      "epoch": 3.7,
-      "learning_rate": 0.0001014450867052023,
-      "loss": 0.1534,
-      "step": 1649
-    },
-    {
-      "epoch": 3.7,
-      "learning_rate": 0.00010127167630057803,
-      "loss": 0.16,
-      "step": 1650
-    },
-    {
-      "epoch": 3.7,
-      "learning_rate": 0.00010109826589595376,
-      "loss": 0.1698,
-      "step": 1651
-    },
-    {
-      "epoch": 3.7,
-      "learning_rate": 0.00010092485549132947,
-      "loss": 0.1082,
-      "step": 1652
-    },
-    {
-      "epoch": 3.71,
-      "learning_rate": 0.0001007514450867052,
-      "loss": 0.1202,
-      "step": 1653
-    },
-    {
-      "epoch": 3.71,
-      "learning_rate": 0.00010057803468208092,
-      "loss": 0.1296,
-      "step": 1654
-    },
-    {
-      "epoch": 3.71,
-      "learning_rate": 0.00010040462427745664,
-      "loss": 0.1141,
-      "step": 1655
-    },
-    {
-      "epoch": 3.71,
-      "learning_rate": 0.00010023121387283236,
-      "loss": 0.1072,
-      "step": 1656
-    },
-    {
-      "epoch": 3.72,
-      "learning_rate": 0.0001000578034682081,
-      "loss": 0.1284,
-      "step": 1657
-    },
-    {
-      "epoch": 3.72,
-      "learning_rate": 9.98843930635838e-05,
-      "loss": 0.1583,
-      "step": 1658
-    },
-    {
-      "epoch": 3.72,
-      "learning_rate": 9.971098265895953e-05,
-      "loss": 0.1118,
-      "step": 1659
-    },
-    {
-      "epoch": 3.72,
-      "learning_rate": 9.953757225433525e-05,
-      "loss": 0.1163,
-      "step": 1660
-    },
-    {
-      "epoch": 3.72,
-      "learning_rate": 9.936416184971097e-05,
-      "loss": 0.0947,
-      "step": 1661
-    },
-    {
-      "epoch": 3.73,
-      "learning_rate": 9.91907514450867e-05,
-      "loss": 0.097,
-      "step": 1662
-    },
-    {
-      "epoch": 3.73,
-      "learning_rate": 9.901734104046241e-05,
-      "loss": 0.1164,
-      "step": 1663
-    },
-    {
-      "epoch": 3.73,
-      "learning_rate": 9.884393063583814e-05,
-      "loss": 0.1529,
-      "step": 1664
-    },
-    {
-      "epoch": 3.73,
-      "learning_rate": 9.867052023121385e-05,
-      "loss": 0.1325,
-      "step": 1665
-    },
-    {
-      "epoch": 3.74,
-      "learning_rate": 9.849710982658958e-05,
-      "loss": 0.1309,
-      "step": 1666
-    },
-    {
-      "epoch": 3.74,
-      "learning_rate": 9.83236994219653e-05,
-      "loss": 0.1572,
-      "step": 1667
-    },
-    {
-      "epoch": 3.74,
-      "learning_rate": 9.815028901734104e-05,
-      "loss": 0.1051,
-      "step": 1668
-    },
-    {
-      "epoch": 3.74,
-      "learning_rate": 9.797687861271675e-05,
-      "loss": 0.1572,
-      "step": 1669
-    },
-    {
-      "epoch": 3.74,
-      "learning_rate": 9.780346820809248e-05,
-      "loss": 0.1192,
-      "step": 1670
-    },
-    {
-      "epoch": 3.75,
-      "learning_rate": 9.763005780346819e-05,
-      "loss": 0.1136,
-      "step": 1671
-    },
-    {
-      "epoch": 3.75,
-      "learning_rate": 9.745664739884392e-05,
-      "loss": 0.1003,
-      "step": 1672
-    },
-    {
-      "epoch": 3.75,
-      "learning_rate": 9.728323699421964e-05,
-      "loss": 0.1195,
-      "step": 1673
-    },
-    {
-      "epoch": 3.75,
-      "learning_rate": 9.710982658959536e-05,
-      "loss": 0.1246,
-      "step": 1674
-    },
-    {
-      "epoch": 3.76,
-      "learning_rate": 9.693641618497108e-05,
-      "loss": 0.1734,
-      "step": 1675
-    },
-    {
-      "epoch": 3.76,
-      "learning_rate": 9.676300578034682e-05,
-      "loss": 0.1257,
-      "step": 1676
-    },
-    {
-      "epoch": 3.76,
-      "learning_rate": 9.658959537572253e-05,
-      "loss": 0.122,
-      "step": 1677
-    },
-    {
-      "epoch": 3.76,
-      "learning_rate": 9.641618497109826e-05,
-      "loss": 0.1254,
-      "step": 1678
-    },
-    {
-      "epoch": 3.76,
-      "learning_rate": 9.624277456647398e-05,
-      "loss": 0.113,
-      "step": 1679
-    },
-    {
-      "epoch": 3.77,
-      "learning_rate": 9.60693641618497e-05,
-      "loss": 0.1115,
-      "step": 1680
-    },
-    {
-      "epoch": 3.77,
-      "learning_rate": 9.589595375722542e-05,
-      "loss": 0.1696,
-      "step": 1681
-    },
-    {
-      "epoch": 3.77,
-      "learning_rate": 9.572254335260116e-05,
-      "loss": 0.1251,
-      "step": 1682
-    },
-    {
-      "epoch": 3.77,
-      "learning_rate": 9.554913294797686e-05,
-      "loss": 0.0788,
-      "step": 1683
-    },
-    {
-      "epoch": 3.78,
-      "learning_rate": 9.53757225433526e-05,
-      "loss": 0.0783,
-      "step": 1684
-    },
-    {
-      "epoch": 3.78,
-      "learning_rate": 9.52023121387283e-05,
-      "loss": 0.0956,
-      "step": 1685
-    },
-    {
-      "epoch": 3.78,
-      "learning_rate": 9.502890173410404e-05,
-      "loss": 0.1233,
-      "step": 1686
-    },
-    {
-      "epoch": 3.78,
-      "learning_rate": 9.485549132947976e-05,
-      "loss": 0.1435,
-      "step": 1687
-    },
-    {
-      "epoch": 3.78,
-      "learning_rate": 9.468208092485548e-05,
-      "loss": 0.2561,
-      "step": 1688
-    },
-    {
-      "epoch": 3.79,
-      "learning_rate": 9.45086705202312e-05,
-      "loss": 0.308,
-      "step": 1689
-    },
-    {
-      "epoch": 3.79,
-      "learning_rate": 9.433526011560693e-05,
-      "loss": 0.2986,
-      "step": 1690
-    },
-    {
-      "epoch": 3.79,
-      "learning_rate": 9.416184971098264e-05,
-      "loss": 0.2049,
-      "step": 1691
-    },
-    {
-      "epoch": 3.79,
-      "learning_rate": 9.398843930635838e-05,
-      "loss": 0.1799,
-      "step": 1692
-    },
-    {
-      "epoch": 3.8,
-      "learning_rate": 9.38150289017341e-05,
-      "loss": 0.1759,
-      "step": 1693
-    },
-    {
-      "epoch": 3.8,
-      "learning_rate": 9.364161849710982e-05,
-      "loss": 0.1845,
-      "step": 1694
-    },
-    {
-      "epoch": 3.8,
-      "learning_rate": 9.346820809248554e-05,
-      "loss": 0.1882,
-      "step": 1695
-    },
-    {
-      "epoch": 3.8,
-      "learning_rate": 9.329479768786127e-05,
-      "loss": 0.1517,
-      "step": 1696
-    },
-    {
-      "epoch": 3.8,
-      "learning_rate": 9.312138728323698e-05,
-      "loss": 0.1269,
-      "step": 1697
-    },
-    {
-      "epoch": 3.81,
-      "learning_rate": 9.294797687861271e-05,
-      "loss": 0.1485,
-      "step": 1698
-    },
-    {
-      "epoch": 3.81,
-      "learning_rate": 9.277456647398842e-05,
-      "loss": 0.1608,
-      "step": 1699
-    },
-    {
-      "epoch": 3.81,
-      "learning_rate": 9.260115606936415e-05,
-      "loss": 0.1159,
-      "step": 1700
-    },
-    {
-      "epoch": 3.81,
-      "learning_rate": 9.242774566473988e-05,
-      "loss": 0.1598,
-      "step": 1701
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 9.22543352601156e-05,
-      "loss": 0.1181,
-      "step": 1702
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 9.208092485549132e-05,
-      "loss": 0.1478,
-      "step": 1703
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 9.190751445086705e-05,
-      "loss": 0.0908,
-      "step": 1704
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 9.173410404624276e-05,
-      "loss": 0.1451,
-      "step": 1705
-    },
-    {
-      "epoch": 3.83,
-      "learning_rate": 9.156069364161849e-05,
-      "loss": 0.1336,
-      "step": 1706
-    },
-    {
-      "epoch": 3.83,
-      "learning_rate": 9.138728323699421e-05,
-      "loss": 0.1112,
-      "step": 1707
-    },
-    {
-      "epoch": 3.83,
-      "learning_rate": 9.121387283236993e-05,
-      "loss": 0.1364,
-      "step": 1708
-    },
-    {
-      "epoch": 3.83,
-      "learning_rate": 9.104046242774565e-05,
-      "loss": 0.1621,
-      "step": 1709
-    },
-    {
-      "epoch": 3.83,
-      "learning_rate": 9.086705202312139e-05,
-      "loss": 0.1221,
-      "step": 1710
-    },
-    {
-      "epoch": 3.84,
-      "learning_rate": 9.06936416184971e-05,
-      "loss": 0.1516,
-      "step": 1711
-    },
-    {
-      "epoch": 3.84,
-      "learning_rate": 9.052023121387283e-05,
-      "loss": 0.1168,
-      "step": 1712
-    },
-    {
-      "epoch": 3.84,
-      "learning_rate": 9.034682080924854e-05,
-      "loss": 0.1298,
-      "step": 1713
-    },
-    {
-      "epoch": 3.84,
-      "learning_rate": 9.017341040462427e-05,
-      "loss": 0.1345,
-      "step": 1714
-    },
-    {
-      "epoch": 3.85,
-      "learning_rate": 8.999999999999999e-05,
-      "loss": 0.1037,
-      "step": 1715
-    },
-    {
-      "epoch": 3.85,
-      "learning_rate": 8.982658959537573e-05,
-      "loss": 0.1118,
-      "step": 1716
-    },
-    {
-      "epoch": 3.85,
-      "learning_rate": 8.965317919075143e-05,
-      "loss": 0.1032,
-      "step": 1717
-    },
-    {
-      "epoch": 3.85,
-      "learning_rate": 8.947976878612717e-05,
-      "loss": 0.1485,
-      "step": 1718
-    },
-    {
-      "epoch": 3.85,
-      "learning_rate": 8.930635838150287e-05,
-      "loss": 0.1301,
-      "step": 1719
-    },
-    {
-      "epoch": 3.86,
-      "learning_rate": 8.913294797687861e-05,
-      "loss": 0.126,
-      "step": 1720
-    },
-    {
-      "epoch": 3.86,
-      "learning_rate": 8.895953757225433e-05,
-      "loss": 0.1263,
-      "step": 1721
-    },
-    {
-      "epoch": 3.86,
-      "learning_rate": 8.878612716763005e-05,
-      "loss": 0.1456,
-      "step": 1722
-    },
-    {
-      "epoch": 3.86,
-      "learning_rate": 8.861271676300577e-05,
-      "loss": 0.1013,
-      "step": 1723
-    },
-    {
-      "epoch": 3.87,
-      "learning_rate": 8.84393063583815e-05,
-      "loss": 0.0905,
-      "step": 1724
-    },
-    {
-      "epoch": 3.87,
-      "learning_rate": 8.826589595375721e-05,
-      "loss": 0.1383,
-      "step": 1725
-    },
-    {
-      "epoch": 3.87,
-      "learning_rate": 8.809248554913295e-05,
-      "loss": 0.0949,
-      "step": 1726
-    },
-    {
-      "epoch": 3.87,
-      "learning_rate": 8.791907514450865e-05,
-      "loss": 0.1031,
-      "step": 1727
-    },
-    {
-      "epoch": 3.87,
-      "learning_rate": 8.774566473988439e-05,
-      "loss": 0.0967,
-      "step": 1728
-    },
-    {
-      "epoch": 3.88,
-      "learning_rate": 8.757225433526011e-05,
-      "loss": 0.0964,
-      "step": 1729
-    },
-    {
-      "epoch": 3.88,
-      "learning_rate": 8.739884393063584e-05,
-      "loss": 0.1521,
-      "step": 1730
-    },
-    {
-      "epoch": 3.88,
-      "learning_rate": 8.722543352601155e-05,
-      "loss": 0.1167,
-      "step": 1731
-    },
-    {
-      "epoch": 3.88,
-      "learning_rate": 8.705202312138728e-05,
-      "loss": 0.1063,
-      "step": 1732
-    },
-    {
-      "epoch": 3.89,
-      "learning_rate": 8.687861271676299e-05,
-      "loss": 0.1102,
-      "step": 1733
-    },
-    {
-      "epoch": 3.89,
-      "learning_rate": 8.670520231213873e-05,
-      "loss": 0.1269,
-      "step": 1734
-    },
-    {
-      "epoch": 3.89,
-      "learning_rate": 8.653179190751445e-05,
-      "loss": 0.151,
-      "step": 1735
-    },
-    {
-      "epoch": 3.89,
-      "learning_rate": 8.635838150289017e-05,
-      "loss": 0.1504,
-      "step": 1736
-    },
-    {
-      "epoch": 3.89,
-      "learning_rate": 8.618497109826589e-05,
-      "loss": 0.1653,
-      "step": 1737
-    },
-    {
-      "epoch": 3.9,
-      "learning_rate": 8.601156069364162e-05,
-      "loss": 0.1394,
-      "step": 1738
-    },
-    {
-      "epoch": 3.9,
-      "learning_rate": 8.583815028901733e-05,
-      "loss": 0.2872,
-      "step": 1739
-    },
-    {
-      "epoch": 3.9,
-      "learning_rate": 8.566473988439306e-05,
-      "loss": 0.2162,
-      "step": 1740
-    },
-    {
-      "epoch": 3.9,
-      "learning_rate": 8.549132947976878e-05,
-      "loss": 0.1601,
-      "step": 1741
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 8.53179190751445e-05,
-      "loss": 0.1933,
-      "step": 1742
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 8.514450867052023e-05,
-      "loss": 0.1663,
-      "step": 1743
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 8.497109826589596e-05,
-      "loss": 0.1629,
-      "step": 1744
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 8.479768786127167e-05,
-      "loss": 0.1432,
-      "step": 1745
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 8.46242774566474e-05,
-      "loss": 0.1363,
-      "step": 1746
-    },
-    {
-      "epoch": 3.92,
-      "learning_rate": 8.445086705202311e-05,
-      "loss": 0.1099,
-      "step": 1747
-    },
-    {
-      "epoch": 3.92,
-      "learning_rate": 8.427745664739884e-05,
-      "loss": 0.1789,
-      "step": 1748
-    },
-    {
-      "epoch": 3.92,
-      "learning_rate": 8.410404624277456e-05,
-      "loss": 0.1364,
-      "step": 1749
-    },
-    {
-      "epoch": 3.92,
-      "learning_rate": 8.393063583815028e-05,
-      "loss": 0.1443,
-      "step": 1750
-    },
-    {
-      "epoch": 3.93,
-      "learning_rate": 8.3757225433526e-05,
-      "loss": 0.1386,
-      "step": 1751
-    },
-    {
-      "epoch": 3.93,
-      "learning_rate": 8.358381502890174e-05,
-      "loss": 0.1371,
-      "step": 1752
-    },
-    {
-      "epoch": 3.93,
-      "learning_rate": 8.341040462427745e-05,
-      "loss": 0.1255,
-      "step": 1753
-    },
-    {
-      "epoch": 3.93,
-      "learning_rate": 8.323699421965317e-05,
-      "loss": 0.1153,
-      "step": 1754
-    },
-    {
-      "epoch": 3.93,
-      "learning_rate": 8.30635838150289e-05,
-      "loss": 0.1359,
-      "step": 1755
-    },
-    {
-      "epoch": 3.94,
-      "learning_rate": 8.289017341040461e-05,
-      "loss": 0.1598,
-      "step": 1756
-    },
-    {
-      "epoch": 3.94,
-      "learning_rate": 8.271676300578034e-05,
-      "loss": 0.1175,
-      "step": 1757
-    },
-    {
-      "epoch": 3.94,
-      "learning_rate": 8.254335260115605e-05,
-      "loss": 0.0973,
-      "step": 1758
-    },
-    {
-      "epoch": 3.94,
-      "learning_rate": 8.236994219653178e-05,
-      "loss": 0.1235,
-      "step": 1759
-    },
-    {
-      "epoch": 3.95,
-      "learning_rate": 8.21965317919075e-05,
-      "loss": 0.128,
-      "step": 1760
-    },
-    {
-      "epoch": 3.95,
-      "learning_rate": 8.202312138728322e-05,
-      "loss": 0.1383,
-      "step": 1761
-    },
-    {
-      "epoch": 3.95,
-      "learning_rate": 8.184971098265895e-05,
-      "loss": 0.1162,
-      "step": 1762
-    },
-    {
-      "epoch": 3.95,
-      "learning_rate": 8.167630057803468e-05,
-      "loss": 0.1021,
-      "step": 1763
-    },
-    {
-      "epoch": 3.96,
-      "learning_rate": 8.150289017341039e-05,
-      "loss": 0.1703,
-      "step": 1764
-    },
-    {
-      "epoch": 3.96,
-      "learning_rate": 8.132947976878612e-05,
-      "loss": 0.1103,
-      "step": 1765
-    },
-    {
-      "epoch": 3.96,
-      "learning_rate": 8.115606936416184e-05,
-      "loss": 0.1656,
-      "step": 1766
-    },
-    {
-      "epoch": 3.96,
-      "learning_rate": 8.098265895953756e-05,
-      "loss": 0.1121,
-      "step": 1767
-    },
-    {
-      "epoch": 3.96,
-      "learning_rate": 8.080924855491328e-05,
-      "loss": 0.0983,
-      "step": 1768
-    },
-    {
-      "epoch": 3.97,
-      "learning_rate": 8.063583815028902e-05,
-      "loss": 0.1085,
-      "step": 1769
-    },
-    {
-      "epoch": 3.97,
-      "learning_rate": 8.046242774566472e-05,
-      "loss": 0.1103,
-      "step": 1770
-    },
-    {
-      "epoch": 3.97,
-      "learning_rate": 8.028901734104046e-05,
-      "loss": 0.1093,
-      "step": 1771
-    },
-    {
-      "epoch": 3.97,
-      "learning_rate": 8.011560693641617e-05,
-      "loss": 0.097,
-      "step": 1772
-    },
-    {
-      "epoch": 3.98,
-      "learning_rate": 7.99421965317919e-05,
-      "loss": 0.102,
-      "step": 1773
-    },
-    {
-      "epoch": 3.98,
-      "learning_rate": 7.976878612716762e-05,
-      "loss": 0.1105,
-      "step": 1774
-    },
-    {
-      "epoch": 3.98,
-      "learning_rate": 7.959537572254334e-05,
-      "loss": 0.1144,
-      "step": 1775
-    },
-    {
-      "epoch": 3.98,
-      "learning_rate": 7.942196531791906e-05,
-      "loss": 0.1052,
-      "step": 1776
-    },
-    {
-      "epoch": 3.98,
-      "learning_rate": 7.92485549132948e-05,
-      "loss": 0.1278,
-      "step": 1777
-    },
-    {
-      "epoch": 3.99,
-      "learning_rate": 7.90751445086705e-05,
-      "loss": 0.1123,
-      "step": 1778
-    },
-    {
-      "epoch": 3.99,
-      "learning_rate": 7.890173410404624e-05,
-      "loss": 0.1038,
-      "step": 1779
-    },
-    {
-      "epoch": 3.99,
-      "learning_rate": 7.872832369942196e-05,
-      "loss": 0.1107,
-      "step": 1780
-    },
-    {
-      "epoch": 3.99,
-      "learning_rate": 7.855491329479768e-05,
-      "loss": 0.0981,
-      "step": 1781
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 7.83815028901734e-05,
-      "loss": 0.0696,
-      "step": 1782
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 7.820809248554913e-05,
-      "loss": 0.0678,
-      "step": 1783
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 7.803468208092484e-05,
-      "loss": 0.1774,
-      "step": 1784
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 7.786127167630058e-05,
-      "loss": 0.1757,
-      "step": 1785
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 7.768786127167628e-05,
-      "loss": 0.1199,
-      "step": 1786
-    },
-    {
-      "epoch": 4.01,
-      "learning_rate": 7.751445086705202e-05,
-      "loss": 0.1245,
-      "step": 1787
-    },
-    {
-      "epoch": 4.01,
-      "learning_rate": 7.734104046242774e-05,
-      "loss": 0.1052,
-      "step": 1788
-    },
-    {
-      "epoch": 4.01,
-      "learning_rate": 7.716763005780346e-05,
-      "loss": 0.0766,
-      "step": 1789
-    },
-    {
-      "epoch": 4.01,
-      "learning_rate": 7.699421965317918e-05,
-      "loss": 0.1122,
-      "step": 1790
-    },
-    {
-      "epoch": 4.02,
-      "learning_rate": 7.682080924855491e-05,
-      "loss": 0.0899,
-      "step": 1791
-    },
-    {
-      "epoch": 4.02,
-      "learning_rate": 7.664739884393062e-05,
-      "loss": 0.0932,
-      "step": 1792
-    },
-    {
-      "epoch": 4.02,
-      "learning_rate": 7.647398843930635e-05,
-      "loss": 0.1327,
-      "step": 1793
-    },
-    {
-      "epoch": 4.02,
-      "learning_rate": 7.630057803468207e-05,
-      "loss": 0.1113,
-      "step": 1794
-    },
-    {
-      "epoch": 4.02,
-      "learning_rate": 7.61271676300578e-05,
-      "loss": 0.087,
-      "step": 1795
-    },
-    {
-      "epoch": 4.03,
-      "learning_rate": 7.595375722543352e-05,
-      "loss": 0.0836,
-      "step": 1796
-    },
-    {
-      "epoch": 4.03,
-      "learning_rate": 7.578034682080925e-05,
-      "loss": 0.0778,
-      "step": 1797
-    },
-    {
-      "epoch": 4.03,
-      "learning_rate": 7.560693641618496e-05,
-      "loss": 0.0863,
-      "step": 1798
-    },
-    {
-      "epoch": 4.03,
-      "learning_rate": 7.543352601156069e-05,
-      "loss": 0.083,
-      "step": 1799
-    },
-    {
-      "epoch": 4.04,
-      "learning_rate": 7.52601156069364e-05,
-      "loss": 0.1162,
-      "step": 1800
-    },
-    {
-      "epoch": 4.04,
-      "learning_rate": 7.508670520231213e-05,
-      "loss": 0.0966,
-      "step": 1801
-    },
-    {
-      "epoch": 4.04,
-      "learning_rate": 7.491329479768785e-05,
-      "loss": 0.0732,
-      "step": 1802
-    },
-    {
-      "epoch": 4.04,
-      "learning_rate": 7.473988439306357e-05,
-      "loss": 0.0816,
-      "step": 1803
-    },
-    {
-      "epoch": 4.04,
-      "learning_rate": 7.45664739884393e-05,
-      "loss": 0.0554,
-      "step": 1804
-    },
-    {
-      "epoch": 4.05,
-      "learning_rate": 7.439306358381502e-05,
-      "loss": 0.087,
-      "step": 1805
-    },
-    {
-      "epoch": 4.05,
-      "learning_rate": 7.421965317919074e-05,
-      "loss": 0.0752,
-      "step": 1806
-    },
-    {
-      "epoch": 4.05,
-      "learning_rate": 7.404624277456646e-05,
-      "loss": 0.0659,
-      "step": 1807
-    },
-    {
-      "epoch": 4.05,
-      "learning_rate": 7.387283236994219e-05,
-      "loss": 0.0671,
-      "step": 1808
-    },
-    {
-      "epoch": 4.06,
-      "learning_rate": 7.369942196531791e-05,
-      "loss": 0.0664,
-      "step": 1809
-    },
-    {
-      "epoch": 4.06,
-      "learning_rate": 7.352601156069363e-05,
-      "loss": 0.0496,
-      "step": 1810
-    },
-    {
-      "epoch": 4.06,
-      "learning_rate": 7.335260115606935e-05,
-      "loss": 0.0615,
-      "step": 1811
-    },
-    {
-      "epoch": 4.06,
-      "learning_rate": 7.317919075144507e-05,
-      "loss": 0.0902,
-      "step": 1812
-    },
-    {
-      "epoch": 4.07,
-      "learning_rate": 7.30057803468208e-05,
-      "loss": 0.0814,
-      "step": 1813
-    },
-    {
-      "epoch": 4.07,
-      "learning_rate": 7.283236994219653e-05,
-      "loss": 0.0685,
-      "step": 1814
-    },
-    {
-      "epoch": 4.07,
-      "learning_rate": 7.265895953757225e-05,
-      "loss": 0.0804,
-      "step": 1815
-    },
-    {
-      "epoch": 4.07,
-      "learning_rate": 7.248554913294797e-05,
-      "loss": 0.0699,
-      "step": 1816
-    },
-    {
-      "epoch": 4.07,
-      "learning_rate": 7.231213872832369e-05,
-      "loss": 0.0642,
-      "step": 1817
-    },
-    {
-      "epoch": 4.08,
-      "learning_rate": 7.213872832369941e-05,
-      "loss": 0.0728,
-      "step": 1818
-    },
-    {
-      "epoch": 4.08,
-      "learning_rate": 7.196531791907513e-05,
-      "loss": 0.0582,
-      "step": 1819
-    },
-    {
-      "epoch": 4.08,
-      "learning_rate": 7.179190751445085e-05,
-      "loss": 0.0681,
-      "step": 1820
-    },
-    {
-      "epoch": 4.08,
-      "learning_rate": 7.161849710982659e-05,
-      "loss": 0.055,
-      "step": 1821
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 7.144508670520231e-05,
-      "loss": 0.053,
-      "step": 1822
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 7.127167630057803e-05,
-      "loss": 0.0802,
-      "step": 1823
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 7.109826589595375e-05,
-      "loss": 0.0759,
-      "step": 1824
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 7.092485549132947e-05,
-      "loss": 0.0842,
-      "step": 1825
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 7.075144508670519e-05,
-      "loss": 0.0695,
-      "step": 1826
-    },
-    {
-      "epoch": 4.1,
-      "learning_rate": 7.057803468208091e-05,
-      "loss": 0.0674,
-      "step": 1827
-    },
-    {
-      "epoch": 4.1,
-      "learning_rate": 7.040462427745665e-05,
-      "loss": 0.522,
-      "step": 1828
-    },
-    {
-      "epoch": 4.1,
-      "learning_rate": 7.023121387283237e-05,
-      "loss": 0.0617,
-      "step": 1829
-    },
-    {
-      "epoch": 4.1,
-      "learning_rate": 7.005780346820809e-05,
-      "loss": 0.074,
-      "step": 1830
-    },
-    {
-      "epoch": 4.11,
-      "learning_rate": 6.988439306358381e-05,
-      "loss": 0.0635,
-      "step": 1831
-    },
-    {
-      "epoch": 4.11,
-      "learning_rate": 6.971098265895953e-05,
-      "loss": 0.0933,
-      "step": 1832
-    },
-    {
-      "epoch": 4.11,
-      "learning_rate": 6.953757225433525e-05,
-      "loss": 0.0421,
-      "step": 1833
-    },
-    {
-      "epoch": 4.11,
-      "learning_rate": 6.936416184971097e-05,
-      "loss": 0.1101,
-      "step": 1834
-    },
-    {
-      "epoch": 4.11,
-      "learning_rate": 6.91907514450867e-05,
-      "loss": 0.1471,
-      "step": 1835
-    },
-    {
-      "epoch": 4.12,
-      "learning_rate": 6.901734104046242e-05,
-      "loss": 0.1476,
-      "step": 1836
-    },
-    {
-      "epoch": 4.12,
-      "learning_rate": 6.884393063583815e-05,
-      "loss": 0.0971,
-      "step": 1837
-    },
-    {
-      "epoch": 4.12,
-      "learning_rate": 6.867052023121387e-05,
-      "loss": 0.092,
-      "step": 1838
-    },
-    {
-      "epoch": 4.12,
-      "learning_rate": 6.849710982658959e-05,
-      "loss": 0.1004,
-      "step": 1839
-    },
-    {
-      "epoch": 4.13,
-      "learning_rate": 6.832369942196531e-05,
-      "loss": 0.1173,
-      "step": 1840
-    },
-    {
-      "epoch": 4.13,
-      "learning_rate": 6.815028901734103e-05,
-      "loss": 0.091,
-      "step": 1841
-    },
-    {
-      "epoch": 4.13,
-      "learning_rate": 6.797687861271676e-05,
-      "loss": 0.1089,
-      "step": 1842
-    },
-    {
-      "epoch": 4.13,
-      "learning_rate": 6.780346820809248e-05,
-      "loss": 0.0859,
-      "step": 1843
-    },
-    {
-      "epoch": 4.13,
-      "learning_rate": 6.76300578034682e-05,
-      "loss": 0.0726,
-      "step": 1844
-    },
-    {
-      "epoch": 4.14,
-      "learning_rate": 6.745664739884392e-05,
-      "loss": 0.084,
-      "step": 1845
-    },
-    {
-      "epoch": 4.14,
-      "learning_rate": 6.728323699421964e-05,
-      "loss": 0.0684,
-      "step": 1846
-    },
-    {
-      "epoch": 4.14,
-      "learning_rate": 6.710982658959537e-05,
-      "loss": 0.0713,
-      "step": 1847
-    },
-    {
-      "epoch": 4.14,
-      "learning_rate": 6.693641618497109e-05,
-      "loss": 0.0874,
-      "step": 1848
-    },
-    {
-      "epoch": 4.15,
-      "learning_rate": 6.676300578034682e-05,
-      "loss": 0.0691,
-      "step": 1849
-    },
-    {
-      "epoch": 4.15,
-      "learning_rate": 6.658959537572254e-05,
-      "loss": 0.0745,
-      "step": 1850
-    },
-    {
-      "epoch": 4.15,
-      "learning_rate": 6.641618497109826e-05,
-      "loss": 0.0953,
-      "step": 1851
-    },
-    {
-      "epoch": 4.15,
-      "learning_rate": 6.624277456647398e-05,
-      "loss": 0.0935,
-      "step": 1852
-    },
-    {
-      "epoch": 4.15,
-      "learning_rate": 6.60693641618497e-05,
-      "loss": 0.0775,
-      "step": 1853
-    },
-    {
-      "epoch": 4.16,
-      "learning_rate": 6.589595375722542e-05,
-      "loss": 0.0828,
-      "step": 1854
-    },
-    {
-      "epoch": 4.16,
-      "learning_rate": 6.572254335260114e-05,
-      "loss": 0.1012,
-      "step": 1855
-    },
-    {
-      "epoch": 4.16,
-      "learning_rate": 6.554913294797688e-05,
-      "loss": 0.1145,
-      "step": 1856
-    },
-    {
-      "epoch": 4.16,
-      "learning_rate": 6.53757225433526e-05,
-      "loss": 0.0819,
-      "step": 1857
-    },
-    {
-      "epoch": 4.17,
-      "learning_rate": 6.520231213872832e-05,
-      "loss": 0.0653,
-      "step": 1858
-    },
-    {
-      "epoch": 4.17,
-      "learning_rate": 6.502890173410404e-05,
-      "loss": 0.095,
-      "step": 1859
-    },
-    {
-      "epoch": 4.17,
-      "learning_rate": 6.485549132947976e-05,
-      "loss": 0.0649,
-      "step": 1860
-    },
-    {
-      "epoch": 4.17,
-      "learning_rate": 6.468208092485548e-05,
-      "loss": 0.0629,
-      "step": 1861
-    },
-    {
-      "epoch": 4.17,
-      "learning_rate": 6.45086705202312e-05,
-      "loss": 0.0555,
-      "step": 1862
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 6.433526011560694e-05,
-      "loss": 0.0947,
-      "step": 1863
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 6.416184971098266e-05,
-      "loss": 0.0661,
-      "step": 1864
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 6.398843930635838e-05,
-      "loss": 0.0607,
-      "step": 1865
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 6.38150289017341e-05,
-      "loss": 0.0694,
-      "step": 1866
-    },
-    {
-      "epoch": 4.19,
-      "learning_rate": 6.364161849710982e-05,
-      "loss": 0.0924,
-      "step": 1867
-    },
-    {
-      "epoch": 4.19,
-      "learning_rate": 6.346820809248554e-05,
-      "loss": 0.089,
-      "step": 1868
-    },
-    {
-      "epoch": 4.19,
-      "learning_rate": 6.329479768786126e-05,
-      "loss": 0.0783,
-      "step": 1869
-    },
-    {
-      "epoch": 4.19,
-      "learning_rate": 6.3121387283237e-05,
-      "loss": 0.0567,
-      "step": 1870
-    },
-    {
-      "epoch": 4.2,
-      "learning_rate": 6.294797687861272e-05,
-      "loss": 0.0759,
-      "step": 1871
-    },
-    {
-      "epoch": 4.2,
-      "learning_rate": 6.277456647398844e-05,
-      "loss": 0.0725,
-      "step": 1872
-    },
-    {
-      "epoch": 4.2,
-      "learning_rate": 6.260115606936416e-05,
-      "loss": 0.0885,
-      "step": 1873
-    },
-    {
-      "epoch": 4.2,
-      "learning_rate": 6.242774566473988e-05,
-      "loss": 0.0624,
-      "step": 1874
-    },
-    {
-      "epoch": 4.2,
-      "learning_rate": 6.22543352601156e-05,
-      "loss": 0.0758,
-      "step": 1875
-    },
-    {
-      "epoch": 4.21,
-      "learning_rate": 6.208092485549133e-05,
-      "loss": 0.0935,
-      "step": 1876
-    },
-    {
-      "epoch": 4.21,
-      "learning_rate": 6.190751445086705e-05,
-      "loss": 0.0601,
-      "step": 1877
-    },
-    {
-      "epoch": 4.21,
-      "learning_rate": 6.173410404624277e-05,
-      "loss": 0.0447,
-      "step": 1878
-    },
-    {
-      "epoch": 4.21,
-      "learning_rate": 6.15606936416185e-05,
-      "loss": 0.0657,
-      "step": 1879
-    },
-    {
-      "epoch": 4.22,
-      "learning_rate": 6.138728323699422e-05,
-      "loss": 0.06,
-      "step": 1880
-    },
-    {
-      "epoch": 4.22,
-      "learning_rate": 6.121387283236994e-05,
-      "loss": 0.0806,
-      "step": 1881
-    },
-    {
-      "epoch": 4.22,
-      "learning_rate": 6.104046242774566e-05,
-      "loss": 0.096,
-      "step": 1882
-    },
-    {
-      "epoch": 4.22,
-      "learning_rate": 6.0867052023121385e-05,
-      "loss": 0.0729,
-      "step": 1883
-    },
-    {
-      "epoch": 4.22,
-      "learning_rate": 6.0693641618497105e-05,
-      "loss": 0.1086,
-      "step": 1884
-    },
-    {
-      "epoch": 4.23,
-      "learning_rate": 6.052023121387283e-05,
-      "loss": 0.1859,
-      "step": 1885
-    },
-    {
-      "epoch": 4.23,
-      "learning_rate": 6.034682080924855e-05,
-      "loss": 0.1361,
-      "step": 1886
-    },
-    {
-      "epoch": 4.23,
-      "learning_rate": 6.0173410404624274e-05,
-      "loss": 0.1013,
-      "step": 1887
-    },
-    {
-      "epoch": 4.23,
-      "learning_rate": 5.9999999999999995e-05,
-      "loss": 0.0918,
-      "step": 1888
-    },
-    {
-      "epoch": 4.24,
-      "learning_rate": 5.982658959537572e-05,
-      "loss": 0.0916,
-      "step": 1889
-    },
-    {
-      "epoch": 4.24,
-      "learning_rate": 5.965317919075144e-05,
-      "loss": 0.0794,
-      "step": 1890
-    },
-    {
-      "epoch": 4.24,
-      "learning_rate": 5.9479768786127164e-05,
-      "loss": 0.1023,
-      "step": 1891
-    },
-    {
-      "epoch": 4.24,
-      "learning_rate": 5.930635838150289e-05,
-      "loss": 0.1047,
-      "step": 1892
-    },
-    {
-      "epoch": 4.24,
-      "learning_rate": 5.913294797687861e-05,
-      "loss": 0.0789,
-      "step": 1893
-    },
-    {
-      "epoch": 4.25,
-      "learning_rate": 5.895953757225433e-05,
-      "loss": 0.0798,
-      "step": 1894
-    },
-    {
-      "epoch": 4.25,
-      "learning_rate": 5.878612716763006e-05,
-      "loss": 0.0895,
-      "step": 1895
-    },
-    {
-      "epoch": 4.25,
-      "learning_rate": 5.861271676300578e-05,
-      "loss": 0.0754,
-      "step": 1896
-    },
-    {
-      "epoch": 4.25,
-      "learning_rate": 5.84393063583815e-05,
-      "loss": 0.1032,
-      "step": 1897
-    },
-    {
-      "epoch": 4.26,
-      "learning_rate": 5.8265895953757215e-05,
-      "loss": 0.0945,
-      "step": 1898
-    },
-    {
-      "epoch": 4.26,
-      "learning_rate": 5.8092485549132936e-05,
-      "loss": 0.0874,
-      "step": 1899
-    },
-    {
-      "epoch": 4.26,
-      "learning_rate": 5.791907514450866e-05,
-      "loss": 0.0859,
-      "step": 1900
-    },
-    {
-      "epoch": 4.26,
-      "learning_rate": 5.7745664739884384e-05,
-      "loss": 0.0794,
-      "step": 1901
-    },
-    {
-      "epoch": 4.26,
-      "learning_rate": 5.7572254335260105e-05,
-      "loss": 0.0629,
-      "step": 1902
-    },
-    {
-      "epoch": 4.27,
-      "learning_rate": 5.739884393063583e-05,
-      "loss": 0.087,
-      "step": 1903
-    },
-    {
-      "epoch": 4.27,
-      "learning_rate": 5.722543352601155e-05,
-      "loss": 0.0803,
-      "step": 1904
-    },
-    {
-      "epoch": 4.27,
-      "learning_rate": 5.705202312138727e-05,
-      "loss": 0.0901,
-      "step": 1905
-    },
-    {
-      "epoch": 4.27,
-      "learning_rate": 5.6878612716762994e-05,
-      "loss": 0.1085,
-      "step": 1906
-    },
-    {
-      "epoch": 4.28,
-      "learning_rate": 5.670520231213872e-05,
-      "loss": 0.0724,
-      "step": 1907
-    },
-    {
-      "epoch": 4.28,
-      "learning_rate": 5.653179190751444e-05,
-      "loss": 0.092,
-      "step": 1908
-    },
-    {
-      "epoch": 4.28,
-      "learning_rate": 5.635838150289016e-05,
-      "loss": 0.0678,
-      "step": 1909
-    },
-    {
-      "epoch": 4.28,
-      "learning_rate": 5.618497109826589e-05,
-      "loss": 0.0715,
-      "step": 1910
-    },
-    {
-      "epoch": 4.28,
-      "learning_rate": 5.601156069364161e-05,
-      "loss": 0.0755,
-      "step": 1911
-    },
-    {
-      "epoch": 4.29,
-      "learning_rate": 5.583815028901733e-05,
-      "loss": 0.0718,
-      "step": 1912
-    },
-    {
-      "epoch": 4.29,
-      "learning_rate": 5.566473988439306e-05,
-      "loss": 0.0771,
-      "step": 1913
-    },
-    {
-      "epoch": 4.29,
-      "learning_rate": 5.549132947976878e-05,
-      "loss": 0.0682,
-      "step": 1914
-    },
-    {
-      "epoch": 4.29,
-      "learning_rate": 5.53179190751445e-05,
-      "loss": 0.087,
-      "step": 1915
-    },
-    {
-      "epoch": 4.3,
-      "learning_rate": 5.514450867052022e-05,
-      "loss": 0.0566,
-      "step": 1916
-    },
-    {
-      "epoch": 4.3,
-      "learning_rate": 5.497109826589595e-05,
-      "loss": 0.0778,
-      "step": 1917
-    },
-    {
-      "epoch": 4.3,
-      "learning_rate": 5.479768786127167e-05,
-      "loss": 0.0692,
-      "step": 1918
-    },
-    {
-      "epoch": 4.3,
-      "learning_rate": 5.462427745664739e-05,
-      "loss": 0.0559,
-      "step": 1919
-    },
-    {
-      "epoch": 4.3,
-      "learning_rate": 5.445086705202312e-05,
-      "loss": 0.0472,
-      "step": 1920
-    },
-    {
-      "epoch": 4.31,
-      "learning_rate": 5.427745664739884e-05,
-      "loss": 0.0515,
-      "step": 1921
-    },
-    {
-      "epoch": 4.31,
-      "learning_rate": 5.410404624277456e-05,
-      "loss": 0.0699,
-      "step": 1922
-    },
-    {
-      "epoch": 4.31,
-      "learning_rate": 5.393063583815028e-05,
-      "loss": 0.0601,
-      "step": 1923
-    },
-    {
-      "epoch": 4.31,
-      "learning_rate": 5.375722543352601e-05,
-      "loss": 0.0811,
-      "step": 1924
-    },
-    {
-      "epoch": 4.32,
-      "learning_rate": 5.358381502890173e-05,
-      "loss": 0.0418,
-      "step": 1925
-    },
-    {
-      "epoch": 4.32,
-      "learning_rate": 5.341040462427745e-05,
-      "loss": 0.0683,
-      "step": 1926
-    },
-    {
-      "epoch": 4.32,
-      "learning_rate": 5.3236994219653176e-05,
-      "loss": 0.0661,
-      "step": 1927
-    },
-    {
-      "epoch": 4.32,
-      "learning_rate": 5.3063583815028896e-05,
-      "loss": 0.098,
-      "step": 1928
-    },
-    {
-      "epoch": 4.33,
-      "learning_rate": 5.289017341040462e-05,
-      "loss": 0.0497,
-      "step": 1929
-    },
-    {
-      "epoch": 4.33,
-      "learning_rate": 5.271676300578034e-05,
-      "loss": 0.0389,
-      "step": 1930
-    },
-    {
-      "epoch": 4.33,
-      "learning_rate": 5.2543352601156065e-05,
-      "loss": 0.041,
-      "step": 1931
-    },
-    {
-      "epoch": 4.33,
-      "learning_rate": 5.2369942196531786e-05,
-      "loss": 0.0961,
-      "step": 1932
-    },
-    {
-      "epoch": 4.33,
-      "learning_rate": 5.2196531791907507e-05,
-      "loss": 0.0785,
-      "step": 1933
-    },
-    {
-      "epoch": 4.34,
-      "learning_rate": 5.2023121387283234e-05,
-      "loss": 0.1104,
-      "step": 1934
-    },
-    {
-      "epoch": 4.34,
-      "learning_rate": 5.1849710982658955e-05,
-      "loss": 0.124,
-      "step": 1935
-    },
-    {
-      "epoch": 4.34,
-      "learning_rate": 5.1676300578034675e-05,
-      "loss": 0.1314,
-      "step": 1936
-    },
-    {
-      "epoch": 4.34,
-      "learning_rate": 5.1502890173410396e-05,
-      "loss": 0.094,
-      "step": 1937
-    },
-    {
-      "epoch": 4.35,
-      "learning_rate": 5.1329479768786124e-05,
-      "loss": 0.0923,
-      "step": 1938
-    },
-    {
-      "epoch": 4.35,
-      "learning_rate": 5.1156069364161844e-05,
-      "loss": 0.1036,
-      "step": 1939
-    },
-    {
-      "epoch": 4.35,
-      "learning_rate": 5.0982658959537565e-05,
-      "loss": 0.0725,
-      "step": 1940
-    },
-    {
-      "epoch": 4.35,
-      "learning_rate": 5.080924855491329e-05,
-      "loss": 0.0853,
-      "step": 1941
-    },
-    {
-      "epoch": 4.35,
-      "learning_rate": 5.063583815028901e-05,
-      "loss": 0.1104,
-      "step": 1942
-    },
-    {
-      "epoch": 4.36,
-      "learning_rate": 5.0462427745664734e-05,
-      "loss": 0.0764,
-      "step": 1943
-    },
-    {
-      "epoch": 4.36,
-      "learning_rate": 5.028901734104046e-05,
-      "loss": 0.0781,
-      "step": 1944
-    },
-    {
-      "epoch": 4.36,
-      "learning_rate": 5.011560693641618e-05,
-      "loss": 0.0802,
-      "step": 1945
-    },
-    {
-      "epoch": 4.36,
-      "learning_rate": 4.99421965317919e-05,
-      "loss": 0.0705,
-      "step": 1946
-    },
-    {
-      "epoch": 4.37,
-      "learning_rate": 4.976878612716762e-05,
-      "loss": 0.0725,
-      "step": 1947
-    },
-    {
-      "epoch": 4.37,
-      "learning_rate": 4.959537572254335e-05,
-      "loss": 0.054,
-      "step": 1948
-    },
-    {
-      "epoch": 4.37,
-      "learning_rate": 4.942196531791907e-05,
-      "loss": 0.0575,
-      "step": 1949
-    },
-    {
-      "epoch": 4.37,
-      "learning_rate": 4.924855491329479e-05,
-      "loss": 0.0801,
-      "step": 1950
-    },
-    {
-      "epoch": 4.37,
-      "learning_rate": 4.907514450867052e-05,
-      "loss": 0.0763,
-      "step": 1951
-    },
-    {
-      "epoch": 4.38,
-      "learning_rate": 4.890173410404624e-05,
-      "loss": 0.0724,
-      "step": 1952
-    },
-    {
-      "epoch": 4.38,
-      "learning_rate": 4.872832369942196e-05,
-      "loss": 0.0776,
-      "step": 1953
-    },
-    {
-      "epoch": 4.38,
-      "learning_rate": 4.855491329479768e-05,
-      "loss": 0.0493,
-      "step": 1954
-    },
-    {
-      "epoch": 4.38,
-      "learning_rate": 4.838150289017341e-05,
-      "loss": 0.0714,
-      "step": 1955
-    },
-    {
-      "epoch": 4.39,
-      "learning_rate": 4.820809248554913e-05,
-      "loss": 0.0701,
-      "step": 1956
-    },
-    {
-      "epoch": 4.39,
-      "learning_rate": 4.803468208092485e-05,
-      "loss": 0.0742,
-      "step": 1957
-    },
-    {
-      "epoch": 4.39,
-      "learning_rate": 4.786127167630058e-05,
-      "loss": 0.0693,
-      "step": 1958
-    },
-    {
-      "epoch": 4.39,
-      "learning_rate": 4.76878612716763e-05,
-      "loss": 0.0687,
-      "step": 1959
-    },
-    {
-      "epoch": 4.39,
-      "learning_rate": 4.751445086705202e-05,
-      "loss": 0.0831,
-      "step": 1960
-    },
-    {
-      "epoch": 4.4,
-      "learning_rate": 4.734104046242774e-05,
-      "loss": 0.0627,
-      "step": 1961
-    },
-    {
-      "epoch": 4.4,
-      "learning_rate": 4.716763005780347e-05,
-      "loss": 0.0741,
-      "step": 1962
-    },
-    {
-      "epoch": 4.4,
-      "learning_rate": 4.699421965317919e-05,
-      "loss": 0.0666,
-      "step": 1963
-    },
-    {
-      "epoch": 4.4,
-      "learning_rate": 4.682080924855491e-05,
-      "loss": 0.0782,
-      "step": 1964
-    },
-    {
-      "epoch": 4.41,
-      "learning_rate": 4.6647398843930636e-05,
-      "loss": 0.0725,
-      "step": 1965
-    },
-    {
-      "epoch": 4.41,
-      "learning_rate": 4.647398843930636e-05,
-      "loss": 0.044,
-      "step": 1966
-    },
-    {
-      "epoch": 4.41,
-      "learning_rate": 4.630057803468208e-05,
-      "loss": 0.0721,
-      "step": 1967
-    },
-    {
-      "epoch": 4.41,
-      "learning_rate": 4.61271676300578e-05,
-      "loss": 0.0641,
-      "step": 1968
-    },
-    {
-      "epoch": 4.41,
-      "learning_rate": 4.5953757225433526e-05,
-      "loss": 0.0749,
-      "step": 1969
-    },
-    {
-      "epoch": 4.42,
-      "learning_rate": 4.5780346820809246e-05,
-      "loss": 0.0604,
-      "step": 1970
-    },
-    {
-      "epoch": 4.42,
-      "learning_rate": 4.560693641618497e-05,
-      "loss": 0.0612,
-      "step": 1971
-    },
-    {
-      "epoch": 4.42,
-      "learning_rate": 4.5433526011560694e-05,
-      "loss": 0.0562,
-      "step": 1972
-    },
-    {
-      "epoch": 4.42,
-      "learning_rate": 4.5260115606936415e-05,
-      "loss": 0.0648,
-      "step": 1973
-    },
-    {
-      "epoch": 4.43,
-      "learning_rate": 4.5086705202312136e-05,
-      "loss": 0.0514,
-      "step": 1974
-    },
-    {
-      "epoch": 4.43,
-      "learning_rate": 4.491329479768786e-05,
-      "loss": 0.0888,
-      "step": 1975
-    },
-    {
-      "epoch": 4.43,
-      "learning_rate": 4.4739884393063584e-05,
-      "loss": 0.0814,
-      "step": 1976
-    },
-    {
-      "epoch": 4.43,
-      "learning_rate": 4.4566473988439305e-05,
-      "loss": 0.0913,
-      "step": 1977
-    },
-    {
-      "epoch": 4.43,
-      "learning_rate": 4.4393063583815025e-05,
-      "loss": 0.0764,
-      "step": 1978
-    },
-    {
-      "epoch": 4.44,
-      "learning_rate": 4.421965317919075e-05,
-      "loss": 0.0806,
-      "step": 1979
-    },
-    {
-      "epoch": 4.44,
-      "learning_rate": 4.404624277456647e-05,
-      "loss": 0.0493,
-      "step": 1980
-    },
-    {
-      "epoch": 4.44,
-      "learning_rate": 4.3872832369942194e-05,
-      "loss": 0.0454,
-      "step": 1981
-    },
-    {
-      "epoch": 4.44,
-      "learning_rate": 4.369942196531792e-05,
-      "loss": 0.0552,
-      "step": 1982
-    },
-    {
-      "epoch": 4.45,
-      "learning_rate": 4.352601156069364e-05,
-      "loss": 0.0891,
-      "step": 1983
-    },
-    {
-      "epoch": 4.45,
-      "learning_rate": 4.335260115606936e-05,
-      "loss": 0.1034,
-      "step": 1984
-    },
-    {
-      "epoch": 4.45,
-      "learning_rate": 4.3179190751445084e-05,
-      "loss": 0.1403,
-      "step": 1985
-    },
-    {
-      "epoch": 4.45,
-      "learning_rate": 4.300578034682081e-05,
-      "loss": 0.0965,
-      "step": 1986
-    },
-    {
-      "epoch": 4.46,
-      "learning_rate": 4.283236994219653e-05,
-      "loss": 0.1017,
-      "step": 1987
-    },
-    {
-      "epoch": 4.46,
-      "learning_rate": 4.265895953757225e-05,
-      "loss": 0.1015,
-      "step": 1988
-    },
-    {
-      "epoch": 4.46,
-      "learning_rate": 4.248554913294798e-05,
-      "loss": 0.0803,
-      "step": 1989
-    },
-    {
-      "epoch": 4.46,
-      "learning_rate": 4.23121387283237e-05,
-      "loss": 0.0917,
-      "step": 1990
-    },
-    {
-      "epoch": 4.46,
-      "learning_rate": 4.213872832369942e-05,
-      "loss": 0.1152,
-      "step": 1991
-    },
-    {
-      "epoch": 4.47,
-      "learning_rate": 4.196531791907514e-05,
-      "loss": 0.0697,
-      "step": 1992
-    },
-    {
-      "epoch": 4.47,
-      "learning_rate": 4.179190751445087e-05,
-      "loss": 0.0904,
-      "step": 1993
-    },
-    {
-      "epoch": 4.47,
-      "learning_rate": 4.161849710982658e-05,
-      "loss": 0.0707,
-      "step": 1994
-    },
-    {
-      "epoch": 4.47,
-      "learning_rate": 4.1445086705202304e-05,
-      "loss": 0.0799,
-      "step": 1995
-    },
-    {
-      "epoch": 4.48,
-      "learning_rate": 4.1271676300578025e-05,
-      "loss": 0.0883,
-      "step": 1996
-    },
-    {
-      "epoch": 4.48,
-      "learning_rate": 4.109826589595375e-05,
-      "loss": 0.1033,
-      "step": 1997
-    },
-    {
-      "epoch": 4.48,
-      "learning_rate": 4.092485549132947e-05,
-      "loss": 0.0797,
-      "step": 1998
-    },
-    {
-      "epoch": 4.48,
-      "learning_rate": 4.075144508670519e-05,
-      "loss": 0.0984,
-      "step": 1999
-    },
-    {
-      "epoch": 4.48,
-      "learning_rate": 4.057803468208092e-05,
-      "loss": 0.0753,
-      "step": 2000
-    },
-    {
-      "epoch": 4.48,
-      "eval_loss": 0.32974740862846375,
-      "eval_runtime": 555.2371,
-      "eval_samples_per_second": 4.758,
-      "eval_steps_per_second": 0.596,
-      "eval_wer": 0.09492264974216581,
-      "step": 2000
-    },
-    {
-      "epoch": 4.49,
-      "learning_rate": 4.040462427745664e-05,
-      "loss": 0.0764,
-      "step": 2001
-    },
-    {
-      "epoch": 4.49,
-      "learning_rate": 4.023121387283236e-05,
-      "loss": 0.0685,
-      "step": 2002
-    },
-    {
-      "epoch": 4.49,
-      "learning_rate": 4.005780346820808e-05,
-      "loss": 0.096,
-      "step": 2003
-    },
-    {
-      "epoch": 4.49,
-      "learning_rate": 3.988439306358381e-05,
-      "loss": 0.0908,
-      "step": 2004
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 3.971098265895953e-05,
-      "loss": 0.0857,
-      "step": 2005
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 3.953757225433525e-05,
-      "loss": 0.0824,
-      "step": 2006
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 3.936416184971098e-05,
-      "loss": 0.0638,
-      "step": 2007
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 3.91907514450867e-05,
-      "loss": 0.0703,
-      "step": 2008
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 3.901734104046242e-05,
-      "loss": 0.0662,
-      "step": 2009
-    },
-    {
-      "epoch": 4.51,
-      "learning_rate": 3.884393063583814e-05,
-      "loss": 0.0581,
-      "step": 2010
-    },
-    {
-      "epoch": 4.51,
-      "learning_rate": 3.867052023121387e-05,
-      "loss": 0.0755,
-      "step": 2011
-    },
-    {
-      "epoch": 4.51,
-      "learning_rate": 3.849710982658959e-05,
-      "loss": 0.0439,
-      "step": 2012
-    },
-    {
-      "epoch": 4.51,
-      "learning_rate": 3.832369942196531e-05,
-      "loss": 0.0588,
-      "step": 2013
-    },
-    {
-      "epoch": 4.52,
-      "learning_rate": 3.815028901734104e-05,
-      "loss": 0.0649,
-      "step": 2014
-    },
-    {
-      "epoch": 4.52,
-      "learning_rate": 3.797687861271676e-05,
-      "loss": 0.0867,
-      "step": 2015
-    },
-    {
-      "epoch": 4.52,
-      "learning_rate": 3.780346820809248e-05,
-      "loss": 0.071,
-      "step": 2016
-    },
-    {
-      "epoch": 4.52,
-      "learning_rate": 3.76300578034682e-05,
-      "loss": 0.09,
-      "step": 2017
-    },
-    {
-      "epoch": 4.52,
-      "learning_rate": 3.745664739884393e-05,
-      "loss": 0.0448,
-      "step": 2018
-    },
-    {
-      "epoch": 4.53,
-      "learning_rate": 3.728323699421965e-05,
-      "loss": 0.0613,
-      "step": 2019
-    },
-    {
-      "epoch": 4.53,
-      "learning_rate": 3.710982658959537e-05,
-      "loss": 0.0503,
-      "step": 2020
-    },
-    {
-      "epoch": 4.53,
-      "learning_rate": 3.6936416184971096e-05,
-      "loss": 0.0799,
-      "step": 2021
-    },
-    {
-      "epoch": 4.53,
-      "learning_rate": 3.6763005780346816e-05,
-      "loss": 0.0628,
-      "step": 2022
-    },
-    {
-      "epoch": 4.54,
-      "learning_rate": 3.658959537572254e-05,
-      "loss": 0.0486,
-      "step": 2023
-    },
-    {
-      "epoch": 4.54,
-      "learning_rate": 3.6416184971098265e-05,
-      "loss": 0.0715,
-      "step": 2024
-    },
-    {
-      "epoch": 4.54,
-      "learning_rate": 3.6242774566473985e-05,
-      "loss": 0.0677,
-      "step": 2025
-    },
-    {
-      "epoch": 4.54,
-      "learning_rate": 3.6069364161849706e-05,
-      "loss": 0.063,
-      "step": 2026
-    },
-    {
-      "epoch": 4.54,
-      "learning_rate": 3.5895953757225427e-05,
-      "loss": 0.0659,
-      "step": 2027
-    },
-    {
-      "epoch": 4.55,
-      "learning_rate": 3.5722543352601154e-05,
-      "loss": 0.7377,
-      "step": 2028
-    },
-    {
-      "epoch": 4.55,
-      "learning_rate": 3.5549132947976875e-05,
-      "loss": 0.093,
-      "step": 2029
-    },
-    {
-      "epoch": 4.55,
-      "learning_rate": 3.5375722543352595e-05,
-      "loss": 0.0447,
-      "step": 2030
-    },
-    {
-      "epoch": 4.55,
-      "learning_rate": 3.520231213872832e-05,
-      "loss": 0.0416,
-      "step": 2031
-    },
-    {
-      "epoch": 4.56,
-      "learning_rate": 3.5028901734104043e-05,
-      "loss": 0.0708,
-      "step": 2032
-    },
-    {
-      "epoch": 4.56,
-      "learning_rate": 3.4855491329479764e-05,
-      "loss": 0.0784,
-      "step": 2033
-    },
-    {
-      "epoch": 4.56,
-      "learning_rate": 3.4682080924855485e-05,
-      "loss": 0.108,
-      "step": 2034
-    },
-    {
-      "epoch": 4.56,
-      "learning_rate": 3.450867052023121e-05,
-      "loss": 0.1217,
-      "step": 2035
-    },
-    {
-      "epoch": 4.57,
-      "learning_rate": 3.433526011560693e-05,
-      "loss": 0.1173,
-      "step": 2036
-    },
-    {
-      "epoch": 4.57,
-      "learning_rate": 3.4161849710982654e-05,
-      "loss": 0.1258,
-      "step": 2037
-    },
-    {
-      "epoch": 4.57,
-      "learning_rate": 3.398843930635838e-05,
-      "loss": 0.0868,
-      "step": 2038
-    },
-    {
-      "epoch": 4.57,
-      "learning_rate": 3.38150289017341e-05,
-      "loss": 0.1049,
-      "step": 2039
-    },
-    {
-      "epoch": 4.57,
-      "learning_rate": 3.364161849710982e-05,
-      "loss": 0.0621,
-      "step": 2040
-    },
-    {
-      "epoch": 4.58,
-      "learning_rate": 3.346820809248554e-05,
-      "loss": 0.0679,
-      "step": 2041
-    },
-    {
-      "epoch": 4.58,
-      "learning_rate": 3.329479768786127e-05,
-      "loss": 0.0786,
-      "step": 2042
-    },
-    {
-      "epoch": 4.58,
-      "learning_rate": 3.312138728323699e-05,
-      "loss": 0.0961,
-      "step": 2043
-    },
-    {
-      "epoch": 4.58,
-      "learning_rate": 3.294797687861271e-05,
-      "loss": 0.098,
-      "step": 2044
-    },
-    {
-      "epoch": 4.59,
-      "learning_rate": 3.277456647398844e-05,
-      "loss": 0.0582,
-      "step": 2045
-    },
-    {
-      "epoch": 4.59,
-      "learning_rate": 3.260115606936416e-05,
-      "loss": 0.0676,
-      "step": 2046
-    },
-    {
-      "epoch": 4.59,
-      "learning_rate": 3.242774566473988e-05,
-      "loss": 0.0756,
-      "step": 2047
-    },
-    {
-      "epoch": 4.59,
-      "learning_rate": 3.22543352601156e-05,
-      "loss": 0.0821,
-      "step": 2048
-    },
-    {
-      "epoch": 4.59,
-      "learning_rate": 3.208092485549133e-05,
-      "loss": 0.0867,
-      "step": 2049
-    },
-    {
-      "epoch": 4.6,
-      "learning_rate": 3.190751445086705e-05,
-      "loss": 0.0538,
-      "step": 2050
-    },
-    {
-      "epoch": 4.6,
-      "learning_rate": 3.173410404624277e-05,
-      "loss": 0.0831,
-      "step": 2051
-    },
-    {
-      "epoch": 4.6,
-      "learning_rate": 3.15606936416185e-05,
-      "loss": 0.0608,
-      "step": 2052
-    },
-    {
-      "epoch": 4.6,
-      "learning_rate": 3.138728323699422e-05,
-      "loss": 0.067,
-      "step": 2053
-    },
-    {
-      "epoch": 4.61,
-      "learning_rate": 3.121387283236994e-05,
-      "loss": 0.0615,
-      "step": 2054
-    },
-    {
-      "epoch": 4.61,
-      "learning_rate": 3.1040462427745667e-05,
-      "loss": 0.0613,
-      "step": 2055
-    },
-    {
-      "epoch": 4.61,
-      "learning_rate": 3.086705202312139e-05,
-      "loss": 0.0709,
-      "step": 2056
-    },
-    {
-      "epoch": 4.61,
-      "learning_rate": 3.069364161849711e-05,
-      "loss": 0.0792,
-      "step": 2057
-    },
-    {
-      "epoch": 4.61,
-      "learning_rate": 3.052023121387283e-05,
-      "loss": 0.0584,
-      "step": 2058
-    },
-    {
-      "epoch": 4.62,
-      "learning_rate": 3.0346820809248553e-05,
-      "loss": 0.071,
-      "step": 2059
-    },
-    {
-      "epoch": 4.62,
-      "learning_rate": 3.0173410404624277e-05,
-      "loss": 0.0718,
-      "step": 2060
-    },
-    {
-      "epoch": 4.62,
-      "learning_rate": 2.9999999999999997e-05,
-      "loss": 0.0671,
-      "step": 2061
-    },
-    {
-      "epoch": 4.62,
-      "learning_rate": 2.982658959537572e-05,
-      "loss": 0.0882,
-      "step": 2062
-    },
-    {
-      "epoch": 4.63,
-      "learning_rate": 2.9653179190751446e-05,
-      "loss": 0.0676,
-      "step": 2063
-    },
-    {
-      "epoch": 4.63,
-      "learning_rate": 2.9479768786127166e-05,
-      "loss": 0.0508,
-      "step": 2064
-    },
-    {
-      "epoch": 4.63,
-      "learning_rate": 2.930635838150289e-05,
-      "loss": 0.0604,
-      "step": 2065
-    },
-    {
-      "epoch": 4.63,
-      "learning_rate": 2.9132947976878608e-05,
-      "loss": 0.0594,
-      "step": 2066
-    },
-    {
-      "epoch": 4.63,
-      "learning_rate": 2.895953757225433e-05,
-      "loss": 0.071,
-      "step": 2067
-    },
-    {
-      "epoch": 4.64,
-      "learning_rate": 2.8786127167630052e-05,
-      "loss": 0.053,
-      "step": 2068
-    },
-    {
-      "epoch": 4.64,
-      "learning_rate": 2.8612716763005776e-05,
-      "loss": 0.0668,
-      "step": 2069
-    },
-    {
-      "epoch": 4.64,
-      "learning_rate": 2.8439306358381497e-05,
-      "loss": 0.054,
-      "step": 2070
-    },
-    {
-      "epoch": 4.64,
-      "learning_rate": 2.826589595375722e-05,
-      "loss": 0.0587,
-      "step": 2071
-    },
-    {
-      "epoch": 4.65,
-      "learning_rate": 2.8092485549132945e-05,
-      "loss": 0.0616,
-      "step": 2072
-    },
-    {
-      "epoch": 4.65,
-      "learning_rate": 2.7919075144508666e-05,
-      "loss": 0.0662,
-      "step": 2073
-    },
-    {
-      "epoch": 4.65,
-      "learning_rate": 2.774566473988439e-05,
-      "loss": 0.0504,
-      "step": 2074
-    },
-    {
-      "epoch": 4.65,
-      "learning_rate": 2.757225433526011e-05,
-      "loss": 0.0569,
-      "step": 2075
-    },
-    {
-      "epoch": 4.65,
-      "learning_rate": 2.7398843930635835e-05,
-      "loss": 0.065,
-      "step": 2076
-    },
-    {
-      "epoch": 4.66,
-      "learning_rate": 2.722543352601156e-05,
-      "loss": 0.0687,
-      "step": 2077
-    },
-    {
-      "epoch": 4.66,
-      "learning_rate": 2.705202312138728e-05,
-      "loss": 0.049,
-      "step": 2078
-    },
-    {
-      "epoch": 4.66,
-      "learning_rate": 2.6878612716763003e-05,
-      "loss": 0.0833,
-      "step": 2079
-    },
-    {
-      "epoch": 4.66,
-      "learning_rate": 2.6705202312138724e-05,
-      "loss": 0.0515,
-      "step": 2080
-    },
-    {
-      "epoch": 4.67,
-      "learning_rate": 2.6531791907514448e-05,
-      "loss": 0.0803,
-      "step": 2081
-    },
-    {
-      "epoch": 4.67,
-      "learning_rate": 2.635838150289017e-05,
-      "loss": 0.0367,
-      "step": 2082
-    },
-    {
-      "epoch": 4.67,
-      "learning_rate": 2.6184971098265893e-05,
-      "loss": 0.0588,
-      "step": 2083
-    },
-    {
-      "epoch": 4.67,
-      "learning_rate": 2.6011560693641617e-05,
-      "loss": 0.085,
-      "step": 2084
-    },
-    {
-      "epoch": 4.67,
-      "learning_rate": 2.5838150289017338e-05,
-      "loss": 0.1184,
-      "step": 2085
-    },
-    {
-      "epoch": 4.68,
-      "learning_rate": 2.5664739884393062e-05,
-      "loss": 0.1096,
-      "step": 2086
-    },
-    {
-      "epoch": 4.68,
-      "learning_rate": 2.5491329479768782e-05,
-      "loss": 0.0876,
-      "step": 2087
-    },
-    {
-      "epoch": 4.68,
-      "learning_rate": 2.5317919075144507e-05,
-      "loss": 0.0747,
-      "step": 2088
-    },
-    {
-      "epoch": 4.68,
-      "learning_rate": 2.514450867052023e-05,
-      "loss": 0.0849,
-      "step": 2089
-    },
-    {
-      "epoch": 4.69,
-      "learning_rate": 2.497109826589595e-05,
-      "loss": 0.0875,
-      "step": 2090
-    },
-    {
-      "epoch": 4.69,
-      "learning_rate": 2.4797687861271675e-05,
-      "loss": 0.1031,
-      "step": 2091
-    },
-    {
-      "epoch": 4.69,
-      "learning_rate": 2.4624277456647396e-05,
-      "loss": 0.1113,
-      "step": 2092
-    },
-    {
-      "epoch": 4.69,
-      "learning_rate": 2.445086705202312e-05,
-      "loss": 0.0534,
-      "step": 2093
-    },
-    {
-      "epoch": 4.7,
-      "learning_rate": 2.427745664739884e-05,
-      "loss": 0.0803,
-      "step": 2094
-    },
-    {
-      "epoch": 4.7,
-      "learning_rate": 2.4104046242774565e-05,
-      "loss": 0.0649,
-      "step": 2095
-    },
-    {
-      "epoch": 4.7,
-      "learning_rate": 2.393063583815029e-05,
-      "loss": 0.0756,
-      "step": 2096
-    },
-    {
-      "epoch": 4.7,
-      "learning_rate": 2.375722543352601e-05,
-      "loss": 0.0671,
-      "step": 2097
-    },
-    {
-      "epoch": 4.7,
-      "learning_rate": 2.3583815028901734e-05,
-      "loss": 0.0649,
-      "step": 2098
-    },
-    {
-      "epoch": 4.71,
-      "learning_rate": 2.3410404624277454e-05,
-      "loss": 0.0595,
-      "step": 2099
-    },
-    {
-      "epoch": 4.71,
-      "learning_rate": 2.323699421965318e-05,
-      "loss": 0.0871,
-      "step": 2100
-    },
-    {
-      "epoch": 4.71,
-      "learning_rate": 2.30635838150289e-05,
-      "loss": 0.0608,
-      "step": 2101
-    },
-    {
-      "epoch": 4.71,
-      "learning_rate": 2.2890173410404623e-05,
-      "loss": 0.0828,
-      "step": 2102
-    },
-    {
-      "epoch": 4.72,
-      "learning_rate": 2.2716763005780347e-05,
-      "loss": 0.0578,
-      "step": 2103
-    },
-    {
-      "epoch": 4.72,
-      "learning_rate": 2.2543352601156068e-05,
-      "loss": 0.0769,
-      "step": 2104
-    },
-    {
-      "epoch": 4.72,
-      "learning_rate": 2.2369942196531792e-05,
-      "loss": 0.0526,
-      "step": 2105
-    },
-    {
-      "epoch": 4.72,
-      "learning_rate": 2.2196531791907513e-05,
-      "loss": 0.0487,
-      "step": 2106
-    },
-    {
-      "epoch": 4.72,
-      "learning_rate": 2.2023121387283237e-05,
-      "loss": 0.0588,
-      "step": 2107
-    },
-    {
-      "epoch": 4.73,
-      "learning_rate": 2.184971098265896e-05,
-      "loss": 0.0664,
-      "step": 2108
-    },
-    {
-      "epoch": 4.73,
-      "learning_rate": 2.167630057803468e-05,
-      "loss": 0.0583,
-      "step": 2109
-    },
-    {
-      "epoch": 4.73,
-      "learning_rate": 2.1502890173410405e-05,
-      "loss": 0.0539,
-      "step": 2110
-    },
-    {
-      "epoch": 4.73,
-      "learning_rate": 2.1329479768786126e-05,
-      "loss": 0.0558,
-      "step": 2111
-    },
-    {
-      "epoch": 4.74,
-      "learning_rate": 2.115606936416185e-05,
-      "loss": 0.0781,
-      "step": 2112
-    },
-    {
-      "epoch": 4.74,
-      "learning_rate": 2.098265895953757e-05,
-      "loss": 0.0674,
-      "step": 2113
-    },
-    {
-      "epoch": 4.74,
-      "learning_rate": 2.080924855491329e-05,
-      "loss": 0.0528,
-      "step": 2114
-    },
-    {
-      "epoch": 4.74,
-      "learning_rate": 2.0635838150289012e-05,
-      "loss": 0.0598,
-      "step": 2115
-    },
-    {
-      "epoch": 4.74,
-      "learning_rate": 2.0462427745664736e-05,
-      "loss": 0.0864,
-      "step": 2116
-    },
-    {
-      "epoch": 4.75,
-      "learning_rate": 2.028901734104046e-05,
-      "loss": 0.0483,
-      "step": 2117
-    },
-    {
-      "epoch": 4.75,
-      "learning_rate": 2.011560693641618e-05,
-      "loss": 0.0676,
-      "step": 2118
-    },
-    {
-      "epoch": 4.75,
-      "learning_rate": 1.9942196531791905e-05,
-      "loss": 0.0697,
-      "step": 2119
-    },
-    {
-      "epoch": 4.75,
-      "learning_rate": 1.9768786127167626e-05,
-      "loss": 0.0706,
-      "step": 2120
-    },
-    {
-      "epoch": 4.76,
-      "learning_rate": 1.959537572254335e-05,
-      "loss": 0.0506,
-      "step": 2121
-    },
-    {
-      "epoch": 4.76,
-      "learning_rate": 1.942196531791907e-05,
-      "loss": 0.0448,
-      "step": 2122
-    },
-    {
-      "epoch": 4.76,
-      "learning_rate": 1.9248554913294795e-05,
-      "loss": 0.0587,
-      "step": 2123
-    },
-    {
-      "epoch": 4.76,
-      "learning_rate": 1.907514450867052e-05,
-      "loss": 0.0716,
-      "step": 2124
-    },
-    {
-      "epoch": 4.76,
-      "learning_rate": 1.890173410404624e-05,
-      "loss": 0.0411,
-      "step": 2125
-    },
-    {
-      "epoch": 4.77,
-      "learning_rate": 1.8728323699421963e-05,
-      "loss": 0.0973,
-      "step": 2126
-    },
-    {
-      "epoch": 4.77,
-      "learning_rate": 1.8554913294797684e-05,
-      "loss": 0.066,
-      "step": 2127
-    },
-    {
-      "epoch": 4.77,
-      "learning_rate": 1.8381502890173408e-05,
-      "loss": 0.039,
-      "step": 2128
-    },
-    {
-      "epoch": 4.77,
-      "learning_rate": 1.8208092485549132e-05,
-      "loss": 0.0588,
-      "step": 2129
-    },
-    {
-      "epoch": 4.78,
-      "learning_rate": 1.8034682080924853e-05,
-      "loss": 0.0343,
-      "step": 2130
-    },
-    {
-      "epoch": 4.78,
-      "learning_rate": 1.7861271676300577e-05,
-      "loss": 0.0909,
-      "step": 2131
-    },
-    {
-      "epoch": 4.78,
-      "learning_rate": 1.7687861271676298e-05,
-      "loss": 0.0581,
-      "step": 2132
-    },
-    {
-      "epoch": 4.78,
-      "learning_rate": 1.7514450867052022e-05,
-      "loss": 0.0546,
-      "step": 2133
-    },
-    {
-      "epoch": 4.78,
-      "learning_rate": 1.7341040462427742e-05,
-      "loss": 0.057,
-      "step": 2134
-    },
-    {
-      "epoch": 4.79,
-      "learning_rate": 1.7167630057803466e-05,
-      "loss": 0.1371,
-      "step": 2135
-    },
-    {
-      "epoch": 4.79,
-      "learning_rate": 1.699421965317919e-05,
-      "loss": 0.1127,
-      "step": 2136
-    },
-    {
-      "epoch": 4.79,
-      "learning_rate": 1.682080924855491e-05,
-      "loss": 0.1117,
-      "step": 2137
-    },
-    {
-      "epoch": 4.79,
-      "learning_rate": 1.6647398843930635e-05,
-      "loss": 0.0863,
-      "step": 2138
-    },
-    {
-      "epoch": 4.8,
-      "learning_rate": 1.6473988439306356e-05,
-      "loss": 0.085,
-      "step": 2139
-    },
-    {
-      "epoch": 4.8,
-      "learning_rate": 1.630057803468208e-05,
-      "loss": 0.0834,
-      "step": 2140
-    },
-    {
-      "epoch": 4.8,
-      "learning_rate": 1.61271676300578e-05,
-      "loss": 0.0748,
-      "step": 2141
-    },
-    {
-      "epoch": 4.8,
-      "learning_rate": 1.5953757225433525e-05,
-      "loss": 0.0827,
-      "step": 2142
-    },
-    {
-      "epoch": 4.8,
-      "learning_rate": 1.578034682080925e-05,
-      "loss": 0.0786,
-      "step": 2143
-    },
-    {
-      "epoch": 4.81,
-      "learning_rate": 1.560693641618497e-05,
-      "loss": 0.0794,
-      "step": 2144
-    },
-    {
-      "epoch": 4.81,
-      "learning_rate": 1.5433526011560694e-05,
-      "loss": 0.0692,
-      "step": 2145
-    },
-    {
-      "epoch": 4.81,
-      "learning_rate": 1.5260115606936414e-05,
-      "loss": 0.091,
-      "step": 2146
-    },
-    {
-      "epoch": 4.81,
-      "learning_rate": 1.5086705202312138e-05,
-      "loss": 0.0572,
-      "step": 2147
-    },
-    {
-      "epoch": 4.82,
-      "learning_rate": 1.491329479768786e-05,
-      "loss": 0.0749,
-      "step": 2148
-    },
-    {
-      "epoch": 4.82,
-      "learning_rate": 1.4739884393063583e-05,
-      "loss": 0.0841,
-      "step": 2149
-    },
-    {
-      "epoch": 4.82,
-      "learning_rate": 1.4566473988439304e-05,
-      "loss": 0.07,
-      "step": 2150
-    },
-    {
-      "epoch": 4.82,
-      "learning_rate": 1.4393063583815026e-05,
-      "loss": 0.0935,
-      "step": 2151
-    },
-    {
-      "epoch": 4.83,
-      "learning_rate": 1.4219653179190749e-05,
-      "loss": 0.0645,
-      "step": 2152
-    },
-    {
-      "epoch": 4.83,
-      "learning_rate": 1.4046242774566473e-05,
-      "loss": 0.4197,
-      "step": 2153
-    },
-    {
-      "epoch": 4.83,
-      "learning_rate": 1.3872832369942195e-05,
-      "loss": 0.0902,
-      "step": 2154
-    },
-    {
-      "epoch": 4.83,
-      "learning_rate": 1.3699421965317917e-05,
-      "loss": 0.0462,
-      "step": 2155
-    },
-    {
-      "epoch": 4.83,
-      "learning_rate": 1.352601156069364e-05,
-      "loss": 0.0749,
-      "step": 2156
-    },
-    {
-      "epoch": 4.84,
-      "learning_rate": 1.3352601156069362e-05,
-      "loss": 0.0589,
-      "step": 2157
-    },
-    {
-      "epoch": 4.84,
-      "learning_rate": 1.3179190751445084e-05,
-      "loss": 0.0734,
-      "step": 2158
-    },
-    {
-      "epoch": 4.84,
-      "learning_rate": 1.3005780346820809e-05,
-      "loss": 0.0546,
-      "step": 2159
-    },
-    {
-      "epoch": 4.84,
-      "learning_rate": 1.2832369942196531e-05,
-      "loss": 0.0643,
-      "step": 2160
-    },
-    {
-      "epoch": 4.85,
-      "learning_rate": 1.2658959537572253e-05,
-      "loss": 0.0613,
-      "step": 2161
-    },
-    {
-      "epoch": 4.85,
-      "learning_rate": 1.2485549132947976e-05,
-      "loss": 0.0652,
-      "step": 2162
-    },
-    {
-      "epoch": 4.85,
-      "learning_rate": 1.2312138728323698e-05,
-      "loss": 0.0587,
-      "step": 2163
-    },
-    {
-      "epoch": 4.85,
-      "learning_rate": 1.213872832369942e-05,
-      "loss": 0.0617,
-      "step": 2164
-    },
-    {
-      "epoch": 4.85,
-      "learning_rate": 1.1965317919075144e-05,
-      "loss": 0.0527,
-      "step": 2165
-    },
-    {
-      "epoch": 4.86,
-      "learning_rate": 1.1791907514450867e-05,
-      "loss": 0.0402,
-      "step": 2166
-    },
-    {
-      "epoch": 4.86,
-      "learning_rate": 1.161849710982659e-05,
-      "loss": 0.0552,
-      "step": 2167
-    },
-    {
-      "epoch": 4.86,
-      "learning_rate": 1.1445086705202312e-05,
-      "loss": 0.0607,
-      "step": 2168
-    },
-    {
-      "epoch": 4.86,
-      "learning_rate": 1.1271676300578034e-05,
-      "loss": 0.056,
-      "step": 2169
-    },
-    {
-      "epoch": 4.87,
-      "learning_rate": 1.1098265895953756e-05,
-      "loss": 0.0579,
-      "step": 2170
-    },
-    {
-      "epoch": 4.87,
-      "learning_rate": 1.092485549132948e-05,
-      "loss": 0.0532,
-      "step": 2171
-    },
-    {
-      "epoch": 4.87,
-      "learning_rate": 1.0751445086705203e-05,
-      "loss": 0.0433,
-      "step": 2172
-    },
-    {
-      "epoch": 4.87,
-      "learning_rate": 1.0578034682080925e-05,
-      "loss": 0.0471,
-      "step": 2173
-    },
-    {
-      "epoch": 4.87,
-      "learning_rate": 1.0404624277456646e-05,
-      "loss": 0.043,
-      "step": 2174
-    },
-    {
-      "epoch": 4.88,
-      "learning_rate": 1.0231213872832368e-05,
-      "loss": 0.0672,
-      "step": 2175
-    },
-    {
-      "epoch": 4.88,
-      "learning_rate": 1.005780346820809e-05,
-      "loss": 0.0645,
-      "step": 2176
-    },
-    {
-      "epoch": 4.88,
-      "learning_rate": 9.884393063583813e-06,
-      "loss": 0.0419,
-      "step": 2177
-    },
-    {
-      "epoch": 4.88,
-      "learning_rate": 9.710982658959535e-06,
-      "loss": 0.0506,
-      "step": 2178
-    },
-    {
-      "epoch": 4.89,
-      "learning_rate": 9.53757225433526e-06,
-      "loss": 0.0312,
-      "step": 2179
-    },
-    {
-      "epoch": 4.89,
-      "learning_rate": 9.364161849710982e-06,
-      "loss": 0.0385,
-      "step": 2180
-    },
-    {
-      "epoch": 4.89,
-      "learning_rate": 9.190751445086704e-06,
-      "loss": 0.0981,
-      "step": 2181
-    },
-    {
-      "epoch": 4.89,
-      "learning_rate": 9.017341040462426e-06,
-      "loss": 0.0456,
-      "step": 2182
-    },
-    {
-      "epoch": 4.89,
-      "learning_rate": 8.843930635838149e-06,
-      "loss": 0.0773,
-      "step": 2183
-    },
-    {
-      "epoch": 4.9,
-      "learning_rate": 8.670520231213871e-06,
-      "loss": 0.0392,
-      "step": 2184
-    },
-    {
-      "epoch": 4.9,
-      "learning_rate": 8.497109826589595e-06,
-      "loss": 0.1151,
-      "step": 2185
-    },
-    {
-      "epoch": 4.9,
-      "learning_rate": 8.323699421965318e-06,
-      "loss": 0.0991,
-      "step": 2186
-    },
-    {
-      "epoch": 4.9,
-      "learning_rate": 8.15028901734104e-06,
-      "loss": 0.0904,
-      "step": 2187
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 7.976878612716762e-06,
-      "loss": 0.0539,
-      "step": 2188
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 7.803468208092485e-06,
-      "loss": 0.0813,
-      "step": 2189
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 7.630057803468207e-06,
-      "loss": 0.0639,
-      "step": 2190
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 7.45664739884393e-06,
-      "loss": 0.0839,
-      "step": 2191
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 7.283236994219652e-06,
-      "loss": 0.0821,
-      "step": 2192
-    },
-    {
-      "epoch": 4.92,
-      "learning_rate": 7.109826589595374e-06,
-      "loss": 0.085,
-      "step": 2193
-    },
-    {
-      "epoch": 4.92,
-      "learning_rate": 6.9364161849710975e-06,
-      "loss": 0.0905,
-      "step": 2194
-    },
-    {
-      "epoch": 4.92,
-      "learning_rate": 6.76300578034682e-06,
-      "loss": 0.0731,
-      "step": 2195
-    },
-    {
-      "epoch": 4.92,
-      "learning_rate": 6.589595375722542e-06,
-      "loss": 0.069,
-      "step": 2196
-    },
-    {
-      "epoch": 4.93,
-      "learning_rate": 6.4161849710982654e-06,
-      "loss": 0.0554,
-      "step": 2197
-    },
-    {
-      "epoch": 4.93,
-      "learning_rate": 6.242774566473988e-06,
-      "loss": 0.0552,
-      "step": 2198
-    },
-    {
-      "epoch": 4.93,
-      "learning_rate": 6.06936416184971e-06,
-      "loss": 0.0705,
-      "step": 2199
-    },
-    {
-      "epoch": 4.93,
-      "learning_rate": 5.895953757225433e-06,
-      "loss": 0.0627,
-      "step": 2200
-    },
-    {
-      "epoch": 4.93,
-      "learning_rate": 5.722543352601156e-06,
-      "loss": 0.0784,
-      "step": 2201
-    },
-    {
-      "epoch": 4.94,
-      "learning_rate": 5.549132947976878e-06,
-      "loss": 0.0665,
-      "step": 2202
-    },
-    {
-      "epoch": 4.94,
-      "learning_rate": 5.375722543352601e-06,
-      "loss": 0.0605,
-      "step": 2203
-    },
-    {
-      "epoch": 4.94,
-      "learning_rate": 5.202312138728323e-06,
-      "loss": 0.0544,
-      "step": 2204
-    },
     {
       "epoch": 4.94,
-      "learning_rate": 5.028901734104045e-06,
-      "loss": 0.0643,
-      "step": 2205
-    },
-    {
-      "epoch": 4.95,
-      "learning_rate": 4.855491329479768e-06,
-      "loss": 0.0615,
-      "step": 2206
-    },
-    {
-      "epoch": 4.95,
-      "learning_rate": 4.682080924855491e-06,
-      "loss": 0.0734,
-      "step": 2207
+      "learning_rate": 8.292682926829267e-06,
+      "loss": 0.1146,
+      "step": 1102
     },
     {
       "epoch": 4.95,
-      "learning_rate": 4.508670520231213e-06,
-      "loss": 0.0615,
-      "step": 2208
+      "learning_rate": 7.804878048780487e-06,
+      "loss": 0.1076,
+      "step": 1103
     },
     {
       "epoch": 4.95,
-      "learning_rate": 4.335260115606936e-06,
-      "loss": 0.053,
-      "step": 2209
-    },
-    {
-      "epoch": 4.96,
-      "learning_rate": 4.161849710982659e-06,
-      "loss": 0.0534,
-      "step": 2210
-    },
-    {
-      "epoch": 4.96,
-      "learning_rate": 3.988439306358381e-06,
-      "loss": 0.0597,
-      "step": 2211
+      "learning_rate": 7.3170731707317065e-06,
+      "loss": 0.0927,
+      "step": 1104
     },
     {
       "epoch": 4.96,
-      "learning_rate": 3.8150289017341036e-06,
-      "loss": 0.0839,
-      "step": 2212
+      "learning_rate": 6.829268292682927e-06,
+      "loss": 0.104,
+      "step": 1105
     },
     {
       "epoch": 4.96,
-      "learning_rate": 3.641618497109826e-06,
-      "loss": 0.0639,
-      "step": 2213
+      "learning_rate": 6.3414634146341454e-06,
+      "loss": 0.1055,
+      "step": 1106
     },
     {
       "epoch": 4.96,
-      "learning_rate": 3.4682080924855487e-06,
-      "loss": 0.0344,
-      "step": 2214
-    },
-    {
-      "epoch": 4.97,
-      "learning_rate": 3.294797687861271e-06,
-      "loss": 0.0682,
-      "step": 2215
-    },
-    {
-      "epoch": 4.97,
-      "learning_rate": 3.121387283236994e-06,
-      "loss": 0.0535,
-      "step": 2216
+      "learning_rate": 5.853658536585366e-06,
+      "loss": 0.0898,
+      "step": 1107
     },
     {
       "epoch": 4.97,
-      "learning_rate": 2.9479768786127167e-06,
-      "loss": 0.06,
-      "step": 2217
+      "learning_rate": 5.365853658536585e-06,
+      "loss": 0.1014,
+      "step": 1108
     },
     {
       "epoch": 4.97,
-      "learning_rate": 2.774566473988439e-06,
-      "loss": 0.0719,
-      "step": 2218
-    },
-    {
-      "epoch": 4.98,
-      "learning_rate": 2.6011560693641614e-06,
-      "loss": 0.0284,
-      "step": 2219
-    },
-    {
-      "epoch": 4.98,
-      "learning_rate": 2.427745664739884e-06,
-      "loss": 0.0614,
-      "step": 2220
-    },
-    {
-      "epoch": 4.98,
-      "learning_rate": 2.2543352601156066e-06,
-      "loss": 0.0555,
-      "step": 2221
+      "learning_rate": 4.878048780487805e-06,
+      "loss": 0.1008,
+      "step": 1109
     },
     {
       "epoch": 4.98,
-      "learning_rate": 2.0809248554913294e-06,
-      "loss": 0.0706,
-      "step": 2222
+      "learning_rate": 4.390243902439024e-06,
+      "loss": 0.0901,
+      "step": 1110
     },
     {
       "epoch": 4.98,
-      "learning_rate": 1.9075144508670518e-06,
-      "loss": 0.0718,
-      "step": 2223
-    },
-    {
-      "epoch": 4.99,
-      "learning_rate": 1.7341040462427744e-06,
-      "loss": 0.0511,
-      "step": 2224
-    },
-    {
-      "epoch": 4.99,
-      "learning_rate": 1.560693641618497e-06,
-      "loss": 0.0561,
-      "step": 2225
+      "learning_rate": 3.9024390243902435e-06,
+      "loss": 0.1198,
+      "step": 1111
     },
     {
       "epoch": 4.99,
-      "learning_rate": 1.3872832369942195e-06,
-      "loss": 0.0437,
-      "step": 2226
+      "learning_rate": 3.4146341463414634e-06,
+      "loss": 0.0944,
+      "step": 1112
     },
     {
       "epoch": 4.99,
-      "learning_rate": 1.213872832369942e-06,
-      "loss": 0.055,
-      "step": 2227
-    },
-    {
-      "epoch": 5.0,
-      "learning_rate": 1.0404624277456647e-06,
-      "loss": 0.0464,
-      "step": 2228
+      "learning_rate": 2.926829268292683e-06,
+      "loss": 0.0886,
+      "step": 1113
     },
     {
       "epoch": 5.0,
-      "learning_rate": 8.670520231213872e-07,
-      "loss": 0.048,
-      "step": 2229
+      "learning_rate": 2.4390243902439023e-06,
+      "loss": 0.0888,
+      "step": 1114
     },
     {
       "epoch": 5.0,
-      "learning_rate": 6.936416184971098e-07,
-      "loss": 0.0607,
-      "step": 2230
+      "learning_rate": 1.9512195121951218e-06,
+      "loss": 0.2069,
+      "step": 1115
     },
     {
       "epoch": 5.0,
-      "step": 2230,
+      "step": 1115,
       "total_flos": 0.0,
-      "train_loss": 1.6266609489166015,
-      "train_runtime": 26664.6603,
-      "train_samples_per_second": 5.351,
-      "train_steps_per_second": 0.084
+      "train_loss": 2.498668275362574,
+      "train_runtime": 25980.5507,
+      "train_samples_per_second": 5.492,
+      "train_steps_per_second": 0.043
     }
   ],
-  "max_steps": 2230,
+  "max_steps": 1115,
   "num_train_epochs": 5,
   "total_flos": 0.0,
   "trial_name": null,