Training in progress, step 500

checkpoint-500/config.json

@@ -0,0 +1,23 @@
+{
+  "_name_or_path": "/home/ubuntu/stanford_alpaca/alpaca-cleaned",
+  "architectures": [
+    "LLaMAForCausalLM"
+  ],
+  "bos_token_id": 0,
+  "eos_token_id": 1,
+  "hidden_act": "silu",
+  "hidden_size": 4096,
+  "initializer_range": 0.02,
+  "intermediate_size": 11008,
+  "max_sequence_length": 2048,
+  "model_type": "llama",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 32,
+  "pad_token_id": -1,
+  "rms_norm_eps": 1e-06,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float32",
+  "transformers_version": "4.27.0.dev0",
+  "use_cache": true,
+  "vocab_size": 32001
+}

checkpoint-500/generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 0,
+  "eos_token_id": 1,
+  "pad_token_id": 0,
+  "transformers_version": "4.27.0.dev0"
+}

checkpoint-500/optimizer.pt

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

checkpoint-500/pytorch_model-00001-of-00003.bin

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

checkpoint-500/pytorch_model-00002-of-00003.bin

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

checkpoint-500/pytorch_model-00003-of-00003.bin

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

checkpoint-500/pytorch_model.bin.index.json

@@ -0,0 +1,330 @@
+{
+  "metadata": {
+    "total_size": 26953699328
+  },
+  "weight_map": {
+    "lm_head.weight": "pytorch_model-00003-of-00003.bin",
+    "model.embed_tokens.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.0.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.0.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.0.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.0.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.0.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.0.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.0.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.0.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.0.self_attn.rotary_emb.inv_freq": "pytorch_model-00001-of-00003.bin",
+    "model.layers.0.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.1.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.1.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.1.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.1.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.1.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.1.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.1.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.1.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.1.self_attn.rotary_emb.inv_freq": "pytorch_model-00001-of-00003.bin",
+    "model.layers.1.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.10.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.10.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.10.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.10.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.10.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.10.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.10.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.10.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.10.self_attn.rotary_emb.inv_freq": "pytorch_model-00001-of-00003.bin",
+    "model.layers.10.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.11.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.11.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.11.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.11.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.11.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.11.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.11.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.11.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.11.self_attn.rotary_emb.inv_freq": "pytorch_model-00001-of-00003.bin",
+    "model.layers.11.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.12.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.12.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.12.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.12.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.12.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.12.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.12.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.12.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.12.self_attn.rotary_emb.inv_freq": "pytorch_model-00002-of-00003.bin",
+    "model.layers.12.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.13.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.13.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.13.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.13.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.13.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.13.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.13.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.13.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.13.self_attn.rotary_emb.inv_freq": "pytorch_model-00002-of-00003.bin",
+    "model.layers.13.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.14.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.14.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.14.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.14.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.14.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.14.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.14.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.14.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.14.self_attn.rotary_emb.inv_freq": "pytorch_model-00002-of-00003.bin",
+    "model.layers.14.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.15.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.15.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.15.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.15.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.15.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.15.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.15.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.15.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.15.self_attn.rotary_emb.inv_freq": "pytorch_model-00002-of-00003.bin",
+    "model.layers.15.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.16.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.16.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.16.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.16.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.16.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.16.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.16.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.16.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.16.self_attn.rotary_emb.inv_freq": "pytorch_model-00002-of-00003.bin",
+    "model.layers.16.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.17.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.17.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.17.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.17.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.17.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.17.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.17.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.17.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.17.self_attn.rotary_emb.inv_freq": "pytorch_model-00002-of-00003.bin",
+    "model.layers.17.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.18.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.18.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.18.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.18.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.18.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.18.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.18.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.18.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.18.self_attn.rotary_emb.inv_freq": "pytorch_model-00002-of-00003.bin",
+    "model.layers.18.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.19.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.19.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.19.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.19.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.19.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.19.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.19.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.19.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.19.self_attn.rotary_emb.inv_freq": "pytorch_model-00002-of-00003.bin",
+    "model.layers.19.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.2.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.2.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.2.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.2.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.2.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.2.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.2.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.2.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.2.self_attn.rotary_emb.inv_freq": "pytorch_model-00001-of-00003.bin",
+    "model.layers.2.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.20.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.20.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.20.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.20.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.20.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.20.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.20.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.20.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.20.self_attn.rotary_emb.inv_freq": "pytorch_model-00002-of-00003.bin",
+    "model.layers.20.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.21.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.21.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.21.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.21.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.21.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.21.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.21.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.21.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.21.self_attn.rotary_emb.inv_freq": "pytorch_model-00002-of-00003.bin",
+    "model.layers.21.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.22.input_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.22.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.22.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.22.mlp.up_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.22.post_attention_layernorm.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.22.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.22.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.22.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.22.self_attn.rotary_emb.inv_freq": "pytorch_model-00002-of-00003.bin",
+    "model.layers.22.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.23.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.23.mlp.down_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.23.mlp.gate_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.23.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.23.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.23.self_attn.k_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.23.self_attn.o_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.23.self_attn.q_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.23.self_attn.rotary_emb.inv_freq": "pytorch_model-00002-of-00003.bin",
+    "model.layers.23.self_attn.v_proj.weight": "pytorch_model-00002-of-00003.bin",
+    "model.layers.24.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.24.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.24.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.24.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.24.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.24.self_attn.k_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.24.self_attn.o_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.24.self_attn.q_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.24.self_attn.rotary_emb.inv_freq": "pytorch_model-00003-of-00003.bin",
+    "model.layers.24.self_attn.v_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.25.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.25.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.25.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.25.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.25.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.25.self_attn.k_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.25.self_attn.o_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.25.self_attn.q_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.25.self_attn.rotary_emb.inv_freq": "pytorch_model-00003-of-00003.bin",
+    "model.layers.25.self_attn.v_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.26.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.26.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.26.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.26.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.26.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.26.self_attn.k_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.26.self_attn.o_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.26.self_attn.q_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.26.self_attn.rotary_emb.inv_freq": "pytorch_model-00003-of-00003.bin",
+    "model.layers.26.self_attn.v_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.27.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.27.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.27.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.27.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.27.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.27.self_attn.k_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.27.self_attn.o_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.27.self_attn.q_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.27.self_attn.rotary_emb.inv_freq": "pytorch_model-00003-of-00003.bin",
+    "model.layers.27.self_attn.v_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.28.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.28.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.28.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.28.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.28.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.28.self_attn.k_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.28.self_attn.o_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.28.self_attn.q_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.28.self_attn.rotary_emb.inv_freq": "pytorch_model-00003-of-00003.bin",
+    "model.layers.28.self_attn.v_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.29.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.29.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.29.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.29.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.29.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.29.self_attn.k_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.29.self_attn.o_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.29.self_attn.q_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.29.self_attn.rotary_emb.inv_freq": "pytorch_model-00003-of-00003.bin",
+    "model.layers.29.self_attn.v_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.3.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.3.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.3.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.3.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.3.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.3.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.3.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.3.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.3.self_attn.rotary_emb.inv_freq": "pytorch_model-00001-of-00003.bin",
+    "model.layers.3.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.30.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.30.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.30.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.30.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.30.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.30.self_attn.k_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.30.self_attn.o_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.30.self_attn.q_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.30.self_attn.rotary_emb.inv_freq": "pytorch_model-00003-of-00003.bin",
+    "model.layers.30.self_attn.v_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.31.input_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.31.mlp.down_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.31.mlp.gate_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.31.mlp.up_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.31.post_attention_layernorm.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.31.self_attn.k_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.31.self_attn.o_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.31.self_attn.q_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.31.self_attn.rotary_emb.inv_freq": "pytorch_model-00003-of-00003.bin",
+    "model.layers.31.self_attn.v_proj.weight": "pytorch_model-00003-of-00003.bin",
+    "model.layers.4.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.4.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.4.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.4.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.4.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.4.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.4.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.4.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.4.self_attn.rotary_emb.inv_freq": "pytorch_model-00001-of-00003.bin",
+    "model.layers.4.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.5.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.5.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.5.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.5.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.5.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.5.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.5.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.5.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.5.self_attn.rotary_emb.inv_freq": "pytorch_model-00001-of-00003.bin",
+    "model.layers.5.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.6.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.6.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.6.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.6.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.6.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.6.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.6.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.6.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.6.self_attn.rotary_emb.inv_freq": "pytorch_model-00001-of-00003.bin",
+    "model.layers.6.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.7.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.7.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.7.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.7.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.7.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.7.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.7.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.7.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.7.self_attn.rotary_emb.inv_freq": "pytorch_model-00001-of-00003.bin",
+    "model.layers.7.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.8.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.8.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.8.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.8.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.8.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.8.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.8.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.8.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.8.self_attn.rotary_emb.inv_freq": "pytorch_model-00001-of-00003.bin",
+    "model.layers.8.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.9.input_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.9.mlp.down_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.9.mlp.gate_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.9.mlp.up_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.9.post_attention_layernorm.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.9.self_attn.k_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.9.self_attn.o_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.9.self_attn.q_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.layers.9.self_attn.rotary_emb.inv_freq": "pytorch_model-00001-of-00003.bin",
+    "model.layers.9.self_attn.v_proj.weight": "pytorch_model-00001-of-00003.bin",
+    "model.norm.weight": "pytorch_model-00003-of-00003.bin"
+  }
+}

checkpoint-500/rng_state_0.pth

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

checkpoint-500/rng_state_1.pth

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

checkpoint-500/rng_state_2.pth

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

checkpoint-500/rng_state_3.pth

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

checkpoint-500/scheduler.pt

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

checkpoint-500/special_tokens_map.json

@@ -0,0 +1,6 @@
+{
+  "bos_token": "</s>",
+  "eos_token": "</s>",
+  "pad_token": "[PAD]",
+  "unk_token": "</s>"
+}

checkpoint-500/tokenizer.model

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

checkpoint-500/tokenizer_config.json

@@ -0,0 +1,9 @@
+{
+  "bos_token": "",
+  "eos_token": "",
+  "model_max_length": 1024,
+  "padding_side": "right",
+  "special_tokens_map_file": "/home/ubuntu/.cache/huggingface/hub/models--decapoda-research--llama-7b-hf/snapshots/5f98eefcc80e437ef68d457ad7bf167c2c6a1348/special_tokens_map.json",
+  "tokenizer_class": "LLaMATokenizer",
+  "unk_token": ""
+}

checkpoint-500/trainer_state.json

@@ -0,0 +1,3016 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.32674399607907206,
+  "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": 4.347826086956522e-07,
+      "loss": 1.658,
+      "step": 1
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 8.695652173913044e-07,
+      "loss": 1.5847,
+      "step": 2
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.3043478260869566e-06,
+      "loss": 1.6487,
+      "step": 3
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 1.7391304347826088e-06,
+      "loss": 1.6376,
+      "step": 4
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 2.173913043478261e-06,
+      "loss": 1.5046,
+      "step": 5
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 2.6086956521739132e-06,
+      "loss": 1.439,
+      "step": 6
+    },
+    {
+      "epoch": 0.0,
+      "learning_rate": 3.043478260869566e-06,
+      "loss": 1.4233,
+      "step": 7
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 3.4782608695652175e-06,
+      "loss": 1.4303,
+      "step": 8
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 3.91304347826087e-06,
+      "loss": 1.4826,
+      "step": 9
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 4.347826086956522e-06,
+      "loss": 1.3716,
+      "step": 10
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 4.782608695652174e-06,
+      "loss": 1.3658,
+      "step": 11
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 5.2173913043478265e-06,
+      "loss": 1.3963,
+      "step": 12
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 5.652173913043479e-06,
+      "loss": 1.4064,
+      "step": 13
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 6.086956521739132e-06,
+      "loss": 1.3534,
+      "step": 14
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 6.521739130434783e-06,
+      "loss": 1.3417,
+      "step": 15
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 6.956521739130435e-06,
+      "loss": 1.3738,
+      "step": 16
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 7.391304347826087e-06,
+      "loss": 1.2103,
+      "step": 17
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 7.82608695652174e-06,
+      "loss": 1.2787,
+      "step": 18
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 8.260869565217392e-06,
+      "loss": 1.4267,
+      "step": 19
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 8.695652173913044e-06,
+      "loss": 1.3208,
+      "step": 20
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 9.130434782608697e-06,
+      "loss": 1.2949,
+      "step": 21
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 9.565217391304349e-06,
+      "loss": 1.3164,
+      "step": 22
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1e-05,
+      "loss": 1.3789,
+      "step": 23
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.0434782608695653e-05,
+      "loss": 1.3058,
+      "step": 24
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.0869565217391305e-05,
+      "loss": 1.3274,
+      "step": 25
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.1304347826086957e-05,
+      "loss": 1.3245,
+      "step": 26
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.1739130434782611e-05,
+      "loss": 1.3816,
+      "step": 27
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.2173913043478263e-05,
+      "loss": 1.3398,
+      "step": 28
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.2608695652173915e-05,
+      "loss": 1.3014,
+      "step": 29
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.3043478260869566e-05,
+      "loss": 1.2833,
+      "step": 30
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.3478260869565218e-05,
+      "loss": 1.3021,
+      "step": 31
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.391304347826087e-05,
+      "loss": 1.2483,
+      "step": 32
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.4347826086956522e-05,
+      "loss": 1.197,
+      "step": 33
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.4782608695652174e-05,
+      "loss": 1.2796,
+      "step": 34
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.5217391304347828e-05,
+      "loss": 1.2436,
+      "step": 35
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.565217391304348e-05,
+      "loss": 1.3143,
+      "step": 36
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.6086956521739132e-05,
+      "loss": 1.2572,
+      "step": 37
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 1.6521739130434785e-05,
+      "loss": 1.2133,
+      "step": 38
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.6956521739130437e-05,
+      "loss": 1.1562,
+      "step": 39
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.739130434782609e-05,
+      "loss": 1.2607,
+      "step": 40
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.782608695652174e-05,
+      "loss": 1.2708,
+      "step": 41
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.8260869565217393e-05,
+      "loss": 1.2855,
+      "step": 42
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.8695652173913045e-05,
+      "loss": 1.2714,
+      "step": 43
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.9130434782608697e-05,
+      "loss": 1.2747,
+      "step": 44
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.956521739130435e-05,
+      "loss": 1.2935,
+      "step": 45
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 2e-05,
+      "loss": 1.3802,
+      "step": 46
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.9999977592067603e-05,
+      "loss": 1.3119,
+      "step": 47
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.9999910368370826e-05,
+      "loss": 1.3561,
+      "step": 48
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.9999798329210938e-05,
+      "loss": 1.3027,
+      "step": 49
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.999964147509006e-05,
+      "loss": 1.3295,
+      "step": 50
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.9999439806711137e-05,
+      "loss": 1.2899,
+      "step": 51
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.9999193324977974e-05,
+      "loss": 1.3297,
+      "step": 52
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 1.999890203099519e-05,
+      "loss": 1.3059,
+      "step": 53
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9998565926068253e-05,
+      "loss": 1.29,
+      "step": 54
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.999818501170344e-05,
+      "loss": 1.2265,
+      "step": 55
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9997759289607854e-05,
+      "loss": 1.2652,
+      "step": 56
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9997288761689404e-05,
+      "loss": 1.3858,
+      "step": 57
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9996773430056806e-05,
+      "loss": 1.3069,
+      "step": 58
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9996213297019556e-05,
+      "loss": 1.2414,
+      "step": 59
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9995608365087945e-05,
+      "loss": 1.3432,
+      "step": 60
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9994958636973024e-05,
+      "loss": 1.4153,
+      "step": 61
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.999426411558661e-05,
+      "loss": 1.3386,
+      "step": 62
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9993524804041255e-05,
+      "loss": 1.2682,
+      "step": 63
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9992740705650252e-05,
+      "loss": 1.3186,
+      "step": 64
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9991911823927607e-05,
+      "loss": 1.4159,
+      "step": 65
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9991038162588018e-05,
+      "loss": 1.2346,
+      "step": 66
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.999011972554688e-05,
+      "loss": 1.3291,
+      "step": 67
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 1.9989156516920248e-05,
+      "loss": 1.2836,
+      "step": 68
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.998814854102482e-05,
+      "loss": 1.2737,
+      "step": 69
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.9987095802377933e-05,
+      "loss": 1.3478,
+      "step": 70
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.998599830569752e-05,
+      "loss": 1.391,
+      "step": 71
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.998485605590211e-05,
+      "loss": 1.2899,
+      "step": 72
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.99836690581108e-05,
+      "loss": 1.3001,
+      "step": 73
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.9982437317643218e-05,
+      "loss": 1.2358,
+      "step": 74
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.9981160840019513e-05,
+      "loss": 1.2875,
+      "step": 75
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.9979839630960333e-05,
+      "loss": 1.2769,
+      "step": 76
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.9978473696386787e-05,
+      "loss": 1.3458,
+      "step": 77
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.9977063042420438e-05,
+      "loss": 1.3433,
+      "step": 78
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.9975607675383243e-05,
+      "loss": 1.3246,
+      "step": 79
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.997410760179756e-05,
+      "loss": 1.2984,
+      "step": 80
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.99725628283861e-05,
+      "loss": 1.2665,
+      "step": 81
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.99709733620719e-05,
+      "loss": 1.3427,
+      "step": 82
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.9969339209978287e-05,
+      "loss": 1.3314,
+      "step": 83
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 1.9967660379428855e-05,
+      "loss": 1.2411,
+      "step": 84
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.996593687794743e-05,
+      "loss": 1.212,
+      "step": 85
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.996416871325803e-05,
+      "loss": 1.3132,
+      "step": 86
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.996235589328484e-05,
+      "loss": 1.2823,
+      "step": 87
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.996049842615217e-05,
+      "loss": 1.4153,
+      "step": 88
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.995859632018442e-05,
+      "loss": 1.3212,
+      "step": 89
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.995664958390604e-05,
+      "loss": 1.1837,
+      "step": 90
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.99546582260415e-05,
+      "loss": 1.3189,
+      "step": 91
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.995262225551524e-05,
+      "loss": 1.2477,
+      "step": 92
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.9950541681451644e-05,
+      "loss": 1.248,
+      "step": 93
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.9948416513174976e-05,
+      "loss": 1.2963,
+      "step": 94
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.994624676020936e-05,
+      "loss": 1.2874,
+      "step": 95
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.9944032432278743e-05,
+      "loss": 1.2947,
+      "step": 96
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.9941773539306818e-05,
+      "loss": 1.309,
+      "step": 97
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.9939470091417012e-05,
+      "loss": 1.3141,
+      "step": 98
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 1.9937122098932428e-05,
+      "loss": 1.2828,
+      "step": 99
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.9934729572375792e-05,
+      "loss": 1.3323,
+      "step": 100
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.9932292522469424e-05,
+      "loss": 1.2671,
+      "step": 101
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.992981096013517e-05,
+      "loss": 1.3129,
+      "step": 102
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.992728489649437e-05,
+      "loss": 1.2566,
+      "step": 103
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.992471434286779e-05,
+      "loss": 1.279,
+      "step": 104
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.99220993107756e-05,
+      "loss": 1.2325,
+      "step": 105
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.9919439811937283e-05,
+      "loss": 1.3848,
+      "step": 106
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.9916735858271615e-05,
+      "loss": 1.2736,
+      "step": 107
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.9913987461896597e-05,
+      "loss": 1.2252,
+      "step": 108
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.991119463512941e-05,
+      "loss": 1.2248,
+      "step": 109
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.9908357390486342e-05,
+      "loss": 1.253,
+      "step": 110
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.9905475740682756e-05,
+      "loss": 1.256,
+      "step": 111
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.990254969863301e-05,
+      "loss": 1.3436,
+      "step": 112
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.9899579277450417e-05,
+      "loss": 1.1671,
+      "step": 113
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 1.989656449044718e-05,
+      "loss": 1.3457,
+      "step": 114
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9893505351134322e-05,
+      "loss": 1.315,
+      "step": 115
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9890401873221642e-05,
+      "loss": 1.2173,
+      "step": 116
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9887254070617644e-05,
+      "loss": 1.3156,
+      "step": 117
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.988406195742948e-05,
+      "loss": 1.3153,
+      "step": 118
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9880825547962882e-05,
+      "loss": 1.2818,
+      "step": 119
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9877544856722094e-05,
+      "loss": 1.3481,
+      "step": 120
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.987421989840982e-05,
+      "loss": 1.3723,
+      "step": 121
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.987085068792715e-05,
+      "loss": 1.2784,
+      "step": 122
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.986743724037349e-05,
+      "loss": 1.2764,
+      "step": 123
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.98639795710465e-05,
+      "loss": 1.2896,
+      "step": 124
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.986047769544203e-05,
+      "loss": 1.3305,
+      "step": 125
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9856931629254032e-05,
+      "loss": 1.2695,
+      "step": 126
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9853341388374504e-05,
+      "loss": 1.2534,
+      "step": 127
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9849706988893433e-05,
+      "loss": 1.2319,
+      "step": 128
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.9846028447098687e-05,
+      "loss": 1.2202,
+      "step": 129
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.984230577947597e-05,
+      "loss": 1.3401,
+      "step": 130
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.983853900270874e-05,
+      "loss": 1.3309,
+      "step": 131
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.983472813367813e-05,
+      "loss": 1.3011,
+      "step": 132
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.9830873189462886e-05,
+      "loss": 1.3471,
+      "step": 133
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.9826974187339267e-05,
+      "loss": 1.2037,
+      "step": 134
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.982303114478099e-05,
+      "loss": 1.2936,
+      "step": 135
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.9819044079459142e-05,
+      "loss": 1.1976,
+      "step": 136
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.9815013009242103e-05,
+      "loss": 1.248,
+      "step": 137
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.981093795219546e-05,
+      "loss": 1.3496,
+      "step": 138
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.980681892658193e-05,
+      "loss": 1.2167,
+      "step": 139
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.980265595086129e-05,
+      "loss": 1.2875,
+      "step": 140
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.9798449043690272e-05,
+      "loss": 1.2559,
+      "step": 141
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.9794198223922496e-05,
+      "loss": 1.313,
+      "step": 142
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.9789903510608374e-05,
+      "loss": 1.2399,
+      "step": 143
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.9785564922995042e-05,
+      "loss": 1.2255,
+      "step": 144
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 1.978118248052625e-05,
+      "loss": 1.3483,
+      "step": 145
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.9776756202842297e-05,
+      "loss": 1.2305,
+      "step": 146
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.977228610977992e-05,
+      "loss": 1.2741,
+      "step": 147
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.976777222137224e-05,
+      "loss": 1.2095,
+      "step": 148
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.9763214557848634e-05,
+      "loss": 1.2364,
+      "step": 149
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.9758613139634662e-05,
+      "loss": 1.3913,
+      "step": 150
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.975396798735198e-05,
+      "loss": 1.2574,
+      "step": 151
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.9749279121818235e-05,
+      "loss": 1.2894,
+      "step": 152
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.9744546564046995e-05,
+      "loss": 1.3109,
+      "step": 153
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.9739770335247616e-05,
+      "loss": 1.3483,
+      "step": 154
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.9734950456825187e-05,
+      "loss": 1.3476,
+      "step": 155
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.9730086950380404e-05,
+      "loss": 1.2305,
+      "step": 156
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.9725179837709494e-05,
+      "loss": 1.2062,
+      "step": 157
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.972022914080411e-05,
+      "loss": 1.2074,
+      "step": 158
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.9715234881851226e-05,
+      "loss": 1.2691,
+      "step": 159
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 1.9710197083233044e-05,
+      "loss": 1.2515,
+      "step": 160
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.9705115767526894e-05,
+      "loss": 1.2018,
+      "step": 161
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.9699990957505136e-05,
+      "loss": 1.1156,
+      "step": 162
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.9694822676135045e-05,
+      "loss": 1.2945,
+      "step": 163
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.9689610946578725e-05,
+      "loss": 1.2831,
+      "step": 164
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.968435579219299e-05,
+      "loss": 1.2824,
+      "step": 165
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.9679057236529266e-05,
+      "loss": 1.2378,
+      "step": 166
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.9673715303333495e-05,
+      "loss": 1.2443,
+      "step": 167
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.9668330016546004e-05,
+      "loss": 1.3786,
+      "step": 168
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.9662901400301432e-05,
+      "loss": 1.2761,
+      "step": 169
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.965742947892858e-05,
+      "loss": 1.2731,
+      "step": 170
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.965191427695035e-05,
+      "loss": 1.2325,
+      "step": 171
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.964635581908359e-05,
+      "loss": 1.4167,
+      "step": 172
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.9640754130239004e-05,
+      "loss": 1.2157,
+      "step": 173
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.9635109235521057e-05,
+      "loss": 1.2924,
+      "step": 174
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.9629421160227825e-05,
+      "loss": 1.3156,
+      "step": 175
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.962368992985091e-05,
+      "loss": 1.3434,
+      "step": 176
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.9617915570075317e-05,
+      "loss": 1.3234,
+      "step": 177
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.961209810677934e-05,
+      "loss": 1.2945,
+      "step": 178
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.9606237566034443e-05,
+      "loss": 1.277,
+      "step": 179
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.9600333974105145e-05,
+      "loss": 1.2148,
+      "step": 180
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.9594387357448908e-05,
+      "loss": 1.2183,
+      "step": 181
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.9588397742716004e-05,
+      "loss": 1.2559,
+      "step": 182
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.958236515674941e-05,
+      "loss": 1.3053,
+      "step": 183
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.9576289626584685e-05,
+      "loss": 1.286,
+      "step": 184
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.957017117944984e-05,
+      "loss": 1.242,
+      "step": 185
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.9564009842765225e-05,
+      "loss": 1.1833,
+      "step": 186
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.9557805644143408e-05,
+      "loss": 1.2265,
+      "step": 187
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.955155861138903e-05,
+      "loss": 1.2032,
+      "step": 188
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.9545268772498723e-05,
+      "loss": 1.3044,
+      "step": 189
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.9538936155660934e-05,
+      "loss": 1.3028,
+      "step": 190
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 1.9532560789255838e-05,
+      "loss": 1.3328,
+      "step": 191
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.9526142701855187e-05,
+      "loss": 1.2071,
+      "step": 192
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.9519681922222195e-05,
+      "loss": 1.3534,
+      "step": 193
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.951317847931141e-05,
+      "loss": 1.2878,
+      "step": 194
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.950663240226857e-05,
+      "loss": 1.2647,
+      "step": 195
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.9500043720430484e-05,
+      "loss": 1.2338,
+      "step": 196
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.9493412463324903e-05,
+      "loss": 1.3765,
+      "step": 197
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.9486738660670373e-05,
+      "loss": 1.1946,
+      "step": 198
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.9480022342376125e-05,
+      "loss": 1.2972,
+      "step": 199
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.9473263538541916e-05,
+      "loss": 1.2961,
+      "step": 200
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.946646227945791e-05,
+      "loss": 1.2324,
+      "step": 201
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.945961859560454e-05,
+      "loss": 1.1474,
+      "step": 202
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.9452732517652366e-05,
+      "loss": 1.359,
+      "step": 203
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.9445804076461946e-05,
+      "loss": 1.3617,
+      "step": 204
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.9438833303083677e-05,
+      "loss": 1.3684,
+      "step": 205
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 1.943182022875769e-05,
+      "loss": 1.2624,
+      "step": 206
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.942476488491369e-05,
+      "loss": 1.2837,
+      "step": 207
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.9417667303170803e-05,
+      "loss": 1.2425,
+      "step": 208
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.941052751533745e-05,
+      "loss": 1.3908,
+      "step": 209
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.940334555341122e-05,
+      "loss": 1.1629,
+      "step": 210
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.9396121449578694e-05,
+      "loss": 1.225,
+      "step": 211
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.938885523621531e-05,
+      "loss": 1.2935,
+      "step": 212
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.9381546945885236e-05,
+      "loss": 1.2493,
+      "step": 213
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.9374196611341212e-05,
+      "loss": 1.1924,
+      "step": 214
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.9366804265524388e-05,
+      "loss": 1.1538,
+      "step": 215
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.935936994156421e-05,
+      "loss": 1.3094,
+      "step": 216
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.935189367277824e-05,
+      "loss": 1.2147,
+      "step": 217
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.9344375492672024e-05,
+      "loss": 1.1849,
+      "step": 218
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.9336815434938935e-05,
+      "loss": 1.2968,
+      "step": 219
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.932921353346003e-05,
+      "loss": 1.2963,
+      "step": 220
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 1.932156982230388e-05,
+      "loss": 1.279,
+      "step": 221
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.9313884335726443e-05,
+      "loss": 1.194,
+      "step": 222
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.9306157108170892e-05,
+      "loss": 1.2834,
+      "step": 223
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.9298388174267463e-05,
+      "loss": 1.1519,
+      "step": 224
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.9290577568833305e-05,
+      "loss": 1.3101,
+      "step": 225
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.9282725326872324e-05,
+      "loss": 1.2542,
+      "step": 226
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.927483148357502e-05,
+      "loss": 1.3028,
+      "step": 227
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.9266896074318335e-05,
+      "loss": 1.2972,
+      "step": 228
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.9258919134665494e-05,
+      "loss": 1.1933,
+      "step": 229
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.9250900700365837e-05,
+      "loss": 1.3796,
+      "step": 230
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.9242840807354676e-05,
+      "loss": 1.271,
+      "step": 231
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.923473949175311e-05,
+      "loss": 1.192,
+      "step": 232
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.9226596789867895e-05,
+      "loss": 1.3255,
+      "step": 233
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.921841273819125e-05,
+      "loss": 1.2359,
+      "step": 234
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.921018737340071e-05,
+      "loss": 1.3459,
+      "step": 235
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.920192073235896e-05,
+      "loss": 1.1901,
+      "step": 236
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 1.9193612852113666e-05,
+      "loss": 1.1438,
+      "step": 237
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.918526376989731e-05,
+      "loss": 1.2968,
+      "step": 238
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.9176873523127025e-05,
+      "loss": 1.2323,
+      "step": 239
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.9168442149404438e-05,
+      "loss": 1.3393,
+      "step": 240
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.9159969686515462e-05,
+      "loss": 1.2142,
+      "step": 241
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.9151456172430186e-05,
+      "loss": 1.1371,
+      "step": 242
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.9142901645302654e-05,
+      "loss": 1.2626,
+      "step": 243
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.9134306143470722e-05,
+      "loss": 1.253,
+      "step": 244
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.912566970545587e-05,
+      "loss": 1.204,
+      "step": 245
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.911699236996305e-05,
+      "loss": 1.2175,
+      "step": 246
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.9108274175880483e-05,
+      "loss": 1.3093,
+      "step": 247
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.9099515162279515e-05,
+      "loss": 1.2699,
+      "step": 248
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.909071536841442e-05,
+      "loss": 1.307,
+      "step": 249
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.9081874833722234e-05,
+      "loss": 1.2227,
+      "step": 250
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.9072993597822586e-05,
+      "loss": 1.1945,
+      "step": 251
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 1.906407170051749e-05,
+      "loss": 1.2266,
+      "step": 252
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.905510918179121e-05,
+      "loss": 1.2669,
+      "step": 253
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.9046106081810047e-05,
+      "loss": 1.2756,
+      "step": 254
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.903706244092217e-05,
+      "loss": 1.2515,
+      "step": 255
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.9027978299657436e-05,
+      "loss": 1.2266,
+      "step": 256
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.9018853698727216e-05,
+      "loss": 1.3235,
+      "step": 257
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.900968867902419e-05,
+      "loss": 1.2487,
+      "step": 258
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.9000483281622198e-05,
+      "loss": 1.2412,
+      "step": 259
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.8991237547776014e-05,
+      "loss": 1.1866,
+      "step": 260
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.8981951518921194e-05,
+      "loss": 1.2566,
+      "step": 261
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.8972625236673887e-05,
+      "loss": 1.2875,
+      "step": 262
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.896325874283063e-05,
+      "loss": 1.1893,
+      "step": 263
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.895385207936817e-05,
+      "loss": 1.3017,
+      "step": 264
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.894440528844329e-05,
+      "loss": 1.2351,
+      "step": 265
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.8934918412392596e-05,
+      "loss": 1.3018,
+      "step": 266
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 1.8925391493732346e-05,
+      "loss": 1.3364,
+      "step": 267
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.891582457515825e-05,
+      "loss": 1.2788,
+      "step": 268
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.890621769954528e-05,
+      "loss": 1.2523,
+      "step": 269
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8896570909947477e-05,
+      "loss": 1.2419,
+      "step": 270
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8886884249597763e-05,
+      "loss": 1.1993,
+      "step": 271
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.887715776190775e-05,
+      "loss": 1.2581,
+      "step": 272
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8867391490467534e-05,
+      "loss": 1.1852,
+      "step": 273
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8857585479045493e-05,
+      "loss": 1.2443,
+      "step": 274
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.884773977158813e-05,
+      "loss": 1.2524,
+      "step": 275
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8837854412219828e-05,
+      "loss": 1.1538,
+      "step": 276
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8827929445242678e-05,
+      "loss": 1.2405,
+      "step": 277
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8817964915136277e-05,
+      "loss": 1.1908,
+      "step": 278
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8807960866557535e-05,
+      "loss": 1.2412,
+      "step": 279
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8797917344340453e-05,
+      "loss": 1.2212,
+      "step": 280
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8787834393495952e-05,
+      "loss": 1.2132,
+      "step": 281
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8777712059211643e-05,
+      "loss": 1.2642,
+      "step": 282
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 1.8767550386851646e-05,
+      "loss": 1.2226,
+      "step": 283
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.875734942195637e-05,
+      "loss": 1.2801,
+      "step": 284
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8747109210242326e-05,
+      "loss": 1.3122,
+      "step": 285
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8736829797601903e-05,
+      "loss": 1.2728,
+      "step": 286
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8726511230103183e-05,
+      "loss": 1.2868,
+      "step": 287
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8716153553989716e-05,
+      "loss": 1.1839,
+      "step": 288
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8705756815680324e-05,
+      "loss": 1.1724,
+      "step": 289
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8695321061768886e-05,
+      "loss": 1.1994,
+      "step": 290
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8684846339024145e-05,
+      "loss": 1.2004,
+      "step": 291
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8674332694389464e-05,
+      "loss": 1.1944,
+      "step": 292
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8663780174982656e-05,
+      "loss": 1.2524,
+      "step": 293
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8653188828095754e-05,
+      "loss": 1.3951,
+      "step": 294
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8642558701194787e-05,
+      "loss": 1.3361,
+      "step": 295
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8631889841919596e-05,
+      "loss": 1.198,
+      "step": 296
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.862118229808359e-05,
+      "loss": 1.2669,
+      "step": 297
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 1.8610436117673557e-05,
+      "loss": 1.2139,
+      "step": 298
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.859965134884943e-05,
+      "loss": 1.3054,
+      "step": 299
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.8588828039944086e-05,
+      "loss": 1.239,
+      "step": 300
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.857796623946312e-05,
+      "loss": 1.2179,
+      "step": 301
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.8567065996084628e-05,
+      "loss": 1.1934,
+      "step": 302
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.8556127358658995e-05,
+      "loss": 1.2408,
+      "step": 303
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.854515037620867e-05,
+      "loss": 1.3218,
+      "step": 304
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.853413509792795e-05,
+      "loss": 1.2498,
+      "step": 305
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.8523081573182754e-05,
+      "loss": 1.1693,
+      "step": 306
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.8511989851510416e-05,
+      "loss": 1.1536,
+      "step": 307
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.8500859982619438e-05,
+      "loss": 1.275,
+      "step": 308
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.848969201638929e-05,
+      "loss": 1.2712,
+      "step": 309
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.847848600287019e-05,
+      "loss": 1.266,
+      "step": 310
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.8467241992282842e-05,
+      "loss": 1.2122,
+      "step": 311
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.845596003501826e-05,
+      "loss": 1.2051,
+      "step": 312
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 1.8444640181637514e-05,
+      "loss": 1.2724,
+      "step": 313
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.8433282482871497e-05,
+      "loss": 1.2526,
+      "step": 314
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.8421886989620724e-05,
+      "loss": 1.1622,
+      "step": 315
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.841045375295508e-05,
+      "loss": 1.2903,
+      "step": 316
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.839898282411361e-05,
+      "loss": 1.2329,
+      "step": 317
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.8387474254504265e-05,
+      "loss": 1.2136,
+      "step": 318
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.8375928095703703e-05,
+      "loss": 1.2846,
+      "step": 319
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.8364344399457028e-05,
+      "loss": 1.2696,
+      "step": 320
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.835272321767758e-05,
+      "loss": 1.187,
+      "step": 321
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.8341064602446686e-05,
+      "loss": 1.2331,
+      "step": 322
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.832936860601344e-05,
+      "loss": 1.2724,
+      "step": 323
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.8317635280794466e-05,
+      "loss": 1.2181,
+      "step": 324
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.8305864679373667e-05,
+      "loss": 1.1992,
+      "step": 325
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.829405685450202e-05,
+      "loss": 1.1572,
+      "step": 326
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.8282211859097308e-05,
+      "loss": 1.1881,
+      "step": 327
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.8270329746243903e-05,
+      "loss": 1.2213,
+      "step": 328
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 1.8258410569192526e-05,
+      "loss": 1.2199,
+      "step": 329
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.824645438135999e-05,
+      "loss": 1.2589,
+      "step": 330
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.8234461236328992e-05,
+      "loss": 1.1814,
+      "step": 331
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.822243118784785e-05,
+      "loss": 1.1442,
+      "step": 332
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.821036428983026e-05,
+      "loss": 1.26,
+      "step": 333
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.8198260596355077e-05,
+      "loss": 1.268,
+      "step": 334
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.818612016166604e-05,
+      "loss": 1.131,
+      "step": 335
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.8173943040171567e-05,
+      "loss": 1.2818,
+      "step": 336
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.8161729286444474e-05,
+      "loss": 1.1387,
+      "step": 337
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.814947895522176e-05,
+      "loss": 1.2065,
+      "step": 338
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.8137192101404343e-05,
+      "loss": 1.2687,
+      "step": 339
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.8124868780056814e-05,
+      "loss": 1.2522,
+      "step": 340
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.811250904640721e-05,
+      "loss": 1.1396,
+      "step": 341
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.8100112955846746e-05,
+      "loss": 1.2635,
+      "step": 342
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.8087680563929574e-05,
+      "loss": 1.2622,
+      "step": 343
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 1.807521192637253e-05,
+      "loss": 1.0812,
+      "step": 344
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.806270709905489e-05,
+      "loss": 1.1504,
+      "step": 345
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.805016613801813e-05,
+      "loss": 1.246,
+      "step": 346
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.8037589099465637e-05,
+      "loss": 1.2893,
+      "step": 347
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.8024976039762507e-05,
+      "loss": 1.228,
+      "step": 348
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.8012327015435255e-05,
+      "loss": 1.1992,
+      "step": 349
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.7999642083171576e-05,
+      "loss": 1.2022,
+      "step": 350
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.798692129982009e-05,
+      "loss": 1.1628,
+      "step": 351
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.7974164722390094e-05,
+      "loss": 1.3536,
+      "step": 352
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.796137240805129e-05,
+      "loss": 1.228,
+      "step": 353
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.7948544414133534e-05,
+      "loss": 1.2561,
+      "step": 354
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.7935680798126597e-05,
+      "loss": 1.1878,
+      "step": 355
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.792278161767989e-05,
+      "loss": 1.3139,
+      "step": 356
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.79098469306022e-05,
+      "loss": 1.2356,
+      "step": 357
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.7896876794861443e-05,
+      "loss": 1.2904,
+      "step": 358
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 1.788387126858441e-05,
+      "loss": 1.3259,
+      "step": 359
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.7870830410056487e-05,
+      "loss": 1.2614,
+      "step": 360
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.7857754277721417e-05,
+      "loss": 1.2127,
+      "step": 361
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.7844642930181008e-05,
+      "loss": 1.2234,
+      "step": 362
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.7831496426194906e-05,
+      "loss": 1.3165,
+      "step": 363
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.78183148246803e-05,
+      "loss": 1.1426,
+      "step": 364
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.780509818471168e-05,
+      "loss": 1.2525,
+      "step": 365
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.779184656552056e-05,
+      "loss": 1.2378,
+      "step": 366
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.7778560026495218e-05,
+      "loss": 1.2249,
+      "step": 367
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.7765238627180424e-05,
+      "loss": 1.1897,
+      "step": 368
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.775188242727719e-05,
+      "loss": 1.2608,
+      "step": 369
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.773849148664247e-05,
+      "loss": 1.2953,
+      "step": 370
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.772506586528893e-05,
+      "loss": 1.2901,
+      "step": 371
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.7711605623384653e-05,
+      "loss": 1.2712,
+      "step": 372
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.7698110821252873e-05,
+      "loss": 1.2286,
+      "step": 373
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 1.7684581519371714e-05,
+      "loss": 1.2791,
+      "step": 374
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.7671017778373914e-05,
+      "loss": 1.2633,
+      "step": 375
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.7657419659046555e-05,
+      "loss": 1.14,
+      "step": 376
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.7643787222330773e-05,
+      "loss": 1.2696,
+      "step": 377
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.7630120529321518e-05,
+      "loss": 1.1512,
+      "step": 378
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.7616419641267267e-05,
+      "loss": 1.1531,
+      "step": 379
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.760268461956972e-05,
+      "loss": 1.1738,
+      "step": 380
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.758891552578357e-05,
+      "loss": 1.2082,
+      "step": 381
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.7575112421616203e-05,
+      "loss": 1.1992,
+      "step": 382
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.7561275368927422e-05,
+      "loss": 1.2195,
+      "step": 383
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.7547404429729174e-05,
+      "loss": 1.1806,
+      "step": 384
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.753349966618528e-05,
+      "loss": 1.1613,
+      "step": 385
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.751956114061113e-05,
+      "loss": 1.2528,
+      "step": 386
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.7505588915473437e-05,
+      "loss": 1.2632,
+      "step": 387
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.7491583053389937e-05,
+      "loss": 1.216,
+      "step": 388
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.747754361712911e-05,
+      "loss": 1.1966,
+      "step": 389
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 1.7463470669609907e-05,
+      "loss": 1.2612,
+      "step": 390
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.7449364273901457e-05,
+      "loss": 1.1057,
+      "step": 391
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.743522449322279e-05,
+      "loss": 1.1335,
+      "step": 392
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.742105139094256e-05,
+      "loss": 1.2405,
+      "step": 393
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.7406845030578747e-05,
+      "loss": 1.2579,
+      "step": 394
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.7392605475798387e-05,
+      "loss": 1.0973,
+      "step": 395
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.7378332790417275e-05,
+      "loss": 1.1534,
+      "step": 396
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.7364027038399682e-05,
+      "loss": 1.216,
+      "step": 397
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.734968828385808e-05,
+      "loss": 1.2669,
+      "step": 398
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.7335316591052828e-05,
+      "loss": 1.1461,
+      "step": 399
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.7320912024391915e-05,
+      "loss": 1.2305,
+      "step": 400
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.7306474648430653e-05,
+      "loss": 1.2888,
+      "step": 401
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.729200452787139e-05,
+      "loss": 1.3148,
+      "step": 402
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.7277501727563227e-05,
+      "loss": 1.1997,
+      "step": 403
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.726296631250171e-05,
+      "loss": 1.208,
+      "step": 404
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 1.724839834782856e-05,
+      "loss": 1.2427,
+      "step": 405
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.7233797898831376e-05,
+      "loss": 1.2214,
+      "step": 406
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.7219165030943327e-05,
+      "loss": 1.1986,
+      "step": 407
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.720449980974288e-05,
+      "loss": 1.2567,
+      "step": 408
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.7189802300953488e-05,
+      "loss": 1.1098,
+      "step": 409
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.717507257044331e-05,
+      "loss": 1.1329,
+      "step": 410
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.716031068422491e-05,
+      "loss": 1.1693,
+      "step": 411
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.7145516708454953e-05,
+      "loss": 1.1304,
+      "step": 412
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.7130690709433927e-05,
+      "loss": 1.2287,
+      "step": 413
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.711583275360582e-05,
+      "loss": 1.2817,
+      "step": 414
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.710094290755785e-05,
+      "loss": 1.2225,
+      "step": 415
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.7086021238020154e-05,
+      "loss": 1.1936,
+      "step": 416
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.7071067811865477e-05,
+      "loss": 1.2964,
+      "step": 417
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.7056082696108896e-05,
+      "loss": 1.2376,
+      "step": 418
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.70410659579075e-05,
+      "loss": 1.1612,
+      "step": 419
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.7026017664560108e-05,
+      "loss": 1.3043,
+      "step": 420
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.7010937883506936e-05,
+      "loss": 1.2882,
+      "step": 421
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.699582668232934e-05,
+      "loss": 1.2565,
+      "step": 422
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.698068412874946e-05,
+      "loss": 1.1343,
+      "step": 423
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.6965510290629973e-05,
+      "loss": 1.2255,
+      "step": 424
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.695030523597374e-05,
+      "loss": 1.2678,
+      "step": 425
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.6935069032923525e-05,
+      "loss": 1.1335,
+      "step": 426
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.6919801749761696e-05,
+      "loss": 1.2225,
+      "step": 427
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.6904503454909905e-05,
+      "loss": 1.1734,
+      "step": 428
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.6889174216928776e-05,
+      "loss": 1.1718,
+      "step": 429
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.6873814104517617e-05,
+      "loss": 1.1448,
+      "step": 430
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.6858423186514108e-05,
+      "loss": 1.1668,
+      "step": 431
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.6843001531893967e-05,
+      "loss": 1.1733,
+      "step": 432
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.6827549209770676e-05,
+      "loss": 1.2871,
+      "step": 433
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.6812066289395157e-05,
+      "loss": 1.2404,
+      "step": 434
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.6796552840155453e-05,
+      "loss": 1.186,
+      "step": 435
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 1.6781008931576433e-05,
+      "loss": 1.3174,
+      "step": 436
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.676543463331946e-05,
+      "loss": 1.1726,
+      "step": 437
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.6749830015182106e-05,
+      "loss": 1.2188,
+      "step": 438
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.6734195147097815e-05,
+      "loss": 1.2124,
+      "step": 439
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.6718530099135596e-05,
+      "loss": 1.1971,
+      "step": 440
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.6702834941499718e-05,
+      "loss": 1.1957,
+      "step": 441
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.6687109744529394e-05,
+      "loss": 1.2282,
+      "step": 442
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.6671354578698444e-05,
+      "loss": 1.2366,
+      "step": 443
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.665556951461501e-05,
+      "loss": 1.1922,
+      "step": 444
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.6639754623021227e-05,
+      "loss": 1.257,
+      "step": 445
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.6623909974792888e-05,
+      "loss": 1.2759,
+      "step": 446
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.6608035640939165e-05,
+      "loss": 1.2557,
+      "step": 447
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.6592131692602257e-05,
+      "loss": 1.1747,
+      "step": 448
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.657619820105708e-05,
+      "loss": 1.2136,
+      "step": 449
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.656023523771095e-05,
+      "loss": 1.3245,
+      "step": 450
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 1.6544242874103278e-05,
+      "loss": 1.1963,
+      "step": 451
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6528221181905217e-05,
+      "loss": 1.2523,
+      "step": 452
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6512170232919372e-05,
+      "loss": 1.2043,
+      "step": 453
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6496090099079452e-05,
+      "loss": 1.2063,
+      "step": 454
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6479980852449974e-05,
+      "loss": 1.0912,
+      "step": 455
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6463842565225914e-05,
+      "loss": 1.1231,
+      "step": 456
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6447675309732403e-05,
+      "loss": 1.2476,
+      "step": 457
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.64314791584244e-05,
+      "loss": 1.1912,
+      "step": 458
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6415254183886356e-05,
+      "loss": 1.2297,
+      "step": 459
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6399000458831893e-05,
+      "loss": 1.1678,
+      "step": 460
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6382718056103486e-05,
+      "loss": 1.227,
+      "step": 461
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6366407048672135e-05,
+      "loss": 1.1352,
+      "step": 462
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6350067509637028e-05,
+      "loss": 1.2545,
+      "step": 463
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6333699512225224e-05,
+      "loss": 1.1916,
+      "step": 464
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6317303129791315e-05,
+      "loss": 1.1412,
+      "step": 465
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 1.6300878435817115e-05,
+      "loss": 1.2384,
+      "step": 466
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.62844255039113e-05,
+      "loss": 1.2379,
+      "step": 467
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.626794440780911e-05,
+      "loss": 1.1708,
+      "step": 468
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.6251435221372007e-05,
+      "loss": 1.1741,
+      "step": 469
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.6234898018587336e-05,
+      "loss": 1.14,
+      "step": 470
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.6218332873568e-05,
+      "loss": 1.1982,
+      "step": 471
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.620173986055213e-05,
+      "loss": 1.1246,
+      "step": 472
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.6185119053902748e-05,
+      "loss": 1.1792,
+      "step": 473
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.616847052810744e-05,
+      "loss": 1.2365,
+      "step": 474
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.6151794357778006e-05,
+      "loss": 1.1503,
+      "step": 475
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.613509061765015e-05,
+      "loss": 1.1668,
+      "step": 476
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.611835938258313e-05,
+      "loss": 1.1859,
+      "step": 477
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.6101600727559423e-05,
+      "loss": 1.1515,
+      "step": 478
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.6084814727684386e-05,
+      "loss": 1.1114,
+      "step": 479
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.6068001458185934e-05,
+      "loss": 1.1375,
+      "step": 480
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.6051160994414187e-05,
+      "loss": 1.2151,
+      "step": 481
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 1.603429341184114e-05,
+      "loss": 1.2092,
+      "step": 482
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.601739878606032e-05,
+      "loss": 1.1854,
+      "step": 483
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.600047719278645e-05,
+      "loss": 1.1809,
+      "step": 484
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.5983528707855128e-05,
+      "loss": 1.1968,
+      "step": 485
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.596655340722244e-05,
+      "loss": 1.1921,
+      "step": 486
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.5949551366964675e-05,
+      "loss": 1.1509,
+      "step": 487
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.593252266327794e-05,
+      "loss": 1.2754,
+      "step": 488
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.5915467372477843e-05,
+      "loss": 1.1864,
+      "step": 489
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.5898385570999146e-05,
+      "loss": 1.2278,
+      "step": 490
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.5881277335395425e-05,
+      "loss": 1.2407,
+      "step": 491
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.5864142742338707e-05,
+      "loss": 1.2947,
+      "step": 492
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.584698186861916e-05,
+      "loss": 1.1966,
+      "step": 493
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.5829794791144723e-05,
+      "loss": 1.1708,
+      "step": 494
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.5812581586940767e-05,
+      "loss": 1.2184,
+      "step": 495
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.5795342333149757e-05,
+      "loss": 1.2204,
+      "step": 496
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 1.57780771070309e-05,
+      "loss": 1.2833,
+      "step": 497
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 1.57607859859598e-05,
+      "loss": 1.1883,
+      "step": 498
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 1.5743469047428113e-05,
+      "loss": 1.2125,
+      "step": 499
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 1.572612636904319e-05,
+      "loss": 1.1894,
+      "step": 500
+    }
+  ],
+  "max_steps": 1530,
+  "num_train_epochs": 1,
+  "total_flos": 6.690027416531763e+16,
+  "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:2a52f9b0047b3906b156d34a4924346c934d856c9f2ee35c75d5183b10ee46ab
+size 3771

config.json

@@ -0,0 +1,23 @@
+{
+  "_name_or_path": "/home/ubuntu/stanford_alpaca/alpaca-cleaned",
+  "architectures": [
+    "LLaMAForCausalLM"
+  ],
+  "bos_token_id": 0,
+  "eos_token_id": 1,
+  "hidden_act": "silu",
+  "hidden_size": 4096,
+  "initializer_range": 0.02,
+  "intermediate_size": 11008,
+  "max_sequence_length": 2048,
+  "model_type": "llama",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 32,
+  "pad_token_id": -1,
+  "rms_norm_eps": 1e-06,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float32",
+  "transformers_version": "4.27.0.dev0",
+  "use_cache": true,
+  "vocab_size": 32001
+}

special_tokens_map.json

@@ -0,0 +1,6 @@
+{
+  "bos_token": "</s>",
+  "eos_token": "</s>",
+  "pad_token": "[PAD]",
+  "unk_token": "</s>"
+}

tokenizer.model

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

tokenizer_config.json

@@ -0,0 +1,9 @@
+{
+  "bos_token": "",
+  "eos_token": "",
+  "model_max_length": 1024,
+  "padding_side": "right",
+  "special_tokens_map_file": "/home/ubuntu/.cache/huggingface/hub/models--decapoda-research--llama-7b-hf/snapshots/5f98eefcc80e437ef68d457ad7bf167c2c6a1348/special_tokens_map.json",
+  "tokenizer_class": "LLaMATokenizer",
+  "unk_token": ""
+}

training_args.bin

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