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-14b-instruct
+tags:
+- llama-factory
+- lora
+- generated_from_trainer
+model-index:
+- name: kto_law
+  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
 
+This model is a fine-tuned version of [openthaigpt/openthaigpt1.5-14b-instruct](https://huggingface.co/openthaigpt/openthaigpt1.5-14b-instruct) on the law_th_kto dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.5569
+- Rewards/chosen: 2.2727
+- Logps/chosen: -127.1557
+- Kl: 25.3337
 
+## 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",
+    "gate_proj",
     "q_proj",
+    "o_proj",
     "k_proj",
-    "v_proj",
-    "up_proj",
-    "gate_proj"
+    "down_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": 25.333707809448242,
+    "eval_logps/chosen": -127.1557421875,
+    "eval_loss": 0.556947648525238,
+    "eval_rewards/chosen": 2.2726934814453124,
+    "eval_runtime": 45.528,
+    "eval_samples_per_second": 2.196,
+    "eval_steps_per_second": 2.196,
+    "total_flos": 7.102230505945498e+16,
+    "train_loss": 0.5188913842042288,
+    "train_runtime": 2226.9521,
+    "train_samples_per_second": 1.212,
+    "train_steps_per_second": 0.151
+}

checkpoint-336/README.md

@@ -0,0 +1,202 @@
+---
+base_model: openthaigpt/openthaigpt1.5-14b-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-14b-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": [
+    "up_proj",
+    "v_proj",
+    "gate_proj",
+    "q_proj",
+    "o_proj",
+    "k_proj",
+    "down_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:982e2161016fcad20fc0b0a234e554a33986d2b3b621a52300ba54c0b94dcad1
+size 137714904

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:680fbe0607b6ed8913038960539d086307b7b84dd07b545263337a9ae091cf5d
+size 275817586

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": 0.9586673974990845,
+      "kl": 0.19199295341968536,
+      "learning_rate": 1.4705882352941177e-06,
+      "logps/chosen": -145.66531982421876,
+      "loss": 0.5076,
+      "rewards/chosen": -0.011382170021533966,
+      "step": 10
+    },
+    {
+      "epoch": 0.17777777777777778,
+      "grad_norm": 0.9881852269172668,
+      "kl": 0.22232285141944885,
+      "learning_rate": 2.9411764705882355e-06,
+      "logps/chosen": -144.47635498046876,
+      "loss": 0.506,
+      "rewards/chosen": -0.0017147257924079895,
+      "step": 20
+    },
+    {
+      "epoch": 0.26666666666666666,
+      "grad_norm": 0.9230504035949707,
+      "kl": 0.2664317190647125,
+      "learning_rate": 4.411764705882353e-06,
+      "logps/chosen": -144.5919189453125,
+      "loss": 0.5053,
+      "rewards/chosen": 0.005377597734332084,
+      "step": 30
+    },
+    {
+      "epoch": 0.35555555555555557,
+      "grad_norm": 0.9962123036384583,
+      "kl": 0.35221290588378906,
+      "learning_rate": 4.995131923687488e-06,
+      "logps/chosen": -142.9361328125,
+      "loss": 0.5012,
+      "rewards/chosen": 0.03048495054244995,
+      "step": 40
+    },
+    {
+      "epoch": 0.4444444444444444,
+      "grad_norm": 1.229694128036499,
+      "kl": 0.8521757125854492,
+      "learning_rate": 4.965451197130373e-06,
+      "logps/chosen": -142.90025634765624,
+      "loss": 0.5014,
+      "rewards/chosen": 0.07963300943374634,
+      "step": 50
+    },
+    {
+      "epoch": 0.5333333333333333,
+      "grad_norm": 1.35447359085083,
+      "kl": 1.8928890228271484,
+      "learning_rate": 4.90911473983908e-06,
+      "logps/chosen": -147.9736572265625,
+      "loss": 0.504,
+      "rewards/chosen": 0.17341512441635132,
+      "step": 60
+    },
+    {
+      "epoch": 0.6222222222222222,
+      "grad_norm": 1.301222324371338,
+      "kl": 3.797394275665283,
+      "learning_rate": 4.826731644963705e-06,
+      "logps/chosen": -144.52886962890625,
+      "loss": 0.514,
+      "rewards/chosen": 0.323371958732605,
+      "step": 70
+    },
+    {
+      "epoch": 0.7111111111111111,
+      "grad_norm": 1.4558587074279785,
+      "kl": 6.093751907348633,
+      "learning_rate": 4.71919261421297e-06,
+      "logps/chosen": -150.49774169921875,
+      "loss": 0.5034,
+      "rewards/chosen": 0.5952839851379395,
+      "step": 80
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 1.3663493394851685,
+      "kl": 8.93967342376709,
+      "learning_rate": 4.587660327850203e-06,
+      "logps/chosen": -134.95494384765624,
+      "loss": 0.5298,
+      "rewards/chosen": 0.7676385402679443,
+      "step": 90
+    },
+    {
+      "epoch": 0.8888888888888888,
+      "grad_norm": 1.547540545463562,
+      "kl": 11.540582656860352,
+      "learning_rate": 4.43355687413747e-06,
+      "logps/chosen": -128.8071044921875,
+      "loss": 0.521,
+      "rewards/chosen": 1.0660431861877442,
+      "step": 100
+    },
+    {
+      "epoch": 0.9777777777777777,
+      "grad_norm": 1.3188670873641968,
+      "kl": 14.06761360168457,
+      "learning_rate": 4.258548374136976e-06,
+      "logps/chosen": -130.93094482421876,
+      "loss": 0.5228,
+      "rewards/chosen": 1.3124106407165528,
+      "step": 110
+    },
+    {
+      "epoch": 1.0666666666666667,
+      "grad_norm": 1.3436963558197021,
+      "kl": 16.5745906829834,
+      "learning_rate": 4.064526968101844e-06,
+      "logps/chosen": -124.19566650390625,
+      "loss": 0.5302,
+      "rewards/chosen": 1.527147102355957,
+      "step": 120
+    },
+    {
+      "epoch": 1.1555555555555554,
+      "grad_norm": 1.3086756467819214,
+      "kl": 18.41370964050293,
+      "learning_rate": 3.853590358214119e-06,
+      "logps/chosen": -129.2854736328125,
+      "loss": 0.5383,
+      "rewards/chosen": 1.6645441055297852,
+      "step": 130
+    },
+    {
+      "epoch": 1.2444444444444445,
+      "grad_norm": 1.2180778980255127,
+      "kl": 18.382841110229492,
+      "learning_rate": 3.6280191288478437e-06,
+      "logps/chosen": -127.15008544921875,
+      "loss": 0.5155,
+      "rewards/chosen": 1.7688350677490234,
+      "step": 140
+    },
+    {
+      "epoch": 1.3333333333333333,
+      "grad_norm": 1.142663836479187,
+      "kl": 20.387691497802734,
+      "learning_rate": 3.3902520895638674e-06,
+      "logps/chosen": -130.60250244140624,
+      "loss": 0.5408,
+      "rewards/chosen": 1.8582941055297852,
+      "step": 150
+    },
+    {
+      "epoch": 1.4222222222222223,
+      "grad_norm": 1.1040987968444824,
+      "kl": 20.953645706176758,
+      "learning_rate": 3.142859907420615e-06,
+      "logps/chosen": -120.3420654296875,
+      "loss": 0.5502,
+      "rewards/chosen": 1.877711868286133,
+      "step": 160
+    },
+    {
+      "epoch": 1.511111111111111,
+      "grad_norm": 1.446460485458374,
+      "kl": 22.705646514892578,
+      "learning_rate": 2.8885173136805126e-06,
+      "logps/chosen": -121.3359130859375,
+      "loss": 0.5312,
+      "rewards/chosen": 2.1138553619384766,
+      "step": 170
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.9512816667556763,
+      "kl": 22.2016658782959,
+      "learning_rate": 2.629974185404951e-06,
+      "logps/chosen": -126.013720703125,
+      "loss": 0.5678,
+      "rewards/chosen": 1.8899778366088866,
+      "step": 180
+    },
+    {
+      "epoch": 1.6888888888888889,
+      "grad_norm": 1.202101469039917,
+      "kl": 22.479812622070312,
+      "learning_rate": 2.3700258145950495e-06,
+      "logps/chosen": -127.17119140625,
+      "loss": 0.4841,
+      "rewards/chosen": 2.30587100982666,
+      "step": 190
+    },
+    {
+      "epoch": 1.7777777777777777,
+      "grad_norm": 1.2024880647659302,
+      "kl": 23.706804275512695,
+      "learning_rate": 2.1114826863194882e-06,
+      "logps/chosen": -119.14400634765624,
+      "loss": 0.5279,
+      "rewards/chosen": 2.2496980667114257,
+      "step": 200
+    },
+    {
+      "epoch": 1.8666666666666667,
+      "grad_norm": 1.2260613441467285,
+      "kl": 22.990591049194336,
+      "learning_rate": 1.8571400925793855e-06,
+      "logps/chosen": -128.0656005859375,
+      "loss": 0.5057,
+      "rewards/chosen": 2.2760807037353517,
+      "step": 210
+    },
+    {
+      "epoch": 1.9555555555555557,
+      "grad_norm": 1.1307311058044434,
+      "kl": 23.434030532836914,
+      "learning_rate": 1.6097479104361328e-06,
+      "logps/chosen": -124.27503662109375,
+      "loss": 0.539,
+      "rewards/chosen": 2.1261518478393553,
+      "step": 220
+    },
+    {
+      "epoch": 2.0444444444444443,
+      "grad_norm": 1.2096738815307617,
+      "kl": 24.883241653442383,
+      "learning_rate": 1.3719808711521573e-06,
+      "logps/chosen": -125.847314453125,
+      "loss": 0.5433,
+      "rewards/chosen": 2.296957015991211,
+      "step": 230
+    },
+    {
+      "epoch": 2.1333333333333333,
+      "grad_norm": 1.2274669408798218,
+      "kl": 23.971763610839844,
+      "learning_rate": 1.1464096417858821e-06,
+      "logps/chosen": -122.934521484375,
+      "loss": 0.4878,
+      "rewards/chosen": 2.448800468444824,
+      "step": 240
+    },
+    {
+      "epoch": 2.2222222222222223,
+      "grad_norm": 1.433097243309021,
+      "kl": 24.515583038330078,
+      "learning_rate": 9.354730318981561e-07,
+      "logps/chosen": -111.53873291015626,
+      "loss": 0.5241,
+      "rewards/chosen": 2.335133171081543,
+      "step": 250
+    },
+    {
+      "epoch": 2.311111111111111,
+      "grad_norm": 1.2111012935638428,
+      "kl": 24.930599212646484,
+      "learning_rate": 7.414516258630245e-07,
+      "logps/chosen": -128.9908935546875,
+      "loss": 0.4901,
+      "rewards/chosen": 2.533807373046875,
+      "step": 260
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 1.340284824371338,
+      "kl": 24.58279037475586,
+      "learning_rate": 5.664431258625305e-07,
+      "logps/chosen": -122.60418701171875,
+      "loss": 0.4913,
+      "rewards/chosen": 2.4873432159423827,
+      "step": 270
+    },
+    {
+      "epoch": 2.488888888888889,
+      "grad_norm": 1.3646727800369263,
+      "kl": 24.800968170166016,
+      "learning_rate": 4.123396721497977e-07,
+      "logps/chosen": -118.5873046875,
+      "loss": 0.5525,
+      "rewards/chosen": 2.2357526779174806,
+      "step": 280
+    },
+    {
+      "epoch": 2.5777777777777775,
+      "grad_norm": 1.2073670625686646,
+      "kl": 25.220767974853516,
+      "learning_rate": 2.8080738578703054e-07,
+      "logps/chosen": -118.42447509765626,
+      "loss": 0.5128,
+      "rewards/chosen": 2.481866455078125,
+      "step": 290
+    },
+    {
+      "epoch": 2.6666666666666665,
+      "grad_norm": 1.2993130683898926,
+      "kl": 25.565937042236328,
+      "learning_rate": 1.7326835503629542e-07,
+      "logps/chosen": -120.5993408203125,
+      "loss": 0.5379,
+      "rewards/chosen": 2.363565444946289,
+      "step": 300
+    },
+    {
+      "epoch": 2.7555555555555555,
+      "grad_norm": 1.1960065364837646,
+      "kl": 23.65550422668457,
+      "learning_rate": 9.088526016092142e-08,
+      "logps/chosen": -128.53665771484376,
+      "loss": 0.4636,
+      "rewards/chosen": 2.549031639099121,
+      "step": 310
+    },
+    {
+      "epoch": 2.8444444444444446,
+      "grad_norm": 1.4810154438018799,
+      "kl": 24.803632736206055,
+      "learning_rate": 3.4548802869627806e-08,
+      "logps/chosen": -119.30946044921875,
+      "loss": 0.5054,
+      "rewards/chosen": 2.466901397705078,
+      "step": 320
+    },
+    {
+      "epoch": 2.9333333333333336,
+      "grad_norm": 0.9765888452529907,
+      "kl": 25.129100799560547,
+      "learning_rate": 4.868076312512515e-09,
+      "logps/chosen": -119.96820068359375,
+      "loss": 0.5472,
+      "rewards/chosen": 2.2562030792236327,
+      "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": 7.102230505945498e+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:86fc7fad0452bb7e904042dcb36aeed12fc5e5fc4af2c25c4a7d9d7169818b62
+size 5368

eval_results.json

@@ -0,0 +1,10 @@
+{
+    "epoch": 2.986666666666667,
+    "eval_kl": 25.333707809448242,
+    "eval_logps/chosen": -127.1557421875,
+    "eval_loss": 0.556947648525238,
+    "eval_rewards/chosen": 2.2726934814453124,
+    "eval_runtime": 45.528,
+    "eval_samples_per_second": 2.196,
+    "eval_steps_per_second": 2.196
+}

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": 7.102230505945498e+16,
+    "train_loss": 0.5188913842042288,
+    "train_runtime": 2226.9521,
+    "train_samples_per_second": 1.212,
+    "train_steps_per_second": 0.151
+}

trainer_log.jsonl

@@ -0,0 +1,34 @@
+{"current_steps": 10, "total_steps": 336, "loss": 0.5076, "learning_rate": 1.4705882352941177e-06, "epoch": 0.08888888888888889, "percentage": 2.98, "elapsed_time": "0:01:07", "remaining_time": "0:36:26"}
+{"current_steps": 20, "total_steps": 336, "loss": 0.506, "learning_rate": 2.9411764705882355e-06, "epoch": 0.17777777777777778, "percentage": 5.95, "elapsed_time": "0:02:12", "remaining_time": "0:35:00"}
+{"current_steps": 30, "total_steps": 336, "loss": 0.5053, "learning_rate": 4.411764705882353e-06, "epoch": 0.26666666666666666, "percentage": 8.93, "elapsed_time": "0:03:24", "remaining_time": "0:34:43"}
+{"current_steps": 40, "total_steps": 336, "loss": 0.5012, "learning_rate": 4.995131923687488e-06, "epoch": 0.35555555555555557, "percentage": 11.9, "elapsed_time": "0:04:31", "remaining_time": "0:33:30"}
+{"current_steps": 50, "total_steps": 336, "loss": 0.5014, "learning_rate": 4.965451197130373e-06, "epoch": 0.4444444444444444, "percentage": 14.88, "elapsed_time": "0:05:37", "remaining_time": "0:32:10"}
+{"current_steps": 60, "total_steps": 336, "loss": 0.504, "learning_rate": 4.90911473983908e-06, "epoch": 0.5333333333333333, "percentage": 17.86, "elapsed_time": "0:06:43", "remaining_time": "0:30:55"}
+{"current_steps": 70, "total_steps": 336, "loss": 0.514, "learning_rate": 4.826731644963705e-06, "epoch": 0.6222222222222222, "percentage": 20.83, "elapsed_time": "0:07:48", "remaining_time": "0:29:38"}
+{"current_steps": 80, "total_steps": 336, "loss": 0.5034, "learning_rate": 4.71919261421297e-06, "epoch": 0.7111111111111111, "percentage": 23.81, "elapsed_time": "0:08:53", "remaining_time": "0:28:26"}
+{"current_steps": 90, "total_steps": 336, "loss": 0.5298, "learning_rate": 4.587660327850203e-06, "epoch": 0.8, "percentage": 26.79, "elapsed_time": "0:09:58", "remaining_time": "0:27:16"}
+{"current_steps": 100, "total_steps": 336, "loss": 0.521, "learning_rate": 4.43355687413747e-06, "epoch": 0.8888888888888888, "percentage": 29.76, "elapsed_time": "0:11:05", "remaining_time": "0:26:10"}
+{"current_steps": 110, "total_steps": 336, "loss": 0.5228, "learning_rate": 4.258548374136976e-06, "epoch": 0.9777777777777777, "percentage": 32.74, "elapsed_time": "0:12:14", "remaining_time": "0:25:09"}
+{"current_steps": 120, "total_steps": 336, "loss": 0.5302, "learning_rate": 4.064526968101844e-06, "epoch": 1.0666666666666667, "percentage": 35.71, "elapsed_time": "0:13:25", "remaining_time": "0:24:10"}
+{"current_steps": 130, "total_steps": 336, "loss": 0.5383, "learning_rate": 3.853590358214119e-06, "epoch": 1.1555555555555554, "percentage": 38.69, "elapsed_time": "0:14:37", "remaining_time": "0:23:10"}
+{"current_steps": 140, "total_steps": 336, "loss": 0.5155, "learning_rate": 3.6280191288478437e-06, "epoch": 1.2444444444444445, "percentage": 41.67, "elapsed_time": "0:15:48", "remaining_time": "0:22:08"}
+{"current_steps": 150, "total_steps": 336, "loss": 0.5408, "learning_rate": 3.3902520895638674e-06, "epoch": 1.3333333333333333, "percentage": 44.64, "elapsed_time": "0:16:53", "remaining_time": "0:20:57"}
+{"current_steps": 160, "total_steps": 336, "loss": 0.5502, "learning_rate": 3.142859907420615e-06, "epoch": 1.4222222222222223, "percentage": 47.62, "elapsed_time": "0:17:58", "remaining_time": "0:19:46"}
+{"current_steps": 170, "total_steps": 336, "loss": 0.5312, "learning_rate": 2.8885173136805126e-06, "epoch": 1.511111111111111, "percentage": 50.6, "elapsed_time": "0:19:04", "remaining_time": "0:18:37"}
+{"current_steps": 180, "total_steps": 336, "loss": 0.5678, "learning_rate": 2.629974185404951e-06, "epoch": 1.6, "percentage": 53.57, "elapsed_time": "0:20:10", "remaining_time": "0:17:28"}
+{"current_steps": 190, "total_steps": 336, "loss": 0.4841, "learning_rate": 2.3700258145950495e-06, "epoch": 1.6888888888888889, "percentage": 56.55, "elapsed_time": "0:21:15", "remaining_time": "0:16:19"}
+{"current_steps": 200, "total_steps": 336, "loss": 0.5279, "learning_rate": 2.1114826863194882e-06, "epoch": 1.7777777777777777, "percentage": 59.52, "elapsed_time": "0:22:20", "remaining_time": "0:15:11"}
+{"current_steps": 210, "total_steps": 336, "loss": 0.5057, "learning_rate": 1.8571400925793855e-06, "epoch": 1.8666666666666667, "percentage": 62.5, "elapsed_time": "0:23:25", "remaining_time": "0:14:03"}
+{"current_steps": 220, "total_steps": 336, "loss": 0.539, "learning_rate": 1.6097479104361328e-06, "epoch": 1.9555555555555557, "percentage": 65.48, "elapsed_time": "0:24:30", "remaining_time": "0:12:55"}
+{"current_steps": 230, "total_steps": 336, "loss": 0.5433, "learning_rate": 1.3719808711521573e-06, "epoch": 2.0444444444444443, "percentage": 68.45, "elapsed_time": "0:25:35", "remaining_time": "0:11:47"}
+{"current_steps": 240, "total_steps": 336, "loss": 0.4878, "learning_rate": 1.1464096417858821e-06, "epoch": 2.1333333333333333, "percentage": 71.43, "elapsed_time": "0:26:42", "remaining_time": "0:10:40"}
+{"current_steps": 250, "total_steps": 336, "loss": 0.5241, "learning_rate": 9.354730318981561e-07, "epoch": 2.2222222222222223, "percentage": 74.4, "elapsed_time": "0:27:47", "remaining_time": "0:09:33"}
+{"current_steps": 260, "total_steps": 336, "loss": 0.4901, "learning_rate": 7.414516258630245e-07, "epoch": 2.311111111111111, "percentage": 77.38, "elapsed_time": "0:28:53", "remaining_time": "0:08:26"}
+{"current_steps": 270, "total_steps": 336, "loss": 0.4913, "learning_rate": 5.664431258625305e-07, "epoch": 2.4, "percentage": 80.36, "elapsed_time": "0:29:58", "remaining_time": "0:07:19"}
+{"current_steps": 280, "total_steps": 336, "loss": 0.5525, "learning_rate": 4.123396721497977e-07, "epoch": 2.488888888888889, "percentage": 83.33, "elapsed_time": "0:31:02", "remaining_time": "0:06:12"}
+{"current_steps": 290, "total_steps": 336, "loss": 0.5128, "learning_rate": 2.8080738578703054e-07, "epoch": 2.5777777777777775, "percentage": 86.31, "elapsed_time": "0:32:07", "remaining_time": "0:05:05"}
+{"current_steps": 300, "total_steps": 336, "loss": 0.5379, "learning_rate": 1.7326835503629542e-07, "epoch": 2.6666666666666665, "percentage": 89.29, "elapsed_time": "0:33:12", "remaining_time": "0:03:59"}
+{"current_steps": 310, "total_steps": 336, "loss": 0.4636, "learning_rate": 9.088526016092142e-08, "epoch": 2.7555555555555555, "percentage": 92.26, "elapsed_time": "0:34:16", "remaining_time": "0:02:52"}
+{"current_steps": 320, "total_steps": 336, "loss": 0.5054, "learning_rate": 3.4548802869627806e-08, "epoch": 2.8444444444444446, "percentage": 95.24, "elapsed_time": "0:35:21", "remaining_time": "0:01:46"}
+{"current_steps": 330, "total_steps": 336, "loss": 0.5472, "learning_rate": 4.868076312512515e-09, "epoch": 2.9333333333333336, "percentage": 98.21, "elapsed_time": "0:36:26", "remaining_time": "0:00:39"}
+{"current_steps": 336, "total_steps": 336, "epoch": 2.986666666666667, "percentage": 100.0, "elapsed_time": "0:37:06", "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": 0.9586673974990845,
+      "kl": 0.19199295341968536,
+      "learning_rate": 1.4705882352941177e-06,
+      "logps/chosen": -145.66531982421876,
+      "loss": 0.5076,
+      "rewards/chosen": -0.011382170021533966,
+      "step": 10
+    },
+    {
+      "epoch": 0.17777777777777778,
+      "grad_norm": 0.9881852269172668,
+      "kl": 0.22232285141944885,
+      "learning_rate": 2.9411764705882355e-06,
+      "logps/chosen": -144.47635498046876,
+      "loss": 0.506,
+      "rewards/chosen": -0.0017147257924079895,
+      "step": 20
+    },
+    {
+      "epoch": 0.26666666666666666,
+      "grad_norm": 0.9230504035949707,
+      "kl": 0.2664317190647125,
+      "learning_rate": 4.411764705882353e-06,
+      "logps/chosen": -144.5919189453125,
+      "loss": 0.5053,
+      "rewards/chosen": 0.005377597734332084,
+      "step": 30
+    },
+    {
+      "epoch": 0.35555555555555557,
+      "grad_norm": 0.9962123036384583,
+      "kl": 0.35221290588378906,
+      "learning_rate": 4.995131923687488e-06,
+      "logps/chosen": -142.9361328125,
+      "loss": 0.5012,
+      "rewards/chosen": 0.03048495054244995,
+      "step": 40
+    },
+    {
+      "epoch": 0.4444444444444444,
+      "grad_norm": 1.229694128036499,
+      "kl": 0.8521757125854492,
+      "learning_rate": 4.965451197130373e-06,
+      "logps/chosen": -142.90025634765624,
+      "loss": 0.5014,
+      "rewards/chosen": 0.07963300943374634,
+      "step": 50
+    },
+    {
+      "epoch": 0.5333333333333333,
+      "grad_norm": 1.35447359085083,
+      "kl": 1.8928890228271484,
+      "learning_rate": 4.90911473983908e-06,
+      "logps/chosen": -147.9736572265625,
+      "loss": 0.504,
+      "rewards/chosen": 0.17341512441635132,
+      "step": 60
+    },
+    {
+      "epoch": 0.6222222222222222,
+      "grad_norm": 1.301222324371338,
+      "kl": 3.797394275665283,
+      "learning_rate": 4.826731644963705e-06,
+      "logps/chosen": -144.52886962890625,
+      "loss": 0.514,
+      "rewards/chosen": 0.323371958732605,
+      "step": 70
+    },
+    {
+      "epoch": 0.7111111111111111,
+      "grad_norm": 1.4558587074279785,
+      "kl": 6.093751907348633,
+      "learning_rate": 4.71919261421297e-06,
+      "logps/chosen": -150.49774169921875,
+      "loss": 0.5034,
+      "rewards/chosen": 0.5952839851379395,
+      "step": 80
+    },
+    {
+      "epoch": 0.8,
+      "grad_norm": 1.3663493394851685,
+      "kl": 8.93967342376709,
+      "learning_rate": 4.587660327850203e-06,
+      "logps/chosen": -134.95494384765624,
+      "loss": 0.5298,
+      "rewards/chosen": 0.7676385402679443,
+      "step": 90
+    },
+    {
+      "epoch": 0.8888888888888888,
+      "grad_norm": 1.547540545463562,
+      "kl": 11.540582656860352,
+      "learning_rate": 4.43355687413747e-06,
+      "logps/chosen": -128.8071044921875,
+      "loss": 0.521,
+      "rewards/chosen": 1.0660431861877442,
+      "step": 100
+    },
+    {
+      "epoch": 0.9777777777777777,
+      "grad_norm": 1.3188670873641968,
+      "kl": 14.06761360168457,
+      "learning_rate": 4.258548374136976e-06,
+      "logps/chosen": -130.93094482421876,
+      "loss": 0.5228,
+      "rewards/chosen": 1.3124106407165528,
+      "step": 110
+    },
+    {
+      "epoch": 1.0666666666666667,
+      "grad_norm": 1.3436963558197021,
+      "kl": 16.5745906829834,
+      "learning_rate": 4.064526968101844e-06,
+      "logps/chosen": -124.19566650390625,
+      "loss": 0.5302,
+      "rewards/chosen": 1.527147102355957,
+      "step": 120
+    },
+    {
+      "epoch": 1.1555555555555554,
+      "grad_norm": 1.3086756467819214,
+      "kl": 18.41370964050293,
+      "learning_rate": 3.853590358214119e-06,
+      "logps/chosen": -129.2854736328125,
+      "loss": 0.5383,
+      "rewards/chosen": 1.6645441055297852,
+      "step": 130
+    },
+    {
+      "epoch": 1.2444444444444445,
+      "grad_norm": 1.2180778980255127,
+      "kl": 18.382841110229492,
+      "learning_rate": 3.6280191288478437e-06,
+      "logps/chosen": -127.15008544921875,
+      "loss": 0.5155,
+      "rewards/chosen": 1.7688350677490234,
+      "step": 140
+    },
+    {
+      "epoch": 1.3333333333333333,
+      "grad_norm": 1.142663836479187,
+      "kl": 20.387691497802734,
+      "learning_rate": 3.3902520895638674e-06,
+      "logps/chosen": -130.60250244140624,
+      "loss": 0.5408,
+      "rewards/chosen": 1.8582941055297852,
+      "step": 150
+    },
+    {
+      "epoch": 1.4222222222222223,
+      "grad_norm": 1.1040987968444824,
+      "kl": 20.953645706176758,
+      "learning_rate": 3.142859907420615e-06,
+      "logps/chosen": -120.3420654296875,
+      "loss": 0.5502,
+      "rewards/chosen": 1.877711868286133,
+      "step": 160
+    },
+    {
+      "epoch": 1.511111111111111,
+      "grad_norm": 1.446460485458374,
+      "kl": 22.705646514892578,
+      "learning_rate": 2.8885173136805126e-06,
+      "logps/chosen": -121.3359130859375,
+      "loss": 0.5312,
+      "rewards/chosen": 2.1138553619384766,
+      "step": 170
+    },
+    {
+      "epoch": 1.6,
+      "grad_norm": 0.9512816667556763,
+      "kl": 22.2016658782959,
+      "learning_rate": 2.629974185404951e-06,
+      "logps/chosen": -126.013720703125,
+      "loss": 0.5678,
+      "rewards/chosen": 1.8899778366088866,
+      "step": 180
+    },
+    {
+      "epoch": 1.6888888888888889,
+      "grad_norm": 1.202101469039917,
+      "kl": 22.479812622070312,
+      "learning_rate": 2.3700258145950495e-06,
+      "logps/chosen": -127.17119140625,
+      "loss": 0.4841,
+      "rewards/chosen": 2.30587100982666,
+      "step": 190
+    },
+    {
+      "epoch": 1.7777777777777777,
+      "grad_norm": 1.2024880647659302,
+      "kl": 23.706804275512695,
+      "learning_rate": 2.1114826863194882e-06,
+      "logps/chosen": -119.14400634765624,
+      "loss": 0.5279,
+      "rewards/chosen": 2.2496980667114257,
+      "step": 200
+    },
+    {
+      "epoch": 1.8666666666666667,
+      "grad_norm": 1.2260613441467285,
+      "kl": 22.990591049194336,
+      "learning_rate": 1.8571400925793855e-06,
+      "logps/chosen": -128.0656005859375,
+      "loss": 0.5057,
+      "rewards/chosen": 2.2760807037353517,
+      "step": 210
+    },
+    {
+      "epoch": 1.9555555555555557,
+      "grad_norm": 1.1307311058044434,
+      "kl": 23.434030532836914,
+      "learning_rate": 1.6097479104361328e-06,
+      "logps/chosen": -124.27503662109375,
+      "loss": 0.539,
+      "rewards/chosen": 2.1261518478393553,
+      "step": 220
+    },
+    {
+      "epoch": 2.0444444444444443,
+      "grad_norm": 1.2096738815307617,
+      "kl": 24.883241653442383,
+      "learning_rate": 1.3719808711521573e-06,
+      "logps/chosen": -125.847314453125,
+      "loss": 0.5433,
+      "rewards/chosen": 2.296957015991211,
+      "step": 230
+    },
+    {
+      "epoch": 2.1333333333333333,
+      "grad_norm": 1.2274669408798218,
+      "kl": 23.971763610839844,
+      "learning_rate": 1.1464096417858821e-06,
+      "logps/chosen": -122.934521484375,
+      "loss": 0.4878,
+      "rewards/chosen": 2.448800468444824,
+      "step": 240
+    },
+    {
+      "epoch": 2.2222222222222223,
+      "grad_norm": 1.433097243309021,
+      "kl": 24.515583038330078,
+      "learning_rate": 9.354730318981561e-07,
+      "logps/chosen": -111.53873291015626,
+      "loss": 0.5241,
+      "rewards/chosen": 2.335133171081543,
+      "step": 250
+    },
+    {
+      "epoch": 2.311111111111111,
+      "grad_norm": 1.2111012935638428,
+      "kl": 24.930599212646484,
+      "learning_rate": 7.414516258630245e-07,
+      "logps/chosen": -128.9908935546875,
+      "loss": 0.4901,
+      "rewards/chosen": 2.533807373046875,
+      "step": 260
+    },
+    {
+      "epoch": 2.4,
+      "grad_norm": 1.340284824371338,
+      "kl": 24.58279037475586,
+      "learning_rate": 5.664431258625305e-07,
+      "logps/chosen": -122.60418701171875,
+      "loss": 0.4913,
+      "rewards/chosen": 2.4873432159423827,
+      "step": 270
+    },
+    {
+      "epoch": 2.488888888888889,
+      "grad_norm": 1.3646727800369263,
+      "kl": 24.800968170166016,
+      "learning_rate": 4.123396721497977e-07,
+      "logps/chosen": -118.5873046875,
+      "loss": 0.5525,
+      "rewards/chosen": 2.2357526779174806,
+      "step": 280
+    },
+    {
+      "epoch": 2.5777777777777775,
+      "grad_norm": 1.2073670625686646,
+      "kl": 25.220767974853516,
+      "learning_rate": 2.8080738578703054e-07,
+      "logps/chosen": -118.42447509765626,
+      "loss": 0.5128,
+      "rewards/chosen": 2.481866455078125,
+      "step": 290
+    },
+    {
+      "epoch": 2.6666666666666665,
+      "grad_norm": 1.2993130683898926,
+      "kl": 25.565937042236328,
+      "learning_rate": 1.7326835503629542e-07,
+      "logps/chosen": -120.5993408203125,
+      "loss": 0.5379,
+      "rewards/chosen": 2.363565444946289,
+      "step": 300
+    },
+    {
+      "epoch": 2.7555555555555555,
+      "grad_norm": 1.1960065364837646,
+      "kl": 23.65550422668457,
+      "learning_rate": 9.088526016092142e-08,
+      "logps/chosen": -128.53665771484376,
+      "loss": 0.4636,
+      "rewards/chosen": 2.549031639099121,
+      "step": 310
+    },
+    {
+      "epoch": 2.8444444444444446,
+      "grad_norm": 1.4810154438018799,
+      "kl": 24.803632736206055,
+      "learning_rate": 3.4548802869627806e-08,
+      "logps/chosen": -119.30946044921875,
+      "loss": 0.5054,
+      "rewards/chosen": 2.466901397705078,
+      "step": 320
+    },
+    {
+      "epoch": 2.9333333333333336,
+      "grad_norm": 0.9765888452529907,
+      "kl": 25.129100799560547,
+      "learning_rate": 4.868076312512515e-09,
+      "logps/chosen": -119.96820068359375,
+      "loss": 0.5472,
+      "rewards/chosen": 2.2562030792236327,
+      "step": 330
+    },
+    {
+      "epoch": 2.986666666666667,
+      "step": 336,
+      "total_flos": 7.102230505945498e+16,
+      "train_loss": 0.5188913842042288,
+      "train_runtime": 2226.9521,
+      "train_samples_per_second": 1.212,
+      "train_steps_per_second": 0.151
+    }
+  ],
+  "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": 7.102230505945498e+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:86fc7fad0452bb7e904042dcb36aeed12fc5e5fc4af2c25c4a7d9d7169818b62
+size 5368