Upload folder using huggingface_hub

checkpoint-1000/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2d72464a6406b03c1da43839501d9043b8ea4a12be7200cc45d05a22455b495b
+size 28967217802

checkpoint-1000/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:689f822f247861108f6b9ec5a032cbaf669ff8a68e9350dc2e902ca0a5f4f1bb
+size 14483616658

checkpoint-1000/pytorch_model_fsdp.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c990feb42bebb1bebb4eec51d4c5ea1f618859009c18a11716e48a657be1cc6b
+size 14483618133

checkpoint-1000/rng_state_0.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8524ba7cd3c1a3f9f34b7f51ba7ad4fd655cd843eb61a115108c0e6586fc3e0e
+size 15984

checkpoint-1000/rng_state_1.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e6d1436f51617661447b79bdc293f26b7a20c8e2dd7962c9703d770922f3b328
+size 15984

checkpoint-1000/rng_state_2.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:65129f8c151e53786769e07beb47a0ea3f78b4022a7f41e08673926b48039ef4
+size 15984

checkpoint-1000/rng_state_3.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7c615965325fbaa1925c3472d69a9ae7b1c3cb4e50596b5666a223a59d7c80f6
+size 15984

checkpoint-1000/rng_state_4.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2b7c7f4e7c56a3a722c9f37750f5a9e2a1e05d0f673db06e40b13e008a45f1de
+size 15984

checkpoint-1000/rng_state_5.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b0439923a2e24591a109b92f0d3dd6a22bae1a2b250879814cb7a04bf4f89a11
+size 15984

checkpoint-1000/rng_state_6.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a6ad1a95779c0860714d577c90821bd26be80105f919f9a98b0574a4fd3365e8
+size 15984

checkpoint-1000/rng_state_7.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:78c0e773004c4053fb08f10db63acc7c9d55a1d2c6c52fafa6cd5e67cbebb729
+size 15984

checkpoint-1000/scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a2895b0e3f1c28bf5c8fcf1e694d1dd1665857bd510ab8f8631b1e346e84826c
+size 1064

checkpoint-1000/special_tokens_map.json

@@ -0,0 +1,24 @@
+{
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": "<s>",
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

checkpoint-1000/tokenizer_config.json

@@ -0,0 +1,43 @@
+{
+  "add_bos_token": true,
+  "add_eos_token": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [],
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "</s>",
+  "legacy": true,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<s>",
+  "padding_side": "right",
+  "sp_model_kwargs": {},
+  "spaces_between_special_tokens": false,
+  "tokenizer_class": "LlamaTokenizer",
+  "unk_token": "<unk>",
+  "use_default_system_prompt": false
+}

checkpoint-1000/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3cf1671e8b74fb4ed4aeaf2b7b46d1549fe3f9d73086e04e9019b6f71e6be72a
+size 4984

checkpoint-500/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ee6071204b64ba451d3e1df016b69673f12ab0a2ebfd175c1dd6326aa00b246f
+size 28967217802

checkpoint-500/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5674bdb2a6ab104170223a3c25ef136efdb6024273cf8f23b144a7473b323c55
+size 14483616658

checkpoint-500/pytorch_model_fsdp.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:752594829cbd8effcf42f325d3c6a0714bd53535d8c52465cff267656a89a640
+size 14483618133

checkpoint-500/rng_state_0.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1d7e7ed1cee0a4e0a0657a08890f6df0eb6af79af079e448e2951adbb47ef3eb
+size 15920

checkpoint-500/rng_state_1.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f4808d228ab65a461951125b6adc1e5e64b8fb6f695c9e28196f3e80c79625c4
+size 15920

checkpoint-500/rng_state_2.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:72c0fe7aa0292034ad622d7e08b10e1b140185fa96aacf6175e929c5b53cd52e
+size 15920

checkpoint-500/rng_state_3.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f475edba6c2f97d481c4ccbd0d45f3b5b535d997206f29ad5ad8d1e4b11d1681
+size 15920

checkpoint-500/rng_state_4.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:93d5a90cf50d5c4f629c1c94e3d82c97e9c7fa8918af5f45a2369e46560d725c
+size 15920

checkpoint-500/rng_state_5.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f0c4aa9c96023a9544f6775e8984bb6fb16d89ce232056beec9627924647629c
+size 15920

checkpoint-500/rng_state_6.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ba63671d61493c22f54c8b91227d1ad3d0fc53e08af654f34ce8b7f59ce6c460
+size 15920

checkpoint-500/rng_state_7.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f417300ba5d28598d5c574db64831f28d4481b10e99760ece1c07c083502e63
+size 15920

checkpoint-500/scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d340feb7bcf864481a7391a71f0217c0632c294086494acb7f8c90d216e571b3
+size 1064

checkpoint-500/special_tokens_map.json

@@ -0,0 +1,24 @@
+{
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": "<s>",
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

checkpoint-500/tokenizer_config.json

@@ -0,0 +1,43 @@
+{
+  "add_bos_token": true,
+  "add_eos_token": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [],
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "</s>",
+  "legacy": true,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<s>",
+  "padding_side": "right",
+  "sp_model_kwargs": {},
+  "spaces_between_special_tokens": false,
+  "tokenizer_class": "LlamaTokenizer",
+  "unk_token": "<unk>",
+  "use_default_system_prompt": false
+}

checkpoint-500/trainer_state.json

@@ -0,0 +1,3021 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.07070135746606335,
+  "eval_steps": 500,
+  "global_step": 500,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 5.263157894736843e-07,
+      "loss": 2.4815,
+      "step": 1
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.0526315789473685e-06,
+      "loss": 4.3537,
+      "step": 2
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.5789473684210526e-06,
+      "loss": 3.3778,
+      "step": 3
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 2.105263157894737e-06,
+      "loss": 3.2835,
+      "step": 4
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 2.631578947368421e-06,
+      "loss": 1.8569,
+      "step": 5
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 3.157894736842105e-06,
+      "loss": 2.9199,
+      "step": 6
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 3.6842105263157896e-06,
+      "loss": 2.058,
+      "step": 7
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 4.210526315789474e-06,
+      "loss": 2.4038,
+      "step": 8
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 4.736842105263158e-06,
+      "loss": 1.2078,
+      "step": 9
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 5.263157894736842e-06,
+      "loss": 1.409,
+      "step": 10
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 5.789473684210527e-06,
+      "loss": 2.1197,
+      "step": 11
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.31578947368421e-06,
+      "loss": 22.4607,
+      "step": 12
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.842105263157896e-06,
+      "loss": 0.9734,
+      "step": 13
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 7.368421052631579e-06,
+      "loss": 4.2703,
+      "step": 14
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 7.894736842105265e-06,
+      "loss": 0.9632,
+      "step": 15
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 8.421052631578948e-06,
+      "loss": 3.3221,
+      "step": 16
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 8.947368421052632e-06,
+      "loss": 1.4652,
+      "step": 17
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 9.473684210526315e-06,
+      "loss": 1.5308,
+      "step": 18
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1e-05,
+      "loss": 0.8635,
+      "step": 19
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.0526315789473684e-05,
+      "loss": 2.3105,
+      "step": 20
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.105263157894737e-05,
+      "loss": 2.4293,
+      "step": 21
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.1578947368421053e-05,
+      "loss": 1.1117,
+      "step": 22
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.2105263157894737e-05,
+      "loss": 0.9303,
+      "step": 23
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.263157894736842e-05,
+      "loss": 1.2474,
+      "step": 24
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.3157894736842108e-05,
+      "loss": 3.766,
+      "step": 25
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.3684210526315791e-05,
+      "loss": 2.8922,
+      "step": 26
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.4210526315789475e-05,
+      "loss": 1.2596,
+      "step": 27
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.4736842105263159e-05,
+      "loss": 3.1363,
+      "step": 28
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.5263157894736846e-05,
+      "loss": 1.2053,
+      "step": 29
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.578947368421053e-05,
+      "loss": 0.9122,
+      "step": 30
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.6315789473684213e-05,
+      "loss": 1.68,
+      "step": 31
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.6842105263157896e-05,
+      "loss": 1.2995,
+      "step": 32
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.736842105263158e-05,
+      "loss": 1.5188,
+      "step": 33
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.7894736842105264e-05,
+      "loss": 3.362,
+      "step": 34
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.8421052631578947e-05,
+      "loss": 1.3297,
+      "step": 35
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.894736842105263e-05,
+      "loss": 1.2883,
+      "step": 36
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9473684210526318e-05,
+      "loss": 1.0937,
+      "step": 37
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 2e-05,
+      "loss": 1.1692,
+      "step": 38
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.998353909465021e-05,
+      "loss": 3.3785,
+      "step": 39
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9967078189300412e-05,
+      "loss": 3.022,
+      "step": 40
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.995061728395062e-05,
+      "loss": 2.2544,
+      "step": 41
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9934156378600823e-05,
+      "loss": 1.1995,
+      "step": 42
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.991769547325103e-05,
+      "loss": 3.618,
+      "step": 43
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9901234567901237e-05,
+      "loss": 2.1835,
+      "step": 44
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.988477366255144e-05,
+      "loss": 1.0323,
+      "step": 45
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9868312757201648e-05,
+      "loss": 1.3887,
+      "step": 46
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9851851851851855e-05,
+      "loss": 1.4793,
+      "step": 47
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.983539094650206e-05,
+      "loss": 1.4738,
+      "step": 48
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9818930041152265e-05,
+      "loss": 2.4011,
+      "step": 49
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9802469135802472e-05,
+      "loss": 1.3168,
+      "step": 50
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9786008230452676e-05,
+      "loss": 0.9198,
+      "step": 51
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9769547325102883e-05,
+      "loss": 1.0118,
+      "step": 52
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9753086419753087e-05,
+      "loss": 1.5369,
+      "step": 53
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9736625514403294e-05,
+      "loss": 1.8322,
+      "step": 54
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.97201646090535e-05,
+      "loss": 0.9453,
+      "step": 55
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9703703703703704e-05,
+      "loss": 1.3244,
+      "step": 56
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.968724279835391e-05,
+      "loss": 1.2034,
+      "step": 57
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9670781893004115e-05,
+      "loss": 1.2124,
+      "step": 58
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9654320987654322e-05,
+      "loss": 1.5657,
+      "step": 59
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.963786008230453e-05,
+      "loss": 1.4496,
+      "step": 60
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9621399176954733e-05,
+      "loss": 1.6916,
+      "step": 61
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.960493827160494e-05,
+      "loss": 1.5062,
+      "step": 62
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9588477366255147e-05,
+      "loss": 1.2756,
+      "step": 63
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9572016460905354e-05,
+      "loss": 1.1265,
+      "step": 64
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9555555555555557e-05,
+      "loss": 1.6872,
+      "step": 65
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9539094650205764e-05,
+      "loss": 1.3783,
+      "step": 66
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9522633744855968e-05,
+      "loss": 3.8868,
+      "step": 67
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9506172839506175e-05,
+      "loss": 2.9365,
+      "step": 68
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9489711934156382e-05,
+      "loss": 2.2621,
+      "step": 69
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9473251028806586e-05,
+      "loss": 1.5057,
+      "step": 70
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9456790123456793e-05,
+      "loss": 1.3916,
+      "step": 71
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9440329218106996e-05,
+      "loss": 1.6518,
+      "step": 72
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9423868312757203e-05,
+      "loss": 2.1652,
+      "step": 73
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9407407407407407e-05,
+      "loss": 0.9224,
+      "step": 74
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9390946502057614e-05,
+      "loss": 1.927,
+      "step": 75
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.937448559670782e-05,
+      "loss": 1.2233,
+      "step": 76
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9358024691358025e-05,
+      "loss": 1.8546,
+      "step": 77
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.934156378600823e-05,
+      "loss": 1.5958,
+      "step": 78
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.932510288065844e-05,
+      "loss": 1.0712,
+      "step": 79
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9308641975308646e-05,
+      "loss": 1.9209,
+      "step": 80
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.929218106995885e-05,
+      "loss": 1.9773,
+      "step": 81
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9275720164609056e-05,
+      "loss": 1.1201,
+      "step": 82
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.925925925925926e-05,
+      "loss": 1.6059,
+      "step": 83
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9242798353909467e-05,
+      "loss": 1.333,
+      "step": 84
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9226337448559674e-05,
+      "loss": 0.9293,
+      "step": 85
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9209876543209878e-05,
+      "loss": 1.9791,
+      "step": 86
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9193415637860085e-05,
+      "loss": 3.4147,
+      "step": 87
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9176954732510288e-05,
+      "loss": 1.2726,
+      "step": 88
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9160493827160495e-05,
+      "loss": 1.3713,
+      "step": 89
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.91440329218107e-05,
+      "loss": 0.9403,
+      "step": 90
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9127572016460906e-05,
+      "loss": 2.3247,
+      "step": 91
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9111111111111113e-05,
+      "loss": 2.0344,
+      "step": 92
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9094650205761317e-05,
+      "loss": 1.3213,
+      "step": 93
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9078189300411524e-05,
+      "loss": 1.8474,
+      "step": 94
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.906172839506173e-05,
+      "loss": 1.3678,
+      "step": 95
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.9045267489711938e-05,
+      "loss": 1.5695,
+      "step": 96
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.902880658436214e-05,
+      "loss": 0.8382,
+      "step": 97
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.901234567901235e-05,
+      "loss": 1.1056,
+      "step": 98
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8995884773662555e-05,
+      "loss": 1.3256,
+      "step": 99
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.897942386831276e-05,
+      "loss": 1.3851,
+      "step": 100
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8962962962962966e-05,
+      "loss": 1.3965,
+      "step": 101
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.894650205761317e-05,
+      "loss": 2.1076,
+      "step": 102
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8930041152263377e-05,
+      "loss": 1.021,
+      "step": 103
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.891358024691358e-05,
+      "loss": 2.3561,
+      "step": 104
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8897119341563787e-05,
+      "loss": 1.3618,
+      "step": 105
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 1.8880658436213994e-05,
+      "loss": 2.7243,
+      "step": 106
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8864197530864198e-05,
+      "loss": 0.9294,
+      "step": 107
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8847736625514405e-05,
+      "loss": 1.4138,
+      "step": 108
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.883127572016461e-05,
+      "loss": 1.4628,
+      "step": 109
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8814814814814816e-05,
+      "loss": 2.6227,
+      "step": 110
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8798353909465023e-05,
+      "loss": 0.8669,
+      "step": 111
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.878189300411523e-05,
+      "loss": 1.2118,
+      "step": 112
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8765432098765433e-05,
+      "loss": 0.874,
+      "step": 113
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.874897119341564e-05,
+      "loss": 1.793,
+      "step": 114
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8732510288065847e-05,
+      "loss": 1.2895,
+      "step": 115
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.871604938271605e-05,
+      "loss": 1.2253,
+      "step": 116
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8699588477366258e-05,
+      "loss": 1.1749,
+      "step": 117
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.868312757201646e-05,
+      "loss": 1.3432,
+      "step": 118
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.866666666666667e-05,
+      "loss": 0.904,
+      "step": 119
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8650205761316872e-05,
+      "loss": 1.9507,
+      "step": 120
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.863374485596708e-05,
+      "loss": 2.6723,
+      "step": 121
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8617283950617286e-05,
+      "loss": 1.222,
+      "step": 122
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.860082304526749e-05,
+      "loss": 2.6814,
+      "step": 123
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8584362139917697e-05,
+      "loss": 1.181,
+      "step": 124
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.85679012345679e-05,
+      "loss": 1.2468,
+      "step": 125
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8551440329218108e-05,
+      "loss": 0.8456,
+      "step": 126
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8534979423868315e-05,
+      "loss": 2.4871,
+      "step": 127
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.851851851851852e-05,
+      "loss": 1.0439,
+      "step": 128
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8502057613168725e-05,
+      "loss": 1.3051,
+      "step": 129
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8485596707818932e-05,
+      "loss": 2.4871,
+      "step": 130
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.846913580246914e-05,
+      "loss": 1.2913,
+      "step": 131
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8452674897119343e-05,
+      "loss": 1.3607,
+      "step": 132
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.843621399176955e-05,
+      "loss": 0.7906,
+      "step": 133
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8419753086419754e-05,
+      "loss": 1.1152,
+      "step": 134
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.840329218106996e-05,
+      "loss": 1.3647,
+      "step": 135
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8386831275720164e-05,
+      "loss": 1.1403,
+      "step": 136
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.837037037037037e-05,
+      "loss": 1.3415,
+      "step": 137
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.835390946502058e-05,
+      "loss": 2.3992,
+      "step": 138
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8337448559670782e-05,
+      "loss": 1.5928,
+      "step": 139
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.832098765432099e-05,
+      "loss": 2.0784,
+      "step": 140
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8304526748971193e-05,
+      "loss": 1.2535,
+      "step": 141
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.82880658436214e-05,
+      "loss": 1.1737,
+      "step": 142
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8271604938271607e-05,
+      "loss": 1.4685,
+      "step": 143
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8255144032921814e-05,
+      "loss": 0.8403,
+      "step": 144
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8238683127572017e-05,
+      "loss": 1.9443,
+      "step": 145
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8222222222222224e-05,
+      "loss": 1.9383,
+      "step": 146
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.820576131687243e-05,
+      "loss": 0.9846,
+      "step": 147
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8189300411522635e-05,
+      "loss": 1.0405,
+      "step": 148
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8172839506172842e-05,
+      "loss": 1.0307,
+      "step": 149
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8156378600823046e-05,
+      "loss": 2.7594,
+      "step": 150
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8139917695473253e-05,
+      "loss": 1.4007,
+      "step": 151
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.812345679012346e-05,
+      "loss": 2.22,
+      "step": 152
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8106995884773663e-05,
+      "loss": 2.2109,
+      "step": 153
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.809053497942387e-05,
+      "loss": 1.3844,
+      "step": 154
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8074074074074074e-05,
+      "loss": 1.1362,
+      "step": 155
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.805761316872428e-05,
+      "loss": 2.0179,
+      "step": 156
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.8041152263374485e-05,
+      "loss": 0.7766,
+      "step": 157
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.802469135802469e-05,
+      "loss": 1.3414,
+      "step": 158
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.80082304526749e-05,
+      "loss": 1.5758,
+      "step": 159
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7991769547325106e-05,
+      "loss": 1.4342,
+      "step": 160
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.797530864197531e-05,
+      "loss": 0.9155,
+      "step": 161
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7958847736625516e-05,
+      "loss": 2.3989,
+      "step": 162
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7942386831275723e-05,
+      "loss": 1.1246,
+      "step": 163
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7925925925925927e-05,
+      "loss": 2.6724,
+      "step": 164
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7909465020576134e-05,
+      "loss": 2.3056,
+      "step": 165
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7893004115226338e-05,
+      "loss": 2.2982,
+      "step": 166
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7876543209876545e-05,
+      "loss": 1.4357,
+      "step": 167
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.786008230452675e-05,
+      "loss": 1.4224,
+      "step": 168
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7843621399176955e-05,
+      "loss": 1.8674,
+      "step": 169
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7827160493827162e-05,
+      "loss": 1.2265,
+      "step": 170
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7810699588477366e-05,
+      "loss": 1.2305,
+      "step": 171
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7794238683127573e-05,
+      "loss": 2.1299,
+      "step": 172
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7777777777777777e-05,
+      "loss": 1.3587,
+      "step": 173
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7761316872427984e-05,
+      "loss": 1.7709,
+      "step": 174
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.774485596707819e-05,
+      "loss": 1.1696,
+      "step": 175
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.7728395061728398e-05,
+      "loss": 1.2198,
+      "step": 176
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.77119341563786e-05,
+      "loss": 2.3921,
+      "step": 177
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.769547325102881e-05,
+      "loss": 2.0317,
+      "step": 178
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7679012345679015e-05,
+      "loss": 1.6844,
+      "step": 179
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.766255144032922e-05,
+      "loss": 2.1612,
+      "step": 180
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7646090534979426e-05,
+      "loss": 1.0552,
+      "step": 181
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7629629629629633e-05,
+      "loss": 1.2516,
+      "step": 182
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7613168724279837e-05,
+      "loss": 1.1556,
+      "step": 183
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7596707818930044e-05,
+      "loss": 0.8491,
+      "step": 184
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7580246913580247e-05,
+      "loss": 1.2157,
+      "step": 185
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7563786008230454e-05,
+      "loss": 2.6295,
+      "step": 186
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7547325102880658e-05,
+      "loss": 0.9006,
+      "step": 187
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7530864197530865e-05,
+      "loss": 0.8407,
+      "step": 188
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7514403292181072e-05,
+      "loss": 1.2994,
+      "step": 189
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7497942386831276e-05,
+      "loss": 1.1556,
+      "step": 190
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7481481481481483e-05,
+      "loss": 1.1404,
+      "step": 191
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.746502057613169e-05,
+      "loss": 1.3204,
+      "step": 192
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7448559670781893e-05,
+      "loss": 2.5288,
+      "step": 193
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.74320987654321e-05,
+      "loss": 1.0881,
+      "step": 194
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7415637860082307e-05,
+      "loss": 2.0757,
+      "step": 195
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.739917695473251e-05,
+      "loss": 1.641,
+      "step": 196
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7382716049382718e-05,
+      "loss": 0.9791,
+      "step": 197
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7366255144032925e-05,
+      "loss": 0.8751,
+      "step": 198
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.734979423868313e-05,
+      "loss": 1.412,
+      "step": 199
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7333333333333336e-05,
+      "loss": 1.2978,
+      "step": 200
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.731687242798354e-05,
+      "loss": 1.3859,
+      "step": 201
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7300411522633746e-05,
+      "loss": 0.8929,
+      "step": 202
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.728395061728395e-05,
+      "loss": 1.1543,
+      "step": 203
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7267489711934157e-05,
+      "loss": 1.0724,
+      "step": 204
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7251028806584364e-05,
+      "loss": 1.2097,
+      "step": 205
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7234567901234568e-05,
+      "loss": 1.9042,
+      "step": 206
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7218106995884775e-05,
+      "loss": 1.7308,
+      "step": 207
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.720164609053498e-05,
+      "loss": 1.2102,
+      "step": 208
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7185185185185185e-05,
+      "loss": 1.2387,
+      "step": 209
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7168724279835392e-05,
+      "loss": 1.1042,
+      "step": 210
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.71522633744856e-05,
+      "loss": 1.0871,
+      "step": 211
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7135802469135806e-05,
+      "loss": 1.8654,
+      "step": 212
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.711934156378601e-05,
+      "loss": 1.8047,
+      "step": 213
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7102880658436217e-05,
+      "loss": 1.9824,
+      "step": 214
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.708641975308642e-05,
+      "loss": 1.3247,
+      "step": 215
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7069958847736628e-05,
+      "loss": 1.2086,
+      "step": 216
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.705349794238683e-05,
+      "loss": 1.6138,
+      "step": 217
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7037037037037038e-05,
+      "loss": 1.1922,
+      "step": 218
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.7020576131687242e-05,
+      "loss": 1.6419,
+      "step": 219
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.700411522633745e-05,
+      "loss": 1.7016,
+      "step": 220
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6987654320987656e-05,
+      "loss": 0.8859,
+      "step": 221
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.697119341563786e-05,
+      "loss": 2.0252,
+      "step": 222
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6954732510288067e-05,
+      "loss": 1.1653,
+      "step": 223
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6938271604938274e-05,
+      "loss": 1.2713,
+      "step": 224
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.692181069958848e-05,
+      "loss": 1.4138,
+      "step": 225
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6905349794238684e-05,
+      "loss": 1.335,
+      "step": 226
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.688888888888889e-05,
+      "loss": 2.347,
+      "step": 227
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.68724279835391e-05,
+      "loss": 0.6731,
+      "step": 228
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6855967078189302e-05,
+      "loss": 1.9767,
+      "step": 229
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.683950617283951e-05,
+      "loss": 1.1805,
+      "step": 230
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6823045267489713e-05,
+      "loss": 1.0245,
+      "step": 231
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.680658436213992e-05,
+      "loss": 0.8785,
+      "step": 232
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6790123456790123e-05,
+      "loss": 1.1263,
+      "step": 233
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.677366255144033e-05,
+      "loss": 1.2684,
+      "step": 234
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6757201646090537e-05,
+      "loss": 1.3128,
+      "step": 235
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.674074074074074e-05,
+      "loss": 1.856,
+      "step": 236
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6724279835390948e-05,
+      "loss": 2.0008,
+      "step": 237
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.670781893004115e-05,
+      "loss": 0.8432,
+      "step": 238
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.669135802469136e-05,
+      "loss": 1.3959,
+      "step": 239
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6674897119341566e-05,
+      "loss": 1.1266,
+      "step": 240
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6658436213991773e-05,
+      "loss": 1.3387,
+      "step": 241
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6641975308641976e-05,
+      "loss": 1.1307,
+      "step": 242
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6625514403292183e-05,
+      "loss": 1.0658,
+      "step": 243
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.660905349794239e-05,
+      "loss": 1.176,
+      "step": 244
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6592592592592594e-05,
+      "loss": 1.1264,
+      "step": 245
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.65761316872428e-05,
+      "loss": 1.2464,
+      "step": 246
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6559670781893005e-05,
+      "loss": 1.0174,
+      "step": 247
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.654320987654321e-05,
+      "loss": 1.1007,
+      "step": 248
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6526748971193415e-05,
+      "loss": 1.1928,
+      "step": 249
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6510288065843622e-05,
+      "loss": 2.14,
+      "step": 250
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.649382716049383e-05,
+      "loss": 0.7998,
+      "step": 251
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6477366255144033e-05,
+      "loss": 1.1035,
+      "step": 252
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.646090534979424e-05,
+      "loss": 1.5725,
+      "step": 253
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6444444444444444e-05,
+      "loss": 0.9665,
+      "step": 254
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.642798353909465e-05,
+      "loss": 2.3221,
+      "step": 255
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6411522633744858e-05,
+      "loss": 0.8597,
+      "step": 256
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6395061728395065e-05,
+      "loss": 1.1366,
+      "step": 257
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6378600823045268e-05,
+      "loss": 0.7674,
+      "step": 258
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6362139917695475e-05,
+      "loss": 1.0137,
+      "step": 259
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6345679012345682e-05,
+      "loss": 1.4277,
+      "step": 260
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6329218106995886e-05,
+      "loss": 1.088,
+      "step": 261
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6312757201646093e-05,
+      "loss": 1.0693,
+      "step": 262
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6296296296296297e-05,
+      "loss": 0.845,
+      "step": 263
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6279835390946504e-05,
+      "loss": 1.1176,
+      "step": 264
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.626337448559671e-05,
+      "loss": 1.2135,
+      "step": 265
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6246913580246914e-05,
+      "loss": 0.7988,
+      "step": 266
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.623045267489712e-05,
+      "loss": 1.1999,
+      "step": 267
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6213991769547325e-05,
+      "loss": 1.0806,
+      "step": 268
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6197530864197532e-05,
+      "loss": 1.0963,
+      "step": 269
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6181069958847736e-05,
+      "loss": 1.709,
+      "step": 270
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6164609053497943e-05,
+      "loss": 0.9721,
+      "step": 271
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.614814814814815e-05,
+      "loss": 1.7105,
+      "step": 272
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6131687242798357e-05,
+      "loss": 0.9966,
+      "step": 273
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.611522633744856e-05,
+      "loss": 2.0088,
+      "step": 274
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6098765432098767e-05,
+      "loss": 1.3689,
+      "step": 275
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6082304526748974e-05,
+      "loss": 0.9826,
+      "step": 276
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6065843621399178e-05,
+      "loss": 0.8722,
+      "step": 277
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6049382716049385e-05,
+      "loss": 1.4338,
+      "step": 278
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.603292181069959e-05,
+      "loss": 1.1227,
+      "step": 279
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6016460905349796e-05,
+      "loss": 1.2619,
+      "step": 280
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.6000000000000003e-05,
+      "loss": 0.9965,
+      "step": 281
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5983539094650206e-05,
+      "loss": 1.4239,
+      "step": 282
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5967078189300413e-05,
+      "loss": 1.1037,
+      "step": 283
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5950617283950617e-05,
+      "loss": 1.9293,
+      "step": 284
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5934156378600824e-05,
+      "loss": 1.133,
+      "step": 285
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5917695473251028e-05,
+      "loss": 2.2002,
+      "step": 286
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5901234567901235e-05,
+      "loss": 1.551,
+      "step": 287
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.588477366255144e-05,
+      "loss": 1.2106,
+      "step": 288
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.586831275720165e-05,
+      "loss": 1.1269,
+      "step": 289
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5851851851851852e-05,
+      "loss": 0.8801,
+      "step": 290
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.583539094650206e-05,
+      "loss": 0.8818,
+      "step": 291
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5818930041152266e-05,
+      "loss": 1.4156,
+      "step": 292
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.580246913580247e-05,
+      "loss": 1.7768,
+      "step": 293
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5786008230452677e-05,
+      "loss": 1.3257,
+      "step": 294
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5769547325102884e-05,
+      "loss": 1.2448,
+      "step": 295
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5753086419753088e-05,
+      "loss": 0.924,
+      "step": 296
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5736625514403295e-05,
+      "loss": 0.8349,
+      "step": 297
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5720164609053498e-05,
+      "loss": 2.0425,
+      "step": 298
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5703703703703705e-05,
+      "loss": 1.8374,
+      "step": 299
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.568724279835391e-05,
+      "loss": 1.3335,
+      "step": 300
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5670781893004116e-05,
+      "loss": 2.097,
+      "step": 301
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.565432098765432e-05,
+      "loss": 1.9508,
+      "step": 302
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5637860082304527e-05,
+      "loss": 1.1492,
+      "step": 303
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5621399176954734e-05,
+      "loss": 1.894,
+      "step": 304
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.560493827160494e-05,
+      "loss": 1.6447,
+      "step": 305
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5588477366255144e-05,
+      "loss": 0.9686,
+      "step": 306
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.557201646090535e-05,
+      "loss": 1.4122,
+      "step": 307
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.555555555555556e-05,
+      "loss": 0.8404,
+      "step": 308
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5539094650205762e-05,
+      "loss": 1.4608,
+      "step": 309
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.552263374485597e-05,
+      "loss": 1.2198,
+      "step": 310
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5506172839506176e-05,
+      "loss": 1.086,
+      "step": 311
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.548971193415638e-05,
+      "loss": 0.9219,
+      "step": 312
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5473251028806587e-05,
+      "loss": 1.5666,
+      "step": 313
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.545679012345679e-05,
+      "loss": 2.3637,
+      "step": 314
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5440329218106997e-05,
+      "loss": 1.1629,
+      "step": 315
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.54238683127572e-05,
+      "loss": 1.1377,
+      "step": 316
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5407407407407408e-05,
+      "loss": 1.1348,
+      "step": 317
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.5390946502057615e-05,
+      "loss": 1.286,
+      "step": 318
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.537448559670782e-05,
+      "loss": 1.2544,
+      "step": 319
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5358024691358026e-05,
+      "loss": 1.1091,
+      "step": 320
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5341563786008233e-05,
+      "loss": 1.2463,
+      "step": 321
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5325102880658436e-05,
+      "loss": 1.0978,
+      "step": 322
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5308641975308643e-05,
+      "loss": 2.5098,
+      "step": 323
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.529218106995885e-05,
+      "loss": 1.0751,
+      "step": 324
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5275720164609057e-05,
+      "loss": 1.1688,
+      "step": 325
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.525925925925926e-05,
+      "loss": 1.2842,
+      "step": 326
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5242798353909466e-05,
+      "loss": 1.4204,
+      "step": 327
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5226337448559672e-05,
+      "loss": 1.0947,
+      "step": 328
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5209876543209879e-05,
+      "loss": 1.5603,
+      "step": 329
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5193415637860082e-05,
+      "loss": 1.2108,
+      "step": 330
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.517695473251029e-05,
+      "loss": 1.6393,
+      "step": 331
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5160493827160495e-05,
+      "loss": 2.0431,
+      "step": 332
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.51440329218107e-05,
+      "loss": 1.1368,
+      "step": 333
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5127572016460907e-05,
+      "loss": 1.2198,
+      "step": 334
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5111111111111112e-05,
+      "loss": 0.8841,
+      "step": 335
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.509465020576132e-05,
+      "loss": 1.077,
+      "step": 336
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5078189300411523e-05,
+      "loss": 0.9249,
+      "step": 337
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.506172839506173e-05,
+      "loss": 1.3955,
+      "step": 338
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5045267489711934e-05,
+      "loss": 1.2066,
+      "step": 339
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.502880658436214e-05,
+      "loss": 1.8491,
+      "step": 340
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.5012345679012348e-05,
+      "loss": 1.8122,
+      "step": 341
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4995884773662553e-05,
+      "loss": 1.9627,
+      "step": 342
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4979423868312758e-05,
+      "loss": 1.0967,
+      "step": 343
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4962962962962964e-05,
+      "loss": 0.9233,
+      "step": 344
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.494650205761317e-05,
+      "loss": 2.1129,
+      "step": 345
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4930041152263374e-05,
+      "loss": 1.6624,
+      "step": 346
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4913580246913581e-05,
+      "loss": 1.1757,
+      "step": 347
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4897119341563788e-05,
+      "loss": 1.0912,
+      "step": 348
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4880658436213992e-05,
+      "loss": 1.097,
+      "step": 349
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4864197530864199e-05,
+      "loss": 1.4303,
+      "step": 350
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4847736625514404e-05,
+      "loss": 1.0809,
+      "step": 351
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4831275720164611e-05,
+      "loss": 1.742,
+      "step": 352
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4814814814814815e-05,
+      "loss": 1.0894,
+      "step": 353
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4798353909465022e-05,
+      "loss": 1.4299,
+      "step": 354
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4781893004115229e-05,
+      "loss": 0.8998,
+      "step": 355
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4765432098765433e-05,
+      "loss": 1.1283,
+      "step": 356
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.474897119341564e-05,
+      "loss": 1.139,
+      "step": 357
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4732510288065845e-05,
+      "loss": 1.4199,
+      "step": 358
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.471604938271605e-05,
+      "loss": 1.1676,
+      "step": 359
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4699588477366256e-05,
+      "loss": 1.9596,
+      "step": 360
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4683127572016463e-05,
+      "loss": 1.042,
+      "step": 361
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4666666666666666e-05,
+      "loss": 1.1311,
+      "step": 362
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4650205761316873e-05,
+      "loss": 1.687,
+      "step": 363
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.463374485596708e-05,
+      "loss": 0.9777,
+      "step": 364
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4617283950617284e-05,
+      "loss": 1.8194,
+      "step": 365
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4600823045267491e-05,
+      "loss": 1.6943,
+      "step": 366
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4584362139917696e-05,
+      "loss": 1.0503,
+      "step": 367
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4567901234567903e-05,
+      "loss": 0.9961,
+      "step": 368
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4551440329218107e-05,
+      "loss": 0.9278,
+      "step": 369
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4534979423868314e-05,
+      "loss": 1.0037,
+      "step": 370
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4518518518518521e-05,
+      "loss": 1.0682,
+      "step": 371
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4502057613168725e-05,
+      "loss": 1.3941,
+      "step": 372
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4485596707818932e-05,
+      "loss": 0.9016,
+      "step": 373
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4469135802469137e-05,
+      "loss": 1.6304,
+      "step": 374
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4452674897119342e-05,
+      "loss": 1.6471,
+      "step": 375
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4436213991769548e-05,
+      "loss": 1.0599,
+      "step": 376
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4419753086419755e-05,
+      "loss": 1.1956,
+      "step": 377
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4403292181069962e-05,
+      "loss": 1.975,
+      "step": 378
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4386831275720165e-05,
+      "loss": 0.8926,
+      "step": 379
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4370370370370372e-05,
+      "loss": 1.1477,
+      "step": 380
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4353909465020578e-05,
+      "loss": 0.7713,
+      "step": 381
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4337448559670783e-05,
+      "loss": 1.0733,
+      "step": 382
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4320987654320988e-05,
+      "loss": 0.9424,
+      "step": 383
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4304526748971195e-05,
+      "loss": 1.1431,
+      "step": 384
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4288065843621399e-05,
+      "loss": 1.1502,
+      "step": 385
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4271604938271606e-05,
+      "loss": 1.9421,
+      "step": 386
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4255144032921813e-05,
+      "loss": 0.9716,
+      "step": 387
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.4238683127572017e-05,
+      "loss": 1.7941,
+      "step": 388
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.4222222222222224e-05,
+      "loss": 1.6438,
+      "step": 389
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.4205761316872429e-05,
+      "loss": 0.9105,
+      "step": 390
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.4189300411522636e-05,
+      "loss": 0.8231,
+      "step": 391
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.417283950617284e-05,
+      "loss": 1.2418,
+      "step": 392
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.4156378600823047e-05,
+      "loss": 1.0835,
+      "step": 393
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.4139917695473254e-05,
+      "loss": 1.0516,
+      "step": 394
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.4123456790123457e-05,
+      "loss": 1.0572,
+      "step": 395
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.4106995884773664e-05,
+      "loss": 1.6052,
+      "step": 396
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.409053497942387e-05,
+      "loss": 1.5509,
+      "step": 397
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.4074074074074075e-05,
+      "loss": 1.5081,
+      "step": 398
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.405761316872428e-05,
+      "loss": 1.1168,
+      "step": 399
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.4041152263374487e-05,
+      "loss": 1.1208,
+      "step": 400
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.4024691358024694e-05,
+      "loss": 1.1381,
+      "step": 401
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.4008230452674898e-05,
+      "loss": 1.1407,
+      "step": 402
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3991769547325105e-05,
+      "loss": 1.3994,
+      "step": 403
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3975308641975309e-05,
+      "loss": 1.4896,
+      "step": 404
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3958847736625516e-05,
+      "loss": 1.6368,
+      "step": 405
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3942386831275721e-05,
+      "loss": 0.9495,
+      "step": 406
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3925925925925928e-05,
+      "loss": 0.8156,
+      "step": 407
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3909465020576133e-05,
+      "loss": 1.452,
+      "step": 408
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3893004115226339e-05,
+      "loss": 1.0892,
+      "step": 409
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3876543209876546e-05,
+      "loss": 1.2386,
+      "step": 410
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.386008230452675e-05,
+      "loss": 1.4466,
+      "step": 411
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3843621399176956e-05,
+      "loss": 1.7343,
+      "step": 412
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3827160493827162e-05,
+      "loss": 1.7157,
+      "step": 413
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3810699588477367e-05,
+      "loss": 1.1039,
+      "step": 414
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3794238683127572e-05,
+      "loss": 0.9846,
+      "step": 415
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.377777777777778e-05,
+      "loss": 1.8517,
+      "step": 416
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3761316872427986e-05,
+      "loss": 0.7884,
+      "step": 417
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.374485596707819e-05,
+      "loss": 1.115,
+      "step": 418
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3728395061728397e-05,
+      "loss": 1.1919,
+      "step": 419
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.37119341563786e-05,
+      "loss": 0.8421,
+      "step": 420
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3695473251028808e-05,
+      "loss": 1.0411,
+      "step": 421
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3679012345679013e-05,
+      "loss": 0.8174,
+      "step": 422
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.366255144032922e-05,
+      "loss": 1.9548,
+      "step": 423
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3646090534979425e-05,
+      "loss": 1.0577,
+      "step": 424
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.362962962962963e-05,
+      "loss": 1.7725,
+      "step": 425
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3613168724279838e-05,
+      "loss": 1.4971,
+      "step": 426
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3596707818930041e-05,
+      "loss": 0.7645,
+      "step": 427
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3580246913580248e-05,
+      "loss": 1.6089,
+      "step": 428
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3563786008230454e-05,
+      "loss": 1.2194,
+      "step": 429
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3547325102880659e-05,
+      "loss": 2.0046,
+      "step": 430
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3530864197530866e-05,
+      "loss": 1.1987,
+      "step": 431
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3514403292181071e-05,
+      "loss": 1.6042,
+      "step": 432
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3497942386831278e-05,
+      "loss": 1.1551,
+      "step": 433
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3481481481481482e-05,
+      "loss": 1.473,
+      "step": 434
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3465020576131689e-05,
+      "loss": 1.9568,
+      "step": 435
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3448559670781893e-05,
+      "loss": 1.6012,
+      "step": 436
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.34320987654321e-05,
+      "loss": 1.1344,
+      "step": 437
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3415637860082307e-05,
+      "loss": 1.8465,
+      "step": 438
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3399176954732512e-05,
+      "loss": 1.4803,
+      "step": 439
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3382716049382717e-05,
+      "loss": 1.0473,
+      "step": 440
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3366255144032923e-05,
+      "loss": 0.7756,
+      "step": 441
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.334979423868313e-05,
+      "loss": 1.13,
+      "step": 442
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3333333333333333e-05,
+      "loss": 0.9533,
+      "step": 443
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.331687242798354e-05,
+      "loss": 0.9758,
+      "step": 444
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3300411522633746e-05,
+      "loss": 1.366,
+      "step": 445
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3283950617283951e-05,
+      "loss": 0.8834,
+      "step": 446
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3267489711934158e-05,
+      "loss": 1.1171,
+      "step": 447
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3251028806584363e-05,
+      "loss": 1.3277,
+      "step": 448
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.323456790123457e-05,
+      "loss": 0.7355,
+      "step": 449
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3218106995884774e-05,
+      "loss": 0.8361,
+      "step": 450
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3201646090534981e-05,
+      "loss": 1.7336,
+      "step": 451
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3185185185185185e-05,
+      "loss": 0.9619,
+      "step": 452
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3168724279835392e-05,
+      "loss": 1.2037,
+      "step": 453
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3152263374485599e-05,
+      "loss": 0.8276,
+      "step": 454
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3135802469135804e-05,
+      "loss": 0.7516,
+      "step": 455
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.311934156378601e-05,
+      "loss": 0.9177,
+      "step": 456
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3102880658436215e-05,
+      "loss": 1.4278,
+      "step": 457
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3086419753086422e-05,
+      "loss": 1.4967,
+      "step": 458
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.3069958847736625e-05,
+      "loss": 1.3179,
+      "step": 459
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.3053497942386832e-05,
+      "loss": 1.1972,
+      "step": 460
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.303703703703704e-05,
+      "loss": 1.0011,
+      "step": 461
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.3020576131687243e-05,
+      "loss": 1.1314,
+      "step": 462
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.300411522633745e-05,
+      "loss": 1.1683,
+      "step": 463
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2987654320987655e-05,
+      "loss": 1.4075,
+      "step": 464
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2971193415637862e-05,
+      "loss": 1.2104,
+      "step": 465
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2954732510288066e-05,
+      "loss": 0.9692,
+      "step": 466
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2938271604938273e-05,
+      "loss": 1.1823,
+      "step": 467
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2921810699588477e-05,
+      "loss": 1.1024,
+      "step": 468
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2905349794238684e-05,
+      "loss": 0.8749,
+      "step": 469
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.288888888888889e-05,
+      "loss": 1.6568,
+      "step": 470
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2872427983539096e-05,
+      "loss": 1.2729,
+      "step": 471
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2855967078189301e-05,
+      "loss": 0.7668,
+      "step": 472
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2839506172839507e-05,
+      "loss": 1.2016,
+      "step": 473
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2823045267489714e-05,
+      "loss": 1.0708,
+      "step": 474
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2806584362139917e-05,
+      "loss": 0.9162,
+      "step": 475
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2790123456790124e-05,
+      "loss": 0.7652,
+      "step": 476
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2773662551440331e-05,
+      "loss": 1.8764,
+      "step": 477
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2757201646090535e-05,
+      "loss": 0.8001,
+      "step": 478
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2740740740740742e-05,
+      "loss": 1.1344,
+      "step": 479
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2724279835390947e-05,
+      "loss": 1.6359,
+      "step": 480
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2707818930041154e-05,
+      "loss": 0.8165,
+      "step": 481
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2691358024691358e-05,
+      "loss": 0.794,
+      "step": 482
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2674897119341565e-05,
+      "loss": 1.0602,
+      "step": 483
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2658436213991772e-05,
+      "loss": 1.1374,
+      "step": 484
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2641975308641976e-05,
+      "loss": 1.3747,
+      "step": 485
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2625514403292183e-05,
+      "loss": 1.5979,
+      "step": 486
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2609053497942388e-05,
+      "loss": 1.416,
+      "step": 487
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2592592592592593e-05,
+      "loss": 1.0315,
+      "step": 488
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2576131687242799e-05,
+      "loss": 1.5533,
+      "step": 489
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2559670781893006e-05,
+      "loss": 1.1579,
+      "step": 490
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2543209876543213e-05,
+      "loss": 1.3781,
+      "step": 491
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2526748971193416e-05,
+      "loss": 0.9454,
+      "step": 492
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2510288065843623e-05,
+      "loss": 1.0059,
+      "step": 493
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2493827160493827e-05,
+      "loss": 0.8041,
+      "step": 494
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2477366255144034e-05,
+      "loss": 0.9324,
+      "step": 495
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.246090534979424e-05,
+      "loss": 0.9093,
+      "step": 496
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2444444444444446e-05,
+      "loss": 1.4422,
+      "step": 497
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.242798353909465e-05,
+      "loss": 0.7431,
+      "step": 498
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2411522633744857e-05,
+      "loss": 1.0068,
+      "step": 499
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.2395061728395064e-05,
+      "loss": 1.3423,
+      "step": 500
+    }
+  ],
+  "logging_steps": 1.0,
+  "max_steps": 1253,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "save_steps": 500,
+  "total_flos": 0.0,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-500/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3cf1671e8b74fb4ed4aeaf2b7b46d1549fe3f9d73086e04e9019b6f71e6be72a
+size 4984

config.json

@@ -0,0 +1,32 @@
+{
+  "_name_or_path": "mistralai/Mistral-7B-v0.1",
+  "architectures": [
+    "MistralForCausalLM"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 1,
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 4096,
+  "id2label": {
+    "0": "LABEL_0"
+  },
+  "initializer_range": 0.02,
+  "intermediate_size": 14336,
+  "label2id": {
+    "LABEL_0": 0
+  },
+  "max_position_embeddings": 32768,
+  "model_type": "mistral",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 32,
+  "num_key_value_heads": 8,
+  "rms_norm_eps": 1e-05,
+  "rope_theta": 10000.0,
+  "sliding_window": 4096,
+  "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.36.2",
+  "use_cache": true,
+  "vocab_size": 32000
+}

dataset_num_samples.json

@@ -0,0 +1 @@
+{"tuluv2.jsonl": 320860, "e5_train_ds_msmarco-doc_format.jsonl": 71564, "e5_train_ds_gpt4-retrieval_format.jsonl": 166602, "e5_train_ds_nq_format.jsonl": 71408, "e5_train_ds_t2ranking_format.jsonl": 112335, "e5_train_ds_squad_format.jsonl": 87599, "e5_train_ds_gpt4-bitext_format.jsonl": 89324, "e5_train_ds_quora_format.jsonl": 14926, "e5_train_ds_gpt4-sts_format.jsonl": 98626, "e5_train_ds_fever_format.jsonl": 71257, "e5_train_ds_mrtydi_format.jsonl": 48729, "e5_train_ds_miracl_format.jsonl": 40203, "e5_train_ds_trivia_format.jsonl": 60296, "e5_train_ds_gpt4-p2p_format.jsonl": 16842, "e5_train_ds_msmarco_format.jsonl": 244582, "e5_train_ds_hotpotqa_format.jsonl": 68659, "e5_train_ds_eli5_format.jsonl": 50293, "e5_train_ds_nli_format.jsonl": 275601, "e5_train_ds_gpt4-s2s_format.jsonl": 13481, "e5_train_ds_gpt4-p2s_format.jsonl": 121878, "e5_train_ds_dureader_format.jsonl": 86395}

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ecb4b0093883d7ff7f6b3d6d8a9f17d79fcec393e6610c41dc303f5a036a370c
+size 14483567250

special_tokens_map.json

@@ -0,0 +1,24 @@
+{
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": "<s>",
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer_config.json

@@ -0,0 +1,43 @@
+{
+  "add_bos_token": true,
+  "add_eos_token": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [],
+  "bos_token": "<s>",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "</s>",
+  "legacy": true,
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<s>",
+  "padding_side": "right",
+  "sp_model_kwargs": {},
+  "spaces_between_special_tokens": false,
+  "tokenizer_class": "LlamaTokenizer",
+  "unk_token": "<unk>",
+  "use_default_system_prompt": false
+}

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3cf1671e8b74fb4ed4aeaf2b7b46d1549fe3f9d73086e04e9019b6f71e6be72a
+size 4984