diff --git "a/checkpoint-600/trainer_state.json" "b/checkpoint-600/trainer_state.json"
--- "a/checkpoint-600/trainer_state.json"
+++ "b/checkpoint-600/trainer_state.json"
@@ -1,7 +1,7 @@
 {
   "best_metric": null,
   "best_model_checkpoint": null,
-  "epoch": 12.140373063547266,
+  "epoch": 6.77846425419241,
   "eval_steps": 500,
   "global_step": 600,
   "is_hyper_param_search": false,
@@ -9,3611 +9,3611 @@
   "is_world_process_zero": true,
   "log_history": [
     {
-      "epoch": 0.02,
-      "learning_rate": 0.00019987244897959184,
-      "loss": 3.2215,
+      "epoch": 0.01,
+      "learning_rate": 0.00019985795454545454,
+      "loss": 3.3254,
       "step": 1
     },
     {
-      "epoch": 0.04,
-      "learning_rate": 0.00019974489795918367,
-      "loss": 2.8365,
+      "epoch": 0.02,
+      "learning_rate": 0.0001997159090909091,
+      "loss": 3.1222,
       "step": 2
     },
     {
-      "epoch": 0.06,
-      "learning_rate": 0.00019961734693877553,
-      "loss": 2.602,
+      "epoch": 0.03,
+      "learning_rate": 0.00019957386363636366,
+      "loss": 2.9506,
       "step": 3
     },
     {
-      "epoch": 0.08,
-      "learning_rate": 0.00019948979591836736,
-      "loss": 2.4196,
+      "epoch": 0.05,
+      "learning_rate": 0.0001994318181818182,
+      "loss": 2.8459,
       "step": 4
     },
     {
-      "epoch": 0.1,
-      "learning_rate": 0.0001993622448979592,
-      "loss": 2.2574,
+      "epoch": 0.06,
+      "learning_rate": 0.00019928977272727275,
+      "loss": 2.7277,
       "step": 5
     },
     {
-      "epoch": 0.12,
-      "learning_rate": 0.00019923469387755102,
-      "loss": 2.2239,
+      "epoch": 0.07,
+      "learning_rate": 0.00019914772727272728,
+      "loss": 2.6184,
       "step": 6
     },
     {
-      "epoch": 0.14,
-      "learning_rate": 0.00019910714285714288,
-      "loss": 2.1661,
+      "epoch": 0.08,
+      "learning_rate": 0.0001990056818181818,
+      "loss": 2.5151,
       "step": 7
     },
     {
-      "epoch": 0.16,
-      "learning_rate": 0.0001989795918367347,
-      "loss": 2.0987,
+      "epoch": 0.09,
+      "learning_rate": 0.00019886363636363637,
+      "loss": 2.4234,
       "step": 8
     },
     {
-      "epoch": 0.18,
-      "learning_rate": 0.00019885204081632654,
-      "loss": 2.015,
+      "epoch": 0.1,
+      "learning_rate": 0.00019872159090909093,
+      "loss": 2.3795,
       "step": 9
     },
     {
-      "epoch": 0.2,
-      "learning_rate": 0.00019872448979591837,
-      "loss": 1.9771,
+      "epoch": 0.11,
+      "learning_rate": 0.00019857954545454546,
+      "loss": 2.3629,
       "step": 10
     },
     {
-      "epoch": 0.22,
-      "learning_rate": 0.00019859693877551023,
-      "loss": 2.0271,
+      "epoch": 0.12,
+      "learning_rate": 0.00019843750000000002,
+      "loss": 2.3246,
       "step": 11
     },
     {
-      "epoch": 0.24,
-      "learning_rate": 0.00019846938775510203,
-      "loss": 1.9812,
+      "epoch": 0.14,
+      "learning_rate": 0.00019829545454545455,
+      "loss": 2.2274,
       "step": 12
     },
     {
-      "epoch": 0.26,
-      "learning_rate": 0.0001983418367346939,
-      "loss": 2.0834,
+      "epoch": 0.15,
+      "learning_rate": 0.00019815340909090908,
+      "loss": 2.2545,
       "step": 13
     },
     {
-      "epoch": 0.28,
-      "learning_rate": 0.00019821428571428572,
-      "loss": 1.9174,
+      "epoch": 0.16,
+      "learning_rate": 0.00019801136363636367,
+      "loss": 2.2814,
       "step": 14
     },
     {
-      "epoch": 0.3,
-      "learning_rate": 0.00019808673469387755,
-      "loss": 1.8409,
+      "epoch": 0.17,
+      "learning_rate": 0.0001978693181818182,
+      "loss": 2.2004,
       "step": 15
     },
     {
-      "epoch": 0.32,
-      "learning_rate": 0.00019795918367346938,
-      "loss": 1.929,
+      "epoch": 0.18,
+      "learning_rate": 0.00019772727272727273,
+      "loss": 2.1897,
       "step": 16
     },
     {
-      "epoch": 0.34,
-      "learning_rate": 0.00019783163265306124,
-      "loss": 2.0041,
+      "epoch": 0.19,
+      "learning_rate": 0.0001975852272727273,
+      "loss": 2.2214,
       "step": 17
     },
     {
-      "epoch": 0.36,
-      "learning_rate": 0.00019770408163265305,
-      "loss": 1.9385,
+      "epoch": 0.2,
+      "learning_rate": 0.00019744318181818182,
+      "loss": 2.2103,
       "step": 18
     },
     {
-      "epoch": 0.38,
-      "learning_rate": 0.0001975765306122449,
-      "loss": 1.9592,
+      "epoch": 0.21,
+      "learning_rate": 0.00019730113636363635,
+      "loss": 2.1747,
       "step": 19
     },
     {
-      "epoch": 0.4,
-      "learning_rate": 0.00019744897959183674,
-      "loss": 1.9701,
+      "epoch": 0.23,
+      "learning_rate": 0.00019715909090909094,
+      "loss": 2.2067,
       "step": 20
     },
     {
-      "epoch": 0.42,
-      "learning_rate": 0.0001973214285714286,
-      "loss": 1.9277,
+      "epoch": 0.24,
+      "learning_rate": 0.00019701704545454547,
+      "loss": 2.1944,
       "step": 21
     },
     {
-      "epoch": 0.45,
-      "learning_rate": 0.00019719387755102042,
-      "loss": 1.8394,
+      "epoch": 0.25,
+      "learning_rate": 0.000196875,
+      "loss": 2.2088,
       "step": 22
     },
     {
-      "epoch": 0.47,
-      "learning_rate": 0.00019706632653061226,
-      "loss": 1.8666,
+      "epoch": 0.26,
+      "learning_rate": 0.00019673295454545456,
+      "loss": 2.1786,
       "step": 23
     },
     {
-      "epoch": 0.49,
-      "learning_rate": 0.00019693877551020409,
-      "loss": 1.8997,
+      "epoch": 0.27,
+      "learning_rate": 0.0001965909090909091,
+      "loss": 2.1242,
       "step": 24
     },
     {
-      "epoch": 0.51,
-      "learning_rate": 0.00019681122448979592,
-      "loss": 1.9432,
+      "epoch": 0.28,
+      "learning_rate": 0.00019644886363636365,
+      "loss": 2.1233,
       "step": 25
     },
     {
-      "epoch": 0.53,
-      "learning_rate": 0.00019668367346938777,
-      "loss": 1.9137,
+      "epoch": 0.29,
+      "learning_rate": 0.0001963068181818182,
+      "loss": 2.1616,
       "step": 26
     },
     {
-      "epoch": 0.55,
-      "learning_rate": 0.0001965561224489796,
-      "loss": 1.905,
+      "epoch": 0.31,
+      "learning_rate": 0.00019616477272727274,
+      "loss": 2.1175,
       "step": 27
     },
     {
-      "epoch": 0.57,
-      "learning_rate": 0.00019642857142857144,
-      "loss": 1.8708,
+      "epoch": 0.32,
+      "learning_rate": 0.00019602272727272727,
+      "loss": 2.1242,
       "step": 28
     },
     {
-      "epoch": 0.59,
-      "learning_rate": 0.00019630102040816327,
-      "loss": 1.9097,
+      "epoch": 0.33,
+      "learning_rate": 0.00019588068181818183,
+      "loss": 2.186,
       "step": 29
     },
     {
-      "epoch": 0.61,
-      "learning_rate": 0.00019617346938775513,
-      "loss": 1.896,
+      "epoch": 0.34,
+      "learning_rate": 0.00019573863636363636,
+      "loss": 2.1319,
       "step": 30
     },
     {
-      "epoch": 0.63,
-      "learning_rate": 0.00019604591836734696,
-      "loss": 1.8834,
+      "epoch": 0.35,
+      "learning_rate": 0.00019559659090909092,
+      "loss": 2.1219,
       "step": 31
     },
     {
-      "epoch": 0.65,
-      "learning_rate": 0.0001959183673469388,
-      "loss": 1.8323,
+      "epoch": 0.36,
+      "learning_rate": 0.00019545454545454548,
+      "loss": 2.1094,
       "step": 32
     },
     {
-      "epoch": 0.67,
-      "learning_rate": 0.00019579081632653062,
-      "loss": 1.804,
+      "epoch": 0.37,
+      "learning_rate": 0.0001953125,
+      "loss": 2.1355,
       "step": 33
     },
     {
-      "epoch": 0.69,
-      "learning_rate": 0.00019566326530612248,
-      "loss": 1.8906,
+      "epoch": 0.38,
+      "learning_rate": 0.00019517045454545454,
+      "loss": 2.1231,
       "step": 34
     },
     {
-      "epoch": 0.71,
-      "learning_rate": 0.00019553571428571428,
-      "loss": 1.8693,
+      "epoch": 0.4,
+      "learning_rate": 0.0001950284090909091,
+      "loss": 2.1089,
       "step": 35
     },
     {
-      "epoch": 0.73,
-      "learning_rate": 0.00019540816326530614,
-      "loss": 1.9308,
+      "epoch": 0.41,
+      "learning_rate": 0.00019488636363636366,
+      "loss": 2.1329,
       "step": 36
     },
     {
-      "epoch": 0.75,
-      "learning_rate": 0.00019528061224489797,
-      "loss": 1.8082,
+      "epoch": 0.42,
+      "learning_rate": 0.0001947443181818182,
+      "loss": 2.1159,
       "step": 37
     },
     {
-      "epoch": 0.77,
-      "learning_rate": 0.0001951530612244898,
-      "loss": 1.848,
+      "epoch": 0.43,
+      "learning_rate": 0.00019460227272727275,
+      "loss": 2.1001,
       "step": 38
     },
     {
-      "epoch": 0.79,
-      "learning_rate": 0.00019502551020408163,
-      "loss": 1.8866,
+      "epoch": 0.44,
+      "learning_rate": 0.00019446022727272728,
+      "loss": 2.1084,
       "step": 39
     },
     {
-      "epoch": 0.81,
-      "learning_rate": 0.0001948979591836735,
-      "loss": 1.7844,
+      "epoch": 0.45,
+      "learning_rate": 0.0001943181818181818,
+      "loss": 2.1431,
       "step": 40
     },
     {
-      "epoch": 0.83,
-      "learning_rate": 0.0001947704081632653,
-      "loss": 1.8485,
+      "epoch": 0.46,
+      "learning_rate": 0.00019417613636363637,
+      "loss": 2.1111,
       "step": 41
     },
     {
-      "epoch": 0.85,
-      "learning_rate": 0.00019464285714285715,
-      "loss": 1.7917,
+      "epoch": 0.47,
+      "learning_rate": 0.00019403409090909093,
+      "loss": 2.1067,
       "step": 42
     },
     {
-      "epoch": 0.87,
-      "learning_rate": 0.00019451530612244898,
-      "loss": 1.7342,
+      "epoch": 0.49,
+      "learning_rate": 0.00019389204545454546,
+      "loss": 2.0974,
       "step": 43
     },
     {
-      "epoch": 0.89,
-      "learning_rate": 0.00019438775510204084,
-      "loss": 1.8479,
+      "epoch": 0.5,
+      "learning_rate": 0.00019375000000000002,
+      "loss": 2.1001,
       "step": 44
     },
     {
-      "epoch": 0.91,
-      "learning_rate": 0.00019426020408163267,
-      "loss": 1.8639,
+      "epoch": 0.51,
+      "learning_rate": 0.00019360795454545455,
+      "loss": 2.0721,
       "step": 45
     },
     {
-      "epoch": 0.93,
-      "learning_rate": 0.0001941326530612245,
-      "loss": 1.8166,
+      "epoch": 0.52,
+      "learning_rate": 0.00019346590909090908,
+      "loss": 2.0786,
       "step": 46
     },
     {
-      "epoch": 0.95,
-      "learning_rate": 0.00019400510204081633,
-      "loss": 1.7566,
+      "epoch": 0.53,
+      "learning_rate": 0.00019332386363636367,
+      "loss": 2.0882,
       "step": 47
     },
     {
-      "epoch": 0.97,
-      "learning_rate": 0.00019387755102040816,
-      "loss": 1.8071,
+      "epoch": 0.54,
+      "learning_rate": 0.0001931818181818182,
+      "loss": 2.083,
       "step": 48
     },
     {
-      "epoch": 0.99,
-      "learning_rate": 0.00019375000000000002,
-      "loss": 1.8612,
+      "epoch": 0.55,
+      "learning_rate": 0.00019303977272727273,
+      "loss": 2.1016,
       "step": 49
     },
     {
-      "epoch": 1.01,
-      "learning_rate": 0.00019362244897959185,
-      "loss": 1.7819,
+      "epoch": 0.56,
+      "learning_rate": 0.0001928977272727273,
+      "loss": 2.0844,
       "step": 50
     },
     {
-      "epoch": 1.03,
-      "learning_rate": 0.00019349489795918368,
-      "loss": 1.8647,
+      "epoch": 0.58,
+      "learning_rate": 0.00019275568181818182,
+      "loss": 2.0891,
       "step": 51
     },
     {
-      "epoch": 1.05,
-      "learning_rate": 0.0001933673469387755,
-      "loss": 1.8196,
+      "epoch": 0.59,
+      "learning_rate": 0.00019261363636363635,
+      "loss": 2.053,
       "step": 52
     },
     {
-      "epoch": 1.07,
-      "learning_rate": 0.00019323979591836737,
-      "loss": 1.8027,
+      "epoch": 0.6,
+      "learning_rate": 0.00019247159090909094,
+      "loss": 2.1013,
       "step": 53
     },
     {
-      "epoch": 1.09,
-      "learning_rate": 0.00019311224489795917,
-      "loss": 1.8927,
+      "epoch": 0.61,
+      "learning_rate": 0.00019232954545454547,
+      "loss": 2.127,
       "step": 54
     },
     {
-      "epoch": 1.11,
-      "learning_rate": 0.00019298469387755103,
-      "loss": 1.8481,
+      "epoch": 0.62,
+      "learning_rate": 0.0001921875,
+      "loss": 2.0909,
       "step": 55
     },
     {
-      "epoch": 1.13,
-      "learning_rate": 0.00019285714285714286,
-      "loss": 1.7781,
+      "epoch": 0.63,
+      "learning_rate": 0.00019204545454545456,
+      "loss": 2.1026,
       "step": 56
     },
     {
-      "epoch": 1.15,
-      "learning_rate": 0.00019272959183673472,
-      "loss": 1.8101,
+      "epoch": 0.64,
+      "learning_rate": 0.0001919034090909091,
+      "loss": 2.0689,
       "step": 57
     },
     {
-      "epoch": 1.17,
-      "learning_rate": 0.00019260204081632653,
-      "loss": 1.7257,
+      "epoch": 0.66,
+      "learning_rate": 0.00019176136363636365,
+      "loss": 2.0475,
       "step": 58
     },
     {
-      "epoch": 1.19,
-      "learning_rate": 0.00019247448979591838,
-      "loss": 1.8185,
+      "epoch": 0.67,
+      "learning_rate": 0.0001916193181818182,
+      "loss": 2.0645,
       "step": 59
     },
     {
-      "epoch": 1.21,
-      "learning_rate": 0.00019234693877551021,
-      "loss": 1.8557,
+      "epoch": 0.68,
+      "learning_rate": 0.00019147727272727274,
+      "loss": 2.0469,
       "step": 60
     },
     {
-      "epoch": 1.23,
-      "learning_rate": 0.00019221938775510204,
-      "loss": 1.7418,
+      "epoch": 0.69,
+      "learning_rate": 0.00019133522727272727,
+      "loss": 2.081,
       "step": 61
     },
     {
-      "epoch": 1.25,
-      "learning_rate": 0.00019209183673469388,
-      "loss": 1.6879,
+      "epoch": 0.7,
+      "learning_rate": 0.00019119318181818183,
+      "loss": 2.0682,
       "step": 62
     },
     {
-      "epoch": 1.27,
-      "learning_rate": 0.00019196428571428573,
-      "loss": 1.7651,
+      "epoch": 0.71,
+      "learning_rate": 0.00019105113636363636,
+      "loss": 2.0794,
       "step": 63
     },
     {
-      "epoch": 1.29,
-      "learning_rate": 0.00019183673469387756,
-      "loss": 1.7759,
+      "epoch": 0.72,
+      "learning_rate": 0.00019090909090909092,
+      "loss": 2.0218,
       "step": 64
     },
     {
-      "epoch": 1.32,
-      "learning_rate": 0.0001917091836734694,
-      "loss": 1.7691,
+      "epoch": 0.73,
+      "learning_rate": 0.00019076704545454548,
+      "loss": 2.0791,
       "step": 65
     },
     {
-      "epoch": 1.34,
-      "learning_rate": 0.00019158163265306123,
-      "loss": 1.7794,
+      "epoch": 0.75,
+      "learning_rate": 0.000190625,
+      "loss": 2.0506,
       "step": 66
     },
     {
-      "epoch": 1.36,
-      "learning_rate": 0.00019145408163265306,
-      "loss": 1.8152,
+      "epoch": 0.76,
+      "learning_rate": 0.00019048295454545454,
+      "loss": 2.0581,
       "step": 67
     },
     {
-      "epoch": 1.38,
-      "learning_rate": 0.00019132653061224492,
-      "loss": 1.8052,
+      "epoch": 0.77,
+      "learning_rate": 0.0001903409090909091,
+      "loss": 2.0614,
       "step": 68
     },
     {
-      "epoch": 1.4,
-      "learning_rate": 0.00019119897959183675,
-      "loss": 1.8054,
+      "epoch": 0.78,
+      "learning_rate": 0.00019019886363636366,
+      "loss": 2.0743,
       "step": 69
     },
     {
-      "epoch": 1.42,
-      "learning_rate": 0.00019107142857142858,
-      "loss": 1.8114,
+      "epoch": 0.79,
+      "learning_rate": 0.0001900568181818182,
+      "loss": 2.0934,
       "step": 70
     },
     {
-      "epoch": 1.44,
-      "learning_rate": 0.0001909438775510204,
-      "loss": 1.7749,
+      "epoch": 0.8,
+      "learning_rate": 0.00018991477272727275,
+      "loss": 2.0695,
       "step": 71
     },
     {
-      "epoch": 1.46,
-      "learning_rate": 0.00019081632653061227,
-      "loss": 1.777,
+      "epoch": 0.81,
+      "learning_rate": 0.00018977272727272728,
+      "loss": 2.0651,
       "step": 72
     },
     {
-      "epoch": 1.48,
-      "learning_rate": 0.0001906887755102041,
-      "loss": 1.7896,
+      "epoch": 0.82,
+      "learning_rate": 0.00018963068181818181,
+      "loss": 2.1002,
       "step": 73
     },
     {
-      "epoch": 1.5,
-      "learning_rate": 0.00019056122448979593,
-      "loss": 1.8335,
+      "epoch": 0.84,
+      "learning_rate": 0.00018948863636363637,
+      "loss": 2.0691,
       "step": 74
     },
     {
-      "epoch": 1.52,
-      "learning_rate": 0.00019043367346938776,
-      "loss": 1.8155,
+      "epoch": 0.85,
+      "learning_rate": 0.00018934659090909093,
+      "loss": 2.0596,
       "step": 75
     },
     {
-      "epoch": 1.54,
-      "learning_rate": 0.00019030612244897962,
-      "loss": 1.8224,
+      "epoch": 0.86,
+      "learning_rate": 0.00018920454545454546,
+      "loss": 2.0542,
       "step": 76
     },
     {
-      "epoch": 1.56,
-      "learning_rate": 0.00019017857142857142,
-      "loss": 1.7889,
+      "epoch": 0.87,
+      "learning_rate": 0.00018906250000000002,
+      "loss": 2.0543,
       "step": 77
     },
     {
-      "epoch": 1.58,
-      "learning_rate": 0.00019005102040816328,
-      "loss": 1.8866,
+      "epoch": 0.88,
+      "learning_rate": 0.00018892045454545455,
+      "loss": 2.0042,
       "step": 78
     },
     {
-      "epoch": 1.6,
-      "learning_rate": 0.0001899234693877551,
-      "loss": 1.8439,
+      "epoch": 0.89,
+      "learning_rate": 0.00018877840909090908,
+      "loss": 2.0072,
       "step": 79
     },
     {
-      "epoch": 1.62,
-      "learning_rate": 0.00018979591836734697,
-      "loss": 1.7906,
+      "epoch": 0.9,
+      "learning_rate": 0.00018863636363636364,
+      "loss": 2.0926,
       "step": 80
     },
     {
-      "epoch": 1.64,
-      "learning_rate": 0.00018966836734693877,
-      "loss": 1.8627,
+      "epoch": 0.92,
+      "learning_rate": 0.0001884943181818182,
+      "loss": 2.0015,
       "step": 81
     },
     {
-      "epoch": 1.66,
-      "learning_rate": 0.00018954081632653063,
-      "loss": 1.7497,
+      "epoch": 0.93,
+      "learning_rate": 0.00018835227272727273,
+      "loss": 2.0591,
       "step": 82
     },
     {
-      "epoch": 1.68,
-      "learning_rate": 0.00018941326530612246,
-      "loss": 1.7936,
+      "epoch": 0.94,
+      "learning_rate": 0.0001882102272727273,
+      "loss": 2.0522,
       "step": 83
     },
     {
-      "epoch": 1.7,
-      "learning_rate": 0.0001892857142857143,
-      "loss": 1.8341,
+      "epoch": 0.95,
+      "learning_rate": 0.00018806818181818182,
+      "loss": 2.0131,
       "step": 84
     },
     {
-      "epoch": 1.72,
-      "learning_rate": 0.00018915816326530612,
-      "loss": 1.7868,
+      "epoch": 0.96,
+      "learning_rate": 0.00018792613636363636,
+      "loss": 2.0572,
       "step": 85
     },
     {
-      "epoch": 1.74,
-      "learning_rate": 0.00018903061224489798,
-      "loss": 1.7493,
+      "epoch": 0.97,
+      "learning_rate": 0.00018778409090909091,
+      "loss": 2.0352,
       "step": 86
     },
     {
-      "epoch": 1.76,
-      "learning_rate": 0.0001889030612244898,
-      "loss": 1.7926,
+      "epoch": 0.98,
+      "learning_rate": 0.00018764204545454547,
+      "loss": 1.9937,
       "step": 87
     },
     {
-      "epoch": 1.78,
-      "learning_rate": 0.00018877551020408164,
-      "loss": 1.8278,
+      "epoch": 0.99,
+      "learning_rate": 0.0001875,
+      "loss": 2.0534,
       "step": 88
     },
     {
-      "epoch": 1.8,
-      "learning_rate": 0.00018864795918367347,
-      "loss": 1.7387,
+      "epoch": 1.01,
+      "learning_rate": 0.00018735795454545456,
+      "loss": 2.0151,
       "step": 89
     },
     {
-      "epoch": 1.82,
-      "learning_rate": 0.0001885204081632653,
-      "loss": 1.7669,
+      "epoch": 1.02,
+      "learning_rate": 0.0001872159090909091,
+      "loss": 2.0281,
       "step": 90
     },
     {
-      "epoch": 1.84,
-      "learning_rate": 0.00018839285714285716,
-      "loss": 1.7686,
+      "epoch": 1.03,
+      "learning_rate": 0.00018707386363636365,
+      "loss": 2.0582,
       "step": 91
     },
     {
-      "epoch": 1.86,
-      "learning_rate": 0.000188265306122449,
-      "loss": 1.7759,
+      "epoch": 1.04,
+      "learning_rate": 0.00018693181818181818,
+      "loss": 2.0173,
       "step": 92
     },
     {
-      "epoch": 1.88,
-      "learning_rate": 0.00018813775510204082,
-      "loss": 1.7016,
+      "epoch": 1.05,
+      "learning_rate": 0.00018678977272727274,
+      "loss": 2.0318,
       "step": 93
     },
     {
-      "epoch": 1.9,
-      "learning_rate": 0.00018801020408163265,
-      "loss": 1.8123,
+      "epoch": 1.06,
+      "learning_rate": 0.00018664772727272727,
+      "loss": 2.0747,
       "step": 94
     },
     {
-      "epoch": 1.92,
-      "learning_rate": 0.0001878826530612245,
-      "loss": 1.8315,
+      "epoch": 1.07,
+      "learning_rate": 0.00018650568181818183,
+      "loss": 2.0036,
       "step": 95
     },
     {
-      "epoch": 1.94,
-      "learning_rate": 0.00018775510204081634,
-      "loss": 1.7679,
+      "epoch": 1.08,
+      "learning_rate": 0.00018636363636363636,
+      "loss": 2.0215,
       "step": 96
     },
     {
-      "epoch": 1.96,
-      "learning_rate": 0.00018762755102040817,
-      "loss": 1.7874,
+      "epoch": 1.1,
+      "learning_rate": 0.00018622159090909092,
+      "loss": 2.0385,
       "step": 97
     },
     {
-      "epoch": 1.98,
-      "learning_rate": 0.0001875,
-      "loss": 1.8008,
+      "epoch": 1.11,
+      "learning_rate": 0.00018607954545454545,
+      "loss": 2.0247,
       "step": 98
     },
     {
-      "epoch": 2.0,
-      "learning_rate": 0.00018737244897959186,
-      "loss": 1.7177,
+      "epoch": 1.12,
+      "learning_rate": 0.0001859375,
+      "loss": 2.0075,
       "step": 99
     },
     {
-      "epoch": 2.02,
-      "learning_rate": 0.00018724489795918367,
-      "loss": 1.7272,
+      "epoch": 1.13,
+      "learning_rate": 0.00018579545454545454,
+      "loss": 2.0134,
       "step": 100
     },
     {
-      "epoch": 2.04,
-      "learning_rate": 0.00018711734693877552,
-      "loss": 1.7848,
+      "epoch": 1.14,
+      "learning_rate": 0.0001856534090909091,
+      "loss": 1.9908,
       "step": 101
     },
     {
-      "epoch": 2.06,
-      "learning_rate": 0.00018698979591836735,
-      "loss": 1.744,
+      "epoch": 1.15,
+      "learning_rate": 0.00018551136363636366,
+      "loss": 2.0048,
       "step": 102
     },
     {
-      "epoch": 2.08,
-      "learning_rate": 0.00018686224489795919,
-      "loss": 1.7005,
+      "epoch": 1.16,
+      "learning_rate": 0.0001853693181818182,
+      "loss": 1.9929,
       "step": 103
     },
     {
-      "epoch": 2.1,
-      "learning_rate": 0.00018673469387755102,
-      "loss": 1.8247,
+      "epoch": 1.17,
+      "learning_rate": 0.00018522727272727273,
+      "loss": 2.0545,
       "step": 104
     },
     {
-      "epoch": 2.12,
-      "learning_rate": 0.00018660714285714287,
-      "loss": 1.6855,
+      "epoch": 1.19,
+      "learning_rate": 0.00018508522727272728,
+      "loss": 2.0212,
       "step": 105
     },
     {
-      "epoch": 2.14,
-      "learning_rate": 0.0001864795918367347,
-      "loss": 1.7627,
+      "epoch": 1.2,
+      "learning_rate": 0.00018494318181818182,
+      "loss": 2.0154,
       "step": 106
     },
     {
-      "epoch": 2.17,
-      "learning_rate": 0.00018635204081632654,
-      "loss": 1.7564,
+      "epoch": 1.21,
+      "learning_rate": 0.00018480113636363637,
+      "loss": 1.988,
       "step": 107
     },
     {
-      "epoch": 2.19,
-      "learning_rate": 0.00018622448979591837,
-      "loss": 1.8237,
+      "epoch": 1.22,
+      "learning_rate": 0.00018465909090909093,
+      "loss": 2.004,
       "step": 108
     },
     {
-      "epoch": 2.21,
-      "learning_rate": 0.00018609693877551022,
-      "loss": 1.7421,
+      "epoch": 1.23,
+      "learning_rate": 0.00018451704545454546,
+      "loss": 1.9902,
       "step": 109
     },
     {
-      "epoch": 2.23,
-      "learning_rate": 0.00018596938775510206,
-      "loss": 1.7517,
+      "epoch": 1.24,
+      "learning_rate": 0.000184375,
+      "loss": 2.0044,
       "step": 110
     },
     {
-      "epoch": 2.25,
-      "learning_rate": 0.0001858418367346939,
-      "loss": 1.7515,
+      "epoch": 1.25,
+      "learning_rate": 0.00018423295454545455,
+      "loss": 2.028,
       "step": 111
     },
     {
-      "epoch": 2.27,
-      "learning_rate": 0.00018571428571428572,
-      "loss": 1.7842,
+      "epoch": 1.27,
+      "learning_rate": 0.00018409090909090909,
+      "loss": 1.975,
       "step": 112
     },
     {
-      "epoch": 2.29,
-      "learning_rate": 0.00018558673469387755,
-      "loss": 1.8001,
+      "epoch": 1.28,
+      "learning_rate": 0.00018394886363636364,
+      "loss": 1.9654,
       "step": 113
     },
     {
-      "epoch": 2.31,
-      "learning_rate": 0.0001854591836734694,
-      "loss": 1.7653,
+      "epoch": 1.29,
+      "learning_rate": 0.0001838068181818182,
+      "loss": 2.013,
       "step": 114
     },
     {
-      "epoch": 2.33,
-      "learning_rate": 0.00018533163265306124,
-      "loss": 1.694,
+      "epoch": 1.3,
+      "learning_rate": 0.00018366477272727273,
+      "loss": 1.9918,
       "step": 115
     },
     {
-      "epoch": 2.35,
-      "learning_rate": 0.00018520408163265307,
-      "loss": 1.7457,
+      "epoch": 1.31,
+      "learning_rate": 0.00018352272727272727,
+      "loss": 2.0028,
       "step": 116
     },
     {
-      "epoch": 2.37,
-      "learning_rate": 0.0001850765306122449,
-      "loss": 1.7899,
+      "epoch": 1.32,
+      "learning_rate": 0.00018338068181818182,
+      "loss": 1.9906,
       "step": 117
     },
     {
-      "epoch": 2.39,
-      "learning_rate": 0.00018494897959183676,
-      "loss": 1.7473,
+      "epoch": 1.33,
+      "learning_rate": 0.00018323863636363636,
+      "loss": 1.9781,
       "step": 118
     },
     {
-      "epoch": 2.41,
-      "learning_rate": 0.0001848214285714286,
-      "loss": 1.6639,
+      "epoch": 1.34,
+      "learning_rate": 0.00018309659090909091,
+      "loss": 1.994,
       "step": 119
     },
     {
-      "epoch": 2.43,
-      "learning_rate": 0.00018469387755102042,
-      "loss": 1.762,
+      "epoch": 1.36,
+      "learning_rate": 0.00018295454545454547,
+      "loss": 1.9732,
       "step": 120
     },
     {
-      "epoch": 2.45,
-      "learning_rate": 0.00018456632653061225,
-      "loss": 1.7378,
+      "epoch": 1.37,
+      "learning_rate": 0.0001828125,
+      "loss": 1.9985,
       "step": 121
     },
     {
-      "epoch": 2.47,
-      "learning_rate": 0.0001844387755102041,
-      "loss": 1.672,
+      "epoch": 1.38,
+      "learning_rate": 0.00018267045454545454,
+      "loss": 2.032,
       "step": 122
     },
     {
-      "epoch": 2.49,
-      "learning_rate": 0.0001843112244897959,
-      "loss": 1.7267,
+      "epoch": 1.39,
+      "learning_rate": 0.0001825284090909091,
+      "loss": 1.9743,
       "step": 123
     },
     {
-      "epoch": 2.51,
-      "learning_rate": 0.00018418367346938777,
-      "loss": 1.7825,
+      "epoch": 1.4,
+      "learning_rate": 0.00018238636363636365,
+      "loss": 1.9857,
       "step": 124
     },
     {
-      "epoch": 2.53,
-      "learning_rate": 0.0001840561224489796,
-      "loss": 1.7566,
+      "epoch": 1.41,
+      "learning_rate": 0.00018224431818181819,
+      "loss": 2.0118,
       "step": 125
     },
     {
-      "epoch": 2.55,
-      "learning_rate": 0.00018392857142857143,
-      "loss": 1.8169,
+      "epoch": 1.42,
+      "learning_rate": 0.00018210227272727274,
+      "loss": 2.0151,
       "step": 126
     },
     {
-      "epoch": 2.57,
-      "learning_rate": 0.00018380102040816326,
-      "loss": 1.6801,
+      "epoch": 1.43,
+      "learning_rate": 0.00018196022727272728,
+      "loss": 1.9863,
       "step": 127
     },
     {
-      "epoch": 2.59,
-      "learning_rate": 0.00018367346938775512,
-      "loss": 1.7292,
+      "epoch": 1.45,
+      "learning_rate": 0.00018181818181818183,
+      "loss": 1.9959,
       "step": 128
     },
     {
-      "epoch": 2.61,
-      "learning_rate": 0.00018354591836734695,
-      "loss": 1.737,
+      "epoch": 1.46,
+      "learning_rate": 0.00018167613636363637,
+      "loss": 1.9642,
       "step": 129
     },
     {
-      "epoch": 2.63,
-      "learning_rate": 0.00018341836734693878,
-      "loss": 1.7696,
+      "epoch": 1.47,
+      "learning_rate": 0.00018153409090909092,
+      "loss": 1.953,
       "step": 130
     },
     {
-      "epoch": 2.65,
-      "learning_rate": 0.0001832908163265306,
-      "loss": 1.7239,
+      "epoch": 1.48,
+      "learning_rate": 0.00018139204545454546,
+      "loss": 1.9994,
       "step": 131
     },
     {
-      "epoch": 2.67,
-      "learning_rate": 0.00018316326530612247,
-      "loss": 1.7441,
+      "epoch": 1.49,
+      "learning_rate": 0.00018125000000000001,
+      "loss": 1.9557,
       "step": 132
     },
     {
-      "epoch": 2.69,
-      "learning_rate": 0.0001830357142857143,
-      "loss": 1.7825,
+      "epoch": 1.5,
+      "learning_rate": 0.00018110795454545455,
+      "loss": 2.0051,
       "step": 133
     },
     {
-      "epoch": 2.71,
-      "learning_rate": 0.00018290816326530613,
-      "loss": 1.7411,
+      "epoch": 1.51,
+      "learning_rate": 0.0001809659090909091,
+      "loss": 1.9799,
       "step": 134
     },
     {
-      "epoch": 2.73,
-      "learning_rate": 0.00018278061224489796,
-      "loss": 1.7119,
+      "epoch": 1.53,
+      "learning_rate": 0.00018082386363636366,
+      "loss": 1.9696,
       "step": 135
     },
     {
-      "epoch": 2.75,
-      "learning_rate": 0.0001826530612244898,
-      "loss": 1.7443,
+      "epoch": 1.54,
+      "learning_rate": 0.0001806818181818182,
+      "loss": 1.9664,
       "step": 136
     },
     {
-      "epoch": 2.77,
-      "learning_rate": 0.00018252551020408165,
-      "loss": 1.7197,
+      "epoch": 1.55,
+      "learning_rate": 0.00018053977272727273,
+      "loss": 1.9619,
       "step": 137
     },
     {
-      "epoch": 2.79,
-      "learning_rate": 0.00018239795918367348,
-      "loss": 1.7273,
+      "epoch": 1.56,
+      "learning_rate": 0.00018039772727272729,
+      "loss": 1.9833,
       "step": 138
     },
     {
-      "epoch": 2.81,
-      "learning_rate": 0.0001822704081632653,
-      "loss": 1.7681,
+      "epoch": 1.57,
+      "learning_rate": 0.00018025568181818182,
+      "loss": 1.9791,
       "step": 139
     },
     {
-      "epoch": 2.83,
-      "learning_rate": 0.00018214285714285714,
-      "loss": 1.8088,
+      "epoch": 1.58,
+      "learning_rate": 0.00018011363636363638,
+      "loss": 1.9777,
       "step": 140
     },
     {
-      "epoch": 2.85,
-      "learning_rate": 0.000182015306122449,
-      "loss": 1.7301,
+      "epoch": 1.59,
+      "learning_rate": 0.00017997159090909093,
+      "loss": 1.9361,
       "step": 141
     },
     {
-      "epoch": 2.87,
-      "learning_rate": 0.00018188775510204083,
-      "loss": 1.6853,
+      "epoch": 1.6,
+      "learning_rate": 0.00017982954545454547,
+      "loss": 1.9449,
       "step": 142
     },
     {
-      "epoch": 2.89,
-      "learning_rate": 0.00018176020408163266,
-      "loss": 1.6966,
+      "epoch": 1.62,
+      "learning_rate": 0.0001796875,
+      "loss": 1.9541,
       "step": 143
     },
     {
-      "epoch": 2.91,
-      "learning_rate": 0.0001816326530612245,
-      "loss": 1.7938,
+      "epoch": 1.63,
+      "learning_rate": 0.00017954545454545456,
+      "loss": 1.9867,
       "step": 144
     },
     {
-      "epoch": 2.93,
-      "learning_rate": 0.00018150510204081635,
-      "loss": 1.7639,
+      "epoch": 1.64,
+      "learning_rate": 0.0001794034090909091,
+      "loss": 1.9433,
       "step": 145
     },
     {
-      "epoch": 2.95,
-      "learning_rate": 0.00018137755102040816,
-      "loss": 1.7527,
+      "epoch": 1.65,
+      "learning_rate": 0.00017926136363636365,
+      "loss": 1.9789,
       "step": 146
     },
     {
-      "epoch": 2.97,
-      "learning_rate": 0.00018125000000000001,
-      "loss": 1.7386,
+      "epoch": 1.66,
+      "learning_rate": 0.0001791193181818182,
+      "loss": 1.9942,
       "step": 147
     },
     {
-      "epoch": 2.99,
-      "learning_rate": 0.00018112244897959185,
-      "loss": 1.7223,
+      "epoch": 1.67,
+      "learning_rate": 0.00017897727272727274,
+      "loss": 1.9724,
       "step": 148
     },
     {
-      "epoch": 3.01,
-      "learning_rate": 0.00018099489795918368,
-      "loss": 1.7571,
+      "epoch": 1.68,
+      "learning_rate": 0.00017883522727272727,
+      "loss": 1.9938,
       "step": 149
     },
     {
-      "epoch": 3.04,
-      "learning_rate": 0.0001808673469387755,
-      "loss": 1.7054,
+      "epoch": 1.69,
+      "learning_rate": 0.00017869318181818183,
+      "loss": 1.9264,
       "step": 150
     },
     {
-      "epoch": 3.06,
-      "learning_rate": 0.00018073979591836737,
-      "loss": 1.6581,
+      "epoch": 1.71,
+      "learning_rate": 0.00017855113636363636,
+      "loss": 1.9372,
       "step": 151
     },
     {
-      "epoch": 3.08,
-      "learning_rate": 0.00018061224489795917,
-      "loss": 1.681,
+      "epoch": 1.72,
+      "learning_rate": 0.00017840909090909092,
+      "loss": 1.9463,
       "step": 152
     },
     {
-      "epoch": 3.1,
-      "learning_rate": 0.00018048469387755103,
-      "loss": 1.7425,
+      "epoch": 1.73,
+      "learning_rate": 0.00017826704545454547,
+      "loss": 1.9244,
       "step": 153
     },
     {
-      "epoch": 3.12,
-      "learning_rate": 0.00018035714285714286,
-      "loss": 1.7108,
+      "epoch": 1.74,
+      "learning_rate": 0.000178125,
+      "loss": 1.9139,
       "step": 154
     },
     {
-      "epoch": 3.14,
-      "learning_rate": 0.00018022959183673472,
-      "loss": 1.7194,
+      "epoch": 1.75,
+      "learning_rate": 0.00017798295454545454,
+      "loss": 1.9612,
       "step": 155
     },
     {
-      "epoch": 3.16,
-      "learning_rate": 0.00018010204081632655,
-      "loss": 1.6953,
+      "epoch": 1.76,
+      "learning_rate": 0.0001778409090909091,
+      "loss": 1.9399,
       "step": 156
     },
     {
-      "epoch": 3.18,
-      "learning_rate": 0.00017997448979591838,
-      "loss": 1.669,
+      "epoch": 1.77,
+      "learning_rate": 0.00017769886363636366,
+      "loss": 1.906,
       "step": 157
     },
     {
-      "epoch": 3.2,
-      "learning_rate": 0.0001798469387755102,
-      "loss": 1.744,
+      "epoch": 1.78,
+      "learning_rate": 0.0001775568181818182,
+      "loss": 1.9294,
       "step": 158
     },
     {
-      "epoch": 3.22,
-      "learning_rate": 0.00017971938775510204,
-      "loss": 1.6467,
+      "epoch": 1.8,
+      "learning_rate": 0.00017741477272727275,
+      "loss": 1.9663,
       "step": 159
     },
     {
-      "epoch": 3.24,
-      "learning_rate": 0.0001795918367346939,
-      "loss": 1.7103,
+      "epoch": 1.81,
+      "learning_rate": 0.00017727272727272728,
+      "loss": 1.9257,
       "step": 160
     },
     {
-      "epoch": 3.26,
-      "learning_rate": 0.00017946428571428573,
-      "loss": 1.6662,
+      "epoch": 1.82,
+      "learning_rate": 0.0001771306818181818,
+      "loss": 1.9416,
       "step": 161
     },
     {
-      "epoch": 3.28,
-      "learning_rate": 0.00017933673469387756,
-      "loss": 1.6657,
+      "epoch": 1.83,
+      "learning_rate": 0.00017698863636363637,
+      "loss": 1.94,
       "step": 162
     },
     {
-      "epoch": 3.3,
-      "learning_rate": 0.0001792091836734694,
-      "loss": 1.791,
+      "epoch": 1.84,
+      "learning_rate": 0.00017684659090909093,
+      "loss": 1.9064,
       "step": 163
     },
     {
-      "epoch": 3.32,
-      "learning_rate": 0.00017908163265306125,
-      "loss": 1.7704,
+      "epoch": 1.85,
+      "learning_rate": 0.00017670454545454546,
+      "loss": 1.9363,
       "step": 164
     },
     {
-      "epoch": 3.34,
-      "learning_rate": 0.00017895408163265305,
-      "loss": 1.7229,
+      "epoch": 1.86,
+      "learning_rate": 0.00017656250000000002,
+      "loss": 1.9414,
       "step": 165
     },
     {
-      "epoch": 3.36,
-      "learning_rate": 0.0001788265306122449,
-      "loss": 1.76,
+      "epoch": 1.88,
+      "learning_rate": 0.00017642045454545455,
+      "loss": 1.9526,
       "step": 166
     },
     {
-      "epoch": 3.38,
-      "learning_rate": 0.00017869897959183674,
-      "loss": 1.6482,
+      "epoch": 1.89,
+      "learning_rate": 0.00017627840909090908,
+      "loss": 1.9263,
       "step": 167
     },
     {
-      "epoch": 3.4,
-      "learning_rate": 0.0001785714285714286,
-      "loss": 1.8076,
+      "epoch": 1.9,
+      "learning_rate": 0.00017613636363636366,
+      "loss": 1.9251,
       "step": 168
     },
     {
-      "epoch": 3.42,
-      "learning_rate": 0.0001784438775510204,
-      "loss": 1.7368,
+      "epoch": 1.91,
+      "learning_rate": 0.0001759943181818182,
+      "loss": 1.9085,
       "step": 169
     },
     {
-      "epoch": 3.44,
-      "learning_rate": 0.00017831632653061226,
-      "loss": 1.6264,
+      "epoch": 1.92,
+      "learning_rate": 0.00017585227272727273,
+      "loss": 1.9287,
       "step": 170
     },
     {
-      "epoch": 3.46,
-      "learning_rate": 0.0001781887755102041,
-      "loss": 1.6289,
+      "epoch": 1.93,
+      "learning_rate": 0.00017571022727272729,
+      "loss": 1.9246,
       "step": 171
     },
     {
-      "epoch": 3.48,
-      "learning_rate": 0.00017806122448979592,
-      "loss": 1.7913,
+      "epoch": 1.94,
+      "learning_rate": 0.00017556818181818182,
+      "loss": 1.916,
       "step": 172
     },
     {
-      "epoch": 3.5,
-      "learning_rate": 0.00017793367346938775,
-      "loss": 1.6985,
+      "epoch": 1.95,
+      "learning_rate": 0.00017542613636363635,
+      "loss": 1.9297,
       "step": 173
     },
     {
-      "epoch": 3.52,
-      "learning_rate": 0.0001778061224489796,
-      "loss": 1.6936,
+      "epoch": 1.97,
+      "learning_rate": 0.00017528409090909094,
+      "loss": 1.8881,
       "step": 174
     },
     {
-      "epoch": 3.54,
-      "learning_rate": 0.00017767857142857141,
-      "loss": 1.8068,
+      "epoch": 1.98,
+      "learning_rate": 0.00017514204545454547,
+      "loss": 1.9208,
       "step": 175
     },
     {
-      "epoch": 3.56,
-      "learning_rate": 0.00017755102040816327,
-      "loss": 1.7243,
+      "epoch": 1.99,
+      "learning_rate": 0.000175,
+      "loss": 1.9233,
       "step": 176
     },
     {
-      "epoch": 3.58,
-      "learning_rate": 0.0001774234693877551,
-      "loss": 1.6893,
+      "epoch": 2.0,
+      "learning_rate": 0.00017485795454545456,
+      "loss": 1.9309,
       "step": 177
     },
     {
-      "epoch": 3.6,
-      "learning_rate": 0.00017729591836734696,
-      "loss": 1.8122,
+      "epoch": 2.01,
+      "learning_rate": 0.0001747159090909091,
+      "loss": 1.877,
       "step": 178
     },
     {
-      "epoch": 3.62,
-      "learning_rate": 0.0001771683673469388,
-      "loss": 1.6562,
+      "epoch": 2.02,
+      "learning_rate": 0.00017457386363636365,
+      "loss": 1.9083,
       "step": 179
     },
     {
-      "epoch": 3.64,
-      "learning_rate": 0.00017704081632653062,
-      "loss": 1.6999,
+      "epoch": 2.03,
+      "learning_rate": 0.0001744318181818182,
+      "loss": 1.8733,
       "step": 180
     },
     {
-      "epoch": 3.66,
-      "learning_rate": 0.00017691326530612245,
-      "loss": 1.7229,
+      "epoch": 2.04,
+      "learning_rate": 0.00017428977272727274,
+      "loss": 1.8905,
       "step": 181
     },
     {
-      "epoch": 3.68,
-      "learning_rate": 0.00017678571428571428,
-      "loss": 1.6764,
+      "epoch": 2.06,
+      "learning_rate": 0.00017414772727272727,
+      "loss": 1.9175,
       "step": 182
     },
     {
-      "epoch": 3.7,
-      "learning_rate": 0.00017665816326530614,
-      "loss": 1.6982,
+      "epoch": 2.07,
+      "learning_rate": 0.00017400568181818183,
+      "loss": 1.8846,
       "step": 183
     },
     {
-      "epoch": 3.72,
-      "learning_rate": 0.00017653061224489797,
-      "loss": 1.696,
+      "epoch": 2.08,
+      "learning_rate": 0.00017386363636363636,
+      "loss": 1.8847,
       "step": 184
     },
     {
-      "epoch": 3.74,
-      "learning_rate": 0.0001764030612244898,
-      "loss": 1.6797,
+      "epoch": 2.09,
+      "learning_rate": 0.00017372159090909092,
+      "loss": 1.8948,
       "step": 185
     },
     {
-      "epoch": 3.76,
-      "learning_rate": 0.00017627551020408164,
-      "loss": 1.637,
+      "epoch": 2.1,
+      "learning_rate": 0.00017357954545454548,
+      "loss": 1.8728,
       "step": 186
     },
     {
-      "epoch": 3.78,
-      "learning_rate": 0.0001761479591836735,
-      "loss": 1.7074,
+      "epoch": 2.11,
+      "learning_rate": 0.0001734375,
+      "loss": 1.8934,
       "step": 187
     },
     {
-      "epoch": 3.8,
-      "learning_rate": 0.0001760204081632653,
-      "loss": 1.705,
+      "epoch": 2.12,
+      "learning_rate": 0.00017329545454545454,
+      "loss": 1.8796,
       "step": 188
     },
     {
-      "epoch": 3.82,
-      "learning_rate": 0.00017589285714285716,
-      "loss": 1.6153,
+      "epoch": 2.14,
+      "learning_rate": 0.0001731534090909091,
+      "loss": 1.902,
       "step": 189
     },
     {
-      "epoch": 3.84,
-      "learning_rate": 0.00017576530612244899,
-      "loss": 1.7354,
+      "epoch": 2.15,
+      "learning_rate": 0.00017301136363636366,
+      "loss": 1.8864,
       "step": 190
     },
     {
-      "epoch": 3.86,
-      "learning_rate": 0.00017563775510204084,
-      "loss": 1.6941,
+      "epoch": 2.16,
+      "learning_rate": 0.0001728693181818182,
+      "loss": 1.8682,
       "step": 191
     },
     {
-      "epoch": 3.88,
-      "learning_rate": 0.00017551020408163265,
-      "loss": 1.7231,
+      "epoch": 2.17,
+      "learning_rate": 0.00017272727272727275,
+      "loss": 1.8662,
       "step": 192
     },
     {
-      "epoch": 3.91,
-      "learning_rate": 0.0001753826530612245,
-      "loss": 1.7663,
+      "epoch": 2.18,
+      "learning_rate": 0.00017258522727272728,
+      "loss": 1.8526,
       "step": 193
     },
     {
-      "epoch": 3.93,
-      "learning_rate": 0.00017525510204081634,
-      "loss": 1.6532,
+      "epoch": 2.19,
+      "learning_rate": 0.0001724431818181818,
+      "loss": 1.8682,
       "step": 194
     },
     {
-      "epoch": 3.95,
-      "learning_rate": 0.00017512755102040817,
-      "loss": 1.7115,
+      "epoch": 2.2,
+      "learning_rate": 0.00017230113636363637,
+      "loss": 1.8205,
       "step": 195
     },
     {
-      "epoch": 3.97,
-      "learning_rate": 0.000175,
-      "loss": 1.6955,
+      "epoch": 2.21,
+      "learning_rate": 0.00017215909090909093,
+      "loss": 1.8726,
       "step": 196
     },
     {
-      "epoch": 3.99,
-      "learning_rate": 0.00017487244897959186,
-      "loss": 1.6863,
+      "epoch": 2.23,
+      "learning_rate": 0.00017201704545454546,
+      "loss": 1.8241,
       "step": 197
     },
     {
-      "epoch": 4.01,
-      "learning_rate": 0.00017474489795918366,
-      "loss": 1.7012,
+      "epoch": 2.24,
+      "learning_rate": 0.00017187500000000002,
+      "loss": 1.9,
       "step": 198
     },
     {
-      "epoch": 4.03,
-      "learning_rate": 0.00017461734693877552,
-      "loss": 1.5927,
+      "epoch": 2.25,
+      "learning_rate": 0.00017173295454545455,
+      "loss": 1.8496,
       "step": 199
     },
     {
-      "epoch": 4.05,
-      "learning_rate": 0.00017448979591836735,
-      "loss": 1.6272,
+      "epoch": 2.26,
+      "learning_rate": 0.00017159090909090908,
+      "loss": 1.8562,
       "step": 200
     },
     {
-      "epoch": 4.07,
-      "learning_rate": 0.00017436224489795918,
-      "loss": 1.5994,
+      "epoch": 2.27,
+      "learning_rate": 0.00017144886363636367,
+      "loss": 1.8594,
       "step": 201
     },
     {
-      "epoch": 4.09,
-      "learning_rate": 0.00017423469387755104,
-      "loss": 1.7141,
+      "epoch": 2.28,
+      "learning_rate": 0.0001713068181818182,
+      "loss": 1.8606,
       "step": 202
     },
     {
-      "epoch": 4.11,
-      "learning_rate": 0.00017410714285714287,
-      "loss": 1.7547,
+      "epoch": 2.29,
+      "learning_rate": 0.00017116477272727273,
+      "loss": 1.8712,
       "step": 203
     },
     {
-      "epoch": 4.13,
-      "learning_rate": 0.0001739795918367347,
-      "loss": 1.6254,
+      "epoch": 2.3,
+      "learning_rate": 0.0001710227272727273,
+      "loss": 1.897,
       "step": 204
     },
     {
-      "epoch": 4.15,
-      "learning_rate": 0.00017385204081632653,
-      "loss": 1.6686,
+      "epoch": 2.32,
+      "learning_rate": 0.00017088068181818182,
+      "loss": 1.8287,
       "step": 205
     },
     {
-      "epoch": 4.17,
-      "learning_rate": 0.0001737244897959184,
-      "loss": 1.6684,
+      "epoch": 2.33,
+      "learning_rate": 0.00017073863636363635,
+      "loss": 1.8698,
       "step": 206
     },
     {
-      "epoch": 4.19,
-      "learning_rate": 0.00017359693877551022,
-      "loss": 1.6724,
+      "epoch": 2.34,
+      "learning_rate": 0.00017059659090909094,
+      "loss": 1.8611,
       "step": 207
     },
     {
-      "epoch": 4.21,
-      "learning_rate": 0.00017346938775510205,
-      "loss": 1.7361,
+      "epoch": 2.35,
+      "learning_rate": 0.00017045454545454547,
+      "loss": 1.8161,
       "step": 208
     },
     {
-      "epoch": 4.23,
-      "learning_rate": 0.00017334183673469388,
-      "loss": 1.7167,
+      "epoch": 2.36,
+      "learning_rate": 0.0001703125,
+      "loss": 1.8303,
       "step": 209
     },
     {
-      "epoch": 4.25,
-      "learning_rate": 0.00017321428571428574,
-      "loss": 1.7226,
+      "epoch": 2.37,
+      "learning_rate": 0.00017017045454545456,
+      "loss": 1.8423,
       "step": 210
     },
     {
-      "epoch": 4.27,
-      "learning_rate": 0.00017308673469387754,
-      "loss": 1.7133,
+      "epoch": 2.38,
+      "learning_rate": 0.0001700284090909091,
+      "loss": 1.861,
       "step": 211
     },
     {
-      "epoch": 4.29,
-      "learning_rate": 0.0001729591836734694,
-      "loss": 1.649,
+      "epoch": 2.4,
+      "learning_rate": 0.00016988636363636365,
+      "loss": 1.864,
       "step": 212
     },
     {
-      "epoch": 4.31,
-      "learning_rate": 0.00017283163265306123,
-      "loss": 1.7104,
+      "epoch": 2.41,
+      "learning_rate": 0.0001697443181818182,
+      "loss": 1.8448,
       "step": 213
     },
     {
-      "epoch": 4.33,
-      "learning_rate": 0.00017270408163265306,
-      "loss": 1.6861,
+      "epoch": 2.42,
+      "learning_rate": 0.00016960227272727274,
+      "loss": 1.8463,
       "step": 214
     },
     {
-      "epoch": 4.35,
-      "learning_rate": 0.0001725765306122449,
-      "loss": 1.648,
+      "epoch": 2.43,
+      "learning_rate": 0.00016946022727272727,
+      "loss": 1.8482,
       "step": 215
     },
     {
-      "epoch": 4.37,
-      "learning_rate": 0.00017244897959183675,
-      "loss": 1.6215,
+      "epoch": 2.44,
+      "learning_rate": 0.00016931818181818183,
+      "loss": 1.8289,
       "step": 216
     },
     {
-      "epoch": 4.39,
-      "learning_rate": 0.00017232142857142858,
-      "loss": 1.6334,
+      "epoch": 2.45,
+      "learning_rate": 0.00016917613636363636,
+      "loss": 1.8352,
       "step": 217
     },
     {
-      "epoch": 4.41,
-      "learning_rate": 0.0001721938775510204,
-      "loss": 1.6283,
+      "epoch": 2.46,
+      "learning_rate": 0.00016903409090909092,
+      "loss": 1.8161,
       "step": 218
     },
     {
-      "epoch": 4.43,
-      "learning_rate": 0.00017206632653061224,
-      "loss": 1.6462,
+      "epoch": 2.47,
+      "learning_rate": 0.00016889204545454548,
+      "loss": 1.8512,
       "step": 219
     },
     {
-      "epoch": 4.45,
-      "learning_rate": 0.0001719387755102041,
-      "loss": 1.7233,
+      "epoch": 2.49,
+      "learning_rate": 0.00016875,
+      "loss": 1.8211,
       "step": 220
     },
     {
-      "epoch": 4.47,
-      "learning_rate": 0.0001718112244897959,
-      "loss": 1.7839,
+      "epoch": 2.5,
+      "learning_rate": 0.00016860795454545454,
+      "loss": 1.7831,
       "step": 221
     },
     {
-      "epoch": 4.49,
-      "learning_rate": 0.00017168367346938776,
-      "loss": 1.7204,
+      "epoch": 2.51,
+      "learning_rate": 0.0001684659090909091,
+      "loss": 1.8232,
       "step": 222
     },
     {
-      "epoch": 4.51,
-      "learning_rate": 0.0001715561224489796,
-      "loss": 1.7671,
+      "epoch": 2.52,
+      "learning_rate": 0.00016832386363636366,
+      "loss": 1.8253,
       "step": 223
     },
     {
-      "epoch": 4.53,
-      "learning_rate": 0.00017142857142857143,
-      "loss": 1.6824,
+      "epoch": 2.53,
+      "learning_rate": 0.0001681818181818182,
+      "loss": 1.7994,
       "step": 224
     },
     {
-      "epoch": 4.55,
-      "learning_rate": 0.00017130102040816328,
-      "loss": 1.7068,
+      "epoch": 2.54,
+      "learning_rate": 0.00016803977272727275,
+      "loss": 1.8405,
       "step": 225
     },
     {
-      "epoch": 4.57,
-      "learning_rate": 0.00017117346938775511,
-      "loss": 1.6515,
+      "epoch": 2.55,
+      "learning_rate": 0.00016789772727272728,
+      "loss": 1.816,
       "step": 226
     },
     {
-      "epoch": 4.59,
-      "learning_rate": 0.00017104591836734694,
-      "loss": 1.6586,
+      "epoch": 2.56,
+      "learning_rate": 0.0001677556818181818,
+      "loss": 1.8343,
       "step": 227
     },
     {
-      "epoch": 4.61,
-      "learning_rate": 0.00017091836734693878,
-      "loss": 1.6355,
+      "epoch": 2.58,
+      "learning_rate": 0.00016761363636363637,
+      "loss": 1.8068,
       "step": 228
     },
     {
-      "epoch": 4.63,
-      "learning_rate": 0.00017079081632653063,
-      "loss": 1.7173,
+      "epoch": 2.59,
+      "learning_rate": 0.00016747159090909093,
+      "loss": 1.8337,
       "step": 229
     },
     {
-      "epoch": 4.65,
-      "learning_rate": 0.00017066326530612246,
-      "loss": 1.6585,
+      "epoch": 2.6,
+      "learning_rate": 0.00016732954545454546,
+      "loss": 1.8269,
       "step": 230
     },
     {
-      "epoch": 4.67,
-      "learning_rate": 0.0001705357142857143,
-      "loss": 1.5856,
+      "epoch": 2.61,
+      "learning_rate": 0.00016718750000000002,
+      "loss": 1.8243,
       "step": 231
     },
     {
-      "epoch": 4.69,
-      "learning_rate": 0.00017040816326530613,
-      "loss": 1.5923,
+      "epoch": 2.62,
+      "learning_rate": 0.00016704545454545455,
+      "loss": 1.7766,
       "step": 232
     },
     {
-      "epoch": 4.71,
-      "learning_rate": 0.00017028061224489798,
-      "loss": 1.7128,
+      "epoch": 2.63,
+      "learning_rate": 0.00016690340909090908,
+      "loss": 1.8144,
       "step": 233
     },
     {
-      "epoch": 4.73,
-      "learning_rate": 0.0001701530612244898,
-      "loss": 1.6971,
+      "epoch": 2.64,
+      "learning_rate": 0.00016676136363636367,
+      "loss": 1.8113,
       "step": 234
     },
     {
-      "epoch": 4.75,
-      "learning_rate": 0.00017002551020408165,
-      "loss": 1.6416,
+      "epoch": 2.65,
+      "learning_rate": 0.0001666193181818182,
+      "loss": 1.8086,
       "step": 235
     },
     {
-      "epoch": 4.78,
-      "learning_rate": 0.00016989795918367348,
-      "loss": 1.645,
+      "epoch": 2.67,
+      "learning_rate": 0.00016647727272727273,
+      "loss": 1.785,
       "step": 236
     },
     {
-      "epoch": 4.8,
-      "learning_rate": 0.0001697704081632653,
-      "loss": 1.6792,
+      "epoch": 2.68,
+      "learning_rate": 0.0001663352272727273,
+      "loss": 1.7884,
       "step": 237
     },
     {
-      "epoch": 4.82,
-      "learning_rate": 0.00016964285714285714,
-      "loss": 1.6522,
+      "epoch": 2.69,
+      "learning_rate": 0.00016619318181818182,
+      "loss": 1.7953,
       "step": 238
     },
     {
-      "epoch": 4.84,
-      "learning_rate": 0.000169515306122449,
-      "loss": 1.6315,
+      "epoch": 2.7,
+      "learning_rate": 0.00016605113636363635,
+      "loss": 1.8013,
       "step": 239
     },
     {
-      "epoch": 4.86,
-      "learning_rate": 0.00016938775510204083,
-      "loss": 1.6622,
+      "epoch": 2.71,
+      "learning_rate": 0.00016590909090909094,
+      "loss": 1.8074,
       "step": 240
     },
     {
-      "epoch": 4.88,
-      "learning_rate": 0.00016926020408163266,
-      "loss": 1.6566,
+      "epoch": 2.72,
+      "learning_rate": 0.00016576704545454547,
+      "loss": 1.82,
       "step": 241
     },
     {
-      "epoch": 4.9,
-      "learning_rate": 0.0001691326530612245,
-      "loss": 1.7141,
+      "epoch": 2.73,
+      "learning_rate": 0.000165625,
+      "loss": 1.7665,
       "step": 242
     },
     {
-      "epoch": 4.92,
-      "learning_rate": 0.00016900510204081635,
-      "loss": 1.5873,
+      "epoch": 2.75,
+      "learning_rate": 0.00016548295454545456,
+      "loss": 1.7638,
       "step": 243
     },
     {
-      "epoch": 4.94,
-      "learning_rate": 0.00016887755102040818,
-      "loss": 1.6571,
+      "epoch": 2.76,
+      "learning_rate": 0.0001653409090909091,
+      "loss": 1.7724,
       "step": 244
     },
     {
-      "epoch": 4.96,
-      "learning_rate": 0.00016875,
-      "loss": 1.6829,
+      "epoch": 2.77,
+      "learning_rate": 0.00016519886363636365,
+      "loss": 1.7917,
       "step": 245
     },
     {
-      "epoch": 4.98,
-      "learning_rate": 0.00016862244897959184,
-      "loss": 1.6935,
+      "epoch": 2.78,
+      "learning_rate": 0.0001650568181818182,
+      "loss": 1.8442,
       "step": 246
     },
     {
-      "epoch": 5.0,
-      "learning_rate": 0.00016849489795918367,
-      "loss": 1.6782,
+      "epoch": 2.79,
+      "learning_rate": 0.00016491477272727274,
+      "loss": 1.7887,
       "step": 247
     },
     {
-      "epoch": 5.02,
-      "learning_rate": 0.00016836734693877553,
-      "loss": 1.622,
+      "epoch": 2.8,
+      "learning_rate": 0.00016477272727272727,
+      "loss": 1.8055,
       "step": 248
     },
     {
-      "epoch": 5.04,
-      "learning_rate": 0.00016823979591836736,
-      "loss": 1.6596,
+      "epoch": 2.81,
+      "learning_rate": 0.00016463068181818183,
+      "loss": 1.7754,
       "step": 249
     },
     {
-      "epoch": 5.06,
-      "learning_rate": 0.0001681122448979592,
-      "loss": 1.5821,
+      "epoch": 2.82,
+      "learning_rate": 0.00016448863636363636,
+      "loss": 1.7948,
       "step": 250
     },
     {
-      "epoch": 5.08,
-      "learning_rate": 0.00016798469387755102,
-      "loss": 1.7292,
+      "epoch": 2.84,
+      "learning_rate": 0.00016434659090909092,
+      "loss": 1.8332,
       "step": 251
     },
     {
-      "epoch": 5.1,
-      "learning_rate": 0.00016785714285714288,
-      "loss": 1.646,
+      "epoch": 2.85,
+      "learning_rate": 0.00016420454545454548,
+      "loss": 1.772,
       "step": 252
     },
     {
-      "epoch": 5.12,
-      "learning_rate": 0.0001677295918367347,
-      "loss": 1.6969,
+      "epoch": 2.86,
+      "learning_rate": 0.0001640625,
+      "loss": 1.7781,
       "step": 253
     },
     {
-      "epoch": 5.14,
-      "learning_rate": 0.00016760204081632654,
-      "loss": 1.6082,
+      "epoch": 2.87,
+      "learning_rate": 0.00016392045454545454,
+      "loss": 1.7714,
       "step": 254
     },
     {
-      "epoch": 5.16,
-      "learning_rate": 0.00016747448979591837,
-      "loss": 1.5843,
+      "epoch": 2.88,
+      "learning_rate": 0.0001637784090909091,
+      "loss": 1.793,
       "step": 255
     },
     {
-      "epoch": 5.18,
-      "learning_rate": 0.00016734693877551023,
-      "loss": 1.6827,
+      "epoch": 2.89,
+      "learning_rate": 0.00016363636363636366,
+      "loss": 1.8038,
       "step": 256
     },
     {
-      "epoch": 5.2,
-      "learning_rate": 0.00016721938775510203,
-      "loss": 1.5824,
+      "epoch": 2.9,
+      "learning_rate": 0.0001634943181818182,
+      "loss": 1.8137,
       "step": 257
     },
     {
-      "epoch": 5.22,
-      "learning_rate": 0.0001670918367346939,
-      "loss": 1.6795,
+      "epoch": 2.91,
+      "learning_rate": 0.00016335227272727275,
+      "loss": 1.7726,
       "step": 258
     },
     {
-      "epoch": 5.24,
-      "learning_rate": 0.00016696428571428572,
-      "loss": 1.5639,
+      "epoch": 2.93,
+      "learning_rate": 0.00016321022727272728,
+      "loss": 1.7753,
       "step": 259
     },
     {
-      "epoch": 5.26,
-      "learning_rate": 0.00016683673469387755,
-      "loss": 1.592,
+      "epoch": 2.94,
+      "learning_rate": 0.0001630681818181818,
+      "loss": 1.7553,
       "step": 260
     },
     {
-      "epoch": 5.28,
-      "learning_rate": 0.00016670918367346938,
-      "loss": 1.65,
+      "epoch": 2.95,
+      "learning_rate": 0.00016292613636363637,
+      "loss": 1.7518,
       "step": 261
     },
     {
-      "epoch": 5.3,
-      "learning_rate": 0.00016658163265306124,
-      "loss": 1.5592,
+      "epoch": 2.96,
+      "learning_rate": 0.00016278409090909093,
+      "loss": 1.7724,
       "step": 262
     },
     {
-      "epoch": 5.32,
-      "learning_rate": 0.00016645408163265305,
-      "loss": 1.5091,
+      "epoch": 2.97,
+      "learning_rate": 0.00016264204545454546,
+      "loss": 1.7266,
       "step": 263
     },
     {
-      "epoch": 5.34,
-      "learning_rate": 0.0001663265306122449,
-      "loss": 1.6138,
+      "epoch": 2.98,
+      "learning_rate": 0.00016250000000000002,
+      "loss": 1.8032,
       "step": 264
     },
     {
-      "epoch": 5.36,
-      "learning_rate": 0.00016619897959183673,
-      "loss": 1.625,
+      "epoch": 2.99,
+      "learning_rate": 0.00016235795454545455,
+      "loss": 1.7345,
       "step": 265
     },
     {
-      "epoch": 5.38,
-      "learning_rate": 0.0001660714285714286,
-      "loss": 1.5757,
+      "epoch": 3.01,
+      "learning_rate": 0.00016221590909090908,
+      "loss": 1.7249,
       "step": 266
     },
     {
-      "epoch": 5.4,
-      "learning_rate": 0.00016594387755102042,
-      "loss": 1.6372,
+      "epoch": 3.02,
+      "learning_rate": 0.00016207386363636364,
+      "loss": 1.7218,
       "step": 267
     },
     {
-      "epoch": 5.42,
-      "learning_rate": 0.00016581632653061225,
-      "loss": 1.5891,
+      "epoch": 3.03,
+      "learning_rate": 0.0001619318181818182,
+      "loss": 1.7092,
       "step": 268
     },
     {
-      "epoch": 5.44,
-      "learning_rate": 0.00016568877551020409,
-      "loss": 1.6893,
+      "epoch": 3.04,
+      "learning_rate": 0.00016178977272727273,
+      "loss": 1.6807,
       "step": 269
     },
     {
-      "epoch": 5.46,
-      "learning_rate": 0.00016556122448979592,
-      "loss": 1.6662,
+      "epoch": 3.05,
+      "learning_rate": 0.0001616477272727273,
+      "loss": 1.7264,
       "step": 270
     },
     {
-      "epoch": 5.48,
-      "learning_rate": 0.00016543367346938777,
-      "loss": 1.7132,
+      "epoch": 3.06,
+      "learning_rate": 0.00016150568181818182,
+      "loss": 1.726,
       "step": 271
     },
     {
-      "epoch": 5.5,
-      "learning_rate": 0.0001653061224489796,
-      "loss": 1.5835,
+      "epoch": 3.07,
+      "learning_rate": 0.00016136363636363635,
+      "loss": 1.6986,
       "step": 272
     },
     {
-      "epoch": 5.52,
-      "learning_rate": 0.00016517857142857144,
-      "loss": 1.6342,
+      "epoch": 3.08,
+      "learning_rate": 0.0001612215909090909,
+      "loss": 1.68,
       "step": 273
     },
     {
-      "epoch": 5.54,
-      "learning_rate": 0.00016505102040816327,
-      "loss": 1.6717,
+      "epoch": 3.1,
+      "learning_rate": 0.00016107954545454547,
+      "loss": 1.6677,
       "step": 274
     },
     {
-      "epoch": 5.56,
-      "learning_rate": 0.00016492346938775512,
-      "loss": 1.6248,
+      "epoch": 3.11,
+      "learning_rate": 0.0001609375,
+      "loss": 1.7137,
       "step": 275
     },
     {
-      "epoch": 5.58,
-      "learning_rate": 0.00016479591836734696,
-      "loss": 1.6117,
+      "epoch": 3.12,
+      "learning_rate": 0.00016079545454545456,
+      "loss": 1.6671,
       "step": 276
     },
     {
-      "epoch": 5.6,
-      "learning_rate": 0.0001646683673469388,
-      "loss": 1.6798,
+      "epoch": 3.13,
+      "learning_rate": 0.0001606534090909091,
+      "loss": 1.6873,
       "step": 277
     },
     {
-      "epoch": 5.63,
-      "learning_rate": 0.00016454081632653062,
-      "loss": 1.6406,
+      "epoch": 3.14,
+      "learning_rate": 0.00016051136363636365,
+      "loss": 1.6694,
       "step": 278
     },
     {
-      "epoch": 5.65,
-      "learning_rate": 0.00016441326530612248,
-      "loss": 1.6512,
+      "epoch": 3.15,
+      "learning_rate": 0.00016036931818181818,
+      "loss": 1.7003,
       "step": 279
     },
     {
-      "epoch": 5.67,
-      "learning_rate": 0.00016428571428571428,
-      "loss": 1.6102,
+      "epoch": 3.16,
+      "learning_rate": 0.00016022727272727274,
+      "loss": 1.6861,
       "step": 280
     },
     {
-      "epoch": 5.69,
-      "learning_rate": 0.00016415816326530614,
-      "loss": 1.6113,
+      "epoch": 3.17,
+      "learning_rate": 0.00016008522727272727,
+      "loss": 1.6881,
       "step": 281
     },
     {
-      "epoch": 5.71,
-      "learning_rate": 0.00016403061224489797,
-      "loss": 1.7116,
+      "epoch": 3.19,
+      "learning_rate": 0.00015994318181818183,
+      "loss": 1.6848,
       "step": 282
     },
     {
-      "epoch": 5.73,
-      "learning_rate": 0.0001639030612244898,
-      "loss": 1.6846,
+      "epoch": 3.2,
+      "learning_rate": 0.00015980113636363636,
+      "loss": 1.6872,
       "step": 283
     },
     {
-      "epoch": 5.75,
-      "learning_rate": 0.00016377551020408163,
-      "loss": 1.6911,
+      "epoch": 3.21,
+      "learning_rate": 0.00015965909090909092,
+      "loss": 1.6975,
       "step": 284
     },
     {
-      "epoch": 5.77,
-      "learning_rate": 0.0001636479591836735,
-      "loss": 1.6202,
+      "epoch": 3.22,
+      "learning_rate": 0.00015951704545454545,
+      "loss": 1.6708,
       "step": 285
     },
     {
-      "epoch": 5.79,
-      "learning_rate": 0.0001635204081632653,
-      "loss": 1.5715,
+      "epoch": 3.23,
+      "learning_rate": 0.000159375,
+      "loss": 1.6985,
       "step": 286
     },
     {
-      "epoch": 5.81,
-      "learning_rate": 0.00016339285714285715,
-      "loss": 1.6461,
+      "epoch": 3.24,
+      "learning_rate": 0.00015923295454545454,
+      "loss": 1.6586,
       "step": 287
     },
     {
-      "epoch": 5.83,
-      "learning_rate": 0.00016326530612244898,
-      "loss": 1.6624,
+      "epoch": 3.25,
+      "learning_rate": 0.0001590909090909091,
+      "loss": 1.6707,
       "step": 288
     },
     {
-      "epoch": 5.85,
-      "learning_rate": 0.00016313775510204084,
-      "loss": 1.6535,
+      "epoch": 3.26,
+      "learning_rate": 0.00015894886363636366,
+      "loss": 1.6576,
       "step": 289
     },
     {
-      "epoch": 5.87,
-      "learning_rate": 0.00016301020408163267,
-      "loss": 1.6275,
+      "epoch": 3.28,
+      "learning_rate": 0.0001588068181818182,
+      "loss": 1.6625,
       "step": 290
     },
     {
-      "epoch": 5.89,
-      "learning_rate": 0.0001628826530612245,
-      "loss": 1.6636,
+      "epoch": 3.29,
+      "learning_rate": 0.00015866477272727275,
+      "loss": 1.677,
       "step": 291
     },
     {
-      "epoch": 5.91,
-      "learning_rate": 0.00016275510204081633,
-      "loss": 1.6546,
+      "epoch": 3.3,
+      "learning_rate": 0.00015852272727272728,
+      "loss": 1.6599,
       "step": 292
     },
     {
-      "epoch": 5.93,
-      "learning_rate": 0.00016262755102040816,
-      "loss": 1.7274,
+      "epoch": 3.31,
+      "learning_rate": 0.0001583806818181818,
+      "loss": 1.6674,
       "step": 293
     },
     {
-      "epoch": 5.95,
-      "learning_rate": 0.00016250000000000002,
-      "loss": 1.5901,
+      "epoch": 3.32,
+      "learning_rate": 0.00015823863636363637,
+      "loss": 1.6707,
       "step": 294
     },
     {
-      "epoch": 5.97,
-      "learning_rate": 0.00016237244897959185,
-      "loss": 1.6046,
+      "epoch": 3.33,
+      "learning_rate": 0.00015809659090909093,
+      "loss": 1.6788,
       "step": 295
     },
     {
-      "epoch": 5.99,
-      "learning_rate": 0.00016224489795918368,
-      "loss": 1.5828,
+      "epoch": 3.34,
+      "learning_rate": 0.00015795454545454546,
+      "loss": 1.6686,
       "step": 296
     },
     {
-      "epoch": 6.01,
-      "learning_rate": 0.0001621173469387755,
-      "loss": 1.6435,
+      "epoch": 3.36,
+      "learning_rate": 0.00015781250000000002,
+      "loss": 1.6488,
       "step": 297
     },
     {
-      "epoch": 6.03,
-      "learning_rate": 0.00016198979591836737,
-      "loss": 1.6263,
+      "epoch": 3.37,
+      "learning_rate": 0.00015767045454545455,
+      "loss": 1.6806,
       "step": 298
     },
     {
-      "epoch": 6.05,
-      "learning_rate": 0.00016186224489795917,
-      "loss": 1.4944,
+      "epoch": 3.38,
+      "learning_rate": 0.00015752840909090908,
+      "loss": 1.6862,
       "step": 299
     },
     {
-      "epoch": 6.07,
-      "learning_rate": 0.00016173469387755103,
-      "loss": 1.6286,
+      "epoch": 3.39,
+      "learning_rate": 0.00015738636363636364,
+      "loss": 1.6499,
       "step": 300
     },
     {
-      "epoch": 6.09,
-      "learning_rate": 0.00016160714285714286,
-      "loss": 1.694,
+      "epoch": 3.4,
+      "learning_rate": 0.0001572443181818182,
+      "loss": 1.6245,
       "step": 301
     },
     {
-      "epoch": 6.11,
-      "learning_rate": 0.00016147959183673472,
-      "loss": 1.6197,
+      "epoch": 3.41,
+      "learning_rate": 0.00015710227272727273,
+      "loss": 1.6268,
       "step": 302
     },
     {
-      "epoch": 6.13,
-      "learning_rate": 0.00016135204081632652,
-      "loss": 1.5597,
+      "epoch": 3.42,
+      "learning_rate": 0.0001569602272727273,
+      "loss": 1.6438,
       "step": 303
     },
     {
-      "epoch": 6.15,
-      "learning_rate": 0.00016122448979591838,
-      "loss": 1.5487,
+      "epoch": 3.43,
+      "learning_rate": 0.00015681818181818182,
+      "loss": 1.6681,
       "step": 304
     },
     {
-      "epoch": 6.17,
-      "learning_rate": 0.0001610969387755102,
-      "loss": 1.5769,
+      "epoch": 3.45,
+      "learning_rate": 0.00015667613636363635,
+      "loss": 1.6582,
       "step": 305
     },
     {
-      "epoch": 6.19,
-      "learning_rate": 0.00016096938775510204,
-      "loss": 1.6367,
+      "epoch": 3.46,
+      "learning_rate": 0.0001565340909090909,
+      "loss": 1.6432,
       "step": 306
     },
     {
-      "epoch": 6.21,
-      "learning_rate": 0.00016084183673469388,
-      "loss": 1.583,
+      "epoch": 3.47,
+      "learning_rate": 0.00015639204545454547,
+      "loss": 1.617,
       "step": 307
     },
     {
-      "epoch": 6.23,
-      "learning_rate": 0.00016071428571428573,
-      "loss": 1.6201,
+      "epoch": 3.48,
+      "learning_rate": 0.00015625,
+      "loss": 1.6569,
       "step": 308
     },
     {
-      "epoch": 6.25,
-      "learning_rate": 0.00016058673469387754,
-      "loss": 1.6586,
+      "epoch": 3.49,
+      "learning_rate": 0.00015610795454545456,
+      "loss": 1.6276,
       "step": 309
     },
     {
-      "epoch": 6.27,
-      "learning_rate": 0.0001604591836734694,
-      "loss": 1.6711,
+      "epoch": 3.5,
+      "learning_rate": 0.0001559659090909091,
+      "loss": 1.6432,
       "step": 310
     },
     {
-      "epoch": 6.29,
-      "learning_rate": 0.00016033163265306123,
-      "loss": 1.6402,
+      "epoch": 3.51,
+      "learning_rate": 0.00015582386363636365,
+      "loss": 1.6132,
       "step": 311
     },
     {
-      "epoch": 6.31,
-      "learning_rate": 0.00016020408163265306,
-      "loss": 1.5247,
+      "epoch": 3.52,
+      "learning_rate": 0.00015568181818181818,
+      "loss": 1.5997,
       "step": 312
     },
     {
-      "epoch": 6.33,
-      "learning_rate": 0.00016007653061224491,
-      "loss": 1.5356,
+      "epoch": 3.54,
+      "learning_rate": 0.00015553977272727274,
+      "loss": 1.6154,
       "step": 313
     },
     {
-      "epoch": 6.35,
-      "learning_rate": 0.00015994897959183675,
-      "loss": 1.564,
+      "epoch": 3.55,
+      "learning_rate": 0.00015539772727272727,
+      "loss": 1.5862,
       "step": 314
     },
     {
-      "epoch": 6.37,
-      "learning_rate": 0.00015982142857142858,
-      "loss": 1.563,
+      "epoch": 3.56,
+      "learning_rate": 0.00015525568181818183,
+      "loss": 1.6233,
       "step": 315
     },
     {
-      "epoch": 6.39,
-      "learning_rate": 0.0001596938775510204,
-      "loss": 1.5198,
+      "epoch": 3.57,
+      "learning_rate": 0.00015511363636363636,
+      "loss": 1.6265,
       "step": 316
     },
     {
-      "epoch": 6.41,
-      "learning_rate": 0.00015956632653061227,
-      "loss": 1.6558,
+      "epoch": 3.58,
+      "learning_rate": 0.00015497159090909092,
+      "loss": 1.6171,
       "step": 317
     },
     {
-      "epoch": 6.43,
-      "learning_rate": 0.0001594387755102041,
-      "loss": 1.5534,
+      "epoch": 3.59,
+      "learning_rate": 0.00015482954545454545,
+      "loss": 1.6303,
       "step": 318
     },
     {
-      "epoch": 6.45,
-      "learning_rate": 0.00015931122448979593,
-      "loss": 1.6239,
+      "epoch": 3.6,
+      "learning_rate": 0.0001546875,
+      "loss": 1.6272,
       "step": 319
     },
     {
-      "epoch": 6.47,
-      "learning_rate": 0.00015918367346938776,
-      "loss": 1.5645,
+      "epoch": 3.62,
+      "learning_rate": 0.00015454545454545454,
+      "loss": 1.6183,
       "step": 320
     },
     {
-      "epoch": 6.5,
-      "learning_rate": 0.00015905612244897962,
-      "loss": 1.5713,
+      "epoch": 3.63,
+      "learning_rate": 0.0001544034090909091,
+      "loss": 1.6205,
       "step": 321
     },
     {
-      "epoch": 6.52,
-      "learning_rate": 0.00015892857142857142,
-      "loss": 1.6176,
+      "epoch": 3.64,
+      "learning_rate": 0.00015426136363636366,
+      "loss": 1.6099,
       "step": 322
     },
     {
-      "epoch": 6.54,
-      "learning_rate": 0.00015880102040816328,
-      "loss": 1.502,
+      "epoch": 3.65,
+      "learning_rate": 0.0001541193181818182,
+      "loss": 1.5973,
       "step": 323
     },
     {
-      "epoch": 6.56,
-      "learning_rate": 0.0001586734693877551,
-      "loss": 1.645,
+      "epoch": 3.66,
+      "learning_rate": 0.00015397727272727272,
+      "loss": 1.6247,
       "step": 324
     },
     {
-      "epoch": 6.58,
-      "learning_rate": 0.00015854591836734697,
-      "loss": 1.5904,
+      "epoch": 3.67,
+      "learning_rate": 0.00015383522727272728,
+      "loss": 1.6041,
       "step": 325
     },
     {
-      "epoch": 6.6,
-      "learning_rate": 0.00015841836734693877,
-      "loss": 1.6149,
+      "epoch": 3.68,
+      "learning_rate": 0.00015369318181818181,
+      "loss": 1.5835,
       "step": 326
     },
     {
-      "epoch": 6.62,
-      "learning_rate": 0.00015829081632653063,
-      "loss": 1.6757,
+      "epoch": 3.69,
+      "learning_rate": 0.00015355113636363637,
+      "loss": 1.608,
       "step": 327
     },
     {
-      "epoch": 6.64,
-      "learning_rate": 0.00015816326530612246,
-      "loss": 1.541,
+      "epoch": 3.71,
+      "learning_rate": 0.00015340909090909093,
+      "loss": 1.6155,
       "step": 328
     },
     {
-      "epoch": 6.66,
-      "learning_rate": 0.0001580357142857143,
-      "loss": 1.5898,
+      "epoch": 3.72,
+      "learning_rate": 0.00015326704545454546,
+      "loss": 1.5777,
       "step": 329
     },
     {
-      "epoch": 6.68,
-      "learning_rate": 0.00015790816326530612,
-      "loss": 1.5441,
+      "epoch": 3.73,
+      "learning_rate": 0.000153125,
+      "loss": 1.5969,
       "step": 330
     },
     {
-      "epoch": 6.7,
-      "learning_rate": 0.00015778061224489798,
-      "loss": 1.61,
+      "epoch": 3.74,
+      "learning_rate": 0.00015298295454545455,
+      "loss": 1.5904,
       "step": 331
     },
     {
-      "epoch": 6.72,
-      "learning_rate": 0.00015765306122448978,
-      "loss": 1.615,
+      "epoch": 3.75,
+      "learning_rate": 0.00015284090909090909,
+      "loss": 1.586,
       "step": 332
     },
     {
-      "epoch": 6.74,
-      "learning_rate": 0.00015752551020408164,
-      "loss": 1.6575,
+      "epoch": 3.76,
+      "learning_rate": 0.00015269886363636364,
+      "loss": 1.582,
       "step": 333
     },
     {
-      "epoch": 6.76,
-      "learning_rate": 0.00015739795918367347,
-      "loss": 1.6702,
+      "epoch": 3.77,
+      "learning_rate": 0.0001525568181818182,
+      "loss": 1.548,
       "step": 334
     },
     {
-      "epoch": 6.78,
-      "learning_rate": 0.0001572704081632653,
-      "loss": 1.6009,
+      "epoch": 3.78,
+      "learning_rate": 0.00015241477272727273,
+      "loss": 1.5564,
       "step": 335
     },
     {
-      "epoch": 6.8,
-      "learning_rate": 0.00015714285714285716,
-      "loss": 1.5568,
+      "epoch": 3.8,
+      "learning_rate": 0.00015227272727272727,
+      "loss": 1.5506,
       "step": 336
     },
     {
-      "epoch": 6.82,
-      "learning_rate": 0.000157015306122449,
-      "loss": 1.619,
+      "epoch": 3.81,
+      "learning_rate": 0.00015213068181818182,
+      "loss": 1.5526,
       "step": 337
     },
     {
-      "epoch": 6.84,
-      "learning_rate": 0.00015688775510204082,
-      "loss": 1.5563,
+      "epoch": 3.82,
+      "learning_rate": 0.00015198863636363636,
+      "loss": 1.5564,
       "step": 338
     },
     {
-      "epoch": 6.86,
-      "learning_rate": 0.00015676020408163265,
-      "loss": 1.6328,
+      "epoch": 3.83,
+      "learning_rate": 0.00015184659090909091,
+      "loss": 1.5598,
       "step": 339
     },
     {
-      "epoch": 6.88,
-      "learning_rate": 0.0001566326530612245,
-      "loss": 1.5726,
+      "epoch": 3.84,
+      "learning_rate": 0.00015170454545454547,
+      "loss": 1.5679,
       "step": 340
     },
     {
-      "epoch": 6.9,
-      "learning_rate": 0.00015650510204081634,
-      "loss": 1.6199,
+      "epoch": 3.85,
+      "learning_rate": 0.0001515625,
+      "loss": 1.549,
       "step": 341
     },
     {
-      "epoch": 6.92,
-      "learning_rate": 0.00015637755102040817,
-      "loss": 1.5722,
+      "epoch": 3.86,
+      "learning_rate": 0.00015142045454545454,
+      "loss": 1.5672,
       "step": 342
     },
     {
-      "epoch": 6.94,
-      "learning_rate": 0.00015625,
-      "loss": 1.5685,
+      "epoch": 3.88,
+      "learning_rate": 0.0001512784090909091,
+      "loss": 1.5399,
       "step": 343
     },
     {
-      "epoch": 6.96,
-      "learning_rate": 0.00015612244897959186,
-      "loss": 1.5615,
+      "epoch": 3.89,
+      "learning_rate": 0.00015113636363636365,
+      "loss": 1.5576,
       "step": 344
     },
     {
-      "epoch": 6.98,
-      "learning_rate": 0.00015599489795918366,
-      "loss": 1.5994,
+      "epoch": 3.9,
+      "learning_rate": 0.00015099431818181818,
+      "loss": 1.549,
       "step": 345
     },
     {
-      "epoch": 7.0,
-      "learning_rate": 0.00015586734693877552,
-      "loss": 1.5579,
+      "epoch": 3.91,
+      "learning_rate": 0.00015085227272727274,
+      "loss": 1.5345,
       "step": 346
     },
     {
-      "epoch": 7.02,
-      "learning_rate": 0.00015573979591836735,
-      "loss": 1.547,
+      "epoch": 3.92,
+      "learning_rate": 0.00015071022727272728,
+      "loss": 1.5015,
       "step": 347
     },
     {
-      "epoch": 7.04,
-      "learning_rate": 0.00015561224489795918,
-      "loss": 1.5292,
+      "epoch": 3.93,
+      "learning_rate": 0.0001505681818181818,
+      "loss": 1.5221,
       "step": 348
     },
     {
-      "epoch": 7.06,
-      "learning_rate": 0.00015548469387755102,
-      "loss": 1.6032,
+      "epoch": 3.94,
+      "learning_rate": 0.00015042613636363637,
+      "loss": 1.556,
       "step": 349
     },
     {
-      "epoch": 7.08,
-      "learning_rate": 0.00015535714285714287,
-      "loss": 1.5149,
+      "epoch": 3.95,
+      "learning_rate": 0.00015028409090909092,
+      "loss": 1.5276,
       "step": 350
     },
     {
-      "epoch": 7.1,
-      "learning_rate": 0.0001552295918367347,
-      "loss": 1.6093,
+      "epoch": 3.97,
+      "learning_rate": 0.00015014204545454546,
+      "loss": 1.552,
       "step": 351
     },
     {
-      "epoch": 7.12,
-      "learning_rate": 0.00015510204081632654,
-      "loss": 1.5421,
+      "epoch": 3.98,
+      "learning_rate": 0.00015000000000000001,
+      "loss": 1.5377,
       "step": 352
     },
     {
-      "epoch": 7.14,
-      "learning_rate": 0.00015497448979591837,
-      "loss": 1.5733,
+      "epoch": 3.99,
+      "learning_rate": 0.00014985795454545455,
+      "loss": 1.5576,
       "step": 353
     },
     {
-      "epoch": 7.16,
-      "learning_rate": 0.00015484693877551022,
-      "loss": 1.5703,
+      "epoch": 4.0,
+      "learning_rate": 0.00014971590909090908,
+      "loss": 1.5295,
       "step": 354
     },
     {
-      "epoch": 7.18,
-      "learning_rate": 0.00015471938775510203,
-      "loss": 1.6141,
+      "epoch": 4.01,
+      "learning_rate": 0.00014957386363636366,
+      "loss": 1.4842,
       "step": 355
     },
     {
-      "epoch": 7.2,
-      "learning_rate": 0.00015459183673469389,
-      "loss": 1.5526,
+      "epoch": 4.02,
+      "learning_rate": 0.0001494318181818182,
+      "loss": 1.4803,
       "step": 356
     },
     {
-      "epoch": 7.22,
-      "learning_rate": 0.00015446428571428572,
-      "loss": 1.5347,
+      "epoch": 4.03,
+      "learning_rate": 0.00014928977272727273,
+      "loss": 1.4559,
       "step": 357
     },
     {
-      "epoch": 7.24,
-      "learning_rate": 0.00015433673469387755,
-      "loss": 1.5682,
+      "epoch": 4.04,
+      "learning_rate": 0.00014914772727272728,
+      "loss": 1.4777,
       "step": 358
     },
     {
-      "epoch": 7.26,
-      "learning_rate": 0.0001542091836734694,
-      "loss": 1.5292,
+      "epoch": 4.06,
+      "learning_rate": 0.00014900568181818182,
+      "loss": 1.4343,
       "step": 359
     },
     {
-      "epoch": 7.28,
-      "learning_rate": 0.00015408163265306124,
-      "loss": 1.499,
+      "epoch": 4.07,
+      "learning_rate": 0.00014886363636363635,
+      "loss": 1.4699,
       "step": 360
     },
     {
-      "epoch": 7.3,
-      "learning_rate": 0.00015395408163265307,
-      "loss": 1.5624,
+      "epoch": 4.08,
+      "learning_rate": 0.00014872159090909093,
+      "loss": 1.4452,
       "step": 361
     },
     {
-      "epoch": 7.32,
-      "learning_rate": 0.0001538265306122449,
-      "loss": 1.627,
+      "epoch": 4.09,
+      "learning_rate": 0.00014857954545454546,
+      "loss": 1.4461,
       "step": 362
     },
     {
-      "epoch": 7.34,
-      "learning_rate": 0.00015369897959183676,
-      "loss": 1.5327,
+      "epoch": 4.1,
+      "learning_rate": 0.0001484375,
+      "loss": 1.4523,
       "step": 363
     },
     {
-      "epoch": 7.37,
-      "learning_rate": 0.0001535714285714286,
-      "loss": 1.5622,
+      "epoch": 4.11,
+      "learning_rate": 0.00014829545454545455,
+      "loss": 1.4425,
       "step": 364
     },
     {
-      "epoch": 7.39,
-      "learning_rate": 0.00015344387755102042,
-      "loss": 1.5659,
+      "epoch": 4.12,
+      "learning_rate": 0.0001481534090909091,
+      "loss": 1.4559,
       "step": 365
     },
     {
-      "epoch": 7.41,
-      "learning_rate": 0.00015331632653061225,
-      "loss": 1.5019,
+      "epoch": 4.13,
+      "learning_rate": 0.00014801136363636365,
+      "loss": 1.4193,
       "step": 366
     },
     {
-      "epoch": 7.43,
-      "learning_rate": 0.0001531887755102041,
-      "loss": 1.5921,
+      "epoch": 4.15,
+      "learning_rate": 0.0001478693181818182,
+      "loss": 1.4136,
       "step": 367
     },
     {
-      "epoch": 7.45,
-      "learning_rate": 0.0001530612244897959,
-      "loss": 1.5914,
+      "epoch": 4.16,
+      "learning_rate": 0.00014772727272727274,
+      "loss": 1.445,
       "step": 368
     },
     {
-      "epoch": 7.47,
-      "learning_rate": 0.00015293367346938777,
-      "loss": 1.5045,
+      "epoch": 4.17,
+      "learning_rate": 0.00014758522727272727,
+      "loss": 1.4304,
       "step": 369
     },
     {
-      "epoch": 7.49,
-      "learning_rate": 0.0001528061224489796,
-      "loss": 1.6209,
+      "epoch": 4.18,
+      "learning_rate": 0.00014744318181818183,
+      "loss": 1.3996,
       "step": 370
     },
     {
-      "epoch": 7.51,
-      "learning_rate": 0.00015267857142857143,
-      "loss": 1.5198,
+      "epoch": 4.19,
+      "learning_rate": 0.00014730113636363636,
+      "loss": 1.4247,
       "step": 371
     },
     {
-      "epoch": 7.53,
-      "learning_rate": 0.00015255102040816326,
-      "loss": 1.5363,
+      "epoch": 4.2,
+      "learning_rate": 0.00014715909090909092,
+      "loss": 1.4303,
       "step": 372
     },
     {
-      "epoch": 7.55,
-      "learning_rate": 0.00015242346938775512,
-      "loss": 1.5391,
+      "epoch": 4.21,
+      "learning_rate": 0.00014701704545454547,
+      "loss": 1.4219,
       "step": 373
     },
     {
-      "epoch": 7.57,
-      "learning_rate": 0.00015229591836734695,
-      "loss": 1.4546,
+      "epoch": 4.23,
+      "learning_rate": 0.000146875,
+      "loss": 1.4538,
       "step": 374
     },
     {
-      "epoch": 7.59,
-      "learning_rate": 0.00015216836734693878,
-      "loss": 1.5546,
+      "epoch": 4.24,
+      "learning_rate": 0.00014673295454545454,
+      "loss": 1.4391,
       "step": 375
     },
     {
-      "epoch": 7.61,
-      "learning_rate": 0.0001520408163265306,
-      "loss": 1.5629,
+      "epoch": 4.25,
+      "learning_rate": 0.0001465909090909091,
+      "loss": 1.4482,
       "step": 376
     },
     {
-      "epoch": 7.63,
-      "learning_rate": 0.00015191326530612247,
-      "loss": 1.6002,
+      "epoch": 4.26,
+      "learning_rate": 0.00014644886363636365,
+      "loss": 1.4208,
       "step": 377
     },
     {
-      "epoch": 7.65,
-      "learning_rate": 0.00015178571428571427,
-      "loss": 1.5543,
+      "epoch": 4.27,
+      "learning_rate": 0.00014630681818181819,
+      "loss": 1.4111,
       "step": 378
     },
     {
-      "epoch": 7.67,
-      "learning_rate": 0.00015165816326530613,
-      "loss": 1.5925,
+      "epoch": 4.28,
+      "learning_rate": 0.00014616477272727274,
+      "loss": 1.4318,
       "step": 379
     },
     {
-      "epoch": 7.69,
-      "learning_rate": 0.00015153061224489796,
-      "loss": 1.5631,
+      "epoch": 4.29,
+      "learning_rate": 0.00014602272727272728,
+      "loss": 1.3913,
       "step": 380
     },
     {
-      "epoch": 7.71,
-      "learning_rate": 0.0001514030612244898,
-      "loss": 1.5677,
+      "epoch": 4.3,
+      "learning_rate": 0.0001458806818181818,
+      "loss": 1.3847,
       "step": 381
     },
     {
-      "epoch": 7.73,
-      "learning_rate": 0.00015127551020408165,
-      "loss": 1.5828,
+      "epoch": 4.32,
+      "learning_rate": 0.00014573863636363637,
+      "loss": 1.4254,
       "step": 382
     },
     {
-      "epoch": 7.75,
-      "learning_rate": 0.00015114795918367348,
-      "loss": 1.6494,
+      "epoch": 4.33,
+      "learning_rate": 0.00014559659090909093,
+      "loss": 1.4143,
       "step": 383
     },
     {
-      "epoch": 7.77,
-      "learning_rate": 0.0001510204081632653,
-      "loss": 1.553,
+      "epoch": 4.34,
+      "learning_rate": 0.00014545454545454546,
+      "loss": 1.4362,
       "step": 384
     },
     {
-      "epoch": 7.79,
-      "learning_rate": 0.00015089285714285714,
-      "loss": 1.6156,
+      "epoch": 4.35,
+      "learning_rate": 0.00014531250000000002,
+      "loss": 1.386,
       "step": 385
     },
     {
-      "epoch": 7.81,
-      "learning_rate": 0.000150765306122449,
-      "loss": 1.5001,
+      "epoch": 4.36,
+      "learning_rate": 0.00014517045454545455,
+      "loss": 1.4009,
       "step": 386
     },
     {
-      "epoch": 7.83,
-      "learning_rate": 0.00015063775510204083,
-      "loss": 1.5321,
+      "epoch": 4.37,
+      "learning_rate": 0.00014502840909090908,
+      "loss": 1.4089,
       "step": 387
     },
     {
-      "epoch": 7.85,
-      "learning_rate": 0.00015051020408163266,
-      "loss": 1.5307,
+      "epoch": 4.38,
+      "learning_rate": 0.00014488636363636366,
+      "loss": 1.4117,
       "step": 388
     },
     {
-      "epoch": 7.87,
-      "learning_rate": 0.0001503826530612245,
-      "loss": 1.5639,
+      "epoch": 4.39,
+      "learning_rate": 0.0001447443181818182,
+      "loss": 1.3788,
       "step": 389
     },
     {
-      "epoch": 7.89,
-      "learning_rate": 0.00015025510204081635,
-      "loss": 1.517,
+      "epoch": 4.41,
+      "learning_rate": 0.00014460227272727273,
+      "loss": 1.3573,
       "step": 390
     },
     {
-      "epoch": 7.91,
-      "learning_rate": 0.00015012755102040816,
-      "loss": 1.4776,
+      "epoch": 4.42,
+      "learning_rate": 0.00014446022727272729,
+      "loss": 1.4133,
       "step": 391
     },
     {
-      "epoch": 7.93,
-      "learning_rate": 0.00015000000000000001,
-      "loss": 1.5368,
+      "epoch": 4.43,
+      "learning_rate": 0.00014431818181818182,
+      "loss": 1.3866,
       "step": 392
     },
     {
-      "epoch": 7.95,
-      "learning_rate": 0.00014987244897959184,
-      "loss": 1.5636,
+      "epoch": 4.44,
+      "learning_rate": 0.00014417613636363635,
+      "loss": 1.3883,
       "step": 393
     },
     {
-      "epoch": 7.97,
-      "learning_rate": 0.00014974489795918368,
-      "loss": 1.6004,
+      "epoch": 4.45,
+      "learning_rate": 0.00014403409090909093,
+      "loss": 1.3741,
       "step": 394
     },
     {
-      "epoch": 7.99,
-      "learning_rate": 0.0001496173469387755,
-      "loss": 1.5524,
+      "epoch": 4.46,
+      "learning_rate": 0.00014389204545454547,
+      "loss": 1.358,
       "step": 395
     },
     {
-      "epoch": 8.01,
-      "learning_rate": 0.00014948979591836736,
-      "loss": 1.5307,
+      "epoch": 4.47,
+      "learning_rate": 0.00014375,
+      "loss": 1.3893,
       "step": 396
     },
     {
-      "epoch": 8.03,
-      "learning_rate": 0.00014936224489795917,
-      "loss": 1.5123,
+      "epoch": 4.49,
+      "learning_rate": 0.00014360795454545456,
+      "loss": 1.4062,
       "step": 397
     },
     {
-      "epoch": 8.05,
-      "learning_rate": 0.00014923469387755103,
-      "loss": 1.5132,
+      "epoch": 4.5,
+      "learning_rate": 0.0001434659090909091,
+      "loss": 1.3795,
       "step": 398
     },
     {
-      "epoch": 8.07,
-      "learning_rate": 0.00014910714285714286,
-      "loss": 1.5109,
+      "epoch": 4.51,
+      "learning_rate": 0.00014332386363636365,
+      "loss": 1.3472,
       "step": 399
     },
     {
-      "epoch": 8.09,
-      "learning_rate": 0.00014897959183673472,
-      "loss": 1.5302,
+      "epoch": 4.52,
+      "learning_rate": 0.0001431818181818182,
+      "loss": 1.3408,
       "step": 400
     },
     {
-      "epoch": 8.11,
-      "learning_rate": 0.00014885204081632652,
-      "loss": 1.5238,
+      "epoch": 4.53,
+      "learning_rate": 0.00014303977272727274,
+      "loss": 1.3801,
       "step": 401
     },
     {
-      "epoch": 8.13,
-      "learning_rate": 0.00014872448979591838,
-      "loss": 1.4781,
+      "epoch": 4.54,
+      "learning_rate": 0.00014289772727272727,
+      "loss": 1.3709,
       "step": 402
     },
     {
-      "epoch": 8.15,
-      "learning_rate": 0.0001485969387755102,
-      "loss": 1.5446,
+      "epoch": 4.55,
+      "learning_rate": 0.00014275568181818183,
+      "loss": 1.3653,
       "step": 403
     },
     {
-      "epoch": 8.17,
-      "learning_rate": 0.00014846938775510204,
-      "loss": 1.5,
+      "epoch": 4.56,
+      "learning_rate": 0.00014261363636363636,
+      "loss": 1.4089,
       "step": 404
     },
     {
-      "epoch": 8.19,
-      "learning_rate": 0.0001483418367346939,
-      "loss": 1.5458,
+      "epoch": 4.58,
+      "learning_rate": 0.00014247159090909092,
+      "loss": 1.3281,
       "step": 405
     },
     {
-      "epoch": 8.21,
-      "learning_rate": 0.00014821428571428573,
-      "loss": 1.5257,
+      "epoch": 4.59,
+      "learning_rate": 0.00014232954545454548,
+      "loss": 1.328,
       "step": 406
     },
     {
-      "epoch": 8.24,
-      "learning_rate": 0.00014808673469387756,
-      "loss": 1.4607,
+      "epoch": 4.6,
+      "learning_rate": 0.0001421875,
+      "loss": 1.3458,
       "step": 407
     },
     {
-      "epoch": 8.26,
-      "learning_rate": 0.0001479591836734694,
-      "loss": 1.4282,
+      "epoch": 4.61,
+      "learning_rate": 0.00014204545454545454,
+      "loss": 1.3425,
       "step": 408
     },
     {
-      "epoch": 8.28,
-      "learning_rate": 0.00014783163265306125,
-      "loss": 1.4519,
+      "epoch": 4.62,
+      "learning_rate": 0.0001419034090909091,
+      "loss": 1.3236,
       "step": 409
     },
     {
-      "epoch": 8.3,
-      "learning_rate": 0.00014770408163265305,
-      "loss": 1.475,
+      "epoch": 4.63,
+      "learning_rate": 0.00014176136363636366,
+      "loss": 1.3439,
       "step": 410
     },
     {
-      "epoch": 8.32,
-      "learning_rate": 0.0001475765306122449,
-      "loss": 1.5425,
+      "epoch": 4.64,
+      "learning_rate": 0.0001416193181818182,
+      "loss": 1.3397,
       "step": 411
     },
     {
-      "epoch": 8.34,
-      "learning_rate": 0.00014744897959183674,
-      "loss": 1.5407,
+      "epoch": 4.65,
+      "learning_rate": 0.00014147727272727275,
+      "loss": 1.329,
       "step": 412
     },
     {
-      "epoch": 8.36,
-      "learning_rate": 0.0001473214285714286,
-      "loss": 1.5698,
+      "epoch": 4.67,
+      "learning_rate": 0.00014133522727272728,
+      "loss": 1.3377,
       "step": 413
     },
     {
-      "epoch": 8.38,
-      "learning_rate": 0.0001471938775510204,
-      "loss": 1.4282,
+      "epoch": 4.68,
+      "learning_rate": 0.0001411931818181818,
+      "loss": 1.343,
       "step": 414
     },
     {
-      "epoch": 8.4,
-      "learning_rate": 0.00014706632653061226,
-      "loss": 1.5301,
+      "epoch": 4.69,
+      "learning_rate": 0.00014105113636363637,
+      "loss": 1.3185,
       "step": 415
     },
     {
-      "epoch": 8.42,
-      "learning_rate": 0.0001469387755102041,
-      "loss": 1.5083,
+      "epoch": 4.7,
+      "learning_rate": 0.00014090909090909093,
+      "loss": 1.3174,
       "step": 416
     },
     {
-      "epoch": 8.44,
-      "learning_rate": 0.00014681122448979592,
-      "loss": 1.5712,
+      "epoch": 4.71,
+      "learning_rate": 0.00014076704545454546,
+      "loss": 1.3231,
       "step": 417
     },
     {
-      "epoch": 8.46,
-      "learning_rate": 0.00014668367346938775,
-      "loss": 1.4363,
+      "epoch": 4.72,
+      "learning_rate": 0.00014062500000000002,
+      "loss": 1.3407,
       "step": 418
     },
     {
-      "epoch": 8.48,
-      "learning_rate": 0.0001465561224489796,
-      "loss": 1.4463,
+      "epoch": 4.73,
+      "learning_rate": 0.00014048295454545455,
+      "loss": 1.3138,
       "step": 419
     },
     {
-      "epoch": 8.5,
-      "learning_rate": 0.00014642857142857141,
-      "loss": 1.4738,
+      "epoch": 4.74,
+      "learning_rate": 0.00014034090909090908,
+      "loss": 1.3134,
       "step": 420
     },
     {
-      "epoch": 8.52,
-      "learning_rate": 0.00014630102040816327,
-      "loss": 1.5396,
+      "epoch": 4.76,
+      "learning_rate": 0.00014019886363636367,
+      "loss": 1.3187,
       "step": 421
     },
     {
-      "epoch": 8.54,
-      "learning_rate": 0.0001461734693877551,
-      "loss": 1.4384,
+      "epoch": 4.77,
+      "learning_rate": 0.0001400568181818182,
+      "loss": 1.2781,
       "step": 422
     },
     {
-      "epoch": 8.56,
-      "learning_rate": 0.00014604591836734696,
-      "loss": 1.5345,
+      "epoch": 4.78,
+      "learning_rate": 0.00013991477272727273,
+      "loss": 1.3254,
       "step": 423
     },
     {
-      "epoch": 8.58,
-      "learning_rate": 0.0001459183673469388,
-      "loss": 1.5355,
+      "epoch": 4.79,
+      "learning_rate": 0.0001397727272727273,
+      "loss": 1.2929,
       "step": 424
     },
     {
-      "epoch": 8.6,
-      "learning_rate": 0.00014579081632653062,
-      "loss": 1.5188,
+      "epoch": 4.8,
+      "learning_rate": 0.00013963068181818182,
+      "loss": 1.2953,
       "step": 425
     },
     {
-      "epoch": 8.62,
-      "learning_rate": 0.00014566326530612245,
-      "loss": 1.5575,
+      "epoch": 4.81,
+      "learning_rate": 0.00013948863636363635,
+      "loss": 1.3202,
       "step": 426
     },
     {
-      "epoch": 8.64,
-      "learning_rate": 0.00014553571428571428,
-      "loss": 1.5279,
+      "epoch": 4.82,
+      "learning_rate": 0.00013934659090909094,
+      "loss": 1.3118,
       "step": 427
     },
     {
-      "epoch": 8.66,
-      "learning_rate": 0.00014540816326530614,
-      "loss": 1.5484,
+      "epoch": 4.84,
+      "learning_rate": 0.00013920454545454547,
+      "loss": 1.3046,
       "step": 428
     },
     {
-      "epoch": 8.68,
-      "learning_rate": 0.00014528061224489797,
-      "loss": 1.4878,
+      "epoch": 4.85,
+      "learning_rate": 0.0001390625,
+      "loss": 1.2708,
       "step": 429
     },
     {
-      "epoch": 8.7,
-      "learning_rate": 0.0001451530612244898,
-      "loss": 1.503,
+      "epoch": 4.86,
+      "learning_rate": 0.00013892045454545456,
+      "loss": 1.2835,
       "step": 430
     },
     {
-      "epoch": 8.72,
-      "learning_rate": 0.00014502551020408163,
-      "loss": 1.4723,
+      "epoch": 4.87,
+      "learning_rate": 0.0001387784090909091,
+      "loss": 1.2728,
       "step": 431
     },
     {
-      "epoch": 8.74,
-      "learning_rate": 0.0001448979591836735,
-      "loss": 1.5579,
+      "epoch": 4.88,
+      "learning_rate": 0.00013863636363636365,
+      "loss": 1.3107,
       "step": 432
     },
     {
-      "epoch": 8.76,
-      "learning_rate": 0.0001447704081632653,
-      "loss": 1.4789,
+      "epoch": 4.89,
+      "learning_rate": 0.0001384943181818182,
+      "loss": 1.2615,
       "step": 433
     },
     {
-      "epoch": 8.78,
-      "learning_rate": 0.00014464285714285715,
-      "loss": 1.5501,
+      "epoch": 4.9,
+      "learning_rate": 0.00013835227272727274,
+      "loss": 1.2754,
       "step": 434
     },
     {
-      "epoch": 8.8,
-      "learning_rate": 0.00014451530612244899,
-      "loss": 1.5204,
+      "epoch": 4.91,
+      "learning_rate": 0.00013821022727272727,
+      "loss": 1.3018,
       "step": 435
     },
     {
-      "epoch": 8.82,
-      "learning_rate": 0.00014438775510204084,
-      "loss": 1.5489,
+      "epoch": 4.93,
+      "learning_rate": 0.00013806818181818183,
+      "loss": 1.2878,
       "step": 436
     },
     {
-      "epoch": 8.84,
-      "learning_rate": 0.00014426020408163265,
-      "loss": 1.5464,
+      "epoch": 4.94,
+      "learning_rate": 0.00013792613636363636,
+      "loss": 1.2595,
       "step": 437
     },
     {
-      "epoch": 8.86,
-      "learning_rate": 0.0001441326530612245,
-      "loss": 1.5896,
+      "epoch": 4.95,
+      "learning_rate": 0.00013778409090909092,
+      "loss": 1.2688,
       "step": 438
     },
     {
-      "epoch": 8.88,
-      "learning_rate": 0.00014400510204081634,
-      "loss": 1.5465,
+      "epoch": 4.96,
+      "learning_rate": 0.00013764204545454548,
+      "loss": 1.2669,
       "step": 439
     },
     {
-      "epoch": 8.9,
-      "learning_rate": 0.00014387755102040817,
-      "loss": 1.5094,
+      "epoch": 4.97,
+      "learning_rate": 0.0001375,
+      "loss": 1.2861,
       "step": 440
     },
     {
-      "epoch": 8.92,
-      "learning_rate": 0.00014375,
-      "loss": 1.5144,
+      "epoch": 4.98,
+      "learning_rate": 0.00013735795454545454,
+      "loss": 1.2536,
       "step": 441
     },
     {
-      "epoch": 8.94,
-      "learning_rate": 0.00014362244897959186,
-      "loss": 1.4919,
+      "epoch": 4.99,
+      "learning_rate": 0.0001372159090909091,
+      "loss": 1.2584,
       "step": 442
     },
     {
-      "epoch": 8.96,
-      "learning_rate": 0.00014349489795918366,
-      "loss": 1.4702,
+      "epoch": 5.0,
+      "learning_rate": 0.00013707386363636366,
+      "loss": 1.2203,
       "step": 443
     },
     {
-      "epoch": 8.98,
-      "learning_rate": 0.00014336734693877552,
-      "loss": 1.4996,
+      "epoch": 5.02,
+      "learning_rate": 0.0001369318181818182,
+      "loss": 1.1796,
       "step": 444
     },
     {
-      "epoch": 9.0,
-      "learning_rate": 0.00014323979591836735,
-      "loss": 1.5503,
+      "epoch": 5.03,
+      "learning_rate": 0.00013678977272727275,
+      "loss": 1.1856,
       "step": 445
     },
     {
-      "epoch": 9.02,
-      "learning_rate": 0.00014311224489795918,
-      "loss": 1.4125,
+      "epoch": 5.04,
+      "learning_rate": 0.00013664772727272728,
+      "loss": 1.1801,
       "step": 446
     },
     {
-      "epoch": 9.04,
-      "learning_rate": 0.00014298469387755104,
-      "loss": 1.4722,
+      "epoch": 5.05,
+      "learning_rate": 0.0001365056818181818,
+      "loss": 1.1761,
       "step": 447
     },
     {
-      "epoch": 9.06,
-      "learning_rate": 0.00014285714285714287,
-      "loss": 1.5199,
+      "epoch": 5.06,
+      "learning_rate": 0.00013636363636363637,
+      "loss": 1.1495,
       "step": 448
     },
     {
-      "epoch": 9.09,
-      "learning_rate": 0.0001427295918367347,
-      "loss": 1.4571,
+      "epoch": 5.07,
+      "learning_rate": 0.00013622159090909093,
+      "loss": 1.1903,
       "step": 449
     },
     {
-      "epoch": 9.11,
-      "learning_rate": 0.00014260204081632653,
-      "loss": 1.4996,
+      "epoch": 5.08,
+      "learning_rate": 0.00013607954545454546,
+      "loss": 1.1778,
       "step": 450
     },
     {
-      "epoch": 9.13,
-      "learning_rate": 0.0001424744897959184,
-      "loss": 1.4092,
+      "epoch": 5.1,
+      "learning_rate": 0.00013593750000000002,
+      "loss": 1.1902,
       "step": 451
     },
     {
-      "epoch": 9.15,
-      "learning_rate": 0.00014234693877551022,
-      "loss": 1.4198,
+      "epoch": 5.11,
+      "learning_rate": 0.00013579545454545455,
+      "loss": 1.1597,
       "step": 452
     },
     {
-      "epoch": 9.17,
-      "learning_rate": 0.00014221938775510205,
-      "loss": 1.4916,
+      "epoch": 5.12,
+      "learning_rate": 0.00013565340909090908,
+      "loss": 1.1529,
       "step": 453
     },
     {
-      "epoch": 9.19,
-      "learning_rate": 0.00014209183673469388,
-      "loss": 1.5051,
+      "epoch": 5.13,
+      "learning_rate": 0.00013551136363636364,
+      "loss": 1.1627,
       "step": 454
     },
     {
-      "epoch": 9.21,
-      "learning_rate": 0.00014196428571428574,
-      "loss": 1.4321,
+      "epoch": 5.14,
+      "learning_rate": 0.0001353693181818182,
+      "loss": 1.1613,
       "step": 455
     },
     {
-      "epoch": 9.23,
-      "learning_rate": 0.00014183673469387754,
-      "loss": 1.4097,
+      "epoch": 5.15,
+      "learning_rate": 0.00013522727272727273,
+      "loss": 1.1336,
       "step": 456
     },
     {
-      "epoch": 9.25,
-      "learning_rate": 0.0001417091836734694,
-      "loss": 1.4853,
+      "epoch": 5.16,
+      "learning_rate": 0.0001350852272727273,
+      "loss": 1.1369,
       "step": 457
     },
     {
-      "epoch": 9.27,
-      "learning_rate": 0.00014158163265306123,
-      "loss": 1.4593,
+      "epoch": 5.17,
+      "learning_rate": 0.00013494318181818182,
+      "loss": 1.1592,
       "step": 458
     },
     {
-      "epoch": 9.29,
-      "learning_rate": 0.00014145408163265306,
-      "loss": 1.3729,
+      "epoch": 5.19,
+      "learning_rate": 0.00013480113636363635,
+      "loss": 1.1482,
       "step": 459
     },
     {
-      "epoch": 9.31,
-      "learning_rate": 0.0001413265306122449,
-      "loss": 1.4467,
+      "epoch": 5.2,
+      "learning_rate": 0.00013465909090909094,
+      "loss": 1.1857,
       "step": 460
     },
     {
-      "epoch": 9.33,
-      "learning_rate": 0.00014119897959183675,
-      "loss": 1.4467,
+      "epoch": 5.21,
+      "learning_rate": 0.00013451704545454547,
+      "loss": 1.1651,
       "step": 461
     },
     {
-      "epoch": 9.35,
-      "learning_rate": 0.00014107142857142858,
-      "loss": 1.4785,
+      "epoch": 5.22,
+      "learning_rate": 0.000134375,
+      "loss": 1.1544,
       "step": 462
     },
     {
-      "epoch": 9.37,
-      "learning_rate": 0.0001409438775510204,
-      "loss": 1.4089,
+      "epoch": 5.23,
+      "learning_rate": 0.00013423295454545456,
+      "loss": 1.125,
       "step": 463
     },
     {
-      "epoch": 9.39,
-      "learning_rate": 0.00014081632653061224,
-      "loss": 1.5026,
+      "epoch": 5.24,
+      "learning_rate": 0.0001340909090909091,
+      "loss": 1.167,
       "step": 464
     },
     {
-      "epoch": 9.41,
-      "learning_rate": 0.0001406887755102041,
-      "loss": 1.4857,
+      "epoch": 5.25,
+      "learning_rate": 0.00013394886363636365,
+      "loss": 1.1316,
       "step": 465
     },
     {
-      "epoch": 9.43,
-      "learning_rate": 0.0001405612244897959,
-      "loss": 1.3745,
+      "epoch": 5.26,
+      "learning_rate": 0.0001338068181818182,
+      "loss": 1.1604,
       "step": 466
     },
     {
-      "epoch": 9.45,
-      "learning_rate": 0.00014043367346938776,
-      "loss": 1.4733,
+      "epoch": 5.28,
+      "learning_rate": 0.00013366477272727274,
+      "loss": 1.2005,
       "step": 467
     },
     {
-      "epoch": 9.47,
-      "learning_rate": 0.0001403061224489796,
-      "loss": 1.5212,
+      "epoch": 5.29,
+      "learning_rate": 0.00013352272727272727,
+      "loss": 1.1496,
       "step": 468
     },
     {
-      "epoch": 9.49,
-      "learning_rate": 0.00014017857142857142,
-      "loss": 1.5398,
+      "epoch": 5.3,
+      "learning_rate": 0.00013338068181818183,
+      "loss": 1.1331,
       "step": 469
     },
     {
-      "epoch": 9.51,
-      "learning_rate": 0.00014005102040816328,
-      "loss": 1.478,
+      "epoch": 5.31,
+      "learning_rate": 0.00013323863636363636,
+      "loss": 1.1414,
       "step": 470
     },
     {
-      "epoch": 9.53,
-      "learning_rate": 0.0001399234693877551,
-      "loss": 1.496,
+      "epoch": 5.32,
+      "learning_rate": 0.00013309659090909092,
+      "loss": 1.0945,
       "step": 471
     },
     {
-      "epoch": 9.55,
-      "learning_rate": 0.00013979591836734694,
-      "loss": 1.4837,
+      "epoch": 5.33,
+      "learning_rate": 0.00013295454545454548,
+      "loss": 1.1305,
       "step": 472
     },
     {
-      "epoch": 9.57,
-      "learning_rate": 0.00013966836734693878,
-      "loss": 1.4724,
+      "epoch": 5.34,
+      "learning_rate": 0.0001328125,
+      "loss": 1.1293,
       "step": 473
     },
     {
-      "epoch": 9.59,
-      "learning_rate": 0.00013954081632653063,
-      "loss": 1.4828,
+      "epoch": 5.35,
+      "learning_rate": 0.00013267045454545454,
+      "loss": 1.163,
       "step": 474
     },
     {
-      "epoch": 9.61,
-      "learning_rate": 0.00013941326530612246,
-      "loss": 1.5012,
+      "epoch": 5.37,
+      "learning_rate": 0.0001325284090909091,
+      "loss": 1.1236,
       "step": 475
     },
     {
-      "epoch": 9.63,
-      "learning_rate": 0.0001392857142857143,
-      "loss": 1.4879,
+      "epoch": 5.38,
+      "learning_rate": 0.00013238636363636366,
+      "loss": 1.1236,
       "step": 476
     },
     {
-      "epoch": 9.65,
-      "learning_rate": 0.00013915816326530613,
-      "loss": 1.4196,
+      "epoch": 5.39,
+      "learning_rate": 0.0001322443181818182,
+      "loss": 1.1228,
       "step": 477
     },
     {
-      "epoch": 9.67,
-      "learning_rate": 0.00013903061224489798,
-      "loss": 1.4915,
+      "epoch": 5.4,
+      "learning_rate": 0.00013210227272727275,
+      "loss": 1.0993,
       "step": 478
     },
     {
-      "epoch": 9.69,
-      "learning_rate": 0.0001389030612244898,
-      "loss": 1.3878,
+      "epoch": 5.41,
+      "learning_rate": 0.00013196022727272728,
+      "loss": 1.1139,
       "step": 479
     },
     {
-      "epoch": 9.71,
-      "learning_rate": 0.00013877551020408165,
-      "loss": 1.466,
+      "epoch": 5.42,
+      "learning_rate": 0.0001318181818181818,
+      "loss": 1.1019,
       "step": 480
     },
     {
-      "epoch": 9.73,
-      "learning_rate": 0.00013864795918367348,
-      "loss": 1.4582,
+      "epoch": 5.43,
+      "learning_rate": 0.00013167613636363637,
+      "loss": 1.0935,
       "step": 481
     },
     {
-      "epoch": 9.75,
-      "learning_rate": 0.0001385204081632653,
-      "loss": 1.533,
+      "epoch": 5.45,
+      "learning_rate": 0.00013153409090909093,
+      "loss": 1.1067,
       "step": 482
     },
     {
-      "epoch": 9.77,
-      "learning_rate": 0.00013839285714285714,
-      "loss": 1.4697,
+      "epoch": 5.46,
+      "learning_rate": 0.00013139204545454546,
+      "loss": 1.0848,
       "step": 483
     },
     {
-      "epoch": 9.79,
-      "learning_rate": 0.000138265306122449,
-      "loss": 1.3989,
+      "epoch": 5.47,
+      "learning_rate": 0.00013125000000000002,
+      "loss": 1.1188,
       "step": 484
     },
     {
-      "epoch": 9.81,
-      "learning_rate": 0.00013813775510204083,
-      "loss": 1.4361,
+      "epoch": 5.48,
+      "learning_rate": 0.00013110795454545455,
+      "loss": 1.1275,
       "step": 485
     },
     {
-      "epoch": 9.83,
-      "learning_rate": 0.00013801020408163266,
-      "loss": 1.5271,
+      "epoch": 5.49,
+      "learning_rate": 0.00013096590909090908,
+      "loss": 1.1211,
       "step": 486
     },
     {
-      "epoch": 9.85,
-      "learning_rate": 0.0001378826530612245,
-      "loss": 1.4905,
+      "epoch": 5.5,
+      "learning_rate": 0.00013082386363636364,
+      "loss": 1.1049,
       "step": 487
     },
     {
-      "epoch": 9.87,
-      "learning_rate": 0.00013775510204081635,
-      "loss": 1.4757,
+      "epoch": 5.51,
+      "learning_rate": 0.0001306818181818182,
+      "loss": 1.1057,
       "step": 488
     },
     {
-      "epoch": 9.89,
-      "learning_rate": 0.00013762755102040815,
-      "loss": 1.5485,
+      "epoch": 5.52,
+      "learning_rate": 0.00013053977272727273,
+      "loss": 1.0909,
       "step": 489
     },
     {
-      "epoch": 9.91,
-      "learning_rate": 0.0001375,
-      "loss": 1.4783,
+      "epoch": 5.54,
+      "learning_rate": 0.0001303977272727273,
+      "loss": 1.1138,
       "step": 490
     },
     {
-      "epoch": 9.93,
-      "learning_rate": 0.00013737244897959184,
-      "loss": 1.4849,
+      "epoch": 5.55,
+      "learning_rate": 0.00013025568181818182,
+      "loss": 1.1094,
       "step": 491
     },
     {
-      "epoch": 9.96,
-      "learning_rate": 0.00013724489795918367,
-      "loss": 1.5382,
+      "epoch": 5.56,
+      "learning_rate": 0.00013011363636363635,
+      "loss": 1.1187,
       "step": 492
     },
     {
-      "epoch": 9.98,
-      "learning_rate": 0.00013711734693877553,
-      "loss": 1.4902,
+      "epoch": 5.57,
+      "learning_rate": 0.0001299715909090909,
+      "loss": 1.1039,
       "step": 493
     },
     {
-      "epoch": 10.0,
-      "learning_rate": 0.00013698979591836736,
-      "loss": 1.4865,
+      "epoch": 5.58,
+      "learning_rate": 0.00012982954545454547,
+      "loss": 1.056,
       "step": 494
     },
     {
-      "epoch": 10.02,
-      "learning_rate": 0.0001368622448979592,
-      "loss": 1.4436,
+      "epoch": 5.59,
+      "learning_rate": 0.0001296875,
+      "loss": 1.0842,
       "step": 495
     },
     {
-      "epoch": 10.04,
-      "learning_rate": 0.00013673469387755102,
-      "loss": 1.408,
+      "epoch": 5.6,
+      "learning_rate": 0.00012954545454545456,
+      "loss": 1.0749,
       "step": 496
     },
     {
-      "epoch": 10.06,
-      "learning_rate": 0.00013660714285714288,
-      "loss": 1.4764,
+      "epoch": 5.61,
+      "learning_rate": 0.0001294034090909091,
+      "loss": 1.1121,
       "step": 497
     },
     {
-      "epoch": 10.08,
-      "learning_rate": 0.0001364795918367347,
-      "loss": 1.4646,
+      "epoch": 5.63,
+      "learning_rate": 0.00012926136363636365,
+      "loss": 1.0772,
       "step": 498
     },
     {
-      "epoch": 10.1,
-      "learning_rate": 0.00013635204081632654,
-      "loss": 1.406,
+      "epoch": 5.64,
+      "learning_rate": 0.00012911931818181818,
+      "loss": 1.0845,
       "step": 499
     },
     {
-      "epoch": 10.12,
-      "learning_rate": 0.00013622448979591837,
-      "loss": 1.4785,
+      "epoch": 5.65,
+      "learning_rate": 0.00012897727272727274,
+      "loss": 1.0534,
       "step": 500
     },
     {
-      "epoch": 10.14,
-      "learning_rate": 0.00013609693877551023,
-      "loss": 1.4117,
+      "epoch": 5.66,
+      "learning_rate": 0.00012883522727272727,
+      "loss": 1.0755,
       "step": 501
     },
     {
-      "epoch": 10.16,
-      "learning_rate": 0.00013596938775510203,
-      "loss": 1.4108,
+      "epoch": 5.67,
+      "learning_rate": 0.00012869318181818183,
+      "loss": 1.0755,
       "step": 502
     },
     {
-      "epoch": 10.18,
-      "learning_rate": 0.0001358418367346939,
-      "loss": 1.4155,
+      "epoch": 5.68,
+      "learning_rate": 0.00012855113636363636,
+      "loss": 1.0869,
       "step": 503
     },
     {
-      "epoch": 10.2,
-      "learning_rate": 0.00013571428571428572,
-      "loss": 1.4021,
+      "epoch": 5.69,
+      "learning_rate": 0.00012840909090909092,
+      "loss": 1.0673,
       "step": 504
     },
     {
-      "epoch": 10.22,
-      "learning_rate": 0.00013558673469387755,
-      "loss": 1.411,
+      "epoch": 5.71,
+      "learning_rate": 0.00012826704545454545,
+      "loss": 1.0692,
       "step": 505
     },
     {
-      "epoch": 10.24,
-      "learning_rate": 0.00013545918367346938,
-      "loss": 1.3851,
+      "epoch": 5.72,
+      "learning_rate": 0.000128125,
+      "loss": 1.0474,
       "step": 506
     },
     {
-      "epoch": 10.26,
-      "learning_rate": 0.00013533163265306124,
-      "loss": 1.387,
+      "epoch": 5.73,
+      "learning_rate": 0.00012798295454545454,
+      "loss": 1.0749,
       "step": 507
     },
     {
-      "epoch": 10.28,
-      "learning_rate": 0.00013520408163265305,
-      "loss": 1.4163,
+      "epoch": 5.74,
+      "learning_rate": 0.0001278409090909091,
+      "loss": 1.0519,
       "step": 508
     },
     {
-      "epoch": 10.3,
-      "learning_rate": 0.0001350765306122449,
-      "loss": 1.3343,
+      "epoch": 5.75,
+      "learning_rate": 0.00012769886363636366,
+      "loss": 1.0566,
       "step": 509
     },
     {
-      "epoch": 10.32,
-      "learning_rate": 0.00013494897959183673,
-      "loss": 1.4811,
+      "epoch": 5.76,
+      "learning_rate": 0.0001275568181818182,
+      "loss": 1.06,
       "step": 510
     },
     {
-      "epoch": 10.34,
-      "learning_rate": 0.0001348214285714286,
-      "loss": 1.4086,
+      "epoch": 5.77,
+      "learning_rate": 0.00012741477272727272,
+      "loss": 1.0618,
       "step": 511
     },
     {
-      "epoch": 10.36,
-      "learning_rate": 0.0001346938775510204,
-      "loss": 1.3879,
+      "epoch": 5.78,
+      "learning_rate": 0.00012727272727272728,
+      "loss": 1.0643,
       "step": 512
     },
     {
-      "epoch": 10.38,
-      "learning_rate": 0.00013456632653061225,
-      "loss": 1.4204,
+      "epoch": 5.8,
+      "learning_rate": 0.0001271306818181818,
+      "loss": 1.026,
       "step": 513
     },
     {
-      "epoch": 10.4,
-      "learning_rate": 0.00013443877551020408,
-      "loss": 1.4158,
+      "epoch": 5.81,
+      "learning_rate": 0.00012698863636363637,
+      "loss": 1.0335,
       "step": 514
     },
     {
-      "epoch": 10.42,
-      "learning_rate": 0.00013431122448979592,
-      "loss": 1.4521,
+      "epoch": 5.82,
+      "learning_rate": 0.00012684659090909093,
+      "loss": 1.0205,
       "step": 515
     },
     {
-      "epoch": 10.44,
-      "learning_rate": 0.00013418367346938777,
-      "loss": 1.4196,
+      "epoch": 5.83,
+      "learning_rate": 0.00012670454545454546,
+      "loss": 1.0594,
       "step": 516
     },
     {
-      "epoch": 10.46,
-      "learning_rate": 0.0001340561224489796,
-      "loss": 1.4361,
+      "epoch": 5.84,
+      "learning_rate": 0.0001265625,
+      "loss": 1.0136,
       "step": 517
     },
     {
-      "epoch": 10.48,
-      "learning_rate": 0.00013392857142857144,
-      "loss": 1.4482,
+      "epoch": 5.85,
+      "learning_rate": 0.00012642045454545455,
+      "loss": 1.0244,
       "step": 518
     },
     {
-      "epoch": 10.5,
-      "learning_rate": 0.00013380102040816327,
-      "loss": 1.4801,
+      "epoch": 5.86,
+      "learning_rate": 0.00012627840909090908,
+      "loss": 1.0569,
       "step": 519
     },
     {
-      "epoch": 10.52,
-      "learning_rate": 0.00013367346938775512,
-      "loss": 1.4556,
+      "epoch": 5.87,
+      "learning_rate": 0.00012613636363636364,
+      "loss": 1.0416,
       "step": 520
     },
     {
-      "epoch": 10.54,
-      "learning_rate": 0.00013354591836734695,
-      "loss": 1.3902,
+      "epoch": 5.89,
+      "learning_rate": 0.0001259943181818182,
+      "loss": 0.9884,
       "step": 521
     },
     {
-      "epoch": 10.56,
-      "learning_rate": 0.00013341836734693879,
-      "loss": 1.4269,
+      "epoch": 5.9,
+      "learning_rate": 0.00012585227272727273,
+      "loss": 1.0351,
       "step": 522
     },
     {
-      "epoch": 10.58,
-      "learning_rate": 0.00013329081632653062,
-      "loss": 1.4899,
+      "epoch": 5.91,
+      "learning_rate": 0.00012571022727272726,
+      "loss": 1.0037,
       "step": 523
     },
     {
-      "epoch": 10.6,
-      "learning_rate": 0.00013316326530612247,
-      "loss": 1.3952,
+      "epoch": 5.92,
+      "learning_rate": 0.00012556818181818182,
+      "loss": 1.0219,
       "step": 524
     },
     {
-      "epoch": 10.62,
-      "learning_rate": 0.00013303571428571428,
-      "loss": 1.4116,
+      "epoch": 5.93,
+      "learning_rate": 0.00012542613636363635,
+      "loss": 1.0533,
       "step": 525
     },
     {
-      "epoch": 10.64,
-      "learning_rate": 0.00013290816326530614,
-      "loss": 1.4583,
+      "epoch": 5.94,
+      "learning_rate": 0.0001252840909090909,
+      "loss": 1.0031,
       "step": 526
     },
     {
-      "epoch": 10.66,
-      "learning_rate": 0.00013278061224489797,
-      "loss": 1.4466,
+      "epoch": 5.95,
+      "learning_rate": 0.00012514204545454547,
+      "loss": 1.0454,
       "step": 527
     },
     {
-      "epoch": 10.68,
-      "learning_rate": 0.0001326530612244898,
-      "loss": 1.4242,
+      "epoch": 5.97,
+      "learning_rate": 0.000125,
+      "loss": 1.0195,
       "step": 528
     },
     {
-      "epoch": 10.7,
-      "learning_rate": 0.00013252551020408163,
-      "loss": 1.3717,
+      "epoch": 5.98,
+      "learning_rate": 0.00012485795454545453,
+      "loss": 1.0076,
       "step": 529
     },
     {
-      "epoch": 10.72,
-      "learning_rate": 0.0001323979591836735,
-      "loss": 1.4583,
+      "epoch": 5.99,
+      "learning_rate": 0.0001247159090909091,
+      "loss": 1.0378,
       "step": 530
     },
     {
-      "epoch": 10.74,
-      "learning_rate": 0.0001322704081632653,
-      "loss": 1.4185,
+      "epoch": 6.0,
+      "learning_rate": 0.00012457386363636365,
+      "loss": 0.9795,
       "step": 531
     },
     {
-      "epoch": 10.76,
-      "learning_rate": 0.00013214285714285715,
-      "loss": 1.4287,
+      "epoch": 6.01,
+      "learning_rate": 0.00012443181818181818,
+      "loss": 0.9405,
       "step": 532
     },
     {
-      "epoch": 10.78,
-      "learning_rate": 0.00013201530612244898,
-      "loss": 1.4385,
+      "epoch": 6.02,
+      "learning_rate": 0.00012428977272727274,
+      "loss": 0.9503,
       "step": 533
     },
     {
-      "epoch": 10.8,
-      "learning_rate": 0.00013188775510204084,
-      "loss": 1.453,
+      "epoch": 6.03,
+      "learning_rate": 0.00012414772727272727,
+      "loss": 0.9456,
       "step": 534
     },
     {
-      "epoch": 10.83,
-      "learning_rate": 0.00013176020408163264,
-      "loss": 1.4161,
+      "epoch": 6.04,
+      "learning_rate": 0.0001240056818181818,
+      "loss": 0.9536,
       "step": 535
     },
     {
-      "epoch": 10.85,
-      "learning_rate": 0.0001316326530612245,
-      "loss": 1.457,
+      "epoch": 6.06,
+      "learning_rate": 0.00012386363636363636,
+      "loss": 0.9412,
       "step": 536
     },
     {
-      "epoch": 10.87,
-      "learning_rate": 0.00013150510204081633,
-      "loss": 1.4367,
+      "epoch": 6.07,
+      "learning_rate": 0.00012372159090909092,
+      "loss": 0.9315,
       "step": 537
     },
     {
-      "epoch": 10.89,
-      "learning_rate": 0.00013137755102040816,
-      "loss": 1.4256,
+      "epoch": 6.08,
+      "learning_rate": 0.00012357954545454545,
+      "loss": 0.9486,
       "step": 538
     },
     {
-      "epoch": 10.91,
-      "learning_rate": 0.00013125000000000002,
-      "loss": 1.424,
+      "epoch": 6.09,
+      "learning_rate": 0.0001234375,
+      "loss": 0.9405,
       "step": 539
     },
     {
-      "epoch": 10.93,
-      "learning_rate": 0.00013112244897959185,
-      "loss": 1.3923,
+      "epoch": 6.1,
+      "learning_rate": 0.00012329545454545454,
+      "loss": 0.9269,
       "step": 540
     },
     {
-      "epoch": 10.95,
-      "learning_rate": 0.00013099489795918368,
-      "loss": 1.4225,
+      "epoch": 6.11,
+      "learning_rate": 0.0001231534090909091,
+      "loss": 0.9378,
       "step": 541
     },
     {
-      "epoch": 10.97,
-      "learning_rate": 0.0001308673469387755,
-      "loss": 1.3969,
+      "epoch": 6.12,
+      "learning_rate": 0.00012301136363636366,
+      "loss": 0.9431,
       "step": 542
     },
     {
-      "epoch": 10.99,
-      "learning_rate": 0.00013073979591836737,
-      "loss": 1.4446,
+      "epoch": 6.13,
+      "learning_rate": 0.0001228693181818182,
+      "loss": 0.9256,
       "step": 543
     },
     {
-      "epoch": 11.01,
-      "learning_rate": 0.00013061224489795917,
-      "loss": 1.4375,
+      "epoch": 6.15,
+      "learning_rate": 0.00012272727272727272,
+      "loss": 0.919,
       "step": 544
     },
     {
-      "epoch": 11.03,
-      "learning_rate": 0.00013048469387755103,
-      "loss": 1.4064,
+      "epoch": 6.16,
+      "learning_rate": 0.00012258522727272728,
+      "loss": 0.9188,
       "step": 545
     },
     {
-      "epoch": 11.05,
-      "learning_rate": 0.00013035714285714286,
-      "loss": 1.3454,
+      "epoch": 6.17,
+      "learning_rate": 0.00012244318181818181,
+      "loss": 0.9447,
       "step": 546
     },
     {
-      "epoch": 11.07,
-      "learning_rate": 0.00013022959183673472,
-      "loss": 1.3234,
+      "epoch": 6.18,
+      "learning_rate": 0.00012230113636363637,
+      "loss": 0.9261,
       "step": 547
     },
     {
-      "epoch": 11.09,
-      "learning_rate": 0.00013010204081632652,
-      "loss": 1.3759,
+      "epoch": 6.19,
+      "learning_rate": 0.00012215909090909093,
+      "loss": 0.9302,
       "step": 548
     },
     {
-      "epoch": 11.11,
-      "learning_rate": 0.00012997448979591838,
-      "loss": 1.4221,
+      "epoch": 6.2,
+      "learning_rate": 0.00012201704545454546,
+      "loss": 0.9161,
       "step": 549
     },
     {
-      "epoch": 11.13,
-      "learning_rate": 0.0001298469387755102,
-      "loss": 1.4261,
+      "epoch": 6.21,
+      "learning_rate": 0.00012187500000000001,
+      "loss": 0.9521,
       "step": 550
     },
     {
-      "epoch": 11.15,
-      "learning_rate": 0.00012971938775510204,
-      "loss": 1.3341,
+      "epoch": 6.22,
+      "learning_rate": 0.00012173295454545455,
+      "loss": 0.9026,
       "step": 551
     },
     {
-      "epoch": 11.17,
-      "learning_rate": 0.00012959183673469387,
-      "loss": 1.3994,
+      "epoch": 6.24,
+      "learning_rate": 0.00012159090909090908,
+      "loss": 0.9361,
       "step": 552
     },
     {
-      "epoch": 11.19,
-      "learning_rate": 0.00012946428571428573,
-      "loss": 1.3894,
+      "epoch": 6.25,
+      "learning_rate": 0.00012144886363636366,
+      "loss": 0.8944,
       "step": 553
     },
     {
-      "epoch": 11.21,
-      "learning_rate": 0.00012933673469387754,
-      "loss": 1.3585,
+      "epoch": 6.26,
+      "learning_rate": 0.00012130681818181819,
+      "loss": 0.895,
       "step": 554
     },
     {
-      "epoch": 11.23,
-      "learning_rate": 0.0001292091836734694,
-      "loss": 1.3763,
+      "epoch": 6.27,
+      "learning_rate": 0.00012116477272727273,
+      "loss": 0.8956,
       "step": 555
     },
     {
-      "epoch": 11.25,
-      "learning_rate": 0.00012908163265306123,
-      "loss": 1.3623,
+      "epoch": 6.28,
+      "learning_rate": 0.00012102272727272728,
+      "loss": 0.8998,
       "step": 556
     },
     {
-      "epoch": 11.27,
-      "learning_rate": 0.00012895408163265306,
-      "loss": 1.3907,
+      "epoch": 6.29,
+      "learning_rate": 0.00012088068181818182,
+      "loss": 0.915,
       "step": 557
     },
     {
-      "epoch": 11.29,
-      "learning_rate": 0.0001288265306122449,
-      "loss": 1.3807,
+      "epoch": 6.3,
+      "learning_rate": 0.00012073863636363636,
+      "loss": 0.9282,
       "step": 558
     },
     {
-      "epoch": 11.31,
-      "learning_rate": 0.00012869897959183674,
-      "loss": 1.4045,
+      "epoch": 6.32,
+      "learning_rate": 0.00012059659090909093,
+      "loss": 0.8938,
       "step": 559
     },
     {
-      "epoch": 11.33,
-      "learning_rate": 0.00012857142857142858,
-      "loss": 1.4038,
+      "epoch": 6.33,
+      "learning_rate": 0.00012045454545454546,
+      "loss": 0.8886,
       "step": 560
     },
     {
-      "epoch": 11.35,
-      "learning_rate": 0.0001284438775510204,
-      "loss": 1.3466,
+      "epoch": 6.34,
+      "learning_rate": 0.0001203125,
+      "loss": 0.8988,
       "step": 561
     },
     {
-      "epoch": 11.37,
-      "learning_rate": 0.00012831632653061226,
-      "loss": 1.3449,
+      "epoch": 6.35,
+      "learning_rate": 0.00012017045454545455,
+      "loss": 0.8852,
       "step": 562
     },
     {
-      "epoch": 11.39,
-      "learning_rate": 0.0001281887755102041,
-      "loss": 1.3866,
+      "epoch": 6.36,
+      "learning_rate": 0.0001200284090909091,
+      "loss": 0.8818,
       "step": 563
     },
     {
-      "epoch": 11.41,
-      "learning_rate": 0.00012806122448979593,
-      "loss": 1.3106,
+      "epoch": 6.37,
+      "learning_rate": 0.00011988636363636365,
+      "loss": 0.8881,
       "step": 564
     },
     {
-      "epoch": 11.43,
-      "learning_rate": 0.00012793367346938776,
-      "loss": 1.4414,
+      "epoch": 6.38,
+      "learning_rate": 0.0001197443181818182,
+      "loss": 0.9226,
       "step": 565
     },
     {
-      "epoch": 11.45,
-      "learning_rate": 0.00012780612244897962,
-      "loss": 1.3737,
+      "epoch": 6.39,
+      "learning_rate": 0.00011960227272727273,
+      "loss": 0.8849,
       "step": 566
     },
     {
-      "epoch": 11.47,
-      "learning_rate": 0.00012767857142857142,
-      "loss": 1.4053,
+      "epoch": 6.41,
+      "learning_rate": 0.00011946022727272727,
+      "loss": 0.8894,
       "step": 567
     },
     {
-      "epoch": 11.49,
-      "learning_rate": 0.00012755102040816328,
-      "loss": 1.4561,
+      "epoch": 6.42,
+      "learning_rate": 0.00011931818181818182,
+      "loss": 0.9207,
       "step": 568
     },
     {
-      "epoch": 11.51,
-      "learning_rate": 0.0001274234693877551,
-      "loss": 1.3684,
+      "epoch": 6.43,
+      "learning_rate": 0.00011917613636363636,
+      "loss": 0.9105,
       "step": 569
     },
     {
-      "epoch": 11.53,
-      "learning_rate": 0.00012729591836734697,
-      "loss": 1.3117,
+      "epoch": 6.44,
+      "learning_rate": 0.00011903409090909092,
+      "loss": 0.8762,
       "step": 570
     },
     {
-      "epoch": 11.55,
-      "learning_rate": 0.00012716836734693877,
-      "loss": 1.3474,
+      "epoch": 6.45,
+      "learning_rate": 0.00011889204545454547,
+      "loss": 0.8926,
       "step": 571
     },
     {
-      "epoch": 11.57,
-      "learning_rate": 0.00012704081632653063,
-      "loss": 1.3804,
+      "epoch": 6.46,
+      "learning_rate": 0.00011875,
+      "loss": 0.8719,
       "step": 572
     },
     {
-      "epoch": 11.59,
-      "learning_rate": 0.00012691326530612246,
-      "loss": 1.3656,
+      "epoch": 6.47,
+      "learning_rate": 0.00011860795454545454,
+      "loss": 0.9198,
       "step": 573
     },
     {
-      "epoch": 11.61,
-      "learning_rate": 0.0001267857142857143,
-      "loss": 1.3133,
+      "epoch": 6.48,
+      "learning_rate": 0.00011846590909090909,
+      "loss": 0.8846,
       "step": 574
     },
     {
-      "epoch": 11.63,
-      "learning_rate": 0.00012665816326530612,
-      "loss": 1.4077,
+      "epoch": 6.5,
+      "learning_rate": 0.00011832386363636365,
+      "loss": 0.8495,
       "step": 575
     },
     {
-      "epoch": 11.65,
-      "learning_rate": 0.00012653061224489798,
-      "loss": 1.4087,
+      "epoch": 6.51,
+      "learning_rate": 0.0001181818181818182,
+      "loss": 0.8953,
       "step": 576
     },
     {
-      "epoch": 11.67,
-      "learning_rate": 0.00012640306122448978,
-      "loss": 1.3524,
+      "epoch": 6.52,
+      "learning_rate": 0.00011803977272727274,
+      "loss": 0.8686,
       "step": 577
     },
     {
-      "epoch": 11.7,
-      "learning_rate": 0.00012627551020408164,
-      "loss": 1.3481,
+      "epoch": 6.53,
+      "learning_rate": 0.00011789772727272727,
+      "loss": 0.8841,
       "step": 578
     },
     {
-      "epoch": 11.72,
-      "learning_rate": 0.00012614795918367347,
-      "loss": 1.4497,
+      "epoch": 6.54,
+      "learning_rate": 0.00011775568181818182,
+      "loss": 0.8681,
       "step": 579
     },
     {
-      "epoch": 11.74,
-      "learning_rate": 0.0001260204081632653,
-      "loss": 1.3866,
+      "epoch": 6.55,
+      "learning_rate": 0.00011761363636363636,
+      "loss": 0.8732,
       "step": 580
     },
     {
-      "epoch": 11.76,
-      "learning_rate": 0.00012589285714285713,
-      "loss": 1.42,
+      "epoch": 6.56,
+      "learning_rate": 0.00011747159090909092,
+      "loss": 0.8582,
       "step": 581
     },
     {
-      "epoch": 11.78,
-      "learning_rate": 0.000125765306122449,
-      "loss": 1.3562,
+      "epoch": 6.58,
+      "learning_rate": 0.00011732954545454546,
+      "loss": 0.8744,
       "step": 582
     },
     {
-      "epoch": 11.8,
-      "learning_rate": 0.00012563775510204082,
-      "loss": 1.3249,
+      "epoch": 6.59,
+      "learning_rate": 0.00011718750000000001,
+      "loss": 0.8694,
       "step": 583
     },
     {
-      "epoch": 11.82,
-      "learning_rate": 0.00012551020408163265,
-      "loss": 1.4277,
+      "epoch": 6.6,
+      "learning_rate": 0.00011704545454545454,
+      "loss": 0.8565,
       "step": 584
     },
     {
-      "epoch": 11.84,
-      "learning_rate": 0.0001253826530612245,
-      "loss": 1.3734,
+      "epoch": 6.61,
+      "learning_rate": 0.00011690340909090909,
+      "loss": 0.8584,
       "step": 585
     },
     {
-      "epoch": 11.86,
-      "learning_rate": 0.00012525510204081634,
-      "loss": 1.3765,
+      "epoch": 6.62,
+      "learning_rate": 0.00011676136363636366,
+      "loss": 0.8859,
       "step": 586
     },
     {
-      "epoch": 11.88,
-      "learning_rate": 0.00012512755102040817,
-      "loss": 1.4153,
+      "epoch": 6.63,
+      "learning_rate": 0.00011661931818181819,
+      "loss": 0.8452,
       "step": 587
     },
     {
-      "epoch": 11.9,
-      "learning_rate": 0.000125,
-      "loss": 1.3847,
+      "epoch": 6.64,
+      "learning_rate": 0.00011647727272727273,
+      "loss": 0.8323,
       "step": 588
     },
     {
-      "epoch": 11.92,
-      "learning_rate": 0.00012487244897959186,
-      "loss": 1.3824,
+      "epoch": 6.65,
+      "learning_rate": 0.00011633522727272728,
+      "loss": 0.8548,
       "step": 589
     },
     {
-      "epoch": 11.94,
-      "learning_rate": 0.00012474489795918366,
-      "loss": 1.3938,
+      "epoch": 6.67,
+      "learning_rate": 0.00011619318181818181,
+      "loss": 0.8506,
       "step": 590
     },
     {
-      "epoch": 11.96,
-      "learning_rate": 0.00012461734693877552,
-      "loss": 1.4143,
+      "epoch": 6.68,
+      "learning_rate": 0.00011605113636363636,
+      "loss": 0.8556,
       "step": 591
     },
     {
-      "epoch": 11.98,
-      "learning_rate": 0.00012448979591836735,
-      "loss": 1.3794,
+      "epoch": 6.69,
+      "learning_rate": 0.00011590909090909093,
+      "loss": 0.8459,
       "step": 592
     },
     {
-      "epoch": 12.0,
-      "learning_rate": 0.00012436224489795918,
-      "loss": 1.3755,
+      "epoch": 6.7,
+      "learning_rate": 0.00011576704545454546,
+      "loss": 0.8432,
       "step": 593
     },
     {
-      "epoch": 12.02,
-      "learning_rate": 0.00012423469387755101,
-      "loss": 1.3736,
+      "epoch": 6.71,
+      "learning_rate": 0.000115625,
+      "loss": 0.8645,
       "step": 594
     },
     {
-      "epoch": 12.04,
-      "learning_rate": 0.00012410714285714287,
-      "loss": 1.2957,
+      "epoch": 6.72,
+      "learning_rate": 0.00011548295454545455,
+      "loss": 0.86,
       "step": 595
     },
     {
-      "epoch": 12.06,
-      "learning_rate": 0.0001239795918367347,
-      "loss": 1.2996,
+      "epoch": 6.73,
+      "learning_rate": 0.00011534090909090908,
+      "loss": 0.8161,
       "step": 596
     },
     {
-      "epoch": 12.08,
-      "learning_rate": 0.00012385204081632653,
-      "loss": 1.3648,
+      "epoch": 6.74,
+      "learning_rate": 0.00011519886363636365,
+      "loss": 0.8133,
       "step": 597
     },
     {
-      "epoch": 12.1,
-      "learning_rate": 0.00012372448979591837,
-      "loss": 1.3031,
+      "epoch": 6.76,
+      "learning_rate": 0.0001150568181818182,
+      "loss": 0.8372,
       "step": 598
     },
     {
-      "epoch": 12.12,
-      "learning_rate": 0.00012359693877551022,
-      "loss": 1.2933,
+      "epoch": 6.77,
+      "learning_rate": 0.00011491477272727273,
+      "loss": 0.8222,
       "step": 599
     },
     {
-      "epoch": 12.14,
-      "learning_rate": 0.00012346938775510203,
-      "loss": 1.322,
+      "epoch": 6.78,
+      "learning_rate": 0.00011477272727272728,
+      "loss": 0.8372,
       "step": 600
     }
   ],
   "logging_steps": 1,
-  "max_steps": 1568,
-  "num_train_epochs": 32,
+  "max_steps": 1408,
+  "num_train_epochs": 16,
   "save_steps": 100,
-  "total_flos": 8.877519690112819e+17,
+  "total_flos": 8.186165472352358e+17,
   "trial_name": null,
   "trial_params": null
 }