Training in progress, step 350, checkpoint

checkpoint-350/README.md

@@ -0,0 +1,202 @@
+---
+base_model: NESPED-GEN/StableCode-text2SQL-alias-indentacao
+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.13.2

checkpoint-350/adapter_config.json

@@ -0,0 +1,37 @@
+{
+  "alpha_pattern": {},
+  "auto_mapping": {
+    "base_model_class": "StableLmForCausalLM",
+    "parent_library": "transformers.models.stablelm.modeling_stablelm"
+  },
+  "base_model_name_or_path": "NESPED-GEN/StableCode-text2SQL-alias-indentacao",
+  "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": 128,
+  "lora_dropout": 0.1,
+  "megatron_config": null,
+  "megatron_core": "megatron.core",
+  "modules_to_save": null,
+  "peft_type": "LORA",
+  "r": 64,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": [
+    "down_proj",
+    "gate_proj",
+    "q_proj",
+    "o_proj",
+    "k_proj",
+    "up_proj",
+    "v_proj"
+  ],
+  "task_type": null,
+  "use_dora": false,
+  "use_rslora": false
+}

checkpoint-350/adapter_model.safetensors

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

checkpoint-350/optimizer.pt

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

checkpoint-350/rng_state.pth

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

checkpoint-350/scheduler.pt

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

checkpoint-350/special_tokens_map.json

@@ -0,0 +1,24 @@
+{
+  "bos_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": "<|endoftext|>",
+  "unk_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

checkpoint-350/tokenizer_config.json

@@ -0,0 +1,392 @@
+{
+  "add_bos_token": false,
+  "add_eos_token": false,
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<|padding|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50254": {
+      "content": "                        ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50255": {
+      "content": "                       ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50256": {
+      "content": "                      ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50257": {
+      "content": "                     ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50258": {
+      "content": "                    ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50259": {
+      "content": "                   ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50260": {
+      "content": "                  ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50261": {
+      "content": "                 ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50262": {
+      "content": "                ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50263": {
+      "content": "               ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50264": {
+      "content": "              ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50265": {
+      "content": "             ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50266": {
+      "content": "            ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50267": {
+      "content": "           ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50268": {
+      "content": "          ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50269": {
+      "content": "         ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50270": {
+      "content": "        ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50271": {
+      "content": "       ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50272": {
+      "content": "      ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50273": {
+      "content": "     ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50274": {
+      "content": "    ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50275": {
+      "content": "   ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50276": {
+      "content": "  ",
+      "lstrip": false,
+      "normalized": true,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50277": {
+      "content": "<fim_prefix>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50278": {
+      "content": "<fim_middle>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50279": {
+      "content": "<fim_suffix>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50280": {
+      "content": "<fim_pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50281": {
+      "content": "<filename>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50282": {
+      "content": "<gh_stars>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50283": {
+      "content": "<issue_start>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50284": {
+      "content": "<issue_comment>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50285": {
+      "content": "<issue_closed>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50286": {
+      "content": "<jupyter_start>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50287": {
+      "content": "<jupyter_text>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50288": {
+      "content": "<jupyter_code>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50289": {
+      "content": "<jupyter_output>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50290": {
+      "content": "<empty_output>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50291": {
+      "content": "<commit_before>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50292": {
+      "content": "<commit_msg>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50293": {
+      "content": "<commit_after>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50294": {
+      "content": "<reponame>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50295": {
+      "content": "<repo_continuation>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50296": {
+      "content": "[PAD]",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "50297": {
+      "content": "<|im_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "50298": {
+      "content": "<|im_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    }
+  },
+  "bos_token": "<|endoftext|>",
+  "chat_template": "{% if messages[0]['role'] == 'system' %}{% set loop_messages = messages[1:] %}{% set system_message = messages[0]['content'] %}{% else %}{% set loop_messages = messages %}{% set system_message = 'You are a helpful assistant.' %}{% endif %}{% if not add_generation_prompt is defined %}{% set add_generation_prompt = false %}{% endif %}{% for message in loop_messages %}{% if loop.index0 == 0 %}{{'<|im_start|>system\n' + system_message + '<|im_end|>\n'}}{% endif %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
+  "clean_up_tokenization_spaces": true,
+  "eos_token": "<|endoftext|>",
+  "map_device": "auto",
+  "model_max_length": 4096,
+  "pad_token": "<|endoftext|>",
+  "tokenizer_class": "GPTNeoXTokenizer",
+  "unk_token": "<|endoftext|>"
+}

checkpoint-350/trainer_state.json

@@ -0,0 +1,250 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.3111111111111111,
+  "eval_steps": 50,
+  "global_step": 350,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.044444444444444446,
+      "grad_norm": 49.98692321777344,
+      "learning_rate": 5e-07,
+      "logits/chosen": -1.6455594301223755,
+      "logits/rejected": -1.8609813451766968,
+      "logps/chosen": -69.99601745605469,
+      "logps/rejected": -28.68402099609375,
+      "loss": 0.4811,
+      "rewards/accuracies": 0.9200000166893005,
+      "rewards/chosen": 0.43002215027809143,
+      "rewards/margins": 0.5248908400535583,
+      "rewards/rejected": -0.09486865252256393,
+      "step": 50
+    },
+    {
+      "epoch": 0.044444444444444446,
+      "eval_logits/chosen": -1.683065414428711,
+      "eval_logits/rejected": -1.8381606340408325,
+      "eval_logps/chosen": -65.2982406616211,
+      "eval_logps/rejected": -31.1202335357666,
+      "eval_loss": 0.26570644974708557,
+      "eval_rewards/accuracies": 1.0,
+      "eval_rewards/chosen": 0.9695585370063782,
+      "eval_rewards/margins": 1.199601650238037,
+      "eval_rewards/rejected": -0.23004300892353058,
+      "eval_runtime": 547.5928,
+      "eval_samples_per_second": 0.51,
+      "eval_steps_per_second": 0.51,
+      "step": 50
+    },
+    {
+      "epoch": 0.08888888888888889,
+      "grad_norm": 18.30071449279785,
+      "learning_rate": 5e-07,
+      "logits/chosen": -1.7492402791976929,
+      "logits/rejected": -1.8840913772583008,
+      "logps/chosen": -62.02247619628906,
+      "logps/rejected": -31.541664123535156,
+      "loss": 0.1648,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": 1.368556022644043,
+      "rewards/margins": 1.8034323453903198,
+      "rewards/rejected": -0.4348761737346649,
+      "step": 100
+    },
+    {
+      "epoch": 0.08888888888888889,
+      "eval_logits/chosen": -1.717586636543274,
+      "eval_logits/rejected": -1.840760588645935,
+      "eval_logps/chosen": -57.23891067504883,
+      "eval_logps/rejected": -35.770687103271484,
+      "eval_loss": 0.0847974419593811,
+      "eval_rewards/accuracies": 1.0,
+      "eval_rewards/chosen": 1.77549147605896,
+      "eval_rewards/margins": 2.4705796241760254,
+      "eval_rewards/rejected": -0.6950880885124207,
+      "eval_runtime": 547.9305,
+      "eval_samples_per_second": 0.509,
+      "eval_steps_per_second": 0.509,
+      "step": 100
+    },
+    {
+      "epoch": 0.13333333333333333,
+      "grad_norm": 6.001527309417725,
+      "learning_rate": 5e-07,
+      "logits/chosen": -1.753422498703003,
+      "logits/rejected": -1.856553316116333,
+      "logps/chosen": -55.98763656616211,
+      "logps/rejected": -39.17717361450195,
+      "loss": 0.0522,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": 2.045351266860962,
+      "rewards/margins": 3.051741361618042,
+      "rewards/rejected": -1.00639009475708,
+      "step": 150
+    },
+    {
+      "epoch": 0.13333333333333333,
+      "eval_logits/chosen": -1.7338056564331055,
+      "eval_logits/rejected": -1.838354229927063,
+      "eval_logps/chosen": -51.332515716552734,
+      "eval_logps/rejected": -42.16679382324219,
+      "eval_loss": 0.026723211631178856,
+      "eval_rewards/accuracies": 1.0,
+      "eval_rewards/chosen": 2.366131067276001,
+      "eval_rewards/margins": 3.7008297443389893,
+      "eval_rewards/rejected": -1.3346984386444092,
+      "eval_runtime": 548.2166,
+      "eval_samples_per_second": 0.509,
+      "eval_steps_per_second": 0.509,
+      "step": 150
+    },
+    {
+      "epoch": 0.17777777777777778,
+      "grad_norm": 3.3345212936401367,
+      "learning_rate": 5e-07,
+      "logits/chosen": -1.7408527135849,
+      "logits/rejected": -1.8468482494354248,
+      "logps/chosen": -51.4506721496582,
+      "logps/rejected": -46.23123550415039,
+      "loss": 0.0157,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": 2.529080867767334,
+      "rewards/margins": 4.344725131988525,
+      "rewards/rejected": -1.8156436681747437,
+      "step": 200
+    },
+    {
+      "epoch": 0.17777777777777778,
+      "eval_logits/chosen": -1.744242787361145,
+      "eval_logits/rejected": -1.8309202194213867,
+      "eval_logps/chosen": -47.80553436279297,
+      "eval_logps/rejected": -54.1657600402832,
+      "eval_loss": 0.006161259487271309,
+      "eval_rewards/accuracies": 1.0,
+      "eval_rewards/chosen": 2.7188286781311035,
+      "eval_rewards/margins": 5.253425121307373,
+      "eval_rewards/rejected": -2.5345959663391113,
+      "eval_runtime": 548.15,
+      "eval_samples_per_second": 0.509,
+      "eval_steps_per_second": 0.509,
+      "step": 200
+    },
+    {
+      "epoch": 0.2222222222222222,
+      "grad_norm": 0.14823301136493683,
+      "learning_rate": 5e-07,
+      "logits/chosen": -1.747748851776123,
+      "logits/rejected": -1.8711057901382446,
+      "logps/chosen": -48.346378326416016,
+      "logps/rejected": -64.187744140625,
+      "loss": 0.0026,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": 2.810575485229492,
+      "rewards/margins": 6.410886764526367,
+      "rewards/rejected": -3.600311040878296,
+      "step": 250
+    },
+    {
+      "epoch": 0.2222222222222222,
+      "eval_logits/chosen": -1.7560131549835205,
+      "eval_logits/rejected": -1.8139592409133911,
+      "eval_logps/chosen": -46.13553237915039,
+      "eval_logps/rejected": -74.509765625,
+      "eval_loss": 0.0008142033475451171,
+      "eval_rewards/accuracies": 1.0,
+      "eval_rewards/chosen": 2.885828733444214,
+      "eval_rewards/margins": 7.454825401306152,
+      "eval_rewards/rejected": -4.568996429443359,
+      "eval_runtime": 548.1607,
+      "eval_samples_per_second": 0.509,
+      "eval_steps_per_second": 0.509,
+      "step": 250
+    },
+    {
+      "epoch": 0.26666666666666666,
+      "grad_norm": 0.07440608739852905,
+      "learning_rate": 5e-07,
+      "logits/chosen": -1.7455374002456665,
+      "logits/rejected": -1.8116660118103027,
+      "logps/chosen": -47.130027770996094,
+      "logps/rejected": -80.37494659423828,
+      "loss": 0.0005,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": 2.917734384536743,
+      "rewards/margins": 8.058551788330078,
+      "rewards/rejected": -5.140817642211914,
+      "step": 300
+    },
+    {
+      "epoch": 0.26666666666666666,
+      "eval_logits/chosen": -1.7649332284927368,
+      "eval_logits/rejected": -1.8052014112472534,
+      "eval_logps/chosen": -45.18332290649414,
+      "eval_logps/rejected": -84.715576171875,
+      "eval_loss": 0.0003367140598129481,
+      "eval_rewards/accuracies": 1.0,
+      "eval_rewards/chosen": 2.9810502529144287,
+      "eval_rewards/margins": 8.570627212524414,
+      "eval_rewards/rejected": -5.589577674865723,
+      "eval_runtime": 548.0227,
+      "eval_samples_per_second": 0.509,
+      "eval_steps_per_second": 0.509,
+      "step": 300
+    },
+    {
+      "epoch": 0.3111111111111111,
+      "grad_norm": 0.23535361886024475,
+      "learning_rate": 5e-07,
+      "logits/chosen": -1.7623131275177002,
+      "logits/rejected": -1.8129371404647827,
+      "logps/chosen": -47.72771072387695,
+      "logps/rejected": -87.96331787109375,
+      "loss": 0.0003,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": 2.999173879623413,
+      "rewards/margins": 8.857995986938477,
+      "rewards/rejected": -5.858822822570801,
+      "step": 350
+    },
+    {
+      "epoch": 0.3111111111111111,
+      "eval_logits/chosen": -1.768768310546875,
+      "eval_logits/rejected": -1.7995201349258423,
+      "eval_logps/chosen": -44.72746276855469,
+      "eval_logps/rejected": -90.03182220458984,
+      "eval_loss": 0.0002303781802766025,
+      "eval_rewards/accuracies": 1.0,
+      "eval_rewards/chosen": 3.0266363620758057,
+      "eval_rewards/margins": 9.14783763885498,
+      "eval_rewards/rejected": -6.121201992034912,
+      "eval_runtime": 538.111,
+      "eval_samples_per_second": 0.518,
+      "eval_steps_per_second": 0.518,
+      "step": 350
+    }
+  ],
+  "logging_steps": 50,
+  "max_steps": 1125,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "save_steps": 50,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": false
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 0.0,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-350/training_args.bin

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