Model save

Browse files

Files changed (13) hide show

README.md +73 -0
adapter_config.json +25 -0
adapter_model.safetensors +3 -0
all_results.json +21 -0
eval_results.json +16 -0
runs/Nov22_09-29-27_401fa5a8015d/events.out.tfevents.1700645508.401fa5a8015d.9788.0 +3 -0
runs/Nov22_09-29-27_401fa5a8015d/events.out.tfevents.1700666050.401fa5a8015d.9788.1 +3 -0
special_tokens_map.json +35 -0
tokenizer.json +0 -0
tokenizer_config.json +45 -0
train_results.json +8 -0
trainer_state.json +1098 -0
training_args.bin +3 -0

README.md ADDED Viewed

	@@ -0,0 +1,73 @@

+---
+license: apache-2.0
+base_model: NousResearch/Yarn-Mistral-7b-128k
+tags:
+- generated_from_trainer
+model-index:
+- name: unraveled-7b-dpo-lora
+  results: []
+---
+<!-- 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. -->
+# unraveled-7b-dpo-lora
+This model is a fine-tuned version of [NousResearch/Yarn-Mistral-7b-128k](https://huggingface.co/NousResearch/Yarn-Mistral-7b-128k) on the None dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.5895
+- Rewards/chosen: 0.1439
+- Rewards/rejected: -0.1833
+- Rewards/accuracies: 0.6880
+- Rewards/margins: 0.3272
+- Logps/rejected: -221.8329
+- Logps/chosen: -266.1414
+- Logits/rejected: -1.9675
+- Logits/chosen: -2.0859
+## Model description
+More information needed
+## Intended uses & limitations
+More information needed
+## Training and evaluation data
+More information needed
+## Training procedure
+### Training hyperparameters
+The following hyperparameters were used during training:
+- learning_rate: 5e-07
+- train_batch_size: 2
+- eval_batch_size: 4
+- seed: 42
+- distributed_type: multi-GPU
+- num_devices: 4
+- gradient_accumulation_steps: 32
+- total_train_batch_size: 256
+- total_eval_batch_size: 16
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- lr_scheduler_warmup_ratio: 0.1
+- num_epochs: 3
+### Training results
+| Training Loss | Epoch | Step | Validation Loss | Rewards/chosen | Rewards/rejected | Rewards/accuracies | Rewards/margins | Logps/rejected | Logps/chosen | Logits/rejected | Logits/chosen |
+|:-------------:|:-----:|:----:|:---------------:|:--------------:|:----------------:|:------------------:|:---------------:|:--------------:|:------------:|:---------------:|:-------------:|
+| 0.6313        | 1.0   | 242  | 0.6318          | 0.1228         | -0.0304          | 0.6600             | 0.1532          | -220.3036      | -266.3521    | -1.9863         | -2.1062       |
+| 0.6013        | 2.0   | 484  | 0.5983          | 0.1484         | -0.1334          | 0.6760             | 0.2819          | -221.3338      | -266.0959    | -1.9723         | -2.0914       |
+| 0.5889        | 3.0   | 726  | 0.5895          | 0.1439         | -0.1833          | 0.6880             | 0.3272          | -221.8329      | -266.1414    | -1.9675         | -2.0859       |
+### Framework versions
+- Transformers 4.35.0
+- Pytorch 2.1.0+cu118
+- Datasets 2.14.6
+- Tokenizers 0.14.1

adapter_config.json ADDED Viewed

	@@ -0,0 +1,25 @@

+{
+  "alpha_pattern": {},
+  "auto_mapping": null,
+  "base_model_name_or_path": "NousResearch/Yarn-Mistral-7b-128k",
+  "bias": "none",
+  "fan_in_fan_out": false,
+  "inference_mode": true,
+  "init_lora_weights": true,
+  "layers_pattern": null,
+  "layers_to_transform": null,
+  "lora_alpha": 16,
+  "lora_dropout": 0.1,
+  "modules_to_save": null,
+  "peft_type": "LORA",
+  "r": 64,
+  "rank_pattern": {},
+  "revision": null,
+  "target_modules": [
+    "o_proj",
+    "q_proj",
+    "k_proj",
+    "v_proj"
+  ],
+  "task_type": "CAUSAL_LM"
+}

adapter_model.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:f8d361ff1dd8577e4bedd4011cfd60a7e4a850379b1c62d1000c06b041e4c858
+size 218138576

all_results.json ADDED Viewed

	@@ -0,0 +1,21 @@

+{
+    "epoch": 3.0,
+    "eval_logits/chosen": -2.085902214050293,
+    "eval_logits/rejected": -1.9675065279006958,
+    "eval_logps/chosen": -266.14141845703125,
+    "eval_logps/rejected": -221.83285522460938,
+    "eval_loss": 0.5895045399665833,
+    "eval_rewards/accuracies": 0.6880000233650208,
+    "eval_rewards/chosen": 0.14388784766197205,
+    "eval_rewards/margins": 0.3272360563278198,
+    "eval_rewards/rejected": -0.1833481788635254,
+    "eval_runtime": 170.0188,
+    "eval_samples": 2000,
+    "eval_samples_per_second": 11.763,
+    "eval_steps_per_second": 0.735,
+    "train_loss": 0.622471270236102,
+    "train_runtime": 20371.8366,
+    "train_samples": 61966,
+    "train_samples_per_second": 9.125,
+    "train_steps_per_second": 0.036
+}

eval_results.json ADDED Viewed

	@@ -0,0 +1,16 @@

+{
+    "epoch": 3.0,
+    "eval_logits/chosen": -2.085902214050293,
+    "eval_logits/rejected": -1.9675065279006958,
+    "eval_logps/chosen": -266.14141845703125,
+    "eval_logps/rejected": -221.83285522460938,
+    "eval_loss": 0.5895045399665833,
+    "eval_rewards/accuracies": 0.6880000233650208,
+    "eval_rewards/chosen": 0.14388784766197205,
+    "eval_rewards/margins": 0.3272360563278198,
+    "eval_rewards/rejected": -0.1833481788635254,
+    "eval_runtime": 170.0188,
+    "eval_samples": 2000,
+    "eval_samples_per_second": 11.763,
+    "eval_steps_per_second": 0.735
+}

runs/Nov22_09-29-27_401fa5a8015d/events.out.tfevents.1700645508.401fa5a8015d.9788.0 ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:85bf84214b62eb5abef7f395c1c7fd693b1b97aee2815078f50035bf105a948f
+size 53448

runs/Nov22_09-29-27_401fa5a8015d/events.out.tfevents.1700666050.401fa5a8015d.9788.1 ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:1cbcea6296bfe57ddf5605e55aef2891478407c1b3e39ef131fdd971da52de35
+size 828

special_tokens_map.json ADDED Viewed

	@@ -0,0 +1,35 @@

+{
+  "additional_special_tokens": [
+    "<unk>",
+    "<s>",
+    "</s>"
+  ],
+  "bos_token": {
+    "content": "<s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer.json ADDED Viewed

The diff for this file is too large to render. See raw diff

tokenizer_config.json ADDED Viewed

	@@ -0,0 +1,45 @@

+{
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "<s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<unk>",
+    "<s>",
+    "</s>"
+  ],
+  "bos_token": "<s>",
+  "chat_template": "{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '<|user|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'system' %}\n{{ '<|system|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'assistant' %}\n{{ '<|assistant|>\n'  + message['content'] + eos_token }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '<|assistant|>' }}\n{% endif %}\n{% endfor %}",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "</s>",
+  "legacy": true,
+  "model_max_length": 2048,
+  "pad_token": "</s>",
+  "sp_model_kwargs": {},
+  "spaces_between_special_tokens": false,
+  "tokenizer_class": "LlamaTokenizer",
+  "unk_token": "<unk>",
+  "use_default_system_prompt": true
+}

train_results.json ADDED Viewed

	@@ -0,0 +1,8 @@

+{
+    "epoch": 3.0,
+    "train_loss": 0.622471270236102,
+    "train_runtime": 20371.8366,
+    "train_samples": 61966,
+    "train_samples_per_second": 9.125,
+    "train_steps_per_second": 0.036
+}

trainer_state.json ADDED Viewed

	@@ -0,0 +1,1098 @@

+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 2.9992254066615027,
+  "eval_steps": 100,
+  "global_step": 726,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0,
+      "learning_rate": 6.84931506849315e-09,
+      "logits/chosen": -2.3569769859313965,
+      "logits/rejected": -2.397932529449463,
+      "logps/chosen": -275.3341369628906,
+      "logps/rejected": -209.60626220703125,
+      "loss": 0.6931,
+      "rewards/accuracies": 0.0,
+      "rewards/chosen": 0.0,
+      "rewards/margins": 0.0,
+      "rewards/rejected": 0.0,
+      "step": 1
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 6.84931506849315e-08,
+      "logits/chosen": -2.4118523597717285,
+      "logits/rejected": -2.3359429836273193,
+      "logps/chosen": -296.8043212890625,
+      "logps/rejected": -226.9486541748047,
+      "loss": 0.6936,
+      "rewards/accuracies": 0.4618055522441864,
+      "rewards/chosen": -0.0005276877782307565,
+      "rewards/margins": -0.0006244900869205594,
+      "rewards/rejected": 9.680193034000695e-05,
+      "step": 10
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.36986301369863e-07,
+      "logits/chosen": -2.4295315742492676,
+      "logits/rejected": -2.379143714904785,
+      "logps/chosen": -281.6752014160156,
+      "logps/rejected": -217.3535919189453,
+      "loss": 0.6938,
+      "rewards/accuracies": 0.4906249940395355,
+      "rewards/chosen": 0.0024631484411656857,
+      "rewards/margins": 5.021132437832421e-06,
+      "rewards/rejected": 0.0024581279139965773,
+      "step": 20
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.054794520547945e-07,
+      "logits/chosen": -2.385784149169922,
+      "logits/rejected": -2.3432505130767822,
+      "logps/chosen": -255.5972442626953,
+      "logps/rejected": -208.0276336669922,
+      "loss": 0.6924,
+      "rewards/accuracies": 0.5406249761581421,
+      "rewards/chosen": 0.0008739754557609558,
+      "rewards/margins": 0.002002383815124631,
+      "rewards/rejected": -0.0011284081265330315,
+      "step": 30
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.73972602739726e-07,
+      "logits/chosen": -2.449314832687378,
+      "logits/rejected": -2.3985402584075928,
+      "logps/chosen": -286.9411926269531,
+      "logps/rejected": -217.51119995117188,
+      "loss": 0.6923,
+      "rewards/accuracies": 0.5218750238418579,
+      "rewards/chosen": 0.0011719572357833385,
+      "rewards/margins": 0.002339401515200734,
+      "rewards/rejected": -0.0011674443958327174,
+      "step": 40
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 3.424657534246575e-07,
+      "logits/chosen": -2.4232876300811768,
+      "logits/rejected": -2.3819985389709473,
+      "logps/chosen": -270.04620361328125,
+      "logps/rejected": -224.44857788085938,
+      "loss": 0.6904,
+      "rewards/accuracies": 0.53125,
+      "rewards/chosen": 0.0045755826868116856,
+      "rewards/margins": 0.007746423594653606,
+      "rewards/rejected": -0.003170841606333852,
+      "step": 50
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 4.10958904109589e-07,
+      "logits/chosen": -2.406489133834839,
+      "logits/rejected": -2.376723289489746,
+      "logps/chosen": -269.7476501464844,
+      "logps/rejected": -216.1673583984375,
+      "loss": 0.6893,
+      "rewards/accuracies": 0.5609375238418579,
+      "rewards/chosen": 0.00798078440129757,
+      "rewards/margins": 0.009231673553586006,
+      "rewards/rejected": -0.0012508893851190805,
+      "step": 60
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 4.794520547945205e-07,
+      "logits/chosen": -2.3951592445373535,
+      "logits/rejected": -2.3819260597229004,
+      "logps/chosen": -257.29620361328125,
+      "logps/rejected": -215.2325897216797,
+      "loss": 0.6874,
+      "rewards/accuracies": 0.5609375238418579,
+      "rewards/chosen": 0.012629570439457893,
+      "rewards/margins": 0.013305542059242725,
+      "rewards/rejected": -0.0006759737734682858,
+      "step": 70
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 4.946401225114854e-07,
+      "logits/chosen": -2.4296867847442627,
+      "logits/rejected": -2.368792772293091,
+      "logps/chosen": -267.8879699707031,
+      "logps/rejected": -219.56350708007812,
+      "loss": 0.6842,
+      "rewards/accuracies": 0.621874988079071,
+      "rewards/chosen": 0.019007008522748947,
+      "rewards/margins": 0.0214321780949831,
+      "rewards/rejected": -0.002425167942419648,
+      "step": 80
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 4.869831546707504e-07,
+      "logits/chosen": -2.463676929473877,
+      "logits/rejected": -2.3972015380859375,
+      "logps/chosen": -274.6397705078125,
+      "logps/rejected": -225.1620635986328,
+      "loss": 0.683,
+      "rewards/accuracies": 0.604687511920929,
+      "rewards/chosen": 0.024952612817287445,
+      "rewards/margins": 0.02081022970378399,
+      "rewards/rejected": 0.004142382647842169,
+      "step": 90
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 4.793261868300153e-07,
+      "logits/chosen": -2.4068779945373535,
+      "logits/rejected": -2.375128746032715,
+      "logps/chosen": -276.1687927246094,
+      "logps/rejected": -228.1005096435547,
+      "loss": 0.6783,
+      "rewards/accuracies": 0.660937488079071,
+      "rewards/chosen": 0.03607472777366638,
+      "rewards/margins": 0.034400396049022675,
+      "rewards/rejected": 0.001674329163506627,
+      "step": 100
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 4.7166921898928023e-07,
+      "logits/chosen": -2.445211887359619,
+      "logits/rejected": -2.3825695514678955,
+      "logps/chosen": -254.85311889648438,
+      "logps/rejected": -222.86117553710938,
+      "loss": 0.6761,
+      "rewards/accuracies": 0.6390625238418579,
+      "rewards/chosen": 0.034680236130952835,
+      "rewards/margins": 0.035091597586870193,
+      "rewards/rejected": -0.00041136034997180104,
+      "step": 110
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 4.640122511485451e-07,
+      "logits/chosen": -2.412264585494995,
+      "logits/rejected": -2.3916800022125244,
+      "logps/chosen": -259.5788879394531,
+      "logps/rejected": -220.30831909179688,
+      "loss": 0.6715,
+      "rewards/accuracies": 0.6421874761581421,
+      "rewards/chosen": 0.04985843971371651,
+      "rewards/margins": 0.050119031220674515,
+      "rewards/rejected": -0.00026059610536322,
+      "step": 120
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 4.563552833078101e-07,
+      "logits/chosen": -2.4411113262176514,
+      "logits/rejected": -2.376892566680908,
+      "logps/chosen": -268.37335205078125,
+      "logps/rejected": -226.0688934326172,
+      "loss": 0.666,
+      "rewards/accuracies": 0.675000011920929,
+      "rewards/chosen": 0.05422825738787651,
+      "rewards/margins": 0.05518989637494087,
+      "rewards/rejected": -0.0009616309544071555,
+      "step": 130
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 4.4869831546707505e-07,
+      "logits/chosen": -2.448073387145996,
+      "logits/rejected": -2.3861663341522217,
+      "logps/chosen": -273.23272705078125,
+      "logps/rejected": -230.17776489257812,
+      "loss": 0.6681,
+      "rewards/accuracies": 0.692187488079071,
+      "rewards/chosen": 0.06362518668174744,
+      "rewards/margins": 0.06625331938266754,
+      "rewards/rejected": -0.002628129906952381,
+      "step": 140
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 4.4104134762633994e-07,
+      "logits/chosen": -2.480207920074463,
+      "logits/rejected": -2.4179327487945557,
+      "logps/chosen": -268.4297790527344,
+      "logps/rejected": -226.82815551757812,
+      "loss": 0.6596,
+      "rewards/accuracies": 0.7250000238418579,
+      "rewards/chosen": 0.07131167501211166,
+      "rewards/margins": 0.0752154216170311,
+      "rewards/rejected": -0.0039037547539919615,
+      "step": 150
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 4.333843797856049e-07,
+      "logits/chosen": -2.438366651535034,
+      "logits/rejected": -2.3843159675598145,
+      "logps/chosen": -281.81756591796875,
+      "logps/rejected": -239.0724334716797,
+      "loss": 0.6555,
+      "rewards/accuracies": 0.7015625238418579,
+      "rewards/chosen": 0.08467759937047958,
+      "rewards/margins": 0.08220230042934418,
+      "rewards/rejected": 0.002475299406796694,
+      "step": 160
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 4.257274119448698e-07,
+      "logits/chosen": -2.4684793949127197,
+      "logits/rejected": -2.433481216430664,
+      "logps/chosen": -272.9942932128906,
+      "logps/rejected": -239.0500946044922,
+      "loss": 0.6542,
+      "rewards/accuracies": 0.6859375238418579,
+      "rewards/chosen": 0.08272445946931839,
+      "rewards/margins": 0.08306626230478287,
+      "rewards/rejected": -0.00034180469810962677,
+      "step": 170
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 4.180704441041347e-07,
+      "logits/chosen": -2.429438829421997,
+      "logits/rejected": -2.378554105758667,
+      "logps/chosen": -259.08294677734375,
+      "logps/rejected": -225.7974853515625,
+      "loss": 0.6565,
+      "rewards/accuracies": 0.6499999761581421,
+      "rewards/chosen": 0.08147875219583511,
+      "rewards/margins": 0.07715155184268951,
+      "rewards/rejected": 0.004327205941081047,
+      "step": 180
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 4.1041347626339966e-07,
+      "logits/chosen": -2.4421346187591553,
+      "logits/rejected": -2.3960843086242676,
+      "logps/chosen": -263.78216552734375,
+      "logps/rejected": -212.2428436279297,
+      "loss": 0.6485,
+      "rewards/accuracies": 0.684374988079071,
+      "rewards/chosen": 0.09069393575191498,
+      "rewards/margins": 0.1026170402765274,
+      "rewards/rejected": -0.011923106387257576,
+      "step": 190
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 4.027565084226646e-07,
+      "logits/chosen": -2.463709831237793,
+      "logits/rejected": -2.404499053955078,
+      "logps/chosen": -264.03118896484375,
+      "logps/rejected": -217.3473358154297,
+      "loss": 0.6415,
+      "rewards/accuracies": 0.7093750238418579,
+      "rewards/chosen": 0.10673630237579346,
+      "rewards/margins": 0.1264755129814148,
+      "rewards/rejected": -0.019739216193556786,
+      "step": 200
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 3.9509954058192954e-07,
+      "logits/chosen": -2.458402633666992,
+      "logits/rejected": -2.4066162109375,
+      "logps/chosen": -286.2770690917969,
+      "logps/rejected": -230.6191864013672,
+      "loss": 0.6401,
+      "rewards/accuracies": 0.6875,
+      "rewards/chosen": 0.10997898876667023,
+      "rewards/margins": 0.12333294004201889,
+      "rewards/rejected": -0.013353955931961536,
+      "step": 210
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 3.874425727411945e-07,
+      "logits/chosen": -2.419983386993408,
+      "logits/rejected": -2.386007308959961,
+      "logps/chosen": -278.9669494628906,
+      "logps/rejected": -227.8882293701172,
+      "loss": 0.6399,
+      "rewards/accuracies": 0.692187488079071,
+      "rewards/chosen": 0.11453185975551605,
+      "rewards/margins": 0.13373538851737976,
+      "rewards/rejected": -0.01920352131128311,
+      "step": 220
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 3.797856049004594e-07,
+      "logits/chosen": -2.4336562156677246,
+      "logits/rejected": -2.36842679977417,
+      "logps/chosen": -266.24053955078125,
+      "logps/rejected": -227.7322235107422,
+      "loss": 0.6365,
+      "rewards/accuracies": 0.6656249761581421,
+      "rewards/chosen": 0.12494595348834991,
+      "rewards/margins": 0.14494453370571136,
+      "rewards/rejected": -0.019998596981167793,
+      "step": 230
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 3.7212863705972436e-07,
+      "logits/chosen": -2.45171856880188,
+      "logits/rejected": -2.4028737545013428,
+      "logps/chosen": -273.0599365234375,
+      "logps/rejected": -221.7194366455078,
+      "loss": 0.6313,
+      "rewards/accuracies": 0.6875,
+      "rewards/chosen": 0.13546457886695862,
+      "rewards/margins": 0.1584644615650177,
+      "rewards/rejected": -0.02299986407160759,
+      "step": 240
+    },
+    {
+      "epoch": 1.0,
+      "eval_logits/chosen": -2.1062474250793457,
+      "eval_logits/rejected": -1.9863277673721313,
+      "eval_logps/chosen": -266.3520812988281,
+      "eval_logps/rejected": -220.30364990234375,
+      "eval_loss": 0.6318375468254089,
+      "eval_rewards/accuracies": 0.6600000262260437,
+      "eval_rewards/chosen": 0.12281632423400879,
+      "eval_rewards/margins": 0.1532445251941681,
+      "eval_rewards/rejected": -0.030428189784288406,
+      "eval_runtime": 170.6685,
+      "eval_samples_per_second": 11.719,
+      "eval_steps_per_second": 0.732,
+      "step": 242
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 3.6447166921898925e-07,
+      "logits/chosen": -2.3959295749664307,
+      "logits/rejected": -2.3386054039001465,
+      "logps/chosen": -259.109619140625,
+      "logps/rejected": -209.0337677001953,
+      "loss": 0.6337,
+      "rewards/accuracies": 0.684374988079071,
+      "rewards/chosen": 0.12544557452201843,
+      "rewards/margins": 0.1500503271818161,
+      "rewards/rejected": -0.024604763835668564,
+      "step": 250
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 3.568147013782542e-07,
+      "logits/chosen": -2.4076590538024902,
+      "logits/rejected": -2.3606529235839844,
+      "logps/chosen": -263.9520568847656,
+      "logps/rejected": -225.4473876953125,
+      "loss": 0.6258,
+      "rewards/accuracies": 0.721875011920929,
+      "rewards/chosen": 0.1394219994544983,
+      "rewards/margins": 0.1773197054862976,
+      "rewards/rejected": -0.03789771348237991,
+      "step": 260
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 3.4915773353751913e-07,
+      "logits/chosen": -2.417232036590576,
+      "logits/rejected": -2.346893072128296,
+      "logps/chosen": -280.55291748046875,
+      "logps/rejected": -234.11996459960938,
+      "loss": 0.6203,
+      "rewards/accuracies": 0.729687511920929,
+      "rewards/chosen": 0.1420687586069107,
+      "rewards/margins": 0.20756664872169495,
+      "rewards/rejected": -0.06549788266420364,
+      "step": 270
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 3.41500765696784e-07,
+      "logits/chosen": -2.4278924465179443,
+      "logits/rejected": -2.3850674629211426,
+      "logps/chosen": -253.51803588867188,
+      "logps/rejected": -224.95968627929688,
+      "loss": 0.6242,
+      "rewards/accuracies": 0.6968749761581421,
+      "rewards/chosen": 0.13101010024547577,
+      "rewards/margins": 0.16972467303276062,
+      "rewards/rejected": -0.03871458023786545,
+      "step": 280
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 3.33843797856049e-07,
+      "logits/chosen": -2.3946681022644043,
+      "logits/rejected": -2.380558967590332,
+      "logps/chosen": -284.79608154296875,
+      "logps/rejected": -226.66262817382812,
+      "loss": 0.6194,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": 0.13935603201389313,
+      "rewards/margins": 0.20914654433727264,
+      "rewards/rejected": -0.06979051232337952,
+      "step": 290
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 3.2618683001531396e-07,
+      "logits/chosen": -2.3978686332702637,
+      "logits/rejected": -2.3249762058258057,
+      "logps/chosen": -263.2580871582031,
+      "logps/rejected": -226.63101196289062,
+      "loss": 0.6169,
+      "rewards/accuracies": 0.706250011920929,
+      "rewards/chosen": 0.14139556884765625,
+      "rewards/margins": 0.1872793436050415,
+      "rewards/rejected": -0.04588378220796585,
+      "step": 300
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 3.1852986217457885e-07,
+      "logits/chosen": -2.4475905895233154,
+      "logits/rejected": -2.363696575164795,
+      "logps/chosen": -262.8795471191406,
+      "logps/rejected": -227.06930541992188,
+      "loss": 0.619,
+      "rewards/accuracies": 0.7203124761581421,
+      "rewards/chosen": 0.15051104128360748,
+      "rewards/margins": 0.1968255192041397,
+      "rewards/rejected": -0.04631447046995163,
+      "step": 310
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 3.108728943338438e-07,
+      "logits/chosen": -2.438253164291382,
+      "logits/rejected": -2.421774387359619,
+      "logps/chosen": -253.95639038085938,
+      "logps/rejected": -236.4005584716797,
+      "loss": 0.6208,
+      "rewards/accuracies": 0.65625,
+      "rewards/chosen": 0.13959024846553802,
+      "rewards/margins": 0.1706792563199997,
+      "rewards/rejected": -0.031088998541235924,
+      "step": 320
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 3.0321592649310873e-07,
+      "logits/chosen": -2.4183051586151123,
+      "logits/rejected": -2.339179039001465,
+      "logps/chosen": -254.692626953125,
+      "logps/rejected": -222.8722381591797,
+      "loss": 0.6141,
+      "rewards/accuracies": 0.699999988079071,
+      "rewards/chosen": 0.1566968411207199,
+      "rewards/margins": 0.19593168795108795,
+      "rewards/rejected": -0.03923482820391655,
+      "step": 330
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.955589586523736e-07,
+      "logits/chosen": -2.4377613067626953,
+      "logits/rejected": -2.366072177886963,
+      "logps/chosen": -267.422607421875,
+      "logps/rejected": -224.8284149169922,
+      "loss": 0.6188,
+      "rewards/accuracies": 0.6890624761581421,
+      "rewards/chosen": 0.14487192034721375,
+      "rewards/margins": 0.2071552723646164,
+      "rewards/rejected": -0.06228336691856384,
+      "step": 340
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.8790199081163856e-07,
+      "logits/chosen": -2.435045003890991,
+      "logits/rejected": -2.3581719398498535,
+      "logps/chosen": -267.0567626953125,
+      "logps/rejected": -223.5650177001953,
+      "loss": 0.6136,
+      "rewards/accuracies": 0.6890624761581421,
+      "rewards/chosen": 0.1428806185722351,
+      "rewards/margins": 0.20400968194007874,
+      "rewards/rejected": -0.06112906336784363,
+      "step": 350
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.802450229709035e-07,
+      "logits/chosen": -2.3934175968170166,
+      "logits/rejected": -2.3315463066101074,
+      "logps/chosen": -256.1235656738281,
+      "logps/rejected": -211.8139190673828,
+      "loss": 0.6093,
+      "rewards/accuracies": 0.6937500238418579,
+      "rewards/chosen": 0.14465923607349396,
+      "rewards/margins": 0.20695683360099792,
+      "rewards/rejected": -0.06229761987924576,
+      "step": 360
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.725880551301684e-07,
+      "logits/chosen": -2.4487271308898926,
+      "logits/rejected": -2.3933699131011963,
+      "logps/chosen": -275.8642578125,
+      "logps/rejected": -231.2057342529297,
+      "loss": 0.6129,
+      "rewards/accuracies": 0.692187488079071,
+      "rewards/chosen": 0.14190760254859924,
+      "rewards/margins": 0.22471928596496582,
+      "rewards/rejected": -0.08281168341636658,
+      "step": 370
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.649310872894334e-07,
+      "logits/chosen": -2.4377894401550293,
+      "logits/rejected": -2.406348466873169,
+      "logps/chosen": -276.6672058105469,
+      "logps/rejected": -221.1224365234375,
+      "loss": 0.603,
+      "rewards/accuracies": 0.7093750238418579,
+      "rewards/chosen": 0.15706932544708252,
+      "rewards/margins": 0.24053001403808594,
+      "rewards/rejected": -0.08346069604158401,
+      "step": 380
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.572741194486983e-07,
+      "logits/chosen": -2.4386019706726074,
+      "logits/rejected": -2.3754172325134277,
+      "logps/chosen": -274.6893005371094,
+      "logps/rejected": -206.1323699951172,
+      "loss": 0.6004,
+      "rewards/accuracies": 0.715624988079071,
+      "rewards/chosen": 0.167318657040596,
+      "rewards/margins": 0.25797349214553833,
+      "rewards/rejected": -0.09065485745668411,
+      "step": 390
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.496171516079632e-07,
+      "logits/chosen": -2.4488420486450195,
+      "logits/rejected": -2.3803772926330566,
+      "logps/chosen": -294.0111389160156,
+      "logps/rejected": -237.24282836914062,
+      "loss": 0.6033,
+      "rewards/accuracies": 0.714062511920929,
+      "rewards/chosen": 0.17194847762584686,
+      "rewards/margins": 0.2693827748298645,
+      "rewards/rejected": -0.09743430465459824,
+      "step": 400
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.4196018376722816e-07,
+      "logits/chosen": -2.413655996322632,
+      "logits/rejected": -2.38279390335083,
+      "logps/chosen": -279.74090576171875,
+      "logps/rejected": -226.3095245361328,
+      "loss": 0.6081,
+      "rewards/accuracies": 0.6953125,
+      "rewards/chosen": 0.16821043193340302,
+      "rewards/margins": 0.26299187541007996,
+      "rewards/rejected": -0.09478144347667694,
+      "step": 410
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.343032159264931e-07,
+      "logits/chosen": -2.390028476715088,
+      "logits/rejected": -2.3162546157836914,
+      "logps/chosen": -270.34320068359375,
+      "logps/rejected": -221.6541290283203,
+      "loss": 0.6019,
+      "rewards/accuracies": 0.721875011920929,
+      "rewards/chosen": 0.17296920716762543,
+      "rewards/margins": 0.2894430458545685,
+      "rewards/rejected": -0.11647380888462067,
+      "step": 420
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.26646248085758e-07,
+      "logits/chosen": -2.3680365085601807,
+      "logits/rejected": -2.3160338401794434,
+      "logps/chosen": -260.14593505859375,
+      "logps/rejected": -214.6702423095703,
+      "loss": 0.5951,
+      "rewards/accuracies": 0.721875011920929,
+      "rewards/chosen": 0.16743981838226318,
+      "rewards/margins": 0.2915259897708893,
+      "rewards/rejected": -0.1240861564874649,
+      "step": 430
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.1898928024502298e-07,
+      "logits/chosen": -2.443164825439453,
+      "logits/rejected": -2.3816096782684326,
+      "logps/chosen": -264.55914306640625,
+      "logps/rejected": -219.47341918945312,
+      "loss": 0.6012,
+      "rewards/accuracies": 0.675000011920929,
+      "rewards/chosen": 0.15291035175323486,
+      "rewards/margins": 0.2545214593410492,
+      "rewards/rejected": -0.10161112248897552,
+      "step": 440
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.113323124042879e-07,
+      "logits/chosen": -2.416532039642334,
+      "logits/rejected": -2.3757832050323486,
+      "logps/chosen": -272.0744934082031,
+      "logps/rejected": -228.9375,
+      "loss": 0.5995,
+      "rewards/accuracies": 0.698437511920929,
+      "rewards/chosen": 0.17259187996387482,
+      "rewards/margins": 0.2831020951271057,
+      "rewards/rejected": -0.1105102151632309,
+      "step": 450
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.036753445635528e-07,
+      "logits/chosen": -2.4263906478881836,
+      "logits/rejected": -2.4025325775146484,
+      "logps/chosen": -284.97576904296875,
+      "logps/rejected": -235.3527069091797,
+      "loss": 0.5904,
+      "rewards/accuracies": 0.706250011920929,
+      "rewards/chosen": 0.16448193788528442,
+      "rewards/margins": 0.32010418176651,
+      "rewards/rejected": -0.1556222140789032,
+      "step": 460
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 1.9601837672281775e-07,
+      "logits/chosen": -2.402890920639038,
+      "logits/rejected": -2.3698153495788574,
+      "logps/chosen": -271.0093688964844,
+      "logps/rejected": -234.02481079101562,
+      "loss": 0.5976,
+      "rewards/accuracies": 0.7250000238418579,
+      "rewards/chosen": 0.14684638381004333,
+      "rewards/margins": 0.2707447409629822,
+      "rewards/rejected": -0.12389836460351944,
+      "step": 470
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 1.883614088820827e-07,
+      "logits/chosen": -2.425776958465576,
+      "logits/rejected": -2.371450901031494,
+      "logps/chosen": -269.1894226074219,
+      "logps/rejected": -222.6921844482422,
+      "loss": 0.6013,
+      "rewards/accuracies": 0.6968749761581421,
+      "rewards/chosen": 0.1555326133966446,
+      "rewards/margins": 0.28726813197135925,
+      "rewards/rejected": -0.13173556327819824,
+      "step": 480
+    },
+    {
+      "epoch": 2.0,
+      "eval_logits/chosen": -2.09140682220459,
+      "eval_logits/rejected": -1.9723255634307861,
+      "eval_logps/chosen": -266.0958557128906,
+      "eval_logps/rejected": -221.3338165283203,
+      "eval_loss": 0.598257839679718,
+      "eval_rewards/accuracies": 0.6759999990463257,
+      "eval_rewards/chosen": 0.14844101667404175,
+      "eval_rewards/margins": 0.28188496828079224,
+      "eval_rewards/rejected": -0.1334439367055893,
+      "eval_runtime": 171.7034,
+      "eval_samples_per_second": 11.648,
+      "eval_steps_per_second": 0.728,
+      "step": 484
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 1.807044410413476e-07,
+      "logits/chosen": -2.3953592777252197,
+      "logits/rejected": -2.363999843597412,
+      "logps/chosen": -258.8542175292969,
+      "logps/rejected": -233.6983642578125,
+      "loss": 0.5992,
+      "rewards/accuracies": 0.7171875238418579,
+      "rewards/chosen": 0.1408676654100418,
+      "rewards/margins": 0.27908438444137573,
+      "rewards/rejected": -0.13821670413017273,
+      "step": 490
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 1.7304747320061255e-07,
+      "logits/chosen": -2.3902556896209717,
+      "logits/rejected": -2.3884072303771973,
+      "logps/chosen": -265.0683288574219,
+      "logps/rejected": -228.423828125,
+      "loss": 0.5955,
+      "rewards/accuracies": 0.684374988079071,
+      "rewards/chosen": 0.17755301296710968,
+      "rewards/margins": 0.3005513548851013,
+      "rewards/rejected": -0.12299831956624985,
+      "step": 500
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 1.6539050535987747e-07,
+      "logits/chosen": -2.473872661590576,
+      "logits/rejected": -2.3617444038391113,
+      "logps/chosen": -270.9250793457031,
+      "logps/rejected": -222.18264770507812,
+      "loss": 0.5912,
+      "rewards/accuracies": 0.714062511920929,
+      "rewards/chosen": 0.16197429597377777,
+      "rewards/margins": 0.3233007788658142,
+      "rewards/rejected": -0.16132643818855286,
+      "step": 510
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 1.5773353751914243e-07,
+      "logits/chosen": -2.4702870845794678,
+      "logits/rejected": -2.4053397178649902,
+      "logps/chosen": -285.6563720703125,
+      "logps/rejected": -227.668212890625,
+      "loss": 0.5944,
+      "rewards/accuracies": 0.7046874761581421,
+      "rewards/chosen": 0.1673307716846466,
+      "rewards/margins": 0.3238461911678314,
+      "rewards/rejected": -0.15651538968086243,
+      "step": 520
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 1.5007656967840735e-07,
+      "logits/chosen": -2.381838798522949,
+      "logits/rejected": -2.3299927711486816,
+      "logps/chosen": -254.9144287109375,
+      "logps/rejected": -228.2997589111328,
+      "loss": 0.5935,
+      "rewards/accuracies": 0.721875011920929,
+      "rewards/chosen": 0.16740316152572632,
+      "rewards/margins": 0.29852622747421265,
+      "rewards/rejected": -0.13112305104732513,
+      "step": 530
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 1.4241960183767226e-07,
+      "logits/chosen": -2.4169716835021973,
+      "logits/rejected": -2.3592681884765625,
+      "logps/chosen": -283.3446350097656,
+      "logps/rejected": -233.24679565429688,
+      "loss": 0.5918,
+      "rewards/accuracies": 0.684374988079071,
+      "rewards/chosen": 0.1610698103904724,
+      "rewards/margins": 0.3080851435661316,
+      "rewards/rejected": -0.1470153033733368,
+      "step": 540
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 1.347626339969372e-07,
+      "logits/chosen": -2.408930540084839,
+      "logits/rejected": -2.37349009513855,
+      "logps/chosen": -271.91973876953125,
+      "logps/rejected": -240.18679809570312,
+      "loss": 0.5949,
+      "rewards/accuracies": 0.692187488079071,
+      "rewards/chosen": 0.17847296595573425,
+      "rewards/margins": 0.29651308059692383,
+      "rewards/rejected": -0.118040069937706,
+      "step": 550
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 1.2710566615620215e-07,
+      "logits/chosen": -2.367671489715576,
+      "logits/rejected": -2.320443868637085,
+      "logps/chosen": -275.58294677734375,
+      "logps/rejected": -222.78842163085938,
+      "loss": 0.5862,
+      "rewards/accuracies": 0.7109375,
+      "rewards/chosen": 0.17732994258403778,
+      "rewards/margins": 0.3325561583042145,
+      "rewards/rejected": -0.1552262306213379,
+      "step": 560
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 1.1944869831546706e-07,
+      "logits/chosen": -2.388644218444824,
+      "logits/rejected": -2.3782949447631836,
+      "logps/chosen": -268.08172607421875,
+      "logps/rejected": -236.77163696289062,
+      "loss": 0.5938,
+      "rewards/accuracies": 0.668749988079071,
+      "rewards/chosen": 0.1551700383424759,
+      "rewards/margins": 0.3083476424217224,
+      "rewards/rejected": -0.15317757427692413,
+      "step": 570
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 1.11791730474732e-07,
+      "logits/chosen": -2.398094654083252,
+      "logits/rejected": -2.3552727699279785,
+      "logps/chosen": -263.4698181152344,
+      "logps/rejected": -223.05538940429688,
+      "loss": 0.5874,
+      "rewards/accuracies": 0.715624988079071,
+      "rewards/chosen": 0.16049396991729736,
+      "rewards/margins": 0.35638219118118286,
+      "rewards/rejected": -0.1958882361650467,
+      "step": 580
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 1.0413476263399694e-07,
+      "logits/chosen": -2.4469425678253174,
+      "logits/rejected": -2.392503261566162,
+      "logps/chosen": -270.57635498046875,
+      "logps/rejected": -216.8152618408203,
+      "loss": 0.5878,
+      "rewards/accuracies": 0.7109375,
+      "rewards/chosen": 0.15162338316440582,
+      "rewards/margins": 0.3167043924331665,
+      "rewards/rejected": -0.1650809943675995,
+      "step": 590
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 9.647779479326186e-08,
+      "logits/chosen": -2.3848066329956055,
+      "logits/rejected": -2.3738114833831787,
+      "logps/chosen": -251.4600372314453,
+      "logps/rejected": -222.6741180419922,
+      "loss": 0.5932,
+      "rewards/accuracies": 0.690625011920929,
+      "rewards/chosen": 0.16963472962379456,
+      "rewards/margins": 0.2706124782562256,
+      "rewards/rejected": -0.10097774118185043,
+      "step": 600
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 8.88208269525268e-08,
+      "logits/chosen": -2.377875804901123,
+      "logits/rejected": -2.374070644378662,
+      "logps/chosen": -261.51788330078125,
+      "logps/rejected": -221.13034057617188,
+      "loss": 0.5909,
+      "rewards/accuracies": 0.690625011920929,
+      "rewards/chosen": 0.1429089605808258,
+      "rewards/margins": 0.29002851247787476,
+      "rewards/rejected": -0.14711955189704895,
+      "step": 610
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 8.116385911179173e-08,
+      "logits/chosen": -2.3656888008117676,
+      "logits/rejected": -2.3523547649383545,
+      "logps/chosen": -281.01617431640625,
+      "logps/rejected": -217.3480682373047,
+      "loss": 0.5924,
+      "rewards/accuracies": 0.7109375,
+      "rewards/chosen": 0.17781397700309753,
+      "rewards/margins": 0.3453444540500641,
+      "rewards/rejected": -0.16753047704696655,
+      "step": 620
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 7.350689127105667e-08,
+      "logits/chosen": -2.4372496604919434,
+      "logits/rejected": -2.3591084480285645,
+      "logps/chosen": -273.20965576171875,
+      "logps/rejected": -229.5889434814453,
+      "loss": 0.5833,
+      "rewards/accuracies": 0.6890624761581421,
+      "rewards/chosen": 0.17219647765159607,
+      "rewards/margins": 0.35740959644317627,
+      "rewards/rejected": -0.185213103890419,
+      "step": 630
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 6.584992343032159e-08,
+      "logits/chosen": -2.351811647415161,
+      "logits/rejected": -2.3350961208343506,
+      "logps/chosen": -263.6435241699219,
+      "logps/rejected": -226.4728240966797,
+      "loss": 0.5973,
+      "rewards/accuracies": 0.682812511920929,
+      "rewards/chosen": 0.15416522324085236,
+      "rewards/margins": 0.29506421089172363,
+      "rewards/rejected": -0.14089898765087128,
+      "step": 640
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 5.819295558958652e-08,
+      "logits/chosen": -2.390491008758545,
+      "logits/rejected": -2.3410990238189697,
+      "logps/chosen": -289.5431823730469,
+      "logps/rejected": -219.86788940429688,
+      "loss": 0.5824,
+      "rewards/accuracies": 0.7328125238418579,
+      "rewards/chosen": 0.18201851844787598,
+      "rewards/margins": 0.38538652658462524,
+      "rewards/rejected": -0.20336803793907166,
+      "step": 650
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 5.0535987748851455e-08,
+      "logits/chosen": -2.449439525604248,
+      "logits/rejected": -2.3970067501068115,
+      "logps/chosen": -276.04620361328125,
+      "logps/rejected": -229.76620483398438,
+      "loss": 0.5751,
+      "rewards/accuracies": 0.7093750238418579,
+      "rewards/chosen": 0.15049387514591217,
+      "rewards/margins": 0.32289570569992065,
+      "rewards/rejected": -0.17240183055400848,
+      "step": 660
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 4.287901990811638e-08,
+      "logits/chosen": -2.436511993408203,
+      "logits/rejected": -2.341296672821045,
+      "logps/chosen": -269.1939392089844,
+      "logps/rejected": -242.32968139648438,
+      "loss": 0.583,
+      "rewards/accuracies": 0.699999988079071,
+      "rewards/chosen": 0.16247320175170898,
+      "rewards/margins": 0.3472925126552582,
+      "rewards/rejected": -0.1848193258047104,
+      "step": 670
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 3.522205206738132e-08,
+      "logits/chosen": -2.3986496925354004,
+      "logits/rejected": -2.3604884147644043,
+      "logps/chosen": -276.03436279296875,
+      "logps/rejected": -227.72573852539062,
+      "loss": 0.5863,
+      "rewards/accuracies": 0.7124999761581421,
+      "rewards/chosen": 0.16009405255317688,
+      "rewards/margins": 0.34014397859573364,
+      "rewards/rejected": -0.18004995584487915,
+      "step": 680
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.7565084226646246e-08,
+      "logits/chosen": -2.375347137451172,
+      "logits/rejected": -2.3558340072631836,
+      "logps/chosen": -265.17486572265625,
+      "logps/rejected": -234.3466796875,
+      "loss": 0.5896,
+      "rewards/accuracies": 0.721875011920929,
+      "rewards/chosen": 0.16287991404533386,
+      "rewards/margins": 0.35990676283836365,
+      "rewards/rejected": -0.19702686369419098,
+      "step": 690
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 1.9908116385911178e-08,
+      "logits/chosen": -2.4065492153167725,
+      "logits/rejected": -2.36779522895813,
+      "logps/chosen": -261.1010437011719,
+      "logps/rejected": -224.9178009033203,
+      "loss": 0.5892,
+      "rewards/accuracies": 0.6546875238418579,
+      "rewards/chosen": 0.15390679240226746,
+      "rewards/margins": 0.2883544862270355,
+      "rewards/rejected": -0.13444769382476807,
+      "step": 700
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 1.225114854517611e-08,
+      "logits/chosen": -2.433964252471924,
+      "logits/rejected": -2.3655142784118652,
+      "logps/chosen": -281.37139892578125,
+      "logps/rejected": -219.88296508789062,
+      "loss": 0.5838,
+      "rewards/accuracies": 0.7093750238418579,
+      "rewards/chosen": 0.1487404853105545,
+      "rewards/margins": 0.35009315609931946,
+      "rewards/rejected": -0.20135268568992615,
+      "step": 710
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 4.594180704441042e-09,
+      "logits/chosen": -2.3966562747955322,
+      "logits/rejected": -2.368652820587158,
+      "logps/chosen": -258.764892578125,
+      "logps/rejected": -223.00405883789062,
+      "loss": 0.5889,
+      "rewards/accuracies": 0.707812488079071,
+      "rewards/chosen": 0.14300301671028137,
+      "rewards/margins": 0.3152904212474823,
+      "rewards/rejected": -0.17228738963603973,
+      "step": 720
+    },
+    {
+      "epoch": 3.0,
+      "eval_logits/chosen": -2.085902214050293,
+      "eval_logits/rejected": -1.9675065279006958,
+      "eval_logps/chosen": -266.14141845703125,
+      "eval_logps/rejected": -221.83285522460938,
+      "eval_loss": 0.5895045399665833,
+      "eval_rewards/accuracies": 0.6880000233650208,
+      "eval_rewards/chosen": 0.14388784766197205,
+      "eval_rewards/margins": 0.3272360563278198,
+      "eval_rewards/rejected": -0.1833481788635254,
+      "eval_runtime": 170.0,
+      "eval_samples_per_second": 11.765,
+      "eval_steps_per_second": 0.735,
+      "step": 726
+    },
+    {
+      "epoch": 3.0,
+      "step": 726,
+      "total_flos": 0.0,
+      "train_loss": 0.622471270236102,
+      "train_runtime": 20371.8366,
+      "train_samples_per_second": 9.125,
+      "train_steps_per_second": 0.036
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 726,
+  "num_train_epochs": 3,
+  "save_steps": 500,
+  "total_flos": 0.0,
+  "trial_name": null,
+  "trial_params": null
+}

training_args.bin ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:3755299ac3d7fee80db1f1444dda220c2fca58cd01a89e4b8cf67c82f95bb3d6
+size 4728