diff --git a/checkpoint-1188/config.json b/checkpoint-1188/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..cb8d52e14263c50fd53dca42b5050e3235115c32
--- /dev/null
+++ b/checkpoint-1188/config.json
@@ -0,0 +1,113 @@
+{
+  "_name_or_path": "facebook/wav2vec2-base-960h",
+  "activation_dropout": 0.1,
+  "adapter_kernel_size": 3,
+  "adapter_stride": 2,
+  "add_adapter": false,
+  "apply_spec_augment": true,
+  "architectures": [
+    "Wav2Vec2ForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 1,
+  "classifier_proj_size": 256,
+  "codevector_dim": 256,
+  "contrastive_logits_temperature": 0.1,
+  "conv_bias": false,
+  "conv_dim": [
+    512,
+    512,
+    512,
+    512,
+    512,
+    512,
+    512
+  ],
+  "conv_kernel": [
+    10,
+    3,
+    3,
+    3,
+    3,
+    2,
+    2
+  ],
+  "conv_stride": [
+    5,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2
+  ],
+  "ctc_loss_reduction": "sum",
+  "ctc_zero_infinity": false,
+  "diversity_loss_weight": 0.1,
+  "do_stable_layer_norm": false,
+  "eos_token_id": 2,
+  "feat_extract_activation": "gelu",
+  "feat_extract_dropout": 0.0,
+  "feat_extract_norm": "group",
+  "feat_proj_dropout": 0.1,
+  "feat_quantizer_dropout": 0.0,
+  "final_dropout": 0.1,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout": 0.1,
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Foreign",
+    "1": "Native"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "layerdrop": 0.1,
+  "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
+  "mask_feature_prob": 0.0,
+  "mask_time_length": 10,
+  "mask_time_min_masks": 2,
+  "mask_time_prob": 0.05,
+  "model_type": "wav2vec2",
+  "num_adapter_layers": 3,
+  "num_attention_heads": 12,
+  "num_codevector_groups": 2,
+  "num_codevectors_per_group": 320,
+  "num_conv_pos_embedding_groups": 16,
+  "num_conv_pos_embeddings": 128,
+  "num_feat_extract_layers": 7,
+  "num_hidden_layers": 12,
+  "num_negatives": 100,
+  "output_hidden_size": 768,
+  "pad_token_id": 0,
+  "proj_codevector_dim": 256,
+  "tdnn_dilation": [
+    1,
+    2,
+    3,
+    1,
+    1
+  ],
+  "tdnn_dim": [
+    512,
+    512,
+    512,
+    512,
+    1500
+  ],
+  "tdnn_kernel": [
+    5,
+    3,
+    3,
+    1,
+    1
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.28.1",
+  "use_weighted_layer_sum": false,
+  "vocab_size": 32,
+  "xvector_output_dim": 512
+}
diff --git a/checkpoint-1188/optimizer.pt b/checkpoint-1188/optimizer.pt
new file mode 100644
index 0000000000000000000000000000000000000000..394ee6249f7abf0dcb0639a824645e6b501c94a1
--- /dev/null
+++ b/checkpoint-1188/optimizer.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6927087e0b8865c0180e983f5ea462633228691b4ffcad35054391c508557549
+size 756677477
diff --git a/checkpoint-1188/preprocessor_config.json b/checkpoint-1188/preprocessor_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..a0b7227fc1d916e469b14f6c154ad6dfea1e6891
--- /dev/null
+++ b/checkpoint-1188/preprocessor_config.json
@@ -0,0 +1,9 @@
+{
+  "do_normalize": true,
+  "feature_extractor_type": "Wav2Vec2FeatureExtractor",
+  "feature_size": 1,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": false,
+  "sampling_rate": 16000
+}
diff --git a/checkpoint-1188/pytorch_model.bin b/checkpoint-1188/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..3eea1ff8fdf9c8740fc84f02436c6c7b2f40bb3e
--- /dev/null
+++ b/checkpoint-1188/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bd0d811da4fb0cfac1f876ced9df766e19da5c76dfb9b9cd94f0bd7b3fdcce39
+size 378348563
diff --git a/checkpoint-1188/rng_state.pth b/checkpoint-1188/rng_state.pth
new file mode 100644
index 0000000000000000000000000000000000000000..df4139620494e6135d196cda2274971d35db3148
--- /dev/null
+++ b/checkpoint-1188/rng_state.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c762f8bb79cfca006e8b4206ef0e021670190bc0b380b067527cfb02322e2fc3
+size 14631
diff --git a/checkpoint-1188/scheduler.pt b/checkpoint-1188/scheduler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..76979bb408c913b551ede57d9fa76a9e6c86cbc7
--- /dev/null
+++ b/checkpoint-1188/scheduler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a97e6e42c229abac35a91863edb487f02fb88823244bbb46e7338ea079d4fb4
+size 623
diff --git a/checkpoint-1188/trainer_state.json b/checkpoint-1188/trainer_state.json
new file mode 100644
index 0000000000000000000000000000000000000000..ad51ab82cf453110cbde8357afa9f7644c5d16fd
--- /dev/null
+++ b/checkpoint-1188/trainer_state.json
@@ -0,0 +1,7171 @@
+{
+  "best_metric": 0.3537317216396332,
+  "best_model_checkpoint": "accent_classification/checkpoint-792",
+  "epoch": 3.0,
+  "global_step": 1188,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.000000000000001e-07,
+      "loss": 0.6956,
+      "step": 1
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.2000000000000002e-06,
+      "loss": 0.6979,
+      "step": 2
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8e-06,
+      "loss": 0.7014,
+      "step": 3
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 2.4000000000000003e-06,
+      "loss": 0.6984,
+      "step": 4
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 3e-06,
+      "loss": 0.6978,
+      "step": 5
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 3.6e-06,
+      "loss": 0.6783,
+      "step": 6
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.2000000000000004e-06,
+      "loss": 0.6956,
+      "step": 7
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.800000000000001e-06,
+      "loss": 0.6994,
+      "step": 8
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 5.4e-06,
+      "loss": 0.6902,
+      "step": 9
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6e-06,
+      "loss": 0.6937,
+      "step": 10
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6.6e-06,
+      "loss": 0.6497,
+      "step": 11
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.2e-06,
+      "loss": 0.6911,
+      "step": 12
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.8e-06,
+      "loss": 0.6904,
+      "step": 13
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 8.400000000000001e-06,
+      "loss": 0.6854,
+      "step": 14
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9e-06,
+      "loss": 0.6843,
+      "step": 15
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9.600000000000001e-06,
+      "loss": 0.6851,
+      "step": 16
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.02e-05,
+      "loss": 0.6733,
+      "step": 17
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.08e-05,
+      "loss": 0.6819,
+      "step": 18
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.1400000000000001e-05,
+      "loss": 0.681,
+      "step": 19
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.6998,
+      "step": 20
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.26e-05,
+      "loss": 0.6743,
+      "step": 21
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.32e-05,
+      "loss": 0.6818,
+      "step": 22
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3800000000000002e-05,
+      "loss": 0.5331,
+      "step": 23
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.44e-05,
+      "loss": 0.6681,
+      "step": 24
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.6955,
+      "step": 25
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.56e-05,
+      "loss": 0.7,
+      "step": 26
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.62e-05,
+      "loss": 0.6975,
+      "step": 27
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.6800000000000002e-05,
+      "loss": 0.6249,
+      "step": 28
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.74e-05,
+      "loss": 0.6979,
+      "step": 29
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.6076,
+      "step": 30
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.86e-05,
+      "loss": 0.6091,
+      "step": 31
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9200000000000003e-05,
+      "loss": 0.6438,
+      "step": 32
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.98e-05,
+      "loss": 0.6624,
+      "step": 33
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.04e-05,
+      "loss": 0.6452,
+      "step": 34
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.6963,
+      "step": 35
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.16e-05,
+      "loss": 0.758,
+      "step": 36
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.22e-05,
+      "loss": 0.6471,
+      "step": 37
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.2800000000000002e-05,
+      "loss": 0.6313,
+      "step": 38
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.3400000000000003e-05,
+      "loss": 0.7019,
+      "step": 39
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.6336,
+      "step": 40
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4599999999999998e-05,
+      "loss": 0.7091,
+      "step": 41
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.52e-05,
+      "loss": 0.701,
+      "step": 42
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.58e-05,
+      "loss": 0.7036,
+      "step": 43
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.64e-05,
+      "loss": 0.7073,
+      "step": 44
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.614,
+      "step": 45
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.7600000000000003e-05,
+      "loss": 0.7543,
+      "step": 46
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.8199999999999998e-05,
+      "loss": 0.538,
+      "step": 47
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.88e-05,
+      "loss": 0.5179,
+      "step": 48
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.94e-05,
+      "loss": 0.6164,
+      "step": 49
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 3e-05,
+      "loss": 0.6093,
+      "step": 50
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.99923273657289e-05,
+      "loss": 0.6983,
+      "step": 51
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9984654731457804e-05,
+      "loss": 0.6069,
+      "step": 52
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9976982097186702e-05,
+      "loss": 0.5933,
+      "step": 53
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.99693094629156e-05,
+      "loss": 0.552,
+      "step": 54
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9961636828644502e-05,
+      "loss": 0.5865,
+      "step": 55
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9953964194373404e-05,
+      "loss": 0.7146,
+      "step": 56
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9946291560102302e-05,
+      "loss": 0.5591,
+      "step": 57
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9938618925831203e-05,
+      "loss": 0.7366,
+      "step": 58
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9930946291560102e-05,
+      "loss": 0.7006,
+      "step": 59
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9923273657289003e-05,
+      "loss": 0.4148,
+      "step": 60
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9915601023017905e-05,
+      "loss": 0.9938,
+      "step": 61
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9907928388746803e-05,
+      "loss": 0.911,
+      "step": 62
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.99002557544757e-05,
+      "loss": 0.5682,
+      "step": 63
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9892583120204606e-05,
+      "loss": 0.5833,
+      "step": 64
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9884910485933505e-05,
+      "loss": 0.7237,
+      "step": 65
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9877237851662403e-05,
+      "loss": 0.4013,
+      "step": 66
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9869565217391304e-05,
+      "loss": 0.9695,
+      "step": 67
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9861892583120206e-05,
+      "loss": 0.568,
+      "step": 68
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9854219948849104e-05,
+      "loss": 0.5697,
+      "step": 69
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9846547314578006e-05,
+      "loss": 0.7675,
+      "step": 70
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9838874680306904e-05,
+      "loss": 0.8831,
+      "step": 71
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9831202046035806e-05,
+      "loss": 0.7008,
+      "step": 72
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9823529411764707e-05,
+      "loss": 0.7085,
+      "step": 73
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9815856777493606e-05,
+      "loss": 0.5374,
+      "step": 74
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9808184143222507e-05,
+      "loss": 0.4131,
+      "step": 75
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.980051150895141e-05,
+      "loss": 0.7309,
+      "step": 76
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9792838874680307e-05,
+      "loss": 0.4279,
+      "step": 77
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.978516624040921e-05,
+      "loss": 0.5813,
+      "step": 78
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.977749360613811e-05,
+      "loss": 0.6276,
+      "step": 79
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976982097186701e-05,
+      "loss": 0.7892,
+      "step": 80
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976214833759591e-05,
+      "loss": 0.7134,
+      "step": 81
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.9754475703324808e-05,
+      "loss": 0.3682,
+      "step": 82
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.974680306905371e-05,
+      "loss": 0.5562,
+      "step": 83
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973913043478261e-05,
+      "loss": 0.7158,
+      "step": 84
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973145780051151e-05,
+      "loss": 0.524,
+      "step": 85
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9723785166240408e-05,
+      "loss": 0.3372,
+      "step": 86
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9716112531969313e-05,
+      "loss": 0.4979,
+      "step": 87
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970843989769821e-05,
+      "loss": 0.3177,
+      "step": 88
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970076726342711e-05,
+      "loss": 0.3488,
+      "step": 89
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.969309462915601e-05,
+      "loss": 0.5158,
+      "step": 90
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9685421994884913e-05,
+      "loss": 0.3208,
+      "step": 91
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.967774936061381e-05,
+      "loss": 0.6654,
+      "step": 92
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9670076726342712e-05,
+      "loss": 0.5581,
+      "step": 93
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.966240409207161e-05,
+      "loss": 0.7456,
+      "step": 94
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9654731457800512e-05,
+      "loss": 0.4437,
+      "step": 95
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9647058823529414e-05,
+      "loss": 0.5525,
+      "step": 96
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9639386189258312e-05,
+      "loss": 0.6673,
+      "step": 97
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.963171355498721e-05,
+      "loss": 0.435,
+      "step": 98
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9624040920716115e-05,
+      "loss": 0.7415,
+      "step": 99
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9616368286445014e-05,
+      "loss": 0.2981,
+      "step": 100
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9608695652173912e-05,
+      "loss": 0.5179,
+      "step": 101
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9601023017902817e-05,
+      "loss": 0.4005,
+      "step": 102
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9593350383631715e-05,
+      "loss": 0.4598,
+      "step": 103
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9585677749360613e-05,
+      "loss": 0.2553,
+      "step": 104
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9578005115089515e-05,
+      "loss": 0.4302,
+      "step": 105
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9570332480818416e-05,
+      "loss": 0.7894,
+      "step": 106
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9562659846547315e-05,
+      "loss": 0.213,
+      "step": 107
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9554987212276216e-05,
+      "loss": 0.5507,
+      "step": 108
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9547314578005115e-05,
+      "loss": 0.7676,
+      "step": 109
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9539641943734016e-05,
+      "loss": 0.3418,
+      "step": 110
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9531969309462918e-05,
+      "loss": 0.8533,
+      "step": 111
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9524296675191816e-05,
+      "loss": 0.3659,
+      "step": 112
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9516624040920714e-05,
+      "loss": 0.6482,
+      "step": 113
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.950895140664962e-05,
+      "loss": 0.5484,
+      "step": 114
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9501278772378517e-05,
+      "loss": 0.4508,
+      "step": 115
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9493606138107416e-05,
+      "loss": 0.4401,
+      "step": 116
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9485933503836317e-05,
+      "loss": 0.5455,
+      "step": 117
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.947826086956522e-05,
+      "loss": 0.722,
+      "step": 118
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9470588235294117e-05,
+      "loss": 0.2874,
+      "step": 119
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.946291560102302e-05,
+      "loss": 0.5973,
+      "step": 120
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.9455242966751917e-05,
+      "loss": 0.5898,
+      "step": 121
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.944757033248082e-05,
+      "loss": 1.0125,
+      "step": 122
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943989769820972e-05,
+      "loss": 0.2511,
+      "step": 123
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943222506393862e-05,
+      "loss": 0.545,
+      "step": 124
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.942455242966752e-05,
+      "loss": 0.593,
+      "step": 125
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.941687979539642e-05,
+      "loss": 0.2319,
+      "step": 126
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.940920716112532e-05,
+      "loss": 0.2215,
+      "step": 127
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.9401534526854218e-05,
+      "loss": 0.1964,
+      "step": 128
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9393861892583123e-05,
+      "loss": 0.1471,
+      "step": 129
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.938618925831202e-05,
+      "loss": 1.5965,
+      "step": 130
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9378516624040923e-05,
+      "loss": 1.0772,
+      "step": 131
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.937084398976982e-05,
+      "loss": 0.2005,
+      "step": 132
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9363171355498723e-05,
+      "loss": 0.9628,
+      "step": 133
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9355498721227624e-05,
+      "loss": 0.1503,
+      "step": 134
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9347826086956523e-05,
+      "loss": 0.6201,
+      "step": 135
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.934015345268542e-05,
+      "loss": 0.6094,
+      "step": 136
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9332480818414326e-05,
+      "loss": 1.385,
+      "step": 137
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9324808184143224e-05,
+      "loss": 0.1552,
+      "step": 138
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9317135549872122e-05,
+      "loss": 0.5828,
+      "step": 139
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9309462915601024e-05,
+      "loss": 0.6018,
+      "step": 140
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9301790281329925e-05,
+      "loss": 1.0614,
+      "step": 141
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9294117647058824e-05,
+      "loss": 0.566,
+      "step": 142
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9286445012787725e-05,
+      "loss": 0.5958,
+      "step": 143
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9278772378516624e-05,
+      "loss": 0.5838,
+      "step": 144
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9271099744245525e-05,
+      "loss": 0.5909,
+      "step": 145
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9263427109974427e-05,
+      "loss": 0.5862,
+      "step": 146
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9255754475703325e-05,
+      "loss": 0.2031,
+      "step": 147
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9248081841432223e-05,
+      "loss": 0.1995,
+      "step": 148
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9240409207161128e-05,
+      "loss": 0.24,
+      "step": 149
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9232736572890026e-05,
+      "loss": 0.5845,
+      "step": 150
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9225063938618925e-05,
+      "loss": 0.964,
+      "step": 151
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.921739130434783e-05,
+      "loss": 0.9595,
+      "step": 152
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9209718670076728e-05,
+      "loss": 0.2033,
+      "step": 153
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9202046035805626e-05,
+      "loss": 0.213,
+      "step": 154
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9194373401534528e-05,
+      "loss": 0.2042,
+      "step": 155
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.918670076726343e-05,
+      "loss": 0.5783,
+      "step": 156
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9179028132992328e-05,
+      "loss": 1.0822,
+      "step": 157
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.917135549872123e-05,
+      "loss": 1.2643,
+      "step": 158
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9163682864450127e-05,
+      "loss": 0.2593,
+      "step": 159
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.915601023017903e-05,
+      "loss": 0.5783,
+      "step": 160
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914833759590793e-05,
+      "loss": 0.5721,
+      "step": 161
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914066496163683e-05,
+      "loss": 0.5808,
+      "step": 162
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9132992327365727e-05,
+      "loss": 1.2578,
+      "step": 163
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9125319693094632e-05,
+      "loss": 0.5709,
+      "step": 164
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.911764705882353e-05,
+      "loss": 0.2159,
+      "step": 165
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910997442455243e-05,
+      "loss": 0.2329,
+      "step": 166
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910230179028133e-05,
+      "loss": 0.926,
+      "step": 167
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.9094629156010232e-05,
+      "loss": 0.9095,
+      "step": 168
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.908695652173913e-05,
+      "loss": 0.5622,
+      "step": 169
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907928388746803e-05,
+      "loss": 0.5703,
+      "step": 170
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907161125319693e-05,
+      "loss": 0.5669,
+      "step": 171
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.906393861892583e-05,
+      "loss": 0.2637,
+      "step": 172
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9056265984654733e-05,
+      "loss": 1.1077,
+      "step": 173
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.904859335038363e-05,
+      "loss": 0.5653,
+      "step": 174
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9040920716112533e-05,
+      "loss": 0.5584,
+      "step": 175
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9033248081841434e-05,
+      "loss": 0.5544,
+      "step": 176
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9025575447570333e-05,
+      "loss": 0.5663,
+      "step": 177
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.901790281329923e-05,
+      "loss": 1.0808,
+      "step": 178
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9010230179028136e-05,
+      "loss": 0.8736,
+      "step": 179
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9002557544757034e-05,
+      "loss": 0.6435,
+      "step": 180
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8994884910485932e-05,
+      "loss": 0.786,
+      "step": 181
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8987212276214834e-05,
+      "loss": 0.5717,
+      "step": 182
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8979539641943736e-05,
+      "loss": 0.5618,
+      "step": 183
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8971867007672637e-05,
+      "loss": 0.5785,
+      "step": 184
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8964194373401535e-05,
+      "loss": 0.7601,
+      "step": 185
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8956521739130434e-05,
+      "loss": 0.8112,
+      "step": 186
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.894884910485934e-05,
+      "loss": 0.5663,
+      "step": 187
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8941176470588237e-05,
+      "loss": 0.5631,
+      "step": 188
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8933503836317135e-05,
+      "loss": 0.7363,
+      "step": 189
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8925831202046037e-05,
+      "loss": 0.4552,
+      "step": 190
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8918158567774938e-05,
+      "loss": 0.4599,
+      "step": 191
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8910485933503836e-05,
+      "loss": 0.7848,
+      "step": 192
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8902813299232738e-05,
+      "loss": 0.5907,
+      "step": 193
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8895140664961636e-05,
+      "loss": 0.4264,
+      "step": 194
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8887468030690538e-05,
+      "loss": 0.5697,
+      "step": 195
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.887979539641944e-05,
+      "loss": 0.7643,
+      "step": 196
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8872122762148338e-05,
+      "loss": 0.3635,
+      "step": 197
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8864450127877236e-05,
+      "loss": 0.3715,
+      "step": 198
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.885677749360614e-05,
+      "loss": 0.554,
+      "step": 199
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.884910485933504e-05,
+      "loss": 0.8044,
+      "step": 200
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8841432225063937e-05,
+      "loss": 0.5673,
+      "step": 201
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8833759590792842e-05,
+      "loss": 0.8042,
+      "step": 202
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.882608695652174e-05,
+      "loss": 0.5638,
+      "step": 203
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881841432225064e-05,
+      "loss": 0.3584,
+      "step": 204
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881074168797954e-05,
+      "loss": 0.8211,
+      "step": 205
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.8803069053708442e-05,
+      "loss": 0.5708,
+      "step": 206
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.879539641943734e-05,
+      "loss": 0.8301,
+      "step": 207
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8787723785166242e-05,
+      "loss": 0.8229,
+      "step": 208
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.878005115089514e-05,
+      "loss": 0.5645,
+      "step": 209
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8772378516624042e-05,
+      "loss": 0.3529,
+      "step": 210
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8764705882352943e-05,
+      "loss": 0.5578,
+      "step": 211
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.875703324808184e-05,
+      "loss": 0.5659,
+      "step": 212
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.874936061381074e-05,
+      "loss": 0.7495,
+      "step": 213
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8741687979539645e-05,
+      "loss": 0.3162,
+      "step": 214
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8734015345268543e-05,
+      "loss": 0.5616,
+      "step": 215
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.872634271099744e-05,
+      "loss": 0.4034,
+      "step": 216
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8718670076726343e-05,
+      "loss": 0.3163,
+      "step": 217
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8710997442455245e-05,
+      "loss": 0.5514,
+      "step": 218
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8703324808184143e-05,
+      "loss": 0.826,
+      "step": 219
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8695652173913044e-05,
+      "loss": 0.8363,
+      "step": 220
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8687979539641943e-05,
+      "loss": 0.8288,
+      "step": 221
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8680306905370844e-05,
+      "loss": 0.5929,
+      "step": 222
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8672634271099746e-05,
+      "loss": 0.565,
+      "step": 223
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8664961636828644e-05,
+      "loss": 0.5642,
+      "step": 224
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8657289002557546e-05,
+      "loss": 0.5609,
+      "step": 225
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8649616368286447e-05,
+      "loss": 0.5641,
+      "step": 226
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8641943734015345e-05,
+      "loss": 0.8535,
+      "step": 227
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8634271099744244e-05,
+      "loss": 0.5652,
+      "step": 228
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.862659846547315e-05,
+      "loss": 0.4397,
+      "step": 229
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8618925831202047e-05,
+      "loss": 0.567,
+      "step": 230
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8611253196930945e-05,
+      "loss": 0.7517,
+      "step": 231
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8603580562659847e-05,
+      "loss": 0.5839,
+      "step": 232
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.859590792838875e-05,
+      "loss": 0.5677,
+      "step": 233
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8588235294117647e-05,
+      "loss": 0.5678,
+      "step": 234
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8580562659846548e-05,
+      "loss": 1.3045,
+      "step": 235
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8572890025575446e-05,
+      "loss": 0.804,
+      "step": 236
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.856521739130435e-05,
+      "loss": 1.0303,
+      "step": 237
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.855754475703325e-05,
+      "loss": 0.3303,
+      "step": 238
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8549872122762148e-05,
+      "loss": 0.5677,
+      "step": 239
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.854219948849105e-05,
+      "loss": 0.7972,
+      "step": 240
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.853452685421995e-05,
+      "loss": 0.5644,
+      "step": 241
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.852685421994885e-05,
+      "loss": 0.9294,
+      "step": 242
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.851918158567775e-05,
+      "loss": 0.5665,
+      "step": 243
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.851150895140665e-05,
+      "loss": 0.7963,
+      "step": 244
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.850383631713555e-05,
+      "loss": 0.5659,
+      "step": 245
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.8496163682864452e-05,
+      "loss": 0.5718,
+      "step": 246
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.848849104859335e-05,
+      "loss": 0.3526,
+      "step": 247
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8480818414322252e-05,
+      "loss": 0.574,
+      "step": 248
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8473145780051154e-05,
+      "loss": 0.9996,
+      "step": 249
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8465473145780052e-05,
+      "loss": 0.7634,
+      "step": 250
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.845780051150895e-05,
+      "loss": 0.5639,
+      "step": 251
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8450127877237855e-05,
+      "loss": 0.5737,
+      "step": 252
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8442455242966753e-05,
+      "loss": 0.7793,
+      "step": 253
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8434782608695652e-05,
+      "loss": 0.6769,
+      "step": 254
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8427109974424553e-05,
+      "loss": 0.5748,
+      "step": 255
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8419437340153455e-05,
+      "loss": 0.3602,
+      "step": 256
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8411764705882353e-05,
+      "loss": 0.5714,
+      "step": 257
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8404092071611255e-05,
+      "loss": 0.5738,
+      "step": 258
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8396419437340153e-05,
+      "loss": 0.5719,
+      "step": 259
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8388746803069055e-05,
+      "loss": 0.5753,
+      "step": 260
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8381074168797956e-05,
+      "loss": 0.5717,
+      "step": 261
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8373401534526854e-05,
+      "loss": 0.9794,
+      "step": 262
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8365728900255753e-05,
+      "loss": 0.5737,
+      "step": 263
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8358056265984658e-05,
+      "loss": 0.5723,
+      "step": 264
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8350383631713556e-05,
+      "loss": 0.5656,
+      "step": 265
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8342710997442454e-05,
+      "loss": 0.4259,
+      "step": 266
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8335038363171356e-05,
+      "loss": 0.3076,
+      "step": 267
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8327365728900257e-05,
+      "loss": 0.338,
+      "step": 268
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8319693094629156e-05,
+      "loss": 0.5612,
+      "step": 269
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8312020460358057e-05,
+      "loss": 0.5249,
+      "step": 270
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8304347826086955e-05,
+      "loss": 0.5671,
+      "step": 271
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8296675191815857e-05,
+      "loss": 0.5574,
+      "step": 272
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.828900255754476e-05,
+      "loss": 0.8152,
+      "step": 273
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8281329923273657e-05,
+      "loss": 0.5647,
+      "step": 274
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.827365728900256e-05,
+      "loss": 0.5632,
+      "step": 275
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.826598465473146e-05,
+      "loss": 0.3391,
+      "step": 276
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8258312020460358e-05,
+      "loss": 0.55,
+      "step": 277
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8250639386189257e-05,
+      "loss": 0.5672,
+      "step": 278
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.824296675191816e-05,
+      "loss": 0.285,
+      "step": 279
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.823529411764706e-05,
+      "loss": 0.2809,
+      "step": 280
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.8227621483375958e-05,
+      "loss": 0.5552,
+      "step": 281
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821994884910486e-05,
+      "loss": 0.563,
+      "step": 282
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821227621483376e-05,
+      "loss": 0.5628,
+      "step": 283
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.820460358056266e-05,
+      "loss": 0.2548,
+      "step": 284
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.819693094629156e-05,
+      "loss": 0.871,
+      "step": 285
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818925831202046e-05,
+      "loss": 0.8797,
+      "step": 286
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818158567774936e-05,
+      "loss": 0.2453,
+      "step": 287
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8173913043478262e-05,
+      "loss": 0.2464,
+      "step": 288
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.816624040920716e-05,
+      "loss": 0.3134,
+      "step": 289
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8158567774936062e-05,
+      "loss": 0.2431,
+      "step": 290
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8150895140664964e-05,
+      "loss": 0.2308,
+      "step": 291
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8143222506393862e-05,
+      "loss": 0.2173,
+      "step": 292
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8135549872122764e-05,
+      "loss": 0.5665,
+      "step": 293
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8127877237851662e-05,
+      "loss": 0.5592,
+      "step": 294
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8120204603580564e-05,
+      "loss": 0.473,
+      "step": 295
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8112531969309465e-05,
+      "loss": 0.9458,
+      "step": 296
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8104859335038363e-05,
+      "loss": 0.1879,
+      "step": 297
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8097186700767265e-05,
+      "loss": 0.6797,
+      "step": 298
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8089514066496167e-05,
+      "loss": 0.7592,
+      "step": 299
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8081841432225065e-05,
+      "loss": 0.4109,
+      "step": 300
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8074168797953963e-05,
+      "loss": 0.5821,
+      "step": 301
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8066496163682868e-05,
+      "loss": 0.2057,
+      "step": 302
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8058823529411766e-05,
+      "loss": 0.5322,
+      "step": 303
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8051150895140665e-05,
+      "loss": 0.5208,
+      "step": 304
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8043478260869566e-05,
+      "loss": 0.6799,
+      "step": 305
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8035805626598468e-05,
+      "loss": 0.5894,
+      "step": 306
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8028132992327366e-05,
+      "loss": 0.4051,
+      "step": 307
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8020460358056268e-05,
+      "loss": 0.3695,
+      "step": 308
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8012787723785166e-05,
+      "loss": 0.4869,
+      "step": 309
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8005115089514067e-05,
+      "loss": 0.6075,
+      "step": 310
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.799744245524297e-05,
+      "loss": 0.4245,
+      "step": 311
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7989769820971867e-05,
+      "loss": 0.5122,
+      "step": 312
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7982097186700765e-05,
+      "loss": 0.452,
+      "step": 313
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.797442455242967e-05,
+      "loss": 0.9568,
+      "step": 314
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.796675191815857e-05,
+      "loss": 0.5846,
+      "step": 315
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.7959079283887467e-05,
+      "loss": 0.3743,
+      "step": 316
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.795140664961637e-05,
+      "loss": 0.1922,
+      "step": 317
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.794373401534527e-05,
+      "loss": 0.5303,
+      "step": 318
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.793606138107417e-05,
+      "loss": 0.7253,
+      "step": 319
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.792838874680307e-05,
+      "loss": 0.2794,
+      "step": 320
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.7920716112531968e-05,
+      "loss": 0.577,
+      "step": 321
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.791304347826087e-05,
+      "loss": 0.614,
+      "step": 322
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.790537084398977e-05,
+      "loss": 0.5882,
+      "step": 323
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789769820971867e-05,
+      "loss": 0.2503,
+      "step": 324
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789002557544757e-05,
+      "loss": 0.4195,
+      "step": 325
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.7882352941176473e-05,
+      "loss": 0.2959,
+      "step": 326
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.787468030690537e-05,
+      "loss": 0.3732,
+      "step": 327
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.786700767263427e-05,
+      "loss": 0.5729,
+      "step": 328
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7859335038363174e-05,
+      "loss": 0.3179,
+      "step": 329
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7851662404092073e-05,
+      "loss": 0.6823,
+      "step": 330
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.784398976982097e-05,
+      "loss": 0.2663,
+      "step": 331
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7836317135549872e-05,
+      "loss": 0.4436,
+      "step": 332
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7828644501278774e-05,
+      "loss": 0.3089,
+      "step": 333
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7820971867007672e-05,
+      "loss": 0.5888,
+      "step": 334
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7813299232736574e-05,
+      "loss": 0.2115,
+      "step": 335
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7805626598465472e-05,
+      "loss": 0.4005,
+      "step": 336
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7797953964194374e-05,
+      "loss": 0.3097,
+      "step": 337
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7790281329923275e-05,
+      "loss": 0.5847,
+      "step": 338
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7782608695652174e-05,
+      "loss": 1.0164,
+      "step": 339
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7774936061381072e-05,
+      "loss": 1.1321,
+      "step": 340
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7767263427109977e-05,
+      "loss": 0.592,
+      "step": 341
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7759590792838875e-05,
+      "loss": 1.0191,
+      "step": 342
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7751918158567777e-05,
+      "loss": 0.9199,
+      "step": 343
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7744245524296675e-05,
+      "loss": 0.2874,
+      "step": 344
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7736572890025576e-05,
+      "loss": 0.2469,
+      "step": 345
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7728900255754478e-05,
+      "loss": 0.8364,
+      "step": 346
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7721227621483376e-05,
+      "loss": 0.7365,
+      "step": 347
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7713554987212278e-05,
+      "loss": 0.5595,
+      "step": 348
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.770588235294118e-05,
+      "loss": 0.2546,
+      "step": 349
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7698209718670078e-05,
+      "loss": 0.8183,
+      "step": 350
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7690537084398976e-05,
+      "loss": 0.5385,
+      "step": 351
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.768286445012788e-05,
+      "loss": 0.2763,
+      "step": 352
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.767519181585678e-05,
+      "loss": 0.3045,
+      "step": 353
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7667519181585677e-05,
+      "loss": 0.3354,
+      "step": 354
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765984654731458e-05,
+      "loss": 0.9366,
+      "step": 355
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765217391304348e-05,
+      "loss": 0.6212,
+      "step": 356
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.764450127877238e-05,
+      "loss": 0.1161,
+      "step": 357
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.763682864450128e-05,
+      "loss": 0.4929,
+      "step": 358
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762915601023018e-05,
+      "loss": 0.8167,
+      "step": 359
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762148337595908e-05,
+      "loss": 0.2521,
+      "step": 360
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.7613810741687982e-05,
+      "loss": 0.4174,
+      "step": 361
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.760613810741688e-05,
+      "loss": 0.3822,
+      "step": 362
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.759846547314578e-05,
+      "loss": 0.5355,
+      "step": 363
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.7590792838874683e-05,
+      "loss": 0.561,
+      "step": 364
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.758312020460358e-05,
+      "loss": 0.5126,
+      "step": 365
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.757544757033248e-05,
+      "loss": 0.2301,
+      "step": 366
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.756777493606138e-05,
+      "loss": 0.1698,
+      "step": 367
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7560102301790283e-05,
+      "loss": 0.1634,
+      "step": 368
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.755242966751918e-05,
+      "loss": 0.5955,
+      "step": 369
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7544757033248083e-05,
+      "loss": 0.1568,
+      "step": 370
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.753708439897698e-05,
+      "loss": 0.5167,
+      "step": 371
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7529411764705883e-05,
+      "loss": 1.0028,
+      "step": 372
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7521739130434784e-05,
+      "loss": 1.2567,
+      "step": 373
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7514066496163682e-05,
+      "loss": 0.6094,
+      "step": 374
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7506393861892584e-05,
+      "loss": 0.6059,
+      "step": 375
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7498721227621486e-05,
+      "loss": 1.0593,
+      "step": 376
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7491048593350384e-05,
+      "loss": 0.1506,
+      "step": 377
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7483375959079282e-05,
+      "loss": 1.5106,
+      "step": 378
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7475703324808187e-05,
+      "loss": 0.1827,
+      "step": 379
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7468030690537085e-05,
+      "loss": 1.4961,
+      "step": 380
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7460358056265984e-05,
+      "loss": 1.0456,
+      "step": 381
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7452685421994885e-05,
+      "loss": 1.2063,
+      "step": 382
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7445012787723787e-05,
+      "loss": 0.5837,
+      "step": 383
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7437340153452685e-05,
+      "loss": 1.4536,
+      "step": 384
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7429667519181587e-05,
+      "loss": 0.5803,
+      "step": 385
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7421994884910485e-05,
+      "loss": 0.5896,
+      "step": 386
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7414322250639386e-05,
+      "loss": 0.1822,
+      "step": 387
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7406649616368288e-05,
+      "loss": 1.7893,
+      "step": 388
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7398976982097186e-05,
+      "loss": 0.9723,
+      "step": 389
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7391304347826085e-05,
+      "loss": 1.2815,
+      "step": 390
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.738363171355499e-05,
+      "loss": 0.5765,
+      "step": 391
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7375959079283888e-05,
+      "loss": 1.2429,
+      "step": 392
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7368286445012786e-05,
+      "loss": 0.1343,
+      "step": 393
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7360613810741688e-05,
+      "loss": 0.2306,
+      "step": 394
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.735294117647059e-05,
+      "loss": 0.624,
+      "step": 395
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.734526854219949e-05,
+      "loss": 0.2371,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7348484848484849,
+      "eval_loss": 0.5804409980773926,
+      "eval_runtime": 57.3336,
+      "eval_samples_per_second": 6.907,
+      "eval_steps_per_second": 1.727,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.733759590792839e-05,
+      "loss": 0.8989,
+      "step": 397
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.732992327365729e-05,
+      "loss": 0.5664,
+      "step": 398
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.7322250639386192e-05,
+      "loss": 0.8852,
+      "step": 399
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.731457800511509e-05,
+      "loss": 0.5661,
+      "step": 400
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.730690537084399e-05,
+      "loss": 0.5648,
+      "step": 401
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7299232736572894e-05,
+      "loss": 0.7333,
+      "step": 402
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7291560102301792e-05,
+      "loss": 0.3037,
+      "step": 403
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.728388746803069e-05,
+      "loss": 0.1463,
+      "step": 404
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7276214833759592e-05,
+      "loss": 0.2702,
+      "step": 405
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7268542199488493e-05,
+      "loss": 0.2693,
+      "step": 406
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.726086956521739e-05,
+      "loss": 0.5647,
+      "step": 407
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7253196930946293e-05,
+      "loss": 0.8507,
+      "step": 408
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.724552429667519e-05,
+      "loss": 0.8645,
+      "step": 409
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7237851662404093e-05,
+      "loss": 0.2649,
+      "step": 410
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7230179028132995e-05,
+      "loss": 0.6942,
+      "step": 411
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7222506393861893e-05,
+      "loss": 0.5656,
+      "step": 412
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.721483375959079e-05,
+      "loss": 0.5489,
+      "step": 413
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7207161125319696e-05,
+      "loss": 0.2652,
+      "step": 414
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7199488491048594e-05,
+      "loss": 0.8676,
+      "step": 415
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7191815856777493e-05,
+      "loss": 0.8621,
+      "step": 416
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7184143222506394e-05,
+      "loss": 0.2641,
+      "step": 417
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7176470588235296e-05,
+      "loss": 0.859,
+      "step": 418
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7168797953964194e-05,
+      "loss": 0.6492,
+      "step": 419
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7161125319693096e-05,
+      "loss": 1.1494,
+      "step": 420
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7153452685421994e-05,
+      "loss": 0.8278,
+      "step": 421
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7145780051150895e-05,
+      "loss": 0.764,
+      "step": 422
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7138107416879797e-05,
+      "loss": 0.564,
+      "step": 423
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7130434782608695e-05,
+      "loss": 0.8051,
+      "step": 424
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7122762148337597e-05,
+      "loss": 0.3052,
+      "step": 425
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.71150895140665e-05,
+      "loss": 0.2945,
+      "step": 426
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7107416879795397e-05,
+      "loss": 0.4118,
+      "step": 427
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7099744245524295e-05,
+      "loss": 0.559,
+      "step": 428
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.70920716112532e-05,
+      "loss": 0.5593,
+      "step": 429
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7084398976982098e-05,
+      "loss": 0.564,
+      "step": 430
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7076726342710996e-05,
+      "loss": 0.4454,
+      "step": 431
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7069053708439898e-05,
+      "loss": 0.4381,
+      "step": 432
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.70613810741688e-05,
+      "loss": 0.3718,
+      "step": 433
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7053708439897698e-05,
+      "loss": 0.6627,
+      "step": 434
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70460358056266e-05,
+      "loss": 0.4004,
+      "step": 435
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7038363171355498e-05,
+      "loss": 0.3855,
+      "step": 436
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70306905370844e-05,
+      "loss": 0.4799,
+      "step": 437
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70230179028133e-05,
+      "loss": 0.6714,
+      "step": 438
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70153452685422e-05,
+      "loss": 0.622,
+      "step": 439
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7007672634271097e-05,
+      "loss": 0.3916,
+      "step": 440
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.5487,
+      "step": 441
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69923273657289e-05,
+      "loss": 0.376,
+      "step": 442
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69846547314578e-05,
+      "loss": 0.3616,
+      "step": 443
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69769820971867e-05,
+      "loss": 0.8747,
+      "step": 444
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.6969309462915602e-05,
+      "loss": 0.283,
+      "step": 445
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.69616368286445e-05,
+      "loss": 0.3988,
+      "step": 446
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6953964194373402e-05,
+      "loss": 0.4569,
+      "step": 447
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6946291560102303e-05,
+      "loss": 0.2005,
+      "step": 448
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6938618925831205e-05,
+      "loss": 0.3087,
+      "step": 449
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6930946291560103e-05,
+      "loss": 0.5814,
+      "step": 450
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6923273657289e-05,
+      "loss": 0.2995,
+      "step": 451
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6915601023017907e-05,
+      "loss": 0.7806,
+      "step": 452
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6907928388746805e-05,
+      "loss": 0.2666,
+      "step": 453
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6900255754475703e-05,
+      "loss": 0.2571,
+      "step": 454
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6892583120204605e-05,
+      "loss": 0.5931,
+      "step": 455
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6884910485933506e-05,
+      "loss": 0.475,
+      "step": 456
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6877237851662404e-05,
+      "loss": 0.2372,
+      "step": 457
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6869565217391306e-05,
+      "loss": 0.1577,
+      "step": 458
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6861892583120204e-05,
+      "loss": 0.1953,
+      "step": 459
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6854219948849106e-05,
+      "loss": 0.1931,
+      "step": 460
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6846547314578008e-05,
+      "loss": 0.1829,
+      "step": 461
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6838874680306906e-05,
+      "loss": 0.1601,
+      "step": 462
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6831202046035804e-05,
+      "loss": 0.2505,
+      "step": 463
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.682352941176471e-05,
+      "loss": 0.181,
+      "step": 464
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6815856777493607e-05,
+      "loss": 0.5978,
+      "step": 465
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6808184143222505e-05,
+      "loss": 0.5458,
+      "step": 466
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6800511508951407e-05,
+      "loss": 0.1289,
+      "step": 467
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.679283887468031e-05,
+      "loss": 0.4383,
+      "step": 468
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6785166240409207e-05,
+      "loss": 0.5892,
+      "step": 469
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.677749360613811e-05,
+      "loss": 1.3565,
+      "step": 470
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.676982097186701e-05,
+      "loss": 0.7267,
+      "step": 471
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.6762148337595908e-05,
+      "loss": 0.1354,
+      "step": 472
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.675447570332481e-05,
+      "loss": 0.9291,
+      "step": 473
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.6746803069053708e-05,
+      "loss": 0.5265,
+      "step": 474
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673913043478261e-05,
+      "loss": 0.4841,
+      "step": 475
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673145780051151e-05,
+      "loss": 1.2391,
+      "step": 476
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.672378516624041e-05,
+      "loss": 0.4924,
+      "step": 477
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6716112531969308e-05,
+      "loss": 1.0076,
+      "step": 478
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6708439897698213e-05,
+      "loss": 0.6314,
+      "step": 479
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.670076726342711e-05,
+      "loss": 0.4237,
+      "step": 480
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.669309462915601e-05,
+      "loss": 0.1077,
+      "step": 481
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.668542199488491e-05,
+      "loss": 0.3038,
+      "step": 482
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6677749360613812e-05,
+      "loss": 0.6081,
+      "step": 483
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.667007672634271e-05,
+      "loss": 0.1078,
+      "step": 484
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6662404092071612e-05,
+      "loss": 0.4296,
+      "step": 485
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.665473145780051e-05,
+      "loss": 0.3981,
+      "step": 486
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6647058823529412e-05,
+      "loss": 0.5853,
+      "step": 487
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6639386189258314e-05,
+      "loss": 0.802,
+      "step": 488
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6631713554987212e-05,
+      "loss": 0.4576,
+      "step": 489
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.662404092071611e-05,
+      "loss": 0.2682,
+      "step": 490
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6616368286445015e-05,
+      "loss": 0.1991,
+      "step": 491
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6608695652173913e-05,
+      "loss": 0.1409,
+      "step": 492
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.660102301790281e-05,
+      "loss": 0.3869,
+      "step": 493
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6593350383631713e-05,
+      "loss": 0.5301,
+      "step": 494
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6585677749360615e-05,
+      "loss": 0.8116,
+      "step": 495
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6578005115089513e-05,
+      "loss": 0.0984,
+      "step": 496
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6570332480818415e-05,
+      "loss": 0.7773,
+      "step": 497
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6562659846547316e-05,
+      "loss": 1.3522,
+      "step": 498
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6554987212276215e-05,
+      "loss": 0.0862,
+      "step": 499
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6547314578005116e-05,
+      "loss": 0.0894,
+      "step": 500
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6539641943734014e-05,
+      "loss": 0.0885,
+      "step": 501
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.653196930946292e-05,
+      "loss": 0.3031,
+      "step": 502
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6524296675191818e-05,
+      "loss": 0.187,
+      "step": 503
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6516624040920716e-05,
+      "loss": 1.3063,
+      "step": 504
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6508951406649617e-05,
+      "loss": 0.6847,
+      "step": 505
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.650127877237852e-05,
+      "loss": 0.078,
+      "step": 506
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6493606138107417e-05,
+      "loss": 0.1265,
+      "step": 507
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.648593350383632e-05,
+      "loss": 0.0804,
+      "step": 508
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.6478260869565217e-05,
+      "loss": 0.6976,
+      "step": 509
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.647058823529412e-05,
+      "loss": 0.7977,
+      "step": 510
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.646291560102302e-05,
+      "loss": 0.413,
+      "step": 511
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.645524296675192e-05,
+      "loss": 0.6944,
+      "step": 512
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6447570332480817e-05,
+      "loss": 0.7177,
+      "step": 513
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6439897698209722e-05,
+      "loss": 0.0581,
+      "step": 514
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.643222506393862e-05,
+      "loss": 0.9308,
+      "step": 515
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6424552429667518e-05,
+      "loss": 0.2518,
+      "step": 516
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.641687979539642e-05,
+      "loss": 0.1014,
+      "step": 517
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640920716112532e-05,
+      "loss": 0.1641,
+      "step": 518
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640153452685422e-05,
+      "loss": 0.101,
+      "step": 519
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.639386189258312e-05,
+      "loss": 0.1097,
+      "step": 520
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6386189258312023e-05,
+      "loss": 0.987,
+      "step": 521
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.637851662404092e-05,
+      "loss": 0.6465,
+      "step": 522
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6370843989769823e-05,
+      "loss": 0.0914,
+      "step": 523
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.636317135549872e-05,
+      "loss": 0.139,
+      "step": 524
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6355498721227623e-05,
+      "loss": 0.6455,
+      "step": 525
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6347826086956524e-05,
+      "loss": 0.204,
+      "step": 526
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6340153452685422e-05,
+      "loss": 0.1269,
+      "step": 527
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.633248081841432e-05,
+      "loss": 0.2214,
+      "step": 528
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6324808184143226e-05,
+      "loss": 0.0736,
+      "step": 529
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6317135549872124e-05,
+      "loss": 0.0714,
+      "step": 530
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6309462915601022e-05,
+      "loss": 0.1388,
+      "step": 531
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6301790281329924e-05,
+      "loss": 0.7758,
+      "step": 532
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6294117647058825e-05,
+      "loss": 0.3388,
+      "step": 533
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6286445012787724e-05,
+      "loss": 1.1831,
+      "step": 534
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6278772378516625e-05,
+      "loss": 0.1166,
+      "step": 535
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6271099744245523e-05,
+      "loss": 0.2389,
+      "step": 536
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6263427109974425e-05,
+      "loss": 0.1622,
+      "step": 537
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6255754475703327e-05,
+      "loss": 0.1015,
+      "step": 538
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6248081841432225e-05,
+      "loss": 0.0658,
+      "step": 539
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6240409207161123e-05,
+      "loss": 0.0954,
+      "step": 540
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6232736572890028e-05,
+      "loss": 0.1182,
+      "step": 541
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6225063938618926e-05,
+      "loss": 0.0605,
+      "step": 542
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6217391304347824e-05,
+      "loss": 0.9265,
+      "step": 543
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6209718670076726e-05,
+      "loss": 1.2994,
+      "step": 544
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6202046035805628e-05,
+      "loss": 0.081,
+      "step": 545
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6194373401534526e-05,
+      "loss": 0.5962,
+      "step": 546
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6186700767263428e-05,
+      "loss": 0.0665,
+      "step": 547
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.617902813299233e-05,
+      "loss": 0.0655,
+      "step": 548
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6171355498721227e-05,
+      "loss": 0.0642,
+      "step": 549
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.616368286445013e-05,
+      "loss": 0.0834,
+      "step": 550
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6156010230179027e-05,
+      "loss": 0.047,
+      "step": 551
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6148337595907932e-05,
+      "loss": 0.0642,
+      "step": 552
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.614066496163683e-05,
+      "loss": 0.0594,
+      "step": 553
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.613299232736573e-05,
+      "loss": 0.0573,
+      "step": 554
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.612531969309463e-05,
+      "loss": 0.1101,
+      "step": 555
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.6117647058823532e-05,
+      "loss": 0.5121,
+      "step": 556
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.610997442455243e-05,
+      "loss": 0.0374,
+      "step": 557
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.6102301790281332e-05,
+      "loss": 0.0957,
+      "step": 558
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.609462915601023e-05,
+      "loss": 0.0488,
+      "step": 559
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.608695652173913e-05,
+      "loss": 0.0482,
+      "step": 560
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6079283887468033e-05,
+      "loss": 1.2041,
+      "step": 561
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.607161125319693e-05,
+      "loss": 0.8833,
+      "step": 562
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.606393861892583e-05,
+      "loss": 0.1544,
+      "step": 563
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6056265984654735e-05,
+      "loss": 0.046,
+      "step": 564
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6048593350383633e-05,
+      "loss": 0.9073,
+      "step": 565
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.604092071611253e-05,
+      "loss": 0.052,
+      "step": 566
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6033248081841433e-05,
+      "loss": 0.0308,
+      "step": 567
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6025575447570334e-05,
+      "loss": 0.7032,
+      "step": 568
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6017902813299232e-05,
+      "loss": 0.6947,
+      "step": 569
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6010230179028134e-05,
+      "loss": 1.5603,
+      "step": 570
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6002557544757036e-05,
+      "loss": 0.8836,
+      "step": 571
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.5994884910485934e-05,
+      "loss": 1.7439,
+      "step": 572
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5987212276214836e-05,
+      "loss": 0.0313,
+      "step": 573
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5979539641943734e-05,
+      "loss": 0.0317,
+      "step": 574
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5971867007672635e-05,
+      "loss": 0.0325,
+      "step": 575
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5964194373401537e-05,
+      "loss": 0.5,
+      "step": 576
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5956521739130435e-05,
+      "loss": 0.0675,
+      "step": 577
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5948849104859333e-05,
+      "loss": 0.6364,
+      "step": 578
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.594117647058824e-05,
+      "loss": 0.0696,
+      "step": 579
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5933503836317137e-05,
+      "loss": 0.6195,
+      "step": 580
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5925831202046035e-05,
+      "loss": 0.5796,
+      "step": 581
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5918158567774937e-05,
+      "loss": 0.0832,
+      "step": 582
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5910485933503838e-05,
+      "loss": 0.0794,
+      "step": 583
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5902813299232736e-05,
+      "loss": 0.5851,
+      "step": 584
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5895140664961638e-05,
+      "loss": 0.104,
+      "step": 585
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5887468030690536e-05,
+      "loss": 1.0492,
+      "step": 586
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5879795396419438e-05,
+      "loss": 0.0645,
+      "step": 587
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.587212276214834e-05,
+      "loss": 0.514,
+      "step": 588
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5864450127877238e-05,
+      "loss": 0.0701,
+      "step": 589
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5856777493606136e-05,
+      "loss": 0.5409,
+      "step": 590
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584910485933504e-05,
+      "loss": 0.0536,
+      "step": 591
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584143222506394e-05,
+      "loss": 0.1728,
+      "step": 592
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.5833759590792837e-05,
+      "loss": 0.0845,
+      "step": 593
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.582608695652174e-05,
+      "loss": 0.2217,
+      "step": 594
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.581841432225064e-05,
+      "loss": 0.1397,
+      "step": 595
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.581074168797954e-05,
+      "loss": 0.4119,
+      "step": 596
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.580306905370844e-05,
+      "loss": 0.3638,
+      "step": 597
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.5795396419437342e-05,
+      "loss": 0.0781,
+      "step": 598
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.578772378516624e-05,
+      "loss": 0.5926,
+      "step": 599
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5780051150895142e-05,
+      "loss": 0.1378,
+      "step": 600
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.577237851662404e-05,
+      "loss": 0.142,
+      "step": 601
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.576470588235294e-05,
+      "loss": 0.1722,
+      "step": 602
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5757033248081843e-05,
+      "loss": 0.1529,
+      "step": 603
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.574936061381074e-05,
+      "loss": 1.6083,
+      "step": 604
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5741687979539643e-05,
+      "loss": 1.8267,
+      "step": 605
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5734015345268545e-05,
+      "loss": 0.0535,
+      "step": 606
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5726342710997443e-05,
+      "loss": 0.9945,
+      "step": 607
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5718670076726345e-05,
+      "loss": 0.9133,
+      "step": 608
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5710997442455243e-05,
+      "loss": 0.5295,
+      "step": 609
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5703324808184144e-05,
+      "loss": 0.058,
+      "step": 610
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5695652173913046e-05,
+      "loss": 0.4789,
+      "step": 611
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5687979539641944e-05,
+      "loss": 0.0572,
+      "step": 612
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5680306905370842e-05,
+      "loss": 0.0897,
+      "step": 613
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5672634271099747e-05,
+      "loss": 0.056,
+      "step": 614
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5664961636828646e-05,
+      "loss": 0.0278,
+      "step": 615
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5657289002557544e-05,
+      "loss": 0.9947,
+      "step": 616
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5649616368286445e-05,
+      "loss": 0.0627,
+      "step": 617
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5641943734015347e-05,
+      "loss": 0.9335,
+      "step": 618
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5634271099744245e-05,
+      "loss": 0.9793,
+      "step": 619
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5626598465473147e-05,
+      "loss": 1.8128,
+      "step": 620
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.561892583120205e-05,
+      "loss": 1.8015,
+      "step": 621
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5611253196930947e-05,
+      "loss": 0.9259,
+      "step": 622
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.560358056265985e-05,
+      "loss": 0.9618,
+      "step": 623
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5595907928388747e-05,
+      "loss": 0.0808,
+      "step": 624
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5588235294117648e-05,
+      "loss": 1.7161,
+      "step": 625
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.558056265984655e-05,
+      "loss": 0.8879,
+      "step": 626
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5572890025575448e-05,
+      "loss": 0.043,
+      "step": 627
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5565217391304346e-05,
+      "loss": 0.9449,
+      "step": 628
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.555754475703325e-05,
+      "loss": 0.0624,
+      "step": 629
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.554987212276215e-05,
+      "loss": 0.0924,
+      "step": 630
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5542199488491048e-05,
+      "loss": 0.0715,
+      "step": 631
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.553452685421995e-05,
+      "loss": 0.4995,
+      "step": 632
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.552685421994885e-05,
+      "loss": 0.5576,
+      "step": 633
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551918158567775e-05,
+      "loss": 0.06,
+      "step": 634
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551150895140665e-05,
+      "loss": 0.0873,
+      "step": 635
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.550383631713555e-05,
+      "loss": 0.6868,
+      "step": 636
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.549616368286445e-05,
+      "loss": 0.8487,
+      "step": 637
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.5488491048593352e-05,
+      "loss": 0.5641,
+      "step": 638
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.548081841432225e-05,
+      "loss": 1.1343,
+      "step": 639
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.547314578005115e-05,
+      "loss": 0.0614,
+      "step": 640
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5465473145780054e-05,
+      "loss": 0.9414,
+      "step": 641
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5457800511508952e-05,
+      "loss": 0.9226,
+      "step": 642
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.545012787723785e-05,
+      "loss": 0.3323,
+      "step": 643
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5442455242966752e-05,
+      "loss": 0.9315,
+      "step": 644
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5434782608695653e-05,
+      "loss": 0.5539,
+      "step": 645
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.542710997442455e-05,
+      "loss": 0.3376,
+      "step": 646
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5419437340153453e-05,
+      "loss": 0.0694,
+      "step": 647
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5411764705882355e-05,
+      "loss": 0.0706,
+      "step": 648
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5404092071611253e-05,
+      "loss": 0.105,
+      "step": 649
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5396419437340155e-05,
+      "loss": 0.912,
+      "step": 650
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5388746803069053e-05,
+      "loss": 0.0998,
+      "step": 651
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5381074168797954e-05,
+      "loss": 0.2247,
+      "step": 652
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5373401534526856e-05,
+      "loss": 0.2869,
+      "step": 653
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5365728900255754e-05,
+      "loss": 0.532,
+      "step": 654
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5358056265984653e-05,
+      "loss": 0.4723,
+      "step": 655
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5350383631713558e-05,
+      "loss": 0.1178,
+      "step": 656
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5342710997442456e-05,
+      "loss": 0.0566,
+      "step": 657
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5335038363171357e-05,
+      "loss": 0.059,
+      "step": 658
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5327365728900256e-05,
+      "loss": 0.0881,
+      "step": 659
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5319693094629157e-05,
+      "loss": 0.5762,
+      "step": 660
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.531202046035806e-05,
+      "loss": 0.0499,
+      "step": 661
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5304347826086957e-05,
+      "loss": 0.087,
+      "step": 662
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5296675191815855e-05,
+      "loss": 1.0719,
+      "step": 663
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528900255754476e-05,
+      "loss": 0.6158,
+      "step": 664
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528132992327366e-05,
+      "loss": 0.6117,
+      "step": 665
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5273657289002557e-05,
+      "loss": 0.0515,
+      "step": 666
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5265984654731458e-05,
+      "loss": 0.0608,
+      "step": 667
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.525831202046036e-05,
+      "loss": 0.0313,
+      "step": 668
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.5250639386189258e-05,
+      "loss": 0.1449,
+      "step": 669
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.524296675191816e-05,
+      "loss": 0.9883,
+      "step": 670
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.523529411764706e-05,
+      "loss": 0.0859,
+      "step": 671
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.522762148337596e-05,
+      "loss": 0.0561,
+      "step": 672
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521994884910486e-05,
+      "loss": 0.0926,
+      "step": 673
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521227621483376e-05,
+      "loss": 0.1101,
+      "step": 674
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.520460358056266e-05,
+      "loss": 0.068,
+      "step": 675
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5196930946291563e-05,
+      "loss": 0.0773,
+      "step": 676
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518925831202046e-05,
+      "loss": 0.0508,
+      "step": 677
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518158567774936e-05,
+      "loss": 0.0375,
+      "step": 678
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5173913043478264e-05,
+      "loss": 0.93,
+      "step": 679
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5166240409207162e-05,
+      "loss": 0.1507,
+      "step": 680
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.515856777493606e-05,
+      "loss": 0.0917,
+      "step": 681
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5150895140664962e-05,
+      "loss": 0.97,
+      "step": 682
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5143222506393864e-05,
+      "loss": 0.5019,
+      "step": 683
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5135549872122762e-05,
+      "loss": 0.0849,
+      "step": 684
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5127877237851664e-05,
+      "loss": 1.2316,
+      "step": 685
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5120204603580562e-05,
+      "loss": 0.0252,
+      "step": 686
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5112531969309463e-05,
+      "loss": 1.0543,
+      "step": 687
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5104859335038365e-05,
+      "loss": 0.0506,
+      "step": 688
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5097186700767263e-05,
+      "loss": 0.0252,
+      "step": 689
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.508951406649616e-05,
+      "loss": 0.6297,
+      "step": 690
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5081841432225066e-05,
+      "loss": 0.4497,
+      "step": 691
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5074168797953965e-05,
+      "loss": 0.0516,
+      "step": 692
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5066496163682863e-05,
+      "loss": 0.5501,
+      "step": 693
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5058823529411768e-05,
+      "loss": 0.5621,
+      "step": 694
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5051150895140666e-05,
+      "loss": 0.0577,
+      "step": 695
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5043478260869564e-05,
+      "loss": 0.0271,
+      "step": 696
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5035805626598466e-05,
+      "loss": 0.9496,
+      "step": 697
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5028132992327368e-05,
+      "loss": 0.0877,
+      "step": 698
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5020460358056266e-05,
+      "loss": 0.0523,
+      "step": 699
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5012787723785167e-05,
+      "loss": 0.0577,
+      "step": 700
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5005115089514066e-05,
+      "loss": 0.0522,
+      "step": 701
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.4997442455242967e-05,
+      "loss": 0.0517,
+      "step": 702
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.498976982097187e-05,
+      "loss": 0.6335,
+      "step": 703
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4982097186700767e-05,
+      "loss": 0.05,
+      "step": 704
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4974424552429665e-05,
+      "loss": 0.0753,
+      "step": 705
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.496675191815857e-05,
+      "loss": 0.0227,
+      "step": 706
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.495907928388747e-05,
+      "loss": 0.0755,
+      "step": 707
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.4951406649616367e-05,
+      "loss": 0.0215,
+      "step": 708
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.494373401534527e-05,
+      "loss": 0.0451,
+      "step": 709
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.493606138107417e-05,
+      "loss": 0.0343,
+      "step": 710
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492838874680307e-05,
+      "loss": 0.0836,
+      "step": 711
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492071611253197e-05,
+      "loss": 0.5287,
+      "step": 712
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4913043478260868e-05,
+      "loss": 0.0686,
+      "step": 713
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4905370843989773e-05,
+      "loss": 0.0211,
+      "step": 714
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489769820971867e-05,
+      "loss": 0.5932,
+      "step": 715
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489002557544757e-05,
+      "loss": 0.0371,
+      "step": 716
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.488235294117647e-05,
+      "loss": 1.0137,
+      "step": 717
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.4874680306905373e-05,
+      "loss": 0.9928,
+      "step": 718
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.486700767263427e-05,
+      "loss": 0.7431,
+      "step": 719
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4859335038363173e-05,
+      "loss": 0.0207,
+      "step": 720
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4851662404092074e-05,
+      "loss": 0.0681,
+      "step": 721
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4843989769820972e-05,
+      "loss": 0.0306,
+      "step": 722
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4836317135549874e-05,
+      "loss": 0.0204,
+      "step": 723
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4828644501278772e-05,
+      "loss": 0.0344,
+      "step": 724
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4820971867007674e-05,
+      "loss": 0.0345,
+      "step": 725
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4813299232736575e-05,
+      "loss": 0.9275,
+      "step": 726
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4805626598465474e-05,
+      "loss": 0.0196,
+      "step": 727
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4797953964194372e-05,
+      "loss": 1.0137,
+      "step": 728
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4790281329923277e-05,
+      "loss": 0.0318,
+      "step": 729
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4782608695652175e-05,
+      "loss": 0.0688,
+      "step": 730
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4774936061381073e-05,
+      "loss": 0.0314,
+      "step": 731
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4767263427109975e-05,
+      "loss": 0.7135,
+      "step": 732
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4759590792838877e-05,
+      "loss": 0.0299,
+      "step": 733
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4751918158567775e-05,
+      "loss": 1.0127,
+      "step": 734
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4744245524296676e-05,
+      "loss": 0.0304,
+      "step": 735
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4736572890025575e-05,
+      "loss": 1.4618,
+      "step": 736
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4728900255754476e-05,
+      "loss": 0.7167,
+      "step": 737
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4721227621483378e-05,
+      "loss": 0.0307,
+      "step": 738
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4713554987212276e-05,
+      "loss": 0.052,
+      "step": 739
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4705882352941174e-05,
+      "loss": 0.778,
+      "step": 740
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.469820971867008e-05,
+      "loss": 0.5482,
+      "step": 741
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4690537084398978e-05,
+      "loss": 1.4314,
+      "step": 742
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4682864450127876e-05,
+      "loss": 0.0422,
+      "step": 743
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.467519181585678e-05,
+      "loss": 0.6614,
+      "step": 744
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.466751918158568e-05,
+      "loss": 0.0312,
+      "step": 745
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4659846547314577e-05,
+      "loss": 0.043,
+      "step": 746
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.465217391304348e-05,
+      "loss": 0.0199,
+      "step": 747
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.464450127877238e-05,
+      "loss": 0.0832,
+      "step": 748
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.463682864450128e-05,
+      "loss": 0.0312,
+      "step": 749
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.462915601023018e-05,
+      "loss": 0.0309,
+      "step": 750
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.462148337595908e-05,
+      "loss": 0.0193,
+      "step": 751
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.461381074168798e-05,
+      "loss": 0.0446,
+      "step": 752
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.4606138107416882e-05,
+      "loss": 0.9993,
+      "step": 753
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.459846547314578e-05,
+      "loss": 0.016,
+      "step": 754
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4590792838874678e-05,
+      "loss": 0.0433,
+      "step": 755
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4583120204603583e-05,
+      "loss": 0.2614,
+      "step": 756
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.457544757033248e-05,
+      "loss": 0.018,
+      "step": 757
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.456777493606138e-05,
+      "loss": 0.6676,
+      "step": 758
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.456010230179028e-05,
+      "loss": 0.0599,
+      "step": 759
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4552429667519183e-05,
+      "loss": 0.0338,
+      "step": 760
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.454475703324808e-05,
+      "loss": 0.0268,
+      "step": 761
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4537084398976983e-05,
+      "loss": 0.0262,
+      "step": 762
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.452941176470588e-05,
+      "loss": 0.0252,
+      "step": 763
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4521739130434786e-05,
+      "loss": 0.0377,
+      "step": 764
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4514066496163684e-05,
+      "loss": 1.4866,
+      "step": 765
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4506393861892582e-05,
+      "loss": 1.0524,
+      "step": 766
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4498721227621484e-05,
+      "loss": 0.0328,
+      "step": 767
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4491048593350386e-05,
+      "loss": 0.024,
+      "step": 768
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4483375959079284e-05,
+      "loss": 0.022,
+      "step": 769
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4475703324808185e-05,
+      "loss": 0.0245,
+      "step": 770
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4468030690537087e-05,
+      "loss": 1.0107,
+      "step": 771
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4460358056265985e-05,
+      "loss": 0.0336,
+      "step": 772
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4452685421994887e-05,
+      "loss": 0.0391,
+      "step": 773
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4445012787723785e-05,
+      "loss": 0.0303,
+      "step": 774
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4437340153452687e-05,
+      "loss": 0.0443,
+      "step": 775
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4429667519181588e-05,
+      "loss": 0.0292,
+      "step": 776
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4421994884910487e-05,
+      "loss": 0.0284,
+      "step": 777
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4414322250639385e-05,
+      "loss": 1.5985,
+      "step": 778
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.440664961636829e-05,
+      "loss": 0.0114,
+      "step": 779
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4398976982097188e-05,
+      "loss": 0.0342,
+      "step": 780
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4391304347826086e-05,
+      "loss": 1.8501,
+      "step": 781
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4383631713554988e-05,
+      "loss": 0.5375,
+      "step": 782
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.437595907928389e-05,
+      "loss": 1.589,
+      "step": 783
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4368286445012788e-05,
+      "loss": 0.0221,
+      "step": 784
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.436061381074169e-05,
+      "loss": 0.0453,
+      "step": 785
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4352941176470587e-05,
+      "loss": 0.796,
+      "step": 786
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.434526854219949e-05,
+      "loss": 0.1249,
+      "step": 787
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.433759590792839e-05,
+      "loss": 0.7963,
+      "step": 788
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.432992327365729e-05,
+      "loss": 0.4231,
+      "step": 789
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.4322250639386187e-05,
+      "loss": 0.0247,
+      "step": 790
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.4314578005115092e-05,
+      "loss": 0.0295,
+      "step": 791
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.430690537084399e-05,
+      "loss": 0.0249,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.3537317216396332,
+      "eval_runtime": 58.1226,
+      "eval_samples_per_second": 6.813,
+      "eval_steps_per_second": 1.703,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.429923273657289e-05,
+      "loss": 0.0157,
+      "step": 793
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4291560102301794e-05,
+      "loss": 0.5222,
+      "step": 794
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4283887468030692e-05,
+      "loss": 0.0226,
+      "step": 795
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.427621483375959e-05,
+      "loss": 0.0293,
+      "step": 796
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.426854219948849e-05,
+      "loss": 0.015,
+      "step": 797
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4260869565217393e-05,
+      "loss": 1.0658,
+      "step": 798
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.425319693094629e-05,
+      "loss": 0.0412,
+      "step": 799
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4245524296675193e-05,
+      "loss": 0.0219,
+      "step": 800
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.423785166240409e-05,
+      "loss": 0.0149,
+      "step": 801
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4230179028132993e-05,
+      "loss": 0.0323,
+      "step": 802
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4222506393861895e-05,
+      "loss": 1.0764,
+      "step": 803
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4214833759590793e-05,
+      "loss": 0.7518,
+      "step": 804
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.420716112531969e-05,
+      "loss": 0.8166,
+      "step": 805
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4199488491048596e-05,
+      "loss": 0.0219,
+      "step": 806
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4191815856777494e-05,
+      "loss": 0.0154,
+      "step": 807
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4184143222506392e-05,
+      "loss": 0.0163,
+      "step": 808
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4176470588235294e-05,
+      "loss": 1.0562,
+      "step": 809
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4168797953964196e-05,
+      "loss": 0.0218,
+      "step": 810
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4161125319693094e-05,
+      "loss": 0.0232,
+      "step": 811
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4153452685421995e-05,
+      "loss": 0.2384,
+      "step": 812
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4145780051150894e-05,
+      "loss": 1.5286,
+      "step": 813
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4138107416879795e-05,
+      "loss": 0.0262,
+      "step": 814
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4130434782608697e-05,
+      "loss": 0.0334,
+      "step": 815
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4122762148337595e-05,
+      "loss": 0.6509,
+      "step": 816
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4115089514066497e-05,
+      "loss": 0.0159,
+      "step": 817
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.41074168797954e-05,
+      "loss": 0.6225,
+      "step": 818
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4099744245524297e-05,
+      "loss": 0.752,
+      "step": 819
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4092071611253198e-05,
+      "loss": 1.0594,
+      "step": 820
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.40843989769821e-05,
+      "loss": 0.0161,
+      "step": 821
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4076726342710998e-05,
+      "loss": 0.7492,
+      "step": 822
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40690537084399e-05,
+      "loss": 0.0302,
+      "step": 823
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4061381074168798e-05,
+      "loss": 0.038,
+      "step": 824
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40537084398977e-05,
+      "loss": 0.0243,
+      "step": 825
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40460358056266e-05,
+      "loss": 0.0331,
+      "step": 826
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40383631713555e-05,
+      "loss": 1.0252,
+      "step": 827
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4030690537084398e-05,
+      "loss": 0.0172,
+      "step": 828
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4023017902813303e-05,
+      "loss": 0.6168,
+      "step": 829
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40153452685422e-05,
+      "loss": 0.3135,
+      "step": 830
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40076726342711e-05,
+      "loss": 0.0168,
+      "step": 831
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.0172,
+      "step": 832
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.3992327365728902e-05,
+      "loss": 1.0349,
+      "step": 833
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39846547314578e-05,
+      "loss": 0.0978,
+      "step": 834
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3976982097186702e-05,
+      "loss": 0.0167,
+      "step": 835
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39693094629156e-05,
+      "loss": 1.0381,
+      "step": 836
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3961636828644502e-05,
+      "loss": 0.6857,
+      "step": 837
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3953964194373404e-05,
+      "loss": 1.0406,
+      "step": 838
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3946291560102302e-05,
+      "loss": 0.071,
+      "step": 839
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.39386189258312e-05,
+      "loss": 0.0193,
+      "step": 840
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3930946291560105e-05,
+      "loss": 0.0193,
+      "step": 841
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3923273657289003e-05,
+      "loss": 0.9916,
+      "step": 842
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.39156010230179e-05,
+      "loss": 0.0723,
+      "step": 843
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3907928388746806e-05,
+      "loss": 0.0626,
+      "step": 844
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3900255754475705e-05,
+      "loss": 0.7264,
+      "step": 845
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3892583120204603e-05,
+      "loss": 0.827,
+      "step": 846
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3884910485933504e-05,
+      "loss": 1.2193,
+      "step": 847
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3877237851662406e-05,
+      "loss": 0.6754,
+      "step": 848
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3869565217391304e-05,
+      "loss": 0.6856,
+      "step": 849
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3861892583120206e-05,
+      "loss": 0.7441,
+      "step": 850
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3854219948849104e-05,
+      "loss": 0.023,
+      "step": 851
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3846547314578006e-05,
+      "loss": 0.4742,
+      "step": 852
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3838874680306907e-05,
+      "loss": 0.0363,
+      "step": 853
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3831202046035806e-05,
+      "loss": 0.0409,
+      "step": 854
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3823529411764704e-05,
+      "loss": 0.8353,
+      "step": 855
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.381585677749361e-05,
+      "loss": 0.0333,
+      "step": 856
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3808184143222507e-05,
+      "loss": 0.0216,
+      "step": 857
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3800511508951405e-05,
+      "loss": 0.9688,
+      "step": 858
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3792838874680307e-05,
+      "loss": 1.0078,
+      "step": 859
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.378516624040921e-05,
+      "loss": 0.0183,
+      "step": 860
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3777493606138107e-05,
+      "loss": 0.0183,
+      "step": 861
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3769820971867008e-05,
+      "loss": 0.9786,
+      "step": 862
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3762148337595907e-05,
+      "loss": 0.9166,
+      "step": 863
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3754475703324808e-05,
+      "loss": 1.1408,
+      "step": 864
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.374680306905371e-05,
+      "loss": 1.1394,
+      "step": 865
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.3739130434782608e-05,
+      "loss": 0.0185,
+      "step": 866
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.373145780051151e-05,
+      "loss": 0.9979,
+      "step": 867
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.372378516624041e-05,
+      "loss": 1.0092,
+      "step": 868
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.371611253196931e-05,
+      "loss": 0.7276,
+      "step": 869
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.370843989769821e-05,
+      "loss": 0.0359,
+      "step": 870
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3700767263427113e-05,
+      "loss": 0.0864,
+      "step": 871
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.369309462915601e-05,
+      "loss": 0.0246,
+      "step": 872
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3685421994884912e-05,
+      "loss": 0.8518,
+      "step": 873
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.367774936061381e-05,
+      "loss": 1.433,
+      "step": 874
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3670076726342712e-05,
+      "loss": 0.035,
+      "step": 875
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3662404092071614e-05,
+      "loss": 0.5922,
+      "step": 876
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3654731457800512e-05,
+      "loss": 0.529,
+      "step": 877
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.364705882352941e-05,
+      "loss": 0.6423,
+      "step": 878
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3639386189258315e-05,
+      "loss": 0.0337,
+      "step": 879
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3631713554987214e-05,
+      "loss": 0.0402,
+      "step": 880
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3624040920716112e-05,
+      "loss": 0.0782,
+      "step": 881
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3616368286445013e-05,
+      "loss": 0.4141,
+      "step": 882
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3608695652173915e-05,
+      "loss": 0.0359,
+      "step": 883
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3601023017902813e-05,
+      "loss": 0.4509,
+      "step": 884
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3593350383631715e-05,
+      "loss": 0.6414,
+      "step": 885
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3585677749360613e-05,
+      "loss": 0.6873,
+      "step": 886
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3578005115089515e-05,
+      "loss": 0.0304,
+      "step": 887
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3570332480818416e-05,
+      "loss": 0.0315,
+      "step": 888
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3562659846547315e-05,
+      "loss": 0.0896,
+      "step": 889
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3554987212276213e-05,
+      "loss": 0.0201,
+      "step": 890
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3547314578005118e-05,
+      "loss": 0.027,
+      "step": 891
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3539641943734016e-05,
+      "loss": 1.9853,
+      "step": 892
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3531969309462914e-05,
+      "loss": 2.9647,
+      "step": 893
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.352429667519182e-05,
+      "loss": 1.9403,
+      "step": 894
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3516624040920717e-05,
+      "loss": 0.0193,
+      "step": 895
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3508951406649616e-05,
+      "loss": 0.9912,
+      "step": 896
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3501278772378517e-05,
+      "loss": 0.02,
+      "step": 897
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.349360613810742e-05,
+      "loss": 0.9015,
+      "step": 898
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3485933503836317e-05,
+      "loss": 0.1294,
+      "step": 899
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.347826086956522e-05,
+      "loss": 0.0559,
+      "step": 900
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.3470588235294117e-05,
+      "loss": 1.8039,
+      "step": 901
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.346291560102302e-05,
+      "loss": 0.0336,
+      "step": 902
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.345524296675192e-05,
+      "loss": 0.0383,
+      "step": 903
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.344757033248082e-05,
+      "loss": 0.0722,
+      "step": 904
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3439897698209717e-05,
+      "loss": 0.0278,
+      "step": 905
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.343222506393862e-05,
+      "loss": 0.0684,
+      "step": 906
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.342455242966752e-05,
+      "loss": 0.0236,
+      "step": 907
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3416879795396418e-05,
+      "loss": 0.0348,
+      "step": 908
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340920716112532e-05,
+      "loss": 0.0204,
+      "step": 909
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340153452685422e-05,
+      "loss": 0.8109,
+      "step": 910
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.339386189258312e-05,
+      "loss": 0.0279,
+      "step": 911
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.338618925831202e-05,
+      "loss": 0.0361,
+      "step": 912
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337851662404092e-05,
+      "loss": 0.0355,
+      "step": 913
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337084398976982e-05,
+      "loss": 0.7958,
+      "step": 914
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.3363171355498723e-05,
+      "loss": 0.0362,
+      "step": 915
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.335549872122762e-05,
+      "loss": 0.0252,
+      "step": 916
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3347826086956522e-05,
+      "loss": 0.8165,
+      "step": 917
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3340153452685424e-05,
+      "loss": 0.0237,
+      "step": 918
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3332480818414322e-05,
+      "loss": 0.8162,
+      "step": 919
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.332480818414322e-05,
+      "loss": 0.7786,
+      "step": 920
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3317135549872125e-05,
+      "loss": 0.0787,
+      "step": 921
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3309462915601024e-05,
+      "loss": 0.1764,
+      "step": 922
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3301790281329925e-05,
+      "loss": 0.7449,
+      "step": 923
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3294117647058824e-05,
+      "loss": 0.0452,
+      "step": 924
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3286445012787725e-05,
+      "loss": 0.0181,
+      "step": 925
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3278772378516627e-05,
+      "loss": 0.1783,
+      "step": 926
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3271099744245525e-05,
+      "loss": 0.5158,
+      "step": 927
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3263427109974423e-05,
+      "loss": 0.7574,
+      "step": 928
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3255754475703328e-05,
+      "loss": 2.6099,
+      "step": 929
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3248081841432226e-05,
+      "loss": 0.0304,
+      "step": 930
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3240409207161125e-05,
+      "loss": 0.0795,
+      "step": 931
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3232736572890026e-05,
+      "loss": 0.0342,
+      "step": 932
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3225063938618928e-05,
+      "loss": 0.0473,
+      "step": 933
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3217391304347826e-05,
+      "loss": 0.0831,
+      "step": 934
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3209718670076728e-05,
+      "loss": 0.0465,
+      "step": 935
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3202046035805626e-05,
+      "loss": 0.6769,
+      "step": 936
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3194373401534528e-05,
+      "loss": 0.0173,
+      "step": 937
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.318670076726343e-05,
+      "loss": 0.0169,
+      "step": 938
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3179028132992327e-05,
+      "loss": 0.0469,
+      "step": 939
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3171355498721226e-05,
+      "loss": 0.048,
+      "step": 940
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.316368286445013e-05,
+      "loss": 1.0704,
+      "step": 941
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.315601023017903e-05,
+      "loss": 1.0738,
+      "step": 942
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3148337595907927e-05,
+      "loss": 0.0163,
+      "step": 943
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3140664961636832e-05,
+      "loss": 2.0702,
+      "step": 944
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.313299232736573e-05,
+      "loss": 1.0433,
+      "step": 945
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.312531969309463e-05,
+      "loss": 1.0351,
+      "step": 946
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.311764705882353e-05,
+      "loss": 1.053,
+      "step": 947
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.3109974424552432e-05,
+      "loss": 0.1797,
+      "step": 948
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.310230179028133e-05,
+      "loss": 1.0259,
+      "step": 949
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.309462915601023e-05,
+      "loss": 0.03,
+      "step": 950
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.308695652173913e-05,
+      "loss": 0.905,
+      "step": 951
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.307928388746803e-05,
+      "loss": 0.0699,
+      "step": 952
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3071611253196933e-05,
+      "loss": 1.0013,
+      "step": 953
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.306393861892583e-05,
+      "loss": 0.0319,
+      "step": 954
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.305626598465473e-05,
+      "loss": 0.0361,
+      "step": 955
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3048593350383634e-05,
+      "loss": 0.0201,
+      "step": 956
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3040920716112533e-05,
+      "loss": 0.033,
+      "step": 957
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.303324808184143e-05,
+      "loss": 0.0411,
+      "step": 958
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3025575447570332e-05,
+      "loss": 0.7138,
+      "step": 959
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3017902813299234e-05,
+      "loss": 0.0434,
+      "step": 960
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3010230179028132e-05,
+      "loss": 0.8445,
+      "step": 961
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3002557544757034e-05,
+      "loss": 0.0323,
+      "step": 962
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2994884910485932e-05,
+      "loss": 0.0203,
+      "step": 963
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2987212276214834e-05,
+      "loss": 0.2302,
+      "step": 964
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2979539641943735e-05,
+      "loss": 0.0293,
+      "step": 965
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2971867007672634e-05,
+      "loss": 0.0593,
+      "step": 966
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2964194373401535e-05,
+      "loss": 0.6505,
+      "step": 967
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2956521739130437e-05,
+      "loss": 0.5891,
+      "step": 968
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2948849104859335e-05,
+      "loss": 0.0191,
+      "step": 969
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2941176470588233e-05,
+      "loss": 0.0188,
+      "step": 970
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2933503836317138e-05,
+      "loss": 0.9537,
+      "step": 971
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2925831202046037e-05,
+      "loss": 0.0186,
+      "step": 972
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2918158567774935e-05,
+      "loss": 1.0237,
+      "step": 973
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2910485933503836e-05,
+      "loss": 0.0417,
+      "step": 974
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2902813299232738e-05,
+      "loss": 0.0369,
+      "step": 975
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.289514066496164e-05,
+      "loss": 0.0184,
+      "step": 976
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2887468030690538e-05,
+      "loss": 0.811,
+      "step": 977
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2879795396419436e-05,
+      "loss": 0.0183,
+      "step": 978
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.287212276214834e-05,
+      "loss": 1.0442,
+      "step": 979
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.286445012787724e-05,
+      "loss": 0.027,
+      "step": 980
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.2856777493606137e-05,
+      "loss": 0.062,
+      "step": 981
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284910485933504e-05,
+      "loss": 0.0186,
+      "step": 982
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284143222506394e-05,
+      "loss": 0.6057,
+      "step": 983
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.283375959079284e-05,
+      "loss": 1.6594,
+      "step": 984
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.282608695652174e-05,
+      "loss": 0.0192,
+      "step": 985
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281841432225064e-05,
+      "loss": 1.5361,
+      "step": 986
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281074168797954e-05,
+      "loss": 0.0384,
+      "step": 987
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.2803069053708442e-05,
+      "loss": 2.0238,
+      "step": 988
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.279539641943734e-05,
+      "loss": 0.0334,
+      "step": 989
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2787723785166242e-05,
+      "loss": 0.0571,
+      "step": 990
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2780051150895143e-05,
+      "loss": 0.0211,
+      "step": 991
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.277237851662404e-05,
+      "loss": 0.0409,
+      "step": 992
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.276470588235294e-05,
+      "loss": 0.1059,
+      "step": 993
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2757033248081845e-05,
+      "loss": 0.0277,
+      "step": 994
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2749360613810743e-05,
+      "loss": 1.0023,
+      "step": 995
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.274168797953964e-05,
+      "loss": 0.0186,
+      "step": 996
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2734015345268543e-05,
+      "loss": 0.0345,
+      "step": 997
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2726342710997445e-05,
+      "loss": 0.0169,
+      "step": 998
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2718670076726343e-05,
+      "loss": 0.7292,
+      "step": 999
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2710997442455244e-05,
+      "loss": 1.2655,
+      "step": 1000
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2703324808184143e-05,
+      "loss": 0.03,
+      "step": 1001
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2695652173913044e-05,
+      "loss": 1.0266,
+      "step": 1002
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2687979539641946e-05,
+      "loss": 0.7645,
+      "step": 1003
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2680306905370844e-05,
+      "loss": 0.7985,
+      "step": 1004
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2672634271099742e-05,
+      "loss": 0.76,
+      "step": 1005
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2664961636828647e-05,
+      "loss": 0.7577,
+      "step": 1006
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2657289002557545e-05,
+      "loss": 0.0248,
+      "step": 1007
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2649616368286444e-05,
+      "loss": 0.0262,
+      "step": 1008
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2641943734015345e-05,
+      "loss": 1.9984,
+      "step": 1009
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2634271099744247e-05,
+      "loss": 0.1489,
+      "step": 1010
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2626598465473145e-05,
+      "loss": 0.0683,
+      "step": 1011
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2618925831202047e-05,
+      "loss": 0.054,
+      "step": 1012
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2611253196930945e-05,
+      "loss": 0.0269,
+      "step": 1013
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2603580562659847e-05,
+      "loss": 1.0378,
+      "step": 1014
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2595907928388748e-05,
+      "loss": 0.0171,
+      "step": 1015
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2588235294117646e-05,
+      "loss": 0.8433,
+      "step": 1016
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2580562659846548e-05,
+      "loss": 0.0468,
+      "step": 1017
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.257289002557545e-05,
+      "loss": 0.0331,
+      "step": 1018
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2565217391304348e-05,
+      "loss": 0.0308,
+      "step": 1019
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2557544757033246e-05,
+      "loss": 1.0007,
+      "step": 1020
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254987212276215e-05,
+      "loss": 2.0242,
+      "step": 1021
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254219948849105e-05,
+      "loss": 0.7713,
+      "step": 1022
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2534526854219948e-05,
+      "loss": 1.1112,
+      "step": 1023
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.252685421994885e-05,
+      "loss": 0.0184,
+      "step": 1024
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251918158567775e-05,
+      "loss": 0.0243,
+      "step": 1025
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251150895140665e-05,
+      "loss": 0.046,
+      "step": 1026
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.250383631713555e-05,
+      "loss": 0.0378,
+      "step": 1027
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.249616368286445e-05,
+      "loss": 0.0329,
+      "step": 1028
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2488491048593354e-05,
+      "loss": 0.0245,
+      "step": 1029
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2480818414322252e-05,
+      "loss": 0.0238,
+      "step": 1030
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.247314578005115e-05,
+      "loss": 1.0628,
+      "step": 1031
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2465473145780052e-05,
+      "loss": 0.0703,
+      "step": 1032
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2457800511508954e-05,
+      "loss": 0.0251,
+      "step": 1033
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2450127877237852e-05,
+      "loss": 1.0561,
+      "step": 1034
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2442455242966753e-05,
+      "loss": 0.0274,
+      "step": 1035
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.243478260869565e-05,
+      "loss": 0.023,
+      "step": 1036
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2427109974424553e-05,
+      "loss": 0.0223,
+      "step": 1037
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2419437340153455e-05,
+      "loss": 0.8113,
+      "step": 1038
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2411764705882353e-05,
+      "loss": 0.0346,
+      "step": 1039
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2404092071611255e-05,
+      "loss": 0.0164,
+      "step": 1040
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2396419437340156e-05,
+      "loss": 0.0237,
+      "step": 1041
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2388746803069054e-05,
+      "loss": 0.0305,
+      "step": 1042
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2381074168797953e-05,
+      "loss": 0.0158,
+      "step": 1043
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2373401534526858e-05,
+      "loss": 0.0225,
+      "step": 1044
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2365728900255756e-05,
+      "loss": 0.0226,
+      "step": 1045
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2358056265984654e-05,
+      "loss": 0.015,
+      "step": 1046
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2350383631713556e-05,
+      "loss": 0.0219,
+      "step": 1047
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2342710997442457e-05,
+      "loss": 1.0694,
+      "step": 1048
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2335038363171356e-05,
+      "loss": 0.0348,
+      "step": 1049
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2327365728900257e-05,
+      "loss": 0.0213,
+      "step": 1050
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2319693094629155e-05,
+      "loss": 0.9405,
+      "step": 1051
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2312020460358057e-05,
+      "loss": 0.0252,
+      "step": 1052
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.230434782608696e-05,
+      "loss": 0.0254,
+      "step": 1053
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2296675191815857e-05,
+      "loss": 0.7353,
+      "step": 1054
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2289002557544755e-05,
+      "loss": 0.0253,
+      "step": 1055
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.228132992327366e-05,
+      "loss": 0.8361,
+      "step": 1056
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2273657289002558e-05,
+      "loss": 0.0196,
+      "step": 1057
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2265984654731457e-05,
+      "loss": 0.014,
+      "step": 1058
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2258312020460358e-05,
+      "loss": 0.8297,
+      "step": 1059
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.225063938618926e-05,
+      "loss": 1.0654,
+      "step": 1060
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2242966751918158e-05,
+      "loss": 0.0248,
+      "step": 1061
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.223529411764706e-05,
+      "loss": 0.0226,
+      "step": 1062
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2227621483375958e-05,
+      "loss": 0.0238,
+      "step": 1063
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221994884910486e-05,
+      "loss": 0.0212,
+      "step": 1064
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221227621483376e-05,
+      "loss": 0.0274,
+      "step": 1065
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.220460358056266e-05,
+      "loss": 1.0217,
+      "step": 1066
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.219693094629156e-05,
+      "loss": 0.0252,
+      "step": 1067
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2189258312020462e-05,
+      "loss": 0.0219,
+      "step": 1068
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.218158567774936e-05,
+      "loss": 0.0349,
+      "step": 1069
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.217391304347826e-05,
+      "loss": 0.0152,
+      "step": 1070
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2166240409207164e-05,
+      "loss": 0.021,
+      "step": 1071
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2158567774936062e-05,
+      "loss": 0.015,
+      "step": 1072
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.215089514066496e-05,
+      "loss": 0.0277,
+      "step": 1073
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2143222506393862e-05,
+      "loss": 0.1858,
+      "step": 1074
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2135549872122764e-05,
+      "loss": 0.0136,
+      "step": 1075
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2127877237851662e-05,
+      "loss": 0.0132,
+      "step": 1076
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2120204603580563e-05,
+      "loss": 0.0126,
+      "step": 1077
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.211253196930946e-05,
+      "loss": 0.5171,
+      "step": 1078
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2104859335038363e-05,
+      "loss": 0.0422,
+      "step": 1079
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2097186700767265e-05,
+      "loss": 0.0186,
+      "step": 1080
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2089514066496163e-05,
+      "loss": 0.8412,
+      "step": 1081
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2081841432225065e-05,
+      "loss": 0.0183,
+      "step": 1082
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2074168797953966e-05,
+      "loss": 1.0243,
+      "step": 1083
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2066496163682865e-05,
+      "loss": 0.0212,
+      "step": 1084
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2058823529411766e-05,
+      "loss": 0.2841,
+      "step": 1085
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2051150895140664e-05,
+      "loss": 0.0118,
+      "step": 1086
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2043478260869566e-05,
+      "loss": 0.0247,
+      "step": 1087
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2035805626598468e-05,
+      "loss": 1.5121,
+      "step": 1088
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2028132992327366e-05,
+      "loss": 0.0304,
+      "step": 1089
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2020460358056267e-05,
+      "loss": 0.0329,
+      "step": 1090
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.201278772378517e-05,
+      "loss": 1.1288,
+      "step": 1091
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.2005115089514067e-05,
+      "loss": 0.0145,
+      "step": 1092
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.1997442455242966e-05,
+      "loss": 0.0498,
+      "step": 1093
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.198976982097187e-05,
+      "loss": 0.0244,
+      "step": 1094
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.198209718670077e-05,
+      "loss": 0.0112,
+      "step": 1095
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.1974424552429667e-05,
+      "loss": 0.8642,
+      "step": 1096
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.196675191815857e-05,
+      "loss": 0.8822,
+      "step": 1097
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.195907928388747e-05,
+      "loss": 0.0278,
+      "step": 1098
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.195140664961637e-05,
+      "loss": 0.6954,
+      "step": 1099
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.194373401534527e-05,
+      "loss": 0.0179,
+      "step": 1100
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.1936061381074168e-05,
+      "loss": 0.5607,
+      "step": 1101
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.192838874680307e-05,
+      "loss": 0.0191,
+      "step": 1102
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.192071611253197e-05,
+      "loss": 1.9519,
+      "step": 1103
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.191304347826087e-05,
+      "loss": 0.8347,
+      "step": 1104
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1905370843989768e-05,
+      "loss": 0.0262,
+      "step": 1105
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1897698209718673e-05,
+      "loss": 0.0477,
+      "step": 1106
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.189002557544757e-05,
+      "loss": 0.0227,
+      "step": 1107
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.188235294117647e-05,
+      "loss": 0.0205,
+      "step": 1108
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.187468030690537e-05,
+      "loss": 0.0208,
+      "step": 1109
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.1867007672634273e-05,
+      "loss": 0.0116,
+      "step": 1110
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.185933503836317e-05,
+      "loss": 1.1546,
+      "step": 1111
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1851662404092072e-05,
+      "loss": 0.0114,
+      "step": 1112
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.184398976982097e-05,
+      "loss": 0.0113,
+      "step": 1113
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1836317135549872e-05,
+      "loss": 0.6881,
+      "step": 1114
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1828644501278774e-05,
+      "loss": 0.0294,
+      "step": 1115
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1820971867007672e-05,
+      "loss": 0.1569,
+      "step": 1116
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1813299232736574e-05,
+      "loss": 1.1362,
+      "step": 1117
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1805626598465475e-05,
+      "loss": 0.0196,
+      "step": 1118
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1797953964194374e-05,
+      "loss": 0.0205,
+      "step": 1119
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1790281329923272e-05,
+      "loss": 0.0198,
+      "step": 1120
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1782608695652177e-05,
+      "loss": 0.0113,
+      "step": 1121
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1774936061381075e-05,
+      "loss": 0.02,
+      "step": 1122
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1767263427109973e-05,
+      "loss": 0.0195,
+      "step": 1123
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1759590792838875e-05,
+      "loss": 0.8016,
+      "step": 1124
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1751918158567776e-05,
+      "loss": 0.0122,
+      "step": 1125
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1744245524296675e-05,
+      "loss": 1.1132,
+      "step": 1126
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1736572890025576e-05,
+      "loss": 0.011,
+      "step": 1127
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1728900255754474e-05,
+      "loss": 0.7705,
+      "step": 1128
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1721227621483376e-05,
+      "loss": 1.1467,
+      "step": 1129
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1713554987212278e-05,
+      "loss": 0.0176,
+      "step": 1130
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1705882352941176e-05,
+      "loss": 0.0293,
+      "step": 1131
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1698209718670074e-05,
+      "loss": 0.0208,
+      "step": 1132
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.169053708439898e-05,
+      "loss": 0.0199,
+      "step": 1133
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1682864450127877e-05,
+      "loss": 0.0431,
+      "step": 1134
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.167519181585678e-05,
+      "loss": 1.6066,
+      "step": 1135
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.1667519181585677e-05,
+      "loss": 0.029,
+      "step": 1136
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165984654731458e-05,
+      "loss": 0.029,
+      "step": 1137
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165217391304348e-05,
+      "loss": 1.1749,
+      "step": 1138
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.164450127877238e-05,
+      "loss": 0.0189,
+      "step": 1139
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.163682864450128e-05,
+      "loss": 0.0309,
+      "step": 1140
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.1629156010230182e-05,
+      "loss": 0.0287,
+      "step": 1141
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.162148337595908e-05,
+      "loss": 0.0202,
+      "step": 1142
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.161381074168798e-05,
+      "loss": 0.7889,
+      "step": 1143
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.1606138107416883e-05,
+      "loss": 0.0285,
+      "step": 1144
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159846547314578e-05,
+      "loss": 0.0535,
+      "step": 1145
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159079283887468e-05,
+      "loss": 1.0829,
+      "step": 1146
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.158312020460358e-05,
+      "loss": 0.0989,
+      "step": 1147
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1575447570332483e-05,
+      "loss": 0.0115,
+      "step": 1148
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.156777493606138e-05,
+      "loss": 0.0227,
+      "step": 1149
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1560102301790283e-05,
+      "loss": 0.5506,
+      "step": 1150
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.155242966751918e-05,
+      "loss": 1.1234,
+      "step": 1151
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1544757033248083e-05,
+      "loss": 0.0564,
+      "step": 1152
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1537084398976984e-05,
+      "loss": 0.0127,
+      "step": 1153
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1529411764705882e-05,
+      "loss": 0.0218,
+      "step": 1154
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.152173913043478e-05,
+      "loss": 0.0124,
+      "step": 1155
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1514066496163686e-05,
+      "loss": 0.0223,
+      "step": 1156
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1506393861892584e-05,
+      "loss": 0.0505,
+      "step": 1157
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1498721227621482e-05,
+      "loss": 0.0225,
+      "step": 1158
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1491048593350384e-05,
+      "loss": 0.0308,
+      "step": 1159
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1483375959079285e-05,
+      "loss": 0.036,
+      "step": 1160
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1475703324808184e-05,
+      "loss": 0.0121,
+      "step": 1161
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1468030690537085e-05,
+      "loss": 0.0274,
+      "step": 1162
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1460358056265983e-05,
+      "loss": 1.0397,
+      "step": 1163
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1452685421994885e-05,
+      "loss": 0.0332,
+      "step": 1164
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1445012787723787e-05,
+      "loss": 0.0182,
+      "step": 1165
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1437340153452685e-05,
+      "loss": 0.018,
+      "step": 1166
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1429667519181587e-05,
+      "loss": 0.859,
+      "step": 1167
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1421994884910488e-05,
+      "loss": 0.0321,
+      "step": 1168
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1414322250639386e-05,
+      "loss": 0.0225,
+      "step": 1169
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1406649616368285e-05,
+      "loss": 0.0224,
+      "step": 1170
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.139897698209719e-05,
+      "loss": 0.03,
+      "step": 1171
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1391304347826088e-05,
+      "loss": 0.7473,
+      "step": 1172
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1383631713554986e-05,
+      "loss": 0.8609,
+      "step": 1173
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1375959079283888e-05,
+      "loss": 0.0105,
+      "step": 1174
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.136828644501279e-05,
+      "loss": 0.0217,
+      "step": 1175
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1360613810741687e-05,
+      "loss": 0.0218,
+      "step": 1176
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.135294117647059e-05,
+      "loss": 0.016,
+      "step": 1177
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1345268542199487e-05,
+      "loss": 0.7844,
+      "step": 1178
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.133759590792839e-05,
+      "loss": 0.01,
+      "step": 1179
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132992327365729e-05,
+      "loss": 0.8774,
+      "step": 1180
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132225063938619e-05,
+      "loss": 0.8375,
+      "step": 1181
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.1314578005115087e-05,
+      "loss": 0.0188,
+      "step": 1182
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.1306905370843992e-05,
+      "loss": 1.1026,
+      "step": 1183
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129923273657289e-05,
+      "loss": 1.0447,
+      "step": 1184
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129156010230179e-05,
+      "loss": 0.0101,
+      "step": 1185
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.128388746803069e-05,
+      "loss": 0.0247,
+      "step": 1186
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.127621483375959e-05,
+      "loss": 0.0295,
+      "step": 1187
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.1268542199488493e-05,
+      "loss": 0.0191,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.37390685081481934,
+      "eval_runtime": 57.4087,
+      "eval_samples_per_second": 6.898,
+      "eval_steps_per_second": 1.724,
+      "step": 1188
+    }
+  ],
+  "max_steps": 3960,
+  "num_train_epochs": 10,
+  "total_flos": 4.311442640736e+17,
+  "trial_name": null,
+  "trial_params": null
+}
diff --git a/checkpoint-1188/training_args.bin b/checkpoint-1188/training_args.bin
new file mode 100644
index 0000000000000000000000000000000000000000..32901f899c91066c808eae9dd019400bfdd17498
--- /dev/null
+++ b/checkpoint-1188/training_args.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db14439c369bfe9c89f64ff38f0b2d3fcc53911fc2c0b12a3a7d04656e7384a
+size 3503
diff --git a/checkpoint-1584/config.json b/checkpoint-1584/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..cb8d52e14263c50fd53dca42b5050e3235115c32
--- /dev/null
+++ b/checkpoint-1584/config.json
@@ -0,0 +1,113 @@
+{
+  "_name_or_path": "facebook/wav2vec2-base-960h",
+  "activation_dropout": 0.1,
+  "adapter_kernel_size": 3,
+  "adapter_stride": 2,
+  "add_adapter": false,
+  "apply_spec_augment": true,
+  "architectures": [
+    "Wav2Vec2ForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 1,
+  "classifier_proj_size": 256,
+  "codevector_dim": 256,
+  "contrastive_logits_temperature": 0.1,
+  "conv_bias": false,
+  "conv_dim": [
+    512,
+    512,
+    512,
+    512,
+    512,
+    512,
+    512
+  ],
+  "conv_kernel": [
+    10,
+    3,
+    3,
+    3,
+    3,
+    2,
+    2
+  ],
+  "conv_stride": [
+    5,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2
+  ],
+  "ctc_loss_reduction": "sum",
+  "ctc_zero_infinity": false,
+  "diversity_loss_weight": 0.1,
+  "do_stable_layer_norm": false,
+  "eos_token_id": 2,
+  "feat_extract_activation": "gelu",
+  "feat_extract_dropout": 0.0,
+  "feat_extract_norm": "group",
+  "feat_proj_dropout": 0.1,
+  "feat_quantizer_dropout": 0.0,
+  "final_dropout": 0.1,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout": 0.1,
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Foreign",
+    "1": "Native"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "layerdrop": 0.1,
+  "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
+  "mask_feature_prob": 0.0,
+  "mask_time_length": 10,
+  "mask_time_min_masks": 2,
+  "mask_time_prob": 0.05,
+  "model_type": "wav2vec2",
+  "num_adapter_layers": 3,
+  "num_attention_heads": 12,
+  "num_codevector_groups": 2,
+  "num_codevectors_per_group": 320,
+  "num_conv_pos_embedding_groups": 16,
+  "num_conv_pos_embeddings": 128,
+  "num_feat_extract_layers": 7,
+  "num_hidden_layers": 12,
+  "num_negatives": 100,
+  "output_hidden_size": 768,
+  "pad_token_id": 0,
+  "proj_codevector_dim": 256,
+  "tdnn_dilation": [
+    1,
+    2,
+    3,
+    1,
+    1
+  ],
+  "tdnn_dim": [
+    512,
+    512,
+    512,
+    512,
+    1500
+  ],
+  "tdnn_kernel": [
+    5,
+    3,
+    3,
+    1,
+    1
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.28.1",
+  "use_weighted_layer_sum": false,
+  "vocab_size": 32,
+  "xvector_output_dim": 512
+}
diff --git a/checkpoint-1584/optimizer.pt b/checkpoint-1584/optimizer.pt
new file mode 100644
index 0000000000000000000000000000000000000000..6495c748166be1b880604cad9e6d2960d39709cc
--- /dev/null
+++ b/checkpoint-1584/optimizer.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:97f0993b621c25d8592902e4e8228f4c10e5d60ccfbbe699e0809df117c76619
+size 756677477
diff --git a/checkpoint-1584/preprocessor_config.json b/checkpoint-1584/preprocessor_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..a0b7227fc1d916e469b14f6c154ad6dfea1e6891
--- /dev/null
+++ b/checkpoint-1584/preprocessor_config.json
@@ -0,0 +1,9 @@
+{
+  "do_normalize": true,
+  "feature_extractor_type": "Wav2Vec2FeatureExtractor",
+  "feature_size": 1,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": false,
+  "sampling_rate": 16000
+}
diff --git a/checkpoint-1584/pytorch_model.bin b/checkpoint-1584/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..e554ec84d5c760be08839a3019b0edc3a9526e9c
--- /dev/null
+++ b/checkpoint-1584/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6d312aab2192f2c6ad837fe9faa7663c78d9b1914e5bb72cd670d6a9d86c877d
+size 378348563
diff --git a/checkpoint-1584/rng_state.pth b/checkpoint-1584/rng_state.pth
new file mode 100644
index 0000000000000000000000000000000000000000..6521078fa0c7890042417386a87138457b933d4b
--- /dev/null
+++ b/checkpoint-1584/rng_state.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0789e5162b4faec730579ea3006c9e972645bf6ca3f783cc8f5e254a0f9f3f02
+size 14567
diff --git a/checkpoint-1584/scheduler.pt b/checkpoint-1584/scheduler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..818dd1da5e1487ccbe711776bf9e409009c88d3b
--- /dev/null
+++ b/checkpoint-1584/scheduler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ac439ca05576f5000d0f9bd8a9fbe56cdda19ab74e39b0502bfc5ea1a4f0613d
+size 623
diff --git a/checkpoint-1584/trainer_state.json b/checkpoint-1584/trainer_state.json
new file mode 100644
index 0000000000000000000000000000000000000000..d1cb3506e27589a494b96a3888162aeaabfae700
--- /dev/null
+++ b/checkpoint-1584/trainer_state.json
@@ -0,0 +1,9556 @@
+{
+  "best_metric": 0.3537317216396332,
+  "best_model_checkpoint": "accent_classification/checkpoint-792",
+  "epoch": 4.0,
+  "global_step": 1584,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.000000000000001e-07,
+      "loss": 0.6956,
+      "step": 1
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.2000000000000002e-06,
+      "loss": 0.6979,
+      "step": 2
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8e-06,
+      "loss": 0.7014,
+      "step": 3
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 2.4000000000000003e-06,
+      "loss": 0.6984,
+      "step": 4
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 3e-06,
+      "loss": 0.6978,
+      "step": 5
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 3.6e-06,
+      "loss": 0.6783,
+      "step": 6
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.2000000000000004e-06,
+      "loss": 0.6956,
+      "step": 7
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.800000000000001e-06,
+      "loss": 0.6994,
+      "step": 8
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 5.4e-06,
+      "loss": 0.6902,
+      "step": 9
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6e-06,
+      "loss": 0.6937,
+      "step": 10
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6.6e-06,
+      "loss": 0.6497,
+      "step": 11
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.2e-06,
+      "loss": 0.6911,
+      "step": 12
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.8e-06,
+      "loss": 0.6904,
+      "step": 13
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 8.400000000000001e-06,
+      "loss": 0.6854,
+      "step": 14
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9e-06,
+      "loss": 0.6843,
+      "step": 15
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9.600000000000001e-06,
+      "loss": 0.6851,
+      "step": 16
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.02e-05,
+      "loss": 0.6733,
+      "step": 17
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.08e-05,
+      "loss": 0.6819,
+      "step": 18
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.1400000000000001e-05,
+      "loss": 0.681,
+      "step": 19
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.6998,
+      "step": 20
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.26e-05,
+      "loss": 0.6743,
+      "step": 21
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.32e-05,
+      "loss": 0.6818,
+      "step": 22
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3800000000000002e-05,
+      "loss": 0.5331,
+      "step": 23
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.44e-05,
+      "loss": 0.6681,
+      "step": 24
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.6955,
+      "step": 25
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.56e-05,
+      "loss": 0.7,
+      "step": 26
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.62e-05,
+      "loss": 0.6975,
+      "step": 27
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.6800000000000002e-05,
+      "loss": 0.6249,
+      "step": 28
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.74e-05,
+      "loss": 0.6979,
+      "step": 29
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.6076,
+      "step": 30
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.86e-05,
+      "loss": 0.6091,
+      "step": 31
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9200000000000003e-05,
+      "loss": 0.6438,
+      "step": 32
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.98e-05,
+      "loss": 0.6624,
+      "step": 33
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.04e-05,
+      "loss": 0.6452,
+      "step": 34
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.6963,
+      "step": 35
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.16e-05,
+      "loss": 0.758,
+      "step": 36
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.22e-05,
+      "loss": 0.6471,
+      "step": 37
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.2800000000000002e-05,
+      "loss": 0.6313,
+      "step": 38
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.3400000000000003e-05,
+      "loss": 0.7019,
+      "step": 39
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.6336,
+      "step": 40
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4599999999999998e-05,
+      "loss": 0.7091,
+      "step": 41
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.52e-05,
+      "loss": 0.701,
+      "step": 42
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.58e-05,
+      "loss": 0.7036,
+      "step": 43
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.64e-05,
+      "loss": 0.7073,
+      "step": 44
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.614,
+      "step": 45
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.7600000000000003e-05,
+      "loss": 0.7543,
+      "step": 46
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.8199999999999998e-05,
+      "loss": 0.538,
+      "step": 47
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.88e-05,
+      "loss": 0.5179,
+      "step": 48
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.94e-05,
+      "loss": 0.6164,
+      "step": 49
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 3e-05,
+      "loss": 0.6093,
+      "step": 50
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.99923273657289e-05,
+      "loss": 0.6983,
+      "step": 51
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9984654731457804e-05,
+      "loss": 0.6069,
+      "step": 52
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9976982097186702e-05,
+      "loss": 0.5933,
+      "step": 53
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.99693094629156e-05,
+      "loss": 0.552,
+      "step": 54
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9961636828644502e-05,
+      "loss": 0.5865,
+      "step": 55
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9953964194373404e-05,
+      "loss": 0.7146,
+      "step": 56
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9946291560102302e-05,
+      "loss": 0.5591,
+      "step": 57
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9938618925831203e-05,
+      "loss": 0.7366,
+      "step": 58
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9930946291560102e-05,
+      "loss": 0.7006,
+      "step": 59
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9923273657289003e-05,
+      "loss": 0.4148,
+      "step": 60
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9915601023017905e-05,
+      "loss": 0.9938,
+      "step": 61
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9907928388746803e-05,
+      "loss": 0.911,
+      "step": 62
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.99002557544757e-05,
+      "loss": 0.5682,
+      "step": 63
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9892583120204606e-05,
+      "loss": 0.5833,
+      "step": 64
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9884910485933505e-05,
+      "loss": 0.7237,
+      "step": 65
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9877237851662403e-05,
+      "loss": 0.4013,
+      "step": 66
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9869565217391304e-05,
+      "loss": 0.9695,
+      "step": 67
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9861892583120206e-05,
+      "loss": 0.568,
+      "step": 68
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9854219948849104e-05,
+      "loss": 0.5697,
+      "step": 69
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9846547314578006e-05,
+      "loss": 0.7675,
+      "step": 70
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9838874680306904e-05,
+      "loss": 0.8831,
+      "step": 71
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9831202046035806e-05,
+      "loss": 0.7008,
+      "step": 72
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9823529411764707e-05,
+      "loss": 0.7085,
+      "step": 73
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9815856777493606e-05,
+      "loss": 0.5374,
+      "step": 74
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9808184143222507e-05,
+      "loss": 0.4131,
+      "step": 75
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.980051150895141e-05,
+      "loss": 0.7309,
+      "step": 76
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9792838874680307e-05,
+      "loss": 0.4279,
+      "step": 77
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.978516624040921e-05,
+      "loss": 0.5813,
+      "step": 78
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.977749360613811e-05,
+      "loss": 0.6276,
+      "step": 79
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976982097186701e-05,
+      "loss": 0.7892,
+      "step": 80
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976214833759591e-05,
+      "loss": 0.7134,
+      "step": 81
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.9754475703324808e-05,
+      "loss": 0.3682,
+      "step": 82
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.974680306905371e-05,
+      "loss": 0.5562,
+      "step": 83
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973913043478261e-05,
+      "loss": 0.7158,
+      "step": 84
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973145780051151e-05,
+      "loss": 0.524,
+      "step": 85
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9723785166240408e-05,
+      "loss": 0.3372,
+      "step": 86
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9716112531969313e-05,
+      "loss": 0.4979,
+      "step": 87
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970843989769821e-05,
+      "loss": 0.3177,
+      "step": 88
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970076726342711e-05,
+      "loss": 0.3488,
+      "step": 89
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.969309462915601e-05,
+      "loss": 0.5158,
+      "step": 90
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9685421994884913e-05,
+      "loss": 0.3208,
+      "step": 91
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.967774936061381e-05,
+      "loss": 0.6654,
+      "step": 92
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9670076726342712e-05,
+      "loss": 0.5581,
+      "step": 93
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.966240409207161e-05,
+      "loss": 0.7456,
+      "step": 94
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9654731457800512e-05,
+      "loss": 0.4437,
+      "step": 95
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9647058823529414e-05,
+      "loss": 0.5525,
+      "step": 96
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9639386189258312e-05,
+      "loss": 0.6673,
+      "step": 97
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.963171355498721e-05,
+      "loss": 0.435,
+      "step": 98
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9624040920716115e-05,
+      "loss": 0.7415,
+      "step": 99
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9616368286445014e-05,
+      "loss": 0.2981,
+      "step": 100
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9608695652173912e-05,
+      "loss": 0.5179,
+      "step": 101
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9601023017902817e-05,
+      "loss": 0.4005,
+      "step": 102
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9593350383631715e-05,
+      "loss": 0.4598,
+      "step": 103
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9585677749360613e-05,
+      "loss": 0.2553,
+      "step": 104
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9578005115089515e-05,
+      "loss": 0.4302,
+      "step": 105
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9570332480818416e-05,
+      "loss": 0.7894,
+      "step": 106
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9562659846547315e-05,
+      "loss": 0.213,
+      "step": 107
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9554987212276216e-05,
+      "loss": 0.5507,
+      "step": 108
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9547314578005115e-05,
+      "loss": 0.7676,
+      "step": 109
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9539641943734016e-05,
+      "loss": 0.3418,
+      "step": 110
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9531969309462918e-05,
+      "loss": 0.8533,
+      "step": 111
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9524296675191816e-05,
+      "loss": 0.3659,
+      "step": 112
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9516624040920714e-05,
+      "loss": 0.6482,
+      "step": 113
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.950895140664962e-05,
+      "loss": 0.5484,
+      "step": 114
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9501278772378517e-05,
+      "loss": 0.4508,
+      "step": 115
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9493606138107416e-05,
+      "loss": 0.4401,
+      "step": 116
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9485933503836317e-05,
+      "loss": 0.5455,
+      "step": 117
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.947826086956522e-05,
+      "loss": 0.722,
+      "step": 118
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9470588235294117e-05,
+      "loss": 0.2874,
+      "step": 119
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.946291560102302e-05,
+      "loss": 0.5973,
+      "step": 120
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.9455242966751917e-05,
+      "loss": 0.5898,
+      "step": 121
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.944757033248082e-05,
+      "loss": 1.0125,
+      "step": 122
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943989769820972e-05,
+      "loss": 0.2511,
+      "step": 123
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943222506393862e-05,
+      "loss": 0.545,
+      "step": 124
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.942455242966752e-05,
+      "loss": 0.593,
+      "step": 125
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.941687979539642e-05,
+      "loss": 0.2319,
+      "step": 126
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.940920716112532e-05,
+      "loss": 0.2215,
+      "step": 127
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.9401534526854218e-05,
+      "loss": 0.1964,
+      "step": 128
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9393861892583123e-05,
+      "loss": 0.1471,
+      "step": 129
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.938618925831202e-05,
+      "loss": 1.5965,
+      "step": 130
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9378516624040923e-05,
+      "loss": 1.0772,
+      "step": 131
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.937084398976982e-05,
+      "loss": 0.2005,
+      "step": 132
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9363171355498723e-05,
+      "loss": 0.9628,
+      "step": 133
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9355498721227624e-05,
+      "loss": 0.1503,
+      "step": 134
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9347826086956523e-05,
+      "loss": 0.6201,
+      "step": 135
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.934015345268542e-05,
+      "loss": 0.6094,
+      "step": 136
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9332480818414326e-05,
+      "loss": 1.385,
+      "step": 137
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9324808184143224e-05,
+      "loss": 0.1552,
+      "step": 138
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9317135549872122e-05,
+      "loss": 0.5828,
+      "step": 139
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9309462915601024e-05,
+      "loss": 0.6018,
+      "step": 140
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9301790281329925e-05,
+      "loss": 1.0614,
+      "step": 141
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9294117647058824e-05,
+      "loss": 0.566,
+      "step": 142
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9286445012787725e-05,
+      "loss": 0.5958,
+      "step": 143
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9278772378516624e-05,
+      "loss": 0.5838,
+      "step": 144
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9271099744245525e-05,
+      "loss": 0.5909,
+      "step": 145
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9263427109974427e-05,
+      "loss": 0.5862,
+      "step": 146
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9255754475703325e-05,
+      "loss": 0.2031,
+      "step": 147
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9248081841432223e-05,
+      "loss": 0.1995,
+      "step": 148
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9240409207161128e-05,
+      "loss": 0.24,
+      "step": 149
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9232736572890026e-05,
+      "loss": 0.5845,
+      "step": 150
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9225063938618925e-05,
+      "loss": 0.964,
+      "step": 151
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.921739130434783e-05,
+      "loss": 0.9595,
+      "step": 152
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9209718670076728e-05,
+      "loss": 0.2033,
+      "step": 153
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9202046035805626e-05,
+      "loss": 0.213,
+      "step": 154
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9194373401534528e-05,
+      "loss": 0.2042,
+      "step": 155
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.918670076726343e-05,
+      "loss": 0.5783,
+      "step": 156
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9179028132992328e-05,
+      "loss": 1.0822,
+      "step": 157
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.917135549872123e-05,
+      "loss": 1.2643,
+      "step": 158
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9163682864450127e-05,
+      "loss": 0.2593,
+      "step": 159
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.915601023017903e-05,
+      "loss": 0.5783,
+      "step": 160
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914833759590793e-05,
+      "loss": 0.5721,
+      "step": 161
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914066496163683e-05,
+      "loss": 0.5808,
+      "step": 162
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9132992327365727e-05,
+      "loss": 1.2578,
+      "step": 163
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9125319693094632e-05,
+      "loss": 0.5709,
+      "step": 164
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.911764705882353e-05,
+      "loss": 0.2159,
+      "step": 165
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910997442455243e-05,
+      "loss": 0.2329,
+      "step": 166
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910230179028133e-05,
+      "loss": 0.926,
+      "step": 167
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.9094629156010232e-05,
+      "loss": 0.9095,
+      "step": 168
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.908695652173913e-05,
+      "loss": 0.5622,
+      "step": 169
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907928388746803e-05,
+      "loss": 0.5703,
+      "step": 170
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907161125319693e-05,
+      "loss": 0.5669,
+      "step": 171
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.906393861892583e-05,
+      "loss": 0.2637,
+      "step": 172
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9056265984654733e-05,
+      "loss": 1.1077,
+      "step": 173
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.904859335038363e-05,
+      "loss": 0.5653,
+      "step": 174
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9040920716112533e-05,
+      "loss": 0.5584,
+      "step": 175
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9033248081841434e-05,
+      "loss": 0.5544,
+      "step": 176
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9025575447570333e-05,
+      "loss": 0.5663,
+      "step": 177
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.901790281329923e-05,
+      "loss": 1.0808,
+      "step": 178
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9010230179028136e-05,
+      "loss": 0.8736,
+      "step": 179
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9002557544757034e-05,
+      "loss": 0.6435,
+      "step": 180
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8994884910485932e-05,
+      "loss": 0.786,
+      "step": 181
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8987212276214834e-05,
+      "loss": 0.5717,
+      "step": 182
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8979539641943736e-05,
+      "loss": 0.5618,
+      "step": 183
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8971867007672637e-05,
+      "loss": 0.5785,
+      "step": 184
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8964194373401535e-05,
+      "loss": 0.7601,
+      "step": 185
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8956521739130434e-05,
+      "loss": 0.8112,
+      "step": 186
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.894884910485934e-05,
+      "loss": 0.5663,
+      "step": 187
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8941176470588237e-05,
+      "loss": 0.5631,
+      "step": 188
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8933503836317135e-05,
+      "loss": 0.7363,
+      "step": 189
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8925831202046037e-05,
+      "loss": 0.4552,
+      "step": 190
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8918158567774938e-05,
+      "loss": 0.4599,
+      "step": 191
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8910485933503836e-05,
+      "loss": 0.7848,
+      "step": 192
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8902813299232738e-05,
+      "loss": 0.5907,
+      "step": 193
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8895140664961636e-05,
+      "loss": 0.4264,
+      "step": 194
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8887468030690538e-05,
+      "loss": 0.5697,
+      "step": 195
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.887979539641944e-05,
+      "loss": 0.7643,
+      "step": 196
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8872122762148338e-05,
+      "loss": 0.3635,
+      "step": 197
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8864450127877236e-05,
+      "loss": 0.3715,
+      "step": 198
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.885677749360614e-05,
+      "loss": 0.554,
+      "step": 199
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.884910485933504e-05,
+      "loss": 0.8044,
+      "step": 200
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8841432225063937e-05,
+      "loss": 0.5673,
+      "step": 201
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8833759590792842e-05,
+      "loss": 0.8042,
+      "step": 202
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.882608695652174e-05,
+      "loss": 0.5638,
+      "step": 203
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881841432225064e-05,
+      "loss": 0.3584,
+      "step": 204
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881074168797954e-05,
+      "loss": 0.8211,
+      "step": 205
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.8803069053708442e-05,
+      "loss": 0.5708,
+      "step": 206
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.879539641943734e-05,
+      "loss": 0.8301,
+      "step": 207
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8787723785166242e-05,
+      "loss": 0.8229,
+      "step": 208
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.878005115089514e-05,
+      "loss": 0.5645,
+      "step": 209
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8772378516624042e-05,
+      "loss": 0.3529,
+      "step": 210
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8764705882352943e-05,
+      "loss": 0.5578,
+      "step": 211
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.875703324808184e-05,
+      "loss": 0.5659,
+      "step": 212
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.874936061381074e-05,
+      "loss": 0.7495,
+      "step": 213
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8741687979539645e-05,
+      "loss": 0.3162,
+      "step": 214
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8734015345268543e-05,
+      "loss": 0.5616,
+      "step": 215
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.872634271099744e-05,
+      "loss": 0.4034,
+      "step": 216
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8718670076726343e-05,
+      "loss": 0.3163,
+      "step": 217
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8710997442455245e-05,
+      "loss": 0.5514,
+      "step": 218
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8703324808184143e-05,
+      "loss": 0.826,
+      "step": 219
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8695652173913044e-05,
+      "loss": 0.8363,
+      "step": 220
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8687979539641943e-05,
+      "loss": 0.8288,
+      "step": 221
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8680306905370844e-05,
+      "loss": 0.5929,
+      "step": 222
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8672634271099746e-05,
+      "loss": 0.565,
+      "step": 223
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8664961636828644e-05,
+      "loss": 0.5642,
+      "step": 224
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8657289002557546e-05,
+      "loss": 0.5609,
+      "step": 225
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8649616368286447e-05,
+      "loss": 0.5641,
+      "step": 226
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8641943734015345e-05,
+      "loss": 0.8535,
+      "step": 227
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8634271099744244e-05,
+      "loss": 0.5652,
+      "step": 228
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.862659846547315e-05,
+      "loss": 0.4397,
+      "step": 229
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8618925831202047e-05,
+      "loss": 0.567,
+      "step": 230
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8611253196930945e-05,
+      "loss": 0.7517,
+      "step": 231
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8603580562659847e-05,
+      "loss": 0.5839,
+      "step": 232
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.859590792838875e-05,
+      "loss": 0.5677,
+      "step": 233
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8588235294117647e-05,
+      "loss": 0.5678,
+      "step": 234
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8580562659846548e-05,
+      "loss": 1.3045,
+      "step": 235
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8572890025575446e-05,
+      "loss": 0.804,
+      "step": 236
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.856521739130435e-05,
+      "loss": 1.0303,
+      "step": 237
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.855754475703325e-05,
+      "loss": 0.3303,
+      "step": 238
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8549872122762148e-05,
+      "loss": 0.5677,
+      "step": 239
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.854219948849105e-05,
+      "loss": 0.7972,
+      "step": 240
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.853452685421995e-05,
+      "loss": 0.5644,
+      "step": 241
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.852685421994885e-05,
+      "loss": 0.9294,
+      "step": 242
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.851918158567775e-05,
+      "loss": 0.5665,
+      "step": 243
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.851150895140665e-05,
+      "loss": 0.7963,
+      "step": 244
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.850383631713555e-05,
+      "loss": 0.5659,
+      "step": 245
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.8496163682864452e-05,
+      "loss": 0.5718,
+      "step": 246
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.848849104859335e-05,
+      "loss": 0.3526,
+      "step": 247
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8480818414322252e-05,
+      "loss": 0.574,
+      "step": 248
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8473145780051154e-05,
+      "loss": 0.9996,
+      "step": 249
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8465473145780052e-05,
+      "loss": 0.7634,
+      "step": 250
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.845780051150895e-05,
+      "loss": 0.5639,
+      "step": 251
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8450127877237855e-05,
+      "loss": 0.5737,
+      "step": 252
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8442455242966753e-05,
+      "loss": 0.7793,
+      "step": 253
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8434782608695652e-05,
+      "loss": 0.6769,
+      "step": 254
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8427109974424553e-05,
+      "loss": 0.5748,
+      "step": 255
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8419437340153455e-05,
+      "loss": 0.3602,
+      "step": 256
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8411764705882353e-05,
+      "loss": 0.5714,
+      "step": 257
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8404092071611255e-05,
+      "loss": 0.5738,
+      "step": 258
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8396419437340153e-05,
+      "loss": 0.5719,
+      "step": 259
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8388746803069055e-05,
+      "loss": 0.5753,
+      "step": 260
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8381074168797956e-05,
+      "loss": 0.5717,
+      "step": 261
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8373401534526854e-05,
+      "loss": 0.9794,
+      "step": 262
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8365728900255753e-05,
+      "loss": 0.5737,
+      "step": 263
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8358056265984658e-05,
+      "loss": 0.5723,
+      "step": 264
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8350383631713556e-05,
+      "loss": 0.5656,
+      "step": 265
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8342710997442454e-05,
+      "loss": 0.4259,
+      "step": 266
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8335038363171356e-05,
+      "loss": 0.3076,
+      "step": 267
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8327365728900257e-05,
+      "loss": 0.338,
+      "step": 268
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8319693094629156e-05,
+      "loss": 0.5612,
+      "step": 269
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8312020460358057e-05,
+      "loss": 0.5249,
+      "step": 270
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8304347826086955e-05,
+      "loss": 0.5671,
+      "step": 271
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8296675191815857e-05,
+      "loss": 0.5574,
+      "step": 272
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.828900255754476e-05,
+      "loss": 0.8152,
+      "step": 273
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8281329923273657e-05,
+      "loss": 0.5647,
+      "step": 274
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.827365728900256e-05,
+      "loss": 0.5632,
+      "step": 275
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.826598465473146e-05,
+      "loss": 0.3391,
+      "step": 276
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8258312020460358e-05,
+      "loss": 0.55,
+      "step": 277
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8250639386189257e-05,
+      "loss": 0.5672,
+      "step": 278
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.824296675191816e-05,
+      "loss": 0.285,
+      "step": 279
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.823529411764706e-05,
+      "loss": 0.2809,
+      "step": 280
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.8227621483375958e-05,
+      "loss": 0.5552,
+      "step": 281
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821994884910486e-05,
+      "loss": 0.563,
+      "step": 282
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821227621483376e-05,
+      "loss": 0.5628,
+      "step": 283
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.820460358056266e-05,
+      "loss": 0.2548,
+      "step": 284
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.819693094629156e-05,
+      "loss": 0.871,
+      "step": 285
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818925831202046e-05,
+      "loss": 0.8797,
+      "step": 286
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818158567774936e-05,
+      "loss": 0.2453,
+      "step": 287
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8173913043478262e-05,
+      "loss": 0.2464,
+      "step": 288
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.816624040920716e-05,
+      "loss": 0.3134,
+      "step": 289
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8158567774936062e-05,
+      "loss": 0.2431,
+      "step": 290
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8150895140664964e-05,
+      "loss": 0.2308,
+      "step": 291
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8143222506393862e-05,
+      "loss": 0.2173,
+      "step": 292
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8135549872122764e-05,
+      "loss": 0.5665,
+      "step": 293
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8127877237851662e-05,
+      "loss": 0.5592,
+      "step": 294
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8120204603580564e-05,
+      "loss": 0.473,
+      "step": 295
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8112531969309465e-05,
+      "loss": 0.9458,
+      "step": 296
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8104859335038363e-05,
+      "loss": 0.1879,
+      "step": 297
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8097186700767265e-05,
+      "loss": 0.6797,
+      "step": 298
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8089514066496167e-05,
+      "loss": 0.7592,
+      "step": 299
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8081841432225065e-05,
+      "loss": 0.4109,
+      "step": 300
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8074168797953963e-05,
+      "loss": 0.5821,
+      "step": 301
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8066496163682868e-05,
+      "loss": 0.2057,
+      "step": 302
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8058823529411766e-05,
+      "loss": 0.5322,
+      "step": 303
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8051150895140665e-05,
+      "loss": 0.5208,
+      "step": 304
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8043478260869566e-05,
+      "loss": 0.6799,
+      "step": 305
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8035805626598468e-05,
+      "loss": 0.5894,
+      "step": 306
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8028132992327366e-05,
+      "loss": 0.4051,
+      "step": 307
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8020460358056268e-05,
+      "loss": 0.3695,
+      "step": 308
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8012787723785166e-05,
+      "loss": 0.4869,
+      "step": 309
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8005115089514067e-05,
+      "loss": 0.6075,
+      "step": 310
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.799744245524297e-05,
+      "loss": 0.4245,
+      "step": 311
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7989769820971867e-05,
+      "loss": 0.5122,
+      "step": 312
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7982097186700765e-05,
+      "loss": 0.452,
+      "step": 313
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.797442455242967e-05,
+      "loss": 0.9568,
+      "step": 314
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.796675191815857e-05,
+      "loss": 0.5846,
+      "step": 315
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.7959079283887467e-05,
+      "loss": 0.3743,
+      "step": 316
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.795140664961637e-05,
+      "loss": 0.1922,
+      "step": 317
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.794373401534527e-05,
+      "loss": 0.5303,
+      "step": 318
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.793606138107417e-05,
+      "loss": 0.7253,
+      "step": 319
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.792838874680307e-05,
+      "loss": 0.2794,
+      "step": 320
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.7920716112531968e-05,
+      "loss": 0.577,
+      "step": 321
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.791304347826087e-05,
+      "loss": 0.614,
+      "step": 322
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.790537084398977e-05,
+      "loss": 0.5882,
+      "step": 323
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789769820971867e-05,
+      "loss": 0.2503,
+      "step": 324
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789002557544757e-05,
+      "loss": 0.4195,
+      "step": 325
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.7882352941176473e-05,
+      "loss": 0.2959,
+      "step": 326
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.787468030690537e-05,
+      "loss": 0.3732,
+      "step": 327
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.786700767263427e-05,
+      "loss": 0.5729,
+      "step": 328
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7859335038363174e-05,
+      "loss": 0.3179,
+      "step": 329
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7851662404092073e-05,
+      "loss": 0.6823,
+      "step": 330
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.784398976982097e-05,
+      "loss": 0.2663,
+      "step": 331
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7836317135549872e-05,
+      "loss": 0.4436,
+      "step": 332
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7828644501278774e-05,
+      "loss": 0.3089,
+      "step": 333
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7820971867007672e-05,
+      "loss": 0.5888,
+      "step": 334
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7813299232736574e-05,
+      "loss": 0.2115,
+      "step": 335
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7805626598465472e-05,
+      "loss": 0.4005,
+      "step": 336
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7797953964194374e-05,
+      "loss": 0.3097,
+      "step": 337
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7790281329923275e-05,
+      "loss": 0.5847,
+      "step": 338
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7782608695652174e-05,
+      "loss": 1.0164,
+      "step": 339
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7774936061381072e-05,
+      "loss": 1.1321,
+      "step": 340
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7767263427109977e-05,
+      "loss": 0.592,
+      "step": 341
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7759590792838875e-05,
+      "loss": 1.0191,
+      "step": 342
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7751918158567777e-05,
+      "loss": 0.9199,
+      "step": 343
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7744245524296675e-05,
+      "loss": 0.2874,
+      "step": 344
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7736572890025576e-05,
+      "loss": 0.2469,
+      "step": 345
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7728900255754478e-05,
+      "loss": 0.8364,
+      "step": 346
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7721227621483376e-05,
+      "loss": 0.7365,
+      "step": 347
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7713554987212278e-05,
+      "loss": 0.5595,
+      "step": 348
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.770588235294118e-05,
+      "loss": 0.2546,
+      "step": 349
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7698209718670078e-05,
+      "loss": 0.8183,
+      "step": 350
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7690537084398976e-05,
+      "loss": 0.5385,
+      "step": 351
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.768286445012788e-05,
+      "loss": 0.2763,
+      "step": 352
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.767519181585678e-05,
+      "loss": 0.3045,
+      "step": 353
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7667519181585677e-05,
+      "loss": 0.3354,
+      "step": 354
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765984654731458e-05,
+      "loss": 0.9366,
+      "step": 355
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765217391304348e-05,
+      "loss": 0.6212,
+      "step": 356
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.764450127877238e-05,
+      "loss": 0.1161,
+      "step": 357
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.763682864450128e-05,
+      "loss": 0.4929,
+      "step": 358
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762915601023018e-05,
+      "loss": 0.8167,
+      "step": 359
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762148337595908e-05,
+      "loss": 0.2521,
+      "step": 360
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.7613810741687982e-05,
+      "loss": 0.4174,
+      "step": 361
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.760613810741688e-05,
+      "loss": 0.3822,
+      "step": 362
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.759846547314578e-05,
+      "loss": 0.5355,
+      "step": 363
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.7590792838874683e-05,
+      "loss": 0.561,
+      "step": 364
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.758312020460358e-05,
+      "loss": 0.5126,
+      "step": 365
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.757544757033248e-05,
+      "loss": 0.2301,
+      "step": 366
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.756777493606138e-05,
+      "loss": 0.1698,
+      "step": 367
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7560102301790283e-05,
+      "loss": 0.1634,
+      "step": 368
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.755242966751918e-05,
+      "loss": 0.5955,
+      "step": 369
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7544757033248083e-05,
+      "loss": 0.1568,
+      "step": 370
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.753708439897698e-05,
+      "loss": 0.5167,
+      "step": 371
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7529411764705883e-05,
+      "loss": 1.0028,
+      "step": 372
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7521739130434784e-05,
+      "loss": 1.2567,
+      "step": 373
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7514066496163682e-05,
+      "loss": 0.6094,
+      "step": 374
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7506393861892584e-05,
+      "loss": 0.6059,
+      "step": 375
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7498721227621486e-05,
+      "loss": 1.0593,
+      "step": 376
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7491048593350384e-05,
+      "loss": 0.1506,
+      "step": 377
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7483375959079282e-05,
+      "loss": 1.5106,
+      "step": 378
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7475703324808187e-05,
+      "loss": 0.1827,
+      "step": 379
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7468030690537085e-05,
+      "loss": 1.4961,
+      "step": 380
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7460358056265984e-05,
+      "loss": 1.0456,
+      "step": 381
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7452685421994885e-05,
+      "loss": 1.2063,
+      "step": 382
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7445012787723787e-05,
+      "loss": 0.5837,
+      "step": 383
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7437340153452685e-05,
+      "loss": 1.4536,
+      "step": 384
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7429667519181587e-05,
+      "loss": 0.5803,
+      "step": 385
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7421994884910485e-05,
+      "loss": 0.5896,
+      "step": 386
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7414322250639386e-05,
+      "loss": 0.1822,
+      "step": 387
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7406649616368288e-05,
+      "loss": 1.7893,
+      "step": 388
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7398976982097186e-05,
+      "loss": 0.9723,
+      "step": 389
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7391304347826085e-05,
+      "loss": 1.2815,
+      "step": 390
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.738363171355499e-05,
+      "loss": 0.5765,
+      "step": 391
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7375959079283888e-05,
+      "loss": 1.2429,
+      "step": 392
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7368286445012786e-05,
+      "loss": 0.1343,
+      "step": 393
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7360613810741688e-05,
+      "loss": 0.2306,
+      "step": 394
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.735294117647059e-05,
+      "loss": 0.624,
+      "step": 395
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.734526854219949e-05,
+      "loss": 0.2371,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7348484848484849,
+      "eval_loss": 0.5804409980773926,
+      "eval_runtime": 57.3336,
+      "eval_samples_per_second": 6.907,
+      "eval_steps_per_second": 1.727,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.733759590792839e-05,
+      "loss": 0.8989,
+      "step": 397
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.732992327365729e-05,
+      "loss": 0.5664,
+      "step": 398
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.7322250639386192e-05,
+      "loss": 0.8852,
+      "step": 399
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.731457800511509e-05,
+      "loss": 0.5661,
+      "step": 400
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.730690537084399e-05,
+      "loss": 0.5648,
+      "step": 401
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7299232736572894e-05,
+      "loss": 0.7333,
+      "step": 402
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7291560102301792e-05,
+      "loss": 0.3037,
+      "step": 403
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.728388746803069e-05,
+      "loss": 0.1463,
+      "step": 404
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7276214833759592e-05,
+      "loss": 0.2702,
+      "step": 405
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7268542199488493e-05,
+      "loss": 0.2693,
+      "step": 406
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.726086956521739e-05,
+      "loss": 0.5647,
+      "step": 407
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7253196930946293e-05,
+      "loss": 0.8507,
+      "step": 408
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.724552429667519e-05,
+      "loss": 0.8645,
+      "step": 409
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7237851662404093e-05,
+      "loss": 0.2649,
+      "step": 410
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7230179028132995e-05,
+      "loss": 0.6942,
+      "step": 411
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7222506393861893e-05,
+      "loss": 0.5656,
+      "step": 412
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.721483375959079e-05,
+      "loss": 0.5489,
+      "step": 413
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7207161125319696e-05,
+      "loss": 0.2652,
+      "step": 414
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7199488491048594e-05,
+      "loss": 0.8676,
+      "step": 415
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7191815856777493e-05,
+      "loss": 0.8621,
+      "step": 416
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7184143222506394e-05,
+      "loss": 0.2641,
+      "step": 417
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7176470588235296e-05,
+      "loss": 0.859,
+      "step": 418
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7168797953964194e-05,
+      "loss": 0.6492,
+      "step": 419
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7161125319693096e-05,
+      "loss": 1.1494,
+      "step": 420
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7153452685421994e-05,
+      "loss": 0.8278,
+      "step": 421
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7145780051150895e-05,
+      "loss": 0.764,
+      "step": 422
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7138107416879797e-05,
+      "loss": 0.564,
+      "step": 423
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7130434782608695e-05,
+      "loss": 0.8051,
+      "step": 424
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7122762148337597e-05,
+      "loss": 0.3052,
+      "step": 425
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.71150895140665e-05,
+      "loss": 0.2945,
+      "step": 426
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7107416879795397e-05,
+      "loss": 0.4118,
+      "step": 427
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7099744245524295e-05,
+      "loss": 0.559,
+      "step": 428
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.70920716112532e-05,
+      "loss": 0.5593,
+      "step": 429
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7084398976982098e-05,
+      "loss": 0.564,
+      "step": 430
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7076726342710996e-05,
+      "loss": 0.4454,
+      "step": 431
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7069053708439898e-05,
+      "loss": 0.4381,
+      "step": 432
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.70613810741688e-05,
+      "loss": 0.3718,
+      "step": 433
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7053708439897698e-05,
+      "loss": 0.6627,
+      "step": 434
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70460358056266e-05,
+      "loss": 0.4004,
+      "step": 435
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7038363171355498e-05,
+      "loss": 0.3855,
+      "step": 436
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70306905370844e-05,
+      "loss": 0.4799,
+      "step": 437
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70230179028133e-05,
+      "loss": 0.6714,
+      "step": 438
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70153452685422e-05,
+      "loss": 0.622,
+      "step": 439
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7007672634271097e-05,
+      "loss": 0.3916,
+      "step": 440
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.5487,
+      "step": 441
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69923273657289e-05,
+      "loss": 0.376,
+      "step": 442
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69846547314578e-05,
+      "loss": 0.3616,
+      "step": 443
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69769820971867e-05,
+      "loss": 0.8747,
+      "step": 444
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.6969309462915602e-05,
+      "loss": 0.283,
+      "step": 445
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.69616368286445e-05,
+      "loss": 0.3988,
+      "step": 446
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6953964194373402e-05,
+      "loss": 0.4569,
+      "step": 447
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6946291560102303e-05,
+      "loss": 0.2005,
+      "step": 448
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6938618925831205e-05,
+      "loss": 0.3087,
+      "step": 449
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6930946291560103e-05,
+      "loss": 0.5814,
+      "step": 450
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6923273657289e-05,
+      "loss": 0.2995,
+      "step": 451
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6915601023017907e-05,
+      "loss": 0.7806,
+      "step": 452
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6907928388746805e-05,
+      "loss": 0.2666,
+      "step": 453
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6900255754475703e-05,
+      "loss": 0.2571,
+      "step": 454
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6892583120204605e-05,
+      "loss": 0.5931,
+      "step": 455
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6884910485933506e-05,
+      "loss": 0.475,
+      "step": 456
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6877237851662404e-05,
+      "loss": 0.2372,
+      "step": 457
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6869565217391306e-05,
+      "loss": 0.1577,
+      "step": 458
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6861892583120204e-05,
+      "loss": 0.1953,
+      "step": 459
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6854219948849106e-05,
+      "loss": 0.1931,
+      "step": 460
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6846547314578008e-05,
+      "loss": 0.1829,
+      "step": 461
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6838874680306906e-05,
+      "loss": 0.1601,
+      "step": 462
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6831202046035804e-05,
+      "loss": 0.2505,
+      "step": 463
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.682352941176471e-05,
+      "loss": 0.181,
+      "step": 464
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6815856777493607e-05,
+      "loss": 0.5978,
+      "step": 465
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6808184143222505e-05,
+      "loss": 0.5458,
+      "step": 466
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6800511508951407e-05,
+      "loss": 0.1289,
+      "step": 467
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.679283887468031e-05,
+      "loss": 0.4383,
+      "step": 468
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6785166240409207e-05,
+      "loss": 0.5892,
+      "step": 469
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.677749360613811e-05,
+      "loss": 1.3565,
+      "step": 470
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.676982097186701e-05,
+      "loss": 0.7267,
+      "step": 471
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.6762148337595908e-05,
+      "loss": 0.1354,
+      "step": 472
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.675447570332481e-05,
+      "loss": 0.9291,
+      "step": 473
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.6746803069053708e-05,
+      "loss": 0.5265,
+      "step": 474
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673913043478261e-05,
+      "loss": 0.4841,
+      "step": 475
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673145780051151e-05,
+      "loss": 1.2391,
+      "step": 476
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.672378516624041e-05,
+      "loss": 0.4924,
+      "step": 477
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6716112531969308e-05,
+      "loss": 1.0076,
+      "step": 478
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6708439897698213e-05,
+      "loss": 0.6314,
+      "step": 479
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.670076726342711e-05,
+      "loss": 0.4237,
+      "step": 480
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.669309462915601e-05,
+      "loss": 0.1077,
+      "step": 481
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.668542199488491e-05,
+      "loss": 0.3038,
+      "step": 482
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6677749360613812e-05,
+      "loss": 0.6081,
+      "step": 483
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.667007672634271e-05,
+      "loss": 0.1078,
+      "step": 484
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6662404092071612e-05,
+      "loss": 0.4296,
+      "step": 485
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.665473145780051e-05,
+      "loss": 0.3981,
+      "step": 486
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6647058823529412e-05,
+      "loss": 0.5853,
+      "step": 487
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6639386189258314e-05,
+      "loss": 0.802,
+      "step": 488
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6631713554987212e-05,
+      "loss": 0.4576,
+      "step": 489
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.662404092071611e-05,
+      "loss": 0.2682,
+      "step": 490
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6616368286445015e-05,
+      "loss": 0.1991,
+      "step": 491
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6608695652173913e-05,
+      "loss": 0.1409,
+      "step": 492
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.660102301790281e-05,
+      "loss": 0.3869,
+      "step": 493
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6593350383631713e-05,
+      "loss": 0.5301,
+      "step": 494
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6585677749360615e-05,
+      "loss": 0.8116,
+      "step": 495
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6578005115089513e-05,
+      "loss": 0.0984,
+      "step": 496
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6570332480818415e-05,
+      "loss": 0.7773,
+      "step": 497
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6562659846547316e-05,
+      "loss": 1.3522,
+      "step": 498
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6554987212276215e-05,
+      "loss": 0.0862,
+      "step": 499
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6547314578005116e-05,
+      "loss": 0.0894,
+      "step": 500
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6539641943734014e-05,
+      "loss": 0.0885,
+      "step": 501
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.653196930946292e-05,
+      "loss": 0.3031,
+      "step": 502
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6524296675191818e-05,
+      "loss": 0.187,
+      "step": 503
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6516624040920716e-05,
+      "loss": 1.3063,
+      "step": 504
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6508951406649617e-05,
+      "loss": 0.6847,
+      "step": 505
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.650127877237852e-05,
+      "loss": 0.078,
+      "step": 506
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6493606138107417e-05,
+      "loss": 0.1265,
+      "step": 507
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.648593350383632e-05,
+      "loss": 0.0804,
+      "step": 508
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.6478260869565217e-05,
+      "loss": 0.6976,
+      "step": 509
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.647058823529412e-05,
+      "loss": 0.7977,
+      "step": 510
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.646291560102302e-05,
+      "loss": 0.413,
+      "step": 511
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.645524296675192e-05,
+      "loss": 0.6944,
+      "step": 512
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6447570332480817e-05,
+      "loss": 0.7177,
+      "step": 513
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6439897698209722e-05,
+      "loss": 0.0581,
+      "step": 514
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.643222506393862e-05,
+      "loss": 0.9308,
+      "step": 515
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6424552429667518e-05,
+      "loss": 0.2518,
+      "step": 516
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.641687979539642e-05,
+      "loss": 0.1014,
+      "step": 517
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640920716112532e-05,
+      "loss": 0.1641,
+      "step": 518
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640153452685422e-05,
+      "loss": 0.101,
+      "step": 519
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.639386189258312e-05,
+      "loss": 0.1097,
+      "step": 520
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6386189258312023e-05,
+      "loss": 0.987,
+      "step": 521
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.637851662404092e-05,
+      "loss": 0.6465,
+      "step": 522
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6370843989769823e-05,
+      "loss": 0.0914,
+      "step": 523
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.636317135549872e-05,
+      "loss": 0.139,
+      "step": 524
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6355498721227623e-05,
+      "loss": 0.6455,
+      "step": 525
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6347826086956524e-05,
+      "loss": 0.204,
+      "step": 526
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6340153452685422e-05,
+      "loss": 0.1269,
+      "step": 527
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.633248081841432e-05,
+      "loss": 0.2214,
+      "step": 528
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6324808184143226e-05,
+      "loss": 0.0736,
+      "step": 529
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6317135549872124e-05,
+      "loss": 0.0714,
+      "step": 530
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6309462915601022e-05,
+      "loss": 0.1388,
+      "step": 531
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6301790281329924e-05,
+      "loss": 0.7758,
+      "step": 532
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6294117647058825e-05,
+      "loss": 0.3388,
+      "step": 533
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6286445012787724e-05,
+      "loss": 1.1831,
+      "step": 534
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6278772378516625e-05,
+      "loss": 0.1166,
+      "step": 535
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6271099744245523e-05,
+      "loss": 0.2389,
+      "step": 536
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6263427109974425e-05,
+      "loss": 0.1622,
+      "step": 537
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6255754475703327e-05,
+      "loss": 0.1015,
+      "step": 538
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6248081841432225e-05,
+      "loss": 0.0658,
+      "step": 539
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6240409207161123e-05,
+      "loss": 0.0954,
+      "step": 540
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6232736572890028e-05,
+      "loss": 0.1182,
+      "step": 541
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6225063938618926e-05,
+      "loss": 0.0605,
+      "step": 542
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6217391304347824e-05,
+      "loss": 0.9265,
+      "step": 543
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6209718670076726e-05,
+      "loss": 1.2994,
+      "step": 544
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6202046035805628e-05,
+      "loss": 0.081,
+      "step": 545
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6194373401534526e-05,
+      "loss": 0.5962,
+      "step": 546
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6186700767263428e-05,
+      "loss": 0.0665,
+      "step": 547
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.617902813299233e-05,
+      "loss": 0.0655,
+      "step": 548
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6171355498721227e-05,
+      "loss": 0.0642,
+      "step": 549
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.616368286445013e-05,
+      "loss": 0.0834,
+      "step": 550
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6156010230179027e-05,
+      "loss": 0.047,
+      "step": 551
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6148337595907932e-05,
+      "loss": 0.0642,
+      "step": 552
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.614066496163683e-05,
+      "loss": 0.0594,
+      "step": 553
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.613299232736573e-05,
+      "loss": 0.0573,
+      "step": 554
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.612531969309463e-05,
+      "loss": 0.1101,
+      "step": 555
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.6117647058823532e-05,
+      "loss": 0.5121,
+      "step": 556
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.610997442455243e-05,
+      "loss": 0.0374,
+      "step": 557
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.6102301790281332e-05,
+      "loss": 0.0957,
+      "step": 558
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.609462915601023e-05,
+      "loss": 0.0488,
+      "step": 559
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.608695652173913e-05,
+      "loss": 0.0482,
+      "step": 560
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6079283887468033e-05,
+      "loss": 1.2041,
+      "step": 561
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.607161125319693e-05,
+      "loss": 0.8833,
+      "step": 562
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.606393861892583e-05,
+      "loss": 0.1544,
+      "step": 563
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6056265984654735e-05,
+      "loss": 0.046,
+      "step": 564
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6048593350383633e-05,
+      "loss": 0.9073,
+      "step": 565
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.604092071611253e-05,
+      "loss": 0.052,
+      "step": 566
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6033248081841433e-05,
+      "loss": 0.0308,
+      "step": 567
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6025575447570334e-05,
+      "loss": 0.7032,
+      "step": 568
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6017902813299232e-05,
+      "loss": 0.6947,
+      "step": 569
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6010230179028134e-05,
+      "loss": 1.5603,
+      "step": 570
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6002557544757036e-05,
+      "loss": 0.8836,
+      "step": 571
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.5994884910485934e-05,
+      "loss": 1.7439,
+      "step": 572
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5987212276214836e-05,
+      "loss": 0.0313,
+      "step": 573
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5979539641943734e-05,
+      "loss": 0.0317,
+      "step": 574
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5971867007672635e-05,
+      "loss": 0.0325,
+      "step": 575
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5964194373401537e-05,
+      "loss": 0.5,
+      "step": 576
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5956521739130435e-05,
+      "loss": 0.0675,
+      "step": 577
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5948849104859333e-05,
+      "loss": 0.6364,
+      "step": 578
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.594117647058824e-05,
+      "loss": 0.0696,
+      "step": 579
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5933503836317137e-05,
+      "loss": 0.6195,
+      "step": 580
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5925831202046035e-05,
+      "loss": 0.5796,
+      "step": 581
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5918158567774937e-05,
+      "loss": 0.0832,
+      "step": 582
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5910485933503838e-05,
+      "loss": 0.0794,
+      "step": 583
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5902813299232736e-05,
+      "loss": 0.5851,
+      "step": 584
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5895140664961638e-05,
+      "loss": 0.104,
+      "step": 585
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5887468030690536e-05,
+      "loss": 1.0492,
+      "step": 586
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5879795396419438e-05,
+      "loss": 0.0645,
+      "step": 587
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.587212276214834e-05,
+      "loss": 0.514,
+      "step": 588
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5864450127877238e-05,
+      "loss": 0.0701,
+      "step": 589
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5856777493606136e-05,
+      "loss": 0.5409,
+      "step": 590
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584910485933504e-05,
+      "loss": 0.0536,
+      "step": 591
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584143222506394e-05,
+      "loss": 0.1728,
+      "step": 592
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.5833759590792837e-05,
+      "loss": 0.0845,
+      "step": 593
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.582608695652174e-05,
+      "loss": 0.2217,
+      "step": 594
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.581841432225064e-05,
+      "loss": 0.1397,
+      "step": 595
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.581074168797954e-05,
+      "loss": 0.4119,
+      "step": 596
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.580306905370844e-05,
+      "loss": 0.3638,
+      "step": 597
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.5795396419437342e-05,
+      "loss": 0.0781,
+      "step": 598
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.578772378516624e-05,
+      "loss": 0.5926,
+      "step": 599
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5780051150895142e-05,
+      "loss": 0.1378,
+      "step": 600
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.577237851662404e-05,
+      "loss": 0.142,
+      "step": 601
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.576470588235294e-05,
+      "loss": 0.1722,
+      "step": 602
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5757033248081843e-05,
+      "loss": 0.1529,
+      "step": 603
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.574936061381074e-05,
+      "loss": 1.6083,
+      "step": 604
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5741687979539643e-05,
+      "loss": 1.8267,
+      "step": 605
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5734015345268545e-05,
+      "loss": 0.0535,
+      "step": 606
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5726342710997443e-05,
+      "loss": 0.9945,
+      "step": 607
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5718670076726345e-05,
+      "loss": 0.9133,
+      "step": 608
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5710997442455243e-05,
+      "loss": 0.5295,
+      "step": 609
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5703324808184144e-05,
+      "loss": 0.058,
+      "step": 610
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5695652173913046e-05,
+      "loss": 0.4789,
+      "step": 611
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5687979539641944e-05,
+      "loss": 0.0572,
+      "step": 612
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5680306905370842e-05,
+      "loss": 0.0897,
+      "step": 613
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5672634271099747e-05,
+      "loss": 0.056,
+      "step": 614
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5664961636828646e-05,
+      "loss": 0.0278,
+      "step": 615
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5657289002557544e-05,
+      "loss": 0.9947,
+      "step": 616
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5649616368286445e-05,
+      "loss": 0.0627,
+      "step": 617
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5641943734015347e-05,
+      "loss": 0.9335,
+      "step": 618
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5634271099744245e-05,
+      "loss": 0.9793,
+      "step": 619
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5626598465473147e-05,
+      "loss": 1.8128,
+      "step": 620
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.561892583120205e-05,
+      "loss": 1.8015,
+      "step": 621
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5611253196930947e-05,
+      "loss": 0.9259,
+      "step": 622
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.560358056265985e-05,
+      "loss": 0.9618,
+      "step": 623
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5595907928388747e-05,
+      "loss": 0.0808,
+      "step": 624
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5588235294117648e-05,
+      "loss": 1.7161,
+      "step": 625
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.558056265984655e-05,
+      "loss": 0.8879,
+      "step": 626
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5572890025575448e-05,
+      "loss": 0.043,
+      "step": 627
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5565217391304346e-05,
+      "loss": 0.9449,
+      "step": 628
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.555754475703325e-05,
+      "loss": 0.0624,
+      "step": 629
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.554987212276215e-05,
+      "loss": 0.0924,
+      "step": 630
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5542199488491048e-05,
+      "loss": 0.0715,
+      "step": 631
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.553452685421995e-05,
+      "loss": 0.4995,
+      "step": 632
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.552685421994885e-05,
+      "loss": 0.5576,
+      "step": 633
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551918158567775e-05,
+      "loss": 0.06,
+      "step": 634
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551150895140665e-05,
+      "loss": 0.0873,
+      "step": 635
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.550383631713555e-05,
+      "loss": 0.6868,
+      "step": 636
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.549616368286445e-05,
+      "loss": 0.8487,
+      "step": 637
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.5488491048593352e-05,
+      "loss": 0.5641,
+      "step": 638
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.548081841432225e-05,
+      "loss": 1.1343,
+      "step": 639
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.547314578005115e-05,
+      "loss": 0.0614,
+      "step": 640
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5465473145780054e-05,
+      "loss": 0.9414,
+      "step": 641
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5457800511508952e-05,
+      "loss": 0.9226,
+      "step": 642
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.545012787723785e-05,
+      "loss": 0.3323,
+      "step": 643
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5442455242966752e-05,
+      "loss": 0.9315,
+      "step": 644
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5434782608695653e-05,
+      "loss": 0.5539,
+      "step": 645
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.542710997442455e-05,
+      "loss": 0.3376,
+      "step": 646
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5419437340153453e-05,
+      "loss": 0.0694,
+      "step": 647
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5411764705882355e-05,
+      "loss": 0.0706,
+      "step": 648
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5404092071611253e-05,
+      "loss": 0.105,
+      "step": 649
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5396419437340155e-05,
+      "loss": 0.912,
+      "step": 650
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5388746803069053e-05,
+      "loss": 0.0998,
+      "step": 651
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5381074168797954e-05,
+      "loss": 0.2247,
+      "step": 652
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5373401534526856e-05,
+      "loss": 0.2869,
+      "step": 653
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5365728900255754e-05,
+      "loss": 0.532,
+      "step": 654
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5358056265984653e-05,
+      "loss": 0.4723,
+      "step": 655
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5350383631713558e-05,
+      "loss": 0.1178,
+      "step": 656
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5342710997442456e-05,
+      "loss": 0.0566,
+      "step": 657
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5335038363171357e-05,
+      "loss": 0.059,
+      "step": 658
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5327365728900256e-05,
+      "loss": 0.0881,
+      "step": 659
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5319693094629157e-05,
+      "loss": 0.5762,
+      "step": 660
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.531202046035806e-05,
+      "loss": 0.0499,
+      "step": 661
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5304347826086957e-05,
+      "loss": 0.087,
+      "step": 662
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5296675191815855e-05,
+      "loss": 1.0719,
+      "step": 663
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528900255754476e-05,
+      "loss": 0.6158,
+      "step": 664
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528132992327366e-05,
+      "loss": 0.6117,
+      "step": 665
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5273657289002557e-05,
+      "loss": 0.0515,
+      "step": 666
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5265984654731458e-05,
+      "loss": 0.0608,
+      "step": 667
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.525831202046036e-05,
+      "loss": 0.0313,
+      "step": 668
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.5250639386189258e-05,
+      "loss": 0.1449,
+      "step": 669
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.524296675191816e-05,
+      "loss": 0.9883,
+      "step": 670
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.523529411764706e-05,
+      "loss": 0.0859,
+      "step": 671
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.522762148337596e-05,
+      "loss": 0.0561,
+      "step": 672
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521994884910486e-05,
+      "loss": 0.0926,
+      "step": 673
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521227621483376e-05,
+      "loss": 0.1101,
+      "step": 674
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.520460358056266e-05,
+      "loss": 0.068,
+      "step": 675
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5196930946291563e-05,
+      "loss": 0.0773,
+      "step": 676
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518925831202046e-05,
+      "loss": 0.0508,
+      "step": 677
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518158567774936e-05,
+      "loss": 0.0375,
+      "step": 678
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5173913043478264e-05,
+      "loss": 0.93,
+      "step": 679
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5166240409207162e-05,
+      "loss": 0.1507,
+      "step": 680
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.515856777493606e-05,
+      "loss": 0.0917,
+      "step": 681
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5150895140664962e-05,
+      "loss": 0.97,
+      "step": 682
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5143222506393864e-05,
+      "loss": 0.5019,
+      "step": 683
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5135549872122762e-05,
+      "loss": 0.0849,
+      "step": 684
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5127877237851664e-05,
+      "loss": 1.2316,
+      "step": 685
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5120204603580562e-05,
+      "loss": 0.0252,
+      "step": 686
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5112531969309463e-05,
+      "loss": 1.0543,
+      "step": 687
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5104859335038365e-05,
+      "loss": 0.0506,
+      "step": 688
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5097186700767263e-05,
+      "loss": 0.0252,
+      "step": 689
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.508951406649616e-05,
+      "loss": 0.6297,
+      "step": 690
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5081841432225066e-05,
+      "loss": 0.4497,
+      "step": 691
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5074168797953965e-05,
+      "loss": 0.0516,
+      "step": 692
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5066496163682863e-05,
+      "loss": 0.5501,
+      "step": 693
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5058823529411768e-05,
+      "loss": 0.5621,
+      "step": 694
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5051150895140666e-05,
+      "loss": 0.0577,
+      "step": 695
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5043478260869564e-05,
+      "loss": 0.0271,
+      "step": 696
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5035805626598466e-05,
+      "loss": 0.9496,
+      "step": 697
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5028132992327368e-05,
+      "loss": 0.0877,
+      "step": 698
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5020460358056266e-05,
+      "loss": 0.0523,
+      "step": 699
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5012787723785167e-05,
+      "loss": 0.0577,
+      "step": 700
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5005115089514066e-05,
+      "loss": 0.0522,
+      "step": 701
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.4997442455242967e-05,
+      "loss": 0.0517,
+      "step": 702
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.498976982097187e-05,
+      "loss": 0.6335,
+      "step": 703
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4982097186700767e-05,
+      "loss": 0.05,
+      "step": 704
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4974424552429665e-05,
+      "loss": 0.0753,
+      "step": 705
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.496675191815857e-05,
+      "loss": 0.0227,
+      "step": 706
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.495907928388747e-05,
+      "loss": 0.0755,
+      "step": 707
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.4951406649616367e-05,
+      "loss": 0.0215,
+      "step": 708
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.494373401534527e-05,
+      "loss": 0.0451,
+      "step": 709
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.493606138107417e-05,
+      "loss": 0.0343,
+      "step": 710
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492838874680307e-05,
+      "loss": 0.0836,
+      "step": 711
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492071611253197e-05,
+      "loss": 0.5287,
+      "step": 712
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4913043478260868e-05,
+      "loss": 0.0686,
+      "step": 713
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4905370843989773e-05,
+      "loss": 0.0211,
+      "step": 714
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489769820971867e-05,
+      "loss": 0.5932,
+      "step": 715
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489002557544757e-05,
+      "loss": 0.0371,
+      "step": 716
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.488235294117647e-05,
+      "loss": 1.0137,
+      "step": 717
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.4874680306905373e-05,
+      "loss": 0.9928,
+      "step": 718
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.486700767263427e-05,
+      "loss": 0.7431,
+      "step": 719
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4859335038363173e-05,
+      "loss": 0.0207,
+      "step": 720
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4851662404092074e-05,
+      "loss": 0.0681,
+      "step": 721
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4843989769820972e-05,
+      "loss": 0.0306,
+      "step": 722
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4836317135549874e-05,
+      "loss": 0.0204,
+      "step": 723
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4828644501278772e-05,
+      "loss": 0.0344,
+      "step": 724
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4820971867007674e-05,
+      "loss": 0.0345,
+      "step": 725
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4813299232736575e-05,
+      "loss": 0.9275,
+      "step": 726
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4805626598465474e-05,
+      "loss": 0.0196,
+      "step": 727
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4797953964194372e-05,
+      "loss": 1.0137,
+      "step": 728
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4790281329923277e-05,
+      "loss": 0.0318,
+      "step": 729
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4782608695652175e-05,
+      "loss": 0.0688,
+      "step": 730
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4774936061381073e-05,
+      "loss": 0.0314,
+      "step": 731
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4767263427109975e-05,
+      "loss": 0.7135,
+      "step": 732
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4759590792838877e-05,
+      "loss": 0.0299,
+      "step": 733
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4751918158567775e-05,
+      "loss": 1.0127,
+      "step": 734
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4744245524296676e-05,
+      "loss": 0.0304,
+      "step": 735
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4736572890025575e-05,
+      "loss": 1.4618,
+      "step": 736
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4728900255754476e-05,
+      "loss": 0.7167,
+      "step": 737
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4721227621483378e-05,
+      "loss": 0.0307,
+      "step": 738
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4713554987212276e-05,
+      "loss": 0.052,
+      "step": 739
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4705882352941174e-05,
+      "loss": 0.778,
+      "step": 740
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.469820971867008e-05,
+      "loss": 0.5482,
+      "step": 741
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4690537084398978e-05,
+      "loss": 1.4314,
+      "step": 742
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4682864450127876e-05,
+      "loss": 0.0422,
+      "step": 743
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.467519181585678e-05,
+      "loss": 0.6614,
+      "step": 744
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.466751918158568e-05,
+      "loss": 0.0312,
+      "step": 745
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4659846547314577e-05,
+      "loss": 0.043,
+      "step": 746
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.465217391304348e-05,
+      "loss": 0.0199,
+      "step": 747
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.464450127877238e-05,
+      "loss": 0.0832,
+      "step": 748
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.463682864450128e-05,
+      "loss": 0.0312,
+      "step": 749
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.462915601023018e-05,
+      "loss": 0.0309,
+      "step": 750
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.462148337595908e-05,
+      "loss": 0.0193,
+      "step": 751
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.461381074168798e-05,
+      "loss": 0.0446,
+      "step": 752
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.4606138107416882e-05,
+      "loss": 0.9993,
+      "step": 753
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.459846547314578e-05,
+      "loss": 0.016,
+      "step": 754
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4590792838874678e-05,
+      "loss": 0.0433,
+      "step": 755
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4583120204603583e-05,
+      "loss": 0.2614,
+      "step": 756
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.457544757033248e-05,
+      "loss": 0.018,
+      "step": 757
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.456777493606138e-05,
+      "loss": 0.6676,
+      "step": 758
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.456010230179028e-05,
+      "loss": 0.0599,
+      "step": 759
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4552429667519183e-05,
+      "loss": 0.0338,
+      "step": 760
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.454475703324808e-05,
+      "loss": 0.0268,
+      "step": 761
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4537084398976983e-05,
+      "loss": 0.0262,
+      "step": 762
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.452941176470588e-05,
+      "loss": 0.0252,
+      "step": 763
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4521739130434786e-05,
+      "loss": 0.0377,
+      "step": 764
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4514066496163684e-05,
+      "loss": 1.4866,
+      "step": 765
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4506393861892582e-05,
+      "loss": 1.0524,
+      "step": 766
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4498721227621484e-05,
+      "loss": 0.0328,
+      "step": 767
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4491048593350386e-05,
+      "loss": 0.024,
+      "step": 768
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4483375959079284e-05,
+      "loss": 0.022,
+      "step": 769
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4475703324808185e-05,
+      "loss": 0.0245,
+      "step": 770
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4468030690537087e-05,
+      "loss": 1.0107,
+      "step": 771
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4460358056265985e-05,
+      "loss": 0.0336,
+      "step": 772
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4452685421994887e-05,
+      "loss": 0.0391,
+      "step": 773
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4445012787723785e-05,
+      "loss": 0.0303,
+      "step": 774
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4437340153452687e-05,
+      "loss": 0.0443,
+      "step": 775
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4429667519181588e-05,
+      "loss": 0.0292,
+      "step": 776
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4421994884910487e-05,
+      "loss": 0.0284,
+      "step": 777
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4414322250639385e-05,
+      "loss": 1.5985,
+      "step": 778
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.440664961636829e-05,
+      "loss": 0.0114,
+      "step": 779
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4398976982097188e-05,
+      "loss": 0.0342,
+      "step": 780
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4391304347826086e-05,
+      "loss": 1.8501,
+      "step": 781
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4383631713554988e-05,
+      "loss": 0.5375,
+      "step": 782
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.437595907928389e-05,
+      "loss": 1.589,
+      "step": 783
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4368286445012788e-05,
+      "loss": 0.0221,
+      "step": 784
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.436061381074169e-05,
+      "loss": 0.0453,
+      "step": 785
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4352941176470587e-05,
+      "loss": 0.796,
+      "step": 786
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.434526854219949e-05,
+      "loss": 0.1249,
+      "step": 787
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.433759590792839e-05,
+      "loss": 0.7963,
+      "step": 788
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.432992327365729e-05,
+      "loss": 0.4231,
+      "step": 789
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.4322250639386187e-05,
+      "loss": 0.0247,
+      "step": 790
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.4314578005115092e-05,
+      "loss": 0.0295,
+      "step": 791
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.430690537084399e-05,
+      "loss": 0.0249,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.3537317216396332,
+      "eval_runtime": 58.1226,
+      "eval_samples_per_second": 6.813,
+      "eval_steps_per_second": 1.703,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.429923273657289e-05,
+      "loss": 0.0157,
+      "step": 793
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4291560102301794e-05,
+      "loss": 0.5222,
+      "step": 794
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4283887468030692e-05,
+      "loss": 0.0226,
+      "step": 795
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.427621483375959e-05,
+      "loss": 0.0293,
+      "step": 796
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.426854219948849e-05,
+      "loss": 0.015,
+      "step": 797
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4260869565217393e-05,
+      "loss": 1.0658,
+      "step": 798
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.425319693094629e-05,
+      "loss": 0.0412,
+      "step": 799
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4245524296675193e-05,
+      "loss": 0.0219,
+      "step": 800
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.423785166240409e-05,
+      "loss": 0.0149,
+      "step": 801
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4230179028132993e-05,
+      "loss": 0.0323,
+      "step": 802
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4222506393861895e-05,
+      "loss": 1.0764,
+      "step": 803
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4214833759590793e-05,
+      "loss": 0.7518,
+      "step": 804
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.420716112531969e-05,
+      "loss": 0.8166,
+      "step": 805
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4199488491048596e-05,
+      "loss": 0.0219,
+      "step": 806
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4191815856777494e-05,
+      "loss": 0.0154,
+      "step": 807
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4184143222506392e-05,
+      "loss": 0.0163,
+      "step": 808
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4176470588235294e-05,
+      "loss": 1.0562,
+      "step": 809
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4168797953964196e-05,
+      "loss": 0.0218,
+      "step": 810
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4161125319693094e-05,
+      "loss": 0.0232,
+      "step": 811
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4153452685421995e-05,
+      "loss": 0.2384,
+      "step": 812
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4145780051150894e-05,
+      "loss": 1.5286,
+      "step": 813
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4138107416879795e-05,
+      "loss": 0.0262,
+      "step": 814
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4130434782608697e-05,
+      "loss": 0.0334,
+      "step": 815
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4122762148337595e-05,
+      "loss": 0.6509,
+      "step": 816
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4115089514066497e-05,
+      "loss": 0.0159,
+      "step": 817
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.41074168797954e-05,
+      "loss": 0.6225,
+      "step": 818
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4099744245524297e-05,
+      "loss": 0.752,
+      "step": 819
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4092071611253198e-05,
+      "loss": 1.0594,
+      "step": 820
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.40843989769821e-05,
+      "loss": 0.0161,
+      "step": 821
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4076726342710998e-05,
+      "loss": 0.7492,
+      "step": 822
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40690537084399e-05,
+      "loss": 0.0302,
+      "step": 823
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4061381074168798e-05,
+      "loss": 0.038,
+      "step": 824
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40537084398977e-05,
+      "loss": 0.0243,
+      "step": 825
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40460358056266e-05,
+      "loss": 0.0331,
+      "step": 826
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40383631713555e-05,
+      "loss": 1.0252,
+      "step": 827
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4030690537084398e-05,
+      "loss": 0.0172,
+      "step": 828
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4023017902813303e-05,
+      "loss": 0.6168,
+      "step": 829
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40153452685422e-05,
+      "loss": 0.3135,
+      "step": 830
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40076726342711e-05,
+      "loss": 0.0168,
+      "step": 831
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.0172,
+      "step": 832
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.3992327365728902e-05,
+      "loss": 1.0349,
+      "step": 833
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39846547314578e-05,
+      "loss": 0.0978,
+      "step": 834
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3976982097186702e-05,
+      "loss": 0.0167,
+      "step": 835
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39693094629156e-05,
+      "loss": 1.0381,
+      "step": 836
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3961636828644502e-05,
+      "loss": 0.6857,
+      "step": 837
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3953964194373404e-05,
+      "loss": 1.0406,
+      "step": 838
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3946291560102302e-05,
+      "loss": 0.071,
+      "step": 839
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.39386189258312e-05,
+      "loss": 0.0193,
+      "step": 840
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3930946291560105e-05,
+      "loss": 0.0193,
+      "step": 841
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3923273657289003e-05,
+      "loss": 0.9916,
+      "step": 842
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.39156010230179e-05,
+      "loss": 0.0723,
+      "step": 843
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3907928388746806e-05,
+      "loss": 0.0626,
+      "step": 844
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3900255754475705e-05,
+      "loss": 0.7264,
+      "step": 845
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3892583120204603e-05,
+      "loss": 0.827,
+      "step": 846
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3884910485933504e-05,
+      "loss": 1.2193,
+      "step": 847
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3877237851662406e-05,
+      "loss": 0.6754,
+      "step": 848
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3869565217391304e-05,
+      "loss": 0.6856,
+      "step": 849
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3861892583120206e-05,
+      "loss": 0.7441,
+      "step": 850
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3854219948849104e-05,
+      "loss": 0.023,
+      "step": 851
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3846547314578006e-05,
+      "loss": 0.4742,
+      "step": 852
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3838874680306907e-05,
+      "loss": 0.0363,
+      "step": 853
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3831202046035806e-05,
+      "loss": 0.0409,
+      "step": 854
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3823529411764704e-05,
+      "loss": 0.8353,
+      "step": 855
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.381585677749361e-05,
+      "loss": 0.0333,
+      "step": 856
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3808184143222507e-05,
+      "loss": 0.0216,
+      "step": 857
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3800511508951405e-05,
+      "loss": 0.9688,
+      "step": 858
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3792838874680307e-05,
+      "loss": 1.0078,
+      "step": 859
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.378516624040921e-05,
+      "loss": 0.0183,
+      "step": 860
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3777493606138107e-05,
+      "loss": 0.0183,
+      "step": 861
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3769820971867008e-05,
+      "loss": 0.9786,
+      "step": 862
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3762148337595907e-05,
+      "loss": 0.9166,
+      "step": 863
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3754475703324808e-05,
+      "loss": 1.1408,
+      "step": 864
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.374680306905371e-05,
+      "loss": 1.1394,
+      "step": 865
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.3739130434782608e-05,
+      "loss": 0.0185,
+      "step": 866
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.373145780051151e-05,
+      "loss": 0.9979,
+      "step": 867
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.372378516624041e-05,
+      "loss": 1.0092,
+      "step": 868
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.371611253196931e-05,
+      "loss": 0.7276,
+      "step": 869
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.370843989769821e-05,
+      "loss": 0.0359,
+      "step": 870
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3700767263427113e-05,
+      "loss": 0.0864,
+      "step": 871
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.369309462915601e-05,
+      "loss": 0.0246,
+      "step": 872
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3685421994884912e-05,
+      "loss": 0.8518,
+      "step": 873
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.367774936061381e-05,
+      "loss": 1.433,
+      "step": 874
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3670076726342712e-05,
+      "loss": 0.035,
+      "step": 875
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3662404092071614e-05,
+      "loss": 0.5922,
+      "step": 876
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3654731457800512e-05,
+      "loss": 0.529,
+      "step": 877
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.364705882352941e-05,
+      "loss": 0.6423,
+      "step": 878
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3639386189258315e-05,
+      "loss": 0.0337,
+      "step": 879
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3631713554987214e-05,
+      "loss": 0.0402,
+      "step": 880
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3624040920716112e-05,
+      "loss": 0.0782,
+      "step": 881
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3616368286445013e-05,
+      "loss": 0.4141,
+      "step": 882
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3608695652173915e-05,
+      "loss": 0.0359,
+      "step": 883
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3601023017902813e-05,
+      "loss": 0.4509,
+      "step": 884
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3593350383631715e-05,
+      "loss": 0.6414,
+      "step": 885
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3585677749360613e-05,
+      "loss": 0.6873,
+      "step": 886
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3578005115089515e-05,
+      "loss": 0.0304,
+      "step": 887
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3570332480818416e-05,
+      "loss": 0.0315,
+      "step": 888
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3562659846547315e-05,
+      "loss": 0.0896,
+      "step": 889
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3554987212276213e-05,
+      "loss": 0.0201,
+      "step": 890
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3547314578005118e-05,
+      "loss": 0.027,
+      "step": 891
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3539641943734016e-05,
+      "loss": 1.9853,
+      "step": 892
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3531969309462914e-05,
+      "loss": 2.9647,
+      "step": 893
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.352429667519182e-05,
+      "loss": 1.9403,
+      "step": 894
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3516624040920717e-05,
+      "loss": 0.0193,
+      "step": 895
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3508951406649616e-05,
+      "loss": 0.9912,
+      "step": 896
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3501278772378517e-05,
+      "loss": 0.02,
+      "step": 897
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.349360613810742e-05,
+      "loss": 0.9015,
+      "step": 898
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3485933503836317e-05,
+      "loss": 0.1294,
+      "step": 899
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.347826086956522e-05,
+      "loss": 0.0559,
+      "step": 900
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.3470588235294117e-05,
+      "loss": 1.8039,
+      "step": 901
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.346291560102302e-05,
+      "loss": 0.0336,
+      "step": 902
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.345524296675192e-05,
+      "loss": 0.0383,
+      "step": 903
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.344757033248082e-05,
+      "loss": 0.0722,
+      "step": 904
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3439897698209717e-05,
+      "loss": 0.0278,
+      "step": 905
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.343222506393862e-05,
+      "loss": 0.0684,
+      "step": 906
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.342455242966752e-05,
+      "loss": 0.0236,
+      "step": 907
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3416879795396418e-05,
+      "loss": 0.0348,
+      "step": 908
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340920716112532e-05,
+      "loss": 0.0204,
+      "step": 909
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340153452685422e-05,
+      "loss": 0.8109,
+      "step": 910
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.339386189258312e-05,
+      "loss": 0.0279,
+      "step": 911
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.338618925831202e-05,
+      "loss": 0.0361,
+      "step": 912
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337851662404092e-05,
+      "loss": 0.0355,
+      "step": 913
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337084398976982e-05,
+      "loss": 0.7958,
+      "step": 914
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.3363171355498723e-05,
+      "loss": 0.0362,
+      "step": 915
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.335549872122762e-05,
+      "loss": 0.0252,
+      "step": 916
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3347826086956522e-05,
+      "loss": 0.8165,
+      "step": 917
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3340153452685424e-05,
+      "loss": 0.0237,
+      "step": 918
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3332480818414322e-05,
+      "loss": 0.8162,
+      "step": 919
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.332480818414322e-05,
+      "loss": 0.7786,
+      "step": 920
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3317135549872125e-05,
+      "loss": 0.0787,
+      "step": 921
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3309462915601024e-05,
+      "loss": 0.1764,
+      "step": 922
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3301790281329925e-05,
+      "loss": 0.7449,
+      "step": 923
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3294117647058824e-05,
+      "loss": 0.0452,
+      "step": 924
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3286445012787725e-05,
+      "loss": 0.0181,
+      "step": 925
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3278772378516627e-05,
+      "loss": 0.1783,
+      "step": 926
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3271099744245525e-05,
+      "loss": 0.5158,
+      "step": 927
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3263427109974423e-05,
+      "loss": 0.7574,
+      "step": 928
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3255754475703328e-05,
+      "loss": 2.6099,
+      "step": 929
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3248081841432226e-05,
+      "loss": 0.0304,
+      "step": 930
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3240409207161125e-05,
+      "loss": 0.0795,
+      "step": 931
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3232736572890026e-05,
+      "loss": 0.0342,
+      "step": 932
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3225063938618928e-05,
+      "loss": 0.0473,
+      "step": 933
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3217391304347826e-05,
+      "loss": 0.0831,
+      "step": 934
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3209718670076728e-05,
+      "loss": 0.0465,
+      "step": 935
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3202046035805626e-05,
+      "loss": 0.6769,
+      "step": 936
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3194373401534528e-05,
+      "loss": 0.0173,
+      "step": 937
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.318670076726343e-05,
+      "loss": 0.0169,
+      "step": 938
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3179028132992327e-05,
+      "loss": 0.0469,
+      "step": 939
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3171355498721226e-05,
+      "loss": 0.048,
+      "step": 940
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.316368286445013e-05,
+      "loss": 1.0704,
+      "step": 941
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.315601023017903e-05,
+      "loss": 1.0738,
+      "step": 942
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3148337595907927e-05,
+      "loss": 0.0163,
+      "step": 943
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3140664961636832e-05,
+      "loss": 2.0702,
+      "step": 944
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.313299232736573e-05,
+      "loss": 1.0433,
+      "step": 945
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.312531969309463e-05,
+      "loss": 1.0351,
+      "step": 946
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.311764705882353e-05,
+      "loss": 1.053,
+      "step": 947
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.3109974424552432e-05,
+      "loss": 0.1797,
+      "step": 948
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.310230179028133e-05,
+      "loss": 1.0259,
+      "step": 949
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.309462915601023e-05,
+      "loss": 0.03,
+      "step": 950
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.308695652173913e-05,
+      "loss": 0.905,
+      "step": 951
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.307928388746803e-05,
+      "loss": 0.0699,
+      "step": 952
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3071611253196933e-05,
+      "loss": 1.0013,
+      "step": 953
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.306393861892583e-05,
+      "loss": 0.0319,
+      "step": 954
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.305626598465473e-05,
+      "loss": 0.0361,
+      "step": 955
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3048593350383634e-05,
+      "loss": 0.0201,
+      "step": 956
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3040920716112533e-05,
+      "loss": 0.033,
+      "step": 957
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.303324808184143e-05,
+      "loss": 0.0411,
+      "step": 958
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3025575447570332e-05,
+      "loss": 0.7138,
+      "step": 959
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3017902813299234e-05,
+      "loss": 0.0434,
+      "step": 960
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3010230179028132e-05,
+      "loss": 0.8445,
+      "step": 961
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3002557544757034e-05,
+      "loss": 0.0323,
+      "step": 962
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2994884910485932e-05,
+      "loss": 0.0203,
+      "step": 963
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2987212276214834e-05,
+      "loss": 0.2302,
+      "step": 964
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2979539641943735e-05,
+      "loss": 0.0293,
+      "step": 965
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2971867007672634e-05,
+      "loss": 0.0593,
+      "step": 966
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2964194373401535e-05,
+      "loss": 0.6505,
+      "step": 967
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2956521739130437e-05,
+      "loss": 0.5891,
+      "step": 968
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2948849104859335e-05,
+      "loss": 0.0191,
+      "step": 969
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2941176470588233e-05,
+      "loss": 0.0188,
+      "step": 970
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2933503836317138e-05,
+      "loss": 0.9537,
+      "step": 971
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2925831202046037e-05,
+      "loss": 0.0186,
+      "step": 972
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2918158567774935e-05,
+      "loss": 1.0237,
+      "step": 973
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2910485933503836e-05,
+      "loss": 0.0417,
+      "step": 974
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2902813299232738e-05,
+      "loss": 0.0369,
+      "step": 975
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.289514066496164e-05,
+      "loss": 0.0184,
+      "step": 976
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2887468030690538e-05,
+      "loss": 0.811,
+      "step": 977
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2879795396419436e-05,
+      "loss": 0.0183,
+      "step": 978
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.287212276214834e-05,
+      "loss": 1.0442,
+      "step": 979
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.286445012787724e-05,
+      "loss": 0.027,
+      "step": 980
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.2856777493606137e-05,
+      "loss": 0.062,
+      "step": 981
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284910485933504e-05,
+      "loss": 0.0186,
+      "step": 982
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284143222506394e-05,
+      "loss": 0.6057,
+      "step": 983
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.283375959079284e-05,
+      "loss": 1.6594,
+      "step": 984
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.282608695652174e-05,
+      "loss": 0.0192,
+      "step": 985
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281841432225064e-05,
+      "loss": 1.5361,
+      "step": 986
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281074168797954e-05,
+      "loss": 0.0384,
+      "step": 987
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.2803069053708442e-05,
+      "loss": 2.0238,
+      "step": 988
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.279539641943734e-05,
+      "loss": 0.0334,
+      "step": 989
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2787723785166242e-05,
+      "loss": 0.0571,
+      "step": 990
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2780051150895143e-05,
+      "loss": 0.0211,
+      "step": 991
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.277237851662404e-05,
+      "loss": 0.0409,
+      "step": 992
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.276470588235294e-05,
+      "loss": 0.1059,
+      "step": 993
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2757033248081845e-05,
+      "loss": 0.0277,
+      "step": 994
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2749360613810743e-05,
+      "loss": 1.0023,
+      "step": 995
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.274168797953964e-05,
+      "loss": 0.0186,
+      "step": 996
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2734015345268543e-05,
+      "loss": 0.0345,
+      "step": 997
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2726342710997445e-05,
+      "loss": 0.0169,
+      "step": 998
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2718670076726343e-05,
+      "loss": 0.7292,
+      "step": 999
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2710997442455244e-05,
+      "loss": 1.2655,
+      "step": 1000
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2703324808184143e-05,
+      "loss": 0.03,
+      "step": 1001
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2695652173913044e-05,
+      "loss": 1.0266,
+      "step": 1002
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2687979539641946e-05,
+      "loss": 0.7645,
+      "step": 1003
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2680306905370844e-05,
+      "loss": 0.7985,
+      "step": 1004
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2672634271099742e-05,
+      "loss": 0.76,
+      "step": 1005
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2664961636828647e-05,
+      "loss": 0.7577,
+      "step": 1006
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2657289002557545e-05,
+      "loss": 0.0248,
+      "step": 1007
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2649616368286444e-05,
+      "loss": 0.0262,
+      "step": 1008
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2641943734015345e-05,
+      "loss": 1.9984,
+      "step": 1009
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2634271099744247e-05,
+      "loss": 0.1489,
+      "step": 1010
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2626598465473145e-05,
+      "loss": 0.0683,
+      "step": 1011
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2618925831202047e-05,
+      "loss": 0.054,
+      "step": 1012
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2611253196930945e-05,
+      "loss": 0.0269,
+      "step": 1013
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2603580562659847e-05,
+      "loss": 1.0378,
+      "step": 1014
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2595907928388748e-05,
+      "loss": 0.0171,
+      "step": 1015
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2588235294117646e-05,
+      "loss": 0.8433,
+      "step": 1016
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2580562659846548e-05,
+      "loss": 0.0468,
+      "step": 1017
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.257289002557545e-05,
+      "loss": 0.0331,
+      "step": 1018
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2565217391304348e-05,
+      "loss": 0.0308,
+      "step": 1019
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2557544757033246e-05,
+      "loss": 1.0007,
+      "step": 1020
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254987212276215e-05,
+      "loss": 2.0242,
+      "step": 1021
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254219948849105e-05,
+      "loss": 0.7713,
+      "step": 1022
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2534526854219948e-05,
+      "loss": 1.1112,
+      "step": 1023
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.252685421994885e-05,
+      "loss": 0.0184,
+      "step": 1024
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251918158567775e-05,
+      "loss": 0.0243,
+      "step": 1025
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251150895140665e-05,
+      "loss": 0.046,
+      "step": 1026
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.250383631713555e-05,
+      "loss": 0.0378,
+      "step": 1027
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.249616368286445e-05,
+      "loss": 0.0329,
+      "step": 1028
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2488491048593354e-05,
+      "loss": 0.0245,
+      "step": 1029
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2480818414322252e-05,
+      "loss": 0.0238,
+      "step": 1030
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.247314578005115e-05,
+      "loss": 1.0628,
+      "step": 1031
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2465473145780052e-05,
+      "loss": 0.0703,
+      "step": 1032
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2457800511508954e-05,
+      "loss": 0.0251,
+      "step": 1033
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2450127877237852e-05,
+      "loss": 1.0561,
+      "step": 1034
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2442455242966753e-05,
+      "loss": 0.0274,
+      "step": 1035
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.243478260869565e-05,
+      "loss": 0.023,
+      "step": 1036
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2427109974424553e-05,
+      "loss": 0.0223,
+      "step": 1037
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2419437340153455e-05,
+      "loss": 0.8113,
+      "step": 1038
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2411764705882353e-05,
+      "loss": 0.0346,
+      "step": 1039
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2404092071611255e-05,
+      "loss": 0.0164,
+      "step": 1040
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2396419437340156e-05,
+      "loss": 0.0237,
+      "step": 1041
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2388746803069054e-05,
+      "loss": 0.0305,
+      "step": 1042
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2381074168797953e-05,
+      "loss": 0.0158,
+      "step": 1043
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2373401534526858e-05,
+      "loss": 0.0225,
+      "step": 1044
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2365728900255756e-05,
+      "loss": 0.0226,
+      "step": 1045
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2358056265984654e-05,
+      "loss": 0.015,
+      "step": 1046
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2350383631713556e-05,
+      "loss": 0.0219,
+      "step": 1047
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2342710997442457e-05,
+      "loss": 1.0694,
+      "step": 1048
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2335038363171356e-05,
+      "loss": 0.0348,
+      "step": 1049
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2327365728900257e-05,
+      "loss": 0.0213,
+      "step": 1050
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2319693094629155e-05,
+      "loss": 0.9405,
+      "step": 1051
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2312020460358057e-05,
+      "loss": 0.0252,
+      "step": 1052
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.230434782608696e-05,
+      "loss": 0.0254,
+      "step": 1053
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2296675191815857e-05,
+      "loss": 0.7353,
+      "step": 1054
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2289002557544755e-05,
+      "loss": 0.0253,
+      "step": 1055
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.228132992327366e-05,
+      "loss": 0.8361,
+      "step": 1056
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2273657289002558e-05,
+      "loss": 0.0196,
+      "step": 1057
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2265984654731457e-05,
+      "loss": 0.014,
+      "step": 1058
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2258312020460358e-05,
+      "loss": 0.8297,
+      "step": 1059
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.225063938618926e-05,
+      "loss": 1.0654,
+      "step": 1060
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2242966751918158e-05,
+      "loss": 0.0248,
+      "step": 1061
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.223529411764706e-05,
+      "loss": 0.0226,
+      "step": 1062
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2227621483375958e-05,
+      "loss": 0.0238,
+      "step": 1063
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221994884910486e-05,
+      "loss": 0.0212,
+      "step": 1064
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221227621483376e-05,
+      "loss": 0.0274,
+      "step": 1065
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.220460358056266e-05,
+      "loss": 1.0217,
+      "step": 1066
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.219693094629156e-05,
+      "loss": 0.0252,
+      "step": 1067
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2189258312020462e-05,
+      "loss": 0.0219,
+      "step": 1068
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.218158567774936e-05,
+      "loss": 0.0349,
+      "step": 1069
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.217391304347826e-05,
+      "loss": 0.0152,
+      "step": 1070
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2166240409207164e-05,
+      "loss": 0.021,
+      "step": 1071
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2158567774936062e-05,
+      "loss": 0.015,
+      "step": 1072
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.215089514066496e-05,
+      "loss": 0.0277,
+      "step": 1073
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2143222506393862e-05,
+      "loss": 0.1858,
+      "step": 1074
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2135549872122764e-05,
+      "loss": 0.0136,
+      "step": 1075
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2127877237851662e-05,
+      "loss": 0.0132,
+      "step": 1076
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2120204603580563e-05,
+      "loss": 0.0126,
+      "step": 1077
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.211253196930946e-05,
+      "loss": 0.5171,
+      "step": 1078
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2104859335038363e-05,
+      "loss": 0.0422,
+      "step": 1079
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2097186700767265e-05,
+      "loss": 0.0186,
+      "step": 1080
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2089514066496163e-05,
+      "loss": 0.8412,
+      "step": 1081
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2081841432225065e-05,
+      "loss": 0.0183,
+      "step": 1082
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2074168797953966e-05,
+      "loss": 1.0243,
+      "step": 1083
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2066496163682865e-05,
+      "loss": 0.0212,
+      "step": 1084
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2058823529411766e-05,
+      "loss": 0.2841,
+      "step": 1085
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2051150895140664e-05,
+      "loss": 0.0118,
+      "step": 1086
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2043478260869566e-05,
+      "loss": 0.0247,
+      "step": 1087
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2035805626598468e-05,
+      "loss": 1.5121,
+      "step": 1088
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2028132992327366e-05,
+      "loss": 0.0304,
+      "step": 1089
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2020460358056267e-05,
+      "loss": 0.0329,
+      "step": 1090
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.201278772378517e-05,
+      "loss": 1.1288,
+      "step": 1091
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.2005115089514067e-05,
+      "loss": 0.0145,
+      "step": 1092
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.1997442455242966e-05,
+      "loss": 0.0498,
+      "step": 1093
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.198976982097187e-05,
+      "loss": 0.0244,
+      "step": 1094
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.198209718670077e-05,
+      "loss": 0.0112,
+      "step": 1095
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.1974424552429667e-05,
+      "loss": 0.8642,
+      "step": 1096
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.196675191815857e-05,
+      "loss": 0.8822,
+      "step": 1097
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.195907928388747e-05,
+      "loss": 0.0278,
+      "step": 1098
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.195140664961637e-05,
+      "loss": 0.6954,
+      "step": 1099
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.194373401534527e-05,
+      "loss": 0.0179,
+      "step": 1100
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.1936061381074168e-05,
+      "loss": 0.5607,
+      "step": 1101
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.192838874680307e-05,
+      "loss": 0.0191,
+      "step": 1102
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.192071611253197e-05,
+      "loss": 1.9519,
+      "step": 1103
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.191304347826087e-05,
+      "loss": 0.8347,
+      "step": 1104
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1905370843989768e-05,
+      "loss": 0.0262,
+      "step": 1105
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1897698209718673e-05,
+      "loss": 0.0477,
+      "step": 1106
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.189002557544757e-05,
+      "loss": 0.0227,
+      "step": 1107
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.188235294117647e-05,
+      "loss": 0.0205,
+      "step": 1108
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.187468030690537e-05,
+      "loss": 0.0208,
+      "step": 1109
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.1867007672634273e-05,
+      "loss": 0.0116,
+      "step": 1110
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.185933503836317e-05,
+      "loss": 1.1546,
+      "step": 1111
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1851662404092072e-05,
+      "loss": 0.0114,
+      "step": 1112
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.184398976982097e-05,
+      "loss": 0.0113,
+      "step": 1113
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1836317135549872e-05,
+      "loss": 0.6881,
+      "step": 1114
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1828644501278774e-05,
+      "loss": 0.0294,
+      "step": 1115
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1820971867007672e-05,
+      "loss": 0.1569,
+      "step": 1116
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1813299232736574e-05,
+      "loss": 1.1362,
+      "step": 1117
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1805626598465475e-05,
+      "loss": 0.0196,
+      "step": 1118
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1797953964194374e-05,
+      "loss": 0.0205,
+      "step": 1119
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1790281329923272e-05,
+      "loss": 0.0198,
+      "step": 1120
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1782608695652177e-05,
+      "loss": 0.0113,
+      "step": 1121
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1774936061381075e-05,
+      "loss": 0.02,
+      "step": 1122
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1767263427109973e-05,
+      "loss": 0.0195,
+      "step": 1123
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1759590792838875e-05,
+      "loss": 0.8016,
+      "step": 1124
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1751918158567776e-05,
+      "loss": 0.0122,
+      "step": 1125
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1744245524296675e-05,
+      "loss": 1.1132,
+      "step": 1126
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1736572890025576e-05,
+      "loss": 0.011,
+      "step": 1127
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1728900255754474e-05,
+      "loss": 0.7705,
+      "step": 1128
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1721227621483376e-05,
+      "loss": 1.1467,
+      "step": 1129
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1713554987212278e-05,
+      "loss": 0.0176,
+      "step": 1130
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1705882352941176e-05,
+      "loss": 0.0293,
+      "step": 1131
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1698209718670074e-05,
+      "loss": 0.0208,
+      "step": 1132
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.169053708439898e-05,
+      "loss": 0.0199,
+      "step": 1133
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1682864450127877e-05,
+      "loss": 0.0431,
+      "step": 1134
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.167519181585678e-05,
+      "loss": 1.6066,
+      "step": 1135
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.1667519181585677e-05,
+      "loss": 0.029,
+      "step": 1136
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165984654731458e-05,
+      "loss": 0.029,
+      "step": 1137
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165217391304348e-05,
+      "loss": 1.1749,
+      "step": 1138
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.164450127877238e-05,
+      "loss": 0.0189,
+      "step": 1139
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.163682864450128e-05,
+      "loss": 0.0309,
+      "step": 1140
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.1629156010230182e-05,
+      "loss": 0.0287,
+      "step": 1141
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.162148337595908e-05,
+      "loss": 0.0202,
+      "step": 1142
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.161381074168798e-05,
+      "loss": 0.7889,
+      "step": 1143
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.1606138107416883e-05,
+      "loss": 0.0285,
+      "step": 1144
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159846547314578e-05,
+      "loss": 0.0535,
+      "step": 1145
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159079283887468e-05,
+      "loss": 1.0829,
+      "step": 1146
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.158312020460358e-05,
+      "loss": 0.0989,
+      "step": 1147
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1575447570332483e-05,
+      "loss": 0.0115,
+      "step": 1148
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.156777493606138e-05,
+      "loss": 0.0227,
+      "step": 1149
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1560102301790283e-05,
+      "loss": 0.5506,
+      "step": 1150
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.155242966751918e-05,
+      "loss": 1.1234,
+      "step": 1151
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1544757033248083e-05,
+      "loss": 0.0564,
+      "step": 1152
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1537084398976984e-05,
+      "loss": 0.0127,
+      "step": 1153
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1529411764705882e-05,
+      "loss": 0.0218,
+      "step": 1154
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.152173913043478e-05,
+      "loss": 0.0124,
+      "step": 1155
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1514066496163686e-05,
+      "loss": 0.0223,
+      "step": 1156
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1506393861892584e-05,
+      "loss": 0.0505,
+      "step": 1157
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1498721227621482e-05,
+      "loss": 0.0225,
+      "step": 1158
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1491048593350384e-05,
+      "loss": 0.0308,
+      "step": 1159
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1483375959079285e-05,
+      "loss": 0.036,
+      "step": 1160
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1475703324808184e-05,
+      "loss": 0.0121,
+      "step": 1161
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1468030690537085e-05,
+      "loss": 0.0274,
+      "step": 1162
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1460358056265983e-05,
+      "loss": 1.0397,
+      "step": 1163
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1452685421994885e-05,
+      "loss": 0.0332,
+      "step": 1164
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1445012787723787e-05,
+      "loss": 0.0182,
+      "step": 1165
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1437340153452685e-05,
+      "loss": 0.018,
+      "step": 1166
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1429667519181587e-05,
+      "loss": 0.859,
+      "step": 1167
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1421994884910488e-05,
+      "loss": 0.0321,
+      "step": 1168
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1414322250639386e-05,
+      "loss": 0.0225,
+      "step": 1169
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1406649616368285e-05,
+      "loss": 0.0224,
+      "step": 1170
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.139897698209719e-05,
+      "loss": 0.03,
+      "step": 1171
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1391304347826088e-05,
+      "loss": 0.7473,
+      "step": 1172
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1383631713554986e-05,
+      "loss": 0.8609,
+      "step": 1173
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1375959079283888e-05,
+      "loss": 0.0105,
+      "step": 1174
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.136828644501279e-05,
+      "loss": 0.0217,
+      "step": 1175
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1360613810741687e-05,
+      "loss": 0.0218,
+      "step": 1176
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.135294117647059e-05,
+      "loss": 0.016,
+      "step": 1177
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1345268542199487e-05,
+      "loss": 0.7844,
+      "step": 1178
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.133759590792839e-05,
+      "loss": 0.01,
+      "step": 1179
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132992327365729e-05,
+      "loss": 0.8774,
+      "step": 1180
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132225063938619e-05,
+      "loss": 0.8375,
+      "step": 1181
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.1314578005115087e-05,
+      "loss": 0.0188,
+      "step": 1182
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.1306905370843992e-05,
+      "loss": 1.1026,
+      "step": 1183
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129923273657289e-05,
+      "loss": 1.0447,
+      "step": 1184
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129156010230179e-05,
+      "loss": 0.0101,
+      "step": 1185
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.128388746803069e-05,
+      "loss": 0.0247,
+      "step": 1186
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.127621483375959e-05,
+      "loss": 0.0295,
+      "step": 1187
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.1268542199488493e-05,
+      "loss": 0.0191,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.37390685081481934,
+      "eval_runtime": 57.4087,
+      "eval_samples_per_second": 6.898,
+      "eval_steps_per_second": 1.724,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.126086956521739e-05,
+      "loss": 0.0103,
+      "step": 1189
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1253196930946293e-05,
+      "loss": 0.0231,
+      "step": 1190
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1245524296675195e-05,
+      "loss": 0.0389,
+      "step": 1191
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1237851662404093e-05,
+      "loss": 0.0298,
+      "step": 1192
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.123017902813299e-05,
+      "loss": 0.0196,
+      "step": 1193
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1222506393861896e-05,
+      "loss": 0.9688,
+      "step": 1194
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1214833759590794e-05,
+      "loss": 0.0458,
+      "step": 1195
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1207161125319693e-05,
+      "loss": 0.0161,
+      "step": 1196
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1199488491048594e-05,
+      "loss": 0.0157,
+      "step": 1197
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1191815856777496e-05,
+      "loss": 0.8732,
+      "step": 1198
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1184143222506394e-05,
+      "loss": 0.0103,
+      "step": 1199
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1176470588235296e-05,
+      "loss": 0.0267,
+      "step": 1200
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1168797953964194e-05,
+      "loss": 0.0266,
+      "step": 1201
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1161125319693095e-05,
+      "loss": 0.0157,
+      "step": 1202
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1153452685421997e-05,
+      "loss": 0.265,
+      "step": 1203
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1145780051150895e-05,
+      "loss": 0.0537,
+      "step": 1204
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1138107416879794e-05,
+      "loss": 0.0306,
+      "step": 1205
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.11304347826087e-05,
+      "loss": 1.1714,
+      "step": 1206
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1122762148337597e-05,
+      "loss": 1.1704,
+      "step": 1207
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1115089514066495e-05,
+      "loss": 0.0154,
+      "step": 1208
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1107416879795397e-05,
+      "loss": 0.0195,
+      "step": 1209
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1099744245524298e-05,
+      "loss": 0.9099,
+      "step": 1210
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1092071611253196e-05,
+      "loss": 0.0112,
+      "step": 1211
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1084398976982098e-05,
+      "loss": 0.1506,
+      "step": 1212
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1076726342710996e-05,
+      "loss": 0.0111,
+      "step": 1213
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1069053708439898e-05,
+      "loss": 0.0244,
+      "step": 1214
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10613810741688e-05,
+      "loss": 0.8889,
+      "step": 1215
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1053708439897698e-05,
+      "loss": 0.0146,
+      "step": 1216
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10460358056266e-05,
+      "loss": 0.0189,
+      "step": 1217
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10383631713555e-05,
+      "loss": 0.0097,
+      "step": 1218
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10306905370844e-05,
+      "loss": 1.1683,
+      "step": 1219
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1023017902813297e-05,
+      "loss": 1.1621,
+      "step": 1220
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1015345268542202e-05,
+      "loss": 0.0194,
+      "step": 1221
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.10076726342711e-05,
+      "loss": 0.0155,
+      "step": 1222
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.0296,
+      "step": 1223
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.09923273657289e-05,
+      "loss": 0.02,
+      "step": 1224
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.0984654731457802e-05,
+      "loss": 0.0154,
+      "step": 1225
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09769820971867e-05,
+      "loss": 0.0151,
+      "step": 1226
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0969309462915602e-05,
+      "loss": 0.0223,
+      "step": 1227
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09616368286445e-05,
+      "loss": 1.7159,
+      "step": 1228
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0953964194373402e-05,
+      "loss": 0.0194,
+      "step": 1229
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0946291560102303e-05,
+      "loss": 0.0637,
+      "step": 1230
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09386189258312e-05,
+      "loss": 0.0156,
+      "step": 1231
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09309462915601e-05,
+      "loss": 1.0612,
+      "step": 1232
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0923273657289005e-05,
+      "loss": 0.8929,
+      "step": 1233
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0915601023017903e-05,
+      "loss": 1.1649,
+      "step": 1234
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.09079283887468e-05,
+      "loss": 2.059,
+      "step": 1235
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0900255754475703e-05,
+      "loss": 0.0169,
+      "step": 1236
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0892583120204604e-05,
+      "loss": 0.0158,
+      "step": 1237
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0884910485933503e-05,
+      "loss": 0.0108,
+      "step": 1238
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0877237851662404e-05,
+      "loss": 0.0152,
+      "step": 1239
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0869565217391306e-05,
+      "loss": 0.0109,
+      "step": 1240
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0861892583120208e-05,
+      "loss": 1.1352,
+      "step": 1241
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0854219948849106e-05,
+      "loss": 0.0226,
+      "step": 1242
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0846547314578004e-05,
+      "loss": 0.0154,
+      "step": 1243
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.083887468030691e-05,
+      "loss": 0.0232,
+      "step": 1244
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0831202046035807e-05,
+      "loss": 0.0142,
+      "step": 1245
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0823529411764705e-05,
+      "loss": 0.9143,
+      "step": 1246
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0815856777493607e-05,
+      "loss": 0.0165,
+      "step": 1247
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.080818414322251e-05,
+      "loss": 0.0446,
+      "step": 1248
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0800511508951407e-05,
+      "loss": 0.0121,
+      "step": 1249
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.079283887468031e-05,
+      "loss": 1.0991,
+      "step": 1250
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.0785166240409207e-05,
+      "loss": 0.0159,
+      "step": 1251
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.077749360613811e-05,
+      "loss": 0.0166,
+      "step": 1252
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.076982097186701e-05,
+      "loss": 0.0171,
+      "step": 1253
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0762148337595908e-05,
+      "loss": 0.7986,
+      "step": 1254
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0754475703324806e-05,
+      "loss": 0.0174,
+      "step": 1255
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.074680306905371e-05,
+      "loss": 1.0431,
+      "step": 1256
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.073913043478261e-05,
+      "loss": 0.1766,
+      "step": 1257
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.0731457800511508e-05,
+      "loss": 0.0209,
+      "step": 1258
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.072378516624041e-05,
+      "loss": 0.0178,
+      "step": 1259
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.071611253196931e-05,
+      "loss": 0.0215,
+      "step": 1260
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.070843989769821e-05,
+      "loss": 0.0112,
+      "step": 1261
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.070076726342711e-05,
+      "loss": 0.0166,
+      "step": 1262
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0693094629156012e-05,
+      "loss": 0.0161,
+      "step": 1263
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.068542199488491e-05,
+      "loss": 0.8922,
+      "step": 1264
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0677749360613812e-05,
+      "loss": 0.0133,
+      "step": 1265
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.067007672634271e-05,
+      "loss": 0.0117,
+      "step": 1266
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0662404092071612e-05,
+      "loss": 0.0153,
+      "step": 1267
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0654731457800514e-05,
+      "loss": 0.0153,
+      "step": 1268
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0647058823529412e-05,
+      "loss": 0.0195,
+      "step": 1269
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.063938618925831e-05,
+      "loss": 1.1511,
+      "step": 1270
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0631713554987215e-05,
+      "loss": 0.0108,
+      "step": 1271
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0624040920716113e-05,
+      "loss": 0.8897,
+      "step": 1272
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.061636828644501e-05,
+      "loss": 0.0157,
+      "step": 1273
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0608695652173913e-05,
+      "loss": 0.0108,
+      "step": 1274
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0601023017902815e-05,
+      "loss": 0.0214,
+      "step": 1275
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0593350383631713e-05,
+      "loss": 0.0186,
+      "step": 1276
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0585677749360615e-05,
+      "loss": 0.0182,
+      "step": 1277
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0578005115089513e-05,
+      "loss": 0.0102,
+      "step": 1278
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0570332480818415e-05,
+      "loss": 1.1313,
+      "step": 1279
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0562659846547316e-05,
+      "loss": 0.0103,
+      "step": 1280
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0554987212276214e-05,
+      "loss": 0.0144,
+      "step": 1281
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0547314578005113e-05,
+      "loss": 0.02,
+      "step": 1282
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0539641943734018e-05,
+      "loss": 0.0292,
+      "step": 1283
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0531969309462916e-05,
+      "loss": 0.0105,
+      "step": 1284
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0524296675191814e-05,
+      "loss": 0.01,
+      "step": 1285
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0516624040920716e-05,
+      "loss": 1.1245,
+      "step": 1286
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0508951406649617e-05,
+      "loss": 0.8544,
+      "step": 1287
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0501278772378516e-05,
+      "loss": 0.0149,
+      "step": 1288
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0493606138107417e-05,
+      "loss": 0.0065,
+      "step": 1289
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.048593350383632e-05,
+      "loss": 0.0113,
+      "step": 1290
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0478260869565217e-05,
+      "loss": 0.0094,
+      "step": 1291
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.047058823529412e-05,
+      "loss": 0.0097,
+      "step": 1292
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0462915601023017e-05,
+      "loss": 0.0812,
+      "step": 1293
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0455242966751922e-05,
+      "loss": 0.0099,
+      "step": 1294
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.044757033248082e-05,
+      "loss": 0.013,
+      "step": 1295
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0439897698209718e-05,
+      "loss": 1.584,
+      "step": 1296
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.043222506393862e-05,
+      "loss": 0.0089,
+      "step": 1297
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.042455242966752e-05,
+      "loss": 0.0089,
+      "step": 1298
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.041687979539642e-05,
+      "loss": 0.0093,
+      "step": 1299
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.040920716112532e-05,
+      "loss": 0.0149,
+      "step": 1300
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.040153452685422e-05,
+      "loss": 1.1671,
+      "step": 1301
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.039386189258312e-05,
+      "loss": 0.0086,
+      "step": 1302
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.0386189258312023e-05,
+      "loss": 0.8177,
+      "step": 1303
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.037851662404092e-05,
+      "loss": 0.0128,
+      "step": 1304
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.037084398976982e-05,
+      "loss": 0.0167,
+      "step": 1305
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0363171355498724e-05,
+      "loss": 0.0085,
+      "step": 1306
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0355498721227622e-05,
+      "loss": 0.0085,
+      "step": 1307
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.034782608695652e-05,
+      "loss": 0.0166,
+      "step": 1308
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0340153452685422e-05,
+      "loss": 0.9448,
+      "step": 1309
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0332480818414324e-05,
+      "loss": 0.0082,
+      "step": 1310
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0324808184143222e-05,
+      "loss": 0.0088,
+      "step": 1311
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0317135549872124e-05,
+      "loss": 0.0082,
+      "step": 1312
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0309462915601025e-05,
+      "loss": 0.0146,
+      "step": 1313
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0301790281329924e-05,
+      "loss": 0.0123,
+      "step": 1314
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0294117647058825e-05,
+      "loss": 0.0101,
+      "step": 1315
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0286445012787723e-05,
+      "loss": 1.2167,
+      "step": 1316
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0278772378516625e-05,
+      "loss": 1.0578,
+      "step": 1317
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0271099744245527e-05,
+      "loss": 0.0163,
+      "step": 1318
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0263427109974425e-05,
+      "loss": 0.013,
+      "step": 1319
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0255754475703323e-05,
+      "loss": 0.0082,
+      "step": 1320
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0248081841432228e-05,
+      "loss": 0.0081,
+      "step": 1321
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0240409207161126e-05,
+      "loss": 0.0124,
+      "step": 1322
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0232736572890024e-05,
+      "loss": 0.0206,
+      "step": 1323
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0225063938618926e-05,
+      "loss": 0.0081,
+      "step": 1324
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0217391304347828e-05,
+      "loss": 0.0122,
+      "step": 1325
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0209718670076726e-05,
+      "loss": 0.0122,
+      "step": 1326
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0202046035805628e-05,
+      "loss": 0.0121,
+      "step": 1327
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0194373401534526e-05,
+      "loss": 1.224,
+      "step": 1328
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0186700767263427e-05,
+      "loss": 2.3502,
+      "step": 1329
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.017902813299233e-05,
+      "loss": 0.0242,
+      "step": 1330
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0171355498721227e-05,
+      "loss": 0.0064,
+      "step": 1331
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0163682864450125e-05,
+      "loss": 0.0186,
+      "step": 1332
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.015601023017903e-05,
+      "loss": 0.012,
+      "step": 1333
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.014833759590793e-05,
+      "loss": 0.0119,
+      "step": 1334
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.0140664961636827e-05,
+      "loss": 0.0159,
+      "step": 1335
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.013299232736573e-05,
+      "loss": 0.0199,
+      "step": 1336
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.012531969309463e-05,
+      "loss": 1.2138,
+      "step": 1337
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.011764705882353e-05,
+      "loss": 0.9491,
+      "step": 1338
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010997442455243e-05,
+      "loss": 0.0083,
+      "step": 1339
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010230179028133e-05,
+      "loss": 0.948,
+      "step": 1340
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.009462915601023e-05,
+      "loss": 0.0112,
+      "step": 1341
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.008695652173913e-05,
+      "loss": 0.8921,
+      "step": 1342
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007928388746803e-05,
+      "loss": 0.0194,
+      "step": 1343
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007161125319693e-05,
+      "loss": 1.0273,
+      "step": 1344
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0063938618925833e-05,
+      "loss": 0.013,
+      "step": 1345
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.005626598465473e-05,
+      "loss": 0.0103,
+      "step": 1346
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0048593350383633e-05,
+      "loss": 0.0114,
+      "step": 1347
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0040920716112534e-05,
+      "loss": 0.0199,
+      "step": 1348
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0033248081841433e-05,
+      "loss": 1.183,
+      "step": 1349
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0025575447570334e-05,
+      "loss": 0.0231,
+      "step": 1350
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0017902813299232e-05,
+      "loss": 1.1794,
+      "step": 1351
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0010230179028134e-05,
+      "loss": 0.0096,
+      "step": 1352
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 2.0002557544757036e-05,
+      "loss": 0.9049,
+      "step": 1353
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9994884910485934e-05,
+      "loss": 0.019,
+      "step": 1354
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9987212276214832e-05,
+      "loss": 0.0152,
+      "step": 1355
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9979539641943737e-05,
+      "loss": 0.8834,
+      "step": 1356
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9971867007672635e-05,
+      "loss": 0.01,
+      "step": 1357
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9964194373401533e-05,
+      "loss": 0.0152,
+      "step": 1358
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9956521739130435e-05,
+      "loss": 0.0161,
+      "step": 1359
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9948849104859337e-05,
+      "loss": 1.6778,
+      "step": 1360
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9941176470588235e-05,
+      "loss": 0.0214,
+      "step": 1361
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9933503836317137e-05,
+      "loss": 0.0156,
+      "step": 1362
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9925831202046038e-05,
+      "loss": 0.026,
+      "step": 1363
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9918158567774936e-05,
+      "loss": 0.0259,
+      "step": 1364
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9910485933503838e-05,
+      "loss": 0.0161,
+      "step": 1365
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9902813299232736e-05,
+      "loss": 0.0108,
+      "step": 1366
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9895140664961638e-05,
+      "loss": 0.0149,
+      "step": 1367
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.988746803069054e-05,
+      "loss": 0.0325,
+      "step": 1368
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9879795396419438e-05,
+      "loss": 0.0153,
+      "step": 1369
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9872122762148336e-05,
+      "loss": 1.7678,
+      "step": 1370
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.986445012787724e-05,
+      "loss": 0.0773,
+      "step": 1371
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.985677749360614e-05,
+      "loss": 0.0244,
+      "step": 1372
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.9849104859335037e-05,
+      "loss": 0.0225,
+      "step": 1373
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.984143222506394e-05,
+      "loss": 0.0254,
+      "step": 1374
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.983375959079284e-05,
+      "loss": 0.0205,
+      "step": 1375
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.982608695652174e-05,
+      "loss": 0.0096,
+      "step": 1376
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981841432225064e-05,
+      "loss": 0.0147,
+      "step": 1377
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981074168797954e-05,
+      "loss": 0.0296,
+      "step": 1378
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.980306905370844e-05,
+      "loss": 0.0227,
+      "step": 1379
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.9795396419437342e-05,
+      "loss": 0.899,
+      "step": 1380
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.978772378516624e-05,
+      "loss": 0.019,
+      "step": 1381
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9780051150895138e-05,
+      "loss": 0.0157,
+      "step": 1382
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9772378516624043e-05,
+      "loss": 1.1897,
+      "step": 1383
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.976470588235294e-05,
+      "loss": 0.014,
+      "step": 1384
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.975703324808184e-05,
+      "loss": 0.0208,
+      "step": 1385
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.974936061381074e-05,
+      "loss": 0.0093,
+      "step": 1386
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.9741687979539643e-05,
+      "loss": 0.0244,
+      "step": 1387
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.973401534526854e-05,
+      "loss": 0.028,
+      "step": 1388
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9726342710997443e-05,
+      "loss": 0.0093,
+      "step": 1389
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9718670076726344e-05,
+      "loss": 0.0184,
+      "step": 1390
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9710997442455243e-05,
+      "loss": 0.9462,
+      "step": 1391
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9703324808184144e-05,
+      "loss": 0.0138,
+      "step": 1392
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9695652173913042e-05,
+      "loss": 0.9561,
+      "step": 1393
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9687979539641944e-05,
+      "loss": 0.0152,
+      "step": 1394
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9680306905370846e-05,
+      "loss": 0.0178,
+      "step": 1395
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9672634271099744e-05,
+      "loss": 0.0166,
+      "step": 1396
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9664961636828642e-05,
+      "loss": 0.92,
+      "step": 1397
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9657289002557547e-05,
+      "loss": 0.0253,
+      "step": 1398
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9649616368286445e-05,
+      "loss": 0.0266,
+      "step": 1399
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9641943734015347e-05,
+      "loss": 0.0105,
+      "step": 1400
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9634271099744245e-05,
+      "loss": 0.0208,
+      "step": 1401
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9626598465473147e-05,
+      "loss": 0.8094,
+      "step": 1402
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.961892583120205e-05,
+      "loss": 0.0103,
+      "step": 1403
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9611253196930947e-05,
+      "loss": 0.0179,
+      "step": 1404
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9603580562659845e-05,
+      "loss": 0.0133,
+      "step": 1405
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.959590792838875e-05,
+      "loss": 0.0086,
+      "step": 1406
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9588235294117648e-05,
+      "loss": 0.0216,
+      "step": 1407
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9580562659846546e-05,
+      "loss": 0.0127,
+      "step": 1408
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9572890025575448e-05,
+      "loss": 0.0127,
+      "step": 1409
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.956521739130435e-05,
+      "loss": 0.0209,
+      "step": 1410
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9557544757033248e-05,
+      "loss": 0.0166,
+      "step": 1411
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954987212276215e-05,
+      "loss": 0.0123,
+      "step": 1412
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954219948849105e-05,
+      "loss": 0.0135,
+      "step": 1413
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.953452685421995e-05,
+      "loss": 0.1253,
+      "step": 1414
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.952685421994885e-05,
+      "loss": 0.0098,
+      "step": 1415
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951918158567775e-05,
+      "loss": 0.0091,
+      "step": 1416
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951150895140665e-05,
+      "loss": 0.9327,
+      "step": 1417
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.9503836317135552e-05,
+      "loss": 0.0119,
+      "step": 1418
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.949616368286445e-05,
+      "loss": 0.0153,
+      "step": 1419
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.948849104859335e-05,
+      "loss": 0.0123,
+      "step": 1420
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9480818414322254e-05,
+      "loss": 0.0194,
+      "step": 1421
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9473145780051152e-05,
+      "loss": 1.2311,
+      "step": 1422
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.946547314578005e-05,
+      "loss": 1.8519,
+      "step": 1423
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9457800511508952e-05,
+      "loss": 0.963,
+      "step": 1424
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9450127877237853e-05,
+      "loss": 0.0075,
+      "step": 1425
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.944245524296675e-05,
+      "loss": 0.0042,
+      "step": 1426
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9434782608695653e-05,
+      "loss": 0.0077,
+      "step": 1427
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.942710997442455e-05,
+      "loss": 0.0152,
+      "step": 1428
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9419437340153453e-05,
+      "loss": 0.0152,
+      "step": 1429
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9411764705882355e-05,
+      "loss": 0.019,
+      "step": 1430
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9404092071611253e-05,
+      "loss": 0.0081,
+      "step": 1431
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.939641943734015e-05,
+      "loss": 0.0076,
+      "step": 1432
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9388746803069056e-05,
+      "loss": 0.0086,
+      "step": 1433
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9381074168797954e-05,
+      "loss": 0.0146,
+      "step": 1434
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9373401534526853e-05,
+      "loss": 0.0074,
+      "step": 1435
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9365728900255754e-05,
+      "loss": 0.1007,
+      "step": 1436
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9358056265984656e-05,
+      "loss": 0.011,
+      "step": 1437
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9350383631713554e-05,
+      "loss": 0.0089,
+      "step": 1438
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9342710997442456e-05,
+      "loss": 0.0182,
+      "step": 1439
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9335038363171357e-05,
+      "loss": 0.0139,
+      "step": 1440
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9327365728900255e-05,
+      "loss": 2.2086,
+      "step": 1441
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9319693094629157e-05,
+      "loss": 0.0178,
+      "step": 1442
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9312020460358055e-05,
+      "loss": 0.0142,
+      "step": 1443
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9304347826086957e-05,
+      "loss": 0.0142,
+      "step": 1444
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.929667519181586e-05,
+      "loss": 0.0242,
+      "step": 1445
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9289002557544757e-05,
+      "loss": 0.0087,
+      "step": 1446
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9281329923273655e-05,
+      "loss": 0.0077,
+      "step": 1447
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.927365728900256e-05,
+      "loss": 0.0104,
+      "step": 1448
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9265984654731458e-05,
+      "loss": 0.0139,
+      "step": 1449
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9258312020460356e-05,
+      "loss": 1.2557,
+      "step": 1450
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9250639386189258e-05,
+      "loss": 0.0068,
+      "step": 1451
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.924296675191816e-05,
+      "loss": 0.0163,
+      "step": 1452
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.923529411764706e-05,
+      "loss": 0.9483,
+      "step": 1453
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.922762148337596e-05,
+      "loss": 1.2181,
+      "step": 1454
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.9219948849104858e-05,
+      "loss": 0.011,
+      "step": 1455
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.9212276214833763e-05,
+      "loss": 0.014,
+      "step": 1456
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.920460358056266e-05,
+      "loss": 0.0131,
+      "step": 1457
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.919693094629156e-05,
+      "loss": 0.0135,
+      "step": 1458
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.918925831202046e-05,
+      "loss": 0.007,
+      "step": 1459
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9181585677749362e-05,
+      "loss": 0.0072,
+      "step": 1460
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.917391304347826e-05,
+      "loss": 0.0135,
+      "step": 1461
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9166240409207162e-05,
+      "loss": 0.9908,
+      "step": 1462
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9158567774936064e-05,
+      "loss": 0.0103,
+      "step": 1463
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9150895140664962e-05,
+      "loss": 0.9867,
+      "step": 1464
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9143222506393864e-05,
+      "loss": 0.0138,
+      "step": 1465
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9135549872122762e-05,
+      "loss": 0.0104,
+      "step": 1466
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9127877237851663e-05,
+      "loss": 0.014,
+      "step": 1467
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9120204603580565e-05,
+      "loss": 0.0071,
+      "step": 1468
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9112531969309463e-05,
+      "loss": 0.0108,
+      "step": 1469
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.910485933503836e-05,
+      "loss": 0.0106,
+      "step": 1470
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9097186700767266e-05,
+      "loss": 0.0404,
+      "step": 1471
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9089514066496165e-05,
+      "loss": 2.2011,
+      "step": 1472
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9081841432225063e-05,
+      "loss": 0.0087,
+      "step": 1473
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9074168797953965e-05,
+      "loss": 0.0071,
+      "step": 1474
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9066496163682866e-05,
+      "loss": 0.0072,
+      "step": 1475
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9058823529411764e-05,
+      "loss": 0.0108,
+      "step": 1476
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9051150895140666e-05,
+      "loss": 0.0108,
+      "step": 1477
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9043478260869564e-05,
+      "loss": 0.0072,
+      "step": 1478
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9035805626598466e-05,
+      "loss": 0.0108,
+      "step": 1479
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9028132992327367e-05,
+      "loss": 0.0073,
+      "step": 1480
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9020460358056266e-05,
+      "loss": 0.9537,
+      "step": 1481
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9012787723785164e-05,
+      "loss": 0.0107,
+      "step": 1482
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.900511508951407e-05,
+      "loss": 0.0145,
+      "step": 1483
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8997442455242967e-05,
+      "loss": 0.007,
+      "step": 1484
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8989769820971865e-05,
+      "loss": 0.0071,
+      "step": 1485
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.898209718670077e-05,
+      "loss": 1.243,
+      "step": 1486
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.897442455242967e-05,
+      "loss": 0.011,
+      "step": 1487
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.8966751918158567e-05,
+      "loss": 0.0077,
+      "step": 1488
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895907928388747e-05,
+      "loss": 0.0073,
+      "step": 1489
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895140664961637e-05,
+      "loss": 1.2486,
+      "step": 1490
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8943734015345268e-05,
+      "loss": 0.0109,
+      "step": 1491
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.893606138107417e-05,
+      "loss": 0.0149,
+      "step": 1492
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8928388746803068e-05,
+      "loss": 0.0125,
+      "step": 1493
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.892071611253197e-05,
+      "loss": 0.0071,
+      "step": 1494
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.891304347826087e-05,
+      "loss": 0.0036,
+      "step": 1495
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.890537084398977e-05,
+      "loss": 0.0149,
+      "step": 1496
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8897698209718668e-05,
+      "loss": 0.015,
+      "step": 1497
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8890025575447573e-05,
+      "loss": 0.9536,
+      "step": 1498
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.888235294117647e-05,
+      "loss": 0.9997,
+      "step": 1499
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.887468030690537e-05,
+      "loss": 0.9621,
+      "step": 1500
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.886700767263427e-05,
+      "loss": 0.0113,
+      "step": 1501
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.8859335038363172e-05,
+      "loss": 0.2056,
+      "step": 1502
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.885166240409207e-05,
+      "loss": 0.0114,
+      "step": 1503
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8843989769820972e-05,
+      "loss": 0.016,
+      "step": 1504
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.883631713554987e-05,
+      "loss": 0.9175,
+      "step": 1505
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8828644501278775e-05,
+      "loss": 0.8959,
+      "step": 1506
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8820971867007674e-05,
+      "loss": 0.0074,
+      "step": 1507
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8813299232736572e-05,
+      "loss": 0.0117,
+      "step": 1508
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8805626598465474e-05,
+      "loss": 0.0075,
+      "step": 1509
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8797953964194375e-05,
+      "loss": 0.0074,
+      "step": 1510
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8790281329923273e-05,
+      "loss": 0.0577,
+      "step": 1511
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8782608695652175e-05,
+      "loss": 0.0128,
+      "step": 1512
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8774936061381077e-05,
+      "loss": 0.0073,
+      "step": 1513
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8767263427109975e-05,
+      "loss": 0.0074,
+      "step": 1514
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8759590792838876e-05,
+      "loss": 0.0073,
+      "step": 1515
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8751918158567775e-05,
+      "loss": 0.0088,
+      "step": 1516
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8744245524296676e-05,
+      "loss": 0.0078,
+      "step": 1517
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8736572890025578e-05,
+      "loss": 0.016,
+      "step": 1518
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8728900255754476e-05,
+      "loss": 1.3842,
+      "step": 1519
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8721227621483374e-05,
+      "loss": 1.2502,
+      "step": 1520
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.871355498721228e-05,
+      "loss": 0.0117,
+      "step": 1521
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8705882352941178e-05,
+      "loss": 0.0113,
+      "step": 1522
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8698209718670076e-05,
+      "loss": 0.0125,
+      "step": 1523
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8690537084398977e-05,
+      "loss": 0.0153,
+      "step": 1524
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.868286445012788e-05,
+      "loss": 0.0072,
+      "step": 1525
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8675191815856777e-05,
+      "loss": 0.0086,
+      "step": 1526
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.866751918158568e-05,
+      "loss": 0.0195,
+      "step": 1527
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.8659846547314577e-05,
+      "loss": 1.2511,
+      "step": 1528
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.865217391304348e-05,
+      "loss": 1.2407,
+      "step": 1529
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.864450127877238e-05,
+      "loss": 0.0112,
+      "step": 1530
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.863682864450128e-05,
+      "loss": 0.0073,
+      "step": 1531
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8629156010230177e-05,
+      "loss": 1.1911,
+      "step": 1532
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8621483375959082e-05,
+      "loss": 0.0168,
+      "step": 1533
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.861381074168798e-05,
+      "loss": 0.0128,
+      "step": 1534
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8606138107416878e-05,
+      "loss": 0.0116,
+      "step": 1535
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8598465473145783e-05,
+      "loss": 1.2286,
+      "step": 1536
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.859079283887468e-05,
+      "loss": 0.0158,
+      "step": 1537
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.858312020460358e-05,
+      "loss": 2.3735,
+      "step": 1538
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.857544757033248e-05,
+      "loss": 1.1972,
+      "step": 1539
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8567774936061383e-05,
+      "loss": 0.011,
+      "step": 1540
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.856010230179028e-05,
+      "loss": 0.0127,
+      "step": 1541
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8552429667519183e-05,
+      "loss": 0.0148,
+      "step": 1542
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.854475703324808e-05,
+      "loss": 0.0111,
+      "step": 1543
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8537084398976983e-05,
+      "loss": 0.0143,
+      "step": 1544
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8529411764705884e-05,
+      "loss": 0.0132,
+      "step": 1545
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8521739130434782e-05,
+      "loss": 0.0152,
+      "step": 1546
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.851406649616368e-05,
+      "loss": 0.0152,
+      "step": 1547
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8506393861892586e-05,
+      "loss": 0.0051,
+      "step": 1548
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8498721227621484e-05,
+      "loss": 0.9576,
+      "step": 1549
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8491048593350382e-05,
+      "loss": 0.0181,
+      "step": 1550
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8483375959079284e-05,
+      "loss": 0.011,
+      "step": 1551
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8475703324808185e-05,
+      "loss": 0.0205,
+      "step": 1552
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8468030690537083e-05,
+      "loss": 2.5631,
+      "step": 1553
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8460358056265985e-05,
+      "loss": 0.0102,
+      "step": 1554
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8452685421994883e-05,
+      "loss": 0.0147,
+      "step": 1555
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8445012787723788e-05,
+      "loss": 0.015,
+      "step": 1556
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8437340153452687e-05,
+      "loss": 0.0122,
+      "step": 1557
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8429667519181585e-05,
+      "loss": 0.0176,
+      "step": 1558
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8421994884910486e-05,
+      "loss": 0.0114,
+      "step": 1559
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8414322250639388e-05,
+      "loss": 0.0252,
+      "step": 1560
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8406649616368286e-05,
+      "loss": 0.013,
+      "step": 1561
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8398976982097188e-05,
+      "loss": 0.0142,
+      "step": 1562
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.839130434782609e-05,
+      "loss": 0.6496,
+      "step": 1563
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8383631713554988e-05,
+      "loss": 0.0149,
+      "step": 1564
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.837595907928389e-05,
+      "loss": 0.0104,
+      "step": 1565
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8368286445012787e-05,
+      "loss": 0.0129,
+      "step": 1566
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.836061381074169e-05,
+      "loss": 0.0197,
+      "step": 1567
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.835294117647059e-05,
+      "loss": 0.009,
+      "step": 1568
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.834526854219949e-05,
+      "loss": 0.0086,
+      "step": 1569
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.8337595907928387e-05,
+      "loss": 0.0087,
+      "step": 1570
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.8329923273657292e-05,
+      "loss": 0.0131,
+      "step": 1571
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.832225063938619e-05,
+      "loss": 1.2046,
+      "step": 1572
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.831457800511509e-05,
+      "loss": 0.0135,
+      "step": 1573
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.830690537084399e-05,
+      "loss": 0.0149,
+      "step": 1574
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.8299232736572892e-05,
+      "loss": 0.0077,
+      "step": 1575
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.829156010230179e-05,
+      "loss": 0.0143,
+      "step": 1576
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.828388746803069e-05,
+      "loss": 0.9806,
+      "step": 1577
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.827621483375959e-05,
+      "loss": 1.3783,
+      "step": 1578
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.826854219948849e-05,
+      "loss": 0.0075,
+      "step": 1579
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.8260869565217393e-05,
+      "loss": 0.0111,
+      "step": 1580
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.825319693094629e-05,
+      "loss": 0.0111,
+      "step": 1581
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.824552429667519e-05,
+      "loss": 0.0144,
+      "step": 1582
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8237851662404095e-05,
+      "loss": 0.0128,
+      "step": 1583
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8230179028132993e-05,
+      "loss": 1.2949,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.40147146582603455,
+      "eval_runtime": 58.0487,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1584
+    }
+  ],
+  "max_steps": 3960,
+  "num_train_epochs": 10,
+  "total_flos": 5.748590187648e+17,
+  "trial_name": null,
+  "trial_params": null
+}
diff --git a/checkpoint-1584/training_args.bin b/checkpoint-1584/training_args.bin
new file mode 100644
index 0000000000000000000000000000000000000000..32901f899c91066c808eae9dd019400bfdd17498
--- /dev/null
+++ b/checkpoint-1584/training_args.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db14439c369bfe9c89f64ff38f0b2d3fcc53911fc2c0b12a3a7d04656e7384a
+size 3503
diff --git a/checkpoint-1980/config.json b/checkpoint-1980/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..cb8d52e14263c50fd53dca42b5050e3235115c32
--- /dev/null
+++ b/checkpoint-1980/config.json
@@ -0,0 +1,113 @@
+{
+  "_name_or_path": "facebook/wav2vec2-base-960h",
+  "activation_dropout": 0.1,
+  "adapter_kernel_size": 3,
+  "adapter_stride": 2,
+  "add_adapter": false,
+  "apply_spec_augment": true,
+  "architectures": [
+    "Wav2Vec2ForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 1,
+  "classifier_proj_size": 256,
+  "codevector_dim": 256,
+  "contrastive_logits_temperature": 0.1,
+  "conv_bias": false,
+  "conv_dim": [
+    512,
+    512,
+    512,
+    512,
+    512,
+    512,
+    512
+  ],
+  "conv_kernel": [
+    10,
+    3,
+    3,
+    3,
+    3,
+    2,
+    2
+  ],
+  "conv_stride": [
+    5,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2
+  ],
+  "ctc_loss_reduction": "sum",
+  "ctc_zero_infinity": false,
+  "diversity_loss_weight": 0.1,
+  "do_stable_layer_norm": false,
+  "eos_token_id": 2,
+  "feat_extract_activation": "gelu",
+  "feat_extract_dropout": 0.0,
+  "feat_extract_norm": "group",
+  "feat_proj_dropout": 0.1,
+  "feat_quantizer_dropout": 0.0,
+  "final_dropout": 0.1,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout": 0.1,
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Foreign",
+    "1": "Native"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "layerdrop": 0.1,
+  "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
+  "mask_feature_prob": 0.0,
+  "mask_time_length": 10,
+  "mask_time_min_masks": 2,
+  "mask_time_prob": 0.05,
+  "model_type": "wav2vec2",
+  "num_adapter_layers": 3,
+  "num_attention_heads": 12,
+  "num_codevector_groups": 2,
+  "num_codevectors_per_group": 320,
+  "num_conv_pos_embedding_groups": 16,
+  "num_conv_pos_embeddings": 128,
+  "num_feat_extract_layers": 7,
+  "num_hidden_layers": 12,
+  "num_negatives": 100,
+  "output_hidden_size": 768,
+  "pad_token_id": 0,
+  "proj_codevector_dim": 256,
+  "tdnn_dilation": [
+    1,
+    2,
+    3,
+    1,
+    1
+  ],
+  "tdnn_dim": [
+    512,
+    512,
+    512,
+    512,
+    1500
+  ],
+  "tdnn_kernel": [
+    5,
+    3,
+    3,
+    1,
+    1
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.28.1",
+  "use_weighted_layer_sum": false,
+  "vocab_size": 32,
+  "xvector_output_dim": 512
+}
diff --git a/checkpoint-1980/optimizer.pt b/checkpoint-1980/optimizer.pt
new file mode 100644
index 0000000000000000000000000000000000000000..5e7d26f2637a6dac82c23278c210f1271e2851e3
--- /dev/null
+++ b/checkpoint-1980/optimizer.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ef2af0605afa1a43a02b20603c395961dc7319d54a4fe06873cb388f2977cdc1
+size 756677477
diff --git a/checkpoint-1980/preprocessor_config.json b/checkpoint-1980/preprocessor_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..a0b7227fc1d916e469b14f6c154ad6dfea1e6891
--- /dev/null
+++ b/checkpoint-1980/preprocessor_config.json
@@ -0,0 +1,9 @@
+{
+  "do_normalize": true,
+  "feature_extractor_type": "Wav2Vec2FeatureExtractor",
+  "feature_size": 1,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": false,
+  "sampling_rate": 16000
+}
diff --git a/checkpoint-1980/pytorch_model.bin b/checkpoint-1980/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..239d68fb4ace0923eff64cecfe962a3793cdedc4
--- /dev/null
+++ b/checkpoint-1980/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:50d57280519ffd40f6fca295963a4db126c049cdcac69c222f30bc68398060c5
+size 378348563
diff --git a/checkpoint-1980/rng_state.pth b/checkpoint-1980/rng_state.pth
new file mode 100644
index 0000000000000000000000000000000000000000..184d283c57c8a97acb942065b36a754bf2562d03
--- /dev/null
+++ b/checkpoint-1980/rng_state.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f4a0d3f377cf41bc60bb1cbe5a284d4a18c47056f8a2e805f0420bd6bd52fdcd
+size 14631
diff --git a/checkpoint-1980/scheduler.pt b/checkpoint-1980/scheduler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..c52dfec22f86021fec1dfa44f28d74474becd4bd
--- /dev/null
+++ b/checkpoint-1980/scheduler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3192417d69cb3637bbe702cbf00e51984fbc3d22ce07e248105a71df9a183a81
+size 623
diff --git a/checkpoint-1980/trainer_state.json b/checkpoint-1980/trainer_state.json
new file mode 100644
index 0000000000000000000000000000000000000000..42624bf2800a138069d6c09a5d4ba327106a4632
--- /dev/null
+++ b/checkpoint-1980/trainer_state.json
@@ -0,0 +1,11941 @@
+{
+  "best_metric": 0.3537317216396332,
+  "best_model_checkpoint": "accent_classification/checkpoint-792",
+  "epoch": 5.0,
+  "global_step": 1980,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.000000000000001e-07,
+      "loss": 0.6956,
+      "step": 1
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.2000000000000002e-06,
+      "loss": 0.6979,
+      "step": 2
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8e-06,
+      "loss": 0.7014,
+      "step": 3
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 2.4000000000000003e-06,
+      "loss": 0.6984,
+      "step": 4
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 3e-06,
+      "loss": 0.6978,
+      "step": 5
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 3.6e-06,
+      "loss": 0.6783,
+      "step": 6
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.2000000000000004e-06,
+      "loss": 0.6956,
+      "step": 7
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.800000000000001e-06,
+      "loss": 0.6994,
+      "step": 8
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 5.4e-06,
+      "loss": 0.6902,
+      "step": 9
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6e-06,
+      "loss": 0.6937,
+      "step": 10
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6.6e-06,
+      "loss": 0.6497,
+      "step": 11
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.2e-06,
+      "loss": 0.6911,
+      "step": 12
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.8e-06,
+      "loss": 0.6904,
+      "step": 13
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 8.400000000000001e-06,
+      "loss": 0.6854,
+      "step": 14
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9e-06,
+      "loss": 0.6843,
+      "step": 15
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9.600000000000001e-06,
+      "loss": 0.6851,
+      "step": 16
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.02e-05,
+      "loss": 0.6733,
+      "step": 17
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.08e-05,
+      "loss": 0.6819,
+      "step": 18
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.1400000000000001e-05,
+      "loss": 0.681,
+      "step": 19
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.6998,
+      "step": 20
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.26e-05,
+      "loss": 0.6743,
+      "step": 21
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.32e-05,
+      "loss": 0.6818,
+      "step": 22
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3800000000000002e-05,
+      "loss": 0.5331,
+      "step": 23
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.44e-05,
+      "loss": 0.6681,
+      "step": 24
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.6955,
+      "step": 25
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.56e-05,
+      "loss": 0.7,
+      "step": 26
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.62e-05,
+      "loss": 0.6975,
+      "step": 27
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.6800000000000002e-05,
+      "loss": 0.6249,
+      "step": 28
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.74e-05,
+      "loss": 0.6979,
+      "step": 29
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.6076,
+      "step": 30
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.86e-05,
+      "loss": 0.6091,
+      "step": 31
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9200000000000003e-05,
+      "loss": 0.6438,
+      "step": 32
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.98e-05,
+      "loss": 0.6624,
+      "step": 33
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.04e-05,
+      "loss": 0.6452,
+      "step": 34
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.6963,
+      "step": 35
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.16e-05,
+      "loss": 0.758,
+      "step": 36
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.22e-05,
+      "loss": 0.6471,
+      "step": 37
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.2800000000000002e-05,
+      "loss": 0.6313,
+      "step": 38
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.3400000000000003e-05,
+      "loss": 0.7019,
+      "step": 39
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.6336,
+      "step": 40
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4599999999999998e-05,
+      "loss": 0.7091,
+      "step": 41
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.52e-05,
+      "loss": 0.701,
+      "step": 42
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.58e-05,
+      "loss": 0.7036,
+      "step": 43
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.64e-05,
+      "loss": 0.7073,
+      "step": 44
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.614,
+      "step": 45
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.7600000000000003e-05,
+      "loss": 0.7543,
+      "step": 46
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.8199999999999998e-05,
+      "loss": 0.538,
+      "step": 47
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.88e-05,
+      "loss": 0.5179,
+      "step": 48
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.94e-05,
+      "loss": 0.6164,
+      "step": 49
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 3e-05,
+      "loss": 0.6093,
+      "step": 50
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.99923273657289e-05,
+      "loss": 0.6983,
+      "step": 51
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9984654731457804e-05,
+      "loss": 0.6069,
+      "step": 52
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9976982097186702e-05,
+      "loss": 0.5933,
+      "step": 53
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.99693094629156e-05,
+      "loss": 0.552,
+      "step": 54
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9961636828644502e-05,
+      "loss": 0.5865,
+      "step": 55
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9953964194373404e-05,
+      "loss": 0.7146,
+      "step": 56
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9946291560102302e-05,
+      "loss": 0.5591,
+      "step": 57
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9938618925831203e-05,
+      "loss": 0.7366,
+      "step": 58
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9930946291560102e-05,
+      "loss": 0.7006,
+      "step": 59
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9923273657289003e-05,
+      "loss": 0.4148,
+      "step": 60
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9915601023017905e-05,
+      "loss": 0.9938,
+      "step": 61
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9907928388746803e-05,
+      "loss": 0.911,
+      "step": 62
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.99002557544757e-05,
+      "loss": 0.5682,
+      "step": 63
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9892583120204606e-05,
+      "loss": 0.5833,
+      "step": 64
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9884910485933505e-05,
+      "loss": 0.7237,
+      "step": 65
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9877237851662403e-05,
+      "loss": 0.4013,
+      "step": 66
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9869565217391304e-05,
+      "loss": 0.9695,
+      "step": 67
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9861892583120206e-05,
+      "loss": 0.568,
+      "step": 68
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9854219948849104e-05,
+      "loss": 0.5697,
+      "step": 69
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9846547314578006e-05,
+      "loss": 0.7675,
+      "step": 70
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9838874680306904e-05,
+      "loss": 0.8831,
+      "step": 71
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9831202046035806e-05,
+      "loss": 0.7008,
+      "step": 72
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9823529411764707e-05,
+      "loss": 0.7085,
+      "step": 73
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9815856777493606e-05,
+      "loss": 0.5374,
+      "step": 74
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9808184143222507e-05,
+      "loss": 0.4131,
+      "step": 75
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.980051150895141e-05,
+      "loss": 0.7309,
+      "step": 76
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9792838874680307e-05,
+      "loss": 0.4279,
+      "step": 77
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.978516624040921e-05,
+      "loss": 0.5813,
+      "step": 78
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.977749360613811e-05,
+      "loss": 0.6276,
+      "step": 79
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976982097186701e-05,
+      "loss": 0.7892,
+      "step": 80
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976214833759591e-05,
+      "loss": 0.7134,
+      "step": 81
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.9754475703324808e-05,
+      "loss": 0.3682,
+      "step": 82
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.974680306905371e-05,
+      "loss": 0.5562,
+      "step": 83
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973913043478261e-05,
+      "loss": 0.7158,
+      "step": 84
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973145780051151e-05,
+      "loss": 0.524,
+      "step": 85
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9723785166240408e-05,
+      "loss": 0.3372,
+      "step": 86
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9716112531969313e-05,
+      "loss": 0.4979,
+      "step": 87
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970843989769821e-05,
+      "loss": 0.3177,
+      "step": 88
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970076726342711e-05,
+      "loss": 0.3488,
+      "step": 89
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.969309462915601e-05,
+      "loss": 0.5158,
+      "step": 90
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9685421994884913e-05,
+      "loss": 0.3208,
+      "step": 91
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.967774936061381e-05,
+      "loss": 0.6654,
+      "step": 92
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9670076726342712e-05,
+      "loss": 0.5581,
+      "step": 93
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.966240409207161e-05,
+      "loss": 0.7456,
+      "step": 94
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9654731457800512e-05,
+      "loss": 0.4437,
+      "step": 95
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9647058823529414e-05,
+      "loss": 0.5525,
+      "step": 96
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9639386189258312e-05,
+      "loss": 0.6673,
+      "step": 97
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.963171355498721e-05,
+      "loss": 0.435,
+      "step": 98
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9624040920716115e-05,
+      "loss": 0.7415,
+      "step": 99
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9616368286445014e-05,
+      "loss": 0.2981,
+      "step": 100
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9608695652173912e-05,
+      "loss": 0.5179,
+      "step": 101
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9601023017902817e-05,
+      "loss": 0.4005,
+      "step": 102
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9593350383631715e-05,
+      "loss": 0.4598,
+      "step": 103
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9585677749360613e-05,
+      "loss": 0.2553,
+      "step": 104
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9578005115089515e-05,
+      "loss": 0.4302,
+      "step": 105
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9570332480818416e-05,
+      "loss": 0.7894,
+      "step": 106
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9562659846547315e-05,
+      "loss": 0.213,
+      "step": 107
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9554987212276216e-05,
+      "loss": 0.5507,
+      "step": 108
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9547314578005115e-05,
+      "loss": 0.7676,
+      "step": 109
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9539641943734016e-05,
+      "loss": 0.3418,
+      "step": 110
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9531969309462918e-05,
+      "loss": 0.8533,
+      "step": 111
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9524296675191816e-05,
+      "loss": 0.3659,
+      "step": 112
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9516624040920714e-05,
+      "loss": 0.6482,
+      "step": 113
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.950895140664962e-05,
+      "loss": 0.5484,
+      "step": 114
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9501278772378517e-05,
+      "loss": 0.4508,
+      "step": 115
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9493606138107416e-05,
+      "loss": 0.4401,
+      "step": 116
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9485933503836317e-05,
+      "loss": 0.5455,
+      "step": 117
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.947826086956522e-05,
+      "loss": 0.722,
+      "step": 118
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9470588235294117e-05,
+      "loss": 0.2874,
+      "step": 119
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.946291560102302e-05,
+      "loss": 0.5973,
+      "step": 120
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.9455242966751917e-05,
+      "loss": 0.5898,
+      "step": 121
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.944757033248082e-05,
+      "loss": 1.0125,
+      "step": 122
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943989769820972e-05,
+      "loss": 0.2511,
+      "step": 123
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943222506393862e-05,
+      "loss": 0.545,
+      "step": 124
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.942455242966752e-05,
+      "loss": 0.593,
+      "step": 125
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.941687979539642e-05,
+      "loss": 0.2319,
+      "step": 126
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.940920716112532e-05,
+      "loss": 0.2215,
+      "step": 127
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.9401534526854218e-05,
+      "loss": 0.1964,
+      "step": 128
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9393861892583123e-05,
+      "loss": 0.1471,
+      "step": 129
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.938618925831202e-05,
+      "loss": 1.5965,
+      "step": 130
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9378516624040923e-05,
+      "loss": 1.0772,
+      "step": 131
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.937084398976982e-05,
+      "loss": 0.2005,
+      "step": 132
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9363171355498723e-05,
+      "loss": 0.9628,
+      "step": 133
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9355498721227624e-05,
+      "loss": 0.1503,
+      "step": 134
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9347826086956523e-05,
+      "loss": 0.6201,
+      "step": 135
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.934015345268542e-05,
+      "loss": 0.6094,
+      "step": 136
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9332480818414326e-05,
+      "loss": 1.385,
+      "step": 137
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9324808184143224e-05,
+      "loss": 0.1552,
+      "step": 138
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9317135549872122e-05,
+      "loss": 0.5828,
+      "step": 139
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9309462915601024e-05,
+      "loss": 0.6018,
+      "step": 140
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9301790281329925e-05,
+      "loss": 1.0614,
+      "step": 141
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9294117647058824e-05,
+      "loss": 0.566,
+      "step": 142
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9286445012787725e-05,
+      "loss": 0.5958,
+      "step": 143
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9278772378516624e-05,
+      "loss": 0.5838,
+      "step": 144
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9271099744245525e-05,
+      "loss": 0.5909,
+      "step": 145
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9263427109974427e-05,
+      "loss": 0.5862,
+      "step": 146
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9255754475703325e-05,
+      "loss": 0.2031,
+      "step": 147
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9248081841432223e-05,
+      "loss": 0.1995,
+      "step": 148
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9240409207161128e-05,
+      "loss": 0.24,
+      "step": 149
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9232736572890026e-05,
+      "loss": 0.5845,
+      "step": 150
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9225063938618925e-05,
+      "loss": 0.964,
+      "step": 151
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.921739130434783e-05,
+      "loss": 0.9595,
+      "step": 152
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9209718670076728e-05,
+      "loss": 0.2033,
+      "step": 153
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9202046035805626e-05,
+      "loss": 0.213,
+      "step": 154
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9194373401534528e-05,
+      "loss": 0.2042,
+      "step": 155
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.918670076726343e-05,
+      "loss": 0.5783,
+      "step": 156
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9179028132992328e-05,
+      "loss": 1.0822,
+      "step": 157
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.917135549872123e-05,
+      "loss": 1.2643,
+      "step": 158
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9163682864450127e-05,
+      "loss": 0.2593,
+      "step": 159
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.915601023017903e-05,
+      "loss": 0.5783,
+      "step": 160
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914833759590793e-05,
+      "loss": 0.5721,
+      "step": 161
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914066496163683e-05,
+      "loss": 0.5808,
+      "step": 162
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9132992327365727e-05,
+      "loss": 1.2578,
+      "step": 163
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9125319693094632e-05,
+      "loss": 0.5709,
+      "step": 164
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.911764705882353e-05,
+      "loss": 0.2159,
+      "step": 165
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910997442455243e-05,
+      "loss": 0.2329,
+      "step": 166
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910230179028133e-05,
+      "loss": 0.926,
+      "step": 167
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.9094629156010232e-05,
+      "loss": 0.9095,
+      "step": 168
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.908695652173913e-05,
+      "loss": 0.5622,
+      "step": 169
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907928388746803e-05,
+      "loss": 0.5703,
+      "step": 170
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907161125319693e-05,
+      "loss": 0.5669,
+      "step": 171
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.906393861892583e-05,
+      "loss": 0.2637,
+      "step": 172
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9056265984654733e-05,
+      "loss": 1.1077,
+      "step": 173
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.904859335038363e-05,
+      "loss": 0.5653,
+      "step": 174
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9040920716112533e-05,
+      "loss": 0.5584,
+      "step": 175
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9033248081841434e-05,
+      "loss": 0.5544,
+      "step": 176
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9025575447570333e-05,
+      "loss": 0.5663,
+      "step": 177
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.901790281329923e-05,
+      "loss": 1.0808,
+      "step": 178
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9010230179028136e-05,
+      "loss": 0.8736,
+      "step": 179
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9002557544757034e-05,
+      "loss": 0.6435,
+      "step": 180
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8994884910485932e-05,
+      "loss": 0.786,
+      "step": 181
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8987212276214834e-05,
+      "loss": 0.5717,
+      "step": 182
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8979539641943736e-05,
+      "loss": 0.5618,
+      "step": 183
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8971867007672637e-05,
+      "loss": 0.5785,
+      "step": 184
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8964194373401535e-05,
+      "loss": 0.7601,
+      "step": 185
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8956521739130434e-05,
+      "loss": 0.8112,
+      "step": 186
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.894884910485934e-05,
+      "loss": 0.5663,
+      "step": 187
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8941176470588237e-05,
+      "loss": 0.5631,
+      "step": 188
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8933503836317135e-05,
+      "loss": 0.7363,
+      "step": 189
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8925831202046037e-05,
+      "loss": 0.4552,
+      "step": 190
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8918158567774938e-05,
+      "loss": 0.4599,
+      "step": 191
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8910485933503836e-05,
+      "loss": 0.7848,
+      "step": 192
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8902813299232738e-05,
+      "loss": 0.5907,
+      "step": 193
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8895140664961636e-05,
+      "loss": 0.4264,
+      "step": 194
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8887468030690538e-05,
+      "loss": 0.5697,
+      "step": 195
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.887979539641944e-05,
+      "loss": 0.7643,
+      "step": 196
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8872122762148338e-05,
+      "loss": 0.3635,
+      "step": 197
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8864450127877236e-05,
+      "loss": 0.3715,
+      "step": 198
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.885677749360614e-05,
+      "loss": 0.554,
+      "step": 199
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.884910485933504e-05,
+      "loss": 0.8044,
+      "step": 200
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8841432225063937e-05,
+      "loss": 0.5673,
+      "step": 201
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8833759590792842e-05,
+      "loss": 0.8042,
+      "step": 202
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.882608695652174e-05,
+      "loss": 0.5638,
+      "step": 203
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881841432225064e-05,
+      "loss": 0.3584,
+      "step": 204
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881074168797954e-05,
+      "loss": 0.8211,
+      "step": 205
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.8803069053708442e-05,
+      "loss": 0.5708,
+      "step": 206
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.879539641943734e-05,
+      "loss": 0.8301,
+      "step": 207
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8787723785166242e-05,
+      "loss": 0.8229,
+      "step": 208
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.878005115089514e-05,
+      "loss": 0.5645,
+      "step": 209
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8772378516624042e-05,
+      "loss": 0.3529,
+      "step": 210
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8764705882352943e-05,
+      "loss": 0.5578,
+      "step": 211
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.875703324808184e-05,
+      "loss": 0.5659,
+      "step": 212
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.874936061381074e-05,
+      "loss": 0.7495,
+      "step": 213
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8741687979539645e-05,
+      "loss": 0.3162,
+      "step": 214
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8734015345268543e-05,
+      "loss": 0.5616,
+      "step": 215
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.872634271099744e-05,
+      "loss": 0.4034,
+      "step": 216
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8718670076726343e-05,
+      "loss": 0.3163,
+      "step": 217
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8710997442455245e-05,
+      "loss": 0.5514,
+      "step": 218
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8703324808184143e-05,
+      "loss": 0.826,
+      "step": 219
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8695652173913044e-05,
+      "loss": 0.8363,
+      "step": 220
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8687979539641943e-05,
+      "loss": 0.8288,
+      "step": 221
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8680306905370844e-05,
+      "loss": 0.5929,
+      "step": 222
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8672634271099746e-05,
+      "loss": 0.565,
+      "step": 223
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8664961636828644e-05,
+      "loss": 0.5642,
+      "step": 224
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8657289002557546e-05,
+      "loss": 0.5609,
+      "step": 225
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8649616368286447e-05,
+      "loss": 0.5641,
+      "step": 226
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8641943734015345e-05,
+      "loss": 0.8535,
+      "step": 227
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8634271099744244e-05,
+      "loss": 0.5652,
+      "step": 228
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.862659846547315e-05,
+      "loss": 0.4397,
+      "step": 229
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8618925831202047e-05,
+      "loss": 0.567,
+      "step": 230
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8611253196930945e-05,
+      "loss": 0.7517,
+      "step": 231
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8603580562659847e-05,
+      "loss": 0.5839,
+      "step": 232
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.859590792838875e-05,
+      "loss": 0.5677,
+      "step": 233
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8588235294117647e-05,
+      "loss": 0.5678,
+      "step": 234
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8580562659846548e-05,
+      "loss": 1.3045,
+      "step": 235
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8572890025575446e-05,
+      "loss": 0.804,
+      "step": 236
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.856521739130435e-05,
+      "loss": 1.0303,
+      "step": 237
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.855754475703325e-05,
+      "loss": 0.3303,
+      "step": 238
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8549872122762148e-05,
+      "loss": 0.5677,
+      "step": 239
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.854219948849105e-05,
+      "loss": 0.7972,
+      "step": 240
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.853452685421995e-05,
+      "loss": 0.5644,
+      "step": 241
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.852685421994885e-05,
+      "loss": 0.9294,
+      "step": 242
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.851918158567775e-05,
+      "loss": 0.5665,
+      "step": 243
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.851150895140665e-05,
+      "loss": 0.7963,
+      "step": 244
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.850383631713555e-05,
+      "loss": 0.5659,
+      "step": 245
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.8496163682864452e-05,
+      "loss": 0.5718,
+      "step": 246
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.848849104859335e-05,
+      "loss": 0.3526,
+      "step": 247
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8480818414322252e-05,
+      "loss": 0.574,
+      "step": 248
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8473145780051154e-05,
+      "loss": 0.9996,
+      "step": 249
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8465473145780052e-05,
+      "loss": 0.7634,
+      "step": 250
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.845780051150895e-05,
+      "loss": 0.5639,
+      "step": 251
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8450127877237855e-05,
+      "loss": 0.5737,
+      "step": 252
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8442455242966753e-05,
+      "loss": 0.7793,
+      "step": 253
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8434782608695652e-05,
+      "loss": 0.6769,
+      "step": 254
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8427109974424553e-05,
+      "loss": 0.5748,
+      "step": 255
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8419437340153455e-05,
+      "loss": 0.3602,
+      "step": 256
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8411764705882353e-05,
+      "loss": 0.5714,
+      "step": 257
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8404092071611255e-05,
+      "loss": 0.5738,
+      "step": 258
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8396419437340153e-05,
+      "loss": 0.5719,
+      "step": 259
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8388746803069055e-05,
+      "loss": 0.5753,
+      "step": 260
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8381074168797956e-05,
+      "loss": 0.5717,
+      "step": 261
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8373401534526854e-05,
+      "loss": 0.9794,
+      "step": 262
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8365728900255753e-05,
+      "loss": 0.5737,
+      "step": 263
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8358056265984658e-05,
+      "loss": 0.5723,
+      "step": 264
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8350383631713556e-05,
+      "loss": 0.5656,
+      "step": 265
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8342710997442454e-05,
+      "loss": 0.4259,
+      "step": 266
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8335038363171356e-05,
+      "loss": 0.3076,
+      "step": 267
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8327365728900257e-05,
+      "loss": 0.338,
+      "step": 268
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8319693094629156e-05,
+      "loss": 0.5612,
+      "step": 269
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8312020460358057e-05,
+      "loss": 0.5249,
+      "step": 270
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8304347826086955e-05,
+      "loss": 0.5671,
+      "step": 271
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8296675191815857e-05,
+      "loss": 0.5574,
+      "step": 272
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.828900255754476e-05,
+      "loss": 0.8152,
+      "step": 273
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8281329923273657e-05,
+      "loss": 0.5647,
+      "step": 274
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.827365728900256e-05,
+      "loss": 0.5632,
+      "step": 275
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.826598465473146e-05,
+      "loss": 0.3391,
+      "step": 276
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8258312020460358e-05,
+      "loss": 0.55,
+      "step": 277
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8250639386189257e-05,
+      "loss": 0.5672,
+      "step": 278
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.824296675191816e-05,
+      "loss": 0.285,
+      "step": 279
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.823529411764706e-05,
+      "loss": 0.2809,
+      "step": 280
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.8227621483375958e-05,
+      "loss": 0.5552,
+      "step": 281
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821994884910486e-05,
+      "loss": 0.563,
+      "step": 282
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821227621483376e-05,
+      "loss": 0.5628,
+      "step": 283
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.820460358056266e-05,
+      "loss": 0.2548,
+      "step": 284
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.819693094629156e-05,
+      "loss": 0.871,
+      "step": 285
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818925831202046e-05,
+      "loss": 0.8797,
+      "step": 286
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818158567774936e-05,
+      "loss": 0.2453,
+      "step": 287
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8173913043478262e-05,
+      "loss": 0.2464,
+      "step": 288
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.816624040920716e-05,
+      "loss": 0.3134,
+      "step": 289
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8158567774936062e-05,
+      "loss": 0.2431,
+      "step": 290
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8150895140664964e-05,
+      "loss": 0.2308,
+      "step": 291
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8143222506393862e-05,
+      "loss": 0.2173,
+      "step": 292
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8135549872122764e-05,
+      "loss": 0.5665,
+      "step": 293
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8127877237851662e-05,
+      "loss": 0.5592,
+      "step": 294
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8120204603580564e-05,
+      "loss": 0.473,
+      "step": 295
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8112531969309465e-05,
+      "loss": 0.9458,
+      "step": 296
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8104859335038363e-05,
+      "loss": 0.1879,
+      "step": 297
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8097186700767265e-05,
+      "loss": 0.6797,
+      "step": 298
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8089514066496167e-05,
+      "loss": 0.7592,
+      "step": 299
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8081841432225065e-05,
+      "loss": 0.4109,
+      "step": 300
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8074168797953963e-05,
+      "loss": 0.5821,
+      "step": 301
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8066496163682868e-05,
+      "loss": 0.2057,
+      "step": 302
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8058823529411766e-05,
+      "loss": 0.5322,
+      "step": 303
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8051150895140665e-05,
+      "loss": 0.5208,
+      "step": 304
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8043478260869566e-05,
+      "loss": 0.6799,
+      "step": 305
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8035805626598468e-05,
+      "loss": 0.5894,
+      "step": 306
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8028132992327366e-05,
+      "loss": 0.4051,
+      "step": 307
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8020460358056268e-05,
+      "loss": 0.3695,
+      "step": 308
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8012787723785166e-05,
+      "loss": 0.4869,
+      "step": 309
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8005115089514067e-05,
+      "loss": 0.6075,
+      "step": 310
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.799744245524297e-05,
+      "loss": 0.4245,
+      "step": 311
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7989769820971867e-05,
+      "loss": 0.5122,
+      "step": 312
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7982097186700765e-05,
+      "loss": 0.452,
+      "step": 313
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.797442455242967e-05,
+      "loss": 0.9568,
+      "step": 314
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.796675191815857e-05,
+      "loss": 0.5846,
+      "step": 315
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.7959079283887467e-05,
+      "loss": 0.3743,
+      "step": 316
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.795140664961637e-05,
+      "loss": 0.1922,
+      "step": 317
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.794373401534527e-05,
+      "loss": 0.5303,
+      "step": 318
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.793606138107417e-05,
+      "loss": 0.7253,
+      "step": 319
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.792838874680307e-05,
+      "loss": 0.2794,
+      "step": 320
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.7920716112531968e-05,
+      "loss": 0.577,
+      "step": 321
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.791304347826087e-05,
+      "loss": 0.614,
+      "step": 322
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.790537084398977e-05,
+      "loss": 0.5882,
+      "step": 323
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789769820971867e-05,
+      "loss": 0.2503,
+      "step": 324
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789002557544757e-05,
+      "loss": 0.4195,
+      "step": 325
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.7882352941176473e-05,
+      "loss": 0.2959,
+      "step": 326
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.787468030690537e-05,
+      "loss": 0.3732,
+      "step": 327
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.786700767263427e-05,
+      "loss": 0.5729,
+      "step": 328
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7859335038363174e-05,
+      "loss": 0.3179,
+      "step": 329
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7851662404092073e-05,
+      "loss": 0.6823,
+      "step": 330
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.784398976982097e-05,
+      "loss": 0.2663,
+      "step": 331
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7836317135549872e-05,
+      "loss": 0.4436,
+      "step": 332
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7828644501278774e-05,
+      "loss": 0.3089,
+      "step": 333
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7820971867007672e-05,
+      "loss": 0.5888,
+      "step": 334
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7813299232736574e-05,
+      "loss": 0.2115,
+      "step": 335
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7805626598465472e-05,
+      "loss": 0.4005,
+      "step": 336
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7797953964194374e-05,
+      "loss": 0.3097,
+      "step": 337
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7790281329923275e-05,
+      "loss": 0.5847,
+      "step": 338
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7782608695652174e-05,
+      "loss": 1.0164,
+      "step": 339
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7774936061381072e-05,
+      "loss": 1.1321,
+      "step": 340
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7767263427109977e-05,
+      "loss": 0.592,
+      "step": 341
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7759590792838875e-05,
+      "loss": 1.0191,
+      "step": 342
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7751918158567777e-05,
+      "loss": 0.9199,
+      "step": 343
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7744245524296675e-05,
+      "loss": 0.2874,
+      "step": 344
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7736572890025576e-05,
+      "loss": 0.2469,
+      "step": 345
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7728900255754478e-05,
+      "loss": 0.8364,
+      "step": 346
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7721227621483376e-05,
+      "loss": 0.7365,
+      "step": 347
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7713554987212278e-05,
+      "loss": 0.5595,
+      "step": 348
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.770588235294118e-05,
+      "loss": 0.2546,
+      "step": 349
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7698209718670078e-05,
+      "loss": 0.8183,
+      "step": 350
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7690537084398976e-05,
+      "loss": 0.5385,
+      "step": 351
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.768286445012788e-05,
+      "loss": 0.2763,
+      "step": 352
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.767519181585678e-05,
+      "loss": 0.3045,
+      "step": 353
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7667519181585677e-05,
+      "loss": 0.3354,
+      "step": 354
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765984654731458e-05,
+      "loss": 0.9366,
+      "step": 355
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765217391304348e-05,
+      "loss": 0.6212,
+      "step": 356
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.764450127877238e-05,
+      "loss": 0.1161,
+      "step": 357
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.763682864450128e-05,
+      "loss": 0.4929,
+      "step": 358
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762915601023018e-05,
+      "loss": 0.8167,
+      "step": 359
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762148337595908e-05,
+      "loss": 0.2521,
+      "step": 360
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.7613810741687982e-05,
+      "loss": 0.4174,
+      "step": 361
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.760613810741688e-05,
+      "loss": 0.3822,
+      "step": 362
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.759846547314578e-05,
+      "loss": 0.5355,
+      "step": 363
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.7590792838874683e-05,
+      "loss": 0.561,
+      "step": 364
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.758312020460358e-05,
+      "loss": 0.5126,
+      "step": 365
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.757544757033248e-05,
+      "loss": 0.2301,
+      "step": 366
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.756777493606138e-05,
+      "loss": 0.1698,
+      "step": 367
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7560102301790283e-05,
+      "loss": 0.1634,
+      "step": 368
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.755242966751918e-05,
+      "loss": 0.5955,
+      "step": 369
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7544757033248083e-05,
+      "loss": 0.1568,
+      "step": 370
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.753708439897698e-05,
+      "loss": 0.5167,
+      "step": 371
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7529411764705883e-05,
+      "loss": 1.0028,
+      "step": 372
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7521739130434784e-05,
+      "loss": 1.2567,
+      "step": 373
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7514066496163682e-05,
+      "loss": 0.6094,
+      "step": 374
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7506393861892584e-05,
+      "loss": 0.6059,
+      "step": 375
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7498721227621486e-05,
+      "loss": 1.0593,
+      "step": 376
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7491048593350384e-05,
+      "loss": 0.1506,
+      "step": 377
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7483375959079282e-05,
+      "loss": 1.5106,
+      "step": 378
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7475703324808187e-05,
+      "loss": 0.1827,
+      "step": 379
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7468030690537085e-05,
+      "loss": 1.4961,
+      "step": 380
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7460358056265984e-05,
+      "loss": 1.0456,
+      "step": 381
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7452685421994885e-05,
+      "loss": 1.2063,
+      "step": 382
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7445012787723787e-05,
+      "loss": 0.5837,
+      "step": 383
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7437340153452685e-05,
+      "loss": 1.4536,
+      "step": 384
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7429667519181587e-05,
+      "loss": 0.5803,
+      "step": 385
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7421994884910485e-05,
+      "loss": 0.5896,
+      "step": 386
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7414322250639386e-05,
+      "loss": 0.1822,
+      "step": 387
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7406649616368288e-05,
+      "loss": 1.7893,
+      "step": 388
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7398976982097186e-05,
+      "loss": 0.9723,
+      "step": 389
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7391304347826085e-05,
+      "loss": 1.2815,
+      "step": 390
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.738363171355499e-05,
+      "loss": 0.5765,
+      "step": 391
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7375959079283888e-05,
+      "loss": 1.2429,
+      "step": 392
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7368286445012786e-05,
+      "loss": 0.1343,
+      "step": 393
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7360613810741688e-05,
+      "loss": 0.2306,
+      "step": 394
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.735294117647059e-05,
+      "loss": 0.624,
+      "step": 395
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.734526854219949e-05,
+      "loss": 0.2371,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7348484848484849,
+      "eval_loss": 0.5804409980773926,
+      "eval_runtime": 57.3336,
+      "eval_samples_per_second": 6.907,
+      "eval_steps_per_second": 1.727,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.733759590792839e-05,
+      "loss": 0.8989,
+      "step": 397
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.732992327365729e-05,
+      "loss": 0.5664,
+      "step": 398
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.7322250639386192e-05,
+      "loss": 0.8852,
+      "step": 399
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.731457800511509e-05,
+      "loss": 0.5661,
+      "step": 400
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.730690537084399e-05,
+      "loss": 0.5648,
+      "step": 401
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7299232736572894e-05,
+      "loss": 0.7333,
+      "step": 402
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7291560102301792e-05,
+      "loss": 0.3037,
+      "step": 403
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.728388746803069e-05,
+      "loss": 0.1463,
+      "step": 404
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7276214833759592e-05,
+      "loss": 0.2702,
+      "step": 405
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7268542199488493e-05,
+      "loss": 0.2693,
+      "step": 406
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.726086956521739e-05,
+      "loss": 0.5647,
+      "step": 407
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7253196930946293e-05,
+      "loss": 0.8507,
+      "step": 408
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.724552429667519e-05,
+      "loss": 0.8645,
+      "step": 409
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7237851662404093e-05,
+      "loss": 0.2649,
+      "step": 410
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7230179028132995e-05,
+      "loss": 0.6942,
+      "step": 411
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7222506393861893e-05,
+      "loss": 0.5656,
+      "step": 412
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.721483375959079e-05,
+      "loss": 0.5489,
+      "step": 413
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7207161125319696e-05,
+      "loss": 0.2652,
+      "step": 414
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7199488491048594e-05,
+      "loss": 0.8676,
+      "step": 415
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7191815856777493e-05,
+      "loss": 0.8621,
+      "step": 416
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7184143222506394e-05,
+      "loss": 0.2641,
+      "step": 417
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7176470588235296e-05,
+      "loss": 0.859,
+      "step": 418
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7168797953964194e-05,
+      "loss": 0.6492,
+      "step": 419
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7161125319693096e-05,
+      "loss": 1.1494,
+      "step": 420
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7153452685421994e-05,
+      "loss": 0.8278,
+      "step": 421
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7145780051150895e-05,
+      "loss": 0.764,
+      "step": 422
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7138107416879797e-05,
+      "loss": 0.564,
+      "step": 423
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7130434782608695e-05,
+      "loss": 0.8051,
+      "step": 424
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7122762148337597e-05,
+      "loss": 0.3052,
+      "step": 425
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.71150895140665e-05,
+      "loss": 0.2945,
+      "step": 426
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7107416879795397e-05,
+      "loss": 0.4118,
+      "step": 427
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7099744245524295e-05,
+      "loss": 0.559,
+      "step": 428
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.70920716112532e-05,
+      "loss": 0.5593,
+      "step": 429
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7084398976982098e-05,
+      "loss": 0.564,
+      "step": 430
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7076726342710996e-05,
+      "loss": 0.4454,
+      "step": 431
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7069053708439898e-05,
+      "loss": 0.4381,
+      "step": 432
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.70613810741688e-05,
+      "loss": 0.3718,
+      "step": 433
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7053708439897698e-05,
+      "loss": 0.6627,
+      "step": 434
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70460358056266e-05,
+      "loss": 0.4004,
+      "step": 435
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7038363171355498e-05,
+      "loss": 0.3855,
+      "step": 436
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70306905370844e-05,
+      "loss": 0.4799,
+      "step": 437
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70230179028133e-05,
+      "loss": 0.6714,
+      "step": 438
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70153452685422e-05,
+      "loss": 0.622,
+      "step": 439
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7007672634271097e-05,
+      "loss": 0.3916,
+      "step": 440
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.5487,
+      "step": 441
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69923273657289e-05,
+      "loss": 0.376,
+      "step": 442
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69846547314578e-05,
+      "loss": 0.3616,
+      "step": 443
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69769820971867e-05,
+      "loss": 0.8747,
+      "step": 444
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.6969309462915602e-05,
+      "loss": 0.283,
+      "step": 445
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.69616368286445e-05,
+      "loss": 0.3988,
+      "step": 446
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6953964194373402e-05,
+      "loss": 0.4569,
+      "step": 447
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6946291560102303e-05,
+      "loss": 0.2005,
+      "step": 448
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6938618925831205e-05,
+      "loss": 0.3087,
+      "step": 449
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6930946291560103e-05,
+      "loss": 0.5814,
+      "step": 450
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6923273657289e-05,
+      "loss": 0.2995,
+      "step": 451
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6915601023017907e-05,
+      "loss": 0.7806,
+      "step": 452
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6907928388746805e-05,
+      "loss": 0.2666,
+      "step": 453
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6900255754475703e-05,
+      "loss": 0.2571,
+      "step": 454
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6892583120204605e-05,
+      "loss": 0.5931,
+      "step": 455
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6884910485933506e-05,
+      "loss": 0.475,
+      "step": 456
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6877237851662404e-05,
+      "loss": 0.2372,
+      "step": 457
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6869565217391306e-05,
+      "loss": 0.1577,
+      "step": 458
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6861892583120204e-05,
+      "loss": 0.1953,
+      "step": 459
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6854219948849106e-05,
+      "loss": 0.1931,
+      "step": 460
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6846547314578008e-05,
+      "loss": 0.1829,
+      "step": 461
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6838874680306906e-05,
+      "loss": 0.1601,
+      "step": 462
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6831202046035804e-05,
+      "loss": 0.2505,
+      "step": 463
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.682352941176471e-05,
+      "loss": 0.181,
+      "step": 464
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6815856777493607e-05,
+      "loss": 0.5978,
+      "step": 465
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6808184143222505e-05,
+      "loss": 0.5458,
+      "step": 466
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6800511508951407e-05,
+      "loss": 0.1289,
+      "step": 467
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.679283887468031e-05,
+      "loss": 0.4383,
+      "step": 468
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6785166240409207e-05,
+      "loss": 0.5892,
+      "step": 469
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.677749360613811e-05,
+      "loss": 1.3565,
+      "step": 470
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.676982097186701e-05,
+      "loss": 0.7267,
+      "step": 471
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.6762148337595908e-05,
+      "loss": 0.1354,
+      "step": 472
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.675447570332481e-05,
+      "loss": 0.9291,
+      "step": 473
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.6746803069053708e-05,
+      "loss": 0.5265,
+      "step": 474
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673913043478261e-05,
+      "loss": 0.4841,
+      "step": 475
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673145780051151e-05,
+      "loss": 1.2391,
+      "step": 476
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.672378516624041e-05,
+      "loss": 0.4924,
+      "step": 477
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6716112531969308e-05,
+      "loss": 1.0076,
+      "step": 478
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6708439897698213e-05,
+      "loss": 0.6314,
+      "step": 479
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.670076726342711e-05,
+      "loss": 0.4237,
+      "step": 480
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.669309462915601e-05,
+      "loss": 0.1077,
+      "step": 481
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.668542199488491e-05,
+      "loss": 0.3038,
+      "step": 482
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6677749360613812e-05,
+      "loss": 0.6081,
+      "step": 483
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.667007672634271e-05,
+      "loss": 0.1078,
+      "step": 484
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6662404092071612e-05,
+      "loss": 0.4296,
+      "step": 485
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.665473145780051e-05,
+      "loss": 0.3981,
+      "step": 486
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6647058823529412e-05,
+      "loss": 0.5853,
+      "step": 487
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6639386189258314e-05,
+      "loss": 0.802,
+      "step": 488
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6631713554987212e-05,
+      "loss": 0.4576,
+      "step": 489
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.662404092071611e-05,
+      "loss": 0.2682,
+      "step": 490
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6616368286445015e-05,
+      "loss": 0.1991,
+      "step": 491
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6608695652173913e-05,
+      "loss": 0.1409,
+      "step": 492
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.660102301790281e-05,
+      "loss": 0.3869,
+      "step": 493
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6593350383631713e-05,
+      "loss": 0.5301,
+      "step": 494
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6585677749360615e-05,
+      "loss": 0.8116,
+      "step": 495
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6578005115089513e-05,
+      "loss": 0.0984,
+      "step": 496
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6570332480818415e-05,
+      "loss": 0.7773,
+      "step": 497
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6562659846547316e-05,
+      "loss": 1.3522,
+      "step": 498
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6554987212276215e-05,
+      "loss": 0.0862,
+      "step": 499
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6547314578005116e-05,
+      "loss": 0.0894,
+      "step": 500
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6539641943734014e-05,
+      "loss": 0.0885,
+      "step": 501
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.653196930946292e-05,
+      "loss": 0.3031,
+      "step": 502
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6524296675191818e-05,
+      "loss": 0.187,
+      "step": 503
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6516624040920716e-05,
+      "loss": 1.3063,
+      "step": 504
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6508951406649617e-05,
+      "loss": 0.6847,
+      "step": 505
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.650127877237852e-05,
+      "loss": 0.078,
+      "step": 506
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6493606138107417e-05,
+      "loss": 0.1265,
+      "step": 507
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.648593350383632e-05,
+      "loss": 0.0804,
+      "step": 508
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.6478260869565217e-05,
+      "loss": 0.6976,
+      "step": 509
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.647058823529412e-05,
+      "loss": 0.7977,
+      "step": 510
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.646291560102302e-05,
+      "loss": 0.413,
+      "step": 511
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.645524296675192e-05,
+      "loss": 0.6944,
+      "step": 512
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6447570332480817e-05,
+      "loss": 0.7177,
+      "step": 513
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6439897698209722e-05,
+      "loss": 0.0581,
+      "step": 514
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.643222506393862e-05,
+      "loss": 0.9308,
+      "step": 515
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6424552429667518e-05,
+      "loss": 0.2518,
+      "step": 516
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.641687979539642e-05,
+      "loss": 0.1014,
+      "step": 517
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640920716112532e-05,
+      "loss": 0.1641,
+      "step": 518
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640153452685422e-05,
+      "loss": 0.101,
+      "step": 519
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.639386189258312e-05,
+      "loss": 0.1097,
+      "step": 520
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6386189258312023e-05,
+      "loss": 0.987,
+      "step": 521
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.637851662404092e-05,
+      "loss": 0.6465,
+      "step": 522
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6370843989769823e-05,
+      "loss": 0.0914,
+      "step": 523
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.636317135549872e-05,
+      "loss": 0.139,
+      "step": 524
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6355498721227623e-05,
+      "loss": 0.6455,
+      "step": 525
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6347826086956524e-05,
+      "loss": 0.204,
+      "step": 526
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6340153452685422e-05,
+      "loss": 0.1269,
+      "step": 527
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.633248081841432e-05,
+      "loss": 0.2214,
+      "step": 528
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6324808184143226e-05,
+      "loss": 0.0736,
+      "step": 529
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6317135549872124e-05,
+      "loss": 0.0714,
+      "step": 530
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6309462915601022e-05,
+      "loss": 0.1388,
+      "step": 531
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6301790281329924e-05,
+      "loss": 0.7758,
+      "step": 532
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6294117647058825e-05,
+      "loss": 0.3388,
+      "step": 533
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6286445012787724e-05,
+      "loss": 1.1831,
+      "step": 534
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6278772378516625e-05,
+      "loss": 0.1166,
+      "step": 535
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6271099744245523e-05,
+      "loss": 0.2389,
+      "step": 536
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6263427109974425e-05,
+      "loss": 0.1622,
+      "step": 537
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6255754475703327e-05,
+      "loss": 0.1015,
+      "step": 538
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6248081841432225e-05,
+      "loss": 0.0658,
+      "step": 539
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6240409207161123e-05,
+      "loss": 0.0954,
+      "step": 540
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6232736572890028e-05,
+      "loss": 0.1182,
+      "step": 541
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6225063938618926e-05,
+      "loss": 0.0605,
+      "step": 542
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6217391304347824e-05,
+      "loss": 0.9265,
+      "step": 543
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6209718670076726e-05,
+      "loss": 1.2994,
+      "step": 544
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6202046035805628e-05,
+      "loss": 0.081,
+      "step": 545
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6194373401534526e-05,
+      "loss": 0.5962,
+      "step": 546
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6186700767263428e-05,
+      "loss": 0.0665,
+      "step": 547
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.617902813299233e-05,
+      "loss": 0.0655,
+      "step": 548
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6171355498721227e-05,
+      "loss": 0.0642,
+      "step": 549
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.616368286445013e-05,
+      "loss": 0.0834,
+      "step": 550
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6156010230179027e-05,
+      "loss": 0.047,
+      "step": 551
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6148337595907932e-05,
+      "loss": 0.0642,
+      "step": 552
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.614066496163683e-05,
+      "loss": 0.0594,
+      "step": 553
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.613299232736573e-05,
+      "loss": 0.0573,
+      "step": 554
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.612531969309463e-05,
+      "loss": 0.1101,
+      "step": 555
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.6117647058823532e-05,
+      "loss": 0.5121,
+      "step": 556
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.610997442455243e-05,
+      "loss": 0.0374,
+      "step": 557
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.6102301790281332e-05,
+      "loss": 0.0957,
+      "step": 558
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.609462915601023e-05,
+      "loss": 0.0488,
+      "step": 559
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.608695652173913e-05,
+      "loss": 0.0482,
+      "step": 560
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6079283887468033e-05,
+      "loss": 1.2041,
+      "step": 561
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.607161125319693e-05,
+      "loss": 0.8833,
+      "step": 562
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.606393861892583e-05,
+      "loss": 0.1544,
+      "step": 563
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6056265984654735e-05,
+      "loss": 0.046,
+      "step": 564
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6048593350383633e-05,
+      "loss": 0.9073,
+      "step": 565
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.604092071611253e-05,
+      "loss": 0.052,
+      "step": 566
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6033248081841433e-05,
+      "loss": 0.0308,
+      "step": 567
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6025575447570334e-05,
+      "loss": 0.7032,
+      "step": 568
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6017902813299232e-05,
+      "loss": 0.6947,
+      "step": 569
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6010230179028134e-05,
+      "loss": 1.5603,
+      "step": 570
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6002557544757036e-05,
+      "loss": 0.8836,
+      "step": 571
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.5994884910485934e-05,
+      "loss": 1.7439,
+      "step": 572
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5987212276214836e-05,
+      "loss": 0.0313,
+      "step": 573
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5979539641943734e-05,
+      "loss": 0.0317,
+      "step": 574
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5971867007672635e-05,
+      "loss": 0.0325,
+      "step": 575
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5964194373401537e-05,
+      "loss": 0.5,
+      "step": 576
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5956521739130435e-05,
+      "loss": 0.0675,
+      "step": 577
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5948849104859333e-05,
+      "loss": 0.6364,
+      "step": 578
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.594117647058824e-05,
+      "loss": 0.0696,
+      "step": 579
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5933503836317137e-05,
+      "loss": 0.6195,
+      "step": 580
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5925831202046035e-05,
+      "loss": 0.5796,
+      "step": 581
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5918158567774937e-05,
+      "loss": 0.0832,
+      "step": 582
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5910485933503838e-05,
+      "loss": 0.0794,
+      "step": 583
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5902813299232736e-05,
+      "loss": 0.5851,
+      "step": 584
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5895140664961638e-05,
+      "loss": 0.104,
+      "step": 585
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5887468030690536e-05,
+      "loss": 1.0492,
+      "step": 586
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5879795396419438e-05,
+      "loss": 0.0645,
+      "step": 587
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.587212276214834e-05,
+      "loss": 0.514,
+      "step": 588
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5864450127877238e-05,
+      "loss": 0.0701,
+      "step": 589
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5856777493606136e-05,
+      "loss": 0.5409,
+      "step": 590
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584910485933504e-05,
+      "loss": 0.0536,
+      "step": 591
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584143222506394e-05,
+      "loss": 0.1728,
+      "step": 592
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.5833759590792837e-05,
+      "loss": 0.0845,
+      "step": 593
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.582608695652174e-05,
+      "loss": 0.2217,
+      "step": 594
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.581841432225064e-05,
+      "loss": 0.1397,
+      "step": 595
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.581074168797954e-05,
+      "loss": 0.4119,
+      "step": 596
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.580306905370844e-05,
+      "loss": 0.3638,
+      "step": 597
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.5795396419437342e-05,
+      "loss": 0.0781,
+      "step": 598
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.578772378516624e-05,
+      "loss": 0.5926,
+      "step": 599
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5780051150895142e-05,
+      "loss": 0.1378,
+      "step": 600
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.577237851662404e-05,
+      "loss": 0.142,
+      "step": 601
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.576470588235294e-05,
+      "loss": 0.1722,
+      "step": 602
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5757033248081843e-05,
+      "loss": 0.1529,
+      "step": 603
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.574936061381074e-05,
+      "loss": 1.6083,
+      "step": 604
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5741687979539643e-05,
+      "loss": 1.8267,
+      "step": 605
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5734015345268545e-05,
+      "loss": 0.0535,
+      "step": 606
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5726342710997443e-05,
+      "loss": 0.9945,
+      "step": 607
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5718670076726345e-05,
+      "loss": 0.9133,
+      "step": 608
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5710997442455243e-05,
+      "loss": 0.5295,
+      "step": 609
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5703324808184144e-05,
+      "loss": 0.058,
+      "step": 610
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5695652173913046e-05,
+      "loss": 0.4789,
+      "step": 611
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5687979539641944e-05,
+      "loss": 0.0572,
+      "step": 612
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5680306905370842e-05,
+      "loss": 0.0897,
+      "step": 613
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5672634271099747e-05,
+      "loss": 0.056,
+      "step": 614
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5664961636828646e-05,
+      "loss": 0.0278,
+      "step": 615
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5657289002557544e-05,
+      "loss": 0.9947,
+      "step": 616
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5649616368286445e-05,
+      "loss": 0.0627,
+      "step": 617
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5641943734015347e-05,
+      "loss": 0.9335,
+      "step": 618
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5634271099744245e-05,
+      "loss": 0.9793,
+      "step": 619
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5626598465473147e-05,
+      "loss": 1.8128,
+      "step": 620
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.561892583120205e-05,
+      "loss": 1.8015,
+      "step": 621
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5611253196930947e-05,
+      "loss": 0.9259,
+      "step": 622
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.560358056265985e-05,
+      "loss": 0.9618,
+      "step": 623
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5595907928388747e-05,
+      "loss": 0.0808,
+      "step": 624
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5588235294117648e-05,
+      "loss": 1.7161,
+      "step": 625
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.558056265984655e-05,
+      "loss": 0.8879,
+      "step": 626
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5572890025575448e-05,
+      "loss": 0.043,
+      "step": 627
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5565217391304346e-05,
+      "loss": 0.9449,
+      "step": 628
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.555754475703325e-05,
+      "loss": 0.0624,
+      "step": 629
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.554987212276215e-05,
+      "loss": 0.0924,
+      "step": 630
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5542199488491048e-05,
+      "loss": 0.0715,
+      "step": 631
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.553452685421995e-05,
+      "loss": 0.4995,
+      "step": 632
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.552685421994885e-05,
+      "loss": 0.5576,
+      "step": 633
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551918158567775e-05,
+      "loss": 0.06,
+      "step": 634
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551150895140665e-05,
+      "loss": 0.0873,
+      "step": 635
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.550383631713555e-05,
+      "loss": 0.6868,
+      "step": 636
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.549616368286445e-05,
+      "loss": 0.8487,
+      "step": 637
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.5488491048593352e-05,
+      "loss": 0.5641,
+      "step": 638
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.548081841432225e-05,
+      "loss": 1.1343,
+      "step": 639
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.547314578005115e-05,
+      "loss": 0.0614,
+      "step": 640
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5465473145780054e-05,
+      "loss": 0.9414,
+      "step": 641
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5457800511508952e-05,
+      "loss": 0.9226,
+      "step": 642
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.545012787723785e-05,
+      "loss": 0.3323,
+      "step": 643
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5442455242966752e-05,
+      "loss": 0.9315,
+      "step": 644
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5434782608695653e-05,
+      "loss": 0.5539,
+      "step": 645
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.542710997442455e-05,
+      "loss": 0.3376,
+      "step": 646
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5419437340153453e-05,
+      "loss": 0.0694,
+      "step": 647
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5411764705882355e-05,
+      "loss": 0.0706,
+      "step": 648
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5404092071611253e-05,
+      "loss": 0.105,
+      "step": 649
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5396419437340155e-05,
+      "loss": 0.912,
+      "step": 650
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5388746803069053e-05,
+      "loss": 0.0998,
+      "step": 651
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5381074168797954e-05,
+      "loss": 0.2247,
+      "step": 652
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5373401534526856e-05,
+      "loss": 0.2869,
+      "step": 653
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5365728900255754e-05,
+      "loss": 0.532,
+      "step": 654
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5358056265984653e-05,
+      "loss": 0.4723,
+      "step": 655
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5350383631713558e-05,
+      "loss": 0.1178,
+      "step": 656
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5342710997442456e-05,
+      "loss": 0.0566,
+      "step": 657
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5335038363171357e-05,
+      "loss": 0.059,
+      "step": 658
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5327365728900256e-05,
+      "loss": 0.0881,
+      "step": 659
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5319693094629157e-05,
+      "loss": 0.5762,
+      "step": 660
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.531202046035806e-05,
+      "loss": 0.0499,
+      "step": 661
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5304347826086957e-05,
+      "loss": 0.087,
+      "step": 662
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5296675191815855e-05,
+      "loss": 1.0719,
+      "step": 663
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528900255754476e-05,
+      "loss": 0.6158,
+      "step": 664
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528132992327366e-05,
+      "loss": 0.6117,
+      "step": 665
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5273657289002557e-05,
+      "loss": 0.0515,
+      "step": 666
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5265984654731458e-05,
+      "loss": 0.0608,
+      "step": 667
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.525831202046036e-05,
+      "loss": 0.0313,
+      "step": 668
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.5250639386189258e-05,
+      "loss": 0.1449,
+      "step": 669
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.524296675191816e-05,
+      "loss": 0.9883,
+      "step": 670
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.523529411764706e-05,
+      "loss": 0.0859,
+      "step": 671
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.522762148337596e-05,
+      "loss": 0.0561,
+      "step": 672
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521994884910486e-05,
+      "loss": 0.0926,
+      "step": 673
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521227621483376e-05,
+      "loss": 0.1101,
+      "step": 674
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.520460358056266e-05,
+      "loss": 0.068,
+      "step": 675
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5196930946291563e-05,
+      "loss": 0.0773,
+      "step": 676
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518925831202046e-05,
+      "loss": 0.0508,
+      "step": 677
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518158567774936e-05,
+      "loss": 0.0375,
+      "step": 678
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5173913043478264e-05,
+      "loss": 0.93,
+      "step": 679
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5166240409207162e-05,
+      "loss": 0.1507,
+      "step": 680
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.515856777493606e-05,
+      "loss": 0.0917,
+      "step": 681
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5150895140664962e-05,
+      "loss": 0.97,
+      "step": 682
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5143222506393864e-05,
+      "loss": 0.5019,
+      "step": 683
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5135549872122762e-05,
+      "loss": 0.0849,
+      "step": 684
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5127877237851664e-05,
+      "loss": 1.2316,
+      "step": 685
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5120204603580562e-05,
+      "loss": 0.0252,
+      "step": 686
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5112531969309463e-05,
+      "loss": 1.0543,
+      "step": 687
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5104859335038365e-05,
+      "loss": 0.0506,
+      "step": 688
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5097186700767263e-05,
+      "loss": 0.0252,
+      "step": 689
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.508951406649616e-05,
+      "loss": 0.6297,
+      "step": 690
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5081841432225066e-05,
+      "loss": 0.4497,
+      "step": 691
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5074168797953965e-05,
+      "loss": 0.0516,
+      "step": 692
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5066496163682863e-05,
+      "loss": 0.5501,
+      "step": 693
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5058823529411768e-05,
+      "loss": 0.5621,
+      "step": 694
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5051150895140666e-05,
+      "loss": 0.0577,
+      "step": 695
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5043478260869564e-05,
+      "loss": 0.0271,
+      "step": 696
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5035805626598466e-05,
+      "loss": 0.9496,
+      "step": 697
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5028132992327368e-05,
+      "loss": 0.0877,
+      "step": 698
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5020460358056266e-05,
+      "loss": 0.0523,
+      "step": 699
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5012787723785167e-05,
+      "loss": 0.0577,
+      "step": 700
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5005115089514066e-05,
+      "loss": 0.0522,
+      "step": 701
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.4997442455242967e-05,
+      "loss": 0.0517,
+      "step": 702
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.498976982097187e-05,
+      "loss": 0.6335,
+      "step": 703
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4982097186700767e-05,
+      "loss": 0.05,
+      "step": 704
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4974424552429665e-05,
+      "loss": 0.0753,
+      "step": 705
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.496675191815857e-05,
+      "loss": 0.0227,
+      "step": 706
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.495907928388747e-05,
+      "loss": 0.0755,
+      "step": 707
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.4951406649616367e-05,
+      "loss": 0.0215,
+      "step": 708
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.494373401534527e-05,
+      "loss": 0.0451,
+      "step": 709
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.493606138107417e-05,
+      "loss": 0.0343,
+      "step": 710
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492838874680307e-05,
+      "loss": 0.0836,
+      "step": 711
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492071611253197e-05,
+      "loss": 0.5287,
+      "step": 712
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4913043478260868e-05,
+      "loss": 0.0686,
+      "step": 713
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4905370843989773e-05,
+      "loss": 0.0211,
+      "step": 714
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489769820971867e-05,
+      "loss": 0.5932,
+      "step": 715
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489002557544757e-05,
+      "loss": 0.0371,
+      "step": 716
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.488235294117647e-05,
+      "loss": 1.0137,
+      "step": 717
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.4874680306905373e-05,
+      "loss": 0.9928,
+      "step": 718
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.486700767263427e-05,
+      "loss": 0.7431,
+      "step": 719
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4859335038363173e-05,
+      "loss": 0.0207,
+      "step": 720
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4851662404092074e-05,
+      "loss": 0.0681,
+      "step": 721
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4843989769820972e-05,
+      "loss": 0.0306,
+      "step": 722
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4836317135549874e-05,
+      "loss": 0.0204,
+      "step": 723
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4828644501278772e-05,
+      "loss": 0.0344,
+      "step": 724
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4820971867007674e-05,
+      "loss": 0.0345,
+      "step": 725
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4813299232736575e-05,
+      "loss": 0.9275,
+      "step": 726
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4805626598465474e-05,
+      "loss": 0.0196,
+      "step": 727
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4797953964194372e-05,
+      "loss": 1.0137,
+      "step": 728
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4790281329923277e-05,
+      "loss": 0.0318,
+      "step": 729
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4782608695652175e-05,
+      "loss": 0.0688,
+      "step": 730
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4774936061381073e-05,
+      "loss": 0.0314,
+      "step": 731
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4767263427109975e-05,
+      "loss": 0.7135,
+      "step": 732
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4759590792838877e-05,
+      "loss": 0.0299,
+      "step": 733
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4751918158567775e-05,
+      "loss": 1.0127,
+      "step": 734
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4744245524296676e-05,
+      "loss": 0.0304,
+      "step": 735
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4736572890025575e-05,
+      "loss": 1.4618,
+      "step": 736
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4728900255754476e-05,
+      "loss": 0.7167,
+      "step": 737
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4721227621483378e-05,
+      "loss": 0.0307,
+      "step": 738
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4713554987212276e-05,
+      "loss": 0.052,
+      "step": 739
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4705882352941174e-05,
+      "loss": 0.778,
+      "step": 740
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.469820971867008e-05,
+      "loss": 0.5482,
+      "step": 741
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4690537084398978e-05,
+      "loss": 1.4314,
+      "step": 742
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4682864450127876e-05,
+      "loss": 0.0422,
+      "step": 743
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.467519181585678e-05,
+      "loss": 0.6614,
+      "step": 744
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.466751918158568e-05,
+      "loss": 0.0312,
+      "step": 745
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4659846547314577e-05,
+      "loss": 0.043,
+      "step": 746
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.465217391304348e-05,
+      "loss": 0.0199,
+      "step": 747
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.464450127877238e-05,
+      "loss": 0.0832,
+      "step": 748
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.463682864450128e-05,
+      "loss": 0.0312,
+      "step": 749
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.462915601023018e-05,
+      "loss": 0.0309,
+      "step": 750
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.462148337595908e-05,
+      "loss": 0.0193,
+      "step": 751
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.461381074168798e-05,
+      "loss": 0.0446,
+      "step": 752
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.4606138107416882e-05,
+      "loss": 0.9993,
+      "step": 753
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.459846547314578e-05,
+      "loss": 0.016,
+      "step": 754
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4590792838874678e-05,
+      "loss": 0.0433,
+      "step": 755
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4583120204603583e-05,
+      "loss": 0.2614,
+      "step": 756
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.457544757033248e-05,
+      "loss": 0.018,
+      "step": 757
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.456777493606138e-05,
+      "loss": 0.6676,
+      "step": 758
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.456010230179028e-05,
+      "loss": 0.0599,
+      "step": 759
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4552429667519183e-05,
+      "loss": 0.0338,
+      "step": 760
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.454475703324808e-05,
+      "loss": 0.0268,
+      "step": 761
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4537084398976983e-05,
+      "loss": 0.0262,
+      "step": 762
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.452941176470588e-05,
+      "loss": 0.0252,
+      "step": 763
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4521739130434786e-05,
+      "loss": 0.0377,
+      "step": 764
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4514066496163684e-05,
+      "loss": 1.4866,
+      "step": 765
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4506393861892582e-05,
+      "loss": 1.0524,
+      "step": 766
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4498721227621484e-05,
+      "loss": 0.0328,
+      "step": 767
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4491048593350386e-05,
+      "loss": 0.024,
+      "step": 768
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4483375959079284e-05,
+      "loss": 0.022,
+      "step": 769
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4475703324808185e-05,
+      "loss": 0.0245,
+      "step": 770
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4468030690537087e-05,
+      "loss": 1.0107,
+      "step": 771
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4460358056265985e-05,
+      "loss": 0.0336,
+      "step": 772
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4452685421994887e-05,
+      "loss": 0.0391,
+      "step": 773
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4445012787723785e-05,
+      "loss": 0.0303,
+      "step": 774
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4437340153452687e-05,
+      "loss": 0.0443,
+      "step": 775
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4429667519181588e-05,
+      "loss": 0.0292,
+      "step": 776
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4421994884910487e-05,
+      "loss": 0.0284,
+      "step": 777
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4414322250639385e-05,
+      "loss": 1.5985,
+      "step": 778
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.440664961636829e-05,
+      "loss": 0.0114,
+      "step": 779
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4398976982097188e-05,
+      "loss": 0.0342,
+      "step": 780
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4391304347826086e-05,
+      "loss": 1.8501,
+      "step": 781
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4383631713554988e-05,
+      "loss": 0.5375,
+      "step": 782
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.437595907928389e-05,
+      "loss": 1.589,
+      "step": 783
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4368286445012788e-05,
+      "loss": 0.0221,
+      "step": 784
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.436061381074169e-05,
+      "loss": 0.0453,
+      "step": 785
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4352941176470587e-05,
+      "loss": 0.796,
+      "step": 786
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.434526854219949e-05,
+      "loss": 0.1249,
+      "step": 787
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.433759590792839e-05,
+      "loss": 0.7963,
+      "step": 788
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.432992327365729e-05,
+      "loss": 0.4231,
+      "step": 789
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.4322250639386187e-05,
+      "loss": 0.0247,
+      "step": 790
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.4314578005115092e-05,
+      "loss": 0.0295,
+      "step": 791
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.430690537084399e-05,
+      "loss": 0.0249,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.3537317216396332,
+      "eval_runtime": 58.1226,
+      "eval_samples_per_second": 6.813,
+      "eval_steps_per_second": 1.703,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.429923273657289e-05,
+      "loss": 0.0157,
+      "step": 793
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4291560102301794e-05,
+      "loss": 0.5222,
+      "step": 794
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4283887468030692e-05,
+      "loss": 0.0226,
+      "step": 795
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.427621483375959e-05,
+      "loss": 0.0293,
+      "step": 796
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.426854219948849e-05,
+      "loss": 0.015,
+      "step": 797
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4260869565217393e-05,
+      "loss": 1.0658,
+      "step": 798
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.425319693094629e-05,
+      "loss": 0.0412,
+      "step": 799
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4245524296675193e-05,
+      "loss": 0.0219,
+      "step": 800
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.423785166240409e-05,
+      "loss": 0.0149,
+      "step": 801
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4230179028132993e-05,
+      "loss": 0.0323,
+      "step": 802
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4222506393861895e-05,
+      "loss": 1.0764,
+      "step": 803
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4214833759590793e-05,
+      "loss": 0.7518,
+      "step": 804
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.420716112531969e-05,
+      "loss": 0.8166,
+      "step": 805
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4199488491048596e-05,
+      "loss": 0.0219,
+      "step": 806
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4191815856777494e-05,
+      "loss": 0.0154,
+      "step": 807
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4184143222506392e-05,
+      "loss": 0.0163,
+      "step": 808
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4176470588235294e-05,
+      "loss": 1.0562,
+      "step": 809
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4168797953964196e-05,
+      "loss": 0.0218,
+      "step": 810
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4161125319693094e-05,
+      "loss": 0.0232,
+      "step": 811
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4153452685421995e-05,
+      "loss": 0.2384,
+      "step": 812
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4145780051150894e-05,
+      "loss": 1.5286,
+      "step": 813
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4138107416879795e-05,
+      "loss": 0.0262,
+      "step": 814
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4130434782608697e-05,
+      "loss": 0.0334,
+      "step": 815
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4122762148337595e-05,
+      "loss": 0.6509,
+      "step": 816
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4115089514066497e-05,
+      "loss": 0.0159,
+      "step": 817
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.41074168797954e-05,
+      "loss": 0.6225,
+      "step": 818
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4099744245524297e-05,
+      "loss": 0.752,
+      "step": 819
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4092071611253198e-05,
+      "loss": 1.0594,
+      "step": 820
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.40843989769821e-05,
+      "loss": 0.0161,
+      "step": 821
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4076726342710998e-05,
+      "loss": 0.7492,
+      "step": 822
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40690537084399e-05,
+      "loss": 0.0302,
+      "step": 823
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4061381074168798e-05,
+      "loss": 0.038,
+      "step": 824
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40537084398977e-05,
+      "loss": 0.0243,
+      "step": 825
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40460358056266e-05,
+      "loss": 0.0331,
+      "step": 826
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40383631713555e-05,
+      "loss": 1.0252,
+      "step": 827
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4030690537084398e-05,
+      "loss": 0.0172,
+      "step": 828
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4023017902813303e-05,
+      "loss": 0.6168,
+      "step": 829
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40153452685422e-05,
+      "loss": 0.3135,
+      "step": 830
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40076726342711e-05,
+      "loss": 0.0168,
+      "step": 831
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.0172,
+      "step": 832
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.3992327365728902e-05,
+      "loss": 1.0349,
+      "step": 833
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39846547314578e-05,
+      "loss": 0.0978,
+      "step": 834
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3976982097186702e-05,
+      "loss": 0.0167,
+      "step": 835
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39693094629156e-05,
+      "loss": 1.0381,
+      "step": 836
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3961636828644502e-05,
+      "loss": 0.6857,
+      "step": 837
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3953964194373404e-05,
+      "loss": 1.0406,
+      "step": 838
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3946291560102302e-05,
+      "loss": 0.071,
+      "step": 839
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.39386189258312e-05,
+      "loss": 0.0193,
+      "step": 840
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3930946291560105e-05,
+      "loss": 0.0193,
+      "step": 841
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3923273657289003e-05,
+      "loss": 0.9916,
+      "step": 842
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.39156010230179e-05,
+      "loss": 0.0723,
+      "step": 843
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3907928388746806e-05,
+      "loss": 0.0626,
+      "step": 844
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3900255754475705e-05,
+      "loss": 0.7264,
+      "step": 845
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3892583120204603e-05,
+      "loss": 0.827,
+      "step": 846
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3884910485933504e-05,
+      "loss": 1.2193,
+      "step": 847
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3877237851662406e-05,
+      "loss": 0.6754,
+      "step": 848
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3869565217391304e-05,
+      "loss": 0.6856,
+      "step": 849
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3861892583120206e-05,
+      "loss": 0.7441,
+      "step": 850
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3854219948849104e-05,
+      "loss": 0.023,
+      "step": 851
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3846547314578006e-05,
+      "loss": 0.4742,
+      "step": 852
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3838874680306907e-05,
+      "loss": 0.0363,
+      "step": 853
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3831202046035806e-05,
+      "loss": 0.0409,
+      "step": 854
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3823529411764704e-05,
+      "loss": 0.8353,
+      "step": 855
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.381585677749361e-05,
+      "loss": 0.0333,
+      "step": 856
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3808184143222507e-05,
+      "loss": 0.0216,
+      "step": 857
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3800511508951405e-05,
+      "loss": 0.9688,
+      "step": 858
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3792838874680307e-05,
+      "loss": 1.0078,
+      "step": 859
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.378516624040921e-05,
+      "loss": 0.0183,
+      "step": 860
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3777493606138107e-05,
+      "loss": 0.0183,
+      "step": 861
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3769820971867008e-05,
+      "loss": 0.9786,
+      "step": 862
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3762148337595907e-05,
+      "loss": 0.9166,
+      "step": 863
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3754475703324808e-05,
+      "loss": 1.1408,
+      "step": 864
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.374680306905371e-05,
+      "loss": 1.1394,
+      "step": 865
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.3739130434782608e-05,
+      "loss": 0.0185,
+      "step": 866
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.373145780051151e-05,
+      "loss": 0.9979,
+      "step": 867
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.372378516624041e-05,
+      "loss": 1.0092,
+      "step": 868
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.371611253196931e-05,
+      "loss": 0.7276,
+      "step": 869
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.370843989769821e-05,
+      "loss": 0.0359,
+      "step": 870
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3700767263427113e-05,
+      "loss": 0.0864,
+      "step": 871
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.369309462915601e-05,
+      "loss": 0.0246,
+      "step": 872
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3685421994884912e-05,
+      "loss": 0.8518,
+      "step": 873
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.367774936061381e-05,
+      "loss": 1.433,
+      "step": 874
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3670076726342712e-05,
+      "loss": 0.035,
+      "step": 875
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3662404092071614e-05,
+      "loss": 0.5922,
+      "step": 876
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3654731457800512e-05,
+      "loss": 0.529,
+      "step": 877
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.364705882352941e-05,
+      "loss": 0.6423,
+      "step": 878
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3639386189258315e-05,
+      "loss": 0.0337,
+      "step": 879
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3631713554987214e-05,
+      "loss": 0.0402,
+      "step": 880
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3624040920716112e-05,
+      "loss": 0.0782,
+      "step": 881
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3616368286445013e-05,
+      "loss": 0.4141,
+      "step": 882
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3608695652173915e-05,
+      "loss": 0.0359,
+      "step": 883
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3601023017902813e-05,
+      "loss": 0.4509,
+      "step": 884
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3593350383631715e-05,
+      "loss": 0.6414,
+      "step": 885
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3585677749360613e-05,
+      "loss": 0.6873,
+      "step": 886
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3578005115089515e-05,
+      "loss": 0.0304,
+      "step": 887
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3570332480818416e-05,
+      "loss": 0.0315,
+      "step": 888
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3562659846547315e-05,
+      "loss": 0.0896,
+      "step": 889
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3554987212276213e-05,
+      "loss": 0.0201,
+      "step": 890
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3547314578005118e-05,
+      "loss": 0.027,
+      "step": 891
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3539641943734016e-05,
+      "loss": 1.9853,
+      "step": 892
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3531969309462914e-05,
+      "loss": 2.9647,
+      "step": 893
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.352429667519182e-05,
+      "loss": 1.9403,
+      "step": 894
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3516624040920717e-05,
+      "loss": 0.0193,
+      "step": 895
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3508951406649616e-05,
+      "loss": 0.9912,
+      "step": 896
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3501278772378517e-05,
+      "loss": 0.02,
+      "step": 897
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.349360613810742e-05,
+      "loss": 0.9015,
+      "step": 898
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3485933503836317e-05,
+      "loss": 0.1294,
+      "step": 899
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.347826086956522e-05,
+      "loss": 0.0559,
+      "step": 900
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.3470588235294117e-05,
+      "loss": 1.8039,
+      "step": 901
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.346291560102302e-05,
+      "loss": 0.0336,
+      "step": 902
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.345524296675192e-05,
+      "loss": 0.0383,
+      "step": 903
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.344757033248082e-05,
+      "loss": 0.0722,
+      "step": 904
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3439897698209717e-05,
+      "loss": 0.0278,
+      "step": 905
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.343222506393862e-05,
+      "loss": 0.0684,
+      "step": 906
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.342455242966752e-05,
+      "loss": 0.0236,
+      "step": 907
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3416879795396418e-05,
+      "loss": 0.0348,
+      "step": 908
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340920716112532e-05,
+      "loss": 0.0204,
+      "step": 909
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340153452685422e-05,
+      "loss": 0.8109,
+      "step": 910
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.339386189258312e-05,
+      "loss": 0.0279,
+      "step": 911
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.338618925831202e-05,
+      "loss": 0.0361,
+      "step": 912
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337851662404092e-05,
+      "loss": 0.0355,
+      "step": 913
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337084398976982e-05,
+      "loss": 0.7958,
+      "step": 914
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.3363171355498723e-05,
+      "loss": 0.0362,
+      "step": 915
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.335549872122762e-05,
+      "loss": 0.0252,
+      "step": 916
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3347826086956522e-05,
+      "loss": 0.8165,
+      "step": 917
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3340153452685424e-05,
+      "loss": 0.0237,
+      "step": 918
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3332480818414322e-05,
+      "loss": 0.8162,
+      "step": 919
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.332480818414322e-05,
+      "loss": 0.7786,
+      "step": 920
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3317135549872125e-05,
+      "loss": 0.0787,
+      "step": 921
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3309462915601024e-05,
+      "loss": 0.1764,
+      "step": 922
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3301790281329925e-05,
+      "loss": 0.7449,
+      "step": 923
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3294117647058824e-05,
+      "loss": 0.0452,
+      "step": 924
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3286445012787725e-05,
+      "loss": 0.0181,
+      "step": 925
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3278772378516627e-05,
+      "loss": 0.1783,
+      "step": 926
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3271099744245525e-05,
+      "loss": 0.5158,
+      "step": 927
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3263427109974423e-05,
+      "loss": 0.7574,
+      "step": 928
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3255754475703328e-05,
+      "loss": 2.6099,
+      "step": 929
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3248081841432226e-05,
+      "loss": 0.0304,
+      "step": 930
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3240409207161125e-05,
+      "loss": 0.0795,
+      "step": 931
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3232736572890026e-05,
+      "loss": 0.0342,
+      "step": 932
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3225063938618928e-05,
+      "loss": 0.0473,
+      "step": 933
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3217391304347826e-05,
+      "loss": 0.0831,
+      "step": 934
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3209718670076728e-05,
+      "loss": 0.0465,
+      "step": 935
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3202046035805626e-05,
+      "loss": 0.6769,
+      "step": 936
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3194373401534528e-05,
+      "loss": 0.0173,
+      "step": 937
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.318670076726343e-05,
+      "loss": 0.0169,
+      "step": 938
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3179028132992327e-05,
+      "loss": 0.0469,
+      "step": 939
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3171355498721226e-05,
+      "loss": 0.048,
+      "step": 940
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.316368286445013e-05,
+      "loss": 1.0704,
+      "step": 941
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.315601023017903e-05,
+      "loss": 1.0738,
+      "step": 942
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3148337595907927e-05,
+      "loss": 0.0163,
+      "step": 943
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3140664961636832e-05,
+      "loss": 2.0702,
+      "step": 944
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.313299232736573e-05,
+      "loss": 1.0433,
+      "step": 945
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.312531969309463e-05,
+      "loss": 1.0351,
+      "step": 946
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.311764705882353e-05,
+      "loss": 1.053,
+      "step": 947
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.3109974424552432e-05,
+      "loss": 0.1797,
+      "step": 948
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.310230179028133e-05,
+      "loss": 1.0259,
+      "step": 949
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.309462915601023e-05,
+      "loss": 0.03,
+      "step": 950
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.308695652173913e-05,
+      "loss": 0.905,
+      "step": 951
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.307928388746803e-05,
+      "loss": 0.0699,
+      "step": 952
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3071611253196933e-05,
+      "loss": 1.0013,
+      "step": 953
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.306393861892583e-05,
+      "loss": 0.0319,
+      "step": 954
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.305626598465473e-05,
+      "loss": 0.0361,
+      "step": 955
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3048593350383634e-05,
+      "loss": 0.0201,
+      "step": 956
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3040920716112533e-05,
+      "loss": 0.033,
+      "step": 957
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.303324808184143e-05,
+      "loss": 0.0411,
+      "step": 958
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3025575447570332e-05,
+      "loss": 0.7138,
+      "step": 959
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3017902813299234e-05,
+      "loss": 0.0434,
+      "step": 960
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3010230179028132e-05,
+      "loss": 0.8445,
+      "step": 961
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3002557544757034e-05,
+      "loss": 0.0323,
+      "step": 962
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2994884910485932e-05,
+      "loss": 0.0203,
+      "step": 963
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2987212276214834e-05,
+      "loss": 0.2302,
+      "step": 964
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2979539641943735e-05,
+      "loss": 0.0293,
+      "step": 965
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2971867007672634e-05,
+      "loss": 0.0593,
+      "step": 966
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2964194373401535e-05,
+      "loss": 0.6505,
+      "step": 967
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2956521739130437e-05,
+      "loss": 0.5891,
+      "step": 968
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2948849104859335e-05,
+      "loss": 0.0191,
+      "step": 969
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2941176470588233e-05,
+      "loss": 0.0188,
+      "step": 970
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2933503836317138e-05,
+      "loss": 0.9537,
+      "step": 971
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2925831202046037e-05,
+      "loss": 0.0186,
+      "step": 972
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2918158567774935e-05,
+      "loss": 1.0237,
+      "step": 973
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2910485933503836e-05,
+      "loss": 0.0417,
+      "step": 974
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2902813299232738e-05,
+      "loss": 0.0369,
+      "step": 975
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.289514066496164e-05,
+      "loss": 0.0184,
+      "step": 976
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2887468030690538e-05,
+      "loss": 0.811,
+      "step": 977
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2879795396419436e-05,
+      "loss": 0.0183,
+      "step": 978
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.287212276214834e-05,
+      "loss": 1.0442,
+      "step": 979
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.286445012787724e-05,
+      "loss": 0.027,
+      "step": 980
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.2856777493606137e-05,
+      "loss": 0.062,
+      "step": 981
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284910485933504e-05,
+      "loss": 0.0186,
+      "step": 982
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284143222506394e-05,
+      "loss": 0.6057,
+      "step": 983
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.283375959079284e-05,
+      "loss": 1.6594,
+      "step": 984
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.282608695652174e-05,
+      "loss": 0.0192,
+      "step": 985
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281841432225064e-05,
+      "loss": 1.5361,
+      "step": 986
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281074168797954e-05,
+      "loss": 0.0384,
+      "step": 987
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.2803069053708442e-05,
+      "loss": 2.0238,
+      "step": 988
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.279539641943734e-05,
+      "loss": 0.0334,
+      "step": 989
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2787723785166242e-05,
+      "loss": 0.0571,
+      "step": 990
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2780051150895143e-05,
+      "loss": 0.0211,
+      "step": 991
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.277237851662404e-05,
+      "loss": 0.0409,
+      "step": 992
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.276470588235294e-05,
+      "loss": 0.1059,
+      "step": 993
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2757033248081845e-05,
+      "loss": 0.0277,
+      "step": 994
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2749360613810743e-05,
+      "loss": 1.0023,
+      "step": 995
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.274168797953964e-05,
+      "loss": 0.0186,
+      "step": 996
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2734015345268543e-05,
+      "loss": 0.0345,
+      "step": 997
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2726342710997445e-05,
+      "loss": 0.0169,
+      "step": 998
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2718670076726343e-05,
+      "loss": 0.7292,
+      "step": 999
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2710997442455244e-05,
+      "loss": 1.2655,
+      "step": 1000
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2703324808184143e-05,
+      "loss": 0.03,
+      "step": 1001
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2695652173913044e-05,
+      "loss": 1.0266,
+      "step": 1002
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2687979539641946e-05,
+      "loss": 0.7645,
+      "step": 1003
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2680306905370844e-05,
+      "loss": 0.7985,
+      "step": 1004
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2672634271099742e-05,
+      "loss": 0.76,
+      "step": 1005
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2664961636828647e-05,
+      "loss": 0.7577,
+      "step": 1006
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2657289002557545e-05,
+      "loss": 0.0248,
+      "step": 1007
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2649616368286444e-05,
+      "loss": 0.0262,
+      "step": 1008
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2641943734015345e-05,
+      "loss": 1.9984,
+      "step": 1009
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2634271099744247e-05,
+      "loss": 0.1489,
+      "step": 1010
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2626598465473145e-05,
+      "loss": 0.0683,
+      "step": 1011
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2618925831202047e-05,
+      "loss": 0.054,
+      "step": 1012
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2611253196930945e-05,
+      "loss": 0.0269,
+      "step": 1013
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2603580562659847e-05,
+      "loss": 1.0378,
+      "step": 1014
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2595907928388748e-05,
+      "loss": 0.0171,
+      "step": 1015
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2588235294117646e-05,
+      "loss": 0.8433,
+      "step": 1016
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2580562659846548e-05,
+      "loss": 0.0468,
+      "step": 1017
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.257289002557545e-05,
+      "loss": 0.0331,
+      "step": 1018
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2565217391304348e-05,
+      "loss": 0.0308,
+      "step": 1019
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2557544757033246e-05,
+      "loss": 1.0007,
+      "step": 1020
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254987212276215e-05,
+      "loss": 2.0242,
+      "step": 1021
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254219948849105e-05,
+      "loss": 0.7713,
+      "step": 1022
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2534526854219948e-05,
+      "loss": 1.1112,
+      "step": 1023
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.252685421994885e-05,
+      "loss": 0.0184,
+      "step": 1024
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251918158567775e-05,
+      "loss": 0.0243,
+      "step": 1025
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251150895140665e-05,
+      "loss": 0.046,
+      "step": 1026
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.250383631713555e-05,
+      "loss": 0.0378,
+      "step": 1027
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.249616368286445e-05,
+      "loss": 0.0329,
+      "step": 1028
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2488491048593354e-05,
+      "loss": 0.0245,
+      "step": 1029
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2480818414322252e-05,
+      "loss": 0.0238,
+      "step": 1030
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.247314578005115e-05,
+      "loss": 1.0628,
+      "step": 1031
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2465473145780052e-05,
+      "loss": 0.0703,
+      "step": 1032
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2457800511508954e-05,
+      "loss": 0.0251,
+      "step": 1033
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2450127877237852e-05,
+      "loss": 1.0561,
+      "step": 1034
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2442455242966753e-05,
+      "loss": 0.0274,
+      "step": 1035
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.243478260869565e-05,
+      "loss": 0.023,
+      "step": 1036
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2427109974424553e-05,
+      "loss": 0.0223,
+      "step": 1037
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2419437340153455e-05,
+      "loss": 0.8113,
+      "step": 1038
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2411764705882353e-05,
+      "loss": 0.0346,
+      "step": 1039
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2404092071611255e-05,
+      "loss": 0.0164,
+      "step": 1040
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2396419437340156e-05,
+      "loss": 0.0237,
+      "step": 1041
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2388746803069054e-05,
+      "loss": 0.0305,
+      "step": 1042
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2381074168797953e-05,
+      "loss": 0.0158,
+      "step": 1043
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2373401534526858e-05,
+      "loss": 0.0225,
+      "step": 1044
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2365728900255756e-05,
+      "loss": 0.0226,
+      "step": 1045
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2358056265984654e-05,
+      "loss": 0.015,
+      "step": 1046
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2350383631713556e-05,
+      "loss": 0.0219,
+      "step": 1047
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2342710997442457e-05,
+      "loss": 1.0694,
+      "step": 1048
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2335038363171356e-05,
+      "loss": 0.0348,
+      "step": 1049
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2327365728900257e-05,
+      "loss": 0.0213,
+      "step": 1050
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2319693094629155e-05,
+      "loss": 0.9405,
+      "step": 1051
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2312020460358057e-05,
+      "loss": 0.0252,
+      "step": 1052
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.230434782608696e-05,
+      "loss": 0.0254,
+      "step": 1053
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2296675191815857e-05,
+      "loss": 0.7353,
+      "step": 1054
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2289002557544755e-05,
+      "loss": 0.0253,
+      "step": 1055
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.228132992327366e-05,
+      "loss": 0.8361,
+      "step": 1056
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2273657289002558e-05,
+      "loss": 0.0196,
+      "step": 1057
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2265984654731457e-05,
+      "loss": 0.014,
+      "step": 1058
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2258312020460358e-05,
+      "loss": 0.8297,
+      "step": 1059
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.225063938618926e-05,
+      "loss": 1.0654,
+      "step": 1060
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2242966751918158e-05,
+      "loss": 0.0248,
+      "step": 1061
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.223529411764706e-05,
+      "loss": 0.0226,
+      "step": 1062
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2227621483375958e-05,
+      "loss": 0.0238,
+      "step": 1063
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221994884910486e-05,
+      "loss": 0.0212,
+      "step": 1064
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221227621483376e-05,
+      "loss": 0.0274,
+      "step": 1065
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.220460358056266e-05,
+      "loss": 1.0217,
+      "step": 1066
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.219693094629156e-05,
+      "loss": 0.0252,
+      "step": 1067
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2189258312020462e-05,
+      "loss": 0.0219,
+      "step": 1068
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.218158567774936e-05,
+      "loss": 0.0349,
+      "step": 1069
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.217391304347826e-05,
+      "loss": 0.0152,
+      "step": 1070
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2166240409207164e-05,
+      "loss": 0.021,
+      "step": 1071
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2158567774936062e-05,
+      "loss": 0.015,
+      "step": 1072
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.215089514066496e-05,
+      "loss": 0.0277,
+      "step": 1073
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2143222506393862e-05,
+      "loss": 0.1858,
+      "step": 1074
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2135549872122764e-05,
+      "loss": 0.0136,
+      "step": 1075
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2127877237851662e-05,
+      "loss": 0.0132,
+      "step": 1076
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2120204603580563e-05,
+      "loss": 0.0126,
+      "step": 1077
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.211253196930946e-05,
+      "loss": 0.5171,
+      "step": 1078
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2104859335038363e-05,
+      "loss": 0.0422,
+      "step": 1079
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2097186700767265e-05,
+      "loss": 0.0186,
+      "step": 1080
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2089514066496163e-05,
+      "loss": 0.8412,
+      "step": 1081
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2081841432225065e-05,
+      "loss": 0.0183,
+      "step": 1082
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2074168797953966e-05,
+      "loss": 1.0243,
+      "step": 1083
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2066496163682865e-05,
+      "loss": 0.0212,
+      "step": 1084
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2058823529411766e-05,
+      "loss": 0.2841,
+      "step": 1085
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2051150895140664e-05,
+      "loss": 0.0118,
+      "step": 1086
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2043478260869566e-05,
+      "loss": 0.0247,
+      "step": 1087
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2035805626598468e-05,
+      "loss": 1.5121,
+      "step": 1088
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2028132992327366e-05,
+      "loss": 0.0304,
+      "step": 1089
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2020460358056267e-05,
+      "loss": 0.0329,
+      "step": 1090
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.201278772378517e-05,
+      "loss": 1.1288,
+      "step": 1091
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.2005115089514067e-05,
+      "loss": 0.0145,
+      "step": 1092
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.1997442455242966e-05,
+      "loss": 0.0498,
+      "step": 1093
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.198976982097187e-05,
+      "loss": 0.0244,
+      "step": 1094
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.198209718670077e-05,
+      "loss": 0.0112,
+      "step": 1095
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.1974424552429667e-05,
+      "loss": 0.8642,
+      "step": 1096
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.196675191815857e-05,
+      "loss": 0.8822,
+      "step": 1097
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.195907928388747e-05,
+      "loss": 0.0278,
+      "step": 1098
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.195140664961637e-05,
+      "loss": 0.6954,
+      "step": 1099
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.194373401534527e-05,
+      "loss": 0.0179,
+      "step": 1100
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.1936061381074168e-05,
+      "loss": 0.5607,
+      "step": 1101
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.192838874680307e-05,
+      "loss": 0.0191,
+      "step": 1102
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.192071611253197e-05,
+      "loss": 1.9519,
+      "step": 1103
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.191304347826087e-05,
+      "loss": 0.8347,
+      "step": 1104
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1905370843989768e-05,
+      "loss": 0.0262,
+      "step": 1105
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1897698209718673e-05,
+      "loss": 0.0477,
+      "step": 1106
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.189002557544757e-05,
+      "loss": 0.0227,
+      "step": 1107
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.188235294117647e-05,
+      "loss": 0.0205,
+      "step": 1108
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.187468030690537e-05,
+      "loss": 0.0208,
+      "step": 1109
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.1867007672634273e-05,
+      "loss": 0.0116,
+      "step": 1110
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.185933503836317e-05,
+      "loss": 1.1546,
+      "step": 1111
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1851662404092072e-05,
+      "loss": 0.0114,
+      "step": 1112
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.184398976982097e-05,
+      "loss": 0.0113,
+      "step": 1113
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1836317135549872e-05,
+      "loss": 0.6881,
+      "step": 1114
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1828644501278774e-05,
+      "loss": 0.0294,
+      "step": 1115
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1820971867007672e-05,
+      "loss": 0.1569,
+      "step": 1116
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1813299232736574e-05,
+      "loss": 1.1362,
+      "step": 1117
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1805626598465475e-05,
+      "loss": 0.0196,
+      "step": 1118
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1797953964194374e-05,
+      "loss": 0.0205,
+      "step": 1119
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1790281329923272e-05,
+      "loss": 0.0198,
+      "step": 1120
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1782608695652177e-05,
+      "loss": 0.0113,
+      "step": 1121
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1774936061381075e-05,
+      "loss": 0.02,
+      "step": 1122
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1767263427109973e-05,
+      "loss": 0.0195,
+      "step": 1123
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1759590792838875e-05,
+      "loss": 0.8016,
+      "step": 1124
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1751918158567776e-05,
+      "loss": 0.0122,
+      "step": 1125
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1744245524296675e-05,
+      "loss": 1.1132,
+      "step": 1126
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1736572890025576e-05,
+      "loss": 0.011,
+      "step": 1127
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1728900255754474e-05,
+      "loss": 0.7705,
+      "step": 1128
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1721227621483376e-05,
+      "loss": 1.1467,
+      "step": 1129
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1713554987212278e-05,
+      "loss": 0.0176,
+      "step": 1130
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1705882352941176e-05,
+      "loss": 0.0293,
+      "step": 1131
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1698209718670074e-05,
+      "loss": 0.0208,
+      "step": 1132
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.169053708439898e-05,
+      "loss": 0.0199,
+      "step": 1133
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1682864450127877e-05,
+      "loss": 0.0431,
+      "step": 1134
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.167519181585678e-05,
+      "loss": 1.6066,
+      "step": 1135
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.1667519181585677e-05,
+      "loss": 0.029,
+      "step": 1136
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165984654731458e-05,
+      "loss": 0.029,
+      "step": 1137
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165217391304348e-05,
+      "loss": 1.1749,
+      "step": 1138
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.164450127877238e-05,
+      "loss": 0.0189,
+      "step": 1139
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.163682864450128e-05,
+      "loss": 0.0309,
+      "step": 1140
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.1629156010230182e-05,
+      "loss": 0.0287,
+      "step": 1141
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.162148337595908e-05,
+      "loss": 0.0202,
+      "step": 1142
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.161381074168798e-05,
+      "loss": 0.7889,
+      "step": 1143
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.1606138107416883e-05,
+      "loss": 0.0285,
+      "step": 1144
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159846547314578e-05,
+      "loss": 0.0535,
+      "step": 1145
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159079283887468e-05,
+      "loss": 1.0829,
+      "step": 1146
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.158312020460358e-05,
+      "loss": 0.0989,
+      "step": 1147
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1575447570332483e-05,
+      "loss": 0.0115,
+      "step": 1148
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.156777493606138e-05,
+      "loss": 0.0227,
+      "step": 1149
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1560102301790283e-05,
+      "loss": 0.5506,
+      "step": 1150
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.155242966751918e-05,
+      "loss": 1.1234,
+      "step": 1151
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1544757033248083e-05,
+      "loss": 0.0564,
+      "step": 1152
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1537084398976984e-05,
+      "loss": 0.0127,
+      "step": 1153
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1529411764705882e-05,
+      "loss": 0.0218,
+      "step": 1154
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.152173913043478e-05,
+      "loss": 0.0124,
+      "step": 1155
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1514066496163686e-05,
+      "loss": 0.0223,
+      "step": 1156
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1506393861892584e-05,
+      "loss": 0.0505,
+      "step": 1157
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1498721227621482e-05,
+      "loss": 0.0225,
+      "step": 1158
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1491048593350384e-05,
+      "loss": 0.0308,
+      "step": 1159
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1483375959079285e-05,
+      "loss": 0.036,
+      "step": 1160
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1475703324808184e-05,
+      "loss": 0.0121,
+      "step": 1161
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1468030690537085e-05,
+      "loss": 0.0274,
+      "step": 1162
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1460358056265983e-05,
+      "loss": 1.0397,
+      "step": 1163
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1452685421994885e-05,
+      "loss": 0.0332,
+      "step": 1164
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1445012787723787e-05,
+      "loss": 0.0182,
+      "step": 1165
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1437340153452685e-05,
+      "loss": 0.018,
+      "step": 1166
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1429667519181587e-05,
+      "loss": 0.859,
+      "step": 1167
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1421994884910488e-05,
+      "loss": 0.0321,
+      "step": 1168
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1414322250639386e-05,
+      "loss": 0.0225,
+      "step": 1169
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1406649616368285e-05,
+      "loss": 0.0224,
+      "step": 1170
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.139897698209719e-05,
+      "loss": 0.03,
+      "step": 1171
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1391304347826088e-05,
+      "loss": 0.7473,
+      "step": 1172
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1383631713554986e-05,
+      "loss": 0.8609,
+      "step": 1173
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1375959079283888e-05,
+      "loss": 0.0105,
+      "step": 1174
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.136828644501279e-05,
+      "loss": 0.0217,
+      "step": 1175
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1360613810741687e-05,
+      "loss": 0.0218,
+      "step": 1176
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.135294117647059e-05,
+      "loss": 0.016,
+      "step": 1177
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1345268542199487e-05,
+      "loss": 0.7844,
+      "step": 1178
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.133759590792839e-05,
+      "loss": 0.01,
+      "step": 1179
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132992327365729e-05,
+      "loss": 0.8774,
+      "step": 1180
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132225063938619e-05,
+      "loss": 0.8375,
+      "step": 1181
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.1314578005115087e-05,
+      "loss": 0.0188,
+      "step": 1182
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.1306905370843992e-05,
+      "loss": 1.1026,
+      "step": 1183
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129923273657289e-05,
+      "loss": 1.0447,
+      "step": 1184
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129156010230179e-05,
+      "loss": 0.0101,
+      "step": 1185
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.128388746803069e-05,
+      "loss": 0.0247,
+      "step": 1186
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.127621483375959e-05,
+      "loss": 0.0295,
+      "step": 1187
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.1268542199488493e-05,
+      "loss": 0.0191,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.37390685081481934,
+      "eval_runtime": 57.4087,
+      "eval_samples_per_second": 6.898,
+      "eval_steps_per_second": 1.724,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.126086956521739e-05,
+      "loss": 0.0103,
+      "step": 1189
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1253196930946293e-05,
+      "loss": 0.0231,
+      "step": 1190
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1245524296675195e-05,
+      "loss": 0.0389,
+      "step": 1191
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1237851662404093e-05,
+      "loss": 0.0298,
+      "step": 1192
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.123017902813299e-05,
+      "loss": 0.0196,
+      "step": 1193
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1222506393861896e-05,
+      "loss": 0.9688,
+      "step": 1194
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1214833759590794e-05,
+      "loss": 0.0458,
+      "step": 1195
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1207161125319693e-05,
+      "loss": 0.0161,
+      "step": 1196
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1199488491048594e-05,
+      "loss": 0.0157,
+      "step": 1197
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1191815856777496e-05,
+      "loss": 0.8732,
+      "step": 1198
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1184143222506394e-05,
+      "loss": 0.0103,
+      "step": 1199
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1176470588235296e-05,
+      "loss": 0.0267,
+      "step": 1200
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1168797953964194e-05,
+      "loss": 0.0266,
+      "step": 1201
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1161125319693095e-05,
+      "loss": 0.0157,
+      "step": 1202
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1153452685421997e-05,
+      "loss": 0.265,
+      "step": 1203
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1145780051150895e-05,
+      "loss": 0.0537,
+      "step": 1204
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1138107416879794e-05,
+      "loss": 0.0306,
+      "step": 1205
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.11304347826087e-05,
+      "loss": 1.1714,
+      "step": 1206
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1122762148337597e-05,
+      "loss": 1.1704,
+      "step": 1207
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1115089514066495e-05,
+      "loss": 0.0154,
+      "step": 1208
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1107416879795397e-05,
+      "loss": 0.0195,
+      "step": 1209
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1099744245524298e-05,
+      "loss": 0.9099,
+      "step": 1210
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1092071611253196e-05,
+      "loss": 0.0112,
+      "step": 1211
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1084398976982098e-05,
+      "loss": 0.1506,
+      "step": 1212
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1076726342710996e-05,
+      "loss": 0.0111,
+      "step": 1213
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1069053708439898e-05,
+      "loss": 0.0244,
+      "step": 1214
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10613810741688e-05,
+      "loss": 0.8889,
+      "step": 1215
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1053708439897698e-05,
+      "loss": 0.0146,
+      "step": 1216
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10460358056266e-05,
+      "loss": 0.0189,
+      "step": 1217
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10383631713555e-05,
+      "loss": 0.0097,
+      "step": 1218
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10306905370844e-05,
+      "loss": 1.1683,
+      "step": 1219
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1023017902813297e-05,
+      "loss": 1.1621,
+      "step": 1220
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1015345268542202e-05,
+      "loss": 0.0194,
+      "step": 1221
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.10076726342711e-05,
+      "loss": 0.0155,
+      "step": 1222
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.0296,
+      "step": 1223
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.09923273657289e-05,
+      "loss": 0.02,
+      "step": 1224
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.0984654731457802e-05,
+      "loss": 0.0154,
+      "step": 1225
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09769820971867e-05,
+      "loss": 0.0151,
+      "step": 1226
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0969309462915602e-05,
+      "loss": 0.0223,
+      "step": 1227
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09616368286445e-05,
+      "loss": 1.7159,
+      "step": 1228
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0953964194373402e-05,
+      "loss": 0.0194,
+      "step": 1229
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0946291560102303e-05,
+      "loss": 0.0637,
+      "step": 1230
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09386189258312e-05,
+      "loss": 0.0156,
+      "step": 1231
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09309462915601e-05,
+      "loss": 1.0612,
+      "step": 1232
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0923273657289005e-05,
+      "loss": 0.8929,
+      "step": 1233
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0915601023017903e-05,
+      "loss": 1.1649,
+      "step": 1234
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.09079283887468e-05,
+      "loss": 2.059,
+      "step": 1235
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0900255754475703e-05,
+      "loss": 0.0169,
+      "step": 1236
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0892583120204604e-05,
+      "loss": 0.0158,
+      "step": 1237
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0884910485933503e-05,
+      "loss": 0.0108,
+      "step": 1238
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0877237851662404e-05,
+      "loss": 0.0152,
+      "step": 1239
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0869565217391306e-05,
+      "loss": 0.0109,
+      "step": 1240
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0861892583120208e-05,
+      "loss": 1.1352,
+      "step": 1241
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0854219948849106e-05,
+      "loss": 0.0226,
+      "step": 1242
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0846547314578004e-05,
+      "loss": 0.0154,
+      "step": 1243
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.083887468030691e-05,
+      "loss": 0.0232,
+      "step": 1244
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0831202046035807e-05,
+      "loss": 0.0142,
+      "step": 1245
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0823529411764705e-05,
+      "loss": 0.9143,
+      "step": 1246
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0815856777493607e-05,
+      "loss": 0.0165,
+      "step": 1247
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.080818414322251e-05,
+      "loss": 0.0446,
+      "step": 1248
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0800511508951407e-05,
+      "loss": 0.0121,
+      "step": 1249
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.079283887468031e-05,
+      "loss": 1.0991,
+      "step": 1250
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.0785166240409207e-05,
+      "loss": 0.0159,
+      "step": 1251
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.077749360613811e-05,
+      "loss": 0.0166,
+      "step": 1252
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.076982097186701e-05,
+      "loss": 0.0171,
+      "step": 1253
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0762148337595908e-05,
+      "loss": 0.7986,
+      "step": 1254
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0754475703324806e-05,
+      "loss": 0.0174,
+      "step": 1255
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.074680306905371e-05,
+      "loss": 1.0431,
+      "step": 1256
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.073913043478261e-05,
+      "loss": 0.1766,
+      "step": 1257
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.0731457800511508e-05,
+      "loss": 0.0209,
+      "step": 1258
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.072378516624041e-05,
+      "loss": 0.0178,
+      "step": 1259
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.071611253196931e-05,
+      "loss": 0.0215,
+      "step": 1260
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.070843989769821e-05,
+      "loss": 0.0112,
+      "step": 1261
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.070076726342711e-05,
+      "loss": 0.0166,
+      "step": 1262
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0693094629156012e-05,
+      "loss": 0.0161,
+      "step": 1263
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.068542199488491e-05,
+      "loss": 0.8922,
+      "step": 1264
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0677749360613812e-05,
+      "loss": 0.0133,
+      "step": 1265
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.067007672634271e-05,
+      "loss": 0.0117,
+      "step": 1266
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0662404092071612e-05,
+      "loss": 0.0153,
+      "step": 1267
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0654731457800514e-05,
+      "loss": 0.0153,
+      "step": 1268
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0647058823529412e-05,
+      "loss": 0.0195,
+      "step": 1269
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.063938618925831e-05,
+      "loss": 1.1511,
+      "step": 1270
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0631713554987215e-05,
+      "loss": 0.0108,
+      "step": 1271
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0624040920716113e-05,
+      "loss": 0.8897,
+      "step": 1272
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.061636828644501e-05,
+      "loss": 0.0157,
+      "step": 1273
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0608695652173913e-05,
+      "loss": 0.0108,
+      "step": 1274
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0601023017902815e-05,
+      "loss": 0.0214,
+      "step": 1275
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0593350383631713e-05,
+      "loss": 0.0186,
+      "step": 1276
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0585677749360615e-05,
+      "loss": 0.0182,
+      "step": 1277
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0578005115089513e-05,
+      "loss": 0.0102,
+      "step": 1278
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0570332480818415e-05,
+      "loss": 1.1313,
+      "step": 1279
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0562659846547316e-05,
+      "loss": 0.0103,
+      "step": 1280
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0554987212276214e-05,
+      "loss": 0.0144,
+      "step": 1281
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0547314578005113e-05,
+      "loss": 0.02,
+      "step": 1282
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0539641943734018e-05,
+      "loss": 0.0292,
+      "step": 1283
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0531969309462916e-05,
+      "loss": 0.0105,
+      "step": 1284
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0524296675191814e-05,
+      "loss": 0.01,
+      "step": 1285
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0516624040920716e-05,
+      "loss": 1.1245,
+      "step": 1286
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0508951406649617e-05,
+      "loss": 0.8544,
+      "step": 1287
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0501278772378516e-05,
+      "loss": 0.0149,
+      "step": 1288
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0493606138107417e-05,
+      "loss": 0.0065,
+      "step": 1289
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.048593350383632e-05,
+      "loss": 0.0113,
+      "step": 1290
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0478260869565217e-05,
+      "loss": 0.0094,
+      "step": 1291
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.047058823529412e-05,
+      "loss": 0.0097,
+      "step": 1292
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0462915601023017e-05,
+      "loss": 0.0812,
+      "step": 1293
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0455242966751922e-05,
+      "loss": 0.0099,
+      "step": 1294
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.044757033248082e-05,
+      "loss": 0.013,
+      "step": 1295
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0439897698209718e-05,
+      "loss": 1.584,
+      "step": 1296
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.043222506393862e-05,
+      "loss": 0.0089,
+      "step": 1297
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.042455242966752e-05,
+      "loss": 0.0089,
+      "step": 1298
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.041687979539642e-05,
+      "loss": 0.0093,
+      "step": 1299
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.040920716112532e-05,
+      "loss": 0.0149,
+      "step": 1300
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.040153452685422e-05,
+      "loss": 1.1671,
+      "step": 1301
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.039386189258312e-05,
+      "loss": 0.0086,
+      "step": 1302
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.0386189258312023e-05,
+      "loss": 0.8177,
+      "step": 1303
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.037851662404092e-05,
+      "loss": 0.0128,
+      "step": 1304
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.037084398976982e-05,
+      "loss": 0.0167,
+      "step": 1305
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0363171355498724e-05,
+      "loss": 0.0085,
+      "step": 1306
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0355498721227622e-05,
+      "loss": 0.0085,
+      "step": 1307
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.034782608695652e-05,
+      "loss": 0.0166,
+      "step": 1308
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0340153452685422e-05,
+      "loss": 0.9448,
+      "step": 1309
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0332480818414324e-05,
+      "loss": 0.0082,
+      "step": 1310
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0324808184143222e-05,
+      "loss": 0.0088,
+      "step": 1311
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0317135549872124e-05,
+      "loss": 0.0082,
+      "step": 1312
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0309462915601025e-05,
+      "loss": 0.0146,
+      "step": 1313
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0301790281329924e-05,
+      "loss": 0.0123,
+      "step": 1314
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0294117647058825e-05,
+      "loss": 0.0101,
+      "step": 1315
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0286445012787723e-05,
+      "loss": 1.2167,
+      "step": 1316
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0278772378516625e-05,
+      "loss": 1.0578,
+      "step": 1317
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0271099744245527e-05,
+      "loss": 0.0163,
+      "step": 1318
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0263427109974425e-05,
+      "loss": 0.013,
+      "step": 1319
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0255754475703323e-05,
+      "loss": 0.0082,
+      "step": 1320
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0248081841432228e-05,
+      "loss": 0.0081,
+      "step": 1321
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0240409207161126e-05,
+      "loss": 0.0124,
+      "step": 1322
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0232736572890024e-05,
+      "loss": 0.0206,
+      "step": 1323
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0225063938618926e-05,
+      "loss": 0.0081,
+      "step": 1324
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0217391304347828e-05,
+      "loss": 0.0122,
+      "step": 1325
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0209718670076726e-05,
+      "loss": 0.0122,
+      "step": 1326
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0202046035805628e-05,
+      "loss": 0.0121,
+      "step": 1327
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0194373401534526e-05,
+      "loss": 1.224,
+      "step": 1328
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0186700767263427e-05,
+      "loss": 2.3502,
+      "step": 1329
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.017902813299233e-05,
+      "loss": 0.0242,
+      "step": 1330
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0171355498721227e-05,
+      "loss": 0.0064,
+      "step": 1331
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0163682864450125e-05,
+      "loss": 0.0186,
+      "step": 1332
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.015601023017903e-05,
+      "loss": 0.012,
+      "step": 1333
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.014833759590793e-05,
+      "loss": 0.0119,
+      "step": 1334
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.0140664961636827e-05,
+      "loss": 0.0159,
+      "step": 1335
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.013299232736573e-05,
+      "loss": 0.0199,
+      "step": 1336
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.012531969309463e-05,
+      "loss": 1.2138,
+      "step": 1337
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.011764705882353e-05,
+      "loss": 0.9491,
+      "step": 1338
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010997442455243e-05,
+      "loss": 0.0083,
+      "step": 1339
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010230179028133e-05,
+      "loss": 0.948,
+      "step": 1340
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.009462915601023e-05,
+      "loss": 0.0112,
+      "step": 1341
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.008695652173913e-05,
+      "loss": 0.8921,
+      "step": 1342
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007928388746803e-05,
+      "loss": 0.0194,
+      "step": 1343
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007161125319693e-05,
+      "loss": 1.0273,
+      "step": 1344
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0063938618925833e-05,
+      "loss": 0.013,
+      "step": 1345
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.005626598465473e-05,
+      "loss": 0.0103,
+      "step": 1346
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0048593350383633e-05,
+      "loss": 0.0114,
+      "step": 1347
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0040920716112534e-05,
+      "loss": 0.0199,
+      "step": 1348
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0033248081841433e-05,
+      "loss": 1.183,
+      "step": 1349
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0025575447570334e-05,
+      "loss": 0.0231,
+      "step": 1350
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0017902813299232e-05,
+      "loss": 1.1794,
+      "step": 1351
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0010230179028134e-05,
+      "loss": 0.0096,
+      "step": 1352
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 2.0002557544757036e-05,
+      "loss": 0.9049,
+      "step": 1353
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9994884910485934e-05,
+      "loss": 0.019,
+      "step": 1354
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9987212276214832e-05,
+      "loss": 0.0152,
+      "step": 1355
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9979539641943737e-05,
+      "loss": 0.8834,
+      "step": 1356
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9971867007672635e-05,
+      "loss": 0.01,
+      "step": 1357
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9964194373401533e-05,
+      "loss": 0.0152,
+      "step": 1358
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9956521739130435e-05,
+      "loss": 0.0161,
+      "step": 1359
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9948849104859337e-05,
+      "loss": 1.6778,
+      "step": 1360
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9941176470588235e-05,
+      "loss": 0.0214,
+      "step": 1361
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9933503836317137e-05,
+      "loss": 0.0156,
+      "step": 1362
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9925831202046038e-05,
+      "loss": 0.026,
+      "step": 1363
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9918158567774936e-05,
+      "loss": 0.0259,
+      "step": 1364
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9910485933503838e-05,
+      "loss": 0.0161,
+      "step": 1365
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9902813299232736e-05,
+      "loss": 0.0108,
+      "step": 1366
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9895140664961638e-05,
+      "loss": 0.0149,
+      "step": 1367
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.988746803069054e-05,
+      "loss": 0.0325,
+      "step": 1368
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9879795396419438e-05,
+      "loss": 0.0153,
+      "step": 1369
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9872122762148336e-05,
+      "loss": 1.7678,
+      "step": 1370
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.986445012787724e-05,
+      "loss": 0.0773,
+      "step": 1371
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.985677749360614e-05,
+      "loss": 0.0244,
+      "step": 1372
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.9849104859335037e-05,
+      "loss": 0.0225,
+      "step": 1373
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.984143222506394e-05,
+      "loss": 0.0254,
+      "step": 1374
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.983375959079284e-05,
+      "loss": 0.0205,
+      "step": 1375
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.982608695652174e-05,
+      "loss": 0.0096,
+      "step": 1376
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981841432225064e-05,
+      "loss": 0.0147,
+      "step": 1377
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981074168797954e-05,
+      "loss": 0.0296,
+      "step": 1378
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.980306905370844e-05,
+      "loss": 0.0227,
+      "step": 1379
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.9795396419437342e-05,
+      "loss": 0.899,
+      "step": 1380
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.978772378516624e-05,
+      "loss": 0.019,
+      "step": 1381
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9780051150895138e-05,
+      "loss": 0.0157,
+      "step": 1382
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9772378516624043e-05,
+      "loss": 1.1897,
+      "step": 1383
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.976470588235294e-05,
+      "loss": 0.014,
+      "step": 1384
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.975703324808184e-05,
+      "loss": 0.0208,
+      "step": 1385
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.974936061381074e-05,
+      "loss": 0.0093,
+      "step": 1386
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.9741687979539643e-05,
+      "loss": 0.0244,
+      "step": 1387
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.973401534526854e-05,
+      "loss": 0.028,
+      "step": 1388
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9726342710997443e-05,
+      "loss": 0.0093,
+      "step": 1389
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9718670076726344e-05,
+      "loss": 0.0184,
+      "step": 1390
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9710997442455243e-05,
+      "loss": 0.9462,
+      "step": 1391
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9703324808184144e-05,
+      "loss": 0.0138,
+      "step": 1392
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9695652173913042e-05,
+      "loss": 0.9561,
+      "step": 1393
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9687979539641944e-05,
+      "loss": 0.0152,
+      "step": 1394
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9680306905370846e-05,
+      "loss": 0.0178,
+      "step": 1395
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9672634271099744e-05,
+      "loss": 0.0166,
+      "step": 1396
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9664961636828642e-05,
+      "loss": 0.92,
+      "step": 1397
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9657289002557547e-05,
+      "loss": 0.0253,
+      "step": 1398
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9649616368286445e-05,
+      "loss": 0.0266,
+      "step": 1399
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9641943734015347e-05,
+      "loss": 0.0105,
+      "step": 1400
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9634271099744245e-05,
+      "loss": 0.0208,
+      "step": 1401
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9626598465473147e-05,
+      "loss": 0.8094,
+      "step": 1402
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.961892583120205e-05,
+      "loss": 0.0103,
+      "step": 1403
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9611253196930947e-05,
+      "loss": 0.0179,
+      "step": 1404
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9603580562659845e-05,
+      "loss": 0.0133,
+      "step": 1405
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.959590792838875e-05,
+      "loss": 0.0086,
+      "step": 1406
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9588235294117648e-05,
+      "loss": 0.0216,
+      "step": 1407
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9580562659846546e-05,
+      "loss": 0.0127,
+      "step": 1408
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9572890025575448e-05,
+      "loss": 0.0127,
+      "step": 1409
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.956521739130435e-05,
+      "loss": 0.0209,
+      "step": 1410
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9557544757033248e-05,
+      "loss": 0.0166,
+      "step": 1411
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954987212276215e-05,
+      "loss": 0.0123,
+      "step": 1412
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954219948849105e-05,
+      "loss": 0.0135,
+      "step": 1413
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.953452685421995e-05,
+      "loss": 0.1253,
+      "step": 1414
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.952685421994885e-05,
+      "loss": 0.0098,
+      "step": 1415
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951918158567775e-05,
+      "loss": 0.0091,
+      "step": 1416
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951150895140665e-05,
+      "loss": 0.9327,
+      "step": 1417
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.9503836317135552e-05,
+      "loss": 0.0119,
+      "step": 1418
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.949616368286445e-05,
+      "loss": 0.0153,
+      "step": 1419
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.948849104859335e-05,
+      "loss": 0.0123,
+      "step": 1420
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9480818414322254e-05,
+      "loss": 0.0194,
+      "step": 1421
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9473145780051152e-05,
+      "loss": 1.2311,
+      "step": 1422
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.946547314578005e-05,
+      "loss": 1.8519,
+      "step": 1423
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9457800511508952e-05,
+      "loss": 0.963,
+      "step": 1424
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9450127877237853e-05,
+      "loss": 0.0075,
+      "step": 1425
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.944245524296675e-05,
+      "loss": 0.0042,
+      "step": 1426
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9434782608695653e-05,
+      "loss": 0.0077,
+      "step": 1427
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.942710997442455e-05,
+      "loss": 0.0152,
+      "step": 1428
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9419437340153453e-05,
+      "loss": 0.0152,
+      "step": 1429
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9411764705882355e-05,
+      "loss": 0.019,
+      "step": 1430
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9404092071611253e-05,
+      "loss": 0.0081,
+      "step": 1431
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.939641943734015e-05,
+      "loss": 0.0076,
+      "step": 1432
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9388746803069056e-05,
+      "loss": 0.0086,
+      "step": 1433
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9381074168797954e-05,
+      "loss": 0.0146,
+      "step": 1434
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9373401534526853e-05,
+      "loss": 0.0074,
+      "step": 1435
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9365728900255754e-05,
+      "loss": 0.1007,
+      "step": 1436
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9358056265984656e-05,
+      "loss": 0.011,
+      "step": 1437
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9350383631713554e-05,
+      "loss": 0.0089,
+      "step": 1438
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9342710997442456e-05,
+      "loss": 0.0182,
+      "step": 1439
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9335038363171357e-05,
+      "loss": 0.0139,
+      "step": 1440
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9327365728900255e-05,
+      "loss": 2.2086,
+      "step": 1441
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9319693094629157e-05,
+      "loss": 0.0178,
+      "step": 1442
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9312020460358055e-05,
+      "loss": 0.0142,
+      "step": 1443
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9304347826086957e-05,
+      "loss": 0.0142,
+      "step": 1444
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.929667519181586e-05,
+      "loss": 0.0242,
+      "step": 1445
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9289002557544757e-05,
+      "loss": 0.0087,
+      "step": 1446
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9281329923273655e-05,
+      "loss": 0.0077,
+      "step": 1447
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.927365728900256e-05,
+      "loss": 0.0104,
+      "step": 1448
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9265984654731458e-05,
+      "loss": 0.0139,
+      "step": 1449
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9258312020460356e-05,
+      "loss": 1.2557,
+      "step": 1450
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9250639386189258e-05,
+      "loss": 0.0068,
+      "step": 1451
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.924296675191816e-05,
+      "loss": 0.0163,
+      "step": 1452
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.923529411764706e-05,
+      "loss": 0.9483,
+      "step": 1453
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.922762148337596e-05,
+      "loss": 1.2181,
+      "step": 1454
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.9219948849104858e-05,
+      "loss": 0.011,
+      "step": 1455
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.9212276214833763e-05,
+      "loss": 0.014,
+      "step": 1456
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.920460358056266e-05,
+      "loss": 0.0131,
+      "step": 1457
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.919693094629156e-05,
+      "loss": 0.0135,
+      "step": 1458
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.918925831202046e-05,
+      "loss": 0.007,
+      "step": 1459
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9181585677749362e-05,
+      "loss": 0.0072,
+      "step": 1460
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.917391304347826e-05,
+      "loss": 0.0135,
+      "step": 1461
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9166240409207162e-05,
+      "loss": 0.9908,
+      "step": 1462
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9158567774936064e-05,
+      "loss": 0.0103,
+      "step": 1463
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9150895140664962e-05,
+      "loss": 0.9867,
+      "step": 1464
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9143222506393864e-05,
+      "loss": 0.0138,
+      "step": 1465
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9135549872122762e-05,
+      "loss": 0.0104,
+      "step": 1466
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9127877237851663e-05,
+      "loss": 0.014,
+      "step": 1467
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9120204603580565e-05,
+      "loss": 0.0071,
+      "step": 1468
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9112531969309463e-05,
+      "loss": 0.0108,
+      "step": 1469
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.910485933503836e-05,
+      "loss": 0.0106,
+      "step": 1470
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9097186700767266e-05,
+      "loss": 0.0404,
+      "step": 1471
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9089514066496165e-05,
+      "loss": 2.2011,
+      "step": 1472
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9081841432225063e-05,
+      "loss": 0.0087,
+      "step": 1473
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9074168797953965e-05,
+      "loss": 0.0071,
+      "step": 1474
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9066496163682866e-05,
+      "loss": 0.0072,
+      "step": 1475
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9058823529411764e-05,
+      "loss": 0.0108,
+      "step": 1476
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9051150895140666e-05,
+      "loss": 0.0108,
+      "step": 1477
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9043478260869564e-05,
+      "loss": 0.0072,
+      "step": 1478
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9035805626598466e-05,
+      "loss": 0.0108,
+      "step": 1479
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9028132992327367e-05,
+      "loss": 0.0073,
+      "step": 1480
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9020460358056266e-05,
+      "loss": 0.9537,
+      "step": 1481
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9012787723785164e-05,
+      "loss": 0.0107,
+      "step": 1482
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.900511508951407e-05,
+      "loss": 0.0145,
+      "step": 1483
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8997442455242967e-05,
+      "loss": 0.007,
+      "step": 1484
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8989769820971865e-05,
+      "loss": 0.0071,
+      "step": 1485
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.898209718670077e-05,
+      "loss": 1.243,
+      "step": 1486
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.897442455242967e-05,
+      "loss": 0.011,
+      "step": 1487
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.8966751918158567e-05,
+      "loss": 0.0077,
+      "step": 1488
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895907928388747e-05,
+      "loss": 0.0073,
+      "step": 1489
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895140664961637e-05,
+      "loss": 1.2486,
+      "step": 1490
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8943734015345268e-05,
+      "loss": 0.0109,
+      "step": 1491
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.893606138107417e-05,
+      "loss": 0.0149,
+      "step": 1492
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8928388746803068e-05,
+      "loss": 0.0125,
+      "step": 1493
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.892071611253197e-05,
+      "loss": 0.0071,
+      "step": 1494
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.891304347826087e-05,
+      "loss": 0.0036,
+      "step": 1495
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.890537084398977e-05,
+      "loss": 0.0149,
+      "step": 1496
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8897698209718668e-05,
+      "loss": 0.015,
+      "step": 1497
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8890025575447573e-05,
+      "loss": 0.9536,
+      "step": 1498
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.888235294117647e-05,
+      "loss": 0.9997,
+      "step": 1499
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.887468030690537e-05,
+      "loss": 0.9621,
+      "step": 1500
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.886700767263427e-05,
+      "loss": 0.0113,
+      "step": 1501
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.8859335038363172e-05,
+      "loss": 0.2056,
+      "step": 1502
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.885166240409207e-05,
+      "loss": 0.0114,
+      "step": 1503
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8843989769820972e-05,
+      "loss": 0.016,
+      "step": 1504
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.883631713554987e-05,
+      "loss": 0.9175,
+      "step": 1505
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8828644501278775e-05,
+      "loss": 0.8959,
+      "step": 1506
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8820971867007674e-05,
+      "loss": 0.0074,
+      "step": 1507
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8813299232736572e-05,
+      "loss": 0.0117,
+      "step": 1508
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8805626598465474e-05,
+      "loss": 0.0075,
+      "step": 1509
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8797953964194375e-05,
+      "loss": 0.0074,
+      "step": 1510
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8790281329923273e-05,
+      "loss": 0.0577,
+      "step": 1511
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8782608695652175e-05,
+      "loss": 0.0128,
+      "step": 1512
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8774936061381077e-05,
+      "loss": 0.0073,
+      "step": 1513
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8767263427109975e-05,
+      "loss": 0.0074,
+      "step": 1514
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8759590792838876e-05,
+      "loss": 0.0073,
+      "step": 1515
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8751918158567775e-05,
+      "loss": 0.0088,
+      "step": 1516
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8744245524296676e-05,
+      "loss": 0.0078,
+      "step": 1517
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8736572890025578e-05,
+      "loss": 0.016,
+      "step": 1518
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8728900255754476e-05,
+      "loss": 1.3842,
+      "step": 1519
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8721227621483374e-05,
+      "loss": 1.2502,
+      "step": 1520
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.871355498721228e-05,
+      "loss": 0.0117,
+      "step": 1521
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8705882352941178e-05,
+      "loss": 0.0113,
+      "step": 1522
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8698209718670076e-05,
+      "loss": 0.0125,
+      "step": 1523
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8690537084398977e-05,
+      "loss": 0.0153,
+      "step": 1524
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.868286445012788e-05,
+      "loss": 0.0072,
+      "step": 1525
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8675191815856777e-05,
+      "loss": 0.0086,
+      "step": 1526
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.866751918158568e-05,
+      "loss": 0.0195,
+      "step": 1527
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.8659846547314577e-05,
+      "loss": 1.2511,
+      "step": 1528
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.865217391304348e-05,
+      "loss": 1.2407,
+      "step": 1529
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.864450127877238e-05,
+      "loss": 0.0112,
+      "step": 1530
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.863682864450128e-05,
+      "loss": 0.0073,
+      "step": 1531
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8629156010230177e-05,
+      "loss": 1.1911,
+      "step": 1532
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8621483375959082e-05,
+      "loss": 0.0168,
+      "step": 1533
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.861381074168798e-05,
+      "loss": 0.0128,
+      "step": 1534
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8606138107416878e-05,
+      "loss": 0.0116,
+      "step": 1535
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8598465473145783e-05,
+      "loss": 1.2286,
+      "step": 1536
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.859079283887468e-05,
+      "loss": 0.0158,
+      "step": 1537
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.858312020460358e-05,
+      "loss": 2.3735,
+      "step": 1538
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.857544757033248e-05,
+      "loss": 1.1972,
+      "step": 1539
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8567774936061383e-05,
+      "loss": 0.011,
+      "step": 1540
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.856010230179028e-05,
+      "loss": 0.0127,
+      "step": 1541
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8552429667519183e-05,
+      "loss": 0.0148,
+      "step": 1542
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.854475703324808e-05,
+      "loss": 0.0111,
+      "step": 1543
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8537084398976983e-05,
+      "loss": 0.0143,
+      "step": 1544
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8529411764705884e-05,
+      "loss": 0.0132,
+      "step": 1545
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8521739130434782e-05,
+      "loss": 0.0152,
+      "step": 1546
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.851406649616368e-05,
+      "loss": 0.0152,
+      "step": 1547
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8506393861892586e-05,
+      "loss": 0.0051,
+      "step": 1548
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8498721227621484e-05,
+      "loss": 0.9576,
+      "step": 1549
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8491048593350382e-05,
+      "loss": 0.0181,
+      "step": 1550
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8483375959079284e-05,
+      "loss": 0.011,
+      "step": 1551
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8475703324808185e-05,
+      "loss": 0.0205,
+      "step": 1552
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8468030690537083e-05,
+      "loss": 2.5631,
+      "step": 1553
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8460358056265985e-05,
+      "loss": 0.0102,
+      "step": 1554
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8452685421994883e-05,
+      "loss": 0.0147,
+      "step": 1555
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8445012787723788e-05,
+      "loss": 0.015,
+      "step": 1556
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8437340153452687e-05,
+      "loss": 0.0122,
+      "step": 1557
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8429667519181585e-05,
+      "loss": 0.0176,
+      "step": 1558
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8421994884910486e-05,
+      "loss": 0.0114,
+      "step": 1559
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8414322250639388e-05,
+      "loss": 0.0252,
+      "step": 1560
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8406649616368286e-05,
+      "loss": 0.013,
+      "step": 1561
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8398976982097188e-05,
+      "loss": 0.0142,
+      "step": 1562
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.839130434782609e-05,
+      "loss": 0.6496,
+      "step": 1563
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8383631713554988e-05,
+      "loss": 0.0149,
+      "step": 1564
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.837595907928389e-05,
+      "loss": 0.0104,
+      "step": 1565
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8368286445012787e-05,
+      "loss": 0.0129,
+      "step": 1566
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.836061381074169e-05,
+      "loss": 0.0197,
+      "step": 1567
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.835294117647059e-05,
+      "loss": 0.009,
+      "step": 1568
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.834526854219949e-05,
+      "loss": 0.0086,
+      "step": 1569
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.8337595907928387e-05,
+      "loss": 0.0087,
+      "step": 1570
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.8329923273657292e-05,
+      "loss": 0.0131,
+      "step": 1571
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.832225063938619e-05,
+      "loss": 1.2046,
+      "step": 1572
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.831457800511509e-05,
+      "loss": 0.0135,
+      "step": 1573
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.830690537084399e-05,
+      "loss": 0.0149,
+      "step": 1574
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.8299232736572892e-05,
+      "loss": 0.0077,
+      "step": 1575
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.829156010230179e-05,
+      "loss": 0.0143,
+      "step": 1576
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.828388746803069e-05,
+      "loss": 0.9806,
+      "step": 1577
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.827621483375959e-05,
+      "loss": 1.3783,
+      "step": 1578
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.826854219948849e-05,
+      "loss": 0.0075,
+      "step": 1579
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.8260869565217393e-05,
+      "loss": 0.0111,
+      "step": 1580
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.825319693094629e-05,
+      "loss": 0.0111,
+      "step": 1581
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.824552429667519e-05,
+      "loss": 0.0144,
+      "step": 1582
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8237851662404095e-05,
+      "loss": 0.0128,
+      "step": 1583
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8230179028132993e-05,
+      "loss": 1.2949,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.40147146582603455,
+      "eval_runtime": 58.0487,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.822250639386189e-05,
+      "loss": 0.0075,
+      "step": 1585
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8214833759590796e-05,
+      "loss": 0.0108,
+      "step": 1586
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8207161125319694e-05,
+      "loss": 0.011,
+      "step": 1587
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8199488491048592e-05,
+      "loss": 0.0148,
+      "step": 1588
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8191815856777494e-05,
+      "loss": 0.0077,
+      "step": 1589
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8184143222506396e-05,
+      "loss": 0.6674,
+      "step": 1590
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8176470588235294e-05,
+      "loss": 0.0109,
+      "step": 1591
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8168797953964195e-05,
+      "loss": 0.0108,
+      "step": 1592
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8161125319693094e-05,
+      "loss": 0.011,
+      "step": 1593
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8153452685421995e-05,
+      "loss": 0.0143,
+      "step": 1594
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8145780051150897e-05,
+      "loss": 0.9728,
+      "step": 1595
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8138107416879795e-05,
+      "loss": 0.0074,
+      "step": 1596
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8130434782608693e-05,
+      "loss": 0.0087,
+      "step": 1597
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.81227621483376e-05,
+      "loss": 0.0107,
+      "step": 1598
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8115089514066497e-05,
+      "loss": 0.0077,
+      "step": 1599
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8107416879795395e-05,
+      "loss": 0.0167,
+      "step": 1600
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8099744245524296e-05,
+      "loss": 1.219,
+      "step": 1601
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8092071611253198e-05,
+      "loss": 1.2244,
+      "step": 1602
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8084398976982096e-05,
+      "loss": 0.0108,
+      "step": 1603
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8076726342710998e-05,
+      "loss": 0.0071,
+      "step": 1604
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8069053708439896e-05,
+      "loss": 0.0109,
+      "step": 1605
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.8061381074168798e-05,
+      "loss": 0.0185,
+      "step": 1606
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.80537084398977e-05,
+      "loss": 1.2711,
+      "step": 1607
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.8046035805626598e-05,
+      "loss": 0.0113,
+      "step": 1608
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.80383631713555e-05,
+      "loss": 0.9575,
+      "step": 1609
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80306905370844e-05,
+      "loss": 1.1369,
+      "step": 1610
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80230179028133e-05,
+      "loss": 0.0114,
+      "step": 1611
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80153452685422e-05,
+      "loss": 1.578,
+      "step": 1612
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.8007672634271102e-05,
+      "loss": 0.0076,
+      "step": 1613
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.0128,
+      "step": 1614
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.7992327365728902e-05,
+      "loss": 0.0157,
+      "step": 1615
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.79846547314578e-05,
+      "loss": 0.0154,
+      "step": 1616
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.7976982097186702e-05,
+      "loss": 0.0125,
+      "step": 1617
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7969309462915604e-05,
+      "loss": 0.0085,
+      "step": 1618
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7961636828644502e-05,
+      "loss": 0.0167,
+      "step": 1619
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.79539641943734e-05,
+      "loss": 1.2275,
+      "step": 1620
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7946291560102305e-05,
+      "loss": 0.0115,
+      "step": 1621
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7938618925831203e-05,
+      "loss": 0.0083,
+      "step": 1622
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.79309462915601e-05,
+      "loss": 0.015,
+      "step": 1623
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7923273657289003e-05,
+      "loss": 1.1705,
+      "step": 1624
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7915601023017905e-05,
+      "loss": 0.0152,
+      "step": 1625
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7907928388746803e-05,
+      "loss": 2.0998,
+      "step": 1626
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7900255754475704e-05,
+      "loss": 0.0072,
+      "step": 1627
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7892583120204603e-05,
+      "loss": 0.0089,
+      "step": 1628
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7884910485933504e-05,
+      "loss": 0.0077,
+      "step": 1629
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7877237851662406e-05,
+      "loss": 0.0126,
+      "step": 1630
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7869565217391304e-05,
+      "loss": 1.2113,
+      "step": 1631
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7861892583120202e-05,
+      "loss": 0.0112,
+      "step": 1632
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7854219948849107e-05,
+      "loss": 0.0099,
+      "step": 1633
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7846547314578006e-05,
+      "loss": 0.0074,
+      "step": 1634
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7838874680306904e-05,
+      "loss": 0.0111,
+      "step": 1635
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.783120204603581e-05,
+      "loss": 0.0075,
+      "step": 1636
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7823529411764707e-05,
+      "loss": 1.2187,
+      "step": 1637
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7815856777493605e-05,
+      "loss": 0.0111,
+      "step": 1638
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7808184143222507e-05,
+      "loss": 0.0184,
+      "step": 1639
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.780051150895141e-05,
+      "loss": 0.0111,
+      "step": 1640
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7792838874680307e-05,
+      "loss": 0.0112,
+      "step": 1641
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.778516624040921e-05,
+      "loss": 0.0149,
+      "step": 1642
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.7777493606138107e-05,
+      "loss": 0.0151,
+      "step": 1643
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.7769820971867008e-05,
+      "loss": 0.0074,
+      "step": 1644
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.776214833759591e-05,
+      "loss": 0.9703,
+      "step": 1645
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.7754475703324808e-05,
+      "loss": 1.9364,
+      "step": 1646
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.7746803069053706e-05,
+      "loss": 0.011,
+      "step": 1647
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.773913043478261e-05,
+      "loss": 0.01,
+      "step": 1648
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.773145780051151e-05,
+      "loss": 0.0111,
+      "step": 1649
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.7723785166240408e-05,
+      "loss": 1.2373,
+      "step": 1650
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.771611253196931e-05,
+      "loss": 0.0074,
+      "step": 1651
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.770843989769821e-05,
+      "loss": 0.0116,
+      "step": 1652
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.770076726342711e-05,
+      "loss": 0.0117,
+      "step": 1653
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.769309462915601e-05,
+      "loss": 1.1636,
+      "step": 1654
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.768542199488491e-05,
+      "loss": 0.0075,
+      "step": 1655
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.767774936061381e-05,
+      "loss": 0.007,
+      "step": 1656
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.7670076726342712e-05,
+      "loss": 0.0174,
+      "step": 1657
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.766240409207161e-05,
+      "loss": 0.0113,
+      "step": 1658
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7654731457800512e-05,
+      "loss": 0.0084,
+      "step": 1659
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7647058823529414e-05,
+      "loss": 0.2994,
+      "step": 1660
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7639386189258312e-05,
+      "loss": 0.0213,
+      "step": 1661
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7631713554987213e-05,
+      "loss": 0.011,
+      "step": 1662
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7624040920716115e-05,
+      "loss": 1.2341,
+      "step": 1663
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7616368286445013e-05,
+      "loss": 1.1774,
+      "step": 1664
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7608695652173915e-05,
+      "loss": 0.0181,
+      "step": 1665
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7601023017902813e-05,
+      "loss": 0.0122,
+      "step": 1666
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7593350383631715e-05,
+      "loss": 0.0119,
+      "step": 1667
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7585677749360616e-05,
+      "loss": 0.0114,
+      "step": 1668
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7578005115089515e-05,
+      "loss": 0.0111,
+      "step": 1669
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7570332480818413e-05,
+      "loss": 0.011,
+      "step": 1670
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7562659846547318e-05,
+      "loss": 0.0113,
+      "step": 1671
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7554987212276216e-05,
+      "loss": 0.0161,
+      "step": 1672
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7547314578005114e-05,
+      "loss": 0.9491,
+      "step": 1673
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7539641943734016e-05,
+      "loss": 0.0108,
+      "step": 1674
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7531969309462917e-05,
+      "loss": 0.0114,
+      "step": 1675
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7524296675191816e-05,
+      "loss": 0.0107,
+      "step": 1676
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7516624040920717e-05,
+      "loss": 0.0171,
+      "step": 1677
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7508951406649616e-05,
+      "loss": 0.017,
+      "step": 1678
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7501278772378517e-05,
+      "loss": 1.939,
+      "step": 1679
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.749360613810742e-05,
+      "loss": 0.0151,
+      "step": 1680
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7485933503836317e-05,
+      "loss": 0.0109,
+      "step": 1681
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.7478260869565215e-05,
+      "loss": 0.0119,
+      "step": 1682
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.747058823529412e-05,
+      "loss": 1.2127,
+      "step": 1683
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.746291560102302e-05,
+      "loss": 0.0075,
+      "step": 1684
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.7455242966751917e-05,
+      "loss": 1.2283,
+      "step": 1685
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.744757033248082e-05,
+      "loss": 0.0107,
+      "step": 1686
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.743989769820972e-05,
+      "loss": 0.0141,
+      "step": 1687
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.7432225063938618e-05,
+      "loss": 1.2117,
+      "step": 1688
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.742455242966752e-05,
+      "loss": 0.8715,
+      "step": 1689
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.741687979539642e-05,
+      "loss": 0.0075,
+      "step": 1690
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.740920716112532e-05,
+      "loss": 1.2208,
+      "step": 1691
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.740153452685422e-05,
+      "loss": 0.0077,
+      "step": 1692
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.739386189258312e-05,
+      "loss": 0.0081,
+      "step": 1693
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.738618925831202e-05,
+      "loss": 0.0104,
+      "step": 1694
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.7378516624040923e-05,
+      "loss": 0.0162,
+      "step": 1695
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.737084398976982e-05,
+      "loss": 0.0173,
+      "step": 1696
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.736317135549872e-05,
+      "loss": 0.0194,
+      "step": 1697
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.7355498721227624e-05,
+      "loss": 0.0174,
+      "step": 1698
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.7347826086956522e-05,
+      "loss": 0.0081,
+      "step": 1699
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.734015345268542e-05,
+      "loss": 0.0121,
+      "step": 1700
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7332480818414322e-05,
+      "loss": 0.0137,
+      "step": 1701
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7324808184143224e-05,
+      "loss": 0.0117,
+      "step": 1702
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7317135549872122e-05,
+      "loss": 0.0133,
+      "step": 1703
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7309462915601024e-05,
+      "loss": 0.0175,
+      "step": 1704
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7301790281329922e-05,
+      "loss": 0.0076,
+      "step": 1705
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7294117647058823e-05,
+      "loss": 0.0179,
+      "step": 1706
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7286445012787725e-05,
+      "loss": 1.0541,
+      "step": 1707
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7278772378516623e-05,
+      "loss": 0.0077,
+      "step": 1708
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7271099744245525e-05,
+      "loss": 0.0079,
+      "step": 1709
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7263427109974426e-05,
+      "loss": 0.0102,
+      "step": 1710
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7255754475703325e-05,
+      "loss": 0.0158,
+      "step": 1711
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7248081841432223e-05,
+      "loss": 0.0107,
+      "step": 1712
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7240409207161128e-05,
+      "loss": 0.0159,
+      "step": 1713
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7232736572890026e-05,
+      "loss": 0.0125,
+      "step": 1714
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7225063938618928e-05,
+      "loss": 0.0081,
+      "step": 1715
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7217391304347826e-05,
+      "loss": 1.1946,
+      "step": 1716
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7209718670076728e-05,
+      "loss": 1.0298,
+      "step": 1717
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.720204603580563e-05,
+      "loss": 1.0207,
+      "step": 1718
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7194373401534527e-05,
+      "loss": 0.0152,
+      "step": 1719
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7186700767263426e-05,
+      "loss": 0.0187,
+      "step": 1720
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.717902813299233e-05,
+      "loss": 0.0134,
+      "step": 1721
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.717135549872123e-05,
+      "loss": 0.0173,
+      "step": 1722
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.7163682864450127e-05,
+      "loss": 1.0206,
+      "step": 1723
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.715601023017903e-05,
+      "loss": 0.0234,
+      "step": 1724
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.714833759590793e-05,
+      "loss": 0.0151,
+      "step": 1725
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.714066496163683e-05,
+      "loss": 0.0074,
+      "step": 1726
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.713299232736573e-05,
+      "loss": 0.0127,
+      "step": 1727
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.712531969309463e-05,
+      "loss": 0.8679,
+      "step": 1728
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.711764705882353e-05,
+      "loss": 0.0498,
+      "step": 1729
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.710997442455243e-05,
+      "loss": 0.0136,
+      "step": 1730
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.710230179028133e-05,
+      "loss": 1.2168,
+      "step": 1731
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.7094629156010228e-05,
+      "loss": 0.0105,
+      "step": 1732
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.7086956521739133e-05,
+      "loss": 0.0107,
+      "step": 1733
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.707928388746803e-05,
+      "loss": 0.0122,
+      "step": 1734
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.707161125319693e-05,
+      "loss": 0.0075,
+      "step": 1735
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.7063938618925834e-05,
+      "loss": 0.0607,
+      "step": 1736
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7056265984654733e-05,
+      "loss": 1.0181,
+      "step": 1737
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.704859335038363e-05,
+      "loss": 0.0095,
+      "step": 1738
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7040920716112533e-05,
+      "loss": 0.0128,
+      "step": 1739
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7033248081841434e-05,
+      "loss": 0.0096,
+      "step": 1740
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7025575447570332e-05,
+      "loss": 1.1781,
+      "step": 1741
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7017902813299234e-05,
+      "loss": 1.0132,
+      "step": 1742
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7010230179028132e-05,
+      "loss": 0.0097,
+      "step": 1743
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7002557544757034e-05,
+      "loss": 0.012,
+      "step": 1744
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6994884910485935e-05,
+      "loss": 0.0128,
+      "step": 1745
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6987212276214834e-05,
+      "loss": 0.0145,
+      "step": 1746
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6979539641943732e-05,
+      "loss": 0.7558,
+      "step": 1747
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6971867007672637e-05,
+      "loss": 0.0107,
+      "step": 1748
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6964194373401535e-05,
+      "loss": 1.0181,
+      "step": 1749
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6956521739130433e-05,
+      "loss": 1.7925,
+      "step": 1750
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6948849104859335e-05,
+      "loss": 0.0141,
+      "step": 1751
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6941176470588237e-05,
+      "loss": 0.0167,
+      "step": 1752
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6933503836317135e-05,
+      "loss": 0.0141,
+      "step": 1753
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6925831202046036e-05,
+      "loss": 0.0149,
+      "step": 1754
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6918158567774935e-05,
+      "loss": 0.0115,
+      "step": 1755
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6910485933503836e-05,
+      "loss": 0.0071,
+      "step": 1756
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6902813299232738e-05,
+      "loss": 0.0155,
+      "step": 1757
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6895140664961636e-05,
+      "loss": 0.0199,
+      "step": 1758
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6887468030690538e-05,
+      "loss": 0.0124,
+      "step": 1759
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.687979539641944e-05,
+      "loss": 0.0086,
+      "step": 1760
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.6872122762148337e-05,
+      "loss": 0.0147,
+      "step": 1761
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.6864450127877236e-05,
+      "loss": 0.0107,
+      "step": 1762
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.685677749360614e-05,
+      "loss": 1.025,
+      "step": 1763
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.684910485933504e-05,
+      "loss": 0.0105,
+      "step": 1764
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.6841432225063937e-05,
+      "loss": 0.0081,
+      "step": 1765
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.683375959079284e-05,
+      "loss": 0.009,
+      "step": 1766
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.682608695652174e-05,
+      "loss": 0.0109,
+      "step": 1767
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.6818414322250642e-05,
+      "loss": 0.008,
+      "step": 1768
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.681074168797954e-05,
+      "loss": 0.0132,
+      "step": 1769
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.680306905370844e-05,
+      "loss": 0.0119,
+      "step": 1770
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.6795396419437343e-05,
+      "loss": 0.0093,
+      "step": 1771
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.678772378516624e-05,
+      "loss": 1.0247,
+      "step": 1772
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.678005115089514e-05,
+      "loss": 1.024,
+      "step": 1773
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.677237851662404e-05,
+      "loss": 0.0106,
+      "step": 1774
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.6764705882352943e-05,
+      "loss": 0.0097,
+      "step": 1775
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.675703324808184e-05,
+      "loss": 0.0098,
+      "step": 1776
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6749360613810743e-05,
+      "loss": 1.0136,
+      "step": 1777
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.674168797953964e-05,
+      "loss": 0.0066,
+      "step": 1778
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6734015345268543e-05,
+      "loss": 0.1284,
+      "step": 1779
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6726342710997444e-05,
+      "loss": 0.9832,
+      "step": 1780
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6718670076726343e-05,
+      "loss": 0.0128,
+      "step": 1781
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6710997442455244e-05,
+      "loss": 0.0103,
+      "step": 1782
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6703324808184146e-05,
+      "loss": 0.0067,
+      "step": 1783
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6695652173913044e-05,
+      "loss": 0.0105,
+      "step": 1784
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6687979539641942e-05,
+      "loss": 0.0146,
+      "step": 1785
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6680306905370847e-05,
+      "loss": 0.0068,
+      "step": 1786
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6672634271099745e-05,
+      "loss": 0.0148,
+      "step": 1787
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6664961636828644e-05,
+      "loss": 0.0107,
+      "step": 1788
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6657289002557545e-05,
+      "loss": 2.4738,
+      "step": 1789
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6649616368286447e-05,
+      "loss": 0.9625,
+      "step": 1790
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6641943734015345e-05,
+      "loss": 0.2578,
+      "step": 1791
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6634271099744247e-05,
+      "loss": 0.0069,
+      "step": 1792
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6626598465473145e-05,
+      "loss": 0.011,
+      "step": 1793
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6618925831202047e-05,
+      "loss": 0.0066,
+      "step": 1794
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6611253196930948e-05,
+      "loss": 0.0069,
+      "step": 1795
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6603580562659846e-05,
+      "loss": 0.0071,
+      "step": 1796
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6595907928388745e-05,
+      "loss": 0.0068,
+      "step": 1797
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.658823529411765e-05,
+      "loss": 0.0232,
+      "step": 1798
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6580562659846548e-05,
+      "loss": 1.1318,
+      "step": 1799
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6572890025575446e-05,
+      "loss": 0.0101,
+      "step": 1800
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6565217391304348e-05,
+      "loss": 0.0187,
+      "step": 1801
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.655754475703325e-05,
+      "loss": 1.0718,
+      "step": 1802
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6549872122762148e-05,
+      "loss": 1.2349,
+      "step": 1803
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.654219948849105e-05,
+      "loss": 0.0148,
+      "step": 1804
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.6534526854219947e-05,
+      "loss": 0.0148,
+      "step": 1805
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.652685421994885e-05,
+      "loss": 0.0067,
+      "step": 1806
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.651918158567775e-05,
+      "loss": 0.0075,
+      "step": 1807
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.651150895140665e-05,
+      "loss": 0.0157,
+      "step": 1808
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.650383631713555e-05,
+      "loss": 0.0084,
+      "step": 1809
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.6496163682864452e-05,
+      "loss": 0.0123,
+      "step": 1810
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.648849104859335e-05,
+      "loss": 0.0065,
+      "step": 1811
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.648081841432225e-05,
+      "loss": 0.014,
+      "step": 1812
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.6473145780051154e-05,
+      "loss": 0.0152,
+      "step": 1813
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.6465473145780052e-05,
+      "loss": 0.0168,
+      "step": 1814
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.645780051150895e-05,
+      "loss": 0.0063,
+      "step": 1815
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.645012787723785e-05,
+      "loss": 0.6994,
+      "step": 1816
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.6442455242966753e-05,
+      "loss": 0.0132,
+      "step": 1817
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.643478260869565e-05,
+      "loss": 0.0103,
+      "step": 1818
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.6427109974424553e-05,
+      "loss": 0.013,
+      "step": 1819
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.641943734015345e-05,
+      "loss": 1.1813,
+      "step": 1820
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6411764705882356e-05,
+      "loss": 0.9809,
+      "step": 1821
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6404092071611254e-05,
+      "loss": 1.2339,
+      "step": 1822
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6396419437340153e-05,
+      "loss": 0.9744,
+      "step": 1823
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6388746803069054e-05,
+      "loss": 0.0098,
+      "step": 1824
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6381074168797956e-05,
+      "loss": 0.0062,
+      "step": 1825
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6373401534526854e-05,
+      "loss": 0.0151,
+      "step": 1826
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6365728900255756e-05,
+      "loss": 0.013,
+      "step": 1827
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6358056265984654e-05,
+      "loss": 0.0071,
+      "step": 1828
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6350383631713556e-05,
+      "loss": 0.0117,
+      "step": 1829
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6342710997442457e-05,
+      "loss": 0.0165,
+      "step": 1830
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6335038363171355e-05,
+      "loss": 0.1519,
+      "step": 1831
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6327365728900257e-05,
+      "loss": 1.0247,
+      "step": 1832
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.631969309462916e-05,
+      "loss": 0.0092,
+      "step": 1833
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6312020460358057e-05,
+      "loss": 0.0098,
+      "step": 1834
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6304347826086955e-05,
+      "loss": 1.1829,
+      "step": 1835
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.629667519181586e-05,
+      "loss": 0.006,
+      "step": 1836
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.628900255754476e-05,
+      "loss": 0.0109,
+      "step": 1837
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.6281329923273657e-05,
+      "loss": 0.6059,
+      "step": 1838
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.6273657289002558e-05,
+      "loss": 0.01,
+      "step": 1839
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.626598465473146e-05,
+      "loss": 0.0149,
+      "step": 1840
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.6258312020460358e-05,
+      "loss": 0.0152,
+      "step": 1841
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.625063938618926e-05,
+      "loss": 0.0171,
+      "step": 1842
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.6242966751918158e-05,
+      "loss": 0.0105,
+      "step": 1843
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.623529411764706e-05,
+      "loss": 0.0093,
+      "step": 1844
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.622762148337596e-05,
+      "loss": 1.4125,
+      "step": 1845
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.621994884910486e-05,
+      "loss": 0.0107,
+      "step": 1846
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.6212276214833757e-05,
+      "loss": 0.0133,
+      "step": 1847
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.6204603580562662e-05,
+      "loss": 0.0063,
+      "step": 1848
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.619693094629156e-05,
+      "loss": 0.0063,
+      "step": 1849
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.618925831202046e-05,
+      "loss": 0.009,
+      "step": 1850
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.618158567774936e-05,
+      "loss": 0.0137,
+      "step": 1851
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.6173913043478262e-05,
+      "loss": 0.0145,
+      "step": 1852
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.616624040920716e-05,
+      "loss": 0.0355,
+      "step": 1853
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.6158567774936062e-05,
+      "loss": 0.9601,
+      "step": 1854
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.615089514066496e-05,
+      "loss": 0.0135,
+      "step": 1855
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6143222506393862e-05,
+      "loss": 1.0129,
+      "step": 1856
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6135549872122763e-05,
+      "loss": 0.0339,
+      "step": 1857
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.612787723785166e-05,
+      "loss": 0.0131,
+      "step": 1858
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6120204603580563e-05,
+      "loss": 0.009,
+      "step": 1859
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6112531969309465e-05,
+      "loss": 0.0087,
+      "step": 1860
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6104859335038363e-05,
+      "loss": 0.012,
+      "step": 1861
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.609718670076726e-05,
+      "loss": 0.0088,
+      "step": 1862
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6089514066496166e-05,
+      "loss": 0.0117,
+      "step": 1863
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6081841432225065e-05,
+      "loss": 2.5214,
+      "step": 1864
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6074168797953963e-05,
+      "loss": 0.0107,
+      "step": 1865
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6066496163682864e-05,
+      "loss": 0.0101,
+      "step": 1866
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6058823529411766e-05,
+      "loss": 0.0118,
+      "step": 1867
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6051150895140664e-05,
+      "loss": 0.0087,
+      "step": 1868
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6043478260869566e-05,
+      "loss": 0.0122,
+      "step": 1869
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6035805626598464e-05,
+      "loss": 0.0084,
+      "step": 1870
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6028132992327366e-05,
+      "loss": 0.0104,
+      "step": 1871
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6020460358056267e-05,
+      "loss": 0.1587,
+      "step": 1872
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6012787723785166e-05,
+      "loss": 0.0057,
+      "step": 1873
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6005115089514067e-05,
+      "loss": 0.0085,
+      "step": 1874
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.599744245524297e-05,
+      "loss": 0.0056,
+      "step": 1875
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.5989769820971867e-05,
+      "loss": 1.2194,
+      "step": 1876
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.598209718670077e-05,
+      "loss": 0.0056,
+      "step": 1877
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.5974424552429667e-05,
+      "loss": 0.0057,
+      "step": 1878
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.596675191815857e-05,
+      "loss": 1.0276,
+      "step": 1879
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.595907928388747e-05,
+      "loss": 0.0083,
+      "step": 1880
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.5951406649616368e-05,
+      "loss": 0.0085,
+      "step": 1881
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.594373401534527e-05,
+      "loss": 0.0363,
+      "step": 1882
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.593606138107417e-05,
+      "loss": 0.0157,
+      "step": 1883
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.592838874680307e-05,
+      "loss": 0.0313,
+      "step": 1884
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.5920716112531968e-05,
+      "loss": 0.0062,
+      "step": 1885
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.5913043478260873e-05,
+      "loss": 0.0195,
+      "step": 1886
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.590537084398977e-05,
+      "loss": 0.0151,
+      "step": 1887
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.589769820971867e-05,
+      "loss": 0.0101,
+      "step": 1888
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.589002557544757e-05,
+      "loss": 1.1092,
+      "step": 1889
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.5882352941176473e-05,
+      "loss": 0.0323,
+      "step": 1890
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.587468030690537e-05,
+      "loss": 0.0086,
+      "step": 1891
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.5867007672634272e-05,
+      "loss": 0.0147,
+      "step": 1892
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.585933503836317e-05,
+      "loss": 0.0084,
+      "step": 1893
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.5851662404092072e-05,
+      "loss": 0.5596,
+      "step": 1894
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5843989769820974e-05,
+      "loss": 0.0114,
+      "step": 1895
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5836317135549872e-05,
+      "loss": 0.0083,
+      "step": 1896
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.582864450127877e-05,
+      "loss": 0.0082,
+      "step": 1897
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5820971867007675e-05,
+      "loss": 0.0146,
+      "step": 1898
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5813299232736574e-05,
+      "loss": 0.0081,
+      "step": 1899
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5805626598465472e-05,
+      "loss": 0.0081,
+      "step": 1900
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5797953964194373e-05,
+      "loss": 0.0116,
+      "step": 1901
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5790281329923275e-05,
+      "loss": 0.0084,
+      "step": 1902
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5782608695652173e-05,
+      "loss": 0.0108,
+      "step": 1903
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5774936061381075e-05,
+      "loss": 0.6146,
+      "step": 1904
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5767263427109973e-05,
+      "loss": 0.0162,
+      "step": 1905
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5759590792838875e-05,
+      "loss": 0.0134,
+      "step": 1906
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5751918158567776e-05,
+      "loss": 0.0047,
+      "step": 1907
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5744245524296674e-05,
+      "loss": 0.0045,
+      "step": 1908
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5736572890025576e-05,
+      "loss": 0.0087,
+      "step": 1909
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5728900255754478e-05,
+      "loss": 2.0913,
+      "step": 1910
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5721227621483376e-05,
+      "loss": 0.0075,
+      "step": 1911
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5713554987212274e-05,
+      "loss": 1.3387,
+      "step": 1912
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.570588235294118e-05,
+      "loss": 0.0102,
+      "step": 1913
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5698209718670077e-05,
+      "loss": 0.0076,
+      "step": 1914
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5690537084398976e-05,
+      "loss": 0.0131,
+      "step": 1915
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5682864450127877e-05,
+      "loss": 0.0079,
+      "step": 1916
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.567519181585678e-05,
+      "loss": 0.0102,
+      "step": 1917
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5667519181585677e-05,
+      "loss": 0.9554,
+      "step": 1918
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.565984654731458e-05,
+      "loss": 0.0088,
+      "step": 1919
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.5652173913043477e-05,
+      "loss": 1.0265,
+      "step": 1920
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.564450127877238e-05,
+      "loss": 0.0155,
+      "step": 1921
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.563682864450128e-05,
+      "loss": 0.0079,
+      "step": 1922
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.562915601023018e-05,
+      "loss": 1.0467,
+      "step": 1923
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.5621483375959077e-05,
+      "loss": 0.0132,
+      "step": 1924
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.561381074168798e-05,
+      "loss": 0.0077,
+      "step": 1925
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.560613810741688e-05,
+      "loss": 0.0064,
+      "step": 1926
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.559846547314578e-05,
+      "loss": 0.0104,
+      "step": 1927
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.559079283887468e-05,
+      "loss": 0.0096,
+      "step": 1928
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.558312020460358e-05,
+      "loss": 0.0049,
+      "step": 1929
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.5575447570332483e-05,
+      "loss": 0.0104,
+      "step": 1930
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.556777493606138e-05,
+      "loss": 1.0424,
+      "step": 1931
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5560102301790283e-05,
+      "loss": 0.0079,
+      "step": 1932
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5552429667519184e-05,
+      "loss": 0.0104,
+      "step": 1933
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5544757033248083e-05,
+      "loss": 0.0077,
+      "step": 1934
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.553708439897698e-05,
+      "loss": 0.005,
+      "step": 1935
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5529411764705886e-05,
+      "loss": 2.6544,
+      "step": 1936
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5521739130434784e-05,
+      "loss": 0.0107,
+      "step": 1937
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5514066496163682e-05,
+      "loss": 0.0124,
+      "step": 1938
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5506393861892584e-05,
+      "loss": 0.0049,
+      "step": 1939
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5498721227621485e-05,
+      "loss": 0.0078,
+      "step": 1940
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5491048593350384e-05,
+      "loss": 2.1009,
+      "step": 1941
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5483375959079285e-05,
+      "loss": 0.0116,
+      "step": 1942
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5475703324808183e-05,
+      "loss": 0.0049,
+      "step": 1943
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5468030690537085e-05,
+      "loss": 0.0079,
+      "step": 1944
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5460358056265987e-05,
+      "loss": 1.0307,
+      "step": 1945
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5452685421994885e-05,
+      "loss": 0.0147,
+      "step": 1946
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5445012787723783e-05,
+      "loss": 0.0084,
+      "step": 1947
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5437340153452688e-05,
+      "loss": 0.0262,
+      "step": 1948
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5429667519181586e-05,
+      "loss": 0.0115,
+      "step": 1949
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5421994884910485e-05,
+      "loss": 1.3221,
+      "step": 1950
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5414322250639386e-05,
+      "loss": 0.7929,
+      "step": 1951
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5406649616368288e-05,
+      "loss": 1.2069,
+      "step": 1952
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5398976982097186e-05,
+      "loss": 0.0081,
+      "step": 1953
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5391304347826088e-05,
+      "loss": 0.0069,
+      "step": 1954
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5383631713554986e-05,
+      "loss": 0.0097,
+      "step": 1955
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5375959079283887e-05,
+      "loss": 0.0082,
+      "step": 1956
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.536828644501279e-05,
+      "loss": 0.0061,
+      "step": 1957
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5360613810741687e-05,
+      "loss": 0.0082,
+      "step": 1958
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.535294117647059e-05,
+      "loss": 0.0113,
+      "step": 1959
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.534526854219949e-05,
+      "loss": 0.0098,
+      "step": 1960
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.533759590792839e-05,
+      "loss": 1.0221,
+      "step": 1961
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.5329923273657287e-05,
+      "loss": 0.0039,
+      "step": 1962
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.5322250639386192e-05,
+      "loss": 0.0127,
+      "step": 1963
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.531457800511509e-05,
+      "loss": 0.0051,
+      "step": 1964
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.530690537084399e-05,
+      "loss": 0.0086,
+      "step": 1965
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.529923273657289e-05,
+      "loss": 0.0096,
+      "step": 1966
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.529156010230179e-05,
+      "loss": 0.0164,
+      "step": 1967
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.528388746803069e-05,
+      "loss": 1.0127,
+      "step": 1968
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.527621483375959e-05,
+      "loss": 0.0149,
+      "step": 1969
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.526854219948849e-05,
+      "loss": 0.0085,
+      "step": 1970
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.526086956521739e-05,
+      "loss": 0.0116,
+      "step": 1971
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5253196930946293e-05,
+      "loss": 0.0168,
+      "step": 1972
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5245524296675191e-05,
+      "loss": 0.0052,
+      "step": 1973
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5237851662404091e-05,
+      "loss": 0.0052,
+      "step": 1974
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5230179028132993e-05,
+      "loss": 1.0013,
+      "step": 1975
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5222506393861893e-05,
+      "loss": 0.0122,
+      "step": 1976
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.521483375959079e-05,
+      "loss": 0.0087,
+      "step": 1977
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5207161125319694e-05,
+      "loss": 1.9624,
+      "step": 1978
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5199488491048592e-05,
+      "loss": 0.0088,
+      "step": 1979
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5191815856777496e-05,
+      "loss": 0.0596,
+      "step": 1980
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9090909090909091,
+      "eval_loss": 0.40464258193969727,
+      "eval_runtime": 58.0491,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1980
+    }
+  ],
+  "max_steps": 3960,
+  "num_train_epochs": 10,
+  "total_flos": 7.18573773456e+17,
+  "trial_name": null,
+  "trial_params": null
+}
diff --git a/checkpoint-1980/training_args.bin b/checkpoint-1980/training_args.bin
new file mode 100644
index 0000000000000000000000000000000000000000..32901f899c91066c808eae9dd019400bfdd17498
--- /dev/null
+++ b/checkpoint-1980/training_args.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db14439c369bfe9c89f64ff38f0b2d3fcc53911fc2c0b12a3a7d04656e7384a
+size 3503
diff --git a/checkpoint-2376/config.json b/checkpoint-2376/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..cb8d52e14263c50fd53dca42b5050e3235115c32
--- /dev/null
+++ b/checkpoint-2376/config.json
@@ -0,0 +1,113 @@
+{
+  "_name_or_path": "facebook/wav2vec2-base-960h",
+  "activation_dropout": 0.1,
+  "adapter_kernel_size": 3,
+  "adapter_stride": 2,
+  "add_adapter": false,
+  "apply_spec_augment": true,
+  "architectures": [
+    "Wav2Vec2ForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 1,
+  "classifier_proj_size": 256,
+  "codevector_dim": 256,
+  "contrastive_logits_temperature": 0.1,
+  "conv_bias": false,
+  "conv_dim": [
+    512,
+    512,
+    512,
+    512,
+    512,
+    512,
+    512
+  ],
+  "conv_kernel": [
+    10,
+    3,
+    3,
+    3,
+    3,
+    2,
+    2
+  ],
+  "conv_stride": [
+    5,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2
+  ],
+  "ctc_loss_reduction": "sum",
+  "ctc_zero_infinity": false,
+  "diversity_loss_weight": 0.1,
+  "do_stable_layer_norm": false,
+  "eos_token_id": 2,
+  "feat_extract_activation": "gelu",
+  "feat_extract_dropout": 0.0,
+  "feat_extract_norm": "group",
+  "feat_proj_dropout": 0.1,
+  "feat_quantizer_dropout": 0.0,
+  "final_dropout": 0.1,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout": 0.1,
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Foreign",
+    "1": "Native"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "layerdrop": 0.1,
+  "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
+  "mask_feature_prob": 0.0,
+  "mask_time_length": 10,
+  "mask_time_min_masks": 2,
+  "mask_time_prob": 0.05,
+  "model_type": "wav2vec2",
+  "num_adapter_layers": 3,
+  "num_attention_heads": 12,
+  "num_codevector_groups": 2,
+  "num_codevectors_per_group": 320,
+  "num_conv_pos_embedding_groups": 16,
+  "num_conv_pos_embeddings": 128,
+  "num_feat_extract_layers": 7,
+  "num_hidden_layers": 12,
+  "num_negatives": 100,
+  "output_hidden_size": 768,
+  "pad_token_id": 0,
+  "proj_codevector_dim": 256,
+  "tdnn_dilation": [
+    1,
+    2,
+    3,
+    1,
+    1
+  ],
+  "tdnn_dim": [
+    512,
+    512,
+    512,
+    512,
+    1500
+  ],
+  "tdnn_kernel": [
+    5,
+    3,
+    3,
+    1,
+    1
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.28.1",
+  "use_weighted_layer_sum": false,
+  "vocab_size": 32,
+  "xvector_output_dim": 512
+}
diff --git a/checkpoint-2376/optimizer.pt b/checkpoint-2376/optimizer.pt
new file mode 100644
index 0000000000000000000000000000000000000000..a64276d9542e1b54fcfeefc035f5e22400089abc
--- /dev/null
+++ b/checkpoint-2376/optimizer.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:364b7ac4a1fc23737a02788a40f1a5c741826bf5b6a9e248d8e3504131444587
+size 756677477
diff --git a/checkpoint-2376/preprocessor_config.json b/checkpoint-2376/preprocessor_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..a0b7227fc1d916e469b14f6c154ad6dfea1e6891
--- /dev/null
+++ b/checkpoint-2376/preprocessor_config.json
@@ -0,0 +1,9 @@
+{
+  "do_normalize": true,
+  "feature_extractor_type": "Wav2Vec2FeatureExtractor",
+  "feature_size": 1,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": false,
+  "sampling_rate": 16000
+}
diff --git a/checkpoint-2376/pytorch_model.bin b/checkpoint-2376/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..1fef170ef7e7594af290734de461df4fe714fad3
--- /dev/null
+++ b/checkpoint-2376/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:415f32f77f5fa3a5892d605f9d3333aa719e7abc96aeb959f4818f6f245a6837
+size 378348563
diff --git a/checkpoint-2376/rng_state.pth b/checkpoint-2376/rng_state.pth
new file mode 100644
index 0000000000000000000000000000000000000000..1a34d3768336cbaea58330c9bb2ee68d1371872a
--- /dev/null
+++ b/checkpoint-2376/rng_state.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d5c9d21d29b879455b34a95f363cc8e14f1416bfc5400268d4f39c9b188a7868
+size 14567
diff --git a/checkpoint-2376/scheduler.pt b/checkpoint-2376/scheduler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..5927b70cb5061769e68afc18ccc2d3b9437e51d2
--- /dev/null
+++ b/checkpoint-2376/scheduler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e2bdd50e7407a607d0e5b5c929ea1ed3e3d72d13d158e9ef94854fe3672d5d9
+size 623
diff --git a/checkpoint-2376/trainer_state.json b/checkpoint-2376/trainer_state.json
new file mode 100644
index 0000000000000000000000000000000000000000..5af577a9ecff38e22afcda1ee141f30bedf22b64
--- /dev/null
+++ b/checkpoint-2376/trainer_state.json
@@ -0,0 +1,14326 @@
+{
+  "best_metric": 0.3489837944507599,
+  "best_model_checkpoint": "accent_classification/checkpoint-2376",
+  "epoch": 6.0,
+  "global_step": 2376,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.000000000000001e-07,
+      "loss": 0.6956,
+      "step": 1
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.2000000000000002e-06,
+      "loss": 0.6979,
+      "step": 2
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8e-06,
+      "loss": 0.7014,
+      "step": 3
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 2.4000000000000003e-06,
+      "loss": 0.6984,
+      "step": 4
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 3e-06,
+      "loss": 0.6978,
+      "step": 5
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 3.6e-06,
+      "loss": 0.6783,
+      "step": 6
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.2000000000000004e-06,
+      "loss": 0.6956,
+      "step": 7
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.800000000000001e-06,
+      "loss": 0.6994,
+      "step": 8
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 5.4e-06,
+      "loss": 0.6902,
+      "step": 9
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6e-06,
+      "loss": 0.6937,
+      "step": 10
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6.6e-06,
+      "loss": 0.6497,
+      "step": 11
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.2e-06,
+      "loss": 0.6911,
+      "step": 12
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.8e-06,
+      "loss": 0.6904,
+      "step": 13
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 8.400000000000001e-06,
+      "loss": 0.6854,
+      "step": 14
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9e-06,
+      "loss": 0.6843,
+      "step": 15
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9.600000000000001e-06,
+      "loss": 0.6851,
+      "step": 16
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.02e-05,
+      "loss": 0.6733,
+      "step": 17
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.08e-05,
+      "loss": 0.6819,
+      "step": 18
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.1400000000000001e-05,
+      "loss": 0.681,
+      "step": 19
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.6998,
+      "step": 20
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.26e-05,
+      "loss": 0.6743,
+      "step": 21
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.32e-05,
+      "loss": 0.6818,
+      "step": 22
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3800000000000002e-05,
+      "loss": 0.5331,
+      "step": 23
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.44e-05,
+      "loss": 0.6681,
+      "step": 24
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.6955,
+      "step": 25
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.56e-05,
+      "loss": 0.7,
+      "step": 26
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.62e-05,
+      "loss": 0.6975,
+      "step": 27
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.6800000000000002e-05,
+      "loss": 0.6249,
+      "step": 28
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.74e-05,
+      "loss": 0.6979,
+      "step": 29
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.6076,
+      "step": 30
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.86e-05,
+      "loss": 0.6091,
+      "step": 31
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9200000000000003e-05,
+      "loss": 0.6438,
+      "step": 32
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.98e-05,
+      "loss": 0.6624,
+      "step": 33
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.04e-05,
+      "loss": 0.6452,
+      "step": 34
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.6963,
+      "step": 35
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.16e-05,
+      "loss": 0.758,
+      "step": 36
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.22e-05,
+      "loss": 0.6471,
+      "step": 37
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.2800000000000002e-05,
+      "loss": 0.6313,
+      "step": 38
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.3400000000000003e-05,
+      "loss": 0.7019,
+      "step": 39
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.6336,
+      "step": 40
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4599999999999998e-05,
+      "loss": 0.7091,
+      "step": 41
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.52e-05,
+      "loss": 0.701,
+      "step": 42
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.58e-05,
+      "loss": 0.7036,
+      "step": 43
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.64e-05,
+      "loss": 0.7073,
+      "step": 44
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.614,
+      "step": 45
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.7600000000000003e-05,
+      "loss": 0.7543,
+      "step": 46
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.8199999999999998e-05,
+      "loss": 0.538,
+      "step": 47
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.88e-05,
+      "loss": 0.5179,
+      "step": 48
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.94e-05,
+      "loss": 0.6164,
+      "step": 49
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 3e-05,
+      "loss": 0.6093,
+      "step": 50
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.99923273657289e-05,
+      "loss": 0.6983,
+      "step": 51
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9984654731457804e-05,
+      "loss": 0.6069,
+      "step": 52
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9976982097186702e-05,
+      "loss": 0.5933,
+      "step": 53
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.99693094629156e-05,
+      "loss": 0.552,
+      "step": 54
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9961636828644502e-05,
+      "loss": 0.5865,
+      "step": 55
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9953964194373404e-05,
+      "loss": 0.7146,
+      "step": 56
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9946291560102302e-05,
+      "loss": 0.5591,
+      "step": 57
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9938618925831203e-05,
+      "loss": 0.7366,
+      "step": 58
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9930946291560102e-05,
+      "loss": 0.7006,
+      "step": 59
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9923273657289003e-05,
+      "loss": 0.4148,
+      "step": 60
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9915601023017905e-05,
+      "loss": 0.9938,
+      "step": 61
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9907928388746803e-05,
+      "loss": 0.911,
+      "step": 62
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.99002557544757e-05,
+      "loss": 0.5682,
+      "step": 63
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9892583120204606e-05,
+      "loss": 0.5833,
+      "step": 64
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9884910485933505e-05,
+      "loss": 0.7237,
+      "step": 65
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9877237851662403e-05,
+      "loss": 0.4013,
+      "step": 66
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9869565217391304e-05,
+      "loss": 0.9695,
+      "step": 67
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9861892583120206e-05,
+      "loss": 0.568,
+      "step": 68
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9854219948849104e-05,
+      "loss": 0.5697,
+      "step": 69
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9846547314578006e-05,
+      "loss": 0.7675,
+      "step": 70
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9838874680306904e-05,
+      "loss": 0.8831,
+      "step": 71
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9831202046035806e-05,
+      "loss": 0.7008,
+      "step": 72
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9823529411764707e-05,
+      "loss": 0.7085,
+      "step": 73
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9815856777493606e-05,
+      "loss": 0.5374,
+      "step": 74
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9808184143222507e-05,
+      "loss": 0.4131,
+      "step": 75
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.980051150895141e-05,
+      "loss": 0.7309,
+      "step": 76
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9792838874680307e-05,
+      "loss": 0.4279,
+      "step": 77
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.978516624040921e-05,
+      "loss": 0.5813,
+      "step": 78
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.977749360613811e-05,
+      "loss": 0.6276,
+      "step": 79
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976982097186701e-05,
+      "loss": 0.7892,
+      "step": 80
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976214833759591e-05,
+      "loss": 0.7134,
+      "step": 81
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.9754475703324808e-05,
+      "loss": 0.3682,
+      "step": 82
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.974680306905371e-05,
+      "loss": 0.5562,
+      "step": 83
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973913043478261e-05,
+      "loss": 0.7158,
+      "step": 84
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973145780051151e-05,
+      "loss": 0.524,
+      "step": 85
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9723785166240408e-05,
+      "loss": 0.3372,
+      "step": 86
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9716112531969313e-05,
+      "loss": 0.4979,
+      "step": 87
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970843989769821e-05,
+      "loss": 0.3177,
+      "step": 88
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970076726342711e-05,
+      "loss": 0.3488,
+      "step": 89
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.969309462915601e-05,
+      "loss": 0.5158,
+      "step": 90
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9685421994884913e-05,
+      "loss": 0.3208,
+      "step": 91
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.967774936061381e-05,
+      "loss": 0.6654,
+      "step": 92
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9670076726342712e-05,
+      "loss": 0.5581,
+      "step": 93
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.966240409207161e-05,
+      "loss": 0.7456,
+      "step": 94
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9654731457800512e-05,
+      "loss": 0.4437,
+      "step": 95
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9647058823529414e-05,
+      "loss": 0.5525,
+      "step": 96
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9639386189258312e-05,
+      "loss": 0.6673,
+      "step": 97
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.963171355498721e-05,
+      "loss": 0.435,
+      "step": 98
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9624040920716115e-05,
+      "loss": 0.7415,
+      "step": 99
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9616368286445014e-05,
+      "loss": 0.2981,
+      "step": 100
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9608695652173912e-05,
+      "loss": 0.5179,
+      "step": 101
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9601023017902817e-05,
+      "loss": 0.4005,
+      "step": 102
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9593350383631715e-05,
+      "loss": 0.4598,
+      "step": 103
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9585677749360613e-05,
+      "loss": 0.2553,
+      "step": 104
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9578005115089515e-05,
+      "loss": 0.4302,
+      "step": 105
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9570332480818416e-05,
+      "loss": 0.7894,
+      "step": 106
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9562659846547315e-05,
+      "loss": 0.213,
+      "step": 107
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9554987212276216e-05,
+      "loss": 0.5507,
+      "step": 108
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9547314578005115e-05,
+      "loss": 0.7676,
+      "step": 109
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9539641943734016e-05,
+      "loss": 0.3418,
+      "step": 110
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9531969309462918e-05,
+      "loss": 0.8533,
+      "step": 111
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9524296675191816e-05,
+      "loss": 0.3659,
+      "step": 112
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9516624040920714e-05,
+      "loss": 0.6482,
+      "step": 113
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.950895140664962e-05,
+      "loss": 0.5484,
+      "step": 114
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9501278772378517e-05,
+      "loss": 0.4508,
+      "step": 115
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9493606138107416e-05,
+      "loss": 0.4401,
+      "step": 116
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9485933503836317e-05,
+      "loss": 0.5455,
+      "step": 117
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.947826086956522e-05,
+      "loss": 0.722,
+      "step": 118
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9470588235294117e-05,
+      "loss": 0.2874,
+      "step": 119
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.946291560102302e-05,
+      "loss": 0.5973,
+      "step": 120
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.9455242966751917e-05,
+      "loss": 0.5898,
+      "step": 121
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.944757033248082e-05,
+      "loss": 1.0125,
+      "step": 122
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943989769820972e-05,
+      "loss": 0.2511,
+      "step": 123
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943222506393862e-05,
+      "loss": 0.545,
+      "step": 124
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.942455242966752e-05,
+      "loss": 0.593,
+      "step": 125
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.941687979539642e-05,
+      "loss": 0.2319,
+      "step": 126
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.940920716112532e-05,
+      "loss": 0.2215,
+      "step": 127
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.9401534526854218e-05,
+      "loss": 0.1964,
+      "step": 128
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9393861892583123e-05,
+      "loss": 0.1471,
+      "step": 129
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.938618925831202e-05,
+      "loss": 1.5965,
+      "step": 130
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9378516624040923e-05,
+      "loss": 1.0772,
+      "step": 131
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.937084398976982e-05,
+      "loss": 0.2005,
+      "step": 132
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9363171355498723e-05,
+      "loss": 0.9628,
+      "step": 133
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9355498721227624e-05,
+      "loss": 0.1503,
+      "step": 134
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9347826086956523e-05,
+      "loss": 0.6201,
+      "step": 135
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.934015345268542e-05,
+      "loss": 0.6094,
+      "step": 136
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9332480818414326e-05,
+      "loss": 1.385,
+      "step": 137
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9324808184143224e-05,
+      "loss": 0.1552,
+      "step": 138
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9317135549872122e-05,
+      "loss": 0.5828,
+      "step": 139
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9309462915601024e-05,
+      "loss": 0.6018,
+      "step": 140
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9301790281329925e-05,
+      "loss": 1.0614,
+      "step": 141
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9294117647058824e-05,
+      "loss": 0.566,
+      "step": 142
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9286445012787725e-05,
+      "loss": 0.5958,
+      "step": 143
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9278772378516624e-05,
+      "loss": 0.5838,
+      "step": 144
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9271099744245525e-05,
+      "loss": 0.5909,
+      "step": 145
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9263427109974427e-05,
+      "loss": 0.5862,
+      "step": 146
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9255754475703325e-05,
+      "loss": 0.2031,
+      "step": 147
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9248081841432223e-05,
+      "loss": 0.1995,
+      "step": 148
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9240409207161128e-05,
+      "loss": 0.24,
+      "step": 149
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9232736572890026e-05,
+      "loss": 0.5845,
+      "step": 150
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9225063938618925e-05,
+      "loss": 0.964,
+      "step": 151
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.921739130434783e-05,
+      "loss": 0.9595,
+      "step": 152
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9209718670076728e-05,
+      "loss": 0.2033,
+      "step": 153
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9202046035805626e-05,
+      "loss": 0.213,
+      "step": 154
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9194373401534528e-05,
+      "loss": 0.2042,
+      "step": 155
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.918670076726343e-05,
+      "loss": 0.5783,
+      "step": 156
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9179028132992328e-05,
+      "loss": 1.0822,
+      "step": 157
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.917135549872123e-05,
+      "loss": 1.2643,
+      "step": 158
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9163682864450127e-05,
+      "loss": 0.2593,
+      "step": 159
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.915601023017903e-05,
+      "loss": 0.5783,
+      "step": 160
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914833759590793e-05,
+      "loss": 0.5721,
+      "step": 161
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914066496163683e-05,
+      "loss": 0.5808,
+      "step": 162
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9132992327365727e-05,
+      "loss": 1.2578,
+      "step": 163
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9125319693094632e-05,
+      "loss": 0.5709,
+      "step": 164
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.911764705882353e-05,
+      "loss": 0.2159,
+      "step": 165
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910997442455243e-05,
+      "loss": 0.2329,
+      "step": 166
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910230179028133e-05,
+      "loss": 0.926,
+      "step": 167
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.9094629156010232e-05,
+      "loss": 0.9095,
+      "step": 168
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.908695652173913e-05,
+      "loss": 0.5622,
+      "step": 169
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907928388746803e-05,
+      "loss": 0.5703,
+      "step": 170
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907161125319693e-05,
+      "loss": 0.5669,
+      "step": 171
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.906393861892583e-05,
+      "loss": 0.2637,
+      "step": 172
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9056265984654733e-05,
+      "loss": 1.1077,
+      "step": 173
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.904859335038363e-05,
+      "loss": 0.5653,
+      "step": 174
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9040920716112533e-05,
+      "loss": 0.5584,
+      "step": 175
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9033248081841434e-05,
+      "loss": 0.5544,
+      "step": 176
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9025575447570333e-05,
+      "loss": 0.5663,
+      "step": 177
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.901790281329923e-05,
+      "loss": 1.0808,
+      "step": 178
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9010230179028136e-05,
+      "loss": 0.8736,
+      "step": 179
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9002557544757034e-05,
+      "loss": 0.6435,
+      "step": 180
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8994884910485932e-05,
+      "loss": 0.786,
+      "step": 181
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8987212276214834e-05,
+      "loss": 0.5717,
+      "step": 182
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8979539641943736e-05,
+      "loss": 0.5618,
+      "step": 183
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8971867007672637e-05,
+      "loss": 0.5785,
+      "step": 184
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8964194373401535e-05,
+      "loss": 0.7601,
+      "step": 185
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8956521739130434e-05,
+      "loss": 0.8112,
+      "step": 186
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.894884910485934e-05,
+      "loss": 0.5663,
+      "step": 187
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8941176470588237e-05,
+      "loss": 0.5631,
+      "step": 188
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8933503836317135e-05,
+      "loss": 0.7363,
+      "step": 189
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8925831202046037e-05,
+      "loss": 0.4552,
+      "step": 190
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8918158567774938e-05,
+      "loss": 0.4599,
+      "step": 191
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8910485933503836e-05,
+      "loss": 0.7848,
+      "step": 192
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8902813299232738e-05,
+      "loss": 0.5907,
+      "step": 193
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8895140664961636e-05,
+      "loss": 0.4264,
+      "step": 194
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8887468030690538e-05,
+      "loss": 0.5697,
+      "step": 195
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.887979539641944e-05,
+      "loss": 0.7643,
+      "step": 196
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8872122762148338e-05,
+      "loss": 0.3635,
+      "step": 197
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8864450127877236e-05,
+      "loss": 0.3715,
+      "step": 198
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.885677749360614e-05,
+      "loss": 0.554,
+      "step": 199
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.884910485933504e-05,
+      "loss": 0.8044,
+      "step": 200
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8841432225063937e-05,
+      "loss": 0.5673,
+      "step": 201
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8833759590792842e-05,
+      "loss": 0.8042,
+      "step": 202
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.882608695652174e-05,
+      "loss": 0.5638,
+      "step": 203
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881841432225064e-05,
+      "loss": 0.3584,
+      "step": 204
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881074168797954e-05,
+      "loss": 0.8211,
+      "step": 205
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.8803069053708442e-05,
+      "loss": 0.5708,
+      "step": 206
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.879539641943734e-05,
+      "loss": 0.8301,
+      "step": 207
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8787723785166242e-05,
+      "loss": 0.8229,
+      "step": 208
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.878005115089514e-05,
+      "loss": 0.5645,
+      "step": 209
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8772378516624042e-05,
+      "loss": 0.3529,
+      "step": 210
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8764705882352943e-05,
+      "loss": 0.5578,
+      "step": 211
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.875703324808184e-05,
+      "loss": 0.5659,
+      "step": 212
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.874936061381074e-05,
+      "loss": 0.7495,
+      "step": 213
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8741687979539645e-05,
+      "loss": 0.3162,
+      "step": 214
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8734015345268543e-05,
+      "loss": 0.5616,
+      "step": 215
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.872634271099744e-05,
+      "loss": 0.4034,
+      "step": 216
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8718670076726343e-05,
+      "loss": 0.3163,
+      "step": 217
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8710997442455245e-05,
+      "loss": 0.5514,
+      "step": 218
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8703324808184143e-05,
+      "loss": 0.826,
+      "step": 219
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8695652173913044e-05,
+      "loss": 0.8363,
+      "step": 220
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8687979539641943e-05,
+      "loss": 0.8288,
+      "step": 221
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8680306905370844e-05,
+      "loss": 0.5929,
+      "step": 222
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8672634271099746e-05,
+      "loss": 0.565,
+      "step": 223
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8664961636828644e-05,
+      "loss": 0.5642,
+      "step": 224
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8657289002557546e-05,
+      "loss": 0.5609,
+      "step": 225
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8649616368286447e-05,
+      "loss": 0.5641,
+      "step": 226
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8641943734015345e-05,
+      "loss": 0.8535,
+      "step": 227
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8634271099744244e-05,
+      "loss": 0.5652,
+      "step": 228
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.862659846547315e-05,
+      "loss": 0.4397,
+      "step": 229
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8618925831202047e-05,
+      "loss": 0.567,
+      "step": 230
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8611253196930945e-05,
+      "loss": 0.7517,
+      "step": 231
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8603580562659847e-05,
+      "loss": 0.5839,
+      "step": 232
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.859590792838875e-05,
+      "loss": 0.5677,
+      "step": 233
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8588235294117647e-05,
+      "loss": 0.5678,
+      "step": 234
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8580562659846548e-05,
+      "loss": 1.3045,
+      "step": 235
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8572890025575446e-05,
+      "loss": 0.804,
+      "step": 236
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.856521739130435e-05,
+      "loss": 1.0303,
+      "step": 237
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.855754475703325e-05,
+      "loss": 0.3303,
+      "step": 238
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8549872122762148e-05,
+      "loss": 0.5677,
+      "step": 239
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.854219948849105e-05,
+      "loss": 0.7972,
+      "step": 240
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.853452685421995e-05,
+      "loss": 0.5644,
+      "step": 241
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.852685421994885e-05,
+      "loss": 0.9294,
+      "step": 242
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.851918158567775e-05,
+      "loss": 0.5665,
+      "step": 243
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.851150895140665e-05,
+      "loss": 0.7963,
+      "step": 244
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.850383631713555e-05,
+      "loss": 0.5659,
+      "step": 245
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.8496163682864452e-05,
+      "loss": 0.5718,
+      "step": 246
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.848849104859335e-05,
+      "loss": 0.3526,
+      "step": 247
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8480818414322252e-05,
+      "loss": 0.574,
+      "step": 248
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8473145780051154e-05,
+      "loss": 0.9996,
+      "step": 249
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8465473145780052e-05,
+      "loss": 0.7634,
+      "step": 250
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.845780051150895e-05,
+      "loss": 0.5639,
+      "step": 251
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8450127877237855e-05,
+      "loss": 0.5737,
+      "step": 252
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8442455242966753e-05,
+      "loss": 0.7793,
+      "step": 253
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8434782608695652e-05,
+      "loss": 0.6769,
+      "step": 254
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8427109974424553e-05,
+      "loss": 0.5748,
+      "step": 255
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8419437340153455e-05,
+      "loss": 0.3602,
+      "step": 256
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8411764705882353e-05,
+      "loss": 0.5714,
+      "step": 257
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8404092071611255e-05,
+      "loss": 0.5738,
+      "step": 258
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8396419437340153e-05,
+      "loss": 0.5719,
+      "step": 259
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8388746803069055e-05,
+      "loss": 0.5753,
+      "step": 260
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8381074168797956e-05,
+      "loss": 0.5717,
+      "step": 261
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8373401534526854e-05,
+      "loss": 0.9794,
+      "step": 262
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8365728900255753e-05,
+      "loss": 0.5737,
+      "step": 263
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8358056265984658e-05,
+      "loss": 0.5723,
+      "step": 264
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8350383631713556e-05,
+      "loss": 0.5656,
+      "step": 265
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8342710997442454e-05,
+      "loss": 0.4259,
+      "step": 266
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8335038363171356e-05,
+      "loss": 0.3076,
+      "step": 267
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8327365728900257e-05,
+      "loss": 0.338,
+      "step": 268
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8319693094629156e-05,
+      "loss": 0.5612,
+      "step": 269
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8312020460358057e-05,
+      "loss": 0.5249,
+      "step": 270
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8304347826086955e-05,
+      "loss": 0.5671,
+      "step": 271
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8296675191815857e-05,
+      "loss": 0.5574,
+      "step": 272
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.828900255754476e-05,
+      "loss": 0.8152,
+      "step": 273
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8281329923273657e-05,
+      "loss": 0.5647,
+      "step": 274
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.827365728900256e-05,
+      "loss": 0.5632,
+      "step": 275
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.826598465473146e-05,
+      "loss": 0.3391,
+      "step": 276
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8258312020460358e-05,
+      "loss": 0.55,
+      "step": 277
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8250639386189257e-05,
+      "loss": 0.5672,
+      "step": 278
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.824296675191816e-05,
+      "loss": 0.285,
+      "step": 279
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.823529411764706e-05,
+      "loss": 0.2809,
+      "step": 280
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.8227621483375958e-05,
+      "loss": 0.5552,
+      "step": 281
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821994884910486e-05,
+      "loss": 0.563,
+      "step": 282
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821227621483376e-05,
+      "loss": 0.5628,
+      "step": 283
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.820460358056266e-05,
+      "loss": 0.2548,
+      "step": 284
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.819693094629156e-05,
+      "loss": 0.871,
+      "step": 285
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818925831202046e-05,
+      "loss": 0.8797,
+      "step": 286
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818158567774936e-05,
+      "loss": 0.2453,
+      "step": 287
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8173913043478262e-05,
+      "loss": 0.2464,
+      "step": 288
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.816624040920716e-05,
+      "loss": 0.3134,
+      "step": 289
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8158567774936062e-05,
+      "loss": 0.2431,
+      "step": 290
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8150895140664964e-05,
+      "loss": 0.2308,
+      "step": 291
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8143222506393862e-05,
+      "loss": 0.2173,
+      "step": 292
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8135549872122764e-05,
+      "loss": 0.5665,
+      "step": 293
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8127877237851662e-05,
+      "loss": 0.5592,
+      "step": 294
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8120204603580564e-05,
+      "loss": 0.473,
+      "step": 295
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8112531969309465e-05,
+      "loss": 0.9458,
+      "step": 296
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8104859335038363e-05,
+      "loss": 0.1879,
+      "step": 297
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8097186700767265e-05,
+      "loss": 0.6797,
+      "step": 298
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8089514066496167e-05,
+      "loss": 0.7592,
+      "step": 299
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8081841432225065e-05,
+      "loss": 0.4109,
+      "step": 300
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8074168797953963e-05,
+      "loss": 0.5821,
+      "step": 301
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8066496163682868e-05,
+      "loss": 0.2057,
+      "step": 302
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8058823529411766e-05,
+      "loss": 0.5322,
+      "step": 303
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8051150895140665e-05,
+      "loss": 0.5208,
+      "step": 304
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8043478260869566e-05,
+      "loss": 0.6799,
+      "step": 305
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8035805626598468e-05,
+      "loss": 0.5894,
+      "step": 306
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8028132992327366e-05,
+      "loss": 0.4051,
+      "step": 307
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8020460358056268e-05,
+      "loss": 0.3695,
+      "step": 308
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8012787723785166e-05,
+      "loss": 0.4869,
+      "step": 309
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8005115089514067e-05,
+      "loss": 0.6075,
+      "step": 310
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.799744245524297e-05,
+      "loss": 0.4245,
+      "step": 311
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7989769820971867e-05,
+      "loss": 0.5122,
+      "step": 312
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7982097186700765e-05,
+      "loss": 0.452,
+      "step": 313
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.797442455242967e-05,
+      "loss": 0.9568,
+      "step": 314
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.796675191815857e-05,
+      "loss": 0.5846,
+      "step": 315
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.7959079283887467e-05,
+      "loss": 0.3743,
+      "step": 316
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.795140664961637e-05,
+      "loss": 0.1922,
+      "step": 317
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.794373401534527e-05,
+      "loss": 0.5303,
+      "step": 318
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.793606138107417e-05,
+      "loss": 0.7253,
+      "step": 319
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.792838874680307e-05,
+      "loss": 0.2794,
+      "step": 320
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.7920716112531968e-05,
+      "loss": 0.577,
+      "step": 321
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.791304347826087e-05,
+      "loss": 0.614,
+      "step": 322
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.790537084398977e-05,
+      "loss": 0.5882,
+      "step": 323
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789769820971867e-05,
+      "loss": 0.2503,
+      "step": 324
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789002557544757e-05,
+      "loss": 0.4195,
+      "step": 325
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.7882352941176473e-05,
+      "loss": 0.2959,
+      "step": 326
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.787468030690537e-05,
+      "loss": 0.3732,
+      "step": 327
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.786700767263427e-05,
+      "loss": 0.5729,
+      "step": 328
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7859335038363174e-05,
+      "loss": 0.3179,
+      "step": 329
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7851662404092073e-05,
+      "loss": 0.6823,
+      "step": 330
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.784398976982097e-05,
+      "loss": 0.2663,
+      "step": 331
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7836317135549872e-05,
+      "loss": 0.4436,
+      "step": 332
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7828644501278774e-05,
+      "loss": 0.3089,
+      "step": 333
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7820971867007672e-05,
+      "loss": 0.5888,
+      "step": 334
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7813299232736574e-05,
+      "loss": 0.2115,
+      "step": 335
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7805626598465472e-05,
+      "loss": 0.4005,
+      "step": 336
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7797953964194374e-05,
+      "loss": 0.3097,
+      "step": 337
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7790281329923275e-05,
+      "loss": 0.5847,
+      "step": 338
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7782608695652174e-05,
+      "loss": 1.0164,
+      "step": 339
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7774936061381072e-05,
+      "loss": 1.1321,
+      "step": 340
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7767263427109977e-05,
+      "loss": 0.592,
+      "step": 341
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7759590792838875e-05,
+      "loss": 1.0191,
+      "step": 342
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7751918158567777e-05,
+      "loss": 0.9199,
+      "step": 343
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7744245524296675e-05,
+      "loss": 0.2874,
+      "step": 344
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7736572890025576e-05,
+      "loss": 0.2469,
+      "step": 345
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7728900255754478e-05,
+      "loss": 0.8364,
+      "step": 346
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7721227621483376e-05,
+      "loss": 0.7365,
+      "step": 347
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7713554987212278e-05,
+      "loss": 0.5595,
+      "step": 348
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.770588235294118e-05,
+      "loss": 0.2546,
+      "step": 349
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7698209718670078e-05,
+      "loss": 0.8183,
+      "step": 350
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7690537084398976e-05,
+      "loss": 0.5385,
+      "step": 351
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.768286445012788e-05,
+      "loss": 0.2763,
+      "step": 352
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.767519181585678e-05,
+      "loss": 0.3045,
+      "step": 353
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7667519181585677e-05,
+      "loss": 0.3354,
+      "step": 354
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765984654731458e-05,
+      "loss": 0.9366,
+      "step": 355
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765217391304348e-05,
+      "loss": 0.6212,
+      "step": 356
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.764450127877238e-05,
+      "loss": 0.1161,
+      "step": 357
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.763682864450128e-05,
+      "loss": 0.4929,
+      "step": 358
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762915601023018e-05,
+      "loss": 0.8167,
+      "step": 359
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762148337595908e-05,
+      "loss": 0.2521,
+      "step": 360
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.7613810741687982e-05,
+      "loss": 0.4174,
+      "step": 361
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.760613810741688e-05,
+      "loss": 0.3822,
+      "step": 362
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.759846547314578e-05,
+      "loss": 0.5355,
+      "step": 363
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.7590792838874683e-05,
+      "loss": 0.561,
+      "step": 364
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.758312020460358e-05,
+      "loss": 0.5126,
+      "step": 365
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.757544757033248e-05,
+      "loss": 0.2301,
+      "step": 366
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.756777493606138e-05,
+      "loss": 0.1698,
+      "step": 367
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7560102301790283e-05,
+      "loss": 0.1634,
+      "step": 368
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.755242966751918e-05,
+      "loss": 0.5955,
+      "step": 369
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7544757033248083e-05,
+      "loss": 0.1568,
+      "step": 370
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.753708439897698e-05,
+      "loss": 0.5167,
+      "step": 371
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7529411764705883e-05,
+      "loss": 1.0028,
+      "step": 372
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7521739130434784e-05,
+      "loss": 1.2567,
+      "step": 373
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7514066496163682e-05,
+      "loss": 0.6094,
+      "step": 374
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7506393861892584e-05,
+      "loss": 0.6059,
+      "step": 375
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7498721227621486e-05,
+      "loss": 1.0593,
+      "step": 376
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7491048593350384e-05,
+      "loss": 0.1506,
+      "step": 377
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7483375959079282e-05,
+      "loss": 1.5106,
+      "step": 378
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7475703324808187e-05,
+      "loss": 0.1827,
+      "step": 379
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7468030690537085e-05,
+      "loss": 1.4961,
+      "step": 380
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7460358056265984e-05,
+      "loss": 1.0456,
+      "step": 381
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7452685421994885e-05,
+      "loss": 1.2063,
+      "step": 382
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7445012787723787e-05,
+      "loss": 0.5837,
+      "step": 383
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7437340153452685e-05,
+      "loss": 1.4536,
+      "step": 384
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7429667519181587e-05,
+      "loss": 0.5803,
+      "step": 385
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7421994884910485e-05,
+      "loss": 0.5896,
+      "step": 386
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7414322250639386e-05,
+      "loss": 0.1822,
+      "step": 387
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7406649616368288e-05,
+      "loss": 1.7893,
+      "step": 388
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7398976982097186e-05,
+      "loss": 0.9723,
+      "step": 389
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7391304347826085e-05,
+      "loss": 1.2815,
+      "step": 390
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.738363171355499e-05,
+      "loss": 0.5765,
+      "step": 391
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7375959079283888e-05,
+      "loss": 1.2429,
+      "step": 392
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7368286445012786e-05,
+      "loss": 0.1343,
+      "step": 393
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7360613810741688e-05,
+      "loss": 0.2306,
+      "step": 394
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.735294117647059e-05,
+      "loss": 0.624,
+      "step": 395
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.734526854219949e-05,
+      "loss": 0.2371,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7348484848484849,
+      "eval_loss": 0.5804409980773926,
+      "eval_runtime": 57.3336,
+      "eval_samples_per_second": 6.907,
+      "eval_steps_per_second": 1.727,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.733759590792839e-05,
+      "loss": 0.8989,
+      "step": 397
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.732992327365729e-05,
+      "loss": 0.5664,
+      "step": 398
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.7322250639386192e-05,
+      "loss": 0.8852,
+      "step": 399
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.731457800511509e-05,
+      "loss": 0.5661,
+      "step": 400
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.730690537084399e-05,
+      "loss": 0.5648,
+      "step": 401
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7299232736572894e-05,
+      "loss": 0.7333,
+      "step": 402
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7291560102301792e-05,
+      "loss": 0.3037,
+      "step": 403
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.728388746803069e-05,
+      "loss": 0.1463,
+      "step": 404
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7276214833759592e-05,
+      "loss": 0.2702,
+      "step": 405
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7268542199488493e-05,
+      "loss": 0.2693,
+      "step": 406
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.726086956521739e-05,
+      "loss": 0.5647,
+      "step": 407
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7253196930946293e-05,
+      "loss": 0.8507,
+      "step": 408
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.724552429667519e-05,
+      "loss": 0.8645,
+      "step": 409
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7237851662404093e-05,
+      "loss": 0.2649,
+      "step": 410
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7230179028132995e-05,
+      "loss": 0.6942,
+      "step": 411
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7222506393861893e-05,
+      "loss": 0.5656,
+      "step": 412
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.721483375959079e-05,
+      "loss": 0.5489,
+      "step": 413
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7207161125319696e-05,
+      "loss": 0.2652,
+      "step": 414
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7199488491048594e-05,
+      "loss": 0.8676,
+      "step": 415
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7191815856777493e-05,
+      "loss": 0.8621,
+      "step": 416
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7184143222506394e-05,
+      "loss": 0.2641,
+      "step": 417
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7176470588235296e-05,
+      "loss": 0.859,
+      "step": 418
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7168797953964194e-05,
+      "loss": 0.6492,
+      "step": 419
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7161125319693096e-05,
+      "loss": 1.1494,
+      "step": 420
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7153452685421994e-05,
+      "loss": 0.8278,
+      "step": 421
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7145780051150895e-05,
+      "loss": 0.764,
+      "step": 422
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7138107416879797e-05,
+      "loss": 0.564,
+      "step": 423
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7130434782608695e-05,
+      "loss": 0.8051,
+      "step": 424
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7122762148337597e-05,
+      "loss": 0.3052,
+      "step": 425
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.71150895140665e-05,
+      "loss": 0.2945,
+      "step": 426
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7107416879795397e-05,
+      "loss": 0.4118,
+      "step": 427
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7099744245524295e-05,
+      "loss": 0.559,
+      "step": 428
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.70920716112532e-05,
+      "loss": 0.5593,
+      "step": 429
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7084398976982098e-05,
+      "loss": 0.564,
+      "step": 430
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7076726342710996e-05,
+      "loss": 0.4454,
+      "step": 431
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7069053708439898e-05,
+      "loss": 0.4381,
+      "step": 432
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.70613810741688e-05,
+      "loss": 0.3718,
+      "step": 433
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7053708439897698e-05,
+      "loss": 0.6627,
+      "step": 434
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70460358056266e-05,
+      "loss": 0.4004,
+      "step": 435
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7038363171355498e-05,
+      "loss": 0.3855,
+      "step": 436
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70306905370844e-05,
+      "loss": 0.4799,
+      "step": 437
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70230179028133e-05,
+      "loss": 0.6714,
+      "step": 438
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70153452685422e-05,
+      "loss": 0.622,
+      "step": 439
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7007672634271097e-05,
+      "loss": 0.3916,
+      "step": 440
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.5487,
+      "step": 441
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69923273657289e-05,
+      "loss": 0.376,
+      "step": 442
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69846547314578e-05,
+      "loss": 0.3616,
+      "step": 443
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69769820971867e-05,
+      "loss": 0.8747,
+      "step": 444
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.6969309462915602e-05,
+      "loss": 0.283,
+      "step": 445
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.69616368286445e-05,
+      "loss": 0.3988,
+      "step": 446
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6953964194373402e-05,
+      "loss": 0.4569,
+      "step": 447
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6946291560102303e-05,
+      "loss": 0.2005,
+      "step": 448
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6938618925831205e-05,
+      "loss": 0.3087,
+      "step": 449
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6930946291560103e-05,
+      "loss": 0.5814,
+      "step": 450
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6923273657289e-05,
+      "loss": 0.2995,
+      "step": 451
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6915601023017907e-05,
+      "loss": 0.7806,
+      "step": 452
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6907928388746805e-05,
+      "loss": 0.2666,
+      "step": 453
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6900255754475703e-05,
+      "loss": 0.2571,
+      "step": 454
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6892583120204605e-05,
+      "loss": 0.5931,
+      "step": 455
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6884910485933506e-05,
+      "loss": 0.475,
+      "step": 456
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6877237851662404e-05,
+      "loss": 0.2372,
+      "step": 457
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6869565217391306e-05,
+      "loss": 0.1577,
+      "step": 458
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6861892583120204e-05,
+      "loss": 0.1953,
+      "step": 459
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6854219948849106e-05,
+      "loss": 0.1931,
+      "step": 460
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6846547314578008e-05,
+      "loss": 0.1829,
+      "step": 461
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6838874680306906e-05,
+      "loss": 0.1601,
+      "step": 462
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6831202046035804e-05,
+      "loss": 0.2505,
+      "step": 463
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.682352941176471e-05,
+      "loss": 0.181,
+      "step": 464
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6815856777493607e-05,
+      "loss": 0.5978,
+      "step": 465
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6808184143222505e-05,
+      "loss": 0.5458,
+      "step": 466
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6800511508951407e-05,
+      "loss": 0.1289,
+      "step": 467
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.679283887468031e-05,
+      "loss": 0.4383,
+      "step": 468
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6785166240409207e-05,
+      "loss": 0.5892,
+      "step": 469
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.677749360613811e-05,
+      "loss": 1.3565,
+      "step": 470
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.676982097186701e-05,
+      "loss": 0.7267,
+      "step": 471
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.6762148337595908e-05,
+      "loss": 0.1354,
+      "step": 472
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.675447570332481e-05,
+      "loss": 0.9291,
+      "step": 473
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.6746803069053708e-05,
+      "loss": 0.5265,
+      "step": 474
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673913043478261e-05,
+      "loss": 0.4841,
+      "step": 475
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673145780051151e-05,
+      "loss": 1.2391,
+      "step": 476
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.672378516624041e-05,
+      "loss": 0.4924,
+      "step": 477
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6716112531969308e-05,
+      "loss": 1.0076,
+      "step": 478
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6708439897698213e-05,
+      "loss": 0.6314,
+      "step": 479
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.670076726342711e-05,
+      "loss": 0.4237,
+      "step": 480
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.669309462915601e-05,
+      "loss": 0.1077,
+      "step": 481
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.668542199488491e-05,
+      "loss": 0.3038,
+      "step": 482
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6677749360613812e-05,
+      "loss": 0.6081,
+      "step": 483
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.667007672634271e-05,
+      "loss": 0.1078,
+      "step": 484
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6662404092071612e-05,
+      "loss": 0.4296,
+      "step": 485
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.665473145780051e-05,
+      "loss": 0.3981,
+      "step": 486
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6647058823529412e-05,
+      "loss": 0.5853,
+      "step": 487
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6639386189258314e-05,
+      "loss": 0.802,
+      "step": 488
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6631713554987212e-05,
+      "loss": 0.4576,
+      "step": 489
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.662404092071611e-05,
+      "loss": 0.2682,
+      "step": 490
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6616368286445015e-05,
+      "loss": 0.1991,
+      "step": 491
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6608695652173913e-05,
+      "loss": 0.1409,
+      "step": 492
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.660102301790281e-05,
+      "loss": 0.3869,
+      "step": 493
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6593350383631713e-05,
+      "loss": 0.5301,
+      "step": 494
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6585677749360615e-05,
+      "loss": 0.8116,
+      "step": 495
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6578005115089513e-05,
+      "loss": 0.0984,
+      "step": 496
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6570332480818415e-05,
+      "loss": 0.7773,
+      "step": 497
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6562659846547316e-05,
+      "loss": 1.3522,
+      "step": 498
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6554987212276215e-05,
+      "loss": 0.0862,
+      "step": 499
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6547314578005116e-05,
+      "loss": 0.0894,
+      "step": 500
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6539641943734014e-05,
+      "loss": 0.0885,
+      "step": 501
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.653196930946292e-05,
+      "loss": 0.3031,
+      "step": 502
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6524296675191818e-05,
+      "loss": 0.187,
+      "step": 503
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6516624040920716e-05,
+      "loss": 1.3063,
+      "step": 504
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6508951406649617e-05,
+      "loss": 0.6847,
+      "step": 505
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.650127877237852e-05,
+      "loss": 0.078,
+      "step": 506
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6493606138107417e-05,
+      "loss": 0.1265,
+      "step": 507
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.648593350383632e-05,
+      "loss": 0.0804,
+      "step": 508
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.6478260869565217e-05,
+      "loss": 0.6976,
+      "step": 509
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.647058823529412e-05,
+      "loss": 0.7977,
+      "step": 510
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.646291560102302e-05,
+      "loss": 0.413,
+      "step": 511
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.645524296675192e-05,
+      "loss": 0.6944,
+      "step": 512
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6447570332480817e-05,
+      "loss": 0.7177,
+      "step": 513
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6439897698209722e-05,
+      "loss": 0.0581,
+      "step": 514
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.643222506393862e-05,
+      "loss": 0.9308,
+      "step": 515
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6424552429667518e-05,
+      "loss": 0.2518,
+      "step": 516
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.641687979539642e-05,
+      "loss": 0.1014,
+      "step": 517
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640920716112532e-05,
+      "loss": 0.1641,
+      "step": 518
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640153452685422e-05,
+      "loss": 0.101,
+      "step": 519
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.639386189258312e-05,
+      "loss": 0.1097,
+      "step": 520
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6386189258312023e-05,
+      "loss": 0.987,
+      "step": 521
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.637851662404092e-05,
+      "loss": 0.6465,
+      "step": 522
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6370843989769823e-05,
+      "loss": 0.0914,
+      "step": 523
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.636317135549872e-05,
+      "loss": 0.139,
+      "step": 524
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6355498721227623e-05,
+      "loss": 0.6455,
+      "step": 525
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6347826086956524e-05,
+      "loss": 0.204,
+      "step": 526
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6340153452685422e-05,
+      "loss": 0.1269,
+      "step": 527
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.633248081841432e-05,
+      "loss": 0.2214,
+      "step": 528
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6324808184143226e-05,
+      "loss": 0.0736,
+      "step": 529
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6317135549872124e-05,
+      "loss": 0.0714,
+      "step": 530
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6309462915601022e-05,
+      "loss": 0.1388,
+      "step": 531
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6301790281329924e-05,
+      "loss": 0.7758,
+      "step": 532
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6294117647058825e-05,
+      "loss": 0.3388,
+      "step": 533
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6286445012787724e-05,
+      "loss": 1.1831,
+      "step": 534
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6278772378516625e-05,
+      "loss": 0.1166,
+      "step": 535
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6271099744245523e-05,
+      "loss": 0.2389,
+      "step": 536
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6263427109974425e-05,
+      "loss": 0.1622,
+      "step": 537
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6255754475703327e-05,
+      "loss": 0.1015,
+      "step": 538
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6248081841432225e-05,
+      "loss": 0.0658,
+      "step": 539
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6240409207161123e-05,
+      "loss": 0.0954,
+      "step": 540
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6232736572890028e-05,
+      "loss": 0.1182,
+      "step": 541
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6225063938618926e-05,
+      "loss": 0.0605,
+      "step": 542
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6217391304347824e-05,
+      "loss": 0.9265,
+      "step": 543
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6209718670076726e-05,
+      "loss": 1.2994,
+      "step": 544
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6202046035805628e-05,
+      "loss": 0.081,
+      "step": 545
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6194373401534526e-05,
+      "loss": 0.5962,
+      "step": 546
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6186700767263428e-05,
+      "loss": 0.0665,
+      "step": 547
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.617902813299233e-05,
+      "loss": 0.0655,
+      "step": 548
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6171355498721227e-05,
+      "loss": 0.0642,
+      "step": 549
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.616368286445013e-05,
+      "loss": 0.0834,
+      "step": 550
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6156010230179027e-05,
+      "loss": 0.047,
+      "step": 551
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6148337595907932e-05,
+      "loss": 0.0642,
+      "step": 552
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.614066496163683e-05,
+      "loss": 0.0594,
+      "step": 553
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.613299232736573e-05,
+      "loss": 0.0573,
+      "step": 554
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.612531969309463e-05,
+      "loss": 0.1101,
+      "step": 555
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.6117647058823532e-05,
+      "loss": 0.5121,
+      "step": 556
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.610997442455243e-05,
+      "loss": 0.0374,
+      "step": 557
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.6102301790281332e-05,
+      "loss": 0.0957,
+      "step": 558
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.609462915601023e-05,
+      "loss": 0.0488,
+      "step": 559
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.608695652173913e-05,
+      "loss": 0.0482,
+      "step": 560
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6079283887468033e-05,
+      "loss": 1.2041,
+      "step": 561
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.607161125319693e-05,
+      "loss": 0.8833,
+      "step": 562
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.606393861892583e-05,
+      "loss": 0.1544,
+      "step": 563
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6056265984654735e-05,
+      "loss": 0.046,
+      "step": 564
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6048593350383633e-05,
+      "loss": 0.9073,
+      "step": 565
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.604092071611253e-05,
+      "loss": 0.052,
+      "step": 566
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6033248081841433e-05,
+      "loss": 0.0308,
+      "step": 567
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6025575447570334e-05,
+      "loss": 0.7032,
+      "step": 568
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6017902813299232e-05,
+      "loss": 0.6947,
+      "step": 569
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6010230179028134e-05,
+      "loss": 1.5603,
+      "step": 570
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6002557544757036e-05,
+      "loss": 0.8836,
+      "step": 571
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.5994884910485934e-05,
+      "loss": 1.7439,
+      "step": 572
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5987212276214836e-05,
+      "loss": 0.0313,
+      "step": 573
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5979539641943734e-05,
+      "loss": 0.0317,
+      "step": 574
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5971867007672635e-05,
+      "loss": 0.0325,
+      "step": 575
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5964194373401537e-05,
+      "loss": 0.5,
+      "step": 576
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5956521739130435e-05,
+      "loss": 0.0675,
+      "step": 577
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5948849104859333e-05,
+      "loss": 0.6364,
+      "step": 578
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.594117647058824e-05,
+      "loss": 0.0696,
+      "step": 579
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5933503836317137e-05,
+      "loss": 0.6195,
+      "step": 580
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5925831202046035e-05,
+      "loss": 0.5796,
+      "step": 581
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5918158567774937e-05,
+      "loss": 0.0832,
+      "step": 582
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5910485933503838e-05,
+      "loss": 0.0794,
+      "step": 583
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5902813299232736e-05,
+      "loss": 0.5851,
+      "step": 584
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5895140664961638e-05,
+      "loss": 0.104,
+      "step": 585
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5887468030690536e-05,
+      "loss": 1.0492,
+      "step": 586
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5879795396419438e-05,
+      "loss": 0.0645,
+      "step": 587
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.587212276214834e-05,
+      "loss": 0.514,
+      "step": 588
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5864450127877238e-05,
+      "loss": 0.0701,
+      "step": 589
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5856777493606136e-05,
+      "loss": 0.5409,
+      "step": 590
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584910485933504e-05,
+      "loss": 0.0536,
+      "step": 591
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584143222506394e-05,
+      "loss": 0.1728,
+      "step": 592
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.5833759590792837e-05,
+      "loss": 0.0845,
+      "step": 593
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.582608695652174e-05,
+      "loss": 0.2217,
+      "step": 594
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.581841432225064e-05,
+      "loss": 0.1397,
+      "step": 595
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.581074168797954e-05,
+      "loss": 0.4119,
+      "step": 596
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.580306905370844e-05,
+      "loss": 0.3638,
+      "step": 597
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.5795396419437342e-05,
+      "loss": 0.0781,
+      "step": 598
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.578772378516624e-05,
+      "loss": 0.5926,
+      "step": 599
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5780051150895142e-05,
+      "loss": 0.1378,
+      "step": 600
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.577237851662404e-05,
+      "loss": 0.142,
+      "step": 601
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.576470588235294e-05,
+      "loss": 0.1722,
+      "step": 602
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5757033248081843e-05,
+      "loss": 0.1529,
+      "step": 603
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.574936061381074e-05,
+      "loss": 1.6083,
+      "step": 604
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5741687979539643e-05,
+      "loss": 1.8267,
+      "step": 605
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5734015345268545e-05,
+      "loss": 0.0535,
+      "step": 606
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5726342710997443e-05,
+      "loss": 0.9945,
+      "step": 607
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5718670076726345e-05,
+      "loss": 0.9133,
+      "step": 608
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5710997442455243e-05,
+      "loss": 0.5295,
+      "step": 609
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5703324808184144e-05,
+      "loss": 0.058,
+      "step": 610
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5695652173913046e-05,
+      "loss": 0.4789,
+      "step": 611
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5687979539641944e-05,
+      "loss": 0.0572,
+      "step": 612
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5680306905370842e-05,
+      "loss": 0.0897,
+      "step": 613
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5672634271099747e-05,
+      "loss": 0.056,
+      "step": 614
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5664961636828646e-05,
+      "loss": 0.0278,
+      "step": 615
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5657289002557544e-05,
+      "loss": 0.9947,
+      "step": 616
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5649616368286445e-05,
+      "loss": 0.0627,
+      "step": 617
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5641943734015347e-05,
+      "loss": 0.9335,
+      "step": 618
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5634271099744245e-05,
+      "loss": 0.9793,
+      "step": 619
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5626598465473147e-05,
+      "loss": 1.8128,
+      "step": 620
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.561892583120205e-05,
+      "loss": 1.8015,
+      "step": 621
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5611253196930947e-05,
+      "loss": 0.9259,
+      "step": 622
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.560358056265985e-05,
+      "loss": 0.9618,
+      "step": 623
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5595907928388747e-05,
+      "loss": 0.0808,
+      "step": 624
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5588235294117648e-05,
+      "loss": 1.7161,
+      "step": 625
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.558056265984655e-05,
+      "loss": 0.8879,
+      "step": 626
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5572890025575448e-05,
+      "loss": 0.043,
+      "step": 627
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5565217391304346e-05,
+      "loss": 0.9449,
+      "step": 628
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.555754475703325e-05,
+      "loss": 0.0624,
+      "step": 629
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.554987212276215e-05,
+      "loss": 0.0924,
+      "step": 630
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5542199488491048e-05,
+      "loss": 0.0715,
+      "step": 631
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.553452685421995e-05,
+      "loss": 0.4995,
+      "step": 632
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.552685421994885e-05,
+      "loss": 0.5576,
+      "step": 633
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551918158567775e-05,
+      "loss": 0.06,
+      "step": 634
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551150895140665e-05,
+      "loss": 0.0873,
+      "step": 635
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.550383631713555e-05,
+      "loss": 0.6868,
+      "step": 636
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.549616368286445e-05,
+      "loss": 0.8487,
+      "step": 637
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.5488491048593352e-05,
+      "loss": 0.5641,
+      "step": 638
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.548081841432225e-05,
+      "loss": 1.1343,
+      "step": 639
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.547314578005115e-05,
+      "loss": 0.0614,
+      "step": 640
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5465473145780054e-05,
+      "loss": 0.9414,
+      "step": 641
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5457800511508952e-05,
+      "loss": 0.9226,
+      "step": 642
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.545012787723785e-05,
+      "loss": 0.3323,
+      "step": 643
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5442455242966752e-05,
+      "loss": 0.9315,
+      "step": 644
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5434782608695653e-05,
+      "loss": 0.5539,
+      "step": 645
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.542710997442455e-05,
+      "loss": 0.3376,
+      "step": 646
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5419437340153453e-05,
+      "loss": 0.0694,
+      "step": 647
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5411764705882355e-05,
+      "loss": 0.0706,
+      "step": 648
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5404092071611253e-05,
+      "loss": 0.105,
+      "step": 649
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5396419437340155e-05,
+      "loss": 0.912,
+      "step": 650
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5388746803069053e-05,
+      "loss": 0.0998,
+      "step": 651
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5381074168797954e-05,
+      "loss": 0.2247,
+      "step": 652
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5373401534526856e-05,
+      "loss": 0.2869,
+      "step": 653
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5365728900255754e-05,
+      "loss": 0.532,
+      "step": 654
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5358056265984653e-05,
+      "loss": 0.4723,
+      "step": 655
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5350383631713558e-05,
+      "loss": 0.1178,
+      "step": 656
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5342710997442456e-05,
+      "loss": 0.0566,
+      "step": 657
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5335038363171357e-05,
+      "loss": 0.059,
+      "step": 658
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5327365728900256e-05,
+      "loss": 0.0881,
+      "step": 659
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5319693094629157e-05,
+      "loss": 0.5762,
+      "step": 660
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.531202046035806e-05,
+      "loss": 0.0499,
+      "step": 661
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5304347826086957e-05,
+      "loss": 0.087,
+      "step": 662
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5296675191815855e-05,
+      "loss": 1.0719,
+      "step": 663
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528900255754476e-05,
+      "loss": 0.6158,
+      "step": 664
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528132992327366e-05,
+      "loss": 0.6117,
+      "step": 665
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5273657289002557e-05,
+      "loss": 0.0515,
+      "step": 666
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5265984654731458e-05,
+      "loss": 0.0608,
+      "step": 667
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.525831202046036e-05,
+      "loss": 0.0313,
+      "step": 668
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.5250639386189258e-05,
+      "loss": 0.1449,
+      "step": 669
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.524296675191816e-05,
+      "loss": 0.9883,
+      "step": 670
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.523529411764706e-05,
+      "loss": 0.0859,
+      "step": 671
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.522762148337596e-05,
+      "loss": 0.0561,
+      "step": 672
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521994884910486e-05,
+      "loss": 0.0926,
+      "step": 673
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521227621483376e-05,
+      "loss": 0.1101,
+      "step": 674
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.520460358056266e-05,
+      "loss": 0.068,
+      "step": 675
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5196930946291563e-05,
+      "loss": 0.0773,
+      "step": 676
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518925831202046e-05,
+      "loss": 0.0508,
+      "step": 677
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518158567774936e-05,
+      "loss": 0.0375,
+      "step": 678
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5173913043478264e-05,
+      "loss": 0.93,
+      "step": 679
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5166240409207162e-05,
+      "loss": 0.1507,
+      "step": 680
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.515856777493606e-05,
+      "loss": 0.0917,
+      "step": 681
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5150895140664962e-05,
+      "loss": 0.97,
+      "step": 682
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5143222506393864e-05,
+      "loss": 0.5019,
+      "step": 683
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5135549872122762e-05,
+      "loss": 0.0849,
+      "step": 684
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5127877237851664e-05,
+      "loss": 1.2316,
+      "step": 685
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5120204603580562e-05,
+      "loss": 0.0252,
+      "step": 686
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5112531969309463e-05,
+      "loss": 1.0543,
+      "step": 687
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5104859335038365e-05,
+      "loss": 0.0506,
+      "step": 688
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5097186700767263e-05,
+      "loss": 0.0252,
+      "step": 689
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.508951406649616e-05,
+      "loss": 0.6297,
+      "step": 690
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5081841432225066e-05,
+      "loss": 0.4497,
+      "step": 691
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5074168797953965e-05,
+      "loss": 0.0516,
+      "step": 692
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5066496163682863e-05,
+      "loss": 0.5501,
+      "step": 693
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5058823529411768e-05,
+      "loss": 0.5621,
+      "step": 694
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5051150895140666e-05,
+      "loss": 0.0577,
+      "step": 695
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5043478260869564e-05,
+      "loss": 0.0271,
+      "step": 696
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5035805626598466e-05,
+      "loss": 0.9496,
+      "step": 697
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5028132992327368e-05,
+      "loss": 0.0877,
+      "step": 698
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5020460358056266e-05,
+      "loss": 0.0523,
+      "step": 699
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5012787723785167e-05,
+      "loss": 0.0577,
+      "step": 700
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5005115089514066e-05,
+      "loss": 0.0522,
+      "step": 701
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.4997442455242967e-05,
+      "loss": 0.0517,
+      "step": 702
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.498976982097187e-05,
+      "loss": 0.6335,
+      "step": 703
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4982097186700767e-05,
+      "loss": 0.05,
+      "step": 704
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4974424552429665e-05,
+      "loss": 0.0753,
+      "step": 705
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.496675191815857e-05,
+      "loss": 0.0227,
+      "step": 706
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.495907928388747e-05,
+      "loss": 0.0755,
+      "step": 707
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.4951406649616367e-05,
+      "loss": 0.0215,
+      "step": 708
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.494373401534527e-05,
+      "loss": 0.0451,
+      "step": 709
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.493606138107417e-05,
+      "loss": 0.0343,
+      "step": 710
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492838874680307e-05,
+      "loss": 0.0836,
+      "step": 711
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492071611253197e-05,
+      "loss": 0.5287,
+      "step": 712
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4913043478260868e-05,
+      "loss": 0.0686,
+      "step": 713
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4905370843989773e-05,
+      "loss": 0.0211,
+      "step": 714
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489769820971867e-05,
+      "loss": 0.5932,
+      "step": 715
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489002557544757e-05,
+      "loss": 0.0371,
+      "step": 716
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.488235294117647e-05,
+      "loss": 1.0137,
+      "step": 717
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.4874680306905373e-05,
+      "loss": 0.9928,
+      "step": 718
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.486700767263427e-05,
+      "loss": 0.7431,
+      "step": 719
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4859335038363173e-05,
+      "loss": 0.0207,
+      "step": 720
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4851662404092074e-05,
+      "loss": 0.0681,
+      "step": 721
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4843989769820972e-05,
+      "loss": 0.0306,
+      "step": 722
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4836317135549874e-05,
+      "loss": 0.0204,
+      "step": 723
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4828644501278772e-05,
+      "loss": 0.0344,
+      "step": 724
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4820971867007674e-05,
+      "loss": 0.0345,
+      "step": 725
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4813299232736575e-05,
+      "loss": 0.9275,
+      "step": 726
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4805626598465474e-05,
+      "loss": 0.0196,
+      "step": 727
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4797953964194372e-05,
+      "loss": 1.0137,
+      "step": 728
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4790281329923277e-05,
+      "loss": 0.0318,
+      "step": 729
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4782608695652175e-05,
+      "loss": 0.0688,
+      "step": 730
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4774936061381073e-05,
+      "loss": 0.0314,
+      "step": 731
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4767263427109975e-05,
+      "loss": 0.7135,
+      "step": 732
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4759590792838877e-05,
+      "loss": 0.0299,
+      "step": 733
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4751918158567775e-05,
+      "loss": 1.0127,
+      "step": 734
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4744245524296676e-05,
+      "loss": 0.0304,
+      "step": 735
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4736572890025575e-05,
+      "loss": 1.4618,
+      "step": 736
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4728900255754476e-05,
+      "loss": 0.7167,
+      "step": 737
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4721227621483378e-05,
+      "loss": 0.0307,
+      "step": 738
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4713554987212276e-05,
+      "loss": 0.052,
+      "step": 739
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4705882352941174e-05,
+      "loss": 0.778,
+      "step": 740
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.469820971867008e-05,
+      "loss": 0.5482,
+      "step": 741
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4690537084398978e-05,
+      "loss": 1.4314,
+      "step": 742
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4682864450127876e-05,
+      "loss": 0.0422,
+      "step": 743
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.467519181585678e-05,
+      "loss": 0.6614,
+      "step": 744
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.466751918158568e-05,
+      "loss": 0.0312,
+      "step": 745
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4659846547314577e-05,
+      "loss": 0.043,
+      "step": 746
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.465217391304348e-05,
+      "loss": 0.0199,
+      "step": 747
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.464450127877238e-05,
+      "loss": 0.0832,
+      "step": 748
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.463682864450128e-05,
+      "loss": 0.0312,
+      "step": 749
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.462915601023018e-05,
+      "loss": 0.0309,
+      "step": 750
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.462148337595908e-05,
+      "loss": 0.0193,
+      "step": 751
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.461381074168798e-05,
+      "loss": 0.0446,
+      "step": 752
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.4606138107416882e-05,
+      "loss": 0.9993,
+      "step": 753
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.459846547314578e-05,
+      "loss": 0.016,
+      "step": 754
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4590792838874678e-05,
+      "loss": 0.0433,
+      "step": 755
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4583120204603583e-05,
+      "loss": 0.2614,
+      "step": 756
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.457544757033248e-05,
+      "loss": 0.018,
+      "step": 757
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.456777493606138e-05,
+      "loss": 0.6676,
+      "step": 758
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.456010230179028e-05,
+      "loss": 0.0599,
+      "step": 759
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4552429667519183e-05,
+      "loss": 0.0338,
+      "step": 760
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.454475703324808e-05,
+      "loss": 0.0268,
+      "step": 761
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4537084398976983e-05,
+      "loss": 0.0262,
+      "step": 762
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.452941176470588e-05,
+      "loss": 0.0252,
+      "step": 763
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4521739130434786e-05,
+      "loss": 0.0377,
+      "step": 764
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4514066496163684e-05,
+      "loss": 1.4866,
+      "step": 765
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4506393861892582e-05,
+      "loss": 1.0524,
+      "step": 766
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4498721227621484e-05,
+      "loss": 0.0328,
+      "step": 767
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4491048593350386e-05,
+      "loss": 0.024,
+      "step": 768
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4483375959079284e-05,
+      "loss": 0.022,
+      "step": 769
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4475703324808185e-05,
+      "loss": 0.0245,
+      "step": 770
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4468030690537087e-05,
+      "loss": 1.0107,
+      "step": 771
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4460358056265985e-05,
+      "loss": 0.0336,
+      "step": 772
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4452685421994887e-05,
+      "loss": 0.0391,
+      "step": 773
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4445012787723785e-05,
+      "loss": 0.0303,
+      "step": 774
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4437340153452687e-05,
+      "loss": 0.0443,
+      "step": 775
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4429667519181588e-05,
+      "loss": 0.0292,
+      "step": 776
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4421994884910487e-05,
+      "loss": 0.0284,
+      "step": 777
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4414322250639385e-05,
+      "loss": 1.5985,
+      "step": 778
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.440664961636829e-05,
+      "loss": 0.0114,
+      "step": 779
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4398976982097188e-05,
+      "loss": 0.0342,
+      "step": 780
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4391304347826086e-05,
+      "loss": 1.8501,
+      "step": 781
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4383631713554988e-05,
+      "loss": 0.5375,
+      "step": 782
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.437595907928389e-05,
+      "loss": 1.589,
+      "step": 783
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4368286445012788e-05,
+      "loss": 0.0221,
+      "step": 784
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.436061381074169e-05,
+      "loss": 0.0453,
+      "step": 785
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4352941176470587e-05,
+      "loss": 0.796,
+      "step": 786
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.434526854219949e-05,
+      "loss": 0.1249,
+      "step": 787
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.433759590792839e-05,
+      "loss": 0.7963,
+      "step": 788
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.432992327365729e-05,
+      "loss": 0.4231,
+      "step": 789
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.4322250639386187e-05,
+      "loss": 0.0247,
+      "step": 790
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.4314578005115092e-05,
+      "loss": 0.0295,
+      "step": 791
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.430690537084399e-05,
+      "loss": 0.0249,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.3537317216396332,
+      "eval_runtime": 58.1226,
+      "eval_samples_per_second": 6.813,
+      "eval_steps_per_second": 1.703,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.429923273657289e-05,
+      "loss": 0.0157,
+      "step": 793
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4291560102301794e-05,
+      "loss": 0.5222,
+      "step": 794
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4283887468030692e-05,
+      "loss": 0.0226,
+      "step": 795
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.427621483375959e-05,
+      "loss": 0.0293,
+      "step": 796
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.426854219948849e-05,
+      "loss": 0.015,
+      "step": 797
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4260869565217393e-05,
+      "loss": 1.0658,
+      "step": 798
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.425319693094629e-05,
+      "loss": 0.0412,
+      "step": 799
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4245524296675193e-05,
+      "loss": 0.0219,
+      "step": 800
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.423785166240409e-05,
+      "loss": 0.0149,
+      "step": 801
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4230179028132993e-05,
+      "loss": 0.0323,
+      "step": 802
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4222506393861895e-05,
+      "loss": 1.0764,
+      "step": 803
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4214833759590793e-05,
+      "loss": 0.7518,
+      "step": 804
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.420716112531969e-05,
+      "loss": 0.8166,
+      "step": 805
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4199488491048596e-05,
+      "loss": 0.0219,
+      "step": 806
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4191815856777494e-05,
+      "loss": 0.0154,
+      "step": 807
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4184143222506392e-05,
+      "loss": 0.0163,
+      "step": 808
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4176470588235294e-05,
+      "loss": 1.0562,
+      "step": 809
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4168797953964196e-05,
+      "loss": 0.0218,
+      "step": 810
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4161125319693094e-05,
+      "loss": 0.0232,
+      "step": 811
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4153452685421995e-05,
+      "loss": 0.2384,
+      "step": 812
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4145780051150894e-05,
+      "loss": 1.5286,
+      "step": 813
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4138107416879795e-05,
+      "loss": 0.0262,
+      "step": 814
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4130434782608697e-05,
+      "loss": 0.0334,
+      "step": 815
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4122762148337595e-05,
+      "loss": 0.6509,
+      "step": 816
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4115089514066497e-05,
+      "loss": 0.0159,
+      "step": 817
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.41074168797954e-05,
+      "loss": 0.6225,
+      "step": 818
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4099744245524297e-05,
+      "loss": 0.752,
+      "step": 819
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4092071611253198e-05,
+      "loss": 1.0594,
+      "step": 820
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.40843989769821e-05,
+      "loss": 0.0161,
+      "step": 821
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4076726342710998e-05,
+      "loss": 0.7492,
+      "step": 822
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40690537084399e-05,
+      "loss": 0.0302,
+      "step": 823
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4061381074168798e-05,
+      "loss": 0.038,
+      "step": 824
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40537084398977e-05,
+      "loss": 0.0243,
+      "step": 825
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40460358056266e-05,
+      "loss": 0.0331,
+      "step": 826
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40383631713555e-05,
+      "loss": 1.0252,
+      "step": 827
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4030690537084398e-05,
+      "loss": 0.0172,
+      "step": 828
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4023017902813303e-05,
+      "loss": 0.6168,
+      "step": 829
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40153452685422e-05,
+      "loss": 0.3135,
+      "step": 830
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40076726342711e-05,
+      "loss": 0.0168,
+      "step": 831
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.0172,
+      "step": 832
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.3992327365728902e-05,
+      "loss": 1.0349,
+      "step": 833
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39846547314578e-05,
+      "loss": 0.0978,
+      "step": 834
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3976982097186702e-05,
+      "loss": 0.0167,
+      "step": 835
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39693094629156e-05,
+      "loss": 1.0381,
+      "step": 836
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3961636828644502e-05,
+      "loss": 0.6857,
+      "step": 837
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3953964194373404e-05,
+      "loss": 1.0406,
+      "step": 838
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3946291560102302e-05,
+      "loss": 0.071,
+      "step": 839
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.39386189258312e-05,
+      "loss": 0.0193,
+      "step": 840
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3930946291560105e-05,
+      "loss": 0.0193,
+      "step": 841
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3923273657289003e-05,
+      "loss": 0.9916,
+      "step": 842
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.39156010230179e-05,
+      "loss": 0.0723,
+      "step": 843
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3907928388746806e-05,
+      "loss": 0.0626,
+      "step": 844
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3900255754475705e-05,
+      "loss": 0.7264,
+      "step": 845
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3892583120204603e-05,
+      "loss": 0.827,
+      "step": 846
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3884910485933504e-05,
+      "loss": 1.2193,
+      "step": 847
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3877237851662406e-05,
+      "loss": 0.6754,
+      "step": 848
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3869565217391304e-05,
+      "loss": 0.6856,
+      "step": 849
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3861892583120206e-05,
+      "loss": 0.7441,
+      "step": 850
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3854219948849104e-05,
+      "loss": 0.023,
+      "step": 851
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3846547314578006e-05,
+      "loss": 0.4742,
+      "step": 852
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3838874680306907e-05,
+      "loss": 0.0363,
+      "step": 853
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3831202046035806e-05,
+      "loss": 0.0409,
+      "step": 854
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3823529411764704e-05,
+      "loss": 0.8353,
+      "step": 855
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.381585677749361e-05,
+      "loss": 0.0333,
+      "step": 856
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3808184143222507e-05,
+      "loss": 0.0216,
+      "step": 857
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3800511508951405e-05,
+      "loss": 0.9688,
+      "step": 858
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3792838874680307e-05,
+      "loss": 1.0078,
+      "step": 859
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.378516624040921e-05,
+      "loss": 0.0183,
+      "step": 860
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3777493606138107e-05,
+      "loss": 0.0183,
+      "step": 861
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3769820971867008e-05,
+      "loss": 0.9786,
+      "step": 862
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3762148337595907e-05,
+      "loss": 0.9166,
+      "step": 863
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3754475703324808e-05,
+      "loss": 1.1408,
+      "step": 864
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.374680306905371e-05,
+      "loss": 1.1394,
+      "step": 865
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.3739130434782608e-05,
+      "loss": 0.0185,
+      "step": 866
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.373145780051151e-05,
+      "loss": 0.9979,
+      "step": 867
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.372378516624041e-05,
+      "loss": 1.0092,
+      "step": 868
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.371611253196931e-05,
+      "loss": 0.7276,
+      "step": 869
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.370843989769821e-05,
+      "loss": 0.0359,
+      "step": 870
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3700767263427113e-05,
+      "loss": 0.0864,
+      "step": 871
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.369309462915601e-05,
+      "loss": 0.0246,
+      "step": 872
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3685421994884912e-05,
+      "loss": 0.8518,
+      "step": 873
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.367774936061381e-05,
+      "loss": 1.433,
+      "step": 874
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3670076726342712e-05,
+      "loss": 0.035,
+      "step": 875
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3662404092071614e-05,
+      "loss": 0.5922,
+      "step": 876
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3654731457800512e-05,
+      "loss": 0.529,
+      "step": 877
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.364705882352941e-05,
+      "loss": 0.6423,
+      "step": 878
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3639386189258315e-05,
+      "loss": 0.0337,
+      "step": 879
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3631713554987214e-05,
+      "loss": 0.0402,
+      "step": 880
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3624040920716112e-05,
+      "loss": 0.0782,
+      "step": 881
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3616368286445013e-05,
+      "loss": 0.4141,
+      "step": 882
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3608695652173915e-05,
+      "loss": 0.0359,
+      "step": 883
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3601023017902813e-05,
+      "loss": 0.4509,
+      "step": 884
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3593350383631715e-05,
+      "loss": 0.6414,
+      "step": 885
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3585677749360613e-05,
+      "loss": 0.6873,
+      "step": 886
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3578005115089515e-05,
+      "loss": 0.0304,
+      "step": 887
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3570332480818416e-05,
+      "loss": 0.0315,
+      "step": 888
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3562659846547315e-05,
+      "loss": 0.0896,
+      "step": 889
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3554987212276213e-05,
+      "loss": 0.0201,
+      "step": 890
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3547314578005118e-05,
+      "loss": 0.027,
+      "step": 891
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3539641943734016e-05,
+      "loss": 1.9853,
+      "step": 892
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3531969309462914e-05,
+      "loss": 2.9647,
+      "step": 893
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.352429667519182e-05,
+      "loss": 1.9403,
+      "step": 894
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3516624040920717e-05,
+      "loss": 0.0193,
+      "step": 895
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3508951406649616e-05,
+      "loss": 0.9912,
+      "step": 896
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3501278772378517e-05,
+      "loss": 0.02,
+      "step": 897
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.349360613810742e-05,
+      "loss": 0.9015,
+      "step": 898
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3485933503836317e-05,
+      "loss": 0.1294,
+      "step": 899
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.347826086956522e-05,
+      "loss": 0.0559,
+      "step": 900
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.3470588235294117e-05,
+      "loss": 1.8039,
+      "step": 901
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.346291560102302e-05,
+      "loss": 0.0336,
+      "step": 902
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.345524296675192e-05,
+      "loss": 0.0383,
+      "step": 903
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.344757033248082e-05,
+      "loss": 0.0722,
+      "step": 904
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3439897698209717e-05,
+      "loss": 0.0278,
+      "step": 905
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.343222506393862e-05,
+      "loss": 0.0684,
+      "step": 906
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.342455242966752e-05,
+      "loss": 0.0236,
+      "step": 907
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3416879795396418e-05,
+      "loss": 0.0348,
+      "step": 908
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340920716112532e-05,
+      "loss": 0.0204,
+      "step": 909
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340153452685422e-05,
+      "loss": 0.8109,
+      "step": 910
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.339386189258312e-05,
+      "loss": 0.0279,
+      "step": 911
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.338618925831202e-05,
+      "loss": 0.0361,
+      "step": 912
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337851662404092e-05,
+      "loss": 0.0355,
+      "step": 913
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337084398976982e-05,
+      "loss": 0.7958,
+      "step": 914
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.3363171355498723e-05,
+      "loss": 0.0362,
+      "step": 915
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.335549872122762e-05,
+      "loss": 0.0252,
+      "step": 916
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3347826086956522e-05,
+      "loss": 0.8165,
+      "step": 917
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3340153452685424e-05,
+      "loss": 0.0237,
+      "step": 918
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3332480818414322e-05,
+      "loss": 0.8162,
+      "step": 919
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.332480818414322e-05,
+      "loss": 0.7786,
+      "step": 920
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3317135549872125e-05,
+      "loss": 0.0787,
+      "step": 921
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3309462915601024e-05,
+      "loss": 0.1764,
+      "step": 922
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3301790281329925e-05,
+      "loss": 0.7449,
+      "step": 923
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3294117647058824e-05,
+      "loss": 0.0452,
+      "step": 924
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3286445012787725e-05,
+      "loss": 0.0181,
+      "step": 925
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3278772378516627e-05,
+      "loss": 0.1783,
+      "step": 926
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3271099744245525e-05,
+      "loss": 0.5158,
+      "step": 927
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3263427109974423e-05,
+      "loss": 0.7574,
+      "step": 928
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3255754475703328e-05,
+      "loss": 2.6099,
+      "step": 929
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3248081841432226e-05,
+      "loss": 0.0304,
+      "step": 930
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3240409207161125e-05,
+      "loss": 0.0795,
+      "step": 931
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3232736572890026e-05,
+      "loss": 0.0342,
+      "step": 932
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3225063938618928e-05,
+      "loss": 0.0473,
+      "step": 933
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3217391304347826e-05,
+      "loss": 0.0831,
+      "step": 934
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3209718670076728e-05,
+      "loss": 0.0465,
+      "step": 935
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3202046035805626e-05,
+      "loss": 0.6769,
+      "step": 936
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3194373401534528e-05,
+      "loss": 0.0173,
+      "step": 937
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.318670076726343e-05,
+      "loss": 0.0169,
+      "step": 938
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3179028132992327e-05,
+      "loss": 0.0469,
+      "step": 939
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3171355498721226e-05,
+      "loss": 0.048,
+      "step": 940
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.316368286445013e-05,
+      "loss": 1.0704,
+      "step": 941
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.315601023017903e-05,
+      "loss": 1.0738,
+      "step": 942
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3148337595907927e-05,
+      "loss": 0.0163,
+      "step": 943
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3140664961636832e-05,
+      "loss": 2.0702,
+      "step": 944
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.313299232736573e-05,
+      "loss": 1.0433,
+      "step": 945
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.312531969309463e-05,
+      "loss": 1.0351,
+      "step": 946
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.311764705882353e-05,
+      "loss": 1.053,
+      "step": 947
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.3109974424552432e-05,
+      "loss": 0.1797,
+      "step": 948
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.310230179028133e-05,
+      "loss": 1.0259,
+      "step": 949
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.309462915601023e-05,
+      "loss": 0.03,
+      "step": 950
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.308695652173913e-05,
+      "loss": 0.905,
+      "step": 951
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.307928388746803e-05,
+      "loss": 0.0699,
+      "step": 952
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3071611253196933e-05,
+      "loss": 1.0013,
+      "step": 953
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.306393861892583e-05,
+      "loss": 0.0319,
+      "step": 954
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.305626598465473e-05,
+      "loss": 0.0361,
+      "step": 955
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3048593350383634e-05,
+      "loss": 0.0201,
+      "step": 956
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3040920716112533e-05,
+      "loss": 0.033,
+      "step": 957
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.303324808184143e-05,
+      "loss": 0.0411,
+      "step": 958
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3025575447570332e-05,
+      "loss": 0.7138,
+      "step": 959
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3017902813299234e-05,
+      "loss": 0.0434,
+      "step": 960
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3010230179028132e-05,
+      "loss": 0.8445,
+      "step": 961
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3002557544757034e-05,
+      "loss": 0.0323,
+      "step": 962
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2994884910485932e-05,
+      "loss": 0.0203,
+      "step": 963
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2987212276214834e-05,
+      "loss": 0.2302,
+      "step": 964
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2979539641943735e-05,
+      "loss": 0.0293,
+      "step": 965
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2971867007672634e-05,
+      "loss": 0.0593,
+      "step": 966
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2964194373401535e-05,
+      "loss": 0.6505,
+      "step": 967
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2956521739130437e-05,
+      "loss": 0.5891,
+      "step": 968
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2948849104859335e-05,
+      "loss": 0.0191,
+      "step": 969
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2941176470588233e-05,
+      "loss": 0.0188,
+      "step": 970
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2933503836317138e-05,
+      "loss": 0.9537,
+      "step": 971
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2925831202046037e-05,
+      "loss": 0.0186,
+      "step": 972
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2918158567774935e-05,
+      "loss": 1.0237,
+      "step": 973
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2910485933503836e-05,
+      "loss": 0.0417,
+      "step": 974
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2902813299232738e-05,
+      "loss": 0.0369,
+      "step": 975
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.289514066496164e-05,
+      "loss": 0.0184,
+      "step": 976
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2887468030690538e-05,
+      "loss": 0.811,
+      "step": 977
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2879795396419436e-05,
+      "loss": 0.0183,
+      "step": 978
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.287212276214834e-05,
+      "loss": 1.0442,
+      "step": 979
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.286445012787724e-05,
+      "loss": 0.027,
+      "step": 980
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.2856777493606137e-05,
+      "loss": 0.062,
+      "step": 981
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284910485933504e-05,
+      "loss": 0.0186,
+      "step": 982
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284143222506394e-05,
+      "loss": 0.6057,
+      "step": 983
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.283375959079284e-05,
+      "loss": 1.6594,
+      "step": 984
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.282608695652174e-05,
+      "loss": 0.0192,
+      "step": 985
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281841432225064e-05,
+      "loss": 1.5361,
+      "step": 986
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281074168797954e-05,
+      "loss": 0.0384,
+      "step": 987
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.2803069053708442e-05,
+      "loss": 2.0238,
+      "step": 988
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.279539641943734e-05,
+      "loss": 0.0334,
+      "step": 989
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2787723785166242e-05,
+      "loss": 0.0571,
+      "step": 990
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2780051150895143e-05,
+      "loss": 0.0211,
+      "step": 991
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.277237851662404e-05,
+      "loss": 0.0409,
+      "step": 992
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.276470588235294e-05,
+      "loss": 0.1059,
+      "step": 993
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2757033248081845e-05,
+      "loss": 0.0277,
+      "step": 994
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2749360613810743e-05,
+      "loss": 1.0023,
+      "step": 995
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.274168797953964e-05,
+      "loss": 0.0186,
+      "step": 996
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2734015345268543e-05,
+      "loss": 0.0345,
+      "step": 997
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2726342710997445e-05,
+      "loss": 0.0169,
+      "step": 998
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2718670076726343e-05,
+      "loss": 0.7292,
+      "step": 999
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2710997442455244e-05,
+      "loss": 1.2655,
+      "step": 1000
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2703324808184143e-05,
+      "loss": 0.03,
+      "step": 1001
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2695652173913044e-05,
+      "loss": 1.0266,
+      "step": 1002
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2687979539641946e-05,
+      "loss": 0.7645,
+      "step": 1003
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2680306905370844e-05,
+      "loss": 0.7985,
+      "step": 1004
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2672634271099742e-05,
+      "loss": 0.76,
+      "step": 1005
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2664961636828647e-05,
+      "loss": 0.7577,
+      "step": 1006
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2657289002557545e-05,
+      "loss": 0.0248,
+      "step": 1007
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2649616368286444e-05,
+      "loss": 0.0262,
+      "step": 1008
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2641943734015345e-05,
+      "loss": 1.9984,
+      "step": 1009
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2634271099744247e-05,
+      "loss": 0.1489,
+      "step": 1010
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2626598465473145e-05,
+      "loss": 0.0683,
+      "step": 1011
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2618925831202047e-05,
+      "loss": 0.054,
+      "step": 1012
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2611253196930945e-05,
+      "loss": 0.0269,
+      "step": 1013
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2603580562659847e-05,
+      "loss": 1.0378,
+      "step": 1014
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2595907928388748e-05,
+      "loss": 0.0171,
+      "step": 1015
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2588235294117646e-05,
+      "loss": 0.8433,
+      "step": 1016
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2580562659846548e-05,
+      "loss": 0.0468,
+      "step": 1017
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.257289002557545e-05,
+      "loss": 0.0331,
+      "step": 1018
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2565217391304348e-05,
+      "loss": 0.0308,
+      "step": 1019
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2557544757033246e-05,
+      "loss": 1.0007,
+      "step": 1020
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254987212276215e-05,
+      "loss": 2.0242,
+      "step": 1021
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254219948849105e-05,
+      "loss": 0.7713,
+      "step": 1022
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2534526854219948e-05,
+      "loss": 1.1112,
+      "step": 1023
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.252685421994885e-05,
+      "loss": 0.0184,
+      "step": 1024
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251918158567775e-05,
+      "loss": 0.0243,
+      "step": 1025
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251150895140665e-05,
+      "loss": 0.046,
+      "step": 1026
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.250383631713555e-05,
+      "loss": 0.0378,
+      "step": 1027
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.249616368286445e-05,
+      "loss": 0.0329,
+      "step": 1028
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2488491048593354e-05,
+      "loss": 0.0245,
+      "step": 1029
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2480818414322252e-05,
+      "loss": 0.0238,
+      "step": 1030
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.247314578005115e-05,
+      "loss": 1.0628,
+      "step": 1031
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2465473145780052e-05,
+      "loss": 0.0703,
+      "step": 1032
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2457800511508954e-05,
+      "loss": 0.0251,
+      "step": 1033
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2450127877237852e-05,
+      "loss": 1.0561,
+      "step": 1034
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2442455242966753e-05,
+      "loss": 0.0274,
+      "step": 1035
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.243478260869565e-05,
+      "loss": 0.023,
+      "step": 1036
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2427109974424553e-05,
+      "loss": 0.0223,
+      "step": 1037
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2419437340153455e-05,
+      "loss": 0.8113,
+      "step": 1038
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2411764705882353e-05,
+      "loss": 0.0346,
+      "step": 1039
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2404092071611255e-05,
+      "loss": 0.0164,
+      "step": 1040
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2396419437340156e-05,
+      "loss": 0.0237,
+      "step": 1041
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2388746803069054e-05,
+      "loss": 0.0305,
+      "step": 1042
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2381074168797953e-05,
+      "loss": 0.0158,
+      "step": 1043
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2373401534526858e-05,
+      "loss": 0.0225,
+      "step": 1044
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2365728900255756e-05,
+      "loss": 0.0226,
+      "step": 1045
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2358056265984654e-05,
+      "loss": 0.015,
+      "step": 1046
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2350383631713556e-05,
+      "loss": 0.0219,
+      "step": 1047
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2342710997442457e-05,
+      "loss": 1.0694,
+      "step": 1048
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2335038363171356e-05,
+      "loss": 0.0348,
+      "step": 1049
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2327365728900257e-05,
+      "loss": 0.0213,
+      "step": 1050
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2319693094629155e-05,
+      "loss": 0.9405,
+      "step": 1051
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2312020460358057e-05,
+      "loss": 0.0252,
+      "step": 1052
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.230434782608696e-05,
+      "loss": 0.0254,
+      "step": 1053
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2296675191815857e-05,
+      "loss": 0.7353,
+      "step": 1054
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2289002557544755e-05,
+      "loss": 0.0253,
+      "step": 1055
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.228132992327366e-05,
+      "loss": 0.8361,
+      "step": 1056
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2273657289002558e-05,
+      "loss": 0.0196,
+      "step": 1057
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2265984654731457e-05,
+      "loss": 0.014,
+      "step": 1058
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2258312020460358e-05,
+      "loss": 0.8297,
+      "step": 1059
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.225063938618926e-05,
+      "loss": 1.0654,
+      "step": 1060
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2242966751918158e-05,
+      "loss": 0.0248,
+      "step": 1061
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.223529411764706e-05,
+      "loss": 0.0226,
+      "step": 1062
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2227621483375958e-05,
+      "loss": 0.0238,
+      "step": 1063
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221994884910486e-05,
+      "loss": 0.0212,
+      "step": 1064
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221227621483376e-05,
+      "loss": 0.0274,
+      "step": 1065
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.220460358056266e-05,
+      "loss": 1.0217,
+      "step": 1066
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.219693094629156e-05,
+      "loss": 0.0252,
+      "step": 1067
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2189258312020462e-05,
+      "loss": 0.0219,
+      "step": 1068
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.218158567774936e-05,
+      "loss": 0.0349,
+      "step": 1069
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.217391304347826e-05,
+      "loss": 0.0152,
+      "step": 1070
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2166240409207164e-05,
+      "loss": 0.021,
+      "step": 1071
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2158567774936062e-05,
+      "loss": 0.015,
+      "step": 1072
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.215089514066496e-05,
+      "loss": 0.0277,
+      "step": 1073
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2143222506393862e-05,
+      "loss": 0.1858,
+      "step": 1074
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2135549872122764e-05,
+      "loss": 0.0136,
+      "step": 1075
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2127877237851662e-05,
+      "loss": 0.0132,
+      "step": 1076
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2120204603580563e-05,
+      "loss": 0.0126,
+      "step": 1077
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.211253196930946e-05,
+      "loss": 0.5171,
+      "step": 1078
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2104859335038363e-05,
+      "loss": 0.0422,
+      "step": 1079
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2097186700767265e-05,
+      "loss": 0.0186,
+      "step": 1080
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2089514066496163e-05,
+      "loss": 0.8412,
+      "step": 1081
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2081841432225065e-05,
+      "loss": 0.0183,
+      "step": 1082
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2074168797953966e-05,
+      "loss": 1.0243,
+      "step": 1083
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2066496163682865e-05,
+      "loss": 0.0212,
+      "step": 1084
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2058823529411766e-05,
+      "loss": 0.2841,
+      "step": 1085
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2051150895140664e-05,
+      "loss": 0.0118,
+      "step": 1086
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2043478260869566e-05,
+      "loss": 0.0247,
+      "step": 1087
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2035805626598468e-05,
+      "loss": 1.5121,
+      "step": 1088
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2028132992327366e-05,
+      "loss": 0.0304,
+      "step": 1089
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2020460358056267e-05,
+      "loss": 0.0329,
+      "step": 1090
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.201278772378517e-05,
+      "loss": 1.1288,
+      "step": 1091
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.2005115089514067e-05,
+      "loss": 0.0145,
+      "step": 1092
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.1997442455242966e-05,
+      "loss": 0.0498,
+      "step": 1093
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.198976982097187e-05,
+      "loss": 0.0244,
+      "step": 1094
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.198209718670077e-05,
+      "loss": 0.0112,
+      "step": 1095
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.1974424552429667e-05,
+      "loss": 0.8642,
+      "step": 1096
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.196675191815857e-05,
+      "loss": 0.8822,
+      "step": 1097
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.195907928388747e-05,
+      "loss": 0.0278,
+      "step": 1098
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.195140664961637e-05,
+      "loss": 0.6954,
+      "step": 1099
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.194373401534527e-05,
+      "loss": 0.0179,
+      "step": 1100
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.1936061381074168e-05,
+      "loss": 0.5607,
+      "step": 1101
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.192838874680307e-05,
+      "loss": 0.0191,
+      "step": 1102
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.192071611253197e-05,
+      "loss": 1.9519,
+      "step": 1103
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.191304347826087e-05,
+      "loss": 0.8347,
+      "step": 1104
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1905370843989768e-05,
+      "loss": 0.0262,
+      "step": 1105
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1897698209718673e-05,
+      "loss": 0.0477,
+      "step": 1106
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.189002557544757e-05,
+      "loss": 0.0227,
+      "step": 1107
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.188235294117647e-05,
+      "loss": 0.0205,
+      "step": 1108
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.187468030690537e-05,
+      "loss": 0.0208,
+      "step": 1109
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.1867007672634273e-05,
+      "loss": 0.0116,
+      "step": 1110
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.185933503836317e-05,
+      "loss": 1.1546,
+      "step": 1111
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1851662404092072e-05,
+      "loss": 0.0114,
+      "step": 1112
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.184398976982097e-05,
+      "loss": 0.0113,
+      "step": 1113
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1836317135549872e-05,
+      "loss": 0.6881,
+      "step": 1114
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1828644501278774e-05,
+      "loss": 0.0294,
+      "step": 1115
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1820971867007672e-05,
+      "loss": 0.1569,
+      "step": 1116
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1813299232736574e-05,
+      "loss": 1.1362,
+      "step": 1117
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1805626598465475e-05,
+      "loss": 0.0196,
+      "step": 1118
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1797953964194374e-05,
+      "loss": 0.0205,
+      "step": 1119
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1790281329923272e-05,
+      "loss": 0.0198,
+      "step": 1120
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1782608695652177e-05,
+      "loss": 0.0113,
+      "step": 1121
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1774936061381075e-05,
+      "loss": 0.02,
+      "step": 1122
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1767263427109973e-05,
+      "loss": 0.0195,
+      "step": 1123
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1759590792838875e-05,
+      "loss": 0.8016,
+      "step": 1124
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1751918158567776e-05,
+      "loss": 0.0122,
+      "step": 1125
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1744245524296675e-05,
+      "loss": 1.1132,
+      "step": 1126
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1736572890025576e-05,
+      "loss": 0.011,
+      "step": 1127
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1728900255754474e-05,
+      "loss": 0.7705,
+      "step": 1128
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1721227621483376e-05,
+      "loss": 1.1467,
+      "step": 1129
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1713554987212278e-05,
+      "loss": 0.0176,
+      "step": 1130
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1705882352941176e-05,
+      "loss": 0.0293,
+      "step": 1131
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1698209718670074e-05,
+      "loss": 0.0208,
+      "step": 1132
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.169053708439898e-05,
+      "loss": 0.0199,
+      "step": 1133
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1682864450127877e-05,
+      "loss": 0.0431,
+      "step": 1134
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.167519181585678e-05,
+      "loss": 1.6066,
+      "step": 1135
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.1667519181585677e-05,
+      "loss": 0.029,
+      "step": 1136
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165984654731458e-05,
+      "loss": 0.029,
+      "step": 1137
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165217391304348e-05,
+      "loss": 1.1749,
+      "step": 1138
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.164450127877238e-05,
+      "loss": 0.0189,
+      "step": 1139
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.163682864450128e-05,
+      "loss": 0.0309,
+      "step": 1140
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.1629156010230182e-05,
+      "loss": 0.0287,
+      "step": 1141
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.162148337595908e-05,
+      "loss": 0.0202,
+      "step": 1142
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.161381074168798e-05,
+      "loss": 0.7889,
+      "step": 1143
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.1606138107416883e-05,
+      "loss": 0.0285,
+      "step": 1144
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159846547314578e-05,
+      "loss": 0.0535,
+      "step": 1145
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159079283887468e-05,
+      "loss": 1.0829,
+      "step": 1146
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.158312020460358e-05,
+      "loss": 0.0989,
+      "step": 1147
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1575447570332483e-05,
+      "loss": 0.0115,
+      "step": 1148
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.156777493606138e-05,
+      "loss": 0.0227,
+      "step": 1149
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1560102301790283e-05,
+      "loss": 0.5506,
+      "step": 1150
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.155242966751918e-05,
+      "loss": 1.1234,
+      "step": 1151
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1544757033248083e-05,
+      "loss": 0.0564,
+      "step": 1152
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1537084398976984e-05,
+      "loss": 0.0127,
+      "step": 1153
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1529411764705882e-05,
+      "loss": 0.0218,
+      "step": 1154
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.152173913043478e-05,
+      "loss": 0.0124,
+      "step": 1155
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1514066496163686e-05,
+      "loss": 0.0223,
+      "step": 1156
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1506393861892584e-05,
+      "loss": 0.0505,
+      "step": 1157
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1498721227621482e-05,
+      "loss": 0.0225,
+      "step": 1158
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1491048593350384e-05,
+      "loss": 0.0308,
+      "step": 1159
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1483375959079285e-05,
+      "loss": 0.036,
+      "step": 1160
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1475703324808184e-05,
+      "loss": 0.0121,
+      "step": 1161
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1468030690537085e-05,
+      "loss": 0.0274,
+      "step": 1162
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1460358056265983e-05,
+      "loss": 1.0397,
+      "step": 1163
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1452685421994885e-05,
+      "loss": 0.0332,
+      "step": 1164
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1445012787723787e-05,
+      "loss": 0.0182,
+      "step": 1165
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1437340153452685e-05,
+      "loss": 0.018,
+      "step": 1166
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1429667519181587e-05,
+      "loss": 0.859,
+      "step": 1167
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1421994884910488e-05,
+      "loss": 0.0321,
+      "step": 1168
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1414322250639386e-05,
+      "loss": 0.0225,
+      "step": 1169
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1406649616368285e-05,
+      "loss": 0.0224,
+      "step": 1170
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.139897698209719e-05,
+      "loss": 0.03,
+      "step": 1171
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1391304347826088e-05,
+      "loss": 0.7473,
+      "step": 1172
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1383631713554986e-05,
+      "loss": 0.8609,
+      "step": 1173
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1375959079283888e-05,
+      "loss": 0.0105,
+      "step": 1174
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.136828644501279e-05,
+      "loss": 0.0217,
+      "step": 1175
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1360613810741687e-05,
+      "loss": 0.0218,
+      "step": 1176
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.135294117647059e-05,
+      "loss": 0.016,
+      "step": 1177
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1345268542199487e-05,
+      "loss": 0.7844,
+      "step": 1178
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.133759590792839e-05,
+      "loss": 0.01,
+      "step": 1179
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132992327365729e-05,
+      "loss": 0.8774,
+      "step": 1180
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132225063938619e-05,
+      "loss": 0.8375,
+      "step": 1181
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.1314578005115087e-05,
+      "loss": 0.0188,
+      "step": 1182
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.1306905370843992e-05,
+      "loss": 1.1026,
+      "step": 1183
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129923273657289e-05,
+      "loss": 1.0447,
+      "step": 1184
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129156010230179e-05,
+      "loss": 0.0101,
+      "step": 1185
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.128388746803069e-05,
+      "loss": 0.0247,
+      "step": 1186
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.127621483375959e-05,
+      "loss": 0.0295,
+      "step": 1187
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.1268542199488493e-05,
+      "loss": 0.0191,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.37390685081481934,
+      "eval_runtime": 57.4087,
+      "eval_samples_per_second": 6.898,
+      "eval_steps_per_second": 1.724,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.126086956521739e-05,
+      "loss": 0.0103,
+      "step": 1189
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1253196930946293e-05,
+      "loss": 0.0231,
+      "step": 1190
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1245524296675195e-05,
+      "loss": 0.0389,
+      "step": 1191
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1237851662404093e-05,
+      "loss": 0.0298,
+      "step": 1192
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.123017902813299e-05,
+      "loss": 0.0196,
+      "step": 1193
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1222506393861896e-05,
+      "loss": 0.9688,
+      "step": 1194
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1214833759590794e-05,
+      "loss": 0.0458,
+      "step": 1195
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1207161125319693e-05,
+      "loss": 0.0161,
+      "step": 1196
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1199488491048594e-05,
+      "loss": 0.0157,
+      "step": 1197
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1191815856777496e-05,
+      "loss": 0.8732,
+      "step": 1198
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1184143222506394e-05,
+      "loss": 0.0103,
+      "step": 1199
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1176470588235296e-05,
+      "loss": 0.0267,
+      "step": 1200
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1168797953964194e-05,
+      "loss": 0.0266,
+      "step": 1201
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1161125319693095e-05,
+      "loss": 0.0157,
+      "step": 1202
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1153452685421997e-05,
+      "loss": 0.265,
+      "step": 1203
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1145780051150895e-05,
+      "loss": 0.0537,
+      "step": 1204
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1138107416879794e-05,
+      "loss": 0.0306,
+      "step": 1205
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.11304347826087e-05,
+      "loss": 1.1714,
+      "step": 1206
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1122762148337597e-05,
+      "loss": 1.1704,
+      "step": 1207
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1115089514066495e-05,
+      "loss": 0.0154,
+      "step": 1208
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1107416879795397e-05,
+      "loss": 0.0195,
+      "step": 1209
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1099744245524298e-05,
+      "loss": 0.9099,
+      "step": 1210
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1092071611253196e-05,
+      "loss": 0.0112,
+      "step": 1211
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1084398976982098e-05,
+      "loss": 0.1506,
+      "step": 1212
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1076726342710996e-05,
+      "loss": 0.0111,
+      "step": 1213
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1069053708439898e-05,
+      "loss": 0.0244,
+      "step": 1214
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10613810741688e-05,
+      "loss": 0.8889,
+      "step": 1215
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1053708439897698e-05,
+      "loss": 0.0146,
+      "step": 1216
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10460358056266e-05,
+      "loss": 0.0189,
+      "step": 1217
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10383631713555e-05,
+      "loss": 0.0097,
+      "step": 1218
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10306905370844e-05,
+      "loss": 1.1683,
+      "step": 1219
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1023017902813297e-05,
+      "loss": 1.1621,
+      "step": 1220
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1015345268542202e-05,
+      "loss": 0.0194,
+      "step": 1221
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.10076726342711e-05,
+      "loss": 0.0155,
+      "step": 1222
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.0296,
+      "step": 1223
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.09923273657289e-05,
+      "loss": 0.02,
+      "step": 1224
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.0984654731457802e-05,
+      "loss": 0.0154,
+      "step": 1225
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09769820971867e-05,
+      "loss": 0.0151,
+      "step": 1226
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0969309462915602e-05,
+      "loss": 0.0223,
+      "step": 1227
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09616368286445e-05,
+      "loss": 1.7159,
+      "step": 1228
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0953964194373402e-05,
+      "loss": 0.0194,
+      "step": 1229
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0946291560102303e-05,
+      "loss": 0.0637,
+      "step": 1230
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09386189258312e-05,
+      "loss": 0.0156,
+      "step": 1231
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09309462915601e-05,
+      "loss": 1.0612,
+      "step": 1232
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0923273657289005e-05,
+      "loss": 0.8929,
+      "step": 1233
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0915601023017903e-05,
+      "loss": 1.1649,
+      "step": 1234
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.09079283887468e-05,
+      "loss": 2.059,
+      "step": 1235
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0900255754475703e-05,
+      "loss": 0.0169,
+      "step": 1236
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0892583120204604e-05,
+      "loss": 0.0158,
+      "step": 1237
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0884910485933503e-05,
+      "loss": 0.0108,
+      "step": 1238
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0877237851662404e-05,
+      "loss": 0.0152,
+      "step": 1239
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0869565217391306e-05,
+      "loss": 0.0109,
+      "step": 1240
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0861892583120208e-05,
+      "loss": 1.1352,
+      "step": 1241
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0854219948849106e-05,
+      "loss": 0.0226,
+      "step": 1242
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0846547314578004e-05,
+      "loss": 0.0154,
+      "step": 1243
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.083887468030691e-05,
+      "loss": 0.0232,
+      "step": 1244
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0831202046035807e-05,
+      "loss": 0.0142,
+      "step": 1245
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0823529411764705e-05,
+      "loss": 0.9143,
+      "step": 1246
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0815856777493607e-05,
+      "loss": 0.0165,
+      "step": 1247
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.080818414322251e-05,
+      "loss": 0.0446,
+      "step": 1248
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0800511508951407e-05,
+      "loss": 0.0121,
+      "step": 1249
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.079283887468031e-05,
+      "loss": 1.0991,
+      "step": 1250
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.0785166240409207e-05,
+      "loss": 0.0159,
+      "step": 1251
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.077749360613811e-05,
+      "loss": 0.0166,
+      "step": 1252
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.076982097186701e-05,
+      "loss": 0.0171,
+      "step": 1253
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0762148337595908e-05,
+      "loss": 0.7986,
+      "step": 1254
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0754475703324806e-05,
+      "loss": 0.0174,
+      "step": 1255
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.074680306905371e-05,
+      "loss": 1.0431,
+      "step": 1256
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.073913043478261e-05,
+      "loss": 0.1766,
+      "step": 1257
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.0731457800511508e-05,
+      "loss": 0.0209,
+      "step": 1258
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.072378516624041e-05,
+      "loss": 0.0178,
+      "step": 1259
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.071611253196931e-05,
+      "loss": 0.0215,
+      "step": 1260
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.070843989769821e-05,
+      "loss": 0.0112,
+      "step": 1261
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.070076726342711e-05,
+      "loss": 0.0166,
+      "step": 1262
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0693094629156012e-05,
+      "loss": 0.0161,
+      "step": 1263
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.068542199488491e-05,
+      "loss": 0.8922,
+      "step": 1264
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0677749360613812e-05,
+      "loss": 0.0133,
+      "step": 1265
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.067007672634271e-05,
+      "loss": 0.0117,
+      "step": 1266
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0662404092071612e-05,
+      "loss": 0.0153,
+      "step": 1267
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0654731457800514e-05,
+      "loss": 0.0153,
+      "step": 1268
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0647058823529412e-05,
+      "loss": 0.0195,
+      "step": 1269
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.063938618925831e-05,
+      "loss": 1.1511,
+      "step": 1270
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0631713554987215e-05,
+      "loss": 0.0108,
+      "step": 1271
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0624040920716113e-05,
+      "loss": 0.8897,
+      "step": 1272
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.061636828644501e-05,
+      "loss": 0.0157,
+      "step": 1273
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0608695652173913e-05,
+      "loss": 0.0108,
+      "step": 1274
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0601023017902815e-05,
+      "loss": 0.0214,
+      "step": 1275
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0593350383631713e-05,
+      "loss": 0.0186,
+      "step": 1276
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0585677749360615e-05,
+      "loss": 0.0182,
+      "step": 1277
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0578005115089513e-05,
+      "loss": 0.0102,
+      "step": 1278
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0570332480818415e-05,
+      "loss": 1.1313,
+      "step": 1279
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0562659846547316e-05,
+      "loss": 0.0103,
+      "step": 1280
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0554987212276214e-05,
+      "loss": 0.0144,
+      "step": 1281
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0547314578005113e-05,
+      "loss": 0.02,
+      "step": 1282
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0539641943734018e-05,
+      "loss": 0.0292,
+      "step": 1283
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0531969309462916e-05,
+      "loss": 0.0105,
+      "step": 1284
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0524296675191814e-05,
+      "loss": 0.01,
+      "step": 1285
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0516624040920716e-05,
+      "loss": 1.1245,
+      "step": 1286
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0508951406649617e-05,
+      "loss": 0.8544,
+      "step": 1287
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0501278772378516e-05,
+      "loss": 0.0149,
+      "step": 1288
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0493606138107417e-05,
+      "loss": 0.0065,
+      "step": 1289
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.048593350383632e-05,
+      "loss": 0.0113,
+      "step": 1290
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0478260869565217e-05,
+      "loss": 0.0094,
+      "step": 1291
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.047058823529412e-05,
+      "loss": 0.0097,
+      "step": 1292
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0462915601023017e-05,
+      "loss": 0.0812,
+      "step": 1293
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0455242966751922e-05,
+      "loss": 0.0099,
+      "step": 1294
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.044757033248082e-05,
+      "loss": 0.013,
+      "step": 1295
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0439897698209718e-05,
+      "loss": 1.584,
+      "step": 1296
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.043222506393862e-05,
+      "loss": 0.0089,
+      "step": 1297
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.042455242966752e-05,
+      "loss": 0.0089,
+      "step": 1298
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.041687979539642e-05,
+      "loss": 0.0093,
+      "step": 1299
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.040920716112532e-05,
+      "loss": 0.0149,
+      "step": 1300
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.040153452685422e-05,
+      "loss": 1.1671,
+      "step": 1301
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.039386189258312e-05,
+      "loss": 0.0086,
+      "step": 1302
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.0386189258312023e-05,
+      "loss": 0.8177,
+      "step": 1303
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.037851662404092e-05,
+      "loss": 0.0128,
+      "step": 1304
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.037084398976982e-05,
+      "loss": 0.0167,
+      "step": 1305
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0363171355498724e-05,
+      "loss": 0.0085,
+      "step": 1306
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0355498721227622e-05,
+      "loss": 0.0085,
+      "step": 1307
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.034782608695652e-05,
+      "loss": 0.0166,
+      "step": 1308
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0340153452685422e-05,
+      "loss": 0.9448,
+      "step": 1309
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0332480818414324e-05,
+      "loss": 0.0082,
+      "step": 1310
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0324808184143222e-05,
+      "loss": 0.0088,
+      "step": 1311
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0317135549872124e-05,
+      "loss": 0.0082,
+      "step": 1312
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0309462915601025e-05,
+      "loss": 0.0146,
+      "step": 1313
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0301790281329924e-05,
+      "loss": 0.0123,
+      "step": 1314
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0294117647058825e-05,
+      "loss": 0.0101,
+      "step": 1315
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0286445012787723e-05,
+      "loss": 1.2167,
+      "step": 1316
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0278772378516625e-05,
+      "loss": 1.0578,
+      "step": 1317
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0271099744245527e-05,
+      "loss": 0.0163,
+      "step": 1318
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0263427109974425e-05,
+      "loss": 0.013,
+      "step": 1319
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0255754475703323e-05,
+      "loss": 0.0082,
+      "step": 1320
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0248081841432228e-05,
+      "loss": 0.0081,
+      "step": 1321
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0240409207161126e-05,
+      "loss": 0.0124,
+      "step": 1322
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0232736572890024e-05,
+      "loss": 0.0206,
+      "step": 1323
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0225063938618926e-05,
+      "loss": 0.0081,
+      "step": 1324
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0217391304347828e-05,
+      "loss": 0.0122,
+      "step": 1325
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0209718670076726e-05,
+      "loss": 0.0122,
+      "step": 1326
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0202046035805628e-05,
+      "loss": 0.0121,
+      "step": 1327
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0194373401534526e-05,
+      "loss": 1.224,
+      "step": 1328
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0186700767263427e-05,
+      "loss": 2.3502,
+      "step": 1329
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.017902813299233e-05,
+      "loss": 0.0242,
+      "step": 1330
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0171355498721227e-05,
+      "loss": 0.0064,
+      "step": 1331
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0163682864450125e-05,
+      "loss": 0.0186,
+      "step": 1332
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.015601023017903e-05,
+      "loss": 0.012,
+      "step": 1333
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.014833759590793e-05,
+      "loss": 0.0119,
+      "step": 1334
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.0140664961636827e-05,
+      "loss": 0.0159,
+      "step": 1335
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.013299232736573e-05,
+      "loss": 0.0199,
+      "step": 1336
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.012531969309463e-05,
+      "loss": 1.2138,
+      "step": 1337
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.011764705882353e-05,
+      "loss": 0.9491,
+      "step": 1338
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010997442455243e-05,
+      "loss": 0.0083,
+      "step": 1339
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010230179028133e-05,
+      "loss": 0.948,
+      "step": 1340
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.009462915601023e-05,
+      "loss": 0.0112,
+      "step": 1341
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.008695652173913e-05,
+      "loss": 0.8921,
+      "step": 1342
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007928388746803e-05,
+      "loss": 0.0194,
+      "step": 1343
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007161125319693e-05,
+      "loss": 1.0273,
+      "step": 1344
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0063938618925833e-05,
+      "loss": 0.013,
+      "step": 1345
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.005626598465473e-05,
+      "loss": 0.0103,
+      "step": 1346
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0048593350383633e-05,
+      "loss": 0.0114,
+      "step": 1347
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0040920716112534e-05,
+      "loss": 0.0199,
+      "step": 1348
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0033248081841433e-05,
+      "loss": 1.183,
+      "step": 1349
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0025575447570334e-05,
+      "loss": 0.0231,
+      "step": 1350
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0017902813299232e-05,
+      "loss": 1.1794,
+      "step": 1351
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0010230179028134e-05,
+      "loss": 0.0096,
+      "step": 1352
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 2.0002557544757036e-05,
+      "loss": 0.9049,
+      "step": 1353
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9994884910485934e-05,
+      "loss": 0.019,
+      "step": 1354
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9987212276214832e-05,
+      "loss": 0.0152,
+      "step": 1355
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9979539641943737e-05,
+      "loss": 0.8834,
+      "step": 1356
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9971867007672635e-05,
+      "loss": 0.01,
+      "step": 1357
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9964194373401533e-05,
+      "loss": 0.0152,
+      "step": 1358
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9956521739130435e-05,
+      "loss": 0.0161,
+      "step": 1359
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9948849104859337e-05,
+      "loss": 1.6778,
+      "step": 1360
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9941176470588235e-05,
+      "loss": 0.0214,
+      "step": 1361
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9933503836317137e-05,
+      "loss": 0.0156,
+      "step": 1362
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9925831202046038e-05,
+      "loss": 0.026,
+      "step": 1363
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9918158567774936e-05,
+      "loss": 0.0259,
+      "step": 1364
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9910485933503838e-05,
+      "loss": 0.0161,
+      "step": 1365
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9902813299232736e-05,
+      "loss": 0.0108,
+      "step": 1366
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9895140664961638e-05,
+      "loss": 0.0149,
+      "step": 1367
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.988746803069054e-05,
+      "loss": 0.0325,
+      "step": 1368
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9879795396419438e-05,
+      "loss": 0.0153,
+      "step": 1369
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9872122762148336e-05,
+      "loss": 1.7678,
+      "step": 1370
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.986445012787724e-05,
+      "loss": 0.0773,
+      "step": 1371
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.985677749360614e-05,
+      "loss": 0.0244,
+      "step": 1372
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.9849104859335037e-05,
+      "loss": 0.0225,
+      "step": 1373
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.984143222506394e-05,
+      "loss": 0.0254,
+      "step": 1374
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.983375959079284e-05,
+      "loss": 0.0205,
+      "step": 1375
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.982608695652174e-05,
+      "loss": 0.0096,
+      "step": 1376
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981841432225064e-05,
+      "loss": 0.0147,
+      "step": 1377
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981074168797954e-05,
+      "loss": 0.0296,
+      "step": 1378
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.980306905370844e-05,
+      "loss": 0.0227,
+      "step": 1379
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.9795396419437342e-05,
+      "loss": 0.899,
+      "step": 1380
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.978772378516624e-05,
+      "loss": 0.019,
+      "step": 1381
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9780051150895138e-05,
+      "loss": 0.0157,
+      "step": 1382
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9772378516624043e-05,
+      "loss": 1.1897,
+      "step": 1383
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.976470588235294e-05,
+      "loss": 0.014,
+      "step": 1384
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.975703324808184e-05,
+      "loss": 0.0208,
+      "step": 1385
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.974936061381074e-05,
+      "loss": 0.0093,
+      "step": 1386
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.9741687979539643e-05,
+      "loss": 0.0244,
+      "step": 1387
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.973401534526854e-05,
+      "loss": 0.028,
+      "step": 1388
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9726342710997443e-05,
+      "loss": 0.0093,
+      "step": 1389
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9718670076726344e-05,
+      "loss": 0.0184,
+      "step": 1390
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9710997442455243e-05,
+      "loss": 0.9462,
+      "step": 1391
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9703324808184144e-05,
+      "loss": 0.0138,
+      "step": 1392
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9695652173913042e-05,
+      "loss": 0.9561,
+      "step": 1393
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9687979539641944e-05,
+      "loss": 0.0152,
+      "step": 1394
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9680306905370846e-05,
+      "loss": 0.0178,
+      "step": 1395
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9672634271099744e-05,
+      "loss": 0.0166,
+      "step": 1396
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9664961636828642e-05,
+      "loss": 0.92,
+      "step": 1397
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9657289002557547e-05,
+      "loss": 0.0253,
+      "step": 1398
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9649616368286445e-05,
+      "loss": 0.0266,
+      "step": 1399
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9641943734015347e-05,
+      "loss": 0.0105,
+      "step": 1400
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9634271099744245e-05,
+      "loss": 0.0208,
+      "step": 1401
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9626598465473147e-05,
+      "loss": 0.8094,
+      "step": 1402
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.961892583120205e-05,
+      "loss": 0.0103,
+      "step": 1403
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9611253196930947e-05,
+      "loss": 0.0179,
+      "step": 1404
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9603580562659845e-05,
+      "loss": 0.0133,
+      "step": 1405
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.959590792838875e-05,
+      "loss": 0.0086,
+      "step": 1406
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9588235294117648e-05,
+      "loss": 0.0216,
+      "step": 1407
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9580562659846546e-05,
+      "loss": 0.0127,
+      "step": 1408
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9572890025575448e-05,
+      "loss": 0.0127,
+      "step": 1409
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.956521739130435e-05,
+      "loss": 0.0209,
+      "step": 1410
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9557544757033248e-05,
+      "loss": 0.0166,
+      "step": 1411
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954987212276215e-05,
+      "loss": 0.0123,
+      "step": 1412
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954219948849105e-05,
+      "loss": 0.0135,
+      "step": 1413
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.953452685421995e-05,
+      "loss": 0.1253,
+      "step": 1414
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.952685421994885e-05,
+      "loss": 0.0098,
+      "step": 1415
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951918158567775e-05,
+      "loss": 0.0091,
+      "step": 1416
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951150895140665e-05,
+      "loss": 0.9327,
+      "step": 1417
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.9503836317135552e-05,
+      "loss": 0.0119,
+      "step": 1418
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.949616368286445e-05,
+      "loss": 0.0153,
+      "step": 1419
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.948849104859335e-05,
+      "loss": 0.0123,
+      "step": 1420
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9480818414322254e-05,
+      "loss": 0.0194,
+      "step": 1421
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9473145780051152e-05,
+      "loss": 1.2311,
+      "step": 1422
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.946547314578005e-05,
+      "loss": 1.8519,
+      "step": 1423
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9457800511508952e-05,
+      "loss": 0.963,
+      "step": 1424
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9450127877237853e-05,
+      "loss": 0.0075,
+      "step": 1425
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.944245524296675e-05,
+      "loss": 0.0042,
+      "step": 1426
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9434782608695653e-05,
+      "loss": 0.0077,
+      "step": 1427
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.942710997442455e-05,
+      "loss": 0.0152,
+      "step": 1428
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9419437340153453e-05,
+      "loss": 0.0152,
+      "step": 1429
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9411764705882355e-05,
+      "loss": 0.019,
+      "step": 1430
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9404092071611253e-05,
+      "loss": 0.0081,
+      "step": 1431
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.939641943734015e-05,
+      "loss": 0.0076,
+      "step": 1432
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9388746803069056e-05,
+      "loss": 0.0086,
+      "step": 1433
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9381074168797954e-05,
+      "loss": 0.0146,
+      "step": 1434
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9373401534526853e-05,
+      "loss": 0.0074,
+      "step": 1435
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9365728900255754e-05,
+      "loss": 0.1007,
+      "step": 1436
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9358056265984656e-05,
+      "loss": 0.011,
+      "step": 1437
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9350383631713554e-05,
+      "loss": 0.0089,
+      "step": 1438
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9342710997442456e-05,
+      "loss": 0.0182,
+      "step": 1439
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9335038363171357e-05,
+      "loss": 0.0139,
+      "step": 1440
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9327365728900255e-05,
+      "loss": 2.2086,
+      "step": 1441
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9319693094629157e-05,
+      "loss": 0.0178,
+      "step": 1442
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9312020460358055e-05,
+      "loss": 0.0142,
+      "step": 1443
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9304347826086957e-05,
+      "loss": 0.0142,
+      "step": 1444
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.929667519181586e-05,
+      "loss": 0.0242,
+      "step": 1445
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9289002557544757e-05,
+      "loss": 0.0087,
+      "step": 1446
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9281329923273655e-05,
+      "loss": 0.0077,
+      "step": 1447
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.927365728900256e-05,
+      "loss": 0.0104,
+      "step": 1448
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9265984654731458e-05,
+      "loss": 0.0139,
+      "step": 1449
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9258312020460356e-05,
+      "loss": 1.2557,
+      "step": 1450
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9250639386189258e-05,
+      "loss": 0.0068,
+      "step": 1451
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.924296675191816e-05,
+      "loss": 0.0163,
+      "step": 1452
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.923529411764706e-05,
+      "loss": 0.9483,
+      "step": 1453
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.922762148337596e-05,
+      "loss": 1.2181,
+      "step": 1454
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.9219948849104858e-05,
+      "loss": 0.011,
+      "step": 1455
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.9212276214833763e-05,
+      "loss": 0.014,
+      "step": 1456
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.920460358056266e-05,
+      "loss": 0.0131,
+      "step": 1457
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.919693094629156e-05,
+      "loss": 0.0135,
+      "step": 1458
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.918925831202046e-05,
+      "loss": 0.007,
+      "step": 1459
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9181585677749362e-05,
+      "loss": 0.0072,
+      "step": 1460
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.917391304347826e-05,
+      "loss": 0.0135,
+      "step": 1461
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9166240409207162e-05,
+      "loss": 0.9908,
+      "step": 1462
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9158567774936064e-05,
+      "loss": 0.0103,
+      "step": 1463
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9150895140664962e-05,
+      "loss": 0.9867,
+      "step": 1464
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9143222506393864e-05,
+      "loss": 0.0138,
+      "step": 1465
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9135549872122762e-05,
+      "loss": 0.0104,
+      "step": 1466
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9127877237851663e-05,
+      "loss": 0.014,
+      "step": 1467
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9120204603580565e-05,
+      "loss": 0.0071,
+      "step": 1468
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9112531969309463e-05,
+      "loss": 0.0108,
+      "step": 1469
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.910485933503836e-05,
+      "loss": 0.0106,
+      "step": 1470
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9097186700767266e-05,
+      "loss": 0.0404,
+      "step": 1471
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9089514066496165e-05,
+      "loss": 2.2011,
+      "step": 1472
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9081841432225063e-05,
+      "loss": 0.0087,
+      "step": 1473
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9074168797953965e-05,
+      "loss": 0.0071,
+      "step": 1474
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9066496163682866e-05,
+      "loss": 0.0072,
+      "step": 1475
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9058823529411764e-05,
+      "loss": 0.0108,
+      "step": 1476
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9051150895140666e-05,
+      "loss": 0.0108,
+      "step": 1477
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9043478260869564e-05,
+      "loss": 0.0072,
+      "step": 1478
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9035805626598466e-05,
+      "loss": 0.0108,
+      "step": 1479
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9028132992327367e-05,
+      "loss": 0.0073,
+      "step": 1480
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9020460358056266e-05,
+      "loss": 0.9537,
+      "step": 1481
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9012787723785164e-05,
+      "loss": 0.0107,
+      "step": 1482
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.900511508951407e-05,
+      "loss": 0.0145,
+      "step": 1483
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8997442455242967e-05,
+      "loss": 0.007,
+      "step": 1484
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8989769820971865e-05,
+      "loss": 0.0071,
+      "step": 1485
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.898209718670077e-05,
+      "loss": 1.243,
+      "step": 1486
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.897442455242967e-05,
+      "loss": 0.011,
+      "step": 1487
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.8966751918158567e-05,
+      "loss": 0.0077,
+      "step": 1488
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895907928388747e-05,
+      "loss": 0.0073,
+      "step": 1489
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895140664961637e-05,
+      "loss": 1.2486,
+      "step": 1490
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8943734015345268e-05,
+      "loss": 0.0109,
+      "step": 1491
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.893606138107417e-05,
+      "loss": 0.0149,
+      "step": 1492
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8928388746803068e-05,
+      "loss": 0.0125,
+      "step": 1493
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.892071611253197e-05,
+      "loss": 0.0071,
+      "step": 1494
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.891304347826087e-05,
+      "loss": 0.0036,
+      "step": 1495
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.890537084398977e-05,
+      "loss": 0.0149,
+      "step": 1496
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8897698209718668e-05,
+      "loss": 0.015,
+      "step": 1497
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8890025575447573e-05,
+      "loss": 0.9536,
+      "step": 1498
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.888235294117647e-05,
+      "loss": 0.9997,
+      "step": 1499
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.887468030690537e-05,
+      "loss": 0.9621,
+      "step": 1500
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.886700767263427e-05,
+      "loss": 0.0113,
+      "step": 1501
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.8859335038363172e-05,
+      "loss": 0.2056,
+      "step": 1502
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.885166240409207e-05,
+      "loss": 0.0114,
+      "step": 1503
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8843989769820972e-05,
+      "loss": 0.016,
+      "step": 1504
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.883631713554987e-05,
+      "loss": 0.9175,
+      "step": 1505
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8828644501278775e-05,
+      "loss": 0.8959,
+      "step": 1506
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8820971867007674e-05,
+      "loss": 0.0074,
+      "step": 1507
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8813299232736572e-05,
+      "loss": 0.0117,
+      "step": 1508
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8805626598465474e-05,
+      "loss": 0.0075,
+      "step": 1509
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8797953964194375e-05,
+      "loss": 0.0074,
+      "step": 1510
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8790281329923273e-05,
+      "loss": 0.0577,
+      "step": 1511
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8782608695652175e-05,
+      "loss": 0.0128,
+      "step": 1512
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8774936061381077e-05,
+      "loss": 0.0073,
+      "step": 1513
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8767263427109975e-05,
+      "loss": 0.0074,
+      "step": 1514
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8759590792838876e-05,
+      "loss": 0.0073,
+      "step": 1515
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8751918158567775e-05,
+      "loss": 0.0088,
+      "step": 1516
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8744245524296676e-05,
+      "loss": 0.0078,
+      "step": 1517
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8736572890025578e-05,
+      "loss": 0.016,
+      "step": 1518
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8728900255754476e-05,
+      "loss": 1.3842,
+      "step": 1519
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8721227621483374e-05,
+      "loss": 1.2502,
+      "step": 1520
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.871355498721228e-05,
+      "loss": 0.0117,
+      "step": 1521
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8705882352941178e-05,
+      "loss": 0.0113,
+      "step": 1522
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8698209718670076e-05,
+      "loss": 0.0125,
+      "step": 1523
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8690537084398977e-05,
+      "loss": 0.0153,
+      "step": 1524
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.868286445012788e-05,
+      "loss": 0.0072,
+      "step": 1525
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8675191815856777e-05,
+      "loss": 0.0086,
+      "step": 1526
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.866751918158568e-05,
+      "loss": 0.0195,
+      "step": 1527
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.8659846547314577e-05,
+      "loss": 1.2511,
+      "step": 1528
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.865217391304348e-05,
+      "loss": 1.2407,
+      "step": 1529
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.864450127877238e-05,
+      "loss": 0.0112,
+      "step": 1530
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.863682864450128e-05,
+      "loss": 0.0073,
+      "step": 1531
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8629156010230177e-05,
+      "loss": 1.1911,
+      "step": 1532
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8621483375959082e-05,
+      "loss": 0.0168,
+      "step": 1533
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.861381074168798e-05,
+      "loss": 0.0128,
+      "step": 1534
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8606138107416878e-05,
+      "loss": 0.0116,
+      "step": 1535
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8598465473145783e-05,
+      "loss": 1.2286,
+      "step": 1536
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.859079283887468e-05,
+      "loss": 0.0158,
+      "step": 1537
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.858312020460358e-05,
+      "loss": 2.3735,
+      "step": 1538
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.857544757033248e-05,
+      "loss": 1.1972,
+      "step": 1539
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8567774936061383e-05,
+      "loss": 0.011,
+      "step": 1540
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.856010230179028e-05,
+      "loss": 0.0127,
+      "step": 1541
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8552429667519183e-05,
+      "loss": 0.0148,
+      "step": 1542
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.854475703324808e-05,
+      "loss": 0.0111,
+      "step": 1543
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8537084398976983e-05,
+      "loss": 0.0143,
+      "step": 1544
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8529411764705884e-05,
+      "loss": 0.0132,
+      "step": 1545
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8521739130434782e-05,
+      "loss": 0.0152,
+      "step": 1546
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.851406649616368e-05,
+      "loss": 0.0152,
+      "step": 1547
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8506393861892586e-05,
+      "loss": 0.0051,
+      "step": 1548
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8498721227621484e-05,
+      "loss": 0.9576,
+      "step": 1549
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8491048593350382e-05,
+      "loss": 0.0181,
+      "step": 1550
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8483375959079284e-05,
+      "loss": 0.011,
+      "step": 1551
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8475703324808185e-05,
+      "loss": 0.0205,
+      "step": 1552
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8468030690537083e-05,
+      "loss": 2.5631,
+      "step": 1553
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8460358056265985e-05,
+      "loss": 0.0102,
+      "step": 1554
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8452685421994883e-05,
+      "loss": 0.0147,
+      "step": 1555
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8445012787723788e-05,
+      "loss": 0.015,
+      "step": 1556
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8437340153452687e-05,
+      "loss": 0.0122,
+      "step": 1557
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8429667519181585e-05,
+      "loss": 0.0176,
+      "step": 1558
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8421994884910486e-05,
+      "loss": 0.0114,
+      "step": 1559
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8414322250639388e-05,
+      "loss": 0.0252,
+      "step": 1560
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8406649616368286e-05,
+      "loss": 0.013,
+      "step": 1561
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8398976982097188e-05,
+      "loss": 0.0142,
+      "step": 1562
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.839130434782609e-05,
+      "loss": 0.6496,
+      "step": 1563
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8383631713554988e-05,
+      "loss": 0.0149,
+      "step": 1564
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.837595907928389e-05,
+      "loss": 0.0104,
+      "step": 1565
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8368286445012787e-05,
+      "loss": 0.0129,
+      "step": 1566
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.836061381074169e-05,
+      "loss": 0.0197,
+      "step": 1567
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.835294117647059e-05,
+      "loss": 0.009,
+      "step": 1568
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.834526854219949e-05,
+      "loss": 0.0086,
+      "step": 1569
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.8337595907928387e-05,
+      "loss": 0.0087,
+      "step": 1570
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.8329923273657292e-05,
+      "loss": 0.0131,
+      "step": 1571
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.832225063938619e-05,
+      "loss": 1.2046,
+      "step": 1572
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.831457800511509e-05,
+      "loss": 0.0135,
+      "step": 1573
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.830690537084399e-05,
+      "loss": 0.0149,
+      "step": 1574
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.8299232736572892e-05,
+      "loss": 0.0077,
+      "step": 1575
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.829156010230179e-05,
+      "loss": 0.0143,
+      "step": 1576
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.828388746803069e-05,
+      "loss": 0.9806,
+      "step": 1577
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.827621483375959e-05,
+      "loss": 1.3783,
+      "step": 1578
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.826854219948849e-05,
+      "loss": 0.0075,
+      "step": 1579
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.8260869565217393e-05,
+      "loss": 0.0111,
+      "step": 1580
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.825319693094629e-05,
+      "loss": 0.0111,
+      "step": 1581
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.824552429667519e-05,
+      "loss": 0.0144,
+      "step": 1582
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8237851662404095e-05,
+      "loss": 0.0128,
+      "step": 1583
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8230179028132993e-05,
+      "loss": 1.2949,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.40147146582603455,
+      "eval_runtime": 58.0487,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.822250639386189e-05,
+      "loss": 0.0075,
+      "step": 1585
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8214833759590796e-05,
+      "loss": 0.0108,
+      "step": 1586
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8207161125319694e-05,
+      "loss": 0.011,
+      "step": 1587
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8199488491048592e-05,
+      "loss": 0.0148,
+      "step": 1588
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8191815856777494e-05,
+      "loss": 0.0077,
+      "step": 1589
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8184143222506396e-05,
+      "loss": 0.6674,
+      "step": 1590
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8176470588235294e-05,
+      "loss": 0.0109,
+      "step": 1591
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8168797953964195e-05,
+      "loss": 0.0108,
+      "step": 1592
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8161125319693094e-05,
+      "loss": 0.011,
+      "step": 1593
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8153452685421995e-05,
+      "loss": 0.0143,
+      "step": 1594
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8145780051150897e-05,
+      "loss": 0.9728,
+      "step": 1595
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8138107416879795e-05,
+      "loss": 0.0074,
+      "step": 1596
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8130434782608693e-05,
+      "loss": 0.0087,
+      "step": 1597
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.81227621483376e-05,
+      "loss": 0.0107,
+      "step": 1598
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8115089514066497e-05,
+      "loss": 0.0077,
+      "step": 1599
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8107416879795395e-05,
+      "loss": 0.0167,
+      "step": 1600
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8099744245524296e-05,
+      "loss": 1.219,
+      "step": 1601
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8092071611253198e-05,
+      "loss": 1.2244,
+      "step": 1602
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8084398976982096e-05,
+      "loss": 0.0108,
+      "step": 1603
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8076726342710998e-05,
+      "loss": 0.0071,
+      "step": 1604
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8069053708439896e-05,
+      "loss": 0.0109,
+      "step": 1605
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.8061381074168798e-05,
+      "loss": 0.0185,
+      "step": 1606
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.80537084398977e-05,
+      "loss": 1.2711,
+      "step": 1607
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.8046035805626598e-05,
+      "loss": 0.0113,
+      "step": 1608
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.80383631713555e-05,
+      "loss": 0.9575,
+      "step": 1609
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80306905370844e-05,
+      "loss": 1.1369,
+      "step": 1610
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80230179028133e-05,
+      "loss": 0.0114,
+      "step": 1611
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80153452685422e-05,
+      "loss": 1.578,
+      "step": 1612
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.8007672634271102e-05,
+      "loss": 0.0076,
+      "step": 1613
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.0128,
+      "step": 1614
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.7992327365728902e-05,
+      "loss": 0.0157,
+      "step": 1615
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.79846547314578e-05,
+      "loss": 0.0154,
+      "step": 1616
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.7976982097186702e-05,
+      "loss": 0.0125,
+      "step": 1617
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7969309462915604e-05,
+      "loss": 0.0085,
+      "step": 1618
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7961636828644502e-05,
+      "loss": 0.0167,
+      "step": 1619
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.79539641943734e-05,
+      "loss": 1.2275,
+      "step": 1620
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7946291560102305e-05,
+      "loss": 0.0115,
+      "step": 1621
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7938618925831203e-05,
+      "loss": 0.0083,
+      "step": 1622
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.79309462915601e-05,
+      "loss": 0.015,
+      "step": 1623
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7923273657289003e-05,
+      "loss": 1.1705,
+      "step": 1624
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7915601023017905e-05,
+      "loss": 0.0152,
+      "step": 1625
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7907928388746803e-05,
+      "loss": 2.0998,
+      "step": 1626
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7900255754475704e-05,
+      "loss": 0.0072,
+      "step": 1627
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7892583120204603e-05,
+      "loss": 0.0089,
+      "step": 1628
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7884910485933504e-05,
+      "loss": 0.0077,
+      "step": 1629
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7877237851662406e-05,
+      "loss": 0.0126,
+      "step": 1630
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7869565217391304e-05,
+      "loss": 1.2113,
+      "step": 1631
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7861892583120202e-05,
+      "loss": 0.0112,
+      "step": 1632
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7854219948849107e-05,
+      "loss": 0.0099,
+      "step": 1633
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7846547314578006e-05,
+      "loss": 0.0074,
+      "step": 1634
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7838874680306904e-05,
+      "loss": 0.0111,
+      "step": 1635
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.783120204603581e-05,
+      "loss": 0.0075,
+      "step": 1636
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7823529411764707e-05,
+      "loss": 1.2187,
+      "step": 1637
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7815856777493605e-05,
+      "loss": 0.0111,
+      "step": 1638
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7808184143222507e-05,
+      "loss": 0.0184,
+      "step": 1639
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.780051150895141e-05,
+      "loss": 0.0111,
+      "step": 1640
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7792838874680307e-05,
+      "loss": 0.0112,
+      "step": 1641
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.778516624040921e-05,
+      "loss": 0.0149,
+      "step": 1642
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.7777493606138107e-05,
+      "loss": 0.0151,
+      "step": 1643
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.7769820971867008e-05,
+      "loss": 0.0074,
+      "step": 1644
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.776214833759591e-05,
+      "loss": 0.9703,
+      "step": 1645
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.7754475703324808e-05,
+      "loss": 1.9364,
+      "step": 1646
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.7746803069053706e-05,
+      "loss": 0.011,
+      "step": 1647
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.773913043478261e-05,
+      "loss": 0.01,
+      "step": 1648
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.773145780051151e-05,
+      "loss": 0.0111,
+      "step": 1649
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.7723785166240408e-05,
+      "loss": 1.2373,
+      "step": 1650
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.771611253196931e-05,
+      "loss": 0.0074,
+      "step": 1651
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.770843989769821e-05,
+      "loss": 0.0116,
+      "step": 1652
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.770076726342711e-05,
+      "loss": 0.0117,
+      "step": 1653
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.769309462915601e-05,
+      "loss": 1.1636,
+      "step": 1654
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.768542199488491e-05,
+      "loss": 0.0075,
+      "step": 1655
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.767774936061381e-05,
+      "loss": 0.007,
+      "step": 1656
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.7670076726342712e-05,
+      "loss": 0.0174,
+      "step": 1657
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.766240409207161e-05,
+      "loss": 0.0113,
+      "step": 1658
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7654731457800512e-05,
+      "loss": 0.0084,
+      "step": 1659
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7647058823529414e-05,
+      "loss": 0.2994,
+      "step": 1660
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7639386189258312e-05,
+      "loss": 0.0213,
+      "step": 1661
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7631713554987213e-05,
+      "loss": 0.011,
+      "step": 1662
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7624040920716115e-05,
+      "loss": 1.2341,
+      "step": 1663
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7616368286445013e-05,
+      "loss": 1.1774,
+      "step": 1664
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7608695652173915e-05,
+      "loss": 0.0181,
+      "step": 1665
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7601023017902813e-05,
+      "loss": 0.0122,
+      "step": 1666
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7593350383631715e-05,
+      "loss": 0.0119,
+      "step": 1667
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7585677749360616e-05,
+      "loss": 0.0114,
+      "step": 1668
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7578005115089515e-05,
+      "loss": 0.0111,
+      "step": 1669
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7570332480818413e-05,
+      "loss": 0.011,
+      "step": 1670
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7562659846547318e-05,
+      "loss": 0.0113,
+      "step": 1671
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7554987212276216e-05,
+      "loss": 0.0161,
+      "step": 1672
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7547314578005114e-05,
+      "loss": 0.9491,
+      "step": 1673
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7539641943734016e-05,
+      "loss": 0.0108,
+      "step": 1674
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7531969309462917e-05,
+      "loss": 0.0114,
+      "step": 1675
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7524296675191816e-05,
+      "loss": 0.0107,
+      "step": 1676
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7516624040920717e-05,
+      "loss": 0.0171,
+      "step": 1677
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7508951406649616e-05,
+      "loss": 0.017,
+      "step": 1678
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7501278772378517e-05,
+      "loss": 1.939,
+      "step": 1679
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.749360613810742e-05,
+      "loss": 0.0151,
+      "step": 1680
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7485933503836317e-05,
+      "loss": 0.0109,
+      "step": 1681
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.7478260869565215e-05,
+      "loss": 0.0119,
+      "step": 1682
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.747058823529412e-05,
+      "loss": 1.2127,
+      "step": 1683
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.746291560102302e-05,
+      "loss": 0.0075,
+      "step": 1684
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.7455242966751917e-05,
+      "loss": 1.2283,
+      "step": 1685
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.744757033248082e-05,
+      "loss": 0.0107,
+      "step": 1686
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.743989769820972e-05,
+      "loss": 0.0141,
+      "step": 1687
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.7432225063938618e-05,
+      "loss": 1.2117,
+      "step": 1688
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.742455242966752e-05,
+      "loss": 0.8715,
+      "step": 1689
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.741687979539642e-05,
+      "loss": 0.0075,
+      "step": 1690
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.740920716112532e-05,
+      "loss": 1.2208,
+      "step": 1691
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.740153452685422e-05,
+      "loss": 0.0077,
+      "step": 1692
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.739386189258312e-05,
+      "loss": 0.0081,
+      "step": 1693
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.738618925831202e-05,
+      "loss": 0.0104,
+      "step": 1694
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.7378516624040923e-05,
+      "loss": 0.0162,
+      "step": 1695
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.737084398976982e-05,
+      "loss": 0.0173,
+      "step": 1696
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.736317135549872e-05,
+      "loss": 0.0194,
+      "step": 1697
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.7355498721227624e-05,
+      "loss": 0.0174,
+      "step": 1698
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.7347826086956522e-05,
+      "loss": 0.0081,
+      "step": 1699
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.734015345268542e-05,
+      "loss": 0.0121,
+      "step": 1700
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7332480818414322e-05,
+      "loss": 0.0137,
+      "step": 1701
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7324808184143224e-05,
+      "loss": 0.0117,
+      "step": 1702
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7317135549872122e-05,
+      "loss": 0.0133,
+      "step": 1703
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7309462915601024e-05,
+      "loss": 0.0175,
+      "step": 1704
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7301790281329922e-05,
+      "loss": 0.0076,
+      "step": 1705
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7294117647058823e-05,
+      "loss": 0.0179,
+      "step": 1706
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7286445012787725e-05,
+      "loss": 1.0541,
+      "step": 1707
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7278772378516623e-05,
+      "loss": 0.0077,
+      "step": 1708
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7271099744245525e-05,
+      "loss": 0.0079,
+      "step": 1709
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7263427109974426e-05,
+      "loss": 0.0102,
+      "step": 1710
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7255754475703325e-05,
+      "loss": 0.0158,
+      "step": 1711
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7248081841432223e-05,
+      "loss": 0.0107,
+      "step": 1712
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7240409207161128e-05,
+      "loss": 0.0159,
+      "step": 1713
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7232736572890026e-05,
+      "loss": 0.0125,
+      "step": 1714
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7225063938618928e-05,
+      "loss": 0.0081,
+      "step": 1715
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7217391304347826e-05,
+      "loss": 1.1946,
+      "step": 1716
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7209718670076728e-05,
+      "loss": 1.0298,
+      "step": 1717
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.720204603580563e-05,
+      "loss": 1.0207,
+      "step": 1718
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7194373401534527e-05,
+      "loss": 0.0152,
+      "step": 1719
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7186700767263426e-05,
+      "loss": 0.0187,
+      "step": 1720
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.717902813299233e-05,
+      "loss": 0.0134,
+      "step": 1721
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.717135549872123e-05,
+      "loss": 0.0173,
+      "step": 1722
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.7163682864450127e-05,
+      "loss": 1.0206,
+      "step": 1723
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.715601023017903e-05,
+      "loss": 0.0234,
+      "step": 1724
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.714833759590793e-05,
+      "loss": 0.0151,
+      "step": 1725
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.714066496163683e-05,
+      "loss": 0.0074,
+      "step": 1726
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.713299232736573e-05,
+      "loss": 0.0127,
+      "step": 1727
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.712531969309463e-05,
+      "loss": 0.8679,
+      "step": 1728
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.711764705882353e-05,
+      "loss": 0.0498,
+      "step": 1729
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.710997442455243e-05,
+      "loss": 0.0136,
+      "step": 1730
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.710230179028133e-05,
+      "loss": 1.2168,
+      "step": 1731
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.7094629156010228e-05,
+      "loss": 0.0105,
+      "step": 1732
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.7086956521739133e-05,
+      "loss": 0.0107,
+      "step": 1733
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.707928388746803e-05,
+      "loss": 0.0122,
+      "step": 1734
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.707161125319693e-05,
+      "loss": 0.0075,
+      "step": 1735
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.7063938618925834e-05,
+      "loss": 0.0607,
+      "step": 1736
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7056265984654733e-05,
+      "loss": 1.0181,
+      "step": 1737
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.704859335038363e-05,
+      "loss": 0.0095,
+      "step": 1738
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7040920716112533e-05,
+      "loss": 0.0128,
+      "step": 1739
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7033248081841434e-05,
+      "loss": 0.0096,
+      "step": 1740
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7025575447570332e-05,
+      "loss": 1.1781,
+      "step": 1741
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7017902813299234e-05,
+      "loss": 1.0132,
+      "step": 1742
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7010230179028132e-05,
+      "loss": 0.0097,
+      "step": 1743
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7002557544757034e-05,
+      "loss": 0.012,
+      "step": 1744
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6994884910485935e-05,
+      "loss": 0.0128,
+      "step": 1745
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6987212276214834e-05,
+      "loss": 0.0145,
+      "step": 1746
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6979539641943732e-05,
+      "loss": 0.7558,
+      "step": 1747
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6971867007672637e-05,
+      "loss": 0.0107,
+      "step": 1748
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6964194373401535e-05,
+      "loss": 1.0181,
+      "step": 1749
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6956521739130433e-05,
+      "loss": 1.7925,
+      "step": 1750
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6948849104859335e-05,
+      "loss": 0.0141,
+      "step": 1751
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6941176470588237e-05,
+      "loss": 0.0167,
+      "step": 1752
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6933503836317135e-05,
+      "loss": 0.0141,
+      "step": 1753
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6925831202046036e-05,
+      "loss": 0.0149,
+      "step": 1754
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6918158567774935e-05,
+      "loss": 0.0115,
+      "step": 1755
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6910485933503836e-05,
+      "loss": 0.0071,
+      "step": 1756
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6902813299232738e-05,
+      "loss": 0.0155,
+      "step": 1757
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6895140664961636e-05,
+      "loss": 0.0199,
+      "step": 1758
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6887468030690538e-05,
+      "loss": 0.0124,
+      "step": 1759
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.687979539641944e-05,
+      "loss": 0.0086,
+      "step": 1760
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.6872122762148337e-05,
+      "loss": 0.0147,
+      "step": 1761
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.6864450127877236e-05,
+      "loss": 0.0107,
+      "step": 1762
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.685677749360614e-05,
+      "loss": 1.025,
+      "step": 1763
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.684910485933504e-05,
+      "loss": 0.0105,
+      "step": 1764
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.6841432225063937e-05,
+      "loss": 0.0081,
+      "step": 1765
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.683375959079284e-05,
+      "loss": 0.009,
+      "step": 1766
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.682608695652174e-05,
+      "loss": 0.0109,
+      "step": 1767
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.6818414322250642e-05,
+      "loss": 0.008,
+      "step": 1768
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.681074168797954e-05,
+      "loss": 0.0132,
+      "step": 1769
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.680306905370844e-05,
+      "loss": 0.0119,
+      "step": 1770
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.6795396419437343e-05,
+      "loss": 0.0093,
+      "step": 1771
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.678772378516624e-05,
+      "loss": 1.0247,
+      "step": 1772
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.678005115089514e-05,
+      "loss": 1.024,
+      "step": 1773
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.677237851662404e-05,
+      "loss": 0.0106,
+      "step": 1774
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.6764705882352943e-05,
+      "loss": 0.0097,
+      "step": 1775
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.675703324808184e-05,
+      "loss": 0.0098,
+      "step": 1776
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6749360613810743e-05,
+      "loss": 1.0136,
+      "step": 1777
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.674168797953964e-05,
+      "loss": 0.0066,
+      "step": 1778
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6734015345268543e-05,
+      "loss": 0.1284,
+      "step": 1779
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6726342710997444e-05,
+      "loss": 0.9832,
+      "step": 1780
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6718670076726343e-05,
+      "loss": 0.0128,
+      "step": 1781
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6710997442455244e-05,
+      "loss": 0.0103,
+      "step": 1782
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6703324808184146e-05,
+      "loss": 0.0067,
+      "step": 1783
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6695652173913044e-05,
+      "loss": 0.0105,
+      "step": 1784
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6687979539641942e-05,
+      "loss": 0.0146,
+      "step": 1785
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6680306905370847e-05,
+      "loss": 0.0068,
+      "step": 1786
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6672634271099745e-05,
+      "loss": 0.0148,
+      "step": 1787
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6664961636828644e-05,
+      "loss": 0.0107,
+      "step": 1788
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6657289002557545e-05,
+      "loss": 2.4738,
+      "step": 1789
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6649616368286447e-05,
+      "loss": 0.9625,
+      "step": 1790
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6641943734015345e-05,
+      "loss": 0.2578,
+      "step": 1791
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6634271099744247e-05,
+      "loss": 0.0069,
+      "step": 1792
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6626598465473145e-05,
+      "loss": 0.011,
+      "step": 1793
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6618925831202047e-05,
+      "loss": 0.0066,
+      "step": 1794
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6611253196930948e-05,
+      "loss": 0.0069,
+      "step": 1795
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6603580562659846e-05,
+      "loss": 0.0071,
+      "step": 1796
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6595907928388745e-05,
+      "loss": 0.0068,
+      "step": 1797
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.658823529411765e-05,
+      "loss": 0.0232,
+      "step": 1798
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6580562659846548e-05,
+      "loss": 1.1318,
+      "step": 1799
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6572890025575446e-05,
+      "loss": 0.0101,
+      "step": 1800
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6565217391304348e-05,
+      "loss": 0.0187,
+      "step": 1801
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.655754475703325e-05,
+      "loss": 1.0718,
+      "step": 1802
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6549872122762148e-05,
+      "loss": 1.2349,
+      "step": 1803
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.654219948849105e-05,
+      "loss": 0.0148,
+      "step": 1804
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.6534526854219947e-05,
+      "loss": 0.0148,
+      "step": 1805
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.652685421994885e-05,
+      "loss": 0.0067,
+      "step": 1806
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.651918158567775e-05,
+      "loss": 0.0075,
+      "step": 1807
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.651150895140665e-05,
+      "loss": 0.0157,
+      "step": 1808
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.650383631713555e-05,
+      "loss": 0.0084,
+      "step": 1809
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.6496163682864452e-05,
+      "loss": 0.0123,
+      "step": 1810
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.648849104859335e-05,
+      "loss": 0.0065,
+      "step": 1811
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.648081841432225e-05,
+      "loss": 0.014,
+      "step": 1812
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.6473145780051154e-05,
+      "loss": 0.0152,
+      "step": 1813
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.6465473145780052e-05,
+      "loss": 0.0168,
+      "step": 1814
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.645780051150895e-05,
+      "loss": 0.0063,
+      "step": 1815
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.645012787723785e-05,
+      "loss": 0.6994,
+      "step": 1816
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.6442455242966753e-05,
+      "loss": 0.0132,
+      "step": 1817
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.643478260869565e-05,
+      "loss": 0.0103,
+      "step": 1818
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.6427109974424553e-05,
+      "loss": 0.013,
+      "step": 1819
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.641943734015345e-05,
+      "loss": 1.1813,
+      "step": 1820
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6411764705882356e-05,
+      "loss": 0.9809,
+      "step": 1821
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6404092071611254e-05,
+      "loss": 1.2339,
+      "step": 1822
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6396419437340153e-05,
+      "loss": 0.9744,
+      "step": 1823
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6388746803069054e-05,
+      "loss": 0.0098,
+      "step": 1824
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6381074168797956e-05,
+      "loss": 0.0062,
+      "step": 1825
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6373401534526854e-05,
+      "loss": 0.0151,
+      "step": 1826
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6365728900255756e-05,
+      "loss": 0.013,
+      "step": 1827
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6358056265984654e-05,
+      "loss": 0.0071,
+      "step": 1828
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6350383631713556e-05,
+      "loss": 0.0117,
+      "step": 1829
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6342710997442457e-05,
+      "loss": 0.0165,
+      "step": 1830
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6335038363171355e-05,
+      "loss": 0.1519,
+      "step": 1831
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6327365728900257e-05,
+      "loss": 1.0247,
+      "step": 1832
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.631969309462916e-05,
+      "loss": 0.0092,
+      "step": 1833
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6312020460358057e-05,
+      "loss": 0.0098,
+      "step": 1834
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6304347826086955e-05,
+      "loss": 1.1829,
+      "step": 1835
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.629667519181586e-05,
+      "loss": 0.006,
+      "step": 1836
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.628900255754476e-05,
+      "loss": 0.0109,
+      "step": 1837
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.6281329923273657e-05,
+      "loss": 0.6059,
+      "step": 1838
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.6273657289002558e-05,
+      "loss": 0.01,
+      "step": 1839
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.626598465473146e-05,
+      "loss": 0.0149,
+      "step": 1840
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.6258312020460358e-05,
+      "loss": 0.0152,
+      "step": 1841
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.625063938618926e-05,
+      "loss": 0.0171,
+      "step": 1842
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.6242966751918158e-05,
+      "loss": 0.0105,
+      "step": 1843
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.623529411764706e-05,
+      "loss": 0.0093,
+      "step": 1844
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.622762148337596e-05,
+      "loss": 1.4125,
+      "step": 1845
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.621994884910486e-05,
+      "loss": 0.0107,
+      "step": 1846
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.6212276214833757e-05,
+      "loss": 0.0133,
+      "step": 1847
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.6204603580562662e-05,
+      "loss": 0.0063,
+      "step": 1848
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.619693094629156e-05,
+      "loss": 0.0063,
+      "step": 1849
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.618925831202046e-05,
+      "loss": 0.009,
+      "step": 1850
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.618158567774936e-05,
+      "loss": 0.0137,
+      "step": 1851
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.6173913043478262e-05,
+      "loss": 0.0145,
+      "step": 1852
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.616624040920716e-05,
+      "loss": 0.0355,
+      "step": 1853
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.6158567774936062e-05,
+      "loss": 0.9601,
+      "step": 1854
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.615089514066496e-05,
+      "loss": 0.0135,
+      "step": 1855
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6143222506393862e-05,
+      "loss": 1.0129,
+      "step": 1856
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6135549872122763e-05,
+      "loss": 0.0339,
+      "step": 1857
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.612787723785166e-05,
+      "loss": 0.0131,
+      "step": 1858
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6120204603580563e-05,
+      "loss": 0.009,
+      "step": 1859
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6112531969309465e-05,
+      "loss": 0.0087,
+      "step": 1860
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6104859335038363e-05,
+      "loss": 0.012,
+      "step": 1861
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.609718670076726e-05,
+      "loss": 0.0088,
+      "step": 1862
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6089514066496166e-05,
+      "loss": 0.0117,
+      "step": 1863
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6081841432225065e-05,
+      "loss": 2.5214,
+      "step": 1864
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6074168797953963e-05,
+      "loss": 0.0107,
+      "step": 1865
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6066496163682864e-05,
+      "loss": 0.0101,
+      "step": 1866
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6058823529411766e-05,
+      "loss": 0.0118,
+      "step": 1867
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6051150895140664e-05,
+      "loss": 0.0087,
+      "step": 1868
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6043478260869566e-05,
+      "loss": 0.0122,
+      "step": 1869
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6035805626598464e-05,
+      "loss": 0.0084,
+      "step": 1870
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6028132992327366e-05,
+      "loss": 0.0104,
+      "step": 1871
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6020460358056267e-05,
+      "loss": 0.1587,
+      "step": 1872
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6012787723785166e-05,
+      "loss": 0.0057,
+      "step": 1873
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6005115089514067e-05,
+      "loss": 0.0085,
+      "step": 1874
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.599744245524297e-05,
+      "loss": 0.0056,
+      "step": 1875
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.5989769820971867e-05,
+      "loss": 1.2194,
+      "step": 1876
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.598209718670077e-05,
+      "loss": 0.0056,
+      "step": 1877
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.5974424552429667e-05,
+      "loss": 0.0057,
+      "step": 1878
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.596675191815857e-05,
+      "loss": 1.0276,
+      "step": 1879
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.595907928388747e-05,
+      "loss": 0.0083,
+      "step": 1880
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.5951406649616368e-05,
+      "loss": 0.0085,
+      "step": 1881
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.594373401534527e-05,
+      "loss": 0.0363,
+      "step": 1882
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.593606138107417e-05,
+      "loss": 0.0157,
+      "step": 1883
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.592838874680307e-05,
+      "loss": 0.0313,
+      "step": 1884
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.5920716112531968e-05,
+      "loss": 0.0062,
+      "step": 1885
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.5913043478260873e-05,
+      "loss": 0.0195,
+      "step": 1886
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.590537084398977e-05,
+      "loss": 0.0151,
+      "step": 1887
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.589769820971867e-05,
+      "loss": 0.0101,
+      "step": 1888
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.589002557544757e-05,
+      "loss": 1.1092,
+      "step": 1889
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.5882352941176473e-05,
+      "loss": 0.0323,
+      "step": 1890
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.587468030690537e-05,
+      "loss": 0.0086,
+      "step": 1891
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.5867007672634272e-05,
+      "loss": 0.0147,
+      "step": 1892
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.585933503836317e-05,
+      "loss": 0.0084,
+      "step": 1893
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.5851662404092072e-05,
+      "loss": 0.5596,
+      "step": 1894
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5843989769820974e-05,
+      "loss": 0.0114,
+      "step": 1895
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5836317135549872e-05,
+      "loss": 0.0083,
+      "step": 1896
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.582864450127877e-05,
+      "loss": 0.0082,
+      "step": 1897
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5820971867007675e-05,
+      "loss": 0.0146,
+      "step": 1898
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5813299232736574e-05,
+      "loss": 0.0081,
+      "step": 1899
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5805626598465472e-05,
+      "loss": 0.0081,
+      "step": 1900
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5797953964194373e-05,
+      "loss": 0.0116,
+      "step": 1901
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5790281329923275e-05,
+      "loss": 0.0084,
+      "step": 1902
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5782608695652173e-05,
+      "loss": 0.0108,
+      "step": 1903
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5774936061381075e-05,
+      "loss": 0.6146,
+      "step": 1904
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5767263427109973e-05,
+      "loss": 0.0162,
+      "step": 1905
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5759590792838875e-05,
+      "loss": 0.0134,
+      "step": 1906
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5751918158567776e-05,
+      "loss": 0.0047,
+      "step": 1907
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5744245524296674e-05,
+      "loss": 0.0045,
+      "step": 1908
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5736572890025576e-05,
+      "loss": 0.0087,
+      "step": 1909
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5728900255754478e-05,
+      "loss": 2.0913,
+      "step": 1910
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5721227621483376e-05,
+      "loss": 0.0075,
+      "step": 1911
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5713554987212274e-05,
+      "loss": 1.3387,
+      "step": 1912
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.570588235294118e-05,
+      "loss": 0.0102,
+      "step": 1913
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5698209718670077e-05,
+      "loss": 0.0076,
+      "step": 1914
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5690537084398976e-05,
+      "loss": 0.0131,
+      "step": 1915
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5682864450127877e-05,
+      "loss": 0.0079,
+      "step": 1916
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.567519181585678e-05,
+      "loss": 0.0102,
+      "step": 1917
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5667519181585677e-05,
+      "loss": 0.9554,
+      "step": 1918
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.565984654731458e-05,
+      "loss": 0.0088,
+      "step": 1919
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.5652173913043477e-05,
+      "loss": 1.0265,
+      "step": 1920
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.564450127877238e-05,
+      "loss": 0.0155,
+      "step": 1921
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.563682864450128e-05,
+      "loss": 0.0079,
+      "step": 1922
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.562915601023018e-05,
+      "loss": 1.0467,
+      "step": 1923
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.5621483375959077e-05,
+      "loss": 0.0132,
+      "step": 1924
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.561381074168798e-05,
+      "loss": 0.0077,
+      "step": 1925
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.560613810741688e-05,
+      "loss": 0.0064,
+      "step": 1926
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.559846547314578e-05,
+      "loss": 0.0104,
+      "step": 1927
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.559079283887468e-05,
+      "loss": 0.0096,
+      "step": 1928
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.558312020460358e-05,
+      "loss": 0.0049,
+      "step": 1929
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.5575447570332483e-05,
+      "loss": 0.0104,
+      "step": 1930
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.556777493606138e-05,
+      "loss": 1.0424,
+      "step": 1931
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5560102301790283e-05,
+      "loss": 0.0079,
+      "step": 1932
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5552429667519184e-05,
+      "loss": 0.0104,
+      "step": 1933
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5544757033248083e-05,
+      "loss": 0.0077,
+      "step": 1934
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.553708439897698e-05,
+      "loss": 0.005,
+      "step": 1935
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5529411764705886e-05,
+      "loss": 2.6544,
+      "step": 1936
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5521739130434784e-05,
+      "loss": 0.0107,
+      "step": 1937
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5514066496163682e-05,
+      "loss": 0.0124,
+      "step": 1938
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5506393861892584e-05,
+      "loss": 0.0049,
+      "step": 1939
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5498721227621485e-05,
+      "loss": 0.0078,
+      "step": 1940
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5491048593350384e-05,
+      "loss": 2.1009,
+      "step": 1941
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5483375959079285e-05,
+      "loss": 0.0116,
+      "step": 1942
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5475703324808183e-05,
+      "loss": 0.0049,
+      "step": 1943
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5468030690537085e-05,
+      "loss": 0.0079,
+      "step": 1944
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5460358056265987e-05,
+      "loss": 1.0307,
+      "step": 1945
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5452685421994885e-05,
+      "loss": 0.0147,
+      "step": 1946
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5445012787723783e-05,
+      "loss": 0.0084,
+      "step": 1947
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5437340153452688e-05,
+      "loss": 0.0262,
+      "step": 1948
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5429667519181586e-05,
+      "loss": 0.0115,
+      "step": 1949
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5421994884910485e-05,
+      "loss": 1.3221,
+      "step": 1950
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5414322250639386e-05,
+      "loss": 0.7929,
+      "step": 1951
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5406649616368288e-05,
+      "loss": 1.2069,
+      "step": 1952
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5398976982097186e-05,
+      "loss": 0.0081,
+      "step": 1953
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5391304347826088e-05,
+      "loss": 0.0069,
+      "step": 1954
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5383631713554986e-05,
+      "loss": 0.0097,
+      "step": 1955
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5375959079283887e-05,
+      "loss": 0.0082,
+      "step": 1956
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.536828644501279e-05,
+      "loss": 0.0061,
+      "step": 1957
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5360613810741687e-05,
+      "loss": 0.0082,
+      "step": 1958
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.535294117647059e-05,
+      "loss": 0.0113,
+      "step": 1959
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.534526854219949e-05,
+      "loss": 0.0098,
+      "step": 1960
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.533759590792839e-05,
+      "loss": 1.0221,
+      "step": 1961
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.5329923273657287e-05,
+      "loss": 0.0039,
+      "step": 1962
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.5322250639386192e-05,
+      "loss": 0.0127,
+      "step": 1963
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.531457800511509e-05,
+      "loss": 0.0051,
+      "step": 1964
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.530690537084399e-05,
+      "loss": 0.0086,
+      "step": 1965
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.529923273657289e-05,
+      "loss": 0.0096,
+      "step": 1966
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.529156010230179e-05,
+      "loss": 0.0164,
+      "step": 1967
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.528388746803069e-05,
+      "loss": 1.0127,
+      "step": 1968
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.527621483375959e-05,
+      "loss": 0.0149,
+      "step": 1969
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.526854219948849e-05,
+      "loss": 0.0085,
+      "step": 1970
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.526086956521739e-05,
+      "loss": 0.0116,
+      "step": 1971
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5253196930946293e-05,
+      "loss": 0.0168,
+      "step": 1972
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5245524296675191e-05,
+      "loss": 0.0052,
+      "step": 1973
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5237851662404091e-05,
+      "loss": 0.0052,
+      "step": 1974
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5230179028132993e-05,
+      "loss": 1.0013,
+      "step": 1975
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5222506393861893e-05,
+      "loss": 0.0122,
+      "step": 1976
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.521483375959079e-05,
+      "loss": 0.0087,
+      "step": 1977
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5207161125319694e-05,
+      "loss": 1.9624,
+      "step": 1978
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5199488491048592e-05,
+      "loss": 0.0088,
+      "step": 1979
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5191815856777496e-05,
+      "loss": 0.0596,
+      "step": 1980
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9090909090909091,
+      "eval_loss": 0.40464258193969727,
+      "eval_runtime": 58.0491,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1980
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5184143222506396e-05,
+      "loss": 0.009,
+      "step": 1981
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5176470588235294e-05,
+      "loss": 0.0063,
+      "step": 1982
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5168797953964197e-05,
+      "loss": 0.0092,
+      "step": 1983
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5161125319693095e-05,
+      "loss": 0.0092,
+      "step": 1984
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5153452685421995e-05,
+      "loss": 0.0259,
+      "step": 1985
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5145780051150897e-05,
+      "loss": 0.0093,
+      "step": 1986
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5138107416879797e-05,
+      "loss": 1.3347,
+      "step": 1987
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5130434782608695e-05,
+      "loss": 0.0135,
+      "step": 1988
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5122762148337598e-05,
+      "loss": 0.0134,
+      "step": 1989
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5115089514066497e-05,
+      "loss": 0.0051,
+      "step": 1990
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5107416879795396e-05,
+      "loss": 0.0106,
+      "step": 1991
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5099744245524298e-05,
+      "loss": 0.013,
+      "step": 1992
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5092071611253198e-05,
+      "loss": 0.0095,
+      "step": 1993
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5084398976982096e-05,
+      "loss": 0.0096,
+      "step": 1994
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5076726342711e-05,
+      "loss": 0.0112,
+      "step": 1995
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5069053708439898e-05,
+      "loss": 0.0137,
+      "step": 1996
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5061381074168798e-05,
+      "loss": 0.9611,
+      "step": 1997
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50537084398977e-05,
+      "loss": 0.009,
+      "step": 1998
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50460358056266e-05,
+      "loss": 0.0066,
+      "step": 1999
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.5038363171355497e-05,
+      "loss": 0.0091,
+      "step": 2000
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50306905370844e-05,
+      "loss": 0.9856,
+      "step": 2001
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5023017902813299e-05,
+      "loss": 0.0137,
+      "step": 2002
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5015345268542199e-05,
+      "loss": 0.0094,
+      "step": 2003
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.50076726342711e-05,
+      "loss": 0.9276,
+      "step": 2004
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.0051,
+      "step": 2005
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4992327365728902e-05,
+      "loss": 0.0097,
+      "step": 2006
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.49846547314578e-05,
+      "loss": 0.8053,
+      "step": 2007
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4976982097186702e-05,
+      "loss": 0.5635,
+      "step": 2008
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4969309462915602e-05,
+      "loss": 0.0099,
+      "step": 2009
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4961636828644502e-05,
+      "loss": 0.0108,
+      "step": 2010
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4953964194373402e-05,
+      "loss": 1.3337,
+      "step": 2011
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4946291560102303e-05,
+      "loss": 0.0098,
+      "step": 2012
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4938618925831201e-05,
+      "loss": 0.0099,
+      "step": 2013
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4930946291560103e-05,
+      "loss": 0.011,
+      "step": 2014
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4923273657289003e-05,
+      "loss": 0.0166,
+      "step": 2015
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4915601023017903e-05,
+      "loss": 1.3541,
+      "step": 2016
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4907928388746803e-05,
+      "loss": 0.0145,
+      "step": 2017
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4900255754475704e-05,
+      "loss": 0.0051,
+      "step": 2018
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4892583120204604e-05,
+      "loss": 0.0051,
+      "step": 2019
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4884910485933504e-05,
+      "loss": 0.0139,
+      "step": 2020
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4877237851662404e-05,
+      "loss": 0.0097,
+      "step": 2021
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4869565217391306e-05,
+      "loss": 0.9418,
+      "step": 2022
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4861892583120204e-05,
+      "loss": 0.9619,
+      "step": 2023
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4854219948849106e-05,
+      "loss": 1.3495,
+      "step": 2024
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4846547314578006e-05,
+      "loss": 0.0051,
+      "step": 2025
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4838874680306905e-05,
+      "loss": 1.3222,
+      "step": 2026
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4831202046035805e-05,
+      "loss": 1.1638,
+      "step": 2027
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4823529411764707e-05,
+      "loss": 0.009,
+      "step": 2028
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4815856777493605e-05,
+      "loss": 0.0059,
+      "step": 2029
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4808184143222507e-05,
+      "loss": 0.9946,
+      "step": 2030
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4800511508951408e-05,
+      "loss": 0.1291,
+      "step": 2031
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4792838874680307e-05,
+      "loss": 0.0089,
+      "step": 2032
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4785166240409208e-05,
+      "loss": 0.8113,
+      "step": 2033
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4777493606138108e-05,
+      "loss": 0.0109,
+      "step": 2034
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4769820971867008e-05,
+      "loss": 0.0061,
+      "step": 2035
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4762148337595908e-05,
+      "loss": 0.0157,
+      "step": 2036
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.475447570332481e-05,
+      "loss": 0.0104,
+      "step": 2037
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.4746803069053708e-05,
+      "loss": 0.012,
+      "step": 2038
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.473913043478261e-05,
+      "loss": 0.0106,
+      "step": 2039
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.473145780051151e-05,
+      "loss": 0.006,
+      "step": 2040
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.472378516624041e-05,
+      "loss": 0.0122,
+      "step": 2041
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.471611253196931e-05,
+      "loss": 0.0103,
+      "step": 2042
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.470843989769821e-05,
+      "loss": 0.9465,
+      "step": 2043
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.4700767263427109e-05,
+      "loss": 0.0149,
+      "step": 2044
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.469309462915601e-05,
+      "loss": 0.0146,
+      "step": 2045
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.468542199488491e-05,
+      "loss": 0.9225,
+      "step": 2046
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.4677749360613812e-05,
+      "loss": 0.0101,
+      "step": 2047
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.467007672634271e-05,
+      "loss": 0.0052,
+      "step": 2048
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.4662404092071612e-05,
+      "loss": 1.682,
+      "step": 2049
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4654731457800512e-05,
+      "loss": 0.9498,
+      "step": 2050
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4647058823529412e-05,
+      "loss": 0.0097,
+      "step": 2051
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4639386189258312e-05,
+      "loss": 0.0067,
+      "step": 2052
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4631713554987213e-05,
+      "loss": 0.0052,
+      "step": 2053
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4624040920716112e-05,
+      "loss": 0.0055,
+      "step": 2054
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4616368286445013e-05,
+      "loss": 0.0225,
+      "step": 2055
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4608695652173915e-05,
+      "loss": 0.0052,
+      "step": 2056
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4601023017902813e-05,
+      "loss": 0.0066,
+      "step": 2057
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4593350383631715e-05,
+      "loss": 0.0112,
+      "step": 2058
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4585677749360615e-05,
+      "loss": 1.3145,
+      "step": 2059
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4578005115089514e-05,
+      "loss": 0.0099,
+      "step": 2060
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4570332480818414e-05,
+      "loss": 0.0055,
+      "step": 2061
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4562659846547316e-05,
+      "loss": 0.0149,
+      "step": 2062
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4554987212276214e-05,
+      "loss": 0.0102,
+      "step": 2063
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4547314578005116e-05,
+      "loss": 0.0108,
+      "step": 2064
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4539641943734016e-05,
+      "loss": 0.0065,
+      "step": 2065
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4531969309462916e-05,
+      "loss": 0.0185,
+      "step": 2066
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4524296675191816e-05,
+      "loss": 0.9659,
+      "step": 2067
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4516624040920717e-05,
+      "loss": 0.0193,
+      "step": 2068
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4508951406649615e-05,
+      "loss": 0.0146,
+      "step": 2069
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4501278772378517e-05,
+      "loss": 0.0061,
+      "step": 2070
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4493606138107417e-05,
+      "loss": 0.0222,
+      "step": 2071
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4485933503836319e-05,
+      "loss": 0.0096,
+      "step": 2072
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4478260869565217e-05,
+      "loss": 0.0096,
+      "step": 2073
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4470588235294118e-05,
+      "loss": 0.005,
+      "step": 2074
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4462915601023018e-05,
+      "loss": 0.8301,
+      "step": 2075
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4455242966751918e-05,
+      "loss": 0.0049,
+      "step": 2076
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4447570332480818e-05,
+      "loss": 0.0097,
+      "step": 2077
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.443989769820972e-05,
+      "loss": 0.0062,
+      "step": 2078
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.4432225063938618e-05,
+      "loss": 1.3459,
+      "step": 2079
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.442455242966752e-05,
+      "loss": 0.0096,
+      "step": 2080
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4416879795396421e-05,
+      "loss": 1.2176,
+      "step": 2081
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.440920716112532e-05,
+      "loss": 0.0114,
+      "step": 2082
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4401534526854221e-05,
+      "loss": 0.0109,
+      "step": 2083
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4393861892583121e-05,
+      "loss": 0.0105,
+      "step": 2084
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.4386189258312021e-05,
+      "loss": 0.0107,
+      "step": 2085
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.437851662404092e-05,
+      "loss": 0.9442,
+      "step": 2086
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.4370843989769822e-05,
+      "loss": 0.01,
+      "step": 2087
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.436317135549872e-05,
+      "loss": 0.0099,
+      "step": 2088
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4355498721227622e-05,
+      "loss": 0.0167,
+      "step": 2089
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4347826086956522e-05,
+      "loss": 0.0051,
+      "step": 2090
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4340153452685422e-05,
+      "loss": 1.333,
+      "step": 2091
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4332480818414322e-05,
+      "loss": 0.0111,
+      "step": 2092
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4324808184143224e-05,
+      "loss": 0.0103,
+      "step": 2093
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4317135549872122e-05,
+      "loss": 0.0111,
+      "step": 2094
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4309462915601023e-05,
+      "loss": 0.0165,
+      "step": 2095
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4301790281329923e-05,
+      "loss": 0.0089,
+      "step": 2096
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4294117647058823e-05,
+      "loss": 0.0143,
+      "step": 2097
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4286445012787723e-05,
+      "loss": 0.4759,
+      "step": 2098
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4278772378516625e-05,
+      "loss": 0.0102,
+      "step": 2099
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4271099744245525e-05,
+      "loss": 0.0059,
+      "step": 2100
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4263427109974425e-05,
+      "loss": 0.0152,
+      "step": 2101
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4255754475703325e-05,
+      "loss": 0.0051,
+      "step": 2102
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4248081841432226e-05,
+      "loss": 0.0098,
+      "step": 2103
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4240409207161126e-05,
+      "loss": 0.0147,
+      "step": 2104
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4232736572890026e-05,
+      "loss": 0.0051,
+      "step": 2105
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4225063938618928e-05,
+      "loss": 1.3384,
+      "step": 2106
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4217391304347826e-05,
+      "loss": 0.0055,
+      "step": 2107
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4209718670076727e-05,
+      "loss": 0.0136,
+      "step": 2108
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4202046035805627e-05,
+      "loss": 0.0056,
+      "step": 2109
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4194373401534527e-05,
+      "loss": 0.0094,
+      "step": 2110
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4186700767263427e-05,
+      "loss": 0.0135,
+      "step": 2111
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4179028132992329e-05,
+      "loss": 0.0129,
+      "step": 2112
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4171355498721227e-05,
+      "loss": 0.0051,
+      "step": 2113
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4163682864450129e-05,
+      "loss": 0.0134,
+      "step": 2114
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4156010230179029e-05,
+      "loss": 0.0091,
+      "step": 2115
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4148337595907929e-05,
+      "loss": 0.0072,
+      "step": 2116
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.4140664961636828e-05,
+      "loss": 0.005,
+      "step": 2117
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.413299232736573e-05,
+      "loss": 0.0112,
+      "step": 2118
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.4125319693094628e-05,
+      "loss": 0.005,
+      "step": 2119
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.411764705882353e-05,
+      "loss": 0.0085,
+      "step": 2120
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.410997442455243e-05,
+      "loss": 0.0121,
+      "step": 2121
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.410230179028133e-05,
+      "loss": 0.0084,
+      "step": 2122
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.409462915601023e-05,
+      "loss": 0.0084,
+      "step": 2123
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.4086956521739131e-05,
+      "loss": 0.0083,
+      "step": 2124
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4079283887468031e-05,
+      "loss": 1.3402,
+      "step": 2125
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4071611253196931e-05,
+      "loss": 0.0124,
+      "step": 2126
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4063938618925831e-05,
+      "loss": 0.0082,
+      "step": 2127
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4056265984654733e-05,
+      "loss": 0.9694,
+      "step": 2128
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4048593350383633e-05,
+      "loss": 0.0081,
+      "step": 2129
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4040920716112532e-05,
+      "loss": 1.2102,
+      "step": 2130
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4033248081841434e-05,
+      "loss": 0.0113,
+      "step": 2131
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4025575447570332e-05,
+      "loss": 0.0114,
+      "step": 2132
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4017902813299234e-05,
+      "loss": 0.0252,
+      "step": 2133
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4010230179028134e-05,
+      "loss": 0.9785,
+      "step": 2134
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4002557544757034e-05,
+      "loss": 0.0061,
+      "step": 2135
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.3994884910485934e-05,
+      "loss": 0.0096,
+      "step": 2136
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3987212276214835e-05,
+      "loss": 0.0088,
+      "step": 2137
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3979539641943733e-05,
+      "loss": 0.0047,
+      "step": 2138
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3971867007672635e-05,
+      "loss": 0.0108,
+      "step": 2139
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3964194373401535e-05,
+      "loss": 0.0081,
+      "step": 2140
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3956521739130435e-05,
+      "loss": 0.0108,
+      "step": 2141
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3948849104859335e-05,
+      "loss": 0.0126,
+      "step": 2142
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3941176470588236e-05,
+      "loss": 0.0046,
+      "step": 2143
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3933503836317135e-05,
+      "loss": 0.0122,
+      "step": 2144
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3925831202046036e-05,
+      "loss": 0.0075,
+      "step": 2145
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3918158567774936e-05,
+      "loss": 0.0123,
+      "step": 2146
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3910485933503836e-05,
+      "loss": 0.0134,
+      "step": 2147
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3902813299232736e-05,
+      "loss": 0.0115,
+      "step": 2148
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3895140664961638e-05,
+      "loss": 0.0044,
+      "step": 2149
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3887468030690536e-05,
+      "loss": 0.0121,
+      "step": 2150
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3879795396419437e-05,
+      "loss": 0.0101,
+      "step": 2151
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3872122762148337e-05,
+      "loss": 0.0099,
+      "step": 2152
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3864450127877239e-05,
+      "loss": 0.0044,
+      "step": 2153
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3856777493606139e-05,
+      "loss": 0.0094,
+      "step": 2154
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3849104859335039e-05,
+      "loss": 0.0124,
+      "step": 2155
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.384143222506394e-05,
+      "loss": 0.0069,
+      "step": 2156
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.3833759590792839e-05,
+      "loss": 0.0068,
+      "step": 2157
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.382608695652174e-05,
+      "loss": 0.0094,
+      "step": 2158
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.381841432225064e-05,
+      "loss": 0.0067,
+      "step": 2159
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.381074168797954e-05,
+      "loss": 0.0044,
+      "step": 2160
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.380306905370844e-05,
+      "loss": 0.0041,
+      "step": 2161
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.3795396419437342e-05,
+      "loss": 0.0041,
+      "step": 2162
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.378772378516624e-05,
+      "loss": 0.009,
+      "step": 2163
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.3780051150895141e-05,
+      "loss": 0.0092,
+      "step": 2164
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3772378516624041e-05,
+      "loss": 0.0101,
+      "step": 2165
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3764705882352941e-05,
+      "loss": 0.0064,
+      "step": 2166
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3757033248081841e-05,
+      "loss": 0.0088,
+      "step": 2167
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3749360613810743e-05,
+      "loss": 0.0059,
+      "step": 2168
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3741687979539641e-05,
+      "loss": 0.0062,
+      "step": 2169
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3734015345268543e-05,
+      "loss": 0.0085,
+      "step": 2170
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3726342710997443e-05,
+      "loss": 0.0141,
+      "step": 2171
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3718670076726343e-05,
+      "loss": 0.0061,
+      "step": 2172
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3710997442455242e-05,
+      "loss": 0.0084,
+      "step": 2173
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3703324808184144e-05,
+      "loss": 0.0038,
+      "step": 2174
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3695652173913042e-05,
+      "loss": 0.006,
+      "step": 2175
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3687979539641944e-05,
+      "loss": 0.0039,
+      "step": 2176
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3680306905370844e-05,
+      "loss": 0.0205,
+      "step": 2177
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3672634271099745e-05,
+      "loss": 0.0062,
+      "step": 2178
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3664961636828645e-05,
+      "loss": 0.0021,
+      "step": 2179
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3657289002557545e-05,
+      "loss": 0.0058,
+      "step": 2180
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3649616368286447e-05,
+      "loss": 0.0101,
+      "step": 2181
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3641943734015345e-05,
+      "loss": 0.0036,
+      "step": 2182
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3634271099744247e-05,
+      "loss": 0.0037,
+      "step": 2183
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3626598465473147e-05,
+      "loss": 0.0036,
+      "step": 2184
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3618925831202047e-05,
+      "loss": 0.0036,
+      "step": 2185
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3611253196930946e-05,
+      "loss": 0.0035,
+      "step": 2186
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3603580562659848e-05,
+      "loss": 0.0059,
+      "step": 2187
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3595907928388746e-05,
+      "loss": 0.0056,
+      "step": 2188
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3588235294117648e-05,
+      "loss": 0.0139,
+      "step": 2189
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3580562659846548e-05,
+      "loss": 0.0558,
+      "step": 2190
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3572890025575448e-05,
+      "loss": 1.419,
+      "step": 2191
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3565217391304348e-05,
+      "loss": 0.0058,
+      "step": 2192
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.355754475703325e-05,
+      "loss": 0.0035,
+      "step": 2193
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3549872122762147e-05,
+      "loss": 0.0055,
+      "step": 2194
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3542199488491049e-05,
+      "loss": 1.2049,
+      "step": 2195
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3534526854219949e-05,
+      "loss": 1.3646,
+      "step": 2196
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3526854219948849e-05,
+      "loss": 1.3803,
+      "step": 2197
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3519181585677749e-05,
+      "loss": 0.0055,
+      "step": 2198
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.351150895140665e-05,
+      "loss": 0.0059,
+      "step": 2199
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.3503836317135549e-05,
+      "loss": 0.0056,
+      "step": 2200
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.349616368286445e-05,
+      "loss": 0.0097,
+      "step": 2201
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.348849104859335e-05,
+      "loss": 1.3497,
+      "step": 2202
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.348081841432225e-05,
+      "loss": 1.0989,
+      "step": 2203
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3473145780051152e-05,
+      "loss": 0.0117,
+      "step": 2204
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3465473145780052e-05,
+      "loss": 0.0038,
+      "step": 2205
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3457800511508953e-05,
+      "loss": 0.0078,
+      "step": 2206
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3450127877237852e-05,
+      "loss": 0.0058,
+      "step": 2207
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3442455242966753e-05,
+      "loss": 0.9775,
+      "step": 2208
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3434782608695653e-05,
+      "loss": 0.0037,
+      "step": 2209
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3427109974424553e-05,
+      "loss": 0.0079,
+      "step": 2210
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3419437340153453e-05,
+      "loss": 0.0663,
+      "step": 2211
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3411764705882354e-05,
+      "loss": 0.0059,
+      "step": 2212
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3404092071611253e-05,
+      "loss": 0.0039,
+      "step": 2213
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3396419437340154e-05,
+      "loss": 1.3169,
+      "step": 2214
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3388746803069054e-05,
+      "loss": 0.0061,
+      "step": 2215
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3381074168797954e-05,
+      "loss": 0.006,
+      "step": 2216
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3373401534526854e-05,
+      "loss": 0.004,
+      "step": 2217
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3365728900255756e-05,
+      "loss": 0.0045,
+      "step": 2218
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3358056265984654e-05,
+      "loss": 0.006,
+      "step": 2219
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3350383631713556e-05,
+      "loss": 0.0058,
+      "step": 2220
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3342710997442455e-05,
+      "loss": 0.014,
+      "step": 2221
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3335038363171355e-05,
+      "loss": 1.1674,
+      "step": 2222
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3327365728900255e-05,
+      "loss": 0.004,
+      "step": 2223
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3319693094629157e-05,
+      "loss": 0.0052,
+      "step": 2224
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3312020460358055e-05,
+      "loss": 0.0095,
+      "step": 2225
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3304347826086957e-05,
+      "loss": 0.0053,
+      "step": 2226
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3296675191815857e-05,
+      "loss": 0.0099,
+      "step": 2227
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3289002557544757e-05,
+      "loss": 1.3879,
+      "step": 2228
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3281329923273658e-05,
+      "loss": 0.0041,
+      "step": 2229
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3273657289002558e-05,
+      "loss": 0.006,
+      "step": 2230
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.326598465473146e-05,
+      "loss": 0.006,
+      "step": 2231
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.3258312020460358e-05,
+      "loss": 0.012,
+      "step": 2232
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.325063938618926e-05,
+      "loss": 0.0107,
+      "step": 2233
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.324296675191816e-05,
+      "loss": 0.0061,
+      "step": 2234
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.323529411764706e-05,
+      "loss": 0.0041,
+      "step": 2235
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.322762148337596e-05,
+      "loss": 0.006,
+      "step": 2236
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.3219948849104861e-05,
+      "loss": 0.0064,
+      "step": 2237
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.3212276214833759e-05,
+      "loss": 0.006,
+      "step": 2238
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.320460358056266e-05,
+      "loss": 0.0035,
+      "step": 2239
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.319693094629156e-05,
+      "loss": 0.8375,
+      "step": 2240
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.318925831202046e-05,
+      "loss": 0.004,
+      "step": 2241
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.318158567774936e-05,
+      "loss": 0.0077,
+      "step": 2242
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.3173913043478262e-05,
+      "loss": 0.006,
+      "step": 2243
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.316624040920716e-05,
+      "loss": 0.0041,
+      "step": 2244
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3158567774936062e-05,
+      "loss": 0.0034,
+      "step": 2245
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3150895140664962e-05,
+      "loss": 0.8219,
+      "step": 2246
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3143222506393862e-05,
+      "loss": 0.004,
+      "step": 2247
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3135549872122762e-05,
+      "loss": 0.004,
+      "step": 2248
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3127877237851663e-05,
+      "loss": 1.1328,
+      "step": 2249
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3120204603580562e-05,
+      "loss": 0.0039,
+      "step": 2250
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3112531969309463e-05,
+      "loss": 0.0283,
+      "step": 2251
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3104859335038363e-05,
+      "loss": 0.0058,
+      "step": 2252
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3097186700767263e-05,
+      "loss": 0.004,
+      "step": 2253
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3089514066496165e-05,
+      "loss": 0.0057,
+      "step": 2254
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3081841432225064e-05,
+      "loss": 0.0076,
+      "step": 2255
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3074168797953966e-05,
+      "loss": 0.004,
+      "step": 2256
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3066496163682864e-05,
+      "loss": 0.0076,
+      "step": 2257
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3058823529411766e-05,
+      "loss": 0.0091,
+      "step": 2258
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3051150895140666e-05,
+      "loss": 0.0038,
+      "step": 2259
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3043478260869566e-05,
+      "loss": 0.0057,
+      "step": 2260
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3035805626598466e-05,
+      "loss": 0.0057,
+      "step": 2261
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3028132992327367e-05,
+      "loss": 1.1251,
+      "step": 2262
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3020460358056266e-05,
+      "loss": 1.1222,
+      "step": 2263
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.3012787723785167e-05,
+      "loss": 0.0047,
+      "step": 2264
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.3005115089514067e-05,
+      "loss": 0.008,
+      "step": 2265
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.2997442455242967e-05,
+      "loss": 0.0061,
+      "step": 2266
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.2989769820971867e-05,
+      "loss": 1.394,
+      "step": 2267
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2982097186700768e-05,
+      "loss": 0.0039,
+      "step": 2268
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2974424552429667e-05,
+      "loss": 0.0103,
+      "step": 2269
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2966751918158568e-05,
+      "loss": 1.1008,
+      "step": 2270
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2959079283887468e-05,
+      "loss": 0.0051,
+      "step": 2271
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2951406649616368e-05,
+      "loss": 0.004,
+      "step": 2272
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2943734015345268e-05,
+      "loss": 0.78,
+      "step": 2273
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.293606138107417e-05,
+      "loss": 0.0069,
+      "step": 2274
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2928388746803068e-05,
+      "loss": 0.0094,
+      "step": 2275
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.292071611253197e-05,
+      "loss": 0.0067,
+      "step": 2276
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.291304347826087e-05,
+      "loss": 0.0041,
+      "step": 2277
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.290537084398977e-05,
+      "loss": 0.0079,
+      "step": 2278
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.2897698209718671e-05,
+      "loss": 0.014,
+      "step": 2279
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.2890025575447571e-05,
+      "loss": 1.3775,
+      "step": 2280
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.288235294117647e-05,
+      "loss": 0.0042,
+      "step": 2281
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.287468030690537e-05,
+      "loss": 1.1862,
+      "step": 2282
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2867007672634272e-05,
+      "loss": 0.0042,
+      "step": 2283
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2859335038363172e-05,
+      "loss": 0.0098,
+      "step": 2284
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2851662404092072e-05,
+      "loss": 0.0038,
+      "step": 2285
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2843989769820972e-05,
+      "loss": 0.0127,
+      "step": 2286
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2836317135549874e-05,
+      "loss": 0.0072,
+      "step": 2287
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2828644501278772e-05,
+      "loss": 0.0098,
+      "step": 2288
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2820971867007674e-05,
+      "loss": 0.0069,
+      "step": 2289
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2813299232736573e-05,
+      "loss": 0.9529,
+      "step": 2290
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2805626598465473e-05,
+      "loss": 0.0096,
+      "step": 2291
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2797953964194373e-05,
+      "loss": 0.0094,
+      "step": 2292
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2790281329923275e-05,
+      "loss": 1.0639,
+      "step": 2293
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2782608695652173e-05,
+      "loss": 0.0043,
+      "step": 2294
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2774936061381075e-05,
+      "loss": 0.0098,
+      "step": 2295
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2767263427109975e-05,
+      "loss": 0.0084,
+      "step": 2296
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2759590792838875e-05,
+      "loss": 0.0071,
+      "step": 2297
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2751918158567774e-05,
+      "loss": 0.0036,
+      "step": 2298
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2744245524296676e-05,
+      "loss": 0.0097,
+      "step": 2299
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2736572890025574e-05,
+      "loss": 0.0072,
+      "step": 2300
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2728900255754476e-05,
+      "loss": 0.0043,
+      "step": 2301
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2721227621483376e-05,
+      "loss": 1.3742,
+      "step": 2302
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2713554987212276e-05,
+      "loss": 0.0086,
+      "step": 2303
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2705882352941177e-05,
+      "loss": 0.0071,
+      "step": 2304
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2698209718670077e-05,
+      "loss": 0.007,
+      "step": 2305
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2690537084398977e-05,
+      "loss": 0.0099,
+      "step": 2306
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2682864450127877e-05,
+      "loss": 1.0437,
+      "step": 2307
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2675191815856779e-05,
+      "loss": 0.01,
+      "step": 2308
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2667519181585679e-05,
+      "loss": 0.1771,
+      "step": 2309
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2659846547314579e-05,
+      "loss": 0.0103,
+      "step": 2310
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.2652173913043479e-05,
+      "loss": 0.0104,
+      "step": 2311
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.264450127877238e-05,
+      "loss": 0.0077,
+      "step": 2312
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.2636828644501278e-05,
+      "loss": 0.0107,
+      "step": 2313
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.262915601023018e-05,
+      "loss": 0.0107,
+      "step": 2314
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.262148337595908e-05,
+      "loss": 0.0135,
+      "step": 2315
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.261381074168798e-05,
+      "loss": 0.0103,
+      "step": 2316
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.260613810741688e-05,
+      "loss": 0.0136,
+      "step": 2317
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.2598465473145781e-05,
+      "loss": 0.0045,
+      "step": 2318
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.259079283887468e-05,
+      "loss": 0.0104,
+      "step": 2319
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2583120204603581e-05,
+      "loss": 0.0073,
+      "step": 2320
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2575447570332481e-05,
+      "loss": 0.9265,
+      "step": 2321
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2567774936061381e-05,
+      "loss": 0.0101,
+      "step": 2322
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2560102301790281e-05,
+      "loss": 1.3704,
+      "step": 2323
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2552429667519183e-05,
+      "loss": 1.415,
+      "step": 2324
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.254475703324808e-05,
+      "loss": 0.0098,
+      "step": 2325
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2537084398976982e-05,
+      "loss": 0.0138,
+      "step": 2326
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2529411764705884e-05,
+      "loss": 0.0064,
+      "step": 2327
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2521739130434782e-05,
+      "loss": 0.0093,
+      "step": 2328
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2514066496163684e-05,
+      "loss": 0.0119,
+      "step": 2329
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2506393861892584e-05,
+      "loss": 0.0097,
+      "step": 2330
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2498721227621484e-05,
+      "loss": 0.0049,
+      "step": 2331
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2491048593350384e-05,
+      "loss": 0.0044,
+      "step": 2332
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2483375959079285e-05,
+      "loss": 0.0067,
+      "step": 2333
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2475703324808183e-05,
+      "loss": 0.0117,
+      "step": 2334
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2468030690537085e-05,
+      "loss": 0.0066,
+      "step": 2335
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2460358056265985e-05,
+      "loss": 1.0804,
+      "step": 2336
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2452685421994887e-05,
+      "loss": 0.0065,
+      "step": 2337
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2445012787723785e-05,
+      "loss": 0.0069,
+      "step": 2338
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2437340153452686e-05,
+      "loss": 0.009,
+      "step": 2339
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2429667519181586e-05,
+      "loss": 0.0067,
+      "step": 2340
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2421994884910486e-05,
+      "loss": 0.0089,
+      "step": 2341
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2414322250639386e-05,
+      "loss": 0.011,
+      "step": 2342
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2406649616368288e-05,
+      "loss": 0.0034,
+      "step": 2343
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2398976982097186e-05,
+      "loss": 1.0842,
+      "step": 2344
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2391304347826088e-05,
+      "loss": 0.0066,
+      "step": 2345
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2383631713554987e-05,
+      "loss": 0.0203,
+      "step": 2346
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2375959079283887e-05,
+      "loss": 0.0046,
+      "step": 2347
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2368286445012787e-05,
+      "loss": 0.9714,
+      "step": 2348
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2360613810741689e-05,
+      "loss": 0.0066,
+      "step": 2349
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2352941176470587e-05,
+      "loss": 0.0091,
+      "step": 2350
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.2345268542199489e-05,
+      "loss": 0.0066,
+      "step": 2351
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.233759590792839e-05,
+      "loss": 0.0081,
+      "step": 2352
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.2329923273657289e-05,
+      "loss": 0.006,
+      "step": 2353
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.232225063938619e-05,
+      "loss": 0.0066,
+      "step": 2354
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.231457800511509e-05,
+      "loss": 0.0114,
+      "step": 2355
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.230690537084399e-05,
+      "loss": 2.7382,
+      "step": 2356
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.229923273657289e-05,
+      "loss": 0.0089,
+      "step": 2357
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.2291560102301792e-05,
+      "loss": 0.0042,
+      "step": 2358
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.228388746803069e-05,
+      "loss": 2.3698,
+      "step": 2359
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2276214833759591e-05,
+      "loss": 0.0065,
+      "step": 2360
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2268542199488491e-05,
+      "loss": 1.0813,
+      "step": 2361
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2260869565217393e-05,
+      "loss": 0.0066,
+      "step": 2362
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2253196930946291e-05,
+      "loss": 0.0092,
+      "step": 2363
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2245524296675193e-05,
+      "loss": 1.3566,
+      "step": 2364
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2237851662404093e-05,
+      "loss": 0.0092,
+      "step": 2365
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2230179028132993e-05,
+      "loss": 1.0739,
+      "step": 2366
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2222506393861893e-05,
+      "loss": 0.0122,
+      "step": 2367
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2214833759590794e-05,
+      "loss": 0.0096,
+      "step": 2368
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2207161125319692e-05,
+      "loss": 0.0094,
+      "step": 2369
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2199488491048594e-05,
+      "loss": 0.0097,
+      "step": 2370
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2191815856777494e-05,
+      "loss": 0.0071,
+      "step": 2371
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2184143222506394e-05,
+      "loss": 0.0085,
+      "step": 2372
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2176470588235294e-05,
+      "loss": 0.0071,
+      "step": 2373
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2168797953964195e-05,
+      "loss": 0.0074,
+      "step": 2374
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2161125319693094e-05,
+      "loss": 0.0084,
+      "step": 2375
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2153452685421995e-05,
+      "loss": 0.008,
+      "step": 2376
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.9267676767676768,
+      "eval_loss": 0.3489837944507599,
+      "eval_runtime": 58.1837,
+      "eval_samples_per_second": 6.806,
+      "eval_steps_per_second": 1.702,
+      "step": 2376
+    }
+  ],
+  "max_steps": 3960,
+  "num_train_epochs": 10,
+  "total_flos": 8.622885281472e+17,
+  "trial_name": null,
+  "trial_params": null
+}
diff --git a/checkpoint-2376/training_args.bin b/checkpoint-2376/training_args.bin
new file mode 100644
index 0000000000000000000000000000000000000000..32901f899c91066c808eae9dd019400bfdd17498
--- /dev/null
+++ b/checkpoint-2376/training_args.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db14439c369bfe9c89f64ff38f0b2d3fcc53911fc2c0b12a3a7d04656e7384a
+size 3503
diff --git a/checkpoint-2772/config.json b/checkpoint-2772/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..cb8d52e14263c50fd53dca42b5050e3235115c32
--- /dev/null
+++ b/checkpoint-2772/config.json
@@ -0,0 +1,113 @@
+{
+  "_name_or_path": "facebook/wav2vec2-base-960h",
+  "activation_dropout": 0.1,
+  "adapter_kernel_size": 3,
+  "adapter_stride": 2,
+  "add_adapter": false,
+  "apply_spec_augment": true,
+  "architectures": [
+    "Wav2Vec2ForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 1,
+  "classifier_proj_size": 256,
+  "codevector_dim": 256,
+  "contrastive_logits_temperature": 0.1,
+  "conv_bias": false,
+  "conv_dim": [
+    512,
+    512,
+    512,
+    512,
+    512,
+    512,
+    512
+  ],
+  "conv_kernel": [
+    10,
+    3,
+    3,
+    3,
+    3,
+    2,
+    2
+  ],
+  "conv_stride": [
+    5,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2
+  ],
+  "ctc_loss_reduction": "sum",
+  "ctc_zero_infinity": false,
+  "diversity_loss_weight": 0.1,
+  "do_stable_layer_norm": false,
+  "eos_token_id": 2,
+  "feat_extract_activation": "gelu",
+  "feat_extract_dropout": 0.0,
+  "feat_extract_norm": "group",
+  "feat_proj_dropout": 0.1,
+  "feat_quantizer_dropout": 0.0,
+  "final_dropout": 0.1,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout": 0.1,
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Foreign",
+    "1": "Native"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "layerdrop": 0.1,
+  "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
+  "mask_feature_prob": 0.0,
+  "mask_time_length": 10,
+  "mask_time_min_masks": 2,
+  "mask_time_prob": 0.05,
+  "model_type": "wav2vec2",
+  "num_adapter_layers": 3,
+  "num_attention_heads": 12,
+  "num_codevector_groups": 2,
+  "num_codevectors_per_group": 320,
+  "num_conv_pos_embedding_groups": 16,
+  "num_conv_pos_embeddings": 128,
+  "num_feat_extract_layers": 7,
+  "num_hidden_layers": 12,
+  "num_negatives": 100,
+  "output_hidden_size": 768,
+  "pad_token_id": 0,
+  "proj_codevector_dim": 256,
+  "tdnn_dilation": [
+    1,
+    2,
+    3,
+    1,
+    1
+  ],
+  "tdnn_dim": [
+    512,
+    512,
+    512,
+    512,
+    1500
+  ],
+  "tdnn_kernel": [
+    5,
+    3,
+    3,
+    1,
+    1
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.28.1",
+  "use_weighted_layer_sum": false,
+  "vocab_size": 32,
+  "xvector_output_dim": 512
+}
diff --git a/checkpoint-2772/optimizer.pt b/checkpoint-2772/optimizer.pt
new file mode 100644
index 0000000000000000000000000000000000000000..05580f0bcd7b17ed0110da9de03d476b6a4e4fa0
--- /dev/null
+++ b/checkpoint-2772/optimizer.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c264e7bf6fe789709415bc8d52c8cca90c135d95e166e43894a890afceea5740
+size 756677477
diff --git a/checkpoint-2772/preprocessor_config.json b/checkpoint-2772/preprocessor_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..a0b7227fc1d916e469b14f6c154ad6dfea1e6891
--- /dev/null
+++ b/checkpoint-2772/preprocessor_config.json
@@ -0,0 +1,9 @@
+{
+  "do_normalize": true,
+  "feature_extractor_type": "Wav2Vec2FeatureExtractor",
+  "feature_size": 1,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": false,
+  "sampling_rate": 16000
+}
diff --git a/checkpoint-2772/pytorch_model.bin b/checkpoint-2772/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..c0050c1d58dd9b5a0b7021c63d0dbccfe142b134
--- /dev/null
+++ b/checkpoint-2772/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:be25b1cf5abc0d6b168daf371d06f0d6bbfe5b96fd13fa28bbcb6bc9412f5a22
+size 378348563
diff --git a/checkpoint-2772/rng_state.pth b/checkpoint-2772/rng_state.pth
new file mode 100644
index 0000000000000000000000000000000000000000..61fc8c713f765b718535e4ef26780171cad7d4c6
--- /dev/null
+++ b/checkpoint-2772/rng_state.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:687640bdbf44b06437495d532a6ac86d76b3463b3987b834b64923f55e0620bc
+size 14567
diff --git a/checkpoint-2772/scheduler.pt b/checkpoint-2772/scheduler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..3e1d70cfc6b4b68924d3f239cf5b1b287a527799
--- /dev/null
+++ b/checkpoint-2772/scheduler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2de1862758096179f75546f36910749e35629f9cdd45e31ebee79dc077f88ab4
+size 623
diff --git a/checkpoint-2772/trainer_state.json b/checkpoint-2772/trainer_state.json
new file mode 100644
index 0000000000000000000000000000000000000000..a90a6784e210fe1485ba2a7182145b8b0ec912c0
--- /dev/null
+++ b/checkpoint-2772/trainer_state.json
@@ -0,0 +1,16711 @@
+{
+  "best_metric": 0.3489837944507599,
+  "best_model_checkpoint": "accent_classification/checkpoint-2376",
+  "epoch": 7.0,
+  "global_step": 2772,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.000000000000001e-07,
+      "loss": 0.6956,
+      "step": 1
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.2000000000000002e-06,
+      "loss": 0.6979,
+      "step": 2
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8e-06,
+      "loss": 0.7014,
+      "step": 3
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 2.4000000000000003e-06,
+      "loss": 0.6984,
+      "step": 4
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 3e-06,
+      "loss": 0.6978,
+      "step": 5
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 3.6e-06,
+      "loss": 0.6783,
+      "step": 6
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.2000000000000004e-06,
+      "loss": 0.6956,
+      "step": 7
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.800000000000001e-06,
+      "loss": 0.6994,
+      "step": 8
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 5.4e-06,
+      "loss": 0.6902,
+      "step": 9
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6e-06,
+      "loss": 0.6937,
+      "step": 10
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6.6e-06,
+      "loss": 0.6497,
+      "step": 11
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.2e-06,
+      "loss": 0.6911,
+      "step": 12
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.8e-06,
+      "loss": 0.6904,
+      "step": 13
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 8.400000000000001e-06,
+      "loss": 0.6854,
+      "step": 14
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9e-06,
+      "loss": 0.6843,
+      "step": 15
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9.600000000000001e-06,
+      "loss": 0.6851,
+      "step": 16
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.02e-05,
+      "loss": 0.6733,
+      "step": 17
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.08e-05,
+      "loss": 0.6819,
+      "step": 18
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.1400000000000001e-05,
+      "loss": 0.681,
+      "step": 19
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.6998,
+      "step": 20
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.26e-05,
+      "loss": 0.6743,
+      "step": 21
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.32e-05,
+      "loss": 0.6818,
+      "step": 22
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3800000000000002e-05,
+      "loss": 0.5331,
+      "step": 23
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.44e-05,
+      "loss": 0.6681,
+      "step": 24
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.6955,
+      "step": 25
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.56e-05,
+      "loss": 0.7,
+      "step": 26
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.62e-05,
+      "loss": 0.6975,
+      "step": 27
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.6800000000000002e-05,
+      "loss": 0.6249,
+      "step": 28
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.74e-05,
+      "loss": 0.6979,
+      "step": 29
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.6076,
+      "step": 30
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.86e-05,
+      "loss": 0.6091,
+      "step": 31
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9200000000000003e-05,
+      "loss": 0.6438,
+      "step": 32
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.98e-05,
+      "loss": 0.6624,
+      "step": 33
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.04e-05,
+      "loss": 0.6452,
+      "step": 34
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.6963,
+      "step": 35
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.16e-05,
+      "loss": 0.758,
+      "step": 36
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.22e-05,
+      "loss": 0.6471,
+      "step": 37
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.2800000000000002e-05,
+      "loss": 0.6313,
+      "step": 38
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.3400000000000003e-05,
+      "loss": 0.7019,
+      "step": 39
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.6336,
+      "step": 40
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4599999999999998e-05,
+      "loss": 0.7091,
+      "step": 41
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.52e-05,
+      "loss": 0.701,
+      "step": 42
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.58e-05,
+      "loss": 0.7036,
+      "step": 43
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.64e-05,
+      "loss": 0.7073,
+      "step": 44
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.614,
+      "step": 45
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.7600000000000003e-05,
+      "loss": 0.7543,
+      "step": 46
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.8199999999999998e-05,
+      "loss": 0.538,
+      "step": 47
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.88e-05,
+      "loss": 0.5179,
+      "step": 48
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.94e-05,
+      "loss": 0.6164,
+      "step": 49
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 3e-05,
+      "loss": 0.6093,
+      "step": 50
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.99923273657289e-05,
+      "loss": 0.6983,
+      "step": 51
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9984654731457804e-05,
+      "loss": 0.6069,
+      "step": 52
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9976982097186702e-05,
+      "loss": 0.5933,
+      "step": 53
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.99693094629156e-05,
+      "loss": 0.552,
+      "step": 54
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9961636828644502e-05,
+      "loss": 0.5865,
+      "step": 55
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9953964194373404e-05,
+      "loss": 0.7146,
+      "step": 56
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9946291560102302e-05,
+      "loss": 0.5591,
+      "step": 57
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9938618925831203e-05,
+      "loss": 0.7366,
+      "step": 58
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9930946291560102e-05,
+      "loss": 0.7006,
+      "step": 59
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9923273657289003e-05,
+      "loss": 0.4148,
+      "step": 60
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9915601023017905e-05,
+      "loss": 0.9938,
+      "step": 61
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9907928388746803e-05,
+      "loss": 0.911,
+      "step": 62
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.99002557544757e-05,
+      "loss": 0.5682,
+      "step": 63
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9892583120204606e-05,
+      "loss": 0.5833,
+      "step": 64
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9884910485933505e-05,
+      "loss": 0.7237,
+      "step": 65
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9877237851662403e-05,
+      "loss": 0.4013,
+      "step": 66
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9869565217391304e-05,
+      "loss": 0.9695,
+      "step": 67
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9861892583120206e-05,
+      "loss": 0.568,
+      "step": 68
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9854219948849104e-05,
+      "loss": 0.5697,
+      "step": 69
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9846547314578006e-05,
+      "loss": 0.7675,
+      "step": 70
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9838874680306904e-05,
+      "loss": 0.8831,
+      "step": 71
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9831202046035806e-05,
+      "loss": 0.7008,
+      "step": 72
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9823529411764707e-05,
+      "loss": 0.7085,
+      "step": 73
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9815856777493606e-05,
+      "loss": 0.5374,
+      "step": 74
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9808184143222507e-05,
+      "loss": 0.4131,
+      "step": 75
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.980051150895141e-05,
+      "loss": 0.7309,
+      "step": 76
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9792838874680307e-05,
+      "loss": 0.4279,
+      "step": 77
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.978516624040921e-05,
+      "loss": 0.5813,
+      "step": 78
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.977749360613811e-05,
+      "loss": 0.6276,
+      "step": 79
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976982097186701e-05,
+      "loss": 0.7892,
+      "step": 80
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976214833759591e-05,
+      "loss": 0.7134,
+      "step": 81
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.9754475703324808e-05,
+      "loss": 0.3682,
+      "step": 82
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.974680306905371e-05,
+      "loss": 0.5562,
+      "step": 83
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973913043478261e-05,
+      "loss": 0.7158,
+      "step": 84
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973145780051151e-05,
+      "loss": 0.524,
+      "step": 85
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9723785166240408e-05,
+      "loss": 0.3372,
+      "step": 86
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9716112531969313e-05,
+      "loss": 0.4979,
+      "step": 87
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970843989769821e-05,
+      "loss": 0.3177,
+      "step": 88
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970076726342711e-05,
+      "loss": 0.3488,
+      "step": 89
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.969309462915601e-05,
+      "loss": 0.5158,
+      "step": 90
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9685421994884913e-05,
+      "loss": 0.3208,
+      "step": 91
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.967774936061381e-05,
+      "loss": 0.6654,
+      "step": 92
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9670076726342712e-05,
+      "loss": 0.5581,
+      "step": 93
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.966240409207161e-05,
+      "loss": 0.7456,
+      "step": 94
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9654731457800512e-05,
+      "loss": 0.4437,
+      "step": 95
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9647058823529414e-05,
+      "loss": 0.5525,
+      "step": 96
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9639386189258312e-05,
+      "loss": 0.6673,
+      "step": 97
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.963171355498721e-05,
+      "loss": 0.435,
+      "step": 98
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9624040920716115e-05,
+      "loss": 0.7415,
+      "step": 99
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9616368286445014e-05,
+      "loss": 0.2981,
+      "step": 100
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9608695652173912e-05,
+      "loss": 0.5179,
+      "step": 101
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9601023017902817e-05,
+      "loss": 0.4005,
+      "step": 102
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9593350383631715e-05,
+      "loss": 0.4598,
+      "step": 103
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9585677749360613e-05,
+      "loss": 0.2553,
+      "step": 104
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9578005115089515e-05,
+      "loss": 0.4302,
+      "step": 105
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9570332480818416e-05,
+      "loss": 0.7894,
+      "step": 106
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9562659846547315e-05,
+      "loss": 0.213,
+      "step": 107
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9554987212276216e-05,
+      "loss": 0.5507,
+      "step": 108
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9547314578005115e-05,
+      "loss": 0.7676,
+      "step": 109
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9539641943734016e-05,
+      "loss": 0.3418,
+      "step": 110
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9531969309462918e-05,
+      "loss": 0.8533,
+      "step": 111
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9524296675191816e-05,
+      "loss": 0.3659,
+      "step": 112
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9516624040920714e-05,
+      "loss": 0.6482,
+      "step": 113
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.950895140664962e-05,
+      "loss": 0.5484,
+      "step": 114
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9501278772378517e-05,
+      "loss": 0.4508,
+      "step": 115
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9493606138107416e-05,
+      "loss": 0.4401,
+      "step": 116
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9485933503836317e-05,
+      "loss": 0.5455,
+      "step": 117
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.947826086956522e-05,
+      "loss": 0.722,
+      "step": 118
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9470588235294117e-05,
+      "loss": 0.2874,
+      "step": 119
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.946291560102302e-05,
+      "loss": 0.5973,
+      "step": 120
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.9455242966751917e-05,
+      "loss": 0.5898,
+      "step": 121
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.944757033248082e-05,
+      "loss": 1.0125,
+      "step": 122
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943989769820972e-05,
+      "loss": 0.2511,
+      "step": 123
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943222506393862e-05,
+      "loss": 0.545,
+      "step": 124
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.942455242966752e-05,
+      "loss": 0.593,
+      "step": 125
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.941687979539642e-05,
+      "loss": 0.2319,
+      "step": 126
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.940920716112532e-05,
+      "loss": 0.2215,
+      "step": 127
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.9401534526854218e-05,
+      "loss": 0.1964,
+      "step": 128
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9393861892583123e-05,
+      "loss": 0.1471,
+      "step": 129
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.938618925831202e-05,
+      "loss": 1.5965,
+      "step": 130
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9378516624040923e-05,
+      "loss": 1.0772,
+      "step": 131
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.937084398976982e-05,
+      "loss": 0.2005,
+      "step": 132
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9363171355498723e-05,
+      "loss": 0.9628,
+      "step": 133
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9355498721227624e-05,
+      "loss": 0.1503,
+      "step": 134
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9347826086956523e-05,
+      "loss": 0.6201,
+      "step": 135
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.934015345268542e-05,
+      "loss": 0.6094,
+      "step": 136
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9332480818414326e-05,
+      "loss": 1.385,
+      "step": 137
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9324808184143224e-05,
+      "loss": 0.1552,
+      "step": 138
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9317135549872122e-05,
+      "loss": 0.5828,
+      "step": 139
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9309462915601024e-05,
+      "loss": 0.6018,
+      "step": 140
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9301790281329925e-05,
+      "loss": 1.0614,
+      "step": 141
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9294117647058824e-05,
+      "loss": 0.566,
+      "step": 142
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9286445012787725e-05,
+      "loss": 0.5958,
+      "step": 143
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9278772378516624e-05,
+      "loss": 0.5838,
+      "step": 144
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9271099744245525e-05,
+      "loss": 0.5909,
+      "step": 145
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9263427109974427e-05,
+      "loss": 0.5862,
+      "step": 146
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9255754475703325e-05,
+      "loss": 0.2031,
+      "step": 147
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9248081841432223e-05,
+      "loss": 0.1995,
+      "step": 148
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9240409207161128e-05,
+      "loss": 0.24,
+      "step": 149
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9232736572890026e-05,
+      "loss": 0.5845,
+      "step": 150
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9225063938618925e-05,
+      "loss": 0.964,
+      "step": 151
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.921739130434783e-05,
+      "loss": 0.9595,
+      "step": 152
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9209718670076728e-05,
+      "loss": 0.2033,
+      "step": 153
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9202046035805626e-05,
+      "loss": 0.213,
+      "step": 154
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9194373401534528e-05,
+      "loss": 0.2042,
+      "step": 155
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.918670076726343e-05,
+      "loss": 0.5783,
+      "step": 156
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9179028132992328e-05,
+      "loss": 1.0822,
+      "step": 157
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.917135549872123e-05,
+      "loss": 1.2643,
+      "step": 158
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9163682864450127e-05,
+      "loss": 0.2593,
+      "step": 159
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.915601023017903e-05,
+      "loss": 0.5783,
+      "step": 160
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914833759590793e-05,
+      "loss": 0.5721,
+      "step": 161
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914066496163683e-05,
+      "loss": 0.5808,
+      "step": 162
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9132992327365727e-05,
+      "loss": 1.2578,
+      "step": 163
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9125319693094632e-05,
+      "loss": 0.5709,
+      "step": 164
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.911764705882353e-05,
+      "loss": 0.2159,
+      "step": 165
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910997442455243e-05,
+      "loss": 0.2329,
+      "step": 166
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910230179028133e-05,
+      "loss": 0.926,
+      "step": 167
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.9094629156010232e-05,
+      "loss": 0.9095,
+      "step": 168
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.908695652173913e-05,
+      "loss": 0.5622,
+      "step": 169
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907928388746803e-05,
+      "loss": 0.5703,
+      "step": 170
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907161125319693e-05,
+      "loss": 0.5669,
+      "step": 171
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.906393861892583e-05,
+      "loss": 0.2637,
+      "step": 172
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9056265984654733e-05,
+      "loss": 1.1077,
+      "step": 173
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.904859335038363e-05,
+      "loss": 0.5653,
+      "step": 174
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9040920716112533e-05,
+      "loss": 0.5584,
+      "step": 175
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9033248081841434e-05,
+      "loss": 0.5544,
+      "step": 176
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9025575447570333e-05,
+      "loss": 0.5663,
+      "step": 177
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.901790281329923e-05,
+      "loss": 1.0808,
+      "step": 178
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9010230179028136e-05,
+      "loss": 0.8736,
+      "step": 179
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9002557544757034e-05,
+      "loss": 0.6435,
+      "step": 180
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8994884910485932e-05,
+      "loss": 0.786,
+      "step": 181
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8987212276214834e-05,
+      "loss": 0.5717,
+      "step": 182
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8979539641943736e-05,
+      "loss": 0.5618,
+      "step": 183
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8971867007672637e-05,
+      "loss": 0.5785,
+      "step": 184
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8964194373401535e-05,
+      "loss": 0.7601,
+      "step": 185
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8956521739130434e-05,
+      "loss": 0.8112,
+      "step": 186
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.894884910485934e-05,
+      "loss": 0.5663,
+      "step": 187
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8941176470588237e-05,
+      "loss": 0.5631,
+      "step": 188
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8933503836317135e-05,
+      "loss": 0.7363,
+      "step": 189
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8925831202046037e-05,
+      "loss": 0.4552,
+      "step": 190
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8918158567774938e-05,
+      "loss": 0.4599,
+      "step": 191
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8910485933503836e-05,
+      "loss": 0.7848,
+      "step": 192
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8902813299232738e-05,
+      "loss": 0.5907,
+      "step": 193
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8895140664961636e-05,
+      "loss": 0.4264,
+      "step": 194
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8887468030690538e-05,
+      "loss": 0.5697,
+      "step": 195
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.887979539641944e-05,
+      "loss": 0.7643,
+      "step": 196
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8872122762148338e-05,
+      "loss": 0.3635,
+      "step": 197
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8864450127877236e-05,
+      "loss": 0.3715,
+      "step": 198
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.885677749360614e-05,
+      "loss": 0.554,
+      "step": 199
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.884910485933504e-05,
+      "loss": 0.8044,
+      "step": 200
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8841432225063937e-05,
+      "loss": 0.5673,
+      "step": 201
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8833759590792842e-05,
+      "loss": 0.8042,
+      "step": 202
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.882608695652174e-05,
+      "loss": 0.5638,
+      "step": 203
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881841432225064e-05,
+      "loss": 0.3584,
+      "step": 204
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881074168797954e-05,
+      "loss": 0.8211,
+      "step": 205
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.8803069053708442e-05,
+      "loss": 0.5708,
+      "step": 206
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.879539641943734e-05,
+      "loss": 0.8301,
+      "step": 207
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8787723785166242e-05,
+      "loss": 0.8229,
+      "step": 208
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.878005115089514e-05,
+      "loss": 0.5645,
+      "step": 209
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8772378516624042e-05,
+      "loss": 0.3529,
+      "step": 210
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8764705882352943e-05,
+      "loss": 0.5578,
+      "step": 211
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.875703324808184e-05,
+      "loss": 0.5659,
+      "step": 212
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.874936061381074e-05,
+      "loss": 0.7495,
+      "step": 213
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8741687979539645e-05,
+      "loss": 0.3162,
+      "step": 214
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8734015345268543e-05,
+      "loss": 0.5616,
+      "step": 215
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.872634271099744e-05,
+      "loss": 0.4034,
+      "step": 216
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8718670076726343e-05,
+      "loss": 0.3163,
+      "step": 217
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8710997442455245e-05,
+      "loss": 0.5514,
+      "step": 218
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8703324808184143e-05,
+      "loss": 0.826,
+      "step": 219
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8695652173913044e-05,
+      "loss": 0.8363,
+      "step": 220
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8687979539641943e-05,
+      "loss": 0.8288,
+      "step": 221
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8680306905370844e-05,
+      "loss": 0.5929,
+      "step": 222
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8672634271099746e-05,
+      "loss": 0.565,
+      "step": 223
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8664961636828644e-05,
+      "loss": 0.5642,
+      "step": 224
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8657289002557546e-05,
+      "loss": 0.5609,
+      "step": 225
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8649616368286447e-05,
+      "loss": 0.5641,
+      "step": 226
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8641943734015345e-05,
+      "loss": 0.8535,
+      "step": 227
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8634271099744244e-05,
+      "loss": 0.5652,
+      "step": 228
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.862659846547315e-05,
+      "loss": 0.4397,
+      "step": 229
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8618925831202047e-05,
+      "loss": 0.567,
+      "step": 230
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8611253196930945e-05,
+      "loss": 0.7517,
+      "step": 231
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8603580562659847e-05,
+      "loss": 0.5839,
+      "step": 232
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.859590792838875e-05,
+      "loss": 0.5677,
+      "step": 233
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8588235294117647e-05,
+      "loss": 0.5678,
+      "step": 234
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8580562659846548e-05,
+      "loss": 1.3045,
+      "step": 235
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8572890025575446e-05,
+      "loss": 0.804,
+      "step": 236
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.856521739130435e-05,
+      "loss": 1.0303,
+      "step": 237
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.855754475703325e-05,
+      "loss": 0.3303,
+      "step": 238
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8549872122762148e-05,
+      "loss": 0.5677,
+      "step": 239
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.854219948849105e-05,
+      "loss": 0.7972,
+      "step": 240
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.853452685421995e-05,
+      "loss": 0.5644,
+      "step": 241
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.852685421994885e-05,
+      "loss": 0.9294,
+      "step": 242
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.851918158567775e-05,
+      "loss": 0.5665,
+      "step": 243
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.851150895140665e-05,
+      "loss": 0.7963,
+      "step": 244
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.850383631713555e-05,
+      "loss": 0.5659,
+      "step": 245
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.8496163682864452e-05,
+      "loss": 0.5718,
+      "step": 246
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.848849104859335e-05,
+      "loss": 0.3526,
+      "step": 247
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8480818414322252e-05,
+      "loss": 0.574,
+      "step": 248
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8473145780051154e-05,
+      "loss": 0.9996,
+      "step": 249
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8465473145780052e-05,
+      "loss": 0.7634,
+      "step": 250
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.845780051150895e-05,
+      "loss": 0.5639,
+      "step": 251
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8450127877237855e-05,
+      "loss": 0.5737,
+      "step": 252
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8442455242966753e-05,
+      "loss": 0.7793,
+      "step": 253
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8434782608695652e-05,
+      "loss": 0.6769,
+      "step": 254
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8427109974424553e-05,
+      "loss": 0.5748,
+      "step": 255
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8419437340153455e-05,
+      "loss": 0.3602,
+      "step": 256
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8411764705882353e-05,
+      "loss": 0.5714,
+      "step": 257
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8404092071611255e-05,
+      "loss": 0.5738,
+      "step": 258
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8396419437340153e-05,
+      "loss": 0.5719,
+      "step": 259
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8388746803069055e-05,
+      "loss": 0.5753,
+      "step": 260
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8381074168797956e-05,
+      "loss": 0.5717,
+      "step": 261
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8373401534526854e-05,
+      "loss": 0.9794,
+      "step": 262
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8365728900255753e-05,
+      "loss": 0.5737,
+      "step": 263
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8358056265984658e-05,
+      "loss": 0.5723,
+      "step": 264
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8350383631713556e-05,
+      "loss": 0.5656,
+      "step": 265
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8342710997442454e-05,
+      "loss": 0.4259,
+      "step": 266
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8335038363171356e-05,
+      "loss": 0.3076,
+      "step": 267
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8327365728900257e-05,
+      "loss": 0.338,
+      "step": 268
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8319693094629156e-05,
+      "loss": 0.5612,
+      "step": 269
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8312020460358057e-05,
+      "loss": 0.5249,
+      "step": 270
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8304347826086955e-05,
+      "loss": 0.5671,
+      "step": 271
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8296675191815857e-05,
+      "loss": 0.5574,
+      "step": 272
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.828900255754476e-05,
+      "loss": 0.8152,
+      "step": 273
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8281329923273657e-05,
+      "loss": 0.5647,
+      "step": 274
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.827365728900256e-05,
+      "loss": 0.5632,
+      "step": 275
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.826598465473146e-05,
+      "loss": 0.3391,
+      "step": 276
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8258312020460358e-05,
+      "loss": 0.55,
+      "step": 277
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8250639386189257e-05,
+      "loss": 0.5672,
+      "step": 278
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.824296675191816e-05,
+      "loss": 0.285,
+      "step": 279
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.823529411764706e-05,
+      "loss": 0.2809,
+      "step": 280
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.8227621483375958e-05,
+      "loss": 0.5552,
+      "step": 281
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821994884910486e-05,
+      "loss": 0.563,
+      "step": 282
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821227621483376e-05,
+      "loss": 0.5628,
+      "step": 283
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.820460358056266e-05,
+      "loss": 0.2548,
+      "step": 284
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.819693094629156e-05,
+      "loss": 0.871,
+      "step": 285
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818925831202046e-05,
+      "loss": 0.8797,
+      "step": 286
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818158567774936e-05,
+      "loss": 0.2453,
+      "step": 287
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8173913043478262e-05,
+      "loss": 0.2464,
+      "step": 288
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.816624040920716e-05,
+      "loss": 0.3134,
+      "step": 289
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8158567774936062e-05,
+      "loss": 0.2431,
+      "step": 290
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8150895140664964e-05,
+      "loss": 0.2308,
+      "step": 291
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8143222506393862e-05,
+      "loss": 0.2173,
+      "step": 292
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8135549872122764e-05,
+      "loss": 0.5665,
+      "step": 293
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8127877237851662e-05,
+      "loss": 0.5592,
+      "step": 294
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8120204603580564e-05,
+      "loss": 0.473,
+      "step": 295
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8112531969309465e-05,
+      "loss": 0.9458,
+      "step": 296
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8104859335038363e-05,
+      "loss": 0.1879,
+      "step": 297
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8097186700767265e-05,
+      "loss": 0.6797,
+      "step": 298
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8089514066496167e-05,
+      "loss": 0.7592,
+      "step": 299
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8081841432225065e-05,
+      "loss": 0.4109,
+      "step": 300
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8074168797953963e-05,
+      "loss": 0.5821,
+      "step": 301
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8066496163682868e-05,
+      "loss": 0.2057,
+      "step": 302
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8058823529411766e-05,
+      "loss": 0.5322,
+      "step": 303
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8051150895140665e-05,
+      "loss": 0.5208,
+      "step": 304
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8043478260869566e-05,
+      "loss": 0.6799,
+      "step": 305
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8035805626598468e-05,
+      "loss": 0.5894,
+      "step": 306
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8028132992327366e-05,
+      "loss": 0.4051,
+      "step": 307
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8020460358056268e-05,
+      "loss": 0.3695,
+      "step": 308
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8012787723785166e-05,
+      "loss": 0.4869,
+      "step": 309
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8005115089514067e-05,
+      "loss": 0.6075,
+      "step": 310
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.799744245524297e-05,
+      "loss": 0.4245,
+      "step": 311
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7989769820971867e-05,
+      "loss": 0.5122,
+      "step": 312
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7982097186700765e-05,
+      "loss": 0.452,
+      "step": 313
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.797442455242967e-05,
+      "loss": 0.9568,
+      "step": 314
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.796675191815857e-05,
+      "loss": 0.5846,
+      "step": 315
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.7959079283887467e-05,
+      "loss": 0.3743,
+      "step": 316
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.795140664961637e-05,
+      "loss": 0.1922,
+      "step": 317
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.794373401534527e-05,
+      "loss": 0.5303,
+      "step": 318
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.793606138107417e-05,
+      "loss": 0.7253,
+      "step": 319
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.792838874680307e-05,
+      "loss": 0.2794,
+      "step": 320
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.7920716112531968e-05,
+      "loss": 0.577,
+      "step": 321
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.791304347826087e-05,
+      "loss": 0.614,
+      "step": 322
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.790537084398977e-05,
+      "loss": 0.5882,
+      "step": 323
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789769820971867e-05,
+      "loss": 0.2503,
+      "step": 324
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789002557544757e-05,
+      "loss": 0.4195,
+      "step": 325
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.7882352941176473e-05,
+      "loss": 0.2959,
+      "step": 326
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.787468030690537e-05,
+      "loss": 0.3732,
+      "step": 327
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.786700767263427e-05,
+      "loss": 0.5729,
+      "step": 328
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7859335038363174e-05,
+      "loss": 0.3179,
+      "step": 329
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7851662404092073e-05,
+      "loss": 0.6823,
+      "step": 330
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.784398976982097e-05,
+      "loss": 0.2663,
+      "step": 331
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7836317135549872e-05,
+      "loss": 0.4436,
+      "step": 332
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7828644501278774e-05,
+      "loss": 0.3089,
+      "step": 333
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7820971867007672e-05,
+      "loss": 0.5888,
+      "step": 334
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7813299232736574e-05,
+      "loss": 0.2115,
+      "step": 335
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7805626598465472e-05,
+      "loss": 0.4005,
+      "step": 336
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7797953964194374e-05,
+      "loss": 0.3097,
+      "step": 337
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7790281329923275e-05,
+      "loss": 0.5847,
+      "step": 338
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7782608695652174e-05,
+      "loss": 1.0164,
+      "step": 339
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7774936061381072e-05,
+      "loss": 1.1321,
+      "step": 340
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7767263427109977e-05,
+      "loss": 0.592,
+      "step": 341
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7759590792838875e-05,
+      "loss": 1.0191,
+      "step": 342
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7751918158567777e-05,
+      "loss": 0.9199,
+      "step": 343
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7744245524296675e-05,
+      "loss": 0.2874,
+      "step": 344
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7736572890025576e-05,
+      "loss": 0.2469,
+      "step": 345
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7728900255754478e-05,
+      "loss": 0.8364,
+      "step": 346
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7721227621483376e-05,
+      "loss": 0.7365,
+      "step": 347
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7713554987212278e-05,
+      "loss": 0.5595,
+      "step": 348
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.770588235294118e-05,
+      "loss": 0.2546,
+      "step": 349
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7698209718670078e-05,
+      "loss": 0.8183,
+      "step": 350
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7690537084398976e-05,
+      "loss": 0.5385,
+      "step": 351
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.768286445012788e-05,
+      "loss": 0.2763,
+      "step": 352
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.767519181585678e-05,
+      "loss": 0.3045,
+      "step": 353
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7667519181585677e-05,
+      "loss": 0.3354,
+      "step": 354
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765984654731458e-05,
+      "loss": 0.9366,
+      "step": 355
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765217391304348e-05,
+      "loss": 0.6212,
+      "step": 356
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.764450127877238e-05,
+      "loss": 0.1161,
+      "step": 357
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.763682864450128e-05,
+      "loss": 0.4929,
+      "step": 358
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762915601023018e-05,
+      "loss": 0.8167,
+      "step": 359
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762148337595908e-05,
+      "loss": 0.2521,
+      "step": 360
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.7613810741687982e-05,
+      "loss": 0.4174,
+      "step": 361
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.760613810741688e-05,
+      "loss": 0.3822,
+      "step": 362
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.759846547314578e-05,
+      "loss": 0.5355,
+      "step": 363
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.7590792838874683e-05,
+      "loss": 0.561,
+      "step": 364
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.758312020460358e-05,
+      "loss": 0.5126,
+      "step": 365
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.757544757033248e-05,
+      "loss": 0.2301,
+      "step": 366
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.756777493606138e-05,
+      "loss": 0.1698,
+      "step": 367
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7560102301790283e-05,
+      "loss": 0.1634,
+      "step": 368
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.755242966751918e-05,
+      "loss": 0.5955,
+      "step": 369
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7544757033248083e-05,
+      "loss": 0.1568,
+      "step": 370
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.753708439897698e-05,
+      "loss": 0.5167,
+      "step": 371
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7529411764705883e-05,
+      "loss": 1.0028,
+      "step": 372
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7521739130434784e-05,
+      "loss": 1.2567,
+      "step": 373
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7514066496163682e-05,
+      "loss": 0.6094,
+      "step": 374
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7506393861892584e-05,
+      "loss": 0.6059,
+      "step": 375
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7498721227621486e-05,
+      "loss": 1.0593,
+      "step": 376
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7491048593350384e-05,
+      "loss": 0.1506,
+      "step": 377
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7483375959079282e-05,
+      "loss": 1.5106,
+      "step": 378
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7475703324808187e-05,
+      "loss": 0.1827,
+      "step": 379
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7468030690537085e-05,
+      "loss": 1.4961,
+      "step": 380
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7460358056265984e-05,
+      "loss": 1.0456,
+      "step": 381
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7452685421994885e-05,
+      "loss": 1.2063,
+      "step": 382
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7445012787723787e-05,
+      "loss": 0.5837,
+      "step": 383
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7437340153452685e-05,
+      "loss": 1.4536,
+      "step": 384
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7429667519181587e-05,
+      "loss": 0.5803,
+      "step": 385
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7421994884910485e-05,
+      "loss": 0.5896,
+      "step": 386
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7414322250639386e-05,
+      "loss": 0.1822,
+      "step": 387
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7406649616368288e-05,
+      "loss": 1.7893,
+      "step": 388
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7398976982097186e-05,
+      "loss": 0.9723,
+      "step": 389
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7391304347826085e-05,
+      "loss": 1.2815,
+      "step": 390
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.738363171355499e-05,
+      "loss": 0.5765,
+      "step": 391
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7375959079283888e-05,
+      "loss": 1.2429,
+      "step": 392
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7368286445012786e-05,
+      "loss": 0.1343,
+      "step": 393
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7360613810741688e-05,
+      "loss": 0.2306,
+      "step": 394
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.735294117647059e-05,
+      "loss": 0.624,
+      "step": 395
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.734526854219949e-05,
+      "loss": 0.2371,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7348484848484849,
+      "eval_loss": 0.5804409980773926,
+      "eval_runtime": 57.3336,
+      "eval_samples_per_second": 6.907,
+      "eval_steps_per_second": 1.727,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.733759590792839e-05,
+      "loss": 0.8989,
+      "step": 397
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.732992327365729e-05,
+      "loss": 0.5664,
+      "step": 398
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.7322250639386192e-05,
+      "loss": 0.8852,
+      "step": 399
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.731457800511509e-05,
+      "loss": 0.5661,
+      "step": 400
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.730690537084399e-05,
+      "loss": 0.5648,
+      "step": 401
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7299232736572894e-05,
+      "loss": 0.7333,
+      "step": 402
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7291560102301792e-05,
+      "loss": 0.3037,
+      "step": 403
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.728388746803069e-05,
+      "loss": 0.1463,
+      "step": 404
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7276214833759592e-05,
+      "loss": 0.2702,
+      "step": 405
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7268542199488493e-05,
+      "loss": 0.2693,
+      "step": 406
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.726086956521739e-05,
+      "loss": 0.5647,
+      "step": 407
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7253196930946293e-05,
+      "loss": 0.8507,
+      "step": 408
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.724552429667519e-05,
+      "loss": 0.8645,
+      "step": 409
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7237851662404093e-05,
+      "loss": 0.2649,
+      "step": 410
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7230179028132995e-05,
+      "loss": 0.6942,
+      "step": 411
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7222506393861893e-05,
+      "loss": 0.5656,
+      "step": 412
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.721483375959079e-05,
+      "loss": 0.5489,
+      "step": 413
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7207161125319696e-05,
+      "loss": 0.2652,
+      "step": 414
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7199488491048594e-05,
+      "loss": 0.8676,
+      "step": 415
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7191815856777493e-05,
+      "loss": 0.8621,
+      "step": 416
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7184143222506394e-05,
+      "loss": 0.2641,
+      "step": 417
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7176470588235296e-05,
+      "loss": 0.859,
+      "step": 418
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7168797953964194e-05,
+      "loss": 0.6492,
+      "step": 419
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7161125319693096e-05,
+      "loss": 1.1494,
+      "step": 420
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7153452685421994e-05,
+      "loss": 0.8278,
+      "step": 421
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7145780051150895e-05,
+      "loss": 0.764,
+      "step": 422
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7138107416879797e-05,
+      "loss": 0.564,
+      "step": 423
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7130434782608695e-05,
+      "loss": 0.8051,
+      "step": 424
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7122762148337597e-05,
+      "loss": 0.3052,
+      "step": 425
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.71150895140665e-05,
+      "loss": 0.2945,
+      "step": 426
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7107416879795397e-05,
+      "loss": 0.4118,
+      "step": 427
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7099744245524295e-05,
+      "loss": 0.559,
+      "step": 428
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.70920716112532e-05,
+      "loss": 0.5593,
+      "step": 429
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7084398976982098e-05,
+      "loss": 0.564,
+      "step": 430
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7076726342710996e-05,
+      "loss": 0.4454,
+      "step": 431
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7069053708439898e-05,
+      "loss": 0.4381,
+      "step": 432
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.70613810741688e-05,
+      "loss": 0.3718,
+      "step": 433
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7053708439897698e-05,
+      "loss": 0.6627,
+      "step": 434
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70460358056266e-05,
+      "loss": 0.4004,
+      "step": 435
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7038363171355498e-05,
+      "loss": 0.3855,
+      "step": 436
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70306905370844e-05,
+      "loss": 0.4799,
+      "step": 437
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70230179028133e-05,
+      "loss": 0.6714,
+      "step": 438
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70153452685422e-05,
+      "loss": 0.622,
+      "step": 439
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7007672634271097e-05,
+      "loss": 0.3916,
+      "step": 440
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.5487,
+      "step": 441
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69923273657289e-05,
+      "loss": 0.376,
+      "step": 442
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69846547314578e-05,
+      "loss": 0.3616,
+      "step": 443
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69769820971867e-05,
+      "loss": 0.8747,
+      "step": 444
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.6969309462915602e-05,
+      "loss": 0.283,
+      "step": 445
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.69616368286445e-05,
+      "loss": 0.3988,
+      "step": 446
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6953964194373402e-05,
+      "loss": 0.4569,
+      "step": 447
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6946291560102303e-05,
+      "loss": 0.2005,
+      "step": 448
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6938618925831205e-05,
+      "loss": 0.3087,
+      "step": 449
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6930946291560103e-05,
+      "loss": 0.5814,
+      "step": 450
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6923273657289e-05,
+      "loss": 0.2995,
+      "step": 451
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6915601023017907e-05,
+      "loss": 0.7806,
+      "step": 452
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6907928388746805e-05,
+      "loss": 0.2666,
+      "step": 453
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6900255754475703e-05,
+      "loss": 0.2571,
+      "step": 454
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6892583120204605e-05,
+      "loss": 0.5931,
+      "step": 455
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6884910485933506e-05,
+      "loss": 0.475,
+      "step": 456
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6877237851662404e-05,
+      "loss": 0.2372,
+      "step": 457
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6869565217391306e-05,
+      "loss": 0.1577,
+      "step": 458
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6861892583120204e-05,
+      "loss": 0.1953,
+      "step": 459
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6854219948849106e-05,
+      "loss": 0.1931,
+      "step": 460
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6846547314578008e-05,
+      "loss": 0.1829,
+      "step": 461
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6838874680306906e-05,
+      "loss": 0.1601,
+      "step": 462
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6831202046035804e-05,
+      "loss": 0.2505,
+      "step": 463
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.682352941176471e-05,
+      "loss": 0.181,
+      "step": 464
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6815856777493607e-05,
+      "loss": 0.5978,
+      "step": 465
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6808184143222505e-05,
+      "loss": 0.5458,
+      "step": 466
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6800511508951407e-05,
+      "loss": 0.1289,
+      "step": 467
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.679283887468031e-05,
+      "loss": 0.4383,
+      "step": 468
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6785166240409207e-05,
+      "loss": 0.5892,
+      "step": 469
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.677749360613811e-05,
+      "loss": 1.3565,
+      "step": 470
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.676982097186701e-05,
+      "loss": 0.7267,
+      "step": 471
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.6762148337595908e-05,
+      "loss": 0.1354,
+      "step": 472
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.675447570332481e-05,
+      "loss": 0.9291,
+      "step": 473
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.6746803069053708e-05,
+      "loss": 0.5265,
+      "step": 474
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673913043478261e-05,
+      "loss": 0.4841,
+      "step": 475
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673145780051151e-05,
+      "loss": 1.2391,
+      "step": 476
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.672378516624041e-05,
+      "loss": 0.4924,
+      "step": 477
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6716112531969308e-05,
+      "loss": 1.0076,
+      "step": 478
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6708439897698213e-05,
+      "loss": 0.6314,
+      "step": 479
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.670076726342711e-05,
+      "loss": 0.4237,
+      "step": 480
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.669309462915601e-05,
+      "loss": 0.1077,
+      "step": 481
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.668542199488491e-05,
+      "loss": 0.3038,
+      "step": 482
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6677749360613812e-05,
+      "loss": 0.6081,
+      "step": 483
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.667007672634271e-05,
+      "loss": 0.1078,
+      "step": 484
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6662404092071612e-05,
+      "loss": 0.4296,
+      "step": 485
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.665473145780051e-05,
+      "loss": 0.3981,
+      "step": 486
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6647058823529412e-05,
+      "loss": 0.5853,
+      "step": 487
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6639386189258314e-05,
+      "loss": 0.802,
+      "step": 488
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6631713554987212e-05,
+      "loss": 0.4576,
+      "step": 489
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.662404092071611e-05,
+      "loss": 0.2682,
+      "step": 490
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6616368286445015e-05,
+      "loss": 0.1991,
+      "step": 491
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6608695652173913e-05,
+      "loss": 0.1409,
+      "step": 492
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.660102301790281e-05,
+      "loss": 0.3869,
+      "step": 493
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6593350383631713e-05,
+      "loss": 0.5301,
+      "step": 494
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6585677749360615e-05,
+      "loss": 0.8116,
+      "step": 495
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6578005115089513e-05,
+      "loss": 0.0984,
+      "step": 496
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6570332480818415e-05,
+      "loss": 0.7773,
+      "step": 497
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6562659846547316e-05,
+      "loss": 1.3522,
+      "step": 498
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6554987212276215e-05,
+      "loss": 0.0862,
+      "step": 499
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6547314578005116e-05,
+      "loss": 0.0894,
+      "step": 500
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6539641943734014e-05,
+      "loss": 0.0885,
+      "step": 501
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.653196930946292e-05,
+      "loss": 0.3031,
+      "step": 502
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6524296675191818e-05,
+      "loss": 0.187,
+      "step": 503
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6516624040920716e-05,
+      "loss": 1.3063,
+      "step": 504
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6508951406649617e-05,
+      "loss": 0.6847,
+      "step": 505
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.650127877237852e-05,
+      "loss": 0.078,
+      "step": 506
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6493606138107417e-05,
+      "loss": 0.1265,
+      "step": 507
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.648593350383632e-05,
+      "loss": 0.0804,
+      "step": 508
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.6478260869565217e-05,
+      "loss": 0.6976,
+      "step": 509
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.647058823529412e-05,
+      "loss": 0.7977,
+      "step": 510
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.646291560102302e-05,
+      "loss": 0.413,
+      "step": 511
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.645524296675192e-05,
+      "loss": 0.6944,
+      "step": 512
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6447570332480817e-05,
+      "loss": 0.7177,
+      "step": 513
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6439897698209722e-05,
+      "loss": 0.0581,
+      "step": 514
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.643222506393862e-05,
+      "loss": 0.9308,
+      "step": 515
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6424552429667518e-05,
+      "loss": 0.2518,
+      "step": 516
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.641687979539642e-05,
+      "loss": 0.1014,
+      "step": 517
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640920716112532e-05,
+      "loss": 0.1641,
+      "step": 518
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640153452685422e-05,
+      "loss": 0.101,
+      "step": 519
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.639386189258312e-05,
+      "loss": 0.1097,
+      "step": 520
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6386189258312023e-05,
+      "loss": 0.987,
+      "step": 521
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.637851662404092e-05,
+      "loss": 0.6465,
+      "step": 522
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6370843989769823e-05,
+      "loss": 0.0914,
+      "step": 523
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.636317135549872e-05,
+      "loss": 0.139,
+      "step": 524
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6355498721227623e-05,
+      "loss": 0.6455,
+      "step": 525
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6347826086956524e-05,
+      "loss": 0.204,
+      "step": 526
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6340153452685422e-05,
+      "loss": 0.1269,
+      "step": 527
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.633248081841432e-05,
+      "loss": 0.2214,
+      "step": 528
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6324808184143226e-05,
+      "loss": 0.0736,
+      "step": 529
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6317135549872124e-05,
+      "loss": 0.0714,
+      "step": 530
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6309462915601022e-05,
+      "loss": 0.1388,
+      "step": 531
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6301790281329924e-05,
+      "loss": 0.7758,
+      "step": 532
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6294117647058825e-05,
+      "loss": 0.3388,
+      "step": 533
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6286445012787724e-05,
+      "loss": 1.1831,
+      "step": 534
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6278772378516625e-05,
+      "loss": 0.1166,
+      "step": 535
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6271099744245523e-05,
+      "loss": 0.2389,
+      "step": 536
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6263427109974425e-05,
+      "loss": 0.1622,
+      "step": 537
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6255754475703327e-05,
+      "loss": 0.1015,
+      "step": 538
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6248081841432225e-05,
+      "loss": 0.0658,
+      "step": 539
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6240409207161123e-05,
+      "loss": 0.0954,
+      "step": 540
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6232736572890028e-05,
+      "loss": 0.1182,
+      "step": 541
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6225063938618926e-05,
+      "loss": 0.0605,
+      "step": 542
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6217391304347824e-05,
+      "loss": 0.9265,
+      "step": 543
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6209718670076726e-05,
+      "loss": 1.2994,
+      "step": 544
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6202046035805628e-05,
+      "loss": 0.081,
+      "step": 545
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6194373401534526e-05,
+      "loss": 0.5962,
+      "step": 546
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6186700767263428e-05,
+      "loss": 0.0665,
+      "step": 547
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.617902813299233e-05,
+      "loss": 0.0655,
+      "step": 548
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6171355498721227e-05,
+      "loss": 0.0642,
+      "step": 549
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.616368286445013e-05,
+      "loss": 0.0834,
+      "step": 550
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6156010230179027e-05,
+      "loss": 0.047,
+      "step": 551
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6148337595907932e-05,
+      "loss": 0.0642,
+      "step": 552
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.614066496163683e-05,
+      "loss": 0.0594,
+      "step": 553
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.613299232736573e-05,
+      "loss": 0.0573,
+      "step": 554
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.612531969309463e-05,
+      "loss": 0.1101,
+      "step": 555
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.6117647058823532e-05,
+      "loss": 0.5121,
+      "step": 556
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.610997442455243e-05,
+      "loss": 0.0374,
+      "step": 557
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.6102301790281332e-05,
+      "loss": 0.0957,
+      "step": 558
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.609462915601023e-05,
+      "loss": 0.0488,
+      "step": 559
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.608695652173913e-05,
+      "loss": 0.0482,
+      "step": 560
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6079283887468033e-05,
+      "loss": 1.2041,
+      "step": 561
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.607161125319693e-05,
+      "loss": 0.8833,
+      "step": 562
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.606393861892583e-05,
+      "loss": 0.1544,
+      "step": 563
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6056265984654735e-05,
+      "loss": 0.046,
+      "step": 564
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6048593350383633e-05,
+      "loss": 0.9073,
+      "step": 565
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.604092071611253e-05,
+      "loss": 0.052,
+      "step": 566
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6033248081841433e-05,
+      "loss": 0.0308,
+      "step": 567
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6025575447570334e-05,
+      "loss": 0.7032,
+      "step": 568
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6017902813299232e-05,
+      "loss": 0.6947,
+      "step": 569
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6010230179028134e-05,
+      "loss": 1.5603,
+      "step": 570
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6002557544757036e-05,
+      "loss": 0.8836,
+      "step": 571
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.5994884910485934e-05,
+      "loss": 1.7439,
+      "step": 572
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5987212276214836e-05,
+      "loss": 0.0313,
+      "step": 573
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5979539641943734e-05,
+      "loss": 0.0317,
+      "step": 574
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5971867007672635e-05,
+      "loss": 0.0325,
+      "step": 575
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5964194373401537e-05,
+      "loss": 0.5,
+      "step": 576
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5956521739130435e-05,
+      "loss": 0.0675,
+      "step": 577
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5948849104859333e-05,
+      "loss": 0.6364,
+      "step": 578
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.594117647058824e-05,
+      "loss": 0.0696,
+      "step": 579
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5933503836317137e-05,
+      "loss": 0.6195,
+      "step": 580
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5925831202046035e-05,
+      "loss": 0.5796,
+      "step": 581
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5918158567774937e-05,
+      "loss": 0.0832,
+      "step": 582
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5910485933503838e-05,
+      "loss": 0.0794,
+      "step": 583
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5902813299232736e-05,
+      "loss": 0.5851,
+      "step": 584
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5895140664961638e-05,
+      "loss": 0.104,
+      "step": 585
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5887468030690536e-05,
+      "loss": 1.0492,
+      "step": 586
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5879795396419438e-05,
+      "loss": 0.0645,
+      "step": 587
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.587212276214834e-05,
+      "loss": 0.514,
+      "step": 588
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5864450127877238e-05,
+      "loss": 0.0701,
+      "step": 589
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5856777493606136e-05,
+      "loss": 0.5409,
+      "step": 590
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584910485933504e-05,
+      "loss": 0.0536,
+      "step": 591
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584143222506394e-05,
+      "loss": 0.1728,
+      "step": 592
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.5833759590792837e-05,
+      "loss": 0.0845,
+      "step": 593
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.582608695652174e-05,
+      "loss": 0.2217,
+      "step": 594
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.581841432225064e-05,
+      "loss": 0.1397,
+      "step": 595
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.581074168797954e-05,
+      "loss": 0.4119,
+      "step": 596
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.580306905370844e-05,
+      "loss": 0.3638,
+      "step": 597
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.5795396419437342e-05,
+      "loss": 0.0781,
+      "step": 598
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.578772378516624e-05,
+      "loss": 0.5926,
+      "step": 599
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5780051150895142e-05,
+      "loss": 0.1378,
+      "step": 600
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.577237851662404e-05,
+      "loss": 0.142,
+      "step": 601
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.576470588235294e-05,
+      "loss": 0.1722,
+      "step": 602
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5757033248081843e-05,
+      "loss": 0.1529,
+      "step": 603
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.574936061381074e-05,
+      "loss": 1.6083,
+      "step": 604
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5741687979539643e-05,
+      "loss": 1.8267,
+      "step": 605
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5734015345268545e-05,
+      "loss": 0.0535,
+      "step": 606
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5726342710997443e-05,
+      "loss": 0.9945,
+      "step": 607
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5718670076726345e-05,
+      "loss": 0.9133,
+      "step": 608
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5710997442455243e-05,
+      "loss": 0.5295,
+      "step": 609
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5703324808184144e-05,
+      "loss": 0.058,
+      "step": 610
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5695652173913046e-05,
+      "loss": 0.4789,
+      "step": 611
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5687979539641944e-05,
+      "loss": 0.0572,
+      "step": 612
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5680306905370842e-05,
+      "loss": 0.0897,
+      "step": 613
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5672634271099747e-05,
+      "loss": 0.056,
+      "step": 614
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5664961636828646e-05,
+      "loss": 0.0278,
+      "step": 615
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5657289002557544e-05,
+      "loss": 0.9947,
+      "step": 616
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5649616368286445e-05,
+      "loss": 0.0627,
+      "step": 617
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5641943734015347e-05,
+      "loss": 0.9335,
+      "step": 618
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5634271099744245e-05,
+      "loss": 0.9793,
+      "step": 619
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5626598465473147e-05,
+      "loss": 1.8128,
+      "step": 620
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.561892583120205e-05,
+      "loss": 1.8015,
+      "step": 621
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5611253196930947e-05,
+      "loss": 0.9259,
+      "step": 622
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.560358056265985e-05,
+      "loss": 0.9618,
+      "step": 623
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5595907928388747e-05,
+      "loss": 0.0808,
+      "step": 624
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5588235294117648e-05,
+      "loss": 1.7161,
+      "step": 625
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.558056265984655e-05,
+      "loss": 0.8879,
+      "step": 626
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5572890025575448e-05,
+      "loss": 0.043,
+      "step": 627
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5565217391304346e-05,
+      "loss": 0.9449,
+      "step": 628
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.555754475703325e-05,
+      "loss": 0.0624,
+      "step": 629
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.554987212276215e-05,
+      "loss": 0.0924,
+      "step": 630
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5542199488491048e-05,
+      "loss": 0.0715,
+      "step": 631
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.553452685421995e-05,
+      "loss": 0.4995,
+      "step": 632
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.552685421994885e-05,
+      "loss": 0.5576,
+      "step": 633
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551918158567775e-05,
+      "loss": 0.06,
+      "step": 634
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551150895140665e-05,
+      "loss": 0.0873,
+      "step": 635
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.550383631713555e-05,
+      "loss": 0.6868,
+      "step": 636
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.549616368286445e-05,
+      "loss": 0.8487,
+      "step": 637
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.5488491048593352e-05,
+      "loss": 0.5641,
+      "step": 638
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.548081841432225e-05,
+      "loss": 1.1343,
+      "step": 639
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.547314578005115e-05,
+      "loss": 0.0614,
+      "step": 640
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5465473145780054e-05,
+      "loss": 0.9414,
+      "step": 641
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5457800511508952e-05,
+      "loss": 0.9226,
+      "step": 642
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.545012787723785e-05,
+      "loss": 0.3323,
+      "step": 643
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5442455242966752e-05,
+      "loss": 0.9315,
+      "step": 644
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5434782608695653e-05,
+      "loss": 0.5539,
+      "step": 645
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.542710997442455e-05,
+      "loss": 0.3376,
+      "step": 646
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5419437340153453e-05,
+      "loss": 0.0694,
+      "step": 647
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5411764705882355e-05,
+      "loss": 0.0706,
+      "step": 648
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5404092071611253e-05,
+      "loss": 0.105,
+      "step": 649
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5396419437340155e-05,
+      "loss": 0.912,
+      "step": 650
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5388746803069053e-05,
+      "loss": 0.0998,
+      "step": 651
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5381074168797954e-05,
+      "loss": 0.2247,
+      "step": 652
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5373401534526856e-05,
+      "loss": 0.2869,
+      "step": 653
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5365728900255754e-05,
+      "loss": 0.532,
+      "step": 654
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5358056265984653e-05,
+      "loss": 0.4723,
+      "step": 655
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5350383631713558e-05,
+      "loss": 0.1178,
+      "step": 656
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5342710997442456e-05,
+      "loss": 0.0566,
+      "step": 657
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5335038363171357e-05,
+      "loss": 0.059,
+      "step": 658
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5327365728900256e-05,
+      "loss": 0.0881,
+      "step": 659
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5319693094629157e-05,
+      "loss": 0.5762,
+      "step": 660
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.531202046035806e-05,
+      "loss": 0.0499,
+      "step": 661
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5304347826086957e-05,
+      "loss": 0.087,
+      "step": 662
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5296675191815855e-05,
+      "loss": 1.0719,
+      "step": 663
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528900255754476e-05,
+      "loss": 0.6158,
+      "step": 664
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528132992327366e-05,
+      "loss": 0.6117,
+      "step": 665
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5273657289002557e-05,
+      "loss": 0.0515,
+      "step": 666
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5265984654731458e-05,
+      "loss": 0.0608,
+      "step": 667
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.525831202046036e-05,
+      "loss": 0.0313,
+      "step": 668
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.5250639386189258e-05,
+      "loss": 0.1449,
+      "step": 669
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.524296675191816e-05,
+      "loss": 0.9883,
+      "step": 670
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.523529411764706e-05,
+      "loss": 0.0859,
+      "step": 671
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.522762148337596e-05,
+      "loss": 0.0561,
+      "step": 672
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521994884910486e-05,
+      "loss": 0.0926,
+      "step": 673
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521227621483376e-05,
+      "loss": 0.1101,
+      "step": 674
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.520460358056266e-05,
+      "loss": 0.068,
+      "step": 675
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5196930946291563e-05,
+      "loss": 0.0773,
+      "step": 676
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518925831202046e-05,
+      "loss": 0.0508,
+      "step": 677
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518158567774936e-05,
+      "loss": 0.0375,
+      "step": 678
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5173913043478264e-05,
+      "loss": 0.93,
+      "step": 679
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5166240409207162e-05,
+      "loss": 0.1507,
+      "step": 680
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.515856777493606e-05,
+      "loss": 0.0917,
+      "step": 681
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5150895140664962e-05,
+      "loss": 0.97,
+      "step": 682
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5143222506393864e-05,
+      "loss": 0.5019,
+      "step": 683
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5135549872122762e-05,
+      "loss": 0.0849,
+      "step": 684
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5127877237851664e-05,
+      "loss": 1.2316,
+      "step": 685
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5120204603580562e-05,
+      "loss": 0.0252,
+      "step": 686
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5112531969309463e-05,
+      "loss": 1.0543,
+      "step": 687
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5104859335038365e-05,
+      "loss": 0.0506,
+      "step": 688
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5097186700767263e-05,
+      "loss": 0.0252,
+      "step": 689
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.508951406649616e-05,
+      "loss": 0.6297,
+      "step": 690
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5081841432225066e-05,
+      "loss": 0.4497,
+      "step": 691
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5074168797953965e-05,
+      "loss": 0.0516,
+      "step": 692
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5066496163682863e-05,
+      "loss": 0.5501,
+      "step": 693
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5058823529411768e-05,
+      "loss": 0.5621,
+      "step": 694
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5051150895140666e-05,
+      "loss": 0.0577,
+      "step": 695
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5043478260869564e-05,
+      "loss": 0.0271,
+      "step": 696
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5035805626598466e-05,
+      "loss": 0.9496,
+      "step": 697
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5028132992327368e-05,
+      "loss": 0.0877,
+      "step": 698
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5020460358056266e-05,
+      "loss": 0.0523,
+      "step": 699
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5012787723785167e-05,
+      "loss": 0.0577,
+      "step": 700
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5005115089514066e-05,
+      "loss": 0.0522,
+      "step": 701
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.4997442455242967e-05,
+      "loss": 0.0517,
+      "step": 702
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.498976982097187e-05,
+      "loss": 0.6335,
+      "step": 703
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4982097186700767e-05,
+      "loss": 0.05,
+      "step": 704
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4974424552429665e-05,
+      "loss": 0.0753,
+      "step": 705
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.496675191815857e-05,
+      "loss": 0.0227,
+      "step": 706
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.495907928388747e-05,
+      "loss": 0.0755,
+      "step": 707
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.4951406649616367e-05,
+      "loss": 0.0215,
+      "step": 708
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.494373401534527e-05,
+      "loss": 0.0451,
+      "step": 709
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.493606138107417e-05,
+      "loss": 0.0343,
+      "step": 710
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492838874680307e-05,
+      "loss": 0.0836,
+      "step": 711
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492071611253197e-05,
+      "loss": 0.5287,
+      "step": 712
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4913043478260868e-05,
+      "loss": 0.0686,
+      "step": 713
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4905370843989773e-05,
+      "loss": 0.0211,
+      "step": 714
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489769820971867e-05,
+      "loss": 0.5932,
+      "step": 715
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489002557544757e-05,
+      "loss": 0.0371,
+      "step": 716
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.488235294117647e-05,
+      "loss": 1.0137,
+      "step": 717
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.4874680306905373e-05,
+      "loss": 0.9928,
+      "step": 718
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.486700767263427e-05,
+      "loss": 0.7431,
+      "step": 719
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4859335038363173e-05,
+      "loss": 0.0207,
+      "step": 720
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4851662404092074e-05,
+      "loss": 0.0681,
+      "step": 721
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4843989769820972e-05,
+      "loss": 0.0306,
+      "step": 722
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4836317135549874e-05,
+      "loss": 0.0204,
+      "step": 723
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4828644501278772e-05,
+      "loss": 0.0344,
+      "step": 724
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4820971867007674e-05,
+      "loss": 0.0345,
+      "step": 725
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4813299232736575e-05,
+      "loss": 0.9275,
+      "step": 726
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4805626598465474e-05,
+      "loss": 0.0196,
+      "step": 727
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4797953964194372e-05,
+      "loss": 1.0137,
+      "step": 728
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4790281329923277e-05,
+      "loss": 0.0318,
+      "step": 729
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4782608695652175e-05,
+      "loss": 0.0688,
+      "step": 730
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4774936061381073e-05,
+      "loss": 0.0314,
+      "step": 731
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4767263427109975e-05,
+      "loss": 0.7135,
+      "step": 732
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4759590792838877e-05,
+      "loss": 0.0299,
+      "step": 733
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4751918158567775e-05,
+      "loss": 1.0127,
+      "step": 734
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4744245524296676e-05,
+      "loss": 0.0304,
+      "step": 735
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4736572890025575e-05,
+      "loss": 1.4618,
+      "step": 736
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4728900255754476e-05,
+      "loss": 0.7167,
+      "step": 737
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4721227621483378e-05,
+      "loss": 0.0307,
+      "step": 738
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4713554987212276e-05,
+      "loss": 0.052,
+      "step": 739
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4705882352941174e-05,
+      "loss": 0.778,
+      "step": 740
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.469820971867008e-05,
+      "loss": 0.5482,
+      "step": 741
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4690537084398978e-05,
+      "loss": 1.4314,
+      "step": 742
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4682864450127876e-05,
+      "loss": 0.0422,
+      "step": 743
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.467519181585678e-05,
+      "loss": 0.6614,
+      "step": 744
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.466751918158568e-05,
+      "loss": 0.0312,
+      "step": 745
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4659846547314577e-05,
+      "loss": 0.043,
+      "step": 746
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.465217391304348e-05,
+      "loss": 0.0199,
+      "step": 747
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.464450127877238e-05,
+      "loss": 0.0832,
+      "step": 748
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.463682864450128e-05,
+      "loss": 0.0312,
+      "step": 749
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.462915601023018e-05,
+      "loss": 0.0309,
+      "step": 750
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.462148337595908e-05,
+      "loss": 0.0193,
+      "step": 751
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.461381074168798e-05,
+      "loss": 0.0446,
+      "step": 752
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.4606138107416882e-05,
+      "loss": 0.9993,
+      "step": 753
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.459846547314578e-05,
+      "loss": 0.016,
+      "step": 754
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4590792838874678e-05,
+      "loss": 0.0433,
+      "step": 755
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4583120204603583e-05,
+      "loss": 0.2614,
+      "step": 756
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.457544757033248e-05,
+      "loss": 0.018,
+      "step": 757
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.456777493606138e-05,
+      "loss": 0.6676,
+      "step": 758
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.456010230179028e-05,
+      "loss": 0.0599,
+      "step": 759
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4552429667519183e-05,
+      "loss": 0.0338,
+      "step": 760
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.454475703324808e-05,
+      "loss": 0.0268,
+      "step": 761
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4537084398976983e-05,
+      "loss": 0.0262,
+      "step": 762
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.452941176470588e-05,
+      "loss": 0.0252,
+      "step": 763
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4521739130434786e-05,
+      "loss": 0.0377,
+      "step": 764
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4514066496163684e-05,
+      "loss": 1.4866,
+      "step": 765
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4506393861892582e-05,
+      "loss": 1.0524,
+      "step": 766
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4498721227621484e-05,
+      "loss": 0.0328,
+      "step": 767
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4491048593350386e-05,
+      "loss": 0.024,
+      "step": 768
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4483375959079284e-05,
+      "loss": 0.022,
+      "step": 769
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4475703324808185e-05,
+      "loss": 0.0245,
+      "step": 770
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4468030690537087e-05,
+      "loss": 1.0107,
+      "step": 771
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4460358056265985e-05,
+      "loss": 0.0336,
+      "step": 772
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4452685421994887e-05,
+      "loss": 0.0391,
+      "step": 773
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4445012787723785e-05,
+      "loss": 0.0303,
+      "step": 774
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4437340153452687e-05,
+      "loss": 0.0443,
+      "step": 775
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4429667519181588e-05,
+      "loss": 0.0292,
+      "step": 776
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4421994884910487e-05,
+      "loss": 0.0284,
+      "step": 777
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4414322250639385e-05,
+      "loss": 1.5985,
+      "step": 778
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.440664961636829e-05,
+      "loss": 0.0114,
+      "step": 779
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4398976982097188e-05,
+      "loss": 0.0342,
+      "step": 780
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4391304347826086e-05,
+      "loss": 1.8501,
+      "step": 781
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4383631713554988e-05,
+      "loss": 0.5375,
+      "step": 782
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.437595907928389e-05,
+      "loss": 1.589,
+      "step": 783
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4368286445012788e-05,
+      "loss": 0.0221,
+      "step": 784
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.436061381074169e-05,
+      "loss": 0.0453,
+      "step": 785
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4352941176470587e-05,
+      "loss": 0.796,
+      "step": 786
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.434526854219949e-05,
+      "loss": 0.1249,
+      "step": 787
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.433759590792839e-05,
+      "loss": 0.7963,
+      "step": 788
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.432992327365729e-05,
+      "loss": 0.4231,
+      "step": 789
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.4322250639386187e-05,
+      "loss": 0.0247,
+      "step": 790
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.4314578005115092e-05,
+      "loss": 0.0295,
+      "step": 791
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.430690537084399e-05,
+      "loss": 0.0249,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.3537317216396332,
+      "eval_runtime": 58.1226,
+      "eval_samples_per_second": 6.813,
+      "eval_steps_per_second": 1.703,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.429923273657289e-05,
+      "loss": 0.0157,
+      "step": 793
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4291560102301794e-05,
+      "loss": 0.5222,
+      "step": 794
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4283887468030692e-05,
+      "loss": 0.0226,
+      "step": 795
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.427621483375959e-05,
+      "loss": 0.0293,
+      "step": 796
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.426854219948849e-05,
+      "loss": 0.015,
+      "step": 797
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4260869565217393e-05,
+      "loss": 1.0658,
+      "step": 798
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.425319693094629e-05,
+      "loss": 0.0412,
+      "step": 799
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4245524296675193e-05,
+      "loss": 0.0219,
+      "step": 800
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.423785166240409e-05,
+      "loss": 0.0149,
+      "step": 801
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4230179028132993e-05,
+      "loss": 0.0323,
+      "step": 802
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4222506393861895e-05,
+      "loss": 1.0764,
+      "step": 803
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4214833759590793e-05,
+      "loss": 0.7518,
+      "step": 804
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.420716112531969e-05,
+      "loss": 0.8166,
+      "step": 805
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4199488491048596e-05,
+      "loss": 0.0219,
+      "step": 806
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4191815856777494e-05,
+      "loss": 0.0154,
+      "step": 807
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4184143222506392e-05,
+      "loss": 0.0163,
+      "step": 808
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4176470588235294e-05,
+      "loss": 1.0562,
+      "step": 809
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4168797953964196e-05,
+      "loss": 0.0218,
+      "step": 810
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4161125319693094e-05,
+      "loss": 0.0232,
+      "step": 811
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4153452685421995e-05,
+      "loss": 0.2384,
+      "step": 812
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4145780051150894e-05,
+      "loss": 1.5286,
+      "step": 813
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4138107416879795e-05,
+      "loss": 0.0262,
+      "step": 814
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4130434782608697e-05,
+      "loss": 0.0334,
+      "step": 815
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4122762148337595e-05,
+      "loss": 0.6509,
+      "step": 816
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4115089514066497e-05,
+      "loss": 0.0159,
+      "step": 817
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.41074168797954e-05,
+      "loss": 0.6225,
+      "step": 818
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4099744245524297e-05,
+      "loss": 0.752,
+      "step": 819
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4092071611253198e-05,
+      "loss": 1.0594,
+      "step": 820
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.40843989769821e-05,
+      "loss": 0.0161,
+      "step": 821
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4076726342710998e-05,
+      "loss": 0.7492,
+      "step": 822
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40690537084399e-05,
+      "loss": 0.0302,
+      "step": 823
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4061381074168798e-05,
+      "loss": 0.038,
+      "step": 824
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40537084398977e-05,
+      "loss": 0.0243,
+      "step": 825
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40460358056266e-05,
+      "loss": 0.0331,
+      "step": 826
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40383631713555e-05,
+      "loss": 1.0252,
+      "step": 827
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4030690537084398e-05,
+      "loss": 0.0172,
+      "step": 828
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4023017902813303e-05,
+      "loss": 0.6168,
+      "step": 829
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40153452685422e-05,
+      "loss": 0.3135,
+      "step": 830
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40076726342711e-05,
+      "loss": 0.0168,
+      "step": 831
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.0172,
+      "step": 832
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.3992327365728902e-05,
+      "loss": 1.0349,
+      "step": 833
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39846547314578e-05,
+      "loss": 0.0978,
+      "step": 834
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3976982097186702e-05,
+      "loss": 0.0167,
+      "step": 835
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39693094629156e-05,
+      "loss": 1.0381,
+      "step": 836
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3961636828644502e-05,
+      "loss": 0.6857,
+      "step": 837
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3953964194373404e-05,
+      "loss": 1.0406,
+      "step": 838
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3946291560102302e-05,
+      "loss": 0.071,
+      "step": 839
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.39386189258312e-05,
+      "loss": 0.0193,
+      "step": 840
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3930946291560105e-05,
+      "loss": 0.0193,
+      "step": 841
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3923273657289003e-05,
+      "loss": 0.9916,
+      "step": 842
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.39156010230179e-05,
+      "loss": 0.0723,
+      "step": 843
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3907928388746806e-05,
+      "loss": 0.0626,
+      "step": 844
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3900255754475705e-05,
+      "loss": 0.7264,
+      "step": 845
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3892583120204603e-05,
+      "loss": 0.827,
+      "step": 846
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3884910485933504e-05,
+      "loss": 1.2193,
+      "step": 847
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3877237851662406e-05,
+      "loss": 0.6754,
+      "step": 848
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3869565217391304e-05,
+      "loss": 0.6856,
+      "step": 849
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3861892583120206e-05,
+      "loss": 0.7441,
+      "step": 850
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3854219948849104e-05,
+      "loss": 0.023,
+      "step": 851
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3846547314578006e-05,
+      "loss": 0.4742,
+      "step": 852
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3838874680306907e-05,
+      "loss": 0.0363,
+      "step": 853
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3831202046035806e-05,
+      "loss": 0.0409,
+      "step": 854
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3823529411764704e-05,
+      "loss": 0.8353,
+      "step": 855
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.381585677749361e-05,
+      "loss": 0.0333,
+      "step": 856
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3808184143222507e-05,
+      "loss": 0.0216,
+      "step": 857
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3800511508951405e-05,
+      "loss": 0.9688,
+      "step": 858
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3792838874680307e-05,
+      "loss": 1.0078,
+      "step": 859
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.378516624040921e-05,
+      "loss": 0.0183,
+      "step": 860
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3777493606138107e-05,
+      "loss": 0.0183,
+      "step": 861
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3769820971867008e-05,
+      "loss": 0.9786,
+      "step": 862
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3762148337595907e-05,
+      "loss": 0.9166,
+      "step": 863
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3754475703324808e-05,
+      "loss": 1.1408,
+      "step": 864
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.374680306905371e-05,
+      "loss": 1.1394,
+      "step": 865
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.3739130434782608e-05,
+      "loss": 0.0185,
+      "step": 866
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.373145780051151e-05,
+      "loss": 0.9979,
+      "step": 867
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.372378516624041e-05,
+      "loss": 1.0092,
+      "step": 868
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.371611253196931e-05,
+      "loss": 0.7276,
+      "step": 869
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.370843989769821e-05,
+      "loss": 0.0359,
+      "step": 870
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3700767263427113e-05,
+      "loss": 0.0864,
+      "step": 871
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.369309462915601e-05,
+      "loss": 0.0246,
+      "step": 872
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3685421994884912e-05,
+      "loss": 0.8518,
+      "step": 873
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.367774936061381e-05,
+      "loss": 1.433,
+      "step": 874
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3670076726342712e-05,
+      "loss": 0.035,
+      "step": 875
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3662404092071614e-05,
+      "loss": 0.5922,
+      "step": 876
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3654731457800512e-05,
+      "loss": 0.529,
+      "step": 877
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.364705882352941e-05,
+      "loss": 0.6423,
+      "step": 878
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3639386189258315e-05,
+      "loss": 0.0337,
+      "step": 879
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3631713554987214e-05,
+      "loss": 0.0402,
+      "step": 880
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3624040920716112e-05,
+      "loss": 0.0782,
+      "step": 881
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3616368286445013e-05,
+      "loss": 0.4141,
+      "step": 882
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3608695652173915e-05,
+      "loss": 0.0359,
+      "step": 883
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3601023017902813e-05,
+      "loss": 0.4509,
+      "step": 884
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3593350383631715e-05,
+      "loss": 0.6414,
+      "step": 885
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3585677749360613e-05,
+      "loss": 0.6873,
+      "step": 886
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3578005115089515e-05,
+      "loss": 0.0304,
+      "step": 887
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3570332480818416e-05,
+      "loss": 0.0315,
+      "step": 888
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3562659846547315e-05,
+      "loss": 0.0896,
+      "step": 889
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3554987212276213e-05,
+      "loss": 0.0201,
+      "step": 890
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3547314578005118e-05,
+      "loss": 0.027,
+      "step": 891
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3539641943734016e-05,
+      "loss": 1.9853,
+      "step": 892
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3531969309462914e-05,
+      "loss": 2.9647,
+      "step": 893
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.352429667519182e-05,
+      "loss": 1.9403,
+      "step": 894
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3516624040920717e-05,
+      "loss": 0.0193,
+      "step": 895
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3508951406649616e-05,
+      "loss": 0.9912,
+      "step": 896
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3501278772378517e-05,
+      "loss": 0.02,
+      "step": 897
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.349360613810742e-05,
+      "loss": 0.9015,
+      "step": 898
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3485933503836317e-05,
+      "loss": 0.1294,
+      "step": 899
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.347826086956522e-05,
+      "loss": 0.0559,
+      "step": 900
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.3470588235294117e-05,
+      "loss": 1.8039,
+      "step": 901
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.346291560102302e-05,
+      "loss": 0.0336,
+      "step": 902
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.345524296675192e-05,
+      "loss": 0.0383,
+      "step": 903
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.344757033248082e-05,
+      "loss": 0.0722,
+      "step": 904
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3439897698209717e-05,
+      "loss": 0.0278,
+      "step": 905
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.343222506393862e-05,
+      "loss": 0.0684,
+      "step": 906
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.342455242966752e-05,
+      "loss": 0.0236,
+      "step": 907
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3416879795396418e-05,
+      "loss": 0.0348,
+      "step": 908
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340920716112532e-05,
+      "loss": 0.0204,
+      "step": 909
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340153452685422e-05,
+      "loss": 0.8109,
+      "step": 910
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.339386189258312e-05,
+      "loss": 0.0279,
+      "step": 911
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.338618925831202e-05,
+      "loss": 0.0361,
+      "step": 912
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337851662404092e-05,
+      "loss": 0.0355,
+      "step": 913
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337084398976982e-05,
+      "loss": 0.7958,
+      "step": 914
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.3363171355498723e-05,
+      "loss": 0.0362,
+      "step": 915
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.335549872122762e-05,
+      "loss": 0.0252,
+      "step": 916
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3347826086956522e-05,
+      "loss": 0.8165,
+      "step": 917
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3340153452685424e-05,
+      "loss": 0.0237,
+      "step": 918
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3332480818414322e-05,
+      "loss": 0.8162,
+      "step": 919
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.332480818414322e-05,
+      "loss": 0.7786,
+      "step": 920
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3317135549872125e-05,
+      "loss": 0.0787,
+      "step": 921
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3309462915601024e-05,
+      "loss": 0.1764,
+      "step": 922
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3301790281329925e-05,
+      "loss": 0.7449,
+      "step": 923
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3294117647058824e-05,
+      "loss": 0.0452,
+      "step": 924
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3286445012787725e-05,
+      "loss": 0.0181,
+      "step": 925
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3278772378516627e-05,
+      "loss": 0.1783,
+      "step": 926
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3271099744245525e-05,
+      "loss": 0.5158,
+      "step": 927
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3263427109974423e-05,
+      "loss": 0.7574,
+      "step": 928
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3255754475703328e-05,
+      "loss": 2.6099,
+      "step": 929
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3248081841432226e-05,
+      "loss": 0.0304,
+      "step": 930
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3240409207161125e-05,
+      "loss": 0.0795,
+      "step": 931
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3232736572890026e-05,
+      "loss": 0.0342,
+      "step": 932
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3225063938618928e-05,
+      "loss": 0.0473,
+      "step": 933
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3217391304347826e-05,
+      "loss": 0.0831,
+      "step": 934
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3209718670076728e-05,
+      "loss": 0.0465,
+      "step": 935
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3202046035805626e-05,
+      "loss": 0.6769,
+      "step": 936
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3194373401534528e-05,
+      "loss": 0.0173,
+      "step": 937
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.318670076726343e-05,
+      "loss": 0.0169,
+      "step": 938
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3179028132992327e-05,
+      "loss": 0.0469,
+      "step": 939
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3171355498721226e-05,
+      "loss": 0.048,
+      "step": 940
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.316368286445013e-05,
+      "loss": 1.0704,
+      "step": 941
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.315601023017903e-05,
+      "loss": 1.0738,
+      "step": 942
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3148337595907927e-05,
+      "loss": 0.0163,
+      "step": 943
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3140664961636832e-05,
+      "loss": 2.0702,
+      "step": 944
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.313299232736573e-05,
+      "loss": 1.0433,
+      "step": 945
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.312531969309463e-05,
+      "loss": 1.0351,
+      "step": 946
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.311764705882353e-05,
+      "loss": 1.053,
+      "step": 947
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.3109974424552432e-05,
+      "loss": 0.1797,
+      "step": 948
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.310230179028133e-05,
+      "loss": 1.0259,
+      "step": 949
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.309462915601023e-05,
+      "loss": 0.03,
+      "step": 950
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.308695652173913e-05,
+      "loss": 0.905,
+      "step": 951
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.307928388746803e-05,
+      "loss": 0.0699,
+      "step": 952
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3071611253196933e-05,
+      "loss": 1.0013,
+      "step": 953
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.306393861892583e-05,
+      "loss": 0.0319,
+      "step": 954
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.305626598465473e-05,
+      "loss": 0.0361,
+      "step": 955
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3048593350383634e-05,
+      "loss": 0.0201,
+      "step": 956
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3040920716112533e-05,
+      "loss": 0.033,
+      "step": 957
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.303324808184143e-05,
+      "loss": 0.0411,
+      "step": 958
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3025575447570332e-05,
+      "loss": 0.7138,
+      "step": 959
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3017902813299234e-05,
+      "loss": 0.0434,
+      "step": 960
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3010230179028132e-05,
+      "loss": 0.8445,
+      "step": 961
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3002557544757034e-05,
+      "loss": 0.0323,
+      "step": 962
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2994884910485932e-05,
+      "loss": 0.0203,
+      "step": 963
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2987212276214834e-05,
+      "loss": 0.2302,
+      "step": 964
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2979539641943735e-05,
+      "loss": 0.0293,
+      "step": 965
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2971867007672634e-05,
+      "loss": 0.0593,
+      "step": 966
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2964194373401535e-05,
+      "loss": 0.6505,
+      "step": 967
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2956521739130437e-05,
+      "loss": 0.5891,
+      "step": 968
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2948849104859335e-05,
+      "loss": 0.0191,
+      "step": 969
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2941176470588233e-05,
+      "loss": 0.0188,
+      "step": 970
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2933503836317138e-05,
+      "loss": 0.9537,
+      "step": 971
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2925831202046037e-05,
+      "loss": 0.0186,
+      "step": 972
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2918158567774935e-05,
+      "loss": 1.0237,
+      "step": 973
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2910485933503836e-05,
+      "loss": 0.0417,
+      "step": 974
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2902813299232738e-05,
+      "loss": 0.0369,
+      "step": 975
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.289514066496164e-05,
+      "loss": 0.0184,
+      "step": 976
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2887468030690538e-05,
+      "loss": 0.811,
+      "step": 977
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2879795396419436e-05,
+      "loss": 0.0183,
+      "step": 978
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.287212276214834e-05,
+      "loss": 1.0442,
+      "step": 979
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.286445012787724e-05,
+      "loss": 0.027,
+      "step": 980
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.2856777493606137e-05,
+      "loss": 0.062,
+      "step": 981
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284910485933504e-05,
+      "loss": 0.0186,
+      "step": 982
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284143222506394e-05,
+      "loss": 0.6057,
+      "step": 983
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.283375959079284e-05,
+      "loss": 1.6594,
+      "step": 984
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.282608695652174e-05,
+      "loss": 0.0192,
+      "step": 985
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281841432225064e-05,
+      "loss": 1.5361,
+      "step": 986
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281074168797954e-05,
+      "loss": 0.0384,
+      "step": 987
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.2803069053708442e-05,
+      "loss": 2.0238,
+      "step": 988
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.279539641943734e-05,
+      "loss": 0.0334,
+      "step": 989
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2787723785166242e-05,
+      "loss": 0.0571,
+      "step": 990
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2780051150895143e-05,
+      "loss": 0.0211,
+      "step": 991
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.277237851662404e-05,
+      "loss": 0.0409,
+      "step": 992
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.276470588235294e-05,
+      "loss": 0.1059,
+      "step": 993
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2757033248081845e-05,
+      "loss": 0.0277,
+      "step": 994
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2749360613810743e-05,
+      "loss": 1.0023,
+      "step": 995
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.274168797953964e-05,
+      "loss": 0.0186,
+      "step": 996
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2734015345268543e-05,
+      "loss": 0.0345,
+      "step": 997
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2726342710997445e-05,
+      "loss": 0.0169,
+      "step": 998
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2718670076726343e-05,
+      "loss": 0.7292,
+      "step": 999
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2710997442455244e-05,
+      "loss": 1.2655,
+      "step": 1000
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2703324808184143e-05,
+      "loss": 0.03,
+      "step": 1001
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2695652173913044e-05,
+      "loss": 1.0266,
+      "step": 1002
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2687979539641946e-05,
+      "loss": 0.7645,
+      "step": 1003
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2680306905370844e-05,
+      "loss": 0.7985,
+      "step": 1004
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2672634271099742e-05,
+      "loss": 0.76,
+      "step": 1005
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2664961636828647e-05,
+      "loss": 0.7577,
+      "step": 1006
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2657289002557545e-05,
+      "loss": 0.0248,
+      "step": 1007
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2649616368286444e-05,
+      "loss": 0.0262,
+      "step": 1008
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2641943734015345e-05,
+      "loss": 1.9984,
+      "step": 1009
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2634271099744247e-05,
+      "loss": 0.1489,
+      "step": 1010
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2626598465473145e-05,
+      "loss": 0.0683,
+      "step": 1011
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2618925831202047e-05,
+      "loss": 0.054,
+      "step": 1012
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2611253196930945e-05,
+      "loss": 0.0269,
+      "step": 1013
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2603580562659847e-05,
+      "loss": 1.0378,
+      "step": 1014
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2595907928388748e-05,
+      "loss": 0.0171,
+      "step": 1015
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2588235294117646e-05,
+      "loss": 0.8433,
+      "step": 1016
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2580562659846548e-05,
+      "loss": 0.0468,
+      "step": 1017
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.257289002557545e-05,
+      "loss": 0.0331,
+      "step": 1018
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2565217391304348e-05,
+      "loss": 0.0308,
+      "step": 1019
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2557544757033246e-05,
+      "loss": 1.0007,
+      "step": 1020
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254987212276215e-05,
+      "loss": 2.0242,
+      "step": 1021
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254219948849105e-05,
+      "loss": 0.7713,
+      "step": 1022
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2534526854219948e-05,
+      "loss": 1.1112,
+      "step": 1023
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.252685421994885e-05,
+      "loss": 0.0184,
+      "step": 1024
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251918158567775e-05,
+      "loss": 0.0243,
+      "step": 1025
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251150895140665e-05,
+      "loss": 0.046,
+      "step": 1026
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.250383631713555e-05,
+      "loss": 0.0378,
+      "step": 1027
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.249616368286445e-05,
+      "loss": 0.0329,
+      "step": 1028
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2488491048593354e-05,
+      "loss": 0.0245,
+      "step": 1029
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2480818414322252e-05,
+      "loss": 0.0238,
+      "step": 1030
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.247314578005115e-05,
+      "loss": 1.0628,
+      "step": 1031
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2465473145780052e-05,
+      "loss": 0.0703,
+      "step": 1032
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2457800511508954e-05,
+      "loss": 0.0251,
+      "step": 1033
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2450127877237852e-05,
+      "loss": 1.0561,
+      "step": 1034
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2442455242966753e-05,
+      "loss": 0.0274,
+      "step": 1035
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.243478260869565e-05,
+      "loss": 0.023,
+      "step": 1036
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2427109974424553e-05,
+      "loss": 0.0223,
+      "step": 1037
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2419437340153455e-05,
+      "loss": 0.8113,
+      "step": 1038
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2411764705882353e-05,
+      "loss": 0.0346,
+      "step": 1039
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2404092071611255e-05,
+      "loss": 0.0164,
+      "step": 1040
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2396419437340156e-05,
+      "loss": 0.0237,
+      "step": 1041
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2388746803069054e-05,
+      "loss": 0.0305,
+      "step": 1042
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2381074168797953e-05,
+      "loss": 0.0158,
+      "step": 1043
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2373401534526858e-05,
+      "loss": 0.0225,
+      "step": 1044
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2365728900255756e-05,
+      "loss": 0.0226,
+      "step": 1045
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2358056265984654e-05,
+      "loss": 0.015,
+      "step": 1046
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2350383631713556e-05,
+      "loss": 0.0219,
+      "step": 1047
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2342710997442457e-05,
+      "loss": 1.0694,
+      "step": 1048
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2335038363171356e-05,
+      "loss": 0.0348,
+      "step": 1049
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2327365728900257e-05,
+      "loss": 0.0213,
+      "step": 1050
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2319693094629155e-05,
+      "loss": 0.9405,
+      "step": 1051
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2312020460358057e-05,
+      "loss": 0.0252,
+      "step": 1052
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.230434782608696e-05,
+      "loss": 0.0254,
+      "step": 1053
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2296675191815857e-05,
+      "loss": 0.7353,
+      "step": 1054
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2289002557544755e-05,
+      "loss": 0.0253,
+      "step": 1055
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.228132992327366e-05,
+      "loss": 0.8361,
+      "step": 1056
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2273657289002558e-05,
+      "loss": 0.0196,
+      "step": 1057
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2265984654731457e-05,
+      "loss": 0.014,
+      "step": 1058
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2258312020460358e-05,
+      "loss": 0.8297,
+      "step": 1059
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.225063938618926e-05,
+      "loss": 1.0654,
+      "step": 1060
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2242966751918158e-05,
+      "loss": 0.0248,
+      "step": 1061
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.223529411764706e-05,
+      "loss": 0.0226,
+      "step": 1062
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2227621483375958e-05,
+      "loss": 0.0238,
+      "step": 1063
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221994884910486e-05,
+      "loss": 0.0212,
+      "step": 1064
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221227621483376e-05,
+      "loss": 0.0274,
+      "step": 1065
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.220460358056266e-05,
+      "loss": 1.0217,
+      "step": 1066
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.219693094629156e-05,
+      "loss": 0.0252,
+      "step": 1067
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2189258312020462e-05,
+      "loss": 0.0219,
+      "step": 1068
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.218158567774936e-05,
+      "loss": 0.0349,
+      "step": 1069
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.217391304347826e-05,
+      "loss": 0.0152,
+      "step": 1070
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2166240409207164e-05,
+      "loss": 0.021,
+      "step": 1071
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2158567774936062e-05,
+      "loss": 0.015,
+      "step": 1072
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.215089514066496e-05,
+      "loss": 0.0277,
+      "step": 1073
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2143222506393862e-05,
+      "loss": 0.1858,
+      "step": 1074
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2135549872122764e-05,
+      "loss": 0.0136,
+      "step": 1075
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2127877237851662e-05,
+      "loss": 0.0132,
+      "step": 1076
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2120204603580563e-05,
+      "loss": 0.0126,
+      "step": 1077
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.211253196930946e-05,
+      "loss": 0.5171,
+      "step": 1078
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2104859335038363e-05,
+      "loss": 0.0422,
+      "step": 1079
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2097186700767265e-05,
+      "loss": 0.0186,
+      "step": 1080
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2089514066496163e-05,
+      "loss": 0.8412,
+      "step": 1081
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2081841432225065e-05,
+      "loss": 0.0183,
+      "step": 1082
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2074168797953966e-05,
+      "loss": 1.0243,
+      "step": 1083
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2066496163682865e-05,
+      "loss": 0.0212,
+      "step": 1084
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2058823529411766e-05,
+      "loss": 0.2841,
+      "step": 1085
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2051150895140664e-05,
+      "loss": 0.0118,
+      "step": 1086
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2043478260869566e-05,
+      "loss": 0.0247,
+      "step": 1087
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2035805626598468e-05,
+      "loss": 1.5121,
+      "step": 1088
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2028132992327366e-05,
+      "loss": 0.0304,
+      "step": 1089
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2020460358056267e-05,
+      "loss": 0.0329,
+      "step": 1090
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.201278772378517e-05,
+      "loss": 1.1288,
+      "step": 1091
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.2005115089514067e-05,
+      "loss": 0.0145,
+      "step": 1092
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.1997442455242966e-05,
+      "loss": 0.0498,
+      "step": 1093
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.198976982097187e-05,
+      "loss": 0.0244,
+      "step": 1094
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.198209718670077e-05,
+      "loss": 0.0112,
+      "step": 1095
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.1974424552429667e-05,
+      "loss": 0.8642,
+      "step": 1096
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.196675191815857e-05,
+      "loss": 0.8822,
+      "step": 1097
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.195907928388747e-05,
+      "loss": 0.0278,
+      "step": 1098
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.195140664961637e-05,
+      "loss": 0.6954,
+      "step": 1099
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.194373401534527e-05,
+      "loss": 0.0179,
+      "step": 1100
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.1936061381074168e-05,
+      "loss": 0.5607,
+      "step": 1101
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.192838874680307e-05,
+      "loss": 0.0191,
+      "step": 1102
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.192071611253197e-05,
+      "loss": 1.9519,
+      "step": 1103
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.191304347826087e-05,
+      "loss": 0.8347,
+      "step": 1104
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1905370843989768e-05,
+      "loss": 0.0262,
+      "step": 1105
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1897698209718673e-05,
+      "loss": 0.0477,
+      "step": 1106
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.189002557544757e-05,
+      "loss": 0.0227,
+      "step": 1107
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.188235294117647e-05,
+      "loss": 0.0205,
+      "step": 1108
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.187468030690537e-05,
+      "loss": 0.0208,
+      "step": 1109
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.1867007672634273e-05,
+      "loss": 0.0116,
+      "step": 1110
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.185933503836317e-05,
+      "loss": 1.1546,
+      "step": 1111
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1851662404092072e-05,
+      "loss": 0.0114,
+      "step": 1112
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.184398976982097e-05,
+      "loss": 0.0113,
+      "step": 1113
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1836317135549872e-05,
+      "loss": 0.6881,
+      "step": 1114
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1828644501278774e-05,
+      "loss": 0.0294,
+      "step": 1115
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1820971867007672e-05,
+      "loss": 0.1569,
+      "step": 1116
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1813299232736574e-05,
+      "loss": 1.1362,
+      "step": 1117
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1805626598465475e-05,
+      "loss": 0.0196,
+      "step": 1118
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1797953964194374e-05,
+      "loss": 0.0205,
+      "step": 1119
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1790281329923272e-05,
+      "loss": 0.0198,
+      "step": 1120
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1782608695652177e-05,
+      "loss": 0.0113,
+      "step": 1121
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1774936061381075e-05,
+      "loss": 0.02,
+      "step": 1122
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1767263427109973e-05,
+      "loss": 0.0195,
+      "step": 1123
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1759590792838875e-05,
+      "loss": 0.8016,
+      "step": 1124
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1751918158567776e-05,
+      "loss": 0.0122,
+      "step": 1125
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1744245524296675e-05,
+      "loss": 1.1132,
+      "step": 1126
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1736572890025576e-05,
+      "loss": 0.011,
+      "step": 1127
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1728900255754474e-05,
+      "loss": 0.7705,
+      "step": 1128
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1721227621483376e-05,
+      "loss": 1.1467,
+      "step": 1129
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1713554987212278e-05,
+      "loss": 0.0176,
+      "step": 1130
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1705882352941176e-05,
+      "loss": 0.0293,
+      "step": 1131
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1698209718670074e-05,
+      "loss": 0.0208,
+      "step": 1132
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.169053708439898e-05,
+      "loss": 0.0199,
+      "step": 1133
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1682864450127877e-05,
+      "loss": 0.0431,
+      "step": 1134
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.167519181585678e-05,
+      "loss": 1.6066,
+      "step": 1135
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.1667519181585677e-05,
+      "loss": 0.029,
+      "step": 1136
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165984654731458e-05,
+      "loss": 0.029,
+      "step": 1137
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165217391304348e-05,
+      "loss": 1.1749,
+      "step": 1138
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.164450127877238e-05,
+      "loss": 0.0189,
+      "step": 1139
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.163682864450128e-05,
+      "loss": 0.0309,
+      "step": 1140
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.1629156010230182e-05,
+      "loss": 0.0287,
+      "step": 1141
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.162148337595908e-05,
+      "loss": 0.0202,
+      "step": 1142
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.161381074168798e-05,
+      "loss": 0.7889,
+      "step": 1143
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.1606138107416883e-05,
+      "loss": 0.0285,
+      "step": 1144
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159846547314578e-05,
+      "loss": 0.0535,
+      "step": 1145
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159079283887468e-05,
+      "loss": 1.0829,
+      "step": 1146
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.158312020460358e-05,
+      "loss": 0.0989,
+      "step": 1147
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1575447570332483e-05,
+      "loss": 0.0115,
+      "step": 1148
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.156777493606138e-05,
+      "loss": 0.0227,
+      "step": 1149
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1560102301790283e-05,
+      "loss": 0.5506,
+      "step": 1150
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.155242966751918e-05,
+      "loss": 1.1234,
+      "step": 1151
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1544757033248083e-05,
+      "loss": 0.0564,
+      "step": 1152
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1537084398976984e-05,
+      "loss": 0.0127,
+      "step": 1153
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1529411764705882e-05,
+      "loss": 0.0218,
+      "step": 1154
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.152173913043478e-05,
+      "loss": 0.0124,
+      "step": 1155
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1514066496163686e-05,
+      "loss": 0.0223,
+      "step": 1156
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1506393861892584e-05,
+      "loss": 0.0505,
+      "step": 1157
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1498721227621482e-05,
+      "loss": 0.0225,
+      "step": 1158
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1491048593350384e-05,
+      "loss": 0.0308,
+      "step": 1159
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1483375959079285e-05,
+      "loss": 0.036,
+      "step": 1160
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1475703324808184e-05,
+      "loss": 0.0121,
+      "step": 1161
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1468030690537085e-05,
+      "loss": 0.0274,
+      "step": 1162
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1460358056265983e-05,
+      "loss": 1.0397,
+      "step": 1163
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1452685421994885e-05,
+      "loss": 0.0332,
+      "step": 1164
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1445012787723787e-05,
+      "loss": 0.0182,
+      "step": 1165
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1437340153452685e-05,
+      "loss": 0.018,
+      "step": 1166
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1429667519181587e-05,
+      "loss": 0.859,
+      "step": 1167
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1421994884910488e-05,
+      "loss": 0.0321,
+      "step": 1168
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1414322250639386e-05,
+      "loss": 0.0225,
+      "step": 1169
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1406649616368285e-05,
+      "loss": 0.0224,
+      "step": 1170
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.139897698209719e-05,
+      "loss": 0.03,
+      "step": 1171
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1391304347826088e-05,
+      "loss": 0.7473,
+      "step": 1172
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1383631713554986e-05,
+      "loss": 0.8609,
+      "step": 1173
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1375959079283888e-05,
+      "loss": 0.0105,
+      "step": 1174
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.136828644501279e-05,
+      "loss": 0.0217,
+      "step": 1175
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1360613810741687e-05,
+      "loss": 0.0218,
+      "step": 1176
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.135294117647059e-05,
+      "loss": 0.016,
+      "step": 1177
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1345268542199487e-05,
+      "loss": 0.7844,
+      "step": 1178
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.133759590792839e-05,
+      "loss": 0.01,
+      "step": 1179
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132992327365729e-05,
+      "loss": 0.8774,
+      "step": 1180
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132225063938619e-05,
+      "loss": 0.8375,
+      "step": 1181
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.1314578005115087e-05,
+      "loss": 0.0188,
+      "step": 1182
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.1306905370843992e-05,
+      "loss": 1.1026,
+      "step": 1183
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129923273657289e-05,
+      "loss": 1.0447,
+      "step": 1184
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129156010230179e-05,
+      "loss": 0.0101,
+      "step": 1185
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.128388746803069e-05,
+      "loss": 0.0247,
+      "step": 1186
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.127621483375959e-05,
+      "loss": 0.0295,
+      "step": 1187
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.1268542199488493e-05,
+      "loss": 0.0191,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.37390685081481934,
+      "eval_runtime": 57.4087,
+      "eval_samples_per_second": 6.898,
+      "eval_steps_per_second": 1.724,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.126086956521739e-05,
+      "loss": 0.0103,
+      "step": 1189
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1253196930946293e-05,
+      "loss": 0.0231,
+      "step": 1190
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1245524296675195e-05,
+      "loss": 0.0389,
+      "step": 1191
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1237851662404093e-05,
+      "loss": 0.0298,
+      "step": 1192
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.123017902813299e-05,
+      "loss": 0.0196,
+      "step": 1193
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1222506393861896e-05,
+      "loss": 0.9688,
+      "step": 1194
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1214833759590794e-05,
+      "loss": 0.0458,
+      "step": 1195
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1207161125319693e-05,
+      "loss": 0.0161,
+      "step": 1196
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1199488491048594e-05,
+      "loss": 0.0157,
+      "step": 1197
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1191815856777496e-05,
+      "loss": 0.8732,
+      "step": 1198
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1184143222506394e-05,
+      "loss": 0.0103,
+      "step": 1199
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1176470588235296e-05,
+      "loss": 0.0267,
+      "step": 1200
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1168797953964194e-05,
+      "loss": 0.0266,
+      "step": 1201
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1161125319693095e-05,
+      "loss": 0.0157,
+      "step": 1202
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1153452685421997e-05,
+      "loss": 0.265,
+      "step": 1203
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1145780051150895e-05,
+      "loss": 0.0537,
+      "step": 1204
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1138107416879794e-05,
+      "loss": 0.0306,
+      "step": 1205
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.11304347826087e-05,
+      "loss": 1.1714,
+      "step": 1206
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1122762148337597e-05,
+      "loss": 1.1704,
+      "step": 1207
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1115089514066495e-05,
+      "loss": 0.0154,
+      "step": 1208
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1107416879795397e-05,
+      "loss": 0.0195,
+      "step": 1209
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1099744245524298e-05,
+      "loss": 0.9099,
+      "step": 1210
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1092071611253196e-05,
+      "loss": 0.0112,
+      "step": 1211
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1084398976982098e-05,
+      "loss": 0.1506,
+      "step": 1212
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1076726342710996e-05,
+      "loss": 0.0111,
+      "step": 1213
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1069053708439898e-05,
+      "loss": 0.0244,
+      "step": 1214
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10613810741688e-05,
+      "loss": 0.8889,
+      "step": 1215
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1053708439897698e-05,
+      "loss": 0.0146,
+      "step": 1216
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10460358056266e-05,
+      "loss": 0.0189,
+      "step": 1217
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10383631713555e-05,
+      "loss": 0.0097,
+      "step": 1218
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10306905370844e-05,
+      "loss": 1.1683,
+      "step": 1219
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1023017902813297e-05,
+      "loss": 1.1621,
+      "step": 1220
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1015345268542202e-05,
+      "loss": 0.0194,
+      "step": 1221
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.10076726342711e-05,
+      "loss": 0.0155,
+      "step": 1222
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.0296,
+      "step": 1223
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.09923273657289e-05,
+      "loss": 0.02,
+      "step": 1224
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.0984654731457802e-05,
+      "loss": 0.0154,
+      "step": 1225
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09769820971867e-05,
+      "loss": 0.0151,
+      "step": 1226
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0969309462915602e-05,
+      "loss": 0.0223,
+      "step": 1227
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09616368286445e-05,
+      "loss": 1.7159,
+      "step": 1228
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0953964194373402e-05,
+      "loss": 0.0194,
+      "step": 1229
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0946291560102303e-05,
+      "loss": 0.0637,
+      "step": 1230
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09386189258312e-05,
+      "loss": 0.0156,
+      "step": 1231
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09309462915601e-05,
+      "loss": 1.0612,
+      "step": 1232
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0923273657289005e-05,
+      "loss": 0.8929,
+      "step": 1233
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0915601023017903e-05,
+      "loss": 1.1649,
+      "step": 1234
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.09079283887468e-05,
+      "loss": 2.059,
+      "step": 1235
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0900255754475703e-05,
+      "loss": 0.0169,
+      "step": 1236
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0892583120204604e-05,
+      "loss": 0.0158,
+      "step": 1237
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0884910485933503e-05,
+      "loss": 0.0108,
+      "step": 1238
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0877237851662404e-05,
+      "loss": 0.0152,
+      "step": 1239
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0869565217391306e-05,
+      "loss": 0.0109,
+      "step": 1240
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0861892583120208e-05,
+      "loss": 1.1352,
+      "step": 1241
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0854219948849106e-05,
+      "loss": 0.0226,
+      "step": 1242
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0846547314578004e-05,
+      "loss": 0.0154,
+      "step": 1243
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.083887468030691e-05,
+      "loss": 0.0232,
+      "step": 1244
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0831202046035807e-05,
+      "loss": 0.0142,
+      "step": 1245
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0823529411764705e-05,
+      "loss": 0.9143,
+      "step": 1246
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0815856777493607e-05,
+      "loss": 0.0165,
+      "step": 1247
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.080818414322251e-05,
+      "loss": 0.0446,
+      "step": 1248
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0800511508951407e-05,
+      "loss": 0.0121,
+      "step": 1249
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.079283887468031e-05,
+      "loss": 1.0991,
+      "step": 1250
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.0785166240409207e-05,
+      "loss": 0.0159,
+      "step": 1251
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.077749360613811e-05,
+      "loss": 0.0166,
+      "step": 1252
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.076982097186701e-05,
+      "loss": 0.0171,
+      "step": 1253
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0762148337595908e-05,
+      "loss": 0.7986,
+      "step": 1254
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0754475703324806e-05,
+      "loss": 0.0174,
+      "step": 1255
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.074680306905371e-05,
+      "loss": 1.0431,
+      "step": 1256
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.073913043478261e-05,
+      "loss": 0.1766,
+      "step": 1257
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.0731457800511508e-05,
+      "loss": 0.0209,
+      "step": 1258
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.072378516624041e-05,
+      "loss": 0.0178,
+      "step": 1259
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.071611253196931e-05,
+      "loss": 0.0215,
+      "step": 1260
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.070843989769821e-05,
+      "loss": 0.0112,
+      "step": 1261
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.070076726342711e-05,
+      "loss": 0.0166,
+      "step": 1262
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0693094629156012e-05,
+      "loss": 0.0161,
+      "step": 1263
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.068542199488491e-05,
+      "loss": 0.8922,
+      "step": 1264
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0677749360613812e-05,
+      "loss": 0.0133,
+      "step": 1265
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.067007672634271e-05,
+      "loss": 0.0117,
+      "step": 1266
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0662404092071612e-05,
+      "loss": 0.0153,
+      "step": 1267
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0654731457800514e-05,
+      "loss": 0.0153,
+      "step": 1268
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0647058823529412e-05,
+      "loss": 0.0195,
+      "step": 1269
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.063938618925831e-05,
+      "loss": 1.1511,
+      "step": 1270
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0631713554987215e-05,
+      "loss": 0.0108,
+      "step": 1271
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0624040920716113e-05,
+      "loss": 0.8897,
+      "step": 1272
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.061636828644501e-05,
+      "loss": 0.0157,
+      "step": 1273
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0608695652173913e-05,
+      "loss": 0.0108,
+      "step": 1274
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0601023017902815e-05,
+      "loss": 0.0214,
+      "step": 1275
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0593350383631713e-05,
+      "loss": 0.0186,
+      "step": 1276
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0585677749360615e-05,
+      "loss": 0.0182,
+      "step": 1277
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0578005115089513e-05,
+      "loss": 0.0102,
+      "step": 1278
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0570332480818415e-05,
+      "loss": 1.1313,
+      "step": 1279
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0562659846547316e-05,
+      "loss": 0.0103,
+      "step": 1280
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0554987212276214e-05,
+      "loss": 0.0144,
+      "step": 1281
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0547314578005113e-05,
+      "loss": 0.02,
+      "step": 1282
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0539641943734018e-05,
+      "loss": 0.0292,
+      "step": 1283
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0531969309462916e-05,
+      "loss": 0.0105,
+      "step": 1284
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0524296675191814e-05,
+      "loss": 0.01,
+      "step": 1285
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0516624040920716e-05,
+      "loss": 1.1245,
+      "step": 1286
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0508951406649617e-05,
+      "loss": 0.8544,
+      "step": 1287
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0501278772378516e-05,
+      "loss": 0.0149,
+      "step": 1288
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0493606138107417e-05,
+      "loss": 0.0065,
+      "step": 1289
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.048593350383632e-05,
+      "loss": 0.0113,
+      "step": 1290
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0478260869565217e-05,
+      "loss": 0.0094,
+      "step": 1291
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.047058823529412e-05,
+      "loss": 0.0097,
+      "step": 1292
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0462915601023017e-05,
+      "loss": 0.0812,
+      "step": 1293
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0455242966751922e-05,
+      "loss": 0.0099,
+      "step": 1294
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.044757033248082e-05,
+      "loss": 0.013,
+      "step": 1295
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0439897698209718e-05,
+      "loss": 1.584,
+      "step": 1296
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.043222506393862e-05,
+      "loss": 0.0089,
+      "step": 1297
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.042455242966752e-05,
+      "loss": 0.0089,
+      "step": 1298
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.041687979539642e-05,
+      "loss": 0.0093,
+      "step": 1299
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.040920716112532e-05,
+      "loss": 0.0149,
+      "step": 1300
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.040153452685422e-05,
+      "loss": 1.1671,
+      "step": 1301
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.039386189258312e-05,
+      "loss": 0.0086,
+      "step": 1302
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.0386189258312023e-05,
+      "loss": 0.8177,
+      "step": 1303
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.037851662404092e-05,
+      "loss": 0.0128,
+      "step": 1304
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.037084398976982e-05,
+      "loss": 0.0167,
+      "step": 1305
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0363171355498724e-05,
+      "loss": 0.0085,
+      "step": 1306
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0355498721227622e-05,
+      "loss": 0.0085,
+      "step": 1307
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.034782608695652e-05,
+      "loss": 0.0166,
+      "step": 1308
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0340153452685422e-05,
+      "loss": 0.9448,
+      "step": 1309
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0332480818414324e-05,
+      "loss": 0.0082,
+      "step": 1310
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0324808184143222e-05,
+      "loss": 0.0088,
+      "step": 1311
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0317135549872124e-05,
+      "loss": 0.0082,
+      "step": 1312
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0309462915601025e-05,
+      "loss": 0.0146,
+      "step": 1313
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0301790281329924e-05,
+      "loss": 0.0123,
+      "step": 1314
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0294117647058825e-05,
+      "loss": 0.0101,
+      "step": 1315
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0286445012787723e-05,
+      "loss": 1.2167,
+      "step": 1316
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0278772378516625e-05,
+      "loss": 1.0578,
+      "step": 1317
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0271099744245527e-05,
+      "loss": 0.0163,
+      "step": 1318
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0263427109974425e-05,
+      "loss": 0.013,
+      "step": 1319
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0255754475703323e-05,
+      "loss": 0.0082,
+      "step": 1320
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0248081841432228e-05,
+      "loss": 0.0081,
+      "step": 1321
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0240409207161126e-05,
+      "loss": 0.0124,
+      "step": 1322
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0232736572890024e-05,
+      "loss": 0.0206,
+      "step": 1323
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0225063938618926e-05,
+      "loss": 0.0081,
+      "step": 1324
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0217391304347828e-05,
+      "loss": 0.0122,
+      "step": 1325
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0209718670076726e-05,
+      "loss": 0.0122,
+      "step": 1326
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0202046035805628e-05,
+      "loss": 0.0121,
+      "step": 1327
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0194373401534526e-05,
+      "loss": 1.224,
+      "step": 1328
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0186700767263427e-05,
+      "loss": 2.3502,
+      "step": 1329
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.017902813299233e-05,
+      "loss": 0.0242,
+      "step": 1330
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0171355498721227e-05,
+      "loss": 0.0064,
+      "step": 1331
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0163682864450125e-05,
+      "loss": 0.0186,
+      "step": 1332
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.015601023017903e-05,
+      "loss": 0.012,
+      "step": 1333
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.014833759590793e-05,
+      "loss": 0.0119,
+      "step": 1334
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.0140664961636827e-05,
+      "loss": 0.0159,
+      "step": 1335
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.013299232736573e-05,
+      "loss": 0.0199,
+      "step": 1336
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.012531969309463e-05,
+      "loss": 1.2138,
+      "step": 1337
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.011764705882353e-05,
+      "loss": 0.9491,
+      "step": 1338
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010997442455243e-05,
+      "loss": 0.0083,
+      "step": 1339
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010230179028133e-05,
+      "loss": 0.948,
+      "step": 1340
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.009462915601023e-05,
+      "loss": 0.0112,
+      "step": 1341
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.008695652173913e-05,
+      "loss": 0.8921,
+      "step": 1342
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007928388746803e-05,
+      "loss": 0.0194,
+      "step": 1343
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007161125319693e-05,
+      "loss": 1.0273,
+      "step": 1344
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0063938618925833e-05,
+      "loss": 0.013,
+      "step": 1345
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.005626598465473e-05,
+      "loss": 0.0103,
+      "step": 1346
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0048593350383633e-05,
+      "loss": 0.0114,
+      "step": 1347
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0040920716112534e-05,
+      "loss": 0.0199,
+      "step": 1348
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0033248081841433e-05,
+      "loss": 1.183,
+      "step": 1349
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0025575447570334e-05,
+      "loss": 0.0231,
+      "step": 1350
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0017902813299232e-05,
+      "loss": 1.1794,
+      "step": 1351
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0010230179028134e-05,
+      "loss": 0.0096,
+      "step": 1352
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 2.0002557544757036e-05,
+      "loss": 0.9049,
+      "step": 1353
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9994884910485934e-05,
+      "loss": 0.019,
+      "step": 1354
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9987212276214832e-05,
+      "loss": 0.0152,
+      "step": 1355
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9979539641943737e-05,
+      "loss": 0.8834,
+      "step": 1356
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9971867007672635e-05,
+      "loss": 0.01,
+      "step": 1357
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9964194373401533e-05,
+      "loss": 0.0152,
+      "step": 1358
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9956521739130435e-05,
+      "loss": 0.0161,
+      "step": 1359
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9948849104859337e-05,
+      "loss": 1.6778,
+      "step": 1360
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9941176470588235e-05,
+      "loss": 0.0214,
+      "step": 1361
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9933503836317137e-05,
+      "loss": 0.0156,
+      "step": 1362
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9925831202046038e-05,
+      "loss": 0.026,
+      "step": 1363
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9918158567774936e-05,
+      "loss": 0.0259,
+      "step": 1364
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9910485933503838e-05,
+      "loss": 0.0161,
+      "step": 1365
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9902813299232736e-05,
+      "loss": 0.0108,
+      "step": 1366
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9895140664961638e-05,
+      "loss": 0.0149,
+      "step": 1367
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.988746803069054e-05,
+      "loss": 0.0325,
+      "step": 1368
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9879795396419438e-05,
+      "loss": 0.0153,
+      "step": 1369
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9872122762148336e-05,
+      "loss": 1.7678,
+      "step": 1370
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.986445012787724e-05,
+      "loss": 0.0773,
+      "step": 1371
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.985677749360614e-05,
+      "loss": 0.0244,
+      "step": 1372
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.9849104859335037e-05,
+      "loss": 0.0225,
+      "step": 1373
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.984143222506394e-05,
+      "loss": 0.0254,
+      "step": 1374
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.983375959079284e-05,
+      "loss": 0.0205,
+      "step": 1375
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.982608695652174e-05,
+      "loss": 0.0096,
+      "step": 1376
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981841432225064e-05,
+      "loss": 0.0147,
+      "step": 1377
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981074168797954e-05,
+      "loss": 0.0296,
+      "step": 1378
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.980306905370844e-05,
+      "loss": 0.0227,
+      "step": 1379
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.9795396419437342e-05,
+      "loss": 0.899,
+      "step": 1380
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.978772378516624e-05,
+      "loss": 0.019,
+      "step": 1381
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9780051150895138e-05,
+      "loss": 0.0157,
+      "step": 1382
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9772378516624043e-05,
+      "loss": 1.1897,
+      "step": 1383
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.976470588235294e-05,
+      "loss": 0.014,
+      "step": 1384
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.975703324808184e-05,
+      "loss": 0.0208,
+      "step": 1385
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.974936061381074e-05,
+      "loss": 0.0093,
+      "step": 1386
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.9741687979539643e-05,
+      "loss": 0.0244,
+      "step": 1387
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.973401534526854e-05,
+      "loss": 0.028,
+      "step": 1388
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9726342710997443e-05,
+      "loss": 0.0093,
+      "step": 1389
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9718670076726344e-05,
+      "loss": 0.0184,
+      "step": 1390
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9710997442455243e-05,
+      "loss": 0.9462,
+      "step": 1391
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9703324808184144e-05,
+      "loss": 0.0138,
+      "step": 1392
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9695652173913042e-05,
+      "loss": 0.9561,
+      "step": 1393
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9687979539641944e-05,
+      "loss": 0.0152,
+      "step": 1394
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9680306905370846e-05,
+      "loss": 0.0178,
+      "step": 1395
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9672634271099744e-05,
+      "loss": 0.0166,
+      "step": 1396
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9664961636828642e-05,
+      "loss": 0.92,
+      "step": 1397
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9657289002557547e-05,
+      "loss": 0.0253,
+      "step": 1398
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9649616368286445e-05,
+      "loss": 0.0266,
+      "step": 1399
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9641943734015347e-05,
+      "loss": 0.0105,
+      "step": 1400
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9634271099744245e-05,
+      "loss": 0.0208,
+      "step": 1401
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9626598465473147e-05,
+      "loss": 0.8094,
+      "step": 1402
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.961892583120205e-05,
+      "loss": 0.0103,
+      "step": 1403
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9611253196930947e-05,
+      "loss": 0.0179,
+      "step": 1404
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9603580562659845e-05,
+      "loss": 0.0133,
+      "step": 1405
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.959590792838875e-05,
+      "loss": 0.0086,
+      "step": 1406
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9588235294117648e-05,
+      "loss": 0.0216,
+      "step": 1407
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9580562659846546e-05,
+      "loss": 0.0127,
+      "step": 1408
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9572890025575448e-05,
+      "loss": 0.0127,
+      "step": 1409
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.956521739130435e-05,
+      "loss": 0.0209,
+      "step": 1410
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9557544757033248e-05,
+      "loss": 0.0166,
+      "step": 1411
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954987212276215e-05,
+      "loss": 0.0123,
+      "step": 1412
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954219948849105e-05,
+      "loss": 0.0135,
+      "step": 1413
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.953452685421995e-05,
+      "loss": 0.1253,
+      "step": 1414
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.952685421994885e-05,
+      "loss": 0.0098,
+      "step": 1415
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951918158567775e-05,
+      "loss": 0.0091,
+      "step": 1416
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951150895140665e-05,
+      "loss": 0.9327,
+      "step": 1417
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.9503836317135552e-05,
+      "loss": 0.0119,
+      "step": 1418
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.949616368286445e-05,
+      "loss": 0.0153,
+      "step": 1419
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.948849104859335e-05,
+      "loss": 0.0123,
+      "step": 1420
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9480818414322254e-05,
+      "loss": 0.0194,
+      "step": 1421
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9473145780051152e-05,
+      "loss": 1.2311,
+      "step": 1422
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.946547314578005e-05,
+      "loss": 1.8519,
+      "step": 1423
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9457800511508952e-05,
+      "loss": 0.963,
+      "step": 1424
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9450127877237853e-05,
+      "loss": 0.0075,
+      "step": 1425
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.944245524296675e-05,
+      "loss": 0.0042,
+      "step": 1426
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9434782608695653e-05,
+      "loss": 0.0077,
+      "step": 1427
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.942710997442455e-05,
+      "loss": 0.0152,
+      "step": 1428
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9419437340153453e-05,
+      "loss": 0.0152,
+      "step": 1429
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9411764705882355e-05,
+      "loss": 0.019,
+      "step": 1430
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9404092071611253e-05,
+      "loss": 0.0081,
+      "step": 1431
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.939641943734015e-05,
+      "loss": 0.0076,
+      "step": 1432
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9388746803069056e-05,
+      "loss": 0.0086,
+      "step": 1433
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9381074168797954e-05,
+      "loss": 0.0146,
+      "step": 1434
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9373401534526853e-05,
+      "loss": 0.0074,
+      "step": 1435
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9365728900255754e-05,
+      "loss": 0.1007,
+      "step": 1436
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9358056265984656e-05,
+      "loss": 0.011,
+      "step": 1437
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9350383631713554e-05,
+      "loss": 0.0089,
+      "step": 1438
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9342710997442456e-05,
+      "loss": 0.0182,
+      "step": 1439
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9335038363171357e-05,
+      "loss": 0.0139,
+      "step": 1440
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9327365728900255e-05,
+      "loss": 2.2086,
+      "step": 1441
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9319693094629157e-05,
+      "loss": 0.0178,
+      "step": 1442
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9312020460358055e-05,
+      "loss": 0.0142,
+      "step": 1443
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9304347826086957e-05,
+      "loss": 0.0142,
+      "step": 1444
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.929667519181586e-05,
+      "loss": 0.0242,
+      "step": 1445
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9289002557544757e-05,
+      "loss": 0.0087,
+      "step": 1446
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9281329923273655e-05,
+      "loss": 0.0077,
+      "step": 1447
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.927365728900256e-05,
+      "loss": 0.0104,
+      "step": 1448
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9265984654731458e-05,
+      "loss": 0.0139,
+      "step": 1449
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9258312020460356e-05,
+      "loss": 1.2557,
+      "step": 1450
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9250639386189258e-05,
+      "loss": 0.0068,
+      "step": 1451
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.924296675191816e-05,
+      "loss": 0.0163,
+      "step": 1452
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.923529411764706e-05,
+      "loss": 0.9483,
+      "step": 1453
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.922762148337596e-05,
+      "loss": 1.2181,
+      "step": 1454
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.9219948849104858e-05,
+      "loss": 0.011,
+      "step": 1455
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.9212276214833763e-05,
+      "loss": 0.014,
+      "step": 1456
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.920460358056266e-05,
+      "loss": 0.0131,
+      "step": 1457
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.919693094629156e-05,
+      "loss": 0.0135,
+      "step": 1458
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.918925831202046e-05,
+      "loss": 0.007,
+      "step": 1459
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9181585677749362e-05,
+      "loss": 0.0072,
+      "step": 1460
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.917391304347826e-05,
+      "loss": 0.0135,
+      "step": 1461
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9166240409207162e-05,
+      "loss": 0.9908,
+      "step": 1462
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9158567774936064e-05,
+      "loss": 0.0103,
+      "step": 1463
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9150895140664962e-05,
+      "loss": 0.9867,
+      "step": 1464
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9143222506393864e-05,
+      "loss": 0.0138,
+      "step": 1465
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9135549872122762e-05,
+      "loss": 0.0104,
+      "step": 1466
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9127877237851663e-05,
+      "loss": 0.014,
+      "step": 1467
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9120204603580565e-05,
+      "loss": 0.0071,
+      "step": 1468
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9112531969309463e-05,
+      "loss": 0.0108,
+      "step": 1469
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.910485933503836e-05,
+      "loss": 0.0106,
+      "step": 1470
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9097186700767266e-05,
+      "loss": 0.0404,
+      "step": 1471
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9089514066496165e-05,
+      "loss": 2.2011,
+      "step": 1472
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9081841432225063e-05,
+      "loss": 0.0087,
+      "step": 1473
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9074168797953965e-05,
+      "loss": 0.0071,
+      "step": 1474
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9066496163682866e-05,
+      "loss": 0.0072,
+      "step": 1475
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9058823529411764e-05,
+      "loss": 0.0108,
+      "step": 1476
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9051150895140666e-05,
+      "loss": 0.0108,
+      "step": 1477
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9043478260869564e-05,
+      "loss": 0.0072,
+      "step": 1478
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9035805626598466e-05,
+      "loss": 0.0108,
+      "step": 1479
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9028132992327367e-05,
+      "loss": 0.0073,
+      "step": 1480
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9020460358056266e-05,
+      "loss": 0.9537,
+      "step": 1481
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9012787723785164e-05,
+      "loss": 0.0107,
+      "step": 1482
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.900511508951407e-05,
+      "loss": 0.0145,
+      "step": 1483
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8997442455242967e-05,
+      "loss": 0.007,
+      "step": 1484
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8989769820971865e-05,
+      "loss": 0.0071,
+      "step": 1485
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.898209718670077e-05,
+      "loss": 1.243,
+      "step": 1486
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.897442455242967e-05,
+      "loss": 0.011,
+      "step": 1487
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.8966751918158567e-05,
+      "loss": 0.0077,
+      "step": 1488
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895907928388747e-05,
+      "loss": 0.0073,
+      "step": 1489
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895140664961637e-05,
+      "loss": 1.2486,
+      "step": 1490
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8943734015345268e-05,
+      "loss": 0.0109,
+      "step": 1491
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.893606138107417e-05,
+      "loss": 0.0149,
+      "step": 1492
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8928388746803068e-05,
+      "loss": 0.0125,
+      "step": 1493
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.892071611253197e-05,
+      "loss": 0.0071,
+      "step": 1494
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.891304347826087e-05,
+      "loss": 0.0036,
+      "step": 1495
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.890537084398977e-05,
+      "loss": 0.0149,
+      "step": 1496
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8897698209718668e-05,
+      "loss": 0.015,
+      "step": 1497
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8890025575447573e-05,
+      "loss": 0.9536,
+      "step": 1498
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.888235294117647e-05,
+      "loss": 0.9997,
+      "step": 1499
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.887468030690537e-05,
+      "loss": 0.9621,
+      "step": 1500
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.886700767263427e-05,
+      "loss": 0.0113,
+      "step": 1501
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.8859335038363172e-05,
+      "loss": 0.2056,
+      "step": 1502
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.885166240409207e-05,
+      "loss": 0.0114,
+      "step": 1503
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8843989769820972e-05,
+      "loss": 0.016,
+      "step": 1504
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.883631713554987e-05,
+      "loss": 0.9175,
+      "step": 1505
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8828644501278775e-05,
+      "loss": 0.8959,
+      "step": 1506
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8820971867007674e-05,
+      "loss": 0.0074,
+      "step": 1507
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8813299232736572e-05,
+      "loss": 0.0117,
+      "step": 1508
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8805626598465474e-05,
+      "loss": 0.0075,
+      "step": 1509
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8797953964194375e-05,
+      "loss": 0.0074,
+      "step": 1510
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8790281329923273e-05,
+      "loss": 0.0577,
+      "step": 1511
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8782608695652175e-05,
+      "loss": 0.0128,
+      "step": 1512
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8774936061381077e-05,
+      "loss": 0.0073,
+      "step": 1513
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8767263427109975e-05,
+      "loss": 0.0074,
+      "step": 1514
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8759590792838876e-05,
+      "loss": 0.0073,
+      "step": 1515
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8751918158567775e-05,
+      "loss": 0.0088,
+      "step": 1516
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8744245524296676e-05,
+      "loss": 0.0078,
+      "step": 1517
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8736572890025578e-05,
+      "loss": 0.016,
+      "step": 1518
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8728900255754476e-05,
+      "loss": 1.3842,
+      "step": 1519
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8721227621483374e-05,
+      "loss": 1.2502,
+      "step": 1520
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.871355498721228e-05,
+      "loss": 0.0117,
+      "step": 1521
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8705882352941178e-05,
+      "loss": 0.0113,
+      "step": 1522
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8698209718670076e-05,
+      "loss": 0.0125,
+      "step": 1523
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8690537084398977e-05,
+      "loss": 0.0153,
+      "step": 1524
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.868286445012788e-05,
+      "loss": 0.0072,
+      "step": 1525
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8675191815856777e-05,
+      "loss": 0.0086,
+      "step": 1526
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.866751918158568e-05,
+      "loss": 0.0195,
+      "step": 1527
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.8659846547314577e-05,
+      "loss": 1.2511,
+      "step": 1528
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.865217391304348e-05,
+      "loss": 1.2407,
+      "step": 1529
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.864450127877238e-05,
+      "loss": 0.0112,
+      "step": 1530
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.863682864450128e-05,
+      "loss": 0.0073,
+      "step": 1531
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8629156010230177e-05,
+      "loss": 1.1911,
+      "step": 1532
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8621483375959082e-05,
+      "loss": 0.0168,
+      "step": 1533
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.861381074168798e-05,
+      "loss": 0.0128,
+      "step": 1534
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8606138107416878e-05,
+      "loss": 0.0116,
+      "step": 1535
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8598465473145783e-05,
+      "loss": 1.2286,
+      "step": 1536
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.859079283887468e-05,
+      "loss": 0.0158,
+      "step": 1537
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.858312020460358e-05,
+      "loss": 2.3735,
+      "step": 1538
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.857544757033248e-05,
+      "loss": 1.1972,
+      "step": 1539
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8567774936061383e-05,
+      "loss": 0.011,
+      "step": 1540
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.856010230179028e-05,
+      "loss": 0.0127,
+      "step": 1541
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8552429667519183e-05,
+      "loss": 0.0148,
+      "step": 1542
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.854475703324808e-05,
+      "loss": 0.0111,
+      "step": 1543
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8537084398976983e-05,
+      "loss": 0.0143,
+      "step": 1544
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8529411764705884e-05,
+      "loss": 0.0132,
+      "step": 1545
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8521739130434782e-05,
+      "loss": 0.0152,
+      "step": 1546
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.851406649616368e-05,
+      "loss": 0.0152,
+      "step": 1547
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8506393861892586e-05,
+      "loss": 0.0051,
+      "step": 1548
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8498721227621484e-05,
+      "loss": 0.9576,
+      "step": 1549
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8491048593350382e-05,
+      "loss": 0.0181,
+      "step": 1550
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8483375959079284e-05,
+      "loss": 0.011,
+      "step": 1551
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8475703324808185e-05,
+      "loss": 0.0205,
+      "step": 1552
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8468030690537083e-05,
+      "loss": 2.5631,
+      "step": 1553
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8460358056265985e-05,
+      "loss": 0.0102,
+      "step": 1554
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8452685421994883e-05,
+      "loss": 0.0147,
+      "step": 1555
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8445012787723788e-05,
+      "loss": 0.015,
+      "step": 1556
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8437340153452687e-05,
+      "loss": 0.0122,
+      "step": 1557
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8429667519181585e-05,
+      "loss": 0.0176,
+      "step": 1558
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8421994884910486e-05,
+      "loss": 0.0114,
+      "step": 1559
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8414322250639388e-05,
+      "loss": 0.0252,
+      "step": 1560
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8406649616368286e-05,
+      "loss": 0.013,
+      "step": 1561
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8398976982097188e-05,
+      "loss": 0.0142,
+      "step": 1562
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.839130434782609e-05,
+      "loss": 0.6496,
+      "step": 1563
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8383631713554988e-05,
+      "loss": 0.0149,
+      "step": 1564
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.837595907928389e-05,
+      "loss": 0.0104,
+      "step": 1565
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8368286445012787e-05,
+      "loss": 0.0129,
+      "step": 1566
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.836061381074169e-05,
+      "loss": 0.0197,
+      "step": 1567
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.835294117647059e-05,
+      "loss": 0.009,
+      "step": 1568
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.834526854219949e-05,
+      "loss": 0.0086,
+      "step": 1569
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.8337595907928387e-05,
+      "loss": 0.0087,
+      "step": 1570
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.8329923273657292e-05,
+      "loss": 0.0131,
+      "step": 1571
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.832225063938619e-05,
+      "loss": 1.2046,
+      "step": 1572
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.831457800511509e-05,
+      "loss": 0.0135,
+      "step": 1573
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.830690537084399e-05,
+      "loss": 0.0149,
+      "step": 1574
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.8299232736572892e-05,
+      "loss": 0.0077,
+      "step": 1575
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.829156010230179e-05,
+      "loss": 0.0143,
+      "step": 1576
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.828388746803069e-05,
+      "loss": 0.9806,
+      "step": 1577
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.827621483375959e-05,
+      "loss": 1.3783,
+      "step": 1578
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.826854219948849e-05,
+      "loss": 0.0075,
+      "step": 1579
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.8260869565217393e-05,
+      "loss": 0.0111,
+      "step": 1580
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.825319693094629e-05,
+      "loss": 0.0111,
+      "step": 1581
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.824552429667519e-05,
+      "loss": 0.0144,
+      "step": 1582
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8237851662404095e-05,
+      "loss": 0.0128,
+      "step": 1583
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8230179028132993e-05,
+      "loss": 1.2949,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.40147146582603455,
+      "eval_runtime": 58.0487,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.822250639386189e-05,
+      "loss": 0.0075,
+      "step": 1585
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8214833759590796e-05,
+      "loss": 0.0108,
+      "step": 1586
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8207161125319694e-05,
+      "loss": 0.011,
+      "step": 1587
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8199488491048592e-05,
+      "loss": 0.0148,
+      "step": 1588
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8191815856777494e-05,
+      "loss": 0.0077,
+      "step": 1589
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8184143222506396e-05,
+      "loss": 0.6674,
+      "step": 1590
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8176470588235294e-05,
+      "loss": 0.0109,
+      "step": 1591
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8168797953964195e-05,
+      "loss": 0.0108,
+      "step": 1592
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8161125319693094e-05,
+      "loss": 0.011,
+      "step": 1593
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8153452685421995e-05,
+      "loss": 0.0143,
+      "step": 1594
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8145780051150897e-05,
+      "loss": 0.9728,
+      "step": 1595
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8138107416879795e-05,
+      "loss": 0.0074,
+      "step": 1596
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8130434782608693e-05,
+      "loss": 0.0087,
+      "step": 1597
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.81227621483376e-05,
+      "loss": 0.0107,
+      "step": 1598
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8115089514066497e-05,
+      "loss": 0.0077,
+      "step": 1599
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8107416879795395e-05,
+      "loss": 0.0167,
+      "step": 1600
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8099744245524296e-05,
+      "loss": 1.219,
+      "step": 1601
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8092071611253198e-05,
+      "loss": 1.2244,
+      "step": 1602
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8084398976982096e-05,
+      "loss": 0.0108,
+      "step": 1603
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8076726342710998e-05,
+      "loss": 0.0071,
+      "step": 1604
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8069053708439896e-05,
+      "loss": 0.0109,
+      "step": 1605
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.8061381074168798e-05,
+      "loss": 0.0185,
+      "step": 1606
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.80537084398977e-05,
+      "loss": 1.2711,
+      "step": 1607
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.8046035805626598e-05,
+      "loss": 0.0113,
+      "step": 1608
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.80383631713555e-05,
+      "loss": 0.9575,
+      "step": 1609
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80306905370844e-05,
+      "loss": 1.1369,
+      "step": 1610
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80230179028133e-05,
+      "loss": 0.0114,
+      "step": 1611
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80153452685422e-05,
+      "loss": 1.578,
+      "step": 1612
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.8007672634271102e-05,
+      "loss": 0.0076,
+      "step": 1613
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.0128,
+      "step": 1614
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.7992327365728902e-05,
+      "loss": 0.0157,
+      "step": 1615
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.79846547314578e-05,
+      "loss": 0.0154,
+      "step": 1616
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.7976982097186702e-05,
+      "loss": 0.0125,
+      "step": 1617
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7969309462915604e-05,
+      "loss": 0.0085,
+      "step": 1618
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7961636828644502e-05,
+      "loss": 0.0167,
+      "step": 1619
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.79539641943734e-05,
+      "loss": 1.2275,
+      "step": 1620
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7946291560102305e-05,
+      "loss": 0.0115,
+      "step": 1621
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7938618925831203e-05,
+      "loss": 0.0083,
+      "step": 1622
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.79309462915601e-05,
+      "loss": 0.015,
+      "step": 1623
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7923273657289003e-05,
+      "loss": 1.1705,
+      "step": 1624
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7915601023017905e-05,
+      "loss": 0.0152,
+      "step": 1625
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7907928388746803e-05,
+      "loss": 2.0998,
+      "step": 1626
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7900255754475704e-05,
+      "loss": 0.0072,
+      "step": 1627
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7892583120204603e-05,
+      "loss": 0.0089,
+      "step": 1628
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7884910485933504e-05,
+      "loss": 0.0077,
+      "step": 1629
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7877237851662406e-05,
+      "loss": 0.0126,
+      "step": 1630
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7869565217391304e-05,
+      "loss": 1.2113,
+      "step": 1631
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7861892583120202e-05,
+      "loss": 0.0112,
+      "step": 1632
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7854219948849107e-05,
+      "loss": 0.0099,
+      "step": 1633
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7846547314578006e-05,
+      "loss": 0.0074,
+      "step": 1634
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7838874680306904e-05,
+      "loss": 0.0111,
+      "step": 1635
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.783120204603581e-05,
+      "loss": 0.0075,
+      "step": 1636
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7823529411764707e-05,
+      "loss": 1.2187,
+      "step": 1637
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7815856777493605e-05,
+      "loss": 0.0111,
+      "step": 1638
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7808184143222507e-05,
+      "loss": 0.0184,
+      "step": 1639
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.780051150895141e-05,
+      "loss": 0.0111,
+      "step": 1640
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7792838874680307e-05,
+      "loss": 0.0112,
+      "step": 1641
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.778516624040921e-05,
+      "loss": 0.0149,
+      "step": 1642
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.7777493606138107e-05,
+      "loss": 0.0151,
+      "step": 1643
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.7769820971867008e-05,
+      "loss": 0.0074,
+      "step": 1644
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.776214833759591e-05,
+      "loss": 0.9703,
+      "step": 1645
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.7754475703324808e-05,
+      "loss": 1.9364,
+      "step": 1646
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.7746803069053706e-05,
+      "loss": 0.011,
+      "step": 1647
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.773913043478261e-05,
+      "loss": 0.01,
+      "step": 1648
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.773145780051151e-05,
+      "loss": 0.0111,
+      "step": 1649
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.7723785166240408e-05,
+      "loss": 1.2373,
+      "step": 1650
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.771611253196931e-05,
+      "loss": 0.0074,
+      "step": 1651
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.770843989769821e-05,
+      "loss": 0.0116,
+      "step": 1652
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.770076726342711e-05,
+      "loss": 0.0117,
+      "step": 1653
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.769309462915601e-05,
+      "loss": 1.1636,
+      "step": 1654
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.768542199488491e-05,
+      "loss": 0.0075,
+      "step": 1655
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.767774936061381e-05,
+      "loss": 0.007,
+      "step": 1656
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.7670076726342712e-05,
+      "loss": 0.0174,
+      "step": 1657
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.766240409207161e-05,
+      "loss": 0.0113,
+      "step": 1658
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7654731457800512e-05,
+      "loss": 0.0084,
+      "step": 1659
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7647058823529414e-05,
+      "loss": 0.2994,
+      "step": 1660
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7639386189258312e-05,
+      "loss": 0.0213,
+      "step": 1661
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7631713554987213e-05,
+      "loss": 0.011,
+      "step": 1662
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7624040920716115e-05,
+      "loss": 1.2341,
+      "step": 1663
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7616368286445013e-05,
+      "loss": 1.1774,
+      "step": 1664
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7608695652173915e-05,
+      "loss": 0.0181,
+      "step": 1665
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7601023017902813e-05,
+      "loss": 0.0122,
+      "step": 1666
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7593350383631715e-05,
+      "loss": 0.0119,
+      "step": 1667
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7585677749360616e-05,
+      "loss": 0.0114,
+      "step": 1668
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7578005115089515e-05,
+      "loss": 0.0111,
+      "step": 1669
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7570332480818413e-05,
+      "loss": 0.011,
+      "step": 1670
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7562659846547318e-05,
+      "loss": 0.0113,
+      "step": 1671
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7554987212276216e-05,
+      "loss": 0.0161,
+      "step": 1672
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7547314578005114e-05,
+      "loss": 0.9491,
+      "step": 1673
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7539641943734016e-05,
+      "loss": 0.0108,
+      "step": 1674
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7531969309462917e-05,
+      "loss": 0.0114,
+      "step": 1675
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7524296675191816e-05,
+      "loss": 0.0107,
+      "step": 1676
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7516624040920717e-05,
+      "loss": 0.0171,
+      "step": 1677
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7508951406649616e-05,
+      "loss": 0.017,
+      "step": 1678
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7501278772378517e-05,
+      "loss": 1.939,
+      "step": 1679
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.749360613810742e-05,
+      "loss": 0.0151,
+      "step": 1680
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7485933503836317e-05,
+      "loss": 0.0109,
+      "step": 1681
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.7478260869565215e-05,
+      "loss": 0.0119,
+      "step": 1682
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.747058823529412e-05,
+      "loss": 1.2127,
+      "step": 1683
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.746291560102302e-05,
+      "loss": 0.0075,
+      "step": 1684
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.7455242966751917e-05,
+      "loss": 1.2283,
+      "step": 1685
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.744757033248082e-05,
+      "loss": 0.0107,
+      "step": 1686
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.743989769820972e-05,
+      "loss": 0.0141,
+      "step": 1687
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.7432225063938618e-05,
+      "loss": 1.2117,
+      "step": 1688
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.742455242966752e-05,
+      "loss": 0.8715,
+      "step": 1689
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.741687979539642e-05,
+      "loss": 0.0075,
+      "step": 1690
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.740920716112532e-05,
+      "loss": 1.2208,
+      "step": 1691
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.740153452685422e-05,
+      "loss": 0.0077,
+      "step": 1692
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.739386189258312e-05,
+      "loss": 0.0081,
+      "step": 1693
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.738618925831202e-05,
+      "loss": 0.0104,
+      "step": 1694
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.7378516624040923e-05,
+      "loss": 0.0162,
+      "step": 1695
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.737084398976982e-05,
+      "loss": 0.0173,
+      "step": 1696
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.736317135549872e-05,
+      "loss": 0.0194,
+      "step": 1697
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.7355498721227624e-05,
+      "loss": 0.0174,
+      "step": 1698
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.7347826086956522e-05,
+      "loss": 0.0081,
+      "step": 1699
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.734015345268542e-05,
+      "loss": 0.0121,
+      "step": 1700
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7332480818414322e-05,
+      "loss": 0.0137,
+      "step": 1701
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7324808184143224e-05,
+      "loss": 0.0117,
+      "step": 1702
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7317135549872122e-05,
+      "loss": 0.0133,
+      "step": 1703
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7309462915601024e-05,
+      "loss": 0.0175,
+      "step": 1704
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7301790281329922e-05,
+      "loss": 0.0076,
+      "step": 1705
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7294117647058823e-05,
+      "loss": 0.0179,
+      "step": 1706
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7286445012787725e-05,
+      "loss": 1.0541,
+      "step": 1707
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7278772378516623e-05,
+      "loss": 0.0077,
+      "step": 1708
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7271099744245525e-05,
+      "loss": 0.0079,
+      "step": 1709
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7263427109974426e-05,
+      "loss": 0.0102,
+      "step": 1710
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7255754475703325e-05,
+      "loss": 0.0158,
+      "step": 1711
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7248081841432223e-05,
+      "loss": 0.0107,
+      "step": 1712
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7240409207161128e-05,
+      "loss": 0.0159,
+      "step": 1713
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7232736572890026e-05,
+      "loss": 0.0125,
+      "step": 1714
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7225063938618928e-05,
+      "loss": 0.0081,
+      "step": 1715
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7217391304347826e-05,
+      "loss": 1.1946,
+      "step": 1716
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7209718670076728e-05,
+      "loss": 1.0298,
+      "step": 1717
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.720204603580563e-05,
+      "loss": 1.0207,
+      "step": 1718
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7194373401534527e-05,
+      "loss": 0.0152,
+      "step": 1719
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7186700767263426e-05,
+      "loss": 0.0187,
+      "step": 1720
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.717902813299233e-05,
+      "loss": 0.0134,
+      "step": 1721
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.717135549872123e-05,
+      "loss": 0.0173,
+      "step": 1722
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.7163682864450127e-05,
+      "loss": 1.0206,
+      "step": 1723
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.715601023017903e-05,
+      "loss": 0.0234,
+      "step": 1724
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.714833759590793e-05,
+      "loss": 0.0151,
+      "step": 1725
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.714066496163683e-05,
+      "loss": 0.0074,
+      "step": 1726
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.713299232736573e-05,
+      "loss": 0.0127,
+      "step": 1727
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.712531969309463e-05,
+      "loss": 0.8679,
+      "step": 1728
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.711764705882353e-05,
+      "loss": 0.0498,
+      "step": 1729
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.710997442455243e-05,
+      "loss": 0.0136,
+      "step": 1730
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.710230179028133e-05,
+      "loss": 1.2168,
+      "step": 1731
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.7094629156010228e-05,
+      "loss": 0.0105,
+      "step": 1732
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.7086956521739133e-05,
+      "loss": 0.0107,
+      "step": 1733
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.707928388746803e-05,
+      "loss": 0.0122,
+      "step": 1734
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.707161125319693e-05,
+      "loss": 0.0075,
+      "step": 1735
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.7063938618925834e-05,
+      "loss": 0.0607,
+      "step": 1736
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7056265984654733e-05,
+      "loss": 1.0181,
+      "step": 1737
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.704859335038363e-05,
+      "loss": 0.0095,
+      "step": 1738
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7040920716112533e-05,
+      "loss": 0.0128,
+      "step": 1739
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7033248081841434e-05,
+      "loss": 0.0096,
+      "step": 1740
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7025575447570332e-05,
+      "loss": 1.1781,
+      "step": 1741
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7017902813299234e-05,
+      "loss": 1.0132,
+      "step": 1742
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7010230179028132e-05,
+      "loss": 0.0097,
+      "step": 1743
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7002557544757034e-05,
+      "loss": 0.012,
+      "step": 1744
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6994884910485935e-05,
+      "loss": 0.0128,
+      "step": 1745
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6987212276214834e-05,
+      "loss": 0.0145,
+      "step": 1746
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6979539641943732e-05,
+      "loss": 0.7558,
+      "step": 1747
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6971867007672637e-05,
+      "loss": 0.0107,
+      "step": 1748
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6964194373401535e-05,
+      "loss": 1.0181,
+      "step": 1749
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6956521739130433e-05,
+      "loss": 1.7925,
+      "step": 1750
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6948849104859335e-05,
+      "loss": 0.0141,
+      "step": 1751
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6941176470588237e-05,
+      "loss": 0.0167,
+      "step": 1752
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6933503836317135e-05,
+      "loss": 0.0141,
+      "step": 1753
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6925831202046036e-05,
+      "loss": 0.0149,
+      "step": 1754
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6918158567774935e-05,
+      "loss": 0.0115,
+      "step": 1755
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6910485933503836e-05,
+      "loss": 0.0071,
+      "step": 1756
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6902813299232738e-05,
+      "loss": 0.0155,
+      "step": 1757
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6895140664961636e-05,
+      "loss": 0.0199,
+      "step": 1758
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6887468030690538e-05,
+      "loss": 0.0124,
+      "step": 1759
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.687979539641944e-05,
+      "loss": 0.0086,
+      "step": 1760
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.6872122762148337e-05,
+      "loss": 0.0147,
+      "step": 1761
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.6864450127877236e-05,
+      "loss": 0.0107,
+      "step": 1762
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.685677749360614e-05,
+      "loss": 1.025,
+      "step": 1763
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.684910485933504e-05,
+      "loss": 0.0105,
+      "step": 1764
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.6841432225063937e-05,
+      "loss": 0.0081,
+      "step": 1765
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.683375959079284e-05,
+      "loss": 0.009,
+      "step": 1766
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.682608695652174e-05,
+      "loss": 0.0109,
+      "step": 1767
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.6818414322250642e-05,
+      "loss": 0.008,
+      "step": 1768
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.681074168797954e-05,
+      "loss": 0.0132,
+      "step": 1769
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.680306905370844e-05,
+      "loss": 0.0119,
+      "step": 1770
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.6795396419437343e-05,
+      "loss": 0.0093,
+      "step": 1771
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.678772378516624e-05,
+      "loss": 1.0247,
+      "step": 1772
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.678005115089514e-05,
+      "loss": 1.024,
+      "step": 1773
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.677237851662404e-05,
+      "loss": 0.0106,
+      "step": 1774
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.6764705882352943e-05,
+      "loss": 0.0097,
+      "step": 1775
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.675703324808184e-05,
+      "loss": 0.0098,
+      "step": 1776
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6749360613810743e-05,
+      "loss": 1.0136,
+      "step": 1777
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.674168797953964e-05,
+      "loss": 0.0066,
+      "step": 1778
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6734015345268543e-05,
+      "loss": 0.1284,
+      "step": 1779
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6726342710997444e-05,
+      "loss": 0.9832,
+      "step": 1780
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6718670076726343e-05,
+      "loss": 0.0128,
+      "step": 1781
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6710997442455244e-05,
+      "loss": 0.0103,
+      "step": 1782
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6703324808184146e-05,
+      "loss": 0.0067,
+      "step": 1783
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6695652173913044e-05,
+      "loss": 0.0105,
+      "step": 1784
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6687979539641942e-05,
+      "loss": 0.0146,
+      "step": 1785
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6680306905370847e-05,
+      "loss": 0.0068,
+      "step": 1786
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6672634271099745e-05,
+      "loss": 0.0148,
+      "step": 1787
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6664961636828644e-05,
+      "loss": 0.0107,
+      "step": 1788
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6657289002557545e-05,
+      "loss": 2.4738,
+      "step": 1789
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6649616368286447e-05,
+      "loss": 0.9625,
+      "step": 1790
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6641943734015345e-05,
+      "loss": 0.2578,
+      "step": 1791
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6634271099744247e-05,
+      "loss": 0.0069,
+      "step": 1792
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6626598465473145e-05,
+      "loss": 0.011,
+      "step": 1793
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6618925831202047e-05,
+      "loss": 0.0066,
+      "step": 1794
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6611253196930948e-05,
+      "loss": 0.0069,
+      "step": 1795
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6603580562659846e-05,
+      "loss": 0.0071,
+      "step": 1796
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6595907928388745e-05,
+      "loss": 0.0068,
+      "step": 1797
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.658823529411765e-05,
+      "loss": 0.0232,
+      "step": 1798
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6580562659846548e-05,
+      "loss": 1.1318,
+      "step": 1799
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6572890025575446e-05,
+      "loss": 0.0101,
+      "step": 1800
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6565217391304348e-05,
+      "loss": 0.0187,
+      "step": 1801
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.655754475703325e-05,
+      "loss": 1.0718,
+      "step": 1802
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6549872122762148e-05,
+      "loss": 1.2349,
+      "step": 1803
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.654219948849105e-05,
+      "loss": 0.0148,
+      "step": 1804
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.6534526854219947e-05,
+      "loss": 0.0148,
+      "step": 1805
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.652685421994885e-05,
+      "loss": 0.0067,
+      "step": 1806
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.651918158567775e-05,
+      "loss": 0.0075,
+      "step": 1807
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.651150895140665e-05,
+      "loss": 0.0157,
+      "step": 1808
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.650383631713555e-05,
+      "loss": 0.0084,
+      "step": 1809
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.6496163682864452e-05,
+      "loss": 0.0123,
+      "step": 1810
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.648849104859335e-05,
+      "loss": 0.0065,
+      "step": 1811
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.648081841432225e-05,
+      "loss": 0.014,
+      "step": 1812
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.6473145780051154e-05,
+      "loss": 0.0152,
+      "step": 1813
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.6465473145780052e-05,
+      "loss": 0.0168,
+      "step": 1814
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.645780051150895e-05,
+      "loss": 0.0063,
+      "step": 1815
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.645012787723785e-05,
+      "loss": 0.6994,
+      "step": 1816
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.6442455242966753e-05,
+      "loss": 0.0132,
+      "step": 1817
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.643478260869565e-05,
+      "loss": 0.0103,
+      "step": 1818
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.6427109974424553e-05,
+      "loss": 0.013,
+      "step": 1819
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.641943734015345e-05,
+      "loss": 1.1813,
+      "step": 1820
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6411764705882356e-05,
+      "loss": 0.9809,
+      "step": 1821
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6404092071611254e-05,
+      "loss": 1.2339,
+      "step": 1822
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6396419437340153e-05,
+      "loss": 0.9744,
+      "step": 1823
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6388746803069054e-05,
+      "loss": 0.0098,
+      "step": 1824
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6381074168797956e-05,
+      "loss": 0.0062,
+      "step": 1825
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6373401534526854e-05,
+      "loss": 0.0151,
+      "step": 1826
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6365728900255756e-05,
+      "loss": 0.013,
+      "step": 1827
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6358056265984654e-05,
+      "loss": 0.0071,
+      "step": 1828
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6350383631713556e-05,
+      "loss": 0.0117,
+      "step": 1829
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6342710997442457e-05,
+      "loss": 0.0165,
+      "step": 1830
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6335038363171355e-05,
+      "loss": 0.1519,
+      "step": 1831
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6327365728900257e-05,
+      "loss": 1.0247,
+      "step": 1832
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.631969309462916e-05,
+      "loss": 0.0092,
+      "step": 1833
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6312020460358057e-05,
+      "loss": 0.0098,
+      "step": 1834
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6304347826086955e-05,
+      "loss": 1.1829,
+      "step": 1835
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.629667519181586e-05,
+      "loss": 0.006,
+      "step": 1836
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.628900255754476e-05,
+      "loss": 0.0109,
+      "step": 1837
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.6281329923273657e-05,
+      "loss": 0.6059,
+      "step": 1838
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.6273657289002558e-05,
+      "loss": 0.01,
+      "step": 1839
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.626598465473146e-05,
+      "loss": 0.0149,
+      "step": 1840
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.6258312020460358e-05,
+      "loss": 0.0152,
+      "step": 1841
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.625063938618926e-05,
+      "loss": 0.0171,
+      "step": 1842
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.6242966751918158e-05,
+      "loss": 0.0105,
+      "step": 1843
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.623529411764706e-05,
+      "loss": 0.0093,
+      "step": 1844
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.622762148337596e-05,
+      "loss": 1.4125,
+      "step": 1845
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.621994884910486e-05,
+      "loss": 0.0107,
+      "step": 1846
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.6212276214833757e-05,
+      "loss": 0.0133,
+      "step": 1847
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.6204603580562662e-05,
+      "loss": 0.0063,
+      "step": 1848
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.619693094629156e-05,
+      "loss": 0.0063,
+      "step": 1849
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.618925831202046e-05,
+      "loss": 0.009,
+      "step": 1850
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.618158567774936e-05,
+      "loss": 0.0137,
+      "step": 1851
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.6173913043478262e-05,
+      "loss": 0.0145,
+      "step": 1852
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.616624040920716e-05,
+      "loss": 0.0355,
+      "step": 1853
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.6158567774936062e-05,
+      "loss": 0.9601,
+      "step": 1854
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.615089514066496e-05,
+      "loss": 0.0135,
+      "step": 1855
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6143222506393862e-05,
+      "loss": 1.0129,
+      "step": 1856
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6135549872122763e-05,
+      "loss": 0.0339,
+      "step": 1857
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.612787723785166e-05,
+      "loss": 0.0131,
+      "step": 1858
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6120204603580563e-05,
+      "loss": 0.009,
+      "step": 1859
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6112531969309465e-05,
+      "loss": 0.0087,
+      "step": 1860
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6104859335038363e-05,
+      "loss": 0.012,
+      "step": 1861
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.609718670076726e-05,
+      "loss": 0.0088,
+      "step": 1862
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6089514066496166e-05,
+      "loss": 0.0117,
+      "step": 1863
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6081841432225065e-05,
+      "loss": 2.5214,
+      "step": 1864
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6074168797953963e-05,
+      "loss": 0.0107,
+      "step": 1865
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6066496163682864e-05,
+      "loss": 0.0101,
+      "step": 1866
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6058823529411766e-05,
+      "loss": 0.0118,
+      "step": 1867
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6051150895140664e-05,
+      "loss": 0.0087,
+      "step": 1868
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6043478260869566e-05,
+      "loss": 0.0122,
+      "step": 1869
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6035805626598464e-05,
+      "loss": 0.0084,
+      "step": 1870
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6028132992327366e-05,
+      "loss": 0.0104,
+      "step": 1871
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6020460358056267e-05,
+      "loss": 0.1587,
+      "step": 1872
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6012787723785166e-05,
+      "loss": 0.0057,
+      "step": 1873
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6005115089514067e-05,
+      "loss": 0.0085,
+      "step": 1874
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.599744245524297e-05,
+      "loss": 0.0056,
+      "step": 1875
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.5989769820971867e-05,
+      "loss": 1.2194,
+      "step": 1876
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.598209718670077e-05,
+      "loss": 0.0056,
+      "step": 1877
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.5974424552429667e-05,
+      "loss": 0.0057,
+      "step": 1878
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.596675191815857e-05,
+      "loss": 1.0276,
+      "step": 1879
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.595907928388747e-05,
+      "loss": 0.0083,
+      "step": 1880
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.5951406649616368e-05,
+      "loss": 0.0085,
+      "step": 1881
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.594373401534527e-05,
+      "loss": 0.0363,
+      "step": 1882
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.593606138107417e-05,
+      "loss": 0.0157,
+      "step": 1883
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.592838874680307e-05,
+      "loss": 0.0313,
+      "step": 1884
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.5920716112531968e-05,
+      "loss": 0.0062,
+      "step": 1885
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.5913043478260873e-05,
+      "loss": 0.0195,
+      "step": 1886
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.590537084398977e-05,
+      "loss": 0.0151,
+      "step": 1887
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.589769820971867e-05,
+      "loss": 0.0101,
+      "step": 1888
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.589002557544757e-05,
+      "loss": 1.1092,
+      "step": 1889
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.5882352941176473e-05,
+      "loss": 0.0323,
+      "step": 1890
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.587468030690537e-05,
+      "loss": 0.0086,
+      "step": 1891
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.5867007672634272e-05,
+      "loss": 0.0147,
+      "step": 1892
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.585933503836317e-05,
+      "loss": 0.0084,
+      "step": 1893
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.5851662404092072e-05,
+      "loss": 0.5596,
+      "step": 1894
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5843989769820974e-05,
+      "loss": 0.0114,
+      "step": 1895
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5836317135549872e-05,
+      "loss": 0.0083,
+      "step": 1896
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.582864450127877e-05,
+      "loss": 0.0082,
+      "step": 1897
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5820971867007675e-05,
+      "loss": 0.0146,
+      "step": 1898
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5813299232736574e-05,
+      "loss": 0.0081,
+      "step": 1899
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5805626598465472e-05,
+      "loss": 0.0081,
+      "step": 1900
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5797953964194373e-05,
+      "loss": 0.0116,
+      "step": 1901
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5790281329923275e-05,
+      "loss": 0.0084,
+      "step": 1902
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5782608695652173e-05,
+      "loss": 0.0108,
+      "step": 1903
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5774936061381075e-05,
+      "loss": 0.6146,
+      "step": 1904
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5767263427109973e-05,
+      "loss": 0.0162,
+      "step": 1905
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5759590792838875e-05,
+      "loss": 0.0134,
+      "step": 1906
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5751918158567776e-05,
+      "loss": 0.0047,
+      "step": 1907
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5744245524296674e-05,
+      "loss": 0.0045,
+      "step": 1908
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5736572890025576e-05,
+      "loss": 0.0087,
+      "step": 1909
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5728900255754478e-05,
+      "loss": 2.0913,
+      "step": 1910
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5721227621483376e-05,
+      "loss": 0.0075,
+      "step": 1911
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5713554987212274e-05,
+      "loss": 1.3387,
+      "step": 1912
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.570588235294118e-05,
+      "loss": 0.0102,
+      "step": 1913
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5698209718670077e-05,
+      "loss": 0.0076,
+      "step": 1914
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5690537084398976e-05,
+      "loss": 0.0131,
+      "step": 1915
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5682864450127877e-05,
+      "loss": 0.0079,
+      "step": 1916
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.567519181585678e-05,
+      "loss": 0.0102,
+      "step": 1917
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5667519181585677e-05,
+      "loss": 0.9554,
+      "step": 1918
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.565984654731458e-05,
+      "loss": 0.0088,
+      "step": 1919
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.5652173913043477e-05,
+      "loss": 1.0265,
+      "step": 1920
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.564450127877238e-05,
+      "loss": 0.0155,
+      "step": 1921
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.563682864450128e-05,
+      "loss": 0.0079,
+      "step": 1922
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.562915601023018e-05,
+      "loss": 1.0467,
+      "step": 1923
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.5621483375959077e-05,
+      "loss": 0.0132,
+      "step": 1924
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.561381074168798e-05,
+      "loss": 0.0077,
+      "step": 1925
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.560613810741688e-05,
+      "loss": 0.0064,
+      "step": 1926
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.559846547314578e-05,
+      "loss": 0.0104,
+      "step": 1927
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.559079283887468e-05,
+      "loss": 0.0096,
+      "step": 1928
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.558312020460358e-05,
+      "loss": 0.0049,
+      "step": 1929
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.5575447570332483e-05,
+      "loss": 0.0104,
+      "step": 1930
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.556777493606138e-05,
+      "loss": 1.0424,
+      "step": 1931
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5560102301790283e-05,
+      "loss": 0.0079,
+      "step": 1932
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5552429667519184e-05,
+      "loss": 0.0104,
+      "step": 1933
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5544757033248083e-05,
+      "loss": 0.0077,
+      "step": 1934
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.553708439897698e-05,
+      "loss": 0.005,
+      "step": 1935
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5529411764705886e-05,
+      "loss": 2.6544,
+      "step": 1936
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5521739130434784e-05,
+      "loss": 0.0107,
+      "step": 1937
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5514066496163682e-05,
+      "loss": 0.0124,
+      "step": 1938
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5506393861892584e-05,
+      "loss": 0.0049,
+      "step": 1939
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5498721227621485e-05,
+      "loss": 0.0078,
+      "step": 1940
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5491048593350384e-05,
+      "loss": 2.1009,
+      "step": 1941
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5483375959079285e-05,
+      "loss": 0.0116,
+      "step": 1942
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5475703324808183e-05,
+      "loss": 0.0049,
+      "step": 1943
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5468030690537085e-05,
+      "loss": 0.0079,
+      "step": 1944
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5460358056265987e-05,
+      "loss": 1.0307,
+      "step": 1945
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5452685421994885e-05,
+      "loss": 0.0147,
+      "step": 1946
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5445012787723783e-05,
+      "loss": 0.0084,
+      "step": 1947
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5437340153452688e-05,
+      "loss": 0.0262,
+      "step": 1948
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5429667519181586e-05,
+      "loss": 0.0115,
+      "step": 1949
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5421994884910485e-05,
+      "loss": 1.3221,
+      "step": 1950
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5414322250639386e-05,
+      "loss": 0.7929,
+      "step": 1951
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5406649616368288e-05,
+      "loss": 1.2069,
+      "step": 1952
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5398976982097186e-05,
+      "loss": 0.0081,
+      "step": 1953
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5391304347826088e-05,
+      "loss": 0.0069,
+      "step": 1954
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5383631713554986e-05,
+      "loss": 0.0097,
+      "step": 1955
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5375959079283887e-05,
+      "loss": 0.0082,
+      "step": 1956
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.536828644501279e-05,
+      "loss": 0.0061,
+      "step": 1957
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5360613810741687e-05,
+      "loss": 0.0082,
+      "step": 1958
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.535294117647059e-05,
+      "loss": 0.0113,
+      "step": 1959
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.534526854219949e-05,
+      "loss": 0.0098,
+      "step": 1960
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.533759590792839e-05,
+      "loss": 1.0221,
+      "step": 1961
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.5329923273657287e-05,
+      "loss": 0.0039,
+      "step": 1962
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.5322250639386192e-05,
+      "loss": 0.0127,
+      "step": 1963
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.531457800511509e-05,
+      "loss": 0.0051,
+      "step": 1964
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.530690537084399e-05,
+      "loss": 0.0086,
+      "step": 1965
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.529923273657289e-05,
+      "loss": 0.0096,
+      "step": 1966
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.529156010230179e-05,
+      "loss": 0.0164,
+      "step": 1967
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.528388746803069e-05,
+      "loss": 1.0127,
+      "step": 1968
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.527621483375959e-05,
+      "loss": 0.0149,
+      "step": 1969
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.526854219948849e-05,
+      "loss": 0.0085,
+      "step": 1970
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.526086956521739e-05,
+      "loss": 0.0116,
+      "step": 1971
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5253196930946293e-05,
+      "loss": 0.0168,
+      "step": 1972
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5245524296675191e-05,
+      "loss": 0.0052,
+      "step": 1973
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5237851662404091e-05,
+      "loss": 0.0052,
+      "step": 1974
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5230179028132993e-05,
+      "loss": 1.0013,
+      "step": 1975
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5222506393861893e-05,
+      "loss": 0.0122,
+      "step": 1976
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.521483375959079e-05,
+      "loss": 0.0087,
+      "step": 1977
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5207161125319694e-05,
+      "loss": 1.9624,
+      "step": 1978
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5199488491048592e-05,
+      "loss": 0.0088,
+      "step": 1979
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5191815856777496e-05,
+      "loss": 0.0596,
+      "step": 1980
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9090909090909091,
+      "eval_loss": 0.40464258193969727,
+      "eval_runtime": 58.0491,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1980
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5184143222506396e-05,
+      "loss": 0.009,
+      "step": 1981
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5176470588235294e-05,
+      "loss": 0.0063,
+      "step": 1982
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5168797953964197e-05,
+      "loss": 0.0092,
+      "step": 1983
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5161125319693095e-05,
+      "loss": 0.0092,
+      "step": 1984
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5153452685421995e-05,
+      "loss": 0.0259,
+      "step": 1985
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5145780051150897e-05,
+      "loss": 0.0093,
+      "step": 1986
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5138107416879797e-05,
+      "loss": 1.3347,
+      "step": 1987
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5130434782608695e-05,
+      "loss": 0.0135,
+      "step": 1988
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5122762148337598e-05,
+      "loss": 0.0134,
+      "step": 1989
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5115089514066497e-05,
+      "loss": 0.0051,
+      "step": 1990
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5107416879795396e-05,
+      "loss": 0.0106,
+      "step": 1991
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5099744245524298e-05,
+      "loss": 0.013,
+      "step": 1992
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5092071611253198e-05,
+      "loss": 0.0095,
+      "step": 1993
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5084398976982096e-05,
+      "loss": 0.0096,
+      "step": 1994
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5076726342711e-05,
+      "loss": 0.0112,
+      "step": 1995
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5069053708439898e-05,
+      "loss": 0.0137,
+      "step": 1996
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5061381074168798e-05,
+      "loss": 0.9611,
+      "step": 1997
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50537084398977e-05,
+      "loss": 0.009,
+      "step": 1998
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50460358056266e-05,
+      "loss": 0.0066,
+      "step": 1999
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.5038363171355497e-05,
+      "loss": 0.0091,
+      "step": 2000
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50306905370844e-05,
+      "loss": 0.9856,
+      "step": 2001
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5023017902813299e-05,
+      "loss": 0.0137,
+      "step": 2002
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5015345268542199e-05,
+      "loss": 0.0094,
+      "step": 2003
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.50076726342711e-05,
+      "loss": 0.9276,
+      "step": 2004
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.0051,
+      "step": 2005
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4992327365728902e-05,
+      "loss": 0.0097,
+      "step": 2006
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.49846547314578e-05,
+      "loss": 0.8053,
+      "step": 2007
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4976982097186702e-05,
+      "loss": 0.5635,
+      "step": 2008
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4969309462915602e-05,
+      "loss": 0.0099,
+      "step": 2009
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4961636828644502e-05,
+      "loss": 0.0108,
+      "step": 2010
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4953964194373402e-05,
+      "loss": 1.3337,
+      "step": 2011
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4946291560102303e-05,
+      "loss": 0.0098,
+      "step": 2012
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4938618925831201e-05,
+      "loss": 0.0099,
+      "step": 2013
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4930946291560103e-05,
+      "loss": 0.011,
+      "step": 2014
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4923273657289003e-05,
+      "loss": 0.0166,
+      "step": 2015
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4915601023017903e-05,
+      "loss": 1.3541,
+      "step": 2016
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4907928388746803e-05,
+      "loss": 0.0145,
+      "step": 2017
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4900255754475704e-05,
+      "loss": 0.0051,
+      "step": 2018
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4892583120204604e-05,
+      "loss": 0.0051,
+      "step": 2019
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4884910485933504e-05,
+      "loss": 0.0139,
+      "step": 2020
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4877237851662404e-05,
+      "loss": 0.0097,
+      "step": 2021
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4869565217391306e-05,
+      "loss": 0.9418,
+      "step": 2022
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4861892583120204e-05,
+      "loss": 0.9619,
+      "step": 2023
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4854219948849106e-05,
+      "loss": 1.3495,
+      "step": 2024
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4846547314578006e-05,
+      "loss": 0.0051,
+      "step": 2025
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4838874680306905e-05,
+      "loss": 1.3222,
+      "step": 2026
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4831202046035805e-05,
+      "loss": 1.1638,
+      "step": 2027
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4823529411764707e-05,
+      "loss": 0.009,
+      "step": 2028
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4815856777493605e-05,
+      "loss": 0.0059,
+      "step": 2029
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4808184143222507e-05,
+      "loss": 0.9946,
+      "step": 2030
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4800511508951408e-05,
+      "loss": 0.1291,
+      "step": 2031
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4792838874680307e-05,
+      "loss": 0.0089,
+      "step": 2032
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4785166240409208e-05,
+      "loss": 0.8113,
+      "step": 2033
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4777493606138108e-05,
+      "loss": 0.0109,
+      "step": 2034
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4769820971867008e-05,
+      "loss": 0.0061,
+      "step": 2035
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4762148337595908e-05,
+      "loss": 0.0157,
+      "step": 2036
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.475447570332481e-05,
+      "loss": 0.0104,
+      "step": 2037
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.4746803069053708e-05,
+      "loss": 0.012,
+      "step": 2038
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.473913043478261e-05,
+      "loss": 0.0106,
+      "step": 2039
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.473145780051151e-05,
+      "loss": 0.006,
+      "step": 2040
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.472378516624041e-05,
+      "loss": 0.0122,
+      "step": 2041
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.471611253196931e-05,
+      "loss": 0.0103,
+      "step": 2042
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.470843989769821e-05,
+      "loss": 0.9465,
+      "step": 2043
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.4700767263427109e-05,
+      "loss": 0.0149,
+      "step": 2044
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.469309462915601e-05,
+      "loss": 0.0146,
+      "step": 2045
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.468542199488491e-05,
+      "loss": 0.9225,
+      "step": 2046
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.4677749360613812e-05,
+      "loss": 0.0101,
+      "step": 2047
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.467007672634271e-05,
+      "loss": 0.0052,
+      "step": 2048
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.4662404092071612e-05,
+      "loss": 1.682,
+      "step": 2049
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4654731457800512e-05,
+      "loss": 0.9498,
+      "step": 2050
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4647058823529412e-05,
+      "loss": 0.0097,
+      "step": 2051
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4639386189258312e-05,
+      "loss": 0.0067,
+      "step": 2052
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4631713554987213e-05,
+      "loss": 0.0052,
+      "step": 2053
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4624040920716112e-05,
+      "loss": 0.0055,
+      "step": 2054
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4616368286445013e-05,
+      "loss": 0.0225,
+      "step": 2055
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4608695652173915e-05,
+      "loss": 0.0052,
+      "step": 2056
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4601023017902813e-05,
+      "loss": 0.0066,
+      "step": 2057
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4593350383631715e-05,
+      "loss": 0.0112,
+      "step": 2058
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4585677749360615e-05,
+      "loss": 1.3145,
+      "step": 2059
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4578005115089514e-05,
+      "loss": 0.0099,
+      "step": 2060
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4570332480818414e-05,
+      "loss": 0.0055,
+      "step": 2061
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4562659846547316e-05,
+      "loss": 0.0149,
+      "step": 2062
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4554987212276214e-05,
+      "loss": 0.0102,
+      "step": 2063
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4547314578005116e-05,
+      "loss": 0.0108,
+      "step": 2064
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4539641943734016e-05,
+      "loss": 0.0065,
+      "step": 2065
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4531969309462916e-05,
+      "loss": 0.0185,
+      "step": 2066
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4524296675191816e-05,
+      "loss": 0.9659,
+      "step": 2067
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4516624040920717e-05,
+      "loss": 0.0193,
+      "step": 2068
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4508951406649615e-05,
+      "loss": 0.0146,
+      "step": 2069
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4501278772378517e-05,
+      "loss": 0.0061,
+      "step": 2070
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4493606138107417e-05,
+      "loss": 0.0222,
+      "step": 2071
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4485933503836319e-05,
+      "loss": 0.0096,
+      "step": 2072
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4478260869565217e-05,
+      "loss": 0.0096,
+      "step": 2073
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4470588235294118e-05,
+      "loss": 0.005,
+      "step": 2074
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4462915601023018e-05,
+      "loss": 0.8301,
+      "step": 2075
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4455242966751918e-05,
+      "loss": 0.0049,
+      "step": 2076
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4447570332480818e-05,
+      "loss": 0.0097,
+      "step": 2077
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.443989769820972e-05,
+      "loss": 0.0062,
+      "step": 2078
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.4432225063938618e-05,
+      "loss": 1.3459,
+      "step": 2079
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.442455242966752e-05,
+      "loss": 0.0096,
+      "step": 2080
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4416879795396421e-05,
+      "loss": 1.2176,
+      "step": 2081
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.440920716112532e-05,
+      "loss": 0.0114,
+      "step": 2082
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4401534526854221e-05,
+      "loss": 0.0109,
+      "step": 2083
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4393861892583121e-05,
+      "loss": 0.0105,
+      "step": 2084
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.4386189258312021e-05,
+      "loss": 0.0107,
+      "step": 2085
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.437851662404092e-05,
+      "loss": 0.9442,
+      "step": 2086
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.4370843989769822e-05,
+      "loss": 0.01,
+      "step": 2087
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.436317135549872e-05,
+      "loss": 0.0099,
+      "step": 2088
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4355498721227622e-05,
+      "loss": 0.0167,
+      "step": 2089
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4347826086956522e-05,
+      "loss": 0.0051,
+      "step": 2090
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4340153452685422e-05,
+      "loss": 1.333,
+      "step": 2091
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4332480818414322e-05,
+      "loss": 0.0111,
+      "step": 2092
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4324808184143224e-05,
+      "loss": 0.0103,
+      "step": 2093
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4317135549872122e-05,
+      "loss": 0.0111,
+      "step": 2094
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4309462915601023e-05,
+      "loss": 0.0165,
+      "step": 2095
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4301790281329923e-05,
+      "loss": 0.0089,
+      "step": 2096
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4294117647058823e-05,
+      "loss": 0.0143,
+      "step": 2097
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4286445012787723e-05,
+      "loss": 0.4759,
+      "step": 2098
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4278772378516625e-05,
+      "loss": 0.0102,
+      "step": 2099
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4271099744245525e-05,
+      "loss": 0.0059,
+      "step": 2100
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4263427109974425e-05,
+      "loss": 0.0152,
+      "step": 2101
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4255754475703325e-05,
+      "loss": 0.0051,
+      "step": 2102
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4248081841432226e-05,
+      "loss": 0.0098,
+      "step": 2103
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4240409207161126e-05,
+      "loss": 0.0147,
+      "step": 2104
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4232736572890026e-05,
+      "loss": 0.0051,
+      "step": 2105
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4225063938618928e-05,
+      "loss": 1.3384,
+      "step": 2106
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4217391304347826e-05,
+      "loss": 0.0055,
+      "step": 2107
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4209718670076727e-05,
+      "loss": 0.0136,
+      "step": 2108
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4202046035805627e-05,
+      "loss": 0.0056,
+      "step": 2109
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4194373401534527e-05,
+      "loss": 0.0094,
+      "step": 2110
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4186700767263427e-05,
+      "loss": 0.0135,
+      "step": 2111
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4179028132992329e-05,
+      "loss": 0.0129,
+      "step": 2112
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4171355498721227e-05,
+      "loss": 0.0051,
+      "step": 2113
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4163682864450129e-05,
+      "loss": 0.0134,
+      "step": 2114
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4156010230179029e-05,
+      "loss": 0.0091,
+      "step": 2115
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4148337595907929e-05,
+      "loss": 0.0072,
+      "step": 2116
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.4140664961636828e-05,
+      "loss": 0.005,
+      "step": 2117
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.413299232736573e-05,
+      "loss": 0.0112,
+      "step": 2118
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.4125319693094628e-05,
+      "loss": 0.005,
+      "step": 2119
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.411764705882353e-05,
+      "loss": 0.0085,
+      "step": 2120
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.410997442455243e-05,
+      "loss": 0.0121,
+      "step": 2121
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.410230179028133e-05,
+      "loss": 0.0084,
+      "step": 2122
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.409462915601023e-05,
+      "loss": 0.0084,
+      "step": 2123
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.4086956521739131e-05,
+      "loss": 0.0083,
+      "step": 2124
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4079283887468031e-05,
+      "loss": 1.3402,
+      "step": 2125
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4071611253196931e-05,
+      "loss": 0.0124,
+      "step": 2126
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4063938618925831e-05,
+      "loss": 0.0082,
+      "step": 2127
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4056265984654733e-05,
+      "loss": 0.9694,
+      "step": 2128
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4048593350383633e-05,
+      "loss": 0.0081,
+      "step": 2129
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4040920716112532e-05,
+      "loss": 1.2102,
+      "step": 2130
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4033248081841434e-05,
+      "loss": 0.0113,
+      "step": 2131
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4025575447570332e-05,
+      "loss": 0.0114,
+      "step": 2132
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4017902813299234e-05,
+      "loss": 0.0252,
+      "step": 2133
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4010230179028134e-05,
+      "loss": 0.9785,
+      "step": 2134
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4002557544757034e-05,
+      "loss": 0.0061,
+      "step": 2135
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.3994884910485934e-05,
+      "loss": 0.0096,
+      "step": 2136
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3987212276214835e-05,
+      "loss": 0.0088,
+      "step": 2137
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3979539641943733e-05,
+      "loss": 0.0047,
+      "step": 2138
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3971867007672635e-05,
+      "loss": 0.0108,
+      "step": 2139
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3964194373401535e-05,
+      "loss": 0.0081,
+      "step": 2140
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3956521739130435e-05,
+      "loss": 0.0108,
+      "step": 2141
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3948849104859335e-05,
+      "loss": 0.0126,
+      "step": 2142
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3941176470588236e-05,
+      "loss": 0.0046,
+      "step": 2143
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3933503836317135e-05,
+      "loss": 0.0122,
+      "step": 2144
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3925831202046036e-05,
+      "loss": 0.0075,
+      "step": 2145
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3918158567774936e-05,
+      "loss": 0.0123,
+      "step": 2146
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3910485933503836e-05,
+      "loss": 0.0134,
+      "step": 2147
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3902813299232736e-05,
+      "loss": 0.0115,
+      "step": 2148
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3895140664961638e-05,
+      "loss": 0.0044,
+      "step": 2149
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3887468030690536e-05,
+      "loss": 0.0121,
+      "step": 2150
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3879795396419437e-05,
+      "loss": 0.0101,
+      "step": 2151
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3872122762148337e-05,
+      "loss": 0.0099,
+      "step": 2152
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3864450127877239e-05,
+      "loss": 0.0044,
+      "step": 2153
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3856777493606139e-05,
+      "loss": 0.0094,
+      "step": 2154
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3849104859335039e-05,
+      "loss": 0.0124,
+      "step": 2155
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.384143222506394e-05,
+      "loss": 0.0069,
+      "step": 2156
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.3833759590792839e-05,
+      "loss": 0.0068,
+      "step": 2157
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.382608695652174e-05,
+      "loss": 0.0094,
+      "step": 2158
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.381841432225064e-05,
+      "loss": 0.0067,
+      "step": 2159
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.381074168797954e-05,
+      "loss": 0.0044,
+      "step": 2160
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.380306905370844e-05,
+      "loss": 0.0041,
+      "step": 2161
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.3795396419437342e-05,
+      "loss": 0.0041,
+      "step": 2162
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.378772378516624e-05,
+      "loss": 0.009,
+      "step": 2163
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.3780051150895141e-05,
+      "loss": 0.0092,
+      "step": 2164
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3772378516624041e-05,
+      "loss": 0.0101,
+      "step": 2165
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3764705882352941e-05,
+      "loss": 0.0064,
+      "step": 2166
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3757033248081841e-05,
+      "loss": 0.0088,
+      "step": 2167
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3749360613810743e-05,
+      "loss": 0.0059,
+      "step": 2168
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3741687979539641e-05,
+      "loss": 0.0062,
+      "step": 2169
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3734015345268543e-05,
+      "loss": 0.0085,
+      "step": 2170
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3726342710997443e-05,
+      "loss": 0.0141,
+      "step": 2171
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3718670076726343e-05,
+      "loss": 0.0061,
+      "step": 2172
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3710997442455242e-05,
+      "loss": 0.0084,
+      "step": 2173
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3703324808184144e-05,
+      "loss": 0.0038,
+      "step": 2174
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3695652173913042e-05,
+      "loss": 0.006,
+      "step": 2175
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3687979539641944e-05,
+      "loss": 0.0039,
+      "step": 2176
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3680306905370844e-05,
+      "loss": 0.0205,
+      "step": 2177
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3672634271099745e-05,
+      "loss": 0.0062,
+      "step": 2178
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3664961636828645e-05,
+      "loss": 0.0021,
+      "step": 2179
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3657289002557545e-05,
+      "loss": 0.0058,
+      "step": 2180
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3649616368286447e-05,
+      "loss": 0.0101,
+      "step": 2181
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3641943734015345e-05,
+      "loss": 0.0036,
+      "step": 2182
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3634271099744247e-05,
+      "loss": 0.0037,
+      "step": 2183
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3626598465473147e-05,
+      "loss": 0.0036,
+      "step": 2184
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3618925831202047e-05,
+      "loss": 0.0036,
+      "step": 2185
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3611253196930946e-05,
+      "loss": 0.0035,
+      "step": 2186
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3603580562659848e-05,
+      "loss": 0.0059,
+      "step": 2187
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3595907928388746e-05,
+      "loss": 0.0056,
+      "step": 2188
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3588235294117648e-05,
+      "loss": 0.0139,
+      "step": 2189
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3580562659846548e-05,
+      "loss": 0.0558,
+      "step": 2190
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3572890025575448e-05,
+      "loss": 1.419,
+      "step": 2191
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3565217391304348e-05,
+      "loss": 0.0058,
+      "step": 2192
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.355754475703325e-05,
+      "loss": 0.0035,
+      "step": 2193
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3549872122762147e-05,
+      "loss": 0.0055,
+      "step": 2194
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3542199488491049e-05,
+      "loss": 1.2049,
+      "step": 2195
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3534526854219949e-05,
+      "loss": 1.3646,
+      "step": 2196
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3526854219948849e-05,
+      "loss": 1.3803,
+      "step": 2197
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3519181585677749e-05,
+      "loss": 0.0055,
+      "step": 2198
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.351150895140665e-05,
+      "loss": 0.0059,
+      "step": 2199
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.3503836317135549e-05,
+      "loss": 0.0056,
+      "step": 2200
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.349616368286445e-05,
+      "loss": 0.0097,
+      "step": 2201
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.348849104859335e-05,
+      "loss": 1.3497,
+      "step": 2202
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.348081841432225e-05,
+      "loss": 1.0989,
+      "step": 2203
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3473145780051152e-05,
+      "loss": 0.0117,
+      "step": 2204
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3465473145780052e-05,
+      "loss": 0.0038,
+      "step": 2205
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3457800511508953e-05,
+      "loss": 0.0078,
+      "step": 2206
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3450127877237852e-05,
+      "loss": 0.0058,
+      "step": 2207
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3442455242966753e-05,
+      "loss": 0.9775,
+      "step": 2208
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3434782608695653e-05,
+      "loss": 0.0037,
+      "step": 2209
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3427109974424553e-05,
+      "loss": 0.0079,
+      "step": 2210
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3419437340153453e-05,
+      "loss": 0.0663,
+      "step": 2211
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3411764705882354e-05,
+      "loss": 0.0059,
+      "step": 2212
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3404092071611253e-05,
+      "loss": 0.0039,
+      "step": 2213
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3396419437340154e-05,
+      "loss": 1.3169,
+      "step": 2214
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3388746803069054e-05,
+      "loss": 0.0061,
+      "step": 2215
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3381074168797954e-05,
+      "loss": 0.006,
+      "step": 2216
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3373401534526854e-05,
+      "loss": 0.004,
+      "step": 2217
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3365728900255756e-05,
+      "loss": 0.0045,
+      "step": 2218
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3358056265984654e-05,
+      "loss": 0.006,
+      "step": 2219
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3350383631713556e-05,
+      "loss": 0.0058,
+      "step": 2220
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3342710997442455e-05,
+      "loss": 0.014,
+      "step": 2221
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3335038363171355e-05,
+      "loss": 1.1674,
+      "step": 2222
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3327365728900255e-05,
+      "loss": 0.004,
+      "step": 2223
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3319693094629157e-05,
+      "loss": 0.0052,
+      "step": 2224
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3312020460358055e-05,
+      "loss": 0.0095,
+      "step": 2225
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3304347826086957e-05,
+      "loss": 0.0053,
+      "step": 2226
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3296675191815857e-05,
+      "loss": 0.0099,
+      "step": 2227
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3289002557544757e-05,
+      "loss": 1.3879,
+      "step": 2228
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3281329923273658e-05,
+      "loss": 0.0041,
+      "step": 2229
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3273657289002558e-05,
+      "loss": 0.006,
+      "step": 2230
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.326598465473146e-05,
+      "loss": 0.006,
+      "step": 2231
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.3258312020460358e-05,
+      "loss": 0.012,
+      "step": 2232
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.325063938618926e-05,
+      "loss": 0.0107,
+      "step": 2233
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.324296675191816e-05,
+      "loss": 0.0061,
+      "step": 2234
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.323529411764706e-05,
+      "loss": 0.0041,
+      "step": 2235
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.322762148337596e-05,
+      "loss": 0.006,
+      "step": 2236
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.3219948849104861e-05,
+      "loss": 0.0064,
+      "step": 2237
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.3212276214833759e-05,
+      "loss": 0.006,
+      "step": 2238
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.320460358056266e-05,
+      "loss": 0.0035,
+      "step": 2239
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.319693094629156e-05,
+      "loss": 0.8375,
+      "step": 2240
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.318925831202046e-05,
+      "loss": 0.004,
+      "step": 2241
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.318158567774936e-05,
+      "loss": 0.0077,
+      "step": 2242
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.3173913043478262e-05,
+      "loss": 0.006,
+      "step": 2243
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.316624040920716e-05,
+      "loss": 0.0041,
+      "step": 2244
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3158567774936062e-05,
+      "loss": 0.0034,
+      "step": 2245
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3150895140664962e-05,
+      "loss": 0.8219,
+      "step": 2246
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3143222506393862e-05,
+      "loss": 0.004,
+      "step": 2247
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3135549872122762e-05,
+      "loss": 0.004,
+      "step": 2248
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3127877237851663e-05,
+      "loss": 1.1328,
+      "step": 2249
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3120204603580562e-05,
+      "loss": 0.0039,
+      "step": 2250
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3112531969309463e-05,
+      "loss": 0.0283,
+      "step": 2251
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3104859335038363e-05,
+      "loss": 0.0058,
+      "step": 2252
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3097186700767263e-05,
+      "loss": 0.004,
+      "step": 2253
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3089514066496165e-05,
+      "loss": 0.0057,
+      "step": 2254
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3081841432225064e-05,
+      "loss": 0.0076,
+      "step": 2255
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3074168797953966e-05,
+      "loss": 0.004,
+      "step": 2256
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3066496163682864e-05,
+      "loss": 0.0076,
+      "step": 2257
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3058823529411766e-05,
+      "loss": 0.0091,
+      "step": 2258
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3051150895140666e-05,
+      "loss": 0.0038,
+      "step": 2259
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3043478260869566e-05,
+      "loss": 0.0057,
+      "step": 2260
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3035805626598466e-05,
+      "loss": 0.0057,
+      "step": 2261
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3028132992327367e-05,
+      "loss": 1.1251,
+      "step": 2262
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3020460358056266e-05,
+      "loss": 1.1222,
+      "step": 2263
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.3012787723785167e-05,
+      "loss": 0.0047,
+      "step": 2264
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.3005115089514067e-05,
+      "loss": 0.008,
+      "step": 2265
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.2997442455242967e-05,
+      "loss": 0.0061,
+      "step": 2266
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.2989769820971867e-05,
+      "loss": 1.394,
+      "step": 2267
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2982097186700768e-05,
+      "loss": 0.0039,
+      "step": 2268
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2974424552429667e-05,
+      "loss": 0.0103,
+      "step": 2269
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2966751918158568e-05,
+      "loss": 1.1008,
+      "step": 2270
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2959079283887468e-05,
+      "loss": 0.0051,
+      "step": 2271
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2951406649616368e-05,
+      "loss": 0.004,
+      "step": 2272
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2943734015345268e-05,
+      "loss": 0.78,
+      "step": 2273
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.293606138107417e-05,
+      "loss": 0.0069,
+      "step": 2274
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2928388746803068e-05,
+      "loss": 0.0094,
+      "step": 2275
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.292071611253197e-05,
+      "loss": 0.0067,
+      "step": 2276
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.291304347826087e-05,
+      "loss": 0.0041,
+      "step": 2277
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.290537084398977e-05,
+      "loss": 0.0079,
+      "step": 2278
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.2897698209718671e-05,
+      "loss": 0.014,
+      "step": 2279
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.2890025575447571e-05,
+      "loss": 1.3775,
+      "step": 2280
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.288235294117647e-05,
+      "loss": 0.0042,
+      "step": 2281
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.287468030690537e-05,
+      "loss": 1.1862,
+      "step": 2282
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2867007672634272e-05,
+      "loss": 0.0042,
+      "step": 2283
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2859335038363172e-05,
+      "loss": 0.0098,
+      "step": 2284
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2851662404092072e-05,
+      "loss": 0.0038,
+      "step": 2285
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2843989769820972e-05,
+      "loss": 0.0127,
+      "step": 2286
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2836317135549874e-05,
+      "loss": 0.0072,
+      "step": 2287
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2828644501278772e-05,
+      "loss": 0.0098,
+      "step": 2288
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2820971867007674e-05,
+      "loss": 0.0069,
+      "step": 2289
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2813299232736573e-05,
+      "loss": 0.9529,
+      "step": 2290
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2805626598465473e-05,
+      "loss": 0.0096,
+      "step": 2291
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2797953964194373e-05,
+      "loss": 0.0094,
+      "step": 2292
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2790281329923275e-05,
+      "loss": 1.0639,
+      "step": 2293
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2782608695652173e-05,
+      "loss": 0.0043,
+      "step": 2294
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2774936061381075e-05,
+      "loss": 0.0098,
+      "step": 2295
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2767263427109975e-05,
+      "loss": 0.0084,
+      "step": 2296
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2759590792838875e-05,
+      "loss": 0.0071,
+      "step": 2297
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2751918158567774e-05,
+      "loss": 0.0036,
+      "step": 2298
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2744245524296676e-05,
+      "loss": 0.0097,
+      "step": 2299
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2736572890025574e-05,
+      "loss": 0.0072,
+      "step": 2300
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2728900255754476e-05,
+      "loss": 0.0043,
+      "step": 2301
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2721227621483376e-05,
+      "loss": 1.3742,
+      "step": 2302
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2713554987212276e-05,
+      "loss": 0.0086,
+      "step": 2303
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2705882352941177e-05,
+      "loss": 0.0071,
+      "step": 2304
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2698209718670077e-05,
+      "loss": 0.007,
+      "step": 2305
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2690537084398977e-05,
+      "loss": 0.0099,
+      "step": 2306
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2682864450127877e-05,
+      "loss": 1.0437,
+      "step": 2307
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2675191815856779e-05,
+      "loss": 0.01,
+      "step": 2308
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2667519181585679e-05,
+      "loss": 0.1771,
+      "step": 2309
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2659846547314579e-05,
+      "loss": 0.0103,
+      "step": 2310
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.2652173913043479e-05,
+      "loss": 0.0104,
+      "step": 2311
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.264450127877238e-05,
+      "loss": 0.0077,
+      "step": 2312
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.2636828644501278e-05,
+      "loss": 0.0107,
+      "step": 2313
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.262915601023018e-05,
+      "loss": 0.0107,
+      "step": 2314
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.262148337595908e-05,
+      "loss": 0.0135,
+      "step": 2315
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.261381074168798e-05,
+      "loss": 0.0103,
+      "step": 2316
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.260613810741688e-05,
+      "loss": 0.0136,
+      "step": 2317
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.2598465473145781e-05,
+      "loss": 0.0045,
+      "step": 2318
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.259079283887468e-05,
+      "loss": 0.0104,
+      "step": 2319
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2583120204603581e-05,
+      "loss": 0.0073,
+      "step": 2320
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2575447570332481e-05,
+      "loss": 0.9265,
+      "step": 2321
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2567774936061381e-05,
+      "loss": 0.0101,
+      "step": 2322
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2560102301790281e-05,
+      "loss": 1.3704,
+      "step": 2323
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2552429667519183e-05,
+      "loss": 1.415,
+      "step": 2324
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.254475703324808e-05,
+      "loss": 0.0098,
+      "step": 2325
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2537084398976982e-05,
+      "loss": 0.0138,
+      "step": 2326
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2529411764705884e-05,
+      "loss": 0.0064,
+      "step": 2327
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2521739130434782e-05,
+      "loss": 0.0093,
+      "step": 2328
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2514066496163684e-05,
+      "loss": 0.0119,
+      "step": 2329
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2506393861892584e-05,
+      "loss": 0.0097,
+      "step": 2330
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2498721227621484e-05,
+      "loss": 0.0049,
+      "step": 2331
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2491048593350384e-05,
+      "loss": 0.0044,
+      "step": 2332
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2483375959079285e-05,
+      "loss": 0.0067,
+      "step": 2333
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2475703324808183e-05,
+      "loss": 0.0117,
+      "step": 2334
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2468030690537085e-05,
+      "loss": 0.0066,
+      "step": 2335
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2460358056265985e-05,
+      "loss": 1.0804,
+      "step": 2336
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2452685421994887e-05,
+      "loss": 0.0065,
+      "step": 2337
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2445012787723785e-05,
+      "loss": 0.0069,
+      "step": 2338
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2437340153452686e-05,
+      "loss": 0.009,
+      "step": 2339
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2429667519181586e-05,
+      "loss": 0.0067,
+      "step": 2340
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2421994884910486e-05,
+      "loss": 0.0089,
+      "step": 2341
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2414322250639386e-05,
+      "loss": 0.011,
+      "step": 2342
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2406649616368288e-05,
+      "loss": 0.0034,
+      "step": 2343
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2398976982097186e-05,
+      "loss": 1.0842,
+      "step": 2344
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2391304347826088e-05,
+      "loss": 0.0066,
+      "step": 2345
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2383631713554987e-05,
+      "loss": 0.0203,
+      "step": 2346
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2375959079283887e-05,
+      "loss": 0.0046,
+      "step": 2347
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2368286445012787e-05,
+      "loss": 0.9714,
+      "step": 2348
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2360613810741689e-05,
+      "loss": 0.0066,
+      "step": 2349
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2352941176470587e-05,
+      "loss": 0.0091,
+      "step": 2350
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.2345268542199489e-05,
+      "loss": 0.0066,
+      "step": 2351
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.233759590792839e-05,
+      "loss": 0.0081,
+      "step": 2352
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.2329923273657289e-05,
+      "loss": 0.006,
+      "step": 2353
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.232225063938619e-05,
+      "loss": 0.0066,
+      "step": 2354
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.231457800511509e-05,
+      "loss": 0.0114,
+      "step": 2355
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.230690537084399e-05,
+      "loss": 2.7382,
+      "step": 2356
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.229923273657289e-05,
+      "loss": 0.0089,
+      "step": 2357
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.2291560102301792e-05,
+      "loss": 0.0042,
+      "step": 2358
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.228388746803069e-05,
+      "loss": 2.3698,
+      "step": 2359
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2276214833759591e-05,
+      "loss": 0.0065,
+      "step": 2360
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2268542199488491e-05,
+      "loss": 1.0813,
+      "step": 2361
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2260869565217393e-05,
+      "loss": 0.0066,
+      "step": 2362
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2253196930946291e-05,
+      "loss": 0.0092,
+      "step": 2363
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2245524296675193e-05,
+      "loss": 1.3566,
+      "step": 2364
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2237851662404093e-05,
+      "loss": 0.0092,
+      "step": 2365
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2230179028132993e-05,
+      "loss": 1.0739,
+      "step": 2366
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2222506393861893e-05,
+      "loss": 0.0122,
+      "step": 2367
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2214833759590794e-05,
+      "loss": 0.0096,
+      "step": 2368
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2207161125319692e-05,
+      "loss": 0.0094,
+      "step": 2369
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2199488491048594e-05,
+      "loss": 0.0097,
+      "step": 2370
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2191815856777494e-05,
+      "loss": 0.0071,
+      "step": 2371
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2184143222506394e-05,
+      "loss": 0.0085,
+      "step": 2372
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2176470588235294e-05,
+      "loss": 0.0071,
+      "step": 2373
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2168797953964195e-05,
+      "loss": 0.0074,
+      "step": 2374
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2161125319693094e-05,
+      "loss": 0.0084,
+      "step": 2375
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2153452685421995e-05,
+      "loss": 0.008,
+      "step": 2376
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.9267676767676768,
+      "eval_loss": 0.3489837944507599,
+      "eval_runtime": 58.1837,
+      "eval_samples_per_second": 6.806,
+      "eval_steps_per_second": 1.702,
+      "step": 2376
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2145780051150897e-05,
+      "loss": 0.0065,
+      "step": 2377
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2138107416879795e-05,
+      "loss": 0.0097,
+      "step": 2378
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2130434782608697e-05,
+      "loss": 1.3744,
+      "step": 2379
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2122762148337597e-05,
+      "loss": 0.0043,
+      "step": 2380
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2115089514066496e-05,
+      "loss": 0.0071,
+      "step": 2381
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2107416879795396e-05,
+      "loss": 1.36,
+      "step": 2382
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2099744245524298e-05,
+      "loss": 0.0044,
+      "step": 2383
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2092071611253196e-05,
+      "loss": 0.0075,
+      "step": 2384
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2084398976982098e-05,
+      "loss": 0.0073,
+      "step": 2385
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.2076726342710998e-05,
+      "loss": 0.0099,
+      "step": 2386
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.2069053708439898e-05,
+      "loss": 0.0048,
+      "step": 2387
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.2061381074168798e-05,
+      "loss": 0.0074,
+      "step": 2388
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.20537084398977e-05,
+      "loss": 0.01,
+      "step": 2389
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.2046035805626599e-05,
+      "loss": 0.0062,
+      "step": 2390
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.2038363171355499e-05,
+      "loss": 0.0039,
+      "step": 2391
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.2030690537084399e-05,
+      "loss": 0.0054,
+      "step": 2392
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.20230179028133e-05,
+      "loss": 1.3442,
+      "step": 2393
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.2015345268542199e-05,
+      "loss": 0.0074,
+      "step": 2394
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.20076726342711e-05,
+      "loss": 0.0063,
+      "step": 2395
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.0075,
+      "step": 2396
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.19923273657289e-05,
+      "loss": 0.008,
+      "step": 2397
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.19846547314578e-05,
+      "loss": 0.0053,
+      "step": 2398
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.1976982097186702e-05,
+      "loss": 0.0049,
+      "step": 2399
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.19693094629156e-05,
+      "loss": 0.0086,
+      "step": 2400
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.1961636828644502e-05,
+      "loss": 0.0098,
+      "step": 2401
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1953964194373403e-05,
+      "loss": 0.0062,
+      "step": 2402
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1946291560102301e-05,
+      "loss": 0.0052,
+      "step": 2403
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1938618925831203e-05,
+      "loss": 0.005,
+      "step": 2404
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1930946291560103e-05,
+      "loss": 1.696,
+      "step": 2405
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1923273657289003e-05,
+      "loss": 0.0076,
+      "step": 2406
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1915601023017903e-05,
+      "loss": 1.1231,
+      "step": 2407
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1907928388746804e-05,
+      "loss": 0.0074,
+      "step": 2408
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1900255754475703e-05,
+      "loss": 0.0097,
+      "step": 2409
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1892583120204604e-05,
+      "loss": 0.0102,
+      "step": 2410
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1884910485933504e-05,
+      "loss": 0.005,
+      "step": 2411
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1877237851662404e-05,
+      "loss": 0.0074,
+      "step": 2412
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1869565217391304e-05,
+      "loss": 0.0071,
+      "step": 2413
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1861892583120206e-05,
+      "loss": 0.0093,
+      "step": 2414
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1854219948849106e-05,
+      "loss": 0.007,
+      "step": 2415
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1846547314578005e-05,
+      "loss": 0.0088,
+      "step": 2416
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1838874680306905e-05,
+      "loss": 0.0049,
+      "step": 2417
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1831202046035807e-05,
+      "loss": 1.2268,
+      "step": 2418
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1823529411764705e-05,
+      "loss": 0.0068,
+      "step": 2419
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1815856777493607e-05,
+      "loss": 0.0112,
+      "step": 2420
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1808184143222507e-05,
+      "loss": 0.0046,
+      "step": 2421
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1800511508951407e-05,
+      "loss": 0.0038,
+      "step": 2422
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1792838874680307e-05,
+      "loss": 0.0088,
+      "step": 2423
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1785166240409208e-05,
+      "loss": 0.0039,
+      "step": 2424
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1777493606138106e-05,
+      "loss": 0.0107,
+      "step": 2425
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.1769820971867008e-05,
+      "loss": 0.0066,
+      "step": 2426
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.176214833759591e-05,
+      "loss": 0.0086,
+      "step": 2427
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.1754475703324808e-05,
+      "loss": 0.0064,
+      "step": 2428
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.174680306905371e-05,
+      "loss": 0.0064,
+      "step": 2429
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.173913043478261e-05,
+      "loss": 0.0042,
+      "step": 2430
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.173145780051151e-05,
+      "loss": 0.0063,
+      "step": 2431
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.172378516624041e-05,
+      "loss": 1.0576,
+      "step": 2432
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.171611253196931e-05,
+      "loss": 0.0041,
+      "step": 2433
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.1708439897698209e-05,
+      "loss": 0.0102,
+      "step": 2434
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.170076726342711e-05,
+      "loss": 1.0312,
+      "step": 2435
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.169309462915601e-05,
+      "loss": 0.1209,
+      "step": 2436
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.168542199488491e-05,
+      "loss": 0.004,
+      "step": 2437
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.167774936061381e-05,
+      "loss": 0.006,
+      "step": 2438
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.1670076726342712e-05,
+      "loss": 0.0081,
+      "step": 2439
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.166240409207161e-05,
+      "loss": 1.1118,
+      "step": 2440
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.1654731457800512e-05,
+      "loss": 0.0059,
+      "step": 2441
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1647058823529412e-05,
+      "loss": 0.0101,
+      "step": 2442
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1639386189258313e-05,
+      "loss": 0.0081,
+      "step": 2443
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1631713554987212e-05,
+      "loss": 0.0032,
+      "step": 2444
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1624040920716113e-05,
+      "loss": 0.0059,
+      "step": 2445
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1616368286445013e-05,
+      "loss": 0.0105,
+      "step": 2446
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1608695652173913e-05,
+      "loss": 0.0038,
+      "step": 2447
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1601023017902813e-05,
+      "loss": 0.0104,
+      "step": 2448
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1593350383631715e-05,
+      "loss": 0.0058,
+      "step": 2449
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1585677749360613e-05,
+      "loss": 0.0058,
+      "step": 2450
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1578005115089514e-05,
+      "loss": 0.0064,
+      "step": 2451
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1570332480818416e-05,
+      "loss": 0.0058,
+      "step": 2452
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1562659846547314e-05,
+      "loss": 0.0077,
+      "step": 2453
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1554987212276216e-05,
+      "loss": 0.0528,
+      "step": 2454
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1547314578005116e-05,
+      "loss": 0.0078,
+      "step": 2455
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1539641943734016e-05,
+      "loss": 0.0023,
+      "step": 2456
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1531969309462916e-05,
+      "loss": 0.0019,
+      "step": 2457
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1524296675191817e-05,
+      "loss": 0.0056,
+      "step": 2458
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1516624040920715e-05,
+      "loss": 0.0116,
+      "step": 2459
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1508951406649617e-05,
+      "loss": 0.0037,
+      "step": 2460
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1501278772378517e-05,
+      "loss": 0.0056,
+      "step": 2461
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1493606138107417e-05,
+      "loss": 1.4099,
+      "step": 2462
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1485933503836317e-05,
+      "loss": 0.0055,
+      "step": 2463
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1478260869565218e-05,
+      "loss": 0.0018,
+      "step": 2464
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1470588235294117e-05,
+      "loss": 0.0081,
+      "step": 2465
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.1462915601023018e-05,
+      "loss": 0.0056,
+      "step": 2466
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.1455242966751918e-05,
+      "loss": 0.0048,
+      "step": 2467
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.144757033248082e-05,
+      "loss": 1.352,
+      "step": 2468
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.1439897698209718e-05,
+      "loss": 1.3368,
+      "step": 2469
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.143222506393862e-05,
+      "loss": 0.0075,
+      "step": 2470
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.142455242966752e-05,
+      "loss": 0.0046,
+      "step": 2471
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.141687979539642e-05,
+      "loss": 0.0036,
+      "step": 2472
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.140920716112532e-05,
+      "loss": 0.0055,
+      "step": 2473
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 1.1401534526854221e-05,
+      "loss": 0.0065,
+      "step": 2474
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 1.1393861892583121e-05,
+      "loss": 0.0055,
+      "step": 2475
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 1.138618925831202e-05,
+      "loss": 0.0057,
+      "step": 2476
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.1378516624040922e-05,
+      "loss": 0.0047,
+      "step": 2477
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.137084398976982e-05,
+      "loss": 0.0036,
+      "step": 2478
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.1363171355498722e-05,
+      "loss": 0.0061,
+      "step": 2479
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.1355498721227622e-05,
+      "loss": 0.0072,
+      "step": 2480
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1347826086956522e-05,
+      "loss": 0.0054,
+      "step": 2481
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1340153452685422e-05,
+      "loss": 0.0072,
+      "step": 2482
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1332480818414324e-05,
+      "loss": 0.0041,
+      "step": 2483
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1324808184143222e-05,
+      "loss": 0.0051,
+      "step": 2484
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1317135549872123e-05,
+      "loss": 1.4203,
+      "step": 2485
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1309462915601023e-05,
+      "loss": 0.0088,
+      "step": 2486
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1301790281329923e-05,
+      "loss": 0.0106,
+      "step": 2487
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1294117647058823e-05,
+      "loss": 0.0071,
+      "step": 2488
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1286445012787725e-05,
+      "loss": 0.0053,
+      "step": 2489
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1278772378516623e-05,
+      "loss": 1.1147,
+      "step": 2490
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1271099744245525e-05,
+      "loss": 0.007,
+      "step": 2491
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1263427109974425e-05,
+      "loss": 0.007,
+      "step": 2492
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1255754475703324e-05,
+      "loss": 0.0035,
+      "step": 2493
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1248081841432224e-05,
+      "loss": 0.0052,
+      "step": 2494
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1240409207161126e-05,
+      "loss": 0.0064,
+      "step": 2495
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1232736572890026e-05,
+      "loss": 0.0046,
+      "step": 2496
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1225063938618926e-05,
+      "loss": 0.0052,
+      "step": 2497
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1217391304347826e-05,
+      "loss": 0.0045,
+      "step": 2498
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1209718670076727e-05,
+      "loss": 0.0052,
+      "step": 2499
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1202046035805627e-05,
+      "loss": 0.0035,
+      "step": 2500
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1194373401534527e-05,
+      "loss": 0.0086,
+      "step": 2501
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1186700767263429e-05,
+      "loss": 0.0068,
+      "step": 2502
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1179028132992327e-05,
+      "loss": 0.0051,
+      "step": 2503
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1171355498721229e-05,
+      "loss": 0.0026,
+      "step": 2504
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.1163682864450129e-05,
+      "loss": 0.0034,
+      "step": 2505
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.1156010230179029e-05,
+      "loss": 0.0052,
+      "step": 2506
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.1148337595907928e-05,
+      "loss": 0.0051,
+      "step": 2507
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.114066496163683e-05,
+      "loss": 0.005,
+      "step": 2508
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.1132992327365728e-05,
+      "loss": 0.005,
+      "step": 2509
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.112531969309463e-05,
+      "loss": 0.0043,
+      "step": 2510
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.111764705882353e-05,
+      "loss": 0.0065,
+      "step": 2511
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.110997442455243e-05,
+      "loss": 0.0039,
+      "step": 2512
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.110230179028133e-05,
+      "loss": 1.4148,
+      "step": 2513
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.1094629156010231e-05,
+      "loss": 0.0049,
+      "step": 2514
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.108695652173913e-05,
+      "loss": 0.0058,
+      "step": 2515
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.1079283887468031e-05,
+      "loss": 1.3611,
+      "step": 2516
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.1071611253196931e-05,
+      "loss": 0.0057,
+      "step": 2517
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.1063938618925831e-05,
+      "loss": 0.0049,
+      "step": 2518
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.105626598465473e-05,
+      "loss": 0.0049,
+      "step": 2519
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.1048593350383632e-05,
+      "loss": 0.0026,
+      "step": 2520
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1040920716112532e-05,
+      "loss": 0.0026,
+      "step": 2521
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1033248081841432e-05,
+      "loss": 0.4305,
+      "step": 2522
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1025575447570332e-05,
+      "loss": 0.0069,
+      "step": 2523
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1017902813299234e-05,
+      "loss": 0.005,
+      "step": 2524
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.1010230179028134e-05,
+      "loss": 1.3999,
+      "step": 2525
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.1002557544757034e-05,
+      "loss": 0.1888,
+      "step": 2526
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.0994884910485935e-05,
+      "loss": 0.0049,
+      "step": 2527
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.0987212276214833e-05,
+      "loss": 0.8899,
+      "step": 2528
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0979539641943735e-05,
+      "loss": 1.0806,
+      "step": 2529
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0971867007672635e-05,
+      "loss": 0.0083,
+      "step": 2530
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0964194373401535e-05,
+      "loss": 0.0034,
+      "step": 2531
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0956521739130435e-05,
+      "loss": 0.005,
+      "step": 2532
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0948849104859336e-05,
+      "loss": 0.0051,
+      "step": 2533
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0941176470588235e-05,
+      "loss": 0.0057,
+      "step": 2534
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0933503836317136e-05,
+      "loss": 0.0058,
+      "step": 2535
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0925831202046036e-05,
+      "loss": 0.0065,
+      "step": 2536
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0918158567774936e-05,
+      "loss": 1.1058,
+      "step": 2537
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0910485933503836e-05,
+      "loss": 0.0052,
+      "step": 2538
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0902813299232738e-05,
+      "loss": 0.0082,
+      "step": 2539
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0895140664961636e-05,
+      "loss": 0.005,
+      "step": 2540
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0887468030690537e-05,
+      "loss": 0.005,
+      "step": 2541
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0879795396419437e-05,
+      "loss": 0.005,
+      "step": 2542
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0872122762148337e-05,
+      "loss": 0.0034,
+      "step": 2543
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0864450127877237e-05,
+      "loss": 0.0066,
+      "step": 2544
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0856777493606139e-05,
+      "loss": 0.0052,
+      "step": 2545
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0849104859335037e-05,
+      "loss": 0.0039,
+      "step": 2546
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0841432225063939e-05,
+      "loss": 0.0039,
+      "step": 2547
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0833759590792839e-05,
+      "loss": 0.0064,
+      "step": 2548
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.082608695652174e-05,
+      "loss": 0.0064,
+      "step": 2549
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.081841432225064e-05,
+      "loss": 0.0052,
+      "step": 2550
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.081074168797954e-05,
+      "loss": 0.0082,
+      "step": 2551
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.0803069053708442e-05,
+      "loss": 0.0035,
+      "step": 2552
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.079539641943734e-05,
+      "loss": 0.0064,
+      "step": 2553
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.0787723785166241e-05,
+      "loss": 0.0064,
+      "step": 2554
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.0780051150895141e-05,
+      "loss": 0.1541,
+      "step": 2555
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.0772378516624041e-05,
+      "loss": 2.4749,
+      "step": 2556
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0764705882352941e-05,
+      "loss": 0.0028,
+      "step": 2557
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0757033248081843e-05,
+      "loss": 0.0034,
+      "step": 2558
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0749360613810741e-05,
+      "loss": 0.0061,
+      "step": 2559
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0741687979539643e-05,
+      "loss": 0.0032,
+      "step": 2560
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0734015345268543e-05,
+      "loss": 0.0055,
+      "step": 2561
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0726342710997443e-05,
+      "loss": 0.0046,
+      "step": 2562
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0718670076726342e-05,
+      "loss": 1.185,
+      "step": 2563
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0710997442455244e-05,
+      "loss": 0.0062,
+      "step": 2564
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0703324808184142e-05,
+      "loss": 0.0024,
+      "step": 2565
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0695652173913044e-05,
+      "loss": 0.0032,
+      "step": 2566
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0687979539641944e-05,
+      "loss": 0.0047,
+      "step": 2567
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0680306905370844e-05,
+      "loss": 0.0047,
+      "step": 2568
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0672634271099744e-05,
+      "loss": 0.0047,
+      "step": 2569
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0664961636828645e-05,
+      "loss": 0.0065,
+      "step": 2570
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0657289002557543e-05,
+      "loss": 0.0061,
+      "step": 2571
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0649616368286445e-05,
+      "loss": 0.0048,
+      "step": 2572
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 1.0641943734015345e-05,
+      "loss": 0.0015,
+      "step": 2573
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 1.0634271099744247e-05,
+      "loss": 0.0046,
+      "step": 2574
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 1.0626598465473147e-05,
+      "loss": 0.0033,
+      "step": 2575
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0618925831202046e-05,
+      "loss": 0.0062,
+      "step": 2576
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0611253196930948e-05,
+      "loss": 1.4243,
+      "step": 2577
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0603580562659846e-05,
+      "loss": 0.012,
+      "step": 2578
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0595907928388748e-05,
+      "loss": 2.3898,
+      "step": 2579
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.0588235294117648e-05,
+      "loss": 0.0032,
+      "step": 2580
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.0580562659846548e-05,
+      "loss": 0.0032,
+      "step": 2581
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.0572890025575448e-05,
+      "loss": 0.0061,
+      "step": 2582
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.056521739130435e-05,
+      "loss": 0.0047,
+      "step": 2583
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0557544757033247e-05,
+      "loss": 0.0047,
+      "step": 2584
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0549872122762149e-05,
+      "loss": 0.2081,
+      "step": 2585
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0542199488491049e-05,
+      "loss": 0.0053,
+      "step": 2586
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0534526854219949e-05,
+      "loss": 0.0047,
+      "step": 2587
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.0526854219948849e-05,
+      "loss": 1.3858,
+      "step": 2588
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.051918158567775e-05,
+      "loss": 0.0062,
+      "step": 2589
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.0511508951406649e-05,
+      "loss": 0.0041,
+      "step": 2590
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.050383631713555e-05,
+      "loss": 1.1796,
+      "step": 2591
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.049616368286445e-05,
+      "loss": 1.0994,
+      "step": 2592
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.048849104859335e-05,
+      "loss": 0.8511,
+      "step": 2593
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.048081841432225e-05,
+      "loss": 1.2895,
+      "step": 2594
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.0473145780051152e-05,
+      "loss": 0.0035,
+      "step": 2595
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.046547314578005e-05,
+      "loss": 0.0065,
+      "step": 2596
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.0457800511508952e-05,
+      "loss": 1.239,
+      "step": 2597
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.0450127877237851e-05,
+      "loss": 1.1663,
+      "step": 2598
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.0442455242966751e-05,
+      "loss": 0.0051,
+      "step": 2599
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0434782608695653e-05,
+      "loss": 1.1103,
+      "step": 2600
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0427109974424553e-05,
+      "loss": 1.4103,
+      "step": 2601
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0419437340153454e-05,
+      "loss": 0.0053,
+      "step": 2602
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0411764705882353e-05,
+      "loss": 0.0072,
+      "step": 2603
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0404092071611254e-05,
+      "loss": 0.0083,
+      "step": 2604
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0396419437340154e-05,
+      "loss": 0.0039,
+      "step": 2605
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0388746803069054e-05,
+      "loss": 0.0049,
+      "step": 2606
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0381074168797954e-05,
+      "loss": 0.0098,
+      "step": 2607
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0373401534526856e-05,
+      "loss": 0.0057,
+      "step": 2608
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0365728900255754e-05,
+      "loss": 0.004,
+      "step": 2609
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0358056265984656e-05,
+      "loss": 0.0098,
+      "step": 2610
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0350383631713555e-05,
+      "loss": 0.0059,
+      "step": 2611
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0342710997442455e-05,
+      "loss": 0.0065,
+      "step": 2612
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0335038363171355e-05,
+      "loss": 0.0059,
+      "step": 2613
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0327365728900257e-05,
+      "loss": 0.0112,
+      "step": 2614
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0319693094629155e-05,
+      "loss": 0.0061,
+      "step": 2615
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0312020460358057e-05,
+      "loss": 0.0039,
+      "step": 2616
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0304347826086957e-05,
+      "loss": 0.0083,
+      "step": 2617
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0296675191815857e-05,
+      "loss": 0.0057,
+      "step": 2618
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0289002557544756e-05,
+      "loss": 1.3969,
+      "step": 2619
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0281329923273658e-05,
+      "loss": 0.0065,
+      "step": 2620
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0273657289002556e-05,
+      "loss": 1.0742,
+      "step": 2621
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0265984654731458e-05,
+      "loss": 0.0039,
+      "step": 2622
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0258312020460358e-05,
+      "loss": 0.0065,
+      "step": 2623
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.0250639386189258e-05,
+      "loss": 0.005,
+      "step": 2624
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.024296675191816e-05,
+      "loss": 0.0077,
+      "step": 2625
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.023529411764706e-05,
+      "loss": 1.2389,
+      "step": 2626
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.0227621483375961e-05,
+      "loss": 0.0076,
+      "step": 2627
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.0219948849104859e-05,
+      "loss": 1.034,
+      "step": 2628
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.021227621483376e-05,
+      "loss": 0.0058,
+      "step": 2629
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.020460358056266e-05,
+      "loss": 0.0059,
+      "step": 2630
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.019693094629156e-05,
+      "loss": 0.0079,
+      "step": 2631
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.018925831202046e-05,
+      "loss": 0.0043,
+      "step": 2632
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.0181585677749362e-05,
+      "loss": 0.004,
+      "step": 2633
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.017391304347826e-05,
+      "loss": 0.0075,
+      "step": 2634
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.0166240409207162e-05,
+      "loss": 0.0076,
+      "step": 2635
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0158567774936062e-05,
+      "loss": 0.0102,
+      "step": 2636
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0150895140664962e-05,
+      "loss": 1.1332,
+      "step": 2637
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0143222506393862e-05,
+      "loss": 0.0096,
+      "step": 2638
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0135549872122763e-05,
+      "loss": 0.0092,
+      "step": 2639
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0127877237851662e-05,
+      "loss": 1.3919,
+      "step": 2640
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0120204603580563e-05,
+      "loss": 0.0058,
+      "step": 2641
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0112531969309463e-05,
+      "loss": 1.0521,
+      "step": 2642
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0104859335038363e-05,
+      "loss": 0.0045,
+      "step": 2643
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0097186700767263e-05,
+      "loss": 0.0059,
+      "step": 2644
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0089514066496164e-05,
+      "loss": 0.005,
+      "step": 2645
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0081841432225063e-05,
+      "loss": 0.0075,
+      "step": 2646
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0074168797953964e-05,
+      "loss": 0.0403,
+      "step": 2647
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0066496163682864e-05,
+      "loss": 0.0079,
+      "step": 2648
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0058823529411764e-05,
+      "loss": 0.0062,
+      "step": 2649
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0051150895140666e-05,
+      "loss": 0.0061,
+      "step": 2650
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0043478260869566e-05,
+      "loss": 0.0064,
+      "step": 2651
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0035805626598466e-05,
+      "loss": 0.0077,
+      "step": 2652
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0028132992327366e-05,
+      "loss": 0.0884,
+      "step": 2653
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0020460358056267e-05,
+      "loss": 0.0062,
+      "step": 2654
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0012787723785167e-05,
+      "loss": 1.1006,
+      "step": 2655
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 1.0005115089514067e-05,
+      "loss": 0.0109,
+      "step": 2656
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 9.997442455242967e-06,
+      "loss": 0.0059,
+      "step": 2657
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 9.989769820971869e-06,
+      "loss": 0.0084,
+      "step": 2658
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 9.982097186700767e-06,
+      "loss": 0.0058,
+      "step": 2659
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.974424552429668e-06,
+      "loss": 0.0059,
+      "step": 2660
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.966751918158568e-06,
+      "loss": 0.0049,
+      "step": 2661
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.959079283887468e-06,
+      "loss": 0.0079,
+      "step": 2662
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.951406649616368e-06,
+      "loss": 1.5801,
+      "step": 2663
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.94373401534527e-06,
+      "loss": 0.0059,
+      "step": 2664
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.936061381074168e-06,
+      "loss": 1.2523,
+      "step": 2665
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.92838874680307e-06,
+      "loss": 0.0079,
+      "step": 2666
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.92071611253197e-06,
+      "loss": 0.0099,
+      "step": 2667
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.91304347826087e-06,
+      "loss": 0.0082,
+      "step": 2668
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.90537084398977e-06,
+      "loss": 0.004,
+      "step": 2669
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.897698209718671e-06,
+      "loss": 0.0064,
+      "step": 2670
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.890025575447569e-06,
+      "loss": 0.004,
+      "step": 2671
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 9.88235294117647e-06,
+      "loss": 0.0943,
+      "step": 2672
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 9.87468030690537e-06,
+      "loss": 0.0059,
+      "step": 2673
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 9.86700767263427e-06,
+      "loss": 0.0098,
+      "step": 2674
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.859335038363172e-06,
+      "loss": 0.0065,
+      "step": 2675
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.851662404092072e-06,
+      "loss": 0.0075,
+      "step": 2676
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.843989769820972e-06,
+      "loss": 0.0044,
+      "step": 2677
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.836317135549872e-06,
+      "loss": 0.0063,
+      "step": 2678
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.828644501278774e-06,
+      "loss": 0.0038,
+      "step": 2679
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.820971867007673e-06,
+      "loss": 0.0067,
+      "step": 2680
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.813299232736573e-06,
+      "loss": 1.1148,
+      "step": 2681
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.805626598465473e-06,
+      "loss": 0.0056,
+      "step": 2682
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.797953964194375e-06,
+      "loss": 0.0096,
+      "step": 2683
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.790281329923273e-06,
+      "loss": 0.011,
+      "step": 2684
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.782608695652175e-06,
+      "loss": 0.0038,
+      "step": 2685
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.774936061381075e-06,
+      "loss": 0.0129,
+      "step": 2686
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.767263427109975e-06,
+      "loss": 0.0063,
+      "step": 2687
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.759590792838875e-06,
+      "loss": 0.0561,
+      "step": 2688
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.751918158567776e-06,
+      "loss": 0.0041,
+      "step": 2689
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.744245524296674e-06,
+      "loss": 0.0048,
+      "step": 2690
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.736572890025576e-06,
+      "loss": 0.0072,
+      "step": 2691
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.728900255754476e-06,
+      "loss": 0.007,
+      "step": 2692
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.721227621483376e-06,
+      "loss": 0.0031,
+      "step": 2693
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.713554987212276e-06,
+      "loss": 1.4136,
+      "step": 2694
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.705882352941177e-06,
+      "loss": 0.0081,
+      "step": 2695
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.698209718670076e-06,
+      "loss": 0.0036,
+      "step": 2696
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.690537084398977e-06,
+      "loss": 1.382,
+      "step": 2697
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.682864450127877e-06,
+      "loss": 0.0054,
+      "step": 2698
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.675191815856777e-06,
+      "loss": 0.0064,
+      "step": 2699
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.667519181585679e-06,
+      "loss": 0.0072,
+      "step": 2700
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.659846547314579e-06,
+      "loss": 0.0054,
+      "step": 2701
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.652173913043478e-06,
+      "loss": 0.0108,
+      "step": 2702
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.644501278772378e-06,
+      "loss": 0.0037,
+      "step": 2703
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.63682864450128e-06,
+      "loss": 0.0071,
+      "step": 2704
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.629156010230178e-06,
+      "loss": 0.0062,
+      "step": 2705
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.62148337595908e-06,
+      "loss": 0.0056,
+      "step": 2706
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.61381074168798e-06,
+      "loss": 1.1276,
+      "step": 2707
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.606138107416881e-06,
+      "loss": 0.0045,
+      "step": 2708
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.59846547314578e-06,
+      "loss": 0.0073,
+      "step": 2709
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.590792838874681e-06,
+      "loss": 0.3343,
+      "step": 2710
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.583120204603581e-06,
+      "loss": 0.0038,
+      "step": 2711
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.575447570332481e-06,
+      "loss": 0.0053,
+      "step": 2712
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.567774936061381e-06,
+      "loss": 0.0037,
+      "step": 2713
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.560102301790283e-06,
+      "loss": 0.0036,
+      "step": 2714
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.55242966751918e-06,
+      "loss": 0.0053,
+      "step": 2715
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.544757033248082e-06,
+      "loss": 0.0069,
+      "step": 2716
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.537084398976982e-06,
+      "loss": 1.3084,
+      "step": 2717
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.529411764705882e-06,
+      "loss": 0.0022,
+      "step": 2718
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.521739130434782e-06,
+      "loss": 0.0046,
+      "step": 2719
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.514066496163684e-06,
+      "loss": 1.1433,
+      "step": 2720
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.506393861892582e-06,
+      "loss": 0.0053,
+      "step": 2721
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.498721227621484e-06,
+      "loss": 0.0053,
+      "step": 2722
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.491048593350385e-06,
+      "loss": 0.0057,
+      "step": 2723
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.483375959079283e-06,
+      "loss": 0.0058,
+      "step": 2724
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.475703324808185e-06,
+      "loss": 0.0029,
+      "step": 2725
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.468030690537085e-06,
+      "loss": 0.0061,
+      "step": 2726
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.460358056265985e-06,
+      "loss": 0.0058,
+      "step": 2727
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.452685421994885e-06,
+      "loss": 0.0074,
+      "step": 2728
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.445012787723786e-06,
+      "loss": 0.007,
+      "step": 2729
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.437340153452685e-06,
+      "loss": 1.3508,
+      "step": 2730
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.429667519181586e-06,
+      "loss": 0.0074,
+      "step": 2731
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.421994884910486e-06,
+      "loss": 0.0037,
+      "step": 2732
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.414322250639388e-06,
+      "loss": 0.003,
+      "step": 2733
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.406649616368286e-06,
+      "loss": 0.0038,
+      "step": 2734
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.398976982097188e-06,
+      "loss": 0.0086,
+      "step": 2735
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.391304347826087e-06,
+      "loss": 0.0058,
+      "step": 2736
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.383631713554987e-06,
+      "loss": 0.0021,
+      "step": 2737
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.375959079283887e-06,
+      "loss": 0.0046,
+      "step": 2738
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.368286445012789e-06,
+      "loss": 0.0035,
+      "step": 2739
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.360613810741687e-06,
+      "loss": 0.0054,
+      "step": 2740
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.352941176470589e-06,
+      "loss": 0.0052,
+      "step": 2741
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.345268542199489e-06,
+      "loss": 0.0072,
+      "step": 2742
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.337595907928389e-06,
+      "loss": 0.1932,
+      "step": 2743
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.329923273657289e-06,
+      "loss": 0.006,
+      "step": 2744
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.32225063938619e-06,
+      "loss": 0.0036,
+      "step": 2745
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.314578005115088e-06,
+      "loss": 1.2929,
+      "step": 2746
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.30690537084399e-06,
+      "loss": 0.0028,
+      "step": 2747
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.299232736572892e-06,
+      "loss": 1.2908,
+      "step": 2748
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.29156010230179e-06,
+      "loss": 0.0067,
+      "step": 2749
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.283887468030691e-06,
+      "loss": 0.0043,
+      "step": 2750
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.276214833759591e-06,
+      "loss": 0.0054,
+      "step": 2751
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.268542199488491e-06,
+      "loss": 0.0055,
+      "step": 2752
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.260869565217391e-06,
+      "loss": 1.1614,
+      "step": 2753
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.253196930946293e-06,
+      "loss": 0.0071,
+      "step": 2754
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.245524296675191e-06,
+      "loss": 0.0035,
+      "step": 2755
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.237851662404093e-06,
+      "loss": 0.0054,
+      "step": 2756
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.230179028132993e-06,
+      "loss": 0.0096,
+      "step": 2757
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.222506393861894e-06,
+      "loss": 0.0035,
+      "step": 2758
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.214833759590792e-06,
+      "loss": 0.0051,
+      "step": 2759
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.207161125319694e-06,
+      "loss": 0.0069,
+      "step": 2760
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.199488491048594e-06,
+      "loss": 0.0018,
+      "step": 2761
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.191815856777494e-06,
+      "loss": 0.0026,
+      "step": 2762
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.184143222506394e-06,
+      "loss": 0.0054,
+      "step": 2763
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.176470588235295e-06,
+      "loss": 0.0085,
+      "step": 2764
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.168797953964194e-06,
+      "loss": 0.008,
+      "step": 2765
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.161125319693095e-06,
+      "loss": 1.3292,
+      "step": 2766
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.153452685421995e-06,
+      "loss": 0.0085,
+      "step": 2767
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.145780051150895e-06,
+      "loss": 0.0034,
+      "step": 2768
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.138107416879795e-06,
+      "loss": 0.0051,
+      "step": 2769
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.130434782608697e-06,
+      "loss": 0.0054,
+      "step": 2770
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 9.122762148337595e-06,
+      "loss": 0.0054,
+      "step": 2771
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 9.115089514066496e-06,
+      "loss": 0.0083,
+      "step": 2772
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.9217171717171717,
+      "eval_loss": 0.37921515107154846,
+      "eval_runtime": 57.416,
+      "eval_samples_per_second": 6.897,
+      "eval_steps_per_second": 1.724,
+      "step": 2772
+    }
+  ],
+  "max_steps": 3960,
+  "num_train_epochs": 10,
+  "total_flos": 1.0060032828384e+18,
+  "trial_name": null,
+  "trial_params": null
+}
diff --git a/checkpoint-2772/training_args.bin b/checkpoint-2772/training_args.bin
new file mode 100644
index 0000000000000000000000000000000000000000..32901f899c91066c808eae9dd019400bfdd17498
--- /dev/null
+++ b/checkpoint-2772/training_args.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db14439c369bfe9c89f64ff38f0b2d3fcc53911fc2c0b12a3a7d04656e7384a
+size 3503
diff --git a/checkpoint-3168/config.json b/checkpoint-3168/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..cb8d52e14263c50fd53dca42b5050e3235115c32
--- /dev/null
+++ b/checkpoint-3168/config.json
@@ -0,0 +1,113 @@
+{
+  "_name_or_path": "facebook/wav2vec2-base-960h",
+  "activation_dropout": 0.1,
+  "adapter_kernel_size": 3,
+  "adapter_stride": 2,
+  "add_adapter": false,
+  "apply_spec_augment": true,
+  "architectures": [
+    "Wav2Vec2ForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 1,
+  "classifier_proj_size": 256,
+  "codevector_dim": 256,
+  "contrastive_logits_temperature": 0.1,
+  "conv_bias": false,
+  "conv_dim": [
+    512,
+    512,
+    512,
+    512,
+    512,
+    512,
+    512
+  ],
+  "conv_kernel": [
+    10,
+    3,
+    3,
+    3,
+    3,
+    2,
+    2
+  ],
+  "conv_stride": [
+    5,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2
+  ],
+  "ctc_loss_reduction": "sum",
+  "ctc_zero_infinity": false,
+  "diversity_loss_weight": 0.1,
+  "do_stable_layer_norm": false,
+  "eos_token_id": 2,
+  "feat_extract_activation": "gelu",
+  "feat_extract_dropout": 0.0,
+  "feat_extract_norm": "group",
+  "feat_proj_dropout": 0.1,
+  "feat_quantizer_dropout": 0.0,
+  "final_dropout": 0.1,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout": 0.1,
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Foreign",
+    "1": "Native"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "layerdrop": 0.1,
+  "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
+  "mask_feature_prob": 0.0,
+  "mask_time_length": 10,
+  "mask_time_min_masks": 2,
+  "mask_time_prob": 0.05,
+  "model_type": "wav2vec2",
+  "num_adapter_layers": 3,
+  "num_attention_heads": 12,
+  "num_codevector_groups": 2,
+  "num_codevectors_per_group": 320,
+  "num_conv_pos_embedding_groups": 16,
+  "num_conv_pos_embeddings": 128,
+  "num_feat_extract_layers": 7,
+  "num_hidden_layers": 12,
+  "num_negatives": 100,
+  "output_hidden_size": 768,
+  "pad_token_id": 0,
+  "proj_codevector_dim": 256,
+  "tdnn_dilation": [
+    1,
+    2,
+    3,
+    1,
+    1
+  ],
+  "tdnn_dim": [
+    512,
+    512,
+    512,
+    512,
+    1500
+  ],
+  "tdnn_kernel": [
+    5,
+    3,
+    3,
+    1,
+    1
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.28.1",
+  "use_weighted_layer_sum": false,
+  "vocab_size": 32,
+  "xvector_output_dim": 512
+}
diff --git a/checkpoint-3168/optimizer.pt b/checkpoint-3168/optimizer.pt
new file mode 100644
index 0000000000000000000000000000000000000000..9ee9af2eb56d38c416d1fe3630091d538b5aeaad
--- /dev/null
+++ b/checkpoint-3168/optimizer.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0d4c416ca93e1fd87f98bb15c654e7ffe94cff6f9a1e3b956ff5e9466c8f8bf6
+size 756677477
diff --git a/checkpoint-3168/preprocessor_config.json b/checkpoint-3168/preprocessor_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..a0b7227fc1d916e469b14f6c154ad6dfea1e6891
--- /dev/null
+++ b/checkpoint-3168/preprocessor_config.json
@@ -0,0 +1,9 @@
+{
+  "do_normalize": true,
+  "feature_extractor_type": "Wav2Vec2FeatureExtractor",
+  "feature_size": 1,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": false,
+  "sampling_rate": 16000
+}
diff --git a/checkpoint-3168/pytorch_model.bin b/checkpoint-3168/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..9f99cbd0051951e62df734377e199895ab1e2a0d
--- /dev/null
+++ b/checkpoint-3168/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c085ef8016a5c49e77e50273e4a0e4c333045fff483f59d6035aa36b20382423
+size 378348563
diff --git a/checkpoint-3168/rng_state.pth b/checkpoint-3168/rng_state.pth
new file mode 100644
index 0000000000000000000000000000000000000000..c0e1421d16bce283b717f8d1dc7b7f007fd2c89b
--- /dev/null
+++ b/checkpoint-3168/rng_state.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9675e845f259340915beb46dbee1633c9941c5cf94e84de3add30732d4a76d6a
+size 14567
diff --git a/checkpoint-3168/scheduler.pt b/checkpoint-3168/scheduler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..ab884997f3788a2dcc38eaa5438eb77d1571e2ce
--- /dev/null
+++ b/checkpoint-3168/scheduler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:703a8c430c945a695d14caa02ab8a8b1465c1b636a8f2e8dfd1aa7709108607b
+size 623
diff --git a/checkpoint-3168/trainer_state.json b/checkpoint-3168/trainer_state.json
new file mode 100644
index 0000000000000000000000000000000000000000..f551f4a1984ccfe1bbe19ee1544b6aee30d5422a
--- /dev/null
+++ b/checkpoint-3168/trainer_state.json
@@ -0,0 +1,19096 @@
+{
+  "best_metric": 0.3489837944507599,
+  "best_model_checkpoint": "accent_classification/checkpoint-2376",
+  "epoch": 8.0,
+  "global_step": 3168,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.000000000000001e-07,
+      "loss": 0.6956,
+      "step": 1
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.2000000000000002e-06,
+      "loss": 0.6979,
+      "step": 2
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8e-06,
+      "loss": 0.7014,
+      "step": 3
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 2.4000000000000003e-06,
+      "loss": 0.6984,
+      "step": 4
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 3e-06,
+      "loss": 0.6978,
+      "step": 5
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 3.6e-06,
+      "loss": 0.6783,
+      "step": 6
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.2000000000000004e-06,
+      "loss": 0.6956,
+      "step": 7
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.800000000000001e-06,
+      "loss": 0.6994,
+      "step": 8
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 5.4e-06,
+      "loss": 0.6902,
+      "step": 9
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6e-06,
+      "loss": 0.6937,
+      "step": 10
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6.6e-06,
+      "loss": 0.6497,
+      "step": 11
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.2e-06,
+      "loss": 0.6911,
+      "step": 12
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.8e-06,
+      "loss": 0.6904,
+      "step": 13
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 8.400000000000001e-06,
+      "loss": 0.6854,
+      "step": 14
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9e-06,
+      "loss": 0.6843,
+      "step": 15
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9.600000000000001e-06,
+      "loss": 0.6851,
+      "step": 16
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.02e-05,
+      "loss": 0.6733,
+      "step": 17
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.08e-05,
+      "loss": 0.6819,
+      "step": 18
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.1400000000000001e-05,
+      "loss": 0.681,
+      "step": 19
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.6998,
+      "step": 20
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.26e-05,
+      "loss": 0.6743,
+      "step": 21
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.32e-05,
+      "loss": 0.6818,
+      "step": 22
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3800000000000002e-05,
+      "loss": 0.5331,
+      "step": 23
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.44e-05,
+      "loss": 0.6681,
+      "step": 24
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.6955,
+      "step": 25
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.56e-05,
+      "loss": 0.7,
+      "step": 26
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.62e-05,
+      "loss": 0.6975,
+      "step": 27
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.6800000000000002e-05,
+      "loss": 0.6249,
+      "step": 28
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.74e-05,
+      "loss": 0.6979,
+      "step": 29
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.6076,
+      "step": 30
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.86e-05,
+      "loss": 0.6091,
+      "step": 31
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9200000000000003e-05,
+      "loss": 0.6438,
+      "step": 32
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.98e-05,
+      "loss": 0.6624,
+      "step": 33
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.04e-05,
+      "loss": 0.6452,
+      "step": 34
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.6963,
+      "step": 35
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.16e-05,
+      "loss": 0.758,
+      "step": 36
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.22e-05,
+      "loss": 0.6471,
+      "step": 37
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.2800000000000002e-05,
+      "loss": 0.6313,
+      "step": 38
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.3400000000000003e-05,
+      "loss": 0.7019,
+      "step": 39
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.6336,
+      "step": 40
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4599999999999998e-05,
+      "loss": 0.7091,
+      "step": 41
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.52e-05,
+      "loss": 0.701,
+      "step": 42
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.58e-05,
+      "loss": 0.7036,
+      "step": 43
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.64e-05,
+      "loss": 0.7073,
+      "step": 44
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.614,
+      "step": 45
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.7600000000000003e-05,
+      "loss": 0.7543,
+      "step": 46
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.8199999999999998e-05,
+      "loss": 0.538,
+      "step": 47
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.88e-05,
+      "loss": 0.5179,
+      "step": 48
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.94e-05,
+      "loss": 0.6164,
+      "step": 49
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 3e-05,
+      "loss": 0.6093,
+      "step": 50
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.99923273657289e-05,
+      "loss": 0.6983,
+      "step": 51
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9984654731457804e-05,
+      "loss": 0.6069,
+      "step": 52
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9976982097186702e-05,
+      "loss": 0.5933,
+      "step": 53
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.99693094629156e-05,
+      "loss": 0.552,
+      "step": 54
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9961636828644502e-05,
+      "loss": 0.5865,
+      "step": 55
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9953964194373404e-05,
+      "loss": 0.7146,
+      "step": 56
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9946291560102302e-05,
+      "loss": 0.5591,
+      "step": 57
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9938618925831203e-05,
+      "loss": 0.7366,
+      "step": 58
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9930946291560102e-05,
+      "loss": 0.7006,
+      "step": 59
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9923273657289003e-05,
+      "loss": 0.4148,
+      "step": 60
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9915601023017905e-05,
+      "loss": 0.9938,
+      "step": 61
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9907928388746803e-05,
+      "loss": 0.911,
+      "step": 62
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.99002557544757e-05,
+      "loss": 0.5682,
+      "step": 63
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9892583120204606e-05,
+      "loss": 0.5833,
+      "step": 64
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9884910485933505e-05,
+      "loss": 0.7237,
+      "step": 65
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9877237851662403e-05,
+      "loss": 0.4013,
+      "step": 66
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9869565217391304e-05,
+      "loss": 0.9695,
+      "step": 67
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9861892583120206e-05,
+      "loss": 0.568,
+      "step": 68
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9854219948849104e-05,
+      "loss": 0.5697,
+      "step": 69
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9846547314578006e-05,
+      "loss": 0.7675,
+      "step": 70
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9838874680306904e-05,
+      "loss": 0.8831,
+      "step": 71
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9831202046035806e-05,
+      "loss": 0.7008,
+      "step": 72
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9823529411764707e-05,
+      "loss": 0.7085,
+      "step": 73
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9815856777493606e-05,
+      "loss": 0.5374,
+      "step": 74
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9808184143222507e-05,
+      "loss": 0.4131,
+      "step": 75
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.980051150895141e-05,
+      "loss": 0.7309,
+      "step": 76
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9792838874680307e-05,
+      "loss": 0.4279,
+      "step": 77
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.978516624040921e-05,
+      "loss": 0.5813,
+      "step": 78
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.977749360613811e-05,
+      "loss": 0.6276,
+      "step": 79
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976982097186701e-05,
+      "loss": 0.7892,
+      "step": 80
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976214833759591e-05,
+      "loss": 0.7134,
+      "step": 81
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.9754475703324808e-05,
+      "loss": 0.3682,
+      "step": 82
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.974680306905371e-05,
+      "loss": 0.5562,
+      "step": 83
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973913043478261e-05,
+      "loss": 0.7158,
+      "step": 84
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973145780051151e-05,
+      "loss": 0.524,
+      "step": 85
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9723785166240408e-05,
+      "loss": 0.3372,
+      "step": 86
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9716112531969313e-05,
+      "loss": 0.4979,
+      "step": 87
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970843989769821e-05,
+      "loss": 0.3177,
+      "step": 88
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970076726342711e-05,
+      "loss": 0.3488,
+      "step": 89
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.969309462915601e-05,
+      "loss": 0.5158,
+      "step": 90
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9685421994884913e-05,
+      "loss": 0.3208,
+      "step": 91
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.967774936061381e-05,
+      "loss": 0.6654,
+      "step": 92
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9670076726342712e-05,
+      "loss": 0.5581,
+      "step": 93
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.966240409207161e-05,
+      "loss": 0.7456,
+      "step": 94
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9654731457800512e-05,
+      "loss": 0.4437,
+      "step": 95
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9647058823529414e-05,
+      "loss": 0.5525,
+      "step": 96
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9639386189258312e-05,
+      "loss": 0.6673,
+      "step": 97
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.963171355498721e-05,
+      "loss": 0.435,
+      "step": 98
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9624040920716115e-05,
+      "loss": 0.7415,
+      "step": 99
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9616368286445014e-05,
+      "loss": 0.2981,
+      "step": 100
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9608695652173912e-05,
+      "loss": 0.5179,
+      "step": 101
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9601023017902817e-05,
+      "loss": 0.4005,
+      "step": 102
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9593350383631715e-05,
+      "loss": 0.4598,
+      "step": 103
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9585677749360613e-05,
+      "loss": 0.2553,
+      "step": 104
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9578005115089515e-05,
+      "loss": 0.4302,
+      "step": 105
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9570332480818416e-05,
+      "loss": 0.7894,
+      "step": 106
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9562659846547315e-05,
+      "loss": 0.213,
+      "step": 107
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9554987212276216e-05,
+      "loss": 0.5507,
+      "step": 108
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9547314578005115e-05,
+      "loss": 0.7676,
+      "step": 109
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9539641943734016e-05,
+      "loss": 0.3418,
+      "step": 110
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9531969309462918e-05,
+      "loss": 0.8533,
+      "step": 111
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9524296675191816e-05,
+      "loss": 0.3659,
+      "step": 112
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9516624040920714e-05,
+      "loss": 0.6482,
+      "step": 113
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.950895140664962e-05,
+      "loss": 0.5484,
+      "step": 114
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9501278772378517e-05,
+      "loss": 0.4508,
+      "step": 115
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9493606138107416e-05,
+      "loss": 0.4401,
+      "step": 116
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9485933503836317e-05,
+      "loss": 0.5455,
+      "step": 117
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.947826086956522e-05,
+      "loss": 0.722,
+      "step": 118
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9470588235294117e-05,
+      "loss": 0.2874,
+      "step": 119
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.946291560102302e-05,
+      "loss": 0.5973,
+      "step": 120
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.9455242966751917e-05,
+      "loss": 0.5898,
+      "step": 121
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.944757033248082e-05,
+      "loss": 1.0125,
+      "step": 122
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943989769820972e-05,
+      "loss": 0.2511,
+      "step": 123
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943222506393862e-05,
+      "loss": 0.545,
+      "step": 124
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.942455242966752e-05,
+      "loss": 0.593,
+      "step": 125
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.941687979539642e-05,
+      "loss": 0.2319,
+      "step": 126
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.940920716112532e-05,
+      "loss": 0.2215,
+      "step": 127
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.9401534526854218e-05,
+      "loss": 0.1964,
+      "step": 128
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9393861892583123e-05,
+      "loss": 0.1471,
+      "step": 129
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.938618925831202e-05,
+      "loss": 1.5965,
+      "step": 130
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9378516624040923e-05,
+      "loss": 1.0772,
+      "step": 131
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.937084398976982e-05,
+      "loss": 0.2005,
+      "step": 132
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9363171355498723e-05,
+      "loss": 0.9628,
+      "step": 133
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9355498721227624e-05,
+      "loss": 0.1503,
+      "step": 134
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9347826086956523e-05,
+      "loss": 0.6201,
+      "step": 135
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.934015345268542e-05,
+      "loss": 0.6094,
+      "step": 136
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9332480818414326e-05,
+      "loss": 1.385,
+      "step": 137
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9324808184143224e-05,
+      "loss": 0.1552,
+      "step": 138
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9317135549872122e-05,
+      "loss": 0.5828,
+      "step": 139
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9309462915601024e-05,
+      "loss": 0.6018,
+      "step": 140
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9301790281329925e-05,
+      "loss": 1.0614,
+      "step": 141
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9294117647058824e-05,
+      "loss": 0.566,
+      "step": 142
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9286445012787725e-05,
+      "loss": 0.5958,
+      "step": 143
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9278772378516624e-05,
+      "loss": 0.5838,
+      "step": 144
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9271099744245525e-05,
+      "loss": 0.5909,
+      "step": 145
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9263427109974427e-05,
+      "loss": 0.5862,
+      "step": 146
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9255754475703325e-05,
+      "loss": 0.2031,
+      "step": 147
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9248081841432223e-05,
+      "loss": 0.1995,
+      "step": 148
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9240409207161128e-05,
+      "loss": 0.24,
+      "step": 149
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9232736572890026e-05,
+      "loss": 0.5845,
+      "step": 150
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9225063938618925e-05,
+      "loss": 0.964,
+      "step": 151
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.921739130434783e-05,
+      "loss": 0.9595,
+      "step": 152
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9209718670076728e-05,
+      "loss": 0.2033,
+      "step": 153
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9202046035805626e-05,
+      "loss": 0.213,
+      "step": 154
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9194373401534528e-05,
+      "loss": 0.2042,
+      "step": 155
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.918670076726343e-05,
+      "loss": 0.5783,
+      "step": 156
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9179028132992328e-05,
+      "loss": 1.0822,
+      "step": 157
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.917135549872123e-05,
+      "loss": 1.2643,
+      "step": 158
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9163682864450127e-05,
+      "loss": 0.2593,
+      "step": 159
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.915601023017903e-05,
+      "loss": 0.5783,
+      "step": 160
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914833759590793e-05,
+      "loss": 0.5721,
+      "step": 161
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914066496163683e-05,
+      "loss": 0.5808,
+      "step": 162
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9132992327365727e-05,
+      "loss": 1.2578,
+      "step": 163
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9125319693094632e-05,
+      "loss": 0.5709,
+      "step": 164
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.911764705882353e-05,
+      "loss": 0.2159,
+      "step": 165
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910997442455243e-05,
+      "loss": 0.2329,
+      "step": 166
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910230179028133e-05,
+      "loss": 0.926,
+      "step": 167
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.9094629156010232e-05,
+      "loss": 0.9095,
+      "step": 168
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.908695652173913e-05,
+      "loss": 0.5622,
+      "step": 169
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907928388746803e-05,
+      "loss": 0.5703,
+      "step": 170
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907161125319693e-05,
+      "loss": 0.5669,
+      "step": 171
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.906393861892583e-05,
+      "loss": 0.2637,
+      "step": 172
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9056265984654733e-05,
+      "loss": 1.1077,
+      "step": 173
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.904859335038363e-05,
+      "loss": 0.5653,
+      "step": 174
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9040920716112533e-05,
+      "loss": 0.5584,
+      "step": 175
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9033248081841434e-05,
+      "loss": 0.5544,
+      "step": 176
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9025575447570333e-05,
+      "loss": 0.5663,
+      "step": 177
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.901790281329923e-05,
+      "loss": 1.0808,
+      "step": 178
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9010230179028136e-05,
+      "loss": 0.8736,
+      "step": 179
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9002557544757034e-05,
+      "loss": 0.6435,
+      "step": 180
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8994884910485932e-05,
+      "loss": 0.786,
+      "step": 181
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8987212276214834e-05,
+      "loss": 0.5717,
+      "step": 182
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8979539641943736e-05,
+      "loss": 0.5618,
+      "step": 183
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8971867007672637e-05,
+      "loss": 0.5785,
+      "step": 184
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8964194373401535e-05,
+      "loss": 0.7601,
+      "step": 185
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8956521739130434e-05,
+      "loss": 0.8112,
+      "step": 186
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.894884910485934e-05,
+      "loss": 0.5663,
+      "step": 187
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8941176470588237e-05,
+      "loss": 0.5631,
+      "step": 188
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8933503836317135e-05,
+      "loss": 0.7363,
+      "step": 189
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8925831202046037e-05,
+      "loss": 0.4552,
+      "step": 190
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8918158567774938e-05,
+      "loss": 0.4599,
+      "step": 191
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8910485933503836e-05,
+      "loss": 0.7848,
+      "step": 192
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8902813299232738e-05,
+      "loss": 0.5907,
+      "step": 193
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8895140664961636e-05,
+      "loss": 0.4264,
+      "step": 194
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8887468030690538e-05,
+      "loss": 0.5697,
+      "step": 195
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.887979539641944e-05,
+      "loss": 0.7643,
+      "step": 196
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8872122762148338e-05,
+      "loss": 0.3635,
+      "step": 197
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8864450127877236e-05,
+      "loss": 0.3715,
+      "step": 198
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.885677749360614e-05,
+      "loss": 0.554,
+      "step": 199
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.884910485933504e-05,
+      "loss": 0.8044,
+      "step": 200
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8841432225063937e-05,
+      "loss": 0.5673,
+      "step": 201
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8833759590792842e-05,
+      "loss": 0.8042,
+      "step": 202
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.882608695652174e-05,
+      "loss": 0.5638,
+      "step": 203
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881841432225064e-05,
+      "loss": 0.3584,
+      "step": 204
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881074168797954e-05,
+      "loss": 0.8211,
+      "step": 205
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.8803069053708442e-05,
+      "loss": 0.5708,
+      "step": 206
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.879539641943734e-05,
+      "loss": 0.8301,
+      "step": 207
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8787723785166242e-05,
+      "loss": 0.8229,
+      "step": 208
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.878005115089514e-05,
+      "loss": 0.5645,
+      "step": 209
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8772378516624042e-05,
+      "loss": 0.3529,
+      "step": 210
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8764705882352943e-05,
+      "loss": 0.5578,
+      "step": 211
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.875703324808184e-05,
+      "loss": 0.5659,
+      "step": 212
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.874936061381074e-05,
+      "loss": 0.7495,
+      "step": 213
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8741687979539645e-05,
+      "loss": 0.3162,
+      "step": 214
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8734015345268543e-05,
+      "loss": 0.5616,
+      "step": 215
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.872634271099744e-05,
+      "loss": 0.4034,
+      "step": 216
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8718670076726343e-05,
+      "loss": 0.3163,
+      "step": 217
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8710997442455245e-05,
+      "loss": 0.5514,
+      "step": 218
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8703324808184143e-05,
+      "loss": 0.826,
+      "step": 219
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8695652173913044e-05,
+      "loss": 0.8363,
+      "step": 220
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8687979539641943e-05,
+      "loss": 0.8288,
+      "step": 221
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8680306905370844e-05,
+      "loss": 0.5929,
+      "step": 222
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8672634271099746e-05,
+      "loss": 0.565,
+      "step": 223
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8664961636828644e-05,
+      "loss": 0.5642,
+      "step": 224
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8657289002557546e-05,
+      "loss": 0.5609,
+      "step": 225
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8649616368286447e-05,
+      "loss": 0.5641,
+      "step": 226
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8641943734015345e-05,
+      "loss": 0.8535,
+      "step": 227
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8634271099744244e-05,
+      "loss": 0.5652,
+      "step": 228
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.862659846547315e-05,
+      "loss": 0.4397,
+      "step": 229
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8618925831202047e-05,
+      "loss": 0.567,
+      "step": 230
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8611253196930945e-05,
+      "loss": 0.7517,
+      "step": 231
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8603580562659847e-05,
+      "loss": 0.5839,
+      "step": 232
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.859590792838875e-05,
+      "loss": 0.5677,
+      "step": 233
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8588235294117647e-05,
+      "loss": 0.5678,
+      "step": 234
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8580562659846548e-05,
+      "loss": 1.3045,
+      "step": 235
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8572890025575446e-05,
+      "loss": 0.804,
+      "step": 236
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.856521739130435e-05,
+      "loss": 1.0303,
+      "step": 237
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.855754475703325e-05,
+      "loss": 0.3303,
+      "step": 238
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8549872122762148e-05,
+      "loss": 0.5677,
+      "step": 239
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.854219948849105e-05,
+      "loss": 0.7972,
+      "step": 240
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.853452685421995e-05,
+      "loss": 0.5644,
+      "step": 241
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.852685421994885e-05,
+      "loss": 0.9294,
+      "step": 242
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.851918158567775e-05,
+      "loss": 0.5665,
+      "step": 243
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.851150895140665e-05,
+      "loss": 0.7963,
+      "step": 244
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.850383631713555e-05,
+      "loss": 0.5659,
+      "step": 245
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.8496163682864452e-05,
+      "loss": 0.5718,
+      "step": 246
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.848849104859335e-05,
+      "loss": 0.3526,
+      "step": 247
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8480818414322252e-05,
+      "loss": 0.574,
+      "step": 248
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8473145780051154e-05,
+      "loss": 0.9996,
+      "step": 249
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8465473145780052e-05,
+      "loss": 0.7634,
+      "step": 250
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.845780051150895e-05,
+      "loss": 0.5639,
+      "step": 251
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8450127877237855e-05,
+      "loss": 0.5737,
+      "step": 252
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8442455242966753e-05,
+      "loss": 0.7793,
+      "step": 253
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8434782608695652e-05,
+      "loss": 0.6769,
+      "step": 254
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8427109974424553e-05,
+      "loss": 0.5748,
+      "step": 255
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8419437340153455e-05,
+      "loss": 0.3602,
+      "step": 256
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8411764705882353e-05,
+      "loss": 0.5714,
+      "step": 257
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8404092071611255e-05,
+      "loss": 0.5738,
+      "step": 258
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8396419437340153e-05,
+      "loss": 0.5719,
+      "step": 259
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8388746803069055e-05,
+      "loss": 0.5753,
+      "step": 260
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8381074168797956e-05,
+      "loss": 0.5717,
+      "step": 261
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8373401534526854e-05,
+      "loss": 0.9794,
+      "step": 262
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8365728900255753e-05,
+      "loss": 0.5737,
+      "step": 263
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8358056265984658e-05,
+      "loss": 0.5723,
+      "step": 264
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8350383631713556e-05,
+      "loss": 0.5656,
+      "step": 265
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8342710997442454e-05,
+      "loss": 0.4259,
+      "step": 266
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8335038363171356e-05,
+      "loss": 0.3076,
+      "step": 267
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8327365728900257e-05,
+      "loss": 0.338,
+      "step": 268
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8319693094629156e-05,
+      "loss": 0.5612,
+      "step": 269
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8312020460358057e-05,
+      "loss": 0.5249,
+      "step": 270
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8304347826086955e-05,
+      "loss": 0.5671,
+      "step": 271
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8296675191815857e-05,
+      "loss": 0.5574,
+      "step": 272
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.828900255754476e-05,
+      "loss": 0.8152,
+      "step": 273
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8281329923273657e-05,
+      "loss": 0.5647,
+      "step": 274
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.827365728900256e-05,
+      "loss": 0.5632,
+      "step": 275
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.826598465473146e-05,
+      "loss": 0.3391,
+      "step": 276
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8258312020460358e-05,
+      "loss": 0.55,
+      "step": 277
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8250639386189257e-05,
+      "loss": 0.5672,
+      "step": 278
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.824296675191816e-05,
+      "loss": 0.285,
+      "step": 279
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.823529411764706e-05,
+      "loss": 0.2809,
+      "step": 280
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.8227621483375958e-05,
+      "loss": 0.5552,
+      "step": 281
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821994884910486e-05,
+      "loss": 0.563,
+      "step": 282
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821227621483376e-05,
+      "loss": 0.5628,
+      "step": 283
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.820460358056266e-05,
+      "loss": 0.2548,
+      "step": 284
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.819693094629156e-05,
+      "loss": 0.871,
+      "step": 285
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818925831202046e-05,
+      "loss": 0.8797,
+      "step": 286
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818158567774936e-05,
+      "loss": 0.2453,
+      "step": 287
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8173913043478262e-05,
+      "loss": 0.2464,
+      "step": 288
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.816624040920716e-05,
+      "loss": 0.3134,
+      "step": 289
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8158567774936062e-05,
+      "loss": 0.2431,
+      "step": 290
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8150895140664964e-05,
+      "loss": 0.2308,
+      "step": 291
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8143222506393862e-05,
+      "loss": 0.2173,
+      "step": 292
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8135549872122764e-05,
+      "loss": 0.5665,
+      "step": 293
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8127877237851662e-05,
+      "loss": 0.5592,
+      "step": 294
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8120204603580564e-05,
+      "loss": 0.473,
+      "step": 295
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8112531969309465e-05,
+      "loss": 0.9458,
+      "step": 296
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8104859335038363e-05,
+      "loss": 0.1879,
+      "step": 297
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8097186700767265e-05,
+      "loss": 0.6797,
+      "step": 298
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8089514066496167e-05,
+      "loss": 0.7592,
+      "step": 299
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8081841432225065e-05,
+      "loss": 0.4109,
+      "step": 300
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8074168797953963e-05,
+      "loss": 0.5821,
+      "step": 301
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8066496163682868e-05,
+      "loss": 0.2057,
+      "step": 302
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8058823529411766e-05,
+      "loss": 0.5322,
+      "step": 303
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8051150895140665e-05,
+      "loss": 0.5208,
+      "step": 304
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8043478260869566e-05,
+      "loss": 0.6799,
+      "step": 305
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8035805626598468e-05,
+      "loss": 0.5894,
+      "step": 306
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8028132992327366e-05,
+      "loss": 0.4051,
+      "step": 307
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8020460358056268e-05,
+      "loss": 0.3695,
+      "step": 308
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8012787723785166e-05,
+      "loss": 0.4869,
+      "step": 309
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8005115089514067e-05,
+      "loss": 0.6075,
+      "step": 310
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.799744245524297e-05,
+      "loss": 0.4245,
+      "step": 311
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7989769820971867e-05,
+      "loss": 0.5122,
+      "step": 312
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7982097186700765e-05,
+      "loss": 0.452,
+      "step": 313
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.797442455242967e-05,
+      "loss": 0.9568,
+      "step": 314
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.796675191815857e-05,
+      "loss": 0.5846,
+      "step": 315
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.7959079283887467e-05,
+      "loss": 0.3743,
+      "step": 316
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.795140664961637e-05,
+      "loss": 0.1922,
+      "step": 317
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.794373401534527e-05,
+      "loss": 0.5303,
+      "step": 318
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.793606138107417e-05,
+      "loss": 0.7253,
+      "step": 319
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.792838874680307e-05,
+      "loss": 0.2794,
+      "step": 320
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.7920716112531968e-05,
+      "loss": 0.577,
+      "step": 321
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.791304347826087e-05,
+      "loss": 0.614,
+      "step": 322
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.790537084398977e-05,
+      "loss": 0.5882,
+      "step": 323
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789769820971867e-05,
+      "loss": 0.2503,
+      "step": 324
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789002557544757e-05,
+      "loss": 0.4195,
+      "step": 325
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.7882352941176473e-05,
+      "loss": 0.2959,
+      "step": 326
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.787468030690537e-05,
+      "loss": 0.3732,
+      "step": 327
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.786700767263427e-05,
+      "loss": 0.5729,
+      "step": 328
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7859335038363174e-05,
+      "loss": 0.3179,
+      "step": 329
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7851662404092073e-05,
+      "loss": 0.6823,
+      "step": 330
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.784398976982097e-05,
+      "loss": 0.2663,
+      "step": 331
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7836317135549872e-05,
+      "loss": 0.4436,
+      "step": 332
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7828644501278774e-05,
+      "loss": 0.3089,
+      "step": 333
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7820971867007672e-05,
+      "loss": 0.5888,
+      "step": 334
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7813299232736574e-05,
+      "loss": 0.2115,
+      "step": 335
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7805626598465472e-05,
+      "loss": 0.4005,
+      "step": 336
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7797953964194374e-05,
+      "loss": 0.3097,
+      "step": 337
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7790281329923275e-05,
+      "loss": 0.5847,
+      "step": 338
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7782608695652174e-05,
+      "loss": 1.0164,
+      "step": 339
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7774936061381072e-05,
+      "loss": 1.1321,
+      "step": 340
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7767263427109977e-05,
+      "loss": 0.592,
+      "step": 341
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7759590792838875e-05,
+      "loss": 1.0191,
+      "step": 342
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7751918158567777e-05,
+      "loss": 0.9199,
+      "step": 343
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7744245524296675e-05,
+      "loss": 0.2874,
+      "step": 344
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7736572890025576e-05,
+      "loss": 0.2469,
+      "step": 345
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7728900255754478e-05,
+      "loss": 0.8364,
+      "step": 346
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7721227621483376e-05,
+      "loss": 0.7365,
+      "step": 347
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7713554987212278e-05,
+      "loss": 0.5595,
+      "step": 348
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.770588235294118e-05,
+      "loss": 0.2546,
+      "step": 349
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7698209718670078e-05,
+      "loss": 0.8183,
+      "step": 350
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7690537084398976e-05,
+      "loss": 0.5385,
+      "step": 351
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.768286445012788e-05,
+      "loss": 0.2763,
+      "step": 352
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.767519181585678e-05,
+      "loss": 0.3045,
+      "step": 353
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7667519181585677e-05,
+      "loss": 0.3354,
+      "step": 354
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765984654731458e-05,
+      "loss": 0.9366,
+      "step": 355
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765217391304348e-05,
+      "loss": 0.6212,
+      "step": 356
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.764450127877238e-05,
+      "loss": 0.1161,
+      "step": 357
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.763682864450128e-05,
+      "loss": 0.4929,
+      "step": 358
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762915601023018e-05,
+      "loss": 0.8167,
+      "step": 359
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762148337595908e-05,
+      "loss": 0.2521,
+      "step": 360
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.7613810741687982e-05,
+      "loss": 0.4174,
+      "step": 361
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.760613810741688e-05,
+      "loss": 0.3822,
+      "step": 362
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.759846547314578e-05,
+      "loss": 0.5355,
+      "step": 363
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.7590792838874683e-05,
+      "loss": 0.561,
+      "step": 364
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.758312020460358e-05,
+      "loss": 0.5126,
+      "step": 365
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.757544757033248e-05,
+      "loss": 0.2301,
+      "step": 366
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.756777493606138e-05,
+      "loss": 0.1698,
+      "step": 367
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7560102301790283e-05,
+      "loss": 0.1634,
+      "step": 368
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.755242966751918e-05,
+      "loss": 0.5955,
+      "step": 369
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7544757033248083e-05,
+      "loss": 0.1568,
+      "step": 370
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.753708439897698e-05,
+      "loss": 0.5167,
+      "step": 371
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7529411764705883e-05,
+      "loss": 1.0028,
+      "step": 372
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7521739130434784e-05,
+      "loss": 1.2567,
+      "step": 373
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7514066496163682e-05,
+      "loss": 0.6094,
+      "step": 374
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7506393861892584e-05,
+      "loss": 0.6059,
+      "step": 375
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7498721227621486e-05,
+      "loss": 1.0593,
+      "step": 376
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7491048593350384e-05,
+      "loss": 0.1506,
+      "step": 377
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7483375959079282e-05,
+      "loss": 1.5106,
+      "step": 378
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7475703324808187e-05,
+      "loss": 0.1827,
+      "step": 379
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7468030690537085e-05,
+      "loss": 1.4961,
+      "step": 380
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7460358056265984e-05,
+      "loss": 1.0456,
+      "step": 381
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7452685421994885e-05,
+      "loss": 1.2063,
+      "step": 382
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7445012787723787e-05,
+      "loss": 0.5837,
+      "step": 383
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7437340153452685e-05,
+      "loss": 1.4536,
+      "step": 384
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7429667519181587e-05,
+      "loss": 0.5803,
+      "step": 385
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7421994884910485e-05,
+      "loss": 0.5896,
+      "step": 386
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7414322250639386e-05,
+      "loss": 0.1822,
+      "step": 387
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7406649616368288e-05,
+      "loss": 1.7893,
+      "step": 388
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7398976982097186e-05,
+      "loss": 0.9723,
+      "step": 389
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7391304347826085e-05,
+      "loss": 1.2815,
+      "step": 390
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.738363171355499e-05,
+      "loss": 0.5765,
+      "step": 391
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7375959079283888e-05,
+      "loss": 1.2429,
+      "step": 392
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7368286445012786e-05,
+      "loss": 0.1343,
+      "step": 393
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7360613810741688e-05,
+      "loss": 0.2306,
+      "step": 394
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.735294117647059e-05,
+      "loss": 0.624,
+      "step": 395
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.734526854219949e-05,
+      "loss": 0.2371,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7348484848484849,
+      "eval_loss": 0.5804409980773926,
+      "eval_runtime": 57.3336,
+      "eval_samples_per_second": 6.907,
+      "eval_steps_per_second": 1.727,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.733759590792839e-05,
+      "loss": 0.8989,
+      "step": 397
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.732992327365729e-05,
+      "loss": 0.5664,
+      "step": 398
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.7322250639386192e-05,
+      "loss": 0.8852,
+      "step": 399
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.731457800511509e-05,
+      "loss": 0.5661,
+      "step": 400
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.730690537084399e-05,
+      "loss": 0.5648,
+      "step": 401
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7299232736572894e-05,
+      "loss": 0.7333,
+      "step": 402
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7291560102301792e-05,
+      "loss": 0.3037,
+      "step": 403
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.728388746803069e-05,
+      "loss": 0.1463,
+      "step": 404
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7276214833759592e-05,
+      "loss": 0.2702,
+      "step": 405
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7268542199488493e-05,
+      "loss": 0.2693,
+      "step": 406
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.726086956521739e-05,
+      "loss": 0.5647,
+      "step": 407
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7253196930946293e-05,
+      "loss": 0.8507,
+      "step": 408
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.724552429667519e-05,
+      "loss": 0.8645,
+      "step": 409
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7237851662404093e-05,
+      "loss": 0.2649,
+      "step": 410
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7230179028132995e-05,
+      "loss": 0.6942,
+      "step": 411
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7222506393861893e-05,
+      "loss": 0.5656,
+      "step": 412
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.721483375959079e-05,
+      "loss": 0.5489,
+      "step": 413
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7207161125319696e-05,
+      "loss": 0.2652,
+      "step": 414
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7199488491048594e-05,
+      "loss": 0.8676,
+      "step": 415
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7191815856777493e-05,
+      "loss": 0.8621,
+      "step": 416
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7184143222506394e-05,
+      "loss": 0.2641,
+      "step": 417
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7176470588235296e-05,
+      "loss": 0.859,
+      "step": 418
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7168797953964194e-05,
+      "loss": 0.6492,
+      "step": 419
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7161125319693096e-05,
+      "loss": 1.1494,
+      "step": 420
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7153452685421994e-05,
+      "loss": 0.8278,
+      "step": 421
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7145780051150895e-05,
+      "loss": 0.764,
+      "step": 422
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7138107416879797e-05,
+      "loss": 0.564,
+      "step": 423
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7130434782608695e-05,
+      "loss": 0.8051,
+      "step": 424
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7122762148337597e-05,
+      "loss": 0.3052,
+      "step": 425
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.71150895140665e-05,
+      "loss": 0.2945,
+      "step": 426
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7107416879795397e-05,
+      "loss": 0.4118,
+      "step": 427
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7099744245524295e-05,
+      "loss": 0.559,
+      "step": 428
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.70920716112532e-05,
+      "loss": 0.5593,
+      "step": 429
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7084398976982098e-05,
+      "loss": 0.564,
+      "step": 430
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7076726342710996e-05,
+      "loss": 0.4454,
+      "step": 431
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7069053708439898e-05,
+      "loss": 0.4381,
+      "step": 432
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.70613810741688e-05,
+      "loss": 0.3718,
+      "step": 433
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7053708439897698e-05,
+      "loss": 0.6627,
+      "step": 434
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70460358056266e-05,
+      "loss": 0.4004,
+      "step": 435
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7038363171355498e-05,
+      "loss": 0.3855,
+      "step": 436
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70306905370844e-05,
+      "loss": 0.4799,
+      "step": 437
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70230179028133e-05,
+      "loss": 0.6714,
+      "step": 438
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70153452685422e-05,
+      "loss": 0.622,
+      "step": 439
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7007672634271097e-05,
+      "loss": 0.3916,
+      "step": 440
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.5487,
+      "step": 441
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69923273657289e-05,
+      "loss": 0.376,
+      "step": 442
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69846547314578e-05,
+      "loss": 0.3616,
+      "step": 443
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69769820971867e-05,
+      "loss": 0.8747,
+      "step": 444
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.6969309462915602e-05,
+      "loss": 0.283,
+      "step": 445
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.69616368286445e-05,
+      "loss": 0.3988,
+      "step": 446
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6953964194373402e-05,
+      "loss": 0.4569,
+      "step": 447
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6946291560102303e-05,
+      "loss": 0.2005,
+      "step": 448
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6938618925831205e-05,
+      "loss": 0.3087,
+      "step": 449
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6930946291560103e-05,
+      "loss": 0.5814,
+      "step": 450
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6923273657289e-05,
+      "loss": 0.2995,
+      "step": 451
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6915601023017907e-05,
+      "loss": 0.7806,
+      "step": 452
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6907928388746805e-05,
+      "loss": 0.2666,
+      "step": 453
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6900255754475703e-05,
+      "loss": 0.2571,
+      "step": 454
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6892583120204605e-05,
+      "loss": 0.5931,
+      "step": 455
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6884910485933506e-05,
+      "loss": 0.475,
+      "step": 456
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6877237851662404e-05,
+      "loss": 0.2372,
+      "step": 457
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6869565217391306e-05,
+      "loss": 0.1577,
+      "step": 458
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6861892583120204e-05,
+      "loss": 0.1953,
+      "step": 459
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6854219948849106e-05,
+      "loss": 0.1931,
+      "step": 460
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6846547314578008e-05,
+      "loss": 0.1829,
+      "step": 461
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6838874680306906e-05,
+      "loss": 0.1601,
+      "step": 462
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6831202046035804e-05,
+      "loss": 0.2505,
+      "step": 463
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.682352941176471e-05,
+      "loss": 0.181,
+      "step": 464
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6815856777493607e-05,
+      "loss": 0.5978,
+      "step": 465
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6808184143222505e-05,
+      "loss": 0.5458,
+      "step": 466
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6800511508951407e-05,
+      "loss": 0.1289,
+      "step": 467
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.679283887468031e-05,
+      "loss": 0.4383,
+      "step": 468
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6785166240409207e-05,
+      "loss": 0.5892,
+      "step": 469
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.677749360613811e-05,
+      "loss": 1.3565,
+      "step": 470
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.676982097186701e-05,
+      "loss": 0.7267,
+      "step": 471
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.6762148337595908e-05,
+      "loss": 0.1354,
+      "step": 472
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.675447570332481e-05,
+      "loss": 0.9291,
+      "step": 473
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.6746803069053708e-05,
+      "loss": 0.5265,
+      "step": 474
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673913043478261e-05,
+      "loss": 0.4841,
+      "step": 475
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673145780051151e-05,
+      "loss": 1.2391,
+      "step": 476
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.672378516624041e-05,
+      "loss": 0.4924,
+      "step": 477
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6716112531969308e-05,
+      "loss": 1.0076,
+      "step": 478
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6708439897698213e-05,
+      "loss": 0.6314,
+      "step": 479
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.670076726342711e-05,
+      "loss": 0.4237,
+      "step": 480
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.669309462915601e-05,
+      "loss": 0.1077,
+      "step": 481
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.668542199488491e-05,
+      "loss": 0.3038,
+      "step": 482
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6677749360613812e-05,
+      "loss": 0.6081,
+      "step": 483
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.667007672634271e-05,
+      "loss": 0.1078,
+      "step": 484
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6662404092071612e-05,
+      "loss": 0.4296,
+      "step": 485
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.665473145780051e-05,
+      "loss": 0.3981,
+      "step": 486
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6647058823529412e-05,
+      "loss": 0.5853,
+      "step": 487
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6639386189258314e-05,
+      "loss": 0.802,
+      "step": 488
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6631713554987212e-05,
+      "loss": 0.4576,
+      "step": 489
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.662404092071611e-05,
+      "loss": 0.2682,
+      "step": 490
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6616368286445015e-05,
+      "loss": 0.1991,
+      "step": 491
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6608695652173913e-05,
+      "loss": 0.1409,
+      "step": 492
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.660102301790281e-05,
+      "loss": 0.3869,
+      "step": 493
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6593350383631713e-05,
+      "loss": 0.5301,
+      "step": 494
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6585677749360615e-05,
+      "loss": 0.8116,
+      "step": 495
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6578005115089513e-05,
+      "loss": 0.0984,
+      "step": 496
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6570332480818415e-05,
+      "loss": 0.7773,
+      "step": 497
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6562659846547316e-05,
+      "loss": 1.3522,
+      "step": 498
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6554987212276215e-05,
+      "loss": 0.0862,
+      "step": 499
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6547314578005116e-05,
+      "loss": 0.0894,
+      "step": 500
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6539641943734014e-05,
+      "loss": 0.0885,
+      "step": 501
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.653196930946292e-05,
+      "loss": 0.3031,
+      "step": 502
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6524296675191818e-05,
+      "loss": 0.187,
+      "step": 503
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6516624040920716e-05,
+      "loss": 1.3063,
+      "step": 504
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6508951406649617e-05,
+      "loss": 0.6847,
+      "step": 505
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.650127877237852e-05,
+      "loss": 0.078,
+      "step": 506
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6493606138107417e-05,
+      "loss": 0.1265,
+      "step": 507
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.648593350383632e-05,
+      "loss": 0.0804,
+      "step": 508
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.6478260869565217e-05,
+      "loss": 0.6976,
+      "step": 509
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.647058823529412e-05,
+      "loss": 0.7977,
+      "step": 510
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.646291560102302e-05,
+      "loss": 0.413,
+      "step": 511
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.645524296675192e-05,
+      "loss": 0.6944,
+      "step": 512
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6447570332480817e-05,
+      "loss": 0.7177,
+      "step": 513
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6439897698209722e-05,
+      "loss": 0.0581,
+      "step": 514
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.643222506393862e-05,
+      "loss": 0.9308,
+      "step": 515
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6424552429667518e-05,
+      "loss": 0.2518,
+      "step": 516
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.641687979539642e-05,
+      "loss": 0.1014,
+      "step": 517
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640920716112532e-05,
+      "loss": 0.1641,
+      "step": 518
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640153452685422e-05,
+      "loss": 0.101,
+      "step": 519
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.639386189258312e-05,
+      "loss": 0.1097,
+      "step": 520
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6386189258312023e-05,
+      "loss": 0.987,
+      "step": 521
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.637851662404092e-05,
+      "loss": 0.6465,
+      "step": 522
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6370843989769823e-05,
+      "loss": 0.0914,
+      "step": 523
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.636317135549872e-05,
+      "loss": 0.139,
+      "step": 524
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6355498721227623e-05,
+      "loss": 0.6455,
+      "step": 525
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6347826086956524e-05,
+      "loss": 0.204,
+      "step": 526
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6340153452685422e-05,
+      "loss": 0.1269,
+      "step": 527
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.633248081841432e-05,
+      "loss": 0.2214,
+      "step": 528
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6324808184143226e-05,
+      "loss": 0.0736,
+      "step": 529
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6317135549872124e-05,
+      "loss": 0.0714,
+      "step": 530
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6309462915601022e-05,
+      "loss": 0.1388,
+      "step": 531
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6301790281329924e-05,
+      "loss": 0.7758,
+      "step": 532
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6294117647058825e-05,
+      "loss": 0.3388,
+      "step": 533
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6286445012787724e-05,
+      "loss": 1.1831,
+      "step": 534
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6278772378516625e-05,
+      "loss": 0.1166,
+      "step": 535
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6271099744245523e-05,
+      "loss": 0.2389,
+      "step": 536
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6263427109974425e-05,
+      "loss": 0.1622,
+      "step": 537
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6255754475703327e-05,
+      "loss": 0.1015,
+      "step": 538
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6248081841432225e-05,
+      "loss": 0.0658,
+      "step": 539
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6240409207161123e-05,
+      "loss": 0.0954,
+      "step": 540
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6232736572890028e-05,
+      "loss": 0.1182,
+      "step": 541
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6225063938618926e-05,
+      "loss": 0.0605,
+      "step": 542
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6217391304347824e-05,
+      "loss": 0.9265,
+      "step": 543
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6209718670076726e-05,
+      "loss": 1.2994,
+      "step": 544
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6202046035805628e-05,
+      "loss": 0.081,
+      "step": 545
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6194373401534526e-05,
+      "loss": 0.5962,
+      "step": 546
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6186700767263428e-05,
+      "loss": 0.0665,
+      "step": 547
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.617902813299233e-05,
+      "loss": 0.0655,
+      "step": 548
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6171355498721227e-05,
+      "loss": 0.0642,
+      "step": 549
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.616368286445013e-05,
+      "loss": 0.0834,
+      "step": 550
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6156010230179027e-05,
+      "loss": 0.047,
+      "step": 551
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6148337595907932e-05,
+      "loss": 0.0642,
+      "step": 552
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.614066496163683e-05,
+      "loss": 0.0594,
+      "step": 553
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.613299232736573e-05,
+      "loss": 0.0573,
+      "step": 554
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.612531969309463e-05,
+      "loss": 0.1101,
+      "step": 555
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.6117647058823532e-05,
+      "loss": 0.5121,
+      "step": 556
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.610997442455243e-05,
+      "loss": 0.0374,
+      "step": 557
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.6102301790281332e-05,
+      "loss": 0.0957,
+      "step": 558
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.609462915601023e-05,
+      "loss": 0.0488,
+      "step": 559
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.608695652173913e-05,
+      "loss": 0.0482,
+      "step": 560
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6079283887468033e-05,
+      "loss": 1.2041,
+      "step": 561
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.607161125319693e-05,
+      "loss": 0.8833,
+      "step": 562
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.606393861892583e-05,
+      "loss": 0.1544,
+      "step": 563
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6056265984654735e-05,
+      "loss": 0.046,
+      "step": 564
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6048593350383633e-05,
+      "loss": 0.9073,
+      "step": 565
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.604092071611253e-05,
+      "loss": 0.052,
+      "step": 566
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6033248081841433e-05,
+      "loss": 0.0308,
+      "step": 567
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6025575447570334e-05,
+      "loss": 0.7032,
+      "step": 568
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6017902813299232e-05,
+      "loss": 0.6947,
+      "step": 569
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6010230179028134e-05,
+      "loss": 1.5603,
+      "step": 570
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6002557544757036e-05,
+      "loss": 0.8836,
+      "step": 571
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.5994884910485934e-05,
+      "loss": 1.7439,
+      "step": 572
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5987212276214836e-05,
+      "loss": 0.0313,
+      "step": 573
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5979539641943734e-05,
+      "loss": 0.0317,
+      "step": 574
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5971867007672635e-05,
+      "loss": 0.0325,
+      "step": 575
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5964194373401537e-05,
+      "loss": 0.5,
+      "step": 576
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5956521739130435e-05,
+      "loss": 0.0675,
+      "step": 577
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5948849104859333e-05,
+      "loss": 0.6364,
+      "step": 578
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.594117647058824e-05,
+      "loss": 0.0696,
+      "step": 579
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5933503836317137e-05,
+      "loss": 0.6195,
+      "step": 580
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5925831202046035e-05,
+      "loss": 0.5796,
+      "step": 581
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5918158567774937e-05,
+      "loss": 0.0832,
+      "step": 582
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5910485933503838e-05,
+      "loss": 0.0794,
+      "step": 583
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5902813299232736e-05,
+      "loss": 0.5851,
+      "step": 584
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5895140664961638e-05,
+      "loss": 0.104,
+      "step": 585
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5887468030690536e-05,
+      "loss": 1.0492,
+      "step": 586
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5879795396419438e-05,
+      "loss": 0.0645,
+      "step": 587
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.587212276214834e-05,
+      "loss": 0.514,
+      "step": 588
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5864450127877238e-05,
+      "loss": 0.0701,
+      "step": 589
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5856777493606136e-05,
+      "loss": 0.5409,
+      "step": 590
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584910485933504e-05,
+      "loss": 0.0536,
+      "step": 591
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584143222506394e-05,
+      "loss": 0.1728,
+      "step": 592
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.5833759590792837e-05,
+      "loss": 0.0845,
+      "step": 593
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.582608695652174e-05,
+      "loss": 0.2217,
+      "step": 594
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.581841432225064e-05,
+      "loss": 0.1397,
+      "step": 595
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.581074168797954e-05,
+      "loss": 0.4119,
+      "step": 596
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.580306905370844e-05,
+      "loss": 0.3638,
+      "step": 597
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.5795396419437342e-05,
+      "loss": 0.0781,
+      "step": 598
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.578772378516624e-05,
+      "loss": 0.5926,
+      "step": 599
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5780051150895142e-05,
+      "loss": 0.1378,
+      "step": 600
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.577237851662404e-05,
+      "loss": 0.142,
+      "step": 601
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.576470588235294e-05,
+      "loss": 0.1722,
+      "step": 602
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5757033248081843e-05,
+      "loss": 0.1529,
+      "step": 603
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.574936061381074e-05,
+      "loss": 1.6083,
+      "step": 604
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5741687979539643e-05,
+      "loss": 1.8267,
+      "step": 605
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5734015345268545e-05,
+      "loss": 0.0535,
+      "step": 606
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5726342710997443e-05,
+      "loss": 0.9945,
+      "step": 607
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5718670076726345e-05,
+      "loss": 0.9133,
+      "step": 608
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5710997442455243e-05,
+      "loss": 0.5295,
+      "step": 609
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5703324808184144e-05,
+      "loss": 0.058,
+      "step": 610
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5695652173913046e-05,
+      "loss": 0.4789,
+      "step": 611
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5687979539641944e-05,
+      "loss": 0.0572,
+      "step": 612
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5680306905370842e-05,
+      "loss": 0.0897,
+      "step": 613
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5672634271099747e-05,
+      "loss": 0.056,
+      "step": 614
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5664961636828646e-05,
+      "loss": 0.0278,
+      "step": 615
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5657289002557544e-05,
+      "loss": 0.9947,
+      "step": 616
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5649616368286445e-05,
+      "loss": 0.0627,
+      "step": 617
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5641943734015347e-05,
+      "loss": 0.9335,
+      "step": 618
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5634271099744245e-05,
+      "loss": 0.9793,
+      "step": 619
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5626598465473147e-05,
+      "loss": 1.8128,
+      "step": 620
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.561892583120205e-05,
+      "loss": 1.8015,
+      "step": 621
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5611253196930947e-05,
+      "loss": 0.9259,
+      "step": 622
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.560358056265985e-05,
+      "loss": 0.9618,
+      "step": 623
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5595907928388747e-05,
+      "loss": 0.0808,
+      "step": 624
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5588235294117648e-05,
+      "loss": 1.7161,
+      "step": 625
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.558056265984655e-05,
+      "loss": 0.8879,
+      "step": 626
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5572890025575448e-05,
+      "loss": 0.043,
+      "step": 627
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5565217391304346e-05,
+      "loss": 0.9449,
+      "step": 628
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.555754475703325e-05,
+      "loss": 0.0624,
+      "step": 629
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.554987212276215e-05,
+      "loss": 0.0924,
+      "step": 630
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5542199488491048e-05,
+      "loss": 0.0715,
+      "step": 631
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.553452685421995e-05,
+      "loss": 0.4995,
+      "step": 632
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.552685421994885e-05,
+      "loss": 0.5576,
+      "step": 633
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551918158567775e-05,
+      "loss": 0.06,
+      "step": 634
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551150895140665e-05,
+      "loss": 0.0873,
+      "step": 635
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.550383631713555e-05,
+      "loss": 0.6868,
+      "step": 636
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.549616368286445e-05,
+      "loss": 0.8487,
+      "step": 637
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.5488491048593352e-05,
+      "loss": 0.5641,
+      "step": 638
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.548081841432225e-05,
+      "loss": 1.1343,
+      "step": 639
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.547314578005115e-05,
+      "loss": 0.0614,
+      "step": 640
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5465473145780054e-05,
+      "loss": 0.9414,
+      "step": 641
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5457800511508952e-05,
+      "loss": 0.9226,
+      "step": 642
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.545012787723785e-05,
+      "loss": 0.3323,
+      "step": 643
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5442455242966752e-05,
+      "loss": 0.9315,
+      "step": 644
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5434782608695653e-05,
+      "loss": 0.5539,
+      "step": 645
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.542710997442455e-05,
+      "loss": 0.3376,
+      "step": 646
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5419437340153453e-05,
+      "loss": 0.0694,
+      "step": 647
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5411764705882355e-05,
+      "loss": 0.0706,
+      "step": 648
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5404092071611253e-05,
+      "loss": 0.105,
+      "step": 649
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5396419437340155e-05,
+      "loss": 0.912,
+      "step": 650
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5388746803069053e-05,
+      "loss": 0.0998,
+      "step": 651
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5381074168797954e-05,
+      "loss": 0.2247,
+      "step": 652
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5373401534526856e-05,
+      "loss": 0.2869,
+      "step": 653
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5365728900255754e-05,
+      "loss": 0.532,
+      "step": 654
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5358056265984653e-05,
+      "loss": 0.4723,
+      "step": 655
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5350383631713558e-05,
+      "loss": 0.1178,
+      "step": 656
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5342710997442456e-05,
+      "loss": 0.0566,
+      "step": 657
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5335038363171357e-05,
+      "loss": 0.059,
+      "step": 658
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5327365728900256e-05,
+      "loss": 0.0881,
+      "step": 659
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5319693094629157e-05,
+      "loss": 0.5762,
+      "step": 660
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.531202046035806e-05,
+      "loss": 0.0499,
+      "step": 661
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5304347826086957e-05,
+      "loss": 0.087,
+      "step": 662
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5296675191815855e-05,
+      "loss": 1.0719,
+      "step": 663
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528900255754476e-05,
+      "loss": 0.6158,
+      "step": 664
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528132992327366e-05,
+      "loss": 0.6117,
+      "step": 665
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5273657289002557e-05,
+      "loss": 0.0515,
+      "step": 666
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5265984654731458e-05,
+      "loss": 0.0608,
+      "step": 667
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.525831202046036e-05,
+      "loss": 0.0313,
+      "step": 668
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.5250639386189258e-05,
+      "loss": 0.1449,
+      "step": 669
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.524296675191816e-05,
+      "loss": 0.9883,
+      "step": 670
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.523529411764706e-05,
+      "loss": 0.0859,
+      "step": 671
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.522762148337596e-05,
+      "loss": 0.0561,
+      "step": 672
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521994884910486e-05,
+      "loss": 0.0926,
+      "step": 673
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521227621483376e-05,
+      "loss": 0.1101,
+      "step": 674
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.520460358056266e-05,
+      "loss": 0.068,
+      "step": 675
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5196930946291563e-05,
+      "loss": 0.0773,
+      "step": 676
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518925831202046e-05,
+      "loss": 0.0508,
+      "step": 677
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518158567774936e-05,
+      "loss": 0.0375,
+      "step": 678
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5173913043478264e-05,
+      "loss": 0.93,
+      "step": 679
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5166240409207162e-05,
+      "loss": 0.1507,
+      "step": 680
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.515856777493606e-05,
+      "loss": 0.0917,
+      "step": 681
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5150895140664962e-05,
+      "loss": 0.97,
+      "step": 682
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5143222506393864e-05,
+      "loss": 0.5019,
+      "step": 683
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5135549872122762e-05,
+      "loss": 0.0849,
+      "step": 684
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5127877237851664e-05,
+      "loss": 1.2316,
+      "step": 685
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5120204603580562e-05,
+      "loss": 0.0252,
+      "step": 686
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5112531969309463e-05,
+      "loss": 1.0543,
+      "step": 687
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5104859335038365e-05,
+      "loss": 0.0506,
+      "step": 688
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5097186700767263e-05,
+      "loss": 0.0252,
+      "step": 689
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.508951406649616e-05,
+      "loss": 0.6297,
+      "step": 690
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5081841432225066e-05,
+      "loss": 0.4497,
+      "step": 691
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5074168797953965e-05,
+      "loss": 0.0516,
+      "step": 692
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5066496163682863e-05,
+      "loss": 0.5501,
+      "step": 693
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5058823529411768e-05,
+      "loss": 0.5621,
+      "step": 694
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5051150895140666e-05,
+      "loss": 0.0577,
+      "step": 695
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5043478260869564e-05,
+      "loss": 0.0271,
+      "step": 696
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5035805626598466e-05,
+      "loss": 0.9496,
+      "step": 697
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5028132992327368e-05,
+      "loss": 0.0877,
+      "step": 698
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5020460358056266e-05,
+      "loss": 0.0523,
+      "step": 699
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5012787723785167e-05,
+      "loss": 0.0577,
+      "step": 700
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5005115089514066e-05,
+      "loss": 0.0522,
+      "step": 701
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.4997442455242967e-05,
+      "loss": 0.0517,
+      "step": 702
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.498976982097187e-05,
+      "loss": 0.6335,
+      "step": 703
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4982097186700767e-05,
+      "loss": 0.05,
+      "step": 704
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4974424552429665e-05,
+      "loss": 0.0753,
+      "step": 705
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.496675191815857e-05,
+      "loss": 0.0227,
+      "step": 706
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.495907928388747e-05,
+      "loss": 0.0755,
+      "step": 707
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.4951406649616367e-05,
+      "loss": 0.0215,
+      "step": 708
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.494373401534527e-05,
+      "loss": 0.0451,
+      "step": 709
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.493606138107417e-05,
+      "loss": 0.0343,
+      "step": 710
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492838874680307e-05,
+      "loss": 0.0836,
+      "step": 711
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492071611253197e-05,
+      "loss": 0.5287,
+      "step": 712
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4913043478260868e-05,
+      "loss": 0.0686,
+      "step": 713
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4905370843989773e-05,
+      "loss": 0.0211,
+      "step": 714
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489769820971867e-05,
+      "loss": 0.5932,
+      "step": 715
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489002557544757e-05,
+      "loss": 0.0371,
+      "step": 716
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.488235294117647e-05,
+      "loss": 1.0137,
+      "step": 717
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.4874680306905373e-05,
+      "loss": 0.9928,
+      "step": 718
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.486700767263427e-05,
+      "loss": 0.7431,
+      "step": 719
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4859335038363173e-05,
+      "loss": 0.0207,
+      "step": 720
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4851662404092074e-05,
+      "loss": 0.0681,
+      "step": 721
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4843989769820972e-05,
+      "loss": 0.0306,
+      "step": 722
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4836317135549874e-05,
+      "loss": 0.0204,
+      "step": 723
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4828644501278772e-05,
+      "loss": 0.0344,
+      "step": 724
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4820971867007674e-05,
+      "loss": 0.0345,
+      "step": 725
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4813299232736575e-05,
+      "loss": 0.9275,
+      "step": 726
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4805626598465474e-05,
+      "loss": 0.0196,
+      "step": 727
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4797953964194372e-05,
+      "loss": 1.0137,
+      "step": 728
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4790281329923277e-05,
+      "loss": 0.0318,
+      "step": 729
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4782608695652175e-05,
+      "loss": 0.0688,
+      "step": 730
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4774936061381073e-05,
+      "loss": 0.0314,
+      "step": 731
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4767263427109975e-05,
+      "loss": 0.7135,
+      "step": 732
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4759590792838877e-05,
+      "loss": 0.0299,
+      "step": 733
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4751918158567775e-05,
+      "loss": 1.0127,
+      "step": 734
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4744245524296676e-05,
+      "loss": 0.0304,
+      "step": 735
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4736572890025575e-05,
+      "loss": 1.4618,
+      "step": 736
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4728900255754476e-05,
+      "loss": 0.7167,
+      "step": 737
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4721227621483378e-05,
+      "loss": 0.0307,
+      "step": 738
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4713554987212276e-05,
+      "loss": 0.052,
+      "step": 739
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4705882352941174e-05,
+      "loss": 0.778,
+      "step": 740
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.469820971867008e-05,
+      "loss": 0.5482,
+      "step": 741
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4690537084398978e-05,
+      "loss": 1.4314,
+      "step": 742
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4682864450127876e-05,
+      "loss": 0.0422,
+      "step": 743
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.467519181585678e-05,
+      "loss": 0.6614,
+      "step": 744
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.466751918158568e-05,
+      "loss": 0.0312,
+      "step": 745
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4659846547314577e-05,
+      "loss": 0.043,
+      "step": 746
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.465217391304348e-05,
+      "loss": 0.0199,
+      "step": 747
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.464450127877238e-05,
+      "loss": 0.0832,
+      "step": 748
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.463682864450128e-05,
+      "loss": 0.0312,
+      "step": 749
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.462915601023018e-05,
+      "loss": 0.0309,
+      "step": 750
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.462148337595908e-05,
+      "loss": 0.0193,
+      "step": 751
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.461381074168798e-05,
+      "loss": 0.0446,
+      "step": 752
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.4606138107416882e-05,
+      "loss": 0.9993,
+      "step": 753
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.459846547314578e-05,
+      "loss": 0.016,
+      "step": 754
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4590792838874678e-05,
+      "loss": 0.0433,
+      "step": 755
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4583120204603583e-05,
+      "loss": 0.2614,
+      "step": 756
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.457544757033248e-05,
+      "loss": 0.018,
+      "step": 757
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.456777493606138e-05,
+      "loss": 0.6676,
+      "step": 758
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.456010230179028e-05,
+      "loss": 0.0599,
+      "step": 759
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4552429667519183e-05,
+      "loss": 0.0338,
+      "step": 760
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.454475703324808e-05,
+      "loss": 0.0268,
+      "step": 761
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4537084398976983e-05,
+      "loss": 0.0262,
+      "step": 762
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.452941176470588e-05,
+      "loss": 0.0252,
+      "step": 763
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4521739130434786e-05,
+      "loss": 0.0377,
+      "step": 764
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4514066496163684e-05,
+      "loss": 1.4866,
+      "step": 765
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4506393861892582e-05,
+      "loss": 1.0524,
+      "step": 766
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4498721227621484e-05,
+      "loss": 0.0328,
+      "step": 767
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4491048593350386e-05,
+      "loss": 0.024,
+      "step": 768
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4483375959079284e-05,
+      "loss": 0.022,
+      "step": 769
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4475703324808185e-05,
+      "loss": 0.0245,
+      "step": 770
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4468030690537087e-05,
+      "loss": 1.0107,
+      "step": 771
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4460358056265985e-05,
+      "loss": 0.0336,
+      "step": 772
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4452685421994887e-05,
+      "loss": 0.0391,
+      "step": 773
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4445012787723785e-05,
+      "loss": 0.0303,
+      "step": 774
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4437340153452687e-05,
+      "loss": 0.0443,
+      "step": 775
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4429667519181588e-05,
+      "loss": 0.0292,
+      "step": 776
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4421994884910487e-05,
+      "loss": 0.0284,
+      "step": 777
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4414322250639385e-05,
+      "loss": 1.5985,
+      "step": 778
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.440664961636829e-05,
+      "loss": 0.0114,
+      "step": 779
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4398976982097188e-05,
+      "loss": 0.0342,
+      "step": 780
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4391304347826086e-05,
+      "loss": 1.8501,
+      "step": 781
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4383631713554988e-05,
+      "loss": 0.5375,
+      "step": 782
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.437595907928389e-05,
+      "loss": 1.589,
+      "step": 783
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4368286445012788e-05,
+      "loss": 0.0221,
+      "step": 784
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.436061381074169e-05,
+      "loss": 0.0453,
+      "step": 785
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4352941176470587e-05,
+      "loss": 0.796,
+      "step": 786
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.434526854219949e-05,
+      "loss": 0.1249,
+      "step": 787
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.433759590792839e-05,
+      "loss": 0.7963,
+      "step": 788
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.432992327365729e-05,
+      "loss": 0.4231,
+      "step": 789
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.4322250639386187e-05,
+      "loss": 0.0247,
+      "step": 790
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.4314578005115092e-05,
+      "loss": 0.0295,
+      "step": 791
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.430690537084399e-05,
+      "loss": 0.0249,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.3537317216396332,
+      "eval_runtime": 58.1226,
+      "eval_samples_per_second": 6.813,
+      "eval_steps_per_second": 1.703,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.429923273657289e-05,
+      "loss": 0.0157,
+      "step": 793
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4291560102301794e-05,
+      "loss": 0.5222,
+      "step": 794
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4283887468030692e-05,
+      "loss": 0.0226,
+      "step": 795
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.427621483375959e-05,
+      "loss": 0.0293,
+      "step": 796
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.426854219948849e-05,
+      "loss": 0.015,
+      "step": 797
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4260869565217393e-05,
+      "loss": 1.0658,
+      "step": 798
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.425319693094629e-05,
+      "loss": 0.0412,
+      "step": 799
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4245524296675193e-05,
+      "loss": 0.0219,
+      "step": 800
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.423785166240409e-05,
+      "loss": 0.0149,
+      "step": 801
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4230179028132993e-05,
+      "loss": 0.0323,
+      "step": 802
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4222506393861895e-05,
+      "loss": 1.0764,
+      "step": 803
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4214833759590793e-05,
+      "loss": 0.7518,
+      "step": 804
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.420716112531969e-05,
+      "loss": 0.8166,
+      "step": 805
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4199488491048596e-05,
+      "loss": 0.0219,
+      "step": 806
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4191815856777494e-05,
+      "loss": 0.0154,
+      "step": 807
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4184143222506392e-05,
+      "loss": 0.0163,
+      "step": 808
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4176470588235294e-05,
+      "loss": 1.0562,
+      "step": 809
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4168797953964196e-05,
+      "loss": 0.0218,
+      "step": 810
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4161125319693094e-05,
+      "loss": 0.0232,
+      "step": 811
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4153452685421995e-05,
+      "loss": 0.2384,
+      "step": 812
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4145780051150894e-05,
+      "loss": 1.5286,
+      "step": 813
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4138107416879795e-05,
+      "loss": 0.0262,
+      "step": 814
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4130434782608697e-05,
+      "loss": 0.0334,
+      "step": 815
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4122762148337595e-05,
+      "loss": 0.6509,
+      "step": 816
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4115089514066497e-05,
+      "loss": 0.0159,
+      "step": 817
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.41074168797954e-05,
+      "loss": 0.6225,
+      "step": 818
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4099744245524297e-05,
+      "loss": 0.752,
+      "step": 819
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4092071611253198e-05,
+      "loss": 1.0594,
+      "step": 820
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.40843989769821e-05,
+      "loss": 0.0161,
+      "step": 821
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4076726342710998e-05,
+      "loss": 0.7492,
+      "step": 822
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40690537084399e-05,
+      "loss": 0.0302,
+      "step": 823
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4061381074168798e-05,
+      "loss": 0.038,
+      "step": 824
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40537084398977e-05,
+      "loss": 0.0243,
+      "step": 825
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40460358056266e-05,
+      "loss": 0.0331,
+      "step": 826
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40383631713555e-05,
+      "loss": 1.0252,
+      "step": 827
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4030690537084398e-05,
+      "loss": 0.0172,
+      "step": 828
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4023017902813303e-05,
+      "loss": 0.6168,
+      "step": 829
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40153452685422e-05,
+      "loss": 0.3135,
+      "step": 830
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40076726342711e-05,
+      "loss": 0.0168,
+      "step": 831
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.0172,
+      "step": 832
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.3992327365728902e-05,
+      "loss": 1.0349,
+      "step": 833
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39846547314578e-05,
+      "loss": 0.0978,
+      "step": 834
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3976982097186702e-05,
+      "loss": 0.0167,
+      "step": 835
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39693094629156e-05,
+      "loss": 1.0381,
+      "step": 836
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3961636828644502e-05,
+      "loss": 0.6857,
+      "step": 837
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3953964194373404e-05,
+      "loss": 1.0406,
+      "step": 838
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3946291560102302e-05,
+      "loss": 0.071,
+      "step": 839
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.39386189258312e-05,
+      "loss": 0.0193,
+      "step": 840
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3930946291560105e-05,
+      "loss": 0.0193,
+      "step": 841
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3923273657289003e-05,
+      "loss": 0.9916,
+      "step": 842
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.39156010230179e-05,
+      "loss": 0.0723,
+      "step": 843
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3907928388746806e-05,
+      "loss": 0.0626,
+      "step": 844
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3900255754475705e-05,
+      "loss": 0.7264,
+      "step": 845
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3892583120204603e-05,
+      "loss": 0.827,
+      "step": 846
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3884910485933504e-05,
+      "loss": 1.2193,
+      "step": 847
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3877237851662406e-05,
+      "loss": 0.6754,
+      "step": 848
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3869565217391304e-05,
+      "loss": 0.6856,
+      "step": 849
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3861892583120206e-05,
+      "loss": 0.7441,
+      "step": 850
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3854219948849104e-05,
+      "loss": 0.023,
+      "step": 851
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3846547314578006e-05,
+      "loss": 0.4742,
+      "step": 852
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3838874680306907e-05,
+      "loss": 0.0363,
+      "step": 853
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3831202046035806e-05,
+      "loss": 0.0409,
+      "step": 854
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3823529411764704e-05,
+      "loss": 0.8353,
+      "step": 855
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.381585677749361e-05,
+      "loss": 0.0333,
+      "step": 856
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3808184143222507e-05,
+      "loss": 0.0216,
+      "step": 857
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3800511508951405e-05,
+      "loss": 0.9688,
+      "step": 858
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3792838874680307e-05,
+      "loss": 1.0078,
+      "step": 859
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.378516624040921e-05,
+      "loss": 0.0183,
+      "step": 860
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3777493606138107e-05,
+      "loss": 0.0183,
+      "step": 861
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3769820971867008e-05,
+      "loss": 0.9786,
+      "step": 862
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3762148337595907e-05,
+      "loss": 0.9166,
+      "step": 863
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3754475703324808e-05,
+      "loss": 1.1408,
+      "step": 864
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.374680306905371e-05,
+      "loss": 1.1394,
+      "step": 865
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.3739130434782608e-05,
+      "loss": 0.0185,
+      "step": 866
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.373145780051151e-05,
+      "loss": 0.9979,
+      "step": 867
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.372378516624041e-05,
+      "loss": 1.0092,
+      "step": 868
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.371611253196931e-05,
+      "loss": 0.7276,
+      "step": 869
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.370843989769821e-05,
+      "loss": 0.0359,
+      "step": 870
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3700767263427113e-05,
+      "loss": 0.0864,
+      "step": 871
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.369309462915601e-05,
+      "loss": 0.0246,
+      "step": 872
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3685421994884912e-05,
+      "loss": 0.8518,
+      "step": 873
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.367774936061381e-05,
+      "loss": 1.433,
+      "step": 874
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3670076726342712e-05,
+      "loss": 0.035,
+      "step": 875
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3662404092071614e-05,
+      "loss": 0.5922,
+      "step": 876
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3654731457800512e-05,
+      "loss": 0.529,
+      "step": 877
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.364705882352941e-05,
+      "loss": 0.6423,
+      "step": 878
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3639386189258315e-05,
+      "loss": 0.0337,
+      "step": 879
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3631713554987214e-05,
+      "loss": 0.0402,
+      "step": 880
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3624040920716112e-05,
+      "loss": 0.0782,
+      "step": 881
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3616368286445013e-05,
+      "loss": 0.4141,
+      "step": 882
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3608695652173915e-05,
+      "loss": 0.0359,
+      "step": 883
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3601023017902813e-05,
+      "loss": 0.4509,
+      "step": 884
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3593350383631715e-05,
+      "loss": 0.6414,
+      "step": 885
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3585677749360613e-05,
+      "loss": 0.6873,
+      "step": 886
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3578005115089515e-05,
+      "loss": 0.0304,
+      "step": 887
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3570332480818416e-05,
+      "loss": 0.0315,
+      "step": 888
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3562659846547315e-05,
+      "loss": 0.0896,
+      "step": 889
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3554987212276213e-05,
+      "loss": 0.0201,
+      "step": 890
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3547314578005118e-05,
+      "loss": 0.027,
+      "step": 891
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3539641943734016e-05,
+      "loss": 1.9853,
+      "step": 892
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3531969309462914e-05,
+      "loss": 2.9647,
+      "step": 893
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.352429667519182e-05,
+      "loss": 1.9403,
+      "step": 894
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3516624040920717e-05,
+      "loss": 0.0193,
+      "step": 895
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3508951406649616e-05,
+      "loss": 0.9912,
+      "step": 896
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3501278772378517e-05,
+      "loss": 0.02,
+      "step": 897
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.349360613810742e-05,
+      "loss": 0.9015,
+      "step": 898
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3485933503836317e-05,
+      "loss": 0.1294,
+      "step": 899
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.347826086956522e-05,
+      "loss": 0.0559,
+      "step": 900
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.3470588235294117e-05,
+      "loss": 1.8039,
+      "step": 901
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.346291560102302e-05,
+      "loss": 0.0336,
+      "step": 902
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.345524296675192e-05,
+      "loss": 0.0383,
+      "step": 903
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.344757033248082e-05,
+      "loss": 0.0722,
+      "step": 904
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3439897698209717e-05,
+      "loss": 0.0278,
+      "step": 905
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.343222506393862e-05,
+      "loss": 0.0684,
+      "step": 906
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.342455242966752e-05,
+      "loss": 0.0236,
+      "step": 907
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3416879795396418e-05,
+      "loss": 0.0348,
+      "step": 908
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340920716112532e-05,
+      "loss": 0.0204,
+      "step": 909
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340153452685422e-05,
+      "loss": 0.8109,
+      "step": 910
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.339386189258312e-05,
+      "loss": 0.0279,
+      "step": 911
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.338618925831202e-05,
+      "loss": 0.0361,
+      "step": 912
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337851662404092e-05,
+      "loss": 0.0355,
+      "step": 913
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337084398976982e-05,
+      "loss": 0.7958,
+      "step": 914
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.3363171355498723e-05,
+      "loss": 0.0362,
+      "step": 915
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.335549872122762e-05,
+      "loss": 0.0252,
+      "step": 916
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3347826086956522e-05,
+      "loss": 0.8165,
+      "step": 917
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3340153452685424e-05,
+      "loss": 0.0237,
+      "step": 918
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3332480818414322e-05,
+      "loss": 0.8162,
+      "step": 919
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.332480818414322e-05,
+      "loss": 0.7786,
+      "step": 920
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3317135549872125e-05,
+      "loss": 0.0787,
+      "step": 921
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3309462915601024e-05,
+      "loss": 0.1764,
+      "step": 922
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3301790281329925e-05,
+      "loss": 0.7449,
+      "step": 923
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3294117647058824e-05,
+      "loss": 0.0452,
+      "step": 924
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3286445012787725e-05,
+      "loss": 0.0181,
+      "step": 925
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3278772378516627e-05,
+      "loss": 0.1783,
+      "step": 926
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3271099744245525e-05,
+      "loss": 0.5158,
+      "step": 927
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3263427109974423e-05,
+      "loss": 0.7574,
+      "step": 928
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3255754475703328e-05,
+      "loss": 2.6099,
+      "step": 929
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3248081841432226e-05,
+      "loss": 0.0304,
+      "step": 930
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3240409207161125e-05,
+      "loss": 0.0795,
+      "step": 931
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3232736572890026e-05,
+      "loss": 0.0342,
+      "step": 932
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3225063938618928e-05,
+      "loss": 0.0473,
+      "step": 933
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3217391304347826e-05,
+      "loss": 0.0831,
+      "step": 934
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3209718670076728e-05,
+      "loss": 0.0465,
+      "step": 935
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3202046035805626e-05,
+      "loss": 0.6769,
+      "step": 936
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3194373401534528e-05,
+      "loss": 0.0173,
+      "step": 937
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.318670076726343e-05,
+      "loss": 0.0169,
+      "step": 938
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3179028132992327e-05,
+      "loss": 0.0469,
+      "step": 939
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3171355498721226e-05,
+      "loss": 0.048,
+      "step": 940
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.316368286445013e-05,
+      "loss": 1.0704,
+      "step": 941
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.315601023017903e-05,
+      "loss": 1.0738,
+      "step": 942
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3148337595907927e-05,
+      "loss": 0.0163,
+      "step": 943
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3140664961636832e-05,
+      "loss": 2.0702,
+      "step": 944
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.313299232736573e-05,
+      "loss": 1.0433,
+      "step": 945
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.312531969309463e-05,
+      "loss": 1.0351,
+      "step": 946
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.311764705882353e-05,
+      "loss": 1.053,
+      "step": 947
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.3109974424552432e-05,
+      "loss": 0.1797,
+      "step": 948
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.310230179028133e-05,
+      "loss": 1.0259,
+      "step": 949
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.309462915601023e-05,
+      "loss": 0.03,
+      "step": 950
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.308695652173913e-05,
+      "loss": 0.905,
+      "step": 951
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.307928388746803e-05,
+      "loss": 0.0699,
+      "step": 952
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3071611253196933e-05,
+      "loss": 1.0013,
+      "step": 953
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.306393861892583e-05,
+      "loss": 0.0319,
+      "step": 954
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.305626598465473e-05,
+      "loss": 0.0361,
+      "step": 955
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3048593350383634e-05,
+      "loss": 0.0201,
+      "step": 956
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3040920716112533e-05,
+      "loss": 0.033,
+      "step": 957
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.303324808184143e-05,
+      "loss": 0.0411,
+      "step": 958
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3025575447570332e-05,
+      "loss": 0.7138,
+      "step": 959
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3017902813299234e-05,
+      "loss": 0.0434,
+      "step": 960
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3010230179028132e-05,
+      "loss": 0.8445,
+      "step": 961
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3002557544757034e-05,
+      "loss": 0.0323,
+      "step": 962
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2994884910485932e-05,
+      "loss": 0.0203,
+      "step": 963
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2987212276214834e-05,
+      "loss": 0.2302,
+      "step": 964
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2979539641943735e-05,
+      "loss": 0.0293,
+      "step": 965
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2971867007672634e-05,
+      "loss": 0.0593,
+      "step": 966
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2964194373401535e-05,
+      "loss": 0.6505,
+      "step": 967
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2956521739130437e-05,
+      "loss": 0.5891,
+      "step": 968
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2948849104859335e-05,
+      "loss": 0.0191,
+      "step": 969
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2941176470588233e-05,
+      "loss": 0.0188,
+      "step": 970
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2933503836317138e-05,
+      "loss": 0.9537,
+      "step": 971
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2925831202046037e-05,
+      "loss": 0.0186,
+      "step": 972
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2918158567774935e-05,
+      "loss": 1.0237,
+      "step": 973
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2910485933503836e-05,
+      "loss": 0.0417,
+      "step": 974
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2902813299232738e-05,
+      "loss": 0.0369,
+      "step": 975
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.289514066496164e-05,
+      "loss": 0.0184,
+      "step": 976
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2887468030690538e-05,
+      "loss": 0.811,
+      "step": 977
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2879795396419436e-05,
+      "loss": 0.0183,
+      "step": 978
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.287212276214834e-05,
+      "loss": 1.0442,
+      "step": 979
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.286445012787724e-05,
+      "loss": 0.027,
+      "step": 980
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.2856777493606137e-05,
+      "loss": 0.062,
+      "step": 981
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284910485933504e-05,
+      "loss": 0.0186,
+      "step": 982
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284143222506394e-05,
+      "loss": 0.6057,
+      "step": 983
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.283375959079284e-05,
+      "loss": 1.6594,
+      "step": 984
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.282608695652174e-05,
+      "loss": 0.0192,
+      "step": 985
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281841432225064e-05,
+      "loss": 1.5361,
+      "step": 986
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281074168797954e-05,
+      "loss": 0.0384,
+      "step": 987
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.2803069053708442e-05,
+      "loss": 2.0238,
+      "step": 988
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.279539641943734e-05,
+      "loss": 0.0334,
+      "step": 989
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2787723785166242e-05,
+      "loss": 0.0571,
+      "step": 990
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2780051150895143e-05,
+      "loss": 0.0211,
+      "step": 991
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.277237851662404e-05,
+      "loss": 0.0409,
+      "step": 992
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.276470588235294e-05,
+      "loss": 0.1059,
+      "step": 993
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2757033248081845e-05,
+      "loss": 0.0277,
+      "step": 994
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2749360613810743e-05,
+      "loss": 1.0023,
+      "step": 995
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.274168797953964e-05,
+      "loss": 0.0186,
+      "step": 996
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2734015345268543e-05,
+      "loss": 0.0345,
+      "step": 997
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2726342710997445e-05,
+      "loss": 0.0169,
+      "step": 998
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2718670076726343e-05,
+      "loss": 0.7292,
+      "step": 999
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2710997442455244e-05,
+      "loss": 1.2655,
+      "step": 1000
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2703324808184143e-05,
+      "loss": 0.03,
+      "step": 1001
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2695652173913044e-05,
+      "loss": 1.0266,
+      "step": 1002
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2687979539641946e-05,
+      "loss": 0.7645,
+      "step": 1003
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2680306905370844e-05,
+      "loss": 0.7985,
+      "step": 1004
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2672634271099742e-05,
+      "loss": 0.76,
+      "step": 1005
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2664961636828647e-05,
+      "loss": 0.7577,
+      "step": 1006
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2657289002557545e-05,
+      "loss": 0.0248,
+      "step": 1007
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2649616368286444e-05,
+      "loss": 0.0262,
+      "step": 1008
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2641943734015345e-05,
+      "loss": 1.9984,
+      "step": 1009
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2634271099744247e-05,
+      "loss": 0.1489,
+      "step": 1010
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2626598465473145e-05,
+      "loss": 0.0683,
+      "step": 1011
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2618925831202047e-05,
+      "loss": 0.054,
+      "step": 1012
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2611253196930945e-05,
+      "loss": 0.0269,
+      "step": 1013
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2603580562659847e-05,
+      "loss": 1.0378,
+      "step": 1014
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2595907928388748e-05,
+      "loss": 0.0171,
+      "step": 1015
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2588235294117646e-05,
+      "loss": 0.8433,
+      "step": 1016
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2580562659846548e-05,
+      "loss": 0.0468,
+      "step": 1017
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.257289002557545e-05,
+      "loss": 0.0331,
+      "step": 1018
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2565217391304348e-05,
+      "loss": 0.0308,
+      "step": 1019
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2557544757033246e-05,
+      "loss": 1.0007,
+      "step": 1020
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254987212276215e-05,
+      "loss": 2.0242,
+      "step": 1021
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254219948849105e-05,
+      "loss": 0.7713,
+      "step": 1022
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2534526854219948e-05,
+      "loss": 1.1112,
+      "step": 1023
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.252685421994885e-05,
+      "loss": 0.0184,
+      "step": 1024
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251918158567775e-05,
+      "loss": 0.0243,
+      "step": 1025
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251150895140665e-05,
+      "loss": 0.046,
+      "step": 1026
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.250383631713555e-05,
+      "loss": 0.0378,
+      "step": 1027
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.249616368286445e-05,
+      "loss": 0.0329,
+      "step": 1028
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2488491048593354e-05,
+      "loss": 0.0245,
+      "step": 1029
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2480818414322252e-05,
+      "loss": 0.0238,
+      "step": 1030
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.247314578005115e-05,
+      "loss": 1.0628,
+      "step": 1031
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2465473145780052e-05,
+      "loss": 0.0703,
+      "step": 1032
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2457800511508954e-05,
+      "loss": 0.0251,
+      "step": 1033
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2450127877237852e-05,
+      "loss": 1.0561,
+      "step": 1034
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2442455242966753e-05,
+      "loss": 0.0274,
+      "step": 1035
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.243478260869565e-05,
+      "loss": 0.023,
+      "step": 1036
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2427109974424553e-05,
+      "loss": 0.0223,
+      "step": 1037
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2419437340153455e-05,
+      "loss": 0.8113,
+      "step": 1038
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2411764705882353e-05,
+      "loss": 0.0346,
+      "step": 1039
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2404092071611255e-05,
+      "loss": 0.0164,
+      "step": 1040
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2396419437340156e-05,
+      "loss": 0.0237,
+      "step": 1041
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2388746803069054e-05,
+      "loss": 0.0305,
+      "step": 1042
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2381074168797953e-05,
+      "loss": 0.0158,
+      "step": 1043
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2373401534526858e-05,
+      "loss": 0.0225,
+      "step": 1044
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2365728900255756e-05,
+      "loss": 0.0226,
+      "step": 1045
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2358056265984654e-05,
+      "loss": 0.015,
+      "step": 1046
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2350383631713556e-05,
+      "loss": 0.0219,
+      "step": 1047
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2342710997442457e-05,
+      "loss": 1.0694,
+      "step": 1048
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2335038363171356e-05,
+      "loss": 0.0348,
+      "step": 1049
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2327365728900257e-05,
+      "loss": 0.0213,
+      "step": 1050
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2319693094629155e-05,
+      "loss": 0.9405,
+      "step": 1051
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2312020460358057e-05,
+      "loss": 0.0252,
+      "step": 1052
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.230434782608696e-05,
+      "loss": 0.0254,
+      "step": 1053
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2296675191815857e-05,
+      "loss": 0.7353,
+      "step": 1054
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2289002557544755e-05,
+      "loss": 0.0253,
+      "step": 1055
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.228132992327366e-05,
+      "loss": 0.8361,
+      "step": 1056
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2273657289002558e-05,
+      "loss": 0.0196,
+      "step": 1057
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2265984654731457e-05,
+      "loss": 0.014,
+      "step": 1058
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2258312020460358e-05,
+      "loss": 0.8297,
+      "step": 1059
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.225063938618926e-05,
+      "loss": 1.0654,
+      "step": 1060
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2242966751918158e-05,
+      "loss": 0.0248,
+      "step": 1061
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.223529411764706e-05,
+      "loss": 0.0226,
+      "step": 1062
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2227621483375958e-05,
+      "loss": 0.0238,
+      "step": 1063
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221994884910486e-05,
+      "loss": 0.0212,
+      "step": 1064
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221227621483376e-05,
+      "loss": 0.0274,
+      "step": 1065
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.220460358056266e-05,
+      "loss": 1.0217,
+      "step": 1066
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.219693094629156e-05,
+      "loss": 0.0252,
+      "step": 1067
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2189258312020462e-05,
+      "loss": 0.0219,
+      "step": 1068
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.218158567774936e-05,
+      "loss": 0.0349,
+      "step": 1069
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.217391304347826e-05,
+      "loss": 0.0152,
+      "step": 1070
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2166240409207164e-05,
+      "loss": 0.021,
+      "step": 1071
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2158567774936062e-05,
+      "loss": 0.015,
+      "step": 1072
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.215089514066496e-05,
+      "loss": 0.0277,
+      "step": 1073
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2143222506393862e-05,
+      "loss": 0.1858,
+      "step": 1074
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2135549872122764e-05,
+      "loss": 0.0136,
+      "step": 1075
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2127877237851662e-05,
+      "loss": 0.0132,
+      "step": 1076
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2120204603580563e-05,
+      "loss": 0.0126,
+      "step": 1077
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.211253196930946e-05,
+      "loss": 0.5171,
+      "step": 1078
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2104859335038363e-05,
+      "loss": 0.0422,
+      "step": 1079
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2097186700767265e-05,
+      "loss": 0.0186,
+      "step": 1080
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2089514066496163e-05,
+      "loss": 0.8412,
+      "step": 1081
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2081841432225065e-05,
+      "loss": 0.0183,
+      "step": 1082
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2074168797953966e-05,
+      "loss": 1.0243,
+      "step": 1083
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2066496163682865e-05,
+      "loss": 0.0212,
+      "step": 1084
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2058823529411766e-05,
+      "loss": 0.2841,
+      "step": 1085
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2051150895140664e-05,
+      "loss": 0.0118,
+      "step": 1086
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2043478260869566e-05,
+      "loss": 0.0247,
+      "step": 1087
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2035805626598468e-05,
+      "loss": 1.5121,
+      "step": 1088
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2028132992327366e-05,
+      "loss": 0.0304,
+      "step": 1089
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2020460358056267e-05,
+      "loss": 0.0329,
+      "step": 1090
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.201278772378517e-05,
+      "loss": 1.1288,
+      "step": 1091
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.2005115089514067e-05,
+      "loss": 0.0145,
+      "step": 1092
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.1997442455242966e-05,
+      "loss": 0.0498,
+      "step": 1093
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.198976982097187e-05,
+      "loss": 0.0244,
+      "step": 1094
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.198209718670077e-05,
+      "loss": 0.0112,
+      "step": 1095
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.1974424552429667e-05,
+      "loss": 0.8642,
+      "step": 1096
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.196675191815857e-05,
+      "loss": 0.8822,
+      "step": 1097
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.195907928388747e-05,
+      "loss": 0.0278,
+      "step": 1098
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.195140664961637e-05,
+      "loss": 0.6954,
+      "step": 1099
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.194373401534527e-05,
+      "loss": 0.0179,
+      "step": 1100
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.1936061381074168e-05,
+      "loss": 0.5607,
+      "step": 1101
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.192838874680307e-05,
+      "loss": 0.0191,
+      "step": 1102
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.192071611253197e-05,
+      "loss": 1.9519,
+      "step": 1103
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.191304347826087e-05,
+      "loss": 0.8347,
+      "step": 1104
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1905370843989768e-05,
+      "loss": 0.0262,
+      "step": 1105
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1897698209718673e-05,
+      "loss": 0.0477,
+      "step": 1106
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.189002557544757e-05,
+      "loss": 0.0227,
+      "step": 1107
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.188235294117647e-05,
+      "loss": 0.0205,
+      "step": 1108
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.187468030690537e-05,
+      "loss": 0.0208,
+      "step": 1109
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.1867007672634273e-05,
+      "loss": 0.0116,
+      "step": 1110
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.185933503836317e-05,
+      "loss": 1.1546,
+      "step": 1111
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1851662404092072e-05,
+      "loss": 0.0114,
+      "step": 1112
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.184398976982097e-05,
+      "loss": 0.0113,
+      "step": 1113
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1836317135549872e-05,
+      "loss": 0.6881,
+      "step": 1114
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1828644501278774e-05,
+      "loss": 0.0294,
+      "step": 1115
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1820971867007672e-05,
+      "loss": 0.1569,
+      "step": 1116
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1813299232736574e-05,
+      "loss": 1.1362,
+      "step": 1117
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1805626598465475e-05,
+      "loss": 0.0196,
+      "step": 1118
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1797953964194374e-05,
+      "loss": 0.0205,
+      "step": 1119
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1790281329923272e-05,
+      "loss": 0.0198,
+      "step": 1120
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1782608695652177e-05,
+      "loss": 0.0113,
+      "step": 1121
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1774936061381075e-05,
+      "loss": 0.02,
+      "step": 1122
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1767263427109973e-05,
+      "loss": 0.0195,
+      "step": 1123
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1759590792838875e-05,
+      "loss": 0.8016,
+      "step": 1124
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1751918158567776e-05,
+      "loss": 0.0122,
+      "step": 1125
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1744245524296675e-05,
+      "loss": 1.1132,
+      "step": 1126
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1736572890025576e-05,
+      "loss": 0.011,
+      "step": 1127
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1728900255754474e-05,
+      "loss": 0.7705,
+      "step": 1128
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1721227621483376e-05,
+      "loss": 1.1467,
+      "step": 1129
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1713554987212278e-05,
+      "loss": 0.0176,
+      "step": 1130
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1705882352941176e-05,
+      "loss": 0.0293,
+      "step": 1131
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1698209718670074e-05,
+      "loss": 0.0208,
+      "step": 1132
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.169053708439898e-05,
+      "loss": 0.0199,
+      "step": 1133
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1682864450127877e-05,
+      "loss": 0.0431,
+      "step": 1134
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.167519181585678e-05,
+      "loss": 1.6066,
+      "step": 1135
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.1667519181585677e-05,
+      "loss": 0.029,
+      "step": 1136
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165984654731458e-05,
+      "loss": 0.029,
+      "step": 1137
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165217391304348e-05,
+      "loss": 1.1749,
+      "step": 1138
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.164450127877238e-05,
+      "loss": 0.0189,
+      "step": 1139
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.163682864450128e-05,
+      "loss": 0.0309,
+      "step": 1140
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.1629156010230182e-05,
+      "loss": 0.0287,
+      "step": 1141
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.162148337595908e-05,
+      "loss": 0.0202,
+      "step": 1142
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.161381074168798e-05,
+      "loss": 0.7889,
+      "step": 1143
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.1606138107416883e-05,
+      "loss": 0.0285,
+      "step": 1144
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159846547314578e-05,
+      "loss": 0.0535,
+      "step": 1145
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159079283887468e-05,
+      "loss": 1.0829,
+      "step": 1146
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.158312020460358e-05,
+      "loss": 0.0989,
+      "step": 1147
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1575447570332483e-05,
+      "loss": 0.0115,
+      "step": 1148
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.156777493606138e-05,
+      "loss": 0.0227,
+      "step": 1149
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1560102301790283e-05,
+      "loss": 0.5506,
+      "step": 1150
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.155242966751918e-05,
+      "loss": 1.1234,
+      "step": 1151
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1544757033248083e-05,
+      "loss": 0.0564,
+      "step": 1152
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1537084398976984e-05,
+      "loss": 0.0127,
+      "step": 1153
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1529411764705882e-05,
+      "loss": 0.0218,
+      "step": 1154
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.152173913043478e-05,
+      "loss": 0.0124,
+      "step": 1155
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1514066496163686e-05,
+      "loss": 0.0223,
+      "step": 1156
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1506393861892584e-05,
+      "loss": 0.0505,
+      "step": 1157
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1498721227621482e-05,
+      "loss": 0.0225,
+      "step": 1158
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1491048593350384e-05,
+      "loss": 0.0308,
+      "step": 1159
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1483375959079285e-05,
+      "loss": 0.036,
+      "step": 1160
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1475703324808184e-05,
+      "loss": 0.0121,
+      "step": 1161
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1468030690537085e-05,
+      "loss": 0.0274,
+      "step": 1162
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1460358056265983e-05,
+      "loss": 1.0397,
+      "step": 1163
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1452685421994885e-05,
+      "loss": 0.0332,
+      "step": 1164
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1445012787723787e-05,
+      "loss": 0.0182,
+      "step": 1165
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1437340153452685e-05,
+      "loss": 0.018,
+      "step": 1166
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1429667519181587e-05,
+      "loss": 0.859,
+      "step": 1167
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1421994884910488e-05,
+      "loss": 0.0321,
+      "step": 1168
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1414322250639386e-05,
+      "loss": 0.0225,
+      "step": 1169
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1406649616368285e-05,
+      "loss": 0.0224,
+      "step": 1170
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.139897698209719e-05,
+      "loss": 0.03,
+      "step": 1171
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1391304347826088e-05,
+      "loss": 0.7473,
+      "step": 1172
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1383631713554986e-05,
+      "loss": 0.8609,
+      "step": 1173
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1375959079283888e-05,
+      "loss": 0.0105,
+      "step": 1174
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.136828644501279e-05,
+      "loss": 0.0217,
+      "step": 1175
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1360613810741687e-05,
+      "loss": 0.0218,
+      "step": 1176
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.135294117647059e-05,
+      "loss": 0.016,
+      "step": 1177
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1345268542199487e-05,
+      "loss": 0.7844,
+      "step": 1178
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.133759590792839e-05,
+      "loss": 0.01,
+      "step": 1179
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132992327365729e-05,
+      "loss": 0.8774,
+      "step": 1180
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132225063938619e-05,
+      "loss": 0.8375,
+      "step": 1181
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.1314578005115087e-05,
+      "loss": 0.0188,
+      "step": 1182
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.1306905370843992e-05,
+      "loss": 1.1026,
+      "step": 1183
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129923273657289e-05,
+      "loss": 1.0447,
+      "step": 1184
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129156010230179e-05,
+      "loss": 0.0101,
+      "step": 1185
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.128388746803069e-05,
+      "loss": 0.0247,
+      "step": 1186
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.127621483375959e-05,
+      "loss": 0.0295,
+      "step": 1187
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.1268542199488493e-05,
+      "loss": 0.0191,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.37390685081481934,
+      "eval_runtime": 57.4087,
+      "eval_samples_per_second": 6.898,
+      "eval_steps_per_second": 1.724,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.126086956521739e-05,
+      "loss": 0.0103,
+      "step": 1189
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1253196930946293e-05,
+      "loss": 0.0231,
+      "step": 1190
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1245524296675195e-05,
+      "loss": 0.0389,
+      "step": 1191
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1237851662404093e-05,
+      "loss": 0.0298,
+      "step": 1192
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.123017902813299e-05,
+      "loss": 0.0196,
+      "step": 1193
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1222506393861896e-05,
+      "loss": 0.9688,
+      "step": 1194
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1214833759590794e-05,
+      "loss": 0.0458,
+      "step": 1195
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1207161125319693e-05,
+      "loss": 0.0161,
+      "step": 1196
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1199488491048594e-05,
+      "loss": 0.0157,
+      "step": 1197
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1191815856777496e-05,
+      "loss": 0.8732,
+      "step": 1198
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1184143222506394e-05,
+      "loss": 0.0103,
+      "step": 1199
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1176470588235296e-05,
+      "loss": 0.0267,
+      "step": 1200
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1168797953964194e-05,
+      "loss": 0.0266,
+      "step": 1201
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1161125319693095e-05,
+      "loss": 0.0157,
+      "step": 1202
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1153452685421997e-05,
+      "loss": 0.265,
+      "step": 1203
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1145780051150895e-05,
+      "loss": 0.0537,
+      "step": 1204
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1138107416879794e-05,
+      "loss": 0.0306,
+      "step": 1205
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.11304347826087e-05,
+      "loss": 1.1714,
+      "step": 1206
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1122762148337597e-05,
+      "loss": 1.1704,
+      "step": 1207
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1115089514066495e-05,
+      "loss": 0.0154,
+      "step": 1208
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1107416879795397e-05,
+      "loss": 0.0195,
+      "step": 1209
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1099744245524298e-05,
+      "loss": 0.9099,
+      "step": 1210
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1092071611253196e-05,
+      "loss": 0.0112,
+      "step": 1211
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1084398976982098e-05,
+      "loss": 0.1506,
+      "step": 1212
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1076726342710996e-05,
+      "loss": 0.0111,
+      "step": 1213
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1069053708439898e-05,
+      "loss": 0.0244,
+      "step": 1214
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10613810741688e-05,
+      "loss": 0.8889,
+      "step": 1215
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1053708439897698e-05,
+      "loss": 0.0146,
+      "step": 1216
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10460358056266e-05,
+      "loss": 0.0189,
+      "step": 1217
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10383631713555e-05,
+      "loss": 0.0097,
+      "step": 1218
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10306905370844e-05,
+      "loss": 1.1683,
+      "step": 1219
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1023017902813297e-05,
+      "loss": 1.1621,
+      "step": 1220
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1015345268542202e-05,
+      "loss": 0.0194,
+      "step": 1221
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.10076726342711e-05,
+      "loss": 0.0155,
+      "step": 1222
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.0296,
+      "step": 1223
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.09923273657289e-05,
+      "loss": 0.02,
+      "step": 1224
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.0984654731457802e-05,
+      "loss": 0.0154,
+      "step": 1225
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09769820971867e-05,
+      "loss": 0.0151,
+      "step": 1226
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0969309462915602e-05,
+      "loss": 0.0223,
+      "step": 1227
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09616368286445e-05,
+      "loss": 1.7159,
+      "step": 1228
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0953964194373402e-05,
+      "loss": 0.0194,
+      "step": 1229
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0946291560102303e-05,
+      "loss": 0.0637,
+      "step": 1230
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09386189258312e-05,
+      "loss": 0.0156,
+      "step": 1231
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09309462915601e-05,
+      "loss": 1.0612,
+      "step": 1232
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0923273657289005e-05,
+      "loss": 0.8929,
+      "step": 1233
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0915601023017903e-05,
+      "loss": 1.1649,
+      "step": 1234
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.09079283887468e-05,
+      "loss": 2.059,
+      "step": 1235
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0900255754475703e-05,
+      "loss": 0.0169,
+      "step": 1236
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0892583120204604e-05,
+      "loss": 0.0158,
+      "step": 1237
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0884910485933503e-05,
+      "loss": 0.0108,
+      "step": 1238
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0877237851662404e-05,
+      "loss": 0.0152,
+      "step": 1239
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0869565217391306e-05,
+      "loss": 0.0109,
+      "step": 1240
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0861892583120208e-05,
+      "loss": 1.1352,
+      "step": 1241
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0854219948849106e-05,
+      "loss": 0.0226,
+      "step": 1242
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0846547314578004e-05,
+      "loss": 0.0154,
+      "step": 1243
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.083887468030691e-05,
+      "loss": 0.0232,
+      "step": 1244
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0831202046035807e-05,
+      "loss": 0.0142,
+      "step": 1245
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0823529411764705e-05,
+      "loss": 0.9143,
+      "step": 1246
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0815856777493607e-05,
+      "loss": 0.0165,
+      "step": 1247
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.080818414322251e-05,
+      "loss": 0.0446,
+      "step": 1248
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0800511508951407e-05,
+      "loss": 0.0121,
+      "step": 1249
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.079283887468031e-05,
+      "loss": 1.0991,
+      "step": 1250
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.0785166240409207e-05,
+      "loss": 0.0159,
+      "step": 1251
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.077749360613811e-05,
+      "loss": 0.0166,
+      "step": 1252
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.076982097186701e-05,
+      "loss": 0.0171,
+      "step": 1253
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0762148337595908e-05,
+      "loss": 0.7986,
+      "step": 1254
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0754475703324806e-05,
+      "loss": 0.0174,
+      "step": 1255
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.074680306905371e-05,
+      "loss": 1.0431,
+      "step": 1256
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.073913043478261e-05,
+      "loss": 0.1766,
+      "step": 1257
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.0731457800511508e-05,
+      "loss": 0.0209,
+      "step": 1258
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.072378516624041e-05,
+      "loss": 0.0178,
+      "step": 1259
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.071611253196931e-05,
+      "loss": 0.0215,
+      "step": 1260
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.070843989769821e-05,
+      "loss": 0.0112,
+      "step": 1261
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.070076726342711e-05,
+      "loss": 0.0166,
+      "step": 1262
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0693094629156012e-05,
+      "loss": 0.0161,
+      "step": 1263
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.068542199488491e-05,
+      "loss": 0.8922,
+      "step": 1264
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0677749360613812e-05,
+      "loss": 0.0133,
+      "step": 1265
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.067007672634271e-05,
+      "loss": 0.0117,
+      "step": 1266
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0662404092071612e-05,
+      "loss": 0.0153,
+      "step": 1267
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0654731457800514e-05,
+      "loss": 0.0153,
+      "step": 1268
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0647058823529412e-05,
+      "loss": 0.0195,
+      "step": 1269
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.063938618925831e-05,
+      "loss": 1.1511,
+      "step": 1270
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0631713554987215e-05,
+      "loss": 0.0108,
+      "step": 1271
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0624040920716113e-05,
+      "loss": 0.8897,
+      "step": 1272
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.061636828644501e-05,
+      "loss": 0.0157,
+      "step": 1273
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0608695652173913e-05,
+      "loss": 0.0108,
+      "step": 1274
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0601023017902815e-05,
+      "loss": 0.0214,
+      "step": 1275
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0593350383631713e-05,
+      "loss": 0.0186,
+      "step": 1276
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0585677749360615e-05,
+      "loss": 0.0182,
+      "step": 1277
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0578005115089513e-05,
+      "loss": 0.0102,
+      "step": 1278
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0570332480818415e-05,
+      "loss": 1.1313,
+      "step": 1279
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0562659846547316e-05,
+      "loss": 0.0103,
+      "step": 1280
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0554987212276214e-05,
+      "loss": 0.0144,
+      "step": 1281
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0547314578005113e-05,
+      "loss": 0.02,
+      "step": 1282
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0539641943734018e-05,
+      "loss": 0.0292,
+      "step": 1283
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0531969309462916e-05,
+      "loss": 0.0105,
+      "step": 1284
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0524296675191814e-05,
+      "loss": 0.01,
+      "step": 1285
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0516624040920716e-05,
+      "loss": 1.1245,
+      "step": 1286
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0508951406649617e-05,
+      "loss": 0.8544,
+      "step": 1287
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0501278772378516e-05,
+      "loss": 0.0149,
+      "step": 1288
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0493606138107417e-05,
+      "loss": 0.0065,
+      "step": 1289
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.048593350383632e-05,
+      "loss": 0.0113,
+      "step": 1290
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0478260869565217e-05,
+      "loss": 0.0094,
+      "step": 1291
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.047058823529412e-05,
+      "loss": 0.0097,
+      "step": 1292
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0462915601023017e-05,
+      "loss": 0.0812,
+      "step": 1293
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0455242966751922e-05,
+      "loss": 0.0099,
+      "step": 1294
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.044757033248082e-05,
+      "loss": 0.013,
+      "step": 1295
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0439897698209718e-05,
+      "loss": 1.584,
+      "step": 1296
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.043222506393862e-05,
+      "loss": 0.0089,
+      "step": 1297
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.042455242966752e-05,
+      "loss": 0.0089,
+      "step": 1298
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.041687979539642e-05,
+      "loss": 0.0093,
+      "step": 1299
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.040920716112532e-05,
+      "loss": 0.0149,
+      "step": 1300
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.040153452685422e-05,
+      "loss": 1.1671,
+      "step": 1301
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.039386189258312e-05,
+      "loss": 0.0086,
+      "step": 1302
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.0386189258312023e-05,
+      "loss": 0.8177,
+      "step": 1303
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.037851662404092e-05,
+      "loss": 0.0128,
+      "step": 1304
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.037084398976982e-05,
+      "loss": 0.0167,
+      "step": 1305
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0363171355498724e-05,
+      "loss": 0.0085,
+      "step": 1306
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0355498721227622e-05,
+      "loss": 0.0085,
+      "step": 1307
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.034782608695652e-05,
+      "loss": 0.0166,
+      "step": 1308
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0340153452685422e-05,
+      "loss": 0.9448,
+      "step": 1309
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0332480818414324e-05,
+      "loss": 0.0082,
+      "step": 1310
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0324808184143222e-05,
+      "loss": 0.0088,
+      "step": 1311
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0317135549872124e-05,
+      "loss": 0.0082,
+      "step": 1312
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0309462915601025e-05,
+      "loss": 0.0146,
+      "step": 1313
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0301790281329924e-05,
+      "loss": 0.0123,
+      "step": 1314
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0294117647058825e-05,
+      "loss": 0.0101,
+      "step": 1315
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0286445012787723e-05,
+      "loss": 1.2167,
+      "step": 1316
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0278772378516625e-05,
+      "loss": 1.0578,
+      "step": 1317
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0271099744245527e-05,
+      "loss": 0.0163,
+      "step": 1318
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0263427109974425e-05,
+      "loss": 0.013,
+      "step": 1319
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0255754475703323e-05,
+      "loss": 0.0082,
+      "step": 1320
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0248081841432228e-05,
+      "loss": 0.0081,
+      "step": 1321
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0240409207161126e-05,
+      "loss": 0.0124,
+      "step": 1322
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0232736572890024e-05,
+      "loss": 0.0206,
+      "step": 1323
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0225063938618926e-05,
+      "loss": 0.0081,
+      "step": 1324
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0217391304347828e-05,
+      "loss": 0.0122,
+      "step": 1325
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0209718670076726e-05,
+      "loss": 0.0122,
+      "step": 1326
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0202046035805628e-05,
+      "loss": 0.0121,
+      "step": 1327
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0194373401534526e-05,
+      "loss": 1.224,
+      "step": 1328
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0186700767263427e-05,
+      "loss": 2.3502,
+      "step": 1329
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.017902813299233e-05,
+      "loss": 0.0242,
+      "step": 1330
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0171355498721227e-05,
+      "loss": 0.0064,
+      "step": 1331
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0163682864450125e-05,
+      "loss": 0.0186,
+      "step": 1332
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.015601023017903e-05,
+      "loss": 0.012,
+      "step": 1333
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.014833759590793e-05,
+      "loss": 0.0119,
+      "step": 1334
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.0140664961636827e-05,
+      "loss": 0.0159,
+      "step": 1335
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.013299232736573e-05,
+      "loss": 0.0199,
+      "step": 1336
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.012531969309463e-05,
+      "loss": 1.2138,
+      "step": 1337
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.011764705882353e-05,
+      "loss": 0.9491,
+      "step": 1338
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010997442455243e-05,
+      "loss": 0.0083,
+      "step": 1339
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010230179028133e-05,
+      "loss": 0.948,
+      "step": 1340
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.009462915601023e-05,
+      "loss": 0.0112,
+      "step": 1341
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.008695652173913e-05,
+      "loss": 0.8921,
+      "step": 1342
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007928388746803e-05,
+      "loss": 0.0194,
+      "step": 1343
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007161125319693e-05,
+      "loss": 1.0273,
+      "step": 1344
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0063938618925833e-05,
+      "loss": 0.013,
+      "step": 1345
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.005626598465473e-05,
+      "loss": 0.0103,
+      "step": 1346
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0048593350383633e-05,
+      "loss": 0.0114,
+      "step": 1347
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0040920716112534e-05,
+      "loss": 0.0199,
+      "step": 1348
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0033248081841433e-05,
+      "loss": 1.183,
+      "step": 1349
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0025575447570334e-05,
+      "loss": 0.0231,
+      "step": 1350
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0017902813299232e-05,
+      "loss": 1.1794,
+      "step": 1351
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0010230179028134e-05,
+      "loss": 0.0096,
+      "step": 1352
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 2.0002557544757036e-05,
+      "loss": 0.9049,
+      "step": 1353
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9994884910485934e-05,
+      "loss": 0.019,
+      "step": 1354
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9987212276214832e-05,
+      "loss": 0.0152,
+      "step": 1355
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9979539641943737e-05,
+      "loss": 0.8834,
+      "step": 1356
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9971867007672635e-05,
+      "loss": 0.01,
+      "step": 1357
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9964194373401533e-05,
+      "loss": 0.0152,
+      "step": 1358
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9956521739130435e-05,
+      "loss": 0.0161,
+      "step": 1359
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9948849104859337e-05,
+      "loss": 1.6778,
+      "step": 1360
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9941176470588235e-05,
+      "loss": 0.0214,
+      "step": 1361
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9933503836317137e-05,
+      "loss": 0.0156,
+      "step": 1362
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9925831202046038e-05,
+      "loss": 0.026,
+      "step": 1363
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9918158567774936e-05,
+      "loss": 0.0259,
+      "step": 1364
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9910485933503838e-05,
+      "loss": 0.0161,
+      "step": 1365
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9902813299232736e-05,
+      "loss": 0.0108,
+      "step": 1366
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9895140664961638e-05,
+      "loss": 0.0149,
+      "step": 1367
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.988746803069054e-05,
+      "loss": 0.0325,
+      "step": 1368
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9879795396419438e-05,
+      "loss": 0.0153,
+      "step": 1369
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9872122762148336e-05,
+      "loss": 1.7678,
+      "step": 1370
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.986445012787724e-05,
+      "loss": 0.0773,
+      "step": 1371
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.985677749360614e-05,
+      "loss": 0.0244,
+      "step": 1372
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.9849104859335037e-05,
+      "loss": 0.0225,
+      "step": 1373
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.984143222506394e-05,
+      "loss": 0.0254,
+      "step": 1374
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.983375959079284e-05,
+      "loss": 0.0205,
+      "step": 1375
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.982608695652174e-05,
+      "loss": 0.0096,
+      "step": 1376
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981841432225064e-05,
+      "loss": 0.0147,
+      "step": 1377
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981074168797954e-05,
+      "loss": 0.0296,
+      "step": 1378
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.980306905370844e-05,
+      "loss": 0.0227,
+      "step": 1379
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.9795396419437342e-05,
+      "loss": 0.899,
+      "step": 1380
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.978772378516624e-05,
+      "loss": 0.019,
+      "step": 1381
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9780051150895138e-05,
+      "loss": 0.0157,
+      "step": 1382
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9772378516624043e-05,
+      "loss": 1.1897,
+      "step": 1383
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.976470588235294e-05,
+      "loss": 0.014,
+      "step": 1384
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.975703324808184e-05,
+      "loss": 0.0208,
+      "step": 1385
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.974936061381074e-05,
+      "loss": 0.0093,
+      "step": 1386
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.9741687979539643e-05,
+      "loss": 0.0244,
+      "step": 1387
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.973401534526854e-05,
+      "loss": 0.028,
+      "step": 1388
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9726342710997443e-05,
+      "loss": 0.0093,
+      "step": 1389
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9718670076726344e-05,
+      "loss": 0.0184,
+      "step": 1390
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9710997442455243e-05,
+      "loss": 0.9462,
+      "step": 1391
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9703324808184144e-05,
+      "loss": 0.0138,
+      "step": 1392
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9695652173913042e-05,
+      "loss": 0.9561,
+      "step": 1393
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9687979539641944e-05,
+      "loss": 0.0152,
+      "step": 1394
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9680306905370846e-05,
+      "loss": 0.0178,
+      "step": 1395
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9672634271099744e-05,
+      "loss": 0.0166,
+      "step": 1396
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9664961636828642e-05,
+      "loss": 0.92,
+      "step": 1397
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9657289002557547e-05,
+      "loss": 0.0253,
+      "step": 1398
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9649616368286445e-05,
+      "loss": 0.0266,
+      "step": 1399
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9641943734015347e-05,
+      "loss": 0.0105,
+      "step": 1400
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9634271099744245e-05,
+      "loss": 0.0208,
+      "step": 1401
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9626598465473147e-05,
+      "loss": 0.8094,
+      "step": 1402
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.961892583120205e-05,
+      "loss": 0.0103,
+      "step": 1403
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9611253196930947e-05,
+      "loss": 0.0179,
+      "step": 1404
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9603580562659845e-05,
+      "loss": 0.0133,
+      "step": 1405
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.959590792838875e-05,
+      "loss": 0.0086,
+      "step": 1406
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9588235294117648e-05,
+      "loss": 0.0216,
+      "step": 1407
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9580562659846546e-05,
+      "loss": 0.0127,
+      "step": 1408
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9572890025575448e-05,
+      "loss": 0.0127,
+      "step": 1409
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.956521739130435e-05,
+      "loss": 0.0209,
+      "step": 1410
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9557544757033248e-05,
+      "loss": 0.0166,
+      "step": 1411
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954987212276215e-05,
+      "loss": 0.0123,
+      "step": 1412
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954219948849105e-05,
+      "loss": 0.0135,
+      "step": 1413
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.953452685421995e-05,
+      "loss": 0.1253,
+      "step": 1414
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.952685421994885e-05,
+      "loss": 0.0098,
+      "step": 1415
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951918158567775e-05,
+      "loss": 0.0091,
+      "step": 1416
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951150895140665e-05,
+      "loss": 0.9327,
+      "step": 1417
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.9503836317135552e-05,
+      "loss": 0.0119,
+      "step": 1418
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.949616368286445e-05,
+      "loss": 0.0153,
+      "step": 1419
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.948849104859335e-05,
+      "loss": 0.0123,
+      "step": 1420
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9480818414322254e-05,
+      "loss": 0.0194,
+      "step": 1421
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9473145780051152e-05,
+      "loss": 1.2311,
+      "step": 1422
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.946547314578005e-05,
+      "loss": 1.8519,
+      "step": 1423
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9457800511508952e-05,
+      "loss": 0.963,
+      "step": 1424
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9450127877237853e-05,
+      "loss": 0.0075,
+      "step": 1425
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.944245524296675e-05,
+      "loss": 0.0042,
+      "step": 1426
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9434782608695653e-05,
+      "loss": 0.0077,
+      "step": 1427
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.942710997442455e-05,
+      "loss": 0.0152,
+      "step": 1428
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9419437340153453e-05,
+      "loss": 0.0152,
+      "step": 1429
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9411764705882355e-05,
+      "loss": 0.019,
+      "step": 1430
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9404092071611253e-05,
+      "loss": 0.0081,
+      "step": 1431
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.939641943734015e-05,
+      "loss": 0.0076,
+      "step": 1432
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9388746803069056e-05,
+      "loss": 0.0086,
+      "step": 1433
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9381074168797954e-05,
+      "loss": 0.0146,
+      "step": 1434
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9373401534526853e-05,
+      "loss": 0.0074,
+      "step": 1435
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9365728900255754e-05,
+      "loss": 0.1007,
+      "step": 1436
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9358056265984656e-05,
+      "loss": 0.011,
+      "step": 1437
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9350383631713554e-05,
+      "loss": 0.0089,
+      "step": 1438
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9342710997442456e-05,
+      "loss": 0.0182,
+      "step": 1439
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9335038363171357e-05,
+      "loss": 0.0139,
+      "step": 1440
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9327365728900255e-05,
+      "loss": 2.2086,
+      "step": 1441
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9319693094629157e-05,
+      "loss": 0.0178,
+      "step": 1442
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9312020460358055e-05,
+      "loss": 0.0142,
+      "step": 1443
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9304347826086957e-05,
+      "loss": 0.0142,
+      "step": 1444
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.929667519181586e-05,
+      "loss": 0.0242,
+      "step": 1445
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9289002557544757e-05,
+      "loss": 0.0087,
+      "step": 1446
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9281329923273655e-05,
+      "loss": 0.0077,
+      "step": 1447
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.927365728900256e-05,
+      "loss": 0.0104,
+      "step": 1448
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9265984654731458e-05,
+      "loss": 0.0139,
+      "step": 1449
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9258312020460356e-05,
+      "loss": 1.2557,
+      "step": 1450
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9250639386189258e-05,
+      "loss": 0.0068,
+      "step": 1451
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.924296675191816e-05,
+      "loss": 0.0163,
+      "step": 1452
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.923529411764706e-05,
+      "loss": 0.9483,
+      "step": 1453
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.922762148337596e-05,
+      "loss": 1.2181,
+      "step": 1454
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.9219948849104858e-05,
+      "loss": 0.011,
+      "step": 1455
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.9212276214833763e-05,
+      "loss": 0.014,
+      "step": 1456
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.920460358056266e-05,
+      "loss": 0.0131,
+      "step": 1457
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.919693094629156e-05,
+      "loss": 0.0135,
+      "step": 1458
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.918925831202046e-05,
+      "loss": 0.007,
+      "step": 1459
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9181585677749362e-05,
+      "loss": 0.0072,
+      "step": 1460
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.917391304347826e-05,
+      "loss": 0.0135,
+      "step": 1461
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9166240409207162e-05,
+      "loss": 0.9908,
+      "step": 1462
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9158567774936064e-05,
+      "loss": 0.0103,
+      "step": 1463
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9150895140664962e-05,
+      "loss": 0.9867,
+      "step": 1464
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9143222506393864e-05,
+      "loss": 0.0138,
+      "step": 1465
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9135549872122762e-05,
+      "loss": 0.0104,
+      "step": 1466
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9127877237851663e-05,
+      "loss": 0.014,
+      "step": 1467
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9120204603580565e-05,
+      "loss": 0.0071,
+      "step": 1468
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9112531969309463e-05,
+      "loss": 0.0108,
+      "step": 1469
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.910485933503836e-05,
+      "loss": 0.0106,
+      "step": 1470
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9097186700767266e-05,
+      "loss": 0.0404,
+      "step": 1471
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9089514066496165e-05,
+      "loss": 2.2011,
+      "step": 1472
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9081841432225063e-05,
+      "loss": 0.0087,
+      "step": 1473
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9074168797953965e-05,
+      "loss": 0.0071,
+      "step": 1474
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9066496163682866e-05,
+      "loss": 0.0072,
+      "step": 1475
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9058823529411764e-05,
+      "loss": 0.0108,
+      "step": 1476
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9051150895140666e-05,
+      "loss": 0.0108,
+      "step": 1477
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9043478260869564e-05,
+      "loss": 0.0072,
+      "step": 1478
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9035805626598466e-05,
+      "loss": 0.0108,
+      "step": 1479
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9028132992327367e-05,
+      "loss": 0.0073,
+      "step": 1480
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9020460358056266e-05,
+      "loss": 0.9537,
+      "step": 1481
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9012787723785164e-05,
+      "loss": 0.0107,
+      "step": 1482
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.900511508951407e-05,
+      "loss": 0.0145,
+      "step": 1483
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8997442455242967e-05,
+      "loss": 0.007,
+      "step": 1484
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8989769820971865e-05,
+      "loss": 0.0071,
+      "step": 1485
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.898209718670077e-05,
+      "loss": 1.243,
+      "step": 1486
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.897442455242967e-05,
+      "loss": 0.011,
+      "step": 1487
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.8966751918158567e-05,
+      "loss": 0.0077,
+      "step": 1488
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895907928388747e-05,
+      "loss": 0.0073,
+      "step": 1489
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895140664961637e-05,
+      "loss": 1.2486,
+      "step": 1490
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8943734015345268e-05,
+      "loss": 0.0109,
+      "step": 1491
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.893606138107417e-05,
+      "loss": 0.0149,
+      "step": 1492
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8928388746803068e-05,
+      "loss": 0.0125,
+      "step": 1493
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.892071611253197e-05,
+      "loss": 0.0071,
+      "step": 1494
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.891304347826087e-05,
+      "loss": 0.0036,
+      "step": 1495
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.890537084398977e-05,
+      "loss": 0.0149,
+      "step": 1496
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8897698209718668e-05,
+      "loss": 0.015,
+      "step": 1497
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8890025575447573e-05,
+      "loss": 0.9536,
+      "step": 1498
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.888235294117647e-05,
+      "loss": 0.9997,
+      "step": 1499
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.887468030690537e-05,
+      "loss": 0.9621,
+      "step": 1500
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.886700767263427e-05,
+      "loss": 0.0113,
+      "step": 1501
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.8859335038363172e-05,
+      "loss": 0.2056,
+      "step": 1502
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.885166240409207e-05,
+      "loss": 0.0114,
+      "step": 1503
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8843989769820972e-05,
+      "loss": 0.016,
+      "step": 1504
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.883631713554987e-05,
+      "loss": 0.9175,
+      "step": 1505
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8828644501278775e-05,
+      "loss": 0.8959,
+      "step": 1506
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8820971867007674e-05,
+      "loss": 0.0074,
+      "step": 1507
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8813299232736572e-05,
+      "loss": 0.0117,
+      "step": 1508
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8805626598465474e-05,
+      "loss": 0.0075,
+      "step": 1509
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8797953964194375e-05,
+      "loss": 0.0074,
+      "step": 1510
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8790281329923273e-05,
+      "loss": 0.0577,
+      "step": 1511
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8782608695652175e-05,
+      "loss": 0.0128,
+      "step": 1512
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8774936061381077e-05,
+      "loss": 0.0073,
+      "step": 1513
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8767263427109975e-05,
+      "loss": 0.0074,
+      "step": 1514
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8759590792838876e-05,
+      "loss": 0.0073,
+      "step": 1515
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8751918158567775e-05,
+      "loss": 0.0088,
+      "step": 1516
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8744245524296676e-05,
+      "loss": 0.0078,
+      "step": 1517
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8736572890025578e-05,
+      "loss": 0.016,
+      "step": 1518
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8728900255754476e-05,
+      "loss": 1.3842,
+      "step": 1519
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8721227621483374e-05,
+      "loss": 1.2502,
+      "step": 1520
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.871355498721228e-05,
+      "loss": 0.0117,
+      "step": 1521
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8705882352941178e-05,
+      "loss": 0.0113,
+      "step": 1522
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8698209718670076e-05,
+      "loss": 0.0125,
+      "step": 1523
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8690537084398977e-05,
+      "loss": 0.0153,
+      "step": 1524
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.868286445012788e-05,
+      "loss": 0.0072,
+      "step": 1525
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8675191815856777e-05,
+      "loss": 0.0086,
+      "step": 1526
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.866751918158568e-05,
+      "loss": 0.0195,
+      "step": 1527
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.8659846547314577e-05,
+      "loss": 1.2511,
+      "step": 1528
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.865217391304348e-05,
+      "loss": 1.2407,
+      "step": 1529
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.864450127877238e-05,
+      "loss": 0.0112,
+      "step": 1530
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.863682864450128e-05,
+      "loss": 0.0073,
+      "step": 1531
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8629156010230177e-05,
+      "loss": 1.1911,
+      "step": 1532
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8621483375959082e-05,
+      "loss": 0.0168,
+      "step": 1533
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.861381074168798e-05,
+      "loss": 0.0128,
+      "step": 1534
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8606138107416878e-05,
+      "loss": 0.0116,
+      "step": 1535
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8598465473145783e-05,
+      "loss": 1.2286,
+      "step": 1536
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.859079283887468e-05,
+      "loss": 0.0158,
+      "step": 1537
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.858312020460358e-05,
+      "loss": 2.3735,
+      "step": 1538
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.857544757033248e-05,
+      "loss": 1.1972,
+      "step": 1539
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8567774936061383e-05,
+      "loss": 0.011,
+      "step": 1540
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.856010230179028e-05,
+      "loss": 0.0127,
+      "step": 1541
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8552429667519183e-05,
+      "loss": 0.0148,
+      "step": 1542
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.854475703324808e-05,
+      "loss": 0.0111,
+      "step": 1543
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8537084398976983e-05,
+      "loss": 0.0143,
+      "step": 1544
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8529411764705884e-05,
+      "loss": 0.0132,
+      "step": 1545
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8521739130434782e-05,
+      "loss": 0.0152,
+      "step": 1546
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.851406649616368e-05,
+      "loss": 0.0152,
+      "step": 1547
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8506393861892586e-05,
+      "loss": 0.0051,
+      "step": 1548
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8498721227621484e-05,
+      "loss": 0.9576,
+      "step": 1549
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8491048593350382e-05,
+      "loss": 0.0181,
+      "step": 1550
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8483375959079284e-05,
+      "loss": 0.011,
+      "step": 1551
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8475703324808185e-05,
+      "loss": 0.0205,
+      "step": 1552
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8468030690537083e-05,
+      "loss": 2.5631,
+      "step": 1553
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8460358056265985e-05,
+      "loss": 0.0102,
+      "step": 1554
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8452685421994883e-05,
+      "loss": 0.0147,
+      "step": 1555
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8445012787723788e-05,
+      "loss": 0.015,
+      "step": 1556
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8437340153452687e-05,
+      "loss": 0.0122,
+      "step": 1557
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8429667519181585e-05,
+      "loss": 0.0176,
+      "step": 1558
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8421994884910486e-05,
+      "loss": 0.0114,
+      "step": 1559
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8414322250639388e-05,
+      "loss": 0.0252,
+      "step": 1560
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8406649616368286e-05,
+      "loss": 0.013,
+      "step": 1561
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8398976982097188e-05,
+      "loss": 0.0142,
+      "step": 1562
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.839130434782609e-05,
+      "loss": 0.6496,
+      "step": 1563
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8383631713554988e-05,
+      "loss": 0.0149,
+      "step": 1564
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.837595907928389e-05,
+      "loss": 0.0104,
+      "step": 1565
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8368286445012787e-05,
+      "loss": 0.0129,
+      "step": 1566
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.836061381074169e-05,
+      "loss": 0.0197,
+      "step": 1567
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.835294117647059e-05,
+      "loss": 0.009,
+      "step": 1568
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.834526854219949e-05,
+      "loss": 0.0086,
+      "step": 1569
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.8337595907928387e-05,
+      "loss": 0.0087,
+      "step": 1570
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.8329923273657292e-05,
+      "loss": 0.0131,
+      "step": 1571
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.832225063938619e-05,
+      "loss": 1.2046,
+      "step": 1572
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.831457800511509e-05,
+      "loss": 0.0135,
+      "step": 1573
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.830690537084399e-05,
+      "loss": 0.0149,
+      "step": 1574
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.8299232736572892e-05,
+      "loss": 0.0077,
+      "step": 1575
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.829156010230179e-05,
+      "loss": 0.0143,
+      "step": 1576
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.828388746803069e-05,
+      "loss": 0.9806,
+      "step": 1577
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.827621483375959e-05,
+      "loss": 1.3783,
+      "step": 1578
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.826854219948849e-05,
+      "loss": 0.0075,
+      "step": 1579
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.8260869565217393e-05,
+      "loss": 0.0111,
+      "step": 1580
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.825319693094629e-05,
+      "loss": 0.0111,
+      "step": 1581
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.824552429667519e-05,
+      "loss": 0.0144,
+      "step": 1582
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8237851662404095e-05,
+      "loss": 0.0128,
+      "step": 1583
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8230179028132993e-05,
+      "loss": 1.2949,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.40147146582603455,
+      "eval_runtime": 58.0487,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.822250639386189e-05,
+      "loss": 0.0075,
+      "step": 1585
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8214833759590796e-05,
+      "loss": 0.0108,
+      "step": 1586
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8207161125319694e-05,
+      "loss": 0.011,
+      "step": 1587
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8199488491048592e-05,
+      "loss": 0.0148,
+      "step": 1588
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8191815856777494e-05,
+      "loss": 0.0077,
+      "step": 1589
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8184143222506396e-05,
+      "loss": 0.6674,
+      "step": 1590
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8176470588235294e-05,
+      "loss": 0.0109,
+      "step": 1591
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8168797953964195e-05,
+      "loss": 0.0108,
+      "step": 1592
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8161125319693094e-05,
+      "loss": 0.011,
+      "step": 1593
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8153452685421995e-05,
+      "loss": 0.0143,
+      "step": 1594
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8145780051150897e-05,
+      "loss": 0.9728,
+      "step": 1595
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8138107416879795e-05,
+      "loss": 0.0074,
+      "step": 1596
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8130434782608693e-05,
+      "loss": 0.0087,
+      "step": 1597
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.81227621483376e-05,
+      "loss": 0.0107,
+      "step": 1598
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8115089514066497e-05,
+      "loss": 0.0077,
+      "step": 1599
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8107416879795395e-05,
+      "loss": 0.0167,
+      "step": 1600
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8099744245524296e-05,
+      "loss": 1.219,
+      "step": 1601
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8092071611253198e-05,
+      "loss": 1.2244,
+      "step": 1602
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8084398976982096e-05,
+      "loss": 0.0108,
+      "step": 1603
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8076726342710998e-05,
+      "loss": 0.0071,
+      "step": 1604
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8069053708439896e-05,
+      "loss": 0.0109,
+      "step": 1605
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.8061381074168798e-05,
+      "loss": 0.0185,
+      "step": 1606
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.80537084398977e-05,
+      "loss": 1.2711,
+      "step": 1607
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.8046035805626598e-05,
+      "loss": 0.0113,
+      "step": 1608
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.80383631713555e-05,
+      "loss": 0.9575,
+      "step": 1609
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80306905370844e-05,
+      "loss": 1.1369,
+      "step": 1610
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80230179028133e-05,
+      "loss": 0.0114,
+      "step": 1611
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80153452685422e-05,
+      "loss": 1.578,
+      "step": 1612
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.8007672634271102e-05,
+      "loss": 0.0076,
+      "step": 1613
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.0128,
+      "step": 1614
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.7992327365728902e-05,
+      "loss": 0.0157,
+      "step": 1615
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.79846547314578e-05,
+      "loss": 0.0154,
+      "step": 1616
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.7976982097186702e-05,
+      "loss": 0.0125,
+      "step": 1617
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7969309462915604e-05,
+      "loss": 0.0085,
+      "step": 1618
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7961636828644502e-05,
+      "loss": 0.0167,
+      "step": 1619
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.79539641943734e-05,
+      "loss": 1.2275,
+      "step": 1620
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7946291560102305e-05,
+      "loss": 0.0115,
+      "step": 1621
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7938618925831203e-05,
+      "loss": 0.0083,
+      "step": 1622
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.79309462915601e-05,
+      "loss": 0.015,
+      "step": 1623
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7923273657289003e-05,
+      "loss": 1.1705,
+      "step": 1624
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7915601023017905e-05,
+      "loss": 0.0152,
+      "step": 1625
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7907928388746803e-05,
+      "loss": 2.0998,
+      "step": 1626
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7900255754475704e-05,
+      "loss": 0.0072,
+      "step": 1627
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7892583120204603e-05,
+      "loss": 0.0089,
+      "step": 1628
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7884910485933504e-05,
+      "loss": 0.0077,
+      "step": 1629
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7877237851662406e-05,
+      "loss": 0.0126,
+      "step": 1630
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7869565217391304e-05,
+      "loss": 1.2113,
+      "step": 1631
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7861892583120202e-05,
+      "loss": 0.0112,
+      "step": 1632
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7854219948849107e-05,
+      "loss": 0.0099,
+      "step": 1633
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7846547314578006e-05,
+      "loss": 0.0074,
+      "step": 1634
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7838874680306904e-05,
+      "loss": 0.0111,
+      "step": 1635
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.783120204603581e-05,
+      "loss": 0.0075,
+      "step": 1636
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7823529411764707e-05,
+      "loss": 1.2187,
+      "step": 1637
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7815856777493605e-05,
+      "loss": 0.0111,
+      "step": 1638
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7808184143222507e-05,
+      "loss": 0.0184,
+      "step": 1639
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.780051150895141e-05,
+      "loss": 0.0111,
+      "step": 1640
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7792838874680307e-05,
+      "loss": 0.0112,
+      "step": 1641
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.778516624040921e-05,
+      "loss": 0.0149,
+      "step": 1642
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.7777493606138107e-05,
+      "loss": 0.0151,
+      "step": 1643
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.7769820971867008e-05,
+      "loss": 0.0074,
+      "step": 1644
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.776214833759591e-05,
+      "loss": 0.9703,
+      "step": 1645
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.7754475703324808e-05,
+      "loss": 1.9364,
+      "step": 1646
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.7746803069053706e-05,
+      "loss": 0.011,
+      "step": 1647
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.773913043478261e-05,
+      "loss": 0.01,
+      "step": 1648
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.773145780051151e-05,
+      "loss": 0.0111,
+      "step": 1649
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.7723785166240408e-05,
+      "loss": 1.2373,
+      "step": 1650
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.771611253196931e-05,
+      "loss": 0.0074,
+      "step": 1651
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.770843989769821e-05,
+      "loss": 0.0116,
+      "step": 1652
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.770076726342711e-05,
+      "loss": 0.0117,
+      "step": 1653
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.769309462915601e-05,
+      "loss": 1.1636,
+      "step": 1654
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.768542199488491e-05,
+      "loss": 0.0075,
+      "step": 1655
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.767774936061381e-05,
+      "loss": 0.007,
+      "step": 1656
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.7670076726342712e-05,
+      "loss": 0.0174,
+      "step": 1657
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.766240409207161e-05,
+      "loss": 0.0113,
+      "step": 1658
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7654731457800512e-05,
+      "loss": 0.0084,
+      "step": 1659
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7647058823529414e-05,
+      "loss": 0.2994,
+      "step": 1660
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7639386189258312e-05,
+      "loss": 0.0213,
+      "step": 1661
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7631713554987213e-05,
+      "loss": 0.011,
+      "step": 1662
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7624040920716115e-05,
+      "loss": 1.2341,
+      "step": 1663
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7616368286445013e-05,
+      "loss": 1.1774,
+      "step": 1664
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7608695652173915e-05,
+      "loss": 0.0181,
+      "step": 1665
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7601023017902813e-05,
+      "loss": 0.0122,
+      "step": 1666
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7593350383631715e-05,
+      "loss": 0.0119,
+      "step": 1667
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7585677749360616e-05,
+      "loss": 0.0114,
+      "step": 1668
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7578005115089515e-05,
+      "loss": 0.0111,
+      "step": 1669
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7570332480818413e-05,
+      "loss": 0.011,
+      "step": 1670
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7562659846547318e-05,
+      "loss": 0.0113,
+      "step": 1671
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7554987212276216e-05,
+      "loss": 0.0161,
+      "step": 1672
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7547314578005114e-05,
+      "loss": 0.9491,
+      "step": 1673
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7539641943734016e-05,
+      "loss": 0.0108,
+      "step": 1674
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7531969309462917e-05,
+      "loss": 0.0114,
+      "step": 1675
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7524296675191816e-05,
+      "loss": 0.0107,
+      "step": 1676
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7516624040920717e-05,
+      "loss": 0.0171,
+      "step": 1677
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7508951406649616e-05,
+      "loss": 0.017,
+      "step": 1678
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7501278772378517e-05,
+      "loss": 1.939,
+      "step": 1679
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.749360613810742e-05,
+      "loss": 0.0151,
+      "step": 1680
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7485933503836317e-05,
+      "loss": 0.0109,
+      "step": 1681
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.7478260869565215e-05,
+      "loss": 0.0119,
+      "step": 1682
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.747058823529412e-05,
+      "loss": 1.2127,
+      "step": 1683
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.746291560102302e-05,
+      "loss": 0.0075,
+      "step": 1684
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.7455242966751917e-05,
+      "loss": 1.2283,
+      "step": 1685
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.744757033248082e-05,
+      "loss": 0.0107,
+      "step": 1686
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.743989769820972e-05,
+      "loss": 0.0141,
+      "step": 1687
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.7432225063938618e-05,
+      "loss": 1.2117,
+      "step": 1688
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.742455242966752e-05,
+      "loss": 0.8715,
+      "step": 1689
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.741687979539642e-05,
+      "loss": 0.0075,
+      "step": 1690
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.740920716112532e-05,
+      "loss": 1.2208,
+      "step": 1691
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.740153452685422e-05,
+      "loss": 0.0077,
+      "step": 1692
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.739386189258312e-05,
+      "loss": 0.0081,
+      "step": 1693
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.738618925831202e-05,
+      "loss": 0.0104,
+      "step": 1694
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.7378516624040923e-05,
+      "loss": 0.0162,
+      "step": 1695
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.737084398976982e-05,
+      "loss": 0.0173,
+      "step": 1696
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.736317135549872e-05,
+      "loss": 0.0194,
+      "step": 1697
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.7355498721227624e-05,
+      "loss": 0.0174,
+      "step": 1698
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.7347826086956522e-05,
+      "loss": 0.0081,
+      "step": 1699
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.734015345268542e-05,
+      "loss": 0.0121,
+      "step": 1700
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7332480818414322e-05,
+      "loss": 0.0137,
+      "step": 1701
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7324808184143224e-05,
+      "loss": 0.0117,
+      "step": 1702
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7317135549872122e-05,
+      "loss": 0.0133,
+      "step": 1703
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7309462915601024e-05,
+      "loss": 0.0175,
+      "step": 1704
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7301790281329922e-05,
+      "loss": 0.0076,
+      "step": 1705
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7294117647058823e-05,
+      "loss": 0.0179,
+      "step": 1706
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7286445012787725e-05,
+      "loss": 1.0541,
+      "step": 1707
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7278772378516623e-05,
+      "loss": 0.0077,
+      "step": 1708
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7271099744245525e-05,
+      "loss": 0.0079,
+      "step": 1709
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7263427109974426e-05,
+      "loss": 0.0102,
+      "step": 1710
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7255754475703325e-05,
+      "loss": 0.0158,
+      "step": 1711
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7248081841432223e-05,
+      "loss": 0.0107,
+      "step": 1712
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7240409207161128e-05,
+      "loss": 0.0159,
+      "step": 1713
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7232736572890026e-05,
+      "loss": 0.0125,
+      "step": 1714
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7225063938618928e-05,
+      "loss": 0.0081,
+      "step": 1715
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7217391304347826e-05,
+      "loss": 1.1946,
+      "step": 1716
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7209718670076728e-05,
+      "loss": 1.0298,
+      "step": 1717
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.720204603580563e-05,
+      "loss": 1.0207,
+      "step": 1718
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7194373401534527e-05,
+      "loss": 0.0152,
+      "step": 1719
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7186700767263426e-05,
+      "loss": 0.0187,
+      "step": 1720
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.717902813299233e-05,
+      "loss": 0.0134,
+      "step": 1721
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.717135549872123e-05,
+      "loss": 0.0173,
+      "step": 1722
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.7163682864450127e-05,
+      "loss": 1.0206,
+      "step": 1723
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.715601023017903e-05,
+      "loss": 0.0234,
+      "step": 1724
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.714833759590793e-05,
+      "loss": 0.0151,
+      "step": 1725
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.714066496163683e-05,
+      "loss": 0.0074,
+      "step": 1726
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.713299232736573e-05,
+      "loss": 0.0127,
+      "step": 1727
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.712531969309463e-05,
+      "loss": 0.8679,
+      "step": 1728
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.711764705882353e-05,
+      "loss": 0.0498,
+      "step": 1729
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.710997442455243e-05,
+      "loss": 0.0136,
+      "step": 1730
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.710230179028133e-05,
+      "loss": 1.2168,
+      "step": 1731
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.7094629156010228e-05,
+      "loss": 0.0105,
+      "step": 1732
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.7086956521739133e-05,
+      "loss": 0.0107,
+      "step": 1733
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.707928388746803e-05,
+      "loss": 0.0122,
+      "step": 1734
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.707161125319693e-05,
+      "loss": 0.0075,
+      "step": 1735
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.7063938618925834e-05,
+      "loss": 0.0607,
+      "step": 1736
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7056265984654733e-05,
+      "loss": 1.0181,
+      "step": 1737
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.704859335038363e-05,
+      "loss": 0.0095,
+      "step": 1738
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7040920716112533e-05,
+      "loss": 0.0128,
+      "step": 1739
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7033248081841434e-05,
+      "loss": 0.0096,
+      "step": 1740
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7025575447570332e-05,
+      "loss": 1.1781,
+      "step": 1741
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7017902813299234e-05,
+      "loss": 1.0132,
+      "step": 1742
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7010230179028132e-05,
+      "loss": 0.0097,
+      "step": 1743
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7002557544757034e-05,
+      "loss": 0.012,
+      "step": 1744
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6994884910485935e-05,
+      "loss": 0.0128,
+      "step": 1745
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6987212276214834e-05,
+      "loss": 0.0145,
+      "step": 1746
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6979539641943732e-05,
+      "loss": 0.7558,
+      "step": 1747
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6971867007672637e-05,
+      "loss": 0.0107,
+      "step": 1748
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6964194373401535e-05,
+      "loss": 1.0181,
+      "step": 1749
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6956521739130433e-05,
+      "loss": 1.7925,
+      "step": 1750
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6948849104859335e-05,
+      "loss": 0.0141,
+      "step": 1751
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6941176470588237e-05,
+      "loss": 0.0167,
+      "step": 1752
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6933503836317135e-05,
+      "loss": 0.0141,
+      "step": 1753
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6925831202046036e-05,
+      "loss": 0.0149,
+      "step": 1754
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6918158567774935e-05,
+      "loss": 0.0115,
+      "step": 1755
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6910485933503836e-05,
+      "loss": 0.0071,
+      "step": 1756
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6902813299232738e-05,
+      "loss": 0.0155,
+      "step": 1757
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6895140664961636e-05,
+      "loss": 0.0199,
+      "step": 1758
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6887468030690538e-05,
+      "loss": 0.0124,
+      "step": 1759
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.687979539641944e-05,
+      "loss": 0.0086,
+      "step": 1760
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.6872122762148337e-05,
+      "loss": 0.0147,
+      "step": 1761
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.6864450127877236e-05,
+      "loss": 0.0107,
+      "step": 1762
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.685677749360614e-05,
+      "loss": 1.025,
+      "step": 1763
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.684910485933504e-05,
+      "loss": 0.0105,
+      "step": 1764
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.6841432225063937e-05,
+      "loss": 0.0081,
+      "step": 1765
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.683375959079284e-05,
+      "loss": 0.009,
+      "step": 1766
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.682608695652174e-05,
+      "loss": 0.0109,
+      "step": 1767
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.6818414322250642e-05,
+      "loss": 0.008,
+      "step": 1768
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.681074168797954e-05,
+      "loss": 0.0132,
+      "step": 1769
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.680306905370844e-05,
+      "loss": 0.0119,
+      "step": 1770
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.6795396419437343e-05,
+      "loss": 0.0093,
+      "step": 1771
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.678772378516624e-05,
+      "loss": 1.0247,
+      "step": 1772
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.678005115089514e-05,
+      "loss": 1.024,
+      "step": 1773
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.677237851662404e-05,
+      "loss": 0.0106,
+      "step": 1774
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.6764705882352943e-05,
+      "loss": 0.0097,
+      "step": 1775
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.675703324808184e-05,
+      "loss": 0.0098,
+      "step": 1776
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6749360613810743e-05,
+      "loss": 1.0136,
+      "step": 1777
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.674168797953964e-05,
+      "loss": 0.0066,
+      "step": 1778
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6734015345268543e-05,
+      "loss": 0.1284,
+      "step": 1779
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6726342710997444e-05,
+      "loss": 0.9832,
+      "step": 1780
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6718670076726343e-05,
+      "loss": 0.0128,
+      "step": 1781
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6710997442455244e-05,
+      "loss": 0.0103,
+      "step": 1782
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6703324808184146e-05,
+      "loss": 0.0067,
+      "step": 1783
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6695652173913044e-05,
+      "loss": 0.0105,
+      "step": 1784
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6687979539641942e-05,
+      "loss": 0.0146,
+      "step": 1785
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6680306905370847e-05,
+      "loss": 0.0068,
+      "step": 1786
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6672634271099745e-05,
+      "loss": 0.0148,
+      "step": 1787
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6664961636828644e-05,
+      "loss": 0.0107,
+      "step": 1788
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6657289002557545e-05,
+      "loss": 2.4738,
+      "step": 1789
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6649616368286447e-05,
+      "loss": 0.9625,
+      "step": 1790
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6641943734015345e-05,
+      "loss": 0.2578,
+      "step": 1791
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6634271099744247e-05,
+      "loss": 0.0069,
+      "step": 1792
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6626598465473145e-05,
+      "loss": 0.011,
+      "step": 1793
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6618925831202047e-05,
+      "loss": 0.0066,
+      "step": 1794
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6611253196930948e-05,
+      "loss": 0.0069,
+      "step": 1795
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6603580562659846e-05,
+      "loss": 0.0071,
+      "step": 1796
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6595907928388745e-05,
+      "loss": 0.0068,
+      "step": 1797
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.658823529411765e-05,
+      "loss": 0.0232,
+      "step": 1798
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6580562659846548e-05,
+      "loss": 1.1318,
+      "step": 1799
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6572890025575446e-05,
+      "loss": 0.0101,
+      "step": 1800
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6565217391304348e-05,
+      "loss": 0.0187,
+      "step": 1801
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.655754475703325e-05,
+      "loss": 1.0718,
+      "step": 1802
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6549872122762148e-05,
+      "loss": 1.2349,
+      "step": 1803
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.654219948849105e-05,
+      "loss": 0.0148,
+      "step": 1804
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.6534526854219947e-05,
+      "loss": 0.0148,
+      "step": 1805
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.652685421994885e-05,
+      "loss": 0.0067,
+      "step": 1806
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.651918158567775e-05,
+      "loss": 0.0075,
+      "step": 1807
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.651150895140665e-05,
+      "loss": 0.0157,
+      "step": 1808
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.650383631713555e-05,
+      "loss": 0.0084,
+      "step": 1809
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.6496163682864452e-05,
+      "loss": 0.0123,
+      "step": 1810
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.648849104859335e-05,
+      "loss": 0.0065,
+      "step": 1811
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.648081841432225e-05,
+      "loss": 0.014,
+      "step": 1812
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.6473145780051154e-05,
+      "loss": 0.0152,
+      "step": 1813
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.6465473145780052e-05,
+      "loss": 0.0168,
+      "step": 1814
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.645780051150895e-05,
+      "loss": 0.0063,
+      "step": 1815
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.645012787723785e-05,
+      "loss": 0.6994,
+      "step": 1816
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.6442455242966753e-05,
+      "loss": 0.0132,
+      "step": 1817
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.643478260869565e-05,
+      "loss": 0.0103,
+      "step": 1818
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.6427109974424553e-05,
+      "loss": 0.013,
+      "step": 1819
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.641943734015345e-05,
+      "loss": 1.1813,
+      "step": 1820
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6411764705882356e-05,
+      "loss": 0.9809,
+      "step": 1821
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6404092071611254e-05,
+      "loss": 1.2339,
+      "step": 1822
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6396419437340153e-05,
+      "loss": 0.9744,
+      "step": 1823
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6388746803069054e-05,
+      "loss": 0.0098,
+      "step": 1824
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6381074168797956e-05,
+      "loss": 0.0062,
+      "step": 1825
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6373401534526854e-05,
+      "loss": 0.0151,
+      "step": 1826
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6365728900255756e-05,
+      "loss": 0.013,
+      "step": 1827
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6358056265984654e-05,
+      "loss": 0.0071,
+      "step": 1828
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6350383631713556e-05,
+      "loss": 0.0117,
+      "step": 1829
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6342710997442457e-05,
+      "loss": 0.0165,
+      "step": 1830
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6335038363171355e-05,
+      "loss": 0.1519,
+      "step": 1831
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6327365728900257e-05,
+      "loss": 1.0247,
+      "step": 1832
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.631969309462916e-05,
+      "loss": 0.0092,
+      "step": 1833
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6312020460358057e-05,
+      "loss": 0.0098,
+      "step": 1834
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6304347826086955e-05,
+      "loss": 1.1829,
+      "step": 1835
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.629667519181586e-05,
+      "loss": 0.006,
+      "step": 1836
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.628900255754476e-05,
+      "loss": 0.0109,
+      "step": 1837
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.6281329923273657e-05,
+      "loss": 0.6059,
+      "step": 1838
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.6273657289002558e-05,
+      "loss": 0.01,
+      "step": 1839
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.626598465473146e-05,
+      "loss": 0.0149,
+      "step": 1840
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.6258312020460358e-05,
+      "loss": 0.0152,
+      "step": 1841
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.625063938618926e-05,
+      "loss": 0.0171,
+      "step": 1842
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.6242966751918158e-05,
+      "loss": 0.0105,
+      "step": 1843
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.623529411764706e-05,
+      "loss": 0.0093,
+      "step": 1844
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.622762148337596e-05,
+      "loss": 1.4125,
+      "step": 1845
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.621994884910486e-05,
+      "loss": 0.0107,
+      "step": 1846
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.6212276214833757e-05,
+      "loss": 0.0133,
+      "step": 1847
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.6204603580562662e-05,
+      "loss": 0.0063,
+      "step": 1848
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.619693094629156e-05,
+      "loss": 0.0063,
+      "step": 1849
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.618925831202046e-05,
+      "loss": 0.009,
+      "step": 1850
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.618158567774936e-05,
+      "loss": 0.0137,
+      "step": 1851
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.6173913043478262e-05,
+      "loss": 0.0145,
+      "step": 1852
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.616624040920716e-05,
+      "loss": 0.0355,
+      "step": 1853
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.6158567774936062e-05,
+      "loss": 0.9601,
+      "step": 1854
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.615089514066496e-05,
+      "loss": 0.0135,
+      "step": 1855
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6143222506393862e-05,
+      "loss": 1.0129,
+      "step": 1856
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6135549872122763e-05,
+      "loss": 0.0339,
+      "step": 1857
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.612787723785166e-05,
+      "loss": 0.0131,
+      "step": 1858
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6120204603580563e-05,
+      "loss": 0.009,
+      "step": 1859
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6112531969309465e-05,
+      "loss": 0.0087,
+      "step": 1860
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6104859335038363e-05,
+      "loss": 0.012,
+      "step": 1861
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.609718670076726e-05,
+      "loss": 0.0088,
+      "step": 1862
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6089514066496166e-05,
+      "loss": 0.0117,
+      "step": 1863
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6081841432225065e-05,
+      "loss": 2.5214,
+      "step": 1864
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6074168797953963e-05,
+      "loss": 0.0107,
+      "step": 1865
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6066496163682864e-05,
+      "loss": 0.0101,
+      "step": 1866
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6058823529411766e-05,
+      "loss": 0.0118,
+      "step": 1867
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6051150895140664e-05,
+      "loss": 0.0087,
+      "step": 1868
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6043478260869566e-05,
+      "loss": 0.0122,
+      "step": 1869
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6035805626598464e-05,
+      "loss": 0.0084,
+      "step": 1870
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6028132992327366e-05,
+      "loss": 0.0104,
+      "step": 1871
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6020460358056267e-05,
+      "loss": 0.1587,
+      "step": 1872
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6012787723785166e-05,
+      "loss": 0.0057,
+      "step": 1873
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6005115089514067e-05,
+      "loss": 0.0085,
+      "step": 1874
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.599744245524297e-05,
+      "loss": 0.0056,
+      "step": 1875
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.5989769820971867e-05,
+      "loss": 1.2194,
+      "step": 1876
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.598209718670077e-05,
+      "loss": 0.0056,
+      "step": 1877
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.5974424552429667e-05,
+      "loss": 0.0057,
+      "step": 1878
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.596675191815857e-05,
+      "loss": 1.0276,
+      "step": 1879
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.595907928388747e-05,
+      "loss": 0.0083,
+      "step": 1880
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.5951406649616368e-05,
+      "loss": 0.0085,
+      "step": 1881
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.594373401534527e-05,
+      "loss": 0.0363,
+      "step": 1882
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.593606138107417e-05,
+      "loss": 0.0157,
+      "step": 1883
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.592838874680307e-05,
+      "loss": 0.0313,
+      "step": 1884
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.5920716112531968e-05,
+      "loss": 0.0062,
+      "step": 1885
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.5913043478260873e-05,
+      "loss": 0.0195,
+      "step": 1886
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.590537084398977e-05,
+      "loss": 0.0151,
+      "step": 1887
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.589769820971867e-05,
+      "loss": 0.0101,
+      "step": 1888
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.589002557544757e-05,
+      "loss": 1.1092,
+      "step": 1889
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.5882352941176473e-05,
+      "loss": 0.0323,
+      "step": 1890
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.587468030690537e-05,
+      "loss": 0.0086,
+      "step": 1891
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.5867007672634272e-05,
+      "loss": 0.0147,
+      "step": 1892
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.585933503836317e-05,
+      "loss": 0.0084,
+      "step": 1893
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.5851662404092072e-05,
+      "loss": 0.5596,
+      "step": 1894
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5843989769820974e-05,
+      "loss": 0.0114,
+      "step": 1895
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5836317135549872e-05,
+      "loss": 0.0083,
+      "step": 1896
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.582864450127877e-05,
+      "loss": 0.0082,
+      "step": 1897
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5820971867007675e-05,
+      "loss": 0.0146,
+      "step": 1898
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5813299232736574e-05,
+      "loss": 0.0081,
+      "step": 1899
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5805626598465472e-05,
+      "loss": 0.0081,
+      "step": 1900
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5797953964194373e-05,
+      "loss": 0.0116,
+      "step": 1901
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5790281329923275e-05,
+      "loss": 0.0084,
+      "step": 1902
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5782608695652173e-05,
+      "loss": 0.0108,
+      "step": 1903
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5774936061381075e-05,
+      "loss": 0.6146,
+      "step": 1904
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5767263427109973e-05,
+      "loss": 0.0162,
+      "step": 1905
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5759590792838875e-05,
+      "loss": 0.0134,
+      "step": 1906
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5751918158567776e-05,
+      "loss": 0.0047,
+      "step": 1907
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5744245524296674e-05,
+      "loss": 0.0045,
+      "step": 1908
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5736572890025576e-05,
+      "loss": 0.0087,
+      "step": 1909
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5728900255754478e-05,
+      "loss": 2.0913,
+      "step": 1910
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5721227621483376e-05,
+      "loss": 0.0075,
+      "step": 1911
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5713554987212274e-05,
+      "loss": 1.3387,
+      "step": 1912
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.570588235294118e-05,
+      "loss": 0.0102,
+      "step": 1913
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5698209718670077e-05,
+      "loss": 0.0076,
+      "step": 1914
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5690537084398976e-05,
+      "loss": 0.0131,
+      "step": 1915
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5682864450127877e-05,
+      "loss": 0.0079,
+      "step": 1916
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.567519181585678e-05,
+      "loss": 0.0102,
+      "step": 1917
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5667519181585677e-05,
+      "loss": 0.9554,
+      "step": 1918
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.565984654731458e-05,
+      "loss": 0.0088,
+      "step": 1919
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.5652173913043477e-05,
+      "loss": 1.0265,
+      "step": 1920
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.564450127877238e-05,
+      "loss": 0.0155,
+      "step": 1921
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.563682864450128e-05,
+      "loss": 0.0079,
+      "step": 1922
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.562915601023018e-05,
+      "loss": 1.0467,
+      "step": 1923
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.5621483375959077e-05,
+      "loss": 0.0132,
+      "step": 1924
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.561381074168798e-05,
+      "loss": 0.0077,
+      "step": 1925
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.560613810741688e-05,
+      "loss": 0.0064,
+      "step": 1926
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.559846547314578e-05,
+      "loss": 0.0104,
+      "step": 1927
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.559079283887468e-05,
+      "loss": 0.0096,
+      "step": 1928
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.558312020460358e-05,
+      "loss": 0.0049,
+      "step": 1929
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.5575447570332483e-05,
+      "loss": 0.0104,
+      "step": 1930
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.556777493606138e-05,
+      "loss": 1.0424,
+      "step": 1931
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5560102301790283e-05,
+      "loss": 0.0079,
+      "step": 1932
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5552429667519184e-05,
+      "loss": 0.0104,
+      "step": 1933
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5544757033248083e-05,
+      "loss": 0.0077,
+      "step": 1934
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.553708439897698e-05,
+      "loss": 0.005,
+      "step": 1935
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5529411764705886e-05,
+      "loss": 2.6544,
+      "step": 1936
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5521739130434784e-05,
+      "loss": 0.0107,
+      "step": 1937
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5514066496163682e-05,
+      "loss": 0.0124,
+      "step": 1938
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5506393861892584e-05,
+      "loss": 0.0049,
+      "step": 1939
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5498721227621485e-05,
+      "loss": 0.0078,
+      "step": 1940
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5491048593350384e-05,
+      "loss": 2.1009,
+      "step": 1941
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5483375959079285e-05,
+      "loss": 0.0116,
+      "step": 1942
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5475703324808183e-05,
+      "loss": 0.0049,
+      "step": 1943
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5468030690537085e-05,
+      "loss": 0.0079,
+      "step": 1944
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5460358056265987e-05,
+      "loss": 1.0307,
+      "step": 1945
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5452685421994885e-05,
+      "loss": 0.0147,
+      "step": 1946
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5445012787723783e-05,
+      "loss": 0.0084,
+      "step": 1947
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5437340153452688e-05,
+      "loss": 0.0262,
+      "step": 1948
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5429667519181586e-05,
+      "loss": 0.0115,
+      "step": 1949
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5421994884910485e-05,
+      "loss": 1.3221,
+      "step": 1950
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5414322250639386e-05,
+      "loss": 0.7929,
+      "step": 1951
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5406649616368288e-05,
+      "loss": 1.2069,
+      "step": 1952
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5398976982097186e-05,
+      "loss": 0.0081,
+      "step": 1953
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5391304347826088e-05,
+      "loss": 0.0069,
+      "step": 1954
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5383631713554986e-05,
+      "loss": 0.0097,
+      "step": 1955
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5375959079283887e-05,
+      "loss": 0.0082,
+      "step": 1956
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.536828644501279e-05,
+      "loss": 0.0061,
+      "step": 1957
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5360613810741687e-05,
+      "loss": 0.0082,
+      "step": 1958
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.535294117647059e-05,
+      "loss": 0.0113,
+      "step": 1959
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.534526854219949e-05,
+      "loss": 0.0098,
+      "step": 1960
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.533759590792839e-05,
+      "loss": 1.0221,
+      "step": 1961
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.5329923273657287e-05,
+      "loss": 0.0039,
+      "step": 1962
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.5322250639386192e-05,
+      "loss": 0.0127,
+      "step": 1963
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.531457800511509e-05,
+      "loss": 0.0051,
+      "step": 1964
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.530690537084399e-05,
+      "loss": 0.0086,
+      "step": 1965
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.529923273657289e-05,
+      "loss": 0.0096,
+      "step": 1966
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.529156010230179e-05,
+      "loss": 0.0164,
+      "step": 1967
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.528388746803069e-05,
+      "loss": 1.0127,
+      "step": 1968
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.527621483375959e-05,
+      "loss": 0.0149,
+      "step": 1969
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.526854219948849e-05,
+      "loss": 0.0085,
+      "step": 1970
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.526086956521739e-05,
+      "loss": 0.0116,
+      "step": 1971
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5253196930946293e-05,
+      "loss": 0.0168,
+      "step": 1972
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5245524296675191e-05,
+      "loss": 0.0052,
+      "step": 1973
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5237851662404091e-05,
+      "loss": 0.0052,
+      "step": 1974
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5230179028132993e-05,
+      "loss": 1.0013,
+      "step": 1975
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5222506393861893e-05,
+      "loss": 0.0122,
+      "step": 1976
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.521483375959079e-05,
+      "loss": 0.0087,
+      "step": 1977
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5207161125319694e-05,
+      "loss": 1.9624,
+      "step": 1978
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5199488491048592e-05,
+      "loss": 0.0088,
+      "step": 1979
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5191815856777496e-05,
+      "loss": 0.0596,
+      "step": 1980
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9090909090909091,
+      "eval_loss": 0.40464258193969727,
+      "eval_runtime": 58.0491,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1980
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5184143222506396e-05,
+      "loss": 0.009,
+      "step": 1981
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5176470588235294e-05,
+      "loss": 0.0063,
+      "step": 1982
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5168797953964197e-05,
+      "loss": 0.0092,
+      "step": 1983
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5161125319693095e-05,
+      "loss": 0.0092,
+      "step": 1984
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5153452685421995e-05,
+      "loss": 0.0259,
+      "step": 1985
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5145780051150897e-05,
+      "loss": 0.0093,
+      "step": 1986
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5138107416879797e-05,
+      "loss": 1.3347,
+      "step": 1987
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5130434782608695e-05,
+      "loss": 0.0135,
+      "step": 1988
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5122762148337598e-05,
+      "loss": 0.0134,
+      "step": 1989
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5115089514066497e-05,
+      "loss": 0.0051,
+      "step": 1990
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5107416879795396e-05,
+      "loss": 0.0106,
+      "step": 1991
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5099744245524298e-05,
+      "loss": 0.013,
+      "step": 1992
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5092071611253198e-05,
+      "loss": 0.0095,
+      "step": 1993
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5084398976982096e-05,
+      "loss": 0.0096,
+      "step": 1994
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5076726342711e-05,
+      "loss": 0.0112,
+      "step": 1995
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5069053708439898e-05,
+      "loss": 0.0137,
+      "step": 1996
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5061381074168798e-05,
+      "loss": 0.9611,
+      "step": 1997
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50537084398977e-05,
+      "loss": 0.009,
+      "step": 1998
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50460358056266e-05,
+      "loss": 0.0066,
+      "step": 1999
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.5038363171355497e-05,
+      "loss": 0.0091,
+      "step": 2000
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50306905370844e-05,
+      "loss": 0.9856,
+      "step": 2001
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5023017902813299e-05,
+      "loss": 0.0137,
+      "step": 2002
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5015345268542199e-05,
+      "loss": 0.0094,
+      "step": 2003
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.50076726342711e-05,
+      "loss": 0.9276,
+      "step": 2004
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.0051,
+      "step": 2005
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4992327365728902e-05,
+      "loss": 0.0097,
+      "step": 2006
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.49846547314578e-05,
+      "loss": 0.8053,
+      "step": 2007
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4976982097186702e-05,
+      "loss": 0.5635,
+      "step": 2008
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4969309462915602e-05,
+      "loss": 0.0099,
+      "step": 2009
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4961636828644502e-05,
+      "loss": 0.0108,
+      "step": 2010
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4953964194373402e-05,
+      "loss": 1.3337,
+      "step": 2011
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4946291560102303e-05,
+      "loss": 0.0098,
+      "step": 2012
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4938618925831201e-05,
+      "loss": 0.0099,
+      "step": 2013
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4930946291560103e-05,
+      "loss": 0.011,
+      "step": 2014
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4923273657289003e-05,
+      "loss": 0.0166,
+      "step": 2015
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4915601023017903e-05,
+      "loss": 1.3541,
+      "step": 2016
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4907928388746803e-05,
+      "loss": 0.0145,
+      "step": 2017
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4900255754475704e-05,
+      "loss": 0.0051,
+      "step": 2018
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4892583120204604e-05,
+      "loss": 0.0051,
+      "step": 2019
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4884910485933504e-05,
+      "loss": 0.0139,
+      "step": 2020
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4877237851662404e-05,
+      "loss": 0.0097,
+      "step": 2021
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4869565217391306e-05,
+      "loss": 0.9418,
+      "step": 2022
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4861892583120204e-05,
+      "loss": 0.9619,
+      "step": 2023
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4854219948849106e-05,
+      "loss": 1.3495,
+      "step": 2024
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4846547314578006e-05,
+      "loss": 0.0051,
+      "step": 2025
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4838874680306905e-05,
+      "loss": 1.3222,
+      "step": 2026
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4831202046035805e-05,
+      "loss": 1.1638,
+      "step": 2027
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4823529411764707e-05,
+      "loss": 0.009,
+      "step": 2028
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4815856777493605e-05,
+      "loss": 0.0059,
+      "step": 2029
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4808184143222507e-05,
+      "loss": 0.9946,
+      "step": 2030
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4800511508951408e-05,
+      "loss": 0.1291,
+      "step": 2031
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4792838874680307e-05,
+      "loss": 0.0089,
+      "step": 2032
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4785166240409208e-05,
+      "loss": 0.8113,
+      "step": 2033
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4777493606138108e-05,
+      "loss": 0.0109,
+      "step": 2034
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4769820971867008e-05,
+      "loss": 0.0061,
+      "step": 2035
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4762148337595908e-05,
+      "loss": 0.0157,
+      "step": 2036
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.475447570332481e-05,
+      "loss": 0.0104,
+      "step": 2037
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.4746803069053708e-05,
+      "loss": 0.012,
+      "step": 2038
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.473913043478261e-05,
+      "loss": 0.0106,
+      "step": 2039
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.473145780051151e-05,
+      "loss": 0.006,
+      "step": 2040
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.472378516624041e-05,
+      "loss": 0.0122,
+      "step": 2041
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.471611253196931e-05,
+      "loss": 0.0103,
+      "step": 2042
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.470843989769821e-05,
+      "loss": 0.9465,
+      "step": 2043
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.4700767263427109e-05,
+      "loss": 0.0149,
+      "step": 2044
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.469309462915601e-05,
+      "loss": 0.0146,
+      "step": 2045
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.468542199488491e-05,
+      "loss": 0.9225,
+      "step": 2046
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.4677749360613812e-05,
+      "loss": 0.0101,
+      "step": 2047
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.467007672634271e-05,
+      "loss": 0.0052,
+      "step": 2048
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.4662404092071612e-05,
+      "loss": 1.682,
+      "step": 2049
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4654731457800512e-05,
+      "loss": 0.9498,
+      "step": 2050
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4647058823529412e-05,
+      "loss": 0.0097,
+      "step": 2051
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4639386189258312e-05,
+      "loss": 0.0067,
+      "step": 2052
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4631713554987213e-05,
+      "loss": 0.0052,
+      "step": 2053
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4624040920716112e-05,
+      "loss": 0.0055,
+      "step": 2054
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4616368286445013e-05,
+      "loss": 0.0225,
+      "step": 2055
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4608695652173915e-05,
+      "loss": 0.0052,
+      "step": 2056
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4601023017902813e-05,
+      "loss": 0.0066,
+      "step": 2057
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4593350383631715e-05,
+      "loss": 0.0112,
+      "step": 2058
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4585677749360615e-05,
+      "loss": 1.3145,
+      "step": 2059
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4578005115089514e-05,
+      "loss": 0.0099,
+      "step": 2060
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4570332480818414e-05,
+      "loss": 0.0055,
+      "step": 2061
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4562659846547316e-05,
+      "loss": 0.0149,
+      "step": 2062
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4554987212276214e-05,
+      "loss": 0.0102,
+      "step": 2063
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4547314578005116e-05,
+      "loss": 0.0108,
+      "step": 2064
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4539641943734016e-05,
+      "loss": 0.0065,
+      "step": 2065
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4531969309462916e-05,
+      "loss": 0.0185,
+      "step": 2066
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4524296675191816e-05,
+      "loss": 0.9659,
+      "step": 2067
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4516624040920717e-05,
+      "loss": 0.0193,
+      "step": 2068
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4508951406649615e-05,
+      "loss": 0.0146,
+      "step": 2069
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4501278772378517e-05,
+      "loss": 0.0061,
+      "step": 2070
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4493606138107417e-05,
+      "loss": 0.0222,
+      "step": 2071
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4485933503836319e-05,
+      "loss": 0.0096,
+      "step": 2072
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4478260869565217e-05,
+      "loss": 0.0096,
+      "step": 2073
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4470588235294118e-05,
+      "loss": 0.005,
+      "step": 2074
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4462915601023018e-05,
+      "loss": 0.8301,
+      "step": 2075
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4455242966751918e-05,
+      "loss": 0.0049,
+      "step": 2076
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4447570332480818e-05,
+      "loss": 0.0097,
+      "step": 2077
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.443989769820972e-05,
+      "loss": 0.0062,
+      "step": 2078
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.4432225063938618e-05,
+      "loss": 1.3459,
+      "step": 2079
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.442455242966752e-05,
+      "loss": 0.0096,
+      "step": 2080
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4416879795396421e-05,
+      "loss": 1.2176,
+      "step": 2081
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.440920716112532e-05,
+      "loss": 0.0114,
+      "step": 2082
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4401534526854221e-05,
+      "loss": 0.0109,
+      "step": 2083
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4393861892583121e-05,
+      "loss": 0.0105,
+      "step": 2084
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.4386189258312021e-05,
+      "loss": 0.0107,
+      "step": 2085
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.437851662404092e-05,
+      "loss": 0.9442,
+      "step": 2086
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.4370843989769822e-05,
+      "loss": 0.01,
+      "step": 2087
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.436317135549872e-05,
+      "loss": 0.0099,
+      "step": 2088
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4355498721227622e-05,
+      "loss": 0.0167,
+      "step": 2089
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4347826086956522e-05,
+      "loss": 0.0051,
+      "step": 2090
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4340153452685422e-05,
+      "loss": 1.333,
+      "step": 2091
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4332480818414322e-05,
+      "loss": 0.0111,
+      "step": 2092
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4324808184143224e-05,
+      "loss": 0.0103,
+      "step": 2093
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4317135549872122e-05,
+      "loss": 0.0111,
+      "step": 2094
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4309462915601023e-05,
+      "loss": 0.0165,
+      "step": 2095
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4301790281329923e-05,
+      "loss": 0.0089,
+      "step": 2096
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4294117647058823e-05,
+      "loss": 0.0143,
+      "step": 2097
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4286445012787723e-05,
+      "loss": 0.4759,
+      "step": 2098
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4278772378516625e-05,
+      "loss": 0.0102,
+      "step": 2099
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4271099744245525e-05,
+      "loss": 0.0059,
+      "step": 2100
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4263427109974425e-05,
+      "loss": 0.0152,
+      "step": 2101
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4255754475703325e-05,
+      "loss": 0.0051,
+      "step": 2102
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4248081841432226e-05,
+      "loss": 0.0098,
+      "step": 2103
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4240409207161126e-05,
+      "loss": 0.0147,
+      "step": 2104
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4232736572890026e-05,
+      "loss": 0.0051,
+      "step": 2105
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4225063938618928e-05,
+      "loss": 1.3384,
+      "step": 2106
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4217391304347826e-05,
+      "loss": 0.0055,
+      "step": 2107
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4209718670076727e-05,
+      "loss": 0.0136,
+      "step": 2108
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4202046035805627e-05,
+      "loss": 0.0056,
+      "step": 2109
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4194373401534527e-05,
+      "loss": 0.0094,
+      "step": 2110
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4186700767263427e-05,
+      "loss": 0.0135,
+      "step": 2111
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4179028132992329e-05,
+      "loss": 0.0129,
+      "step": 2112
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4171355498721227e-05,
+      "loss": 0.0051,
+      "step": 2113
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4163682864450129e-05,
+      "loss": 0.0134,
+      "step": 2114
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4156010230179029e-05,
+      "loss": 0.0091,
+      "step": 2115
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4148337595907929e-05,
+      "loss": 0.0072,
+      "step": 2116
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.4140664961636828e-05,
+      "loss": 0.005,
+      "step": 2117
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.413299232736573e-05,
+      "loss": 0.0112,
+      "step": 2118
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.4125319693094628e-05,
+      "loss": 0.005,
+      "step": 2119
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.411764705882353e-05,
+      "loss": 0.0085,
+      "step": 2120
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.410997442455243e-05,
+      "loss": 0.0121,
+      "step": 2121
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.410230179028133e-05,
+      "loss": 0.0084,
+      "step": 2122
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.409462915601023e-05,
+      "loss": 0.0084,
+      "step": 2123
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.4086956521739131e-05,
+      "loss": 0.0083,
+      "step": 2124
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4079283887468031e-05,
+      "loss": 1.3402,
+      "step": 2125
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4071611253196931e-05,
+      "loss": 0.0124,
+      "step": 2126
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4063938618925831e-05,
+      "loss": 0.0082,
+      "step": 2127
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4056265984654733e-05,
+      "loss": 0.9694,
+      "step": 2128
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4048593350383633e-05,
+      "loss": 0.0081,
+      "step": 2129
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4040920716112532e-05,
+      "loss": 1.2102,
+      "step": 2130
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4033248081841434e-05,
+      "loss": 0.0113,
+      "step": 2131
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4025575447570332e-05,
+      "loss": 0.0114,
+      "step": 2132
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4017902813299234e-05,
+      "loss": 0.0252,
+      "step": 2133
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4010230179028134e-05,
+      "loss": 0.9785,
+      "step": 2134
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4002557544757034e-05,
+      "loss": 0.0061,
+      "step": 2135
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.3994884910485934e-05,
+      "loss": 0.0096,
+      "step": 2136
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3987212276214835e-05,
+      "loss": 0.0088,
+      "step": 2137
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3979539641943733e-05,
+      "loss": 0.0047,
+      "step": 2138
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3971867007672635e-05,
+      "loss": 0.0108,
+      "step": 2139
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3964194373401535e-05,
+      "loss": 0.0081,
+      "step": 2140
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3956521739130435e-05,
+      "loss": 0.0108,
+      "step": 2141
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3948849104859335e-05,
+      "loss": 0.0126,
+      "step": 2142
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3941176470588236e-05,
+      "loss": 0.0046,
+      "step": 2143
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3933503836317135e-05,
+      "loss": 0.0122,
+      "step": 2144
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3925831202046036e-05,
+      "loss": 0.0075,
+      "step": 2145
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3918158567774936e-05,
+      "loss": 0.0123,
+      "step": 2146
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3910485933503836e-05,
+      "loss": 0.0134,
+      "step": 2147
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3902813299232736e-05,
+      "loss": 0.0115,
+      "step": 2148
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3895140664961638e-05,
+      "loss": 0.0044,
+      "step": 2149
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3887468030690536e-05,
+      "loss": 0.0121,
+      "step": 2150
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3879795396419437e-05,
+      "loss": 0.0101,
+      "step": 2151
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3872122762148337e-05,
+      "loss": 0.0099,
+      "step": 2152
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3864450127877239e-05,
+      "loss": 0.0044,
+      "step": 2153
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3856777493606139e-05,
+      "loss": 0.0094,
+      "step": 2154
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3849104859335039e-05,
+      "loss": 0.0124,
+      "step": 2155
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.384143222506394e-05,
+      "loss": 0.0069,
+      "step": 2156
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.3833759590792839e-05,
+      "loss": 0.0068,
+      "step": 2157
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.382608695652174e-05,
+      "loss": 0.0094,
+      "step": 2158
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.381841432225064e-05,
+      "loss": 0.0067,
+      "step": 2159
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.381074168797954e-05,
+      "loss": 0.0044,
+      "step": 2160
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.380306905370844e-05,
+      "loss": 0.0041,
+      "step": 2161
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.3795396419437342e-05,
+      "loss": 0.0041,
+      "step": 2162
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.378772378516624e-05,
+      "loss": 0.009,
+      "step": 2163
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.3780051150895141e-05,
+      "loss": 0.0092,
+      "step": 2164
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3772378516624041e-05,
+      "loss": 0.0101,
+      "step": 2165
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3764705882352941e-05,
+      "loss": 0.0064,
+      "step": 2166
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3757033248081841e-05,
+      "loss": 0.0088,
+      "step": 2167
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3749360613810743e-05,
+      "loss": 0.0059,
+      "step": 2168
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3741687979539641e-05,
+      "loss": 0.0062,
+      "step": 2169
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3734015345268543e-05,
+      "loss": 0.0085,
+      "step": 2170
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3726342710997443e-05,
+      "loss": 0.0141,
+      "step": 2171
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3718670076726343e-05,
+      "loss": 0.0061,
+      "step": 2172
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3710997442455242e-05,
+      "loss": 0.0084,
+      "step": 2173
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3703324808184144e-05,
+      "loss": 0.0038,
+      "step": 2174
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3695652173913042e-05,
+      "loss": 0.006,
+      "step": 2175
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3687979539641944e-05,
+      "loss": 0.0039,
+      "step": 2176
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3680306905370844e-05,
+      "loss": 0.0205,
+      "step": 2177
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3672634271099745e-05,
+      "loss": 0.0062,
+      "step": 2178
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3664961636828645e-05,
+      "loss": 0.0021,
+      "step": 2179
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3657289002557545e-05,
+      "loss": 0.0058,
+      "step": 2180
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3649616368286447e-05,
+      "loss": 0.0101,
+      "step": 2181
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3641943734015345e-05,
+      "loss": 0.0036,
+      "step": 2182
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3634271099744247e-05,
+      "loss": 0.0037,
+      "step": 2183
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3626598465473147e-05,
+      "loss": 0.0036,
+      "step": 2184
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3618925831202047e-05,
+      "loss": 0.0036,
+      "step": 2185
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3611253196930946e-05,
+      "loss": 0.0035,
+      "step": 2186
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3603580562659848e-05,
+      "loss": 0.0059,
+      "step": 2187
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3595907928388746e-05,
+      "loss": 0.0056,
+      "step": 2188
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3588235294117648e-05,
+      "loss": 0.0139,
+      "step": 2189
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3580562659846548e-05,
+      "loss": 0.0558,
+      "step": 2190
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3572890025575448e-05,
+      "loss": 1.419,
+      "step": 2191
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3565217391304348e-05,
+      "loss": 0.0058,
+      "step": 2192
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.355754475703325e-05,
+      "loss": 0.0035,
+      "step": 2193
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3549872122762147e-05,
+      "loss": 0.0055,
+      "step": 2194
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3542199488491049e-05,
+      "loss": 1.2049,
+      "step": 2195
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3534526854219949e-05,
+      "loss": 1.3646,
+      "step": 2196
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3526854219948849e-05,
+      "loss": 1.3803,
+      "step": 2197
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3519181585677749e-05,
+      "loss": 0.0055,
+      "step": 2198
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.351150895140665e-05,
+      "loss": 0.0059,
+      "step": 2199
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.3503836317135549e-05,
+      "loss": 0.0056,
+      "step": 2200
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.349616368286445e-05,
+      "loss": 0.0097,
+      "step": 2201
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.348849104859335e-05,
+      "loss": 1.3497,
+      "step": 2202
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.348081841432225e-05,
+      "loss": 1.0989,
+      "step": 2203
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3473145780051152e-05,
+      "loss": 0.0117,
+      "step": 2204
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3465473145780052e-05,
+      "loss": 0.0038,
+      "step": 2205
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3457800511508953e-05,
+      "loss": 0.0078,
+      "step": 2206
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3450127877237852e-05,
+      "loss": 0.0058,
+      "step": 2207
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3442455242966753e-05,
+      "loss": 0.9775,
+      "step": 2208
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3434782608695653e-05,
+      "loss": 0.0037,
+      "step": 2209
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3427109974424553e-05,
+      "loss": 0.0079,
+      "step": 2210
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3419437340153453e-05,
+      "loss": 0.0663,
+      "step": 2211
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3411764705882354e-05,
+      "loss": 0.0059,
+      "step": 2212
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3404092071611253e-05,
+      "loss": 0.0039,
+      "step": 2213
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3396419437340154e-05,
+      "loss": 1.3169,
+      "step": 2214
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3388746803069054e-05,
+      "loss": 0.0061,
+      "step": 2215
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3381074168797954e-05,
+      "loss": 0.006,
+      "step": 2216
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3373401534526854e-05,
+      "loss": 0.004,
+      "step": 2217
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3365728900255756e-05,
+      "loss": 0.0045,
+      "step": 2218
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3358056265984654e-05,
+      "loss": 0.006,
+      "step": 2219
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3350383631713556e-05,
+      "loss": 0.0058,
+      "step": 2220
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3342710997442455e-05,
+      "loss": 0.014,
+      "step": 2221
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3335038363171355e-05,
+      "loss": 1.1674,
+      "step": 2222
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3327365728900255e-05,
+      "loss": 0.004,
+      "step": 2223
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3319693094629157e-05,
+      "loss": 0.0052,
+      "step": 2224
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3312020460358055e-05,
+      "loss": 0.0095,
+      "step": 2225
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3304347826086957e-05,
+      "loss": 0.0053,
+      "step": 2226
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3296675191815857e-05,
+      "loss": 0.0099,
+      "step": 2227
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3289002557544757e-05,
+      "loss": 1.3879,
+      "step": 2228
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3281329923273658e-05,
+      "loss": 0.0041,
+      "step": 2229
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3273657289002558e-05,
+      "loss": 0.006,
+      "step": 2230
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.326598465473146e-05,
+      "loss": 0.006,
+      "step": 2231
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.3258312020460358e-05,
+      "loss": 0.012,
+      "step": 2232
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.325063938618926e-05,
+      "loss": 0.0107,
+      "step": 2233
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.324296675191816e-05,
+      "loss": 0.0061,
+      "step": 2234
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.323529411764706e-05,
+      "loss": 0.0041,
+      "step": 2235
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.322762148337596e-05,
+      "loss": 0.006,
+      "step": 2236
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.3219948849104861e-05,
+      "loss": 0.0064,
+      "step": 2237
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.3212276214833759e-05,
+      "loss": 0.006,
+      "step": 2238
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.320460358056266e-05,
+      "loss": 0.0035,
+      "step": 2239
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.319693094629156e-05,
+      "loss": 0.8375,
+      "step": 2240
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.318925831202046e-05,
+      "loss": 0.004,
+      "step": 2241
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.318158567774936e-05,
+      "loss": 0.0077,
+      "step": 2242
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.3173913043478262e-05,
+      "loss": 0.006,
+      "step": 2243
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.316624040920716e-05,
+      "loss": 0.0041,
+      "step": 2244
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3158567774936062e-05,
+      "loss": 0.0034,
+      "step": 2245
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3150895140664962e-05,
+      "loss": 0.8219,
+      "step": 2246
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3143222506393862e-05,
+      "loss": 0.004,
+      "step": 2247
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3135549872122762e-05,
+      "loss": 0.004,
+      "step": 2248
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3127877237851663e-05,
+      "loss": 1.1328,
+      "step": 2249
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3120204603580562e-05,
+      "loss": 0.0039,
+      "step": 2250
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3112531969309463e-05,
+      "loss": 0.0283,
+      "step": 2251
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3104859335038363e-05,
+      "loss": 0.0058,
+      "step": 2252
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3097186700767263e-05,
+      "loss": 0.004,
+      "step": 2253
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3089514066496165e-05,
+      "loss": 0.0057,
+      "step": 2254
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3081841432225064e-05,
+      "loss": 0.0076,
+      "step": 2255
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3074168797953966e-05,
+      "loss": 0.004,
+      "step": 2256
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3066496163682864e-05,
+      "loss": 0.0076,
+      "step": 2257
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3058823529411766e-05,
+      "loss": 0.0091,
+      "step": 2258
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3051150895140666e-05,
+      "loss": 0.0038,
+      "step": 2259
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3043478260869566e-05,
+      "loss": 0.0057,
+      "step": 2260
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3035805626598466e-05,
+      "loss": 0.0057,
+      "step": 2261
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3028132992327367e-05,
+      "loss": 1.1251,
+      "step": 2262
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3020460358056266e-05,
+      "loss": 1.1222,
+      "step": 2263
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.3012787723785167e-05,
+      "loss": 0.0047,
+      "step": 2264
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.3005115089514067e-05,
+      "loss": 0.008,
+      "step": 2265
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.2997442455242967e-05,
+      "loss": 0.0061,
+      "step": 2266
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.2989769820971867e-05,
+      "loss": 1.394,
+      "step": 2267
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2982097186700768e-05,
+      "loss": 0.0039,
+      "step": 2268
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2974424552429667e-05,
+      "loss": 0.0103,
+      "step": 2269
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2966751918158568e-05,
+      "loss": 1.1008,
+      "step": 2270
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2959079283887468e-05,
+      "loss": 0.0051,
+      "step": 2271
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2951406649616368e-05,
+      "loss": 0.004,
+      "step": 2272
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2943734015345268e-05,
+      "loss": 0.78,
+      "step": 2273
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.293606138107417e-05,
+      "loss": 0.0069,
+      "step": 2274
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2928388746803068e-05,
+      "loss": 0.0094,
+      "step": 2275
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.292071611253197e-05,
+      "loss": 0.0067,
+      "step": 2276
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.291304347826087e-05,
+      "loss": 0.0041,
+      "step": 2277
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.290537084398977e-05,
+      "loss": 0.0079,
+      "step": 2278
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.2897698209718671e-05,
+      "loss": 0.014,
+      "step": 2279
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.2890025575447571e-05,
+      "loss": 1.3775,
+      "step": 2280
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.288235294117647e-05,
+      "loss": 0.0042,
+      "step": 2281
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.287468030690537e-05,
+      "loss": 1.1862,
+      "step": 2282
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2867007672634272e-05,
+      "loss": 0.0042,
+      "step": 2283
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2859335038363172e-05,
+      "loss": 0.0098,
+      "step": 2284
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2851662404092072e-05,
+      "loss": 0.0038,
+      "step": 2285
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2843989769820972e-05,
+      "loss": 0.0127,
+      "step": 2286
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2836317135549874e-05,
+      "loss": 0.0072,
+      "step": 2287
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2828644501278772e-05,
+      "loss": 0.0098,
+      "step": 2288
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2820971867007674e-05,
+      "loss": 0.0069,
+      "step": 2289
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2813299232736573e-05,
+      "loss": 0.9529,
+      "step": 2290
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2805626598465473e-05,
+      "loss": 0.0096,
+      "step": 2291
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2797953964194373e-05,
+      "loss": 0.0094,
+      "step": 2292
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2790281329923275e-05,
+      "loss": 1.0639,
+      "step": 2293
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2782608695652173e-05,
+      "loss": 0.0043,
+      "step": 2294
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2774936061381075e-05,
+      "loss": 0.0098,
+      "step": 2295
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2767263427109975e-05,
+      "loss": 0.0084,
+      "step": 2296
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2759590792838875e-05,
+      "loss": 0.0071,
+      "step": 2297
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2751918158567774e-05,
+      "loss": 0.0036,
+      "step": 2298
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2744245524296676e-05,
+      "loss": 0.0097,
+      "step": 2299
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2736572890025574e-05,
+      "loss": 0.0072,
+      "step": 2300
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2728900255754476e-05,
+      "loss": 0.0043,
+      "step": 2301
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2721227621483376e-05,
+      "loss": 1.3742,
+      "step": 2302
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2713554987212276e-05,
+      "loss": 0.0086,
+      "step": 2303
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2705882352941177e-05,
+      "loss": 0.0071,
+      "step": 2304
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2698209718670077e-05,
+      "loss": 0.007,
+      "step": 2305
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2690537084398977e-05,
+      "loss": 0.0099,
+      "step": 2306
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2682864450127877e-05,
+      "loss": 1.0437,
+      "step": 2307
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2675191815856779e-05,
+      "loss": 0.01,
+      "step": 2308
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2667519181585679e-05,
+      "loss": 0.1771,
+      "step": 2309
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2659846547314579e-05,
+      "loss": 0.0103,
+      "step": 2310
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.2652173913043479e-05,
+      "loss": 0.0104,
+      "step": 2311
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.264450127877238e-05,
+      "loss": 0.0077,
+      "step": 2312
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.2636828644501278e-05,
+      "loss": 0.0107,
+      "step": 2313
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.262915601023018e-05,
+      "loss": 0.0107,
+      "step": 2314
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.262148337595908e-05,
+      "loss": 0.0135,
+      "step": 2315
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.261381074168798e-05,
+      "loss": 0.0103,
+      "step": 2316
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.260613810741688e-05,
+      "loss": 0.0136,
+      "step": 2317
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.2598465473145781e-05,
+      "loss": 0.0045,
+      "step": 2318
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.259079283887468e-05,
+      "loss": 0.0104,
+      "step": 2319
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2583120204603581e-05,
+      "loss": 0.0073,
+      "step": 2320
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2575447570332481e-05,
+      "loss": 0.9265,
+      "step": 2321
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2567774936061381e-05,
+      "loss": 0.0101,
+      "step": 2322
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2560102301790281e-05,
+      "loss": 1.3704,
+      "step": 2323
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2552429667519183e-05,
+      "loss": 1.415,
+      "step": 2324
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.254475703324808e-05,
+      "loss": 0.0098,
+      "step": 2325
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2537084398976982e-05,
+      "loss": 0.0138,
+      "step": 2326
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2529411764705884e-05,
+      "loss": 0.0064,
+      "step": 2327
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2521739130434782e-05,
+      "loss": 0.0093,
+      "step": 2328
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2514066496163684e-05,
+      "loss": 0.0119,
+      "step": 2329
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2506393861892584e-05,
+      "loss": 0.0097,
+      "step": 2330
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2498721227621484e-05,
+      "loss": 0.0049,
+      "step": 2331
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2491048593350384e-05,
+      "loss": 0.0044,
+      "step": 2332
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2483375959079285e-05,
+      "loss": 0.0067,
+      "step": 2333
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2475703324808183e-05,
+      "loss": 0.0117,
+      "step": 2334
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2468030690537085e-05,
+      "loss": 0.0066,
+      "step": 2335
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2460358056265985e-05,
+      "loss": 1.0804,
+      "step": 2336
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2452685421994887e-05,
+      "loss": 0.0065,
+      "step": 2337
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2445012787723785e-05,
+      "loss": 0.0069,
+      "step": 2338
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2437340153452686e-05,
+      "loss": 0.009,
+      "step": 2339
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2429667519181586e-05,
+      "loss": 0.0067,
+      "step": 2340
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2421994884910486e-05,
+      "loss": 0.0089,
+      "step": 2341
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2414322250639386e-05,
+      "loss": 0.011,
+      "step": 2342
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2406649616368288e-05,
+      "loss": 0.0034,
+      "step": 2343
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2398976982097186e-05,
+      "loss": 1.0842,
+      "step": 2344
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2391304347826088e-05,
+      "loss": 0.0066,
+      "step": 2345
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2383631713554987e-05,
+      "loss": 0.0203,
+      "step": 2346
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2375959079283887e-05,
+      "loss": 0.0046,
+      "step": 2347
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2368286445012787e-05,
+      "loss": 0.9714,
+      "step": 2348
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2360613810741689e-05,
+      "loss": 0.0066,
+      "step": 2349
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2352941176470587e-05,
+      "loss": 0.0091,
+      "step": 2350
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.2345268542199489e-05,
+      "loss": 0.0066,
+      "step": 2351
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.233759590792839e-05,
+      "loss": 0.0081,
+      "step": 2352
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.2329923273657289e-05,
+      "loss": 0.006,
+      "step": 2353
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.232225063938619e-05,
+      "loss": 0.0066,
+      "step": 2354
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.231457800511509e-05,
+      "loss": 0.0114,
+      "step": 2355
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.230690537084399e-05,
+      "loss": 2.7382,
+      "step": 2356
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.229923273657289e-05,
+      "loss": 0.0089,
+      "step": 2357
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.2291560102301792e-05,
+      "loss": 0.0042,
+      "step": 2358
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.228388746803069e-05,
+      "loss": 2.3698,
+      "step": 2359
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2276214833759591e-05,
+      "loss": 0.0065,
+      "step": 2360
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2268542199488491e-05,
+      "loss": 1.0813,
+      "step": 2361
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2260869565217393e-05,
+      "loss": 0.0066,
+      "step": 2362
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2253196930946291e-05,
+      "loss": 0.0092,
+      "step": 2363
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2245524296675193e-05,
+      "loss": 1.3566,
+      "step": 2364
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2237851662404093e-05,
+      "loss": 0.0092,
+      "step": 2365
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2230179028132993e-05,
+      "loss": 1.0739,
+      "step": 2366
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2222506393861893e-05,
+      "loss": 0.0122,
+      "step": 2367
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2214833759590794e-05,
+      "loss": 0.0096,
+      "step": 2368
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2207161125319692e-05,
+      "loss": 0.0094,
+      "step": 2369
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2199488491048594e-05,
+      "loss": 0.0097,
+      "step": 2370
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2191815856777494e-05,
+      "loss": 0.0071,
+      "step": 2371
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2184143222506394e-05,
+      "loss": 0.0085,
+      "step": 2372
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2176470588235294e-05,
+      "loss": 0.0071,
+      "step": 2373
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2168797953964195e-05,
+      "loss": 0.0074,
+      "step": 2374
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2161125319693094e-05,
+      "loss": 0.0084,
+      "step": 2375
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2153452685421995e-05,
+      "loss": 0.008,
+      "step": 2376
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.9267676767676768,
+      "eval_loss": 0.3489837944507599,
+      "eval_runtime": 58.1837,
+      "eval_samples_per_second": 6.806,
+      "eval_steps_per_second": 1.702,
+      "step": 2376
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2145780051150897e-05,
+      "loss": 0.0065,
+      "step": 2377
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2138107416879795e-05,
+      "loss": 0.0097,
+      "step": 2378
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2130434782608697e-05,
+      "loss": 1.3744,
+      "step": 2379
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2122762148337597e-05,
+      "loss": 0.0043,
+      "step": 2380
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2115089514066496e-05,
+      "loss": 0.0071,
+      "step": 2381
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2107416879795396e-05,
+      "loss": 1.36,
+      "step": 2382
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2099744245524298e-05,
+      "loss": 0.0044,
+      "step": 2383
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2092071611253196e-05,
+      "loss": 0.0075,
+      "step": 2384
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2084398976982098e-05,
+      "loss": 0.0073,
+      "step": 2385
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.2076726342710998e-05,
+      "loss": 0.0099,
+      "step": 2386
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.2069053708439898e-05,
+      "loss": 0.0048,
+      "step": 2387
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.2061381074168798e-05,
+      "loss": 0.0074,
+      "step": 2388
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.20537084398977e-05,
+      "loss": 0.01,
+      "step": 2389
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.2046035805626599e-05,
+      "loss": 0.0062,
+      "step": 2390
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.2038363171355499e-05,
+      "loss": 0.0039,
+      "step": 2391
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.2030690537084399e-05,
+      "loss": 0.0054,
+      "step": 2392
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.20230179028133e-05,
+      "loss": 1.3442,
+      "step": 2393
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.2015345268542199e-05,
+      "loss": 0.0074,
+      "step": 2394
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.20076726342711e-05,
+      "loss": 0.0063,
+      "step": 2395
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.0075,
+      "step": 2396
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.19923273657289e-05,
+      "loss": 0.008,
+      "step": 2397
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.19846547314578e-05,
+      "loss": 0.0053,
+      "step": 2398
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.1976982097186702e-05,
+      "loss": 0.0049,
+      "step": 2399
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.19693094629156e-05,
+      "loss": 0.0086,
+      "step": 2400
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.1961636828644502e-05,
+      "loss": 0.0098,
+      "step": 2401
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1953964194373403e-05,
+      "loss": 0.0062,
+      "step": 2402
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1946291560102301e-05,
+      "loss": 0.0052,
+      "step": 2403
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1938618925831203e-05,
+      "loss": 0.005,
+      "step": 2404
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1930946291560103e-05,
+      "loss": 1.696,
+      "step": 2405
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1923273657289003e-05,
+      "loss": 0.0076,
+      "step": 2406
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1915601023017903e-05,
+      "loss": 1.1231,
+      "step": 2407
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1907928388746804e-05,
+      "loss": 0.0074,
+      "step": 2408
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1900255754475703e-05,
+      "loss": 0.0097,
+      "step": 2409
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1892583120204604e-05,
+      "loss": 0.0102,
+      "step": 2410
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1884910485933504e-05,
+      "loss": 0.005,
+      "step": 2411
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1877237851662404e-05,
+      "loss": 0.0074,
+      "step": 2412
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1869565217391304e-05,
+      "loss": 0.0071,
+      "step": 2413
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1861892583120206e-05,
+      "loss": 0.0093,
+      "step": 2414
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1854219948849106e-05,
+      "loss": 0.007,
+      "step": 2415
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1846547314578005e-05,
+      "loss": 0.0088,
+      "step": 2416
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1838874680306905e-05,
+      "loss": 0.0049,
+      "step": 2417
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1831202046035807e-05,
+      "loss": 1.2268,
+      "step": 2418
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1823529411764705e-05,
+      "loss": 0.0068,
+      "step": 2419
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1815856777493607e-05,
+      "loss": 0.0112,
+      "step": 2420
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1808184143222507e-05,
+      "loss": 0.0046,
+      "step": 2421
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1800511508951407e-05,
+      "loss": 0.0038,
+      "step": 2422
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1792838874680307e-05,
+      "loss": 0.0088,
+      "step": 2423
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1785166240409208e-05,
+      "loss": 0.0039,
+      "step": 2424
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1777493606138106e-05,
+      "loss": 0.0107,
+      "step": 2425
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.1769820971867008e-05,
+      "loss": 0.0066,
+      "step": 2426
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.176214833759591e-05,
+      "loss": 0.0086,
+      "step": 2427
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.1754475703324808e-05,
+      "loss": 0.0064,
+      "step": 2428
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.174680306905371e-05,
+      "loss": 0.0064,
+      "step": 2429
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.173913043478261e-05,
+      "loss": 0.0042,
+      "step": 2430
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.173145780051151e-05,
+      "loss": 0.0063,
+      "step": 2431
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.172378516624041e-05,
+      "loss": 1.0576,
+      "step": 2432
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.171611253196931e-05,
+      "loss": 0.0041,
+      "step": 2433
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.1708439897698209e-05,
+      "loss": 0.0102,
+      "step": 2434
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.170076726342711e-05,
+      "loss": 1.0312,
+      "step": 2435
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.169309462915601e-05,
+      "loss": 0.1209,
+      "step": 2436
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.168542199488491e-05,
+      "loss": 0.004,
+      "step": 2437
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.167774936061381e-05,
+      "loss": 0.006,
+      "step": 2438
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.1670076726342712e-05,
+      "loss": 0.0081,
+      "step": 2439
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.166240409207161e-05,
+      "loss": 1.1118,
+      "step": 2440
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.1654731457800512e-05,
+      "loss": 0.0059,
+      "step": 2441
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1647058823529412e-05,
+      "loss": 0.0101,
+      "step": 2442
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1639386189258313e-05,
+      "loss": 0.0081,
+      "step": 2443
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1631713554987212e-05,
+      "loss": 0.0032,
+      "step": 2444
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1624040920716113e-05,
+      "loss": 0.0059,
+      "step": 2445
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1616368286445013e-05,
+      "loss": 0.0105,
+      "step": 2446
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1608695652173913e-05,
+      "loss": 0.0038,
+      "step": 2447
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1601023017902813e-05,
+      "loss": 0.0104,
+      "step": 2448
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1593350383631715e-05,
+      "loss": 0.0058,
+      "step": 2449
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1585677749360613e-05,
+      "loss": 0.0058,
+      "step": 2450
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1578005115089514e-05,
+      "loss": 0.0064,
+      "step": 2451
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1570332480818416e-05,
+      "loss": 0.0058,
+      "step": 2452
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1562659846547314e-05,
+      "loss": 0.0077,
+      "step": 2453
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1554987212276216e-05,
+      "loss": 0.0528,
+      "step": 2454
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1547314578005116e-05,
+      "loss": 0.0078,
+      "step": 2455
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1539641943734016e-05,
+      "loss": 0.0023,
+      "step": 2456
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1531969309462916e-05,
+      "loss": 0.0019,
+      "step": 2457
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1524296675191817e-05,
+      "loss": 0.0056,
+      "step": 2458
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1516624040920715e-05,
+      "loss": 0.0116,
+      "step": 2459
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1508951406649617e-05,
+      "loss": 0.0037,
+      "step": 2460
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1501278772378517e-05,
+      "loss": 0.0056,
+      "step": 2461
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1493606138107417e-05,
+      "loss": 1.4099,
+      "step": 2462
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1485933503836317e-05,
+      "loss": 0.0055,
+      "step": 2463
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1478260869565218e-05,
+      "loss": 0.0018,
+      "step": 2464
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1470588235294117e-05,
+      "loss": 0.0081,
+      "step": 2465
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.1462915601023018e-05,
+      "loss": 0.0056,
+      "step": 2466
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.1455242966751918e-05,
+      "loss": 0.0048,
+      "step": 2467
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.144757033248082e-05,
+      "loss": 1.352,
+      "step": 2468
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.1439897698209718e-05,
+      "loss": 1.3368,
+      "step": 2469
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.143222506393862e-05,
+      "loss": 0.0075,
+      "step": 2470
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.142455242966752e-05,
+      "loss": 0.0046,
+      "step": 2471
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.141687979539642e-05,
+      "loss": 0.0036,
+      "step": 2472
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.140920716112532e-05,
+      "loss": 0.0055,
+      "step": 2473
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 1.1401534526854221e-05,
+      "loss": 0.0065,
+      "step": 2474
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 1.1393861892583121e-05,
+      "loss": 0.0055,
+      "step": 2475
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 1.138618925831202e-05,
+      "loss": 0.0057,
+      "step": 2476
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.1378516624040922e-05,
+      "loss": 0.0047,
+      "step": 2477
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.137084398976982e-05,
+      "loss": 0.0036,
+      "step": 2478
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.1363171355498722e-05,
+      "loss": 0.0061,
+      "step": 2479
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.1355498721227622e-05,
+      "loss": 0.0072,
+      "step": 2480
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1347826086956522e-05,
+      "loss": 0.0054,
+      "step": 2481
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1340153452685422e-05,
+      "loss": 0.0072,
+      "step": 2482
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1332480818414324e-05,
+      "loss": 0.0041,
+      "step": 2483
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1324808184143222e-05,
+      "loss": 0.0051,
+      "step": 2484
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1317135549872123e-05,
+      "loss": 1.4203,
+      "step": 2485
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1309462915601023e-05,
+      "loss": 0.0088,
+      "step": 2486
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1301790281329923e-05,
+      "loss": 0.0106,
+      "step": 2487
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1294117647058823e-05,
+      "loss": 0.0071,
+      "step": 2488
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1286445012787725e-05,
+      "loss": 0.0053,
+      "step": 2489
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1278772378516623e-05,
+      "loss": 1.1147,
+      "step": 2490
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1271099744245525e-05,
+      "loss": 0.007,
+      "step": 2491
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1263427109974425e-05,
+      "loss": 0.007,
+      "step": 2492
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1255754475703324e-05,
+      "loss": 0.0035,
+      "step": 2493
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1248081841432224e-05,
+      "loss": 0.0052,
+      "step": 2494
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1240409207161126e-05,
+      "loss": 0.0064,
+      "step": 2495
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1232736572890026e-05,
+      "loss": 0.0046,
+      "step": 2496
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1225063938618926e-05,
+      "loss": 0.0052,
+      "step": 2497
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1217391304347826e-05,
+      "loss": 0.0045,
+      "step": 2498
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1209718670076727e-05,
+      "loss": 0.0052,
+      "step": 2499
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1202046035805627e-05,
+      "loss": 0.0035,
+      "step": 2500
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1194373401534527e-05,
+      "loss": 0.0086,
+      "step": 2501
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1186700767263429e-05,
+      "loss": 0.0068,
+      "step": 2502
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1179028132992327e-05,
+      "loss": 0.0051,
+      "step": 2503
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1171355498721229e-05,
+      "loss": 0.0026,
+      "step": 2504
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.1163682864450129e-05,
+      "loss": 0.0034,
+      "step": 2505
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.1156010230179029e-05,
+      "loss": 0.0052,
+      "step": 2506
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.1148337595907928e-05,
+      "loss": 0.0051,
+      "step": 2507
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.114066496163683e-05,
+      "loss": 0.005,
+      "step": 2508
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.1132992327365728e-05,
+      "loss": 0.005,
+      "step": 2509
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.112531969309463e-05,
+      "loss": 0.0043,
+      "step": 2510
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.111764705882353e-05,
+      "loss": 0.0065,
+      "step": 2511
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.110997442455243e-05,
+      "loss": 0.0039,
+      "step": 2512
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.110230179028133e-05,
+      "loss": 1.4148,
+      "step": 2513
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.1094629156010231e-05,
+      "loss": 0.0049,
+      "step": 2514
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.108695652173913e-05,
+      "loss": 0.0058,
+      "step": 2515
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.1079283887468031e-05,
+      "loss": 1.3611,
+      "step": 2516
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.1071611253196931e-05,
+      "loss": 0.0057,
+      "step": 2517
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.1063938618925831e-05,
+      "loss": 0.0049,
+      "step": 2518
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.105626598465473e-05,
+      "loss": 0.0049,
+      "step": 2519
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.1048593350383632e-05,
+      "loss": 0.0026,
+      "step": 2520
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1040920716112532e-05,
+      "loss": 0.0026,
+      "step": 2521
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1033248081841432e-05,
+      "loss": 0.4305,
+      "step": 2522
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1025575447570332e-05,
+      "loss": 0.0069,
+      "step": 2523
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1017902813299234e-05,
+      "loss": 0.005,
+      "step": 2524
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.1010230179028134e-05,
+      "loss": 1.3999,
+      "step": 2525
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.1002557544757034e-05,
+      "loss": 0.1888,
+      "step": 2526
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.0994884910485935e-05,
+      "loss": 0.0049,
+      "step": 2527
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.0987212276214833e-05,
+      "loss": 0.8899,
+      "step": 2528
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0979539641943735e-05,
+      "loss": 1.0806,
+      "step": 2529
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0971867007672635e-05,
+      "loss": 0.0083,
+      "step": 2530
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0964194373401535e-05,
+      "loss": 0.0034,
+      "step": 2531
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0956521739130435e-05,
+      "loss": 0.005,
+      "step": 2532
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0948849104859336e-05,
+      "loss": 0.0051,
+      "step": 2533
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0941176470588235e-05,
+      "loss": 0.0057,
+      "step": 2534
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0933503836317136e-05,
+      "loss": 0.0058,
+      "step": 2535
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0925831202046036e-05,
+      "loss": 0.0065,
+      "step": 2536
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0918158567774936e-05,
+      "loss": 1.1058,
+      "step": 2537
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0910485933503836e-05,
+      "loss": 0.0052,
+      "step": 2538
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0902813299232738e-05,
+      "loss": 0.0082,
+      "step": 2539
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0895140664961636e-05,
+      "loss": 0.005,
+      "step": 2540
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0887468030690537e-05,
+      "loss": 0.005,
+      "step": 2541
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0879795396419437e-05,
+      "loss": 0.005,
+      "step": 2542
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0872122762148337e-05,
+      "loss": 0.0034,
+      "step": 2543
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0864450127877237e-05,
+      "loss": 0.0066,
+      "step": 2544
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0856777493606139e-05,
+      "loss": 0.0052,
+      "step": 2545
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0849104859335037e-05,
+      "loss": 0.0039,
+      "step": 2546
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0841432225063939e-05,
+      "loss": 0.0039,
+      "step": 2547
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0833759590792839e-05,
+      "loss": 0.0064,
+      "step": 2548
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.082608695652174e-05,
+      "loss": 0.0064,
+      "step": 2549
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.081841432225064e-05,
+      "loss": 0.0052,
+      "step": 2550
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.081074168797954e-05,
+      "loss": 0.0082,
+      "step": 2551
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.0803069053708442e-05,
+      "loss": 0.0035,
+      "step": 2552
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.079539641943734e-05,
+      "loss": 0.0064,
+      "step": 2553
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.0787723785166241e-05,
+      "loss": 0.0064,
+      "step": 2554
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.0780051150895141e-05,
+      "loss": 0.1541,
+      "step": 2555
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.0772378516624041e-05,
+      "loss": 2.4749,
+      "step": 2556
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0764705882352941e-05,
+      "loss": 0.0028,
+      "step": 2557
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0757033248081843e-05,
+      "loss": 0.0034,
+      "step": 2558
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0749360613810741e-05,
+      "loss": 0.0061,
+      "step": 2559
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0741687979539643e-05,
+      "loss": 0.0032,
+      "step": 2560
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0734015345268543e-05,
+      "loss": 0.0055,
+      "step": 2561
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0726342710997443e-05,
+      "loss": 0.0046,
+      "step": 2562
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0718670076726342e-05,
+      "loss": 1.185,
+      "step": 2563
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0710997442455244e-05,
+      "loss": 0.0062,
+      "step": 2564
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0703324808184142e-05,
+      "loss": 0.0024,
+      "step": 2565
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0695652173913044e-05,
+      "loss": 0.0032,
+      "step": 2566
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0687979539641944e-05,
+      "loss": 0.0047,
+      "step": 2567
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0680306905370844e-05,
+      "loss": 0.0047,
+      "step": 2568
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0672634271099744e-05,
+      "loss": 0.0047,
+      "step": 2569
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0664961636828645e-05,
+      "loss": 0.0065,
+      "step": 2570
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0657289002557543e-05,
+      "loss": 0.0061,
+      "step": 2571
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0649616368286445e-05,
+      "loss": 0.0048,
+      "step": 2572
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 1.0641943734015345e-05,
+      "loss": 0.0015,
+      "step": 2573
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 1.0634271099744247e-05,
+      "loss": 0.0046,
+      "step": 2574
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 1.0626598465473147e-05,
+      "loss": 0.0033,
+      "step": 2575
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0618925831202046e-05,
+      "loss": 0.0062,
+      "step": 2576
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0611253196930948e-05,
+      "loss": 1.4243,
+      "step": 2577
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0603580562659846e-05,
+      "loss": 0.012,
+      "step": 2578
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0595907928388748e-05,
+      "loss": 2.3898,
+      "step": 2579
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.0588235294117648e-05,
+      "loss": 0.0032,
+      "step": 2580
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.0580562659846548e-05,
+      "loss": 0.0032,
+      "step": 2581
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.0572890025575448e-05,
+      "loss": 0.0061,
+      "step": 2582
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.056521739130435e-05,
+      "loss": 0.0047,
+      "step": 2583
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0557544757033247e-05,
+      "loss": 0.0047,
+      "step": 2584
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0549872122762149e-05,
+      "loss": 0.2081,
+      "step": 2585
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0542199488491049e-05,
+      "loss": 0.0053,
+      "step": 2586
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0534526854219949e-05,
+      "loss": 0.0047,
+      "step": 2587
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.0526854219948849e-05,
+      "loss": 1.3858,
+      "step": 2588
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.051918158567775e-05,
+      "loss": 0.0062,
+      "step": 2589
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.0511508951406649e-05,
+      "loss": 0.0041,
+      "step": 2590
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.050383631713555e-05,
+      "loss": 1.1796,
+      "step": 2591
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.049616368286445e-05,
+      "loss": 1.0994,
+      "step": 2592
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.048849104859335e-05,
+      "loss": 0.8511,
+      "step": 2593
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.048081841432225e-05,
+      "loss": 1.2895,
+      "step": 2594
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.0473145780051152e-05,
+      "loss": 0.0035,
+      "step": 2595
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.046547314578005e-05,
+      "loss": 0.0065,
+      "step": 2596
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.0457800511508952e-05,
+      "loss": 1.239,
+      "step": 2597
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.0450127877237851e-05,
+      "loss": 1.1663,
+      "step": 2598
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.0442455242966751e-05,
+      "loss": 0.0051,
+      "step": 2599
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0434782608695653e-05,
+      "loss": 1.1103,
+      "step": 2600
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0427109974424553e-05,
+      "loss": 1.4103,
+      "step": 2601
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0419437340153454e-05,
+      "loss": 0.0053,
+      "step": 2602
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0411764705882353e-05,
+      "loss": 0.0072,
+      "step": 2603
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0404092071611254e-05,
+      "loss": 0.0083,
+      "step": 2604
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0396419437340154e-05,
+      "loss": 0.0039,
+      "step": 2605
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0388746803069054e-05,
+      "loss": 0.0049,
+      "step": 2606
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0381074168797954e-05,
+      "loss": 0.0098,
+      "step": 2607
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0373401534526856e-05,
+      "loss": 0.0057,
+      "step": 2608
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0365728900255754e-05,
+      "loss": 0.004,
+      "step": 2609
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0358056265984656e-05,
+      "loss": 0.0098,
+      "step": 2610
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0350383631713555e-05,
+      "loss": 0.0059,
+      "step": 2611
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0342710997442455e-05,
+      "loss": 0.0065,
+      "step": 2612
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0335038363171355e-05,
+      "loss": 0.0059,
+      "step": 2613
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0327365728900257e-05,
+      "loss": 0.0112,
+      "step": 2614
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0319693094629155e-05,
+      "loss": 0.0061,
+      "step": 2615
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0312020460358057e-05,
+      "loss": 0.0039,
+      "step": 2616
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0304347826086957e-05,
+      "loss": 0.0083,
+      "step": 2617
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0296675191815857e-05,
+      "loss": 0.0057,
+      "step": 2618
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0289002557544756e-05,
+      "loss": 1.3969,
+      "step": 2619
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0281329923273658e-05,
+      "loss": 0.0065,
+      "step": 2620
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0273657289002556e-05,
+      "loss": 1.0742,
+      "step": 2621
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0265984654731458e-05,
+      "loss": 0.0039,
+      "step": 2622
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0258312020460358e-05,
+      "loss": 0.0065,
+      "step": 2623
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.0250639386189258e-05,
+      "loss": 0.005,
+      "step": 2624
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.024296675191816e-05,
+      "loss": 0.0077,
+      "step": 2625
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.023529411764706e-05,
+      "loss": 1.2389,
+      "step": 2626
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.0227621483375961e-05,
+      "loss": 0.0076,
+      "step": 2627
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.0219948849104859e-05,
+      "loss": 1.034,
+      "step": 2628
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.021227621483376e-05,
+      "loss": 0.0058,
+      "step": 2629
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.020460358056266e-05,
+      "loss": 0.0059,
+      "step": 2630
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.019693094629156e-05,
+      "loss": 0.0079,
+      "step": 2631
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.018925831202046e-05,
+      "loss": 0.0043,
+      "step": 2632
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.0181585677749362e-05,
+      "loss": 0.004,
+      "step": 2633
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.017391304347826e-05,
+      "loss": 0.0075,
+      "step": 2634
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.0166240409207162e-05,
+      "loss": 0.0076,
+      "step": 2635
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0158567774936062e-05,
+      "loss": 0.0102,
+      "step": 2636
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0150895140664962e-05,
+      "loss": 1.1332,
+      "step": 2637
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0143222506393862e-05,
+      "loss": 0.0096,
+      "step": 2638
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0135549872122763e-05,
+      "loss": 0.0092,
+      "step": 2639
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0127877237851662e-05,
+      "loss": 1.3919,
+      "step": 2640
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0120204603580563e-05,
+      "loss": 0.0058,
+      "step": 2641
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0112531969309463e-05,
+      "loss": 1.0521,
+      "step": 2642
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0104859335038363e-05,
+      "loss": 0.0045,
+      "step": 2643
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0097186700767263e-05,
+      "loss": 0.0059,
+      "step": 2644
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0089514066496164e-05,
+      "loss": 0.005,
+      "step": 2645
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0081841432225063e-05,
+      "loss": 0.0075,
+      "step": 2646
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0074168797953964e-05,
+      "loss": 0.0403,
+      "step": 2647
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0066496163682864e-05,
+      "loss": 0.0079,
+      "step": 2648
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0058823529411764e-05,
+      "loss": 0.0062,
+      "step": 2649
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0051150895140666e-05,
+      "loss": 0.0061,
+      "step": 2650
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0043478260869566e-05,
+      "loss": 0.0064,
+      "step": 2651
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0035805626598466e-05,
+      "loss": 0.0077,
+      "step": 2652
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0028132992327366e-05,
+      "loss": 0.0884,
+      "step": 2653
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0020460358056267e-05,
+      "loss": 0.0062,
+      "step": 2654
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0012787723785167e-05,
+      "loss": 1.1006,
+      "step": 2655
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 1.0005115089514067e-05,
+      "loss": 0.0109,
+      "step": 2656
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 9.997442455242967e-06,
+      "loss": 0.0059,
+      "step": 2657
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 9.989769820971869e-06,
+      "loss": 0.0084,
+      "step": 2658
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 9.982097186700767e-06,
+      "loss": 0.0058,
+      "step": 2659
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.974424552429668e-06,
+      "loss": 0.0059,
+      "step": 2660
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.966751918158568e-06,
+      "loss": 0.0049,
+      "step": 2661
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.959079283887468e-06,
+      "loss": 0.0079,
+      "step": 2662
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.951406649616368e-06,
+      "loss": 1.5801,
+      "step": 2663
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.94373401534527e-06,
+      "loss": 0.0059,
+      "step": 2664
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.936061381074168e-06,
+      "loss": 1.2523,
+      "step": 2665
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.92838874680307e-06,
+      "loss": 0.0079,
+      "step": 2666
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.92071611253197e-06,
+      "loss": 0.0099,
+      "step": 2667
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.91304347826087e-06,
+      "loss": 0.0082,
+      "step": 2668
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.90537084398977e-06,
+      "loss": 0.004,
+      "step": 2669
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.897698209718671e-06,
+      "loss": 0.0064,
+      "step": 2670
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.890025575447569e-06,
+      "loss": 0.004,
+      "step": 2671
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 9.88235294117647e-06,
+      "loss": 0.0943,
+      "step": 2672
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 9.87468030690537e-06,
+      "loss": 0.0059,
+      "step": 2673
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 9.86700767263427e-06,
+      "loss": 0.0098,
+      "step": 2674
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.859335038363172e-06,
+      "loss": 0.0065,
+      "step": 2675
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.851662404092072e-06,
+      "loss": 0.0075,
+      "step": 2676
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.843989769820972e-06,
+      "loss": 0.0044,
+      "step": 2677
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.836317135549872e-06,
+      "loss": 0.0063,
+      "step": 2678
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.828644501278774e-06,
+      "loss": 0.0038,
+      "step": 2679
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.820971867007673e-06,
+      "loss": 0.0067,
+      "step": 2680
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.813299232736573e-06,
+      "loss": 1.1148,
+      "step": 2681
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.805626598465473e-06,
+      "loss": 0.0056,
+      "step": 2682
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.797953964194375e-06,
+      "loss": 0.0096,
+      "step": 2683
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.790281329923273e-06,
+      "loss": 0.011,
+      "step": 2684
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.782608695652175e-06,
+      "loss": 0.0038,
+      "step": 2685
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.774936061381075e-06,
+      "loss": 0.0129,
+      "step": 2686
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.767263427109975e-06,
+      "loss": 0.0063,
+      "step": 2687
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.759590792838875e-06,
+      "loss": 0.0561,
+      "step": 2688
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.751918158567776e-06,
+      "loss": 0.0041,
+      "step": 2689
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.744245524296674e-06,
+      "loss": 0.0048,
+      "step": 2690
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.736572890025576e-06,
+      "loss": 0.0072,
+      "step": 2691
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.728900255754476e-06,
+      "loss": 0.007,
+      "step": 2692
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.721227621483376e-06,
+      "loss": 0.0031,
+      "step": 2693
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.713554987212276e-06,
+      "loss": 1.4136,
+      "step": 2694
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.705882352941177e-06,
+      "loss": 0.0081,
+      "step": 2695
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.698209718670076e-06,
+      "loss": 0.0036,
+      "step": 2696
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.690537084398977e-06,
+      "loss": 1.382,
+      "step": 2697
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.682864450127877e-06,
+      "loss": 0.0054,
+      "step": 2698
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.675191815856777e-06,
+      "loss": 0.0064,
+      "step": 2699
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.667519181585679e-06,
+      "loss": 0.0072,
+      "step": 2700
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.659846547314579e-06,
+      "loss": 0.0054,
+      "step": 2701
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.652173913043478e-06,
+      "loss": 0.0108,
+      "step": 2702
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.644501278772378e-06,
+      "loss": 0.0037,
+      "step": 2703
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.63682864450128e-06,
+      "loss": 0.0071,
+      "step": 2704
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.629156010230178e-06,
+      "loss": 0.0062,
+      "step": 2705
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.62148337595908e-06,
+      "loss": 0.0056,
+      "step": 2706
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.61381074168798e-06,
+      "loss": 1.1276,
+      "step": 2707
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.606138107416881e-06,
+      "loss": 0.0045,
+      "step": 2708
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.59846547314578e-06,
+      "loss": 0.0073,
+      "step": 2709
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.590792838874681e-06,
+      "loss": 0.3343,
+      "step": 2710
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.583120204603581e-06,
+      "loss": 0.0038,
+      "step": 2711
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.575447570332481e-06,
+      "loss": 0.0053,
+      "step": 2712
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.567774936061381e-06,
+      "loss": 0.0037,
+      "step": 2713
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.560102301790283e-06,
+      "loss": 0.0036,
+      "step": 2714
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.55242966751918e-06,
+      "loss": 0.0053,
+      "step": 2715
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.544757033248082e-06,
+      "loss": 0.0069,
+      "step": 2716
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.537084398976982e-06,
+      "loss": 1.3084,
+      "step": 2717
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.529411764705882e-06,
+      "loss": 0.0022,
+      "step": 2718
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.521739130434782e-06,
+      "loss": 0.0046,
+      "step": 2719
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.514066496163684e-06,
+      "loss": 1.1433,
+      "step": 2720
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.506393861892582e-06,
+      "loss": 0.0053,
+      "step": 2721
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.498721227621484e-06,
+      "loss": 0.0053,
+      "step": 2722
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.491048593350385e-06,
+      "loss": 0.0057,
+      "step": 2723
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.483375959079283e-06,
+      "loss": 0.0058,
+      "step": 2724
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.475703324808185e-06,
+      "loss": 0.0029,
+      "step": 2725
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.468030690537085e-06,
+      "loss": 0.0061,
+      "step": 2726
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.460358056265985e-06,
+      "loss": 0.0058,
+      "step": 2727
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.452685421994885e-06,
+      "loss": 0.0074,
+      "step": 2728
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.445012787723786e-06,
+      "loss": 0.007,
+      "step": 2729
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.437340153452685e-06,
+      "loss": 1.3508,
+      "step": 2730
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.429667519181586e-06,
+      "loss": 0.0074,
+      "step": 2731
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.421994884910486e-06,
+      "loss": 0.0037,
+      "step": 2732
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.414322250639388e-06,
+      "loss": 0.003,
+      "step": 2733
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.406649616368286e-06,
+      "loss": 0.0038,
+      "step": 2734
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.398976982097188e-06,
+      "loss": 0.0086,
+      "step": 2735
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.391304347826087e-06,
+      "loss": 0.0058,
+      "step": 2736
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.383631713554987e-06,
+      "loss": 0.0021,
+      "step": 2737
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.375959079283887e-06,
+      "loss": 0.0046,
+      "step": 2738
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.368286445012789e-06,
+      "loss": 0.0035,
+      "step": 2739
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.360613810741687e-06,
+      "loss": 0.0054,
+      "step": 2740
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.352941176470589e-06,
+      "loss": 0.0052,
+      "step": 2741
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.345268542199489e-06,
+      "loss": 0.0072,
+      "step": 2742
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.337595907928389e-06,
+      "loss": 0.1932,
+      "step": 2743
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.329923273657289e-06,
+      "loss": 0.006,
+      "step": 2744
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.32225063938619e-06,
+      "loss": 0.0036,
+      "step": 2745
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.314578005115088e-06,
+      "loss": 1.2929,
+      "step": 2746
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.30690537084399e-06,
+      "loss": 0.0028,
+      "step": 2747
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.299232736572892e-06,
+      "loss": 1.2908,
+      "step": 2748
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.29156010230179e-06,
+      "loss": 0.0067,
+      "step": 2749
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.283887468030691e-06,
+      "loss": 0.0043,
+      "step": 2750
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.276214833759591e-06,
+      "loss": 0.0054,
+      "step": 2751
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.268542199488491e-06,
+      "loss": 0.0055,
+      "step": 2752
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.260869565217391e-06,
+      "loss": 1.1614,
+      "step": 2753
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.253196930946293e-06,
+      "loss": 0.0071,
+      "step": 2754
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.245524296675191e-06,
+      "loss": 0.0035,
+      "step": 2755
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.237851662404093e-06,
+      "loss": 0.0054,
+      "step": 2756
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.230179028132993e-06,
+      "loss": 0.0096,
+      "step": 2757
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.222506393861894e-06,
+      "loss": 0.0035,
+      "step": 2758
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.214833759590792e-06,
+      "loss": 0.0051,
+      "step": 2759
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.207161125319694e-06,
+      "loss": 0.0069,
+      "step": 2760
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.199488491048594e-06,
+      "loss": 0.0018,
+      "step": 2761
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.191815856777494e-06,
+      "loss": 0.0026,
+      "step": 2762
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.184143222506394e-06,
+      "loss": 0.0054,
+      "step": 2763
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.176470588235295e-06,
+      "loss": 0.0085,
+      "step": 2764
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.168797953964194e-06,
+      "loss": 0.008,
+      "step": 2765
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.161125319693095e-06,
+      "loss": 1.3292,
+      "step": 2766
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.153452685421995e-06,
+      "loss": 0.0085,
+      "step": 2767
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.145780051150895e-06,
+      "loss": 0.0034,
+      "step": 2768
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.138107416879795e-06,
+      "loss": 0.0051,
+      "step": 2769
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.130434782608697e-06,
+      "loss": 0.0054,
+      "step": 2770
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 9.122762148337595e-06,
+      "loss": 0.0054,
+      "step": 2771
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 9.115089514066496e-06,
+      "loss": 0.0083,
+      "step": 2772
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.9217171717171717,
+      "eval_loss": 0.37921515107154846,
+      "eval_runtime": 57.416,
+      "eval_samples_per_second": 6.897,
+      "eval_steps_per_second": 1.724,
+      "step": 2772
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 9.107416879795398e-06,
+      "loss": 1.4643,
+      "step": 2773
+    },
+    {
+      "epoch": 7.01,
+      "learning_rate": 9.099744245524296e-06,
+      "loss": 0.0067,
+      "step": 2774
+    },
+    {
+      "epoch": 7.01,
+      "learning_rate": 9.092071611253198e-06,
+      "loss": 1.362,
+      "step": 2775
+    },
+    {
+      "epoch": 7.01,
+      "learning_rate": 9.084398976982098e-06,
+      "loss": 0.0035,
+      "step": 2776
+    },
+    {
+      "epoch": 7.01,
+      "learning_rate": 9.076726342710998e-06,
+      "loss": 0.0043,
+      "step": 2777
+    },
+    {
+      "epoch": 7.02,
+      "learning_rate": 9.069053708439898e-06,
+      "loss": 0.005,
+      "step": 2778
+    },
+    {
+      "epoch": 7.02,
+      "learning_rate": 9.0613810741688e-06,
+      "loss": 0.0049,
+      "step": 2779
+    },
+    {
+      "epoch": 7.02,
+      "learning_rate": 9.053708439897697e-06,
+      "loss": 0.007,
+      "step": 2780
+    },
+    {
+      "epoch": 7.02,
+      "learning_rate": 9.046035805626599e-06,
+      "loss": 0.0038,
+      "step": 2781
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 9.038363171355499e-06,
+      "loss": 0.7287,
+      "step": 2782
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 9.030690537084399e-06,
+      "loss": 0.0089,
+      "step": 2783
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 9.023017902813299e-06,
+      "loss": 0.0067,
+      "step": 2784
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 9.0153452685422e-06,
+      "loss": 0.0065,
+      "step": 2785
+    },
+    {
+      "epoch": 7.04,
+      "learning_rate": 9.0076726342711e-06,
+      "loss": 0.0066,
+      "step": 2786
+    },
+    {
+      "epoch": 7.04,
+      "learning_rate": 9e-06,
+      "loss": 1.341,
+      "step": 2787
+    },
+    {
+      "epoch": 7.04,
+      "learning_rate": 8.9923273657289e-06,
+      "loss": 0.0039,
+      "step": 2788
+    },
+    {
+      "epoch": 7.04,
+      "learning_rate": 8.984654731457802e-06,
+      "loss": 0.0053,
+      "step": 2789
+    },
+    {
+      "epoch": 7.05,
+      "learning_rate": 8.9769820971867e-06,
+      "loss": 0.0058,
+      "step": 2790
+    },
+    {
+      "epoch": 7.05,
+      "learning_rate": 8.969309462915602e-06,
+      "loss": 1.0835,
+      "step": 2791
+    },
+    {
+      "epoch": 7.05,
+      "learning_rate": 8.961636828644502e-06,
+      "loss": 0.0059,
+      "step": 2792
+    },
+    {
+      "epoch": 7.05,
+      "learning_rate": 8.953964194373401e-06,
+      "loss": 0.0067,
+      "step": 2793
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 8.946291560102301e-06,
+      "loss": 0.0059,
+      "step": 2794
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 8.938618925831203e-06,
+      "loss": 0.0031,
+      "step": 2795
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 8.930946291560101e-06,
+      "loss": 0.005,
+      "step": 2796
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 8.923273657289003e-06,
+      "loss": 0.008,
+      "step": 2797
+    },
+    {
+      "epoch": 7.07,
+      "learning_rate": 8.915601023017904e-06,
+      "loss": 0.0052,
+      "step": 2798
+    },
+    {
+      "epoch": 7.07,
+      "learning_rate": 8.907928388746803e-06,
+      "loss": 0.0634,
+      "step": 2799
+    },
+    {
+      "epoch": 7.07,
+      "learning_rate": 8.900255754475704e-06,
+      "loss": 0.0053,
+      "step": 2800
+    },
+    {
+      "epoch": 7.07,
+      "learning_rate": 8.892583120204604e-06,
+      "loss": 0.0054,
+      "step": 2801
+    },
+    {
+      "epoch": 7.08,
+      "learning_rate": 8.884910485933504e-06,
+      "loss": 0.0066,
+      "step": 2802
+    },
+    {
+      "epoch": 7.08,
+      "learning_rate": 8.877237851662404e-06,
+      "loss": 0.0067,
+      "step": 2803
+    },
+    {
+      "epoch": 7.08,
+      "learning_rate": 8.869565217391306e-06,
+      "loss": 0.0488,
+      "step": 2804
+    },
+    {
+      "epoch": 7.08,
+      "learning_rate": 8.861892583120204e-06,
+      "loss": 0.0048,
+      "step": 2805
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 8.854219948849105e-06,
+      "loss": 0.0065,
+      "step": 2806
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 8.846547314578005e-06,
+      "loss": 0.0089,
+      "step": 2807
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 8.838874680306905e-06,
+      "loss": 0.0052,
+      "step": 2808
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 8.831202046035805e-06,
+      "loss": 0.006,
+      "step": 2809
+    },
+    {
+      "epoch": 7.1,
+      "learning_rate": 8.823529411764707e-06,
+      "loss": 0.0032,
+      "step": 2810
+    },
+    {
+      "epoch": 7.1,
+      "learning_rate": 8.815856777493607e-06,
+      "loss": 0.0036,
+      "step": 2811
+    },
+    {
+      "epoch": 7.1,
+      "learning_rate": 8.808184143222507e-06,
+      "loss": 0.0032,
+      "step": 2812
+    },
+    {
+      "epoch": 7.1,
+      "learning_rate": 8.800511508951407e-06,
+      "loss": 0.0035,
+      "step": 2813
+    },
+    {
+      "epoch": 7.11,
+      "learning_rate": 8.792838874680308e-06,
+      "loss": 1.4438,
+      "step": 2814
+    },
+    {
+      "epoch": 7.11,
+      "learning_rate": 8.785166240409206e-06,
+      "loss": 0.0048,
+      "step": 2815
+    },
+    {
+      "epoch": 7.11,
+      "learning_rate": 8.777493606138108e-06,
+      "loss": 0.0052,
+      "step": 2816
+    },
+    {
+      "epoch": 7.11,
+      "learning_rate": 8.769820971867008e-06,
+      "loss": 0.0038,
+      "step": 2817
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 8.762148337595908e-06,
+      "loss": 0.7281,
+      "step": 2818
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 8.754475703324808e-06,
+      "loss": 1.3888,
+      "step": 2819
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 8.74680306905371e-06,
+      "loss": 0.0016,
+      "step": 2820
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 8.739130434782608e-06,
+      "loss": 0.0065,
+      "step": 2821
+    },
+    {
+      "epoch": 7.13,
+      "learning_rate": 8.73145780051151e-06,
+      "loss": 1.1887,
+      "step": 2822
+    },
+    {
+      "epoch": 7.13,
+      "learning_rate": 8.72378516624041e-06,
+      "loss": 0.0069,
+      "step": 2823
+    },
+    {
+      "epoch": 7.13,
+      "learning_rate": 8.716112531969309e-06,
+      "loss": 0.0034,
+      "step": 2824
+    },
+    {
+      "epoch": 7.13,
+      "learning_rate": 8.70843989769821e-06,
+      "loss": 1.1691,
+      "step": 2825
+    },
+    {
+      "epoch": 7.14,
+      "learning_rate": 8.70076726342711e-06,
+      "loss": 0.0033,
+      "step": 2826
+    },
+    {
+      "epoch": 7.14,
+      "learning_rate": 8.69309462915601e-06,
+      "loss": 0.0088,
+      "step": 2827
+    },
+    {
+      "epoch": 7.14,
+      "learning_rate": 8.68542199488491e-06,
+      "loss": 0.0071,
+      "step": 2828
+    },
+    {
+      "epoch": 7.14,
+      "learning_rate": 8.677749360613812e-06,
+      "loss": 0.0023,
+      "step": 2829
+    },
+    {
+      "epoch": 7.15,
+      "learning_rate": 8.67007672634271e-06,
+      "loss": 0.0036,
+      "step": 2830
+    },
+    {
+      "epoch": 7.15,
+      "learning_rate": 8.662404092071612e-06,
+      "loss": 0.0088,
+      "step": 2831
+    },
+    {
+      "epoch": 7.15,
+      "learning_rate": 8.654731457800512e-06,
+      "loss": 1.1488,
+      "step": 2832
+    },
+    {
+      "epoch": 7.15,
+      "learning_rate": 8.647058823529412e-06,
+      "loss": 0.0067,
+      "step": 2833
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 8.639386189258312e-06,
+      "loss": 0.0031,
+      "step": 2834
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 8.631713554987213e-06,
+      "loss": 0.0036,
+      "step": 2835
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 8.624040920716111e-06,
+      "loss": 0.0058,
+      "step": 2836
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 8.616368286445013e-06,
+      "loss": 0.03,
+      "step": 2837
+    },
+    {
+      "epoch": 7.17,
+      "learning_rate": 8.608695652173913e-06,
+      "loss": 0.0046,
+      "step": 2838
+    },
+    {
+      "epoch": 7.17,
+      "learning_rate": 8.601023017902815e-06,
+      "loss": 0.0093,
+      "step": 2839
+    },
+    {
+      "epoch": 7.17,
+      "learning_rate": 8.593350383631713e-06,
+      "loss": 0.0054,
+      "step": 2840
+    },
+    {
+      "epoch": 7.17,
+      "learning_rate": 8.585677749360614e-06,
+      "loss": 0.0051,
+      "step": 2841
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 8.578005115089514e-06,
+      "loss": 0.0036,
+      "step": 2842
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 8.570332480818414e-06,
+      "loss": 0.0016,
+      "step": 2843
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 8.562659846547314e-06,
+      "loss": 0.0057,
+      "step": 2844
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 8.554987212276216e-06,
+      "loss": 0.0041,
+      "step": 2845
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 8.547314578005114e-06,
+      "loss": 0.0035,
+      "step": 2846
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 8.539641943734016e-06,
+      "loss": 0.0084,
+      "step": 2847
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 8.531969309462917e-06,
+      "loss": 0.0065,
+      "step": 2848
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 8.524296675191815e-06,
+      "loss": 0.0066,
+      "step": 2849
+    },
+    {
+      "epoch": 7.2,
+      "learning_rate": 8.516624040920717e-06,
+      "loss": 0.0035,
+      "step": 2850
+    },
+    {
+      "epoch": 7.2,
+      "learning_rate": 8.508951406649617e-06,
+      "loss": 0.0052,
+      "step": 2851
+    },
+    {
+      "epoch": 7.2,
+      "learning_rate": 8.501278772378517e-06,
+      "loss": 0.0065,
+      "step": 2852
+    },
+    {
+      "epoch": 7.2,
+      "learning_rate": 8.493606138107417e-06,
+      "loss": 0.0064,
+      "step": 2853
+    },
+    {
+      "epoch": 7.21,
+      "learning_rate": 8.485933503836318e-06,
+      "loss": 0.0154,
+      "step": 2854
+    },
+    {
+      "epoch": 7.21,
+      "learning_rate": 8.478260869565217e-06,
+      "loss": 0.0114,
+      "step": 2855
+    },
+    {
+      "epoch": 7.21,
+      "learning_rate": 8.470588235294118e-06,
+      "loss": 0.0063,
+      "step": 2856
+    },
+    {
+      "epoch": 7.21,
+      "learning_rate": 8.462915601023018e-06,
+      "loss": 0.0097,
+      "step": 2857
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 8.455242966751918e-06,
+      "loss": 0.0051,
+      "step": 2858
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 8.447570332480818e-06,
+      "loss": 0.0081,
+      "step": 2859
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 8.43989769820972e-06,
+      "loss": 0.0046,
+      "step": 2860
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 8.432225063938618e-06,
+      "loss": 0.0034,
+      "step": 2861
+    },
+    {
+      "epoch": 7.23,
+      "learning_rate": 8.42455242966752e-06,
+      "loss": 0.0053,
+      "step": 2862
+    },
+    {
+      "epoch": 7.23,
+      "learning_rate": 8.41687979539642e-06,
+      "loss": 0.0071,
+      "step": 2863
+    },
+    {
+      "epoch": 7.23,
+      "learning_rate": 8.409207161125321e-06,
+      "loss": 0.0062,
+      "step": 2864
+    },
+    {
+      "epoch": 7.23,
+      "learning_rate": 8.40153452685422e-06,
+      "loss": 0.0106,
+      "step": 2865
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 8.39386189258312e-06,
+      "loss": 0.0038,
+      "step": 2866
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 8.38618925831202e-06,
+      "loss": 0.003,
+      "step": 2867
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 8.37851662404092e-06,
+      "loss": 0.0044,
+      "step": 2868
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 8.37084398976982e-06,
+      "loss": 0.0061,
+      "step": 2869
+    },
+    {
+      "epoch": 7.25,
+      "learning_rate": 8.363171355498722e-06,
+      "loss": 0.0061,
+      "step": 2870
+    },
+    {
+      "epoch": 7.25,
+      "learning_rate": 8.355498721227622e-06,
+      "loss": 0.0071,
+      "step": 2871
+    },
+    {
+      "epoch": 7.25,
+      "learning_rate": 8.347826086956522e-06,
+      "loss": 0.0044,
+      "step": 2872
+    },
+    {
+      "epoch": 7.26,
+      "learning_rate": 8.340153452685424e-06,
+      "loss": 0.0429,
+      "step": 2873
+    },
+    {
+      "epoch": 7.26,
+      "learning_rate": 8.332480818414322e-06,
+      "loss": 0.002,
+      "step": 2874
+    },
+    {
+      "epoch": 7.26,
+      "learning_rate": 8.324808184143223e-06,
+      "loss": 0.0056,
+      "step": 2875
+    },
+    {
+      "epoch": 7.26,
+      "learning_rate": 8.317135549872123e-06,
+      "loss": 0.0161,
+      "step": 2876
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 8.309462915601023e-06,
+      "loss": 0.0049,
+      "step": 2877
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 8.301790281329923e-06,
+      "loss": 0.0082,
+      "step": 2878
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 8.294117647058825e-06,
+      "loss": 0.0055,
+      "step": 2879
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 8.286445012787723e-06,
+      "loss": 0.0029,
+      "step": 2880
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 8.278772378516625e-06,
+      "loss": 0.0067,
+      "step": 2881
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 8.271099744245525e-06,
+      "loss": 0.0046,
+      "step": 2882
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 8.263427109974425e-06,
+      "loss": 0.0029,
+      "step": 2883
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 8.255754475703324e-06,
+      "loss": 1.4067,
+      "step": 2884
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 8.248081841432226e-06,
+      "loss": 0.0056,
+      "step": 2885
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 8.240409207161124e-06,
+      "loss": 0.007,
+      "step": 2886
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 8.232736572890026e-06,
+      "loss": 0.0044,
+      "step": 2887
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 8.225063938618926e-06,
+      "loss": 0.2182,
+      "step": 2888
+    },
+    {
+      "epoch": 7.3,
+      "learning_rate": 8.217391304347826e-06,
+      "loss": 0.0028,
+      "step": 2889
+    },
+    {
+      "epoch": 7.3,
+      "learning_rate": 8.209718670076726e-06,
+      "loss": 0.0028,
+      "step": 2890
+    },
+    {
+      "epoch": 7.3,
+      "learning_rate": 8.202046035805627e-06,
+      "loss": 0.0085,
+      "step": 2891
+    },
+    {
+      "epoch": 7.3,
+      "learning_rate": 8.194373401534527e-06,
+      "loss": 0.0059,
+      "step": 2892
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 8.186700767263427e-06,
+      "loss": 0.004,
+      "step": 2893
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 8.179028132992327e-06,
+      "loss": 0.0057,
+      "step": 2894
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 8.171355498721229e-06,
+      "loss": 0.0052,
+      "step": 2895
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 8.163682864450129e-06,
+      "loss": 0.0041,
+      "step": 2896
+    },
+    {
+      "epoch": 7.32,
+      "learning_rate": 8.156010230179028e-06,
+      "loss": 0.0042,
+      "step": 2897
+    },
+    {
+      "epoch": 7.32,
+      "learning_rate": 8.14833759590793e-06,
+      "loss": 0.0066,
+      "step": 2898
+    },
+    {
+      "epoch": 7.32,
+      "learning_rate": 8.140664961636828e-06,
+      "loss": 0.0042,
+      "step": 2899
+    },
+    {
+      "epoch": 7.32,
+      "learning_rate": 8.13299232736573e-06,
+      "loss": 0.0042,
+      "step": 2900
+    },
+    {
+      "epoch": 7.33,
+      "learning_rate": 8.12531969309463e-06,
+      "loss": 0.0019,
+      "step": 2901
+    },
+    {
+      "epoch": 7.33,
+      "learning_rate": 8.11764705882353e-06,
+      "loss": 0.0054,
+      "step": 2902
+    },
+    {
+      "epoch": 7.33,
+      "learning_rate": 8.10997442455243e-06,
+      "loss": 0.0039,
+      "step": 2903
+    },
+    {
+      "epoch": 7.33,
+      "learning_rate": 8.102301790281331e-06,
+      "loss": 1.1882,
+      "step": 2904
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 8.09462915601023e-06,
+      "loss": 0.0034,
+      "step": 2905
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 8.086956521739131e-06,
+      "loss": 0.0066,
+      "step": 2906
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 8.079283887468031e-06,
+      "loss": 0.0041,
+      "step": 2907
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 8.071611253196931e-06,
+      "loss": 0.0039,
+      "step": 2908
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 8.06393861892583e-06,
+      "loss": 0.0062,
+      "step": 2909
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 8.056265984654732e-06,
+      "loss": 0.0053,
+      "step": 2910
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 8.04859335038363e-06,
+      "loss": 0.0041,
+      "step": 2911
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 8.040920716112532e-06,
+      "loss": 0.059,
+      "step": 2912
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 8.033248081841432e-06,
+      "loss": 0.0027,
+      "step": 2913
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 8.025575447570332e-06,
+      "loss": 0.005,
+      "step": 2914
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 8.017902813299232e-06,
+      "loss": 0.0061,
+      "step": 2915
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 8.010230179028134e-06,
+      "loss": 0.0063,
+      "step": 2916
+    },
+    {
+      "epoch": 7.37,
+      "learning_rate": 8.002557544757034e-06,
+      "loss": 0.0026,
+      "step": 2917
+    },
+    {
+      "epoch": 7.37,
+      "learning_rate": 7.994884910485933e-06,
+      "loss": 0.0037,
+      "step": 2918
+    },
+    {
+      "epoch": 7.37,
+      "learning_rate": 7.987212276214833e-06,
+      "loss": 1.2357,
+      "step": 2919
+    },
+    {
+      "epoch": 7.37,
+      "learning_rate": 7.979539641943735e-06,
+      "loss": 0.0037,
+      "step": 2920
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 7.971867007672635e-06,
+      "loss": 0.0031,
+      "step": 2921
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 7.964194373401535e-06,
+      "loss": 0.0026,
+      "step": 2922
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 7.956521739130436e-06,
+      "loss": 0.006,
+      "step": 2923
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 7.948849104859335e-06,
+      "loss": 0.0032,
+      "step": 2924
+    },
+    {
+      "epoch": 7.39,
+      "learning_rate": 7.941176470588236e-06,
+      "loss": 0.0049,
+      "step": 2925
+    },
+    {
+      "epoch": 7.39,
+      "learning_rate": 7.933503836317136e-06,
+      "loss": 0.0053,
+      "step": 2926
+    },
+    {
+      "epoch": 7.39,
+      "learning_rate": 7.925831202046036e-06,
+      "loss": 0.0037,
+      "step": 2927
+    },
+    {
+      "epoch": 7.39,
+      "learning_rate": 7.918158567774936e-06,
+      "loss": 0.0049,
+      "step": 2928
+    },
+    {
+      "epoch": 7.4,
+      "learning_rate": 7.910485933503838e-06,
+      "loss": 0.0048,
+      "step": 2929
+    },
+    {
+      "epoch": 7.4,
+      "learning_rate": 7.902813299232736e-06,
+      "loss": 0.0059,
+      "step": 2930
+    },
+    {
+      "epoch": 7.4,
+      "learning_rate": 7.895140664961637e-06,
+      "loss": 0.0037,
+      "step": 2931
+    },
+    {
+      "epoch": 7.4,
+      "learning_rate": 7.887468030690537e-06,
+      "loss": 0.0021,
+      "step": 2932
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 7.879795396419437e-06,
+      "loss": 0.0037,
+      "step": 2933
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 7.872122762148337e-06,
+      "loss": 0.0033,
+      "step": 2934
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 7.864450127877239e-06,
+      "loss": 0.0048,
+      "step": 2935
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 7.856777493606137e-06,
+      "loss": 0.0025,
+      "step": 2936
+    },
+    {
+      "epoch": 7.42,
+      "learning_rate": 7.849104859335039e-06,
+      "loss": 0.0025,
+      "step": 2937
+    },
+    {
+      "epoch": 7.42,
+      "learning_rate": 7.841432225063939e-06,
+      "loss": 0.0036,
+      "step": 2938
+    },
+    {
+      "epoch": 7.42,
+      "learning_rate": 7.833759590792839e-06,
+      "loss": 0.0025,
+      "step": 2939
+    },
+    {
+      "epoch": 7.42,
+      "learning_rate": 7.826086956521738e-06,
+      "loss": 1.0675,
+      "step": 2940
+    },
+    {
+      "epoch": 7.43,
+      "learning_rate": 7.81841432225064e-06,
+      "loss": 0.003,
+      "step": 2941
+    },
+    {
+      "epoch": 7.43,
+      "learning_rate": 7.810741687979538e-06,
+      "loss": 0.0032,
+      "step": 2942
+    },
+    {
+      "epoch": 7.43,
+      "learning_rate": 7.80306905370844e-06,
+      "loss": 0.0036,
+      "step": 2943
+    },
+    {
+      "epoch": 7.43,
+      "learning_rate": 7.79539641943734e-06,
+      "loss": 0.0046,
+      "step": 2944
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 7.787723785166241e-06,
+      "loss": 0.0036,
+      "step": 2945
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 7.780051150895141e-06,
+      "loss": 0.0036,
+      "step": 2946
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 7.772378516624041e-06,
+      "loss": 0.0025,
+      "step": 2947
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 7.764705882352943e-06,
+      "loss": 0.0046,
+      "step": 2948
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 7.757033248081841e-06,
+      "loss": 0.0025,
+      "step": 2949
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 7.749360613810743e-06,
+      "loss": 0.0031,
+      "step": 2950
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 7.741687979539643e-06,
+      "loss": 0.0024,
+      "step": 2951
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 7.734015345268543e-06,
+      "loss": 0.003,
+      "step": 2952
+    },
+    {
+      "epoch": 7.46,
+      "learning_rate": 7.726342710997442e-06,
+      "loss": 0.0024,
+      "step": 2953
+    },
+    {
+      "epoch": 7.46,
+      "learning_rate": 7.718670076726344e-06,
+      "loss": 0.0028,
+      "step": 2954
+    },
+    {
+      "epoch": 7.46,
+      "learning_rate": 7.710997442455242e-06,
+      "loss": 0.0046,
+      "step": 2955
+    },
+    {
+      "epoch": 7.46,
+      "learning_rate": 7.703324808184144e-06,
+      "loss": 0.0017,
+      "step": 2956
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 7.695652173913044e-06,
+      "loss": 1.2377,
+      "step": 2957
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 7.687979539641944e-06,
+      "loss": 0.0034,
+      "step": 2958
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 7.680306905370844e-06,
+      "loss": 2.3609,
+      "step": 2959
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 7.672634271099745e-06,
+      "loss": 0.0045,
+      "step": 2960
+    },
+    {
+      "epoch": 7.48,
+      "learning_rate": 7.664961636828643e-06,
+      "loss": 0.0036,
+      "step": 2961
+    },
+    {
+      "epoch": 7.48,
+      "learning_rate": 7.657289002557545e-06,
+      "loss": 0.0056,
+      "step": 2962
+    },
+    {
+      "epoch": 7.48,
+      "learning_rate": 7.649616368286445e-06,
+      "loss": 0.0024,
+      "step": 2963
+    },
+    {
+      "epoch": 7.48,
+      "learning_rate": 7.641943734015345e-06,
+      "loss": 0.0036,
+      "step": 2964
+    },
+    {
+      "epoch": 7.49,
+      "learning_rate": 7.634271099744245e-06,
+      "loss": 0.0034,
+      "step": 2965
+    },
+    {
+      "epoch": 7.49,
+      "learning_rate": 7.6265984654731465e-06,
+      "loss": 0.0034,
+      "step": 2966
+    },
+    {
+      "epoch": 7.49,
+      "learning_rate": 7.6189258312020455e-06,
+      "loss": 0.0034,
+      "step": 2967
+    },
+    {
+      "epoch": 7.49,
+      "learning_rate": 7.611253196930946e-06,
+      "loss": 0.0024,
+      "step": 2968
+    },
+    {
+      "epoch": 7.5,
+      "learning_rate": 7.603580562659847e-06,
+      "loss": 0.0034,
+      "step": 2969
+    },
+    {
+      "epoch": 7.5,
+      "learning_rate": 7.595907928388748e-06,
+      "loss": 0.0041,
+      "step": 2970
+    },
+    {
+      "epoch": 7.5,
+      "learning_rate": 7.588235294117647e-06,
+      "loss": 0.0057,
+      "step": 2971
+    },
+    {
+      "epoch": 7.51,
+      "learning_rate": 7.580562659846548e-06,
+      "loss": 1.2636,
+      "step": 2972
+    },
+    {
+      "epoch": 7.51,
+      "learning_rate": 7.5728900255754484e-06,
+      "loss": 0.0044,
+      "step": 2973
+    },
+    {
+      "epoch": 7.51,
+      "learning_rate": 7.5652173913043475e-06,
+      "loss": 0.0034,
+      "step": 2974
+    },
+    {
+      "epoch": 7.51,
+      "learning_rate": 7.557544757033248e-06,
+      "loss": 0.0035,
+      "step": 2975
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 7.549872122762149e-06,
+      "loss": 0.0034,
+      "step": 2976
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 7.542199488491048e-06,
+      "loss": 0.0045,
+      "step": 2977
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 7.534526854219949e-06,
+      "loss": 0.0024,
+      "step": 2978
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 7.52685421994885e-06,
+      "loss": 0.0038,
+      "step": 2979
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 7.519181585677749e-06,
+      "loss": 0.0034,
+      "step": 2980
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 7.5115089514066495e-06,
+      "loss": 0.0029,
+      "step": 2981
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 7.50383631713555e-06,
+      "loss": 0.0034,
+      "step": 2982
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 7.496163682864451e-06,
+      "loss": 0.0034,
+      "step": 2983
+    },
+    {
+      "epoch": 7.54,
+      "learning_rate": 7.488491048593351e-06,
+      "loss": 0.0056,
+      "step": 2984
+    },
+    {
+      "epoch": 7.54,
+      "learning_rate": 7.480818414322251e-06,
+      "loss": 0.0035,
+      "step": 2985
+    },
+    {
+      "epoch": 7.54,
+      "learning_rate": 7.473145780051152e-06,
+      "loss": 0.0045,
+      "step": 2986
+    },
+    {
+      "epoch": 7.54,
+      "learning_rate": 7.4654731457800515e-06,
+      "loss": 0.0034,
+      "step": 2987
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 7.4578005115089514e-06,
+      "loss": 1.4972,
+      "step": 2988
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 7.450127877237852e-06,
+      "loss": 0.0034,
+      "step": 2989
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 7.442455242966752e-06,
+      "loss": 0.0045,
+      "step": 2990
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 7.434782608695653e-06,
+      "loss": 0.0026,
+      "step": 2991
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 7.427109974424553e-06,
+      "loss": 0.0051,
+      "step": 2992
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 7.419437340153453e-06,
+      "loss": 0.0028,
+      "step": 2993
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 7.4117647058823535e-06,
+      "loss": 0.0023,
+      "step": 2994
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 7.404092071611253e-06,
+      "loss": 0.0034,
+      "step": 2995
+    },
+    {
+      "epoch": 7.57,
+      "learning_rate": 7.396419437340153e-06,
+      "loss": 0.0023,
+      "step": 2996
+    },
+    {
+      "epoch": 7.57,
+      "learning_rate": 7.388746803069054e-06,
+      "loss": 0.0045,
+      "step": 2997
+    },
+    {
+      "epoch": 7.57,
+      "learning_rate": 7.381074168797954e-06,
+      "loss": 0.0224,
+      "step": 2998
+    },
+    {
+      "epoch": 7.57,
+      "learning_rate": 7.373401534526854e-06,
+      "loss": 1.444,
+      "step": 2999
+    },
+    {
+      "epoch": 7.58,
+      "learning_rate": 7.365728900255755e-06,
+      "loss": 0.0016,
+      "step": 3000
+    },
+    {
+      "epoch": 7.58,
+      "learning_rate": 7.358056265984655e-06,
+      "loss": 0.0034,
+      "step": 3001
+    },
+    {
+      "epoch": 7.58,
+      "learning_rate": 7.3503836317135545e-06,
+      "loss": 0.0051,
+      "step": 3002
+    },
+    {
+      "epoch": 7.58,
+      "learning_rate": 7.342710997442455e-06,
+      "loss": 0.0035,
+      "step": 3003
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 7.335038363171355e-06,
+      "loss": 0.0038,
+      "step": 3004
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 7.327365728900256e-06,
+      "loss": 0.0034,
+      "step": 3005
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 7.319693094629156e-06,
+      "loss": 0.0034,
+      "step": 3006
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 7.312020460358056e-06,
+      "loss": 0.0034,
+      "step": 3007
+    },
+    {
+      "epoch": 7.6,
+      "learning_rate": 7.304347826086957e-06,
+      "loss": 0.0037,
+      "step": 3008
+    },
+    {
+      "epoch": 7.6,
+      "learning_rate": 7.296675191815857e-06,
+      "loss": 0.0034,
+      "step": 3009
+    },
+    {
+      "epoch": 7.6,
+      "learning_rate": 7.289002557544757e-06,
+      "loss": 1.2445,
+      "step": 3010
+    },
+    {
+      "epoch": 7.6,
+      "learning_rate": 7.281329923273658e-06,
+      "loss": 0.0034,
+      "step": 3011
+    },
+    {
+      "epoch": 7.61,
+      "learning_rate": 7.273657289002558e-06,
+      "loss": 0.0034,
+      "step": 3012
+    },
+    {
+      "epoch": 7.61,
+      "learning_rate": 7.265984654731458e-06,
+      "loss": 0.0019,
+      "step": 3013
+    },
+    {
+      "epoch": 7.61,
+      "learning_rate": 7.258312020460359e-06,
+      "loss": 1.5106,
+      "step": 3014
+    },
+    {
+      "epoch": 7.61,
+      "learning_rate": 7.2506393861892585e-06,
+      "loss": 1.5062,
+      "step": 3015
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 7.242966751918159e-06,
+      "loss": 0.0035,
+      "step": 3016
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 7.235294117647059e-06,
+      "loss": 0.0027,
+      "step": 3017
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 7.227621483375959e-06,
+      "loss": 0.0025,
+      "step": 3018
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 7.21994884910486e-06,
+      "loss": 0.0035,
+      "step": 3019
+    },
+    {
+      "epoch": 7.63,
+      "learning_rate": 7.21227621483376e-06,
+      "loss": 0.0046,
+      "step": 3020
+    },
+    {
+      "epoch": 7.63,
+      "learning_rate": 7.20460358056266e-06,
+      "loss": 1.2443,
+      "step": 3021
+    },
+    {
+      "epoch": 7.63,
+      "learning_rate": 7.1969309462915605e-06,
+      "loss": 0.004,
+      "step": 3022
+    },
+    {
+      "epoch": 7.63,
+      "learning_rate": 7.18925831202046e-06,
+      "loss": 0.0028,
+      "step": 3023
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 7.18158567774936e-06,
+      "loss": 0.0059,
+      "step": 3024
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 7.173913043478261e-06,
+      "loss": 0.0062,
+      "step": 3025
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 7.166240409207161e-06,
+      "loss": 0.0017,
+      "step": 3026
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 7.158567774936061e-06,
+      "loss": 1.2106,
+      "step": 3027
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 7.150895140664962e-06,
+      "loss": 0.0049,
+      "step": 3028
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 7.143222506393862e-06,
+      "loss": 0.0022,
+      "step": 3029
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 7.135549872122762e-06,
+      "loss": 0.0038,
+      "step": 3030
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 7.127877237851662e-06,
+      "loss": 0.004,
+      "step": 3031
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 7.120204603580563e-06,
+      "loss": 0.0018,
+      "step": 3032
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 7.112531969309464e-06,
+      "loss": 0.0038,
+      "step": 3033
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 7.104859335038364e-06,
+      "loss": 0.0011,
+      "step": 3034
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 7.097186700767264e-06,
+      "loss": 0.0033,
+      "step": 3035
+    },
+    {
+      "epoch": 7.67,
+      "learning_rate": 7.089514066496164e-06,
+      "loss": 0.0052,
+      "step": 3036
+    },
+    {
+      "epoch": 7.67,
+      "learning_rate": 7.081841432225064e-06,
+      "loss": 1.4336,
+      "step": 3037
+    },
+    {
+      "epoch": 7.67,
+      "learning_rate": 7.074168797953964e-06,
+      "loss": 0.0017,
+      "step": 3038
+    },
+    {
+      "epoch": 7.67,
+      "learning_rate": 7.066496163682865e-06,
+      "loss": 0.0039,
+      "step": 3039
+    },
+    {
+      "epoch": 7.68,
+      "learning_rate": 7.058823529411765e-06,
+      "loss": 1.1887,
+      "step": 3040
+    },
+    {
+      "epoch": 7.68,
+      "learning_rate": 7.051150895140665e-06,
+      "loss": 0.0031,
+      "step": 3041
+    },
+    {
+      "epoch": 7.68,
+      "learning_rate": 7.043478260869566e-06,
+      "loss": 0.005,
+      "step": 3042
+    },
+    {
+      "epoch": 7.68,
+      "learning_rate": 7.0358056265984655e-06,
+      "loss": 0.0036,
+      "step": 3043
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 7.028132992327366e-06,
+      "loss": 0.0042,
+      "step": 3044
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 7.020460358056266e-06,
+      "loss": 0.0027,
+      "step": 3045
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 7.012787723785166e-06,
+      "loss": 1.1529,
+      "step": 3046
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 7.005115089514067e-06,
+      "loss": 0.0043,
+      "step": 3047
+    },
+    {
+      "epoch": 7.7,
+      "learning_rate": 6.997442455242967e-06,
+      "loss": 0.0064,
+      "step": 3048
+    },
+    {
+      "epoch": 7.7,
+      "learning_rate": 6.989769820971867e-06,
+      "loss": 0.0045,
+      "step": 3049
+    },
+    {
+      "epoch": 7.7,
+      "learning_rate": 6.9820971867007675e-06,
+      "loss": 0.0029,
+      "step": 3050
+    },
+    {
+      "epoch": 7.7,
+      "learning_rate": 6.974424552429667e-06,
+      "loss": 0.0044,
+      "step": 3051
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 6.966751918158567e-06,
+      "loss": 0.0012,
+      "step": 3052
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 6.959079283887468e-06,
+      "loss": 0.0018,
+      "step": 3053
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 6.951406649616368e-06,
+      "loss": 1.1802,
+      "step": 3054
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 6.943734015345268e-06,
+      "loss": 0.0048,
+      "step": 3055
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 6.936061381074169e-06,
+      "loss": 1.1027,
+      "step": 3056
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 6.9283887468030695e-06,
+      "loss": 0.0026,
+      "step": 3057
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 6.92071611253197e-06,
+      "loss": 1.24,
+      "step": 3058
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 6.91304347826087e-06,
+      "loss": 1.6565,
+      "step": 3059
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 6.90537084398977e-06,
+      "loss": 0.0047,
+      "step": 3060
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 6.897698209718671e-06,
+      "loss": 1.3994,
+      "step": 3061
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 6.890025575447571e-06,
+      "loss": 1.3895,
+      "step": 3062
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 6.882352941176471e-06,
+      "loss": 0.0073,
+      "step": 3063
+    },
+    {
+      "epoch": 7.74,
+      "learning_rate": 6.8746803069053714e-06,
+      "loss": 0.0075,
+      "step": 3064
+    },
+    {
+      "epoch": 7.74,
+      "learning_rate": 6.867007672634271e-06,
+      "loss": 0.0192,
+      "step": 3065
+    },
+    {
+      "epoch": 7.74,
+      "learning_rate": 6.859335038363171e-06,
+      "loss": 0.005,
+      "step": 3066
+    },
+    {
+      "epoch": 7.74,
+      "learning_rate": 6.851662404092072e-06,
+      "loss": 0.0079,
+      "step": 3067
+    },
+    {
+      "epoch": 7.75,
+      "learning_rate": 6.843989769820972e-06,
+      "loss": 0.005,
+      "step": 3068
+    },
+    {
+      "epoch": 7.75,
+      "learning_rate": 6.836317135549873e-06,
+      "loss": 0.0051,
+      "step": 3069
+    },
+    {
+      "epoch": 7.75,
+      "learning_rate": 6.828644501278773e-06,
+      "loss": 1.4104,
+      "step": 3070
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 6.8209718670076725e-06,
+      "loss": 0.0014,
+      "step": 3071
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 6.813299232736573e-06,
+      "loss": 0.0072,
+      "step": 3072
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 6.805626598465473e-06,
+      "loss": 0.006,
+      "step": 3073
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 6.797953964194373e-06,
+      "loss": 0.0046,
+      "step": 3074
+    },
+    {
+      "epoch": 7.77,
+      "learning_rate": 6.790281329923274e-06,
+      "loss": 0.0092,
+      "step": 3075
+    },
+    {
+      "epoch": 7.77,
+      "learning_rate": 6.782608695652174e-06,
+      "loss": 0.003,
+      "step": 3076
+    },
+    {
+      "epoch": 7.77,
+      "learning_rate": 6.774936061381074e-06,
+      "loss": 0.0071,
+      "step": 3077
+    },
+    {
+      "epoch": 7.77,
+      "learning_rate": 6.7672634271099745e-06,
+      "loss": 0.0089,
+      "step": 3078
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 6.7595907928388744e-06,
+      "loss": 0.0051,
+      "step": 3079
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 6.751918158567774e-06,
+      "loss": 0.0074,
+      "step": 3080
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 6.744245524296675e-06,
+      "loss": 0.0082,
+      "step": 3081
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 6.736572890025576e-06,
+      "loss": 0.0047,
+      "step": 3082
+    },
+    {
+      "epoch": 7.79,
+      "learning_rate": 6.728900255754477e-06,
+      "loss": 0.0071,
+      "step": 3083
+    },
+    {
+      "epoch": 7.79,
+      "learning_rate": 6.7212276214833766e-06,
+      "loss": 0.0029,
+      "step": 3084
+    },
+    {
+      "epoch": 7.79,
+      "learning_rate": 6.7135549872122765e-06,
+      "loss": 0.0071,
+      "step": 3085
+    },
+    {
+      "epoch": 7.79,
+      "learning_rate": 6.705882352941177e-06,
+      "loss": 0.0062,
+      "step": 3086
+    },
+    {
+      "epoch": 7.8,
+      "learning_rate": 6.698209718670077e-06,
+      "loss": 0.0046,
+      "step": 3087
+    },
+    {
+      "epoch": 7.8,
+      "learning_rate": 6.690537084398977e-06,
+      "loss": 0.0028,
+      "step": 3088
+    },
+    {
+      "epoch": 7.8,
+      "learning_rate": 6.682864450127878e-06,
+      "loss": 0.0049,
+      "step": 3089
+    },
+    {
+      "epoch": 7.8,
+      "learning_rate": 6.675191815856778e-06,
+      "loss": 0.0078,
+      "step": 3090
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 6.667519181585678e-06,
+      "loss": 0.0388,
+      "step": 3091
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 6.6598465473145784e-06,
+      "loss": 0.007,
+      "step": 3092
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 6.652173913043478e-06,
+      "loss": 0.003,
+      "step": 3093
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 6.644501278772378e-06,
+      "loss": 0.0028,
+      "step": 3094
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 6.636828644501279e-06,
+      "loss": 1.424,
+      "step": 3095
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 6.629156010230179e-06,
+      "loss": 0.016,
+      "step": 3096
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 6.62148337595908e-06,
+      "loss": 0.0046,
+      "step": 3097
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 6.61381074168798e-06,
+      "loss": 0.0048,
+      "step": 3098
+    },
+    {
+      "epoch": 7.83,
+      "learning_rate": 6.6061381074168796e-06,
+      "loss": 0.0066,
+      "step": 3099
+    },
+    {
+      "epoch": 7.83,
+      "learning_rate": 6.59846547314578e-06,
+      "loss": 0.0047,
+      "step": 3100
+    },
+    {
+      "epoch": 7.83,
+      "learning_rate": 6.59079283887468e-06,
+      "loss": 0.0029,
+      "step": 3101
+    },
+    {
+      "epoch": 7.83,
+      "learning_rate": 6.58312020460358e-06,
+      "loss": 0.0042,
+      "step": 3102
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 6.575447570332481e-06,
+      "loss": 0.003,
+      "step": 3103
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 6.567774936061381e-06,
+      "loss": 0.0028,
+      "step": 3104
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 6.560102301790281e-06,
+      "loss": 0.0042,
+      "step": 3105
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 6.5524296675191815e-06,
+      "loss": 0.0032,
+      "step": 3106
+    },
+    {
+      "epoch": 7.85,
+      "learning_rate": 6.544757033248082e-06,
+      "loss": 0.0041,
+      "step": 3107
+    },
+    {
+      "epoch": 7.85,
+      "learning_rate": 6.537084398976983e-06,
+      "loss": 0.005,
+      "step": 3108
+    },
+    {
+      "epoch": 7.85,
+      "learning_rate": 6.529411764705883e-06,
+      "loss": 0.0035,
+      "step": 3109
+    },
+    {
+      "epoch": 7.85,
+      "learning_rate": 6.521739130434783e-06,
+      "loss": 1.2368,
+      "step": 3110
+    },
+    {
+      "epoch": 7.86,
+      "learning_rate": 6.514066496163684e-06,
+      "loss": 0.0041,
+      "step": 3111
+    },
+    {
+      "epoch": 7.86,
+      "learning_rate": 6.5063938618925836e-06,
+      "loss": 0.0054,
+      "step": 3112
+    },
+    {
+      "epoch": 7.86,
+      "learning_rate": 6.4987212276214835e-06,
+      "loss": 1.4303,
+      "step": 3113
+    },
+    {
+      "epoch": 7.86,
+      "learning_rate": 6.491048593350384e-06,
+      "loss": 0.0041,
+      "step": 3114
+    },
+    {
+      "epoch": 7.87,
+      "learning_rate": 6.483375959079284e-06,
+      "loss": 0.0042,
+      "step": 3115
+    },
+    {
+      "epoch": 7.87,
+      "learning_rate": 6.475703324808184e-06,
+      "loss": 0.0034,
+      "step": 3116
+    },
+    {
+      "epoch": 7.87,
+      "learning_rate": 6.468030690537085e-06,
+      "loss": 0.0046,
+      "step": 3117
+    },
+    {
+      "epoch": 7.87,
+      "learning_rate": 6.460358056265985e-06,
+      "loss": 0.0037,
+      "step": 3118
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 6.452685421994885e-06,
+      "loss": 0.0027,
+      "step": 3119
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 6.4450127877237854e-06,
+      "loss": 0.0045,
+      "step": 3120
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 6.437340153452685e-06,
+      "loss": 0.0056,
+      "step": 3121
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 6.429667519181586e-06,
+      "loss": 0.0166,
+      "step": 3122
+    },
+    {
+      "epoch": 7.89,
+      "learning_rate": 6.421994884910486e-06,
+      "loss": 0.0031,
+      "step": 3123
+    },
+    {
+      "epoch": 7.89,
+      "learning_rate": 6.414322250639386e-06,
+      "loss": 1.2067,
+      "step": 3124
+    },
+    {
+      "epoch": 7.89,
+      "learning_rate": 6.406649616368287e-06,
+      "loss": 0.0932,
+      "step": 3125
+    },
+    {
+      "epoch": 7.89,
+      "learning_rate": 6.398976982097187e-06,
+      "loss": 0.0028,
+      "step": 3126
+    },
+    {
+      "epoch": 7.9,
+      "learning_rate": 6.3913043478260866e-06,
+      "loss": 0.0056,
+      "step": 3127
+    },
+    {
+      "epoch": 7.9,
+      "learning_rate": 6.383631713554987e-06,
+      "loss": 0.0022,
+      "step": 3128
+    },
+    {
+      "epoch": 7.9,
+      "learning_rate": 6.375959079283887e-06,
+      "loss": 0.0042,
+      "step": 3129
+    },
+    {
+      "epoch": 7.9,
+      "learning_rate": 6.368286445012787e-06,
+      "loss": 0.0056,
+      "step": 3130
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 6.360613810741688e-06,
+      "loss": 0.0042,
+      "step": 3131
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 6.352941176470589e-06,
+      "loss": 1.4088,
+      "step": 3132
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 6.345268542199489e-06,
+      "loss": 0.0022,
+      "step": 3133
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 6.337595907928389e-06,
+      "loss": 0.0062,
+      "step": 3134
+    },
+    {
+      "epoch": 7.92,
+      "learning_rate": 6.329923273657289e-06,
+      "loss": 0.0042,
+      "step": 3135
+    },
+    {
+      "epoch": 7.92,
+      "learning_rate": 6.32225063938619e-06,
+      "loss": 0.0048,
+      "step": 3136
+    },
+    {
+      "epoch": 7.92,
+      "learning_rate": 6.31457800511509e-06,
+      "loss": 1.4579,
+      "step": 3137
+    },
+    {
+      "epoch": 7.92,
+      "learning_rate": 6.30690537084399e-06,
+      "loss": 0.0043,
+      "step": 3138
+    },
+    {
+      "epoch": 7.93,
+      "learning_rate": 6.299232736572891e-06,
+      "loss": 0.0056,
+      "step": 3139
+    },
+    {
+      "epoch": 7.93,
+      "learning_rate": 6.291560102301791e-06,
+      "loss": 0.0043,
+      "step": 3140
+    },
+    {
+      "epoch": 7.93,
+      "learning_rate": 6.2838874680306905e-06,
+      "loss": 0.0057,
+      "step": 3141
+    },
+    {
+      "epoch": 7.93,
+      "learning_rate": 6.276214833759591e-06,
+      "loss": 0.0043,
+      "step": 3142
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 6.268542199488491e-06,
+      "loss": 0.0703,
+      "step": 3143
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 6.260869565217391e-06,
+      "loss": 0.0023,
+      "step": 3144
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 6.253196930946292e-06,
+      "loss": 0.0081,
+      "step": 3145
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 6.245524296675192e-06,
+      "loss": 0.0046,
+      "step": 3146
+    },
+    {
+      "epoch": 7.95,
+      "learning_rate": 6.237851662404092e-06,
+      "loss": 0.0047,
+      "step": 3147
+    },
+    {
+      "epoch": 7.95,
+      "learning_rate": 6.2301790281329925e-06,
+      "loss": 1.3925,
+      "step": 3148
+    },
+    {
+      "epoch": 7.95,
+      "learning_rate": 6.222506393861892e-06,
+      "loss": 0.0077,
+      "step": 3149
+    },
+    {
+      "epoch": 7.95,
+      "learning_rate": 6.214833759590793e-06,
+      "loss": 0.0051,
+      "step": 3150
+    },
+    {
+      "epoch": 7.96,
+      "learning_rate": 6.207161125319693e-06,
+      "loss": 0.0052,
+      "step": 3151
+    },
+    {
+      "epoch": 7.96,
+      "learning_rate": 6.199488491048593e-06,
+      "loss": 0.0037,
+      "step": 3152
+    },
+    {
+      "epoch": 7.96,
+      "learning_rate": 6.191815856777494e-06,
+      "loss": 0.007,
+      "step": 3153
+    },
+    {
+      "epoch": 7.96,
+      "learning_rate": 6.184143222506394e-06,
+      "loss": 0.0051,
+      "step": 3154
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 6.176470588235294e-06,
+      "loss": 1.4096,
+      "step": 3155
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 6.168797953964195e-06,
+      "loss": 0.0043,
+      "step": 3156
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 6.161125319693095e-06,
+      "loss": 0.0029,
+      "step": 3157
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 6.153452685421995e-06,
+      "loss": 0.0089,
+      "step": 3158
+    },
+    {
+      "epoch": 7.98,
+      "learning_rate": 6.145780051150896e-06,
+      "loss": 0.0043,
+      "step": 3159
+    },
+    {
+      "epoch": 7.98,
+      "learning_rate": 6.138107416879796e-06,
+      "loss": 0.009,
+      "step": 3160
+    },
+    {
+      "epoch": 7.98,
+      "learning_rate": 6.1304347826086965e-06,
+      "loss": 0.0045,
+      "step": 3161
+    },
+    {
+      "epoch": 7.98,
+      "learning_rate": 6.122762148337596e-06,
+      "loss": 0.0059,
+      "step": 3162
+    },
+    {
+      "epoch": 7.99,
+      "learning_rate": 6.115089514066496e-06,
+      "loss": 0.0049,
+      "step": 3163
+    },
+    {
+      "epoch": 7.99,
+      "learning_rate": 6.107416879795397e-06,
+      "loss": 0.0059,
+      "step": 3164
+    },
+    {
+      "epoch": 7.99,
+      "learning_rate": 6.099744245524297e-06,
+      "loss": 0.0033,
+      "step": 3165
+    },
+    {
+      "epoch": 7.99,
+      "learning_rate": 6.092071611253197e-06,
+      "loss": 0.0056,
+      "step": 3166
+    },
+    {
+      "epoch": 8.0,
+      "learning_rate": 6.084398976982098e-06,
+      "loss": 0.0053,
+      "step": 3167
+    },
+    {
+      "epoch": 8.0,
+      "learning_rate": 6.076726342710998e-06,
+      "loss": 0.0046,
+      "step": 3168
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 0.9242424242424242,
+      "eval_loss": 0.3901217579841614,
+      "eval_runtime": 57.8076,
+      "eval_samples_per_second": 6.85,
+      "eval_steps_per_second": 1.713,
+      "step": 3168
+    }
+  ],
+  "max_steps": 3960,
+  "num_train_epochs": 10,
+  "total_flos": 1.1497180375296e+18,
+  "trial_name": null,
+  "trial_params": null
+}
diff --git a/checkpoint-3168/training_args.bin b/checkpoint-3168/training_args.bin
new file mode 100644
index 0000000000000000000000000000000000000000..32901f899c91066c808eae9dd019400bfdd17498
--- /dev/null
+++ b/checkpoint-3168/training_args.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db14439c369bfe9c89f64ff38f0b2d3fcc53911fc2c0b12a3a7d04656e7384a
+size 3503
diff --git a/checkpoint-3564/config.json b/checkpoint-3564/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..cb8d52e14263c50fd53dca42b5050e3235115c32
--- /dev/null
+++ b/checkpoint-3564/config.json
@@ -0,0 +1,113 @@
+{
+  "_name_or_path": "facebook/wav2vec2-base-960h",
+  "activation_dropout": 0.1,
+  "adapter_kernel_size": 3,
+  "adapter_stride": 2,
+  "add_adapter": false,
+  "apply_spec_augment": true,
+  "architectures": [
+    "Wav2Vec2ForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 1,
+  "classifier_proj_size": 256,
+  "codevector_dim": 256,
+  "contrastive_logits_temperature": 0.1,
+  "conv_bias": false,
+  "conv_dim": [
+    512,
+    512,
+    512,
+    512,
+    512,
+    512,
+    512
+  ],
+  "conv_kernel": [
+    10,
+    3,
+    3,
+    3,
+    3,
+    2,
+    2
+  ],
+  "conv_stride": [
+    5,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2
+  ],
+  "ctc_loss_reduction": "sum",
+  "ctc_zero_infinity": false,
+  "diversity_loss_weight": 0.1,
+  "do_stable_layer_norm": false,
+  "eos_token_id": 2,
+  "feat_extract_activation": "gelu",
+  "feat_extract_dropout": 0.0,
+  "feat_extract_norm": "group",
+  "feat_proj_dropout": 0.1,
+  "feat_quantizer_dropout": 0.0,
+  "final_dropout": 0.1,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout": 0.1,
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Foreign",
+    "1": "Native"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "layerdrop": 0.1,
+  "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
+  "mask_feature_prob": 0.0,
+  "mask_time_length": 10,
+  "mask_time_min_masks": 2,
+  "mask_time_prob": 0.05,
+  "model_type": "wav2vec2",
+  "num_adapter_layers": 3,
+  "num_attention_heads": 12,
+  "num_codevector_groups": 2,
+  "num_codevectors_per_group": 320,
+  "num_conv_pos_embedding_groups": 16,
+  "num_conv_pos_embeddings": 128,
+  "num_feat_extract_layers": 7,
+  "num_hidden_layers": 12,
+  "num_negatives": 100,
+  "output_hidden_size": 768,
+  "pad_token_id": 0,
+  "proj_codevector_dim": 256,
+  "tdnn_dilation": [
+    1,
+    2,
+    3,
+    1,
+    1
+  ],
+  "tdnn_dim": [
+    512,
+    512,
+    512,
+    512,
+    1500
+  ],
+  "tdnn_kernel": [
+    5,
+    3,
+    3,
+    1,
+    1
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.28.1",
+  "use_weighted_layer_sum": false,
+  "vocab_size": 32,
+  "xvector_output_dim": 512
+}
diff --git a/checkpoint-3564/optimizer.pt b/checkpoint-3564/optimizer.pt
new file mode 100644
index 0000000000000000000000000000000000000000..2eba9d0c49038ea8899c1d6b653d81b77dc4936c
--- /dev/null
+++ b/checkpoint-3564/optimizer.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3d9bd6b60cd229d78ae29b1f74924d0c6347b2f90c1716edb3614940feee3bc8
+size 756677477
diff --git a/checkpoint-3564/preprocessor_config.json b/checkpoint-3564/preprocessor_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..a0b7227fc1d916e469b14f6c154ad6dfea1e6891
--- /dev/null
+++ b/checkpoint-3564/preprocessor_config.json
@@ -0,0 +1,9 @@
+{
+  "do_normalize": true,
+  "feature_extractor_type": "Wav2Vec2FeatureExtractor",
+  "feature_size": 1,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": false,
+  "sampling_rate": 16000
+}
diff --git a/checkpoint-3564/pytorch_model.bin b/checkpoint-3564/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..8f2e336fa680a3e214e4e65d10217c66e69be51f
--- /dev/null
+++ b/checkpoint-3564/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3a39bb9aef4ad32b917d83708dac2aaae3d1468bb39c888a537b5904622e0ad0
+size 378348563
diff --git a/checkpoint-3564/rng_state.pth b/checkpoint-3564/rng_state.pth
new file mode 100644
index 0000000000000000000000000000000000000000..0463716f6b53c1a1e87fa6e0eecfac1b7520a178
--- /dev/null
+++ b/checkpoint-3564/rng_state.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:74f0a9f7d7fc2ce78f50cf9df2e3701a2d0012178ef3b16f0df6ad2563ce9e84
+size 14567
diff --git a/checkpoint-3564/scheduler.pt b/checkpoint-3564/scheduler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..db80da334cadee0b5f271056aed437f9d9d3f990
--- /dev/null
+++ b/checkpoint-3564/scheduler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:343f0c137ed31ca490c895b4a0f7ac28716cc3121954d0171b3c4703aa7614d0
+size 623
diff --git a/checkpoint-3564/trainer_state.json b/checkpoint-3564/trainer_state.json
new file mode 100644
index 0000000000000000000000000000000000000000..820a5917fc478e44328be5cbd6b3f009ef348dcf
--- /dev/null
+++ b/checkpoint-3564/trainer_state.json
@@ -0,0 +1,21481 @@
+{
+  "best_metric": 0.3489837944507599,
+  "best_model_checkpoint": "accent_classification/checkpoint-2376",
+  "epoch": 9.0,
+  "global_step": 3564,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.000000000000001e-07,
+      "loss": 0.6956,
+      "step": 1
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.2000000000000002e-06,
+      "loss": 0.6979,
+      "step": 2
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8e-06,
+      "loss": 0.7014,
+      "step": 3
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 2.4000000000000003e-06,
+      "loss": 0.6984,
+      "step": 4
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 3e-06,
+      "loss": 0.6978,
+      "step": 5
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 3.6e-06,
+      "loss": 0.6783,
+      "step": 6
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.2000000000000004e-06,
+      "loss": 0.6956,
+      "step": 7
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.800000000000001e-06,
+      "loss": 0.6994,
+      "step": 8
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 5.4e-06,
+      "loss": 0.6902,
+      "step": 9
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6e-06,
+      "loss": 0.6937,
+      "step": 10
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6.6e-06,
+      "loss": 0.6497,
+      "step": 11
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.2e-06,
+      "loss": 0.6911,
+      "step": 12
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.8e-06,
+      "loss": 0.6904,
+      "step": 13
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 8.400000000000001e-06,
+      "loss": 0.6854,
+      "step": 14
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9e-06,
+      "loss": 0.6843,
+      "step": 15
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9.600000000000001e-06,
+      "loss": 0.6851,
+      "step": 16
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.02e-05,
+      "loss": 0.6733,
+      "step": 17
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.08e-05,
+      "loss": 0.6819,
+      "step": 18
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.1400000000000001e-05,
+      "loss": 0.681,
+      "step": 19
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.6998,
+      "step": 20
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.26e-05,
+      "loss": 0.6743,
+      "step": 21
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.32e-05,
+      "loss": 0.6818,
+      "step": 22
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3800000000000002e-05,
+      "loss": 0.5331,
+      "step": 23
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.44e-05,
+      "loss": 0.6681,
+      "step": 24
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.6955,
+      "step": 25
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.56e-05,
+      "loss": 0.7,
+      "step": 26
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.62e-05,
+      "loss": 0.6975,
+      "step": 27
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.6800000000000002e-05,
+      "loss": 0.6249,
+      "step": 28
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.74e-05,
+      "loss": 0.6979,
+      "step": 29
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.6076,
+      "step": 30
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.86e-05,
+      "loss": 0.6091,
+      "step": 31
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9200000000000003e-05,
+      "loss": 0.6438,
+      "step": 32
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.98e-05,
+      "loss": 0.6624,
+      "step": 33
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.04e-05,
+      "loss": 0.6452,
+      "step": 34
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.6963,
+      "step": 35
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.16e-05,
+      "loss": 0.758,
+      "step": 36
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.22e-05,
+      "loss": 0.6471,
+      "step": 37
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.2800000000000002e-05,
+      "loss": 0.6313,
+      "step": 38
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.3400000000000003e-05,
+      "loss": 0.7019,
+      "step": 39
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.6336,
+      "step": 40
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4599999999999998e-05,
+      "loss": 0.7091,
+      "step": 41
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.52e-05,
+      "loss": 0.701,
+      "step": 42
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.58e-05,
+      "loss": 0.7036,
+      "step": 43
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.64e-05,
+      "loss": 0.7073,
+      "step": 44
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.614,
+      "step": 45
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.7600000000000003e-05,
+      "loss": 0.7543,
+      "step": 46
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.8199999999999998e-05,
+      "loss": 0.538,
+      "step": 47
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.88e-05,
+      "loss": 0.5179,
+      "step": 48
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.94e-05,
+      "loss": 0.6164,
+      "step": 49
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 3e-05,
+      "loss": 0.6093,
+      "step": 50
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.99923273657289e-05,
+      "loss": 0.6983,
+      "step": 51
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9984654731457804e-05,
+      "loss": 0.6069,
+      "step": 52
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9976982097186702e-05,
+      "loss": 0.5933,
+      "step": 53
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.99693094629156e-05,
+      "loss": 0.552,
+      "step": 54
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9961636828644502e-05,
+      "loss": 0.5865,
+      "step": 55
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9953964194373404e-05,
+      "loss": 0.7146,
+      "step": 56
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9946291560102302e-05,
+      "loss": 0.5591,
+      "step": 57
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9938618925831203e-05,
+      "loss": 0.7366,
+      "step": 58
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9930946291560102e-05,
+      "loss": 0.7006,
+      "step": 59
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9923273657289003e-05,
+      "loss": 0.4148,
+      "step": 60
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9915601023017905e-05,
+      "loss": 0.9938,
+      "step": 61
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9907928388746803e-05,
+      "loss": 0.911,
+      "step": 62
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.99002557544757e-05,
+      "loss": 0.5682,
+      "step": 63
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9892583120204606e-05,
+      "loss": 0.5833,
+      "step": 64
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9884910485933505e-05,
+      "loss": 0.7237,
+      "step": 65
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9877237851662403e-05,
+      "loss": 0.4013,
+      "step": 66
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9869565217391304e-05,
+      "loss": 0.9695,
+      "step": 67
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9861892583120206e-05,
+      "loss": 0.568,
+      "step": 68
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9854219948849104e-05,
+      "loss": 0.5697,
+      "step": 69
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9846547314578006e-05,
+      "loss": 0.7675,
+      "step": 70
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9838874680306904e-05,
+      "loss": 0.8831,
+      "step": 71
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9831202046035806e-05,
+      "loss": 0.7008,
+      "step": 72
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9823529411764707e-05,
+      "loss": 0.7085,
+      "step": 73
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9815856777493606e-05,
+      "loss": 0.5374,
+      "step": 74
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9808184143222507e-05,
+      "loss": 0.4131,
+      "step": 75
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.980051150895141e-05,
+      "loss": 0.7309,
+      "step": 76
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9792838874680307e-05,
+      "loss": 0.4279,
+      "step": 77
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.978516624040921e-05,
+      "loss": 0.5813,
+      "step": 78
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.977749360613811e-05,
+      "loss": 0.6276,
+      "step": 79
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976982097186701e-05,
+      "loss": 0.7892,
+      "step": 80
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976214833759591e-05,
+      "loss": 0.7134,
+      "step": 81
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.9754475703324808e-05,
+      "loss": 0.3682,
+      "step": 82
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.974680306905371e-05,
+      "loss": 0.5562,
+      "step": 83
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973913043478261e-05,
+      "loss": 0.7158,
+      "step": 84
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973145780051151e-05,
+      "loss": 0.524,
+      "step": 85
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9723785166240408e-05,
+      "loss": 0.3372,
+      "step": 86
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9716112531969313e-05,
+      "loss": 0.4979,
+      "step": 87
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970843989769821e-05,
+      "loss": 0.3177,
+      "step": 88
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970076726342711e-05,
+      "loss": 0.3488,
+      "step": 89
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.969309462915601e-05,
+      "loss": 0.5158,
+      "step": 90
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9685421994884913e-05,
+      "loss": 0.3208,
+      "step": 91
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.967774936061381e-05,
+      "loss": 0.6654,
+      "step": 92
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9670076726342712e-05,
+      "loss": 0.5581,
+      "step": 93
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.966240409207161e-05,
+      "loss": 0.7456,
+      "step": 94
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9654731457800512e-05,
+      "loss": 0.4437,
+      "step": 95
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9647058823529414e-05,
+      "loss": 0.5525,
+      "step": 96
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9639386189258312e-05,
+      "loss": 0.6673,
+      "step": 97
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.963171355498721e-05,
+      "loss": 0.435,
+      "step": 98
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9624040920716115e-05,
+      "loss": 0.7415,
+      "step": 99
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9616368286445014e-05,
+      "loss": 0.2981,
+      "step": 100
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9608695652173912e-05,
+      "loss": 0.5179,
+      "step": 101
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9601023017902817e-05,
+      "loss": 0.4005,
+      "step": 102
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9593350383631715e-05,
+      "loss": 0.4598,
+      "step": 103
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9585677749360613e-05,
+      "loss": 0.2553,
+      "step": 104
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9578005115089515e-05,
+      "loss": 0.4302,
+      "step": 105
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9570332480818416e-05,
+      "loss": 0.7894,
+      "step": 106
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9562659846547315e-05,
+      "loss": 0.213,
+      "step": 107
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9554987212276216e-05,
+      "loss": 0.5507,
+      "step": 108
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9547314578005115e-05,
+      "loss": 0.7676,
+      "step": 109
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9539641943734016e-05,
+      "loss": 0.3418,
+      "step": 110
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9531969309462918e-05,
+      "loss": 0.8533,
+      "step": 111
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9524296675191816e-05,
+      "loss": 0.3659,
+      "step": 112
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9516624040920714e-05,
+      "loss": 0.6482,
+      "step": 113
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.950895140664962e-05,
+      "loss": 0.5484,
+      "step": 114
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9501278772378517e-05,
+      "loss": 0.4508,
+      "step": 115
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9493606138107416e-05,
+      "loss": 0.4401,
+      "step": 116
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9485933503836317e-05,
+      "loss": 0.5455,
+      "step": 117
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.947826086956522e-05,
+      "loss": 0.722,
+      "step": 118
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9470588235294117e-05,
+      "loss": 0.2874,
+      "step": 119
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.946291560102302e-05,
+      "loss": 0.5973,
+      "step": 120
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.9455242966751917e-05,
+      "loss": 0.5898,
+      "step": 121
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.944757033248082e-05,
+      "loss": 1.0125,
+      "step": 122
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943989769820972e-05,
+      "loss": 0.2511,
+      "step": 123
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943222506393862e-05,
+      "loss": 0.545,
+      "step": 124
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.942455242966752e-05,
+      "loss": 0.593,
+      "step": 125
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.941687979539642e-05,
+      "loss": 0.2319,
+      "step": 126
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.940920716112532e-05,
+      "loss": 0.2215,
+      "step": 127
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.9401534526854218e-05,
+      "loss": 0.1964,
+      "step": 128
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9393861892583123e-05,
+      "loss": 0.1471,
+      "step": 129
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.938618925831202e-05,
+      "loss": 1.5965,
+      "step": 130
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9378516624040923e-05,
+      "loss": 1.0772,
+      "step": 131
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.937084398976982e-05,
+      "loss": 0.2005,
+      "step": 132
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9363171355498723e-05,
+      "loss": 0.9628,
+      "step": 133
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9355498721227624e-05,
+      "loss": 0.1503,
+      "step": 134
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9347826086956523e-05,
+      "loss": 0.6201,
+      "step": 135
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.934015345268542e-05,
+      "loss": 0.6094,
+      "step": 136
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9332480818414326e-05,
+      "loss": 1.385,
+      "step": 137
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9324808184143224e-05,
+      "loss": 0.1552,
+      "step": 138
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9317135549872122e-05,
+      "loss": 0.5828,
+      "step": 139
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9309462915601024e-05,
+      "loss": 0.6018,
+      "step": 140
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9301790281329925e-05,
+      "loss": 1.0614,
+      "step": 141
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9294117647058824e-05,
+      "loss": 0.566,
+      "step": 142
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9286445012787725e-05,
+      "loss": 0.5958,
+      "step": 143
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9278772378516624e-05,
+      "loss": 0.5838,
+      "step": 144
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9271099744245525e-05,
+      "loss": 0.5909,
+      "step": 145
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9263427109974427e-05,
+      "loss": 0.5862,
+      "step": 146
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9255754475703325e-05,
+      "loss": 0.2031,
+      "step": 147
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9248081841432223e-05,
+      "loss": 0.1995,
+      "step": 148
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9240409207161128e-05,
+      "loss": 0.24,
+      "step": 149
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9232736572890026e-05,
+      "loss": 0.5845,
+      "step": 150
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9225063938618925e-05,
+      "loss": 0.964,
+      "step": 151
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.921739130434783e-05,
+      "loss": 0.9595,
+      "step": 152
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9209718670076728e-05,
+      "loss": 0.2033,
+      "step": 153
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9202046035805626e-05,
+      "loss": 0.213,
+      "step": 154
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9194373401534528e-05,
+      "loss": 0.2042,
+      "step": 155
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.918670076726343e-05,
+      "loss": 0.5783,
+      "step": 156
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9179028132992328e-05,
+      "loss": 1.0822,
+      "step": 157
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.917135549872123e-05,
+      "loss": 1.2643,
+      "step": 158
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9163682864450127e-05,
+      "loss": 0.2593,
+      "step": 159
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.915601023017903e-05,
+      "loss": 0.5783,
+      "step": 160
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914833759590793e-05,
+      "loss": 0.5721,
+      "step": 161
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914066496163683e-05,
+      "loss": 0.5808,
+      "step": 162
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9132992327365727e-05,
+      "loss": 1.2578,
+      "step": 163
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9125319693094632e-05,
+      "loss": 0.5709,
+      "step": 164
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.911764705882353e-05,
+      "loss": 0.2159,
+      "step": 165
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910997442455243e-05,
+      "loss": 0.2329,
+      "step": 166
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910230179028133e-05,
+      "loss": 0.926,
+      "step": 167
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.9094629156010232e-05,
+      "loss": 0.9095,
+      "step": 168
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.908695652173913e-05,
+      "loss": 0.5622,
+      "step": 169
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907928388746803e-05,
+      "loss": 0.5703,
+      "step": 170
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907161125319693e-05,
+      "loss": 0.5669,
+      "step": 171
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.906393861892583e-05,
+      "loss": 0.2637,
+      "step": 172
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9056265984654733e-05,
+      "loss": 1.1077,
+      "step": 173
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.904859335038363e-05,
+      "loss": 0.5653,
+      "step": 174
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9040920716112533e-05,
+      "loss": 0.5584,
+      "step": 175
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9033248081841434e-05,
+      "loss": 0.5544,
+      "step": 176
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9025575447570333e-05,
+      "loss": 0.5663,
+      "step": 177
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.901790281329923e-05,
+      "loss": 1.0808,
+      "step": 178
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9010230179028136e-05,
+      "loss": 0.8736,
+      "step": 179
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9002557544757034e-05,
+      "loss": 0.6435,
+      "step": 180
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8994884910485932e-05,
+      "loss": 0.786,
+      "step": 181
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8987212276214834e-05,
+      "loss": 0.5717,
+      "step": 182
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8979539641943736e-05,
+      "loss": 0.5618,
+      "step": 183
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8971867007672637e-05,
+      "loss": 0.5785,
+      "step": 184
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8964194373401535e-05,
+      "loss": 0.7601,
+      "step": 185
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8956521739130434e-05,
+      "loss": 0.8112,
+      "step": 186
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.894884910485934e-05,
+      "loss": 0.5663,
+      "step": 187
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8941176470588237e-05,
+      "loss": 0.5631,
+      "step": 188
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8933503836317135e-05,
+      "loss": 0.7363,
+      "step": 189
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8925831202046037e-05,
+      "loss": 0.4552,
+      "step": 190
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8918158567774938e-05,
+      "loss": 0.4599,
+      "step": 191
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8910485933503836e-05,
+      "loss": 0.7848,
+      "step": 192
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8902813299232738e-05,
+      "loss": 0.5907,
+      "step": 193
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8895140664961636e-05,
+      "loss": 0.4264,
+      "step": 194
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8887468030690538e-05,
+      "loss": 0.5697,
+      "step": 195
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.887979539641944e-05,
+      "loss": 0.7643,
+      "step": 196
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8872122762148338e-05,
+      "loss": 0.3635,
+      "step": 197
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8864450127877236e-05,
+      "loss": 0.3715,
+      "step": 198
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.885677749360614e-05,
+      "loss": 0.554,
+      "step": 199
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.884910485933504e-05,
+      "loss": 0.8044,
+      "step": 200
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8841432225063937e-05,
+      "loss": 0.5673,
+      "step": 201
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8833759590792842e-05,
+      "loss": 0.8042,
+      "step": 202
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.882608695652174e-05,
+      "loss": 0.5638,
+      "step": 203
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881841432225064e-05,
+      "loss": 0.3584,
+      "step": 204
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881074168797954e-05,
+      "loss": 0.8211,
+      "step": 205
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.8803069053708442e-05,
+      "loss": 0.5708,
+      "step": 206
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.879539641943734e-05,
+      "loss": 0.8301,
+      "step": 207
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8787723785166242e-05,
+      "loss": 0.8229,
+      "step": 208
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.878005115089514e-05,
+      "loss": 0.5645,
+      "step": 209
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8772378516624042e-05,
+      "loss": 0.3529,
+      "step": 210
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8764705882352943e-05,
+      "loss": 0.5578,
+      "step": 211
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.875703324808184e-05,
+      "loss": 0.5659,
+      "step": 212
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.874936061381074e-05,
+      "loss": 0.7495,
+      "step": 213
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8741687979539645e-05,
+      "loss": 0.3162,
+      "step": 214
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8734015345268543e-05,
+      "loss": 0.5616,
+      "step": 215
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.872634271099744e-05,
+      "loss": 0.4034,
+      "step": 216
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8718670076726343e-05,
+      "loss": 0.3163,
+      "step": 217
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8710997442455245e-05,
+      "loss": 0.5514,
+      "step": 218
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8703324808184143e-05,
+      "loss": 0.826,
+      "step": 219
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8695652173913044e-05,
+      "loss": 0.8363,
+      "step": 220
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8687979539641943e-05,
+      "loss": 0.8288,
+      "step": 221
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8680306905370844e-05,
+      "loss": 0.5929,
+      "step": 222
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8672634271099746e-05,
+      "loss": 0.565,
+      "step": 223
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8664961636828644e-05,
+      "loss": 0.5642,
+      "step": 224
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8657289002557546e-05,
+      "loss": 0.5609,
+      "step": 225
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8649616368286447e-05,
+      "loss": 0.5641,
+      "step": 226
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8641943734015345e-05,
+      "loss": 0.8535,
+      "step": 227
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8634271099744244e-05,
+      "loss": 0.5652,
+      "step": 228
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.862659846547315e-05,
+      "loss": 0.4397,
+      "step": 229
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8618925831202047e-05,
+      "loss": 0.567,
+      "step": 230
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8611253196930945e-05,
+      "loss": 0.7517,
+      "step": 231
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8603580562659847e-05,
+      "loss": 0.5839,
+      "step": 232
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.859590792838875e-05,
+      "loss": 0.5677,
+      "step": 233
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8588235294117647e-05,
+      "loss": 0.5678,
+      "step": 234
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8580562659846548e-05,
+      "loss": 1.3045,
+      "step": 235
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8572890025575446e-05,
+      "loss": 0.804,
+      "step": 236
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.856521739130435e-05,
+      "loss": 1.0303,
+      "step": 237
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.855754475703325e-05,
+      "loss": 0.3303,
+      "step": 238
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8549872122762148e-05,
+      "loss": 0.5677,
+      "step": 239
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.854219948849105e-05,
+      "loss": 0.7972,
+      "step": 240
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.853452685421995e-05,
+      "loss": 0.5644,
+      "step": 241
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.852685421994885e-05,
+      "loss": 0.9294,
+      "step": 242
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.851918158567775e-05,
+      "loss": 0.5665,
+      "step": 243
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.851150895140665e-05,
+      "loss": 0.7963,
+      "step": 244
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.850383631713555e-05,
+      "loss": 0.5659,
+      "step": 245
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.8496163682864452e-05,
+      "loss": 0.5718,
+      "step": 246
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.848849104859335e-05,
+      "loss": 0.3526,
+      "step": 247
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8480818414322252e-05,
+      "loss": 0.574,
+      "step": 248
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8473145780051154e-05,
+      "loss": 0.9996,
+      "step": 249
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8465473145780052e-05,
+      "loss": 0.7634,
+      "step": 250
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.845780051150895e-05,
+      "loss": 0.5639,
+      "step": 251
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8450127877237855e-05,
+      "loss": 0.5737,
+      "step": 252
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8442455242966753e-05,
+      "loss": 0.7793,
+      "step": 253
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8434782608695652e-05,
+      "loss": 0.6769,
+      "step": 254
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8427109974424553e-05,
+      "loss": 0.5748,
+      "step": 255
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8419437340153455e-05,
+      "loss": 0.3602,
+      "step": 256
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8411764705882353e-05,
+      "loss": 0.5714,
+      "step": 257
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8404092071611255e-05,
+      "loss": 0.5738,
+      "step": 258
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8396419437340153e-05,
+      "loss": 0.5719,
+      "step": 259
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8388746803069055e-05,
+      "loss": 0.5753,
+      "step": 260
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8381074168797956e-05,
+      "loss": 0.5717,
+      "step": 261
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8373401534526854e-05,
+      "loss": 0.9794,
+      "step": 262
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8365728900255753e-05,
+      "loss": 0.5737,
+      "step": 263
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8358056265984658e-05,
+      "loss": 0.5723,
+      "step": 264
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8350383631713556e-05,
+      "loss": 0.5656,
+      "step": 265
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8342710997442454e-05,
+      "loss": 0.4259,
+      "step": 266
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8335038363171356e-05,
+      "loss": 0.3076,
+      "step": 267
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8327365728900257e-05,
+      "loss": 0.338,
+      "step": 268
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8319693094629156e-05,
+      "loss": 0.5612,
+      "step": 269
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8312020460358057e-05,
+      "loss": 0.5249,
+      "step": 270
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8304347826086955e-05,
+      "loss": 0.5671,
+      "step": 271
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8296675191815857e-05,
+      "loss": 0.5574,
+      "step": 272
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.828900255754476e-05,
+      "loss": 0.8152,
+      "step": 273
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8281329923273657e-05,
+      "loss": 0.5647,
+      "step": 274
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.827365728900256e-05,
+      "loss": 0.5632,
+      "step": 275
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.826598465473146e-05,
+      "loss": 0.3391,
+      "step": 276
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8258312020460358e-05,
+      "loss": 0.55,
+      "step": 277
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8250639386189257e-05,
+      "loss": 0.5672,
+      "step": 278
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.824296675191816e-05,
+      "loss": 0.285,
+      "step": 279
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.823529411764706e-05,
+      "loss": 0.2809,
+      "step": 280
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.8227621483375958e-05,
+      "loss": 0.5552,
+      "step": 281
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821994884910486e-05,
+      "loss": 0.563,
+      "step": 282
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821227621483376e-05,
+      "loss": 0.5628,
+      "step": 283
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.820460358056266e-05,
+      "loss": 0.2548,
+      "step": 284
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.819693094629156e-05,
+      "loss": 0.871,
+      "step": 285
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818925831202046e-05,
+      "loss": 0.8797,
+      "step": 286
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818158567774936e-05,
+      "loss": 0.2453,
+      "step": 287
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8173913043478262e-05,
+      "loss": 0.2464,
+      "step": 288
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.816624040920716e-05,
+      "loss": 0.3134,
+      "step": 289
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8158567774936062e-05,
+      "loss": 0.2431,
+      "step": 290
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8150895140664964e-05,
+      "loss": 0.2308,
+      "step": 291
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8143222506393862e-05,
+      "loss": 0.2173,
+      "step": 292
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8135549872122764e-05,
+      "loss": 0.5665,
+      "step": 293
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8127877237851662e-05,
+      "loss": 0.5592,
+      "step": 294
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8120204603580564e-05,
+      "loss": 0.473,
+      "step": 295
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8112531969309465e-05,
+      "loss": 0.9458,
+      "step": 296
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8104859335038363e-05,
+      "loss": 0.1879,
+      "step": 297
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8097186700767265e-05,
+      "loss": 0.6797,
+      "step": 298
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8089514066496167e-05,
+      "loss": 0.7592,
+      "step": 299
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8081841432225065e-05,
+      "loss": 0.4109,
+      "step": 300
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8074168797953963e-05,
+      "loss": 0.5821,
+      "step": 301
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8066496163682868e-05,
+      "loss": 0.2057,
+      "step": 302
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8058823529411766e-05,
+      "loss": 0.5322,
+      "step": 303
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8051150895140665e-05,
+      "loss": 0.5208,
+      "step": 304
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8043478260869566e-05,
+      "loss": 0.6799,
+      "step": 305
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8035805626598468e-05,
+      "loss": 0.5894,
+      "step": 306
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8028132992327366e-05,
+      "loss": 0.4051,
+      "step": 307
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8020460358056268e-05,
+      "loss": 0.3695,
+      "step": 308
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8012787723785166e-05,
+      "loss": 0.4869,
+      "step": 309
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8005115089514067e-05,
+      "loss": 0.6075,
+      "step": 310
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.799744245524297e-05,
+      "loss": 0.4245,
+      "step": 311
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7989769820971867e-05,
+      "loss": 0.5122,
+      "step": 312
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7982097186700765e-05,
+      "loss": 0.452,
+      "step": 313
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.797442455242967e-05,
+      "loss": 0.9568,
+      "step": 314
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.796675191815857e-05,
+      "loss": 0.5846,
+      "step": 315
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.7959079283887467e-05,
+      "loss": 0.3743,
+      "step": 316
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.795140664961637e-05,
+      "loss": 0.1922,
+      "step": 317
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.794373401534527e-05,
+      "loss": 0.5303,
+      "step": 318
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.793606138107417e-05,
+      "loss": 0.7253,
+      "step": 319
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.792838874680307e-05,
+      "loss": 0.2794,
+      "step": 320
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.7920716112531968e-05,
+      "loss": 0.577,
+      "step": 321
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.791304347826087e-05,
+      "loss": 0.614,
+      "step": 322
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.790537084398977e-05,
+      "loss": 0.5882,
+      "step": 323
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789769820971867e-05,
+      "loss": 0.2503,
+      "step": 324
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789002557544757e-05,
+      "loss": 0.4195,
+      "step": 325
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.7882352941176473e-05,
+      "loss": 0.2959,
+      "step": 326
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.787468030690537e-05,
+      "loss": 0.3732,
+      "step": 327
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.786700767263427e-05,
+      "loss": 0.5729,
+      "step": 328
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7859335038363174e-05,
+      "loss": 0.3179,
+      "step": 329
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7851662404092073e-05,
+      "loss": 0.6823,
+      "step": 330
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.784398976982097e-05,
+      "loss": 0.2663,
+      "step": 331
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7836317135549872e-05,
+      "loss": 0.4436,
+      "step": 332
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7828644501278774e-05,
+      "loss": 0.3089,
+      "step": 333
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7820971867007672e-05,
+      "loss": 0.5888,
+      "step": 334
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7813299232736574e-05,
+      "loss": 0.2115,
+      "step": 335
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7805626598465472e-05,
+      "loss": 0.4005,
+      "step": 336
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7797953964194374e-05,
+      "loss": 0.3097,
+      "step": 337
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7790281329923275e-05,
+      "loss": 0.5847,
+      "step": 338
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7782608695652174e-05,
+      "loss": 1.0164,
+      "step": 339
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7774936061381072e-05,
+      "loss": 1.1321,
+      "step": 340
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7767263427109977e-05,
+      "loss": 0.592,
+      "step": 341
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7759590792838875e-05,
+      "loss": 1.0191,
+      "step": 342
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7751918158567777e-05,
+      "loss": 0.9199,
+      "step": 343
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7744245524296675e-05,
+      "loss": 0.2874,
+      "step": 344
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7736572890025576e-05,
+      "loss": 0.2469,
+      "step": 345
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7728900255754478e-05,
+      "loss": 0.8364,
+      "step": 346
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7721227621483376e-05,
+      "loss": 0.7365,
+      "step": 347
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7713554987212278e-05,
+      "loss": 0.5595,
+      "step": 348
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.770588235294118e-05,
+      "loss": 0.2546,
+      "step": 349
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7698209718670078e-05,
+      "loss": 0.8183,
+      "step": 350
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7690537084398976e-05,
+      "loss": 0.5385,
+      "step": 351
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.768286445012788e-05,
+      "loss": 0.2763,
+      "step": 352
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.767519181585678e-05,
+      "loss": 0.3045,
+      "step": 353
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7667519181585677e-05,
+      "loss": 0.3354,
+      "step": 354
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765984654731458e-05,
+      "loss": 0.9366,
+      "step": 355
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765217391304348e-05,
+      "loss": 0.6212,
+      "step": 356
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.764450127877238e-05,
+      "loss": 0.1161,
+      "step": 357
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.763682864450128e-05,
+      "loss": 0.4929,
+      "step": 358
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762915601023018e-05,
+      "loss": 0.8167,
+      "step": 359
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762148337595908e-05,
+      "loss": 0.2521,
+      "step": 360
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.7613810741687982e-05,
+      "loss": 0.4174,
+      "step": 361
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.760613810741688e-05,
+      "loss": 0.3822,
+      "step": 362
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.759846547314578e-05,
+      "loss": 0.5355,
+      "step": 363
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.7590792838874683e-05,
+      "loss": 0.561,
+      "step": 364
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.758312020460358e-05,
+      "loss": 0.5126,
+      "step": 365
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.757544757033248e-05,
+      "loss": 0.2301,
+      "step": 366
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.756777493606138e-05,
+      "loss": 0.1698,
+      "step": 367
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7560102301790283e-05,
+      "loss": 0.1634,
+      "step": 368
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.755242966751918e-05,
+      "loss": 0.5955,
+      "step": 369
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7544757033248083e-05,
+      "loss": 0.1568,
+      "step": 370
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.753708439897698e-05,
+      "loss": 0.5167,
+      "step": 371
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7529411764705883e-05,
+      "loss": 1.0028,
+      "step": 372
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7521739130434784e-05,
+      "loss": 1.2567,
+      "step": 373
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7514066496163682e-05,
+      "loss": 0.6094,
+      "step": 374
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7506393861892584e-05,
+      "loss": 0.6059,
+      "step": 375
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7498721227621486e-05,
+      "loss": 1.0593,
+      "step": 376
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7491048593350384e-05,
+      "loss": 0.1506,
+      "step": 377
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7483375959079282e-05,
+      "loss": 1.5106,
+      "step": 378
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7475703324808187e-05,
+      "loss": 0.1827,
+      "step": 379
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7468030690537085e-05,
+      "loss": 1.4961,
+      "step": 380
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7460358056265984e-05,
+      "loss": 1.0456,
+      "step": 381
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7452685421994885e-05,
+      "loss": 1.2063,
+      "step": 382
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7445012787723787e-05,
+      "loss": 0.5837,
+      "step": 383
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7437340153452685e-05,
+      "loss": 1.4536,
+      "step": 384
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7429667519181587e-05,
+      "loss": 0.5803,
+      "step": 385
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7421994884910485e-05,
+      "loss": 0.5896,
+      "step": 386
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7414322250639386e-05,
+      "loss": 0.1822,
+      "step": 387
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7406649616368288e-05,
+      "loss": 1.7893,
+      "step": 388
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7398976982097186e-05,
+      "loss": 0.9723,
+      "step": 389
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7391304347826085e-05,
+      "loss": 1.2815,
+      "step": 390
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.738363171355499e-05,
+      "loss": 0.5765,
+      "step": 391
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7375959079283888e-05,
+      "loss": 1.2429,
+      "step": 392
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7368286445012786e-05,
+      "loss": 0.1343,
+      "step": 393
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7360613810741688e-05,
+      "loss": 0.2306,
+      "step": 394
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.735294117647059e-05,
+      "loss": 0.624,
+      "step": 395
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.734526854219949e-05,
+      "loss": 0.2371,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7348484848484849,
+      "eval_loss": 0.5804409980773926,
+      "eval_runtime": 57.3336,
+      "eval_samples_per_second": 6.907,
+      "eval_steps_per_second": 1.727,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.733759590792839e-05,
+      "loss": 0.8989,
+      "step": 397
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.732992327365729e-05,
+      "loss": 0.5664,
+      "step": 398
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.7322250639386192e-05,
+      "loss": 0.8852,
+      "step": 399
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.731457800511509e-05,
+      "loss": 0.5661,
+      "step": 400
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.730690537084399e-05,
+      "loss": 0.5648,
+      "step": 401
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7299232736572894e-05,
+      "loss": 0.7333,
+      "step": 402
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7291560102301792e-05,
+      "loss": 0.3037,
+      "step": 403
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.728388746803069e-05,
+      "loss": 0.1463,
+      "step": 404
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7276214833759592e-05,
+      "loss": 0.2702,
+      "step": 405
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7268542199488493e-05,
+      "loss": 0.2693,
+      "step": 406
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.726086956521739e-05,
+      "loss": 0.5647,
+      "step": 407
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7253196930946293e-05,
+      "loss": 0.8507,
+      "step": 408
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.724552429667519e-05,
+      "loss": 0.8645,
+      "step": 409
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7237851662404093e-05,
+      "loss": 0.2649,
+      "step": 410
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7230179028132995e-05,
+      "loss": 0.6942,
+      "step": 411
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7222506393861893e-05,
+      "loss": 0.5656,
+      "step": 412
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.721483375959079e-05,
+      "loss": 0.5489,
+      "step": 413
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7207161125319696e-05,
+      "loss": 0.2652,
+      "step": 414
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7199488491048594e-05,
+      "loss": 0.8676,
+      "step": 415
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7191815856777493e-05,
+      "loss": 0.8621,
+      "step": 416
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7184143222506394e-05,
+      "loss": 0.2641,
+      "step": 417
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7176470588235296e-05,
+      "loss": 0.859,
+      "step": 418
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7168797953964194e-05,
+      "loss": 0.6492,
+      "step": 419
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7161125319693096e-05,
+      "loss": 1.1494,
+      "step": 420
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7153452685421994e-05,
+      "loss": 0.8278,
+      "step": 421
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7145780051150895e-05,
+      "loss": 0.764,
+      "step": 422
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7138107416879797e-05,
+      "loss": 0.564,
+      "step": 423
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7130434782608695e-05,
+      "loss": 0.8051,
+      "step": 424
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7122762148337597e-05,
+      "loss": 0.3052,
+      "step": 425
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.71150895140665e-05,
+      "loss": 0.2945,
+      "step": 426
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7107416879795397e-05,
+      "loss": 0.4118,
+      "step": 427
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7099744245524295e-05,
+      "loss": 0.559,
+      "step": 428
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.70920716112532e-05,
+      "loss": 0.5593,
+      "step": 429
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7084398976982098e-05,
+      "loss": 0.564,
+      "step": 430
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7076726342710996e-05,
+      "loss": 0.4454,
+      "step": 431
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7069053708439898e-05,
+      "loss": 0.4381,
+      "step": 432
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.70613810741688e-05,
+      "loss": 0.3718,
+      "step": 433
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7053708439897698e-05,
+      "loss": 0.6627,
+      "step": 434
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70460358056266e-05,
+      "loss": 0.4004,
+      "step": 435
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7038363171355498e-05,
+      "loss": 0.3855,
+      "step": 436
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70306905370844e-05,
+      "loss": 0.4799,
+      "step": 437
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70230179028133e-05,
+      "loss": 0.6714,
+      "step": 438
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70153452685422e-05,
+      "loss": 0.622,
+      "step": 439
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7007672634271097e-05,
+      "loss": 0.3916,
+      "step": 440
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.5487,
+      "step": 441
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69923273657289e-05,
+      "loss": 0.376,
+      "step": 442
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69846547314578e-05,
+      "loss": 0.3616,
+      "step": 443
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69769820971867e-05,
+      "loss": 0.8747,
+      "step": 444
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.6969309462915602e-05,
+      "loss": 0.283,
+      "step": 445
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.69616368286445e-05,
+      "loss": 0.3988,
+      "step": 446
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6953964194373402e-05,
+      "loss": 0.4569,
+      "step": 447
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6946291560102303e-05,
+      "loss": 0.2005,
+      "step": 448
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6938618925831205e-05,
+      "loss": 0.3087,
+      "step": 449
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6930946291560103e-05,
+      "loss": 0.5814,
+      "step": 450
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6923273657289e-05,
+      "loss": 0.2995,
+      "step": 451
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6915601023017907e-05,
+      "loss": 0.7806,
+      "step": 452
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6907928388746805e-05,
+      "loss": 0.2666,
+      "step": 453
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6900255754475703e-05,
+      "loss": 0.2571,
+      "step": 454
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6892583120204605e-05,
+      "loss": 0.5931,
+      "step": 455
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6884910485933506e-05,
+      "loss": 0.475,
+      "step": 456
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6877237851662404e-05,
+      "loss": 0.2372,
+      "step": 457
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6869565217391306e-05,
+      "loss": 0.1577,
+      "step": 458
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6861892583120204e-05,
+      "loss": 0.1953,
+      "step": 459
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6854219948849106e-05,
+      "loss": 0.1931,
+      "step": 460
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6846547314578008e-05,
+      "loss": 0.1829,
+      "step": 461
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6838874680306906e-05,
+      "loss": 0.1601,
+      "step": 462
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6831202046035804e-05,
+      "loss": 0.2505,
+      "step": 463
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.682352941176471e-05,
+      "loss": 0.181,
+      "step": 464
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6815856777493607e-05,
+      "loss": 0.5978,
+      "step": 465
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6808184143222505e-05,
+      "loss": 0.5458,
+      "step": 466
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6800511508951407e-05,
+      "loss": 0.1289,
+      "step": 467
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.679283887468031e-05,
+      "loss": 0.4383,
+      "step": 468
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6785166240409207e-05,
+      "loss": 0.5892,
+      "step": 469
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.677749360613811e-05,
+      "loss": 1.3565,
+      "step": 470
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.676982097186701e-05,
+      "loss": 0.7267,
+      "step": 471
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.6762148337595908e-05,
+      "loss": 0.1354,
+      "step": 472
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.675447570332481e-05,
+      "loss": 0.9291,
+      "step": 473
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.6746803069053708e-05,
+      "loss": 0.5265,
+      "step": 474
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673913043478261e-05,
+      "loss": 0.4841,
+      "step": 475
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673145780051151e-05,
+      "loss": 1.2391,
+      "step": 476
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.672378516624041e-05,
+      "loss": 0.4924,
+      "step": 477
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6716112531969308e-05,
+      "loss": 1.0076,
+      "step": 478
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6708439897698213e-05,
+      "loss": 0.6314,
+      "step": 479
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.670076726342711e-05,
+      "loss": 0.4237,
+      "step": 480
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.669309462915601e-05,
+      "loss": 0.1077,
+      "step": 481
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.668542199488491e-05,
+      "loss": 0.3038,
+      "step": 482
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6677749360613812e-05,
+      "loss": 0.6081,
+      "step": 483
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.667007672634271e-05,
+      "loss": 0.1078,
+      "step": 484
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6662404092071612e-05,
+      "loss": 0.4296,
+      "step": 485
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.665473145780051e-05,
+      "loss": 0.3981,
+      "step": 486
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6647058823529412e-05,
+      "loss": 0.5853,
+      "step": 487
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6639386189258314e-05,
+      "loss": 0.802,
+      "step": 488
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6631713554987212e-05,
+      "loss": 0.4576,
+      "step": 489
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.662404092071611e-05,
+      "loss": 0.2682,
+      "step": 490
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6616368286445015e-05,
+      "loss": 0.1991,
+      "step": 491
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6608695652173913e-05,
+      "loss": 0.1409,
+      "step": 492
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.660102301790281e-05,
+      "loss": 0.3869,
+      "step": 493
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6593350383631713e-05,
+      "loss": 0.5301,
+      "step": 494
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6585677749360615e-05,
+      "loss": 0.8116,
+      "step": 495
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6578005115089513e-05,
+      "loss": 0.0984,
+      "step": 496
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6570332480818415e-05,
+      "loss": 0.7773,
+      "step": 497
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6562659846547316e-05,
+      "loss": 1.3522,
+      "step": 498
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6554987212276215e-05,
+      "loss": 0.0862,
+      "step": 499
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6547314578005116e-05,
+      "loss": 0.0894,
+      "step": 500
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6539641943734014e-05,
+      "loss": 0.0885,
+      "step": 501
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.653196930946292e-05,
+      "loss": 0.3031,
+      "step": 502
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6524296675191818e-05,
+      "loss": 0.187,
+      "step": 503
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6516624040920716e-05,
+      "loss": 1.3063,
+      "step": 504
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6508951406649617e-05,
+      "loss": 0.6847,
+      "step": 505
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.650127877237852e-05,
+      "loss": 0.078,
+      "step": 506
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6493606138107417e-05,
+      "loss": 0.1265,
+      "step": 507
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.648593350383632e-05,
+      "loss": 0.0804,
+      "step": 508
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.6478260869565217e-05,
+      "loss": 0.6976,
+      "step": 509
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.647058823529412e-05,
+      "loss": 0.7977,
+      "step": 510
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.646291560102302e-05,
+      "loss": 0.413,
+      "step": 511
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.645524296675192e-05,
+      "loss": 0.6944,
+      "step": 512
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6447570332480817e-05,
+      "loss": 0.7177,
+      "step": 513
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6439897698209722e-05,
+      "loss": 0.0581,
+      "step": 514
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.643222506393862e-05,
+      "loss": 0.9308,
+      "step": 515
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6424552429667518e-05,
+      "loss": 0.2518,
+      "step": 516
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.641687979539642e-05,
+      "loss": 0.1014,
+      "step": 517
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640920716112532e-05,
+      "loss": 0.1641,
+      "step": 518
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640153452685422e-05,
+      "loss": 0.101,
+      "step": 519
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.639386189258312e-05,
+      "loss": 0.1097,
+      "step": 520
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6386189258312023e-05,
+      "loss": 0.987,
+      "step": 521
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.637851662404092e-05,
+      "loss": 0.6465,
+      "step": 522
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6370843989769823e-05,
+      "loss": 0.0914,
+      "step": 523
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.636317135549872e-05,
+      "loss": 0.139,
+      "step": 524
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6355498721227623e-05,
+      "loss": 0.6455,
+      "step": 525
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6347826086956524e-05,
+      "loss": 0.204,
+      "step": 526
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6340153452685422e-05,
+      "loss": 0.1269,
+      "step": 527
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.633248081841432e-05,
+      "loss": 0.2214,
+      "step": 528
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6324808184143226e-05,
+      "loss": 0.0736,
+      "step": 529
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6317135549872124e-05,
+      "loss": 0.0714,
+      "step": 530
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6309462915601022e-05,
+      "loss": 0.1388,
+      "step": 531
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6301790281329924e-05,
+      "loss": 0.7758,
+      "step": 532
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6294117647058825e-05,
+      "loss": 0.3388,
+      "step": 533
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6286445012787724e-05,
+      "loss": 1.1831,
+      "step": 534
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6278772378516625e-05,
+      "loss": 0.1166,
+      "step": 535
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6271099744245523e-05,
+      "loss": 0.2389,
+      "step": 536
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6263427109974425e-05,
+      "loss": 0.1622,
+      "step": 537
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6255754475703327e-05,
+      "loss": 0.1015,
+      "step": 538
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6248081841432225e-05,
+      "loss": 0.0658,
+      "step": 539
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6240409207161123e-05,
+      "loss": 0.0954,
+      "step": 540
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6232736572890028e-05,
+      "loss": 0.1182,
+      "step": 541
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6225063938618926e-05,
+      "loss": 0.0605,
+      "step": 542
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6217391304347824e-05,
+      "loss": 0.9265,
+      "step": 543
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6209718670076726e-05,
+      "loss": 1.2994,
+      "step": 544
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6202046035805628e-05,
+      "loss": 0.081,
+      "step": 545
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6194373401534526e-05,
+      "loss": 0.5962,
+      "step": 546
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6186700767263428e-05,
+      "loss": 0.0665,
+      "step": 547
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.617902813299233e-05,
+      "loss": 0.0655,
+      "step": 548
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6171355498721227e-05,
+      "loss": 0.0642,
+      "step": 549
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.616368286445013e-05,
+      "loss": 0.0834,
+      "step": 550
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6156010230179027e-05,
+      "loss": 0.047,
+      "step": 551
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6148337595907932e-05,
+      "loss": 0.0642,
+      "step": 552
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.614066496163683e-05,
+      "loss": 0.0594,
+      "step": 553
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.613299232736573e-05,
+      "loss": 0.0573,
+      "step": 554
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.612531969309463e-05,
+      "loss": 0.1101,
+      "step": 555
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.6117647058823532e-05,
+      "loss": 0.5121,
+      "step": 556
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.610997442455243e-05,
+      "loss": 0.0374,
+      "step": 557
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.6102301790281332e-05,
+      "loss": 0.0957,
+      "step": 558
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.609462915601023e-05,
+      "loss": 0.0488,
+      "step": 559
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.608695652173913e-05,
+      "loss": 0.0482,
+      "step": 560
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6079283887468033e-05,
+      "loss": 1.2041,
+      "step": 561
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.607161125319693e-05,
+      "loss": 0.8833,
+      "step": 562
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.606393861892583e-05,
+      "loss": 0.1544,
+      "step": 563
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6056265984654735e-05,
+      "loss": 0.046,
+      "step": 564
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6048593350383633e-05,
+      "loss": 0.9073,
+      "step": 565
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.604092071611253e-05,
+      "loss": 0.052,
+      "step": 566
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6033248081841433e-05,
+      "loss": 0.0308,
+      "step": 567
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6025575447570334e-05,
+      "loss": 0.7032,
+      "step": 568
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6017902813299232e-05,
+      "loss": 0.6947,
+      "step": 569
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6010230179028134e-05,
+      "loss": 1.5603,
+      "step": 570
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6002557544757036e-05,
+      "loss": 0.8836,
+      "step": 571
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.5994884910485934e-05,
+      "loss": 1.7439,
+      "step": 572
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5987212276214836e-05,
+      "loss": 0.0313,
+      "step": 573
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5979539641943734e-05,
+      "loss": 0.0317,
+      "step": 574
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5971867007672635e-05,
+      "loss": 0.0325,
+      "step": 575
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5964194373401537e-05,
+      "loss": 0.5,
+      "step": 576
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5956521739130435e-05,
+      "loss": 0.0675,
+      "step": 577
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5948849104859333e-05,
+      "loss": 0.6364,
+      "step": 578
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.594117647058824e-05,
+      "loss": 0.0696,
+      "step": 579
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5933503836317137e-05,
+      "loss": 0.6195,
+      "step": 580
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5925831202046035e-05,
+      "loss": 0.5796,
+      "step": 581
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5918158567774937e-05,
+      "loss": 0.0832,
+      "step": 582
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5910485933503838e-05,
+      "loss": 0.0794,
+      "step": 583
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5902813299232736e-05,
+      "loss": 0.5851,
+      "step": 584
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5895140664961638e-05,
+      "loss": 0.104,
+      "step": 585
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5887468030690536e-05,
+      "loss": 1.0492,
+      "step": 586
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5879795396419438e-05,
+      "loss": 0.0645,
+      "step": 587
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.587212276214834e-05,
+      "loss": 0.514,
+      "step": 588
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5864450127877238e-05,
+      "loss": 0.0701,
+      "step": 589
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5856777493606136e-05,
+      "loss": 0.5409,
+      "step": 590
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584910485933504e-05,
+      "loss": 0.0536,
+      "step": 591
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584143222506394e-05,
+      "loss": 0.1728,
+      "step": 592
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.5833759590792837e-05,
+      "loss": 0.0845,
+      "step": 593
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.582608695652174e-05,
+      "loss": 0.2217,
+      "step": 594
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.581841432225064e-05,
+      "loss": 0.1397,
+      "step": 595
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.581074168797954e-05,
+      "loss": 0.4119,
+      "step": 596
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.580306905370844e-05,
+      "loss": 0.3638,
+      "step": 597
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.5795396419437342e-05,
+      "loss": 0.0781,
+      "step": 598
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.578772378516624e-05,
+      "loss": 0.5926,
+      "step": 599
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5780051150895142e-05,
+      "loss": 0.1378,
+      "step": 600
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.577237851662404e-05,
+      "loss": 0.142,
+      "step": 601
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.576470588235294e-05,
+      "loss": 0.1722,
+      "step": 602
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5757033248081843e-05,
+      "loss": 0.1529,
+      "step": 603
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.574936061381074e-05,
+      "loss": 1.6083,
+      "step": 604
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5741687979539643e-05,
+      "loss": 1.8267,
+      "step": 605
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5734015345268545e-05,
+      "loss": 0.0535,
+      "step": 606
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5726342710997443e-05,
+      "loss": 0.9945,
+      "step": 607
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5718670076726345e-05,
+      "loss": 0.9133,
+      "step": 608
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5710997442455243e-05,
+      "loss": 0.5295,
+      "step": 609
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5703324808184144e-05,
+      "loss": 0.058,
+      "step": 610
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5695652173913046e-05,
+      "loss": 0.4789,
+      "step": 611
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5687979539641944e-05,
+      "loss": 0.0572,
+      "step": 612
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5680306905370842e-05,
+      "loss": 0.0897,
+      "step": 613
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5672634271099747e-05,
+      "loss": 0.056,
+      "step": 614
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5664961636828646e-05,
+      "loss": 0.0278,
+      "step": 615
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5657289002557544e-05,
+      "loss": 0.9947,
+      "step": 616
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5649616368286445e-05,
+      "loss": 0.0627,
+      "step": 617
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5641943734015347e-05,
+      "loss": 0.9335,
+      "step": 618
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5634271099744245e-05,
+      "loss": 0.9793,
+      "step": 619
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5626598465473147e-05,
+      "loss": 1.8128,
+      "step": 620
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.561892583120205e-05,
+      "loss": 1.8015,
+      "step": 621
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5611253196930947e-05,
+      "loss": 0.9259,
+      "step": 622
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.560358056265985e-05,
+      "loss": 0.9618,
+      "step": 623
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5595907928388747e-05,
+      "loss": 0.0808,
+      "step": 624
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5588235294117648e-05,
+      "loss": 1.7161,
+      "step": 625
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.558056265984655e-05,
+      "loss": 0.8879,
+      "step": 626
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5572890025575448e-05,
+      "loss": 0.043,
+      "step": 627
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5565217391304346e-05,
+      "loss": 0.9449,
+      "step": 628
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.555754475703325e-05,
+      "loss": 0.0624,
+      "step": 629
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.554987212276215e-05,
+      "loss": 0.0924,
+      "step": 630
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5542199488491048e-05,
+      "loss": 0.0715,
+      "step": 631
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.553452685421995e-05,
+      "loss": 0.4995,
+      "step": 632
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.552685421994885e-05,
+      "loss": 0.5576,
+      "step": 633
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551918158567775e-05,
+      "loss": 0.06,
+      "step": 634
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551150895140665e-05,
+      "loss": 0.0873,
+      "step": 635
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.550383631713555e-05,
+      "loss": 0.6868,
+      "step": 636
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.549616368286445e-05,
+      "loss": 0.8487,
+      "step": 637
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.5488491048593352e-05,
+      "loss": 0.5641,
+      "step": 638
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.548081841432225e-05,
+      "loss": 1.1343,
+      "step": 639
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.547314578005115e-05,
+      "loss": 0.0614,
+      "step": 640
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5465473145780054e-05,
+      "loss": 0.9414,
+      "step": 641
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5457800511508952e-05,
+      "loss": 0.9226,
+      "step": 642
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.545012787723785e-05,
+      "loss": 0.3323,
+      "step": 643
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5442455242966752e-05,
+      "loss": 0.9315,
+      "step": 644
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5434782608695653e-05,
+      "loss": 0.5539,
+      "step": 645
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.542710997442455e-05,
+      "loss": 0.3376,
+      "step": 646
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5419437340153453e-05,
+      "loss": 0.0694,
+      "step": 647
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5411764705882355e-05,
+      "loss": 0.0706,
+      "step": 648
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5404092071611253e-05,
+      "loss": 0.105,
+      "step": 649
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5396419437340155e-05,
+      "loss": 0.912,
+      "step": 650
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5388746803069053e-05,
+      "loss": 0.0998,
+      "step": 651
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5381074168797954e-05,
+      "loss": 0.2247,
+      "step": 652
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5373401534526856e-05,
+      "loss": 0.2869,
+      "step": 653
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5365728900255754e-05,
+      "loss": 0.532,
+      "step": 654
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5358056265984653e-05,
+      "loss": 0.4723,
+      "step": 655
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5350383631713558e-05,
+      "loss": 0.1178,
+      "step": 656
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5342710997442456e-05,
+      "loss": 0.0566,
+      "step": 657
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5335038363171357e-05,
+      "loss": 0.059,
+      "step": 658
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5327365728900256e-05,
+      "loss": 0.0881,
+      "step": 659
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5319693094629157e-05,
+      "loss": 0.5762,
+      "step": 660
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.531202046035806e-05,
+      "loss": 0.0499,
+      "step": 661
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5304347826086957e-05,
+      "loss": 0.087,
+      "step": 662
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5296675191815855e-05,
+      "loss": 1.0719,
+      "step": 663
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528900255754476e-05,
+      "loss": 0.6158,
+      "step": 664
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528132992327366e-05,
+      "loss": 0.6117,
+      "step": 665
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5273657289002557e-05,
+      "loss": 0.0515,
+      "step": 666
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5265984654731458e-05,
+      "loss": 0.0608,
+      "step": 667
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.525831202046036e-05,
+      "loss": 0.0313,
+      "step": 668
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.5250639386189258e-05,
+      "loss": 0.1449,
+      "step": 669
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.524296675191816e-05,
+      "loss": 0.9883,
+      "step": 670
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.523529411764706e-05,
+      "loss": 0.0859,
+      "step": 671
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.522762148337596e-05,
+      "loss": 0.0561,
+      "step": 672
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521994884910486e-05,
+      "loss": 0.0926,
+      "step": 673
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521227621483376e-05,
+      "loss": 0.1101,
+      "step": 674
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.520460358056266e-05,
+      "loss": 0.068,
+      "step": 675
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5196930946291563e-05,
+      "loss": 0.0773,
+      "step": 676
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518925831202046e-05,
+      "loss": 0.0508,
+      "step": 677
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518158567774936e-05,
+      "loss": 0.0375,
+      "step": 678
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5173913043478264e-05,
+      "loss": 0.93,
+      "step": 679
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5166240409207162e-05,
+      "loss": 0.1507,
+      "step": 680
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.515856777493606e-05,
+      "loss": 0.0917,
+      "step": 681
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5150895140664962e-05,
+      "loss": 0.97,
+      "step": 682
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5143222506393864e-05,
+      "loss": 0.5019,
+      "step": 683
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5135549872122762e-05,
+      "loss": 0.0849,
+      "step": 684
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5127877237851664e-05,
+      "loss": 1.2316,
+      "step": 685
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5120204603580562e-05,
+      "loss": 0.0252,
+      "step": 686
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5112531969309463e-05,
+      "loss": 1.0543,
+      "step": 687
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5104859335038365e-05,
+      "loss": 0.0506,
+      "step": 688
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5097186700767263e-05,
+      "loss": 0.0252,
+      "step": 689
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.508951406649616e-05,
+      "loss": 0.6297,
+      "step": 690
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5081841432225066e-05,
+      "loss": 0.4497,
+      "step": 691
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5074168797953965e-05,
+      "loss": 0.0516,
+      "step": 692
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5066496163682863e-05,
+      "loss": 0.5501,
+      "step": 693
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5058823529411768e-05,
+      "loss": 0.5621,
+      "step": 694
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5051150895140666e-05,
+      "loss": 0.0577,
+      "step": 695
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5043478260869564e-05,
+      "loss": 0.0271,
+      "step": 696
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5035805626598466e-05,
+      "loss": 0.9496,
+      "step": 697
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5028132992327368e-05,
+      "loss": 0.0877,
+      "step": 698
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5020460358056266e-05,
+      "loss": 0.0523,
+      "step": 699
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5012787723785167e-05,
+      "loss": 0.0577,
+      "step": 700
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5005115089514066e-05,
+      "loss": 0.0522,
+      "step": 701
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.4997442455242967e-05,
+      "loss": 0.0517,
+      "step": 702
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.498976982097187e-05,
+      "loss": 0.6335,
+      "step": 703
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4982097186700767e-05,
+      "loss": 0.05,
+      "step": 704
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4974424552429665e-05,
+      "loss": 0.0753,
+      "step": 705
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.496675191815857e-05,
+      "loss": 0.0227,
+      "step": 706
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.495907928388747e-05,
+      "loss": 0.0755,
+      "step": 707
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.4951406649616367e-05,
+      "loss": 0.0215,
+      "step": 708
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.494373401534527e-05,
+      "loss": 0.0451,
+      "step": 709
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.493606138107417e-05,
+      "loss": 0.0343,
+      "step": 710
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492838874680307e-05,
+      "loss": 0.0836,
+      "step": 711
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492071611253197e-05,
+      "loss": 0.5287,
+      "step": 712
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4913043478260868e-05,
+      "loss": 0.0686,
+      "step": 713
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4905370843989773e-05,
+      "loss": 0.0211,
+      "step": 714
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489769820971867e-05,
+      "loss": 0.5932,
+      "step": 715
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489002557544757e-05,
+      "loss": 0.0371,
+      "step": 716
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.488235294117647e-05,
+      "loss": 1.0137,
+      "step": 717
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.4874680306905373e-05,
+      "loss": 0.9928,
+      "step": 718
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.486700767263427e-05,
+      "loss": 0.7431,
+      "step": 719
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4859335038363173e-05,
+      "loss": 0.0207,
+      "step": 720
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4851662404092074e-05,
+      "loss": 0.0681,
+      "step": 721
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4843989769820972e-05,
+      "loss": 0.0306,
+      "step": 722
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4836317135549874e-05,
+      "loss": 0.0204,
+      "step": 723
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4828644501278772e-05,
+      "loss": 0.0344,
+      "step": 724
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4820971867007674e-05,
+      "loss": 0.0345,
+      "step": 725
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4813299232736575e-05,
+      "loss": 0.9275,
+      "step": 726
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4805626598465474e-05,
+      "loss": 0.0196,
+      "step": 727
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4797953964194372e-05,
+      "loss": 1.0137,
+      "step": 728
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4790281329923277e-05,
+      "loss": 0.0318,
+      "step": 729
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4782608695652175e-05,
+      "loss": 0.0688,
+      "step": 730
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4774936061381073e-05,
+      "loss": 0.0314,
+      "step": 731
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4767263427109975e-05,
+      "loss": 0.7135,
+      "step": 732
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4759590792838877e-05,
+      "loss": 0.0299,
+      "step": 733
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4751918158567775e-05,
+      "loss": 1.0127,
+      "step": 734
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4744245524296676e-05,
+      "loss": 0.0304,
+      "step": 735
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4736572890025575e-05,
+      "loss": 1.4618,
+      "step": 736
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4728900255754476e-05,
+      "loss": 0.7167,
+      "step": 737
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4721227621483378e-05,
+      "loss": 0.0307,
+      "step": 738
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4713554987212276e-05,
+      "loss": 0.052,
+      "step": 739
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4705882352941174e-05,
+      "loss": 0.778,
+      "step": 740
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.469820971867008e-05,
+      "loss": 0.5482,
+      "step": 741
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4690537084398978e-05,
+      "loss": 1.4314,
+      "step": 742
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4682864450127876e-05,
+      "loss": 0.0422,
+      "step": 743
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.467519181585678e-05,
+      "loss": 0.6614,
+      "step": 744
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.466751918158568e-05,
+      "loss": 0.0312,
+      "step": 745
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4659846547314577e-05,
+      "loss": 0.043,
+      "step": 746
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.465217391304348e-05,
+      "loss": 0.0199,
+      "step": 747
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.464450127877238e-05,
+      "loss": 0.0832,
+      "step": 748
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.463682864450128e-05,
+      "loss": 0.0312,
+      "step": 749
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.462915601023018e-05,
+      "loss": 0.0309,
+      "step": 750
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.462148337595908e-05,
+      "loss": 0.0193,
+      "step": 751
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.461381074168798e-05,
+      "loss": 0.0446,
+      "step": 752
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.4606138107416882e-05,
+      "loss": 0.9993,
+      "step": 753
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.459846547314578e-05,
+      "loss": 0.016,
+      "step": 754
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4590792838874678e-05,
+      "loss": 0.0433,
+      "step": 755
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4583120204603583e-05,
+      "loss": 0.2614,
+      "step": 756
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.457544757033248e-05,
+      "loss": 0.018,
+      "step": 757
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.456777493606138e-05,
+      "loss": 0.6676,
+      "step": 758
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.456010230179028e-05,
+      "loss": 0.0599,
+      "step": 759
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4552429667519183e-05,
+      "loss": 0.0338,
+      "step": 760
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.454475703324808e-05,
+      "loss": 0.0268,
+      "step": 761
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4537084398976983e-05,
+      "loss": 0.0262,
+      "step": 762
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.452941176470588e-05,
+      "loss": 0.0252,
+      "step": 763
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4521739130434786e-05,
+      "loss": 0.0377,
+      "step": 764
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4514066496163684e-05,
+      "loss": 1.4866,
+      "step": 765
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4506393861892582e-05,
+      "loss": 1.0524,
+      "step": 766
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4498721227621484e-05,
+      "loss": 0.0328,
+      "step": 767
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4491048593350386e-05,
+      "loss": 0.024,
+      "step": 768
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4483375959079284e-05,
+      "loss": 0.022,
+      "step": 769
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4475703324808185e-05,
+      "loss": 0.0245,
+      "step": 770
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4468030690537087e-05,
+      "loss": 1.0107,
+      "step": 771
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4460358056265985e-05,
+      "loss": 0.0336,
+      "step": 772
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4452685421994887e-05,
+      "loss": 0.0391,
+      "step": 773
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4445012787723785e-05,
+      "loss": 0.0303,
+      "step": 774
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4437340153452687e-05,
+      "loss": 0.0443,
+      "step": 775
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4429667519181588e-05,
+      "loss": 0.0292,
+      "step": 776
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4421994884910487e-05,
+      "loss": 0.0284,
+      "step": 777
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4414322250639385e-05,
+      "loss": 1.5985,
+      "step": 778
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.440664961636829e-05,
+      "loss": 0.0114,
+      "step": 779
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4398976982097188e-05,
+      "loss": 0.0342,
+      "step": 780
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4391304347826086e-05,
+      "loss": 1.8501,
+      "step": 781
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4383631713554988e-05,
+      "loss": 0.5375,
+      "step": 782
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.437595907928389e-05,
+      "loss": 1.589,
+      "step": 783
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4368286445012788e-05,
+      "loss": 0.0221,
+      "step": 784
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.436061381074169e-05,
+      "loss": 0.0453,
+      "step": 785
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4352941176470587e-05,
+      "loss": 0.796,
+      "step": 786
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.434526854219949e-05,
+      "loss": 0.1249,
+      "step": 787
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.433759590792839e-05,
+      "loss": 0.7963,
+      "step": 788
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.432992327365729e-05,
+      "loss": 0.4231,
+      "step": 789
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.4322250639386187e-05,
+      "loss": 0.0247,
+      "step": 790
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.4314578005115092e-05,
+      "loss": 0.0295,
+      "step": 791
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.430690537084399e-05,
+      "loss": 0.0249,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.3537317216396332,
+      "eval_runtime": 58.1226,
+      "eval_samples_per_second": 6.813,
+      "eval_steps_per_second": 1.703,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.429923273657289e-05,
+      "loss": 0.0157,
+      "step": 793
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4291560102301794e-05,
+      "loss": 0.5222,
+      "step": 794
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4283887468030692e-05,
+      "loss": 0.0226,
+      "step": 795
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.427621483375959e-05,
+      "loss": 0.0293,
+      "step": 796
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.426854219948849e-05,
+      "loss": 0.015,
+      "step": 797
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4260869565217393e-05,
+      "loss": 1.0658,
+      "step": 798
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.425319693094629e-05,
+      "loss": 0.0412,
+      "step": 799
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4245524296675193e-05,
+      "loss": 0.0219,
+      "step": 800
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.423785166240409e-05,
+      "loss": 0.0149,
+      "step": 801
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4230179028132993e-05,
+      "loss": 0.0323,
+      "step": 802
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4222506393861895e-05,
+      "loss": 1.0764,
+      "step": 803
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4214833759590793e-05,
+      "loss": 0.7518,
+      "step": 804
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.420716112531969e-05,
+      "loss": 0.8166,
+      "step": 805
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4199488491048596e-05,
+      "loss": 0.0219,
+      "step": 806
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4191815856777494e-05,
+      "loss": 0.0154,
+      "step": 807
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4184143222506392e-05,
+      "loss": 0.0163,
+      "step": 808
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4176470588235294e-05,
+      "loss": 1.0562,
+      "step": 809
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4168797953964196e-05,
+      "loss": 0.0218,
+      "step": 810
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4161125319693094e-05,
+      "loss": 0.0232,
+      "step": 811
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4153452685421995e-05,
+      "loss": 0.2384,
+      "step": 812
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4145780051150894e-05,
+      "loss": 1.5286,
+      "step": 813
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4138107416879795e-05,
+      "loss": 0.0262,
+      "step": 814
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4130434782608697e-05,
+      "loss": 0.0334,
+      "step": 815
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4122762148337595e-05,
+      "loss": 0.6509,
+      "step": 816
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4115089514066497e-05,
+      "loss": 0.0159,
+      "step": 817
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.41074168797954e-05,
+      "loss": 0.6225,
+      "step": 818
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4099744245524297e-05,
+      "loss": 0.752,
+      "step": 819
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4092071611253198e-05,
+      "loss": 1.0594,
+      "step": 820
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.40843989769821e-05,
+      "loss": 0.0161,
+      "step": 821
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4076726342710998e-05,
+      "loss": 0.7492,
+      "step": 822
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40690537084399e-05,
+      "loss": 0.0302,
+      "step": 823
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4061381074168798e-05,
+      "loss": 0.038,
+      "step": 824
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40537084398977e-05,
+      "loss": 0.0243,
+      "step": 825
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40460358056266e-05,
+      "loss": 0.0331,
+      "step": 826
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40383631713555e-05,
+      "loss": 1.0252,
+      "step": 827
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4030690537084398e-05,
+      "loss": 0.0172,
+      "step": 828
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4023017902813303e-05,
+      "loss": 0.6168,
+      "step": 829
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40153452685422e-05,
+      "loss": 0.3135,
+      "step": 830
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40076726342711e-05,
+      "loss": 0.0168,
+      "step": 831
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.0172,
+      "step": 832
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.3992327365728902e-05,
+      "loss": 1.0349,
+      "step": 833
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39846547314578e-05,
+      "loss": 0.0978,
+      "step": 834
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3976982097186702e-05,
+      "loss": 0.0167,
+      "step": 835
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39693094629156e-05,
+      "loss": 1.0381,
+      "step": 836
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3961636828644502e-05,
+      "loss": 0.6857,
+      "step": 837
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3953964194373404e-05,
+      "loss": 1.0406,
+      "step": 838
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3946291560102302e-05,
+      "loss": 0.071,
+      "step": 839
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.39386189258312e-05,
+      "loss": 0.0193,
+      "step": 840
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3930946291560105e-05,
+      "loss": 0.0193,
+      "step": 841
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3923273657289003e-05,
+      "loss": 0.9916,
+      "step": 842
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.39156010230179e-05,
+      "loss": 0.0723,
+      "step": 843
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3907928388746806e-05,
+      "loss": 0.0626,
+      "step": 844
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3900255754475705e-05,
+      "loss": 0.7264,
+      "step": 845
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3892583120204603e-05,
+      "loss": 0.827,
+      "step": 846
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3884910485933504e-05,
+      "loss": 1.2193,
+      "step": 847
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3877237851662406e-05,
+      "loss": 0.6754,
+      "step": 848
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3869565217391304e-05,
+      "loss": 0.6856,
+      "step": 849
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3861892583120206e-05,
+      "loss": 0.7441,
+      "step": 850
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3854219948849104e-05,
+      "loss": 0.023,
+      "step": 851
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3846547314578006e-05,
+      "loss": 0.4742,
+      "step": 852
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3838874680306907e-05,
+      "loss": 0.0363,
+      "step": 853
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3831202046035806e-05,
+      "loss": 0.0409,
+      "step": 854
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3823529411764704e-05,
+      "loss": 0.8353,
+      "step": 855
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.381585677749361e-05,
+      "loss": 0.0333,
+      "step": 856
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3808184143222507e-05,
+      "loss": 0.0216,
+      "step": 857
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3800511508951405e-05,
+      "loss": 0.9688,
+      "step": 858
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3792838874680307e-05,
+      "loss": 1.0078,
+      "step": 859
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.378516624040921e-05,
+      "loss": 0.0183,
+      "step": 860
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3777493606138107e-05,
+      "loss": 0.0183,
+      "step": 861
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3769820971867008e-05,
+      "loss": 0.9786,
+      "step": 862
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3762148337595907e-05,
+      "loss": 0.9166,
+      "step": 863
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3754475703324808e-05,
+      "loss": 1.1408,
+      "step": 864
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.374680306905371e-05,
+      "loss": 1.1394,
+      "step": 865
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.3739130434782608e-05,
+      "loss": 0.0185,
+      "step": 866
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.373145780051151e-05,
+      "loss": 0.9979,
+      "step": 867
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.372378516624041e-05,
+      "loss": 1.0092,
+      "step": 868
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.371611253196931e-05,
+      "loss": 0.7276,
+      "step": 869
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.370843989769821e-05,
+      "loss": 0.0359,
+      "step": 870
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3700767263427113e-05,
+      "loss": 0.0864,
+      "step": 871
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.369309462915601e-05,
+      "loss": 0.0246,
+      "step": 872
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3685421994884912e-05,
+      "loss": 0.8518,
+      "step": 873
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.367774936061381e-05,
+      "loss": 1.433,
+      "step": 874
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3670076726342712e-05,
+      "loss": 0.035,
+      "step": 875
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3662404092071614e-05,
+      "loss": 0.5922,
+      "step": 876
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3654731457800512e-05,
+      "loss": 0.529,
+      "step": 877
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.364705882352941e-05,
+      "loss": 0.6423,
+      "step": 878
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3639386189258315e-05,
+      "loss": 0.0337,
+      "step": 879
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3631713554987214e-05,
+      "loss": 0.0402,
+      "step": 880
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3624040920716112e-05,
+      "loss": 0.0782,
+      "step": 881
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3616368286445013e-05,
+      "loss": 0.4141,
+      "step": 882
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3608695652173915e-05,
+      "loss": 0.0359,
+      "step": 883
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3601023017902813e-05,
+      "loss": 0.4509,
+      "step": 884
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3593350383631715e-05,
+      "loss": 0.6414,
+      "step": 885
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3585677749360613e-05,
+      "loss": 0.6873,
+      "step": 886
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3578005115089515e-05,
+      "loss": 0.0304,
+      "step": 887
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3570332480818416e-05,
+      "loss": 0.0315,
+      "step": 888
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3562659846547315e-05,
+      "loss": 0.0896,
+      "step": 889
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3554987212276213e-05,
+      "loss": 0.0201,
+      "step": 890
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3547314578005118e-05,
+      "loss": 0.027,
+      "step": 891
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3539641943734016e-05,
+      "loss": 1.9853,
+      "step": 892
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3531969309462914e-05,
+      "loss": 2.9647,
+      "step": 893
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.352429667519182e-05,
+      "loss": 1.9403,
+      "step": 894
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3516624040920717e-05,
+      "loss": 0.0193,
+      "step": 895
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3508951406649616e-05,
+      "loss": 0.9912,
+      "step": 896
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3501278772378517e-05,
+      "loss": 0.02,
+      "step": 897
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.349360613810742e-05,
+      "loss": 0.9015,
+      "step": 898
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3485933503836317e-05,
+      "loss": 0.1294,
+      "step": 899
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.347826086956522e-05,
+      "loss": 0.0559,
+      "step": 900
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.3470588235294117e-05,
+      "loss": 1.8039,
+      "step": 901
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.346291560102302e-05,
+      "loss": 0.0336,
+      "step": 902
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.345524296675192e-05,
+      "loss": 0.0383,
+      "step": 903
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.344757033248082e-05,
+      "loss": 0.0722,
+      "step": 904
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3439897698209717e-05,
+      "loss": 0.0278,
+      "step": 905
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.343222506393862e-05,
+      "loss": 0.0684,
+      "step": 906
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.342455242966752e-05,
+      "loss": 0.0236,
+      "step": 907
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3416879795396418e-05,
+      "loss": 0.0348,
+      "step": 908
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340920716112532e-05,
+      "loss": 0.0204,
+      "step": 909
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340153452685422e-05,
+      "loss": 0.8109,
+      "step": 910
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.339386189258312e-05,
+      "loss": 0.0279,
+      "step": 911
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.338618925831202e-05,
+      "loss": 0.0361,
+      "step": 912
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337851662404092e-05,
+      "loss": 0.0355,
+      "step": 913
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337084398976982e-05,
+      "loss": 0.7958,
+      "step": 914
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.3363171355498723e-05,
+      "loss": 0.0362,
+      "step": 915
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.335549872122762e-05,
+      "loss": 0.0252,
+      "step": 916
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3347826086956522e-05,
+      "loss": 0.8165,
+      "step": 917
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3340153452685424e-05,
+      "loss": 0.0237,
+      "step": 918
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3332480818414322e-05,
+      "loss": 0.8162,
+      "step": 919
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.332480818414322e-05,
+      "loss": 0.7786,
+      "step": 920
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3317135549872125e-05,
+      "loss": 0.0787,
+      "step": 921
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3309462915601024e-05,
+      "loss": 0.1764,
+      "step": 922
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3301790281329925e-05,
+      "loss": 0.7449,
+      "step": 923
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3294117647058824e-05,
+      "loss": 0.0452,
+      "step": 924
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3286445012787725e-05,
+      "loss": 0.0181,
+      "step": 925
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3278772378516627e-05,
+      "loss": 0.1783,
+      "step": 926
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3271099744245525e-05,
+      "loss": 0.5158,
+      "step": 927
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3263427109974423e-05,
+      "loss": 0.7574,
+      "step": 928
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3255754475703328e-05,
+      "loss": 2.6099,
+      "step": 929
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3248081841432226e-05,
+      "loss": 0.0304,
+      "step": 930
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3240409207161125e-05,
+      "loss": 0.0795,
+      "step": 931
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3232736572890026e-05,
+      "loss": 0.0342,
+      "step": 932
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3225063938618928e-05,
+      "loss": 0.0473,
+      "step": 933
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3217391304347826e-05,
+      "loss": 0.0831,
+      "step": 934
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3209718670076728e-05,
+      "loss": 0.0465,
+      "step": 935
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3202046035805626e-05,
+      "loss": 0.6769,
+      "step": 936
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3194373401534528e-05,
+      "loss": 0.0173,
+      "step": 937
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.318670076726343e-05,
+      "loss": 0.0169,
+      "step": 938
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3179028132992327e-05,
+      "loss": 0.0469,
+      "step": 939
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3171355498721226e-05,
+      "loss": 0.048,
+      "step": 940
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.316368286445013e-05,
+      "loss": 1.0704,
+      "step": 941
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.315601023017903e-05,
+      "loss": 1.0738,
+      "step": 942
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3148337595907927e-05,
+      "loss": 0.0163,
+      "step": 943
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3140664961636832e-05,
+      "loss": 2.0702,
+      "step": 944
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.313299232736573e-05,
+      "loss": 1.0433,
+      "step": 945
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.312531969309463e-05,
+      "loss": 1.0351,
+      "step": 946
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.311764705882353e-05,
+      "loss": 1.053,
+      "step": 947
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.3109974424552432e-05,
+      "loss": 0.1797,
+      "step": 948
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.310230179028133e-05,
+      "loss": 1.0259,
+      "step": 949
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.309462915601023e-05,
+      "loss": 0.03,
+      "step": 950
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.308695652173913e-05,
+      "loss": 0.905,
+      "step": 951
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.307928388746803e-05,
+      "loss": 0.0699,
+      "step": 952
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3071611253196933e-05,
+      "loss": 1.0013,
+      "step": 953
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.306393861892583e-05,
+      "loss": 0.0319,
+      "step": 954
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.305626598465473e-05,
+      "loss": 0.0361,
+      "step": 955
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3048593350383634e-05,
+      "loss": 0.0201,
+      "step": 956
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3040920716112533e-05,
+      "loss": 0.033,
+      "step": 957
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.303324808184143e-05,
+      "loss": 0.0411,
+      "step": 958
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3025575447570332e-05,
+      "loss": 0.7138,
+      "step": 959
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3017902813299234e-05,
+      "loss": 0.0434,
+      "step": 960
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3010230179028132e-05,
+      "loss": 0.8445,
+      "step": 961
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3002557544757034e-05,
+      "loss": 0.0323,
+      "step": 962
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2994884910485932e-05,
+      "loss": 0.0203,
+      "step": 963
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2987212276214834e-05,
+      "loss": 0.2302,
+      "step": 964
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2979539641943735e-05,
+      "loss": 0.0293,
+      "step": 965
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2971867007672634e-05,
+      "loss": 0.0593,
+      "step": 966
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2964194373401535e-05,
+      "loss": 0.6505,
+      "step": 967
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2956521739130437e-05,
+      "loss": 0.5891,
+      "step": 968
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2948849104859335e-05,
+      "loss": 0.0191,
+      "step": 969
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2941176470588233e-05,
+      "loss": 0.0188,
+      "step": 970
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2933503836317138e-05,
+      "loss": 0.9537,
+      "step": 971
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2925831202046037e-05,
+      "loss": 0.0186,
+      "step": 972
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2918158567774935e-05,
+      "loss": 1.0237,
+      "step": 973
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2910485933503836e-05,
+      "loss": 0.0417,
+      "step": 974
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2902813299232738e-05,
+      "loss": 0.0369,
+      "step": 975
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.289514066496164e-05,
+      "loss": 0.0184,
+      "step": 976
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2887468030690538e-05,
+      "loss": 0.811,
+      "step": 977
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2879795396419436e-05,
+      "loss": 0.0183,
+      "step": 978
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.287212276214834e-05,
+      "loss": 1.0442,
+      "step": 979
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.286445012787724e-05,
+      "loss": 0.027,
+      "step": 980
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.2856777493606137e-05,
+      "loss": 0.062,
+      "step": 981
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284910485933504e-05,
+      "loss": 0.0186,
+      "step": 982
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284143222506394e-05,
+      "loss": 0.6057,
+      "step": 983
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.283375959079284e-05,
+      "loss": 1.6594,
+      "step": 984
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.282608695652174e-05,
+      "loss": 0.0192,
+      "step": 985
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281841432225064e-05,
+      "loss": 1.5361,
+      "step": 986
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281074168797954e-05,
+      "loss": 0.0384,
+      "step": 987
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.2803069053708442e-05,
+      "loss": 2.0238,
+      "step": 988
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.279539641943734e-05,
+      "loss": 0.0334,
+      "step": 989
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2787723785166242e-05,
+      "loss": 0.0571,
+      "step": 990
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2780051150895143e-05,
+      "loss": 0.0211,
+      "step": 991
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.277237851662404e-05,
+      "loss": 0.0409,
+      "step": 992
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.276470588235294e-05,
+      "loss": 0.1059,
+      "step": 993
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2757033248081845e-05,
+      "loss": 0.0277,
+      "step": 994
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2749360613810743e-05,
+      "loss": 1.0023,
+      "step": 995
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.274168797953964e-05,
+      "loss": 0.0186,
+      "step": 996
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2734015345268543e-05,
+      "loss": 0.0345,
+      "step": 997
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2726342710997445e-05,
+      "loss": 0.0169,
+      "step": 998
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2718670076726343e-05,
+      "loss": 0.7292,
+      "step": 999
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2710997442455244e-05,
+      "loss": 1.2655,
+      "step": 1000
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2703324808184143e-05,
+      "loss": 0.03,
+      "step": 1001
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2695652173913044e-05,
+      "loss": 1.0266,
+      "step": 1002
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2687979539641946e-05,
+      "loss": 0.7645,
+      "step": 1003
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2680306905370844e-05,
+      "loss": 0.7985,
+      "step": 1004
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2672634271099742e-05,
+      "loss": 0.76,
+      "step": 1005
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2664961636828647e-05,
+      "loss": 0.7577,
+      "step": 1006
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2657289002557545e-05,
+      "loss": 0.0248,
+      "step": 1007
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2649616368286444e-05,
+      "loss": 0.0262,
+      "step": 1008
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2641943734015345e-05,
+      "loss": 1.9984,
+      "step": 1009
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2634271099744247e-05,
+      "loss": 0.1489,
+      "step": 1010
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2626598465473145e-05,
+      "loss": 0.0683,
+      "step": 1011
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2618925831202047e-05,
+      "loss": 0.054,
+      "step": 1012
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2611253196930945e-05,
+      "loss": 0.0269,
+      "step": 1013
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2603580562659847e-05,
+      "loss": 1.0378,
+      "step": 1014
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2595907928388748e-05,
+      "loss": 0.0171,
+      "step": 1015
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2588235294117646e-05,
+      "loss": 0.8433,
+      "step": 1016
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2580562659846548e-05,
+      "loss": 0.0468,
+      "step": 1017
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.257289002557545e-05,
+      "loss": 0.0331,
+      "step": 1018
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2565217391304348e-05,
+      "loss": 0.0308,
+      "step": 1019
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2557544757033246e-05,
+      "loss": 1.0007,
+      "step": 1020
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254987212276215e-05,
+      "loss": 2.0242,
+      "step": 1021
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254219948849105e-05,
+      "loss": 0.7713,
+      "step": 1022
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2534526854219948e-05,
+      "loss": 1.1112,
+      "step": 1023
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.252685421994885e-05,
+      "loss": 0.0184,
+      "step": 1024
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251918158567775e-05,
+      "loss": 0.0243,
+      "step": 1025
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251150895140665e-05,
+      "loss": 0.046,
+      "step": 1026
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.250383631713555e-05,
+      "loss": 0.0378,
+      "step": 1027
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.249616368286445e-05,
+      "loss": 0.0329,
+      "step": 1028
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2488491048593354e-05,
+      "loss": 0.0245,
+      "step": 1029
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2480818414322252e-05,
+      "loss": 0.0238,
+      "step": 1030
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.247314578005115e-05,
+      "loss": 1.0628,
+      "step": 1031
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2465473145780052e-05,
+      "loss": 0.0703,
+      "step": 1032
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2457800511508954e-05,
+      "loss": 0.0251,
+      "step": 1033
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2450127877237852e-05,
+      "loss": 1.0561,
+      "step": 1034
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2442455242966753e-05,
+      "loss": 0.0274,
+      "step": 1035
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.243478260869565e-05,
+      "loss": 0.023,
+      "step": 1036
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2427109974424553e-05,
+      "loss": 0.0223,
+      "step": 1037
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2419437340153455e-05,
+      "loss": 0.8113,
+      "step": 1038
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2411764705882353e-05,
+      "loss": 0.0346,
+      "step": 1039
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2404092071611255e-05,
+      "loss": 0.0164,
+      "step": 1040
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2396419437340156e-05,
+      "loss": 0.0237,
+      "step": 1041
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2388746803069054e-05,
+      "loss": 0.0305,
+      "step": 1042
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2381074168797953e-05,
+      "loss": 0.0158,
+      "step": 1043
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2373401534526858e-05,
+      "loss": 0.0225,
+      "step": 1044
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2365728900255756e-05,
+      "loss": 0.0226,
+      "step": 1045
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2358056265984654e-05,
+      "loss": 0.015,
+      "step": 1046
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2350383631713556e-05,
+      "loss": 0.0219,
+      "step": 1047
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2342710997442457e-05,
+      "loss": 1.0694,
+      "step": 1048
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2335038363171356e-05,
+      "loss": 0.0348,
+      "step": 1049
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2327365728900257e-05,
+      "loss": 0.0213,
+      "step": 1050
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2319693094629155e-05,
+      "loss": 0.9405,
+      "step": 1051
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2312020460358057e-05,
+      "loss": 0.0252,
+      "step": 1052
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.230434782608696e-05,
+      "loss": 0.0254,
+      "step": 1053
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2296675191815857e-05,
+      "loss": 0.7353,
+      "step": 1054
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2289002557544755e-05,
+      "loss": 0.0253,
+      "step": 1055
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.228132992327366e-05,
+      "loss": 0.8361,
+      "step": 1056
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2273657289002558e-05,
+      "loss": 0.0196,
+      "step": 1057
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2265984654731457e-05,
+      "loss": 0.014,
+      "step": 1058
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2258312020460358e-05,
+      "loss": 0.8297,
+      "step": 1059
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.225063938618926e-05,
+      "loss": 1.0654,
+      "step": 1060
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2242966751918158e-05,
+      "loss": 0.0248,
+      "step": 1061
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.223529411764706e-05,
+      "loss": 0.0226,
+      "step": 1062
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2227621483375958e-05,
+      "loss": 0.0238,
+      "step": 1063
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221994884910486e-05,
+      "loss": 0.0212,
+      "step": 1064
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221227621483376e-05,
+      "loss": 0.0274,
+      "step": 1065
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.220460358056266e-05,
+      "loss": 1.0217,
+      "step": 1066
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.219693094629156e-05,
+      "loss": 0.0252,
+      "step": 1067
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2189258312020462e-05,
+      "loss": 0.0219,
+      "step": 1068
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.218158567774936e-05,
+      "loss": 0.0349,
+      "step": 1069
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.217391304347826e-05,
+      "loss": 0.0152,
+      "step": 1070
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2166240409207164e-05,
+      "loss": 0.021,
+      "step": 1071
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2158567774936062e-05,
+      "loss": 0.015,
+      "step": 1072
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.215089514066496e-05,
+      "loss": 0.0277,
+      "step": 1073
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2143222506393862e-05,
+      "loss": 0.1858,
+      "step": 1074
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2135549872122764e-05,
+      "loss": 0.0136,
+      "step": 1075
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2127877237851662e-05,
+      "loss": 0.0132,
+      "step": 1076
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2120204603580563e-05,
+      "loss": 0.0126,
+      "step": 1077
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.211253196930946e-05,
+      "loss": 0.5171,
+      "step": 1078
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2104859335038363e-05,
+      "loss": 0.0422,
+      "step": 1079
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2097186700767265e-05,
+      "loss": 0.0186,
+      "step": 1080
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2089514066496163e-05,
+      "loss": 0.8412,
+      "step": 1081
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2081841432225065e-05,
+      "loss": 0.0183,
+      "step": 1082
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2074168797953966e-05,
+      "loss": 1.0243,
+      "step": 1083
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2066496163682865e-05,
+      "loss": 0.0212,
+      "step": 1084
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2058823529411766e-05,
+      "loss": 0.2841,
+      "step": 1085
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2051150895140664e-05,
+      "loss": 0.0118,
+      "step": 1086
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2043478260869566e-05,
+      "loss": 0.0247,
+      "step": 1087
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2035805626598468e-05,
+      "loss": 1.5121,
+      "step": 1088
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2028132992327366e-05,
+      "loss": 0.0304,
+      "step": 1089
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2020460358056267e-05,
+      "loss": 0.0329,
+      "step": 1090
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.201278772378517e-05,
+      "loss": 1.1288,
+      "step": 1091
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.2005115089514067e-05,
+      "loss": 0.0145,
+      "step": 1092
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.1997442455242966e-05,
+      "loss": 0.0498,
+      "step": 1093
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.198976982097187e-05,
+      "loss": 0.0244,
+      "step": 1094
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.198209718670077e-05,
+      "loss": 0.0112,
+      "step": 1095
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.1974424552429667e-05,
+      "loss": 0.8642,
+      "step": 1096
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.196675191815857e-05,
+      "loss": 0.8822,
+      "step": 1097
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.195907928388747e-05,
+      "loss": 0.0278,
+      "step": 1098
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.195140664961637e-05,
+      "loss": 0.6954,
+      "step": 1099
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.194373401534527e-05,
+      "loss": 0.0179,
+      "step": 1100
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.1936061381074168e-05,
+      "loss": 0.5607,
+      "step": 1101
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.192838874680307e-05,
+      "loss": 0.0191,
+      "step": 1102
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.192071611253197e-05,
+      "loss": 1.9519,
+      "step": 1103
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.191304347826087e-05,
+      "loss": 0.8347,
+      "step": 1104
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1905370843989768e-05,
+      "loss": 0.0262,
+      "step": 1105
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1897698209718673e-05,
+      "loss": 0.0477,
+      "step": 1106
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.189002557544757e-05,
+      "loss": 0.0227,
+      "step": 1107
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.188235294117647e-05,
+      "loss": 0.0205,
+      "step": 1108
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.187468030690537e-05,
+      "loss": 0.0208,
+      "step": 1109
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.1867007672634273e-05,
+      "loss": 0.0116,
+      "step": 1110
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.185933503836317e-05,
+      "loss": 1.1546,
+      "step": 1111
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1851662404092072e-05,
+      "loss": 0.0114,
+      "step": 1112
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.184398976982097e-05,
+      "loss": 0.0113,
+      "step": 1113
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1836317135549872e-05,
+      "loss": 0.6881,
+      "step": 1114
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1828644501278774e-05,
+      "loss": 0.0294,
+      "step": 1115
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1820971867007672e-05,
+      "loss": 0.1569,
+      "step": 1116
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1813299232736574e-05,
+      "loss": 1.1362,
+      "step": 1117
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1805626598465475e-05,
+      "loss": 0.0196,
+      "step": 1118
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1797953964194374e-05,
+      "loss": 0.0205,
+      "step": 1119
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1790281329923272e-05,
+      "loss": 0.0198,
+      "step": 1120
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1782608695652177e-05,
+      "loss": 0.0113,
+      "step": 1121
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1774936061381075e-05,
+      "loss": 0.02,
+      "step": 1122
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1767263427109973e-05,
+      "loss": 0.0195,
+      "step": 1123
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1759590792838875e-05,
+      "loss": 0.8016,
+      "step": 1124
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1751918158567776e-05,
+      "loss": 0.0122,
+      "step": 1125
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1744245524296675e-05,
+      "loss": 1.1132,
+      "step": 1126
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1736572890025576e-05,
+      "loss": 0.011,
+      "step": 1127
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1728900255754474e-05,
+      "loss": 0.7705,
+      "step": 1128
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1721227621483376e-05,
+      "loss": 1.1467,
+      "step": 1129
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1713554987212278e-05,
+      "loss": 0.0176,
+      "step": 1130
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1705882352941176e-05,
+      "loss": 0.0293,
+      "step": 1131
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1698209718670074e-05,
+      "loss": 0.0208,
+      "step": 1132
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.169053708439898e-05,
+      "loss": 0.0199,
+      "step": 1133
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1682864450127877e-05,
+      "loss": 0.0431,
+      "step": 1134
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.167519181585678e-05,
+      "loss": 1.6066,
+      "step": 1135
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.1667519181585677e-05,
+      "loss": 0.029,
+      "step": 1136
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165984654731458e-05,
+      "loss": 0.029,
+      "step": 1137
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165217391304348e-05,
+      "loss": 1.1749,
+      "step": 1138
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.164450127877238e-05,
+      "loss": 0.0189,
+      "step": 1139
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.163682864450128e-05,
+      "loss": 0.0309,
+      "step": 1140
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.1629156010230182e-05,
+      "loss": 0.0287,
+      "step": 1141
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.162148337595908e-05,
+      "loss": 0.0202,
+      "step": 1142
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.161381074168798e-05,
+      "loss": 0.7889,
+      "step": 1143
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.1606138107416883e-05,
+      "loss": 0.0285,
+      "step": 1144
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159846547314578e-05,
+      "loss": 0.0535,
+      "step": 1145
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159079283887468e-05,
+      "loss": 1.0829,
+      "step": 1146
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.158312020460358e-05,
+      "loss": 0.0989,
+      "step": 1147
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1575447570332483e-05,
+      "loss": 0.0115,
+      "step": 1148
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.156777493606138e-05,
+      "loss": 0.0227,
+      "step": 1149
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1560102301790283e-05,
+      "loss": 0.5506,
+      "step": 1150
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.155242966751918e-05,
+      "loss": 1.1234,
+      "step": 1151
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1544757033248083e-05,
+      "loss": 0.0564,
+      "step": 1152
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1537084398976984e-05,
+      "loss": 0.0127,
+      "step": 1153
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1529411764705882e-05,
+      "loss": 0.0218,
+      "step": 1154
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.152173913043478e-05,
+      "loss": 0.0124,
+      "step": 1155
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1514066496163686e-05,
+      "loss": 0.0223,
+      "step": 1156
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1506393861892584e-05,
+      "loss": 0.0505,
+      "step": 1157
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1498721227621482e-05,
+      "loss": 0.0225,
+      "step": 1158
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1491048593350384e-05,
+      "loss": 0.0308,
+      "step": 1159
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1483375959079285e-05,
+      "loss": 0.036,
+      "step": 1160
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1475703324808184e-05,
+      "loss": 0.0121,
+      "step": 1161
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1468030690537085e-05,
+      "loss": 0.0274,
+      "step": 1162
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1460358056265983e-05,
+      "loss": 1.0397,
+      "step": 1163
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1452685421994885e-05,
+      "loss": 0.0332,
+      "step": 1164
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1445012787723787e-05,
+      "loss": 0.0182,
+      "step": 1165
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1437340153452685e-05,
+      "loss": 0.018,
+      "step": 1166
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1429667519181587e-05,
+      "loss": 0.859,
+      "step": 1167
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1421994884910488e-05,
+      "loss": 0.0321,
+      "step": 1168
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1414322250639386e-05,
+      "loss": 0.0225,
+      "step": 1169
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1406649616368285e-05,
+      "loss": 0.0224,
+      "step": 1170
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.139897698209719e-05,
+      "loss": 0.03,
+      "step": 1171
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1391304347826088e-05,
+      "loss": 0.7473,
+      "step": 1172
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1383631713554986e-05,
+      "loss": 0.8609,
+      "step": 1173
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1375959079283888e-05,
+      "loss": 0.0105,
+      "step": 1174
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.136828644501279e-05,
+      "loss": 0.0217,
+      "step": 1175
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1360613810741687e-05,
+      "loss": 0.0218,
+      "step": 1176
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.135294117647059e-05,
+      "loss": 0.016,
+      "step": 1177
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1345268542199487e-05,
+      "loss": 0.7844,
+      "step": 1178
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.133759590792839e-05,
+      "loss": 0.01,
+      "step": 1179
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132992327365729e-05,
+      "loss": 0.8774,
+      "step": 1180
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132225063938619e-05,
+      "loss": 0.8375,
+      "step": 1181
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.1314578005115087e-05,
+      "loss": 0.0188,
+      "step": 1182
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.1306905370843992e-05,
+      "loss": 1.1026,
+      "step": 1183
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129923273657289e-05,
+      "loss": 1.0447,
+      "step": 1184
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129156010230179e-05,
+      "loss": 0.0101,
+      "step": 1185
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.128388746803069e-05,
+      "loss": 0.0247,
+      "step": 1186
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.127621483375959e-05,
+      "loss": 0.0295,
+      "step": 1187
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.1268542199488493e-05,
+      "loss": 0.0191,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.37390685081481934,
+      "eval_runtime": 57.4087,
+      "eval_samples_per_second": 6.898,
+      "eval_steps_per_second": 1.724,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.126086956521739e-05,
+      "loss": 0.0103,
+      "step": 1189
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1253196930946293e-05,
+      "loss": 0.0231,
+      "step": 1190
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1245524296675195e-05,
+      "loss": 0.0389,
+      "step": 1191
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1237851662404093e-05,
+      "loss": 0.0298,
+      "step": 1192
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.123017902813299e-05,
+      "loss": 0.0196,
+      "step": 1193
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1222506393861896e-05,
+      "loss": 0.9688,
+      "step": 1194
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1214833759590794e-05,
+      "loss": 0.0458,
+      "step": 1195
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1207161125319693e-05,
+      "loss": 0.0161,
+      "step": 1196
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1199488491048594e-05,
+      "loss": 0.0157,
+      "step": 1197
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1191815856777496e-05,
+      "loss": 0.8732,
+      "step": 1198
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1184143222506394e-05,
+      "loss": 0.0103,
+      "step": 1199
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1176470588235296e-05,
+      "loss": 0.0267,
+      "step": 1200
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1168797953964194e-05,
+      "loss": 0.0266,
+      "step": 1201
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1161125319693095e-05,
+      "loss": 0.0157,
+      "step": 1202
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1153452685421997e-05,
+      "loss": 0.265,
+      "step": 1203
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1145780051150895e-05,
+      "loss": 0.0537,
+      "step": 1204
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1138107416879794e-05,
+      "loss": 0.0306,
+      "step": 1205
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.11304347826087e-05,
+      "loss": 1.1714,
+      "step": 1206
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1122762148337597e-05,
+      "loss": 1.1704,
+      "step": 1207
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1115089514066495e-05,
+      "loss": 0.0154,
+      "step": 1208
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1107416879795397e-05,
+      "loss": 0.0195,
+      "step": 1209
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1099744245524298e-05,
+      "loss": 0.9099,
+      "step": 1210
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1092071611253196e-05,
+      "loss": 0.0112,
+      "step": 1211
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1084398976982098e-05,
+      "loss": 0.1506,
+      "step": 1212
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1076726342710996e-05,
+      "loss": 0.0111,
+      "step": 1213
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1069053708439898e-05,
+      "loss": 0.0244,
+      "step": 1214
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10613810741688e-05,
+      "loss": 0.8889,
+      "step": 1215
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1053708439897698e-05,
+      "loss": 0.0146,
+      "step": 1216
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10460358056266e-05,
+      "loss": 0.0189,
+      "step": 1217
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10383631713555e-05,
+      "loss": 0.0097,
+      "step": 1218
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10306905370844e-05,
+      "loss": 1.1683,
+      "step": 1219
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1023017902813297e-05,
+      "loss": 1.1621,
+      "step": 1220
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1015345268542202e-05,
+      "loss": 0.0194,
+      "step": 1221
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.10076726342711e-05,
+      "loss": 0.0155,
+      "step": 1222
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.0296,
+      "step": 1223
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.09923273657289e-05,
+      "loss": 0.02,
+      "step": 1224
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.0984654731457802e-05,
+      "loss": 0.0154,
+      "step": 1225
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09769820971867e-05,
+      "loss": 0.0151,
+      "step": 1226
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0969309462915602e-05,
+      "loss": 0.0223,
+      "step": 1227
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09616368286445e-05,
+      "loss": 1.7159,
+      "step": 1228
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0953964194373402e-05,
+      "loss": 0.0194,
+      "step": 1229
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0946291560102303e-05,
+      "loss": 0.0637,
+      "step": 1230
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09386189258312e-05,
+      "loss": 0.0156,
+      "step": 1231
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09309462915601e-05,
+      "loss": 1.0612,
+      "step": 1232
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0923273657289005e-05,
+      "loss": 0.8929,
+      "step": 1233
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0915601023017903e-05,
+      "loss": 1.1649,
+      "step": 1234
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.09079283887468e-05,
+      "loss": 2.059,
+      "step": 1235
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0900255754475703e-05,
+      "loss": 0.0169,
+      "step": 1236
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0892583120204604e-05,
+      "loss": 0.0158,
+      "step": 1237
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0884910485933503e-05,
+      "loss": 0.0108,
+      "step": 1238
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0877237851662404e-05,
+      "loss": 0.0152,
+      "step": 1239
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0869565217391306e-05,
+      "loss": 0.0109,
+      "step": 1240
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0861892583120208e-05,
+      "loss": 1.1352,
+      "step": 1241
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0854219948849106e-05,
+      "loss": 0.0226,
+      "step": 1242
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0846547314578004e-05,
+      "loss": 0.0154,
+      "step": 1243
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.083887468030691e-05,
+      "loss": 0.0232,
+      "step": 1244
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0831202046035807e-05,
+      "loss": 0.0142,
+      "step": 1245
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0823529411764705e-05,
+      "loss": 0.9143,
+      "step": 1246
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0815856777493607e-05,
+      "loss": 0.0165,
+      "step": 1247
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.080818414322251e-05,
+      "loss": 0.0446,
+      "step": 1248
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0800511508951407e-05,
+      "loss": 0.0121,
+      "step": 1249
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.079283887468031e-05,
+      "loss": 1.0991,
+      "step": 1250
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.0785166240409207e-05,
+      "loss": 0.0159,
+      "step": 1251
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.077749360613811e-05,
+      "loss": 0.0166,
+      "step": 1252
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.076982097186701e-05,
+      "loss": 0.0171,
+      "step": 1253
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0762148337595908e-05,
+      "loss": 0.7986,
+      "step": 1254
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0754475703324806e-05,
+      "loss": 0.0174,
+      "step": 1255
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.074680306905371e-05,
+      "loss": 1.0431,
+      "step": 1256
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.073913043478261e-05,
+      "loss": 0.1766,
+      "step": 1257
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.0731457800511508e-05,
+      "loss": 0.0209,
+      "step": 1258
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.072378516624041e-05,
+      "loss": 0.0178,
+      "step": 1259
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.071611253196931e-05,
+      "loss": 0.0215,
+      "step": 1260
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.070843989769821e-05,
+      "loss": 0.0112,
+      "step": 1261
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.070076726342711e-05,
+      "loss": 0.0166,
+      "step": 1262
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0693094629156012e-05,
+      "loss": 0.0161,
+      "step": 1263
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.068542199488491e-05,
+      "loss": 0.8922,
+      "step": 1264
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0677749360613812e-05,
+      "loss": 0.0133,
+      "step": 1265
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.067007672634271e-05,
+      "loss": 0.0117,
+      "step": 1266
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0662404092071612e-05,
+      "loss": 0.0153,
+      "step": 1267
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0654731457800514e-05,
+      "loss": 0.0153,
+      "step": 1268
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0647058823529412e-05,
+      "loss": 0.0195,
+      "step": 1269
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.063938618925831e-05,
+      "loss": 1.1511,
+      "step": 1270
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0631713554987215e-05,
+      "loss": 0.0108,
+      "step": 1271
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0624040920716113e-05,
+      "loss": 0.8897,
+      "step": 1272
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.061636828644501e-05,
+      "loss": 0.0157,
+      "step": 1273
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0608695652173913e-05,
+      "loss": 0.0108,
+      "step": 1274
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0601023017902815e-05,
+      "loss": 0.0214,
+      "step": 1275
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0593350383631713e-05,
+      "loss": 0.0186,
+      "step": 1276
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0585677749360615e-05,
+      "loss": 0.0182,
+      "step": 1277
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0578005115089513e-05,
+      "loss": 0.0102,
+      "step": 1278
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0570332480818415e-05,
+      "loss": 1.1313,
+      "step": 1279
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0562659846547316e-05,
+      "loss": 0.0103,
+      "step": 1280
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0554987212276214e-05,
+      "loss": 0.0144,
+      "step": 1281
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0547314578005113e-05,
+      "loss": 0.02,
+      "step": 1282
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0539641943734018e-05,
+      "loss": 0.0292,
+      "step": 1283
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0531969309462916e-05,
+      "loss": 0.0105,
+      "step": 1284
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0524296675191814e-05,
+      "loss": 0.01,
+      "step": 1285
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0516624040920716e-05,
+      "loss": 1.1245,
+      "step": 1286
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0508951406649617e-05,
+      "loss": 0.8544,
+      "step": 1287
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0501278772378516e-05,
+      "loss": 0.0149,
+      "step": 1288
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0493606138107417e-05,
+      "loss": 0.0065,
+      "step": 1289
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.048593350383632e-05,
+      "loss": 0.0113,
+      "step": 1290
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0478260869565217e-05,
+      "loss": 0.0094,
+      "step": 1291
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.047058823529412e-05,
+      "loss": 0.0097,
+      "step": 1292
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0462915601023017e-05,
+      "loss": 0.0812,
+      "step": 1293
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0455242966751922e-05,
+      "loss": 0.0099,
+      "step": 1294
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.044757033248082e-05,
+      "loss": 0.013,
+      "step": 1295
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0439897698209718e-05,
+      "loss": 1.584,
+      "step": 1296
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.043222506393862e-05,
+      "loss": 0.0089,
+      "step": 1297
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.042455242966752e-05,
+      "loss": 0.0089,
+      "step": 1298
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.041687979539642e-05,
+      "loss": 0.0093,
+      "step": 1299
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.040920716112532e-05,
+      "loss": 0.0149,
+      "step": 1300
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.040153452685422e-05,
+      "loss": 1.1671,
+      "step": 1301
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.039386189258312e-05,
+      "loss": 0.0086,
+      "step": 1302
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.0386189258312023e-05,
+      "loss": 0.8177,
+      "step": 1303
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.037851662404092e-05,
+      "loss": 0.0128,
+      "step": 1304
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.037084398976982e-05,
+      "loss": 0.0167,
+      "step": 1305
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0363171355498724e-05,
+      "loss": 0.0085,
+      "step": 1306
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0355498721227622e-05,
+      "loss": 0.0085,
+      "step": 1307
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.034782608695652e-05,
+      "loss": 0.0166,
+      "step": 1308
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0340153452685422e-05,
+      "loss": 0.9448,
+      "step": 1309
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0332480818414324e-05,
+      "loss": 0.0082,
+      "step": 1310
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0324808184143222e-05,
+      "loss": 0.0088,
+      "step": 1311
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0317135549872124e-05,
+      "loss": 0.0082,
+      "step": 1312
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0309462915601025e-05,
+      "loss": 0.0146,
+      "step": 1313
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0301790281329924e-05,
+      "loss": 0.0123,
+      "step": 1314
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0294117647058825e-05,
+      "loss": 0.0101,
+      "step": 1315
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0286445012787723e-05,
+      "loss": 1.2167,
+      "step": 1316
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0278772378516625e-05,
+      "loss": 1.0578,
+      "step": 1317
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0271099744245527e-05,
+      "loss": 0.0163,
+      "step": 1318
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0263427109974425e-05,
+      "loss": 0.013,
+      "step": 1319
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0255754475703323e-05,
+      "loss": 0.0082,
+      "step": 1320
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0248081841432228e-05,
+      "loss": 0.0081,
+      "step": 1321
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0240409207161126e-05,
+      "loss": 0.0124,
+      "step": 1322
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0232736572890024e-05,
+      "loss": 0.0206,
+      "step": 1323
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0225063938618926e-05,
+      "loss": 0.0081,
+      "step": 1324
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0217391304347828e-05,
+      "loss": 0.0122,
+      "step": 1325
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0209718670076726e-05,
+      "loss": 0.0122,
+      "step": 1326
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0202046035805628e-05,
+      "loss": 0.0121,
+      "step": 1327
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0194373401534526e-05,
+      "loss": 1.224,
+      "step": 1328
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0186700767263427e-05,
+      "loss": 2.3502,
+      "step": 1329
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.017902813299233e-05,
+      "loss": 0.0242,
+      "step": 1330
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0171355498721227e-05,
+      "loss": 0.0064,
+      "step": 1331
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0163682864450125e-05,
+      "loss": 0.0186,
+      "step": 1332
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.015601023017903e-05,
+      "loss": 0.012,
+      "step": 1333
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.014833759590793e-05,
+      "loss": 0.0119,
+      "step": 1334
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.0140664961636827e-05,
+      "loss": 0.0159,
+      "step": 1335
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.013299232736573e-05,
+      "loss": 0.0199,
+      "step": 1336
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.012531969309463e-05,
+      "loss": 1.2138,
+      "step": 1337
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.011764705882353e-05,
+      "loss": 0.9491,
+      "step": 1338
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010997442455243e-05,
+      "loss": 0.0083,
+      "step": 1339
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010230179028133e-05,
+      "loss": 0.948,
+      "step": 1340
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.009462915601023e-05,
+      "loss": 0.0112,
+      "step": 1341
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.008695652173913e-05,
+      "loss": 0.8921,
+      "step": 1342
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007928388746803e-05,
+      "loss": 0.0194,
+      "step": 1343
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007161125319693e-05,
+      "loss": 1.0273,
+      "step": 1344
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0063938618925833e-05,
+      "loss": 0.013,
+      "step": 1345
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.005626598465473e-05,
+      "loss": 0.0103,
+      "step": 1346
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0048593350383633e-05,
+      "loss": 0.0114,
+      "step": 1347
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0040920716112534e-05,
+      "loss": 0.0199,
+      "step": 1348
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0033248081841433e-05,
+      "loss": 1.183,
+      "step": 1349
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0025575447570334e-05,
+      "loss": 0.0231,
+      "step": 1350
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0017902813299232e-05,
+      "loss": 1.1794,
+      "step": 1351
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0010230179028134e-05,
+      "loss": 0.0096,
+      "step": 1352
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 2.0002557544757036e-05,
+      "loss": 0.9049,
+      "step": 1353
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9994884910485934e-05,
+      "loss": 0.019,
+      "step": 1354
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9987212276214832e-05,
+      "loss": 0.0152,
+      "step": 1355
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9979539641943737e-05,
+      "loss": 0.8834,
+      "step": 1356
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9971867007672635e-05,
+      "loss": 0.01,
+      "step": 1357
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9964194373401533e-05,
+      "loss": 0.0152,
+      "step": 1358
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9956521739130435e-05,
+      "loss": 0.0161,
+      "step": 1359
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9948849104859337e-05,
+      "loss": 1.6778,
+      "step": 1360
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9941176470588235e-05,
+      "loss": 0.0214,
+      "step": 1361
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9933503836317137e-05,
+      "loss": 0.0156,
+      "step": 1362
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9925831202046038e-05,
+      "loss": 0.026,
+      "step": 1363
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9918158567774936e-05,
+      "loss": 0.0259,
+      "step": 1364
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9910485933503838e-05,
+      "loss": 0.0161,
+      "step": 1365
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9902813299232736e-05,
+      "loss": 0.0108,
+      "step": 1366
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9895140664961638e-05,
+      "loss": 0.0149,
+      "step": 1367
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.988746803069054e-05,
+      "loss": 0.0325,
+      "step": 1368
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9879795396419438e-05,
+      "loss": 0.0153,
+      "step": 1369
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9872122762148336e-05,
+      "loss": 1.7678,
+      "step": 1370
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.986445012787724e-05,
+      "loss": 0.0773,
+      "step": 1371
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.985677749360614e-05,
+      "loss": 0.0244,
+      "step": 1372
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.9849104859335037e-05,
+      "loss": 0.0225,
+      "step": 1373
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.984143222506394e-05,
+      "loss": 0.0254,
+      "step": 1374
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.983375959079284e-05,
+      "loss": 0.0205,
+      "step": 1375
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.982608695652174e-05,
+      "loss": 0.0096,
+      "step": 1376
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981841432225064e-05,
+      "loss": 0.0147,
+      "step": 1377
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981074168797954e-05,
+      "loss": 0.0296,
+      "step": 1378
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.980306905370844e-05,
+      "loss": 0.0227,
+      "step": 1379
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.9795396419437342e-05,
+      "loss": 0.899,
+      "step": 1380
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.978772378516624e-05,
+      "loss": 0.019,
+      "step": 1381
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9780051150895138e-05,
+      "loss": 0.0157,
+      "step": 1382
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9772378516624043e-05,
+      "loss": 1.1897,
+      "step": 1383
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.976470588235294e-05,
+      "loss": 0.014,
+      "step": 1384
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.975703324808184e-05,
+      "loss": 0.0208,
+      "step": 1385
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.974936061381074e-05,
+      "loss": 0.0093,
+      "step": 1386
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.9741687979539643e-05,
+      "loss": 0.0244,
+      "step": 1387
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.973401534526854e-05,
+      "loss": 0.028,
+      "step": 1388
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9726342710997443e-05,
+      "loss": 0.0093,
+      "step": 1389
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9718670076726344e-05,
+      "loss": 0.0184,
+      "step": 1390
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9710997442455243e-05,
+      "loss": 0.9462,
+      "step": 1391
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9703324808184144e-05,
+      "loss": 0.0138,
+      "step": 1392
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9695652173913042e-05,
+      "loss": 0.9561,
+      "step": 1393
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9687979539641944e-05,
+      "loss": 0.0152,
+      "step": 1394
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9680306905370846e-05,
+      "loss": 0.0178,
+      "step": 1395
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9672634271099744e-05,
+      "loss": 0.0166,
+      "step": 1396
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9664961636828642e-05,
+      "loss": 0.92,
+      "step": 1397
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9657289002557547e-05,
+      "loss": 0.0253,
+      "step": 1398
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9649616368286445e-05,
+      "loss": 0.0266,
+      "step": 1399
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9641943734015347e-05,
+      "loss": 0.0105,
+      "step": 1400
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9634271099744245e-05,
+      "loss": 0.0208,
+      "step": 1401
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9626598465473147e-05,
+      "loss": 0.8094,
+      "step": 1402
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.961892583120205e-05,
+      "loss": 0.0103,
+      "step": 1403
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9611253196930947e-05,
+      "loss": 0.0179,
+      "step": 1404
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9603580562659845e-05,
+      "loss": 0.0133,
+      "step": 1405
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.959590792838875e-05,
+      "loss": 0.0086,
+      "step": 1406
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9588235294117648e-05,
+      "loss": 0.0216,
+      "step": 1407
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9580562659846546e-05,
+      "loss": 0.0127,
+      "step": 1408
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9572890025575448e-05,
+      "loss": 0.0127,
+      "step": 1409
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.956521739130435e-05,
+      "loss": 0.0209,
+      "step": 1410
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9557544757033248e-05,
+      "loss": 0.0166,
+      "step": 1411
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954987212276215e-05,
+      "loss": 0.0123,
+      "step": 1412
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954219948849105e-05,
+      "loss": 0.0135,
+      "step": 1413
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.953452685421995e-05,
+      "loss": 0.1253,
+      "step": 1414
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.952685421994885e-05,
+      "loss": 0.0098,
+      "step": 1415
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951918158567775e-05,
+      "loss": 0.0091,
+      "step": 1416
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951150895140665e-05,
+      "loss": 0.9327,
+      "step": 1417
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.9503836317135552e-05,
+      "loss": 0.0119,
+      "step": 1418
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.949616368286445e-05,
+      "loss": 0.0153,
+      "step": 1419
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.948849104859335e-05,
+      "loss": 0.0123,
+      "step": 1420
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9480818414322254e-05,
+      "loss": 0.0194,
+      "step": 1421
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9473145780051152e-05,
+      "loss": 1.2311,
+      "step": 1422
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.946547314578005e-05,
+      "loss": 1.8519,
+      "step": 1423
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9457800511508952e-05,
+      "loss": 0.963,
+      "step": 1424
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9450127877237853e-05,
+      "loss": 0.0075,
+      "step": 1425
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.944245524296675e-05,
+      "loss": 0.0042,
+      "step": 1426
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9434782608695653e-05,
+      "loss": 0.0077,
+      "step": 1427
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.942710997442455e-05,
+      "loss": 0.0152,
+      "step": 1428
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9419437340153453e-05,
+      "loss": 0.0152,
+      "step": 1429
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9411764705882355e-05,
+      "loss": 0.019,
+      "step": 1430
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9404092071611253e-05,
+      "loss": 0.0081,
+      "step": 1431
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.939641943734015e-05,
+      "loss": 0.0076,
+      "step": 1432
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9388746803069056e-05,
+      "loss": 0.0086,
+      "step": 1433
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9381074168797954e-05,
+      "loss": 0.0146,
+      "step": 1434
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9373401534526853e-05,
+      "loss": 0.0074,
+      "step": 1435
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9365728900255754e-05,
+      "loss": 0.1007,
+      "step": 1436
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9358056265984656e-05,
+      "loss": 0.011,
+      "step": 1437
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9350383631713554e-05,
+      "loss": 0.0089,
+      "step": 1438
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9342710997442456e-05,
+      "loss": 0.0182,
+      "step": 1439
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9335038363171357e-05,
+      "loss": 0.0139,
+      "step": 1440
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9327365728900255e-05,
+      "loss": 2.2086,
+      "step": 1441
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9319693094629157e-05,
+      "loss": 0.0178,
+      "step": 1442
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9312020460358055e-05,
+      "loss": 0.0142,
+      "step": 1443
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9304347826086957e-05,
+      "loss": 0.0142,
+      "step": 1444
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.929667519181586e-05,
+      "loss": 0.0242,
+      "step": 1445
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9289002557544757e-05,
+      "loss": 0.0087,
+      "step": 1446
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9281329923273655e-05,
+      "loss": 0.0077,
+      "step": 1447
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.927365728900256e-05,
+      "loss": 0.0104,
+      "step": 1448
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9265984654731458e-05,
+      "loss": 0.0139,
+      "step": 1449
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9258312020460356e-05,
+      "loss": 1.2557,
+      "step": 1450
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9250639386189258e-05,
+      "loss": 0.0068,
+      "step": 1451
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.924296675191816e-05,
+      "loss": 0.0163,
+      "step": 1452
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.923529411764706e-05,
+      "loss": 0.9483,
+      "step": 1453
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.922762148337596e-05,
+      "loss": 1.2181,
+      "step": 1454
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.9219948849104858e-05,
+      "loss": 0.011,
+      "step": 1455
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.9212276214833763e-05,
+      "loss": 0.014,
+      "step": 1456
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.920460358056266e-05,
+      "loss": 0.0131,
+      "step": 1457
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.919693094629156e-05,
+      "loss": 0.0135,
+      "step": 1458
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.918925831202046e-05,
+      "loss": 0.007,
+      "step": 1459
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9181585677749362e-05,
+      "loss": 0.0072,
+      "step": 1460
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.917391304347826e-05,
+      "loss": 0.0135,
+      "step": 1461
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9166240409207162e-05,
+      "loss": 0.9908,
+      "step": 1462
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9158567774936064e-05,
+      "loss": 0.0103,
+      "step": 1463
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9150895140664962e-05,
+      "loss": 0.9867,
+      "step": 1464
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9143222506393864e-05,
+      "loss": 0.0138,
+      "step": 1465
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9135549872122762e-05,
+      "loss": 0.0104,
+      "step": 1466
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9127877237851663e-05,
+      "loss": 0.014,
+      "step": 1467
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9120204603580565e-05,
+      "loss": 0.0071,
+      "step": 1468
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9112531969309463e-05,
+      "loss": 0.0108,
+      "step": 1469
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.910485933503836e-05,
+      "loss": 0.0106,
+      "step": 1470
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9097186700767266e-05,
+      "loss": 0.0404,
+      "step": 1471
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9089514066496165e-05,
+      "loss": 2.2011,
+      "step": 1472
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9081841432225063e-05,
+      "loss": 0.0087,
+      "step": 1473
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9074168797953965e-05,
+      "loss": 0.0071,
+      "step": 1474
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9066496163682866e-05,
+      "loss": 0.0072,
+      "step": 1475
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9058823529411764e-05,
+      "loss": 0.0108,
+      "step": 1476
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9051150895140666e-05,
+      "loss": 0.0108,
+      "step": 1477
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9043478260869564e-05,
+      "loss": 0.0072,
+      "step": 1478
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9035805626598466e-05,
+      "loss": 0.0108,
+      "step": 1479
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9028132992327367e-05,
+      "loss": 0.0073,
+      "step": 1480
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9020460358056266e-05,
+      "loss": 0.9537,
+      "step": 1481
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9012787723785164e-05,
+      "loss": 0.0107,
+      "step": 1482
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.900511508951407e-05,
+      "loss": 0.0145,
+      "step": 1483
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8997442455242967e-05,
+      "loss": 0.007,
+      "step": 1484
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8989769820971865e-05,
+      "loss": 0.0071,
+      "step": 1485
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.898209718670077e-05,
+      "loss": 1.243,
+      "step": 1486
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.897442455242967e-05,
+      "loss": 0.011,
+      "step": 1487
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.8966751918158567e-05,
+      "loss": 0.0077,
+      "step": 1488
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895907928388747e-05,
+      "loss": 0.0073,
+      "step": 1489
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895140664961637e-05,
+      "loss": 1.2486,
+      "step": 1490
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8943734015345268e-05,
+      "loss": 0.0109,
+      "step": 1491
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.893606138107417e-05,
+      "loss": 0.0149,
+      "step": 1492
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8928388746803068e-05,
+      "loss": 0.0125,
+      "step": 1493
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.892071611253197e-05,
+      "loss": 0.0071,
+      "step": 1494
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.891304347826087e-05,
+      "loss": 0.0036,
+      "step": 1495
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.890537084398977e-05,
+      "loss": 0.0149,
+      "step": 1496
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8897698209718668e-05,
+      "loss": 0.015,
+      "step": 1497
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8890025575447573e-05,
+      "loss": 0.9536,
+      "step": 1498
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.888235294117647e-05,
+      "loss": 0.9997,
+      "step": 1499
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.887468030690537e-05,
+      "loss": 0.9621,
+      "step": 1500
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.886700767263427e-05,
+      "loss": 0.0113,
+      "step": 1501
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.8859335038363172e-05,
+      "loss": 0.2056,
+      "step": 1502
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.885166240409207e-05,
+      "loss": 0.0114,
+      "step": 1503
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8843989769820972e-05,
+      "loss": 0.016,
+      "step": 1504
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.883631713554987e-05,
+      "loss": 0.9175,
+      "step": 1505
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8828644501278775e-05,
+      "loss": 0.8959,
+      "step": 1506
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8820971867007674e-05,
+      "loss": 0.0074,
+      "step": 1507
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8813299232736572e-05,
+      "loss": 0.0117,
+      "step": 1508
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8805626598465474e-05,
+      "loss": 0.0075,
+      "step": 1509
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8797953964194375e-05,
+      "loss": 0.0074,
+      "step": 1510
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8790281329923273e-05,
+      "loss": 0.0577,
+      "step": 1511
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8782608695652175e-05,
+      "loss": 0.0128,
+      "step": 1512
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8774936061381077e-05,
+      "loss": 0.0073,
+      "step": 1513
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8767263427109975e-05,
+      "loss": 0.0074,
+      "step": 1514
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8759590792838876e-05,
+      "loss": 0.0073,
+      "step": 1515
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8751918158567775e-05,
+      "loss": 0.0088,
+      "step": 1516
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8744245524296676e-05,
+      "loss": 0.0078,
+      "step": 1517
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8736572890025578e-05,
+      "loss": 0.016,
+      "step": 1518
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8728900255754476e-05,
+      "loss": 1.3842,
+      "step": 1519
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8721227621483374e-05,
+      "loss": 1.2502,
+      "step": 1520
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.871355498721228e-05,
+      "loss": 0.0117,
+      "step": 1521
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8705882352941178e-05,
+      "loss": 0.0113,
+      "step": 1522
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8698209718670076e-05,
+      "loss": 0.0125,
+      "step": 1523
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8690537084398977e-05,
+      "loss": 0.0153,
+      "step": 1524
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.868286445012788e-05,
+      "loss": 0.0072,
+      "step": 1525
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8675191815856777e-05,
+      "loss": 0.0086,
+      "step": 1526
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.866751918158568e-05,
+      "loss": 0.0195,
+      "step": 1527
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.8659846547314577e-05,
+      "loss": 1.2511,
+      "step": 1528
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.865217391304348e-05,
+      "loss": 1.2407,
+      "step": 1529
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.864450127877238e-05,
+      "loss": 0.0112,
+      "step": 1530
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.863682864450128e-05,
+      "loss": 0.0073,
+      "step": 1531
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8629156010230177e-05,
+      "loss": 1.1911,
+      "step": 1532
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8621483375959082e-05,
+      "loss": 0.0168,
+      "step": 1533
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.861381074168798e-05,
+      "loss": 0.0128,
+      "step": 1534
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8606138107416878e-05,
+      "loss": 0.0116,
+      "step": 1535
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8598465473145783e-05,
+      "loss": 1.2286,
+      "step": 1536
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.859079283887468e-05,
+      "loss": 0.0158,
+      "step": 1537
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.858312020460358e-05,
+      "loss": 2.3735,
+      "step": 1538
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.857544757033248e-05,
+      "loss": 1.1972,
+      "step": 1539
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8567774936061383e-05,
+      "loss": 0.011,
+      "step": 1540
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.856010230179028e-05,
+      "loss": 0.0127,
+      "step": 1541
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8552429667519183e-05,
+      "loss": 0.0148,
+      "step": 1542
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.854475703324808e-05,
+      "loss": 0.0111,
+      "step": 1543
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8537084398976983e-05,
+      "loss": 0.0143,
+      "step": 1544
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8529411764705884e-05,
+      "loss": 0.0132,
+      "step": 1545
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8521739130434782e-05,
+      "loss": 0.0152,
+      "step": 1546
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.851406649616368e-05,
+      "loss": 0.0152,
+      "step": 1547
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8506393861892586e-05,
+      "loss": 0.0051,
+      "step": 1548
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8498721227621484e-05,
+      "loss": 0.9576,
+      "step": 1549
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8491048593350382e-05,
+      "loss": 0.0181,
+      "step": 1550
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8483375959079284e-05,
+      "loss": 0.011,
+      "step": 1551
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8475703324808185e-05,
+      "loss": 0.0205,
+      "step": 1552
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8468030690537083e-05,
+      "loss": 2.5631,
+      "step": 1553
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8460358056265985e-05,
+      "loss": 0.0102,
+      "step": 1554
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8452685421994883e-05,
+      "loss": 0.0147,
+      "step": 1555
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8445012787723788e-05,
+      "loss": 0.015,
+      "step": 1556
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8437340153452687e-05,
+      "loss": 0.0122,
+      "step": 1557
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8429667519181585e-05,
+      "loss": 0.0176,
+      "step": 1558
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8421994884910486e-05,
+      "loss": 0.0114,
+      "step": 1559
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8414322250639388e-05,
+      "loss": 0.0252,
+      "step": 1560
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8406649616368286e-05,
+      "loss": 0.013,
+      "step": 1561
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8398976982097188e-05,
+      "loss": 0.0142,
+      "step": 1562
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.839130434782609e-05,
+      "loss": 0.6496,
+      "step": 1563
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8383631713554988e-05,
+      "loss": 0.0149,
+      "step": 1564
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.837595907928389e-05,
+      "loss": 0.0104,
+      "step": 1565
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8368286445012787e-05,
+      "loss": 0.0129,
+      "step": 1566
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.836061381074169e-05,
+      "loss": 0.0197,
+      "step": 1567
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.835294117647059e-05,
+      "loss": 0.009,
+      "step": 1568
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.834526854219949e-05,
+      "loss": 0.0086,
+      "step": 1569
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.8337595907928387e-05,
+      "loss": 0.0087,
+      "step": 1570
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.8329923273657292e-05,
+      "loss": 0.0131,
+      "step": 1571
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.832225063938619e-05,
+      "loss": 1.2046,
+      "step": 1572
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.831457800511509e-05,
+      "loss": 0.0135,
+      "step": 1573
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.830690537084399e-05,
+      "loss": 0.0149,
+      "step": 1574
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.8299232736572892e-05,
+      "loss": 0.0077,
+      "step": 1575
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.829156010230179e-05,
+      "loss": 0.0143,
+      "step": 1576
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.828388746803069e-05,
+      "loss": 0.9806,
+      "step": 1577
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.827621483375959e-05,
+      "loss": 1.3783,
+      "step": 1578
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.826854219948849e-05,
+      "loss": 0.0075,
+      "step": 1579
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.8260869565217393e-05,
+      "loss": 0.0111,
+      "step": 1580
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.825319693094629e-05,
+      "loss": 0.0111,
+      "step": 1581
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.824552429667519e-05,
+      "loss": 0.0144,
+      "step": 1582
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8237851662404095e-05,
+      "loss": 0.0128,
+      "step": 1583
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8230179028132993e-05,
+      "loss": 1.2949,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.40147146582603455,
+      "eval_runtime": 58.0487,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.822250639386189e-05,
+      "loss": 0.0075,
+      "step": 1585
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8214833759590796e-05,
+      "loss": 0.0108,
+      "step": 1586
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8207161125319694e-05,
+      "loss": 0.011,
+      "step": 1587
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8199488491048592e-05,
+      "loss": 0.0148,
+      "step": 1588
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8191815856777494e-05,
+      "loss": 0.0077,
+      "step": 1589
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8184143222506396e-05,
+      "loss": 0.6674,
+      "step": 1590
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8176470588235294e-05,
+      "loss": 0.0109,
+      "step": 1591
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8168797953964195e-05,
+      "loss": 0.0108,
+      "step": 1592
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8161125319693094e-05,
+      "loss": 0.011,
+      "step": 1593
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8153452685421995e-05,
+      "loss": 0.0143,
+      "step": 1594
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8145780051150897e-05,
+      "loss": 0.9728,
+      "step": 1595
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8138107416879795e-05,
+      "loss": 0.0074,
+      "step": 1596
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8130434782608693e-05,
+      "loss": 0.0087,
+      "step": 1597
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.81227621483376e-05,
+      "loss": 0.0107,
+      "step": 1598
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8115089514066497e-05,
+      "loss": 0.0077,
+      "step": 1599
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8107416879795395e-05,
+      "loss": 0.0167,
+      "step": 1600
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8099744245524296e-05,
+      "loss": 1.219,
+      "step": 1601
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8092071611253198e-05,
+      "loss": 1.2244,
+      "step": 1602
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8084398976982096e-05,
+      "loss": 0.0108,
+      "step": 1603
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8076726342710998e-05,
+      "loss": 0.0071,
+      "step": 1604
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8069053708439896e-05,
+      "loss": 0.0109,
+      "step": 1605
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.8061381074168798e-05,
+      "loss": 0.0185,
+      "step": 1606
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.80537084398977e-05,
+      "loss": 1.2711,
+      "step": 1607
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.8046035805626598e-05,
+      "loss": 0.0113,
+      "step": 1608
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.80383631713555e-05,
+      "loss": 0.9575,
+      "step": 1609
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80306905370844e-05,
+      "loss": 1.1369,
+      "step": 1610
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80230179028133e-05,
+      "loss": 0.0114,
+      "step": 1611
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80153452685422e-05,
+      "loss": 1.578,
+      "step": 1612
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.8007672634271102e-05,
+      "loss": 0.0076,
+      "step": 1613
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.0128,
+      "step": 1614
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.7992327365728902e-05,
+      "loss": 0.0157,
+      "step": 1615
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.79846547314578e-05,
+      "loss": 0.0154,
+      "step": 1616
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.7976982097186702e-05,
+      "loss": 0.0125,
+      "step": 1617
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7969309462915604e-05,
+      "loss": 0.0085,
+      "step": 1618
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7961636828644502e-05,
+      "loss": 0.0167,
+      "step": 1619
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.79539641943734e-05,
+      "loss": 1.2275,
+      "step": 1620
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7946291560102305e-05,
+      "loss": 0.0115,
+      "step": 1621
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7938618925831203e-05,
+      "loss": 0.0083,
+      "step": 1622
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.79309462915601e-05,
+      "loss": 0.015,
+      "step": 1623
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7923273657289003e-05,
+      "loss": 1.1705,
+      "step": 1624
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7915601023017905e-05,
+      "loss": 0.0152,
+      "step": 1625
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7907928388746803e-05,
+      "loss": 2.0998,
+      "step": 1626
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7900255754475704e-05,
+      "loss": 0.0072,
+      "step": 1627
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7892583120204603e-05,
+      "loss": 0.0089,
+      "step": 1628
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7884910485933504e-05,
+      "loss": 0.0077,
+      "step": 1629
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7877237851662406e-05,
+      "loss": 0.0126,
+      "step": 1630
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7869565217391304e-05,
+      "loss": 1.2113,
+      "step": 1631
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7861892583120202e-05,
+      "loss": 0.0112,
+      "step": 1632
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7854219948849107e-05,
+      "loss": 0.0099,
+      "step": 1633
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7846547314578006e-05,
+      "loss": 0.0074,
+      "step": 1634
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7838874680306904e-05,
+      "loss": 0.0111,
+      "step": 1635
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.783120204603581e-05,
+      "loss": 0.0075,
+      "step": 1636
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7823529411764707e-05,
+      "loss": 1.2187,
+      "step": 1637
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7815856777493605e-05,
+      "loss": 0.0111,
+      "step": 1638
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7808184143222507e-05,
+      "loss": 0.0184,
+      "step": 1639
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.780051150895141e-05,
+      "loss": 0.0111,
+      "step": 1640
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7792838874680307e-05,
+      "loss": 0.0112,
+      "step": 1641
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.778516624040921e-05,
+      "loss": 0.0149,
+      "step": 1642
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.7777493606138107e-05,
+      "loss": 0.0151,
+      "step": 1643
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.7769820971867008e-05,
+      "loss": 0.0074,
+      "step": 1644
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.776214833759591e-05,
+      "loss": 0.9703,
+      "step": 1645
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.7754475703324808e-05,
+      "loss": 1.9364,
+      "step": 1646
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.7746803069053706e-05,
+      "loss": 0.011,
+      "step": 1647
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.773913043478261e-05,
+      "loss": 0.01,
+      "step": 1648
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.773145780051151e-05,
+      "loss": 0.0111,
+      "step": 1649
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.7723785166240408e-05,
+      "loss": 1.2373,
+      "step": 1650
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.771611253196931e-05,
+      "loss": 0.0074,
+      "step": 1651
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.770843989769821e-05,
+      "loss": 0.0116,
+      "step": 1652
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.770076726342711e-05,
+      "loss": 0.0117,
+      "step": 1653
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.769309462915601e-05,
+      "loss": 1.1636,
+      "step": 1654
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.768542199488491e-05,
+      "loss": 0.0075,
+      "step": 1655
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.767774936061381e-05,
+      "loss": 0.007,
+      "step": 1656
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.7670076726342712e-05,
+      "loss": 0.0174,
+      "step": 1657
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.766240409207161e-05,
+      "loss": 0.0113,
+      "step": 1658
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7654731457800512e-05,
+      "loss": 0.0084,
+      "step": 1659
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7647058823529414e-05,
+      "loss": 0.2994,
+      "step": 1660
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7639386189258312e-05,
+      "loss": 0.0213,
+      "step": 1661
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7631713554987213e-05,
+      "loss": 0.011,
+      "step": 1662
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7624040920716115e-05,
+      "loss": 1.2341,
+      "step": 1663
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7616368286445013e-05,
+      "loss": 1.1774,
+      "step": 1664
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7608695652173915e-05,
+      "loss": 0.0181,
+      "step": 1665
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7601023017902813e-05,
+      "loss": 0.0122,
+      "step": 1666
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7593350383631715e-05,
+      "loss": 0.0119,
+      "step": 1667
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7585677749360616e-05,
+      "loss": 0.0114,
+      "step": 1668
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7578005115089515e-05,
+      "loss": 0.0111,
+      "step": 1669
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7570332480818413e-05,
+      "loss": 0.011,
+      "step": 1670
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7562659846547318e-05,
+      "loss": 0.0113,
+      "step": 1671
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7554987212276216e-05,
+      "loss": 0.0161,
+      "step": 1672
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7547314578005114e-05,
+      "loss": 0.9491,
+      "step": 1673
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7539641943734016e-05,
+      "loss": 0.0108,
+      "step": 1674
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7531969309462917e-05,
+      "loss": 0.0114,
+      "step": 1675
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7524296675191816e-05,
+      "loss": 0.0107,
+      "step": 1676
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7516624040920717e-05,
+      "loss": 0.0171,
+      "step": 1677
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7508951406649616e-05,
+      "loss": 0.017,
+      "step": 1678
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7501278772378517e-05,
+      "loss": 1.939,
+      "step": 1679
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.749360613810742e-05,
+      "loss": 0.0151,
+      "step": 1680
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7485933503836317e-05,
+      "loss": 0.0109,
+      "step": 1681
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.7478260869565215e-05,
+      "loss": 0.0119,
+      "step": 1682
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.747058823529412e-05,
+      "loss": 1.2127,
+      "step": 1683
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.746291560102302e-05,
+      "loss": 0.0075,
+      "step": 1684
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.7455242966751917e-05,
+      "loss": 1.2283,
+      "step": 1685
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.744757033248082e-05,
+      "loss": 0.0107,
+      "step": 1686
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.743989769820972e-05,
+      "loss": 0.0141,
+      "step": 1687
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.7432225063938618e-05,
+      "loss": 1.2117,
+      "step": 1688
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.742455242966752e-05,
+      "loss": 0.8715,
+      "step": 1689
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.741687979539642e-05,
+      "loss": 0.0075,
+      "step": 1690
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.740920716112532e-05,
+      "loss": 1.2208,
+      "step": 1691
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.740153452685422e-05,
+      "loss": 0.0077,
+      "step": 1692
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.739386189258312e-05,
+      "loss": 0.0081,
+      "step": 1693
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.738618925831202e-05,
+      "loss": 0.0104,
+      "step": 1694
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.7378516624040923e-05,
+      "loss": 0.0162,
+      "step": 1695
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.737084398976982e-05,
+      "loss": 0.0173,
+      "step": 1696
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.736317135549872e-05,
+      "loss": 0.0194,
+      "step": 1697
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.7355498721227624e-05,
+      "loss": 0.0174,
+      "step": 1698
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.7347826086956522e-05,
+      "loss": 0.0081,
+      "step": 1699
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.734015345268542e-05,
+      "loss": 0.0121,
+      "step": 1700
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7332480818414322e-05,
+      "loss": 0.0137,
+      "step": 1701
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7324808184143224e-05,
+      "loss": 0.0117,
+      "step": 1702
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7317135549872122e-05,
+      "loss": 0.0133,
+      "step": 1703
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7309462915601024e-05,
+      "loss": 0.0175,
+      "step": 1704
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7301790281329922e-05,
+      "loss": 0.0076,
+      "step": 1705
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7294117647058823e-05,
+      "loss": 0.0179,
+      "step": 1706
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7286445012787725e-05,
+      "loss": 1.0541,
+      "step": 1707
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7278772378516623e-05,
+      "loss": 0.0077,
+      "step": 1708
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7271099744245525e-05,
+      "loss": 0.0079,
+      "step": 1709
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7263427109974426e-05,
+      "loss": 0.0102,
+      "step": 1710
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7255754475703325e-05,
+      "loss": 0.0158,
+      "step": 1711
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7248081841432223e-05,
+      "loss": 0.0107,
+      "step": 1712
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7240409207161128e-05,
+      "loss": 0.0159,
+      "step": 1713
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7232736572890026e-05,
+      "loss": 0.0125,
+      "step": 1714
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7225063938618928e-05,
+      "loss": 0.0081,
+      "step": 1715
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7217391304347826e-05,
+      "loss": 1.1946,
+      "step": 1716
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7209718670076728e-05,
+      "loss": 1.0298,
+      "step": 1717
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.720204603580563e-05,
+      "loss": 1.0207,
+      "step": 1718
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7194373401534527e-05,
+      "loss": 0.0152,
+      "step": 1719
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7186700767263426e-05,
+      "loss": 0.0187,
+      "step": 1720
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.717902813299233e-05,
+      "loss": 0.0134,
+      "step": 1721
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.717135549872123e-05,
+      "loss": 0.0173,
+      "step": 1722
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.7163682864450127e-05,
+      "loss": 1.0206,
+      "step": 1723
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.715601023017903e-05,
+      "loss": 0.0234,
+      "step": 1724
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.714833759590793e-05,
+      "loss": 0.0151,
+      "step": 1725
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.714066496163683e-05,
+      "loss": 0.0074,
+      "step": 1726
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.713299232736573e-05,
+      "loss": 0.0127,
+      "step": 1727
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.712531969309463e-05,
+      "loss": 0.8679,
+      "step": 1728
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.711764705882353e-05,
+      "loss": 0.0498,
+      "step": 1729
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.710997442455243e-05,
+      "loss": 0.0136,
+      "step": 1730
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.710230179028133e-05,
+      "loss": 1.2168,
+      "step": 1731
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.7094629156010228e-05,
+      "loss": 0.0105,
+      "step": 1732
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.7086956521739133e-05,
+      "loss": 0.0107,
+      "step": 1733
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.707928388746803e-05,
+      "loss": 0.0122,
+      "step": 1734
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.707161125319693e-05,
+      "loss": 0.0075,
+      "step": 1735
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.7063938618925834e-05,
+      "loss": 0.0607,
+      "step": 1736
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7056265984654733e-05,
+      "loss": 1.0181,
+      "step": 1737
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.704859335038363e-05,
+      "loss": 0.0095,
+      "step": 1738
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7040920716112533e-05,
+      "loss": 0.0128,
+      "step": 1739
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7033248081841434e-05,
+      "loss": 0.0096,
+      "step": 1740
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7025575447570332e-05,
+      "loss": 1.1781,
+      "step": 1741
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7017902813299234e-05,
+      "loss": 1.0132,
+      "step": 1742
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7010230179028132e-05,
+      "loss": 0.0097,
+      "step": 1743
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7002557544757034e-05,
+      "loss": 0.012,
+      "step": 1744
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6994884910485935e-05,
+      "loss": 0.0128,
+      "step": 1745
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6987212276214834e-05,
+      "loss": 0.0145,
+      "step": 1746
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6979539641943732e-05,
+      "loss": 0.7558,
+      "step": 1747
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6971867007672637e-05,
+      "loss": 0.0107,
+      "step": 1748
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6964194373401535e-05,
+      "loss": 1.0181,
+      "step": 1749
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6956521739130433e-05,
+      "loss": 1.7925,
+      "step": 1750
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6948849104859335e-05,
+      "loss": 0.0141,
+      "step": 1751
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6941176470588237e-05,
+      "loss": 0.0167,
+      "step": 1752
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6933503836317135e-05,
+      "loss": 0.0141,
+      "step": 1753
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6925831202046036e-05,
+      "loss": 0.0149,
+      "step": 1754
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6918158567774935e-05,
+      "loss": 0.0115,
+      "step": 1755
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6910485933503836e-05,
+      "loss": 0.0071,
+      "step": 1756
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6902813299232738e-05,
+      "loss": 0.0155,
+      "step": 1757
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6895140664961636e-05,
+      "loss": 0.0199,
+      "step": 1758
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6887468030690538e-05,
+      "loss": 0.0124,
+      "step": 1759
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.687979539641944e-05,
+      "loss": 0.0086,
+      "step": 1760
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.6872122762148337e-05,
+      "loss": 0.0147,
+      "step": 1761
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.6864450127877236e-05,
+      "loss": 0.0107,
+      "step": 1762
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.685677749360614e-05,
+      "loss": 1.025,
+      "step": 1763
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.684910485933504e-05,
+      "loss": 0.0105,
+      "step": 1764
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.6841432225063937e-05,
+      "loss": 0.0081,
+      "step": 1765
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.683375959079284e-05,
+      "loss": 0.009,
+      "step": 1766
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.682608695652174e-05,
+      "loss": 0.0109,
+      "step": 1767
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.6818414322250642e-05,
+      "loss": 0.008,
+      "step": 1768
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.681074168797954e-05,
+      "loss": 0.0132,
+      "step": 1769
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.680306905370844e-05,
+      "loss": 0.0119,
+      "step": 1770
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.6795396419437343e-05,
+      "loss": 0.0093,
+      "step": 1771
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.678772378516624e-05,
+      "loss": 1.0247,
+      "step": 1772
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.678005115089514e-05,
+      "loss": 1.024,
+      "step": 1773
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.677237851662404e-05,
+      "loss": 0.0106,
+      "step": 1774
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.6764705882352943e-05,
+      "loss": 0.0097,
+      "step": 1775
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.675703324808184e-05,
+      "loss": 0.0098,
+      "step": 1776
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6749360613810743e-05,
+      "loss": 1.0136,
+      "step": 1777
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.674168797953964e-05,
+      "loss": 0.0066,
+      "step": 1778
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6734015345268543e-05,
+      "loss": 0.1284,
+      "step": 1779
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6726342710997444e-05,
+      "loss": 0.9832,
+      "step": 1780
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6718670076726343e-05,
+      "loss": 0.0128,
+      "step": 1781
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6710997442455244e-05,
+      "loss": 0.0103,
+      "step": 1782
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6703324808184146e-05,
+      "loss": 0.0067,
+      "step": 1783
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6695652173913044e-05,
+      "loss": 0.0105,
+      "step": 1784
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6687979539641942e-05,
+      "loss": 0.0146,
+      "step": 1785
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6680306905370847e-05,
+      "loss": 0.0068,
+      "step": 1786
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6672634271099745e-05,
+      "loss": 0.0148,
+      "step": 1787
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6664961636828644e-05,
+      "loss": 0.0107,
+      "step": 1788
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6657289002557545e-05,
+      "loss": 2.4738,
+      "step": 1789
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6649616368286447e-05,
+      "loss": 0.9625,
+      "step": 1790
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6641943734015345e-05,
+      "loss": 0.2578,
+      "step": 1791
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6634271099744247e-05,
+      "loss": 0.0069,
+      "step": 1792
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6626598465473145e-05,
+      "loss": 0.011,
+      "step": 1793
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6618925831202047e-05,
+      "loss": 0.0066,
+      "step": 1794
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6611253196930948e-05,
+      "loss": 0.0069,
+      "step": 1795
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6603580562659846e-05,
+      "loss": 0.0071,
+      "step": 1796
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6595907928388745e-05,
+      "loss": 0.0068,
+      "step": 1797
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.658823529411765e-05,
+      "loss": 0.0232,
+      "step": 1798
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6580562659846548e-05,
+      "loss": 1.1318,
+      "step": 1799
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6572890025575446e-05,
+      "loss": 0.0101,
+      "step": 1800
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6565217391304348e-05,
+      "loss": 0.0187,
+      "step": 1801
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.655754475703325e-05,
+      "loss": 1.0718,
+      "step": 1802
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6549872122762148e-05,
+      "loss": 1.2349,
+      "step": 1803
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.654219948849105e-05,
+      "loss": 0.0148,
+      "step": 1804
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.6534526854219947e-05,
+      "loss": 0.0148,
+      "step": 1805
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.652685421994885e-05,
+      "loss": 0.0067,
+      "step": 1806
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.651918158567775e-05,
+      "loss": 0.0075,
+      "step": 1807
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.651150895140665e-05,
+      "loss": 0.0157,
+      "step": 1808
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.650383631713555e-05,
+      "loss": 0.0084,
+      "step": 1809
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.6496163682864452e-05,
+      "loss": 0.0123,
+      "step": 1810
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.648849104859335e-05,
+      "loss": 0.0065,
+      "step": 1811
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.648081841432225e-05,
+      "loss": 0.014,
+      "step": 1812
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.6473145780051154e-05,
+      "loss": 0.0152,
+      "step": 1813
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.6465473145780052e-05,
+      "loss": 0.0168,
+      "step": 1814
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.645780051150895e-05,
+      "loss": 0.0063,
+      "step": 1815
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.645012787723785e-05,
+      "loss": 0.6994,
+      "step": 1816
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.6442455242966753e-05,
+      "loss": 0.0132,
+      "step": 1817
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.643478260869565e-05,
+      "loss": 0.0103,
+      "step": 1818
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.6427109974424553e-05,
+      "loss": 0.013,
+      "step": 1819
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.641943734015345e-05,
+      "loss": 1.1813,
+      "step": 1820
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6411764705882356e-05,
+      "loss": 0.9809,
+      "step": 1821
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6404092071611254e-05,
+      "loss": 1.2339,
+      "step": 1822
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6396419437340153e-05,
+      "loss": 0.9744,
+      "step": 1823
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6388746803069054e-05,
+      "loss": 0.0098,
+      "step": 1824
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6381074168797956e-05,
+      "loss": 0.0062,
+      "step": 1825
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6373401534526854e-05,
+      "loss": 0.0151,
+      "step": 1826
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6365728900255756e-05,
+      "loss": 0.013,
+      "step": 1827
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6358056265984654e-05,
+      "loss": 0.0071,
+      "step": 1828
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6350383631713556e-05,
+      "loss": 0.0117,
+      "step": 1829
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6342710997442457e-05,
+      "loss": 0.0165,
+      "step": 1830
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6335038363171355e-05,
+      "loss": 0.1519,
+      "step": 1831
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6327365728900257e-05,
+      "loss": 1.0247,
+      "step": 1832
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.631969309462916e-05,
+      "loss": 0.0092,
+      "step": 1833
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6312020460358057e-05,
+      "loss": 0.0098,
+      "step": 1834
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6304347826086955e-05,
+      "loss": 1.1829,
+      "step": 1835
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.629667519181586e-05,
+      "loss": 0.006,
+      "step": 1836
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.628900255754476e-05,
+      "loss": 0.0109,
+      "step": 1837
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.6281329923273657e-05,
+      "loss": 0.6059,
+      "step": 1838
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.6273657289002558e-05,
+      "loss": 0.01,
+      "step": 1839
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.626598465473146e-05,
+      "loss": 0.0149,
+      "step": 1840
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.6258312020460358e-05,
+      "loss": 0.0152,
+      "step": 1841
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.625063938618926e-05,
+      "loss": 0.0171,
+      "step": 1842
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.6242966751918158e-05,
+      "loss": 0.0105,
+      "step": 1843
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.623529411764706e-05,
+      "loss": 0.0093,
+      "step": 1844
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.622762148337596e-05,
+      "loss": 1.4125,
+      "step": 1845
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.621994884910486e-05,
+      "loss": 0.0107,
+      "step": 1846
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.6212276214833757e-05,
+      "loss": 0.0133,
+      "step": 1847
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.6204603580562662e-05,
+      "loss": 0.0063,
+      "step": 1848
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.619693094629156e-05,
+      "loss": 0.0063,
+      "step": 1849
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.618925831202046e-05,
+      "loss": 0.009,
+      "step": 1850
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.618158567774936e-05,
+      "loss": 0.0137,
+      "step": 1851
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.6173913043478262e-05,
+      "loss": 0.0145,
+      "step": 1852
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.616624040920716e-05,
+      "loss": 0.0355,
+      "step": 1853
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.6158567774936062e-05,
+      "loss": 0.9601,
+      "step": 1854
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.615089514066496e-05,
+      "loss": 0.0135,
+      "step": 1855
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6143222506393862e-05,
+      "loss": 1.0129,
+      "step": 1856
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6135549872122763e-05,
+      "loss": 0.0339,
+      "step": 1857
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.612787723785166e-05,
+      "loss": 0.0131,
+      "step": 1858
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6120204603580563e-05,
+      "loss": 0.009,
+      "step": 1859
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6112531969309465e-05,
+      "loss": 0.0087,
+      "step": 1860
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6104859335038363e-05,
+      "loss": 0.012,
+      "step": 1861
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.609718670076726e-05,
+      "loss": 0.0088,
+      "step": 1862
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6089514066496166e-05,
+      "loss": 0.0117,
+      "step": 1863
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6081841432225065e-05,
+      "loss": 2.5214,
+      "step": 1864
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6074168797953963e-05,
+      "loss": 0.0107,
+      "step": 1865
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6066496163682864e-05,
+      "loss": 0.0101,
+      "step": 1866
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6058823529411766e-05,
+      "loss": 0.0118,
+      "step": 1867
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6051150895140664e-05,
+      "loss": 0.0087,
+      "step": 1868
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6043478260869566e-05,
+      "loss": 0.0122,
+      "step": 1869
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6035805626598464e-05,
+      "loss": 0.0084,
+      "step": 1870
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6028132992327366e-05,
+      "loss": 0.0104,
+      "step": 1871
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6020460358056267e-05,
+      "loss": 0.1587,
+      "step": 1872
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6012787723785166e-05,
+      "loss": 0.0057,
+      "step": 1873
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6005115089514067e-05,
+      "loss": 0.0085,
+      "step": 1874
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.599744245524297e-05,
+      "loss": 0.0056,
+      "step": 1875
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.5989769820971867e-05,
+      "loss": 1.2194,
+      "step": 1876
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.598209718670077e-05,
+      "loss": 0.0056,
+      "step": 1877
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.5974424552429667e-05,
+      "loss": 0.0057,
+      "step": 1878
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.596675191815857e-05,
+      "loss": 1.0276,
+      "step": 1879
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.595907928388747e-05,
+      "loss": 0.0083,
+      "step": 1880
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.5951406649616368e-05,
+      "loss": 0.0085,
+      "step": 1881
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.594373401534527e-05,
+      "loss": 0.0363,
+      "step": 1882
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.593606138107417e-05,
+      "loss": 0.0157,
+      "step": 1883
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.592838874680307e-05,
+      "loss": 0.0313,
+      "step": 1884
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.5920716112531968e-05,
+      "loss": 0.0062,
+      "step": 1885
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.5913043478260873e-05,
+      "loss": 0.0195,
+      "step": 1886
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.590537084398977e-05,
+      "loss": 0.0151,
+      "step": 1887
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.589769820971867e-05,
+      "loss": 0.0101,
+      "step": 1888
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.589002557544757e-05,
+      "loss": 1.1092,
+      "step": 1889
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.5882352941176473e-05,
+      "loss": 0.0323,
+      "step": 1890
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.587468030690537e-05,
+      "loss": 0.0086,
+      "step": 1891
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.5867007672634272e-05,
+      "loss": 0.0147,
+      "step": 1892
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.585933503836317e-05,
+      "loss": 0.0084,
+      "step": 1893
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.5851662404092072e-05,
+      "loss": 0.5596,
+      "step": 1894
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5843989769820974e-05,
+      "loss": 0.0114,
+      "step": 1895
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5836317135549872e-05,
+      "loss": 0.0083,
+      "step": 1896
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.582864450127877e-05,
+      "loss": 0.0082,
+      "step": 1897
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5820971867007675e-05,
+      "loss": 0.0146,
+      "step": 1898
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5813299232736574e-05,
+      "loss": 0.0081,
+      "step": 1899
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5805626598465472e-05,
+      "loss": 0.0081,
+      "step": 1900
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5797953964194373e-05,
+      "loss": 0.0116,
+      "step": 1901
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5790281329923275e-05,
+      "loss": 0.0084,
+      "step": 1902
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5782608695652173e-05,
+      "loss": 0.0108,
+      "step": 1903
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5774936061381075e-05,
+      "loss": 0.6146,
+      "step": 1904
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5767263427109973e-05,
+      "loss": 0.0162,
+      "step": 1905
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5759590792838875e-05,
+      "loss": 0.0134,
+      "step": 1906
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5751918158567776e-05,
+      "loss": 0.0047,
+      "step": 1907
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5744245524296674e-05,
+      "loss": 0.0045,
+      "step": 1908
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5736572890025576e-05,
+      "loss": 0.0087,
+      "step": 1909
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5728900255754478e-05,
+      "loss": 2.0913,
+      "step": 1910
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5721227621483376e-05,
+      "loss": 0.0075,
+      "step": 1911
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5713554987212274e-05,
+      "loss": 1.3387,
+      "step": 1912
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.570588235294118e-05,
+      "loss": 0.0102,
+      "step": 1913
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5698209718670077e-05,
+      "loss": 0.0076,
+      "step": 1914
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5690537084398976e-05,
+      "loss": 0.0131,
+      "step": 1915
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5682864450127877e-05,
+      "loss": 0.0079,
+      "step": 1916
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.567519181585678e-05,
+      "loss": 0.0102,
+      "step": 1917
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5667519181585677e-05,
+      "loss": 0.9554,
+      "step": 1918
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.565984654731458e-05,
+      "loss": 0.0088,
+      "step": 1919
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.5652173913043477e-05,
+      "loss": 1.0265,
+      "step": 1920
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.564450127877238e-05,
+      "loss": 0.0155,
+      "step": 1921
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.563682864450128e-05,
+      "loss": 0.0079,
+      "step": 1922
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.562915601023018e-05,
+      "loss": 1.0467,
+      "step": 1923
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.5621483375959077e-05,
+      "loss": 0.0132,
+      "step": 1924
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.561381074168798e-05,
+      "loss": 0.0077,
+      "step": 1925
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.560613810741688e-05,
+      "loss": 0.0064,
+      "step": 1926
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.559846547314578e-05,
+      "loss": 0.0104,
+      "step": 1927
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.559079283887468e-05,
+      "loss": 0.0096,
+      "step": 1928
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.558312020460358e-05,
+      "loss": 0.0049,
+      "step": 1929
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.5575447570332483e-05,
+      "loss": 0.0104,
+      "step": 1930
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.556777493606138e-05,
+      "loss": 1.0424,
+      "step": 1931
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5560102301790283e-05,
+      "loss": 0.0079,
+      "step": 1932
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5552429667519184e-05,
+      "loss": 0.0104,
+      "step": 1933
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5544757033248083e-05,
+      "loss": 0.0077,
+      "step": 1934
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.553708439897698e-05,
+      "loss": 0.005,
+      "step": 1935
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5529411764705886e-05,
+      "loss": 2.6544,
+      "step": 1936
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5521739130434784e-05,
+      "loss": 0.0107,
+      "step": 1937
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5514066496163682e-05,
+      "loss": 0.0124,
+      "step": 1938
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5506393861892584e-05,
+      "loss": 0.0049,
+      "step": 1939
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5498721227621485e-05,
+      "loss": 0.0078,
+      "step": 1940
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5491048593350384e-05,
+      "loss": 2.1009,
+      "step": 1941
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5483375959079285e-05,
+      "loss": 0.0116,
+      "step": 1942
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5475703324808183e-05,
+      "loss": 0.0049,
+      "step": 1943
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5468030690537085e-05,
+      "loss": 0.0079,
+      "step": 1944
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5460358056265987e-05,
+      "loss": 1.0307,
+      "step": 1945
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5452685421994885e-05,
+      "loss": 0.0147,
+      "step": 1946
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5445012787723783e-05,
+      "loss": 0.0084,
+      "step": 1947
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5437340153452688e-05,
+      "loss": 0.0262,
+      "step": 1948
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5429667519181586e-05,
+      "loss": 0.0115,
+      "step": 1949
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5421994884910485e-05,
+      "loss": 1.3221,
+      "step": 1950
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5414322250639386e-05,
+      "loss": 0.7929,
+      "step": 1951
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5406649616368288e-05,
+      "loss": 1.2069,
+      "step": 1952
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5398976982097186e-05,
+      "loss": 0.0081,
+      "step": 1953
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5391304347826088e-05,
+      "loss": 0.0069,
+      "step": 1954
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5383631713554986e-05,
+      "loss": 0.0097,
+      "step": 1955
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5375959079283887e-05,
+      "loss": 0.0082,
+      "step": 1956
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.536828644501279e-05,
+      "loss": 0.0061,
+      "step": 1957
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5360613810741687e-05,
+      "loss": 0.0082,
+      "step": 1958
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.535294117647059e-05,
+      "loss": 0.0113,
+      "step": 1959
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.534526854219949e-05,
+      "loss": 0.0098,
+      "step": 1960
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.533759590792839e-05,
+      "loss": 1.0221,
+      "step": 1961
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.5329923273657287e-05,
+      "loss": 0.0039,
+      "step": 1962
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.5322250639386192e-05,
+      "loss": 0.0127,
+      "step": 1963
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.531457800511509e-05,
+      "loss": 0.0051,
+      "step": 1964
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.530690537084399e-05,
+      "loss": 0.0086,
+      "step": 1965
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.529923273657289e-05,
+      "loss": 0.0096,
+      "step": 1966
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.529156010230179e-05,
+      "loss": 0.0164,
+      "step": 1967
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.528388746803069e-05,
+      "loss": 1.0127,
+      "step": 1968
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.527621483375959e-05,
+      "loss": 0.0149,
+      "step": 1969
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.526854219948849e-05,
+      "loss": 0.0085,
+      "step": 1970
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.526086956521739e-05,
+      "loss": 0.0116,
+      "step": 1971
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5253196930946293e-05,
+      "loss": 0.0168,
+      "step": 1972
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5245524296675191e-05,
+      "loss": 0.0052,
+      "step": 1973
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5237851662404091e-05,
+      "loss": 0.0052,
+      "step": 1974
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5230179028132993e-05,
+      "loss": 1.0013,
+      "step": 1975
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5222506393861893e-05,
+      "loss": 0.0122,
+      "step": 1976
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.521483375959079e-05,
+      "loss": 0.0087,
+      "step": 1977
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5207161125319694e-05,
+      "loss": 1.9624,
+      "step": 1978
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5199488491048592e-05,
+      "loss": 0.0088,
+      "step": 1979
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5191815856777496e-05,
+      "loss": 0.0596,
+      "step": 1980
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9090909090909091,
+      "eval_loss": 0.40464258193969727,
+      "eval_runtime": 58.0491,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1980
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5184143222506396e-05,
+      "loss": 0.009,
+      "step": 1981
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5176470588235294e-05,
+      "loss": 0.0063,
+      "step": 1982
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5168797953964197e-05,
+      "loss": 0.0092,
+      "step": 1983
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5161125319693095e-05,
+      "loss": 0.0092,
+      "step": 1984
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5153452685421995e-05,
+      "loss": 0.0259,
+      "step": 1985
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5145780051150897e-05,
+      "loss": 0.0093,
+      "step": 1986
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5138107416879797e-05,
+      "loss": 1.3347,
+      "step": 1987
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5130434782608695e-05,
+      "loss": 0.0135,
+      "step": 1988
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5122762148337598e-05,
+      "loss": 0.0134,
+      "step": 1989
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5115089514066497e-05,
+      "loss": 0.0051,
+      "step": 1990
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5107416879795396e-05,
+      "loss": 0.0106,
+      "step": 1991
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5099744245524298e-05,
+      "loss": 0.013,
+      "step": 1992
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5092071611253198e-05,
+      "loss": 0.0095,
+      "step": 1993
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5084398976982096e-05,
+      "loss": 0.0096,
+      "step": 1994
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5076726342711e-05,
+      "loss": 0.0112,
+      "step": 1995
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5069053708439898e-05,
+      "loss": 0.0137,
+      "step": 1996
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5061381074168798e-05,
+      "loss": 0.9611,
+      "step": 1997
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50537084398977e-05,
+      "loss": 0.009,
+      "step": 1998
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50460358056266e-05,
+      "loss": 0.0066,
+      "step": 1999
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.5038363171355497e-05,
+      "loss": 0.0091,
+      "step": 2000
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50306905370844e-05,
+      "loss": 0.9856,
+      "step": 2001
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5023017902813299e-05,
+      "loss": 0.0137,
+      "step": 2002
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5015345268542199e-05,
+      "loss": 0.0094,
+      "step": 2003
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.50076726342711e-05,
+      "loss": 0.9276,
+      "step": 2004
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.0051,
+      "step": 2005
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4992327365728902e-05,
+      "loss": 0.0097,
+      "step": 2006
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.49846547314578e-05,
+      "loss": 0.8053,
+      "step": 2007
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4976982097186702e-05,
+      "loss": 0.5635,
+      "step": 2008
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4969309462915602e-05,
+      "loss": 0.0099,
+      "step": 2009
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4961636828644502e-05,
+      "loss": 0.0108,
+      "step": 2010
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4953964194373402e-05,
+      "loss": 1.3337,
+      "step": 2011
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4946291560102303e-05,
+      "loss": 0.0098,
+      "step": 2012
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4938618925831201e-05,
+      "loss": 0.0099,
+      "step": 2013
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4930946291560103e-05,
+      "loss": 0.011,
+      "step": 2014
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4923273657289003e-05,
+      "loss": 0.0166,
+      "step": 2015
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4915601023017903e-05,
+      "loss": 1.3541,
+      "step": 2016
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4907928388746803e-05,
+      "loss": 0.0145,
+      "step": 2017
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4900255754475704e-05,
+      "loss": 0.0051,
+      "step": 2018
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4892583120204604e-05,
+      "loss": 0.0051,
+      "step": 2019
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4884910485933504e-05,
+      "loss": 0.0139,
+      "step": 2020
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4877237851662404e-05,
+      "loss": 0.0097,
+      "step": 2021
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4869565217391306e-05,
+      "loss": 0.9418,
+      "step": 2022
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4861892583120204e-05,
+      "loss": 0.9619,
+      "step": 2023
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4854219948849106e-05,
+      "loss": 1.3495,
+      "step": 2024
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4846547314578006e-05,
+      "loss": 0.0051,
+      "step": 2025
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4838874680306905e-05,
+      "loss": 1.3222,
+      "step": 2026
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4831202046035805e-05,
+      "loss": 1.1638,
+      "step": 2027
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4823529411764707e-05,
+      "loss": 0.009,
+      "step": 2028
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4815856777493605e-05,
+      "loss": 0.0059,
+      "step": 2029
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4808184143222507e-05,
+      "loss": 0.9946,
+      "step": 2030
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4800511508951408e-05,
+      "loss": 0.1291,
+      "step": 2031
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4792838874680307e-05,
+      "loss": 0.0089,
+      "step": 2032
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4785166240409208e-05,
+      "loss": 0.8113,
+      "step": 2033
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4777493606138108e-05,
+      "loss": 0.0109,
+      "step": 2034
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4769820971867008e-05,
+      "loss": 0.0061,
+      "step": 2035
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4762148337595908e-05,
+      "loss": 0.0157,
+      "step": 2036
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.475447570332481e-05,
+      "loss": 0.0104,
+      "step": 2037
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.4746803069053708e-05,
+      "loss": 0.012,
+      "step": 2038
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.473913043478261e-05,
+      "loss": 0.0106,
+      "step": 2039
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.473145780051151e-05,
+      "loss": 0.006,
+      "step": 2040
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.472378516624041e-05,
+      "loss": 0.0122,
+      "step": 2041
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.471611253196931e-05,
+      "loss": 0.0103,
+      "step": 2042
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.470843989769821e-05,
+      "loss": 0.9465,
+      "step": 2043
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.4700767263427109e-05,
+      "loss": 0.0149,
+      "step": 2044
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.469309462915601e-05,
+      "loss": 0.0146,
+      "step": 2045
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.468542199488491e-05,
+      "loss": 0.9225,
+      "step": 2046
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.4677749360613812e-05,
+      "loss": 0.0101,
+      "step": 2047
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.467007672634271e-05,
+      "loss": 0.0052,
+      "step": 2048
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.4662404092071612e-05,
+      "loss": 1.682,
+      "step": 2049
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4654731457800512e-05,
+      "loss": 0.9498,
+      "step": 2050
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4647058823529412e-05,
+      "loss": 0.0097,
+      "step": 2051
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4639386189258312e-05,
+      "loss": 0.0067,
+      "step": 2052
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4631713554987213e-05,
+      "loss": 0.0052,
+      "step": 2053
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4624040920716112e-05,
+      "loss": 0.0055,
+      "step": 2054
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4616368286445013e-05,
+      "loss": 0.0225,
+      "step": 2055
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4608695652173915e-05,
+      "loss": 0.0052,
+      "step": 2056
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4601023017902813e-05,
+      "loss": 0.0066,
+      "step": 2057
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4593350383631715e-05,
+      "loss": 0.0112,
+      "step": 2058
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4585677749360615e-05,
+      "loss": 1.3145,
+      "step": 2059
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4578005115089514e-05,
+      "loss": 0.0099,
+      "step": 2060
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4570332480818414e-05,
+      "loss": 0.0055,
+      "step": 2061
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4562659846547316e-05,
+      "loss": 0.0149,
+      "step": 2062
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4554987212276214e-05,
+      "loss": 0.0102,
+      "step": 2063
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4547314578005116e-05,
+      "loss": 0.0108,
+      "step": 2064
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4539641943734016e-05,
+      "loss": 0.0065,
+      "step": 2065
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4531969309462916e-05,
+      "loss": 0.0185,
+      "step": 2066
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4524296675191816e-05,
+      "loss": 0.9659,
+      "step": 2067
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4516624040920717e-05,
+      "loss": 0.0193,
+      "step": 2068
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4508951406649615e-05,
+      "loss": 0.0146,
+      "step": 2069
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4501278772378517e-05,
+      "loss": 0.0061,
+      "step": 2070
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4493606138107417e-05,
+      "loss": 0.0222,
+      "step": 2071
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4485933503836319e-05,
+      "loss": 0.0096,
+      "step": 2072
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4478260869565217e-05,
+      "loss": 0.0096,
+      "step": 2073
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4470588235294118e-05,
+      "loss": 0.005,
+      "step": 2074
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4462915601023018e-05,
+      "loss": 0.8301,
+      "step": 2075
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4455242966751918e-05,
+      "loss": 0.0049,
+      "step": 2076
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4447570332480818e-05,
+      "loss": 0.0097,
+      "step": 2077
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.443989769820972e-05,
+      "loss": 0.0062,
+      "step": 2078
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.4432225063938618e-05,
+      "loss": 1.3459,
+      "step": 2079
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.442455242966752e-05,
+      "loss": 0.0096,
+      "step": 2080
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4416879795396421e-05,
+      "loss": 1.2176,
+      "step": 2081
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.440920716112532e-05,
+      "loss": 0.0114,
+      "step": 2082
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4401534526854221e-05,
+      "loss": 0.0109,
+      "step": 2083
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4393861892583121e-05,
+      "loss": 0.0105,
+      "step": 2084
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.4386189258312021e-05,
+      "loss": 0.0107,
+      "step": 2085
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.437851662404092e-05,
+      "loss": 0.9442,
+      "step": 2086
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.4370843989769822e-05,
+      "loss": 0.01,
+      "step": 2087
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.436317135549872e-05,
+      "loss": 0.0099,
+      "step": 2088
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4355498721227622e-05,
+      "loss": 0.0167,
+      "step": 2089
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4347826086956522e-05,
+      "loss": 0.0051,
+      "step": 2090
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4340153452685422e-05,
+      "loss": 1.333,
+      "step": 2091
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4332480818414322e-05,
+      "loss": 0.0111,
+      "step": 2092
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4324808184143224e-05,
+      "loss": 0.0103,
+      "step": 2093
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4317135549872122e-05,
+      "loss": 0.0111,
+      "step": 2094
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4309462915601023e-05,
+      "loss": 0.0165,
+      "step": 2095
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4301790281329923e-05,
+      "loss": 0.0089,
+      "step": 2096
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4294117647058823e-05,
+      "loss": 0.0143,
+      "step": 2097
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4286445012787723e-05,
+      "loss": 0.4759,
+      "step": 2098
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4278772378516625e-05,
+      "loss": 0.0102,
+      "step": 2099
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4271099744245525e-05,
+      "loss": 0.0059,
+      "step": 2100
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4263427109974425e-05,
+      "loss": 0.0152,
+      "step": 2101
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4255754475703325e-05,
+      "loss": 0.0051,
+      "step": 2102
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4248081841432226e-05,
+      "loss": 0.0098,
+      "step": 2103
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4240409207161126e-05,
+      "loss": 0.0147,
+      "step": 2104
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4232736572890026e-05,
+      "loss": 0.0051,
+      "step": 2105
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4225063938618928e-05,
+      "loss": 1.3384,
+      "step": 2106
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4217391304347826e-05,
+      "loss": 0.0055,
+      "step": 2107
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4209718670076727e-05,
+      "loss": 0.0136,
+      "step": 2108
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4202046035805627e-05,
+      "loss": 0.0056,
+      "step": 2109
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4194373401534527e-05,
+      "loss": 0.0094,
+      "step": 2110
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4186700767263427e-05,
+      "loss": 0.0135,
+      "step": 2111
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4179028132992329e-05,
+      "loss": 0.0129,
+      "step": 2112
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4171355498721227e-05,
+      "loss": 0.0051,
+      "step": 2113
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4163682864450129e-05,
+      "loss": 0.0134,
+      "step": 2114
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4156010230179029e-05,
+      "loss": 0.0091,
+      "step": 2115
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4148337595907929e-05,
+      "loss": 0.0072,
+      "step": 2116
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.4140664961636828e-05,
+      "loss": 0.005,
+      "step": 2117
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.413299232736573e-05,
+      "loss": 0.0112,
+      "step": 2118
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.4125319693094628e-05,
+      "loss": 0.005,
+      "step": 2119
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.411764705882353e-05,
+      "loss": 0.0085,
+      "step": 2120
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.410997442455243e-05,
+      "loss": 0.0121,
+      "step": 2121
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.410230179028133e-05,
+      "loss": 0.0084,
+      "step": 2122
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.409462915601023e-05,
+      "loss": 0.0084,
+      "step": 2123
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.4086956521739131e-05,
+      "loss": 0.0083,
+      "step": 2124
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4079283887468031e-05,
+      "loss": 1.3402,
+      "step": 2125
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4071611253196931e-05,
+      "loss": 0.0124,
+      "step": 2126
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4063938618925831e-05,
+      "loss": 0.0082,
+      "step": 2127
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4056265984654733e-05,
+      "loss": 0.9694,
+      "step": 2128
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4048593350383633e-05,
+      "loss": 0.0081,
+      "step": 2129
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4040920716112532e-05,
+      "loss": 1.2102,
+      "step": 2130
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4033248081841434e-05,
+      "loss": 0.0113,
+      "step": 2131
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4025575447570332e-05,
+      "loss": 0.0114,
+      "step": 2132
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4017902813299234e-05,
+      "loss": 0.0252,
+      "step": 2133
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4010230179028134e-05,
+      "loss": 0.9785,
+      "step": 2134
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4002557544757034e-05,
+      "loss": 0.0061,
+      "step": 2135
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.3994884910485934e-05,
+      "loss": 0.0096,
+      "step": 2136
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3987212276214835e-05,
+      "loss": 0.0088,
+      "step": 2137
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3979539641943733e-05,
+      "loss": 0.0047,
+      "step": 2138
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3971867007672635e-05,
+      "loss": 0.0108,
+      "step": 2139
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3964194373401535e-05,
+      "loss": 0.0081,
+      "step": 2140
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3956521739130435e-05,
+      "loss": 0.0108,
+      "step": 2141
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3948849104859335e-05,
+      "loss": 0.0126,
+      "step": 2142
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3941176470588236e-05,
+      "loss": 0.0046,
+      "step": 2143
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3933503836317135e-05,
+      "loss": 0.0122,
+      "step": 2144
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3925831202046036e-05,
+      "loss": 0.0075,
+      "step": 2145
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3918158567774936e-05,
+      "loss": 0.0123,
+      "step": 2146
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3910485933503836e-05,
+      "loss": 0.0134,
+      "step": 2147
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3902813299232736e-05,
+      "loss": 0.0115,
+      "step": 2148
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3895140664961638e-05,
+      "loss": 0.0044,
+      "step": 2149
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3887468030690536e-05,
+      "loss": 0.0121,
+      "step": 2150
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3879795396419437e-05,
+      "loss": 0.0101,
+      "step": 2151
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3872122762148337e-05,
+      "loss": 0.0099,
+      "step": 2152
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3864450127877239e-05,
+      "loss": 0.0044,
+      "step": 2153
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3856777493606139e-05,
+      "loss": 0.0094,
+      "step": 2154
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3849104859335039e-05,
+      "loss": 0.0124,
+      "step": 2155
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.384143222506394e-05,
+      "loss": 0.0069,
+      "step": 2156
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.3833759590792839e-05,
+      "loss": 0.0068,
+      "step": 2157
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.382608695652174e-05,
+      "loss": 0.0094,
+      "step": 2158
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.381841432225064e-05,
+      "loss": 0.0067,
+      "step": 2159
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.381074168797954e-05,
+      "loss": 0.0044,
+      "step": 2160
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.380306905370844e-05,
+      "loss": 0.0041,
+      "step": 2161
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.3795396419437342e-05,
+      "loss": 0.0041,
+      "step": 2162
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.378772378516624e-05,
+      "loss": 0.009,
+      "step": 2163
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.3780051150895141e-05,
+      "loss": 0.0092,
+      "step": 2164
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3772378516624041e-05,
+      "loss": 0.0101,
+      "step": 2165
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3764705882352941e-05,
+      "loss": 0.0064,
+      "step": 2166
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3757033248081841e-05,
+      "loss": 0.0088,
+      "step": 2167
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3749360613810743e-05,
+      "loss": 0.0059,
+      "step": 2168
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3741687979539641e-05,
+      "loss": 0.0062,
+      "step": 2169
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3734015345268543e-05,
+      "loss": 0.0085,
+      "step": 2170
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3726342710997443e-05,
+      "loss": 0.0141,
+      "step": 2171
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3718670076726343e-05,
+      "loss": 0.0061,
+      "step": 2172
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3710997442455242e-05,
+      "loss": 0.0084,
+      "step": 2173
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3703324808184144e-05,
+      "loss": 0.0038,
+      "step": 2174
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3695652173913042e-05,
+      "loss": 0.006,
+      "step": 2175
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3687979539641944e-05,
+      "loss": 0.0039,
+      "step": 2176
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3680306905370844e-05,
+      "loss": 0.0205,
+      "step": 2177
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3672634271099745e-05,
+      "loss": 0.0062,
+      "step": 2178
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3664961636828645e-05,
+      "loss": 0.0021,
+      "step": 2179
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3657289002557545e-05,
+      "loss": 0.0058,
+      "step": 2180
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3649616368286447e-05,
+      "loss": 0.0101,
+      "step": 2181
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3641943734015345e-05,
+      "loss": 0.0036,
+      "step": 2182
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3634271099744247e-05,
+      "loss": 0.0037,
+      "step": 2183
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3626598465473147e-05,
+      "loss": 0.0036,
+      "step": 2184
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3618925831202047e-05,
+      "loss": 0.0036,
+      "step": 2185
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3611253196930946e-05,
+      "loss": 0.0035,
+      "step": 2186
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3603580562659848e-05,
+      "loss": 0.0059,
+      "step": 2187
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3595907928388746e-05,
+      "loss": 0.0056,
+      "step": 2188
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3588235294117648e-05,
+      "loss": 0.0139,
+      "step": 2189
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3580562659846548e-05,
+      "loss": 0.0558,
+      "step": 2190
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3572890025575448e-05,
+      "loss": 1.419,
+      "step": 2191
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3565217391304348e-05,
+      "loss": 0.0058,
+      "step": 2192
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.355754475703325e-05,
+      "loss": 0.0035,
+      "step": 2193
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3549872122762147e-05,
+      "loss": 0.0055,
+      "step": 2194
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3542199488491049e-05,
+      "loss": 1.2049,
+      "step": 2195
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3534526854219949e-05,
+      "loss": 1.3646,
+      "step": 2196
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3526854219948849e-05,
+      "loss": 1.3803,
+      "step": 2197
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3519181585677749e-05,
+      "loss": 0.0055,
+      "step": 2198
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.351150895140665e-05,
+      "loss": 0.0059,
+      "step": 2199
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.3503836317135549e-05,
+      "loss": 0.0056,
+      "step": 2200
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.349616368286445e-05,
+      "loss": 0.0097,
+      "step": 2201
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.348849104859335e-05,
+      "loss": 1.3497,
+      "step": 2202
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.348081841432225e-05,
+      "loss": 1.0989,
+      "step": 2203
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3473145780051152e-05,
+      "loss": 0.0117,
+      "step": 2204
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3465473145780052e-05,
+      "loss": 0.0038,
+      "step": 2205
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3457800511508953e-05,
+      "loss": 0.0078,
+      "step": 2206
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3450127877237852e-05,
+      "loss": 0.0058,
+      "step": 2207
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3442455242966753e-05,
+      "loss": 0.9775,
+      "step": 2208
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3434782608695653e-05,
+      "loss": 0.0037,
+      "step": 2209
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3427109974424553e-05,
+      "loss": 0.0079,
+      "step": 2210
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3419437340153453e-05,
+      "loss": 0.0663,
+      "step": 2211
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3411764705882354e-05,
+      "loss": 0.0059,
+      "step": 2212
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3404092071611253e-05,
+      "loss": 0.0039,
+      "step": 2213
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3396419437340154e-05,
+      "loss": 1.3169,
+      "step": 2214
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3388746803069054e-05,
+      "loss": 0.0061,
+      "step": 2215
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3381074168797954e-05,
+      "loss": 0.006,
+      "step": 2216
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3373401534526854e-05,
+      "loss": 0.004,
+      "step": 2217
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3365728900255756e-05,
+      "loss": 0.0045,
+      "step": 2218
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3358056265984654e-05,
+      "loss": 0.006,
+      "step": 2219
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3350383631713556e-05,
+      "loss": 0.0058,
+      "step": 2220
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3342710997442455e-05,
+      "loss": 0.014,
+      "step": 2221
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3335038363171355e-05,
+      "loss": 1.1674,
+      "step": 2222
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3327365728900255e-05,
+      "loss": 0.004,
+      "step": 2223
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3319693094629157e-05,
+      "loss": 0.0052,
+      "step": 2224
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3312020460358055e-05,
+      "loss": 0.0095,
+      "step": 2225
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3304347826086957e-05,
+      "loss": 0.0053,
+      "step": 2226
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3296675191815857e-05,
+      "loss": 0.0099,
+      "step": 2227
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3289002557544757e-05,
+      "loss": 1.3879,
+      "step": 2228
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3281329923273658e-05,
+      "loss": 0.0041,
+      "step": 2229
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3273657289002558e-05,
+      "loss": 0.006,
+      "step": 2230
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.326598465473146e-05,
+      "loss": 0.006,
+      "step": 2231
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.3258312020460358e-05,
+      "loss": 0.012,
+      "step": 2232
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.325063938618926e-05,
+      "loss": 0.0107,
+      "step": 2233
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.324296675191816e-05,
+      "loss": 0.0061,
+      "step": 2234
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.323529411764706e-05,
+      "loss": 0.0041,
+      "step": 2235
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.322762148337596e-05,
+      "loss": 0.006,
+      "step": 2236
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.3219948849104861e-05,
+      "loss": 0.0064,
+      "step": 2237
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.3212276214833759e-05,
+      "loss": 0.006,
+      "step": 2238
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.320460358056266e-05,
+      "loss": 0.0035,
+      "step": 2239
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.319693094629156e-05,
+      "loss": 0.8375,
+      "step": 2240
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.318925831202046e-05,
+      "loss": 0.004,
+      "step": 2241
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.318158567774936e-05,
+      "loss": 0.0077,
+      "step": 2242
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.3173913043478262e-05,
+      "loss": 0.006,
+      "step": 2243
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.316624040920716e-05,
+      "loss": 0.0041,
+      "step": 2244
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3158567774936062e-05,
+      "loss": 0.0034,
+      "step": 2245
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3150895140664962e-05,
+      "loss": 0.8219,
+      "step": 2246
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3143222506393862e-05,
+      "loss": 0.004,
+      "step": 2247
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3135549872122762e-05,
+      "loss": 0.004,
+      "step": 2248
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3127877237851663e-05,
+      "loss": 1.1328,
+      "step": 2249
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3120204603580562e-05,
+      "loss": 0.0039,
+      "step": 2250
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3112531969309463e-05,
+      "loss": 0.0283,
+      "step": 2251
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3104859335038363e-05,
+      "loss": 0.0058,
+      "step": 2252
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3097186700767263e-05,
+      "loss": 0.004,
+      "step": 2253
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3089514066496165e-05,
+      "loss": 0.0057,
+      "step": 2254
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3081841432225064e-05,
+      "loss": 0.0076,
+      "step": 2255
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3074168797953966e-05,
+      "loss": 0.004,
+      "step": 2256
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3066496163682864e-05,
+      "loss": 0.0076,
+      "step": 2257
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3058823529411766e-05,
+      "loss": 0.0091,
+      "step": 2258
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3051150895140666e-05,
+      "loss": 0.0038,
+      "step": 2259
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3043478260869566e-05,
+      "loss": 0.0057,
+      "step": 2260
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3035805626598466e-05,
+      "loss": 0.0057,
+      "step": 2261
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3028132992327367e-05,
+      "loss": 1.1251,
+      "step": 2262
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3020460358056266e-05,
+      "loss": 1.1222,
+      "step": 2263
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.3012787723785167e-05,
+      "loss": 0.0047,
+      "step": 2264
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.3005115089514067e-05,
+      "loss": 0.008,
+      "step": 2265
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.2997442455242967e-05,
+      "loss": 0.0061,
+      "step": 2266
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.2989769820971867e-05,
+      "loss": 1.394,
+      "step": 2267
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2982097186700768e-05,
+      "loss": 0.0039,
+      "step": 2268
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2974424552429667e-05,
+      "loss": 0.0103,
+      "step": 2269
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2966751918158568e-05,
+      "loss": 1.1008,
+      "step": 2270
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2959079283887468e-05,
+      "loss": 0.0051,
+      "step": 2271
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2951406649616368e-05,
+      "loss": 0.004,
+      "step": 2272
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2943734015345268e-05,
+      "loss": 0.78,
+      "step": 2273
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.293606138107417e-05,
+      "loss": 0.0069,
+      "step": 2274
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2928388746803068e-05,
+      "loss": 0.0094,
+      "step": 2275
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.292071611253197e-05,
+      "loss": 0.0067,
+      "step": 2276
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.291304347826087e-05,
+      "loss": 0.0041,
+      "step": 2277
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.290537084398977e-05,
+      "loss": 0.0079,
+      "step": 2278
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.2897698209718671e-05,
+      "loss": 0.014,
+      "step": 2279
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.2890025575447571e-05,
+      "loss": 1.3775,
+      "step": 2280
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.288235294117647e-05,
+      "loss": 0.0042,
+      "step": 2281
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.287468030690537e-05,
+      "loss": 1.1862,
+      "step": 2282
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2867007672634272e-05,
+      "loss": 0.0042,
+      "step": 2283
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2859335038363172e-05,
+      "loss": 0.0098,
+      "step": 2284
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2851662404092072e-05,
+      "loss": 0.0038,
+      "step": 2285
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2843989769820972e-05,
+      "loss": 0.0127,
+      "step": 2286
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2836317135549874e-05,
+      "loss": 0.0072,
+      "step": 2287
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2828644501278772e-05,
+      "loss": 0.0098,
+      "step": 2288
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2820971867007674e-05,
+      "loss": 0.0069,
+      "step": 2289
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2813299232736573e-05,
+      "loss": 0.9529,
+      "step": 2290
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2805626598465473e-05,
+      "loss": 0.0096,
+      "step": 2291
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2797953964194373e-05,
+      "loss": 0.0094,
+      "step": 2292
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2790281329923275e-05,
+      "loss": 1.0639,
+      "step": 2293
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2782608695652173e-05,
+      "loss": 0.0043,
+      "step": 2294
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2774936061381075e-05,
+      "loss": 0.0098,
+      "step": 2295
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2767263427109975e-05,
+      "loss": 0.0084,
+      "step": 2296
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2759590792838875e-05,
+      "loss": 0.0071,
+      "step": 2297
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2751918158567774e-05,
+      "loss": 0.0036,
+      "step": 2298
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2744245524296676e-05,
+      "loss": 0.0097,
+      "step": 2299
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2736572890025574e-05,
+      "loss": 0.0072,
+      "step": 2300
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2728900255754476e-05,
+      "loss": 0.0043,
+      "step": 2301
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2721227621483376e-05,
+      "loss": 1.3742,
+      "step": 2302
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2713554987212276e-05,
+      "loss": 0.0086,
+      "step": 2303
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2705882352941177e-05,
+      "loss": 0.0071,
+      "step": 2304
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2698209718670077e-05,
+      "loss": 0.007,
+      "step": 2305
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2690537084398977e-05,
+      "loss": 0.0099,
+      "step": 2306
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2682864450127877e-05,
+      "loss": 1.0437,
+      "step": 2307
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2675191815856779e-05,
+      "loss": 0.01,
+      "step": 2308
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2667519181585679e-05,
+      "loss": 0.1771,
+      "step": 2309
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2659846547314579e-05,
+      "loss": 0.0103,
+      "step": 2310
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.2652173913043479e-05,
+      "loss": 0.0104,
+      "step": 2311
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.264450127877238e-05,
+      "loss": 0.0077,
+      "step": 2312
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.2636828644501278e-05,
+      "loss": 0.0107,
+      "step": 2313
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.262915601023018e-05,
+      "loss": 0.0107,
+      "step": 2314
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.262148337595908e-05,
+      "loss": 0.0135,
+      "step": 2315
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.261381074168798e-05,
+      "loss": 0.0103,
+      "step": 2316
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.260613810741688e-05,
+      "loss": 0.0136,
+      "step": 2317
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.2598465473145781e-05,
+      "loss": 0.0045,
+      "step": 2318
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.259079283887468e-05,
+      "loss": 0.0104,
+      "step": 2319
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2583120204603581e-05,
+      "loss": 0.0073,
+      "step": 2320
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2575447570332481e-05,
+      "loss": 0.9265,
+      "step": 2321
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2567774936061381e-05,
+      "loss": 0.0101,
+      "step": 2322
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2560102301790281e-05,
+      "loss": 1.3704,
+      "step": 2323
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2552429667519183e-05,
+      "loss": 1.415,
+      "step": 2324
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.254475703324808e-05,
+      "loss": 0.0098,
+      "step": 2325
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2537084398976982e-05,
+      "loss": 0.0138,
+      "step": 2326
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2529411764705884e-05,
+      "loss": 0.0064,
+      "step": 2327
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2521739130434782e-05,
+      "loss": 0.0093,
+      "step": 2328
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2514066496163684e-05,
+      "loss": 0.0119,
+      "step": 2329
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2506393861892584e-05,
+      "loss": 0.0097,
+      "step": 2330
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2498721227621484e-05,
+      "loss": 0.0049,
+      "step": 2331
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2491048593350384e-05,
+      "loss": 0.0044,
+      "step": 2332
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2483375959079285e-05,
+      "loss": 0.0067,
+      "step": 2333
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2475703324808183e-05,
+      "loss": 0.0117,
+      "step": 2334
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2468030690537085e-05,
+      "loss": 0.0066,
+      "step": 2335
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2460358056265985e-05,
+      "loss": 1.0804,
+      "step": 2336
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2452685421994887e-05,
+      "loss": 0.0065,
+      "step": 2337
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2445012787723785e-05,
+      "loss": 0.0069,
+      "step": 2338
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2437340153452686e-05,
+      "loss": 0.009,
+      "step": 2339
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2429667519181586e-05,
+      "loss": 0.0067,
+      "step": 2340
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2421994884910486e-05,
+      "loss": 0.0089,
+      "step": 2341
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2414322250639386e-05,
+      "loss": 0.011,
+      "step": 2342
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2406649616368288e-05,
+      "loss": 0.0034,
+      "step": 2343
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2398976982097186e-05,
+      "loss": 1.0842,
+      "step": 2344
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2391304347826088e-05,
+      "loss": 0.0066,
+      "step": 2345
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2383631713554987e-05,
+      "loss": 0.0203,
+      "step": 2346
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2375959079283887e-05,
+      "loss": 0.0046,
+      "step": 2347
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2368286445012787e-05,
+      "loss": 0.9714,
+      "step": 2348
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2360613810741689e-05,
+      "loss": 0.0066,
+      "step": 2349
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2352941176470587e-05,
+      "loss": 0.0091,
+      "step": 2350
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.2345268542199489e-05,
+      "loss": 0.0066,
+      "step": 2351
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.233759590792839e-05,
+      "loss": 0.0081,
+      "step": 2352
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.2329923273657289e-05,
+      "loss": 0.006,
+      "step": 2353
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.232225063938619e-05,
+      "loss": 0.0066,
+      "step": 2354
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.231457800511509e-05,
+      "loss": 0.0114,
+      "step": 2355
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.230690537084399e-05,
+      "loss": 2.7382,
+      "step": 2356
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.229923273657289e-05,
+      "loss": 0.0089,
+      "step": 2357
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.2291560102301792e-05,
+      "loss": 0.0042,
+      "step": 2358
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.228388746803069e-05,
+      "loss": 2.3698,
+      "step": 2359
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2276214833759591e-05,
+      "loss": 0.0065,
+      "step": 2360
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2268542199488491e-05,
+      "loss": 1.0813,
+      "step": 2361
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2260869565217393e-05,
+      "loss": 0.0066,
+      "step": 2362
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2253196930946291e-05,
+      "loss": 0.0092,
+      "step": 2363
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2245524296675193e-05,
+      "loss": 1.3566,
+      "step": 2364
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2237851662404093e-05,
+      "loss": 0.0092,
+      "step": 2365
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2230179028132993e-05,
+      "loss": 1.0739,
+      "step": 2366
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2222506393861893e-05,
+      "loss": 0.0122,
+      "step": 2367
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2214833759590794e-05,
+      "loss": 0.0096,
+      "step": 2368
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2207161125319692e-05,
+      "loss": 0.0094,
+      "step": 2369
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2199488491048594e-05,
+      "loss": 0.0097,
+      "step": 2370
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2191815856777494e-05,
+      "loss": 0.0071,
+      "step": 2371
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2184143222506394e-05,
+      "loss": 0.0085,
+      "step": 2372
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2176470588235294e-05,
+      "loss": 0.0071,
+      "step": 2373
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2168797953964195e-05,
+      "loss": 0.0074,
+      "step": 2374
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2161125319693094e-05,
+      "loss": 0.0084,
+      "step": 2375
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2153452685421995e-05,
+      "loss": 0.008,
+      "step": 2376
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.9267676767676768,
+      "eval_loss": 0.3489837944507599,
+      "eval_runtime": 58.1837,
+      "eval_samples_per_second": 6.806,
+      "eval_steps_per_second": 1.702,
+      "step": 2376
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2145780051150897e-05,
+      "loss": 0.0065,
+      "step": 2377
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2138107416879795e-05,
+      "loss": 0.0097,
+      "step": 2378
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2130434782608697e-05,
+      "loss": 1.3744,
+      "step": 2379
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2122762148337597e-05,
+      "loss": 0.0043,
+      "step": 2380
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2115089514066496e-05,
+      "loss": 0.0071,
+      "step": 2381
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2107416879795396e-05,
+      "loss": 1.36,
+      "step": 2382
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2099744245524298e-05,
+      "loss": 0.0044,
+      "step": 2383
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2092071611253196e-05,
+      "loss": 0.0075,
+      "step": 2384
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2084398976982098e-05,
+      "loss": 0.0073,
+      "step": 2385
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.2076726342710998e-05,
+      "loss": 0.0099,
+      "step": 2386
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.2069053708439898e-05,
+      "loss": 0.0048,
+      "step": 2387
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.2061381074168798e-05,
+      "loss": 0.0074,
+      "step": 2388
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.20537084398977e-05,
+      "loss": 0.01,
+      "step": 2389
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.2046035805626599e-05,
+      "loss": 0.0062,
+      "step": 2390
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.2038363171355499e-05,
+      "loss": 0.0039,
+      "step": 2391
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.2030690537084399e-05,
+      "loss": 0.0054,
+      "step": 2392
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.20230179028133e-05,
+      "loss": 1.3442,
+      "step": 2393
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.2015345268542199e-05,
+      "loss": 0.0074,
+      "step": 2394
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.20076726342711e-05,
+      "loss": 0.0063,
+      "step": 2395
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.0075,
+      "step": 2396
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.19923273657289e-05,
+      "loss": 0.008,
+      "step": 2397
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.19846547314578e-05,
+      "loss": 0.0053,
+      "step": 2398
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.1976982097186702e-05,
+      "loss": 0.0049,
+      "step": 2399
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.19693094629156e-05,
+      "loss": 0.0086,
+      "step": 2400
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.1961636828644502e-05,
+      "loss": 0.0098,
+      "step": 2401
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1953964194373403e-05,
+      "loss": 0.0062,
+      "step": 2402
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1946291560102301e-05,
+      "loss": 0.0052,
+      "step": 2403
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1938618925831203e-05,
+      "loss": 0.005,
+      "step": 2404
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1930946291560103e-05,
+      "loss": 1.696,
+      "step": 2405
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1923273657289003e-05,
+      "loss": 0.0076,
+      "step": 2406
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1915601023017903e-05,
+      "loss": 1.1231,
+      "step": 2407
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1907928388746804e-05,
+      "loss": 0.0074,
+      "step": 2408
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1900255754475703e-05,
+      "loss": 0.0097,
+      "step": 2409
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1892583120204604e-05,
+      "loss": 0.0102,
+      "step": 2410
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1884910485933504e-05,
+      "loss": 0.005,
+      "step": 2411
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1877237851662404e-05,
+      "loss": 0.0074,
+      "step": 2412
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1869565217391304e-05,
+      "loss": 0.0071,
+      "step": 2413
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1861892583120206e-05,
+      "loss": 0.0093,
+      "step": 2414
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1854219948849106e-05,
+      "loss": 0.007,
+      "step": 2415
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1846547314578005e-05,
+      "loss": 0.0088,
+      "step": 2416
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1838874680306905e-05,
+      "loss": 0.0049,
+      "step": 2417
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1831202046035807e-05,
+      "loss": 1.2268,
+      "step": 2418
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1823529411764705e-05,
+      "loss": 0.0068,
+      "step": 2419
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1815856777493607e-05,
+      "loss": 0.0112,
+      "step": 2420
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1808184143222507e-05,
+      "loss": 0.0046,
+      "step": 2421
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1800511508951407e-05,
+      "loss": 0.0038,
+      "step": 2422
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1792838874680307e-05,
+      "loss": 0.0088,
+      "step": 2423
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1785166240409208e-05,
+      "loss": 0.0039,
+      "step": 2424
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1777493606138106e-05,
+      "loss": 0.0107,
+      "step": 2425
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.1769820971867008e-05,
+      "loss": 0.0066,
+      "step": 2426
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.176214833759591e-05,
+      "loss": 0.0086,
+      "step": 2427
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.1754475703324808e-05,
+      "loss": 0.0064,
+      "step": 2428
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.174680306905371e-05,
+      "loss": 0.0064,
+      "step": 2429
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.173913043478261e-05,
+      "loss": 0.0042,
+      "step": 2430
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.173145780051151e-05,
+      "loss": 0.0063,
+      "step": 2431
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.172378516624041e-05,
+      "loss": 1.0576,
+      "step": 2432
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.171611253196931e-05,
+      "loss": 0.0041,
+      "step": 2433
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.1708439897698209e-05,
+      "loss": 0.0102,
+      "step": 2434
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.170076726342711e-05,
+      "loss": 1.0312,
+      "step": 2435
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.169309462915601e-05,
+      "loss": 0.1209,
+      "step": 2436
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.168542199488491e-05,
+      "loss": 0.004,
+      "step": 2437
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.167774936061381e-05,
+      "loss": 0.006,
+      "step": 2438
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.1670076726342712e-05,
+      "loss": 0.0081,
+      "step": 2439
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.166240409207161e-05,
+      "loss": 1.1118,
+      "step": 2440
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.1654731457800512e-05,
+      "loss": 0.0059,
+      "step": 2441
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1647058823529412e-05,
+      "loss": 0.0101,
+      "step": 2442
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1639386189258313e-05,
+      "loss": 0.0081,
+      "step": 2443
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1631713554987212e-05,
+      "loss": 0.0032,
+      "step": 2444
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1624040920716113e-05,
+      "loss": 0.0059,
+      "step": 2445
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1616368286445013e-05,
+      "loss": 0.0105,
+      "step": 2446
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1608695652173913e-05,
+      "loss": 0.0038,
+      "step": 2447
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1601023017902813e-05,
+      "loss": 0.0104,
+      "step": 2448
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1593350383631715e-05,
+      "loss": 0.0058,
+      "step": 2449
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1585677749360613e-05,
+      "loss": 0.0058,
+      "step": 2450
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1578005115089514e-05,
+      "loss": 0.0064,
+      "step": 2451
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1570332480818416e-05,
+      "loss": 0.0058,
+      "step": 2452
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1562659846547314e-05,
+      "loss": 0.0077,
+      "step": 2453
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1554987212276216e-05,
+      "loss": 0.0528,
+      "step": 2454
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1547314578005116e-05,
+      "loss": 0.0078,
+      "step": 2455
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1539641943734016e-05,
+      "loss": 0.0023,
+      "step": 2456
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1531969309462916e-05,
+      "loss": 0.0019,
+      "step": 2457
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1524296675191817e-05,
+      "loss": 0.0056,
+      "step": 2458
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1516624040920715e-05,
+      "loss": 0.0116,
+      "step": 2459
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1508951406649617e-05,
+      "loss": 0.0037,
+      "step": 2460
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1501278772378517e-05,
+      "loss": 0.0056,
+      "step": 2461
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1493606138107417e-05,
+      "loss": 1.4099,
+      "step": 2462
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1485933503836317e-05,
+      "loss": 0.0055,
+      "step": 2463
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1478260869565218e-05,
+      "loss": 0.0018,
+      "step": 2464
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1470588235294117e-05,
+      "loss": 0.0081,
+      "step": 2465
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.1462915601023018e-05,
+      "loss": 0.0056,
+      "step": 2466
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.1455242966751918e-05,
+      "loss": 0.0048,
+      "step": 2467
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.144757033248082e-05,
+      "loss": 1.352,
+      "step": 2468
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.1439897698209718e-05,
+      "loss": 1.3368,
+      "step": 2469
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.143222506393862e-05,
+      "loss": 0.0075,
+      "step": 2470
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.142455242966752e-05,
+      "loss": 0.0046,
+      "step": 2471
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.141687979539642e-05,
+      "loss": 0.0036,
+      "step": 2472
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.140920716112532e-05,
+      "loss": 0.0055,
+      "step": 2473
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 1.1401534526854221e-05,
+      "loss": 0.0065,
+      "step": 2474
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 1.1393861892583121e-05,
+      "loss": 0.0055,
+      "step": 2475
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 1.138618925831202e-05,
+      "loss": 0.0057,
+      "step": 2476
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.1378516624040922e-05,
+      "loss": 0.0047,
+      "step": 2477
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.137084398976982e-05,
+      "loss": 0.0036,
+      "step": 2478
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.1363171355498722e-05,
+      "loss": 0.0061,
+      "step": 2479
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.1355498721227622e-05,
+      "loss": 0.0072,
+      "step": 2480
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1347826086956522e-05,
+      "loss": 0.0054,
+      "step": 2481
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1340153452685422e-05,
+      "loss": 0.0072,
+      "step": 2482
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1332480818414324e-05,
+      "loss": 0.0041,
+      "step": 2483
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1324808184143222e-05,
+      "loss": 0.0051,
+      "step": 2484
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1317135549872123e-05,
+      "loss": 1.4203,
+      "step": 2485
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1309462915601023e-05,
+      "loss": 0.0088,
+      "step": 2486
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1301790281329923e-05,
+      "loss": 0.0106,
+      "step": 2487
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1294117647058823e-05,
+      "loss": 0.0071,
+      "step": 2488
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1286445012787725e-05,
+      "loss": 0.0053,
+      "step": 2489
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1278772378516623e-05,
+      "loss": 1.1147,
+      "step": 2490
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1271099744245525e-05,
+      "loss": 0.007,
+      "step": 2491
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1263427109974425e-05,
+      "loss": 0.007,
+      "step": 2492
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1255754475703324e-05,
+      "loss": 0.0035,
+      "step": 2493
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1248081841432224e-05,
+      "loss": 0.0052,
+      "step": 2494
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1240409207161126e-05,
+      "loss": 0.0064,
+      "step": 2495
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1232736572890026e-05,
+      "loss": 0.0046,
+      "step": 2496
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1225063938618926e-05,
+      "loss": 0.0052,
+      "step": 2497
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1217391304347826e-05,
+      "loss": 0.0045,
+      "step": 2498
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1209718670076727e-05,
+      "loss": 0.0052,
+      "step": 2499
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1202046035805627e-05,
+      "loss": 0.0035,
+      "step": 2500
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1194373401534527e-05,
+      "loss": 0.0086,
+      "step": 2501
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1186700767263429e-05,
+      "loss": 0.0068,
+      "step": 2502
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1179028132992327e-05,
+      "loss": 0.0051,
+      "step": 2503
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1171355498721229e-05,
+      "loss": 0.0026,
+      "step": 2504
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.1163682864450129e-05,
+      "loss": 0.0034,
+      "step": 2505
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.1156010230179029e-05,
+      "loss": 0.0052,
+      "step": 2506
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.1148337595907928e-05,
+      "loss": 0.0051,
+      "step": 2507
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.114066496163683e-05,
+      "loss": 0.005,
+      "step": 2508
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.1132992327365728e-05,
+      "loss": 0.005,
+      "step": 2509
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.112531969309463e-05,
+      "loss": 0.0043,
+      "step": 2510
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.111764705882353e-05,
+      "loss": 0.0065,
+      "step": 2511
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.110997442455243e-05,
+      "loss": 0.0039,
+      "step": 2512
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.110230179028133e-05,
+      "loss": 1.4148,
+      "step": 2513
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.1094629156010231e-05,
+      "loss": 0.0049,
+      "step": 2514
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.108695652173913e-05,
+      "loss": 0.0058,
+      "step": 2515
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.1079283887468031e-05,
+      "loss": 1.3611,
+      "step": 2516
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.1071611253196931e-05,
+      "loss": 0.0057,
+      "step": 2517
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.1063938618925831e-05,
+      "loss": 0.0049,
+      "step": 2518
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.105626598465473e-05,
+      "loss": 0.0049,
+      "step": 2519
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.1048593350383632e-05,
+      "loss": 0.0026,
+      "step": 2520
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1040920716112532e-05,
+      "loss": 0.0026,
+      "step": 2521
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1033248081841432e-05,
+      "loss": 0.4305,
+      "step": 2522
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1025575447570332e-05,
+      "loss": 0.0069,
+      "step": 2523
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1017902813299234e-05,
+      "loss": 0.005,
+      "step": 2524
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.1010230179028134e-05,
+      "loss": 1.3999,
+      "step": 2525
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.1002557544757034e-05,
+      "loss": 0.1888,
+      "step": 2526
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.0994884910485935e-05,
+      "loss": 0.0049,
+      "step": 2527
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.0987212276214833e-05,
+      "loss": 0.8899,
+      "step": 2528
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0979539641943735e-05,
+      "loss": 1.0806,
+      "step": 2529
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0971867007672635e-05,
+      "loss": 0.0083,
+      "step": 2530
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0964194373401535e-05,
+      "loss": 0.0034,
+      "step": 2531
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0956521739130435e-05,
+      "loss": 0.005,
+      "step": 2532
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0948849104859336e-05,
+      "loss": 0.0051,
+      "step": 2533
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0941176470588235e-05,
+      "loss": 0.0057,
+      "step": 2534
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0933503836317136e-05,
+      "loss": 0.0058,
+      "step": 2535
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0925831202046036e-05,
+      "loss": 0.0065,
+      "step": 2536
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0918158567774936e-05,
+      "loss": 1.1058,
+      "step": 2537
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0910485933503836e-05,
+      "loss": 0.0052,
+      "step": 2538
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0902813299232738e-05,
+      "loss": 0.0082,
+      "step": 2539
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0895140664961636e-05,
+      "loss": 0.005,
+      "step": 2540
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0887468030690537e-05,
+      "loss": 0.005,
+      "step": 2541
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0879795396419437e-05,
+      "loss": 0.005,
+      "step": 2542
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0872122762148337e-05,
+      "loss": 0.0034,
+      "step": 2543
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0864450127877237e-05,
+      "loss": 0.0066,
+      "step": 2544
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0856777493606139e-05,
+      "loss": 0.0052,
+      "step": 2545
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0849104859335037e-05,
+      "loss": 0.0039,
+      "step": 2546
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0841432225063939e-05,
+      "loss": 0.0039,
+      "step": 2547
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0833759590792839e-05,
+      "loss": 0.0064,
+      "step": 2548
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.082608695652174e-05,
+      "loss": 0.0064,
+      "step": 2549
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.081841432225064e-05,
+      "loss": 0.0052,
+      "step": 2550
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.081074168797954e-05,
+      "loss": 0.0082,
+      "step": 2551
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.0803069053708442e-05,
+      "loss": 0.0035,
+      "step": 2552
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.079539641943734e-05,
+      "loss": 0.0064,
+      "step": 2553
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.0787723785166241e-05,
+      "loss": 0.0064,
+      "step": 2554
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.0780051150895141e-05,
+      "loss": 0.1541,
+      "step": 2555
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.0772378516624041e-05,
+      "loss": 2.4749,
+      "step": 2556
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0764705882352941e-05,
+      "loss": 0.0028,
+      "step": 2557
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0757033248081843e-05,
+      "loss": 0.0034,
+      "step": 2558
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0749360613810741e-05,
+      "loss": 0.0061,
+      "step": 2559
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0741687979539643e-05,
+      "loss": 0.0032,
+      "step": 2560
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0734015345268543e-05,
+      "loss": 0.0055,
+      "step": 2561
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0726342710997443e-05,
+      "loss": 0.0046,
+      "step": 2562
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0718670076726342e-05,
+      "loss": 1.185,
+      "step": 2563
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0710997442455244e-05,
+      "loss": 0.0062,
+      "step": 2564
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0703324808184142e-05,
+      "loss": 0.0024,
+      "step": 2565
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0695652173913044e-05,
+      "loss": 0.0032,
+      "step": 2566
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0687979539641944e-05,
+      "loss": 0.0047,
+      "step": 2567
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0680306905370844e-05,
+      "loss": 0.0047,
+      "step": 2568
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0672634271099744e-05,
+      "loss": 0.0047,
+      "step": 2569
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0664961636828645e-05,
+      "loss": 0.0065,
+      "step": 2570
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0657289002557543e-05,
+      "loss": 0.0061,
+      "step": 2571
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0649616368286445e-05,
+      "loss": 0.0048,
+      "step": 2572
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 1.0641943734015345e-05,
+      "loss": 0.0015,
+      "step": 2573
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 1.0634271099744247e-05,
+      "loss": 0.0046,
+      "step": 2574
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 1.0626598465473147e-05,
+      "loss": 0.0033,
+      "step": 2575
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0618925831202046e-05,
+      "loss": 0.0062,
+      "step": 2576
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0611253196930948e-05,
+      "loss": 1.4243,
+      "step": 2577
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0603580562659846e-05,
+      "loss": 0.012,
+      "step": 2578
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0595907928388748e-05,
+      "loss": 2.3898,
+      "step": 2579
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.0588235294117648e-05,
+      "loss": 0.0032,
+      "step": 2580
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.0580562659846548e-05,
+      "loss": 0.0032,
+      "step": 2581
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.0572890025575448e-05,
+      "loss": 0.0061,
+      "step": 2582
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.056521739130435e-05,
+      "loss": 0.0047,
+      "step": 2583
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0557544757033247e-05,
+      "loss": 0.0047,
+      "step": 2584
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0549872122762149e-05,
+      "loss": 0.2081,
+      "step": 2585
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0542199488491049e-05,
+      "loss": 0.0053,
+      "step": 2586
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0534526854219949e-05,
+      "loss": 0.0047,
+      "step": 2587
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.0526854219948849e-05,
+      "loss": 1.3858,
+      "step": 2588
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.051918158567775e-05,
+      "loss": 0.0062,
+      "step": 2589
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.0511508951406649e-05,
+      "loss": 0.0041,
+      "step": 2590
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.050383631713555e-05,
+      "loss": 1.1796,
+      "step": 2591
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.049616368286445e-05,
+      "loss": 1.0994,
+      "step": 2592
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.048849104859335e-05,
+      "loss": 0.8511,
+      "step": 2593
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.048081841432225e-05,
+      "loss": 1.2895,
+      "step": 2594
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.0473145780051152e-05,
+      "loss": 0.0035,
+      "step": 2595
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.046547314578005e-05,
+      "loss": 0.0065,
+      "step": 2596
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.0457800511508952e-05,
+      "loss": 1.239,
+      "step": 2597
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.0450127877237851e-05,
+      "loss": 1.1663,
+      "step": 2598
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.0442455242966751e-05,
+      "loss": 0.0051,
+      "step": 2599
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0434782608695653e-05,
+      "loss": 1.1103,
+      "step": 2600
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0427109974424553e-05,
+      "loss": 1.4103,
+      "step": 2601
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0419437340153454e-05,
+      "loss": 0.0053,
+      "step": 2602
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0411764705882353e-05,
+      "loss": 0.0072,
+      "step": 2603
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0404092071611254e-05,
+      "loss": 0.0083,
+      "step": 2604
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0396419437340154e-05,
+      "loss": 0.0039,
+      "step": 2605
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0388746803069054e-05,
+      "loss": 0.0049,
+      "step": 2606
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0381074168797954e-05,
+      "loss": 0.0098,
+      "step": 2607
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0373401534526856e-05,
+      "loss": 0.0057,
+      "step": 2608
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0365728900255754e-05,
+      "loss": 0.004,
+      "step": 2609
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0358056265984656e-05,
+      "loss": 0.0098,
+      "step": 2610
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0350383631713555e-05,
+      "loss": 0.0059,
+      "step": 2611
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0342710997442455e-05,
+      "loss": 0.0065,
+      "step": 2612
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0335038363171355e-05,
+      "loss": 0.0059,
+      "step": 2613
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0327365728900257e-05,
+      "loss": 0.0112,
+      "step": 2614
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0319693094629155e-05,
+      "loss": 0.0061,
+      "step": 2615
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0312020460358057e-05,
+      "loss": 0.0039,
+      "step": 2616
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0304347826086957e-05,
+      "loss": 0.0083,
+      "step": 2617
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0296675191815857e-05,
+      "loss": 0.0057,
+      "step": 2618
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0289002557544756e-05,
+      "loss": 1.3969,
+      "step": 2619
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0281329923273658e-05,
+      "loss": 0.0065,
+      "step": 2620
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0273657289002556e-05,
+      "loss": 1.0742,
+      "step": 2621
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0265984654731458e-05,
+      "loss": 0.0039,
+      "step": 2622
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0258312020460358e-05,
+      "loss": 0.0065,
+      "step": 2623
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.0250639386189258e-05,
+      "loss": 0.005,
+      "step": 2624
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.024296675191816e-05,
+      "loss": 0.0077,
+      "step": 2625
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.023529411764706e-05,
+      "loss": 1.2389,
+      "step": 2626
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.0227621483375961e-05,
+      "loss": 0.0076,
+      "step": 2627
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.0219948849104859e-05,
+      "loss": 1.034,
+      "step": 2628
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.021227621483376e-05,
+      "loss": 0.0058,
+      "step": 2629
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.020460358056266e-05,
+      "loss": 0.0059,
+      "step": 2630
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.019693094629156e-05,
+      "loss": 0.0079,
+      "step": 2631
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.018925831202046e-05,
+      "loss": 0.0043,
+      "step": 2632
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.0181585677749362e-05,
+      "loss": 0.004,
+      "step": 2633
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.017391304347826e-05,
+      "loss": 0.0075,
+      "step": 2634
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.0166240409207162e-05,
+      "loss": 0.0076,
+      "step": 2635
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0158567774936062e-05,
+      "loss": 0.0102,
+      "step": 2636
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0150895140664962e-05,
+      "loss": 1.1332,
+      "step": 2637
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0143222506393862e-05,
+      "loss": 0.0096,
+      "step": 2638
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0135549872122763e-05,
+      "loss": 0.0092,
+      "step": 2639
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0127877237851662e-05,
+      "loss": 1.3919,
+      "step": 2640
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0120204603580563e-05,
+      "loss": 0.0058,
+      "step": 2641
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0112531969309463e-05,
+      "loss": 1.0521,
+      "step": 2642
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0104859335038363e-05,
+      "loss": 0.0045,
+      "step": 2643
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0097186700767263e-05,
+      "loss": 0.0059,
+      "step": 2644
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0089514066496164e-05,
+      "loss": 0.005,
+      "step": 2645
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0081841432225063e-05,
+      "loss": 0.0075,
+      "step": 2646
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0074168797953964e-05,
+      "loss": 0.0403,
+      "step": 2647
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0066496163682864e-05,
+      "loss": 0.0079,
+      "step": 2648
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0058823529411764e-05,
+      "loss": 0.0062,
+      "step": 2649
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0051150895140666e-05,
+      "loss": 0.0061,
+      "step": 2650
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0043478260869566e-05,
+      "loss": 0.0064,
+      "step": 2651
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0035805626598466e-05,
+      "loss": 0.0077,
+      "step": 2652
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0028132992327366e-05,
+      "loss": 0.0884,
+      "step": 2653
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0020460358056267e-05,
+      "loss": 0.0062,
+      "step": 2654
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0012787723785167e-05,
+      "loss": 1.1006,
+      "step": 2655
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 1.0005115089514067e-05,
+      "loss": 0.0109,
+      "step": 2656
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 9.997442455242967e-06,
+      "loss": 0.0059,
+      "step": 2657
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 9.989769820971869e-06,
+      "loss": 0.0084,
+      "step": 2658
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 9.982097186700767e-06,
+      "loss": 0.0058,
+      "step": 2659
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.974424552429668e-06,
+      "loss": 0.0059,
+      "step": 2660
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.966751918158568e-06,
+      "loss": 0.0049,
+      "step": 2661
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.959079283887468e-06,
+      "loss": 0.0079,
+      "step": 2662
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.951406649616368e-06,
+      "loss": 1.5801,
+      "step": 2663
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.94373401534527e-06,
+      "loss": 0.0059,
+      "step": 2664
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.936061381074168e-06,
+      "loss": 1.2523,
+      "step": 2665
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.92838874680307e-06,
+      "loss": 0.0079,
+      "step": 2666
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.92071611253197e-06,
+      "loss": 0.0099,
+      "step": 2667
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.91304347826087e-06,
+      "loss": 0.0082,
+      "step": 2668
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.90537084398977e-06,
+      "loss": 0.004,
+      "step": 2669
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.897698209718671e-06,
+      "loss": 0.0064,
+      "step": 2670
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.890025575447569e-06,
+      "loss": 0.004,
+      "step": 2671
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 9.88235294117647e-06,
+      "loss": 0.0943,
+      "step": 2672
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 9.87468030690537e-06,
+      "loss": 0.0059,
+      "step": 2673
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 9.86700767263427e-06,
+      "loss": 0.0098,
+      "step": 2674
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.859335038363172e-06,
+      "loss": 0.0065,
+      "step": 2675
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.851662404092072e-06,
+      "loss": 0.0075,
+      "step": 2676
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.843989769820972e-06,
+      "loss": 0.0044,
+      "step": 2677
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.836317135549872e-06,
+      "loss": 0.0063,
+      "step": 2678
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.828644501278774e-06,
+      "loss": 0.0038,
+      "step": 2679
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.820971867007673e-06,
+      "loss": 0.0067,
+      "step": 2680
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.813299232736573e-06,
+      "loss": 1.1148,
+      "step": 2681
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.805626598465473e-06,
+      "loss": 0.0056,
+      "step": 2682
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.797953964194375e-06,
+      "loss": 0.0096,
+      "step": 2683
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.790281329923273e-06,
+      "loss": 0.011,
+      "step": 2684
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.782608695652175e-06,
+      "loss": 0.0038,
+      "step": 2685
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.774936061381075e-06,
+      "loss": 0.0129,
+      "step": 2686
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.767263427109975e-06,
+      "loss": 0.0063,
+      "step": 2687
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.759590792838875e-06,
+      "loss": 0.0561,
+      "step": 2688
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.751918158567776e-06,
+      "loss": 0.0041,
+      "step": 2689
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.744245524296674e-06,
+      "loss": 0.0048,
+      "step": 2690
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.736572890025576e-06,
+      "loss": 0.0072,
+      "step": 2691
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.728900255754476e-06,
+      "loss": 0.007,
+      "step": 2692
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.721227621483376e-06,
+      "loss": 0.0031,
+      "step": 2693
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.713554987212276e-06,
+      "loss": 1.4136,
+      "step": 2694
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.705882352941177e-06,
+      "loss": 0.0081,
+      "step": 2695
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.698209718670076e-06,
+      "loss": 0.0036,
+      "step": 2696
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.690537084398977e-06,
+      "loss": 1.382,
+      "step": 2697
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.682864450127877e-06,
+      "loss": 0.0054,
+      "step": 2698
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.675191815856777e-06,
+      "loss": 0.0064,
+      "step": 2699
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.667519181585679e-06,
+      "loss": 0.0072,
+      "step": 2700
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.659846547314579e-06,
+      "loss": 0.0054,
+      "step": 2701
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.652173913043478e-06,
+      "loss": 0.0108,
+      "step": 2702
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.644501278772378e-06,
+      "loss": 0.0037,
+      "step": 2703
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.63682864450128e-06,
+      "loss": 0.0071,
+      "step": 2704
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.629156010230178e-06,
+      "loss": 0.0062,
+      "step": 2705
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.62148337595908e-06,
+      "loss": 0.0056,
+      "step": 2706
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.61381074168798e-06,
+      "loss": 1.1276,
+      "step": 2707
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.606138107416881e-06,
+      "loss": 0.0045,
+      "step": 2708
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.59846547314578e-06,
+      "loss": 0.0073,
+      "step": 2709
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.590792838874681e-06,
+      "loss": 0.3343,
+      "step": 2710
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.583120204603581e-06,
+      "loss": 0.0038,
+      "step": 2711
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.575447570332481e-06,
+      "loss": 0.0053,
+      "step": 2712
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.567774936061381e-06,
+      "loss": 0.0037,
+      "step": 2713
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.560102301790283e-06,
+      "loss": 0.0036,
+      "step": 2714
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.55242966751918e-06,
+      "loss": 0.0053,
+      "step": 2715
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.544757033248082e-06,
+      "loss": 0.0069,
+      "step": 2716
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.537084398976982e-06,
+      "loss": 1.3084,
+      "step": 2717
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.529411764705882e-06,
+      "loss": 0.0022,
+      "step": 2718
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.521739130434782e-06,
+      "loss": 0.0046,
+      "step": 2719
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.514066496163684e-06,
+      "loss": 1.1433,
+      "step": 2720
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.506393861892582e-06,
+      "loss": 0.0053,
+      "step": 2721
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.498721227621484e-06,
+      "loss": 0.0053,
+      "step": 2722
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.491048593350385e-06,
+      "loss": 0.0057,
+      "step": 2723
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.483375959079283e-06,
+      "loss": 0.0058,
+      "step": 2724
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.475703324808185e-06,
+      "loss": 0.0029,
+      "step": 2725
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.468030690537085e-06,
+      "loss": 0.0061,
+      "step": 2726
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.460358056265985e-06,
+      "loss": 0.0058,
+      "step": 2727
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.452685421994885e-06,
+      "loss": 0.0074,
+      "step": 2728
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.445012787723786e-06,
+      "loss": 0.007,
+      "step": 2729
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.437340153452685e-06,
+      "loss": 1.3508,
+      "step": 2730
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.429667519181586e-06,
+      "loss": 0.0074,
+      "step": 2731
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.421994884910486e-06,
+      "loss": 0.0037,
+      "step": 2732
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.414322250639388e-06,
+      "loss": 0.003,
+      "step": 2733
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.406649616368286e-06,
+      "loss": 0.0038,
+      "step": 2734
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.398976982097188e-06,
+      "loss": 0.0086,
+      "step": 2735
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.391304347826087e-06,
+      "loss": 0.0058,
+      "step": 2736
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.383631713554987e-06,
+      "loss": 0.0021,
+      "step": 2737
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.375959079283887e-06,
+      "loss": 0.0046,
+      "step": 2738
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.368286445012789e-06,
+      "loss": 0.0035,
+      "step": 2739
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.360613810741687e-06,
+      "loss": 0.0054,
+      "step": 2740
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.352941176470589e-06,
+      "loss": 0.0052,
+      "step": 2741
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.345268542199489e-06,
+      "loss": 0.0072,
+      "step": 2742
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.337595907928389e-06,
+      "loss": 0.1932,
+      "step": 2743
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.329923273657289e-06,
+      "loss": 0.006,
+      "step": 2744
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.32225063938619e-06,
+      "loss": 0.0036,
+      "step": 2745
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.314578005115088e-06,
+      "loss": 1.2929,
+      "step": 2746
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.30690537084399e-06,
+      "loss": 0.0028,
+      "step": 2747
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.299232736572892e-06,
+      "loss": 1.2908,
+      "step": 2748
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.29156010230179e-06,
+      "loss": 0.0067,
+      "step": 2749
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.283887468030691e-06,
+      "loss": 0.0043,
+      "step": 2750
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.276214833759591e-06,
+      "loss": 0.0054,
+      "step": 2751
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.268542199488491e-06,
+      "loss": 0.0055,
+      "step": 2752
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.260869565217391e-06,
+      "loss": 1.1614,
+      "step": 2753
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.253196930946293e-06,
+      "loss": 0.0071,
+      "step": 2754
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.245524296675191e-06,
+      "loss": 0.0035,
+      "step": 2755
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.237851662404093e-06,
+      "loss": 0.0054,
+      "step": 2756
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.230179028132993e-06,
+      "loss": 0.0096,
+      "step": 2757
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.222506393861894e-06,
+      "loss": 0.0035,
+      "step": 2758
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.214833759590792e-06,
+      "loss": 0.0051,
+      "step": 2759
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.207161125319694e-06,
+      "loss": 0.0069,
+      "step": 2760
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.199488491048594e-06,
+      "loss": 0.0018,
+      "step": 2761
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.191815856777494e-06,
+      "loss": 0.0026,
+      "step": 2762
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.184143222506394e-06,
+      "loss": 0.0054,
+      "step": 2763
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.176470588235295e-06,
+      "loss": 0.0085,
+      "step": 2764
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.168797953964194e-06,
+      "loss": 0.008,
+      "step": 2765
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.161125319693095e-06,
+      "loss": 1.3292,
+      "step": 2766
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.153452685421995e-06,
+      "loss": 0.0085,
+      "step": 2767
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.145780051150895e-06,
+      "loss": 0.0034,
+      "step": 2768
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.138107416879795e-06,
+      "loss": 0.0051,
+      "step": 2769
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.130434782608697e-06,
+      "loss": 0.0054,
+      "step": 2770
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 9.122762148337595e-06,
+      "loss": 0.0054,
+      "step": 2771
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 9.115089514066496e-06,
+      "loss": 0.0083,
+      "step": 2772
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.9217171717171717,
+      "eval_loss": 0.37921515107154846,
+      "eval_runtime": 57.416,
+      "eval_samples_per_second": 6.897,
+      "eval_steps_per_second": 1.724,
+      "step": 2772
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 9.107416879795398e-06,
+      "loss": 1.4643,
+      "step": 2773
+    },
+    {
+      "epoch": 7.01,
+      "learning_rate": 9.099744245524296e-06,
+      "loss": 0.0067,
+      "step": 2774
+    },
+    {
+      "epoch": 7.01,
+      "learning_rate": 9.092071611253198e-06,
+      "loss": 1.362,
+      "step": 2775
+    },
+    {
+      "epoch": 7.01,
+      "learning_rate": 9.084398976982098e-06,
+      "loss": 0.0035,
+      "step": 2776
+    },
+    {
+      "epoch": 7.01,
+      "learning_rate": 9.076726342710998e-06,
+      "loss": 0.0043,
+      "step": 2777
+    },
+    {
+      "epoch": 7.02,
+      "learning_rate": 9.069053708439898e-06,
+      "loss": 0.005,
+      "step": 2778
+    },
+    {
+      "epoch": 7.02,
+      "learning_rate": 9.0613810741688e-06,
+      "loss": 0.0049,
+      "step": 2779
+    },
+    {
+      "epoch": 7.02,
+      "learning_rate": 9.053708439897697e-06,
+      "loss": 0.007,
+      "step": 2780
+    },
+    {
+      "epoch": 7.02,
+      "learning_rate": 9.046035805626599e-06,
+      "loss": 0.0038,
+      "step": 2781
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 9.038363171355499e-06,
+      "loss": 0.7287,
+      "step": 2782
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 9.030690537084399e-06,
+      "loss": 0.0089,
+      "step": 2783
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 9.023017902813299e-06,
+      "loss": 0.0067,
+      "step": 2784
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 9.0153452685422e-06,
+      "loss": 0.0065,
+      "step": 2785
+    },
+    {
+      "epoch": 7.04,
+      "learning_rate": 9.0076726342711e-06,
+      "loss": 0.0066,
+      "step": 2786
+    },
+    {
+      "epoch": 7.04,
+      "learning_rate": 9e-06,
+      "loss": 1.341,
+      "step": 2787
+    },
+    {
+      "epoch": 7.04,
+      "learning_rate": 8.9923273657289e-06,
+      "loss": 0.0039,
+      "step": 2788
+    },
+    {
+      "epoch": 7.04,
+      "learning_rate": 8.984654731457802e-06,
+      "loss": 0.0053,
+      "step": 2789
+    },
+    {
+      "epoch": 7.05,
+      "learning_rate": 8.9769820971867e-06,
+      "loss": 0.0058,
+      "step": 2790
+    },
+    {
+      "epoch": 7.05,
+      "learning_rate": 8.969309462915602e-06,
+      "loss": 1.0835,
+      "step": 2791
+    },
+    {
+      "epoch": 7.05,
+      "learning_rate": 8.961636828644502e-06,
+      "loss": 0.0059,
+      "step": 2792
+    },
+    {
+      "epoch": 7.05,
+      "learning_rate": 8.953964194373401e-06,
+      "loss": 0.0067,
+      "step": 2793
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 8.946291560102301e-06,
+      "loss": 0.0059,
+      "step": 2794
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 8.938618925831203e-06,
+      "loss": 0.0031,
+      "step": 2795
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 8.930946291560101e-06,
+      "loss": 0.005,
+      "step": 2796
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 8.923273657289003e-06,
+      "loss": 0.008,
+      "step": 2797
+    },
+    {
+      "epoch": 7.07,
+      "learning_rate": 8.915601023017904e-06,
+      "loss": 0.0052,
+      "step": 2798
+    },
+    {
+      "epoch": 7.07,
+      "learning_rate": 8.907928388746803e-06,
+      "loss": 0.0634,
+      "step": 2799
+    },
+    {
+      "epoch": 7.07,
+      "learning_rate": 8.900255754475704e-06,
+      "loss": 0.0053,
+      "step": 2800
+    },
+    {
+      "epoch": 7.07,
+      "learning_rate": 8.892583120204604e-06,
+      "loss": 0.0054,
+      "step": 2801
+    },
+    {
+      "epoch": 7.08,
+      "learning_rate": 8.884910485933504e-06,
+      "loss": 0.0066,
+      "step": 2802
+    },
+    {
+      "epoch": 7.08,
+      "learning_rate": 8.877237851662404e-06,
+      "loss": 0.0067,
+      "step": 2803
+    },
+    {
+      "epoch": 7.08,
+      "learning_rate": 8.869565217391306e-06,
+      "loss": 0.0488,
+      "step": 2804
+    },
+    {
+      "epoch": 7.08,
+      "learning_rate": 8.861892583120204e-06,
+      "loss": 0.0048,
+      "step": 2805
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 8.854219948849105e-06,
+      "loss": 0.0065,
+      "step": 2806
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 8.846547314578005e-06,
+      "loss": 0.0089,
+      "step": 2807
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 8.838874680306905e-06,
+      "loss": 0.0052,
+      "step": 2808
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 8.831202046035805e-06,
+      "loss": 0.006,
+      "step": 2809
+    },
+    {
+      "epoch": 7.1,
+      "learning_rate": 8.823529411764707e-06,
+      "loss": 0.0032,
+      "step": 2810
+    },
+    {
+      "epoch": 7.1,
+      "learning_rate": 8.815856777493607e-06,
+      "loss": 0.0036,
+      "step": 2811
+    },
+    {
+      "epoch": 7.1,
+      "learning_rate": 8.808184143222507e-06,
+      "loss": 0.0032,
+      "step": 2812
+    },
+    {
+      "epoch": 7.1,
+      "learning_rate": 8.800511508951407e-06,
+      "loss": 0.0035,
+      "step": 2813
+    },
+    {
+      "epoch": 7.11,
+      "learning_rate": 8.792838874680308e-06,
+      "loss": 1.4438,
+      "step": 2814
+    },
+    {
+      "epoch": 7.11,
+      "learning_rate": 8.785166240409206e-06,
+      "loss": 0.0048,
+      "step": 2815
+    },
+    {
+      "epoch": 7.11,
+      "learning_rate": 8.777493606138108e-06,
+      "loss": 0.0052,
+      "step": 2816
+    },
+    {
+      "epoch": 7.11,
+      "learning_rate": 8.769820971867008e-06,
+      "loss": 0.0038,
+      "step": 2817
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 8.762148337595908e-06,
+      "loss": 0.7281,
+      "step": 2818
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 8.754475703324808e-06,
+      "loss": 1.3888,
+      "step": 2819
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 8.74680306905371e-06,
+      "loss": 0.0016,
+      "step": 2820
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 8.739130434782608e-06,
+      "loss": 0.0065,
+      "step": 2821
+    },
+    {
+      "epoch": 7.13,
+      "learning_rate": 8.73145780051151e-06,
+      "loss": 1.1887,
+      "step": 2822
+    },
+    {
+      "epoch": 7.13,
+      "learning_rate": 8.72378516624041e-06,
+      "loss": 0.0069,
+      "step": 2823
+    },
+    {
+      "epoch": 7.13,
+      "learning_rate": 8.716112531969309e-06,
+      "loss": 0.0034,
+      "step": 2824
+    },
+    {
+      "epoch": 7.13,
+      "learning_rate": 8.70843989769821e-06,
+      "loss": 1.1691,
+      "step": 2825
+    },
+    {
+      "epoch": 7.14,
+      "learning_rate": 8.70076726342711e-06,
+      "loss": 0.0033,
+      "step": 2826
+    },
+    {
+      "epoch": 7.14,
+      "learning_rate": 8.69309462915601e-06,
+      "loss": 0.0088,
+      "step": 2827
+    },
+    {
+      "epoch": 7.14,
+      "learning_rate": 8.68542199488491e-06,
+      "loss": 0.0071,
+      "step": 2828
+    },
+    {
+      "epoch": 7.14,
+      "learning_rate": 8.677749360613812e-06,
+      "loss": 0.0023,
+      "step": 2829
+    },
+    {
+      "epoch": 7.15,
+      "learning_rate": 8.67007672634271e-06,
+      "loss": 0.0036,
+      "step": 2830
+    },
+    {
+      "epoch": 7.15,
+      "learning_rate": 8.662404092071612e-06,
+      "loss": 0.0088,
+      "step": 2831
+    },
+    {
+      "epoch": 7.15,
+      "learning_rate": 8.654731457800512e-06,
+      "loss": 1.1488,
+      "step": 2832
+    },
+    {
+      "epoch": 7.15,
+      "learning_rate": 8.647058823529412e-06,
+      "loss": 0.0067,
+      "step": 2833
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 8.639386189258312e-06,
+      "loss": 0.0031,
+      "step": 2834
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 8.631713554987213e-06,
+      "loss": 0.0036,
+      "step": 2835
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 8.624040920716111e-06,
+      "loss": 0.0058,
+      "step": 2836
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 8.616368286445013e-06,
+      "loss": 0.03,
+      "step": 2837
+    },
+    {
+      "epoch": 7.17,
+      "learning_rate": 8.608695652173913e-06,
+      "loss": 0.0046,
+      "step": 2838
+    },
+    {
+      "epoch": 7.17,
+      "learning_rate": 8.601023017902815e-06,
+      "loss": 0.0093,
+      "step": 2839
+    },
+    {
+      "epoch": 7.17,
+      "learning_rate": 8.593350383631713e-06,
+      "loss": 0.0054,
+      "step": 2840
+    },
+    {
+      "epoch": 7.17,
+      "learning_rate": 8.585677749360614e-06,
+      "loss": 0.0051,
+      "step": 2841
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 8.578005115089514e-06,
+      "loss": 0.0036,
+      "step": 2842
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 8.570332480818414e-06,
+      "loss": 0.0016,
+      "step": 2843
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 8.562659846547314e-06,
+      "loss": 0.0057,
+      "step": 2844
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 8.554987212276216e-06,
+      "loss": 0.0041,
+      "step": 2845
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 8.547314578005114e-06,
+      "loss": 0.0035,
+      "step": 2846
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 8.539641943734016e-06,
+      "loss": 0.0084,
+      "step": 2847
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 8.531969309462917e-06,
+      "loss": 0.0065,
+      "step": 2848
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 8.524296675191815e-06,
+      "loss": 0.0066,
+      "step": 2849
+    },
+    {
+      "epoch": 7.2,
+      "learning_rate": 8.516624040920717e-06,
+      "loss": 0.0035,
+      "step": 2850
+    },
+    {
+      "epoch": 7.2,
+      "learning_rate": 8.508951406649617e-06,
+      "loss": 0.0052,
+      "step": 2851
+    },
+    {
+      "epoch": 7.2,
+      "learning_rate": 8.501278772378517e-06,
+      "loss": 0.0065,
+      "step": 2852
+    },
+    {
+      "epoch": 7.2,
+      "learning_rate": 8.493606138107417e-06,
+      "loss": 0.0064,
+      "step": 2853
+    },
+    {
+      "epoch": 7.21,
+      "learning_rate": 8.485933503836318e-06,
+      "loss": 0.0154,
+      "step": 2854
+    },
+    {
+      "epoch": 7.21,
+      "learning_rate": 8.478260869565217e-06,
+      "loss": 0.0114,
+      "step": 2855
+    },
+    {
+      "epoch": 7.21,
+      "learning_rate": 8.470588235294118e-06,
+      "loss": 0.0063,
+      "step": 2856
+    },
+    {
+      "epoch": 7.21,
+      "learning_rate": 8.462915601023018e-06,
+      "loss": 0.0097,
+      "step": 2857
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 8.455242966751918e-06,
+      "loss": 0.0051,
+      "step": 2858
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 8.447570332480818e-06,
+      "loss": 0.0081,
+      "step": 2859
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 8.43989769820972e-06,
+      "loss": 0.0046,
+      "step": 2860
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 8.432225063938618e-06,
+      "loss": 0.0034,
+      "step": 2861
+    },
+    {
+      "epoch": 7.23,
+      "learning_rate": 8.42455242966752e-06,
+      "loss": 0.0053,
+      "step": 2862
+    },
+    {
+      "epoch": 7.23,
+      "learning_rate": 8.41687979539642e-06,
+      "loss": 0.0071,
+      "step": 2863
+    },
+    {
+      "epoch": 7.23,
+      "learning_rate": 8.409207161125321e-06,
+      "loss": 0.0062,
+      "step": 2864
+    },
+    {
+      "epoch": 7.23,
+      "learning_rate": 8.40153452685422e-06,
+      "loss": 0.0106,
+      "step": 2865
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 8.39386189258312e-06,
+      "loss": 0.0038,
+      "step": 2866
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 8.38618925831202e-06,
+      "loss": 0.003,
+      "step": 2867
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 8.37851662404092e-06,
+      "loss": 0.0044,
+      "step": 2868
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 8.37084398976982e-06,
+      "loss": 0.0061,
+      "step": 2869
+    },
+    {
+      "epoch": 7.25,
+      "learning_rate": 8.363171355498722e-06,
+      "loss": 0.0061,
+      "step": 2870
+    },
+    {
+      "epoch": 7.25,
+      "learning_rate": 8.355498721227622e-06,
+      "loss": 0.0071,
+      "step": 2871
+    },
+    {
+      "epoch": 7.25,
+      "learning_rate": 8.347826086956522e-06,
+      "loss": 0.0044,
+      "step": 2872
+    },
+    {
+      "epoch": 7.26,
+      "learning_rate": 8.340153452685424e-06,
+      "loss": 0.0429,
+      "step": 2873
+    },
+    {
+      "epoch": 7.26,
+      "learning_rate": 8.332480818414322e-06,
+      "loss": 0.002,
+      "step": 2874
+    },
+    {
+      "epoch": 7.26,
+      "learning_rate": 8.324808184143223e-06,
+      "loss": 0.0056,
+      "step": 2875
+    },
+    {
+      "epoch": 7.26,
+      "learning_rate": 8.317135549872123e-06,
+      "loss": 0.0161,
+      "step": 2876
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 8.309462915601023e-06,
+      "loss": 0.0049,
+      "step": 2877
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 8.301790281329923e-06,
+      "loss": 0.0082,
+      "step": 2878
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 8.294117647058825e-06,
+      "loss": 0.0055,
+      "step": 2879
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 8.286445012787723e-06,
+      "loss": 0.0029,
+      "step": 2880
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 8.278772378516625e-06,
+      "loss": 0.0067,
+      "step": 2881
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 8.271099744245525e-06,
+      "loss": 0.0046,
+      "step": 2882
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 8.263427109974425e-06,
+      "loss": 0.0029,
+      "step": 2883
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 8.255754475703324e-06,
+      "loss": 1.4067,
+      "step": 2884
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 8.248081841432226e-06,
+      "loss": 0.0056,
+      "step": 2885
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 8.240409207161124e-06,
+      "loss": 0.007,
+      "step": 2886
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 8.232736572890026e-06,
+      "loss": 0.0044,
+      "step": 2887
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 8.225063938618926e-06,
+      "loss": 0.2182,
+      "step": 2888
+    },
+    {
+      "epoch": 7.3,
+      "learning_rate": 8.217391304347826e-06,
+      "loss": 0.0028,
+      "step": 2889
+    },
+    {
+      "epoch": 7.3,
+      "learning_rate": 8.209718670076726e-06,
+      "loss": 0.0028,
+      "step": 2890
+    },
+    {
+      "epoch": 7.3,
+      "learning_rate": 8.202046035805627e-06,
+      "loss": 0.0085,
+      "step": 2891
+    },
+    {
+      "epoch": 7.3,
+      "learning_rate": 8.194373401534527e-06,
+      "loss": 0.0059,
+      "step": 2892
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 8.186700767263427e-06,
+      "loss": 0.004,
+      "step": 2893
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 8.179028132992327e-06,
+      "loss": 0.0057,
+      "step": 2894
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 8.171355498721229e-06,
+      "loss": 0.0052,
+      "step": 2895
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 8.163682864450129e-06,
+      "loss": 0.0041,
+      "step": 2896
+    },
+    {
+      "epoch": 7.32,
+      "learning_rate": 8.156010230179028e-06,
+      "loss": 0.0042,
+      "step": 2897
+    },
+    {
+      "epoch": 7.32,
+      "learning_rate": 8.14833759590793e-06,
+      "loss": 0.0066,
+      "step": 2898
+    },
+    {
+      "epoch": 7.32,
+      "learning_rate": 8.140664961636828e-06,
+      "loss": 0.0042,
+      "step": 2899
+    },
+    {
+      "epoch": 7.32,
+      "learning_rate": 8.13299232736573e-06,
+      "loss": 0.0042,
+      "step": 2900
+    },
+    {
+      "epoch": 7.33,
+      "learning_rate": 8.12531969309463e-06,
+      "loss": 0.0019,
+      "step": 2901
+    },
+    {
+      "epoch": 7.33,
+      "learning_rate": 8.11764705882353e-06,
+      "loss": 0.0054,
+      "step": 2902
+    },
+    {
+      "epoch": 7.33,
+      "learning_rate": 8.10997442455243e-06,
+      "loss": 0.0039,
+      "step": 2903
+    },
+    {
+      "epoch": 7.33,
+      "learning_rate": 8.102301790281331e-06,
+      "loss": 1.1882,
+      "step": 2904
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 8.09462915601023e-06,
+      "loss": 0.0034,
+      "step": 2905
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 8.086956521739131e-06,
+      "loss": 0.0066,
+      "step": 2906
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 8.079283887468031e-06,
+      "loss": 0.0041,
+      "step": 2907
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 8.071611253196931e-06,
+      "loss": 0.0039,
+      "step": 2908
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 8.06393861892583e-06,
+      "loss": 0.0062,
+      "step": 2909
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 8.056265984654732e-06,
+      "loss": 0.0053,
+      "step": 2910
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 8.04859335038363e-06,
+      "loss": 0.0041,
+      "step": 2911
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 8.040920716112532e-06,
+      "loss": 0.059,
+      "step": 2912
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 8.033248081841432e-06,
+      "loss": 0.0027,
+      "step": 2913
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 8.025575447570332e-06,
+      "loss": 0.005,
+      "step": 2914
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 8.017902813299232e-06,
+      "loss": 0.0061,
+      "step": 2915
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 8.010230179028134e-06,
+      "loss": 0.0063,
+      "step": 2916
+    },
+    {
+      "epoch": 7.37,
+      "learning_rate": 8.002557544757034e-06,
+      "loss": 0.0026,
+      "step": 2917
+    },
+    {
+      "epoch": 7.37,
+      "learning_rate": 7.994884910485933e-06,
+      "loss": 0.0037,
+      "step": 2918
+    },
+    {
+      "epoch": 7.37,
+      "learning_rate": 7.987212276214833e-06,
+      "loss": 1.2357,
+      "step": 2919
+    },
+    {
+      "epoch": 7.37,
+      "learning_rate": 7.979539641943735e-06,
+      "loss": 0.0037,
+      "step": 2920
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 7.971867007672635e-06,
+      "loss": 0.0031,
+      "step": 2921
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 7.964194373401535e-06,
+      "loss": 0.0026,
+      "step": 2922
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 7.956521739130436e-06,
+      "loss": 0.006,
+      "step": 2923
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 7.948849104859335e-06,
+      "loss": 0.0032,
+      "step": 2924
+    },
+    {
+      "epoch": 7.39,
+      "learning_rate": 7.941176470588236e-06,
+      "loss": 0.0049,
+      "step": 2925
+    },
+    {
+      "epoch": 7.39,
+      "learning_rate": 7.933503836317136e-06,
+      "loss": 0.0053,
+      "step": 2926
+    },
+    {
+      "epoch": 7.39,
+      "learning_rate": 7.925831202046036e-06,
+      "loss": 0.0037,
+      "step": 2927
+    },
+    {
+      "epoch": 7.39,
+      "learning_rate": 7.918158567774936e-06,
+      "loss": 0.0049,
+      "step": 2928
+    },
+    {
+      "epoch": 7.4,
+      "learning_rate": 7.910485933503838e-06,
+      "loss": 0.0048,
+      "step": 2929
+    },
+    {
+      "epoch": 7.4,
+      "learning_rate": 7.902813299232736e-06,
+      "loss": 0.0059,
+      "step": 2930
+    },
+    {
+      "epoch": 7.4,
+      "learning_rate": 7.895140664961637e-06,
+      "loss": 0.0037,
+      "step": 2931
+    },
+    {
+      "epoch": 7.4,
+      "learning_rate": 7.887468030690537e-06,
+      "loss": 0.0021,
+      "step": 2932
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 7.879795396419437e-06,
+      "loss": 0.0037,
+      "step": 2933
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 7.872122762148337e-06,
+      "loss": 0.0033,
+      "step": 2934
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 7.864450127877239e-06,
+      "loss": 0.0048,
+      "step": 2935
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 7.856777493606137e-06,
+      "loss": 0.0025,
+      "step": 2936
+    },
+    {
+      "epoch": 7.42,
+      "learning_rate": 7.849104859335039e-06,
+      "loss": 0.0025,
+      "step": 2937
+    },
+    {
+      "epoch": 7.42,
+      "learning_rate": 7.841432225063939e-06,
+      "loss": 0.0036,
+      "step": 2938
+    },
+    {
+      "epoch": 7.42,
+      "learning_rate": 7.833759590792839e-06,
+      "loss": 0.0025,
+      "step": 2939
+    },
+    {
+      "epoch": 7.42,
+      "learning_rate": 7.826086956521738e-06,
+      "loss": 1.0675,
+      "step": 2940
+    },
+    {
+      "epoch": 7.43,
+      "learning_rate": 7.81841432225064e-06,
+      "loss": 0.003,
+      "step": 2941
+    },
+    {
+      "epoch": 7.43,
+      "learning_rate": 7.810741687979538e-06,
+      "loss": 0.0032,
+      "step": 2942
+    },
+    {
+      "epoch": 7.43,
+      "learning_rate": 7.80306905370844e-06,
+      "loss": 0.0036,
+      "step": 2943
+    },
+    {
+      "epoch": 7.43,
+      "learning_rate": 7.79539641943734e-06,
+      "loss": 0.0046,
+      "step": 2944
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 7.787723785166241e-06,
+      "loss": 0.0036,
+      "step": 2945
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 7.780051150895141e-06,
+      "loss": 0.0036,
+      "step": 2946
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 7.772378516624041e-06,
+      "loss": 0.0025,
+      "step": 2947
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 7.764705882352943e-06,
+      "loss": 0.0046,
+      "step": 2948
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 7.757033248081841e-06,
+      "loss": 0.0025,
+      "step": 2949
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 7.749360613810743e-06,
+      "loss": 0.0031,
+      "step": 2950
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 7.741687979539643e-06,
+      "loss": 0.0024,
+      "step": 2951
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 7.734015345268543e-06,
+      "loss": 0.003,
+      "step": 2952
+    },
+    {
+      "epoch": 7.46,
+      "learning_rate": 7.726342710997442e-06,
+      "loss": 0.0024,
+      "step": 2953
+    },
+    {
+      "epoch": 7.46,
+      "learning_rate": 7.718670076726344e-06,
+      "loss": 0.0028,
+      "step": 2954
+    },
+    {
+      "epoch": 7.46,
+      "learning_rate": 7.710997442455242e-06,
+      "loss": 0.0046,
+      "step": 2955
+    },
+    {
+      "epoch": 7.46,
+      "learning_rate": 7.703324808184144e-06,
+      "loss": 0.0017,
+      "step": 2956
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 7.695652173913044e-06,
+      "loss": 1.2377,
+      "step": 2957
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 7.687979539641944e-06,
+      "loss": 0.0034,
+      "step": 2958
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 7.680306905370844e-06,
+      "loss": 2.3609,
+      "step": 2959
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 7.672634271099745e-06,
+      "loss": 0.0045,
+      "step": 2960
+    },
+    {
+      "epoch": 7.48,
+      "learning_rate": 7.664961636828643e-06,
+      "loss": 0.0036,
+      "step": 2961
+    },
+    {
+      "epoch": 7.48,
+      "learning_rate": 7.657289002557545e-06,
+      "loss": 0.0056,
+      "step": 2962
+    },
+    {
+      "epoch": 7.48,
+      "learning_rate": 7.649616368286445e-06,
+      "loss": 0.0024,
+      "step": 2963
+    },
+    {
+      "epoch": 7.48,
+      "learning_rate": 7.641943734015345e-06,
+      "loss": 0.0036,
+      "step": 2964
+    },
+    {
+      "epoch": 7.49,
+      "learning_rate": 7.634271099744245e-06,
+      "loss": 0.0034,
+      "step": 2965
+    },
+    {
+      "epoch": 7.49,
+      "learning_rate": 7.6265984654731465e-06,
+      "loss": 0.0034,
+      "step": 2966
+    },
+    {
+      "epoch": 7.49,
+      "learning_rate": 7.6189258312020455e-06,
+      "loss": 0.0034,
+      "step": 2967
+    },
+    {
+      "epoch": 7.49,
+      "learning_rate": 7.611253196930946e-06,
+      "loss": 0.0024,
+      "step": 2968
+    },
+    {
+      "epoch": 7.5,
+      "learning_rate": 7.603580562659847e-06,
+      "loss": 0.0034,
+      "step": 2969
+    },
+    {
+      "epoch": 7.5,
+      "learning_rate": 7.595907928388748e-06,
+      "loss": 0.0041,
+      "step": 2970
+    },
+    {
+      "epoch": 7.5,
+      "learning_rate": 7.588235294117647e-06,
+      "loss": 0.0057,
+      "step": 2971
+    },
+    {
+      "epoch": 7.51,
+      "learning_rate": 7.580562659846548e-06,
+      "loss": 1.2636,
+      "step": 2972
+    },
+    {
+      "epoch": 7.51,
+      "learning_rate": 7.5728900255754484e-06,
+      "loss": 0.0044,
+      "step": 2973
+    },
+    {
+      "epoch": 7.51,
+      "learning_rate": 7.5652173913043475e-06,
+      "loss": 0.0034,
+      "step": 2974
+    },
+    {
+      "epoch": 7.51,
+      "learning_rate": 7.557544757033248e-06,
+      "loss": 0.0035,
+      "step": 2975
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 7.549872122762149e-06,
+      "loss": 0.0034,
+      "step": 2976
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 7.542199488491048e-06,
+      "loss": 0.0045,
+      "step": 2977
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 7.534526854219949e-06,
+      "loss": 0.0024,
+      "step": 2978
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 7.52685421994885e-06,
+      "loss": 0.0038,
+      "step": 2979
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 7.519181585677749e-06,
+      "loss": 0.0034,
+      "step": 2980
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 7.5115089514066495e-06,
+      "loss": 0.0029,
+      "step": 2981
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 7.50383631713555e-06,
+      "loss": 0.0034,
+      "step": 2982
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 7.496163682864451e-06,
+      "loss": 0.0034,
+      "step": 2983
+    },
+    {
+      "epoch": 7.54,
+      "learning_rate": 7.488491048593351e-06,
+      "loss": 0.0056,
+      "step": 2984
+    },
+    {
+      "epoch": 7.54,
+      "learning_rate": 7.480818414322251e-06,
+      "loss": 0.0035,
+      "step": 2985
+    },
+    {
+      "epoch": 7.54,
+      "learning_rate": 7.473145780051152e-06,
+      "loss": 0.0045,
+      "step": 2986
+    },
+    {
+      "epoch": 7.54,
+      "learning_rate": 7.4654731457800515e-06,
+      "loss": 0.0034,
+      "step": 2987
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 7.4578005115089514e-06,
+      "loss": 1.4972,
+      "step": 2988
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 7.450127877237852e-06,
+      "loss": 0.0034,
+      "step": 2989
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 7.442455242966752e-06,
+      "loss": 0.0045,
+      "step": 2990
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 7.434782608695653e-06,
+      "loss": 0.0026,
+      "step": 2991
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 7.427109974424553e-06,
+      "loss": 0.0051,
+      "step": 2992
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 7.419437340153453e-06,
+      "loss": 0.0028,
+      "step": 2993
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 7.4117647058823535e-06,
+      "loss": 0.0023,
+      "step": 2994
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 7.404092071611253e-06,
+      "loss": 0.0034,
+      "step": 2995
+    },
+    {
+      "epoch": 7.57,
+      "learning_rate": 7.396419437340153e-06,
+      "loss": 0.0023,
+      "step": 2996
+    },
+    {
+      "epoch": 7.57,
+      "learning_rate": 7.388746803069054e-06,
+      "loss": 0.0045,
+      "step": 2997
+    },
+    {
+      "epoch": 7.57,
+      "learning_rate": 7.381074168797954e-06,
+      "loss": 0.0224,
+      "step": 2998
+    },
+    {
+      "epoch": 7.57,
+      "learning_rate": 7.373401534526854e-06,
+      "loss": 1.444,
+      "step": 2999
+    },
+    {
+      "epoch": 7.58,
+      "learning_rate": 7.365728900255755e-06,
+      "loss": 0.0016,
+      "step": 3000
+    },
+    {
+      "epoch": 7.58,
+      "learning_rate": 7.358056265984655e-06,
+      "loss": 0.0034,
+      "step": 3001
+    },
+    {
+      "epoch": 7.58,
+      "learning_rate": 7.3503836317135545e-06,
+      "loss": 0.0051,
+      "step": 3002
+    },
+    {
+      "epoch": 7.58,
+      "learning_rate": 7.342710997442455e-06,
+      "loss": 0.0035,
+      "step": 3003
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 7.335038363171355e-06,
+      "loss": 0.0038,
+      "step": 3004
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 7.327365728900256e-06,
+      "loss": 0.0034,
+      "step": 3005
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 7.319693094629156e-06,
+      "loss": 0.0034,
+      "step": 3006
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 7.312020460358056e-06,
+      "loss": 0.0034,
+      "step": 3007
+    },
+    {
+      "epoch": 7.6,
+      "learning_rate": 7.304347826086957e-06,
+      "loss": 0.0037,
+      "step": 3008
+    },
+    {
+      "epoch": 7.6,
+      "learning_rate": 7.296675191815857e-06,
+      "loss": 0.0034,
+      "step": 3009
+    },
+    {
+      "epoch": 7.6,
+      "learning_rate": 7.289002557544757e-06,
+      "loss": 1.2445,
+      "step": 3010
+    },
+    {
+      "epoch": 7.6,
+      "learning_rate": 7.281329923273658e-06,
+      "loss": 0.0034,
+      "step": 3011
+    },
+    {
+      "epoch": 7.61,
+      "learning_rate": 7.273657289002558e-06,
+      "loss": 0.0034,
+      "step": 3012
+    },
+    {
+      "epoch": 7.61,
+      "learning_rate": 7.265984654731458e-06,
+      "loss": 0.0019,
+      "step": 3013
+    },
+    {
+      "epoch": 7.61,
+      "learning_rate": 7.258312020460359e-06,
+      "loss": 1.5106,
+      "step": 3014
+    },
+    {
+      "epoch": 7.61,
+      "learning_rate": 7.2506393861892585e-06,
+      "loss": 1.5062,
+      "step": 3015
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 7.242966751918159e-06,
+      "loss": 0.0035,
+      "step": 3016
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 7.235294117647059e-06,
+      "loss": 0.0027,
+      "step": 3017
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 7.227621483375959e-06,
+      "loss": 0.0025,
+      "step": 3018
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 7.21994884910486e-06,
+      "loss": 0.0035,
+      "step": 3019
+    },
+    {
+      "epoch": 7.63,
+      "learning_rate": 7.21227621483376e-06,
+      "loss": 0.0046,
+      "step": 3020
+    },
+    {
+      "epoch": 7.63,
+      "learning_rate": 7.20460358056266e-06,
+      "loss": 1.2443,
+      "step": 3021
+    },
+    {
+      "epoch": 7.63,
+      "learning_rate": 7.1969309462915605e-06,
+      "loss": 0.004,
+      "step": 3022
+    },
+    {
+      "epoch": 7.63,
+      "learning_rate": 7.18925831202046e-06,
+      "loss": 0.0028,
+      "step": 3023
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 7.18158567774936e-06,
+      "loss": 0.0059,
+      "step": 3024
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 7.173913043478261e-06,
+      "loss": 0.0062,
+      "step": 3025
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 7.166240409207161e-06,
+      "loss": 0.0017,
+      "step": 3026
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 7.158567774936061e-06,
+      "loss": 1.2106,
+      "step": 3027
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 7.150895140664962e-06,
+      "loss": 0.0049,
+      "step": 3028
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 7.143222506393862e-06,
+      "loss": 0.0022,
+      "step": 3029
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 7.135549872122762e-06,
+      "loss": 0.0038,
+      "step": 3030
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 7.127877237851662e-06,
+      "loss": 0.004,
+      "step": 3031
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 7.120204603580563e-06,
+      "loss": 0.0018,
+      "step": 3032
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 7.112531969309464e-06,
+      "loss": 0.0038,
+      "step": 3033
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 7.104859335038364e-06,
+      "loss": 0.0011,
+      "step": 3034
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 7.097186700767264e-06,
+      "loss": 0.0033,
+      "step": 3035
+    },
+    {
+      "epoch": 7.67,
+      "learning_rate": 7.089514066496164e-06,
+      "loss": 0.0052,
+      "step": 3036
+    },
+    {
+      "epoch": 7.67,
+      "learning_rate": 7.081841432225064e-06,
+      "loss": 1.4336,
+      "step": 3037
+    },
+    {
+      "epoch": 7.67,
+      "learning_rate": 7.074168797953964e-06,
+      "loss": 0.0017,
+      "step": 3038
+    },
+    {
+      "epoch": 7.67,
+      "learning_rate": 7.066496163682865e-06,
+      "loss": 0.0039,
+      "step": 3039
+    },
+    {
+      "epoch": 7.68,
+      "learning_rate": 7.058823529411765e-06,
+      "loss": 1.1887,
+      "step": 3040
+    },
+    {
+      "epoch": 7.68,
+      "learning_rate": 7.051150895140665e-06,
+      "loss": 0.0031,
+      "step": 3041
+    },
+    {
+      "epoch": 7.68,
+      "learning_rate": 7.043478260869566e-06,
+      "loss": 0.005,
+      "step": 3042
+    },
+    {
+      "epoch": 7.68,
+      "learning_rate": 7.0358056265984655e-06,
+      "loss": 0.0036,
+      "step": 3043
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 7.028132992327366e-06,
+      "loss": 0.0042,
+      "step": 3044
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 7.020460358056266e-06,
+      "loss": 0.0027,
+      "step": 3045
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 7.012787723785166e-06,
+      "loss": 1.1529,
+      "step": 3046
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 7.005115089514067e-06,
+      "loss": 0.0043,
+      "step": 3047
+    },
+    {
+      "epoch": 7.7,
+      "learning_rate": 6.997442455242967e-06,
+      "loss": 0.0064,
+      "step": 3048
+    },
+    {
+      "epoch": 7.7,
+      "learning_rate": 6.989769820971867e-06,
+      "loss": 0.0045,
+      "step": 3049
+    },
+    {
+      "epoch": 7.7,
+      "learning_rate": 6.9820971867007675e-06,
+      "loss": 0.0029,
+      "step": 3050
+    },
+    {
+      "epoch": 7.7,
+      "learning_rate": 6.974424552429667e-06,
+      "loss": 0.0044,
+      "step": 3051
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 6.966751918158567e-06,
+      "loss": 0.0012,
+      "step": 3052
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 6.959079283887468e-06,
+      "loss": 0.0018,
+      "step": 3053
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 6.951406649616368e-06,
+      "loss": 1.1802,
+      "step": 3054
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 6.943734015345268e-06,
+      "loss": 0.0048,
+      "step": 3055
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 6.936061381074169e-06,
+      "loss": 1.1027,
+      "step": 3056
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 6.9283887468030695e-06,
+      "loss": 0.0026,
+      "step": 3057
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 6.92071611253197e-06,
+      "loss": 1.24,
+      "step": 3058
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 6.91304347826087e-06,
+      "loss": 1.6565,
+      "step": 3059
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 6.90537084398977e-06,
+      "loss": 0.0047,
+      "step": 3060
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 6.897698209718671e-06,
+      "loss": 1.3994,
+      "step": 3061
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 6.890025575447571e-06,
+      "loss": 1.3895,
+      "step": 3062
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 6.882352941176471e-06,
+      "loss": 0.0073,
+      "step": 3063
+    },
+    {
+      "epoch": 7.74,
+      "learning_rate": 6.8746803069053714e-06,
+      "loss": 0.0075,
+      "step": 3064
+    },
+    {
+      "epoch": 7.74,
+      "learning_rate": 6.867007672634271e-06,
+      "loss": 0.0192,
+      "step": 3065
+    },
+    {
+      "epoch": 7.74,
+      "learning_rate": 6.859335038363171e-06,
+      "loss": 0.005,
+      "step": 3066
+    },
+    {
+      "epoch": 7.74,
+      "learning_rate": 6.851662404092072e-06,
+      "loss": 0.0079,
+      "step": 3067
+    },
+    {
+      "epoch": 7.75,
+      "learning_rate": 6.843989769820972e-06,
+      "loss": 0.005,
+      "step": 3068
+    },
+    {
+      "epoch": 7.75,
+      "learning_rate": 6.836317135549873e-06,
+      "loss": 0.0051,
+      "step": 3069
+    },
+    {
+      "epoch": 7.75,
+      "learning_rate": 6.828644501278773e-06,
+      "loss": 1.4104,
+      "step": 3070
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 6.8209718670076725e-06,
+      "loss": 0.0014,
+      "step": 3071
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 6.813299232736573e-06,
+      "loss": 0.0072,
+      "step": 3072
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 6.805626598465473e-06,
+      "loss": 0.006,
+      "step": 3073
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 6.797953964194373e-06,
+      "loss": 0.0046,
+      "step": 3074
+    },
+    {
+      "epoch": 7.77,
+      "learning_rate": 6.790281329923274e-06,
+      "loss": 0.0092,
+      "step": 3075
+    },
+    {
+      "epoch": 7.77,
+      "learning_rate": 6.782608695652174e-06,
+      "loss": 0.003,
+      "step": 3076
+    },
+    {
+      "epoch": 7.77,
+      "learning_rate": 6.774936061381074e-06,
+      "loss": 0.0071,
+      "step": 3077
+    },
+    {
+      "epoch": 7.77,
+      "learning_rate": 6.7672634271099745e-06,
+      "loss": 0.0089,
+      "step": 3078
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 6.7595907928388744e-06,
+      "loss": 0.0051,
+      "step": 3079
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 6.751918158567774e-06,
+      "loss": 0.0074,
+      "step": 3080
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 6.744245524296675e-06,
+      "loss": 0.0082,
+      "step": 3081
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 6.736572890025576e-06,
+      "loss": 0.0047,
+      "step": 3082
+    },
+    {
+      "epoch": 7.79,
+      "learning_rate": 6.728900255754477e-06,
+      "loss": 0.0071,
+      "step": 3083
+    },
+    {
+      "epoch": 7.79,
+      "learning_rate": 6.7212276214833766e-06,
+      "loss": 0.0029,
+      "step": 3084
+    },
+    {
+      "epoch": 7.79,
+      "learning_rate": 6.7135549872122765e-06,
+      "loss": 0.0071,
+      "step": 3085
+    },
+    {
+      "epoch": 7.79,
+      "learning_rate": 6.705882352941177e-06,
+      "loss": 0.0062,
+      "step": 3086
+    },
+    {
+      "epoch": 7.8,
+      "learning_rate": 6.698209718670077e-06,
+      "loss": 0.0046,
+      "step": 3087
+    },
+    {
+      "epoch": 7.8,
+      "learning_rate": 6.690537084398977e-06,
+      "loss": 0.0028,
+      "step": 3088
+    },
+    {
+      "epoch": 7.8,
+      "learning_rate": 6.682864450127878e-06,
+      "loss": 0.0049,
+      "step": 3089
+    },
+    {
+      "epoch": 7.8,
+      "learning_rate": 6.675191815856778e-06,
+      "loss": 0.0078,
+      "step": 3090
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 6.667519181585678e-06,
+      "loss": 0.0388,
+      "step": 3091
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 6.6598465473145784e-06,
+      "loss": 0.007,
+      "step": 3092
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 6.652173913043478e-06,
+      "loss": 0.003,
+      "step": 3093
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 6.644501278772378e-06,
+      "loss": 0.0028,
+      "step": 3094
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 6.636828644501279e-06,
+      "loss": 1.424,
+      "step": 3095
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 6.629156010230179e-06,
+      "loss": 0.016,
+      "step": 3096
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 6.62148337595908e-06,
+      "loss": 0.0046,
+      "step": 3097
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 6.61381074168798e-06,
+      "loss": 0.0048,
+      "step": 3098
+    },
+    {
+      "epoch": 7.83,
+      "learning_rate": 6.6061381074168796e-06,
+      "loss": 0.0066,
+      "step": 3099
+    },
+    {
+      "epoch": 7.83,
+      "learning_rate": 6.59846547314578e-06,
+      "loss": 0.0047,
+      "step": 3100
+    },
+    {
+      "epoch": 7.83,
+      "learning_rate": 6.59079283887468e-06,
+      "loss": 0.0029,
+      "step": 3101
+    },
+    {
+      "epoch": 7.83,
+      "learning_rate": 6.58312020460358e-06,
+      "loss": 0.0042,
+      "step": 3102
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 6.575447570332481e-06,
+      "loss": 0.003,
+      "step": 3103
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 6.567774936061381e-06,
+      "loss": 0.0028,
+      "step": 3104
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 6.560102301790281e-06,
+      "loss": 0.0042,
+      "step": 3105
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 6.5524296675191815e-06,
+      "loss": 0.0032,
+      "step": 3106
+    },
+    {
+      "epoch": 7.85,
+      "learning_rate": 6.544757033248082e-06,
+      "loss": 0.0041,
+      "step": 3107
+    },
+    {
+      "epoch": 7.85,
+      "learning_rate": 6.537084398976983e-06,
+      "loss": 0.005,
+      "step": 3108
+    },
+    {
+      "epoch": 7.85,
+      "learning_rate": 6.529411764705883e-06,
+      "loss": 0.0035,
+      "step": 3109
+    },
+    {
+      "epoch": 7.85,
+      "learning_rate": 6.521739130434783e-06,
+      "loss": 1.2368,
+      "step": 3110
+    },
+    {
+      "epoch": 7.86,
+      "learning_rate": 6.514066496163684e-06,
+      "loss": 0.0041,
+      "step": 3111
+    },
+    {
+      "epoch": 7.86,
+      "learning_rate": 6.5063938618925836e-06,
+      "loss": 0.0054,
+      "step": 3112
+    },
+    {
+      "epoch": 7.86,
+      "learning_rate": 6.4987212276214835e-06,
+      "loss": 1.4303,
+      "step": 3113
+    },
+    {
+      "epoch": 7.86,
+      "learning_rate": 6.491048593350384e-06,
+      "loss": 0.0041,
+      "step": 3114
+    },
+    {
+      "epoch": 7.87,
+      "learning_rate": 6.483375959079284e-06,
+      "loss": 0.0042,
+      "step": 3115
+    },
+    {
+      "epoch": 7.87,
+      "learning_rate": 6.475703324808184e-06,
+      "loss": 0.0034,
+      "step": 3116
+    },
+    {
+      "epoch": 7.87,
+      "learning_rate": 6.468030690537085e-06,
+      "loss": 0.0046,
+      "step": 3117
+    },
+    {
+      "epoch": 7.87,
+      "learning_rate": 6.460358056265985e-06,
+      "loss": 0.0037,
+      "step": 3118
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 6.452685421994885e-06,
+      "loss": 0.0027,
+      "step": 3119
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 6.4450127877237854e-06,
+      "loss": 0.0045,
+      "step": 3120
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 6.437340153452685e-06,
+      "loss": 0.0056,
+      "step": 3121
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 6.429667519181586e-06,
+      "loss": 0.0166,
+      "step": 3122
+    },
+    {
+      "epoch": 7.89,
+      "learning_rate": 6.421994884910486e-06,
+      "loss": 0.0031,
+      "step": 3123
+    },
+    {
+      "epoch": 7.89,
+      "learning_rate": 6.414322250639386e-06,
+      "loss": 1.2067,
+      "step": 3124
+    },
+    {
+      "epoch": 7.89,
+      "learning_rate": 6.406649616368287e-06,
+      "loss": 0.0932,
+      "step": 3125
+    },
+    {
+      "epoch": 7.89,
+      "learning_rate": 6.398976982097187e-06,
+      "loss": 0.0028,
+      "step": 3126
+    },
+    {
+      "epoch": 7.9,
+      "learning_rate": 6.3913043478260866e-06,
+      "loss": 0.0056,
+      "step": 3127
+    },
+    {
+      "epoch": 7.9,
+      "learning_rate": 6.383631713554987e-06,
+      "loss": 0.0022,
+      "step": 3128
+    },
+    {
+      "epoch": 7.9,
+      "learning_rate": 6.375959079283887e-06,
+      "loss": 0.0042,
+      "step": 3129
+    },
+    {
+      "epoch": 7.9,
+      "learning_rate": 6.368286445012787e-06,
+      "loss": 0.0056,
+      "step": 3130
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 6.360613810741688e-06,
+      "loss": 0.0042,
+      "step": 3131
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 6.352941176470589e-06,
+      "loss": 1.4088,
+      "step": 3132
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 6.345268542199489e-06,
+      "loss": 0.0022,
+      "step": 3133
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 6.337595907928389e-06,
+      "loss": 0.0062,
+      "step": 3134
+    },
+    {
+      "epoch": 7.92,
+      "learning_rate": 6.329923273657289e-06,
+      "loss": 0.0042,
+      "step": 3135
+    },
+    {
+      "epoch": 7.92,
+      "learning_rate": 6.32225063938619e-06,
+      "loss": 0.0048,
+      "step": 3136
+    },
+    {
+      "epoch": 7.92,
+      "learning_rate": 6.31457800511509e-06,
+      "loss": 1.4579,
+      "step": 3137
+    },
+    {
+      "epoch": 7.92,
+      "learning_rate": 6.30690537084399e-06,
+      "loss": 0.0043,
+      "step": 3138
+    },
+    {
+      "epoch": 7.93,
+      "learning_rate": 6.299232736572891e-06,
+      "loss": 0.0056,
+      "step": 3139
+    },
+    {
+      "epoch": 7.93,
+      "learning_rate": 6.291560102301791e-06,
+      "loss": 0.0043,
+      "step": 3140
+    },
+    {
+      "epoch": 7.93,
+      "learning_rate": 6.2838874680306905e-06,
+      "loss": 0.0057,
+      "step": 3141
+    },
+    {
+      "epoch": 7.93,
+      "learning_rate": 6.276214833759591e-06,
+      "loss": 0.0043,
+      "step": 3142
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 6.268542199488491e-06,
+      "loss": 0.0703,
+      "step": 3143
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 6.260869565217391e-06,
+      "loss": 0.0023,
+      "step": 3144
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 6.253196930946292e-06,
+      "loss": 0.0081,
+      "step": 3145
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 6.245524296675192e-06,
+      "loss": 0.0046,
+      "step": 3146
+    },
+    {
+      "epoch": 7.95,
+      "learning_rate": 6.237851662404092e-06,
+      "loss": 0.0047,
+      "step": 3147
+    },
+    {
+      "epoch": 7.95,
+      "learning_rate": 6.2301790281329925e-06,
+      "loss": 1.3925,
+      "step": 3148
+    },
+    {
+      "epoch": 7.95,
+      "learning_rate": 6.222506393861892e-06,
+      "loss": 0.0077,
+      "step": 3149
+    },
+    {
+      "epoch": 7.95,
+      "learning_rate": 6.214833759590793e-06,
+      "loss": 0.0051,
+      "step": 3150
+    },
+    {
+      "epoch": 7.96,
+      "learning_rate": 6.207161125319693e-06,
+      "loss": 0.0052,
+      "step": 3151
+    },
+    {
+      "epoch": 7.96,
+      "learning_rate": 6.199488491048593e-06,
+      "loss": 0.0037,
+      "step": 3152
+    },
+    {
+      "epoch": 7.96,
+      "learning_rate": 6.191815856777494e-06,
+      "loss": 0.007,
+      "step": 3153
+    },
+    {
+      "epoch": 7.96,
+      "learning_rate": 6.184143222506394e-06,
+      "loss": 0.0051,
+      "step": 3154
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 6.176470588235294e-06,
+      "loss": 1.4096,
+      "step": 3155
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 6.168797953964195e-06,
+      "loss": 0.0043,
+      "step": 3156
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 6.161125319693095e-06,
+      "loss": 0.0029,
+      "step": 3157
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 6.153452685421995e-06,
+      "loss": 0.0089,
+      "step": 3158
+    },
+    {
+      "epoch": 7.98,
+      "learning_rate": 6.145780051150896e-06,
+      "loss": 0.0043,
+      "step": 3159
+    },
+    {
+      "epoch": 7.98,
+      "learning_rate": 6.138107416879796e-06,
+      "loss": 0.009,
+      "step": 3160
+    },
+    {
+      "epoch": 7.98,
+      "learning_rate": 6.1304347826086965e-06,
+      "loss": 0.0045,
+      "step": 3161
+    },
+    {
+      "epoch": 7.98,
+      "learning_rate": 6.122762148337596e-06,
+      "loss": 0.0059,
+      "step": 3162
+    },
+    {
+      "epoch": 7.99,
+      "learning_rate": 6.115089514066496e-06,
+      "loss": 0.0049,
+      "step": 3163
+    },
+    {
+      "epoch": 7.99,
+      "learning_rate": 6.107416879795397e-06,
+      "loss": 0.0059,
+      "step": 3164
+    },
+    {
+      "epoch": 7.99,
+      "learning_rate": 6.099744245524297e-06,
+      "loss": 0.0033,
+      "step": 3165
+    },
+    {
+      "epoch": 7.99,
+      "learning_rate": 6.092071611253197e-06,
+      "loss": 0.0056,
+      "step": 3166
+    },
+    {
+      "epoch": 8.0,
+      "learning_rate": 6.084398976982098e-06,
+      "loss": 0.0053,
+      "step": 3167
+    },
+    {
+      "epoch": 8.0,
+      "learning_rate": 6.076726342710998e-06,
+      "loss": 0.0046,
+      "step": 3168
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 0.9242424242424242,
+      "eval_loss": 0.3901217579841614,
+      "eval_runtime": 57.8076,
+      "eval_samples_per_second": 6.85,
+      "eval_steps_per_second": 1.713,
+      "step": 3168
+    },
+    {
+      "epoch": 8.0,
+      "learning_rate": 6.0690537084398975e-06,
+      "loss": 0.0068,
+      "step": 3169
+    },
+    {
+      "epoch": 8.01,
+      "learning_rate": 6.061381074168798e-06,
+      "loss": 0.0034,
+      "step": 3170
+    },
+    {
+      "epoch": 8.01,
+      "learning_rate": 6.053708439897698e-06,
+      "loss": 0.0057,
+      "step": 3171
+    },
+    {
+      "epoch": 8.01,
+      "learning_rate": 6.046035805626598e-06,
+      "loss": 0.0055,
+      "step": 3172
+    },
+    {
+      "epoch": 8.01,
+      "learning_rate": 6.038363171355499e-06,
+      "loss": 0.0042,
+      "step": 3173
+    },
+    {
+      "epoch": 8.02,
+      "learning_rate": 6.030690537084399e-06,
+      "loss": 0.0054,
+      "step": 3174
+    },
+    {
+      "epoch": 8.02,
+      "learning_rate": 6.0230179028132996e-06,
+      "loss": 0.0029,
+      "step": 3175
+    },
+    {
+      "epoch": 8.02,
+      "learning_rate": 6.0153452685421995e-06,
+      "loss": 0.0029,
+      "step": 3176
+    },
+    {
+      "epoch": 8.02,
+      "learning_rate": 6.007672634271099e-06,
+      "loss": 0.0041,
+      "step": 3177
+    },
+    {
+      "epoch": 8.03,
+      "learning_rate": 6e-06,
+      "loss": 0.0054,
+      "step": 3178
+    },
+    {
+      "epoch": 8.03,
+      "learning_rate": 5.9923273657289e-06,
+      "loss": 0.0034,
+      "step": 3179
+    },
+    {
+      "epoch": 8.03,
+      "learning_rate": 5.9846547314578e-06,
+      "loss": 0.0218,
+      "step": 3180
+    },
+    {
+      "epoch": 8.03,
+      "learning_rate": 5.976982097186702e-06,
+      "loss": 0.0044,
+      "step": 3181
+    },
+    {
+      "epoch": 8.04,
+      "learning_rate": 5.9693094629156015e-06,
+      "loss": 0.0064,
+      "step": 3182
+    },
+    {
+      "epoch": 8.04,
+      "learning_rate": 5.9616368286445014e-06,
+      "loss": 0.0051,
+      "step": 3183
+    },
+    {
+      "epoch": 8.04,
+      "learning_rate": 5.953964194373402e-06,
+      "loss": 0.0039,
+      "step": 3184
+    },
+    {
+      "epoch": 8.04,
+      "learning_rate": 5.946291560102302e-06,
+      "loss": 0.0028,
+      "step": 3185
+    },
+    {
+      "epoch": 8.05,
+      "learning_rate": 5.938618925831202e-06,
+      "loss": 0.0043,
+      "step": 3186
+    },
+    {
+      "epoch": 8.05,
+      "learning_rate": 5.930946291560103e-06,
+      "loss": 0.0027,
+      "step": 3187
+    },
+    {
+      "epoch": 8.05,
+      "learning_rate": 5.923273657289003e-06,
+      "loss": 0.0044,
+      "step": 3188
+    },
+    {
+      "epoch": 8.05,
+      "learning_rate": 5.9156010230179035e-06,
+      "loss": 0.0063,
+      "step": 3189
+    },
+    {
+      "epoch": 8.06,
+      "learning_rate": 5.907928388746803e-06,
+      "loss": 0.0054,
+      "step": 3190
+    },
+    {
+      "epoch": 8.06,
+      "learning_rate": 5.900255754475703e-06,
+      "loss": 0.0047,
+      "step": 3191
+    },
+    {
+      "epoch": 8.06,
+      "learning_rate": 5.892583120204604e-06,
+      "loss": 0.0048,
+      "step": 3192
+    },
+    {
+      "epoch": 8.06,
+      "learning_rate": 5.884910485933504e-06,
+      "loss": 1.4901,
+      "step": 3193
+    },
+    {
+      "epoch": 8.07,
+      "learning_rate": 5.877237851662404e-06,
+      "loss": 0.0022,
+      "step": 3194
+    },
+    {
+      "epoch": 8.07,
+      "learning_rate": 5.869565217391305e-06,
+      "loss": 1.2931,
+      "step": 3195
+    },
+    {
+      "epoch": 8.07,
+      "learning_rate": 5.861892583120205e-06,
+      "loss": 0.0054,
+      "step": 3196
+    },
+    {
+      "epoch": 8.07,
+      "learning_rate": 5.8542199488491045e-06,
+      "loss": 0.0051,
+      "step": 3197
+    },
+    {
+      "epoch": 8.08,
+      "learning_rate": 5.846547314578005e-06,
+      "loss": 0.0028,
+      "step": 3198
+    },
+    {
+      "epoch": 8.08,
+      "learning_rate": 5.838874680306905e-06,
+      "loss": 0.0042,
+      "step": 3199
+    },
+    {
+      "epoch": 8.08,
+      "learning_rate": 5.831202046035805e-06,
+      "loss": 0.0043,
+      "step": 3200
+    },
+    {
+      "epoch": 8.08,
+      "learning_rate": 5.823529411764706e-06,
+      "loss": 0.0045,
+      "step": 3201
+    },
+    {
+      "epoch": 8.09,
+      "learning_rate": 5.815856777493606e-06,
+      "loss": 0.0045,
+      "step": 3202
+    },
+    {
+      "epoch": 8.09,
+      "learning_rate": 5.8081841432225066e-06,
+      "loss": 1.5693,
+      "step": 3203
+    },
+    {
+      "epoch": 8.09,
+      "learning_rate": 5.8005115089514065e-06,
+      "loss": 0.0051,
+      "step": 3204
+    },
+    {
+      "epoch": 8.09,
+      "learning_rate": 5.792838874680306e-06,
+      "loss": 0.0051,
+      "step": 3205
+    },
+    {
+      "epoch": 8.1,
+      "learning_rate": 5.785166240409208e-06,
+      "loss": 0.004,
+      "step": 3206
+    },
+    {
+      "epoch": 8.1,
+      "learning_rate": 5.777493606138108e-06,
+      "loss": 0.0045,
+      "step": 3207
+    },
+    {
+      "epoch": 8.1,
+      "learning_rate": 5.769820971867008e-06,
+      "loss": 0.0021,
+      "step": 3208
+    },
+    {
+      "epoch": 8.1,
+      "learning_rate": 5.762148337595909e-06,
+      "loss": 0.0054,
+      "step": 3209
+    },
+    {
+      "epoch": 8.11,
+      "learning_rate": 5.7544757033248085e-06,
+      "loss": 0.0057,
+      "step": 3210
+    },
+    {
+      "epoch": 8.11,
+      "learning_rate": 5.7468030690537084e-06,
+      "loss": 0.0044,
+      "step": 3211
+    },
+    {
+      "epoch": 8.11,
+      "learning_rate": 5.739130434782609e-06,
+      "loss": 0.005,
+      "step": 3212
+    },
+    {
+      "epoch": 8.11,
+      "learning_rate": 5.731457800511509e-06,
+      "loss": 0.0027,
+      "step": 3213
+    },
+    {
+      "epoch": 8.12,
+      "learning_rate": 5.72378516624041e-06,
+      "loss": 0.0027,
+      "step": 3214
+    },
+    {
+      "epoch": 8.12,
+      "learning_rate": 5.71611253196931e-06,
+      "loss": 0.005,
+      "step": 3215
+    },
+    {
+      "epoch": 8.12,
+      "learning_rate": 5.70843989769821e-06,
+      "loss": 0.0052,
+      "step": 3216
+    },
+    {
+      "epoch": 8.12,
+      "learning_rate": 5.7007672634271105e-06,
+      "loss": 0.004,
+      "step": 3217
+    },
+    {
+      "epoch": 8.13,
+      "learning_rate": 5.69309462915601e-06,
+      "loss": 0.0064,
+      "step": 3218
+    },
+    {
+      "epoch": 8.13,
+      "learning_rate": 5.68542199488491e-06,
+      "loss": 0.0038,
+      "step": 3219
+    },
+    {
+      "epoch": 8.13,
+      "learning_rate": 5.677749360613811e-06,
+      "loss": 0.0038,
+      "step": 3220
+    },
+    {
+      "epoch": 8.13,
+      "learning_rate": 5.670076726342711e-06,
+      "loss": 0.004,
+      "step": 3221
+    },
+    {
+      "epoch": 8.14,
+      "learning_rate": 5.662404092071611e-06,
+      "loss": 0.0054,
+      "step": 3222
+    },
+    {
+      "epoch": 8.14,
+      "learning_rate": 5.654731457800512e-06,
+      "loss": 1.4265,
+      "step": 3223
+    },
+    {
+      "epoch": 8.14,
+      "learning_rate": 5.647058823529412e-06,
+      "loss": 0.5258,
+      "step": 3224
+    },
+    {
+      "epoch": 8.14,
+      "learning_rate": 5.6393861892583115e-06,
+      "loss": 0.0028,
+      "step": 3225
+    },
+    {
+      "epoch": 8.15,
+      "learning_rate": 5.631713554987212e-06,
+      "loss": 0.0032,
+      "step": 3226
+    },
+    {
+      "epoch": 8.15,
+      "learning_rate": 5.624040920716112e-06,
+      "loss": 0.0048,
+      "step": 3227
+    },
+    {
+      "epoch": 8.15,
+      "learning_rate": 5.616368286445013e-06,
+      "loss": 0.0044,
+      "step": 3228
+    },
+    {
+      "epoch": 8.15,
+      "learning_rate": 5.608695652173913e-06,
+      "loss": 0.0042,
+      "step": 3229
+    },
+    {
+      "epoch": 8.16,
+      "learning_rate": 5.601023017902814e-06,
+      "loss": 0.0031,
+      "step": 3230
+    },
+    {
+      "epoch": 8.16,
+      "learning_rate": 5.593350383631714e-06,
+      "loss": 0.0054,
+      "step": 3231
+    },
+    {
+      "epoch": 8.16,
+      "learning_rate": 5.585677749360614e-06,
+      "loss": 0.0032,
+      "step": 3232
+    },
+    {
+      "epoch": 8.16,
+      "learning_rate": 5.578005115089514e-06,
+      "loss": 0.0026,
+      "step": 3233
+    },
+    {
+      "epoch": 8.17,
+      "learning_rate": 5.570332480818415e-06,
+      "loss": 0.0037,
+      "step": 3234
+    },
+    {
+      "epoch": 8.17,
+      "learning_rate": 5.562659846547315e-06,
+      "loss": 0.0053,
+      "step": 3235
+    },
+    {
+      "epoch": 8.17,
+      "learning_rate": 5.554987212276215e-06,
+      "loss": 0.0039,
+      "step": 3236
+    },
+    {
+      "epoch": 8.17,
+      "learning_rate": 5.547314578005116e-06,
+      "loss": 0.0048,
+      "step": 3237
+    },
+    {
+      "epoch": 8.18,
+      "learning_rate": 5.5396419437340155e-06,
+      "loss": 0.0048,
+      "step": 3238
+    },
+    {
+      "epoch": 8.18,
+      "learning_rate": 5.5319693094629155e-06,
+      "loss": 0.0047,
+      "step": 3239
+    },
+    {
+      "epoch": 8.18,
+      "learning_rate": 5.524296675191816e-06,
+      "loss": 0.0038,
+      "step": 3240
+    },
+    {
+      "epoch": 8.18,
+      "learning_rate": 5.516624040920716e-06,
+      "loss": 0.0037,
+      "step": 3241
+    },
+    {
+      "epoch": 8.19,
+      "learning_rate": 5.508951406649617e-06,
+      "loss": 0.0036,
+      "step": 3242
+    },
+    {
+      "epoch": 8.19,
+      "learning_rate": 5.501278772378517e-06,
+      "loss": 0.0036,
+      "step": 3243
+    },
+    {
+      "epoch": 8.19,
+      "learning_rate": 5.493606138107417e-06,
+      "loss": 0.0025,
+      "step": 3244
+    },
+    {
+      "epoch": 8.19,
+      "learning_rate": 5.4859335038363175e-06,
+      "loss": 0.0036,
+      "step": 3245
+    },
+    {
+      "epoch": 8.2,
+      "learning_rate": 5.478260869565217e-06,
+      "loss": 0.0033,
+      "step": 3246
+    },
+    {
+      "epoch": 8.2,
+      "learning_rate": 5.470588235294117e-06,
+      "loss": 0.0053,
+      "step": 3247
+    },
+    {
+      "epoch": 8.2,
+      "learning_rate": 5.462915601023018e-06,
+      "loss": 0.0046,
+      "step": 3248
+    },
+    {
+      "epoch": 8.2,
+      "learning_rate": 5.455242966751918e-06,
+      "loss": 0.0057,
+      "step": 3249
+    },
+    {
+      "epoch": 8.21,
+      "learning_rate": 5.447570332480818e-06,
+      "loss": 0.0049,
+      "step": 3250
+    },
+    {
+      "epoch": 8.21,
+      "learning_rate": 5.439897698209719e-06,
+      "loss": 0.0051,
+      "step": 3251
+    },
+    {
+      "epoch": 8.21,
+      "learning_rate": 5.432225063938619e-06,
+      "loss": 0.0045,
+      "step": 3252
+    },
+    {
+      "epoch": 8.21,
+      "learning_rate": 5.4245524296675185e-06,
+      "loss": 0.0025,
+      "step": 3253
+    },
+    {
+      "epoch": 8.22,
+      "learning_rate": 5.416879795396419e-06,
+      "loss": 0.0033,
+      "step": 3254
+    },
+    {
+      "epoch": 8.22,
+      "learning_rate": 5.40920716112532e-06,
+      "loss": 0.0042,
+      "step": 3255
+    },
+    {
+      "epoch": 8.22,
+      "learning_rate": 5.401534526854221e-06,
+      "loss": 0.0038,
+      "step": 3256
+    },
+    {
+      "epoch": 8.22,
+      "learning_rate": 5.393861892583121e-06,
+      "loss": 0.0036,
+      "step": 3257
+    },
+    {
+      "epoch": 8.23,
+      "learning_rate": 5.386189258312021e-06,
+      "loss": 0.0035,
+      "step": 3258
+    },
+    {
+      "epoch": 8.23,
+      "learning_rate": 5.3785166240409214e-06,
+      "loss": 0.029,
+      "step": 3259
+    },
+    {
+      "epoch": 8.23,
+      "learning_rate": 5.370843989769821e-06,
+      "loss": 0.0038,
+      "step": 3260
+    },
+    {
+      "epoch": 8.23,
+      "learning_rate": 5.363171355498721e-06,
+      "loss": 0.0024,
+      "step": 3261
+    },
+    {
+      "epoch": 8.24,
+      "learning_rate": 5.355498721227622e-06,
+      "loss": 0.0058,
+      "step": 3262
+    },
+    {
+      "epoch": 8.24,
+      "learning_rate": 5.347826086956522e-06,
+      "loss": 0.0034,
+      "step": 3263
+    },
+    {
+      "epoch": 8.24,
+      "learning_rate": 5.340153452685422e-06,
+      "loss": 0.0047,
+      "step": 3264
+    },
+    {
+      "epoch": 8.24,
+      "learning_rate": 5.332480818414323e-06,
+      "loss": 1.5064,
+      "step": 3265
+    },
+    {
+      "epoch": 8.25,
+      "learning_rate": 5.3248081841432225e-06,
+      "loss": 0.0041,
+      "step": 3266
+    },
+    {
+      "epoch": 8.25,
+      "learning_rate": 5.317135549872123e-06,
+      "loss": 0.0065,
+      "step": 3267
+    },
+    {
+      "epoch": 8.25,
+      "learning_rate": 5.309462915601023e-06,
+      "loss": 0.0034,
+      "step": 3268
+    },
+    {
+      "epoch": 8.26,
+      "learning_rate": 5.301790281329923e-06,
+      "loss": 0.0038,
+      "step": 3269
+    },
+    {
+      "epoch": 8.26,
+      "learning_rate": 5.294117647058824e-06,
+      "loss": 1.1837,
+      "step": 3270
+    },
+    {
+      "epoch": 8.26,
+      "learning_rate": 5.286445012787724e-06,
+      "loss": 0.0034,
+      "step": 3271
+    },
+    {
+      "epoch": 8.26,
+      "learning_rate": 5.278772378516624e-06,
+      "loss": 0.0025,
+      "step": 3272
+    },
+    {
+      "epoch": 8.27,
+      "learning_rate": 5.2710997442455245e-06,
+      "loss": 0.0049,
+      "step": 3273
+    },
+    {
+      "epoch": 8.27,
+      "learning_rate": 5.2634271099744244e-06,
+      "loss": 0.0028,
+      "step": 3274
+    },
+    {
+      "epoch": 8.27,
+      "learning_rate": 5.255754475703324e-06,
+      "loss": 0.0046,
+      "step": 3275
+    },
+    {
+      "epoch": 8.27,
+      "learning_rate": 5.248081841432225e-06,
+      "loss": 0.0052,
+      "step": 3276
+    },
+    {
+      "epoch": 8.28,
+      "learning_rate": 5.240409207161125e-06,
+      "loss": 0.0207,
+      "step": 3277
+    },
+    {
+      "epoch": 8.28,
+      "learning_rate": 5.232736572890025e-06,
+      "loss": 0.0027,
+      "step": 3278
+    },
+    {
+      "epoch": 8.28,
+      "learning_rate": 5.225063938618926e-06,
+      "loss": 0.0022,
+      "step": 3279
+    },
+    {
+      "epoch": 8.28,
+      "learning_rate": 5.2173913043478265e-06,
+      "loss": 0.0034,
+      "step": 3280
+    },
+    {
+      "epoch": 8.29,
+      "learning_rate": 5.209718670076727e-06,
+      "loss": 0.0035,
+      "step": 3281
+    },
+    {
+      "epoch": 8.29,
+      "learning_rate": 5.202046035805627e-06,
+      "loss": 0.0027,
+      "step": 3282
+    },
+    {
+      "epoch": 8.29,
+      "learning_rate": 5.194373401534527e-06,
+      "loss": 0.0031,
+      "step": 3283
+    },
+    {
+      "epoch": 8.29,
+      "learning_rate": 5.186700767263428e-06,
+      "loss": 0.0036,
+      "step": 3284
+    },
+    {
+      "epoch": 8.3,
+      "learning_rate": 5.179028132992328e-06,
+      "loss": 0.003,
+      "step": 3285
+    },
+    {
+      "epoch": 8.3,
+      "learning_rate": 5.171355498721228e-06,
+      "loss": 0.0024,
+      "step": 3286
+    },
+    {
+      "epoch": 8.3,
+      "learning_rate": 5.1636828644501284e-06,
+      "loss": 0.0043,
+      "step": 3287
+    },
+    {
+      "epoch": 8.3,
+      "learning_rate": 5.156010230179028e-06,
+      "loss": 0.0026,
+      "step": 3288
+    },
+    {
+      "epoch": 8.31,
+      "learning_rate": 5.148337595907928e-06,
+      "loss": 0.0024,
+      "step": 3289
+    },
+    {
+      "epoch": 8.31,
+      "learning_rate": 5.140664961636829e-06,
+      "loss": 0.0025,
+      "step": 3290
+    },
+    {
+      "epoch": 8.31,
+      "learning_rate": 5.132992327365729e-06,
+      "loss": 0.0034,
+      "step": 3291
+    },
+    {
+      "epoch": 8.31,
+      "learning_rate": 5.125319693094629e-06,
+      "loss": 0.0033,
+      "step": 3292
+    },
+    {
+      "epoch": 8.32,
+      "learning_rate": 5.11764705882353e-06,
+      "loss": 1.6,
+      "step": 3293
+    },
+    {
+      "epoch": 8.32,
+      "learning_rate": 5.1099744245524296e-06,
+      "loss": 0.0191,
+      "step": 3294
+    },
+    {
+      "epoch": 8.32,
+      "learning_rate": 5.10230179028133e-06,
+      "loss": 0.0219,
+      "step": 3295
+    },
+    {
+      "epoch": 8.32,
+      "learning_rate": 5.09462915601023e-06,
+      "loss": 0.0052,
+      "step": 3296
+    },
+    {
+      "epoch": 8.33,
+      "learning_rate": 5.08695652173913e-06,
+      "loss": 0.0043,
+      "step": 3297
+    },
+    {
+      "epoch": 8.33,
+      "learning_rate": 5.079283887468031e-06,
+      "loss": 0.025,
+      "step": 3298
+    },
+    {
+      "epoch": 8.33,
+      "learning_rate": 5.071611253196931e-06,
+      "loss": 0.0037,
+      "step": 3299
+    },
+    {
+      "epoch": 8.33,
+      "learning_rate": 5.063938618925831e-06,
+      "loss": 0.0025,
+      "step": 3300
+    },
+    {
+      "epoch": 8.34,
+      "learning_rate": 5.0562659846547315e-06,
+      "loss": 0.0035,
+      "step": 3301
+    },
+    {
+      "epoch": 8.34,
+      "learning_rate": 5.0485933503836314e-06,
+      "loss": 0.0024,
+      "step": 3302
+    },
+    {
+      "epoch": 8.34,
+      "learning_rate": 5.040920716112531e-06,
+      "loss": 0.0052,
+      "step": 3303
+    },
+    {
+      "epoch": 8.34,
+      "learning_rate": 5.033248081841432e-06,
+      "loss": 1.2789,
+      "step": 3304
+    },
+    {
+      "epoch": 8.35,
+      "learning_rate": 5.025575447570333e-06,
+      "loss": 0.003,
+      "step": 3305
+    },
+    {
+      "epoch": 8.35,
+      "learning_rate": 5.017902813299233e-06,
+      "loss": 0.004,
+      "step": 3306
+    },
+    {
+      "epoch": 8.35,
+      "learning_rate": 5.0102301790281336e-06,
+      "loss": 0.0114,
+      "step": 3307
+    },
+    {
+      "epoch": 8.35,
+      "learning_rate": 5.0025575447570335e-06,
+      "loss": 0.0033,
+      "step": 3308
+    },
+    {
+      "epoch": 8.36,
+      "learning_rate": 4.994884910485934e-06,
+      "loss": 0.0034,
+      "step": 3309
+    },
+    {
+      "epoch": 8.36,
+      "learning_rate": 4.987212276214834e-06,
+      "loss": 1.4496,
+      "step": 3310
+    },
+    {
+      "epoch": 8.36,
+      "learning_rate": 4.979539641943734e-06,
+      "loss": 0.0034,
+      "step": 3311
+    },
+    {
+      "epoch": 8.36,
+      "learning_rate": 4.971867007672635e-06,
+      "loss": 0.0038,
+      "step": 3312
+    },
+    {
+      "epoch": 8.37,
+      "learning_rate": 4.964194373401535e-06,
+      "loss": 0.0035,
+      "step": 3313
+    },
+    {
+      "epoch": 8.37,
+      "learning_rate": 4.956521739130435e-06,
+      "loss": 0.0057,
+      "step": 3314
+    },
+    {
+      "epoch": 8.37,
+      "learning_rate": 4.9488491048593355e-06,
+      "loss": 0.0041,
+      "step": 3315
+    },
+    {
+      "epoch": 8.37,
+      "learning_rate": 4.941176470588235e-06,
+      "loss": 0.0034,
+      "step": 3316
+    },
+    {
+      "epoch": 8.38,
+      "learning_rate": 4.933503836317135e-06,
+      "loss": 0.004,
+      "step": 3317
+    },
+    {
+      "epoch": 8.38,
+      "learning_rate": 4.925831202046036e-06,
+      "loss": 0.0026,
+      "step": 3318
+    },
+    {
+      "epoch": 8.38,
+      "learning_rate": 4.918158567774936e-06,
+      "loss": 0.0024,
+      "step": 3319
+    },
+    {
+      "epoch": 8.38,
+      "learning_rate": 4.910485933503837e-06,
+      "loss": 0.003,
+      "step": 3320
+    },
+    {
+      "epoch": 8.39,
+      "learning_rate": 4.902813299232737e-06,
+      "loss": 0.0026,
+      "step": 3321
+    },
+    {
+      "epoch": 8.39,
+      "learning_rate": 4.8951406649616366e-06,
+      "loss": 0.0036,
+      "step": 3322
+    },
+    {
+      "epoch": 8.39,
+      "learning_rate": 4.887468030690537e-06,
+      "loss": 1.2734,
+      "step": 3323
+    },
+    {
+      "epoch": 8.39,
+      "learning_rate": 4.879795396419437e-06,
+      "loss": 0.0029,
+      "step": 3324
+    },
+    {
+      "epoch": 8.4,
+      "learning_rate": 4.872122762148337e-06,
+      "loss": 0.0028,
+      "step": 3325
+    },
+    {
+      "epoch": 8.4,
+      "learning_rate": 4.864450127877238e-06,
+      "loss": 1.2688,
+      "step": 3326
+    },
+    {
+      "epoch": 8.4,
+      "learning_rate": 4.856777493606138e-06,
+      "loss": 0.0054,
+      "step": 3327
+    },
+    {
+      "epoch": 8.4,
+      "learning_rate": 4.849104859335038e-06,
+      "loss": 0.0026,
+      "step": 3328
+    },
+    {
+      "epoch": 8.41,
+      "learning_rate": 4.8414322250639385e-06,
+      "loss": 0.0026,
+      "step": 3329
+    },
+    {
+      "epoch": 8.41,
+      "learning_rate": 4.833759590792839e-06,
+      "loss": 0.0035,
+      "step": 3330
+    },
+    {
+      "epoch": 8.41,
+      "learning_rate": 4.826086956521739e-06,
+      "loss": 0.0036,
+      "step": 3331
+    },
+    {
+      "epoch": 8.41,
+      "learning_rate": 4.81841432225064e-06,
+      "loss": 0.0051,
+      "step": 3332
+    },
+    {
+      "epoch": 8.42,
+      "learning_rate": 4.81074168797954e-06,
+      "loss": 1.2525,
+      "step": 3333
+    },
+    {
+      "epoch": 8.42,
+      "learning_rate": 4.803069053708441e-06,
+      "loss": 0.0058,
+      "step": 3334
+    },
+    {
+      "epoch": 8.42,
+      "learning_rate": 4.795396419437341e-06,
+      "loss": 0.0025,
+      "step": 3335
+    },
+    {
+      "epoch": 8.42,
+      "learning_rate": 4.7877237851662405e-06,
+      "loss": 0.0323,
+      "step": 3336
+    },
+    {
+      "epoch": 8.43,
+      "learning_rate": 4.780051150895141e-06,
+      "loss": 0.0037,
+      "step": 3337
+    },
+    {
+      "epoch": 8.43,
+      "learning_rate": 4.772378516624041e-06,
+      "loss": 0.0046,
+      "step": 3338
+    },
+    {
+      "epoch": 8.43,
+      "learning_rate": 4.764705882352941e-06,
+      "loss": 0.003,
+      "step": 3339
+    },
+    {
+      "epoch": 8.43,
+      "learning_rate": 4.757033248081842e-06,
+      "loss": 0.0023,
+      "step": 3340
+    },
+    {
+      "epoch": 8.44,
+      "learning_rate": 4.749360613810742e-06,
+      "loss": 0.003,
+      "step": 3341
+    },
+    {
+      "epoch": 8.44,
+      "learning_rate": 4.741687979539642e-06,
+      "loss": 0.0025,
+      "step": 3342
+    },
+    {
+      "epoch": 8.44,
+      "learning_rate": 4.7340153452685425e-06,
+      "loss": 0.0026,
+      "step": 3343
+    },
+    {
+      "epoch": 8.44,
+      "learning_rate": 4.726342710997442e-06,
+      "loss": 1.2155,
+      "step": 3344
+    },
+    {
+      "epoch": 8.45,
+      "learning_rate": 4.718670076726342e-06,
+      "loss": 0.0049,
+      "step": 3345
+    },
+    {
+      "epoch": 8.45,
+      "learning_rate": 4.710997442455243e-06,
+      "loss": 1.2437,
+      "step": 3346
+    },
+    {
+      "epoch": 8.45,
+      "learning_rate": 4.703324808184143e-06,
+      "loss": 0.006,
+      "step": 3347
+    },
+    {
+      "epoch": 8.45,
+      "learning_rate": 4.695652173913044e-06,
+      "loss": 0.0036,
+      "step": 3348
+    },
+    {
+      "epoch": 8.46,
+      "learning_rate": 4.687979539641944e-06,
+      "loss": 0.0032,
+      "step": 3349
+    },
+    {
+      "epoch": 8.46,
+      "learning_rate": 4.680306905370844e-06,
+      "loss": 0.0048,
+      "step": 3350
+    },
+    {
+      "epoch": 8.46,
+      "learning_rate": 4.672634271099744e-06,
+      "loss": 0.0059,
+      "step": 3351
+    },
+    {
+      "epoch": 8.46,
+      "learning_rate": 4.664961636828644e-06,
+      "loss": 0.005,
+      "step": 3352
+    },
+    {
+      "epoch": 8.47,
+      "learning_rate": 4.657289002557544e-06,
+      "loss": 0.0048,
+      "step": 3353
+    },
+    {
+      "epoch": 8.47,
+      "learning_rate": 4.649616368286446e-06,
+      "loss": 0.0037,
+      "step": 3354
+    },
+    {
+      "epoch": 8.47,
+      "learning_rate": 4.641943734015346e-06,
+      "loss": 1.2189,
+      "step": 3355
+    },
+    {
+      "epoch": 8.47,
+      "learning_rate": 4.634271099744246e-06,
+      "loss": 0.0083,
+      "step": 3356
+    },
+    {
+      "epoch": 8.48,
+      "learning_rate": 4.626598465473146e-06,
+      "loss": 0.0062,
+      "step": 3357
+    },
+    {
+      "epoch": 8.48,
+      "learning_rate": 4.618925831202046e-06,
+      "loss": 1.6984,
+      "step": 3358
+    },
+    {
+      "epoch": 8.48,
+      "learning_rate": 4.611253196930947e-06,
+      "loss": 0.0385,
+      "step": 3359
+    },
+    {
+      "epoch": 8.48,
+      "learning_rate": 4.603580562659847e-06,
+      "loss": 0.0106,
+      "step": 3360
+    },
+    {
+      "epoch": 8.49,
+      "learning_rate": 4.595907928388747e-06,
+      "loss": 0.0041,
+      "step": 3361
+    },
+    {
+      "epoch": 8.49,
+      "learning_rate": 4.588235294117648e-06,
+      "loss": 0.0048,
+      "step": 3362
+    },
+    {
+      "epoch": 8.49,
+      "learning_rate": 4.580562659846548e-06,
+      "loss": 0.003,
+      "step": 3363
+    },
+    {
+      "epoch": 8.49,
+      "learning_rate": 4.5728900255754475e-06,
+      "loss": 0.0039,
+      "step": 3364
+    },
+    {
+      "epoch": 8.5,
+      "learning_rate": 4.565217391304348e-06,
+      "loss": 0.005,
+      "step": 3365
+    },
+    {
+      "epoch": 8.5,
+      "learning_rate": 4.557544757033248e-06,
+      "loss": 0.0037,
+      "step": 3366
+    },
+    {
+      "epoch": 8.5,
+      "learning_rate": 4.549872122762148e-06,
+      "loss": 0.0048,
+      "step": 3367
+    },
+    {
+      "epoch": 8.51,
+      "learning_rate": 4.542199488491049e-06,
+      "loss": 0.0054,
+      "step": 3368
+    },
+    {
+      "epoch": 8.51,
+      "learning_rate": 4.534526854219949e-06,
+      "loss": 0.0025,
+      "step": 3369
+    },
+    {
+      "epoch": 8.51,
+      "learning_rate": 4.526854219948849e-06,
+      "loss": 0.0093,
+      "step": 3370
+    },
+    {
+      "epoch": 8.51,
+      "learning_rate": 4.5191815856777495e-06,
+      "loss": 0.0042,
+      "step": 3371
+    },
+    {
+      "epoch": 8.52,
+      "learning_rate": 4.511508951406649e-06,
+      "loss": 0.0025,
+      "step": 3372
+    },
+    {
+      "epoch": 8.52,
+      "learning_rate": 4.50383631713555e-06,
+      "loss": 0.0048,
+      "step": 3373
+    },
+    {
+      "epoch": 8.52,
+      "learning_rate": 4.49616368286445e-06,
+      "loss": 0.0024,
+      "step": 3374
+    },
+    {
+      "epoch": 8.52,
+      "learning_rate": 4.48849104859335e-06,
+      "loss": 0.0048,
+      "step": 3375
+    },
+    {
+      "epoch": 8.53,
+      "learning_rate": 4.480818414322251e-06,
+      "loss": 0.0026,
+      "step": 3376
+    },
+    {
+      "epoch": 8.53,
+      "learning_rate": 4.473145780051151e-06,
+      "loss": 1.1559,
+      "step": 3377
+    },
+    {
+      "epoch": 8.53,
+      "learning_rate": 4.465473145780051e-06,
+      "loss": 0.0048,
+      "step": 3378
+    },
+    {
+      "epoch": 8.53,
+      "learning_rate": 4.457800511508952e-06,
+      "loss": 0.0021,
+      "step": 3379
+    },
+    {
+      "epoch": 8.54,
+      "learning_rate": 4.450127877237852e-06,
+      "loss": 0.0028,
+      "step": 3380
+    },
+    {
+      "epoch": 8.54,
+      "learning_rate": 4.442455242966752e-06,
+      "loss": 0.0798,
+      "step": 3381
+    },
+    {
+      "epoch": 8.54,
+      "learning_rate": 4.434782608695653e-06,
+      "loss": 0.0053,
+      "step": 3382
+    },
+    {
+      "epoch": 8.54,
+      "learning_rate": 4.427109974424553e-06,
+      "loss": 0.0024,
+      "step": 3383
+    },
+    {
+      "epoch": 8.55,
+      "learning_rate": 4.419437340153453e-06,
+      "loss": 0.0049,
+      "step": 3384
+    },
+    {
+      "epoch": 8.55,
+      "learning_rate": 4.411764705882353e-06,
+      "loss": 0.0036,
+      "step": 3385
+    },
+    {
+      "epoch": 8.55,
+      "learning_rate": 4.404092071611253e-06,
+      "loss": 1.5139,
+      "step": 3386
+    },
+    {
+      "epoch": 8.55,
+      "learning_rate": 4.396419437340154e-06,
+      "loss": 0.0035,
+      "step": 3387
+    },
+    {
+      "epoch": 8.56,
+      "learning_rate": 4.388746803069054e-06,
+      "loss": 0.0024,
+      "step": 3388
+    },
+    {
+      "epoch": 8.56,
+      "learning_rate": 4.381074168797954e-06,
+      "loss": 0.0051,
+      "step": 3389
+    },
+    {
+      "epoch": 8.56,
+      "learning_rate": 4.373401534526855e-06,
+      "loss": 0.0059,
+      "step": 3390
+    },
+    {
+      "epoch": 8.56,
+      "learning_rate": 4.365728900255755e-06,
+      "loss": 0.0059,
+      "step": 3391
+    },
+    {
+      "epoch": 8.57,
+      "learning_rate": 4.3580562659846545e-06,
+      "loss": 0.0025,
+      "step": 3392
+    },
+    {
+      "epoch": 8.57,
+      "learning_rate": 4.350383631713555e-06,
+      "loss": 1.2465,
+      "step": 3393
+    },
+    {
+      "epoch": 8.57,
+      "learning_rate": 4.342710997442455e-06,
+      "loss": 0.0038,
+      "step": 3394
+    },
+    {
+      "epoch": 8.57,
+      "learning_rate": 4.335038363171355e-06,
+      "loss": 0.0024,
+      "step": 3395
+    },
+    {
+      "epoch": 8.58,
+      "learning_rate": 4.327365728900256e-06,
+      "loss": 0.0052,
+      "step": 3396
+    },
+    {
+      "epoch": 8.58,
+      "learning_rate": 4.319693094629156e-06,
+      "loss": 0.0024,
+      "step": 3397
+    },
+    {
+      "epoch": 8.58,
+      "learning_rate": 4.312020460358056e-06,
+      "loss": 0.0051,
+      "step": 3398
+    },
+    {
+      "epoch": 8.58,
+      "learning_rate": 4.3043478260869565e-06,
+      "loss": 0.0035,
+      "step": 3399
+    },
+    {
+      "epoch": 8.59,
+      "learning_rate": 4.296675191815856e-06,
+      "loss": 0.0166,
+      "step": 3400
+    },
+    {
+      "epoch": 8.59,
+      "learning_rate": 4.289002557544757e-06,
+      "loss": 0.0045,
+      "step": 3401
+    },
+    {
+      "epoch": 8.59,
+      "learning_rate": 4.281329923273657e-06,
+      "loss": 0.0043,
+      "step": 3402
+    },
+    {
+      "epoch": 8.59,
+      "learning_rate": 4.273657289002557e-06,
+      "loss": 0.003,
+      "step": 3403
+    },
+    {
+      "epoch": 8.6,
+      "learning_rate": 4.265984654731459e-06,
+      "loss": 0.0071,
+      "step": 3404
+    },
+    {
+      "epoch": 8.6,
+      "learning_rate": 4.2583120204603585e-06,
+      "loss": 0.0036,
+      "step": 3405
+    },
+    {
+      "epoch": 8.6,
+      "learning_rate": 4.2506393861892584e-06,
+      "loss": 0.0049,
+      "step": 3406
+    },
+    {
+      "epoch": 8.6,
+      "learning_rate": 4.242966751918159e-06,
+      "loss": 0.0039,
+      "step": 3407
+    },
+    {
+      "epoch": 8.61,
+      "learning_rate": 4.235294117647059e-06,
+      "loss": 0.0036,
+      "step": 3408
+    },
+    {
+      "epoch": 8.61,
+      "learning_rate": 4.227621483375959e-06,
+      "loss": 0.0026,
+      "step": 3409
+    },
+    {
+      "epoch": 8.61,
+      "learning_rate": 4.21994884910486e-06,
+      "loss": 0.0028,
+      "step": 3410
+    },
+    {
+      "epoch": 8.61,
+      "learning_rate": 4.21227621483376e-06,
+      "loss": 0.0024,
+      "step": 3411
+    },
+    {
+      "epoch": 8.62,
+      "learning_rate": 4.2046035805626605e-06,
+      "loss": 0.0044,
+      "step": 3412
+    },
+    {
+      "epoch": 8.62,
+      "learning_rate": 4.19693094629156e-06,
+      "loss": 0.0025,
+      "step": 3413
+    },
+    {
+      "epoch": 8.62,
+      "learning_rate": 4.18925831202046e-06,
+      "loss": 0.0029,
+      "step": 3414
+    },
+    {
+      "epoch": 8.62,
+      "learning_rate": 4.181585677749361e-06,
+      "loss": 0.0019,
+      "step": 3415
+    },
+    {
+      "epoch": 8.63,
+      "learning_rate": 4.173913043478261e-06,
+      "loss": 0.0025,
+      "step": 3416
+    },
+    {
+      "epoch": 8.63,
+      "learning_rate": 4.166240409207161e-06,
+      "loss": 0.0043,
+      "step": 3417
+    },
+    {
+      "epoch": 8.63,
+      "learning_rate": 4.158567774936062e-06,
+      "loss": 0.0037,
+      "step": 3418
+    },
+    {
+      "epoch": 8.63,
+      "learning_rate": 4.150895140664962e-06,
+      "loss": 0.0024,
+      "step": 3419
+    },
+    {
+      "epoch": 8.64,
+      "learning_rate": 4.1432225063938615e-06,
+      "loss": 0.0024,
+      "step": 3420
+    },
+    {
+      "epoch": 8.64,
+      "learning_rate": 4.135549872122762e-06,
+      "loss": 0.0023,
+      "step": 3421
+    },
+    {
+      "epoch": 8.64,
+      "learning_rate": 4.127877237851662e-06,
+      "loss": 0.0024,
+      "step": 3422
+    },
+    {
+      "epoch": 8.64,
+      "learning_rate": 4.120204603580562e-06,
+      "loss": 0.0039,
+      "step": 3423
+    },
+    {
+      "epoch": 8.65,
+      "learning_rate": 4.112531969309463e-06,
+      "loss": 0.0047,
+      "step": 3424
+    },
+    {
+      "epoch": 8.65,
+      "learning_rate": 4.104859335038363e-06,
+      "loss": 0.0225,
+      "step": 3425
+    },
+    {
+      "epoch": 8.65,
+      "learning_rate": 4.0971867007672636e-06,
+      "loss": 0.0298,
+      "step": 3426
+    },
+    {
+      "epoch": 8.65,
+      "learning_rate": 4.0895140664961635e-06,
+      "loss": 1.1412,
+      "step": 3427
+    },
+    {
+      "epoch": 8.66,
+      "learning_rate": 4.081841432225064e-06,
+      "loss": 0.0038,
+      "step": 3428
+    },
+    {
+      "epoch": 8.66,
+      "learning_rate": 4.074168797953965e-06,
+      "loss": 0.0027,
+      "step": 3429
+    },
+    {
+      "epoch": 8.66,
+      "learning_rate": 4.066496163682865e-06,
+      "loss": 0.0042,
+      "step": 3430
+    },
+    {
+      "epoch": 8.66,
+      "learning_rate": 4.058823529411765e-06,
+      "loss": 0.0035,
+      "step": 3431
+    },
+    {
+      "epoch": 8.67,
+      "learning_rate": 4.051150895140666e-06,
+      "loss": 0.0038,
+      "step": 3432
+    },
+    {
+      "epoch": 8.67,
+      "learning_rate": 4.0434782608695655e-06,
+      "loss": 0.0034,
+      "step": 3433
+    },
+    {
+      "epoch": 8.67,
+      "learning_rate": 4.0358056265984655e-06,
+      "loss": 0.0025,
+      "step": 3434
+    },
+    {
+      "epoch": 8.67,
+      "learning_rate": 4.028132992327366e-06,
+      "loss": 0.0046,
+      "step": 3435
+    },
+    {
+      "epoch": 8.68,
+      "learning_rate": 4.020460358056266e-06,
+      "loss": 0.0023,
+      "step": 3436
+    },
+    {
+      "epoch": 8.68,
+      "learning_rate": 4.012787723785166e-06,
+      "loss": 0.004,
+      "step": 3437
+    },
+    {
+      "epoch": 8.68,
+      "learning_rate": 4.005115089514067e-06,
+      "loss": 0.0035,
+      "step": 3438
+    },
+    {
+      "epoch": 8.68,
+      "learning_rate": 3.997442455242967e-06,
+      "loss": 0.0057,
+      "step": 3439
+    },
+    {
+      "epoch": 8.69,
+      "learning_rate": 3.9897698209718675e-06,
+      "loss": 1.4514,
+      "step": 3440
+    },
+    {
+      "epoch": 8.69,
+      "learning_rate": 3.982097186700767e-06,
+      "loss": 0.006,
+      "step": 3441
+    },
+    {
+      "epoch": 8.69,
+      "learning_rate": 3.974424552429667e-06,
+      "loss": 0.0025,
+      "step": 3442
+    },
+    {
+      "epoch": 8.69,
+      "learning_rate": 3.966751918158568e-06,
+      "loss": 0.0043,
+      "step": 3443
+    },
+    {
+      "epoch": 8.7,
+      "learning_rate": 3.959079283887468e-06,
+      "loss": 0.0035,
+      "step": 3444
+    },
+    {
+      "epoch": 8.7,
+      "learning_rate": 3.951406649616368e-06,
+      "loss": 0.0036,
+      "step": 3445
+    },
+    {
+      "epoch": 8.7,
+      "learning_rate": 3.943734015345269e-06,
+      "loss": 0.0034,
+      "step": 3446
+    },
+    {
+      "epoch": 8.7,
+      "learning_rate": 3.936061381074169e-06,
+      "loss": 0.0045,
+      "step": 3447
+    },
+    {
+      "epoch": 8.71,
+      "learning_rate": 3.9283887468030685e-06,
+      "loss": 0.0056,
+      "step": 3448
+    },
+    {
+      "epoch": 8.71,
+      "learning_rate": 3.920716112531969e-06,
+      "loss": 0.003,
+      "step": 3449
+    },
+    {
+      "epoch": 8.71,
+      "learning_rate": 3.913043478260869e-06,
+      "loss": 1.316,
+      "step": 3450
+    },
+    {
+      "epoch": 8.71,
+      "learning_rate": 3.905370843989769e-06,
+      "loss": 0.0041,
+      "step": 3451
+    },
+    {
+      "epoch": 8.72,
+      "learning_rate": 3.89769820971867e-06,
+      "loss": 0.0046,
+      "step": 3452
+    },
+    {
+      "epoch": 8.72,
+      "learning_rate": 3.890025575447571e-06,
+      "loss": 0.0037,
+      "step": 3453
+    },
+    {
+      "epoch": 8.72,
+      "learning_rate": 3.8823529411764714e-06,
+      "loss": 0.0023,
+      "step": 3454
+    },
+    {
+      "epoch": 8.72,
+      "learning_rate": 3.874680306905371e-06,
+      "loss": 0.0037,
+      "step": 3455
+    },
+    {
+      "epoch": 8.73,
+      "learning_rate": 3.867007672634271e-06,
+      "loss": 0.0038,
+      "step": 3456
+    },
+    {
+      "epoch": 8.73,
+      "learning_rate": 3.859335038363172e-06,
+      "loss": 0.0045,
+      "step": 3457
+    },
+    {
+      "epoch": 8.73,
+      "learning_rate": 3.851662404092072e-06,
+      "loss": 0.0054,
+      "step": 3458
+    },
+    {
+      "epoch": 8.73,
+      "learning_rate": 3.843989769820972e-06,
+      "loss": 0.0023,
+      "step": 3459
+    },
+    {
+      "epoch": 8.74,
+      "learning_rate": 3.836317135549873e-06,
+      "loss": 0.0058,
+      "step": 3460
+    },
+    {
+      "epoch": 8.74,
+      "learning_rate": 3.8286445012787726e-06,
+      "loss": 0.0033,
+      "step": 3461
+    },
+    {
+      "epoch": 8.74,
+      "learning_rate": 3.8209718670076725e-06,
+      "loss": 0.0045,
+      "step": 3462
+    },
+    {
+      "epoch": 8.74,
+      "learning_rate": 3.8132992327365732e-06,
+      "loss": 0.0037,
+      "step": 3463
+    },
+    {
+      "epoch": 8.75,
+      "learning_rate": 3.805626598465473e-06,
+      "loss": 0.0034,
+      "step": 3464
+    },
+    {
+      "epoch": 8.75,
+      "learning_rate": 3.797953964194374e-06,
+      "loss": 0.0025,
+      "step": 3465
+    },
+    {
+      "epoch": 8.75,
+      "learning_rate": 3.790281329923274e-06,
+      "loss": 0.005,
+      "step": 3466
+    },
+    {
+      "epoch": 8.76,
+      "learning_rate": 3.7826086956521738e-06,
+      "loss": 0.0036,
+      "step": 3467
+    },
+    {
+      "epoch": 8.76,
+      "learning_rate": 3.7749360613810745e-06,
+      "loss": 0.0023,
+      "step": 3468
+    },
+    {
+      "epoch": 8.76,
+      "learning_rate": 3.7672634271099744e-06,
+      "loss": 0.0022,
+      "step": 3469
+    },
+    {
+      "epoch": 8.76,
+      "learning_rate": 3.7595907928388744e-06,
+      "loss": 0.0038,
+      "step": 3470
+    },
+    {
+      "epoch": 8.77,
+      "learning_rate": 3.751918158567775e-06,
+      "loss": 1.5236,
+      "step": 3471
+    },
+    {
+      "epoch": 8.77,
+      "learning_rate": 3.7442455242966755e-06,
+      "loss": 0.0033,
+      "step": 3472
+    },
+    {
+      "epoch": 8.77,
+      "learning_rate": 3.736572890025576e-06,
+      "loss": 0.0066,
+      "step": 3473
+    },
+    {
+      "epoch": 8.77,
+      "learning_rate": 3.7289002557544757e-06,
+      "loss": 0.0059,
+      "step": 3474
+    },
+    {
+      "epoch": 8.78,
+      "learning_rate": 3.721227621483376e-06,
+      "loss": 0.0034,
+      "step": 3475
+    },
+    {
+      "epoch": 8.78,
+      "learning_rate": 3.7135549872122764e-06,
+      "loss": 0.0033,
+      "step": 3476
+    },
+    {
+      "epoch": 8.78,
+      "learning_rate": 3.7058823529411767e-06,
+      "loss": 0.0034,
+      "step": 3477
+    },
+    {
+      "epoch": 8.78,
+      "learning_rate": 3.6982097186700767e-06,
+      "loss": 0.0053,
+      "step": 3478
+    },
+    {
+      "epoch": 8.79,
+      "learning_rate": 3.690537084398977e-06,
+      "loss": 0.0066,
+      "step": 3479
+    },
+    {
+      "epoch": 8.79,
+      "learning_rate": 3.6828644501278773e-06,
+      "loss": 0.0047,
+      "step": 3480
+    },
+    {
+      "epoch": 8.79,
+      "learning_rate": 3.6751918158567773e-06,
+      "loss": 0.0022,
+      "step": 3481
+    },
+    {
+      "epoch": 8.79,
+      "learning_rate": 3.6675191815856776e-06,
+      "loss": 0.0034,
+      "step": 3482
+    },
+    {
+      "epoch": 8.8,
+      "learning_rate": 3.659846547314578e-06,
+      "loss": 0.0034,
+      "step": 3483
+    },
+    {
+      "epoch": 8.8,
+      "learning_rate": 3.6521739130434787e-06,
+      "loss": 0.0079,
+      "step": 3484
+    },
+    {
+      "epoch": 8.8,
+      "learning_rate": 3.6445012787723786e-06,
+      "loss": 0.0045,
+      "step": 3485
+    },
+    {
+      "epoch": 8.8,
+      "learning_rate": 3.636828644501279e-06,
+      "loss": 0.0033,
+      "step": 3486
+    },
+    {
+      "epoch": 8.81,
+      "learning_rate": 3.6291560102301793e-06,
+      "loss": 0.0029,
+      "step": 3487
+    },
+    {
+      "epoch": 8.81,
+      "learning_rate": 3.6214833759590796e-06,
+      "loss": 0.0047,
+      "step": 3488
+    },
+    {
+      "epoch": 8.81,
+      "learning_rate": 3.6138107416879796e-06,
+      "loss": 0.0027,
+      "step": 3489
+    },
+    {
+      "epoch": 8.81,
+      "learning_rate": 3.60613810741688e-06,
+      "loss": 0.0043,
+      "step": 3490
+    },
+    {
+      "epoch": 8.82,
+      "learning_rate": 3.5984654731457802e-06,
+      "loss": 0.0033,
+      "step": 3491
+    },
+    {
+      "epoch": 8.82,
+      "learning_rate": 3.59079283887468e-06,
+      "loss": 0.0033,
+      "step": 3492
+    },
+    {
+      "epoch": 8.82,
+      "learning_rate": 3.5831202046035805e-06,
+      "loss": 0.0044,
+      "step": 3493
+    },
+    {
+      "epoch": 8.82,
+      "learning_rate": 3.575447570332481e-06,
+      "loss": 0.0025,
+      "step": 3494
+    },
+    {
+      "epoch": 8.83,
+      "learning_rate": 3.567774936061381e-06,
+      "loss": 0.0023,
+      "step": 3495
+    },
+    {
+      "epoch": 8.83,
+      "learning_rate": 3.5601023017902815e-06,
+      "loss": 1.2509,
+      "step": 3496
+    },
+    {
+      "epoch": 8.83,
+      "learning_rate": 3.552429667519182e-06,
+      "loss": 0.003,
+      "step": 3497
+    },
+    {
+      "epoch": 8.83,
+      "learning_rate": 3.544757033248082e-06,
+      "loss": 0.0024,
+      "step": 3498
+    },
+    {
+      "epoch": 8.84,
+      "learning_rate": 3.537084398976982e-06,
+      "loss": 1.388,
+      "step": 3499
+    },
+    {
+      "epoch": 8.84,
+      "learning_rate": 3.5294117647058825e-06,
+      "loss": 0.0043,
+      "step": 3500
+    },
+    {
+      "epoch": 8.84,
+      "learning_rate": 3.521739130434783e-06,
+      "loss": 0.0043,
+      "step": 3501
+    },
+    {
+      "epoch": 8.84,
+      "learning_rate": 3.514066496163683e-06,
+      "loss": 0.0044,
+      "step": 3502
+    },
+    {
+      "epoch": 8.85,
+      "learning_rate": 3.506393861892583e-06,
+      "loss": 0.0043,
+      "step": 3503
+    },
+    {
+      "epoch": 8.85,
+      "learning_rate": 3.4987212276214834e-06,
+      "loss": 1.3346,
+      "step": 3504
+    },
+    {
+      "epoch": 8.85,
+      "learning_rate": 3.4910485933503837e-06,
+      "loss": 0.0024,
+      "step": 3505
+    },
+    {
+      "epoch": 8.85,
+      "learning_rate": 3.4833759590792837e-06,
+      "loss": 0.0031,
+      "step": 3506
+    },
+    {
+      "epoch": 8.86,
+      "learning_rate": 3.475703324808184e-06,
+      "loss": 0.0034,
+      "step": 3507
+    },
+    {
+      "epoch": 8.86,
+      "learning_rate": 3.4680306905370843e-06,
+      "loss": 0.0033,
+      "step": 3508
+    },
+    {
+      "epoch": 8.86,
+      "learning_rate": 3.460358056265985e-06,
+      "loss": 0.0043,
+      "step": 3509
+    },
+    {
+      "epoch": 8.86,
+      "learning_rate": 3.452685421994885e-06,
+      "loss": 0.0035,
+      "step": 3510
+    },
+    {
+      "epoch": 8.87,
+      "learning_rate": 3.4450127877237854e-06,
+      "loss": 0.0045,
+      "step": 3511
+    },
+    {
+      "epoch": 8.87,
+      "learning_rate": 3.4373401534526857e-06,
+      "loss": 0.0033,
+      "step": 3512
+    },
+    {
+      "epoch": 8.87,
+      "learning_rate": 3.4296675191815856e-06,
+      "loss": 0.0033,
+      "step": 3513
+    },
+    {
+      "epoch": 8.87,
+      "learning_rate": 3.421994884910486e-06,
+      "loss": 1.4988,
+      "step": 3514
+    },
+    {
+      "epoch": 8.88,
+      "learning_rate": 3.4143222506393863e-06,
+      "loss": 0.0048,
+      "step": 3515
+    },
+    {
+      "epoch": 8.88,
+      "learning_rate": 3.4066496163682867e-06,
+      "loss": 0.0017,
+      "step": 3516
+    },
+    {
+      "epoch": 8.88,
+      "learning_rate": 3.3989769820971866e-06,
+      "loss": 0.0043,
+      "step": 3517
+    },
+    {
+      "epoch": 8.88,
+      "learning_rate": 3.391304347826087e-06,
+      "loss": 1.4504,
+      "step": 3518
+    },
+    {
+      "epoch": 8.89,
+      "learning_rate": 3.3836317135549873e-06,
+      "loss": 0.0033,
+      "step": 3519
+    },
+    {
+      "epoch": 8.89,
+      "learning_rate": 3.375959079283887e-06,
+      "loss": 0.0043,
+      "step": 3520
+    },
+    {
+      "epoch": 8.89,
+      "learning_rate": 3.368286445012788e-06,
+      "loss": 0.0042,
+      "step": 3521
+    },
+    {
+      "epoch": 8.89,
+      "learning_rate": 3.3606138107416883e-06,
+      "loss": 0.0023,
+      "step": 3522
+    },
+    {
+      "epoch": 8.9,
+      "learning_rate": 3.3529411764705886e-06,
+      "loss": 0.0084,
+      "step": 3523
+    },
+    {
+      "epoch": 8.9,
+      "learning_rate": 3.3452685421994885e-06,
+      "loss": 0.0049,
+      "step": 3524
+    },
+    {
+      "epoch": 8.9,
+      "learning_rate": 3.337595907928389e-06,
+      "loss": 0.0048,
+      "step": 3525
+    },
+    {
+      "epoch": 8.9,
+      "learning_rate": 3.3299232736572892e-06,
+      "loss": 0.0043,
+      "step": 3526
+    },
+    {
+      "epoch": 8.91,
+      "learning_rate": 3.322250639386189e-06,
+      "loss": 0.0037,
+      "step": 3527
+    },
+    {
+      "epoch": 8.91,
+      "learning_rate": 3.3145780051150895e-06,
+      "loss": 0.005,
+      "step": 3528
+    },
+    {
+      "epoch": 8.91,
+      "learning_rate": 3.30690537084399e-06,
+      "loss": 0.0024,
+      "step": 3529
+    },
+    {
+      "epoch": 8.91,
+      "learning_rate": 3.29923273657289e-06,
+      "loss": 0.0017,
+      "step": 3530
+    },
+    {
+      "epoch": 8.92,
+      "learning_rate": 3.29156010230179e-06,
+      "loss": 0.0037,
+      "step": 3531
+    },
+    {
+      "epoch": 8.92,
+      "learning_rate": 3.2838874680306904e-06,
+      "loss": 1.2684,
+      "step": 3532
+    },
+    {
+      "epoch": 8.92,
+      "learning_rate": 3.2762148337595908e-06,
+      "loss": 0.0043,
+      "step": 3533
+    },
+    {
+      "epoch": 8.92,
+      "learning_rate": 3.2685421994884915e-06,
+      "loss": 0.0058,
+      "step": 3534
+    },
+    {
+      "epoch": 8.93,
+      "learning_rate": 3.2608695652173914e-06,
+      "loss": 1.2339,
+      "step": 3535
+    },
+    {
+      "epoch": 8.93,
+      "learning_rate": 3.2531969309462918e-06,
+      "loss": 0.0033,
+      "step": 3536
+    },
+    {
+      "epoch": 8.93,
+      "learning_rate": 3.245524296675192e-06,
+      "loss": 0.0043,
+      "step": 3537
+    },
+    {
+      "epoch": 8.93,
+      "learning_rate": 3.237851662404092e-06,
+      "loss": 0.003,
+      "step": 3538
+    },
+    {
+      "epoch": 8.94,
+      "learning_rate": 3.2301790281329924e-06,
+      "loss": 0.0049,
+      "step": 3539
+    },
+    {
+      "epoch": 8.94,
+      "learning_rate": 3.2225063938618927e-06,
+      "loss": 0.0039,
+      "step": 3540
+    },
+    {
+      "epoch": 8.94,
+      "learning_rate": 3.214833759590793e-06,
+      "loss": 0.0034,
+      "step": 3541
+    },
+    {
+      "epoch": 8.94,
+      "learning_rate": 3.207161125319693e-06,
+      "loss": 0.0058,
+      "step": 3542
+    },
+    {
+      "epoch": 8.95,
+      "learning_rate": 3.1994884910485933e-06,
+      "loss": 0.0034,
+      "step": 3543
+    },
+    {
+      "epoch": 8.95,
+      "learning_rate": 3.1918158567774937e-06,
+      "loss": 0.0039,
+      "step": 3544
+    },
+    {
+      "epoch": 8.95,
+      "learning_rate": 3.1841432225063936e-06,
+      "loss": 0.003,
+      "step": 3545
+    },
+    {
+      "epoch": 8.95,
+      "learning_rate": 3.1764705882352943e-06,
+      "loss": 0.0033,
+      "step": 3546
+    },
+    {
+      "epoch": 8.96,
+      "learning_rate": 3.1687979539641947e-06,
+      "loss": 0.0041,
+      "step": 3547
+    },
+    {
+      "epoch": 8.96,
+      "learning_rate": 3.161125319693095e-06,
+      "loss": 0.0033,
+      "step": 3548
+    },
+    {
+      "epoch": 8.96,
+      "learning_rate": 3.153452685421995e-06,
+      "loss": 0.0022,
+      "step": 3549
+    },
+    {
+      "epoch": 8.96,
+      "learning_rate": 3.1457800511508953e-06,
+      "loss": 0.0029,
+      "step": 3550
+    },
+    {
+      "epoch": 8.97,
+      "learning_rate": 3.1381074168797956e-06,
+      "loss": 0.0033,
+      "step": 3551
+    },
+    {
+      "epoch": 8.97,
+      "learning_rate": 3.1304347826086955e-06,
+      "loss": 0.0033,
+      "step": 3552
+    },
+    {
+      "epoch": 8.97,
+      "learning_rate": 3.122762148337596e-06,
+      "loss": 0.0047,
+      "step": 3553
+    },
+    {
+      "epoch": 8.97,
+      "learning_rate": 3.1150895140664962e-06,
+      "loss": 0.0022,
+      "step": 3554
+    },
+    {
+      "epoch": 8.98,
+      "learning_rate": 3.1074168797953966e-06,
+      "loss": 0.0033,
+      "step": 3555
+    },
+    {
+      "epoch": 8.98,
+      "learning_rate": 3.0997442455242965e-06,
+      "loss": 0.004,
+      "step": 3556
+    },
+    {
+      "epoch": 8.98,
+      "learning_rate": 3.092071611253197e-06,
+      "loss": 0.0022,
+      "step": 3557
+    },
+    {
+      "epoch": 8.98,
+      "learning_rate": 3.0843989769820976e-06,
+      "loss": 0.0024,
+      "step": 3558
+    },
+    {
+      "epoch": 8.99,
+      "learning_rate": 3.0767263427109975e-06,
+      "loss": 0.0036,
+      "step": 3559
+    },
+    {
+      "epoch": 8.99,
+      "learning_rate": 3.069053708439898e-06,
+      "loss": 0.0057,
+      "step": 3560
+    },
+    {
+      "epoch": 8.99,
+      "learning_rate": 3.061381074168798e-06,
+      "loss": 0.0033,
+      "step": 3561
+    },
+    {
+      "epoch": 8.99,
+      "learning_rate": 3.0537084398976985e-06,
+      "loss": 0.0023,
+      "step": 3562
+    },
+    {
+      "epoch": 9.0,
+      "learning_rate": 3.0460358056265985e-06,
+      "loss": 0.0035,
+      "step": 3563
+    },
+    {
+      "epoch": 9.0,
+      "learning_rate": 3.038363171355499e-06,
+      "loss": 0.0038,
+      "step": 3564
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 0.9242424242424242,
+      "eval_loss": 0.4058963358402252,
+      "eval_runtime": 57.7406,
+      "eval_samples_per_second": 6.858,
+      "eval_steps_per_second": 1.715,
+      "step": 3564
+    }
+  ],
+  "max_steps": 3960,
+  "num_train_epochs": 10,
+  "total_flos": 1.2934327922208e+18,
+  "trial_name": null,
+  "trial_params": null
+}
diff --git a/checkpoint-3564/training_args.bin b/checkpoint-3564/training_args.bin
new file mode 100644
index 0000000000000000000000000000000000000000..32901f899c91066c808eae9dd019400bfdd17498
--- /dev/null
+++ b/checkpoint-3564/training_args.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db14439c369bfe9c89f64ff38f0b2d3fcc53911fc2c0b12a3a7d04656e7384a
+size 3503
diff --git a/checkpoint-396/config.json b/checkpoint-396/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..cb8d52e14263c50fd53dca42b5050e3235115c32
--- /dev/null
+++ b/checkpoint-396/config.json
@@ -0,0 +1,113 @@
+{
+  "_name_or_path": "facebook/wav2vec2-base-960h",
+  "activation_dropout": 0.1,
+  "adapter_kernel_size": 3,
+  "adapter_stride": 2,
+  "add_adapter": false,
+  "apply_spec_augment": true,
+  "architectures": [
+    "Wav2Vec2ForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 1,
+  "classifier_proj_size": 256,
+  "codevector_dim": 256,
+  "contrastive_logits_temperature": 0.1,
+  "conv_bias": false,
+  "conv_dim": [
+    512,
+    512,
+    512,
+    512,
+    512,
+    512,
+    512
+  ],
+  "conv_kernel": [
+    10,
+    3,
+    3,
+    3,
+    3,
+    2,
+    2
+  ],
+  "conv_stride": [
+    5,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2
+  ],
+  "ctc_loss_reduction": "sum",
+  "ctc_zero_infinity": false,
+  "diversity_loss_weight": 0.1,
+  "do_stable_layer_norm": false,
+  "eos_token_id": 2,
+  "feat_extract_activation": "gelu",
+  "feat_extract_dropout": 0.0,
+  "feat_extract_norm": "group",
+  "feat_proj_dropout": 0.1,
+  "feat_quantizer_dropout": 0.0,
+  "final_dropout": 0.1,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout": 0.1,
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Foreign",
+    "1": "Native"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "layerdrop": 0.1,
+  "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
+  "mask_feature_prob": 0.0,
+  "mask_time_length": 10,
+  "mask_time_min_masks": 2,
+  "mask_time_prob": 0.05,
+  "model_type": "wav2vec2",
+  "num_adapter_layers": 3,
+  "num_attention_heads": 12,
+  "num_codevector_groups": 2,
+  "num_codevectors_per_group": 320,
+  "num_conv_pos_embedding_groups": 16,
+  "num_conv_pos_embeddings": 128,
+  "num_feat_extract_layers": 7,
+  "num_hidden_layers": 12,
+  "num_negatives": 100,
+  "output_hidden_size": 768,
+  "pad_token_id": 0,
+  "proj_codevector_dim": 256,
+  "tdnn_dilation": [
+    1,
+    2,
+    3,
+    1,
+    1
+  ],
+  "tdnn_dim": [
+    512,
+    512,
+    512,
+    512,
+    1500
+  ],
+  "tdnn_kernel": [
+    5,
+    3,
+    3,
+    1,
+    1
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.28.1",
+  "use_weighted_layer_sum": false,
+  "vocab_size": 32,
+  "xvector_output_dim": 512
+}
diff --git a/checkpoint-396/optimizer.pt b/checkpoint-396/optimizer.pt
new file mode 100644
index 0000000000000000000000000000000000000000..d4328f33e1ad4e04a690aec55599f683c969d8ef
--- /dev/null
+++ b/checkpoint-396/optimizer.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c338614b720c662f4c27b653def73289b30aba26b54b21570de1a7eb770b6bcd
+size 756677477
diff --git a/checkpoint-396/preprocessor_config.json b/checkpoint-396/preprocessor_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..a0b7227fc1d916e469b14f6c154ad6dfea1e6891
--- /dev/null
+++ b/checkpoint-396/preprocessor_config.json
@@ -0,0 +1,9 @@
+{
+  "do_normalize": true,
+  "feature_extractor_type": "Wav2Vec2FeatureExtractor",
+  "feature_size": 1,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": false,
+  "sampling_rate": 16000
+}
diff --git a/checkpoint-396/pytorch_model.bin b/checkpoint-396/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..abdf2e42aa7e2e337c14c4f8eb7af93b494ff5ad
--- /dev/null
+++ b/checkpoint-396/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b472fde9fc3d14bcdfe46e0277a01733c313cb4e30adac5150155219fa144a46
+size 378348563
diff --git a/checkpoint-396/rng_state.pth b/checkpoint-396/rng_state.pth
new file mode 100644
index 0000000000000000000000000000000000000000..c336b452db019105521809ccc3657a21cf0d0031
--- /dev/null
+++ b/checkpoint-396/rng_state.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c037bf99c9c06329fd35485b0d4917d06571c04b00a23355b4edd375afa0a18
+size 14503
diff --git a/checkpoint-396/scheduler.pt b/checkpoint-396/scheduler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..f706ea7d0ecbb2a414b02ce971da02b40d391cc7
--- /dev/null
+++ b/checkpoint-396/scheduler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:74ee58aa35faa0337f54225142bd03b3ae1e78d7d0b70eedb8bf49fe38ebb61f
+size 623
diff --git a/checkpoint-396/trainer_state.json b/checkpoint-396/trainer_state.json
new file mode 100644
index 0000000000000000000000000000000000000000..fe2e4b8b48e0ca177b91469521e03d04d754c2e1
--- /dev/null
+++ b/checkpoint-396/trainer_state.json
@@ -0,0 +1,2401 @@
+{
+  "best_metric": 0.5804409980773926,
+  "best_model_checkpoint": "accent_classification/checkpoint-396",
+  "epoch": 1.0,
+  "global_step": 396,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.000000000000001e-07,
+      "loss": 0.6956,
+      "step": 1
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.2000000000000002e-06,
+      "loss": 0.6979,
+      "step": 2
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8e-06,
+      "loss": 0.7014,
+      "step": 3
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 2.4000000000000003e-06,
+      "loss": 0.6984,
+      "step": 4
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 3e-06,
+      "loss": 0.6978,
+      "step": 5
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 3.6e-06,
+      "loss": 0.6783,
+      "step": 6
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.2000000000000004e-06,
+      "loss": 0.6956,
+      "step": 7
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.800000000000001e-06,
+      "loss": 0.6994,
+      "step": 8
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 5.4e-06,
+      "loss": 0.6902,
+      "step": 9
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6e-06,
+      "loss": 0.6937,
+      "step": 10
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6.6e-06,
+      "loss": 0.6497,
+      "step": 11
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.2e-06,
+      "loss": 0.6911,
+      "step": 12
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.8e-06,
+      "loss": 0.6904,
+      "step": 13
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 8.400000000000001e-06,
+      "loss": 0.6854,
+      "step": 14
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9e-06,
+      "loss": 0.6843,
+      "step": 15
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9.600000000000001e-06,
+      "loss": 0.6851,
+      "step": 16
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.02e-05,
+      "loss": 0.6733,
+      "step": 17
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.08e-05,
+      "loss": 0.6819,
+      "step": 18
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.1400000000000001e-05,
+      "loss": 0.681,
+      "step": 19
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.6998,
+      "step": 20
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.26e-05,
+      "loss": 0.6743,
+      "step": 21
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.32e-05,
+      "loss": 0.6818,
+      "step": 22
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3800000000000002e-05,
+      "loss": 0.5331,
+      "step": 23
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.44e-05,
+      "loss": 0.6681,
+      "step": 24
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.6955,
+      "step": 25
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.56e-05,
+      "loss": 0.7,
+      "step": 26
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.62e-05,
+      "loss": 0.6975,
+      "step": 27
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.6800000000000002e-05,
+      "loss": 0.6249,
+      "step": 28
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.74e-05,
+      "loss": 0.6979,
+      "step": 29
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.6076,
+      "step": 30
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.86e-05,
+      "loss": 0.6091,
+      "step": 31
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9200000000000003e-05,
+      "loss": 0.6438,
+      "step": 32
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.98e-05,
+      "loss": 0.6624,
+      "step": 33
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.04e-05,
+      "loss": 0.6452,
+      "step": 34
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.6963,
+      "step": 35
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.16e-05,
+      "loss": 0.758,
+      "step": 36
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.22e-05,
+      "loss": 0.6471,
+      "step": 37
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.2800000000000002e-05,
+      "loss": 0.6313,
+      "step": 38
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.3400000000000003e-05,
+      "loss": 0.7019,
+      "step": 39
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.6336,
+      "step": 40
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4599999999999998e-05,
+      "loss": 0.7091,
+      "step": 41
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.52e-05,
+      "loss": 0.701,
+      "step": 42
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.58e-05,
+      "loss": 0.7036,
+      "step": 43
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.64e-05,
+      "loss": 0.7073,
+      "step": 44
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.614,
+      "step": 45
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.7600000000000003e-05,
+      "loss": 0.7543,
+      "step": 46
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.8199999999999998e-05,
+      "loss": 0.538,
+      "step": 47
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.88e-05,
+      "loss": 0.5179,
+      "step": 48
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.94e-05,
+      "loss": 0.6164,
+      "step": 49
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 3e-05,
+      "loss": 0.6093,
+      "step": 50
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.99923273657289e-05,
+      "loss": 0.6983,
+      "step": 51
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9984654731457804e-05,
+      "loss": 0.6069,
+      "step": 52
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9976982097186702e-05,
+      "loss": 0.5933,
+      "step": 53
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.99693094629156e-05,
+      "loss": 0.552,
+      "step": 54
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9961636828644502e-05,
+      "loss": 0.5865,
+      "step": 55
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9953964194373404e-05,
+      "loss": 0.7146,
+      "step": 56
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9946291560102302e-05,
+      "loss": 0.5591,
+      "step": 57
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9938618925831203e-05,
+      "loss": 0.7366,
+      "step": 58
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9930946291560102e-05,
+      "loss": 0.7006,
+      "step": 59
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9923273657289003e-05,
+      "loss": 0.4148,
+      "step": 60
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9915601023017905e-05,
+      "loss": 0.9938,
+      "step": 61
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9907928388746803e-05,
+      "loss": 0.911,
+      "step": 62
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.99002557544757e-05,
+      "loss": 0.5682,
+      "step": 63
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9892583120204606e-05,
+      "loss": 0.5833,
+      "step": 64
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9884910485933505e-05,
+      "loss": 0.7237,
+      "step": 65
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9877237851662403e-05,
+      "loss": 0.4013,
+      "step": 66
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9869565217391304e-05,
+      "loss": 0.9695,
+      "step": 67
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9861892583120206e-05,
+      "loss": 0.568,
+      "step": 68
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9854219948849104e-05,
+      "loss": 0.5697,
+      "step": 69
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9846547314578006e-05,
+      "loss": 0.7675,
+      "step": 70
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9838874680306904e-05,
+      "loss": 0.8831,
+      "step": 71
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9831202046035806e-05,
+      "loss": 0.7008,
+      "step": 72
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9823529411764707e-05,
+      "loss": 0.7085,
+      "step": 73
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9815856777493606e-05,
+      "loss": 0.5374,
+      "step": 74
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9808184143222507e-05,
+      "loss": 0.4131,
+      "step": 75
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.980051150895141e-05,
+      "loss": 0.7309,
+      "step": 76
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9792838874680307e-05,
+      "loss": 0.4279,
+      "step": 77
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.978516624040921e-05,
+      "loss": 0.5813,
+      "step": 78
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.977749360613811e-05,
+      "loss": 0.6276,
+      "step": 79
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976982097186701e-05,
+      "loss": 0.7892,
+      "step": 80
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976214833759591e-05,
+      "loss": 0.7134,
+      "step": 81
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.9754475703324808e-05,
+      "loss": 0.3682,
+      "step": 82
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.974680306905371e-05,
+      "loss": 0.5562,
+      "step": 83
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973913043478261e-05,
+      "loss": 0.7158,
+      "step": 84
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973145780051151e-05,
+      "loss": 0.524,
+      "step": 85
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9723785166240408e-05,
+      "loss": 0.3372,
+      "step": 86
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9716112531969313e-05,
+      "loss": 0.4979,
+      "step": 87
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970843989769821e-05,
+      "loss": 0.3177,
+      "step": 88
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970076726342711e-05,
+      "loss": 0.3488,
+      "step": 89
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.969309462915601e-05,
+      "loss": 0.5158,
+      "step": 90
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9685421994884913e-05,
+      "loss": 0.3208,
+      "step": 91
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.967774936061381e-05,
+      "loss": 0.6654,
+      "step": 92
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9670076726342712e-05,
+      "loss": 0.5581,
+      "step": 93
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.966240409207161e-05,
+      "loss": 0.7456,
+      "step": 94
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9654731457800512e-05,
+      "loss": 0.4437,
+      "step": 95
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9647058823529414e-05,
+      "loss": 0.5525,
+      "step": 96
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9639386189258312e-05,
+      "loss": 0.6673,
+      "step": 97
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.963171355498721e-05,
+      "loss": 0.435,
+      "step": 98
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9624040920716115e-05,
+      "loss": 0.7415,
+      "step": 99
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9616368286445014e-05,
+      "loss": 0.2981,
+      "step": 100
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9608695652173912e-05,
+      "loss": 0.5179,
+      "step": 101
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9601023017902817e-05,
+      "loss": 0.4005,
+      "step": 102
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9593350383631715e-05,
+      "loss": 0.4598,
+      "step": 103
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9585677749360613e-05,
+      "loss": 0.2553,
+      "step": 104
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9578005115089515e-05,
+      "loss": 0.4302,
+      "step": 105
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9570332480818416e-05,
+      "loss": 0.7894,
+      "step": 106
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9562659846547315e-05,
+      "loss": 0.213,
+      "step": 107
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9554987212276216e-05,
+      "loss": 0.5507,
+      "step": 108
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9547314578005115e-05,
+      "loss": 0.7676,
+      "step": 109
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9539641943734016e-05,
+      "loss": 0.3418,
+      "step": 110
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9531969309462918e-05,
+      "loss": 0.8533,
+      "step": 111
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9524296675191816e-05,
+      "loss": 0.3659,
+      "step": 112
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9516624040920714e-05,
+      "loss": 0.6482,
+      "step": 113
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.950895140664962e-05,
+      "loss": 0.5484,
+      "step": 114
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9501278772378517e-05,
+      "loss": 0.4508,
+      "step": 115
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9493606138107416e-05,
+      "loss": 0.4401,
+      "step": 116
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9485933503836317e-05,
+      "loss": 0.5455,
+      "step": 117
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.947826086956522e-05,
+      "loss": 0.722,
+      "step": 118
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9470588235294117e-05,
+      "loss": 0.2874,
+      "step": 119
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.946291560102302e-05,
+      "loss": 0.5973,
+      "step": 120
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.9455242966751917e-05,
+      "loss": 0.5898,
+      "step": 121
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.944757033248082e-05,
+      "loss": 1.0125,
+      "step": 122
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943989769820972e-05,
+      "loss": 0.2511,
+      "step": 123
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943222506393862e-05,
+      "loss": 0.545,
+      "step": 124
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.942455242966752e-05,
+      "loss": 0.593,
+      "step": 125
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.941687979539642e-05,
+      "loss": 0.2319,
+      "step": 126
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.940920716112532e-05,
+      "loss": 0.2215,
+      "step": 127
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.9401534526854218e-05,
+      "loss": 0.1964,
+      "step": 128
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9393861892583123e-05,
+      "loss": 0.1471,
+      "step": 129
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.938618925831202e-05,
+      "loss": 1.5965,
+      "step": 130
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9378516624040923e-05,
+      "loss": 1.0772,
+      "step": 131
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.937084398976982e-05,
+      "loss": 0.2005,
+      "step": 132
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9363171355498723e-05,
+      "loss": 0.9628,
+      "step": 133
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9355498721227624e-05,
+      "loss": 0.1503,
+      "step": 134
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9347826086956523e-05,
+      "loss": 0.6201,
+      "step": 135
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.934015345268542e-05,
+      "loss": 0.6094,
+      "step": 136
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9332480818414326e-05,
+      "loss": 1.385,
+      "step": 137
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9324808184143224e-05,
+      "loss": 0.1552,
+      "step": 138
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9317135549872122e-05,
+      "loss": 0.5828,
+      "step": 139
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9309462915601024e-05,
+      "loss": 0.6018,
+      "step": 140
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9301790281329925e-05,
+      "loss": 1.0614,
+      "step": 141
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9294117647058824e-05,
+      "loss": 0.566,
+      "step": 142
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9286445012787725e-05,
+      "loss": 0.5958,
+      "step": 143
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9278772378516624e-05,
+      "loss": 0.5838,
+      "step": 144
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9271099744245525e-05,
+      "loss": 0.5909,
+      "step": 145
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9263427109974427e-05,
+      "loss": 0.5862,
+      "step": 146
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9255754475703325e-05,
+      "loss": 0.2031,
+      "step": 147
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9248081841432223e-05,
+      "loss": 0.1995,
+      "step": 148
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9240409207161128e-05,
+      "loss": 0.24,
+      "step": 149
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9232736572890026e-05,
+      "loss": 0.5845,
+      "step": 150
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9225063938618925e-05,
+      "loss": 0.964,
+      "step": 151
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.921739130434783e-05,
+      "loss": 0.9595,
+      "step": 152
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9209718670076728e-05,
+      "loss": 0.2033,
+      "step": 153
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9202046035805626e-05,
+      "loss": 0.213,
+      "step": 154
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9194373401534528e-05,
+      "loss": 0.2042,
+      "step": 155
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.918670076726343e-05,
+      "loss": 0.5783,
+      "step": 156
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9179028132992328e-05,
+      "loss": 1.0822,
+      "step": 157
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.917135549872123e-05,
+      "loss": 1.2643,
+      "step": 158
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9163682864450127e-05,
+      "loss": 0.2593,
+      "step": 159
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.915601023017903e-05,
+      "loss": 0.5783,
+      "step": 160
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914833759590793e-05,
+      "loss": 0.5721,
+      "step": 161
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914066496163683e-05,
+      "loss": 0.5808,
+      "step": 162
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9132992327365727e-05,
+      "loss": 1.2578,
+      "step": 163
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9125319693094632e-05,
+      "loss": 0.5709,
+      "step": 164
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.911764705882353e-05,
+      "loss": 0.2159,
+      "step": 165
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910997442455243e-05,
+      "loss": 0.2329,
+      "step": 166
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910230179028133e-05,
+      "loss": 0.926,
+      "step": 167
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.9094629156010232e-05,
+      "loss": 0.9095,
+      "step": 168
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.908695652173913e-05,
+      "loss": 0.5622,
+      "step": 169
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907928388746803e-05,
+      "loss": 0.5703,
+      "step": 170
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907161125319693e-05,
+      "loss": 0.5669,
+      "step": 171
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.906393861892583e-05,
+      "loss": 0.2637,
+      "step": 172
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9056265984654733e-05,
+      "loss": 1.1077,
+      "step": 173
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.904859335038363e-05,
+      "loss": 0.5653,
+      "step": 174
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9040920716112533e-05,
+      "loss": 0.5584,
+      "step": 175
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9033248081841434e-05,
+      "loss": 0.5544,
+      "step": 176
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9025575447570333e-05,
+      "loss": 0.5663,
+      "step": 177
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.901790281329923e-05,
+      "loss": 1.0808,
+      "step": 178
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9010230179028136e-05,
+      "loss": 0.8736,
+      "step": 179
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9002557544757034e-05,
+      "loss": 0.6435,
+      "step": 180
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8994884910485932e-05,
+      "loss": 0.786,
+      "step": 181
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8987212276214834e-05,
+      "loss": 0.5717,
+      "step": 182
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8979539641943736e-05,
+      "loss": 0.5618,
+      "step": 183
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8971867007672637e-05,
+      "loss": 0.5785,
+      "step": 184
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8964194373401535e-05,
+      "loss": 0.7601,
+      "step": 185
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8956521739130434e-05,
+      "loss": 0.8112,
+      "step": 186
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.894884910485934e-05,
+      "loss": 0.5663,
+      "step": 187
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8941176470588237e-05,
+      "loss": 0.5631,
+      "step": 188
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8933503836317135e-05,
+      "loss": 0.7363,
+      "step": 189
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8925831202046037e-05,
+      "loss": 0.4552,
+      "step": 190
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8918158567774938e-05,
+      "loss": 0.4599,
+      "step": 191
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8910485933503836e-05,
+      "loss": 0.7848,
+      "step": 192
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8902813299232738e-05,
+      "loss": 0.5907,
+      "step": 193
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8895140664961636e-05,
+      "loss": 0.4264,
+      "step": 194
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8887468030690538e-05,
+      "loss": 0.5697,
+      "step": 195
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.887979539641944e-05,
+      "loss": 0.7643,
+      "step": 196
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8872122762148338e-05,
+      "loss": 0.3635,
+      "step": 197
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8864450127877236e-05,
+      "loss": 0.3715,
+      "step": 198
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.885677749360614e-05,
+      "loss": 0.554,
+      "step": 199
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.884910485933504e-05,
+      "loss": 0.8044,
+      "step": 200
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8841432225063937e-05,
+      "loss": 0.5673,
+      "step": 201
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8833759590792842e-05,
+      "loss": 0.8042,
+      "step": 202
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.882608695652174e-05,
+      "loss": 0.5638,
+      "step": 203
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881841432225064e-05,
+      "loss": 0.3584,
+      "step": 204
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881074168797954e-05,
+      "loss": 0.8211,
+      "step": 205
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.8803069053708442e-05,
+      "loss": 0.5708,
+      "step": 206
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.879539641943734e-05,
+      "loss": 0.8301,
+      "step": 207
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8787723785166242e-05,
+      "loss": 0.8229,
+      "step": 208
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.878005115089514e-05,
+      "loss": 0.5645,
+      "step": 209
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8772378516624042e-05,
+      "loss": 0.3529,
+      "step": 210
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8764705882352943e-05,
+      "loss": 0.5578,
+      "step": 211
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.875703324808184e-05,
+      "loss": 0.5659,
+      "step": 212
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.874936061381074e-05,
+      "loss": 0.7495,
+      "step": 213
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8741687979539645e-05,
+      "loss": 0.3162,
+      "step": 214
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8734015345268543e-05,
+      "loss": 0.5616,
+      "step": 215
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.872634271099744e-05,
+      "loss": 0.4034,
+      "step": 216
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8718670076726343e-05,
+      "loss": 0.3163,
+      "step": 217
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8710997442455245e-05,
+      "loss": 0.5514,
+      "step": 218
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8703324808184143e-05,
+      "loss": 0.826,
+      "step": 219
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8695652173913044e-05,
+      "loss": 0.8363,
+      "step": 220
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8687979539641943e-05,
+      "loss": 0.8288,
+      "step": 221
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8680306905370844e-05,
+      "loss": 0.5929,
+      "step": 222
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8672634271099746e-05,
+      "loss": 0.565,
+      "step": 223
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8664961636828644e-05,
+      "loss": 0.5642,
+      "step": 224
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8657289002557546e-05,
+      "loss": 0.5609,
+      "step": 225
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8649616368286447e-05,
+      "loss": 0.5641,
+      "step": 226
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8641943734015345e-05,
+      "loss": 0.8535,
+      "step": 227
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8634271099744244e-05,
+      "loss": 0.5652,
+      "step": 228
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.862659846547315e-05,
+      "loss": 0.4397,
+      "step": 229
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8618925831202047e-05,
+      "loss": 0.567,
+      "step": 230
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8611253196930945e-05,
+      "loss": 0.7517,
+      "step": 231
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8603580562659847e-05,
+      "loss": 0.5839,
+      "step": 232
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.859590792838875e-05,
+      "loss": 0.5677,
+      "step": 233
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8588235294117647e-05,
+      "loss": 0.5678,
+      "step": 234
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8580562659846548e-05,
+      "loss": 1.3045,
+      "step": 235
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8572890025575446e-05,
+      "loss": 0.804,
+      "step": 236
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.856521739130435e-05,
+      "loss": 1.0303,
+      "step": 237
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.855754475703325e-05,
+      "loss": 0.3303,
+      "step": 238
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8549872122762148e-05,
+      "loss": 0.5677,
+      "step": 239
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.854219948849105e-05,
+      "loss": 0.7972,
+      "step": 240
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.853452685421995e-05,
+      "loss": 0.5644,
+      "step": 241
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.852685421994885e-05,
+      "loss": 0.9294,
+      "step": 242
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.851918158567775e-05,
+      "loss": 0.5665,
+      "step": 243
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.851150895140665e-05,
+      "loss": 0.7963,
+      "step": 244
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.850383631713555e-05,
+      "loss": 0.5659,
+      "step": 245
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.8496163682864452e-05,
+      "loss": 0.5718,
+      "step": 246
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.848849104859335e-05,
+      "loss": 0.3526,
+      "step": 247
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8480818414322252e-05,
+      "loss": 0.574,
+      "step": 248
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8473145780051154e-05,
+      "loss": 0.9996,
+      "step": 249
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8465473145780052e-05,
+      "loss": 0.7634,
+      "step": 250
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.845780051150895e-05,
+      "loss": 0.5639,
+      "step": 251
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8450127877237855e-05,
+      "loss": 0.5737,
+      "step": 252
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8442455242966753e-05,
+      "loss": 0.7793,
+      "step": 253
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8434782608695652e-05,
+      "loss": 0.6769,
+      "step": 254
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8427109974424553e-05,
+      "loss": 0.5748,
+      "step": 255
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8419437340153455e-05,
+      "loss": 0.3602,
+      "step": 256
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8411764705882353e-05,
+      "loss": 0.5714,
+      "step": 257
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8404092071611255e-05,
+      "loss": 0.5738,
+      "step": 258
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8396419437340153e-05,
+      "loss": 0.5719,
+      "step": 259
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8388746803069055e-05,
+      "loss": 0.5753,
+      "step": 260
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8381074168797956e-05,
+      "loss": 0.5717,
+      "step": 261
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8373401534526854e-05,
+      "loss": 0.9794,
+      "step": 262
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8365728900255753e-05,
+      "loss": 0.5737,
+      "step": 263
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8358056265984658e-05,
+      "loss": 0.5723,
+      "step": 264
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8350383631713556e-05,
+      "loss": 0.5656,
+      "step": 265
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8342710997442454e-05,
+      "loss": 0.4259,
+      "step": 266
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8335038363171356e-05,
+      "loss": 0.3076,
+      "step": 267
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8327365728900257e-05,
+      "loss": 0.338,
+      "step": 268
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8319693094629156e-05,
+      "loss": 0.5612,
+      "step": 269
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8312020460358057e-05,
+      "loss": 0.5249,
+      "step": 270
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8304347826086955e-05,
+      "loss": 0.5671,
+      "step": 271
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8296675191815857e-05,
+      "loss": 0.5574,
+      "step": 272
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.828900255754476e-05,
+      "loss": 0.8152,
+      "step": 273
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8281329923273657e-05,
+      "loss": 0.5647,
+      "step": 274
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.827365728900256e-05,
+      "loss": 0.5632,
+      "step": 275
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.826598465473146e-05,
+      "loss": 0.3391,
+      "step": 276
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8258312020460358e-05,
+      "loss": 0.55,
+      "step": 277
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8250639386189257e-05,
+      "loss": 0.5672,
+      "step": 278
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.824296675191816e-05,
+      "loss": 0.285,
+      "step": 279
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.823529411764706e-05,
+      "loss": 0.2809,
+      "step": 280
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.8227621483375958e-05,
+      "loss": 0.5552,
+      "step": 281
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821994884910486e-05,
+      "loss": 0.563,
+      "step": 282
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821227621483376e-05,
+      "loss": 0.5628,
+      "step": 283
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.820460358056266e-05,
+      "loss": 0.2548,
+      "step": 284
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.819693094629156e-05,
+      "loss": 0.871,
+      "step": 285
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818925831202046e-05,
+      "loss": 0.8797,
+      "step": 286
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818158567774936e-05,
+      "loss": 0.2453,
+      "step": 287
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8173913043478262e-05,
+      "loss": 0.2464,
+      "step": 288
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.816624040920716e-05,
+      "loss": 0.3134,
+      "step": 289
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8158567774936062e-05,
+      "loss": 0.2431,
+      "step": 290
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8150895140664964e-05,
+      "loss": 0.2308,
+      "step": 291
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8143222506393862e-05,
+      "loss": 0.2173,
+      "step": 292
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8135549872122764e-05,
+      "loss": 0.5665,
+      "step": 293
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8127877237851662e-05,
+      "loss": 0.5592,
+      "step": 294
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8120204603580564e-05,
+      "loss": 0.473,
+      "step": 295
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8112531969309465e-05,
+      "loss": 0.9458,
+      "step": 296
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8104859335038363e-05,
+      "loss": 0.1879,
+      "step": 297
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8097186700767265e-05,
+      "loss": 0.6797,
+      "step": 298
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8089514066496167e-05,
+      "loss": 0.7592,
+      "step": 299
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8081841432225065e-05,
+      "loss": 0.4109,
+      "step": 300
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8074168797953963e-05,
+      "loss": 0.5821,
+      "step": 301
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8066496163682868e-05,
+      "loss": 0.2057,
+      "step": 302
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8058823529411766e-05,
+      "loss": 0.5322,
+      "step": 303
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8051150895140665e-05,
+      "loss": 0.5208,
+      "step": 304
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8043478260869566e-05,
+      "loss": 0.6799,
+      "step": 305
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8035805626598468e-05,
+      "loss": 0.5894,
+      "step": 306
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8028132992327366e-05,
+      "loss": 0.4051,
+      "step": 307
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8020460358056268e-05,
+      "loss": 0.3695,
+      "step": 308
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8012787723785166e-05,
+      "loss": 0.4869,
+      "step": 309
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8005115089514067e-05,
+      "loss": 0.6075,
+      "step": 310
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.799744245524297e-05,
+      "loss": 0.4245,
+      "step": 311
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7989769820971867e-05,
+      "loss": 0.5122,
+      "step": 312
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7982097186700765e-05,
+      "loss": 0.452,
+      "step": 313
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.797442455242967e-05,
+      "loss": 0.9568,
+      "step": 314
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.796675191815857e-05,
+      "loss": 0.5846,
+      "step": 315
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.7959079283887467e-05,
+      "loss": 0.3743,
+      "step": 316
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.795140664961637e-05,
+      "loss": 0.1922,
+      "step": 317
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.794373401534527e-05,
+      "loss": 0.5303,
+      "step": 318
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.793606138107417e-05,
+      "loss": 0.7253,
+      "step": 319
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.792838874680307e-05,
+      "loss": 0.2794,
+      "step": 320
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.7920716112531968e-05,
+      "loss": 0.577,
+      "step": 321
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.791304347826087e-05,
+      "loss": 0.614,
+      "step": 322
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.790537084398977e-05,
+      "loss": 0.5882,
+      "step": 323
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789769820971867e-05,
+      "loss": 0.2503,
+      "step": 324
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789002557544757e-05,
+      "loss": 0.4195,
+      "step": 325
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.7882352941176473e-05,
+      "loss": 0.2959,
+      "step": 326
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.787468030690537e-05,
+      "loss": 0.3732,
+      "step": 327
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.786700767263427e-05,
+      "loss": 0.5729,
+      "step": 328
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7859335038363174e-05,
+      "loss": 0.3179,
+      "step": 329
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7851662404092073e-05,
+      "loss": 0.6823,
+      "step": 330
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.784398976982097e-05,
+      "loss": 0.2663,
+      "step": 331
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7836317135549872e-05,
+      "loss": 0.4436,
+      "step": 332
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7828644501278774e-05,
+      "loss": 0.3089,
+      "step": 333
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7820971867007672e-05,
+      "loss": 0.5888,
+      "step": 334
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7813299232736574e-05,
+      "loss": 0.2115,
+      "step": 335
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7805626598465472e-05,
+      "loss": 0.4005,
+      "step": 336
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7797953964194374e-05,
+      "loss": 0.3097,
+      "step": 337
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7790281329923275e-05,
+      "loss": 0.5847,
+      "step": 338
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7782608695652174e-05,
+      "loss": 1.0164,
+      "step": 339
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7774936061381072e-05,
+      "loss": 1.1321,
+      "step": 340
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7767263427109977e-05,
+      "loss": 0.592,
+      "step": 341
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7759590792838875e-05,
+      "loss": 1.0191,
+      "step": 342
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7751918158567777e-05,
+      "loss": 0.9199,
+      "step": 343
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7744245524296675e-05,
+      "loss": 0.2874,
+      "step": 344
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7736572890025576e-05,
+      "loss": 0.2469,
+      "step": 345
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7728900255754478e-05,
+      "loss": 0.8364,
+      "step": 346
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7721227621483376e-05,
+      "loss": 0.7365,
+      "step": 347
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7713554987212278e-05,
+      "loss": 0.5595,
+      "step": 348
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.770588235294118e-05,
+      "loss": 0.2546,
+      "step": 349
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7698209718670078e-05,
+      "loss": 0.8183,
+      "step": 350
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7690537084398976e-05,
+      "loss": 0.5385,
+      "step": 351
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.768286445012788e-05,
+      "loss": 0.2763,
+      "step": 352
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.767519181585678e-05,
+      "loss": 0.3045,
+      "step": 353
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7667519181585677e-05,
+      "loss": 0.3354,
+      "step": 354
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765984654731458e-05,
+      "loss": 0.9366,
+      "step": 355
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765217391304348e-05,
+      "loss": 0.6212,
+      "step": 356
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.764450127877238e-05,
+      "loss": 0.1161,
+      "step": 357
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.763682864450128e-05,
+      "loss": 0.4929,
+      "step": 358
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762915601023018e-05,
+      "loss": 0.8167,
+      "step": 359
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762148337595908e-05,
+      "loss": 0.2521,
+      "step": 360
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.7613810741687982e-05,
+      "loss": 0.4174,
+      "step": 361
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.760613810741688e-05,
+      "loss": 0.3822,
+      "step": 362
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.759846547314578e-05,
+      "loss": 0.5355,
+      "step": 363
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.7590792838874683e-05,
+      "loss": 0.561,
+      "step": 364
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.758312020460358e-05,
+      "loss": 0.5126,
+      "step": 365
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.757544757033248e-05,
+      "loss": 0.2301,
+      "step": 366
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.756777493606138e-05,
+      "loss": 0.1698,
+      "step": 367
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7560102301790283e-05,
+      "loss": 0.1634,
+      "step": 368
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.755242966751918e-05,
+      "loss": 0.5955,
+      "step": 369
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7544757033248083e-05,
+      "loss": 0.1568,
+      "step": 370
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.753708439897698e-05,
+      "loss": 0.5167,
+      "step": 371
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7529411764705883e-05,
+      "loss": 1.0028,
+      "step": 372
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7521739130434784e-05,
+      "loss": 1.2567,
+      "step": 373
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7514066496163682e-05,
+      "loss": 0.6094,
+      "step": 374
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7506393861892584e-05,
+      "loss": 0.6059,
+      "step": 375
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7498721227621486e-05,
+      "loss": 1.0593,
+      "step": 376
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7491048593350384e-05,
+      "loss": 0.1506,
+      "step": 377
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7483375959079282e-05,
+      "loss": 1.5106,
+      "step": 378
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7475703324808187e-05,
+      "loss": 0.1827,
+      "step": 379
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7468030690537085e-05,
+      "loss": 1.4961,
+      "step": 380
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7460358056265984e-05,
+      "loss": 1.0456,
+      "step": 381
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7452685421994885e-05,
+      "loss": 1.2063,
+      "step": 382
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7445012787723787e-05,
+      "loss": 0.5837,
+      "step": 383
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7437340153452685e-05,
+      "loss": 1.4536,
+      "step": 384
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7429667519181587e-05,
+      "loss": 0.5803,
+      "step": 385
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7421994884910485e-05,
+      "loss": 0.5896,
+      "step": 386
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7414322250639386e-05,
+      "loss": 0.1822,
+      "step": 387
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7406649616368288e-05,
+      "loss": 1.7893,
+      "step": 388
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7398976982097186e-05,
+      "loss": 0.9723,
+      "step": 389
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7391304347826085e-05,
+      "loss": 1.2815,
+      "step": 390
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.738363171355499e-05,
+      "loss": 0.5765,
+      "step": 391
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7375959079283888e-05,
+      "loss": 1.2429,
+      "step": 392
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7368286445012786e-05,
+      "loss": 0.1343,
+      "step": 393
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7360613810741688e-05,
+      "loss": 0.2306,
+      "step": 394
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.735294117647059e-05,
+      "loss": 0.624,
+      "step": 395
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.734526854219949e-05,
+      "loss": 0.2371,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7348484848484849,
+      "eval_loss": 0.5804409980773926,
+      "eval_runtime": 57.3336,
+      "eval_samples_per_second": 6.907,
+      "eval_steps_per_second": 1.727,
+      "step": 396
+    }
+  ],
+  "max_steps": 3960,
+  "num_train_epochs": 10,
+  "total_flos": 1.437147546912e+17,
+  "trial_name": null,
+  "trial_params": null
+}
diff --git a/checkpoint-396/training_args.bin b/checkpoint-396/training_args.bin
new file mode 100644
index 0000000000000000000000000000000000000000..32901f899c91066c808eae9dd019400bfdd17498
--- /dev/null
+++ b/checkpoint-396/training_args.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db14439c369bfe9c89f64ff38f0b2d3fcc53911fc2c0b12a3a7d04656e7384a
+size 3503
diff --git a/checkpoint-3960/config.json b/checkpoint-3960/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..cb8d52e14263c50fd53dca42b5050e3235115c32
--- /dev/null
+++ b/checkpoint-3960/config.json
@@ -0,0 +1,113 @@
+{
+  "_name_or_path": "facebook/wav2vec2-base-960h",
+  "activation_dropout": 0.1,
+  "adapter_kernel_size": 3,
+  "adapter_stride": 2,
+  "add_adapter": false,
+  "apply_spec_augment": true,
+  "architectures": [
+    "Wav2Vec2ForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 1,
+  "classifier_proj_size": 256,
+  "codevector_dim": 256,
+  "contrastive_logits_temperature": 0.1,
+  "conv_bias": false,
+  "conv_dim": [
+    512,
+    512,
+    512,
+    512,
+    512,
+    512,
+    512
+  ],
+  "conv_kernel": [
+    10,
+    3,
+    3,
+    3,
+    3,
+    2,
+    2
+  ],
+  "conv_stride": [
+    5,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2
+  ],
+  "ctc_loss_reduction": "sum",
+  "ctc_zero_infinity": false,
+  "diversity_loss_weight": 0.1,
+  "do_stable_layer_norm": false,
+  "eos_token_id": 2,
+  "feat_extract_activation": "gelu",
+  "feat_extract_dropout": 0.0,
+  "feat_extract_norm": "group",
+  "feat_proj_dropout": 0.1,
+  "feat_quantizer_dropout": 0.0,
+  "final_dropout": 0.1,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout": 0.1,
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Foreign",
+    "1": "Native"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "layerdrop": 0.1,
+  "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
+  "mask_feature_prob": 0.0,
+  "mask_time_length": 10,
+  "mask_time_min_masks": 2,
+  "mask_time_prob": 0.05,
+  "model_type": "wav2vec2",
+  "num_adapter_layers": 3,
+  "num_attention_heads": 12,
+  "num_codevector_groups": 2,
+  "num_codevectors_per_group": 320,
+  "num_conv_pos_embedding_groups": 16,
+  "num_conv_pos_embeddings": 128,
+  "num_feat_extract_layers": 7,
+  "num_hidden_layers": 12,
+  "num_negatives": 100,
+  "output_hidden_size": 768,
+  "pad_token_id": 0,
+  "proj_codevector_dim": 256,
+  "tdnn_dilation": [
+    1,
+    2,
+    3,
+    1,
+    1
+  ],
+  "tdnn_dim": [
+    512,
+    512,
+    512,
+    512,
+    1500
+  ],
+  "tdnn_kernel": [
+    5,
+    3,
+    3,
+    1,
+    1
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.28.1",
+  "use_weighted_layer_sum": false,
+  "vocab_size": 32,
+  "xvector_output_dim": 512
+}
diff --git a/checkpoint-3960/optimizer.pt b/checkpoint-3960/optimizer.pt
new file mode 100644
index 0000000000000000000000000000000000000000..b8e872d5e89c462b45d945a567aa88c7cd2261c5
--- /dev/null
+++ b/checkpoint-3960/optimizer.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ca294facb628f80e1804f516942d9e38d9235ccab313be5725b0f278e11924fe
+size 756677477
diff --git a/checkpoint-3960/preprocessor_config.json b/checkpoint-3960/preprocessor_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..a0b7227fc1d916e469b14f6c154ad6dfea1e6891
--- /dev/null
+++ b/checkpoint-3960/preprocessor_config.json
@@ -0,0 +1,9 @@
+{
+  "do_normalize": true,
+  "feature_extractor_type": "Wav2Vec2FeatureExtractor",
+  "feature_size": 1,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": false,
+  "sampling_rate": 16000
+}
diff --git a/checkpoint-3960/pytorch_model.bin b/checkpoint-3960/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..e289a881a8bf4c7cb4994eecdc4f606ff65cf609
--- /dev/null
+++ b/checkpoint-3960/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1e0b340e7c20045e2d3ff9865a6d73c3381c46db3b8aba8c216dca05aea30852
+size 378348563
diff --git a/checkpoint-3960/rng_state.pth b/checkpoint-3960/rng_state.pth
new file mode 100644
index 0000000000000000000000000000000000000000..04e9ccb4c0f6eef292b9dafdfa0f85ae042e2e9f
--- /dev/null
+++ b/checkpoint-3960/rng_state.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa474d4e8fb7b47025c11527a901e64b5a17ec12e1dc6fc599a99d66d5016a9e
+size 14567
diff --git a/checkpoint-3960/scheduler.pt b/checkpoint-3960/scheduler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..faf40a30978543549d0ab71108c237fe93151c97
--- /dev/null
+++ b/checkpoint-3960/scheduler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e84fd5fe3fb5a73ac4b7d72352490ea5cbb5727ab2aee6702ff93331179a638b
+size 623
diff --git a/checkpoint-3960/trainer_state.json b/checkpoint-3960/trainer_state.json
new file mode 100644
index 0000000000000000000000000000000000000000..b8854d0814402119e3f359cabf504560eeea4766
--- /dev/null
+++ b/checkpoint-3960/trainer_state.json
@@ -0,0 +1,23866 @@
+{
+  "best_metric": 0.3489837944507599,
+  "best_model_checkpoint": "accent_classification/checkpoint-2376",
+  "epoch": 10.0,
+  "global_step": 3960,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.000000000000001e-07,
+      "loss": 0.6956,
+      "step": 1
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.2000000000000002e-06,
+      "loss": 0.6979,
+      "step": 2
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8e-06,
+      "loss": 0.7014,
+      "step": 3
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 2.4000000000000003e-06,
+      "loss": 0.6984,
+      "step": 4
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 3e-06,
+      "loss": 0.6978,
+      "step": 5
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 3.6e-06,
+      "loss": 0.6783,
+      "step": 6
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.2000000000000004e-06,
+      "loss": 0.6956,
+      "step": 7
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.800000000000001e-06,
+      "loss": 0.6994,
+      "step": 8
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 5.4e-06,
+      "loss": 0.6902,
+      "step": 9
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6e-06,
+      "loss": 0.6937,
+      "step": 10
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6.6e-06,
+      "loss": 0.6497,
+      "step": 11
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.2e-06,
+      "loss": 0.6911,
+      "step": 12
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.8e-06,
+      "loss": 0.6904,
+      "step": 13
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 8.400000000000001e-06,
+      "loss": 0.6854,
+      "step": 14
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9e-06,
+      "loss": 0.6843,
+      "step": 15
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9.600000000000001e-06,
+      "loss": 0.6851,
+      "step": 16
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.02e-05,
+      "loss": 0.6733,
+      "step": 17
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.08e-05,
+      "loss": 0.6819,
+      "step": 18
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.1400000000000001e-05,
+      "loss": 0.681,
+      "step": 19
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.6998,
+      "step": 20
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.26e-05,
+      "loss": 0.6743,
+      "step": 21
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.32e-05,
+      "loss": 0.6818,
+      "step": 22
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3800000000000002e-05,
+      "loss": 0.5331,
+      "step": 23
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.44e-05,
+      "loss": 0.6681,
+      "step": 24
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.6955,
+      "step": 25
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.56e-05,
+      "loss": 0.7,
+      "step": 26
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.62e-05,
+      "loss": 0.6975,
+      "step": 27
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.6800000000000002e-05,
+      "loss": 0.6249,
+      "step": 28
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.74e-05,
+      "loss": 0.6979,
+      "step": 29
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.6076,
+      "step": 30
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.86e-05,
+      "loss": 0.6091,
+      "step": 31
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9200000000000003e-05,
+      "loss": 0.6438,
+      "step": 32
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.98e-05,
+      "loss": 0.6624,
+      "step": 33
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.04e-05,
+      "loss": 0.6452,
+      "step": 34
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.6963,
+      "step": 35
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.16e-05,
+      "loss": 0.758,
+      "step": 36
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.22e-05,
+      "loss": 0.6471,
+      "step": 37
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.2800000000000002e-05,
+      "loss": 0.6313,
+      "step": 38
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.3400000000000003e-05,
+      "loss": 0.7019,
+      "step": 39
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.6336,
+      "step": 40
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4599999999999998e-05,
+      "loss": 0.7091,
+      "step": 41
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.52e-05,
+      "loss": 0.701,
+      "step": 42
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.58e-05,
+      "loss": 0.7036,
+      "step": 43
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.64e-05,
+      "loss": 0.7073,
+      "step": 44
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.614,
+      "step": 45
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.7600000000000003e-05,
+      "loss": 0.7543,
+      "step": 46
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.8199999999999998e-05,
+      "loss": 0.538,
+      "step": 47
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.88e-05,
+      "loss": 0.5179,
+      "step": 48
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.94e-05,
+      "loss": 0.6164,
+      "step": 49
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 3e-05,
+      "loss": 0.6093,
+      "step": 50
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.99923273657289e-05,
+      "loss": 0.6983,
+      "step": 51
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9984654731457804e-05,
+      "loss": 0.6069,
+      "step": 52
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9976982097186702e-05,
+      "loss": 0.5933,
+      "step": 53
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.99693094629156e-05,
+      "loss": 0.552,
+      "step": 54
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9961636828644502e-05,
+      "loss": 0.5865,
+      "step": 55
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9953964194373404e-05,
+      "loss": 0.7146,
+      "step": 56
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9946291560102302e-05,
+      "loss": 0.5591,
+      "step": 57
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9938618925831203e-05,
+      "loss": 0.7366,
+      "step": 58
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9930946291560102e-05,
+      "loss": 0.7006,
+      "step": 59
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9923273657289003e-05,
+      "loss": 0.4148,
+      "step": 60
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9915601023017905e-05,
+      "loss": 0.9938,
+      "step": 61
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9907928388746803e-05,
+      "loss": 0.911,
+      "step": 62
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.99002557544757e-05,
+      "loss": 0.5682,
+      "step": 63
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9892583120204606e-05,
+      "loss": 0.5833,
+      "step": 64
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9884910485933505e-05,
+      "loss": 0.7237,
+      "step": 65
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9877237851662403e-05,
+      "loss": 0.4013,
+      "step": 66
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9869565217391304e-05,
+      "loss": 0.9695,
+      "step": 67
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9861892583120206e-05,
+      "loss": 0.568,
+      "step": 68
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9854219948849104e-05,
+      "loss": 0.5697,
+      "step": 69
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9846547314578006e-05,
+      "loss": 0.7675,
+      "step": 70
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9838874680306904e-05,
+      "loss": 0.8831,
+      "step": 71
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9831202046035806e-05,
+      "loss": 0.7008,
+      "step": 72
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9823529411764707e-05,
+      "loss": 0.7085,
+      "step": 73
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9815856777493606e-05,
+      "loss": 0.5374,
+      "step": 74
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9808184143222507e-05,
+      "loss": 0.4131,
+      "step": 75
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.980051150895141e-05,
+      "loss": 0.7309,
+      "step": 76
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9792838874680307e-05,
+      "loss": 0.4279,
+      "step": 77
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.978516624040921e-05,
+      "loss": 0.5813,
+      "step": 78
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.977749360613811e-05,
+      "loss": 0.6276,
+      "step": 79
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976982097186701e-05,
+      "loss": 0.7892,
+      "step": 80
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976214833759591e-05,
+      "loss": 0.7134,
+      "step": 81
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.9754475703324808e-05,
+      "loss": 0.3682,
+      "step": 82
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.974680306905371e-05,
+      "loss": 0.5562,
+      "step": 83
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973913043478261e-05,
+      "loss": 0.7158,
+      "step": 84
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973145780051151e-05,
+      "loss": 0.524,
+      "step": 85
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9723785166240408e-05,
+      "loss": 0.3372,
+      "step": 86
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9716112531969313e-05,
+      "loss": 0.4979,
+      "step": 87
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970843989769821e-05,
+      "loss": 0.3177,
+      "step": 88
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970076726342711e-05,
+      "loss": 0.3488,
+      "step": 89
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.969309462915601e-05,
+      "loss": 0.5158,
+      "step": 90
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9685421994884913e-05,
+      "loss": 0.3208,
+      "step": 91
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.967774936061381e-05,
+      "loss": 0.6654,
+      "step": 92
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9670076726342712e-05,
+      "loss": 0.5581,
+      "step": 93
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.966240409207161e-05,
+      "loss": 0.7456,
+      "step": 94
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9654731457800512e-05,
+      "loss": 0.4437,
+      "step": 95
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9647058823529414e-05,
+      "loss": 0.5525,
+      "step": 96
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9639386189258312e-05,
+      "loss": 0.6673,
+      "step": 97
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.963171355498721e-05,
+      "loss": 0.435,
+      "step": 98
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9624040920716115e-05,
+      "loss": 0.7415,
+      "step": 99
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9616368286445014e-05,
+      "loss": 0.2981,
+      "step": 100
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9608695652173912e-05,
+      "loss": 0.5179,
+      "step": 101
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9601023017902817e-05,
+      "loss": 0.4005,
+      "step": 102
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9593350383631715e-05,
+      "loss": 0.4598,
+      "step": 103
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9585677749360613e-05,
+      "loss": 0.2553,
+      "step": 104
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9578005115089515e-05,
+      "loss": 0.4302,
+      "step": 105
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9570332480818416e-05,
+      "loss": 0.7894,
+      "step": 106
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9562659846547315e-05,
+      "loss": 0.213,
+      "step": 107
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9554987212276216e-05,
+      "loss": 0.5507,
+      "step": 108
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9547314578005115e-05,
+      "loss": 0.7676,
+      "step": 109
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9539641943734016e-05,
+      "loss": 0.3418,
+      "step": 110
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9531969309462918e-05,
+      "loss": 0.8533,
+      "step": 111
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9524296675191816e-05,
+      "loss": 0.3659,
+      "step": 112
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9516624040920714e-05,
+      "loss": 0.6482,
+      "step": 113
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.950895140664962e-05,
+      "loss": 0.5484,
+      "step": 114
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9501278772378517e-05,
+      "loss": 0.4508,
+      "step": 115
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9493606138107416e-05,
+      "loss": 0.4401,
+      "step": 116
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9485933503836317e-05,
+      "loss": 0.5455,
+      "step": 117
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.947826086956522e-05,
+      "loss": 0.722,
+      "step": 118
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9470588235294117e-05,
+      "loss": 0.2874,
+      "step": 119
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.946291560102302e-05,
+      "loss": 0.5973,
+      "step": 120
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.9455242966751917e-05,
+      "loss": 0.5898,
+      "step": 121
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.944757033248082e-05,
+      "loss": 1.0125,
+      "step": 122
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943989769820972e-05,
+      "loss": 0.2511,
+      "step": 123
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943222506393862e-05,
+      "loss": 0.545,
+      "step": 124
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.942455242966752e-05,
+      "loss": 0.593,
+      "step": 125
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.941687979539642e-05,
+      "loss": 0.2319,
+      "step": 126
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.940920716112532e-05,
+      "loss": 0.2215,
+      "step": 127
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.9401534526854218e-05,
+      "loss": 0.1964,
+      "step": 128
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9393861892583123e-05,
+      "loss": 0.1471,
+      "step": 129
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.938618925831202e-05,
+      "loss": 1.5965,
+      "step": 130
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9378516624040923e-05,
+      "loss": 1.0772,
+      "step": 131
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.937084398976982e-05,
+      "loss": 0.2005,
+      "step": 132
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9363171355498723e-05,
+      "loss": 0.9628,
+      "step": 133
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9355498721227624e-05,
+      "loss": 0.1503,
+      "step": 134
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9347826086956523e-05,
+      "loss": 0.6201,
+      "step": 135
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.934015345268542e-05,
+      "loss": 0.6094,
+      "step": 136
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9332480818414326e-05,
+      "loss": 1.385,
+      "step": 137
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9324808184143224e-05,
+      "loss": 0.1552,
+      "step": 138
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9317135549872122e-05,
+      "loss": 0.5828,
+      "step": 139
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9309462915601024e-05,
+      "loss": 0.6018,
+      "step": 140
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9301790281329925e-05,
+      "loss": 1.0614,
+      "step": 141
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9294117647058824e-05,
+      "loss": 0.566,
+      "step": 142
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9286445012787725e-05,
+      "loss": 0.5958,
+      "step": 143
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9278772378516624e-05,
+      "loss": 0.5838,
+      "step": 144
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9271099744245525e-05,
+      "loss": 0.5909,
+      "step": 145
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9263427109974427e-05,
+      "loss": 0.5862,
+      "step": 146
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9255754475703325e-05,
+      "loss": 0.2031,
+      "step": 147
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9248081841432223e-05,
+      "loss": 0.1995,
+      "step": 148
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9240409207161128e-05,
+      "loss": 0.24,
+      "step": 149
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9232736572890026e-05,
+      "loss": 0.5845,
+      "step": 150
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9225063938618925e-05,
+      "loss": 0.964,
+      "step": 151
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.921739130434783e-05,
+      "loss": 0.9595,
+      "step": 152
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9209718670076728e-05,
+      "loss": 0.2033,
+      "step": 153
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9202046035805626e-05,
+      "loss": 0.213,
+      "step": 154
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9194373401534528e-05,
+      "loss": 0.2042,
+      "step": 155
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.918670076726343e-05,
+      "loss": 0.5783,
+      "step": 156
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9179028132992328e-05,
+      "loss": 1.0822,
+      "step": 157
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.917135549872123e-05,
+      "loss": 1.2643,
+      "step": 158
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9163682864450127e-05,
+      "loss": 0.2593,
+      "step": 159
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.915601023017903e-05,
+      "loss": 0.5783,
+      "step": 160
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914833759590793e-05,
+      "loss": 0.5721,
+      "step": 161
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914066496163683e-05,
+      "loss": 0.5808,
+      "step": 162
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9132992327365727e-05,
+      "loss": 1.2578,
+      "step": 163
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9125319693094632e-05,
+      "loss": 0.5709,
+      "step": 164
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.911764705882353e-05,
+      "loss": 0.2159,
+      "step": 165
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910997442455243e-05,
+      "loss": 0.2329,
+      "step": 166
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910230179028133e-05,
+      "loss": 0.926,
+      "step": 167
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.9094629156010232e-05,
+      "loss": 0.9095,
+      "step": 168
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.908695652173913e-05,
+      "loss": 0.5622,
+      "step": 169
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907928388746803e-05,
+      "loss": 0.5703,
+      "step": 170
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907161125319693e-05,
+      "loss": 0.5669,
+      "step": 171
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.906393861892583e-05,
+      "loss": 0.2637,
+      "step": 172
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9056265984654733e-05,
+      "loss": 1.1077,
+      "step": 173
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.904859335038363e-05,
+      "loss": 0.5653,
+      "step": 174
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9040920716112533e-05,
+      "loss": 0.5584,
+      "step": 175
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9033248081841434e-05,
+      "loss": 0.5544,
+      "step": 176
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9025575447570333e-05,
+      "loss": 0.5663,
+      "step": 177
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.901790281329923e-05,
+      "loss": 1.0808,
+      "step": 178
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9010230179028136e-05,
+      "loss": 0.8736,
+      "step": 179
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9002557544757034e-05,
+      "loss": 0.6435,
+      "step": 180
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8994884910485932e-05,
+      "loss": 0.786,
+      "step": 181
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8987212276214834e-05,
+      "loss": 0.5717,
+      "step": 182
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8979539641943736e-05,
+      "loss": 0.5618,
+      "step": 183
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8971867007672637e-05,
+      "loss": 0.5785,
+      "step": 184
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8964194373401535e-05,
+      "loss": 0.7601,
+      "step": 185
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8956521739130434e-05,
+      "loss": 0.8112,
+      "step": 186
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.894884910485934e-05,
+      "loss": 0.5663,
+      "step": 187
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8941176470588237e-05,
+      "loss": 0.5631,
+      "step": 188
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8933503836317135e-05,
+      "loss": 0.7363,
+      "step": 189
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8925831202046037e-05,
+      "loss": 0.4552,
+      "step": 190
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8918158567774938e-05,
+      "loss": 0.4599,
+      "step": 191
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8910485933503836e-05,
+      "loss": 0.7848,
+      "step": 192
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8902813299232738e-05,
+      "loss": 0.5907,
+      "step": 193
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8895140664961636e-05,
+      "loss": 0.4264,
+      "step": 194
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8887468030690538e-05,
+      "loss": 0.5697,
+      "step": 195
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.887979539641944e-05,
+      "loss": 0.7643,
+      "step": 196
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8872122762148338e-05,
+      "loss": 0.3635,
+      "step": 197
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8864450127877236e-05,
+      "loss": 0.3715,
+      "step": 198
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.885677749360614e-05,
+      "loss": 0.554,
+      "step": 199
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.884910485933504e-05,
+      "loss": 0.8044,
+      "step": 200
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8841432225063937e-05,
+      "loss": 0.5673,
+      "step": 201
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8833759590792842e-05,
+      "loss": 0.8042,
+      "step": 202
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.882608695652174e-05,
+      "loss": 0.5638,
+      "step": 203
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881841432225064e-05,
+      "loss": 0.3584,
+      "step": 204
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881074168797954e-05,
+      "loss": 0.8211,
+      "step": 205
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.8803069053708442e-05,
+      "loss": 0.5708,
+      "step": 206
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.879539641943734e-05,
+      "loss": 0.8301,
+      "step": 207
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8787723785166242e-05,
+      "loss": 0.8229,
+      "step": 208
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.878005115089514e-05,
+      "loss": 0.5645,
+      "step": 209
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8772378516624042e-05,
+      "loss": 0.3529,
+      "step": 210
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8764705882352943e-05,
+      "loss": 0.5578,
+      "step": 211
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.875703324808184e-05,
+      "loss": 0.5659,
+      "step": 212
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.874936061381074e-05,
+      "loss": 0.7495,
+      "step": 213
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8741687979539645e-05,
+      "loss": 0.3162,
+      "step": 214
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8734015345268543e-05,
+      "loss": 0.5616,
+      "step": 215
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.872634271099744e-05,
+      "loss": 0.4034,
+      "step": 216
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8718670076726343e-05,
+      "loss": 0.3163,
+      "step": 217
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8710997442455245e-05,
+      "loss": 0.5514,
+      "step": 218
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8703324808184143e-05,
+      "loss": 0.826,
+      "step": 219
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8695652173913044e-05,
+      "loss": 0.8363,
+      "step": 220
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8687979539641943e-05,
+      "loss": 0.8288,
+      "step": 221
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8680306905370844e-05,
+      "loss": 0.5929,
+      "step": 222
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8672634271099746e-05,
+      "loss": 0.565,
+      "step": 223
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8664961636828644e-05,
+      "loss": 0.5642,
+      "step": 224
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8657289002557546e-05,
+      "loss": 0.5609,
+      "step": 225
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8649616368286447e-05,
+      "loss": 0.5641,
+      "step": 226
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8641943734015345e-05,
+      "loss": 0.8535,
+      "step": 227
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8634271099744244e-05,
+      "loss": 0.5652,
+      "step": 228
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.862659846547315e-05,
+      "loss": 0.4397,
+      "step": 229
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8618925831202047e-05,
+      "loss": 0.567,
+      "step": 230
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8611253196930945e-05,
+      "loss": 0.7517,
+      "step": 231
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8603580562659847e-05,
+      "loss": 0.5839,
+      "step": 232
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.859590792838875e-05,
+      "loss": 0.5677,
+      "step": 233
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8588235294117647e-05,
+      "loss": 0.5678,
+      "step": 234
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8580562659846548e-05,
+      "loss": 1.3045,
+      "step": 235
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8572890025575446e-05,
+      "loss": 0.804,
+      "step": 236
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.856521739130435e-05,
+      "loss": 1.0303,
+      "step": 237
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.855754475703325e-05,
+      "loss": 0.3303,
+      "step": 238
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8549872122762148e-05,
+      "loss": 0.5677,
+      "step": 239
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.854219948849105e-05,
+      "loss": 0.7972,
+      "step": 240
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.853452685421995e-05,
+      "loss": 0.5644,
+      "step": 241
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.852685421994885e-05,
+      "loss": 0.9294,
+      "step": 242
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.851918158567775e-05,
+      "loss": 0.5665,
+      "step": 243
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.851150895140665e-05,
+      "loss": 0.7963,
+      "step": 244
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.850383631713555e-05,
+      "loss": 0.5659,
+      "step": 245
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.8496163682864452e-05,
+      "loss": 0.5718,
+      "step": 246
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.848849104859335e-05,
+      "loss": 0.3526,
+      "step": 247
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8480818414322252e-05,
+      "loss": 0.574,
+      "step": 248
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8473145780051154e-05,
+      "loss": 0.9996,
+      "step": 249
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8465473145780052e-05,
+      "loss": 0.7634,
+      "step": 250
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.845780051150895e-05,
+      "loss": 0.5639,
+      "step": 251
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8450127877237855e-05,
+      "loss": 0.5737,
+      "step": 252
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8442455242966753e-05,
+      "loss": 0.7793,
+      "step": 253
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8434782608695652e-05,
+      "loss": 0.6769,
+      "step": 254
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8427109974424553e-05,
+      "loss": 0.5748,
+      "step": 255
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8419437340153455e-05,
+      "loss": 0.3602,
+      "step": 256
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8411764705882353e-05,
+      "loss": 0.5714,
+      "step": 257
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8404092071611255e-05,
+      "loss": 0.5738,
+      "step": 258
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8396419437340153e-05,
+      "loss": 0.5719,
+      "step": 259
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8388746803069055e-05,
+      "loss": 0.5753,
+      "step": 260
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8381074168797956e-05,
+      "loss": 0.5717,
+      "step": 261
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8373401534526854e-05,
+      "loss": 0.9794,
+      "step": 262
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8365728900255753e-05,
+      "loss": 0.5737,
+      "step": 263
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8358056265984658e-05,
+      "loss": 0.5723,
+      "step": 264
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8350383631713556e-05,
+      "loss": 0.5656,
+      "step": 265
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8342710997442454e-05,
+      "loss": 0.4259,
+      "step": 266
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8335038363171356e-05,
+      "loss": 0.3076,
+      "step": 267
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8327365728900257e-05,
+      "loss": 0.338,
+      "step": 268
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8319693094629156e-05,
+      "loss": 0.5612,
+      "step": 269
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8312020460358057e-05,
+      "loss": 0.5249,
+      "step": 270
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8304347826086955e-05,
+      "loss": 0.5671,
+      "step": 271
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8296675191815857e-05,
+      "loss": 0.5574,
+      "step": 272
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.828900255754476e-05,
+      "loss": 0.8152,
+      "step": 273
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8281329923273657e-05,
+      "loss": 0.5647,
+      "step": 274
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.827365728900256e-05,
+      "loss": 0.5632,
+      "step": 275
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.826598465473146e-05,
+      "loss": 0.3391,
+      "step": 276
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8258312020460358e-05,
+      "loss": 0.55,
+      "step": 277
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8250639386189257e-05,
+      "loss": 0.5672,
+      "step": 278
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.824296675191816e-05,
+      "loss": 0.285,
+      "step": 279
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.823529411764706e-05,
+      "loss": 0.2809,
+      "step": 280
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.8227621483375958e-05,
+      "loss": 0.5552,
+      "step": 281
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821994884910486e-05,
+      "loss": 0.563,
+      "step": 282
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821227621483376e-05,
+      "loss": 0.5628,
+      "step": 283
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.820460358056266e-05,
+      "loss": 0.2548,
+      "step": 284
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.819693094629156e-05,
+      "loss": 0.871,
+      "step": 285
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818925831202046e-05,
+      "loss": 0.8797,
+      "step": 286
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818158567774936e-05,
+      "loss": 0.2453,
+      "step": 287
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8173913043478262e-05,
+      "loss": 0.2464,
+      "step": 288
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.816624040920716e-05,
+      "loss": 0.3134,
+      "step": 289
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8158567774936062e-05,
+      "loss": 0.2431,
+      "step": 290
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8150895140664964e-05,
+      "loss": 0.2308,
+      "step": 291
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8143222506393862e-05,
+      "loss": 0.2173,
+      "step": 292
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8135549872122764e-05,
+      "loss": 0.5665,
+      "step": 293
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8127877237851662e-05,
+      "loss": 0.5592,
+      "step": 294
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8120204603580564e-05,
+      "loss": 0.473,
+      "step": 295
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8112531969309465e-05,
+      "loss": 0.9458,
+      "step": 296
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8104859335038363e-05,
+      "loss": 0.1879,
+      "step": 297
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8097186700767265e-05,
+      "loss": 0.6797,
+      "step": 298
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8089514066496167e-05,
+      "loss": 0.7592,
+      "step": 299
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8081841432225065e-05,
+      "loss": 0.4109,
+      "step": 300
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8074168797953963e-05,
+      "loss": 0.5821,
+      "step": 301
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8066496163682868e-05,
+      "loss": 0.2057,
+      "step": 302
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8058823529411766e-05,
+      "loss": 0.5322,
+      "step": 303
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8051150895140665e-05,
+      "loss": 0.5208,
+      "step": 304
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8043478260869566e-05,
+      "loss": 0.6799,
+      "step": 305
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8035805626598468e-05,
+      "loss": 0.5894,
+      "step": 306
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8028132992327366e-05,
+      "loss": 0.4051,
+      "step": 307
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8020460358056268e-05,
+      "loss": 0.3695,
+      "step": 308
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8012787723785166e-05,
+      "loss": 0.4869,
+      "step": 309
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8005115089514067e-05,
+      "loss": 0.6075,
+      "step": 310
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.799744245524297e-05,
+      "loss": 0.4245,
+      "step": 311
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7989769820971867e-05,
+      "loss": 0.5122,
+      "step": 312
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7982097186700765e-05,
+      "loss": 0.452,
+      "step": 313
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.797442455242967e-05,
+      "loss": 0.9568,
+      "step": 314
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.796675191815857e-05,
+      "loss": 0.5846,
+      "step": 315
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.7959079283887467e-05,
+      "loss": 0.3743,
+      "step": 316
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.795140664961637e-05,
+      "loss": 0.1922,
+      "step": 317
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.794373401534527e-05,
+      "loss": 0.5303,
+      "step": 318
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.793606138107417e-05,
+      "loss": 0.7253,
+      "step": 319
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.792838874680307e-05,
+      "loss": 0.2794,
+      "step": 320
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.7920716112531968e-05,
+      "loss": 0.577,
+      "step": 321
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.791304347826087e-05,
+      "loss": 0.614,
+      "step": 322
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.790537084398977e-05,
+      "loss": 0.5882,
+      "step": 323
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789769820971867e-05,
+      "loss": 0.2503,
+      "step": 324
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789002557544757e-05,
+      "loss": 0.4195,
+      "step": 325
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.7882352941176473e-05,
+      "loss": 0.2959,
+      "step": 326
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.787468030690537e-05,
+      "loss": 0.3732,
+      "step": 327
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.786700767263427e-05,
+      "loss": 0.5729,
+      "step": 328
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7859335038363174e-05,
+      "loss": 0.3179,
+      "step": 329
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7851662404092073e-05,
+      "loss": 0.6823,
+      "step": 330
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.784398976982097e-05,
+      "loss": 0.2663,
+      "step": 331
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7836317135549872e-05,
+      "loss": 0.4436,
+      "step": 332
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7828644501278774e-05,
+      "loss": 0.3089,
+      "step": 333
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7820971867007672e-05,
+      "loss": 0.5888,
+      "step": 334
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7813299232736574e-05,
+      "loss": 0.2115,
+      "step": 335
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7805626598465472e-05,
+      "loss": 0.4005,
+      "step": 336
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7797953964194374e-05,
+      "loss": 0.3097,
+      "step": 337
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7790281329923275e-05,
+      "loss": 0.5847,
+      "step": 338
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7782608695652174e-05,
+      "loss": 1.0164,
+      "step": 339
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7774936061381072e-05,
+      "loss": 1.1321,
+      "step": 340
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7767263427109977e-05,
+      "loss": 0.592,
+      "step": 341
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7759590792838875e-05,
+      "loss": 1.0191,
+      "step": 342
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7751918158567777e-05,
+      "loss": 0.9199,
+      "step": 343
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7744245524296675e-05,
+      "loss": 0.2874,
+      "step": 344
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7736572890025576e-05,
+      "loss": 0.2469,
+      "step": 345
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7728900255754478e-05,
+      "loss": 0.8364,
+      "step": 346
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7721227621483376e-05,
+      "loss": 0.7365,
+      "step": 347
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7713554987212278e-05,
+      "loss": 0.5595,
+      "step": 348
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.770588235294118e-05,
+      "loss": 0.2546,
+      "step": 349
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7698209718670078e-05,
+      "loss": 0.8183,
+      "step": 350
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7690537084398976e-05,
+      "loss": 0.5385,
+      "step": 351
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.768286445012788e-05,
+      "loss": 0.2763,
+      "step": 352
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.767519181585678e-05,
+      "loss": 0.3045,
+      "step": 353
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7667519181585677e-05,
+      "loss": 0.3354,
+      "step": 354
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765984654731458e-05,
+      "loss": 0.9366,
+      "step": 355
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765217391304348e-05,
+      "loss": 0.6212,
+      "step": 356
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.764450127877238e-05,
+      "loss": 0.1161,
+      "step": 357
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.763682864450128e-05,
+      "loss": 0.4929,
+      "step": 358
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762915601023018e-05,
+      "loss": 0.8167,
+      "step": 359
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762148337595908e-05,
+      "loss": 0.2521,
+      "step": 360
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.7613810741687982e-05,
+      "loss": 0.4174,
+      "step": 361
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.760613810741688e-05,
+      "loss": 0.3822,
+      "step": 362
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.759846547314578e-05,
+      "loss": 0.5355,
+      "step": 363
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.7590792838874683e-05,
+      "loss": 0.561,
+      "step": 364
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.758312020460358e-05,
+      "loss": 0.5126,
+      "step": 365
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.757544757033248e-05,
+      "loss": 0.2301,
+      "step": 366
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.756777493606138e-05,
+      "loss": 0.1698,
+      "step": 367
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7560102301790283e-05,
+      "loss": 0.1634,
+      "step": 368
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.755242966751918e-05,
+      "loss": 0.5955,
+      "step": 369
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7544757033248083e-05,
+      "loss": 0.1568,
+      "step": 370
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.753708439897698e-05,
+      "loss": 0.5167,
+      "step": 371
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7529411764705883e-05,
+      "loss": 1.0028,
+      "step": 372
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7521739130434784e-05,
+      "loss": 1.2567,
+      "step": 373
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7514066496163682e-05,
+      "loss": 0.6094,
+      "step": 374
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7506393861892584e-05,
+      "loss": 0.6059,
+      "step": 375
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7498721227621486e-05,
+      "loss": 1.0593,
+      "step": 376
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7491048593350384e-05,
+      "loss": 0.1506,
+      "step": 377
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7483375959079282e-05,
+      "loss": 1.5106,
+      "step": 378
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7475703324808187e-05,
+      "loss": 0.1827,
+      "step": 379
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7468030690537085e-05,
+      "loss": 1.4961,
+      "step": 380
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7460358056265984e-05,
+      "loss": 1.0456,
+      "step": 381
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7452685421994885e-05,
+      "loss": 1.2063,
+      "step": 382
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7445012787723787e-05,
+      "loss": 0.5837,
+      "step": 383
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7437340153452685e-05,
+      "loss": 1.4536,
+      "step": 384
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7429667519181587e-05,
+      "loss": 0.5803,
+      "step": 385
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7421994884910485e-05,
+      "loss": 0.5896,
+      "step": 386
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7414322250639386e-05,
+      "loss": 0.1822,
+      "step": 387
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7406649616368288e-05,
+      "loss": 1.7893,
+      "step": 388
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7398976982097186e-05,
+      "loss": 0.9723,
+      "step": 389
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7391304347826085e-05,
+      "loss": 1.2815,
+      "step": 390
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.738363171355499e-05,
+      "loss": 0.5765,
+      "step": 391
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7375959079283888e-05,
+      "loss": 1.2429,
+      "step": 392
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7368286445012786e-05,
+      "loss": 0.1343,
+      "step": 393
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7360613810741688e-05,
+      "loss": 0.2306,
+      "step": 394
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.735294117647059e-05,
+      "loss": 0.624,
+      "step": 395
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.734526854219949e-05,
+      "loss": 0.2371,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7348484848484849,
+      "eval_loss": 0.5804409980773926,
+      "eval_runtime": 57.3336,
+      "eval_samples_per_second": 6.907,
+      "eval_steps_per_second": 1.727,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.733759590792839e-05,
+      "loss": 0.8989,
+      "step": 397
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.732992327365729e-05,
+      "loss": 0.5664,
+      "step": 398
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.7322250639386192e-05,
+      "loss": 0.8852,
+      "step": 399
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.731457800511509e-05,
+      "loss": 0.5661,
+      "step": 400
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.730690537084399e-05,
+      "loss": 0.5648,
+      "step": 401
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7299232736572894e-05,
+      "loss": 0.7333,
+      "step": 402
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7291560102301792e-05,
+      "loss": 0.3037,
+      "step": 403
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.728388746803069e-05,
+      "loss": 0.1463,
+      "step": 404
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7276214833759592e-05,
+      "loss": 0.2702,
+      "step": 405
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7268542199488493e-05,
+      "loss": 0.2693,
+      "step": 406
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.726086956521739e-05,
+      "loss": 0.5647,
+      "step": 407
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7253196930946293e-05,
+      "loss": 0.8507,
+      "step": 408
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.724552429667519e-05,
+      "loss": 0.8645,
+      "step": 409
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7237851662404093e-05,
+      "loss": 0.2649,
+      "step": 410
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7230179028132995e-05,
+      "loss": 0.6942,
+      "step": 411
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7222506393861893e-05,
+      "loss": 0.5656,
+      "step": 412
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.721483375959079e-05,
+      "loss": 0.5489,
+      "step": 413
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7207161125319696e-05,
+      "loss": 0.2652,
+      "step": 414
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7199488491048594e-05,
+      "loss": 0.8676,
+      "step": 415
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7191815856777493e-05,
+      "loss": 0.8621,
+      "step": 416
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7184143222506394e-05,
+      "loss": 0.2641,
+      "step": 417
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7176470588235296e-05,
+      "loss": 0.859,
+      "step": 418
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7168797953964194e-05,
+      "loss": 0.6492,
+      "step": 419
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7161125319693096e-05,
+      "loss": 1.1494,
+      "step": 420
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7153452685421994e-05,
+      "loss": 0.8278,
+      "step": 421
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7145780051150895e-05,
+      "loss": 0.764,
+      "step": 422
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7138107416879797e-05,
+      "loss": 0.564,
+      "step": 423
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7130434782608695e-05,
+      "loss": 0.8051,
+      "step": 424
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7122762148337597e-05,
+      "loss": 0.3052,
+      "step": 425
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.71150895140665e-05,
+      "loss": 0.2945,
+      "step": 426
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7107416879795397e-05,
+      "loss": 0.4118,
+      "step": 427
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7099744245524295e-05,
+      "loss": 0.559,
+      "step": 428
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.70920716112532e-05,
+      "loss": 0.5593,
+      "step": 429
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7084398976982098e-05,
+      "loss": 0.564,
+      "step": 430
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7076726342710996e-05,
+      "loss": 0.4454,
+      "step": 431
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7069053708439898e-05,
+      "loss": 0.4381,
+      "step": 432
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.70613810741688e-05,
+      "loss": 0.3718,
+      "step": 433
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7053708439897698e-05,
+      "loss": 0.6627,
+      "step": 434
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70460358056266e-05,
+      "loss": 0.4004,
+      "step": 435
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7038363171355498e-05,
+      "loss": 0.3855,
+      "step": 436
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70306905370844e-05,
+      "loss": 0.4799,
+      "step": 437
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70230179028133e-05,
+      "loss": 0.6714,
+      "step": 438
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70153452685422e-05,
+      "loss": 0.622,
+      "step": 439
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7007672634271097e-05,
+      "loss": 0.3916,
+      "step": 440
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.5487,
+      "step": 441
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69923273657289e-05,
+      "loss": 0.376,
+      "step": 442
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69846547314578e-05,
+      "loss": 0.3616,
+      "step": 443
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69769820971867e-05,
+      "loss": 0.8747,
+      "step": 444
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.6969309462915602e-05,
+      "loss": 0.283,
+      "step": 445
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.69616368286445e-05,
+      "loss": 0.3988,
+      "step": 446
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6953964194373402e-05,
+      "loss": 0.4569,
+      "step": 447
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6946291560102303e-05,
+      "loss": 0.2005,
+      "step": 448
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6938618925831205e-05,
+      "loss": 0.3087,
+      "step": 449
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6930946291560103e-05,
+      "loss": 0.5814,
+      "step": 450
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6923273657289e-05,
+      "loss": 0.2995,
+      "step": 451
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6915601023017907e-05,
+      "loss": 0.7806,
+      "step": 452
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6907928388746805e-05,
+      "loss": 0.2666,
+      "step": 453
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6900255754475703e-05,
+      "loss": 0.2571,
+      "step": 454
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6892583120204605e-05,
+      "loss": 0.5931,
+      "step": 455
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6884910485933506e-05,
+      "loss": 0.475,
+      "step": 456
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6877237851662404e-05,
+      "loss": 0.2372,
+      "step": 457
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6869565217391306e-05,
+      "loss": 0.1577,
+      "step": 458
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6861892583120204e-05,
+      "loss": 0.1953,
+      "step": 459
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6854219948849106e-05,
+      "loss": 0.1931,
+      "step": 460
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6846547314578008e-05,
+      "loss": 0.1829,
+      "step": 461
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6838874680306906e-05,
+      "loss": 0.1601,
+      "step": 462
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6831202046035804e-05,
+      "loss": 0.2505,
+      "step": 463
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.682352941176471e-05,
+      "loss": 0.181,
+      "step": 464
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6815856777493607e-05,
+      "loss": 0.5978,
+      "step": 465
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6808184143222505e-05,
+      "loss": 0.5458,
+      "step": 466
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6800511508951407e-05,
+      "loss": 0.1289,
+      "step": 467
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.679283887468031e-05,
+      "loss": 0.4383,
+      "step": 468
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6785166240409207e-05,
+      "loss": 0.5892,
+      "step": 469
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.677749360613811e-05,
+      "loss": 1.3565,
+      "step": 470
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.676982097186701e-05,
+      "loss": 0.7267,
+      "step": 471
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.6762148337595908e-05,
+      "loss": 0.1354,
+      "step": 472
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.675447570332481e-05,
+      "loss": 0.9291,
+      "step": 473
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.6746803069053708e-05,
+      "loss": 0.5265,
+      "step": 474
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673913043478261e-05,
+      "loss": 0.4841,
+      "step": 475
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673145780051151e-05,
+      "loss": 1.2391,
+      "step": 476
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.672378516624041e-05,
+      "loss": 0.4924,
+      "step": 477
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6716112531969308e-05,
+      "loss": 1.0076,
+      "step": 478
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6708439897698213e-05,
+      "loss": 0.6314,
+      "step": 479
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.670076726342711e-05,
+      "loss": 0.4237,
+      "step": 480
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.669309462915601e-05,
+      "loss": 0.1077,
+      "step": 481
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.668542199488491e-05,
+      "loss": 0.3038,
+      "step": 482
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6677749360613812e-05,
+      "loss": 0.6081,
+      "step": 483
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.667007672634271e-05,
+      "loss": 0.1078,
+      "step": 484
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6662404092071612e-05,
+      "loss": 0.4296,
+      "step": 485
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.665473145780051e-05,
+      "loss": 0.3981,
+      "step": 486
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6647058823529412e-05,
+      "loss": 0.5853,
+      "step": 487
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6639386189258314e-05,
+      "loss": 0.802,
+      "step": 488
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6631713554987212e-05,
+      "loss": 0.4576,
+      "step": 489
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.662404092071611e-05,
+      "loss": 0.2682,
+      "step": 490
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6616368286445015e-05,
+      "loss": 0.1991,
+      "step": 491
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6608695652173913e-05,
+      "loss": 0.1409,
+      "step": 492
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.660102301790281e-05,
+      "loss": 0.3869,
+      "step": 493
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6593350383631713e-05,
+      "loss": 0.5301,
+      "step": 494
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6585677749360615e-05,
+      "loss": 0.8116,
+      "step": 495
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6578005115089513e-05,
+      "loss": 0.0984,
+      "step": 496
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6570332480818415e-05,
+      "loss": 0.7773,
+      "step": 497
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6562659846547316e-05,
+      "loss": 1.3522,
+      "step": 498
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6554987212276215e-05,
+      "loss": 0.0862,
+      "step": 499
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6547314578005116e-05,
+      "loss": 0.0894,
+      "step": 500
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6539641943734014e-05,
+      "loss": 0.0885,
+      "step": 501
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.653196930946292e-05,
+      "loss": 0.3031,
+      "step": 502
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6524296675191818e-05,
+      "loss": 0.187,
+      "step": 503
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6516624040920716e-05,
+      "loss": 1.3063,
+      "step": 504
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6508951406649617e-05,
+      "loss": 0.6847,
+      "step": 505
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.650127877237852e-05,
+      "loss": 0.078,
+      "step": 506
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6493606138107417e-05,
+      "loss": 0.1265,
+      "step": 507
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.648593350383632e-05,
+      "loss": 0.0804,
+      "step": 508
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.6478260869565217e-05,
+      "loss": 0.6976,
+      "step": 509
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.647058823529412e-05,
+      "loss": 0.7977,
+      "step": 510
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.646291560102302e-05,
+      "loss": 0.413,
+      "step": 511
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.645524296675192e-05,
+      "loss": 0.6944,
+      "step": 512
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6447570332480817e-05,
+      "loss": 0.7177,
+      "step": 513
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6439897698209722e-05,
+      "loss": 0.0581,
+      "step": 514
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.643222506393862e-05,
+      "loss": 0.9308,
+      "step": 515
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6424552429667518e-05,
+      "loss": 0.2518,
+      "step": 516
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.641687979539642e-05,
+      "loss": 0.1014,
+      "step": 517
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640920716112532e-05,
+      "loss": 0.1641,
+      "step": 518
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640153452685422e-05,
+      "loss": 0.101,
+      "step": 519
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.639386189258312e-05,
+      "loss": 0.1097,
+      "step": 520
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6386189258312023e-05,
+      "loss": 0.987,
+      "step": 521
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.637851662404092e-05,
+      "loss": 0.6465,
+      "step": 522
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6370843989769823e-05,
+      "loss": 0.0914,
+      "step": 523
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.636317135549872e-05,
+      "loss": 0.139,
+      "step": 524
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6355498721227623e-05,
+      "loss": 0.6455,
+      "step": 525
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6347826086956524e-05,
+      "loss": 0.204,
+      "step": 526
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6340153452685422e-05,
+      "loss": 0.1269,
+      "step": 527
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.633248081841432e-05,
+      "loss": 0.2214,
+      "step": 528
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6324808184143226e-05,
+      "loss": 0.0736,
+      "step": 529
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6317135549872124e-05,
+      "loss": 0.0714,
+      "step": 530
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6309462915601022e-05,
+      "loss": 0.1388,
+      "step": 531
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6301790281329924e-05,
+      "loss": 0.7758,
+      "step": 532
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6294117647058825e-05,
+      "loss": 0.3388,
+      "step": 533
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6286445012787724e-05,
+      "loss": 1.1831,
+      "step": 534
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6278772378516625e-05,
+      "loss": 0.1166,
+      "step": 535
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6271099744245523e-05,
+      "loss": 0.2389,
+      "step": 536
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6263427109974425e-05,
+      "loss": 0.1622,
+      "step": 537
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6255754475703327e-05,
+      "loss": 0.1015,
+      "step": 538
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6248081841432225e-05,
+      "loss": 0.0658,
+      "step": 539
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6240409207161123e-05,
+      "loss": 0.0954,
+      "step": 540
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6232736572890028e-05,
+      "loss": 0.1182,
+      "step": 541
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6225063938618926e-05,
+      "loss": 0.0605,
+      "step": 542
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6217391304347824e-05,
+      "loss": 0.9265,
+      "step": 543
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6209718670076726e-05,
+      "loss": 1.2994,
+      "step": 544
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6202046035805628e-05,
+      "loss": 0.081,
+      "step": 545
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6194373401534526e-05,
+      "loss": 0.5962,
+      "step": 546
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6186700767263428e-05,
+      "loss": 0.0665,
+      "step": 547
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.617902813299233e-05,
+      "loss": 0.0655,
+      "step": 548
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6171355498721227e-05,
+      "loss": 0.0642,
+      "step": 549
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.616368286445013e-05,
+      "loss": 0.0834,
+      "step": 550
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6156010230179027e-05,
+      "loss": 0.047,
+      "step": 551
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6148337595907932e-05,
+      "loss": 0.0642,
+      "step": 552
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.614066496163683e-05,
+      "loss": 0.0594,
+      "step": 553
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.613299232736573e-05,
+      "loss": 0.0573,
+      "step": 554
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.612531969309463e-05,
+      "loss": 0.1101,
+      "step": 555
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.6117647058823532e-05,
+      "loss": 0.5121,
+      "step": 556
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.610997442455243e-05,
+      "loss": 0.0374,
+      "step": 557
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.6102301790281332e-05,
+      "loss": 0.0957,
+      "step": 558
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.609462915601023e-05,
+      "loss": 0.0488,
+      "step": 559
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.608695652173913e-05,
+      "loss": 0.0482,
+      "step": 560
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6079283887468033e-05,
+      "loss": 1.2041,
+      "step": 561
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.607161125319693e-05,
+      "loss": 0.8833,
+      "step": 562
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.606393861892583e-05,
+      "loss": 0.1544,
+      "step": 563
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6056265984654735e-05,
+      "loss": 0.046,
+      "step": 564
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6048593350383633e-05,
+      "loss": 0.9073,
+      "step": 565
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.604092071611253e-05,
+      "loss": 0.052,
+      "step": 566
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6033248081841433e-05,
+      "loss": 0.0308,
+      "step": 567
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6025575447570334e-05,
+      "loss": 0.7032,
+      "step": 568
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6017902813299232e-05,
+      "loss": 0.6947,
+      "step": 569
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6010230179028134e-05,
+      "loss": 1.5603,
+      "step": 570
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6002557544757036e-05,
+      "loss": 0.8836,
+      "step": 571
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.5994884910485934e-05,
+      "loss": 1.7439,
+      "step": 572
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5987212276214836e-05,
+      "loss": 0.0313,
+      "step": 573
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5979539641943734e-05,
+      "loss": 0.0317,
+      "step": 574
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5971867007672635e-05,
+      "loss": 0.0325,
+      "step": 575
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5964194373401537e-05,
+      "loss": 0.5,
+      "step": 576
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5956521739130435e-05,
+      "loss": 0.0675,
+      "step": 577
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5948849104859333e-05,
+      "loss": 0.6364,
+      "step": 578
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.594117647058824e-05,
+      "loss": 0.0696,
+      "step": 579
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5933503836317137e-05,
+      "loss": 0.6195,
+      "step": 580
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5925831202046035e-05,
+      "loss": 0.5796,
+      "step": 581
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5918158567774937e-05,
+      "loss": 0.0832,
+      "step": 582
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5910485933503838e-05,
+      "loss": 0.0794,
+      "step": 583
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5902813299232736e-05,
+      "loss": 0.5851,
+      "step": 584
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5895140664961638e-05,
+      "loss": 0.104,
+      "step": 585
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5887468030690536e-05,
+      "loss": 1.0492,
+      "step": 586
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5879795396419438e-05,
+      "loss": 0.0645,
+      "step": 587
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.587212276214834e-05,
+      "loss": 0.514,
+      "step": 588
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5864450127877238e-05,
+      "loss": 0.0701,
+      "step": 589
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5856777493606136e-05,
+      "loss": 0.5409,
+      "step": 590
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584910485933504e-05,
+      "loss": 0.0536,
+      "step": 591
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584143222506394e-05,
+      "loss": 0.1728,
+      "step": 592
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.5833759590792837e-05,
+      "loss": 0.0845,
+      "step": 593
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.582608695652174e-05,
+      "loss": 0.2217,
+      "step": 594
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.581841432225064e-05,
+      "loss": 0.1397,
+      "step": 595
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.581074168797954e-05,
+      "loss": 0.4119,
+      "step": 596
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.580306905370844e-05,
+      "loss": 0.3638,
+      "step": 597
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.5795396419437342e-05,
+      "loss": 0.0781,
+      "step": 598
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.578772378516624e-05,
+      "loss": 0.5926,
+      "step": 599
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5780051150895142e-05,
+      "loss": 0.1378,
+      "step": 600
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.577237851662404e-05,
+      "loss": 0.142,
+      "step": 601
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.576470588235294e-05,
+      "loss": 0.1722,
+      "step": 602
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5757033248081843e-05,
+      "loss": 0.1529,
+      "step": 603
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.574936061381074e-05,
+      "loss": 1.6083,
+      "step": 604
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5741687979539643e-05,
+      "loss": 1.8267,
+      "step": 605
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5734015345268545e-05,
+      "loss": 0.0535,
+      "step": 606
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5726342710997443e-05,
+      "loss": 0.9945,
+      "step": 607
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5718670076726345e-05,
+      "loss": 0.9133,
+      "step": 608
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5710997442455243e-05,
+      "loss": 0.5295,
+      "step": 609
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5703324808184144e-05,
+      "loss": 0.058,
+      "step": 610
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5695652173913046e-05,
+      "loss": 0.4789,
+      "step": 611
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5687979539641944e-05,
+      "loss": 0.0572,
+      "step": 612
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5680306905370842e-05,
+      "loss": 0.0897,
+      "step": 613
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5672634271099747e-05,
+      "loss": 0.056,
+      "step": 614
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5664961636828646e-05,
+      "loss": 0.0278,
+      "step": 615
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5657289002557544e-05,
+      "loss": 0.9947,
+      "step": 616
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5649616368286445e-05,
+      "loss": 0.0627,
+      "step": 617
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5641943734015347e-05,
+      "loss": 0.9335,
+      "step": 618
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5634271099744245e-05,
+      "loss": 0.9793,
+      "step": 619
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5626598465473147e-05,
+      "loss": 1.8128,
+      "step": 620
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.561892583120205e-05,
+      "loss": 1.8015,
+      "step": 621
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5611253196930947e-05,
+      "loss": 0.9259,
+      "step": 622
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.560358056265985e-05,
+      "loss": 0.9618,
+      "step": 623
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5595907928388747e-05,
+      "loss": 0.0808,
+      "step": 624
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5588235294117648e-05,
+      "loss": 1.7161,
+      "step": 625
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.558056265984655e-05,
+      "loss": 0.8879,
+      "step": 626
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5572890025575448e-05,
+      "loss": 0.043,
+      "step": 627
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5565217391304346e-05,
+      "loss": 0.9449,
+      "step": 628
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.555754475703325e-05,
+      "loss": 0.0624,
+      "step": 629
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.554987212276215e-05,
+      "loss": 0.0924,
+      "step": 630
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5542199488491048e-05,
+      "loss": 0.0715,
+      "step": 631
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.553452685421995e-05,
+      "loss": 0.4995,
+      "step": 632
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.552685421994885e-05,
+      "loss": 0.5576,
+      "step": 633
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551918158567775e-05,
+      "loss": 0.06,
+      "step": 634
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551150895140665e-05,
+      "loss": 0.0873,
+      "step": 635
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.550383631713555e-05,
+      "loss": 0.6868,
+      "step": 636
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.549616368286445e-05,
+      "loss": 0.8487,
+      "step": 637
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.5488491048593352e-05,
+      "loss": 0.5641,
+      "step": 638
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.548081841432225e-05,
+      "loss": 1.1343,
+      "step": 639
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.547314578005115e-05,
+      "loss": 0.0614,
+      "step": 640
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5465473145780054e-05,
+      "loss": 0.9414,
+      "step": 641
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5457800511508952e-05,
+      "loss": 0.9226,
+      "step": 642
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.545012787723785e-05,
+      "loss": 0.3323,
+      "step": 643
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5442455242966752e-05,
+      "loss": 0.9315,
+      "step": 644
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5434782608695653e-05,
+      "loss": 0.5539,
+      "step": 645
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.542710997442455e-05,
+      "loss": 0.3376,
+      "step": 646
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5419437340153453e-05,
+      "loss": 0.0694,
+      "step": 647
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5411764705882355e-05,
+      "loss": 0.0706,
+      "step": 648
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5404092071611253e-05,
+      "loss": 0.105,
+      "step": 649
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5396419437340155e-05,
+      "loss": 0.912,
+      "step": 650
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5388746803069053e-05,
+      "loss": 0.0998,
+      "step": 651
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5381074168797954e-05,
+      "loss": 0.2247,
+      "step": 652
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5373401534526856e-05,
+      "loss": 0.2869,
+      "step": 653
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5365728900255754e-05,
+      "loss": 0.532,
+      "step": 654
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5358056265984653e-05,
+      "loss": 0.4723,
+      "step": 655
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5350383631713558e-05,
+      "loss": 0.1178,
+      "step": 656
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5342710997442456e-05,
+      "loss": 0.0566,
+      "step": 657
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5335038363171357e-05,
+      "loss": 0.059,
+      "step": 658
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5327365728900256e-05,
+      "loss": 0.0881,
+      "step": 659
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5319693094629157e-05,
+      "loss": 0.5762,
+      "step": 660
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.531202046035806e-05,
+      "loss": 0.0499,
+      "step": 661
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5304347826086957e-05,
+      "loss": 0.087,
+      "step": 662
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5296675191815855e-05,
+      "loss": 1.0719,
+      "step": 663
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528900255754476e-05,
+      "loss": 0.6158,
+      "step": 664
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528132992327366e-05,
+      "loss": 0.6117,
+      "step": 665
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5273657289002557e-05,
+      "loss": 0.0515,
+      "step": 666
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5265984654731458e-05,
+      "loss": 0.0608,
+      "step": 667
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.525831202046036e-05,
+      "loss": 0.0313,
+      "step": 668
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.5250639386189258e-05,
+      "loss": 0.1449,
+      "step": 669
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.524296675191816e-05,
+      "loss": 0.9883,
+      "step": 670
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.523529411764706e-05,
+      "loss": 0.0859,
+      "step": 671
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.522762148337596e-05,
+      "loss": 0.0561,
+      "step": 672
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521994884910486e-05,
+      "loss": 0.0926,
+      "step": 673
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521227621483376e-05,
+      "loss": 0.1101,
+      "step": 674
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.520460358056266e-05,
+      "loss": 0.068,
+      "step": 675
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5196930946291563e-05,
+      "loss": 0.0773,
+      "step": 676
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518925831202046e-05,
+      "loss": 0.0508,
+      "step": 677
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518158567774936e-05,
+      "loss": 0.0375,
+      "step": 678
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5173913043478264e-05,
+      "loss": 0.93,
+      "step": 679
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5166240409207162e-05,
+      "loss": 0.1507,
+      "step": 680
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.515856777493606e-05,
+      "loss": 0.0917,
+      "step": 681
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5150895140664962e-05,
+      "loss": 0.97,
+      "step": 682
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5143222506393864e-05,
+      "loss": 0.5019,
+      "step": 683
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5135549872122762e-05,
+      "loss": 0.0849,
+      "step": 684
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5127877237851664e-05,
+      "loss": 1.2316,
+      "step": 685
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5120204603580562e-05,
+      "loss": 0.0252,
+      "step": 686
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5112531969309463e-05,
+      "loss": 1.0543,
+      "step": 687
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5104859335038365e-05,
+      "loss": 0.0506,
+      "step": 688
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5097186700767263e-05,
+      "loss": 0.0252,
+      "step": 689
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.508951406649616e-05,
+      "loss": 0.6297,
+      "step": 690
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5081841432225066e-05,
+      "loss": 0.4497,
+      "step": 691
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5074168797953965e-05,
+      "loss": 0.0516,
+      "step": 692
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5066496163682863e-05,
+      "loss": 0.5501,
+      "step": 693
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5058823529411768e-05,
+      "loss": 0.5621,
+      "step": 694
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5051150895140666e-05,
+      "loss": 0.0577,
+      "step": 695
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5043478260869564e-05,
+      "loss": 0.0271,
+      "step": 696
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5035805626598466e-05,
+      "loss": 0.9496,
+      "step": 697
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5028132992327368e-05,
+      "loss": 0.0877,
+      "step": 698
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5020460358056266e-05,
+      "loss": 0.0523,
+      "step": 699
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5012787723785167e-05,
+      "loss": 0.0577,
+      "step": 700
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5005115089514066e-05,
+      "loss": 0.0522,
+      "step": 701
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.4997442455242967e-05,
+      "loss": 0.0517,
+      "step": 702
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.498976982097187e-05,
+      "loss": 0.6335,
+      "step": 703
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4982097186700767e-05,
+      "loss": 0.05,
+      "step": 704
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4974424552429665e-05,
+      "loss": 0.0753,
+      "step": 705
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.496675191815857e-05,
+      "loss": 0.0227,
+      "step": 706
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.495907928388747e-05,
+      "loss": 0.0755,
+      "step": 707
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.4951406649616367e-05,
+      "loss": 0.0215,
+      "step": 708
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.494373401534527e-05,
+      "loss": 0.0451,
+      "step": 709
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.493606138107417e-05,
+      "loss": 0.0343,
+      "step": 710
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492838874680307e-05,
+      "loss": 0.0836,
+      "step": 711
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492071611253197e-05,
+      "loss": 0.5287,
+      "step": 712
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4913043478260868e-05,
+      "loss": 0.0686,
+      "step": 713
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4905370843989773e-05,
+      "loss": 0.0211,
+      "step": 714
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489769820971867e-05,
+      "loss": 0.5932,
+      "step": 715
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489002557544757e-05,
+      "loss": 0.0371,
+      "step": 716
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.488235294117647e-05,
+      "loss": 1.0137,
+      "step": 717
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.4874680306905373e-05,
+      "loss": 0.9928,
+      "step": 718
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.486700767263427e-05,
+      "loss": 0.7431,
+      "step": 719
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4859335038363173e-05,
+      "loss": 0.0207,
+      "step": 720
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4851662404092074e-05,
+      "loss": 0.0681,
+      "step": 721
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4843989769820972e-05,
+      "loss": 0.0306,
+      "step": 722
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4836317135549874e-05,
+      "loss": 0.0204,
+      "step": 723
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4828644501278772e-05,
+      "loss": 0.0344,
+      "step": 724
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4820971867007674e-05,
+      "loss": 0.0345,
+      "step": 725
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4813299232736575e-05,
+      "loss": 0.9275,
+      "step": 726
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4805626598465474e-05,
+      "loss": 0.0196,
+      "step": 727
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4797953964194372e-05,
+      "loss": 1.0137,
+      "step": 728
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4790281329923277e-05,
+      "loss": 0.0318,
+      "step": 729
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4782608695652175e-05,
+      "loss": 0.0688,
+      "step": 730
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4774936061381073e-05,
+      "loss": 0.0314,
+      "step": 731
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4767263427109975e-05,
+      "loss": 0.7135,
+      "step": 732
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4759590792838877e-05,
+      "loss": 0.0299,
+      "step": 733
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4751918158567775e-05,
+      "loss": 1.0127,
+      "step": 734
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4744245524296676e-05,
+      "loss": 0.0304,
+      "step": 735
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4736572890025575e-05,
+      "loss": 1.4618,
+      "step": 736
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4728900255754476e-05,
+      "loss": 0.7167,
+      "step": 737
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4721227621483378e-05,
+      "loss": 0.0307,
+      "step": 738
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4713554987212276e-05,
+      "loss": 0.052,
+      "step": 739
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4705882352941174e-05,
+      "loss": 0.778,
+      "step": 740
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.469820971867008e-05,
+      "loss": 0.5482,
+      "step": 741
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4690537084398978e-05,
+      "loss": 1.4314,
+      "step": 742
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4682864450127876e-05,
+      "loss": 0.0422,
+      "step": 743
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.467519181585678e-05,
+      "loss": 0.6614,
+      "step": 744
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.466751918158568e-05,
+      "loss": 0.0312,
+      "step": 745
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4659846547314577e-05,
+      "loss": 0.043,
+      "step": 746
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.465217391304348e-05,
+      "loss": 0.0199,
+      "step": 747
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.464450127877238e-05,
+      "loss": 0.0832,
+      "step": 748
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.463682864450128e-05,
+      "loss": 0.0312,
+      "step": 749
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.462915601023018e-05,
+      "loss": 0.0309,
+      "step": 750
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.462148337595908e-05,
+      "loss": 0.0193,
+      "step": 751
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.461381074168798e-05,
+      "loss": 0.0446,
+      "step": 752
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.4606138107416882e-05,
+      "loss": 0.9993,
+      "step": 753
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.459846547314578e-05,
+      "loss": 0.016,
+      "step": 754
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4590792838874678e-05,
+      "loss": 0.0433,
+      "step": 755
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4583120204603583e-05,
+      "loss": 0.2614,
+      "step": 756
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.457544757033248e-05,
+      "loss": 0.018,
+      "step": 757
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.456777493606138e-05,
+      "loss": 0.6676,
+      "step": 758
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.456010230179028e-05,
+      "loss": 0.0599,
+      "step": 759
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4552429667519183e-05,
+      "loss": 0.0338,
+      "step": 760
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.454475703324808e-05,
+      "loss": 0.0268,
+      "step": 761
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4537084398976983e-05,
+      "loss": 0.0262,
+      "step": 762
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.452941176470588e-05,
+      "loss": 0.0252,
+      "step": 763
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4521739130434786e-05,
+      "loss": 0.0377,
+      "step": 764
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4514066496163684e-05,
+      "loss": 1.4866,
+      "step": 765
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4506393861892582e-05,
+      "loss": 1.0524,
+      "step": 766
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4498721227621484e-05,
+      "loss": 0.0328,
+      "step": 767
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4491048593350386e-05,
+      "loss": 0.024,
+      "step": 768
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4483375959079284e-05,
+      "loss": 0.022,
+      "step": 769
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4475703324808185e-05,
+      "loss": 0.0245,
+      "step": 770
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4468030690537087e-05,
+      "loss": 1.0107,
+      "step": 771
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4460358056265985e-05,
+      "loss": 0.0336,
+      "step": 772
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4452685421994887e-05,
+      "loss": 0.0391,
+      "step": 773
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4445012787723785e-05,
+      "loss": 0.0303,
+      "step": 774
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4437340153452687e-05,
+      "loss": 0.0443,
+      "step": 775
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4429667519181588e-05,
+      "loss": 0.0292,
+      "step": 776
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4421994884910487e-05,
+      "loss": 0.0284,
+      "step": 777
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4414322250639385e-05,
+      "loss": 1.5985,
+      "step": 778
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.440664961636829e-05,
+      "loss": 0.0114,
+      "step": 779
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4398976982097188e-05,
+      "loss": 0.0342,
+      "step": 780
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4391304347826086e-05,
+      "loss": 1.8501,
+      "step": 781
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4383631713554988e-05,
+      "loss": 0.5375,
+      "step": 782
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.437595907928389e-05,
+      "loss": 1.589,
+      "step": 783
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4368286445012788e-05,
+      "loss": 0.0221,
+      "step": 784
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.436061381074169e-05,
+      "loss": 0.0453,
+      "step": 785
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4352941176470587e-05,
+      "loss": 0.796,
+      "step": 786
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.434526854219949e-05,
+      "loss": 0.1249,
+      "step": 787
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.433759590792839e-05,
+      "loss": 0.7963,
+      "step": 788
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.432992327365729e-05,
+      "loss": 0.4231,
+      "step": 789
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.4322250639386187e-05,
+      "loss": 0.0247,
+      "step": 790
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.4314578005115092e-05,
+      "loss": 0.0295,
+      "step": 791
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.430690537084399e-05,
+      "loss": 0.0249,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.3537317216396332,
+      "eval_runtime": 58.1226,
+      "eval_samples_per_second": 6.813,
+      "eval_steps_per_second": 1.703,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.429923273657289e-05,
+      "loss": 0.0157,
+      "step": 793
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4291560102301794e-05,
+      "loss": 0.5222,
+      "step": 794
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.4283887468030692e-05,
+      "loss": 0.0226,
+      "step": 795
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.427621483375959e-05,
+      "loss": 0.0293,
+      "step": 796
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 2.426854219948849e-05,
+      "loss": 0.015,
+      "step": 797
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4260869565217393e-05,
+      "loss": 1.0658,
+      "step": 798
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.425319693094629e-05,
+      "loss": 0.0412,
+      "step": 799
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.4245524296675193e-05,
+      "loss": 0.0219,
+      "step": 800
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 2.423785166240409e-05,
+      "loss": 0.0149,
+      "step": 801
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4230179028132993e-05,
+      "loss": 0.0323,
+      "step": 802
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4222506393861895e-05,
+      "loss": 1.0764,
+      "step": 803
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.4214833759590793e-05,
+      "loss": 0.7518,
+      "step": 804
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 2.420716112531969e-05,
+      "loss": 0.8166,
+      "step": 805
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4199488491048596e-05,
+      "loss": 0.0219,
+      "step": 806
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4191815856777494e-05,
+      "loss": 0.0154,
+      "step": 807
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4184143222506392e-05,
+      "loss": 0.0163,
+      "step": 808
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 2.4176470588235294e-05,
+      "loss": 1.0562,
+      "step": 809
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4168797953964196e-05,
+      "loss": 0.0218,
+      "step": 810
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4161125319693094e-05,
+      "loss": 0.0232,
+      "step": 811
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4153452685421995e-05,
+      "loss": 0.2384,
+      "step": 812
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 2.4145780051150894e-05,
+      "loss": 1.5286,
+      "step": 813
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4138107416879795e-05,
+      "loss": 0.0262,
+      "step": 814
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4130434782608697e-05,
+      "loss": 0.0334,
+      "step": 815
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4122762148337595e-05,
+      "loss": 0.6509,
+      "step": 816
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 2.4115089514066497e-05,
+      "loss": 0.0159,
+      "step": 817
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.41074168797954e-05,
+      "loss": 0.6225,
+      "step": 818
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4099744245524297e-05,
+      "loss": 0.752,
+      "step": 819
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.4092071611253198e-05,
+      "loss": 1.0594,
+      "step": 820
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 2.40843989769821e-05,
+      "loss": 0.0161,
+      "step": 821
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4076726342710998e-05,
+      "loss": 0.7492,
+      "step": 822
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40690537084399e-05,
+      "loss": 0.0302,
+      "step": 823
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.4061381074168798e-05,
+      "loss": 0.038,
+      "step": 824
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 2.40537084398977e-05,
+      "loss": 0.0243,
+      "step": 825
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40460358056266e-05,
+      "loss": 0.0331,
+      "step": 826
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.40383631713555e-05,
+      "loss": 1.0252,
+      "step": 827
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4030690537084398e-05,
+      "loss": 0.0172,
+      "step": 828
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 2.4023017902813303e-05,
+      "loss": 0.6168,
+      "step": 829
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40153452685422e-05,
+      "loss": 0.3135,
+      "step": 830
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.40076726342711e-05,
+      "loss": 0.0168,
+      "step": 831
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.0172,
+      "step": 832
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 2.3992327365728902e-05,
+      "loss": 1.0349,
+      "step": 833
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39846547314578e-05,
+      "loss": 0.0978,
+      "step": 834
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3976982097186702e-05,
+      "loss": 0.0167,
+      "step": 835
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.39693094629156e-05,
+      "loss": 1.0381,
+      "step": 836
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 2.3961636828644502e-05,
+      "loss": 0.6857,
+      "step": 837
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3953964194373404e-05,
+      "loss": 1.0406,
+      "step": 838
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3946291560102302e-05,
+      "loss": 0.071,
+      "step": 839
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.39386189258312e-05,
+      "loss": 0.0193,
+      "step": 840
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 2.3930946291560105e-05,
+      "loss": 0.0193,
+      "step": 841
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3923273657289003e-05,
+      "loss": 0.9916,
+      "step": 842
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.39156010230179e-05,
+      "loss": 0.0723,
+      "step": 843
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3907928388746806e-05,
+      "loss": 0.0626,
+      "step": 844
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 2.3900255754475705e-05,
+      "loss": 0.7264,
+      "step": 845
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3892583120204603e-05,
+      "loss": 0.827,
+      "step": 846
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3884910485933504e-05,
+      "loss": 1.2193,
+      "step": 847
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3877237851662406e-05,
+      "loss": 0.6754,
+      "step": 848
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 2.3869565217391304e-05,
+      "loss": 0.6856,
+      "step": 849
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3861892583120206e-05,
+      "loss": 0.7441,
+      "step": 850
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3854219948849104e-05,
+      "loss": 0.023,
+      "step": 851
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3846547314578006e-05,
+      "loss": 0.4742,
+      "step": 852
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 2.3838874680306907e-05,
+      "loss": 0.0363,
+      "step": 853
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3831202046035806e-05,
+      "loss": 0.0409,
+      "step": 854
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3823529411764704e-05,
+      "loss": 0.8353,
+      "step": 855
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.381585677749361e-05,
+      "loss": 0.0333,
+      "step": 856
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 2.3808184143222507e-05,
+      "loss": 0.0216,
+      "step": 857
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3800511508951405e-05,
+      "loss": 0.9688,
+      "step": 858
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3792838874680307e-05,
+      "loss": 1.0078,
+      "step": 859
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.378516624040921e-05,
+      "loss": 0.0183,
+      "step": 860
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 2.3777493606138107e-05,
+      "loss": 0.0183,
+      "step": 861
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3769820971867008e-05,
+      "loss": 0.9786,
+      "step": 862
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3762148337595907e-05,
+      "loss": 0.9166,
+      "step": 863
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.3754475703324808e-05,
+      "loss": 1.1408,
+      "step": 864
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 2.374680306905371e-05,
+      "loss": 1.1394,
+      "step": 865
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.3739130434782608e-05,
+      "loss": 0.0185,
+      "step": 866
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.373145780051151e-05,
+      "loss": 0.9979,
+      "step": 867
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.372378516624041e-05,
+      "loss": 1.0092,
+      "step": 868
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 2.371611253196931e-05,
+      "loss": 0.7276,
+      "step": 869
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.370843989769821e-05,
+      "loss": 0.0359,
+      "step": 870
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3700767263427113e-05,
+      "loss": 0.0864,
+      "step": 871
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.369309462915601e-05,
+      "loss": 0.0246,
+      "step": 872
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 2.3685421994884912e-05,
+      "loss": 0.8518,
+      "step": 873
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.367774936061381e-05,
+      "loss": 1.433,
+      "step": 874
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3670076726342712e-05,
+      "loss": 0.035,
+      "step": 875
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3662404092071614e-05,
+      "loss": 0.5922,
+      "step": 876
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 2.3654731457800512e-05,
+      "loss": 0.529,
+      "step": 877
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.364705882352941e-05,
+      "loss": 0.6423,
+      "step": 878
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3639386189258315e-05,
+      "loss": 0.0337,
+      "step": 879
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3631713554987214e-05,
+      "loss": 0.0402,
+      "step": 880
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 2.3624040920716112e-05,
+      "loss": 0.0782,
+      "step": 881
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3616368286445013e-05,
+      "loss": 0.4141,
+      "step": 882
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3608695652173915e-05,
+      "loss": 0.0359,
+      "step": 883
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3601023017902813e-05,
+      "loss": 0.4509,
+      "step": 884
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 2.3593350383631715e-05,
+      "loss": 0.6414,
+      "step": 885
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3585677749360613e-05,
+      "loss": 0.6873,
+      "step": 886
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3578005115089515e-05,
+      "loss": 0.0304,
+      "step": 887
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3570332480818416e-05,
+      "loss": 0.0315,
+      "step": 888
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 2.3562659846547315e-05,
+      "loss": 0.0896,
+      "step": 889
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3554987212276213e-05,
+      "loss": 0.0201,
+      "step": 890
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3547314578005118e-05,
+      "loss": 0.027,
+      "step": 891
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 2.3539641943734016e-05,
+      "loss": 1.9853,
+      "step": 892
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3531969309462914e-05,
+      "loss": 2.9647,
+      "step": 893
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.352429667519182e-05,
+      "loss": 1.9403,
+      "step": 894
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3516624040920717e-05,
+      "loss": 0.0193,
+      "step": 895
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 2.3508951406649616e-05,
+      "loss": 0.9912,
+      "step": 896
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3501278772378517e-05,
+      "loss": 0.02,
+      "step": 897
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.349360613810742e-05,
+      "loss": 0.9015,
+      "step": 898
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.3485933503836317e-05,
+      "loss": 0.1294,
+      "step": 899
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 2.347826086956522e-05,
+      "loss": 0.0559,
+      "step": 900
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.3470588235294117e-05,
+      "loss": 1.8039,
+      "step": 901
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.346291560102302e-05,
+      "loss": 0.0336,
+      "step": 902
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.345524296675192e-05,
+      "loss": 0.0383,
+      "step": 903
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 2.344757033248082e-05,
+      "loss": 0.0722,
+      "step": 904
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3439897698209717e-05,
+      "loss": 0.0278,
+      "step": 905
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.343222506393862e-05,
+      "loss": 0.0684,
+      "step": 906
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.342455242966752e-05,
+      "loss": 0.0236,
+      "step": 907
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 2.3416879795396418e-05,
+      "loss": 0.0348,
+      "step": 908
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340920716112532e-05,
+      "loss": 0.0204,
+      "step": 909
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.340153452685422e-05,
+      "loss": 0.8109,
+      "step": 910
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.339386189258312e-05,
+      "loss": 0.0279,
+      "step": 911
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 2.338618925831202e-05,
+      "loss": 0.0361,
+      "step": 912
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337851662404092e-05,
+      "loss": 0.0355,
+      "step": 913
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.337084398976982e-05,
+      "loss": 0.7958,
+      "step": 914
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.3363171355498723e-05,
+      "loss": 0.0362,
+      "step": 915
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 2.335549872122762e-05,
+      "loss": 0.0252,
+      "step": 916
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3347826086956522e-05,
+      "loss": 0.8165,
+      "step": 917
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3340153452685424e-05,
+      "loss": 0.0237,
+      "step": 918
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.3332480818414322e-05,
+      "loss": 0.8162,
+      "step": 919
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 2.332480818414322e-05,
+      "loss": 0.7786,
+      "step": 920
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3317135549872125e-05,
+      "loss": 0.0787,
+      "step": 921
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3309462915601024e-05,
+      "loss": 0.1764,
+      "step": 922
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3301790281329925e-05,
+      "loss": 0.7449,
+      "step": 923
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 2.3294117647058824e-05,
+      "loss": 0.0452,
+      "step": 924
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3286445012787725e-05,
+      "loss": 0.0181,
+      "step": 925
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3278772378516627e-05,
+      "loss": 0.1783,
+      "step": 926
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3271099744245525e-05,
+      "loss": 0.5158,
+      "step": 927
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 2.3263427109974423e-05,
+      "loss": 0.7574,
+      "step": 928
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3255754475703328e-05,
+      "loss": 2.6099,
+      "step": 929
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3248081841432226e-05,
+      "loss": 0.0304,
+      "step": 930
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3240409207161125e-05,
+      "loss": 0.0795,
+      "step": 931
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 2.3232736572890026e-05,
+      "loss": 0.0342,
+      "step": 932
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3225063938618928e-05,
+      "loss": 0.0473,
+      "step": 933
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3217391304347826e-05,
+      "loss": 0.0831,
+      "step": 934
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3209718670076728e-05,
+      "loss": 0.0465,
+      "step": 935
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 2.3202046035805626e-05,
+      "loss": 0.6769,
+      "step": 936
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3194373401534528e-05,
+      "loss": 0.0173,
+      "step": 937
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.318670076726343e-05,
+      "loss": 0.0169,
+      "step": 938
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3179028132992327e-05,
+      "loss": 0.0469,
+      "step": 939
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 2.3171355498721226e-05,
+      "loss": 0.048,
+      "step": 940
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.316368286445013e-05,
+      "loss": 1.0704,
+      "step": 941
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.315601023017903e-05,
+      "loss": 1.0738,
+      "step": 942
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3148337595907927e-05,
+      "loss": 0.0163,
+      "step": 943
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 2.3140664961636832e-05,
+      "loss": 2.0702,
+      "step": 944
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.313299232736573e-05,
+      "loss": 1.0433,
+      "step": 945
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.312531969309463e-05,
+      "loss": 1.0351,
+      "step": 946
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.311764705882353e-05,
+      "loss": 1.053,
+      "step": 947
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 2.3109974424552432e-05,
+      "loss": 0.1797,
+      "step": 948
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.310230179028133e-05,
+      "loss": 1.0259,
+      "step": 949
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.309462915601023e-05,
+      "loss": 0.03,
+      "step": 950
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.308695652173913e-05,
+      "loss": 0.905,
+      "step": 951
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 2.307928388746803e-05,
+      "loss": 0.0699,
+      "step": 952
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3071611253196933e-05,
+      "loss": 1.0013,
+      "step": 953
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.306393861892583e-05,
+      "loss": 0.0319,
+      "step": 954
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.305626598465473e-05,
+      "loss": 0.0361,
+      "step": 955
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 2.3048593350383634e-05,
+      "loss": 0.0201,
+      "step": 956
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3040920716112533e-05,
+      "loss": 0.033,
+      "step": 957
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.303324808184143e-05,
+      "loss": 0.0411,
+      "step": 958
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3025575447570332e-05,
+      "loss": 0.7138,
+      "step": 959
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 2.3017902813299234e-05,
+      "loss": 0.0434,
+      "step": 960
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3010230179028132e-05,
+      "loss": 0.8445,
+      "step": 961
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.3002557544757034e-05,
+      "loss": 0.0323,
+      "step": 962
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2994884910485932e-05,
+      "loss": 0.0203,
+      "step": 963
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 2.2987212276214834e-05,
+      "loss": 0.2302,
+      "step": 964
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2979539641943735e-05,
+      "loss": 0.0293,
+      "step": 965
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2971867007672634e-05,
+      "loss": 0.0593,
+      "step": 966
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2964194373401535e-05,
+      "loss": 0.6505,
+      "step": 967
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 2.2956521739130437e-05,
+      "loss": 0.5891,
+      "step": 968
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2948849104859335e-05,
+      "loss": 0.0191,
+      "step": 969
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2941176470588233e-05,
+      "loss": 0.0188,
+      "step": 970
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2933503836317138e-05,
+      "loss": 0.9537,
+      "step": 971
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 2.2925831202046037e-05,
+      "loss": 0.0186,
+      "step": 972
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2918158567774935e-05,
+      "loss": 1.0237,
+      "step": 973
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2910485933503836e-05,
+      "loss": 0.0417,
+      "step": 974
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.2902813299232738e-05,
+      "loss": 0.0369,
+      "step": 975
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 2.289514066496164e-05,
+      "loss": 0.0184,
+      "step": 976
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2887468030690538e-05,
+      "loss": 0.811,
+      "step": 977
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.2879795396419436e-05,
+      "loss": 0.0183,
+      "step": 978
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.287212276214834e-05,
+      "loss": 1.0442,
+      "step": 979
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 2.286445012787724e-05,
+      "loss": 0.027,
+      "step": 980
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.2856777493606137e-05,
+      "loss": 0.062,
+      "step": 981
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284910485933504e-05,
+      "loss": 0.0186,
+      "step": 982
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.284143222506394e-05,
+      "loss": 0.6057,
+      "step": 983
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 2.283375959079284e-05,
+      "loss": 1.6594,
+      "step": 984
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.282608695652174e-05,
+      "loss": 0.0192,
+      "step": 985
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281841432225064e-05,
+      "loss": 1.5361,
+      "step": 986
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.281074168797954e-05,
+      "loss": 0.0384,
+      "step": 987
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 2.2803069053708442e-05,
+      "loss": 2.0238,
+      "step": 988
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.279539641943734e-05,
+      "loss": 0.0334,
+      "step": 989
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2787723785166242e-05,
+      "loss": 0.0571,
+      "step": 990
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 2.2780051150895143e-05,
+      "loss": 0.0211,
+      "step": 991
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.277237851662404e-05,
+      "loss": 0.0409,
+      "step": 992
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.276470588235294e-05,
+      "loss": 0.1059,
+      "step": 993
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2757033248081845e-05,
+      "loss": 0.0277,
+      "step": 994
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 2.2749360613810743e-05,
+      "loss": 1.0023,
+      "step": 995
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.274168797953964e-05,
+      "loss": 0.0186,
+      "step": 996
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2734015345268543e-05,
+      "loss": 0.0345,
+      "step": 997
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2726342710997445e-05,
+      "loss": 0.0169,
+      "step": 998
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 2.2718670076726343e-05,
+      "loss": 0.7292,
+      "step": 999
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2710997442455244e-05,
+      "loss": 1.2655,
+      "step": 1000
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2703324808184143e-05,
+      "loss": 0.03,
+      "step": 1001
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2695652173913044e-05,
+      "loss": 1.0266,
+      "step": 1002
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 2.2687979539641946e-05,
+      "loss": 0.7645,
+      "step": 1003
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2680306905370844e-05,
+      "loss": 0.7985,
+      "step": 1004
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2672634271099742e-05,
+      "loss": 0.76,
+      "step": 1005
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2664961636828647e-05,
+      "loss": 0.7577,
+      "step": 1006
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 2.2657289002557545e-05,
+      "loss": 0.0248,
+      "step": 1007
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2649616368286444e-05,
+      "loss": 0.0262,
+      "step": 1008
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2641943734015345e-05,
+      "loss": 1.9984,
+      "step": 1009
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2634271099744247e-05,
+      "loss": 0.1489,
+      "step": 1010
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 2.2626598465473145e-05,
+      "loss": 0.0683,
+      "step": 1011
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2618925831202047e-05,
+      "loss": 0.054,
+      "step": 1012
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2611253196930945e-05,
+      "loss": 0.0269,
+      "step": 1013
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2603580562659847e-05,
+      "loss": 1.0378,
+      "step": 1014
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 2.2595907928388748e-05,
+      "loss": 0.0171,
+      "step": 1015
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2588235294117646e-05,
+      "loss": 0.8433,
+      "step": 1016
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2580562659846548e-05,
+      "loss": 0.0468,
+      "step": 1017
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.257289002557545e-05,
+      "loss": 0.0331,
+      "step": 1018
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 2.2565217391304348e-05,
+      "loss": 0.0308,
+      "step": 1019
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2557544757033246e-05,
+      "loss": 1.0007,
+      "step": 1020
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254987212276215e-05,
+      "loss": 2.0242,
+      "step": 1021
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.254219948849105e-05,
+      "loss": 0.7713,
+      "step": 1022
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 2.2534526854219948e-05,
+      "loss": 1.1112,
+      "step": 1023
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.252685421994885e-05,
+      "loss": 0.0184,
+      "step": 1024
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251918158567775e-05,
+      "loss": 0.0243,
+      "step": 1025
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.251150895140665e-05,
+      "loss": 0.046,
+      "step": 1026
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 2.250383631713555e-05,
+      "loss": 0.0378,
+      "step": 1027
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.249616368286445e-05,
+      "loss": 0.0329,
+      "step": 1028
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2488491048593354e-05,
+      "loss": 0.0245,
+      "step": 1029
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.2480818414322252e-05,
+      "loss": 0.0238,
+      "step": 1030
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 2.247314578005115e-05,
+      "loss": 1.0628,
+      "step": 1031
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2465473145780052e-05,
+      "loss": 0.0703,
+      "step": 1032
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2457800511508954e-05,
+      "loss": 0.0251,
+      "step": 1033
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2450127877237852e-05,
+      "loss": 1.0561,
+      "step": 1034
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 2.2442455242966753e-05,
+      "loss": 0.0274,
+      "step": 1035
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.243478260869565e-05,
+      "loss": 0.023,
+      "step": 1036
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2427109974424553e-05,
+      "loss": 0.0223,
+      "step": 1037
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2419437340153455e-05,
+      "loss": 0.8113,
+      "step": 1038
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 2.2411764705882353e-05,
+      "loss": 0.0346,
+      "step": 1039
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2404092071611255e-05,
+      "loss": 0.0164,
+      "step": 1040
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2396419437340156e-05,
+      "loss": 0.0237,
+      "step": 1041
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2388746803069054e-05,
+      "loss": 0.0305,
+      "step": 1042
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 2.2381074168797953e-05,
+      "loss": 0.0158,
+      "step": 1043
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2373401534526858e-05,
+      "loss": 0.0225,
+      "step": 1044
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2365728900255756e-05,
+      "loss": 0.0226,
+      "step": 1045
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2358056265984654e-05,
+      "loss": 0.015,
+      "step": 1046
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 2.2350383631713556e-05,
+      "loss": 0.0219,
+      "step": 1047
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2342710997442457e-05,
+      "loss": 1.0694,
+      "step": 1048
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2335038363171356e-05,
+      "loss": 0.0348,
+      "step": 1049
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2327365728900257e-05,
+      "loss": 0.0213,
+      "step": 1050
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 2.2319693094629155e-05,
+      "loss": 0.9405,
+      "step": 1051
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2312020460358057e-05,
+      "loss": 0.0252,
+      "step": 1052
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.230434782608696e-05,
+      "loss": 0.0254,
+      "step": 1053
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2296675191815857e-05,
+      "loss": 0.7353,
+      "step": 1054
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 2.2289002557544755e-05,
+      "loss": 0.0253,
+      "step": 1055
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.228132992327366e-05,
+      "loss": 0.8361,
+      "step": 1056
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2273657289002558e-05,
+      "loss": 0.0196,
+      "step": 1057
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2265984654731457e-05,
+      "loss": 0.014,
+      "step": 1058
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 2.2258312020460358e-05,
+      "loss": 0.8297,
+      "step": 1059
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.225063938618926e-05,
+      "loss": 1.0654,
+      "step": 1060
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2242966751918158e-05,
+      "loss": 0.0248,
+      "step": 1061
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.223529411764706e-05,
+      "loss": 0.0226,
+      "step": 1062
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 2.2227621483375958e-05,
+      "loss": 0.0238,
+      "step": 1063
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221994884910486e-05,
+      "loss": 0.0212,
+      "step": 1064
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.221227621483376e-05,
+      "loss": 0.0274,
+      "step": 1065
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.220460358056266e-05,
+      "loss": 1.0217,
+      "step": 1066
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 2.219693094629156e-05,
+      "loss": 0.0252,
+      "step": 1067
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2189258312020462e-05,
+      "loss": 0.0219,
+      "step": 1068
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.218158567774936e-05,
+      "loss": 0.0349,
+      "step": 1069
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.217391304347826e-05,
+      "loss": 0.0152,
+      "step": 1070
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 2.2166240409207164e-05,
+      "loss": 0.021,
+      "step": 1071
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2158567774936062e-05,
+      "loss": 0.015,
+      "step": 1072
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.215089514066496e-05,
+      "loss": 0.0277,
+      "step": 1073
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2143222506393862e-05,
+      "loss": 0.1858,
+      "step": 1074
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 2.2135549872122764e-05,
+      "loss": 0.0136,
+      "step": 1075
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2127877237851662e-05,
+      "loss": 0.0132,
+      "step": 1076
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2120204603580563e-05,
+      "loss": 0.0126,
+      "step": 1077
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.211253196930946e-05,
+      "loss": 0.5171,
+      "step": 1078
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 2.2104859335038363e-05,
+      "loss": 0.0422,
+      "step": 1079
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2097186700767265e-05,
+      "loss": 0.0186,
+      "step": 1080
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2089514066496163e-05,
+      "loss": 0.8412,
+      "step": 1081
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2081841432225065e-05,
+      "loss": 0.0183,
+      "step": 1082
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 2.2074168797953966e-05,
+      "loss": 1.0243,
+      "step": 1083
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2066496163682865e-05,
+      "loss": 0.0212,
+      "step": 1084
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2058823529411766e-05,
+      "loss": 0.2841,
+      "step": 1085
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2051150895140664e-05,
+      "loss": 0.0118,
+      "step": 1086
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 2.2043478260869566e-05,
+      "loss": 0.0247,
+      "step": 1087
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2035805626598468e-05,
+      "loss": 1.5121,
+      "step": 1088
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2028132992327366e-05,
+      "loss": 0.0304,
+      "step": 1089
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 2.2020460358056267e-05,
+      "loss": 0.0329,
+      "step": 1090
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.201278772378517e-05,
+      "loss": 1.1288,
+      "step": 1091
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.2005115089514067e-05,
+      "loss": 0.0145,
+      "step": 1092
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.1997442455242966e-05,
+      "loss": 0.0498,
+      "step": 1093
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 2.198976982097187e-05,
+      "loss": 0.0244,
+      "step": 1094
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.198209718670077e-05,
+      "loss": 0.0112,
+      "step": 1095
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.1974424552429667e-05,
+      "loss": 0.8642,
+      "step": 1096
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.196675191815857e-05,
+      "loss": 0.8822,
+      "step": 1097
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 2.195907928388747e-05,
+      "loss": 0.0278,
+      "step": 1098
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.195140664961637e-05,
+      "loss": 0.6954,
+      "step": 1099
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.194373401534527e-05,
+      "loss": 0.0179,
+      "step": 1100
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.1936061381074168e-05,
+      "loss": 0.5607,
+      "step": 1101
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 2.192838874680307e-05,
+      "loss": 0.0191,
+      "step": 1102
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.192071611253197e-05,
+      "loss": 1.9519,
+      "step": 1103
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.191304347826087e-05,
+      "loss": 0.8347,
+      "step": 1104
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1905370843989768e-05,
+      "loss": 0.0262,
+      "step": 1105
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 2.1897698209718673e-05,
+      "loss": 0.0477,
+      "step": 1106
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.189002557544757e-05,
+      "loss": 0.0227,
+      "step": 1107
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.188235294117647e-05,
+      "loss": 0.0205,
+      "step": 1108
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.187468030690537e-05,
+      "loss": 0.0208,
+      "step": 1109
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 2.1867007672634273e-05,
+      "loss": 0.0116,
+      "step": 1110
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.185933503836317e-05,
+      "loss": 1.1546,
+      "step": 1111
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1851662404092072e-05,
+      "loss": 0.0114,
+      "step": 1112
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.184398976982097e-05,
+      "loss": 0.0113,
+      "step": 1113
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 2.1836317135549872e-05,
+      "loss": 0.6881,
+      "step": 1114
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1828644501278774e-05,
+      "loss": 0.0294,
+      "step": 1115
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1820971867007672e-05,
+      "loss": 0.1569,
+      "step": 1116
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1813299232736574e-05,
+      "loss": 1.1362,
+      "step": 1117
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 2.1805626598465475e-05,
+      "loss": 0.0196,
+      "step": 1118
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1797953964194374e-05,
+      "loss": 0.0205,
+      "step": 1119
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1790281329923272e-05,
+      "loss": 0.0198,
+      "step": 1120
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1782608695652177e-05,
+      "loss": 0.0113,
+      "step": 1121
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 2.1774936061381075e-05,
+      "loss": 0.02,
+      "step": 1122
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1767263427109973e-05,
+      "loss": 0.0195,
+      "step": 1123
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1759590792838875e-05,
+      "loss": 0.8016,
+      "step": 1124
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1751918158567776e-05,
+      "loss": 0.0122,
+      "step": 1125
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 2.1744245524296675e-05,
+      "loss": 1.1132,
+      "step": 1126
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1736572890025576e-05,
+      "loss": 0.011,
+      "step": 1127
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1728900255754474e-05,
+      "loss": 0.7705,
+      "step": 1128
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1721227621483376e-05,
+      "loss": 1.1467,
+      "step": 1129
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.1713554987212278e-05,
+      "loss": 0.0176,
+      "step": 1130
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1705882352941176e-05,
+      "loss": 0.0293,
+      "step": 1131
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1698209718670074e-05,
+      "loss": 0.0208,
+      "step": 1132
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.169053708439898e-05,
+      "loss": 0.0199,
+      "step": 1133
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 2.1682864450127877e-05,
+      "loss": 0.0431,
+      "step": 1134
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.167519181585678e-05,
+      "loss": 1.6066,
+      "step": 1135
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.1667519181585677e-05,
+      "loss": 0.029,
+      "step": 1136
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165984654731458e-05,
+      "loss": 0.029,
+      "step": 1137
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 2.165217391304348e-05,
+      "loss": 1.1749,
+      "step": 1138
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.164450127877238e-05,
+      "loss": 0.0189,
+      "step": 1139
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.163682864450128e-05,
+      "loss": 0.0309,
+      "step": 1140
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.1629156010230182e-05,
+      "loss": 0.0287,
+      "step": 1141
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 2.162148337595908e-05,
+      "loss": 0.0202,
+      "step": 1142
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.161381074168798e-05,
+      "loss": 0.7889,
+      "step": 1143
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.1606138107416883e-05,
+      "loss": 0.0285,
+      "step": 1144
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159846547314578e-05,
+      "loss": 0.0535,
+      "step": 1145
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 2.159079283887468e-05,
+      "loss": 1.0829,
+      "step": 1146
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.158312020460358e-05,
+      "loss": 0.0989,
+      "step": 1147
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1575447570332483e-05,
+      "loss": 0.0115,
+      "step": 1148
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.156777493606138e-05,
+      "loss": 0.0227,
+      "step": 1149
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 2.1560102301790283e-05,
+      "loss": 0.5506,
+      "step": 1150
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.155242966751918e-05,
+      "loss": 1.1234,
+      "step": 1151
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1544757033248083e-05,
+      "loss": 0.0564,
+      "step": 1152
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1537084398976984e-05,
+      "loss": 0.0127,
+      "step": 1153
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 2.1529411764705882e-05,
+      "loss": 0.0218,
+      "step": 1154
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.152173913043478e-05,
+      "loss": 0.0124,
+      "step": 1155
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1514066496163686e-05,
+      "loss": 0.0223,
+      "step": 1156
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1506393861892584e-05,
+      "loss": 0.0505,
+      "step": 1157
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 2.1498721227621482e-05,
+      "loss": 0.0225,
+      "step": 1158
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1491048593350384e-05,
+      "loss": 0.0308,
+      "step": 1159
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1483375959079285e-05,
+      "loss": 0.036,
+      "step": 1160
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1475703324808184e-05,
+      "loss": 0.0121,
+      "step": 1161
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 2.1468030690537085e-05,
+      "loss": 0.0274,
+      "step": 1162
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1460358056265983e-05,
+      "loss": 1.0397,
+      "step": 1163
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1452685421994885e-05,
+      "loss": 0.0332,
+      "step": 1164
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1445012787723787e-05,
+      "loss": 0.0182,
+      "step": 1165
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 2.1437340153452685e-05,
+      "loss": 0.018,
+      "step": 1166
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1429667519181587e-05,
+      "loss": 0.859,
+      "step": 1167
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1421994884910488e-05,
+      "loss": 0.0321,
+      "step": 1168
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1414322250639386e-05,
+      "loss": 0.0225,
+      "step": 1169
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 2.1406649616368285e-05,
+      "loss": 0.0224,
+      "step": 1170
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.139897698209719e-05,
+      "loss": 0.03,
+      "step": 1171
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1391304347826088e-05,
+      "loss": 0.7473,
+      "step": 1172
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1383631713554986e-05,
+      "loss": 0.8609,
+      "step": 1173
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 2.1375959079283888e-05,
+      "loss": 0.0105,
+      "step": 1174
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.136828644501279e-05,
+      "loss": 0.0217,
+      "step": 1175
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1360613810741687e-05,
+      "loss": 0.0218,
+      "step": 1176
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.135294117647059e-05,
+      "loss": 0.016,
+      "step": 1177
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 2.1345268542199487e-05,
+      "loss": 0.7844,
+      "step": 1178
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.133759590792839e-05,
+      "loss": 0.01,
+      "step": 1179
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132992327365729e-05,
+      "loss": 0.8774,
+      "step": 1180
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.132225063938619e-05,
+      "loss": 0.8375,
+      "step": 1181
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 2.1314578005115087e-05,
+      "loss": 0.0188,
+      "step": 1182
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.1306905370843992e-05,
+      "loss": 1.1026,
+      "step": 1183
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129923273657289e-05,
+      "loss": 1.0447,
+      "step": 1184
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.129156010230179e-05,
+      "loss": 0.0101,
+      "step": 1185
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 2.128388746803069e-05,
+      "loss": 0.0247,
+      "step": 1186
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.127621483375959e-05,
+      "loss": 0.0295,
+      "step": 1187
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.1268542199488493e-05,
+      "loss": 0.0191,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.37390685081481934,
+      "eval_runtime": 57.4087,
+      "eval_samples_per_second": 6.898,
+      "eval_steps_per_second": 1.724,
+      "step": 1188
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 2.126086956521739e-05,
+      "loss": 0.0103,
+      "step": 1189
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1253196930946293e-05,
+      "loss": 0.0231,
+      "step": 1190
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1245524296675195e-05,
+      "loss": 0.0389,
+      "step": 1191
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.1237851662404093e-05,
+      "loss": 0.0298,
+      "step": 1192
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 2.123017902813299e-05,
+      "loss": 0.0196,
+      "step": 1193
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1222506393861896e-05,
+      "loss": 0.9688,
+      "step": 1194
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1214833759590794e-05,
+      "loss": 0.0458,
+      "step": 1195
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1207161125319693e-05,
+      "loss": 0.0161,
+      "step": 1196
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 2.1199488491048594e-05,
+      "loss": 0.0157,
+      "step": 1197
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1191815856777496e-05,
+      "loss": 0.8732,
+      "step": 1198
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1184143222506394e-05,
+      "loss": 0.0103,
+      "step": 1199
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1176470588235296e-05,
+      "loss": 0.0267,
+      "step": 1200
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 2.1168797953964194e-05,
+      "loss": 0.0266,
+      "step": 1201
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1161125319693095e-05,
+      "loss": 0.0157,
+      "step": 1202
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1153452685421997e-05,
+      "loss": 0.265,
+      "step": 1203
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1145780051150895e-05,
+      "loss": 0.0537,
+      "step": 1204
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 2.1138107416879794e-05,
+      "loss": 0.0306,
+      "step": 1205
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.11304347826087e-05,
+      "loss": 1.1714,
+      "step": 1206
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1122762148337597e-05,
+      "loss": 1.1704,
+      "step": 1207
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1115089514066495e-05,
+      "loss": 0.0154,
+      "step": 1208
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 2.1107416879795397e-05,
+      "loss": 0.0195,
+      "step": 1209
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1099744245524298e-05,
+      "loss": 0.9099,
+      "step": 1210
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1092071611253196e-05,
+      "loss": 0.0112,
+      "step": 1211
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1084398976982098e-05,
+      "loss": 0.1506,
+      "step": 1212
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 2.1076726342710996e-05,
+      "loss": 0.0111,
+      "step": 1213
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1069053708439898e-05,
+      "loss": 0.0244,
+      "step": 1214
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10613810741688e-05,
+      "loss": 0.8889,
+      "step": 1215
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.1053708439897698e-05,
+      "loss": 0.0146,
+      "step": 1216
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 2.10460358056266e-05,
+      "loss": 0.0189,
+      "step": 1217
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10383631713555e-05,
+      "loss": 0.0097,
+      "step": 1218
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.10306905370844e-05,
+      "loss": 1.1683,
+      "step": 1219
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1023017902813297e-05,
+      "loss": 1.1621,
+      "step": 1220
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 2.1015345268542202e-05,
+      "loss": 0.0194,
+      "step": 1221
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.10076726342711e-05,
+      "loss": 0.0155,
+      "step": 1222
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.0296,
+      "step": 1223
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.09923273657289e-05,
+      "loss": 0.02,
+      "step": 1224
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 2.0984654731457802e-05,
+      "loss": 0.0154,
+      "step": 1225
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09769820971867e-05,
+      "loss": 0.0151,
+      "step": 1226
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0969309462915602e-05,
+      "loss": 0.0223,
+      "step": 1227
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.09616368286445e-05,
+      "loss": 1.7159,
+      "step": 1228
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 2.0953964194373402e-05,
+      "loss": 0.0194,
+      "step": 1229
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0946291560102303e-05,
+      "loss": 0.0637,
+      "step": 1230
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09386189258312e-05,
+      "loss": 0.0156,
+      "step": 1231
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.09309462915601e-05,
+      "loss": 1.0612,
+      "step": 1232
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 2.0923273657289005e-05,
+      "loss": 0.8929,
+      "step": 1233
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0915601023017903e-05,
+      "loss": 1.1649,
+      "step": 1234
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.09079283887468e-05,
+      "loss": 2.059,
+      "step": 1235
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0900255754475703e-05,
+      "loss": 0.0169,
+      "step": 1236
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 2.0892583120204604e-05,
+      "loss": 0.0158,
+      "step": 1237
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0884910485933503e-05,
+      "loss": 0.0108,
+      "step": 1238
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0877237851662404e-05,
+      "loss": 0.0152,
+      "step": 1239
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0869565217391306e-05,
+      "loss": 0.0109,
+      "step": 1240
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 2.0861892583120208e-05,
+      "loss": 1.1352,
+      "step": 1241
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0854219948849106e-05,
+      "loss": 0.0226,
+      "step": 1242
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0846547314578004e-05,
+      "loss": 0.0154,
+      "step": 1243
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.083887468030691e-05,
+      "loss": 0.0232,
+      "step": 1244
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 2.0831202046035807e-05,
+      "loss": 0.0142,
+      "step": 1245
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0823529411764705e-05,
+      "loss": 0.9143,
+      "step": 1246
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0815856777493607e-05,
+      "loss": 0.0165,
+      "step": 1247
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.080818414322251e-05,
+      "loss": 0.0446,
+      "step": 1248
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 2.0800511508951407e-05,
+      "loss": 0.0121,
+      "step": 1249
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.079283887468031e-05,
+      "loss": 1.0991,
+      "step": 1250
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.0785166240409207e-05,
+      "loss": 0.0159,
+      "step": 1251
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.077749360613811e-05,
+      "loss": 0.0166,
+      "step": 1252
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 2.076982097186701e-05,
+      "loss": 0.0171,
+      "step": 1253
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0762148337595908e-05,
+      "loss": 0.7986,
+      "step": 1254
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.0754475703324806e-05,
+      "loss": 0.0174,
+      "step": 1255
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.074680306905371e-05,
+      "loss": 1.0431,
+      "step": 1256
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 2.073913043478261e-05,
+      "loss": 0.1766,
+      "step": 1257
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.0731457800511508e-05,
+      "loss": 0.0209,
+      "step": 1258
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.072378516624041e-05,
+      "loss": 0.0178,
+      "step": 1259
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.071611253196931e-05,
+      "loss": 0.0215,
+      "step": 1260
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 2.070843989769821e-05,
+      "loss": 0.0112,
+      "step": 1261
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.070076726342711e-05,
+      "loss": 0.0166,
+      "step": 1262
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0693094629156012e-05,
+      "loss": 0.0161,
+      "step": 1263
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.068542199488491e-05,
+      "loss": 0.8922,
+      "step": 1264
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 2.0677749360613812e-05,
+      "loss": 0.0133,
+      "step": 1265
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.067007672634271e-05,
+      "loss": 0.0117,
+      "step": 1266
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0662404092071612e-05,
+      "loss": 0.0153,
+      "step": 1267
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0654731457800514e-05,
+      "loss": 0.0153,
+      "step": 1268
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 2.0647058823529412e-05,
+      "loss": 0.0195,
+      "step": 1269
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.063938618925831e-05,
+      "loss": 1.1511,
+      "step": 1270
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0631713554987215e-05,
+      "loss": 0.0108,
+      "step": 1271
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.0624040920716113e-05,
+      "loss": 0.8897,
+      "step": 1272
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 2.061636828644501e-05,
+      "loss": 0.0157,
+      "step": 1273
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0608695652173913e-05,
+      "loss": 0.0108,
+      "step": 1274
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0601023017902815e-05,
+      "loss": 0.0214,
+      "step": 1275
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0593350383631713e-05,
+      "loss": 0.0186,
+      "step": 1276
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 2.0585677749360615e-05,
+      "loss": 0.0182,
+      "step": 1277
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0578005115089513e-05,
+      "loss": 0.0102,
+      "step": 1278
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0570332480818415e-05,
+      "loss": 1.1313,
+      "step": 1279
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0562659846547316e-05,
+      "loss": 0.0103,
+      "step": 1280
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 2.0554987212276214e-05,
+      "loss": 0.0144,
+      "step": 1281
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0547314578005113e-05,
+      "loss": 0.02,
+      "step": 1282
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0539641943734018e-05,
+      "loss": 0.0292,
+      "step": 1283
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0531969309462916e-05,
+      "loss": 0.0105,
+      "step": 1284
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 2.0524296675191814e-05,
+      "loss": 0.01,
+      "step": 1285
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0516624040920716e-05,
+      "loss": 1.1245,
+      "step": 1286
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0508951406649617e-05,
+      "loss": 0.8544,
+      "step": 1287
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 2.0501278772378516e-05,
+      "loss": 0.0149,
+      "step": 1288
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0493606138107417e-05,
+      "loss": 0.0065,
+      "step": 1289
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.048593350383632e-05,
+      "loss": 0.0113,
+      "step": 1290
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.0478260869565217e-05,
+      "loss": 0.0094,
+      "step": 1291
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 2.047058823529412e-05,
+      "loss": 0.0097,
+      "step": 1292
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0462915601023017e-05,
+      "loss": 0.0812,
+      "step": 1293
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0455242966751922e-05,
+      "loss": 0.0099,
+      "step": 1294
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.044757033248082e-05,
+      "loss": 0.013,
+      "step": 1295
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 2.0439897698209718e-05,
+      "loss": 1.584,
+      "step": 1296
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.043222506393862e-05,
+      "loss": 0.0089,
+      "step": 1297
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.042455242966752e-05,
+      "loss": 0.0089,
+      "step": 1298
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.041687979539642e-05,
+      "loss": 0.0093,
+      "step": 1299
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 2.040920716112532e-05,
+      "loss": 0.0149,
+      "step": 1300
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.040153452685422e-05,
+      "loss": 1.1671,
+      "step": 1301
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.039386189258312e-05,
+      "loss": 0.0086,
+      "step": 1302
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.0386189258312023e-05,
+      "loss": 0.8177,
+      "step": 1303
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 2.037851662404092e-05,
+      "loss": 0.0128,
+      "step": 1304
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.037084398976982e-05,
+      "loss": 0.0167,
+      "step": 1305
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0363171355498724e-05,
+      "loss": 0.0085,
+      "step": 1306
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.0355498721227622e-05,
+      "loss": 0.0085,
+      "step": 1307
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 2.034782608695652e-05,
+      "loss": 0.0166,
+      "step": 1308
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0340153452685422e-05,
+      "loss": 0.9448,
+      "step": 1309
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0332480818414324e-05,
+      "loss": 0.0082,
+      "step": 1310
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0324808184143222e-05,
+      "loss": 0.0088,
+      "step": 1311
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 2.0317135549872124e-05,
+      "loss": 0.0082,
+      "step": 1312
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0309462915601025e-05,
+      "loss": 0.0146,
+      "step": 1313
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0301790281329924e-05,
+      "loss": 0.0123,
+      "step": 1314
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0294117647058825e-05,
+      "loss": 0.0101,
+      "step": 1315
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 2.0286445012787723e-05,
+      "loss": 1.2167,
+      "step": 1316
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0278772378516625e-05,
+      "loss": 1.0578,
+      "step": 1317
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0271099744245527e-05,
+      "loss": 0.0163,
+      "step": 1318
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0263427109974425e-05,
+      "loss": 0.013,
+      "step": 1319
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 2.0255754475703323e-05,
+      "loss": 0.0082,
+      "step": 1320
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0248081841432228e-05,
+      "loss": 0.0081,
+      "step": 1321
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0240409207161126e-05,
+      "loss": 0.0124,
+      "step": 1322
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0232736572890024e-05,
+      "loss": 0.0206,
+      "step": 1323
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 2.0225063938618926e-05,
+      "loss": 0.0081,
+      "step": 1324
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0217391304347828e-05,
+      "loss": 0.0122,
+      "step": 1325
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0209718670076726e-05,
+      "loss": 0.0122,
+      "step": 1326
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0202046035805628e-05,
+      "loss": 0.0121,
+      "step": 1327
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 2.0194373401534526e-05,
+      "loss": 1.224,
+      "step": 1328
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0186700767263427e-05,
+      "loss": 2.3502,
+      "step": 1329
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.017902813299233e-05,
+      "loss": 0.0242,
+      "step": 1330
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0171355498721227e-05,
+      "loss": 0.0064,
+      "step": 1331
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 2.0163682864450125e-05,
+      "loss": 0.0186,
+      "step": 1332
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.015601023017903e-05,
+      "loss": 0.012,
+      "step": 1333
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.014833759590793e-05,
+      "loss": 0.0119,
+      "step": 1334
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.0140664961636827e-05,
+      "loss": 0.0159,
+      "step": 1335
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 2.013299232736573e-05,
+      "loss": 0.0199,
+      "step": 1336
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.012531969309463e-05,
+      "loss": 1.2138,
+      "step": 1337
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.011764705882353e-05,
+      "loss": 0.9491,
+      "step": 1338
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010997442455243e-05,
+      "loss": 0.0083,
+      "step": 1339
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 2.010230179028133e-05,
+      "loss": 0.948,
+      "step": 1340
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.009462915601023e-05,
+      "loss": 0.0112,
+      "step": 1341
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.008695652173913e-05,
+      "loss": 0.8921,
+      "step": 1342
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007928388746803e-05,
+      "loss": 0.0194,
+      "step": 1343
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 2.007161125319693e-05,
+      "loss": 1.0273,
+      "step": 1344
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0063938618925833e-05,
+      "loss": 0.013,
+      "step": 1345
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.005626598465473e-05,
+      "loss": 0.0103,
+      "step": 1346
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0048593350383633e-05,
+      "loss": 0.0114,
+      "step": 1347
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 2.0040920716112534e-05,
+      "loss": 0.0199,
+      "step": 1348
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0033248081841433e-05,
+      "loss": 1.183,
+      "step": 1349
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0025575447570334e-05,
+      "loss": 0.0231,
+      "step": 1350
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0017902813299232e-05,
+      "loss": 1.1794,
+      "step": 1351
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 2.0010230179028134e-05,
+      "loss": 0.0096,
+      "step": 1352
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 2.0002557544757036e-05,
+      "loss": 0.9049,
+      "step": 1353
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9994884910485934e-05,
+      "loss": 0.019,
+      "step": 1354
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9987212276214832e-05,
+      "loss": 0.0152,
+      "step": 1355
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 1.9979539641943737e-05,
+      "loss": 0.8834,
+      "step": 1356
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9971867007672635e-05,
+      "loss": 0.01,
+      "step": 1357
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9964194373401533e-05,
+      "loss": 0.0152,
+      "step": 1358
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9956521739130435e-05,
+      "loss": 0.0161,
+      "step": 1359
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 1.9948849104859337e-05,
+      "loss": 1.6778,
+      "step": 1360
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9941176470588235e-05,
+      "loss": 0.0214,
+      "step": 1361
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9933503836317137e-05,
+      "loss": 0.0156,
+      "step": 1362
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9925831202046038e-05,
+      "loss": 0.026,
+      "step": 1363
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 1.9918158567774936e-05,
+      "loss": 0.0259,
+      "step": 1364
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9910485933503838e-05,
+      "loss": 0.0161,
+      "step": 1365
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9902813299232736e-05,
+      "loss": 0.0108,
+      "step": 1366
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.9895140664961638e-05,
+      "loss": 0.0149,
+      "step": 1367
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.988746803069054e-05,
+      "loss": 0.0325,
+      "step": 1368
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9879795396419438e-05,
+      "loss": 0.0153,
+      "step": 1369
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.9872122762148336e-05,
+      "loss": 1.7678,
+      "step": 1370
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.986445012787724e-05,
+      "loss": 0.0773,
+      "step": 1371
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 1.985677749360614e-05,
+      "loss": 0.0244,
+      "step": 1372
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.9849104859335037e-05,
+      "loss": 0.0225,
+      "step": 1373
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.984143222506394e-05,
+      "loss": 0.0254,
+      "step": 1374
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.983375959079284e-05,
+      "loss": 0.0205,
+      "step": 1375
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 1.982608695652174e-05,
+      "loss": 0.0096,
+      "step": 1376
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981841432225064e-05,
+      "loss": 0.0147,
+      "step": 1377
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.981074168797954e-05,
+      "loss": 0.0296,
+      "step": 1378
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.980306905370844e-05,
+      "loss": 0.0227,
+      "step": 1379
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 1.9795396419437342e-05,
+      "loss": 0.899,
+      "step": 1380
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.978772378516624e-05,
+      "loss": 0.019,
+      "step": 1381
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9780051150895138e-05,
+      "loss": 0.0157,
+      "step": 1382
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.9772378516624043e-05,
+      "loss": 1.1897,
+      "step": 1383
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 1.976470588235294e-05,
+      "loss": 0.014,
+      "step": 1384
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.975703324808184e-05,
+      "loss": 0.0208,
+      "step": 1385
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.974936061381074e-05,
+      "loss": 0.0093,
+      "step": 1386
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 1.9741687979539643e-05,
+      "loss": 0.0244,
+      "step": 1387
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.973401534526854e-05,
+      "loss": 0.028,
+      "step": 1388
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9726342710997443e-05,
+      "loss": 0.0093,
+      "step": 1389
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9718670076726344e-05,
+      "loss": 0.0184,
+      "step": 1390
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 1.9710997442455243e-05,
+      "loss": 0.9462,
+      "step": 1391
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9703324808184144e-05,
+      "loss": 0.0138,
+      "step": 1392
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9695652173913042e-05,
+      "loss": 0.9561,
+      "step": 1393
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9687979539641944e-05,
+      "loss": 0.0152,
+      "step": 1394
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 1.9680306905370846e-05,
+      "loss": 0.0178,
+      "step": 1395
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9672634271099744e-05,
+      "loss": 0.0166,
+      "step": 1396
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9664961636828642e-05,
+      "loss": 0.92,
+      "step": 1397
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9657289002557547e-05,
+      "loss": 0.0253,
+      "step": 1398
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 1.9649616368286445e-05,
+      "loss": 0.0266,
+      "step": 1399
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9641943734015347e-05,
+      "loss": 0.0105,
+      "step": 1400
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9634271099744245e-05,
+      "loss": 0.0208,
+      "step": 1401
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.9626598465473147e-05,
+      "loss": 0.8094,
+      "step": 1402
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 1.961892583120205e-05,
+      "loss": 0.0103,
+      "step": 1403
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9611253196930947e-05,
+      "loss": 0.0179,
+      "step": 1404
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9603580562659845e-05,
+      "loss": 0.0133,
+      "step": 1405
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.959590792838875e-05,
+      "loss": 0.0086,
+      "step": 1406
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.9588235294117648e-05,
+      "loss": 0.0216,
+      "step": 1407
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9580562659846546e-05,
+      "loss": 0.0127,
+      "step": 1408
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9572890025575448e-05,
+      "loss": 0.0127,
+      "step": 1409
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.956521739130435e-05,
+      "loss": 0.0209,
+      "step": 1410
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 1.9557544757033248e-05,
+      "loss": 0.0166,
+      "step": 1411
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954987212276215e-05,
+      "loss": 0.0123,
+      "step": 1412
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.954219948849105e-05,
+      "loss": 0.0135,
+      "step": 1413
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.953452685421995e-05,
+      "loss": 0.1253,
+      "step": 1414
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 1.952685421994885e-05,
+      "loss": 0.0098,
+      "step": 1415
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951918158567775e-05,
+      "loss": 0.0091,
+      "step": 1416
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.951150895140665e-05,
+      "loss": 0.9327,
+      "step": 1417
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.9503836317135552e-05,
+      "loss": 0.0119,
+      "step": 1418
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 1.949616368286445e-05,
+      "loss": 0.0153,
+      "step": 1419
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.948849104859335e-05,
+      "loss": 0.0123,
+      "step": 1420
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9480818414322254e-05,
+      "loss": 0.0194,
+      "step": 1421
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.9473145780051152e-05,
+      "loss": 1.2311,
+      "step": 1422
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 1.946547314578005e-05,
+      "loss": 1.8519,
+      "step": 1423
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9457800511508952e-05,
+      "loss": 0.963,
+      "step": 1424
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9450127877237853e-05,
+      "loss": 0.0075,
+      "step": 1425
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.944245524296675e-05,
+      "loss": 0.0042,
+      "step": 1426
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 1.9434782608695653e-05,
+      "loss": 0.0077,
+      "step": 1427
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.942710997442455e-05,
+      "loss": 0.0152,
+      "step": 1428
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9419437340153453e-05,
+      "loss": 0.0152,
+      "step": 1429
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9411764705882355e-05,
+      "loss": 0.019,
+      "step": 1430
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 1.9404092071611253e-05,
+      "loss": 0.0081,
+      "step": 1431
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.939641943734015e-05,
+      "loss": 0.0076,
+      "step": 1432
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9388746803069056e-05,
+      "loss": 0.0086,
+      "step": 1433
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9381074168797954e-05,
+      "loss": 0.0146,
+      "step": 1434
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 1.9373401534526853e-05,
+      "loss": 0.0074,
+      "step": 1435
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9365728900255754e-05,
+      "loss": 0.1007,
+      "step": 1436
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9358056265984656e-05,
+      "loss": 0.011,
+      "step": 1437
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9350383631713554e-05,
+      "loss": 0.0089,
+      "step": 1438
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 1.9342710997442456e-05,
+      "loss": 0.0182,
+      "step": 1439
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9335038363171357e-05,
+      "loss": 0.0139,
+      "step": 1440
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9327365728900255e-05,
+      "loss": 2.2086,
+      "step": 1441
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9319693094629157e-05,
+      "loss": 0.0178,
+      "step": 1442
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.9312020460358055e-05,
+      "loss": 0.0142,
+      "step": 1443
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9304347826086957e-05,
+      "loss": 0.0142,
+      "step": 1444
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.929667519181586e-05,
+      "loss": 0.0242,
+      "step": 1445
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9289002557544757e-05,
+      "loss": 0.0087,
+      "step": 1446
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 1.9281329923273655e-05,
+      "loss": 0.0077,
+      "step": 1447
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.927365728900256e-05,
+      "loss": 0.0104,
+      "step": 1448
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9265984654731458e-05,
+      "loss": 0.0139,
+      "step": 1449
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9258312020460356e-05,
+      "loss": 1.2557,
+      "step": 1450
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 1.9250639386189258e-05,
+      "loss": 0.0068,
+      "step": 1451
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.924296675191816e-05,
+      "loss": 0.0163,
+      "step": 1452
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.923529411764706e-05,
+      "loss": 0.9483,
+      "step": 1453
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.922762148337596e-05,
+      "loss": 1.2181,
+      "step": 1454
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 1.9219948849104858e-05,
+      "loss": 0.011,
+      "step": 1455
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.9212276214833763e-05,
+      "loss": 0.014,
+      "step": 1456
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.920460358056266e-05,
+      "loss": 0.0131,
+      "step": 1457
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.919693094629156e-05,
+      "loss": 0.0135,
+      "step": 1458
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 1.918925831202046e-05,
+      "loss": 0.007,
+      "step": 1459
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9181585677749362e-05,
+      "loss": 0.0072,
+      "step": 1460
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.917391304347826e-05,
+      "loss": 0.0135,
+      "step": 1461
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9166240409207162e-05,
+      "loss": 0.9908,
+      "step": 1462
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 1.9158567774936064e-05,
+      "loss": 0.0103,
+      "step": 1463
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9150895140664962e-05,
+      "loss": 0.9867,
+      "step": 1464
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9143222506393864e-05,
+      "loss": 0.0138,
+      "step": 1465
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9135549872122762e-05,
+      "loss": 0.0104,
+      "step": 1466
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 1.9127877237851663e-05,
+      "loss": 0.014,
+      "step": 1467
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9120204603580565e-05,
+      "loss": 0.0071,
+      "step": 1468
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9112531969309463e-05,
+      "loss": 0.0108,
+      "step": 1469
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.910485933503836e-05,
+      "loss": 0.0106,
+      "step": 1470
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 1.9097186700767266e-05,
+      "loss": 0.0404,
+      "step": 1471
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9089514066496165e-05,
+      "loss": 2.2011,
+      "step": 1472
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9081841432225063e-05,
+      "loss": 0.0087,
+      "step": 1473
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9074168797953965e-05,
+      "loss": 0.0071,
+      "step": 1474
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 1.9066496163682866e-05,
+      "loss": 0.0072,
+      "step": 1475
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9058823529411764e-05,
+      "loss": 0.0108,
+      "step": 1476
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9051150895140666e-05,
+      "loss": 0.0108,
+      "step": 1477
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9043478260869564e-05,
+      "loss": 0.0072,
+      "step": 1478
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.9035805626598466e-05,
+      "loss": 0.0108,
+      "step": 1479
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9028132992327367e-05,
+      "loss": 0.0073,
+      "step": 1480
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9020460358056266e-05,
+      "loss": 0.9537,
+      "step": 1481
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.9012787723785164e-05,
+      "loss": 0.0107,
+      "step": 1482
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 1.900511508951407e-05,
+      "loss": 0.0145,
+      "step": 1483
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8997442455242967e-05,
+      "loss": 0.007,
+      "step": 1484
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.8989769820971865e-05,
+      "loss": 0.0071,
+      "step": 1485
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 1.898209718670077e-05,
+      "loss": 1.243,
+      "step": 1486
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.897442455242967e-05,
+      "loss": 0.011,
+      "step": 1487
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.8966751918158567e-05,
+      "loss": 0.0077,
+      "step": 1488
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895907928388747e-05,
+      "loss": 0.0073,
+      "step": 1489
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 1.895140664961637e-05,
+      "loss": 1.2486,
+      "step": 1490
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8943734015345268e-05,
+      "loss": 0.0109,
+      "step": 1491
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.893606138107417e-05,
+      "loss": 0.0149,
+      "step": 1492
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.8928388746803068e-05,
+      "loss": 0.0125,
+      "step": 1493
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 1.892071611253197e-05,
+      "loss": 0.0071,
+      "step": 1494
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.891304347826087e-05,
+      "loss": 0.0036,
+      "step": 1495
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.890537084398977e-05,
+      "loss": 0.0149,
+      "step": 1496
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8897698209718668e-05,
+      "loss": 0.015,
+      "step": 1497
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 1.8890025575447573e-05,
+      "loss": 0.9536,
+      "step": 1498
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.888235294117647e-05,
+      "loss": 0.9997,
+      "step": 1499
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.887468030690537e-05,
+      "loss": 0.9621,
+      "step": 1500
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.886700767263427e-05,
+      "loss": 0.0113,
+      "step": 1501
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 1.8859335038363172e-05,
+      "loss": 0.2056,
+      "step": 1502
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.885166240409207e-05,
+      "loss": 0.0114,
+      "step": 1503
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8843989769820972e-05,
+      "loss": 0.016,
+      "step": 1504
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.883631713554987e-05,
+      "loss": 0.9175,
+      "step": 1505
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 1.8828644501278775e-05,
+      "loss": 0.8959,
+      "step": 1506
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8820971867007674e-05,
+      "loss": 0.0074,
+      "step": 1507
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8813299232736572e-05,
+      "loss": 0.0117,
+      "step": 1508
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8805626598465474e-05,
+      "loss": 0.0075,
+      "step": 1509
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 1.8797953964194375e-05,
+      "loss": 0.0074,
+      "step": 1510
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8790281329923273e-05,
+      "loss": 0.0577,
+      "step": 1511
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8782608695652175e-05,
+      "loss": 0.0128,
+      "step": 1512
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8774936061381077e-05,
+      "loss": 0.0073,
+      "step": 1513
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.8767263427109975e-05,
+      "loss": 0.0074,
+      "step": 1514
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8759590792838876e-05,
+      "loss": 0.0073,
+      "step": 1515
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8751918158567775e-05,
+      "loss": 0.0088,
+      "step": 1516
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8744245524296676e-05,
+      "loss": 0.0078,
+      "step": 1517
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 1.8736572890025578e-05,
+      "loss": 0.016,
+      "step": 1518
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8728900255754476e-05,
+      "loss": 1.3842,
+      "step": 1519
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8721227621483374e-05,
+      "loss": 1.2502,
+      "step": 1520
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.871355498721228e-05,
+      "loss": 0.0117,
+      "step": 1521
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 1.8705882352941178e-05,
+      "loss": 0.0113,
+      "step": 1522
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8698209718670076e-05,
+      "loss": 0.0125,
+      "step": 1523
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8690537084398977e-05,
+      "loss": 0.0153,
+      "step": 1524
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.868286445012788e-05,
+      "loss": 0.0072,
+      "step": 1525
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 1.8675191815856777e-05,
+      "loss": 0.0086,
+      "step": 1526
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.866751918158568e-05,
+      "loss": 0.0195,
+      "step": 1527
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.8659846547314577e-05,
+      "loss": 1.2511,
+      "step": 1528
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.865217391304348e-05,
+      "loss": 1.2407,
+      "step": 1529
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 1.864450127877238e-05,
+      "loss": 0.0112,
+      "step": 1530
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.863682864450128e-05,
+      "loss": 0.0073,
+      "step": 1531
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8629156010230177e-05,
+      "loss": 1.1911,
+      "step": 1532
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.8621483375959082e-05,
+      "loss": 0.0168,
+      "step": 1533
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 1.861381074168798e-05,
+      "loss": 0.0128,
+      "step": 1534
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8606138107416878e-05,
+      "loss": 0.0116,
+      "step": 1535
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.8598465473145783e-05,
+      "loss": 1.2286,
+      "step": 1536
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.859079283887468e-05,
+      "loss": 0.0158,
+      "step": 1537
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 1.858312020460358e-05,
+      "loss": 2.3735,
+      "step": 1538
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.857544757033248e-05,
+      "loss": 1.1972,
+      "step": 1539
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8567774936061383e-05,
+      "loss": 0.011,
+      "step": 1540
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.856010230179028e-05,
+      "loss": 0.0127,
+      "step": 1541
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 1.8552429667519183e-05,
+      "loss": 0.0148,
+      "step": 1542
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.854475703324808e-05,
+      "loss": 0.0111,
+      "step": 1543
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8537084398976983e-05,
+      "loss": 0.0143,
+      "step": 1544
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8529411764705884e-05,
+      "loss": 0.0132,
+      "step": 1545
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 1.8521739130434782e-05,
+      "loss": 0.0152,
+      "step": 1546
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.851406649616368e-05,
+      "loss": 0.0152,
+      "step": 1547
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8506393861892586e-05,
+      "loss": 0.0051,
+      "step": 1548
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8498721227621484e-05,
+      "loss": 0.9576,
+      "step": 1549
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.8491048593350382e-05,
+      "loss": 0.0181,
+      "step": 1550
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8483375959079284e-05,
+      "loss": 0.011,
+      "step": 1551
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8475703324808185e-05,
+      "loss": 0.0205,
+      "step": 1552
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8468030690537083e-05,
+      "loss": 2.5631,
+      "step": 1553
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 1.8460358056265985e-05,
+      "loss": 0.0102,
+      "step": 1554
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8452685421994883e-05,
+      "loss": 0.0147,
+      "step": 1555
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8445012787723788e-05,
+      "loss": 0.015,
+      "step": 1556
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8437340153452687e-05,
+      "loss": 0.0122,
+      "step": 1557
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 1.8429667519181585e-05,
+      "loss": 0.0176,
+      "step": 1558
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8421994884910486e-05,
+      "loss": 0.0114,
+      "step": 1559
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8414322250639388e-05,
+      "loss": 0.0252,
+      "step": 1560
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8406649616368286e-05,
+      "loss": 0.013,
+      "step": 1561
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 1.8398976982097188e-05,
+      "loss": 0.0142,
+      "step": 1562
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.839130434782609e-05,
+      "loss": 0.6496,
+      "step": 1563
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8383631713554988e-05,
+      "loss": 0.0149,
+      "step": 1564
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.837595907928389e-05,
+      "loss": 0.0104,
+      "step": 1565
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 1.8368286445012787e-05,
+      "loss": 0.0129,
+      "step": 1566
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.836061381074169e-05,
+      "loss": 0.0197,
+      "step": 1567
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.835294117647059e-05,
+      "loss": 0.009,
+      "step": 1568
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.834526854219949e-05,
+      "loss": 0.0086,
+      "step": 1569
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 1.8337595907928387e-05,
+      "loss": 0.0087,
+      "step": 1570
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.8329923273657292e-05,
+      "loss": 0.0131,
+      "step": 1571
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.832225063938619e-05,
+      "loss": 1.2046,
+      "step": 1572
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.831457800511509e-05,
+      "loss": 0.0135,
+      "step": 1573
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 1.830690537084399e-05,
+      "loss": 0.0149,
+      "step": 1574
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.8299232736572892e-05,
+      "loss": 0.0077,
+      "step": 1575
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.829156010230179e-05,
+      "loss": 0.0143,
+      "step": 1576
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.828388746803069e-05,
+      "loss": 0.9806,
+      "step": 1577
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 1.827621483375959e-05,
+      "loss": 1.3783,
+      "step": 1578
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.826854219948849e-05,
+      "loss": 0.0075,
+      "step": 1579
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.8260869565217393e-05,
+      "loss": 0.0111,
+      "step": 1580
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.825319693094629e-05,
+      "loss": 0.0111,
+      "step": 1581
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 1.824552429667519e-05,
+      "loss": 0.0144,
+      "step": 1582
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8237851662404095e-05,
+      "loss": 0.0128,
+      "step": 1583
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.8230179028132993e-05,
+      "loss": 1.2949,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.40147146582603455,
+      "eval_runtime": 58.0487,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1584
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.822250639386189e-05,
+      "loss": 0.0075,
+      "step": 1585
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8214833759590796e-05,
+      "loss": 0.0108,
+      "step": 1586
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8207161125319694e-05,
+      "loss": 0.011,
+      "step": 1587
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8199488491048592e-05,
+      "loss": 0.0148,
+      "step": 1588
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 1.8191815856777494e-05,
+      "loss": 0.0077,
+      "step": 1589
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8184143222506396e-05,
+      "loss": 0.6674,
+      "step": 1590
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8176470588235294e-05,
+      "loss": 0.0109,
+      "step": 1591
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8168797953964195e-05,
+      "loss": 0.0108,
+      "step": 1592
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 1.8161125319693094e-05,
+      "loss": 0.011,
+      "step": 1593
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8153452685421995e-05,
+      "loss": 0.0143,
+      "step": 1594
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8145780051150897e-05,
+      "loss": 0.9728,
+      "step": 1595
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8138107416879795e-05,
+      "loss": 0.0074,
+      "step": 1596
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 1.8130434782608693e-05,
+      "loss": 0.0087,
+      "step": 1597
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.81227621483376e-05,
+      "loss": 0.0107,
+      "step": 1598
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8115089514066497e-05,
+      "loss": 0.0077,
+      "step": 1599
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8107416879795395e-05,
+      "loss": 0.0167,
+      "step": 1600
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 1.8099744245524296e-05,
+      "loss": 1.219,
+      "step": 1601
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8092071611253198e-05,
+      "loss": 1.2244,
+      "step": 1602
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8084398976982096e-05,
+      "loss": 0.0108,
+      "step": 1603
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8076726342710998e-05,
+      "loss": 0.0071,
+      "step": 1604
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 1.8069053708439896e-05,
+      "loss": 0.0109,
+      "step": 1605
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.8061381074168798e-05,
+      "loss": 0.0185,
+      "step": 1606
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.80537084398977e-05,
+      "loss": 1.2711,
+      "step": 1607
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.8046035805626598e-05,
+      "loss": 0.0113,
+      "step": 1608
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 1.80383631713555e-05,
+      "loss": 0.9575,
+      "step": 1609
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80306905370844e-05,
+      "loss": 1.1369,
+      "step": 1610
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80230179028133e-05,
+      "loss": 0.0114,
+      "step": 1611
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.80153452685422e-05,
+      "loss": 1.578,
+      "step": 1612
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 1.8007672634271102e-05,
+      "loss": 0.0076,
+      "step": 1613
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.0128,
+      "step": 1614
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.7992327365728902e-05,
+      "loss": 0.0157,
+      "step": 1615
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.79846547314578e-05,
+      "loss": 0.0154,
+      "step": 1616
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 1.7976982097186702e-05,
+      "loss": 0.0125,
+      "step": 1617
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7969309462915604e-05,
+      "loss": 0.0085,
+      "step": 1618
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7961636828644502e-05,
+      "loss": 0.0167,
+      "step": 1619
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.79539641943734e-05,
+      "loss": 1.2275,
+      "step": 1620
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.7946291560102305e-05,
+      "loss": 0.0115,
+      "step": 1621
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7938618925831203e-05,
+      "loss": 0.0083,
+      "step": 1622
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.79309462915601e-05,
+      "loss": 0.015,
+      "step": 1623
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7923273657289003e-05,
+      "loss": 1.1705,
+      "step": 1624
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 1.7915601023017905e-05,
+      "loss": 0.0152,
+      "step": 1625
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7907928388746803e-05,
+      "loss": 2.0998,
+      "step": 1626
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7900255754475704e-05,
+      "loss": 0.0072,
+      "step": 1627
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7892583120204603e-05,
+      "loss": 0.0089,
+      "step": 1628
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 1.7884910485933504e-05,
+      "loss": 0.0077,
+      "step": 1629
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7877237851662406e-05,
+      "loss": 0.0126,
+      "step": 1630
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7869565217391304e-05,
+      "loss": 1.2113,
+      "step": 1631
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7861892583120202e-05,
+      "loss": 0.0112,
+      "step": 1632
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 1.7854219948849107e-05,
+      "loss": 0.0099,
+      "step": 1633
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7846547314578006e-05,
+      "loss": 0.0074,
+      "step": 1634
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7838874680306904e-05,
+      "loss": 0.0111,
+      "step": 1635
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.783120204603581e-05,
+      "loss": 0.0075,
+      "step": 1636
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 1.7823529411764707e-05,
+      "loss": 1.2187,
+      "step": 1637
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7815856777493605e-05,
+      "loss": 0.0111,
+      "step": 1638
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7808184143222507e-05,
+      "loss": 0.0184,
+      "step": 1639
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.780051150895141e-05,
+      "loss": 0.0111,
+      "step": 1640
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 1.7792838874680307e-05,
+      "loss": 0.0112,
+      "step": 1641
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.778516624040921e-05,
+      "loss": 0.0149,
+      "step": 1642
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.7777493606138107e-05,
+      "loss": 0.0151,
+      "step": 1643
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.7769820971867008e-05,
+      "loss": 0.0074,
+      "step": 1644
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 1.776214833759591e-05,
+      "loss": 0.9703,
+      "step": 1645
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.7754475703324808e-05,
+      "loss": 1.9364,
+      "step": 1646
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.7746803069053706e-05,
+      "loss": 0.011,
+      "step": 1647
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.773913043478261e-05,
+      "loss": 0.01,
+      "step": 1648
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 1.773145780051151e-05,
+      "loss": 0.0111,
+      "step": 1649
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.7723785166240408e-05,
+      "loss": 1.2373,
+      "step": 1650
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.771611253196931e-05,
+      "loss": 0.0074,
+      "step": 1651
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.770843989769821e-05,
+      "loss": 0.0116,
+      "step": 1652
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 1.770076726342711e-05,
+      "loss": 0.0117,
+      "step": 1653
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.769309462915601e-05,
+      "loss": 1.1636,
+      "step": 1654
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.768542199488491e-05,
+      "loss": 0.0075,
+      "step": 1655
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.767774936061381e-05,
+      "loss": 0.007,
+      "step": 1656
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.7670076726342712e-05,
+      "loss": 0.0174,
+      "step": 1657
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.766240409207161e-05,
+      "loss": 0.0113,
+      "step": 1658
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7654731457800512e-05,
+      "loss": 0.0084,
+      "step": 1659
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7647058823529414e-05,
+      "loss": 0.2994,
+      "step": 1660
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 1.7639386189258312e-05,
+      "loss": 0.0213,
+      "step": 1661
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7631713554987213e-05,
+      "loss": 0.011,
+      "step": 1662
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7624040920716115e-05,
+      "loss": 1.2341,
+      "step": 1663
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7616368286445013e-05,
+      "loss": 1.1774,
+      "step": 1664
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 1.7608695652173915e-05,
+      "loss": 0.0181,
+      "step": 1665
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7601023017902813e-05,
+      "loss": 0.0122,
+      "step": 1666
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7593350383631715e-05,
+      "loss": 0.0119,
+      "step": 1667
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7585677749360616e-05,
+      "loss": 0.0114,
+      "step": 1668
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 1.7578005115089515e-05,
+      "loss": 0.0111,
+      "step": 1669
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7570332480818413e-05,
+      "loss": 0.011,
+      "step": 1670
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7562659846547318e-05,
+      "loss": 0.0113,
+      "step": 1671
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7554987212276216e-05,
+      "loss": 0.0161,
+      "step": 1672
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 1.7547314578005114e-05,
+      "loss": 0.9491,
+      "step": 1673
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7539641943734016e-05,
+      "loss": 0.0108,
+      "step": 1674
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7531969309462917e-05,
+      "loss": 0.0114,
+      "step": 1675
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7524296675191816e-05,
+      "loss": 0.0107,
+      "step": 1676
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 1.7516624040920717e-05,
+      "loss": 0.0171,
+      "step": 1677
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7508951406649616e-05,
+      "loss": 0.017,
+      "step": 1678
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7501278772378517e-05,
+      "loss": 1.939,
+      "step": 1679
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.749360613810742e-05,
+      "loss": 0.0151,
+      "step": 1680
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 1.7485933503836317e-05,
+      "loss": 0.0109,
+      "step": 1681
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.7478260869565215e-05,
+      "loss": 0.0119,
+      "step": 1682
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.747058823529412e-05,
+      "loss": 1.2127,
+      "step": 1683
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 1.746291560102302e-05,
+      "loss": 0.0075,
+      "step": 1684
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.7455242966751917e-05,
+      "loss": 1.2283,
+      "step": 1685
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.744757033248082e-05,
+      "loss": 0.0107,
+      "step": 1686
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.743989769820972e-05,
+      "loss": 0.0141,
+      "step": 1687
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 1.7432225063938618e-05,
+      "loss": 1.2117,
+      "step": 1688
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.742455242966752e-05,
+      "loss": 0.8715,
+      "step": 1689
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.741687979539642e-05,
+      "loss": 0.0075,
+      "step": 1690
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.740920716112532e-05,
+      "loss": 1.2208,
+      "step": 1691
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.740153452685422e-05,
+      "loss": 0.0077,
+      "step": 1692
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.739386189258312e-05,
+      "loss": 0.0081,
+      "step": 1693
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.738618925831202e-05,
+      "loss": 0.0104,
+      "step": 1694
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.7378516624040923e-05,
+      "loss": 0.0162,
+      "step": 1695
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 1.737084398976982e-05,
+      "loss": 0.0173,
+      "step": 1696
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.736317135549872e-05,
+      "loss": 0.0194,
+      "step": 1697
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.7355498721227624e-05,
+      "loss": 0.0174,
+      "step": 1698
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.7347826086956522e-05,
+      "loss": 0.0081,
+      "step": 1699
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 1.734015345268542e-05,
+      "loss": 0.0121,
+      "step": 1700
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7332480818414322e-05,
+      "loss": 0.0137,
+      "step": 1701
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7324808184143224e-05,
+      "loss": 0.0117,
+      "step": 1702
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7317135549872122e-05,
+      "loss": 0.0133,
+      "step": 1703
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 1.7309462915601024e-05,
+      "loss": 0.0175,
+      "step": 1704
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7301790281329922e-05,
+      "loss": 0.0076,
+      "step": 1705
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7294117647058823e-05,
+      "loss": 0.0179,
+      "step": 1706
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7286445012787725e-05,
+      "loss": 1.0541,
+      "step": 1707
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 1.7278772378516623e-05,
+      "loss": 0.0077,
+      "step": 1708
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7271099744245525e-05,
+      "loss": 0.0079,
+      "step": 1709
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7263427109974426e-05,
+      "loss": 0.0102,
+      "step": 1710
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7255754475703325e-05,
+      "loss": 0.0158,
+      "step": 1711
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 1.7248081841432223e-05,
+      "loss": 0.0107,
+      "step": 1712
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7240409207161128e-05,
+      "loss": 0.0159,
+      "step": 1713
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7232736572890026e-05,
+      "loss": 0.0125,
+      "step": 1714
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7225063938618928e-05,
+      "loss": 0.0081,
+      "step": 1715
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 1.7217391304347826e-05,
+      "loss": 1.1946,
+      "step": 1716
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7209718670076728e-05,
+      "loss": 1.0298,
+      "step": 1717
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.720204603580563e-05,
+      "loss": 1.0207,
+      "step": 1718
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7194373401534527e-05,
+      "loss": 0.0152,
+      "step": 1719
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 1.7186700767263426e-05,
+      "loss": 0.0187,
+      "step": 1720
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.717902813299233e-05,
+      "loss": 0.0134,
+      "step": 1721
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.717135549872123e-05,
+      "loss": 0.0173,
+      "step": 1722
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.7163682864450127e-05,
+      "loss": 1.0206,
+      "step": 1723
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 1.715601023017903e-05,
+      "loss": 0.0234,
+      "step": 1724
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.714833759590793e-05,
+      "loss": 0.0151,
+      "step": 1725
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.714066496163683e-05,
+      "loss": 0.0074,
+      "step": 1726
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.713299232736573e-05,
+      "loss": 0.0127,
+      "step": 1727
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.712531969309463e-05,
+      "loss": 0.8679,
+      "step": 1728
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.711764705882353e-05,
+      "loss": 0.0498,
+      "step": 1729
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.710997442455243e-05,
+      "loss": 0.0136,
+      "step": 1730
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.710230179028133e-05,
+      "loss": 1.2168,
+      "step": 1731
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 1.7094629156010228e-05,
+      "loss": 0.0105,
+      "step": 1732
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.7086956521739133e-05,
+      "loss": 0.0107,
+      "step": 1733
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.707928388746803e-05,
+      "loss": 0.0122,
+      "step": 1734
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.707161125319693e-05,
+      "loss": 0.0075,
+      "step": 1735
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 1.7063938618925834e-05,
+      "loss": 0.0607,
+      "step": 1736
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7056265984654733e-05,
+      "loss": 1.0181,
+      "step": 1737
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.704859335038363e-05,
+      "loss": 0.0095,
+      "step": 1738
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7040920716112533e-05,
+      "loss": 0.0128,
+      "step": 1739
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 1.7033248081841434e-05,
+      "loss": 0.0096,
+      "step": 1740
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7025575447570332e-05,
+      "loss": 1.1781,
+      "step": 1741
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7017902813299234e-05,
+      "loss": 1.0132,
+      "step": 1742
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7010230179028132e-05,
+      "loss": 0.0097,
+      "step": 1743
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 1.7002557544757034e-05,
+      "loss": 0.012,
+      "step": 1744
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6994884910485935e-05,
+      "loss": 0.0128,
+      "step": 1745
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6987212276214834e-05,
+      "loss": 0.0145,
+      "step": 1746
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6979539641943732e-05,
+      "loss": 0.7558,
+      "step": 1747
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 1.6971867007672637e-05,
+      "loss": 0.0107,
+      "step": 1748
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6964194373401535e-05,
+      "loss": 1.0181,
+      "step": 1749
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6956521739130433e-05,
+      "loss": 1.7925,
+      "step": 1750
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6948849104859335e-05,
+      "loss": 0.0141,
+      "step": 1751
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 1.6941176470588237e-05,
+      "loss": 0.0167,
+      "step": 1752
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6933503836317135e-05,
+      "loss": 0.0141,
+      "step": 1753
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6925831202046036e-05,
+      "loss": 0.0149,
+      "step": 1754
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6918158567774935e-05,
+      "loss": 0.0115,
+      "step": 1755
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 1.6910485933503836e-05,
+      "loss": 0.0071,
+      "step": 1756
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6902813299232738e-05,
+      "loss": 0.0155,
+      "step": 1757
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6895140664961636e-05,
+      "loss": 0.0199,
+      "step": 1758
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.6887468030690538e-05,
+      "loss": 0.0124,
+      "step": 1759
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 1.687979539641944e-05,
+      "loss": 0.0086,
+      "step": 1760
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.6872122762148337e-05,
+      "loss": 0.0147,
+      "step": 1761
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.6864450127877236e-05,
+      "loss": 0.0107,
+      "step": 1762
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.685677749360614e-05,
+      "loss": 1.025,
+      "step": 1763
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.684910485933504e-05,
+      "loss": 0.0105,
+      "step": 1764
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.6841432225063937e-05,
+      "loss": 0.0081,
+      "step": 1765
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.683375959079284e-05,
+      "loss": 0.009,
+      "step": 1766
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.682608695652174e-05,
+      "loss": 0.0109,
+      "step": 1767
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 1.6818414322250642e-05,
+      "loss": 0.008,
+      "step": 1768
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.681074168797954e-05,
+      "loss": 0.0132,
+      "step": 1769
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.680306905370844e-05,
+      "loss": 0.0119,
+      "step": 1770
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.6795396419437343e-05,
+      "loss": 0.0093,
+      "step": 1771
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 1.678772378516624e-05,
+      "loss": 1.0247,
+      "step": 1772
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.678005115089514e-05,
+      "loss": 1.024,
+      "step": 1773
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.677237851662404e-05,
+      "loss": 0.0106,
+      "step": 1774
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.6764705882352943e-05,
+      "loss": 0.0097,
+      "step": 1775
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 1.675703324808184e-05,
+      "loss": 0.0098,
+      "step": 1776
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6749360613810743e-05,
+      "loss": 1.0136,
+      "step": 1777
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.674168797953964e-05,
+      "loss": 0.0066,
+      "step": 1778
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6734015345268543e-05,
+      "loss": 0.1284,
+      "step": 1779
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 1.6726342710997444e-05,
+      "loss": 0.9832,
+      "step": 1780
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6718670076726343e-05,
+      "loss": 0.0128,
+      "step": 1781
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6710997442455244e-05,
+      "loss": 0.0103,
+      "step": 1782
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 1.6703324808184146e-05,
+      "loss": 0.0067,
+      "step": 1783
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6695652173913044e-05,
+      "loss": 0.0105,
+      "step": 1784
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6687979539641942e-05,
+      "loss": 0.0146,
+      "step": 1785
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6680306905370847e-05,
+      "loss": 0.0068,
+      "step": 1786
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 1.6672634271099745e-05,
+      "loss": 0.0148,
+      "step": 1787
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6664961636828644e-05,
+      "loss": 0.0107,
+      "step": 1788
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6657289002557545e-05,
+      "loss": 2.4738,
+      "step": 1789
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6649616368286447e-05,
+      "loss": 0.9625,
+      "step": 1790
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 1.6641943734015345e-05,
+      "loss": 0.2578,
+      "step": 1791
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6634271099744247e-05,
+      "loss": 0.0069,
+      "step": 1792
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6626598465473145e-05,
+      "loss": 0.011,
+      "step": 1793
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6618925831202047e-05,
+      "loss": 0.0066,
+      "step": 1794
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 1.6611253196930948e-05,
+      "loss": 0.0069,
+      "step": 1795
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6603580562659846e-05,
+      "loss": 0.0071,
+      "step": 1796
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6595907928388745e-05,
+      "loss": 0.0068,
+      "step": 1797
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.658823529411765e-05,
+      "loss": 0.0232,
+      "step": 1798
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 1.6580562659846548e-05,
+      "loss": 1.1318,
+      "step": 1799
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6572890025575446e-05,
+      "loss": 0.0101,
+      "step": 1800
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6565217391304348e-05,
+      "loss": 0.0187,
+      "step": 1801
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.655754475703325e-05,
+      "loss": 1.0718,
+      "step": 1802
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.6549872122762148e-05,
+      "loss": 1.2349,
+      "step": 1803
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.654219948849105e-05,
+      "loss": 0.0148,
+      "step": 1804
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.6534526854219947e-05,
+      "loss": 0.0148,
+      "step": 1805
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.652685421994885e-05,
+      "loss": 0.0067,
+      "step": 1806
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 1.651918158567775e-05,
+      "loss": 0.0075,
+      "step": 1807
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.651150895140665e-05,
+      "loss": 0.0157,
+      "step": 1808
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.650383631713555e-05,
+      "loss": 0.0084,
+      "step": 1809
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.6496163682864452e-05,
+      "loss": 0.0123,
+      "step": 1810
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 1.648849104859335e-05,
+      "loss": 0.0065,
+      "step": 1811
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.648081841432225e-05,
+      "loss": 0.014,
+      "step": 1812
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.6473145780051154e-05,
+      "loss": 0.0152,
+      "step": 1813
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.6465473145780052e-05,
+      "loss": 0.0168,
+      "step": 1814
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 1.645780051150895e-05,
+      "loss": 0.0063,
+      "step": 1815
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.645012787723785e-05,
+      "loss": 0.6994,
+      "step": 1816
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.6442455242966753e-05,
+      "loss": 0.0132,
+      "step": 1817
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.643478260869565e-05,
+      "loss": 0.0103,
+      "step": 1818
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 1.6427109974424553e-05,
+      "loss": 0.013,
+      "step": 1819
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.641943734015345e-05,
+      "loss": 1.1813,
+      "step": 1820
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6411764705882356e-05,
+      "loss": 0.9809,
+      "step": 1821
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6404092071611254e-05,
+      "loss": 1.2339,
+      "step": 1822
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 1.6396419437340153e-05,
+      "loss": 0.9744,
+      "step": 1823
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6388746803069054e-05,
+      "loss": 0.0098,
+      "step": 1824
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6381074168797956e-05,
+      "loss": 0.0062,
+      "step": 1825
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6373401534526854e-05,
+      "loss": 0.0151,
+      "step": 1826
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 1.6365728900255756e-05,
+      "loss": 0.013,
+      "step": 1827
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6358056265984654e-05,
+      "loss": 0.0071,
+      "step": 1828
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6350383631713556e-05,
+      "loss": 0.0117,
+      "step": 1829
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6342710997442457e-05,
+      "loss": 0.0165,
+      "step": 1830
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 1.6335038363171355e-05,
+      "loss": 0.1519,
+      "step": 1831
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6327365728900257e-05,
+      "loss": 1.0247,
+      "step": 1832
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.631969309462916e-05,
+      "loss": 0.0092,
+      "step": 1833
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6312020460358057e-05,
+      "loss": 0.0098,
+      "step": 1834
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 1.6304347826086955e-05,
+      "loss": 1.1829,
+      "step": 1835
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.629667519181586e-05,
+      "loss": 0.006,
+      "step": 1836
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.628900255754476e-05,
+      "loss": 0.0109,
+      "step": 1837
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.6281329923273657e-05,
+      "loss": 0.6059,
+      "step": 1838
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.6273657289002558e-05,
+      "loss": 0.01,
+      "step": 1839
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.626598465473146e-05,
+      "loss": 0.0149,
+      "step": 1840
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.6258312020460358e-05,
+      "loss": 0.0152,
+      "step": 1841
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.625063938618926e-05,
+      "loss": 0.0171,
+      "step": 1842
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 1.6242966751918158e-05,
+      "loss": 0.0105,
+      "step": 1843
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.623529411764706e-05,
+      "loss": 0.0093,
+      "step": 1844
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.622762148337596e-05,
+      "loss": 1.4125,
+      "step": 1845
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.621994884910486e-05,
+      "loss": 0.0107,
+      "step": 1846
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 1.6212276214833757e-05,
+      "loss": 0.0133,
+      "step": 1847
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.6204603580562662e-05,
+      "loss": 0.0063,
+      "step": 1848
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.619693094629156e-05,
+      "loss": 0.0063,
+      "step": 1849
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.618925831202046e-05,
+      "loss": 0.009,
+      "step": 1850
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 1.618158567774936e-05,
+      "loss": 0.0137,
+      "step": 1851
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.6173913043478262e-05,
+      "loss": 0.0145,
+      "step": 1852
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.616624040920716e-05,
+      "loss": 0.0355,
+      "step": 1853
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.6158567774936062e-05,
+      "loss": 0.9601,
+      "step": 1854
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 1.615089514066496e-05,
+      "loss": 0.0135,
+      "step": 1855
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6143222506393862e-05,
+      "loss": 1.0129,
+      "step": 1856
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6135549872122763e-05,
+      "loss": 0.0339,
+      "step": 1857
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.612787723785166e-05,
+      "loss": 0.0131,
+      "step": 1858
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 1.6120204603580563e-05,
+      "loss": 0.009,
+      "step": 1859
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6112531969309465e-05,
+      "loss": 0.0087,
+      "step": 1860
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6104859335038363e-05,
+      "loss": 0.012,
+      "step": 1861
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.609718670076726e-05,
+      "loss": 0.0088,
+      "step": 1862
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 1.6089514066496166e-05,
+      "loss": 0.0117,
+      "step": 1863
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6081841432225065e-05,
+      "loss": 2.5214,
+      "step": 1864
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6074168797953963e-05,
+      "loss": 0.0107,
+      "step": 1865
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6066496163682864e-05,
+      "loss": 0.0101,
+      "step": 1866
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 1.6058823529411766e-05,
+      "loss": 0.0118,
+      "step": 1867
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6051150895140664e-05,
+      "loss": 0.0087,
+      "step": 1868
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6043478260869566e-05,
+      "loss": 0.0122,
+      "step": 1869
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6035805626598464e-05,
+      "loss": 0.0084,
+      "step": 1870
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 1.6028132992327366e-05,
+      "loss": 0.0104,
+      "step": 1871
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6020460358056267e-05,
+      "loss": 0.1587,
+      "step": 1872
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6012787723785166e-05,
+      "loss": 0.0057,
+      "step": 1873
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.6005115089514067e-05,
+      "loss": 0.0085,
+      "step": 1874
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.599744245524297e-05,
+      "loss": 0.0056,
+      "step": 1875
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.5989769820971867e-05,
+      "loss": 1.2194,
+      "step": 1876
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.598209718670077e-05,
+      "loss": 0.0056,
+      "step": 1877
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.5974424552429667e-05,
+      "loss": 0.0057,
+      "step": 1878
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 1.596675191815857e-05,
+      "loss": 1.0276,
+      "step": 1879
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.595907928388747e-05,
+      "loss": 0.0083,
+      "step": 1880
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.5951406649616368e-05,
+      "loss": 0.0085,
+      "step": 1881
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 1.594373401534527e-05,
+      "loss": 0.0363,
+      "step": 1882
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.593606138107417e-05,
+      "loss": 0.0157,
+      "step": 1883
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.592838874680307e-05,
+      "loss": 0.0313,
+      "step": 1884
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.5920716112531968e-05,
+      "loss": 0.0062,
+      "step": 1885
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 1.5913043478260873e-05,
+      "loss": 0.0195,
+      "step": 1886
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.590537084398977e-05,
+      "loss": 0.0151,
+      "step": 1887
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.589769820971867e-05,
+      "loss": 0.0101,
+      "step": 1888
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.589002557544757e-05,
+      "loss": 1.1092,
+      "step": 1889
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 1.5882352941176473e-05,
+      "loss": 0.0323,
+      "step": 1890
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.587468030690537e-05,
+      "loss": 0.0086,
+      "step": 1891
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.5867007672634272e-05,
+      "loss": 0.0147,
+      "step": 1892
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.585933503836317e-05,
+      "loss": 0.0084,
+      "step": 1893
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 1.5851662404092072e-05,
+      "loss": 0.5596,
+      "step": 1894
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5843989769820974e-05,
+      "loss": 0.0114,
+      "step": 1895
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5836317135549872e-05,
+      "loss": 0.0083,
+      "step": 1896
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.582864450127877e-05,
+      "loss": 0.0082,
+      "step": 1897
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 1.5820971867007675e-05,
+      "loss": 0.0146,
+      "step": 1898
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5813299232736574e-05,
+      "loss": 0.0081,
+      "step": 1899
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5805626598465472e-05,
+      "loss": 0.0081,
+      "step": 1900
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5797953964194373e-05,
+      "loss": 0.0116,
+      "step": 1901
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 1.5790281329923275e-05,
+      "loss": 0.0084,
+      "step": 1902
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5782608695652173e-05,
+      "loss": 0.0108,
+      "step": 1903
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5774936061381075e-05,
+      "loss": 0.6146,
+      "step": 1904
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5767263427109973e-05,
+      "loss": 0.0162,
+      "step": 1905
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 1.5759590792838875e-05,
+      "loss": 0.0134,
+      "step": 1906
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5751918158567776e-05,
+      "loss": 0.0047,
+      "step": 1907
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5744245524296674e-05,
+      "loss": 0.0045,
+      "step": 1908
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5736572890025576e-05,
+      "loss": 0.0087,
+      "step": 1909
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.5728900255754478e-05,
+      "loss": 2.0913,
+      "step": 1910
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5721227621483376e-05,
+      "loss": 0.0075,
+      "step": 1911
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5713554987212274e-05,
+      "loss": 1.3387,
+      "step": 1912
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.570588235294118e-05,
+      "loss": 0.0102,
+      "step": 1913
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 1.5698209718670077e-05,
+      "loss": 0.0076,
+      "step": 1914
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5690537084398976e-05,
+      "loss": 0.0131,
+      "step": 1915
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5682864450127877e-05,
+      "loss": 0.0079,
+      "step": 1916
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.567519181585678e-05,
+      "loss": 0.0102,
+      "step": 1917
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 1.5667519181585677e-05,
+      "loss": 0.9554,
+      "step": 1918
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.565984654731458e-05,
+      "loss": 0.0088,
+      "step": 1919
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.5652173913043477e-05,
+      "loss": 1.0265,
+      "step": 1920
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.564450127877238e-05,
+      "loss": 0.0155,
+      "step": 1921
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 1.563682864450128e-05,
+      "loss": 0.0079,
+      "step": 1922
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.562915601023018e-05,
+      "loss": 1.0467,
+      "step": 1923
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.5621483375959077e-05,
+      "loss": 0.0132,
+      "step": 1924
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.561381074168798e-05,
+      "loss": 0.0077,
+      "step": 1925
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 1.560613810741688e-05,
+      "loss": 0.0064,
+      "step": 1926
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.559846547314578e-05,
+      "loss": 0.0104,
+      "step": 1927
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.559079283887468e-05,
+      "loss": 0.0096,
+      "step": 1928
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.558312020460358e-05,
+      "loss": 0.0049,
+      "step": 1929
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 1.5575447570332483e-05,
+      "loss": 0.0104,
+      "step": 1930
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.556777493606138e-05,
+      "loss": 1.0424,
+      "step": 1931
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5560102301790283e-05,
+      "loss": 0.0079,
+      "step": 1932
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5552429667519184e-05,
+      "loss": 0.0104,
+      "step": 1933
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 1.5544757033248083e-05,
+      "loss": 0.0077,
+      "step": 1934
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.553708439897698e-05,
+      "loss": 0.005,
+      "step": 1935
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5529411764705886e-05,
+      "loss": 2.6544,
+      "step": 1936
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5521739130434784e-05,
+      "loss": 0.0107,
+      "step": 1937
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 1.5514066496163682e-05,
+      "loss": 0.0124,
+      "step": 1938
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5506393861892584e-05,
+      "loss": 0.0049,
+      "step": 1939
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5498721227621485e-05,
+      "loss": 0.0078,
+      "step": 1940
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5491048593350384e-05,
+      "loss": 2.1009,
+      "step": 1941
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 1.5483375959079285e-05,
+      "loss": 0.0116,
+      "step": 1942
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5475703324808183e-05,
+      "loss": 0.0049,
+      "step": 1943
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5468030690537085e-05,
+      "loss": 0.0079,
+      "step": 1944
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5460358056265987e-05,
+      "loss": 1.0307,
+      "step": 1945
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.5452685421994885e-05,
+      "loss": 0.0147,
+      "step": 1946
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5445012787723783e-05,
+      "loss": 0.0084,
+      "step": 1947
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5437340153452688e-05,
+      "loss": 0.0262,
+      "step": 1948
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5429667519181586e-05,
+      "loss": 0.0115,
+      "step": 1949
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 1.5421994884910485e-05,
+      "loss": 1.3221,
+      "step": 1950
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5414322250639386e-05,
+      "loss": 0.7929,
+      "step": 1951
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5406649616368288e-05,
+      "loss": 1.2069,
+      "step": 1952
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5398976982097186e-05,
+      "loss": 0.0081,
+      "step": 1953
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 1.5391304347826088e-05,
+      "loss": 0.0069,
+      "step": 1954
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5383631713554986e-05,
+      "loss": 0.0097,
+      "step": 1955
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5375959079283887e-05,
+      "loss": 0.0082,
+      "step": 1956
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.536828644501279e-05,
+      "loss": 0.0061,
+      "step": 1957
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 1.5360613810741687e-05,
+      "loss": 0.0082,
+      "step": 1958
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.535294117647059e-05,
+      "loss": 0.0113,
+      "step": 1959
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.534526854219949e-05,
+      "loss": 0.0098,
+      "step": 1960
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.533759590792839e-05,
+      "loss": 1.0221,
+      "step": 1961
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 1.5329923273657287e-05,
+      "loss": 0.0039,
+      "step": 1962
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.5322250639386192e-05,
+      "loss": 0.0127,
+      "step": 1963
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.531457800511509e-05,
+      "loss": 0.0051,
+      "step": 1964
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.530690537084399e-05,
+      "loss": 0.0086,
+      "step": 1965
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 1.529923273657289e-05,
+      "loss": 0.0096,
+      "step": 1966
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.529156010230179e-05,
+      "loss": 0.0164,
+      "step": 1967
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.528388746803069e-05,
+      "loss": 1.0127,
+      "step": 1968
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.527621483375959e-05,
+      "loss": 0.0149,
+      "step": 1969
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 1.526854219948849e-05,
+      "loss": 0.0085,
+      "step": 1970
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.526086956521739e-05,
+      "loss": 0.0116,
+      "step": 1971
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5253196930946293e-05,
+      "loss": 0.0168,
+      "step": 1972
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5245524296675191e-05,
+      "loss": 0.0052,
+      "step": 1973
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 1.5237851662404091e-05,
+      "loss": 0.0052,
+      "step": 1974
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5230179028132993e-05,
+      "loss": 1.0013,
+      "step": 1975
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5222506393861893e-05,
+      "loss": 0.0122,
+      "step": 1976
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.521483375959079e-05,
+      "loss": 0.0087,
+      "step": 1977
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 1.5207161125319694e-05,
+      "loss": 1.9624,
+      "step": 1978
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5199488491048592e-05,
+      "loss": 0.0088,
+      "step": 1979
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5191815856777496e-05,
+      "loss": 0.0596,
+      "step": 1980
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9090909090909091,
+      "eval_loss": 0.40464258193969727,
+      "eval_runtime": 58.0491,
+      "eval_samples_per_second": 6.822,
+      "eval_steps_per_second": 1.705,
+      "step": 1980
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 1.5184143222506396e-05,
+      "loss": 0.009,
+      "step": 1981
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5176470588235294e-05,
+      "loss": 0.0063,
+      "step": 1982
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5168797953964197e-05,
+      "loss": 0.0092,
+      "step": 1983
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5161125319693095e-05,
+      "loss": 0.0092,
+      "step": 1984
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 1.5153452685421995e-05,
+      "loss": 0.0259,
+      "step": 1985
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5145780051150897e-05,
+      "loss": 0.0093,
+      "step": 1986
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5138107416879797e-05,
+      "loss": 1.3347,
+      "step": 1987
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5130434782608695e-05,
+      "loss": 0.0135,
+      "step": 1988
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 1.5122762148337598e-05,
+      "loss": 0.0134,
+      "step": 1989
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5115089514066497e-05,
+      "loss": 0.0051,
+      "step": 1990
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5107416879795396e-05,
+      "loss": 0.0106,
+      "step": 1991
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5099744245524298e-05,
+      "loss": 0.013,
+      "step": 1992
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 1.5092071611253198e-05,
+      "loss": 0.0095,
+      "step": 1993
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5084398976982096e-05,
+      "loss": 0.0096,
+      "step": 1994
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5076726342711e-05,
+      "loss": 0.0112,
+      "step": 1995
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5069053708439898e-05,
+      "loss": 0.0137,
+      "step": 1996
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 1.5061381074168798e-05,
+      "loss": 0.9611,
+      "step": 1997
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50537084398977e-05,
+      "loss": 0.009,
+      "step": 1998
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50460358056266e-05,
+      "loss": 0.0066,
+      "step": 1999
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.5038363171355497e-05,
+      "loss": 0.0091,
+      "step": 2000
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 1.50306905370844e-05,
+      "loss": 0.9856,
+      "step": 2001
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5023017902813299e-05,
+      "loss": 0.0137,
+      "step": 2002
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5015345268542199e-05,
+      "loss": 0.0094,
+      "step": 2003
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.50076726342711e-05,
+      "loss": 0.9276,
+      "step": 2004
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.0051,
+      "step": 2005
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4992327365728902e-05,
+      "loss": 0.0097,
+      "step": 2006
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.49846547314578e-05,
+      "loss": 0.8053,
+      "step": 2007
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4976982097186702e-05,
+      "loss": 0.5635,
+      "step": 2008
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 1.4969309462915602e-05,
+      "loss": 0.0099,
+      "step": 2009
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4961636828644502e-05,
+      "loss": 0.0108,
+      "step": 2010
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4953964194373402e-05,
+      "loss": 1.3337,
+      "step": 2011
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4946291560102303e-05,
+      "loss": 0.0098,
+      "step": 2012
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 1.4938618925831201e-05,
+      "loss": 0.0099,
+      "step": 2013
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4930946291560103e-05,
+      "loss": 0.011,
+      "step": 2014
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4923273657289003e-05,
+      "loss": 0.0166,
+      "step": 2015
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4915601023017903e-05,
+      "loss": 1.3541,
+      "step": 2016
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 1.4907928388746803e-05,
+      "loss": 0.0145,
+      "step": 2017
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4900255754475704e-05,
+      "loss": 0.0051,
+      "step": 2018
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4892583120204604e-05,
+      "loss": 0.0051,
+      "step": 2019
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4884910485933504e-05,
+      "loss": 0.0139,
+      "step": 2020
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 1.4877237851662404e-05,
+      "loss": 0.0097,
+      "step": 2021
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4869565217391306e-05,
+      "loss": 0.9418,
+      "step": 2022
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4861892583120204e-05,
+      "loss": 0.9619,
+      "step": 2023
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4854219948849106e-05,
+      "loss": 1.3495,
+      "step": 2024
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 1.4846547314578006e-05,
+      "loss": 0.0051,
+      "step": 2025
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4838874680306905e-05,
+      "loss": 1.3222,
+      "step": 2026
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4831202046035805e-05,
+      "loss": 1.1638,
+      "step": 2027
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4823529411764707e-05,
+      "loss": 0.009,
+      "step": 2028
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 1.4815856777493605e-05,
+      "loss": 0.0059,
+      "step": 2029
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4808184143222507e-05,
+      "loss": 0.9946,
+      "step": 2030
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4800511508951408e-05,
+      "loss": 0.1291,
+      "step": 2031
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4792838874680307e-05,
+      "loss": 0.0089,
+      "step": 2032
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 1.4785166240409208e-05,
+      "loss": 0.8113,
+      "step": 2033
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4777493606138108e-05,
+      "loss": 0.0109,
+      "step": 2034
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4769820971867008e-05,
+      "loss": 0.0061,
+      "step": 2035
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.4762148337595908e-05,
+      "loss": 0.0157,
+      "step": 2036
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 1.475447570332481e-05,
+      "loss": 0.0104,
+      "step": 2037
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.4746803069053708e-05,
+      "loss": 0.012,
+      "step": 2038
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.473913043478261e-05,
+      "loss": 0.0106,
+      "step": 2039
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.473145780051151e-05,
+      "loss": 0.006,
+      "step": 2040
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 1.472378516624041e-05,
+      "loss": 0.0122,
+      "step": 2041
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.471611253196931e-05,
+      "loss": 0.0103,
+      "step": 2042
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.470843989769821e-05,
+      "loss": 0.9465,
+      "step": 2043
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.4700767263427109e-05,
+      "loss": 0.0149,
+      "step": 2044
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 1.469309462915601e-05,
+      "loss": 0.0146,
+      "step": 2045
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.468542199488491e-05,
+      "loss": 0.9225,
+      "step": 2046
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.4677749360613812e-05,
+      "loss": 0.0101,
+      "step": 2047
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.467007672634271e-05,
+      "loss": 0.0052,
+      "step": 2048
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 1.4662404092071612e-05,
+      "loss": 1.682,
+      "step": 2049
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4654731457800512e-05,
+      "loss": 0.9498,
+      "step": 2050
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4647058823529412e-05,
+      "loss": 0.0097,
+      "step": 2051
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4639386189258312e-05,
+      "loss": 0.0067,
+      "step": 2052
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 1.4631713554987213e-05,
+      "loss": 0.0052,
+      "step": 2053
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4624040920716112e-05,
+      "loss": 0.0055,
+      "step": 2054
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4616368286445013e-05,
+      "loss": 0.0225,
+      "step": 2055
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4608695652173915e-05,
+      "loss": 0.0052,
+      "step": 2056
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 1.4601023017902813e-05,
+      "loss": 0.0066,
+      "step": 2057
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4593350383631715e-05,
+      "loss": 0.0112,
+      "step": 2058
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4585677749360615e-05,
+      "loss": 1.3145,
+      "step": 2059
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4578005115089514e-05,
+      "loss": 0.0099,
+      "step": 2060
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 1.4570332480818414e-05,
+      "loss": 0.0055,
+      "step": 2061
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4562659846547316e-05,
+      "loss": 0.0149,
+      "step": 2062
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4554987212276214e-05,
+      "loss": 0.0102,
+      "step": 2063
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4547314578005116e-05,
+      "loss": 0.0108,
+      "step": 2064
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 1.4539641943734016e-05,
+      "loss": 0.0065,
+      "step": 2065
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4531969309462916e-05,
+      "loss": 0.0185,
+      "step": 2066
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4524296675191816e-05,
+      "loss": 0.9659,
+      "step": 2067
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4516624040920717e-05,
+      "loss": 0.0193,
+      "step": 2068
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 1.4508951406649615e-05,
+      "loss": 0.0146,
+      "step": 2069
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4501278772378517e-05,
+      "loss": 0.0061,
+      "step": 2070
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4493606138107417e-05,
+      "loss": 0.0222,
+      "step": 2071
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4485933503836319e-05,
+      "loss": 0.0096,
+      "step": 2072
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 1.4478260869565217e-05,
+      "loss": 0.0096,
+      "step": 2073
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4470588235294118e-05,
+      "loss": 0.005,
+      "step": 2074
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4462915601023018e-05,
+      "loss": 0.8301,
+      "step": 2075
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4455242966751918e-05,
+      "loss": 0.0049,
+      "step": 2076
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 1.4447570332480818e-05,
+      "loss": 0.0097,
+      "step": 2077
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.443989769820972e-05,
+      "loss": 0.0062,
+      "step": 2078
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.4432225063938618e-05,
+      "loss": 1.3459,
+      "step": 2079
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 1.442455242966752e-05,
+      "loss": 0.0096,
+      "step": 2080
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4416879795396421e-05,
+      "loss": 1.2176,
+      "step": 2081
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.440920716112532e-05,
+      "loss": 0.0114,
+      "step": 2082
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4401534526854221e-05,
+      "loss": 0.0109,
+      "step": 2083
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 1.4393861892583121e-05,
+      "loss": 0.0105,
+      "step": 2084
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.4386189258312021e-05,
+      "loss": 0.0107,
+      "step": 2085
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.437851662404092e-05,
+      "loss": 0.9442,
+      "step": 2086
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.4370843989769822e-05,
+      "loss": 0.01,
+      "step": 2087
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 1.436317135549872e-05,
+      "loss": 0.0099,
+      "step": 2088
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4355498721227622e-05,
+      "loss": 0.0167,
+      "step": 2089
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4347826086956522e-05,
+      "loss": 0.0051,
+      "step": 2090
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4340153452685422e-05,
+      "loss": 1.333,
+      "step": 2091
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 1.4332480818414322e-05,
+      "loss": 0.0111,
+      "step": 2092
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4324808184143224e-05,
+      "loss": 0.0103,
+      "step": 2093
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4317135549872122e-05,
+      "loss": 0.0111,
+      "step": 2094
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4309462915601023e-05,
+      "loss": 0.0165,
+      "step": 2095
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 1.4301790281329923e-05,
+      "loss": 0.0089,
+      "step": 2096
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4294117647058823e-05,
+      "loss": 0.0143,
+      "step": 2097
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4286445012787723e-05,
+      "loss": 0.4759,
+      "step": 2098
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4278772378516625e-05,
+      "loss": 0.0102,
+      "step": 2099
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 1.4271099744245525e-05,
+      "loss": 0.0059,
+      "step": 2100
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4263427109974425e-05,
+      "loss": 0.0152,
+      "step": 2101
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4255754475703325e-05,
+      "loss": 0.0051,
+      "step": 2102
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4248081841432226e-05,
+      "loss": 0.0098,
+      "step": 2103
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 1.4240409207161126e-05,
+      "loss": 0.0147,
+      "step": 2104
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4232736572890026e-05,
+      "loss": 0.0051,
+      "step": 2105
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4225063938618928e-05,
+      "loss": 1.3384,
+      "step": 2106
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4217391304347826e-05,
+      "loss": 0.0055,
+      "step": 2107
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 1.4209718670076727e-05,
+      "loss": 0.0136,
+      "step": 2108
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4202046035805627e-05,
+      "loss": 0.0056,
+      "step": 2109
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4194373401534527e-05,
+      "loss": 0.0094,
+      "step": 2110
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4186700767263427e-05,
+      "loss": 0.0135,
+      "step": 2111
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 1.4179028132992329e-05,
+      "loss": 0.0129,
+      "step": 2112
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4171355498721227e-05,
+      "loss": 0.0051,
+      "step": 2113
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4163682864450129e-05,
+      "loss": 0.0134,
+      "step": 2114
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4156010230179029e-05,
+      "loss": 0.0091,
+      "step": 2115
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 1.4148337595907929e-05,
+      "loss": 0.0072,
+      "step": 2116
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.4140664961636828e-05,
+      "loss": 0.005,
+      "step": 2117
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.413299232736573e-05,
+      "loss": 0.0112,
+      "step": 2118
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.4125319693094628e-05,
+      "loss": 0.005,
+      "step": 2119
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 1.411764705882353e-05,
+      "loss": 0.0085,
+      "step": 2120
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.410997442455243e-05,
+      "loss": 0.0121,
+      "step": 2121
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.410230179028133e-05,
+      "loss": 0.0084,
+      "step": 2122
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.409462915601023e-05,
+      "loss": 0.0084,
+      "step": 2123
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 1.4086956521739131e-05,
+      "loss": 0.0083,
+      "step": 2124
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4079283887468031e-05,
+      "loss": 1.3402,
+      "step": 2125
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4071611253196931e-05,
+      "loss": 0.0124,
+      "step": 2126
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4063938618925831e-05,
+      "loss": 0.0082,
+      "step": 2127
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 1.4056265984654733e-05,
+      "loss": 0.9694,
+      "step": 2128
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4048593350383633e-05,
+      "loss": 0.0081,
+      "step": 2129
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4040920716112532e-05,
+      "loss": 1.2102,
+      "step": 2130
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4033248081841434e-05,
+      "loss": 0.0113,
+      "step": 2131
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 1.4025575447570332e-05,
+      "loss": 0.0114,
+      "step": 2132
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4017902813299234e-05,
+      "loss": 0.0252,
+      "step": 2133
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4010230179028134e-05,
+      "loss": 0.9785,
+      "step": 2134
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.4002557544757034e-05,
+      "loss": 0.0061,
+      "step": 2135
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 1.3994884910485934e-05,
+      "loss": 0.0096,
+      "step": 2136
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3987212276214835e-05,
+      "loss": 0.0088,
+      "step": 2137
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3979539641943733e-05,
+      "loss": 0.0047,
+      "step": 2138
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3971867007672635e-05,
+      "loss": 0.0108,
+      "step": 2139
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 1.3964194373401535e-05,
+      "loss": 0.0081,
+      "step": 2140
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3956521739130435e-05,
+      "loss": 0.0108,
+      "step": 2141
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3948849104859335e-05,
+      "loss": 0.0126,
+      "step": 2142
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3941176470588236e-05,
+      "loss": 0.0046,
+      "step": 2143
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 1.3933503836317135e-05,
+      "loss": 0.0122,
+      "step": 2144
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3925831202046036e-05,
+      "loss": 0.0075,
+      "step": 2145
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3918158567774936e-05,
+      "loss": 0.0123,
+      "step": 2146
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3910485933503836e-05,
+      "loss": 0.0134,
+      "step": 2147
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 1.3902813299232736e-05,
+      "loss": 0.0115,
+      "step": 2148
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3895140664961638e-05,
+      "loss": 0.0044,
+      "step": 2149
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3887468030690536e-05,
+      "loss": 0.0121,
+      "step": 2150
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3879795396419437e-05,
+      "loss": 0.0101,
+      "step": 2151
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 1.3872122762148337e-05,
+      "loss": 0.0099,
+      "step": 2152
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3864450127877239e-05,
+      "loss": 0.0044,
+      "step": 2153
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3856777493606139e-05,
+      "loss": 0.0094,
+      "step": 2154
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.3849104859335039e-05,
+      "loss": 0.0124,
+      "step": 2155
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 1.384143222506394e-05,
+      "loss": 0.0069,
+      "step": 2156
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.3833759590792839e-05,
+      "loss": 0.0068,
+      "step": 2157
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.382608695652174e-05,
+      "loss": 0.0094,
+      "step": 2158
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.381841432225064e-05,
+      "loss": 0.0067,
+      "step": 2159
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 1.381074168797954e-05,
+      "loss": 0.0044,
+      "step": 2160
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.380306905370844e-05,
+      "loss": 0.0041,
+      "step": 2161
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.3795396419437342e-05,
+      "loss": 0.0041,
+      "step": 2162
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.378772378516624e-05,
+      "loss": 0.009,
+      "step": 2163
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 1.3780051150895141e-05,
+      "loss": 0.0092,
+      "step": 2164
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3772378516624041e-05,
+      "loss": 0.0101,
+      "step": 2165
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3764705882352941e-05,
+      "loss": 0.0064,
+      "step": 2166
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3757033248081841e-05,
+      "loss": 0.0088,
+      "step": 2167
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 1.3749360613810743e-05,
+      "loss": 0.0059,
+      "step": 2168
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3741687979539641e-05,
+      "loss": 0.0062,
+      "step": 2169
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3734015345268543e-05,
+      "loss": 0.0085,
+      "step": 2170
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3726342710997443e-05,
+      "loss": 0.0141,
+      "step": 2171
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 1.3718670076726343e-05,
+      "loss": 0.0061,
+      "step": 2172
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3710997442455242e-05,
+      "loss": 0.0084,
+      "step": 2173
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3703324808184144e-05,
+      "loss": 0.0038,
+      "step": 2174
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3695652173913042e-05,
+      "loss": 0.006,
+      "step": 2175
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 1.3687979539641944e-05,
+      "loss": 0.0039,
+      "step": 2176
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3680306905370844e-05,
+      "loss": 0.0205,
+      "step": 2177
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3672634271099745e-05,
+      "loss": 0.0062,
+      "step": 2178
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 1.3664961636828645e-05,
+      "loss": 0.0021,
+      "step": 2179
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3657289002557545e-05,
+      "loss": 0.0058,
+      "step": 2180
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3649616368286447e-05,
+      "loss": 0.0101,
+      "step": 2181
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3641943734015345e-05,
+      "loss": 0.0036,
+      "step": 2182
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 1.3634271099744247e-05,
+      "loss": 0.0037,
+      "step": 2183
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3626598465473147e-05,
+      "loss": 0.0036,
+      "step": 2184
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3618925831202047e-05,
+      "loss": 0.0036,
+      "step": 2185
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3611253196930946e-05,
+      "loss": 0.0035,
+      "step": 2186
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 1.3603580562659848e-05,
+      "loss": 0.0059,
+      "step": 2187
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3595907928388746e-05,
+      "loss": 0.0056,
+      "step": 2188
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3588235294117648e-05,
+      "loss": 0.0139,
+      "step": 2189
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3580562659846548e-05,
+      "loss": 0.0558,
+      "step": 2190
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 1.3572890025575448e-05,
+      "loss": 1.419,
+      "step": 2191
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3565217391304348e-05,
+      "loss": 0.0058,
+      "step": 2192
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.355754475703325e-05,
+      "loss": 0.0035,
+      "step": 2193
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3549872122762147e-05,
+      "loss": 0.0055,
+      "step": 2194
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 1.3542199488491049e-05,
+      "loss": 1.2049,
+      "step": 2195
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3534526854219949e-05,
+      "loss": 1.3646,
+      "step": 2196
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3526854219948849e-05,
+      "loss": 1.3803,
+      "step": 2197
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.3519181585677749e-05,
+      "loss": 0.0055,
+      "step": 2198
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 1.351150895140665e-05,
+      "loss": 0.0059,
+      "step": 2199
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.3503836317135549e-05,
+      "loss": 0.0056,
+      "step": 2200
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.349616368286445e-05,
+      "loss": 0.0097,
+      "step": 2201
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.348849104859335e-05,
+      "loss": 1.3497,
+      "step": 2202
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 1.348081841432225e-05,
+      "loss": 1.0989,
+      "step": 2203
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3473145780051152e-05,
+      "loss": 0.0117,
+      "step": 2204
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3465473145780052e-05,
+      "loss": 0.0038,
+      "step": 2205
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3457800511508953e-05,
+      "loss": 0.0078,
+      "step": 2206
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 1.3450127877237852e-05,
+      "loss": 0.0058,
+      "step": 2207
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3442455242966753e-05,
+      "loss": 0.9775,
+      "step": 2208
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3434782608695653e-05,
+      "loss": 0.0037,
+      "step": 2209
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3427109974424553e-05,
+      "loss": 0.0079,
+      "step": 2210
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 1.3419437340153453e-05,
+      "loss": 0.0663,
+      "step": 2211
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3411764705882354e-05,
+      "loss": 0.0059,
+      "step": 2212
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3404092071611253e-05,
+      "loss": 0.0039,
+      "step": 2213
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3396419437340154e-05,
+      "loss": 1.3169,
+      "step": 2214
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 1.3388746803069054e-05,
+      "loss": 0.0061,
+      "step": 2215
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3381074168797954e-05,
+      "loss": 0.006,
+      "step": 2216
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3373401534526854e-05,
+      "loss": 0.004,
+      "step": 2217
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3365728900255756e-05,
+      "loss": 0.0045,
+      "step": 2218
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 1.3358056265984654e-05,
+      "loss": 0.006,
+      "step": 2219
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3350383631713556e-05,
+      "loss": 0.0058,
+      "step": 2220
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3342710997442455e-05,
+      "loss": 0.014,
+      "step": 2221
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3335038363171355e-05,
+      "loss": 1.1674,
+      "step": 2222
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 1.3327365728900255e-05,
+      "loss": 0.004,
+      "step": 2223
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3319693094629157e-05,
+      "loss": 0.0052,
+      "step": 2224
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3312020460358055e-05,
+      "loss": 0.0095,
+      "step": 2225
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3304347826086957e-05,
+      "loss": 0.0053,
+      "step": 2226
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 1.3296675191815857e-05,
+      "loss": 0.0099,
+      "step": 2227
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3289002557544757e-05,
+      "loss": 1.3879,
+      "step": 2228
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3281329923273658e-05,
+      "loss": 0.0041,
+      "step": 2229
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.3273657289002558e-05,
+      "loss": 0.006,
+      "step": 2230
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 1.326598465473146e-05,
+      "loss": 0.006,
+      "step": 2231
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.3258312020460358e-05,
+      "loss": 0.012,
+      "step": 2232
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.325063938618926e-05,
+      "loss": 0.0107,
+      "step": 2233
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.324296675191816e-05,
+      "loss": 0.0061,
+      "step": 2234
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 1.323529411764706e-05,
+      "loss": 0.0041,
+      "step": 2235
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.322762148337596e-05,
+      "loss": 0.006,
+      "step": 2236
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.3219948849104861e-05,
+      "loss": 0.0064,
+      "step": 2237
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.3212276214833759e-05,
+      "loss": 0.006,
+      "step": 2238
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 1.320460358056266e-05,
+      "loss": 0.0035,
+      "step": 2239
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.319693094629156e-05,
+      "loss": 0.8375,
+      "step": 2240
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.318925831202046e-05,
+      "loss": 0.004,
+      "step": 2241
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.318158567774936e-05,
+      "loss": 0.0077,
+      "step": 2242
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 1.3173913043478262e-05,
+      "loss": 0.006,
+      "step": 2243
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.316624040920716e-05,
+      "loss": 0.0041,
+      "step": 2244
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3158567774936062e-05,
+      "loss": 0.0034,
+      "step": 2245
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3150895140664962e-05,
+      "loss": 0.8219,
+      "step": 2246
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 1.3143222506393862e-05,
+      "loss": 0.004,
+      "step": 2247
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3135549872122762e-05,
+      "loss": 0.004,
+      "step": 2248
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3127877237851663e-05,
+      "loss": 1.1328,
+      "step": 2249
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3120204603580562e-05,
+      "loss": 0.0039,
+      "step": 2250
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 1.3112531969309463e-05,
+      "loss": 0.0283,
+      "step": 2251
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3104859335038363e-05,
+      "loss": 0.0058,
+      "step": 2252
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3097186700767263e-05,
+      "loss": 0.004,
+      "step": 2253
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3089514066496165e-05,
+      "loss": 0.0057,
+      "step": 2254
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 1.3081841432225064e-05,
+      "loss": 0.0076,
+      "step": 2255
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3074168797953966e-05,
+      "loss": 0.004,
+      "step": 2256
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3066496163682864e-05,
+      "loss": 0.0076,
+      "step": 2257
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3058823529411766e-05,
+      "loss": 0.0091,
+      "step": 2258
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 1.3051150895140666e-05,
+      "loss": 0.0038,
+      "step": 2259
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3043478260869566e-05,
+      "loss": 0.0057,
+      "step": 2260
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3035805626598466e-05,
+      "loss": 0.0057,
+      "step": 2261
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3028132992327367e-05,
+      "loss": 1.1251,
+      "step": 2262
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 1.3020460358056266e-05,
+      "loss": 1.1222,
+      "step": 2263
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.3012787723785167e-05,
+      "loss": 0.0047,
+      "step": 2264
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.3005115089514067e-05,
+      "loss": 0.008,
+      "step": 2265
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.2997442455242967e-05,
+      "loss": 0.0061,
+      "step": 2266
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 1.2989769820971867e-05,
+      "loss": 1.394,
+      "step": 2267
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2982097186700768e-05,
+      "loss": 0.0039,
+      "step": 2268
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2974424552429667e-05,
+      "loss": 0.0103,
+      "step": 2269
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2966751918158568e-05,
+      "loss": 1.1008,
+      "step": 2270
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 1.2959079283887468e-05,
+      "loss": 0.0051,
+      "step": 2271
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2951406649616368e-05,
+      "loss": 0.004,
+      "step": 2272
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2943734015345268e-05,
+      "loss": 0.78,
+      "step": 2273
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.293606138107417e-05,
+      "loss": 0.0069,
+      "step": 2274
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 1.2928388746803068e-05,
+      "loss": 0.0094,
+      "step": 2275
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.292071611253197e-05,
+      "loss": 0.0067,
+      "step": 2276
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.291304347826087e-05,
+      "loss": 0.0041,
+      "step": 2277
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 1.290537084398977e-05,
+      "loss": 0.0079,
+      "step": 2278
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.2897698209718671e-05,
+      "loss": 0.014,
+      "step": 2279
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.2890025575447571e-05,
+      "loss": 1.3775,
+      "step": 2280
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.288235294117647e-05,
+      "loss": 0.0042,
+      "step": 2281
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 1.287468030690537e-05,
+      "loss": 1.1862,
+      "step": 2282
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2867007672634272e-05,
+      "loss": 0.0042,
+      "step": 2283
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2859335038363172e-05,
+      "loss": 0.0098,
+      "step": 2284
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2851662404092072e-05,
+      "loss": 0.0038,
+      "step": 2285
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 1.2843989769820972e-05,
+      "loss": 0.0127,
+      "step": 2286
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2836317135549874e-05,
+      "loss": 0.0072,
+      "step": 2287
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2828644501278772e-05,
+      "loss": 0.0098,
+      "step": 2288
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2820971867007674e-05,
+      "loss": 0.0069,
+      "step": 2289
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 1.2813299232736573e-05,
+      "loss": 0.9529,
+      "step": 2290
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2805626598465473e-05,
+      "loss": 0.0096,
+      "step": 2291
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2797953964194373e-05,
+      "loss": 0.0094,
+      "step": 2292
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2790281329923275e-05,
+      "loss": 1.0639,
+      "step": 2293
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 1.2782608695652173e-05,
+      "loss": 0.0043,
+      "step": 2294
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2774936061381075e-05,
+      "loss": 0.0098,
+      "step": 2295
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2767263427109975e-05,
+      "loss": 0.0084,
+      "step": 2296
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2759590792838875e-05,
+      "loss": 0.0071,
+      "step": 2297
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 1.2751918158567774e-05,
+      "loss": 0.0036,
+      "step": 2298
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2744245524296676e-05,
+      "loss": 0.0097,
+      "step": 2299
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2736572890025574e-05,
+      "loss": 0.0072,
+      "step": 2300
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2728900255754476e-05,
+      "loss": 0.0043,
+      "step": 2301
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 1.2721227621483376e-05,
+      "loss": 1.3742,
+      "step": 2302
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2713554987212276e-05,
+      "loss": 0.0086,
+      "step": 2303
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2705882352941177e-05,
+      "loss": 0.0071,
+      "step": 2304
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2698209718670077e-05,
+      "loss": 0.007,
+      "step": 2305
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 1.2690537084398977e-05,
+      "loss": 0.0099,
+      "step": 2306
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2682864450127877e-05,
+      "loss": 1.0437,
+      "step": 2307
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2675191815856779e-05,
+      "loss": 0.01,
+      "step": 2308
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2667519181585679e-05,
+      "loss": 0.1771,
+      "step": 2309
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 1.2659846547314579e-05,
+      "loss": 0.0103,
+      "step": 2310
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.2652173913043479e-05,
+      "loss": 0.0104,
+      "step": 2311
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.264450127877238e-05,
+      "loss": 0.0077,
+      "step": 2312
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.2636828644501278e-05,
+      "loss": 0.0107,
+      "step": 2313
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 1.262915601023018e-05,
+      "loss": 0.0107,
+      "step": 2314
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.262148337595908e-05,
+      "loss": 0.0135,
+      "step": 2315
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.261381074168798e-05,
+      "loss": 0.0103,
+      "step": 2316
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.260613810741688e-05,
+      "loss": 0.0136,
+      "step": 2317
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 1.2598465473145781e-05,
+      "loss": 0.0045,
+      "step": 2318
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.259079283887468e-05,
+      "loss": 0.0104,
+      "step": 2319
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2583120204603581e-05,
+      "loss": 0.0073,
+      "step": 2320
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2575447570332481e-05,
+      "loss": 0.9265,
+      "step": 2321
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 1.2567774936061381e-05,
+      "loss": 0.0101,
+      "step": 2322
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2560102301790281e-05,
+      "loss": 1.3704,
+      "step": 2323
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2552429667519183e-05,
+      "loss": 1.415,
+      "step": 2324
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.254475703324808e-05,
+      "loss": 0.0098,
+      "step": 2325
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 1.2537084398976982e-05,
+      "loss": 0.0138,
+      "step": 2326
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2529411764705884e-05,
+      "loss": 0.0064,
+      "step": 2327
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2521739130434782e-05,
+      "loss": 0.0093,
+      "step": 2328
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2514066496163684e-05,
+      "loss": 0.0119,
+      "step": 2329
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 1.2506393861892584e-05,
+      "loss": 0.0097,
+      "step": 2330
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2498721227621484e-05,
+      "loss": 0.0049,
+      "step": 2331
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2491048593350384e-05,
+      "loss": 0.0044,
+      "step": 2332
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2483375959079285e-05,
+      "loss": 0.0067,
+      "step": 2333
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 1.2475703324808183e-05,
+      "loss": 0.0117,
+      "step": 2334
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2468030690537085e-05,
+      "loss": 0.0066,
+      "step": 2335
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2460358056265985e-05,
+      "loss": 1.0804,
+      "step": 2336
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2452685421994887e-05,
+      "loss": 0.0065,
+      "step": 2337
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 1.2445012787723785e-05,
+      "loss": 0.0069,
+      "step": 2338
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2437340153452686e-05,
+      "loss": 0.009,
+      "step": 2339
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2429667519181586e-05,
+      "loss": 0.0067,
+      "step": 2340
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2421994884910486e-05,
+      "loss": 0.0089,
+      "step": 2341
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 1.2414322250639386e-05,
+      "loss": 0.011,
+      "step": 2342
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2406649616368288e-05,
+      "loss": 0.0034,
+      "step": 2343
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2398976982097186e-05,
+      "loss": 1.0842,
+      "step": 2344
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2391304347826088e-05,
+      "loss": 0.0066,
+      "step": 2345
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 1.2383631713554987e-05,
+      "loss": 0.0203,
+      "step": 2346
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2375959079283887e-05,
+      "loss": 0.0046,
+      "step": 2347
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2368286445012787e-05,
+      "loss": 0.9714,
+      "step": 2348
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2360613810741689e-05,
+      "loss": 0.0066,
+      "step": 2349
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 1.2352941176470587e-05,
+      "loss": 0.0091,
+      "step": 2350
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.2345268542199489e-05,
+      "loss": 0.0066,
+      "step": 2351
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.233759590792839e-05,
+      "loss": 0.0081,
+      "step": 2352
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.2329923273657289e-05,
+      "loss": 0.006,
+      "step": 2353
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 1.232225063938619e-05,
+      "loss": 0.0066,
+      "step": 2354
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.231457800511509e-05,
+      "loss": 0.0114,
+      "step": 2355
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.230690537084399e-05,
+      "loss": 2.7382,
+      "step": 2356
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.229923273657289e-05,
+      "loss": 0.0089,
+      "step": 2357
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 1.2291560102301792e-05,
+      "loss": 0.0042,
+      "step": 2358
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.228388746803069e-05,
+      "loss": 2.3698,
+      "step": 2359
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2276214833759591e-05,
+      "loss": 0.0065,
+      "step": 2360
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2268542199488491e-05,
+      "loss": 1.0813,
+      "step": 2361
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 1.2260869565217393e-05,
+      "loss": 0.0066,
+      "step": 2362
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2253196930946291e-05,
+      "loss": 0.0092,
+      "step": 2363
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2245524296675193e-05,
+      "loss": 1.3566,
+      "step": 2364
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2237851662404093e-05,
+      "loss": 0.0092,
+      "step": 2365
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 1.2230179028132993e-05,
+      "loss": 1.0739,
+      "step": 2366
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2222506393861893e-05,
+      "loss": 0.0122,
+      "step": 2367
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2214833759590794e-05,
+      "loss": 0.0096,
+      "step": 2368
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2207161125319692e-05,
+      "loss": 0.0094,
+      "step": 2369
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 1.2199488491048594e-05,
+      "loss": 0.0097,
+      "step": 2370
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2191815856777494e-05,
+      "loss": 0.0071,
+      "step": 2371
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2184143222506394e-05,
+      "loss": 0.0085,
+      "step": 2372
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2176470588235294e-05,
+      "loss": 0.0071,
+      "step": 2373
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 1.2168797953964195e-05,
+      "loss": 0.0074,
+      "step": 2374
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2161125319693094e-05,
+      "loss": 0.0084,
+      "step": 2375
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2153452685421995e-05,
+      "loss": 0.008,
+      "step": 2376
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.9267676767676768,
+      "eval_loss": 0.3489837944507599,
+      "eval_runtime": 58.1837,
+      "eval_samples_per_second": 6.806,
+      "eval_steps_per_second": 1.702,
+      "step": 2376
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 1.2145780051150897e-05,
+      "loss": 0.0065,
+      "step": 2377
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2138107416879795e-05,
+      "loss": 0.0097,
+      "step": 2378
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2130434782608697e-05,
+      "loss": 1.3744,
+      "step": 2379
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2122762148337597e-05,
+      "loss": 0.0043,
+      "step": 2380
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 1.2115089514066496e-05,
+      "loss": 0.0071,
+      "step": 2381
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2107416879795396e-05,
+      "loss": 1.36,
+      "step": 2382
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2099744245524298e-05,
+      "loss": 0.0044,
+      "step": 2383
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2092071611253196e-05,
+      "loss": 0.0075,
+      "step": 2384
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 1.2084398976982098e-05,
+      "loss": 0.0073,
+      "step": 2385
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.2076726342710998e-05,
+      "loss": 0.0099,
+      "step": 2386
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.2069053708439898e-05,
+      "loss": 0.0048,
+      "step": 2387
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.2061381074168798e-05,
+      "loss": 0.0074,
+      "step": 2388
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 1.20537084398977e-05,
+      "loss": 0.01,
+      "step": 2389
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.2046035805626599e-05,
+      "loss": 0.0062,
+      "step": 2390
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.2038363171355499e-05,
+      "loss": 0.0039,
+      "step": 2391
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.2030690537084399e-05,
+      "loss": 0.0054,
+      "step": 2392
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 1.20230179028133e-05,
+      "loss": 1.3442,
+      "step": 2393
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.2015345268542199e-05,
+      "loss": 0.0074,
+      "step": 2394
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.20076726342711e-05,
+      "loss": 0.0063,
+      "step": 2395
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.0075,
+      "step": 2396
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 1.19923273657289e-05,
+      "loss": 0.008,
+      "step": 2397
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.19846547314578e-05,
+      "loss": 0.0053,
+      "step": 2398
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.1976982097186702e-05,
+      "loss": 0.0049,
+      "step": 2399
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.19693094629156e-05,
+      "loss": 0.0086,
+      "step": 2400
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 1.1961636828644502e-05,
+      "loss": 0.0098,
+      "step": 2401
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1953964194373403e-05,
+      "loss": 0.0062,
+      "step": 2402
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1946291560102301e-05,
+      "loss": 0.0052,
+      "step": 2403
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1938618925831203e-05,
+      "loss": 0.005,
+      "step": 2404
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 1.1930946291560103e-05,
+      "loss": 1.696,
+      "step": 2405
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1923273657289003e-05,
+      "loss": 0.0076,
+      "step": 2406
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1915601023017903e-05,
+      "loss": 1.1231,
+      "step": 2407
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1907928388746804e-05,
+      "loss": 0.0074,
+      "step": 2408
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 1.1900255754475703e-05,
+      "loss": 0.0097,
+      "step": 2409
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1892583120204604e-05,
+      "loss": 0.0102,
+      "step": 2410
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1884910485933504e-05,
+      "loss": 0.005,
+      "step": 2411
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1877237851662404e-05,
+      "loss": 0.0074,
+      "step": 2412
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 1.1869565217391304e-05,
+      "loss": 0.0071,
+      "step": 2413
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1861892583120206e-05,
+      "loss": 0.0093,
+      "step": 2414
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1854219948849106e-05,
+      "loss": 0.007,
+      "step": 2415
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1846547314578005e-05,
+      "loss": 0.0088,
+      "step": 2416
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 1.1838874680306905e-05,
+      "loss": 0.0049,
+      "step": 2417
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1831202046035807e-05,
+      "loss": 1.2268,
+      "step": 2418
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1823529411764705e-05,
+      "loss": 0.0068,
+      "step": 2419
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1815856777493607e-05,
+      "loss": 0.0112,
+      "step": 2420
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 1.1808184143222507e-05,
+      "loss": 0.0046,
+      "step": 2421
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1800511508951407e-05,
+      "loss": 0.0038,
+      "step": 2422
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1792838874680307e-05,
+      "loss": 0.0088,
+      "step": 2423
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1785166240409208e-05,
+      "loss": 0.0039,
+      "step": 2424
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 1.1777493606138106e-05,
+      "loss": 0.0107,
+      "step": 2425
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.1769820971867008e-05,
+      "loss": 0.0066,
+      "step": 2426
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.176214833759591e-05,
+      "loss": 0.0086,
+      "step": 2427
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.1754475703324808e-05,
+      "loss": 0.0064,
+      "step": 2428
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 1.174680306905371e-05,
+      "loss": 0.0064,
+      "step": 2429
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.173913043478261e-05,
+      "loss": 0.0042,
+      "step": 2430
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.173145780051151e-05,
+      "loss": 0.0063,
+      "step": 2431
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.172378516624041e-05,
+      "loss": 1.0576,
+      "step": 2432
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 1.171611253196931e-05,
+      "loss": 0.0041,
+      "step": 2433
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.1708439897698209e-05,
+      "loss": 0.0102,
+      "step": 2434
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.170076726342711e-05,
+      "loss": 1.0312,
+      "step": 2435
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.169309462915601e-05,
+      "loss": 0.1209,
+      "step": 2436
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 1.168542199488491e-05,
+      "loss": 0.004,
+      "step": 2437
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.167774936061381e-05,
+      "loss": 0.006,
+      "step": 2438
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.1670076726342712e-05,
+      "loss": 0.0081,
+      "step": 2439
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.166240409207161e-05,
+      "loss": 1.1118,
+      "step": 2440
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 1.1654731457800512e-05,
+      "loss": 0.0059,
+      "step": 2441
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1647058823529412e-05,
+      "loss": 0.0101,
+      "step": 2442
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1639386189258313e-05,
+      "loss": 0.0081,
+      "step": 2443
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1631713554987212e-05,
+      "loss": 0.0032,
+      "step": 2444
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 1.1624040920716113e-05,
+      "loss": 0.0059,
+      "step": 2445
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1616368286445013e-05,
+      "loss": 0.0105,
+      "step": 2446
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1608695652173913e-05,
+      "loss": 0.0038,
+      "step": 2447
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1601023017902813e-05,
+      "loss": 0.0104,
+      "step": 2448
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 1.1593350383631715e-05,
+      "loss": 0.0058,
+      "step": 2449
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1585677749360613e-05,
+      "loss": 0.0058,
+      "step": 2450
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1578005115089514e-05,
+      "loss": 0.0064,
+      "step": 2451
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1570332480818416e-05,
+      "loss": 0.0058,
+      "step": 2452
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 1.1562659846547314e-05,
+      "loss": 0.0077,
+      "step": 2453
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1554987212276216e-05,
+      "loss": 0.0528,
+      "step": 2454
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1547314578005116e-05,
+      "loss": 0.0078,
+      "step": 2455
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1539641943734016e-05,
+      "loss": 0.0023,
+      "step": 2456
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 1.1531969309462916e-05,
+      "loss": 0.0019,
+      "step": 2457
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1524296675191817e-05,
+      "loss": 0.0056,
+      "step": 2458
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1516624040920715e-05,
+      "loss": 0.0116,
+      "step": 2459
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1508951406649617e-05,
+      "loss": 0.0037,
+      "step": 2460
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 1.1501278772378517e-05,
+      "loss": 0.0056,
+      "step": 2461
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1493606138107417e-05,
+      "loss": 1.4099,
+      "step": 2462
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1485933503836317e-05,
+      "loss": 0.0055,
+      "step": 2463
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1478260869565218e-05,
+      "loss": 0.0018,
+      "step": 2464
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 1.1470588235294117e-05,
+      "loss": 0.0081,
+      "step": 2465
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.1462915601023018e-05,
+      "loss": 0.0056,
+      "step": 2466
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.1455242966751918e-05,
+      "loss": 0.0048,
+      "step": 2467
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.144757033248082e-05,
+      "loss": 1.352,
+      "step": 2468
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 1.1439897698209718e-05,
+      "loss": 1.3368,
+      "step": 2469
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.143222506393862e-05,
+      "loss": 0.0075,
+      "step": 2470
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.142455242966752e-05,
+      "loss": 0.0046,
+      "step": 2471
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.141687979539642e-05,
+      "loss": 0.0036,
+      "step": 2472
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 1.140920716112532e-05,
+      "loss": 0.0055,
+      "step": 2473
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 1.1401534526854221e-05,
+      "loss": 0.0065,
+      "step": 2474
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 1.1393861892583121e-05,
+      "loss": 0.0055,
+      "step": 2475
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 1.138618925831202e-05,
+      "loss": 0.0057,
+      "step": 2476
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.1378516624040922e-05,
+      "loss": 0.0047,
+      "step": 2477
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.137084398976982e-05,
+      "loss": 0.0036,
+      "step": 2478
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.1363171355498722e-05,
+      "loss": 0.0061,
+      "step": 2479
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 1.1355498721227622e-05,
+      "loss": 0.0072,
+      "step": 2480
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1347826086956522e-05,
+      "loss": 0.0054,
+      "step": 2481
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1340153452685422e-05,
+      "loss": 0.0072,
+      "step": 2482
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1332480818414324e-05,
+      "loss": 0.0041,
+      "step": 2483
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 1.1324808184143222e-05,
+      "loss": 0.0051,
+      "step": 2484
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1317135549872123e-05,
+      "loss": 1.4203,
+      "step": 2485
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1309462915601023e-05,
+      "loss": 0.0088,
+      "step": 2486
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1301790281329923e-05,
+      "loss": 0.0106,
+      "step": 2487
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 1.1294117647058823e-05,
+      "loss": 0.0071,
+      "step": 2488
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1286445012787725e-05,
+      "loss": 0.0053,
+      "step": 2489
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1278772378516623e-05,
+      "loss": 1.1147,
+      "step": 2490
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1271099744245525e-05,
+      "loss": 0.007,
+      "step": 2491
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 1.1263427109974425e-05,
+      "loss": 0.007,
+      "step": 2492
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1255754475703324e-05,
+      "loss": 0.0035,
+      "step": 2493
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1248081841432224e-05,
+      "loss": 0.0052,
+      "step": 2494
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1240409207161126e-05,
+      "loss": 0.0064,
+      "step": 2495
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 1.1232736572890026e-05,
+      "loss": 0.0046,
+      "step": 2496
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1225063938618926e-05,
+      "loss": 0.0052,
+      "step": 2497
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1217391304347826e-05,
+      "loss": 0.0045,
+      "step": 2498
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1209718670076727e-05,
+      "loss": 0.0052,
+      "step": 2499
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 1.1202046035805627e-05,
+      "loss": 0.0035,
+      "step": 2500
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1194373401534527e-05,
+      "loss": 0.0086,
+      "step": 2501
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1186700767263429e-05,
+      "loss": 0.0068,
+      "step": 2502
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1179028132992327e-05,
+      "loss": 0.0051,
+      "step": 2503
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 1.1171355498721229e-05,
+      "loss": 0.0026,
+      "step": 2504
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.1163682864450129e-05,
+      "loss": 0.0034,
+      "step": 2505
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.1156010230179029e-05,
+      "loss": 0.0052,
+      "step": 2506
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.1148337595907928e-05,
+      "loss": 0.0051,
+      "step": 2507
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 1.114066496163683e-05,
+      "loss": 0.005,
+      "step": 2508
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.1132992327365728e-05,
+      "loss": 0.005,
+      "step": 2509
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.112531969309463e-05,
+      "loss": 0.0043,
+      "step": 2510
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.111764705882353e-05,
+      "loss": 0.0065,
+      "step": 2511
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 1.110997442455243e-05,
+      "loss": 0.0039,
+      "step": 2512
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.110230179028133e-05,
+      "loss": 1.4148,
+      "step": 2513
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.1094629156010231e-05,
+      "loss": 0.0049,
+      "step": 2514
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.108695652173913e-05,
+      "loss": 0.0058,
+      "step": 2515
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 1.1079283887468031e-05,
+      "loss": 1.3611,
+      "step": 2516
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.1071611253196931e-05,
+      "loss": 0.0057,
+      "step": 2517
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.1063938618925831e-05,
+      "loss": 0.0049,
+      "step": 2518
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.105626598465473e-05,
+      "loss": 0.0049,
+      "step": 2519
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 1.1048593350383632e-05,
+      "loss": 0.0026,
+      "step": 2520
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1040920716112532e-05,
+      "loss": 0.0026,
+      "step": 2521
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1033248081841432e-05,
+      "loss": 0.4305,
+      "step": 2522
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1025575447570332e-05,
+      "loss": 0.0069,
+      "step": 2523
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 1.1017902813299234e-05,
+      "loss": 0.005,
+      "step": 2524
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.1010230179028134e-05,
+      "loss": 1.3999,
+      "step": 2525
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.1002557544757034e-05,
+      "loss": 0.1888,
+      "step": 2526
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.0994884910485935e-05,
+      "loss": 0.0049,
+      "step": 2527
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 1.0987212276214833e-05,
+      "loss": 0.8899,
+      "step": 2528
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0979539641943735e-05,
+      "loss": 1.0806,
+      "step": 2529
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0971867007672635e-05,
+      "loss": 0.0083,
+      "step": 2530
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0964194373401535e-05,
+      "loss": 0.0034,
+      "step": 2531
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 1.0956521739130435e-05,
+      "loss": 0.005,
+      "step": 2532
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0948849104859336e-05,
+      "loss": 0.0051,
+      "step": 2533
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0941176470588235e-05,
+      "loss": 0.0057,
+      "step": 2534
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0933503836317136e-05,
+      "loss": 0.0058,
+      "step": 2535
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 1.0925831202046036e-05,
+      "loss": 0.0065,
+      "step": 2536
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0918158567774936e-05,
+      "loss": 1.1058,
+      "step": 2537
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0910485933503836e-05,
+      "loss": 0.0052,
+      "step": 2538
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0902813299232738e-05,
+      "loss": 0.0082,
+      "step": 2539
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 1.0895140664961636e-05,
+      "loss": 0.005,
+      "step": 2540
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0887468030690537e-05,
+      "loss": 0.005,
+      "step": 2541
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0879795396419437e-05,
+      "loss": 0.005,
+      "step": 2542
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0872122762148337e-05,
+      "loss": 0.0034,
+      "step": 2543
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 1.0864450127877237e-05,
+      "loss": 0.0066,
+      "step": 2544
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0856777493606139e-05,
+      "loss": 0.0052,
+      "step": 2545
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0849104859335037e-05,
+      "loss": 0.0039,
+      "step": 2546
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0841432225063939e-05,
+      "loss": 0.0039,
+      "step": 2547
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 1.0833759590792839e-05,
+      "loss": 0.0064,
+      "step": 2548
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.082608695652174e-05,
+      "loss": 0.0064,
+      "step": 2549
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.081841432225064e-05,
+      "loss": 0.0052,
+      "step": 2550
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.081074168797954e-05,
+      "loss": 0.0082,
+      "step": 2551
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 1.0803069053708442e-05,
+      "loss": 0.0035,
+      "step": 2552
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.079539641943734e-05,
+      "loss": 0.0064,
+      "step": 2553
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.0787723785166241e-05,
+      "loss": 0.0064,
+      "step": 2554
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.0780051150895141e-05,
+      "loss": 0.1541,
+      "step": 2555
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 1.0772378516624041e-05,
+      "loss": 2.4749,
+      "step": 2556
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0764705882352941e-05,
+      "loss": 0.0028,
+      "step": 2557
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0757033248081843e-05,
+      "loss": 0.0034,
+      "step": 2558
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0749360613810741e-05,
+      "loss": 0.0061,
+      "step": 2559
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 1.0741687979539643e-05,
+      "loss": 0.0032,
+      "step": 2560
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0734015345268543e-05,
+      "loss": 0.0055,
+      "step": 2561
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0726342710997443e-05,
+      "loss": 0.0046,
+      "step": 2562
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0718670076726342e-05,
+      "loss": 1.185,
+      "step": 2563
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 1.0710997442455244e-05,
+      "loss": 0.0062,
+      "step": 2564
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0703324808184142e-05,
+      "loss": 0.0024,
+      "step": 2565
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0695652173913044e-05,
+      "loss": 0.0032,
+      "step": 2566
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0687979539641944e-05,
+      "loss": 0.0047,
+      "step": 2567
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 1.0680306905370844e-05,
+      "loss": 0.0047,
+      "step": 2568
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0672634271099744e-05,
+      "loss": 0.0047,
+      "step": 2569
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0664961636828645e-05,
+      "loss": 0.0065,
+      "step": 2570
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0657289002557543e-05,
+      "loss": 0.0061,
+      "step": 2571
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 1.0649616368286445e-05,
+      "loss": 0.0048,
+      "step": 2572
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 1.0641943734015345e-05,
+      "loss": 0.0015,
+      "step": 2573
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 1.0634271099744247e-05,
+      "loss": 0.0046,
+      "step": 2574
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 1.0626598465473147e-05,
+      "loss": 0.0033,
+      "step": 2575
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0618925831202046e-05,
+      "loss": 0.0062,
+      "step": 2576
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0611253196930948e-05,
+      "loss": 1.4243,
+      "step": 2577
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0603580562659846e-05,
+      "loss": 0.012,
+      "step": 2578
+    },
+    {
+      "epoch": 6.51,
+      "learning_rate": 1.0595907928388748e-05,
+      "loss": 2.3898,
+      "step": 2579
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.0588235294117648e-05,
+      "loss": 0.0032,
+      "step": 2580
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.0580562659846548e-05,
+      "loss": 0.0032,
+      "step": 2581
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.0572890025575448e-05,
+      "loss": 0.0061,
+      "step": 2582
+    },
+    {
+      "epoch": 6.52,
+      "learning_rate": 1.056521739130435e-05,
+      "loss": 0.0047,
+      "step": 2583
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0557544757033247e-05,
+      "loss": 0.0047,
+      "step": 2584
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0549872122762149e-05,
+      "loss": 0.2081,
+      "step": 2585
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0542199488491049e-05,
+      "loss": 0.0053,
+      "step": 2586
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 1.0534526854219949e-05,
+      "loss": 0.0047,
+      "step": 2587
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.0526854219948849e-05,
+      "loss": 1.3858,
+      "step": 2588
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.051918158567775e-05,
+      "loss": 0.0062,
+      "step": 2589
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.0511508951406649e-05,
+      "loss": 0.0041,
+      "step": 2590
+    },
+    {
+      "epoch": 6.54,
+      "learning_rate": 1.050383631713555e-05,
+      "loss": 1.1796,
+      "step": 2591
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.049616368286445e-05,
+      "loss": 1.0994,
+      "step": 2592
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.048849104859335e-05,
+      "loss": 0.8511,
+      "step": 2593
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.048081841432225e-05,
+      "loss": 1.2895,
+      "step": 2594
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 1.0473145780051152e-05,
+      "loss": 0.0035,
+      "step": 2595
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.046547314578005e-05,
+      "loss": 0.0065,
+      "step": 2596
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.0457800511508952e-05,
+      "loss": 1.239,
+      "step": 2597
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.0450127877237851e-05,
+      "loss": 1.1663,
+      "step": 2598
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 1.0442455242966751e-05,
+      "loss": 0.0051,
+      "step": 2599
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0434782608695653e-05,
+      "loss": 1.1103,
+      "step": 2600
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0427109974424553e-05,
+      "loss": 1.4103,
+      "step": 2601
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0419437340153454e-05,
+      "loss": 0.0053,
+      "step": 2602
+    },
+    {
+      "epoch": 6.57,
+      "learning_rate": 1.0411764705882353e-05,
+      "loss": 0.0072,
+      "step": 2603
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0404092071611254e-05,
+      "loss": 0.0083,
+      "step": 2604
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0396419437340154e-05,
+      "loss": 0.0039,
+      "step": 2605
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0388746803069054e-05,
+      "loss": 0.0049,
+      "step": 2606
+    },
+    {
+      "epoch": 6.58,
+      "learning_rate": 1.0381074168797954e-05,
+      "loss": 0.0098,
+      "step": 2607
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0373401534526856e-05,
+      "loss": 0.0057,
+      "step": 2608
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0365728900255754e-05,
+      "loss": 0.004,
+      "step": 2609
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0358056265984656e-05,
+      "loss": 0.0098,
+      "step": 2610
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 1.0350383631713555e-05,
+      "loss": 0.0059,
+      "step": 2611
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0342710997442455e-05,
+      "loss": 0.0065,
+      "step": 2612
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0335038363171355e-05,
+      "loss": 0.0059,
+      "step": 2613
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0327365728900257e-05,
+      "loss": 0.0112,
+      "step": 2614
+    },
+    {
+      "epoch": 6.6,
+      "learning_rate": 1.0319693094629155e-05,
+      "loss": 0.0061,
+      "step": 2615
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0312020460358057e-05,
+      "loss": 0.0039,
+      "step": 2616
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0304347826086957e-05,
+      "loss": 0.0083,
+      "step": 2617
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0296675191815857e-05,
+      "loss": 0.0057,
+      "step": 2618
+    },
+    {
+      "epoch": 6.61,
+      "learning_rate": 1.0289002557544756e-05,
+      "loss": 1.3969,
+      "step": 2619
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0281329923273658e-05,
+      "loss": 0.0065,
+      "step": 2620
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0273657289002556e-05,
+      "loss": 1.0742,
+      "step": 2621
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0265984654731458e-05,
+      "loss": 0.0039,
+      "step": 2622
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 1.0258312020460358e-05,
+      "loss": 0.0065,
+      "step": 2623
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.0250639386189258e-05,
+      "loss": 0.005,
+      "step": 2624
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.024296675191816e-05,
+      "loss": 0.0077,
+      "step": 2625
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.023529411764706e-05,
+      "loss": 1.2389,
+      "step": 2626
+    },
+    {
+      "epoch": 6.63,
+      "learning_rate": 1.0227621483375961e-05,
+      "loss": 0.0076,
+      "step": 2627
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.0219948849104859e-05,
+      "loss": 1.034,
+      "step": 2628
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.021227621483376e-05,
+      "loss": 0.0058,
+      "step": 2629
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.020460358056266e-05,
+      "loss": 0.0059,
+      "step": 2630
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 1.019693094629156e-05,
+      "loss": 0.0079,
+      "step": 2631
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.018925831202046e-05,
+      "loss": 0.0043,
+      "step": 2632
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.0181585677749362e-05,
+      "loss": 0.004,
+      "step": 2633
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.017391304347826e-05,
+      "loss": 0.0075,
+      "step": 2634
+    },
+    {
+      "epoch": 6.65,
+      "learning_rate": 1.0166240409207162e-05,
+      "loss": 0.0076,
+      "step": 2635
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0158567774936062e-05,
+      "loss": 0.0102,
+      "step": 2636
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0150895140664962e-05,
+      "loss": 1.1332,
+      "step": 2637
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0143222506393862e-05,
+      "loss": 0.0096,
+      "step": 2638
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 1.0135549872122763e-05,
+      "loss": 0.0092,
+      "step": 2639
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0127877237851662e-05,
+      "loss": 1.3919,
+      "step": 2640
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0120204603580563e-05,
+      "loss": 0.0058,
+      "step": 2641
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0112531969309463e-05,
+      "loss": 1.0521,
+      "step": 2642
+    },
+    {
+      "epoch": 6.67,
+      "learning_rate": 1.0104859335038363e-05,
+      "loss": 0.0045,
+      "step": 2643
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0097186700767263e-05,
+      "loss": 0.0059,
+      "step": 2644
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0089514066496164e-05,
+      "loss": 0.005,
+      "step": 2645
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0081841432225063e-05,
+      "loss": 0.0075,
+      "step": 2646
+    },
+    {
+      "epoch": 6.68,
+      "learning_rate": 1.0074168797953964e-05,
+      "loss": 0.0403,
+      "step": 2647
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0066496163682864e-05,
+      "loss": 0.0079,
+      "step": 2648
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0058823529411764e-05,
+      "loss": 0.0062,
+      "step": 2649
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0051150895140666e-05,
+      "loss": 0.0061,
+      "step": 2650
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 1.0043478260869566e-05,
+      "loss": 0.0064,
+      "step": 2651
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0035805626598466e-05,
+      "loss": 0.0077,
+      "step": 2652
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0028132992327366e-05,
+      "loss": 0.0884,
+      "step": 2653
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0020460358056267e-05,
+      "loss": 0.0062,
+      "step": 2654
+    },
+    {
+      "epoch": 6.7,
+      "learning_rate": 1.0012787723785167e-05,
+      "loss": 1.1006,
+      "step": 2655
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 1.0005115089514067e-05,
+      "loss": 0.0109,
+      "step": 2656
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 9.997442455242967e-06,
+      "loss": 0.0059,
+      "step": 2657
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 9.989769820971869e-06,
+      "loss": 0.0084,
+      "step": 2658
+    },
+    {
+      "epoch": 6.71,
+      "learning_rate": 9.982097186700767e-06,
+      "loss": 0.0058,
+      "step": 2659
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.974424552429668e-06,
+      "loss": 0.0059,
+      "step": 2660
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.966751918158568e-06,
+      "loss": 0.0049,
+      "step": 2661
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.959079283887468e-06,
+      "loss": 0.0079,
+      "step": 2662
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 9.951406649616368e-06,
+      "loss": 1.5801,
+      "step": 2663
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.94373401534527e-06,
+      "loss": 0.0059,
+      "step": 2664
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.936061381074168e-06,
+      "loss": 1.2523,
+      "step": 2665
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.92838874680307e-06,
+      "loss": 0.0079,
+      "step": 2666
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 9.92071611253197e-06,
+      "loss": 0.0099,
+      "step": 2667
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.91304347826087e-06,
+      "loss": 0.0082,
+      "step": 2668
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.90537084398977e-06,
+      "loss": 0.004,
+      "step": 2669
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.897698209718671e-06,
+      "loss": 0.0064,
+      "step": 2670
+    },
+    {
+      "epoch": 6.74,
+      "learning_rate": 9.890025575447569e-06,
+      "loss": 0.004,
+      "step": 2671
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 9.88235294117647e-06,
+      "loss": 0.0943,
+      "step": 2672
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 9.87468030690537e-06,
+      "loss": 0.0059,
+      "step": 2673
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 9.86700767263427e-06,
+      "loss": 0.0098,
+      "step": 2674
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.859335038363172e-06,
+      "loss": 0.0065,
+      "step": 2675
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.851662404092072e-06,
+      "loss": 0.0075,
+      "step": 2676
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.843989769820972e-06,
+      "loss": 0.0044,
+      "step": 2677
+    },
+    {
+      "epoch": 6.76,
+      "learning_rate": 9.836317135549872e-06,
+      "loss": 0.0063,
+      "step": 2678
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.828644501278774e-06,
+      "loss": 0.0038,
+      "step": 2679
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.820971867007673e-06,
+      "loss": 0.0067,
+      "step": 2680
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.813299232736573e-06,
+      "loss": 1.1148,
+      "step": 2681
+    },
+    {
+      "epoch": 6.77,
+      "learning_rate": 9.805626598465473e-06,
+      "loss": 0.0056,
+      "step": 2682
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.797953964194375e-06,
+      "loss": 0.0096,
+      "step": 2683
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.790281329923273e-06,
+      "loss": 0.011,
+      "step": 2684
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.782608695652175e-06,
+      "loss": 0.0038,
+      "step": 2685
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 9.774936061381075e-06,
+      "loss": 0.0129,
+      "step": 2686
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.767263427109975e-06,
+      "loss": 0.0063,
+      "step": 2687
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.759590792838875e-06,
+      "loss": 0.0561,
+      "step": 2688
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.751918158567776e-06,
+      "loss": 0.0041,
+      "step": 2689
+    },
+    {
+      "epoch": 6.79,
+      "learning_rate": 9.744245524296674e-06,
+      "loss": 0.0048,
+      "step": 2690
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.736572890025576e-06,
+      "loss": 0.0072,
+      "step": 2691
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.728900255754476e-06,
+      "loss": 0.007,
+      "step": 2692
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.721227621483376e-06,
+      "loss": 0.0031,
+      "step": 2693
+    },
+    {
+      "epoch": 6.8,
+      "learning_rate": 9.713554987212276e-06,
+      "loss": 1.4136,
+      "step": 2694
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.705882352941177e-06,
+      "loss": 0.0081,
+      "step": 2695
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.698209718670076e-06,
+      "loss": 0.0036,
+      "step": 2696
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.690537084398977e-06,
+      "loss": 1.382,
+      "step": 2697
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 9.682864450127877e-06,
+      "loss": 0.0054,
+      "step": 2698
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.675191815856777e-06,
+      "loss": 0.0064,
+      "step": 2699
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.667519181585679e-06,
+      "loss": 0.0072,
+      "step": 2700
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.659846547314579e-06,
+      "loss": 0.0054,
+      "step": 2701
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 9.652173913043478e-06,
+      "loss": 0.0108,
+      "step": 2702
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.644501278772378e-06,
+      "loss": 0.0037,
+      "step": 2703
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.63682864450128e-06,
+      "loss": 0.0071,
+      "step": 2704
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.629156010230178e-06,
+      "loss": 0.0062,
+      "step": 2705
+    },
+    {
+      "epoch": 6.83,
+      "learning_rate": 9.62148337595908e-06,
+      "loss": 0.0056,
+      "step": 2706
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.61381074168798e-06,
+      "loss": 1.1276,
+      "step": 2707
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.606138107416881e-06,
+      "loss": 0.0045,
+      "step": 2708
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.59846547314578e-06,
+      "loss": 0.0073,
+      "step": 2709
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 9.590792838874681e-06,
+      "loss": 0.3343,
+      "step": 2710
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.583120204603581e-06,
+      "loss": 0.0038,
+      "step": 2711
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.575447570332481e-06,
+      "loss": 0.0053,
+      "step": 2712
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.567774936061381e-06,
+      "loss": 0.0037,
+      "step": 2713
+    },
+    {
+      "epoch": 6.85,
+      "learning_rate": 9.560102301790283e-06,
+      "loss": 0.0036,
+      "step": 2714
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.55242966751918e-06,
+      "loss": 0.0053,
+      "step": 2715
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.544757033248082e-06,
+      "loss": 0.0069,
+      "step": 2716
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.537084398976982e-06,
+      "loss": 1.3084,
+      "step": 2717
+    },
+    {
+      "epoch": 6.86,
+      "learning_rate": 9.529411764705882e-06,
+      "loss": 0.0022,
+      "step": 2718
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.521739130434782e-06,
+      "loss": 0.0046,
+      "step": 2719
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.514066496163684e-06,
+      "loss": 1.1433,
+      "step": 2720
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.506393861892582e-06,
+      "loss": 0.0053,
+      "step": 2721
+    },
+    {
+      "epoch": 6.87,
+      "learning_rate": 9.498721227621484e-06,
+      "loss": 0.0053,
+      "step": 2722
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.491048593350385e-06,
+      "loss": 0.0057,
+      "step": 2723
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.483375959079283e-06,
+      "loss": 0.0058,
+      "step": 2724
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.475703324808185e-06,
+      "loss": 0.0029,
+      "step": 2725
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 9.468030690537085e-06,
+      "loss": 0.0061,
+      "step": 2726
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.460358056265985e-06,
+      "loss": 0.0058,
+      "step": 2727
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.452685421994885e-06,
+      "loss": 0.0074,
+      "step": 2728
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.445012787723786e-06,
+      "loss": 0.007,
+      "step": 2729
+    },
+    {
+      "epoch": 6.89,
+      "learning_rate": 9.437340153452685e-06,
+      "loss": 1.3508,
+      "step": 2730
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.429667519181586e-06,
+      "loss": 0.0074,
+      "step": 2731
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.421994884910486e-06,
+      "loss": 0.0037,
+      "step": 2732
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.414322250639388e-06,
+      "loss": 0.003,
+      "step": 2733
+    },
+    {
+      "epoch": 6.9,
+      "learning_rate": 9.406649616368286e-06,
+      "loss": 0.0038,
+      "step": 2734
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.398976982097188e-06,
+      "loss": 0.0086,
+      "step": 2735
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.391304347826087e-06,
+      "loss": 0.0058,
+      "step": 2736
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.383631713554987e-06,
+      "loss": 0.0021,
+      "step": 2737
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 9.375959079283887e-06,
+      "loss": 0.0046,
+      "step": 2738
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.368286445012789e-06,
+      "loss": 0.0035,
+      "step": 2739
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.360613810741687e-06,
+      "loss": 0.0054,
+      "step": 2740
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.352941176470589e-06,
+      "loss": 0.0052,
+      "step": 2741
+    },
+    {
+      "epoch": 6.92,
+      "learning_rate": 9.345268542199489e-06,
+      "loss": 0.0072,
+      "step": 2742
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.337595907928389e-06,
+      "loss": 0.1932,
+      "step": 2743
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.329923273657289e-06,
+      "loss": 0.006,
+      "step": 2744
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.32225063938619e-06,
+      "loss": 0.0036,
+      "step": 2745
+    },
+    {
+      "epoch": 6.93,
+      "learning_rate": 9.314578005115088e-06,
+      "loss": 1.2929,
+      "step": 2746
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.30690537084399e-06,
+      "loss": 0.0028,
+      "step": 2747
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.299232736572892e-06,
+      "loss": 1.2908,
+      "step": 2748
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.29156010230179e-06,
+      "loss": 0.0067,
+      "step": 2749
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 9.283887468030691e-06,
+      "loss": 0.0043,
+      "step": 2750
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.276214833759591e-06,
+      "loss": 0.0054,
+      "step": 2751
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.268542199488491e-06,
+      "loss": 0.0055,
+      "step": 2752
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.260869565217391e-06,
+      "loss": 1.1614,
+      "step": 2753
+    },
+    {
+      "epoch": 6.95,
+      "learning_rate": 9.253196930946293e-06,
+      "loss": 0.0071,
+      "step": 2754
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.245524296675191e-06,
+      "loss": 0.0035,
+      "step": 2755
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.237851662404093e-06,
+      "loss": 0.0054,
+      "step": 2756
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.230179028132993e-06,
+      "loss": 0.0096,
+      "step": 2757
+    },
+    {
+      "epoch": 6.96,
+      "learning_rate": 9.222506393861894e-06,
+      "loss": 0.0035,
+      "step": 2758
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.214833759590792e-06,
+      "loss": 0.0051,
+      "step": 2759
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.207161125319694e-06,
+      "loss": 0.0069,
+      "step": 2760
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.199488491048594e-06,
+      "loss": 0.0018,
+      "step": 2761
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 9.191815856777494e-06,
+      "loss": 0.0026,
+      "step": 2762
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.184143222506394e-06,
+      "loss": 0.0054,
+      "step": 2763
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.176470588235295e-06,
+      "loss": 0.0085,
+      "step": 2764
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.168797953964194e-06,
+      "loss": 0.008,
+      "step": 2765
+    },
+    {
+      "epoch": 6.98,
+      "learning_rate": 9.161125319693095e-06,
+      "loss": 1.3292,
+      "step": 2766
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.153452685421995e-06,
+      "loss": 0.0085,
+      "step": 2767
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.145780051150895e-06,
+      "loss": 0.0034,
+      "step": 2768
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.138107416879795e-06,
+      "loss": 0.0051,
+      "step": 2769
+    },
+    {
+      "epoch": 6.99,
+      "learning_rate": 9.130434782608697e-06,
+      "loss": 0.0054,
+      "step": 2770
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 9.122762148337595e-06,
+      "loss": 0.0054,
+      "step": 2771
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 9.115089514066496e-06,
+      "loss": 0.0083,
+      "step": 2772
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.9217171717171717,
+      "eval_loss": 0.37921515107154846,
+      "eval_runtime": 57.416,
+      "eval_samples_per_second": 6.897,
+      "eval_steps_per_second": 1.724,
+      "step": 2772
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 9.107416879795398e-06,
+      "loss": 1.4643,
+      "step": 2773
+    },
+    {
+      "epoch": 7.01,
+      "learning_rate": 9.099744245524296e-06,
+      "loss": 0.0067,
+      "step": 2774
+    },
+    {
+      "epoch": 7.01,
+      "learning_rate": 9.092071611253198e-06,
+      "loss": 1.362,
+      "step": 2775
+    },
+    {
+      "epoch": 7.01,
+      "learning_rate": 9.084398976982098e-06,
+      "loss": 0.0035,
+      "step": 2776
+    },
+    {
+      "epoch": 7.01,
+      "learning_rate": 9.076726342710998e-06,
+      "loss": 0.0043,
+      "step": 2777
+    },
+    {
+      "epoch": 7.02,
+      "learning_rate": 9.069053708439898e-06,
+      "loss": 0.005,
+      "step": 2778
+    },
+    {
+      "epoch": 7.02,
+      "learning_rate": 9.0613810741688e-06,
+      "loss": 0.0049,
+      "step": 2779
+    },
+    {
+      "epoch": 7.02,
+      "learning_rate": 9.053708439897697e-06,
+      "loss": 0.007,
+      "step": 2780
+    },
+    {
+      "epoch": 7.02,
+      "learning_rate": 9.046035805626599e-06,
+      "loss": 0.0038,
+      "step": 2781
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 9.038363171355499e-06,
+      "loss": 0.7287,
+      "step": 2782
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 9.030690537084399e-06,
+      "loss": 0.0089,
+      "step": 2783
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 9.023017902813299e-06,
+      "loss": 0.0067,
+      "step": 2784
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 9.0153452685422e-06,
+      "loss": 0.0065,
+      "step": 2785
+    },
+    {
+      "epoch": 7.04,
+      "learning_rate": 9.0076726342711e-06,
+      "loss": 0.0066,
+      "step": 2786
+    },
+    {
+      "epoch": 7.04,
+      "learning_rate": 9e-06,
+      "loss": 1.341,
+      "step": 2787
+    },
+    {
+      "epoch": 7.04,
+      "learning_rate": 8.9923273657289e-06,
+      "loss": 0.0039,
+      "step": 2788
+    },
+    {
+      "epoch": 7.04,
+      "learning_rate": 8.984654731457802e-06,
+      "loss": 0.0053,
+      "step": 2789
+    },
+    {
+      "epoch": 7.05,
+      "learning_rate": 8.9769820971867e-06,
+      "loss": 0.0058,
+      "step": 2790
+    },
+    {
+      "epoch": 7.05,
+      "learning_rate": 8.969309462915602e-06,
+      "loss": 1.0835,
+      "step": 2791
+    },
+    {
+      "epoch": 7.05,
+      "learning_rate": 8.961636828644502e-06,
+      "loss": 0.0059,
+      "step": 2792
+    },
+    {
+      "epoch": 7.05,
+      "learning_rate": 8.953964194373401e-06,
+      "loss": 0.0067,
+      "step": 2793
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 8.946291560102301e-06,
+      "loss": 0.0059,
+      "step": 2794
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 8.938618925831203e-06,
+      "loss": 0.0031,
+      "step": 2795
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 8.930946291560101e-06,
+      "loss": 0.005,
+      "step": 2796
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 8.923273657289003e-06,
+      "loss": 0.008,
+      "step": 2797
+    },
+    {
+      "epoch": 7.07,
+      "learning_rate": 8.915601023017904e-06,
+      "loss": 0.0052,
+      "step": 2798
+    },
+    {
+      "epoch": 7.07,
+      "learning_rate": 8.907928388746803e-06,
+      "loss": 0.0634,
+      "step": 2799
+    },
+    {
+      "epoch": 7.07,
+      "learning_rate": 8.900255754475704e-06,
+      "loss": 0.0053,
+      "step": 2800
+    },
+    {
+      "epoch": 7.07,
+      "learning_rate": 8.892583120204604e-06,
+      "loss": 0.0054,
+      "step": 2801
+    },
+    {
+      "epoch": 7.08,
+      "learning_rate": 8.884910485933504e-06,
+      "loss": 0.0066,
+      "step": 2802
+    },
+    {
+      "epoch": 7.08,
+      "learning_rate": 8.877237851662404e-06,
+      "loss": 0.0067,
+      "step": 2803
+    },
+    {
+      "epoch": 7.08,
+      "learning_rate": 8.869565217391306e-06,
+      "loss": 0.0488,
+      "step": 2804
+    },
+    {
+      "epoch": 7.08,
+      "learning_rate": 8.861892583120204e-06,
+      "loss": 0.0048,
+      "step": 2805
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 8.854219948849105e-06,
+      "loss": 0.0065,
+      "step": 2806
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 8.846547314578005e-06,
+      "loss": 0.0089,
+      "step": 2807
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 8.838874680306905e-06,
+      "loss": 0.0052,
+      "step": 2808
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 8.831202046035805e-06,
+      "loss": 0.006,
+      "step": 2809
+    },
+    {
+      "epoch": 7.1,
+      "learning_rate": 8.823529411764707e-06,
+      "loss": 0.0032,
+      "step": 2810
+    },
+    {
+      "epoch": 7.1,
+      "learning_rate": 8.815856777493607e-06,
+      "loss": 0.0036,
+      "step": 2811
+    },
+    {
+      "epoch": 7.1,
+      "learning_rate": 8.808184143222507e-06,
+      "loss": 0.0032,
+      "step": 2812
+    },
+    {
+      "epoch": 7.1,
+      "learning_rate": 8.800511508951407e-06,
+      "loss": 0.0035,
+      "step": 2813
+    },
+    {
+      "epoch": 7.11,
+      "learning_rate": 8.792838874680308e-06,
+      "loss": 1.4438,
+      "step": 2814
+    },
+    {
+      "epoch": 7.11,
+      "learning_rate": 8.785166240409206e-06,
+      "loss": 0.0048,
+      "step": 2815
+    },
+    {
+      "epoch": 7.11,
+      "learning_rate": 8.777493606138108e-06,
+      "loss": 0.0052,
+      "step": 2816
+    },
+    {
+      "epoch": 7.11,
+      "learning_rate": 8.769820971867008e-06,
+      "loss": 0.0038,
+      "step": 2817
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 8.762148337595908e-06,
+      "loss": 0.7281,
+      "step": 2818
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 8.754475703324808e-06,
+      "loss": 1.3888,
+      "step": 2819
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 8.74680306905371e-06,
+      "loss": 0.0016,
+      "step": 2820
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 8.739130434782608e-06,
+      "loss": 0.0065,
+      "step": 2821
+    },
+    {
+      "epoch": 7.13,
+      "learning_rate": 8.73145780051151e-06,
+      "loss": 1.1887,
+      "step": 2822
+    },
+    {
+      "epoch": 7.13,
+      "learning_rate": 8.72378516624041e-06,
+      "loss": 0.0069,
+      "step": 2823
+    },
+    {
+      "epoch": 7.13,
+      "learning_rate": 8.716112531969309e-06,
+      "loss": 0.0034,
+      "step": 2824
+    },
+    {
+      "epoch": 7.13,
+      "learning_rate": 8.70843989769821e-06,
+      "loss": 1.1691,
+      "step": 2825
+    },
+    {
+      "epoch": 7.14,
+      "learning_rate": 8.70076726342711e-06,
+      "loss": 0.0033,
+      "step": 2826
+    },
+    {
+      "epoch": 7.14,
+      "learning_rate": 8.69309462915601e-06,
+      "loss": 0.0088,
+      "step": 2827
+    },
+    {
+      "epoch": 7.14,
+      "learning_rate": 8.68542199488491e-06,
+      "loss": 0.0071,
+      "step": 2828
+    },
+    {
+      "epoch": 7.14,
+      "learning_rate": 8.677749360613812e-06,
+      "loss": 0.0023,
+      "step": 2829
+    },
+    {
+      "epoch": 7.15,
+      "learning_rate": 8.67007672634271e-06,
+      "loss": 0.0036,
+      "step": 2830
+    },
+    {
+      "epoch": 7.15,
+      "learning_rate": 8.662404092071612e-06,
+      "loss": 0.0088,
+      "step": 2831
+    },
+    {
+      "epoch": 7.15,
+      "learning_rate": 8.654731457800512e-06,
+      "loss": 1.1488,
+      "step": 2832
+    },
+    {
+      "epoch": 7.15,
+      "learning_rate": 8.647058823529412e-06,
+      "loss": 0.0067,
+      "step": 2833
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 8.639386189258312e-06,
+      "loss": 0.0031,
+      "step": 2834
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 8.631713554987213e-06,
+      "loss": 0.0036,
+      "step": 2835
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 8.624040920716111e-06,
+      "loss": 0.0058,
+      "step": 2836
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 8.616368286445013e-06,
+      "loss": 0.03,
+      "step": 2837
+    },
+    {
+      "epoch": 7.17,
+      "learning_rate": 8.608695652173913e-06,
+      "loss": 0.0046,
+      "step": 2838
+    },
+    {
+      "epoch": 7.17,
+      "learning_rate": 8.601023017902815e-06,
+      "loss": 0.0093,
+      "step": 2839
+    },
+    {
+      "epoch": 7.17,
+      "learning_rate": 8.593350383631713e-06,
+      "loss": 0.0054,
+      "step": 2840
+    },
+    {
+      "epoch": 7.17,
+      "learning_rate": 8.585677749360614e-06,
+      "loss": 0.0051,
+      "step": 2841
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 8.578005115089514e-06,
+      "loss": 0.0036,
+      "step": 2842
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 8.570332480818414e-06,
+      "loss": 0.0016,
+      "step": 2843
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 8.562659846547314e-06,
+      "loss": 0.0057,
+      "step": 2844
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 8.554987212276216e-06,
+      "loss": 0.0041,
+      "step": 2845
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 8.547314578005114e-06,
+      "loss": 0.0035,
+      "step": 2846
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 8.539641943734016e-06,
+      "loss": 0.0084,
+      "step": 2847
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 8.531969309462917e-06,
+      "loss": 0.0065,
+      "step": 2848
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 8.524296675191815e-06,
+      "loss": 0.0066,
+      "step": 2849
+    },
+    {
+      "epoch": 7.2,
+      "learning_rate": 8.516624040920717e-06,
+      "loss": 0.0035,
+      "step": 2850
+    },
+    {
+      "epoch": 7.2,
+      "learning_rate": 8.508951406649617e-06,
+      "loss": 0.0052,
+      "step": 2851
+    },
+    {
+      "epoch": 7.2,
+      "learning_rate": 8.501278772378517e-06,
+      "loss": 0.0065,
+      "step": 2852
+    },
+    {
+      "epoch": 7.2,
+      "learning_rate": 8.493606138107417e-06,
+      "loss": 0.0064,
+      "step": 2853
+    },
+    {
+      "epoch": 7.21,
+      "learning_rate": 8.485933503836318e-06,
+      "loss": 0.0154,
+      "step": 2854
+    },
+    {
+      "epoch": 7.21,
+      "learning_rate": 8.478260869565217e-06,
+      "loss": 0.0114,
+      "step": 2855
+    },
+    {
+      "epoch": 7.21,
+      "learning_rate": 8.470588235294118e-06,
+      "loss": 0.0063,
+      "step": 2856
+    },
+    {
+      "epoch": 7.21,
+      "learning_rate": 8.462915601023018e-06,
+      "loss": 0.0097,
+      "step": 2857
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 8.455242966751918e-06,
+      "loss": 0.0051,
+      "step": 2858
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 8.447570332480818e-06,
+      "loss": 0.0081,
+      "step": 2859
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 8.43989769820972e-06,
+      "loss": 0.0046,
+      "step": 2860
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 8.432225063938618e-06,
+      "loss": 0.0034,
+      "step": 2861
+    },
+    {
+      "epoch": 7.23,
+      "learning_rate": 8.42455242966752e-06,
+      "loss": 0.0053,
+      "step": 2862
+    },
+    {
+      "epoch": 7.23,
+      "learning_rate": 8.41687979539642e-06,
+      "loss": 0.0071,
+      "step": 2863
+    },
+    {
+      "epoch": 7.23,
+      "learning_rate": 8.409207161125321e-06,
+      "loss": 0.0062,
+      "step": 2864
+    },
+    {
+      "epoch": 7.23,
+      "learning_rate": 8.40153452685422e-06,
+      "loss": 0.0106,
+      "step": 2865
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 8.39386189258312e-06,
+      "loss": 0.0038,
+      "step": 2866
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 8.38618925831202e-06,
+      "loss": 0.003,
+      "step": 2867
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 8.37851662404092e-06,
+      "loss": 0.0044,
+      "step": 2868
+    },
+    {
+      "epoch": 7.24,
+      "learning_rate": 8.37084398976982e-06,
+      "loss": 0.0061,
+      "step": 2869
+    },
+    {
+      "epoch": 7.25,
+      "learning_rate": 8.363171355498722e-06,
+      "loss": 0.0061,
+      "step": 2870
+    },
+    {
+      "epoch": 7.25,
+      "learning_rate": 8.355498721227622e-06,
+      "loss": 0.0071,
+      "step": 2871
+    },
+    {
+      "epoch": 7.25,
+      "learning_rate": 8.347826086956522e-06,
+      "loss": 0.0044,
+      "step": 2872
+    },
+    {
+      "epoch": 7.26,
+      "learning_rate": 8.340153452685424e-06,
+      "loss": 0.0429,
+      "step": 2873
+    },
+    {
+      "epoch": 7.26,
+      "learning_rate": 8.332480818414322e-06,
+      "loss": 0.002,
+      "step": 2874
+    },
+    {
+      "epoch": 7.26,
+      "learning_rate": 8.324808184143223e-06,
+      "loss": 0.0056,
+      "step": 2875
+    },
+    {
+      "epoch": 7.26,
+      "learning_rate": 8.317135549872123e-06,
+      "loss": 0.0161,
+      "step": 2876
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 8.309462915601023e-06,
+      "loss": 0.0049,
+      "step": 2877
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 8.301790281329923e-06,
+      "loss": 0.0082,
+      "step": 2878
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 8.294117647058825e-06,
+      "loss": 0.0055,
+      "step": 2879
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 8.286445012787723e-06,
+      "loss": 0.0029,
+      "step": 2880
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 8.278772378516625e-06,
+      "loss": 0.0067,
+      "step": 2881
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 8.271099744245525e-06,
+      "loss": 0.0046,
+      "step": 2882
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 8.263427109974425e-06,
+      "loss": 0.0029,
+      "step": 2883
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 8.255754475703324e-06,
+      "loss": 1.4067,
+      "step": 2884
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 8.248081841432226e-06,
+      "loss": 0.0056,
+      "step": 2885
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 8.240409207161124e-06,
+      "loss": 0.007,
+      "step": 2886
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 8.232736572890026e-06,
+      "loss": 0.0044,
+      "step": 2887
+    },
+    {
+      "epoch": 7.29,
+      "learning_rate": 8.225063938618926e-06,
+      "loss": 0.2182,
+      "step": 2888
+    },
+    {
+      "epoch": 7.3,
+      "learning_rate": 8.217391304347826e-06,
+      "loss": 0.0028,
+      "step": 2889
+    },
+    {
+      "epoch": 7.3,
+      "learning_rate": 8.209718670076726e-06,
+      "loss": 0.0028,
+      "step": 2890
+    },
+    {
+      "epoch": 7.3,
+      "learning_rate": 8.202046035805627e-06,
+      "loss": 0.0085,
+      "step": 2891
+    },
+    {
+      "epoch": 7.3,
+      "learning_rate": 8.194373401534527e-06,
+      "loss": 0.0059,
+      "step": 2892
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 8.186700767263427e-06,
+      "loss": 0.004,
+      "step": 2893
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 8.179028132992327e-06,
+      "loss": 0.0057,
+      "step": 2894
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 8.171355498721229e-06,
+      "loss": 0.0052,
+      "step": 2895
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 8.163682864450129e-06,
+      "loss": 0.0041,
+      "step": 2896
+    },
+    {
+      "epoch": 7.32,
+      "learning_rate": 8.156010230179028e-06,
+      "loss": 0.0042,
+      "step": 2897
+    },
+    {
+      "epoch": 7.32,
+      "learning_rate": 8.14833759590793e-06,
+      "loss": 0.0066,
+      "step": 2898
+    },
+    {
+      "epoch": 7.32,
+      "learning_rate": 8.140664961636828e-06,
+      "loss": 0.0042,
+      "step": 2899
+    },
+    {
+      "epoch": 7.32,
+      "learning_rate": 8.13299232736573e-06,
+      "loss": 0.0042,
+      "step": 2900
+    },
+    {
+      "epoch": 7.33,
+      "learning_rate": 8.12531969309463e-06,
+      "loss": 0.0019,
+      "step": 2901
+    },
+    {
+      "epoch": 7.33,
+      "learning_rate": 8.11764705882353e-06,
+      "loss": 0.0054,
+      "step": 2902
+    },
+    {
+      "epoch": 7.33,
+      "learning_rate": 8.10997442455243e-06,
+      "loss": 0.0039,
+      "step": 2903
+    },
+    {
+      "epoch": 7.33,
+      "learning_rate": 8.102301790281331e-06,
+      "loss": 1.1882,
+      "step": 2904
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 8.09462915601023e-06,
+      "loss": 0.0034,
+      "step": 2905
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 8.086956521739131e-06,
+      "loss": 0.0066,
+      "step": 2906
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 8.079283887468031e-06,
+      "loss": 0.0041,
+      "step": 2907
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 8.071611253196931e-06,
+      "loss": 0.0039,
+      "step": 2908
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 8.06393861892583e-06,
+      "loss": 0.0062,
+      "step": 2909
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 8.056265984654732e-06,
+      "loss": 0.0053,
+      "step": 2910
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 8.04859335038363e-06,
+      "loss": 0.0041,
+      "step": 2911
+    },
+    {
+      "epoch": 7.35,
+      "learning_rate": 8.040920716112532e-06,
+      "loss": 0.059,
+      "step": 2912
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 8.033248081841432e-06,
+      "loss": 0.0027,
+      "step": 2913
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 8.025575447570332e-06,
+      "loss": 0.005,
+      "step": 2914
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 8.017902813299232e-06,
+      "loss": 0.0061,
+      "step": 2915
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 8.010230179028134e-06,
+      "loss": 0.0063,
+      "step": 2916
+    },
+    {
+      "epoch": 7.37,
+      "learning_rate": 8.002557544757034e-06,
+      "loss": 0.0026,
+      "step": 2917
+    },
+    {
+      "epoch": 7.37,
+      "learning_rate": 7.994884910485933e-06,
+      "loss": 0.0037,
+      "step": 2918
+    },
+    {
+      "epoch": 7.37,
+      "learning_rate": 7.987212276214833e-06,
+      "loss": 1.2357,
+      "step": 2919
+    },
+    {
+      "epoch": 7.37,
+      "learning_rate": 7.979539641943735e-06,
+      "loss": 0.0037,
+      "step": 2920
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 7.971867007672635e-06,
+      "loss": 0.0031,
+      "step": 2921
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 7.964194373401535e-06,
+      "loss": 0.0026,
+      "step": 2922
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 7.956521739130436e-06,
+      "loss": 0.006,
+      "step": 2923
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 7.948849104859335e-06,
+      "loss": 0.0032,
+      "step": 2924
+    },
+    {
+      "epoch": 7.39,
+      "learning_rate": 7.941176470588236e-06,
+      "loss": 0.0049,
+      "step": 2925
+    },
+    {
+      "epoch": 7.39,
+      "learning_rate": 7.933503836317136e-06,
+      "loss": 0.0053,
+      "step": 2926
+    },
+    {
+      "epoch": 7.39,
+      "learning_rate": 7.925831202046036e-06,
+      "loss": 0.0037,
+      "step": 2927
+    },
+    {
+      "epoch": 7.39,
+      "learning_rate": 7.918158567774936e-06,
+      "loss": 0.0049,
+      "step": 2928
+    },
+    {
+      "epoch": 7.4,
+      "learning_rate": 7.910485933503838e-06,
+      "loss": 0.0048,
+      "step": 2929
+    },
+    {
+      "epoch": 7.4,
+      "learning_rate": 7.902813299232736e-06,
+      "loss": 0.0059,
+      "step": 2930
+    },
+    {
+      "epoch": 7.4,
+      "learning_rate": 7.895140664961637e-06,
+      "loss": 0.0037,
+      "step": 2931
+    },
+    {
+      "epoch": 7.4,
+      "learning_rate": 7.887468030690537e-06,
+      "loss": 0.0021,
+      "step": 2932
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 7.879795396419437e-06,
+      "loss": 0.0037,
+      "step": 2933
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 7.872122762148337e-06,
+      "loss": 0.0033,
+      "step": 2934
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 7.864450127877239e-06,
+      "loss": 0.0048,
+      "step": 2935
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 7.856777493606137e-06,
+      "loss": 0.0025,
+      "step": 2936
+    },
+    {
+      "epoch": 7.42,
+      "learning_rate": 7.849104859335039e-06,
+      "loss": 0.0025,
+      "step": 2937
+    },
+    {
+      "epoch": 7.42,
+      "learning_rate": 7.841432225063939e-06,
+      "loss": 0.0036,
+      "step": 2938
+    },
+    {
+      "epoch": 7.42,
+      "learning_rate": 7.833759590792839e-06,
+      "loss": 0.0025,
+      "step": 2939
+    },
+    {
+      "epoch": 7.42,
+      "learning_rate": 7.826086956521738e-06,
+      "loss": 1.0675,
+      "step": 2940
+    },
+    {
+      "epoch": 7.43,
+      "learning_rate": 7.81841432225064e-06,
+      "loss": 0.003,
+      "step": 2941
+    },
+    {
+      "epoch": 7.43,
+      "learning_rate": 7.810741687979538e-06,
+      "loss": 0.0032,
+      "step": 2942
+    },
+    {
+      "epoch": 7.43,
+      "learning_rate": 7.80306905370844e-06,
+      "loss": 0.0036,
+      "step": 2943
+    },
+    {
+      "epoch": 7.43,
+      "learning_rate": 7.79539641943734e-06,
+      "loss": 0.0046,
+      "step": 2944
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 7.787723785166241e-06,
+      "loss": 0.0036,
+      "step": 2945
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 7.780051150895141e-06,
+      "loss": 0.0036,
+      "step": 2946
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 7.772378516624041e-06,
+      "loss": 0.0025,
+      "step": 2947
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 7.764705882352943e-06,
+      "loss": 0.0046,
+      "step": 2948
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 7.757033248081841e-06,
+      "loss": 0.0025,
+      "step": 2949
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 7.749360613810743e-06,
+      "loss": 0.0031,
+      "step": 2950
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 7.741687979539643e-06,
+      "loss": 0.0024,
+      "step": 2951
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 7.734015345268543e-06,
+      "loss": 0.003,
+      "step": 2952
+    },
+    {
+      "epoch": 7.46,
+      "learning_rate": 7.726342710997442e-06,
+      "loss": 0.0024,
+      "step": 2953
+    },
+    {
+      "epoch": 7.46,
+      "learning_rate": 7.718670076726344e-06,
+      "loss": 0.0028,
+      "step": 2954
+    },
+    {
+      "epoch": 7.46,
+      "learning_rate": 7.710997442455242e-06,
+      "loss": 0.0046,
+      "step": 2955
+    },
+    {
+      "epoch": 7.46,
+      "learning_rate": 7.703324808184144e-06,
+      "loss": 0.0017,
+      "step": 2956
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 7.695652173913044e-06,
+      "loss": 1.2377,
+      "step": 2957
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 7.687979539641944e-06,
+      "loss": 0.0034,
+      "step": 2958
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 7.680306905370844e-06,
+      "loss": 2.3609,
+      "step": 2959
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 7.672634271099745e-06,
+      "loss": 0.0045,
+      "step": 2960
+    },
+    {
+      "epoch": 7.48,
+      "learning_rate": 7.664961636828643e-06,
+      "loss": 0.0036,
+      "step": 2961
+    },
+    {
+      "epoch": 7.48,
+      "learning_rate": 7.657289002557545e-06,
+      "loss": 0.0056,
+      "step": 2962
+    },
+    {
+      "epoch": 7.48,
+      "learning_rate": 7.649616368286445e-06,
+      "loss": 0.0024,
+      "step": 2963
+    },
+    {
+      "epoch": 7.48,
+      "learning_rate": 7.641943734015345e-06,
+      "loss": 0.0036,
+      "step": 2964
+    },
+    {
+      "epoch": 7.49,
+      "learning_rate": 7.634271099744245e-06,
+      "loss": 0.0034,
+      "step": 2965
+    },
+    {
+      "epoch": 7.49,
+      "learning_rate": 7.6265984654731465e-06,
+      "loss": 0.0034,
+      "step": 2966
+    },
+    {
+      "epoch": 7.49,
+      "learning_rate": 7.6189258312020455e-06,
+      "loss": 0.0034,
+      "step": 2967
+    },
+    {
+      "epoch": 7.49,
+      "learning_rate": 7.611253196930946e-06,
+      "loss": 0.0024,
+      "step": 2968
+    },
+    {
+      "epoch": 7.5,
+      "learning_rate": 7.603580562659847e-06,
+      "loss": 0.0034,
+      "step": 2969
+    },
+    {
+      "epoch": 7.5,
+      "learning_rate": 7.595907928388748e-06,
+      "loss": 0.0041,
+      "step": 2970
+    },
+    {
+      "epoch": 7.5,
+      "learning_rate": 7.588235294117647e-06,
+      "loss": 0.0057,
+      "step": 2971
+    },
+    {
+      "epoch": 7.51,
+      "learning_rate": 7.580562659846548e-06,
+      "loss": 1.2636,
+      "step": 2972
+    },
+    {
+      "epoch": 7.51,
+      "learning_rate": 7.5728900255754484e-06,
+      "loss": 0.0044,
+      "step": 2973
+    },
+    {
+      "epoch": 7.51,
+      "learning_rate": 7.5652173913043475e-06,
+      "loss": 0.0034,
+      "step": 2974
+    },
+    {
+      "epoch": 7.51,
+      "learning_rate": 7.557544757033248e-06,
+      "loss": 0.0035,
+      "step": 2975
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 7.549872122762149e-06,
+      "loss": 0.0034,
+      "step": 2976
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 7.542199488491048e-06,
+      "loss": 0.0045,
+      "step": 2977
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 7.534526854219949e-06,
+      "loss": 0.0024,
+      "step": 2978
+    },
+    {
+      "epoch": 7.52,
+      "learning_rate": 7.52685421994885e-06,
+      "loss": 0.0038,
+      "step": 2979
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 7.519181585677749e-06,
+      "loss": 0.0034,
+      "step": 2980
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 7.5115089514066495e-06,
+      "loss": 0.0029,
+      "step": 2981
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 7.50383631713555e-06,
+      "loss": 0.0034,
+      "step": 2982
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 7.496163682864451e-06,
+      "loss": 0.0034,
+      "step": 2983
+    },
+    {
+      "epoch": 7.54,
+      "learning_rate": 7.488491048593351e-06,
+      "loss": 0.0056,
+      "step": 2984
+    },
+    {
+      "epoch": 7.54,
+      "learning_rate": 7.480818414322251e-06,
+      "loss": 0.0035,
+      "step": 2985
+    },
+    {
+      "epoch": 7.54,
+      "learning_rate": 7.473145780051152e-06,
+      "loss": 0.0045,
+      "step": 2986
+    },
+    {
+      "epoch": 7.54,
+      "learning_rate": 7.4654731457800515e-06,
+      "loss": 0.0034,
+      "step": 2987
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 7.4578005115089514e-06,
+      "loss": 1.4972,
+      "step": 2988
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 7.450127877237852e-06,
+      "loss": 0.0034,
+      "step": 2989
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 7.442455242966752e-06,
+      "loss": 0.0045,
+      "step": 2990
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 7.434782608695653e-06,
+      "loss": 0.0026,
+      "step": 2991
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 7.427109974424553e-06,
+      "loss": 0.0051,
+      "step": 2992
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 7.419437340153453e-06,
+      "loss": 0.0028,
+      "step": 2993
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 7.4117647058823535e-06,
+      "loss": 0.0023,
+      "step": 2994
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 7.404092071611253e-06,
+      "loss": 0.0034,
+      "step": 2995
+    },
+    {
+      "epoch": 7.57,
+      "learning_rate": 7.396419437340153e-06,
+      "loss": 0.0023,
+      "step": 2996
+    },
+    {
+      "epoch": 7.57,
+      "learning_rate": 7.388746803069054e-06,
+      "loss": 0.0045,
+      "step": 2997
+    },
+    {
+      "epoch": 7.57,
+      "learning_rate": 7.381074168797954e-06,
+      "loss": 0.0224,
+      "step": 2998
+    },
+    {
+      "epoch": 7.57,
+      "learning_rate": 7.373401534526854e-06,
+      "loss": 1.444,
+      "step": 2999
+    },
+    {
+      "epoch": 7.58,
+      "learning_rate": 7.365728900255755e-06,
+      "loss": 0.0016,
+      "step": 3000
+    },
+    {
+      "epoch": 7.58,
+      "learning_rate": 7.358056265984655e-06,
+      "loss": 0.0034,
+      "step": 3001
+    },
+    {
+      "epoch": 7.58,
+      "learning_rate": 7.3503836317135545e-06,
+      "loss": 0.0051,
+      "step": 3002
+    },
+    {
+      "epoch": 7.58,
+      "learning_rate": 7.342710997442455e-06,
+      "loss": 0.0035,
+      "step": 3003
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 7.335038363171355e-06,
+      "loss": 0.0038,
+      "step": 3004
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 7.327365728900256e-06,
+      "loss": 0.0034,
+      "step": 3005
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 7.319693094629156e-06,
+      "loss": 0.0034,
+      "step": 3006
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 7.312020460358056e-06,
+      "loss": 0.0034,
+      "step": 3007
+    },
+    {
+      "epoch": 7.6,
+      "learning_rate": 7.304347826086957e-06,
+      "loss": 0.0037,
+      "step": 3008
+    },
+    {
+      "epoch": 7.6,
+      "learning_rate": 7.296675191815857e-06,
+      "loss": 0.0034,
+      "step": 3009
+    },
+    {
+      "epoch": 7.6,
+      "learning_rate": 7.289002557544757e-06,
+      "loss": 1.2445,
+      "step": 3010
+    },
+    {
+      "epoch": 7.6,
+      "learning_rate": 7.281329923273658e-06,
+      "loss": 0.0034,
+      "step": 3011
+    },
+    {
+      "epoch": 7.61,
+      "learning_rate": 7.273657289002558e-06,
+      "loss": 0.0034,
+      "step": 3012
+    },
+    {
+      "epoch": 7.61,
+      "learning_rate": 7.265984654731458e-06,
+      "loss": 0.0019,
+      "step": 3013
+    },
+    {
+      "epoch": 7.61,
+      "learning_rate": 7.258312020460359e-06,
+      "loss": 1.5106,
+      "step": 3014
+    },
+    {
+      "epoch": 7.61,
+      "learning_rate": 7.2506393861892585e-06,
+      "loss": 1.5062,
+      "step": 3015
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 7.242966751918159e-06,
+      "loss": 0.0035,
+      "step": 3016
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 7.235294117647059e-06,
+      "loss": 0.0027,
+      "step": 3017
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 7.227621483375959e-06,
+      "loss": 0.0025,
+      "step": 3018
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 7.21994884910486e-06,
+      "loss": 0.0035,
+      "step": 3019
+    },
+    {
+      "epoch": 7.63,
+      "learning_rate": 7.21227621483376e-06,
+      "loss": 0.0046,
+      "step": 3020
+    },
+    {
+      "epoch": 7.63,
+      "learning_rate": 7.20460358056266e-06,
+      "loss": 1.2443,
+      "step": 3021
+    },
+    {
+      "epoch": 7.63,
+      "learning_rate": 7.1969309462915605e-06,
+      "loss": 0.004,
+      "step": 3022
+    },
+    {
+      "epoch": 7.63,
+      "learning_rate": 7.18925831202046e-06,
+      "loss": 0.0028,
+      "step": 3023
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 7.18158567774936e-06,
+      "loss": 0.0059,
+      "step": 3024
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 7.173913043478261e-06,
+      "loss": 0.0062,
+      "step": 3025
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 7.166240409207161e-06,
+      "loss": 0.0017,
+      "step": 3026
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 7.158567774936061e-06,
+      "loss": 1.2106,
+      "step": 3027
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 7.150895140664962e-06,
+      "loss": 0.0049,
+      "step": 3028
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 7.143222506393862e-06,
+      "loss": 0.0022,
+      "step": 3029
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 7.135549872122762e-06,
+      "loss": 0.0038,
+      "step": 3030
+    },
+    {
+      "epoch": 7.65,
+      "learning_rate": 7.127877237851662e-06,
+      "loss": 0.004,
+      "step": 3031
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 7.120204603580563e-06,
+      "loss": 0.0018,
+      "step": 3032
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 7.112531969309464e-06,
+      "loss": 0.0038,
+      "step": 3033
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 7.104859335038364e-06,
+      "loss": 0.0011,
+      "step": 3034
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 7.097186700767264e-06,
+      "loss": 0.0033,
+      "step": 3035
+    },
+    {
+      "epoch": 7.67,
+      "learning_rate": 7.089514066496164e-06,
+      "loss": 0.0052,
+      "step": 3036
+    },
+    {
+      "epoch": 7.67,
+      "learning_rate": 7.081841432225064e-06,
+      "loss": 1.4336,
+      "step": 3037
+    },
+    {
+      "epoch": 7.67,
+      "learning_rate": 7.074168797953964e-06,
+      "loss": 0.0017,
+      "step": 3038
+    },
+    {
+      "epoch": 7.67,
+      "learning_rate": 7.066496163682865e-06,
+      "loss": 0.0039,
+      "step": 3039
+    },
+    {
+      "epoch": 7.68,
+      "learning_rate": 7.058823529411765e-06,
+      "loss": 1.1887,
+      "step": 3040
+    },
+    {
+      "epoch": 7.68,
+      "learning_rate": 7.051150895140665e-06,
+      "loss": 0.0031,
+      "step": 3041
+    },
+    {
+      "epoch": 7.68,
+      "learning_rate": 7.043478260869566e-06,
+      "loss": 0.005,
+      "step": 3042
+    },
+    {
+      "epoch": 7.68,
+      "learning_rate": 7.0358056265984655e-06,
+      "loss": 0.0036,
+      "step": 3043
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 7.028132992327366e-06,
+      "loss": 0.0042,
+      "step": 3044
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 7.020460358056266e-06,
+      "loss": 0.0027,
+      "step": 3045
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 7.012787723785166e-06,
+      "loss": 1.1529,
+      "step": 3046
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 7.005115089514067e-06,
+      "loss": 0.0043,
+      "step": 3047
+    },
+    {
+      "epoch": 7.7,
+      "learning_rate": 6.997442455242967e-06,
+      "loss": 0.0064,
+      "step": 3048
+    },
+    {
+      "epoch": 7.7,
+      "learning_rate": 6.989769820971867e-06,
+      "loss": 0.0045,
+      "step": 3049
+    },
+    {
+      "epoch": 7.7,
+      "learning_rate": 6.9820971867007675e-06,
+      "loss": 0.0029,
+      "step": 3050
+    },
+    {
+      "epoch": 7.7,
+      "learning_rate": 6.974424552429667e-06,
+      "loss": 0.0044,
+      "step": 3051
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 6.966751918158567e-06,
+      "loss": 0.0012,
+      "step": 3052
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 6.959079283887468e-06,
+      "loss": 0.0018,
+      "step": 3053
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 6.951406649616368e-06,
+      "loss": 1.1802,
+      "step": 3054
+    },
+    {
+      "epoch": 7.71,
+      "learning_rate": 6.943734015345268e-06,
+      "loss": 0.0048,
+      "step": 3055
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 6.936061381074169e-06,
+      "loss": 1.1027,
+      "step": 3056
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 6.9283887468030695e-06,
+      "loss": 0.0026,
+      "step": 3057
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 6.92071611253197e-06,
+      "loss": 1.24,
+      "step": 3058
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 6.91304347826087e-06,
+      "loss": 1.6565,
+      "step": 3059
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 6.90537084398977e-06,
+      "loss": 0.0047,
+      "step": 3060
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 6.897698209718671e-06,
+      "loss": 1.3994,
+      "step": 3061
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 6.890025575447571e-06,
+      "loss": 1.3895,
+      "step": 3062
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 6.882352941176471e-06,
+      "loss": 0.0073,
+      "step": 3063
+    },
+    {
+      "epoch": 7.74,
+      "learning_rate": 6.8746803069053714e-06,
+      "loss": 0.0075,
+      "step": 3064
+    },
+    {
+      "epoch": 7.74,
+      "learning_rate": 6.867007672634271e-06,
+      "loss": 0.0192,
+      "step": 3065
+    },
+    {
+      "epoch": 7.74,
+      "learning_rate": 6.859335038363171e-06,
+      "loss": 0.005,
+      "step": 3066
+    },
+    {
+      "epoch": 7.74,
+      "learning_rate": 6.851662404092072e-06,
+      "loss": 0.0079,
+      "step": 3067
+    },
+    {
+      "epoch": 7.75,
+      "learning_rate": 6.843989769820972e-06,
+      "loss": 0.005,
+      "step": 3068
+    },
+    {
+      "epoch": 7.75,
+      "learning_rate": 6.836317135549873e-06,
+      "loss": 0.0051,
+      "step": 3069
+    },
+    {
+      "epoch": 7.75,
+      "learning_rate": 6.828644501278773e-06,
+      "loss": 1.4104,
+      "step": 3070
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 6.8209718670076725e-06,
+      "loss": 0.0014,
+      "step": 3071
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 6.813299232736573e-06,
+      "loss": 0.0072,
+      "step": 3072
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 6.805626598465473e-06,
+      "loss": 0.006,
+      "step": 3073
+    },
+    {
+      "epoch": 7.76,
+      "learning_rate": 6.797953964194373e-06,
+      "loss": 0.0046,
+      "step": 3074
+    },
+    {
+      "epoch": 7.77,
+      "learning_rate": 6.790281329923274e-06,
+      "loss": 0.0092,
+      "step": 3075
+    },
+    {
+      "epoch": 7.77,
+      "learning_rate": 6.782608695652174e-06,
+      "loss": 0.003,
+      "step": 3076
+    },
+    {
+      "epoch": 7.77,
+      "learning_rate": 6.774936061381074e-06,
+      "loss": 0.0071,
+      "step": 3077
+    },
+    {
+      "epoch": 7.77,
+      "learning_rate": 6.7672634271099745e-06,
+      "loss": 0.0089,
+      "step": 3078
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 6.7595907928388744e-06,
+      "loss": 0.0051,
+      "step": 3079
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 6.751918158567774e-06,
+      "loss": 0.0074,
+      "step": 3080
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 6.744245524296675e-06,
+      "loss": 0.0082,
+      "step": 3081
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 6.736572890025576e-06,
+      "loss": 0.0047,
+      "step": 3082
+    },
+    {
+      "epoch": 7.79,
+      "learning_rate": 6.728900255754477e-06,
+      "loss": 0.0071,
+      "step": 3083
+    },
+    {
+      "epoch": 7.79,
+      "learning_rate": 6.7212276214833766e-06,
+      "loss": 0.0029,
+      "step": 3084
+    },
+    {
+      "epoch": 7.79,
+      "learning_rate": 6.7135549872122765e-06,
+      "loss": 0.0071,
+      "step": 3085
+    },
+    {
+      "epoch": 7.79,
+      "learning_rate": 6.705882352941177e-06,
+      "loss": 0.0062,
+      "step": 3086
+    },
+    {
+      "epoch": 7.8,
+      "learning_rate": 6.698209718670077e-06,
+      "loss": 0.0046,
+      "step": 3087
+    },
+    {
+      "epoch": 7.8,
+      "learning_rate": 6.690537084398977e-06,
+      "loss": 0.0028,
+      "step": 3088
+    },
+    {
+      "epoch": 7.8,
+      "learning_rate": 6.682864450127878e-06,
+      "loss": 0.0049,
+      "step": 3089
+    },
+    {
+      "epoch": 7.8,
+      "learning_rate": 6.675191815856778e-06,
+      "loss": 0.0078,
+      "step": 3090
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 6.667519181585678e-06,
+      "loss": 0.0388,
+      "step": 3091
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 6.6598465473145784e-06,
+      "loss": 0.007,
+      "step": 3092
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 6.652173913043478e-06,
+      "loss": 0.003,
+      "step": 3093
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 6.644501278772378e-06,
+      "loss": 0.0028,
+      "step": 3094
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 6.636828644501279e-06,
+      "loss": 1.424,
+      "step": 3095
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 6.629156010230179e-06,
+      "loss": 0.016,
+      "step": 3096
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 6.62148337595908e-06,
+      "loss": 0.0046,
+      "step": 3097
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 6.61381074168798e-06,
+      "loss": 0.0048,
+      "step": 3098
+    },
+    {
+      "epoch": 7.83,
+      "learning_rate": 6.6061381074168796e-06,
+      "loss": 0.0066,
+      "step": 3099
+    },
+    {
+      "epoch": 7.83,
+      "learning_rate": 6.59846547314578e-06,
+      "loss": 0.0047,
+      "step": 3100
+    },
+    {
+      "epoch": 7.83,
+      "learning_rate": 6.59079283887468e-06,
+      "loss": 0.0029,
+      "step": 3101
+    },
+    {
+      "epoch": 7.83,
+      "learning_rate": 6.58312020460358e-06,
+      "loss": 0.0042,
+      "step": 3102
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 6.575447570332481e-06,
+      "loss": 0.003,
+      "step": 3103
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 6.567774936061381e-06,
+      "loss": 0.0028,
+      "step": 3104
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 6.560102301790281e-06,
+      "loss": 0.0042,
+      "step": 3105
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 6.5524296675191815e-06,
+      "loss": 0.0032,
+      "step": 3106
+    },
+    {
+      "epoch": 7.85,
+      "learning_rate": 6.544757033248082e-06,
+      "loss": 0.0041,
+      "step": 3107
+    },
+    {
+      "epoch": 7.85,
+      "learning_rate": 6.537084398976983e-06,
+      "loss": 0.005,
+      "step": 3108
+    },
+    {
+      "epoch": 7.85,
+      "learning_rate": 6.529411764705883e-06,
+      "loss": 0.0035,
+      "step": 3109
+    },
+    {
+      "epoch": 7.85,
+      "learning_rate": 6.521739130434783e-06,
+      "loss": 1.2368,
+      "step": 3110
+    },
+    {
+      "epoch": 7.86,
+      "learning_rate": 6.514066496163684e-06,
+      "loss": 0.0041,
+      "step": 3111
+    },
+    {
+      "epoch": 7.86,
+      "learning_rate": 6.5063938618925836e-06,
+      "loss": 0.0054,
+      "step": 3112
+    },
+    {
+      "epoch": 7.86,
+      "learning_rate": 6.4987212276214835e-06,
+      "loss": 1.4303,
+      "step": 3113
+    },
+    {
+      "epoch": 7.86,
+      "learning_rate": 6.491048593350384e-06,
+      "loss": 0.0041,
+      "step": 3114
+    },
+    {
+      "epoch": 7.87,
+      "learning_rate": 6.483375959079284e-06,
+      "loss": 0.0042,
+      "step": 3115
+    },
+    {
+      "epoch": 7.87,
+      "learning_rate": 6.475703324808184e-06,
+      "loss": 0.0034,
+      "step": 3116
+    },
+    {
+      "epoch": 7.87,
+      "learning_rate": 6.468030690537085e-06,
+      "loss": 0.0046,
+      "step": 3117
+    },
+    {
+      "epoch": 7.87,
+      "learning_rate": 6.460358056265985e-06,
+      "loss": 0.0037,
+      "step": 3118
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 6.452685421994885e-06,
+      "loss": 0.0027,
+      "step": 3119
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 6.4450127877237854e-06,
+      "loss": 0.0045,
+      "step": 3120
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 6.437340153452685e-06,
+      "loss": 0.0056,
+      "step": 3121
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 6.429667519181586e-06,
+      "loss": 0.0166,
+      "step": 3122
+    },
+    {
+      "epoch": 7.89,
+      "learning_rate": 6.421994884910486e-06,
+      "loss": 0.0031,
+      "step": 3123
+    },
+    {
+      "epoch": 7.89,
+      "learning_rate": 6.414322250639386e-06,
+      "loss": 1.2067,
+      "step": 3124
+    },
+    {
+      "epoch": 7.89,
+      "learning_rate": 6.406649616368287e-06,
+      "loss": 0.0932,
+      "step": 3125
+    },
+    {
+      "epoch": 7.89,
+      "learning_rate": 6.398976982097187e-06,
+      "loss": 0.0028,
+      "step": 3126
+    },
+    {
+      "epoch": 7.9,
+      "learning_rate": 6.3913043478260866e-06,
+      "loss": 0.0056,
+      "step": 3127
+    },
+    {
+      "epoch": 7.9,
+      "learning_rate": 6.383631713554987e-06,
+      "loss": 0.0022,
+      "step": 3128
+    },
+    {
+      "epoch": 7.9,
+      "learning_rate": 6.375959079283887e-06,
+      "loss": 0.0042,
+      "step": 3129
+    },
+    {
+      "epoch": 7.9,
+      "learning_rate": 6.368286445012787e-06,
+      "loss": 0.0056,
+      "step": 3130
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 6.360613810741688e-06,
+      "loss": 0.0042,
+      "step": 3131
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 6.352941176470589e-06,
+      "loss": 1.4088,
+      "step": 3132
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 6.345268542199489e-06,
+      "loss": 0.0022,
+      "step": 3133
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 6.337595907928389e-06,
+      "loss": 0.0062,
+      "step": 3134
+    },
+    {
+      "epoch": 7.92,
+      "learning_rate": 6.329923273657289e-06,
+      "loss": 0.0042,
+      "step": 3135
+    },
+    {
+      "epoch": 7.92,
+      "learning_rate": 6.32225063938619e-06,
+      "loss": 0.0048,
+      "step": 3136
+    },
+    {
+      "epoch": 7.92,
+      "learning_rate": 6.31457800511509e-06,
+      "loss": 1.4579,
+      "step": 3137
+    },
+    {
+      "epoch": 7.92,
+      "learning_rate": 6.30690537084399e-06,
+      "loss": 0.0043,
+      "step": 3138
+    },
+    {
+      "epoch": 7.93,
+      "learning_rate": 6.299232736572891e-06,
+      "loss": 0.0056,
+      "step": 3139
+    },
+    {
+      "epoch": 7.93,
+      "learning_rate": 6.291560102301791e-06,
+      "loss": 0.0043,
+      "step": 3140
+    },
+    {
+      "epoch": 7.93,
+      "learning_rate": 6.2838874680306905e-06,
+      "loss": 0.0057,
+      "step": 3141
+    },
+    {
+      "epoch": 7.93,
+      "learning_rate": 6.276214833759591e-06,
+      "loss": 0.0043,
+      "step": 3142
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 6.268542199488491e-06,
+      "loss": 0.0703,
+      "step": 3143
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 6.260869565217391e-06,
+      "loss": 0.0023,
+      "step": 3144
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 6.253196930946292e-06,
+      "loss": 0.0081,
+      "step": 3145
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 6.245524296675192e-06,
+      "loss": 0.0046,
+      "step": 3146
+    },
+    {
+      "epoch": 7.95,
+      "learning_rate": 6.237851662404092e-06,
+      "loss": 0.0047,
+      "step": 3147
+    },
+    {
+      "epoch": 7.95,
+      "learning_rate": 6.2301790281329925e-06,
+      "loss": 1.3925,
+      "step": 3148
+    },
+    {
+      "epoch": 7.95,
+      "learning_rate": 6.222506393861892e-06,
+      "loss": 0.0077,
+      "step": 3149
+    },
+    {
+      "epoch": 7.95,
+      "learning_rate": 6.214833759590793e-06,
+      "loss": 0.0051,
+      "step": 3150
+    },
+    {
+      "epoch": 7.96,
+      "learning_rate": 6.207161125319693e-06,
+      "loss": 0.0052,
+      "step": 3151
+    },
+    {
+      "epoch": 7.96,
+      "learning_rate": 6.199488491048593e-06,
+      "loss": 0.0037,
+      "step": 3152
+    },
+    {
+      "epoch": 7.96,
+      "learning_rate": 6.191815856777494e-06,
+      "loss": 0.007,
+      "step": 3153
+    },
+    {
+      "epoch": 7.96,
+      "learning_rate": 6.184143222506394e-06,
+      "loss": 0.0051,
+      "step": 3154
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 6.176470588235294e-06,
+      "loss": 1.4096,
+      "step": 3155
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 6.168797953964195e-06,
+      "loss": 0.0043,
+      "step": 3156
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 6.161125319693095e-06,
+      "loss": 0.0029,
+      "step": 3157
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 6.153452685421995e-06,
+      "loss": 0.0089,
+      "step": 3158
+    },
+    {
+      "epoch": 7.98,
+      "learning_rate": 6.145780051150896e-06,
+      "loss": 0.0043,
+      "step": 3159
+    },
+    {
+      "epoch": 7.98,
+      "learning_rate": 6.138107416879796e-06,
+      "loss": 0.009,
+      "step": 3160
+    },
+    {
+      "epoch": 7.98,
+      "learning_rate": 6.1304347826086965e-06,
+      "loss": 0.0045,
+      "step": 3161
+    },
+    {
+      "epoch": 7.98,
+      "learning_rate": 6.122762148337596e-06,
+      "loss": 0.0059,
+      "step": 3162
+    },
+    {
+      "epoch": 7.99,
+      "learning_rate": 6.115089514066496e-06,
+      "loss": 0.0049,
+      "step": 3163
+    },
+    {
+      "epoch": 7.99,
+      "learning_rate": 6.107416879795397e-06,
+      "loss": 0.0059,
+      "step": 3164
+    },
+    {
+      "epoch": 7.99,
+      "learning_rate": 6.099744245524297e-06,
+      "loss": 0.0033,
+      "step": 3165
+    },
+    {
+      "epoch": 7.99,
+      "learning_rate": 6.092071611253197e-06,
+      "loss": 0.0056,
+      "step": 3166
+    },
+    {
+      "epoch": 8.0,
+      "learning_rate": 6.084398976982098e-06,
+      "loss": 0.0053,
+      "step": 3167
+    },
+    {
+      "epoch": 8.0,
+      "learning_rate": 6.076726342710998e-06,
+      "loss": 0.0046,
+      "step": 3168
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 0.9242424242424242,
+      "eval_loss": 0.3901217579841614,
+      "eval_runtime": 57.8076,
+      "eval_samples_per_second": 6.85,
+      "eval_steps_per_second": 1.713,
+      "step": 3168
+    },
+    {
+      "epoch": 8.0,
+      "learning_rate": 6.0690537084398975e-06,
+      "loss": 0.0068,
+      "step": 3169
+    },
+    {
+      "epoch": 8.01,
+      "learning_rate": 6.061381074168798e-06,
+      "loss": 0.0034,
+      "step": 3170
+    },
+    {
+      "epoch": 8.01,
+      "learning_rate": 6.053708439897698e-06,
+      "loss": 0.0057,
+      "step": 3171
+    },
+    {
+      "epoch": 8.01,
+      "learning_rate": 6.046035805626598e-06,
+      "loss": 0.0055,
+      "step": 3172
+    },
+    {
+      "epoch": 8.01,
+      "learning_rate": 6.038363171355499e-06,
+      "loss": 0.0042,
+      "step": 3173
+    },
+    {
+      "epoch": 8.02,
+      "learning_rate": 6.030690537084399e-06,
+      "loss": 0.0054,
+      "step": 3174
+    },
+    {
+      "epoch": 8.02,
+      "learning_rate": 6.0230179028132996e-06,
+      "loss": 0.0029,
+      "step": 3175
+    },
+    {
+      "epoch": 8.02,
+      "learning_rate": 6.0153452685421995e-06,
+      "loss": 0.0029,
+      "step": 3176
+    },
+    {
+      "epoch": 8.02,
+      "learning_rate": 6.007672634271099e-06,
+      "loss": 0.0041,
+      "step": 3177
+    },
+    {
+      "epoch": 8.03,
+      "learning_rate": 6e-06,
+      "loss": 0.0054,
+      "step": 3178
+    },
+    {
+      "epoch": 8.03,
+      "learning_rate": 5.9923273657289e-06,
+      "loss": 0.0034,
+      "step": 3179
+    },
+    {
+      "epoch": 8.03,
+      "learning_rate": 5.9846547314578e-06,
+      "loss": 0.0218,
+      "step": 3180
+    },
+    {
+      "epoch": 8.03,
+      "learning_rate": 5.976982097186702e-06,
+      "loss": 0.0044,
+      "step": 3181
+    },
+    {
+      "epoch": 8.04,
+      "learning_rate": 5.9693094629156015e-06,
+      "loss": 0.0064,
+      "step": 3182
+    },
+    {
+      "epoch": 8.04,
+      "learning_rate": 5.9616368286445014e-06,
+      "loss": 0.0051,
+      "step": 3183
+    },
+    {
+      "epoch": 8.04,
+      "learning_rate": 5.953964194373402e-06,
+      "loss": 0.0039,
+      "step": 3184
+    },
+    {
+      "epoch": 8.04,
+      "learning_rate": 5.946291560102302e-06,
+      "loss": 0.0028,
+      "step": 3185
+    },
+    {
+      "epoch": 8.05,
+      "learning_rate": 5.938618925831202e-06,
+      "loss": 0.0043,
+      "step": 3186
+    },
+    {
+      "epoch": 8.05,
+      "learning_rate": 5.930946291560103e-06,
+      "loss": 0.0027,
+      "step": 3187
+    },
+    {
+      "epoch": 8.05,
+      "learning_rate": 5.923273657289003e-06,
+      "loss": 0.0044,
+      "step": 3188
+    },
+    {
+      "epoch": 8.05,
+      "learning_rate": 5.9156010230179035e-06,
+      "loss": 0.0063,
+      "step": 3189
+    },
+    {
+      "epoch": 8.06,
+      "learning_rate": 5.907928388746803e-06,
+      "loss": 0.0054,
+      "step": 3190
+    },
+    {
+      "epoch": 8.06,
+      "learning_rate": 5.900255754475703e-06,
+      "loss": 0.0047,
+      "step": 3191
+    },
+    {
+      "epoch": 8.06,
+      "learning_rate": 5.892583120204604e-06,
+      "loss": 0.0048,
+      "step": 3192
+    },
+    {
+      "epoch": 8.06,
+      "learning_rate": 5.884910485933504e-06,
+      "loss": 1.4901,
+      "step": 3193
+    },
+    {
+      "epoch": 8.07,
+      "learning_rate": 5.877237851662404e-06,
+      "loss": 0.0022,
+      "step": 3194
+    },
+    {
+      "epoch": 8.07,
+      "learning_rate": 5.869565217391305e-06,
+      "loss": 1.2931,
+      "step": 3195
+    },
+    {
+      "epoch": 8.07,
+      "learning_rate": 5.861892583120205e-06,
+      "loss": 0.0054,
+      "step": 3196
+    },
+    {
+      "epoch": 8.07,
+      "learning_rate": 5.8542199488491045e-06,
+      "loss": 0.0051,
+      "step": 3197
+    },
+    {
+      "epoch": 8.08,
+      "learning_rate": 5.846547314578005e-06,
+      "loss": 0.0028,
+      "step": 3198
+    },
+    {
+      "epoch": 8.08,
+      "learning_rate": 5.838874680306905e-06,
+      "loss": 0.0042,
+      "step": 3199
+    },
+    {
+      "epoch": 8.08,
+      "learning_rate": 5.831202046035805e-06,
+      "loss": 0.0043,
+      "step": 3200
+    },
+    {
+      "epoch": 8.08,
+      "learning_rate": 5.823529411764706e-06,
+      "loss": 0.0045,
+      "step": 3201
+    },
+    {
+      "epoch": 8.09,
+      "learning_rate": 5.815856777493606e-06,
+      "loss": 0.0045,
+      "step": 3202
+    },
+    {
+      "epoch": 8.09,
+      "learning_rate": 5.8081841432225066e-06,
+      "loss": 1.5693,
+      "step": 3203
+    },
+    {
+      "epoch": 8.09,
+      "learning_rate": 5.8005115089514065e-06,
+      "loss": 0.0051,
+      "step": 3204
+    },
+    {
+      "epoch": 8.09,
+      "learning_rate": 5.792838874680306e-06,
+      "loss": 0.0051,
+      "step": 3205
+    },
+    {
+      "epoch": 8.1,
+      "learning_rate": 5.785166240409208e-06,
+      "loss": 0.004,
+      "step": 3206
+    },
+    {
+      "epoch": 8.1,
+      "learning_rate": 5.777493606138108e-06,
+      "loss": 0.0045,
+      "step": 3207
+    },
+    {
+      "epoch": 8.1,
+      "learning_rate": 5.769820971867008e-06,
+      "loss": 0.0021,
+      "step": 3208
+    },
+    {
+      "epoch": 8.1,
+      "learning_rate": 5.762148337595909e-06,
+      "loss": 0.0054,
+      "step": 3209
+    },
+    {
+      "epoch": 8.11,
+      "learning_rate": 5.7544757033248085e-06,
+      "loss": 0.0057,
+      "step": 3210
+    },
+    {
+      "epoch": 8.11,
+      "learning_rate": 5.7468030690537084e-06,
+      "loss": 0.0044,
+      "step": 3211
+    },
+    {
+      "epoch": 8.11,
+      "learning_rate": 5.739130434782609e-06,
+      "loss": 0.005,
+      "step": 3212
+    },
+    {
+      "epoch": 8.11,
+      "learning_rate": 5.731457800511509e-06,
+      "loss": 0.0027,
+      "step": 3213
+    },
+    {
+      "epoch": 8.12,
+      "learning_rate": 5.72378516624041e-06,
+      "loss": 0.0027,
+      "step": 3214
+    },
+    {
+      "epoch": 8.12,
+      "learning_rate": 5.71611253196931e-06,
+      "loss": 0.005,
+      "step": 3215
+    },
+    {
+      "epoch": 8.12,
+      "learning_rate": 5.70843989769821e-06,
+      "loss": 0.0052,
+      "step": 3216
+    },
+    {
+      "epoch": 8.12,
+      "learning_rate": 5.7007672634271105e-06,
+      "loss": 0.004,
+      "step": 3217
+    },
+    {
+      "epoch": 8.13,
+      "learning_rate": 5.69309462915601e-06,
+      "loss": 0.0064,
+      "step": 3218
+    },
+    {
+      "epoch": 8.13,
+      "learning_rate": 5.68542199488491e-06,
+      "loss": 0.0038,
+      "step": 3219
+    },
+    {
+      "epoch": 8.13,
+      "learning_rate": 5.677749360613811e-06,
+      "loss": 0.0038,
+      "step": 3220
+    },
+    {
+      "epoch": 8.13,
+      "learning_rate": 5.670076726342711e-06,
+      "loss": 0.004,
+      "step": 3221
+    },
+    {
+      "epoch": 8.14,
+      "learning_rate": 5.662404092071611e-06,
+      "loss": 0.0054,
+      "step": 3222
+    },
+    {
+      "epoch": 8.14,
+      "learning_rate": 5.654731457800512e-06,
+      "loss": 1.4265,
+      "step": 3223
+    },
+    {
+      "epoch": 8.14,
+      "learning_rate": 5.647058823529412e-06,
+      "loss": 0.5258,
+      "step": 3224
+    },
+    {
+      "epoch": 8.14,
+      "learning_rate": 5.6393861892583115e-06,
+      "loss": 0.0028,
+      "step": 3225
+    },
+    {
+      "epoch": 8.15,
+      "learning_rate": 5.631713554987212e-06,
+      "loss": 0.0032,
+      "step": 3226
+    },
+    {
+      "epoch": 8.15,
+      "learning_rate": 5.624040920716112e-06,
+      "loss": 0.0048,
+      "step": 3227
+    },
+    {
+      "epoch": 8.15,
+      "learning_rate": 5.616368286445013e-06,
+      "loss": 0.0044,
+      "step": 3228
+    },
+    {
+      "epoch": 8.15,
+      "learning_rate": 5.608695652173913e-06,
+      "loss": 0.0042,
+      "step": 3229
+    },
+    {
+      "epoch": 8.16,
+      "learning_rate": 5.601023017902814e-06,
+      "loss": 0.0031,
+      "step": 3230
+    },
+    {
+      "epoch": 8.16,
+      "learning_rate": 5.593350383631714e-06,
+      "loss": 0.0054,
+      "step": 3231
+    },
+    {
+      "epoch": 8.16,
+      "learning_rate": 5.585677749360614e-06,
+      "loss": 0.0032,
+      "step": 3232
+    },
+    {
+      "epoch": 8.16,
+      "learning_rate": 5.578005115089514e-06,
+      "loss": 0.0026,
+      "step": 3233
+    },
+    {
+      "epoch": 8.17,
+      "learning_rate": 5.570332480818415e-06,
+      "loss": 0.0037,
+      "step": 3234
+    },
+    {
+      "epoch": 8.17,
+      "learning_rate": 5.562659846547315e-06,
+      "loss": 0.0053,
+      "step": 3235
+    },
+    {
+      "epoch": 8.17,
+      "learning_rate": 5.554987212276215e-06,
+      "loss": 0.0039,
+      "step": 3236
+    },
+    {
+      "epoch": 8.17,
+      "learning_rate": 5.547314578005116e-06,
+      "loss": 0.0048,
+      "step": 3237
+    },
+    {
+      "epoch": 8.18,
+      "learning_rate": 5.5396419437340155e-06,
+      "loss": 0.0048,
+      "step": 3238
+    },
+    {
+      "epoch": 8.18,
+      "learning_rate": 5.5319693094629155e-06,
+      "loss": 0.0047,
+      "step": 3239
+    },
+    {
+      "epoch": 8.18,
+      "learning_rate": 5.524296675191816e-06,
+      "loss": 0.0038,
+      "step": 3240
+    },
+    {
+      "epoch": 8.18,
+      "learning_rate": 5.516624040920716e-06,
+      "loss": 0.0037,
+      "step": 3241
+    },
+    {
+      "epoch": 8.19,
+      "learning_rate": 5.508951406649617e-06,
+      "loss": 0.0036,
+      "step": 3242
+    },
+    {
+      "epoch": 8.19,
+      "learning_rate": 5.501278772378517e-06,
+      "loss": 0.0036,
+      "step": 3243
+    },
+    {
+      "epoch": 8.19,
+      "learning_rate": 5.493606138107417e-06,
+      "loss": 0.0025,
+      "step": 3244
+    },
+    {
+      "epoch": 8.19,
+      "learning_rate": 5.4859335038363175e-06,
+      "loss": 0.0036,
+      "step": 3245
+    },
+    {
+      "epoch": 8.2,
+      "learning_rate": 5.478260869565217e-06,
+      "loss": 0.0033,
+      "step": 3246
+    },
+    {
+      "epoch": 8.2,
+      "learning_rate": 5.470588235294117e-06,
+      "loss": 0.0053,
+      "step": 3247
+    },
+    {
+      "epoch": 8.2,
+      "learning_rate": 5.462915601023018e-06,
+      "loss": 0.0046,
+      "step": 3248
+    },
+    {
+      "epoch": 8.2,
+      "learning_rate": 5.455242966751918e-06,
+      "loss": 0.0057,
+      "step": 3249
+    },
+    {
+      "epoch": 8.21,
+      "learning_rate": 5.447570332480818e-06,
+      "loss": 0.0049,
+      "step": 3250
+    },
+    {
+      "epoch": 8.21,
+      "learning_rate": 5.439897698209719e-06,
+      "loss": 0.0051,
+      "step": 3251
+    },
+    {
+      "epoch": 8.21,
+      "learning_rate": 5.432225063938619e-06,
+      "loss": 0.0045,
+      "step": 3252
+    },
+    {
+      "epoch": 8.21,
+      "learning_rate": 5.4245524296675185e-06,
+      "loss": 0.0025,
+      "step": 3253
+    },
+    {
+      "epoch": 8.22,
+      "learning_rate": 5.416879795396419e-06,
+      "loss": 0.0033,
+      "step": 3254
+    },
+    {
+      "epoch": 8.22,
+      "learning_rate": 5.40920716112532e-06,
+      "loss": 0.0042,
+      "step": 3255
+    },
+    {
+      "epoch": 8.22,
+      "learning_rate": 5.401534526854221e-06,
+      "loss": 0.0038,
+      "step": 3256
+    },
+    {
+      "epoch": 8.22,
+      "learning_rate": 5.393861892583121e-06,
+      "loss": 0.0036,
+      "step": 3257
+    },
+    {
+      "epoch": 8.23,
+      "learning_rate": 5.386189258312021e-06,
+      "loss": 0.0035,
+      "step": 3258
+    },
+    {
+      "epoch": 8.23,
+      "learning_rate": 5.3785166240409214e-06,
+      "loss": 0.029,
+      "step": 3259
+    },
+    {
+      "epoch": 8.23,
+      "learning_rate": 5.370843989769821e-06,
+      "loss": 0.0038,
+      "step": 3260
+    },
+    {
+      "epoch": 8.23,
+      "learning_rate": 5.363171355498721e-06,
+      "loss": 0.0024,
+      "step": 3261
+    },
+    {
+      "epoch": 8.24,
+      "learning_rate": 5.355498721227622e-06,
+      "loss": 0.0058,
+      "step": 3262
+    },
+    {
+      "epoch": 8.24,
+      "learning_rate": 5.347826086956522e-06,
+      "loss": 0.0034,
+      "step": 3263
+    },
+    {
+      "epoch": 8.24,
+      "learning_rate": 5.340153452685422e-06,
+      "loss": 0.0047,
+      "step": 3264
+    },
+    {
+      "epoch": 8.24,
+      "learning_rate": 5.332480818414323e-06,
+      "loss": 1.5064,
+      "step": 3265
+    },
+    {
+      "epoch": 8.25,
+      "learning_rate": 5.3248081841432225e-06,
+      "loss": 0.0041,
+      "step": 3266
+    },
+    {
+      "epoch": 8.25,
+      "learning_rate": 5.317135549872123e-06,
+      "loss": 0.0065,
+      "step": 3267
+    },
+    {
+      "epoch": 8.25,
+      "learning_rate": 5.309462915601023e-06,
+      "loss": 0.0034,
+      "step": 3268
+    },
+    {
+      "epoch": 8.26,
+      "learning_rate": 5.301790281329923e-06,
+      "loss": 0.0038,
+      "step": 3269
+    },
+    {
+      "epoch": 8.26,
+      "learning_rate": 5.294117647058824e-06,
+      "loss": 1.1837,
+      "step": 3270
+    },
+    {
+      "epoch": 8.26,
+      "learning_rate": 5.286445012787724e-06,
+      "loss": 0.0034,
+      "step": 3271
+    },
+    {
+      "epoch": 8.26,
+      "learning_rate": 5.278772378516624e-06,
+      "loss": 0.0025,
+      "step": 3272
+    },
+    {
+      "epoch": 8.27,
+      "learning_rate": 5.2710997442455245e-06,
+      "loss": 0.0049,
+      "step": 3273
+    },
+    {
+      "epoch": 8.27,
+      "learning_rate": 5.2634271099744244e-06,
+      "loss": 0.0028,
+      "step": 3274
+    },
+    {
+      "epoch": 8.27,
+      "learning_rate": 5.255754475703324e-06,
+      "loss": 0.0046,
+      "step": 3275
+    },
+    {
+      "epoch": 8.27,
+      "learning_rate": 5.248081841432225e-06,
+      "loss": 0.0052,
+      "step": 3276
+    },
+    {
+      "epoch": 8.28,
+      "learning_rate": 5.240409207161125e-06,
+      "loss": 0.0207,
+      "step": 3277
+    },
+    {
+      "epoch": 8.28,
+      "learning_rate": 5.232736572890025e-06,
+      "loss": 0.0027,
+      "step": 3278
+    },
+    {
+      "epoch": 8.28,
+      "learning_rate": 5.225063938618926e-06,
+      "loss": 0.0022,
+      "step": 3279
+    },
+    {
+      "epoch": 8.28,
+      "learning_rate": 5.2173913043478265e-06,
+      "loss": 0.0034,
+      "step": 3280
+    },
+    {
+      "epoch": 8.29,
+      "learning_rate": 5.209718670076727e-06,
+      "loss": 0.0035,
+      "step": 3281
+    },
+    {
+      "epoch": 8.29,
+      "learning_rate": 5.202046035805627e-06,
+      "loss": 0.0027,
+      "step": 3282
+    },
+    {
+      "epoch": 8.29,
+      "learning_rate": 5.194373401534527e-06,
+      "loss": 0.0031,
+      "step": 3283
+    },
+    {
+      "epoch": 8.29,
+      "learning_rate": 5.186700767263428e-06,
+      "loss": 0.0036,
+      "step": 3284
+    },
+    {
+      "epoch": 8.3,
+      "learning_rate": 5.179028132992328e-06,
+      "loss": 0.003,
+      "step": 3285
+    },
+    {
+      "epoch": 8.3,
+      "learning_rate": 5.171355498721228e-06,
+      "loss": 0.0024,
+      "step": 3286
+    },
+    {
+      "epoch": 8.3,
+      "learning_rate": 5.1636828644501284e-06,
+      "loss": 0.0043,
+      "step": 3287
+    },
+    {
+      "epoch": 8.3,
+      "learning_rate": 5.156010230179028e-06,
+      "loss": 0.0026,
+      "step": 3288
+    },
+    {
+      "epoch": 8.31,
+      "learning_rate": 5.148337595907928e-06,
+      "loss": 0.0024,
+      "step": 3289
+    },
+    {
+      "epoch": 8.31,
+      "learning_rate": 5.140664961636829e-06,
+      "loss": 0.0025,
+      "step": 3290
+    },
+    {
+      "epoch": 8.31,
+      "learning_rate": 5.132992327365729e-06,
+      "loss": 0.0034,
+      "step": 3291
+    },
+    {
+      "epoch": 8.31,
+      "learning_rate": 5.125319693094629e-06,
+      "loss": 0.0033,
+      "step": 3292
+    },
+    {
+      "epoch": 8.32,
+      "learning_rate": 5.11764705882353e-06,
+      "loss": 1.6,
+      "step": 3293
+    },
+    {
+      "epoch": 8.32,
+      "learning_rate": 5.1099744245524296e-06,
+      "loss": 0.0191,
+      "step": 3294
+    },
+    {
+      "epoch": 8.32,
+      "learning_rate": 5.10230179028133e-06,
+      "loss": 0.0219,
+      "step": 3295
+    },
+    {
+      "epoch": 8.32,
+      "learning_rate": 5.09462915601023e-06,
+      "loss": 0.0052,
+      "step": 3296
+    },
+    {
+      "epoch": 8.33,
+      "learning_rate": 5.08695652173913e-06,
+      "loss": 0.0043,
+      "step": 3297
+    },
+    {
+      "epoch": 8.33,
+      "learning_rate": 5.079283887468031e-06,
+      "loss": 0.025,
+      "step": 3298
+    },
+    {
+      "epoch": 8.33,
+      "learning_rate": 5.071611253196931e-06,
+      "loss": 0.0037,
+      "step": 3299
+    },
+    {
+      "epoch": 8.33,
+      "learning_rate": 5.063938618925831e-06,
+      "loss": 0.0025,
+      "step": 3300
+    },
+    {
+      "epoch": 8.34,
+      "learning_rate": 5.0562659846547315e-06,
+      "loss": 0.0035,
+      "step": 3301
+    },
+    {
+      "epoch": 8.34,
+      "learning_rate": 5.0485933503836314e-06,
+      "loss": 0.0024,
+      "step": 3302
+    },
+    {
+      "epoch": 8.34,
+      "learning_rate": 5.040920716112531e-06,
+      "loss": 0.0052,
+      "step": 3303
+    },
+    {
+      "epoch": 8.34,
+      "learning_rate": 5.033248081841432e-06,
+      "loss": 1.2789,
+      "step": 3304
+    },
+    {
+      "epoch": 8.35,
+      "learning_rate": 5.025575447570333e-06,
+      "loss": 0.003,
+      "step": 3305
+    },
+    {
+      "epoch": 8.35,
+      "learning_rate": 5.017902813299233e-06,
+      "loss": 0.004,
+      "step": 3306
+    },
+    {
+      "epoch": 8.35,
+      "learning_rate": 5.0102301790281336e-06,
+      "loss": 0.0114,
+      "step": 3307
+    },
+    {
+      "epoch": 8.35,
+      "learning_rate": 5.0025575447570335e-06,
+      "loss": 0.0033,
+      "step": 3308
+    },
+    {
+      "epoch": 8.36,
+      "learning_rate": 4.994884910485934e-06,
+      "loss": 0.0034,
+      "step": 3309
+    },
+    {
+      "epoch": 8.36,
+      "learning_rate": 4.987212276214834e-06,
+      "loss": 1.4496,
+      "step": 3310
+    },
+    {
+      "epoch": 8.36,
+      "learning_rate": 4.979539641943734e-06,
+      "loss": 0.0034,
+      "step": 3311
+    },
+    {
+      "epoch": 8.36,
+      "learning_rate": 4.971867007672635e-06,
+      "loss": 0.0038,
+      "step": 3312
+    },
+    {
+      "epoch": 8.37,
+      "learning_rate": 4.964194373401535e-06,
+      "loss": 0.0035,
+      "step": 3313
+    },
+    {
+      "epoch": 8.37,
+      "learning_rate": 4.956521739130435e-06,
+      "loss": 0.0057,
+      "step": 3314
+    },
+    {
+      "epoch": 8.37,
+      "learning_rate": 4.9488491048593355e-06,
+      "loss": 0.0041,
+      "step": 3315
+    },
+    {
+      "epoch": 8.37,
+      "learning_rate": 4.941176470588235e-06,
+      "loss": 0.0034,
+      "step": 3316
+    },
+    {
+      "epoch": 8.38,
+      "learning_rate": 4.933503836317135e-06,
+      "loss": 0.004,
+      "step": 3317
+    },
+    {
+      "epoch": 8.38,
+      "learning_rate": 4.925831202046036e-06,
+      "loss": 0.0026,
+      "step": 3318
+    },
+    {
+      "epoch": 8.38,
+      "learning_rate": 4.918158567774936e-06,
+      "loss": 0.0024,
+      "step": 3319
+    },
+    {
+      "epoch": 8.38,
+      "learning_rate": 4.910485933503837e-06,
+      "loss": 0.003,
+      "step": 3320
+    },
+    {
+      "epoch": 8.39,
+      "learning_rate": 4.902813299232737e-06,
+      "loss": 0.0026,
+      "step": 3321
+    },
+    {
+      "epoch": 8.39,
+      "learning_rate": 4.8951406649616366e-06,
+      "loss": 0.0036,
+      "step": 3322
+    },
+    {
+      "epoch": 8.39,
+      "learning_rate": 4.887468030690537e-06,
+      "loss": 1.2734,
+      "step": 3323
+    },
+    {
+      "epoch": 8.39,
+      "learning_rate": 4.879795396419437e-06,
+      "loss": 0.0029,
+      "step": 3324
+    },
+    {
+      "epoch": 8.4,
+      "learning_rate": 4.872122762148337e-06,
+      "loss": 0.0028,
+      "step": 3325
+    },
+    {
+      "epoch": 8.4,
+      "learning_rate": 4.864450127877238e-06,
+      "loss": 1.2688,
+      "step": 3326
+    },
+    {
+      "epoch": 8.4,
+      "learning_rate": 4.856777493606138e-06,
+      "loss": 0.0054,
+      "step": 3327
+    },
+    {
+      "epoch": 8.4,
+      "learning_rate": 4.849104859335038e-06,
+      "loss": 0.0026,
+      "step": 3328
+    },
+    {
+      "epoch": 8.41,
+      "learning_rate": 4.8414322250639385e-06,
+      "loss": 0.0026,
+      "step": 3329
+    },
+    {
+      "epoch": 8.41,
+      "learning_rate": 4.833759590792839e-06,
+      "loss": 0.0035,
+      "step": 3330
+    },
+    {
+      "epoch": 8.41,
+      "learning_rate": 4.826086956521739e-06,
+      "loss": 0.0036,
+      "step": 3331
+    },
+    {
+      "epoch": 8.41,
+      "learning_rate": 4.81841432225064e-06,
+      "loss": 0.0051,
+      "step": 3332
+    },
+    {
+      "epoch": 8.42,
+      "learning_rate": 4.81074168797954e-06,
+      "loss": 1.2525,
+      "step": 3333
+    },
+    {
+      "epoch": 8.42,
+      "learning_rate": 4.803069053708441e-06,
+      "loss": 0.0058,
+      "step": 3334
+    },
+    {
+      "epoch": 8.42,
+      "learning_rate": 4.795396419437341e-06,
+      "loss": 0.0025,
+      "step": 3335
+    },
+    {
+      "epoch": 8.42,
+      "learning_rate": 4.7877237851662405e-06,
+      "loss": 0.0323,
+      "step": 3336
+    },
+    {
+      "epoch": 8.43,
+      "learning_rate": 4.780051150895141e-06,
+      "loss": 0.0037,
+      "step": 3337
+    },
+    {
+      "epoch": 8.43,
+      "learning_rate": 4.772378516624041e-06,
+      "loss": 0.0046,
+      "step": 3338
+    },
+    {
+      "epoch": 8.43,
+      "learning_rate": 4.764705882352941e-06,
+      "loss": 0.003,
+      "step": 3339
+    },
+    {
+      "epoch": 8.43,
+      "learning_rate": 4.757033248081842e-06,
+      "loss": 0.0023,
+      "step": 3340
+    },
+    {
+      "epoch": 8.44,
+      "learning_rate": 4.749360613810742e-06,
+      "loss": 0.003,
+      "step": 3341
+    },
+    {
+      "epoch": 8.44,
+      "learning_rate": 4.741687979539642e-06,
+      "loss": 0.0025,
+      "step": 3342
+    },
+    {
+      "epoch": 8.44,
+      "learning_rate": 4.7340153452685425e-06,
+      "loss": 0.0026,
+      "step": 3343
+    },
+    {
+      "epoch": 8.44,
+      "learning_rate": 4.726342710997442e-06,
+      "loss": 1.2155,
+      "step": 3344
+    },
+    {
+      "epoch": 8.45,
+      "learning_rate": 4.718670076726342e-06,
+      "loss": 0.0049,
+      "step": 3345
+    },
+    {
+      "epoch": 8.45,
+      "learning_rate": 4.710997442455243e-06,
+      "loss": 1.2437,
+      "step": 3346
+    },
+    {
+      "epoch": 8.45,
+      "learning_rate": 4.703324808184143e-06,
+      "loss": 0.006,
+      "step": 3347
+    },
+    {
+      "epoch": 8.45,
+      "learning_rate": 4.695652173913044e-06,
+      "loss": 0.0036,
+      "step": 3348
+    },
+    {
+      "epoch": 8.46,
+      "learning_rate": 4.687979539641944e-06,
+      "loss": 0.0032,
+      "step": 3349
+    },
+    {
+      "epoch": 8.46,
+      "learning_rate": 4.680306905370844e-06,
+      "loss": 0.0048,
+      "step": 3350
+    },
+    {
+      "epoch": 8.46,
+      "learning_rate": 4.672634271099744e-06,
+      "loss": 0.0059,
+      "step": 3351
+    },
+    {
+      "epoch": 8.46,
+      "learning_rate": 4.664961636828644e-06,
+      "loss": 0.005,
+      "step": 3352
+    },
+    {
+      "epoch": 8.47,
+      "learning_rate": 4.657289002557544e-06,
+      "loss": 0.0048,
+      "step": 3353
+    },
+    {
+      "epoch": 8.47,
+      "learning_rate": 4.649616368286446e-06,
+      "loss": 0.0037,
+      "step": 3354
+    },
+    {
+      "epoch": 8.47,
+      "learning_rate": 4.641943734015346e-06,
+      "loss": 1.2189,
+      "step": 3355
+    },
+    {
+      "epoch": 8.47,
+      "learning_rate": 4.634271099744246e-06,
+      "loss": 0.0083,
+      "step": 3356
+    },
+    {
+      "epoch": 8.48,
+      "learning_rate": 4.626598465473146e-06,
+      "loss": 0.0062,
+      "step": 3357
+    },
+    {
+      "epoch": 8.48,
+      "learning_rate": 4.618925831202046e-06,
+      "loss": 1.6984,
+      "step": 3358
+    },
+    {
+      "epoch": 8.48,
+      "learning_rate": 4.611253196930947e-06,
+      "loss": 0.0385,
+      "step": 3359
+    },
+    {
+      "epoch": 8.48,
+      "learning_rate": 4.603580562659847e-06,
+      "loss": 0.0106,
+      "step": 3360
+    },
+    {
+      "epoch": 8.49,
+      "learning_rate": 4.595907928388747e-06,
+      "loss": 0.0041,
+      "step": 3361
+    },
+    {
+      "epoch": 8.49,
+      "learning_rate": 4.588235294117648e-06,
+      "loss": 0.0048,
+      "step": 3362
+    },
+    {
+      "epoch": 8.49,
+      "learning_rate": 4.580562659846548e-06,
+      "loss": 0.003,
+      "step": 3363
+    },
+    {
+      "epoch": 8.49,
+      "learning_rate": 4.5728900255754475e-06,
+      "loss": 0.0039,
+      "step": 3364
+    },
+    {
+      "epoch": 8.5,
+      "learning_rate": 4.565217391304348e-06,
+      "loss": 0.005,
+      "step": 3365
+    },
+    {
+      "epoch": 8.5,
+      "learning_rate": 4.557544757033248e-06,
+      "loss": 0.0037,
+      "step": 3366
+    },
+    {
+      "epoch": 8.5,
+      "learning_rate": 4.549872122762148e-06,
+      "loss": 0.0048,
+      "step": 3367
+    },
+    {
+      "epoch": 8.51,
+      "learning_rate": 4.542199488491049e-06,
+      "loss": 0.0054,
+      "step": 3368
+    },
+    {
+      "epoch": 8.51,
+      "learning_rate": 4.534526854219949e-06,
+      "loss": 0.0025,
+      "step": 3369
+    },
+    {
+      "epoch": 8.51,
+      "learning_rate": 4.526854219948849e-06,
+      "loss": 0.0093,
+      "step": 3370
+    },
+    {
+      "epoch": 8.51,
+      "learning_rate": 4.5191815856777495e-06,
+      "loss": 0.0042,
+      "step": 3371
+    },
+    {
+      "epoch": 8.52,
+      "learning_rate": 4.511508951406649e-06,
+      "loss": 0.0025,
+      "step": 3372
+    },
+    {
+      "epoch": 8.52,
+      "learning_rate": 4.50383631713555e-06,
+      "loss": 0.0048,
+      "step": 3373
+    },
+    {
+      "epoch": 8.52,
+      "learning_rate": 4.49616368286445e-06,
+      "loss": 0.0024,
+      "step": 3374
+    },
+    {
+      "epoch": 8.52,
+      "learning_rate": 4.48849104859335e-06,
+      "loss": 0.0048,
+      "step": 3375
+    },
+    {
+      "epoch": 8.53,
+      "learning_rate": 4.480818414322251e-06,
+      "loss": 0.0026,
+      "step": 3376
+    },
+    {
+      "epoch": 8.53,
+      "learning_rate": 4.473145780051151e-06,
+      "loss": 1.1559,
+      "step": 3377
+    },
+    {
+      "epoch": 8.53,
+      "learning_rate": 4.465473145780051e-06,
+      "loss": 0.0048,
+      "step": 3378
+    },
+    {
+      "epoch": 8.53,
+      "learning_rate": 4.457800511508952e-06,
+      "loss": 0.0021,
+      "step": 3379
+    },
+    {
+      "epoch": 8.54,
+      "learning_rate": 4.450127877237852e-06,
+      "loss": 0.0028,
+      "step": 3380
+    },
+    {
+      "epoch": 8.54,
+      "learning_rate": 4.442455242966752e-06,
+      "loss": 0.0798,
+      "step": 3381
+    },
+    {
+      "epoch": 8.54,
+      "learning_rate": 4.434782608695653e-06,
+      "loss": 0.0053,
+      "step": 3382
+    },
+    {
+      "epoch": 8.54,
+      "learning_rate": 4.427109974424553e-06,
+      "loss": 0.0024,
+      "step": 3383
+    },
+    {
+      "epoch": 8.55,
+      "learning_rate": 4.419437340153453e-06,
+      "loss": 0.0049,
+      "step": 3384
+    },
+    {
+      "epoch": 8.55,
+      "learning_rate": 4.411764705882353e-06,
+      "loss": 0.0036,
+      "step": 3385
+    },
+    {
+      "epoch": 8.55,
+      "learning_rate": 4.404092071611253e-06,
+      "loss": 1.5139,
+      "step": 3386
+    },
+    {
+      "epoch": 8.55,
+      "learning_rate": 4.396419437340154e-06,
+      "loss": 0.0035,
+      "step": 3387
+    },
+    {
+      "epoch": 8.56,
+      "learning_rate": 4.388746803069054e-06,
+      "loss": 0.0024,
+      "step": 3388
+    },
+    {
+      "epoch": 8.56,
+      "learning_rate": 4.381074168797954e-06,
+      "loss": 0.0051,
+      "step": 3389
+    },
+    {
+      "epoch": 8.56,
+      "learning_rate": 4.373401534526855e-06,
+      "loss": 0.0059,
+      "step": 3390
+    },
+    {
+      "epoch": 8.56,
+      "learning_rate": 4.365728900255755e-06,
+      "loss": 0.0059,
+      "step": 3391
+    },
+    {
+      "epoch": 8.57,
+      "learning_rate": 4.3580562659846545e-06,
+      "loss": 0.0025,
+      "step": 3392
+    },
+    {
+      "epoch": 8.57,
+      "learning_rate": 4.350383631713555e-06,
+      "loss": 1.2465,
+      "step": 3393
+    },
+    {
+      "epoch": 8.57,
+      "learning_rate": 4.342710997442455e-06,
+      "loss": 0.0038,
+      "step": 3394
+    },
+    {
+      "epoch": 8.57,
+      "learning_rate": 4.335038363171355e-06,
+      "loss": 0.0024,
+      "step": 3395
+    },
+    {
+      "epoch": 8.58,
+      "learning_rate": 4.327365728900256e-06,
+      "loss": 0.0052,
+      "step": 3396
+    },
+    {
+      "epoch": 8.58,
+      "learning_rate": 4.319693094629156e-06,
+      "loss": 0.0024,
+      "step": 3397
+    },
+    {
+      "epoch": 8.58,
+      "learning_rate": 4.312020460358056e-06,
+      "loss": 0.0051,
+      "step": 3398
+    },
+    {
+      "epoch": 8.58,
+      "learning_rate": 4.3043478260869565e-06,
+      "loss": 0.0035,
+      "step": 3399
+    },
+    {
+      "epoch": 8.59,
+      "learning_rate": 4.296675191815856e-06,
+      "loss": 0.0166,
+      "step": 3400
+    },
+    {
+      "epoch": 8.59,
+      "learning_rate": 4.289002557544757e-06,
+      "loss": 0.0045,
+      "step": 3401
+    },
+    {
+      "epoch": 8.59,
+      "learning_rate": 4.281329923273657e-06,
+      "loss": 0.0043,
+      "step": 3402
+    },
+    {
+      "epoch": 8.59,
+      "learning_rate": 4.273657289002557e-06,
+      "loss": 0.003,
+      "step": 3403
+    },
+    {
+      "epoch": 8.6,
+      "learning_rate": 4.265984654731459e-06,
+      "loss": 0.0071,
+      "step": 3404
+    },
+    {
+      "epoch": 8.6,
+      "learning_rate": 4.2583120204603585e-06,
+      "loss": 0.0036,
+      "step": 3405
+    },
+    {
+      "epoch": 8.6,
+      "learning_rate": 4.2506393861892584e-06,
+      "loss": 0.0049,
+      "step": 3406
+    },
+    {
+      "epoch": 8.6,
+      "learning_rate": 4.242966751918159e-06,
+      "loss": 0.0039,
+      "step": 3407
+    },
+    {
+      "epoch": 8.61,
+      "learning_rate": 4.235294117647059e-06,
+      "loss": 0.0036,
+      "step": 3408
+    },
+    {
+      "epoch": 8.61,
+      "learning_rate": 4.227621483375959e-06,
+      "loss": 0.0026,
+      "step": 3409
+    },
+    {
+      "epoch": 8.61,
+      "learning_rate": 4.21994884910486e-06,
+      "loss": 0.0028,
+      "step": 3410
+    },
+    {
+      "epoch": 8.61,
+      "learning_rate": 4.21227621483376e-06,
+      "loss": 0.0024,
+      "step": 3411
+    },
+    {
+      "epoch": 8.62,
+      "learning_rate": 4.2046035805626605e-06,
+      "loss": 0.0044,
+      "step": 3412
+    },
+    {
+      "epoch": 8.62,
+      "learning_rate": 4.19693094629156e-06,
+      "loss": 0.0025,
+      "step": 3413
+    },
+    {
+      "epoch": 8.62,
+      "learning_rate": 4.18925831202046e-06,
+      "loss": 0.0029,
+      "step": 3414
+    },
+    {
+      "epoch": 8.62,
+      "learning_rate": 4.181585677749361e-06,
+      "loss": 0.0019,
+      "step": 3415
+    },
+    {
+      "epoch": 8.63,
+      "learning_rate": 4.173913043478261e-06,
+      "loss": 0.0025,
+      "step": 3416
+    },
+    {
+      "epoch": 8.63,
+      "learning_rate": 4.166240409207161e-06,
+      "loss": 0.0043,
+      "step": 3417
+    },
+    {
+      "epoch": 8.63,
+      "learning_rate": 4.158567774936062e-06,
+      "loss": 0.0037,
+      "step": 3418
+    },
+    {
+      "epoch": 8.63,
+      "learning_rate": 4.150895140664962e-06,
+      "loss": 0.0024,
+      "step": 3419
+    },
+    {
+      "epoch": 8.64,
+      "learning_rate": 4.1432225063938615e-06,
+      "loss": 0.0024,
+      "step": 3420
+    },
+    {
+      "epoch": 8.64,
+      "learning_rate": 4.135549872122762e-06,
+      "loss": 0.0023,
+      "step": 3421
+    },
+    {
+      "epoch": 8.64,
+      "learning_rate": 4.127877237851662e-06,
+      "loss": 0.0024,
+      "step": 3422
+    },
+    {
+      "epoch": 8.64,
+      "learning_rate": 4.120204603580562e-06,
+      "loss": 0.0039,
+      "step": 3423
+    },
+    {
+      "epoch": 8.65,
+      "learning_rate": 4.112531969309463e-06,
+      "loss": 0.0047,
+      "step": 3424
+    },
+    {
+      "epoch": 8.65,
+      "learning_rate": 4.104859335038363e-06,
+      "loss": 0.0225,
+      "step": 3425
+    },
+    {
+      "epoch": 8.65,
+      "learning_rate": 4.0971867007672636e-06,
+      "loss": 0.0298,
+      "step": 3426
+    },
+    {
+      "epoch": 8.65,
+      "learning_rate": 4.0895140664961635e-06,
+      "loss": 1.1412,
+      "step": 3427
+    },
+    {
+      "epoch": 8.66,
+      "learning_rate": 4.081841432225064e-06,
+      "loss": 0.0038,
+      "step": 3428
+    },
+    {
+      "epoch": 8.66,
+      "learning_rate": 4.074168797953965e-06,
+      "loss": 0.0027,
+      "step": 3429
+    },
+    {
+      "epoch": 8.66,
+      "learning_rate": 4.066496163682865e-06,
+      "loss": 0.0042,
+      "step": 3430
+    },
+    {
+      "epoch": 8.66,
+      "learning_rate": 4.058823529411765e-06,
+      "loss": 0.0035,
+      "step": 3431
+    },
+    {
+      "epoch": 8.67,
+      "learning_rate": 4.051150895140666e-06,
+      "loss": 0.0038,
+      "step": 3432
+    },
+    {
+      "epoch": 8.67,
+      "learning_rate": 4.0434782608695655e-06,
+      "loss": 0.0034,
+      "step": 3433
+    },
+    {
+      "epoch": 8.67,
+      "learning_rate": 4.0358056265984655e-06,
+      "loss": 0.0025,
+      "step": 3434
+    },
+    {
+      "epoch": 8.67,
+      "learning_rate": 4.028132992327366e-06,
+      "loss": 0.0046,
+      "step": 3435
+    },
+    {
+      "epoch": 8.68,
+      "learning_rate": 4.020460358056266e-06,
+      "loss": 0.0023,
+      "step": 3436
+    },
+    {
+      "epoch": 8.68,
+      "learning_rate": 4.012787723785166e-06,
+      "loss": 0.004,
+      "step": 3437
+    },
+    {
+      "epoch": 8.68,
+      "learning_rate": 4.005115089514067e-06,
+      "loss": 0.0035,
+      "step": 3438
+    },
+    {
+      "epoch": 8.68,
+      "learning_rate": 3.997442455242967e-06,
+      "loss": 0.0057,
+      "step": 3439
+    },
+    {
+      "epoch": 8.69,
+      "learning_rate": 3.9897698209718675e-06,
+      "loss": 1.4514,
+      "step": 3440
+    },
+    {
+      "epoch": 8.69,
+      "learning_rate": 3.982097186700767e-06,
+      "loss": 0.006,
+      "step": 3441
+    },
+    {
+      "epoch": 8.69,
+      "learning_rate": 3.974424552429667e-06,
+      "loss": 0.0025,
+      "step": 3442
+    },
+    {
+      "epoch": 8.69,
+      "learning_rate": 3.966751918158568e-06,
+      "loss": 0.0043,
+      "step": 3443
+    },
+    {
+      "epoch": 8.7,
+      "learning_rate": 3.959079283887468e-06,
+      "loss": 0.0035,
+      "step": 3444
+    },
+    {
+      "epoch": 8.7,
+      "learning_rate": 3.951406649616368e-06,
+      "loss": 0.0036,
+      "step": 3445
+    },
+    {
+      "epoch": 8.7,
+      "learning_rate": 3.943734015345269e-06,
+      "loss": 0.0034,
+      "step": 3446
+    },
+    {
+      "epoch": 8.7,
+      "learning_rate": 3.936061381074169e-06,
+      "loss": 0.0045,
+      "step": 3447
+    },
+    {
+      "epoch": 8.71,
+      "learning_rate": 3.9283887468030685e-06,
+      "loss": 0.0056,
+      "step": 3448
+    },
+    {
+      "epoch": 8.71,
+      "learning_rate": 3.920716112531969e-06,
+      "loss": 0.003,
+      "step": 3449
+    },
+    {
+      "epoch": 8.71,
+      "learning_rate": 3.913043478260869e-06,
+      "loss": 1.316,
+      "step": 3450
+    },
+    {
+      "epoch": 8.71,
+      "learning_rate": 3.905370843989769e-06,
+      "loss": 0.0041,
+      "step": 3451
+    },
+    {
+      "epoch": 8.72,
+      "learning_rate": 3.89769820971867e-06,
+      "loss": 0.0046,
+      "step": 3452
+    },
+    {
+      "epoch": 8.72,
+      "learning_rate": 3.890025575447571e-06,
+      "loss": 0.0037,
+      "step": 3453
+    },
+    {
+      "epoch": 8.72,
+      "learning_rate": 3.8823529411764714e-06,
+      "loss": 0.0023,
+      "step": 3454
+    },
+    {
+      "epoch": 8.72,
+      "learning_rate": 3.874680306905371e-06,
+      "loss": 0.0037,
+      "step": 3455
+    },
+    {
+      "epoch": 8.73,
+      "learning_rate": 3.867007672634271e-06,
+      "loss": 0.0038,
+      "step": 3456
+    },
+    {
+      "epoch": 8.73,
+      "learning_rate": 3.859335038363172e-06,
+      "loss": 0.0045,
+      "step": 3457
+    },
+    {
+      "epoch": 8.73,
+      "learning_rate": 3.851662404092072e-06,
+      "loss": 0.0054,
+      "step": 3458
+    },
+    {
+      "epoch": 8.73,
+      "learning_rate": 3.843989769820972e-06,
+      "loss": 0.0023,
+      "step": 3459
+    },
+    {
+      "epoch": 8.74,
+      "learning_rate": 3.836317135549873e-06,
+      "loss": 0.0058,
+      "step": 3460
+    },
+    {
+      "epoch": 8.74,
+      "learning_rate": 3.8286445012787726e-06,
+      "loss": 0.0033,
+      "step": 3461
+    },
+    {
+      "epoch": 8.74,
+      "learning_rate": 3.8209718670076725e-06,
+      "loss": 0.0045,
+      "step": 3462
+    },
+    {
+      "epoch": 8.74,
+      "learning_rate": 3.8132992327365732e-06,
+      "loss": 0.0037,
+      "step": 3463
+    },
+    {
+      "epoch": 8.75,
+      "learning_rate": 3.805626598465473e-06,
+      "loss": 0.0034,
+      "step": 3464
+    },
+    {
+      "epoch": 8.75,
+      "learning_rate": 3.797953964194374e-06,
+      "loss": 0.0025,
+      "step": 3465
+    },
+    {
+      "epoch": 8.75,
+      "learning_rate": 3.790281329923274e-06,
+      "loss": 0.005,
+      "step": 3466
+    },
+    {
+      "epoch": 8.76,
+      "learning_rate": 3.7826086956521738e-06,
+      "loss": 0.0036,
+      "step": 3467
+    },
+    {
+      "epoch": 8.76,
+      "learning_rate": 3.7749360613810745e-06,
+      "loss": 0.0023,
+      "step": 3468
+    },
+    {
+      "epoch": 8.76,
+      "learning_rate": 3.7672634271099744e-06,
+      "loss": 0.0022,
+      "step": 3469
+    },
+    {
+      "epoch": 8.76,
+      "learning_rate": 3.7595907928388744e-06,
+      "loss": 0.0038,
+      "step": 3470
+    },
+    {
+      "epoch": 8.77,
+      "learning_rate": 3.751918158567775e-06,
+      "loss": 1.5236,
+      "step": 3471
+    },
+    {
+      "epoch": 8.77,
+      "learning_rate": 3.7442455242966755e-06,
+      "loss": 0.0033,
+      "step": 3472
+    },
+    {
+      "epoch": 8.77,
+      "learning_rate": 3.736572890025576e-06,
+      "loss": 0.0066,
+      "step": 3473
+    },
+    {
+      "epoch": 8.77,
+      "learning_rate": 3.7289002557544757e-06,
+      "loss": 0.0059,
+      "step": 3474
+    },
+    {
+      "epoch": 8.78,
+      "learning_rate": 3.721227621483376e-06,
+      "loss": 0.0034,
+      "step": 3475
+    },
+    {
+      "epoch": 8.78,
+      "learning_rate": 3.7135549872122764e-06,
+      "loss": 0.0033,
+      "step": 3476
+    },
+    {
+      "epoch": 8.78,
+      "learning_rate": 3.7058823529411767e-06,
+      "loss": 0.0034,
+      "step": 3477
+    },
+    {
+      "epoch": 8.78,
+      "learning_rate": 3.6982097186700767e-06,
+      "loss": 0.0053,
+      "step": 3478
+    },
+    {
+      "epoch": 8.79,
+      "learning_rate": 3.690537084398977e-06,
+      "loss": 0.0066,
+      "step": 3479
+    },
+    {
+      "epoch": 8.79,
+      "learning_rate": 3.6828644501278773e-06,
+      "loss": 0.0047,
+      "step": 3480
+    },
+    {
+      "epoch": 8.79,
+      "learning_rate": 3.6751918158567773e-06,
+      "loss": 0.0022,
+      "step": 3481
+    },
+    {
+      "epoch": 8.79,
+      "learning_rate": 3.6675191815856776e-06,
+      "loss": 0.0034,
+      "step": 3482
+    },
+    {
+      "epoch": 8.8,
+      "learning_rate": 3.659846547314578e-06,
+      "loss": 0.0034,
+      "step": 3483
+    },
+    {
+      "epoch": 8.8,
+      "learning_rate": 3.6521739130434787e-06,
+      "loss": 0.0079,
+      "step": 3484
+    },
+    {
+      "epoch": 8.8,
+      "learning_rate": 3.6445012787723786e-06,
+      "loss": 0.0045,
+      "step": 3485
+    },
+    {
+      "epoch": 8.8,
+      "learning_rate": 3.636828644501279e-06,
+      "loss": 0.0033,
+      "step": 3486
+    },
+    {
+      "epoch": 8.81,
+      "learning_rate": 3.6291560102301793e-06,
+      "loss": 0.0029,
+      "step": 3487
+    },
+    {
+      "epoch": 8.81,
+      "learning_rate": 3.6214833759590796e-06,
+      "loss": 0.0047,
+      "step": 3488
+    },
+    {
+      "epoch": 8.81,
+      "learning_rate": 3.6138107416879796e-06,
+      "loss": 0.0027,
+      "step": 3489
+    },
+    {
+      "epoch": 8.81,
+      "learning_rate": 3.60613810741688e-06,
+      "loss": 0.0043,
+      "step": 3490
+    },
+    {
+      "epoch": 8.82,
+      "learning_rate": 3.5984654731457802e-06,
+      "loss": 0.0033,
+      "step": 3491
+    },
+    {
+      "epoch": 8.82,
+      "learning_rate": 3.59079283887468e-06,
+      "loss": 0.0033,
+      "step": 3492
+    },
+    {
+      "epoch": 8.82,
+      "learning_rate": 3.5831202046035805e-06,
+      "loss": 0.0044,
+      "step": 3493
+    },
+    {
+      "epoch": 8.82,
+      "learning_rate": 3.575447570332481e-06,
+      "loss": 0.0025,
+      "step": 3494
+    },
+    {
+      "epoch": 8.83,
+      "learning_rate": 3.567774936061381e-06,
+      "loss": 0.0023,
+      "step": 3495
+    },
+    {
+      "epoch": 8.83,
+      "learning_rate": 3.5601023017902815e-06,
+      "loss": 1.2509,
+      "step": 3496
+    },
+    {
+      "epoch": 8.83,
+      "learning_rate": 3.552429667519182e-06,
+      "loss": 0.003,
+      "step": 3497
+    },
+    {
+      "epoch": 8.83,
+      "learning_rate": 3.544757033248082e-06,
+      "loss": 0.0024,
+      "step": 3498
+    },
+    {
+      "epoch": 8.84,
+      "learning_rate": 3.537084398976982e-06,
+      "loss": 1.388,
+      "step": 3499
+    },
+    {
+      "epoch": 8.84,
+      "learning_rate": 3.5294117647058825e-06,
+      "loss": 0.0043,
+      "step": 3500
+    },
+    {
+      "epoch": 8.84,
+      "learning_rate": 3.521739130434783e-06,
+      "loss": 0.0043,
+      "step": 3501
+    },
+    {
+      "epoch": 8.84,
+      "learning_rate": 3.514066496163683e-06,
+      "loss": 0.0044,
+      "step": 3502
+    },
+    {
+      "epoch": 8.85,
+      "learning_rate": 3.506393861892583e-06,
+      "loss": 0.0043,
+      "step": 3503
+    },
+    {
+      "epoch": 8.85,
+      "learning_rate": 3.4987212276214834e-06,
+      "loss": 1.3346,
+      "step": 3504
+    },
+    {
+      "epoch": 8.85,
+      "learning_rate": 3.4910485933503837e-06,
+      "loss": 0.0024,
+      "step": 3505
+    },
+    {
+      "epoch": 8.85,
+      "learning_rate": 3.4833759590792837e-06,
+      "loss": 0.0031,
+      "step": 3506
+    },
+    {
+      "epoch": 8.86,
+      "learning_rate": 3.475703324808184e-06,
+      "loss": 0.0034,
+      "step": 3507
+    },
+    {
+      "epoch": 8.86,
+      "learning_rate": 3.4680306905370843e-06,
+      "loss": 0.0033,
+      "step": 3508
+    },
+    {
+      "epoch": 8.86,
+      "learning_rate": 3.460358056265985e-06,
+      "loss": 0.0043,
+      "step": 3509
+    },
+    {
+      "epoch": 8.86,
+      "learning_rate": 3.452685421994885e-06,
+      "loss": 0.0035,
+      "step": 3510
+    },
+    {
+      "epoch": 8.87,
+      "learning_rate": 3.4450127877237854e-06,
+      "loss": 0.0045,
+      "step": 3511
+    },
+    {
+      "epoch": 8.87,
+      "learning_rate": 3.4373401534526857e-06,
+      "loss": 0.0033,
+      "step": 3512
+    },
+    {
+      "epoch": 8.87,
+      "learning_rate": 3.4296675191815856e-06,
+      "loss": 0.0033,
+      "step": 3513
+    },
+    {
+      "epoch": 8.87,
+      "learning_rate": 3.421994884910486e-06,
+      "loss": 1.4988,
+      "step": 3514
+    },
+    {
+      "epoch": 8.88,
+      "learning_rate": 3.4143222506393863e-06,
+      "loss": 0.0048,
+      "step": 3515
+    },
+    {
+      "epoch": 8.88,
+      "learning_rate": 3.4066496163682867e-06,
+      "loss": 0.0017,
+      "step": 3516
+    },
+    {
+      "epoch": 8.88,
+      "learning_rate": 3.3989769820971866e-06,
+      "loss": 0.0043,
+      "step": 3517
+    },
+    {
+      "epoch": 8.88,
+      "learning_rate": 3.391304347826087e-06,
+      "loss": 1.4504,
+      "step": 3518
+    },
+    {
+      "epoch": 8.89,
+      "learning_rate": 3.3836317135549873e-06,
+      "loss": 0.0033,
+      "step": 3519
+    },
+    {
+      "epoch": 8.89,
+      "learning_rate": 3.375959079283887e-06,
+      "loss": 0.0043,
+      "step": 3520
+    },
+    {
+      "epoch": 8.89,
+      "learning_rate": 3.368286445012788e-06,
+      "loss": 0.0042,
+      "step": 3521
+    },
+    {
+      "epoch": 8.89,
+      "learning_rate": 3.3606138107416883e-06,
+      "loss": 0.0023,
+      "step": 3522
+    },
+    {
+      "epoch": 8.9,
+      "learning_rate": 3.3529411764705886e-06,
+      "loss": 0.0084,
+      "step": 3523
+    },
+    {
+      "epoch": 8.9,
+      "learning_rate": 3.3452685421994885e-06,
+      "loss": 0.0049,
+      "step": 3524
+    },
+    {
+      "epoch": 8.9,
+      "learning_rate": 3.337595907928389e-06,
+      "loss": 0.0048,
+      "step": 3525
+    },
+    {
+      "epoch": 8.9,
+      "learning_rate": 3.3299232736572892e-06,
+      "loss": 0.0043,
+      "step": 3526
+    },
+    {
+      "epoch": 8.91,
+      "learning_rate": 3.322250639386189e-06,
+      "loss": 0.0037,
+      "step": 3527
+    },
+    {
+      "epoch": 8.91,
+      "learning_rate": 3.3145780051150895e-06,
+      "loss": 0.005,
+      "step": 3528
+    },
+    {
+      "epoch": 8.91,
+      "learning_rate": 3.30690537084399e-06,
+      "loss": 0.0024,
+      "step": 3529
+    },
+    {
+      "epoch": 8.91,
+      "learning_rate": 3.29923273657289e-06,
+      "loss": 0.0017,
+      "step": 3530
+    },
+    {
+      "epoch": 8.92,
+      "learning_rate": 3.29156010230179e-06,
+      "loss": 0.0037,
+      "step": 3531
+    },
+    {
+      "epoch": 8.92,
+      "learning_rate": 3.2838874680306904e-06,
+      "loss": 1.2684,
+      "step": 3532
+    },
+    {
+      "epoch": 8.92,
+      "learning_rate": 3.2762148337595908e-06,
+      "loss": 0.0043,
+      "step": 3533
+    },
+    {
+      "epoch": 8.92,
+      "learning_rate": 3.2685421994884915e-06,
+      "loss": 0.0058,
+      "step": 3534
+    },
+    {
+      "epoch": 8.93,
+      "learning_rate": 3.2608695652173914e-06,
+      "loss": 1.2339,
+      "step": 3535
+    },
+    {
+      "epoch": 8.93,
+      "learning_rate": 3.2531969309462918e-06,
+      "loss": 0.0033,
+      "step": 3536
+    },
+    {
+      "epoch": 8.93,
+      "learning_rate": 3.245524296675192e-06,
+      "loss": 0.0043,
+      "step": 3537
+    },
+    {
+      "epoch": 8.93,
+      "learning_rate": 3.237851662404092e-06,
+      "loss": 0.003,
+      "step": 3538
+    },
+    {
+      "epoch": 8.94,
+      "learning_rate": 3.2301790281329924e-06,
+      "loss": 0.0049,
+      "step": 3539
+    },
+    {
+      "epoch": 8.94,
+      "learning_rate": 3.2225063938618927e-06,
+      "loss": 0.0039,
+      "step": 3540
+    },
+    {
+      "epoch": 8.94,
+      "learning_rate": 3.214833759590793e-06,
+      "loss": 0.0034,
+      "step": 3541
+    },
+    {
+      "epoch": 8.94,
+      "learning_rate": 3.207161125319693e-06,
+      "loss": 0.0058,
+      "step": 3542
+    },
+    {
+      "epoch": 8.95,
+      "learning_rate": 3.1994884910485933e-06,
+      "loss": 0.0034,
+      "step": 3543
+    },
+    {
+      "epoch": 8.95,
+      "learning_rate": 3.1918158567774937e-06,
+      "loss": 0.0039,
+      "step": 3544
+    },
+    {
+      "epoch": 8.95,
+      "learning_rate": 3.1841432225063936e-06,
+      "loss": 0.003,
+      "step": 3545
+    },
+    {
+      "epoch": 8.95,
+      "learning_rate": 3.1764705882352943e-06,
+      "loss": 0.0033,
+      "step": 3546
+    },
+    {
+      "epoch": 8.96,
+      "learning_rate": 3.1687979539641947e-06,
+      "loss": 0.0041,
+      "step": 3547
+    },
+    {
+      "epoch": 8.96,
+      "learning_rate": 3.161125319693095e-06,
+      "loss": 0.0033,
+      "step": 3548
+    },
+    {
+      "epoch": 8.96,
+      "learning_rate": 3.153452685421995e-06,
+      "loss": 0.0022,
+      "step": 3549
+    },
+    {
+      "epoch": 8.96,
+      "learning_rate": 3.1457800511508953e-06,
+      "loss": 0.0029,
+      "step": 3550
+    },
+    {
+      "epoch": 8.97,
+      "learning_rate": 3.1381074168797956e-06,
+      "loss": 0.0033,
+      "step": 3551
+    },
+    {
+      "epoch": 8.97,
+      "learning_rate": 3.1304347826086955e-06,
+      "loss": 0.0033,
+      "step": 3552
+    },
+    {
+      "epoch": 8.97,
+      "learning_rate": 3.122762148337596e-06,
+      "loss": 0.0047,
+      "step": 3553
+    },
+    {
+      "epoch": 8.97,
+      "learning_rate": 3.1150895140664962e-06,
+      "loss": 0.0022,
+      "step": 3554
+    },
+    {
+      "epoch": 8.98,
+      "learning_rate": 3.1074168797953966e-06,
+      "loss": 0.0033,
+      "step": 3555
+    },
+    {
+      "epoch": 8.98,
+      "learning_rate": 3.0997442455242965e-06,
+      "loss": 0.004,
+      "step": 3556
+    },
+    {
+      "epoch": 8.98,
+      "learning_rate": 3.092071611253197e-06,
+      "loss": 0.0022,
+      "step": 3557
+    },
+    {
+      "epoch": 8.98,
+      "learning_rate": 3.0843989769820976e-06,
+      "loss": 0.0024,
+      "step": 3558
+    },
+    {
+      "epoch": 8.99,
+      "learning_rate": 3.0767263427109975e-06,
+      "loss": 0.0036,
+      "step": 3559
+    },
+    {
+      "epoch": 8.99,
+      "learning_rate": 3.069053708439898e-06,
+      "loss": 0.0057,
+      "step": 3560
+    },
+    {
+      "epoch": 8.99,
+      "learning_rate": 3.061381074168798e-06,
+      "loss": 0.0033,
+      "step": 3561
+    },
+    {
+      "epoch": 8.99,
+      "learning_rate": 3.0537084398976985e-06,
+      "loss": 0.0023,
+      "step": 3562
+    },
+    {
+      "epoch": 9.0,
+      "learning_rate": 3.0460358056265985e-06,
+      "loss": 0.0035,
+      "step": 3563
+    },
+    {
+      "epoch": 9.0,
+      "learning_rate": 3.038363171355499e-06,
+      "loss": 0.0038,
+      "step": 3564
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 0.9242424242424242,
+      "eval_loss": 0.4058963358402252,
+      "eval_runtime": 57.7406,
+      "eval_samples_per_second": 6.858,
+      "eval_steps_per_second": 1.715,
+      "step": 3564
+    },
+    {
+      "epoch": 9.0,
+      "learning_rate": 3.030690537084399e-06,
+      "loss": 0.0089,
+      "step": 3565
+    },
+    {
+      "epoch": 9.01,
+      "learning_rate": 3.023017902813299e-06,
+      "loss": 0.0032,
+      "step": 3566
+    },
+    {
+      "epoch": 9.01,
+      "learning_rate": 3.0153452685421994e-06,
+      "loss": 0.0032,
+      "step": 3567
+    },
+    {
+      "epoch": 9.01,
+      "learning_rate": 3.0076726342710997e-06,
+      "loss": 0.0023,
+      "step": 3568
+    },
+    {
+      "epoch": 9.01,
+      "learning_rate": 3e-06,
+      "loss": 1.4444,
+      "step": 3569
+    },
+    {
+      "epoch": 9.02,
+      "learning_rate": 2.9923273657289e-06,
+      "loss": 0.0044,
+      "step": 3570
+    },
+    {
+      "epoch": 9.02,
+      "learning_rate": 2.9846547314578008e-06,
+      "loss": 0.0046,
+      "step": 3571
+    },
+    {
+      "epoch": 9.02,
+      "learning_rate": 2.976982097186701e-06,
+      "loss": 0.0034,
+      "step": 3572
+    },
+    {
+      "epoch": 9.02,
+      "learning_rate": 2.969309462915601e-06,
+      "loss": 0.0025,
+      "step": 3573
+    },
+    {
+      "epoch": 9.03,
+      "learning_rate": 2.9616368286445014e-06,
+      "loss": 0.0032,
+      "step": 3574
+    },
+    {
+      "epoch": 9.03,
+      "learning_rate": 2.9539641943734017e-06,
+      "loss": 0.0053,
+      "step": 3575
+    },
+    {
+      "epoch": 9.03,
+      "learning_rate": 2.946291560102302e-06,
+      "loss": 0.0032,
+      "step": 3576
+    },
+    {
+      "epoch": 9.03,
+      "learning_rate": 2.938618925831202e-06,
+      "loss": 0.0023,
+      "step": 3577
+    },
+    {
+      "epoch": 9.04,
+      "learning_rate": 2.9309462915601023e-06,
+      "loss": 0.0035,
+      "step": 3578
+    },
+    {
+      "epoch": 9.04,
+      "learning_rate": 2.9232736572890026e-06,
+      "loss": 0.0025,
+      "step": 3579
+    },
+    {
+      "epoch": 9.04,
+      "learning_rate": 2.9156010230179026e-06,
+      "loss": 0.0022,
+      "step": 3580
+    },
+    {
+      "epoch": 9.04,
+      "learning_rate": 2.907928388746803e-06,
+      "loss": 0.0033,
+      "step": 3581
+    },
+    {
+      "epoch": 9.05,
+      "learning_rate": 2.9002557544757032e-06,
+      "loss": 1.4673,
+      "step": 3582
+    },
+    {
+      "epoch": 9.05,
+      "learning_rate": 2.892583120204604e-06,
+      "loss": 0.004,
+      "step": 3583
+    },
+    {
+      "epoch": 9.05,
+      "learning_rate": 2.884910485933504e-06,
+      "loss": 0.0046,
+      "step": 3584
+    },
+    {
+      "epoch": 9.05,
+      "learning_rate": 2.8772378516624043e-06,
+      "loss": 0.0042,
+      "step": 3585
+    },
+    {
+      "epoch": 9.06,
+      "learning_rate": 2.8695652173913046e-06,
+      "loss": 0.0032,
+      "step": 3586
+    },
+    {
+      "epoch": 9.06,
+      "learning_rate": 2.861892583120205e-06,
+      "loss": 0.0035,
+      "step": 3587
+    },
+    {
+      "epoch": 9.06,
+      "learning_rate": 2.854219948849105e-06,
+      "loss": 0.0032,
+      "step": 3588
+    },
+    {
+      "epoch": 9.06,
+      "learning_rate": 2.846547314578005e-06,
+      "loss": 0.0042,
+      "step": 3589
+    },
+    {
+      "epoch": 9.07,
+      "learning_rate": 2.8388746803069055e-06,
+      "loss": 0.0042,
+      "step": 3590
+    },
+    {
+      "epoch": 9.07,
+      "learning_rate": 2.8312020460358055e-06,
+      "loss": 0.0048,
+      "step": 3591
+    },
+    {
+      "epoch": 9.07,
+      "learning_rate": 2.823529411764706e-06,
+      "loss": 0.0042,
+      "step": 3592
+    },
+    {
+      "epoch": 9.07,
+      "learning_rate": 2.815856777493606e-06,
+      "loss": 0.0032,
+      "step": 3593
+    },
+    {
+      "epoch": 9.08,
+      "learning_rate": 2.8081841432225065e-06,
+      "loss": 0.0047,
+      "step": 3594
+    },
+    {
+      "epoch": 9.08,
+      "learning_rate": 2.800511508951407e-06,
+      "loss": 0.0033,
+      "step": 3595
+    },
+    {
+      "epoch": 9.08,
+      "learning_rate": 2.792838874680307e-06,
+      "loss": 2.3099,
+      "step": 3596
+    },
+    {
+      "epoch": 9.08,
+      "learning_rate": 2.7851662404092075e-06,
+      "loss": 0.0035,
+      "step": 3597
+    },
+    {
+      "epoch": 9.09,
+      "learning_rate": 2.7774936061381074e-06,
+      "loss": 0.0044,
+      "step": 3598
+    },
+    {
+      "epoch": 9.09,
+      "learning_rate": 2.7698209718670078e-06,
+      "loss": 0.0032,
+      "step": 3599
+    },
+    {
+      "epoch": 9.09,
+      "learning_rate": 2.762148337595908e-06,
+      "loss": 0.0065,
+      "step": 3600
+    },
+    {
+      "epoch": 9.09,
+      "learning_rate": 2.7544757033248085e-06,
+      "loss": 0.0042,
+      "step": 3601
+    },
+    {
+      "epoch": 9.1,
+      "learning_rate": 2.7468030690537084e-06,
+      "loss": 0.0022,
+      "step": 3602
+    },
+    {
+      "epoch": 9.1,
+      "learning_rate": 2.7391304347826087e-06,
+      "loss": 0.0042,
+      "step": 3603
+    },
+    {
+      "epoch": 9.1,
+      "learning_rate": 2.731457800511509e-06,
+      "loss": 0.0033,
+      "step": 3604
+    },
+    {
+      "epoch": 9.1,
+      "learning_rate": 2.723785166240409e-06,
+      "loss": 0.004,
+      "step": 3605
+    },
+    {
+      "epoch": 9.11,
+      "learning_rate": 2.7161125319693093e-06,
+      "loss": 0.0037,
+      "step": 3606
+    },
+    {
+      "epoch": 9.11,
+      "learning_rate": 2.7084398976982097e-06,
+      "loss": 0.0042,
+      "step": 3607
+    },
+    {
+      "epoch": 9.11,
+      "learning_rate": 2.7007672634271104e-06,
+      "loss": 0.0034,
+      "step": 3608
+    },
+    {
+      "epoch": 9.11,
+      "learning_rate": 2.6930946291560103e-06,
+      "loss": 0.0041,
+      "step": 3609
+    },
+    {
+      "epoch": 9.12,
+      "learning_rate": 2.6854219948849107e-06,
+      "loss": 0.0029,
+      "step": 3610
+    },
+    {
+      "epoch": 9.12,
+      "learning_rate": 2.677749360613811e-06,
+      "loss": 0.0049,
+      "step": 3611
+    },
+    {
+      "epoch": 9.12,
+      "learning_rate": 2.670076726342711e-06,
+      "loss": 0.0056,
+      "step": 3612
+    },
+    {
+      "epoch": 9.12,
+      "learning_rate": 2.6624040920716113e-06,
+      "loss": 0.0047,
+      "step": 3613
+    },
+    {
+      "epoch": 9.13,
+      "learning_rate": 2.6547314578005116e-06,
+      "loss": 0.0503,
+      "step": 3614
+    },
+    {
+      "epoch": 9.13,
+      "learning_rate": 2.647058823529412e-06,
+      "loss": 0.0031,
+      "step": 3615
+    },
+    {
+      "epoch": 9.13,
+      "learning_rate": 2.639386189258312e-06,
+      "loss": 0.0041,
+      "step": 3616
+    },
+    {
+      "epoch": 9.13,
+      "learning_rate": 2.6317135549872122e-06,
+      "loss": 0.0044,
+      "step": 3617
+    },
+    {
+      "epoch": 9.14,
+      "learning_rate": 2.6240409207161126e-06,
+      "loss": 0.0021,
+      "step": 3618
+    },
+    {
+      "epoch": 9.14,
+      "learning_rate": 2.6163682864450125e-06,
+      "loss": 0.002,
+      "step": 3619
+    },
+    {
+      "epoch": 9.14,
+      "learning_rate": 2.6086956521739132e-06,
+      "loss": 0.0031,
+      "step": 3620
+    },
+    {
+      "epoch": 9.14,
+      "learning_rate": 2.6010230179028136e-06,
+      "loss": 0.0047,
+      "step": 3621
+    },
+    {
+      "epoch": 9.15,
+      "learning_rate": 2.593350383631714e-06,
+      "loss": 0.0031,
+      "step": 3622
+    },
+    {
+      "epoch": 9.15,
+      "learning_rate": 2.585677749360614e-06,
+      "loss": 0.0031,
+      "step": 3623
+    },
+    {
+      "epoch": 9.15,
+      "learning_rate": 2.578005115089514e-06,
+      "loss": 0.0041,
+      "step": 3624
+    },
+    {
+      "epoch": 9.15,
+      "learning_rate": 2.5703324808184145e-06,
+      "loss": 0.005,
+      "step": 3625
+    },
+    {
+      "epoch": 9.16,
+      "learning_rate": 2.5626598465473144e-06,
+      "loss": 0.003,
+      "step": 3626
+    },
+    {
+      "epoch": 9.16,
+      "learning_rate": 2.5549872122762148e-06,
+      "loss": 0.0033,
+      "step": 3627
+    },
+    {
+      "epoch": 9.16,
+      "learning_rate": 2.547314578005115e-06,
+      "loss": 0.0041,
+      "step": 3628
+    },
+    {
+      "epoch": 9.16,
+      "learning_rate": 2.5396419437340155e-06,
+      "loss": 0.0035,
+      "step": 3629
+    },
+    {
+      "epoch": 9.17,
+      "learning_rate": 2.5319693094629154e-06,
+      "loss": 0.0037,
+      "step": 3630
+    },
+    {
+      "epoch": 9.17,
+      "learning_rate": 2.5242966751918157e-06,
+      "loss": 0.0038,
+      "step": 3631
+    },
+    {
+      "epoch": 9.17,
+      "learning_rate": 2.516624040920716e-06,
+      "loss": 0.0024,
+      "step": 3632
+    },
+    {
+      "epoch": 9.17,
+      "learning_rate": 2.5089514066496164e-06,
+      "loss": 0.0023,
+      "step": 3633
+    },
+    {
+      "epoch": 9.18,
+      "learning_rate": 2.5012787723785167e-06,
+      "loss": 0.0042,
+      "step": 3634
+    },
+    {
+      "epoch": 9.18,
+      "learning_rate": 2.493606138107417e-06,
+      "loss": 0.0024,
+      "step": 3635
+    },
+    {
+      "epoch": 9.18,
+      "learning_rate": 2.4859335038363174e-06,
+      "loss": 0.0032,
+      "step": 3636
+    },
+    {
+      "epoch": 9.18,
+      "learning_rate": 2.4782608695652173e-06,
+      "loss": 0.0044,
+      "step": 3637
+    },
+    {
+      "epoch": 9.19,
+      "learning_rate": 2.4705882352941177e-06,
+      "loss": 0.0035,
+      "step": 3638
+    },
+    {
+      "epoch": 9.19,
+      "learning_rate": 2.462915601023018e-06,
+      "loss": 0.0059,
+      "step": 3639
+    },
+    {
+      "epoch": 9.19,
+      "learning_rate": 2.4552429667519184e-06,
+      "loss": 0.004,
+      "step": 3640
+    },
+    {
+      "epoch": 9.19,
+      "learning_rate": 2.4475703324808183e-06,
+      "loss": 0.0028,
+      "step": 3641
+    },
+    {
+      "epoch": 9.2,
+      "learning_rate": 2.4398976982097186e-06,
+      "loss": 0.0033,
+      "step": 3642
+    },
+    {
+      "epoch": 9.2,
+      "learning_rate": 2.432225063938619e-06,
+      "loss": 0.0024,
+      "step": 3643
+    },
+    {
+      "epoch": 9.2,
+      "learning_rate": 2.424552429667519e-06,
+      "loss": 0.0041,
+      "step": 3644
+    },
+    {
+      "epoch": 9.2,
+      "learning_rate": 2.4168797953964196e-06,
+      "loss": 0.003,
+      "step": 3645
+    },
+    {
+      "epoch": 9.21,
+      "learning_rate": 2.40920716112532e-06,
+      "loss": 1.4877,
+      "step": 3646
+    },
+    {
+      "epoch": 9.21,
+      "learning_rate": 2.4015345268542203e-06,
+      "loss": 0.0031,
+      "step": 3647
+    },
+    {
+      "epoch": 9.21,
+      "learning_rate": 2.3938618925831202e-06,
+      "loss": 0.0033,
+      "step": 3648
+    },
+    {
+      "epoch": 9.21,
+      "learning_rate": 2.3861892583120206e-06,
+      "loss": 0.0021,
+      "step": 3649
+    },
+    {
+      "epoch": 9.22,
+      "learning_rate": 2.378516624040921e-06,
+      "loss": 0.0041,
+      "step": 3650
+    },
+    {
+      "epoch": 9.22,
+      "learning_rate": 2.370843989769821e-06,
+      "loss": 0.0033,
+      "step": 3651
+    },
+    {
+      "epoch": 9.22,
+      "learning_rate": 2.363171355498721e-06,
+      "loss": 0.004,
+      "step": 3652
+    },
+    {
+      "epoch": 9.22,
+      "learning_rate": 2.3554987212276215e-06,
+      "loss": 0.0037,
+      "step": 3653
+    },
+    {
+      "epoch": 9.23,
+      "learning_rate": 2.347826086956522e-06,
+      "loss": 0.0052,
+      "step": 3654
+    },
+    {
+      "epoch": 9.23,
+      "learning_rate": 2.340153452685422e-06,
+      "loss": 0.003,
+      "step": 3655
+    },
+    {
+      "epoch": 9.23,
+      "learning_rate": 2.332480818414322e-06,
+      "loss": 0.0036,
+      "step": 3656
+    },
+    {
+      "epoch": 9.23,
+      "learning_rate": 2.324808184143223e-06,
+      "loss": 0.0022,
+      "step": 3657
+    },
+    {
+      "epoch": 9.24,
+      "learning_rate": 2.317135549872123e-06,
+      "loss": 0.0023,
+      "step": 3658
+    },
+    {
+      "epoch": 9.24,
+      "learning_rate": 2.309462915601023e-06,
+      "loss": 0.0033,
+      "step": 3659
+    },
+    {
+      "epoch": 9.24,
+      "learning_rate": 2.3017902813299235e-06,
+      "loss": 0.0047,
+      "step": 3660
+    },
+    {
+      "epoch": 9.24,
+      "learning_rate": 2.294117647058824e-06,
+      "loss": 0.003,
+      "step": 3661
+    },
+    {
+      "epoch": 9.25,
+      "learning_rate": 2.2864450127877238e-06,
+      "loss": 0.0039,
+      "step": 3662
+    },
+    {
+      "epoch": 9.25,
+      "learning_rate": 2.278772378516624e-06,
+      "loss": 1.4507,
+      "step": 3663
+    },
+    {
+      "epoch": 9.25,
+      "learning_rate": 2.2710997442455244e-06,
+      "loss": 0.002,
+      "step": 3664
+    },
+    {
+      "epoch": 9.26,
+      "learning_rate": 2.2634271099744244e-06,
+      "loss": 0.0045,
+      "step": 3665
+    },
+    {
+      "epoch": 9.26,
+      "learning_rate": 2.2557544757033247e-06,
+      "loss": 0.0021,
+      "step": 3666
+    },
+    {
+      "epoch": 9.26,
+      "learning_rate": 2.248081841432225e-06,
+      "loss": 0.0021,
+      "step": 3667
+    },
+    {
+      "epoch": 9.26,
+      "learning_rate": 2.2404092071611254e-06,
+      "loss": 0.0039,
+      "step": 3668
+    },
+    {
+      "epoch": 9.27,
+      "learning_rate": 2.2327365728900253e-06,
+      "loss": 0.003,
+      "step": 3669
+    },
+    {
+      "epoch": 9.27,
+      "learning_rate": 2.225063938618926e-06,
+      "loss": 0.0047,
+      "step": 3670
+    },
+    {
+      "epoch": 9.27,
+      "learning_rate": 2.2173913043478264e-06,
+      "loss": 0.0033,
+      "step": 3671
+    },
+    {
+      "epoch": 9.27,
+      "learning_rate": 2.2097186700767263e-06,
+      "loss": 0.0034,
+      "step": 3672
+    },
+    {
+      "epoch": 9.28,
+      "learning_rate": 2.2020460358056267e-06,
+      "loss": 0.0067,
+      "step": 3673
+    },
+    {
+      "epoch": 9.28,
+      "learning_rate": 2.194373401534527e-06,
+      "loss": 0.0048,
+      "step": 3674
+    },
+    {
+      "epoch": 9.28,
+      "learning_rate": 2.1867007672634273e-06,
+      "loss": 0.0039,
+      "step": 3675
+    },
+    {
+      "epoch": 9.28,
+      "learning_rate": 2.1790281329923273e-06,
+      "loss": 0.003,
+      "step": 3676
+    },
+    {
+      "epoch": 9.29,
+      "learning_rate": 2.1713554987212276e-06,
+      "loss": 0.0039,
+      "step": 3677
+    },
+    {
+      "epoch": 9.29,
+      "learning_rate": 2.163682864450128e-06,
+      "loss": 0.0049,
+      "step": 3678
+    },
+    {
+      "epoch": 9.29,
+      "learning_rate": 2.156010230179028e-06,
+      "loss": 0.0021,
+      "step": 3679
+    },
+    {
+      "epoch": 9.29,
+      "learning_rate": 2.148337595907928e-06,
+      "loss": 1.5285,
+      "step": 3680
+    },
+    {
+      "epoch": 9.3,
+      "learning_rate": 2.1406649616368285e-06,
+      "loss": 0.0046,
+      "step": 3681
+    },
+    {
+      "epoch": 9.3,
+      "learning_rate": 2.1329923273657293e-06,
+      "loss": 0.0044,
+      "step": 3682
+    },
+    {
+      "epoch": 9.3,
+      "learning_rate": 2.1253196930946292e-06,
+      "loss": 0.0031,
+      "step": 3683
+    },
+    {
+      "epoch": 9.3,
+      "learning_rate": 2.1176470588235296e-06,
+      "loss": 0.0031,
+      "step": 3684
+    },
+    {
+      "epoch": 9.31,
+      "learning_rate": 2.10997442455243e-06,
+      "loss": 0.0033,
+      "step": 3685
+    },
+    {
+      "epoch": 9.31,
+      "learning_rate": 2.1023017902813302e-06,
+      "loss": 0.003,
+      "step": 3686
+    },
+    {
+      "epoch": 9.31,
+      "learning_rate": 2.09462915601023e-06,
+      "loss": 0.0048,
+      "step": 3687
+    },
+    {
+      "epoch": 9.31,
+      "learning_rate": 2.0869565217391305e-06,
+      "loss": 0.0027,
+      "step": 3688
+    },
+    {
+      "epoch": 9.32,
+      "learning_rate": 2.079283887468031e-06,
+      "loss": 0.0032,
+      "step": 3689
+    },
+    {
+      "epoch": 9.32,
+      "learning_rate": 2.0716112531969308e-06,
+      "loss": 0.0043,
+      "step": 3690
+    },
+    {
+      "epoch": 9.32,
+      "learning_rate": 2.063938618925831e-06,
+      "loss": 0.0022,
+      "step": 3691
+    },
+    {
+      "epoch": 9.32,
+      "learning_rate": 2.0562659846547314e-06,
+      "loss": 0.0029,
+      "step": 3692
+    },
+    {
+      "epoch": 9.33,
+      "learning_rate": 2.0485933503836318e-06,
+      "loss": 0.003,
+      "step": 3693
+    },
+    {
+      "epoch": 9.33,
+      "learning_rate": 2.040920716112532e-06,
+      "loss": 0.0039,
+      "step": 3694
+    },
+    {
+      "epoch": 9.33,
+      "learning_rate": 2.0332480818414325e-06,
+      "loss": 0.0039,
+      "step": 3695
+    },
+    {
+      "epoch": 9.33,
+      "learning_rate": 2.025575447570333e-06,
+      "loss": 0.0042,
+      "step": 3696
+    },
+    {
+      "epoch": 9.34,
+      "learning_rate": 2.0179028132992327e-06,
+      "loss": 0.003,
+      "step": 3697
+    },
+    {
+      "epoch": 9.34,
+      "learning_rate": 2.010230179028133e-06,
+      "loss": 0.0038,
+      "step": 3698
+    },
+    {
+      "epoch": 9.34,
+      "learning_rate": 2.0025575447570334e-06,
+      "loss": 0.0032,
+      "step": 3699
+    },
+    {
+      "epoch": 9.34,
+      "learning_rate": 1.9948849104859338e-06,
+      "loss": 0.0021,
+      "step": 3700
+    },
+    {
+      "epoch": 9.35,
+      "learning_rate": 1.9872122762148337e-06,
+      "loss": 0.0029,
+      "step": 3701
+    },
+    {
+      "epoch": 9.35,
+      "learning_rate": 1.979539641943734e-06,
+      "loss": 0.003,
+      "step": 3702
+    },
+    {
+      "epoch": 9.35,
+      "learning_rate": 1.9718670076726344e-06,
+      "loss": 1.2736,
+      "step": 3703
+    },
+    {
+      "epoch": 9.35,
+      "learning_rate": 1.9641943734015343e-06,
+      "loss": 0.0034,
+      "step": 3704
+    },
+    {
+      "epoch": 9.36,
+      "learning_rate": 1.9565217391304346e-06,
+      "loss": 1.2829,
+      "step": 3705
+    },
+    {
+      "epoch": 9.36,
+      "learning_rate": 1.948849104859335e-06,
+      "loss": 0.0033,
+      "step": 3706
+    },
+    {
+      "epoch": 9.36,
+      "learning_rate": 1.9411764705882357e-06,
+      "loss": 0.0041,
+      "step": 3707
+    },
+    {
+      "epoch": 9.36,
+      "learning_rate": 1.9335038363171356e-06,
+      "loss": 0.0039,
+      "step": 3708
+    },
+    {
+      "epoch": 9.37,
+      "learning_rate": 1.925831202046036e-06,
+      "loss": 0.0021,
+      "step": 3709
+    },
+    {
+      "epoch": 9.37,
+      "learning_rate": 1.9181585677749363e-06,
+      "loss": 0.0023,
+      "step": 3710
+    },
+    {
+      "epoch": 9.37,
+      "learning_rate": 1.9104859335038362e-06,
+      "loss": 0.0032,
+      "step": 3711
+    },
+    {
+      "epoch": 9.37,
+      "learning_rate": 1.9028132992327366e-06,
+      "loss": 0.0051,
+      "step": 3712
+    },
+    {
+      "epoch": 9.38,
+      "learning_rate": 1.895140664961637e-06,
+      "loss": 0.0039,
+      "step": 3713
+    },
+    {
+      "epoch": 9.38,
+      "learning_rate": 1.8874680306905373e-06,
+      "loss": 0.0263,
+      "step": 3714
+    },
+    {
+      "epoch": 9.38,
+      "learning_rate": 1.8797953964194372e-06,
+      "loss": 0.0021,
+      "step": 3715
+    },
+    {
+      "epoch": 9.38,
+      "learning_rate": 1.8721227621483377e-06,
+      "loss": 0.0032,
+      "step": 3716
+    },
+    {
+      "epoch": 9.39,
+      "learning_rate": 1.8644501278772379e-06,
+      "loss": 0.0021,
+      "step": 3717
+    },
+    {
+      "epoch": 9.39,
+      "learning_rate": 1.8567774936061382e-06,
+      "loss": 0.0039,
+      "step": 3718
+    },
+    {
+      "epoch": 9.39,
+      "learning_rate": 1.8491048593350383e-06,
+      "loss": 0.0029,
+      "step": 3719
+    },
+    {
+      "epoch": 9.39,
+      "learning_rate": 1.8414322250639387e-06,
+      "loss": 0.003,
+      "step": 3720
+    },
+    {
+      "epoch": 9.4,
+      "learning_rate": 1.8337595907928388e-06,
+      "loss": 0.0033,
+      "step": 3721
+    },
+    {
+      "epoch": 9.4,
+      "learning_rate": 1.8260869565217394e-06,
+      "loss": 0.0022,
+      "step": 3722
+    },
+    {
+      "epoch": 9.4,
+      "learning_rate": 1.8184143222506395e-06,
+      "loss": 0.0025,
+      "step": 3723
+    },
+    {
+      "epoch": 9.4,
+      "learning_rate": 1.8107416879795398e-06,
+      "loss": 0.003,
+      "step": 3724
+    },
+    {
+      "epoch": 9.41,
+      "learning_rate": 1.80306905370844e-06,
+      "loss": 0.002,
+      "step": 3725
+    },
+    {
+      "epoch": 9.41,
+      "learning_rate": 1.79539641943734e-06,
+      "loss": 0.0021,
+      "step": 3726
+    },
+    {
+      "epoch": 9.41,
+      "learning_rate": 1.7877237851662404e-06,
+      "loss": 0.0033,
+      "step": 3727
+    },
+    {
+      "epoch": 9.41,
+      "learning_rate": 1.7800511508951408e-06,
+      "loss": 0.004,
+      "step": 3728
+    },
+    {
+      "epoch": 9.42,
+      "learning_rate": 1.772378516624041e-06,
+      "loss": 0.0039,
+      "step": 3729
+    },
+    {
+      "epoch": 9.42,
+      "learning_rate": 1.7647058823529412e-06,
+      "loss": 0.0029,
+      "step": 3730
+    },
+    {
+      "epoch": 9.42,
+      "learning_rate": 1.7570332480818416e-06,
+      "loss": 0.0035,
+      "step": 3731
+    },
+    {
+      "epoch": 9.42,
+      "learning_rate": 1.7493606138107417e-06,
+      "loss": 0.0041,
+      "step": 3732
+    },
+    {
+      "epoch": 9.43,
+      "learning_rate": 1.7416879795396418e-06,
+      "loss": 0.0054,
+      "step": 3733
+    },
+    {
+      "epoch": 9.43,
+      "learning_rate": 1.7340153452685422e-06,
+      "loss": 0.0038,
+      "step": 3734
+    },
+    {
+      "epoch": 9.43,
+      "learning_rate": 1.7263427109974425e-06,
+      "loss": 0.0038,
+      "step": 3735
+    },
+    {
+      "epoch": 9.43,
+      "learning_rate": 1.7186700767263429e-06,
+      "loss": 0.0041,
+      "step": 3736
+    },
+    {
+      "epoch": 9.44,
+      "learning_rate": 1.710997442455243e-06,
+      "loss": 0.0033,
+      "step": 3737
+    },
+    {
+      "epoch": 9.44,
+      "learning_rate": 1.7033248081841433e-06,
+      "loss": 0.0029,
+      "step": 3738
+    },
+    {
+      "epoch": 9.44,
+      "learning_rate": 1.6956521739130435e-06,
+      "loss": 0.0022,
+      "step": 3739
+    },
+    {
+      "epoch": 9.44,
+      "learning_rate": 1.6879795396419436e-06,
+      "loss": 0.0038,
+      "step": 3740
+    },
+    {
+      "epoch": 9.45,
+      "learning_rate": 1.6803069053708441e-06,
+      "loss": 0.0028,
+      "step": 3741
+    },
+    {
+      "epoch": 9.45,
+      "learning_rate": 1.6726342710997443e-06,
+      "loss": 0.0042,
+      "step": 3742
+    },
+    {
+      "epoch": 9.45,
+      "learning_rate": 1.6649616368286446e-06,
+      "loss": 1.2593,
+      "step": 3743
+    },
+    {
+      "epoch": 9.45,
+      "learning_rate": 1.6572890025575447e-06,
+      "loss": 0.003,
+      "step": 3744
+    },
+    {
+      "epoch": 9.46,
+      "learning_rate": 1.649616368286445e-06,
+      "loss": 0.004,
+      "step": 3745
+    },
+    {
+      "epoch": 9.46,
+      "learning_rate": 1.6419437340153452e-06,
+      "loss": 0.2219,
+      "step": 3746
+    },
+    {
+      "epoch": 9.46,
+      "learning_rate": 1.6342710997442458e-06,
+      "loss": 0.0061,
+      "step": 3747
+    },
+    {
+      "epoch": 9.46,
+      "learning_rate": 1.6265984654731459e-06,
+      "loss": 0.0031,
+      "step": 3748
+    },
+    {
+      "epoch": 9.47,
+      "learning_rate": 1.618925831202046e-06,
+      "loss": 0.0039,
+      "step": 3749
+    },
+    {
+      "epoch": 9.47,
+      "learning_rate": 1.6112531969309464e-06,
+      "loss": 0.0035,
+      "step": 3750
+    },
+    {
+      "epoch": 9.47,
+      "learning_rate": 1.6035805626598465e-06,
+      "loss": 0.0023,
+      "step": 3751
+    },
+    {
+      "epoch": 9.47,
+      "learning_rate": 1.5959079283887468e-06,
+      "loss": 0.0028,
+      "step": 3752
+    },
+    {
+      "epoch": 9.48,
+      "learning_rate": 1.5882352941176472e-06,
+      "loss": 0.0039,
+      "step": 3753
+    },
+    {
+      "epoch": 9.48,
+      "learning_rate": 1.5805626598465475e-06,
+      "loss": 0.002,
+      "step": 3754
+    },
+    {
+      "epoch": 9.48,
+      "learning_rate": 1.5728900255754476e-06,
+      "loss": 0.0018,
+      "step": 3755
+    },
+    {
+      "epoch": 9.48,
+      "learning_rate": 1.5652173913043478e-06,
+      "loss": 1.2381,
+      "step": 3756
+    },
+    {
+      "epoch": 9.49,
+      "learning_rate": 1.5575447570332481e-06,
+      "loss": 0.0029,
+      "step": 3757
+    },
+    {
+      "epoch": 9.49,
+      "learning_rate": 1.5498721227621482e-06,
+      "loss": 0.0038,
+      "step": 3758
+    },
+    {
+      "epoch": 9.49,
+      "learning_rate": 1.5421994884910488e-06,
+      "loss": 0.0029,
+      "step": 3759
+    },
+    {
+      "epoch": 9.49,
+      "learning_rate": 1.534526854219949e-06,
+      "loss": 0.0027,
+      "step": 3760
+    },
+    {
+      "epoch": 9.5,
+      "learning_rate": 1.5268542199488493e-06,
+      "loss": 0.0029,
+      "step": 3761
+    },
+    {
+      "epoch": 9.5,
+      "learning_rate": 1.5191815856777494e-06,
+      "loss": 0.003,
+      "step": 3762
+    },
+    {
+      "epoch": 9.5,
+      "learning_rate": 1.5115089514066495e-06,
+      "loss": 0.0051,
+      "step": 3763
+    },
+    {
+      "epoch": 9.51,
+      "learning_rate": 1.5038363171355499e-06,
+      "loss": 1.1782,
+      "step": 3764
+    },
+    {
+      "epoch": 9.51,
+      "learning_rate": 1.49616368286445e-06,
+      "loss": 0.0029,
+      "step": 3765
+    },
+    {
+      "epoch": 9.51,
+      "learning_rate": 1.4884910485933505e-06,
+      "loss": 0.0029,
+      "step": 3766
+    },
+    {
+      "epoch": 9.51,
+      "learning_rate": 1.4808184143222507e-06,
+      "loss": 0.0032,
+      "step": 3767
+    },
+    {
+      "epoch": 9.52,
+      "learning_rate": 1.473145780051151e-06,
+      "loss": 0.0042,
+      "step": 3768
+    },
+    {
+      "epoch": 9.52,
+      "learning_rate": 1.4654731457800511e-06,
+      "loss": 0.0037,
+      "step": 3769
+    },
+    {
+      "epoch": 9.52,
+      "learning_rate": 1.4578005115089513e-06,
+      "loss": 0.0029,
+      "step": 3770
+    },
+    {
+      "epoch": 9.52,
+      "learning_rate": 1.4501278772378516e-06,
+      "loss": 0.0029,
+      "step": 3771
+    },
+    {
+      "epoch": 9.53,
+      "learning_rate": 1.442455242966752e-06,
+      "loss": 0.005,
+      "step": 3772
+    },
+    {
+      "epoch": 9.53,
+      "learning_rate": 1.4347826086956523e-06,
+      "loss": 0.0034,
+      "step": 3773
+    },
+    {
+      "epoch": 9.53,
+      "learning_rate": 1.4271099744245524e-06,
+      "loss": 0.0153,
+      "step": 3774
+    },
+    {
+      "epoch": 9.53,
+      "learning_rate": 1.4194373401534528e-06,
+      "loss": 0.0038,
+      "step": 3775
+    },
+    {
+      "epoch": 9.54,
+      "learning_rate": 1.411764705882353e-06,
+      "loss": 0.002,
+      "step": 3776
+    },
+    {
+      "epoch": 9.54,
+      "learning_rate": 1.4040920716112532e-06,
+      "loss": 0.0041,
+      "step": 3777
+    },
+    {
+      "epoch": 9.54,
+      "learning_rate": 1.3964194373401536e-06,
+      "loss": 0.0036,
+      "step": 3778
+    },
+    {
+      "epoch": 9.54,
+      "learning_rate": 1.3887468030690537e-06,
+      "loss": 0.0028,
+      "step": 3779
+    },
+    {
+      "epoch": 9.55,
+      "learning_rate": 1.381074168797954e-06,
+      "loss": 0.002,
+      "step": 3780
+    },
+    {
+      "epoch": 9.55,
+      "learning_rate": 1.3734015345268542e-06,
+      "loss": 0.0038,
+      "step": 3781
+    },
+    {
+      "epoch": 9.55,
+      "learning_rate": 1.3657289002557545e-06,
+      "loss": 0.0029,
+      "step": 3782
+    },
+    {
+      "epoch": 9.55,
+      "learning_rate": 1.3580562659846547e-06,
+      "loss": 0.0038,
+      "step": 3783
+    },
+    {
+      "epoch": 9.56,
+      "learning_rate": 1.3503836317135552e-06,
+      "loss": 0.0031,
+      "step": 3784
+    },
+    {
+      "epoch": 9.56,
+      "learning_rate": 1.3427109974424553e-06,
+      "loss": 0.0042,
+      "step": 3785
+    },
+    {
+      "epoch": 9.56,
+      "learning_rate": 1.3350383631713555e-06,
+      "loss": 1.5125,
+      "step": 3786
+    },
+    {
+      "epoch": 9.56,
+      "learning_rate": 1.3273657289002558e-06,
+      "loss": 0.0043,
+      "step": 3787
+    },
+    {
+      "epoch": 9.57,
+      "learning_rate": 1.319693094629156e-06,
+      "loss": 0.002,
+      "step": 3788
+    },
+    {
+      "epoch": 9.57,
+      "learning_rate": 1.3120204603580563e-06,
+      "loss": 0.0047,
+      "step": 3789
+    },
+    {
+      "epoch": 9.57,
+      "learning_rate": 1.3043478260869566e-06,
+      "loss": 0.002,
+      "step": 3790
+    },
+    {
+      "epoch": 9.57,
+      "learning_rate": 1.296675191815857e-06,
+      "loss": 0.0018,
+      "step": 3791
+    },
+    {
+      "epoch": 9.58,
+      "learning_rate": 1.289002557544757e-06,
+      "loss": 0.0021,
+      "step": 3792
+    },
+    {
+      "epoch": 9.58,
+      "learning_rate": 1.2813299232736572e-06,
+      "loss": 0.0032,
+      "step": 3793
+    },
+    {
+      "epoch": 9.58,
+      "learning_rate": 1.2736572890025576e-06,
+      "loss": 0.0035,
+      "step": 3794
+    },
+    {
+      "epoch": 9.58,
+      "learning_rate": 1.2659846547314577e-06,
+      "loss": 0.005,
+      "step": 3795
+    },
+    {
+      "epoch": 9.59,
+      "learning_rate": 1.258312020460358e-06,
+      "loss": 0.0038,
+      "step": 3796
+    },
+    {
+      "epoch": 9.59,
+      "learning_rate": 1.2506393861892584e-06,
+      "loss": 0.0029,
+      "step": 3797
+    },
+    {
+      "epoch": 9.59,
+      "learning_rate": 1.2429667519181587e-06,
+      "loss": 0.0041,
+      "step": 3798
+    },
+    {
+      "epoch": 9.59,
+      "learning_rate": 1.2352941176470588e-06,
+      "loss": 0.0035,
+      "step": 3799
+    },
+    {
+      "epoch": 9.6,
+      "learning_rate": 1.2276214833759592e-06,
+      "loss": 0.002,
+      "step": 3800
+    },
+    {
+      "epoch": 9.6,
+      "learning_rate": 1.2199488491048593e-06,
+      "loss": 0.0028,
+      "step": 3801
+    },
+    {
+      "epoch": 9.6,
+      "learning_rate": 1.2122762148337594e-06,
+      "loss": 0.002,
+      "step": 3802
+    },
+    {
+      "epoch": 9.6,
+      "learning_rate": 1.20460358056266e-06,
+      "loss": 0.0034,
+      "step": 3803
+    },
+    {
+      "epoch": 9.61,
+      "learning_rate": 1.1969309462915601e-06,
+      "loss": 0.0018,
+      "step": 3804
+    },
+    {
+      "epoch": 9.61,
+      "learning_rate": 1.1892583120204605e-06,
+      "loss": 0.0029,
+      "step": 3805
+    },
+    {
+      "epoch": 9.61,
+      "learning_rate": 1.1815856777493606e-06,
+      "loss": 0.002,
+      "step": 3806
+    },
+    {
+      "epoch": 9.61,
+      "learning_rate": 1.173913043478261e-06,
+      "loss": 0.0037,
+      "step": 3807
+    },
+    {
+      "epoch": 9.62,
+      "learning_rate": 1.166240409207161e-06,
+      "loss": 0.0046,
+      "step": 3808
+    },
+    {
+      "epoch": 9.62,
+      "learning_rate": 1.1585677749360614e-06,
+      "loss": 0.0023,
+      "step": 3809
+    },
+    {
+      "epoch": 9.62,
+      "learning_rate": 1.1508951406649617e-06,
+      "loss": 0.0028,
+      "step": 3810
+    },
+    {
+      "epoch": 9.62,
+      "learning_rate": 1.1432225063938619e-06,
+      "loss": 0.0021,
+      "step": 3811
+    },
+    {
+      "epoch": 9.63,
+      "learning_rate": 1.1355498721227622e-06,
+      "loss": 0.0036,
+      "step": 3812
+    },
+    {
+      "epoch": 9.63,
+      "learning_rate": 1.1278772378516623e-06,
+      "loss": 0.0037,
+      "step": 3813
+    },
+    {
+      "epoch": 9.63,
+      "learning_rate": 1.1202046035805627e-06,
+      "loss": 0.0019,
+      "step": 3814
+    },
+    {
+      "epoch": 9.63,
+      "learning_rate": 1.112531969309463e-06,
+      "loss": 0.0037,
+      "step": 3815
+    },
+    {
+      "epoch": 9.64,
+      "learning_rate": 1.1048593350383632e-06,
+      "loss": 0.005,
+      "step": 3816
+    },
+    {
+      "epoch": 9.64,
+      "learning_rate": 1.0971867007672635e-06,
+      "loss": 0.002,
+      "step": 3817
+    },
+    {
+      "epoch": 9.64,
+      "learning_rate": 1.0895140664961636e-06,
+      "loss": 0.0019,
+      "step": 3818
+    },
+    {
+      "epoch": 9.64,
+      "learning_rate": 1.081841432225064e-06,
+      "loss": 0.995,
+      "step": 3819
+    },
+    {
+      "epoch": 9.65,
+      "learning_rate": 1.074168797953964e-06,
+      "loss": 0.0054,
+      "step": 3820
+    },
+    {
+      "epoch": 9.65,
+      "learning_rate": 1.0664961636828647e-06,
+      "loss": 0.0019,
+      "step": 3821
+    },
+    {
+      "epoch": 9.65,
+      "learning_rate": 1.0588235294117648e-06,
+      "loss": 0.003,
+      "step": 3822
+    },
+    {
+      "epoch": 9.65,
+      "learning_rate": 1.0511508951406651e-06,
+      "loss": 0.0031,
+      "step": 3823
+    },
+    {
+      "epoch": 9.66,
+      "learning_rate": 1.0434782608695653e-06,
+      "loss": 0.0028,
+      "step": 3824
+    },
+    {
+      "epoch": 9.66,
+      "learning_rate": 1.0358056265984654e-06,
+      "loss": 0.0031,
+      "step": 3825
+    },
+    {
+      "epoch": 9.66,
+      "learning_rate": 1.0281329923273657e-06,
+      "loss": 0.0028,
+      "step": 3826
+    },
+    {
+      "epoch": 9.66,
+      "learning_rate": 1.020460358056266e-06,
+      "loss": 0.002,
+      "step": 3827
+    },
+    {
+      "epoch": 9.67,
+      "learning_rate": 1.0127877237851664e-06,
+      "loss": 0.0026,
+      "step": 3828
+    },
+    {
+      "epoch": 9.67,
+      "learning_rate": 1.0051150895140665e-06,
+      "loss": 0.0037,
+      "step": 3829
+    },
+    {
+      "epoch": 9.67,
+      "learning_rate": 9.974424552429669e-07,
+      "loss": 0.0046,
+      "step": 3830
+    },
+    {
+      "epoch": 9.67,
+      "learning_rate": 9.89769820971867e-07,
+      "loss": 0.0019,
+      "step": 3831
+    },
+    {
+      "epoch": 9.68,
+      "learning_rate": 9.820971867007671e-07,
+      "loss": 0.0046,
+      "step": 3832
+    },
+    {
+      "epoch": 9.68,
+      "learning_rate": 9.744245524296675e-07,
+      "loss": 0.0028,
+      "step": 3833
+    },
+    {
+      "epoch": 9.68,
+      "learning_rate": 9.667519181585678e-07,
+      "loss": 0.002,
+      "step": 3834
+    },
+    {
+      "epoch": 9.68,
+      "learning_rate": 9.590792838874682e-07,
+      "loss": 0.0028,
+      "step": 3835
+    },
+    {
+      "epoch": 9.69,
+      "learning_rate": 9.514066496163683e-07,
+      "loss": 0.0037,
+      "step": 3836
+    },
+    {
+      "epoch": 9.69,
+      "learning_rate": 9.437340153452686e-07,
+      "loss": 0.0035,
+      "step": 3837
+    },
+    {
+      "epoch": 9.69,
+      "learning_rate": 9.360613810741689e-07,
+      "loss": 1.0014,
+      "step": 3838
+    },
+    {
+      "epoch": 9.69,
+      "learning_rate": 9.283887468030691e-07,
+      "loss": 0.0032,
+      "step": 3839
+    },
+    {
+      "epoch": 9.7,
+      "learning_rate": 9.207161125319693e-07,
+      "loss": 0.019,
+      "step": 3840
+    },
+    {
+      "epoch": 9.7,
+      "learning_rate": 9.130434782608697e-07,
+      "loss": 0.0032,
+      "step": 3841
+    },
+    {
+      "epoch": 9.7,
+      "learning_rate": 9.053708439897699e-07,
+      "loss": 0.0028,
+      "step": 3842
+    },
+    {
+      "epoch": 9.7,
+      "learning_rate": 8.9769820971867e-07,
+      "loss": 0.0028,
+      "step": 3843
+    },
+    {
+      "epoch": 9.71,
+      "learning_rate": 8.900255754475704e-07,
+      "loss": 0.0041,
+      "step": 3844
+    },
+    {
+      "epoch": 9.71,
+      "learning_rate": 8.823529411764706e-07,
+      "loss": 0.0028,
+      "step": 3845
+    },
+    {
+      "epoch": 9.71,
+      "learning_rate": 8.746803069053709e-07,
+      "loss": 0.0028,
+      "step": 3846
+    },
+    {
+      "epoch": 9.71,
+      "learning_rate": 8.670076726342711e-07,
+      "loss": 0.0037,
+      "step": 3847
+    },
+    {
+      "epoch": 9.72,
+      "learning_rate": 8.593350383631714e-07,
+      "loss": 0.003,
+      "step": 3848
+    },
+    {
+      "epoch": 9.72,
+      "learning_rate": 8.516624040920717e-07,
+      "loss": 0.0029,
+      "step": 3849
+    },
+    {
+      "epoch": 9.72,
+      "learning_rate": 8.439897698209718e-07,
+      "loss": 0.0034,
+      "step": 3850
+    },
+    {
+      "epoch": 9.72,
+      "learning_rate": 8.363171355498721e-07,
+      "loss": 0.0019,
+      "step": 3851
+    },
+    {
+      "epoch": 9.73,
+      "learning_rate": 8.286445012787724e-07,
+      "loss": 0.0019,
+      "step": 3852
+    },
+    {
+      "epoch": 9.73,
+      "learning_rate": 8.209718670076726e-07,
+      "loss": 0.0029,
+      "step": 3853
+    },
+    {
+      "epoch": 9.73,
+      "learning_rate": 8.132992327365729e-07,
+      "loss": 0.0026,
+      "step": 3854
+    },
+    {
+      "epoch": 9.73,
+      "learning_rate": 8.056265984654732e-07,
+      "loss": 0.0046,
+      "step": 3855
+    },
+    {
+      "epoch": 9.74,
+      "learning_rate": 7.979539641943734e-07,
+      "loss": 0.0028,
+      "step": 3856
+    },
+    {
+      "epoch": 9.74,
+      "learning_rate": 7.902813299232738e-07,
+      "loss": 0.002,
+      "step": 3857
+    },
+    {
+      "epoch": 9.74,
+      "learning_rate": 7.826086956521739e-07,
+      "loss": 2.5479,
+      "step": 3858
+    },
+    {
+      "epoch": 9.74,
+      "learning_rate": 7.749360613810741e-07,
+      "loss": 1.5623,
+      "step": 3859
+    },
+    {
+      "epoch": 9.75,
+      "learning_rate": 7.672634271099745e-07,
+      "loss": 0.0022,
+      "step": 3860
+    },
+    {
+      "epoch": 9.75,
+      "learning_rate": 7.595907928388747e-07,
+      "loss": 0.0028,
+      "step": 3861
+    },
+    {
+      "epoch": 9.75,
+      "learning_rate": 7.519181585677749e-07,
+      "loss": 0.0037,
+      "step": 3862
+    },
+    {
+      "epoch": 9.76,
+      "learning_rate": 7.442455242966753e-07,
+      "loss": 1.3087,
+      "step": 3863
+    },
+    {
+      "epoch": 9.76,
+      "learning_rate": 7.365728900255755e-07,
+      "loss": 0.841,
+      "step": 3864
+    },
+    {
+      "epoch": 9.76,
+      "learning_rate": 7.289002557544756e-07,
+      "loss": 0.0033,
+      "step": 3865
+    },
+    {
+      "epoch": 9.76,
+      "learning_rate": 7.21227621483376e-07,
+      "loss": 0.0028,
+      "step": 3866
+    },
+    {
+      "epoch": 9.77,
+      "learning_rate": 7.135549872122762e-07,
+      "loss": 0.0026,
+      "step": 3867
+    },
+    {
+      "epoch": 9.77,
+      "learning_rate": 7.058823529411765e-07,
+      "loss": 0.0041,
+      "step": 3868
+    },
+    {
+      "epoch": 9.77,
+      "learning_rate": 6.982097186700768e-07,
+      "loss": 0.0034,
+      "step": 3869
+    },
+    {
+      "epoch": 9.77,
+      "learning_rate": 6.90537084398977e-07,
+      "loss": 0.0026,
+      "step": 3870
+    },
+    {
+      "epoch": 9.78,
+      "learning_rate": 6.828644501278773e-07,
+      "loss": 0.0033,
+      "step": 3871
+    },
+    {
+      "epoch": 9.78,
+      "learning_rate": 6.751918158567776e-07,
+      "loss": 0.0062,
+      "step": 3872
+    },
+    {
+      "epoch": 9.78,
+      "learning_rate": 6.675191815856777e-07,
+      "loss": 0.0046,
+      "step": 3873
+    },
+    {
+      "epoch": 9.78,
+      "learning_rate": 6.59846547314578e-07,
+      "loss": 0.0029,
+      "step": 3874
+    },
+    {
+      "epoch": 9.79,
+      "learning_rate": 6.521739130434783e-07,
+      "loss": 0.0014,
+      "step": 3875
+    },
+    {
+      "epoch": 9.79,
+      "learning_rate": 6.445012787723785e-07,
+      "loss": 0.0031,
+      "step": 3876
+    },
+    {
+      "epoch": 9.79,
+      "learning_rate": 6.368286445012788e-07,
+      "loss": 1.2915,
+      "step": 3877
+    },
+    {
+      "epoch": 9.79,
+      "learning_rate": 6.29156010230179e-07,
+      "loss": 0.004,
+      "step": 3878
+    },
+    {
+      "epoch": 9.8,
+      "learning_rate": 6.214833759590794e-07,
+      "loss": 0.056,
+      "step": 3879
+    },
+    {
+      "epoch": 9.8,
+      "learning_rate": 6.138107416879796e-07,
+      "loss": 0.0019,
+      "step": 3880
+    },
+    {
+      "epoch": 9.8,
+      "learning_rate": 6.061381074168797e-07,
+      "loss": 0.0037,
+      "step": 3881
+    },
+    {
+      "epoch": 9.8,
+      "learning_rate": 5.984654731457801e-07,
+      "loss": 0.0028,
+      "step": 3882
+    },
+    {
+      "epoch": 9.81,
+      "learning_rate": 5.907928388746803e-07,
+      "loss": 0.0028,
+      "step": 3883
+    },
+    {
+      "epoch": 9.81,
+      "learning_rate": 5.831202046035805e-07,
+      "loss": 0.0032,
+      "step": 3884
+    },
+    {
+      "epoch": 9.81,
+      "learning_rate": 5.754475703324809e-07,
+      "loss": 0.0028,
+      "step": 3885
+    },
+    {
+      "epoch": 9.81,
+      "learning_rate": 5.677749360613811e-07,
+      "loss": 0.0028,
+      "step": 3886
+    },
+    {
+      "epoch": 9.82,
+      "learning_rate": 5.601023017902813e-07,
+      "loss": 0.0029,
+      "step": 3887
+    },
+    {
+      "epoch": 9.82,
+      "learning_rate": 5.524296675191816e-07,
+      "loss": 0.0041,
+      "step": 3888
+    },
+    {
+      "epoch": 9.82,
+      "learning_rate": 5.447570332480818e-07,
+      "loss": 0.0041,
+      "step": 3889
+    },
+    {
+      "epoch": 9.82,
+      "learning_rate": 5.37084398976982e-07,
+      "loss": 0.0046,
+      "step": 3890
+    },
+    {
+      "epoch": 9.83,
+      "learning_rate": 5.294117647058824e-07,
+      "loss": 0.0037,
+      "step": 3891
+    },
+    {
+      "epoch": 9.83,
+      "learning_rate": 5.217391304347826e-07,
+      "loss": 1.5666,
+      "step": 3892
+    },
+    {
+      "epoch": 9.83,
+      "learning_rate": 5.140664961636829e-07,
+      "loss": 0.0028,
+      "step": 3893
+    },
+    {
+      "epoch": 9.83,
+      "learning_rate": 5.063938618925832e-07,
+      "loss": 0.0037,
+      "step": 3894
+    },
+    {
+      "epoch": 9.84,
+      "learning_rate": 4.987212276214834e-07,
+      "loss": 0.0028,
+      "step": 3895
+    },
+    {
+      "epoch": 9.84,
+      "learning_rate": 4.910485933503836e-07,
+      "loss": 1.4502,
+      "step": 3896
+    },
+    {
+      "epoch": 9.84,
+      "learning_rate": 4.833759590792839e-07,
+      "loss": 0.0051,
+      "step": 3897
+    },
+    {
+      "epoch": 9.84,
+      "learning_rate": 4.7570332480818414e-07,
+      "loss": 0.0039,
+      "step": 3898
+    },
+    {
+      "epoch": 9.85,
+      "learning_rate": 4.6803069053708443e-07,
+      "loss": 0.0021,
+      "step": 3899
+    },
+    {
+      "epoch": 9.85,
+      "learning_rate": 4.6035805626598467e-07,
+      "loss": 0.0014,
+      "step": 3900
+    },
+    {
+      "epoch": 9.85,
+      "learning_rate": 4.5268542199488496e-07,
+      "loss": 0.0062,
+      "step": 3901
+    },
+    {
+      "epoch": 9.85,
+      "learning_rate": 4.450127877237852e-07,
+      "loss": 0.003,
+      "step": 3902
+    },
+    {
+      "epoch": 9.86,
+      "learning_rate": 4.373401534526854e-07,
+      "loss": 0.0034,
+      "step": 3903
+    },
+    {
+      "epoch": 9.86,
+      "learning_rate": 4.296675191815857e-07,
+      "loss": 0.0051,
+      "step": 3904
+    },
+    {
+      "epoch": 9.86,
+      "learning_rate": 4.219948849104859e-07,
+      "loss": 0.0041,
+      "step": 3905
+    },
+    {
+      "epoch": 9.86,
+      "learning_rate": 4.143222506393862e-07,
+      "loss": 0.0031,
+      "step": 3906
+    },
+    {
+      "epoch": 9.87,
+      "learning_rate": 4.0664961636828647e-07,
+      "loss": 0.0028,
+      "step": 3907
+    },
+    {
+      "epoch": 9.87,
+      "learning_rate": 3.989769820971867e-07,
+      "loss": 0.0027,
+      "step": 3908
+    },
+    {
+      "epoch": 9.87,
+      "learning_rate": 3.9130434782608694e-07,
+      "loss": 0.0038,
+      "step": 3909
+    },
+    {
+      "epoch": 9.87,
+      "learning_rate": 3.8363171355498723e-07,
+      "loss": 0.0225,
+      "step": 3910
+    },
+    {
+      "epoch": 9.88,
+      "learning_rate": 3.7595907928388747e-07,
+      "loss": 0.0038,
+      "step": 3911
+    },
+    {
+      "epoch": 9.88,
+      "learning_rate": 3.6828644501278776e-07,
+      "loss": 0.002,
+      "step": 3912
+    },
+    {
+      "epoch": 9.88,
+      "learning_rate": 3.60613810741688e-07,
+      "loss": 0.0029,
+      "step": 3913
+    },
+    {
+      "epoch": 9.88,
+      "learning_rate": 3.529411764705882e-07,
+      "loss": 0.0019,
+      "step": 3914
+    },
+    {
+      "epoch": 9.89,
+      "learning_rate": 3.452685421994885e-07,
+      "loss": 1.4576,
+      "step": 3915
+    },
+    {
+      "epoch": 9.89,
+      "learning_rate": 3.375959079283888e-07,
+      "loss": 0.002,
+      "step": 3916
+    },
+    {
+      "epoch": 9.89,
+      "learning_rate": 3.29923273657289e-07,
+      "loss": 0.065,
+      "step": 3917
+    },
+    {
+      "epoch": 9.89,
+      "learning_rate": 3.2225063938618927e-07,
+      "loss": 0.0029,
+      "step": 3918
+    },
+    {
+      "epoch": 9.9,
+      "learning_rate": 3.145780051150895e-07,
+      "loss": 0.0032,
+      "step": 3919
+    },
+    {
+      "epoch": 9.9,
+      "learning_rate": 3.069053708439898e-07,
+      "loss": 0.0032,
+      "step": 3920
+    },
+    {
+      "epoch": 9.9,
+      "learning_rate": 2.9923273657289003e-07,
+      "loss": 0.0037,
+      "step": 3921
+    },
+    {
+      "epoch": 9.9,
+      "learning_rate": 2.9156010230179027e-07,
+      "loss": 2.4894,
+      "step": 3922
+    },
+    {
+      "epoch": 9.91,
+      "learning_rate": 2.8388746803069055e-07,
+      "loss": 0.0109,
+      "step": 3923
+    },
+    {
+      "epoch": 9.91,
+      "learning_rate": 2.762148337595908e-07,
+      "loss": 0.0043,
+      "step": 3924
+    },
+    {
+      "epoch": 9.91,
+      "learning_rate": 2.68542199488491e-07,
+      "loss": 0.0029,
+      "step": 3925
+    },
+    {
+      "epoch": 9.91,
+      "learning_rate": 2.608695652173913e-07,
+      "loss": 0.0026,
+      "step": 3926
+    },
+    {
+      "epoch": 9.92,
+      "learning_rate": 2.531969309462916e-07,
+      "loss": 0.0029,
+      "step": 3927
+    },
+    {
+      "epoch": 9.92,
+      "learning_rate": 2.455242966751918e-07,
+      "loss": 0.0018,
+      "step": 3928
+    },
+    {
+      "epoch": 9.92,
+      "learning_rate": 2.3785166240409207e-07,
+      "loss": 0.0028,
+      "step": 3929
+    },
+    {
+      "epoch": 9.92,
+      "learning_rate": 2.3017902813299233e-07,
+      "loss": 0.0015,
+      "step": 3930
+    },
+    {
+      "epoch": 9.93,
+      "learning_rate": 2.225063938618926e-07,
+      "loss": 0.0041,
+      "step": 3931
+    },
+    {
+      "epoch": 9.93,
+      "learning_rate": 2.1483375959079286e-07,
+      "loss": 0.0039,
+      "step": 3932
+    },
+    {
+      "epoch": 9.93,
+      "learning_rate": 2.071611253196931e-07,
+      "loss": 0.0028,
+      "step": 3933
+    },
+    {
+      "epoch": 9.93,
+      "learning_rate": 1.9948849104859335e-07,
+      "loss": 0.0037,
+      "step": 3934
+    },
+    {
+      "epoch": 9.94,
+      "learning_rate": 1.9181585677749362e-07,
+      "loss": 0.002,
+      "step": 3935
+    },
+    {
+      "epoch": 9.94,
+      "learning_rate": 1.8414322250639388e-07,
+      "loss": 0.0019,
+      "step": 3936
+    },
+    {
+      "epoch": 9.94,
+      "learning_rate": 1.764705882352941e-07,
+      "loss": 0.0032,
+      "step": 3937
+    },
+    {
+      "epoch": 9.94,
+      "learning_rate": 1.687979539641944e-07,
+      "loss": 0.0037,
+      "step": 3938
+    },
+    {
+      "epoch": 9.95,
+      "learning_rate": 1.6112531969309464e-07,
+      "loss": 0.002,
+      "step": 3939
+    },
+    {
+      "epoch": 9.95,
+      "learning_rate": 1.534526854219949e-07,
+      "loss": 1.3031,
+      "step": 3940
+    },
+    {
+      "epoch": 9.95,
+      "learning_rate": 1.4578005115089513e-07,
+      "loss": 0.003,
+      "step": 3941
+    },
+    {
+      "epoch": 9.95,
+      "learning_rate": 1.381074168797954e-07,
+      "loss": 0.004,
+      "step": 3942
+    },
+    {
+      "epoch": 9.96,
+      "learning_rate": 1.3043478260869566e-07,
+      "loss": 0.0016,
+      "step": 3943
+    },
+    {
+      "epoch": 9.96,
+      "learning_rate": 1.227621483375959e-07,
+      "loss": 0.0023,
+      "step": 3944
+    },
+    {
+      "epoch": 9.96,
+      "learning_rate": 1.1508951406649617e-07,
+      "loss": 0.0019,
+      "step": 3945
+    },
+    {
+      "epoch": 9.96,
+      "learning_rate": 1.0741687979539643e-07,
+      "loss": 0.0028,
+      "step": 3946
+    },
+    {
+      "epoch": 9.97,
+      "learning_rate": 9.974424552429668e-08,
+      "loss": 1.5173,
+      "step": 3947
+    },
+    {
+      "epoch": 9.97,
+      "learning_rate": 9.207161125319694e-08,
+      "loss": 0.0021,
+      "step": 3948
+    },
+    {
+      "epoch": 9.97,
+      "learning_rate": 8.43989769820972e-08,
+      "loss": 0.0039,
+      "step": 3949
+    },
+    {
+      "epoch": 9.97,
+      "learning_rate": 7.672634271099745e-08,
+      "loss": 0.0046,
+      "step": 3950
+    },
+    {
+      "epoch": 9.98,
+      "learning_rate": 6.90537084398977e-08,
+      "loss": 0.0028,
+      "step": 3951
+    },
+    {
+      "epoch": 9.98,
+      "learning_rate": 6.138107416879795e-08,
+      "loss": 0.0037,
+      "step": 3952
+    },
+    {
+      "epoch": 9.98,
+      "learning_rate": 5.3708439897698214e-08,
+      "loss": 0.0019,
+      "step": 3953
+    },
+    {
+      "epoch": 9.98,
+      "learning_rate": 4.603580562659847e-08,
+      "loss": 0.0028,
+      "step": 3954
+    },
+    {
+      "epoch": 9.99,
+      "learning_rate": 3.8363171355498724e-08,
+      "loss": 0.0019,
+      "step": 3955
+    },
+    {
+      "epoch": 9.99,
+      "learning_rate": 3.069053708439897e-08,
+      "loss": 0.007,
+      "step": 3956
+    },
+    {
+      "epoch": 9.99,
+      "learning_rate": 2.3017902813299235e-08,
+      "loss": 0.002,
+      "step": 3957
+    },
+    {
+      "epoch": 9.99,
+      "learning_rate": 1.5345268542199486e-08,
+      "loss": 0.0029,
+      "step": 3958
+    },
+    {
+      "epoch": 10.0,
+      "learning_rate": 7.672634271099743e-09,
+      "loss": 0.0039,
+      "step": 3959
+    },
+    {
+      "epoch": 10.0,
+      "learning_rate": 0.0,
+      "loss": 0.0031,
+      "step": 3960
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 0.9141414141414141,
+      "eval_loss": 0.46265003085136414,
+      "eval_runtime": 58.0919,
+      "eval_samples_per_second": 6.817,
+      "eval_steps_per_second": 1.704,
+      "step": 3960
+    }
+  ],
+  "max_steps": 3960,
+  "num_train_epochs": 10,
+  "total_flos": 1.437147546912e+18,
+  "trial_name": null,
+  "trial_params": null
+}
diff --git a/checkpoint-3960/training_args.bin b/checkpoint-3960/training_args.bin
new file mode 100644
index 0000000000000000000000000000000000000000..32901f899c91066c808eae9dd019400bfdd17498
--- /dev/null
+++ b/checkpoint-3960/training_args.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db14439c369bfe9c89f64ff38f0b2d3fcc53911fc2c0b12a3a7d04656e7384a
+size 3503
diff --git a/checkpoint-792/config.json b/checkpoint-792/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..cb8d52e14263c50fd53dca42b5050e3235115c32
--- /dev/null
+++ b/checkpoint-792/config.json
@@ -0,0 +1,113 @@
+{
+  "_name_or_path": "facebook/wav2vec2-base-960h",
+  "activation_dropout": 0.1,
+  "adapter_kernel_size": 3,
+  "adapter_stride": 2,
+  "add_adapter": false,
+  "apply_spec_augment": true,
+  "architectures": [
+    "Wav2Vec2ForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 1,
+  "classifier_proj_size": 256,
+  "codevector_dim": 256,
+  "contrastive_logits_temperature": 0.1,
+  "conv_bias": false,
+  "conv_dim": [
+    512,
+    512,
+    512,
+    512,
+    512,
+    512,
+    512
+  ],
+  "conv_kernel": [
+    10,
+    3,
+    3,
+    3,
+    3,
+    2,
+    2
+  ],
+  "conv_stride": [
+    5,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2
+  ],
+  "ctc_loss_reduction": "sum",
+  "ctc_zero_infinity": false,
+  "diversity_loss_weight": 0.1,
+  "do_stable_layer_norm": false,
+  "eos_token_id": 2,
+  "feat_extract_activation": "gelu",
+  "feat_extract_dropout": 0.0,
+  "feat_extract_norm": "group",
+  "feat_proj_dropout": 0.1,
+  "feat_quantizer_dropout": 0.0,
+  "final_dropout": 0.1,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout": 0.1,
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Foreign",
+    "1": "Native"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "layerdrop": 0.1,
+  "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
+  "mask_feature_prob": 0.0,
+  "mask_time_length": 10,
+  "mask_time_min_masks": 2,
+  "mask_time_prob": 0.05,
+  "model_type": "wav2vec2",
+  "num_adapter_layers": 3,
+  "num_attention_heads": 12,
+  "num_codevector_groups": 2,
+  "num_codevectors_per_group": 320,
+  "num_conv_pos_embedding_groups": 16,
+  "num_conv_pos_embeddings": 128,
+  "num_feat_extract_layers": 7,
+  "num_hidden_layers": 12,
+  "num_negatives": 100,
+  "output_hidden_size": 768,
+  "pad_token_id": 0,
+  "proj_codevector_dim": 256,
+  "tdnn_dilation": [
+    1,
+    2,
+    3,
+    1,
+    1
+  ],
+  "tdnn_dim": [
+    512,
+    512,
+    512,
+    512,
+    1500
+  ],
+  "tdnn_kernel": [
+    5,
+    3,
+    3,
+    1,
+    1
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.28.1",
+  "use_weighted_layer_sum": false,
+  "vocab_size": 32,
+  "xvector_output_dim": 512
+}
diff --git a/checkpoint-792/optimizer.pt b/checkpoint-792/optimizer.pt
new file mode 100644
index 0000000000000000000000000000000000000000..523e39236727a81f77fae588c8484015ac69202c
--- /dev/null
+++ b/checkpoint-792/optimizer.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf7bbf3b4dd6c569fd11ebd2948bf322f3f6465ec1a6166b30b910fbff2d5d6a
+size 756677477
diff --git a/checkpoint-792/preprocessor_config.json b/checkpoint-792/preprocessor_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..a0b7227fc1d916e469b14f6c154ad6dfea1e6891
--- /dev/null
+++ b/checkpoint-792/preprocessor_config.json
@@ -0,0 +1,9 @@
+{
+  "do_normalize": true,
+  "feature_extractor_type": "Wav2Vec2FeatureExtractor",
+  "feature_size": 1,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": false,
+  "sampling_rate": 16000
+}
diff --git a/checkpoint-792/pytorch_model.bin b/checkpoint-792/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..f734f655de81a3cb7cf75499c3e6a76a15ed04d3
--- /dev/null
+++ b/checkpoint-792/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:52f50a631c7d9c5cf12650f14ff42c798bbccbe4f20700bfd455ffd5ec67f931
+size 378348563
diff --git a/checkpoint-792/rng_state.pth b/checkpoint-792/rng_state.pth
new file mode 100644
index 0000000000000000000000000000000000000000..80e12aa3c9bf44d481fc6b0ecd941c6764ddc20a
--- /dev/null
+++ b/checkpoint-792/rng_state.pth
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8f16f32de637fd83fc5d800b18745f1aa20b3fac786697c16ccad1885d6eb138
+size 14567
diff --git a/checkpoint-792/scheduler.pt b/checkpoint-792/scheduler.pt
new file mode 100644
index 0000000000000000000000000000000000000000..3ce200ba17ade7148721fc8f77e257ea0fad051f
--- /dev/null
+++ b/checkpoint-792/scheduler.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ed0778c0c030f2d2b07f1e5e2839a06e9830fbd5f27d191057c0b564ce009d3b
+size 623
diff --git a/checkpoint-792/trainer_state.json b/checkpoint-792/trainer_state.json
new file mode 100644
index 0000000000000000000000000000000000000000..0ff20f9484269e03f478ab178fa9118e67e66f91
--- /dev/null
+++ b/checkpoint-792/trainer_state.json
@@ -0,0 +1,4786 @@
+{
+  "best_metric": 0.3537317216396332,
+  "best_model_checkpoint": "accent_classification/checkpoint-792",
+  "epoch": 2.0,
+  "global_step": 792,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.000000000000001e-07,
+      "loss": 0.6956,
+      "step": 1
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.2000000000000002e-06,
+      "loss": 0.6979,
+      "step": 2
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8e-06,
+      "loss": 0.7014,
+      "step": 3
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 2.4000000000000003e-06,
+      "loss": 0.6984,
+      "step": 4
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 3e-06,
+      "loss": 0.6978,
+      "step": 5
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 3.6e-06,
+      "loss": 0.6783,
+      "step": 6
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.2000000000000004e-06,
+      "loss": 0.6956,
+      "step": 7
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 4.800000000000001e-06,
+      "loss": 0.6994,
+      "step": 8
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 5.4e-06,
+      "loss": 0.6902,
+      "step": 9
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6e-06,
+      "loss": 0.6937,
+      "step": 10
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 6.6e-06,
+      "loss": 0.6497,
+      "step": 11
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.2e-06,
+      "loss": 0.6911,
+      "step": 12
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 7.8e-06,
+      "loss": 0.6904,
+      "step": 13
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 8.400000000000001e-06,
+      "loss": 0.6854,
+      "step": 14
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9e-06,
+      "loss": 0.6843,
+      "step": 15
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9.600000000000001e-06,
+      "loss": 0.6851,
+      "step": 16
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.02e-05,
+      "loss": 0.6733,
+      "step": 17
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.08e-05,
+      "loss": 0.6819,
+      "step": 18
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.1400000000000001e-05,
+      "loss": 0.681,
+      "step": 19
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.2e-05,
+      "loss": 0.6998,
+      "step": 20
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.26e-05,
+      "loss": 0.6743,
+      "step": 21
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.32e-05,
+      "loss": 0.6818,
+      "step": 22
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3800000000000002e-05,
+      "loss": 0.5331,
+      "step": 23
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.44e-05,
+      "loss": 0.6681,
+      "step": 24
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.5e-05,
+      "loss": 0.6955,
+      "step": 25
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.56e-05,
+      "loss": 0.7,
+      "step": 26
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.62e-05,
+      "loss": 0.6975,
+      "step": 27
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.6800000000000002e-05,
+      "loss": 0.6249,
+      "step": 28
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.74e-05,
+      "loss": 0.6979,
+      "step": 29
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.8e-05,
+      "loss": 0.6076,
+      "step": 30
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.86e-05,
+      "loss": 0.6091,
+      "step": 31
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9200000000000003e-05,
+      "loss": 0.6438,
+      "step": 32
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.98e-05,
+      "loss": 0.6624,
+      "step": 33
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.04e-05,
+      "loss": 0.6452,
+      "step": 34
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.1e-05,
+      "loss": 0.6963,
+      "step": 35
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.16e-05,
+      "loss": 0.758,
+      "step": 36
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 2.22e-05,
+      "loss": 0.6471,
+      "step": 37
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.2800000000000002e-05,
+      "loss": 0.6313,
+      "step": 38
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.3400000000000003e-05,
+      "loss": 0.7019,
+      "step": 39
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4e-05,
+      "loss": 0.6336,
+      "step": 40
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.4599999999999998e-05,
+      "loss": 0.7091,
+      "step": 41
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.52e-05,
+      "loss": 0.701,
+      "step": 42
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.58e-05,
+      "loss": 0.7036,
+      "step": 43
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.64e-05,
+      "loss": 0.7073,
+      "step": 44
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.614,
+      "step": 45
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.7600000000000003e-05,
+      "loss": 0.7543,
+      "step": 46
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.8199999999999998e-05,
+      "loss": 0.538,
+      "step": 47
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.88e-05,
+      "loss": 0.5179,
+      "step": 48
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.94e-05,
+      "loss": 0.6164,
+      "step": 49
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 3e-05,
+      "loss": 0.6093,
+      "step": 50
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.99923273657289e-05,
+      "loss": 0.6983,
+      "step": 51
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9984654731457804e-05,
+      "loss": 0.6069,
+      "step": 52
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 2.9976982097186702e-05,
+      "loss": 0.5933,
+      "step": 53
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.99693094629156e-05,
+      "loss": 0.552,
+      "step": 54
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9961636828644502e-05,
+      "loss": 0.5865,
+      "step": 55
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9953964194373404e-05,
+      "loss": 0.7146,
+      "step": 56
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 2.9946291560102302e-05,
+      "loss": 0.5591,
+      "step": 57
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9938618925831203e-05,
+      "loss": 0.7366,
+      "step": 58
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9930946291560102e-05,
+      "loss": 0.7006,
+      "step": 59
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9923273657289003e-05,
+      "loss": 0.4148,
+      "step": 60
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 2.9915601023017905e-05,
+      "loss": 0.9938,
+      "step": 61
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9907928388746803e-05,
+      "loss": 0.911,
+      "step": 62
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.99002557544757e-05,
+      "loss": 0.5682,
+      "step": 63
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9892583120204606e-05,
+      "loss": 0.5833,
+      "step": 64
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 2.9884910485933505e-05,
+      "loss": 0.7237,
+      "step": 65
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9877237851662403e-05,
+      "loss": 0.4013,
+      "step": 66
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9869565217391304e-05,
+      "loss": 0.9695,
+      "step": 67
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9861892583120206e-05,
+      "loss": 0.568,
+      "step": 68
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.9854219948849104e-05,
+      "loss": 0.5697,
+      "step": 69
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9846547314578006e-05,
+      "loss": 0.7675,
+      "step": 70
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9838874680306904e-05,
+      "loss": 0.8831,
+      "step": 71
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9831202046035806e-05,
+      "loss": 0.7008,
+      "step": 72
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 2.9823529411764707e-05,
+      "loss": 0.7085,
+      "step": 73
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9815856777493606e-05,
+      "loss": 0.5374,
+      "step": 74
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9808184143222507e-05,
+      "loss": 0.4131,
+      "step": 75
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.980051150895141e-05,
+      "loss": 0.7309,
+      "step": 76
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 2.9792838874680307e-05,
+      "loss": 0.4279,
+      "step": 77
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.978516624040921e-05,
+      "loss": 0.5813,
+      "step": 78
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.977749360613811e-05,
+      "loss": 0.6276,
+      "step": 79
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976982097186701e-05,
+      "loss": 0.7892,
+      "step": 80
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 2.976214833759591e-05,
+      "loss": 0.7134,
+      "step": 81
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.9754475703324808e-05,
+      "loss": 0.3682,
+      "step": 82
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.974680306905371e-05,
+      "loss": 0.5562,
+      "step": 83
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973913043478261e-05,
+      "loss": 0.7158,
+      "step": 84
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 2.973145780051151e-05,
+      "loss": 0.524,
+      "step": 85
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9723785166240408e-05,
+      "loss": 0.3372,
+      "step": 86
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.9716112531969313e-05,
+      "loss": 0.4979,
+      "step": 87
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970843989769821e-05,
+      "loss": 0.3177,
+      "step": 88
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 2.970076726342711e-05,
+      "loss": 0.3488,
+      "step": 89
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.969309462915601e-05,
+      "loss": 0.5158,
+      "step": 90
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9685421994884913e-05,
+      "loss": 0.3208,
+      "step": 91
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.967774936061381e-05,
+      "loss": 0.6654,
+      "step": 92
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 2.9670076726342712e-05,
+      "loss": 0.5581,
+      "step": 93
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.966240409207161e-05,
+      "loss": 0.7456,
+      "step": 94
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9654731457800512e-05,
+      "loss": 0.4437,
+      "step": 95
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9647058823529414e-05,
+      "loss": 0.5525,
+      "step": 96
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 2.9639386189258312e-05,
+      "loss": 0.6673,
+      "step": 97
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.963171355498721e-05,
+      "loss": 0.435,
+      "step": 98
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9624040920716115e-05,
+      "loss": 0.7415,
+      "step": 99
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 2.9616368286445014e-05,
+      "loss": 0.2981,
+      "step": 100
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9608695652173912e-05,
+      "loss": 0.5179,
+      "step": 101
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9601023017902817e-05,
+      "loss": 0.4005,
+      "step": 102
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9593350383631715e-05,
+      "loss": 0.4598,
+      "step": 103
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 2.9585677749360613e-05,
+      "loss": 0.2553,
+      "step": 104
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9578005115089515e-05,
+      "loss": 0.4302,
+      "step": 105
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9570332480818416e-05,
+      "loss": 0.7894,
+      "step": 106
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9562659846547315e-05,
+      "loss": 0.213,
+      "step": 107
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 2.9554987212276216e-05,
+      "loss": 0.5507,
+      "step": 108
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9547314578005115e-05,
+      "loss": 0.7676,
+      "step": 109
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9539641943734016e-05,
+      "loss": 0.3418,
+      "step": 110
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9531969309462918e-05,
+      "loss": 0.8533,
+      "step": 111
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 2.9524296675191816e-05,
+      "loss": 0.3659,
+      "step": 112
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9516624040920714e-05,
+      "loss": 0.6482,
+      "step": 113
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.950895140664962e-05,
+      "loss": 0.5484,
+      "step": 114
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9501278772378517e-05,
+      "loss": 0.4508,
+      "step": 115
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 2.9493606138107416e-05,
+      "loss": 0.4401,
+      "step": 116
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9485933503836317e-05,
+      "loss": 0.5455,
+      "step": 117
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.947826086956522e-05,
+      "loss": 0.722,
+      "step": 118
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.9470588235294117e-05,
+      "loss": 0.2874,
+      "step": 119
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 2.946291560102302e-05,
+      "loss": 0.5973,
+      "step": 120
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.9455242966751917e-05,
+      "loss": 0.5898,
+      "step": 121
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.944757033248082e-05,
+      "loss": 1.0125,
+      "step": 122
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943989769820972e-05,
+      "loss": 0.2511,
+      "step": 123
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 2.943222506393862e-05,
+      "loss": 0.545,
+      "step": 124
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.942455242966752e-05,
+      "loss": 0.593,
+      "step": 125
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.941687979539642e-05,
+      "loss": 0.2319,
+      "step": 126
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.940920716112532e-05,
+      "loss": 0.2215,
+      "step": 127
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 2.9401534526854218e-05,
+      "loss": 0.1964,
+      "step": 128
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9393861892583123e-05,
+      "loss": 0.1471,
+      "step": 129
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.938618925831202e-05,
+      "loss": 1.5965,
+      "step": 130
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.9378516624040923e-05,
+      "loss": 1.0772,
+      "step": 131
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 2.937084398976982e-05,
+      "loss": 0.2005,
+      "step": 132
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9363171355498723e-05,
+      "loss": 0.9628,
+      "step": 133
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9355498721227624e-05,
+      "loss": 0.1503,
+      "step": 134
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.9347826086956523e-05,
+      "loss": 0.6201,
+      "step": 135
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 2.934015345268542e-05,
+      "loss": 0.6094,
+      "step": 136
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9332480818414326e-05,
+      "loss": 1.385,
+      "step": 137
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9324808184143224e-05,
+      "loss": 0.1552,
+      "step": 138
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9317135549872122e-05,
+      "loss": 0.5828,
+      "step": 139
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 2.9309462915601024e-05,
+      "loss": 0.6018,
+      "step": 140
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9301790281329925e-05,
+      "loss": 1.0614,
+      "step": 141
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9294117647058824e-05,
+      "loss": 0.566,
+      "step": 142
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9286445012787725e-05,
+      "loss": 0.5958,
+      "step": 143
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 2.9278772378516624e-05,
+      "loss": 0.5838,
+      "step": 144
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9271099744245525e-05,
+      "loss": 0.5909,
+      "step": 145
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9263427109974427e-05,
+      "loss": 0.5862,
+      "step": 146
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9255754475703325e-05,
+      "loss": 0.2031,
+      "step": 147
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 2.9248081841432223e-05,
+      "loss": 0.1995,
+      "step": 148
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9240409207161128e-05,
+      "loss": 0.24,
+      "step": 149
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9232736572890026e-05,
+      "loss": 0.5845,
+      "step": 150
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.9225063938618925e-05,
+      "loss": 0.964,
+      "step": 151
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 2.921739130434783e-05,
+      "loss": 0.9595,
+      "step": 152
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9209718670076728e-05,
+      "loss": 0.2033,
+      "step": 153
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9202046035805626e-05,
+      "loss": 0.213,
+      "step": 154
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.9194373401534528e-05,
+      "loss": 0.2042,
+      "step": 155
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 2.918670076726343e-05,
+      "loss": 0.5783,
+      "step": 156
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9179028132992328e-05,
+      "loss": 1.0822,
+      "step": 157
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.917135549872123e-05,
+      "loss": 1.2643,
+      "step": 158
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.9163682864450127e-05,
+      "loss": 0.2593,
+      "step": 159
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 2.915601023017903e-05,
+      "loss": 0.5783,
+      "step": 160
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914833759590793e-05,
+      "loss": 0.5721,
+      "step": 161
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.914066496163683e-05,
+      "loss": 0.5808,
+      "step": 162
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9132992327365727e-05,
+      "loss": 1.2578,
+      "step": 163
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 2.9125319693094632e-05,
+      "loss": 0.5709,
+      "step": 164
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.911764705882353e-05,
+      "loss": 0.2159,
+      "step": 165
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910997442455243e-05,
+      "loss": 0.2329,
+      "step": 166
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.910230179028133e-05,
+      "loss": 0.926,
+      "step": 167
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 2.9094629156010232e-05,
+      "loss": 0.9095,
+      "step": 168
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.908695652173913e-05,
+      "loss": 0.5622,
+      "step": 169
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907928388746803e-05,
+      "loss": 0.5703,
+      "step": 170
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.907161125319693e-05,
+      "loss": 0.5669,
+      "step": 171
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 2.906393861892583e-05,
+      "loss": 0.2637,
+      "step": 172
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9056265984654733e-05,
+      "loss": 1.1077,
+      "step": 173
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.904859335038363e-05,
+      "loss": 0.5653,
+      "step": 174
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9040920716112533e-05,
+      "loss": 0.5584,
+      "step": 175
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 2.9033248081841434e-05,
+      "loss": 0.5544,
+      "step": 176
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9025575447570333e-05,
+      "loss": 0.5663,
+      "step": 177
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.901790281329923e-05,
+      "loss": 1.0808,
+      "step": 178
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9010230179028136e-05,
+      "loss": 0.8736,
+      "step": 179
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 2.9002557544757034e-05,
+      "loss": 0.6435,
+      "step": 180
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8994884910485932e-05,
+      "loss": 0.786,
+      "step": 181
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8987212276214834e-05,
+      "loss": 0.5717,
+      "step": 182
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8979539641943736e-05,
+      "loss": 0.5618,
+      "step": 183
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 2.8971867007672637e-05,
+      "loss": 0.5785,
+      "step": 184
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8964194373401535e-05,
+      "loss": 0.7601,
+      "step": 185
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8956521739130434e-05,
+      "loss": 0.8112,
+      "step": 186
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.894884910485934e-05,
+      "loss": 0.5663,
+      "step": 187
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 2.8941176470588237e-05,
+      "loss": 0.5631,
+      "step": 188
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8933503836317135e-05,
+      "loss": 0.7363,
+      "step": 189
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8925831202046037e-05,
+      "loss": 0.4552,
+      "step": 190
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8918158567774938e-05,
+      "loss": 0.4599,
+      "step": 191
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 2.8910485933503836e-05,
+      "loss": 0.7848,
+      "step": 192
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8902813299232738e-05,
+      "loss": 0.5907,
+      "step": 193
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8895140664961636e-05,
+      "loss": 0.4264,
+      "step": 194
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.8887468030690538e-05,
+      "loss": 0.5697,
+      "step": 195
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 2.887979539641944e-05,
+      "loss": 0.7643,
+      "step": 196
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8872122762148338e-05,
+      "loss": 0.3635,
+      "step": 197
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.8864450127877236e-05,
+      "loss": 0.3715,
+      "step": 198
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 2.885677749360614e-05,
+      "loss": 0.554,
+      "step": 199
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.884910485933504e-05,
+      "loss": 0.8044,
+      "step": 200
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8841432225063937e-05,
+      "loss": 0.5673,
+      "step": 201
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.8833759590792842e-05,
+      "loss": 0.8042,
+      "step": 202
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.882608695652174e-05,
+      "loss": 0.5638,
+      "step": 203
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881841432225064e-05,
+      "loss": 0.3584,
+      "step": 204
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.881074168797954e-05,
+      "loss": 0.8211,
+      "step": 205
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.8803069053708442e-05,
+      "loss": 0.5708,
+      "step": 206
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 2.879539641943734e-05,
+      "loss": 0.8301,
+      "step": 207
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8787723785166242e-05,
+      "loss": 0.8229,
+      "step": 208
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.878005115089514e-05,
+      "loss": 0.5645,
+      "step": 209
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8772378516624042e-05,
+      "loss": 0.3529,
+      "step": 210
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 2.8764705882352943e-05,
+      "loss": 0.5578,
+      "step": 211
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.875703324808184e-05,
+      "loss": 0.5659,
+      "step": 212
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.874936061381074e-05,
+      "loss": 0.7495,
+      "step": 213
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8741687979539645e-05,
+      "loss": 0.3162,
+      "step": 214
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 2.8734015345268543e-05,
+      "loss": 0.5616,
+      "step": 215
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.872634271099744e-05,
+      "loss": 0.4034,
+      "step": 216
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8718670076726343e-05,
+      "loss": 0.3163,
+      "step": 217
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8710997442455245e-05,
+      "loss": 0.5514,
+      "step": 218
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.8703324808184143e-05,
+      "loss": 0.826,
+      "step": 219
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8695652173913044e-05,
+      "loss": 0.8363,
+      "step": 220
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8687979539641943e-05,
+      "loss": 0.8288,
+      "step": 221
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8680306905370844e-05,
+      "loss": 0.5929,
+      "step": 222
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 2.8672634271099746e-05,
+      "loss": 0.565,
+      "step": 223
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8664961636828644e-05,
+      "loss": 0.5642,
+      "step": 224
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8657289002557546e-05,
+      "loss": 0.5609,
+      "step": 225
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8649616368286447e-05,
+      "loss": 0.5641,
+      "step": 226
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 2.8641943734015345e-05,
+      "loss": 0.8535,
+      "step": 227
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8634271099744244e-05,
+      "loss": 0.5652,
+      "step": 228
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.862659846547315e-05,
+      "loss": 0.4397,
+      "step": 229
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8618925831202047e-05,
+      "loss": 0.567,
+      "step": 230
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.8611253196930945e-05,
+      "loss": 0.7517,
+      "step": 231
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8603580562659847e-05,
+      "loss": 0.5839,
+      "step": 232
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.859590792838875e-05,
+      "loss": 0.5677,
+      "step": 233
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8588235294117647e-05,
+      "loss": 0.5678,
+      "step": 234
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 2.8580562659846548e-05,
+      "loss": 1.3045,
+      "step": 235
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8572890025575446e-05,
+      "loss": 0.804,
+      "step": 236
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.856521739130435e-05,
+      "loss": 1.0303,
+      "step": 237
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.855754475703325e-05,
+      "loss": 0.3303,
+      "step": 238
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 2.8549872122762148e-05,
+      "loss": 0.5677,
+      "step": 239
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.854219948849105e-05,
+      "loss": 0.7972,
+      "step": 240
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.853452685421995e-05,
+      "loss": 0.5644,
+      "step": 241
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.852685421994885e-05,
+      "loss": 0.9294,
+      "step": 242
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 2.851918158567775e-05,
+      "loss": 0.5665,
+      "step": 243
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.851150895140665e-05,
+      "loss": 0.7963,
+      "step": 244
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.850383631713555e-05,
+      "loss": 0.5659,
+      "step": 245
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.8496163682864452e-05,
+      "loss": 0.5718,
+      "step": 246
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 2.848849104859335e-05,
+      "loss": 0.3526,
+      "step": 247
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8480818414322252e-05,
+      "loss": 0.574,
+      "step": 248
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8473145780051154e-05,
+      "loss": 0.9996,
+      "step": 249
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.8465473145780052e-05,
+      "loss": 0.7634,
+      "step": 250
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 2.845780051150895e-05,
+      "loss": 0.5639,
+      "step": 251
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8450127877237855e-05,
+      "loss": 0.5737,
+      "step": 252
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8442455242966753e-05,
+      "loss": 0.7793,
+      "step": 253
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8434782608695652e-05,
+      "loss": 0.6769,
+      "step": 254
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.8427109974424553e-05,
+      "loss": 0.5748,
+      "step": 255
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8419437340153455e-05,
+      "loss": 0.3602,
+      "step": 256
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8411764705882353e-05,
+      "loss": 0.5714,
+      "step": 257
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8404092071611255e-05,
+      "loss": 0.5738,
+      "step": 258
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 2.8396419437340153e-05,
+      "loss": 0.5719,
+      "step": 259
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8388746803069055e-05,
+      "loss": 0.5753,
+      "step": 260
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8381074168797956e-05,
+      "loss": 0.5717,
+      "step": 261
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8373401534526854e-05,
+      "loss": 0.9794,
+      "step": 262
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 2.8365728900255753e-05,
+      "loss": 0.5737,
+      "step": 263
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8358056265984658e-05,
+      "loss": 0.5723,
+      "step": 264
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8350383631713556e-05,
+      "loss": 0.5656,
+      "step": 265
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8342710997442454e-05,
+      "loss": 0.4259,
+      "step": 266
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 2.8335038363171356e-05,
+      "loss": 0.3076,
+      "step": 267
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8327365728900257e-05,
+      "loss": 0.338,
+      "step": 268
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8319693094629156e-05,
+      "loss": 0.5612,
+      "step": 269
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8312020460358057e-05,
+      "loss": 0.5249,
+      "step": 270
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 2.8304347826086955e-05,
+      "loss": 0.5671,
+      "step": 271
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8296675191815857e-05,
+      "loss": 0.5574,
+      "step": 272
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.828900255754476e-05,
+      "loss": 0.8152,
+      "step": 273
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.8281329923273657e-05,
+      "loss": 0.5647,
+      "step": 274
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 2.827365728900256e-05,
+      "loss": 0.5632,
+      "step": 275
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.826598465473146e-05,
+      "loss": 0.3391,
+      "step": 276
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8258312020460358e-05,
+      "loss": 0.55,
+      "step": 277
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.8250639386189257e-05,
+      "loss": 0.5672,
+      "step": 278
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 2.824296675191816e-05,
+      "loss": 0.285,
+      "step": 279
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.823529411764706e-05,
+      "loss": 0.2809,
+      "step": 280
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.8227621483375958e-05,
+      "loss": 0.5552,
+      "step": 281
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821994884910486e-05,
+      "loss": 0.563,
+      "step": 282
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 2.821227621483376e-05,
+      "loss": 0.5628,
+      "step": 283
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.820460358056266e-05,
+      "loss": 0.2548,
+      "step": 284
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.819693094629156e-05,
+      "loss": 0.871,
+      "step": 285
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818925831202046e-05,
+      "loss": 0.8797,
+      "step": 286
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 2.818158567774936e-05,
+      "loss": 0.2453,
+      "step": 287
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8173913043478262e-05,
+      "loss": 0.2464,
+      "step": 288
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.816624040920716e-05,
+      "loss": 0.3134,
+      "step": 289
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8158567774936062e-05,
+      "loss": 0.2431,
+      "step": 290
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.8150895140664964e-05,
+      "loss": 0.2308,
+      "step": 291
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8143222506393862e-05,
+      "loss": 0.2173,
+      "step": 292
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8135549872122764e-05,
+      "loss": 0.5665,
+      "step": 293
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8127877237851662e-05,
+      "loss": 0.5592,
+      "step": 294
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 2.8120204603580564e-05,
+      "loss": 0.473,
+      "step": 295
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8112531969309465e-05,
+      "loss": 0.9458,
+      "step": 296
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8104859335038363e-05,
+      "loss": 0.1879,
+      "step": 297
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 2.8097186700767265e-05,
+      "loss": 0.6797,
+      "step": 298
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8089514066496167e-05,
+      "loss": 0.7592,
+      "step": 299
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8081841432225065e-05,
+      "loss": 0.4109,
+      "step": 300
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8074168797953963e-05,
+      "loss": 0.5821,
+      "step": 301
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 2.8066496163682868e-05,
+      "loss": 0.2057,
+      "step": 302
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8058823529411766e-05,
+      "loss": 0.5322,
+      "step": 303
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8051150895140665e-05,
+      "loss": 0.5208,
+      "step": 304
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8043478260869566e-05,
+      "loss": 0.6799,
+      "step": 305
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 2.8035805626598468e-05,
+      "loss": 0.5894,
+      "step": 306
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8028132992327366e-05,
+      "loss": 0.4051,
+      "step": 307
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8020460358056268e-05,
+      "loss": 0.3695,
+      "step": 308
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8012787723785166e-05,
+      "loss": 0.4869,
+      "step": 309
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 2.8005115089514067e-05,
+      "loss": 0.6075,
+      "step": 310
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.799744245524297e-05,
+      "loss": 0.4245,
+      "step": 311
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7989769820971867e-05,
+      "loss": 0.5122,
+      "step": 312
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.7982097186700765e-05,
+      "loss": 0.452,
+      "step": 313
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 2.797442455242967e-05,
+      "loss": 0.9568,
+      "step": 314
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.796675191815857e-05,
+      "loss": 0.5846,
+      "step": 315
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.7959079283887467e-05,
+      "loss": 0.3743,
+      "step": 316
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.795140664961637e-05,
+      "loss": 0.1922,
+      "step": 317
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 2.794373401534527e-05,
+      "loss": 0.5303,
+      "step": 318
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.793606138107417e-05,
+      "loss": 0.7253,
+      "step": 319
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.792838874680307e-05,
+      "loss": 0.2794,
+      "step": 320
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.7920716112531968e-05,
+      "loss": 0.577,
+      "step": 321
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 2.791304347826087e-05,
+      "loss": 0.614,
+      "step": 322
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.790537084398977e-05,
+      "loss": 0.5882,
+      "step": 323
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789769820971867e-05,
+      "loss": 0.2503,
+      "step": 324
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.789002557544757e-05,
+      "loss": 0.4195,
+      "step": 325
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 2.7882352941176473e-05,
+      "loss": 0.2959,
+      "step": 326
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.787468030690537e-05,
+      "loss": 0.3732,
+      "step": 327
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.786700767263427e-05,
+      "loss": 0.5729,
+      "step": 328
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7859335038363174e-05,
+      "loss": 0.3179,
+      "step": 329
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 2.7851662404092073e-05,
+      "loss": 0.6823,
+      "step": 330
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.784398976982097e-05,
+      "loss": 0.2663,
+      "step": 331
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7836317135549872e-05,
+      "loss": 0.4436,
+      "step": 332
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7828644501278774e-05,
+      "loss": 0.3089,
+      "step": 333
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 2.7820971867007672e-05,
+      "loss": 0.5888,
+      "step": 334
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7813299232736574e-05,
+      "loss": 0.2115,
+      "step": 335
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7805626598465472e-05,
+      "loss": 0.4005,
+      "step": 336
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7797953964194374e-05,
+      "loss": 0.3097,
+      "step": 337
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 2.7790281329923275e-05,
+      "loss": 0.5847,
+      "step": 338
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7782608695652174e-05,
+      "loss": 1.0164,
+      "step": 339
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7774936061381072e-05,
+      "loss": 1.1321,
+      "step": 340
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7767263427109977e-05,
+      "loss": 0.592,
+      "step": 341
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 2.7759590792838875e-05,
+      "loss": 1.0191,
+      "step": 342
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7751918158567777e-05,
+      "loss": 0.9199,
+      "step": 343
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7744245524296675e-05,
+      "loss": 0.2874,
+      "step": 344
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7736572890025576e-05,
+      "loss": 0.2469,
+      "step": 345
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 2.7728900255754478e-05,
+      "loss": 0.8364,
+      "step": 346
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7721227621483376e-05,
+      "loss": 0.7365,
+      "step": 347
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7713554987212278e-05,
+      "loss": 0.5595,
+      "step": 348
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.770588235294118e-05,
+      "loss": 0.2546,
+      "step": 349
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 2.7698209718670078e-05,
+      "loss": 0.8183,
+      "step": 350
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7690537084398976e-05,
+      "loss": 0.5385,
+      "step": 351
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.768286445012788e-05,
+      "loss": 0.2763,
+      "step": 352
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.767519181585678e-05,
+      "loss": 0.3045,
+      "step": 353
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 2.7667519181585677e-05,
+      "loss": 0.3354,
+      "step": 354
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765984654731458e-05,
+      "loss": 0.9366,
+      "step": 355
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.765217391304348e-05,
+      "loss": 0.6212,
+      "step": 356
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.764450127877238e-05,
+      "loss": 0.1161,
+      "step": 357
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 2.763682864450128e-05,
+      "loss": 0.4929,
+      "step": 358
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762915601023018e-05,
+      "loss": 0.8167,
+      "step": 359
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.762148337595908e-05,
+      "loss": 0.2521,
+      "step": 360
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.7613810741687982e-05,
+      "loss": 0.4174,
+      "step": 361
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 2.760613810741688e-05,
+      "loss": 0.3822,
+      "step": 362
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.759846547314578e-05,
+      "loss": 0.5355,
+      "step": 363
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.7590792838874683e-05,
+      "loss": 0.561,
+      "step": 364
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.758312020460358e-05,
+      "loss": 0.5126,
+      "step": 365
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 2.757544757033248e-05,
+      "loss": 0.2301,
+      "step": 366
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.756777493606138e-05,
+      "loss": 0.1698,
+      "step": 367
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7560102301790283e-05,
+      "loss": 0.1634,
+      "step": 368
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.755242966751918e-05,
+      "loss": 0.5955,
+      "step": 369
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 2.7544757033248083e-05,
+      "loss": 0.1568,
+      "step": 370
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.753708439897698e-05,
+      "loss": 0.5167,
+      "step": 371
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7529411764705883e-05,
+      "loss": 1.0028,
+      "step": 372
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7521739130434784e-05,
+      "loss": 1.2567,
+      "step": 373
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 2.7514066496163682e-05,
+      "loss": 0.6094,
+      "step": 374
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7506393861892584e-05,
+      "loss": 0.6059,
+      "step": 375
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7498721227621486e-05,
+      "loss": 1.0593,
+      "step": 376
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7491048593350384e-05,
+      "loss": 0.1506,
+      "step": 377
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 2.7483375959079282e-05,
+      "loss": 1.5106,
+      "step": 378
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7475703324808187e-05,
+      "loss": 0.1827,
+      "step": 379
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7468030690537085e-05,
+      "loss": 1.4961,
+      "step": 380
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7460358056265984e-05,
+      "loss": 1.0456,
+      "step": 381
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.7452685421994885e-05,
+      "loss": 1.2063,
+      "step": 382
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7445012787723787e-05,
+      "loss": 0.5837,
+      "step": 383
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7437340153452685e-05,
+      "loss": 1.4536,
+      "step": 384
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7429667519181587e-05,
+      "loss": 0.5803,
+      "step": 385
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 2.7421994884910485e-05,
+      "loss": 0.5896,
+      "step": 386
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7414322250639386e-05,
+      "loss": 0.1822,
+      "step": 387
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7406649616368288e-05,
+      "loss": 1.7893,
+      "step": 388
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7398976982097186e-05,
+      "loss": 0.9723,
+      "step": 389
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 2.7391304347826085e-05,
+      "loss": 1.2815,
+      "step": 390
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.738363171355499e-05,
+      "loss": 0.5765,
+      "step": 391
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7375959079283888e-05,
+      "loss": 1.2429,
+      "step": 392
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7368286445012786e-05,
+      "loss": 0.1343,
+      "step": 393
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 2.7360613810741688e-05,
+      "loss": 0.2306,
+      "step": 394
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.735294117647059e-05,
+      "loss": 0.624,
+      "step": 395
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.734526854219949e-05,
+      "loss": 0.2371,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7348484848484849,
+      "eval_loss": 0.5804409980773926,
+      "eval_runtime": 57.3336,
+      "eval_samples_per_second": 6.907,
+      "eval_steps_per_second": 1.727,
+      "step": 396
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 2.733759590792839e-05,
+      "loss": 0.8989,
+      "step": 397
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.732992327365729e-05,
+      "loss": 0.5664,
+      "step": 398
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.7322250639386192e-05,
+      "loss": 0.8852,
+      "step": 399
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.731457800511509e-05,
+      "loss": 0.5661,
+      "step": 400
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 2.730690537084399e-05,
+      "loss": 0.5648,
+      "step": 401
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7299232736572894e-05,
+      "loss": 0.7333,
+      "step": 402
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7291560102301792e-05,
+      "loss": 0.3037,
+      "step": 403
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.728388746803069e-05,
+      "loss": 0.1463,
+      "step": 404
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 2.7276214833759592e-05,
+      "loss": 0.2702,
+      "step": 405
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7268542199488493e-05,
+      "loss": 0.2693,
+      "step": 406
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.726086956521739e-05,
+      "loss": 0.5647,
+      "step": 407
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.7253196930946293e-05,
+      "loss": 0.8507,
+      "step": 408
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 2.724552429667519e-05,
+      "loss": 0.8645,
+      "step": 409
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7237851662404093e-05,
+      "loss": 0.2649,
+      "step": 410
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7230179028132995e-05,
+      "loss": 0.6942,
+      "step": 411
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7222506393861893e-05,
+      "loss": 0.5656,
+      "step": 412
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.721483375959079e-05,
+      "loss": 0.5489,
+      "step": 413
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7207161125319696e-05,
+      "loss": 0.2652,
+      "step": 414
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7199488491048594e-05,
+      "loss": 0.8676,
+      "step": 415
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7191815856777493e-05,
+      "loss": 0.8621,
+      "step": 416
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 2.7184143222506394e-05,
+      "loss": 0.2641,
+      "step": 417
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7176470588235296e-05,
+      "loss": 0.859,
+      "step": 418
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7168797953964194e-05,
+      "loss": 0.6492,
+      "step": 419
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7161125319693096e-05,
+      "loss": 1.1494,
+      "step": 420
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 2.7153452685421994e-05,
+      "loss": 0.8278,
+      "step": 421
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7145780051150895e-05,
+      "loss": 0.764,
+      "step": 422
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7138107416879797e-05,
+      "loss": 0.564,
+      "step": 423
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7130434782608695e-05,
+      "loss": 0.8051,
+      "step": 424
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 2.7122762148337597e-05,
+      "loss": 0.3052,
+      "step": 425
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.71150895140665e-05,
+      "loss": 0.2945,
+      "step": 426
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7107416879795397e-05,
+      "loss": 0.4118,
+      "step": 427
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.7099744245524295e-05,
+      "loss": 0.559,
+      "step": 428
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 2.70920716112532e-05,
+      "loss": 0.5593,
+      "step": 429
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7084398976982098e-05,
+      "loss": 0.564,
+      "step": 430
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7076726342710996e-05,
+      "loss": 0.4454,
+      "step": 431
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.7069053708439898e-05,
+      "loss": 0.4381,
+      "step": 432
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 2.70613810741688e-05,
+      "loss": 0.3718,
+      "step": 433
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7053708439897698e-05,
+      "loss": 0.6627,
+      "step": 434
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70460358056266e-05,
+      "loss": 0.4004,
+      "step": 435
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.7038363171355498e-05,
+      "loss": 0.3855,
+      "step": 436
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 2.70306905370844e-05,
+      "loss": 0.4799,
+      "step": 437
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70230179028133e-05,
+      "loss": 0.6714,
+      "step": 438
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.70153452685422e-05,
+      "loss": 0.622,
+      "step": 439
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7007672634271097e-05,
+      "loss": 0.3916,
+      "step": 440
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 2.7000000000000002e-05,
+      "loss": 0.5487,
+      "step": 441
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69923273657289e-05,
+      "loss": 0.376,
+      "step": 442
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69846547314578e-05,
+      "loss": 0.3616,
+      "step": 443
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.69769820971867e-05,
+      "loss": 0.8747,
+      "step": 444
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 2.6969309462915602e-05,
+      "loss": 0.283,
+      "step": 445
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.69616368286445e-05,
+      "loss": 0.3988,
+      "step": 446
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6953964194373402e-05,
+      "loss": 0.4569,
+      "step": 447
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6946291560102303e-05,
+      "loss": 0.2005,
+      "step": 448
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 2.6938618925831205e-05,
+      "loss": 0.3087,
+      "step": 449
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6930946291560103e-05,
+      "loss": 0.5814,
+      "step": 450
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6923273657289e-05,
+      "loss": 0.2995,
+      "step": 451
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6915601023017907e-05,
+      "loss": 0.7806,
+      "step": 452
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 2.6907928388746805e-05,
+      "loss": 0.2666,
+      "step": 453
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6900255754475703e-05,
+      "loss": 0.2571,
+      "step": 454
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6892583120204605e-05,
+      "loss": 0.5931,
+      "step": 455
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6884910485933506e-05,
+      "loss": 0.475,
+      "step": 456
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.6877237851662404e-05,
+      "loss": 0.2372,
+      "step": 457
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6869565217391306e-05,
+      "loss": 0.1577,
+      "step": 458
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6861892583120204e-05,
+      "loss": 0.1953,
+      "step": 459
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6854219948849106e-05,
+      "loss": 0.1931,
+      "step": 460
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 2.6846547314578008e-05,
+      "loss": 0.1829,
+      "step": 461
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6838874680306906e-05,
+      "loss": 0.1601,
+      "step": 462
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6831202046035804e-05,
+      "loss": 0.2505,
+      "step": 463
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.682352941176471e-05,
+      "loss": 0.181,
+      "step": 464
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 2.6815856777493607e-05,
+      "loss": 0.5978,
+      "step": 465
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6808184143222505e-05,
+      "loss": 0.5458,
+      "step": 466
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6800511508951407e-05,
+      "loss": 0.1289,
+      "step": 467
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.679283887468031e-05,
+      "loss": 0.4383,
+      "step": 468
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 2.6785166240409207e-05,
+      "loss": 0.5892,
+      "step": 469
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.677749360613811e-05,
+      "loss": 1.3565,
+      "step": 470
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.676982097186701e-05,
+      "loss": 0.7267,
+      "step": 471
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.6762148337595908e-05,
+      "loss": 0.1354,
+      "step": 472
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 2.675447570332481e-05,
+      "loss": 0.9291,
+      "step": 473
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.6746803069053708e-05,
+      "loss": 0.5265,
+      "step": 474
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673913043478261e-05,
+      "loss": 0.4841,
+      "step": 475
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.673145780051151e-05,
+      "loss": 1.2391,
+      "step": 476
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 2.672378516624041e-05,
+      "loss": 0.4924,
+      "step": 477
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6716112531969308e-05,
+      "loss": 1.0076,
+      "step": 478
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.6708439897698213e-05,
+      "loss": 0.6314,
+      "step": 479
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.670076726342711e-05,
+      "loss": 0.4237,
+      "step": 480
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 2.669309462915601e-05,
+      "loss": 0.1077,
+      "step": 481
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.668542199488491e-05,
+      "loss": 0.3038,
+      "step": 482
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6677749360613812e-05,
+      "loss": 0.6081,
+      "step": 483
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.667007672634271e-05,
+      "loss": 0.1078,
+      "step": 484
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 2.6662404092071612e-05,
+      "loss": 0.4296,
+      "step": 485
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.665473145780051e-05,
+      "loss": 0.3981,
+      "step": 486
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6647058823529412e-05,
+      "loss": 0.5853,
+      "step": 487
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6639386189258314e-05,
+      "loss": 0.802,
+      "step": 488
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 2.6631713554987212e-05,
+      "loss": 0.4576,
+      "step": 489
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.662404092071611e-05,
+      "loss": 0.2682,
+      "step": 490
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6616368286445015e-05,
+      "loss": 0.1991,
+      "step": 491
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.6608695652173913e-05,
+      "loss": 0.1409,
+      "step": 492
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 2.660102301790281e-05,
+      "loss": 0.3869,
+      "step": 493
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6593350383631713e-05,
+      "loss": 0.5301,
+      "step": 494
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6585677749360615e-05,
+      "loss": 0.8116,
+      "step": 495
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 2.6578005115089513e-05,
+      "loss": 0.0984,
+      "step": 496
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6570332480818415e-05,
+      "loss": 0.7773,
+      "step": 497
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6562659846547316e-05,
+      "loss": 1.3522,
+      "step": 498
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6554987212276215e-05,
+      "loss": 0.0862,
+      "step": 499
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 2.6547314578005116e-05,
+      "loss": 0.0894,
+      "step": 500
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6539641943734014e-05,
+      "loss": 0.0885,
+      "step": 501
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.653196930946292e-05,
+      "loss": 0.3031,
+      "step": 502
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6524296675191818e-05,
+      "loss": 0.187,
+      "step": 503
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 2.6516624040920716e-05,
+      "loss": 1.3063,
+      "step": 504
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6508951406649617e-05,
+      "loss": 0.6847,
+      "step": 505
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.650127877237852e-05,
+      "loss": 0.078,
+      "step": 506
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.6493606138107417e-05,
+      "loss": 0.1265,
+      "step": 507
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 2.648593350383632e-05,
+      "loss": 0.0804,
+      "step": 508
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.6478260869565217e-05,
+      "loss": 0.6976,
+      "step": 509
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.647058823529412e-05,
+      "loss": 0.7977,
+      "step": 510
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.646291560102302e-05,
+      "loss": 0.413,
+      "step": 511
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 2.645524296675192e-05,
+      "loss": 0.6944,
+      "step": 512
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6447570332480817e-05,
+      "loss": 0.7177,
+      "step": 513
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6439897698209722e-05,
+      "loss": 0.0581,
+      "step": 514
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.643222506393862e-05,
+      "loss": 0.9308,
+      "step": 515
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 2.6424552429667518e-05,
+      "loss": 0.2518,
+      "step": 516
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.641687979539642e-05,
+      "loss": 0.1014,
+      "step": 517
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640920716112532e-05,
+      "loss": 0.1641,
+      "step": 518
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.640153452685422e-05,
+      "loss": 0.101,
+      "step": 519
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 2.639386189258312e-05,
+      "loss": 0.1097,
+      "step": 520
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6386189258312023e-05,
+      "loss": 0.987,
+      "step": 521
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.637851662404092e-05,
+      "loss": 0.6465,
+      "step": 522
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.6370843989769823e-05,
+      "loss": 0.0914,
+      "step": 523
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 2.636317135549872e-05,
+      "loss": 0.139,
+      "step": 524
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6355498721227623e-05,
+      "loss": 0.6455,
+      "step": 525
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6347826086956524e-05,
+      "loss": 0.204,
+      "step": 526
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.6340153452685422e-05,
+      "loss": 0.1269,
+      "step": 527
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 2.633248081841432e-05,
+      "loss": 0.2214,
+      "step": 528
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6324808184143226e-05,
+      "loss": 0.0736,
+      "step": 529
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6317135549872124e-05,
+      "loss": 0.0714,
+      "step": 530
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6309462915601022e-05,
+      "loss": 0.1388,
+      "step": 531
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 2.6301790281329924e-05,
+      "loss": 0.7758,
+      "step": 532
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6294117647058825e-05,
+      "loss": 0.3388,
+      "step": 533
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6286445012787724e-05,
+      "loss": 1.1831,
+      "step": 534
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6278772378516625e-05,
+      "loss": 0.1166,
+      "step": 535
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 2.6271099744245523e-05,
+      "loss": 0.2389,
+      "step": 536
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6263427109974425e-05,
+      "loss": 0.1622,
+      "step": 537
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6255754475703327e-05,
+      "loss": 0.1015,
+      "step": 538
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6248081841432225e-05,
+      "loss": 0.0658,
+      "step": 539
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 2.6240409207161123e-05,
+      "loss": 0.0954,
+      "step": 540
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6232736572890028e-05,
+      "loss": 0.1182,
+      "step": 541
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6225063938618926e-05,
+      "loss": 0.0605,
+      "step": 542
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6217391304347824e-05,
+      "loss": 0.9265,
+      "step": 543
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 2.6209718670076726e-05,
+      "loss": 1.2994,
+      "step": 544
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6202046035805628e-05,
+      "loss": 0.081,
+      "step": 545
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6194373401534526e-05,
+      "loss": 0.5962,
+      "step": 546
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.6186700767263428e-05,
+      "loss": 0.0665,
+      "step": 547
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 2.617902813299233e-05,
+      "loss": 0.0655,
+      "step": 548
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6171355498721227e-05,
+      "loss": 0.0642,
+      "step": 549
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.616368286445013e-05,
+      "loss": 0.0834,
+      "step": 550
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6156010230179027e-05,
+      "loss": 0.047,
+      "step": 551
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.6148337595907932e-05,
+      "loss": 0.0642,
+      "step": 552
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.614066496163683e-05,
+      "loss": 0.0594,
+      "step": 553
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.613299232736573e-05,
+      "loss": 0.0573,
+      "step": 554
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.612531969309463e-05,
+      "loss": 0.1101,
+      "step": 555
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.6117647058823532e-05,
+      "loss": 0.5121,
+      "step": 556
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.610997442455243e-05,
+      "loss": 0.0374,
+      "step": 557
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.6102301790281332e-05,
+      "loss": 0.0957,
+      "step": 558
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.609462915601023e-05,
+      "loss": 0.0488,
+      "step": 559
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 2.608695652173913e-05,
+      "loss": 0.0482,
+      "step": 560
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6079283887468033e-05,
+      "loss": 1.2041,
+      "step": 561
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.607161125319693e-05,
+      "loss": 0.8833,
+      "step": 562
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.606393861892583e-05,
+      "loss": 0.1544,
+      "step": 563
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.6056265984654735e-05,
+      "loss": 0.046,
+      "step": 564
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6048593350383633e-05,
+      "loss": 0.9073,
+      "step": 565
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.604092071611253e-05,
+      "loss": 0.052,
+      "step": 566
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6033248081841433e-05,
+      "loss": 0.0308,
+      "step": 567
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 2.6025575447570334e-05,
+      "loss": 0.7032,
+      "step": 568
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6017902813299232e-05,
+      "loss": 0.6947,
+      "step": 569
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6010230179028134e-05,
+      "loss": 1.5603,
+      "step": 570
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.6002557544757036e-05,
+      "loss": 0.8836,
+      "step": 571
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 2.5994884910485934e-05,
+      "loss": 1.7439,
+      "step": 572
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5987212276214836e-05,
+      "loss": 0.0313,
+      "step": 573
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5979539641943734e-05,
+      "loss": 0.0317,
+      "step": 574
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5971867007672635e-05,
+      "loss": 0.0325,
+      "step": 575
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.5964194373401537e-05,
+      "loss": 0.5,
+      "step": 576
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5956521739130435e-05,
+      "loss": 0.0675,
+      "step": 577
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5948849104859333e-05,
+      "loss": 0.6364,
+      "step": 578
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.594117647058824e-05,
+      "loss": 0.0696,
+      "step": 579
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 2.5933503836317137e-05,
+      "loss": 0.6195,
+      "step": 580
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5925831202046035e-05,
+      "loss": 0.5796,
+      "step": 581
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5918158567774937e-05,
+      "loss": 0.0832,
+      "step": 582
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5910485933503838e-05,
+      "loss": 0.0794,
+      "step": 583
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 2.5902813299232736e-05,
+      "loss": 0.5851,
+      "step": 584
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5895140664961638e-05,
+      "loss": 0.104,
+      "step": 585
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5887468030690536e-05,
+      "loss": 1.0492,
+      "step": 586
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.5879795396419438e-05,
+      "loss": 0.0645,
+      "step": 587
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 2.587212276214834e-05,
+      "loss": 0.514,
+      "step": 588
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5864450127877238e-05,
+      "loss": 0.0701,
+      "step": 589
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.5856777493606136e-05,
+      "loss": 0.5409,
+      "step": 590
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584910485933504e-05,
+      "loss": 0.0536,
+      "step": 591
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.584143222506394e-05,
+      "loss": 0.1728,
+      "step": 592
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.5833759590792837e-05,
+      "loss": 0.0845,
+      "step": 593
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.582608695652174e-05,
+      "loss": 0.2217,
+      "step": 594
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 2.581841432225064e-05,
+      "loss": 0.1397,
+      "step": 595
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.581074168797954e-05,
+      "loss": 0.4119,
+      "step": 596
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.580306905370844e-05,
+      "loss": 0.3638,
+      "step": 597
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.5795396419437342e-05,
+      "loss": 0.0781,
+      "step": 598
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 2.578772378516624e-05,
+      "loss": 0.5926,
+      "step": 599
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5780051150895142e-05,
+      "loss": 0.1378,
+      "step": 600
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.577237851662404e-05,
+      "loss": 0.142,
+      "step": 601
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.576470588235294e-05,
+      "loss": 0.1722,
+      "step": 602
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 2.5757033248081843e-05,
+      "loss": 0.1529,
+      "step": 603
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.574936061381074e-05,
+      "loss": 1.6083,
+      "step": 604
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5741687979539643e-05,
+      "loss": 1.8267,
+      "step": 605
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5734015345268545e-05,
+      "loss": 0.0535,
+      "step": 606
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.5726342710997443e-05,
+      "loss": 0.9945,
+      "step": 607
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5718670076726345e-05,
+      "loss": 0.9133,
+      "step": 608
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5710997442455243e-05,
+      "loss": 0.5295,
+      "step": 609
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5703324808184144e-05,
+      "loss": 0.058,
+      "step": 610
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 2.5695652173913046e-05,
+      "loss": 0.4789,
+      "step": 611
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5687979539641944e-05,
+      "loss": 0.0572,
+      "step": 612
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5680306905370842e-05,
+      "loss": 0.0897,
+      "step": 613
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5672634271099747e-05,
+      "loss": 0.056,
+      "step": 614
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 2.5664961636828646e-05,
+      "loss": 0.0278,
+      "step": 615
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5657289002557544e-05,
+      "loss": 0.9947,
+      "step": 616
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5649616368286445e-05,
+      "loss": 0.0627,
+      "step": 617
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5641943734015347e-05,
+      "loss": 0.9335,
+      "step": 618
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 2.5634271099744245e-05,
+      "loss": 0.9793,
+      "step": 619
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5626598465473147e-05,
+      "loss": 1.8128,
+      "step": 620
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.561892583120205e-05,
+      "loss": 1.8015,
+      "step": 621
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.5611253196930947e-05,
+      "loss": 0.9259,
+      "step": 622
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.560358056265985e-05,
+      "loss": 0.9618,
+      "step": 623
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5595907928388747e-05,
+      "loss": 0.0808,
+      "step": 624
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5588235294117648e-05,
+      "loss": 1.7161,
+      "step": 625
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.558056265984655e-05,
+      "loss": 0.8879,
+      "step": 626
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 2.5572890025575448e-05,
+      "loss": 0.043,
+      "step": 627
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5565217391304346e-05,
+      "loss": 0.9449,
+      "step": 628
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.555754475703325e-05,
+      "loss": 0.0624,
+      "step": 629
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.554987212276215e-05,
+      "loss": 0.0924,
+      "step": 630
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.5542199488491048e-05,
+      "loss": 0.0715,
+      "step": 631
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.553452685421995e-05,
+      "loss": 0.4995,
+      "step": 632
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.552685421994885e-05,
+      "loss": 0.5576,
+      "step": 633
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551918158567775e-05,
+      "loss": 0.06,
+      "step": 634
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 2.551150895140665e-05,
+      "loss": 0.0873,
+      "step": 635
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.550383631713555e-05,
+      "loss": 0.6868,
+      "step": 636
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.549616368286445e-05,
+      "loss": 0.8487,
+      "step": 637
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.5488491048593352e-05,
+      "loss": 0.5641,
+      "step": 638
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.548081841432225e-05,
+      "loss": 1.1343,
+      "step": 639
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.547314578005115e-05,
+      "loss": 0.0614,
+      "step": 640
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5465473145780054e-05,
+      "loss": 0.9414,
+      "step": 641
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.5457800511508952e-05,
+      "loss": 0.9226,
+      "step": 642
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 2.545012787723785e-05,
+      "loss": 0.3323,
+      "step": 643
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5442455242966752e-05,
+      "loss": 0.9315,
+      "step": 644
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5434782608695653e-05,
+      "loss": 0.5539,
+      "step": 645
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.542710997442455e-05,
+      "loss": 0.3376,
+      "step": 646
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 2.5419437340153453e-05,
+      "loss": 0.0694,
+      "step": 647
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5411764705882355e-05,
+      "loss": 0.0706,
+      "step": 648
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5404092071611253e-05,
+      "loss": 0.105,
+      "step": 649
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5396419437340155e-05,
+      "loss": 0.912,
+      "step": 650
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.5388746803069053e-05,
+      "loss": 0.0998,
+      "step": 651
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5381074168797954e-05,
+      "loss": 0.2247,
+      "step": 652
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5373401534526856e-05,
+      "loss": 0.2869,
+      "step": 653
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5365728900255754e-05,
+      "loss": 0.532,
+      "step": 654
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.5358056265984653e-05,
+      "loss": 0.4723,
+      "step": 655
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5350383631713558e-05,
+      "loss": 0.1178,
+      "step": 656
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5342710997442456e-05,
+      "loss": 0.0566,
+      "step": 657
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5335038363171357e-05,
+      "loss": 0.059,
+      "step": 658
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 2.5327365728900256e-05,
+      "loss": 0.0881,
+      "step": 659
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5319693094629157e-05,
+      "loss": 0.5762,
+      "step": 660
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.531202046035806e-05,
+      "loss": 0.0499,
+      "step": 661
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5304347826086957e-05,
+      "loss": 0.087,
+      "step": 662
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 2.5296675191815855e-05,
+      "loss": 1.0719,
+      "step": 663
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528900255754476e-05,
+      "loss": 0.6158,
+      "step": 664
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.528132992327366e-05,
+      "loss": 0.6117,
+      "step": 665
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5273657289002557e-05,
+      "loss": 0.0515,
+      "step": 666
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 2.5265984654731458e-05,
+      "loss": 0.0608,
+      "step": 667
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.525831202046036e-05,
+      "loss": 0.0313,
+      "step": 668
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.5250639386189258e-05,
+      "loss": 0.1449,
+      "step": 669
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.524296675191816e-05,
+      "loss": 0.9883,
+      "step": 670
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.523529411764706e-05,
+      "loss": 0.0859,
+      "step": 671
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.522762148337596e-05,
+      "loss": 0.0561,
+      "step": 672
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521994884910486e-05,
+      "loss": 0.0926,
+      "step": 673
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.521227621483376e-05,
+      "loss": 0.1101,
+      "step": 674
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 2.520460358056266e-05,
+      "loss": 0.068,
+      "step": 675
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5196930946291563e-05,
+      "loss": 0.0773,
+      "step": 676
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518925831202046e-05,
+      "loss": 0.0508,
+      "step": 677
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.518158567774936e-05,
+      "loss": 0.0375,
+      "step": 678
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 2.5173913043478264e-05,
+      "loss": 0.93,
+      "step": 679
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5166240409207162e-05,
+      "loss": 0.1507,
+      "step": 680
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.515856777493606e-05,
+      "loss": 0.0917,
+      "step": 681
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5150895140664962e-05,
+      "loss": 0.97,
+      "step": 682
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 2.5143222506393864e-05,
+      "loss": 0.5019,
+      "step": 683
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5135549872122762e-05,
+      "loss": 0.0849,
+      "step": 684
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5127877237851664e-05,
+      "loss": 1.2316,
+      "step": 685
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5120204603580562e-05,
+      "loss": 0.0252,
+      "step": 686
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 2.5112531969309463e-05,
+      "loss": 1.0543,
+      "step": 687
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5104859335038365e-05,
+      "loss": 0.0506,
+      "step": 688
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5097186700767263e-05,
+      "loss": 0.0252,
+      "step": 689
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.508951406649616e-05,
+      "loss": 0.6297,
+      "step": 690
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.5081841432225066e-05,
+      "loss": 0.4497,
+      "step": 691
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5074168797953965e-05,
+      "loss": 0.0516,
+      "step": 692
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5066496163682863e-05,
+      "loss": 0.5501,
+      "step": 693
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.5058823529411768e-05,
+      "loss": 0.5621,
+      "step": 694
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5051150895140666e-05,
+      "loss": 0.0577,
+      "step": 695
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5043478260869564e-05,
+      "loss": 0.0271,
+      "step": 696
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5035805626598466e-05,
+      "loss": 0.9496,
+      "step": 697
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 2.5028132992327368e-05,
+      "loss": 0.0877,
+      "step": 698
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5020460358056266e-05,
+      "loss": 0.0523,
+      "step": 699
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5012787723785167e-05,
+      "loss": 0.0577,
+      "step": 700
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.5005115089514066e-05,
+      "loss": 0.0522,
+      "step": 701
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 2.4997442455242967e-05,
+      "loss": 0.0517,
+      "step": 702
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.498976982097187e-05,
+      "loss": 0.6335,
+      "step": 703
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4982097186700767e-05,
+      "loss": 0.05,
+      "step": 704
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.4974424552429665e-05,
+      "loss": 0.0753,
+      "step": 705
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.496675191815857e-05,
+      "loss": 0.0227,
+      "step": 706
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.495907928388747e-05,
+      "loss": 0.0755,
+      "step": 707
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.4951406649616367e-05,
+      "loss": 0.0215,
+      "step": 708
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.494373401534527e-05,
+      "loss": 0.0451,
+      "step": 709
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.493606138107417e-05,
+      "loss": 0.0343,
+      "step": 710
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492838874680307e-05,
+      "loss": 0.0836,
+      "step": 711
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.492071611253197e-05,
+      "loss": 0.5287,
+      "step": 712
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4913043478260868e-05,
+      "loss": 0.0686,
+      "step": 713
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 2.4905370843989773e-05,
+      "loss": 0.0211,
+      "step": 714
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489769820971867e-05,
+      "loss": 0.5932,
+      "step": 715
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.489002557544757e-05,
+      "loss": 0.0371,
+      "step": 716
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.488235294117647e-05,
+      "loss": 1.0137,
+      "step": 717
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 2.4874680306905373e-05,
+      "loss": 0.9928,
+      "step": 718
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.486700767263427e-05,
+      "loss": 0.7431,
+      "step": 719
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4859335038363173e-05,
+      "loss": 0.0207,
+      "step": 720
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4851662404092074e-05,
+      "loss": 0.0681,
+      "step": 721
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.4843989769820972e-05,
+      "loss": 0.0306,
+      "step": 722
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4836317135549874e-05,
+      "loss": 0.0204,
+      "step": 723
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4828644501278772e-05,
+      "loss": 0.0344,
+      "step": 724
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4820971867007674e-05,
+      "loss": 0.0345,
+      "step": 725
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 2.4813299232736575e-05,
+      "loss": 0.9275,
+      "step": 726
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4805626598465474e-05,
+      "loss": 0.0196,
+      "step": 727
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4797953964194372e-05,
+      "loss": 1.0137,
+      "step": 728
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4790281329923277e-05,
+      "loss": 0.0318,
+      "step": 729
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 2.4782608695652175e-05,
+      "loss": 0.0688,
+      "step": 730
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4774936061381073e-05,
+      "loss": 0.0314,
+      "step": 731
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4767263427109975e-05,
+      "loss": 0.7135,
+      "step": 732
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4759590792838877e-05,
+      "loss": 0.0299,
+      "step": 733
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 2.4751918158567775e-05,
+      "loss": 1.0127,
+      "step": 734
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4744245524296676e-05,
+      "loss": 0.0304,
+      "step": 735
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4736572890025575e-05,
+      "loss": 1.4618,
+      "step": 736
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4728900255754476e-05,
+      "loss": 0.7167,
+      "step": 737
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.4721227621483378e-05,
+      "loss": 0.0307,
+      "step": 738
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4713554987212276e-05,
+      "loss": 0.052,
+      "step": 739
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4705882352941174e-05,
+      "loss": 0.778,
+      "step": 740
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.469820971867008e-05,
+      "loss": 0.5482,
+      "step": 741
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 2.4690537084398978e-05,
+      "loss": 1.4314,
+      "step": 742
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4682864450127876e-05,
+      "loss": 0.0422,
+      "step": 743
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.467519181585678e-05,
+      "loss": 0.6614,
+      "step": 744
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.466751918158568e-05,
+      "loss": 0.0312,
+      "step": 745
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 2.4659846547314577e-05,
+      "loss": 0.043,
+      "step": 746
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.465217391304348e-05,
+      "loss": 0.0199,
+      "step": 747
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.464450127877238e-05,
+      "loss": 0.0832,
+      "step": 748
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.463682864450128e-05,
+      "loss": 0.0312,
+      "step": 749
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 2.462915601023018e-05,
+      "loss": 0.0309,
+      "step": 750
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.462148337595908e-05,
+      "loss": 0.0193,
+      "step": 751
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.461381074168798e-05,
+      "loss": 0.0446,
+      "step": 752
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.4606138107416882e-05,
+      "loss": 0.9993,
+      "step": 753
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.459846547314578e-05,
+      "loss": 0.016,
+      "step": 754
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4590792838874678e-05,
+      "loss": 0.0433,
+      "step": 755
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.4583120204603583e-05,
+      "loss": 0.2614,
+      "step": 756
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.457544757033248e-05,
+      "loss": 0.018,
+      "step": 757
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 2.456777493606138e-05,
+      "loss": 0.6676,
+      "step": 758
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.456010230179028e-05,
+      "loss": 0.0599,
+      "step": 759
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4552429667519183e-05,
+      "loss": 0.0338,
+      "step": 760
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.454475703324808e-05,
+      "loss": 0.0268,
+      "step": 761
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 2.4537084398976983e-05,
+      "loss": 0.0262,
+      "step": 762
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.452941176470588e-05,
+      "loss": 0.0252,
+      "step": 763
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4521739130434786e-05,
+      "loss": 0.0377,
+      "step": 764
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4514066496163684e-05,
+      "loss": 1.4866,
+      "step": 765
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 2.4506393861892582e-05,
+      "loss": 1.0524,
+      "step": 766
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4498721227621484e-05,
+      "loss": 0.0328,
+      "step": 767
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4491048593350386e-05,
+      "loss": 0.024,
+      "step": 768
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4483375959079284e-05,
+      "loss": 0.022,
+      "step": 769
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 2.4475703324808185e-05,
+      "loss": 0.0245,
+      "step": 770
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4468030690537087e-05,
+      "loss": 1.0107,
+      "step": 771
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4460358056265985e-05,
+      "loss": 0.0336,
+      "step": 772
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4452685421994887e-05,
+      "loss": 0.0391,
+      "step": 773
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 2.4445012787723785e-05,
+      "loss": 0.0303,
+      "step": 774
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4437340153452687e-05,
+      "loss": 0.0443,
+      "step": 775
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4429667519181588e-05,
+      "loss": 0.0292,
+      "step": 776
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4421994884910487e-05,
+      "loss": 0.0284,
+      "step": 777
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 2.4414322250639385e-05,
+      "loss": 1.5985,
+      "step": 778
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.440664961636829e-05,
+      "loss": 0.0114,
+      "step": 779
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4398976982097188e-05,
+      "loss": 0.0342,
+      "step": 780
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4391304347826086e-05,
+      "loss": 1.8501,
+      "step": 781
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 2.4383631713554988e-05,
+      "loss": 0.5375,
+      "step": 782
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.437595907928389e-05,
+      "loss": 1.589,
+      "step": 783
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4368286445012788e-05,
+      "loss": 0.0221,
+      "step": 784
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.436061381074169e-05,
+      "loss": 0.0453,
+      "step": 785
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 2.4352941176470587e-05,
+      "loss": 0.796,
+      "step": 786
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.434526854219949e-05,
+      "loss": 0.1249,
+      "step": 787
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.433759590792839e-05,
+      "loss": 0.7963,
+      "step": 788
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.432992327365729e-05,
+      "loss": 0.4231,
+      "step": 789
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 2.4322250639386187e-05,
+      "loss": 0.0247,
+      "step": 790
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.4314578005115092e-05,
+      "loss": 0.0295,
+      "step": 791
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 2.430690537084399e-05,
+      "loss": 0.0249,
+      "step": 792
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9065656565656566,
+      "eval_loss": 0.3537317216396332,
+      "eval_runtime": 58.1226,
+      "eval_samples_per_second": 6.813,
+      "eval_steps_per_second": 1.703,
+      "step": 792
+    }
+  ],
+  "max_steps": 3960,
+  "num_train_epochs": 10,
+  "total_flos": 2.874295093824e+17,
+  "trial_name": null,
+  "trial_params": null
+}
diff --git a/checkpoint-792/training_args.bin b/checkpoint-792/training_args.bin
new file mode 100644
index 0000000000000000000000000000000000000000..32901f899c91066c808eae9dd019400bfdd17498
--- /dev/null
+++ b/checkpoint-792/training_args.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db14439c369bfe9c89f64ff38f0b2d3fcc53911fc2c0b12a3a7d04656e7384a
+size 3503
diff --git a/config.json b/config.json
new file mode 100644
index 0000000000000000000000000000000000000000..cb8d52e14263c50fd53dca42b5050e3235115c32
--- /dev/null
+++ b/config.json
@@ -0,0 +1,113 @@
+{
+  "_name_or_path": "facebook/wav2vec2-base-960h",
+  "activation_dropout": 0.1,
+  "adapter_kernel_size": 3,
+  "adapter_stride": 2,
+  "add_adapter": false,
+  "apply_spec_augment": true,
+  "architectures": [
+    "Wav2Vec2ForSequenceClassification"
+  ],
+  "attention_dropout": 0.1,
+  "bos_token_id": 1,
+  "classifier_proj_size": 256,
+  "codevector_dim": 256,
+  "contrastive_logits_temperature": 0.1,
+  "conv_bias": false,
+  "conv_dim": [
+    512,
+    512,
+    512,
+    512,
+    512,
+    512,
+    512
+  ],
+  "conv_kernel": [
+    10,
+    3,
+    3,
+    3,
+    3,
+    2,
+    2
+  ],
+  "conv_stride": [
+    5,
+    2,
+    2,
+    2,
+    2,
+    2,
+    2
+  ],
+  "ctc_loss_reduction": "sum",
+  "ctc_zero_infinity": false,
+  "diversity_loss_weight": 0.1,
+  "do_stable_layer_norm": false,
+  "eos_token_id": 2,
+  "feat_extract_activation": "gelu",
+  "feat_extract_dropout": 0.0,
+  "feat_extract_norm": "group",
+  "feat_proj_dropout": 0.1,
+  "feat_quantizer_dropout": 0.0,
+  "final_dropout": 0.1,
+  "gradient_checkpointing": false,
+  "hidden_act": "gelu",
+  "hidden_dropout": 0.1,
+  "hidden_dropout_prob": 0.1,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Foreign",
+    "1": "Native"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "layerdrop": 0.1,
+  "mask_feature_length": 10,
+  "mask_feature_min_masks": 0,
+  "mask_feature_prob": 0.0,
+  "mask_time_length": 10,
+  "mask_time_min_masks": 2,
+  "mask_time_prob": 0.05,
+  "model_type": "wav2vec2",
+  "num_adapter_layers": 3,
+  "num_attention_heads": 12,
+  "num_codevector_groups": 2,
+  "num_codevectors_per_group": 320,
+  "num_conv_pos_embedding_groups": 16,
+  "num_conv_pos_embeddings": 128,
+  "num_feat_extract_layers": 7,
+  "num_hidden_layers": 12,
+  "num_negatives": 100,
+  "output_hidden_size": 768,
+  "pad_token_id": 0,
+  "proj_codevector_dim": 256,
+  "tdnn_dilation": [
+    1,
+    2,
+    3,
+    1,
+    1
+  ],
+  "tdnn_dim": [
+    512,
+    512,
+    512,
+    512,
+    1500
+  ],
+  "tdnn_kernel": [
+    5,
+    3,
+    3,
+    1,
+    1
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.28.1",
+  "use_weighted_layer_sum": false,
+  "vocab_size": 32,
+  "xvector_output_dim": 512
+}
diff --git a/preprocessor_config.json b/preprocessor_config.json
new file mode 100644
index 0000000000000000000000000000000000000000..a0b7227fc1d916e469b14f6c154ad6dfea1e6891
--- /dev/null
+++ b/preprocessor_config.json
@@ -0,0 +1,9 @@
+{
+  "do_normalize": true,
+  "feature_extractor_type": "Wav2Vec2FeatureExtractor",
+  "feature_size": 1,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": false,
+  "sampling_rate": 16000
+}
diff --git a/pytorch_model.bin b/pytorch_model.bin
new file mode 100644
index 0000000000000000000000000000000000000000..1fef170ef7e7594af290734de461df4fe714fad3
--- /dev/null
+++ b/pytorch_model.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:415f32f77f5fa3a5892d605f9d3333aa719e7abc96aeb959f4818f6f245a6837
+size 378348563
diff --git a/training_args.bin b/training_args.bin
new file mode 100644
index 0000000000000000000000000000000000000000..32901f899c91066c808eae9dd019400bfdd17498
--- /dev/null
+++ b/training_args.bin
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7db14439c369bfe9c89f64ff38f0b2d3fcc53911fc2c0b12a3a7d04656e7384a
+size 3503