grail_v2op_fcur_sr_b64_sft_e3_ck640

.gitattributes

@@ -80,3 +80,4 @@ grail_v1_trajq_rtx_b64_sft_e3_ck300/tokenizer.json filter=lfs diff=lfs merge=lfs
 grail_v2op_fcur_b64_sft_e3_ck640/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 grail_v5_fcur_op48k_b64_sft_e3_ck800/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 grail_v2op_fcur_long_b64_sft_e3_ck1280/tokenizer.json filter=lfs diff=lfs merge=lfs -text
+grail_v2op_fcur_sr_b64_sft_e3_ck640/tokenizer.json filter=lfs diff=lfs merge=lfs -text

grail_v2op_fcur_sr_b64_sft_e3_ck640/all_results.json

@@ -0,0 +1,12 @@
+{
+    "epoch": 1.007877116975187,
+    "eval_loss": 0.034945208579301834,
+    "eval_runtime": 154.7177,
+    "eval_samples_per_second": 13.819,
+    "eval_steps_per_second": 3.458,
+    "total_flos": 281838925119488.0,
+    "train_loss": 0.0994124161079526,
+    "train_runtime": 7659.8586,
+    "train_samples_per_second": 5.347,
+    "train_steps_per_second": 0.084
+}

grail_v2op_fcur_sr_b64_sft_e3_ck640/chat_template.jinja

@@ -0,0 +1,120 @@
+{%- if tools %}
+    {{- '<|im_start|>system\n' }}
+    {%- if messages[0].role == 'system' %}
+        {%- if messages[0].content is string %}
+            {{- messages[0].content }}
+        {%- else %}
+            {%- for content in messages[0].content %}
+                {%- if 'text' in content %}
+                    {{- content.text }}
+                {%- endif %}
+            {%- endfor %}
+        {%- endif %}
+        {{- '\n\n' }}
+    {%- endif %}
+    {{- "# 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' }}
+        {%- if messages[0].content is string %}
+            {{- messages[0].content }}
+        {%- else %}
+            {%- for content in messages[0].content %}
+                {%- if 'text' in content %}
+                    {{- content.text }}
+                {%- endif %}
+            {%- endfor %}
+        {%- endif %}
+        {{- '<|im_end|>\n' }}
+    {%- endif %}
+{%- endif %}
+{%- set image_count = namespace(value=0) %}
+{%- set video_count = namespace(value=0) %}
+{%- for message in messages %}
+    {%- if message.role == "user" %}
+        {{- '<|im_start|>' + message.role + '\n' }}
+        {%- if message.content is string %}
+            {{- message.content }}
+        {%- else %}
+            {%- for content in message.content %}
+                {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}
+                    {%- set image_count.value = image_count.value + 1 %}
+                    {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}
+                    <|vision_start|><|image_pad|><|vision_end|>
+                {%- elif content.type == 'video' or 'video' in content %}
+                    {%- set video_count.value = video_count.value + 1 %}
+                    {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}
+                    <|vision_start|><|video_pad|><|vision_end|>
+                {%- elif 'text' in content %}
+                    {{- content.text }}
+                {%- endif %}
+            {%- endfor %}
+        {%- endif %}
+        {{- '<|im_end|>\n' }}
+    {%- elif message.role == "assistant" %}
+        {{- '<|im_start|>' + message.role + '\n' }}
+        {%- if message.content is string %}
+            {{- message.content }}
+        {%- else %}
+            {%- for content_item in message.content %}
+                {%- if 'text' in content_item %}
+                    {{- content_item.text }}
+                {%- endif %}
+            {%- endfor %}
+        {%- endif %}
+        {%- if message.tool_calls %}
+            {%- for tool_call in message.tool_calls %}
+                {%- if (loop.first and message.content) or (not loop.first) %}
+                    {{- '\n' }}
+                {%- endif %}
+                {%- if tool_call.function %}
+                    {%- set tool_call = tool_call.function %}
+                {%- endif %}
+                {{- '<tool_call>\n{"name": "' }}
+                {{- tool_call.name }}
+                {{- '", "arguments": ' }}
+                {%- if tool_call.arguments is string %}
+                    {{- tool_call.arguments }}
+                {%- else %}
+                    {{- tool_call.arguments | tojson }}
+                {%- endif %}
+                {{- '}\n</tool_call>' }}
+            {%- endfor %}
+        {%- endif %}
+        {{- '<|im_end|>\n' }}
+    {%- elif message.role == "tool" %}
+        {%- if loop.first or (messages[loop.index0 - 1].role != "tool") %}
+            {{- '<|im_start|>user' }}
+        {%- endif %}
+        {{- '\n<tool_response>\n' }}
+        {%- if message.content is string %}
+            {{- message.content }}
+        {%- else %}
+            {%- for content in message.content %}
+                {%- if content.type == 'image' or 'image' in content or 'image_url' in content %}
+                    {%- set image_count.value = image_count.value + 1 %}
+                    {%- if add_vision_id %}Picture {{ image_count.value }}: {% endif -%}
+                    <|vision_start|><|image_pad|><|vision_end|>
+                {%- elif content.type == 'video' or 'video' in content %}
+                    {%- set video_count.value = video_count.value + 1 %}
+                    {%- if add_vision_id %}Video {{ video_count.value }}: {% endif -%}
+                    <|vision_start|><|video_pad|><|vision_end|>
+                {%- elif 'text' in content %}
+                    {{- content.text }}
+                {%- endif %}
+            {%- endfor %}
+        {%- endif %}
+        {{- '\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 %}

grail_v2op_fcur_sr_b64_sft_e3_ck640/config.json

@@ -0,0 +1,70 @@
+{
+  "architectures": [
+    "Qwen3VLForConditionalGeneration"
+  ],
+  "bos_token_id": null,
+  "dtype": "bfloat16",
+  "eos_token_id": 151645,
+  "hidden_size": 4096,
+  "image_token_id": 151655,
+  "model_type": "qwen3_vl",
+  "pad_token_id": 151643,
+  "text_config": {
+    "attention_bias": false,
+    "attention_dropout": 0.0,
+    "bos_token_id": 151643,
+    "dtype": "bfloat16",
+    "eos_token_id": 151645,
+    "head_dim": 128,
+    "hidden_act": "silu",
+    "hidden_size": 4096,
+    "initializer_range": 0.02,
+    "intermediate_size": 12288,
+    "max_position_embeddings": 262144,
+    "model_type": "qwen3_vl_text",
+    "num_attention_heads": 32,
+    "num_hidden_layers": 36,
+    "num_key_value_heads": 8,
+    "pad_token_id": null,
+    "rms_norm_eps": 1e-06,
+    "rope_parameters": {
+      "mrope_interleaved": true,
+      "mrope_section": [
+        24,
+        20,
+        20
+      ],
+      "rope_theta": 5000000,
+      "rope_type": "default"
+    },
+    "use_cache": false,
+    "vocab_size": 151936
+  },
+  "tie_word_embeddings": false,
+  "transformers_version": "5.8.0",
+  "use_cache": false,
+  "video_token_id": 151656,
+  "vision_config": {
+    "deepstack_visual_indexes": [
+      8,
+      16,
+      24
+    ],
+    "depth": 27,
+    "dtype": "bfloat16",
+    "hidden_act": "gelu_pytorch_tanh",
+    "hidden_size": 1152,
+    "in_channels": 3,
+    "initializer_range": 0.02,
+    "intermediate_size": 4304,
+    "model_type": "qwen3_vl_vision",
+    "num_heads": 16,
+    "num_position_embeddings": 2304,
+    "out_hidden_size": 4096,
+    "patch_size": 16,
+    "spatial_merge_size": 2,
+    "temporal_patch_size": 2
+  },
+  "vision_end_token_id": 151653,
+  "vision_start_token_id": 151652
+}

grail_v2op_fcur_sr_b64_sft_e3_ck640/eval_results.json

@@ -0,0 +1,7 @@
+{
+    "epoch": 1.007877116975187,
+    "eval_loss": 0.034945208579301834,
+    "eval_runtime": 154.7177,
+    "eval_samples_per_second": 13.819,
+    "eval_steps_per_second": 3.458
+}

grail_v2op_fcur_sr_b64_sft_e3_ck640/generation_config.json

@@ -0,0 +1,14 @@
+{
+  "do_sample": true,
+  "eos_token_id": [
+    151645,
+    151645,
+    151643
+  ],
+  "pad_token_id": 151643,
+  "repetition_penalty": 1.0,
+  "temperature": 0.7,
+  "top_k": 20,
+  "top_p": 0.8,
+  "transformers_version": "5.8.0"
+}

grail_v2op_fcur_sr_b64_sft_e3_ck640/model.safetensors

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

grail_v2op_fcur_sr_b64_sft_e3_ck640/processor_config.json

@@ -0,0 +1,60 @@
+{
+  "image_processor": {
+    "do_convert_rgb": true,
+    "do_normalize": true,
+    "do_rescale": true,
+    "do_resize": true,
+    "image_mean": [
+      0.5,
+      0.5,
+      0.5
+    ],
+    "image_processor_type": "Qwen2VLImageProcessor",
+    "image_std": [
+      0.5,
+      0.5,
+      0.5
+    ],
+    "merge_size": 2,
+    "patch_size": 16,
+    "resample": 3,
+    "rescale_factor": 0.00392156862745098,
+    "size": {
+      "longest_edge": 16777216,
+      "shortest_edge": 65536
+    },
+    "temporal_patch_size": 2
+  },
+  "processor_class": "Qwen3VLProcessor",
+  "video_processor": {
+    "do_convert_rgb": true,
+    "do_normalize": true,
+    "do_rescale": true,
+    "do_resize": true,
+    "do_sample_frames": true,
+    "fps": 2,
+    "image_mean": [
+      0.5,
+      0.5,
+      0.5
+    ],
+    "image_std": [
+      0.5,
+      0.5,
+      0.5
+    ],
+    "max_frames": 768,
+    "merge_size": 2,
+    "min_frames": 4,
+    "patch_size": 16,
+    "resample": 3,
+    "rescale_factor": 0.00392156862745098,
+    "return_metadata": false,
+    "size": {
+      "longest_edge": 25165824,
+      "shortest_edge": 4096
+    },
+    "temporal_patch_size": 2,
+    "video_processor_type": "Qwen3VLVideoProcessor"
+  }
+}

grail_v2op_fcur_sr_b64_sft_e3_ck640/sft.yaml

@@ -0,0 +1,59 @@
+# LLaMA-Factory full-parameter SFT of Qwen3-VL-8B-Instruct on one HER-final dataset.
+# Four L40S (46 GB), one run at a time: DeepSpeed ZeRO-3 with optimizer offload to host
+# RAM (~130 GB per run, so two runs do not fit in 372 GB); the vision tower and projector
+# stay frozen (LLaMA-Factory's default for VLMs), the LLM trains.
+# Filled per run by launch_sft.sh: grail_v2op_fcur_sr_sft_e3, exps/sft/grail_v2op_fcur_sr_sft_e3_k25515, /scratch/kwc5354/viewagent/viewagent_private/exps/sft_runs/grail_v2op_fcur_sr_b64_sft_e3.
+### model
+model_name_or_path: /scratch/kwc5354/viewagent/viewagent_private/exps/models/graph_nemi_sft_e2
+template: qwen3_vl_nothink
+trust_remote_code: true
+image_max_pixels: 65536          # frames are 256x256 already; never upscale
+image_min_pixels: 4096
+
+### method
+stage: sft
+do_train: true
+finetuning_type: full
+freeze_vision_tower: true
+freeze_multi_modal_projector: true
+freeze_language_model: false
+deepspeed: /scratch/kwc5354/viewagent/viewagent_private/LLaMA-Factory/examples/deepspeed/ds_z3_config.json
+
+### dataset
+dataset: grail_v2op_fcur_sr_sft_e3
+dataset_dir: exps/sft/grail_v2op_fcur_sr_sft_e3_k25515
+media_dir: /scratch/kwc5354/viewagent/viewagent_private
+cutoff_len: 8192
+overwrite_cache: true
+preprocessing_num_workers: 16
+dataloader_num_workers: 4
+
+### output
+output_dir: /scratch/kwc5354/viewagent/viewagent_private/exps/sft_runs/grail_v2op_fcur_sr_b64_sft_e3
+logging_steps: 10
+save_strategy: steps
+save_steps: 320
+save_total_limit: 100
+save_only_model: true
+plot_loss: true
+overwrite_output_dir: true
+report_to: wandb
+
+### train
+per_device_train_batch_size: 1
+gradient_accumulation_steps: 16
+max_steps: 640
+learning_rate: 1.0e-5
+lr_scheduler_type: cosine
+warmup_ratio: 0.03
+weight_decay: 0.01
+bf16: true
+gradient_checkpointing: true
+ddp_timeout: 180000000
+seed: 0
+val_size: 0.05
+eval_strategy: steps
+eval_steps: 320
+per_device_eval_batch_size: 1
+mask_reasoning_span: true
+run_name: grail_v2op_fcur_sr_b64_sft_e3

grail_v2op_fcur_sr_b64_sft_e3_ck640/tokenizer.json

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

grail_v2op_fcur_sr_b64_sft_e3_ck640/tokenizer_config.json

@@ -0,0 +1,32 @@
+{
+  "add_prefix_space": false,
+  "backend": "tokenizers",
+  "bos_token": null,
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|im_end|>",
+  "errors": "replace",
+  "extra_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|>"
+  ],
+  "is_local": true,
+  "local_files_only": false,
+  "model_max_length": 262144,
+  "pad_token": "<|endoftext|>",
+  "padding_side": "right",
+  "processor_class": "Qwen3VLProcessor",
+  "split_special_tokens": false,
+  "tokenizer_class": "Qwen2Tokenizer",
+  "unk_token": null
+}

grail_v2op_fcur_sr_b64_sft_e3_ck640/train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 1.007877116975187,
+    "total_flos": 281838925119488.0,
+    "train_loss": 0.0994124161079526,
+    "train_runtime": 7659.8586,
+    "train_samples_per_second": 5.347,
+    "train_steps_per_second": 0.084
+}

grail_v2op_fcur_sr_b64_sft_e3_ck640/trainer_state.json

@@ -0,0 +1,507 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 1.007877116975187,
+  "eval_steps": 320,
+  "global_step": 640,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.015754233950374164,
+      "grad_norm": 6.630710419241484,
+      "learning_rate": 4.5e-06,
+      "loss": 1.1664539337158204,
+      "step": 10
+    },
+    {
+      "epoch": 0.03150846790074833,
+      "grad_norm": 1.3460756643807419,
+      "learning_rate": 9.5e-06,
+      "loss": 0.3788366079330444,
+      "step": 20
+    },
+    {
+      "epoch": 0.04726270185112249,
+      "grad_norm": 0.7951640996022633,
+      "learning_rate": 9.994801642711945e-06,
+      "loss": 0.2180910587310791,
+      "step": 30
+    },
+    {
+      "epoch": 0.06301693580149666,
+      "grad_norm": 0.7713198722707091,
+      "learning_rate": 9.976845889479778e-06,
+      "loss": 0.1960616111755371,
+      "step": 40
+    },
+    {
+      "epoch": 0.07877116975187082,
+      "grad_norm": 0.6059415755466836,
+      "learning_rate": 9.946114644227765e-06,
+      "loss": 0.17394330501556396,
+      "step": 50
+    },
+    {
+      "epoch": 0.09452540370224498,
+      "grad_norm": 0.5989499182762709,
+      "learning_rate": 9.90268679362172e-06,
+      "loss": 0.1669042468070984,
+      "step": 60
+    },
+    {
+      "epoch": 0.11027963765261914,
+      "grad_norm": 0.8092490910478523,
+      "learning_rate": 9.846673816331297e-06,
+      "loss": 0.1597401738166809,
+      "step": 70
+    },
+    {
+      "epoch": 0.1260338716029933,
+      "grad_norm": 0.5322210975894792,
+      "learning_rate": 9.778219496865864e-06,
+      "loss": 0.14796795845031738,
+      "step": 80
+    },
+    {
+      "epoch": 0.14178810555336746,
+      "grad_norm": 0.7201538787618418,
+      "learning_rate": 9.6974995564817e-06,
+      "loss": 0.1454394817352295,
+      "step": 90
+    },
+    {
+      "epoch": 0.15754233950374164,
+      "grad_norm": 0.5947952547655874,
+      "learning_rate": 9.604721202108084e-06,
+      "loss": 0.13459253311157227,
+      "step": 100
+    },
+    {
+      "epoch": 0.17329657345411578,
+      "grad_norm": 0.6295764614836088,
+      "learning_rate": 9.500122594450056e-06,
+      "loss": 0.13184187412261963,
+      "step": 110
+    },
+    {
+      "epoch": 0.18905080740448996,
+      "grad_norm": 0.6220010769800145,
+      "learning_rate": 9.383972236633308e-06,
+      "loss": 0.12013475894927979,
+      "step": 120
+    },
+    {
+      "epoch": 0.20480504135486413,
+      "grad_norm": 0.5816895886550543,
+      "learning_rate": 9.256568284960513e-06,
+      "loss": 0.11741031408309936,
+      "step": 130
+    },
+    {
+      "epoch": 0.22055927530523828,
+      "grad_norm": 0.7971152324421312,
+      "learning_rate": 9.118237783548376e-06,
+      "loss": 0.11438512802124023,
+      "step": 140
+    },
+    {
+      "epoch": 0.23631350925561245,
+      "grad_norm": 0.6429191736824276,
+      "learning_rate": 8.969335824810075e-06,
+      "loss": 0.10405704975128174,
+      "step": 150
+    },
+    {
+      "epoch": 0.2520677432059866,
+      "grad_norm": 0.744676044644892,
+      "learning_rate": 8.810244637938117e-06,
+      "loss": 0.10688095092773438,
+      "step": 160
+    },
+    {
+      "epoch": 0.26782197715636075,
+      "grad_norm": 0.7611860854334,
+      "learning_rate": 8.641372607727503e-06,
+      "loss": 0.10522496700286865,
+      "step": 170
+    },
+    {
+      "epoch": 0.2835762111067349,
+      "grad_norm": 0.6015052851357979,
+      "learning_rate": 8.463153226257805e-06,
+      "loss": 0.09823083281517028,
+      "step": 180
+    },
+    {
+      "epoch": 0.2993304450571091,
+      "grad_norm": 0.697377301904915,
+      "learning_rate": 8.276043980125213e-06,
+      "loss": 0.0936966896057129,
+      "step": 190
+    },
+    {
+      "epoch": 0.31508467900748327,
+      "grad_norm": 0.8463864443848728,
+      "learning_rate": 8.080525176081006e-06,
+      "loss": 0.09509625434875488,
+      "step": 200
+    },
+    {
+      "epoch": 0.33083891295785745,
+      "grad_norm": 0.7531425155224758,
+      "learning_rate": 7.877098708091017e-06,
+      "loss": 0.09124931097030639,
+      "step": 210
+    },
+    {
+      "epoch": 0.34659314690823156,
+      "grad_norm": 0.7286969427717154,
+      "learning_rate": 7.666286768981012e-06,
+      "loss": 0.08264161944389344,
+      "step": 220
+    },
+    {
+      "epoch": 0.36234738085860574,
+      "grad_norm": 0.7406454120838915,
+      "learning_rate": 7.448630509975199e-06,
+      "loss": 0.08563262820243836,
+      "step": 230
+    },
+    {
+      "epoch": 0.3781016148089799,
+      "grad_norm": 0.8350820971178975,
+      "learning_rate": 7.224688651568797e-06,
+      "loss": 0.08674737811088562,
+      "step": 240
+    },
+    {
+      "epoch": 0.3938558487593541,
+      "grad_norm": 0.7073103891625422,
+      "learning_rate": 6.995036049300539e-06,
+      "loss": 0.08587347269058228,
+      "step": 250
+    },
+    {
+      "epoch": 0.40961008270972826,
+      "grad_norm": 0.7991561537465599,
+      "learning_rate": 6.7602622181067425e-06,
+      "loss": 0.07690019011497498,
+      "step": 260
+    },
+    {
+      "epoch": 0.4253643166601024,
+      "grad_norm": 0.6691617369942527,
+      "learning_rate": 6.520969819044933e-06,
+      "loss": 0.07653147578239441,
+      "step": 270
+    },
+    {
+      "epoch": 0.44111855061047656,
+      "grad_norm": 0.6729927000086433,
+      "learning_rate": 6.277773112271545e-06,
+      "loss": 0.06907791495323182,
+      "step": 280
+    },
+    {
+      "epoch": 0.45687278456085073,
+      "grad_norm": 0.802972645929226,
+      "learning_rate": 6.031296380244915e-06,
+      "loss": 0.07677772641181946,
+      "step": 290
+    },
+    {
+      "epoch": 0.4726270185112249,
+      "grad_norm": 0.727978091232625,
+      "learning_rate": 5.782172325201155e-06,
+      "loss": 0.06885179877281189,
+      "step": 300
+    },
+    {
+      "epoch": 0.4883812524615991,
+      "grad_norm": 0.6821748662778541,
+      "learning_rate": 5.531040445016604e-06,
+      "loss": 0.0691904902458191,
+      "step": 310
+    },
+    {
+      "epoch": 0.5041354864119733,
+      "grad_norm": 0.6817424786022891,
+      "learning_rate": 5.278545391625931e-06,
+      "loss": 0.06935135126113892,
+      "step": 320
+    },
+    {
+      "epoch": 0.5041354864119733,
+      "eval_loss": 0.0637066513299942,
+      "eval_runtime": 155.2286,
+      "eval_samples_per_second": 13.773,
+      "eval_steps_per_second": 3.447,
+      "step": 320
+    },
+    {
+      "epoch": 0.5198897203623474,
+      "grad_norm": 0.7826529450720553,
+      "learning_rate": 5.025335316209883e-06,
+      "loss": 0.062109851837158205,
+      "step": 330
+    },
+    {
+      "epoch": 0.5356439543127215,
+      "grad_norm": 0.7599652234172967,
+      "learning_rate": 4.7720602054004865e-06,
+      "loss": 0.06312604546546936,
+      "step": 340
+    },
+    {
+      "epoch": 0.5513981882630957,
+      "grad_norm": 0.7077668033901435,
+      "learning_rate": 4.51937021277469e-06,
+      "loss": 0.05922434329986572,
+      "step": 350
+    },
+    {
+      "epoch": 0.5671524222134698,
+      "grad_norm": 0.6313171156396902,
+      "learning_rate": 4.267913989919457e-06,
+      "loss": 0.0586701512336731,
+      "step": 360
+    },
+    {
+      "epoch": 0.5829066561638441,
+      "grad_norm": 0.8984275220186211,
+      "learning_rate": 4.018337021352447e-06,
+      "loss": 0.05281572937965393,
+      "step": 370
+    },
+    {
+      "epoch": 0.5986608901142182,
+      "grad_norm": 0.7043760499100633,
+      "learning_rate": 3.7712799675725105e-06,
+      "loss": 0.057857978343963626,
+      "step": 380
+    },
+    {
+      "epoch": 0.6144151240645923,
+      "grad_norm": 0.6644628535647129,
+      "learning_rate": 3.5273770204933814e-06,
+      "loss": 0.055921196937561035,
+      "step": 390
+    },
+    {
+      "epoch": 0.6301693580149665,
+      "grad_norm": 0.7468237057784437,
+      "learning_rate": 3.2872542754821334e-06,
+      "loss": 0.053208005428314206,
+      "step": 400
+    },
+    {
+      "epoch": 0.6459235919653407,
+      "grad_norm": 0.6403112901590272,
+      "learning_rate": 3.051528124181342e-06,
+      "loss": 0.0492661327123642,
+      "step": 410
+    },
+    {
+      "epoch": 0.6616778259157149,
+      "grad_norm": 0.7250515503198126,
+      "learning_rate": 2.8208036722405895e-06,
+      "loss": 0.04837639927864075,
+      "step": 420
+    },
+    {
+      "epoch": 0.677432059866089,
+      "grad_norm": 0.7407330985264863,
+      "learning_rate": 2.595673186018929e-06,
+      "loss": 0.051023554801940915,
+      "step": 430
+    },
+    {
+      "epoch": 0.6931862938164631,
+      "grad_norm": 0.8474659674545724,
+      "learning_rate": 2.376714572245651e-06,
+      "loss": 0.047341907024383546,
+      "step": 440
+    },
+    {
+      "epoch": 0.7089405277668374,
+      "grad_norm": 0.665762930843984,
+      "learning_rate": 2.1644898945419814e-06,
+      "loss": 0.046429941058158876,
+      "step": 450
+    },
+    {
+      "epoch": 0.7246947617172115,
+      "grad_norm": 0.7171367768909648,
+      "learning_rate": 1.959543930611835e-06,
+      "loss": 0.045801252126693726,
+      "step": 460
+    },
+    {
+      "epoch": 0.7404489956675857,
+      "grad_norm": 0.7641716788751164,
+      "learning_rate": 1.7624027738052669e-06,
+      "loss": 0.048598605394363406,
+      "step": 470
+    },
+    {
+      "epoch": 0.7562032296179598,
+      "grad_norm": 1.0531059069415996,
+      "learning_rate": 1.5735724826443788e-06,
+      "loss": 0.04257847666740418,
+      "step": 480
+    },
+    {
+      "epoch": 0.771957463568334,
+      "grad_norm": 1.0055061980534017,
+      "learning_rate": 1.3935377817783585e-06,
+      "loss": 0.04214349389076233,
+      "step": 490
+    },
+    {
+      "epoch": 0.7877116975187082,
+      "grad_norm": 0.7957055061410945,
+      "learning_rate": 1.2227608177022522e-06,
+      "loss": 0.04379300773143768,
+      "step": 500
+    },
+    {
+      "epoch": 0.8034659314690823,
+      "grad_norm": 0.773698868748556,
+      "learning_rate": 1.0616799724335387e-06,
+      "loss": 0.04371005296707153,
+      "step": 510
+    },
+    {
+      "epoch": 0.8192201654194565,
+      "grad_norm": 0.764622043081019,
+      "learning_rate": 9.107087381917878e-07,
+      "loss": 0.037135356664657594,
+      "step": 520
+    },
+    {
+      "epoch": 0.8349743993698306,
+      "grad_norm": 0.7587792734685739,
+      "learning_rate": 7.702346559700502e-07,
+      "loss": 0.03855884075164795,
+      "step": 530
+    },
+    {
+      "epoch": 0.8507286333202048,
+      "grad_norm": 0.7903359573546342,
+      "learning_rate": 6.406183207226973e-07,
+      "loss": 0.03782544732093811,
+      "step": 540
+    },
+    {
+      "epoch": 0.866482867270579,
+      "grad_norm": 0.831670667536907,
+      "learning_rate": 5.22192455723336e-07,
+      "loss": 0.037650269269943235,
+      "step": 550
+    },
+    {
+      "epoch": 0.8822371012209531,
+      "grad_norm": 0.7139728620722768,
+      "learning_rate": 4.152610584689443e-07,
+      "loss": 0.038959911465644835,
+      "step": 560
+    },
+    {
+      "epoch": 0.8979913351713273,
+      "grad_norm": 0.8364214935219988,
+      "learning_rate": 3.200986203226536e-07,
+      "loss": 0.03320388793945313,
+      "step": 570
+    },
+    {
+      "epoch": 0.9137455691217015,
+      "grad_norm": 0.8024842256204965,
+      "learning_rate": 2.3694942189835423e-07,
+      "loss": 0.040011551976203916,
+      "step": 580
+    },
+    {
+      "epoch": 0.9294998030720756,
+      "grad_norm": 0.863094126501592,
+      "learning_rate": 1.6602690599586747e-07,
+      "loss": 0.03752725124359131,
+      "step": 590
+    },
+    {
+      "epoch": 0.9452540370224498,
+      "grad_norm": 0.7672903898976177,
+      "learning_rate": 1.0751312969633665e-07,
+      "loss": 0.034224200248718264,
+      "step": 600
+    },
+    {
+      "epoch": 0.9610082709728239,
+      "grad_norm": 0.8235865521685797,
+      "learning_rate": 6.15582970243117e-08,
+      "loss": 0.037143072485923766,
+      "step": 610
+    },
+    {
+      "epoch": 0.9767625049231982,
+      "grad_norm": 0.813399272968933,
+      "learning_rate": 2.828037337617273e-08,
+      "loss": 0.03858485221862793,
+      "step": 620
+    },
+    {
+      "epoch": 0.9925167388735723,
+      "grad_norm": 0.95869301557528,
+      "learning_rate": 7.764782704649532e-09,
+      "loss": 0.03263449668884277,
+      "step": 630
+    },
+    {
+      "epoch": 1.007877116975187,
+      "grad_norm": 0.49499409916829495,
+      "learning_rate": 6.418823675857777e-11,
+      "loss": 0.03312427997589111,
+      "step": 640
+    },
+    {
+      "epoch": 1.007877116975187,
+      "eval_loss": 0.034945208579301834,
+      "eval_runtime": 154.7642,
+      "eval_samples_per_second": 13.815,
+      "eval_steps_per_second": 3.457,
+      "step": 640
+    },
+    {
+      "epoch": 1.007877116975187,
+      "step": 640,
+      "total_flos": 281838925119488.0,
+      "train_loss": 0.0994124161079526,
+      "train_runtime": 7659.8586,
+      "train_samples_per_second": 5.347,
+      "train_steps_per_second": 0.084
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 640,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2,
+  "save_steps": 320,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 281838925119488.0,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}