Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+tokenizer.json filter=lfs diff=lfs merge=lfs -text

added_tokens.json

@@ -0,0 +1,28 @@
+{
+  "</think>": 151668,
+  "</tool_call>": 151658,
+  "</tool_response>": 151666,
+  "<think>": 151667,
+  "<tool_call>": 151657,
+  "<tool_response>": 151665,
+  "<|box_end|>": 151649,
+  "<|box_start|>": 151648,
+  "<|endoftext|>": 151643,
+  "<|file_sep|>": 151664,
+  "<|fim_middle|>": 151660,
+  "<|fim_pad|>": 151662,
+  "<|fim_prefix|>": 151659,
+  "<|fim_suffix|>": 151661,
+  "<|im_end|>": 151645,
+  "<|im_start|>": 151644,
+  "<|image_pad|>": 151655,
+  "<|object_ref_end|>": 151647,
+  "<|object_ref_start|>": 151646,
+  "<|quad_end|>": 151651,
+  "<|quad_start|>": 151650,
+  "<|repo_name|>": 151663,
+  "<|video_pad|>": 151656,
+  "<|vision_end|>": 151653,
+  "<|vision_pad|>": 151654,
+  "<|vision_start|>": 151652
+}

config.json

@@ -0,0 +1,38 @@
+{
+  "architectures": [
+    "Qwen3MoeForCausalLM"
+  ],
+  "attention_bias": false,
+  "attention_dropout": 0.0,
+  "bos_token_id": 151643,
+  "decoder_sparse_step": 1,
+  "eos_token_id": 151643,
+  "head_dim": 128,
+  "hidden_act": "silu",
+  "hidden_size": 2048,
+  "initializer_range": 0.02,
+  "intermediate_size": 6144,
+  "max_position_embeddings": 32768,
+  "max_window_layers": 48,
+  "mlp_only_layers": [],
+  "model_type": "qwen3_moe",
+  "moe_intermediate_size": 768,
+  "norm_topk_prob": true,
+  "num_attention_heads": 32,
+  "num_experts": 128,
+  "num_experts_per_tok": 8,
+  "num_hidden_layers": 48,
+  "num_key_value_heads": 4,
+  "output_router_logits": false,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": null,
+  "rope_theta": 1000000.0,
+  "router_aux_loss_coef": 0.001,
+  "sliding_window": null,
+  "tie_word_embeddings": false,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.51.3",
+  "use_cache": false,
+  "use_sliding_window": false,
+  "vocab_size": 151936
+}

generation_config.json

@@ -0,0 +1,6 @@
+{
+  "bos_token_id": 151643,
+  "eos_token_id": 151643,
+  "max_new_tokens": 2048,
+  "transformers_version": "4.51.3"
+}

model-00001-of-00013.safetensors

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

model-00002-of-00013.safetensors

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

model-00003-of-00013.safetensors

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

model-00004-of-00013.safetensors

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

model-00005-of-00013.safetensors

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

model-00006-of-00013.safetensors

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

model-00007-of-00013.safetensors

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

model-00008-of-00013.safetensors

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

model-00009-of-00013.safetensors

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

model-00010-of-00013.safetensors

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

model-00011-of-00013.safetensors

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

model-00012-of-00013.safetensors

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

model-00013-of-00013.safetensors

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

special_tokens_map.json

@@ -0,0 +1,31 @@
+{
+  "additional_special_tokens": [
+    "<|im_start|>",
+    "<|im_end|>",
+    "<|object_ref_start|>",
+    "<|object_ref_end|>",
+    "<|box_start|>",
+    "<|box_end|>",
+    "<|quad_start|>",
+    "<|quad_end|>",
+    "<|vision_start|>",
+    "<|vision_end|>",
+    "<|vision_pad|>",
+    "<|image_pad|>",
+    "<|video_pad|>"
+  ],
+  "eos_token": {
+    "content": "<|im_end|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer.json

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

tokenizer_config.json

@@ -0,0 +1,241 @@
+{
+  "add_bos_token": false,
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "151643": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151644": {
+      "content": "<|im_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151645": {
+      "content": "<|im_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151646": {
+      "content": "<|object_ref_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151647": {
+      "content": "<|object_ref_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151648": {
+      "content": "<|box_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151649": {
+      "content": "<|box_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151650": {
+      "content": "<|quad_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151651": {
+      "content": "<|quad_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151652": {
+      "content": "<|vision_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151653": {
+      "content": "<|vision_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151654": {
+      "content": "<|vision_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151655": {
+      "content": "<|image_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151656": {
+      "content": "<|video_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151657": {
+      "content": "<tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151658": {
+      "content": "</tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151659": {
+      "content": "<|fim_prefix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151660": {
+      "content": "<|fim_middle|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151661": {
+      "content": "<|fim_suffix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151662": {
+      "content": "<|fim_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151663": {
+      "content": "<|repo_name|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151664": {
+      "content": "<|file_sep|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151665": {
+      "content": "<tool_response>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151666": {
+      "content": "</tool_response>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151667": {
+      "content": "<think>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151668": {
+      "content": "</think>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    }
+  },
+  "additional_special_tokens": [
+    "<|im_start|>",
+    "<|im_end|>",
+    "<|object_ref_start|>",
+    "<|object_ref_end|>",
+    "<|box_start|>",
+    "<|box_end|>",
+    "<|quad_start|>",
+    "<|quad_end|>",
+    "<|vision_start|>",
+    "<|vision_end|>",
+    "<|vision_pad|>",
+    "<|image_pad|>",
+    "<|video_pad|>"
+  ],
+  "bos_token": null,
+  "chat_template": "{%- if tools %}\n    {{- '<|im_start|>system\\n' }}\n    {%- if messages[0].role == 'system' %}\n        {{- messages[0].content + '\\n\\n' }}\n    {%- endif %}\n    {{- \"# Tools\\n\\nYou may call one or more functions to assist with the user query.\\n\\nYou are provided with function signatures within <tools></tools> XML tags:\\n<tools>\" }}\n    {%- for tool in tools %}\n        {{- \"\\n\" }}\n        {{- tool | tojson }}\n    {%- endfor %}\n    {{- \"\\n</tools>\\n\\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\\n<tool_call>\\n{\\\"name\\\": <function-name>, \\\"arguments\\\": <args-json-object>}\\n</tool_call><|im_end|>\\n\" }}\n{%- else %}\n    {%- if messages[0].role == 'system' %}\n        {{- '<|im_start|>system\\n' + messages[0].content + '<|im_end|>\\n' }}\n    {%- endif %}\n{%- endif %}\n{%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}\n{%- for message in messages[::-1] %}\n    {%- set index = (messages|length - 1) - loop.index0 %}\n    {%- if ns.multi_step_tool and message.role == \"user\" and not(message.content.startswith('<tool_response>') and message.content.endswith('</tool_response>')) %}\n        {%- set ns.multi_step_tool = false %}\n        {%- set ns.last_query_index = index %}\n    {%- endif %}\n{%- endfor %}\n{%- for message in messages %}\n    {%- if (message.role == \"user\") or (message.role == \"system\" and not loop.first) %}\n        {{- '<|im_start|>' + message.role + '\\n' + message.content + '<|im_end|>' + '\\n' }}\n    {%- elif message.role == \"assistant\" %}\n        {%- set content = message.content %}\n        {%- set reasoning_content = '' %}\n        {%- if message.reasoning_content is defined and message.reasoning_content is not none %}\n            {%- set reasoning_content = message.reasoning_content %}\n        {%- else %}\n            {%- if '</think>' in message.content %}\n                {%- set content = message.content.split('</think>')[-1].lstrip('\\n') %}\n                {%- set reasoning_content = message.content.split('</think>')[0].rstrip('\\n').split('<think>')[-1].lstrip('\\n') %}\n            {%- endif %}\n        {%- endif %}\n        {%- if loop.index0 > ns.last_query_index %}\n            {%- if loop.last or (not loop.last and reasoning_content) %}\n                {{- '<|im_start|>' + message.role + '\\n<think>\\n' + reasoning_content.strip('\\n') + '\\n</think>\\n\\n' + content.lstrip('\\n') }}\n            {%- else %}\n                {{- '<|im_start|>' + message.role + '\\n' + content }}\n            {%- endif %}\n        {%- else %}\n            {{- '<|im_start|>' + message.role + '\\n' + content }}\n        {%- endif %}\n        {%- if message.tool_calls %}\n            {%- for tool_call in message.tool_calls %}\n                {%- if (loop.first and content) or (not loop.first) %}\n                    {{- '\\n' }}\n                {%- endif %}\n                {%- if tool_call.function %}\n                    {%- set tool_call = tool_call.function %}\n                {%- endif %}\n                {{- '<tool_call>\\n{\"name\": \"' }}\n                {{- tool_call.name }}\n                {{- '\", \"arguments\": ' }}\n                {%- if tool_call.arguments is string %}\n                    {{- tool_call.arguments }}\n                {%- else %}\n                    {{- tool_call.arguments | tojson }}\n                {%- endif %}\n                {{- '}\\n</tool_call>' }}\n            {%- endfor %}\n        {%- endif %}\n        {{- '<|im_end|>\\n' }}\n    {%- elif message.role == \"tool\" %}\n        {%- if loop.first or (messages[loop.index0 - 1].role != \"tool\") %}\n            {{- '<|im_start|>user' }}\n        {%- endif %}\n        {{- '\\n<tool_response>\\n' }}\n        {{- message.content }}\n        {{- '\\n</tool_response>' }}\n        {%- if loop.last or (messages[loop.index0 + 1].role != \"tool\") %}\n            {{- '<|im_end|>\\n' }}\n        {%- endif %}\n    {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n    {{- '<|im_start|>assistant\\n' }}\n    {%- if enable_thinking is defined and enable_thinking is false %}\n        {{- '<think>\\n\\n</think>\\n\\n' }}\n    {%- endif %}\n{%- endif %}",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|im_end|>",
+  "errors": "replace",
+  "extra_special_tokens": {},
+  "model_max_length": 131072,
+  "pad_token": "<|endoftext|>",
+  "padding_side": "right",
+  "split_special_tokens": false,
+  "tokenizer_class": "Qwen2Tokenizer",
+  "unk_token": null
+}

trainer_state.json

@@ -0,0 +1,1798 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 4.0,
+  "eval_steps": 500,
+  "global_step": 252,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.015873015873015872,
+      "grad_norm": 2.355871594757968,
+      "learning_rate": 0.0,
+      "loss": 1.0469,
+      "step": 1
+    },
+    {
+      "epoch": 0.031746031746031744,
+      "grad_norm": 2.29666205825573,
+      "learning_rate": 1.5625e-06,
+      "loss": 0.9929,
+      "step": 2
+    },
+    {
+      "epoch": 0.047619047619047616,
+      "grad_norm": 2.535252155158116,
+      "learning_rate": 3.125e-06,
+      "loss": 1.0185,
+      "step": 3
+    },
+    {
+      "epoch": 0.06349206349206349,
+      "grad_norm": 2.106429965859871,
+      "learning_rate": 4.6875000000000004e-06,
+      "loss": 0.9046,
+      "step": 4
+    },
+    {
+      "epoch": 0.07936507936507936,
+      "grad_norm": 1.7754252196610092,
+      "learning_rate": 6.25e-06,
+      "loss": 1.0312,
+      "step": 5
+    },
+    {
+      "epoch": 0.09523809523809523,
+      "grad_norm": 1.5212254656019404,
+      "learning_rate": 7.8125e-06,
+      "loss": 0.9741,
+      "step": 6
+    },
+    {
+      "epoch": 0.1111111111111111,
+      "grad_norm": 1.2286700272250457,
+      "learning_rate": 9.375000000000001e-06,
+      "loss": 0.924,
+      "step": 7
+    },
+    {
+      "epoch": 0.12698412698412698,
+      "grad_norm": 1.239763672997659,
+      "learning_rate": 1.09375e-05,
+      "loss": 0.9112,
+      "step": 8
+    },
+    {
+      "epoch": 0.14285714285714285,
+      "grad_norm": 1.7804807603399884,
+      "learning_rate": 1.25e-05,
+      "loss": 0.9631,
+      "step": 9
+    },
+    {
+      "epoch": 0.15873015873015872,
+      "grad_norm": 1.6096276860914913,
+      "learning_rate": 1.4062500000000001e-05,
+      "loss": 0.8914,
+      "step": 10
+    },
+    {
+      "epoch": 0.1746031746031746,
+      "grad_norm": 1.4685478602306294,
+      "learning_rate": 1.5625e-05,
+      "loss": 0.8313,
+      "step": 11
+    },
+    {
+      "epoch": 0.19047619047619047,
+      "grad_norm": 1.797675338951107,
+      "learning_rate": 1.71875e-05,
+      "loss": 0.9572,
+      "step": 12
+    },
+    {
+      "epoch": 0.20634920634920634,
+      "grad_norm": 1.5020710264445867,
+      "learning_rate": 1.8750000000000002e-05,
+      "loss": 0.8929,
+      "step": 13
+    },
+    {
+      "epoch": 0.2222222222222222,
+      "grad_norm": 1.389913325647765,
+      "learning_rate": 2.0312500000000002e-05,
+      "loss": 1.0575,
+      "step": 14
+    },
+    {
+      "epoch": 0.23809523809523808,
+      "grad_norm": 1.0364172775494112,
+      "learning_rate": 2.1875e-05,
+      "loss": 0.9367,
+      "step": 15
+    },
+    {
+      "epoch": 0.25396825396825395,
+      "grad_norm": 1.1754008678466592,
+      "learning_rate": 2.34375e-05,
+      "loss": 1.0082,
+      "step": 16
+    },
+    {
+      "epoch": 0.2698412698412698,
+      "grad_norm": 1.0227204004003465,
+      "learning_rate": 2.5e-05,
+      "loss": 0.9148,
+      "step": 17
+    },
+    {
+      "epoch": 0.2857142857142857,
+      "grad_norm": 1.0820292890536618,
+      "learning_rate": 2.6562500000000002e-05,
+      "loss": 0.9748,
+      "step": 18
+    },
+    {
+      "epoch": 0.30158730158730157,
+      "grad_norm": 0.945837076799249,
+      "learning_rate": 2.8125000000000003e-05,
+      "loss": 0.7132,
+      "step": 19
+    },
+    {
+      "epoch": 0.31746031746031744,
+      "grad_norm": 1.0248571882025115,
+      "learning_rate": 2.96875e-05,
+      "loss": 0.8881,
+      "step": 20
+    },
+    {
+      "epoch": 0.3333333333333333,
+      "grad_norm": 0.7825673389448284,
+      "learning_rate": 3.125e-05,
+      "loss": 0.9596,
+      "step": 21
+    },
+    {
+      "epoch": 0.3492063492063492,
+      "grad_norm": 0.9610289643147956,
+      "learning_rate": 3.2812500000000005e-05,
+      "loss": 1.0261,
+      "step": 22
+    },
+    {
+      "epoch": 0.36507936507936506,
+      "grad_norm": 0.9669319262880848,
+      "learning_rate": 3.4375e-05,
+      "loss": 0.8504,
+      "step": 23
+    },
+    {
+      "epoch": 0.38095238095238093,
+      "grad_norm": 0.7449356400588687,
+      "learning_rate": 3.59375e-05,
+      "loss": 0.8124,
+      "step": 24
+    },
+    {
+      "epoch": 0.3968253968253968,
+      "grad_norm": 0.83693613965816,
+      "learning_rate": 3.7500000000000003e-05,
+      "loss": 1.0061,
+      "step": 25
+    },
+    {
+      "epoch": 0.4126984126984127,
+      "grad_norm": 0.732383235323442,
+      "learning_rate": 3.90625e-05,
+      "loss": 0.8433,
+      "step": 26
+    },
+    {
+      "epoch": 0.42857142857142855,
+      "grad_norm": 0.7794935876422098,
+      "learning_rate": 4.0625000000000005e-05,
+      "loss": 0.7042,
+      "step": 27
+    },
+    {
+      "epoch": 0.4444444444444444,
+      "grad_norm": 0.7604958575268047,
+      "learning_rate": 4.21875e-05,
+      "loss": 0.9629,
+      "step": 28
+    },
+    {
+      "epoch": 0.4603174603174603,
+      "grad_norm": 0.6984385536231642,
+      "learning_rate": 4.375e-05,
+      "loss": 0.8909,
+      "step": 29
+    },
+    {
+      "epoch": 0.47619047619047616,
+      "grad_norm": 0.7941140442495155,
+      "learning_rate": 4.5312500000000004e-05,
+      "loss": 0.9318,
+      "step": 30
+    },
+    {
+      "epoch": 0.49206349206349204,
+      "grad_norm": 0.7485507149365567,
+      "learning_rate": 4.6875e-05,
+      "loss": 0.9032,
+      "step": 31
+    },
+    {
+      "epoch": 0.5079365079365079,
+      "grad_norm": 0.7312955579208906,
+      "learning_rate": 4.8437500000000005e-05,
+      "loss": 0.8793,
+      "step": 32
+    },
+    {
+      "epoch": 0.5238095238095238,
+      "grad_norm": 0.728955139017027,
+      "learning_rate": 5e-05,
+      "loss": 1.0292,
+      "step": 33
+    },
+    {
+      "epoch": 0.5396825396825397,
+      "grad_norm": 0.6223071714803796,
+      "learning_rate": 4.999845960383972e-05,
+      "loss": 0.829,
+      "step": 34
+    },
+    {
+      "epoch": 0.5555555555555556,
+      "grad_norm": 0.7081482857388056,
+      "learning_rate": 4.999383860518451e-05,
+      "loss": 0.821,
+      "step": 35
+    },
+    {
+      "epoch": 0.5714285714285714,
+      "grad_norm": 0.7021438050343367,
+      "learning_rate": 4.998613757348784e-05,
+      "loss": 0.9658,
+      "step": 36
+    },
+    {
+      "epoch": 0.5873015873015873,
+      "grad_norm": 0.6005545292155794,
+      "learning_rate": 4.9975357457760896e-05,
+      "loss": 0.9805,
+      "step": 37
+    },
+    {
+      "epoch": 0.6031746031746031,
+      "grad_norm": 0.66159765943155,
+      "learning_rate": 4.9961499586455584e-05,
+      "loss": 0.8479,
+      "step": 38
+    },
+    {
+      "epoch": 0.6190476190476191,
+      "grad_norm": 0.6859851915992995,
+      "learning_rate": 4.994456566730085e-05,
+      "loss": 0.9104,
+      "step": 39
+    },
+    {
+      "epoch": 0.6349206349206349,
+      "grad_norm": 0.5770937588145084,
+      "learning_rate": 4.9924557787092216e-05,
+      "loss": 0.7705,
+      "step": 40
+    },
+    {
+      "epoch": 0.6507936507936508,
+      "grad_norm": 0.5587380636863567,
+      "learning_rate": 4.990147841143462e-05,
+      "loss": 0.8406,
+      "step": 41
+    },
+    {
+      "epoch": 0.6666666666666666,
+      "grad_norm": 0.5939117737515469,
+      "learning_rate": 4.9875330384438604e-05,
+      "loss": 0.9143,
+      "step": 42
+    },
+    {
+      "epoch": 0.6825396825396826,
+      "grad_norm": 0.5421902034671652,
+      "learning_rate": 4.984611692836979e-05,
+      "loss": 0.8323,
+      "step": 43
+    },
+    {
+      "epoch": 0.6984126984126984,
+      "grad_norm": 0.5523631282552526,
+      "learning_rate": 4.9813841643251836e-05,
+      "loss": 0.7557,
+      "step": 44
+    },
+    {
+      "epoch": 0.7142857142857143,
+      "grad_norm": 0.5991896816540728,
+      "learning_rate": 4.977850850642275e-05,
+      "loss": 0.8471,
+      "step": 45
+    },
+    {
+      "epoch": 0.7301587301587301,
+      "grad_norm": 0.6364621652121066,
+      "learning_rate": 4.97401218720448e-05,
+      "loss": 0.8013,
+      "step": 46
+    },
+    {
+      "epoch": 0.746031746031746,
+      "grad_norm": 0.5534873919742207,
+      "learning_rate": 4.969868647056792e-05,
+      "loss": 0.8996,
+      "step": 47
+    },
+    {
+      "epoch": 0.7619047619047619,
+      "grad_norm": 0.5035034360883922,
+      "learning_rate": 4.965420740814679e-05,
+      "loss": 0.8944,
+      "step": 48
+    },
+    {
+      "epoch": 0.7777777777777778,
+      "grad_norm": 0.5729489774574688,
+      "learning_rate": 4.9606690166011546e-05,
+      "loss": 0.7887,
+      "step": 49
+    },
+    {
+      "epoch": 0.7936507936507936,
+      "grad_norm": 0.6004405465949967,
+      "learning_rate": 4.95561405997924e-05,
+      "loss": 0.831,
+      "step": 50
+    },
+    {
+      "epoch": 0.8095238095238095,
+      "grad_norm": 0.725673276310606,
+      "learning_rate": 4.9502564938797946e-05,
+      "loss": 0.8849,
+      "step": 51
+    },
+    {
+      "epoch": 0.8253968253968254,
+      "grad_norm": 0.4990176056299948,
+      "learning_rate": 4.9445969785247595e-05,
+      "loss": 0.8298,
+      "step": 52
+    },
+    {
+      "epoch": 0.8412698412698413,
+      "grad_norm": 0.6071056927514447,
+      "learning_rate": 4.938636211345792e-05,
+      "loss": 0.944,
+      "step": 53
+    },
+    {
+      "epoch": 0.8571428571428571,
+      "grad_norm": 0.6345636016627114,
+      "learning_rate": 4.932374926898321e-05,
+      "loss": 0.8069,
+      "step": 54
+    },
+    {
+      "epoch": 0.873015873015873,
+      "grad_norm": 0.6333202856879505,
+      "learning_rate": 4.9258138967710306e-05,
+      "loss": 0.7373,
+      "step": 55
+    },
+    {
+      "epoch": 0.8888888888888888,
+      "grad_norm": 0.5152048396222051,
+      "learning_rate": 4.918953929490768e-05,
+      "loss": 0.8855,
+      "step": 56
+    },
+    {
+      "epoch": 0.9047619047619048,
+      "grad_norm": 0.5806639114473304,
+      "learning_rate": 4.9117958704229154e-05,
+      "loss": 0.8731,
+      "step": 57
+    },
+    {
+      "epoch": 0.9206349206349206,
+      "grad_norm": 0.6001547400645247,
+      "learning_rate": 4.904340601667208e-05,
+      "loss": 0.816,
+      "step": 58
+    },
+    {
+      "epoch": 0.9365079365079365,
+      "grad_norm": 0.470270082255384,
+      "learning_rate": 4.8965890419490354e-05,
+      "loss": 0.8847,
+      "step": 59
+    },
+    {
+      "epoch": 0.9523809523809523,
+      "grad_norm": 0.5015222585808989,
+      "learning_rate": 4.8885421465062236e-05,
+      "loss": 0.8507,
+      "step": 60
+    },
+    {
+      "epoch": 0.9682539682539683,
+      "grad_norm": 0.48808541077856465,
+      "learning_rate": 4.88020090697132e-05,
+      "loss": 0.774,
+      "step": 61
+    },
+    {
+      "epoch": 0.9841269841269841,
+      "grad_norm": 0.5837250225569859,
+      "learning_rate": 4.871566351249393e-05,
+      "loss": 0.8844,
+      "step": 62
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 0.5121395435398158,
+      "learning_rate": 4.86263954339136e-05,
+      "loss": 0.7565,
+      "step": 63
+    },
+    {
+      "epoch": 1.0158730158730158,
+      "grad_norm": 0.858989786993005,
+      "learning_rate": 4.853421583462866e-05,
+      "loss": 0.7147,
+      "step": 64
+    },
+    {
+      "epoch": 1.0317460317460316,
+      "grad_norm": 0.6819369439457027,
+      "learning_rate": 4.843913607408717e-05,
+      "loss": 0.7632,
+      "step": 65
+    },
+    {
+      "epoch": 1.0476190476190477,
+      "grad_norm": 0.8758215098676306,
+      "learning_rate": 4.834116786912897e-05,
+      "loss": 0.7952,
+      "step": 66
+    },
+    {
+      "epoch": 1.0634920634920635,
+      "grad_norm": 0.5971995504609642,
+      "learning_rate": 4.82403232925418e-05,
+      "loss": 0.6327,
+      "step": 67
+    },
+    {
+      "epoch": 1.0793650793650793,
+      "grad_norm": 0.5651140794936931,
+      "learning_rate": 4.8136614771573554e-05,
+      "loss": 0.6429,
+      "step": 68
+    },
+    {
+      "epoch": 1.0952380952380953,
+      "grad_norm": 0.882155318220025,
+      "learning_rate": 4.8030055086400826e-05,
+      "loss": 0.8083,
+      "step": 69
+    },
+    {
+      "epoch": 1.1111111111111112,
+      "grad_norm": 0.7712424011094504,
+      "learning_rate": 4.7920657368554e-05,
+      "loss": 0.6792,
+      "step": 70
+    },
+    {
+      "epoch": 1.126984126984127,
+      "grad_norm": 0.5624839067932064,
+      "learning_rate": 4.7808435099299045e-05,
+      "loss": 0.6928,
+      "step": 71
+    },
+    {
+      "epoch": 1.1428571428571428,
+      "grad_norm": 0.5634668991057131,
+      "learning_rate": 4.7693402107976174e-05,
+      "loss": 0.7191,
+      "step": 72
+    },
+    {
+      "epoch": 1.1587301587301586,
+      "grad_norm": 0.6557940355672685,
+      "learning_rate": 4.757557257029564e-05,
+      "loss": 0.7138,
+      "step": 73
+    },
+    {
+      "epoch": 1.1746031746031746,
+      "grad_norm": 0.5576756408219278,
+      "learning_rate": 4.745496100659082e-05,
+      "loss": 0.6904,
+      "step": 74
+    },
+    {
+      "epoch": 1.1904761904761905,
+      "grad_norm": 0.5394945009236098,
+      "learning_rate": 4.7331582280028906e-05,
+      "loss": 0.6684,
+      "step": 75
+    },
+    {
+      "epoch": 1.2063492063492063,
+      "grad_norm": 0.5616484225088534,
+      "learning_rate": 4.720545159477922e-05,
+      "loss": 0.6123,
+      "step": 76
+    },
+    {
+      "epoch": 1.2222222222222223,
+      "grad_norm": 0.6137522865571288,
+      "learning_rate": 4.7076584494139606e-05,
+      "loss": 0.6639,
+      "step": 77
+    },
+    {
+      "epoch": 1.2380952380952381,
+      "grad_norm": 0.5520563824703866,
+      "learning_rate": 4.694499685862106e-05,
+      "loss": 0.7504,
+      "step": 78
+    },
+    {
+      "epoch": 1.253968253968254,
+      "grad_norm": 0.6398395502333153,
+      "learning_rate": 4.6810704903990634e-05,
+      "loss": 0.8063,
+      "step": 79
+    },
+    {
+      "epoch": 1.2698412698412698,
+      "grad_norm": 0.5170068764903927,
+      "learning_rate": 4.667372517927323e-05,
+      "loss": 0.682,
+      "step": 80
+    },
+    {
+      "epoch": 1.2857142857142856,
+      "grad_norm": 0.5470142100714379,
+      "learning_rate": 4.653407456471222e-05,
+      "loss": 0.74,
+      "step": 81
+    },
+    {
+      "epoch": 1.3015873015873016,
+      "grad_norm": 0.5344699813999502,
+      "learning_rate": 4.6391770269689236e-05,
+      "loss": 0.7019,
+      "step": 82
+    },
+    {
+      "epoch": 1.3174603174603174,
+      "grad_norm": 0.572242361510543,
+      "learning_rate": 4.6246829830603454e-05,
+      "loss": 0.7612,
+      "step": 83
+    },
+    {
+      "epoch": 1.3333333333333333,
+      "grad_norm": 0.5918389811121451,
+      "learning_rate": 4.609927110871053e-05,
+      "loss": 0.7285,
+      "step": 84
+    },
+    {
+      "epoch": 1.3492063492063493,
+      "grad_norm": 0.4866622733839569,
+      "learning_rate": 4.594911228792156e-05,
+      "loss": 0.656,
+      "step": 85
+    },
+    {
+      "epoch": 1.3650793650793651,
+      "grad_norm": 0.5275973128412808,
+      "learning_rate": 4.579637187256222e-05,
+      "loss": 0.576,
+      "step": 86
+    },
+    {
+      "epoch": 1.380952380952381,
+      "grad_norm": 0.5488033313933244,
+      "learning_rate": 4.564106868509246e-05,
+      "loss": 0.7176,
+      "step": 87
+    },
+    {
+      "epoch": 1.3968253968253967,
+      "grad_norm": 0.6920680886388099,
+      "learning_rate": 4.548322186378697e-05,
+      "loss": 0.7926,
+      "step": 88
+    },
+    {
+      "epoch": 1.4126984126984126,
+      "grad_norm": 0.774236286341856,
+      "learning_rate": 4.532285086037674e-05,
+      "loss": 0.7313,
+      "step": 89
+    },
+    {
+      "epoch": 1.4285714285714286,
+      "grad_norm": 0.7799985101441093,
+      "learning_rate": 4.5159975437652017e-05,
+      "loss": 0.717,
+      "step": 90
+    },
+    {
+      "epoch": 1.4444444444444444,
+      "grad_norm": 0.5542880474504607,
+      "learning_rate": 4.499461566702685e-05,
+      "loss": 0.5701,
+      "step": 91
+    },
+    {
+      "epoch": 1.4603174603174602,
+      "grad_norm": 0.6118784803715336,
+      "learning_rate": 4.48267919260657e-05,
+      "loss": 0.7026,
+      "step": 92
+    },
+    {
+      "epoch": 1.4761904761904763,
+      "grad_norm": 0.6649407372701752,
+      "learning_rate": 4.4656524895972255e-05,
+      "loss": 0.7061,
+      "step": 93
+    },
+    {
+      "epoch": 1.492063492063492,
+      "grad_norm": 0.47895549061427817,
+      "learning_rate": 4.448383555904088e-05,
+      "loss": 0.6387,
+      "step": 94
+    },
+    {
+      "epoch": 1.507936507936508,
+      "grad_norm": 0.5811905127982857,
+      "learning_rate": 4.430874519607089e-05,
+      "loss": 0.8295,
+      "step": 95
+    },
+    {
+      "epoch": 1.5238095238095237,
+      "grad_norm": 0.6782303990614583,
+      "learning_rate": 4.413127538374411e-05,
+      "loss": 0.7571,
+      "step": 96
+    },
+    {
+      "epoch": 1.5396825396825395,
+      "grad_norm": 0.5127521378901643,
+      "learning_rate": 4.395144799196593e-05,
+      "loss": 0.6259,
+      "step": 97
+    },
+    {
+      "epoch": 1.5555555555555556,
+      "grad_norm": 1.1333047155488623,
+      "learning_rate": 4.376928518117027e-05,
+      "loss": 0.7263,
+      "step": 98
+    },
+    {
+      "epoch": 1.5714285714285714,
+      "grad_norm": 0.9285725347831506,
+      "learning_rate": 4.358480939958867e-05,
+      "loss": 0.6652,
+      "step": 99
+    },
+    {
+      "epoch": 1.5873015873015874,
+      "grad_norm": 0.5756004852981952,
+      "learning_rate": 4.3398043380483965e-05,
+      "loss": 0.7045,
+      "step": 100
+    },
+    {
+      "epoch": 1.6031746031746033,
+      "grad_norm": 0.5732214165272889,
+      "learning_rate": 4.320901013934887e-05,
+      "loss": 0.7096,
+      "step": 101
+    },
+    {
+      "epoch": 1.619047619047619,
+      "grad_norm": 0.5729960942540637,
+      "learning_rate": 4.3017732971069674e-05,
+      "loss": 0.7797,
+      "step": 102
+    },
+    {
+      "epoch": 1.6349206349206349,
+      "grad_norm": 0.6414711093856016,
+      "learning_rate": 4.282423544705564e-05,
+      "loss": 0.75,
+      "step": 103
+    },
+    {
+      "epoch": 1.6507936507936507,
+      "grad_norm": 0.48208643464721207,
+      "learning_rate": 4.262854141233419e-05,
+      "loss": 0.7566,
+      "step": 104
+    },
+    {
+      "epoch": 1.6666666666666665,
+      "grad_norm": 0.5650935089244651,
+      "learning_rate": 4.2430674982612506e-05,
+      "loss": 0.7282,
+      "step": 105
+    },
+    {
+      "epoch": 1.6825396825396826,
+      "grad_norm": 0.6266660832902915,
+      "learning_rate": 4.223066054130568e-05,
+      "loss": 0.7098,
+      "step": 106
+    },
+    {
+      "epoch": 1.6984126984126984,
+      "grad_norm": 1.1721379739857851,
+      "learning_rate": 4.2028522736531895e-05,
+      "loss": 0.7244,
+      "step": 107
+    },
+    {
+      "epoch": 1.7142857142857144,
+      "grad_norm": 0.5821734258045642,
+      "learning_rate": 4.1824286478075025e-05,
+      "loss": 0.7918,
+      "step": 108
+    },
+    {
+      "epoch": 1.7301587301587302,
+      "grad_norm": 0.5675713145707344,
+      "learning_rate": 4.161797693431493e-05,
+      "loss": 0.7271,
+      "step": 109
+    },
+    {
+      "epoch": 1.746031746031746,
+      "grad_norm": 0.5948316073979433,
+      "learning_rate": 4.140961952912594e-05,
+      "loss": 0.6585,
+      "step": 110
+    },
+    {
+      "epoch": 1.7619047619047619,
+      "grad_norm": 0.6142301353494962,
+      "learning_rate": 4.1199239938743797e-05,
+      "loss": 0.6636,
+      "step": 111
+    },
+    {
+      "epoch": 1.7777777777777777,
+      "grad_norm": 0.6089779675550376,
+      "learning_rate": 4.0986864088601564e-05,
+      "loss": 0.6483,
+      "step": 112
+    },
+    {
+      "epoch": 1.7936507936507935,
+      "grad_norm": 0.5407284417083653,
+      "learning_rate": 4.077251815013477e-05,
+      "loss": 0.6273,
+      "step": 113
+    },
+    {
+      "epoch": 1.8095238095238095,
+      "grad_norm": 0.5554496556982176,
+      "learning_rate": 4.0556228537556266e-05,
+      "loss": 0.6301,
+      "step": 114
+    },
+    {
+      "epoch": 1.8253968253968254,
+      "grad_norm": 0.5507291755794801,
+      "learning_rate": 4.0338021904601145e-05,
+      "loss": 0.6909,
+      "step": 115
+    },
+    {
+      "epoch": 1.8412698412698414,
+      "grad_norm": 0.4537046958331062,
+      "learning_rate": 4.0117925141242174e-05,
+      "loss": 0.7491,
+      "step": 116
+    },
+    {
+      "epoch": 1.8571428571428572,
+      "grad_norm": 0.5463750469400513,
+      "learning_rate": 3.9895965370376076e-05,
+      "loss": 0.8075,
+      "step": 117
+    },
+    {
+      "epoch": 1.873015873015873,
+      "grad_norm": 0.5027853264982712,
+      "learning_rate": 3.9672169944481165e-05,
+      "loss": 0.5601,
+      "step": 118
+    },
+    {
+      "epoch": 1.8888888888888888,
+      "grad_norm": 0.5661467536668536,
+      "learning_rate": 3.944656644224661e-05,
+      "loss": 0.7353,
+      "step": 119
+    },
+    {
+      "epoch": 1.9047619047619047,
+      "grad_norm": 0.48806777854236366,
+      "learning_rate": 3.921918266517391e-05,
+      "loss": 0.6765,
+      "step": 120
+    },
+    {
+      "epoch": 1.9206349206349205,
+      "grad_norm": 0.5877767380659309,
+      "learning_rate": 3.899004663415084e-05,
+      "loss": 0.707,
+      "step": 121
+    },
+    {
+      "epoch": 1.9365079365079365,
+      "grad_norm": 0.5311318468752881,
+      "learning_rate": 3.875918658599837e-05,
+      "loss": 0.7882,
+      "step": 122
+    },
+    {
+      "epoch": 1.9523809523809523,
+      "grad_norm": 0.5682753083153059,
+      "learning_rate": 3.8526630969991035e-05,
+      "loss": 0.6445,
+      "step": 123
+    },
+    {
+      "epoch": 1.9682539682539684,
+      "grad_norm": 0.4740591298744526,
+      "learning_rate": 3.8292408444351094e-05,
+      "loss": 0.6718,
+      "step": 124
+    },
+    {
+      "epoch": 1.9841269841269842,
+      "grad_norm": 0.45882583861407755,
+      "learning_rate": 3.805654787271686e-05,
+      "loss": 0.6762,
+      "step": 125
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 0.5675883245832433,
+      "learning_rate": 3.781907832058587e-05,
+      "loss": 0.7781,
+      "step": 126
+    },
+    {
+      "epoch": 2.015873015873016,
+      "grad_norm": 1.1104924115901638,
+      "learning_rate": 3.758002905173299e-05,
+      "loss": 0.4988,
+      "step": 127
+    },
+    {
+      "epoch": 2.0317460317460316,
+      "grad_norm": 2.7413879763258584,
+      "learning_rate": 3.7339429524604316e-05,
+      "loss": 0.5467,
+      "step": 128
+    },
+    {
+      "epoch": 2.0476190476190474,
+      "grad_norm": 2.2935599186627647,
+      "learning_rate": 3.7097309388686864e-05,
+      "loss": 0.6297,
+      "step": 129
+    },
+    {
+      "epoch": 2.0634920634920633,
+      "grad_norm": 0.8617968906197346,
+      "learning_rate": 3.6853698480854855e-05,
+      "loss": 0.5621,
+      "step": 130
+    },
+    {
+      "epoch": 2.0793650793650795,
+      "grad_norm": 0.8914264101265775,
+      "learning_rate": 3.660862682169282e-05,
+      "loss": 0.4258,
+      "step": 131
+    },
+    {
+      "epoch": 2.0952380952380953,
+      "grad_norm": 0.8467461851206783,
+      "learning_rate": 3.6362124611796225e-05,
+      "loss": 0.4557,
+      "step": 132
+    },
+    {
+      "epoch": 2.111111111111111,
+      "grad_norm": 0.7089992750499162,
+      "learning_rate": 3.6114222228049656e-05,
+      "loss": 0.5554,
+      "step": 133
+    },
+    {
+      "epoch": 2.126984126984127,
+      "grad_norm": 1.140458615229716,
+      "learning_rate": 3.586495021988351e-05,
+      "loss": 0.5374,
+      "step": 134
+    },
+    {
+      "epoch": 2.142857142857143,
+      "grad_norm": 0.7757982307284462,
+      "learning_rate": 3.5614339305509336e-05,
+      "loss": 0.5589,
+      "step": 135
+    },
+    {
+      "epoch": 2.1587301587301586,
+      "grad_norm": 0.6933824916021677,
+      "learning_rate": 3.5362420368134356e-05,
+      "loss": 0.4544,
+      "step": 136
+    },
+    {
+      "epoch": 2.1746031746031744,
+      "grad_norm": 0.719692057444369,
+      "learning_rate": 3.510922445215568e-05,
+      "loss": 0.531,
+      "step": 137
+    },
+    {
+      "epoch": 2.1904761904761907,
+      "grad_norm": 0.5869409227388236,
+      "learning_rate": 3.4854782759334627e-05,
+      "loss": 0.5112,
+      "step": 138
+    },
+    {
+      "epoch": 2.2063492063492065,
+      "grad_norm": 0.5457151982279885,
+      "learning_rate": 3.459912664495176e-05,
+      "loss": 0.4499,
+      "step": 139
+    },
+    {
+      "epoch": 2.2222222222222223,
+      "grad_norm": 0.7006218084957239,
+      "learning_rate": 3.4342287613942805e-05,
+      "loss": 0.5074,
+      "step": 140
+    },
+    {
+      "epoch": 2.238095238095238,
+      "grad_norm": 0.5496478791456875,
+      "learning_rate": 3.408429731701635e-05,
+      "loss": 0.4533,
+      "step": 141
+    },
+    {
+      "epoch": 2.253968253968254,
+      "grad_norm": 0.7267666056198808,
+      "learning_rate": 3.3825187546753425e-05,
+      "loss": 0.5618,
+      "step": 142
+    },
+    {
+      "epoch": 2.2698412698412698,
+      "grad_norm": 0.5231531289243824,
+      "learning_rate": 3.356499023368963e-05,
+      "loss": 0.4319,
+      "step": 143
+    },
+    {
+      "epoch": 2.2857142857142856,
+      "grad_norm": 0.586670521348103,
+      "learning_rate": 3.330373744238033e-05,
+      "loss": 0.4441,
+      "step": 144
+    },
+    {
+      "epoch": 2.3015873015873014,
+      "grad_norm": 0.5837432088571016,
+      "learning_rate": 3.3041461367449254e-05,
+      "loss": 0.4027,
+      "step": 145
+    },
+    {
+      "epoch": 2.317460317460317,
+      "grad_norm": 0.5743190095438013,
+      "learning_rate": 3.2778194329621104e-05,
+      "loss": 0.4944,
+      "step": 146
+    },
+    {
+      "epoch": 2.3333333333333335,
+      "grad_norm": 0.6587815114542759,
+      "learning_rate": 3.251396877173861e-05,
+      "loss": 0.5504,
+      "step": 147
+    },
+    {
+      "epoch": 2.3492063492063493,
+      "grad_norm": 0.6550600819194445,
+      "learning_rate": 3.2248817254764554e-05,
+      "loss": 0.4673,
+      "step": 148
+    },
+    {
+      "epoch": 2.365079365079365,
+      "grad_norm": 0.5454074517099673,
+      "learning_rate": 3.1982772453769236e-05,
+      "loss": 0.4607,
+      "step": 149
+    },
+    {
+      "epoch": 2.380952380952381,
+      "grad_norm": 0.6024175453719594,
+      "learning_rate": 3.171586715390384e-05,
+      "loss": 0.603,
+      "step": 150
+    },
+    {
+      "epoch": 2.3968253968253967,
+      "grad_norm": 0.6278682196249654,
+      "learning_rate": 3.144813424636031e-05,
+      "loss": 0.5712,
+      "step": 151
+    },
+    {
+      "epoch": 2.4126984126984126,
+      "grad_norm": 0.6240251741716067,
+      "learning_rate": 3.117960672431805e-05,
+      "loss": 0.5312,
+      "step": 152
+    },
+    {
+      "epoch": 2.4285714285714284,
+      "grad_norm": 0.5826262197215396,
+      "learning_rate": 3.091031767887817e-05,
+      "loss": 0.5541,
+      "step": 153
+    },
+    {
+      "epoch": 2.4444444444444446,
+      "grad_norm": 0.5973198102264671,
+      "learning_rate": 3.064030029498561e-05,
+      "loss": 0.5684,
+      "step": 154
+    },
+    {
+      "epoch": 2.4603174603174605,
+      "grad_norm": 0.5452803248570526,
+      "learning_rate": 3.036958784733967e-05,
+      "loss": 0.579,
+      "step": 155
+    },
+    {
+      "epoch": 2.4761904761904763,
+      "grad_norm": 0.4826933541939396,
+      "learning_rate": 3.0098213696293542e-05,
+      "loss": 0.4995,
+      "step": 156
+    },
+    {
+      "epoch": 2.492063492063492,
+      "grad_norm": 0.5045447989769657,
+      "learning_rate": 2.982621128374325e-05,
+      "loss": 0.5211,
+      "step": 157
+    },
+    {
+      "epoch": 2.507936507936508,
+      "grad_norm": 0.5512726317886238,
+      "learning_rate": 2.955361412900654e-05,
+      "loss": 0.4766,
+      "step": 158
+    },
+    {
+      "epoch": 2.5238095238095237,
+      "grad_norm": 0.5720105255698517,
+      "learning_rate": 2.9280455824692256e-05,
+      "loss": 0.4498,
+      "step": 159
+    },
+    {
+      "epoch": 2.5396825396825395,
+      "grad_norm": 0.5440271425737038,
+      "learning_rate": 2.9006770032560637e-05,
+      "loss": 0.4801,
+      "step": 160
+    },
+    {
+      "epoch": 2.5555555555555554,
+      "grad_norm": 0.5161357261610209,
+      "learning_rate": 2.8732590479375165e-05,
+      "loss": 0.4853,
+      "step": 161
+    },
+    {
+      "epoch": 2.571428571428571,
+      "grad_norm": 0.506946956779649,
+      "learning_rate": 2.8457950952746292e-05,
+      "loss": 0.4138,
+      "step": 162
+    },
+    {
+      "epoch": 2.5873015873015874,
+      "grad_norm": 0.6232427008678014,
+      "learning_rate": 2.818288529696783e-05,
+      "loss": 0.3632,
+      "step": 163
+    },
+    {
+      "epoch": 2.6031746031746033,
+      "grad_norm": 0.5079770086478173,
+      "learning_rate": 2.7907427408846155e-05,
+      "loss": 0.4947,
+      "step": 164
+    },
+    {
+      "epoch": 2.619047619047619,
+      "grad_norm": 0.49058848467902455,
+      "learning_rate": 2.7631611233523136e-05,
+      "loss": 0.5522,
+      "step": 165
+    },
+    {
+      "epoch": 2.634920634920635,
+      "grad_norm": 0.5385726031164884,
+      "learning_rate": 2.7355470760292956e-05,
+      "loss": 0.4761,
+      "step": 166
+    },
+    {
+      "epoch": 2.6507936507936507,
+      "grad_norm": 0.6602291225270063,
+      "learning_rate": 2.7079040018413587e-05,
+      "loss": 0.5188,
+      "step": 167
+    },
+    {
+      "epoch": 2.6666666666666665,
+      "grad_norm": 0.5175508130590559,
+      "learning_rate": 2.6802353072913305e-05,
+      "loss": 0.5124,
+      "step": 168
+    },
+    {
+      "epoch": 2.682539682539683,
+      "grad_norm": 0.6455239838067437,
+      "learning_rate": 2.652544402039279e-05,
+      "loss": 0.4907,
+      "step": 169
+    },
+    {
+      "epoch": 2.6984126984126986,
+      "grad_norm": 0.5830651033904475,
+      "learning_rate": 2.6248346984823324e-05,
+      "loss": 0.4195,
+      "step": 170
+    },
+    {
+      "epoch": 2.7142857142857144,
+      "grad_norm": 0.5497318122287652,
+      "learning_rate": 2.597109611334169e-05,
+      "loss": 0.5326,
+      "step": 171
+    },
+    {
+      "epoch": 2.7301587301587302,
+      "grad_norm": 0.5205142114766783,
+      "learning_rate": 2.569372557204213e-05,
+      "loss": 0.4012,
+      "step": 172
+    },
+    {
+      "epoch": 2.746031746031746,
+      "grad_norm": 0.5128168951414211,
+      "learning_rate": 2.5416269541765964e-05,
+      "loss": 0.4625,
+      "step": 173
+    },
+    {
+      "epoch": 2.761904761904762,
+      "grad_norm": 0.5348874040766619,
+      "learning_rate": 2.5138762213889494e-05,
+      "loss": 0.5216,
+      "step": 174
+    },
+    {
+      "epoch": 2.7777777777777777,
+      "grad_norm": 0.4946974445313897,
+      "learning_rate": 2.4861237786110512e-05,
+      "loss": 0.4782,
+      "step": 175
+    },
+    {
+      "epoch": 2.7936507936507935,
+      "grad_norm": 0.5128385397220351,
+      "learning_rate": 2.458373045823404e-05,
+      "loss": 0.4704,
+      "step": 176
+    },
+    {
+      "epoch": 2.8095238095238093,
+      "grad_norm": 0.4597694930950496,
+      "learning_rate": 2.4306274427957878e-05,
+      "loss": 0.5063,
+      "step": 177
+    },
+    {
+      "epoch": 2.825396825396825,
+      "grad_norm": 0.5788602736229355,
+      "learning_rate": 2.4028903886658312e-05,
+      "loss": 0.5135,
+      "step": 178
+    },
+    {
+      "epoch": 2.8412698412698414,
+      "grad_norm": 0.513112821698425,
+      "learning_rate": 2.375165301517668e-05,
+      "loss": 0.4996,
+      "step": 179
+    },
+    {
+      "epoch": 2.857142857142857,
+      "grad_norm": 0.44596052374356865,
+      "learning_rate": 2.3474555979607214e-05,
+      "loss": 0.5051,
+      "step": 180
+    },
+    {
+      "epoch": 2.873015873015873,
+      "grad_norm": 0.49975482790762493,
+      "learning_rate": 2.3197646927086697e-05,
+      "loss": 0.3964,
+      "step": 181
+    },
+    {
+      "epoch": 2.888888888888889,
+      "grad_norm": 0.9671546735894949,
+      "learning_rate": 2.2920959981586422e-05,
+      "loss": 0.4667,
+      "step": 182
+    },
+    {
+      "epoch": 2.9047619047619047,
+      "grad_norm": 0.5958277372123243,
+      "learning_rate": 2.264452923970705e-05,
+      "loss": 0.5834,
+      "step": 183
+    },
+    {
+      "epoch": 2.9206349206349205,
+      "grad_norm": 0.5045651744357849,
+      "learning_rate": 2.2368388766476873e-05,
+      "loss": 0.4309,
+      "step": 184
+    },
+    {
+      "epoch": 2.9365079365079367,
+      "grad_norm": 0.47069616911716566,
+      "learning_rate": 2.2092572591153844e-05,
+      "loss": 0.4961,
+      "step": 185
+    },
+    {
+      "epoch": 2.9523809523809526,
+      "grad_norm": 0.5526655604005396,
+      "learning_rate": 2.1817114703032176e-05,
+      "loss": 0.5278,
+      "step": 186
+    },
+    {
+      "epoch": 2.9682539682539684,
+      "grad_norm": 0.5731260555494184,
+      "learning_rate": 2.154204904725371e-05,
+      "loss": 0.5567,
+      "step": 187
+    },
+    {
+      "epoch": 2.984126984126984,
+      "grad_norm": 0.5077085930555579,
+      "learning_rate": 2.1267409520624844e-05,
+      "loss": 0.4869,
+      "step": 188
+    },
+    {
+      "epoch": 3.0,
+      "grad_norm": 0.5249184159517826,
+      "learning_rate": 2.099322996743936e-05,
+      "loss": 0.3362,
+      "step": 189
+    },
+    {
+      "epoch": 3.015873015873016,
+      "grad_norm": 1.1451953853847987,
+      "learning_rate": 2.0719544175307754e-05,
+      "loss": 0.3188,
+      "step": 190
+    },
+    {
+      "epoch": 3.0317460317460316,
+      "grad_norm": 0.8148236258777549,
+      "learning_rate": 2.0446385870993467e-05,
+      "loss": 0.3492,
+      "step": 191
+    },
+    {
+      "epoch": 3.0476190476190474,
+      "grad_norm": 0.7899864704952979,
+      "learning_rate": 2.0173788716256756e-05,
+      "loss": 0.377,
+      "step": 192
+    },
+    {
+      "epoch": 3.0634920634920633,
+      "grad_norm": 1.293037201092889,
+      "learning_rate": 1.9901786303706467e-05,
+      "loss": 0.3286,
+      "step": 193
+    },
+    {
+      "epoch": 3.0793650793650795,
+      "grad_norm": 1.5199646910005133,
+      "learning_rate": 1.9630412152660332e-05,
+      "loss": 0.2886,
+      "step": 194
+    },
+    {
+      "epoch": 3.0952380952380953,
+      "grad_norm": 0.7493819364448684,
+      "learning_rate": 1.9359699705014397e-05,
+      "loss": 0.2698,
+      "step": 195
+    },
+    {
+      "epoch": 3.111111111111111,
+      "grad_norm": 0.700366824905691,
+      "learning_rate": 1.9089682321121834e-05,
+      "loss": 0.3111,
+      "step": 196
+    },
+    {
+      "epoch": 3.126984126984127,
+      "grad_norm": 0.8429958495049307,
+      "learning_rate": 1.882039327568195e-05,
+      "loss": 0.3393,
+      "step": 197
+    },
+    {
+      "epoch": 3.142857142857143,
+      "grad_norm": 0.7306551929825829,
+      "learning_rate": 1.8551865753639693e-05,
+      "loss": 0.316,
+      "step": 198
+    },
+    {
+      "epoch": 3.1587301587301586,
+      "grad_norm": 0.6244862767673323,
+      "learning_rate": 1.8284132846096164e-05,
+      "loss": 0.2946,
+      "step": 199
+    },
+    {
+      "epoch": 3.1746031746031744,
+      "grad_norm": 0.590947513131452,
+      "learning_rate": 1.801722754623077e-05,
+      "loss": 0.2896,
+      "step": 200
+    },
+    {
+      "epoch": 3.1904761904761907,
+      "grad_norm": 0.5741314659912675,
+      "learning_rate": 1.775118274523545e-05,
+      "loss": 0.3371,
+      "step": 201
+    },
+    {
+      "epoch": 3.2063492063492065,
+      "grad_norm": 0.6471899564343424,
+      "learning_rate": 1.74860312282614e-05,
+      "loss": 0.2351,
+      "step": 202
+    },
+    {
+      "epoch": 3.2222222222222223,
+      "grad_norm": 0.6604007565403088,
+      "learning_rate": 1.72218056703789e-05,
+      "loss": 0.3245,
+      "step": 203
+    },
+    {
+      "epoch": 3.238095238095238,
+      "grad_norm": 0.7669449937781672,
+      "learning_rate": 1.6958538632550752e-05,
+      "loss": 0.3746,
+      "step": 204
+    },
+    {
+      "epoch": 3.253968253968254,
+      "grad_norm": 0.5725384176741873,
+      "learning_rate": 1.6696262557619676e-05,
+      "loss": 0.2268,
+      "step": 205
+    },
+    {
+      "epoch": 3.2698412698412698,
+      "grad_norm": 0.5160814026986275,
+      "learning_rate": 1.643500976631037e-05,
+      "loss": 0.3461,
+      "step": 206
+    },
+    {
+      "epoch": 3.2857142857142856,
+      "grad_norm": 0.4819957072062547,
+      "learning_rate": 1.6174812453246584e-05,
+      "loss": 0.2906,
+      "step": 207
+    },
+    {
+      "epoch": 3.3015873015873014,
+      "grad_norm": 0.6080164945661113,
+      "learning_rate": 1.5915702682983658e-05,
+      "loss": 0.2132,
+      "step": 208
+    },
+    {
+      "epoch": 3.317460317460317,
+      "grad_norm": 0.6680029345374708,
+      "learning_rate": 1.56577123860572e-05,
+      "loss": 0.3023,
+      "step": 209
+    },
+    {
+      "epoch": 3.3333333333333335,
+      "grad_norm": 0.5355326751881537,
+      "learning_rate": 1.540087335504825e-05,
+      "loss": 0.2133,
+      "step": 210
+    },
+    {
+      "epoch": 3.3492063492063493,
+      "grad_norm": 0.5783500445358497,
+      "learning_rate": 1.514521724066537e-05,
+      "loss": 0.2582,
+      "step": 211
+    },
+    {
+      "epoch": 3.365079365079365,
+      "grad_norm": 0.5063022372874914,
+      "learning_rate": 1.4890775547844322e-05,
+      "loss": 0.2529,
+      "step": 212
+    },
+    {
+      "epoch": 3.380952380952381,
+      "grad_norm": 0.5796557988965374,
+      "learning_rate": 1.4637579631865645e-05,
+      "loss": 0.2897,
+      "step": 213
+    },
+    {
+      "epoch": 3.3968253968253967,
+      "grad_norm": 0.603714087084395,
+      "learning_rate": 1.4385660694490666e-05,
+      "loss": 0.2719,
+      "step": 214
+    },
+    {
+      "epoch": 3.4126984126984126,
+      "grad_norm": 0.6026614021745487,
+      "learning_rate": 1.4135049780116497e-05,
+      "loss": 0.3395,
+      "step": 215
+    },
+    {
+      "epoch": 3.4285714285714284,
+      "grad_norm": 0.7436598149589313,
+      "learning_rate": 1.3885777771950348e-05,
+      "loss": 0.337,
+      "step": 216
+    },
+    {
+      "epoch": 3.4444444444444446,
+      "grad_norm": 0.4834765774588162,
+      "learning_rate": 1.3637875388203782e-05,
+      "loss": 0.2668,
+      "step": 217
+    },
+    {
+      "epoch": 3.4603174603174605,
+      "grad_norm": 0.5451634141994833,
+      "learning_rate": 1.3391373178307181e-05,
+      "loss": 0.2384,
+      "step": 218
+    },
+    {
+      "epoch": 3.4761904761904763,
+      "grad_norm": 0.5698078470898527,
+      "learning_rate": 1.3146301519145151e-05,
+      "loss": 0.3243,
+      "step": 219
+    },
+    {
+      "epoch": 3.492063492063492,
+      "grad_norm": 0.5619505062332472,
+      "learning_rate": 1.2902690611313135e-05,
+      "loss": 0.2749,
+      "step": 220
+    },
+    {
+      "epoch": 3.507936507936508,
+      "grad_norm": 0.8210565503394927,
+      "learning_rate": 1.2660570475395683e-05,
+      "loss": 0.2303,
+      "step": 221
+    },
+    {
+      "epoch": 3.5238095238095237,
+      "grad_norm": 0.49079475108361625,
+      "learning_rate": 1.2419970948267012e-05,
+      "loss": 0.2464,
+      "step": 222
+    },
+    {
+      "epoch": 3.5396825396825395,
+      "grad_norm": 0.436922487388573,
+      "learning_rate": 1.2180921679414142e-05,
+      "loss": 0.25,
+      "step": 223
+    },
+    {
+      "epoch": 3.5555555555555554,
+      "grad_norm": 0.5240708542070043,
+      "learning_rate": 1.1943452127283145e-05,
+      "loss": 0.3265,
+      "step": 224
+    },
+    {
+      "epoch": 3.571428571428571,
+      "grad_norm": 0.5949138600847729,
+      "learning_rate": 1.1707591555648905e-05,
+      "loss": 0.2194,
+      "step": 225
+    },
+    {
+      "epoch": 3.5873015873015874,
+      "grad_norm": 0.5046744300340641,
+      "learning_rate": 1.1473369030008974e-05,
+      "loss": 0.3074,
+      "step": 226
+    },
+    {
+      "epoch": 3.6031746031746033,
+      "grad_norm": 0.5719832566771118,
+      "learning_rate": 1.1240813414001649e-05,
+      "loss": 0.2806,
+      "step": 227
+    },
+    {
+      "epoch": 3.619047619047619,
+      "grad_norm": 0.5543183797876515,
+      "learning_rate": 1.1009953365849169e-05,
+      "loss": 0.2832,
+      "step": 228
+    },
+    {
+      "epoch": 3.634920634920635,
+      "grad_norm": 0.4453621166767317,
+      "learning_rate": 1.0780817334826088e-05,
+      "loss": 0.289,
+      "step": 229
+    },
+    {
+      "epoch": 3.6507936507936507,
+      "grad_norm": 0.7163652417066158,
+      "learning_rate": 1.055343355775339e-05,
+      "loss": 0.3219,
+      "step": 230
+    },
+    {
+      "epoch": 3.6666666666666665,
+      "grad_norm": 0.5328475907166573,
+      "learning_rate": 1.0327830055518842e-05,
+      "loss": 0.2549,
+      "step": 231
+    },
+    {
+      "epoch": 3.682539682539683,
+      "grad_norm": 0.4796570396334715,
+      "learning_rate": 1.0104034629623932e-05,
+      "loss": 0.2796,
+      "step": 232
+    },
+    {
+      "epoch": 3.6984126984126986,
+      "grad_norm": 0.5103221963717066,
+      "learning_rate": 9.882074858757839e-06,
+      "loss": 0.2729,
+      "step": 233
+    },
+    {
+      "epoch": 3.7142857142857144,
+      "grad_norm": 0.4996694005868267,
+      "learning_rate": 9.661978095398854e-06,
+      "loss": 0.2868,
+      "step": 234
+    },
+    {
+      "epoch": 3.7301587301587302,
+      "grad_norm": 0.5364352563243058,
+      "learning_rate": 9.443771462443743e-06,
+      "loss": 0.3097,
+      "step": 235
+    },
+    {
+      "epoch": 3.746031746031746,
+      "grad_norm": 0.4755469936560106,
+      "learning_rate": 9.227481849865235e-06,
+      "loss": 0.2957,
+      "step": 236
+    },
+    {
+      "epoch": 3.761904761904762,
+      "grad_norm": 0.4651127214450329,
+      "learning_rate": 9.013135911398435e-06,
+      "loss": 0.2227,
+      "step": 237
+    },
+    {
+      "epoch": 3.7777777777777777,
+      "grad_norm": 0.5362240727276808,
+      "learning_rate": 8.800760061256206e-06,
+      "loss": 0.1757,
+      "step": 238
+    },
+    {
+      "epoch": 3.7936507936507935,
+      "grad_norm": 0.5787446958564049,
+      "learning_rate": 8.590380470874066e-06,
+      "loss": 0.2754,
+      "step": 239
+    },
+    {
+      "epoch": 3.8095238095238093,
+      "grad_norm": 0.5904647055376557,
+      "learning_rate": 8.38202306568507e-06,
+      "loss": 0.3051,
+      "step": 240
+    },
+    {
+      "epoch": 3.825396825396825,
+      "grad_norm": 0.5206106513004272,
+      "learning_rate": 8.175713521924978e-06,
+      "loss": 0.2922,
+      "step": 241
+    },
+    {
+      "epoch": 3.8412698412698414,
+      "grad_norm": 0.5012140513116363,
+      "learning_rate": 7.971477263468108e-06,
+      "loss": 0.2858,
+      "step": 242
+    },
+    {
+      "epoch": 3.857142857142857,
+      "grad_norm": 0.8553908081088378,
+      "learning_rate": 7.76933945869432e-06,
+      "loss": 0.3107,
+      "step": 243
+    },
+    {
+      "epoch": 3.873015873015873,
+      "grad_norm": 0.47754509677743895,
+      "learning_rate": 7.569325017387502e-06,
+      "loss": 0.2306,
+      "step": 244
+    },
+    {
+      "epoch": 3.888888888888889,
+      "grad_norm": 0.5126677036627112,
+      "learning_rate": 7.371458587665822e-06,
+      "loss": 0.3294,
+      "step": 245
+    },
+    {
+      "epoch": 3.9047619047619047,
+      "grad_norm": 0.4464738010436707,
+      "learning_rate": 7.1757645529443665e-06,
+      "loss": 0.3642,
+      "step": 246
+    },
+    {
+      "epoch": 3.9206349206349205,
+      "grad_norm": 0.4320951524772534,
+      "learning_rate": 6.982267028930325e-06,
+      "loss": 0.2424,
+      "step": 247
+    },
+    {
+      "epoch": 3.9365079365079367,
+      "grad_norm": 0.4366593119524442,
+      "learning_rate": 6.790989860651142e-06,
+      "loss": 0.277,
+      "step": 248
+    },
+    {
+      "epoch": 3.9523809523809526,
+      "grad_norm": 0.49933857198819703,
+      "learning_rate": 6.601956619516036e-06,
+      "loss": 0.2885,
+      "step": 249
+    },
+    {
+      "epoch": 3.9682539682539684,
+      "grad_norm": 0.5167959983988006,
+      "learning_rate": 6.4151906004113385e-06,
+      "loss": 0.2631,
+      "step": 250
+    },
+    {
+      "epoch": 3.984126984126984,
+      "grad_norm": 0.4656615515512403,
+      "learning_rate": 6.230714818829733e-06,
+      "loss": 0.2491,
+      "step": 251
+    },
+    {
+      "epoch": 4.0,
+      "grad_norm": 0.5702401686100438,
+      "learning_rate": 6.048552008034072e-06,
+      "loss": 0.2445,
+      "step": 252
+    }
+  ],
+  "logging_steps": 1,
+  "max_steps": 315,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 5,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 62903662804992.0,
+  "train_batch_size": 2,
+  "trial_name": null,
+  "trial_params": null
+}

training_args.bin

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