Add files using upload-large-folder tool

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-489/added_tokens.json

@@ -0,0 +1,24 @@
+{
+  "</tool_call>": 151658,
+  "<tool_call>": 151657,
+  "<|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
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-489/chat_template.jinja

@@ -0,0 +1,54 @@
+{%- if tools %}
+    {{- '<|im_start|>system\n' }}
+    {%- if messages[0]['role'] == 'system' %}
+        {{- messages[0]['content'] }}
+    {%- else %}
+        {{- 'You are Qwen, created by Alibaba Cloud. You are a helpful assistant.' }}
+    {%- endif %}
+    {{- "\n\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>" }}
+    {%- for tool in tools %}
+        {{- "\n" }}
+        {{- tool | tojson }}
+    {%- endfor %}
+    {{- "\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" }}
+{%- else %}
+    {%- if messages[0]['role'] == 'system' %}
+        {{- '<|im_start|>system\n' + messages[0]['content'] + '<|im_end|>\n' }}
+    {%- else %}
+        {{- '<|im_start|>system\nYou are Qwen, created by Alibaba Cloud. You are a helpful assistant.<|im_end|>\n' }}
+    {%- endif %}
+{%- endif %}
+{%- for message in messages %}
+    {%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %}
+        {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
+    {%- elif message.role == "assistant" %}
+        {{- '<|im_start|>' + message.role }}
+        {%- if message.content %}
+            {{- '\n' + message.content }}
+        {%- endif %}
+        {%- for tool_call in message.tool_calls %}
+            {%- if tool_call.function is defined %}
+                {%- set tool_call = tool_call.function %}
+            {%- endif %}
+            {{- '\n<tool_call>\n{"name": "' }}
+            {{- tool_call.name }}
+            {{- '", "arguments": ' }}
+            {{- tool_call.arguments | tojson }}
+            {{- '}\n</tool_call>' }}
+        {%- endfor %}
+        {{- '<|im_end|>\n' }}
+    {%- elif message.role == "tool" %}
+        {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %}
+            {{- '<|im_start|>user' }}
+        {%- endif %}
+        {{- '\n<tool_response>\n' }}
+        {{- message.content }}
+        {{- '\n</tool_response>' }}
+        {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
+            {{- '<|im_end|>\n' }}
+        {%- endif %}
+    {%- endif %}
+{%- endfor %}
+{%- if add_generation_prompt %}
+    {{- '<|im_start|>assistant\n' }}
+{%- endif %}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-489/config.json

@@ -0,0 +1,58 @@
+{
+  "architectures": [
+    "Qwen2ForCausalLM"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 151643,
+  "eos_token_id": 151645,
+  "hidden_act": "silu",
+  "hidden_size": 3584,
+  "initializer_range": 0.02,
+  "intermediate_size": 18944,
+  "layer_types": [
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention"
+  ],
+  "max_position_embeddings": 32768,
+  "max_window_layers": 28,
+  "model_type": "qwen2",
+  "num_attention_heads": 28,
+  "num_hidden_layers": 28,
+  "num_key_value_heads": 4,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": null,
+  "rope_theta": 1000000.0,
+  "sliding_window": null,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float32",
+  "transformers_version": "4.55.0",
+  "use_cache": false,
+  "use_sliding_window": false,
+  "vocab_size": 152064
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-489/generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 151643,
+  "eos_token_id": 151645,
+  "transformers_version": "4.55.0",
+  "use_cache": false
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-489/model.safetensors.index.json

@@ -0,0 +1,347 @@
+{
+  "metadata": {
+    "total_parameters": 1903904128,
+    "total_size": 30462466048
+  },
+  "weight_map": {
+    "lm_head.weight": "model-00007-of-00007.safetensors",
+    "model.embed_tokens.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.input_layernorm.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.k_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.q_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.v_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.input_layernorm.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.k_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.q_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.v_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.10.input_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.input_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.input_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.13.input_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.13.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.13.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.13.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.13.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.13.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.13.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.13.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.13.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.13.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.13.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.13.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.14.input_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.input_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.input_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.input_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.18.input_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.18.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.18.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.18.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.18.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.18.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.18.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.18.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.18.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.18.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.18.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.18.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.19.input_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.2.input_layernorm.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.k_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.q_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.v_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.20.input_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.input_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.input_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.input_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.24.input_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.24.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.24.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.24.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.24.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.24.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.24.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.24.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.24.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.24.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.24.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.24.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.25.input_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.k_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.q_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.v_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.input_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.k_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.q_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.v_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.input_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.k_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.q_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.v_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.3.input_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.input_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.input_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.input_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.input_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.8.input_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.8.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.8.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.8.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.8.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.8.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.8.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.8.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.8.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.8.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.8.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.8.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.9.input_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+    "model.norm.weight": "model-00006-of-00007.safetensors"
+  }
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-489/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
+  }
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-489/tokenizer_config.json

@@ -0,0 +1,208 @@
+{
+  "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
+    }
+  },
+  "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,
+  "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
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-489/trainer_state.json

@@ -0,0 +1,3457 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 3.0,
+  "eval_steps": 500,
+  "global_step": 489,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.006134969325153374,
+      "grad_norm": 6.9923529624938965,
+      "learning_rate": 5e-06,
+      "loss": 0.8044,
+      "step": 1
+    },
+    {
+      "epoch": 0.012269938650306749,
+      "grad_norm": 5.830728054046631,
+      "learning_rate": 4.999995356617983e-06,
+      "loss": 0.7551,
+      "step": 2
+    },
+    {
+      "epoch": 0.018404907975460124,
+      "grad_norm": 4.734502792358398,
+      "learning_rate": 4.999981426489179e-06,
+      "loss": 1.154,
+      "step": 3
+    },
+    {
+      "epoch": 0.024539877300613498,
+      "grad_norm": 3.2161216735839844,
+      "learning_rate": 4.999958209665336e-06,
+      "loss": 0.6391,
+      "step": 4
+    },
+    {
+      "epoch": 0.03067484662576687,
+      "grad_norm": 3.7258894443511963,
+      "learning_rate": 4.999925706232695e-06,
+      "loss": 0.7033,
+      "step": 5
+    },
+    {
+      "epoch": 0.03680981595092025,
+      "grad_norm": 4.751665115356445,
+      "learning_rate": 4.999883916312e-06,
+      "loss": 0.9834,
+      "step": 6
+    },
+    {
+      "epoch": 0.04294478527607362,
+      "grad_norm": 4.036141872406006,
+      "learning_rate": 4.9998328400584864e-06,
+      "loss": 0.729,
+      "step": 7
+    },
+    {
+      "epoch": 0.049079754601226995,
+      "grad_norm": 4.16400146484375,
+      "learning_rate": 4.999772477661888e-06,
+      "loss": 0.7127,
+      "step": 8
+    },
+    {
+      "epoch": 0.05521472392638037,
+      "grad_norm": 3.304521322250366,
+      "learning_rate": 4.999702829346432e-06,
+      "loss": 0.8199,
+      "step": 9
+    },
+    {
+      "epoch": 0.06134969325153374,
+      "grad_norm": 3.010204315185547,
+      "learning_rate": 4.999623895370843e-06,
+      "loss": 0.8398,
+      "step": 10
+    },
+    {
+      "epoch": 0.06748466257668712,
+      "grad_norm": 2.022270441055298,
+      "learning_rate": 4.999535676028338e-06,
+      "loss": 0.5905,
+      "step": 11
+    },
+    {
+      "epoch": 0.0736196319018405,
+      "grad_norm": 2.387634754180908,
+      "learning_rate": 4.999438171646624e-06,
+      "loss": 0.7349,
+      "step": 12
+    },
+    {
+      "epoch": 0.07975460122699386,
+      "grad_norm": 3.5500376224517822,
+      "learning_rate": 4.999331382587901e-06,
+      "loss": 0.684,
+      "step": 13
+    },
+    {
+      "epoch": 0.08588957055214724,
+      "grad_norm": 2.570266008377075,
+      "learning_rate": 4.999215309248861e-06,
+      "loss": 0.6058,
+      "step": 14
+    },
+    {
+      "epoch": 0.09202453987730061,
+      "grad_norm": 3.104067802429199,
+      "learning_rate": 4.999089952060681e-06,
+      "loss": 0.7153,
+      "step": 15
+    },
+    {
+      "epoch": 0.09815950920245399,
+      "grad_norm": 2.6090075969696045,
+      "learning_rate": 4.998955311489025e-06,
+      "loss": 0.6817,
+      "step": 16
+    },
+    {
+      "epoch": 0.10429447852760736,
+      "grad_norm": 2.0669596195220947,
+      "learning_rate": 4.998811388034046e-06,
+      "loss": 0.5734,
+      "step": 17
+    },
+    {
+      "epoch": 0.11042944785276074,
+      "grad_norm": 1.6724618673324585,
+      "learning_rate": 4.9986581822303746e-06,
+      "loss": 0.5803,
+      "step": 18
+    },
+    {
+      "epoch": 0.1165644171779141,
+      "grad_norm": 2.2240495681762695,
+      "learning_rate": 4.998495694647127e-06,
+      "loss": 0.8003,
+      "step": 19
+    },
+    {
+      "epoch": 0.12269938650306748,
+      "grad_norm": 2.3481221199035645,
+      "learning_rate": 4.998323925887895e-06,
+      "loss": 1.1675,
+      "step": 20
+    },
+    {
+      "epoch": 0.12883435582822086,
+      "grad_norm": 2.2566070556640625,
+      "learning_rate": 4.998142876590749e-06,
+      "loss": 0.4282,
+      "step": 21
+    },
+    {
+      "epoch": 0.13496932515337423,
+      "grad_norm": 1.4764599800109863,
+      "learning_rate": 4.997952547428236e-06,
+      "loss": 0.3757,
+      "step": 22
+    },
+    {
+      "epoch": 0.1411042944785276,
+      "grad_norm": 1.742960810661316,
+      "learning_rate": 4.997752939107372e-06,
+      "loss": 0.519,
+      "step": 23
+    },
+    {
+      "epoch": 0.147239263803681,
+      "grad_norm": 2.158172845840454,
+      "learning_rate": 4.997544052369642e-06,
+      "loss": 0.747,
+      "step": 24
+    },
+    {
+      "epoch": 0.15337423312883436,
+      "grad_norm": 2.3133811950683594,
+      "learning_rate": 4.997325887990999e-06,
+      "loss": 0.7648,
+      "step": 25
+    },
+    {
+      "epoch": 0.15950920245398773,
+      "grad_norm": 1.7891194820404053,
+      "learning_rate": 4.997098446781861e-06,
+      "loss": 0.5636,
+      "step": 26
+    },
+    {
+      "epoch": 0.1656441717791411,
+      "grad_norm": 1.831465721130371,
+      "learning_rate": 4.996861729587103e-06,
+      "loss": 0.5758,
+      "step": 27
+    },
+    {
+      "epoch": 0.17177914110429449,
+      "grad_norm": 1.8467206954956055,
+      "learning_rate": 4.996615737286061e-06,
+      "loss": 0.4968,
+      "step": 28
+    },
+    {
+      "epoch": 0.17791411042944785,
+      "grad_norm": 1.4383373260498047,
+      "learning_rate": 4.996360470792524e-06,
+      "loss": 0.4592,
+      "step": 29
+    },
+    {
+      "epoch": 0.18404907975460122,
+      "grad_norm": 1.9713517427444458,
+      "learning_rate": 4.996095931054731e-06,
+      "loss": 0.5191,
+      "step": 30
+    },
+    {
+      "epoch": 0.1901840490797546,
+      "grad_norm": 1.5770502090454102,
+      "learning_rate": 4.9958221190553705e-06,
+      "loss": 0.5222,
+      "step": 31
+    },
+    {
+      "epoch": 0.19631901840490798,
+      "grad_norm": 1.8299424648284912,
+      "learning_rate": 4.995539035811572e-06,
+      "loss": 0.5229,
+      "step": 32
+    },
+    {
+      "epoch": 0.20245398773006135,
+      "grad_norm": 1.9029279947280884,
+      "learning_rate": 4.9952466823749076e-06,
+      "loss": 0.4647,
+      "step": 33
+    },
+    {
+      "epoch": 0.2085889570552147,
+      "grad_norm": 2.3988473415374756,
+      "learning_rate": 4.9949450598313835e-06,
+      "loss": 0.5923,
+      "step": 34
+    },
+    {
+      "epoch": 0.2147239263803681,
+      "grad_norm": 1.842591643333435,
+      "learning_rate": 4.994634169301439e-06,
+      "loss": 0.6943,
+      "step": 35
+    },
+    {
+      "epoch": 0.22085889570552147,
+      "grad_norm": 2.348345994949341,
+      "learning_rate": 4.994314011939941e-06,
+      "loss": 0.8154,
+      "step": 36
+    },
+    {
+      "epoch": 0.22699386503067484,
+      "grad_norm": 1.7699306011199951,
+      "learning_rate": 4.99398458893618e-06,
+      "loss": 0.673,
+      "step": 37
+    },
+    {
+      "epoch": 0.2331288343558282,
+      "grad_norm": 2.050677537918091,
+      "learning_rate": 4.993645901513865e-06,
+      "loss": 0.7914,
+      "step": 38
+    },
+    {
+      "epoch": 0.2392638036809816,
+      "grad_norm": 2.040433168411255,
+      "learning_rate": 4.993297950931121e-06,
+      "loss": 0.5829,
+      "step": 39
+    },
+    {
+      "epoch": 0.24539877300613497,
+      "grad_norm": 1.7105989456176758,
+      "learning_rate": 4.9929407384804806e-06,
+      "loss": 0.7158,
+      "step": 40
+    },
+    {
+      "epoch": 0.25153374233128833,
+      "grad_norm": 1.5662792921066284,
+      "learning_rate": 4.992574265488883e-06,
+      "loss": 0.5753,
+      "step": 41
+    },
+    {
+      "epoch": 0.25766871165644173,
+      "grad_norm": 1.6192151308059692,
+      "learning_rate": 4.9921985333176694e-06,
+      "loss": 0.4917,
+      "step": 42
+    },
+    {
+      "epoch": 0.26380368098159507,
+      "grad_norm": 1.558786392211914,
+      "learning_rate": 4.991813543362572e-06,
+      "loss": 0.4652,
+      "step": 43
+    },
+    {
+      "epoch": 0.26993865030674846,
+      "grad_norm": 1.7447681427001953,
+      "learning_rate": 4.991419297053716e-06,
+      "loss": 0.6159,
+      "step": 44
+    },
+    {
+      "epoch": 0.27607361963190186,
+      "grad_norm": 1.6782058477401733,
+      "learning_rate": 4.991015795855611e-06,
+      "loss": 0.5053,
+      "step": 45
+    },
+    {
+      "epoch": 0.2822085889570552,
+      "grad_norm": 1.6515028476715088,
+      "learning_rate": 4.990603041267144e-06,
+      "loss": 0.5635,
+      "step": 46
+    },
+    {
+      "epoch": 0.2883435582822086,
+      "grad_norm": 1.4562774896621704,
+      "learning_rate": 4.990181034821578e-06,
+      "loss": 0.4378,
+      "step": 47
+    },
+    {
+      "epoch": 0.294478527607362,
+      "grad_norm": 1.8532418012619019,
+      "learning_rate": 4.98974977808654e-06,
+      "loss": 0.5417,
+      "step": 48
+    },
+    {
+      "epoch": 0.3006134969325153,
+      "grad_norm": 1.7704395055770874,
+      "learning_rate": 4.989309272664026e-06,
+      "loss": 0.5658,
+      "step": 49
+    },
+    {
+      "epoch": 0.3067484662576687,
+      "grad_norm": 2.7695469856262207,
+      "learning_rate": 4.988859520190381e-06,
+      "loss": 0.8042,
+      "step": 50
+    },
+    {
+      "epoch": 0.3128834355828221,
+      "grad_norm": 1.949196219444275,
+      "learning_rate": 4.988400522336304e-06,
+      "loss": 0.6629,
+      "step": 51
+    },
+    {
+      "epoch": 0.31901840490797545,
+      "grad_norm": 2.0406854152679443,
+      "learning_rate": 4.9879322808068365e-06,
+      "loss": 0.6338,
+      "step": 52
+    },
+    {
+      "epoch": 0.32515337423312884,
+      "grad_norm": 1.505089282989502,
+      "learning_rate": 4.987454797341358e-06,
+      "loss": 0.4152,
+      "step": 53
+    },
+    {
+      "epoch": 0.3312883435582822,
+      "grad_norm": 1.9921419620513916,
+      "learning_rate": 4.98696807371358e-06,
+      "loss": 0.6265,
+      "step": 54
+    },
+    {
+      "epoch": 0.3374233128834356,
+      "grad_norm": 1.885555624961853,
+      "learning_rate": 4.986472111731536e-06,
+      "loss": 0.6234,
+      "step": 55
+    },
+    {
+      "epoch": 0.34355828220858897,
+      "grad_norm": 1.969870924949646,
+      "learning_rate": 4.985966913237581e-06,
+      "loss": 0.4797,
+      "step": 56
+    },
+    {
+      "epoch": 0.3496932515337423,
+      "grad_norm": 1.6400866508483887,
+      "learning_rate": 4.985452480108376e-06,
+      "loss": 0.5631,
+      "step": 57
+    },
+    {
+      "epoch": 0.3558282208588957,
+      "grad_norm": 1.6498558521270752,
+      "learning_rate": 4.984928814254889e-06,
+      "loss": 0.5423,
+      "step": 58
+    },
+    {
+      "epoch": 0.3619631901840491,
+      "grad_norm": 2.8436930179595947,
+      "learning_rate": 4.984395917622387e-06,
+      "loss": 0.6798,
+      "step": 59
+    },
+    {
+      "epoch": 0.36809815950920244,
+      "grad_norm": 1.96578049659729,
+      "learning_rate": 4.9838537921904206e-06,
+      "loss": 0.6465,
+      "step": 60
+    },
+    {
+      "epoch": 0.37423312883435583,
+      "grad_norm": 1.9277716875076294,
+      "learning_rate": 4.9833024399728295e-06,
+      "loss": 0.7242,
+      "step": 61
+    },
+    {
+      "epoch": 0.3803680981595092,
+      "grad_norm": 1.6608657836914062,
+      "learning_rate": 4.982741863017722e-06,
+      "loss": 0.4709,
+      "step": 62
+    },
+    {
+      "epoch": 0.38650306748466257,
+      "grad_norm": 1.7999095916748047,
+      "learning_rate": 4.982172063407479e-06,
+      "loss": 0.7222,
+      "step": 63
+    },
+    {
+      "epoch": 0.39263803680981596,
+      "grad_norm": 1.6642130613327026,
+      "learning_rate": 4.9815930432587365e-06,
+      "loss": 0.4714,
+      "step": 64
+    },
+    {
+      "epoch": 0.3987730061349693,
+      "grad_norm": 1.509299874305725,
+      "learning_rate": 4.981004804722384e-06,
+      "loss": 0.4674,
+      "step": 65
+    },
+    {
+      "epoch": 0.4049079754601227,
+      "grad_norm": 1.6120250225067139,
+      "learning_rate": 4.980407349983556e-06,
+      "loss": 0.5602,
+      "step": 66
+    },
+    {
+      "epoch": 0.4110429447852761,
+      "grad_norm": 1.5738871097564697,
+      "learning_rate": 4.979800681261619e-06,
+      "loss": 0.5359,
+      "step": 67
+    },
+    {
+      "epoch": 0.4171779141104294,
+      "grad_norm": 1.5814478397369385,
+      "learning_rate": 4.9791848008101705e-06,
+      "loss": 0.4154,
+      "step": 68
+    },
+    {
+      "epoch": 0.4233128834355828,
+      "grad_norm": 1.4176090955734253,
+      "learning_rate": 4.978559710917024e-06,
+      "loss": 0.5474,
+      "step": 69
+    },
+    {
+      "epoch": 0.4294478527607362,
+      "grad_norm": 2.648993492126465,
+      "learning_rate": 4.977925413904205e-06,
+      "loss": 0.7893,
+      "step": 70
+    },
+    {
+      "epoch": 0.43558282208588955,
+      "grad_norm": 1.9104279279708862,
+      "learning_rate": 4.9772819121279395e-06,
+      "loss": 0.6817,
+      "step": 71
+    },
+    {
+      "epoch": 0.44171779141104295,
+      "grad_norm": 1.9021060466766357,
+      "learning_rate": 4.976629207978648e-06,
+      "loss": 0.5561,
+      "step": 72
+    },
+    {
+      "epoch": 0.44785276073619634,
+      "grad_norm": 1.9069586992263794,
+      "learning_rate": 4.975967303880933e-06,
+      "loss": 0.4215,
+      "step": 73
+    },
+    {
+      "epoch": 0.4539877300613497,
+      "grad_norm": 1.6458418369293213,
+      "learning_rate": 4.975296202293575e-06,
+      "loss": 0.5461,
+      "step": 74
+    },
+    {
+      "epoch": 0.4601226993865031,
+      "grad_norm": 1.8261065483093262,
+      "learning_rate": 4.974615905709518e-06,
+      "loss": 0.582,
+      "step": 75
+    },
+    {
+      "epoch": 0.4662576687116564,
+      "grad_norm": 1.709266185760498,
+      "learning_rate": 4.973926416655863e-06,
+      "loss": 0.5228,
+      "step": 76
+    },
+    {
+      "epoch": 0.4723926380368098,
+      "grad_norm": 1.83115553855896,
+      "learning_rate": 4.973227737693858e-06,
+      "loss": 0.4916,
+      "step": 77
+    },
+    {
+      "epoch": 0.4785276073619632,
+      "grad_norm": 2.158396005630493,
+      "learning_rate": 4.972519871418894e-06,
+      "loss": 0.6101,
+      "step": 78
+    },
+    {
+      "epoch": 0.48466257668711654,
+      "grad_norm": 1.3959546089172363,
+      "learning_rate": 4.971802820460481e-06,
+      "loss": 0.4591,
+      "step": 79
+    },
+    {
+      "epoch": 0.49079754601226994,
+      "grad_norm": 1.7163394689559937,
+      "learning_rate": 4.971076587482254e-06,
+      "loss": 0.4643,
+      "step": 80
+    },
+    {
+      "epoch": 0.49693251533742333,
+      "grad_norm": 2.133448362350464,
+      "learning_rate": 4.970341175181957e-06,
+      "loss": 0.5664,
+      "step": 81
+    },
+    {
+      "epoch": 0.5030674846625767,
+      "grad_norm": 2.017498016357422,
+      "learning_rate": 4.969596586291425e-06,
+      "loss": 0.6118,
+      "step": 82
+    },
+    {
+      "epoch": 0.50920245398773,
+      "grad_norm": 2.5697553157806396,
+      "learning_rate": 4.968842823576592e-06,
+      "loss": 0.6501,
+      "step": 83
+    },
+    {
+      "epoch": 0.5153374233128835,
+      "grad_norm": 1.6452585458755493,
+      "learning_rate": 4.968079889837461e-06,
+      "loss": 0.6388,
+      "step": 84
+    },
+    {
+      "epoch": 0.5214723926380368,
+      "grad_norm": 1.419793725013733,
+      "learning_rate": 4.967307787908108e-06,
+      "loss": 0.5053,
+      "step": 85
+    },
+    {
+      "epoch": 0.5276073619631901,
+      "grad_norm": 2.2178261280059814,
+      "learning_rate": 4.966526520656663e-06,
+      "loss": 0.6009,
+      "step": 86
+    },
+    {
+      "epoch": 0.5337423312883436,
+      "grad_norm": 2.477430820465088,
+      "learning_rate": 4.965736090985305e-06,
+      "loss": 0.7252,
+      "step": 87
+    },
+    {
+      "epoch": 0.5398773006134969,
+      "grad_norm": 1.6238809823989868,
+      "learning_rate": 4.964936501830246e-06,
+      "loss": 0.4776,
+      "step": 88
+    },
+    {
+      "epoch": 0.5460122699386503,
+      "grad_norm": 1.4969124794006348,
+      "learning_rate": 4.964127756161727e-06,
+      "loss": 0.4766,
+      "step": 89
+    },
+    {
+      "epoch": 0.5521472392638037,
+      "grad_norm": 1.8127515316009521,
+      "learning_rate": 4.963309856983998e-06,
+      "loss": 0.5787,
+      "step": 90
+    },
+    {
+      "epoch": 0.558282208588957,
+      "grad_norm": 1.8815516233444214,
+      "learning_rate": 4.9624828073353144e-06,
+      "loss": 0.6682,
+      "step": 91
+    },
+    {
+      "epoch": 0.5644171779141104,
+      "grad_norm": 1.7554055452346802,
+      "learning_rate": 4.961646610287922e-06,
+      "loss": 0.5965,
+      "step": 92
+    },
+    {
+      "epoch": 0.5705521472392638,
+      "grad_norm": 1.7013734579086304,
+      "learning_rate": 4.960801268948047e-06,
+      "loss": 0.5281,
+      "step": 93
+    },
+    {
+      "epoch": 0.5766871165644172,
+      "grad_norm": 1.7376574277877808,
+      "learning_rate": 4.959946786455882e-06,
+      "loss": 0.3861,
+      "step": 94
+    },
+    {
+      "epoch": 0.5828220858895705,
+      "grad_norm": 1.668974757194519,
+      "learning_rate": 4.959083165985581e-06,
+      "loss": 0.4878,
+      "step": 95
+    },
+    {
+      "epoch": 0.588957055214724,
+      "grad_norm": 1.6796667575836182,
+      "learning_rate": 4.958210410745237e-06,
+      "loss": 0.636,
+      "step": 96
+    },
+    {
+      "epoch": 0.5950920245398773,
+      "grad_norm": 1.555358648300171,
+      "learning_rate": 4.957328523976879e-06,
+      "loss": 0.3842,
+      "step": 97
+    },
+    {
+      "epoch": 0.6012269938650306,
+      "grad_norm": 1.42095148563385,
+      "learning_rate": 4.956437508956458e-06,
+      "loss": 0.531,
+      "step": 98
+    },
+    {
+      "epoch": 0.6073619631901841,
+      "grad_norm": 1.4321339130401611,
+      "learning_rate": 4.9555373689938325e-06,
+      "loss": 0.4642,
+      "step": 99
+    },
+    {
+      "epoch": 0.6134969325153374,
+      "grad_norm": 1.768678069114685,
+      "learning_rate": 4.954628107432757e-06,
+      "loss": 0.7276,
+      "step": 100
+    },
+    {
+      "epoch": 0.6196319018404908,
+      "grad_norm": 1.768539547920227,
+      "learning_rate": 4.95370972765087e-06,
+      "loss": 0.5959,
+      "step": 101
+    },
+    {
+      "epoch": 0.6257668711656442,
+      "grad_norm": 1.848176121711731,
+      "learning_rate": 4.952782233059683e-06,
+      "loss": 0.4058,
+      "step": 102
+    },
+    {
+      "epoch": 0.6319018404907976,
+      "grad_norm": 2.5831241607666016,
+      "learning_rate": 4.951845627104565e-06,
+      "loss": 0.6992,
+      "step": 103
+    },
+    {
+      "epoch": 0.6380368098159509,
+      "grad_norm": 1.7924416065216064,
+      "learning_rate": 4.95089991326473e-06,
+      "loss": 0.7086,
+      "step": 104
+    },
+    {
+      "epoch": 0.6441717791411042,
+      "grad_norm": 1.745954155921936,
+      "learning_rate": 4.9499450950532305e-06,
+      "loss": 0.6957,
+      "step": 105
+    },
+    {
+      "epoch": 0.6503067484662577,
+      "grad_norm": 1.3389885425567627,
+      "learning_rate": 4.94898117601693e-06,
+      "loss": 0.5959,
+      "step": 106
+    },
+    {
+      "epoch": 0.656441717791411,
+      "grad_norm": 1.700087070465088,
+      "learning_rate": 4.948008159736507e-06,
+      "loss": 0.4963,
+      "step": 107
+    },
+    {
+      "epoch": 0.6625766871165644,
+      "grad_norm": 1.5351721048355103,
+      "learning_rate": 4.94702604982643e-06,
+      "loss": 0.4459,
+      "step": 108
+    },
+    {
+      "epoch": 0.6687116564417178,
+      "grad_norm": 1.7808955907821655,
+      "learning_rate": 4.9460348499349485e-06,
+      "loss": 0.618,
+      "step": 109
+    },
+    {
+      "epoch": 0.6748466257668712,
+      "grad_norm": 1.672696828842163,
+      "learning_rate": 4.945034563744077e-06,
+      "loss": 0.4872,
+      "step": 110
+    },
+    {
+      "epoch": 0.6809815950920245,
+      "grad_norm": 1.7819981575012207,
+      "learning_rate": 4.944025194969586e-06,
+      "loss": 0.4841,
+      "step": 111
+    },
+    {
+      "epoch": 0.6871165644171779,
+      "grad_norm": 1.878158688545227,
+      "learning_rate": 4.9430067473609825e-06,
+      "loss": 0.6865,
+      "step": 112
+    },
+    {
+      "epoch": 0.6932515337423313,
+      "grad_norm": 1.541741132736206,
+      "learning_rate": 4.941979224701499e-06,
+      "loss": 0.635,
+      "step": 113
+    },
+    {
+      "epoch": 0.6993865030674846,
+      "grad_norm": 1.5765048265457153,
+      "learning_rate": 4.94094263080808e-06,
+      "loss": 0.523,
+      "step": 114
+    },
+    {
+      "epoch": 0.7055214723926381,
+      "grad_norm": 1.8286758661270142,
+      "learning_rate": 4.939896969531367e-06,
+      "loss": 0.8787,
+      "step": 115
+    },
+    {
+      "epoch": 0.7116564417177914,
+      "grad_norm": 1.7785090208053589,
+      "learning_rate": 4.938842244755683e-06,
+      "loss": 0.6749,
+      "step": 116
+    },
+    {
+      "epoch": 0.7177914110429447,
+      "grad_norm": 1.766484022140503,
+      "learning_rate": 4.937778460399022e-06,
+      "loss": 0.7022,
+      "step": 117
+    },
+    {
+      "epoch": 0.7239263803680982,
+      "grad_norm": 2.141592502593994,
+      "learning_rate": 4.936705620413028e-06,
+      "loss": 0.4353,
+      "step": 118
+    },
+    {
+      "epoch": 0.7300613496932515,
+      "grad_norm": 1.4721283912658691,
+      "learning_rate": 4.935623728782986e-06,
+      "loss": 0.4014,
+      "step": 119
+    },
+    {
+      "epoch": 0.7361963190184049,
+      "grad_norm": 1.8953309059143066,
+      "learning_rate": 4.934532789527805e-06,
+      "loss": 0.6815,
+      "step": 120
+    },
+    {
+      "epoch": 0.7423312883435583,
+      "grad_norm": 1.426912784576416,
+      "learning_rate": 4.933432806700004e-06,
+      "loss": 0.4061,
+      "step": 121
+    },
+    {
+      "epoch": 0.7484662576687117,
+      "grad_norm": 1.8180159330368042,
+      "learning_rate": 4.932323784385693e-06,
+      "loss": 0.5724,
+      "step": 122
+    },
+    {
+      "epoch": 0.754601226993865,
+      "grad_norm": 1.8672186136245728,
+      "learning_rate": 4.931205726704566e-06,
+      "loss": 0.5498,
+      "step": 123
+    },
+    {
+      "epoch": 0.7607361963190185,
+      "grad_norm": 1.728786587715149,
+      "learning_rate": 4.930078637809878e-06,
+      "loss": 0.6044,
+      "step": 124
+    },
+    {
+      "epoch": 0.7668711656441718,
+      "grad_norm": 1.6010749340057373,
+      "learning_rate": 4.928942521888431e-06,
+      "loss": 0.5287,
+      "step": 125
+    },
+    {
+      "epoch": 0.7730061349693251,
+      "grad_norm": 1.8714640140533447,
+      "learning_rate": 4.927797383160561e-06,
+      "loss": 0.7863,
+      "step": 126
+    },
+    {
+      "epoch": 0.7791411042944786,
+      "grad_norm": 1.6601593494415283,
+      "learning_rate": 4.926643225880123e-06,
+      "loss": 0.4406,
+      "step": 127
+    },
+    {
+      "epoch": 0.7852760736196319,
+      "grad_norm": 1.7373054027557373,
+      "learning_rate": 4.925480054334471e-06,
+      "loss": 0.4517,
+      "step": 128
+    },
+    {
+      "epoch": 0.7914110429447853,
+      "grad_norm": 2.2996363639831543,
+      "learning_rate": 4.924307872844444e-06,
+      "loss": 0.8735,
+      "step": 129
+    },
+    {
+      "epoch": 0.7975460122699386,
+      "grad_norm": 1.764446496963501,
+      "learning_rate": 4.923126685764351e-06,
+      "loss": 0.5479,
+      "step": 130
+    },
+    {
+      "epoch": 0.803680981595092,
+      "grad_norm": 1.4068061113357544,
+      "learning_rate": 4.921936497481956e-06,
+      "loss": 0.4837,
+      "step": 131
+    },
+    {
+      "epoch": 0.8098159509202454,
+      "grad_norm": 1.430518388748169,
+      "learning_rate": 4.920737312418456e-06,
+      "loss": 0.5074,
+      "step": 132
+    },
+    {
+      "epoch": 0.8159509202453987,
+      "grad_norm": 1.5003552436828613,
+      "learning_rate": 4.919529135028473e-06,
+      "loss": 0.5877,
+      "step": 133
+    },
+    {
+      "epoch": 0.8220858895705522,
+      "grad_norm": 2.147737503051758,
+      "learning_rate": 4.918311969800027e-06,
+      "loss": 0.5747,
+      "step": 134
+    },
+    {
+      "epoch": 0.8282208588957055,
+      "grad_norm": 1.451145887374878,
+      "learning_rate": 4.917085821254532e-06,
+      "loss": 0.5395,
+      "step": 135
+    },
+    {
+      "epoch": 0.8343558282208589,
+      "grad_norm": 1.1156998872756958,
+      "learning_rate": 4.915850693946766e-06,
+      "loss": 0.298,
+      "step": 136
+    },
+    {
+      "epoch": 0.8404907975460123,
+      "grad_norm": 2.0387191772460938,
+      "learning_rate": 4.914606592464865e-06,
+      "loss": 0.5912,
+      "step": 137
+    },
+    {
+      "epoch": 0.8466257668711656,
+      "grad_norm": 2.7926077842712402,
+      "learning_rate": 4.9133535214303e-06,
+      "loss": 0.7912,
+      "step": 138
+    },
+    {
+      "epoch": 0.852760736196319,
+      "grad_norm": 1.437120795249939,
+      "learning_rate": 4.91209148549786e-06,
+      "loss": 0.5394,
+      "step": 139
+    },
+    {
+      "epoch": 0.8588957055214724,
+      "grad_norm": 1.5832260847091675,
+      "learning_rate": 4.910820489355637e-06,
+      "loss": 0.5652,
+      "step": 140
+    },
+    {
+      "epoch": 0.8650306748466258,
+      "grad_norm": 1.995019793510437,
+      "learning_rate": 4.909540537725007e-06,
+      "loss": 0.4534,
+      "step": 141
+    },
+    {
+      "epoch": 0.8711656441717791,
+      "grad_norm": 1.9998847246170044,
+      "learning_rate": 4.908251635360616e-06,
+      "loss": 0.8508,
+      "step": 142
+    },
+    {
+      "epoch": 0.8773006134969326,
+      "grad_norm": 1.9584916830062866,
+      "learning_rate": 4.906953787050354e-06,
+      "loss": 0.5554,
+      "step": 143
+    },
+    {
+      "epoch": 0.8834355828220859,
+      "grad_norm": 2.025939702987671,
+      "learning_rate": 4.905646997615347e-06,
+      "loss": 0.4616,
+      "step": 144
+    },
+    {
+      "epoch": 0.8895705521472392,
+      "grad_norm": 1.6374311447143555,
+      "learning_rate": 4.904331271909932e-06,
+      "loss": 0.5916,
+      "step": 145
+    },
+    {
+      "epoch": 0.8957055214723927,
+      "grad_norm": 1.8247662782669067,
+      "learning_rate": 4.903006614821645e-06,
+      "loss": 0.5175,
+      "step": 146
+    },
+    {
+      "epoch": 0.901840490797546,
+      "grad_norm": 1.7309168577194214,
+      "learning_rate": 4.901673031271194e-06,
+      "loss": 0.4879,
+      "step": 147
+    },
+    {
+      "epoch": 0.9079754601226994,
+      "grad_norm": 2.1945736408233643,
+      "learning_rate": 4.900330526212451e-06,
+      "loss": 0.4516,
+      "step": 148
+    },
+    {
+      "epoch": 0.9141104294478528,
+      "grad_norm": 1.799201488494873,
+      "learning_rate": 4.898979104632427e-06,
+      "loss": 0.6766,
+      "step": 149
+    },
+    {
+      "epoch": 0.9202453987730062,
+      "grad_norm": 1.8236511945724487,
+      "learning_rate": 4.897618771551255e-06,
+      "loss": 0.4401,
+      "step": 150
+    },
+    {
+      "epoch": 0.9263803680981595,
+      "grad_norm": 1.6799453496932983,
+      "learning_rate": 4.8962495320221714e-06,
+      "loss": 0.4828,
+      "step": 151
+    },
+    {
+      "epoch": 0.9325153374233128,
+      "grad_norm": 1.7452043294906616,
+      "learning_rate": 4.8948713911315e-06,
+      "loss": 0.6206,
+      "step": 152
+    },
+    {
+      "epoch": 0.9386503067484663,
+      "grad_norm": 1.7726504802703857,
+      "learning_rate": 4.8934843539986266e-06,
+      "loss": 0.5375,
+      "step": 153
+    },
+    {
+      "epoch": 0.9447852760736196,
+      "grad_norm": 1.99801504611969,
+      "learning_rate": 4.892088425775986e-06,
+      "loss": 0.6845,
+      "step": 154
+    },
+    {
+      "epoch": 0.950920245398773,
+      "grad_norm": 2.2314271926879883,
+      "learning_rate": 4.890683611649041e-06,
+      "loss": 0.6218,
+      "step": 155
+    },
+    {
+      "epoch": 0.9570552147239264,
+      "grad_norm": 1.858280897140503,
+      "learning_rate": 4.8892699168362626e-06,
+      "loss": 0.4607,
+      "step": 156
+    },
+    {
+      "epoch": 0.9631901840490797,
+      "grad_norm": 2.170034170150757,
+      "learning_rate": 4.887847346589111e-06,
+      "loss": 0.498,
+      "step": 157
+    },
+    {
+      "epoch": 0.9693251533742331,
+      "grad_norm": 1.6477316617965698,
+      "learning_rate": 4.886415906192015e-06,
+      "loss": 0.3307,
+      "step": 158
+    },
+    {
+      "epoch": 0.9754601226993865,
+      "grad_norm": 1.852866768836975,
+      "learning_rate": 4.884975600962355e-06,
+      "loss": 0.6712,
+      "step": 159
+    },
+    {
+      "epoch": 0.9815950920245399,
+      "grad_norm": 2.0444769859313965,
+      "learning_rate": 4.883526436250441e-06,
+      "loss": 0.5598,
+      "step": 160
+    },
+    {
+      "epoch": 0.9877300613496932,
+      "grad_norm": 1.5719209909439087,
+      "learning_rate": 4.8820684174394935e-06,
+      "loss": 0.5916,
+      "step": 161
+    },
+    {
+      "epoch": 0.9938650306748467,
+      "grad_norm": 2.230224370956421,
+      "learning_rate": 4.880601549945622e-06,
+      "loss": 0.7047,
+      "step": 162
+    },
+    {
+      "epoch": 1.0,
+      "grad_norm": 1.6576162576675415,
+      "learning_rate": 4.879125839217808e-06,
+      "loss": 0.6195,
+      "step": 163
+    },
+    {
+      "epoch": 1.0061349693251533,
+      "grad_norm": 1.9678032398223877,
+      "learning_rate": 4.8776412907378845e-06,
+      "loss": 0.6454,
+      "step": 164
+    },
+    {
+      "epoch": 1.0122699386503067,
+      "grad_norm": 2.2545297145843506,
+      "learning_rate": 4.8761479100205085e-06,
+      "loss": 0.6233,
+      "step": 165
+    },
+    {
+      "epoch": 1.01840490797546,
+      "grad_norm": 1.4177260398864746,
+      "learning_rate": 4.874645702613152e-06,
+      "loss": 0.3,
+      "step": 166
+    },
+    {
+      "epoch": 1.0245398773006136,
+      "grad_norm": 1.2438665628433228,
+      "learning_rate": 4.873134674096072e-06,
+      "loss": 0.2664,
+      "step": 167
+    },
+    {
+      "epoch": 1.030674846625767,
+      "grad_norm": 1.7756118774414062,
+      "learning_rate": 4.871614830082297e-06,
+      "loss": 0.486,
+      "step": 168
+    },
+    {
+      "epoch": 1.0368098159509203,
+      "grad_norm": 2.1051673889160156,
+      "learning_rate": 4.870086176217597e-06,
+      "loss": 0.5931,
+      "step": 169
+    },
+    {
+      "epoch": 1.0429447852760736,
+      "grad_norm": 1.842322587966919,
+      "learning_rate": 4.868548718180473e-06,
+      "loss": 0.4285,
+      "step": 170
+    },
+    {
+      "epoch": 1.049079754601227,
+      "grad_norm": 1.5782299041748047,
+      "learning_rate": 4.867002461682129e-06,
+      "loss": 0.3152,
+      "step": 171
+    },
+    {
+      "epoch": 1.0552147239263803,
+      "grad_norm": 1.3892101049423218,
+      "learning_rate": 4.8654474124664505e-06,
+      "loss": 0.3344,
+      "step": 172
+    },
+    {
+      "epoch": 1.0613496932515338,
+      "grad_norm": 1.085908055305481,
+      "learning_rate": 4.863883576309991e-06,
+      "loss": 0.2761,
+      "step": 173
+    },
+    {
+      "epoch": 1.0674846625766872,
+      "grad_norm": 3.142186403274536,
+      "learning_rate": 4.8623109590219395e-06,
+      "loss": 0.3567,
+      "step": 174
+    },
+    {
+      "epoch": 1.0736196319018405,
+      "grad_norm": 2.2522873878479004,
+      "learning_rate": 4.860729566444106e-06,
+      "loss": 0.4063,
+      "step": 175
+    },
+    {
+      "epoch": 1.0797546012269938,
+      "grad_norm": 1.9580739736557007,
+      "learning_rate": 4.8591394044508985e-06,
+      "loss": 0.367,
+      "step": 176
+    },
+    {
+      "epoch": 1.0858895705521472,
+      "grad_norm": 1.4515142440795898,
+      "learning_rate": 4.857540478949302e-06,
+      "loss": 0.3272,
+      "step": 177
+    },
+    {
+      "epoch": 1.0920245398773005,
+      "grad_norm": 1.7119684219360352,
+      "learning_rate": 4.855932795878852e-06,
+      "loss": 0.6205,
+      "step": 178
+    },
+    {
+      "epoch": 1.098159509202454,
+      "grad_norm": 1.5967235565185547,
+      "learning_rate": 4.854316361211619e-06,
+      "loss": 0.3414,
+      "step": 179
+    },
+    {
+      "epoch": 1.1042944785276074,
+      "grad_norm": 1.6185253858566284,
+      "learning_rate": 4.852691180952183e-06,
+      "loss": 0.452,
+      "step": 180
+    },
+    {
+      "epoch": 1.1104294478527608,
+      "grad_norm": 1.9637706279754639,
+      "learning_rate": 4.851057261137608e-06,
+      "loss": 0.3391,
+      "step": 181
+    },
+    {
+      "epoch": 1.116564417177914,
+      "grad_norm": 1.6783651113510132,
+      "learning_rate": 4.8494146078374274e-06,
+      "loss": 0.3259,
+      "step": 182
+    },
+    {
+      "epoch": 1.1226993865030674,
+      "grad_norm": 1.4616082906723022,
+      "learning_rate": 4.847763227153612e-06,
+      "loss": 0.3545,
+      "step": 183
+    },
+    {
+      "epoch": 1.1288343558282208,
+      "grad_norm": 1.4015353918075562,
+      "learning_rate": 4.846103125220557e-06,
+      "loss": 0.2772,
+      "step": 184
+    },
+    {
+      "epoch": 1.1349693251533743,
+      "grad_norm": 1.8387027978897095,
+      "learning_rate": 4.844434308205052e-06,
+      "loss": 0.4568,
+      "step": 185
+    },
+    {
+      "epoch": 1.1411042944785277,
+      "grad_norm": 1.3309922218322754,
+      "learning_rate": 4.842756782306261e-06,
+      "loss": 0.3697,
+      "step": 186
+    },
+    {
+      "epoch": 1.147239263803681,
+      "grad_norm": 1.3558727502822876,
+      "learning_rate": 4.841070553755697e-06,
+      "loss": 0.293,
+      "step": 187
+    },
+    {
+      "epoch": 1.1533742331288344,
+      "grad_norm": 1.476223111152649,
+      "learning_rate": 4.839375628817205e-06,
+      "loss": 0.38,
+      "step": 188
+    },
+    {
+      "epoch": 1.1595092024539877,
+      "grad_norm": 1.7385125160217285,
+      "learning_rate": 4.837672013786931e-06,
+      "loss": 0.4321,
+      "step": 189
+    },
+    {
+      "epoch": 1.165644171779141,
+      "grad_norm": 1.4258180856704712,
+      "learning_rate": 4.835959714993305e-06,
+      "loss": 0.3884,
+      "step": 190
+    },
+    {
+      "epoch": 1.1717791411042944,
+      "grad_norm": 1.6587196588516235,
+      "learning_rate": 4.8342387387970105e-06,
+      "loss": 0.3894,
+      "step": 191
+    },
+    {
+      "epoch": 1.177914110429448,
+      "grad_norm": 2.1183578968048096,
+      "learning_rate": 4.832509091590968e-06,
+      "loss": 0.5337,
+      "step": 192
+    },
+    {
+      "epoch": 1.1840490797546013,
+      "grad_norm": 1.9579923152923584,
+      "learning_rate": 4.830770779800309e-06,
+      "loss": 0.6102,
+      "step": 193
+    },
+    {
+      "epoch": 1.1901840490797546,
+      "grad_norm": 1.6721203327178955,
+      "learning_rate": 4.829023809882349e-06,
+      "loss": 0.3467,
+      "step": 194
+    },
+    {
+      "epoch": 1.196319018404908,
+      "grad_norm": 1.3854533433914185,
+      "learning_rate": 4.827268188326567e-06,
+      "loss": 0.4115,
+      "step": 195
+    },
+    {
+      "epoch": 1.2024539877300613,
+      "grad_norm": 1.651465892791748,
+      "learning_rate": 4.825503921654582e-06,
+      "loss": 0.5211,
+      "step": 196
+    },
+    {
+      "epoch": 1.2085889570552146,
+      "grad_norm": 2.229555606842041,
+      "learning_rate": 4.823731016420122e-06,
+      "loss": 0.6709,
+      "step": 197
+    },
+    {
+      "epoch": 1.2147239263803682,
+      "grad_norm": 1.436557412147522,
+      "learning_rate": 4.821949479209011e-06,
+      "loss": 0.3011,
+      "step": 198
+    },
+    {
+      "epoch": 1.2208588957055215,
+      "grad_norm": 1.7224531173706055,
+      "learning_rate": 4.820159316639133e-06,
+      "loss": 0.4072,
+      "step": 199
+    },
+    {
+      "epoch": 1.2269938650306749,
+      "grad_norm": 1.3606412410736084,
+      "learning_rate": 4.818360535360418e-06,
+      "loss": 0.3754,
+      "step": 200
+    },
+    {
+      "epoch": 1.2331288343558282,
+      "grad_norm": 1.4690773487091064,
+      "learning_rate": 4.816553142054806e-06,
+      "loss": 0.2789,
+      "step": 201
+    },
+    {
+      "epoch": 1.2392638036809815,
+      "grad_norm": 1.5914554595947266,
+      "learning_rate": 4.814737143436232e-06,
+      "loss": 0.7622,
+      "step": 202
+    },
+    {
+      "epoch": 1.2453987730061349,
+      "grad_norm": 1.682158350944519,
+      "learning_rate": 4.812912546250595e-06,
+      "loss": 0.5109,
+      "step": 203
+    },
+    {
+      "epoch": 1.2515337423312882,
+      "grad_norm": 2.007913112640381,
+      "learning_rate": 4.81107935727574e-06,
+      "loss": 0.8327,
+      "step": 204
+    },
+    {
+      "epoch": 1.2576687116564418,
+      "grad_norm": 0.9559780359268188,
+      "learning_rate": 4.809237583321421e-06,
+      "loss": 0.1719,
+      "step": 205
+    },
+    {
+      "epoch": 1.2638036809815951,
+      "grad_norm": 1.6852002143859863,
+      "learning_rate": 4.807387231229287e-06,
+      "loss": 0.6202,
+      "step": 206
+    },
+    {
+      "epoch": 1.2699386503067485,
+      "grad_norm": 1.6032408475875854,
+      "learning_rate": 4.8055283078728525e-06,
+      "loss": 0.3352,
+      "step": 207
+    },
+    {
+      "epoch": 1.2760736196319018,
+      "grad_norm": 1.7607511281967163,
+      "learning_rate": 4.803660820157468e-06,
+      "loss": 0.6079,
+      "step": 208
+    },
+    {
+      "epoch": 1.2822085889570551,
+      "grad_norm": 1.6461058855056763,
+      "learning_rate": 4.801784775020303e-06,
+      "loss": 0.5819,
+      "step": 209
+    },
+    {
+      "epoch": 1.2883435582822087,
+      "grad_norm": 1.4451910257339478,
+      "learning_rate": 4.799900179430312e-06,
+      "loss": 0.316,
+      "step": 210
+    },
+    {
+      "epoch": 1.294478527607362,
+      "grad_norm": 1.8077638149261475,
+      "learning_rate": 4.798007040388212e-06,
+      "loss": 0.5785,
+      "step": 211
+    },
+    {
+      "epoch": 1.3006134969325154,
+      "grad_norm": 1.281840205192566,
+      "learning_rate": 4.7961053649264585e-06,
+      "loss": 0.5042,
+      "step": 212
+    },
+    {
+      "epoch": 1.3067484662576687,
+      "grad_norm": 1.6926276683807373,
+      "learning_rate": 4.794195160109215e-06,
+      "loss": 0.5844,
+      "step": 213
+    },
+    {
+      "epoch": 1.312883435582822,
+      "grad_norm": 1.3358347415924072,
+      "learning_rate": 4.7922764330323315e-06,
+      "loss": 0.3266,
+      "step": 214
+    },
+    {
+      "epoch": 1.3190184049079754,
+      "grad_norm": 2.0009665489196777,
+      "learning_rate": 4.790349190823313e-06,
+      "loss": 0.3916,
+      "step": 215
+    },
+    {
+      "epoch": 1.3251533742331287,
+      "grad_norm": 1.8225066661834717,
+      "learning_rate": 4.788413440641297e-06,
+      "loss": 0.5581,
+      "step": 216
+    },
+    {
+      "epoch": 1.331288343558282,
+      "grad_norm": 1.732446312904358,
+      "learning_rate": 4.786469189677026e-06,
+      "loss": 0.5667,
+      "step": 217
+    },
+    {
+      "epoch": 1.3374233128834356,
+      "grad_norm": 1.8246673345565796,
+      "learning_rate": 4.784516445152821e-06,
+      "loss": 0.3854,
+      "step": 218
+    },
+    {
+      "epoch": 1.343558282208589,
+      "grad_norm": 1.5674211978912354,
+      "learning_rate": 4.78255521432255e-06,
+      "loss": 0.4581,
+      "step": 219
+    },
+    {
+      "epoch": 1.3496932515337423,
+      "grad_norm": 1.581689715385437,
+      "learning_rate": 4.780585504471612e-06,
+      "loss": 0.6761,
+      "step": 220
+    },
+    {
+      "epoch": 1.3558282208588956,
+      "grad_norm": 1.56204092502594,
+      "learning_rate": 4.778607322916896e-06,
+      "loss": 0.3545,
+      "step": 221
+    },
+    {
+      "epoch": 1.3619631901840492,
+      "grad_norm": 1.5203900337219238,
+      "learning_rate": 4.776620677006766e-06,
+      "loss": 0.4182,
+      "step": 222
+    },
+    {
+      "epoch": 1.3680981595092025,
+      "grad_norm": 1.3753960132598877,
+      "learning_rate": 4.7746255741210256e-06,
+      "loss": 0.4366,
+      "step": 223
+    },
+    {
+      "epoch": 1.3742331288343559,
+      "grad_norm": 1.5160562992095947,
+      "learning_rate": 4.772622021670897e-06,
+      "loss": 0.3919,
+      "step": 224
+    },
+    {
+      "epoch": 1.3803680981595092,
+      "grad_norm": 1.787851095199585,
+      "learning_rate": 4.770610027098983e-06,
+      "loss": 0.4604,
+      "step": 225
+    },
+    {
+      "epoch": 1.3865030674846626,
+      "grad_norm": 1.2753666639328003,
+      "learning_rate": 4.7685895978792564e-06,
+      "loss": 0.4043,
+      "step": 226
+    },
+    {
+      "epoch": 1.392638036809816,
+      "grad_norm": 1.6157721281051636,
+      "learning_rate": 4.766560741517014e-06,
+      "loss": 0.5779,
+      "step": 227
+    },
+    {
+      "epoch": 1.3987730061349692,
+      "grad_norm": 1.9472342729568481,
+      "learning_rate": 4.76452346554886e-06,
+      "loss": 0.4489,
+      "step": 228
+    },
+    {
+      "epoch": 1.4049079754601226,
+      "grad_norm": 1.9271211624145508,
+      "learning_rate": 4.762477777542676e-06,
+      "loss": 0.4052,
+      "step": 229
+    },
+    {
+      "epoch": 1.4110429447852761,
+      "grad_norm": 1.738323450088501,
+      "learning_rate": 4.7604236850975905e-06,
+      "loss": 0.4796,
+      "step": 230
+    },
+    {
+      "epoch": 1.4171779141104295,
+      "grad_norm": 1.401344895362854,
+      "learning_rate": 4.7583611958439514e-06,
+      "loss": 0.4581,
+      "step": 231
+    },
+    {
+      "epoch": 1.4233128834355828,
+      "grad_norm": 1.653306484222412,
+      "learning_rate": 4.7562903174433e-06,
+      "loss": 0.4598,
+      "step": 232
+    },
+    {
+      "epoch": 1.4294478527607362,
+      "grad_norm": 1.6614420413970947,
+      "learning_rate": 4.75421105758834e-06,
+      "loss": 0.3902,
+      "step": 233
+    },
+    {
+      "epoch": 1.4355828220858895,
+      "grad_norm": 1.6880271434783936,
+      "learning_rate": 4.752123424002908e-06,
+      "loss": 0.34,
+      "step": 234
+    },
+    {
+      "epoch": 1.441717791411043,
+      "grad_norm": 1.9208685159683228,
+      "learning_rate": 4.750027424441949e-06,
+      "loss": 0.599,
+      "step": 235
+    },
+    {
+      "epoch": 1.4478527607361964,
+      "grad_norm": 1.4582723379135132,
+      "learning_rate": 4.747923066691487e-06,
+      "loss": 0.4305,
+      "step": 236
+    },
+    {
+      "epoch": 1.4539877300613497,
+      "grad_norm": 1.7404050827026367,
+      "learning_rate": 4.745810358568588e-06,
+      "loss": 0.5674,
+      "step": 237
+    },
+    {
+      "epoch": 1.460122699386503,
+      "grad_norm": 1.3999745845794678,
+      "learning_rate": 4.743689307921342e-06,
+      "loss": 0.3324,
+      "step": 238
+    },
+    {
+      "epoch": 1.4662576687116564,
+      "grad_norm": 1.839109182357788,
+      "learning_rate": 4.741559922628828e-06,
+      "loss": 0.4068,
+      "step": 239
+    },
+    {
+      "epoch": 1.4723926380368098,
+      "grad_norm": 1.6012935638427734,
+      "learning_rate": 4.739422210601085e-06,
+      "loss": 0.3567,
+      "step": 240
+    },
+    {
+      "epoch": 1.478527607361963,
+      "grad_norm": 1.4487239122390747,
+      "learning_rate": 4.7372761797790836e-06,
+      "loss": 0.3539,
+      "step": 241
+    },
+    {
+      "epoch": 1.4846625766871164,
+      "grad_norm": 1.4824320077896118,
+      "learning_rate": 4.735121838134697e-06,
+      "loss": 0.363,
+      "step": 242
+    },
+    {
+      "epoch": 1.49079754601227,
+      "grad_norm": 1.7434751987457275,
+      "learning_rate": 4.732959193670672e-06,
+      "loss": 0.3935,
+      "step": 243
+    },
+    {
+      "epoch": 1.4969325153374233,
+      "grad_norm": 1.821610927581787,
+      "learning_rate": 4.730788254420593e-06,
+      "loss": 0.3938,
+      "step": 244
+    },
+    {
+      "epoch": 1.5030674846625767,
+      "grad_norm": 1.3840686082839966,
+      "learning_rate": 4.728609028448862e-06,
+      "loss": 0.3826,
+      "step": 245
+    },
+    {
+      "epoch": 1.50920245398773,
+      "grad_norm": 1.3281679153442383,
+      "learning_rate": 4.726421523850662e-06,
+      "loss": 0.5849,
+      "step": 246
+    },
+    {
+      "epoch": 1.5153374233128836,
+      "grad_norm": 1.365745186805725,
+      "learning_rate": 4.7242257487519275e-06,
+      "loss": 0.3154,
+      "step": 247
+    },
+    {
+      "epoch": 1.521472392638037,
+      "grad_norm": 1.6890219449996948,
+      "learning_rate": 4.722021711309317e-06,
+      "loss": 0.4836,
+      "step": 248
+    },
+    {
+      "epoch": 1.5276073619631902,
+      "grad_norm": 1.4030988216400146,
+      "learning_rate": 4.7198094197101826e-06,
+      "loss": 0.3707,
+      "step": 249
+    },
+    {
+      "epoch": 1.5337423312883436,
+      "grad_norm": 1.5182822942733765,
+      "learning_rate": 4.7175888821725335e-06,
+      "loss": 0.3076,
+      "step": 250
+    },
+    {
+      "epoch": 1.539877300613497,
+      "grad_norm": 2.029749631881714,
+      "learning_rate": 4.715360106945015e-06,
+      "loss": 0.7769,
+      "step": 251
+    },
+    {
+      "epoch": 1.5460122699386503,
+      "grad_norm": 1.884514331817627,
+      "learning_rate": 4.713123102306869e-06,
+      "loss": 0.3958,
+      "step": 252
+    },
+    {
+      "epoch": 1.5521472392638036,
+      "grad_norm": 1.5089579820632935,
+      "learning_rate": 4.710877876567912e-06,
+      "loss": 0.3622,
+      "step": 253
+    },
+    {
+      "epoch": 1.558282208588957,
+      "grad_norm": 1.5756475925445557,
+      "learning_rate": 4.708624438068494e-06,
+      "loss": 0.3598,
+      "step": 254
+    },
+    {
+      "epoch": 1.5644171779141103,
+      "grad_norm": 1.944437861442566,
+      "learning_rate": 4.706362795179476e-06,
+      "loss": 0.4663,
+      "step": 255
+    },
+    {
+      "epoch": 1.5705521472392638,
+      "grad_norm": 1.4686535596847534,
+      "learning_rate": 4.7040929563021975e-06,
+      "loss": 0.3772,
+      "step": 256
+    },
+    {
+      "epoch": 1.5766871165644172,
+      "grad_norm": 1.6413816213607788,
+      "learning_rate": 4.70181492986844e-06,
+      "loss": 0.5029,
+      "step": 257
+    },
+    {
+      "epoch": 1.5828220858895705,
+      "grad_norm": 1.6235861778259277,
+      "learning_rate": 4.699528724340401e-06,
+      "loss": 0.3689,
+      "step": 258
+    },
+    {
+      "epoch": 1.588957055214724,
+      "grad_norm": 1.6241801977157593,
+      "learning_rate": 4.6972343482106615e-06,
+      "loss": 0.3999,
+      "step": 259
+    },
+    {
+      "epoch": 1.5950920245398774,
+      "grad_norm": 1.5499999523162842,
+      "learning_rate": 4.6949318100021546e-06,
+      "loss": 0.5234,
+      "step": 260
+    },
+    {
+      "epoch": 1.6012269938650308,
+      "grad_norm": 1.6335983276367188,
+      "learning_rate": 4.6926211182681295e-06,
+      "loss": 0.4691,
+      "step": 261
+    },
+    {
+      "epoch": 1.607361963190184,
+      "grad_norm": 1.815408706665039,
+      "learning_rate": 4.690302281592128e-06,
+      "loss": 0.5875,
+      "step": 262
+    },
+    {
+      "epoch": 1.6134969325153374,
+      "grad_norm": 1.715320110321045,
+      "learning_rate": 4.687975308587944e-06,
+      "loss": 0.4065,
+      "step": 263
+    },
+    {
+      "epoch": 1.6196319018404908,
+      "grad_norm": 1.5997414588928223,
+      "learning_rate": 4.685640207899598e-06,
+      "loss": 0.4049,
+      "step": 264
+    },
+    {
+      "epoch": 1.6257668711656441,
+      "grad_norm": 1.8126248121261597,
+      "learning_rate": 4.683296988201301e-06,
+      "loss": 0.3087,
+      "step": 265
+    },
+    {
+      "epoch": 1.6319018404907975,
+      "grad_norm": 2.1788690090179443,
+      "learning_rate": 4.680945658197425e-06,
+      "loss": 0.6583,
+      "step": 266
+    },
+    {
+      "epoch": 1.6380368098159508,
+      "grad_norm": 1.4271981716156006,
+      "learning_rate": 4.6785862266224695e-06,
+      "loss": 0.4414,
+      "step": 267
+    },
+    {
+      "epoch": 1.6441717791411041,
+      "grad_norm": 1.856804609298706,
+      "learning_rate": 4.676218702241026e-06,
+      "loss": 0.7841,
+      "step": 268
+    },
+    {
+      "epoch": 1.6503067484662577,
+      "grad_norm": 1.598827600479126,
+      "learning_rate": 4.673843093847753e-06,
+      "loss": 0.3861,
+      "step": 269
+    },
+    {
+      "epoch": 1.656441717791411,
+      "grad_norm": 1.4012119770050049,
+      "learning_rate": 4.6714594102673355e-06,
+      "loss": 0.4428,
+      "step": 270
+    },
+    {
+      "epoch": 1.6625766871165644,
+      "grad_norm": 1.854849100112915,
+      "learning_rate": 4.669067660354456e-06,
+      "loss": 0.3792,
+      "step": 271
+    },
+    {
+      "epoch": 1.668711656441718,
+      "grad_norm": 1.586449384689331,
+      "learning_rate": 4.666667852993761e-06,
+      "loss": 0.3957,
+      "step": 272
+    },
+    {
+      "epoch": 1.6748466257668713,
+      "grad_norm": 1.2575863599777222,
+      "learning_rate": 4.664259997099829e-06,
+      "loss": 0.4009,
+      "step": 273
+    },
+    {
+      "epoch": 1.6809815950920246,
+      "grad_norm": 1.533814787864685,
+      "learning_rate": 4.661844101617135e-06,
+      "loss": 0.515,
+      "step": 274
+    },
+    {
+      "epoch": 1.687116564417178,
+      "grad_norm": 1.3487898111343384,
+      "learning_rate": 4.6594201755200205e-06,
+      "loss": 0.4041,
+      "step": 275
+    },
+    {
+      "epoch": 1.6932515337423313,
+      "grad_norm": 1.5758376121520996,
+      "learning_rate": 4.656988227812658e-06,
+      "loss": 0.3834,
+      "step": 276
+    },
+    {
+      "epoch": 1.6993865030674846,
+      "grad_norm": 1.9094839096069336,
+      "learning_rate": 4.654548267529015e-06,
+      "loss": 0.4345,
+      "step": 277
+    },
+    {
+      "epoch": 1.705521472392638,
+      "grad_norm": 1.68153977394104,
+      "learning_rate": 4.652100303732827e-06,
+      "loss": 0.4362,
+      "step": 278
+    },
+    {
+      "epoch": 1.7116564417177913,
+      "grad_norm": 2.002685070037842,
+      "learning_rate": 4.64964434551756e-06,
+      "loss": 0.7991,
+      "step": 279
+    },
+    {
+      "epoch": 1.7177914110429446,
+      "grad_norm": 1.3374216556549072,
+      "learning_rate": 4.647180402006372e-06,
+      "loss": 0.3379,
+      "step": 280
+    },
+    {
+      "epoch": 1.7239263803680982,
+      "grad_norm": 1.9104384183883667,
+      "learning_rate": 4.644708482352093e-06,
+      "loss": 0.5705,
+      "step": 281
+    },
+    {
+      "epoch": 1.7300613496932515,
+      "grad_norm": 1.747294545173645,
+      "learning_rate": 4.6422285957371735e-06,
+      "loss": 0.3882,
+      "step": 282
+    },
+    {
+      "epoch": 1.7361963190184049,
+      "grad_norm": 1.3395034074783325,
+      "learning_rate": 4.639740751373663e-06,
+      "loss": 0.3524,
+      "step": 283
+    },
+    {
+      "epoch": 1.7423312883435584,
+      "grad_norm": 1.3480404615402222,
+      "learning_rate": 4.63724495850317e-06,
+      "loss": 0.4185,
+      "step": 284
+    },
+    {
+      "epoch": 1.7484662576687118,
+      "grad_norm": 1.6395094394683838,
+      "learning_rate": 4.634741226396832e-06,
+      "loss": 0.4902,
+      "step": 285
+    },
+    {
+      "epoch": 1.7546012269938651,
+      "grad_norm": 3.175724983215332,
+      "learning_rate": 4.632229564355275e-06,
+      "loss": 0.3556,
+      "step": 286
+    },
+    {
+      "epoch": 1.7607361963190185,
+      "grad_norm": 2.0897467136383057,
+      "learning_rate": 4.629709981708586e-06,
+      "loss": 0.7291,
+      "step": 287
+    },
+    {
+      "epoch": 1.7668711656441718,
+      "grad_norm": 1.5657193660736084,
+      "learning_rate": 4.6271824878162704e-06,
+      "loss": 0.4604,
+      "step": 288
+    },
+    {
+      "epoch": 1.7730061349693251,
+      "grad_norm": 1.3878685235977173,
+      "learning_rate": 4.624647092067226e-06,
+      "loss": 0.2647,
+      "step": 289
+    },
+    {
+      "epoch": 1.7791411042944785,
+      "grad_norm": 1.4577692747116089,
+      "learning_rate": 4.622103803879702e-06,
+      "loss": 0.2972,
+      "step": 290
+    },
+    {
+      "epoch": 1.7852760736196318,
+      "grad_norm": 1.7305755615234375,
+      "learning_rate": 4.619552632701263e-06,
+      "loss": 0.4544,
+      "step": 291
+    },
+    {
+      "epoch": 1.7914110429447851,
+      "grad_norm": 1.969154953956604,
+      "learning_rate": 4.61699358800876e-06,
+      "loss": 0.5592,
+      "step": 292
+    },
+    {
+      "epoch": 1.7975460122699385,
+      "grad_norm": 1.5827393531799316,
+      "learning_rate": 4.614426679308291e-06,
+      "loss": 0.4336,
+      "step": 293
+    },
+    {
+      "epoch": 1.803680981595092,
+      "grad_norm": 1.8153727054595947,
+      "learning_rate": 4.611851916135166e-06,
+      "loss": 0.4222,
+      "step": 294
+    },
+    {
+      "epoch": 1.8098159509202454,
+      "grad_norm": 1.2182137966156006,
+      "learning_rate": 4.609269308053872e-06,
+      "loss": 0.4382,
+      "step": 295
+    },
+    {
+      "epoch": 1.8159509202453987,
+      "grad_norm": 1.521155595779419,
+      "learning_rate": 4.606678864658039e-06,
+      "loss": 0.5281,
+      "step": 296
+    },
+    {
+      "epoch": 1.8220858895705523,
+      "grad_norm": 2.7794649600982666,
+      "learning_rate": 4.604080595570399e-06,
+      "loss": 0.5388,
+      "step": 297
+    },
+    {
+      "epoch": 1.8282208588957056,
+      "grad_norm": 1.675002932548523,
+      "learning_rate": 4.601474510442759e-06,
+      "loss": 0.3114,
+      "step": 298
+    },
+    {
+      "epoch": 1.834355828220859,
+      "grad_norm": 1.3150911331176758,
+      "learning_rate": 4.598860618955957e-06,
+      "loss": 0.4758,
+      "step": 299
+    },
+    {
+      "epoch": 1.8404907975460123,
+      "grad_norm": 1.495651364326477,
+      "learning_rate": 4.596238930819832e-06,
+      "loss": 0.4235,
+      "step": 300
+    },
+    {
+      "epoch": 1.8466257668711656,
+      "grad_norm": 1.874261736869812,
+      "learning_rate": 4.5936094557731815e-06,
+      "loss": 0.6002,
+      "step": 301
+    },
+    {
+      "epoch": 1.852760736196319,
+      "grad_norm": 1.473367691040039,
+      "learning_rate": 4.590972203583732e-06,
+      "loss": 0.4014,
+      "step": 302
+    },
+    {
+      "epoch": 1.8588957055214723,
+      "grad_norm": 1.536181926727295,
+      "learning_rate": 4.588327184048099e-06,
+      "loss": 0.5189,
+      "step": 303
+    },
+    {
+      "epoch": 1.8650306748466257,
+      "grad_norm": 1.3161652088165283,
+      "learning_rate": 4.585674406991752e-06,
+      "loss": 0.3588,
+      "step": 304
+    },
+    {
+      "epoch": 1.871165644171779,
+      "grad_norm": 1.5687568187713623,
+      "learning_rate": 4.5830138822689755e-06,
+      "loss": 0.5033,
+      "step": 305
+    },
+    {
+      "epoch": 1.8773006134969326,
+      "grad_norm": 1.545177936553955,
+      "learning_rate": 4.5803456197628374e-06,
+      "loss": 0.346,
+      "step": 306
+    },
+    {
+      "epoch": 1.883435582822086,
+      "grad_norm": 1.0760587453842163,
+      "learning_rate": 4.577669629385145e-06,
+      "loss": 0.2668,
+      "step": 307
+    },
+    {
+      "epoch": 1.8895705521472392,
+      "grad_norm": 1.7441003322601318,
+      "learning_rate": 4.574985921076418e-06,
+      "loss": 0.4313,
+      "step": 308
+    },
+    {
+      "epoch": 1.8957055214723928,
+      "grad_norm": 1.603339672088623,
+      "learning_rate": 4.572294504805841e-06,
+      "loss": 0.6583,
+      "step": 309
+    },
+    {
+      "epoch": 1.9018404907975461,
+      "grad_norm": 1.780927300453186,
+      "learning_rate": 4.569595390571232e-06,
+      "loss": 0.4172,
+      "step": 310
+    },
+    {
+      "epoch": 1.9079754601226995,
+      "grad_norm": 1.9701824188232422,
+      "learning_rate": 4.566888588399007e-06,
+      "loss": 0.596,
+      "step": 311
+    },
+    {
+      "epoch": 1.9141104294478528,
+      "grad_norm": 1.222953200340271,
+      "learning_rate": 4.564174108344139e-06,
+      "loss": 0.2627,
+      "step": 312
+    },
+    {
+      "epoch": 1.9202453987730062,
+      "grad_norm": 1.6144652366638184,
+      "learning_rate": 4.561451960490123e-06,
+      "loss": 0.5249,
+      "step": 313
+    },
+    {
+      "epoch": 1.9263803680981595,
+      "grad_norm": 1.619401216506958,
+      "learning_rate": 4.558722154948937e-06,
+      "loss": 0.4996,
+      "step": 314
+    },
+    {
+      "epoch": 1.9325153374233128,
+      "grad_norm": 1.4417848587036133,
+      "learning_rate": 4.5559847018610034e-06,
+      "loss": 0.3363,
+      "step": 315
+    },
+    {
+      "epoch": 1.9386503067484662,
+      "grad_norm": 1.9679949283599854,
+      "learning_rate": 4.553239611395156e-06,
+      "loss": 0.4989,
+      "step": 316
+    },
+    {
+      "epoch": 1.9447852760736195,
+      "grad_norm": 1.3560220003128052,
+      "learning_rate": 4.550486893748596e-06,
+      "loss": 0.2803,
+      "step": 317
+    },
+    {
+      "epoch": 1.9509202453987728,
+      "grad_norm": 1.53361976146698,
+      "learning_rate": 4.547726559146862e-06,
+      "loss": 0.2808,
+      "step": 318
+    },
+    {
+      "epoch": 1.9570552147239264,
+      "grad_norm": 1.320259928703308,
+      "learning_rate": 4.544958617843782e-06,
+      "loss": 0.2725,
+      "step": 319
+    },
+    {
+      "epoch": 1.9631901840490797,
+      "grad_norm": 2.153022527694702,
+      "learning_rate": 4.542183080121444e-06,
+      "loss": 0.5694,
+      "step": 320
+    },
+    {
+      "epoch": 1.969325153374233,
+      "grad_norm": 1.5384567975997925,
+      "learning_rate": 4.539399956290152e-06,
+      "loss": 0.5557,
+      "step": 321
+    },
+    {
+      "epoch": 1.9754601226993866,
+      "grad_norm": 1.2425167560577393,
+      "learning_rate": 4.536609256688396e-06,
+      "loss": 0.3398,
+      "step": 322
+    },
+    {
+      "epoch": 1.98159509202454,
+      "grad_norm": 2.326434850692749,
+      "learning_rate": 4.533810991682799e-06,
+      "loss": 0.4658,
+      "step": 323
+    },
+    {
+      "epoch": 1.9877300613496933,
+      "grad_norm": 1.734060287475586,
+      "learning_rate": 4.531005171668093e-06,
+      "loss": 0.2193,
+      "step": 324
+    },
+    {
+      "epoch": 1.9938650306748467,
+      "grad_norm": 1.6828219890594482,
+      "learning_rate": 4.528191807067074e-06,
+      "loss": 0.4298,
+      "step": 325
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 1.529443621635437,
+      "learning_rate": 4.525370908330564e-06,
+      "loss": 0.346,
+      "step": 326
+    },
+    {
+      "epoch": 2.0061349693251533,
+      "grad_norm": 1.4995561838150024,
+      "learning_rate": 4.522542485937369e-06,
+      "loss": 0.2368,
+      "step": 327
+    },
+    {
+      "epoch": 2.0122699386503067,
+      "grad_norm": 1.757996916770935,
+      "learning_rate": 4.519706550394248e-06,
+      "loss": 0.3509,
+      "step": 328
+    },
+    {
+      "epoch": 2.01840490797546,
+      "grad_norm": 1.7020138502120972,
+      "learning_rate": 4.516863112235864e-06,
+      "loss": 0.4645,
+      "step": 329
+    },
+    {
+      "epoch": 2.0245398773006134,
+      "grad_norm": 1.5239059925079346,
+      "learning_rate": 4.514012182024756e-06,
+      "loss": 0.2381,
+      "step": 330
+    },
+    {
+      "epoch": 2.0306748466257667,
+      "grad_norm": 1.6800508499145508,
+      "learning_rate": 4.511153770351288e-06,
+      "loss": 0.3471,
+      "step": 331
+    },
+    {
+      "epoch": 2.03680981595092,
+      "grad_norm": 2.0719175338745117,
+      "learning_rate": 4.508287887833619e-06,
+      "loss": 0.4505,
+      "step": 332
+    },
+    {
+      "epoch": 2.042944785276074,
+      "grad_norm": 2.1750898361206055,
+      "learning_rate": 4.505414545117658e-06,
+      "loss": 0.4058,
+      "step": 333
+    },
+    {
+      "epoch": 2.049079754601227,
+      "grad_norm": 1.5316275358200073,
+      "learning_rate": 4.502533752877028e-06,
+      "loss": 0.2321,
+      "step": 334
+    },
+    {
+      "epoch": 2.0552147239263805,
+      "grad_norm": 1.8096753358840942,
+      "learning_rate": 4.499645521813024e-06,
+      "loss": 0.2843,
+      "step": 335
+    },
+    {
+      "epoch": 2.061349693251534,
+      "grad_norm": 1.8904895782470703,
+      "learning_rate": 4.496749862654574e-06,
+      "loss": 0.4246,
+      "step": 336
+    },
+    {
+      "epoch": 2.067484662576687,
+      "grad_norm": 2.3869524002075195,
+      "learning_rate": 4.4938467861582e-06,
+      "loss": 0.3769,
+      "step": 337
+    },
+    {
+      "epoch": 2.0736196319018405,
+      "grad_norm": 1.9226906299591064,
+      "learning_rate": 4.490936303107975e-06,
+      "loss": 0.5615,
+      "step": 338
+    },
+    {
+      "epoch": 2.079754601226994,
+      "grad_norm": 1.8863157033920288,
+      "learning_rate": 4.488018424315488e-06,
+      "loss": 0.3174,
+      "step": 339
+    },
+    {
+      "epoch": 2.085889570552147,
+      "grad_norm": 1.8585947751998901,
+      "learning_rate": 4.4850931606198e-06,
+      "loss": 0.3145,
+      "step": 340
+    },
+    {
+      "epoch": 2.0920245398773005,
+      "grad_norm": 1.7092756032943726,
+      "learning_rate": 4.482160522887404e-06,
+      "loss": 0.306,
+      "step": 341
+    },
+    {
+      "epoch": 2.098159509202454,
+      "grad_norm": 1.763797402381897,
+      "learning_rate": 4.479220522012185e-06,
+      "loss": 0.225,
+      "step": 342
+    },
+    {
+      "epoch": 2.104294478527607,
+      "grad_norm": 2.366706371307373,
+      "learning_rate": 4.476273168915382e-06,
+      "loss": 0.4664,
+      "step": 343
+    },
+    {
+      "epoch": 2.1104294478527605,
+      "grad_norm": 2.0161778926849365,
+      "learning_rate": 4.473318474545544e-06,
+      "loss": 0.2855,
+      "step": 344
+    },
+    {
+      "epoch": 2.116564417177914,
+      "grad_norm": 1.492847204208374,
+      "learning_rate": 4.470356449878489e-06,
+      "loss": 0.2564,
+      "step": 345
+    },
+    {
+      "epoch": 2.1226993865030677,
+      "grad_norm": 1.842887043952942,
+      "learning_rate": 4.467387105917269e-06,
+      "loss": 0.2546,
+      "step": 346
+    },
+    {
+      "epoch": 2.128834355828221,
+      "grad_norm": 1.837159276008606,
+      "learning_rate": 4.464410453692122e-06,
+      "loss": 0.4798,
+      "step": 347
+    },
+    {
+      "epoch": 2.1349693251533743,
+      "grad_norm": 1.8717197179794312,
+      "learning_rate": 4.461426504260434e-06,
+      "loss": 0.3368,
+      "step": 348
+    },
+    {
+      "epoch": 2.1411042944785277,
+      "grad_norm": 2.6156680583953857,
+      "learning_rate": 4.458435268706699e-06,
+      "loss": 0.3138,
+      "step": 349
+    },
+    {
+      "epoch": 2.147239263803681,
+      "grad_norm": 1.300620675086975,
+      "learning_rate": 4.455436758142477e-06,
+      "loss": 0.1833,
+      "step": 350
+    },
+    {
+      "epoch": 2.1533742331288344,
+      "grad_norm": 1.4636272192001343,
+      "learning_rate": 4.452430983706351e-06,
+      "loss": 0.2729,
+      "step": 351
+    },
+    {
+      "epoch": 2.1595092024539877,
+      "grad_norm": 1.232196569442749,
+      "learning_rate": 4.44941795656389e-06,
+      "loss": 0.1936,
+      "step": 352
+    },
+    {
+      "epoch": 2.165644171779141,
+      "grad_norm": 2.260737657546997,
+      "learning_rate": 4.446397687907601e-06,
+      "loss": 0.5558,
+      "step": 353
+    },
+    {
+      "epoch": 2.1717791411042944,
+      "grad_norm": 1.4225959777832031,
+      "learning_rate": 4.4433701889568935e-06,
+      "loss": 0.1668,
+      "step": 354
+    },
+    {
+      "epoch": 2.1779141104294477,
+      "grad_norm": 1.796919345855713,
+      "learning_rate": 4.440335470958035e-06,
+      "loss": 0.263,
+      "step": 355
+    },
+    {
+      "epoch": 2.184049079754601,
+      "grad_norm": 1.6292144060134888,
+      "learning_rate": 4.437293545184111e-06,
+      "loss": 0.2887,
+      "step": 356
+    },
+    {
+      "epoch": 2.190184049079755,
+      "grad_norm": 1.4105199575424194,
+      "learning_rate": 4.434244422934976e-06,
+      "loss": 0.2808,
+      "step": 357
+    },
+    {
+      "epoch": 2.196319018404908,
+      "grad_norm": 2.0672659873962402,
+      "learning_rate": 4.431188115537226e-06,
+      "loss": 0.5076,
+      "step": 358
+    },
+    {
+      "epoch": 2.2024539877300615,
+      "grad_norm": 1.6706706285476685,
+      "learning_rate": 4.428124634344141e-06,
+      "loss": 0.1688,
+      "step": 359
+    },
+    {
+      "epoch": 2.208588957055215,
+      "grad_norm": 1.5746945142745972,
+      "learning_rate": 4.425053990735653e-06,
+      "loss": 0.1529,
+      "step": 360
+    },
+    {
+      "epoch": 2.214723926380368,
+      "grad_norm": 1.9231823682785034,
+      "learning_rate": 4.421976196118297e-06,
+      "loss": 0.4083,
+      "step": 361
+    },
+    {
+      "epoch": 2.2208588957055215,
+      "grad_norm": 1.7074155807495117,
+      "learning_rate": 4.4188912619251765e-06,
+      "loss": 0.3044,
+      "step": 362
+    },
+    {
+      "epoch": 2.226993865030675,
+      "grad_norm": 1.7478076219558716,
+      "learning_rate": 4.415799199615912e-06,
+      "loss": 0.2681,
+      "step": 363
+    },
+    {
+      "epoch": 2.233128834355828,
+      "grad_norm": 1.8380169868469238,
+      "learning_rate": 4.4127000206766055e-06,
+      "loss": 0.3466,
+      "step": 364
+    },
+    {
+      "epoch": 2.2392638036809815,
+      "grad_norm": 1.4243403673171997,
+      "learning_rate": 4.409593736619795e-06,
+      "loss": 0.2478,
+      "step": 365
+    },
+    {
+      "epoch": 2.245398773006135,
+      "grad_norm": 1.4997358322143555,
+      "learning_rate": 4.40648035898441e-06,
+      "loss": 0.2759,
+      "step": 366
+    },
+    {
+      "epoch": 2.2515337423312882,
+      "grad_norm": 1.3192161321640015,
+      "learning_rate": 4.403359899335732e-06,
+      "loss": 0.2551,
+      "step": 367
+    },
+    {
+      "epoch": 2.2576687116564416,
+      "grad_norm": 1.8495427370071411,
+      "learning_rate": 4.400232369265351e-06,
+      "loss": 0.2319,
+      "step": 368
+    },
+    {
+      "epoch": 2.263803680981595,
+      "grad_norm": 1.434638261795044,
+      "learning_rate": 4.39709778039112e-06,
+      "loss": 0.3061,
+      "step": 369
+    },
+    {
+      "epoch": 2.2699386503067487,
+      "grad_norm": 3.4450862407684326,
+      "learning_rate": 4.393956144357113e-06,
+      "loss": 0.4023,
+      "step": 370
+    },
+    {
+      "epoch": 2.276073619631902,
+      "grad_norm": 1.3243643045425415,
+      "learning_rate": 4.390807472833585e-06,
+      "loss": 0.3033,
+      "step": 371
+    },
+    {
+      "epoch": 2.2822085889570554,
+      "grad_norm": 1.4476289749145508,
+      "learning_rate": 4.3876517775169216e-06,
+      "loss": 0.2242,
+      "step": 372
+    },
+    {
+      "epoch": 2.2883435582822087,
+      "grad_norm": 1.6636766195297241,
+      "learning_rate": 4.384489070129604e-06,
+      "loss": 0.1947,
+      "step": 373
+    },
+    {
+      "epoch": 2.294478527607362,
+      "grad_norm": 1.710154414176941,
+      "learning_rate": 4.381319362420158e-06,
+      "loss": 0.3506,
+      "step": 374
+    },
+    {
+      "epoch": 2.3006134969325154,
+      "grad_norm": 1.2876087427139282,
+      "learning_rate": 4.378142666163114e-06,
+      "loss": 0.1701,
+      "step": 375
+    },
+    {
+      "epoch": 2.3067484662576687,
+      "grad_norm": 1.6479226350784302,
+      "learning_rate": 4.374958993158965e-06,
+      "loss": 0.458,
+      "step": 376
+    },
+    {
+      "epoch": 2.312883435582822,
+      "grad_norm": 1.3171360492706299,
+      "learning_rate": 4.371768355234116e-06,
+      "loss": 0.1446,
+      "step": 377
+    },
+    {
+      "epoch": 2.3190184049079754,
+      "grad_norm": 1.6110225915908813,
+      "learning_rate": 4.368570764240852e-06,
+      "loss": 0.5192,
+      "step": 378
+    },
+    {
+      "epoch": 2.3251533742331287,
+      "grad_norm": 1.7204008102416992,
+      "learning_rate": 4.365366232057279e-06,
+      "loss": 0.5703,
+      "step": 379
+    },
+    {
+      "epoch": 2.331288343558282,
+      "grad_norm": 1.3460030555725098,
+      "learning_rate": 4.3621547705872915e-06,
+      "loss": 0.207,
+      "step": 380
+    },
+    {
+      "epoch": 2.3374233128834354,
+      "grad_norm": 1.599328637123108,
+      "learning_rate": 4.358936391760524e-06,
+      "loss": 0.2842,
+      "step": 381
+    },
+    {
+      "epoch": 2.3435582822085887,
+      "grad_norm": 1.440152883529663,
+      "learning_rate": 4.355711107532305e-06,
+      "loss": 0.3172,
+      "step": 382
+    },
+    {
+      "epoch": 2.3496932515337425,
+      "grad_norm": 1.3458422422409058,
+      "learning_rate": 4.3524789298836175e-06,
+      "loss": 0.2926,
+      "step": 383
+    },
+    {
+      "epoch": 2.355828220858896,
+      "grad_norm": 1.8103442192077637,
+      "learning_rate": 4.349239870821049e-06,
+      "loss": 0.3151,
+      "step": 384
+    },
+    {
+      "epoch": 2.361963190184049,
+      "grad_norm": 1.3873928785324097,
+      "learning_rate": 4.345993942376752e-06,
+      "loss": 0.2372,
+      "step": 385
+    },
+    {
+      "epoch": 2.3680981595092025,
+      "grad_norm": 2.0732638835906982,
+      "learning_rate": 4.342741156608392e-06,
+      "loss": 0.5985,
+      "step": 386
+    },
+    {
+      "epoch": 2.374233128834356,
+      "grad_norm": 1.370980978012085,
+      "learning_rate": 4.3394815255991135e-06,
+      "loss": 0.2267,
+      "step": 387
+    },
+    {
+      "epoch": 2.3803680981595092,
+      "grad_norm": 1.2192656993865967,
+      "learning_rate": 4.336215061457485e-06,
+      "loss": 0.1905,
+      "step": 388
+    },
+    {
+      "epoch": 2.3865030674846626,
+      "grad_norm": 2.077481508255005,
+      "learning_rate": 4.332941776317458e-06,
+      "loss": 0.3766,
+      "step": 389
+    },
+    {
+      "epoch": 2.392638036809816,
+      "grad_norm": 1.49028480052948,
+      "learning_rate": 4.329661682338325e-06,
+      "loss": 0.2559,
+      "step": 390
+    },
+    {
+      "epoch": 2.3987730061349692,
+      "grad_norm": 2.1250507831573486,
+      "learning_rate": 4.32637479170467e-06,
+      "loss": 0.3021,
+      "step": 391
+    },
+    {
+      "epoch": 2.4049079754601226,
+      "grad_norm": 1.6332277059555054,
+      "learning_rate": 4.323081116626322e-06,
+      "loss": 0.2961,
+      "step": 392
+    },
+    {
+      "epoch": 2.411042944785276,
+      "grad_norm": 1.7477848529815674,
+      "learning_rate": 4.319780669338316e-06,
+      "loss": 0.2753,
+      "step": 393
+    },
+    {
+      "epoch": 2.4171779141104293,
+      "grad_norm": 2.097752094268799,
+      "learning_rate": 4.31647346210084e-06,
+      "loss": 0.7489,
+      "step": 394
+    },
+    {
+      "epoch": 2.4233128834355826,
+      "grad_norm": 1.4820683002471924,
+      "learning_rate": 4.313159507199197e-06,
+      "loss": 0.2,
+      "step": 395
+    },
+    {
+      "epoch": 2.4294478527607364,
+      "grad_norm": 1.348221778869629,
+      "learning_rate": 4.309838816943755e-06,
+      "loss": 0.2193,
+      "step": 396
+    },
+    {
+      "epoch": 2.4355828220858897,
+      "grad_norm": 1.6083935499191284,
+      "learning_rate": 4.306511403669897e-06,
+      "loss": 0.2359,
+      "step": 397
+    },
+    {
+      "epoch": 2.441717791411043,
+      "grad_norm": 1.9602519273757935,
+      "learning_rate": 4.303177279737988e-06,
+      "loss": 0.4448,
+      "step": 398
+    },
+    {
+      "epoch": 2.4478527607361964,
+      "grad_norm": 1.6498708724975586,
+      "learning_rate": 4.299836457533313e-06,
+      "loss": 0.2562,
+      "step": 399
+    },
+    {
+      "epoch": 2.4539877300613497,
+      "grad_norm": 1.685544490814209,
+      "learning_rate": 4.296488949466046e-06,
+      "loss": 0.3617,
+      "step": 400
+    },
+    {
+      "epoch": 2.460122699386503,
+      "grad_norm": 1.6809372901916504,
+      "learning_rate": 4.293134767971193e-06,
+      "loss": 0.3189,
+      "step": 401
+    },
+    {
+      "epoch": 2.4662576687116564,
+      "grad_norm": 1.984054446220398,
+      "learning_rate": 4.28977392550855e-06,
+      "loss": 0.3658,
+      "step": 402
+    },
+    {
+      "epoch": 2.4723926380368098,
+      "grad_norm": 2.175684928894043,
+      "learning_rate": 4.286406434562659e-06,
+      "loss": 0.21,
+      "step": 403
+    },
+    {
+      "epoch": 2.478527607361963,
+      "grad_norm": 1.513929009437561,
+      "learning_rate": 4.283032307642756e-06,
+      "loss": 0.2439,
+      "step": 404
+    },
+    {
+      "epoch": 2.4846625766871164,
+      "grad_norm": 2.032034158706665,
+      "learning_rate": 4.2796515572827305e-06,
+      "loss": 0.2158,
+      "step": 405
+    },
+    {
+      "epoch": 2.4907975460122698,
+      "grad_norm": 1.3851574659347534,
+      "learning_rate": 4.276264196041074e-06,
+      "loss": 0.1417,
+      "step": 406
+    },
+    {
+      "epoch": 2.4969325153374236,
+      "grad_norm": 1.7157762050628662,
+      "learning_rate": 4.2728702365008356e-06,
+      "loss": 0.4353,
+      "step": 407
+    },
+    {
+      "epoch": 2.5030674846625764,
+      "grad_norm": 2.0632054805755615,
+      "learning_rate": 4.269469691269577e-06,
+      "loss": 0.3277,
+      "step": 408
+    },
+    {
+      "epoch": 2.5092024539877302,
+      "grad_norm": 2.2436537742614746,
+      "learning_rate": 4.266062572979323e-06,
+      "loss": 0.4356,
+      "step": 409
+    },
+    {
+      "epoch": 2.5153374233128836,
+      "grad_norm": 1.5137197971343994,
+      "learning_rate": 4.262648894286515e-06,
+      "loss": 0.175,
+      "step": 410
+    },
+    {
+      "epoch": 2.521472392638037,
+      "grad_norm": 1.4493860006332397,
+      "learning_rate": 4.259228667871963e-06,
+      "loss": 0.2174,
+      "step": 411
+    },
+    {
+      "epoch": 2.5276073619631902,
+      "grad_norm": 1.3004103899002075,
+      "learning_rate": 4.255801906440803e-06,
+      "loss": 0.1816,
+      "step": 412
+    },
+    {
+      "epoch": 2.5337423312883436,
+      "grad_norm": 1.3862189054489136,
+      "learning_rate": 4.252368622722443e-06,
+      "loss": 0.322,
+      "step": 413
+    },
+    {
+      "epoch": 2.539877300613497,
+      "grad_norm": 1.1516449451446533,
+      "learning_rate": 4.248928829470522e-06,
+      "loss": 0.2127,
+      "step": 414
+    },
+    {
+      "epoch": 2.5460122699386503,
+      "grad_norm": 1.4983477592468262,
+      "learning_rate": 4.245482539462861e-06,
+      "loss": 0.3524,
+      "step": 415
+    },
+    {
+      "epoch": 2.5521472392638036,
+      "grad_norm": 1.7523754835128784,
+      "learning_rate": 4.242029765501411e-06,
+      "loss": 0.3279,
+      "step": 416
+    },
+    {
+      "epoch": 2.558282208588957,
+      "grad_norm": 1.2261234521865845,
+      "learning_rate": 4.2385705204122104e-06,
+      "loss": 0.2657,
+      "step": 417
+    },
+    {
+      "epoch": 2.5644171779141103,
+      "grad_norm": 2.011181592941284,
+      "learning_rate": 4.235104817045338e-06,
+      "loss": 0.4862,
+      "step": 418
+    },
+    {
+      "epoch": 2.5705521472392636,
+      "grad_norm": 1.5264896154403687,
+      "learning_rate": 4.231632668274861e-06,
+      "loss": 0.3103,
+      "step": 419
+    },
+    {
+      "epoch": 2.5766871165644174,
+      "grad_norm": 1.350937008857727,
+      "learning_rate": 4.22815408699879e-06,
+      "loss": 0.1807,
+      "step": 420
+    },
+    {
+      "epoch": 2.5828220858895703,
+      "grad_norm": 1.3427659273147583,
+      "learning_rate": 4.22466908613903e-06,
+      "loss": 0.2463,
+      "step": 421
+    },
+    {
+      "epoch": 2.588957055214724,
+      "grad_norm": 1.6290534734725952,
+      "learning_rate": 4.221177678641333e-06,
+      "loss": 0.2521,
+      "step": 422
+    },
+    {
+      "epoch": 2.5950920245398774,
+      "grad_norm": 1.3621251583099365,
+      "learning_rate": 4.217679877475251e-06,
+      "loss": 0.1904,
+      "step": 423
+    },
+    {
+      "epoch": 2.6012269938650308,
+      "grad_norm": 1.8909351825714111,
+      "learning_rate": 4.214175695634084e-06,
+      "loss": 0.2176,
+      "step": 424
+    },
+    {
+      "epoch": 2.607361963190184,
+      "grad_norm": 1.4748960733413696,
+      "learning_rate": 4.210665146134838e-06,
+      "loss": 0.1709,
+      "step": 425
+    },
+    {
+      "epoch": 2.6134969325153374,
+      "grad_norm": 1.4763535261154175,
+      "learning_rate": 4.20714824201817e-06,
+      "loss": 0.1469,
+      "step": 426
+    },
+    {
+      "epoch": 2.6196319018404908,
+      "grad_norm": 1.6110737323760986,
+      "learning_rate": 4.203624996348343e-06,
+      "loss": 0.3355,
+      "step": 427
+    },
+    {
+      "epoch": 2.625766871165644,
+      "grad_norm": 1.786254644393921,
+      "learning_rate": 4.200095422213177e-06,
+      "loss": 0.2299,
+      "step": 428
+    },
+    {
+      "epoch": 2.6319018404907975,
+      "grad_norm": 2.2847800254821777,
+      "learning_rate": 4.196559532724004e-06,
+      "loss": 0.5011,
+      "step": 429
+    },
+    {
+      "epoch": 2.638036809815951,
+      "grad_norm": 1.5624232292175293,
+      "learning_rate": 4.193017341015608e-06,
+      "loss": 0.3945,
+      "step": 430
+    },
+    {
+      "epoch": 2.644171779141104,
+      "grad_norm": 1.4905847311019897,
+      "learning_rate": 4.189468860246192e-06,
+      "loss": 0.3572,
+      "step": 431
+    },
+    {
+      "epoch": 2.6503067484662575,
+      "grad_norm": 1.5950927734375,
+      "learning_rate": 4.185914103597316e-06,
+      "loss": 0.2352,
+      "step": 432
+    },
+    {
+      "epoch": 2.6564417177914113,
+      "grad_norm": 1.6100196838378906,
+      "learning_rate": 4.182353084273855e-06,
+      "loss": 0.4812,
+      "step": 433
+    },
+    {
+      "epoch": 2.662576687116564,
+      "grad_norm": 1.8571834564208984,
+      "learning_rate": 4.178785815503946e-06,
+      "loss": 0.327,
+      "step": 434
+    },
+    {
+      "epoch": 2.668711656441718,
+      "grad_norm": 1.401564121246338,
+      "learning_rate": 4.1752123105389444e-06,
+      "loss": 0.3473,
+      "step": 435
+    },
+    {
+      "epoch": 2.6748466257668713,
+      "grad_norm": 1.6578110456466675,
+      "learning_rate": 4.171632582653368e-06,
+      "loss": 0.2416,
+      "step": 436
+    },
+    {
+      "epoch": 2.6809815950920246,
+      "grad_norm": 1.9427868127822876,
+      "learning_rate": 4.168046645144851e-06,
+      "loss": 0.2937,
+      "step": 437
+    },
+    {
+      "epoch": 2.687116564417178,
+      "grad_norm": 1.5119742155075073,
+      "learning_rate": 4.164454511334098e-06,
+      "loss": 0.2928,
+      "step": 438
+    },
+    {
+      "epoch": 2.6932515337423313,
+      "grad_norm": 1.7040557861328125,
+      "learning_rate": 4.160856194564828e-06,
+      "loss": 0.478,
+      "step": 439
+    },
+    {
+      "epoch": 2.6993865030674846,
+      "grad_norm": 1.6068278551101685,
+      "learning_rate": 4.157251708203728e-06,
+      "loss": 0.3757,
+      "step": 440
+    },
+    {
+      "epoch": 2.705521472392638,
+      "grad_norm": 2.0017125606536865,
+      "learning_rate": 4.153641065640402e-06,
+      "loss": 0.5441,
+      "step": 441
+    },
+    {
+      "epoch": 2.7116564417177913,
+      "grad_norm": 1.7390366792678833,
+      "learning_rate": 4.150024280287327e-06,
+      "loss": 0.1915,
+      "step": 442
+    },
+    {
+      "epoch": 2.7177914110429446,
+      "grad_norm": 1.5149441957473755,
+      "learning_rate": 4.146401365579795e-06,
+      "loss": 0.1871,
+      "step": 443
+    },
+    {
+      "epoch": 2.7239263803680984,
+      "grad_norm": 1.7839851379394531,
+      "learning_rate": 4.142772334975868e-06,
+      "loss": 0.3467,
+      "step": 444
+    },
+    {
+      "epoch": 2.7300613496932513,
+      "grad_norm": 1.6114904880523682,
+      "learning_rate": 4.139137201956324e-06,
+      "loss": 0.2356,
+      "step": 445
+    },
+    {
+      "epoch": 2.736196319018405,
+      "grad_norm": 1.484224557876587,
+      "learning_rate": 4.1354959800246155e-06,
+      "loss": 0.2542,
+      "step": 446
+    },
+    {
+      "epoch": 2.7423312883435584,
+      "grad_norm": 1.8020520210266113,
+      "learning_rate": 4.131848682706807e-06,
+      "loss": 0.2444,
+      "step": 447
+    },
+    {
+      "epoch": 2.7484662576687118,
+      "grad_norm": 1.686484456062317,
+      "learning_rate": 4.128195323551536e-06,
+      "loss": 0.2315,
+      "step": 448
+    },
+    {
+      "epoch": 2.754601226993865,
+      "grad_norm": 1.5548804998397827,
+      "learning_rate": 4.1245359161299555e-06,
+      "loss": 0.4567,
+      "step": 449
+    },
+    {
+      "epoch": 2.7607361963190185,
+      "grad_norm": 2.001708984375,
+      "learning_rate": 4.120870474035687e-06,
+      "loss": 0.3574,
+      "step": 450
+    },
+    {
+      "epoch": 2.766871165644172,
+      "grad_norm": 1.7765088081359863,
+      "learning_rate": 4.1171990108847705e-06,
+      "loss": 0.5253,
+      "step": 451
+    },
+    {
+      "epoch": 2.773006134969325,
+      "grad_norm": 1.8426337242126465,
+      "learning_rate": 4.113521540315609e-06,
+      "loss": 0.246,
+      "step": 452
+    },
+    {
+      "epoch": 2.7791411042944785,
+      "grad_norm": 1.6934659481048584,
+      "learning_rate": 4.109838075988922e-06,
+      "loss": 0.4699,
+      "step": 453
+    },
+    {
+      "epoch": 2.785276073619632,
+      "grad_norm": 1.723349690437317,
+      "learning_rate": 4.106148631587697e-06,
+      "loss": 0.3245,
+      "step": 454
+    },
+    {
+      "epoch": 2.791411042944785,
+      "grad_norm": 2.0301599502563477,
+      "learning_rate": 4.102453220817134e-06,
+      "loss": 0.3997,
+      "step": 455
+    },
+    {
+      "epoch": 2.7975460122699385,
+      "grad_norm": 1.7088685035705566,
+      "learning_rate": 4.098751857404595e-06,
+      "loss": 0.2232,
+      "step": 456
+    },
+    {
+      "epoch": 2.8036809815950923,
+      "grad_norm": 1.403787612915039,
+      "learning_rate": 4.0950445550995566e-06,
+      "loss": 0.2832,
+      "step": 457
+    },
+    {
+      "epoch": 2.809815950920245,
+      "grad_norm": 1.2241122722625732,
+      "learning_rate": 4.091331327673554e-06,
+      "loss": 0.2125,
+      "step": 458
+    },
+    {
+      "epoch": 2.815950920245399,
+      "grad_norm": 1.1791332960128784,
+      "learning_rate": 4.087612188920135e-06,
+      "loss": 0.2807,
+      "step": 459
+    },
+    {
+      "epoch": 2.8220858895705523,
+      "grad_norm": 2.1663200855255127,
+      "learning_rate": 4.083887152654804e-06,
+      "loss": 0.2626,
+      "step": 460
+    },
+    {
+      "epoch": 2.8282208588957056,
+      "grad_norm": 1.9503370523452759,
+      "learning_rate": 4.080156232714976e-06,
+      "loss": 0.2926,
+      "step": 461
+    },
+    {
+      "epoch": 2.834355828220859,
+      "grad_norm": 1.3069140911102295,
+      "learning_rate": 4.07641944295992e-06,
+      "loss": 0.2139,
+      "step": 462
+    },
+    {
+      "epoch": 2.8404907975460123,
+      "grad_norm": 1.5166411399841309,
+      "learning_rate": 4.072676797270708e-06,
+      "loss": 0.1883,
+      "step": 463
+    },
+    {
+      "epoch": 2.8466257668711656,
+      "grad_norm": 1.6423773765563965,
+      "learning_rate": 4.0689283095501684e-06,
+      "loss": 0.3312,
+      "step": 464
+    },
+    {
+      "epoch": 2.852760736196319,
+      "grad_norm": 1.7567787170410156,
+      "learning_rate": 4.06517399372283e-06,
+      "loss": 0.2336,
+      "step": 465
+    },
+    {
+      "epoch": 2.8588957055214723,
+      "grad_norm": 1.8323341608047485,
+      "learning_rate": 4.061413863734869e-06,
+      "loss": 0.1874,
+      "step": 466
+    },
+    {
+      "epoch": 2.8650306748466257,
+      "grad_norm": 1.9533634185791016,
+      "learning_rate": 4.057647933554063e-06,
+      "loss": 0.3139,
+      "step": 467
+    },
+    {
+      "epoch": 2.871165644171779,
+      "grad_norm": 1.379435658454895,
+      "learning_rate": 4.053876217169734e-06,
+      "loss": 0.359,
+      "step": 468
+    },
+    {
+      "epoch": 2.8773006134969323,
+      "grad_norm": 1.2710524797439575,
+      "learning_rate": 4.050098728592698e-06,
+      "loss": 0.1438,
+      "step": 469
+    },
+    {
+      "epoch": 2.883435582822086,
+      "grad_norm": 1.7576236724853516,
+      "learning_rate": 4.046315481855211e-06,
+      "loss": 0.483,
+      "step": 470
+    },
+    {
+      "epoch": 2.889570552147239,
+      "grad_norm": 1.8108645677566528,
+      "learning_rate": 4.0425264910109245e-06,
+      "loss": 0.3779,
+      "step": 471
+    },
+    {
+      "epoch": 2.895705521472393,
+      "grad_norm": 1.674763798713684,
+      "learning_rate": 4.03873177013482e-06,
+      "loss": 0.2121,
+      "step": 472
+    },
+    {
+      "epoch": 2.901840490797546,
+      "grad_norm": 1.643700122833252,
+      "learning_rate": 4.034931333323173e-06,
+      "loss": 0.2777,
+      "step": 473
+    },
+    {
+      "epoch": 2.9079754601226995,
+      "grad_norm": 1.3550535440444946,
+      "learning_rate": 4.031125194693484e-06,
+      "loss": 0.2281,
+      "step": 474
+    },
+    {
+      "epoch": 2.914110429447853,
+      "grad_norm": 1.610695481300354,
+      "learning_rate": 4.0273133683844375e-06,
+      "loss": 0.3369,
+      "step": 475
+    },
+    {
+      "epoch": 2.920245398773006,
+      "grad_norm": 2.248654842376709,
+      "learning_rate": 4.023495868555848e-06,
+      "loss": 0.2674,
+      "step": 476
+    },
+    {
+      "epoch": 2.9263803680981595,
+      "grad_norm": 1.8112519979476929,
+      "learning_rate": 4.0196727093886024e-06,
+      "loss": 0.4803,
+      "step": 477
+    },
+    {
+      "epoch": 2.932515337423313,
+      "grad_norm": 1.595954418182373,
+      "learning_rate": 4.015843905084612e-06,
+      "loss": 0.3961,
+      "step": 478
+    },
+    {
+      "epoch": 2.938650306748466,
+      "grad_norm": 1.717560887336731,
+      "learning_rate": 4.012009469866756e-06,
+      "loss": 0.3542,
+      "step": 479
+    },
+    {
+      "epoch": 2.9447852760736195,
+      "grad_norm": 2.2247915267944336,
+      "learning_rate": 4.008169417978836e-06,
+      "loss": 0.498,
+      "step": 480
+    },
+    {
+      "epoch": 2.950920245398773,
+      "grad_norm": 1.5092260837554932,
+      "learning_rate": 4.004323763685511e-06,
+      "loss": 0.4167,
+      "step": 481
+    },
+    {
+      "epoch": 2.957055214723926,
+      "grad_norm": 1.624187707901001,
+      "learning_rate": 4.0004725212722565e-06,
+      "loss": 0.2129,
+      "step": 482
+    },
+    {
+      "epoch": 2.96319018404908,
+      "grad_norm": 1.04583740234375,
+      "learning_rate": 3.996615705045302e-06,
+      "loss": 0.2277,
+      "step": 483
+    },
+    {
+      "epoch": 2.969325153374233,
+      "grad_norm": 1.4794758558273315,
+      "learning_rate": 3.992753329331588e-06,
+      "loss": 0.2771,
+      "step": 484
+    },
+    {
+      "epoch": 2.9754601226993866,
+      "grad_norm": 1.7486788034439087,
+      "learning_rate": 3.9888854084786995e-06,
+      "loss": 0.5283,
+      "step": 485
+    },
+    {
+      "epoch": 2.98159509202454,
+      "grad_norm": 2.020996570587158,
+      "learning_rate": 3.985011956854826e-06,
+      "loss": 0.6443,
+      "step": 486
+    },
+    {
+      "epoch": 2.9877300613496933,
+      "grad_norm": 1.2538446187973022,
+      "learning_rate": 3.9811329888487004e-06,
+      "loss": 0.247,
+      "step": 487
+    },
+    {
+      "epoch": 2.9938650306748467,
+      "grad_norm": 1.605323314666748,
+      "learning_rate": 3.977248518869545e-06,
+      "loss": 0.3421,
+      "step": 488
+    },
+    {
+      "epoch": 3.0,
+      "grad_norm": 2.032585620880127,
+      "learning_rate": 3.973358561347024e-06,
+      "loss": 0.684,
+      "step": 489
+    }
+  ],
+  "logging_steps": 1,
+  "max_steps": 1630,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 10,
+  "save_steps": 208,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 1.2495424094155571e+17,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-652/config.json

@@ -0,0 +1,58 @@
+{
+  "architectures": [
+    "Qwen2ForCausalLM"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 151643,
+  "eos_token_id": 151645,
+  "hidden_act": "silu",
+  "hidden_size": 3584,
+  "initializer_range": 0.02,
+  "intermediate_size": 18944,
+  "layer_types": [
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention"
+  ],
+  "max_position_embeddings": 32768,
+  "max_window_layers": 28,
+  "model_type": "qwen2",
+  "num_attention_heads": 28,
+  "num_hidden_layers": 28,
+  "num_key_value_heads": 4,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": null,
+  "rope_theta": 1000000.0,
+  "sliding_window": null,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float32",
+  "transformers_version": "4.55.0",
+  "use_cache": false,
+  "use_sliding_window": false,
+  "vocab_size": 152064
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-815/added_tokens.json

@@ -0,0 +1,24 @@
+{
+  "</tool_call>": 151658,
+  "<tool_call>": 151657,
+  "<|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
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-815/chat_template.jinja

@@ -0,0 +1,54 @@
+{%- if tools %}
+    {{- '<|im_start|>system\n' }}
+    {%- if messages[0]['role'] == 'system' %}
+        {{- messages[0]['content'] }}
+    {%- else %}
+        {{- 'You are Qwen, created by Alibaba Cloud. You are a helpful assistant.' }}
+    {%- endif %}
+    {{- "\n\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>" }}
+    {%- for tool in tools %}
+        {{- "\n" }}
+        {{- tool | tojson }}
+    {%- endfor %}
+    {{- "\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" }}
+{%- else %}
+    {%- if messages[0]['role'] == 'system' %}
+        {{- '<|im_start|>system\n' + messages[0]['content'] + '<|im_end|>\n' }}
+    {%- else %}
+        {{- '<|im_start|>system\nYou are Qwen, created by Alibaba Cloud. You are a helpful assistant.<|im_end|>\n' }}
+    {%- endif %}
+{%- endif %}
+{%- for message in messages %}
+    {%- if (message.role == "user") or (message.role == "system" and not loop.first) or (message.role == "assistant" and not message.tool_calls) %}
+        {{- '<|im_start|>' + message.role + '\n' + message.content + '<|im_end|>' + '\n' }}
+    {%- elif message.role == "assistant" %}
+        {{- '<|im_start|>' + message.role }}
+        {%- if message.content %}
+            {{- '\n' + message.content }}
+        {%- endif %}
+        {%- for tool_call in message.tool_calls %}
+            {%- if tool_call.function is defined %}
+                {%- set tool_call = tool_call.function %}
+            {%- endif %}
+            {{- '\n<tool_call>\n{"name": "' }}
+            {{- tool_call.name }}
+            {{- '", "arguments": ' }}
+            {{- tool_call.arguments | tojson }}
+            {{- '}\n</tool_call>' }}
+        {%- endfor %}
+        {{- '<|im_end|>\n' }}
+    {%- elif message.role == "tool" %}
+        {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != "tool") %}
+            {{- '<|im_start|>user' }}
+        {%- endif %}
+        {{- '\n<tool_response>\n' }}
+        {{- message.content }}
+        {{- '\n</tool_response>' }}
+        {%- if loop.last or (messages[loop.index0 + 1].role != "tool") %}
+            {{- '<|im_end|>\n' }}
+        {%- endif %}
+    {%- endif %}
+{%- endfor %}
+{%- if add_generation_prompt %}
+    {{- '<|im_start|>assistant\n' }}
+{%- endif %}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-815/config.json

@@ -0,0 +1,58 @@
+{
+  "architectures": [
+    "Qwen2ForCausalLM"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 151643,
+  "eos_token_id": 151645,
+  "hidden_act": "silu",
+  "hidden_size": 3584,
+  "initializer_range": 0.02,
+  "intermediate_size": 18944,
+  "layer_types": [
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention",
+    "full_attention"
+  ],
+  "max_position_embeddings": 32768,
+  "max_window_layers": 28,
+  "model_type": "qwen2",
+  "num_attention_heads": 28,
+  "num_hidden_layers": 28,
+  "num_key_value_heads": 4,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": null,
+  "rope_theta": 1000000.0,
+  "sliding_window": null,
+  "tie_word_embeddings": false,
+  "torch_dtype": "float32",
+  "transformers_version": "4.55.0",
+  "use_cache": false,
+  "use_sliding_window": false,
+  "vocab_size": 152064
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-815/generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 151643,
+  "eos_token_id": 151645,
+  "transformers_version": "4.55.0",
+  "use_cache": false
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-815/model.safetensors.index.json

@@ -0,0 +1,347 @@
+{
+  "metadata": {
+    "total_parameters": 1903904128,
+    "total_size": 30462466048
+  },
+  "weight_map": {
+    "lm_head.weight": "model-00007-of-00007.safetensors",
+    "model.embed_tokens.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.input_layernorm.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.k_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.q_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.v_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.input_layernorm.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.k_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.q_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.v_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.10.input_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.10.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.input_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.11.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.input_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.12.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.13.input_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.13.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.13.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.13.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.13.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.13.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.13.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.13.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.13.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.13.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.13.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.13.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.14.input_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.14.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.input_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.15.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.input_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.16.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.input_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.mlp.down_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.post_attention_layernorm.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.17.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.18.input_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.18.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.18.mlp.gate_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.18.mlp.up_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.18.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.18.self_attn.k_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.18.self_attn.k_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.18.self_attn.o_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.18.self_attn.q_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.18.self_attn.q_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.18.self_attn.v_proj.bias": "model-00004-of-00007.safetensors",
+    "model.layers.18.self_attn.v_proj.weight": "model-00004-of-00007.safetensors",
+    "model.layers.19.input_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.19.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.2.input_layernorm.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.mlp.down_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.mlp.up_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.k_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.q_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.v_proj.bias": "model-00001-of-00007.safetensors",
+    "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00007.safetensors",
+    "model.layers.20.input_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.20.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.input_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.21.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.input_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.22.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.input_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.mlp.down_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.mlp.gate_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.mlp.up_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.post_attention_layernorm.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.o_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.23.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.24.input_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.24.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.24.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.24.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.24.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.24.self_attn.k_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.24.self_attn.k_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.24.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.24.self_attn.q_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.24.self_attn.q_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.24.self_attn.v_proj.bias": "model-00005-of-00007.safetensors",
+    "model.layers.24.self_attn.v_proj.weight": "model-00005-of-00007.safetensors",
+    "model.layers.25.input_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.k_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.q_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.v_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.25.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.input_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.k_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.q_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.v_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.26.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.input_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.mlp.down_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.mlp.gate_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.mlp.up_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.post_attention_layernorm.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.k_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.k_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.o_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.q_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.q_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.v_proj.bias": "model-00006-of-00007.safetensors",
+    "model.layers.27.self_attn.v_proj.weight": "model-00006-of-00007.safetensors",
+    "model.layers.3.input_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.3.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.input_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.4.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.input_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.5.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.input_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.6.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.input_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.mlp.down_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.mlp.gate_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.mlp.up_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.post_attention_layernorm.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.7.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.8.input_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.8.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.8.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.8.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.8.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.8.self_attn.k_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.8.self_attn.k_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.8.self_attn.o_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.8.self_attn.q_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.8.self_attn.q_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.8.self_attn.v_proj.bias": "model-00002-of-00007.safetensors",
+    "model.layers.8.self_attn.v_proj.weight": "model-00002-of-00007.safetensors",
+    "model.layers.9.input_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.mlp.down_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.mlp.gate_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.mlp.up_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.post_attention_layernorm.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.k_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.k_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.o_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.q_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.q_proj.weight": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.v_proj.bias": "model-00003-of-00007.safetensors",
+    "model.layers.9.self_attn.v_proj.weight": "model-00003-of-00007.safetensors",
+    "model.norm.weight": "model-00006-of-00007.safetensors"
+  }
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-815/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
+  }
+}

qwen2_5_7b_instruct/limo_filtered_correct/checkpoint-815/tokenizer_config.json

@@ -0,0 +1,208 @@
+{
+  "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
+    }
+  },
+  "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,
+  "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
+}