diff --git a/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/0_trainer_state.json b/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/0_trainer_state.json
new file mode 100644
index 0000000000000000000000000000000000000000..26015c2a40abc231ed285598eaef55e0e79147ae
--- /dev/null
+++ b/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/0_trainer_state.json
@@ -0,0 +1,7532 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 1.0,
+  "eval_steps": 500,
+  "global_step": 2500,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0008,
+      "learning_rate": 2.357535430610912e-06,
+      "loss": 0.0114,
+      "step": 2
+    },
+    {
+      "epoch": 0.0016,
+      "learning_rate": 2.3755748898855234e-06,
+      "loss": 0.0538,
+      "step": 4
+    },
+    {
+      "epoch": 0.0024,
+      "learning_rate": 2.3936738059587174e-06,
+      "loss": 0.1818,
+      "step": 6
+    },
+    {
+      "epoch": 0.0032,
+      "learning_rate": 2.411832037691545e-06,
+      "loss": 0.5497,
+      "step": 8
+    },
+    {
+      "epoch": 0.004,
+      "learning_rate": 2.430049443482434e-06,
+      "loss": 0.0445,
+      "step": 10
+    },
+    {
+      "epoch": 0.0048,
+      "learning_rate": 2.448325881268406e-06,
+      "loss": 0.2739,
+      "step": 12
+    },
+    {
+      "epoch": 0.0056,
+      "learning_rate": 2.4666612085261277e-06,
+      "loss": 0.0153,
+      "step": 14
+    },
+    {
+      "epoch": 0.0064,
+      "learning_rate": 2.4850552822730346e-06,
+      "loss": 0.025,
+      "step": 16
+    },
+    {
+      "epoch": 0.0072,
+      "learning_rate": 2.503507959068455e-06,
+      "loss": 0.5164,
+      "step": 18
+    },
+    {
+      "epoch": 0.008,
+      "learning_rate": 2.522019095014686e-06,
+      "loss": 0.0066,
+      "step": 20
+    },
+    {
+      "epoch": 0.0088,
+      "learning_rate": 2.5405885457581814e-06,
+      "loss": 0.0245,
+      "step": 22
+    },
+    {
+      "epoch": 0.0096,
+      "learning_rate": 2.5592161664906243e-06,
+      "loss": 0.5499,
+      "step": 24
+    },
+    {
+      "epoch": 0.0104,
+      "learning_rate": 2.5779018119501086e-06,
+      "loss": 0.0008,
+      "step": 26
+    },
+    {
+      "epoch": 0.0112,
+      "learning_rate": 2.596645336422219e-06,
+      "loss": 1.0019,
+      "step": 28
+    },
+    {
+      "epoch": 0.012,
+      "learning_rate": 2.615446593741161e-06,
+      "loss": 0.0006,
+      "step": 30
+    },
+    {
+      "epoch": 0.0128,
+      "learning_rate": 2.6343054372909648e-06,
+      "loss": 0.0102,
+      "step": 32
+    },
+    {
+      "epoch": 0.0136,
+      "learning_rate": 2.6532217200065826e-06,
+      "loss": 0.0069,
+      "step": 34
+    },
+    {
+      "epoch": 0.0144,
+      "learning_rate": 2.6721952943750396e-06,
+      "loss": 0.0537,
+      "step": 36
+    },
+    {
+      "epoch": 0.0152,
+      "learning_rate": 2.691226012436604e-06,
+      "loss": 0.0036,
+      "step": 38
+    },
+    {
+      "epoch": 0.016,
+      "learning_rate": 2.7103137257858893e-06,
+      "loss": 1.0076,
+      "step": 40
+    },
+    {
+      "epoch": 0.0168,
+      "learning_rate": 2.7294582855730733e-06,
+      "loss": 0.0316,
+      "step": 42
+    },
+    {
+      "epoch": 0.0176,
+      "learning_rate": 2.7486595425050566e-06,
+      "loss": 0.0013,
+      "step": 44
+    },
+    {
+      "epoch": 0.0184,
+      "learning_rate": 2.7679173468465813e-06,
+      "loss": 0.4162,
+      "step": 46
+    },
+    {
+      "epoch": 0.0192,
+      "learning_rate": 2.7872315484213954e-06,
+      "loss": 0.3843,
+      "step": 48
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 2.8066019966134873e-06,
+      "loss": 0.0093,
+      "step": 50
+    },
+    {
+      "epoch": 0.0208,
+      "learning_rate": 2.826028540368212e-06,
+      "loss": 0.4165,
+      "step": 52
+    },
+    {
+      "epoch": 0.0216,
+      "learning_rate": 2.845511028193477e-06,
+      "loss": 0.204,
+      "step": 54
+    },
+    {
+      "epoch": 0.0224,
+      "learning_rate": 2.865049308160931e-06,
+      "loss": 0.0037,
+      "step": 56
+    },
+    {
+      "epoch": 0.0232,
+      "learning_rate": 2.8846432279071533e-06,
+      "loss": 0.0029,
+      "step": 58
+    },
+    {
+      "epoch": 0.024,
+      "learning_rate": 2.9042926346347835e-06,
+      "loss": 0.334,
+      "step": 60
+    },
+    {
+      "epoch": 0.0248,
+      "learning_rate": 2.9239973751138397e-06,
+      "loss": 0.1269,
+      "step": 62
+    },
+    {
+      "epoch": 0.0256,
+      "learning_rate": 2.943757295682783e-06,
+      "loss": 0.011,
+      "step": 64
+    },
+    {
+      "epoch": 0.0264,
+      "learning_rate": 2.9635722422497983e-06,
+      "loss": 0.0114,
+      "step": 66
+    },
+    {
+      "epoch": 0.0272,
+      "learning_rate": 2.983442060293926e-06,
+      "loss": 0.0149,
+      "step": 68
+    },
+    {
+      "epoch": 0.028,
+      "learning_rate": 3.003366594866345e-06,
+      "loss": 0.0478,
+      "step": 70
+    },
+    {
+      "epoch": 0.0288,
+      "learning_rate": 3.0233456905915338e-06,
+      "loss": 0.3687,
+      "step": 72
+    },
+    {
+      "epoch": 0.0296,
+      "learning_rate": 3.0433791916684885e-06,
+      "loss": 0.0143,
+      "step": 74
+    },
+    {
+      "epoch": 0.0304,
+      "learning_rate": 3.0634669418719453e-06,
+      "loss": 0.0017,
+      "step": 76
+    },
+    {
+      "epoch": 0.0312,
+      "learning_rate": 3.0836087845535933e-06,
+      "loss": 0.1096,
+      "step": 78
+    },
+    {
+      "epoch": 0.032,
+      "learning_rate": 3.1038045626432945e-06,
+      "loss": 0.3132,
+      "step": 80
+    },
+    {
+      "epoch": 0.0328,
+      "learning_rate": 3.1240541186503173e-06,
+      "loss": 0.0037,
+      "step": 82
+    },
+    {
+      "epoch": 0.0336,
+      "learning_rate": 3.1443572946645683e-06,
+      "loss": 0.1466,
+      "step": 84
+    },
+    {
+      "epoch": 0.0344,
+      "learning_rate": 3.164713932357776e-06,
+      "loss": 0.0059,
+      "step": 86
+    },
+    {
+      "epoch": 0.0352,
+      "learning_rate": 3.1851238729848033e-06,
+      "loss": 0.0342,
+      "step": 88
+    },
+    {
+      "epoch": 0.036,
+      "learning_rate": 3.205586957384834e-06,
+      "loss": 0.0024,
+      "step": 90
+    },
+    {
+      "epoch": 0.0368,
+      "learning_rate": 3.2261030259826253e-06,
+      "loss": 0.0054,
+      "step": 92
+    },
+    {
+      "epoch": 0.0376,
+      "learning_rate": 3.246671918789752e-06,
+      "loss": 0.0084,
+      "step": 94
+    },
+    {
+      "epoch": 0.0384,
+      "learning_rate": 3.267293475405858e-06,
+      "loss": 0.0054,
+      "step": 96
+    },
+    {
+      "epoch": 0.0392,
+      "learning_rate": 3.2879675350199004e-06,
+      "loss": 0.0032,
+      "step": 98
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 3.3086939364114113e-06,
+      "loss": 0.0102,
+      "step": 100
+    },
+    {
+      "epoch": 0.0408,
+      "learning_rate": 3.329472517951747e-06,
+      "loss": 0.0022,
+      "step": 102
+    },
+    {
+      "epoch": 0.0416,
+      "learning_rate": 3.350303117605369e-06,
+      "loss": 0.0495,
+      "step": 104
+    },
+    {
+      "epoch": 0.0424,
+      "learning_rate": 3.3711855729310503e-06,
+      "loss": 0.036,
+      "step": 106
+    },
+    {
+      "epoch": 0.0432,
+      "learning_rate": 3.3921197210832235e-06,
+      "loss": 0.3466,
+      "step": 108
+    },
+    {
+      "epoch": 0.044,
+      "learning_rate": 3.4131053988131947e-06,
+      "loss": 0.1809,
+      "step": 110
+    },
+    {
+      "epoch": 0.0448,
+      "learning_rate": 3.434142442470434e-06,
+      "loss": 0.0235,
+      "step": 112
+    },
+    {
+      "epoch": 0.0456,
+      "learning_rate": 3.455230688003849e-06,
+      "loss": 0.0472,
+      "step": 114
+    },
+    {
+      "epoch": 0.0464,
+      "learning_rate": 3.476369970963065e-06,
+      "loss": 0.004,
+      "step": 116
+    },
+    {
+      "epoch": 0.0472,
+      "learning_rate": 3.497560126499706e-06,
+      "loss": 0.0425,
+      "step": 118
+    },
+    {
+      "epoch": 0.048,
+      "learning_rate": 3.5188009893686836e-06,
+      "loss": 0.0014,
+      "step": 120
+    },
+    {
+      "epoch": 0.0488,
+      "learning_rate": 3.5400923939294827e-06,
+      "loss": 0.0007,
+      "step": 122
+    },
+    {
+      "epoch": 0.0496,
+      "learning_rate": 3.5614341741474667e-06,
+      "loss": 0.1245,
+      "step": 124
+    },
+    {
+      "epoch": 0.0504,
+      "learning_rate": 3.5828261635951177e-06,
+      "loss": 0.0381,
+      "step": 126
+    },
+    {
+      "epoch": 0.0512,
+      "learning_rate": 3.604268195453421e-06,
+      "loss": 0.119,
+      "step": 128
+    },
+    {
+      "epoch": 0.052,
+      "learning_rate": 3.6257601025130893e-06,
+      "loss": 0.0039,
+      "step": 130
+    },
+    {
+      "epoch": 0.0528,
+      "learning_rate": 3.647301717175955e-06,
+      "loss": 0.0322,
+      "step": 132
+    },
+    {
+      "epoch": 0.0536,
+      "learning_rate": 3.66889287145614e-06,
+      "loss": 0.0574,
+      "step": 134
+    },
+    {
+      "epoch": 0.0544,
+      "learning_rate": 3.6905333969814995e-06,
+      "loss": 0.0746,
+      "step": 136
+    },
+    {
+      "epoch": 0.0552,
+      "learning_rate": 3.712223124994867e-06,
+      "loss": 0.0444,
+      "step": 138
+    },
+    {
+      "epoch": 0.056,
+      "learning_rate": 3.7339618863553885e-06,
+      "loss": 0.0497,
+      "step": 140
+    },
+    {
+      "epoch": 0.0568,
+      "learning_rate": 3.755749511539848e-06,
+      "loss": 0.001,
+      "step": 142
+    },
+    {
+      "epoch": 0.0576,
+      "learning_rate": 3.7775858306439404e-06,
+      "loss": 0.1217,
+      "step": 144
+    },
+    {
+      "epoch": 0.0584,
+      "learning_rate": 3.799470673383677e-06,
+      "loss": 0.1533,
+      "step": 146
+    },
+    {
+      "epoch": 0.0592,
+      "learning_rate": 3.821403869096644e-06,
+      "loss": 0.0017,
+      "step": 148
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 3.8433852467434175e-06,
+      "loss": 0.0614,
+      "step": 150
+    },
+    {
+      "epoch": 0.0608,
+      "learning_rate": 3.865414634908756e-06,
+      "loss": 0.0003,
+      "step": 152
+    },
+    {
+      "epoch": 0.0616,
+      "learning_rate": 3.887491861803081e-06,
+      "loss": 0.0377,
+      "step": 154
+    },
+    {
+      "epoch": 0.0624,
+      "learning_rate": 3.909616755263741e-06,
+      "loss": 0.027,
+      "step": 156
+    },
+    {
+      "epoch": 0.0632,
+      "learning_rate": 3.9317891427563725e-06,
+      "loss": 0.6366,
+      "step": 158
+    },
+    {
+      "epoch": 0.064,
+      "learning_rate": 3.954008851376244e-06,
+      "loss": 0.0048,
+      "step": 160
+    },
+    {
+      "epoch": 0.0648,
+      "learning_rate": 3.976275707849619e-06,
+      "loss": 0.0695,
+      "step": 162
+    },
+    {
+      "epoch": 0.0656,
+      "learning_rate": 3.99858953853505e-06,
+      "loss": 0.0022,
+      "step": 164
+    },
+    {
+      "epoch": 0.0664,
+      "learning_rate": 4.0209501694248e-06,
+      "loss": 0.0024,
+      "step": 166
+    },
+    {
+      "epoch": 0.0672,
+      "learning_rate": 4.043357426146209e-06,
+      "loss": 0.0038,
+      "step": 168
+    },
+    {
+      "epoch": 0.068,
+      "learning_rate": 4.065811133962987e-06,
+      "loss": 0.2259,
+      "step": 170
+    },
+    {
+      "epoch": 0.0688,
+      "learning_rate": 4.08831111777658e-06,
+      "loss": 0.0097,
+      "step": 172
+    },
+    {
+      "epoch": 0.0696,
+      "learning_rate": 4.110857202127611e-06,
+      "loss": 0.0053,
+      "step": 174
+    },
+    {
+      "epoch": 0.0704,
+      "learning_rate": 4.133449211197183e-06,
+      "loss": 0.0052,
+      "step": 176
+    },
+    {
+      "epoch": 0.0712,
+      "learning_rate": 4.156086968808274e-06,
+      "loss": 0.0007,
+      "step": 178
+    },
+    {
+      "epoch": 0.072,
+      "learning_rate": 4.178770298427114e-06,
+      "loss": 0.0012,
+      "step": 180
+    },
+    {
+      "epoch": 0.0728,
+      "learning_rate": 4.201499023164515e-06,
+      "loss": 0.0042,
+      "step": 182
+    },
+    {
+      "epoch": 0.0736,
+      "learning_rate": 4.224272965777315e-06,
+      "loss": 0.0049,
+      "step": 184
+    },
+    {
+      "epoch": 0.0744,
+      "learning_rate": 4.247091948669764e-06,
+      "loss": 0.143,
+      "step": 186
+    },
+    {
+      "epoch": 0.0752,
+      "learning_rate": 4.269955793894849e-06,
+      "loss": 0.0142,
+      "step": 188
+    },
+    {
+      "epoch": 0.076,
+      "learning_rate": 4.292864323155684e-06,
+      "loss": 0.014,
+      "step": 190
+    },
+    {
+      "epoch": 0.0768,
+      "learning_rate": 4.3158173578069696e-06,
+      "loss": 0.0904,
+      "step": 192
+    },
+    {
+      "epoch": 0.0776,
+      "learning_rate": 4.338814718856333e-06,
+      "loss": 0.0013,
+      "step": 194
+    },
+    {
+      "epoch": 0.0784,
+      "learning_rate": 4.3618562269657285e-06,
+      "loss": 0.0035,
+      "step": 196
+    },
+    {
+      "epoch": 0.0792,
+      "learning_rate": 4.384941702452852e-06,
+      "loss": 0.0274,
+      "step": 198
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 4.408070965292526e-06,
+      "loss": 0.0032,
+      "step": 200
+    },
+    {
+      "epoch": 0.0808,
+      "learning_rate": 4.431243835118112e-06,
+      "loss": 0.0075,
+      "step": 202
+    },
+    {
+      "epoch": 0.0816,
+      "learning_rate": 4.4544601312229185e-06,
+      "loss": 0.8027,
+      "step": 204
+    },
+    {
+      "epoch": 0.0824,
+      "learning_rate": 4.477719672561602e-06,
+      "loss": 0.0012,
+      "step": 206
+    },
+    {
+      "epoch": 0.0832,
+      "learning_rate": 4.501022277751605e-06,
+      "loss": 0.0023,
+      "step": 208
+    },
+    {
+      "epoch": 0.084,
+      "learning_rate": 4.524367765074499e-06,
+      "loss": 0.0265,
+      "step": 210
+    },
+    {
+      "epoch": 0.0848,
+      "learning_rate": 4.5477559524775e-06,
+      "loss": 0.2539,
+      "step": 212
+    },
+    {
+      "epoch": 0.0856,
+      "learning_rate": 4.571186657574823e-06,
+      "loss": 0.0004,
+      "step": 214
+    },
+    {
+      "epoch": 0.0864,
+      "learning_rate": 4.5946596976491254e-06,
+      "loss": 0.0025,
+      "step": 216
+    },
+    {
+      "epoch": 0.0872,
+      "learning_rate": 4.618174889652924e-06,
+      "loss": 0.0023,
+      "step": 218
+    },
+    {
+      "epoch": 0.088,
+      "learning_rate": 4.6417320502100286e-06,
+      "loss": 0.0594,
+      "step": 220
+    },
+    {
+      "epoch": 0.0888,
+      "learning_rate": 4.665330995616967e-06,
+      "loss": 0.0184,
+      "step": 222
+    },
+    {
+      "epoch": 0.0896,
+      "learning_rate": 4.688971541844424e-06,
+      "loss": 0.001,
+      "step": 224
+    },
+    {
+      "epoch": 0.0904,
+      "learning_rate": 4.712653504538672e-06,
+      "loss": 0.0034,
+      "step": 226
+    },
+    {
+      "epoch": 0.0912,
+      "learning_rate": 4.736376699023023e-06,
+      "loss": 0.034,
+      "step": 228
+    },
+    {
+      "epoch": 0.092,
+      "learning_rate": 4.76014094029921e-06,
+      "loss": 0.008,
+      "step": 230
+    },
+    {
+      "epoch": 0.0928,
+      "learning_rate": 4.7839460430489216e-06,
+      "loss": 0.012,
+      "step": 232
+    },
+    {
+      "epoch": 0.0936,
+      "learning_rate": 4.807791821635185e-06,
+      "loss": 0.0023,
+      "step": 234
+    },
+    {
+      "epoch": 0.0944,
+      "learning_rate": 4.831678090103828e-06,
+      "loss": 0.3672,
+      "step": 236
+    },
+    {
+      "epoch": 0.0952,
+      "learning_rate": 4.855604662184931e-06,
+      "loss": 0.0014,
+      "step": 238
+    },
+    {
+      "epoch": 0.096,
+      "learning_rate": 4.8795713512942785e-06,
+      "loss": 0.0039,
+      "step": 240
+    },
+    {
+      "epoch": 0.0968,
+      "learning_rate": 4.903577970534815e-06,
+      "loss": 0.0008,
+      "step": 242
+    },
+    {
+      "epoch": 0.0976,
+      "learning_rate": 4.9276243326981e-06,
+      "loss": 0.0046,
+      "step": 244
+    },
+    {
+      "epoch": 0.0984,
+      "learning_rate": 4.951710250265788e-06,
+      "loss": 0.0463,
+      "step": 246
+    },
+    {
+      "epoch": 0.0992,
+      "learning_rate": 4.975835535411023e-06,
+      "loss": 0.004,
+      "step": 248
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 5.000000000000003e-06,
+      "loss": 0.0037,
+      "step": 250
+    },
+    {
+      "epoch": 0.1008,
+      "learning_rate": 5.024203455593375e-06,
+      "loss": 0.0231,
+      "step": 252
+    },
+    {
+      "epoch": 0.1016,
+      "learning_rate": 5.048445713447734e-06,
+      "loss": 0.0795,
+      "step": 254
+    },
+    {
+      "epoch": 0.1024,
+      "learning_rate": 5.072726584517083e-06,
+      "loss": 0.7435,
+      "step": 256
+    },
+    {
+      "epoch": 0.1032,
+      "learning_rate": 5.097045879454308e-06,
+      "loss": 0.0049,
+      "step": 258
+    },
+    {
+      "epoch": 0.104,
+      "learning_rate": 5.1214034086126685e-06,
+      "loss": 0.0009,
+      "step": 260
+    },
+    {
+      "epoch": 0.1048,
+      "learning_rate": 5.145798982047253e-06,
+      "loss": 0.0047,
+      "step": 262
+    },
+    {
+      "epoch": 0.1056,
+      "learning_rate": 5.170232409516483e-06,
+      "loss": 0.0286,
+      "step": 264
+    },
+    {
+      "epoch": 0.1064,
+      "learning_rate": 5.194703500483597e-06,
+      "loss": 0.0018,
+      "step": 266
+    },
+    {
+      "epoch": 0.1072,
+      "learning_rate": 5.219212064118082e-06,
+      "loss": 0.0011,
+      "step": 268
+    },
+    {
+      "epoch": 0.108,
+      "learning_rate": 5.24375790929725e-06,
+      "loss": 0.0004,
+      "step": 270
+    },
+    {
+      "epoch": 0.1088,
+      "learning_rate": 5.268340844607653e-06,
+      "loss": 0.0462,
+      "step": 272
+    },
+    {
+      "epoch": 0.1096,
+      "learning_rate": 5.2929606783466735e-06,
+      "loss": 0.0012,
+      "step": 274
+    },
+    {
+      "epoch": 0.1104,
+      "learning_rate": 5.317617218523853e-06,
+      "loss": 0.0096,
+      "step": 276
+    },
+    {
+      "epoch": 0.1112,
+      "learning_rate": 5.342310272862553e-06,
+      "loss": 0.0264,
+      "step": 278
+    },
+    {
+      "epoch": 0.112,
+      "learning_rate": 5.367039648801377e-06,
+      "loss": 0.0003,
+      "step": 280
+    },
+    {
+      "epoch": 0.1128,
+      "learning_rate": 5.391805153495684e-06,
+      "loss": 0.0022,
+      "step": 282
+    },
+    {
+      "epoch": 0.1136,
+      "learning_rate": 5.416606593819109e-06,
+      "loss": 0.0209,
+      "step": 284
+    },
+    {
+      "epoch": 0.1144,
+      "learning_rate": 5.441443776365005e-06,
+      "loss": 0.4095,
+      "step": 286
+    },
+    {
+      "epoch": 0.1152,
+      "learning_rate": 5.466316507448053e-06,
+      "loss": 0.0015,
+      "step": 288
+    },
+    {
+      "epoch": 0.116,
+      "learning_rate": 5.49122459310568e-06,
+      "loss": 0.0059,
+      "step": 290
+    },
+    {
+      "epoch": 0.1168,
+      "learning_rate": 5.516167839099662e-06,
+      "loss": 0.0031,
+      "step": 292
+    },
+    {
+      "epoch": 0.1176,
+      "learning_rate": 5.5411460509175605e-06,
+      "loss": 0.0229,
+      "step": 294
+    },
+    {
+      "epoch": 0.1184,
+      "learning_rate": 5.5661590337742255e-06,
+      "loss": 0.0019,
+      "step": 296
+    },
+    {
+      "epoch": 0.1192,
+      "learning_rate": 5.591206592613412e-06,
+      "loss": 0.0674,
+      "step": 298
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 5.616288532109221e-06,
+      "loss": 0.0055,
+      "step": 300
+    },
+    {
+      "epoch": 0.1208,
+      "learning_rate": 5.641404656667652e-06,
+      "loss": 0.0002,
+      "step": 302
+    },
+    {
+      "epoch": 0.1216,
+      "learning_rate": 5.666554770428136e-06,
+      "loss": 0.0066,
+      "step": 304
+    },
+    {
+      "epoch": 0.1224,
+      "learning_rate": 5.6917386772650015e-06,
+      "loss": 0.0032,
+      "step": 306
+    },
+    {
+      "epoch": 0.1232,
+      "learning_rate": 5.716956180789086e-06,
+      "loss": 0.0002,
+      "step": 308
+    },
+    {
+      "epoch": 0.124,
+      "learning_rate": 5.74220708434926e-06,
+      "loss": 0.0293,
+      "step": 310
+    },
+    {
+      "epoch": 0.1248,
+      "learning_rate": 5.767491191033909e-06,
+      "loss": 0.0051,
+      "step": 312
+    },
+    {
+      "epoch": 0.1256,
+      "learning_rate": 5.7928083036724535e-06,
+      "loss": 0.278,
+      "step": 314
+    },
+    {
+      "epoch": 0.1264,
+      "learning_rate": 5.818158224836983e-06,
+      "loss": 0.0008,
+      "step": 316
+    },
+    {
+      "epoch": 0.1272,
+      "learning_rate": 5.8435407568437194e-06,
+      "loss": 0.0324,
+      "step": 318
+    },
+    {
+      "epoch": 0.128,
+      "learning_rate": 5.868955701754577e-06,
+      "loss": 0.0127,
+      "step": 320
+    },
+    {
+      "epoch": 0.1288,
+      "learning_rate": 5.894402861378714e-06,
+      "loss": 0.0164,
+      "step": 322
+    },
+    {
+      "epoch": 0.1296,
+      "learning_rate": 5.919882037274065e-06,
+      "loss": 0.8316,
+      "step": 324
+    },
+    {
+      "epoch": 0.1304,
+      "learning_rate": 5.9453930307488985e-06,
+      "loss": 0.0853,
+      "step": 326
+    },
+    {
+      "epoch": 0.1312,
+      "learning_rate": 5.970935642863362e-06,
+      "loss": 0.0004,
+      "step": 328
+    },
+    {
+      "epoch": 0.132,
+      "learning_rate": 5.996509674431038e-06,
+      "loss": 0.0261,
+      "step": 330
+    },
+    {
+      "epoch": 0.1328,
+      "learning_rate": 6.022114926020505e-06,
+      "loss": 0.7622,
+      "step": 332
+    },
+    {
+      "epoch": 0.1336,
+      "learning_rate": 6.047751197956836e-06,
+      "loss": 0.0873,
+      "step": 334
+    },
+    {
+      "epoch": 0.1344,
+      "learning_rate": 6.0734182903232475e-06,
+      "loss": 0.0004,
+      "step": 336
+    },
+    {
+      "epoch": 0.1352,
+      "learning_rate": 6.0991160029626e-06,
+      "loss": 0.0301,
+      "step": 338
+    },
+    {
+      "epoch": 0.136,
+      "learning_rate": 6.124844135478966e-06,
+      "loss": 0.0036,
+      "step": 340
+    },
+    {
+      "epoch": 0.1368,
+      "learning_rate": 6.1506024872392e-06,
+      "loss": 0.0004,
+      "step": 342
+    },
+    {
+      "epoch": 0.1376,
+      "learning_rate": 6.176390857374501e-06,
+      "loss": 0.0043,
+      "step": 344
+    },
+    {
+      "epoch": 0.1384,
+      "learning_rate": 6.202209044781979e-06,
+      "loss": 0.0001,
+      "step": 346
+    },
+    {
+      "epoch": 0.1392,
+      "learning_rate": 6.228056848126223e-06,
+      "loss": 0.0667,
+      "step": 348
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 6.253934065840883e-06,
+      "loss": 0.0061,
+      "step": 350
+    },
+    {
+      "epoch": 0.1408,
+      "learning_rate": 6.279840496130188e-06,
+      "loss": 0.0089,
+      "step": 352
+    },
+    {
+      "epoch": 0.1416,
+      "learning_rate": 6.305775936970606e-06,
+      "loss": 0.0907,
+      "step": 354
+    },
+    {
+      "epoch": 0.1424,
+      "learning_rate": 6.331740186112359e-06,
+      "loss": 0.0035,
+      "step": 356
+    },
+    {
+      "epoch": 0.1432,
+      "learning_rate": 6.357733041081015e-06,
+      "loss": 0.5141,
+      "step": 358
+    },
+    {
+      "epoch": 0.144,
+      "learning_rate": 6.383754299179072e-06,
+      "loss": 0.1489,
+      "step": 360
+    },
+    {
+      "epoch": 0.1448,
+      "learning_rate": 6.409803757487532e-06,
+      "loss": 0.0288,
+      "step": 362
+    },
+    {
+      "epoch": 0.1456,
+      "learning_rate": 6.435881212867485e-06,
+      "loss": 0.1001,
+      "step": 364
+    },
+    {
+      "epoch": 0.1464,
+      "learning_rate": 6.4619864619616975e-06,
+      "loss": 0.0039,
+      "step": 366
+    },
+    {
+      "epoch": 0.1472,
+      "learning_rate": 6.48811930119619e-06,
+      "loss": 0.2385,
+      "step": 368
+    },
+    {
+      "epoch": 0.148,
+      "learning_rate": 6.514279526781853e-06,
+      "loss": 0.0015,
+      "step": 370
+    },
+    {
+      "epoch": 0.1488,
+      "learning_rate": 6.540466934715955e-06,
+      "loss": 0.0478,
+      "step": 372
+    },
+    {
+      "epoch": 0.1496,
+      "learning_rate": 6.566681320783848e-06,
+      "loss": 0.0036,
+      "step": 374
+    },
+    {
+      "epoch": 0.1504,
+      "learning_rate": 6.592922480560483e-06,
+      "loss": 0.0085,
+      "step": 376
+    },
+    {
+      "epoch": 0.1512,
+      "learning_rate": 6.619190209412025e-06,
+      "loss": 0.0015,
+      "step": 378
+    },
+    {
+      "epoch": 0.152,
+      "learning_rate": 6.6454843024974465e-06,
+      "loss": 0.0013,
+      "step": 380
+    },
+    {
+      "epoch": 0.1528,
+      "learning_rate": 6.671804554770128e-06,
+      "loss": 0.0002,
+      "step": 382
+    },
+    {
+      "epoch": 0.1536,
+      "learning_rate": 6.698150760979456e-06,
+      "loss": 0.0011,
+      "step": 384
+    },
+    {
+      "epoch": 0.1544,
+      "learning_rate": 6.724522715672421e-06,
+      "loss": 0.0012,
+      "step": 386
+    },
+    {
+      "epoch": 0.1552,
+      "learning_rate": 6.750920213195242e-06,
+      "loss": 0.3655,
+      "step": 388
+    },
+    {
+      "epoch": 0.156,
+      "learning_rate": 6.777343047694894e-06,
+      "loss": 0.0018,
+      "step": 390
+    },
+    {
+      "epoch": 0.1568,
+      "learning_rate": 6.803791013120824e-06,
+      "loss": 0.2722,
+      "step": 392
+    },
+    {
+      "epoch": 0.1576,
+      "learning_rate": 6.8302639032264836e-06,
+      "loss": 0.0345,
+      "step": 394
+    },
+    {
+      "epoch": 0.1584,
+      "learning_rate": 6.856761511570944e-06,
+      "loss": 0.0025,
+      "step": 396
+    },
+    {
+      "epoch": 0.1592,
+      "learning_rate": 6.883283631520579e-06,
+      "loss": 0.2781,
+      "step": 398
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 6.909830056250522e-06,
+      "loss": 0.006,
+      "step": 400
+    },
+    {
+      "epoch": 0.1608,
+      "learning_rate": 6.936400578746436e-06,
+      "loss": 0.0012,
+      "step": 402
+    },
+    {
+      "epoch": 0.1616,
+      "learning_rate": 6.96299499180605e-06,
+      "loss": 0.0173,
+      "step": 404
+    },
+    {
+      "epoch": 0.1624,
+      "learning_rate": 6.989613088040787e-06,
+      "loss": 0.0631,
+      "step": 406
+    },
+    {
+      "epoch": 0.1632,
+      "learning_rate": 7.016254659877404e-06,
+      "loss": 0.0364,
+      "step": 408
+    },
+    {
+      "epoch": 0.164,
+      "learning_rate": 7.042919499559539e-06,
+      "loss": 0.0771,
+      "step": 410
+    },
+    {
+      "epoch": 0.1648,
+      "learning_rate": 7.06960739914943e-06,
+      "loss": 0.0028,
+      "step": 412
+    },
+    {
+      "epoch": 0.1656,
+      "learning_rate": 7.09631815052946e-06,
+      "loss": 0.0001,
+      "step": 414
+    },
+    {
+      "epoch": 0.1664,
+      "learning_rate": 7.123051545403873e-06,
+      "loss": 0.0004,
+      "step": 416
+    },
+    {
+      "epoch": 0.1672,
+      "learning_rate": 7.1498073753002375e-06,
+      "loss": 0.3526,
+      "step": 418
+    },
+    {
+      "epoch": 0.168,
+      "learning_rate": 7.1765854315712325e-06,
+      "loss": 0.0049,
+      "step": 420
+    },
+    {
+      "epoch": 0.1688,
+      "learning_rate": 7.203385505396197e-06,
+      "loss": 0.001,
+      "step": 422
+    },
+    {
+      "epoch": 0.1696,
+      "learning_rate": 7.230207387782771e-06,
+      "loss": 0.3729,
+      "step": 424
+    },
+    {
+      "epoch": 0.1704,
+      "learning_rate": 7.257050869568527e-06,
+      "loss": 0.0225,
+      "step": 426
+    },
+    {
+      "epoch": 0.1712,
+      "learning_rate": 7.28391574142262e-06,
+      "loss": 0.0008,
+      "step": 428
+    },
+    {
+      "epoch": 0.172,
+      "learning_rate": 7.3108017938473485e-06,
+      "loss": 0.0008,
+      "step": 430
+    },
+    {
+      "epoch": 0.1728,
+      "learning_rate": 7.337708817179875e-06,
+      "loss": 0.0014,
+      "step": 432
+    },
+    {
+      "epoch": 0.1736,
+      "learning_rate": 7.36463660159386e-06,
+      "loss": 0.0002,
+      "step": 434
+    },
+    {
+      "epoch": 0.1744,
+      "learning_rate": 7.39158493710103e-06,
+      "loss": 0.0004,
+      "step": 436
+    },
+    {
+      "epoch": 0.1752,
+      "learning_rate": 7.418553613552822e-06,
+      "loss": 0.0077,
+      "step": 438
+    },
+    {
+      "epoch": 0.176,
+      "learning_rate": 7.445542420642091e-06,
+      "loss": 0.4597,
+      "step": 440
+    },
+    {
+      "epoch": 0.1768,
+      "learning_rate": 7.472551147904703e-06,
+      "loss": 0.1823,
+      "step": 442
+    },
+    {
+      "epoch": 0.1776,
+      "learning_rate": 7.499579584721173e-06,
+      "loss": 0.0015,
+      "step": 444
+    },
+    {
+      "epoch": 0.1784,
+      "learning_rate": 7.5266275203183395e-06,
+      "loss": 0.0297,
+      "step": 446
+    },
+    {
+      "epoch": 0.1792,
+      "learning_rate": 7.553694743770917e-06,
+      "loss": 0.0674,
+      "step": 448
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 7.580781044003312e-06,
+      "loss": 0.0713,
+      "step": 450
+    },
+    {
+      "epoch": 0.1808,
+      "learning_rate": 7.607886209791095e-06,
+      "loss": 0.001,
+      "step": 452
+    },
+    {
+      "epoch": 0.1816,
+      "learning_rate": 7.635010029762755e-06,
+      "loss": 0.0012,
+      "step": 454
+    },
+    {
+      "epoch": 0.1824,
+      "learning_rate": 7.662152292401265e-06,
+      "loss": 0.0021,
+      "step": 456
+    },
+    {
+      "epoch": 0.1832,
+      "learning_rate": 7.689312786045822e-06,
+      "loss": 0.0148,
+      "step": 458
+    },
+    {
+      "epoch": 0.184,
+      "learning_rate": 7.716491298893441e-06,
+      "loss": 0.0057,
+      "step": 460
+    },
+    {
+      "epoch": 0.1848,
+      "learning_rate": 7.74368761900062e-06,
+      "loss": 0.0339,
+      "step": 462
+    },
+    {
+      "epoch": 0.1856,
+      "learning_rate": 7.770901534284991e-06,
+      "loss": 0.003,
+      "step": 464
+    },
+    {
+      "epoch": 0.1864,
+      "learning_rate": 7.798132832526976e-06,
+      "loss": 0.0326,
+      "step": 466
+    },
+    {
+      "epoch": 0.1872,
+      "learning_rate": 7.825381301371444e-06,
+      "loss": 0.0095,
+      "step": 468
+    },
+    {
+      "epoch": 0.188,
+      "learning_rate": 7.852646728329358e-06,
+      "loss": 0.0037,
+      "step": 470
+    },
+    {
+      "epoch": 0.1888,
+      "learning_rate": 7.879928900779441e-06,
+      "loss": 0.0264,
+      "step": 472
+    },
+    {
+      "epoch": 0.1896,
+      "learning_rate": 7.907227605969852e-06,
+      "loss": 0.0059,
+      "step": 474
+    },
+    {
+      "epoch": 0.1904,
+      "learning_rate": 7.934542631019767e-06,
+      "loss": 0.014,
+      "step": 476
+    },
+    {
+      "epoch": 0.1912,
+      "learning_rate": 7.961873762921153e-06,
+      "loss": 0.0004,
+      "step": 478
+    },
+    {
+      "epoch": 0.192,
+      "learning_rate": 7.989220788540351e-06,
+      "loss": 0.0831,
+      "step": 480
+    },
+    {
+      "epoch": 0.1928,
+      "learning_rate": 8.016583494619764e-06,
+      "loss": 0.0018,
+      "step": 482
+    },
+    {
+      "epoch": 0.1936,
+      "learning_rate": 8.043961667779511e-06,
+      "loss": 0.1637,
+      "step": 484
+    },
+    {
+      "epoch": 0.1944,
+      "learning_rate": 8.071355094519103e-06,
+      "loss": 0.6775,
+      "step": 486
+    },
+    {
+      "epoch": 0.1952,
+      "learning_rate": 8.098763561219089e-06,
+      "loss": 0.0003,
+      "step": 488
+    },
+    {
+      "epoch": 0.196,
+      "learning_rate": 8.126186854142744e-06,
+      "loss": 0.0012,
+      "step": 490
+    },
+    {
+      "epoch": 0.1968,
+      "learning_rate": 8.153624759437718e-06,
+      "loss": 0.0007,
+      "step": 492
+    },
+    {
+      "epoch": 0.1976,
+      "learning_rate": 8.181077063137735e-06,
+      "loss": 0.9133,
+      "step": 494
+    },
+    {
+      "epoch": 0.1984,
+      "learning_rate": 8.208543551164178e-06,
+      "loss": 0.0029,
+      "step": 496
+    },
+    {
+      "epoch": 0.1992,
+      "learning_rate": 8.236024009327877e-06,
+      "loss": 0.0982,
+      "step": 498
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 8.263518223330695e-06,
+      "loss": 0.0166,
+      "step": 500
+    },
+    {
+      "epoch": 0.2008,
+      "learning_rate": 8.29102597876723e-06,
+      "loss": 0.0606,
+      "step": 502
+    },
+    {
+      "epoch": 0.2016,
+      "learning_rate": 8.31854706112648e-06,
+      "loss": 0.0004,
+      "step": 504
+    },
+    {
+      "epoch": 0.2024,
+      "learning_rate": 8.346081255793516e-06,
+      "loss": 0.001,
+      "step": 506
+    },
+    {
+      "epoch": 0.2032,
+      "learning_rate": 8.373628348051156e-06,
+      "loss": 0.0188,
+      "step": 508
+    },
+    {
+      "epoch": 0.204,
+      "learning_rate": 8.401188123081642e-06,
+      "loss": 0.0079,
+      "step": 510
+    },
+    {
+      "epoch": 0.2048,
+      "learning_rate": 8.428760365968329e-06,
+      "loss": 0.003,
+      "step": 512
+    },
+    {
+      "epoch": 0.2056,
+      "learning_rate": 8.456344861697293e-06,
+      "loss": 0.0823,
+      "step": 514
+    },
+    {
+      "epoch": 0.2064,
+      "learning_rate": 8.483941395159114e-06,
+      "loss": 0.043,
+      "step": 516
+    },
+    {
+      "epoch": 0.2072,
+      "learning_rate": 8.511549751150478e-06,
+      "loss": 0.0042,
+      "step": 518
+    },
+    {
+      "epoch": 0.208,
+      "learning_rate": 8.539169714375883e-06,
+      "loss": 0.0027,
+      "step": 520
+    },
+    {
+      "epoch": 0.2088,
+      "learning_rate": 8.566801069449304e-06,
+      "loss": 0.9017,
+      "step": 522
+    },
+    {
+      "epoch": 0.2096,
+      "learning_rate": 8.594443600895886e-06,
+      "loss": 0.0148,
+      "step": 524
+    },
+    {
+      "epoch": 0.2104,
+      "learning_rate": 8.622097093153612e-06,
+      "loss": 0.0513,
+      "step": 526
+    },
+    {
+      "epoch": 0.2112,
+      "learning_rate": 8.649761330575e-06,
+      "loss": 0.6009,
+      "step": 528
+    },
+    {
+      "epoch": 0.212,
+      "learning_rate": 8.677436097428766e-06,
+      "loss": 0.0026,
+      "step": 530
+    },
+    {
+      "epoch": 0.2128,
+      "learning_rate": 8.705121177901537e-06,
+      "loss": 0.0009,
+      "step": 532
+    },
+    {
+      "epoch": 0.2136,
+      "learning_rate": 8.732816356099459e-06,
+      "loss": 0.0011,
+      "step": 534
+    },
+    {
+      "epoch": 0.2144,
+      "learning_rate": 8.760521416049986e-06,
+      "loss": 0.1258,
+      "step": 536
+    },
+    {
+      "epoch": 0.2152,
+      "learning_rate": 8.788236141703477e-06,
+      "loss": 0.0101,
+      "step": 538
+    },
+    {
+      "epoch": 0.216,
+      "learning_rate": 8.81596031693499e-06,
+      "loss": 0.0406,
+      "step": 540
+    },
+    {
+      "epoch": 0.2168,
+      "learning_rate": 8.84369372554578e-06,
+      "loss": 0.0734,
+      "step": 542
+    },
+    {
+      "epoch": 0.2176,
+      "learning_rate": 8.87143615126518e-06,
+      "loss": 0.0003,
+      "step": 544
+    },
+    {
+      "epoch": 0.2184,
+      "learning_rate": 8.899187377752173e-06,
+      "loss": 0.0004,
+      "step": 546
+    },
+    {
+      "epoch": 0.2192,
+      "learning_rate": 8.926947188597127e-06,
+      "loss": 0.0299,
+      "step": 548
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 8.954715367323473e-06,
+      "loss": 0.122,
+      "step": 550
+    },
+    {
+      "epoch": 0.2208,
+      "learning_rate": 8.982491697389344e-06,
+      "loss": 0.0006,
+      "step": 552
+    },
+    {
+      "epoch": 0.2216,
+      "learning_rate": 9.010275962189356e-06,
+      "loss": 0.0203,
+      "step": 554
+    },
+    {
+      "epoch": 0.2224,
+      "learning_rate": 9.03806794505621e-06,
+      "loss": 0.0011,
+      "step": 556
+    },
+    {
+      "epoch": 0.2232,
+      "learning_rate": 9.065867429262497e-06,
+      "loss": 0.4826,
+      "step": 558
+    },
+    {
+      "epoch": 0.224,
+      "learning_rate": 9.093674198022198e-06,
+      "loss": 0.0016,
+      "step": 560
+    },
+    {
+      "epoch": 0.2248,
+      "learning_rate": 9.121488034492567e-06,
+      "loss": 0.4759,
+      "step": 562
+    },
+    {
+      "epoch": 0.2256,
+      "learning_rate": 9.149308721775717e-06,
+      "loss": 0.0429,
+      "step": 564
+    },
+    {
+      "epoch": 0.2264,
+      "learning_rate": 9.177136042920338e-06,
+      "loss": 0.0005,
+      "step": 566
+    },
+    {
+      "epoch": 0.2272,
+      "learning_rate": 9.204969780923396e-06,
+      "loss": 0.0174,
+      "step": 568
+    },
+    {
+      "epoch": 0.228,
+      "learning_rate": 9.232809718731822e-06,
+      "loss": 1.223,
+      "step": 570
+    },
+    {
+      "epoch": 0.2288,
+      "learning_rate": 9.26065563924414e-06,
+      "loss": 0.0002,
+      "step": 572
+    },
+    {
+      "epoch": 0.2296,
+      "learning_rate": 9.288507325312319e-06,
+      "loss": 0.0051,
+      "step": 574
+    },
+    {
+      "epoch": 0.2304,
+      "learning_rate": 9.316364559743298e-06,
+      "loss": 0.0003,
+      "step": 576
+    },
+    {
+      "epoch": 0.2312,
+      "learning_rate": 9.344227125300788e-06,
+      "loss": 0.0137,
+      "step": 578
+    },
+    {
+      "epoch": 0.232,
+      "learning_rate": 9.372094804706867e-06,
+      "loss": 0.0003,
+      "step": 580
+    },
+    {
+      "epoch": 0.2328,
+      "learning_rate": 9.39996738064379e-06,
+      "loss": 0.7711,
+      "step": 582
+    },
+    {
+      "epoch": 0.2336,
+      "learning_rate": 9.427844635755615e-06,
+      "loss": 0.0449,
+      "step": 584
+    },
+    {
+      "epoch": 0.2344,
+      "learning_rate": 9.455726352649904e-06,
+      "loss": 0.825,
+      "step": 586
+    },
+    {
+      "epoch": 0.2352,
+      "learning_rate": 9.483612313899446e-06,
+      "loss": 0.0013,
+      "step": 588
+    },
+    {
+      "epoch": 0.236,
+      "learning_rate": 9.511502302043859e-06,
+      "loss": 0.156,
+      "step": 590
+    },
+    {
+      "epoch": 0.2368,
+      "learning_rate": 9.539396099591469e-06,
+      "loss": 0.512,
+      "step": 592
+    },
+    {
+      "epoch": 0.2376,
+      "learning_rate": 9.567293489020816e-06,
+      "loss": 0.0858,
+      "step": 594
+    },
+    {
+      "epoch": 0.2384,
+      "learning_rate": 9.595194252782461e-06,
+      "loss": 0.1478,
+      "step": 596
+    },
+    {
+      "epoch": 0.2392,
+      "learning_rate": 9.623098173300656e-06,
+      "loss": 0.022,
+      "step": 598
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 9.651005032974991e-06,
+      "loss": 0.0778,
+      "step": 600
+    },
+    {
+      "epoch": 0.2408,
+      "learning_rate": 9.678914614182184e-06,
+      "loss": 0.0113,
+      "step": 602
+    },
+    {
+      "epoch": 0.2416,
+      "learning_rate": 9.706826699277714e-06,
+      "loss": 0.2379,
+      "step": 604
+    },
+    {
+      "epoch": 0.2424,
+      "learning_rate": 9.734741070597535e-06,
+      "loss": 0.0122,
+      "step": 606
+    },
+    {
+      "epoch": 0.2432,
+      "learning_rate": 9.762657510459774e-06,
+      "loss": 0.0637,
+      "step": 608
+    },
+    {
+      "epoch": 0.244,
+      "learning_rate": 9.790575801166422e-06,
+      "loss": 0.0143,
+      "step": 610
+    },
+    {
+      "epoch": 0.2448,
+      "learning_rate": 9.818495725005043e-06,
+      "loss": 0.2477,
+      "step": 612
+    },
+    {
+      "epoch": 0.2456,
+      "learning_rate": 9.846417064250459e-06,
+      "loss": 1.1105,
+      "step": 614
+    },
+    {
+      "epoch": 0.2464,
+      "learning_rate": 9.874339601166479e-06,
+      "loss": 0.0052,
+      "step": 616
+    },
+    {
+      "epoch": 0.2472,
+      "learning_rate": 9.902263118007513e-06,
+      "loss": 0.0181,
+      "step": 618
+    },
+    {
+      "epoch": 0.248,
+      "learning_rate": 9.930187397020385e-06,
+      "loss": 0.0374,
+      "step": 620
+    },
+    {
+      "epoch": 0.2488,
+      "learning_rate": 9.95811222044596e-06,
+      "loss": 0.089,
+      "step": 622
+    },
+    {
+      "epoch": 0.2496,
+      "learning_rate": 9.986037370520855e-06,
+      "loss": 0.0948,
+      "step": 624
+    },
+    {
+      "epoch": 0.2504,
+      "learning_rate": 1.0013962629479139e-05,
+      "loss": 0.0015,
+      "step": 626
+    },
+    {
+      "epoch": 0.2512,
+      "learning_rate": 1.0041887779554034e-05,
+      "loss": 0.1123,
+      "step": 628
+    },
+    {
+      "epoch": 0.252,
+      "learning_rate": 1.0069812602979607e-05,
+      "loss": 0.0043,
+      "step": 630
+    },
+    {
+      "epoch": 0.2528,
+      "learning_rate": 1.0097736881992482e-05,
+      "loss": 0.0043,
+      "step": 632
+    },
+    {
+      "epoch": 0.2536,
+      "learning_rate": 1.0125660398833514e-05,
+      "loss": 0.0142,
+      "step": 634
+    },
+    {
+      "epoch": 0.2544,
+      "learning_rate": 1.0153582935749533e-05,
+      "loss": 0.0108,
+      "step": 636
+    },
+    {
+      "epoch": 0.2552,
+      "learning_rate": 1.0181504274994952e-05,
+      "loss": 0.1254,
+      "step": 638
+    },
+    {
+      "epoch": 0.256,
+      "learning_rate": 1.0209424198833571e-05,
+      "loss": 0.083,
+      "step": 640
+    },
+    {
+      "epoch": 0.2568,
+      "learning_rate": 1.0237342489540218e-05,
+      "loss": 0.179,
+      "step": 642
+    },
+    {
+      "epoch": 0.2576,
+      "learning_rate": 1.0265258929402458e-05,
+      "loss": 0.0012,
+      "step": 644
+    },
+    {
+      "epoch": 0.2584,
+      "learning_rate": 1.029317330072228e-05,
+      "loss": 0.0353,
+      "step": 646
+    },
+    {
+      "epoch": 0.2592,
+      "learning_rate": 1.0321085385817811e-05,
+      "loss": 0.1669,
+      "step": 648
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.0348994967025004e-05,
+      "loss": 0.0032,
+      "step": 650
+    },
+    {
+      "epoch": 0.2608,
+      "learning_rate": 1.0376901826699337e-05,
+      "loss": 0.01,
+      "step": 652
+    },
+    {
+      "epoch": 0.2616,
+      "learning_rate": 1.0404805747217532e-05,
+      "loss": 0.0292,
+      "step": 654
+    },
+    {
+      "epoch": 0.2624,
+      "learning_rate": 1.0432706510979175e-05,
+      "loss": 0.0006,
+      "step": 656
+    },
+    {
+      "epoch": 0.2632,
+      "learning_rate": 1.0460603900408526e-05,
+      "loss": 0.0018,
+      "step": 658
+    },
+    {
+      "epoch": 0.264,
+      "learning_rate": 1.0488497697956134e-05,
+      "loss": 0.0007,
+      "step": 660
+    },
+    {
+      "epoch": 0.2648,
+      "learning_rate": 1.0516387686100549e-05,
+      "loss": 0.0087,
+      "step": 662
+    },
+    {
+      "epoch": 0.2656,
+      "learning_rate": 1.054427364735009e-05,
+      "loss": 0.058,
+      "step": 664
+    },
+    {
+      "epoch": 0.2664,
+      "learning_rate": 1.0572155364244378e-05,
+      "loss": 0.0566,
+      "step": 666
+    },
+    {
+      "epoch": 0.2672,
+      "learning_rate": 1.0600032619356203e-05,
+      "loss": 0.0087,
+      "step": 668
+    },
+    {
+      "epoch": 0.268,
+      "learning_rate": 1.0627905195293127e-05,
+      "loss": 0.0131,
+      "step": 670
+    },
+    {
+      "epoch": 0.2688,
+      "learning_rate": 1.0655772874699206e-05,
+      "loss": 0.0989,
+      "step": 672
+    },
+    {
+      "epoch": 0.2696,
+      "learning_rate": 1.0683635440256694e-05,
+      "loss": 0.4608,
+      "step": 674
+    },
+    {
+      "epoch": 0.2704,
+      "learning_rate": 1.0711492674687674e-05,
+      "loss": 0.0039,
+      "step": 676
+    },
+    {
+      "epoch": 0.2712,
+      "learning_rate": 1.0739344360755855e-05,
+      "loss": 0.0584,
+      "step": 678
+    },
+    {
+      "epoch": 0.272,
+      "learning_rate": 1.0767190281268171e-05,
+      "loss": 0.0007,
+      "step": 680
+    },
+    {
+      "epoch": 0.2728,
+      "learning_rate": 1.07950302190766e-05,
+      "loss": 0.0037,
+      "step": 682
+    },
+    {
+      "epoch": 0.2736,
+      "learning_rate": 1.0822863957079654e-05,
+      "loss": 0.0758,
+      "step": 684
+    },
+    {
+      "epoch": 0.2744,
+      "learning_rate": 1.0850691278224277e-05,
+      "loss": 0.0008,
+      "step": 686
+    },
+    {
+      "epoch": 0.2752,
+      "learning_rate": 1.0878511965507428e-05,
+      "loss": 0.2452,
+      "step": 688
+    },
+    {
+      "epoch": 0.276,
+      "learning_rate": 1.0906325801977795e-05,
+      "loss": 0.0012,
+      "step": 690
+    },
+    {
+      "epoch": 0.2768,
+      "learning_rate": 1.0934132570737497e-05,
+      "loss": 0.066,
+      "step": 692
+    },
+    {
+      "epoch": 0.2776,
+      "learning_rate": 1.0961932054943785e-05,
+      "loss": 0.002,
+      "step": 694
+    },
+    {
+      "epoch": 0.2784,
+      "learning_rate": 1.098972403781064e-05,
+      "loss": 0.022,
+      "step": 696
+    },
+    {
+      "epoch": 0.2792,
+      "learning_rate": 1.101750830261065e-05,
+      "loss": 0.0027,
+      "step": 698
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.104528463267652e-05,
+      "loss": 0.0079,
+      "step": 700
+    },
+    {
+      "epoch": 0.2808,
+      "learning_rate": 1.1073052811402867e-05,
+      "loss": 0.0002,
+      "step": 702
+    },
+    {
+      "epoch": 0.2816,
+      "learning_rate": 1.1100812622247821e-05,
+      "loss": 1.9113,
+      "step": 704
+    },
+    {
+      "epoch": 0.2824,
+      "learning_rate": 1.1128563848734815e-05,
+      "loss": 0.0007,
+      "step": 706
+    },
+    {
+      "epoch": 0.2832,
+      "learning_rate": 1.1156306274454211e-05,
+      "loss": 0.0143,
+      "step": 708
+    },
+    {
+      "epoch": 0.284,
+      "learning_rate": 1.1184039683065002e-05,
+      "loss": 0.0064,
+      "step": 710
+    },
+    {
+      "epoch": 0.2848,
+      "learning_rate": 1.1211763858296516e-05,
+      "loss": 0.7138,
+      "step": 712
+    },
+    {
+      "epoch": 0.2856,
+      "learning_rate": 1.1239478583950007e-05,
+      "loss": 0.2771,
+      "step": 714
+    },
+    {
+      "epoch": 0.2864,
+      "learning_rate": 1.1267183643900534e-05,
+      "loss": 0.0005,
+      "step": 716
+    },
+    {
+      "epoch": 0.2872,
+      "learning_rate": 1.1294878822098456e-05,
+      "loss": 0.0612,
+      "step": 718
+    },
+    {
+      "epoch": 0.288,
+      "learning_rate": 1.1322563902571227e-05,
+      "loss": 0.0012,
+      "step": 720
+    },
+    {
+      "epoch": 0.2888,
+      "learning_rate": 1.1350238669424993e-05,
+      "loss": 0.0094,
+      "step": 722
+    },
+    {
+      "epoch": 0.2896,
+      "learning_rate": 1.137790290684638e-05,
+      "loss": 0.3117,
+      "step": 724
+    },
+    {
+      "epoch": 0.2904,
+      "learning_rate": 1.1405556399104108e-05,
+      "loss": 0.0388,
+      "step": 726
+    },
+    {
+      "epoch": 0.2912,
+      "learning_rate": 1.143319893055069e-05,
+      "loss": 0.0009,
+      "step": 728
+    },
+    {
+      "epoch": 0.292,
+      "learning_rate": 1.1460830285624112e-05,
+      "loss": 0.0194,
+      "step": 730
+    },
+    {
+      "epoch": 0.2928,
+      "learning_rate": 1.1488450248849515e-05,
+      "loss": 0.0013,
+      "step": 732
+    },
+    {
+      "epoch": 0.2936,
+      "learning_rate": 1.1516058604840881e-05,
+      "loss": 0.0026,
+      "step": 734
+    },
+    {
+      "epoch": 0.2944,
+      "learning_rate": 1.15436551383027e-05,
+      "loss": 0.1476,
+      "step": 736
+    },
+    {
+      "epoch": 0.2952,
+      "learning_rate": 1.1571239634031666e-05,
+      "loss": 0.0232,
+      "step": 738
+    },
+    {
+      "epoch": 0.296,
+      "learning_rate": 1.1598811876918352e-05,
+      "loss": 0.0117,
+      "step": 740
+    },
+    {
+      "epoch": 0.2968,
+      "learning_rate": 1.1626371651948839e-05,
+      "loss": 0.0124,
+      "step": 742
+    },
+    {
+      "epoch": 0.2976,
+      "learning_rate": 1.1653918744206478e-05,
+      "loss": 0.0187,
+      "step": 744
+    },
+    {
+      "epoch": 0.2984,
+      "learning_rate": 1.1681452938873515e-05,
+      "loss": 0.0075,
+      "step": 746
+    },
+    {
+      "epoch": 0.2992,
+      "learning_rate": 1.1708974021232763e-05,
+      "loss": 0.0358,
+      "step": 748
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.1736481776669297e-05,
+      "loss": 0.0565,
+      "step": 750
+    },
+    {
+      "epoch": 0.3008,
+      "learning_rate": 1.1763975990672116e-05,
+      "loss": 0.4557,
+      "step": 752
+    },
+    {
+      "epoch": 0.3016,
+      "learning_rate": 1.1791456448835815e-05,
+      "loss": 0.0485,
+      "step": 754
+    },
+    {
+      "epoch": 0.3024,
+      "learning_rate": 1.1818922936862258e-05,
+      "loss": 0.3115,
+      "step": 756
+    },
+    {
+      "epoch": 0.3032,
+      "learning_rate": 1.1846375240562274e-05,
+      "loss": 0.0016,
+      "step": 758
+    },
+    {
+      "epoch": 0.304,
+      "learning_rate": 1.187381314585725e-05,
+      "loss": 0.0022,
+      "step": 760
+    },
+    {
+      "epoch": 0.3048,
+      "learning_rate": 1.1901236438780906e-05,
+      "loss": 0.0065,
+      "step": 762
+    },
+    {
+      "epoch": 0.3056,
+      "learning_rate": 1.192864490548089e-05,
+      "loss": 0.0273,
+      "step": 764
+    },
+    {
+      "epoch": 0.3064,
+      "learning_rate": 1.195603833222048e-05,
+      "loss": 0.0972,
+      "step": 766
+    },
+    {
+      "epoch": 0.3072,
+      "learning_rate": 1.198341650538023e-05,
+      "loss": 0.0071,
+      "step": 768
+    },
+    {
+      "epoch": 0.308,
+      "learning_rate": 1.2010779211459642e-05,
+      "loss": 0.0115,
+      "step": 770
+    },
+    {
+      "epoch": 0.3088,
+      "learning_rate": 1.203812623707884e-05,
+      "loss": 0.2521,
+      "step": 772
+    },
+    {
+      "epoch": 0.3096,
+      "learning_rate": 1.2065457368980227e-05,
+      "loss": 0.0017,
+      "step": 774
+    },
+    {
+      "epoch": 0.3104,
+      "learning_rate": 1.2092772394030141e-05,
+      "loss": 0.0025,
+      "step": 776
+    },
+    {
+      "epoch": 0.3112,
+      "learning_rate": 1.2120071099220552e-05,
+      "loss": 0.0036,
+      "step": 778
+    },
+    {
+      "epoch": 0.312,
+      "learning_rate": 1.2147353271670637e-05,
+      "loss": 1.2653,
+      "step": 780
+    },
+    {
+      "epoch": 0.3128,
+      "learning_rate": 1.217461869862855e-05,
+      "loss": 0.0179,
+      "step": 782
+    },
+    {
+      "epoch": 0.3136,
+      "learning_rate": 1.2201867167473015e-05,
+      "loss": 0.2921,
+      "step": 784
+    },
+    {
+      "epoch": 0.3144,
+      "learning_rate": 1.2229098465715002e-05,
+      "loss": 0.0273,
+      "step": 786
+    },
+    {
+      "epoch": 0.3152,
+      "learning_rate": 1.2256312380999373e-05,
+      "loss": 0.0678,
+      "step": 788
+    },
+    {
+      "epoch": 0.316,
+      "learning_rate": 1.2283508701106552e-05,
+      "loss": 0.227,
+      "step": 790
+    },
+    {
+      "epoch": 0.3168,
+      "learning_rate": 1.2310687213954173e-05,
+      "loss": 0.0005,
+      "step": 792
+    },
+    {
+      "epoch": 0.3176,
+      "learning_rate": 1.233784770759873e-05,
+      "loss": 0.1092,
+      "step": 794
+    },
+    {
+      "epoch": 0.3184,
+      "learning_rate": 1.2364989970237238e-05,
+      "loss": 0.0441,
+      "step": 796
+    },
+    {
+      "epoch": 0.3192,
+      "learning_rate": 1.23921137902089e-05,
+      "loss": 0.2983,
+      "step": 798
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.241921895599668e-05,
+      "loss": 0.1579,
+      "step": 800
+    },
+    {
+      "epoch": 0.3208,
+      "learning_rate": 1.2446305256229076e-05,
+      "loss": 0.0502,
+      "step": 802
+    },
+    {
+      "epoch": 0.3216,
+      "learning_rate": 1.2473372479681653e-05,
+      "loss": 0.0675,
+      "step": 804
+    },
+    {
+      "epoch": 0.3224,
+      "learning_rate": 1.2500420415278822e-05,
+      "loss": 0.6894,
+      "step": 806
+    },
+    {
+      "epoch": 0.3232,
+      "learning_rate": 1.2527448852095292e-05,
+      "loss": 0.1998,
+      "step": 808
+    },
+    {
+      "epoch": 0.324,
+      "learning_rate": 1.2554457579357902e-05,
+      "loss": 0.0055,
+      "step": 810
+    },
+    {
+      "epoch": 0.3248,
+      "learning_rate": 1.2581446386447171e-05,
+      "loss": 0.9854,
+      "step": 812
+    },
+    {
+      "epoch": 0.3256,
+      "learning_rate": 1.2608415062898963e-05,
+      "loss": 0.0063,
+      "step": 814
+    },
+    {
+      "epoch": 0.3264,
+      "learning_rate": 1.2635363398406133e-05,
+      "loss": 0.2309,
+      "step": 816
+    },
+    {
+      "epoch": 0.3272,
+      "learning_rate": 1.266229118282012e-05,
+      "loss": 0.1528,
+      "step": 818
+    },
+    {
+      "epoch": 0.328,
+      "learning_rate": 1.2689198206152644e-05,
+      "loss": 0.0011,
+      "step": 820
+    },
+    {
+      "epoch": 0.3288,
+      "learning_rate": 1.2716084258577373e-05,
+      "loss": 0.0162,
+      "step": 822
+    },
+    {
+      "epoch": 0.3296,
+      "learning_rate": 1.2742949130431468e-05,
+      "loss": 0.0097,
+      "step": 824
+    },
+    {
+      "epoch": 0.3304,
+      "learning_rate": 1.2769792612217224e-05,
+      "loss": 0.2707,
+      "step": 826
+    },
+    {
+      "epoch": 0.3312,
+      "learning_rate": 1.2796614494603795e-05,
+      "loss": 0.2288,
+      "step": 828
+    },
+    {
+      "epoch": 0.332,
+      "learning_rate": 1.282341456842876e-05,
+      "loss": 0.0007,
+      "step": 830
+    },
+    {
+      "epoch": 0.3328,
+      "learning_rate": 1.2850192624699756e-05,
+      "loss": 0.0066,
+      "step": 832
+    },
+    {
+      "epoch": 0.3336,
+      "learning_rate": 1.2876948454596122e-05,
+      "loss": 0.0135,
+      "step": 834
+    },
+    {
+      "epoch": 0.3344,
+      "learning_rate": 1.2903681849470535e-05,
+      "loss": 0.2187,
+      "step": 836
+    },
+    {
+      "epoch": 0.3352,
+      "learning_rate": 1.2930392600850565e-05,
+      "loss": 0.0007,
+      "step": 838
+    },
+    {
+      "epoch": 0.336,
+      "learning_rate": 1.2957080500440455e-05,
+      "loss": 0.0367,
+      "step": 840
+    },
+    {
+      "epoch": 0.3368,
+      "learning_rate": 1.2983745340122589e-05,
+      "loss": 0.604,
+      "step": 842
+    },
+    {
+      "epoch": 0.3376,
+      "learning_rate": 1.3010386911959205e-05,
+      "loss": 0.1659,
+      "step": 844
+    },
+    {
+      "epoch": 0.3384,
+      "learning_rate": 1.3037005008193944e-05,
+      "loss": 0.4783,
+      "step": 846
+    },
+    {
+      "epoch": 0.3392,
+      "learning_rate": 1.3063599421253556e-05,
+      "loss": 0.0374,
+      "step": 848
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 1.309016994374947e-05,
+      "loss": 0.0644,
+      "step": 850
+    },
+    {
+      "epoch": 0.3408,
+      "learning_rate": 1.3116716368479415e-05,
+      "loss": 0.0113,
+      "step": 852
+    },
+    {
+      "epoch": 0.3416,
+      "learning_rate": 1.3143238488429049e-05,
+      "loss": 0.1769,
+      "step": 854
+    },
+    {
+      "epoch": 0.3424,
+      "learning_rate": 1.316973609677351e-05,
+      "loss": 0.006,
+      "step": 856
+    },
+    {
+      "epoch": 0.3432,
+      "learning_rate": 1.319620898687917e-05,
+      "loss": 0.0015,
+      "step": 858
+    },
+    {
+      "epoch": 0.344,
+      "learning_rate": 1.32226569523051e-05,
+      "loss": 0.0007,
+      "step": 860
+    },
+    {
+      "epoch": 0.3448,
+      "learning_rate": 1.324907978680475e-05,
+      "loss": 0.0015,
+      "step": 862
+    },
+    {
+      "epoch": 0.3456,
+      "learning_rate": 1.3275477284327572e-05,
+      "loss": 0.0753,
+      "step": 864
+    },
+    {
+      "epoch": 0.3464,
+      "learning_rate": 1.3301849239020537e-05,
+      "loss": 0.029,
+      "step": 866
+    },
+    {
+      "epoch": 0.3472,
+      "learning_rate": 1.3328195445229865e-05,
+      "loss": 0.0157,
+      "step": 868
+    },
+    {
+      "epoch": 0.348,
+      "learning_rate": 1.3354515697502548e-05,
+      "loss": 0.0014,
+      "step": 870
+    },
+    {
+      "epoch": 0.3488,
+      "learning_rate": 1.338080979058797e-05,
+      "loss": 0.1665,
+      "step": 872
+    },
+    {
+      "epoch": 0.3496,
+      "learning_rate": 1.340707751943951e-05,
+      "loss": 0.0365,
+      "step": 874
+    },
+    {
+      "epoch": 0.3504,
+      "learning_rate": 1.3433318679216145e-05,
+      "loss": 0.3169,
+      "step": 876
+    },
+    {
+      "epoch": 0.3512,
+      "learning_rate": 1.3459533065284039e-05,
+      "loss": 0.0014,
+      "step": 878
+    },
+    {
+      "epoch": 0.352,
+      "learning_rate": 1.348572047321814e-05,
+      "loss": 0.0664,
+      "step": 880
+    },
+    {
+      "epoch": 0.3528,
+      "learning_rate": 1.3511880698803803e-05,
+      "loss": 0.1472,
+      "step": 882
+    },
+    {
+      "epoch": 0.3536,
+      "learning_rate": 1.3538013538038296e-05,
+      "loss": 0.0087,
+      "step": 884
+    },
+    {
+      "epoch": 0.3544,
+      "learning_rate": 1.3564118787132507e-05,
+      "loss": 0.0016,
+      "step": 886
+    },
+    {
+      "epoch": 0.3552,
+      "learning_rate": 1.3590196242512461e-05,
+      "loss": 0.0363,
+      "step": 888
+    },
+    {
+      "epoch": 0.356,
+      "learning_rate": 1.361624570082092e-05,
+      "loss": 0.0002,
+      "step": 890
+    },
+    {
+      "epoch": 0.3568,
+      "learning_rate": 1.364226695891898e-05,
+      "loss": 0.0014,
+      "step": 892
+    },
+    {
+      "epoch": 0.3576,
+      "learning_rate": 1.3668259813887637e-05,
+      "loss": 0.0156,
+      "step": 894
+    },
+    {
+      "epoch": 0.3584,
+      "learning_rate": 1.3694224063029386e-05,
+      "loss": 0.003,
+      "step": 896
+    },
+    {
+      "epoch": 0.3592,
+      "learning_rate": 1.3720159503869806e-05,
+      "loss": 0.003,
+      "step": 898
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 1.374606593415911e-05,
+      "loss": 0.0041,
+      "step": 900
+    },
+    {
+      "epoch": 0.3608,
+      "learning_rate": 1.377194315187377e-05,
+      "loss": 0.0173,
+      "step": 902
+    },
+    {
+      "epoch": 0.3616,
+      "learning_rate": 1.3797790955218014e-05,
+      "loss": 0.0012,
+      "step": 904
+    },
+    {
+      "epoch": 0.3624,
+      "learning_rate": 1.3823609142625492e-05,
+      "loss": 0.004,
+      "step": 906
+    },
+    {
+      "epoch": 0.3632,
+      "learning_rate": 1.3849397512760793e-05,
+      "loss": 0.6563,
+      "step": 908
+    },
+    {
+      "epoch": 0.364,
+      "learning_rate": 1.3875155864521027e-05,
+      "loss": 0.0011,
+      "step": 910
+    },
+    {
+      "epoch": 0.3648,
+      "learning_rate": 1.3900883997037393e-05,
+      "loss": 0.1612,
+      "step": 912
+    },
+    {
+      "epoch": 0.3656,
+      "learning_rate": 1.3926581709676746e-05,
+      "loss": 0.0024,
+      "step": 914
+    },
+    {
+      "epoch": 0.3664,
+      "learning_rate": 1.3952248802043158e-05,
+      "loss": 0.4061,
+      "step": 916
+    },
+    {
+      "epoch": 0.3672,
+      "learning_rate": 1.397788507397949e-05,
+      "loss": 0.0048,
+      "step": 918
+    },
+    {
+      "epoch": 0.368,
+      "learning_rate": 1.4003490325568956e-05,
+      "loss": 0.0213,
+      "step": 920
+    },
+    {
+      "epoch": 0.3688,
+      "learning_rate": 1.4029064357136632e-05,
+      "loss": 0.0237,
+      "step": 922
+    },
+    {
+      "epoch": 0.3696,
+      "learning_rate": 1.4054606969251096e-05,
+      "loss": 0.4146,
+      "step": 924
+    },
+    {
+      "epoch": 0.3704,
+      "learning_rate": 1.4080117962725929e-05,
+      "loss": 0.0252,
+      "step": 926
+    },
+    {
+      "epoch": 0.3712,
+      "learning_rate": 1.4105597138621281e-05,
+      "loss": 0.0316,
+      "step": 928
+    },
+    {
+      "epoch": 0.372,
+      "learning_rate": 1.4131044298245416e-05,
+      "loss": 0.0291,
+      "step": 930
+    },
+    {
+      "epoch": 0.3728,
+      "learning_rate": 1.4156459243156275e-05,
+      "loss": 0.7741,
+      "step": 932
+    },
+    {
+      "epoch": 0.3736,
+      "learning_rate": 1.418184177516301e-05,
+      "loss": 0.0015,
+      "step": 934
+    },
+    {
+      "epoch": 0.3744,
+      "learning_rate": 1.420719169632754e-05,
+      "loss": 0.0003,
+      "step": 936
+    },
+    {
+      "epoch": 0.3752,
+      "learning_rate": 1.4232508808966085e-05,
+      "loss": 0.0707,
+      "step": 938
+    },
+    {
+      "epoch": 0.376,
+      "learning_rate": 1.4257792915650735e-05,
+      "loss": 0.0455,
+      "step": 940
+    },
+    {
+      "epoch": 0.3768,
+      "learning_rate": 1.4283043819210906e-05,
+      "loss": 0.0389,
+      "step": 942
+    },
+    {
+      "epoch": 0.3776,
+      "learning_rate": 1.430826132273499e-05,
+      "loss": 0.1369,
+      "step": 944
+    },
+    {
+      "epoch": 0.3784,
+      "learning_rate": 1.4333445229571857e-05,
+      "loss": 0.1176,
+      "step": 946
+    },
+    {
+      "epoch": 0.3792,
+      "learning_rate": 1.4358595343332342e-05,
+      "loss": 0.2449,
+      "step": 948
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 1.4383711467890772e-05,
+      "loss": 0.8199,
+      "step": 950
+    },
+    {
+      "epoch": 0.3808,
+      "learning_rate": 1.4408793407386584e-05,
+      "loss": 0.0006,
+      "step": 952
+    },
+    {
+      "epoch": 0.3816,
+      "learning_rate": 1.4433840966225767e-05,
+      "loss": 0.0957,
+      "step": 954
+    },
+    {
+      "epoch": 0.3824,
+      "learning_rate": 1.4458853949082434e-05,
+      "loss": 0.0109,
+      "step": 956
+    },
+    {
+      "epoch": 0.3832,
+      "learning_rate": 1.4483832160900332e-05,
+      "loss": 0.864,
+      "step": 958
+    },
+    {
+      "epoch": 0.384,
+      "learning_rate": 1.4508775406894315e-05,
+      "loss": 0.1186,
+      "step": 960
+    },
+    {
+      "epoch": 0.3848,
+      "learning_rate": 1.4533683492551942e-05,
+      "loss": 0.0543,
+      "step": 962
+    },
+    {
+      "epoch": 0.3856,
+      "learning_rate": 1.4558556223634988e-05,
+      "loss": 0.0087,
+      "step": 964
+    },
+    {
+      "epoch": 0.3864,
+      "learning_rate": 1.4583393406180886e-05,
+      "loss": 0.8009,
+      "step": 966
+    },
+    {
+      "epoch": 0.3872,
+      "learning_rate": 1.460819484650431e-05,
+      "loss": 0.0273,
+      "step": 968
+    },
+    {
+      "epoch": 0.388,
+      "learning_rate": 1.4632960351198618e-05,
+      "loss": 0.3166,
+      "step": 970
+    },
+    {
+      "epoch": 0.3888,
+      "learning_rate": 1.4657689727137441e-05,
+      "loss": 0.0277,
+      "step": 972
+    },
+    {
+      "epoch": 0.3896,
+      "learning_rate": 1.468238278147614e-05,
+      "loss": 1.1738,
+      "step": 974
+    },
+    {
+      "epoch": 0.3904,
+      "learning_rate": 1.470703932165332e-05,
+      "loss": 0.0129,
+      "step": 976
+    },
+    {
+      "epoch": 0.3912,
+      "learning_rate": 1.4731659155392339e-05,
+      "loss": 0.0066,
+      "step": 978
+    },
+    {
+      "epoch": 0.392,
+      "learning_rate": 1.4756242090702744e-05,
+      "loss": 0.0411,
+      "step": 980
+    },
+    {
+      "epoch": 0.3928,
+      "learning_rate": 1.4780787935881913e-05,
+      "loss": 0.0114,
+      "step": 982
+    },
+    {
+      "epoch": 0.3936,
+      "learning_rate": 1.4805296499516397e-05,
+      "loss": 0.8405,
+      "step": 984
+    },
+    {
+      "epoch": 0.3944,
+      "learning_rate": 1.482976759048351e-05,
+      "loss": 0.0832,
+      "step": 986
+    },
+    {
+      "epoch": 0.3952,
+      "learning_rate": 1.485420101795274e-05,
+      "loss": 0.189,
+      "step": 988
+    },
+    {
+      "epoch": 0.396,
+      "learning_rate": 1.4878596591387327e-05,
+      "loss": 0.1741,
+      "step": 990
+    },
+    {
+      "epoch": 0.3968,
+      "learning_rate": 1.4902954120545686e-05,
+      "loss": 0.0904,
+      "step": 992
+    },
+    {
+      "epoch": 0.3976,
+      "learning_rate": 1.4927273415482913e-05,
+      "loss": 0.2758,
+      "step": 994
+    },
+    {
+      "epoch": 0.3984,
+      "learning_rate": 1.4951554286552261e-05,
+      "loss": 0.1394,
+      "step": 996
+    },
+    {
+      "epoch": 0.3992,
+      "learning_rate": 1.4975796544406617e-05,
+      "loss": 0.0227,
+      "step": 998
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 1.4999999999999992e-05,
+      "loss": 0.0007,
+      "step": 1000
+    },
+    {
+      "epoch": 0.4008,
+      "learning_rate": 1.502416446458897e-05,
+      "loss": 0.6995,
+      "step": 1002
+    },
+    {
+      "epoch": 0.4016,
+      "learning_rate": 1.5048289749734206e-05,
+      "loss": 0.1991,
+      "step": 1004
+    },
+    {
+      "epoch": 0.4024,
+      "learning_rate": 1.5072375667301895e-05,
+      "loss": 0.8636,
+      "step": 1006
+    },
+    {
+      "epoch": 0.4032,
+      "learning_rate": 1.5096422029465178e-05,
+      "loss": 0.0666,
+      "step": 1008
+    },
+    {
+      "epoch": 0.404,
+      "learning_rate": 1.5120428648705714e-05,
+      "loss": 0.2682,
+      "step": 1010
+    },
+    {
+      "epoch": 0.4048,
+      "learning_rate": 1.5144395337815064e-05,
+      "loss": 0.2553,
+      "step": 1012
+    },
+    {
+      "epoch": 0.4056,
+      "learning_rate": 1.5168321909896166e-05,
+      "loss": 0.1212,
+      "step": 1014
+    },
+    {
+      "epoch": 0.4064,
+      "learning_rate": 1.5192208178364808e-05,
+      "loss": 0.1705,
+      "step": 1016
+    },
+    {
+      "epoch": 0.4072,
+      "learning_rate": 1.521605395695107e-05,
+      "loss": 0.0247,
+      "step": 1018
+    },
+    {
+      "epoch": 0.408,
+      "learning_rate": 1.5239859059700784e-05,
+      "loss": 0.3914,
+      "step": 1020
+    },
+    {
+      "epoch": 0.4088,
+      "learning_rate": 1.526362330097697e-05,
+      "loss": 0.0678,
+      "step": 1022
+    },
+    {
+      "epoch": 0.4096,
+      "learning_rate": 1.5287346495461322e-05,
+      "loss": 0.0016,
+      "step": 1024
+    },
+    {
+      "epoch": 0.4104,
+      "learning_rate": 1.531102845815557e-05,
+      "loss": 0.0028,
+      "step": 1026
+    },
+    {
+      "epoch": 0.4112,
+      "learning_rate": 1.5334669004383025e-05,
+      "loss": 0.0065,
+      "step": 1028
+    },
+    {
+      "epoch": 0.412,
+      "learning_rate": 1.5358267949789968e-05,
+      "loss": 0.0158,
+      "step": 1030
+    },
+    {
+      "epoch": 0.4128,
+      "learning_rate": 1.5381825110347072e-05,
+      "loss": 0.0158,
+      "step": 1032
+    },
+    {
+      "epoch": 0.4136,
+      "learning_rate": 1.540534030235087e-05,
+      "loss": 0.4769,
+      "step": 1034
+    },
+    {
+      "epoch": 0.4144,
+      "learning_rate": 1.542881334242517e-05,
+      "loss": 0.024,
+      "step": 1036
+    },
+    {
+      "epoch": 0.4152,
+      "learning_rate": 1.5452244047522493e-05,
+      "loss": 0.211,
+      "step": 1038
+    },
+    {
+      "epoch": 0.416,
+      "learning_rate": 1.5475632234925495e-05,
+      "loss": 0.5014,
+      "step": 1040
+    },
+    {
+      "epoch": 0.4168,
+      "learning_rate": 1.5498977722248388e-05,
+      "loss": 0.7562,
+      "step": 1042
+    },
+    {
+      "epoch": 0.4176,
+      "learning_rate": 1.552228032743839e-05,
+      "loss": 0.0957,
+      "step": 1044
+    },
+    {
+      "epoch": 0.4184,
+      "learning_rate": 1.5545539868777075e-05,
+      "loss": 0.9688,
+      "step": 1046
+    },
+    {
+      "epoch": 0.4192,
+      "learning_rate": 1.556875616488188e-05,
+      "loss": 0.1644,
+      "step": 1048
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 1.5591929034707468e-05,
+      "loss": 0.0653,
+      "step": 1050
+    },
+    {
+      "epoch": 0.4208,
+      "learning_rate": 1.5615058297547144e-05,
+      "loss": 0.7905,
+      "step": 1052
+    },
+    {
+      "epoch": 0.4216,
+      "learning_rate": 1.5638143773034268e-05,
+      "loss": 0.1003,
+      "step": 1054
+    },
+    {
+      "epoch": 0.4224,
+      "learning_rate": 1.5661185281143663e-05,
+      "loss": 0.0503,
+      "step": 1056
+    },
+    {
+      "epoch": 0.4232,
+      "learning_rate": 1.5684182642193024e-05,
+      "loss": 0.0621,
+      "step": 1058
+    },
+    {
+      "epoch": 0.424,
+      "learning_rate": 1.5707135676844312e-05,
+      "loss": 0.0158,
+      "step": 1060
+    },
+    {
+      "epoch": 0.4248,
+      "learning_rate": 1.5730044206105146e-05,
+      "loss": 0.0271,
+      "step": 1062
+    },
+    {
+      "epoch": 0.4256,
+      "learning_rate": 1.5752908051330232e-05,
+      "loss": 0.0148,
+      "step": 1064
+    },
+    {
+      "epoch": 0.4264,
+      "learning_rate": 1.577572703422268e-05,
+      "loss": 0.6018,
+      "step": 1066
+    },
+    {
+      "epoch": 0.4272,
+      "learning_rate": 1.579850097683548e-05,
+      "loss": 0.0818,
+      "step": 1068
+    },
+    {
+      "epoch": 0.428,
+      "learning_rate": 1.582122970157288e-05,
+      "loss": 0.1039,
+      "step": 1070
+    },
+    {
+      "epoch": 0.4288,
+      "learning_rate": 1.5843913031191722e-05,
+      "loss": 0.0046,
+      "step": 1072
+    },
+    {
+      "epoch": 0.4296,
+      "learning_rate": 1.586655078880281e-05,
+      "loss": 0.0247,
+      "step": 1074
+    },
+    {
+      "epoch": 0.4304,
+      "learning_rate": 1.5889142797872383e-05,
+      "loss": 0.0405,
+      "step": 1076
+    },
+    {
+      "epoch": 0.4312,
+      "learning_rate": 1.5911688882223415e-05,
+      "loss": 0.1715,
+      "step": 1078
+    },
+    {
+      "epoch": 0.432,
+      "learning_rate": 1.5934188866037007e-05,
+      "loss": 0.0853,
+      "step": 1080
+    },
+    {
+      "epoch": 0.4328,
+      "learning_rate": 1.5956642573853787e-05,
+      "loss": 0.1593,
+      "step": 1082
+    },
+    {
+      "epoch": 0.4336,
+      "learning_rate": 1.5979049830575193e-05,
+      "loss": 0.0041,
+      "step": 1084
+    },
+    {
+      "epoch": 0.4344,
+      "learning_rate": 1.6001410461464945e-05,
+      "loss": 0.0077,
+      "step": 1086
+    },
+    {
+      "epoch": 0.4352,
+      "learning_rate": 1.6023724292150377e-05,
+      "loss": 0.0052,
+      "step": 1088
+    },
+    {
+      "epoch": 0.436,
+      "learning_rate": 1.604599114862375e-05,
+      "loss": 0.0109,
+      "step": 1090
+    },
+    {
+      "epoch": 0.4368,
+      "learning_rate": 1.606821085724362e-05,
+      "loss": 0.1346,
+      "step": 1092
+    },
+    {
+      "epoch": 0.4376,
+      "learning_rate": 1.6090383244736253e-05,
+      "loss": 0.0207,
+      "step": 1094
+    },
+    {
+      "epoch": 0.4384,
+      "learning_rate": 1.6112508138196912e-05,
+      "loss": 0.0765,
+      "step": 1096
+    },
+    {
+      "epoch": 0.4392,
+      "learning_rate": 1.613458536509124e-05,
+      "loss": 0.0292,
+      "step": 1098
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 1.615661475325658e-05,
+      "loss": 0.0049,
+      "step": 1100
+    },
+    {
+      "epoch": 0.4408,
+      "learning_rate": 1.6178596130903352e-05,
+      "loss": 0.0016,
+      "step": 1102
+    },
+    {
+      "epoch": 0.4416,
+      "learning_rate": 1.620052932661632e-05,
+      "loss": 0.0067,
+      "step": 1104
+    },
+    {
+      "epoch": 0.4424,
+      "learning_rate": 1.6222414169356056e-05,
+      "loss": 0.0091,
+      "step": 1106
+    },
+    {
+      "epoch": 0.4432,
+      "learning_rate": 1.6244250488460146e-05,
+      "loss": 0.0003,
+      "step": 1108
+    },
+    {
+      "epoch": 0.444,
+      "learning_rate": 1.6266038113644605e-05,
+      "loss": 0.0207,
+      "step": 1110
+    },
+    {
+      "epoch": 0.4448,
+      "learning_rate": 1.6287776875005127e-05,
+      "loss": 0.0036,
+      "step": 1112
+    },
+    {
+      "epoch": 0.4456,
+      "learning_rate": 1.6309466603018497e-05,
+      "loss": 1.3406,
+      "step": 1114
+    },
+    {
+      "epoch": 0.4464,
+      "learning_rate": 1.6331107128543856e-05,
+      "loss": 0.107,
+      "step": 1116
+    },
+    {
+      "epoch": 0.4472,
+      "learning_rate": 1.635269828282404e-05,
+      "loss": 0.6438,
+      "step": 1118
+    },
+    {
+      "epoch": 0.448,
+      "learning_rate": 1.6374239897486905e-05,
+      "loss": 0.2638,
+      "step": 1120
+    },
+    {
+      "epoch": 0.4488,
+      "learning_rate": 1.6395731804546575e-05,
+      "loss": 0.4416,
+      "step": 1122
+    },
+    {
+      "epoch": 0.4496,
+      "learning_rate": 1.6417173836404878e-05,
+      "loss": 0.0075,
+      "step": 1124
+    },
+    {
+      "epoch": 0.4504,
+      "learning_rate": 1.643856582585253e-05,
+      "loss": 0.0039,
+      "step": 1126
+    },
+    {
+      "epoch": 0.4512,
+      "learning_rate": 1.6459907606070513e-05,
+      "loss": 0.1285,
+      "step": 1128
+    },
+    {
+      "epoch": 0.452,
+      "learning_rate": 1.6481199010631312e-05,
+      "loss": 0.0005,
+      "step": 1130
+    },
+    {
+      "epoch": 0.4528,
+      "learning_rate": 1.650243987350029e-05,
+      "loss": 0.0033,
+      "step": 1132
+    },
+    {
+      "epoch": 0.4536,
+      "learning_rate": 1.652363002903693e-05,
+      "loss": 0.0251,
+      "step": 1134
+    },
+    {
+      "epoch": 0.4544,
+      "learning_rate": 1.6544769311996146e-05,
+      "loss": 0.0014,
+      "step": 1136
+    },
+    {
+      "epoch": 0.4552,
+      "learning_rate": 1.656585755752956e-05,
+      "loss": 0.0683,
+      "step": 1138
+    },
+    {
+      "epoch": 0.456,
+      "learning_rate": 1.65868946011868e-05,
+      "loss": 0.0005,
+      "step": 1140
+    },
+    {
+      "epoch": 0.4568,
+      "learning_rate": 1.660788027891677e-05,
+      "loss": 0.0002,
+      "step": 1142
+    },
+    {
+      "epoch": 0.4576,
+      "learning_rate": 1.6628814427068944e-05,
+      "loss": 0.0422,
+      "step": 1144
+    },
+    {
+      "epoch": 0.4584,
+      "learning_rate": 1.6649696882394625e-05,
+      "loss": 0.0005,
+      "step": 1146
+    },
+    {
+      "epoch": 0.4592,
+      "learning_rate": 1.667052748204825e-05,
+      "loss": 1.2971,
+      "step": 1148
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 1.6691306063588583e-05,
+      "loss": 0.6126,
+      "step": 1150
+    },
+    {
+      "epoch": 0.4608,
+      "learning_rate": 1.6712032464980094e-05,
+      "loss": 0.2187,
+      "step": 1152
+    },
+    {
+      "epoch": 0.4616,
+      "learning_rate": 1.6732706524594138e-05,
+      "loss": 0.1372,
+      "step": 1154
+    },
+    {
+      "epoch": 0.4624,
+      "learning_rate": 1.6753328081210244e-05,
+      "loss": 0.1888,
+      "step": 1156
+    },
+    {
+      "epoch": 0.4632,
+      "learning_rate": 1.6773896974017373e-05,
+      "loss": 0.0294,
+      "step": 1158
+    },
+    {
+      "epoch": 0.464,
+      "learning_rate": 1.679441304261516e-05,
+      "loss": 0.0481,
+      "step": 1160
+    },
+    {
+      "epoch": 0.4648,
+      "learning_rate": 1.681487612701519e-05,
+      "loss": 0.011,
+      "step": 1162
+    },
+    {
+      "epoch": 0.4656,
+      "learning_rate": 1.683528606764222e-05,
+      "loss": 0.0404,
+      "step": 1164
+    },
+    {
+      "epoch": 0.4664,
+      "learning_rate": 1.6855642705335428e-05,
+      "loss": 0.0818,
+      "step": 1166
+    },
+    {
+      "epoch": 0.4672,
+      "learning_rate": 1.687594588134968e-05,
+      "loss": 0.0033,
+      "step": 1168
+    },
+    {
+      "epoch": 0.468,
+      "learning_rate": 1.68961954373567e-05,
+      "loss": 0.3934,
+      "step": 1170
+    },
+    {
+      "epoch": 0.4688,
+      "learning_rate": 1.6916391215446403e-05,
+      "loss": 0.0475,
+      "step": 1172
+    },
+    {
+      "epoch": 0.4696,
+      "learning_rate": 1.693653305812805e-05,
+      "loss": 0.028,
+      "step": 1174
+    },
+    {
+      "epoch": 0.4704,
+      "learning_rate": 1.6956620808331505e-05,
+      "loss": 1.1289,
+      "step": 1176
+    },
+    {
+      "epoch": 0.4712,
+      "learning_rate": 1.697665430940846e-05,
+      "loss": 0.002,
+      "step": 1178
+    },
+    {
+      "epoch": 0.472,
+      "learning_rate": 1.699663340513365e-05,
+      "loss": 0.5752,
+      "step": 1180
+    },
+    {
+      "epoch": 0.4728,
+      "learning_rate": 1.7016557939706068e-05,
+      "loss": 0.8214,
+      "step": 1182
+    },
+    {
+      "epoch": 0.4736,
+      "learning_rate": 1.7036427757750198e-05,
+      "loss": 0.0217,
+      "step": 1184
+    },
+    {
+      "epoch": 0.4744,
+      "learning_rate": 1.7056242704317212e-05,
+      "loss": 0.5819,
+      "step": 1186
+    },
+    {
+      "epoch": 0.4752,
+      "learning_rate": 1.7076002624886156e-05,
+      "loss": 0.0412,
+      "step": 1188
+    },
+    {
+      "epoch": 0.476,
+      "learning_rate": 1.709570736536521e-05,
+      "loss": 0.0078,
+      "step": 1190
+    },
+    {
+      "epoch": 0.4768,
+      "learning_rate": 1.7115356772092844e-05,
+      "loss": 0.1368,
+      "step": 1192
+    },
+    {
+      "epoch": 0.4776,
+      "learning_rate": 1.7134950691839063e-05,
+      "loss": 0.3109,
+      "step": 1194
+    },
+    {
+      "epoch": 0.4784,
+      "learning_rate": 1.7154488971806518e-05,
+      "loss": 0.3393,
+      "step": 1196
+    },
+    {
+      "epoch": 0.4792,
+      "learning_rate": 1.7173971459631783e-05,
+      "loss": 0.1565,
+      "step": 1198
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 1.7193398003386507e-05,
+      "loss": 0.0589,
+      "step": 1200
+    },
+    {
+      "epoch": 0.4808,
+      "learning_rate": 1.7212768451578602e-05,
+      "loss": 0.5378,
+      "step": 1202
+    },
+    {
+      "epoch": 0.4816,
+      "learning_rate": 1.7232082653153416e-05,
+      "loss": 0.4679,
+      "step": 1204
+    },
+    {
+      "epoch": 0.4824,
+      "learning_rate": 1.7251340457494937e-05,
+      "loss": 0.023,
+      "step": 1206
+    },
+    {
+      "epoch": 0.4832,
+      "learning_rate": 1.7270541714426923e-05,
+      "loss": 0.0611,
+      "step": 1208
+    },
+    {
+      "epoch": 0.484,
+      "learning_rate": 1.7289686274214106e-05,
+      "loss": 0.3156,
+      "step": 1210
+    },
+    {
+      "epoch": 0.4848,
+      "learning_rate": 1.7308773987563393e-05,
+      "loss": 0.4025,
+      "step": 1212
+    },
+    {
+      "epoch": 0.4856,
+      "learning_rate": 1.732780470562496e-05,
+      "loss": 0.2821,
+      "step": 1214
+    },
+    {
+      "epoch": 0.4864,
+      "learning_rate": 1.7346778279993413e-05,
+      "loss": 0.1697,
+      "step": 1216
+    },
+    {
+      "epoch": 0.4872,
+      "learning_rate": 1.736569456270903e-05,
+      "loss": 0.0026,
+      "step": 1218
+    },
+    {
+      "epoch": 0.488,
+      "learning_rate": 1.7384553406258836e-05,
+      "loss": 0.004,
+      "step": 1220
+    },
+    {
+      "epoch": 0.4888,
+      "learning_rate": 1.740335466357778e-05,
+      "loss": 0.2519,
+      "step": 1222
+    },
+    {
+      "epoch": 0.4896,
+      "learning_rate": 1.7422098188049888e-05,
+      "loss": 0.0024,
+      "step": 1224
+    },
+    {
+      "epoch": 0.4904,
+      "learning_rate": 1.7440783833509373e-05,
+      "loss": 0.0013,
+      "step": 1226
+    },
+    {
+      "epoch": 0.4912,
+      "learning_rate": 1.7459411454241816e-05,
+      "loss": 0.0287,
+      "step": 1228
+    },
+    {
+      "epoch": 0.492,
+      "learning_rate": 1.747798090498531e-05,
+      "loss": 0.0411,
+      "step": 1230
+    },
+    {
+      "epoch": 0.4928,
+      "learning_rate": 1.749649204093154e-05,
+      "loss": 0.1829,
+      "step": 1232
+    },
+    {
+      "epoch": 0.4936,
+      "learning_rate": 1.7514944717726962e-05,
+      "loss": 0.006,
+      "step": 1234
+    },
+    {
+      "epoch": 0.4944,
+      "learning_rate": 1.753333879147387e-05,
+      "loss": 0.0149,
+      "step": 1236
+    },
+    {
+      "epoch": 0.4952,
+      "learning_rate": 1.755167411873159e-05,
+      "loss": 0.0105,
+      "step": 1238
+    },
+    {
+      "epoch": 0.496,
+      "learning_rate": 1.7569950556517563e-05,
+      "loss": 0.07,
+      "step": 1240
+    },
+    {
+      "epoch": 0.4968,
+      "learning_rate": 1.758816796230845e-05,
+      "loss": 0.0456,
+      "step": 1242
+    },
+    {
+      "epoch": 0.4976,
+      "learning_rate": 1.7606326194041278e-05,
+      "loss": 0.0071,
+      "step": 1244
+    },
+    {
+      "epoch": 0.4984,
+      "learning_rate": 1.762442511011447e-05,
+      "loss": 0.0006,
+      "step": 1246
+    },
+    {
+      "epoch": 0.4992,
+      "learning_rate": 1.7642464569389083e-05,
+      "loss": 0.0042,
+      "step": 1248
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 1.766044443118977e-05,
+      "loss": 0.2683,
+      "step": 1250
+    },
+    {
+      "epoch": 0.5008,
+      "learning_rate": 1.767836455530598e-05,
+      "loss": 0.0009,
+      "step": 1252
+    },
+    {
+      "epoch": 0.5016,
+      "learning_rate": 1.7696224801992947e-05,
+      "loss": 0.0747,
+      "step": 1254
+    },
+    {
+      "epoch": 0.5024,
+      "learning_rate": 1.77140250319729e-05,
+      "loss": 0.0087,
+      "step": 1256
+    },
+    {
+      "epoch": 0.5032,
+      "learning_rate": 1.7731765106436073e-05,
+      "loss": 0.0002,
+      "step": 1258
+    },
+    {
+      "epoch": 0.504,
+      "learning_rate": 1.7749444887041793e-05,
+      "loss": 0.0036,
+      "step": 1260
+    },
+    {
+      "epoch": 0.5048,
+      "learning_rate": 1.776706423591959e-05,
+      "loss": 0.0039,
+      "step": 1262
+    },
+    {
+      "epoch": 0.5056,
+      "learning_rate": 1.778462301567023e-05,
+      "loss": 0.0016,
+      "step": 1264
+    },
+    {
+      "epoch": 0.5064,
+      "learning_rate": 1.7802121089366832e-05,
+      "loss": 0.0084,
+      "step": 1266
+    },
+    {
+      "epoch": 0.5072,
+      "learning_rate": 1.7819558320555895e-05,
+      "loss": 0.0001,
+      "step": 1268
+    },
+    {
+      "epoch": 0.508,
+      "learning_rate": 1.7836934573258392e-05,
+      "loss": 0.0192,
+      "step": 1270
+    },
+    {
+      "epoch": 0.5088,
+      "learning_rate": 1.785424971197082e-05,
+      "loss": 0.0047,
+      "step": 1272
+    },
+    {
+      "epoch": 0.5096,
+      "learning_rate": 1.7871503601666233e-05,
+      "loss": 0.0763,
+      "step": 1274
+    },
+    {
+      "epoch": 0.5104,
+      "learning_rate": 1.7888696107795343e-05,
+      "loss": 0.0017,
+      "step": 1276
+    },
+    {
+      "epoch": 0.5112,
+      "learning_rate": 1.790582709628753e-05,
+      "loss": 0.0333,
+      "step": 1278
+    },
+    {
+      "epoch": 0.512,
+      "learning_rate": 1.7922896433551903e-05,
+      "loss": 0.0246,
+      "step": 1280
+    },
+    {
+      "epoch": 0.5128,
+      "learning_rate": 1.793990398647835e-05,
+      "loss": 0.0006,
+      "step": 1282
+    },
+    {
+      "epoch": 0.5136,
+      "learning_rate": 1.795684962243855e-05,
+      "loss": 0.0006,
+      "step": 1284
+    },
+    {
+      "epoch": 0.5144,
+      "learning_rate": 1.7973733209287032e-05,
+      "loss": 0.0152,
+      "step": 1286
+    },
+    {
+      "epoch": 0.5152,
+      "learning_rate": 1.7990554615362193e-05,
+      "loss": 0.0005,
+      "step": 1288
+    },
+    {
+      "epoch": 0.516,
+      "learning_rate": 1.800731370948734e-05,
+      "loss": 0.0001,
+      "step": 1290
+    },
+    {
+      "epoch": 0.5168,
+      "learning_rate": 1.802401036097167e-05,
+      "loss": 0.0484,
+      "step": 1292
+    },
+    {
+      "epoch": 0.5176,
+      "learning_rate": 1.804064443961135e-05,
+      "loss": 0.0445,
+      "step": 1294
+    },
+    {
+      "epoch": 0.5184,
+      "learning_rate": 1.8057215815690494e-05,
+      "loss": 0.0345,
+      "step": 1296
+    },
+    {
+      "epoch": 0.5192,
+      "learning_rate": 1.8073724359982184e-05,
+      "loss": 0.0209,
+      "step": 1298
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 1.809016994374947e-05,
+      "loss": 0.0064,
+      "step": 1300
+    },
+    {
+      "epoch": 0.5208,
+      "learning_rate": 1.81065524387464e-05,
+      "loss": 0.5277,
+      "step": 1302
+    },
+    {
+      "epoch": 0.5216,
+      "learning_rate": 1.8122871717218968e-05,
+      "loss": 0.0001,
+      "step": 1304
+    },
+    {
+      "epoch": 0.5224,
+      "learning_rate": 1.8139127651906176e-05,
+      "loss": 0.0005,
+      "step": 1306
+    },
+    {
+      "epoch": 0.5232,
+      "learning_rate": 1.8155320116040976e-05,
+      "loss": 0.0008,
+      "step": 1308
+    },
+    {
+      "epoch": 0.524,
+      "learning_rate": 1.817144898335129e-05,
+      "loss": 0.0044,
+      "step": 1310
+    },
+    {
+      "epoch": 0.5248,
+      "learning_rate": 1.818751412806095e-05,
+      "loss": 0.0013,
+      "step": 1312
+    },
+    {
+      "epoch": 0.5256,
+      "learning_rate": 1.8203515424890738e-05,
+      "loss": 0.4533,
+      "step": 1314
+    },
+    {
+      "epoch": 0.5264,
+      "learning_rate": 1.8219452749059322e-05,
+      "loss": 1.586,
+      "step": 1316
+    },
+    {
+      "epoch": 0.5272,
+      "learning_rate": 1.8235325976284276e-05,
+      "loss": 0.2815,
+      "step": 1318
+    },
+    {
+      "epoch": 0.528,
+      "learning_rate": 1.8251134982782952e-05,
+      "loss": 0.0044,
+      "step": 1320
+    },
+    {
+      "epoch": 0.5288,
+      "learning_rate": 1.826687964527355e-05,
+      "loss": 0.0002,
+      "step": 1322
+    },
+    {
+      "epoch": 0.5296,
+      "learning_rate": 1.828255984097604e-05,
+      "loss": 0.0003,
+      "step": 1324
+    },
+    {
+      "epoch": 0.5304,
+      "learning_rate": 1.8298175447613093e-05,
+      "loss": 0.6127,
+      "step": 1326
+    },
+    {
+      "epoch": 0.5312,
+      "learning_rate": 1.8313726343411092e-05,
+      "loss": 0.3639,
+      "step": 1328
+    },
+    {
+      "epoch": 0.532,
+      "learning_rate": 1.8329212407101e-05,
+      "loss": 0.4435,
+      "step": 1330
+    },
+    {
+      "epoch": 0.5328,
+      "learning_rate": 1.8344633517919394e-05,
+      "loss": 0.0088,
+      "step": 1332
+    },
+    {
+      "epoch": 0.5336,
+      "learning_rate": 1.8359989555609344e-05,
+      "loss": 0.0047,
+      "step": 1334
+    },
+    {
+      "epoch": 0.5344,
+      "learning_rate": 1.8375280400421407e-05,
+      "loss": 0.3612,
+      "step": 1336
+    },
+    {
+      "epoch": 0.5352,
+      "learning_rate": 1.8390505933114503e-05,
+      "loss": 0.0592,
+      "step": 1338
+    },
+    {
+      "epoch": 0.536,
+      "learning_rate": 1.8405666034956842e-05,
+      "loss": 0.1241,
+      "step": 1340
+    },
+    {
+      "epoch": 0.5368,
+      "learning_rate": 1.842076058772692e-05,
+      "loss": 0.6893,
+      "step": 1342
+    },
+    {
+      "epoch": 0.5376,
+      "learning_rate": 1.8435789473714384e-05,
+      "loss": 0.1428,
+      "step": 1344
+    },
+    {
+      "epoch": 0.5384,
+      "learning_rate": 1.8450752575720964e-05,
+      "loss": 0.0038,
+      "step": 1346
+    },
+    {
+      "epoch": 0.5392,
+      "learning_rate": 1.8465649777061384e-05,
+      "loss": 0.2682,
+      "step": 1348
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 1.8480480961564266e-05,
+      "loss": 0.0566,
+      "step": 1350
+    },
+    {
+      "epoch": 0.5408,
+      "learning_rate": 1.8495246013573047e-05,
+      "loss": 0.0086,
+      "step": 1352
+    },
+    {
+      "epoch": 0.5416,
+      "learning_rate": 1.850994481794691e-05,
+      "loss": 0.0004,
+      "step": 1354
+    },
+    {
+      "epoch": 0.5424,
+      "learning_rate": 1.8524577260061628e-05,
+      "loss": 0.0132,
+      "step": 1356
+    },
+    {
+      "epoch": 0.5432,
+      "learning_rate": 1.8539143225810453e-05,
+      "loss": 0.5804,
+      "step": 1358
+    },
+    {
+      "epoch": 0.544,
+      "learning_rate": 1.8553642601605066e-05,
+      "loss": 0.0579,
+      "step": 1360
+    },
+    {
+      "epoch": 0.5448,
+      "learning_rate": 1.856807527437643e-05,
+      "loss": 0.4187,
+      "step": 1362
+    },
+    {
+      "epoch": 0.5456,
+      "learning_rate": 1.8582441131575658e-05,
+      "loss": 0.0008,
+      "step": 1364
+    },
+    {
+      "epoch": 0.5464,
+      "learning_rate": 1.859674006117491e-05,
+      "loss": 0.0948,
+      "step": 1366
+    },
+    {
+      "epoch": 0.5472,
+      "learning_rate": 1.8610971951668268e-05,
+      "loss": 0.0008,
+      "step": 1368
+    },
+    {
+      "epoch": 0.548,
+      "learning_rate": 1.862513669207257e-05,
+      "loss": 0.4324,
+      "step": 1370
+    },
+    {
+      "epoch": 0.5488,
+      "learning_rate": 1.8639234171928348e-05,
+      "loss": 0.0195,
+      "step": 1372
+    },
+    {
+      "epoch": 0.5496,
+      "learning_rate": 1.8653264281300612e-05,
+      "loss": 0.0012,
+      "step": 1374
+    },
+    {
+      "epoch": 0.5504,
+      "learning_rate": 1.8667226910779767e-05,
+      "loss": 0.1332,
+      "step": 1376
+    },
+    {
+      "epoch": 0.5512,
+      "learning_rate": 1.8681121951482393e-05,
+      "loss": 0.0032,
+      "step": 1378
+    },
+    {
+      "epoch": 0.552,
+      "learning_rate": 1.869494929505219e-05,
+      "loss": 0.6811,
+      "step": 1380
+    },
+    {
+      "epoch": 0.5528,
+      "learning_rate": 1.870870883366075e-05,
+      "loss": 0.6594,
+      "step": 1382
+    },
+    {
+      "epoch": 0.5536,
+      "learning_rate": 1.8722400460008434e-05,
+      "loss": 0.5689,
+      "step": 1384
+    },
+    {
+      "epoch": 0.5544,
+      "learning_rate": 1.8736024067325195e-05,
+      "loss": 0.0255,
+      "step": 1386
+    },
+    {
+      "epoch": 0.5552,
+      "learning_rate": 1.8749579549371373e-05,
+      "loss": 0.0003,
+      "step": 1388
+    },
+    {
+      "epoch": 0.556,
+      "learning_rate": 1.876306680043863e-05,
+      "loss": 0.1605,
+      "step": 1390
+    },
+    {
+      "epoch": 0.5568,
+      "learning_rate": 1.8776485715350665e-05,
+      "loss": 0.054,
+      "step": 1392
+    },
+    {
+      "epoch": 0.5576,
+      "learning_rate": 1.878983618946409e-05,
+      "loss": 0.009,
+      "step": 1394
+    },
+    {
+      "epoch": 0.5584,
+      "learning_rate": 1.8803118118669203e-05,
+      "loss": 0.3037,
+      "step": 1396
+    },
+    {
+      "epoch": 0.5592,
+      "learning_rate": 1.881633139939087e-05,
+      "loss": 0.0373,
+      "step": 1398
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 1.882947592858927e-05,
+      "loss": 0.1047,
+      "step": 1400
+    },
+    {
+      "epoch": 0.5608,
+      "learning_rate": 1.884255160376072e-05,
+      "loss": 0.0505,
+      "step": 1402
+    },
+    {
+      "epoch": 0.5616,
+      "learning_rate": 1.885555832293849e-05,
+      "loss": 0.517,
+      "step": 1404
+    },
+    {
+      "epoch": 0.5624,
+      "learning_rate": 1.886849598469356e-05,
+      "loss": 0.4008,
+      "step": 1406
+    },
+    {
+      "epoch": 0.5632,
+      "learning_rate": 1.888136448813544e-05,
+      "loss": 0.1585,
+      "step": 1408
+    },
+    {
+      "epoch": 0.564,
+      "learning_rate": 1.8894163732912972e-05,
+      "loss": 0.002,
+      "step": 1410
+    },
+    {
+      "epoch": 0.5648,
+      "learning_rate": 1.890689361921506e-05,
+      "loss": 0.3848,
+      "step": 1412
+    },
+    {
+      "epoch": 0.5656,
+      "learning_rate": 1.891955404777151e-05,
+      "loss": 0.5114,
+      "step": 1414
+    },
+    {
+      "epoch": 0.5664,
+      "learning_rate": 1.893214491985374e-05,
+      "loss": 0.1321,
+      "step": 1416
+    },
+    {
+      "epoch": 0.5672,
+      "learning_rate": 1.89446661372756e-05,
+      "loss": 0.5882,
+      "step": 1418
+    },
+    {
+      "epoch": 0.568,
+      "learning_rate": 1.895711760239413e-05,
+      "loss": 0.0188,
+      "step": 1420
+    },
+    {
+      "epoch": 0.5688,
+      "learning_rate": 1.89694992181103e-05,
+      "loss": 0.0756,
+      "step": 1422
+    },
+    {
+      "epoch": 0.5696,
+      "learning_rate": 1.8981810887869784e-05,
+      "loss": 0.5708,
+      "step": 1424
+    },
+    {
+      "epoch": 0.5704,
+      "learning_rate": 1.8994052515663708e-05,
+      "loss": 0.07,
+      "step": 1426
+    },
+    {
+      "epoch": 0.5712,
+      "learning_rate": 1.90062240060294e-05,
+      "loss": 0.005,
+      "step": 1428
+    },
+    {
+      "epoch": 0.572,
+      "learning_rate": 1.9018325264051136e-05,
+      "loss": 0.3297,
+      "step": 1430
+    },
+    {
+      "epoch": 0.5728,
+      "learning_rate": 1.9030356195360868e-05,
+      "loss": 0.0366,
+      "step": 1432
+    },
+    {
+      "epoch": 0.5736,
+      "learning_rate": 1.904231670613899e-05,
+      "loss": 0.0119,
+      "step": 1434
+    },
+    {
+      "epoch": 0.5744,
+      "learning_rate": 1.905420670311502e-05,
+      "loss": 0.0041,
+      "step": 1436
+    },
+    {
+      "epoch": 0.5752,
+      "learning_rate": 1.906602609356838e-05,
+      "loss": 0.067,
+      "step": 1438
+    },
+    {
+      "epoch": 0.576,
+      "learning_rate": 1.9077774785329078e-05,
+      "loss": 0.2302,
+      "step": 1440
+    },
+    {
+      "epoch": 0.5768,
+      "learning_rate": 1.9089452686778487e-05,
+      "loss": 0.0009,
+      "step": 1442
+    },
+    {
+      "epoch": 0.5776,
+      "learning_rate": 1.9101059706849957e-05,
+      "loss": 0.0141,
+      "step": 1444
+    },
+    {
+      "epoch": 0.5784,
+      "learning_rate": 1.911259575502962e-05,
+      "loss": 0.2043,
+      "step": 1446
+    },
+    {
+      "epoch": 0.5792,
+      "learning_rate": 1.912406074135706e-05,
+      "loss": 0.0001,
+      "step": 1448
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 1.9135454576426006e-05,
+      "loss": 0.059,
+      "step": 1450
+    },
+    {
+      "epoch": 0.5808,
+      "learning_rate": 1.9146777171385053e-05,
+      "loss": 0.2296,
+      "step": 1452
+    },
+    {
+      "epoch": 0.5816,
+      "learning_rate": 1.9158028437938316e-05,
+      "loss": 0.2977,
+      "step": 1454
+    },
+    {
+      "epoch": 0.5824,
+      "learning_rate": 1.9169208288346168e-05,
+      "loss": 0.0002,
+      "step": 1456
+    },
+    {
+      "epoch": 0.5832,
+      "learning_rate": 1.9180316635425876e-05,
+      "loss": 0.133,
+      "step": 1458
+    },
+    {
+      "epoch": 0.584,
+      "learning_rate": 1.9191353392552346e-05,
+      "loss": 0.0047,
+      "step": 1460
+    },
+    {
+      "epoch": 0.5848,
+      "learning_rate": 1.9202318473658703e-05,
+      "loss": 0.0105,
+      "step": 1462
+    },
+    {
+      "epoch": 0.5856,
+      "learning_rate": 1.9213211793237052e-05,
+      "loss": 0.5803,
+      "step": 1464
+    },
+    {
+      "epoch": 0.5864,
+      "learning_rate": 1.92240332663391e-05,
+      "loss": 0.8232,
+      "step": 1466
+    },
+    {
+      "epoch": 0.5872,
+      "learning_rate": 1.923478280857682e-05,
+      "loss": 0.2902,
+      "step": 1468
+    },
+    {
+      "epoch": 0.588,
+      "learning_rate": 1.924546033612313e-05,
+      "loss": 0.0084,
+      "step": 1470
+    },
+    {
+      "epoch": 0.5888,
+      "learning_rate": 1.9256065765712524e-05,
+      "loss": 0.0606,
+      "step": 1472
+    },
+    {
+      "epoch": 0.5896,
+      "learning_rate": 1.9266599014641724e-05,
+      "loss": 0.3786,
+      "step": 1474
+    },
+    {
+      "epoch": 0.5904,
+      "learning_rate": 1.927706000077034e-05,
+      "loss": 0.0146,
+      "step": 1476
+    },
+    {
+      "epoch": 0.5912,
+      "learning_rate": 1.9287448642521507e-05,
+      "loss": 0.0747,
+      "step": 1478
+    },
+    {
+      "epoch": 0.592,
+      "learning_rate": 1.9297764858882516e-05,
+      "loss": 0.0042,
+      "step": 1480
+    },
+    {
+      "epoch": 0.5928,
+      "learning_rate": 1.9308008569405424e-05,
+      "loss": 0.057,
+      "step": 1482
+    },
+    {
+      "epoch": 0.5936,
+      "learning_rate": 1.9318179694207722e-05,
+      "loss": 0.6622,
+      "step": 1484
+    },
+    {
+      "epoch": 0.5944,
+      "learning_rate": 1.9328278153972943e-05,
+      "loss": 0.3217,
+      "step": 1486
+    },
+    {
+      "epoch": 0.5952,
+      "learning_rate": 1.9338303869951266e-05,
+      "loss": 0.1116,
+      "step": 1488
+    },
+    {
+      "epoch": 0.596,
+      "learning_rate": 1.934825676396015e-05,
+      "loss": 0.1194,
+      "step": 1490
+    },
+    {
+      "epoch": 0.5968,
+      "learning_rate": 1.935813675838491e-05,
+      "loss": 0.0122,
+      "step": 1492
+    },
+    {
+      "epoch": 0.5976,
+      "learning_rate": 1.9367943776179375e-05,
+      "loss": 0.0017,
+      "step": 1494
+    },
+    {
+      "epoch": 0.5984,
+      "learning_rate": 1.9377677740866457e-05,
+      "loss": 0.0015,
+      "step": 1496
+    },
+    {
+      "epoch": 0.5992,
+      "learning_rate": 1.9387338576538743e-05,
+      "loss": 0.1951,
+      "step": 1498
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 1.9396926207859085e-05,
+      "loss": 0.0009,
+      "step": 1500
+    },
+    {
+      "epoch": 0.6008,
+      "learning_rate": 1.9406440560061214e-05,
+      "loss": 0.0017,
+      "step": 1502
+    },
+    {
+      "epoch": 0.6016,
+      "learning_rate": 1.9415881558950302e-05,
+      "loss": 0.4027,
+      "step": 1504
+    },
+    {
+      "epoch": 0.6024,
+      "learning_rate": 1.942524913090354e-05,
+      "loss": 0.4808,
+      "step": 1506
+    },
+    {
+      "epoch": 0.6032,
+      "learning_rate": 1.9434543202870723e-05,
+      "loss": 0.0004,
+      "step": 1508
+    },
+    {
+      "epoch": 0.604,
+      "learning_rate": 1.9443763702374815e-05,
+      "loss": 0.0057,
+      "step": 1510
+    },
+    {
+      "epoch": 0.6048,
+      "learning_rate": 1.9452910557512494e-05,
+      "loss": 0.0264,
+      "step": 1512
+    },
+    {
+      "epoch": 0.6056,
+      "learning_rate": 1.9461983696954756e-05,
+      "loss": 0.0214,
+      "step": 1514
+    },
+    {
+      "epoch": 0.6064,
+      "learning_rate": 1.947098304994744e-05,
+      "loss": 0.0286,
+      "step": 1516
+    },
+    {
+      "epoch": 0.6072,
+      "learning_rate": 1.9479908546311787e-05,
+      "loss": 0.0015,
+      "step": 1518
+    },
+    {
+      "epoch": 0.608,
+      "learning_rate": 1.9488760116444966e-05,
+      "loss": 0.5635,
+      "step": 1520
+    },
+    {
+      "epoch": 0.6088,
+      "learning_rate": 1.949753769132067e-05,
+      "loss": 0.0213,
+      "step": 1522
+    },
+    {
+      "epoch": 0.6096,
+      "learning_rate": 1.95062412024896e-05,
+      "loss": 0.1775,
+      "step": 1524
+    },
+    {
+      "epoch": 0.6104,
+      "learning_rate": 1.951487058208003e-05,
+      "loss": 0.0096,
+      "step": 1526
+    },
+    {
+      "epoch": 0.6112,
+      "learning_rate": 1.952342576279833e-05,
+      "loss": 0.0098,
+      "step": 1528
+    },
+    {
+      "epoch": 0.612,
+      "learning_rate": 1.953190667792947e-05,
+      "loss": 0.001,
+      "step": 1530
+    },
+    {
+      "epoch": 0.6128,
+      "learning_rate": 1.9540313261337578e-05,
+      "loss": 0.006,
+      "step": 1532
+    },
+    {
+      "epoch": 0.6136,
+      "learning_rate": 1.954864544746643e-05,
+      "loss": 0.003,
+      "step": 1534
+    },
+    {
+      "epoch": 0.6144,
+      "learning_rate": 1.955690317133996e-05,
+      "loss": 0.0077,
+      "step": 1536
+    },
+    {
+      "epoch": 0.6152,
+      "learning_rate": 1.956508636856278e-05,
+      "loss": 0.0034,
+      "step": 1538
+    },
+    {
+      "epoch": 0.616,
+      "learning_rate": 1.9573194975320672e-05,
+      "loss": 0.125,
+      "step": 1540
+    },
+    {
+      "epoch": 0.6168,
+      "learning_rate": 1.95812289283811e-05,
+      "loss": 0.0174,
+      "step": 1542
+    },
+    {
+      "epoch": 0.6176,
+      "learning_rate": 1.958918816509367e-05,
+      "loss": 0.7194,
+      "step": 1544
+    },
+    {
+      "epoch": 0.6184,
+      "learning_rate": 1.9597072623390668e-05,
+      "loss": 0.2466,
+      "step": 1546
+    },
+    {
+      "epoch": 0.6192,
+      "learning_rate": 1.9604882241787496e-05,
+      "loss": 0.0222,
+      "step": 1548
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 1.9612616959383187e-05,
+      "loss": 0.3027,
+      "step": 1550
+    },
+    {
+      "epoch": 0.6208,
+      "learning_rate": 1.9620276715860856e-05,
+      "loss": 0.0285,
+      "step": 1552
+    },
+    {
+      "epoch": 0.6216,
+      "learning_rate": 1.9627861451488187e-05,
+      "loss": 0.0427,
+      "step": 1554
+    },
+    {
+      "epoch": 0.6224,
+      "learning_rate": 1.963537110711789e-05,
+      "loss": 0.182,
+      "step": 1556
+    },
+    {
+      "epoch": 0.6232,
+      "learning_rate": 1.964280562418815e-05,
+      "loss": 0.0003,
+      "step": 1558
+    },
+    {
+      "epoch": 0.624,
+      "learning_rate": 1.9650164944723116e-05,
+      "loss": 0.0442,
+      "step": 1560
+    },
+    {
+      "epoch": 0.6248,
+      "learning_rate": 1.9657449011333328e-05,
+      "loss": 0.2273,
+      "step": 1562
+    },
+    {
+      "epoch": 0.6256,
+      "learning_rate": 1.9664657767216176e-05,
+      "loss": 0.0003,
+      "step": 1564
+    },
+    {
+      "epoch": 0.6264,
+      "learning_rate": 1.967179115615633e-05,
+      "loss": 0.0438,
+      "step": 1566
+    },
+    {
+      "epoch": 0.6272,
+      "learning_rate": 1.967884912252619e-05,
+      "loss": 0.0003,
+      "step": 1568
+    },
+    {
+      "epoch": 0.628,
+      "learning_rate": 1.9685831611286312e-05,
+      "loss": 0.2332,
+      "step": 1570
+    },
+    {
+      "epoch": 0.6288,
+      "learning_rate": 1.969273856798585e-05,
+      "loss": 0.3103,
+      "step": 1572
+    },
+    {
+      "epoch": 0.6296,
+      "learning_rate": 1.9699569938762972e-05,
+      "loss": 0.2202,
+      "step": 1574
+    },
+    {
+      "epoch": 0.6304,
+      "learning_rate": 1.9706325670345276e-05,
+      "loss": 0.0534,
+      "step": 1576
+    },
+    {
+      "epoch": 0.6312,
+      "learning_rate": 1.9713005710050203e-05,
+      "loss": 0.3785,
+      "step": 1578
+    },
+    {
+      "epoch": 0.632,
+      "learning_rate": 1.9719610005785466e-05,
+      "loss": 0.506,
+      "step": 1580
+    },
+    {
+      "epoch": 0.6328,
+      "learning_rate": 1.9726138506049434e-05,
+      "loss": 0.15,
+      "step": 1582
+    },
+    {
+      "epoch": 0.6336,
+      "learning_rate": 1.9732591159931564e-05,
+      "loss": 0.2677,
+      "step": 1584
+    },
+    {
+      "epoch": 0.6344,
+      "learning_rate": 1.9738967917112752e-05,
+      "loss": 0.2498,
+      "step": 1586
+    },
+    {
+      "epoch": 0.6352,
+      "learning_rate": 1.974526872786577e-05,
+      "loss": 0.0013,
+      "step": 1588
+    },
+    {
+      "epoch": 0.636,
+      "learning_rate": 1.9751493543055634e-05,
+      "loss": 0.0054,
+      "step": 1590
+    },
+    {
+      "epoch": 0.6368,
+      "learning_rate": 1.9757642314139977e-05,
+      "loss": 0.3639,
+      "step": 1592
+    },
+    {
+      "epoch": 0.6376,
+      "learning_rate": 1.976371499316945e-05,
+      "loss": 0.0004,
+      "step": 1594
+    },
+    {
+      "epoch": 0.6384,
+      "learning_rate": 1.9769711532788083e-05,
+      "loss": 0.3692,
+      "step": 1596
+    },
+    {
+      "epoch": 0.6392,
+      "learning_rate": 1.9775631886233655e-05,
+      "loss": 0.191,
+      "step": 1598
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 1.9781476007338054e-05,
+      "loss": 0.0002,
+      "step": 1600
+    },
+    {
+      "epoch": 0.6408,
+      "learning_rate": 1.978724385052766e-05,
+      "loss": 0.0002,
+      "step": 1602
+    },
+    {
+      "epoch": 0.6416,
+      "learning_rate": 1.9792935370823673e-05,
+      "loss": 0.0026,
+      "step": 1604
+    },
+    {
+      "epoch": 0.6424,
+      "learning_rate": 1.979855052384247e-05,
+      "loss": 0.0347,
+      "step": 1606
+    },
+    {
+      "epoch": 0.6432,
+      "learning_rate": 1.9804089265795956e-05,
+      "loss": 0.0011,
+      "step": 1608
+    },
+    {
+      "epoch": 0.644,
+      "learning_rate": 1.9809551553491918e-05,
+      "loss": 0.2278,
+      "step": 1610
+    },
+    {
+      "epoch": 0.6448,
+      "learning_rate": 1.981493734433433e-05,
+      "loss": 0.0023,
+      "step": 1612
+    },
+    {
+      "epoch": 0.6456,
+      "learning_rate": 1.982024659632372e-05,
+      "loss": 0.0313,
+      "step": 1614
+    },
+    {
+      "epoch": 0.6464,
+      "learning_rate": 1.9825479268057472e-05,
+      "loss": 0.0001,
+      "step": 1616
+    },
+    {
+      "epoch": 0.6472,
+      "learning_rate": 1.9830635318730155e-05,
+      "loss": 0.301,
+      "step": 1618
+    },
+    {
+      "epoch": 0.648,
+      "learning_rate": 1.9835714708133858e-05,
+      "loss": 0.0276,
+      "step": 1620
+    },
+    {
+      "epoch": 0.6488,
+      "learning_rate": 1.9840717396658483e-05,
+      "loss": 1.7505,
+      "step": 1622
+    },
+    {
+      "epoch": 0.6496,
+      "learning_rate": 1.9845643345292055e-05,
+      "loss": 0.0024,
+      "step": 1624
+    },
+    {
+      "epoch": 0.6504,
+      "learning_rate": 1.9850492515621038e-05,
+      "loss": 0.0959,
+      "step": 1626
+    },
+    {
+      "epoch": 0.6512,
+      "learning_rate": 1.985526486983063e-05,
+      "loss": 0.0063,
+      "step": 1628
+    },
+    {
+      "epoch": 0.652,
+      "learning_rate": 1.985996037070505e-05,
+      "loss": 0.0406,
+      "step": 1630
+    },
+    {
+      "epoch": 0.6528,
+      "learning_rate": 1.9864578981627844e-05,
+      "loss": 0.1483,
+      "step": 1632
+    },
+    {
+      "epoch": 0.6536,
+      "learning_rate": 1.9869120666582153e-05,
+      "loss": 0.4189,
+      "step": 1634
+    },
+    {
+      "epoch": 0.6544,
+      "learning_rate": 1.9873585390151003e-05,
+      "loss": 0.0329,
+      "step": 1636
+    },
+    {
+      "epoch": 0.6552,
+      "learning_rate": 1.987797311751759e-05,
+      "loss": 1.1702,
+      "step": 1638
+    },
+    {
+      "epoch": 0.656,
+      "learning_rate": 1.9882283814465528e-05,
+      "loss": 0.0006,
+      "step": 1640
+    },
+    {
+      "epoch": 0.6568,
+      "learning_rate": 1.988651744737914e-05,
+      "loss": 0.1924,
+      "step": 1642
+    },
+    {
+      "epoch": 0.6576,
+      "learning_rate": 1.9890673983243704e-05,
+      "loss": 0.2227,
+      "step": 1644
+    },
+    {
+      "epoch": 0.6584,
+      "learning_rate": 1.9894753389645723e-05,
+      "loss": 0.0678,
+      "step": 1646
+    },
+    {
+      "epoch": 0.6592,
+      "learning_rate": 1.9898755634773155e-05,
+      "loss": 0.1466,
+      "step": 1648
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 1.9902680687415704e-05,
+      "loss": 0.4467,
+      "step": 1650
+    },
+    {
+      "epoch": 0.6608,
+      "learning_rate": 1.9906528516965014e-05,
+      "loss": 1.5136,
+      "step": 1652
+    },
+    {
+      "epoch": 0.6616,
+      "learning_rate": 1.9910299093414926e-05,
+      "loss": 0.0969,
+      "step": 1654
+    },
+    {
+      "epoch": 0.6624,
+      "learning_rate": 1.9913992387361744e-05,
+      "loss": 0.16,
+      "step": 1656
+    },
+    {
+      "epoch": 0.6632,
+      "learning_rate": 1.9917608370004414e-05,
+      "loss": 0.0317,
+      "step": 1658
+    },
+    {
+      "epoch": 0.664,
+      "learning_rate": 1.9921147013144782e-05,
+      "loss": 0.0569,
+      "step": 1660
+    },
+    {
+      "epoch": 0.6648,
+      "learning_rate": 1.9924608289187786e-05,
+      "loss": 0.0147,
+      "step": 1662
+    },
+    {
+      "epoch": 0.6656,
+      "learning_rate": 1.9927992171141707e-05,
+      "loss": 0.2525,
+      "step": 1664
+    },
+    {
+      "epoch": 0.6664,
+      "learning_rate": 1.9931298632618355e-05,
+      "loss": 0.0582,
+      "step": 1666
+    },
+    {
+      "epoch": 0.6672,
+      "learning_rate": 1.9934527647833276e-05,
+      "loss": 0.1197,
+      "step": 1668
+    },
+    {
+      "epoch": 0.668,
+      "learning_rate": 1.9937679191605964e-05,
+      "loss": 0.0994,
+      "step": 1670
+    },
+    {
+      "epoch": 0.6688,
+      "learning_rate": 1.9940753239360047e-05,
+      "loss": 0.1577,
+      "step": 1672
+    },
+    {
+      "epoch": 0.6696,
+      "learning_rate": 1.994374976712348e-05,
+      "loss": 1.0883,
+      "step": 1674
+    },
+    {
+      "epoch": 0.6704,
+      "learning_rate": 1.994666875152874e-05,
+      "loss": 0.0405,
+      "step": 1676
+    },
+    {
+      "epoch": 0.6712,
+      "learning_rate": 1.9949510169813003e-05,
+      "loss": 0.0261,
+      "step": 1678
+    },
+    {
+      "epoch": 0.672,
+      "learning_rate": 1.9952273999818312e-05,
+      "loss": 0.012,
+      "step": 1680
+    },
+    {
+      "epoch": 0.6728,
+      "learning_rate": 1.995496021999177e-05,
+      "loss": 0.3207,
+      "step": 1682
+    },
+    {
+      "epoch": 0.6736,
+      "learning_rate": 1.9957568809385693e-05,
+      "loss": 0.049,
+      "step": 1684
+    },
+    {
+      "epoch": 0.6744,
+      "learning_rate": 1.9960099747657774e-05,
+      "loss": 0.0295,
+      "step": 1686
+    },
+    {
+      "epoch": 0.6752,
+      "learning_rate": 1.996255301507125e-05,
+      "loss": 0.1288,
+      "step": 1688
+    },
+    {
+      "epoch": 0.676,
+      "learning_rate": 1.9964928592495046e-05,
+      "loss": 0.0928,
+      "step": 1690
+    },
+    {
+      "epoch": 0.6768,
+      "learning_rate": 1.9967226461403934e-05,
+      "loss": 0.281,
+      "step": 1692
+    },
+    {
+      "epoch": 0.6776,
+      "learning_rate": 1.996944660387867e-05,
+      "loss": 0.0964,
+      "step": 1694
+    },
+    {
+      "epoch": 0.6784,
+      "learning_rate": 1.997158900260614e-05,
+      "loss": 0.735,
+      "step": 1696
+    },
+    {
+      "epoch": 0.6792,
+      "learning_rate": 1.9973653640879486e-05,
+      "loss": 0.1437,
+      "step": 1698
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 1.9975640502598246e-05,
+      "loss": 0.0645,
+      "step": 1700
+    },
+    {
+      "epoch": 0.6808,
+      "learning_rate": 1.997754957226847e-05,
+      "loss": 0.0176,
+      "step": 1702
+    },
+    {
+      "epoch": 0.6816,
+      "learning_rate": 1.9979380835002846e-05,
+      "loss": 0.0703,
+      "step": 1704
+    },
+    {
+      "epoch": 0.6824,
+      "learning_rate": 1.9981134276520828e-05,
+      "loss": 0.0106,
+      "step": 1706
+    },
+    {
+      "epoch": 0.6832,
+      "learning_rate": 1.998280988314872e-05,
+      "loss": 0.7959,
+      "step": 1708
+    },
+    {
+      "epoch": 0.684,
+      "learning_rate": 1.998440764181981e-05,
+      "loss": 0.1008,
+      "step": 1710
+    },
+    {
+      "epoch": 0.6848,
+      "learning_rate": 1.9985927540074453e-05,
+      "loss": 0.3584,
+      "step": 1712
+    },
+    {
+      "epoch": 0.6856,
+      "learning_rate": 1.998736956606018e-05,
+      "loss": 0.0058,
+      "step": 1714
+    },
+    {
+      "epoch": 0.6864,
+      "learning_rate": 1.9988733708531772e-05,
+      "loss": 0.0196,
+      "step": 1716
+    },
+    {
+      "epoch": 0.6872,
+      "learning_rate": 1.9990019956851384e-05,
+      "loss": 0.0115,
+      "step": 1718
+    },
+    {
+      "epoch": 0.688,
+      "learning_rate": 1.9991228300988586e-05,
+      "loss": 0.0118,
+      "step": 1720
+    },
+    {
+      "epoch": 0.6888,
+      "learning_rate": 1.999235873152047e-05,
+      "loss": 0.2053,
+      "step": 1722
+    },
+    {
+      "epoch": 0.6896,
+      "learning_rate": 1.9993411239631713e-05,
+      "loss": 0.0303,
+      "step": 1724
+    },
+    {
+      "epoch": 0.6904,
+      "learning_rate": 1.9994385817114644e-05,
+      "loss": 0.023,
+      "step": 1726
+    },
+    {
+      "epoch": 0.6912,
+      "learning_rate": 1.9995282456369313e-05,
+      "loss": 0.0311,
+      "step": 1728
+    },
+    {
+      "epoch": 0.692,
+      "learning_rate": 1.9996101150403543e-05,
+      "loss": 0.1754,
+      "step": 1730
+    },
+    {
+      "epoch": 0.6928,
+      "learning_rate": 1.9996841892833e-05,
+      "loss": 0.4787,
+      "step": 1732
+    },
+    {
+      "epoch": 0.6936,
+      "learning_rate": 1.9997504677881224e-05,
+      "loss": 0.055,
+      "step": 1734
+    },
+    {
+      "epoch": 0.6944,
+      "learning_rate": 1.999808950037968e-05,
+      "loss": 0.9505,
+      "step": 1736
+    },
+    {
+      "epoch": 0.6952,
+      "learning_rate": 1.9998596355767805e-05,
+      "loss": 0.0058,
+      "step": 1738
+    },
+    {
+      "epoch": 0.696,
+      "learning_rate": 1.9999025240093045e-05,
+      "loss": 0.0023,
+      "step": 1740
+    },
+    {
+      "epoch": 0.6968,
+      "learning_rate": 1.9999376150010868e-05,
+      "loss": 0.0462,
+      "step": 1742
+    },
+    {
+      "epoch": 0.6976,
+      "learning_rate": 1.9999649082784807e-05,
+      "loss": 0.0009,
+      "step": 1744
+    },
+    {
+      "epoch": 0.6984,
+      "learning_rate": 1.9999844036286483e-05,
+      "loss": 0.4746,
+      "step": 1746
+    },
+    {
+      "epoch": 0.6992,
+      "learning_rate": 1.9999961008995607e-05,
+      "loss": 0.314,
+      "step": 1748
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2e-05,
+      "loss": 0.0017,
+      "step": 1750
+    },
+    {
+      "epoch": 0.7008,
+      "learning_rate": 1.9999961008995607e-05,
+      "loss": 2.2469,
+      "step": 1752
+    },
+    {
+      "epoch": 0.7016,
+      "learning_rate": 1.9999844036286483e-05,
+      "loss": 0.0295,
+      "step": 1754
+    },
+    {
+      "epoch": 0.7024,
+      "learning_rate": 1.9999649082784807e-05,
+      "loss": 0.2414,
+      "step": 1756
+    },
+    {
+      "epoch": 0.7032,
+      "learning_rate": 1.9999376150010868e-05,
+      "loss": 0.1735,
+      "step": 1758
+    },
+    {
+      "epoch": 0.704,
+      "learning_rate": 1.9999025240093045e-05,
+      "loss": 0.009,
+      "step": 1760
+    },
+    {
+      "epoch": 0.7048,
+      "learning_rate": 1.9998596355767805e-05,
+      "loss": 0.1329,
+      "step": 1762
+    },
+    {
+      "epoch": 0.7056,
+      "learning_rate": 1.999808950037968e-05,
+      "loss": 0.2628,
+      "step": 1764
+    },
+    {
+      "epoch": 0.7064,
+      "learning_rate": 1.9997504677881224e-05,
+      "loss": 0.3569,
+      "step": 1766
+    },
+    {
+      "epoch": 0.7072,
+      "learning_rate": 1.9996841892833e-05,
+      "loss": 0.0721,
+      "step": 1768
+    },
+    {
+      "epoch": 0.708,
+      "learning_rate": 1.9996101150403547e-05,
+      "loss": 0.0184,
+      "step": 1770
+    },
+    {
+      "epoch": 0.7088,
+      "learning_rate": 1.9995282456369313e-05,
+      "loss": 0.6307,
+      "step": 1772
+    },
+    {
+      "epoch": 0.7096,
+      "learning_rate": 1.9994385817114644e-05,
+      "loss": 0.0453,
+      "step": 1774
+    },
+    {
+      "epoch": 0.7104,
+      "learning_rate": 1.9993411239631713e-05,
+      "loss": 0.0117,
+      "step": 1776
+    },
+    {
+      "epoch": 0.7112,
+      "learning_rate": 1.999235873152047e-05,
+      "loss": 0.0416,
+      "step": 1778
+    },
+    {
+      "epoch": 0.712,
+      "learning_rate": 1.9991228300988586e-05,
+      "loss": 0.2019,
+      "step": 1780
+    },
+    {
+      "epoch": 0.7128,
+      "learning_rate": 1.9990019956851384e-05,
+      "loss": 0.0016,
+      "step": 1782
+    },
+    {
+      "epoch": 0.7136,
+      "learning_rate": 1.9988733708531772e-05,
+      "loss": 0.0211,
+      "step": 1784
+    },
+    {
+      "epoch": 0.7144,
+      "learning_rate": 1.998736956606018e-05,
+      "loss": 0.0076,
+      "step": 1786
+    },
+    {
+      "epoch": 0.7152,
+      "learning_rate": 1.9985927540074453e-05,
+      "loss": 0.2906,
+      "step": 1788
+    },
+    {
+      "epoch": 0.716,
+      "learning_rate": 1.9984407641819812e-05,
+      "loss": 0.5363,
+      "step": 1790
+    },
+    {
+      "epoch": 0.7168,
+      "learning_rate": 1.998280988314872e-05,
+      "loss": 0.0181,
+      "step": 1792
+    },
+    {
+      "epoch": 0.7176,
+      "learning_rate": 1.9981134276520828e-05,
+      "loss": 1.7796,
+      "step": 1794
+    },
+    {
+      "epoch": 0.7184,
+      "learning_rate": 1.9979380835002846e-05,
+      "loss": 0.2532,
+      "step": 1796
+    },
+    {
+      "epoch": 0.7192,
+      "learning_rate": 1.9977549572268467e-05,
+      "loss": 0.0862,
+      "step": 1798
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 1.9975640502598246e-05,
+      "loss": 0.216,
+      "step": 1800
+    },
+    {
+      "epoch": 0.7208,
+      "learning_rate": 1.9973653640879486e-05,
+      "loss": 0.013,
+      "step": 1802
+    },
+    {
+      "epoch": 0.7216,
+      "learning_rate": 1.997158900260614e-05,
+      "loss": 0.0234,
+      "step": 1804
+    },
+    {
+      "epoch": 0.7224,
+      "learning_rate": 1.9969446603878673e-05,
+      "loss": 0.026,
+      "step": 1806
+    },
+    {
+      "epoch": 0.7232,
+      "learning_rate": 1.9967226461403934e-05,
+      "loss": 0.0468,
+      "step": 1808
+    },
+    {
+      "epoch": 0.724,
+      "learning_rate": 1.9964928592495046e-05,
+      "loss": 0.0468,
+      "step": 1810
+    },
+    {
+      "epoch": 0.7248,
+      "learning_rate": 1.996255301507125e-05,
+      "loss": 0.0512,
+      "step": 1812
+    },
+    {
+      "epoch": 0.7256,
+      "learning_rate": 1.9960099747657774e-05,
+      "loss": 0.0127,
+      "step": 1814
+    },
+    {
+      "epoch": 0.7264,
+      "learning_rate": 1.9957568809385693e-05,
+      "loss": 0.2308,
+      "step": 1816
+    },
+    {
+      "epoch": 0.7272,
+      "learning_rate": 1.995496021999177e-05,
+      "loss": 0.1829,
+      "step": 1818
+    },
+    {
+      "epoch": 0.728,
+      "learning_rate": 1.9952273999818312e-05,
+      "loss": 0.9208,
+      "step": 1820
+    },
+    {
+      "epoch": 0.7288,
+      "learning_rate": 1.9949510169813006e-05,
+      "loss": 0.0892,
+      "step": 1822
+    },
+    {
+      "epoch": 0.7296,
+      "learning_rate": 1.9946668751528745e-05,
+      "loss": 0.1839,
+      "step": 1824
+    },
+    {
+      "epoch": 0.7304,
+      "learning_rate": 1.994374976712348e-05,
+      "loss": 0.7965,
+      "step": 1826
+    },
+    {
+      "epoch": 0.7312,
+      "learning_rate": 1.9940753239360047e-05,
+      "loss": 0.0167,
+      "step": 1828
+    },
+    {
+      "epoch": 0.732,
+      "learning_rate": 1.9937679191605964e-05,
+      "loss": 0.0144,
+      "step": 1830
+    },
+    {
+      "epoch": 0.7328,
+      "learning_rate": 1.993452764783328e-05,
+      "loss": 0.1671,
+      "step": 1832
+    },
+    {
+      "epoch": 0.7336,
+      "learning_rate": 1.9931298632618352e-05,
+      "loss": 0.1504,
+      "step": 1834
+    },
+    {
+      "epoch": 0.7344,
+      "learning_rate": 1.9927992171141707e-05,
+      "loss": 0.8673,
+      "step": 1836
+    },
+    {
+      "epoch": 0.7352,
+      "learning_rate": 1.9924608289187786e-05,
+      "loss": 0.0088,
+      "step": 1838
+    },
+    {
+      "epoch": 0.736,
+      "learning_rate": 1.9921147013144782e-05,
+      "loss": 0.2308,
+      "step": 1840
+    },
+    {
+      "epoch": 0.7368,
+      "learning_rate": 1.9917608370004417e-05,
+      "loss": 0.1666,
+      "step": 1842
+    },
+    {
+      "epoch": 0.7376,
+      "learning_rate": 1.9913992387361747e-05,
+      "loss": 0.3304,
+      "step": 1844
+    },
+    {
+      "epoch": 0.7384,
+      "learning_rate": 1.9910299093414932e-05,
+      "loss": 0.1237,
+      "step": 1846
+    },
+    {
+      "epoch": 0.7392,
+      "learning_rate": 1.990652851696501e-05,
+      "loss": 0.2194,
+      "step": 1848
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 1.9902680687415704e-05,
+      "loss": 0.0359,
+      "step": 1850
+    },
+    {
+      "epoch": 0.7408,
+      "learning_rate": 1.9898755634773155e-05,
+      "loss": 0.1059,
+      "step": 1852
+    },
+    {
+      "epoch": 0.7416,
+      "learning_rate": 1.9894753389645723e-05,
+      "loss": 0.0054,
+      "step": 1854
+    },
+    {
+      "epoch": 0.7424,
+      "learning_rate": 1.9890673983243704e-05,
+      "loss": 0.0806,
+      "step": 1856
+    },
+    {
+      "epoch": 0.7432,
+      "learning_rate": 1.9886517447379143e-05,
+      "loss": 0.3198,
+      "step": 1858
+    },
+    {
+      "epoch": 0.744,
+      "learning_rate": 1.988228381446553e-05,
+      "loss": 0.5058,
+      "step": 1860
+    },
+    {
+      "epoch": 0.7448,
+      "learning_rate": 1.987797311751759e-05,
+      "loss": 0.3095,
+      "step": 1862
+    },
+    {
+      "epoch": 0.7456,
+      "learning_rate": 1.9873585390151007e-05,
+      "loss": 0.2754,
+      "step": 1864
+    },
+    {
+      "epoch": 0.7464,
+      "learning_rate": 1.9869120666582153e-05,
+      "loss": 0.1577,
+      "step": 1866
+    },
+    {
+      "epoch": 0.7472,
+      "learning_rate": 1.9864578981627844e-05,
+      "loss": 0.026,
+      "step": 1868
+    },
+    {
+      "epoch": 0.748,
+      "learning_rate": 1.985996037070505e-05,
+      "loss": 0.1339,
+      "step": 1870
+    },
+    {
+      "epoch": 0.7488,
+      "learning_rate": 1.985526486983063e-05,
+      "loss": 0.0163,
+      "step": 1872
+    },
+    {
+      "epoch": 0.7496,
+      "learning_rate": 1.9850492515621038e-05,
+      "loss": 0.019,
+      "step": 1874
+    },
+    {
+      "epoch": 0.7504,
+      "learning_rate": 1.9845643345292055e-05,
+      "loss": 0.3126,
+      "step": 1876
+    },
+    {
+      "epoch": 0.7512,
+      "learning_rate": 1.9840717396658483e-05,
+      "loss": 0.2677,
+      "step": 1878
+    },
+    {
+      "epoch": 0.752,
+      "learning_rate": 1.983571470813386e-05,
+      "loss": 0.3568,
+      "step": 1880
+    },
+    {
+      "epoch": 0.7528,
+      "learning_rate": 1.983063531873016e-05,
+      "loss": 0.2941,
+      "step": 1882
+    },
+    {
+      "epoch": 0.7536,
+      "learning_rate": 1.982547926805747e-05,
+      "loss": 0.0069,
+      "step": 1884
+    },
+    {
+      "epoch": 0.7544,
+      "learning_rate": 1.9820246596323724e-05,
+      "loss": 0.0106,
+      "step": 1886
+    },
+    {
+      "epoch": 0.7552,
+      "learning_rate": 1.981493734433433e-05,
+      "loss": 0.0587,
+      "step": 1888
+    },
+    {
+      "epoch": 0.756,
+      "learning_rate": 1.9809551553491918e-05,
+      "loss": 0.0023,
+      "step": 1890
+    },
+    {
+      "epoch": 0.7568,
+      "learning_rate": 1.9804089265795963e-05,
+      "loss": 0.1224,
+      "step": 1892
+    },
+    {
+      "epoch": 0.7576,
+      "learning_rate": 1.979855052384247e-05,
+      "loss": 0.003,
+      "step": 1894
+    },
+    {
+      "epoch": 0.7584,
+      "learning_rate": 1.979293537082368e-05,
+      "loss": 0.3377,
+      "step": 1896
+    },
+    {
+      "epoch": 0.7592,
+      "learning_rate": 1.9787243850527663e-05,
+      "loss": 0.0331,
+      "step": 1898
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 1.978147600733806e-05,
+      "loss": 0.0491,
+      "step": 1900
+    },
+    {
+      "epoch": 0.7608,
+      "learning_rate": 1.977563188623365e-05,
+      "loss": 0.0005,
+      "step": 1902
+    },
+    {
+      "epoch": 0.7616,
+      "learning_rate": 1.9769711532788086e-05,
+      "loss": 0.0682,
+      "step": 1904
+    },
+    {
+      "epoch": 0.7624,
+      "learning_rate": 1.9763714993169448e-05,
+      "loss": 0.0827,
+      "step": 1906
+    },
+    {
+      "epoch": 0.7632,
+      "learning_rate": 1.9757642314139977e-05,
+      "loss": 0.9237,
+      "step": 1908
+    },
+    {
+      "epoch": 0.764,
+      "learning_rate": 1.9751493543055638e-05,
+      "loss": 0.0071,
+      "step": 1910
+    },
+    {
+      "epoch": 0.7648,
+      "learning_rate": 1.9745268727865774e-05,
+      "loss": 0.0193,
+      "step": 1912
+    },
+    {
+      "epoch": 0.7656,
+      "learning_rate": 1.973896791711276e-05,
+      "loss": 0.0005,
+      "step": 1914
+    },
+    {
+      "epoch": 0.7664,
+      "learning_rate": 1.9732591159931567e-05,
+      "loss": 0.1188,
+      "step": 1916
+    },
+    {
+      "epoch": 0.7672,
+      "learning_rate": 1.972613850604944e-05,
+      "loss": 0.0178,
+      "step": 1918
+    },
+    {
+      "epoch": 0.768,
+      "learning_rate": 1.9719610005785463e-05,
+      "loss": 0.3305,
+      "step": 1920
+    },
+    {
+      "epoch": 0.7688,
+      "learning_rate": 1.9713005710050206e-05,
+      "loss": 0.0029,
+      "step": 1922
+    },
+    {
+      "epoch": 0.7696,
+      "learning_rate": 1.9706325670345276e-05,
+      "loss": 0.0447,
+      "step": 1924
+    },
+    {
+      "epoch": 0.7704,
+      "learning_rate": 1.9699569938762975e-05,
+      "loss": 0.0011,
+      "step": 1926
+    },
+    {
+      "epoch": 0.7712,
+      "learning_rate": 1.969273856798586e-05,
+      "loss": 0.0123,
+      "step": 1928
+    },
+    {
+      "epoch": 0.772,
+      "learning_rate": 1.9685831611286312e-05,
+      "loss": 0.0023,
+      "step": 1930
+    },
+    {
+      "epoch": 0.7728,
+      "learning_rate": 1.9678849122526195e-05,
+      "loss": 0.0008,
+      "step": 1932
+    },
+    {
+      "epoch": 0.7736,
+      "learning_rate": 1.967179115615633e-05,
+      "loss": 0.579,
+      "step": 1934
+    },
+    {
+      "epoch": 0.7744,
+      "learning_rate": 1.966465776721618e-05,
+      "loss": 0.0196,
+      "step": 1936
+    },
+    {
+      "epoch": 0.7752,
+      "learning_rate": 1.9657449011333328e-05,
+      "loss": 0.0382,
+      "step": 1938
+    },
+    {
+      "epoch": 0.776,
+      "learning_rate": 1.965016494472312e-05,
+      "loss": 0.0026,
+      "step": 1940
+    },
+    {
+      "epoch": 0.7768,
+      "learning_rate": 1.964280562418815e-05,
+      "loss": 0.0063,
+      "step": 1942
+    },
+    {
+      "epoch": 0.7776,
+      "learning_rate": 1.963537110711789e-05,
+      "loss": 0.0025,
+      "step": 1944
+    },
+    {
+      "epoch": 0.7784,
+      "learning_rate": 1.9627861451488194e-05,
+      "loss": 0.0163,
+      "step": 1946
+    },
+    {
+      "epoch": 0.7792,
+      "learning_rate": 1.962027671586086e-05,
+      "loss": 0.0006,
+      "step": 1948
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 1.9612616959383194e-05,
+      "loss": 0.0035,
+      "step": 1950
+    },
+    {
+      "epoch": 0.7808,
+      "learning_rate": 1.96048822417875e-05,
+      "loss": 0.0256,
+      "step": 1952
+    },
+    {
+      "epoch": 0.7816,
+      "learning_rate": 1.9597072623390668e-05,
+      "loss": 0.4352,
+      "step": 1954
+    },
+    {
+      "epoch": 0.7824,
+      "learning_rate": 1.9589188165093666e-05,
+      "loss": 0.0044,
+      "step": 1956
+    },
+    {
+      "epoch": 0.7832,
+      "learning_rate": 1.95812289283811e-05,
+      "loss": 0.0061,
+      "step": 1958
+    },
+    {
+      "epoch": 0.784,
+      "learning_rate": 1.9573194975320672e-05,
+      "loss": 0.0048,
+      "step": 1960
+    },
+    {
+      "epoch": 0.7848,
+      "learning_rate": 1.9565086368562784e-05,
+      "loss": 0.1914,
+      "step": 1962
+    },
+    {
+      "epoch": 0.7856,
+      "learning_rate": 1.9556903171339966e-05,
+      "loss": 0.0944,
+      "step": 1964
+    },
+    {
+      "epoch": 0.7864,
+      "learning_rate": 1.954864544746643e-05,
+      "loss": 0.0012,
+      "step": 1966
+    },
+    {
+      "epoch": 0.7872,
+      "learning_rate": 1.9540313261337585e-05,
+      "loss": 0.0027,
+      "step": 1968
+    },
+    {
+      "epoch": 0.788,
+      "learning_rate": 1.9531906677929472e-05,
+      "loss": 0.0148,
+      "step": 1970
+    },
+    {
+      "epoch": 0.7888,
+      "learning_rate": 1.9523425762798335e-05,
+      "loss": 0.0012,
+      "step": 1972
+    },
+    {
+      "epoch": 0.7896,
+      "learning_rate": 1.9514870582080035e-05,
+      "loss": 0.0018,
+      "step": 1974
+    },
+    {
+      "epoch": 0.7904,
+      "learning_rate": 1.95062412024896e-05,
+      "loss": 0.0305,
+      "step": 1976
+    },
+    {
+      "epoch": 0.7912,
+      "learning_rate": 1.9497537691320667e-05,
+      "loss": 0.0019,
+      "step": 1978
+    },
+    {
+      "epoch": 0.792,
+      "learning_rate": 1.948876011644497e-05,
+      "loss": 0.1104,
+      "step": 1980
+    },
+    {
+      "epoch": 0.7928,
+      "learning_rate": 1.9479908546311787e-05,
+      "loss": 0.0005,
+      "step": 1982
+    },
+    {
+      "epoch": 0.7936,
+      "learning_rate": 1.9470983049947443e-05,
+      "loss": 0.0132,
+      "step": 1984
+    },
+    {
+      "epoch": 0.7944,
+      "learning_rate": 1.9461983696954767e-05,
+      "loss": 0.0727,
+      "step": 1986
+    },
+    {
+      "epoch": 0.7952,
+      "learning_rate": 1.9452910557512494e-05,
+      "loss": 0.3567,
+      "step": 1988
+    },
+    {
+      "epoch": 0.796,
+      "learning_rate": 1.9443763702374818e-05,
+      "loss": 0.0384,
+      "step": 1990
+    },
+    {
+      "epoch": 0.7968,
+      "learning_rate": 1.9434543202870726e-05,
+      "loss": 0.2872,
+      "step": 1992
+    },
+    {
+      "epoch": 0.7976,
+      "learning_rate": 1.9425249130903544e-05,
+      "loss": 0.0145,
+      "step": 1994
+    },
+    {
+      "epoch": 0.7984,
+      "learning_rate": 1.94158815589503e-05,
+      "loss": 1.0319,
+      "step": 1996
+    },
+    {
+      "epoch": 0.7992,
+      "learning_rate": 1.940644056006122e-05,
+      "loss": 0.0323,
+      "step": 1998
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 1.939692620785909e-05,
+      "loss": 0.9458,
+      "step": 2000
+    },
+    {
+      "epoch": 0.8008,
+      "learning_rate": 1.9387338576538746e-05,
+      "loss": 0.0499,
+      "step": 2002
+    },
+    {
+      "epoch": 0.8016,
+      "learning_rate": 1.9377677740866464e-05,
+      "loss": 0.0003,
+      "step": 2004
+    },
+    {
+      "epoch": 0.8024,
+      "learning_rate": 1.936794377617938e-05,
+      "loss": 0.0219,
+      "step": 2006
+    },
+    {
+      "epoch": 0.8032,
+      "learning_rate": 1.9358136758384917e-05,
+      "loss": 0.001,
+      "step": 2008
+    },
+    {
+      "epoch": 0.804,
+      "learning_rate": 1.9348256763960146e-05,
+      "loss": 0.0657,
+      "step": 2010
+    },
+    {
+      "epoch": 0.8048,
+      "learning_rate": 1.9338303869951273e-05,
+      "loss": 0.0478,
+      "step": 2012
+    },
+    {
+      "epoch": 0.8056,
+      "learning_rate": 1.9328278153972943e-05,
+      "loss": 0.1628,
+      "step": 2014
+    },
+    {
+      "epoch": 0.8064,
+      "learning_rate": 1.931817969420773e-05,
+      "loss": 0.0626,
+      "step": 2016
+    },
+    {
+      "epoch": 0.8072,
+      "learning_rate": 1.930800856940543e-05,
+      "loss": 0.001,
+      "step": 2018
+    },
+    {
+      "epoch": 0.808,
+      "learning_rate": 1.929776485888252e-05,
+      "loss": 0.0201,
+      "step": 2020
+    },
+    {
+      "epoch": 0.8088,
+      "learning_rate": 1.9287448642521517e-05,
+      "loss": 0.3468,
+      "step": 2022
+    },
+    {
+      "epoch": 0.8096,
+      "learning_rate": 1.9277060000770342e-05,
+      "loss": 0.5697,
+      "step": 2024
+    },
+    {
+      "epoch": 0.8104,
+      "learning_rate": 1.9266599014641727e-05,
+      "loss": 0.5123,
+      "step": 2026
+    },
+    {
+      "epoch": 0.8112,
+      "learning_rate": 1.925606576571252e-05,
+      "loss": 0.2776,
+      "step": 2028
+    },
+    {
+      "epoch": 0.812,
+      "learning_rate": 1.9245460336123136e-05,
+      "loss": 0.218,
+      "step": 2030
+    },
+    {
+      "epoch": 0.8128,
+      "learning_rate": 1.923478280857682e-05,
+      "loss": 0.5923,
+      "step": 2032
+    },
+    {
+      "epoch": 0.8136,
+      "learning_rate": 1.9224033266339103e-05,
+      "loss": 0.2937,
+      "step": 2034
+    },
+    {
+      "epoch": 0.8144,
+      "learning_rate": 1.9213211793237066e-05,
+      "loss": 0.1209,
+      "step": 2036
+    },
+    {
+      "epoch": 0.8152,
+      "learning_rate": 1.9202318473658707e-05,
+      "loss": 0.0074,
+      "step": 2038
+    },
+    {
+      "epoch": 0.816,
+      "learning_rate": 1.919135339255235e-05,
+      "loss": 0.3764,
+      "step": 2040
+    },
+    {
+      "epoch": 0.8168,
+      "learning_rate": 1.918031663542588e-05,
+      "loss": 0.4123,
+      "step": 2042
+    },
+    {
+      "epoch": 0.8176,
+      "learning_rate": 1.916920828834617e-05,
+      "loss": 0.5164,
+      "step": 2044
+    },
+    {
+      "epoch": 0.8184,
+      "learning_rate": 1.9158028437938313e-05,
+      "loss": 1.1912,
+      "step": 2046
+    },
+    {
+      "epoch": 0.8192,
+      "learning_rate": 1.9146777171385057e-05,
+      "loss": 1.1282,
+      "step": 2048
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 1.913545457642601e-05,
+      "loss": 0.2692,
+      "step": 2050
+    },
+    {
+      "epoch": 0.8208,
+      "learning_rate": 1.9124060741357065e-05,
+      "loss": 0.1678,
+      "step": 2052
+    },
+    {
+      "epoch": 0.8216,
+      "learning_rate": 1.911259575502963e-05,
+      "loss": 0.2135,
+      "step": 2054
+    },
+    {
+      "epoch": 0.8224,
+      "learning_rate": 1.910105970684996e-05,
+      "loss": 0.0102,
+      "step": 2056
+    },
+    {
+      "epoch": 0.8232,
+      "learning_rate": 1.908945268677849e-05,
+      "loss": 0.2763,
+      "step": 2058
+    },
+    {
+      "epoch": 0.824,
+      "learning_rate": 1.9077774785329085e-05,
+      "loss": 0.1061,
+      "step": 2060
+    },
+    {
+      "epoch": 0.8248,
+      "learning_rate": 1.9066026093568383e-05,
+      "loss": 0.1281,
+      "step": 2062
+    },
+    {
+      "epoch": 0.8256,
+      "learning_rate": 1.9054206703115013e-05,
+      "loss": 0.0895,
+      "step": 2064
+    },
+    {
+      "epoch": 0.8264,
+      "learning_rate": 1.9042316706138994e-05,
+      "loss": 0.2634,
+      "step": 2066
+    },
+    {
+      "epoch": 0.8272,
+      "learning_rate": 1.903035619536087e-05,
+      "loss": 0.0122,
+      "step": 2068
+    },
+    {
+      "epoch": 0.828,
+      "learning_rate": 1.901832526405114e-05,
+      "loss": 0.0331,
+      "step": 2070
+    },
+    {
+      "epoch": 0.8288,
+      "learning_rate": 1.9006224006029414e-05,
+      "loss": 0.0773,
+      "step": 2072
+    },
+    {
+      "epoch": 0.8296,
+      "learning_rate": 1.899405251566371e-05,
+      "loss": 0.023,
+      "step": 2074
+    },
+    {
+      "epoch": 0.8304,
+      "learning_rate": 1.8981810887869797e-05,
+      "loss": 0.245,
+      "step": 2076
+    },
+    {
+      "epoch": 0.8312,
+      "learning_rate": 1.8969499218110302e-05,
+      "loss": 0.0226,
+      "step": 2078
+    },
+    {
+      "epoch": 0.832,
+      "learning_rate": 1.8957117602394133e-05,
+      "loss": 0.0033,
+      "step": 2080
+    },
+    {
+      "epoch": 0.8328,
+      "learning_rate": 1.8944666137275596e-05,
+      "loss": 0.0674,
+      "step": 2082
+    },
+    {
+      "epoch": 0.8336,
+      "learning_rate": 1.8932144919853744e-05,
+      "loss": 0.008,
+      "step": 2084
+    },
+    {
+      "epoch": 0.8344,
+      "learning_rate": 1.8919554047771508e-05,
+      "loss": 0.0182,
+      "step": 2086
+    },
+    {
+      "epoch": 0.8352,
+      "learning_rate": 1.890689361921507e-05,
+      "loss": 0.0252,
+      "step": 2088
+    },
+    {
+      "epoch": 0.836,
+      "learning_rate": 1.8894163732912986e-05,
+      "loss": 0.0042,
+      "step": 2090
+    },
+    {
+      "epoch": 0.8368,
+      "learning_rate": 1.8881364488135445e-05,
+      "loss": 0.0898,
+      "step": 2092
+    },
+    {
+      "epoch": 0.8376,
+      "learning_rate": 1.886849598469357e-05,
+      "loss": 0.0248,
+      "step": 2094
+    },
+    {
+      "epoch": 0.8384,
+      "learning_rate": 1.8855558322938492e-05,
+      "loss": 0.0324,
+      "step": 2096
+    },
+    {
+      "epoch": 0.8392,
+      "learning_rate": 1.8842551603760725e-05,
+      "loss": 0.0027,
+      "step": 2098
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 1.8829475928589265e-05,
+      "loss": 0.0028,
+      "step": 2100
+    },
+    {
+      "epoch": 0.8408,
+      "learning_rate": 1.8816331399390874e-05,
+      "loss": 0.0142,
+      "step": 2102
+    },
+    {
+      "epoch": 0.8416,
+      "learning_rate": 1.88031181186692e-05,
+      "loss": 0.8458,
+      "step": 2104
+    },
+    {
+      "epoch": 0.8424,
+      "learning_rate": 1.8789836189464092e-05,
+      "loss": 0.0154,
+      "step": 2106
+    },
+    {
+      "epoch": 0.8432,
+      "learning_rate": 1.877648571535068e-05,
+      "loss": 0.0126,
+      "step": 2108
+    },
+    {
+      "epoch": 0.844,
+      "learning_rate": 1.8763066800438638e-05,
+      "loss": 0.003,
+      "step": 2110
+    },
+    {
+      "epoch": 0.8448,
+      "learning_rate": 1.8749579549371387e-05,
+      "loss": 0.0466,
+      "step": 2112
+    },
+    {
+      "epoch": 0.8456,
+      "learning_rate": 1.8736024067325188e-05,
+      "loss": 0.0049,
+      "step": 2114
+    },
+    {
+      "epoch": 0.8464,
+      "learning_rate": 1.8722400460008437e-05,
+      "loss": 0.005,
+      "step": 2116
+    },
+    {
+      "epoch": 0.8472,
+      "learning_rate": 1.8708708833660748e-05,
+      "loss": 0.0026,
+      "step": 2118
+    },
+    {
+      "epoch": 0.848,
+      "learning_rate": 1.8694949295052198e-05,
+      "loss": 0.7495,
+      "step": 2120
+    },
+    {
+      "epoch": 0.8488,
+      "learning_rate": 1.868112195148239e-05,
+      "loss": 0.0057,
+      "step": 2122
+    },
+    {
+      "epoch": 0.8496,
+      "learning_rate": 1.866722691077977e-05,
+      "loss": 0.6767,
+      "step": 2124
+    },
+    {
+      "epoch": 0.8504,
+      "learning_rate": 1.8653264281300626e-05,
+      "loss": 0.6432,
+      "step": 2126
+    },
+    {
+      "epoch": 0.8512,
+      "learning_rate": 1.8639234171928355e-05,
+      "loss": 0.4428,
+      "step": 2128
+    },
+    {
+      "epoch": 0.852,
+      "learning_rate": 1.8625136692072587e-05,
+      "loss": 0.0113,
+      "step": 2130
+    },
+    {
+      "epoch": 0.8528,
+      "learning_rate": 1.8610971951668265e-05,
+      "loss": 0.0022,
+      "step": 2132
+    },
+    {
+      "epoch": 0.8536,
+      "learning_rate": 1.8596740061174912e-05,
+      "loss": 0.1897,
+      "step": 2134
+    },
+    {
+      "epoch": 0.8544,
+      "learning_rate": 1.858244113157566e-05,
+      "loss": 0.4108,
+      "step": 2136
+    },
+    {
+      "epoch": 0.8552,
+      "learning_rate": 1.8568075274376432e-05,
+      "loss": 0.0664,
+      "step": 2138
+    },
+    {
+      "epoch": 0.856,
+      "learning_rate": 1.8553642601605083e-05,
+      "loss": 0.3855,
+      "step": 2140
+    },
+    {
+      "epoch": 0.8568,
+      "learning_rate": 1.8539143225810457e-05,
+      "loss": 0.1407,
+      "step": 2142
+    },
+    {
+      "epoch": 0.8576,
+      "learning_rate": 1.852457726006163e-05,
+      "loss": 0.0698,
+      "step": 2144
+    },
+    {
+      "epoch": 0.8584,
+      "learning_rate": 1.8509944817946917e-05,
+      "loss": 1.2363,
+      "step": 2146
+    },
+    {
+      "epoch": 0.8592,
+      "learning_rate": 1.8495246013573064e-05,
+      "loss": 0.1597,
+      "step": 2148
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 1.848048096156426e-05,
+      "loss": 0.4943,
+      "step": 2150
+    },
+    {
+      "epoch": 0.8608,
+      "learning_rate": 1.8465649777061387e-05,
+      "loss": 0.0143,
+      "step": 2152
+    },
+    {
+      "epoch": 0.8616,
+      "learning_rate": 1.8450752575720967e-05,
+      "loss": 0.047,
+      "step": 2154
+    },
+    {
+      "epoch": 0.8624,
+      "learning_rate": 1.843578947371439e-05,
+      "loss": 0.5615,
+      "step": 2156
+    },
+    {
+      "epoch": 0.8632,
+      "learning_rate": 1.8420760587726935e-05,
+      "loss": 0.5071,
+      "step": 2158
+    },
+    {
+      "epoch": 0.864,
+      "learning_rate": 1.8405666034956846e-05,
+      "loss": 0.0564,
+      "step": 2160
+    },
+    {
+      "epoch": 0.8648,
+      "learning_rate": 1.8390505933114507e-05,
+      "loss": 0.0319,
+      "step": 2162
+    },
+    {
+      "epoch": 0.8656,
+      "learning_rate": 1.8375280400421414e-05,
+      "loss": 0.0658,
+      "step": 2164
+    },
+    {
+      "epoch": 0.8664,
+      "learning_rate": 1.8359989555609365e-05,
+      "loss": 0.0327,
+      "step": 2166
+    },
+    {
+      "epoch": 0.8672,
+      "learning_rate": 1.834463351791939e-05,
+      "loss": 0.1069,
+      "step": 2168
+    },
+    {
+      "epoch": 0.868,
+      "learning_rate": 1.8329212407101006e-05,
+      "loss": 0.0228,
+      "step": 2170
+    },
+    {
+      "epoch": 0.8688,
+      "learning_rate": 1.8313726343411085e-05,
+      "loss": 0.6702,
+      "step": 2172
+    },
+    {
+      "epoch": 0.8696,
+      "learning_rate": 1.82981754476131e-05,
+      "loss": 0.0255,
+      "step": 2174
+    },
+    {
+      "epoch": 0.8704,
+      "learning_rate": 1.8282559840976053e-05,
+      "loss": 0.7038,
+      "step": 2176
+    },
+    {
+      "epoch": 0.8712,
+      "learning_rate": 1.8266879645273557e-05,
+      "loss": 0.2695,
+      "step": 2178
+    },
+    {
+      "epoch": 0.872,
+      "learning_rate": 1.8251134982782966e-05,
+      "loss": 0.0189,
+      "step": 2180
+    },
+    {
+      "epoch": 0.8728,
+      "learning_rate": 1.823532597628428e-05,
+      "loss": 0.0351,
+      "step": 2182
+    },
+    {
+      "epoch": 0.8736,
+      "learning_rate": 1.8219452749059336e-05,
+      "loss": 0.3323,
+      "step": 2184
+    },
+    {
+      "epoch": 0.8744,
+      "learning_rate": 1.8203515424890734e-05,
+      "loss": 0.2399,
+      "step": 2186
+    },
+    {
+      "epoch": 0.8752,
+      "learning_rate": 1.8187514128060956e-05,
+      "loss": 0.0123,
+      "step": 2188
+    },
+    {
+      "epoch": 0.876,
+      "learning_rate": 1.8171448983351284e-05,
+      "loss": 0.3425,
+      "step": 2190
+    },
+    {
+      "epoch": 0.8768,
+      "learning_rate": 1.8155320116040983e-05,
+      "loss": 0.0112,
+      "step": 2192
+    },
+    {
+      "epoch": 0.8776,
+      "learning_rate": 1.8139127651906193e-05,
+      "loss": 0.0814,
+      "step": 2194
+    },
+    {
+      "epoch": 0.8784,
+      "learning_rate": 1.8122871717218974e-05,
+      "loss": 0.0233,
+      "step": 2196
+    },
+    {
+      "epoch": 0.8792,
+      "learning_rate": 1.8106552438746413e-05,
+      "loss": 0.0063,
+      "step": 2198
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 1.8090169943749477e-05,
+      "loss": 0.0577,
+      "step": 2200
+    },
+    {
+      "epoch": 0.8808,
+      "learning_rate": 1.807372435998219e-05,
+      "loss": 0.0751,
+      "step": 2202
+    },
+    {
+      "epoch": 0.8816,
+      "learning_rate": 1.8057215815690487e-05,
+      "loss": 0.0273,
+      "step": 2204
+    },
+    {
+      "epoch": 0.8824,
+      "learning_rate": 1.8040644439611355e-05,
+      "loss": 0.008,
+      "step": 2206
+    },
+    {
+      "epoch": 0.8832,
+      "learning_rate": 1.8024010360971665e-05,
+      "loss": 0.0224,
+      "step": 2208
+    },
+    {
+      "epoch": 0.884,
+      "learning_rate": 1.8007313709487345e-05,
+      "loss": 0.2811,
+      "step": 2210
+    },
+    {
+      "epoch": 0.8848,
+      "learning_rate": 1.7990554615362207e-05,
+      "loss": 0.4007,
+      "step": 2212
+    },
+    {
+      "epoch": 0.8856,
+      "learning_rate": 1.7973733209287036e-05,
+      "loss": 0.0078,
+      "step": 2214
+    },
+    {
+      "epoch": 0.8864,
+      "learning_rate": 1.7956849622438568e-05,
+      "loss": 0.161,
+      "step": 2216
+    },
+    {
+      "epoch": 0.8872,
+      "learning_rate": 1.7939903986478357e-05,
+      "loss": 0.5141,
+      "step": 2218
+    },
+    {
+      "epoch": 0.888,
+      "learning_rate": 1.7922896433551913e-05,
+      "loss": 0.0496,
+      "step": 2220
+    },
+    {
+      "epoch": 0.8888,
+      "learning_rate": 1.7905827096287525e-05,
+      "loss": 0.0023,
+      "step": 2222
+    },
+    {
+      "epoch": 0.8896,
+      "learning_rate": 1.7888696107795347e-05,
+      "loss": 0.0266,
+      "step": 2224
+    },
+    {
+      "epoch": 0.8904,
+      "learning_rate": 1.787150360166623e-05,
+      "loss": 0.2279,
+      "step": 2226
+    },
+    {
+      "epoch": 0.8912,
+      "learning_rate": 1.7854249711970826e-05,
+      "loss": 0.0023,
+      "step": 2228
+    },
+    {
+      "epoch": 0.892,
+      "learning_rate": 1.783693457325841e-05,
+      "loss": 0.0088,
+      "step": 2230
+    },
+    {
+      "epoch": 0.8928,
+      "learning_rate": 1.7819558320555902e-05,
+      "loss": 0.1709,
+      "step": 2232
+    },
+    {
+      "epoch": 0.8936,
+      "learning_rate": 1.780212108936685e-05,
+      "loss": 0.0026,
+      "step": 2234
+    },
+    {
+      "epoch": 0.8944,
+      "learning_rate": 1.7784623015670237e-05,
+      "loss": 0.0001,
+      "step": 2236
+    },
+    {
+      "epoch": 0.8952,
+      "learning_rate": 1.7767064235919594e-05,
+      "loss": 0.0021,
+      "step": 2238
+    },
+    {
+      "epoch": 0.896,
+      "learning_rate": 1.77494448870418e-05,
+      "loss": 0.5575,
+      "step": 2240
+    },
+    {
+      "epoch": 0.8968,
+      "learning_rate": 1.773176510643608e-05,
+      "loss": 0.8552,
+      "step": 2242
+    },
+    {
+      "epoch": 0.8976,
+      "learning_rate": 1.7714025031972894e-05,
+      "loss": 0.006,
+      "step": 2244
+    },
+    {
+      "epoch": 0.8984,
+      "learning_rate": 1.769622480199295e-05,
+      "loss": 0.4348,
+      "step": 2246
+    },
+    {
+      "epoch": 0.8992,
+      "learning_rate": 1.7678364555305982e-05,
+      "loss": 0.0395,
+      "step": 2248
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 1.7660444431189777e-05,
+      "loss": 0.0625,
+      "step": 2250
+    },
+    {
+      "epoch": 0.9008,
+      "learning_rate": 1.76424645693891e-05,
+      "loss": 0.4662,
+      "step": 2252
+    },
+    {
+      "epoch": 0.9016,
+      "learning_rate": 1.762442511011448e-05,
+      "loss": 0.0017,
+      "step": 2254
+    },
+    {
+      "epoch": 0.9024,
+      "learning_rate": 1.7606326194041285e-05,
+      "loss": 0.1559,
+      "step": 2256
+    },
+    {
+      "epoch": 0.9032,
+      "learning_rate": 1.7588167962308458e-05,
+      "loss": 0.6804,
+      "step": 2258
+    },
+    {
+      "epoch": 0.904,
+      "learning_rate": 1.756995055651757e-05,
+      "loss": 0.0232,
+      "step": 2260
+    },
+    {
+      "epoch": 0.9048,
+      "learning_rate": 1.7551674118731585e-05,
+      "loss": 0.0438,
+      "step": 2262
+    },
+    {
+      "epoch": 0.9056,
+      "learning_rate": 1.7533338791473875e-05,
+      "loss": 0.0052,
+      "step": 2264
+    },
+    {
+      "epoch": 0.9064,
+      "learning_rate": 1.751494471772697e-05,
+      "loss": 0.1633,
+      "step": 2266
+    },
+    {
+      "epoch": 0.9072,
+      "learning_rate": 1.7496492040931548e-05,
+      "loss": 0.1294,
+      "step": 2268
+    },
+    {
+      "epoch": 0.908,
+      "learning_rate": 1.747798090498533e-05,
+      "loss": 0.0381,
+      "step": 2270
+    },
+    {
+      "epoch": 0.9088,
+      "learning_rate": 1.745941145424182e-05,
+      "loss": 0.473,
+      "step": 2272
+    },
+    {
+      "epoch": 0.9096,
+      "learning_rate": 1.744078383350938e-05,
+      "loss": 0.247,
+      "step": 2274
+    },
+    {
+      "epoch": 0.9104,
+      "learning_rate": 1.7422098188049885e-05,
+      "loss": 0.0817,
+      "step": 2276
+    },
+    {
+      "epoch": 0.9112,
+      "learning_rate": 1.7403354663577782e-05,
+      "loss": 0.004,
+      "step": 2278
+    },
+    {
+      "epoch": 0.912,
+      "learning_rate": 1.738455340625883e-05,
+      "loss": 0.4078,
+      "step": 2280
+    },
+    {
+      "epoch": 0.9128,
+      "learning_rate": 1.7365694562709038e-05,
+      "loss": 0.0037,
+      "step": 2282
+    },
+    {
+      "epoch": 0.9136,
+      "learning_rate": 1.7346778279993433e-05,
+      "loss": 0.0082,
+      "step": 2284
+    },
+    {
+      "epoch": 0.9144,
+      "learning_rate": 1.7327804705624962e-05,
+      "loss": 0.0443,
+      "step": 2286
+    },
+    {
+      "epoch": 0.9152,
+      "learning_rate": 1.730877398756341e-05,
+      "loss": 0.0072,
+      "step": 2288
+    },
+    {
+      "epoch": 0.916,
+      "learning_rate": 1.7289686274214113e-05,
+      "loss": 0.0628,
+      "step": 2290
+    },
+    {
+      "epoch": 0.9168,
+      "learning_rate": 1.727054171442693e-05,
+      "loss": 0.2357,
+      "step": 2292
+    },
+    {
+      "epoch": 0.9176,
+      "learning_rate": 1.7251340457494934e-05,
+      "loss": 0.3079,
+      "step": 2294
+    },
+    {
+      "epoch": 0.9184,
+      "learning_rate": 1.7232082653153422e-05,
+      "loss": 0.2869,
+      "step": 2296
+    },
+    {
+      "epoch": 0.9192,
+      "learning_rate": 1.7212768451578595e-05,
+      "loss": 0.0453,
+      "step": 2298
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 1.7193398003386517e-05,
+      "loss": 0.0131,
+      "step": 2300
+    },
+    {
+      "epoch": 0.9208,
+      "learning_rate": 1.7173971459631803e-05,
+      "loss": 0.0003,
+      "step": 2302
+    },
+    {
+      "epoch": 0.9216,
+      "learning_rate": 1.7154488971806525e-05,
+      "loss": 0.0726,
+      "step": 2304
+    },
+    {
+      "epoch": 0.9224,
+      "learning_rate": 1.713495069183907e-05,
+      "loss": 0.0008,
+      "step": 2306
+    },
+    {
+      "epoch": 0.9232,
+      "learning_rate": 1.7115356772092847e-05,
+      "loss": 0.0026,
+      "step": 2308
+    },
+    {
+      "epoch": 0.924,
+      "learning_rate": 1.709570736536522e-05,
+      "loss": 0.266,
+      "step": 2310
+    },
+    {
+      "epoch": 0.9248,
+      "learning_rate": 1.7076002624886152e-05,
+      "loss": 0.005,
+      "step": 2312
+    },
+    {
+      "epoch": 0.9256,
+      "learning_rate": 1.705624270431722e-05,
+      "loss": 0.0005,
+      "step": 2314
+    },
+    {
+      "epoch": 0.9264,
+      "learning_rate": 1.70364277577502e-05,
+      "loss": 0.0004,
+      "step": 2316
+    },
+    {
+      "epoch": 0.9272,
+      "learning_rate": 1.7016557939706078e-05,
+      "loss": 0.0146,
+      "step": 2318
+    },
+    {
+      "epoch": 0.928,
+      "learning_rate": 1.6996633405133673e-05,
+      "loss": 0.0005,
+      "step": 2320
+    },
+    {
+      "epoch": 0.9288,
+      "learning_rate": 1.6976654309408468e-05,
+      "loss": 0.0005,
+      "step": 2322
+    },
+    {
+      "epoch": 0.9296,
+      "learning_rate": 1.6956620808331515e-05,
+      "loss": 0.0188,
+      "step": 2324
+    },
+    {
+      "epoch": 0.9304,
+      "learning_rate": 1.6936533058128042e-05,
+      "loss": 0.0016,
+      "step": 2326
+    },
+    {
+      "epoch": 0.9312,
+      "learning_rate": 1.691639121544641e-05,
+      "loss": 0.0101,
+      "step": 2328
+    },
+    {
+      "epoch": 0.932,
+      "learning_rate": 1.6896195437356696e-05,
+      "loss": 0.0724,
+      "step": 2330
+    },
+    {
+      "epoch": 0.9328,
+      "learning_rate": 1.6875945881349686e-05,
+      "loss": 0.5693,
+      "step": 2332
+    },
+    {
+      "epoch": 0.9336,
+      "learning_rate": 1.6855642705335435e-05,
+      "loss": 0.0031,
+      "step": 2334
+    },
+    {
+      "epoch": 0.9344,
+      "learning_rate": 1.6835286067642228e-05,
+      "loss": 0.0002,
+      "step": 2336
+    },
+    {
+      "epoch": 0.9352,
+      "learning_rate": 1.681487612701521e-05,
+      "loss": 0.2462,
+      "step": 2338
+    },
+    {
+      "epoch": 0.936,
+      "learning_rate": 1.6794413042615168e-05,
+      "loss": 0.004,
+      "step": 2340
+    },
+    {
+      "epoch": 0.9368,
+      "learning_rate": 1.677389697401739e-05,
+      "loss": 0.0028,
+      "step": 2342
+    },
+    {
+      "epoch": 0.9376,
+      "learning_rate": 1.675332808121025e-05,
+      "loss": 0.0787,
+      "step": 2344
+    },
+    {
+      "epoch": 0.9384,
+      "learning_rate": 1.6732706524594145e-05,
+      "loss": 0.4253,
+      "step": 2346
+    },
+    {
+      "epoch": 0.9392,
+      "learning_rate": 1.671203246498009e-05,
+      "loss": 0.0914,
+      "step": 2348
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 1.6691306063588593e-05,
+      "loss": 0.0402,
+      "step": 2350
+    },
+    {
+      "epoch": 0.9408,
+      "learning_rate": 1.6670527482048242e-05,
+      "loss": 0.003,
+      "step": 2352
+    },
+    {
+      "epoch": 0.9416,
+      "learning_rate": 1.6649696882394635e-05,
+      "loss": 0.0003,
+      "step": 2354
+    },
+    {
+      "epoch": 0.9424,
+      "learning_rate": 1.6628814427068968e-05,
+      "loss": 0.0497,
+      "step": 2356
+    },
+    {
+      "epoch": 0.9432,
+      "learning_rate": 1.6607880278916778e-05,
+      "loss": 0.031,
+      "step": 2358
+    },
+    {
+      "epoch": 0.944,
+      "learning_rate": 1.6586894601186824e-05,
+      "loss": 0.0002,
+      "step": 2360
+    },
+    {
+      "epoch": 0.9448,
+      "learning_rate": 1.656585755752957e-05,
+      "loss": 0.0034,
+      "step": 2362
+    },
+    {
+      "epoch": 0.9456,
+      "learning_rate": 1.6544769311996153e-05,
+      "loss": 0.0004,
+      "step": 2364
+    },
+    {
+      "epoch": 0.9464,
+      "learning_rate": 1.6523630029036924e-05,
+      "loss": 0.0001,
+      "step": 2366
+    },
+    {
+      "epoch": 0.9472,
+      "learning_rate": 1.6502439873500294e-05,
+      "loss": 1.1633,
+      "step": 2368
+    },
+    {
+      "epoch": 0.948,
+      "learning_rate": 1.6481199010631305e-05,
+      "loss": 0.0428,
+      "step": 2370
+    },
+    {
+      "epoch": 0.9488,
+      "learning_rate": 1.645990760607052e-05,
+      "loss": 0.0014,
+      "step": 2372
+    },
+    {
+      "epoch": 0.9496,
+      "learning_rate": 1.643856582585255e-05,
+      "loss": 0.0009,
+      "step": 2374
+    },
+    {
+      "epoch": 0.9504,
+      "learning_rate": 1.641717383640488e-05,
+      "loss": 0.1227,
+      "step": 2376
+    },
+    {
+      "epoch": 0.9512,
+      "learning_rate": 1.6395731804546596e-05,
+      "loss": 0.0003,
+      "step": 2378
+    },
+    {
+      "epoch": 0.952,
+      "learning_rate": 1.63742398974869e-05,
+      "loss": 0.2144,
+      "step": 2380
+    },
+    {
+      "epoch": 0.9528,
+      "learning_rate": 1.6352698282824045e-05,
+      "loss": 0.0011,
+      "step": 2382
+    },
+    {
+      "epoch": 0.9536,
+      "learning_rate": 1.633110712854385e-05,
+      "loss": 0.1121,
+      "step": 2384
+    },
+    {
+      "epoch": 0.9544,
+      "learning_rate": 1.6309466603018504e-05,
+      "loss": 0.1026,
+      "step": 2386
+    },
+    {
+      "epoch": 0.9552,
+      "learning_rate": 1.6287776875005148e-05,
+      "loss": 0.2669,
+      "step": 2388
+    },
+    {
+      "epoch": 0.956,
+      "learning_rate": 1.6266038113644612e-05,
+      "loss": 0.3882,
+      "step": 2390
+    },
+    {
+      "epoch": 0.9568,
+      "learning_rate": 1.624425048846017e-05,
+      "loss": 0.01,
+      "step": 2392
+    },
+    {
+      "epoch": 0.9576,
+      "learning_rate": 1.6222414169356063e-05,
+      "loss": 0.0017,
+      "step": 2394
+    },
+    {
+      "epoch": 0.9584,
+      "learning_rate": 1.6200529326616343e-05,
+      "loss": 0.0099,
+      "step": 2396
+    },
+    {
+      "epoch": 0.9592,
+      "learning_rate": 1.6178596130903345e-05,
+      "loss": 0.0004,
+      "step": 2398
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 1.6156614753256587e-05,
+      "loss": 0.13,
+      "step": 2400
+    },
+    {
+      "epoch": 0.9608,
+      "learning_rate": 1.613458536509123e-05,
+      "loss": 0.0671,
+      "step": 2402
+    },
+    {
+      "epoch": 0.9616,
+      "learning_rate": 1.6112508138196922e-05,
+      "loss": 2.566,
+      "step": 2404
+    },
+    {
+      "epoch": 0.9624,
+      "learning_rate": 1.6090383244736277e-05,
+      "loss": 0.001,
+      "step": 2406
+    },
+    {
+      "epoch": 0.9632,
+      "learning_rate": 1.606821085724363e-05,
+      "loss": 0.4241,
+      "step": 2408
+    },
+    {
+      "epoch": 0.964,
+      "learning_rate": 1.6045991148623756e-05,
+      "loss": 1.1747,
+      "step": 2410
+    },
+    {
+      "epoch": 0.9648,
+      "learning_rate": 1.602372429215038e-05,
+      "loss": 0.3063,
+      "step": 2412
+    },
+    {
+      "epoch": 0.9656,
+      "learning_rate": 1.600141046146497e-05,
+      "loss": 0.6455,
+      "step": 2414
+    },
+    {
+      "epoch": 0.9664,
+      "learning_rate": 1.597904983057519e-05,
+      "loss": 0.5169,
+      "step": 2416
+    },
+    {
+      "epoch": 0.9672,
+      "learning_rate": 1.5956642573853794e-05,
+      "loss": 0.1549,
+      "step": 2418
+    },
+    {
+      "epoch": 0.968,
+      "learning_rate": 1.5934188866037014e-05,
+      "loss": 0.0871,
+      "step": 2420
+    },
+    {
+      "epoch": 0.9688,
+      "learning_rate": 1.591168888222342e-05,
+      "loss": 0.0502,
+      "step": 2422
+    },
+    {
+      "epoch": 0.9696,
+      "learning_rate": 1.5889142797872407e-05,
+      "loss": 0.4109,
+      "step": 2424
+    },
+    {
+      "epoch": 0.9704,
+      "learning_rate": 1.5866550788802818e-05,
+      "loss": 0.2261,
+      "step": 2426
+    },
+    {
+      "epoch": 0.9712,
+      "learning_rate": 1.584391303119173e-05,
+      "loss": 0.0975,
+      "step": 2428
+    },
+    {
+      "epoch": 0.972,
+      "learning_rate": 1.582122970157289e-05,
+      "loss": 0.2487,
+      "step": 2430
+    },
+    {
+      "epoch": 0.9728,
+      "learning_rate": 1.5798500976835503e-05,
+      "loss": 0.0141,
+      "step": 2432
+    },
+    {
+      "epoch": 0.9736,
+      "learning_rate": 1.577572703422267e-05,
+      "loss": 0.6673,
+      "step": 2434
+    },
+    {
+      "epoch": 0.9744,
+      "learning_rate": 1.575290805133024e-05,
+      "loss": 0.018,
+      "step": 2436
+    },
+    {
+      "epoch": 0.9752,
+      "learning_rate": 1.5730044206105156e-05,
+      "loss": 0.1152,
+      "step": 2438
+    },
+    {
+      "epoch": 0.976,
+      "learning_rate": 1.570713567684432e-05,
+      "loss": 0.4949,
+      "step": 2440
+    },
+    {
+      "epoch": 0.9768,
+      "learning_rate": 1.5684182642193047e-05,
+      "loss": 0.1438,
+      "step": 2442
+    },
+    {
+      "epoch": 0.9776,
+      "learning_rate": 1.566118528114367e-05,
+      "loss": 0.248,
+      "step": 2444
+    },
+    {
+      "epoch": 0.9784,
+      "learning_rate": 1.563814377303429e-05,
+      "loss": 0.0297,
+      "step": 2446
+    },
+    {
+      "epoch": 0.9792,
+      "learning_rate": 1.561505829754715e-05,
+      "loss": 0.1958,
+      "step": 2448
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 1.5591929034707475e-05,
+      "loss": 0.0327,
+      "step": 2450
+    },
+    {
+      "epoch": 0.9808,
+      "learning_rate": 1.5568756164881874e-05,
+      "loss": 0.084,
+      "step": 2452
+    },
+    {
+      "epoch": 0.9816,
+      "learning_rate": 1.5545539868777085e-05,
+      "loss": 0.0017,
+      "step": 2454
+    },
+    {
+      "epoch": 0.9824,
+      "learning_rate": 1.5522280327438384e-05,
+      "loss": 0.0033,
+      "step": 2456
+    },
+    {
+      "epoch": 0.9832,
+      "learning_rate": 1.5498977722248398e-05,
+      "loss": 0.0753,
+      "step": 2458
+    },
+    {
+      "epoch": 0.984,
+      "learning_rate": 1.547563223492552e-05,
+      "loss": 0.1011,
+      "step": 2460
+    },
+    {
+      "epoch": 0.9848,
+      "learning_rate": 1.5452244047522504e-05,
+      "loss": 0.1606,
+      "step": 2462
+    },
+    {
+      "epoch": 0.9856,
+      "learning_rate": 1.5428813342425194e-05,
+      "loss": 0.0043,
+      "step": 2464
+    },
+    {
+      "epoch": 0.9864,
+      "learning_rate": 1.5405340302350876e-05,
+      "loss": 0.0161,
+      "step": 2466
+    },
+    {
+      "epoch": 0.9872,
+      "learning_rate": 1.538182511034708e-05,
+      "loss": 0.0018,
+      "step": 2468
+    },
+    {
+      "epoch": 0.988,
+      "learning_rate": 1.535826794978996e-05,
+      "loss": 0.0519,
+      "step": 2470
+    },
+    {
+      "epoch": 0.9888,
+      "learning_rate": 1.5334669004383036e-05,
+      "loss": 0.0014,
+      "step": 2472
+    },
+    {
+      "epoch": 0.9896,
+      "learning_rate": 1.5311028458155564e-05,
+      "loss": 0.0853,
+      "step": 2474
+    },
+    {
+      "epoch": 0.9904,
+      "learning_rate": 1.528734649546133e-05,
+      "loss": 0.0207,
+      "step": 2476
+    },
+    {
+      "epoch": 0.9912,
+      "learning_rate": 1.5263623300976997e-05,
+      "loss": 0.013,
+      "step": 2478
+    },
+    {
+      "epoch": 0.992,
+      "learning_rate": 1.5239859059700792e-05,
+      "loss": 0.002,
+      "step": 2480
+    },
+    {
+      "epoch": 0.9928,
+      "learning_rate": 1.5216053956951096e-05,
+      "loss": 0.0041,
+      "step": 2482
+    },
+    {
+      "epoch": 0.9936,
+      "learning_rate": 1.5192208178364819e-05,
+      "loss": 0.0017,
+      "step": 2484
+    },
+    {
+      "epoch": 0.9944,
+      "learning_rate": 1.5168321909896176e-05,
+      "loss": 0.0465,
+      "step": 2486
+    },
+    {
+      "epoch": 0.9952,
+      "learning_rate": 1.5144395337815057e-05,
+      "loss": 0.0026,
+      "step": 2488
+    },
+    {
+      "epoch": 0.996,
+      "learning_rate": 1.5120428648705722e-05,
+      "loss": 0.4636,
+      "step": 2490
+    },
+    {
+      "epoch": 0.9968,
+      "learning_rate": 1.5096422029465171e-05,
+      "loss": 0.2338,
+      "step": 2492
+    },
+    {
+      "epoch": 0.9976,
+      "learning_rate": 1.5072375667301904e-05,
+      "loss": 0.0026,
+      "step": 2494
+    },
+    {
+      "epoch": 0.9984,
+      "learning_rate": 1.5048289749734231e-05,
+      "loss": 0.0015,
+      "step": 2496
+    },
+    {
+      "epoch": 0.9992,
+      "learning_rate": 1.502416446458898e-05,
+      "loss": 0.2254,
+      "step": 2498
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 1.5000000000000014e-05,
+      "loss": 0.0009,
+      "step": 2500
+    },
+    {
+      "epoch": 1.0,
+      "step": 2500,
+      "total_flos": 1.1095339016650752e+16,
+      "train_loss": 0.14856628477514022,
+      "train_runtime": 12512.2958,
+      "train_samples_per_second": 3.197,
+      "train_steps_per_second": 0.2
+    }
+  ],
+  "logging_steps": 2,
+  "max_steps": 2500,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "save_steps": 500,
+  "stateful_callbacks": {},
+  "total_flos": 1.1095339016650752e+16,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}
diff --git a/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/server_model_round0.pth b/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/server_model_round0.pth
new file mode 100644
index 0000000000000000000000000000000000000000..58477712887115aafec1c46fbcd552d1b478eab9
--- /dev/null
+++ b/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/server_model_round0.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:23e3aa6ba728f20117c11a282ee63f55f9ff4dbbcb82c987e3ed9d378d92f896
+size 1279587682
diff --git a/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/server_model_round1.pth b/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/server_model_round1.pth
new file mode 100644
index 0000000000000000000000000000000000000000..09f20625ce98ebea41fa283bf098c8ef010edb54
--- /dev/null
+++ b/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/server_model_round1.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dbc06e1fc98f41656fccb78d839ef3e94621728f5e097a6745738e186f8109b1
+size 1279587682
diff --git a/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/server_model_round2.pth b/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/server_model_round2.pth
new file mode 100644
index 0000000000000000000000000000000000000000..47e557ee0062c8d4e608e316223c4f02ffc25dc9
--- /dev/null
+++ b/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/server_model_round2.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:89034dd8c7b24f83108cd4522fe63ff5f056122818c218027b30ee3593938d75
+size 1279587682
diff --git a/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/server_model_round3.pth b/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/server_model_round3.pth
new file mode 100644
index 0000000000000000000000000000000000000000..67220ffbea50b09920177395d0a2996e714fce47
--- /dev/null
+++ b/client_states_v9_NEURIPS_DISJOINT_Memonly_LORA_llava_lr2e-5_bs1_gradacc32_iter0_25_selfsup_scenario18_new_10000_random0_0625_seed1/server_model_round3.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ffa0fcfe130463ff30bc3ce9cf1d5dce42cf9255c165c67b2a1e2f98ac62c1ab
+size 1279587682