Upload folder using huggingface_hub

README.md

@@ -1,199 +1,64 @@
 ---
-library_name: transformers
-tags: []
+library_name: peft
+license: other
+base_model: openthaigpt/openthaigpt1.5-7b-instruct
+tags:
+- llama-factory
+- lora
+- generated_from_trainer
+model-index:
+- name: kto_law_7b
+  results: []
 ---
 
-# Model Card for Model ID
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
 
-<!-- Provide a quick summary of what the model is/does. -->
+# kto_law_7b
 
+This model is a fine-tuned version of [openthaigpt/openthaigpt1.5-7b-instruct](https://huggingface.co/openthaigpt/openthaigpt1.5-7b-instruct) on the law_th_kto dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.5288
+- Rewards/chosen: 2.6834
+- Logps/chosen: -142.3163
+- Kl: 28.1784
 
+## Model description
 
-## Model Details
+More information needed
 
-### Model Description
+## Intended uses & limitations
 
-<!-- Provide a longer summary of what this model is. -->
+More information needed
 
-This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.
+## Training and evaluation data
 
-- **Developed by:** [More Information Needed]
-- **Funded by [optional]:** [More Information Needed]
-- **Shared by [optional]:** [More Information Needed]
-- **Model type:** [More Information Needed]
-- **Language(s) (NLP):** [More Information Needed]
-- **License:** [More Information Needed]
-- **Finetuned from model [optional]:** [More Information Needed]
+More information needed
 
-### Model Sources [optional]
+## Training procedure
 
-<!-- Provide the basic links for the model. -->
+### Training hyperparameters
 
-- **Repository:** [More Information Needed]
-- **Paper [optional]:** [More Information Needed]
-- **Demo [optional]:** [More Information Needed]
+The following hyperparameters were used during training:
+- learning_rate: 5e-06
+- train_batch_size: 1
+- eval_batch_size: 1
+- seed: 42
+- gradient_accumulation_steps: 8
+- total_train_batch_size: 8
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: cosine
+- lr_scheduler_warmup_ratio: 0.1
+- num_epochs: 3.0
 
-## Uses
+### Training results
 
-<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 
-### Direct Use
 
-<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
+### Framework versions
 
-[More Information Needed]
-
-### Downstream Use [optional]
-
-<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
-
-[More Information Needed]
-
-### Out-of-Scope Use
-
-<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
-
-[More Information Needed]
-
-## Bias, Risks, and Limitations
-
-<!-- This section is meant to convey both technical and sociotechnical limitations. -->
-
-[More Information Needed]
-
-### Recommendations
-
-<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
-
-Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
-
-## How to Get Started with the Model
-
-Use the code below to get started with the model.
-
-[More Information Needed]
-
-## Training Details
-
-### Training Data
-
-<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
-
-[More Information Needed]
-
-### Training Procedure
-
-<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
-
-#### Preprocessing [optional]
-
-[More Information Needed]
-
-
-#### Training Hyperparameters
-
-- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
-
-#### Speeds, Sizes, Times [optional]
-
-<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
-
-[More Information Needed]
-
-## Evaluation
-
-<!-- This section describes the evaluation protocols and provides the results. -->
-
-### Testing Data, Factors & Metrics
-
-#### Testing Data
-
-<!-- This should link to a Dataset Card if possible. -->
-
-[More Information Needed]
-
-#### Factors
-
-<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
-
-[More Information Needed]
-
-#### Metrics
-
-<!-- These are the evaluation metrics being used, ideally with a description of why. -->
-
-[More Information Needed]
-
-### Results
-
-[More Information Needed]
-
-#### Summary
-
-
-
-## Model Examination [optional]
-
-<!-- Relevant interpretability work for the model goes here -->
-
-[More Information Needed]
-
-## Environmental Impact
-
-<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
-
-Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
-
-- **Hardware Type:** [More Information Needed]
-- **Hours used:** [More Information Needed]
-- **Cloud Provider:** [More Information Needed]
-- **Compute Region:** [More Information Needed]
-- **Carbon Emitted:** [More Information Needed]
-
-## Technical Specifications [optional]
-
-### Model Architecture and Objective
-
-[More Information Needed]
-
-### Compute Infrastructure
-
-[More Information Needed]
-
-#### Hardware
-
-[More Information Needed]
-
-#### Software
-
-[More Information Needed]
-
-## Citation [optional]
-
-<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
-
-**BibTeX:**
-
-[More Information Needed]
-
-**APA:**
-
-[More Information Needed]
-
-## Glossary [optional]
-
-<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
-
-[More Information Needed]
-
-## More Information [optional]
-
-[More Information Needed]
-
-## Model Card Authors [optional]
-
-[More Information Needed]
-
-## Model Card Contact
-
-[More Information Needed]
+- PEFT 0.12.0
+- Transformers 4.44.2
+- Pytorch 2.4.1+cu121
+- Datasets 2.21.0
+- Tokenizers 0.19.1

adapter_config.json

@@ -20,13 +20,13 @@
   "rank_pattern": {},
   "revision": null,
   "target_modules": [
-    "o_proj",
+    "down_proj",
     "up_proj",
-    "v_proj",
     "k_proj",
+    "o_proj",
     "q_proj",
     "gate_proj",
-    "down_proj"
+    "v_proj"
   ],
   "task_type": "CAUSAL_LM",
   "use_dora": false,

added_tokens.json

@@ -0,0 +1,25 @@
+{
+  "</tool_call>": 151658,
+  "<tool_call>": 151657,
+  "<|box_end|>": 151649,
+  "<|box_start|>": 151648,
+  "<|endoftext|>": 151643,
+  "<|eot_id|>": 151665,
+  "<|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
+}

all_results.json

@@ -0,0 +1,15 @@
+{
+    "epoch": 2.986666666666667,
+    "eval_kl": 28.178438186645508,
+    "eval_logps/chosen": -142.316337890625,
+    "eval_loss": 0.5287721157073975,
+    "eval_rewards/chosen": 2.6833560180664064,
+    "eval_runtime": 25.1552,
+    "eval_samples_per_second": 3.975,
+    "eval_steps_per_second": 3.975,
+    "total_flos": 3.5905450080116736e+16,
+    "train_loss": 0.5264021051781518,
+    "train_runtime": 1249.11,
+    "train_samples_per_second": 2.162,
+    "train_steps_per_second": 0.269
+}

checkpoint-336/README.md

@@ -0,0 +1,202 @@
+---
+base_model: openthaigpt/openthaigpt1.5-7b-instruct
+library_name: peft
+---
+
+# Model Card for Model ID
+
+<!-- Provide a quick summary of what the model is/does. -->
+
+
+
+## Model Details
+
+### Model Description
+
+<!-- Provide a longer summary of what this model is. -->
+
+
+
+- **Developed by:** [More Information Needed]
+- **Funded by [optional]:** [More Information Needed]
+- **Shared by [optional]:** [More Information Needed]
+- **Model type:** [More Information Needed]
+- **Language(s) (NLP):** [More Information Needed]
+- **License:** [More Information Needed]
+- **Finetuned from model [optional]:** [More Information Needed]
+
+### Model Sources [optional]
+
+<!-- Provide the basic links for the model. -->
+
+- **Repository:** [More Information Needed]
+- **Paper [optional]:** [More Information Needed]
+- **Demo [optional]:** [More Information Needed]
+
+## Uses
+
+<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
+
+### Direct Use
+
+<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
+
+[More Information Needed]
+
+### Downstream Use [optional]
+
+<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
+
+[More Information Needed]
+
+### Out-of-Scope Use
+
+<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
+
+[More Information Needed]
+
+## Bias, Risks, and Limitations
+
+<!-- This section is meant to convey both technical and sociotechnical limitations. -->
+
+[More Information Needed]
+
+### Recommendations
+
+<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
+
+Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
+
+## How to Get Started with the Model
+
+Use the code below to get started with the model.
+
+[More Information Needed]
+
+## Training Details
+
+### Training Data
+
+<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
+
+[More Information Needed]
+
+### Training Procedure
+
+<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
+
+#### Preprocessing [optional]
+
+[More Information Needed]
+
+
+#### Training Hyperparameters
+
+- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
+
+#### Speeds, Sizes, Times [optional]
+
+<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
+
+[More Information Needed]
+
+## Evaluation
+
+<!-- This section describes the evaluation protocols and provides the results. -->
+
+### Testing Data, Factors & Metrics
+
+#### Testing Data
+
+<!-- This should link to a Dataset Card if possible. -->
+
+[More Information Needed]
+
+#### Factors
+
+<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
+
+[More Information Needed]
+
+#### Metrics
+
+<!-- These are the evaluation metrics being used, ideally with a description of why. -->
+
+[More Information Needed]
+
+### Results
+
+[More Information Needed]
+
+#### Summary
+
+
+
+## Model Examination [optional]
+
+<!-- Relevant interpretability work for the model goes here -->
+
+[More Information Needed]
+
+## Environmental Impact
+
+<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
+
+Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
+
+- **Hardware Type:** [More Information Needed]
+- **Hours used:** [More Information Needed]
+- **Cloud Provider:** [More Information Needed]
+- **Compute Region:** [More Information Needed]
+- **Carbon Emitted:** [More Information Needed]
+
+## Technical Specifications [optional]
+
+### Model Architecture and Objective
+
+[More Information Needed]
+
+### Compute Infrastructure
+
+[More Information Needed]
+
+#### Hardware
+
+[More Information Needed]
+
+#### Software
+
+[More Information Needed]
+
+## Citation [optional]
+
+<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
+
+**BibTeX:**
+
+[More Information Needed]
+
+**APA:**
+
+[More Information Needed]
+
+## Glossary [optional]
+
+<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
+
+[More Information Needed]
+
+## More Information [optional]
+
+[More Information Needed]
+
+## Model Card Authors [optional]
+
+[More Information Needed]
+
+## Model Card Contact
+
+[More Information Needed]
+### Framework versions
+
+- PEFT 0.12.0

checkpoint-336/adapter_config.json

@@ -0,0 +1,34 @@
+{
+  "alpha_pattern": {},
+  "auto_mapping": null,
+  "base_model_name_or_path": "openthaigpt/openthaigpt1.5-7b-instruct",
+  "bias": "none",
+  "fan_in_fan_out": false,
+  "inference_mode": true,
+  "init_lora_weights": true,
+  "layer_replication": null,
+  "layers_pattern": null,
+  "layers_to_transform": null,
+  "loftq_config": {},
+  "lora_alpha": 16,
+  "lora_dropout": 0.0,
+  "megatron_config": null,
+  "megatron_core": "megatron.core",
+  "modules_to_save": null,
+  "peft_type": "LORA",
+  "r": 8,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": [
+    "down_proj",
+    "up_proj",
+    "k_proj",
+    "o_proj",
+    "q_proj",
+    "gate_proj",
+    "v_proj"
+  ],
+  "task_type": "CAUSAL_LM",
+  "use_dora": false,
+  "use_rslora": false
+}

checkpoint-336/adapter_model.safetensors

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

checkpoint-336/added_tokens.json

@@ -0,0 +1,25 @@
+{
+  "</tool_call>": 151658,
+  "<tool_call>": 151657,
+  "<|box_end|>": 151649,
+  "<|box_start|>": 151648,
+  "<|endoftext|>": 151643,
+  "<|eot_id|>": 151665,
+  "<|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
+}

checkpoint-336/optimizer.pt

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

checkpoint-336/rng_state.pth

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

checkpoint-336/scheduler.pt

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

checkpoint-336/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": "<|eot_id|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

checkpoint-336/tokenizer_config.json

@@ -0,0 +1,216 @@
+{
+  "add_bos_token": false,
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "151643": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151644": {
+      "content": "<|im_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151645": {
+      "content": "<|im_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151646": {
+      "content": "<|object_ref_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151647": {
+      "content": "<|object_ref_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151648": {
+      "content": "<|box_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151649": {
+      "content": "<|box_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151650": {
+      "content": "<|quad_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151651": {
+      "content": "<|quad_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151652": {
+      "content": "<|vision_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151653": {
+      "content": "<|vision_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151654": {
+      "content": "<|vision_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151655": {
+      "content": "<|image_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151656": {
+      "content": "<|video_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151657": {
+      "content": "<tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151658": {
+      "content": "</tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151659": {
+      "content": "<|fim_prefix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151660": {
+      "content": "<|fim_middle|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151661": {
+      "content": "<|fim_suffix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151662": {
+      "content": "<|fim_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151663": {
+      "content": "<|repo_name|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151664": {
+      "content": "<|file_sep|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151665": {
+      "content": "<|eot_id|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<|im_start|>",
+    "<|im_end|>",
+    "<|object_ref_start|>",
+    "<|object_ref_end|>",
+    "<|box_start|>",
+    "<|box_end|>",
+    "<|quad_start|>",
+    "<|quad_end|>",
+    "<|vision_start|>",
+    "<|vision_end|>",
+    "<|vision_pad|>",
+    "<|image_pad|>",
+    "<|video_pad|>"
+  ],
+  "bos_token": null,
+  "chat_template": "{% if messages[0]['role'] == 'system' %}{% set loop_messages = messages[1:] %}{% set system_message = messages[0]['content'] %}{% else %}{% set loop_messages = messages %}{% endif %}{% if system_message is defined %}{{ '<|start_header_id|>system<|end_header_id|>\n\n' + system_message + '<|eot_id|>' }}{% endif %}{% for message in loop_messages %}{% set content = message['content'] %}{% if message['role'] == 'user' %}{{ '<|start_header_id|>user<|end_header_id|>\n\n' + content + '<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n' }}{% elif message['role'] == 'assistant' %}{{ content + '<|eot_id|>' }}{% endif %}{% endfor %}",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|eot_id|>",
+  "errors": "replace",
+  "model_max_length": 131072,
+  "pad_token": "<|endoftext|>",
+  "padding_side": "right",
+  "split_special_tokens": false,
+  "tokenizer_class": "Qwen2Tokenizer",
+  "unk_token": null
+}

checkpoint-336/trainer_state.json

@@ -0,0 +1,363 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 2.986666666666667,
+  "eval_steps": 500,
+  "global_step": 336,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.08888888888888889,
+      "grad_norm": 1.792281150817871,
+      "kl": 0.15646514296531677,
+      "learning_rate": 1.4705882352941177e-06,
+      "logps/chosen": -164.1169189453125,
+      "loss": 0.5043,
+      "rewards/chosen": -0.0013721466064453125,
+      "step": 10
+    },
+    {
+      "epoch": 0.17777777777777778,
+      "grad_norm": 1.7645841836929321,
+      "kl": 0.23952817916870117,
+      "learning_rate": 2.9411764705882355e-06,
+      "logps/chosen": -161.21431884765624,
+      "loss": 0.506,
+      "rewards/chosen": -0.000222191633656621,
+      "step": 20
+    },
+    {
+      "epoch": 0.26666666666666666,
+      "grad_norm": 1.6355303525924683,
+      "kl": 0.2789602279663086,
+      "learning_rate": 4.411764705882353e-06,
+      "logps/chosen": -161.8250244140625,
+      "loss": 0.5046,
+      "rewards/chosen": 0.009497338533401489,
+      "step": 30
+    },
+    {
+      "epoch": 0.35555555555555557,
+      "grad_norm": 1.8716518878936768,
+      "kl": 0.5364601016044617,
+      "learning_rate": 4.995131923687488e-06,
+      "logps/chosen": -160.8272705078125,
+      "loss": 0.5011,
+      "rewards/chosen": 0.04936442375183105,
+      "step": 40
+    },
+    {
+      "epoch": 0.4444444444444444,
+      "grad_norm": 1.7810463905334473,
+      "kl": 1.5138229131698608,
+      "learning_rate": 4.965451197130373e-06,
+      "logps/chosen": -159.2620361328125,
+      "loss": 0.5081,
+      "rewards/chosen": 0.11891223192214966,
+      "step": 50
+    },
+    {
+      "epoch": 0.5333333333333333,
+      "grad_norm": 2.2072041034698486,
+      "kl": 3.1553397178649902,
+      "learning_rate": 4.90911473983908e-06,
+      "logps/chosen": -163.2381103515625,
+      "loss": 0.5049,
+      "rewards/chosen": 0.295930004119873,
+      "step": 60
+    },
+    {
+      "epoch": 0.6222222222222222,
+      "grad_norm": 1.858336329460144,
+      "kl": 6.15082311630249,
+      "learning_rate": 4.826731644963705e-06,
+      "logps/chosen": -160.2790771484375,
+      "loss": 0.5238,
+      "rewards/chosen": 0.517311429977417,
+      "step": 70
+    },
+    {
+      "epoch": 0.7111111111111111,
+      "grad_norm": 2.1484944820404053,
+      "kl": 9.094244003295898,
+      "learning_rate": 4.71919261421297e-06,
+      "logps/chosen": -165.65592041015626,
+      "loss": 0.4949,
+      "rewards/chosen": 0.9306423187255859,
+      "step": 80
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 1.9223041534423828,
+      "kl": 12.533430099487305,
+      "learning_rate": 4.587660327850203e-06,
+      "logps/chosen": -147.84144287109376,
+      "loss": 0.5327,
+      "rewards/chosen": 1.1160342216491699,
+      "step": 90
+    },
+    {
+      "epoch": 0.8888888888888888,
+      "grad_norm": 2.052889823913574,
+      "kl": 15.455533981323242,
+      "learning_rate": 4.43355687413747e-06,
+      "logps/chosen": -141.32998046875,
+      "loss": 0.5258,
+      "rewards/chosen": 1.4277078628540039,
+      "step": 100
+    },
+    {
+      "epoch": 0.9777777777777777,
+      "grad_norm": 1.8008378744125366,
+      "kl": 18.05435562133789,
+      "learning_rate": 4.258548374136976e-06,
+      "logps/chosen": -142.55394287109374,
+      "loss": 0.5491,
+      "rewards/chosen": 1.5885238647460938,
+      "step": 110
+    },
+    {
+      "epoch": 1.0666666666666667,
+      "grad_norm": 1.9117740392684937,
+      "kl": 20.216676712036133,
+      "learning_rate": 4.064526968101844e-06,
+      "logps/chosen": -138.30635986328124,
+      "loss": 0.5418,
+      "rewards/chosen": 1.8262062072753906,
+      "step": 120
+    },
+    {
+      "epoch": 1.1555555555555554,
+      "grad_norm": 2.0452499389648438,
+      "kl": 22.620920181274414,
+      "learning_rate": 3.853590358214119e-06,
+      "logps/chosen": -142.5436767578125,
+      "loss": 0.5414,
+      "rewards/chosen": 2.02716178894043,
+      "step": 130
+    },
+    {
+      "epoch": 1.2444444444444445,
+      "grad_norm": 1.5861475467681885,
+      "kl": 21.474929809570312,
+      "learning_rate": 3.6280191288478437e-06,
+      "logps/chosen": -139.66346435546876,
+      "loss": 0.5436,
+      "rewards/chosen": 1.9418670654296875,
+      "step": 140
+    },
+    {
+      "epoch": 1.3333333333333333,
+      "grad_norm": 2.040761947631836,
+      "kl": 24.884174346923828,
+      "learning_rate": 3.3902520895638674e-06,
+      "logps/chosen": -144.4580078125,
+      "loss": 0.5551,
+      "rewards/chosen": 2.2067359924316405,
+      "step": 150
+    },
+    {
+      "epoch": 1.4222222222222223,
+      "grad_norm": 1.4137938022613525,
+      "kl": 24.920988082885742,
+      "learning_rate": 3.142859907420615e-06,
+      "logps/chosen": -134.41856689453124,
+      "loss": 0.5452,
+      "rewards/chosen": 2.2827281951904297,
+      "step": 160
+    },
+    {
+      "epoch": 1.511111111111111,
+      "grad_norm": 1.9247097969055176,
+      "kl": 27.40714454650879,
+      "learning_rate": 2.8885173136805126e-06,
+      "logps/chosen": -134.56304931640625,
+      "loss": 0.5554,
+      "rewards/chosen": 2.478318786621094,
+      "step": 170
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 1.7178384065628052,
+      "kl": 26.36468505859375,
+      "learning_rate": 2.629974185404951e-06,
+      "logps/chosen": -140.7825927734375,
+      "loss": 0.5482,
+      "rewards/chosen": 2.3276464462280275,
+      "step": 180
+    },
+    {
+      "epoch": 1.6888888888888889,
+      "grad_norm": 1.899584174156189,
+      "kl": 27.777027130126953,
+      "learning_rate": 2.3700258145950495e-06,
+      "logps/chosen": -140.53466796875,
+      "loss": 0.5618,
+      "rewards/chosen": 2.424651336669922,
+      "step": 190
+    },
+    {
+      "epoch": 1.7777777777777777,
+      "grad_norm": 1.7362196445465088,
+      "kl": 27.96476173400879,
+      "learning_rate": 2.1114826863194882e-06,
+      "logps/chosen": -132.751318359375,
+      "loss": 0.5549,
+      "rewards/chosen": 2.5185680389404297,
+      "step": 200
+    },
+    {
+      "epoch": 1.8666666666666667,
+      "grad_norm": 1.5568628311157227,
+      "kl": 25.686458587646484,
+      "learning_rate": 1.8571400925793855e-06,
+      "logps/chosen": -141.460986328125,
+      "loss": 0.4799,
+      "rewards/chosen": 2.678207778930664,
+      "step": 210
+    },
+    {
+      "epoch": 1.9555555555555557,
+      "grad_norm": 1.4741252660751343,
+      "kl": 27.350994110107422,
+      "learning_rate": 1.6097479104361328e-06,
+      "logps/chosen": -138.36417236328126,
+      "loss": 0.5164,
+      "rewards/chosen": 2.5984352111816404,
+      "step": 220
+    },
+    {
+      "epoch": 2.0444444444444443,
+      "grad_norm": 1.4206268787384033,
+      "kl": 27.860248565673828,
+      "learning_rate": 1.3719808711521573e-06,
+      "logps/chosen": -140.159765625,
+      "loss": 0.5315,
+      "rewards/chosen": 2.6533029556274412,
+      "step": 230
+    },
+    {
+      "epoch": 2.1333333333333333,
+      "grad_norm": 1.964991569519043,
+      "kl": 28.185623168945312,
+      "learning_rate": 1.1464096417858821e-06,
+      "logps/chosen": -136.175439453125,
+      "loss": 0.5194,
+      "rewards/chosen": 2.697865676879883,
+      "step": 240
+    },
+    {
+      "epoch": 2.2222222222222223,
+      "grad_norm": 2.0303757190704346,
+      "kl": 29.453914642333984,
+      "learning_rate": 9.354730318981561e-07,
+      "logps/chosen": -124.939111328125,
+      "loss": 0.5565,
+      "rewards/chosen": 2.6518789291381837,
+      "step": 250
+    },
+    {
+      "epoch": 2.311111111111111,
+      "grad_norm": 1.9753375053405762,
+      "kl": 28.224218368530273,
+      "learning_rate": 7.414516258630245e-07,
+      "logps/chosen": -143.79691162109376,
+      "loss": 0.4645,
+      "rewards/chosen": 2.974603271484375,
+      "step": 260
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 1.5573641061782837,
+      "kl": 28.22760581970215,
+      "learning_rate": 5.664431258625305e-07,
+      "logps/chosen": -135.832470703125,
+      "loss": 0.5039,
+      "rewards/chosen": 2.787263107299805,
+      "step": 270
+    },
+    {
+      "epoch": 2.488888888888889,
+      "grad_norm": 1.8748925924301147,
+      "kl": 29.364704132080078,
+      "learning_rate": 4.123396721497977e-07,
+      "logps/chosen": -132.555029296875,
+      "loss": 0.5659,
+      "rewards/chosen": 2.5855674743652344,
+      "step": 280
+    },
+    {
+      "epoch": 2.5777777777777775,
+      "grad_norm": 1.9802337884902954,
+      "kl": 29.822818756103516,
+      "learning_rate": 2.8080738578703054e-07,
+      "logps/chosen": -132.623876953125,
+      "loss": 0.5254,
+      "rewards/chosen": 2.7944366455078127,
+      "step": 290
+    },
+    {
+      "epoch": 2.6666666666666665,
+      "grad_norm": 1.6388640403747559,
+      "kl": 29.55389404296875,
+      "learning_rate": 1.7326835503629542e-07,
+      "logps/chosen": -132.94588623046874,
+      "loss": 0.5661,
+      "rewards/chosen": 2.6015438079833983,
+      "step": 300
+    },
+    {
+      "epoch": 2.7555555555555555,
+      "grad_norm": 2.059163808822632,
+      "kl": 28.346817016601562,
+      "learning_rate": 9.088526016092142e-08,
+      "logps/chosen": -142.41676025390626,
+      "loss": 0.4982,
+      "rewards/chosen": 2.8422063827514648,
+      "step": 310
+    },
+    {
+      "epoch": 2.8444444444444446,
+      "grad_norm": 1.8374332189559937,
+      "kl": 28.879934310913086,
+      "learning_rate": 3.4548802869627806e-08,
+      "logps/chosen": -133.9089599609375,
+      "loss": 0.508,
+      "rewards/chosen": 2.8151947021484376,
+      "step": 320
+    },
+    {
+      "epoch": 2.9333333333333336,
+      "grad_norm": 1.6760042905807495,
+      "kl": 28.577600479125977,
+      "learning_rate": 4.868076312512515e-09,
+      "logps/chosen": -133.6552001953125,
+      "loss": 0.5269,
+      "rewards/chosen": 2.694635200500488,
+      "step": 330
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 336,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 3,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 3.5905450080116736e+16,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-336/training_args.bin

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

eval_results.json

@@ -0,0 +1,10 @@
+{
+    "epoch": 2.986666666666667,
+    "eval_kl": 28.178438186645508,
+    "eval_logps/chosen": -142.316337890625,
+    "eval_loss": 0.5287721157073975,
+    "eval_rewards/chosen": 2.6833560180664064,
+    "eval_runtime": 25.1552,
+    "eval_samples_per_second": 3.975,
+    "eval_steps_per_second": 3.975
+}

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": "<|eot_id|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer_config.json

@@ -0,0 +1,216 @@
+{
+  "add_bos_token": false,
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "151643": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151644": {
+      "content": "<|im_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151645": {
+      "content": "<|im_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151646": {
+      "content": "<|object_ref_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151647": {
+      "content": "<|object_ref_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151648": {
+      "content": "<|box_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151649": {
+      "content": "<|box_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151650": {
+      "content": "<|quad_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151651": {
+      "content": "<|quad_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151652": {
+      "content": "<|vision_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151653": {
+      "content": "<|vision_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151654": {
+      "content": "<|vision_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151655": {
+      "content": "<|image_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151656": {
+      "content": "<|video_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151657": {
+      "content": "<tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151658": {
+      "content": "</tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151659": {
+      "content": "<|fim_prefix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151660": {
+      "content": "<|fim_middle|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151661": {
+      "content": "<|fim_suffix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151662": {
+      "content": "<|fim_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151663": {
+      "content": "<|repo_name|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151664": {
+      "content": "<|file_sep|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151665": {
+      "content": "<|eot_id|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<|im_start|>",
+    "<|im_end|>",
+    "<|object_ref_start|>",
+    "<|object_ref_end|>",
+    "<|box_start|>",
+    "<|box_end|>",
+    "<|quad_start|>",
+    "<|quad_end|>",
+    "<|vision_start|>",
+    "<|vision_end|>",
+    "<|vision_pad|>",
+    "<|image_pad|>",
+    "<|video_pad|>"
+  ],
+  "bos_token": null,
+  "chat_template": "{% if messages[0]['role'] == 'system' %}{% set loop_messages = messages[1:] %}{% set system_message = messages[0]['content'] %}{% else %}{% set loop_messages = messages %}{% endif %}{% if system_message is defined %}{{ '<|start_header_id|>system<|end_header_id|>\n\n' + system_message + '<|eot_id|>' }}{% endif %}{% for message in loop_messages %}{% set content = message['content'] %}{% if message['role'] == 'user' %}{{ '<|start_header_id|>user<|end_header_id|>\n\n' + content + '<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n' }}{% elif message['role'] == 'assistant' %}{{ content + '<|eot_id|>' }}{% endif %}{% endfor %}",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|eot_id|>",
+  "errors": "replace",
+  "model_max_length": 131072,
+  "pad_token": "<|endoftext|>",
+  "padding_side": "right",
+  "split_special_tokens": false,
+  "tokenizer_class": "Qwen2Tokenizer",
+  "unk_token": null
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 2.986666666666667,
+    "total_flos": 3.5905450080116736e+16,
+    "train_loss": 0.5264021051781518,
+    "train_runtime": 1249.11,
+    "train_samples_per_second": 2.162,
+    "train_steps_per_second": 0.269
+}

trainer_log.jsonl

@@ -0,0 +1,34 @@
+{"current_steps": 10, "total_steps": 336, "loss": 0.5043, "learning_rate": 1.4705882352941177e-06, "epoch": 0.08888888888888889, "percentage": 2.98, "elapsed_time": "0:00:38", "remaining_time": "0:21:04"}
+{"current_steps": 20, "total_steps": 336, "loss": 0.506, "learning_rate": 2.9411764705882355e-06, "epoch": 0.17777777777777778, "percentage": 5.95, "elapsed_time": "0:01:16", "remaining_time": "0:20:03"}
+{"current_steps": 30, "total_steps": 336, "loss": 0.5046, "learning_rate": 4.411764705882353e-06, "epoch": 0.26666666666666666, "percentage": 8.93, "elapsed_time": "0:01:53", "remaining_time": "0:19:15"}
+{"current_steps": 40, "total_steps": 336, "loss": 0.5011, "learning_rate": 4.995131923687488e-06, "epoch": 0.35555555555555557, "percentage": 11.9, "elapsed_time": "0:02:30", "remaining_time": "0:18:31"}
+{"current_steps": 50, "total_steps": 336, "loss": 0.5081, "learning_rate": 4.965451197130373e-06, "epoch": 0.4444444444444444, "percentage": 14.88, "elapsed_time": "0:03:07", "remaining_time": "0:17:50"}
+{"current_steps": 60, "total_steps": 336, "loss": 0.5049, "learning_rate": 4.90911473983908e-06, "epoch": 0.5333333333333333, "percentage": 17.86, "elapsed_time": "0:03:44", "remaining_time": "0:17:11"}
+{"current_steps": 70, "total_steps": 336, "loss": 0.5238, "learning_rate": 4.826731644963705e-06, "epoch": 0.6222222222222222, "percentage": 20.83, "elapsed_time": "0:04:21", "remaining_time": "0:16:33"}
+{"current_steps": 80, "total_steps": 336, "loss": 0.4949, "learning_rate": 4.71919261421297e-06, "epoch": 0.7111111111111111, "percentage": 23.81, "elapsed_time": "0:04:58", "remaining_time": "0:15:55"}
+{"current_steps": 90, "total_steps": 336, "loss": 0.5327, "learning_rate": 4.587660327850203e-06, "epoch": 0.8, "percentage": 26.79, "elapsed_time": "0:05:35", "remaining_time": "0:15:17"}
+{"current_steps": 100, "total_steps": 336, "loss": 0.5258, "learning_rate": 4.43355687413747e-06, "epoch": 0.8888888888888888, "percentage": 29.76, "elapsed_time": "0:06:12", "remaining_time": "0:14:39"}
+{"current_steps": 110, "total_steps": 336, "loss": 0.5491, "learning_rate": 4.258548374136976e-06, "epoch": 0.9777777777777777, "percentage": 32.74, "elapsed_time": "0:06:49", "remaining_time": "0:14:01"}
+{"current_steps": 120, "total_steps": 336, "loss": 0.5418, "learning_rate": 4.064526968101844e-06, "epoch": 1.0666666666666667, "percentage": 35.71, "elapsed_time": "0:07:27", "remaining_time": "0:13:24"}
+{"current_steps": 130, "total_steps": 336, "loss": 0.5414, "learning_rate": 3.853590358214119e-06, "epoch": 1.1555555555555554, "percentage": 38.69, "elapsed_time": "0:08:04", "remaining_time": "0:12:48"}
+{"current_steps": 140, "total_steps": 336, "loss": 0.5436, "learning_rate": 3.6280191288478437e-06, "epoch": 1.2444444444444445, "percentage": 41.67, "elapsed_time": "0:08:42", "remaining_time": "0:12:11"}
+{"current_steps": 150, "total_steps": 336, "loss": 0.5551, "learning_rate": 3.3902520895638674e-06, "epoch": 1.3333333333333333, "percentage": 44.64, "elapsed_time": "0:09:19", "remaining_time": "0:11:33"}
+{"current_steps": 160, "total_steps": 336, "loss": 0.5452, "learning_rate": 3.142859907420615e-06, "epoch": 1.4222222222222223, "percentage": 47.62, "elapsed_time": "0:09:56", "remaining_time": "0:10:55"}
+{"current_steps": 170, "total_steps": 336, "loss": 0.5554, "learning_rate": 2.8885173136805126e-06, "epoch": 1.511111111111111, "percentage": 50.6, "elapsed_time": "0:10:33", "remaining_time": "0:10:18"}
+{"current_steps": 180, "total_steps": 336, "loss": 0.5482, "learning_rate": 2.629974185404951e-06, "epoch": 1.6, "percentage": 53.57, "elapsed_time": "0:11:10", "remaining_time": "0:09:40"}
+{"current_steps": 190, "total_steps": 336, "loss": 0.5618, "learning_rate": 2.3700258145950495e-06, "epoch": 1.6888888888888889, "percentage": 56.55, "elapsed_time": "0:11:47", "remaining_time": "0:09:03"}
+{"current_steps": 200, "total_steps": 336, "loss": 0.5549, "learning_rate": 2.1114826863194882e-06, "epoch": 1.7777777777777777, "percentage": 59.52, "elapsed_time": "0:12:24", "remaining_time": "0:08:26"}
+{"current_steps": 210, "total_steps": 336, "loss": 0.4799, "learning_rate": 1.8571400925793855e-06, "epoch": 1.8666666666666667, "percentage": 62.5, "elapsed_time": "0:13:01", "remaining_time": "0:07:49"}
+{"current_steps": 220, "total_steps": 336, "loss": 0.5164, "learning_rate": 1.6097479104361328e-06, "epoch": 1.9555555555555557, "percentage": 65.48, "elapsed_time": "0:13:38", "remaining_time": "0:07:11"}
+{"current_steps": 230, "total_steps": 336, "loss": 0.5315, "learning_rate": 1.3719808711521573e-06, "epoch": 2.0444444444444443, "percentage": 68.45, "elapsed_time": "0:14:15", "remaining_time": "0:06:34"}
+{"current_steps": 240, "total_steps": 336, "loss": 0.5194, "learning_rate": 1.1464096417858821e-06, "epoch": 2.1333333333333333, "percentage": 71.43, "elapsed_time": "0:14:52", "remaining_time": "0:05:57"}
+{"current_steps": 250, "total_steps": 336, "loss": 0.5565, "learning_rate": 9.354730318981561e-07, "epoch": 2.2222222222222223, "percentage": 74.4, "elapsed_time": "0:15:30", "remaining_time": "0:05:20"}
+{"current_steps": 260, "total_steps": 336, "loss": 0.4645, "learning_rate": 7.414516258630245e-07, "epoch": 2.311111111111111, "percentage": 77.38, "elapsed_time": "0:16:07", "remaining_time": "0:04:42"}
+{"current_steps": 270, "total_steps": 336, "loss": 0.5039, "learning_rate": 5.664431258625305e-07, "epoch": 2.4, "percentage": 80.36, "elapsed_time": "0:16:44", "remaining_time": "0:04:05"}
+{"current_steps": 280, "total_steps": 336, "loss": 0.5659, "learning_rate": 4.123396721497977e-07, "epoch": 2.488888888888889, "percentage": 83.33, "elapsed_time": "0:17:21", "remaining_time": "0:03:28"}
+{"current_steps": 290, "total_steps": 336, "loss": 0.5254, "learning_rate": 2.8080738578703054e-07, "epoch": 2.5777777777777775, "percentage": 86.31, "elapsed_time": "0:17:58", "remaining_time": "0:02:51"}
+{"current_steps": 300, "total_steps": 336, "loss": 0.5661, "learning_rate": 1.7326835503629542e-07, "epoch": 2.6666666666666665, "percentage": 89.29, "elapsed_time": "0:18:35", "remaining_time": "0:02:13"}
+{"current_steps": 310, "total_steps": 336, "loss": 0.4982, "learning_rate": 9.088526016092142e-08, "epoch": 2.7555555555555555, "percentage": 92.26, "elapsed_time": "0:19:12", "remaining_time": "0:01:36"}
+{"current_steps": 320, "total_steps": 336, "loss": 0.508, "learning_rate": 3.4548802869627806e-08, "epoch": 2.8444444444444446, "percentage": 95.24, "elapsed_time": "0:19:49", "remaining_time": "0:00:59"}
+{"current_steps": 330, "total_steps": 336, "loss": 0.5269, "learning_rate": 4.868076312512515e-09, "epoch": 2.9333333333333336, "percentage": 98.21, "elapsed_time": "0:20:26", "remaining_time": "0:00:22"}
+{"current_steps": 336, "total_steps": 336, "epoch": 2.986666666666667, "percentage": 100.0, "elapsed_time": "0:20:49", "remaining_time": "0:00:00"}

trainer_state.json

@@ -0,0 +1,372 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 2.986666666666667,
+  "eval_steps": 500,
+  "global_step": 336,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.08888888888888889,
+      "grad_norm": 1.792281150817871,
+      "kl": 0.15646514296531677,
+      "learning_rate": 1.4705882352941177e-06,
+      "logps/chosen": -164.1169189453125,
+      "loss": 0.5043,
+      "rewards/chosen": -0.0013721466064453125,
+      "step": 10
+    },
+    {
+      "epoch": 0.17777777777777778,
+      "grad_norm": 1.7645841836929321,
+      "kl": 0.23952817916870117,
+      "learning_rate": 2.9411764705882355e-06,
+      "logps/chosen": -161.21431884765624,
+      "loss": 0.506,
+      "rewards/chosen": -0.000222191633656621,
+      "step": 20
+    },
+    {
+      "epoch": 0.26666666666666666,
+      "grad_norm": 1.6355303525924683,
+      "kl": 0.2789602279663086,
+      "learning_rate": 4.411764705882353e-06,
+      "logps/chosen": -161.8250244140625,
+      "loss": 0.5046,
+      "rewards/chosen": 0.009497338533401489,
+      "step": 30
+    },
+    {
+      "epoch": 0.35555555555555557,
+      "grad_norm": 1.8716518878936768,
+      "kl": 0.5364601016044617,
+      "learning_rate": 4.995131923687488e-06,
+      "logps/chosen": -160.8272705078125,
+      "loss": 0.5011,
+      "rewards/chosen": 0.04936442375183105,
+      "step": 40
+    },
+    {
+      "epoch": 0.4444444444444444,
+      "grad_norm": 1.7810463905334473,
+      "kl": 1.5138229131698608,
+      "learning_rate": 4.965451197130373e-06,
+      "logps/chosen": -159.2620361328125,
+      "loss": 0.5081,
+      "rewards/chosen": 0.11891223192214966,
+      "step": 50
+    },
+    {
+      "epoch": 0.5333333333333333,
+      "grad_norm": 2.2072041034698486,
+      "kl": 3.1553397178649902,
+      "learning_rate": 4.90911473983908e-06,
+      "logps/chosen": -163.2381103515625,
+      "loss": 0.5049,
+      "rewards/chosen": 0.295930004119873,
+      "step": 60
+    },
+    {
+      "epoch": 0.6222222222222222,
+      "grad_norm": 1.858336329460144,
+      "kl": 6.15082311630249,
+      "learning_rate": 4.826731644963705e-06,
+      "logps/chosen": -160.2790771484375,
+      "loss": 0.5238,
+      "rewards/chosen": 0.517311429977417,
+      "step": 70
+    },
+    {
+      "epoch": 0.7111111111111111,
+      "grad_norm": 2.1484944820404053,
+      "kl": 9.094244003295898,
+      "learning_rate": 4.71919261421297e-06,
+      "logps/chosen": -165.65592041015626,
+      "loss": 0.4949,
+      "rewards/chosen": 0.9306423187255859,
+      "step": 80
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 1.9223041534423828,
+      "kl": 12.533430099487305,
+      "learning_rate": 4.587660327850203e-06,
+      "logps/chosen": -147.84144287109376,
+      "loss": 0.5327,
+      "rewards/chosen": 1.1160342216491699,
+      "step": 90
+    },
+    {
+      "epoch": 0.8888888888888888,
+      "grad_norm": 2.052889823913574,
+      "kl": 15.455533981323242,
+      "learning_rate": 4.43355687413747e-06,
+      "logps/chosen": -141.32998046875,
+      "loss": 0.5258,
+      "rewards/chosen": 1.4277078628540039,
+      "step": 100
+    },
+    {
+      "epoch": 0.9777777777777777,
+      "grad_norm": 1.8008378744125366,
+      "kl": 18.05435562133789,
+      "learning_rate": 4.258548374136976e-06,
+      "logps/chosen": -142.55394287109374,
+      "loss": 0.5491,
+      "rewards/chosen": 1.5885238647460938,
+      "step": 110
+    },
+    {
+      "epoch": 1.0666666666666667,
+      "grad_norm": 1.9117740392684937,
+      "kl": 20.216676712036133,
+      "learning_rate": 4.064526968101844e-06,
+      "logps/chosen": -138.30635986328124,
+      "loss": 0.5418,
+      "rewards/chosen": 1.8262062072753906,
+      "step": 120
+    },
+    {
+      "epoch": 1.1555555555555554,
+      "grad_norm": 2.0452499389648438,
+      "kl": 22.620920181274414,
+      "learning_rate": 3.853590358214119e-06,
+      "logps/chosen": -142.5436767578125,
+      "loss": 0.5414,
+      "rewards/chosen": 2.02716178894043,
+      "step": 130
+    },
+    {
+      "epoch": 1.2444444444444445,
+      "grad_norm": 1.5861475467681885,
+      "kl": 21.474929809570312,
+      "learning_rate": 3.6280191288478437e-06,
+      "logps/chosen": -139.66346435546876,
+      "loss": 0.5436,
+      "rewards/chosen": 1.9418670654296875,
+      "step": 140
+    },
+    {
+      "epoch": 1.3333333333333333,
+      "grad_norm": 2.040761947631836,
+      "kl": 24.884174346923828,
+      "learning_rate": 3.3902520895638674e-06,
+      "logps/chosen": -144.4580078125,
+      "loss": 0.5551,
+      "rewards/chosen": 2.2067359924316405,
+      "step": 150
+    },
+    {
+      "epoch": 1.4222222222222223,
+      "grad_norm": 1.4137938022613525,
+      "kl": 24.920988082885742,
+      "learning_rate": 3.142859907420615e-06,
+      "logps/chosen": -134.41856689453124,
+      "loss": 0.5452,
+      "rewards/chosen": 2.2827281951904297,
+      "step": 160
+    },
+    {
+      "epoch": 1.511111111111111,
+      "grad_norm": 1.9247097969055176,
+      "kl": 27.40714454650879,
+      "learning_rate": 2.8885173136805126e-06,
+      "logps/chosen": -134.56304931640625,
+      "loss": 0.5554,
+      "rewards/chosen": 2.478318786621094,
+      "step": 170
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 1.7178384065628052,
+      "kl": 26.36468505859375,
+      "learning_rate": 2.629974185404951e-06,
+      "logps/chosen": -140.7825927734375,
+      "loss": 0.5482,
+      "rewards/chosen": 2.3276464462280275,
+      "step": 180
+    },
+    {
+      "epoch": 1.6888888888888889,
+      "grad_norm": 1.899584174156189,
+      "kl": 27.777027130126953,
+      "learning_rate": 2.3700258145950495e-06,
+      "logps/chosen": -140.53466796875,
+      "loss": 0.5618,
+      "rewards/chosen": 2.424651336669922,
+      "step": 190
+    },
+    {
+      "epoch": 1.7777777777777777,
+      "grad_norm": 1.7362196445465088,
+      "kl": 27.96476173400879,
+      "learning_rate": 2.1114826863194882e-06,
+      "logps/chosen": -132.751318359375,
+      "loss": 0.5549,
+      "rewards/chosen": 2.5185680389404297,
+      "step": 200
+    },
+    {
+      "epoch": 1.8666666666666667,
+      "grad_norm": 1.5568628311157227,
+      "kl": 25.686458587646484,
+      "learning_rate": 1.8571400925793855e-06,
+      "logps/chosen": -141.460986328125,
+      "loss": 0.4799,
+      "rewards/chosen": 2.678207778930664,
+      "step": 210
+    },
+    {
+      "epoch": 1.9555555555555557,
+      "grad_norm": 1.4741252660751343,
+      "kl": 27.350994110107422,
+      "learning_rate": 1.6097479104361328e-06,
+      "logps/chosen": -138.36417236328126,
+      "loss": 0.5164,
+      "rewards/chosen": 2.5984352111816404,
+      "step": 220
+    },
+    {
+      "epoch": 2.0444444444444443,
+      "grad_norm": 1.4206268787384033,
+      "kl": 27.860248565673828,
+      "learning_rate": 1.3719808711521573e-06,
+      "logps/chosen": -140.159765625,
+      "loss": 0.5315,
+      "rewards/chosen": 2.6533029556274412,
+      "step": 230
+    },
+    {
+      "epoch": 2.1333333333333333,
+      "grad_norm": 1.964991569519043,
+      "kl": 28.185623168945312,
+      "learning_rate": 1.1464096417858821e-06,
+      "logps/chosen": -136.175439453125,
+      "loss": 0.5194,
+      "rewards/chosen": 2.697865676879883,
+      "step": 240
+    },
+    {
+      "epoch": 2.2222222222222223,
+      "grad_norm": 2.0303757190704346,
+      "kl": 29.453914642333984,
+      "learning_rate": 9.354730318981561e-07,
+      "logps/chosen": -124.939111328125,
+      "loss": 0.5565,
+      "rewards/chosen": 2.6518789291381837,
+      "step": 250
+    },
+    {
+      "epoch": 2.311111111111111,
+      "grad_norm": 1.9753375053405762,
+      "kl": 28.224218368530273,
+      "learning_rate": 7.414516258630245e-07,
+      "logps/chosen": -143.79691162109376,
+      "loss": 0.4645,
+      "rewards/chosen": 2.974603271484375,
+      "step": 260
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 1.5573641061782837,
+      "kl": 28.22760581970215,
+      "learning_rate": 5.664431258625305e-07,
+      "logps/chosen": -135.832470703125,
+      "loss": 0.5039,
+      "rewards/chosen": 2.787263107299805,
+      "step": 270
+    },
+    {
+      "epoch": 2.488888888888889,
+      "grad_norm": 1.8748925924301147,
+      "kl": 29.364704132080078,
+      "learning_rate": 4.123396721497977e-07,
+      "logps/chosen": -132.555029296875,
+      "loss": 0.5659,
+      "rewards/chosen": 2.5855674743652344,
+      "step": 280
+    },
+    {
+      "epoch": 2.5777777777777775,
+      "grad_norm": 1.9802337884902954,
+      "kl": 29.822818756103516,
+      "learning_rate": 2.8080738578703054e-07,
+      "logps/chosen": -132.623876953125,
+      "loss": 0.5254,
+      "rewards/chosen": 2.7944366455078127,
+      "step": 290
+    },
+    {
+      "epoch": 2.6666666666666665,
+      "grad_norm": 1.6388640403747559,
+      "kl": 29.55389404296875,
+      "learning_rate": 1.7326835503629542e-07,
+      "logps/chosen": -132.94588623046874,
+      "loss": 0.5661,
+      "rewards/chosen": 2.6015438079833983,
+      "step": 300
+    },
+    {
+      "epoch": 2.7555555555555555,
+      "grad_norm": 2.059163808822632,
+      "kl": 28.346817016601562,
+      "learning_rate": 9.088526016092142e-08,
+      "logps/chosen": -142.41676025390626,
+      "loss": 0.4982,
+      "rewards/chosen": 2.8422063827514648,
+      "step": 310
+    },
+    {
+      "epoch": 2.8444444444444446,
+      "grad_norm": 1.8374332189559937,
+      "kl": 28.879934310913086,
+      "learning_rate": 3.4548802869627806e-08,
+      "logps/chosen": -133.9089599609375,
+      "loss": 0.508,
+      "rewards/chosen": 2.8151947021484376,
+      "step": 320
+    },
+    {
+      "epoch": 2.9333333333333336,
+      "grad_norm": 1.6760042905807495,
+      "kl": 28.577600479125977,
+      "learning_rate": 4.868076312512515e-09,
+      "logps/chosen": -133.6552001953125,
+      "loss": 0.5269,
+      "rewards/chosen": 2.694635200500488,
+      "step": 330
+    },
+    {
+      "epoch": 2.986666666666667,
+      "step": 336,
+      "total_flos": 3.5905450080116736e+16,
+      "train_loss": 0.5264021051781518,
+      "train_runtime": 1249.11,
+      "train_samples_per_second": 2.162,
+      "train_steps_per_second": 0.269
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 336,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 3,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 3.5905450080116736e+16,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}

training_args.bin

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