diff --git "a/vicuna-33B-lora/trainer_state.json" "b/vicuna-33B-lora/trainer_state.json"
deleted file mode 100644--- "a/vicuna-33B-lora/trainer_state.json"
+++ /dev/null
@@ -1,23727 +0,0 @@
-{
-  "best_metric": null,
-  "best_model_checkpoint": null,
-  "epoch": 6.99866844207723,
-  "eval_steps": 500,
-  "global_step": 3942,
-  "is_hyper_param_search": false,
-  "is_local_process_zero": true,
-  "is_world_process_zero": true,
-  "log_history": [
-    {
-      "epoch": 0.0,
-      "learning_rate": 4.728132387706856e-08,
-      "loss": 1.6865,
-      "step": 1
-    },
-    {
-      "epoch": 0.0,
-      "learning_rate": 9.456264775413712e-08,
-      "loss": 1.6434,
-      "step": 2
-    },
-    {
-      "epoch": 0.01,
-      "learning_rate": 1.4184397163120568e-07,
-      "loss": 1.7062,
-      "step": 3
-    },
-    {
-      "epoch": 0.01,
-      "learning_rate": 1.8912529550827425e-07,
-      "loss": 1.7049,
-      "step": 4
-    },
-    {
-      "epoch": 0.01,
-      "learning_rate": 2.364066193853428e-07,
-      "loss": 1.7256,
-      "step": 5
-    },
-    {
-      "epoch": 0.01,
-      "learning_rate": 2.8368794326241136e-07,
-      "loss": 1.6642,
-      "step": 6
-    },
-    {
-      "epoch": 0.01,
-      "learning_rate": 3.309692671394799e-07,
-      "loss": 1.716,
-      "step": 7
-    },
-    {
-      "epoch": 0.01,
-      "learning_rate": 3.782505910165485e-07,
-      "loss": 1.7305,
-      "step": 8
-    },
-    {
-      "epoch": 0.02,
-      "learning_rate": 4.2553191489361704e-07,
-      "loss": 1.7293,
-      "step": 9
-    },
-    {
-      "epoch": 0.02,
-      "learning_rate": 4.728132387706856e-07,
-      "loss": 1.664,
-      "step": 10
-    },
-    {
-      "epoch": 0.02,
-      "learning_rate": 5.200945626477541e-07,
-      "loss": 1.6494,
-      "step": 11
-    },
-    {
-      "epoch": 0.02,
-      "learning_rate": 5.673758865248227e-07,
-      "loss": 1.67,
-      "step": 12
-    },
-    {
-      "epoch": 0.02,
-      "learning_rate": 6.146572104018913e-07,
-      "loss": 1.7033,
-      "step": 13
-    },
-    {
-      "epoch": 0.02,
-      "learning_rate": 6.619385342789598e-07,
-      "loss": 1.6824,
-      "step": 14
-    },
-    {
-      "epoch": 0.03,
-      "learning_rate": 7.092198581560285e-07,
-      "loss": 1.689,
-      "step": 15
-    },
-    {
-      "epoch": 0.03,
-      "learning_rate": 7.56501182033097e-07,
-      "loss": 1.6849,
-      "step": 16
-    },
-    {
-      "epoch": 0.03,
-      "learning_rate": 8.037825059101656e-07,
-      "loss": 1.6966,
-      "step": 17
-    },
-    {
-      "epoch": 0.03,
-      "learning_rate": 8.510638297872341e-07,
-      "loss": 1.7069,
-      "step": 18
-    },
-    {
-      "epoch": 0.03,
-      "learning_rate": 8.983451536643027e-07,
-      "loss": 1.6271,
-      "step": 19
-    },
-    {
-      "epoch": 0.04,
-      "learning_rate": 9.456264775413712e-07,
-      "loss": 1.7064,
-      "step": 20
-    },
-    {
-      "epoch": 0.04,
-      "learning_rate": 9.929078014184399e-07,
-      "loss": 1.712,
-      "step": 21
-    },
-    {
-      "epoch": 0.04,
-      "learning_rate": 1.0401891252955083e-06,
-      "loss": 1.6596,
-      "step": 22
-    },
-    {
-      "epoch": 0.04,
-      "learning_rate": 1.087470449172577e-06,
-      "loss": 1.6598,
-      "step": 23
-    },
-    {
-      "epoch": 0.04,
-      "learning_rate": 1.1347517730496454e-06,
-      "loss": 1.6642,
-      "step": 24
-    },
-    {
-      "epoch": 0.04,
-      "learning_rate": 1.182033096926714e-06,
-      "loss": 1.705,
-      "step": 25
-    },
-    {
-      "epoch": 0.05,
-      "learning_rate": 1.2293144208037826e-06,
-      "loss": 1.646,
-      "step": 26
-    },
-    {
-      "epoch": 0.05,
-      "learning_rate": 1.276595744680851e-06,
-      "loss": 1.7031,
-      "step": 27
-    },
-    {
-      "epoch": 0.05,
-      "learning_rate": 1.3238770685579196e-06,
-      "loss": 1.7459,
-      "step": 28
-    },
-    {
-      "epoch": 0.05,
-      "learning_rate": 1.3711583924349884e-06,
-      "loss": 1.6929,
-      "step": 29
-    },
-    {
-      "epoch": 0.05,
-      "learning_rate": 1.418439716312057e-06,
-      "loss": 1.6768,
-      "step": 30
-    },
-    {
-      "epoch": 0.06,
-      "learning_rate": 1.4657210401891254e-06,
-      "loss": 1.6704,
-      "step": 31
-    },
-    {
-      "epoch": 0.06,
-      "learning_rate": 1.513002364066194e-06,
-      "loss": 1.6803,
-      "step": 32
-    },
-    {
-      "epoch": 0.06,
-      "learning_rate": 1.5602836879432626e-06,
-      "loss": 1.7174,
-      "step": 33
-    },
-    {
-      "epoch": 0.06,
-      "learning_rate": 1.6075650118203312e-06,
-      "loss": 1.6946,
-      "step": 34
-    },
-    {
-      "epoch": 0.06,
-      "learning_rate": 1.6548463356973996e-06,
-      "loss": 1.6502,
-      "step": 35
-    },
-    {
-      "epoch": 0.06,
-      "learning_rate": 1.7021276595744682e-06,
-      "loss": 1.6575,
-      "step": 36
-    },
-    {
-      "epoch": 0.07,
-      "learning_rate": 1.7494089834515368e-06,
-      "loss": 1.6671,
-      "step": 37
-    },
-    {
-      "epoch": 0.07,
-      "learning_rate": 1.7966903073286054e-06,
-      "loss": 1.7631,
-      "step": 38
-    },
-    {
-      "epoch": 0.07,
-      "learning_rate": 1.8439716312056737e-06,
-      "loss": 1.6923,
-      "step": 39
-    },
-    {
-      "epoch": 0.07,
-      "learning_rate": 1.8912529550827423e-06,
-      "loss": 1.6643,
-      "step": 40
-    },
-    {
-      "epoch": 0.07,
-      "learning_rate": 1.938534278959811e-06,
-      "loss": 1.6965,
-      "step": 41
-    },
-    {
-      "epoch": 0.07,
-      "learning_rate": 1.9858156028368797e-06,
-      "loss": 1.7151,
-      "step": 42
-    },
-    {
-      "epoch": 0.08,
-      "learning_rate": 2.033096926713948e-06,
-      "loss": 1.673,
-      "step": 43
-    },
-    {
-      "epoch": 0.08,
-      "learning_rate": 2.0803782505910165e-06,
-      "loss": 1.7265,
-      "step": 44
-    },
-    {
-      "epoch": 0.08,
-      "learning_rate": 2.1276595744680853e-06,
-      "loss": 1.7043,
-      "step": 45
-    },
-    {
-      "epoch": 0.08,
-      "learning_rate": 2.174940898345154e-06,
-      "loss": 1.6742,
-      "step": 46
-    },
-    {
-      "epoch": 0.08,
-      "learning_rate": 2.222222222222222e-06,
-      "loss": 1.6945,
-      "step": 47
-    },
-    {
-      "epoch": 0.09,
-      "learning_rate": 2.269503546099291e-06,
-      "loss": 1.6766,
-      "step": 48
-    },
-    {
-      "epoch": 0.09,
-      "learning_rate": 2.3167848699763597e-06,
-      "loss": 1.7333,
-      "step": 49
-    },
-    {
-      "epoch": 0.09,
-      "learning_rate": 2.364066193853428e-06,
-      "loss": 1.7095,
-      "step": 50
-    },
-    {
-      "epoch": 0.09,
-      "learning_rate": 2.4113475177304965e-06,
-      "loss": 1.7088,
-      "step": 51
-    },
-    {
-      "epoch": 0.09,
-      "learning_rate": 2.4586288416075653e-06,
-      "loss": 1.7187,
-      "step": 52
-    },
-    {
-      "epoch": 0.09,
-      "learning_rate": 2.5059101654846336e-06,
-      "loss": 1.6437,
-      "step": 53
-    },
-    {
-      "epoch": 0.1,
-      "learning_rate": 2.553191489361702e-06,
-      "loss": 1.6646,
-      "step": 54
-    },
-    {
-      "epoch": 0.1,
-      "learning_rate": 2.600472813238771e-06,
-      "loss": 1.6744,
-      "step": 55
-    },
-    {
-      "epoch": 0.1,
-      "learning_rate": 2.6477541371158392e-06,
-      "loss": 1.663,
-      "step": 56
-    },
-    {
-      "epoch": 0.1,
-      "learning_rate": 2.695035460992908e-06,
-      "loss": 1.6856,
-      "step": 57
-    },
-    {
-      "epoch": 0.1,
-      "learning_rate": 2.742316784869977e-06,
-      "loss": 1.6995,
-      "step": 58
-    },
-    {
-      "epoch": 0.1,
-      "learning_rate": 2.7895981087470452e-06,
-      "loss": 1.7023,
-      "step": 59
-    },
-    {
-      "epoch": 0.11,
-      "learning_rate": 2.836879432624114e-06,
-      "loss": 1.707,
-      "step": 60
-    },
-    {
-      "epoch": 0.11,
-      "learning_rate": 2.8841607565011824e-06,
-      "loss": 1.6746,
-      "step": 61
-    },
-    {
-      "epoch": 0.11,
-      "learning_rate": 2.931442080378251e-06,
-      "loss": 1.688,
-      "step": 62
-    },
-    {
-      "epoch": 0.11,
-      "learning_rate": 2.978723404255319e-06,
-      "loss": 1.6785,
-      "step": 63
-    },
-    {
-      "epoch": 0.11,
-      "learning_rate": 3.026004728132388e-06,
-      "loss": 1.6359,
-      "step": 64
-    },
-    {
-      "epoch": 0.12,
-      "learning_rate": 3.0732860520094564e-06,
-      "loss": 1.6657,
-      "step": 65
-    },
-    {
-      "epoch": 0.12,
-      "learning_rate": 3.120567375886525e-06,
-      "loss": 1.6816,
-      "step": 66
-    },
-    {
-      "epoch": 0.12,
-      "learning_rate": 3.167848699763594e-06,
-      "loss": 1.702,
-      "step": 67
-    },
-    {
-      "epoch": 0.12,
-      "learning_rate": 3.2151300236406624e-06,
-      "loss": 1.6748,
-      "step": 68
-    },
-    {
-      "epoch": 0.12,
-      "learning_rate": 3.262411347517731e-06,
-      "loss": 1.701,
-      "step": 69
-    },
-    {
-      "epoch": 0.12,
-      "learning_rate": 3.309692671394799e-06,
-      "loss": 1.7053,
-      "step": 70
-    },
-    {
-      "epoch": 0.13,
-      "learning_rate": 3.356973995271868e-06,
-      "loss": 1.6736,
-      "step": 71
-    },
-    {
-      "epoch": 0.13,
-      "learning_rate": 3.4042553191489363e-06,
-      "loss": 1.7129,
-      "step": 72
-    },
-    {
-      "epoch": 0.13,
-      "learning_rate": 3.451536643026005e-06,
-      "loss": 1.684,
-      "step": 73
-    },
-    {
-      "epoch": 0.13,
-      "learning_rate": 3.4988179669030735e-06,
-      "loss": 1.6599,
-      "step": 74
-    },
-    {
-      "epoch": 0.13,
-      "learning_rate": 3.5460992907801423e-06,
-      "loss": 1.6522,
-      "step": 75
-    },
-    {
-      "epoch": 0.13,
-      "learning_rate": 3.5933806146572107e-06,
-      "loss": 1.6679,
-      "step": 76
-    },
-    {
-      "epoch": 0.14,
-      "learning_rate": 3.6406619385342795e-06,
-      "loss": 1.6652,
-      "step": 77
-    },
-    {
-      "epoch": 0.14,
-      "learning_rate": 3.6879432624113475e-06,
-      "loss": 1.6346,
-      "step": 78
-    },
-    {
-      "epoch": 0.14,
-      "learning_rate": 3.7352245862884163e-06,
-      "loss": 1.6526,
-      "step": 79
-    },
-    {
-      "epoch": 0.14,
-      "learning_rate": 3.7825059101654847e-06,
-      "loss": 1.7132,
-      "step": 80
-    },
-    {
-      "epoch": 0.14,
-      "learning_rate": 3.8297872340425535e-06,
-      "loss": 1.6648,
-      "step": 81
-    },
-    {
-      "epoch": 0.15,
-      "learning_rate": 3.877068557919622e-06,
-      "loss": 1.6323,
-      "step": 82
-    },
-    {
-      "epoch": 0.15,
-      "learning_rate": 3.924349881796691e-06,
-      "loss": 1.6766,
-      "step": 83
-    },
-    {
-      "epoch": 0.15,
-      "learning_rate": 3.9716312056737595e-06,
-      "loss": 1.6633,
-      "step": 84
-    },
-    {
-      "epoch": 0.15,
-      "learning_rate": 4.018912529550828e-06,
-      "loss": 1.6544,
-      "step": 85
-    },
-    {
-      "epoch": 0.15,
-      "learning_rate": 4.066193853427896e-06,
-      "loss": 1.6443,
-      "step": 86
-    },
-    {
-      "epoch": 0.15,
-      "learning_rate": 4.113475177304965e-06,
-      "loss": 1.6711,
-      "step": 87
-    },
-    {
-      "epoch": 0.16,
-      "learning_rate": 4.160756501182033e-06,
-      "loss": 1.6611,
-      "step": 88
-    },
-    {
-      "epoch": 0.16,
-      "learning_rate": 4.208037825059102e-06,
-      "loss": 1.6033,
-      "step": 89
-    },
-    {
-      "epoch": 0.16,
-      "learning_rate": 4.255319148936171e-06,
-      "loss": 1.6512,
-      "step": 90
-    },
-    {
-      "epoch": 0.16,
-      "learning_rate": 4.302600472813239e-06,
-      "loss": 1.6802,
-      "step": 91
-    },
-    {
-      "epoch": 0.16,
-      "learning_rate": 4.349881796690308e-06,
-      "loss": 1.6359,
-      "step": 92
-    },
-    {
-      "epoch": 0.17,
-      "learning_rate": 4.397163120567377e-06,
-      "loss": 1.633,
-      "step": 93
-    },
-    {
-      "epoch": 0.17,
-      "learning_rate": 4.444444444444444e-06,
-      "loss": 1.6181,
-      "step": 94
-    },
-    {
-      "epoch": 0.17,
-      "learning_rate": 4.491725768321513e-06,
-      "loss": 1.6368,
-      "step": 95
-    },
-    {
-      "epoch": 0.17,
-      "learning_rate": 4.539007092198582e-06,
-      "loss": 1.6496,
-      "step": 96
-    },
-    {
-      "epoch": 0.17,
-      "learning_rate": 4.58628841607565e-06,
-      "loss": 1.6017,
-      "step": 97
-    },
-    {
-      "epoch": 0.17,
-      "learning_rate": 4.633569739952719e-06,
-      "loss": 1.6274,
-      "step": 98
-    },
-    {
-      "epoch": 0.18,
-      "learning_rate": 4.680851063829788e-06,
-      "loss": 1.6176,
-      "step": 99
-    },
-    {
-      "epoch": 0.18,
-      "learning_rate": 4.728132387706856e-06,
-      "loss": 1.5796,
-      "step": 100
-    },
-    {
-      "epoch": 0.18,
-      "learning_rate": 4.775413711583925e-06,
-      "loss": 1.6057,
-      "step": 101
-    },
-    {
-      "epoch": 0.18,
-      "learning_rate": 4.822695035460993e-06,
-      "loss": 1.6145,
-      "step": 102
-    },
-    {
-      "epoch": 0.18,
-      "learning_rate": 4.869976359338061e-06,
-      "loss": 1.5817,
-      "step": 103
-    },
-    {
-      "epoch": 0.18,
-      "learning_rate": 4.9172576832151305e-06,
-      "loss": 1.5529,
-      "step": 104
-    },
-    {
-      "epoch": 0.19,
-      "learning_rate": 4.964539007092199e-06,
-      "loss": 1.5753,
-      "step": 105
-    },
-    {
-      "epoch": 0.19,
-      "learning_rate": 5.011820330969267e-06,
-      "loss": 1.5933,
-      "step": 106
-    },
-    {
-      "epoch": 0.19,
-      "learning_rate": 5.059101654846336e-06,
-      "loss": 1.5487,
-      "step": 107
-    },
-    {
-      "epoch": 0.19,
-      "learning_rate": 5.106382978723404e-06,
-      "loss": 1.5498,
-      "step": 108
-    },
-    {
-      "epoch": 0.19,
-      "learning_rate": 5.153664302600473e-06,
-      "loss": 1.5477,
-      "step": 109
-    },
-    {
-      "epoch": 0.2,
-      "learning_rate": 5.200945626477542e-06,
-      "loss": 1.5677,
-      "step": 110
-    },
-    {
-      "epoch": 0.2,
-      "learning_rate": 5.24822695035461e-06,
-      "loss": 1.5645,
-      "step": 111
-    },
-    {
-      "epoch": 0.2,
-      "learning_rate": 5.2955082742316784e-06,
-      "loss": 1.5034,
-      "step": 112
-    },
-    {
-      "epoch": 0.2,
-      "learning_rate": 5.342789598108748e-06,
-      "loss": 1.5522,
-      "step": 113
-    },
-    {
-      "epoch": 0.2,
-      "learning_rate": 5.390070921985816e-06,
-      "loss": 1.507,
-      "step": 114
-    },
-    {
-      "epoch": 0.2,
-      "learning_rate": 5.4373522458628844e-06,
-      "loss": 1.5375,
-      "step": 115
-    },
-    {
-      "epoch": 0.21,
-      "learning_rate": 5.484633569739954e-06,
-      "loss": 1.5016,
-      "step": 116
-    },
-    {
-      "epoch": 0.21,
-      "learning_rate": 5.531914893617022e-06,
-      "loss": 1.5283,
-      "step": 117
-    },
-    {
-      "epoch": 0.21,
-      "learning_rate": 5.5791962174940904e-06,
-      "loss": 1.521,
-      "step": 118
-    },
-    {
-      "epoch": 0.21,
-      "learning_rate": 5.626477541371159e-06,
-      "loss": 1.5166,
-      "step": 119
-    },
-    {
-      "epoch": 0.21,
-      "learning_rate": 5.673758865248228e-06,
-      "loss": 1.4447,
-      "step": 120
-    },
-    {
-      "epoch": 0.21,
-      "learning_rate": 5.7210401891252964e-06,
-      "loss": 1.5281,
-      "step": 121
-    },
-    {
-      "epoch": 0.22,
-      "learning_rate": 5.768321513002365e-06,
-      "loss": 1.4504,
-      "step": 122
-    },
-    {
-      "epoch": 0.22,
-      "learning_rate": 5.815602836879432e-06,
-      "loss": 1.4717,
-      "step": 123
-    },
-    {
-      "epoch": 0.22,
-      "learning_rate": 5.862884160756502e-06,
-      "loss": 1.4371,
-      "step": 124
-    },
-    {
-      "epoch": 0.22,
-      "learning_rate": 5.91016548463357e-06,
-      "loss": 1.4725,
-      "step": 125
-    },
-    {
-      "epoch": 0.22,
-      "learning_rate": 5.957446808510638e-06,
-      "loss": 1.4551,
-      "step": 126
-    },
-    {
-      "epoch": 0.23,
-      "learning_rate": 6.004728132387707e-06,
-      "loss": 1.4531,
-      "step": 127
-    },
-    {
-      "epoch": 0.23,
-      "learning_rate": 6.052009456264776e-06,
-      "loss": 1.4799,
-      "step": 128
-    },
-    {
-      "epoch": 0.23,
-      "learning_rate": 6.099290780141844e-06,
-      "loss": 1.4334,
-      "step": 129
-    },
-    {
-      "epoch": 0.23,
-      "learning_rate": 6.146572104018913e-06,
-      "loss": 1.4047,
-      "step": 130
-    },
-    {
-      "epoch": 0.23,
-      "learning_rate": 6.193853427895982e-06,
-      "loss": 1.4176,
-      "step": 131
-    },
-    {
-      "epoch": 0.23,
-      "learning_rate": 6.24113475177305e-06,
-      "loss": 1.3964,
-      "step": 132
-    },
-    {
-      "epoch": 0.24,
-      "learning_rate": 6.288416075650119e-06,
-      "loss": 1.3788,
-      "step": 133
-    },
-    {
-      "epoch": 0.24,
-      "learning_rate": 6.335697399527188e-06,
-      "loss": 1.3766,
-      "step": 134
-    },
-    {
-      "epoch": 0.24,
-      "learning_rate": 6.382978723404256e-06,
-      "loss": 1.3991,
-      "step": 135
-    },
-    {
-      "epoch": 0.24,
-      "learning_rate": 6.430260047281325e-06,
-      "loss": 1.382,
-      "step": 136
-    },
-    {
-      "epoch": 0.24,
-      "learning_rate": 6.477541371158393e-06,
-      "loss": 1.3919,
-      "step": 137
-    },
-    {
-      "epoch": 0.25,
-      "learning_rate": 6.524822695035462e-06,
-      "loss": 1.3731,
-      "step": 138
-    },
-    {
-      "epoch": 0.25,
-      "learning_rate": 6.57210401891253e-06,
-      "loss": 1.3417,
-      "step": 139
-    },
-    {
-      "epoch": 0.25,
-      "learning_rate": 6.619385342789598e-06,
-      "loss": 1.3178,
-      "step": 140
-    },
-    {
-      "epoch": 0.25,
-      "learning_rate": 6.666666666666667e-06,
-      "loss": 1.3421,
-      "step": 141
-    },
-    {
-      "epoch": 0.25,
-      "learning_rate": 6.713947990543736e-06,
-      "loss": 1.3609,
-      "step": 142
-    },
-    {
-      "epoch": 0.25,
-      "learning_rate": 6.761229314420804e-06,
-      "loss": 1.3476,
-      "step": 143
-    },
-    {
-      "epoch": 0.26,
-      "learning_rate": 6.808510638297873e-06,
-      "loss": 1.3395,
-      "step": 144
-    },
-    {
-      "epoch": 0.26,
-      "learning_rate": 6.855791962174941e-06,
-      "loss": 1.3151,
-      "step": 145
-    },
-    {
-      "epoch": 0.26,
-      "learning_rate": 6.90307328605201e-06,
-      "loss": 1.326,
-      "step": 146
-    },
-    {
-      "epoch": 0.26,
-      "learning_rate": 6.950354609929079e-06,
-      "loss": 1.2917,
-      "step": 147
-    },
-    {
-      "epoch": 0.26,
-      "learning_rate": 6.997635933806147e-06,
-      "loss": 1.3112,
-      "step": 148
-    },
-    {
-      "epoch": 0.26,
-      "learning_rate": 7.044917257683216e-06,
-      "loss": 1.2951,
-      "step": 149
-    },
-    {
-      "epoch": 0.27,
-      "learning_rate": 7.092198581560285e-06,
-      "loss": 1.3028,
-      "step": 150
-    },
-    {
-      "epoch": 0.27,
-      "learning_rate": 7.139479905437353e-06,
-      "loss": 1.2671,
-      "step": 151
-    },
-    {
-      "epoch": 0.27,
-      "learning_rate": 7.186761229314421e-06,
-      "loss": 1.2745,
-      "step": 152
-    },
-    {
-      "epoch": 0.27,
-      "learning_rate": 7.234042553191491e-06,
-      "loss": 1.2703,
-      "step": 153
-    },
-    {
-      "epoch": 0.27,
-      "learning_rate": 7.281323877068559e-06,
-      "loss": 1.2861,
-      "step": 154
-    },
-    {
-      "epoch": 0.28,
-      "learning_rate": 7.3286052009456266e-06,
-      "loss": 1.2308,
-      "step": 155
-    },
-    {
-      "epoch": 0.28,
-      "learning_rate": 7.375886524822695e-06,
-      "loss": 1.2557,
-      "step": 156
-    },
-    {
-      "epoch": 0.28,
-      "learning_rate": 7.423167848699764e-06,
-      "loss": 1.2295,
-      "step": 157
-    },
-    {
-      "epoch": 0.28,
-      "learning_rate": 7.4704491725768326e-06,
-      "loss": 1.2186,
-      "step": 158
-    },
-    {
-      "epoch": 0.28,
-      "learning_rate": 7.517730496453901e-06,
-      "loss": 1.2425,
-      "step": 159
-    },
-    {
-      "epoch": 0.28,
-      "learning_rate": 7.565011820330969e-06,
-      "loss": 1.2669,
-      "step": 160
-    },
-    {
-      "epoch": 0.29,
-      "learning_rate": 7.6122931442080386e-06,
-      "loss": 1.2357,
-      "step": 161
-    },
-    {
-      "epoch": 0.29,
-      "learning_rate": 7.659574468085107e-06,
-      "loss": 1.1846,
-      "step": 162
-    },
-    {
-      "epoch": 0.29,
-      "learning_rate": 7.706855791962176e-06,
-      "loss": 1.2153,
-      "step": 163
-    },
-    {
-      "epoch": 0.29,
-      "learning_rate": 7.754137115839244e-06,
-      "loss": 1.2404,
-      "step": 164
-    },
-    {
-      "epoch": 0.29,
-      "learning_rate": 7.801418439716313e-06,
-      "loss": 1.2312,
-      "step": 165
-    },
-    {
-      "epoch": 0.29,
-      "learning_rate": 7.848699763593382e-06,
-      "loss": 1.1853,
-      "step": 166
-    },
-    {
-      "epoch": 0.3,
-      "learning_rate": 7.89598108747045e-06,
-      "loss": 1.2032,
-      "step": 167
-    },
-    {
-      "epoch": 0.3,
-      "learning_rate": 7.943262411347519e-06,
-      "loss": 1.1866,
-      "step": 168
-    },
-    {
-      "epoch": 0.3,
-      "learning_rate": 7.990543735224588e-06,
-      "loss": 1.1753,
-      "step": 169
-    },
-    {
-      "epoch": 0.3,
-      "learning_rate": 8.037825059101656e-06,
-      "loss": 1.1363,
-      "step": 170
-    },
-    {
-      "epoch": 0.3,
-      "learning_rate": 8.085106382978723e-06,
-      "loss": 1.1782,
-      "step": 171
-    },
-    {
-      "epoch": 0.31,
-      "learning_rate": 8.132387706855792e-06,
-      "loss": 1.1623,
-      "step": 172
-    },
-    {
-      "epoch": 0.31,
-      "learning_rate": 8.17966903073286e-06,
-      "loss": 1.1751,
-      "step": 173
-    },
-    {
-      "epoch": 0.31,
-      "learning_rate": 8.22695035460993e-06,
-      "loss": 1.1478,
-      "step": 174
-    },
-    {
-      "epoch": 0.31,
-      "learning_rate": 8.274231678486998e-06,
-      "loss": 1.1418,
-      "step": 175
-    },
-    {
-      "epoch": 0.31,
-      "learning_rate": 8.321513002364066e-06,
-      "loss": 1.1827,
-      "step": 176
-    },
-    {
-      "epoch": 0.31,
-      "learning_rate": 8.368794326241135e-06,
-      "loss": 1.1514,
-      "step": 177
-    },
-    {
-      "epoch": 0.32,
-      "learning_rate": 8.416075650118204e-06,
-      "loss": 1.1389,
-      "step": 178
-    },
-    {
-      "epoch": 0.32,
-      "learning_rate": 8.463356973995272e-06,
-      "loss": 1.0831,
-      "step": 179
-    },
-    {
-      "epoch": 0.32,
-      "learning_rate": 8.510638297872341e-06,
-      "loss": 1.1216,
-      "step": 180
-    },
-    {
-      "epoch": 0.32,
-      "learning_rate": 8.55791962174941e-06,
-      "loss": 1.1188,
-      "step": 181
-    },
-    {
-      "epoch": 0.32,
-      "learning_rate": 8.605200945626478e-06,
-      "loss": 1.1417,
-      "step": 182
-    },
-    {
-      "epoch": 0.32,
-      "learning_rate": 8.652482269503547e-06,
-      "loss": 1.0844,
-      "step": 183
-    },
-    {
-      "epoch": 0.33,
-      "learning_rate": 8.699763593380616e-06,
-      "loss": 1.0685,
-      "step": 184
-    },
-    {
-      "epoch": 0.33,
-      "learning_rate": 8.747044917257684e-06,
-      "loss": 1.1316,
-      "step": 185
-    },
-    {
-      "epoch": 0.33,
-      "learning_rate": 8.794326241134753e-06,
-      "loss": 1.0882,
-      "step": 186
-    },
-    {
-      "epoch": 0.33,
-      "learning_rate": 8.84160756501182e-06,
-      "loss": 1.0891,
-      "step": 187
-    },
-    {
-      "epoch": 0.33,
-      "learning_rate": 8.888888888888888e-06,
-      "loss": 1.0951,
-      "step": 188
-    },
-    {
-      "epoch": 0.34,
-      "learning_rate": 8.936170212765958e-06,
-      "loss": 1.0857,
-      "step": 189
-    },
-    {
-      "epoch": 0.34,
-      "learning_rate": 8.983451536643027e-06,
-      "loss": 1.0693,
-      "step": 190
-    },
-    {
-      "epoch": 0.34,
-      "learning_rate": 9.030732860520094e-06,
-      "loss": 1.0469,
-      "step": 191
-    },
-    {
-      "epoch": 0.34,
-      "learning_rate": 9.078014184397164e-06,
-      "loss": 1.0561,
-      "step": 192
-    },
-    {
-      "epoch": 0.34,
-      "learning_rate": 9.125295508274233e-06,
-      "loss": 1.0898,
-      "step": 193
-    },
-    {
-      "epoch": 0.34,
-      "learning_rate": 9.1725768321513e-06,
-      "loss": 1.0712,
-      "step": 194
-    },
-    {
-      "epoch": 0.35,
-      "learning_rate": 9.21985815602837e-06,
-      "loss": 1.0386,
-      "step": 195
-    },
-    {
-      "epoch": 0.35,
-      "learning_rate": 9.267139479905439e-06,
-      "loss": 1.0142,
-      "step": 196
-    },
-    {
-      "epoch": 0.35,
-      "learning_rate": 9.314420803782506e-06,
-      "loss": 1.0559,
-      "step": 197
-    },
-    {
-      "epoch": 0.35,
-      "learning_rate": 9.361702127659576e-06,
-      "loss": 1.0253,
-      "step": 198
-    },
-    {
-      "epoch": 0.35,
-      "learning_rate": 9.408983451536645e-06,
-      "loss": 1.0349,
-      "step": 199
-    },
-    {
-      "epoch": 0.36,
-      "learning_rate": 9.456264775413712e-06,
-      "loss": 1.0346,
-      "step": 200
-    },
-    {
-      "epoch": 0.36,
-      "learning_rate": 9.503546099290782e-06,
-      "loss": 1.0597,
-      "step": 201
-    },
-    {
-      "epoch": 0.36,
-      "learning_rate": 9.55082742316785e-06,
-      "loss": 1.0072,
-      "step": 202
-    },
-    {
-      "epoch": 0.36,
-      "learning_rate": 9.598108747044918e-06,
-      "loss": 1.0393,
-      "step": 203
-    },
-    {
-      "epoch": 0.36,
-      "learning_rate": 9.645390070921986e-06,
-      "loss": 1.0246,
-      "step": 204
-    },
-    {
-      "epoch": 0.36,
-      "learning_rate": 9.692671394799055e-06,
-      "loss": 1.0064,
-      "step": 205
-    },
-    {
-      "epoch": 0.37,
-      "learning_rate": 9.739952718676123e-06,
-      "loss": 1.0485,
-      "step": 206
-    },
-    {
-      "epoch": 0.37,
-      "learning_rate": 9.787234042553192e-06,
-      "loss": 1.0353,
-      "step": 207
-    },
-    {
-      "epoch": 0.37,
-      "learning_rate": 9.834515366430261e-06,
-      "loss": 0.9979,
-      "step": 208
-    },
-    {
-      "epoch": 0.37,
-      "learning_rate": 9.881796690307329e-06,
-      "loss": 0.9859,
-      "step": 209
-    },
-    {
-      "epoch": 0.37,
-      "learning_rate": 9.929078014184398e-06,
-      "loss": 0.9585,
-      "step": 210
-    },
-    {
-      "epoch": 0.37,
-      "learning_rate": 9.976359338061467e-06,
-      "loss": 0.9838,
-      "step": 211
-    },
-    {
-      "epoch": 0.38,
-      "learning_rate": 1.0023640661938535e-05,
-      "loss": 0.9708,
-      "step": 212
-    },
-    {
-      "epoch": 0.38,
-      "learning_rate": 1.0070921985815602e-05,
-      "loss": 0.9753,
-      "step": 213
-    },
-    {
-      "epoch": 0.38,
-      "learning_rate": 1.0118203309692671e-05,
-      "loss": 0.9577,
-      "step": 214
-    },
-    {
-      "epoch": 0.38,
-      "learning_rate": 1.016548463356974e-05,
-      "loss": 0.9933,
-      "step": 215
-    },
-    {
-      "epoch": 0.38,
-      "learning_rate": 1.0212765957446808e-05,
-      "loss": 0.9679,
-      "step": 216
-    },
-    {
-      "epoch": 0.39,
-      "learning_rate": 1.0260047281323877e-05,
-      "loss": 0.9511,
-      "step": 217
-    },
-    {
-      "epoch": 0.39,
-      "learning_rate": 1.0307328605200947e-05,
-      "loss": 0.9708,
-      "step": 218
-    },
-    {
-      "epoch": 0.39,
-      "learning_rate": 1.0354609929078014e-05,
-      "loss": 0.966,
-      "step": 219
-    },
-    {
-      "epoch": 0.39,
-      "learning_rate": 1.0401891252955083e-05,
-      "loss": 0.9696,
-      "step": 220
-    },
-    {
-      "epoch": 0.39,
-      "learning_rate": 1.0449172576832151e-05,
-      "loss": 0.9158,
-      "step": 221
-    },
-    {
-      "epoch": 0.39,
-      "learning_rate": 1.049645390070922e-05,
-      "loss": 0.9576,
-      "step": 222
-    },
-    {
-      "epoch": 0.4,
-      "learning_rate": 1.054373522458629e-05,
-      "loss": 0.9127,
-      "step": 223
-    },
-    {
-      "epoch": 0.4,
-      "learning_rate": 1.0591016548463357e-05,
-      "loss": 0.9716,
-      "step": 224
-    },
-    {
-      "epoch": 0.4,
-      "learning_rate": 1.0638297872340426e-05,
-      "loss": 0.9133,
-      "step": 225
-    },
-    {
-      "epoch": 0.4,
-      "learning_rate": 1.0685579196217495e-05,
-      "loss": 0.9123,
-      "step": 226
-    },
-    {
-      "epoch": 0.4,
-      "learning_rate": 1.0732860520094563e-05,
-      "loss": 0.934,
-      "step": 227
-    },
-    {
-      "epoch": 0.4,
-      "learning_rate": 1.0780141843971632e-05,
-      "loss": 0.9427,
-      "step": 228
-    },
-    {
-      "epoch": 0.41,
-      "learning_rate": 1.0827423167848701e-05,
-      "loss": 0.9192,
-      "step": 229
-    },
-    {
-      "epoch": 0.41,
-      "learning_rate": 1.0874704491725769e-05,
-      "loss": 0.8922,
-      "step": 230
-    },
-    {
-      "epoch": 0.41,
-      "learning_rate": 1.0921985815602838e-05,
-      "loss": 0.9188,
-      "step": 231
-    },
-    {
-      "epoch": 0.41,
-      "learning_rate": 1.0969267139479907e-05,
-      "loss": 0.9435,
-      "step": 232
-    },
-    {
-      "epoch": 0.41,
-      "learning_rate": 1.1016548463356975e-05,
-      "loss": 0.8918,
-      "step": 233
-    },
-    {
-      "epoch": 0.42,
-      "learning_rate": 1.1063829787234044e-05,
-      "loss": 0.9229,
-      "step": 234
-    },
-    {
-      "epoch": 0.42,
-      "learning_rate": 1.1111111111111113e-05,
-      "loss": 0.9139,
-      "step": 235
-    },
-    {
-      "epoch": 0.42,
-      "learning_rate": 1.1158392434988181e-05,
-      "loss": 0.9064,
-      "step": 236
-    },
-    {
-      "epoch": 0.42,
-      "learning_rate": 1.120567375886525e-05,
-      "loss": 0.8945,
-      "step": 237
-    },
-    {
-      "epoch": 0.42,
-      "learning_rate": 1.1252955082742318e-05,
-      "loss": 0.9115,
-      "step": 238
-    },
-    {
-      "epoch": 0.42,
-      "learning_rate": 1.1300236406619387e-05,
-      "loss": 0.9127,
-      "step": 239
-    },
-    {
-      "epoch": 0.43,
-      "learning_rate": 1.1347517730496456e-05,
-      "loss": 0.8804,
-      "step": 240
-    },
-    {
-      "epoch": 0.43,
-      "learning_rate": 1.1394799054373524e-05,
-      "loss": 0.9221,
-      "step": 241
-    },
-    {
-      "epoch": 0.43,
-      "learning_rate": 1.1442080378250593e-05,
-      "loss": 0.8999,
-      "step": 242
-    },
-    {
-      "epoch": 0.43,
-      "learning_rate": 1.1489361702127662e-05,
-      "loss": 0.9172,
-      "step": 243
-    },
-    {
-      "epoch": 0.43,
-      "learning_rate": 1.153664302600473e-05,
-      "loss": 0.8696,
-      "step": 244
-    },
-    {
-      "epoch": 0.43,
-      "learning_rate": 1.1583924349881797e-05,
-      "loss": 0.8919,
-      "step": 245
-    },
-    {
-      "epoch": 0.44,
-      "learning_rate": 1.1631205673758865e-05,
-      "loss": 0.8784,
-      "step": 246
-    },
-    {
-      "epoch": 0.44,
-      "learning_rate": 1.1678486997635934e-05,
-      "loss": 0.8869,
-      "step": 247
-    },
-    {
-      "epoch": 0.44,
-      "learning_rate": 1.1725768321513003e-05,
-      "loss": 0.913,
-      "step": 248
-    },
-    {
-      "epoch": 0.44,
-      "learning_rate": 1.177304964539007e-05,
-      "loss": 0.8671,
-      "step": 249
-    },
-    {
-      "epoch": 0.44,
-      "learning_rate": 1.182033096926714e-05,
-      "loss": 0.8967,
-      "step": 250
-    },
-    {
-      "epoch": 0.45,
-      "learning_rate": 1.186761229314421e-05,
-      "loss": 0.8543,
-      "step": 251
-    },
-    {
-      "epoch": 0.45,
-      "learning_rate": 1.1914893617021277e-05,
-      "loss": 0.868,
-      "step": 252
-    },
-    {
-      "epoch": 0.45,
-      "learning_rate": 1.1962174940898346e-05,
-      "loss": 0.8705,
-      "step": 253
-    },
-    {
-      "epoch": 0.45,
-      "learning_rate": 1.2009456264775413e-05,
-      "loss": 0.8738,
-      "step": 254
-    },
-    {
-      "epoch": 0.45,
-      "learning_rate": 1.2056737588652483e-05,
-      "loss": 0.8323,
-      "step": 255
-    },
-    {
-      "epoch": 0.45,
-      "learning_rate": 1.2104018912529552e-05,
-      "loss": 0.8502,
-      "step": 256
-    },
-    {
-      "epoch": 0.46,
-      "learning_rate": 1.215130023640662e-05,
-      "loss": 0.8402,
-      "step": 257
-    },
-    {
-      "epoch": 0.46,
-      "learning_rate": 1.2198581560283689e-05,
-      "loss": 0.8314,
-      "step": 258
-    },
-    {
-      "epoch": 0.46,
-      "learning_rate": 1.2245862884160758e-05,
-      "loss": 0.8295,
-      "step": 259
-    },
-    {
-      "epoch": 0.46,
-      "learning_rate": 1.2293144208037825e-05,
-      "loss": 0.8335,
-      "step": 260
-    },
-    {
-      "epoch": 0.46,
-      "learning_rate": 1.2340425531914895e-05,
-      "loss": 0.8218,
-      "step": 261
-    },
-    {
-      "epoch": 0.47,
-      "learning_rate": 1.2387706855791964e-05,
-      "loss": 0.8781,
-      "step": 262
-    },
-    {
-      "epoch": 0.47,
-      "learning_rate": 1.2434988179669031e-05,
-      "loss": 0.8645,
-      "step": 263
-    },
-    {
-      "epoch": 0.47,
-      "learning_rate": 1.24822695035461e-05,
-      "loss": 0.8579,
-      "step": 264
-    },
-    {
-      "epoch": 0.47,
-      "learning_rate": 1.252955082742317e-05,
-      "loss": 0.8684,
-      "step": 265
-    },
-    {
-      "epoch": 0.47,
-      "learning_rate": 1.2576832151300237e-05,
-      "loss": 0.8202,
-      "step": 266
-    },
-    {
-      "epoch": 0.47,
-      "learning_rate": 1.2624113475177307e-05,
-      "loss": 0.8679,
-      "step": 267
-    },
-    {
-      "epoch": 0.48,
-      "learning_rate": 1.2671394799054376e-05,
-      "loss": 0.8203,
-      "step": 268
-    },
-    {
-      "epoch": 0.48,
-      "learning_rate": 1.2718676122931443e-05,
-      "loss": 0.8233,
-      "step": 269
-    },
-    {
-      "epoch": 0.48,
-      "learning_rate": 1.2765957446808513e-05,
-      "loss": 0.8536,
-      "step": 270
-    },
-    {
-      "epoch": 0.48,
-      "learning_rate": 1.281323877068558e-05,
-      "loss": 0.8138,
-      "step": 271
-    },
-    {
-      "epoch": 0.48,
-      "learning_rate": 1.286052009456265e-05,
-      "loss": 0.8269,
-      "step": 272
-    },
-    {
-      "epoch": 0.48,
-      "learning_rate": 1.2907801418439719e-05,
-      "loss": 0.8068,
-      "step": 273
-    },
-    {
-      "epoch": 0.49,
-      "learning_rate": 1.2955082742316786e-05,
-      "loss": 0.8068,
-      "step": 274
-    },
-    {
-      "epoch": 0.49,
-      "learning_rate": 1.3002364066193855e-05,
-      "loss": 0.8055,
-      "step": 275
-    },
-    {
-      "epoch": 0.49,
-      "learning_rate": 1.3049645390070925e-05,
-      "loss": 0.8057,
-      "step": 276
-    },
-    {
-      "epoch": 0.49,
-      "learning_rate": 1.309692671394799e-05,
-      "loss": 0.8159,
-      "step": 277
-    },
-    {
-      "epoch": 0.49,
-      "learning_rate": 1.314420803782506e-05,
-      "loss": 0.8331,
-      "step": 278
-    },
-    {
-      "epoch": 0.5,
-      "learning_rate": 1.3191489361702127e-05,
-      "loss": 0.8376,
-      "step": 279
-    },
-    {
-      "epoch": 0.5,
-      "learning_rate": 1.3238770685579197e-05,
-      "loss": 0.8314,
-      "step": 280
-    },
-    {
-      "epoch": 0.5,
-      "learning_rate": 1.3286052009456266e-05,
-      "loss": 0.8254,
-      "step": 281
-    },
-    {
-      "epoch": 0.5,
-      "learning_rate": 1.3333333333333333e-05,
-      "loss": 0.8307,
-      "step": 282
-    },
-    {
-      "epoch": 0.5,
-      "learning_rate": 1.3380614657210403e-05,
-      "loss": 0.7884,
-      "step": 283
-    },
-    {
-      "epoch": 0.5,
-      "learning_rate": 1.3427895981087472e-05,
-      "loss": 0.8049,
-      "step": 284
-    },
-    {
-      "epoch": 0.51,
-      "learning_rate": 1.347517730496454e-05,
-      "loss": 0.8204,
-      "step": 285
-    },
-    {
-      "epoch": 0.51,
-      "learning_rate": 1.3522458628841609e-05,
-      "loss": 0.8123,
-      "step": 286
-    },
-    {
-      "epoch": 0.51,
-      "learning_rate": 1.3569739952718676e-05,
-      "loss": 0.8158,
-      "step": 287
-    },
-    {
-      "epoch": 0.51,
-      "learning_rate": 1.3617021276595745e-05,
-      "loss": 0.7879,
-      "step": 288
-    },
-    {
-      "epoch": 0.51,
-      "learning_rate": 1.3664302600472815e-05,
-      "loss": 0.8056,
-      "step": 289
-    },
-    {
-      "epoch": 0.51,
-      "learning_rate": 1.3711583924349882e-05,
-      "loss": 0.7756,
-      "step": 290
-    },
-    {
-      "epoch": 0.52,
-      "learning_rate": 1.3758865248226951e-05,
-      "loss": 0.7695,
-      "step": 291
-    },
-    {
-      "epoch": 0.52,
-      "learning_rate": 1.380614657210402e-05,
-      "loss": 0.8254,
-      "step": 292
-    },
-    {
-      "epoch": 0.52,
-      "learning_rate": 1.3853427895981088e-05,
-      "loss": 0.769,
-      "step": 293
-    },
-    {
-      "epoch": 0.52,
-      "learning_rate": 1.3900709219858157e-05,
-      "loss": 0.7873,
-      "step": 294
-    },
-    {
-      "epoch": 0.52,
-      "learning_rate": 1.3947990543735227e-05,
-      "loss": 0.7799,
-      "step": 295
-    },
-    {
-      "epoch": 0.53,
-      "learning_rate": 1.3995271867612294e-05,
-      "loss": 0.8157,
-      "step": 296
-    },
-    {
-      "epoch": 0.53,
-      "learning_rate": 1.4042553191489363e-05,
-      "loss": 0.7977,
-      "step": 297
-    },
-    {
-      "epoch": 0.53,
-      "learning_rate": 1.4089834515366433e-05,
-      "loss": 0.8075,
-      "step": 298
-    },
-    {
-      "epoch": 0.53,
-      "learning_rate": 1.41371158392435e-05,
-      "loss": 0.7364,
-      "step": 299
-    },
-    {
-      "epoch": 0.53,
-      "learning_rate": 1.418439716312057e-05,
-      "loss": 0.7782,
-      "step": 300
-    },
-    {
-      "epoch": 0.53,
-      "learning_rate": 1.4231678486997637e-05,
-      "loss": 0.7639,
-      "step": 301
-    },
-    {
-      "epoch": 0.54,
-      "learning_rate": 1.4278959810874706e-05,
-      "loss": 0.7389,
-      "step": 302
-    },
-    {
-      "epoch": 0.54,
-      "learning_rate": 1.4326241134751775e-05,
-      "loss": 0.7675,
-      "step": 303
-    },
-    {
-      "epoch": 0.54,
-      "learning_rate": 1.4373522458628843e-05,
-      "loss": 0.7576,
-      "step": 304
-    },
-    {
-      "epoch": 0.54,
-      "learning_rate": 1.4420803782505912e-05,
-      "loss": 0.7744,
-      "step": 305
-    },
-    {
-      "epoch": 0.54,
-      "learning_rate": 1.4468085106382981e-05,
-      "loss": 0.741,
-      "step": 306
-    },
-    {
-      "epoch": 0.55,
-      "learning_rate": 1.4515366430260049e-05,
-      "loss": 0.7645,
-      "step": 307
-    },
-    {
-      "epoch": 0.55,
-      "learning_rate": 1.4562647754137118e-05,
-      "loss": 0.757,
-      "step": 308
-    },
-    {
-      "epoch": 0.55,
-      "learning_rate": 1.4609929078014187e-05,
-      "loss": 0.7619,
-      "step": 309
-    },
-    {
-      "epoch": 0.55,
-      "learning_rate": 1.4657210401891253e-05,
-      "loss": 0.7905,
-      "step": 310
-    },
-    {
-      "epoch": 0.55,
-      "learning_rate": 1.4704491725768322e-05,
-      "loss": 0.7784,
-      "step": 311
-    },
-    {
-      "epoch": 0.55,
-      "learning_rate": 1.475177304964539e-05,
-      "loss": 0.7665,
-      "step": 312
-    },
-    {
-      "epoch": 0.56,
-      "learning_rate": 1.4799054373522459e-05,
-      "loss": 0.7598,
-      "step": 313
-    },
-    {
-      "epoch": 0.56,
-      "learning_rate": 1.4846335697399528e-05,
-      "loss": 0.755,
-      "step": 314
-    },
-    {
-      "epoch": 0.56,
-      "learning_rate": 1.4893617021276596e-05,
-      "loss": 0.7687,
-      "step": 315
-    },
-    {
-      "epoch": 0.56,
-      "learning_rate": 1.4940898345153665e-05,
-      "loss": 0.7691,
-      "step": 316
-    },
-    {
-      "epoch": 0.56,
-      "learning_rate": 1.4988179669030734e-05,
-      "loss": 0.7386,
-      "step": 317
-    },
-    {
-      "epoch": 0.56,
-      "learning_rate": 1.5035460992907802e-05,
-      "loss": 0.7706,
-      "step": 318
-    },
-    {
-      "epoch": 0.57,
-      "learning_rate": 1.5082742316784871e-05,
-      "loss": 0.7762,
-      "step": 319
-    },
-    {
-      "epoch": 0.57,
-      "learning_rate": 1.5130023640661939e-05,
-      "loss": 0.7095,
-      "step": 320
-    },
-    {
-      "epoch": 0.57,
-      "learning_rate": 1.5177304964539008e-05,
-      "loss": 0.7664,
-      "step": 321
-    },
-    {
-      "epoch": 0.57,
-      "learning_rate": 1.5224586288416077e-05,
-      "loss": 0.7227,
-      "step": 322
-    },
-    {
-      "epoch": 0.57,
-      "learning_rate": 1.5271867612293146e-05,
-      "loss": 0.7245,
-      "step": 323
-    },
-    {
-      "epoch": 0.58,
-      "learning_rate": 1.5319148936170214e-05,
-      "loss": 0.7479,
-      "step": 324
-    },
-    {
-      "epoch": 0.58,
-      "learning_rate": 1.536643026004728e-05,
-      "loss": 0.7473,
-      "step": 325
-    },
-    {
-      "epoch": 0.58,
-      "learning_rate": 1.5413711583924352e-05,
-      "loss": 0.7424,
-      "step": 326
-    },
-    {
-      "epoch": 0.58,
-      "learning_rate": 1.546099290780142e-05,
-      "loss": 0.7338,
-      "step": 327
-    },
-    {
-      "epoch": 0.58,
-      "learning_rate": 1.5508274231678487e-05,
-      "loss": 0.7282,
-      "step": 328
-    },
-    {
-      "epoch": 0.58,
-      "learning_rate": 1.555555555555556e-05,
-      "loss": 0.7161,
-      "step": 329
-    },
-    {
-      "epoch": 0.59,
-      "learning_rate": 1.5602836879432626e-05,
-      "loss": 0.755,
-      "step": 330
-    },
-    {
-      "epoch": 0.59,
-      "learning_rate": 1.5650118203309693e-05,
-      "loss": 0.746,
-      "step": 331
-    },
-    {
-      "epoch": 0.59,
-      "learning_rate": 1.5697399527186764e-05,
-      "loss": 0.7293,
-      "step": 332
-    },
-    {
-      "epoch": 0.59,
-      "learning_rate": 1.5744680851063832e-05,
-      "loss": 0.7189,
-      "step": 333
-    },
-    {
-      "epoch": 0.59,
-      "learning_rate": 1.57919621749409e-05,
-      "loss": 0.734,
-      "step": 334
-    },
-    {
-      "epoch": 0.59,
-      "learning_rate": 1.583924349881797e-05,
-      "loss": 0.7286,
-      "step": 335
-    },
-    {
-      "epoch": 0.6,
-      "learning_rate": 1.5886524822695038e-05,
-      "loss": 0.7264,
-      "step": 336
-    },
-    {
-      "epoch": 0.6,
-      "learning_rate": 1.5933806146572105e-05,
-      "loss": 0.7538,
-      "step": 337
-    },
-    {
-      "epoch": 0.6,
-      "learning_rate": 1.5981087470449176e-05,
-      "loss": 0.7266,
-      "step": 338
-    },
-    {
-      "epoch": 0.6,
-      "learning_rate": 1.6028368794326244e-05,
-      "loss": 0.7031,
-      "step": 339
-    },
-    {
-      "epoch": 0.6,
-      "learning_rate": 1.607565011820331e-05,
-      "loss": 0.722,
-      "step": 340
-    },
-    {
-      "epoch": 0.61,
-      "learning_rate": 1.6122931442080382e-05,
-      "loss": 0.702,
-      "step": 341
-    },
-    {
-      "epoch": 0.61,
-      "learning_rate": 1.6170212765957446e-05,
-      "loss": 0.6998,
-      "step": 342
-    },
-    {
-      "epoch": 0.61,
-      "learning_rate": 1.6217494089834514e-05,
-      "loss": 0.7173,
-      "step": 343
-    },
-    {
-      "epoch": 0.61,
-      "learning_rate": 1.6264775413711585e-05,
-      "loss": 0.7194,
-      "step": 344
-    },
-    {
-      "epoch": 0.61,
-      "learning_rate": 1.6312056737588652e-05,
-      "loss": 0.7231,
-      "step": 345
-    },
-    {
-      "epoch": 0.61,
-      "learning_rate": 1.635933806146572e-05,
-      "loss": 0.6913,
-      "step": 346
-    },
-    {
-      "epoch": 0.62,
-      "learning_rate": 1.640661938534279e-05,
-      "loss": 0.7171,
-      "step": 347
-    },
-    {
-      "epoch": 0.62,
-      "learning_rate": 1.645390070921986e-05,
-      "loss": 0.7188,
-      "step": 348
-    },
-    {
-      "epoch": 0.62,
-      "learning_rate": 1.6501182033096926e-05,
-      "loss": 0.7125,
-      "step": 349
-    },
-    {
-      "epoch": 0.62,
-      "learning_rate": 1.6548463356973997e-05,
-      "loss": 0.7211,
-      "step": 350
-    },
-    {
-      "epoch": 0.62,
-      "learning_rate": 1.6595744680851064e-05,
-      "loss": 0.7019,
-      "step": 351
-    },
-    {
-      "epoch": 0.62,
-      "learning_rate": 1.6643026004728132e-05,
-      "loss": 0.6957,
-      "step": 352
-    },
-    {
-      "epoch": 0.63,
-      "learning_rate": 1.6690307328605203e-05,
-      "loss": 0.7074,
-      "step": 353
-    },
-    {
-      "epoch": 0.63,
-      "learning_rate": 1.673758865248227e-05,
-      "loss": 0.6837,
-      "step": 354
-    },
-    {
-      "epoch": 0.63,
-      "learning_rate": 1.6784869976359338e-05,
-      "loss": 0.7022,
-      "step": 355
-    },
-    {
-      "epoch": 0.63,
-      "learning_rate": 1.683215130023641e-05,
-      "loss": 0.6923,
-      "step": 356
-    },
-    {
-      "epoch": 0.63,
-      "learning_rate": 1.6879432624113476e-05,
-      "loss": 0.6791,
-      "step": 357
-    },
-    {
-      "epoch": 0.64,
-      "learning_rate": 1.6926713947990544e-05,
-      "loss": 0.7162,
-      "step": 358
-    },
-    {
-      "epoch": 0.64,
-      "learning_rate": 1.6973995271867615e-05,
-      "loss": 0.713,
-      "step": 359
-    },
-    {
-      "epoch": 0.64,
-      "learning_rate": 1.7021276595744682e-05,
-      "loss": 0.713,
-      "step": 360
-    },
-    {
-      "epoch": 0.64,
-      "learning_rate": 1.706855791962175e-05,
-      "loss": 0.6995,
-      "step": 361
-    },
-    {
-      "epoch": 0.64,
-      "learning_rate": 1.711583924349882e-05,
-      "loss": 0.7018,
-      "step": 362
-    },
-    {
-      "epoch": 0.64,
-      "learning_rate": 1.716312056737589e-05,
-      "loss": 0.6833,
-      "step": 363
-    },
-    {
-      "epoch": 0.65,
-      "learning_rate": 1.7210401891252956e-05,
-      "loss": 0.6949,
-      "step": 364
-    },
-    {
-      "epoch": 0.65,
-      "learning_rate": 1.7257683215130027e-05,
-      "loss": 0.7002,
-      "step": 365
-    },
-    {
-      "epoch": 0.65,
-      "learning_rate": 1.7304964539007094e-05,
-      "loss": 0.6981,
-      "step": 366
-    },
-    {
-      "epoch": 0.65,
-      "learning_rate": 1.7352245862884162e-05,
-      "loss": 0.6695,
-      "step": 367
-    },
-    {
-      "epoch": 0.65,
-      "learning_rate": 1.7399527186761233e-05,
-      "loss": 0.6819,
-      "step": 368
-    },
-    {
-      "epoch": 0.66,
-      "learning_rate": 1.74468085106383e-05,
-      "loss": 0.6982,
-      "step": 369
-    },
-    {
-      "epoch": 0.66,
-      "learning_rate": 1.7494089834515368e-05,
-      "loss": 0.6822,
-      "step": 370
-    },
-    {
-      "epoch": 0.66,
-      "learning_rate": 1.754137115839244e-05,
-      "loss": 0.6708,
-      "step": 371
-    },
-    {
-      "epoch": 0.66,
-      "learning_rate": 1.7588652482269506e-05,
-      "loss": 0.6927,
-      "step": 372
-    },
-    {
-      "epoch": 0.66,
-      "learning_rate": 1.7635933806146574e-05,
-      "loss": 0.6801,
-      "step": 373
-    },
-    {
-      "epoch": 0.66,
-      "learning_rate": 1.768321513002364e-05,
-      "loss": 0.6953,
-      "step": 374
-    },
-    {
-      "epoch": 0.67,
-      "learning_rate": 1.773049645390071e-05,
-      "loss": 0.6935,
-      "step": 375
-    },
-    {
-      "epoch": 0.67,
-      "learning_rate": 1.7777777777777777e-05,
-      "loss": 0.6826,
-      "step": 376
-    },
-    {
-      "epoch": 0.67,
-      "learning_rate": 1.7825059101654848e-05,
-      "loss": 0.6681,
-      "step": 377
-    },
-    {
-      "epoch": 0.67,
-      "learning_rate": 1.7872340425531915e-05,
-      "loss": 0.6733,
-      "step": 378
-    },
-    {
-      "epoch": 0.67,
-      "learning_rate": 1.7919621749408983e-05,
-      "loss": 0.6745,
-      "step": 379
-    },
-    {
-      "epoch": 0.67,
-      "learning_rate": 1.7966903073286054e-05,
-      "loss": 0.6666,
-      "step": 380
-    },
-    {
-      "epoch": 0.68,
-      "learning_rate": 1.801418439716312e-05,
-      "loss": 0.6772,
-      "step": 381
-    },
-    {
-      "epoch": 0.68,
-      "learning_rate": 1.806146572104019e-05,
-      "loss": 0.6761,
-      "step": 382
-    },
-    {
-      "epoch": 0.68,
-      "learning_rate": 1.810874704491726e-05,
-      "loss": 0.6506,
-      "step": 383
-    },
-    {
-      "epoch": 0.68,
-      "learning_rate": 1.8156028368794327e-05,
-      "loss": 0.6872,
-      "step": 384
-    },
-    {
-      "epoch": 0.68,
-      "learning_rate": 1.8203309692671395e-05,
-      "loss": 0.686,
-      "step": 385
-    },
-    {
-      "epoch": 0.69,
-      "learning_rate": 1.8250591016548466e-05,
-      "loss": 0.6717,
-      "step": 386
-    },
-    {
-      "epoch": 0.69,
-      "learning_rate": 1.8297872340425533e-05,
-      "loss": 0.6828,
-      "step": 387
-    },
-    {
-      "epoch": 0.69,
-      "learning_rate": 1.83451536643026e-05,
-      "loss": 0.6579,
-      "step": 388
-    },
-    {
-      "epoch": 0.69,
-      "learning_rate": 1.839243498817967e-05,
-      "loss": 0.6817,
-      "step": 389
-    },
-    {
-      "epoch": 0.69,
-      "learning_rate": 1.843971631205674e-05,
-      "loss": 0.686,
-      "step": 390
-    },
-    {
-      "epoch": 0.69,
-      "learning_rate": 1.8486997635933807e-05,
-      "loss": 0.6688,
-      "step": 391
-    },
-    {
-      "epoch": 0.7,
-      "learning_rate": 1.8534278959810878e-05,
-      "loss": 0.6892,
-      "step": 392
-    },
-    {
-      "epoch": 0.7,
-      "learning_rate": 1.8581560283687945e-05,
-      "loss": 0.6642,
-      "step": 393
-    },
-    {
-      "epoch": 0.7,
-      "learning_rate": 1.8628841607565013e-05,
-      "loss": 0.6225,
-      "step": 394
-    },
-    {
-      "epoch": 0.7,
-      "learning_rate": 1.8676122931442084e-05,
-      "loss": 0.6613,
-      "step": 395
-    },
-    {
-      "epoch": 0.7,
-      "learning_rate": 1.872340425531915e-05,
-      "loss": 0.6599,
-      "step": 396
-    },
-    {
-      "epoch": 0.7,
-      "learning_rate": 1.877068557919622e-05,
-      "loss": 0.6575,
-      "step": 397
-    },
-    {
-      "epoch": 0.71,
-      "learning_rate": 1.881796690307329e-05,
-      "loss": 0.6252,
-      "step": 398
-    },
-    {
-      "epoch": 0.71,
-      "learning_rate": 1.8865248226950357e-05,
-      "loss": 0.612,
-      "step": 399
-    },
-    {
-      "epoch": 0.71,
-      "learning_rate": 1.8912529550827425e-05,
-      "loss": 0.6328,
-      "step": 400
-    },
-    {
-      "epoch": 0.71,
-      "learning_rate": 1.8959810874704496e-05,
-      "loss": 0.6708,
-      "step": 401
-    },
-    {
-      "epoch": 0.71,
-      "learning_rate": 1.9007092198581563e-05,
-      "loss": 0.6587,
-      "step": 402
-    },
-    {
-      "epoch": 0.72,
-      "learning_rate": 1.905437352245863e-05,
-      "loss": 0.6649,
-      "step": 403
-    },
-    {
-      "epoch": 0.72,
-      "learning_rate": 1.91016548463357e-05,
-      "loss": 0.6589,
-      "step": 404
-    },
-    {
-      "epoch": 0.72,
-      "learning_rate": 1.914893617021277e-05,
-      "loss": 0.6507,
-      "step": 405
-    },
-    {
-      "epoch": 0.72,
-      "learning_rate": 1.9196217494089837e-05,
-      "loss": 0.6523,
-      "step": 406
-    },
-    {
-      "epoch": 0.72,
-      "learning_rate": 1.9243498817966904e-05,
-      "loss": 0.6641,
-      "step": 407
-    },
-    {
-      "epoch": 0.72,
-      "learning_rate": 1.929078014184397e-05,
-      "loss": 0.631,
-      "step": 408
-    },
-    {
-      "epoch": 0.73,
-      "learning_rate": 1.933806146572104e-05,
-      "loss": 0.6603,
-      "step": 409
-    },
-    {
-      "epoch": 0.73,
-      "learning_rate": 1.938534278959811e-05,
-      "loss": 0.6254,
-      "step": 410
-    },
-    {
-      "epoch": 0.73,
-      "learning_rate": 1.9432624113475178e-05,
-      "loss": 0.6369,
-      "step": 411
-    },
-    {
-      "epoch": 0.73,
-      "learning_rate": 1.9479905437352245e-05,
-      "loss": 0.6332,
-      "step": 412
-    },
-    {
-      "epoch": 0.73,
-      "learning_rate": 1.9527186761229316e-05,
-      "loss": 0.6278,
-      "step": 413
-    },
-    {
-      "epoch": 0.74,
-      "learning_rate": 1.9574468085106384e-05,
-      "loss": 0.6357,
-      "step": 414
-    },
-    {
-      "epoch": 0.74,
-      "learning_rate": 1.962174940898345e-05,
-      "loss": 0.6294,
-      "step": 415
-    },
-    {
-      "epoch": 0.74,
-      "learning_rate": 1.9669030732860522e-05,
-      "loss": 0.6708,
-      "step": 416
-    },
-    {
-      "epoch": 0.74,
-      "learning_rate": 1.971631205673759e-05,
-      "loss": 0.6455,
-      "step": 417
-    },
-    {
-      "epoch": 0.74,
-      "learning_rate": 1.9763593380614657e-05,
-      "loss": 0.6442,
-      "step": 418
-    },
-    {
-      "epoch": 0.74,
-      "learning_rate": 1.9810874704491728e-05,
-      "loss": 0.6219,
-      "step": 419
-    },
-    {
-      "epoch": 0.75,
-      "learning_rate": 1.9858156028368796e-05,
-      "loss": 0.6351,
-      "step": 420
-    },
-    {
-      "epoch": 0.75,
-      "learning_rate": 1.9905437352245863e-05,
-      "loss": 0.6068,
-      "step": 421
-    },
-    {
-      "epoch": 0.75,
-      "learning_rate": 1.9952718676122934e-05,
-      "loss": 0.6219,
-      "step": 422
-    },
-    {
-      "epoch": 0.75,
-      "learning_rate": 2e-05,
-      "loss": 0.6292,
-      "step": 423
-    },
-    {
-      "epoch": 0.75,
-      "learning_rate": 1.9999999233160583e-05,
-      "loss": 0.6364,
-      "step": 424
-    },
-    {
-      "epoch": 0.75,
-      "learning_rate": 1.9999996932642446e-05,
-      "loss": 0.6329,
-      "step": 425
-    },
-    {
-      "epoch": 0.76,
-      "learning_rate": 1.9999993098445946e-05,
-      "loss": 0.6357,
-      "step": 426
-    },
-    {
-      "epoch": 0.76,
-      "learning_rate": 1.9999987730571666e-05,
-      "loss": 0.6171,
-      "step": 427
-    },
-    {
-      "epoch": 0.76,
-      "learning_rate": 1.9999980829020433e-05,
-      "loss": 0.6357,
-      "step": 428
-    },
-    {
-      "epoch": 0.76,
-      "learning_rate": 1.9999972393793303e-05,
-      "loss": 0.6289,
-      "step": 429
-    },
-    {
-      "epoch": 0.76,
-      "learning_rate": 1.999996242489157e-05,
-      "loss": 0.6188,
-      "step": 430
-    },
-    {
-      "epoch": 0.77,
-      "learning_rate": 1.999995092231677e-05,
-      "loss": 0.6264,
-      "step": 431
-    },
-    {
-      "epoch": 0.77,
-      "learning_rate": 1.9999937886070655e-05,
-      "loss": 0.6227,
-      "step": 432
-    },
-    {
-      "epoch": 0.77,
-      "learning_rate": 1.9999923316155235e-05,
-      "loss": 0.6106,
-      "step": 433
-    },
-    {
-      "epoch": 0.77,
-      "learning_rate": 1.9999907212572734e-05,
-      "loss": 0.6301,
-      "step": 434
-    },
-    {
-      "epoch": 0.77,
-      "learning_rate": 1.9999889575325633e-05,
-      "loss": 0.6232,
-      "step": 435
-    },
-    {
-      "epoch": 0.77,
-      "learning_rate": 1.999987040441663e-05,
-      "loss": 0.6217,
-      "step": 436
-    },
-    {
-      "epoch": 0.78,
-      "learning_rate": 1.9999849699848662e-05,
-      "loss": 0.6499,
-      "step": 437
-    },
-    {
-      "epoch": 0.78,
-      "learning_rate": 1.9999827461624918e-05,
-      "loss": 0.6178,
-      "step": 438
-    },
-    {
-      "epoch": 0.78,
-      "learning_rate": 1.9999803689748795e-05,
-      "loss": 0.6222,
-      "step": 439
-    },
-    {
-      "epoch": 0.78,
-      "learning_rate": 1.999977838422394e-05,
-      "loss": 0.6256,
-      "step": 440
-    },
-    {
-      "epoch": 0.78,
-      "learning_rate": 1.9999751545054247e-05,
-      "loss": 0.6478,
-      "step": 441
-    },
-    {
-      "epoch": 0.78,
-      "learning_rate": 1.999972317224382e-05,
-      "loss": 0.6145,
-      "step": 442
-    },
-    {
-      "epoch": 0.79,
-      "learning_rate": 1.999969326579701e-05,
-      "loss": 0.6076,
-      "step": 443
-    },
-    {
-      "epoch": 0.79,
-      "learning_rate": 1.9999661825718408e-05,
-      "loss": 0.6054,
-      "step": 444
-    },
-    {
-      "epoch": 0.79,
-      "learning_rate": 1.9999628852012834e-05,
-      "loss": 0.6584,
-      "step": 445
-    },
-    {
-      "epoch": 0.79,
-      "learning_rate": 1.999959434468535e-05,
-      "loss": 0.629,
-      "step": 446
-    },
-    {
-      "epoch": 0.79,
-      "learning_rate": 1.9999558303741244e-05,
-      "loss": 0.6506,
-      "step": 447
-    },
-    {
-      "epoch": 0.8,
-      "learning_rate": 1.999952072918604e-05,
-      "loss": 0.6394,
-      "step": 448
-    },
-    {
-      "epoch": 0.8,
-      "learning_rate": 1.999948162102551e-05,
-      "loss": 0.6483,
-      "step": 449
-    },
-    {
-      "epoch": 0.8,
-      "learning_rate": 1.9999440979265646e-05,
-      "loss": 0.6211,
-      "step": 450
-    },
-    {
-      "epoch": 0.8,
-      "learning_rate": 1.999939880391268e-05,
-      "loss": 0.6207,
-      "step": 451
-    },
-    {
-      "epoch": 0.8,
-      "learning_rate": 1.9999355094973082e-05,
-      "loss": 0.6063,
-      "step": 452
-    },
-    {
-      "epoch": 0.8,
-      "learning_rate": 1.9999309852453556e-05,
-      "loss": 0.6081,
-      "step": 453
-    },
-    {
-      "epoch": 0.81,
-      "learning_rate": 1.9999263076361046e-05,
-      "loss": 0.6235,
-      "step": 454
-    },
-    {
-      "epoch": 0.81,
-      "learning_rate": 1.9999214766702716e-05,
-      "loss": 0.6279,
-      "step": 455
-    },
-    {
-      "epoch": 0.81,
-      "learning_rate": 1.999916492348598e-05,
-      "loss": 0.6236,
-      "step": 456
-    },
-    {
-      "epoch": 0.81,
-      "learning_rate": 1.9999113546718478e-05,
-      "loss": 0.633,
-      "step": 457
-    },
-    {
-      "epoch": 0.81,
-      "learning_rate": 1.99990606364081e-05,
-      "loss": 0.63,
-      "step": 458
-    },
-    {
-      "epoch": 0.81,
-      "learning_rate": 1.9999006192562948e-05,
-      "loss": 0.5975,
-      "step": 459
-    },
-    {
-      "epoch": 0.82,
-      "learning_rate": 1.9998950215191383e-05,
-      "loss": 0.6178,
-      "step": 460
-    },
-    {
-      "epoch": 0.82,
-      "learning_rate": 1.9998892704301984e-05,
-      "loss": 0.5916,
-      "step": 461
-    },
-    {
-      "epoch": 0.82,
-      "learning_rate": 1.999883365990357e-05,
-      "loss": 0.6077,
-      "step": 462
-    },
-    {
-      "epoch": 0.82,
-      "learning_rate": 1.9998773082005205e-05,
-      "loss": 0.6011,
-      "step": 463
-    },
-    {
-      "epoch": 0.82,
-      "learning_rate": 1.999871097061617e-05,
-      "loss": 0.6109,
-      "step": 464
-    },
-    {
-      "epoch": 0.83,
-      "learning_rate": 1.9998647325745995e-05,
-      "loss": 0.5994,
-      "step": 465
-    },
-    {
-      "epoch": 0.83,
-      "learning_rate": 1.999858214740444e-05,
-      "loss": 0.5895,
-      "step": 466
-    },
-    {
-      "epoch": 0.83,
-      "learning_rate": 1.9998515435601505e-05,
-      "loss": 0.6405,
-      "step": 467
-    },
-    {
-      "epoch": 0.83,
-      "learning_rate": 1.9998447190347416e-05,
-      "loss": 0.6105,
-      "step": 468
-    },
-    {
-      "epoch": 0.83,
-      "learning_rate": 1.9998377411652644e-05,
-      "loss": 0.6158,
-      "step": 469
-    },
-    {
-      "epoch": 0.83,
-      "learning_rate": 1.999830609952789e-05,
-      "loss": 0.6179,
-      "step": 470
-    },
-    {
-      "epoch": 0.84,
-      "learning_rate": 1.9998233253984088e-05,
-      "loss": 0.6081,
-      "step": 471
-    },
-    {
-      "epoch": 0.84,
-      "learning_rate": 1.999815887503241e-05,
-      "loss": 0.5728,
-      "step": 472
-    },
-    {
-      "epoch": 0.84,
-      "learning_rate": 1.999808296268427e-05,
-      "loss": 0.5977,
-      "step": 473
-    },
-    {
-      "epoch": 0.84,
-      "learning_rate": 1.9998005516951303e-05,
-      "loss": 0.5753,
-      "step": 474
-    },
-    {
-      "epoch": 0.84,
-      "learning_rate": 1.999792653784539e-05,
-      "loss": 0.5827,
-      "step": 475
-    },
-    {
-      "epoch": 0.85,
-      "learning_rate": 1.9997846025378648e-05,
-      "loss": 0.6259,
-      "step": 476
-    },
-    {
-      "epoch": 0.85,
-      "learning_rate": 1.9997763979563418e-05,
-      "loss": 0.5997,
-      "step": 477
-    },
-    {
-      "epoch": 0.85,
-      "learning_rate": 1.9997680400412283e-05,
-      "loss": 0.6014,
-      "step": 478
-    },
-    {
-      "epoch": 0.85,
-      "learning_rate": 1.9997595287938062e-05,
-      "loss": 0.5851,
-      "step": 479
-    },
-    {
-      "epoch": 0.85,
-      "learning_rate": 1.999750864215382e-05,
-      "loss": 0.6525,
-      "step": 480
-    },
-    {
-      "epoch": 0.85,
-      "learning_rate": 1.9997420463072828e-05,
-      "loss": 0.6072,
-      "step": 481
-    },
-    {
-      "epoch": 0.86,
-      "learning_rate": 1.9997330750708623e-05,
-      "loss": 0.6276,
-      "step": 482
-    },
-    {
-      "epoch": 0.86,
-      "learning_rate": 1.9997239505074954e-05,
-      "loss": 0.6331,
-      "step": 483
-    },
-    {
-      "epoch": 0.86,
-      "learning_rate": 1.9997146726185826e-05,
-      "loss": 0.5996,
-      "step": 484
-    },
-    {
-      "epoch": 0.86,
-      "learning_rate": 1.999705241405546e-05,
-      "loss": 0.6122,
-      "step": 485
-    },
-    {
-      "epoch": 0.86,
-      "learning_rate": 1.9996956568698325e-05,
-      "loss": 0.5888,
-      "step": 486
-    },
-    {
-      "epoch": 0.86,
-      "learning_rate": 1.9996859190129118e-05,
-      "loss": 0.6154,
-      "step": 487
-    },
-    {
-      "epoch": 0.87,
-      "learning_rate": 1.999676027836277e-05,
-      "loss": 0.6008,
-      "step": 488
-    },
-    {
-      "epoch": 0.87,
-      "learning_rate": 1.999665983341446e-05,
-      "loss": 0.6225,
-      "step": 489
-    },
-    {
-      "epoch": 0.87,
-      "learning_rate": 1.999655785529959e-05,
-      "loss": 0.6065,
-      "step": 490
-    },
-    {
-      "epoch": 0.87,
-      "learning_rate": 1.9996454344033794e-05,
-      "loss": 0.5996,
-      "step": 491
-    },
-    {
-      "epoch": 0.87,
-      "learning_rate": 1.9996349299632953e-05,
-      "loss": 0.5845,
-      "step": 492
-    },
-    {
-      "epoch": 0.88,
-      "learning_rate": 1.9996242722113177e-05,
-      "loss": 0.6225,
-      "step": 493
-    },
-    {
-      "epoch": 0.88,
-      "learning_rate": 1.999613461149081e-05,
-      "loss": 0.602,
-      "step": 494
-    },
-    {
-      "epoch": 0.88,
-      "learning_rate": 1.9996024967782436e-05,
-      "loss": 0.6028,
-      "step": 495
-    },
-    {
-      "epoch": 0.88,
-      "learning_rate": 1.9995913791004867e-05,
-      "loss": 0.6112,
-      "step": 496
-    },
-    {
-      "epoch": 0.88,
-      "learning_rate": 1.9995801081175154e-05,
-      "loss": 0.6162,
-      "step": 497
-    },
-    {
-      "epoch": 0.88,
-      "learning_rate": 1.9995686838310586e-05,
-      "loss": 0.5914,
-      "step": 498
-    },
-    {
-      "epoch": 0.89,
-      "learning_rate": 1.9995571062428682e-05,
-      "loss": 0.6041,
-      "step": 499
-    },
-    {
-      "epoch": 0.89,
-      "learning_rate": 1.9995453753547198e-05,
-      "loss": 0.6,
-      "step": 500
-    },
-    {
-      "epoch": 0.89,
-      "learning_rate": 1.9995334911684127e-05,
-      "loss": 0.6035,
-      "step": 501
-    },
-    {
-      "epoch": 0.89,
-      "learning_rate": 1.9995214536857698e-05,
-      "loss": 0.5888,
-      "step": 502
-    },
-    {
-      "epoch": 0.89,
-      "learning_rate": 1.9995092629086365e-05,
-      "loss": 0.585,
-      "step": 503
-    },
-    {
-      "epoch": 0.89,
-      "learning_rate": 1.9994969188388828e-05,
-      "loss": 0.5802,
-      "step": 504
-    },
-    {
-      "epoch": 0.9,
-      "learning_rate": 1.9994844214784022e-05,
-      "loss": 0.6,
-      "step": 505
-    },
-    {
-      "epoch": 0.9,
-      "learning_rate": 1.9994717708291112e-05,
-      "loss": 0.5846,
-      "step": 506
-    },
-    {
-      "epoch": 0.9,
-      "learning_rate": 1.99945896689295e-05,
-      "loss": 0.6126,
-      "step": 507
-    },
-    {
-      "epoch": 0.9,
-      "learning_rate": 1.9994460096718827e-05,
-      "loss": 0.588,
-      "step": 508
-    },
-    {
-      "epoch": 0.9,
-      "learning_rate": 1.9994328991678957e-05,
-      "loss": 0.6372,
-      "step": 509
-    },
-    {
-      "epoch": 0.91,
-      "learning_rate": 1.999419635383e-05,
-      "loss": 0.6032,
-      "step": 510
-    },
-    {
-      "epoch": 0.91,
-      "learning_rate": 1.9994062183192303e-05,
-      "loss": 0.5952,
-      "step": 511
-    },
-    {
-      "epoch": 0.91,
-      "learning_rate": 1.999392647978644e-05,
-      "loss": 0.5611,
-      "step": 512
-    },
-    {
-      "epoch": 0.91,
-      "learning_rate": 1.9993789243633227e-05,
-      "loss": 0.5745,
-      "step": 513
-    },
-    {
-      "epoch": 0.91,
-      "learning_rate": 1.9993650474753705e-05,
-      "loss": 0.5806,
-      "step": 514
-    },
-    {
-      "epoch": 0.91,
-      "learning_rate": 1.999351017316916e-05,
-      "loss": 0.6006,
-      "step": 515
-    },
-    {
-      "epoch": 0.92,
-      "learning_rate": 1.9993368338901114e-05,
-      "loss": 0.588,
-      "step": 516
-    },
-    {
-      "epoch": 0.92,
-      "learning_rate": 1.999322497197131e-05,
-      "loss": 0.5888,
-      "step": 517
-    },
-    {
-      "epoch": 0.92,
-      "learning_rate": 1.9993080072401746e-05,
-      "loss": 0.5921,
-      "step": 518
-    },
-    {
-      "epoch": 0.92,
-      "learning_rate": 1.999293364021464e-05,
-      "loss": 0.5797,
-      "step": 519
-    },
-    {
-      "epoch": 0.92,
-      "learning_rate": 1.999278567543245e-05,
-      "loss": 0.5983,
-      "step": 520
-    },
-    {
-      "epoch": 0.92,
-      "learning_rate": 1.999263617807787e-05,
-      "loss": 0.5924,
-      "step": 521
-    },
-    {
-      "epoch": 0.93,
-      "learning_rate": 1.999248514817383e-05,
-      "loss": 0.5977,
-      "step": 522
-    },
-    {
-      "epoch": 0.93,
-      "learning_rate": 1.9992332585743486e-05,
-      "loss": 0.5923,
-      "step": 523
-    },
-    {
-      "epoch": 0.93,
-      "learning_rate": 1.9992178490810244e-05,
-      "loss": 0.5973,
-      "step": 524
-    },
-    {
-      "epoch": 0.93,
-      "learning_rate": 1.9992022863397737e-05,
-      "loss": 0.6248,
-      "step": 525
-    },
-    {
-      "epoch": 0.93,
-      "learning_rate": 1.9991865703529826e-05,
-      "loss": 0.5642,
-      "step": 526
-    },
-    {
-      "epoch": 0.94,
-      "learning_rate": 1.999170701123062e-05,
-      "loss": 0.568,
-      "step": 527
-    },
-    {
-      "epoch": 0.94,
-      "learning_rate": 1.999154678652446e-05,
-      "loss": 0.609,
-      "step": 528
-    },
-    {
-      "epoch": 0.94,
-      "learning_rate": 1.999138502943591e-05,
-      "loss": 0.5823,
-      "step": 529
-    },
-    {
-      "epoch": 0.94,
-      "learning_rate": 1.999122173998979e-05,
-      "loss": 0.5592,
-      "step": 530
-    },
-    {
-      "epoch": 0.94,
-      "learning_rate": 1.999105691821113e-05,
-      "loss": 0.5698,
-      "step": 531
-    },
-    {
-      "epoch": 0.94,
-      "learning_rate": 1.999089056412522e-05,
-      "loss": 0.5987,
-      "step": 532
-    },
-    {
-      "epoch": 0.95,
-      "learning_rate": 1.9990722677757572e-05,
-      "loss": 0.5695,
-      "step": 533
-    },
-    {
-      "epoch": 0.95,
-      "learning_rate": 1.9990553259133927e-05,
-      "loss": 0.5921,
-      "step": 534
-    },
-    {
-      "epoch": 0.95,
-      "learning_rate": 1.9990382308280272e-05,
-      "loss": 0.5887,
-      "step": 535
-    },
-    {
-      "epoch": 0.95,
-      "learning_rate": 1.999020982522283e-05,
-      "loss": 0.5859,
-      "step": 536
-    },
-    {
-      "epoch": 0.95,
-      "learning_rate": 1.9990035809988045e-05,
-      "loss": 0.6055,
-      "step": 537
-    },
-    {
-      "epoch": 0.96,
-      "learning_rate": 1.9989860262602616e-05,
-      "loss": 0.5835,
-      "step": 538
-    },
-    {
-      "epoch": 0.96,
-      "learning_rate": 1.998968318309346e-05,
-      "loss": 0.5651,
-      "step": 539
-    },
-    {
-      "epoch": 0.96,
-      "learning_rate": 1.9989504571487736e-05,
-      "loss": 0.5715,
-      "step": 540
-    },
-    {
-      "epoch": 0.96,
-      "learning_rate": 1.9989324427812836e-05,
-      "loss": 0.5785,
-      "step": 541
-    },
-    {
-      "epoch": 0.96,
-      "learning_rate": 1.998914275209639e-05,
-      "loss": 0.5611,
-      "step": 542
-    },
-    {
-      "epoch": 0.96,
-      "learning_rate": 1.9988959544366265e-05,
-      "loss": 0.5964,
-      "step": 543
-    },
-    {
-      "epoch": 0.97,
-      "learning_rate": 1.9988774804650552e-05,
-      "loss": 0.5999,
-      "step": 544
-    },
-    {
-      "epoch": 0.97,
-      "learning_rate": 1.998858853297759e-05,
-      "loss": 0.5519,
-      "step": 545
-    },
-    {
-      "epoch": 0.97,
-      "learning_rate": 1.9988400729375944e-05,
-      "loss": 0.5767,
-      "step": 546
-    },
-    {
-      "epoch": 0.97,
-      "learning_rate": 1.9988211393874413e-05,
-      "loss": 0.5756,
-      "step": 547
-    },
-    {
-      "epoch": 0.97,
-      "learning_rate": 1.9988020526502043e-05,
-      "loss": 0.5968,
-      "step": 548
-    },
-    {
-      "epoch": 0.97,
-      "learning_rate": 1.9987828127288105e-05,
-      "loss": 0.5844,
-      "step": 549
-    },
-    {
-      "epoch": 0.98,
-      "learning_rate": 1.9987634196262106e-05,
-      "loss": 0.5768,
-      "step": 550
-    },
-    {
-      "epoch": 0.98,
-      "learning_rate": 1.9987438733453786e-05,
-      "loss": 0.5916,
-      "step": 551
-    },
-    {
-      "epoch": 0.98,
-      "learning_rate": 1.9987241738893124e-05,
-      "loss": 0.5877,
-      "step": 552
-    },
-    {
-      "epoch": 0.98,
-      "learning_rate": 1.9987043212610336e-05,
-      "loss": 0.5747,
-      "step": 553
-    },
-    {
-      "epoch": 0.98,
-      "learning_rate": 1.998684315463587e-05,
-      "loss": 0.58,
-      "step": 554
-    },
-    {
-      "epoch": 0.99,
-      "learning_rate": 1.99866415650004e-05,
-      "loss": 0.5531,
-      "step": 555
-    },
-    {
-      "epoch": 0.99,
-      "learning_rate": 1.998643844373485e-05,
-      "loss": 0.5776,
-      "step": 556
-    },
-    {
-      "epoch": 0.99,
-      "learning_rate": 1.9986233790870372e-05,
-      "loss": 0.6022,
-      "step": 557
-    },
-    {
-      "epoch": 0.99,
-      "learning_rate": 1.9986027606438353e-05,
-      "loss": 0.5771,
-      "step": 558
-    },
-    {
-      "epoch": 0.99,
-      "learning_rate": 1.9985819890470416e-05,
-      "loss": 0.5716,
-      "step": 559
-    },
-    {
-      "epoch": 0.99,
-      "learning_rate": 1.9985610642998412e-05,
-      "loss": 0.5843,
-      "step": 560
-    },
-    {
-      "epoch": 1.0,
-      "learning_rate": 1.998539986405444e-05,
-      "loss": 0.5697,
-      "step": 561
-    },
-    {
-      "epoch": 1.0,
-      "learning_rate": 1.9985187553670825e-05,
-      "loss": 0.5678,
-      "step": 562
-    },
-    {
-      "epoch": 1.0,
-      "learning_rate": 1.998497371188013e-05,
-      "loss": 0.558,
-      "step": 563
-    },
-    {
-      "epoch": 1.0,
-      "eval_loss": 0.609296441078186,
-      "eval_runtime": 3448.5758,
-      "eval_samples_per_second": 4.16,
-      "eval_steps_per_second": 1.04,
-      "step": 563
-    },
-    {
-      "epoch": 1.0,
-      "learning_rate": 1.9984758338715144e-05,
-      "loss": 0.5845,
-      "step": 564
-    },
-    {
-      "epoch": 1.0,
-      "learning_rate": 1.9984541434208903e-05,
-      "loss": 0.5839,
-      "step": 565
-    },
-    {
-      "epoch": 1.0,
-      "learning_rate": 1.9984322998394677e-05,
-      "loss": 0.5877,
-      "step": 566
-    },
-    {
-      "epoch": 1.01,
-      "learning_rate": 1.998410303130596e-05,
-      "loss": 0.5895,
-      "step": 567
-    },
-    {
-      "epoch": 1.01,
-      "learning_rate": 1.9983881532976497e-05,
-      "loss": 0.6125,
-      "step": 568
-    },
-    {
-      "epoch": 1.01,
-      "learning_rate": 1.9983658503440248e-05,
-      "loss": 0.5905,
-      "step": 569
-    },
-    {
-      "epoch": 1.01,
-      "learning_rate": 1.9983433942731427e-05,
-      "loss": 0.5596,
-      "step": 570
-    },
-    {
-      "epoch": 1.01,
-      "learning_rate": 1.9983207850884468e-05,
-      "loss": 0.5795,
-      "step": 571
-    },
-    {
-      "epoch": 1.02,
-      "learning_rate": 1.9982980227934052e-05,
-      "loss": 0.5799,
-      "step": 572
-    },
-    {
-      "epoch": 1.02,
-      "learning_rate": 1.9982751073915087e-05,
-      "loss": 0.5843,
-      "step": 573
-    },
-    {
-      "epoch": 1.02,
-      "learning_rate": 1.9982520388862718e-05,
-      "loss": 0.5754,
-      "step": 574
-    },
-    {
-      "epoch": 1.02,
-      "learning_rate": 1.9982288172812323e-05,
-      "loss": 0.5881,
-      "step": 575
-    },
-    {
-      "epoch": 1.02,
-      "learning_rate": 1.9982054425799516e-05,
-      "loss": 0.5854,
-      "step": 576
-    },
-    {
-      "epoch": 1.02,
-      "learning_rate": 1.998181914786015e-05,
-      "loss": 0.6005,
-      "step": 577
-    },
-    {
-      "epoch": 1.03,
-      "learning_rate": 1.9981582339030307e-05,
-      "loss": 0.55,
-      "step": 578
-    },
-    {
-      "epoch": 1.03,
-      "learning_rate": 1.9981343999346302e-05,
-      "loss": 0.5438,
-      "step": 579
-    },
-    {
-      "epoch": 1.03,
-      "learning_rate": 1.9981104128844694e-05,
-      "loss": 0.5905,
-      "step": 580
-    },
-    {
-      "epoch": 1.03,
-      "learning_rate": 1.9980862727562274e-05,
-      "loss": 0.5614,
-      "step": 581
-    },
-    {
-      "epoch": 1.03,
-      "learning_rate": 1.998061979553606e-05,
-      "loss": 0.5818,
-      "step": 582
-    },
-    {
-      "epoch": 1.04,
-      "learning_rate": 1.9980375332803305e-05,
-      "loss": 0.5584,
-      "step": 583
-    },
-    {
-      "epoch": 1.04,
-      "learning_rate": 1.9980129339401512e-05,
-      "loss": 0.5843,
-      "step": 584
-    },
-    {
-      "epoch": 1.04,
-      "learning_rate": 1.9979881815368406e-05,
-      "loss": 0.6011,
-      "step": 585
-    },
-    {
-      "epoch": 1.04,
-      "learning_rate": 1.9979632760741942e-05,
-      "loss": 0.5964,
-      "step": 586
-    },
-    {
-      "epoch": 1.04,
-      "learning_rate": 1.997938217556033e-05,
-      "loss": 0.5641,
-      "step": 587
-    },
-    {
-      "epoch": 1.04,
-      "learning_rate": 1.997913005986199e-05,
-      "loss": 0.5634,
-      "step": 588
-    },
-    {
-      "epoch": 1.05,
-      "learning_rate": 1.997887641368559e-05,
-      "loss": 0.578,
-      "step": 589
-    },
-    {
-      "epoch": 1.05,
-      "learning_rate": 1.9978621237070037e-05,
-      "loss": 0.5802,
-      "step": 590
-    },
-    {
-      "epoch": 1.05,
-      "learning_rate": 1.9978364530054465e-05,
-      "loss": 0.5661,
-      "step": 591
-    },
-    {
-      "epoch": 1.05,
-      "learning_rate": 1.997810629267824e-05,
-      "loss": 0.57,
-      "step": 592
-    },
-    {
-      "epoch": 1.05,
-      "learning_rate": 1.9977846524980974e-05,
-      "loss": 0.5736,
-      "step": 593
-    },
-    {
-      "epoch": 1.05,
-      "learning_rate": 1.99775852270025e-05,
-      "loss": 0.5762,
-      "step": 594
-    },
-    {
-      "epoch": 1.06,
-      "learning_rate": 1.9977322398782898e-05,
-      "loss": 0.5744,
-      "step": 595
-    },
-    {
-      "epoch": 1.06,
-      "learning_rate": 1.997705804036248e-05,
-      "loss": 0.544,
-      "step": 596
-    },
-    {
-      "epoch": 1.06,
-      "learning_rate": 1.9976792151781778e-05,
-      "loss": 0.5597,
-      "step": 597
-    },
-    {
-      "epoch": 1.06,
-      "learning_rate": 1.9976524733081584e-05,
-      "loss": 0.5928,
-      "step": 598
-    },
-    {
-      "epoch": 1.06,
-      "learning_rate": 1.9976255784302904e-05,
-      "loss": 0.5637,
-      "step": 599
-    },
-    {
-      "epoch": 1.07,
-      "learning_rate": 1.9975985305486988e-05,
-      "loss": 0.6159,
-      "step": 600
-    },
-    {
-      "epoch": 1.07,
-      "learning_rate": 1.997571329667532e-05,
-      "loss": 0.5793,
-      "step": 601
-    },
-    {
-      "epoch": 1.07,
-      "learning_rate": 1.997543975790962e-05,
-      "loss": 0.5722,
-      "step": 602
-    },
-    {
-      "epoch": 1.07,
-      "learning_rate": 1.997516468923183e-05,
-      "loss": 0.565,
-      "step": 603
-    },
-    {
-      "epoch": 1.07,
-      "learning_rate": 1.9974888090684145e-05,
-      "loss": 0.5605,
-      "step": 604
-    },
-    {
-      "epoch": 1.07,
-      "learning_rate": 1.9974609962308986e-05,
-      "loss": 0.5722,
-      "step": 605
-    },
-    {
-      "epoch": 1.08,
-      "learning_rate": 1.9974330304149004e-05,
-      "loss": 0.5645,
-      "step": 606
-    },
-    {
-      "epoch": 1.08,
-      "learning_rate": 1.9974049116247096e-05,
-      "loss": 0.5708,
-      "step": 607
-    },
-    {
-      "epoch": 1.08,
-      "learning_rate": 1.9973766398646386e-05,
-      "loss": 0.551,
-      "step": 608
-    },
-    {
-      "epoch": 1.08,
-      "learning_rate": 1.9973482151390228e-05,
-      "loss": 0.5604,
-      "step": 609
-    },
-    {
-      "epoch": 1.08,
-      "learning_rate": 1.9973196374522224e-05,
-      "loss": 0.572,
-      "step": 610
-    },
-    {
-      "epoch": 1.08,
-      "learning_rate": 1.9972909068086197e-05,
-      "loss": 0.5849,
-      "step": 611
-    },
-    {
-      "epoch": 1.09,
-      "learning_rate": 1.9972620232126215e-05,
-      "loss": 0.5786,
-      "step": 612
-    },
-    {
-      "epoch": 1.09,
-      "learning_rate": 1.997232986668657e-05,
-      "loss": 0.5884,
-      "step": 613
-    },
-    {
-      "epoch": 1.09,
-      "learning_rate": 1.9972037971811802e-05,
-      "loss": 0.5855,
-      "step": 614
-    },
-    {
-      "epoch": 1.09,
-      "learning_rate": 1.9971744547546676e-05,
-      "loss": 0.5811,
-      "step": 615
-    },
-    {
-      "epoch": 1.09,
-      "learning_rate": 1.9971449593936193e-05,
-      "loss": 0.5738,
-      "step": 616
-    },
-    {
-      "epoch": 1.1,
-      "learning_rate": 1.9971153111025586e-05,
-      "loss": 0.5696,
-      "step": 617
-    },
-    {
-      "epoch": 1.1,
-      "learning_rate": 1.9970855098860335e-05,
-      "loss": 0.5812,
-      "step": 618
-    },
-    {
-      "epoch": 1.1,
-      "learning_rate": 1.9970555557486137e-05,
-      "loss": 0.587,
-      "step": 619
-    },
-    {
-      "epoch": 1.1,
-      "learning_rate": 1.9970254486948933e-05,
-      "loss": 0.5543,
-      "step": 620
-    },
-    {
-      "epoch": 1.1,
-      "learning_rate": 1.99699518872949e-05,
-      "loss": 0.5564,
-      "step": 621
-    },
-    {
-      "epoch": 1.1,
-      "learning_rate": 1.9969647758570447e-05,
-      "loss": 0.5518,
-      "step": 622
-    },
-    {
-      "epoch": 1.11,
-      "learning_rate": 1.9969342100822217e-05,
-      "loss": 0.5821,
-      "step": 623
-    },
-    {
-      "epoch": 1.11,
-      "learning_rate": 1.996903491409709e-05,
-      "loss": 0.5742,
-      "step": 624
-    },
-    {
-      "epoch": 1.11,
-      "learning_rate": 1.9968726198442173e-05,
-      "loss": 0.5715,
-      "step": 625
-    },
-    {
-      "epoch": 1.11,
-      "learning_rate": 1.9968415953904823e-05,
-      "loss": 0.5878,
-      "step": 626
-    },
-    {
-      "epoch": 1.11,
-      "learning_rate": 1.996810418053261e-05,
-      "loss": 0.5405,
-      "step": 627
-    },
-    {
-      "epoch": 1.11,
-      "learning_rate": 1.9967790878373358e-05,
-      "loss": 0.5635,
-      "step": 628
-    },
-    {
-      "epoch": 1.12,
-      "learning_rate": 1.9967476047475115e-05,
-      "loss": 0.5753,
-      "step": 629
-    },
-    {
-      "epoch": 1.12,
-      "learning_rate": 1.996715968788616e-05,
-      "loss": 0.5644,
-      "step": 630
-    },
-    {
-      "epoch": 1.12,
-      "learning_rate": 1.996684179965503e-05,
-      "loss": 0.5659,
-      "step": 631
-    },
-    {
-      "epoch": 1.12,
-      "learning_rate": 1.9966522382830455e-05,
-      "loss": 0.5614,
-      "step": 632
-    },
-    {
-      "epoch": 1.12,
-      "learning_rate": 1.996620143746144e-05,
-      "loss": 0.568,
-      "step": 633
-    },
-    {
-      "epoch": 1.13,
-      "learning_rate": 1.9965878963597208e-05,
-      "loss": 0.5503,
-      "step": 634
-    },
-    {
-      "epoch": 1.13,
-      "learning_rate": 1.9965554961287204e-05,
-      "loss": 0.5798,
-      "step": 635
-    },
-    {
-      "epoch": 1.13,
-      "learning_rate": 1.996522943058113e-05,
-      "loss": 0.5646,
-      "step": 636
-    },
-    {
-      "epoch": 1.13,
-      "learning_rate": 1.996490237152891e-05,
-      "loss": 0.5843,
-      "step": 637
-    },
-    {
-      "epoch": 1.13,
-      "learning_rate": 1.9964573784180702e-05,
-      "loss": 0.5771,
-      "step": 638
-    },
-    {
-      "epoch": 1.13,
-      "learning_rate": 1.99642436685869e-05,
-      "loss": 0.572,
-      "step": 639
-    },
-    {
-      "epoch": 1.14,
-      "learning_rate": 1.9963912024798136e-05,
-      "loss": 0.5708,
-      "step": 640
-    },
-    {
-      "epoch": 1.14,
-      "learning_rate": 1.9963578852865273e-05,
-      "loss": 0.5645,
-      "step": 641
-    },
-    {
-      "epoch": 1.14,
-      "learning_rate": 1.9963244152839407e-05,
-      "loss": 0.5667,
-      "step": 642
-    },
-    {
-      "epoch": 1.14,
-      "learning_rate": 1.9962907924771873e-05,
-      "loss": 0.5616,
-      "step": 643
-    },
-    {
-      "epoch": 1.14,
-      "learning_rate": 1.9962570168714237e-05,
-      "loss": 0.5833,
-      "step": 644
-    },
-    {
-      "epoch": 1.15,
-      "learning_rate": 1.9962230884718296e-05,
-      "loss": 0.5856,
-      "step": 645
-    },
-    {
-      "epoch": 1.15,
-      "learning_rate": 1.996189007283609e-05,
-      "loss": 0.5677,
-      "step": 646
-    },
-    {
-      "epoch": 1.15,
-      "learning_rate": 1.996154773311989e-05,
-      "loss": 0.5838,
-      "step": 647
-    },
-    {
-      "epoch": 1.15,
-      "learning_rate": 1.9961203865622193e-05,
-      "loss": 0.6084,
-      "step": 648
-    },
-    {
-      "epoch": 1.15,
-      "learning_rate": 1.996085847039574e-05,
-      "loss": 0.5714,
-      "step": 649
-    },
-    {
-      "epoch": 1.15,
-      "learning_rate": 1.9960511547493508e-05,
-      "loss": 0.5611,
-      "step": 650
-    },
-    {
-      "epoch": 1.16,
-      "learning_rate": 1.9960163096968702e-05,
-      "loss": 0.5492,
-      "step": 651
-    },
-    {
-      "epoch": 1.16,
-      "learning_rate": 1.995981311887476e-05,
-      "loss": 0.5741,
-      "step": 652
-    },
-    {
-      "epoch": 1.16,
-      "learning_rate": 1.9959461613265353e-05,
-      "loss": 0.5629,
-      "step": 653
-    },
-    {
-      "epoch": 1.16,
-      "learning_rate": 1.9959108580194403e-05,
-      "loss": 0.5844,
-      "step": 654
-    },
-    {
-      "epoch": 1.16,
-      "learning_rate": 1.9958754019716047e-05,
-      "loss": 0.5785,
-      "step": 655
-    },
-    {
-      "epoch": 1.16,
-      "learning_rate": 1.9958397931884663e-05,
-      "loss": 0.5506,
-      "step": 656
-    },
-    {
-      "epoch": 1.17,
-      "learning_rate": 1.9958040316754866e-05,
-      "loss": 0.5564,
-      "step": 657
-    },
-    {
-      "epoch": 1.17,
-      "learning_rate": 1.9957681174381497e-05,
-      "loss": 0.5702,
-      "step": 658
-    },
-    {
-      "epoch": 1.17,
-      "learning_rate": 1.9957320504819643e-05,
-      "loss": 0.5885,
-      "step": 659
-    },
-    {
-      "epoch": 1.17,
-      "learning_rate": 1.9956958308124617e-05,
-      "loss": 0.5588,
-      "step": 660
-    },
-    {
-      "epoch": 1.17,
-      "learning_rate": 1.995659458435197e-05,
-      "loss": 0.5559,
-      "step": 661
-    },
-    {
-      "epoch": 1.18,
-      "learning_rate": 1.9956229333557478e-05,
-      "loss": 0.5268,
-      "step": 662
-    },
-    {
-      "epoch": 1.18,
-      "learning_rate": 1.995586255579717e-05,
-      "loss": 0.5382,
-      "step": 663
-    },
-    {
-      "epoch": 1.18,
-      "learning_rate": 1.995549425112729e-05,
-      "loss": 0.556,
-      "step": 664
-    },
-    {
-      "epoch": 1.18,
-      "learning_rate": 1.995512441960433e-05,
-      "loss": 0.5332,
-      "step": 665
-    },
-    {
-      "epoch": 1.18,
-      "learning_rate": 1.9954753061285004e-05,
-      "loss": 0.5486,
-      "step": 666
-    },
-    {
-      "epoch": 1.18,
-      "learning_rate": 1.995438017622627e-05,
-      "loss": 0.5704,
-      "step": 667
-    },
-    {
-      "epoch": 1.19,
-      "learning_rate": 1.9954005764485313e-05,
-      "loss": 0.5624,
-      "step": 668
-    },
-    {
-      "epoch": 1.19,
-      "learning_rate": 1.9953629826119562e-05,
-      "loss": 0.5529,
-      "step": 669
-    },
-    {
-      "epoch": 1.19,
-      "learning_rate": 1.995325236118667e-05,
-      "loss": 0.546,
-      "step": 670
-    },
-    {
-      "epoch": 1.19,
-      "learning_rate": 1.9952873369744528e-05,
-      "loss": 0.5704,
-      "step": 671
-    },
-    {
-      "epoch": 1.19,
-      "learning_rate": 1.995249285185126e-05,
-      "loss": 0.5608,
-      "step": 672
-    },
-    {
-      "epoch": 1.19,
-      "learning_rate": 1.9952110807565233e-05,
-      "loss": 0.568,
-      "step": 673
-    },
-    {
-      "epoch": 1.2,
-      "learning_rate": 1.995172723694503e-05,
-      "loss": 0.5269,
-      "step": 674
-    },
-    {
-      "epoch": 1.2,
-      "learning_rate": 1.9951342140049483e-05,
-      "loss": 0.5709,
-      "step": 675
-    },
-    {
-      "epoch": 1.2,
-      "learning_rate": 1.995095551693765e-05,
-      "loss": 0.5613,
-      "step": 676
-    },
-    {
-      "epoch": 1.2,
-      "learning_rate": 1.995056736766883e-05,
-      "loss": 0.57,
-      "step": 677
-    },
-    {
-      "epoch": 1.2,
-      "learning_rate": 1.9950177692302557e-05,
-      "loss": 0.5464,
-      "step": 678
-    },
-    {
-      "epoch": 1.21,
-      "learning_rate": 1.9949786490898587e-05,
-      "loss": 0.5985,
-      "step": 679
-    },
-    {
-      "epoch": 1.21,
-      "learning_rate": 1.994939376351692e-05,
-      "loss": 0.5664,
-      "step": 680
-    },
-    {
-      "epoch": 1.21,
-      "learning_rate": 1.994899951021779e-05,
-      "loss": 0.5698,
-      "step": 681
-    },
-    {
-      "epoch": 1.21,
-      "learning_rate": 1.994860373106166e-05,
-      "loss": 0.5629,
-      "step": 682
-    },
-    {
-      "epoch": 1.21,
-      "learning_rate": 1.9948206426109234e-05,
-      "loss": 0.553,
-      "step": 683
-    },
-    {
-      "epoch": 1.21,
-      "learning_rate": 1.994780759542144e-05,
-      "loss": 0.5793,
-      "step": 684
-    },
-    {
-      "epoch": 1.22,
-      "learning_rate": 1.994740723905945e-05,
-      "loss": 0.5595,
-      "step": 685
-    },
-    {
-      "epoch": 1.22,
-      "learning_rate": 1.9947005357084666e-05,
-      "loss": 0.5897,
-      "step": 686
-    },
-    {
-      "epoch": 1.22,
-      "learning_rate": 1.9946601949558722e-05,
-      "loss": 0.5477,
-      "step": 687
-    },
-    {
-      "epoch": 1.22,
-      "learning_rate": 1.9946197016543486e-05,
-      "loss": 0.5537,
-      "step": 688
-    },
-    {
-      "epoch": 1.22,
-      "learning_rate": 1.9945790558101065e-05,
-      "loss": 0.5498,
-      "step": 689
-    },
-    {
-      "epoch": 1.23,
-      "learning_rate": 1.994538257429379e-05,
-      "loss": 0.5658,
-      "step": 690
-    },
-    {
-      "epoch": 1.23,
-      "learning_rate": 1.9944973065184246e-05,
-      "loss": 0.5897,
-      "step": 691
-    },
-    {
-      "epoch": 1.23,
-      "learning_rate": 1.9944562030835226e-05,
-      "loss": 0.577,
-      "step": 692
-    },
-    {
-      "epoch": 1.23,
-      "learning_rate": 1.9944149471309777e-05,
-      "loss": 0.5663,
-      "step": 693
-    },
-    {
-      "epoch": 1.23,
-      "learning_rate": 1.9943735386671166e-05,
-      "loss": 0.5668,
-      "step": 694
-    },
-    {
-      "epoch": 1.23,
-      "learning_rate": 1.9943319776982907e-05,
-      "loss": 0.548,
-      "step": 695
-    },
-    {
-      "epoch": 1.24,
-      "learning_rate": 1.9942902642308737e-05,
-      "loss": 0.5431,
-      "step": 696
-    },
-    {
-      "epoch": 1.24,
-      "learning_rate": 1.994248398271263e-05,
-      "loss": 0.572,
-      "step": 697
-    },
-    {
-      "epoch": 1.24,
-      "learning_rate": 1.99420637982588e-05,
-      "loss": 0.5623,
-      "step": 698
-    },
-    {
-      "epoch": 1.24,
-      "learning_rate": 1.9941642089011684e-05,
-      "loss": 0.5731,
-      "step": 699
-    },
-    {
-      "epoch": 1.24,
-      "learning_rate": 1.9941218855035963e-05,
-      "loss": 0.5774,
-      "step": 700
-    },
-    {
-      "epoch": 1.24,
-      "learning_rate": 1.9940794096396545e-05,
-      "loss": 0.5643,
-      "step": 701
-    },
-    {
-      "epoch": 1.25,
-      "learning_rate": 1.9940367813158578e-05,
-      "loss": 0.5385,
-      "step": 702
-    },
-    {
-      "epoch": 1.25,
-      "learning_rate": 1.9939940005387436e-05,
-      "loss": 0.5717,
-      "step": 703
-    },
-    {
-      "epoch": 1.25,
-      "learning_rate": 1.993951067314873e-05,
-      "loss": 0.5602,
-      "step": 704
-    },
-    {
-      "epoch": 1.25,
-      "learning_rate": 1.9939079816508312e-05,
-      "loss": 0.5543,
-      "step": 705
-    },
-    {
-      "epoch": 1.25,
-      "learning_rate": 1.9938647435532254e-05,
-      "loss": 0.5442,
-      "step": 706
-    },
-    {
-      "epoch": 1.26,
-      "learning_rate": 1.9938213530286875e-05,
-      "loss": 0.5799,
-      "step": 707
-    },
-    {
-      "epoch": 1.26,
-      "learning_rate": 1.9937778100838717e-05,
-      "loss": 0.5712,
-      "step": 708
-    },
-    {
-      "epoch": 1.26,
-      "learning_rate": 1.993734114725457e-05,
-      "loss": 0.5577,
-      "step": 709
-    },
-    {
-      "epoch": 1.26,
-      "learning_rate": 1.9936902669601436e-05,
-      "loss": 0.5588,
-      "step": 710
-    },
-    {
-      "epoch": 1.26,
-      "learning_rate": 1.9936462667946574e-05,
-      "loss": 0.5485,
-      "step": 711
-    },
-    {
-      "epoch": 1.26,
-      "learning_rate": 1.993602114235746e-05,
-      "loss": 0.5572,
-      "step": 712
-    },
-    {
-      "epoch": 1.27,
-      "learning_rate": 1.9935578092901817e-05,
-      "loss": 0.5341,
-      "step": 713
-    },
-    {
-      "epoch": 1.27,
-      "learning_rate": 1.9935133519647586e-05,
-      "loss": 0.5544,
-      "step": 714
-    },
-    {
-      "epoch": 1.27,
-      "learning_rate": 1.9934687422662954e-05,
-      "loss": 0.5655,
-      "step": 715
-    },
-    {
-      "epoch": 1.27,
-      "learning_rate": 1.9934239802016337e-05,
-      "loss": 0.5723,
-      "step": 716
-    },
-    {
-      "epoch": 1.27,
-      "learning_rate": 1.993379065777639e-05,
-      "loss": 0.5392,
-      "step": 717
-    },
-    {
-      "epoch": 1.27,
-      "learning_rate": 1.993333999001199e-05,
-      "loss": 0.554,
-      "step": 718
-    },
-    {
-      "epoch": 1.28,
-      "learning_rate": 1.993288779879226e-05,
-      "loss": 0.5446,
-      "step": 719
-    },
-    {
-      "epoch": 1.28,
-      "learning_rate": 1.9932434084186553e-05,
-      "loss": 0.5335,
-      "step": 720
-    },
-    {
-      "epoch": 1.28,
-      "learning_rate": 1.9931978846264446e-05,
-      "loss": 0.5403,
-      "step": 721
-    },
-    {
-      "epoch": 1.28,
-      "learning_rate": 1.9931522085095766e-05,
-      "loss": 0.5456,
-      "step": 722
-    },
-    {
-      "epoch": 1.28,
-      "learning_rate": 1.9931063800750564e-05,
-      "loss": 0.5446,
-      "step": 723
-    },
-    {
-      "epoch": 1.29,
-      "learning_rate": 1.9930603993299125e-05,
-      "loss": 0.5825,
-      "step": 724
-    },
-    {
-      "epoch": 1.29,
-      "learning_rate": 1.9930142662811967e-05,
-      "loss": 0.5584,
-      "step": 725
-    },
-    {
-      "epoch": 1.29,
-      "learning_rate": 1.9929679809359848e-05,
-      "loss": 0.565,
-      "step": 726
-    },
-    {
-      "epoch": 1.29,
-      "learning_rate": 1.9929215433013753e-05,
-      "loss": 0.5444,
-      "step": 727
-    },
-    {
-      "epoch": 1.29,
-      "learning_rate": 1.9928749533844895e-05,
-      "loss": 0.5479,
-      "step": 728
-    },
-    {
-      "epoch": 1.29,
-      "learning_rate": 1.9928282111924736e-05,
-      "loss": 0.5452,
-      "step": 729
-    },
-    {
-      "epoch": 1.3,
-      "learning_rate": 1.9927813167324967e-05,
-      "loss": 0.566,
-      "step": 730
-    },
-    {
-      "epoch": 1.3,
-      "learning_rate": 1.99273427001175e-05,
-      "loss": 0.559,
-      "step": 731
-    },
-    {
-      "epoch": 1.3,
-      "learning_rate": 1.9926870710374493e-05,
-      "loss": 0.5456,
-      "step": 732
-    },
-    {
-      "epoch": 1.3,
-      "learning_rate": 1.9926397198168333e-05,
-      "loss": 0.574,
-      "step": 733
-    },
-    {
-      "epoch": 1.3,
-      "learning_rate": 1.992592216357165e-05,
-      "loss": 0.5463,
-      "step": 734
-    },
-    {
-      "epoch": 1.3,
-      "learning_rate": 1.9925445606657286e-05,
-      "loss": 0.5374,
-      "step": 735
-    },
-    {
-      "epoch": 1.31,
-      "learning_rate": 1.992496752749834e-05,
-      "loss": 0.5512,
-      "step": 736
-    },
-    {
-      "epoch": 1.31,
-      "learning_rate": 1.9924487926168125e-05,
-      "loss": 0.5577,
-      "step": 737
-    },
-    {
-      "epoch": 1.31,
-      "learning_rate": 1.9924006802740203e-05,
-      "loss": 0.5637,
-      "step": 738
-    },
-    {
-      "epoch": 1.31,
-      "learning_rate": 1.992352415728836e-05,
-      "loss": 0.5527,
-      "step": 739
-    },
-    {
-      "epoch": 1.31,
-      "learning_rate": 1.9923039989886618e-05,
-      "loss": 0.563,
-      "step": 740
-    },
-    {
-      "epoch": 1.32,
-      "learning_rate": 1.9922554300609237e-05,
-      "loss": 0.5437,
-      "step": 741
-    },
-    {
-      "epoch": 1.32,
-      "learning_rate": 1.99220670895307e-05,
-      "loss": 0.5675,
-      "step": 742
-    },
-    {
-      "epoch": 1.32,
-      "learning_rate": 1.9921578356725736e-05,
-      "loss": 0.5622,
-      "step": 743
-    },
-    {
-      "epoch": 1.32,
-      "learning_rate": 1.9921088102269296e-05,
-      "loss": 0.5397,
-      "step": 744
-    },
-    {
-      "epoch": 1.32,
-      "learning_rate": 1.992059632623657e-05,
-      "loss": 0.5497,
-      "step": 745
-    },
-    {
-      "epoch": 1.32,
-      "learning_rate": 1.9920103028702983e-05,
-      "loss": 0.5496,
-      "step": 746
-    },
-    {
-      "epoch": 1.33,
-      "learning_rate": 1.991960820974419e-05,
-      "loss": 0.5832,
-      "step": 747
-    },
-    {
-      "epoch": 1.33,
-      "learning_rate": 1.9919111869436075e-05,
-      "loss": 0.5394,
-      "step": 748
-    },
-    {
-      "epoch": 1.33,
-      "learning_rate": 1.991861400785477e-05,
-      "loss": 0.5659,
-      "step": 749
-    },
-    {
-      "epoch": 1.33,
-      "learning_rate": 1.9918114625076627e-05,
-      "loss": 0.5397,
-      "step": 750
-    },
-    {
-      "epoch": 1.33,
-      "learning_rate": 1.9917613721178233e-05,
-      "loss": 0.5491,
-      "step": 751
-    },
-    {
-      "epoch": 1.34,
-      "learning_rate": 1.9917111296236413e-05,
-      "loss": 0.5711,
-      "step": 752
-    },
-    {
-      "epoch": 1.34,
-      "learning_rate": 1.991660735032822e-05,
-      "loss": 0.5498,
-      "step": 753
-    },
-    {
-      "epoch": 1.34,
-      "learning_rate": 1.9916101883530946e-05,
-      "loss": 0.5569,
-      "step": 754
-    },
-    {
-      "epoch": 1.34,
-      "learning_rate": 1.9915594895922113e-05,
-      "loss": 0.5775,
-      "step": 755
-    },
-    {
-      "epoch": 1.34,
-      "learning_rate": 1.9915086387579478e-05,
-      "loss": 0.566,
-      "step": 756
-    },
-    {
-      "epoch": 1.34,
-      "learning_rate": 1.991457635858103e-05,
-      "loss": 0.5549,
-      "step": 757
-    },
-    {
-      "epoch": 1.35,
-      "learning_rate": 1.9914064809004984e-05,
-      "loss": 0.5527,
-      "step": 758
-    },
-    {
-      "epoch": 1.35,
-      "learning_rate": 1.9913551738929803e-05,
-      "loss": 0.5535,
-      "step": 759
-    },
-    {
-      "epoch": 1.35,
-      "learning_rate": 1.9913037148434174e-05,
-      "loss": 0.5533,
-      "step": 760
-    },
-    {
-      "epoch": 1.35,
-      "learning_rate": 1.9912521037597015e-05,
-      "loss": 0.5449,
-      "step": 761
-    },
-    {
-      "epoch": 1.35,
-      "learning_rate": 1.9912003406497484e-05,
-      "loss": 0.5524,
-      "step": 762
-    },
-    {
-      "epoch": 1.35,
-      "learning_rate": 1.9911484255214968e-05,
-      "loss": 0.5341,
-      "step": 763
-    },
-    {
-      "epoch": 1.36,
-      "learning_rate": 1.9910963583829092e-05,
-      "loss": 0.5545,
-      "step": 764
-    },
-    {
-      "epoch": 1.36,
-      "learning_rate": 1.9910441392419702e-05,
-      "loss": 0.5479,
-      "step": 765
-    },
-    {
-      "epoch": 1.36,
-      "learning_rate": 1.9909917681066894e-05,
-      "loss": 0.5451,
-      "step": 766
-    },
-    {
-      "epoch": 1.36,
-      "learning_rate": 1.9909392449850983e-05,
-      "loss": 0.554,
-      "step": 767
-    },
-    {
-      "epoch": 1.36,
-      "learning_rate": 1.990886569885252e-05,
-      "loss": 0.5402,
-      "step": 768
-    },
-    {
-      "epoch": 1.37,
-      "learning_rate": 1.9908337428152304e-05,
-      "loss": 0.5343,
-      "step": 769
-    },
-    {
-      "epoch": 1.37,
-      "learning_rate": 1.9907807637831342e-05,
-      "loss": 0.5502,
-      "step": 770
-    },
-    {
-      "epoch": 1.37,
-      "learning_rate": 1.9907276327970892e-05,
-      "loss": 0.5579,
-      "step": 771
-    },
-    {
-      "epoch": 1.37,
-      "learning_rate": 1.9906743498652444e-05,
-      "loss": 0.5768,
-      "step": 772
-    },
-    {
-      "epoch": 1.37,
-      "learning_rate": 1.9906209149957707e-05,
-      "loss": 0.5563,
-      "step": 773
-    },
-    {
-      "epoch": 1.37,
-      "learning_rate": 1.990567328196864e-05,
-      "loss": 0.5409,
-      "step": 774
-    },
-    {
-      "epoch": 1.38,
-      "learning_rate": 1.9905135894767424e-05,
-      "loss": 0.5601,
-      "step": 775
-    },
-    {
-      "epoch": 1.38,
-      "learning_rate": 1.9904596988436483e-05,
-      "loss": 0.5435,
-      "step": 776
-    },
-    {
-      "epoch": 1.38,
-      "learning_rate": 1.990405656305846e-05,
-      "loss": 0.5813,
-      "step": 777
-    },
-    {
-      "epoch": 1.38,
-      "learning_rate": 1.9903514618716248e-05,
-      "loss": 0.5887,
-      "step": 778
-    },
-    {
-      "epoch": 1.38,
-      "learning_rate": 1.9902971155492955e-05,
-      "loss": 0.5595,
-      "step": 779
-    },
-    {
-      "epoch": 1.38,
-      "learning_rate": 1.9902426173471933e-05,
-      "loss": 0.5474,
-      "step": 780
-    },
-    {
-      "epoch": 1.39,
-      "learning_rate": 1.990187967273677e-05,
-      "loss": 0.5672,
-      "step": 781
-    },
-    {
-      "epoch": 1.39,
-      "learning_rate": 1.9901331653371275e-05,
-      "loss": 0.5767,
-      "step": 782
-    },
-    {
-      "epoch": 1.39,
-      "learning_rate": 1.9900782115459503e-05,
-      "loss": 0.5445,
-      "step": 783
-    },
-    {
-      "epoch": 1.39,
-      "learning_rate": 1.990023105908573e-05,
-      "loss": 0.5753,
-      "step": 784
-    },
-    {
-      "epoch": 1.39,
-      "learning_rate": 1.989967848433447e-05,
-      "loss": 0.5317,
-      "step": 785
-    },
-    {
-      "epoch": 1.4,
-      "learning_rate": 1.9899124391290478e-05,
-      "loss": 0.5382,
-      "step": 786
-    },
-    {
-      "epoch": 1.4,
-      "learning_rate": 1.9898568780038725e-05,
-      "loss": 0.5176,
-      "step": 787
-    },
-    {
-      "epoch": 1.4,
-      "learning_rate": 1.989801165066443e-05,
-      "loss": 0.5586,
-      "step": 788
-    },
-    {
-      "epoch": 1.4,
-      "learning_rate": 1.9897453003253035e-05,
-      "loss": 0.5409,
-      "step": 789
-    },
-    {
-      "epoch": 1.4,
-      "learning_rate": 1.989689283789022e-05,
-      "loss": 0.5384,
-      "step": 790
-    },
-    {
-      "epoch": 1.4,
-      "learning_rate": 1.9896331154661897e-05,
-      "loss": 0.5671,
-      "step": 791
-    },
-    {
-      "epoch": 1.41,
-      "learning_rate": 1.989576795365421e-05,
-      "loss": 0.5261,
-      "step": 792
-    },
-    {
-      "epoch": 1.41,
-      "learning_rate": 1.9895203234953533e-05,
-      "loss": 0.5591,
-      "step": 793
-    },
-    {
-      "epoch": 1.41,
-      "learning_rate": 1.989463699864648e-05,
-      "loss": 0.5226,
-      "step": 794
-    },
-    {
-      "epoch": 1.41,
-      "learning_rate": 1.9894069244819893e-05,
-      "loss": 0.5771,
-      "step": 795
-    },
-    {
-      "epoch": 1.41,
-      "learning_rate": 1.989349997356084e-05,
-      "loss": 0.548,
-      "step": 796
-    },
-    {
-      "epoch": 1.42,
-      "learning_rate": 1.989292918495664e-05,
-      "loss": 0.5622,
-      "step": 797
-    },
-    {
-      "epoch": 1.42,
-      "learning_rate": 1.9892356879094827e-05,
-      "loss": 0.5368,
-      "step": 798
-    },
-    {
-      "epoch": 1.42,
-      "learning_rate": 1.9891783056063178e-05,
-      "loss": 0.5706,
-      "step": 799
-    },
-    {
-      "epoch": 1.42,
-      "learning_rate": 1.9891207715949692e-05,
-      "loss": 0.5297,
-      "step": 800
-    },
-    {
-      "epoch": 1.42,
-      "learning_rate": 1.9890630858842614e-05,
-      "loss": 0.5603,
-      "step": 801
-    },
-    {
-      "epoch": 1.42,
-      "learning_rate": 1.9890052484830417e-05,
-      "loss": 0.5489,
-      "step": 802
-    },
-    {
-      "epoch": 1.43,
-      "learning_rate": 1.9889472594001797e-05,
-      "loss": 0.5609,
-      "step": 803
-    },
-    {
-      "epoch": 1.43,
-      "learning_rate": 1.9888891186445698e-05,
-      "loss": 0.5347,
-      "step": 804
-    },
-    {
-      "epoch": 1.43,
-      "learning_rate": 1.9888308262251286e-05,
-      "loss": 0.5424,
-      "step": 805
-    },
-    {
-      "epoch": 1.43,
-      "learning_rate": 1.9887723821507964e-05,
-      "loss": 0.5368,
-      "step": 806
-    },
-    {
-      "epoch": 1.43,
-      "learning_rate": 1.9887137864305365e-05,
-      "loss": 0.5808,
-      "step": 807
-    },
-    {
-      "epoch": 1.43,
-      "learning_rate": 1.9886550390733358e-05,
-      "loss": 0.5494,
-      "step": 808
-    },
-    {
-      "epoch": 1.44,
-      "learning_rate": 1.9885961400882045e-05,
-      "loss": 0.56,
-      "step": 809
-    },
-    {
-      "epoch": 1.44,
-      "learning_rate": 1.988537089484175e-05,
-      "loss": 0.5708,
-      "step": 810
-    },
-    {
-      "epoch": 1.44,
-      "learning_rate": 1.9884778872703043e-05,
-      "loss": 0.5431,
-      "step": 811
-    },
-    {
-      "epoch": 1.44,
-      "learning_rate": 1.988418533455672e-05,
-      "loss": 0.5458,
-      "step": 812
-    },
-    {
-      "epoch": 1.44,
-      "learning_rate": 1.9883590280493815e-05,
-      "loss": 0.545,
-      "step": 813
-    },
-    {
-      "epoch": 1.45,
-      "learning_rate": 1.9882993710605582e-05,
-      "loss": 0.5182,
-      "step": 814
-    },
-    {
-      "epoch": 1.45,
-      "learning_rate": 1.9882395624983522e-05,
-      "loss": 0.5436,
-      "step": 815
-    },
-    {
-      "epoch": 1.45,
-      "learning_rate": 1.988179602371936e-05,
-      "loss": 0.5528,
-      "step": 816
-    },
-    {
-      "epoch": 1.45,
-      "learning_rate": 1.9881194906905053e-05,
-      "loss": 0.5451,
-      "step": 817
-    },
-    {
-      "epoch": 1.45,
-      "learning_rate": 1.98805922746328e-05,
-      "loss": 0.5435,
-      "step": 818
-    },
-    {
-      "epoch": 1.45,
-      "learning_rate": 1.9879988126995023e-05,
-      "loss": 0.5592,
-      "step": 819
-    },
-    {
-      "epoch": 1.46,
-      "learning_rate": 1.987938246408437e-05,
-      "loss": 0.5448,
-      "step": 820
-    },
-    {
-      "epoch": 1.46,
-      "learning_rate": 1.9878775285993744e-05,
-      "loss": 0.5685,
-      "step": 821
-    },
-    {
-      "epoch": 1.46,
-      "learning_rate": 1.9878166592816255e-05,
-      "loss": 0.5578,
-      "step": 822
-    },
-    {
-      "epoch": 1.46,
-      "learning_rate": 1.9877556384645265e-05,
-      "loss": 0.5477,
-      "step": 823
-    },
-    {
-      "epoch": 1.46,
-      "learning_rate": 1.9876944661574355e-05,
-      "loss": 0.5414,
-      "step": 824
-    },
-    {
-      "epoch": 1.46,
-      "learning_rate": 1.9876331423697345e-05,
-      "loss": 0.5594,
-      "step": 825
-    },
-    {
-      "epoch": 1.47,
-      "learning_rate": 1.987571667110829e-05,
-      "loss": 0.5477,
-      "step": 826
-    },
-    {
-      "epoch": 1.47,
-      "learning_rate": 1.987510040390147e-05,
-      "loss": 0.5373,
-      "step": 827
-    },
-    {
-      "epoch": 1.47,
-      "learning_rate": 1.9874482622171397e-05,
-      "loss": 0.5511,
-      "step": 828
-    },
-    {
-      "epoch": 1.47,
-      "learning_rate": 1.987386332601283e-05,
-      "loss": 0.5458,
-      "step": 829
-    },
-    {
-      "epoch": 1.47,
-      "learning_rate": 1.9873242515520738e-05,
-      "loss": 0.5557,
-      "step": 830
-    },
-    {
-      "epoch": 1.48,
-      "learning_rate": 1.9872620190790334e-05,
-      "loss": 0.5256,
-      "step": 831
-    },
-    {
-      "epoch": 1.48,
-      "learning_rate": 1.9871996351917068e-05,
-      "loss": 0.5621,
-      "step": 832
-    },
-    {
-      "epoch": 1.48,
-      "learning_rate": 1.9871370998996616e-05,
-      "loss": 0.5475,
-      "step": 833
-    },
-    {
-      "epoch": 1.48,
-      "learning_rate": 1.9870744132124885e-05,
-      "loss": 0.5237,
-      "step": 834
-    },
-    {
-      "epoch": 1.48,
-      "learning_rate": 1.9870115751398017e-05,
-      "loss": 0.5714,
-      "step": 835
-    },
-    {
-      "epoch": 1.48,
-      "learning_rate": 1.9869485856912383e-05,
-      "loss": 0.5568,
-      "step": 836
-    },
-    {
-      "epoch": 1.49,
-      "learning_rate": 1.9868854448764594e-05,
-      "loss": 0.5658,
-      "step": 837
-    },
-    {
-      "epoch": 1.49,
-      "learning_rate": 1.9868221527051484e-05,
-      "loss": 0.5283,
-      "step": 838
-    },
-    {
-      "epoch": 1.49,
-      "learning_rate": 1.9867587091870126e-05,
-      "loss": 0.5403,
-      "step": 839
-    },
-    {
-      "epoch": 1.49,
-      "learning_rate": 1.9866951143317813e-05,
-      "loss": 0.5475,
-      "step": 840
-    },
-    {
-      "epoch": 1.49,
-      "learning_rate": 1.986631368149209e-05,
-      "loss": 0.5524,
-      "step": 841
-    },
-    {
-      "epoch": 1.49,
-      "learning_rate": 1.9865674706490717e-05,
-      "loss": 0.5608,
-      "step": 842
-    },
-    {
-      "epoch": 1.5,
-      "learning_rate": 1.9865034218411698e-05,
-      "loss": 0.5653,
-      "step": 843
-    },
-    {
-      "epoch": 1.5,
-      "learning_rate": 1.9864392217353256e-05,
-      "loss": 0.5537,
-      "step": 844
-    },
-    {
-      "epoch": 1.5,
-      "learning_rate": 1.9863748703413858e-05,
-      "loss": 0.5429,
-      "step": 845
-    },
-    {
-      "epoch": 1.5,
-      "learning_rate": 1.9863103676692197e-05,
-      "loss": 0.5359,
-      "step": 846
-    },
-    {
-      "epoch": 1.5,
-      "learning_rate": 1.9862457137287202e-05,
-      "loss": 0.5241,
-      "step": 847
-    },
-    {
-      "epoch": 1.51,
-      "learning_rate": 1.9861809085298022e-05,
-      "loss": 0.5236,
-      "step": 848
-    },
-    {
-      "epoch": 1.51,
-      "learning_rate": 1.986115952082406e-05,
-      "loss": 0.5588,
-      "step": 849
-    },
-    {
-      "epoch": 1.51,
-      "learning_rate": 1.986050844396493e-05,
-      "loss": 0.5342,
-      "step": 850
-    },
-    {
-      "epoch": 1.51,
-      "learning_rate": 1.985985585482049e-05,
-      "loss": 0.5404,
-      "step": 851
-    },
-    {
-      "epoch": 1.51,
-      "learning_rate": 1.9859201753490824e-05,
-      "loss": 0.567,
-      "step": 852
-    },
-    {
-      "epoch": 1.51,
-      "learning_rate": 1.985854614007625e-05,
-      "loss": 0.5653,
-      "step": 853
-    },
-    {
-      "epoch": 1.52,
-      "learning_rate": 1.9857889014677324e-05,
-      "loss": 0.5704,
-      "step": 854
-    },
-    {
-      "epoch": 1.52,
-      "learning_rate": 1.985723037739482e-05,
-      "loss": 0.5175,
-      "step": 855
-    },
-    {
-      "epoch": 1.52,
-      "learning_rate": 1.9856570228329756e-05,
-      "loss": 0.5447,
-      "step": 856
-    },
-    {
-      "epoch": 1.52,
-      "learning_rate": 1.9855908567583374e-05,
-      "loss": 0.516,
-      "step": 857
-    },
-    {
-      "epoch": 1.52,
-      "learning_rate": 1.9855245395257156e-05,
-      "loss": 0.5243,
-      "step": 858
-    },
-    {
-      "epoch": 1.53,
-      "learning_rate": 1.9854580711452808e-05,
-      "loss": 0.5383,
-      "step": 859
-    },
-    {
-      "epoch": 1.53,
-      "learning_rate": 1.9853914516272275e-05,
-      "loss": 0.5581,
-      "step": 860
-    },
-    {
-      "epoch": 1.53,
-      "learning_rate": 1.9853246809817725e-05,
-      "loss": 0.5591,
-      "step": 861
-    },
-    {
-      "epoch": 1.53,
-      "learning_rate": 1.985257759219157e-05,
-      "loss": 0.5448,
-      "step": 862
-    },
-    {
-      "epoch": 1.53,
-      "learning_rate": 1.9851906863496434e-05,
-      "loss": 0.5651,
-      "step": 863
-    },
-    {
-      "epoch": 1.53,
-      "learning_rate": 1.98512346238352e-05,
-      "loss": 0.5672,
-      "step": 864
-    },
-    {
-      "epoch": 1.54,
-      "learning_rate": 1.9850560873310955e-05,
-      "loss": 0.5604,
-      "step": 865
-    },
-    {
-      "epoch": 1.54,
-      "learning_rate": 1.984988561202704e-05,
-      "loss": 0.5461,
-      "step": 866
-    },
-    {
-      "epoch": 1.54,
-      "learning_rate": 1.9849208840087018e-05,
-      "loss": 0.5668,
-      "step": 867
-    },
-    {
-      "epoch": 1.54,
-      "learning_rate": 1.9848530557594678e-05,
-      "loss": 0.543,
-      "step": 868
-    },
-    {
-      "epoch": 1.54,
-      "learning_rate": 1.984785076465405e-05,
-      "loss": 0.559,
-      "step": 869
-    },
-    {
-      "epoch": 1.54,
-      "learning_rate": 1.9847169461369394e-05,
-      "loss": 0.5556,
-      "step": 870
-    },
-    {
-      "epoch": 1.55,
-      "learning_rate": 1.9846486647845197e-05,
-      "loss": 0.5606,
-      "step": 871
-    },
-    {
-      "epoch": 1.55,
-      "learning_rate": 1.984580232418618e-05,
-      "loss": 0.5434,
-      "step": 872
-    },
-    {
-      "epoch": 1.55,
-      "learning_rate": 1.98451164904973e-05,
-      "loss": 0.5579,
-      "step": 873
-    },
-    {
-      "epoch": 1.55,
-      "learning_rate": 1.984442914688374e-05,
-      "loss": 0.5425,
-      "step": 874
-    },
-    {
-      "epoch": 1.55,
-      "learning_rate": 1.9843740293450916e-05,
-      "loss": 0.5637,
-      "step": 875
-    },
-    {
-      "epoch": 1.56,
-      "learning_rate": 1.9843049930304475e-05,
-      "loss": 0.5448,
-      "step": 876
-    },
-    {
-      "epoch": 1.56,
-      "learning_rate": 1.9842358057550305e-05,
-      "loss": 0.5363,
-      "step": 877
-    },
-    {
-      "epoch": 1.56,
-      "learning_rate": 1.9841664675294505e-05,
-      "loss": 0.5426,
-      "step": 878
-    },
-    {
-      "epoch": 1.56,
-      "learning_rate": 1.984096978364342e-05,
-      "loss": 0.5603,
-      "step": 879
-    },
-    {
-      "epoch": 1.56,
-      "learning_rate": 1.984027338270363e-05,
-      "loss": 0.53,
-      "step": 880
-    },
-    {
-      "epoch": 1.56,
-      "learning_rate": 1.983957547258194e-05,
-      "loss": 0.5656,
-      "step": 881
-    },
-    {
-      "epoch": 1.57,
-      "learning_rate": 1.983887605338538e-05,
-      "loss": 0.541,
-      "step": 882
-    },
-    {
-      "epoch": 1.57,
-      "learning_rate": 1.9838175125221225e-05,
-      "loss": 0.5296,
-      "step": 883
-    },
-    {
-      "epoch": 1.57,
-      "learning_rate": 1.983747268819697e-05,
-      "loss": 0.5287,
-      "step": 884
-    },
-    {
-      "epoch": 1.57,
-      "learning_rate": 1.9836768742420355e-05,
-      "loss": 0.5292,
-      "step": 885
-    },
-    {
-      "epoch": 1.57,
-      "learning_rate": 1.983606328799933e-05,
-      "loss": 0.5441,
-      "step": 886
-    },
-    {
-      "epoch": 1.57,
-      "learning_rate": 1.98353563250421e-05,
-      "loss": 0.5354,
-      "step": 887
-    },
-    {
-      "epoch": 1.58,
-      "learning_rate": 1.9834647853657084e-05,
-      "loss": 0.5522,
-      "step": 888
-    },
-    {
-      "epoch": 1.58,
-      "learning_rate": 1.9833937873952943e-05,
-      "loss": 0.5564,
-      "step": 889
-    },
-    {
-      "epoch": 1.58,
-      "learning_rate": 1.983322638603856e-05,
-      "loss": 0.5308,
-      "step": 890
-    },
-    {
-      "epoch": 1.58,
-      "learning_rate": 1.983251339002306e-05,
-      "loss": 0.5516,
-      "step": 891
-    },
-    {
-      "epoch": 1.58,
-      "learning_rate": 1.983179888601579e-05,
-      "loss": 0.5396,
-      "step": 892
-    },
-    {
-      "epoch": 1.59,
-      "learning_rate": 1.983108287412633e-05,
-      "loss": 0.5539,
-      "step": 893
-    },
-    {
-      "epoch": 1.59,
-      "learning_rate": 1.9830365354464498e-05,
-      "loss": 0.5192,
-      "step": 894
-    },
-    {
-      "epoch": 1.59,
-      "learning_rate": 1.9829646327140336e-05,
-      "loss": 0.5398,
-      "step": 895
-    },
-    {
-      "epoch": 1.59,
-      "learning_rate": 1.9828925792264123e-05,
-      "loss": 0.5156,
-      "step": 896
-    },
-    {
-      "epoch": 1.59,
-      "learning_rate": 1.9828203749946362e-05,
-      "loss": 0.5512,
-      "step": 897
-    },
-    {
-      "epoch": 1.59,
-      "learning_rate": 1.9827480200297794e-05,
-      "loss": 0.5675,
-      "step": 898
-    },
-    {
-      "epoch": 1.6,
-      "learning_rate": 1.982675514342938e-05,
-      "loss": 0.5537,
-      "step": 899
-    },
-    {
-      "epoch": 1.6,
-      "learning_rate": 1.9826028579452335e-05,
-      "loss": 0.5445,
-      "step": 900
-    },
-    {
-      "epoch": 1.6,
-      "learning_rate": 1.982530050847808e-05,
-      "loss": 0.5714,
-      "step": 901
-    },
-    {
-      "epoch": 1.6,
-      "learning_rate": 1.9824570930618278e-05,
-      "loss": 0.5377,
-      "step": 902
-    },
-    {
-      "epoch": 1.6,
-      "learning_rate": 1.982383984598483e-05,
-      "loss": 0.5446,
-      "step": 903
-    },
-    {
-      "epoch": 1.6,
-      "learning_rate": 1.982310725468985e-05,
-      "loss": 0.5656,
-      "step": 904
-    },
-    {
-      "epoch": 1.61,
-      "learning_rate": 1.9822373156845705e-05,
-      "loss": 0.5434,
-      "step": 905
-    },
-    {
-      "epoch": 1.61,
-      "learning_rate": 1.9821637552564973e-05,
-      "loss": 0.5627,
-      "step": 906
-    },
-    {
-      "epoch": 1.61,
-      "learning_rate": 1.982090044196048e-05,
-      "loss": 0.5428,
-      "step": 907
-    },
-    {
-      "epoch": 1.61,
-      "learning_rate": 1.982016182514527e-05,
-      "loss": 0.5471,
-      "step": 908
-    },
-    {
-      "epoch": 1.61,
-      "learning_rate": 1.9819421702232624e-05,
-      "loss": 0.5395,
-      "step": 909
-    },
-    {
-      "epoch": 1.62,
-      "learning_rate": 1.9818680073336054e-05,
-      "loss": 0.5527,
-      "step": 910
-    },
-    {
-      "epoch": 1.62,
-      "learning_rate": 1.98179369385693e-05,
-      "loss": 0.5515,
-      "step": 911
-    },
-    {
-      "epoch": 1.62,
-      "learning_rate": 1.9817192298046336e-05,
-      "loss": 0.5406,
-      "step": 912
-    },
-    {
-      "epoch": 1.62,
-      "learning_rate": 1.9816446151881367e-05,
-      "loss": 0.5498,
-      "step": 913
-    },
-    {
-      "epoch": 1.62,
-      "learning_rate": 1.981569850018883e-05,
-      "loss": 0.516,
-      "step": 914
-    },
-    {
-      "epoch": 1.62,
-      "learning_rate": 1.9814949343083383e-05,
-      "loss": 0.5496,
-      "step": 915
-    },
-    {
-      "epoch": 1.63,
-      "learning_rate": 1.9814198680679934e-05,
-      "loss": 0.5635,
-      "step": 916
-    },
-    {
-      "epoch": 1.63,
-      "learning_rate": 1.98134465130936e-05,
-      "loss": 0.5736,
-      "step": 917
-    },
-    {
-      "epoch": 1.63,
-      "learning_rate": 1.9812692840439743e-05,
-      "loss": 0.5527,
-      "step": 918
-    },
-    {
-      "epoch": 1.63,
-      "learning_rate": 1.9811937662833958e-05,
-      "loss": 0.547,
-      "step": 919
-    },
-    {
-      "epoch": 1.63,
-      "learning_rate": 1.9811180980392054e-05,
-      "loss": 0.5592,
-      "step": 920
-    },
-    {
-      "epoch": 1.64,
-      "learning_rate": 1.981042279323009e-05,
-      "loss": 0.5452,
-      "step": 921
-    },
-    {
-      "epoch": 1.64,
-      "learning_rate": 1.9809663101464345e-05,
-      "loss": 0.5391,
-      "step": 922
-    },
-    {
-      "epoch": 1.64,
-      "learning_rate": 1.9808901905211333e-05,
-      "loss": 0.5437,
-      "step": 923
-    },
-    {
-      "epoch": 1.64,
-      "learning_rate": 1.9808139204587792e-05,
-      "loss": 0.5477,
-      "step": 924
-    },
-    {
-      "epoch": 1.64,
-      "learning_rate": 1.98073749997107e-05,
-      "loss": 0.5634,
-      "step": 925
-    },
-    {
-      "epoch": 1.64,
-      "learning_rate": 1.980660929069726e-05,
-      "loss": 0.5299,
-      "step": 926
-    },
-    {
-      "epoch": 1.65,
-      "learning_rate": 1.9805842077664913e-05,
-      "loss": 0.547,
-      "step": 927
-    },
-    {
-      "epoch": 1.65,
-      "learning_rate": 1.9805073360731316e-05,
-      "loss": 0.5399,
-      "step": 928
-    },
-    {
-      "epoch": 1.65,
-      "learning_rate": 1.9804303140014368e-05,
-      "loss": 0.5491,
-      "step": 929
-    },
-    {
-      "epoch": 1.65,
-      "learning_rate": 1.98035314156322e-05,
-      "loss": 0.5293,
-      "step": 930
-    },
-    {
-      "epoch": 1.65,
-      "learning_rate": 1.9802758187703163e-05,
-      "loss": 0.5506,
-      "step": 931
-    },
-    {
-      "epoch": 1.65,
-      "learning_rate": 1.9801983456345852e-05,
-      "loss": 0.5358,
-      "step": 932
-    },
-    {
-      "epoch": 1.66,
-      "learning_rate": 1.9801207221679085e-05,
-      "loss": 0.5515,
-      "step": 933
-    },
-    {
-      "epoch": 1.66,
-      "learning_rate": 1.980042948382191e-05,
-      "loss": 0.5226,
-      "step": 934
-    },
-    {
-      "epoch": 1.66,
-      "learning_rate": 1.9799650242893602e-05,
-      "loss": 0.5393,
-      "step": 935
-    },
-    {
-      "epoch": 1.66,
-      "learning_rate": 1.979886949901368e-05,
-      "loss": 0.5271,
-      "step": 936
-    },
-    {
-      "epoch": 1.66,
-      "learning_rate": 1.979808725230188e-05,
-      "loss": 0.5485,
-      "step": 937
-    },
-    {
-      "epoch": 1.67,
-      "learning_rate": 1.9797303502878174e-05,
-      "loss": 0.5521,
-      "step": 938
-    },
-    {
-      "epoch": 1.67,
-      "learning_rate": 1.979651825086277e-05,
-      "loss": 0.5267,
-      "step": 939
-    },
-    {
-      "epoch": 1.67,
-      "learning_rate": 1.979573149637609e-05,
-      "loss": 0.5548,
-      "step": 940
-    },
-    {
-      "epoch": 1.67,
-      "learning_rate": 1.9794943239538803e-05,
-      "loss": 0.5325,
-      "step": 941
-    },
-    {
-      "epoch": 1.67,
-      "learning_rate": 1.97941534804718e-05,
-      "loss": 0.5117,
-      "step": 942
-    },
-    {
-      "epoch": 1.67,
-      "learning_rate": 1.9793362219296212e-05,
-      "loss": 0.523,
-      "step": 943
-    },
-    {
-      "epoch": 1.68,
-      "learning_rate": 1.979256945613338e-05,
-      "loss": 0.5686,
-      "step": 944
-    },
-    {
-      "epoch": 1.68,
-      "learning_rate": 1.97917751911049e-05,
-      "loss": 0.5316,
-      "step": 945
-    },
-    {
-      "epoch": 1.68,
-      "learning_rate": 1.9790979424332583e-05,
-      "loss": 0.5402,
-      "step": 946
-    },
-    {
-      "epoch": 1.68,
-      "learning_rate": 1.9790182155938467e-05,
-      "loss": 0.5488,
-      "step": 947
-    },
-    {
-      "epoch": 1.68,
-      "learning_rate": 1.978938338604484e-05,
-      "loss": 0.535,
-      "step": 948
-    },
-    {
-      "epoch": 1.68,
-      "learning_rate": 1.9788583114774203e-05,
-      "loss": 0.5597,
-      "step": 949
-    },
-    {
-      "epoch": 1.69,
-      "learning_rate": 1.9787781342249285e-05,
-      "loss": 0.5322,
-      "step": 950
-    },
-    {
-      "epoch": 1.69,
-      "learning_rate": 1.9786978068593062e-05,
-      "loss": 0.5417,
-      "step": 951
-    },
-    {
-      "epoch": 1.69,
-      "learning_rate": 1.978617329392872e-05,
-      "loss": 0.5581,
-      "step": 952
-    },
-    {
-      "epoch": 1.69,
-      "learning_rate": 1.97853670183797e-05,
-      "loss": 0.5318,
-      "step": 953
-    },
-    {
-      "epoch": 1.69,
-      "learning_rate": 1.9784559242069644e-05,
-      "loss": 0.5513,
-      "step": 954
-    },
-    {
-      "epoch": 1.7,
-      "learning_rate": 1.9783749965122444e-05,
-      "loss": 0.5463,
-      "step": 955
-    },
-    {
-      "epoch": 1.7,
-      "learning_rate": 1.9782939187662223e-05,
-      "loss": 0.5254,
-      "step": 956
-    },
-    {
-      "epoch": 1.7,
-      "learning_rate": 1.978212690981332e-05,
-      "loss": 0.5798,
-      "step": 957
-    },
-    {
-      "epoch": 1.7,
-      "learning_rate": 1.9781313131700317e-05,
-      "loss": 0.55,
-      "step": 958
-    },
-    {
-      "epoch": 1.7,
-      "learning_rate": 1.978049785344802e-05,
-      "loss": 0.5318,
-      "step": 959
-    },
-    {
-      "epoch": 1.7,
-      "learning_rate": 1.9779681075181466e-05,
-      "loss": 0.5186,
-      "step": 960
-    },
-    {
-      "epoch": 1.71,
-      "learning_rate": 1.9778862797025925e-05,
-      "loss": 0.5378,
-      "step": 961
-    },
-    {
-      "epoch": 1.71,
-      "learning_rate": 1.9778043019106895e-05,
-      "loss": 0.5374,
-      "step": 962
-    },
-    {
-      "epoch": 1.71,
-      "learning_rate": 1.9777221741550096e-05,
-      "loss": 0.5418,
-      "step": 963
-    },
-    {
-      "epoch": 1.71,
-      "learning_rate": 1.9776398964481493e-05,
-      "loss": 0.5583,
-      "step": 964
-    },
-    {
-      "epoch": 1.71,
-      "learning_rate": 1.9775574688027275e-05,
-      "loss": 0.525,
-      "step": 965
-    },
-    {
-      "epoch": 1.72,
-      "learning_rate": 1.977474891231385e-05,
-      "loss": 0.5455,
-      "step": 966
-    },
-    {
-      "epoch": 1.72,
-      "learning_rate": 1.9773921637467877e-05,
-      "loss": 0.5268,
-      "step": 967
-    },
-    {
-      "epoch": 1.72,
-      "learning_rate": 1.9773092863616226e-05,
-      "loss": 0.5241,
-      "step": 968
-    },
-    {
-      "epoch": 1.72,
-      "learning_rate": 1.9772262590886006e-05,
-      "loss": 0.546,
-      "step": 969
-    },
-    {
-      "epoch": 1.72,
-      "learning_rate": 1.977143081940456e-05,
-      "loss": 0.55,
-      "step": 970
-    },
-    {
-      "epoch": 1.72,
-      "learning_rate": 1.9770597549299442e-05,
-      "loss": 0.5309,
-      "step": 971
-    },
-    {
-      "epoch": 1.73,
-      "learning_rate": 1.9769762780698458e-05,
-      "loss": 0.5306,
-      "step": 972
-    },
-    {
-      "epoch": 1.73,
-      "learning_rate": 1.9768926513729637e-05,
-      "loss": 0.518,
-      "step": 973
-    },
-    {
-      "epoch": 1.73,
-      "learning_rate": 1.9768088748521225e-05,
-      "loss": 0.5366,
-      "step": 974
-    },
-    {
-      "epoch": 1.73,
-      "learning_rate": 1.976724948520172e-05,
-      "loss": 0.5363,
-      "step": 975
-    },
-    {
-      "epoch": 1.73,
-      "learning_rate": 1.976640872389983e-05,
-      "loss": 0.529,
-      "step": 976
-    },
-    {
-      "epoch": 1.73,
-      "learning_rate": 1.9765566464744508e-05,
-      "loss": 0.5572,
-      "step": 977
-    },
-    {
-      "epoch": 1.74,
-      "learning_rate": 1.9764722707864923e-05,
-      "loss": 0.5454,
-      "step": 978
-    },
-    {
-      "epoch": 1.74,
-      "learning_rate": 1.9763877453390485e-05,
-      "loss": 0.5694,
-      "step": 979
-    },
-    {
-      "epoch": 1.74,
-      "learning_rate": 1.9763030701450822e-05,
-      "loss": 0.5484,
-      "step": 980
-    },
-    {
-      "epoch": 1.74,
-      "learning_rate": 1.9762182452175806e-05,
-      "loss": 0.5332,
-      "step": 981
-    },
-    {
-      "epoch": 1.74,
-      "learning_rate": 1.9761332705695523e-05,
-      "loss": 0.5655,
-      "step": 982
-    },
-    {
-      "epoch": 1.75,
-      "learning_rate": 1.9760481462140305e-05,
-      "loss": 0.5548,
-      "step": 983
-    },
-    {
-      "epoch": 1.75,
-      "learning_rate": 1.9759628721640703e-05,
-      "loss": 0.5458,
-      "step": 984
-    },
-    {
-      "epoch": 1.75,
-      "learning_rate": 1.97587744843275e-05,
-      "loss": 0.5459,
-      "step": 985
-    },
-    {
-      "epoch": 1.75,
-      "learning_rate": 1.9757918750331707e-05,
-      "loss": 0.5499,
-      "step": 986
-    },
-    {
-      "epoch": 1.75,
-      "learning_rate": 1.9757061519784568e-05,
-      "loss": 0.5365,
-      "step": 987
-    },
-    {
-      "epoch": 1.75,
-      "learning_rate": 1.975620279281755e-05,
-      "loss": 0.5421,
-      "step": 988
-    },
-    {
-      "epoch": 1.76,
-      "learning_rate": 1.975534256956236e-05,
-      "loss": 0.5107,
-      "step": 989
-    },
-    {
-      "epoch": 1.76,
-      "learning_rate": 1.975448085015093e-05,
-      "loss": 0.5203,
-      "step": 990
-    },
-    {
-      "epoch": 1.76,
-      "learning_rate": 1.975361763471541e-05,
-      "loss": 0.5419,
-      "step": 991
-    },
-    {
-      "epoch": 1.76,
-      "learning_rate": 1.97527529233882e-05,
-      "loss": 0.5411,
-      "step": 992
-    },
-    {
-      "epoch": 1.76,
-      "learning_rate": 1.9751886716301914e-05,
-      "loss": 0.5443,
-      "step": 993
-    },
-    {
-      "epoch": 1.76,
-      "learning_rate": 1.9751019013589403e-05,
-      "loss": 0.5383,
-      "step": 994
-    },
-    {
-      "epoch": 1.77,
-      "learning_rate": 1.975014981538374e-05,
-      "loss": 0.5259,
-      "step": 995
-    },
-    {
-      "epoch": 1.77,
-      "learning_rate": 1.9749279121818235e-05,
-      "loss": 0.5176,
-      "step": 996
-    },
-    {
-      "epoch": 1.77,
-      "learning_rate": 1.974840693302643e-05,
-      "loss": 0.5359,
-      "step": 997
-    },
-    {
-      "epoch": 1.77,
-      "learning_rate": 1.974753324914208e-05,
-      "loss": 0.5858,
-      "step": 998
-    },
-    {
-      "epoch": 1.77,
-      "learning_rate": 1.9746658070299188e-05,
-      "loss": 0.5611,
-      "step": 999
-    },
-    {
-      "epoch": 1.78,
-      "learning_rate": 1.9745781396631974e-05,
-      "loss": 0.528,
-      "step": 1000
-    },
-    {
-      "epoch": 1.78,
-      "learning_rate": 1.974490322827489e-05,
-      "loss": 0.5337,
-      "step": 1001
-    },
-    {
-      "epoch": 1.78,
-      "learning_rate": 1.974402356536263e-05,
-      "loss": 0.5418,
-      "step": 1002
-    },
-    {
-      "epoch": 1.78,
-      "learning_rate": 1.9743142408030095e-05,
-      "loss": 0.5418,
-      "step": 1003
-    },
-    {
-      "epoch": 1.78,
-      "learning_rate": 1.9742259756412426e-05,
-      "loss": 0.515,
-      "step": 1004
-    },
-    {
-      "epoch": 1.78,
-      "learning_rate": 1.9741375610645e-05,
-      "loss": 0.5478,
-      "step": 1005
-    },
-    {
-      "epoch": 1.79,
-      "learning_rate": 1.974048997086341e-05,
-      "loss": 0.5475,
-      "step": 1006
-    },
-    {
-      "epoch": 1.79,
-      "learning_rate": 1.973960283720349e-05,
-      "loss": 0.5368,
-      "step": 1007
-    },
-    {
-      "epoch": 1.79,
-      "learning_rate": 1.97387142098013e-05,
-      "loss": 0.5399,
-      "step": 1008
-    },
-    {
-      "epoch": 1.79,
-      "learning_rate": 1.9737824088793117e-05,
-      "loss": 0.5464,
-      "step": 1009
-    },
-    {
-      "epoch": 1.79,
-      "learning_rate": 1.9736932474315464e-05,
-      "loss": 0.5153,
-      "step": 1010
-    },
-    {
-      "epoch": 1.79,
-      "learning_rate": 1.9736039366505087e-05,
-      "loss": 0.543,
-      "step": 1011
-    },
-    {
-      "epoch": 1.8,
-      "learning_rate": 1.9735144765498956e-05,
-      "loss": 0.523,
-      "step": 1012
-    },
-    {
-      "epoch": 1.8,
-      "learning_rate": 1.9734248671434275e-05,
-      "loss": 0.5481,
-      "step": 1013
-    },
-    {
-      "epoch": 1.8,
-      "learning_rate": 1.973335108444848e-05,
-      "loss": 0.5317,
-      "step": 1014
-    },
-    {
-      "epoch": 1.8,
-      "learning_rate": 1.9732452004679223e-05,
-      "loss": 0.5266,
-      "step": 1015
-    },
-    {
-      "epoch": 1.8,
-      "learning_rate": 1.9731551432264405e-05,
-      "loss": 0.566,
-      "step": 1016
-    },
-    {
-      "epoch": 1.81,
-      "learning_rate": 1.9730649367342137e-05,
-      "loss": 0.5417,
-      "step": 1017
-    },
-    {
-      "epoch": 1.81,
-      "learning_rate": 1.9729745810050772e-05,
-      "loss": 0.5206,
-      "step": 1018
-    },
-    {
-      "epoch": 1.81,
-      "learning_rate": 1.9728840760528884e-05,
-      "loss": 0.5432,
-      "step": 1019
-    },
-    {
-      "epoch": 1.81,
-      "learning_rate": 1.972793421891528e-05,
-      "loss": 0.5337,
-      "step": 1020
-    },
-    {
-      "epoch": 1.81,
-      "learning_rate": 1.972702618534899e-05,
-      "loss": 0.5596,
-      "step": 1021
-    },
-    {
-      "epoch": 1.81,
-      "learning_rate": 1.972611665996928e-05,
-      "loss": 0.548,
-      "step": 1022
-    },
-    {
-      "epoch": 1.82,
-      "learning_rate": 1.9725205642915645e-05,
-      "loss": 0.5447,
-      "step": 1023
-    },
-    {
-      "epoch": 1.82,
-      "learning_rate": 1.9724293134327802e-05,
-      "loss": 0.5471,
-      "step": 1024
-    },
-    {
-      "epoch": 1.82,
-      "learning_rate": 1.9723379134345698e-05,
-      "loss": 0.5405,
-      "step": 1025
-    },
-    {
-      "epoch": 1.82,
-      "learning_rate": 1.972246364310952e-05,
-      "loss": 0.5572,
-      "step": 1026
-    },
-    {
-      "epoch": 1.82,
-      "learning_rate": 1.972154666075967e-05,
-      "loss": 0.5469,
-      "step": 1027
-    },
-    {
-      "epoch": 1.83,
-      "learning_rate": 1.9720628187436782e-05,
-      "loss": 0.5514,
-      "step": 1028
-    },
-    {
-      "epoch": 1.83,
-      "learning_rate": 1.971970822328172e-05,
-      "loss": 0.5554,
-      "step": 1029
-    },
-    {
-      "epoch": 1.83,
-      "learning_rate": 1.971878676843558e-05,
-      "loss": 0.5643,
-      "step": 1030
-    },
-    {
-      "epoch": 1.83,
-      "learning_rate": 1.971786382303968e-05,
-      "loss": 0.5351,
-      "step": 1031
-    },
-    {
-      "epoch": 1.83,
-      "learning_rate": 1.9716939387235573e-05,
-      "loss": 0.5321,
-      "step": 1032
-    },
-    {
-      "epoch": 1.83,
-      "learning_rate": 1.9716013461165043e-05,
-      "loss": 0.5278,
-      "step": 1033
-    },
-    {
-      "epoch": 1.84,
-      "learning_rate": 1.9715086044970086e-05,
-      "loss": 0.528,
-      "step": 1034
-    },
-    {
-      "epoch": 1.84,
-      "learning_rate": 1.9714157138792946e-05,
-      "loss": 0.5151,
-      "step": 1035
-    },
-    {
-      "epoch": 1.84,
-      "learning_rate": 1.971322674277608e-05,
-      "loss": 0.5226,
-      "step": 1036
-    },
-    {
-      "epoch": 1.84,
-      "learning_rate": 1.971229485706219e-05,
-      "loss": 0.5508,
-      "step": 1037
-    },
-    {
-      "epoch": 1.84,
-      "learning_rate": 1.9711361481794193e-05,
-      "loss": 0.5013,
-      "step": 1038
-    },
-    {
-      "epoch": 1.84,
-      "learning_rate": 1.9710426617115237e-05,
-      "loss": 0.512,
-      "step": 1039
-    },
-    {
-      "epoch": 1.85,
-      "learning_rate": 1.9709490263168703e-05,
-      "loss": 0.5602,
-      "step": 1040
-    },
-    {
-      "epoch": 1.85,
-      "learning_rate": 1.9708552420098196e-05,
-      "loss": 0.5221,
-      "step": 1041
-    },
-    {
-      "epoch": 1.85,
-      "learning_rate": 1.970761308804755e-05,
-      "loss": 0.5488,
-      "step": 1042
-    },
-    {
-      "epoch": 1.85,
-      "learning_rate": 1.970667226716083e-05,
-      "loss": 0.5177,
-      "step": 1043
-    },
-    {
-      "epoch": 1.85,
-      "learning_rate": 1.970572995758233e-05,
-      "loss": 0.5394,
-      "step": 1044
-    },
-    {
-      "epoch": 1.86,
-      "learning_rate": 1.9704786159456568e-05,
-      "loss": 0.5637,
-      "step": 1045
-    },
-    {
-      "epoch": 1.86,
-      "learning_rate": 1.970384087292829e-05,
-      "loss": 0.5218,
-      "step": 1046
-    },
-    {
-      "epoch": 1.86,
-      "learning_rate": 1.9702894098142476e-05,
-      "loss": 0.5369,
-      "step": 1047
-    },
-    {
-      "epoch": 1.86,
-      "learning_rate": 1.970194583524433e-05,
-      "loss": 0.5327,
-      "step": 1048
-    },
-    {
-      "epoch": 1.86,
-      "learning_rate": 1.9700996084379283e-05,
-      "loss": 0.518,
-      "step": 1049
-    },
-    {
-      "epoch": 1.86,
-      "learning_rate": 1.9700044845692997e-05,
-      "loss": 0.5543,
-      "step": 1050
-    },
-    {
-      "epoch": 1.87,
-      "learning_rate": 1.9699092119331364e-05,
-      "loss": 0.5332,
-      "step": 1051
-    },
-    {
-      "epoch": 1.87,
-      "learning_rate": 1.9698137905440502e-05,
-      "loss": 0.5437,
-      "step": 1052
-    },
-    {
-      "epoch": 1.87,
-      "learning_rate": 1.969718220416675e-05,
-      "loss": 0.5337,
-      "step": 1053
-    },
-    {
-      "epoch": 1.87,
-      "learning_rate": 1.969622501565669e-05,
-      "loss": 0.5432,
-      "step": 1054
-    },
-    {
-      "epoch": 1.87,
-      "learning_rate": 1.969526634005712e-05,
-      "loss": 0.516,
-      "step": 1055
-    },
-    {
-      "epoch": 1.87,
-      "learning_rate": 1.9694306177515072e-05,
-      "loss": 0.5435,
-      "step": 1056
-    },
-    {
-      "epoch": 1.88,
-      "learning_rate": 1.96933445281778e-05,
-      "loss": 0.5517,
-      "step": 1057
-    },
-    {
-      "epoch": 1.88,
-      "learning_rate": 1.969238139219279e-05,
-      "loss": 0.5458,
-      "step": 1058
-    },
-    {
-      "epoch": 1.88,
-      "learning_rate": 1.9691416769707763e-05,
-      "loss": 0.5375,
-      "step": 1059
-    },
-    {
-      "epoch": 1.88,
-      "learning_rate": 1.9690450660870657e-05,
-      "loss": 0.5372,
-      "step": 1060
-    },
-    {
-      "epoch": 1.88,
-      "learning_rate": 1.968948306582964e-05,
-      "loss": 0.5231,
-      "step": 1061
-    },
-    {
-      "epoch": 1.89,
-      "learning_rate": 1.9688513984733113e-05,
-      "loss": 0.5431,
-      "step": 1062
-    },
-    {
-      "epoch": 1.89,
-      "learning_rate": 1.9687543417729702e-05,
-      "loss": 0.5388,
-      "step": 1063
-    },
-    {
-      "epoch": 1.89,
-      "learning_rate": 1.9686571364968258e-05,
-      "loss": 0.5543,
-      "step": 1064
-    },
-    {
-      "epoch": 1.89,
-      "learning_rate": 1.968559782659787e-05,
-      "loss": 0.5407,
-      "step": 1065
-    },
-    {
-      "epoch": 1.89,
-      "learning_rate": 1.9684622802767834e-05,
-      "loss": 0.5185,
-      "step": 1066
-    },
-    {
-      "epoch": 1.89,
-      "learning_rate": 1.96836462936277e-05,
-      "loss": 0.5389,
-      "step": 1067
-    },
-    {
-      "epoch": 1.9,
-      "learning_rate": 1.9682668299327227e-05,
-      "loss": 0.5284,
-      "step": 1068
-    },
-    {
-      "epoch": 1.9,
-      "learning_rate": 1.968168882001641e-05,
-      "loss": 0.5307,
-      "step": 1069
-    },
-    {
-      "epoch": 1.9,
-      "learning_rate": 1.968070785584547e-05,
-      "loss": 0.5258,
-      "step": 1070
-    },
-    {
-      "epoch": 1.9,
-      "learning_rate": 1.9679725406964853e-05,
-      "loss": 0.5502,
-      "step": 1071
-    },
-    {
-      "epoch": 1.9,
-      "learning_rate": 1.967874147352524e-05,
-      "loss": 0.5246,
-      "step": 1072
-    },
-    {
-      "epoch": 1.91,
-      "learning_rate": 1.9677756055677525e-05,
-      "loss": 0.5424,
-      "step": 1073
-    },
-    {
-      "epoch": 1.91,
-      "learning_rate": 1.9676769153572853e-05,
-      "loss": 0.5439,
-      "step": 1074
-    },
-    {
-      "epoch": 1.91,
-      "learning_rate": 1.967578076736257e-05,
-      "loss": 0.5328,
-      "step": 1075
-    },
-    {
-      "epoch": 1.91,
-      "learning_rate": 1.967479089719827e-05,
-      "loss": 0.5302,
-      "step": 1076
-    },
-    {
-      "epoch": 1.91,
-      "learning_rate": 1.967379954323177e-05,
-      "loss": 0.5187,
-      "step": 1077
-    },
-    {
-      "epoch": 1.91,
-      "learning_rate": 1.96728067056151e-05,
-      "loss": 0.5327,
-      "step": 1078
-    },
-    {
-      "epoch": 1.92,
-      "learning_rate": 1.9671812384500543e-05,
-      "loss": 0.5072,
-      "step": 1079
-    },
-    {
-      "epoch": 1.92,
-      "learning_rate": 1.9670816580040592e-05,
-      "loss": 0.5285,
-      "step": 1080
-    },
-    {
-      "epoch": 1.92,
-      "learning_rate": 1.9669819292387965e-05,
-      "loss": 0.5274,
-      "step": 1081
-    },
-    {
-      "epoch": 1.92,
-      "learning_rate": 1.966882052169562e-05,
-      "loss": 0.5154,
-      "step": 1082
-    },
-    {
-      "epoch": 1.92,
-      "learning_rate": 1.9667820268116732e-05,
-      "loss": 0.539,
-      "step": 1083
-    },
-    {
-      "epoch": 1.92,
-      "learning_rate": 1.9666818531804715e-05,
-      "loss": 0.5399,
-      "step": 1084
-    },
-    {
-      "epoch": 1.93,
-      "learning_rate": 1.9665815312913197e-05,
-      "loss": 0.5508,
-      "step": 1085
-    },
-    {
-      "epoch": 1.93,
-      "learning_rate": 1.9664810611596042e-05,
-      "loss": 0.5415,
-      "step": 1086
-    },
-    {
-      "epoch": 1.93,
-      "learning_rate": 1.9663804428007335e-05,
-      "loss": 0.5179,
-      "step": 1087
-    },
-    {
-      "epoch": 1.93,
-      "learning_rate": 1.9662796762301395e-05,
-      "loss": 0.5321,
-      "step": 1088
-    },
-    {
-      "epoch": 1.93,
-      "learning_rate": 1.966178761463277e-05,
-      "loss": 0.5248,
-      "step": 1089
-    },
-    {
-      "epoch": 1.94,
-      "learning_rate": 1.9660776985156226e-05,
-      "loss": 0.5122,
-      "step": 1090
-    },
-    {
-      "epoch": 1.94,
-      "learning_rate": 1.965976487402676e-05,
-      "loss": 0.542,
-      "step": 1091
-    },
-    {
-      "epoch": 1.94,
-      "learning_rate": 1.9658751281399603e-05,
-      "loss": 0.5326,
-      "step": 1092
-    },
-    {
-      "epoch": 1.94,
-      "learning_rate": 1.96577362074302e-05,
-      "loss": 0.5136,
-      "step": 1093
-    },
-    {
-      "epoch": 1.94,
-      "learning_rate": 1.9656719652274238e-05,
-      "loss": 0.5252,
-      "step": 1094
-    },
-    {
-      "epoch": 1.94,
-      "learning_rate": 1.965570161608762e-05,
-      "loss": 0.5483,
-      "step": 1095
-    },
-    {
-      "epoch": 1.95,
-      "learning_rate": 1.9654682099026477e-05,
-      "loss": 0.5342,
-      "step": 1096
-    },
-    {
-      "epoch": 1.95,
-      "learning_rate": 1.9653661101247178e-05,
-      "loss": 0.5503,
-      "step": 1097
-    },
-    {
-      "epoch": 1.95,
-      "learning_rate": 1.9652638622906304e-05,
-      "loss": 0.518,
-      "step": 1098
-    },
-    {
-      "epoch": 1.95,
-      "learning_rate": 1.9651614664160674e-05,
-      "loss": 0.5462,
-      "step": 1099
-    },
-    {
-      "epoch": 1.95,
-      "learning_rate": 1.965058922516733e-05,
-      "loss": 0.5389,
-      "step": 1100
-    },
-    {
-      "epoch": 1.95,
-      "learning_rate": 1.964956230608354e-05,
-      "loss": 0.5482,
-      "step": 1101
-    },
-    {
-      "epoch": 1.96,
-      "learning_rate": 1.9648533907066804e-05,
-      "loss": 0.5453,
-      "step": 1102
-    },
-    {
-      "epoch": 1.96,
-      "learning_rate": 1.964750402827484e-05,
-      "loss": 0.5448,
-      "step": 1103
-    },
-    {
-      "epoch": 1.96,
-      "learning_rate": 1.9646472669865604e-05,
-      "loss": 0.5428,
-      "step": 1104
-    },
-    {
-      "epoch": 1.96,
-      "learning_rate": 1.964543983199727e-05,
-      "loss": 0.5341,
-      "step": 1105
-    },
-    {
-      "epoch": 1.96,
-      "learning_rate": 1.9644405514828243e-05,
-      "loss": 0.5448,
-      "step": 1106
-    },
-    {
-      "epoch": 1.97,
-      "learning_rate": 1.9643369718517152e-05,
-      "loss": 0.547,
-      "step": 1107
-    },
-    {
-      "epoch": 1.97,
-      "learning_rate": 1.9642332443222852e-05,
-      "loss": 0.5108,
-      "step": 1108
-    },
-    {
-      "epoch": 1.97,
-      "learning_rate": 1.964129368910444e-05,
-      "loss": 0.5383,
-      "step": 1109
-    },
-    {
-      "epoch": 1.97,
-      "learning_rate": 1.9640253456321216e-05,
-      "loss": 0.5171,
-      "step": 1110
-    },
-    {
-      "epoch": 1.97,
-      "learning_rate": 1.9639211745032725e-05,
-      "loss": 0.5239,
-      "step": 1111
-    },
-    {
-      "epoch": 1.97,
-      "learning_rate": 1.9638168555398728e-05,
-      "loss": 0.5125,
-      "step": 1112
-    },
-    {
-      "epoch": 1.98,
-      "learning_rate": 1.9637123887579217e-05,
-      "loss": 0.5325,
-      "step": 1113
-    },
-    {
-      "epoch": 1.98,
-      "learning_rate": 1.9636077741734413e-05,
-      "loss": 0.5517,
-      "step": 1114
-    },
-    {
-      "epoch": 1.98,
-      "learning_rate": 1.9635030118024757e-05,
-      "loss": 0.5201,
-      "step": 1115
-    },
-    {
-      "epoch": 1.98,
-      "learning_rate": 1.9633981016610926e-05,
-      "loss": 0.5536,
-      "step": 1116
-    },
-    {
-      "epoch": 1.98,
-      "learning_rate": 1.9632930437653814e-05,
-      "loss": 0.5533,
-      "step": 1117
-    },
-    {
-      "epoch": 1.98,
-      "learning_rate": 1.9631878381314552e-05,
-      "loss": 0.5191,
-      "step": 1118
-    },
-    {
-      "epoch": 1.99,
-      "learning_rate": 1.963082484775448e-05,
-      "loss": 0.529,
-      "step": 1119
-    },
-    {
-      "epoch": 1.99,
-      "learning_rate": 1.962976983713519e-05,
-      "loss": 0.5388,
-      "step": 1120
-    },
-    {
-      "epoch": 1.99,
-      "learning_rate": 1.962871334961848e-05,
-      "loss": 0.516,
-      "step": 1121
-    },
-    {
-      "epoch": 1.99,
-      "learning_rate": 1.962765538536638e-05,
-      "loss": 0.5261,
-      "step": 1122
-    },
-    {
-      "epoch": 1.99,
-      "learning_rate": 1.962659594454115e-05,
-      "loss": 0.5312,
-      "step": 1123
-    },
-    {
-      "epoch": 2.0,
-      "learning_rate": 1.9625535027305272e-05,
-      "loss": 0.5231,
-      "step": 1124
-    },
-    {
-      "epoch": 2.0,
-      "learning_rate": 1.9624472633821464e-05,
-      "loss": 0.5401,
-      "step": 1125
-    },
-    {
-      "epoch": 2.0,
-      "learning_rate": 1.9623408764252652e-05,
-      "loss": 0.5383,
-      "step": 1126
-    },
-    {
-      "epoch": 2.0,
-      "eval_loss": 0.5581398606300354,
-      "eval_runtime": 3447.8302,
-      "eval_samples_per_second": 4.161,
-      "eval_steps_per_second": 1.04,
-      "step": 1126
-    },
-    {
-      "epoch": 2.0,
-      "learning_rate": 1.9622343418762008e-05,
-      "loss": 0.529,
-      "step": 1127
-    },
-    {
-      "epoch": 2.0,
-      "learning_rate": 1.9621276597512917e-05,
-      "loss": 0.5141,
-      "step": 1128
-    },
-    {
-      "epoch": 2.0,
-      "learning_rate": 1.9620208300669e-05,
-      "loss": 0.5177,
-      "step": 1129
-    },
-    {
-      "epoch": 2.01,
-      "learning_rate": 1.961913852839409e-05,
-      "loss": 0.5393,
-      "step": 1130
-    },
-    {
-      "epoch": 2.01,
-      "learning_rate": 1.961806728085227e-05,
-      "loss": 0.523,
-      "step": 1131
-    },
-    {
-      "epoch": 2.01,
-      "learning_rate": 1.961699455820782e-05,
-      "loss": 0.5238,
-      "step": 1132
-    },
-    {
-      "epoch": 2.01,
-      "learning_rate": 1.961592036062527e-05,
-      "loss": 0.5229,
-      "step": 1133
-    },
-    {
-      "epoch": 2.01,
-      "learning_rate": 1.961484468826937e-05,
-      "loss": 0.5297,
-      "step": 1134
-    },
-    {
-      "epoch": 2.02,
-      "learning_rate": 1.9613767541305084e-05,
-      "loss": 0.5316,
-      "step": 1135
-    },
-    {
-      "epoch": 2.02,
-      "learning_rate": 1.9612688919897618e-05,
-      "loss": 0.5078,
-      "step": 1136
-    },
-    {
-      "epoch": 2.02,
-      "learning_rate": 1.9611608824212395e-05,
-      "loss": 0.4892,
-      "step": 1137
-    },
-    {
-      "epoch": 2.02,
-      "learning_rate": 1.9610527254415072e-05,
-      "loss": 0.553,
-      "step": 1138
-    },
-    {
-      "epoch": 2.02,
-      "learning_rate": 1.9609444210671523e-05,
-      "loss": 0.5133,
-      "step": 1139
-    },
-    {
-      "epoch": 2.02,
-      "learning_rate": 1.960835969314785e-05,
-      "loss": 0.5322,
-      "step": 1140
-    },
-    {
-      "epoch": 2.03,
-      "learning_rate": 1.960727370201039e-05,
-      "loss": 0.5193,
-      "step": 1141
-    },
-    {
-      "epoch": 2.03,
-      "learning_rate": 1.9606186237425694e-05,
-      "loss": 0.573,
-      "step": 1142
-    },
-    {
-      "epoch": 2.03,
-      "learning_rate": 1.9605097299560545e-05,
-      "loss": 0.5379,
-      "step": 1143
-    },
-    {
-      "epoch": 2.03,
-      "learning_rate": 1.960400688858195e-05,
-      "loss": 0.5072,
-      "step": 1144
-    },
-    {
-      "epoch": 2.03,
-      "learning_rate": 1.9602915004657146e-05,
-      "loss": 0.5233,
-      "step": 1145
-    },
-    {
-      "epoch": 2.03,
-      "learning_rate": 1.960182164795359e-05,
-      "loss": 0.5332,
-      "step": 1146
-    },
-    {
-      "epoch": 2.04,
-      "learning_rate": 1.9600726818638974e-05,
-      "loss": 0.5056,
-      "step": 1147
-    },
-    {
-      "epoch": 2.04,
-      "learning_rate": 1.95996305168812e-05,
-      "loss": 0.5464,
-      "step": 1148
-    },
-    {
-      "epoch": 2.04,
-      "learning_rate": 1.959853274284841e-05,
-      "loss": 0.5317,
-      "step": 1149
-    },
-    {
-      "epoch": 2.04,
-      "learning_rate": 1.959743349670897e-05,
-      "loss": 0.5143,
-      "step": 1150
-    },
-    {
-      "epoch": 2.04,
-      "learning_rate": 1.9596332778631467e-05,
-      "loss": 0.5269,
-      "step": 1151
-    },
-    {
-      "epoch": 2.05,
-      "learning_rate": 1.9595230588784715e-05,
-      "loss": 0.5387,
-      "step": 1152
-    },
-    {
-      "epoch": 2.05,
-      "learning_rate": 1.9594126927337753e-05,
-      "loss": 0.5005,
-      "step": 1153
-    },
-    {
-      "epoch": 2.05,
-      "learning_rate": 1.9593021794459853e-05,
-      "loss": 0.5355,
-      "step": 1154
-    },
-    {
-      "epoch": 2.05,
-      "learning_rate": 1.95919151903205e-05,
-      "loss": 0.52,
-      "step": 1155
-    },
-    {
-      "epoch": 2.05,
-      "learning_rate": 1.9590807115089417e-05,
-      "loss": 0.5417,
-      "step": 1156
-    },
-    {
-      "epoch": 2.05,
-      "learning_rate": 1.9589697568936542e-05,
-      "loss": 0.5102,
-      "step": 1157
-    },
-    {
-      "epoch": 2.06,
-      "learning_rate": 1.958858655203205e-05,
-      "loss": 0.4903,
-      "step": 1158
-    },
-    {
-      "epoch": 2.06,
-      "learning_rate": 1.958747406454633e-05,
-      "loss": 0.5315,
-      "step": 1159
-    },
-    {
-      "epoch": 2.06,
-      "learning_rate": 1.9586360106650003e-05,
-      "loss": 0.512,
-      "step": 1160
-    },
-    {
-      "epoch": 2.06,
-      "learning_rate": 1.9585244678513913e-05,
-      "loss": 0.5317,
-      "step": 1161
-    },
-    {
-      "epoch": 2.06,
-      "learning_rate": 1.9584127780309138e-05,
-      "loss": 0.5339,
-      "step": 1162
-    },
-    {
-      "epoch": 2.06,
-      "learning_rate": 1.9583009412206964e-05,
-      "loss": 0.55,
-      "step": 1163
-    },
-    {
-      "epoch": 2.07,
-      "learning_rate": 1.9581889574378926e-05,
-      "loss": 0.513,
-      "step": 1164
-    },
-    {
-      "epoch": 2.07,
-      "learning_rate": 1.958076826699676e-05,
-      "loss": 0.5399,
-      "step": 1165
-    },
-    {
-      "epoch": 2.07,
-      "learning_rate": 1.9579645490232435e-05,
-      "loss": 0.5253,
-      "step": 1166
-    },
-    {
-      "epoch": 2.07,
-      "learning_rate": 1.9578521244258162e-05,
-      "loss": 0.5368,
-      "step": 1167
-    },
-    {
-      "epoch": 2.07,
-      "learning_rate": 1.9577395529246358e-05,
-      "loss": 0.5298,
-      "step": 1168
-    },
-    {
-      "epoch": 2.08,
-      "learning_rate": 1.9576268345369668e-05,
-      "loss": 0.5108,
-      "step": 1169
-    },
-    {
-      "epoch": 2.08,
-      "learning_rate": 1.9575139692800972e-05,
-      "loss": 0.5284,
-      "step": 1170
-    },
-    {
-      "epoch": 2.08,
-      "learning_rate": 1.9574009571713365e-05,
-      "loss": 0.5407,
-      "step": 1171
-    },
-    {
-      "epoch": 2.08,
-      "learning_rate": 1.9572877982280172e-05,
-      "loss": 0.5336,
-      "step": 1172
-    },
-    {
-      "epoch": 2.08,
-      "learning_rate": 1.9571744924674943e-05,
-      "loss": 0.5413,
-      "step": 1173
-    },
-    {
-      "epoch": 2.08,
-      "learning_rate": 1.957061039907145e-05,
-      "loss": 0.525,
-      "step": 1174
-    },
-    {
-      "epoch": 2.09,
-      "learning_rate": 1.95694744056437e-05,
-      "loss": 0.5223,
-      "step": 1175
-    },
-    {
-      "epoch": 2.09,
-      "learning_rate": 1.9568336944565913e-05,
-      "loss": 0.5337,
-      "step": 1176
-    },
-    {
-      "epoch": 2.09,
-      "learning_rate": 1.9567198016012537e-05,
-      "loss": 0.5306,
-      "step": 1177
-    },
-    {
-      "epoch": 2.09,
-      "learning_rate": 1.9566057620158253e-05,
-      "loss": 0.5491,
-      "step": 1178
-    },
-    {
-      "epoch": 2.09,
-      "learning_rate": 1.9564915757177955e-05,
-      "loss": 0.523,
-      "step": 1179
-    },
-    {
-      "epoch": 2.09,
-      "learning_rate": 1.956377242724677e-05,
-      "loss": 0.5167,
-      "step": 1180
-    },
-    {
-      "epoch": 2.1,
-      "learning_rate": 1.956262763054005e-05,
-      "loss": 0.5207,
-      "step": 1181
-    },
-    {
-      "epoch": 2.1,
-      "learning_rate": 1.9561481367233367e-05,
-      "loss": 0.5027,
-      "step": 1182
-    },
-    {
-      "epoch": 2.1,
-      "learning_rate": 1.9560333637502526e-05,
-      "loss": 0.5435,
-      "step": 1183
-    },
-    {
-      "epoch": 2.1,
-      "learning_rate": 1.9559184441523544e-05,
-      "loss": 0.5183,
-      "step": 1184
-    },
-    {
-      "epoch": 2.1,
-      "learning_rate": 1.955803377947268e-05,
-      "loss": 0.5411,
-      "step": 1185
-    },
-    {
-      "epoch": 2.11,
-      "learning_rate": 1.95568816515264e-05,
-      "loss": 0.5482,
-      "step": 1186
-    },
-    {
-      "epoch": 2.11,
-      "learning_rate": 1.955572805786141e-05,
-      "loss": 0.5235,
-      "step": 1187
-    },
-    {
-      "epoch": 2.11,
-      "learning_rate": 1.9554572998654627e-05,
-      "loss": 0.5151,
-      "step": 1188
-    },
-    {
-      "epoch": 2.11,
-      "learning_rate": 1.955341647408321e-05,
-      "loss": 0.4954,
-      "step": 1189
-    },
-    {
-      "epoch": 2.11,
-      "learning_rate": 1.9552258484324526e-05,
-      "loss": 0.5456,
-      "step": 1190
-    },
-    {
-      "epoch": 2.11,
-      "learning_rate": 1.955109902955617e-05,
-      "loss": 0.5252,
-      "step": 1191
-    },
-    {
-      "epoch": 2.12,
-      "learning_rate": 1.954993810995598e-05,
-      "loss": 0.525,
-      "step": 1192
-    },
-    {
-      "epoch": 2.12,
-      "learning_rate": 1.9548775725701986e-05,
-      "loss": 0.5331,
-      "step": 1193
-    },
-    {
-      "epoch": 2.12,
-      "learning_rate": 1.954761187697247e-05,
-      "loss": 0.4938,
-      "step": 1194
-    },
-    {
-      "epoch": 2.12,
-      "learning_rate": 1.9546446563945926e-05,
-      "loss": 0.52,
-      "step": 1195
-    },
-    {
-      "epoch": 2.12,
-      "learning_rate": 1.9545279786801078e-05,
-      "loss": 0.5232,
-      "step": 1196
-    },
-    {
-      "epoch": 2.13,
-      "learning_rate": 1.954411154571687e-05,
-      "loss": 0.5159,
-      "step": 1197
-    },
-    {
-      "epoch": 2.13,
-      "learning_rate": 1.9542941840872474e-05,
-      "loss": 0.5202,
-      "step": 1198
-    },
-    {
-      "epoch": 2.13,
-      "learning_rate": 1.9541770672447288e-05,
-      "loss": 0.5164,
-      "step": 1199
-    },
-    {
-      "epoch": 2.13,
-      "learning_rate": 1.954059804062092e-05,
-      "loss": 0.5467,
-      "step": 1200
-    },
-    {
-      "epoch": 2.13,
-      "learning_rate": 1.953942394557323e-05,
-      "loss": 0.5022,
-      "step": 1201
-    },
-    {
-      "epoch": 2.13,
-      "learning_rate": 1.9538248387484275e-05,
-      "loss": 0.5324,
-      "step": 1202
-    },
-    {
-      "epoch": 2.14,
-      "learning_rate": 1.9537071366534352e-05,
-      "loss": 0.5062,
-      "step": 1203
-    },
-    {
-      "epoch": 2.14,
-      "learning_rate": 1.9535892882903978e-05,
-      "loss": 0.5387,
-      "step": 1204
-    },
-    {
-      "epoch": 2.14,
-      "learning_rate": 1.9534712936773897e-05,
-      "loss": 0.507,
-      "step": 1205
-    },
-    {
-      "epoch": 2.14,
-      "learning_rate": 1.9533531528325068e-05,
-      "loss": 0.5266,
-      "step": 1206
-    },
-    {
-      "epoch": 2.14,
-      "learning_rate": 1.953234865773869e-05,
-      "loss": 0.5161,
-      "step": 1207
-    },
-    {
-      "epoch": 2.14,
-      "learning_rate": 1.9531164325196167e-05,
-      "loss": 0.5511,
-      "step": 1208
-    },
-    {
-      "epoch": 2.15,
-      "learning_rate": 1.9529978530879144e-05,
-      "loss": 0.5225,
-      "step": 1209
-    },
-    {
-      "epoch": 2.15,
-      "learning_rate": 1.9528791274969487e-05,
-      "loss": 0.5365,
-      "step": 1210
-    },
-    {
-      "epoch": 2.15,
-      "learning_rate": 1.9527602557649275e-05,
-      "loss": 0.5245,
-      "step": 1211
-    },
-    {
-      "epoch": 2.15,
-      "learning_rate": 1.9526412379100827e-05,
-      "loss": 0.5138,
-      "step": 1212
-    },
-    {
-      "epoch": 2.15,
-      "learning_rate": 1.9525220739506674e-05,
-      "loss": 0.5369,
-      "step": 1213
-    },
-    {
-      "epoch": 2.16,
-      "learning_rate": 1.9524027639049576e-05,
-      "loss": 0.5403,
-      "step": 1214
-    },
-    {
-      "epoch": 2.16,
-      "learning_rate": 1.9522833077912512e-05,
-      "loss": 0.5517,
-      "step": 1215
-    },
-    {
-      "epoch": 2.16,
-      "learning_rate": 1.9521637056278693e-05,
-      "loss": 0.4965,
-      "step": 1216
-    },
-    {
-      "epoch": 2.16,
-      "learning_rate": 1.9520439574331554e-05,
-      "loss": 0.5408,
-      "step": 1217
-    },
-    {
-      "epoch": 2.16,
-      "learning_rate": 1.9519240632254744e-05,
-      "loss": 0.5274,
-      "step": 1218
-    },
-    {
-      "epoch": 2.16,
-      "learning_rate": 1.9518040230232145e-05,
-      "loss": 0.5208,
-      "step": 1219
-    },
-    {
-      "epoch": 2.17,
-      "learning_rate": 1.951683836844786e-05,
-      "loss": 0.5243,
-      "step": 1220
-    },
-    {
-      "epoch": 2.17,
-      "learning_rate": 1.951563504708622e-05,
-      "loss": 0.4931,
-      "step": 1221
-    },
-    {
-      "epoch": 2.17,
-      "learning_rate": 1.9514430266331766e-05,
-      "loss": 0.5402,
-      "step": 1222
-    },
-    {
-      "epoch": 2.17,
-      "learning_rate": 1.9513224026369278e-05,
-      "loss": 0.5363,
-      "step": 1223
-    },
-    {
-      "epoch": 2.17,
-      "learning_rate": 1.951201632738376e-05,
-      "loss": 0.5269,
-      "step": 1224
-    },
-    {
-      "epoch": 2.17,
-      "learning_rate": 1.9510807169560426e-05,
-      "loss": 0.5391,
-      "step": 1225
-    },
-    {
-      "epoch": 2.18,
-      "learning_rate": 1.9509596553084728e-05,
-      "loss": 0.5265,
-      "step": 1226
-    },
-    {
-      "epoch": 2.18,
-      "learning_rate": 1.950838447814233e-05,
-      "loss": 0.5403,
-      "step": 1227
-    },
-    {
-      "epoch": 2.18,
-      "learning_rate": 1.9507170944919127e-05,
-      "loss": 0.5293,
-      "step": 1228
-    },
-    {
-      "epoch": 2.18,
-      "learning_rate": 1.9505955953601238e-05,
-      "loss": 0.5352,
-      "step": 1229
-    },
-    {
-      "epoch": 2.18,
-      "learning_rate": 1.9504739504375004e-05,
-      "loss": 0.5348,
-      "step": 1230
-    },
-    {
-      "epoch": 2.19,
-      "learning_rate": 1.950352159742699e-05,
-      "loss": 0.5132,
-      "step": 1231
-    },
-    {
-      "epoch": 2.19,
-      "learning_rate": 1.950230223294398e-05,
-      "loss": 0.5154,
-      "step": 1232
-    },
-    {
-      "epoch": 2.19,
-      "learning_rate": 1.9501081411112987e-05,
-      "loss": 0.5342,
-      "step": 1233
-    },
-    {
-      "epoch": 2.19,
-      "learning_rate": 1.9499859132121247e-05,
-      "loss": 0.505,
-      "step": 1234
-    },
-    {
-      "epoch": 2.19,
-      "learning_rate": 1.9498635396156217e-05,
-      "loss": 0.5356,
-      "step": 1235
-    },
-    {
-      "epoch": 2.19,
-      "learning_rate": 1.949741020340558e-05,
-      "loss": 0.5351,
-      "step": 1236
-    },
-    {
-      "epoch": 2.2,
-      "learning_rate": 1.949618355405724e-05,
-      "loss": 0.5203,
-      "step": 1237
-    },
-    {
-      "epoch": 2.2,
-      "learning_rate": 1.9494955448299325e-05,
-      "loss": 0.5268,
-      "step": 1238
-    },
-    {
-      "epoch": 2.2,
-      "learning_rate": 1.9493725886320193e-05,
-      "loss": 0.531,
-      "step": 1239
-    },
-    {
-      "epoch": 2.2,
-      "learning_rate": 1.9492494868308406e-05,
-      "loss": 0.531,
-      "step": 1240
-    },
-    {
-      "epoch": 2.2,
-      "learning_rate": 1.9491262394452778e-05,
-      "loss": 0.5346,
-      "step": 1241
-    },
-    {
-      "epoch": 2.21,
-      "learning_rate": 1.9490028464942322e-05,
-      "loss": 0.5121,
-      "step": 1242
-    },
-    {
-      "epoch": 2.21,
-      "learning_rate": 1.9488793079966283e-05,
-      "loss": 0.511,
-      "step": 1243
-    },
-    {
-      "epoch": 2.21,
-      "learning_rate": 1.948755623971413e-05,
-      "loss": 0.5092,
-      "step": 1244
-    },
-    {
-      "epoch": 2.21,
-      "learning_rate": 1.9486317944375563e-05,
-      "loss": 0.5356,
-      "step": 1245
-    },
-    {
-      "epoch": 2.21,
-      "learning_rate": 1.9485078194140484e-05,
-      "loss": 0.5458,
-      "step": 1246
-    },
-    {
-      "epoch": 2.21,
-      "learning_rate": 1.948383698919904e-05,
-      "loss": 0.509,
-      "step": 1247
-    },
-    {
-      "epoch": 2.22,
-      "learning_rate": 1.9482594329741587e-05,
-      "loss": 0.5336,
-      "step": 1248
-    },
-    {
-      "epoch": 2.22,
-      "learning_rate": 1.9481350215958713e-05,
-      "loss": 0.5311,
-      "step": 1249
-    },
-    {
-      "epoch": 2.22,
-      "learning_rate": 1.948010464804122e-05,
-      "loss": 0.5438,
-      "step": 1250
-    },
-    {
-      "epoch": 2.22,
-      "learning_rate": 1.947885762618014e-05,
-      "loss": 0.5034,
-      "step": 1251
-    },
-    {
-      "epoch": 2.22,
-      "learning_rate": 1.947760915056673e-05,
-      "loss": 0.5237,
-      "step": 1252
-    },
-    {
-      "epoch": 2.22,
-      "learning_rate": 1.947635922139246e-05,
-      "loss": 0.5274,
-      "step": 1253
-    },
-    {
-      "epoch": 2.23,
-      "learning_rate": 1.9475107838849035e-05,
-      "loss": 0.531,
-      "step": 1254
-    },
-    {
-      "epoch": 2.23,
-      "learning_rate": 1.947385500312837e-05,
-      "loss": 0.5149,
-      "step": 1255
-    },
-    {
-      "epoch": 2.23,
-      "learning_rate": 1.9472600714422618e-05,
-      "loss": 0.5173,
-      "step": 1256
-    },
-    {
-      "epoch": 2.23,
-      "learning_rate": 1.9471344972924142e-05,
-      "loss": 0.5282,
-      "step": 1257
-    },
-    {
-      "epoch": 2.23,
-      "learning_rate": 1.947008777882553e-05,
-      "loss": 0.5214,
-      "step": 1258
-    },
-    {
-      "epoch": 2.24,
-      "learning_rate": 1.94688291323196e-05,
-      "loss": 0.5198,
-      "step": 1259
-    },
-    {
-      "epoch": 2.24,
-      "learning_rate": 1.9467569033599386e-05,
-      "loss": 0.5429,
-      "step": 1260
-    },
-    {
-      "epoch": 2.24,
-      "learning_rate": 1.9466307482858144e-05,
-      "loss": 0.5148,
-      "step": 1261
-    },
-    {
-      "epoch": 2.24,
-      "learning_rate": 1.946504448028936e-05,
-      "loss": 0.5135,
-      "step": 1262
-    },
-    {
-      "epoch": 2.24,
-      "learning_rate": 1.9463780026086735e-05,
-      "loss": 0.5217,
-      "step": 1263
-    },
-    {
-      "epoch": 2.24,
-      "learning_rate": 1.94625141204442e-05,
-      "loss": 0.5256,
-      "step": 1264
-    },
-    {
-      "epoch": 2.25,
-      "learning_rate": 1.9461246763555897e-05,
-      "loss": 0.4913,
-      "step": 1265
-    },
-    {
-      "epoch": 2.25,
-      "learning_rate": 1.9459977955616202e-05,
-      "loss": 0.5138,
-      "step": 1266
-    },
-    {
-      "epoch": 2.25,
-      "learning_rate": 1.945870769681971e-05,
-      "loss": 0.5417,
-      "step": 1267
-    },
-    {
-      "epoch": 2.25,
-      "learning_rate": 1.9457435987361238e-05,
-      "loss": 0.5231,
-      "step": 1268
-    },
-    {
-      "epoch": 2.25,
-      "learning_rate": 1.945616282743582e-05,
-      "loss": 0.5273,
-      "step": 1269
-    },
-    {
-      "epoch": 2.25,
-      "learning_rate": 1.945488821723873e-05,
-      "loss": 0.5223,
-      "step": 1270
-    },
-    {
-      "epoch": 2.26,
-      "learning_rate": 1.945361215696544e-05,
-      "loss": 0.508,
-      "step": 1271
-    },
-    {
-      "epoch": 2.26,
-      "learning_rate": 1.945233464681166e-05,
-      "loss": 0.5215,
-      "step": 1272
-    },
-    {
-      "epoch": 2.26,
-      "learning_rate": 1.9451055686973323e-05,
-      "loss": 0.5088,
-      "step": 1273
-    },
-    {
-      "epoch": 2.26,
-      "learning_rate": 1.9449775277646578e-05,
-      "loss": 0.5133,
-      "step": 1274
-    },
-    {
-      "epoch": 2.26,
-      "learning_rate": 1.9448493419027795e-05,
-      "loss": 0.5148,
-      "step": 1275
-    },
-    {
-      "epoch": 2.27,
-      "learning_rate": 1.9447210111313573e-05,
-      "loss": 0.5308,
-      "step": 1276
-    },
-    {
-      "epoch": 2.27,
-      "learning_rate": 1.9445925354700733e-05,
-      "loss": 0.5092,
-      "step": 1277
-    },
-    {
-      "epoch": 2.27,
-      "learning_rate": 1.9444639149386312e-05,
-      "loss": 0.5468,
-      "step": 1278
-    },
-    {
-      "epoch": 2.27,
-      "learning_rate": 1.9443351495567576e-05,
-      "loss": 0.5011,
-      "step": 1279
-    },
-    {
-      "epoch": 2.27,
-      "learning_rate": 1.9442062393442004e-05,
-      "loss": 0.5281,
-      "step": 1280
-    },
-    {
-      "epoch": 2.27,
-      "learning_rate": 1.9440771843207305e-05,
-      "loss": 0.5345,
-      "step": 1281
-    },
-    {
-      "epoch": 2.28,
-      "learning_rate": 1.943947984506141e-05,
-      "loss": 0.5158,
-      "step": 1282
-    },
-    {
-      "epoch": 2.28,
-      "learning_rate": 1.943818639920247e-05,
-      "loss": 0.5102,
-      "step": 1283
-    },
-    {
-      "epoch": 2.28,
-      "learning_rate": 1.9436891505828854e-05,
-      "loss": 0.5014,
-      "step": 1284
-    },
-    {
-      "epoch": 2.28,
-      "learning_rate": 1.9435595165139163e-05,
-      "loss": 0.5547,
-      "step": 1285
-    },
-    {
-      "epoch": 2.28,
-      "learning_rate": 1.943429737733221e-05,
-      "loss": 0.4966,
-      "step": 1286
-    },
-    {
-      "epoch": 2.28,
-      "learning_rate": 1.9432998142607036e-05,
-      "loss": 0.5491,
-      "step": 1287
-    },
-    {
-      "epoch": 2.29,
-      "learning_rate": 1.9431697461162898e-05,
-      "loss": 0.5193,
-      "step": 1288
-    },
-    {
-      "epoch": 2.29,
-      "learning_rate": 1.9430395333199284e-05,
-      "loss": 0.5586,
-      "step": 1289
-    },
-    {
-      "epoch": 2.29,
-      "learning_rate": 1.9429091758915902e-05,
-      "loss": 0.4913,
-      "step": 1290
-    },
-    {
-      "epoch": 2.29,
-      "learning_rate": 1.9427786738512667e-05,
-      "loss": 0.526,
-      "step": 1291
-    },
-    {
-      "epoch": 2.29,
-      "learning_rate": 1.942648027218973e-05,
-      "loss": 0.5252,
-      "step": 1292
-    },
-    {
-      "epoch": 2.3,
-      "learning_rate": 1.9425172360147467e-05,
-      "loss": 0.5337,
-      "step": 1293
-    },
-    {
-      "epoch": 2.3,
-      "learning_rate": 1.942386300258647e-05,
-      "loss": 0.5066,
-      "step": 1294
-    },
-    {
-      "epoch": 2.3,
-      "learning_rate": 1.9422552199707545e-05,
-      "loss": 0.5348,
-      "step": 1295
-    },
-    {
-      "epoch": 2.3,
-      "learning_rate": 1.942123995171173e-05,
-      "loss": 0.5182,
-      "step": 1296
-    },
-    {
-      "epoch": 2.3,
-      "learning_rate": 1.9419926258800286e-05,
-      "loss": 0.4796,
-      "step": 1297
-    },
-    {
-      "epoch": 2.3,
-      "learning_rate": 1.9418611121174688e-05,
-      "loss": 0.531,
-      "step": 1298
-    },
-    {
-      "epoch": 2.31,
-      "learning_rate": 1.9417294539036634e-05,
-      "loss": 0.4857,
-      "step": 1299
-    },
-    {
-      "epoch": 2.31,
-      "learning_rate": 1.9415976512588045e-05,
-      "loss": 0.5662,
-      "step": 1300
-    },
-    {
-      "epoch": 2.31,
-      "learning_rate": 1.941465704203107e-05,
-      "loss": 0.5168,
-      "step": 1301
-    },
-    {
-      "epoch": 2.31,
-      "learning_rate": 1.941333612756807e-05,
-      "loss": 0.5502,
-      "step": 1302
-    },
-    {
-      "epoch": 2.31,
-      "learning_rate": 1.941201376940163e-05,
-      "loss": 0.5216,
-      "step": 1303
-    },
-    {
-      "epoch": 2.32,
-      "learning_rate": 1.941068996773456e-05,
-      "loss": 0.5274,
-      "step": 1304
-    },
-    {
-      "epoch": 2.32,
-      "learning_rate": 1.9409364722769882e-05,
-      "loss": 0.5168,
-      "step": 1305
-    },
-    {
-      "epoch": 2.32,
-      "learning_rate": 1.9408038034710853e-05,
-      "loss": 0.5238,
-      "step": 1306
-    },
-    {
-      "epoch": 2.32,
-      "learning_rate": 1.9406709903760942e-05,
-      "loss": 0.5076,
-      "step": 1307
-    },
-    {
-      "epoch": 2.32,
-      "learning_rate": 1.9405380330123838e-05,
-      "loss": 0.5307,
-      "step": 1308
-    },
-    {
-      "epoch": 2.32,
-      "learning_rate": 1.9404049314003465e-05,
-      "loss": 0.5409,
-      "step": 1309
-    },
-    {
-      "epoch": 2.33,
-      "learning_rate": 1.9402716855603945e-05,
-      "loss": 0.5146,
-      "step": 1310
-    },
-    {
-      "epoch": 2.33,
-      "learning_rate": 1.9401382955129646e-05,
-      "loss": 0.5187,
-      "step": 1311
-    },
-    {
-      "epoch": 2.33,
-      "learning_rate": 1.9400047612785138e-05,
-      "loss": 0.5249,
-      "step": 1312
-    },
-    {
-      "epoch": 2.33,
-      "learning_rate": 1.939871082877522e-05,
-      "loss": 0.5292,
-      "step": 1313
-    },
-    {
-      "epoch": 2.33,
-      "learning_rate": 1.9397372603304917e-05,
-      "loss": 0.5282,
-      "step": 1314
-    },
-    {
-      "epoch": 2.33,
-      "learning_rate": 1.9396032936579464e-05,
-      "loss": 0.507,
-      "step": 1315
-    },
-    {
-      "epoch": 2.34,
-      "learning_rate": 1.9394691828804325e-05,
-      "loss": 0.505,
-      "step": 1316
-    },
-    {
-      "epoch": 2.34,
-      "learning_rate": 1.9393349280185187e-05,
-      "loss": 0.5303,
-      "step": 1317
-    },
-    {
-      "epoch": 2.34,
-      "learning_rate": 1.939200529092795e-05,
-      "loss": 0.5355,
-      "step": 1318
-    },
-    {
-      "epoch": 2.34,
-      "learning_rate": 1.9390659861238735e-05,
-      "loss": 0.497,
-      "step": 1319
-    },
-    {
-      "epoch": 2.34,
-      "learning_rate": 1.9389312991323895e-05,
-      "loss": 0.5345,
-      "step": 1320
-    },
-    {
-      "epoch": 2.35,
-      "learning_rate": 1.938796468138999e-05,
-      "loss": 0.513,
-      "step": 1321
-    },
-    {
-      "epoch": 2.35,
-      "learning_rate": 1.9386614931643814e-05,
-      "loss": 0.5299,
-      "step": 1322
-    },
-    {
-      "epoch": 2.35,
-      "learning_rate": 1.9385263742292368e-05,
-      "loss": 0.5226,
-      "step": 1323
-    },
-    {
-      "epoch": 2.35,
-      "learning_rate": 1.938391111354289e-05,
-      "loss": 0.5221,
-      "step": 1324
-    },
-    {
-      "epoch": 2.35,
-      "learning_rate": 1.9382557045602816e-05,
-      "loss": 0.5002,
-      "step": 1325
-    },
-    {
-      "epoch": 2.35,
-      "learning_rate": 1.938120153867983e-05,
-      "loss": 0.5279,
-      "step": 1326
-    },
-    {
-      "epoch": 2.36,
-      "learning_rate": 1.937984459298182e-05,
-      "loss": 0.4825,
-      "step": 1327
-    },
-    {
-      "epoch": 2.36,
-      "learning_rate": 1.9378486208716896e-05,
-      "loss": 0.5513,
-      "step": 1328
-    },
-    {
-      "epoch": 2.36,
-      "learning_rate": 1.937712638609339e-05,
-      "loss": 0.5037,
-      "step": 1329
-    },
-    {
-      "epoch": 2.36,
-      "learning_rate": 1.937576512531986e-05,
-      "loss": 0.5137,
-      "step": 1330
-    },
-    {
-      "epoch": 2.36,
-      "learning_rate": 1.9374402426605072e-05,
-      "loss": 0.5307,
-      "step": 1331
-    },
-    {
-      "epoch": 2.36,
-      "learning_rate": 1.9373038290158024e-05,
-      "loss": 0.4891,
-      "step": 1332
-    },
-    {
-      "epoch": 2.37,
-      "learning_rate": 1.937167271618793e-05,
-      "loss": 0.5044,
-      "step": 1333
-    },
-    {
-      "epoch": 2.37,
-      "learning_rate": 1.9370305704904227e-05,
-      "loss": 0.5228,
-      "step": 1334
-    },
-    {
-      "epoch": 2.37,
-      "learning_rate": 1.9368937256516567e-05,
-      "loss": 0.5322,
-      "step": 1335
-    },
-    {
-      "epoch": 2.37,
-      "learning_rate": 1.936756737123483e-05,
-      "loss": 0.519,
-      "step": 1336
-    },
-    {
-      "epoch": 2.37,
-      "learning_rate": 1.936619604926911e-05,
-      "loss": 0.5228,
-      "step": 1337
-    },
-    {
-      "epoch": 2.38,
-      "learning_rate": 1.936482329082973e-05,
-      "loss": 0.5052,
-      "step": 1338
-    },
-    {
-      "epoch": 2.38,
-      "learning_rate": 1.9363449096127214e-05,
-      "loss": 0.5281,
-      "step": 1339
-    },
-    {
-      "epoch": 2.38,
-      "learning_rate": 1.936207346537233e-05,
-      "loss": 0.5294,
-      "step": 1340
-    },
-    {
-      "epoch": 2.38,
-      "learning_rate": 1.9360696398776056e-05,
-      "loss": 0.5171,
-      "step": 1341
-    },
-    {
-      "epoch": 2.38,
-      "learning_rate": 1.935931789654958e-05,
-      "loss": 0.5157,
-      "step": 1342
-    },
-    {
-      "epoch": 2.38,
-      "learning_rate": 1.935793795890433e-05,
-      "loss": 0.5079,
-      "step": 1343
-    },
-    {
-      "epoch": 2.39,
-      "learning_rate": 1.9356556586051942e-05,
-      "loss": 0.5213,
-      "step": 1344
-    },
-    {
-      "epoch": 2.39,
-      "learning_rate": 1.935517377820427e-05,
-      "loss": 0.5129,
-      "step": 1345
-    },
-    {
-      "epoch": 2.39,
-      "learning_rate": 1.9353789535573394e-05,
-      "loss": 0.5241,
-      "step": 1346
-    },
-    {
-      "epoch": 2.39,
-      "learning_rate": 1.9352403858371618e-05,
-      "loss": 0.5166,
-      "step": 1347
-    },
-    {
-      "epoch": 2.39,
-      "learning_rate": 1.9351016746811452e-05,
-      "loss": 0.5439,
-      "step": 1348
-    },
-    {
-      "epoch": 2.4,
-      "learning_rate": 1.934962820110564e-05,
-      "loss": 0.5269,
-      "step": 1349
-    },
-    {
-      "epoch": 2.4,
-      "learning_rate": 1.9348238221467138e-05,
-      "loss": 0.5185,
-      "step": 1350
-    },
-    {
-      "epoch": 2.4,
-      "learning_rate": 1.9346846808109122e-05,
-      "loss": 0.5195,
-      "step": 1351
-    },
-    {
-      "epoch": 2.4,
-      "learning_rate": 1.9345453961244994e-05,
-      "loss": 0.5037,
-      "step": 1352
-    },
-    {
-      "epoch": 2.4,
-      "learning_rate": 1.9344059681088372e-05,
-      "loss": 0.4921,
-      "step": 1353
-    },
-    {
-      "epoch": 2.4,
-      "learning_rate": 1.9342663967853096e-05,
-      "loss": 0.5191,
-      "step": 1354
-    },
-    {
-      "epoch": 2.41,
-      "learning_rate": 1.9341266821753216e-05,
-      "loss": 0.521,
-      "step": 1355
-    },
-    {
-      "epoch": 2.41,
-      "learning_rate": 1.9339868243003012e-05,
-      "loss": 0.5297,
-      "step": 1356
-    },
-    {
-      "epoch": 2.41,
-      "learning_rate": 1.9338468231816984e-05,
-      "loss": 0.5347,
-      "step": 1357
-    },
-    {
-      "epoch": 2.41,
-      "learning_rate": 1.933706678840985e-05,
-      "loss": 0.5208,
-      "step": 1358
-    },
-    {
-      "epoch": 2.41,
-      "learning_rate": 1.933566391299654e-05,
-      "loss": 0.5207,
-      "step": 1359
-    },
-    {
-      "epoch": 2.41,
-      "learning_rate": 1.9334259605792218e-05,
-      "loss": 0.5112,
-      "step": 1360
-    },
-    {
-      "epoch": 2.42,
-      "learning_rate": 1.9332853867012253e-05,
-      "loss": 0.4984,
-      "step": 1361
-    },
-    {
-      "epoch": 2.42,
-      "learning_rate": 1.933144669687224e-05,
-      "loss": 0.4784,
-      "step": 1362
-    },
-    {
-      "epoch": 2.42,
-      "learning_rate": 1.9330038095588e-05,
-      "loss": 0.5151,
-      "step": 1363
-    },
-    {
-      "epoch": 2.42,
-      "learning_rate": 1.932862806337556e-05,
-      "loss": 0.51,
-      "step": 1364
-    },
-    {
-      "epoch": 2.42,
-      "learning_rate": 1.9327216600451177e-05,
-      "loss": 0.5264,
-      "step": 1365
-    },
-    {
-      "epoch": 2.43,
-      "learning_rate": 1.932580370703133e-05,
-      "loss": 0.5255,
-      "step": 1366
-    },
-    {
-      "epoch": 2.43,
-      "learning_rate": 1.93243893833327e-05,
-      "loss": 0.5257,
-      "step": 1367
-    },
-    {
-      "epoch": 2.43,
-      "learning_rate": 1.9322973629572207e-05,
-      "loss": 0.5219,
-      "step": 1368
-    },
-    {
-      "epoch": 2.43,
-      "learning_rate": 1.9321556445966978e-05,
-      "loss": 0.5414,
-      "step": 1369
-    },
-    {
-      "epoch": 2.43,
-      "learning_rate": 1.9320137832734363e-05,
-      "loss": 0.5093,
-      "step": 1370
-    },
-    {
-      "epoch": 2.43,
-      "learning_rate": 1.931871779009194e-05,
-      "loss": 0.5128,
-      "step": 1371
-    },
-    {
-      "epoch": 2.44,
-      "learning_rate": 1.931729631825749e-05,
-      "loss": 0.4952,
-      "step": 1372
-    },
-    {
-      "epoch": 2.44,
-      "learning_rate": 1.931587341744902e-05,
-      "loss": 0.5061,
-      "step": 1373
-    },
-    {
-      "epoch": 2.44,
-      "learning_rate": 1.931444908788476e-05,
-      "loss": 0.5104,
-      "step": 1374
-    },
-    {
-      "epoch": 2.44,
-      "learning_rate": 1.931302332978316e-05,
-      "loss": 0.5288,
-      "step": 1375
-    },
-    {
-      "epoch": 2.44,
-      "learning_rate": 1.931159614336288e-05,
-      "loss": 0.5183,
-      "step": 1376
-    },
-    {
-      "epoch": 2.44,
-      "learning_rate": 1.9310167528842808e-05,
-      "loss": 0.5114,
-      "step": 1377
-    },
-    {
-      "epoch": 2.45,
-      "learning_rate": 1.9308737486442045e-05,
-      "loss": 0.5232,
-      "step": 1378
-    },
-    {
-      "epoch": 2.45,
-      "learning_rate": 1.9307306016379913e-05,
-      "loss": 0.5281,
-      "step": 1379
-    },
-    {
-      "epoch": 2.45,
-      "learning_rate": 1.930587311887596e-05,
-      "loss": 0.5333,
-      "step": 1380
-    },
-    {
-      "epoch": 2.45,
-      "learning_rate": 1.9304438794149938e-05,
-      "loss": 0.5034,
-      "step": 1381
-    },
-    {
-      "epoch": 2.45,
-      "learning_rate": 1.9303003042421832e-05,
-      "loss": 0.5084,
-      "step": 1382
-    },
-    {
-      "epoch": 2.46,
-      "learning_rate": 1.9301565863911836e-05,
-      "loss": 0.4917,
-      "step": 1383
-    },
-    {
-      "epoch": 2.46,
-      "learning_rate": 1.9300127258840372e-05,
-      "loss": 0.5231,
-      "step": 1384
-    },
-    {
-      "epoch": 2.46,
-      "learning_rate": 1.9298687227428072e-05,
-      "loss": 0.5187,
-      "step": 1385
-    },
-    {
-      "epoch": 2.46,
-      "learning_rate": 1.9297245769895793e-05,
-      "loss": 0.5201,
-      "step": 1386
-    },
-    {
-      "epoch": 2.46,
-      "learning_rate": 1.9295802886464602e-05,
-      "loss": 0.5263,
-      "step": 1387
-    },
-    {
-      "epoch": 2.46,
-      "learning_rate": 1.92943585773558e-05,
-      "loss": 0.522,
-      "step": 1388
-    },
-    {
-      "epoch": 2.47,
-      "learning_rate": 1.9292912842790893e-05,
-      "loss": 0.5301,
-      "step": 1389
-    },
-    {
-      "epoch": 2.47,
-      "learning_rate": 1.9291465682991614e-05,
-      "loss": 0.5101,
-      "step": 1390
-    },
-    {
-      "epoch": 2.47,
-      "learning_rate": 1.9290017098179904e-05,
-      "loss": 0.5199,
-      "step": 1391
-    },
-    {
-      "epoch": 2.47,
-      "learning_rate": 1.9288567088577935e-05,
-      "loss": 0.5221,
-      "step": 1392
-    },
-    {
-      "epoch": 2.47,
-      "learning_rate": 1.928711565440809e-05,
-      "loss": 0.5059,
-      "step": 1393
-    },
-    {
-      "epoch": 2.47,
-      "learning_rate": 1.928566279589297e-05,
-      "loss": 0.5088,
-      "step": 1394
-    },
-    {
-      "epoch": 2.48,
-      "learning_rate": 1.92842085132554e-05,
-      "loss": 0.5168,
-      "step": 1395
-    },
-    {
-      "epoch": 2.48,
-      "learning_rate": 1.928275280671842e-05,
-      "loss": 0.5268,
-      "step": 1396
-    },
-    {
-      "epoch": 2.48,
-      "learning_rate": 1.928129567650529e-05,
-      "loss": 0.5031,
-      "step": 1397
-    },
-    {
-      "epoch": 2.48,
-      "learning_rate": 1.9279837122839485e-05,
-      "loss": 0.501,
-      "step": 1398
-    },
-    {
-      "epoch": 2.48,
-      "learning_rate": 1.9278377145944697e-05,
-      "loss": 0.5391,
-      "step": 1399
-    },
-    {
-      "epoch": 2.49,
-      "learning_rate": 1.9276915746044847e-05,
-      "loss": 0.5144,
-      "step": 1400
-    },
-    {
-      "epoch": 2.49,
-      "learning_rate": 1.927545292336406e-05,
-      "loss": 0.5157,
-      "step": 1401
-    },
-    {
-      "epoch": 2.49,
-      "learning_rate": 1.927398867812669e-05,
-      "loss": 0.4874,
-      "step": 1402
-    },
-    {
-      "epoch": 2.49,
-      "learning_rate": 1.9272523010557305e-05,
-      "loss": 0.4914,
-      "step": 1403
-    },
-    {
-      "epoch": 2.49,
-      "learning_rate": 1.9271055920880684e-05,
-      "loss": 0.51,
-      "step": 1404
-    },
-    {
-      "epoch": 2.49,
-      "learning_rate": 1.9269587409321845e-05,
-      "loss": 0.5141,
-      "step": 1405
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 1.9268117476106e-05,
-      "loss": 0.4962,
-      "step": 1406
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 1.9266646121458597e-05,
-      "loss": 0.52,
-      "step": 1407
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 1.9265173345605285e-05,
-      "loss": 0.5091,
-      "step": 1408
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 1.926369914877195e-05,
-      "loss": 0.5235,
-      "step": 1409
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 1.9262223531184678e-05,
-      "loss": 0.5258,
-      "step": 1410
-    },
-    {
-      "epoch": 2.51,
-      "learning_rate": 1.9260746493069787e-05,
-      "loss": 0.527,
-      "step": 1411
-    },
-    {
-      "epoch": 2.51,
-      "learning_rate": 1.9259268034653808e-05,
-      "loss": 0.5239,
-      "step": 1412
-    },
-    {
-      "epoch": 2.51,
-      "learning_rate": 1.9257788156163484e-05,
-      "loss": 0.5134,
-      "step": 1413
-    },
-    {
-      "epoch": 2.51,
-      "learning_rate": 1.9256306857825787e-05,
-      "loss": 0.5374,
-      "step": 1414
-    },
-    {
-      "epoch": 2.51,
-      "learning_rate": 1.9254824139867897e-05,
-      "loss": 0.5348,
-      "step": 1415
-    },
-    {
-      "epoch": 2.51,
-      "learning_rate": 1.9253340002517213e-05,
-      "loss": 0.4989,
-      "step": 1416
-    },
-    {
-      "epoch": 2.52,
-      "learning_rate": 1.9251854446001358e-05,
-      "loss": 0.5146,
-      "step": 1417
-    },
-    {
-      "epoch": 2.52,
-      "learning_rate": 1.925036747054817e-05,
-      "loss": 0.5092,
-      "step": 1418
-    },
-    {
-      "epoch": 2.52,
-      "learning_rate": 1.9248879076385696e-05,
-      "loss": 0.518,
-      "step": 1419
-    },
-    {
-      "epoch": 2.52,
-      "learning_rate": 1.9247389263742215e-05,
-      "loss": 0.5313,
-      "step": 1420
-    },
-    {
-      "epoch": 2.52,
-      "learning_rate": 1.9245898032846212e-05,
-      "loss": 0.5308,
-      "step": 1421
-    },
-    {
-      "epoch": 2.52,
-      "learning_rate": 1.9244405383926396e-05,
-      "loss": 0.511,
-      "step": 1422
-    },
-    {
-      "epoch": 2.53,
-      "learning_rate": 1.9242911317211696e-05,
-      "loss": 0.5223,
-      "step": 1423
-    },
-    {
-      "epoch": 2.53,
-      "learning_rate": 1.9241415832931244e-05,
-      "loss": 0.5307,
-      "step": 1424
-    },
-    {
-      "epoch": 2.53,
-      "learning_rate": 1.9239918931314404e-05,
-      "loss": 0.5238,
-      "step": 1425
-    },
-    {
-      "epoch": 2.53,
-      "learning_rate": 1.9238420612590758e-05,
-      "loss": 0.5214,
-      "step": 1426
-    },
-    {
-      "epoch": 2.53,
-      "learning_rate": 1.923692087699009e-05,
-      "loss": 0.5082,
-      "step": 1427
-    },
-    {
-      "epoch": 2.54,
-      "learning_rate": 1.923541972474242e-05,
-      "loss": 0.5283,
-      "step": 1428
-    },
-    {
-      "epoch": 2.54,
-      "learning_rate": 1.923391715607797e-05,
-      "loss": 0.5212,
-      "step": 1429
-    },
-    {
-      "epoch": 2.54,
-      "learning_rate": 1.923241317122719e-05,
-      "loss": 0.5256,
-      "step": 1430
-    },
-    {
-      "epoch": 2.54,
-      "learning_rate": 1.9230907770420737e-05,
-      "loss": 0.5368,
-      "step": 1431
-    },
-    {
-      "epoch": 2.54,
-      "learning_rate": 1.92294009538895e-05,
-      "loss": 0.4992,
-      "step": 1432
-    },
-    {
-      "epoch": 2.54,
-      "learning_rate": 1.9227892721864572e-05,
-      "loss": 0.5014,
-      "step": 1433
-    },
-    {
-      "epoch": 2.55,
-      "learning_rate": 1.9226383074577266e-05,
-      "loss": 0.5134,
-      "step": 1434
-    },
-    {
-      "epoch": 2.55,
-      "learning_rate": 1.9224872012259114e-05,
-      "loss": 0.5356,
-      "step": 1435
-    },
-    {
-      "epoch": 2.55,
-      "learning_rate": 1.9223359535141868e-05,
-      "loss": 0.5197,
-      "step": 1436
-    },
-    {
-      "epoch": 2.55,
-      "learning_rate": 1.9221845643457485e-05,
-      "loss": 0.51,
-      "step": 1437
-    },
-    {
-      "epoch": 2.55,
-      "learning_rate": 1.9220330337438158e-05,
-      "loss": 0.5045,
-      "step": 1438
-    },
-    {
-      "epoch": 2.55,
-      "learning_rate": 1.921881361731628e-05,
-      "loss": 0.5146,
-      "step": 1439
-    },
-    {
-      "epoch": 2.56,
-      "learning_rate": 1.9217295483324467e-05,
-      "loss": 0.5332,
-      "step": 1440
-    },
-    {
-      "epoch": 2.56,
-      "learning_rate": 1.9215775935695557e-05,
-      "loss": 0.5153,
-      "step": 1441
-    },
-    {
-      "epoch": 2.56,
-      "learning_rate": 1.9214254974662592e-05,
-      "loss": 0.5157,
-      "step": 1442
-    },
-    {
-      "epoch": 2.56,
-      "learning_rate": 1.9212732600458845e-05,
-      "loss": 0.5036,
-      "step": 1443
-    },
-    {
-      "epoch": 2.56,
-      "learning_rate": 1.9211208813317795e-05,
-      "loss": 0.5324,
-      "step": 1444
-    },
-    {
-      "epoch": 2.57,
-      "learning_rate": 1.9209683613473143e-05,
-      "loss": 0.5298,
-      "step": 1445
-    },
-    {
-      "epoch": 2.57,
-      "learning_rate": 1.9208157001158808e-05,
-      "loss": 0.5147,
-      "step": 1446
-    },
-    {
-      "epoch": 2.57,
-      "learning_rate": 1.920662897660892e-05,
-      "loss": 0.4774,
-      "step": 1447
-    },
-    {
-      "epoch": 2.57,
-      "learning_rate": 1.9205099540057833e-05,
-      "loss": 0.5285,
-      "step": 1448
-    },
-    {
-      "epoch": 2.57,
-      "learning_rate": 1.9203568691740115e-05,
-      "loss": 0.5261,
-      "step": 1449
-    },
-    {
-      "epoch": 2.57,
-      "learning_rate": 1.920203643189054e-05,
-      "loss": 0.5207,
-      "step": 1450
-    },
-    {
-      "epoch": 2.58,
-      "learning_rate": 1.9200502760744114e-05,
-      "loss": 0.5221,
-      "step": 1451
-    },
-    {
-      "epoch": 2.58,
-      "learning_rate": 1.9198967678536054e-05,
-      "loss": 0.5136,
-      "step": 1452
-    },
-    {
-      "epoch": 2.58,
-      "learning_rate": 1.9197431185501786e-05,
-      "loss": 0.523,
-      "step": 1453
-    },
-    {
-      "epoch": 2.58,
-      "learning_rate": 1.9195893281876964e-05,
-      "loss": 0.517,
-      "step": 1454
-    },
-    {
-      "epoch": 2.58,
-      "learning_rate": 1.9194353967897453e-05,
-      "loss": 0.52,
-      "step": 1455
-    },
-    {
-      "epoch": 2.58,
-      "learning_rate": 1.9192813243799334e-05,
-      "loss": 0.5085,
-      "step": 1456
-    },
-    {
-      "epoch": 2.59,
-      "learning_rate": 1.91912711098189e-05,
-      "loss": 0.5271,
-      "step": 1457
-    },
-    {
-      "epoch": 2.59,
-      "learning_rate": 1.918972756619267e-05,
-      "loss": 0.5293,
-      "step": 1458
-    },
-    {
-      "epoch": 2.59,
-      "learning_rate": 1.9188182613157374e-05,
-      "loss": 0.5218,
-      "step": 1459
-    },
-    {
-      "epoch": 2.59,
-      "learning_rate": 1.9186636250949955e-05,
-      "loss": 0.5277,
-      "step": 1460
-    },
-    {
-      "epoch": 2.59,
-      "learning_rate": 1.9185088479807577e-05,
-      "loss": 0.5507,
-      "step": 1461
-    },
-    {
-      "epoch": 2.6,
-      "learning_rate": 1.9183539299967618e-05,
-      "loss": 0.5192,
-      "step": 1462
-    },
-    {
-      "epoch": 2.6,
-      "learning_rate": 1.918198871166767e-05,
-      "loss": 0.5328,
-      "step": 1463
-    },
-    {
-      "epoch": 2.6,
-      "learning_rate": 1.9180436715145552e-05,
-      "loss": 0.5118,
-      "step": 1464
-    },
-    {
-      "epoch": 2.6,
-      "learning_rate": 1.917888331063928e-05,
-      "loss": 0.5249,
-      "step": 1465
-    },
-    {
-      "epoch": 2.6,
-      "learning_rate": 1.91773284983871e-05,
-      "loss": 0.5057,
-      "step": 1466
-    },
-    {
-      "epoch": 2.6,
-      "learning_rate": 1.9175772278627477e-05,
-      "loss": 0.505,
-      "step": 1467
-    },
-    {
-      "epoch": 2.61,
-      "learning_rate": 1.9174214651599073e-05,
-      "loss": 0.5188,
-      "step": 1468
-    },
-    {
-      "epoch": 2.61,
-      "learning_rate": 1.917265561754079e-05,
-      "loss": 0.5304,
-      "step": 1469
-    },
-    {
-      "epoch": 2.61,
-      "learning_rate": 1.9171095176691726e-05,
-      "loss": 0.5249,
-      "step": 1470
-    },
-    {
-      "epoch": 2.61,
-      "learning_rate": 1.9169533329291204e-05,
-      "loss": 0.5171,
-      "step": 1471
-    },
-    {
-      "epoch": 2.61,
-      "learning_rate": 1.916797007557876e-05,
-      "loss": 0.5,
-      "step": 1472
-    },
-    {
-      "epoch": 2.62,
-      "learning_rate": 1.916640541579415e-05,
-      "loss": 0.5077,
-      "step": 1473
-    },
-    {
-      "epoch": 2.62,
-      "learning_rate": 1.916483935017734e-05,
-      "loss": 0.489,
-      "step": 1474
-    },
-    {
-      "epoch": 2.62,
-      "learning_rate": 1.916327187896852e-05,
-      "loss": 0.5192,
-      "step": 1475
-    },
-    {
-      "epoch": 2.62,
-      "learning_rate": 1.9161703002408082e-05,
-      "loss": 0.5146,
-      "step": 1476
-    },
-    {
-      "epoch": 2.62,
-      "learning_rate": 1.9160132720736642e-05,
-      "loss": 0.5264,
-      "step": 1477
-    },
-    {
-      "epoch": 2.62,
-      "learning_rate": 1.9158561034195035e-05,
-      "loss": 0.5184,
-      "step": 1478
-    },
-    {
-      "epoch": 2.63,
-      "learning_rate": 1.9156987943024305e-05,
-      "loss": 0.4975,
-      "step": 1479
-    },
-    {
-      "epoch": 2.63,
-      "learning_rate": 1.9155413447465715e-05,
-      "loss": 0.5221,
-      "step": 1480
-    },
-    {
-      "epoch": 2.63,
-      "learning_rate": 1.9153837547760744e-05,
-      "loss": 0.5344,
-      "step": 1481
-    },
-    {
-      "epoch": 2.63,
-      "learning_rate": 1.9152260244151077e-05,
-      "loss": 0.5065,
-      "step": 1482
-    },
-    {
-      "epoch": 2.63,
-      "learning_rate": 1.9150681536878628e-05,
-      "loss": 0.5161,
-      "step": 1483
-    },
-    {
-      "epoch": 2.63,
-      "learning_rate": 1.914910142618552e-05,
-      "loss": 0.5205,
-      "step": 1484
-    },
-    {
-      "epoch": 2.64,
-      "learning_rate": 1.9147519912314086e-05,
-      "loss": 0.5277,
-      "step": 1485
-    },
-    {
-      "epoch": 2.64,
-      "learning_rate": 1.9145936995506887e-05,
-      "loss": 0.5052,
-      "step": 1486
-    },
-    {
-      "epoch": 2.64,
-      "learning_rate": 1.9144352676006683e-05,
-      "loss": 0.5207,
-      "step": 1487
-    },
-    {
-      "epoch": 2.64,
-      "learning_rate": 1.9142766954056468e-05,
-      "loss": 0.5335,
-      "step": 1488
-    },
-    {
-      "epoch": 2.64,
-      "learning_rate": 1.914117982989943e-05,
-      "loss": 0.5144,
-      "step": 1489
-    },
-    {
-      "epoch": 2.65,
-      "learning_rate": 1.913959130377899e-05,
-      "loss": 0.5119,
-      "step": 1490
-    },
-    {
-      "epoch": 2.65,
-      "learning_rate": 1.9138001375938773e-05,
-      "loss": 0.4974,
-      "step": 1491
-    },
-    {
-      "epoch": 2.65,
-      "learning_rate": 1.9136410046622627e-05,
-      "loss": 0.5362,
-      "step": 1492
-    },
-    {
-      "epoch": 2.65,
-      "learning_rate": 1.9134817316074605e-05,
-      "loss": 0.4987,
-      "step": 1493
-    },
-    {
-      "epoch": 2.65,
-      "learning_rate": 1.913322318453899e-05,
-      "loss": 0.532,
-      "step": 1494
-    },
-    {
-      "epoch": 2.65,
-      "learning_rate": 1.913162765226026e-05,
-      "loss": 0.5136,
-      "step": 1495
-    },
-    {
-      "epoch": 2.66,
-      "learning_rate": 1.9130030719483125e-05,
-      "loss": 0.4961,
-      "step": 1496
-    },
-    {
-      "epoch": 2.66,
-      "learning_rate": 1.9128432386452503e-05,
-      "loss": 0.5273,
-      "step": 1497
-    },
-    {
-      "epoch": 2.66,
-      "learning_rate": 1.9126832653413523e-05,
-      "loss": 0.5151,
-      "step": 1498
-    },
-    {
-      "epoch": 2.66,
-      "learning_rate": 1.9125231520611534e-05,
-      "loss": 0.5047,
-      "step": 1499
-    },
-    {
-      "epoch": 2.66,
-      "learning_rate": 1.91236289882921e-05,
-      "loss": 0.5215,
-      "step": 1500
-    },
-    {
-      "epoch": 2.66,
-      "learning_rate": 1.9122025056701e-05,
-      "loss": 0.524,
-      "step": 1501
-    },
-    {
-      "epoch": 2.67,
-      "learning_rate": 1.9120419726084224e-05,
-      "loss": 0.5022,
-      "step": 1502
-    },
-    {
-      "epoch": 2.67,
-      "learning_rate": 1.9118812996687975e-05,
-      "loss": 0.5306,
-      "step": 1503
-    },
-    {
-      "epoch": 2.67,
-      "learning_rate": 1.9117204868758673e-05,
-      "loss": 0.5248,
-      "step": 1504
-    },
-    {
-      "epoch": 2.67,
-      "learning_rate": 1.911559534254296e-05,
-      "loss": 0.5184,
-      "step": 1505
-    },
-    {
-      "epoch": 2.67,
-      "learning_rate": 1.911398441828768e-05,
-      "loss": 0.5353,
-      "step": 1506
-    },
-    {
-      "epoch": 2.68,
-      "learning_rate": 1.91123720962399e-05,
-      "loss": 0.5315,
-      "step": 1507
-    },
-    {
-      "epoch": 2.68,
-      "learning_rate": 1.9110758376646897e-05,
-      "loss": 0.4985,
-      "step": 1508
-    },
-    {
-      "epoch": 2.68,
-      "learning_rate": 1.910914325975616e-05,
-      "loss": 0.4985,
-      "step": 1509
-    },
-    {
-      "epoch": 2.68,
-      "learning_rate": 1.9107526745815404e-05,
-      "loss": 0.5082,
-      "step": 1510
-    },
-    {
-      "epoch": 2.68,
-      "learning_rate": 1.9105908835072545e-05,
-      "loss": 0.5012,
-      "step": 1511
-    },
-    {
-      "epoch": 2.68,
-      "learning_rate": 1.910428952777572e-05,
-      "loss": 0.5385,
-      "step": 1512
-    },
-    {
-      "epoch": 2.69,
-      "learning_rate": 1.9102668824173278e-05,
-      "loss": 0.5141,
-      "step": 1513
-    },
-    {
-      "epoch": 2.69,
-      "learning_rate": 1.9101046724513783e-05,
-      "loss": 0.5233,
-      "step": 1514
-    },
-    {
-      "epoch": 2.69,
-      "learning_rate": 1.9099423229046015e-05,
-      "loss": 0.4893,
-      "step": 1515
-    },
-    {
-      "epoch": 2.69,
-      "learning_rate": 1.9097798338018963e-05,
-      "loss": 0.5126,
-      "step": 1516
-    },
-    {
-      "epoch": 2.69,
-      "learning_rate": 1.9096172051681832e-05,
-      "loss": 0.5307,
-      "step": 1517
-    },
-    {
-      "epoch": 2.7,
-      "learning_rate": 1.909454437028405e-05,
-      "loss": 0.5206,
-      "step": 1518
-    },
-    {
-      "epoch": 2.7,
-      "learning_rate": 1.9092915294075246e-05,
-      "loss": 0.4884,
-      "step": 1519
-    },
-    {
-      "epoch": 2.7,
-      "learning_rate": 1.909128482330526e-05,
-      "loss": 0.5523,
-      "step": 1520
-    },
-    {
-      "epoch": 2.7,
-      "learning_rate": 1.908965295822417e-05,
-      "loss": 0.5332,
-      "step": 1521
-    },
-    {
-      "epoch": 2.7,
-      "learning_rate": 1.908801969908224e-05,
-      "loss": 0.4861,
-      "step": 1522
-    },
-    {
-      "epoch": 2.7,
-      "learning_rate": 1.9086385046129963e-05,
-      "loss": 0.5346,
-      "step": 1523
-    },
-    {
-      "epoch": 2.71,
-      "learning_rate": 1.908474899961804e-05,
-      "loss": 0.504,
-      "step": 1524
-    },
-    {
-      "epoch": 2.71,
-      "learning_rate": 1.9083111559797394e-05,
-      "loss": 0.5326,
-      "step": 1525
-    },
-    {
-      "epoch": 2.71,
-      "learning_rate": 1.908147272691915e-05,
-      "loss": 0.5355,
-      "step": 1526
-    },
-    {
-      "epoch": 2.71,
-      "learning_rate": 1.907983250123465e-05,
-      "loss": 0.504,
-      "step": 1527
-    },
-    {
-      "epoch": 2.71,
-      "learning_rate": 1.907819088299546e-05,
-      "loss": 0.5167,
-      "step": 1528
-    },
-    {
-      "epoch": 2.71,
-      "learning_rate": 1.907654787245335e-05,
-      "loss": 0.4925,
-      "step": 1529
-    },
-    {
-      "epoch": 2.72,
-      "learning_rate": 1.9074903469860297e-05,
-      "loss": 0.5162,
-      "step": 1530
-    },
-    {
-      "epoch": 2.72,
-      "learning_rate": 1.9073257675468506e-05,
-      "loss": 0.5183,
-      "step": 1531
-    },
-    {
-      "epoch": 2.72,
-      "learning_rate": 1.907161048953039e-05,
-      "loss": 0.533,
-      "step": 1532
-    },
-    {
-      "epoch": 2.72,
-      "learning_rate": 1.906996191229857e-05,
-      "loss": 0.518,
-      "step": 1533
-    },
-    {
-      "epoch": 2.72,
-      "learning_rate": 1.906831194402589e-05,
-      "loss": 0.5321,
-      "step": 1534
-    },
-    {
-      "epoch": 2.73,
-      "learning_rate": 1.9066660584965397e-05,
-      "loss": 0.4977,
-      "step": 1535
-    },
-    {
-      "epoch": 2.73,
-      "learning_rate": 1.9065007835370358e-05,
-      "loss": 0.542,
-      "step": 1536
-    },
-    {
-      "epoch": 2.73,
-      "learning_rate": 1.9063353695494256e-05,
-      "loss": 0.525,
-      "step": 1537
-    },
-    {
-      "epoch": 2.73,
-      "learning_rate": 1.906169816559078e-05,
-      "loss": 0.5139,
-      "step": 1538
-    },
-    {
-      "epoch": 2.73,
-      "learning_rate": 1.906004124591383e-05,
-      "loss": 0.5358,
-      "step": 1539
-    },
-    {
-      "epoch": 2.73,
-      "learning_rate": 1.9058382936717532e-05,
-      "loss": 0.528,
-      "step": 1540
-    },
-    {
-      "epoch": 2.74,
-      "learning_rate": 1.9056723238256215e-05,
-      "loss": 0.507,
-      "step": 1541
-    },
-    {
-      "epoch": 2.74,
-      "learning_rate": 1.905506215078442e-05,
-      "loss": 0.4894,
-      "step": 1542
-    },
-    {
-      "epoch": 2.74,
-      "learning_rate": 1.905339967455691e-05,
-      "loss": 0.5128,
-      "step": 1543
-    },
-    {
-      "epoch": 2.74,
-      "learning_rate": 1.905173580982865e-05,
-      "loss": 0.5311,
-      "step": 1544
-    },
-    {
-      "epoch": 2.74,
-      "learning_rate": 1.905007055685483e-05,
-      "loss": 0.5141,
-      "step": 1545
-    },
-    {
-      "epoch": 2.74,
-      "learning_rate": 1.9048403915890838e-05,
-      "loss": 0.5293,
-      "step": 1546
-    },
-    {
-      "epoch": 2.75,
-      "learning_rate": 1.9046735887192288e-05,
-      "loss": 0.4963,
-      "step": 1547
-    },
-    {
-      "epoch": 2.75,
-      "learning_rate": 1.9045066471015e-05,
-      "loss": 0.5252,
-      "step": 1548
-    },
-    {
-      "epoch": 2.75,
-      "learning_rate": 1.9043395667615016e-05,
-      "loss": 0.4968,
-      "step": 1549
-    },
-    {
-      "epoch": 2.75,
-      "learning_rate": 1.9041723477248575e-05,
-      "loss": 0.5352,
-      "step": 1550
-    },
-    {
-      "epoch": 2.75,
-      "learning_rate": 1.904004990017214e-05,
-      "loss": 0.5339,
-      "step": 1551
-    },
-    {
-      "epoch": 2.76,
-      "learning_rate": 1.903837493664238e-05,
-      "loss": 0.4995,
-      "step": 1552
-    },
-    {
-      "epoch": 2.76,
-      "learning_rate": 1.9036698586916195e-05,
-      "loss": 0.5274,
-      "step": 1553
-    },
-    {
-      "epoch": 2.76,
-      "learning_rate": 1.9035020851250668e-05,
-      "loss": 0.5204,
-      "step": 1554
-    },
-    {
-      "epoch": 2.76,
-      "learning_rate": 1.9033341729903112e-05,
-      "loss": 0.5018,
-      "step": 1555
-    },
-    {
-      "epoch": 2.76,
-      "learning_rate": 1.9031661223131056e-05,
-      "loss": 0.4883,
-      "step": 1556
-    },
-    {
-      "epoch": 2.76,
-      "learning_rate": 1.902997933119223e-05,
-      "loss": 0.5363,
-      "step": 1557
-    },
-    {
-      "epoch": 2.77,
-      "learning_rate": 1.9028296054344588e-05,
-      "loss": 0.521,
-      "step": 1558
-    },
-    {
-      "epoch": 2.77,
-      "learning_rate": 1.9026611392846287e-05,
-      "loss": 0.5318,
-      "step": 1559
-    },
-    {
-      "epoch": 2.77,
-      "learning_rate": 1.9024925346955703e-05,
-      "loss": 0.5097,
-      "step": 1560
-    },
-    {
-      "epoch": 2.77,
-      "learning_rate": 1.9023237916931413e-05,
-      "loss": 0.4973,
-      "step": 1561
-    },
-    {
-      "epoch": 2.77,
-      "learning_rate": 1.9021549103032225e-05,
-      "loss": 0.5124,
-      "step": 1562
-    },
-    {
-      "epoch": 2.77,
-      "learning_rate": 1.9019858905517146e-05,
-      "loss": 0.5077,
-      "step": 1563
-    },
-    {
-      "epoch": 2.78,
-      "learning_rate": 1.9018167324645394e-05,
-      "loss": 0.5394,
-      "step": 1564
-    },
-    {
-      "epoch": 2.78,
-      "learning_rate": 1.9016474360676406e-05,
-      "loss": 0.5347,
-      "step": 1565
-    },
-    {
-      "epoch": 2.78,
-      "learning_rate": 1.9014780013869827e-05,
-      "loss": 0.5146,
-      "step": 1566
-    },
-    {
-      "epoch": 2.78,
-      "learning_rate": 1.9013084284485518e-05,
-      "loss": 0.5032,
-      "step": 1567
-    },
-    {
-      "epoch": 2.78,
-      "learning_rate": 1.9011387172783544e-05,
-      "loss": 0.4874,
-      "step": 1568
-    },
-    {
-      "epoch": 2.79,
-      "learning_rate": 1.900968867902419e-05,
-      "loss": 0.5317,
-      "step": 1569
-    },
-    {
-      "epoch": 2.79,
-      "learning_rate": 1.9007988803467954e-05,
-      "loss": 0.4973,
-      "step": 1570
-    },
-    {
-      "epoch": 2.79,
-      "learning_rate": 1.9006287546375546e-05,
-      "loss": 0.5024,
-      "step": 1571
-    },
-    {
-      "epoch": 2.79,
-      "learning_rate": 1.900458490800787e-05,
-      "loss": 0.5314,
-      "step": 1572
-    },
-    {
-      "epoch": 2.79,
-      "learning_rate": 1.900288088862607e-05,
-      "loss": 0.5158,
-      "step": 1573
-    },
-    {
-      "epoch": 2.79,
-      "learning_rate": 1.9001175488491475e-05,
-      "loss": 0.5146,
-      "step": 1574
-    },
-    {
-      "epoch": 2.8,
-      "learning_rate": 1.899946870786565e-05,
-      "loss": 0.5237,
-      "step": 1575
-    },
-    {
-      "epoch": 2.8,
-      "learning_rate": 1.8997760547010354e-05,
-      "loss": 0.5208,
-      "step": 1576
-    },
-    {
-      "epoch": 2.8,
-      "learning_rate": 1.8996051006187567e-05,
-      "loss": 0.5112,
-      "step": 1577
-    },
-    {
-      "epoch": 2.8,
-      "learning_rate": 1.8994340085659474e-05,
-      "loss": 0.5243,
-      "step": 1578
-    },
-    {
-      "epoch": 2.8,
-      "learning_rate": 1.8992627785688482e-05,
-      "loss": 0.5169,
-      "step": 1579
-    },
-    {
-      "epoch": 2.81,
-      "learning_rate": 1.89909141065372e-05,
-      "loss": 0.5122,
-      "step": 1580
-    },
-    {
-      "epoch": 2.81,
-      "learning_rate": 1.8989199048468443e-05,
-      "loss": 0.5113,
-      "step": 1581
-    },
-    {
-      "epoch": 2.81,
-      "learning_rate": 1.8987482611745255e-05,
-      "loss": 0.5029,
-      "step": 1582
-    },
-    {
-      "epoch": 2.81,
-      "learning_rate": 1.8985764796630884e-05,
-      "loss": 0.5101,
-      "step": 1583
-    },
-    {
-      "epoch": 2.81,
-      "learning_rate": 1.8984045603388778e-05,
-      "loss": 0.5059,
-      "step": 1584
-    },
-    {
-      "epoch": 2.81,
-      "learning_rate": 1.8982325032282616e-05,
-      "loss": 0.487,
-      "step": 1585
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 1.898060308357627e-05,
-      "loss": 0.5076,
-      "step": 1586
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 1.8978879757533838e-05,
-      "loss": 0.5287,
-      "step": 1587
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 1.8977155054419626e-05,
-      "loss": 0.5286,
-      "step": 1588
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 1.8975428974498136e-05,
-      "loss": 0.5213,
-      "step": 1589
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 1.89737015180341e-05,
-      "loss": 0.5303,
-      "step": 1590
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 1.8971972685292454e-05,
-      "loss": 0.5215,
-      "step": 1591
-    },
-    {
-      "epoch": 2.83,
-      "learning_rate": 1.897024247653835e-05,
-      "loss": 0.5144,
-      "step": 1592
-    },
-    {
-      "epoch": 2.83,
-      "learning_rate": 1.8968510892037136e-05,
-      "loss": 0.5332,
-      "step": 1593
-    },
-    {
-      "epoch": 2.83,
-      "learning_rate": 1.896677793205439e-05,
-      "loss": 0.5447,
-      "step": 1594
-    },
-    {
-      "epoch": 2.83,
-      "learning_rate": 1.8965043596855894e-05,
-      "loss": 0.5082,
-      "step": 1595
-    },
-    {
-      "epoch": 2.83,
-      "learning_rate": 1.8963307886707633e-05,
-      "loss": 0.5209,
-      "step": 1596
-    },
-    {
-      "epoch": 2.84,
-      "learning_rate": 1.8961570801875808e-05,
-      "loss": 0.5248,
-      "step": 1597
-    },
-    {
-      "epoch": 2.84,
-      "learning_rate": 1.8959832342626838e-05,
-      "loss": 0.5091,
-      "step": 1598
-    },
-    {
-      "epoch": 2.84,
-      "learning_rate": 1.8958092509227347e-05,
-      "loss": 0.5102,
-      "step": 1599
-    },
-    {
-      "epoch": 2.84,
-      "learning_rate": 1.8956351301944162e-05,
-      "loss": 0.4996,
-      "step": 1600
-    },
-    {
-      "epoch": 2.84,
-      "learning_rate": 1.8954608721044337e-05,
-      "loss": 0.5119,
-      "step": 1601
-    },
-    {
-      "epoch": 2.84,
-      "learning_rate": 1.895286476679512e-05,
-      "loss": 0.5284,
-      "step": 1602
-    },
-    {
-      "epoch": 2.85,
-      "learning_rate": 1.8951119439463984e-05,
-      "loss": 0.5076,
-      "step": 1603
-    },
-    {
-      "epoch": 2.85,
-      "learning_rate": 1.8949372739318602e-05,
-      "loss": 0.5298,
-      "step": 1604
-    },
-    {
-      "epoch": 2.85,
-      "learning_rate": 1.8947624666626866e-05,
-      "loss": 0.5251,
-      "step": 1605
-    },
-    {
-      "epoch": 2.85,
-      "learning_rate": 1.8945875221656873e-05,
-      "loss": 0.5097,
-      "step": 1606
-    },
-    {
-      "epoch": 2.85,
-      "learning_rate": 1.8944124404676925e-05,
-      "loss": 0.518,
-      "step": 1607
-    },
-    {
-      "epoch": 2.85,
-      "learning_rate": 1.894237221595555e-05,
-      "loss": 0.5291,
-      "step": 1608
-    },
-    {
-      "epoch": 2.86,
-      "learning_rate": 1.8940618655761475e-05,
-      "loss": 0.5227,
-      "step": 1609
-    },
-    {
-      "epoch": 2.86,
-      "learning_rate": 1.8938863724363638e-05,
-      "loss": 0.5194,
-      "step": 1610
-    },
-    {
-      "epoch": 2.86,
-      "learning_rate": 1.893710742203119e-05,
-      "loss": 0.4958,
-      "step": 1611
-    },
-    {
-      "epoch": 2.86,
-      "learning_rate": 1.893534974903349e-05,
-      "loss": 0.5098,
-      "step": 1612
-    },
-    {
-      "epoch": 2.86,
-      "learning_rate": 1.893359070564011e-05,
-      "loss": 0.5077,
-      "step": 1613
-    },
-    {
-      "epoch": 2.87,
-      "learning_rate": 1.893183029212083e-05,
-      "loss": 0.4954,
-      "step": 1614
-    },
-    {
-      "epoch": 2.87,
-      "learning_rate": 1.8930068508745643e-05,
-      "loss": 0.5246,
-      "step": 1615
-    },
-    {
-      "epoch": 2.87,
-      "learning_rate": 1.8928305355784748e-05,
-      "loss": 0.5041,
-      "step": 1616
-    },
-    {
-      "epoch": 2.87,
-      "learning_rate": 1.8926540833508557e-05,
-      "loss": 0.5141,
-      "step": 1617
-    },
-    {
-      "epoch": 2.87,
-      "learning_rate": 1.8924774942187688e-05,
-      "loss": 0.5089,
-      "step": 1618
-    },
-    {
-      "epoch": 2.87,
-      "learning_rate": 1.8923007682092977e-05,
-      "loss": 0.5128,
-      "step": 1619
-    },
-    {
-      "epoch": 2.88,
-      "learning_rate": 1.8921239053495465e-05,
-      "loss": 0.5053,
-      "step": 1620
-    },
-    {
-      "epoch": 2.88,
-      "learning_rate": 1.8919469056666395e-05,
-      "loss": 0.5125,
-      "step": 1621
-    },
-    {
-      "epoch": 2.88,
-      "learning_rate": 1.8917697691877235e-05,
-      "loss": 0.5265,
-      "step": 1622
-    },
-    {
-      "epoch": 2.88,
-      "learning_rate": 1.8915924959399656e-05,
-      "loss": 0.5241,
-      "step": 1623
-    },
-    {
-      "epoch": 2.88,
-      "learning_rate": 1.8914150859505534e-05,
-      "loss": 0.4839,
-      "step": 1624
-    },
-    {
-      "epoch": 2.89,
-      "learning_rate": 1.8912375392466957e-05,
-      "loss": 0.5179,
-      "step": 1625
-    },
-    {
-      "epoch": 2.89,
-      "learning_rate": 1.8910598558556232e-05,
-      "loss": 0.5077,
-      "step": 1626
-    },
-    {
-      "epoch": 2.89,
-      "learning_rate": 1.8908820358045864e-05,
-      "loss": 0.5198,
-      "step": 1627
-    },
-    {
-      "epoch": 2.89,
-      "learning_rate": 1.8907040791208577e-05,
-      "loss": 0.5172,
-      "step": 1628
-    },
-    {
-      "epoch": 2.89,
-      "learning_rate": 1.8905259858317287e-05,
-      "loss": 0.5241,
-      "step": 1629
-    },
-    {
-      "epoch": 2.89,
-      "learning_rate": 1.890347755964514e-05,
-      "loss": 0.5119,
-      "step": 1630
-    },
-    {
-      "epoch": 2.9,
-      "learning_rate": 1.8901693895465488e-05,
-      "loss": 0.5284,
-      "step": 1631
-    },
-    {
-      "epoch": 2.9,
-      "learning_rate": 1.889990886605188e-05,
-      "loss": 0.5079,
-      "step": 1632
-    },
-    {
-      "epoch": 2.9,
-      "learning_rate": 1.8898122471678087e-05,
-      "loss": 0.4953,
-      "step": 1633
-    },
-    {
-      "epoch": 2.9,
-      "learning_rate": 1.8896334712618082e-05,
-      "loss": 0.5078,
-      "step": 1634
-    },
-    {
-      "epoch": 2.9,
-      "learning_rate": 1.889454558914605e-05,
-      "loss": 0.525,
-      "step": 1635
-    },
-    {
-      "epoch": 2.9,
-      "learning_rate": 1.8892755101536387e-05,
-      "loss": 0.5106,
-      "step": 1636
-    },
-    {
-      "epoch": 2.91,
-      "learning_rate": 1.889096325006369e-05,
-      "loss": 0.5095,
-      "step": 1637
-    },
-    {
-      "epoch": 2.91,
-      "learning_rate": 1.8889170035002783e-05,
-      "loss": 0.5205,
-      "step": 1638
-    },
-    {
-      "epoch": 2.91,
-      "learning_rate": 1.8887375456628675e-05,
-      "loss": 0.5236,
-      "step": 1639
-    },
-    {
-      "epoch": 2.91,
-      "learning_rate": 1.8885579515216606e-05,
-      "loss": 0.4948,
-      "step": 1640
-    },
-    {
-      "epoch": 2.91,
-      "learning_rate": 1.888378221104201e-05,
-      "loss": 0.5143,
-      "step": 1641
-    },
-    {
-      "epoch": 2.92,
-      "learning_rate": 1.888198354438054e-05,
-      "loss": 0.5058,
-      "step": 1642
-    },
-    {
-      "epoch": 2.92,
-      "learning_rate": 1.888018351550805e-05,
-      "loss": 0.5215,
-      "step": 1643
-    },
-    {
-      "epoch": 2.92,
-      "learning_rate": 1.887838212470061e-05,
-      "loss": 0.5304,
-      "step": 1644
-    },
-    {
-      "epoch": 2.92,
-      "learning_rate": 1.887657937223449e-05,
-      "loss": 0.5211,
-      "step": 1645
-    },
-    {
-      "epoch": 2.92,
-      "learning_rate": 1.8874775258386183e-05,
-      "loss": 0.5015,
-      "step": 1646
-    },
-    {
-      "epoch": 2.92,
-      "learning_rate": 1.8872969783432376e-05,
-      "loss": 0.498,
-      "step": 1647
-    },
-    {
-      "epoch": 2.93,
-      "learning_rate": 1.887116294764997e-05,
-      "loss": 0.5117,
-      "step": 1648
-    },
-    {
-      "epoch": 2.93,
-      "learning_rate": 1.8869354751316083e-05,
-      "loss": 0.5188,
-      "step": 1649
-    },
-    {
-      "epoch": 2.93,
-      "learning_rate": 1.8867545194708024e-05,
-      "loss": 0.4955,
-      "step": 1650
-    },
-    {
-      "epoch": 2.93,
-      "learning_rate": 1.8865734278103323e-05,
-      "loss": 0.4956,
-      "step": 1651
-    },
-    {
-      "epoch": 2.93,
-      "learning_rate": 1.8863922001779726e-05,
-      "loss": 0.5262,
-      "step": 1652
-    },
-    {
-      "epoch": 2.93,
-      "learning_rate": 1.886210836601517e-05,
-      "loss": 0.5164,
-      "step": 1653
-    },
-    {
-      "epoch": 2.94,
-      "learning_rate": 1.8860293371087808e-05,
-      "loss": 0.4965,
-      "step": 1654
-    },
-    {
-      "epoch": 2.94,
-      "learning_rate": 1.8858477017276002e-05,
-      "loss": 0.5007,
-      "step": 1655
-    },
-    {
-      "epoch": 2.94,
-      "learning_rate": 1.8856659304858328e-05,
-      "loss": 0.5001,
-      "step": 1656
-    },
-    {
-      "epoch": 2.94,
-      "learning_rate": 1.8854840234113558e-05,
-      "loss": 0.5209,
-      "step": 1657
-    },
-    {
-      "epoch": 2.94,
-      "learning_rate": 1.8853019805320684e-05,
-      "loss": 0.523,
-      "step": 1658
-    },
-    {
-      "epoch": 2.95,
-      "learning_rate": 1.8851198018758898e-05,
-      "loss": 0.4789,
-      "step": 1659
-    },
-    {
-      "epoch": 2.95,
-      "learning_rate": 1.8849374874707602e-05,
-      "loss": 0.5222,
-      "step": 1660
-    },
-    {
-      "epoch": 2.95,
-      "learning_rate": 1.8847550373446414e-05,
-      "loss": 0.522,
-      "step": 1661
-    },
-    {
-      "epoch": 2.95,
-      "learning_rate": 1.8845724515255147e-05,
-      "loss": 0.5272,
-      "step": 1662
-    },
-    {
-      "epoch": 2.95,
-      "learning_rate": 1.8843897300413837e-05,
-      "loss": 0.4953,
-      "step": 1663
-    },
-    {
-      "epoch": 2.95,
-      "learning_rate": 1.8842068729202712e-05,
-      "loss": 0.5186,
-      "step": 1664
-    },
-    {
-      "epoch": 2.96,
-      "learning_rate": 1.884023880190222e-05,
-      "loss": 0.4997,
-      "step": 1665
-    },
-    {
-      "epoch": 2.96,
-      "learning_rate": 1.8838407518793008e-05,
-      "loss": 0.5208,
-      "step": 1666
-    },
-    {
-      "epoch": 2.96,
-      "learning_rate": 1.8836574880155948e-05,
-      "loss": 0.528,
-      "step": 1667
-    },
-    {
-      "epoch": 2.96,
-      "learning_rate": 1.8834740886272098e-05,
-      "loss": 0.4996,
-      "step": 1668
-    },
-    {
-      "epoch": 2.96,
-      "learning_rate": 1.8832905537422734e-05,
-      "loss": 0.5242,
-      "step": 1669
-    },
-    {
-      "epoch": 2.96,
-      "learning_rate": 1.8831068833889343e-05,
-      "loss": 0.516,
-      "step": 1670
-    },
-    {
-      "epoch": 2.97,
-      "learning_rate": 1.8829230775953616e-05,
-      "loss": 0.5285,
-      "step": 1671
-    },
-    {
-      "epoch": 2.97,
-      "learning_rate": 1.882739136389745e-05,
-      "loss": 0.4988,
-      "step": 1672
-    },
-    {
-      "epoch": 2.97,
-      "learning_rate": 1.8825550598002955e-05,
-      "loss": 0.5152,
-      "step": 1673
-    },
-    {
-      "epoch": 2.97,
-      "learning_rate": 1.8823708478552442e-05,
-      "loss": 0.5174,
-      "step": 1674
-    },
-    {
-      "epoch": 2.97,
-      "learning_rate": 1.882186500582843e-05,
-      "loss": 0.505,
-      "step": 1675
-    },
-    {
-      "epoch": 2.98,
-      "learning_rate": 1.882002018011366e-05,
-      "loss": 0.5075,
-      "step": 1676
-    },
-    {
-      "epoch": 2.98,
-      "learning_rate": 1.8818174001691055e-05,
-      "loss": 0.4984,
-      "step": 1677
-    },
-    {
-      "epoch": 2.98,
-      "learning_rate": 1.881632647084377e-05,
-      "loss": 0.5235,
-      "step": 1678
-    },
-    {
-      "epoch": 2.98,
-      "learning_rate": 1.8814477587855154e-05,
-      "loss": 0.5341,
-      "step": 1679
-    },
-    {
-      "epoch": 2.98,
-      "learning_rate": 1.8812627353008764e-05,
-      "loss": 0.5097,
-      "step": 1680
-    },
-    {
-      "epoch": 2.98,
-      "learning_rate": 1.8810775766588366e-05,
-      "loss": 0.5183,
-      "step": 1681
-    },
-    {
-      "epoch": 2.99,
-      "learning_rate": 1.880892282887794e-05,
-      "loss": 0.5308,
-      "step": 1682
-    },
-    {
-      "epoch": 2.99,
-      "learning_rate": 1.880706854016166e-05,
-      "loss": 0.5148,
-      "step": 1683
-    },
-    {
-      "epoch": 2.99,
-      "learning_rate": 1.880521290072392e-05,
-      "loss": 0.5069,
-      "step": 1684
-    },
-    {
-      "epoch": 2.99,
-      "learning_rate": 1.880335591084931e-05,
-      "loss": 0.5137,
-      "step": 1685
-    },
-    {
-      "epoch": 2.99,
-      "learning_rate": 1.8801497570822636e-05,
-      "loss": 0.5048,
-      "step": 1686
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 1.879963788092891e-05,
-      "loss": 0.5292,
-      "step": 1687
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 1.879777684145334e-05,
-      "loss": 0.5147,
-      "step": 1688
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 1.8795914452681362e-05,
-      "loss": 0.499,
-      "step": 1689
-    },
-    {
-      "epoch": 3.0,
-      "eval_loss": 0.541254460811615,
-      "eval_runtime": 3448.6866,
-      "eval_samples_per_second": 4.16,
-      "eval_steps_per_second": 1.04,
-      "step": 1689
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 1.8794050714898596e-05,
-      "loss": 0.5265,
-      "step": 1690
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 1.8792185628390886e-05,
-      "loss": 0.4948,
-      "step": 1691
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 1.879031919344427e-05,
-      "loss": 0.5133,
-      "step": 1692
-    },
-    {
-      "epoch": 3.01,
-      "learning_rate": 1.8788451410345007e-05,
-      "loss": 0.502,
-      "step": 1693
-    },
-    {
-      "epoch": 3.01,
-      "learning_rate": 1.878658227937955e-05,
-      "loss": 0.5059,
-      "step": 1694
-    },
-    {
-      "epoch": 3.01,
-      "learning_rate": 1.8784711800834564e-05,
-      "loss": 0.5083,
-      "step": 1695
-    },
-    {
-      "epoch": 3.01,
-      "learning_rate": 1.8782839974996924e-05,
-      "loss": 0.5116,
-      "step": 1696
-    },
-    {
-      "epoch": 3.01,
-      "learning_rate": 1.87809668021537e-05,
-      "loss": 0.4733,
-      "step": 1697
-    },
-    {
-      "epoch": 3.01,
-      "learning_rate": 1.8779092282592186e-05,
-      "loss": 0.5231,
-      "step": 1698
-    },
-    {
-      "epoch": 3.02,
-      "learning_rate": 1.8777216416599868e-05,
-      "loss": 0.5013,
-      "step": 1699
-    },
-    {
-      "epoch": 3.02,
-      "learning_rate": 1.8775339204464446e-05,
-      "loss": 0.4987,
-      "step": 1700
-    },
-    {
-      "epoch": 3.02,
-      "learning_rate": 1.877346064647382e-05,
-      "loss": 0.4876,
-      "step": 1701
-    },
-    {
-      "epoch": 3.02,
-      "learning_rate": 1.87715807429161e-05,
-      "loss": 0.4903,
-      "step": 1702
-    },
-    {
-      "epoch": 3.02,
-      "learning_rate": 1.8769699494079612e-05,
-      "loss": 0.5009,
-      "step": 1703
-    },
-    {
-      "epoch": 3.03,
-      "learning_rate": 1.876781690025287e-05,
-      "loss": 0.5019,
-      "step": 1704
-    },
-    {
-      "epoch": 3.03,
-      "learning_rate": 1.8765932961724607e-05,
-      "loss": 0.4904,
-      "step": 1705
-    },
-    {
-      "epoch": 3.03,
-      "learning_rate": 1.8764047678783758e-05,
-      "loss": 0.4938,
-      "step": 1706
-    },
-    {
-      "epoch": 3.03,
-      "learning_rate": 1.8762161051719464e-05,
-      "loss": 0.5264,
-      "step": 1707
-    },
-    {
-      "epoch": 3.03,
-      "learning_rate": 1.8760273080821072e-05,
-      "loss": 0.5037,
-      "step": 1708
-    },
-    {
-      "epoch": 3.03,
-      "learning_rate": 1.8758383766378138e-05,
-      "loss": 0.5122,
-      "step": 1709
-    },
-    {
-      "epoch": 3.04,
-      "learning_rate": 1.8756493108680426e-05,
-      "loss": 0.4981,
-      "step": 1710
-    },
-    {
-      "epoch": 3.04,
-      "learning_rate": 1.8754601108017895e-05,
-      "loss": 0.4948,
-      "step": 1711
-    },
-    {
-      "epoch": 3.04,
-      "learning_rate": 1.875270776468072e-05,
-      "loss": 0.4999,
-      "step": 1712
-    },
-    {
-      "epoch": 3.04,
-      "learning_rate": 1.8750813078959282e-05,
-      "loss": 0.5133,
-      "step": 1713
-    },
-    {
-      "epoch": 3.04,
-      "learning_rate": 1.8748917051144164e-05,
-      "loss": 0.4899,
-      "step": 1714
-    },
-    {
-      "epoch": 3.04,
-      "learning_rate": 1.874701968152615e-05,
-      "loss": 0.5277,
-      "step": 1715
-    },
-    {
-      "epoch": 3.05,
-      "learning_rate": 1.8745120970396238e-05,
-      "loss": 0.4941,
-      "step": 1716
-    },
-    {
-      "epoch": 3.05,
-      "learning_rate": 1.8743220918045633e-05,
-      "loss": 0.5253,
-      "step": 1717
-    },
-    {
-      "epoch": 3.05,
-      "learning_rate": 1.8741319524765745e-05,
-      "loss": 0.5126,
-      "step": 1718
-    },
-    {
-      "epoch": 3.05,
-      "learning_rate": 1.8739416790848177e-05,
-      "loss": 0.5295,
-      "step": 1719
-    },
-    {
-      "epoch": 3.05,
-      "learning_rate": 1.873751271658475e-05,
-      "loss": 0.5035,
-      "step": 1720
-    },
-    {
-      "epoch": 3.06,
-      "learning_rate": 1.8735607302267495e-05,
-      "loss": 0.5238,
-      "step": 1721
-    },
-    {
-      "epoch": 3.06,
-      "learning_rate": 1.8733700548188634e-05,
-      "loss": 0.5005,
-      "step": 1722
-    },
-    {
-      "epoch": 3.06,
-      "learning_rate": 1.87317924546406e-05,
-      "loss": 0.5389,
-      "step": 1723
-    },
-    {
-      "epoch": 3.06,
-      "learning_rate": 1.872988302191604e-05,
-      "loss": 0.4926,
-      "step": 1724
-    },
-    {
-      "epoch": 3.06,
-      "learning_rate": 1.87279722503078e-05,
-      "loss": 0.5264,
-      "step": 1725
-    },
-    {
-      "epoch": 3.06,
-      "learning_rate": 1.8726060140108927e-05,
-      "loss": 0.5183,
-      "step": 1726
-    },
-    {
-      "epoch": 3.07,
-      "learning_rate": 1.8724146691612676e-05,
-      "loss": 0.513,
-      "step": 1727
-    },
-    {
-      "epoch": 3.07,
-      "learning_rate": 1.8722231905112512e-05,
-      "loss": 0.5264,
-      "step": 1728
-    },
-    {
-      "epoch": 3.07,
-      "learning_rate": 1.87203157809021e-05,
-      "loss": 0.4934,
-      "step": 1729
-    },
-    {
-      "epoch": 3.07,
-      "learning_rate": 1.8718398319275312e-05,
-      "loss": 0.5163,
-      "step": 1730
-    },
-    {
-      "epoch": 3.07,
-      "learning_rate": 1.8716479520526227e-05,
-      "loss": 0.4983,
-      "step": 1731
-    },
-    {
-      "epoch": 3.08,
-      "learning_rate": 1.8714559384949125e-05,
-      "loss": 0.4983,
-      "step": 1732
-    },
-    {
-      "epoch": 3.08,
-      "learning_rate": 1.8712637912838492e-05,
-      "loss": 0.5076,
-      "step": 1733
-    },
-    {
-      "epoch": 3.08,
-      "learning_rate": 1.8710715104489027e-05,
-      "loss": 0.5114,
-      "step": 1734
-    },
-    {
-      "epoch": 3.08,
-      "learning_rate": 1.8708790960195616e-05,
-      "loss": 0.512,
-      "step": 1735
-    },
-    {
-      "epoch": 3.08,
-      "learning_rate": 1.870686548025337e-05,
-      "loss": 0.4942,
-      "step": 1736
-    },
-    {
-      "epoch": 3.08,
-      "learning_rate": 1.870493866495759e-05,
-      "loss": 0.5146,
-      "step": 1737
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 1.8703010514603792e-05,
-      "loss": 0.5092,
-      "step": 1738
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 1.8701081029487688e-05,
-      "loss": 0.5179,
-      "step": 1739
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 1.8699150209905205e-05,
-      "loss": 0.5075,
-      "step": 1740
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 1.8697218056152464e-05,
-      "loss": 0.5269,
-      "step": 1741
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 1.8695284568525797e-05,
-      "loss": 0.4937,
-      "step": 1742
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 1.8693349747321737e-05,
-      "loss": 0.5213,
-      "step": 1743
-    },
-    {
-      "epoch": 3.1,
-      "learning_rate": 1.8691413592837026e-05,
-      "loss": 0.4982,
-      "step": 1744
-    },
-    {
-      "epoch": 3.1,
-      "learning_rate": 1.8689476105368607e-05,
-      "loss": 0.511,
-      "step": 1745
-    },
-    {
-      "epoch": 3.1,
-      "learning_rate": 1.8687537285213627e-05,
-      "loss": 0.5059,
-      "step": 1746
-    },
-    {
-      "epoch": 3.1,
-      "learning_rate": 1.868559713266944e-05,
-      "loss": 0.4999,
-      "step": 1747
-    },
-    {
-      "epoch": 3.1,
-      "learning_rate": 1.8683655648033604e-05,
-      "loss": 0.4762,
-      "step": 1748
-    },
-    {
-      "epoch": 3.11,
-      "learning_rate": 1.868171283160388e-05,
-      "loss": 0.4997,
-      "step": 1749
-    },
-    {
-      "epoch": 3.11,
-      "learning_rate": 1.867976868367823e-05,
-      "loss": 0.5005,
-      "step": 1750
-    },
-    {
-      "epoch": 3.11,
-      "learning_rate": 1.8677823204554834e-05,
-      "loss": 0.4991,
-      "step": 1751
-    },
-    {
-      "epoch": 3.11,
-      "learning_rate": 1.8675876394532053e-05,
-      "loss": 0.5093,
-      "step": 1752
-    },
-    {
-      "epoch": 3.11,
-      "learning_rate": 1.867392825390847e-05,
-      "loss": 0.4984,
-      "step": 1753
-    },
-    {
-      "epoch": 3.11,
-      "learning_rate": 1.867197878298287e-05,
-      "loss": 0.502,
-      "step": 1754
-    },
-    {
-      "epoch": 3.12,
-      "learning_rate": 1.867002798205424e-05,
-      "loss": 0.506,
-      "step": 1755
-    },
-    {
-      "epoch": 3.12,
-      "learning_rate": 1.8668075851421767e-05,
-      "loss": 0.5128,
-      "step": 1756
-    },
-    {
-      "epoch": 3.12,
-      "learning_rate": 1.8666122391384846e-05,
-      "loss": 0.5193,
-      "step": 1757
-    },
-    {
-      "epoch": 3.12,
-      "learning_rate": 1.866416760224307e-05,
-      "loss": 0.4996,
-      "step": 1758
-    },
-    {
-      "epoch": 3.12,
-      "learning_rate": 1.866221148429625e-05,
-      "loss": 0.5252,
-      "step": 1759
-    },
-    {
-      "epoch": 3.12,
-      "learning_rate": 1.866025403784439e-05,
-      "loss": 0.5126,
-      "step": 1760
-    },
-    {
-      "epoch": 3.13,
-      "learning_rate": 1.8658295263187692e-05,
-      "loss": 0.4973,
-      "step": 1761
-    },
-    {
-      "epoch": 3.13,
-      "learning_rate": 1.8656335160626577e-05,
-      "loss": 0.5157,
-      "step": 1762
-    },
-    {
-      "epoch": 3.13,
-      "learning_rate": 1.8654373730461656e-05,
-      "loss": 0.5092,
-      "step": 1763
-    },
-    {
-      "epoch": 3.13,
-      "learning_rate": 1.865241097299375e-05,
-      "loss": 0.5269,
-      "step": 1764
-    },
-    {
-      "epoch": 3.13,
-      "learning_rate": 1.8650446888523893e-05,
-      "loss": 0.4971,
-      "step": 1765
-    },
-    {
-      "epoch": 3.14,
-      "learning_rate": 1.86484814773533e-05,
-      "loss": 0.5104,
-      "step": 1766
-    },
-    {
-      "epoch": 3.14,
-      "learning_rate": 1.8646514739783404e-05,
-      "loss": 0.5274,
-      "step": 1767
-    },
-    {
-      "epoch": 3.14,
-      "learning_rate": 1.8644546676115847e-05,
-      "loss": 0.5139,
-      "step": 1768
-    },
-    {
-      "epoch": 3.14,
-      "learning_rate": 1.864257728665246e-05,
-      "loss": 0.5208,
-      "step": 1769
-    },
-    {
-      "epoch": 3.14,
-      "learning_rate": 1.8640606571695285e-05,
-      "loss": 0.5247,
-      "step": 1770
-    },
-    {
-      "epoch": 3.14,
-      "learning_rate": 1.8638634531546566e-05,
-      "loss": 0.5104,
-      "step": 1771
-    },
-    {
-      "epoch": 3.15,
-      "learning_rate": 1.863666116650875e-05,
-      "loss": 0.4942,
-      "step": 1772
-    },
-    {
-      "epoch": 3.15,
-      "learning_rate": 1.8634686476884497e-05,
-      "loss": 0.5074,
-      "step": 1773
-    },
-    {
-      "epoch": 3.15,
-      "learning_rate": 1.8632710462976648e-05,
-      "loss": 0.5373,
-      "step": 1774
-    },
-    {
-      "epoch": 3.15,
-      "learning_rate": 1.8630733125088265e-05,
-      "loss": 0.4973,
-      "step": 1775
-    },
-    {
-      "epoch": 3.15,
-      "learning_rate": 1.862875446352261e-05,
-      "loss": 0.4955,
-      "step": 1776
-    },
-    {
-      "epoch": 3.15,
-      "learning_rate": 1.8626774478583148e-05,
-      "loss": 0.5328,
-      "step": 1777
-    },
-    {
-      "epoch": 3.16,
-      "learning_rate": 1.862479317057354e-05,
-      "loss": 0.5172,
-      "step": 1778
-    },
-    {
-      "epoch": 3.16,
-      "learning_rate": 1.8622810539797656e-05,
-      "loss": 0.5037,
-      "step": 1779
-    },
-    {
-      "epoch": 3.16,
-      "learning_rate": 1.862082658655957e-05,
-      "loss": 0.51,
-      "step": 1780
-    },
-    {
-      "epoch": 3.16,
-      "learning_rate": 1.8618841311163554e-05,
-      "loss": 0.5202,
-      "step": 1781
-    },
-    {
-      "epoch": 3.16,
-      "learning_rate": 1.8616854713914088e-05,
-      "loss": 0.5144,
-      "step": 1782
-    },
-    {
-      "epoch": 3.17,
-      "learning_rate": 1.8614866795115852e-05,
-      "loss": 0.514,
-      "step": 1783
-    },
-    {
-      "epoch": 3.17,
-      "learning_rate": 1.8612877555073728e-05,
-      "loss": 0.5252,
-      "step": 1784
-    },
-    {
-      "epoch": 3.17,
-      "learning_rate": 1.8610886994092802e-05,
-      "loss": 0.4957,
-      "step": 1785
-    },
-    {
-      "epoch": 3.17,
-      "learning_rate": 1.860889511247836e-05,
-      "loss": 0.4931,
-      "step": 1786
-    },
-    {
-      "epoch": 3.17,
-      "learning_rate": 1.8606901910535894e-05,
-      "loss": 0.5059,
-      "step": 1787
-    },
-    {
-      "epoch": 3.17,
-      "learning_rate": 1.8604907388571097e-05,
-      "loss": 0.5015,
-      "step": 1788
-    },
-    {
-      "epoch": 3.18,
-      "learning_rate": 1.860291154688987e-05,
-      "loss": 0.5269,
-      "step": 1789
-    },
-    {
-      "epoch": 3.18,
-      "learning_rate": 1.8600914385798305e-05,
-      "loss": 0.5212,
-      "step": 1790
-    },
-    {
-      "epoch": 3.18,
-      "learning_rate": 1.8598915905602704e-05,
-      "loss": 0.5013,
-      "step": 1791
-    },
-    {
-      "epoch": 3.18,
-      "learning_rate": 1.8596916106609568e-05,
-      "loss": 0.505,
-      "step": 1792
-    },
-    {
-      "epoch": 3.18,
-      "learning_rate": 1.85949149891256e-05,
-      "loss": 0.4985,
-      "step": 1793
-    },
-    {
-      "epoch": 3.19,
-      "learning_rate": 1.8592912553457717e-05,
-      "loss": 0.5058,
-      "step": 1794
-    },
-    {
-      "epoch": 3.19,
-      "learning_rate": 1.859090879991302e-05,
-      "loss": 0.509,
-      "step": 1795
-    },
-    {
-      "epoch": 3.19,
-      "learning_rate": 1.858890372879882e-05,
-      "loss": 0.4796,
-      "step": 1796
-    },
-    {
-      "epoch": 3.19,
-      "learning_rate": 1.858689734042263e-05,
-      "loss": 0.494,
-      "step": 1797
-    },
-    {
-      "epoch": 3.19,
-      "learning_rate": 1.8584889635092174e-05,
-      "loss": 0.5201,
-      "step": 1798
-    },
-    {
-      "epoch": 3.19,
-      "learning_rate": 1.8582880613115363e-05,
-      "loss": 0.5012,
-      "step": 1799
-    },
-    {
-      "epoch": 3.2,
-      "learning_rate": 1.8580870274800315e-05,
-      "loss": 0.4986,
-      "step": 1800
-    },
-    {
-      "epoch": 3.2,
-      "learning_rate": 1.8578858620455355e-05,
-      "loss": 0.5067,
-      "step": 1801
-    },
-    {
-      "epoch": 3.2,
-      "learning_rate": 1.8576845650389005e-05,
-      "loss": 0.5027,
-      "step": 1802
-    },
-    {
-      "epoch": 3.2,
-      "learning_rate": 1.8574831364909988e-05,
-      "loss": 0.5006,
-      "step": 1803
-    },
-    {
-      "epoch": 3.2,
-      "learning_rate": 1.8572815764327235e-05,
-      "loss": 0.4984,
-      "step": 1804
-    },
-    {
-      "epoch": 3.2,
-      "learning_rate": 1.857079884894987e-05,
-      "loss": 0.524,
-      "step": 1805
-    },
-    {
-      "epoch": 3.21,
-      "learning_rate": 1.8568780619087222e-05,
-      "loss": 0.4755,
-      "step": 1806
-    },
-    {
-      "epoch": 3.21,
-      "learning_rate": 1.856676107504883e-05,
-      "loss": 0.5047,
-      "step": 1807
-    },
-    {
-      "epoch": 3.21,
-      "learning_rate": 1.856474021714442e-05,
-      "loss": 0.5241,
-      "step": 1808
-    },
-    {
-      "epoch": 3.21,
-      "learning_rate": 1.8562718045683933e-05,
-      "loss": 0.5098,
-      "step": 1809
-    },
-    {
-      "epoch": 3.21,
-      "learning_rate": 1.8560694560977497e-05,
-      "loss": 0.4969,
-      "step": 1810
-    },
-    {
-      "epoch": 3.22,
-      "learning_rate": 1.8558669763335458e-05,
-      "loss": 0.5194,
-      "step": 1811
-    },
-    {
-      "epoch": 3.22,
-      "learning_rate": 1.8556643653068347e-05,
-      "loss": 0.5281,
-      "step": 1812
-    },
-    {
-      "epoch": 3.22,
-      "learning_rate": 1.8554616230486912e-05,
-      "loss": 0.5195,
-      "step": 1813
-    },
-    {
-      "epoch": 3.22,
-      "learning_rate": 1.8552587495902087e-05,
-      "loss": 0.5143,
-      "step": 1814
-    },
-    {
-      "epoch": 3.22,
-      "learning_rate": 1.855055744962502e-05,
-      "loss": 0.505,
-      "step": 1815
-    },
-    {
-      "epoch": 3.22,
-      "learning_rate": 1.8548526091967058e-05,
-      "loss": 0.5021,
-      "step": 1816
-    },
-    {
-      "epoch": 3.23,
-      "learning_rate": 1.8546493423239738e-05,
-      "loss": 0.5192,
-      "step": 1817
-    },
-    {
-      "epoch": 3.23,
-      "learning_rate": 1.8544459443754808e-05,
-      "loss": 0.5088,
-      "step": 1818
-    },
-    {
-      "epoch": 3.23,
-      "learning_rate": 1.854242415382422e-05,
-      "loss": 0.508,
-      "step": 1819
-    },
-    {
-      "epoch": 3.23,
-      "learning_rate": 1.8540387553760116e-05,
-      "loss": 0.4993,
-      "step": 1820
-    },
-    {
-      "epoch": 3.23,
-      "learning_rate": 1.8538349643874845e-05,
-      "loss": 0.5094,
-      "step": 1821
-    },
-    {
-      "epoch": 3.23,
-      "learning_rate": 1.8536310424480968e-05,
-      "loss": 0.4977,
-      "step": 1822
-    },
-    {
-      "epoch": 3.24,
-      "learning_rate": 1.8534269895891222e-05,
-      "loss": 0.5025,
-      "step": 1823
-    },
-    {
-      "epoch": 3.24,
-      "learning_rate": 1.8532228058418563e-05,
-      "loss": 0.5158,
-      "step": 1824
-    },
-    {
-      "epoch": 3.24,
-      "learning_rate": 1.853018491237615e-05,
-      "loss": 0.5088,
-      "step": 1825
-    },
-    {
-      "epoch": 3.24,
-      "learning_rate": 1.8528140458077324e-05,
-      "loss": 0.4919,
-      "step": 1826
-    },
-    {
-      "epoch": 3.24,
-      "learning_rate": 1.8526094695835647e-05,
-      "loss": 0.5063,
-      "step": 1827
-    },
-    {
-      "epoch": 3.25,
-      "learning_rate": 1.8524047625964872e-05,
-      "loss": 0.496,
-      "step": 1828
-    },
-    {
-      "epoch": 3.25,
-      "learning_rate": 1.8521999248778954e-05,
-      "loss": 0.4667,
-      "step": 1829
-    },
-    {
-      "epoch": 3.25,
-      "learning_rate": 1.8519949564592047e-05,
-      "loss": 0.5083,
-      "step": 1830
-    },
-    {
-      "epoch": 3.25,
-      "learning_rate": 1.8517898573718504e-05,
-      "loss": 0.4987,
-      "step": 1831
-    },
-    {
-      "epoch": 3.25,
-      "learning_rate": 1.8515846276472888e-05,
-      "loss": 0.5093,
-      "step": 1832
-    },
-    {
-      "epoch": 3.25,
-      "learning_rate": 1.851379267316995e-05,
-      "loss": 0.5137,
-      "step": 1833
-    },
-    {
-      "epoch": 3.26,
-      "learning_rate": 1.8511737764124646e-05,
-      "loss": 0.4885,
-      "step": 1834
-    },
-    {
-      "epoch": 3.26,
-      "learning_rate": 1.8509681549652136e-05,
-      "loss": 0.5015,
-      "step": 1835
-    },
-    {
-      "epoch": 3.26,
-      "learning_rate": 1.850762403006778e-05,
-      "loss": 0.5223,
-      "step": 1836
-    },
-    {
-      "epoch": 3.26,
-      "learning_rate": 1.850556520568713e-05,
-      "loss": 0.5046,
-      "step": 1837
-    },
-    {
-      "epoch": 3.26,
-      "learning_rate": 1.8503505076825946e-05,
-      "loss": 0.501,
-      "step": 1838
-    },
-    {
-      "epoch": 3.26,
-      "learning_rate": 1.8501443643800185e-05,
-      "loss": 0.5329,
-      "step": 1839
-    },
-    {
-      "epoch": 3.27,
-      "learning_rate": 1.8499380906926006e-05,
-      "loss": 0.5034,
-      "step": 1840
-    },
-    {
-      "epoch": 3.27,
-      "learning_rate": 1.8497316866519764e-05,
-      "loss": 0.5008,
-      "step": 1841
-    },
-    {
-      "epoch": 3.27,
-      "learning_rate": 1.8495251522898017e-05,
-      "loss": 0.4827,
-      "step": 1842
-    },
-    {
-      "epoch": 3.27,
-      "learning_rate": 1.8493184876377527e-05,
-      "loss": 0.5086,
-      "step": 1843
-    },
-    {
-      "epoch": 3.27,
-      "learning_rate": 1.8491116927275245e-05,
-      "loss": 0.507,
-      "step": 1844
-    },
-    {
-      "epoch": 3.28,
-      "learning_rate": 1.8489047675908328e-05,
-      "loss": 0.5103,
-      "step": 1845
-    },
-    {
-      "epoch": 3.28,
-      "learning_rate": 1.848697712259414e-05,
-      "loss": 0.5027,
-      "step": 1846
-    },
-    {
-      "epoch": 3.28,
-      "learning_rate": 1.848490526765023e-05,
-      "loss": 0.5115,
-      "step": 1847
-    },
-    {
-      "epoch": 3.28,
-      "learning_rate": 1.8482832111394356e-05,
-      "loss": 0.5091,
-      "step": 1848
-    },
-    {
-      "epoch": 3.28,
-      "learning_rate": 1.848075765414448e-05,
-      "loss": 0.4869,
-      "step": 1849
-    },
-    {
-      "epoch": 3.28,
-      "learning_rate": 1.8478681896218746e-05,
-      "loss": 0.5048,
-      "step": 1850
-    },
-    {
-      "epoch": 3.29,
-      "learning_rate": 1.8476604837935515e-05,
-      "loss": 0.4929,
-      "step": 1851
-    },
-    {
-      "epoch": 3.29,
-      "learning_rate": 1.847452647961334e-05,
-      "loss": 0.5054,
-      "step": 1852
-    },
-    {
-      "epoch": 3.29,
-      "learning_rate": 1.8472446821570976e-05,
-      "loss": 0.4839,
-      "step": 1853
-    },
-    {
-      "epoch": 3.29,
-      "learning_rate": 1.847036586412737e-05,
-      "loss": 0.4966,
-      "step": 1854
-    },
-    {
-      "epoch": 3.29,
-      "learning_rate": 1.846828360760168e-05,
-      "loss": 0.503,
-      "step": 1855
-    },
-    {
-      "epoch": 3.3,
-      "learning_rate": 1.8466200052313256e-05,
-      "loss": 0.4995,
-      "step": 1856
-    },
-    {
-      "epoch": 3.3,
-      "learning_rate": 1.846411519858165e-05,
-      "loss": 0.5028,
-      "step": 1857
-    },
-    {
-      "epoch": 3.3,
-      "learning_rate": 1.8462029046726606e-05,
-      "loss": 0.518,
-      "step": 1858
-    },
-    {
-      "epoch": 3.3,
-      "learning_rate": 1.8459941597068078e-05,
-      "loss": 0.4931,
-      "step": 1859
-    },
-    {
-      "epoch": 3.3,
-      "learning_rate": 1.845785284992621e-05,
-      "loss": 0.4961,
-      "step": 1860
-    },
-    {
-      "epoch": 3.3,
-      "learning_rate": 1.8455762805621354e-05,
-      "loss": 0.4817,
-      "step": 1861
-    },
-    {
-      "epoch": 3.31,
-      "learning_rate": 1.845367146447405e-05,
-      "loss": 0.5058,
-      "step": 1862
-    },
-    {
-      "epoch": 3.31,
-      "learning_rate": 1.8451578826805046e-05,
-      "loss": 0.512,
-      "step": 1863
-    },
-    {
-      "epoch": 3.31,
-      "learning_rate": 1.8449484892935282e-05,
-      "loss": 0.5179,
-      "step": 1864
-    },
-    {
-      "epoch": 3.31,
-      "learning_rate": 1.8447389663185905e-05,
-      "loss": 0.5168,
-      "step": 1865
-    },
-    {
-      "epoch": 3.31,
-      "learning_rate": 1.8445293137878253e-05,
-      "loss": 0.504,
-      "step": 1866
-    },
-    {
-      "epoch": 3.31,
-      "learning_rate": 1.844319531733387e-05,
-      "loss": 0.4923,
-      "step": 1867
-    },
-    {
-      "epoch": 3.32,
-      "learning_rate": 1.844109620187448e-05,
-      "loss": 0.4963,
-      "step": 1868
-    },
-    {
-      "epoch": 3.32,
-      "learning_rate": 1.843899579182204e-05,
-      "loss": 0.5147,
-      "step": 1869
-    },
-    {
-      "epoch": 3.32,
-      "learning_rate": 1.8436894087498673e-05,
-      "loss": 0.5053,
-      "step": 1870
-    },
-    {
-      "epoch": 3.32,
-      "learning_rate": 1.8434791089226712e-05,
-      "loss": 0.4884,
-      "step": 1871
-    },
-    {
-      "epoch": 3.32,
-      "learning_rate": 1.8432686797328697e-05,
-      "loss": 0.507,
-      "step": 1872
-    },
-    {
-      "epoch": 3.33,
-      "learning_rate": 1.8430581212127358e-05,
-      "loss": 0.4983,
-      "step": 1873
-    },
-    {
-      "epoch": 3.33,
-      "learning_rate": 1.8428474333945616e-05,
-      "loss": 0.5207,
-      "step": 1874
-    },
-    {
-      "epoch": 3.33,
-      "learning_rate": 1.8426366163106603e-05,
-      "loss": 0.5126,
-      "step": 1875
-    },
-    {
-      "epoch": 3.33,
-      "learning_rate": 1.842425669993365e-05,
-      "loss": 0.5062,
-      "step": 1876
-    },
-    {
-      "epoch": 3.33,
-      "learning_rate": 1.842214594475027e-05,
-      "loss": 0.5108,
-      "step": 1877
-    },
-    {
-      "epoch": 3.33,
-      "learning_rate": 1.8420033897880193e-05,
-      "loss": 0.5078,
-      "step": 1878
-    },
-    {
-      "epoch": 3.34,
-      "learning_rate": 1.8417920559647343e-05,
-      "loss": 0.5009,
-      "step": 1879
-    },
-    {
-      "epoch": 3.34,
-      "learning_rate": 1.8415805930375827e-05,
-      "loss": 0.5268,
-      "step": 1880
-    },
-    {
-      "epoch": 3.34,
-      "learning_rate": 1.841369001038997e-05,
-      "loss": 0.5079,
-      "step": 1881
-    },
-    {
-      "epoch": 3.34,
-      "learning_rate": 1.841157280001428e-05,
-      "loss": 0.4811,
-      "step": 1882
-    },
-    {
-      "epoch": 3.34,
-      "learning_rate": 1.840945429957347e-05,
-      "loss": 0.5295,
-      "step": 1883
-    },
-    {
-      "epoch": 3.34,
-      "learning_rate": 1.8407334509392454e-05,
-      "loss": 0.5016,
-      "step": 1884
-    },
-    {
-      "epoch": 3.35,
-      "learning_rate": 1.840521342979634e-05,
-      "loss": 0.5031,
-      "step": 1885
-    },
-    {
-      "epoch": 3.35,
-      "learning_rate": 1.8403091061110427e-05,
-      "loss": 0.5277,
-      "step": 1886
-    },
-    {
-      "epoch": 3.35,
-      "learning_rate": 1.8400967403660228e-05,
-      "loss": 0.5207,
-      "step": 1887
-    },
-    {
-      "epoch": 3.35,
-      "learning_rate": 1.8398842457771434e-05,
-      "loss": 0.5115,
-      "step": 1888
-    },
-    {
-      "epoch": 3.35,
-      "learning_rate": 1.8396716223769947e-05,
-      "loss": 0.4963,
-      "step": 1889
-    },
-    {
-      "epoch": 3.36,
-      "learning_rate": 1.8394588701981867e-05,
-      "loss": 0.4938,
-      "step": 1890
-    },
-    {
-      "epoch": 3.36,
-      "learning_rate": 1.8392459892733483e-05,
-      "loss": 0.4964,
-      "step": 1891
-    },
-    {
-      "epoch": 3.36,
-      "learning_rate": 1.8390329796351285e-05,
-      "loss": 0.4982,
-      "step": 1892
-    },
-    {
-      "epoch": 3.36,
-      "learning_rate": 1.8388198413161962e-05,
-      "loss": 0.507,
-      "step": 1893
-    },
-    {
-      "epoch": 3.36,
-      "learning_rate": 1.8386065743492406e-05,
-      "loss": 0.4981,
-      "step": 1894
-    },
-    {
-      "epoch": 3.36,
-      "learning_rate": 1.8383931787669694e-05,
-      "loss": 0.5074,
-      "step": 1895
-    },
-    {
-      "epoch": 3.37,
-      "learning_rate": 1.8381796546021107e-05,
-      "loss": 0.506,
-      "step": 1896
-    },
-    {
-      "epoch": 3.37,
-      "learning_rate": 1.8379660018874124e-05,
-      "loss": 0.5222,
-      "step": 1897
-    },
-    {
-      "epoch": 3.37,
-      "learning_rate": 1.8377522206556416e-05,
-      "loss": 0.5113,
-      "step": 1898
-    },
-    {
-      "epoch": 3.37,
-      "learning_rate": 1.837538310939586e-05,
-      "loss": 0.5179,
-      "step": 1899
-    },
-    {
-      "epoch": 3.37,
-      "learning_rate": 1.837324272772052e-05,
-      "loss": 0.5077,
-      "step": 1900
-    },
-    {
-      "epoch": 3.38,
-      "learning_rate": 1.8371101061858667e-05,
-      "loss": 0.4908,
-      "step": 1901
-    },
-    {
-      "epoch": 3.38,
-      "learning_rate": 1.8368958112138758e-05,
-      "loss": 0.4876,
-      "step": 1902
-    },
-    {
-      "epoch": 3.38,
-      "learning_rate": 1.8366813878889457e-05,
-      "loss": 0.5062,
-      "step": 1903
-    },
-    {
-      "epoch": 3.38,
-      "learning_rate": 1.836466836243962e-05,
-      "loss": 0.5149,
-      "step": 1904
-    },
-    {
-      "epoch": 3.38,
-      "learning_rate": 1.8362521563118298e-05,
-      "loss": 0.4845,
-      "step": 1905
-    },
-    {
-      "epoch": 3.38,
-      "learning_rate": 1.836037348125474e-05,
-      "loss": 0.5178,
-      "step": 1906
-    },
-    {
-      "epoch": 3.39,
-      "learning_rate": 1.83582241171784e-05,
-      "loss": 0.4934,
-      "step": 1907
-    },
-    {
-      "epoch": 3.39,
-      "learning_rate": 1.8356073471218914e-05,
-      "loss": 0.5217,
-      "step": 1908
-    },
-    {
-      "epoch": 3.39,
-      "learning_rate": 1.8353921543706127e-05,
-      "loss": 0.491,
-      "step": 1909
-    },
-    {
-      "epoch": 3.39,
-      "learning_rate": 1.8351768334970067e-05,
-      "loss": 0.5013,
-      "step": 1910
-    },
-    {
-      "epoch": 3.39,
-      "learning_rate": 1.834961384534098e-05,
-      "loss": 0.503,
-      "step": 1911
-    },
-    {
-      "epoch": 3.39,
-      "learning_rate": 1.8347458075149283e-05,
-      "loss": 0.5137,
-      "step": 1912
-    },
-    {
-      "epoch": 3.4,
-      "learning_rate": 1.834530102472561e-05,
-      "loss": 0.4885,
-      "step": 1913
-    },
-    {
-      "epoch": 3.4,
-      "learning_rate": 1.8343142694400784e-05,
-      "loss": 0.4937,
-      "step": 1914
-    },
-    {
-      "epoch": 3.4,
-      "learning_rate": 1.8340983084505817e-05,
-      "loss": 0.5204,
-      "step": 1915
-    },
-    {
-      "epoch": 3.4,
-      "learning_rate": 1.833882219537193e-05,
-      "loss": 0.5006,
-      "step": 1916
-    },
-    {
-      "epoch": 3.4,
-      "learning_rate": 1.8336660027330525e-05,
-      "loss": 0.4829,
-      "step": 1917
-    },
-    {
-      "epoch": 3.41,
-      "learning_rate": 1.833449658071322e-05,
-      "loss": 0.4962,
-      "step": 1918
-    },
-    {
-      "epoch": 3.41,
-      "learning_rate": 1.8332331855851816e-05,
-      "loss": 0.5033,
-      "step": 1919
-    },
-    {
-      "epoch": 3.41,
-      "learning_rate": 1.8330165853078306e-05,
-      "loss": 0.5139,
-      "step": 1920
-    },
-    {
-      "epoch": 3.41,
-      "learning_rate": 1.832799857272489e-05,
-      "loss": 0.489,
-      "step": 1921
-    },
-    {
-      "epoch": 3.41,
-      "learning_rate": 1.8325830015123958e-05,
-      "loss": 0.5215,
-      "step": 1922
-    },
-    {
-      "epoch": 3.41,
-      "learning_rate": 1.83236601806081e-05,
-      "loss": 0.4847,
-      "step": 1923
-    },
-    {
-      "epoch": 3.42,
-      "learning_rate": 1.8321489069510096e-05,
-      "loss": 0.4784,
-      "step": 1924
-    },
-    {
-      "epoch": 3.42,
-      "learning_rate": 1.8319316682162924e-05,
-      "loss": 0.4814,
-      "step": 1925
-    },
-    {
-      "epoch": 3.42,
-      "learning_rate": 1.8317143018899757e-05,
-      "loss": 0.5192,
-      "step": 1926
-    },
-    {
-      "epoch": 3.42,
-      "learning_rate": 1.831496808005397e-05,
-      "loss": 0.4733,
-      "step": 1927
-    },
-    {
-      "epoch": 3.42,
-      "learning_rate": 1.8312791865959127e-05,
-      "loss": 0.4932,
-      "step": 1928
-    },
-    {
-      "epoch": 3.42,
-      "learning_rate": 1.8310614376948986e-05,
-      "loss": 0.5011,
-      "step": 1929
-    },
-    {
-      "epoch": 3.43,
-      "learning_rate": 1.830843561335751e-05,
-      "loss": 0.5457,
-      "step": 1930
-    },
-    {
-      "epoch": 3.43,
-      "learning_rate": 1.8306255575518845e-05,
-      "loss": 0.4986,
-      "step": 1931
-    },
-    {
-      "epoch": 3.43,
-      "learning_rate": 1.8304074263767343e-05,
-      "loss": 0.5034,
-      "step": 1932
-    },
-    {
-      "epoch": 3.43,
-      "learning_rate": 1.8301891678437548e-05,
-      "loss": 0.5134,
-      "step": 1933
-    },
-    {
-      "epoch": 3.43,
-      "learning_rate": 1.8299707819864196e-05,
-      "loss": 0.5194,
-      "step": 1934
-    },
-    {
-      "epoch": 3.44,
-      "learning_rate": 1.829752268838222e-05,
-      "loss": 0.5122,
-      "step": 1935
-    },
-    {
-      "epoch": 3.44,
-      "learning_rate": 1.829533628432675e-05,
-      "loss": 0.492,
-      "step": 1936
-    },
-    {
-      "epoch": 3.44,
-      "learning_rate": 1.8293148608033112e-05,
-      "loss": 0.492,
-      "step": 1937
-    },
-    {
-      "epoch": 3.44,
-      "learning_rate": 1.829095965983682e-05,
-      "loss": 0.4913,
-      "step": 1938
-    },
-    {
-      "epoch": 3.44,
-      "learning_rate": 1.8288769440073595e-05,
-      "loss": 0.4876,
-      "step": 1939
-    },
-    {
-      "epoch": 3.44,
-      "learning_rate": 1.8286577949079342e-05,
-      "loss": 0.5015,
-      "step": 1940
-    },
-    {
-      "epoch": 3.45,
-      "learning_rate": 1.8284385187190168e-05,
-      "loss": 0.4859,
-      "step": 1941
-    },
-    {
-      "epoch": 3.45,
-      "learning_rate": 1.828219115474237e-05,
-      "loss": 0.5117,
-      "step": 1942
-    },
-    {
-      "epoch": 3.45,
-      "learning_rate": 1.8279995852072442e-05,
-      "loss": 0.5519,
-      "step": 1943
-    },
-    {
-      "epoch": 3.45,
-      "learning_rate": 1.8277799279517078e-05,
-      "loss": 0.4853,
-      "step": 1944
-    },
-    {
-      "epoch": 3.45,
-      "learning_rate": 1.8275601437413155e-05,
-      "loss": 0.5213,
-      "step": 1945
-    },
-    {
-      "epoch": 3.45,
-      "learning_rate": 1.8273402326097756e-05,
-      "loss": 0.4965,
-      "step": 1946
-    },
-    {
-      "epoch": 3.46,
-      "learning_rate": 1.8271201945908152e-05,
-      "loss": 0.4954,
-      "step": 1947
-    },
-    {
-      "epoch": 3.46,
-      "learning_rate": 1.826900029718181e-05,
-      "loss": 0.5239,
-      "step": 1948
-    },
-    {
-      "epoch": 3.46,
-      "learning_rate": 1.8266797380256388e-05,
-      "loss": 0.5104,
-      "step": 1949
-    },
-    {
-      "epoch": 3.46,
-      "learning_rate": 1.8264593195469755e-05,
-      "loss": 0.491,
-      "step": 1950
-    },
-    {
-      "epoch": 3.46,
-      "learning_rate": 1.826238774315995e-05,
-      "loss": 0.4811,
-      "step": 1951
-    },
-    {
-      "epoch": 3.47,
-      "learning_rate": 1.8260181023665223e-05,
-      "loss": 0.5294,
-      "step": 1952
-    },
-    {
-      "epoch": 3.47,
-      "learning_rate": 1.825797303732402e-05,
-      "loss": 0.4857,
-      "step": 1953
-    },
-    {
-      "epoch": 3.47,
-      "learning_rate": 1.8255763784474964e-05,
-      "loss": 0.495,
-      "step": 1954
-    },
-    {
-      "epoch": 3.47,
-      "learning_rate": 1.8253553265456892e-05,
-      "loss": 0.5028,
-      "step": 1955
-    },
-    {
-      "epoch": 3.47,
-      "learning_rate": 1.8251341480608823e-05,
-      "loss": 0.4727,
-      "step": 1956
-    },
-    {
-      "epoch": 3.47,
-      "learning_rate": 1.8249128430269977e-05,
-      "loss": 0.4893,
-      "step": 1957
-    },
-    {
-      "epoch": 3.48,
-      "learning_rate": 1.8246914114779756e-05,
-      "loss": 0.5234,
-      "step": 1958
-    },
-    {
-      "epoch": 3.48,
-      "learning_rate": 1.8244698534477776e-05,
-      "loss": 0.4931,
-      "step": 1959
-    },
-    {
-      "epoch": 3.48,
-      "learning_rate": 1.824248168970383e-05,
-      "loss": 0.525,
-      "step": 1960
-    },
-    {
-      "epoch": 3.48,
-      "learning_rate": 1.824026358079791e-05,
-      "loss": 0.5148,
-      "step": 1961
-    },
-    {
-      "epoch": 3.48,
-      "learning_rate": 1.8238044208100204e-05,
-      "loss": 0.5095,
-      "step": 1962
-    },
-    {
-      "epoch": 3.49,
-      "learning_rate": 1.8235823571951095e-05,
-      "loss": 0.5274,
-      "step": 1963
-    },
-    {
-      "epoch": 3.49,
-      "learning_rate": 1.823360167269115e-05,
-      "loss": 0.5036,
-      "step": 1964
-    },
-    {
-      "epoch": 3.49,
-      "learning_rate": 1.823137851066115e-05,
-      "loss": 0.5028,
-      "step": 1965
-    },
-    {
-      "epoch": 3.49,
-      "learning_rate": 1.8229154086202044e-05,
-      "loss": 0.5086,
-      "step": 1966
-    },
-    {
-      "epoch": 3.49,
-      "learning_rate": 1.8226928399654996e-05,
-      "loss": 0.5185,
-      "step": 1967
-    },
-    {
-      "epoch": 3.49,
-      "learning_rate": 1.8224701451361348e-05,
-      "loss": 0.497,
-      "step": 1968
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 1.8222473241662643e-05,
-      "loss": 0.4964,
-      "step": 1969
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 1.8220243770900623e-05,
-      "loss": 0.505,
-      "step": 1970
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 1.821801303941721e-05,
-      "loss": 0.5485,
-      "step": 1971
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 1.8215781047554534e-05,
-      "loss": 0.5019,
-      "step": 1972
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 1.8213547795654903e-05,
-      "loss": 0.4836,
-      "step": 1973
-    },
-    {
-      "epoch": 3.5,
-      "learning_rate": 1.8211313284060832e-05,
-      "loss": 0.5026,
-      "step": 1974
-    },
-    {
-      "epoch": 3.51,
-      "learning_rate": 1.8209077513115018e-05,
-      "loss": 0.5001,
-      "step": 1975
-    },
-    {
-      "epoch": 3.51,
-      "learning_rate": 1.8206840483160362e-05,
-      "loss": 0.5055,
-      "step": 1976
-    },
-    {
-      "epoch": 3.51,
-      "learning_rate": 1.8204602194539948e-05,
-      "loss": 0.4825,
-      "step": 1977
-    },
-    {
-      "epoch": 3.51,
-      "learning_rate": 1.8202362647597064e-05,
-      "loss": 0.4825,
-      "step": 1978
-    },
-    {
-      "epoch": 3.51,
-      "learning_rate": 1.8200121842675176e-05,
-      "loss": 0.5195,
-      "step": 1979
-    },
-    {
-      "epoch": 3.52,
-      "learning_rate": 1.819787978011796e-05,
-      "loss": 0.5108,
-      "step": 1980
-    },
-    {
-      "epoch": 3.52,
-      "learning_rate": 1.8195636460269268e-05,
-      "loss": 0.4966,
-      "step": 1981
-    },
-    {
-      "epoch": 3.52,
-      "learning_rate": 1.8193391883473158e-05,
-      "loss": 0.5209,
-      "step": 1982
-    },
-    {
-      "epoch": 3.52,
-      "learning_rate": 1.819114605007388e-05,
-      "loss": 0.4858,
-      "step": 1983
-    },
-    {
-      "epoch": 3.52,
-      "learning_rate": 1.8188898960415865e-05,
-      "loss": 0.4934,
-      "step": 1984
-    },
-    {
-      "epoch": 3.52,
-      "learning_rate": 1.8186650614843744e-05,
-      "loss": 0.5048,
-      "step": 1985
-    },
-    {
-      "epoch": 3.53,
-      "learning_rate": 1.8184401013702348e-05,
-      "loss": 0.4735,
-      "step": 1986
-    },
-    {
-      "epoch": 3.53,
-      "learning_rate": 1.8182150157336692e-05,
-      "loss": 0.52,
-      "step": 1987
-    },
-    {
-      "epoch": 3.53,
-      "learning_rate": 1.817989804609198e-05,
-      "loss": 0.5124,
-      "step": 1988
-    },
-    {
-      "epoch": 3.53,
-      "learning_rate": 1.8177644680313618e-05,
-      "loss": 0.5383,
-      "step": 1989
-    },
-    {
-      "epoch": 3.53,
-      "learning_rate": 1.8175390060347196e-05,
-      "loss": 0.5083,
-      "step": 1990
-    },
-    {
-      "epoch": 3.53,
-      "learning_rate": 1.8173134186538505e-05,
-      "loss": 0.4944,
-      "step": 1991
-    },
-    {
-      "epoch": 3.54,
-      "learning_rate": 1.8170877059233522e-05,
-      "loss": 0.4814,
-      "step": 1992
-    },
-    {
-      "epoch": 3.54,
-      "learning_rate": 1.8168618678778418e-05,
-      "loss": 0.5172,
-      "step": 1993
-    },
-    {
-      "epoch": 3.54,
-      "learning_rate": 1.8166359045519555e-05,
-      "loss": 0.4837,
-      "step": 1994
-    },
-    {
-      "epoch": 3.54,
-      "learning_rate": 1.8164098159803485e-05,
-      "loss": 0.4789,
-      "step": 1995
-    },
-    {
-      "epoch": 3.54,
-      "learning_rate": 1.8161836021976966e-05,
-      "loss": 0.4728,
-      "step": 1996
-    },
-    {
-      "epoch": 3.55,
-      "learning_rate": 1.8159572632386923e-05,
-      "loss": 0.4878,
-      "step": 1997
-    },
-    {
-      "epoch": 3.55,
-      "learning_rate": 1.8157307991380496e-05,
-      "loss": 0.5097,
-      "step": 1998
-    },
-    {
-      "epoch": 3.55,
-      "learning_rate": 1.815504209930501e-05,
-      "loss": 0.488,
-      "step": 1999
-    },
-    {
-      "epoch": 3.55,
-      "learning_rate": 1.8152774956507974e-05,
-      "loss": 0.5101,
-      "step": 2000
-    },
-    {
-      "epoch": 3.55,
-      "learning_rate": 1.8150506563337095e-05,
-      "loss": 0.5121,
-      "step": 2001
-    },
-    {
-      "epoch": 3.55,
-      "learning_rate": 1.8148236920140275e-05,
-      "loss": 0.4991,
-      "step": 2002
-    },
-    {
-      "epoch": 3.56,
-      "learning_rate": 1.8145966027265607e-05,
-      "loss": 0.484,
-      "step": 2003
-    },
-    {
-      "epoch": 3.56,
-      "learning_rate": 1.814369388506137e-05,
-      "loss": 0.5108,
-      "step": 2004
-    },
-    {
-      "epoch": 3.56,
-      "learning_rate": 1.8141420493876035e-05,
-      "loss": 0.5108,
-      "step": 2005
-    },
-    {
-      "epoch": 3.56,
-      "learning_rate": 1.8139145854058273e-05,
-      "loss": 0.5019,
-      "step": 2006
-    },
-    {
-      "epoch": 3.56,
-      "learning_rate": 1.813686996595693e-05,
-      "loss": 0.5236,
-      "step": 2007
-    },
-    {
-      "epoch": 3.57,
-      "learning_rate": 1.8134592829921065e-05,
-      "loss": 0.5012,
-      "step": 2008
-    },
-    {
-      "epoch": 3.57,
-      "learning_rate": 1.8132314446299912e-05,
-      "loss": 0.5256,
-      "step": 2009
-    },
-    {
-      "epoch": 3.57,
-      "learning_rate": 1.8130034815442908e-05,
-      "loss": 0.5089,
-      "step": 2010
-    },
-    {
-      "epoch": 3.57,
-      "learning_rate": 1.8127753937699667e-05,
-      "loss": 0.4973,
-      "step": 2011
-    },
-    {
-      "epoch": 3.57,
-      "learning_rate": 1.812547181342001e-05,
-      "loss": 0.5227,
-      "step": 2012
-    },
-    {
-      "epoch": 3.57,
-      "learning_rate": 1.8123188442953935e-05,
-      "loss": 0.4954,
-      "step": 2013
-    },
-    {
-      "epoch": 3.58,
-      "learning_rate": 1.812090382665164e-05,
-      "loss": 0.497,
-      "step": 2014
-    },
-    {
-      "epoch": 3.58,
-      "learning_rate": 1.811861796486351e-05,
-      "loss": 0.5043,
-      "step": 2015
-    },
-    {
-      "epoch": 3.58,
-      "learning_rate": 1.811633085794013e-05,
-      "loss": 0.527,
-      "step": 2016
-    },
-    {
-      "epoch": 3.58,
-      "learning_rate": 1.8114042506232264e-05,
-      "loss": 0.4959,
-      "step": 2017
-    },
-    {
-      "epoch": 3.58,
-      "learning_rate": 1.8111752910090868e-05,
-      "loss": 0.4934,
-      "step": 2018
-    },
-    {
-      "epoch": 3.58,
-      "learning_rate": 1.81094620698671e-05,
-      "loss": 0.4962,
-      "step": 2019
-    },
-    {
-      "epoch": 3.59,
-      "learning_rate": 1.8107169985912293e-05,
-      "loss": 0.5151,
-      "step": 2020
-    },
-    {
-      "epoch": 3.59,
-      "learning_rate": 1.8104876658577987e-05,
-      "loss": 0.5018,
-      "step": 2021
-    },
-    {
-      "epoch": 3.59,
-      "learning_rate": 1.8102582088215896e-05,
-      "loss": 0.4905,
-      "step": 2022
-    },
-    {
-      "epoch": 3.59,
-      "learning_rate": 1.8100286275177943e-05,
-      "loss": 0.516,
-      "step": 2023
-    },
-    {
-      "epoch": 3.59,
-      "learning_rate": 1.8097989219816226e-05,
-      "loss": 0.5085,
-      "step": 2024
-    },
-    {
-      "epoch": 3.6,
-      "learning_rate": 1.809569092248304e-05,
-      "loss": 0.5206,
-      "step": 2025
-    },
-    {
-      "epoch": 3.6,
-      "learning_rate": 1.809339138353087e-05,
-      "loss": 0.5159,
-      "step": 2026
-    },
-    {
-      "epoch": 3.6,
-      "learning_rate": 1.8091090603312396e-05,
-      "loss": 0.4906,
-      "step": 2027
-    },
-    {
-      "epoch": 3.6,
-      "learning_rate": 1.808878858218048e-05,
-      "loss": 0.5172,
-      "step": 2028
-    },
-    {
-      "epoch": 3.6,
-      "learning_rate": 1.8086485320488176e-05,
-      "loss": 0.5147,
-      "step": 2029
-    },
-    {
-      "epoch": 3.6,
-      "learning_rate": 1.808418081858873e-05,
-      "loss": 0.5129,
-      "step": 2030
-    },
-    {
-      "epoch": 3.61,
-      "learning_rate": 1.8081875076835587e-05,
-      "loss": 0.493,
-      "step": 2031
-    },
-    {
-      "epoch": 3.61,
-      "learning_rate": 1.807956809558237e-05,
-      "loss": 0.5157,
-      "step": 2032
-    },
-    {
-      "epoch": 3.61,
-      "learning_rate": 1.807725987518289e-05,
-      "loss": 0.4941,
-      "step": 2033
-    },
-    {
-      "epoch": 3.61,
-      "learning_rate": 1.8074950415991157e-05,
-      "loss": 0.5077,
-      "step": 2034
-    },
-    {
-      "epoch": 3.61,
-      "learning_rate": 1.807263971836137e-05,
-      "loss": 0.5155,
-      "step": 2035
-    },
-    {
-      "epoch": 3.61,
-      "learning_rate": 1.8070327782647912e-05,
-      "loss": 0.4948,
-      "step": 2036
-    },
-    {
-      "epoch": 3.62,
-      "learning_rate": 1.8068014609205363e-05,
-      "loss": 0.507,
-      "step": 2037
-    },
-    {
-      "epoch": 3.62,
-      "learning_rate": 1.8065700198388488e-05,
-      "loss": 0.4888,
-      "step": 2038
-    },
-    {
-      "epoch": 3.62,
-      "learning_rate": 1.8063384550552246e-05,
-      "loss": 0.4929,
-      "step": 2039
-    },
-    {
-      "epoch": 3.62,
-      "learning_rate": 1.806106766605178e-05,
-      "loss": 0.512,
-      "step": 2040
-    },
-    {
-      "epoch": 3.62,
-      "learning_rate": 1.8058749545242428e-05,
-      "loss": 0.5142,
-      "step": 2041
-    },
-    {
-      "epoch": 3.63,
-      "learning_rate": 1.8056430188479712e-05,
-      "loss": 0.4826,
-      "step": 2042
-    },
-    {
-      "epoch": 3.63,
-      "learning_rate": 1.805410959611935e-05,
-      "loss": 0.5239,
-      "step": 2043
-    },
-    {
-      "epoch": 3.63,
-      "learning_rate": 1.8051787768517245e-05,
-      "loss": 0.4938,
-      "step": 2044
-    },
-    {
-      "epoch": 3.63,
-      "learning_rate": 1.804946470602949e-05,
-      "loss": 0.5023,
-      "step": 2045
-    },
-    {
-      "epoch": 3.63,
-      "learning_rate": 1.8047140409012367e-05,
-      "loss": 0.5014,
-      "step": 2046
-    },
-    {
-      "epoch": 3.63,
-      "learning_rate": 1.8044814877822353e-05,
-      "loss": 0.5028,
-      "step": 2047
-    },
-    {
-      "epoch": 3.64,
-      "learning_rate": 1.8042488112816107e-05,
-      "loss": 0.5225,
-      "step": 2048
-    },
-    {
-      "epoch": 3.64,
-      "learning_rate": 1.804016011435048e-05,
-      "loss": 0.4954,
-      "step": 2049
-    },
-    {
-      "epoch": 3.64,
-      "learning_rate": 1.8037830882782512e-05,
-      "loss": 0.5004,
-      "step": 2050
-    },
-    {
-      "epoch": 3.64,
-      "learning_rate": 1.8035500418469435e-05,
-      "loss": 0.5206,
-      "step": 2051
-    },
-    {
-      "epoch": 3.64,
-      "learning_rate": 1.803316872176866e-05,
-      "loss": 0.5202,
-      "step": 2052
-    },
-    {
-      "epoch": 3.64,
-      "learning_rate": 1.8030835793037805e-05,
-      "loss": 0.4972,
-      "step": 2053
-    },
-    {
-      "epoch": 3.65,
-      "learning_rate": 1.802850163263466e-05,
-      "loss": 0.5023,
-      "step": 2054
-    },
-    {
-      "epoch": 3.65,
-      "learning_rate": 1.802616624091721e-05,
-      "loss": 0.5158,
-      "step": 2055
-    },
-    {
-      "epoch": 3.65,
-      "learning_rate": 1.802382961824363e-05,
-      "loss": 0.5199,
-      "step": 2056
-    },
-    {
-      "epoch": 3.65,
-      "learning_rate": 1.8021491764972286e-05,
-      "loss": 0.4746,
-      "step": 2057
-    },
-    {
-      "epoch": 3.65,
-      "learning_rate": 1.8019152681461724e-05,
-      "loss": 0.4982,
-      "step": 2058
-    },
-    {
-      "epoch": 3.66,
-      "learning_rate": 1.8016812368070686e-05,
-      "loss": 0.5072,
-      "step": 2059
-    },
-    {
-      "epoch": 3.66,
-      "learning_rate": 1.8014470825158104e-05,
-      "loss": 0.5206,
-      "step": 2060
-    },
-    {
-      "epoch": 3.66,
-      "learning_rate": 1.8012128053083097e-05,
-      "loss": 0.4892,
-      "step": 2061
-    },
-    {
-      "epoch": 3.66,
-      "learning_rate": 1.8009784052204963e-05,
-      "loss": 0.5042,
-      "step": 2062
-    },
-    {
-      "epoch": 3.66,
-      "learning_rate": 1.80074388228832e-05,
-      "loss": 0.5004,
-      "step": 2063
-    },
-    {
-      "epoch": 3.66,
-      "learning_rate": 1.800509236547749e-05,
-      "loss": 0.4944,
-      "step": 2064
-    },
-    {
-      "epoch": 3.67,
-      "learning_rate": 1.8002744680347712e-05,
-      "loss": 0.508,
-      "step": 2065
-    },
-    {
-      "epoch": 3.67,
-      "learning_rate": 1.8000395767853914e-05,
-      "loss": 0.4843,
-      "step": 2066
-    },
-    {
-      "epoch": 3.67,
-      "learning_rate": 1.7998045628356355e-05,
-      "loss": 0.5165,
-      "step": 2067
-    },
-    {
-      "epoch": 3.67,
-      "learning_rate": 1.7995694262215462e-05,
-      "loss": 0.5064,
-      "step": 2068
-    },
-    {
-      "epoch": 3.67,
-      "learning_rate": 1.799334166979186e-05,
-      "loss": 0.5009,
-      "step": 2069
-    },
-    {
-      "epoch": 3.68,
-      "learning_rate": 1.7990987851446367e-05,
-      "loss": 0.512,
-      "step": 2070
-    },
-    {
-      "epoch": 3.68,
-      "learning_rate": 1.7988632807539975e-05,
-      "loss": 0.5121,
-      "step": 2071
-    },
-    {
-      "epoch": 3.68,
-      "learning_rate": 1.7986276538433883e-05,
-      "loss": 0.5278,
-      "step": 2072
-    },
-    {
-      "epoch": 3.68,
-      "learning_rate": 1.7983919044489455e-05,
-      "loss": 0.5307,
-      "step": 2073
-    },
-    {
-      "epoch": 3.68,
-      "learning_rate": 1.7981560326068262e-05,
-      "loss": 0.524,
-      "step": 2074
-    },
-    {
-      "epoch": 3.68,
-      "learning_rate": 1.7979200383532055e-05,
-      "loss": 0.5004,
-      "step": 2075
-    },
-    {
-      "epoch": 3.69,
-      "learning_rate": 1.7976839217242772e-05,
-      "loss": 0.5068,
-      "step": 2076
-    },
-    {
-      "epoch": 3.69,
-      "learning_rate": 1.797447682756254e-05,
-      "loss": 0.5064,
-      "step": 2077
-    },
-    {
-      "epoch": 3.69,
-      "learning_rate": 1.7972113214853674e-05,
-      "loss": 0.527,
-      "step": 2078
-    },
-    {
-      "epoch": 3.69,
-      "learning_rate": 1.7969748379478675e-05,
-      "loss": 0.506,
-      "step": 2079
-    },
-    {
-      "epoch": 3.69,
-      "learning_rate": 1.796738232180024e-05,
-      "loss": 0.5346,
-      "step": 2080
-    },
-    {
-      "epoch": 3.69,
-      "learning_rate": 1.7965015042181234e-05,
-      "loss": 0.5093,
-      "step": 2081
-    },
-    {
-      "epoch": 3.7,
-      "learning_rate": 1.7962646540984733e-05,
-      "loss": 0.5038,
-      "step": 2082
-    },
-    {
-      "epoch": 3.7,
-      "learning_rate": 1.7960276818573978e-05,
-      "loss": 0.5038,
-      "step": 2083
-    },
-    {
-      "epoch": 3.7,
-      "learning_rate": 1.795790587531242e-05,
-      "loss": 0.4743,
-      "step": 2084
-    },
-    {
-      "epoch": 3.7,
-      "learning_rate": 1.795553371156368e-05,
-      "loss": 0.5023,
-      "step": 2085
-    },
-    {
-      "epoch": 3.7,
-      "learning_rate": 1.795316032769157e-05,
-      "loss": 0.5095,
-      "step": 2086
-    },
-    {
-      "epoch": 3.71,
-      "learning_rate": 1.795078572406009e-05,
-      "loss": 0.4869,
-      "step": 2087
-    },
-    {
-      "epoch": 3.71,
-      "learning_rate": 1.7948409901033436e-05,
-      "loss": 0.489,
-      "step": 2088
-    },
-    {
-      "epoch": 3.71,
-      "learning_rate": 1.7946032858975978e-05,
-      "loss": 0.5166,
-      "step": 2089
-    },
-    {
-      "epoch": 3.71,
-      "learning_rate": 1.7943654598252272e-05,
-      "loss": 0.4959,
-      "step": 2090
-    },
-    {
-      "epoch": 3.71,
-      "learning_rate": 1.7941275119227076e-05,
-      "loss": 0.5063,
-      "step": 2091
-    },
-    {
-      "epoch": 3.71,
-      "learning_rate": 1.793889442226532e-05,
-      "loss": 0.5168,
-      "step": 2092
-    },
-    {
-      "epoch": 3.72,
-      "learning_rate": 1.793651250773213e-05,
-      "loss": 0.4915,
-      "step": 2093
-    },
-    {
-      "epoch": 3.72,
-      "learning_rate": 1.7934129375992814e-05,
-      "loss": 0.4845,
-      "step": 2094
-    },
-    {
-      "epoch": 3.72,
-      "learning_rate": 1.7931745027412866e-05,
-      "loss": 0.517,
-      "step": 2095
-    },
-    {
-      "epoch": 3.72,
-      "learning_rate": 1.792935946235797e-05,
-      "loss": 0.4999,
-      "step": 2096
-    },
-    {
-      "epoch": 3.72,
-      "learning_rate": 1.7926972681193996e-05,
-      "loss": 0.5096,
-      "step": 2097
-    },
-    {
-      "epoch": 3.72,
-      "learning_rate": 1.7924584684286995e-05,
-      "loss": 0.5101,
-      "step": 2098
-    },
-    {
-      "epoch": 3.73,
-      "learning_rate": 1.792219547200322e-05,
-      "loss": 0.4733,
-      "step": 2099
-    },
-    {
-      "epoch": 3.73,
-      "learning_rate": 1.7919805044709083e-05,
-      "loss": 0.4878,
-      "step": 2100
-    },
-    {
-      "epoch": 3.73,
-      "learning_rate": 1.7917413402771215e-05,
-      "loss": 0.5137,
-      "step": 2101
-    },
-    {
-      "epoch": 3.73,
-      "learning_rate": 1.7915020546556407e-05,
-      "loss": 0.4892,
-      "step": 2102
-    },
-    {
-      "epoch": 3.73,
-      "learning_rate": 1.7912626476431648e-05,
-      "loss": 0.4838,
-      "step": 2103
-    },
-    {
-      "epoch": 3.74,
-      "learning_rate": 1.7910231192764117e-05,
-      "loss": 0.4862,
-      "step": 2104
-    },
-    {
-      "epoch": 3.74,
-      "learning_rate": 1.7907834695921166e-05,
-      "loss": 0.5129,
-      "step": 2105
-    },
-    {
-      "epoch": 3.74,
-      "learning_rate": 1.7905436986270342e-05,
-      "loss": 0.4964,
-      "step": 2106
-    },
-    {
-      "epoch": 3.74,
-      "learning_rate": 1.790303806417938e-05,
-      "loss": 0.4842,
-      "step": 2107
-    },
-    {
-      "epoch": 3.74,
-      "learning_rate": 1.79006379300162e-05,
-      "loss": 0.5015,
-      "step": 2108
-    },
-    {
-      "epoch": 3.74,
-      "learning_rate": 1.78982365841489e-05,
-      "loss": 0.5044,
-      "step": 2109
-    },
-    {
-      "epoch": 3.75,
-      "learning_rate": 1.789583402694577e-05,
-      "loss": 0.5052,
-      "step": 2110
-    },
-    {
-      "epoch": 3.75,
-      "learning_rate": 1.7893430258775283e-05,
-      "loss": 0.5148,
-      "step": 2111
-    },
-    {
-      "epoch": 3.75,
-      "learning_rate": 1.7891025280006108e-05,
-      "loss": 0.4947,
-      "step": 2112
-    },
-    {
-      "epoch": 3.75,
-      "learning_rate": 1.7888619091007082e-05,
-      "loss": 0.5018,
-      "step": 2113
-    },
-    {
-      "epoch": 3.75,
-      "learning_rate": 1.7886211692147245e-05,
-      "loss": 0.5107,
-      "step": 2114
-    },
-    {
-      "epoch": 3.75,
-      "learning_rate": 1.788380308379581e-05,
-      "loss": 0.5314,
-      "step": 2115
-    },
-    {
-      "epoch": 3.76,
-      "learning_rate": 1.788139326632218e-05,
-      "loss": 0.4779,
-      "step": 2116
-    },
-    {
-      "epoch": 3.76,
-      "learning_rate": 1.7878982240095948e-05,
-      "loss": 0.5169,
-      "step": 2117
-    },
-    {
-      "epoch": 3.76,
-      "learning_rate": 1.7876570005486882e-05,
-      "loss": 0.5296,
-      "step": 2118
-    },
-    {
-      "epoch": 3.76,
-      "learning_rate": 1.7874156562864946e-05,
-      "loss": 0.4858,
-      "step": 2119
-    },
-    {
-      "epoch": 3.76,
-      "learning_rate": 1.7871741912600283e-05,
-      "loss": 0.5042,
-      "step": 2120
-    },
-    {
-      "epoch": 3.77,
-      "learning_rate": 1.7869326055063223e-05,
-      "loss": 0.4914,
-      "step": 2121
-    },
-    {
-      "epoch": 3.77,
-      "learning_rate": 1.7866908990624278e-05,
-      "loss": 0.4863,
-      "step": 2122
-    },
-    {
-      "epoch": 3.77,
-      "learning_rate": 1.786449071965415e-05,
-      "loss": 0.4994,
-      "step": 2123
-    },
-    {
-      "epoch": 3.77,
-      "learning_rate": 1.786207124252373e-05,
-      "loss": 0.4989,
-      "step": 2124
-    },
-    {
-      "epoch": 3.77,
-      "learning_rate": 1.785965055960408e-05,
-      "loss": 0.4981,
-      "step": 2125
-    },
-    {
-      "epoch": 3.77,
-      "learning_rate": 1.785722867126646e-05,
-      "loss": 0.5024,
-      "step": 2126
-    },
-    {
-      "epoch": 3.78,
-      "learning_rate": 1.7854805577882307e-05,
-      "loss": 0.4896,
-      "step": 2127
-    },
-    {
-      "epoch": 3.78,
-      "learning_rate": 1.7852381279823243e-05,
-      "loss": 0.5072,
-      "step": 2128
-    },
-    {
-      "epoch": 3.78,
-      "learning_rate": 1.7849955777461083e-05,
-      "loss": 0.5091,
-      "step": 2129
-    },
-    {
-      "epoch": 3.78,
-      "learning_rate": 1.784752907116782e-05,
-      "loss": 0.5154,
-      "step": 2130
-    },
-    {
-      "epoch": 3.78,
-      "learning_rate": 1.784510116131563e-05,
-      "loss": 0.4884,
-      "step": 2131
-    },
-    {
-      "epoch": 3.79,
-      "learning_rate": 1.784267204827688e-05,
-      "loss": 0.5114,
-      "step": 2132
-    },
-    {
-      "epoch": 3.79,
-      "learning_rate": 1.784024173242412e-05,
-      "loss": 0.5235,
-      "step": 2133
-    },
-    {
-      "epoch": 3.79,
-      "learning_rate": 1.7837810214130073e-05,
-      "loss": 0.5097,
-      "step": 2134
-    },
-    {
-      "epoch": 3.79,
-      "learning_rate": 1.7835377493767664e-05,
-      "loss": 0.4994,
-      "step": 2135
-    },
-    {
-      "epoch": 3.79,
-      "learning_rate": 1.7832943571709988e-05,
-      "loss": 0.5075,
-      "step": 2136
-    },
-    {
-      "epoch": 3.79,
-      "learning_rate": 1.7830508448330338e-05,
-      "loss": 0.4831,
-      "step": 2137
-    },
-    {
-      "epoch": 3.8,
-      "learning_rate": 1.7828072124002177e-05,
-      "loss": 0.4954,
-      "step": 2138
-    },
-    {
-      "epoch": 3.8,
-      "learning_rate": 1.7825634599099167e-05,
-      "loss": 0.4961,
-      "step": 2139
-    },
-    {
-      "epoch": 3.8,
-      "learning_rate": 1.7823195873995136e-05,
-      "loss": 0.5022,
-      "step": 2140
-    },
-    {
-      "epoch": 3.8,
-      "learning_rate": 1.782075594906411e-05,
-      "loss": 0.5008,
-      "step": 2141
-    },
-    {
-      "epoch": 3.8,
-      "learning_rate": 1.78183148246803e-05,
-      "loss": 0.5107,
-      "step": 2142
-    },
-    {
-      "epoch": 3.8,
-      "learning_rate": 1.781587250121809e-05,
-      "loss": 0.5335,
-      "step": 2143
-    },
-    {
-      "epoch": 3.81,
-      "learning_rate": 1.7813428979052056e-05,
-      "loss": 0.555,
-      "step": 2144
-    },
-    {
-      "epoch": 3.81,
-      "learning_rate": 1.7810984258556955e-05,
-      "loss": 0.5072,
-      "step": 2145
-    },
-    {
-      "epoch": 3.81,
-      "learning_rate": 1.780853834010773e-05,
-      "loss": 0.4884,
-      "step": 2146
-    },
-    {
-      "epoch": 3.81,
-      "learning_rate": 1.7806091224079508e-05,
-      "loss": 0.4942,
-      "step": 2147
-    },
-    {
-      "epoch": 3.81,
-      "learning_rate": 1.7803642910847595e-05,
-      "loss": 0.4838,
-      "step": 2148
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 1.7801193400787485e-05,
-      "loss": 0.5217,
-      "step": 2149
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 1.7798742694274852e-05,
-      "loss": 0.4861,
-      "step": 2150
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 1.779629079168556e-05,
-      "loss": 0.4951,
-      "step": 2151
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 1.7793837693395645e-05,
-      "loss": 0.5073,
-      "step": 2152
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 1.779138339978134e-05,
-      "loss": 0.5214,
-      "step": 2153
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 1.7788927911219053e-05,
-      "loss": 0.4945,
-      "step": 2154
-    },
-    {
-      "epoch": 3.83,
-      "learning_rate": 1.7786471228085377e-05,
-      "loss": 0.4918,
-      "step": 2155
-    },
-    {
-      "epoch": 3.83,
-      "learning_rate": 1.7784013350757086e-05,
-      "loss": 0.4993,
-      "step": 2156
-    },
-    {
-      "epoch": 3.83,
-      "learning_rate": 1.7781554279611143e-05,
-      "loss": 0.4874,
-      "step": 2157
-    },
-    {
-      "epoch": 3.83,
-      "learning_rate": 1.7779094015024688e-05,
-      "loss": 0.5091,
-      "step": 2158
-    },
-    {
-      "epoch": 3.83,
-      "learning_rate": 1.777663255737505e-05,
-      "loss": 0.5208,
-      "step": 2159
-    },
-    {
-      "epoch": 3.83,
-      "learning_rate": 1.7774169907039735e-05,
-      "loss": 0.516,
-      "step": 2160
-    },
-    {
-      "epoch": 3.84,
-      "learning_rate": 1.7771706064396433e-05,
-      "loss": 0.4874,
-      "step": 2161
-    },
-    {
-      "epoch": 3.84,
-      "learning_rate": 1.7769241029823022e-05,
-      "loss": 0.5025,
-      "step": 2162
-    },
-    {
-      "epoch": 3.84,
-      "learning_rate": 1.7766774803697555e-05,
-      "loss": 0.4952,
-      "step": 2163
-    },
-    {
-      "epoch": 3.84,
-      "learning_rate": 1.7764307386398275e-05,
-      "loss": 0.5118,
-      "step": 2164
-    },
-    {
-      "epoch": 3.84,
-      "learning_rate": 1.7761838778303603e-05,
-      "loss": 0.5243,
-      "step": 2165
-    },
-    {
-      "epoch": 3.85,
-      "learning_rate": 1.7759368979792145e-05,
-      "loss": 0.5014,
-      "step": 2166
-    },
-    {
-      "epoch": 3.85,
-      "learning_rate": 1.775689799124269e-05,
-      "loss": 0.4891,
-      "step": 2167
-    },
-    {
-      "epoch": 3.85,
-      "learning_rate": 1.7754425813034206e-05,
-      "loss": 0.4923,
-      "step": 2168
-    },
-    {
-      "epoch": 3.85,
-      "learning_rate": 1.7751952445545846e-05,
-      "loss": 0.5113,
-      "step": 2169
-    },
-    {
-      "epoch": 3.85,
-      "learning_rate": 1.7749477889156947e-05,
-      "loss": 0.5023,
-      "step": 2170
-    },
-    {
-      "epoch": 3.85,
-      "learning_rate": 1.7747002144247024e-05,
-      "loss": 0.518,
-      "step": 2171
-    },
-    {
-      "epoch": 3.86,
-      "learning_rate": 1.7744525211195777e-05,
-      "loss": 0.476,
-      "step": 2172
-    },
-    {
-      "epoch": 3.86,
-      "learning_rate": 1.7742047090383094e-05,
-      "loss": 0.4715,
-      "step": 2173
-    },
-    {
-      "epoch": 3.86,
-      "learning_rate": 1.7739567782189026e-05,
-      "loss": 0.5022,
-      "step": 2174
-    },
-    {
-      "epoch": 3.86,
-      "learning_rate": 1.7737087286993832e-05,
-      "loss": 0.4948,
-      "step": 2175
-    },
-    {
-      "epoch": 3.86,
-      "learning_rate": 1.7734605605177937e-05,
-      "loss": 0.522,
-      "step": 2176
-    },
-    {
-      "epoch": 3.87,
-      "learning_rate": 1.7732122737121948e-05,
-      "loss": 0.5269,
-      "step": 2177
-    },
-    {
-      "epoch": 3.87,
-      "learning_rate": 1.772963868320666e-05,
-      "loss": 0.4839,
-      "step": 2178
-    },
-    {
-      "epoch": 3.87,
-      "learning_rate": 1.7727153443813043e-05,
-      "loss": 0.4956,
-      "step": 2179
-    },
-    {
-      "epoch": 3.87,
-      "learning_rate": 1.7724667019322258e-05,
-      "loss": 0.4967,
-      "step": 2180
-    },
-    {
-      "epoch": 3.87,
-      "learning_rate": 1.7722179410115644e-05,
-      "loss": 0.4971,
-      "step": 2181
-    },
-    {
-      "epoch": 3.87,
-      "learning_rate": 1.7719690616574715e-05,
-      "loss": 0.5056,
-      "step": 2182
-    },
-    {
-      "epoch": 3.88,
-      "learning_rate": 1.7717200639081172e-05,
-      "loss": 0.4609,
-      "step": 2183
-    },
-    {
-      "epoch": 3.88,
-      "learning_rate": 1.77147094780169e-05,
-      "loss": 0.4833,
-      "step": 2184
-    },
-    {
-      "epoch": 3.88,
-      "learning_rate": 1.7712217133763964e-05,
-      "loss": 0.4843,
-      "step": 2185
-    },
-    {
-      "epoch": 3.88,
-      "learning_rate": 1.770972360670461e-05,
-      "loss": 0.4995,
-      "step": 2186
-    },
-    {
-      "epoch": 3.88,
-      "learning_rate": 1.770722889722126e-05,
-      "loss": 0.4961,
-      "step": 2187
-    },
-    {
-      "epoch": 3.88,
-      "learning_rate": 1.770473300569653e-05,
-      "loss": 0.5019,
-      "step": 2188
-    },
-    {
-      "epoch": 3.89,
-      "learning_rate": 1.7702235932513202e-05,
-      "loss": 0.4965,
-      "step": 2189
-    },
-    {
-      "epoch": 3.89,
-      "learning_rate": 1.7699737678054256e-05,
-      "loss": 0.4854,
-      "step": 2190
-    },
-    {
-      "epoch": 3.89,
-      "learning_rate": 1.7697238242702828e-05,
-      "loss": 0.4904,
-      "step": 2191
-    },
-    {
-      "epoch": 3.89,
-      "learning_rate": 1.769473762684227e-05,
-      "loss": 0.4924,
-      "step": 2192
-    },
-    {
-      "epoch": 3.89,
-      "learning_rate": 1.769223583085608e-05,
-      "loss": 0.4757,
-      "step": 2193
-    },
-    {
-      "epoch": 3.9,
-      "learning_rate": 1.7689732855127965e-05,
-      "loss": 0.4785,
-      "step": 2194
-    },
-    {
-      "epoch": 3.9,
-      "learning_rate": 1.76872287000418e-05,
-      "loss": 0.5078,
-      "step": 2195
-    },
-    {
-      "epoch": 3.9,
-      "learning_rate": 1.7684723365981632e-05,
-      "loss": 0.5049,
-      "step": 2196
-    },
-    {
-      "epoch": 3.9,
-      "learning_rate": 1.768221685333171e-05,
-      "loss": 0.5192,
-      "step": 2197
-    },
-    {
-      "epoch": 3.9,
-      "learning_rate": 1.7679709162476444e-05,
-      "loss": 0.5321,
-      "step": 2198
-    },
-    {
-      "epoch": 3.9,
-      "learning_rate": 1.767720029380044e-05,
-      "loss": 0.4844,
-      "step": 2199
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 1.7674690247688475e-05,
-      "loss": 0.4986,
-      "step": 2200
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 1.7672179024525506e-05,
-      "loss": 0.4928,
-      "step": 2201
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 1.7669666624696683e-05,
-      "loss": 0.5123,
-      "step": 2202
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 1.7667153048587318e-05,
-      "loss": 0.4868,
-      "step": 2203
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 1.7664638296582916e-05,
-      "loss": 0.507,
-      "step": 2204
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 1.7662122369069164e-05,
-      "loss": 0.4915,
-      "step": 2205
-    },
-    {
-      "epoch": 3.92,
-      "learning_rate": 1.7659605266431916e-05,
-      "loss": 0.5236,
-      "step": 2206
-    },
-    {
-      "epoch": 3.92,
-      "learning_rate": 1.765708698905722e-05,
-      "loss": 0.4943,
-      "step": 2207
-    },
-    {
-      "epoch": 3.92,
-      "learning_rate": 1.76545675373313e-05,
-      "loss": 0.5041,
-      "step": 2208
-    },
-    {
-      "epoch": 3.92,
-      "learning_rate": 1.7652046911640557e-05,
-      "loss": 0.5148,
-      "step": 2209
-    },
-    {
-      "epoch": 3.92,
-      "learning_rate": 1.764952511237157e-05,
-      "loss": 0.4837,
-      "step": 2210
-    },
-    {
-      "epoch": 3.93,
-      "learning_rate": 1.764700213991111e-05,
-      "loss": 0.5128,
-      "step": 2211
-    },
-    {
-      "epoch": 3.93,
-      "learning_rate": 1.7644477994646115e-05,
-      "loss": 0.5048,
-      "step": 2212
-    },
-    {
-      "epoch": 3.93,
-      "learning_rate": 1.764195267696371e-05,
-      "loss": 0.4842,
-      "step": 2213
-    },
-    {
-      "epoch": 3.93,
-      "learning_rate": 1.76394261872512e-05,
-      "loss": 0.5001,
-      "step": 2214
-    },
-    {
-      "epoch": 3.93,
-      "learning_rate": 1.7636898525896057e-05,
-      "loss": 0.5199,
-      "step": 2215
-    },
-    {
-      "epoch": 3.93,
-      "learning_rate": 1.7634369693285954e-05,
-      "loss": 0.4942,
-      "step": 2216
-    },
-    {
-      "epoch": 3.94,
-      "learning_rate": 1.7631839689808724e-05,
-      "loss": 0.4975,
-      "step": 2217
-    },
-    {
-      "epoch": 3.94,
-      "learning_rate": 1.76293085158524e-05,
-      "loss": 0.4922,
-      "step": 2218
-    },
-    {
-      "epoch": 3.94,
-      "learning_rate": 1.7626776171805166e-05,
-      "loss": 0.5213,
-      "step": 2219
-    },
-    {
-      "epoch": 3.94,
-      "learning_rate": 1.762424265805542e-05,
-      "loss": 0.4921,
-      "step": 2220
-    },
-    {
-      "epoch": 3.94,
-      "learning_rate": 1.7621707974991708e-05,
-      "loss": 0.5055,
-      "step": 2221
-    },
-    {
-      "epoch": 3.94,
-      "learning_rate": 1.761917212300278e-05,
-      "loss": 0.5489,
-      "step": 2222
-    },
-    {
-      "epoch": 3.95,
-      "learning_rate": 1.7616635102477545e-05,
-      "loss": 0.5093,
-      "step": 2223
-    },
-    {
-      "epoch": 3.95,
-      "learning_rate": 1.7614096913805104e-05,
-      "loss": 0.5061,
-      "step": 2224
-    },
-    {
-      "epoch": 3.95,
-      "learning_rate": 1.7611557557374735e-05,
-      "loss": 0.5184,
-      "step": 2225
-    },
-    {
-      "epoch": 3.95,
-      "learning_rate": 1.760901703357589e-05,
-      "loss": 0.5003,
-      "step": 2226
-    },
-    {
-      "epoch": 3.95,
-      "learning_rate": 1.760647534279821e-05,
-      "loss": 0.5182,
-      "step": 2227
-    },
-    {
-      "epoch": 3.96,
-      "learning_rate": 1.7603932485431506e-05,
-      "loss": 0.5045,
-      "step": 2228
-    },
-    {
-      "epoch": 3.96,
-      "learning_rate": 1.760138846186577e-05,
-      "loss": 0.5103,
-      "step": 2229
-    },
-    {
-      "epoch": 3.96,
-      "learning_rate": 1.759884327249117e-05,
-      "loss": 0.4824,
-      "step": 2230
-    },
-    {
-      "epoch": 3.96,
-      "learning_rate": 1.759629691769806e-05,
-      "loss": 0.5073,
-      "step": 2231
-    },
-    {
-      "epoch": 3.96,
-      "learning_rate": 1.759374939787697e-05,
-      "loss": 0.5195,
-      "step": 2232
-    },
-    {
-      "epoch": 3.96,
-      "learning_rate": 1.7591200713418604e-05,
-      "loss": 0.4909,
-      "step": 2233
-    },
-    {
-      "epoch": 3.97,
-      "learning_rate": 1.7588650864713855e-05,
-      "loss": 0.4992,
-      "step": 2234
-    },
-    {
-      "epoch": 3.97,
-      "learning_rate": 1.7586099852153778e-05,
-      "loss": 0.5052,
-      "step": 2235
-    },
-    {
-      "epoch": 3.97,
-      "learning_rate": 1.7583547676129624e-05,
-      "loss": 0.5009,
-      "step": 2236
-    },
-    {
-      "epoch": 3.97,
-      "learning_rate": 1.7580994337032814e-05,
-      "loss": 0.4828,
-      "step": 2237
-    },
-    {
-      "epoch": 3.97,
-      "learning_rate": 1.757843983525495e-05,
-      "loss": 0.4865,
-      "step": 2238
-    },
-    {
-      "epoch": 3.98,
-      "learning_rate": 1.7575884171187803e-05,
-      "loss": 0.477,
-      "step": 2239
-    },
-    {
-      "epoch": 3.98,
-      "learning_rate": 1.7573327345223335e-05,
-      "loss": 0.5019,
-      "step": 2240
-    },
-    {
-      "epoch": 3.98,
-      "learning_rate": 1.7570769357753682e-05,
-      "loss": 0.4643,
-      "step": 2241
-    },
-    {
-      "epoch": 3.98,
-      "learning_rate": 1.7568210209171154e-05,
-      "loss": 0.5128,
-      "step": 2242
-    },
-    {
-      "epoch": 3.98,
-      "learning_rate": 1.756564989986824e-05,
-      "loss": 0.4936,
-      "step": 2243
-    },
-    {
-      "epoch": 3.98,
-      "learning_rate": 1.7563088430237618e-05,
-      "loss": 0.521,
-      "step": 2244
-    },
-    {
-      "epoch": 3.99,
-      "learning_rate": 1.756052580067213e-05,
-      "loss": 0.5092,
-      "step": 2245
-    },
-    {
-      "epoch": 3.99,
-      "learning_rate": 1.75579620115648e-05,
-      "loss": 0.5031,
-      "step": 2246
-    },
-    {
-      "epoch": 3.99,
-      "learning_rate": 1.755539706330883e-05,
-      "loss": 0.5005,
-      "step": 2247
-    },
-    {
-      "epoch": 3.99,
-      "learning_rate": 1.7552830956297607e-05,
-      "loss": 0.5107,
-      "step": 2248
-    },
-    {
-      "epoch": 3.99,
-      "learning_rate": 1.7550263690924683e-05,
-      "loss": 0.5157,
-      "step": 2249
-    },
-    {
-      "epoch": 3.99,
-      "learning_rate": 1.7547695267583794e-05,
-      "loss": 0.5228,
-      "step": 2250
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 1.754512568666886e-05,
-      "loss": 0.5025,
-      "step": 2251
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 1.754255494857396e-05,
-      "loss": 0.4806,
-      "step": 2252
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 1.753998305369338e-05,
-      "loss": 0.4808,
-      "step": 2253
-    },
-    {
-      "epoch": 4.0,
-      "eval_loss": 0.5313146710395813,
-      "eval_runtime": 3447.53,
-      "eval_samples_per_second": 4.161,
-      "eval_steps_per_second": 1.04,
-      "step": 2253
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 1.7537410002421552e-05,
-      "loss": 0.4933,
-      "step": 2254
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 1.7534835795153107e-05,
-      "loss": 0.4998,
-      "step": 2255
-    },
-    {
-      "epoch": 4.01,
-      "learning_rate": 1.753226043228284e-05,
-      "loss": 0.5108,
-      "step": 2256
-    },
-    {
-      "epoch": 4.01,
-      "learning_rate": 1.7529683914205732e-05,
-      "loss": 0.5066,
-      "step": 2257
-    },
-    {
-      "epoch": 4.01,
-      "learning_rate": 1.752710624131694e-05,
-      "loss": 0.5071,
-      "step": 2258
-    },
-    {
-      "epoch": 4.01,
-      "learning_rate": 1.752452741401179e-05,
-      "loss": 0.4903,
-      "step": 2259
-    },
-    {
-      "epoch": 4.01,
-      "learning_rate": 1.7521947432685803e-05,
-      "loss": 0.4796,
-      "step": 2260
-    },
-    {
-      "epoch": 4.01,
-      "learning_rate": 1.7519366297734653e-05,
-      "loss": 0.5101,
-      "step": 2261
-    },
-    {
-      "epoch": 4.02,
-      "learning_rate": 1.7516784009554207e-05,
-      "loss": 0.4827,
-      "step": 2262
-    },
-    {
-      "epoch": 4.02,
-      "learning_rate": 1.751420056854051e-05,
-      "loss": 0.4889,
-      "step": 2263
-    },
-    {
-      "epoch": 4.02,
-      "learning_rate": 1.7511615975089768e-05,
-      "loss": 0.4798,
-      "step": 2264
-    },
-    {
-      "epoch": 4.02,
-      "learning_rate": 1.7509030229598387e-05,
-      "loss": 0.5135,
-      "step": 2265
-    },
-    {
-      "epoch": 4.02,
-      "learning_rate": 1.7506443332462932e-05,
-      "loss": 0.5165,
-      "step": 2266
-    },
-    {
-      "epoch": 4.02,
-      "learning_rate": 1.750385528408015e-05,
-      "loss": 0.4934,
-      "step": 2267
-    },
-    {
-      "epoch": 4.03,
-      "learning_rate": 1.750126608484696e-05,
-      "loss": 0.508,
-      "step": 2268
-    },
-    {
-      "epoch": 4.03,
-      "learning_rate": 1.7498675735160468e-05,
-      "loss": 0.5035,
-      "step": 2269
-    },
-    {
-      "epoch": 4.03,
-      "learning_rate": 1.749608423541795e-05,
-      "loss": 0.5043,
-      "step": 2270
-    },
-    {
-      "epoch": 4.03,
-      "learning_rate": 1.749349158601686e-05,
-      "loss": 0.4904,
-      "step": 2271
-    },
-    {
-      "epoch": 4.03,
-      "learning_rate": 1.749089778735482e-05,
-      "loss": 0.471,
-      "step": 2272
-    },
-    {
-      "epoch": 4.04,
-      "learning_rate": 1.7488302839829642e-05,
-      "loss": 0.4817,
-      "step": 2273
-    },
-    {
-      "epoch": 4.04,
-      "learning_rate": 1.7485706743839307e-05,
-      "loss": 0.5028,
-      "step": 2274
-    },
-    {
-      "epoch": 4.04,
-      "learning_rate": 1.7483109499781972e-05,
-      "loss": 0.5039,
-      "step": 2275
-    },
-    {
-      "epoch": 4.04,
-      "learning_rate": 1.748051110805597e-05,
-      "loss": 0.5033,
-      "step": 2276
-    },
-    {
-      "epoch": 4.04,
-      "learning_rate": 1.7477911569059808e-05,
-      "loss": 0.4938,
-      "step": 2277
-    },
-    {
-      "epoch": 4.04,
-      "learning_rate": 1.7475310883192177e-05,
-      "loss": 0.4908,
-      "step": 2278
-    },
-    {
-      "epoch": 4.05,
-      "learning_rate": 1.7472709050851936e-05,
-      "loss": 0.4869,
-      "step": 2279
-    },
-    {
-      "epoch": 4.05,
-      "learning_rate": 1.7470106072438126e-05,
-      "loss": 0.5079,
-      "step": 2280
-    },
-    {
-      "epoch": 4.05,
-      "learning_rate": 1.7467501948349954e-05,
-      "loss": 0.4911,
-      "step": 2281
-    },
-    {
-      "epoch": 4.05,
-      "learning_rate": 1.7464896678986813e-05,
-      "loss": 0.4888,
-      "step": 2282
-    },
-    {
-      "epoch": 4.05,
-      "learning_rate": 1.746229026474827e-05,
-      "loss": 0.5002,
-      "step": 2283
-    },
-    {
-      "epoch": 4.06,
-      "learning_rate": 1.745968270603406e-05,
-      "loss": 0.5099,
-      "step": 2284
-    },
-    {
-      "epoch": 4.06,
-      "learning_rate": 1.74570740032441e-05,
-      "loss": 0.5016,
-      "step": 2285
-    },
-    {
-      "epoch": 4.06,
-      "learning_rate": 1.7454464156778485e-05,
-      "loss": 0.485,
-      "step": 2286
-    },
-    {
-      "epoch": 4.06,
-      "learning_rate": 1.745185316703748e-05,
-      "loss": 0.4831,
-      "step": 2287
-    },
-    {
-      "epoch": 4.06,
-      "learning_rate": 1.744924103442152e-05,
-      "loss": 0.5048,
-      "step": 2288
-    },
-    {
-      "epoch": 4.06,
-      "learning_rate": 1.744662775933123e-05,
-      "loss": 0.4977,
-      "step": 2289
-    },
-    {
-      "epoch": 4.07,
-      "learning_rate": 1.7444013342167406e-05,
-      "loss": 0.5058,
-      "step": 2290
-    },
-    {
-      "epoch": 4.07,
-      "learning_rate": 1.7441397783331008e-05,
-      "loss": 0.4793,
-      "step": 2291
-    },
-    {
-      "epoch": 4.07,
-      "learning_rate": 1.743878108322318e-05,
-      "loss": 0.5086,
-      "step": 2292
-    },
-    {
-      "epoch": 4.07,
-      "learning_rate": 1.743616324224524e-05,
-      "loss": 0.5136,
-      "step": 2293
-    },
-    {
-      "epoch": 4.07,
-      "learning_rate": 1.7433544260798682e-05,
-      "loss": 0.4766,
-      "step": 2294
-    },
-    {
-      "epoch": 4.07,
-      "learning_rate": 1.743092413928517e-05,
-      "loss": 0.4948,
-      "step": 2295
-    },
-    {
-      "epoch": 4.08,
-      "learning_rate": 1.7428302878106557e-05,
-      "loss": 0.4839,
-      "step": 2296
-    },
-    {
-      "epoch": 4.08,
-      "learning_rate": 1.7425680477664847e-05,
-      "loss": 0.5046,
-      "step": 2297
-    },
-    {
-      "epoch": 4.08,
-      "learning_rate": 1.742305693836224e-05,
-      "loss": 0.478,
-      "step": 2298
-    },
-    {
-      "epoch": 4.08,
-      "learning_rate": 1.7420432260601102e-05,
-      "loss": 0.4995,
-      "step": 2299
-    },
-    {
-      "epoch": 4.08,
-      "learning_rate": 1.7417806444783968e-05,
-      "loss": 0.4954,
-      "step": 2300
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 1.741517949131356e-05,
-      "loss": 0.5002,
-      "step": 2301
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 1.741255140059277e-05,
-      "loss": 0.4705,
-      "step": 2302
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 1.7409922173024658e-05,
-      "loss": 0.4809,
-      "step": 2303
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 1.7407291809012465e-05,
-      "loss": 0.5012,
-      "step": 2304
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 1.7404660308959608e-05,
-      "loss": 0.4798,
-      "step": 2305
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 1.7402027673269665e-05,
-      "loss": 0.4871,
-      "step": 2306
-    },
-    {
-      "epoch": 4.1,
-      "learning_rate": 1.7399393902346403e-05,
-      "loss": 0.4899,
-      "step": 2307
-    },
-    {
-      "epoch": 4.1,
-      "learning_rate": 1.7396758996593762e-05,
-      "loss": 0.4777,
-      "step": 2308
-    },
-    {
-      "epoch": 4.1,
-      "learning_rate": 1.7394122956415848e-05,
-      "loss": 0.4902,
-      "step": 2309
-    },
-    {
-      "epoch": 4.1,
-      "learning_rate": 1.7391485782216943e-05,
-      "loss": 0.5098,
-      "step": 2310
-    },
-    {
-      "epoch": 4.1,
-      "learning_rate": 1.7388847474401507e-05,
-      "loss": 0.4631,
-      "step": 2311
-    },
-    {
-      "epoch": 4.1,
-      "learning_rate": 1.7386208033374174e-05,
-      "loss": 0.5137,
-      "step": 2312
-    },
-    {
-      "epoch": 4.11,
-      "learning_rate": 1.738356745953975e-05,
-      "loss": 0.4862,
-      "step": 2313
-    },
-    {
-      "epoch": 4.11,
-      "learning_rate": 1.7380925753303205e-05,
-      "loss": 0.5007,
-      "step": 2314
-    },
-    {
-      "epoch": 4.11,
-      "learning_rate": 1.73782829150697e-05,
-      "loss": 0.5039,
-      "step": 2315
-    },
-    {
-      "epoch": 4.11,
-      "learning_rate": 1.737563894524456e-05,
-      "loss": 0.5093,
-      "step": 2316
-    },
-    {
-      "epoch": 4.11,
-      "learning_rate": 1.7372993844233285e-05,
-      "loss": 0.4929,
-      "step": 2317
-    },
-    {
-      "epoch": 4.12,
-      "learning_rate": 1.737034761244155e-05,
-      "loss": 0.4917,
-      "step": 2318
-    },
-    {
-      "epoch": 4.12,
-      "learning_rate": 1.7367700250275196e-05,
-      "loss": 0.4752,
-      "step": 2319
-    },
-    {
-      "epoch": 4.12,
-      "learning_rate": 1.736505175814025e-05,
-      "loss": 0.5012,
-      "step": 2320
-    },
-    {
-      "epoch": 4.12,
-      "learning_rate": 1.7362402136442903e-05,
-      "loss": 0.4927,
-      "step": 2321
-    },
-    {
-      "epoch": 4.12,
-      "learning_rate": 1.735975138558953e-05,
-      "loss": 0.5016,
-      "step": 2322
-    },
-    {
-      "epoch": 4.12,
-      "learning_rate": 1.735709950598665e-05,
-      "loss": 0.5057,
-      "step": 2323
-    },
-    {
-      "epoch": 4.13,
-      "learning_rate": 1.7354446498040997e-05,
-      "loss": 0.4999,
-      "step": 2324
-    },
-    {
-      "epoch": 4.13,
-      "learning_rate": 1.735179236215945e-05,
-      "loss": 0.4842,
-      "step": 2325
-    },
-    {
-      "epoch": 4.13,
-      "learning_rate": 1.7349137098749067e-05,
-      "loss": 0.4892,
-      "step": 2326
-    },
-    {
-      "epoch": 4.13,
-      "learning_rate": 1.7346480708217077e-05,
-      "loss": 0.4931,
-      "step": 2327
-    },
-    {
-      "epoch": 4.13,
-      "learning_rate": 1.7343823190970894e-05,
-      "loss": 0.5014,
-      "step": 2328
-    },
-    {
-      "epoch": 4.13,
-      "learning_rate": 1.734116454741809e-05,
-      "loss": 0.5016,
-      "step": 2329
-    },
-    {
-      "epoch": 4.14,
-      "learning_rate": 1.7338504777966414e-05,
-      "loss": 0.5108,
-      "step": 2330
-    },
-    {
-      "epoch": 4.14,
-      "learning_rate": 1.7335843883023792e-05,
-      "loss": 0.4927,
-      "step": 2331
-    },
-    {
-      "epoch": 4.14,
-      "learning_rate": 1.733318186299832e-05,
-      "loss": 0.4869,
-      "step": 2332
-    },
-    {
-      "epoch": 4.14,
-      "learning_rate": 1.7330518718298263e-05,
-      "loss": 0.5029,
-      "step": 2333
-    },
-    {
-      "epoch": 4.14,
-      "learning_rate": 1.7327854449332067e-05,
-      "loss": 0.5039,
-      "step": 2334
-    },
-    {
-      "epoch": 4.15,
-      "learning_rate": 1.7325189056508344e-05,
-      "loss": 0.5011,
-      "step": 2335
-    },
-    {
-      "epoch": 4.15,
-      "learning_rate": 1.7322522540235878e-05,
-      "loss": 0.516,
-      "step": 2336
-    },
-    {
-      "epoch": 4.15,
-      "learning_rate": 1.7319854900923627e-05,
-      "loss": 0.4959,
-      "step": 2337
-    },
-    {
-      "epoch": 4.15,
-      "learning_rate": 1.731718613898072e-05,
-      "loss": 0.4467,
-      "step": 2338
-    },
-    {
-      "epoch": 4.15,
-      "learning_rate": 1.7314516254816464e-05,
-      "loss": 0.5126,
-      "step": 2339
-    },
-    {
-      "epoch": 4.15,
-      "learning_rate": 1.731184524884033e-05,
-      "loss": 0.4882,
-      "step": 2340
-    },
-    {
-      "epoch": 4.16,
-      "learning_rate": 1.7309173121461963e-05,
-      "loss": 0.4863,
-      "step": 2341
-    },
-    {
-      "epoch": 4.16,
-      "learning_rate": 1.7306499873091182e-05,
-      "loss": 0.5087,
-      "step": 2342
-    },
-    {
-      "epoch": 4.16,
-      "learning_rate": 1.7303825504137985e-05,
-      "loss": 0.4889,
-      "step": 2343
-    },
-    {
-      "epoch": 4.16,
-      "learning_rate": 1.7301150015012523e-05,
-      "loss": 0.4792,
-      "step": 2344
-    },
-    {
-      "epoch": 4.16,
-      "learning_rate": 1.729847340612514e-05,
-      "loss": 0.4975,
-      "step": 2345
-    },
-    {
-      "epoch": 4.17,
-      "learning_rate": 1.7295795677886334e-05,
-      "loss": 0.4868,
-      "step": 2346
-    },
-    {
-      "epoch": 4.17,
-      "learning_rate": 1.7293116830706783e-05,
-      "loss": 0.498,
-      "step": 2347
-    },
-    {
-      "epoch": 4.17,
-      "learning_rate": 1.7290436864997345e-05,
-      "loss": 0.5017,
-      "step": 2348
-    },
-    {
-      "epoch": 4.17,
-      "learning_rate": 1.728775578116903e-05,
-      "loss": 0.5222,
-      "step": 2349
-    },
-    {
-      "epoch": 4.17,
-      "learning_rate": 1.7285073579633036e-05,
-      "loss": 0.5012,
-      "step": 2350
-    },
-    {
-      "epoch": 4.17,
-      "learning_rate": 1.7282390260800728e-05,
-      "loss": 0.4866,
-      "step": 2351
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 1.7279705825083634e-05,
-      "loss": 0.5019,
-      "step": 2352
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 1.727702027289347e-05,
-      "loss": 0.5054,
-      "step": 2353
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 1.72743336046421e-05,
-      "loss": 0.5101,
-      "step": 2354
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 1.7271645820741586e-05,
-      "loss": 0.5105,
-      "step": 2355
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 1.7268956921604142e-05,
-      "loss": 0.4949,
-      "step": 2356
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 1.7266266907642157e-05,
-      "loss": 0.4958,
-      "step": 2357
-    },
-    {
-      "epoch": 4.19,
-      "learning_rate": 1.7263575779268197e-05,
-      "loss": 0.5039,
-      "step": 2358
-    },
-    {
-      "epoch": 4.19,
-      "learning_rate": 1.726088353689499e-05,
-      "loss": 0.4845,
-      "step": 2359
-    },
-    {
-      "epoch": 4.19,
-      "learning_rate": 1.7258190180935443e-05,
-      "loss": 0.5078,
-      "step": 2360
-    },
-    {
-      "epoch": 4.19,
-      "learning_rate": 1.7255495711802627e-05,
-      "loss": 0.4901,
-      "step": 2361
-    },
-    {
-      "epoch": 4.19,
-      "learning_rate": 1.7252800129909794e-05,
-      "loss": 0.4993,
-      "step": 2362
-    },
-    {
-      "epoch": 4.2,
-      "learning_rate": 1.725010343567035e-05,
-      "loss": 0.5222,
-      "step": 2363
-    },
-    {
-      "epoch": 4.2,
-      "learning_rate": 1.7247405629497892e-05,
-      "loss": 0.4919,
-      "step": 2364
-    },
-    {
-      "epoch": 4.2,
-      "learning_rate": 1.7244706711806168e-05,
-      "loss": 0.5019,
-      "step": 2365
-    },
-    {
-      "epoch": 4.2,
-      "learning_rate": 1.7242006683009108e-05,
-      "loss": 0.4881,
-      "step": 2366
-    },
-    {
-      "epoch": 4.2,
-      "learning_rate": 1.723930554352081e-05,
-      "loss": 0.4836,
-      "step": 2367
-    },
-    {
-      "epoch": 4.2,
-      "learning_rate": 1.7236603293755544e-05,
-      "loss": 0.5119,
-      "step": 2368
-    },
-    {
-      "epoch": 4.21,
-      "learning_rate": 1.7233899934127743e-05,
-      "loss": 0.4917,
-      "step": 2369
-    },
-    {
-      "epoch": 4.21,
-      "learning_rate": 1.7231195465052023e-05,
-      "loss": 0.4889,
-      "step": 2370
-    },
-    {
-      "epoch": 4.21,
-      "learning_rate": 1.7228489886943154e-05,
-      "loss": 0.4999,
-      "step": 2371
-    },
-    {
-      "epoch": 4.21,
-      "learning_rate": 1.7225783200216092e-05,
-      "loss": 0.4913,
-      "step": 2372
-    },
-    {
-      "epoch": 4.21,
-      "learning_rate": 1.7223075405285956e-05,
-      "loss": 0.5369,
-      "step": 2373
-    },
-    {
-      "epoch": 4.21,
-      "learning_rate": 1.7220366502568028e-05,
-      "loss": 0.4738,
-      "step": 2374
-    },
-    {
-      "epoch": 4.22,
-      "learning_rate": 1.7217656492477772e-05,
-      "loss": 0.5013,
-      "step": 2375
-    },
-    {
-      "epoch": 4.22,
-      "learning_rate": 1.7214945375430816e-05,
-      "loss": 0.5032,
-      "step": 2376
-    },
-    {
-      "epoch": 4.22,
-      "learning_rate": 1.7212233151842956e-05,
-      "loss": 0.4819,
-      "step": 2377
-    },
-    {
-      "epoch": 4.22,
-      "learning_rate": 1.7209519822130162e-05,
-      "loss": 0.511,
-      "step": 2378
-    },
-    {
-      "epoch": 4.22,
-      "learning_rate": 1.7206805386708572e-05,
-      "loss": 0.4851,
-      "step": 2379
-    },
-    {
-      "epoch": 4.23,
-      "learning_rate": 1.7204089845994492e-05,
-      "loss": 0.4903,
-      "step": 2380
-    },
-    {
-      "epoch": 4.23,
-      "learning_rate": 1.72013732004044e-05,
-      "loss": 0.4755,
-      "step": 2381
-    },
-    {
-      "epoch": 4.23,
-      "learning_rate": 1.719865545035494e-05,
-      "loss": 0.4968,
-      "step": 2382
-    },
-    {
-      "epoch": 4.23,
-      "learning_rate": 1.719593659626293e-05,
-      "loss": 0.5024,
-      "step": 2383
-    },
-    {
-      "epoch": 4.23,
-      "learning_rate": 1.7193216638545352e-05,
-      "loss": 0.5014,
-      "step": 2384
-    },
-    {
-      "epoch": 4.23,
-      "learning_rate": 1.7190495577619364e-05,
-      "loss": 0.4921,
-      "step": 2385
-    },
-    {
-      "epoch": 4.24,
-      "learning_rate": 1.7187773413902288e-05,
-      "loss": 0.4966,
-      "step": 2386
-    },
-    {
-      "epoch": 4.24,
-      "learning_rate": 1.7185050147811612e-05,
-      "loss": 0.505,
-      "step": 2387
-    },
-    {
-      "epoch": 4.24,
-      "learning_rate": 1.7182325779765004e-05,
-      "loss": 0.4967,
-      "step": 2388
-    },
-    {
-      "epoch": 4.24,
-      "learning_rate": 1.717960031018029e-05,
-      "loss": 0.4798,
-      "step": 2389
-    },
-    {
-      "epoch": 4.24,
-      "learning_rate": 1.7176873739475475e-05,
-      "loss": 0.4871,
-      "step": 2390
-    },
-    {
-      "epoch": 4.25,
-      "learning_rate": 1.717414606806872e-05,
-      "loss": 0.485,
-      "step": 2391
-    },
-    {
-      "epoch": 4.25,
-      "learning_rate": 1.717141729637837e-05,
-      "loss": 0.4743,
-      "step": 2392
-    },
-    {
-      "epoch": 4.25,
-      "learning_rate": 1.7168687424822922e-05,
-      "loss": 0.5069,
-      "step": 2393
-    },
-    {
-      "epoch": 4.25,
-      "learning_rate": 1.7165956453821057e-05,
-      "loss": 0.4899,
-      "step": 2394
-    },
-    {
-      "epoch": 4.25,
-      "learning_rate": 1.716322438379162e-05,
-      "loss": 0.5031,
-      "step": 2395
-    },
-    {
-      "epoch": 4.25,
-      "learning_rate": 1.7160491215153617e-05,
-      "loss": 0.4809,
-      "step": 2396
-    },
-    {
-      "epoch": 4.26,
-      "learning_rate": 1.715775694832623e-05,
-      "loss": 0.4884,
-      "step": 2397
-    },
-    {
-      "epoch": 4.26,
-      "learning_rate": 1.715502158372881e-05,
-      "loss": 0.4745,
-      "step": 2398
-    },
-    {
-      "epoch": 4.26,
-      "learning_rate": 1.7152285121780872e-05,
-      "loss": 0.5016,
-      "step": 2399
-    },
-    {
-      "epoch": 4.26,
-      "learning_rate": 1.7149547562902102e-05,
-      "loss": 0.4866,
-      "step": 2400
-    },
-    {
-      "epoch": 4.26,
-      "learning_rate": 1.7146808907512355e-05,
-      "loss": 0.4837,
-      "step": 2401
-    },
-    {
-      "epoch": 4.26,
-      "learning_rate": 1.7144069156031647e-05,
-      "loss": 0.4919,
-      "step": 2402
-    },
-    {
-      "epoch": 4.27,
-      "learning_rate": 1.7141328308880178e-05,
-      "loss": 0.5102,
-      "step": 2403
-    },
-    {
-      "epoch": 4.27,
-      "learning_rate": 1.71385863664783e-05,
-      "loss": 0.4868,
-      "step": 2404
-    },
-    {
-      "epoch": 4.27,
-      "learning_rate": 1.7135843329246535e-05,
-      "loss": 0.487,
-      "step": 2405
-    },
-    {
-      "epoch": 4.27,
-      "learning_rate": 1.7133099197605584e-05,
-      "loss": 0.5121,
-      "step": 2406
-    },
-    {
-      "epoch": 4.27,
-      "learning_rate": 1.7130353971976307e-05,
-      "loss": 0.4936,
-      "step": 2407
-    },
-    {
-      "epoch": 4.28,
-      "learning_rate": 1.7127607652779728e-05,
-      "loss": 0.4919,
-      "step": 2408
-    },
-    {
-      "epoch": 4.28,
-      "learning_rate": 1.712486024043705e-05,
-      "loss": 0.5152,
-      "step": 2409
-    },
-    {
-      "epoch": 4.28,
-      "learning_rate": 1.7122111735369634e-05,
-      "loss": 0.4843,
-      "step": 2410
-    },
-    {
-      "epoch": 4.28,
-      "learning_rate": 1.7119362137999018e-05,
-      "loss": 0.4878,
-      "step": 2411
-    },
-    {
-      "epoch": 4.28,
-      "learning_rate": 1.7116611448746896e-05,
-      "loss": 0.4993,
-      "step": 2412
-    },
-    {
-      "epoch": 4.28,
-      "learning_rate": 1.7113859668035136e-05,
-      "loss": 0.5075,
-      "step": 2413
-    },
-    {
-      "epoch": 4.29,
-      "learning_rate": 1.711110679628578e-05,
-      "loss": 0.4946,
-      "step": 2414
-    },
-    {
-      "epoch": 4.29,
-      "learning_rate": 1.7108352833921022e-05,
-      "loss": 0.5047,
-      "step": 2415
-    },
-    {
-      "epoch": 4.29,
-      "learning_rate": 1.710559778136323e-05,
-      "loss": 0.4704,
-      "step": 2416
-    },
-    {
-      "epoch": 4.29,
-      "learning_rate": 1.710284163903495e-05,
-      "loss": 0.5061,
-      "step": 2417
-    },
-    {
-      "epoch": 4.29,
-      "learning_rate": 1.7100084407358882e-05,
-      "loss": 0.4752,
-      "step": 2418
-    },
-    {
-      "epoch": 4.29,
-      "learning_rate": 1.709732608675789e-05,
-      "loss": 0.5201,
-      "step": 2419
-    },
-    {
-      "epoch": 4.3,
-      "learning_rate": 1.709456667765502e-05,
-      "loss": 0.4911,
-      "step": 2420
-    },
-    {
-      "epoch": 4.3,
-      "learning_rate": 1.7091806180473472e-05,
-      "loss": 0.5022,
-      "step": 2421
-    },
-    {
-      "epoch": 4.3,
-      "learning_rate": 1.708904459563662e-05,
-      "loss": 0.503,
-      "step": 2422
-    },
-    {
-      "epoch": 4.3,
-      "learning_rate": 1.7086281923568e-05,
-      "loss": 0.5002,
-      "step": 2423
-    },
-    {
-      "epoch": 4.3,
-      "learning_rate": 1.708351816469132e-05,
-      "loss": 0.4686,
-      "step": 2424
-    },
-    {
-      "epoch": 4.31,
-      "learning_rate": 1.7080753319430452e-05,
-      "loss": 0.496,
-      "step": 2425
-    },
-    {
-      "epoch": 4.31,
-      "learning_rate": 1.7077987388209432e-05,
-      "loss": 0.4881,
-      "step": 2426
-    },
-    {
-      "epoch": 4.31,
-      "learning_rate": 1.7075220371452465e-05,
-      "loss": 0.5217,
-      "step": 2427
-    },
-    {
-      "epoch": 4.31,
-      "learning_rate": 1.7072452269583924e-05,
-      "loss": 0.5134,
-      "step": 2428
-    },
-    {
-      "epoch": 4.31,
-      "learning_rate": 1.706968308302835e-05,
-      "loss": 0.4821,
-      "step": 2429
-    },
-    {
-      "epoch": 4.31,
-      "learning_rate": 1.706691281221044e-05,
-      "loss": 0.4811,
-      "step": 2430
-    },
-    {
-      "epoch": 4.32,
-      "learning_rate": 1.706414145755507e-05,
-      "loss": 0.498,
-      "step": 2431
-    },
-    {
-      "epoch": 4.32,
-      "learning_rate": 1.7061369019487278e-05,
-      "loss": 0.4793,
-      "step": 2432
-    },
-    {
-      "epoch": 4.32,
-      "learning_rate": 1.705859549843226e-05,
-      "loss": 0.5042,
-      "step": 2433
-    },
-    {
-      "epoch": 4.32,
-      "learning_rate": 1.705582089481539e-05,
-      "loss": 0.4875,
-      "step": 2434
-    },
-    {
-      "epoch": 4.32,
-      "learning_rate": 1.7053045209062203e-05,
-      "loss": 0.4715,
-      "step": 2435
-    },
-    {
-      "epoch": 4.32,
-      "learning_rate": 1.7050268441598402e-05,
-      "loss": 0.4918,
-      "step": 2436
-    },
-    {
-      "epoch": 4.33,
-      "learning_rate": 1.7047490592849847e-05,
-      "loss": 0.4903,
-      "step": 2437
-    },
-    {
-      "epoch": 4.33,
-      "learning_rate": 1.7044711663242576e-05,
-      "loss": 0.5048,
-      "step": 2438
-    },
-    {
-      "epoch": 4.33,
-      "learning_rate": 1.7041931653202788e-05,
-      "loss": 0.4956,
-      "step": 2439
-    },
-    {
-      "epoch": 4.33,
-      "learning_rate": 1.7039150563156846e-05,
-      "loss": 0.4896,
-      "step": 2440
-    },
-    {
-      "epoch": 4.33,
-      "learning_rate": 1.703636839353128e-05,
-      "loss": 0.4755,
-      "step": 2441
-    },
-    {
-      "epoch": 4.34,
-      "learning_rate": 1.7033585144752782e-05,
-      "loss": 0.5131,
-      "step": 2442
-    },
-    {
-      "epoch": 4.34,
-      "learning_rate": 1.7030800817248223e-05,
-      "loss": 0.5142,
-      "step": 2443
-    },
-    {
-      "epoch": 4.34,
-      "learning_rate": 1.7028015411444616e-05,
-      "loss": 0.5055,
-      "step": 2444
-    },
-    {
-      "epoch": 4.34,
-      "learning_rate": 1.702522892776916e-05,
-      "loss": 0.4938,
-      "step": 2445
-    },
-    {
-      "epoch": 4.34,
-      "learning_rate": 1.7022441366649216e-05,
-      "loss": 0.5084,
-      "step": 2446
-    },
-    {
-      "epoch": 4.34,
-      "learning_rate": 1.70196527285123e-05,
-      "loss": 0.4999,
-      "step": 2447
-    },
-    {
-      "epoch": 4.35,
-      "learning_rate": 1.70168630137861e-05,
-      "loss": 0.4815,
-      "step": 2448
-    },
-    {
-      "epoch": 4.35,
-      "learning_rate": 1.7014072222898472e-05,
-      "loss": 0.5133,
-      "step": 2449
-    },
-    {
-      "epoch": 4.35,
-      "learning_rate": 1.7011280356277432e-05,
-      "loss": 0.4936,
-      "step": 2450
-    },
-    {
-      "epoch": 4.35,
-      "learning_rate": 1.7008487414351163e-05,
-      "loss": 0.5301,
-      "step": 2451
-    },
-    {
-      "epoch": 4.35,
-      "learning_rate": 1.700569339754801e-05,
-      "loss": 0.4843,
-      "step": 2452
-    },
-    {
-      "epoch": 4.36,
-      "learning_rate": 1.7002898306296486e-05,
-      "loss": 0.4823,
-      "step": 2453
-    },
-    {
-      "epoch": 4.36,
-      "learning_rate": 1.7000102141025272e-05,
-      "loss": 0.4983,
-      "step": 2454
-    },
-    {
-      "epoch": 4.36,
-      "learning_rate": 1.6997304902163206e-05,
-      "loss": 0.4709,
-      "step": 2455
-    },
-    {
-      "epoch": 4.36,
-      "learning_rate": 1.69945065901393e-05,
-      "loss": 0.5011,
-      "step": 2456
-    },
-    {
-      "epoch": 4.36,
-      "learning_rate": 1.6991707205382723e-05,
-      "loss": 0.4845,
-      "step": 2457
-    },
-    {
-      "epoch": 4.36,
-      "learning_rate": 1.6988906748322803e-05,
-      "loss": 0.4864,
-      "step": 2458
-    },
-    {
-      "epoch": 4.37,
-      "learning_rate": 1.698610521938905e-05,
-      "loss": 0.4947,
-      "step": 2459
-    },
-    {
-      "epoch": 4.37,
-      "learning_rate": 1.6983302619011125e-05,
-      "loss": 0.4824,
-      "step": 2460
-    },
-    {
-      "epoch": 4.37,
-      "learning_rate": 1.6980498947618854e-05,
-      "loss": 0.4799,
-      "step": 2461
-    },
-    {
-      "epoch": 4.37,
-      "learning_rate": 1.6977694205642237e-05,
-      "loss": 0.476,
-      "step": 2462
-    },
-    {
-      "epoch": 4.37,
-      "learning_rate": 1.6974888393511425e-05,
-      "loss": 0.4672,
-      "step": 2463
-    },
-    {
-      "epoch": 4.37,
-      "learning_rate": 1.697208151165674e-05,
-      "loss": 0.4551,
-      "step": 2464
-    },
-    {
-      "epoch": 4.38,
-      "learning_rate": 1.6969273560508674e-05,
-      "loss": 0.4946,
-      "step": 2465
-    },
-    {
-      "epoch": 4.38,
-      "learning_rate": 1.6966464540497865e-05,
-      "loss": 0.4959,
-      "step": 2466
-    },
-    {
-      "epoch": 4.38,
-      "learning_rate": 1.696365445205514e-05,
-      "loss": 0.5026,
-      "step": 2467
-    },
-    {
-      "epoch": 4.38,
-      "learning_rate": 1.6960843295611466e-05,
-      "loss": 0.4972,
-      "step": 2468
-    },
-    {
-      "epoch": 4.38,
-      "learning_rate": 1.695803107159799e-05,
-      "loss": 0.4959,
-      "step": 2469
-    },
-    {
-      "epoch": 4.39,
-      "learning_rate": 1.6955217780446013e-05,
-      "loss": 0.5018,
-      "step": 2470
-    },
-    {
-      "epoch": 4.39,
-      "learning_rate": 1.6952403422587006e-05,
-      "loss": 0.4885,
-      "step": 2471
-    },
-    {
-      "epoch": 4.39,
-      "learning_rate": 1.6949587998452598e-05,
-      "loss": 0.5082,
-      "step": 2472
-    },
-    {
-      "epoch": 4.39,
-      "learning_rate": 1.6946771508474592e-05,
-      "loss": 0.5147,
-      "step": 2473
-    },
-    {
-      "epoch": 4.39,
-      "learning_rate": 1.694395395308494e-05,
-      "loss": 0.4952,
-      "step": 2474
-    },
-    {
-      "epoch": 4.39,
-      "learning_rate": 1.694113533271576e-05,
-      "loss": 0.5022,
-      "step": 2475
-    },
-    {
-      "epoch": 4.4,
-      "learning_rate": 1.6938315647799354e-05,
-      "loss": 0.4703,
-      "step": 2476
-    },
-    {
-      "epoch": 4.4,
-      "learning_rate": 1.6935494898768155e-05,
-      "loss": 0.5059,
-      "step": 2477
-    },
-    {
-      "epoch": 4.4,
-      "learning_rate": 1.6932673086054784e-05,
-      "loss": 0.4956,
-      "step": 2478
-    },
-    {
-      "epoch": 4.4,
-      "learning_rate": 1.6929850210092013e-05,
-      "loss": 0.4802,
-      "step": 2479
-    },
-    {
-      "epoch": 4.4,
-      "learning_rate": 1.6927026271312785e-05,
-      "loss": 0.4846,
-      "step": 2480
-    },
-    {
-      "epoch": 4.4,
-      "learning_rate": 1.6924201270150194e-05,
-      "loss": 0.4859,
-      "step": 2481
-    },
-    {
-      "epoch": 4.41,
-      "learning_rate": 1.6921375207037507e-05,
-      "loss": 0.4903,
-      "step": 2482
-    },
-    {
-      "epoch": 4.41,
-      "learning_rate": 1.6918548082408156e-05,
-      "loss": 0.5012,
-      "step": 2483
-    },
-    {
-      "epoch": 4.41,
-      "learning_rate": 1.691571989669573e-05,
-      "loss": 0.4662,
-      "step": 2484
-    },
-    {
-      "epoch": 4.41,
-      "learning_rate": 1.691289065033398e-05,
-      "loss": 0.471,
-      "step": 2485
-    },
-    {
-      "epoch": 4.41,
-      "learning_rate": 1.691006034375682e-05,
-      "loss": 0.4837,
-      "step": 2486
-    },
-    {
-      "epoch": 4.42,
-      "learning_rate": 1.6907228977398324e-05,
-      "loss": 0.491,
-      "step": 2487
-    },
-    {
-      "epoch": 4.42,
-      "learning_rate": 1.6904396551692744e-05,
-      "loss": 0.5036,
-      "step": 2488
-    },
-    {
-      "epoch": 4.42,
-      "learning_rate": 1.690156306707448e-05,
-      "loss": 0.489,
-      "step": 2489
-    },
-    {
-      "epoch": 4.42,
-      "learning_rate": 1.689872852397809e-05,
-      "loss": 0.4912,
-      "step": 2490
-    },
-    {
-      "epoch": 4.42,
-      "learning_rate": 1.689589292283831e-05,
-      "loss": 0.4811,
-      "step": 2491
-    },
-    {
-      "epoch": 4.42,
-      "learning_rate": 1.6893056264090025e-05,
-      "loss": 0.4956,
-      "step": 2492
-    },
-    {
-      "epoch": 4.43,
-      "learning_rate": 1.689021854816829e-05,
-      "loss": 0.4816,
-      "step": 2493
-    },
-    {
-      "epoch": 4.43,
-      "learning_rate": 1.6887379775508316e-05,
-      "loss": 0.4861,
-      "step": 2494
-    },
-    {
-      "epoch": 4.43,
-      "learning_rate": 1.6884539946545486e-05,
-      "loss": 0.5166,
-      "step": 2495
-    },
-    {
-      "epoch": 4.43,
-      "learning_rate": 1.6881699061715334e-05,
-      "loss": 0.5224,
-      "step": 2496
-    },
-    {
-      "epoch": 4.43,
-      "learning_rate": 1.687885712145356e-05,
-      "loss": 0.4967,
-      "step": 2497
-    },
-    {
-      "epoch": 4.43,
-      "learning_rate": 1.687601412619603e-05,
-      "loss": 0.4672,
-      "step": 2498
-    },
-    {
-      "epoch": 4.44,
-      "learning_rate": 1.6873170076378764e-05,
-      "loss": 0.4679,
-      "step": 2499
-    },
-    {
-      "epoch": 4.44,
-      "learning_rate": 1.687032497243795e-05,
-      "loss": 0.4967,
-      "step": 2500
-    },
-    {
-      "epoch": 4.44,
-      "learning_rate": 1.6867478814809937e-05,
-      "loss": 0.4967,
-      "step": 2501
-    },
-    {
-      "epoch": 4.44,
-      "learning_rate": 1.686463160393123e-05,
-      "loss": 0.5059,
-      "step": 2502
-    },
-    {
-      "epoch": 4.44,
-      "learning_rate": 1.6861783340238504e-05,
-      "loss": 0.5125,
-      "step": 2503
-    },
-    {
-      "epoch": 4.45,
-      "learning_rate": 1.6858934024168593e-05,
-      "loss": 0.4876,
-      "step": 2504
-    },
-    {
-      "epoch": 4.45,
-      "learning_rate": 1.685608365615848e-05,
-      "loss": 0.4763,
-      "step": 2505
-    },
-    {
-      "epoch": 4.45,
-      "learning_rate": 1.6853232236645333e-05,
-      "loss": 0.5086,
-      "step": 2506
-    },
-    {
-      "epoch": 4.45,
-      "learning_rate": 1.6850379766066457e-05,
-      "loss": 0.5147,
-      "step": 2507
-    },
-    {
-      "epoch": 4.45,
-      "learning_rate": 1.6847526244859338e-05,
-      "loss": 0.5092,
-      "step": 2508
-    },
-    {
-      "epoch": 4.45,
-      "learning_rate": 1.6844671673461607e-05,
-      "loss": 0.499,
-      "step": 2509
-    },
-    {
-      "epoch": 4.46,
-      "learning_rate": 1.6841816052311073e-05,
-      "loss": 0.4846,
-      "step": 2510
-    },
-    {
-      "epoch": 4.46,
-      "learning_rate": 1.683895938184569e-05,
-      "loss": 0.4908,
-      "step": 2511
-    },
-    {
-      "epoch": 4.46,
-      "learning_rate": 1.683610166250358e-05,
-      "loss": 0.4983,
-      "step": 2512
-    },
-    {
-      "epoch": 4.46,
-      "learning_rate": 1.683324289472302e-05,
-      "loss": 0.5029,
-      "step": 2513
-    },
-    {
-      "epoch": 4.46,
-      "learning_rate": 1.683038307894247e-05,
-      "loss": 0.5059,
-      "step": 2514
-    },
-    {
-      "epoch": 4.47,
-      "learning_rate": 1.6827522215600515e-05,
-      "loss": 0.4786,
-      "step": 2515
-    },
-    {
-      "epoch": 4.47,
-      "learning_rate": 1.682466030513593e-05,
-      "loss": 0.5136,
-      "step": 2516
-    },
-    {
-      "epoch": 4.47,
-      "learning_rate": 1.6821797347987637e-05,
-      "loss": 0.4529,
-      "step": 2517
-    },
-    {
-      "epoch": 4.47,
-      "learning_rate": 1.6818933344594724e-05,
-      "loss": 0.4722,
-      "step": 2518
-    },
-    {
-      "epoch": 4.47,
-      "learning_rate": 1.6816068295396434e-05,
-      "loss": 0.4815,
-      "step": 2519
-    },
-    {
-      "epoch": 4.47,
-      "learning_rate": 1.6813202200832175e-05,
-      "loss": 0.4888,
-      "step": 2520
-    },
-    {
-      "epoch": 4.48,
-      "learning_rate": 1.6810335061341512e-05,
-      "loss": 0.4948,
-      "step": 2521
-    },
-    {
-      "epoch": 4.48,
-      "learning_rate": 1.6807466877364182e-05,
-      "loss": 0.5163,
-      "step": 2522
-    },
-    {
-      "epoch": 4.48,
-      "learning_rate": 1.680459764934006e-05,
-      "loss": 0.4857,
-      "step": 2523
-    },
-    {
-      "epoch": 4.48,
-      "learning_rate": 1.6801727377709195e-05,
-      "loss": 0.4895,
-      "step": 2524
-    },
-    {
-      "epoch": 4.48,
-      "learning_rate": 1.6798856062911802e-05,
-      "loss": 0.4757,
-      "step": 2525
-    },
-    {
-      "epoch": 4.48,
-      "learning_rate": 1.6795983705388242e-05,
-      "loss": 0.4943,
-      "step": 2526
-    },
-    {
-      "epoch": 4.49,
-      "learning_rate": 1.6793110305579043e-05,
-      "loss": 0.4782,
-      "step": 2527
-    },
-    {
-      "epoch": 4.49,
-      "learning_rate": 1.679023586392489e-05,
-      "loss": 0.5191,
-      "step": 2528
-    },
-    {
-      "epoch": 4.49,
-      "learning_rate": 1.678736038086664e-05,
-      "loss": 0.5214,
-      "step": 2529
-    },
-    {
-      "epoch": 4.49,
-      "learning_rate": 1.6784483856845287e-05,
-      "loss": 0.4964,
-      "step": 2530
-    },
-    {
-      "epoch": 4.49,
-      "learning_rate": 1.6781606292302012e-05,
-      "loss": 0.4923,
-      "step": 2531
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 1.6778727687678125e-05,
-      "loss": 0.4851,
-      "step": 2532
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 1.6775848043415122e-05,
-      "loss": 0.5021,
-      "step": 2533
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 1.6772967359954647e-05,
-      "loss": 0.5039,
-      "step": 2534
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 1.67700856377385e-05,
-      "loss": 0.5003,
-      "step": 2535
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 1.6767202877208645e-05,
-      "loss": 0.4948,
-      "step": 2536
-    },
-    {
-      "epoch": 4.5,
-      "learning_rate": 1.676431907880721e-05,
-      "loss": 0.4781,
-      "step": 2537
-    },
-    {
-      "epoch": 4.51,
-      "learning_rate": 1.6761434242976474e-05,
-      "loss": 0.4904,
-      "step": 2538
-    },
-    {
-      "epoch": 4.51,
-      "learning_rate": 1.6758548370158877e-05,
-      "loss": 0.4888,
-      "step": 2539
-    },
-    {
-      "epoch": 4.51,
-      "learning_rate": 1.675566146079702e-05,
-      "loss": 0.4967,
-      "step": 2540
-    },
-    {
-      "epoch": 4.51,
-      "learning_rate": 1.6752773515333665e-05,
-      "loss": 0.4931,
-      "step": 2541
-    },
-    {
-      "epoch": 4.51,
-      "learning_rate": 1.6749884534211723e-05,
-      "loss": 0.5009,
-      "step": 2542
-    },
-    {
-      "epoch": 4.51,
-      "learning_rate": 1.674699451787428e-05,
-      "loss": 0.5092,
-      "step": 2543
-    },
-    {
-      "epoch": 4.52,
-      "learning_rate": 1.6744103466764566e-05,
-      "loss": 0.4981,
-      "step": 2544
-    },
-    {
-      "epoch": 4.52,
-      "learning_rate": 1.6741211381325974e-05,
-      "loss": 0.5101,
-      "step": 2545
-    },
-    {
-      "epoch": 4.52,
-      "learning_rate": 1.673831826200206e-05,
-      "loss": 0.4873,
-      "step": 2546
-    },
-    {
-      "epoch": 4.52,
-      "learning_rate": 1.6735424109236538e-05,
-      "loss": 0.468,
-      "step": 2547
-    },
-    {
-      "epoch": 4.52,
-      "learning_rate": 1.6732528923473276e-05,
-      "loss": 0.4971,
-      "step": 2548
-    },
-    {
-      "epoch": 4.53,
-      "learning_rate": 1.6729632705156298e-05,
-      "loss": 0.4987,
-      "step": 2549
-    },
-    {
-      "epoch": 4.53,
-      "learning_rate": 1.6726735454729796e-05,
-      "loss": 0.5085,
-      "step": 2550
-    },
-    {
-      "epoch": 4.53,
-      "learning_rate": 1.6723837172638117e-05,
-      "loss": 0.4714,
-      "step": 2551
-    },
-    {
-      "epoch": 4.53,
-      "learning_rate": 1.672093785932576e-05,
-      "loss": 0.4986,
-      "step": 2552
-    },
-    {
-      "epoch": 4.53,
-      "learning_rate": 1.671803751523738e-05,
-      "loss": 0.4859,
-      "step": 2553
-    },
-    {
-      "epoch": 4.53,
-      "learning_rate": 1.6715136140817815e-05,
-      "loss": 0.4998,
-      "step": 2554
-    },
-    {
-      "epoch": 4.54,
-      "learning_rate": 1.6712233736512023e-05,
-      "loss": 0.4808,
-      "step": 2555
-    },
-    {
-      "epoch": 4.54,
-      "learning_rate": 1.6709330302765156e-05,
-      "loss": 0.5111,
-      "step": 2556
-    },
-    {
-      "epoch": 4.54,
-      "learning_rate": 1.6706425840022498e-05,
-      "loss": 0.4768,
-      "step": 2557
-    },
-    {
-      "epoch": 4.54,
-      "learning_rate": 1.6703520348729504e-05,
-      "loss": 0.5068,
-      "step": 2558
-    },
-    {
-      "epoch": 4.54,
-      "learning_rate": 1.6700613829331778e-05,
-      "loss": 0.5259,
-      "step": 2559
-    },
-    {
-      "epoch": 4.55,
-      "learning_rate": 1.669770628227509e-05,
-      "loss": 0.4935,
-      "step": 2560
-    },
-    {
-      "epoch": 4.55,
-      "learning_rate": 1.6694797708005367e-05,
-      "loss": 0.4992,
-      "step": 2561
-    },
-    {
-      "epoch": 4.55,
-      "learning_rate": 1.6691888106968686e-05,
-      "loss": 0.4956,
-      "step": 2562
-    },
-    {
-      "epoch": 4.55,
-      "learning_rate": 1.668897747961129e-05,
-      "loss": 0.5213,
-      "step": 2563
-    },
-    {
-      "epoch": 4.55,
-      "learning_rate": 1.6686065826379577e-05,
-      "loss": 0.4911,
-      "step": 2564
-    },
-    {
-      "epoch": 4.55,
-      "learning_rate": 1.6683153147720098e-05,
-      "loss": 0.4743,
-      "step": 2565
-    },
-    {
-      "epoch": 4.56,
-      "learning_rate": 1.6680239444079564e-05,
-      "loss": 0.5032,
-      "step": 2566
-    },
-    {
-      "epoch": 4.56,
-      "learning_rate": 1.6677324715904843e-05,
-      "loss": 0.4872,
-      "step": 2567
-    },
-    {
-      "epoch": 4.56,
-      "learning_rate": 1.6674408963642963e-05,
-      "loss": 0.4878,
-      "step": 2568
-    },
-    {
-      "epoch": 4.56,
-      "learning_rate": 1.6671492187741108e-05,
-      "loss": 0.5087,
-      "step": 2569
-    },
-    {
-      "epoch": 4.56,
-      "learning_rate": 1.6668574388646615e-05,
-      "loss": 0.501,
-      "step": 2570
-    },
-    {
-      "epoch": 4.56,
-      "learning_rate": 1.666565556680698e-05,
-      "loss": 0.503,
-      "step": 2571
-    },
-    {
-      "epoch": 4.57,
-      "learning_rate": 1.6662735722669857e-05,
-      "loss": 0.4826,
-      "step": 2572
-    },
-    {
-      "epoch": 4.57,
-      "learning_rate": 1.6659814856683062e-05,
-      "loss": 0.4917,
-      "step": 2573
-    },
-    {
-      "epoch": 4.57,
-      "learning_rate": 1.6656892969294555e-05,
-      "loss": 0.4894,
-      "step": 2574
-    },
-    {
-      "epoch": 4.57,
-      "learning_rate": 1.665397006095246e-05,
-      "loss": 0.5011,
-      "step": 2575
-    },
-    {
-      "epoch": 4.57,
-      "learning_rate": 1.665104613210506e-05,
-      "loss": 0.5073,
-      "step": 2576
-    },
-    {
-      "epoch": 4.58,
-      "learning_rate": 1.664812118320079e-05,
-      "loss": 0.4969,
-      "step": 2577
-    },
-    {
-      "epoch": 4.58,
-      "learning_rate": 1.6645195214688247e-05,
-      "loss": 0.4853,
-      "step": 2578
-    },
-    {
-      "epoch": 4.58,
-      "learning_rate": 1.6642268227016177e-05,
-      "loss": 0.5046,
-      "step": 2579
-    },
-    {
-      "epoch": 4.58,
-      "learning_rate": 1.6639340220633486e-05,
-      "loss": 0.4941,
-      "step": 2580
-    },
-    {
-      "epoch": 4.58,
-      "learning_rate": 1.663641119598924e-05,
-      "loss": 0.4849,
-      "step": 2581
-    },
-    {
-      "epoch": 4.58,
-      "learning_rate": 1.663348115353265e-05,
-      "loss": 0.4818,
-      "step": 2582
-    },
-    {
-      "epoch": 4.59,
-      "learning_rate": 1.6630550093713096e-05,
-      "loss": 0.5193,
-      "step": 2583
-    },
-    {
-      "epoch": 4.59,
-      "learning_rate": 1.6627618016980106e-05,
-      "loss": 0.4881,
-      "step": 2584
-    },
-    {
-      "epoch": 4.59,
-      "learning_rate": 1.6624684923783368e-05,
-      "loss": 0.4954,
-      "step": 2585
-    },
-    {
-      "epoch": 4.59,
-      "learning_rate": 1.6621750814572728e-05,
-      "loss": 0.506,
-      "step": 2586
-    },
-    {
-      "epoch": 4.59,
-      "learning_rate": 1.6618815689798176e-05,
-      "loss": 0.4931,
-      "step": 2587
-    },
-    {
-      "epoch": 4.59,
-      "learning_rate": 1.661587954990987e-05,
-      "loss": 0.5093,
-      "step": 2588
-    },
-    {
-      "epoch": 4.6,
-      "learning_rate": 1.661294239535812e-05,
-      "loss": 0.5002,
-      "step": 2589
-    },
-    {
-      "epoch": 4.6,
-      "learning_rate": 1.661000422659339e-05,
-      "loss": 0.4718,
-      "step": 2590
-    },
-    {
-      "epoch": 4.6,
-      "learning_rate": 1.6607065044066303e-05,
-      "loss": 0.5284,
-      "step": 2591
-    },
-    {
-      "epoch": 4.6,
-      "learning_rate": 1.660412484822763e-05,
-      "loss": 0.5158,
-      "step": 2592
-    },
-    {
-      "epoch": 4.6,
-      "learning_rate": 1.660118363952831e-05,
-      "loss": 0.4799,
-      "step": 2593
-    },
-    {
-      "epoch": 4.61,
-      "learning_rate": 1.6598241418419424e-05,
-      "loss": 0.5116,
-      "step": 2594
-    },
-    {
-      "epoch": 4.61,
-      "learning_rate": 1.6595298185352216e-05,
-      "loss": 0.4942,
-      "step": 2595
-    },
-    {
-      "epoch": 4.61,
-      "learning_rate": 1.659235394077808e-05,
-      "loss": 0.5242,
-      "step": 2596
-    },
-    {
-      "epoch": 4.61,
-      "learning_rate": 1.658940868514858e-05,
-      "loss": 0.4821,
-      "step": 2597
-    },
-    {
-      "epoch": 4.61,
-      "learning_rate": 1.658646241891541e-05,
-      "loss": 0.4702,
-      "step": 2598
-    },
-    {
-      "epoch": 4.61,
-      "learning_rate": 1.658351514253044e-05,
-      "loss": 0.4975,
-      "step": 2599
-    },
-    {
-      "epoch": 4.62,
-      "learning_rate": 1.6580566856445684e-05,
-      "loss": 0.5075,
-      "step": 2600
-    },
-    {
-      "epoch": 4.62,
-      "learning_rate": 1.6577617561113322e-05,
-      "loss": 0.4929,
-      "step": 2601
-    },
-    {
-      "epoch": 4.62,
-      "learning_rate": 1.657466725698567e-05,
-      "loss": 0.5232,
-      "step": 2602
-    },
-    {
-      "epoch": 4.62,
-      "learning_rate": 1.6571715944515218e-05,
-      "loss": 0.5005,
-      "step": 2603
-    },
-    {
-      "epoch": 4.62,
-      "learning_rate": 1.65687636241546e-05,
-      "loss": 0.4831,
-      "step": 2604
-    },
-    {
-      "epoch": 4.62,
-      "learning_rate": 1.6565810296356612e-05,
-      "loss": 0.5249,
-      "step": 2605
-    },
-    {
-      "epoch": 4.63,
-      "learning_rate": 1.6562855961574188e-05,
-      "loss": 0.5065,
-      "step": 2606
-    },
-    {
-      "epoch": 4.63,
-      "learning_rate": 1.6559900620260435e-05,
-      "loss": 0.4731,
-      "step": 2607
-    },
-    {
-      "epoch": 4.63,
-      "learning_rate": 1.6556944272868606e-05,
-      "loss": 0.4713,
-      "step": 2608
-    },
-    {
-      "epoch": 4.63,
-      "learning_rate": 1.655398691985212e-05,
-      "loss": 0.4889,
-      "step": 2609
-    },
-    {
-      "epoch": 4.63,
-      "learning_rate": 1.6551028561664523e-05,
-      "loss": 0.4783,
-      "step": 2610
-    },
-    {
-      "epoch": 4.64,
-      "learning_rate": 1.654806919875954e-05,
-      "loss": 0.483,
-      "step": 2611
-    },
-    {
-      "epoch": 4.64,
-      "learning_rate": 1.6545108831591042e-05,
-      "loss": 0.4734,
-      "step": 2612
-    },
-    {
-      "epoch": 4.64,
-      "learning_rate": 1.654214746061306e-05,
-      "loss": 0.4788,
-      "step": 2613
-    },
-    {
-      "epoch": 4.64,
-      "learning_rate": 1.6539185086279767e-05,
-      "loss": 0.5103,
-      "step": 2614
-    },
-    {
-      "epoch": 4.64,
-      "learning_rate": 1.6536221709045493e-05,
-      "loss": 0.4878,
-      "step": 2615
-    },
-    {
-      "epoch": 4.64,
-      "learning_rate": 1.6533257329364732e-05,
-      "loss": 0.491,
-      "step": 2616
-    },
-    {
-      "epoch": 4.65,
-      "learning_rate": 1.653029194769212e-05,
-      "loss": 0.4934,
-      "step": 2617
-    },
-    {
-      "epoch": 4.65,
-      "learning_rate": 1.6527325564482453e-05,
-      "loss": 0.5096,
-      "step": 2618
-    },
-    {
-      "epoch": 4.65,
-      "learning_rate": 1.652435818019068e-05,
-      "loss": 0.4978,
-      "step": 2619
-    },
-    {
-      "epoch": 4.65,
-      "learning_rate": 1.65213897952719e-05,
-      "loss": 0.4972,
-      "step": 2620
-    },
-    {
-      "epoch": 4.65,
-      "learning_rate": 1.6518420410181365e-05,
-      "loss": 0.5035,
-      "step": 2621
-    },
-    {
-      "epoch": 4.66,
-      "learning_rate": 1.651545002537449e-05,
-      "loss": 0.4847,
-      "step": 2622
-    },
-    {
-      "epoch": 4.66,
-      "learning_rate": 1.6512478641306837e-05,
-      "loss": 0.5006,
-      "step": 2623
-    },
-    {
-      "epoch": 4.66,
-      "learning_rate": 1.6509506258434114e-05,
-      "loss": 0.4839,
-      "step": 2624
-    },
-    {
-      "epoch": 4.66,
-      "learning_rate": 1.6506532877212195e-05,
-      "loss": 0.5019,
-      "step": 2625
-    },
-    {
-      "epoch": 4.66,
-      "learning_rate": 1.65035584980971e-05,
-      "loss": 0.489,
-      "step": 2626
-    },
-    {
-      "epoch": 4.66,
-      "learning_rate": 1.6500583121545e-05,
-      "loss": 0.4969,
-      "step": 2627
-    },
-    {
-      "epoch": 4.67,
-      "learning_rate": 1.6497606748012227e-05,
-      "loss": 0.4925,
-      "step": 2628
-    },
-    {
-      "epoch": 4.67,
-      "learning_rate": 1.6494629377955255e-05,
-      "loss": 0.4887,
-      "step": 2629
-    },
-    {
-      "epoch": 4.67,
-      "learning_rate": 1.6491651011830725e-05,
-      "loss": 0.5015,
-      "step": 2630
-    },
-    {
-      "epoch": 4.67,
-      "learning_rate": 1.6488671650095416e-05,
-      "loss": 0.4971,
-      "step": 2631
-    },
-    {
-      "epoch": 4.67,
-      "learning_rate": 1.648569129320627e-05,
-      "loss": 0.4927,
-      "step": 2632
-    },
-    {
-      "epoch": 4.67,
-      "learning_rate": 1.6482709941620373e-05,
-      "loss": 0.489,
-      "step": 2633
-    },
-    {
-      "epoch": 4.68,
-      "learning_rate": 1.6479727595794974e-05,
-      "loss": 0.4681,
-      "step": 2634
-    },
-    {
-      "epoch": 4.68,
-      "learning_rate": 1.647674425618747e-05,
-      "loss": 0.4872,
-      "step": 2635
-    },
-    {
-      "epoch": 4.68,
-      "learning_rate": 1.6473759923255405e-05,
-      "loss": 0.4933,
-      "step": 2636
-    },
-    {
-      "epoch": 4.68,
-      "learning_rate": 1.6470774597456482e-05,
-      "loss": 0.4939,
-      "step": 2637
-    },
-    {
-      "epoch": 4.68,
-      "learning_rate": 1.6467788279248554e-05,
-      "loss": 0.4898,
-      "step": 2638
-    },
-    {
-      "epoch": 4.69,
-      "learning_rate": 1.6464800969089628e-05,
-      "loss": 0.499,
-      "step": 2639
-    },
-    {
-      "epoch": 4.69,
-      "learning_rate": 1.6461812667437854e-05,
-      "loss": 0.493,
-      "step": 2640
-    },
-    {
-      "epoch": 4.69,
-      "learning_rate": 1.645882337475155e-05,
-      "loss": 0.4842,
-      "step": 2641
-    },
-    {
-      "epoch": 4.69,
-      "learning_rate": 1.6455833091489177e-05,
-      "loss": 0.4826,
-      "step": 2642
-    },
-    {
-      "epoch": 4.69,
-      "learning_rate": 1.6452841818109342e-05,
-      "loss": 0.4903,
-      "step": 2643
-    },
-    {
-      "epoch": 4.69,
-      "learning_rate": 1.6449849555070814e-05,
-      "loss": 0.5076,
-      "step": 2644
-    },
-    {
-      "epoch": 4.7,
-      "learning_rate": 1.6446856302832513e-05,
-      "loss": 0.5071,
-      "step": 2645
-    },
-    {
-      "epoch": 4.7,
-      "learning_rate": 1.6443862061853507e-05,
-      "loss": 0.4698,
-      "step": 2646
-    },
-    {
-      "epoch": 4.7,
-      "learning_rate": 1.644086683259301e-05,
-      "loss": 0.5143,
-      "step": 2647
-    },
-    {
-      "epoch": 4.7,
-      "learning_rate": 1.64378706155104e-05,
-      "loss": 0.5161,
-      "step": 2648
-    },
-    {
-      "epoch": 4.7,
-      "learning_rate": 1.64348734110652e-05,
-      "loss": 0.4736,
-      "step": 2649
-    },
-    {
-      "epoch": 4.7,
-      "learning_rate": 1.6431875219717083e-05,
-      "loss": 0.4978,
-      "step": 2650
-    },
-    {
-      "epoch": 4.71,
-      "learning_rate": 1.6428876041925877e-05,
-      "loss": 0.4805,
-      "step": 2651
-    },
-    {
-      "epoch": 4.71,
-      "learning_rate": 1.6425875878151555e-05,
-      "loss": 0.4838,
-      "step": 2652
-    },
-    {
-      "epoch": 4.71,
-      "learning_rate": 1.6422874728854254e-05,
-      "loss": 0.5164,
-      "step": 2653
-    },
-    {
-      "epoch": 4.71,
-      "learning_rate": 1.6419872594494247e-05,
-      "loss": 0.5162,
-      "step": 2654
-    },
-    {
-      "epoch": 4.71,
-      "learning_rate": 1.6416869475531964e-05,
-      "loss": 0.493,
-      "step": 2655
-    },
-    {
-      "epoch": 4.72,
-      "learning_rate": 1.6413865372427996e-05,
-      "loss": 0.4912,
-      "step": 2656
-    },
-    {
-      "epoch": 4.72,
-      "learning_rate": 1.6410860285643067e-05,
-      "loss": 0.5078,
-      "step": 2657
-    },
-    {
-      "epoch": 4.72,
-      "learning_rate": 1.6407854215638065e-05,
-      "loss": 0.4952,
-      "step": 2658
-    },
-    {
-      "epoch": 4.72,
-      "learning_rate": 1.6404847162874022e-05,
-      "loss": 0.4927,
-      "step": 2659
-    },
-    {
-      "epoch": 4.72,
-      "learning_rate": 1.6401839127812125e-05,
-      "loss": 0.4654,
-      "step": 2660
-    },
-    {
-      "epoch": 4.72,
-      "learning_rate": 1.639883011091371e-05,
-      "loss": 0.4913,
-      "step": 2661
-    },
-    {
-      "epoch": 4.73,
-      "learning_rate": 1.6395820112640265e-05,
-      "loss": 0.4868,
-      "step": 2662
-    },
-    {
-      "epoch": 4.73,
-      "learning_rate": 1.6392809133453423e-05,
-      "loss": 0.4978,
-      "step": 2663
-    },
-    {
-      "epoch": 4.73,
-      "learning_rate": 1.6389797173814978e-05,
-      "loss": 0.4947,
-      "step": 2664
-    },
-    {
-      "epoch": 4.73,
-      "learning_rate": 1.6386784234186863e-05,
-      "loss": 0.5176,
-      "step": 2665
-    },
-    {
-      "epoch": 4.73,
-      "learning_rate": 1.6383770315031165e-05,
-      "loss": 0.5214,
-      "step": 2666
-    },
-    {
-      "epoch": 4.74,
-      "learning_rate": 1.6380755416810125e-05,
-      "loss": 0.4939,
-      "step": 2667
-    },
-    {
-      "epoch": 4.74,
-      "learning_rate": 1.637773953998613e-05,
-      "loss": 0.5161,
-      "step": 2668
-    },
-    {
-      "epoch": 4.74,
-      "learning_rate": 1.6374722685021722e-05,
-      "loss": 0.4997,
-      "step": 2669
-    },
-    {
-      "epoch": 4.74,
-      "learning_rate": 1.6371704852379587e-05,
-      "loss": 0.4981,
-      "step": 2670
-    },
-    {
-      "epoch": 4.74,
-      "learning_rate": 1.636868604252256e-05,
-      "loss": 0.4794,
-      "step": 2671
-    },
-    {
-      "epoch": 4.74,
-      "learning_rate": 1.636566625591364e-05,
-      "loss": 0.4993,
-      "step": 2672
-    },
-    {
-      "epoch": 4.75,
-      "learning_rate": 1.6362645493015955e-05,
-      "loss": 0.5214,
-      "step": 2673
-    },
-    {
-      "epoch": 4.75,
-      "learning_rate": 1.6359623754292795e-05,
-      "loss": 0.4975,
-      "step": 2674
-    },
-    {
-      "epoch": 4.75,
-      "learning_rate": 1.63566010402076e-05,
-      "loss": 0.5114,
-      "step": 2675
-    },
-    {
-      "epoch": 4.75,
-      "learning_rate": 1.635357735122396e-05,
-      "loss": 0.4959,
-      "step": 2676
-    },
-    {
-      "epoch": 4.75,
-      "learning_rate": 1.6350552687805604e-05,
-      "loss": 0.5092,
-      "step": 2677
-    },
-    {
-      "epoch": 4.75,
-      "learning_rate": 1.6347527050416427e-05,
-      "loss": 0.4939,
-      "step": 2678
-    },
-    {
-      "epoch": 4.76,
-      "learning_rate": 1.634450043952046e-05,
-      "loss": 0.4977,
-      "step": 2679
-    },
-    {
-      "epoch": 4.76,
-      "learning_rate": 1.6341472855581884e-05,
-      "loss": 0.4836,
-      "step": 2680
-    },
-    {
-      "epoch": 4.76,
-      "learning_rate": 1.633844429906504e-05,
-      "loss": 0.5013,
-      "step": 2681
-    },
-    {
-      "epoch": 4.76,
-      "learning_rate": 1.6335414770434407e-05,
-      "loss": 0.4858,
-      "step": 2682
-    },
-    {
-      "epoch": 4.76,
-      "learning_rate": 1.633238427015462e-05,
-      "loss": 0.5024,
-      "step": 2683
-    },
-    {
-      "epoch": 4.77,
-      "learning_rate": 1.632935279869046e-05,
-      "loss": 0.5029,
-      "step": 2684
-    },
-    {
-      "epoch": 4.77,
-      "learning_rate": 1.6326320356506858e-05,
-      "loss": 0.4891,
-      "step": 2685
-    },
-    {
-      "epoch": 4.77,
-      "learning_rate": 1.6323286944068888e-05,
-      "loss": 0.4918,
-      "step": 2686
-    },
-    {
-      "epoch": 4.77,
-      "learning_rate": 1.6320252561841784e-05,
-      "loss": 0.5005,
-      "step": 2687
-    },
-    {
-      "epoch": 4.77,
-      "learning_rate": 1.6317217210290922e-05,
-      "loss": 0.5011,
-      "step": 2688
-    },
-    {
-      "epoch": 4.77,
-      "learning_rate": 1.631418088988183e-05,
-      "loss": 0.4951,
-      "step": 2689
-    },
-    {
-      "epoch": 4.78,
-      "learning_rate": 1.631114360108017e-05,
-      "loss": 0.484,
-      "step": 2690
-    },
-    {
-      "epoch": 4.78,
-      "learning_rate": 1.6308105344351776e-05,
-      "loss": 0.486,
-      "step": 2691
-    },
-    {
-      "epoch": 4.78,
-      "learning_rate": 1.630506612016262e-05,
-      "loss": 0.4843,
-      "step": 2692
-    },
-    {
-      "epoch": 4.78,
-      "learning_rate": 1.630202592897881e-05,
-      "loss": 0.506,
-      "step": 2693
-    },
-    {
-      "epoch": 4.78,
-      "learning_rate": 1.6298984771266623e-05,
-      "loss": 0.4988,
-      "step": 2694
-    },
-    {
-      "epoch": 4.78,
-      "learning_rate": 1.6295942647492475e-05,
-      "loss": 0.4884,
-      "step": 2695
-    },
-    {
-      "epoch": 4.79,
-      "learning_rate": 1.629289955812293e-05,
-      "loss": 0.4836,
-      "step": 2696
-    },
-    {
-      "epoch": 4.79,
-      "learning_rate": 1.6289855503624694e-05,
-      "loss": 0.4936,
-      "step": 2697
-    },
-    {
-      "epoch": 4.79,
-      "learning_rate": 1.6286810484464633e-05,
-      "loss": 0.49,
-      "step": 2698
-    },
-    {
-      "epoch": 4.79,
-      "learning_rate": 1.6283764501109753e-05,
-      "loss": 0.4905,
-      "step": 2699
-    },
-    {
-      "epoch": 4.79,
-      "learning_rate": 1.6280717554027213e-05,
-      "loss": 0.4876,
-      "step": 2700
-    },
-    {
-      "epoch": 4.8,
-      "learning_rate": 1.627766964368431e-05,
-      "loss": 0.4663,
-      "step": 2701
-    },
-    {
-      "epoch": 4.8,
-      "learning_rate": 1.62746207705485e-05,
-      "loss": 0.4993,
-      "step": 2702
-    },
-    {
-      "epoch": 4.8,
-      "learning_rate": 1.6271570935087385e-05,
-      "loss": 0.5138,
-      "step": 2703
-    },
-    {
-      "epoch": 4.8,
-      "learning_rate": 1.626852013776871e-05,
-      "loss": 0.5119,
-      "step": 2704
-    },
-    {
-      "epoch": 4.8,
-      "learning_rate": 1.6265468379060364e-05,
-      "loss": 0.4841,
-      "step": 2705
-    },
-    {
-      "epoch": 4.8,
-      "learning_rate": 1.6262415659430396e-05,
-      "loss": 0.4794,
-      "step": 2706
-    },
-    {
-      "epoch": 4.81,
-      "learning_rate": 1.6259361979346994e-05,
-      "loss": 0.5029,
-      "step": 2707
-    },
-    {
-      "epoch": 4.81,
-      "learning_rate": 1.6256307339278486e-05,
-      "loss": 0.4785,
-      "step": 2708
-    },
-    {
-      "epoch": 4.81,
-      "learning_rate": 1.6253251739693366e-05,
-      "loss": 0.4899,
-      "step": 2709
-    },
-    {
-      "epoch": 4.81,
-      "learning_rate": 1.625019518106026e-05,
-      "loss": 0.5133,
-      "step": 2710
-    },
-    {
-      "epoch": 4.81,
-      "learning_rate": 1.624713766384795e-05,
-      "loss": 0.5006,
-      "step": 2711
-    },
-    {
-      "epoch": 4.81,
-      "learning_rate": 1.6244079188525358e-05,
-      "loss": 0.4945,
-      "step": 2712
-    },
-    {
-      "epoch": 4.82,
-      "learning_rate": 1.6241019755561552e-05,
-      "loss": 0.4835,
-      "step": 2713
-    },
-    {
-      "epoch": 4.82,
-      "learning_rate": 1.623795936542576e-05,
-      "loss": 0.4984,
-      "step": 2714
-    },
-    {
-      "epoch": 4.82,
-      "learning_rate": 1.6234898018587336e-05,
-      "loss": 0.4892,
-      "step": 2715
-    },
-    {
-      "epoch": 4.82,
-      "learning_rate": 1.6231835715515803e-05,
-      "loss": 0.4863,
-      "step": 2716
-    },
-    {
-      "epoch": 4.82,
-      "learning_rate": 1.6228772456680815e-05,
-      "loss": 0.483,
-      "step": 2717
-    },
-    {
-      "epoch": 4.83,
-      "learning_rate": 1.6225708242552175e-05,
-      "loss": 0.5023,
-      "step": 2718
-    },
-    {
-      "epoch": 4.83,
-      "learning_rate": 1.6222643073599842e-05,
-      "loss": 0.4698,
-      "step": 2719
-    },
-    {
-      "epoch": 4.83,
-      "learning_rate": 1.6219576950293908e-05,
-      "loss": 0.4819,
-      "step": 2720
-    },
-    {
-      "epoch": 4.83,
-      "learning_rate": 1.621650987310462e-05,
-      "loss": 0.4855,
-      "step": 2721
-    },
-    {
-      "epoch": 4.83,
-      "learning_rate": 1.621344184250237e-05,
-      "loss": 0.488,
-      "step": 2722
-    },
-    {
-      "epoch": 4.83,
-      "learning_rate": 1.6210372858957694e-05,
-      "loss": 0.4955,
-      "step": 2723
-    },
-    {
-      "epoch": 4.84,
-      "learning_rate": 1.6207302922941277e-05,
-      "loss": 0.4911,
-      "step": 2724
-    },
-    {
-      "epoch": 4.84,
-      "learning_rate": 1.620423203492395e-05,
-      "loss": 0.5054,
-      "step": 2725
-    },
-    {
-      "epoch": 4.84,
-      "learning_rate": 1.6201160195376685e-05,
-      "loss": 0.5275,
-      "step": 2726
-    },
-    {
-      "epoch": 4.84,
-      "learning_rate": 1.6198087404770602e-05,
-      "loss": 0.4931,
-      "step": 2727
-    },
-    {
-      "epoch": 4.84,
-      "learning_rate": 1.6195013663576973e-05,
-      "loss": 0.4949,
-      "step": 2728
-    },
-    {
-      "epoch": 4.85,
-      "learning_rate": 1.6191938972267213e-05,
-      "loss": 0.5011,
-      "step": 2729
-    },
-    {
-      "epoch": 4.85,
-      "learning_rate": 1.6188863331312874e-05,
-      "loss": 0.4661,
-      "step": 2730
-    },
-    {
-      "epoch": 4.85,
-      "learning_rate": 1.6185786741185662e-05,
-      "loss": 0.482,
-      "step": 2731
-    },
-    {
-      "epoch": 4.85,
-      "learning_rate": 1.6182709202357433e-05,
-      "loss": 0.4848,
-      "step": 2732
-    },
-    {
-      "epoch": 4.85,
-      "learning_rate": 1.617963071530018e-05,
-      "loss": 0.4957,
-      "step": 2733
-    },
-    {
-      "epoch": 4.85,
-      "learning_rate": 1.617655128048604e-05,
-      "loss": 0.4873,
-      "step": 2734
-    },
-    {
-      "epoch": 4.86,
-      "learning_rate": 1.6173470898387303e-05,
-      "loss": 0.5249,
-      "step": 2735
-    },
-    {
-      "epoch": 4.86,
-      "learning_rate": 1.61703895694764e-05,
-      "loss": 0.4924,
-      "step": 2736
-    },
-    {
-      "epoch": 4.86,
-      "learning_rate": 1.616730729422591e-05,
-      "loss": 0.4911,
-      "step": 2737
-    },
-    {
-      "epoch": 4.86,
-      "learning_rate": 1.6164224073108548e-05,
-      "loss": 0.5103,
-      "step": 2738
-    },
-    {
-      "epoch": 4.86,
-      "learning_rate": 1.616113990659719e-05,
-      "loss": 0.5154,
-      "step": 2739
-    },
-    {
-      "epoch": 4.86,
-      "learning_rate": 1.615805479516484e-05,
-      "loss": 0.4718,
-      "step": 2740
-    },
-    {
-      "epoch": 4.87,
-      "learning_rate": 1.6154968739284664e-05,
-      "loss": 0.4859,
-      "step": 2741
-    },
-    {
-      "epoch": 4.87,
-      "learning_rate": 1.6151881739429956e-05,
-      "loss": 0.4857,
-      "step": 2742
-    },
-    {
-      "epoch": 4.87,
-      "learning_rate": 1.6148793796074166e-05,
-      "loss": 0.514,
-      "step": 2743
-    },
-    {
-      "epoch": 4.87,
-      "learning_rate": 1.6145704909690883e-05,
-      "loss": 0.4872,
-      "step": 2744
-    },
-    {
-      "epoch": 4.87,
-      "learning_rate": 1.6142615080753846e-05,
-      "loss": 0.5041,
-      "step": 2745
-    },
-    {
-      "epoch": 4.88,
-      "learning_rate": 1.6139524309736936e-05,
-      "loss": 0.5023,
-      "step": 2746
-    },
-    {
-      "epoch": 4.88,
-      "learning_rate": 1.6136432597114176e-05,
-      "loss": 0.4877,
-      "step": 2747
-    },
-    {
-      "epoch": 4.88,
-      "learning_rate": 1.6133339943359733e-05,
-      "loss": 0.4856,
-      "step": 2748
-    },
-    {
-      "epoch": 4.88,
-      "learning_rate": 1.6130246348947927e-05,
-      "loss": 0.4911,
-      "step": 2749
-    },
-    {
-      "epoch": 4.88,
-      "learning_rate": 1.612715181435321e-05,
-      "loss": 0.4772,
-      "step": 2750
-    },
-    {
-      "epoch": 4.88,
-      "learning_rate": 1.6124056340050186e-05,
-      "loss": 0.5181,
-      "step": 2751
-    },
-    {
-      "epoch": 4.89,
-      "learning_rate": 1.6120959926513604e-05,
-      "loss": 0.4834,
-      "step": 2752
-    },
-    {
-      "epoch": 4.89,
-      "learning_rate": 1.6117862574218356e-05,
-      "loss": 0.4907,
-      "step": 2753
-    },
-    {
-      "epoch": 4.89,
-      "learning_rate": 1.6114764283639467e-05,
-      "loss": 0.4969,
-      "step": 2754
-    },
-    {
-      "epoch": 4.89,
-      "learning_rate": 1.6111665055252123e-05,
-      "loss": 0.47,
-      "step": 2755
-    },
-    {
-      "epoch": 4.89,
-      "learning_rate": 1.6108564889531648e-05,
-      "loss": 0.499,
-      "step": 2756
-    },
-    {
-      "epoch": 4.89,
-      "learning_rate": 1.6105463786953502e-05,
-      "loss": 0.4881,
-      "step": 2757
-    },
-    {
-      "epoch": 4.9,
-      "learning_rate": 1.6102361747993295e-05,
-      "loss": 0.4804,
-      "step": 2758
-    },
-    {
-      "epoch": 4.9,
-      "learning_rate": 1.6099258773126783e-05,
-      "loss": 0.4924,
-      "step": 2759
-    },
-    {
-      "epoch": 4.9,
-      "learning_rate": 1.609615486282986e-05,
-      "loss": 0.4751,
-      "step": 2760
-    },
-    {
-      "epoch": 4.9,
-      "learning_rate": 1.6093050017578566e-05,
-      "loss": 0.5138,
-      "step": 2761
-    },
-    {
-      "epoch": 4.9,
-      "learning_rate": 1.608994423784909e-05,
-      "loss": 0.5086,
-      "step": 2762
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 1.6086837524117755e-05,
-      "loss": 0.5118,
-      "step": 2763
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 1.608372987686103e-05,
-      "loss": 0.4933,
-      "step": 2764
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 1.608062129655553e-05,
-      "loss": 0.4928,
-      "step": 2765
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 1.6077511783678007e-05,
-      "loss": 0.4938,
-      "step": 2766
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 1.607440133870537e-05,
-      "loss": 0.5002,
-      "step": 2767
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 1.6071289962114655e-05,
-      "loss": 0.4697,
-      "step": 2768
-    },
-    {
-      "epoch": 4.92,
-      "learning_rate": 1.6068177654383042e-05,
-      "loss": 0.505,
-      "step": 2769
-    },
-    {
-      "epoch": 4.92,
-      "learning_rate": 1.606506441598787e-05,
-      "loss": 0.4794,
-      "step": 2770
-    },
-    {
-      "epoch": 4.92,
-      "learning_rate": 1.6061950247406604e-05,
-      "loss": 0.5095,
-      "step": 2771
-    },
-    {
-      "epoch": 4.92,
-      "learning_rate": 1.605883514911686e-05,
-      "loss": 0.5008,
-      "step": 2772
-    },
-    {
-      "epoch": 4.92,
-      "learning_rate": 1.6055719121596385e-05,
-      "loss": 0.504,
-      "step": 2773
-    },
-    {
-      "epoch": 4.92,
-      "learning_rate": 1.605260216532309e-05,
-      "loss": 0.5019,
-      "step": 2774
-    },
-    {
-      "epoch": 4.93,
-      "learning_rate": 1.6049484280775012e-05,
-      "loss": 0.5055,
-      "step": 2775
-    },
-    {
-      "epoch": 4.93,
-      "learning_rate": 1.6046365468430328e-05,
-      "loss": 0.5128,
-      "step": 2776
-    },
-    {
-      "epoch": 4.93,
-      "learning_rate": 1.6043245728767374e-05,
-      "loss": 0.4639,
-      "step": 2777
-    },
-    {
-      "epoch": 4.93,
-      "learning_rate": 1.604012506226461e-05,
-      "loss": 0.4948,
-      "step": 2778
-    },
-    {
-      "epoch": 4.93,
-      "learning_rate": 1.603700346940065e-05,
-      "loss": 0.4849,
-      "step": 2779
-    },
-    {
-      "epoch": 4.94,
-      "learning_rate": 1.6033880950654243e-05,
-      "loss": 0.4784,
-      "step": 2780
-    },
-    {
-      "epoch": 4.94,
-      "learning_rate": 1.6030757506504284e-05,
-      "loss": 0.4889,
-      "step": 2781
-    },
-    {
-      "epoch": 4.94,
-      "learning_rate": 1.602763313742981e-05,
-      "loss": 0.4759,
-      "step": 2782
-    },
-    {
-      "epoch": 4.94,
-      "learning_rate": 1.6024507843910002e-05,
-      "loss": 0.4866,
-      "step": 2783
-    },
-    {
-      "epoch": 4.94,
-      "learning_rate": 1.6021381626424175e-05,
-      "loss": 0.4987,
-      "step": 2784
-    },
-    {
-      "epoch": 4.94,
-      "learning_rate": 1.601825448545179e-05,
-      "loss": 0.4974,
-      "step": 2785
-    },
-    {
-      "epoch": 4.95,
-      "learning_rate": 1.6015126421472454e-05,
-      "loss": 0.5026,
-      "step": 2786
-    },
-    {
-      "epoch": 4.95,
-      "learning_rate": 1.6011997434965906e-05,
-      "loss": 0.497,
-      "step": 2787
-    },
-    {
-      "epoch": 4.95,
-      "learning_rate": 1.600886752641204e-05,
-      "loss": 0.5278,
-      "step": 2788
-    },
-    {
-      "epoch": 4.95,
-      "learning_rate": 1.600573669629087e-05,
-      "loss": 0.4812,
-      "step": 2789
-    },
-    {
-      "epoch": 4.95,
-      "learning_rate": 1.600260494508258e-05,
-      "loss": 0.5043,
-      "step": 2790
-    },
-    {
-      "epoch": 4.96,
-      "learning_rate": 1.5999472273267477e-05,
-      "loss": 0.4918,
-      "step": 2791
-    },
-    {
-      "epoch": 4.96,
-      "learning_rate": 1.5996338681326e-05,
-      "loss": 0.4845,
-      "step": 2792
-    },
-    {
-      "epoch": 4.96,
-      "learning_rate": 1.5993204169738757e-05,
-      "loss": 0.4662,
-      "step": 2793
-    },
-    {
-      "epoch": 4.96,
-      "learning_rate": 1.5990068738986472e-05,
-      "loss": 0.503,
-      "step": 2794
-    },
-    {
-      "epoch": 4.96,
-      "learning_rate": 1.598693238955002e-05,
-      "loss": 0.4877,
-      "step": 2795
-    },
-    {
-      "epoch": 4.96,
-      "learning_rate": 1.598379512191042e-05,
-      "loss": 0.4857,
-      "step": 2796
-    },
-    {
-      "epoch": 4.97,
-      "learning_rate": 1.5980656936548826e-05,
-      "loss": 0.5096,
-      "step": 2797
-    },
-    {
-      "epoch": 4.97,
-      "learning_rate": 1.5977517833946536e-05,
-      "loss": 0.5035,
-      "step": 2798
-    },
-    {
-      "epoch": 4.97,
-      "learning_rate": 1.5974377814584986e-05,
-      "loss": 0.4785,
-      "step": 2799
-    },
-    {
-      "epoch": 4.97,
-      "learning_rate": 1.5971236878945756e-05,
-      "loss": 0.5007,
-      "step": 2800
-    },
-    {
-      "epoch": 4.97,
-      "learning_rate": 1.5968095027510563e-05,
-      "loss": 0.5176,
-      "step": 2801
-    },
-    {
-      "epoch": 4.97,
-      "learning_rate": 1.5964952260761267e-05,
-      "loss": 0.461,
-      "step": 2802
-    },
-    {
-      "epoch": 4.98,
-      "learning_rate": 1.5961808579179864e-05,
-      "loss": 0.4809,
-      "step": 2803
-    },
-    {
-      "epoch": 4.98,
-      "learning_rate": 1.59586639832485e-05,
-      "loss": 0.5016,
-      "step": 2804
-    },
-    {
-      "epoch": 4.98,
-      "learning_rate": 1.5955518473449448e-05,
-      "loss": 0.492,
-      "step": 2805
-    },
-    {
-      "epoch": 4.98,
-      "learning_rate": 1.5952372050265137e-05,
-      "loss": 0.4929,
-      "step": 2806
-    },
-    {
-      "epoch": 4.98,
-      "learning_rate": 1.594922471417812e-05,
-      "loss": 0.5,
-      "step": 2807
-    },
-    {
-      "epoch": 4.99,
-      "learning_rate": 1.5946076465671102e-05,
-      "loss": 0.4964,
-      "step": 2808
-    },
-    {
-      "epoch": 4.99,
-      "learning_rate": 1.5942927305226916e-05,
-      "loss": 0.4938,
-      "step": 2809
-    },
-    {
-      "epoch": 4.99,
-      "learning_rate": 1.593977723332855e-05,
-      "loss": 0.5072,
-      "step": 2810
-    },
-    {
-      "epoch": 4.99,
-      "learning_rate": 1.593662625045912e-05,
-      "loss": 0.4919,
-      "step": 2811
-    },
-    {
-      "epoch": 4.99,
-      "learning_rate": 1.5933474357101885e-05,
-      "loss": 0.4729,
-      "step": 2812
-    },
-    {
-      "epoch": 4.99,
-      "learning_rate": 1.593032155374025e-05,
-      "loss": 0.4949,
-      "step": 2813
-    },
-    {
-      "epoch": 5.0,
-      "learning_rate": 1.5927167840857744e-05,
-      "loss": 0.4864,
-      "step": 2814
-    },
-    {
-      "epoch": 5.0,
-      "learning_rate": 1.592401321893805e-05,
-      "loss": 0.4994,
-      "step": 2815
-    },
-    {
-      "epoch": 5.0,
-      "learning_rate": 1.5920857688464992e-05,
-      "loss": 0.498,
-      "step": 2816
-    },
-    {
-      "epoch": 5.0,
-      "eval_loss": 0.5260665416717529,
-      "eval_runtime": 3445.8104,
-      "eval_samples_per_second": 4.163,
-      "eval_steps_per_second": 1.041,
-      "step": 2816
-    },
-    {
-      "epoch": 5.0,
-      "learning_rate": 1.591770124992252e-05,
-      "loss": 0.5158,
-      "step": 2817
-    },
-    {
-      "epoch": 5.0,
-      "learning_rate": 1.591454390379473e-05,
-      "loss": 0.4784,
-      "step": 2818
-    },
-    {
-      "epoch": 5.0,
-      "learning_rate": 1.5911385650565855e-05,
-      "loss": 0.4766,
-      "step": 2819
-    },
-    {
-      "epoch": 5.01,
-      "learning_rate": 1.590822649072028e-05,
-      "loss": 0.5023,
-      "step": 2820
-    },
-    {
-      "epoch": 5.01,
-      "learning_rate": 1.590506642474251e-05,
-      "loss": 0.4897,
-      "step": 2821
-    },
-    {
-      "epoch": 5.01,
-      "learning_rate": 1.59019054531172e-05,
-      "loss": 0.4717,
-      "step": 2822
-    },
-    {
-      "epoch": 5.01,
-      "learning_rate": 1.589874357632914e-05,
-      "loss": 0.4961,
-      "step": 2823
-    },
-    {
-      "epoch": 5.01,
-      "learning_rate": 1.5895580794863262e-05,
-      "loss": 0.4871,
-      "step": 2824
-    },
-    {
-      "epoch": 5.02,
-      "learning_rate": 1.5892417109204638e-05,
-      "loss": 0.4921,
-      "step": 2825
-    },
-    {
-      "epoch": 5.02,
-      "learning_rate": 1.588925251983847e-05,
-      "loss": 0.5048,
-      "step": 2826
-    },
-    {
-      "epoch": 5.02,
-      "learning_rate": 1.588608702725011e-05,
-      "loss": 0.4824,
-      "step": 2827
-    },
-    {
-      "epoch": 5.02,
-      "learning_rate": 1.588292063192504e-05,
-      "loss": 0.4974,
-      "step": 2828
-    },
-    {
-      "epoch": 5.02,
-      "learning_rate": 1.587975333434888e-05,
-      "loss": 0.4719,
-      "step": 2829
-    },
-    {
-      "epoch": 5.02,
-      "learning_rate": 1.5876585135007398e-05,
-      "loss": 0.5094,
-      "step": 2830
-    },
-    {
-      "epoch": 5.03,
-      "learning_rate": 1.5873416034386485e-05,
-      "loss": 0.5085,
-      "step": 2831
-    },
-    {
-      "epoch": 5.03,
-      "learning_rate": 1.5870246032972192e-05,
-      "loss": 0.4979,
-      "step": 2832
-    },
-    {
-      "epoch": 5.03,
-      "learning_rate": 1.5867075131250687e-05,
-      "loss": 0.491,
-      "step": 2833
-    },
-    {
-      "epoch": 5.03,
-      "learning_rate": 1.5863903329708287e-05,
-      "loss": 0.5128,
-      "step": 2834
-    },
-    {
-      "epoch": 5.03,
-      "learning_rate": 1.586073062883144e-05,
-      "loss": 0.4768,
-      "step": 2835
-    },
-    {
-      "epoch": 5.04,
-      "learning_rate": 1.5857557029106743e-05,
-      "loss": 0.4977,
-      "step": 2836
-    },
-    {
-      "epoch": 5.04,
-      "learning_rate": 1.5854382531020922e-05,
-      "loss": 0.5253,
-      "step": 2837
-    },
-    {
-      "epoch": 5.04,
-      "learning_rate": 1.585120713506084e-05,
-      "loss": 0.46,
-      "step": 2838
-    },
-    {
-      "epoch": 5.04,
-      "learning_rate": 1.5848030841713506e-05,
-      "loss": 0.4803,
-      "step": 2839
-    },
-    {
-      "epoch": 5.04,
-      "learning_rate": 1.5844853651466056e-05,
-      "loss": 0.4875,
-      "step": 2840
-    },
-    {
-      "epoch": 5.04,
-      "learning_rate": 1.5841675564805773e-05,
-      "loss": 0.4876,
-      "step": 2841
-    },
-    {
-      "epoch": 5.05,
-      "learning_rate": 1.5838496582220073e-05,
-      "loss": 0.4863,
-      "step": 2842
-    },
-    {
-      "epoch": 5.05,
-      "learning_rate": 1.583531670419651e-05,
-      "loss": 0.497,
-      "step": 2843
-    },
-    {
-      "epoch": 5.05,
-      "learning_rate": 1.583213593122277e-05,
-      "loss": 0.4954,
-      "step": 2844
-    },
-    {
-      "epoch": 5.05,
-      "learning_rate": 1.5828954263786688e-05,
-      "loss": 0.52,
-      "step": 2845
-    },
-    {
-      "epoch": 5.05,
-      "learning_rate": 1.5825771702376224e-05,
-      "loss": 0.497,
-      "step": 2846
-    },
-    {
-      "epoch": 5.05,
-      "learning_rate": 1.582258824747949e-05,
-      "loss": 0.471,
-      "step": 2847
-    },
-    {
-      "epoch": 5.06,
-      "learning_rate": 1.5819403899584712e-05,
-      "loss": 0.4977,
-      "step": 2848
-    },
-    {
-      "epoch": 5.06,
-      "learning_rate": 1.581621865918028e-05,
-      "loss": 0.519,
-      "step": 2849
-    },
-    {
-      "epoch": 5.06,
-      "learning_rate": 1.5813032526754696e-05,
-      "loss": 0.4988,
-      "step": 2850
-    },
-    {
-      "epoch": 5.06,
-      "learning_rate": 1.5809845502796625e-05,
-      "loss": 0.4725,
-      "step": 2851
-    },
-    {
-      "epoch": 5.06,
-      "learning_rate": 1.5806657587794836e-05,
-      "loss": 0.4905,
-      "step": 2852
-    },
-    {
-      "epoch": 5.07,
-      "learning_rate": 1.580346878223827e-05,
-      "loss": 0.466,
-      "step": 2853
-    },
-    {
-      "epoch": 5.07,
-      "learning_rate": 1.580027908661597e-05,
-      "loss": 0.5019,
-      "step": 2854
-    },
-    {
-      "epoch": 5.07,
-      "learning_rate": 1.579708850141715e-05,
-      "loss": 0.4553,
-      "step": 2855
-    },
-    {
-      "epoch": 5.07,
-      "learning_rate": 1.5793897027131133e-05,
-      "loss": 0.4854,
-      "step": 2856
-    },
-    {
-      "epoch": 5.07,
-      "learning_rate": 1.579070466424739e-05,
-      "loss": 0.5006,
-      "step": 2857
-    },
-    {
-      "epoch": 5.07,
-      "learning_rate": 1.578751141325553e-05,
-      "loss": 0.5051,
-      "step": 2858
-    },
-    {
-      "epoch": 5.08,
-      "learning_rate": 1.5784317274645294e-05,
-      "loss": 0.4939,
-      "step": 2859
-    },
-    {
-      "epoch": 5.08,
-      "learning_rate": 1.5781122248906555e-05,
-      "loss": 0.4802,
-      "step": 2860
-    },
-    {
-      "epoch": 5.08,
-      "learning_rate": 1.5777926336529335e-05,
-      "loss": 0.4833,
-      "step": 2861
-    },
-    {
-      "epoch": 5.08,
-      "learning_rate": 1.577472953800378e-05,
-      "loss": 0.4988,
-      "step": 2862
-    },
-    {
-      "epoch": 5.08,
-      "learning_rate": 1.5771531853820183e-05,
-      "loss": 0.5049,
-      "step": 2863
-    },
-    {
-      "epoch": 5.08,
-      "learning_rate": 1.5768333284468954e-05,
-      "loss": 0.4874,
-      "step": 2864
-    },
-    {
-      "epoch": 5.09,
-      "learning_rate": 1.576513383044066e-05,
-      "loss": 0.4736,
-      "step": 2865
-    },
-    {
-      "epoch": 5.09,
-      "learning_rate": 1.576193349222599e-05,
-      "loss": 0.4981,
-      "step": 2866
-    },
-    {
-      "epoch": 5.09,
-      "learning_rate": 1.575873227031578e-05,
-      "loss": 0.498,
-      "step": 2867
-    },
-    {
-      "epoch": 5.09,
-      "learning_rate": 1.5755530165200983e-05,
-      "loss": 0.4914,
-      "step": 2868
-    },
-    {
-      "epoch": 5.09,
-      "learning_rate": 1.575232717737271e-05,
-      "loss": 0.4804,
-      "step": 2869
-    },
-    {
-      "epoch": 5.1,
-      "learning_rate": 1.574912330732219e-05,
-      "loss": 0.4698,
-      "step": 2870
-    },
-    {
-      "epoch": 5.1,
-      "learning_rate": 1.5745918555540796e-05,
-      "loss": 0.487,
-      "step": 2871
-    },
-    {
-      "epoch": 5.1,
-      "learning_rate": 1.5742712922520033e-05,
-      "loss": 0.4677,
-      "step": 2872
-    },
-    {
-      "epoch": 5.1,
-      "learning_rate": 1.5739506408751545e-05,
-      "loss": 0.4832,
-      "step": 2873
-    },
-    {
-      "epoch": 5.1,
-      "learning_rate": 1.5736299014727102e-05,
-      "loss": 0.4989,
-      "step": 2874
-    },
-    {
-      "epoch": 5.1,
-      "learning_rate": 1.5733090740938624e-05,
-      "loss": 0.5099,
-      "step": 2875
-    },
-    {
-      "epoch": 5.11,
-      "learning_rate": 1.572988158787815e-05,
-      "loss": 0.4643,
-      "step": 2876
-    },
-    {
-      "epoch": 5.11,
-      "learning_rate": 1.572667155603787e-05,
-      "loss": 0.4996,
-      "step": 2877
-    },
-    {
-      "epoch": 5.11,
-      "learning_rate": 1.5723460645910087e-05,
-      "loss": 0.4887,
-      "step": 2878
-    },
-    {
-      "epoch": 5.11,
-      "learning_rate": 1.572024885798726e-05,
-      "loss": 0.4663,
-      "step": 2879
-    },
-    {
-      "epoch": 5.11,
-      "learning_rate": 1.571703619276197e-05,
-      "loss": 0.4992,
-      "step": 2880
-    },
-    {
-      "epoch": 5.11,
-      "learning_rate": 1.571382265072694e-05,
-      "loss": 0.4529,
-      "step": 2881
-    },
-    {
-      "epoch": 5.12,
-      "learning_rate": 1.5710608232375022e-05,
-      "loss": 0.5201,
-      "step": 2882
-    },
-    {
-      "epoch": 5.12,
-      "learning_rate": 1.5707392938199208e-05,
-      "loss": 0.4876,
-      "step": 2883
-    },
-    {
-      "epoch": 5.12,
-      "learning_rate": 1.5704176768692617e-05,
-      "loss": 0.5028,
-      "step": 2884
-    },
-    {
-      "epoch": 5.12,
-      "learning_rate": 1.5700959724348508e-05,
-      "loss": 0.4885,
-      "step": 2885
-    },
-    {
-      "epoch": 5.12,
-      "learning_rate": 1.569774180566027e-05,
-      "loss": 0.4766,
-      "step": 2886
-    },
-    {
-      "epoch": 5.13,
-      "learning_rate": 1.569452301312143e-05,
-      "loss": 0.486,
-      "step": 2887
-    },
-    {
-      "epoch": 5.13,
-      "learning_rate": 1.5691303347225647e-05,
-      "loss": 0.4805,
-      "step": 2888
-    },
-    {
-      "epoch": 5.13,
-      "learning_rate": 1.5688082808466714e-05,
-      "loss": 0.5202,
-      "step": 2889
-    },
-    {
-      "epoch": 5.13,
-      "learning_rate": 1.568486139733856e-05,
-      "loss": 0.4961,
-      "step": 2890
-    },
-    {
-      "epoch": 5.13,
-      "learning_rate": 1.5681639114335247e-05,
-      "loss": 0.4789,
-      "step": 2891
-    },
-    {
-      "epoch": 5.13,
-      "learning_rate": 1.5678415959950967e-05,
-      "loss": 0.4883,
-      "step": 2892
-    },
-    {
-      "epoch": 5.14,
-      "learning_rate": 1.5675191934680048e-05,
-      "loss": 0.4699,
-      "step": 2893
-    },
-    {
-      "epoch": 5.14,
-      "learning_rate": 1.5671967039016955e-05,
-      "loss": 0.4907,
-      "step": 2894
-    },
-    {
-      "epoch": 5.14,
-      "learning_rate": 1.566874127345628e-05,
-      "loss": 0.4842,
-      "step": 2895
-    },
-    {
-      "epoch": 5.14,
-      "learning_rate": 1.566551463849275e-05,
-      "loss": 0.4803,
-      "step": 2896
-    },
-    {
-      "epoch": 5.14,
-      "learning_rate": 1.5662287134621235e-05,
-      "loss": 0.4834,
-      "step": 2897
-    },
-    {
-      "epoch": 5.15,
-      "learning_rate": 1.5659058762336726e-05,
-      "loss": 0.5137,
-      "step": 2898
-    },
-    {
-      "epoch": 5.15,
-      "learning_rate": 1.565582952213435e-05,
-      "loss": 0.4674,
-      "step": 2899
-    },
-    {
-      "epoch": 5.15,
-      "learning_rate": 1.565259941450937e-05,
-      "loss": 0.5037,
-      "step": 2900
-    },
-    {
-      "epoch": 5.15,
-      "learning_rate": 1.5649368439957182e-05,
-      "loss": 0.5121,
-      "step": 2901
-    },
-    {
-      "epoch": 5.15,
-      "learning_rate": 1.5646136598973312e-05,
-      "loss": 0.4655,
-      "step": 2902
-    },
-    {
-      "epoch": 5.15,
-      "learning_rate": 1.564290389205342e-05,
-      "loss": 0.4754,
-      "step": 2903
-    },
-    {
-      "epoch": 5.16,
-      "learning_rate": 1.5639670319693304e-05,
-      "loss": 0.4593,
-      "step": 2904
-    },
-    {
-      "epoch": 5.16,
-      "learning_rate": 1.5636435882388882e-05,
-      "loss": 0.4864,
-      "step": 2905
-    },
-    {
-      "epoch": 5.16,
-      "learning_rate": 1.563320058063622e-05,
-      "loss": 0.4822,
-      "step": 2906
-    },
-    {
-      "epoch": 5.16,
-      "learning_rate": 1.562996441493151e-05,
-      "loss": 0.5046,
-      "step": 2907
-    },
-    {
-      "epoch": 5.16,
-      "learning_rate": 1.562672738577107e-05,
-      "loss": 0.5251,
-      "step": 2908
-    },
-    {
-      "epoch": 5.16,
-      "learning_rate": 1.562348949365136e-05,
-      "loss": 0.5133,
-      "step": 2909
-    },
-    {
-      "epoch": 5.17,
-      "learning_rate": 1.562025073906897e-05,
-      "loss": 0.4991,
-      "step": 2910
-    },
-    {
-      "epoch": 5.17,
-      "learning_rate": 1.5617011122520615e-05,
-      "loss": 0.5002,
-      "step": 2911
-    },
-    {
-      "epoch": 5.17,
-      "learning_rate": 1.5613770644503156e-05,
-      "loss": 0.4797,
-      "step": 2912
-    },
-    {
-      "epoch": 5.17,
-      "learning_rate": 1.561052930551357e-05,
-      "loss": 0.4788,
-      "step": 2913
-    },
-    {
-      "epoch": 5.17,
-      "learning_rate": 1.560728710604898e-05,
-      "loss": 0.4989,
-      "step": 2914
-    },
-    {
-      "epoch": 5.18,
-      "learning_rate": 1.5604044046606638e-05,
-      "loss": 0.4653,
-      "step": 2915
-    },
-    {
-      "epoch": 5.18,
-      "learning_rate": 1.5600800127683916e-05,
-      "loss": 0.4708,
-      "step": 2916
-    },
-    {
-      "epoch": 5.18,
-      "learning_rate": 1.5597555349778334e-05,
-      "loss": 0.494,
-      "step": 2917
-    },
-    {
-      "epoch": 5.18,
-      "learning_rate": 1.5594309713387535e-05,
-      "loss": 0.4676,
-      "step": 2918
-    },
-    {
-      "epoch": 5.18,
-      "learning_rate": 1.5591063219009296e-05,
-      "loss": 0.4827,
-      "step": 2919
-    },
-    {
-      "epoch": 5.18,
-      "learning_rate": 1.558781586714152e-05,
-      "loss": 0.4983,
-      "step": 2920
-    },
-    {
-      "epoch": 5.19,
-      "learning_rate": 1.5584567658282253e-05,
-      "loss": 0.4879,
-      "step": 2921
-    },
-    {
-      "epoch": 5.19,
-      "learning_rate": 1.5581318592929665e-05,
-      "loss": 0.4845,
-      "step": 2922
-    },
-    {
-      "epoch": 5.19,
-      "learning_rate": 1.5578068671582054e-05,
-      "loss": 0.4745,
-      "step": 2923
-    },
-    {
-      "epoch": 5.19,
-      "learning_rate": 1.5574817894737858e-05,
-      "loss": 0.4986,
-      "step": 2924
-    },
-    {
-      "epoch": 5.19,
-      "learning_rate": 1.5571566262895637e-05,
-      "loss": 0.4876,
-      "step": 2925
-    },
-    {
-      "epoch": 5.19,
-      "learning_rate": 1.5568313776554097e-05,
-      "loss": 0.4892,
-      "step": 2926
-    },
-    {
-      "epoch": 5.2,
-      "learning_rate": 1.5565060436212052e-05,
-      "loss": 0.4794,
-      "step": 2927
-    },
-    {
-      "epoch": 5.2,
-      "learning_rate": 1.5561806242368466e-05,
-      "loss": 0.4563,
-      "step": 2928
-    },
-    {
-      "epoch": 5.2,
-      "learning_rate": 1.555855119552243e-05,
-      "loss": 0.4888,
-      "step": 2929
-    },
-    {
-      "epoch": 5.2,
-      "learning_rate": 1.5555295296173164e-05,
-      "loss": 0.5118,
-      "step": 2930
-    },
-    {
-      "epoch": 5.2,
-      "learning_rate": 1.5552038544820008e-05,
-      "loss": 0.4651,
-      "step": 2931
-    },
-    {
-      "epoch": 5.21,
-      "learning_rate": 1.5548780941962455e-05,
-      "loss": 0.4912,
-      "step": 2932
-    },
-    {
-      "epoch": 5.21,
-      "learning_rate": 1.5545522488100112e-05,
-      "loss": 0.4923,
-      "step": 2933
-    },
-    {
-      "epoch": 5.21,
-      "learning_rate": 1.5542263183732723e-05,
-      "loss": 0.485,
-      "step": 2934
-    },
-    {
-      "epoch": 5.21,
-      "learning_rate": 1.5539003029360157e-05,
-      "loss": 0.4725,
-      "step": 2935
-    },
-    {
-      "epoch": 5.21,
-      "learning_rate": 1.553574202548242e-05,
-      "loss": 0.4771,
-      "step": 2936
-    },
-    {
-      "epoch": 5.21,
-      "learning_rate": 1.5532480172599646e-05,
-      "loss": 0.4948,
-      "step": 2937
-    },
-    {
-      "epoch": 5.22,
-      "learning_rate": 1.5529217471212096e-05,
-      "loss": 0.4884,
-      "step": 2938
-    },
-    {
-      "epoch": 5.22,
-      "learning_rate": 1.5525953921820164e-05,
-      "loss": 0.4842,
-      "step": 2939
-    },
-    {
-      "epoch": 5.22,
-      "learning_rate": 1.5522689524924376e-05,
-      "loss": 0.5167,
-      "step": 2940
-    },
-    {
-      "epoch": 5.22,
-      "learning_rate": 1.551942428102538e-05,
-      "loss": 0.4812,
-      "step": 2941
-    },
-    {
-      "epoch": 5.22,
-      "learning_rate": 1.5516158190623966e-05,
-      "loss": 0.4924,
-      "step": 2942
-    },
-    {
-      "epoch": 5.23,
-      "learning_rate": 1.5512891254221046e-05,
-      "loss": 0.4679,
-      "step": 2943
-    },
-    {
-      "epoch": 5.23,
-      "learning_rate": 1.5509623472317658e-05,
-      "loss": 0.4856,
-      "step": 2944
-    },
-    {
-      "epoch": 5.23,
-      "learning_rate": 1.550635484541498e-05,
-      "loss": 0.4919,
-      "step": 2945
-    },
-    {
-      "epoch": 5.23,
-      "learning_rate": 1.5503085374014316e-05,
-      "loss": 0.4828,
-      "step": 2946
-    },
-    {
-      "epoch": 5.23,
-      "learning_rate": 1.5499815058617094e-05,
-      "loss": 0.4966,
-      "step": 2947
-    },
-    {
-      "epoch": 5.23,
-      "learning_rate": 1.5496543899724872e-05,
-      "loss": 0.4908,
-      "step": 2948
-    },
-    {
-      "epoch": 5.24,
-      "learning_rate": 1.549327189783935e-05,
-      "loss": 0.4933,
-      "step": 2949
-    },
-    {
-      "epoch": 5.24,
-      "learning_rate": 1.548999905346234e-05,
-      "loss": 0.4998,
-      "step": 2950
-    },
-    {
-      "epoch": 5.24,
-      "learning_rate": 1.5486725367095796e-05,
-      "loss": 0.4859,
-      "step": 2951
-    },
-    {
-      "epoch": 5.24,
-      "learning_rate": 1.5483450839241794e-05,
-      "loss": 0.481,
-      "step": 2952
-    },
-    {
-      "epoch": 5.24,
-      "learning_rate": 1.5480175470402542e-05,
-      "loss": 0.4933,
-      "step": 2953
-    },
-    {
-      "epoch": 5.24,
-      "learning_rate": 1.5476899261080377e-05,
-      "loss": 0.4964,
-      "step": 2954
-    },
-    {
-      "epoch": 5.25,
-      "learning_rate": 1.5473622211777763e-05,
-      "loss": 0.4819,
-      "step": 2955
-    },
-    {
-      "epoch": 5.25,
-      "learning_rate": 1.5470344322997295e-05,
-      "loss": 0.4764,
-      "step": 2956
-    },
-    {
-      "epoch": 5.25,
-      "learning_rate": 1.5467065595241698e-05,
-      "loss": 0.5076,
-      "step": 2957
-    },
-    {
-      "epoch": 5.25,
-      "learning_rate": 1.5463786029013817e-05,
-      "loss": 0.4882,
-      "step": 2958
-    },
-    {
-      "epoch": 5.25,
-      "learning_rate": 1.5460505624816638e-05,
-      "loss": 0.4736,
-      "step": 2959
-    },
-    {
-      "epoch": 5.26,
-      "learning_rate": 1.5457224383153267e-05,
-      "loss": 0.5056,
-      "step": 2960
-    },
-    {
-      "epoch": 5.26,
-      "learning_rate": 1.5453942304526945e-05,
-      "loss": 0.4904,
-      "step": 2961
-    },
-    {
-      "epoch": 5.26,
-      "learning_rate": 1.545065938944103e-05,
-      "loss": 0.4771,
-      "step": 2962
-    },
-    {
-      "epoch": 5.26,
-      "learning_rate": 1.5447375638399024e-05,
-      "loss": 0.5113,
-      "step": 2963
-    },
-    {
-      "epoch": 5.26,
-      "learning_rate": 1.5444091051904545e-05,
-      "loss": 0.4954,
-      "step": 2964
-    },
-    {
-      "epoch": 5.26,
-      "learning_rate": 1.5440805630461344e-05,
-      "loss": 0.4891,
-      "step": 2965
-    },
-    {
-      "epoch": 5.27,
-      "learning_rate": 1.5437519374573295e-05,
-      "loss": 0.4956,
-      "step": 2966
-    },
-    {
-      "epoch": 5.27,
-      "learning_rate": 1.543423228474441e-05,
-      "loss": 0.4908,
-      "step": 2967
-    },
-    {
-      "epoch": 5.27,
-      "learning_rate": 1.543094436147882e-05,
-      "loss": 0.4984,
-      "step": 2968
-    },
-    {
-      "epoch": 5.27,
-      "learning_rate": 1.5427655605280788e-05,
-      "loss": 0.4648,
-      "step": 2969
-    },
-    {
-      "epoch": 5.27,
-      "learning_rate": 1.5424366016654702e-05,
-      "loss": 0.4685,
-      "step": 2970
-    },
-    {
-      "epoch": 5.27,
-      "learning_rate": 1.542107559610508e-05,
-      "loss": 0.4834,
-      "step": 2971
-    },
-    {
-      "epoch": 5.28,
-      "learning_rate": 1.5417784344136567e-05,
-      "loss": 0.4828,
-      "step": 2972
-    },
-    {
-      "epoch": 5.28,
-      "learning_rate": 1.5414492261253937e-05,
-      "loss": 0.4933,
-      "step": 2973
-    },
-    {
-      "epoch": 5.28,
-      "learning_rate": 1.5411199347962086e-05,
-      "loss": 0.5165,
-      "step": 2974
-    },
-    {
-      "epoch": 5.28,
-      "learning_rate": 1.5407905604766043e-05,
-      "loss": 0.4884,
-      "step": 2975
-    },
-    {
-      "epoch": 5.28,
-      "learning_rate": 1.5404611032170967e-05,
-      "loss": 0.4818,
-      "step": 2976
-    },
-    {
-      "epoch": 5.29,
-      "learning_rate": 1.540131563068213e-05,
-      "loss": 0.4961,
-      "step": 2977
-    },
-    {
-      "epoch": 5.29,
-      "learning_rate": 1.539801940080495e-05,
-      "loss": 0.4942,
-      "step": 2978
-    },
-    {
-      "epoch": 5.29,
-      "learning_rate": 1.539472234304496e-05,
-      "loss": 0.5171,
-      "step": 2979
-    },
-    {
-      "epoch": 5.29,
-      "learning_rate": 1.539142445790782e-05,
-      "loss": 0.4858,
-      "step": 2980
-    },
-    {
-      "epoch": 5.29,
-      "learning_rate": 1.538812574589932e-05,
-      "loss": 0.4984,
-      "step": 2981
-    },
-    {
-      "epoch": 5.29,
-      "learning_rate": 1.5384826207525376e-05,
-      "loss": 0.4823,
-      "step": 2982
-    },
-    {
-      "epoch": 5.3,
-      "learning_rate": 1.5381525843292036e-05,
-      "loss": 0.4838,
-      "step": 2983
-    },
-    {
-      "epoch": 5.3,
-      "learning_rate": 1.5378224653705467e-05,
-      "loss": 0.4582,
-      "step": 2984
-    },
-    {
-      "epoch": 5.3,
-      "learning_rate": 1.537492263927196e-05,
-      "loss": 0.4921,
-      "step": 2985
-    },
-    {
-      "epoch": 5.3,
-      "learning_rate": 1.537161980049795e-05,
-      "loss": 0.4859,
-      "step": 2986
-    },
-    {
-      "epoch": 5.3,
-      "learning_rate": 1.536831613788998e-05,
-      "loss": 0.498,
-      "step": 2987
-    },
-    {
-      "epoch": 5.3,
-      "learning_rate": 1.5365011651954723e-05,
-      "loss": 0.4821,
-      "step": 2988
-    },
-    {
-      "epoch": 5.31,
-      "learning_rate": 1.5361706343198982e-05,
-      "loss": 0.4837,
-      "step": 2989
-    },
-    {
-      "epoch": 5.31,
-      "learning_rate": 1.535840021212969e-05,
-      "loss": 0.4833,
-      "step": 2990
-    },
-    {
-      "epoch": 5.31,
-      "learning_rate": 1.5355093259253897e-05,
-      "loss": 0.4675,
-      "step": 2991
-    },
-    {
-      "epoch": 5.31,
-      "learning_rate": 1.5351785485078784e-05,
-      "loss": 0.4766,
-      "step": 2992
-    },
-    {
-      "epoch": 5.31,
-      "learning_rate": 1.5348476890111655e-05,
-      "loss": 0.5157,
-      "step": 2993
-    },
-    {
-      "epoch": 5.32,
-      "learning_rate": 1.534516747485995e-05,
-      "loss": 0.4775,
-      "step": 2994
-    },
-    {
-      "epoch": 5.32,
-      "learning_rate": 1.5341857239831218e-05,
-      "loss": 0.4903,
-      "step": 2995
-    },
-    {
-      "epoch": 5.32,
-      "learning_rate": 1.5338546185533146e-05,
-      "loss": 0.4963,
-      "step": 2996
-    },
-    {
-      "epoch": 5.32,
-      "learning_rate": 1.533523431247355e-05,
-      "loss": 0.4904,
-      "step": 2997
-    },
-    {
-      "epoch": 5.32,
-      "learning_rate": 1.5331921621160355e-05,
-      "loss": 0.4952,
-      "step": 2998
-    },
-    {
-      "epoch": 5.32,
-      "learning_rate": 1.5328608112101625e-05,
-      "loss": 0.5038,
-      "step": 2999
-    },
-    {
-      "epoch": 5.33,
-      "learning_rate": 1.5325293785805548e-05,
-      "loss": 0.4871,
-      "step": 3000
-    },
-    {
-      "epoch": 5.33,
-      "learning_rate": 1.5321978642780432e-05,
-      "loss": 0.4854,
-      "step": 3001
-    },
-    {
-      "epoch": 5.33,
-      "learning_rate": 1.5318662683534716e-05,
-      "loss": 0.4888,
-      "step": 3002
-    },
-    {
-      "epoch": 5.33,
-      "learning_rate": 1.531534590857696e-05,
-      "loss": 0.5017,
-      "step": 3003
-    },
-    {
-      "epoch": 5.33,
-      "learning_rate": 1.531202831841585e-05,
-      "loss": 0.4943,
-      "step": 3004
-    },
-    {
-      "epoch": 5.34,
-      "learning_rate": 1.53087099135602e-05,
-      "loss": 0.4758,
-      "step": 3005
-    },
-    {
-      "epoch": 5.34,
-      "learning_rate": 1.5305390694518953e-05,
-      "loss": 0.4904,
-      "step": 3006
-    },
-    {
-      "epoch": 5.34,
-      "learning_rate": 1.5302070661801155e-05,
-      "loss": 0.4864,
-      "step": 3007
-    },
-    {
-      "epoch": 5.34,
-      "learning_rate": 1.5298749815916007e-05,
-      "loss": 0.4899,
-      "step": 3008
-    },
-    {
-      "epoch": 5.34,
-      "learning_rate": 1.529542815737281e-05,
-      "loss": 0.4699,
-      "step": 3009
-    },
-    {
-      "epoch": 5.34,
-      "learning_rate": 1.5292105686681004e-05,
-      "loss": 0.5059,
-      "step": 3010
-    },
-    {
-      "epoch": 5.35,
-      "learning_rate": 1.528878240435015e-05,
-      "loss": 0.4991,
-      "step": 3011
-    },
-    {
-      "epoch": 5.35,
-      "learning_rate": 1.5285458310889928e-05,
-      "loss": 0.5048,
-      "step": 3012
-    },
-    {
-      "epoch": 5.35,
-      "learning_rate": 1.5282133406810152e-05,
-      "loss": 0.4866,
-      "step": 3013
-    },
-    {
-      "epoch": 5.35,
-      "learning_rate": 1.527880769262076e-05,
-      "loss": 0.4757,
-      "step": 3014
-    },
-    {
-      "epoch": 5.35,
-      "learning_rate": 1.52754811688318e-05,
-      "loss": 0.4822,
-      "step": 3015
-    },
-    {
-      "epoch": 5.35,
-      "learning_rate": 1.527215383595346e-05,
-      "loss": 0.4883,
-      "step": 3016
-    },
-    {
-      "epoch": 5.36,
-      "learning_rate": 1.5268825694496046e-05,
-      "loss": 0.4926,
-      "step": 3017
-    },
-    {
-      "epoch": 5.36,
-      "learning_rate": 1.526549674496998e-05,
-      "loss": 0.4607,
-      "step": 3018
-    },
-    {
-      "epoch": 5.36,
-      "learning_rate": 1.5262166987885827e-05,
-      "loss": 0.4702,
-      "step": 3019
-    },
-    {
-      "epoch": 5.36,
-      "learning_rate": 1.5258836423754258e-05,
-      "loss": 0.49,
-      "step": 3020
-    },
-    {
-      "epoch": 5.36,
-      "learning_rate": 1.525550505308608e-05,
-      "loss": 0.4735,
-      "step": 3021
-    },
-    {
-      "epoch": 5.37,
-      "learning_rate": 1.525217287639221e-05,
-      "loss": 0.466,
-      "step": 3022
-    },
-    {
-      "epoch": 5.37,
-      "learning_rate": 1.5248839894183704e-05,
-      "loss": 0.4926,
-      "step": 3023
-    },
-    {
-      "epoch": 5.37,
-      "learning_rate": 1.5245506106971733e-05,
-      "loss": 0.4922,
-      "step": 3024
-    },
-    {
-      "epoch": 5.37,
-      "learning_rate": 1.5242171515267593e-05,
-      "loss": 0.4977,
-      "step": 3025
-    },
-    {
-      "epoch": 5.37,
-      "learning_rate": 1.5238836119582698e-05,
-      "loss": 0.4803,
-      "step": 3026
-    },
-    {
-      "epoch": 5.37,
-      "learning_rate": 1.52354999204286e-05,
-      "loss": 0.481,
-      "step": 3027
-    },
-    {
-      "epoch": 5.38,
-      "learning_rate": 1.5232162918316955e-05,
-      "loss": 0.5019,
-      "step": 3028
-    },
-    {
-      "epoch": 5.38,
-      "learning_rate": 1.522882511375956e-05,
-      "loss": 0.4754,
-      "step": 3029
-    },
-    {
-      "epoch": 5.38,
-      "learning_rate": 1.5225486507268322e-05,
-      "loss": 0.4997,
-      "step": 3030
-    },
-    {
-      "epoch": 5.38,
-      "learning_rate": 1.5222147099355277e-05,
-      "loss": 0.489,
-      "step": 3031
-    },
-    {
-      "epoch": 5.38,
-      "learning_rate": 1.5218806890532584e-05,
-      "loss": 0.4726,
-      "step": 3032
-    },
-    {
-      "epoch": 5.38,
-      "learning_rate": 1.5215465881312526e-05,
-      "loss": 0.4847,
-      "step": 3033
-    },
-    {
-      "epoch": 5.39,
-      "learning_rate": 1.5212124072207499e-05,
-      "loss": 0.4834,
-      "step": 3034
-    },
-    {
-      "epoch": 5.39,
-      "learning_rate": 1.5208781463730038e-05,
-      "loss": 0.4825,
-      "step": 3035
-    },
-    {
-      "epoch": 5.39,
-      "learning_rate": 1.5205438056392788e-05,
-      "loss": 0.4949,
-      "step": 3036
-    },
-    {
-      "epoch": 5.39,
-      "learning_rate": 1.5202093850708517e-05,
-      "loss": 0.4873,
-      "step": 3037
-    },
-    {
-      "epoch": 5.39,
-      "learning_rate": 1.5198748847190121e-05,
-      "loss": 0.4581,
-      "step": 3038
-    },
-    {
-      "epoch": 5.4,
-      "learning_rate": 1.519540304635062e-05,
-      "loss": 0.4806,
-      "step": 3039
-    },
-    {
-      "epoch": 5.4,
-      "learning_rate": 1.519205644870315e-05,
-      "loss": 0.5415,
-      "step": 3040
-    },
-    {
-      "epoch": 5.4,
-      "learning_rate": 1.518870905476097e-05,
-      "loss": 0.4715,
-      "step": 3041
-    },
-    {
-      "epoch": 5.4,
-      "learning_rate": 1.5185360865037462e-05,
-      "loss": 0.4747,
-      "step": 3042
-    },
-    {
-      "epoch": 5.4,
-      "learning_rate": 1.5182011880046132e-05,
-      "loss": 0.4894,
-      "step": 3043
-    },
-    {
-      "epoch": 5.4,
-      "learning_rate": 1.5178662100300612e-05,
-      "loss": 0.4925,
-      "step": 3044
-    },
-    {
-      "epoch": 5.41,
-      "learning_rate": 1.5175311526314642e-05,
-      "loss": 0.4638,
-      "step": 3045
-    },
-    {
-      "epoch": 5.41,
-      "learning_rate": 1.5171960158602097e-05,
-      "loss": 0.4897,
-      "step": 3046
-    },
-    {
-      "epoch": 5.41,
-      "learning_rate": 1.5168607997676969e-05,
-      "loss": 0.4904,
-      "step": 3047
-    },
-    {
-      "epoch": 5.41,
-      "learning_rate": 1.5165255044053373e-05,
-      "loss": 0.4879,
-      "step": 3048
-    },
-    {
-      "epoch": 5.41,
-      "learning_rate": 1.5161901298245538e-05,
-      "loss": 0.4966,
-      "step": 3049
-    },
-    {
-      "epoch": 5.42,
-      "learning_rate": 1.5158546760767827e-05,
-      "loss": 0.4859,
-      "step": 3050
-    },
-    {
-      "epoch": 5.42,
-      "learning_rate": 1.5155191432134721e-05,
-      "loss": 0.4781,
-      "step": 3051
-    },
-    {
-      "epoch": 5.42,
-      "learning_rate": 1.5151835312860811e-05,
-      "loss": 0.5022,
-      "step": 3052
-    },
-    {
-      "epoch": 5.42,
-      "learning_rate": 1.5148478403460824e-05,
-      "loss": 0.4839,
-      "step": 3053
-    },
-    {
-      "epoch": 5.42,
-      "learning_rate": 1.51451207044496e-05,
-      "loss": 0.474,
-      "step": 3054
-    },
-    {
-      "epoch": 5.42,
-      "learning_rate": 1.5141762216342107e-05,
-      "loss": 0.4771,
-      "step": 3055
-    },
-    {
-      "epoch": 5.43,
-      "learning_rate": 1.5138402939653421e-05,
-      "loss": 0.481,
-      "step": 3056
-    },
-    {
-      "epoch": 5.43,
-      "learning_rate": 1.513504287489875e-05,
-      "loss": 0.4899,
-      "step": 3057
-    },
-    {
-      "epoch": 5.43,
-      "learning_rate": 1.5131682022593422e-05,
-      "loss": 0.4684,
-      "step": 3058
-    },
-    {
-      "epoch": 5.43,
-      "learning_rate": 1.5128320383252886e-05,
-      "loss": 0.5143,
-      "step": 3059
-    },
-    {
-      "epoch": 5.43,
-      "learning_rate": 1.5124957957392703e-05,
-      "loss": 0.5034,
-      "step": 3060
-    },
-    {
-      "epoch": 5.43,
-      "learning_rate": 1.5121594745528566e-05,
-      "loss": 0.4816,
-      "step": 3061
-    },
-    {
-      "epoch": 5.44,
-      "learning_rate": 1.511823074817628e-05,
-      "loss": 0.4631,
-      "step": 3062
-    },
-    {
-      "epoch": 5.44,
-      "learning_rate": 1.5114865965851779e-05,
-      "loss": 0.4854,
-      "step": 3063
-    },
-    {
-      "epoch": 5.44,
-      "learning_rate": 1.5111500399071109e-05,
-      "loss": 0.4792,
-      "step": 3064
-    },
-    {
-      "epoch": 5.44,
-      "learning_rate": 1.510813404835044e-05,
-      "loss": 0.499,
-      "step": 3065
-    },
-    {
-      "epoch": 5.44,
-      "learning_rate": 1.5104766914206063e-05,
-      "loss": 0.4789,
-      "step": 3066
-    },
-    {
-      "epoch": 5.45,
-      "learning_rate": 1.510139899715439e-05,
-      "loss": 0.4849,
-      "step": 3067
-    },
-    {
-      "epoch": 5.45,
-      "learning_rate": 1.5098030297711948e-05,
-      "loss": 0.4845,
-      "step": 3068
-    },
-    {
-      "epoch": 5.45,
-      "learning_rate": 1.509466081639539e-05,
-      "loss": 0.5021,
-      "step": 3069
-    },
-    {
-      "epoch": 5.45,
-      "learning_rate": 1.5091290553721485e-05,
-      "loss": 0.4816,
-      "step": 3070
-    },
-    {
-      "epoch": 5.45,
-      "learning_rate": 1.5087919510207121e-05,
-      "loss": 0.5147,
-      "step": 3071
-    },
-    {
-      "epoch": 5.45,
-      "learning_rate": 1.5084547686369311e-05,
-      "loss": 0.4909,
-      "step": 3072
-    },
-    {
-      "epoch": 5.46,
-      "learning_rate": 1.5081175082725184e-05,
-      "loss": 0.4819,
-      "step": 3073
-    },
-    {
-      "epoch": 5.46,
-      "learning_rate": 1.5077801699791988e-05,
-      "loss": 0.4927,
-      "step": 3074
-    },
-    {
-      "epoch": 5.46,
-      "learning_rate": 1.507442753808709e-05,
-      "loss": 0.4863,
-      "step": 3075
-    },
-    {
-      "epoch": 5.46,
-      "learning_rate": 1.5071052598127981e-05,
-      "loss": 0.4977,
-      "step": 3076
-    },
-    {
-      "epoch": 5.46,
-      "learning_rate": 1.5067676880432269e-05,
-      "loss": 0.4589,
-      "step": 3077
-    },
-    {
-      "epoch": 5.46,
-      "learning_rate": 1.506430038551768e-05,
-      "loss": 0.4816,
-      "step": 3078
-    },
-    {
-      "epoch": 5.47,
-      "learning_rate": 1.5060923113902055e-05,
-      "loss": 0.4888,
-      "step": 3079
-    },
-    {
-      "epoch": 5.47,
-      "learning_rate": 1.5057545066103365e-05,
-      "loss": 0.4961,
-      "step": 3080
-    },
-    {
-      "epoch": 5.47,
-      "learning_rate": 1.5054166242639694e-05,
-      "loss": 0.4751,
-      "step": 3081
-    },
-    {
-      "epoch": 5.47,
-      "learning_rate": 1.505078664402924e-05,
-      "loss": 0.4566,
-      "step": 3082
-    },
-    {
-      "epoch": 5.47,
-      "learning_rate": 1.5047406270790332e-05,
-      "loss": 0.4836,
-      "step": 3083
-    },
-    {
-      "epoch": 5.48,
-      "learning_rate": 1.5044025123441403e-05,
-      "loss": 0.4891,
-      "step": 3084
-    },
-    {
-      "epoch": 5.48,
-      "learning_rate": 1.5040643202501018e-05,
-      "loss": 0.4894,
-      "step": 3085
-    },
-    {
-      "epoch": 5.48,
-      "learning_rate": 1.5037260508487852e-05,
-      "loss": 0.4926,
-      "step": 3086
-    },
-    {
-      "epoch": 5.48,
-      "learning_rate": 1.5033877041920703e-05,
-      "loss": 0.4869,
-      "step": 3087
-    },
-    {
-      "epoch": 5.48,
-      "learning_rate": 1.5030492803318486e-05,
-      "loss": 0.4983,
-      "step": 3088
-    },
-    {
-      "epoch": 5.48,
-      "learning_rate": 1.5027107793200235e-05,
-      "loss": 0.4817,
-      "step": 3089
-    },
-    {
-      "epoch": 5.49,
-      "learning_rate": 1.5023722012085098e-05,
-      "loss": 0.4777,
-      "step": 3090
-    },
-    {
-      "epoch": 5.49,
-      "learning_rate": 1.5020335460492352e-05,
-      "loss": 0.4785,
-      "step": 3091
-    },
-    {
-      "epoch": 5.49,
-      "learning_rate": 1.5016948138941379e-05,
-      "loss": 0.4778,
-      "step": 3092
-    },
-    {
-      "epoch": 5.49,
-      "learning_rate": 1.5013560047951689e-05,
-      "loss": 0.4695,
-      "step": 3093
-    },
-    {
-      "epoch": 5.49,
-      "learning_rate": 1.5010171188042906e-05,
-      "loss": 0.4892,
-      "step": 3094
-    },
-    {
-      "epoch": 5.49,
-      "learning_rate": 1.5006781559734769e-05,
-      "loss": 0.4682,
-      "step": 3095
-    },
-    {
-      "epoch": 5.5,
-      "learning_rate": 1.5003391163547142e-05,
-      "loss": 0.4718,
-      "step": 3096
-    },
-    {
-      "epoch": 5.5,
-      "learning_rate": 1.5000000000000002e-05,
-      "loss": 0.5005,
-      "step": 3097
-    },
-    {
-      "epoch": 5.5,
-      "learning_rate": 1.4996608069613443e-05,
-      "loss": 0.5192,
-      "step": 3098
-    },
-    {
-      "epoch": 5.5,
-      "learning_rate": 1.499321537290768e-05,
-      "loss": 0.4901,
-      "step": 3099
-    },
-    {
-      "epoch": 5.5,
-      "learning_rate": 1.4989821910403042e-05,
-      "loss": 0.5107,
-      "step": 3100
-    },
-    {
-      "epoch": 5.51,
-      "learning_rate": 1.498642768261998e-05,
-      "loss": 0.476,
-      "step": 3101
-    },
-    {
-      "epoch": 5.51,
-      "learning_rate": 1.4983032690079056e-05,
-      "loss": 0.4916,
-      "step": 3102
-    },
-    {
-      "epoch": 5.51,
-      "learning_rate": 1.4979636933300954e-05,
-      "loss": 0.4687,
-      "step": 3103
-    },
-    {
-      "epoch": 5.51,
-      "learning_rate": 1.4976240412806476e-05,
-      "loss": 0.4885,
-      "step": 3104
-    },
-    {
-      "epoch": 5.51,
-      "learning_rate": 1.4972843129116537e-05,
-      "loss": 0.4896,
-      "step": 3105
-    },
-    {
-      "epoch": 5.51,
-      "learning_rate": 1.4969445082752172e-05,
-      "loss": 0.4782,
-      "step": 3106
-    },
-    {
-      "epoch": 5.52,
-      "learning_rate": 1.496604627423453e-05,
-      "loss": 0.4596,
-      "step": 3107
-    },
-    {
-      "epoch": 5.52,
-      "learning_rate": 1.4962646704084884e-05,
-      "loss": 0.4866,
-      "step": 3108
-    },
-    {
-      "epoch": 5.52,
-      "learning_rate": 1.4959246372824616e-05,
-      "loss": 0.4904,
-      "step": 3109
-    },
-    {
-      "epoch": 5.52,
-      "learning_rate": 1.4955845280975225e-05,
-      "loss": 0.483,
-      "step": 3110
-    },
-    {
-      "epoch": 5.52,
-      "learning_rate": 1.4952443429058334e-05,
-      "loss": 0.4869,
-      "step": 3111
-    },
-    {
-      "epoch": 5.53,
-      "learning_rate": 1.4949040817595674e-05,
-      "loss": 0.4833,
-      "step": 3112
-    },
-    {
-      "epoch": 5.53,
-      "learning_rate": 1.4945637447109102e-05,
-      "loss": 0.4802,
-      "step": 3113
-    },
-    {
-      "epoch": 5.53,
-      "learning_rate": 1.4942233318120578e-05,
-      "loss": 0.4852,
-      "step": 3114
-    },
-    {
-      "epoch": 5.53,
-      "learning_rate": 1.4938828431152191e-05,
-      "loss": 0.4909,
-      "step": 3115
-    },
-    {
-      "epoch": 5.53,
-      "learning_rate": 1.4935422786726139e-05,
-      "loss": 0.4842,
-      "step": 3116
-    },
-    {
-      "epoch": 5.53,
-      "learning_rate": 1.493201638536474e-05,
-      "loss": 0.4923,
-      "step": 3117
-    },
-    {
-      "epoch": 5.54,
-      "learning_rate": 1.4928609227590427e-05,
-      "loss": 0.4813,
-      "step": 3118
-    },
-    {
-      "epoch": 5.54,
-      "learning_rate": 1.4925201313925745e-05,
-      "loss": 0.4959,
-      "step": 3119
-    },
-    {
-      "epoch": 5.54,
-      "learning_rate": 1.4921792644893364e-05,
-      "loss": 0.5007,
-      "step": 3120
-    },
-    {
-      "epoch": 5.54,
-      "learning_rate": 1.4918383221016062e-05,
-      "loss": 0.4827,
-      "step": 3121
-    },
-    {
-      "epoch": 5.54,
-      "learning_rate": 1.4914973042816731e-05,
-      "loss": 0.4898,
-      "step": 3122
-    },
-    {
-      "epoch": 5.54,
-      "learning_rate": 1.4911562110818392e-05,
-      "loss": 0.4924,
-      "step": 3123
-    },
-    {
-      "epoch": 5.55,
-      "learning_rate": 1.4908150425544163e-05,
-      "loss": 0.4805,
-      "step": 3124
-    },
-    {
-      "epoch": 5.55,
-      "learning_rate": 1.4904737987517293e-05,
-      "loss": 0.4921,
-      "step": 3125
-    },
-    {
-      "epoch": 5.55,
-      "learning_rate": 1.4901324797261137e-05,
-      "loss": 0.4868,
-      "step": 3126
-    },
-    {
-      "epoch": 5.55,
-      "learning_rate": 1.489791085529917e-05,
-      "loss": 0.4976,
-      "step": 3127
-    },
-    {
-      "epoch": 5.55,
-      "learning_rate": 1.4894496162154981e-05,
-      "loss": 0.4987,
-      "step": 3128
-    },
-    {
-      "epoch": 5.56,
-      "learning_rate": 1.4891080718352275e-05,
-      "loss": 0.5076,
-      "step": 3129
-    },
-    {
-      "epoch": 5.56,
-      "learning_rate": 1.4887664524414869e-05,
-      "loss": 0.4738,
-      "step": 3130
-    },
-    {
-      "epoch": 5.56,
-      "learning_rate": 1.4884247580866702e-05,
-      "loss": 0.4833,
-      "step": 3131
-    },
-    {
-      "epoch": 5.56,
-      "learning_rate": 1.4880829888231818e-05,
-      "loss": 0.4679,
-      "step": 3132
-    },
-    {
-      "epoch": 5.56,
-      "learning_rate": 1.4877411447034385e-05,
-      "loss": 0.4858,
-      "step": 3133
-    },
-    {
-      "epoch": 5.56,
-      "learning_rate": 1.4873992257798682e-05,
-      "loss": 0.5144,
-      "step": 3134
-    },
-    {
-      "epoch": 5.57,
-      "learning_rate": 1.4870572321049102e-05,
-      "loss": 0.4915,
-      "step": 3135
-    },
-    {
-      "epoch": 5.57,
-      "learning_rate": 1.486715163731015e-05,
-      "loss": 0.4819,
-      "step": 3136
-    },
-    {
-      "epoch": 5.57,
-      "learning_rate": 1.4863730207106456e-05,
-      "loss": 0.4787,
-      "step": 3137
-    },
-    {
-      "epoch": 5.57,
-      "learning_rate": 1.4860308030962752e-05,
-      "loss": 0.4857,
-      "step": 3138
-    },
-    {
-      "epoch": 5.57,
-      "learning_rate": 1.485688510940389e-05,
-      "loss": 0.4642,
-      "step": 3139
-    },
-    {
-      "epoch": 5.57,
-      "learning_rate": 1.4853461442954842e-05,
-      "loss": 0.4821,
-      "step": 3140
-    },
-    {
-      "epoch": 5.58,
-      "learning_rate": 1.4850037032140683e-05,
-      "loss": 0.482,
-      "step": 3141
-    },
-    {
-      "epoch": 5.58,
-      "learning_rate": 1.4846611877486607e-05,
-      "loss": 0.4848,
-      "step": 3142
-    },
-    {
-      "epoch": 5.58,
-      "learning_rate": 1.4843185979517927e-05,
-      "loss": 0.4945,
-      "step": 3143
-    },
-    {
-      "epoch": 5.58,
-      "learning_rate": 1.483975933876006e-05,
-      "loss": 0.472,
-      "step": 3144
-    },
-    {
-      "epoch": 5.58,
-      "learning_rate": 1.483633195573855e-05,
-      "loss": 0.4998,
-      "step": 3145
-    },
-    {
-      "epoch": 5.59,
-      "learning_rate": 1.4832903830979042e-05,
-      "loss": 0.4893,
-      "step": 3146
-    },
-    {
-      "epoch": 5.59,
-      "learning_rate": 1.4829474965007303e-05,
-      "loss": 0.4816,
-      "step": 3147
-    },
-    {
-      "epoch": 5.59,
-      "learning_rate": 1.4826045358349208e-05,
-      "loss": 0.5073,
-      "step": 3148
-    },
-    {
-      "epoch": 5.59,
-      "learning_rate": 1.482261501153075e-05,
-      "loss": 0.4963,
-      "step": 3149
-    },
-    {
-      "epoch": 5.59,
-      "learning_rate": 1.4819183925078035e-05,
-      "loss": 0.4819,
-      "step": 3150
-    },
-    {
-      "epoch": 5.59,
-      "learning_rate": 1.4815752099517283e-05,
-      "loss": 0.4738,
-      "step": 3151
-    },
-    {
-      "epoch": 5.6,
-      "learning_rate": 1.4812319535374819e-05,
-      "loss": 0.5041,
-      "step": 3152
-    },
-    {
-      "epoch": 5.6,
-      "learning_rate": 1.4808886233177096e-05,
-      "loss": 0.475,
-      "step": 3153
-    },
-    {
-      "epoch": 5.6,
-      "learning_rate": 1.480545219345067e-05,
-      "loss": 0.5127,
-      "step": 3154
-    },
-    {
-      "epoch": 5.6,
-      "learning_rate": 1.4802017416722207e-05,
-      "loss": 0.48,
-      "step": 3155
-    },
-    {
-      "epoch": 5.6,
-      "learning_rate": 1.47985819035185e-05,
-      "loss": 0.4858,
-      "step": 3156
-    },
-    {
-      "epoch": 5.6,
-      "learning_rate": 1.4795145654366443e-05,
-      "loss": 0.49,
-      "step": 3157
-    },
-    {
-      "epoch": 5.61,
-      "learning_rate": 1.479170866979304e-05,
-      "loss": 0.4798,
-      "step": 3158
-    },
-    {
-      "epoch": 5.61,
-      "learning_rate": 1.4788270950325424e-05,
-      "loss": 0.4945,
-      "step": 3159
-    },
-    {
-      "epoch": 5.61,
-      "learning_rate": 1.4784832496490824e-05,
-      "loss": 0.4993,
-      "step": 3160
-    },
-    {
-      "epoch": 5.61,
-      "learning_rate": 1.4781393308816594e-05,
-      "loss": 0.4833,
-      "step": 3161
-    },
-    {
-      "epoch": 5.61,
-      "learning_rate": 1.4777953387830189e-05,
-      "loss": 0.5051,
-      "step": 3162
-    },
-    {
-      "epoch": 5.62,
-      "learning_rate": 1.4774512734059183e-05,
-      "loss": 0.5005,
-      "step": 3163
-    },
-    {
-      "epoch": 5.62,
-      "learning_rate": 1.4771071348031268e-05,
-      "loss": 0.4826,
-      "step": 3164
-    },
-    {
-      "epoch": 5.62,
-      "learning_rate": 1.4767629230274236e-05,
-      "loss": 0.497,
-      "step": 3165
-    },
-    {
-      "epoch": 5.62,
-      "learning_rate": 1.4764186381316e-05,
-      "loss": 0.496,
-      "step": 3166
-    },
-    {
-      "epoch": 5.62,
-      "learning_rate": 1.476074280168458e-05,
-      "loss": 0.468,
-      "step": 3167
-    },
-    {
-      "epoch": 5.62,
-      "learning_rate": 1.4757298491908114e-05,
-      "loss": 0.4798,
-      "step": 3168
-    },
-    {
-      "epoch": 5.63,
-      "learning_rate": 1.4753853452514846e-05,
-      "loss": 0.4717,
-      "step": 3169
-    },
-    {
-      "epoch": 5.63,
-      "learning_rate": 1.4750407684033135e-05,
-      "loss": 0.4844,
-      "step": 3170
-    },
-    {
-      "epoch": 5.63,
-      "learning_rate": 1.474696118699145e-05,
-      "loss": 0.4876,
-      "step": 3171
-    },
-    {
-      "epoch": 5.63,
-      "learning_rate": 1.4743513961918375e-05,
-      "loss": 0.4949,
-      "step": 3172
-    },
-    {
-      "epoch": 5.63,
-      "learning_rate": 1.4740066009342604e-05,
-      "loss": 0.5002,
-      "step": 3173
-    },
-    {
-      "epoch": 5.64,
-      "learning_rate": 1.4736617329792942e-05,
-      "loss": 0.4996,
-      "step": 3174
-    },
-    {
-      "epoch": 5.64,
-      "learning_rate": 1.4733167923798301e-05,
-      "loss": 0.4929,
-      "step": 3175
-    },
-    {
-      "epoch": 5.64,
-      "learning_rate": 1.4729717791887716e-05,
-      "loss": 0.5048,
-      "step": 3176
-    },
-    {
-      "epoch": 5.64,
-      "learning_rate": 1.4726266934590321e-05,
-      "loss": 0.4708,
-      "step": 3177
-    },
-    {
-      "epoch": 5.64,
-      "learning_rate": 1.472281535243537e-05,
-      "loss": 0.466,
-      "step": 3178
-    },
-    {
-      "epoch": 5.64,
-      "learning_rate": 1.4719363045952224e-05,
-      "loss": 0.5012,
-      "step": 3179
-    },
-    {
-      "epoch": 5.65,
-      "learning_rate": 1.4715910015670357e-05,
-      "loss": 0.4632,
-      "step": 3180
-    },
-    {
-      "epoch": 5.65,
-      "learning_rate": 1.471245626211935e-05,
-      "loss": 0.4975,
-      "step": 3181
-    },
-    {
-      "epoch": 5.65,
-      "learning_rate": 1.4709001785828898e-05,
-      "loss": 0.4943,
-      "step": 3182
-    },
-    {
-      "epoch": 5.65,
-      "learning_rate": 1.4705546587328811e-05,
-      "loss": 0.4732,
-      "step": 3183
-    },
-    {
-      "epoch": 5.65,
-      "learning_rate": 1.4702090667149003e-05,
-      "loss": 0.4804,
-      "step": 3184
-    },
-    {
-      "epoch": 5.65,
-      "learning_rate": 1.4698634025819499e-05,
-      "loss": 0.4945,
-      "step": 3185
-    },
-    {
-      "epoch": 5.66,
-      "learning_rate": 1.4695176663870438e-05,
-      "loss": 0.4775,
-      "step": 3186
-    },
-    {
-      "epoch": 5.66,
-      "learning_rate": 1.469171858183207e-05,
-      "loss": 0.4838,
-      "step": 3187
-    },
-    {
-      "epoch": 5.66,
-      "learning_rate": 1.4688259780234754e-05,
-      "loss": 0.49,
-      "step": 3188
-    },
-    {
-      "epoch": 5.66,
-      "learning_rate": 1.4684800259608958e-05,
-      "loss": 0.4664,
-      "step": 3189
-    },
-    {
-      "epoch": 5.66,
-      "learning_rate": 1.468134002048526e-05,
-      "loss": 0.4634,
-      "step": 3190
-    },
-    {
-      "epoch": 5.67,
-      "learning_rate": 1.4677879063394351e-05,
-      "loss": 0.4785,
-      "step": 3191
-    },
-    {
-      "epoch": 5.67,
-      "learning_rate": 1.467441738886703e-05,
-      "loss": 0.4959,
-      "step": 3192
-    },
-    {
-      "epoch": 5.67,
-      "learning_rate": 1.4670954997434208e-05,
-      "loss": 0.4763,
-      "step": 3193
-    },
-    {
-      "epoch": 5.67,
-      "learning_rate": 1.4667491889626904e-05,
-      "loss": 0.4809,
-      "step": 3194
-    },
-    {
-      "epoch": 5.67,
-      "learning_rate": 1.4664028065976245e-05,
-      "loss": 0.5046,
-      "step": 3195
-    },
-    {
-      "epoch": 5.67,
-      "learning_rate": 1.4660563527013477e-05,
-      "loss": 0.5024,
-      "step": 3196
-    },
-    {
-      "epoch": 5.68,
-      "learning_rate": 1.4657098273269939e-05,
-      "loss": 0.4971,
-      "step": 3197
-    },
-    {
-      "epoch": 5.68,
-      "learning_rate": 1.4653632305277097e-05,
-      "loss": 0.5016,
-      "step": 3198
-    },
-    {
-      "epoch": 5.68,
-      "learning_rate": 1.465016562356652e-05,
-      "loss": 0.5028,
-      "step": 3199
-    },
-    {
-      "epoch": 5.68,
-      "learning_rate": 1.4646698228669878e-05,
-      "loss": 0.483,
-      "step": 3200
-    },
-    {
-      "epoch": 5.68,
-      "learning_rate": 1.4643230121118965e-05,
-      "loss": 0.4988,
-      "step": 3201
-    },
-    {
-      "epoch": 5.68,
-      "learning_rate": 1.4639761301445677e-05,
-      "loss": 0.4924,
-      "step": 3202
-    },
-    {
-      "epoch": 5.69,
-      "learning_rate": 1.4636291770182017e-05,
-      "loss": 0.4864,
-      "step": 3203
-    },
-    {
-      "epoch": 5.69,
-      "learning_rate": 1.46328215278601e-05,
-      "loss": 0.4781,
-      "step": 3204
-    },
-    {
-      "epoch": 5.69,
-      "learning_rate": 1.4629350575012149e-05,
-      "loss": 0.4966,
-      "step": 3205
-    },
-    {
-      "epoch": 5.69,
-      "learning_rate": 1.4625878912170499e-05,
-      "loss": 0.4953,
-      "step": 3206
-    },
-    {
-      "epoch": 5.69,
-      "learning_rate": 1.4622406539867593e-05,
-      "loss": 0.4776,
-      "step": 3207
-    },
-    {
-      "epoch": 5.7,
-      "learning_rate": 1.4618933458635976e-05,
-      "loss": 0.478,
-      "step": 3208
-    },
-    {
-      "epoch": 5.7,
-      "learning_rate": 1.461545966900831e-05,
-      "loss": 0.5091,
-      "step": 3209
-    },
-    {
-      "epoch": 5.7,
-      "learning_rate": 1.4611985171517364e-05,
-      "loss": 0.4953,
-      "step": 3210
-    },
-    {
-      "epoch": 5.7,
-      "learning_rate": 1.4608509966696015e-05,
-      "loss": 0.4664,
-      "step": 3211
-    },
-    {
-      "epoch": 5.7,
-      "learning_rate": 1.4605034055077241e-05,
-      "loss": 0.4985,
-      "step": 3212
-    },
-    {
-      "epoch": 5.7,
-      "learning_rate": 1.460155743719414e-05,
-      "loss": 0.4792,
-      "step": 3213
-    },
-    {
-      "epoch": 5.71,
-      "learning_rate": 1.459808011357992e-05,
-      "loss": 0.5043,
-      "step": 3214
-    },
-    {
-      "epoch": 5.71,
-      "learning_rate": 1.4594602084767879e-05,
-      "loss": 0.4828,
-      "step": 3215
-    },
-    {
-      "epoch": 5.71,
-      "learning_rate": 1.459112335129144e-05,
-      "loss": 0.4715,
-      "step": 3216
-    },
-    {
-      "epoch": 5.71,
-      "learning_rate": 1.4587643913684134e-05,
-      "loss": 0.4845,
-      "step": 3217
-    },
-    {
-      "epoch": 5.71,
-      "learning_rate": 1.4584163772479588e-05,
-      "loss": 0.499,
-      "step": 3218
-    },
-    {
-      "epoch": 5.72,
-      "learning_rate": 1.4580682928211545e-05,
-      "loss": 0.4824,
-      "step": 3219
-    },
-    {
-      "epoch": 5.72,
-      "learning_rate": 1.4577201381413855e-05,
-      "loss": 0.4686,
-      "step": 3220
-    },
-    {
-      "epoch": 5.72,
-      "learning_rate": 1.4573719132620478e-05,
-      "loss": 0.4905,
-      "step": 3221
-    },
-    {
-      "epoch": 5.72,
-      "learning_rate": 1.4570236182365479e-05,
-      "loss": 0.4954,
-      "step": 3222
-    },
-    {
-      "epoch": 5.72,
-      "learning_rate": 1.4566752531183028e-05,
-      "loss": 0.5028,
-      "step": 3223
-    },
-    {
-      "epoch": 5.72,
-      "learning_rate": 1.4563268179607406e-05,
-      "loss": 0.4835,
-      "step": 3224
-    },
-    {
-      "epoch": 5.73,
-      "learning_rate": 1.4559783128173e-05,
-      "loss": 0.5164,
-      "step": 3225
-    },
-    {
-      "epoch": 5.73,
-      "learning_rate": 1.4556297377414308e-05,
-      "loss": 0.4892,
-      "step": 3226
-    },
-    {
-      "epoch": 5.73,
-      "learning_rate": 1.455281092786593e-05,
-      "loss": 0.4662,
-      "step": 3227
-    },
-    {
-      "epoch": 5.73,
-      "learning_rate": 1.4549323780062575e-05,
-      "loss": 0.4947,
-      "step": 3228
-    },
-    {
-      "epoch": 5.73,
-      "learning_rate": 1.4545835934539062e-05,
-      "loss": 0.4818,
-      "step": 3229
-    },
-    {
-      "epoch": 5.73,
-      "learning_rate": 1.4542347391830308e-05,
-      "loss": 0.4722,
-      "step": 3230
-    },
-    {
-      "epoch": 5.74,
-      "learning_rate": 1.4538858152471349e-05,
-      "loss": 0.4861,
-      "step": 3231
-    },
-    {
-      "epoch": 5.74,
-      "learning_rate": 1.4535368216997323e-05,
-      "loss": 0.4954,
-      "step": 3232
-    },
-    {
-      "epoch": 5.74,
-      "learning_rate": 1.4531877585943472e-05,
-      "loss": 0.481,
-      "step": 3233
-    },
-    {
-      "epoch": 5.74,
-      "learning_rate": 1.4528386259845146e-05,
-      "loss": 0.4773,
-      "step": 3234
-    },
-    {
-      "epoch": 5.74,
-      "learning_rate": 1.4524894239237806e-05,
-      "loss": 0.4608,
-      "step": 3235
-    },
-    {
-      "epoch": 5.75,
-      "learning_rate": 1.4521401524657011e-05,
-      "loss": 0.4944,
-      "step": 3236
-    },
-    {
-      "epoch": 5.75,
-      "learning_rate": 1.4517908116638433e-05,
-      "loss": 0.4767,
-      "step": 3237
-    },
-    {
-      "epoch": 5.75,
-      "learning_rate": 1.4514414015717848e-05,
-      "loss": 0.507,
-      "step": 3238
-    },
-    {
-      "epoch": 5.75,
-      "learning_rate": 1.451091922243114e-05,
-      "loss": 0.4769,
-      "step": 3239
-    },
-    {
-      "epoch": 5.75,
-      "learning_rate": 1.4507423737314299e-05,
-      "loss": 0.4869,
-      "step": 3240
-    },
-    {
-      "epoch": 5.75,
-      "learning_rate": 1.4503927560903421e-05,
-      "loss": 0.4701,
-      "step": 3241
-    },
-    {
-      "epoch": 5.76,
-      "learning_rate": 1.4500430693734702e-05,
-      "loss": 0.5001,
-      "step": 3242
-    },
-    {
-      "epoch": 5.76,
-      "learning_rate": 1.4496933136344453e-05,
-      "loss": 0.492,
-      "step": 3243
-    },
-    {
-      "epoch": 5.76,
-      "learning_rate": 1.4493434889269087e-05,
-      "loss": 0.477,
-      "step": 3244
-    },
-    {
-      "epoch": 5.76,
-      "learning_rate": 1.4489935953045122e-05,
-      "loss": 0.4823,
-      "step": 3245
-    },
-    {
-      "epoch": 5.76,
-      "learning_rate": 1.4486436328209181e-05,
-      "loss": 0.4819,
-      "step": 3246
-    },
-    {
-      "epoch": 5.76,
-      "learning_rate": 1.4482936015297994e-05,
-      "loss": 0.4928,
-      "step": 3247
-    },
-    {
-      "epoch": 5.77,
-      "learning_rate": 1.4479435014848404e-05,
-      "loss": 0.4866,
-      "step": 3248
-    },
-    {
-      "epoch": 5.77,
-      "learning_rate": 1.4475933327397342e-05,
-      "loss": 0.4846,
-      "step": 3249
-    },
-    {
-      "epoch": 5.77,
-      "learning_rate": 1.4472430953481859e-05,
-      "loss": 0.495,
-      "step": 3250
-    },
-    {
-      "epoch": 5.77,
-      "learning_rate": 1.4468927893639106e-05,
-      "loss": 0.4841,
-      "step": 3251
-    },
-    {
-      "epoch": 5.77,
-      "learning_rate": 1.446542414840634e-05,
-      "loss": 0.4867,
-      "step": 3252
-    },
-    {
-      "epoch": 5.78,
-      "learning_rate": 1.4461919718320924e-05,
-      "loss": 0.4654,
-      "step": 3253
-    },
-    {
-      "epoch": 5.78,
-      "learning_rate": 1.4458414603920323e-05,
-      "loss": 0.4805,
-      "step": 3254
-    },
-    {
-      "epoch": 5.78,
-      "learning_rate": 1.445490880574211e-05,
-      "loss": 0.4768,
-      "step": 3255
-    },
-    {
-      "epoch": 5.78,
-      "learning_rate": 1.4451402324323964e-05,
-      "loss": 0.4765,
-      "step": 3256
-    },
-    {
-      "epoch": 5.78,
-      "learning_rate": 1.4447895160203662e-05,
-      "loss": 0.4986,
-      "step": 3257
-    },
-    {
-      "epoch": 5.78,
-      "learning_rate": 1.4444387313919092e-05,
-      "loss": 0.4745,
-      "step": 3258
-    },
-    {
-      "epoch": 5.79,
-      "learning_rate": 1.4440878786008248e-05,
-      "loss": 0.471,
-      "step": 3259
-    },
-    {
-      "epoch": 5.79,
-      "learning_rate": 1.4437369577009223e-05,
-      "loss": 0.4807,
-      "step": 3260
-    },
-    {
-      "epoch": 5.79,
-      "learning_rate": 1.4433859687460216e-05,
-      "loss": 0.4954,
-      "step": 3261
-    },
-    {
-      "epoch": 5.79,
-      "learning_rate": 1.4430349117899533e-05,
-      "loss": 0.4797,
-      "step": 3262
-    },
-    {
-      "epoch": 5.79,
-      "learning_rate": 1.4426837868865584e-05,
-      "loss": 0.4912,
-      "step": 3263
-    },
-    {
-      "epoch": 5.79,
-      "learning_rate": 1.4423325940896878e-05,
-      "loss": 0.4877,
-      "step": 3264
-    },
-    {
-      "epoch": 5.8,
-      "learning_rate": 1.4419813334532037e-05,
-      "loss": 0.4942,
-      "step": 3265
-    },
-    {
-      "epoch": 5.8,
-      "learning_rate": 1.4416300050309776e-05,
-      "loss": 0.4839,
-      "step": 3266
-    },
-    {
-      "epoch": 5.8,
-      "learning_rate": 1.4412786088768923e-05,
-      "loss": 0.4846,
-      "step": 3267
-    },
-    {
-      "epoch": 5.8,
-      "learning_rate": 1.4409271450448405e-05,
-      "loss": 0.4765,
-      "step": 3268
-    },
-    {
-      "epoch": 5.8,
-      "learning_rate": 1.4405756135887261e-05,
-      "loss": 0.4694,
-      "step": 3269
-    },
-    {
-      "epoch": 5.81,
-      "learning_rate": 1.440224014562462e-05,
-      "loss": 0.5013,
-      "step": 3270
-    },
-    {
-      "epoch": 5.81,
-      "learning_rate": 1.4398723480199725e-05,
-      "loss": 0.4794,
-      "step": 3271
-    },
-    {
-      "epoch": 5.81,
-      "learning_rate": 1.439520614015192e-05,
-      "loss": 0.4731,
-      "step": 3272
-    },
-    {
-      "epoch": 5.81,
-      "learning_rate": 1.439168812602065e-05,
-      "loss": 0.4802,
-      "step": 3273
-    },
-    {
-      "epoch": 5.81,
-      "learning_rate": 1.4388169438345467e-05,
-      "loss": 0.4766,
-      "step": 3274
-    },
-    {
-      "epoch": 5.81,
-      "learning_rate": 1.438465007766603e-05,
-      "loss": 0.4925,
-      "step": 3275
-    },
-    {
-      "epoch": 5.82,
-      "learning_rate": 1.4381130044522085e-05,
-      "loss": 0.4909,
-      "step": 3276
-    },
-    {
-      "epoch": 5.82,
-      "learning_rate": 1.4377609339453498e-05,
-      "loss": 0.499,
-      "step": 3277
-    },
-    {
-      "epoch": 5.82,
-      "learning_rate": 1.437408796300023e-05,
-      "loss": 0.4883,
-      "step": 3278
-    },
-    {
-      "epoch": 5.82,
-      "learning_rate": 1.437056591570235e-05,
-      "loss": 0.4899,
-      "step": 3279
-    },
-    {
-      "epoch": 5.82,
-      "learning_rate": 1.4367043198100028e-05,
-      "loss": 0.4805,
-      "step": 3280
-    },
-    {
-      "epoch": 5.83,
-      "learning_rate": 1.4363519810733529e-05,
-      "loss": 0.4901,
-      "step": 3281
-    },
-    {
-      "epoch": 5.83,
-      "learning_rate": 1.4359995754143234e-05,
-      "loss": 0.5088,
-      "step": 3282
-    },
-    {
-      "epoch": 5.83,
-      "learning_rate": 1.4356471028869617e-05,
-      "loss": 0.4948,
-      "step": 3283
-    },
-    {
-      "epoch": 5.83,
-      "learning_rate": 1.4352945635453257e-05,
-      "loss": 0.4801,
-      "step": 3284
-    },
-    {
-      "epoch": 5.83,
-      "learning_rate": 1.4349419574434838e-05,
-      "loss": 0.4647,
-      "step": 3285
-    },
-    {
-      "epoch": 5.83,
-      "learning_rate": 1.4345892846355147e-05,
-      "loss": 0.4943,
-      "step": 3286
-    },
-    {
-      "epoch": 5.84,
-      "learning_rate": 1.4342365451755064e-05,
-      "loss": 0.4676,
-      "step": 3287
-    },
-    {
-      "epoch": 5.84,
-      "learning_rate": 1.4338837391175582e-05,
-      "loss": 0.482,
-      "step": 3288
-    },
-    {
-      "epoch": 5.84,
-      "learning_rate": 1.4335308665157794e-05,
-      "loss": 0.4788,
-      "step": 3289
-    },
-    {
-      "epoch": 5.84,
-      "learning_rate": 1.4331779274242892e-05,
-      "loss": 0.5033,
-      "step": 3290
-    },
-    {
-      "epoch": 5.84,
-      "learning_rate": 1.4328249218972168e-05,
-      "loss": 0.4998,
-      "step": 3291
-    },
-    {
-      "epoch": 5.84,
-      "learning_rate": 1.4324718499887022e-05,
-      "loss": 0.4594,
-      "step": 3292
-    },
-    {
-      "epoch": 5.85,
-      "learning_rate": 1.4321187117528954e-05,
-      "loss": 0.4706,
-      "step": 3293
-    },
-    {
-      "epoch": 5.85,
-      "learning_rate": 1.4317655072439564e-05,
-      "loss": 0.4946,
-      "step": 3294
-    },
-    {
-      "epoch": 5.85,
-      "learning_rate": 1.4314122365160551e-05,
-      "loss": 0.4892,
-      "step": 3295
-    },
-    {
-      "epoch": 5.85,
-      "learning_rate": 1.4310588996233724e-05,
-      "loss": 0.5042,
-      "step": 3296
-    },
-    {
-      "epoch": 5.85,
-      "learning_rate": 1.4307054966200984e-05,
-      "loss": 0.4784,
-      "step": 3297
-    },
-    {
-      "epoch": 5.86,
-      "learning_rate": 1.4303520275604339e-05,
-      "loss": 0.4799,
-      "step": 3298
-    },
-    {
-      "epoch": 5.86,
-      "learning_rate": 1.4299984924985897e-05,
-      "loss": 0.4883,
-      "step": 3299
-    },
-    {
-      "epoch": 5.86,
-      "learning_rate": 1.4296448914887866e-05,
-      "loss": 0.4792,
-      "step": 3300
-    },
-    {
-      "epoch": 5.86,
-      "learning_rate": 1.4292912245852564e-05,
-      "loss": 0.4936,
-      "step": 3301
-    },
-    {
-      "epoch": 5.86,
-      "learning_rate": 1.428937491842239e-05,
-      "loss": 0.4956,
-      "step": 3302
-    },
-    {
-      "epoch": 5.86,
-      "learning_rate": 1.4285836933139865e-05,
-      "loss": 0.4829,
-      "step": 3303
-    },
-    {
-      "epoch": 5.87,
-      "learning_rate": 1.4282298290547603e-05,
-      "loss": 0.4864,
-      "step": 3304
-    },
-    {
-      "epoch": 5.87,
-      "learning_rate": 1.4278758991188314e-05,
-      "loss": 0.4892,
-      "step": 3305
-    },
-    {
-      "epoch": 5.87,
-      "learning_rate": 1.427521903560481e-05,
-      "loss": 0.4668,
-      "step": 3306
-    },
-    {
-      "epoch": 5.87,
-      "learning_rate": 1.4271678424340016e-05,
-      "loss": 0.4849,
-      "step": 3307
-    },
-    {
-      "epoch": 5.87,
-      "learning_rate": 1.426813715793694e-05,
-      "loss": 0.499,
-      "step": 3308
-    },
-    {
-      "epoch": 5.87,
-      "learning_rate": 1.42645952369387e-05,
-      "loss": 0.4882,
-      "step": 3309
-    },
-    {
-      "epoch": 5.88,
-      "learning_rate": 1.4261052661888516e-05,
-      "loss": 0.4868,
-      "step": 3310
-    },
-    {
-      "epoch": 5.88,
-      "learning_rate": 1.4257509433329704e-05,
-      "loss": 0.4979,
-      "step": 3311
-    },
-    {
-      "epoch": 5.88,
-      "learning_rate": 1.425396555180568e-05,
-      "loss": 0.4918,
-      "step": 3312
-    },
-    {
-      "epoch": 5.88,
-      "learning_rate": 1.4250421017859963e-05,
-      "loss": 0.4806,
-      "step": 3313
-    },
-    {
-      "epoch": 5.88,
-      "learning_rate": 1.4246875832036169e-05,
-      "loss": 0.473,
-      "step": 3314
-    },
-    {
-      "epoch": 5.89,
-      "learning_rate": 1.4243329994878017e-05,
-      "loss": 0.4892,
-      "step": 3315
-    },
-    {
-      "epoch": 5.89,
-      "learning_rate": 1.4239783506929326e-05,
-      "loss": 0.4966,
-      "step": 3316
-    },
-    {
-      "epoch": 5.89,
-      "learning_rate": 1.4236236368734009e-05,
-      "loss": 0.4876,
-      "step": 3317
-    },
-    {
-      "epoch": 5.89,
-      "learning_rate": 1.4232688580836085e-05,
-      "loss": 0.4801,
-      "step": 3318
-    },
-    {
-      "epoch": 5.89,
-      "learning_rate": 1.4229140143779674e-05,
-      "loss": 0.4971,
-      "step": 3319
-    },
-    {
-      "epoch": 5.89,
-      "learning_rate": 1.422559105810899e-05,
-      "loss": 0.5041,
-      "step": 3320
-    },
-    {
-      "epoch": 5.9,
-      "learning_rate": 1.4222041324368347e-05,
-      "loss": 0.492,
-      "step": 3321
-    },
-    {
-      "epoch": 5.9,
-      "learning_rate": 1.421849094310216e-05,
-      "loss": 0.4872,
-      "step": 3322
-    },
-    {
-      "epoch": 5.9,
-      "learning_rate": 1.4214939914854948e-05,
-      "loss": 0.4887,
-      "step": 3323
-    },
-    {
-      "epoch": 5.9,
-      "learning_rate": 1.4211388240171321e-05,
-      "loss": 0.4922,
-      "step": 3324
-    },
-    {
-      "epoch": 5.9,
-      "learning_rate": 1.4207835919595994e-05,
-      "loss": 0.4973,
-      "step": 3325
-    },
-    {
-      "epoch": 5.91,
-      "learning_rate": 1.4204282953673777e-05,
-      "loss": 0.5007,
-      "step": 3326
-    },
-    {
-      "epoch": 5.91,
-      "learning_rate": 1.4200729342949577e-05,
-      "loss": 0.4703,
-      "step": 3327
-    },
-    {
-      "epoch": 5.91,
-      "learning_rate": 1.4197175087968415e-05,
-      "loss": 0.4986,
-      "step": 3328
-    },
-    {
-      "epoch": 5.91,
-      "learning_rate": 1.4193620189275388e-05,
-      "loss": 0.4873,
-      "step": 3329
-    },
-    {
-      "epoch": 5.91,
-      "learning_rate": 1.419006464741571e-05,
-      "loss": 0.4625,
-      "step": 3330
-    },
-    {
-      "epoch": 5.91,
-      "learning_rate": 1.4186508462934686e-05,
-      "loss": 0.5143,
-      "step": 3331
-    },
-    {
-      "epoch": 5.92,
-      "learning_rate": 1.4182951636377716e-05,
-      "loss": 0.479,
-      "step": 3332
-    },
-    {
-      "epoch": 5.92,
-      "learning_rate": 1.4179394168290309e-05,
-      "loss": 0.4758,
-      "step": 3333
-    },
-    {
-      "epoch": 5.92,
-      "learning_rate": 1.4175836059218061e-05,
-      "loss": 0.4815,
-      "step": 3334
-    },
-    {
-      "epoch": 5.92,
-      "learning_rate": 1.4172277309706677e-05,
-      "loss": 0.4885,
-      "step": 3335
-    },
-    {
-      "epoch": 5.92,
-      "learning_rate": 1.4168717920301953e-05,
-      "loss": 0.4787,
-      "step": 3336
-    },
-    {
-      "epoch": 5.92,
-      "learning_rate": 1.4165157891549784e-05,
-      "loss": 0.4688,
-      "step": 3337
-    },
-    {
-      "epoch": 5.93,
-      "learning_rate": 1.4161597223996162e-05,
-      "loss": 0.4827,
-      "step": 3338
-    },
-    {
-      "epoch": 5.93,
-      "learning_rate": 1.4158035918187182e-05,
-      "loss": 0.4877,
-      "step": 3339
-    },
-    {
-      "epoch": 5.93,
-      "learning_rate": 1.4154473974669034e-05,
-      "loss": 0.4772,
-      "step": 3340
-    },
-    {
-      "epoch": 5.93,
-      "learning_rate": 1.4150911393988004e-05,
-      "loss": 0.506,
-      "step": 3341
-    },
-    {
-      "epoch": 5.93,
-      "learning_rate": 1.4147348176690479e-05,
-      "loss": 0.4952,
-      "step": 3342
-    },
-    {
-      "epoch": 5.94,
-      "learning_rate": 1.4143784323322941e-05,
-      "loss": 0.4638,
-      "step": 3343
-    },
-    {
-      "epoch": 5.94,
-      "learning_rate": 1.4140219834431974e-05,
-      "loss": 0.4835,
-      "step": 3344
-    },
-    {
-      "epoch": 5.94,
-      "learning_rate": 1.4136654710564251e-05,
-      "loss": 0.496,
-      "step": 3345
-    },
-    {
-      "epoch": 5.94,
-      "learning_rate": 1.413308895226655e-05,
-      "loss": 0.5023,
-      "step": 3346
-    },
-    {
-      "epoch": 5.94,
-      "learning_rate": 1.4129522560085747e-05,
-      "loss": 0.4829,
-      "step": 3347
-    },
-    {
-      "epoch": 5.94,
-      "learning_rate": 1.4125955534568805e-05,
-      "loss": 0.4803,
-      "step": 3348
-    },
-    {
-      "epoch": 5.95,
-      "learning_rate": 1.4122387876262794e-05,
-      "loss": 0.4905,
-      "step": 3349
-    },
-    {
-      "epoch": 5.95,
-      "learning_rate": 1.4118819585714882e-05,
-      "loss": 0.5111,
-      "step": 3350
-    },
-    {
-      "epoch": 5.95,
-      "learning_rate": 1.4115250663472326e-05,
-      "loss": 0.5103,
-      "step": 3351
-    },
-    {
-      "epoch": 5.95,
-      "learning_rate": 1.4111681110082484e-05,
-      "loss": 0.5085,
-      "step": 3352
-    },
-    {
-      "epoch": 5.95,
-      "learning_rate": 1.4108110926092816e-05,
-      "loss": 0.4926,
-      "step": 3353
-    },
-    {
-      "epoch": 5.95,
-      "learning_rate": 1.4104540112050866e-05,
-      "loss": 0.4892,
-      "step": 3354
-    },
-    {
-      "epoch": 5.96,
-      "learning_rate": 1.4100968668504287e-05,
-      "loss": 0.4805,
-      "step": 3355
-    },
-    {
-      "epoch": 5.96,
-      "learning_rate": 1.4097396596000822e-05,
-      "loss": 0.4974,
-      "step": 3356
-    },
-    {
-      "epoch": 5.96,
-      "learning_rate": 1.4093823895088315e-05,
-      "loss": 0.4801,
-      "step": 3357
-    },
-    {
-      "epoch": 5.96,
-      "learning_rate": 1.40902505663147e-05,
-      "loss": 0.4823,
-      "step": 3358
-    },
-    {
-      "epoch": 5.96,
-      "learning_rate": 1.4086676610228012e-05,
-      "loss": 0.4949,
-      "step": 3359
-    },
-    {
-      "epoch": 5.97,
-      "learning_rate": 1.408310202737638e-05,
-      "loss": 0.4706,
-      "step": 3360
-    },
-    {
-      "epoch": 5.97,
-      "learning_rate": 1.4079526818308036e-05,
-      "loss": 0.4816,
-      "step": 3361
-    },
-    {
-      "epoch": 5.97,
-      "learning_rate": 1.4075950983571294e-05,
-      "loss": 0.4881,
-      "step": 3362
-    },
-    {
-      "epoch": 5.97,
-      "learning_rate": 1.4072374523714577e-05,
-      "loss": 0.4753,
-      "step": 3363
-    },
-    {
-      "epoch": 5.97,
-      "learning_rate": 1.40687974392864e-05,
-      "loss": 0.4814,
-      "step": 3364
-    },
-    {
-      "epoch": 5.97,
-      "learning_rate": 1.4065219730835369e-05,
-      "loss": 0.4904,
-      "step": 3365
-    },
-    {
-      "epoch": 5.98,
-      "learning_rate": 1.4061641398910193e-05,
-      "loss": 0.4932,
-      "step": 3366
-    },
-    {
-      "epoch": 5.98,
-      "learning_rate": 1.4058062444059668e-05,
-      "loss": 0.4776,
-      "step": 3367
-    },
-    {
-      "epoch": 5.98,
-      "learning_rate": 1.4054482866832698e-05,
-      "loss": 0.5122,
-      "step": 3368
-    },
-    {
-      "epoch": 5.98,
-      "learning_rate": 1.4050902667778272e-05,
-      "loss": 0.4815,
-      "step": 3369
-    },
-    {
-      "epoch": 5.98,
-      "learning_rate": 1.4047321847445474e-05,
-      "loss": 0.47,
-      "step": 3370
-    },
-    {
-      "epoch": 5.98,
-      "learning_rate": 1.4043740406383493e-05,
-      "loss": 0.4818,
-      "step": 3371
-    },
-    {
-      "epoch": 5.99,
-      "learning_rate": 1.4040158345141601e-05,
-      "loss": 0.4757,
-      "step": 3372
-    },
-    {
-      "epoch": 5.99,
-      "learning_rate": 1.4036575664269179e-05,
-      "loss": 0.4995,
-      "step": 3373
-    },
-    {
-      "epoch": 5.99,
-      "learning_rate": 1.4032992364315687e-05,
-      "loss": 0.473,
-      "step": 3374
-    },
-    {
-      "epoch": 5.99,
-      "learning_rate": 1.402940844583069e-05,
-      "loss": 0.5011,
-      "step": 3375
-    },
-    {
-      "epoch": 5.99,
-      "learning_rate": 1.4025823909363849e-05,
-      "loss": 0.4771,
-      "step": 3376
-    },
-    {
-      "epoch": 6.0,
-      "learning_rate": 1.4022238755464918e-05,
-      "loss": 0.4895,
-      "step": 3377
-    },
-    {
-      "epoch": 6.0,
-      "learning_rate": 1.401865298468374e-05,
-      "loss": 0.4912,
-      "step": 3378
-    },
-    {
-      "epoch": 6.0,
-      "learning_rate": 1.4015066597570258e-05,
-      "loss": 0.489,
-      "step": 3379
-    },
-    {
-      "epoch": 6.0,
-      "eval_loss": 0.521112859249115,
-      "eval_runtime": 3449.2583,
-      "eval_samples_per_second": 4.159,
-      "eval_steps_per_second": 1.04,
-      "step": 3379
-    },
-    {
-      "epoch": 6.0,
-      "learning_rate": 1.401147959467451e-05,
-      "loss": 0.4911,
-      "step": 3380
-    },
-    {
-      "epoch": 6.0,
-      "learning_rate": 1.4007891976546627e-05,
-      "loss": 0.5081,
-      "step": 3381
-    },
-    {
-      "epoch": 6.0,
-      "learning_rate": 1.4004303743736833e-05,
-      "loss": 0.4997,
-      "step": 3382
-    },
-    {
-      "epoch": 6.01,
-      "learning_rate": 1.4000714896795448e-05,
-      "loss": 0.4742,
-      "step": 3383
-    },
-    {
-      "epoch": 6.01,
-      "learning_rate": 1.399712543627289e-05,
-      "loss": 0.499,
-      "step": 3384
-    },
-    {
-      "epoch": 6.01,
-      "learning_rate": 1.3993535362719659e-05,
-      "loss": 0.4829,
-      "step": 3385
-    },
-    {
-      "epoch": 6.01,
-      "learning_rate": 1.3989944676686363e-05,
-      "loss": 0.4728,
-      "step": 3386
-    },
-    {
-      "epoch": 6.01,
-      "learning_rate": 1.3986353378723696e-05,
-      "loss": 0.4768,
-      "step": 3387
-    },
-    {
-      "epoch": 6.02,
-      "learning_rate": 1.3982761469382448e-05,
-      "loss": 0.487,
-      "step": 3388
-    },
-    {
-      "epoch": 6.02,
-      "learning_rate": 1.3979168949213504e-05,
-      "loss": 0.4942,
-      "step": 3389
-    },
-    {
-      "epoch": 6.02,
-      "learning_rate": 1.3975575818767841e-05,
-      "loss": 0.4815,
-      "step": 3390
-    },
-    {
-      "epoch": 6.02,
-      "learning_rate": 1.3971982078596526e-05,
-      "loss": 0.4999,
-      "step": 3391
-    },
-    {
-      "epoch": 6.02,
-      "learning_rate": 1.3968387729250728e-05,
-      "loss": 0.4962,
-      "step": 3392
-    },
-    {
-      "epoch": 6.02,
-      "learning_rate": 1.3964792771281702e-05,
-      "loss": 0.4678,
-      "step": 3393
-    },
-    {
-      "epoch": 6.03,
-      "learning_rate": 1.3961197205240797e-05,
-      "loss": 0.4593,
-      "step": 3394
-    },
-    {
-      "epoch": 6.03,
-      "learning_rate": 1.395760103167946e-05,
-      "loss": 0.4768,
-      "step": 3395
-    },
-    {
-      "epoch": 6.03,
-      "learning_rate": 1.3954004251149233e-05,
-      "loss": 0.4755,
-      "step": 3396
-    },
-    {
-      "epoch": 6.03,
-      "learning_rate": 1.395040686420174e-05,
-      "loss": 0.4709,
-      "step": 3397
-    },
-    {
-      "epoch": 6.03,
-      "learning_rate": 1.3946808871388706e-05,
-      "loss": 0.4705,
-      "step": 3398
-    },
-    {
-      "epoch": 6.03,
-      "learning_rate": 1.394321027326195e-05,
-      "loss": 0.4966,
-      "step": 3399
-    },
-    {
-      "epoch": 6.04,
-      "learning_rate": 1.393961107037338e-05,
-      "loss": 0.4662,
-      "step": 3400
-    },
-    {
-      "epoch": 6.04,
-      "learning_rate": 1.3936011263274996e-05,
-      "loss": 0.46,
-      "step": 3401
-    },
-    {
-      "epoch": 6.04,
-      "learning_rate": 1.3932410852518896e-05,
-      "loss": 0.4623,
-      "step": 3402
-    },
-    {
-      "epoch": 6.04,
-      "learning_rate": 1.3928809838657266e-05,
-      "loss": 0.4957,
-      "step": 3403
-    },
-    {
-      "epoch": 6.04,
-      "learning_rate": 1.3925208222242387e-05,
-      "loss": 0.4858,
-      "step": 3404
-    },
-    {
-      "epoch": 6.05,
-      "learning_rate": 1.392160600382663e-05,
-      "loss": 0.4755,
-      "step": 3405
-    },
-    {
-      "epoch": 6.05,
-      "learning_rate": 1.3918003183962459e-05,
-      "loss": 0.489,
-      "step": 3406
-    },
-    {
-      "epoch": 6.05,
-      "learning_rate": 1.3914399763202431e-05,
-      "loss": 0.5055,
-      "step": 3407
-    },
-    {
-      "epoch": 6.05,
-      "learning_rate": 1.3910795742099198e-05,
-      "loss": 0.477,
-      "step": 3408
-    },
-    {
-      "epoch": 6.05,
-      "learning_rate": 1.3907191121205497e-05,
-      "loss": 0.4804,
-      "step": 3409
-    },
-    {
-      "epoch": 6.05,
-      "learning_rate": 1.3903585901074163e-05,
-      "loss": 0.4792,
-      "step": 3410
-    },
-    {
-      "epoch": 6.06,
-      "learning_rate": 1.3899980082258122e-05,
-      "loss": 0.4803,
-      "step": 3411
-    },
-    {
-      "epoch": 6.06,
-      "learning_rate": 1.389637366531039e-05,
-      "loss": 0.4874,
-      "step": 3412
-    },
-    {
-      "epoch": 6.06,
-      "learning_rate": 1.3892766650784072e-05,
-      "loss": 0.4848,
-      "step": 3413
-    },
-    {
-      "epoch": 6.06,
-      "learning_rate": 1.3889159039232376e-05,
-      "loss": 0.4681,
-      "step": 3414
-    },
-    {
-      "epoch": 6.06,
-      "learning_rate": 1.3885550831208588e-05,
-      "loss": 0.4876,
-      "step": 3415
-    },
-    {
-      "epoch": 6.06,
-      "learning_rate": 1.388194202726609e-05,
-      "loss": 0.4985,
-      "step": 3416
-    },
-    {
-      "epoch": 6.07,
-      "learning_rate": 1.3878332627958365e-05,
-      "loss": 0.4777,
-      "step": 3417
-    },
-    {
-      "epoch": 6.07,
-      "learning_rate": 1.3874722633838966e-05,
-      "loss": 0.4885,
-      "step": 3418
-    },
-    {
-      "epoch": 6.07,
-      "learning_rate": 1.387111204546156e-05,
-      "loss": 0.4707,
-      "step": 3419
-    },
-    {
-      "epoch": 6.07,
-      "learning_rate": 1.3867500863379894e-05,
-      "loss": 0.4977,
-      "step": 3420
-    },
-    {
-      "epoch": 6.07,
-      "learning_rate": 1.3863889088147801e-05,
-      "loss": 0.4911,
-      "step": 3421
-    },
-    {
-      "epoch": 6.08,
-      "learning_rate": 1.3860276720319223e-05,
-      "loss": 0.4506,
-      "step": 3422
-    },
-    {
-      "epoch": 6.08,
-      "learning_rate": 1.385666376044817e-05,
-      "loss": 0.4827,
-      "step": 3423
-    },
-    {
-      "epoch": 6.08,
-      "learning_rate": 1.385305020908876e-05,
-      "loss": 0.4783,
-      "step": 3424
-    },
-    {
-      "epoch": 6.08,
-      "learning_rate": 1.384943606679519e-05,
-      "loss": 0.4953,
-      "step": 3425
-    },
-    {
-      "epoch": 6.08,
-      "learning_rate": 1.3845821334121763e-05,
-      "loss": 0.4869,
-      "step": 3426
-    },
-    {
-      "epoch": 6.08,
-      "learning_rate": 1.3842206011622855e-05,
-      "loss": 0.4918,
-      "step": 3427
-    },
-    {
-      "epoch": 6.09,
-      "learning_rate": 1.383859009985294e-05,
-      "loss": 0.4685,
-      "step": 3428
-    },
-    {
-      "epoch": 6.09,
-      "learning_rate": 1.3834973599366588e-05,
-      "loss": 0.49,
-      "step": 3429
-    },
-    {
-      "epoch": 6.09,
-      "learning_rate": 1.3831356510718452e-05,
-      "loss": 0.4896,
-      "step": 3430
-    },
-    {
-      "epoch": 6.09,
-      "learning_rate": 1.3827738834463275e-05,
-      "loss": 0.4599,
-      "step": 3431
-    },
-    {
-      "epoch": 6.09,
-      "learning_rate": 1.3824120571155895e-05,
-      "loss": 0.4913,
-      "step": 3432
-    },
-    {
-      "epoch": 6.09,
-      "learning_rate": 1.3820501721351236e-05,
-      "loss": 0.4583,
-      "step": 3433
-    },
-    {
-      "epoch": 6.1,
-      "learning_rate": 1.3816882285604316e-05,
-      "loss": 0.4709,
-      "step": 3434
-    },
-    {
-      "epoch": 6.1,
-      "learning_rate": 1.3813262264470235e-05,
-      "loss": 0.4777,
-      "step": 3435
-    },
-    {
-      "epoch": 6.1,
-      "learning_rate": 1.3809641658504192e-05,
-      "loss": 0.4929,
-      "step": 3436
-    },
-    {
-      "epoch": 6.1,
-      "learning_rate": 1.3806020468261471e-05,
-      "loss": 0.469,
-      "step": 3437
-    },
-    {
-      "epoch": 6.1,
-      "learning_rate": 1.3802398694297445e-05,
-      "loss": 0.4895,
-      "step": 3438
-    },
-    {
-      "epoch": 6.11,
-      "learning_rate": 1.3798776337167579e-05,
-      "loss": 0.5123,
-      "step": 3439
-    },
-    {
-      "epoch": 6.11,
-      "learning_rate": 1.3795153397427426e-05,
-      "loss": 0.4918,
-      "step": 3440
-    },
-    {
-      "epoch": 6.11,
-      "learning_rate": 1.3791529875632628e-05,
-      "loss": 0.4834,
-      "step": 3441
-    },
-    {
-      "epoch": 6.11,
-      "learning_rate": 1.3787905772338918e-05,
-      "loss": 0.476,
-      "step": 3442
-    },
-    {
-      "epoch": 6.11,
-      "learning_rate": 1.3784281088102115e-05,
-      "loss": 0.4841,
-      "step": 3443
-    },
-    {
-      "epoch": 6.11,
-      "learning_rate": 1.3780655823478135e-05,
-      "loss": 0.4852,
-      "step": 3444
-    },
-    {
-      "epoch": 6.12,
-      "learning_rate": 1.3777029979022969e-05,
-      "loss": 0.4696,
-      "step": 3445
-    },
-    {
-      "epoch": 6.12,
-      "learning_rate": 1.377340355529271e-05,
-      "loss": 0.5076,
-      "step": 3446
-    },
-    {
-      "epoch": 6.12,
-      "learning_rate": 1.3769776552843532e-05,
-      "loss": 0.4577,
-      "step": 3447
-    },
-    {
-      "epoch": 6.12,
-      "learning_rate": 1.3766148972231706e-05,
-      "loss": 0.455,
-      "step": 3448
-    },
-    {
-      "epoch": 6.12,
-      "learning_rate": 1.376252081401358e-05,
-      "loss": 0.4878,
-      "step": 3449
-    },
-    {
-      "epoch": 6.13,
-      "learning_rate": 1.37588920787456e-05,
-      "loss": 0.4821,
-      "step": 3450
-    },
-    {
-      "epoch": 6.13,
-      "learning_rate": 1.3755262766984298e-05,
-      "loss": 0.4754,
-      "step": 3451
-    },
-    {
-      "epoch": 6.13,
-      "learning_rate": 1.3751632879286291e-05,
-      "loss": 0.4709,
-      "step": 3452
-    },
-    {
-      "epoch": 6.13,
-      "learning_rate": 1.3748002416208292e-05,
-      "loss": 0.4832,
-      "step": 3453
-    },
-    {
-      "epoch": 6.13,
-      "learning_rate": 1.3744371378307095e-05,
-      "loss": 0.4959,
-      "step": 3454
-    },
-    {
-      "epoch": 6.13,
-      "learning_rate": 1.3740739766139583e-05,
-      "loss": 0.4804,
-      "step": 3455
-    },
-    {
-      "epoch": 6.14,
-      "learning_rate": 1.3737107580262731e-05,
-      "loss": 0.4804,
-      "step": 3456
-    },
-    {
-      "epoch": 6.14,
-      "learning_rate": 1.3733474821233596e-05,
-      "loss": 0.4876,
-      "step": 3457
-    },
-    {
-      "epoch": 6.14,
-      "learning_rate": 1.3729841489609332e-05,
-      "loss": 0.478,
-      "step": 3458
-    },
-    {
-      "epoch": 6.14,
-      "learning_rate": 1.372620758594717e-05,
-      "loss": 0.4767,
-      "step": 3459
-    },
-    {
-      "epoch": 6.14,
-      "learning_rate": 1.372257311080444e-05,
-      "loss": 0.49,
-      "step": 3460
-    },
-    {
-      "epoch": 6.14,
-      "learning_rate": 1.371893806473855e-05,
-      "loss": 0.4833,
-      "step": 3461
-    },
-    {
-      "epoch": 6.15,
-      "learning_rate": 1.3715302448306998e-05,
-      "loss": 0.4745,
-      "step": 3462
-    },
-    {
-      "epoch": 6.15,
-      "learning_rate": 1.371166626206737e-05,
-      "loss": 0.4874,
-      "step": 3463
-    },
-    {
-      "epoch": 6.15,
-      "learning_rate": 1.3708029506577349e-05,
-      "loss": 0.4716,
-      "step": 3464
-    },
-    {
-      "epoch": 6.15,
-      "learning_rate": 1.3704392182394686e-05,
-      "loss": 0.4917,
-      "step": 3465
-    },
-    {
-      "epoch": 6.15,
-      "learning_rate": 1.3700754290077236e-05,
-      "loss": 0.4677,
-      "step": 3466
-    },
-    {
-      "epoch": 6.16,
-      "learning_rate": 1.369711583018293e-05,
-      "loss": 0.4847,
-      "step": 3467
-    },
-    {
-      "epoch": 6.16,
-      "learning_rate": 1.3693476803269799e-05,
-      "loss": 0.489,
-      "step": 3468
-    },
-    {
-      "epoch": 6.16,
-      "learning_rate": 1.3689837209895944e-05,
-      "loss": 0.4881,
-      "step": 3469
-    },
-    {
-      "epoch": 6.16,
-      "learning_rate": 1.3686197050619564e-05,
-      "loss": 0.4937,
-      "step": 3470
-    },
-    {
-      "epoch": 6.16,
-      "learning_rate": 1.3682556325998947e-05,
-      "loss": 0.501,
-      "step": 3471
-    },
-    {
-      "epoch": 6.16,
-      "learning_rate": 1.367891503659246e-05,
-      "loss": 0.4707,
-      "step": 3472
-    },
-    {
-      "epoch": 6.17,
-      "learning_rate": 1.367527318295856e-05,
-      "loss": 0.4741,
-      "step": 3473
-    },
-    {
-      "epoch": 6.17,
-      "learning_rate": 1.3671630765655788e-05,
-      "loss": 0.4656,
-      "step": 3474
-    },
-    {
-      "epoch": 6.17,
-      "learning_rate": 1.3667987785242776e-05,
-      "loss": 0.4758,
-      "step": 3475
-    },
-    {
-      "epoch": 6.17,
-      "learning_rate": 1.3664344242278243e-05,
-      "loss": 0.4925,
-      "step": 3476
-    },
-    {
-      "epoch": 6.17,
-      "learning_rate": 1.3660700137320986e-05,
-      "loss": 0.4711,
-      "step": 3477
-    },
-    {
-      "epoch": 6.17,
-      "learning_rate": 1.3657055470929899e-05,
-      "loss": 0.4747,
-      "step": 3478
-    },
-    {
-      "epoch": 6.18,
-      "learning_rate": 1.3653410243663953e-05,
-      "loss": 0.4908,
-      "step": 3479
-    },
-    {
-      "epoch": 6.18,
-      "learning_rate": 1.3649764456082211e-05,
-      "loss": 0.5134,
-      "step": 3480
-    },
-    {
-      "epoch": 6.18,
-      "learning_rate": 1.3646118108743815e-05,
-      "loss": 0.4778,
-      "step": 3481
-    },
-    {
-      "epoch": 6.18,
-      "learning_rate": 1.3642471202208002e-05,
-      "loss": 0.497,
-      "step": 3482
-    },
-    {
-      "epoch": 6.18,
-      "learning_rate": 1.3638823737034095e-05,
-      "loss": 0.4684,
-      "step": 3483
-    },
-    {
-      "epoch": 6.19,
-      "learning_rate": 1.363517571378149e-05,
-      "loss": 0.4713,
-      "step": 3484
-    },
-    {
-      "epoch": 6.19,
-      "learning_rate": 1.3631527133009677e-05,
-      "loss": 0.5007,
-      "step": 3485
-    },
-    {
-      "epoch": 6.19,
-      "learning_rate": 1.3627877995278235e-05,
-      "loss": 0.484,
-      "step": 3486
-    },
-    {
-      "epoch": 6.19,
-      "learning_rate": 1.3624228301146824e-05,
-      "loss": 0.4791,
-      "step": 3487
-    },
-    {
-      "epoch": 6.19,
-      "learning_rate": 1.3620578051175189e-05,
-      "loss": 0.4733,
-      "step": 3488
-    },
-    {
-      "epoch": 6.19,
-      "learning_rate": 1.3616927245923157e-05,
-      "loss": 0.4636,
-      "step": 3489
-    },
-    {
-      "epoch": 6.2,
-      "learning_rate": 1.3613275885950654e-05,
-      "loss": 0.4932,
-      "step": 3490
-    },
-    {
-      "epoch": 6.2,
-      "learning_rate": 1.3609623971817672e-05,
-      "loss": 0.5081,
-      "step": 3491
-    },
-    {
-      "epoch": 6.2,
-      "learning_rate": 1.3605971504084302e-05,
-      "loss": 0.5022,
-      "step": 3492
-    },
-    {
-      "epoch": 6.2,
-      "learning_rate": 1.360231848331071e-05,
-      "loss": 0.4957,
-      "step": 3493
-    },
-    {
-      "epoch": 6.2,
-      "learning_rate": 1.3598664910057163e-05,
-      "loss": 0.4685,
-      "step": 3494
-    },
-    {
-      "epoch": 6.21,
-      "learning_rate": 1.359501078488399e-05,
-      "loss": 0.4914,
-      "step": 3495
-    },
-    {
-      "epoch": 6.21,
-      "learning_rate": 1.3591356108351622e-05,
-      "loss": 0.4578,
-      "step": 3496
-    },
-    {
-      "epoch": 6.21,
-      "learning_rate": 1.3587700881020572e-05,
-      "loss": 0.4963,
-      "step": 3497
-    },
-    {
-      "epoch": 6.21,
-      "learning_rate": 1.358404510345143e-05,
-      "loss": 0.4962,
-      "step": 3498
-    },
-    {
-      "epoch": 6.21,
-      "learning_rate": 1.358038877620487e-05,
-      "loss": 0.4955,
-      "step": 3499
-    },
-    {
-      "epoch": 6.21,
-      "learning_rate": 1.3576731899841668e-05,
-      "loss": 0.4687,
-      "step": 3500
-    },
-    {
-      "epoch": 6.22,
-      "learning_rate": 1.357307447492266e-05,
-      "loss": 0.4879,
-      "step": 3501
-    },
-    {
-      "epoch": 6.22,
-      "learning_rate": 1.3569416502008785e-05,
-      "loss": 0.451,
-      "step": 3502
-    },
-    {
-      "epoch": 6.22,
-      "learning_rate": 1.3565757981661054e-05,
-      "loss": 0.4767,
-      "step": 3503
-    },
-    {
-      "epoch": 6.22,
-      "learning_rate": 1.3562098914440567e-05,
-      "loss": 0.4807,
-      "step": 3504
-    },
-    {
-      "epoch": 6.22,
-      "learning_rate": 1.3558439300908511e-05,
-      "loss": 0.4624,
-      "step": 3505
-    },
-    {
-      "epoch": 6.22,
-      "learning_rate": 1.3554779141626149e-05,
-      "loss": 0.4806,
-      "step": 3506
-    },
-    {
-      "epoch": 6.23,
-      "learning_rate": 1.3551118437154833e-05,
-      "loss": 0.4844,
-      "step": 3507
-    },
-    {
-      "epoch": 6.23,
-      "learning_rate": 1.3547457188055996e-05,
-      "loss": 0.4821,
-      "step": 3508
-    },
-    {
-      "epoch": 6.23,
-      "learning_rate": 1.3543795394891162e-05,
-      "loss": 0.5036,
-      "step": 3509
-    },
-    {
-      "epoch": 6.23,
-      "learning_rate": 1.3540133058221927e-05,
-      "loss": 0.4598,
-      "step": 3510
-    },
-    {
-      "epoch": 6.23,
-      "learning_rate": 1.3536470178609975e-05,
-      "loss": 0.4878,
-      "step": 3511
-    },
-    {
-      "epoch": 6.24,
-      "learning_rate": 1.3532806756617078e-05,
-      "loss": 0.4681,
-      "step": 3512
-    },
-    {
-      "epoch": 6.24,
-      "learning_rate": 1.3529142792805087e-05,
-      "loss": 0.4718,
-      "step": 3513
-    },
-    {
-      "epoch": 6.24,
-      "learning_rate": 1.3525478287735934e-05,
-      "loss": 0.4812,
-      "step": 3514
-    },
-    {
-      "epoch": 6.24,
-      "learning_rate": 1.3521813241971635e-05,
-      "loss": 0.4828,
-      "step": 3515
-    },
-    {
-      "epoch": 6.24,
-      "learning_rate": 1.3518147656074294e-05,
-      "loss": 0.4973,
-      "step": 3516
-    },
-    {
-      "epoch": 6.24,
-      "learning_rate": 1.3514481530606094e-05,
-      "loss": 0.4768,
-      "step": 3517
-    },
-    {
-      "epoch": 6.25,
-      "learning_rate": 1.3510814866129298e-05,
-      "loss": 0.4815,
-      "step": 3518
-    },
-    {
-      "epoch": 6.25,
-      "learning_rate": 1.3507147663206257e-05,
-      "loss": 0.4757,
-      "step": 3519
-    },
-    {
-      "epoch": 6.25,
-      "learning_rate": 1.3503479922399402e-05,
-      "loss": 0.4803,
-      "step": 3520
-    },
-    {
-      "epoch": 6.25,
-      "learning_rate": 1.3499811644271248e-05,
-      "loss": 0.4926,
-      "step": 3521
-    },
-    {
-      "epoch": 6.25,
-      "learning_rate": 1.3496142829384387e-05,
-      "loss": 0.4806,
-      "step": 3522
-    },
-    {
-      "epoch": 6.25,
-      "learning_rate": 1.3492473478301497e-05,
-      "loss": 0.4906,
-      "step": 3523
-    },
-    {
-      "epoch": 6.26,
-      "learning_rate": 1.3488803591585346e-05,
-      "loss": 0.4765,
-      "step": 3524
-    },
-    {
-      "epoch": 6.26,
-      "learning_rate": 1.348513316979877e-05,
-      "loss": 0.4816,
-      "step": 3525
-    },
-    {
-      "epoch": 6.26,
-      "learning_rate": 1.3481462213504694e-05,
-      "loss": 0.4699,
-      "step": 3526
-    },
-    {
-      "epoch": 6.26,
-      "learning_rate": 1.3477790723266129e-05,
-      "loss": 0.5049,
-      "step": 3527
-    },
-    {
-      "epoch": 6.26,
-      "learning_rate": 1.347411869964616e-05,
-      "loss": 0.4819,
-      "step": 3528
-    },
-    {
-      "epoch": 6.27,
-      "learning_rate": 1.3470446143207958e-05,
-      "loss": 0.464,
-      "step": 3529
-    },
-    {
-      "epoch": 6.27,
-      "learning_rate": 1.3466773054514777e-05,
-      "loss": 0.5099,
-      "step": 3530
-    },
-    {
-      "epoch": 6.27,
-      "learning_rate": 1.346309943412995e-05,
-      "loss": 0.4729,
-      "step": 3531
-    },
-    {
-      "epoch": 6.27,
-      "learning_rate": 1.3459425282616894e-05,
-      "loss": 0.48,
-      "step": 3532
-    },
-    {
-      "epoch": 6.27,
-      "learning_rate": 1.34557506005391e-05,
-      "loss": 0.4703,
-      "step": 3533
-    },
-    {
-      "epoch": 6.27,
-      "learning_rate": 1.3452075388460151e-05,
-      "loss": 0.4622,
-      "step": 3534
-    },
-    {
-      "epoch": 6.28,
-      "learning_rate": 1.3448399646943706e-05,
-      "loss": 0.4498,
-      "step": 3535
-    },
-    {
-      "epoch": 6.28,
-      "learning_rate": 1.3444723376553506e-05,
-      "loss": 0.4728,
-      "step": 3536
-    },
-    {
-      "epoch": 6.28,
-      "learning_rate": 1.3441046577853371e-05,
-      "loss": 0.5032,
-      "step": 3537
-    },
-    {
-      "epoch": 6.28,
-      "learning_rate": 1.3437369251407207e-05,
-      "loss": 0.5203,
-      "step": 3538
-    },
-    {
-      "epoch": 6.28,
-      "learning_rate": 1.3433691397778993e-05,
-      "loss": 0.5014,
-      "step": 3539
-    },
-    {
-      "epoch": 6.28,
-      "learning_rate": 1.3430013017532798e-05,
-      "loss": 0.4679,
-      "step": 3540
-    },
-    {
-      "epoch": 6.29,
-      "learning_rate": 1.3426334111232767e-05,
-      "loss": 0.4922,
-      "step": 3541
-    },
-    {
-      "epoch": 6.29,
-      "learning_rate": 1.3422654679443123e-05,
-      "loss": 0.4926,
-      "step": 3542
-    },
-    {
-      "epoch": 6.29,
-      "learning_rate": 1.3418974722728177e-05,
-      "loss": 0.4797,
-      "step": 3543
-    },
-    {
-      "epoch": 6.29,
-      "learning_rate": 1.3415294241652312e-05,
-      "loss": 0.5044,
-      "step": 3544
-    },
-    {
-      "epoch": 6.29,
-      "learning_rate": 1.3411613236779996e-05,
-      "loss": 0.4895,
-      "step": 3545
-    },
-    {
-      "epoch": 6.3,
-      "learning_rate": 1.340793170867578e-05,
-      "loss": 0.4702,
-      "step": 3546
-    },
-    {
-      "epoch": 6.3,
-      "learning_rate": 1.3404249657904292e-05,
-      "loss": 0.5006,
-      "step": 3547
-    },
-    {
-      "epoch": 6.3,
-      "learning_rate": 1.3400567085030236e-05,
-      "loss": 0.4598,
-      "step": 3548
-    },
-    {
-      "epoch": 6.3,
-      "learning_rate": 1.3396883990618404e-05,
-      "loss": 0.5019,
-      "step": 3549
-    },
-    {
-      "epoch": 6.3,
-      "learning_rate": 1.3393200375233662e-05,
-      "loss": 0.4836,
-      "step": 3550
-    },
-    {
-      "epoch": 6.3,
-      "learning_rate": 1.3389516239440962e-05,
-      "loss": 0.473,
-      "step": 3551
-    },
-    {
-      "epoch": 6.31,
-      "learning_rate": 1.3385831583805329e-05,
-      "loss": 0.4929,
-      "step": 3552
-    },
-    {
-      "epoch": 6.31,
-      "learning_rate": 1.3382146408891871e-05,
-      "loss": 0.4931,
-      "step": 3553
-    },
-    {
-      "epoch": 6.31,
-      "learning_rate": 1.337846071526578e-05,
-      "loss": 0.4615,
-      "step": 3554
-    },
-    {
-      "epoch": 6.31,
-      "learning_rate": 1.3374774503492316e-05,
-      "loss": 0.4925,
-      "step": 3555
-    },
-    {
-      "epoch": 6.31,
-      "learning_rate": 1.3371087774136829e-05,
-      "loss": 0.499,
-      "step": 3556
-    },
-    {
-      "epoch": 6.32,
-      "learning_rate": 1.3367400527764747e-05,
-      "loss": 0.5012,
-      "step": 3557
-    },
-    {
-      "epoch": 6.32,
-      "learning_rate": 1.3363712764941568e-05,
-      "loss": 0.4797,
-      "step": 3558
-    },
-    {
-      "epoch": 6.32,
-      "learning_rate": 1.3360024486232885e-05,
-      "loss": 0.4801,
-      "step": 3559
-    },
-    {
-      "epoch": 6.32,
-      "learning_rate": 1.335633569220436e-05,
-      "loss": 0.4755,
-      "step": 3560
-    },
-    {
-      "epoch": 6.32,
-      "learning_rate": 1.3352646383421728e-05,
-      "loss": 0.4686,
-      "step": 3561
-    },
-    {
-      "epoch": 6.32,
-      "learning_rate": 1.3348956560450818e-05,
-      "loss": 0.4834,
-      "step": 3562
-    },
-    {
-      "epoch": 6.33,
-      "learning_rate": 1.3345266223857528e-05,
-      "loss": 0.5071,
-      "step": 3563
-    },
-    {
-      "epoch": 6.33,
-      "learning_rate": 1.3341575374207837e-05,
-      "loss": 0.4671,
-      "step": 3564
-    },
-    {
-      "epoch": 6.33,
-      "learning_rate": 1.3337884012067803e-05,
-      "loss": 0.4697,
-      "step": 3565
-    },
-    {
-      "epoch": 6.33,
-      "learning_rate": 1.3334192138003563e-05,
-      "loss": 0.4641,
-      "step": 3566
-    },
-    {
-      "epoch": 6.33,
-      "learning_rate": 1.3330499752581331e-05,
-      "loss": 0.4727,
-      "step": 3567
-    },
-    {
-      "epoch": 6.33,
-      "learning_rate": 1.33268068563674e-05,
-      "loss": 0.4659,
-      "step": 3568
-    },
-    {
-      "epoch": 6.34,
-      "learning_rate": 1.3323113449928142e-05,
-      "loss": 0.4538,
-      "step": 3569
-    },
-    {
-      "epoch": 6.34,
-      "learning_rate": 1.331941953383001e-05,
-      "loss": 0.4859,
-      "step": 3570
-    },
-    {
-      "epoch": 6.34,
-      "learning_rate": 1.3315725108639526e-05,
-      "loss": 0.4718,
-      "step": 3571
-    },
-    {
-      "epoch": 6.34,
-      "learning_rate": 1.33120301749233e-05,
-      "loss": 0.5018,
-      "step": 3572
-    },
-    {
-      "epoch": 6.34,
-      "learning_rate": 1.3308334733248019e-05,
-      "loss": 0.4834,
-      "step": 3573
-    },
-    {
-      "epoch": 6.35,
-      "learning_rate": 1.330463878418044e-05,
-      "loss": 0.4704,
-      "step": 3574
-    },
-    {
-      "epoch": 6.35,
-      "learning_rate": 1.3300942328287404e-05,
-      "loss": 0.4685,
-      "step": 3575
-    },
-    {
-      "epoch": 6.35,
-      "learning_rate": 1.3297245366135828e-05,
-      "loss": 0.4751,
-      "step": 3576
-    },
-    {
-      "epoch": 6.35,
-      "learning_rate": 1.329354789829271e-05,
-      "loss": 0.4818,
-      "step": 3577
-    },
-    {
-      "epoch": 6.35,
-      "learning_rate": 1.328984992532512e-05,
-      "loss": 0.4929,
-      "step": 3578
-    },
-    {
-      "epoch": 6.35,
-      "learning_rate": 1.3286151447800211e-05,
-      "loss": 0.482,
-      "step": 3579
-    },
-    {
-      "epoch": 6.36,
-      "learning_rate": 1.328245246628521e-05,
-      "loss": 0.4635,
-      "step": 3580
-    },
-    {
-      "epoch": 6.36,
-      "learning_rate": 1.327875298134742e-05,
-      "loss": 0.5076,
-      "step": 3581
-    },
-    {
-      "epoch": 6.36,
-      "learning_rate": 1.3275052993554226e-05,
-      "loss": 0.4842,
-      "step": 3582
-    },
-    {
-      "epoch": 6.36,
-      "learning_rate": 1.3271352503473082e-05,
-      "loss": 0.4783,
-      "step": 3583
-    },
-    {
-      "epoch": 6.36,
-      "learning_rate": 1.3267651511671534e-05,
-      "loss": 0.4707,
-      "step": 3584
-    },
-    {
-      "epoch": 6.36,
-      "learning_rate": 1.3263950018717184e-05,
-      "loss": 0.4779,
-      "step": 3585
-    },
-    {
-      "epoch": 6.37,
-      "learning_rate": 1.3260248025177731e-05,
-      "loss": 0.4917,
-      "step": 3586
-    },
-    {
-      "epoch": 6.37,
-      "learning_rate": 1.3256545531620936e-05,
-      "loss": 0.4873,
-      "step": 3587
-    },
-    {
-      "epoch": 6.37,
-      "learning_rate": 1.3252842538614646e-05,
-      "loss": 0.4965,
-      "step": 3588
-    },
-    {
-      "epoch": 6.37,
-      "learning_rate": 1.3249139046726779e-05,
-      "loss": 0.483,
-      "step": 3589
-    },
-    {
-      "epoch": 6.37,
-      "learning_rate": 1.3245435056525335e-05,
-      "loss": 0.4921,
-      "step": 3590
-    },
-    {
-      "epoch": 6.38,
-      "learning_rate": 1.3241730568578383e-05,
-      "loss": 0.4595,
-      "step": 3591
-    },
-    {
-      "epoch": 6.38,
-      "learning_rate": 1.3238025583454074e-05,
-      "loss": 0.4692,
-      "step": 3592
-    },
-    {
-      "epoch": 6.38,
-      "learning_rate": 1.3234320101720637e-05,
-      "loss": 0.4784,
-      "step": 3593
-    },
-    {
-      "epoch": 6.38,
-      "learning_rate": 1.323061412394637e-05,
-      "loss": 0.4784,
-      "step": 3594
-    },
-    {
-      "epoch": 6.38,
-      "learning_rate": 1.3226907650699653e-05,
-      "loss": 0.4888,
-      "step": 3595
-    },
-    {
-      "epoch": 6.38,
-      "learning_rate": 1.3223200682548938e-05,
-      "loss": 0.4775,
-      "step": 3596
-    },
-    {
-      "epoch": 6.39,
-      "learning_rate": 1.3219493220062756e-05,
-      "loss": 0.4793,
-      "step": 3597
-    },
-    {
-      "epoch": 6.39,
-      "learning_rate": 1.3215785263809711e-05,
-      "loss": 0.4711,
-      "step": 3598
-    },
-    {
-      "epoch": 6.39,
-      "learning_rate": 1.3212076814358488e-05,
-      "loss": 0.4884,
-      "step": 3599
-    },
-    {
-      "epoch": 6.39,
-      "learning_rate": 1.3208367872277843e-05,
-      "loss": 0.4647,
-      "step": 3600
-    },
-    {
-      "epoch": 6.39,
-      "learning_rate": 1.3204658438136604e-05,
-      "loss": 0.4915,
-      "step": 3601
-    },
-    {
-      "epoch": 6.4,
-      "learning_rate": 1.3200948512503686e-05,
-      "loss": 0.4865,
-      "step": 3602
-    },
-    {
-      "epoch": 6.4,
-      "learning_rate": 1.3197238095948066e-05,
-      "loss": 0.465,
-      "step": 3603
-    },
-    {
-      "epoch": 6.4,
-      "learning_rate": 1.3193527189038811e-05,
-      "loss": 0.5063,
-      "step": 3604
-    },
-    {
-      "epoch": 6.4,
-      "learning_rate": 1.3189815792345044e-05,
-      "loss": 0.4951,
-      "step": 3605
-    },
-    {
-      "epoch": 6.4,
-      "learning_rate": 1.3186103906435983e-05,
-      "loss": 0.5008,
-      "step": 3606
-    },
-    {
-      "epoch": 6.4,
-      "learning_rate": 1.3182391531880908e-05,
-      "loss": 0.4764,
-      "step": 3607
-    },
-    {
-      "epoch": 6.41,
-      "learning_rate": 1.3178678669249181e-05,
-      "loss": 0.4569,
-      "step": 3608
-    },
-    {
-      "epoch": 6.41,
-      "learning_rate": 1.3174965319110231e-05,
-      "loss": 0.4738,
-      "step": 3609
-    },
-    {
-      "epoch": 6.41,
-      "learning_rate": 1.3171251482033572e-05,
-      "loss": 0.4996,
-      "step": 3610
-    },
-    {
-      "epoch": 6.41,
-      "learning_rate": 1.3167537158588786e-05,
-      "loss": 0.4716,
-      "step": 3611
-    },
-    {
-      "epoch": 6.41,
-      "learning_rate": 1.3163822349345525e-05,
-      "loss": 0.487,
-      "step": 3612
-    },
-    {
-      "epoch": 6.41,
-      "learning_rate": 1.3160107054873532e-05,
-      "loss": 0.4888,
-      "step": 3613
-    },
-    {
-      "epoch": 6.42,
-      "learning_rate": 1.3156391275742605e-05,
-      "loss": 0.471,
-      "step": 3614
-    },
-    {
-      "epoch": 6.42,
-      "learning_rate": 1.3152675012522629e-05,
-      "loss": 0.4708,
-      "step": 3615
-    },
-    {
-      "epoch": 6.42,
-      "learning_rate": 1.3148958265783556e-05,
-      "loss": 0.4894,
-      "step": 3616
-    },
-    {
-      "epoch": 6.42,
-      "learning_rate": 1.314524103609542e-05,
-      "loss": 0.4724,
-      "step": 3617
-    },
-    {
-      "epoch": 6.42,
-      "learning_rate": 1.3141523324028323e-05,
-      "loss": 0.4881,
-      "step": 3618
-    },
-    {
-      "epoch": 6.43,
-      "learning_rate": 1.3137805130152443e-05,
-      "loss": 0.4624,
-      "step": 3619
-    },
-    {
-      "epoch": 6.43,
-      "learning_rate": 1.313408645503803e-05,
-      "loss": 0.4913,
-      "step": 3620
-    },
-    {
-      "epoch": 6.43,
-      "learning_rate": 1.313036729925541e-05,
-      "loss": 0.4576,
-      "step": 3621
-    },
-    {
-      "epoch": 6.43,
-      "learning_rate": 1.3126647663374985e-05,
-      "loss": 0.4618,
-      "step": 3622
-    },
-    {
-      "epoch": 6.43,
-      "learning_rate": 1.3122927547967223e-05,
-      "loss": 0.4855,
-      "step": 3623
-    },
-    {
-      "epoch": 6.43,
-      "learning_rate": 1.3119206953602675e-05,
-      "loss": 0.4959,
-      "step": 3624
-    },
-    {
-      "epoch": 6.44,
-      "learning_rate": 1.3115485880851955e-05,
-      "loss": 0.502,
-      "step": 3625
-    },
-    {
-      "epoch": 6.44,
-      "learning_rate": 1.311176433028576e-05,
-      "loss": 0.4781,
-      "step": 3626
-    },
-    {
-      "epoch": 6.44,
-      "learning_rate": 1.3108042302474858e-05,
-      "loss": 0.4826,
-      "step": 3627
-    },
-    {
-      "epoch": 6.44,
-      "learning_rate": 1.3104319797990081e-05,
-      "loss": 0.4893,
-      "step": 3628
-    },
-    {
-      "epoch": 6.44,
-      "learning_rate": 1.310059681740235e-05,
-      "loss": 0.4718,
-      "step": 3629
-    },
-    {
-      "epoch": 6.44,
-      "learning_rate": 1.3096873361282647e-05,
-      "loss": 0.4815,
-      "step": 3630
-    },
-    {
-      "epoch": 6.45,
-      "learning_rate": 1.3093149430202031e-05,
-      "loss": 0.4939,
-      "step": 3631
-    },
-    {
-      "epoch": 6.45,
-      "learning_rate": 1.308942502473163e-05,
-      "loss": 0.491,
-      "step": 3632
-    },
-    {
-      "epoch": 6.45,
-      "learning_rate": 1.3085700145442653e-05,
-      "loss": 0.5043,
-      "step": 3633
-    },
-    {
-      "epoch": 6.45,
-      "learning_rate": 1.3081974792906378e-05,
-      "loss": 0.4809,
-      "step": 3634
-    },
-    {
-      "epoch": 6.45,
-      "learning_rate": 1.307824896769415e-05,
-      "loss": 0.4882,
-      "step": 3635
-    },
-    {
-      "epoch": 6.46,
-      "learning_rate": 1.3074522670377392e-05,
-      "loss": 0.4695,
-      "step": 3636
-    },
-    {
-      "epoch": 6.46,
-      "learning_rate": 1.3070795901527598e-05,
-      "loss": 0.5131,
-      "step": 3637
-    },
-    {
-      "epoch": 6.46,
-      "learning_rate": 1.3067068661716338e-05,
-      "loss": 0.4879,
-      "step": 3638
-    },
-    {
-      "epoch": 6.46,
-      "learning_rate": 1.3063340951515246e-05,
-      "loss": 0.4868,
-      "step": 3639
-    },
-    {
-      "epoch": 6.46,
-      "learning_rate": 1.3059612771496042e-05,
-      "loss": 0.4685,
-      "step": 3640
-    },
-    {
-      "epoch": 6.46,
-      "learning_rate": 1.3055884122230496e-05,
-      "loss": 0.4959,
-      "step": 3641
-    },
-    {
-      "epoch": 6.47,
-      "learning_rate": 1.3052155004290471e-05,
-      "loss": 0.4747,
-      "step": 3642
-    },
-    {
-      "epoch": 6.47,
-      "learning_rate": 1.3048425418247893e-05,
-      "loss": 0.4957,
-      "step": 3643
-    },
-    {
-      "epoch": 6.47,
-      "learning_rate": 1.3044695364674762e-05,
-      "loss": 0.4807,
-      "step": 3644
-    },
-    {
-      "epoch": 6.47,
-      "learning_rate": 1.3040964844143144e-05,
-      "loss": 0.481,
-      "step": 3645
-    },
-    {
-      "epoch": 6.47,
-      "learning_rate": 1.3037233857225186e-05,
-      "loss": 0.4846,
-      "step": 3646
-    },
-    {
-      "epoch": 6.47,
-      "learning_rate": 1.30335024044931e-05,
-      "loss": 0.4678,
-      "step": 3647
-    },
-    {
-      "epoch": 6.48,
-      "learning_rate": 1.3029770486519167e-05,
-      "loss": 0.4913,
-      "step": 3648
-    },
-    {
-      "epoch": 6.48,
-      "learning_rate": 1.302603810387575e-05,
-      "loss": 0.466,
-      "step": 3649
-    },
-    {
-      "epoch": 6.48,
-      "learning_rate": 1.302230525713527e-05,
-      "loss": 0.4808,
-      "step": 3650
-    },
-    {
-      "epoch": 6.48,
-      "learning_rate": 1.301857194687023e-05,
-      "loss": 0.4718,
-      "step": 3651
-    },
-    {
-      "epoch": 6.48,
-      "learning_rate": 1.30148381736532e-05,
-      "loss": 0.4799,
-      "step": 3652
-    },
-    {
-      "epoch": 6.49,
-      "learning_rate": 1.3011103938056822e-05,
-      "loss": 0.4843,
-      "step": 3653
-    },
-    {
-      "epoch": 6.49,
-      "learning_rate": 1.3007369240653802e-05,
-      "loss": 0.478,
-      "step": 3654
-    },
-    {
-      "epoch": 6.49,
-      "learning_rate": 1.3003634082016926e-05,
-      "loss": 0.4737,
-      "step": 3655
-    },
-    {
-      "epoch": 6.49,
-      "learning_rate": 1.2999898462719051e-05,
-      "loss": 0.497,
-      "step": 3656
-    },
-    {
-      "epoch": 6.49,
-      "learning_rate": 1.2996162383333097e-05,
-      "loss": 0.4528,
-      "step": 3657
-    },
-    {
-      "epoch": 6.49,
-      "learning_rate": 1.299242584443206e-05,
-      "loss": 0.5017,
-      "step": 3658
-    },
-    {
-      "epoch": 6.5,
-      "learning_rate": 1.2988688846589e-05,
-      "loss": 0.4752,
-      "step": 3659
-    },
-    {
-      "epoch": 6.5,
-      "learning_rate": 1.2984951390377061e-05,
-      "loss": 0.477,
-      "step": 3660
-    },
-    {
-      "epoch": 6.5,
-      "learning_rate": 1.2981213476369445e-05,
-      "loss": 0.4944,
-      "step": 3661
-    },
-    {
-      "epoch": 6.5,
-      "learning_rate": 1.2977475105139425e-05,
-      "loss": 0.4746,
-      "step": 3662
-    },
-    {
-      "epoch": 6.5,
-      "learning_rate": 1.2973736277260351e-05,
-      "loss": 0.4633,
-      "step": 3663
-    },
-    {
-      "epoch": 6.51,
-      "learning_rate": 1.296999699330564e-05,
-      "loss": 0.5061,
-      "step": 3664
-    },
-    {
-      "epoch": 6.51,
-      "learning_rate": 1.2966257253848773e-05,
-      "loss": 0.4739,
-      "step": 3665
-    },
-    {
-      "epoch": 6.51,
-      "learning_rate": 1.296251705946331e-05,
-      "loss": 0.4968,
-      "step": 3666
-    },
-    {
-      "epoch": 6.51,
-      "learning_rate": 1.2958776410722878e-05,
-      "loss": 0.4731,
-      "step": 3667
-    },
-    {
-      "epoch": 6.51,
-      "learning_rate": 1.2955035308201165e-05,
-      "loss": 0.4906,
-      "step": 3668
-    },
-    {
-      "epoch": 6.51,
-      "learning_rate": 1.295129375247194e-05,
-      "loss": 0.4966,
-      "step": 3669
-    },
-    {
-      "epoch": 6.52,
-      "learning_rate": 1.2947551744109044e-05,
-      "loss": 0.4816,
-      "step": 3670
-    },
-    {
-      "epoch": 6.52,
-      "learning_rate": 1.2943809283686373e-05,
-      "loss": 0.4816,
-      "step": 3671
-    },
-    {
-      "epoch": 6.52,
-      "learning_rate": 1.2940066371777902e-05,
-      "loss": 0.4802,
-      "step": 3672
-    },
-    {
-      "epoch": 6.52,
-      "learning_rate": 1.2936323008957675e-05,
-      "loss": 0.4837,
-      "step": 3673
-    },
-    {
-      "epoch": 6.52,
-      "learning_rate": 1.2932579195799799e-05,
-      "loss": 0.4756,
-      "step": 3674
-    },
-    {
-      "epoch": 6.52,
-      "learning_rate": 1.292883493287846e-05,
-      "loss": 0.4732,
-      "step": 3675
-    },
-    {
-      "epoch": 6.53,
-      "learning_rate": 1.2925090220767908e-05,
-      "loss": 0.4685,
-      "step": 3676
-    },
-    {
-      "epoch": 6.53,
-      "learning_rate": 1.2921345060042458e-05,
-      "loss": 0.5008,
-      "step": 3677
-    },
-    {
-      "epoch": 6.53,
-      "learning_rate": 1.2917599451276498e-05,
-      "loss": 0.4698,
-      "step": 3678
-    },
-    {
-      "epoch": 6.53,
-      "learning_rate": 1.2913853395044486e-05,
-      "loss": 0.4663,
-      "step": 3679
-    },
-    {
-      "epoch": 6.53,
-      "learning_rate": 1.2910106891920947e-05,
-      "loss": 0.5144,
-      "step": 3680
-    },
-    {
-      "epoch": 6.54,
-      "learning_rate": 1.290635994248047e-05,
-      "loss": 0.4773,
-      "step": 3681
-    },
-    {
-      "epoch": 6.54,
-      "learning_rate": 1.2902612547297722e-05,
-      "loss": 0.4872,
-      "step": 3682
-    },
-    {
-      "epoch": 6.54,
-      "learning_rate": 1.2898864706947431e-05,
-      "loss": 0.4644,
-      "step": 3683
-    },
-    {
-      "epoch": 6.54,
-      "learning_rate": 1.2895116422004393e-05,
-      "loss": 0.4763,
-      "step": 3684
-    },
-    {
-      "epoch": 6.54,
-      "learning_rate": 1.2891367693043477e-05,
-      "loss": 0.4787,
-      "step": 3685
-    },
-    {
-      "epoch": 6.54,
-      "learning_rate": 1.2887618520639617e-05,
-      "loss": 0.4654,
-      "step": 3686
-    },
-    {
-      "epoch": 6.55,
-      "learning_rate": 1.2883868905367818e-05,
-      "loss": 0.5021,
-      "step": 3687
-    },
-    {
-      "epoch": 6.55,
-      "learning_rate": 1.2880118847803147e-05,
-      "loss": 0.4602,
-      "step": 3688
-    },
-    {
-      "epoch": 6.55,
-      "learning_rate": 1.2876368348520744e-05,
-      "loss": 0.4802,
-      "step": 3689
-    },
-    {
-      "epoch": 6.55,
-      "learning_rate": 1.2872617408095813e-05,
-      "loss": 0.4898,
-      "step": 3690
-    },
-    {
-      "epoch": 6.55,
-      "learning_rate": 1.2868866027103635e-05,
-      "loss": 0.4736,
-      "step": 3691
-    },
-    {
-      "epoch": 6.55,
-      "learning_rate": 1.2865114206119542e-05,
-      "loss": 0.4769,
-      "step": 3692
-    },
-    {
-      "epoch": 6.56,
-      "learning_rate": 1.2861361945718947e-05,
-      "loss": 0.4963,
-      "step": 3693
-    },
-    {
-      "epoch": 6.56,
-      "learning_rate": 1.285760924647733e-05,
-      "loss": 0.4803,
-      "step": 3694
-    },
-    {
-      "epoch": 6.56,
-      "learning_rate": 1.2853856108970227e-05,
-      "loss": 0.4945,
-      "step": 3695
-    },
-    {
-      "epoch": 6.56,
-      "learning_rate": 1.2850102533773253e-05,
-      "loss": 0.4907,
-      "step": 3696
-    },
-    {
-      "epoch": 6.56,
-      "learning_rate": 1.2846348521462087e-05,
-      "loss": 0.5027,
-      "step": 3697
-    },
-    {
-      "epoch": 6.57,
-      "learning_rate": 1.2842594072612469e-05,
-      "loss": 0.4681,
-      "step": 3698
-    },
-    {
-      "epoch": 6.57,
-      "learning_rate": 1.2838839187800218e-05,
-      "loss": 0.4966,
-      "step": 3699
-    },
-    {
-      "epoch": 6.57,
-      "learning_rate": 1.2835083867601206e-05,
-      "loss": 0.4934,
-      "step": 3700
-    },
-    {
-      "epoch": 6.57,
-      "learning_rate": 1.2831328112591382e-05,
-      "loss": 0.4989,
-      "step": 3701
-    },
-    {
-      "epoch": 6.57,
-      "learning_rate": 1.282757192334676e-05,
-      "loss": 0.4802,
-      "step": 3702
-    },
-    {
-      "epoch": 6.57,
-      "learning_rate": 1.2823815300443413e-05,
-      "loss": 0.4604,
-      "step": 3703
-    },
-    {
-      "epoch": 6.58,
-      "learning_rate": 1.282005824445749e-05,
-      "loss": 0.4962,
-      "step": 3704
-    },
-    {
-      "epoch": 6.58,
-      "learning_rate": 1.2816300755965202e-05,
-      "loss": 0.494,
-      "step": 3705
-    },
-    {
-      "epoch": 6.58,
-      "learning_rate": 1.2812542835542833e-05,
-      "loss": 0.4841,
-      "step": 3706
-    },
-    {
-      "epoch": 6.58,
-      "learning_rate": 1.2808784483766716e-05,
-      "loss": 0.4746,
-      "step": 3707
-    },
-    {
-      "epoch": 6.58,
-      "learning_rate": 1.280502570121327e-05,
-      "loss": 0.4922,
-      "step": 3708
-    },
-    {
-      "epoch": 6.58,
-      "learning_rate": 1.2801266488458968e-05,
-      "loss": 0.4846,
-      "step": 3709
-    },
-    {
-      "epoch": 6.59,
-      "learning_rate": 1.2797506846080354e-05,
-      "loss": 0.4557,
-      "step": 3710
-    },
-    {
-      "epoch": 6.59,
-      "learning_rate": 1.2793746774654034e-05,
-      "loss": 0.4768,
-      "step": 3711
-    },
-    {
-      "epoch": 6.59,
-      "learning_rate": 1.2789986274756685e-05,
-      "loss": 0.4773,
-      "step": 3712
-    },
-    {
-      "epoch": 6.59,
-      "learning_rate": 1.2786225346965044e-05,
-      "loss": 0.4898,
-      "step": 3713
-    },
-    {
-      "epoch": 6.59,
-      "learning_rate": 1.278246399185592e-05,
-      "loss": 0.4889,
-      "step": 3714
-    },
-    {
-      "epoch": 6.6,
-      "learning_rate": 1.2778702210006184e-05,
-      "loss": 0.4648,
-      "step": 3715
-    },
-    {
-      "epoch": 6.6,
-      "learning_rate": 1.2774940001992768e-05,
-      "loss": 0.4898,
-      "step": 3716
-    },
-    {
-      "epoch": 6.6,
-      "learning_rate": 1.277117736839268e-05,
-      "loss": 0.4818,
-      "step": 3717
-    },
-    {
-      "epoch": 6.6,
-      "learning_rate": 1.2767414309782979e-05,
-      "loss": 0.4824,
-      "step": 3718
-    },
-    {
-      "epoch": 6.6,
-      "learning_rate": 1.2763650826740807e-05,
-      "loss": 0.479,
-      "step": 3719
-    },
-    {
-      "epoch": 6.6,
-      "learning_rate": 1.2759886919843354e-05,
-      "loss": 0.4895,
-      "step": 3720
-    },
-    {
-      "epoch": 6.61,
-      "learning_rate": 1.2756122589667888e-05,
-      "loss": 0.4802,
-      "step": 3721
-    },
-    {
-      "epoch": 6.61,
-      "learning_rate": 1.2752357836791732e-05,
-      "loss": 0.4865,
-      "step": 3722
-    },
-    {
-      "epoch": 6.61,
-      "learning_rate": 1.2748592661792278e-05,
-      "loss": 0.4995,
-      "step": 3723
-    },
-    {
-      "epoch": 6.61,
-      "learning_rate": 1.2744827065246988e-05,
-      "loss": 0.4691,
-      "step": 3724
-    },
-    {
-      "epoch": 6.61,
-      "learning_rate": 1.2741061047733377e-05,
-      "loss": 0.4819,
-      "step": 3725
-    },
-    {
-      "epoch": 6.62,
-      "learning_rate": 1.2737294609829035e-05,
-      "loss": 0.4701,
-      "step": 3726
-    },
-    {
-      "epoch": 6.62,
-      "learning_rate": 1.2733527752111611e-05,
-      "loss": 0.4691,
-      "step": 3727
-    },
-    {
-      "epoch": 6.62,
-      "learning_rate": 1.2729760475158824e-05,
-      "loss": 0.4877,
-      "step": 3728
-    },
-    {
-      "epoch": 6.62,
-      "learning_rate": 1.2725992779548451e-05,
-      "loss": 0.4908,
-      "step": 3729
-    },
-    {
-      "epoch": 6.62,
-      "learning_rate": 1.272222466585833e-05,
-      "loss": 0.4713,
-      "step": 3730
-    },
-    {
-      "epoch": 6.62,
-      "learning_rate": 1.2718456134666375e-05,
-      "loss": 0.4884,
-      "step": 3731
-    },
-    {
-      "epoch": 6.63,
-      "learning_rate": 1.2714687186550556e-05,
-      "loss": 0.4727,
-      "step": 3732
-    },
-    {
-      "epoch": 6.63,
-      "learning_rate": 1.271091782208891e-05,
-      "loss": 0.4796,
-      "step": 3733
-    },
-    {
-      "epoch": 6.63,
-      "learning_rate": 1.2707148041859533e-05,
-      "loss": 0.4946,
-      "step": 3734
-    },
-    {
-      "epoch": 6.63,
-      "learning_rate": 1.2703377846440591e-05,
-      "loss": 0.4934,
-      "step": 3735
-    },
-    {
-      "epoch": 6.63,
-      "learning_rate": 1.2699607236410315e-05,
-      "loss": 0.4628,
-      "step": 3736
-    },
-    {
-      "epoch": 6.63,
-      "learning_rate": 1.2695836212346984e-05,
-      "loss": 0.4874,
-      "step": 3737
-    },
-    {
-      "epoch": 6.64,
-      "learning_rate": 1.269206477482896e-05,
-      "loss": 0.4943,
-      "step": 3738
-    },
-    {
-      "epoch": 6.64,
-      "learning_rate": 1.268829292443466e-05,
-      "loss": 0.4833,
-      "step": 3739
-    },
-    {
-      "epoch": 6.64,
-      "learning_rate": 1.2684520661742564e-05,
-      "loss": 0.4932,
-      "step": 3740
-    },
-    {
-      "epoch": 6.64,
-      "learning_rate": 1.2680747987331215e-05,
-      "loss": 0.4745,
-      "step": 3741
-    },
-    {
-      "epoch": 6.64,
-      "learning_rate": 1.267697490177922e-05,
-      "loss": 0.471,
-      "step": 3742
-    },
-    {
-      "epoch": 6.65,
-      "learning_rate": 1.267320140566525e-05,
-      "loss": 0.4777,
-      "step": 3743
-    },
-    {
-      "epoch": 6.65,
-      "learning_rate": 1.2669427499568042e-05,
-      "loss": 0.4821,
-      "step": 3744
-    },
-    {
-      "epoch": 6.65,
-      "learning_rate": 1.2665653184066383e-05,
-      "loss": 0.4893,
-      "step": 3745
-    },
-    {
-      "epoch": 6.65,
-      "learning_rate": 1.2661878459739136e-05,
-      "loss": 0.4886,
-      "step": 3746
-    },
-    {
-      "epoch": 6.65,
-      "learning_rate": 1.2658103327165226e-05,
-      "loss": 0.5063,
-      "step": 3747
-    },
-    {
-      "epoch": 6.65,
-      "learning_rate": 1.2654327786923638e-05,
-      "loss": 0.4719,
-      "step": 3748
-    },
-    {
-      "epoch": 6.66,
-      "learning_rate": 1.2650551839593408e-05,
-      "loss": 0.4799,
-      "step": 3749
-    },
-    {
-      "epoch": 6.66,
-      "learning_rate": 1.2646775485753655e-05,
-      "loss": 0.4606,
-      "step": 3750
-    },
-    {
-      "epoch": 6.66,
-      "learning_rate": 1.2642998725983549e-05,
-      "loss": 0.4676,
-      "step": 3751
-    },
-    {
-      "epoch": 6.66,
-      "learning_rate": 1.263922156086232e-05,
-      "loss": 0.4833,
-      "step": 3752
-    },
-    {
-      "epoch": 6.66,
-      "learning_rate": 1.2635443990969266e-05,
-      "loss": 0.484,
-      "step": 3753
-    },
-    {
-      "epoch": 6.66,
-      "learning_rate": 1.2631666016883749e-05,
-      "loss": 0.4759,
-      "step": 3754
-    },
-    {
-      "epoch": 6.67,
-      "learning_rate": 1.262788763918518e-05,
-      "loss": 0.4853,
-      "step": 3755
-    },
-    {
-      "epoch": 6.67,
-      "learning_rate": 1.2624108858453045e-05,
-      "loss": 0.4683,
-      "step": 3756
-    },
-    {
-      "epoch": 6.67,
-      "learning_rate": 1.262032967526689e-05,
-      "loss": 0.4755,
-      "step": 3757
-    },
-    {
-      "epoch": 6.67,
-      "learning_rate": 1.2616550090206316e-05,
-      "loss": 0.4623,
-      "step": 3758
-    },
-    {
-      "epoch": 6.67,
-      "learning_rate": 1.2612770103850998e-05,
-      "loss": 0.4769,
-      "step": 3759
-    },
-    {
-      "epoch": 6.68,
-      "learning_rate": 1.2608989716780653e-05,
-      "loss": 0.4952,
-      "step": 3760
-    },
-    {
-      "epoch": 6.68,
-      "learning_rate": 1.2605208929575077e-05,
-      "loss": 0.494,
-      "step": 3761
-    },
-    {
-      "epoch": 6.68,
-      "learning_rate": 1.2601427742814123e-05,
-      "loss": 0.4953,
-      "step": 3762
-    },
-    {
-      "epoch": 6.68,
-      "learning_rate": 1.2597646157077703e-05,
-      "loss": 0.4989,
-      "step": 3763
-    },
-    {
-      "epoch": 6.68,
-      "learning_rate": 1.259386417294579e-05,
-      "loss": 0.4751,
-      "step": 3764
-    },
-    {
-      "epoch": 6.68,
-      "learning_rate": 1.2590081790998414e-05,
-      "loss": 0.4671,
-      "step": 3765
-    },
-    {
-      "epoch": 6.69,
-      "learning_rate": 1.2586299011815677e-05,
-      "loss": 0.4715,
-      "step": 3766
-    },
-    {
-      "epoch": 6.69,
-      "learning_rate": 1.2582515835977737e-05,
-      "loss": 0.4766,
-      "step": 3767
-    },
-    {
-      "epoch": 6.69,
-      "learning_rate": 1.2578732264064808e-05,
-      "loss": 0.4827,
-      "step": 3768
-    },
-    {
-      "epoch": 6.69,
-      "learning_rate": 1.2574948296657167e-05,
-      "loss": 0.4725,
-      "step": 3769
-    },
-    {
-      "epoch": 6.69,
-      "learning_rate": 1.257116393433516e-05,
-      "loss": 0.5088,
-      "step": 3770
-    },
-    {
-      "epoch": 6.7,
-      "learning_rate": 1.2567379177679178e-05,
-      "loss": 0.4678,
-      "step": 3771
-    },
-    {
-      "epoch": 6.7,
-      "learning_rate": 1.2563594027269685e-05,
-      "loss": 0.4513,
-      "step": 3772
-    },
-    {
-      "epoch": 6.7,
-      "learning_rate": 1.2559808483687204e-05,
-      "loss": 0.4724,
-      "step": 3773
-    },
-    {
-      "epoch": 6.7,
-      "learning_rate": 1.2556022547512312e-05,
-      "loss": 0.504,
-      "step": 3774
-    },
-    {
-      "epoch": 6.7,
-      "learning_rate": 1.2552236219325651e-05,
-      "loss": 0.5016,
-      "step": 3775
-    },
-    {
-      "epoch": 6.7,
-      "learning_rate": 1.2548449499707921e-05,
-      "loss": 0.467,
-      "step": 3776
-    },
-    {
-      "epoch": 6.71,
-      "learning_rate": 1.254466238923989e-05,
-      "loss": 0.4485,
-      "step": 3777
-    },
-    {
-      "epoch": 6.71,
-      "learning_rate": 1.254087488850237e-05,
-      "loss": 0.4558,
-      "step": 3778
-    },
-    {
-      "epoch": 6.71,
-      "learning_rate": 1.253708699807625e-05,
-      "loss": 0.5067,
-      "step": 3779
-    },
-    {
-      "epoch": 6.71,
-      "learning_rate": 1.253329871854246e-05,
-      "loss": 0.467,
-      "step": 3780
-    },
-    {
-      "epoch": 6.71,
-      "learning_rate": 1.2529510050482012e-05,
-      "loss": 0.4937,
-      "step": 3781
-    },
-    {
-      "epoch": 6.71,
-      "learning_rate": 1.2525720994475963e-05,
-      "loss": 0.4866,
-      "step": 3782
-    },
-    {
-      "epoch": 6.72,
-      "learning_rate": 1.2521931551105427e-05,
-      "loss": 0.4648,
-      "step": 3783
-    },
-    {
-      "epoch": 6.72,
-      "learning_rate": 1.2518141720951585e-05,
-      "loss": 0.4939,
-      "step": 3784
-    },
-    {
-      "epoch": 6.72,
-      "learning_rate": 1.2514351504595676e-05,
-      "loss": 0.4925,
-      "step": 3785
-    },
-    {
-      "epoch": 6.72,
-      "learning_rate": 1.2510560902619002e-05,
-      "loss": 0.4582,
-      "step": 3786
-    },
-    {
-      "epoch": 6.72,
-      "learning_rate": 1.2506769915602914e-05,
-      "loss": 0.5056,
-      "step": 3787
-    },
-    {
-      "epoch": 6.73,
-      "learning_rate": 1.2502978544128829e-05,
-      "loss": 0.4933,
-      "step": 3788
-    },
-    {
-      "epoch": 6.73,
-      "learning_rate": 1.2499186788778223e-05,
-      "loss": 0.4678,
-      "step": 3789
-    },
-    {
-      "epoch": 6.73,
-      "learning_rate": 1.2495394650132628e-05,
-      "loss": 0.4895,
-      "step": 3790
-    },
-    {
-      "epoch": 6.73,
-      "learning_rate": 1.2491602128773636e-05,
-      "loss": 0.4917,
-      "step": 3791
-    },
-    {
-      "epoch": 6.73,
-      "learning_rate": 1.24878092252829e-05,
-      "loss": 0.492,
-      "step": 3792
-    },
-    {
-      "epoch": 6.73,
-      "learning_rate": 1.2484015940242129e-05,
-      "loss": 0.4675,
-      "step": 3793
-    },
-    {
-      "epoch": 6.74,
-      "learning_rate": 1.248022227423309e-05,
-      "loss": 0.4629,
-      "step": 3794
-    },
-    {
-      "epoch": 6.74,
-      "learning_rate": 1.2476428227837607e-05,
-      "loss": 0.4645,
-      "step": 3795
-    },
-    {
-      "epoch": 6.74,
-      "learning_rate": 1.247263380163757e-05,
-      "loss": 0.5074,
-      "step": 3796
-    },
-    {
-      "epoch": 6.74,
-      "learning_rate": 1.2468838996214923e-05,
-      "loss": 0.4788,
-      "step": 3797
-    },
-    {
-      "epoch": 6.74,
-      "learning_rate": 1.246504381215166e-05,
-      "loss": 0.4738,
-      "step": 3798
-    },
-    {
-      "epoch": 6.74,
-      "learning_rate": 1.2461248250029845e-05,
-      "loss": 0.4802,
-      "step": 3799
-    },
-    {
-      "epoch": 6.75,
-      "learning_rate": 1.2457452310431594e-05,
-      "loss": 0.4687,
-      "step": 3800
-    },
-    {
-      "epoch": 6.75,
-      "learning_rate": 1.2453655993939088e-05,
-      "loss": 0.5073,
-      "step": 3801
-    },
-    {
-      "epoch": 6.75,
-      "learning_rate": 1.244985930113455e-05,
-      "loss": 0.4679,
-      "step": 3802
-    },
-    {
-      "epoch": 6.75,
-      "learning_rate": 1.2446062232600277e-05,
-      "loss": 0.4891,
-      "step": 3803
-    },
-    {
-      "epoch": 6.75,
-      "learning_rate": 1.244226478891862e-05,
-      "loss": 0.4854,
-      "step": 3804
-    },
-    {
-      "epoch": 6.76,
-      "learning_rate": 1.2438466970671978e-05,
-      "loss": 0.4903,
-      "step": 3805
-    },
-    {
-      "epoch": 6.76,
-      "learning_rate": 1.2434668778442818e-05,
-      "loss": 0.498,
-      "step": 3806
-    },
-    {
-      "epoch": 6.76,
-      "learning_rate": 1.2430870212813661e-05,
-      "loss": 0.4568,
-      "step": 3807
-    },
-    {
-      "epoch": 6.76,
-      "learning_rate": 1.2427071274367083e-05,
-      "loss": 0.4738,
-      "step": 3808
-    },
-    {
-      "epoch": 6.76,
-      "learning_rate": 1.242327196368572e-05,
-      "loss": 0.5051,
-      "step": 3809
-    },
-    {
-      "epoch": 6.76,
-      "learning_rate": 1.2419472281352265e-05,
-      "loss": 0.4822,
-      "step": 3810
-    },
-    {
-      "epoch": 6.77,
-      "learning_rate": 1.2415672227949467e-05,
-      "loss": 0.463,
-      "step": 3811
-    },
-    {
-      "epoch": 6.77,
-      "learning_rate": 1.2411871804060135e-05,
-      "loss": 0.4589,
-      "step": 3812
-    },
-    {
-      "epoch": 6.77,
-      "learning_rate": 1.2408071010267125e-05,
-      "loss": 0.4763,
-      "step": 3813
-    },
-    {
-      "epoch": 6.77,
-      "learning_rate": 1.240426984715336e-05,
-      "loss": 0.4805,
-      "step": 3814
-    },
-    {
-      "epoch": 6.77,
-      "learning_rate": 1.2400468315301816e-05,
-      "loss": 0.4887,
-      "step": 3815
-    },
-    {
-      "epoch": 6.77,
-      "learning_rate": 1.2396666415295532e-05,
-      "loss": 0.4731,
-      "step": 3816
-    },
-    {
-      "epoch": 6.78,
-      "learning_rate": 1.2392864147717589e-05,
-      "loss": 0.518,
-      "step": 3817
-    },
-    {
-      "epoch": 6.78,
-      "learning_rate": 1.2389061513151135e-05,
-      "loss": 0.4908,
-      "step": 3818
-    },
-    {
-      "epoch": 6.78,
-      "learning_rate": 1.2385258512179375e-05,
-      "loss": 0.4932,
-      "step": 3819
-    },
-    {
-      "epoch": 6.78,
-      "learning_rate": 1.2381455145385565e-05,
-      "loss": 0.4789,
-      "step": 3820
-    },
-    {
-      "epoch": 6.78,
-      "learning_rate": 1.2377651413353018e-05,
-      "loss": 0.4908,
-      "step": 3821
-    },
-    {
-      "epoch": 6.79,
-      "learning_rate": 1.2373847316665106e-05,
-      "loss": 0.4774,
-      "step": 3822
-    },
-    {
-      "epoch": 6.79,
-      "learning_rate": 1.2370042855905258e-05,
-      "loss": 0.482,
-      "step": 3823
-    },
-    {
-      "epoch": 6.79,
-      "learning_rate": 1.2366238031656953e-05,
-      "loss": 0.4729,
-      "step": 3824
-    },
-    {
-      "epoch": 6.79,
-      "learning_rate": 1.2362432844503725e-05,
-      "loss": 0.4812,
-      "step": 3825
-    },
-    {
-      "epoch": 6.79,
-      "learning_rate": 1.2358627295029173e-05,
-      "loss": 0.4904,
-      "step": 3826
-    },
-    {
-      "epoch": 6.79,
-      "learning_rate": 1.2354821383816946e-05,
-      "loss": 0.4804,
-      "step": 3827
-    },
-    {
-      "epoch": 6.8,
-      "learning_rate": 1.2351015111450745e-05,
-      "loss": 0.4748,
-      "step": 3828
-    },
-    {
-      "epoch": 6.8,
-      "learning_rate": 1.2347208478514334e-05,
-      "loss": 0.4562,
-      "step": 3829
-    },
-    {
-      "epoch": 6.8,
-      "learning_rate": 1.2343401485591524e-05,
-      "loss": 0.4802,
-      "step": 3830
-    },
-    {
-      "epoch": 6.8,
-      "learning_rate": 1.233959413326619e-05,
-      "loss": 0.4743,
-      "step": 3831
-    },
-    {
-      "epoch": 6.8,
-      "learning_rate": 1.2335786422122255e-05,
-      "loss": 0.4679,
-      "step": 3832
-    },
-    {
-      "epoch": 6.81,
-      "learning_rate": 1.23319783527437e-05,
-      "loss": 0.5024,
-      "step": 3833
-    },
-    {
-      "epoch": 6.81,
-      "learning_rate": 1.2328169925714561e-05,
-      "loss": 0.4892,
-      "step": 3834
-    },
-    {
-      "epoch": 6.81,
-      "learning_rate": 1.2324361141618925e-05,
-      "loss": 0.4712,
-      "step": 3835
-    },
-    {
-      "epoch": 6.81,
-      "learning_rate": 1.2320552001040945e-05,
-      "loss": 0.489,
-      "step": 3836
-    },
-    {
-      "epoch": 6.81,
-      "learning_rate": 1.2316742504564811e-05,
-      "loss": 0.4643,
-      "step": 3837
-    },
-    {
-      "epoch": 6.81,
-      "learning_rate": 1.2312932652774785e-05,
-      "loss": 0.4745,
-      "step": 3838
-    },
-    {
-      "epoch": 6.82,
-      "learning_rate": 1.2309122446255174e-05,
-      "loss": 0.4475,
-      "step": 3839
-    },
-    {
-      "epoch": 6.82,
-      "learning_rate": 1.2305311885590337e-05,
-      "loss": 0.4851,
-      "step": 3840
-    },
-    {
-      "epoch": 6.82,
-      "learning_rate": 1.2301500971364698e-05,
-      "loss": 0.4796,
-      "step": 3841
-    },
-    {
-      "epoch": 6.82,
-      "learning_rate": 1.2297689704162725e-05,
-      "loss": 0.5114,
-      "step": 3842
-    },
-    {
-      "epoch": 6.82,
-      "learning_rate": 1.2293878084568944e-05,
-      "loss": 0.4792,
-      "step": 3843
-    },
-    {
-      "epoch": 6.82,
-      "learning_rate": 1.2290066113167935e-05,
-      "loss": 0.4939,
-      "step": 3844
-    },
-    {
-      "epoch": 6.83,
-      "learning_rate": 1.2286253790544333e-05,
-      "loss": 0.4794,
-      "step": 3845
-    },
-    {
-      "epoch": 6.83,
-      "learning_rate": 1.2282441117282831e-05,
-      "loss": 0.4768,
-      "step": 3846
-    },
-    {
-      "epoch": 6.83,
-      "learning_rate": 1.227862809396816e-05,
-      "loss": 0.477,
-      "step": 3847
-    },
-    {
-      "epoch": 6.83,
-      "learning_rate": 1.2274814721185122e-05,
-      "loss": 0.4932,
-      "step": 3848
-    },
-    {
-      "epoch": 6.83,
-      "learning_rate": 1.2271000999518563e-05,
-      "loss": 0.4824,
-      "step": 3849
-    },
-    {
-      "epoch": 6.84,
-      "learning_rate": 1.2267186929553392e-05,
-      "loss": 0.4916,
-      "step": 3850
-    },
-    {
-      "epoch": 6.84,
-      "learning_rate": 1.2263372511874557e-05,
-      "loss": 0.4844,
-      "step": 3851
-    },
-    {
-      "epoch": 6.84,
-      "learning_rate": 1.2259557747067069e-05,
-      "loss": 0.4564,
-      "step": 3852
-    },
-    {
-      "epoch": 6.84,
-      "learning_rate": 1.2255742635715993e-05,
-      "loss": 0.4933,
-      "step": 3853
-    },
-    {
-      "epoch": 6.84,
-      "learning_rate": 1.2251927178406442e-05,
-      "loss": 0.4746,
-      "step": 3854
-    },
-    {
-      "epoch": 6.84,
-      "learning_rate": 1.2248111375723585e-05,
-      "loss": 0.4826,
-      "step": 3855
-    },
-    {
-      "epoch": 6.85,
-      "learning_rate": 1.2244295228252647e-05,
-      "loss": 0.4743,
-      "step": 3856
-    },
-    {
-      "epoch": 6.85,
-      "learning_rate": 1.22404787365789e-05,
-      "loss": 0.4907,
-      "step": 3857
-    },
-    {
-      "epoch": 6.85,
-      "learning_rate": 1.2236661901287668e-05,
-      "loss": 0.489,
-      "step": 3858
-    },
-    {
-      "epoch": 6.85,
-      "learning_rate": 1.2232844722964335e-05,
-      "loss": 0.4807,
-      "step": 3859
-    },
-    {
-      "epoch": 6.85,
-      "learning_rate": 1.222902720219433e-05,
-      "loss": 0.4901,
-      "step": 3860
-    },
-    {
-      "epoch": 6.85,
-      "learning_rate": 1.2225209339563144e-05,
-      "loss": 0.4812,
-      "step": 3861
-    },
-    {
-      "epoch": 6.86,
-      "learning_rate": 1.2221391135656311e-05,
-      "loss": 0.5047,
-      "step": 3862
-    },
-    {
-      "epoch": 6.86,
-      "learning_rate": 1.2217572591059421e-05,
-      "loss": 0.4914,
-      "step": 3863
-    },
-    {
-      "epoch": 6.86,
-      "learning_rate": 1.2213753706358116e-05,
-      "loss": 0.4825,
-      "step": 3864
-    },
-    {
-      "epoch": 6.86,
-      "learning_rate": 1.2209934482138089e-05,
-      "loss": 0.4866,
-      "step": 3865
-    },
-    {
-      "epoch": 6.86,
-      "learning_rate": 1.2206114918985087e-05,
-      "loss": 0.4778,
-      "step": 3866
-    },
-    {
-      "epoch": 6.87,
-      "learning_rate": 1.2202295017484911e-05,
-      "loss": 0.4891,
-      "step": 3867
-    },
-    {
-      "epoch": 6.87,
-      "learning_rate": 1.2198474778223407e-05,
-      "loss": 0.4711,
-      "step": 3868
-    },
-    {
-      "epoch": 6.87,
-      "learning_rate": 1.219465420178648e-05,
-      "loss": 0.4835,
-      "step": 3869
-    },
-    {
-      "epoch": 6.87,
-      "learning_rate": 1.2190833288760081e-05,
-      "loss": 0.4916,
-      "step": 3870
-    },
-    {
-      "epoch": 6.87,
-      "learning_rate": 1.218701203973022e-05,
-      "loss": 0.4928,
-      "step": 3871
-    },
-    {
-      "epoch": 6.87,
-      "learning_rate": 1.218319045528295e-05,
-      "loss": 0.4704,
-      "step": 3872
-    },
-    {
-      "epoch": 6.88,
-      "learning_rate": 1.2179368536004382e-05,
-      "loss": 0.4664,
-      "step": 3873
-    },
-    {
-      "epoch": 6.88,
-      "learning_rate": 1.2175546282480673e-05,
-      "loss": 0.475,
-      "step": 3874
-    },
-    {
-      "epoch": 6.88,
-      "learning_rate": 1.2171723695298032e-05,
-      "loss": 0.48,
-      "step": 3875
-    },
-    {
-      "epoch": 6.88,
-      "learning_rate": 1.2167900775042729e-05,
-      "loss": 0.4941,
-      "step": 3876
-    },
-    {
-      "epoch": 6.88,
-      "learning_rate": 1.2164077522301069e-05,
-      "loss": 0.4852,
-      "step": 3877
-    },
-    {
-      "epoch": 6.89,
-      "learning_rate": 1.2160253937659419e-05,
-      "loss": 0.5134,
-      "step": 3878
-    },
-    {
-      "epoch": 6.89,
-      "learning_rate": 1.2156430021704196e-05,
-      "loss": 0.4868,
-      "step": 3879
-    },
-    {
-      "epoch": 6.89,
-      "learning_rate": 1.2152605775021867e-05,
-      "loss": 0.4856,
-      "step": 3880
-    },
-    {
-      "epoch": 6.89,
-      "learning_rate": 1.2148781198198943e-05,
-      "loss": 0.485,
-      "step": 3881
-    },
-    {
-      "epoch": 6.89,
-      "learning_rate": 1.2144956291821994e-05,
-      "loss": 0.4604,
-      "step": 3882
-    },
-    {
-      "epoch": 6.89,
-      "learning_rate": 1.2141131056477637e-05,
-      "loss": 0.484,
-      "step": 3883
-    },
-    {
-      "epoch": 6.9,
-      "learning_rate": 1.2137305492752547e-05,
-      "loss": 0.5036,
-      "step": 3884
-    },
-    {
-      "epoch": 6.9,
-      "learning_rate": 1.2133479601233431e-05,
-      "loss": 0.462,
-      "step": 3885
-    },
-    {
-      "epoch": 6.9,
-      "learning_rate": 1.2129653382507067e-05,
-      "loss": 0.4666,
-      "step": 3886
-    },
-    {
-      "epoch": 6.9,
-      "learning_rate": 1.2125826837160268e-05,
-      "loss": 0.4552,
-      "step": 3887
-    },
-    {
-      "epoch": 6.9,
-      "learning_rate": 1.212199996577991e-05,
-      "loss": 0.4776,
-      "step": 3888
-    },
-    {
-      "epoch": 6.9,
-      "learning_rate": 1.2118172768952907e-05,
-      "loss": 0.4678,
-      "step": 3889
-    },
-    {
-      "epoch": 6.91,
-      "learning_rate": 1.211434524726623e-05,
-      "loss": 0.4651,
-      "step": 3890
-    },
-    {
-      "epoch": 6.91,
-      "learning_rate": 1.2110517401306896e-05,
-      "loss": 0.4748,
-      "step": 3891
-    },
-    {
-      "epoch": 6.91,
-      "learning_rate": 1.2106689231661975e-05,
-      "loss": 0.4821,
-      "step": 3892
-    },
-    {
-      "epoch": 6.91,
-      "learning_rate": 1.2102860738918585e-05,
-      "loss": 0.4866,
-      "step": 3893
-    },
-    {
-      "epoch": 6.91,
-      "learning_rate": 1.2099031923663893e-05,
-      "loss": 0.4689,
-      "step": 3894
-    },
-    {
-      "epoch": 6.92,
-      "learning_rate": 1.209520278648512e-05,
-      "loss": 0.4797,
-      "step": 3895
-    },
-    {
-      "epoch": 6.92,
-      "learning_rate": 1.209137332796953e-05,
-      "loss": 0.4629,
-      "step": 3896
-    },
-    {
-      "epoch": 6.92,
-      "learning_rate": 1.2087543548704435e-05,
-      "loss": 0.4919,
-      "step": 3897
-    },
-    {
-      "epoch": 6.92,
-      "learning_rate": 1.2083713449277205e-05,
-      "loss": 0.464,
-      "step": 3898
-    },
-    {
-      "epoch": 6.92,
-      "learning_rate": 1.2079883030275258e-05,
-      "loss": 0.4769,
-      "step": 3899
-    },
-    {
-      "epoch": 6.92,
-      "learning_rate": 1.2076052292286047e-05,
-      "loss": 0.495,
-      "step": 3900
-    },
-    {
-      "epoch": 6.93,
-      "learning_rate": 1.2072221235897092e-05,
-      "loss": 0.5055,
-      "step": 3901
-    },
-    {
-      "epoch": 6.93,
-      "learning_rate": 1.2068389861695951e-05,
-      "loss": 0.4766,
-      "step": 3902
-    },
-    {
-      "epoch": 6.93,
-      "learning_rate": 1.2064558170270236e-05,
-      "loss": 0.4917,
-      "step": 3903
-    },
-    {
-      "epoch": 6.93,
-      "learning_rate": 1.2060726162207601e-05,
-      "loss": 0.4938,
-      "step": 3904
-    },
-    {
-      "epoch": 6.93,
-      "learning_rate": 1.2056893838095758e-05,
-      "loss": 0.4854,
-      "step": 3905
-    },
-    {
-      "epoch": 6.93,
-      "learning_rate": 1.2053061198522459e-05,
-      "loss": 0.495,
-      "step": 3906
-    },
-    {
-      "epoch": 6.94,
-      "learning_rate": 1.2049228244075513e-05,
-      "loss": 0.4833,
-      "step": 3907
-    },
-    {
-      "epoch": 6.94,
-      "learning_rate": 1.2045394975342764e-05,
-      "loss": 0.4942,
-      "step": 3908
-    },
-    {
-      "epoch": 6.94,
-      "learning_rate": 1.2041561392912118e-05,
-      "loss": 0.4967,
-      "step": 3909
-    },
-    {
-      "epoch": 6.94,
-      "learning_rate": 1.2037727497371523e-05,
-      "loss": 0.4821,
-      "step": 3910
-    },
-    {
-      "epoch": 6.94,
-      "learning_rate": 1.2033893289308971e-05,
-      "loss": 0.4693,
-      "step": 3911
-    },
-    {
-      "epoch": 6.95,
-      "learning_rate": 1.2030058769312511e-05,
-      "loss": 0.4884,
-      "step": 3912
-    },
-    {
-      "epoch": 6.95,
-      "learning_rate": 1.2026223937970236e-05,
-      "loss": 0.4837,
-      "step": 3913
-    },
-    {
-      "epoch": 6.95,
-      "learning_rate": 1.2022388795870283e-05,
-      "loss": 0.5075,
-      "step": 3914
-    },
-    {
-      "epoch": 6.95,
-      "learning_rate": 1.2018553343600841e-05,
-      "loss": 0.4689,
-      "step": 3915
-    },
-    {
-      "epoch": 6.95,
-      "learning_rate": 1.2014717581750143e-05,
-      "loss": 0.4816,
-      "step": 3916
-    },
-    {
-      "epoch": 6.95,
-      "learning_rate": 1.2010881510906476e-05,
-      "loss": 0.4804,
-      "step": 3917
-    },
-    {
-      "epoch": 6.96,
-      "learning_rate": 1.2007045131658167e-05,
-      "loss": 0.5141,
-      "step": 3918
-    },
-    {
-      "epoch": 6.96,
-      "learning_rate": 1.200320844459359e-05,
-      "loss": 0.4552,
-      "step": 3919
-    },
-    {
-      "epoch": 6.96,
-      "learning_rate": 1.199937145030118e-05,
-      "loss": 0.4618,
-      "step": 3920
-    },
-    {
-      "epoch": 6.96,
-      "learning_rate": 1.1995534149369397e-05,
-      "loss": 0.4768,
-      "step": 3921
-    },
-    {
-      "epoch": 6.96,
-      "learning_rate": 1.1991696542386768e-05,
-      "loss": 0.4773,
-      "step": 3922
-    },
-    {
-      "epoch": 6.96,
-      "learning_rate": 1.1987858629941853e-05,
-      "loss": 0.4939,
-      "step": 3923
-    },
-    {
-      "epoch": 6.97,
-      "learning_rate": 1.1984020412623268e-05,
-      "loss": 0.4805,
-      "step": 3924
-    },
-    {
-      "epoch": 6.97,
-      "learning_rate": 1.198018189101967e-05,
-      "loss": 0.4809,
-      "step": 3925
-    },
-    {
-      "epoch": 6.97,
-      "learning_rate": 1.197634306571977e-05,
-      "loss": 0.4711,
-      "step": 3926
-    },
-    {
-      "epoch": 6.97,
-      "learning_rate": 1.1972503937312315e-05,
-      "loss": 0.4695,
-      "step": 3927
-    },
-    {
-      "epoch": 6.97,
-      "learning_rate": 1.1968664506386102e-05,
-      "loss": 0.4898,
-      "step": 3928
-    },
-    {
-      "epoch": 6.98,
-      "learning_rate": 1.1964824773529986e-05,
-      "loss": 0.5153,
-      "step": 3929
-    },
-    {
-      "epoch": 6.98,
-      "learning_rate": 1.1960984739332851e-05,
-      "loss": 0.4816,
-      "step": 3930
-    },
-    {
-      "epoch": 6.98,
-      "learning_rate": 1.1957144404383634e-05,
-      "loss": 0.4681,
-      "step": 3931
-    },
-    {
-      "epoch": 6.98,
-      "learning_rate": 1.1953303769271324e-05,
-      "loss": 0.4807,
-      "step": 3932
-    },
-    {
-      "epoch": 6.98,
-      "learning_rate": 1.194946283458495e-05,
-      "loss": 0.4756,
-      "step": 3933
-    },
-    {
-      "epoch": 6.98,
-      "learning_rate": 1.1945621600913584e-05,
-      "loss": 0.461,
-      "step": 3934
-    },
-    {
-      "epoch": 6.99,
-      "learning_rate": 1.1941780068846353e-05,
-      "loss": 0.4879,
-      "step": 3935
-    },
-    {
-      "epoch": 6.99,
-      "learning_rate": 1.1937938238972421e-05,
-      "loss": 0.474,
-      "step": 3936
-    },
-    {
-      "epoch": 6.99,
-      "learning_rate": 1.1934096111881006e-05,
-      "loss": 0.4977,
-      "step": 3937
-    },
-    {
-      "epoch": 6.99,
-      "learning_rate": 1.193025368816136e-05,
-      "loss": 0.4942,
-      "step": 3938
-    },
-    {
-      "epoch": 6.99,
-      "learning_rate": 1.1926410968402791e-05,
-      "loss": 0.4656,
-      "step": 3939
-    },
-    {
-      "epoch": 7.0,
-      "learning_rate": 1.192256795319465e-05,
-      "loss": 0.4826,
-      "step": 3940
-    },
-    {
-      "epoch": 7.0,
-      "learning_rate": 1.1918724643126334e-05,
-      "loss": 0.4904,
-      "step": 3941
-    },
-    {
-      "epoch": 7.0,
-      "learning_rate": 1.1914881038787277e-05,
-      "loss": 0.4993,
-      "step": 3942
-    },
-    {
-      "epoch": 7.0,
-      "eval_loss": 0.5171352028846741,
-      "eval_runtime": 3446.2381,
-      "eval_samples_per_second": 4.163,
-      "eval_steps_per_second": 1.041,
-      "step": 3942
-    }
-  ],
-  "logging_steps": 1.0,
-  "max_steps": 8445,
-  "num_train_epochs": 15,
-  "save_steps": 500,
-  "total_flos": 1.0118215300419053e+20,
-  "trial_name": null,
-  "trial_params": null
-}