Model save

README.md

@@ -0,0 +1,82 @@
+---
+license: apache-2.0
+library_name: peft
+tags:
+- trl-dpo
+- generated_from_trainer
+base_model: mistralai/Mistral-7B-v0.1
+model-index:
+- name: zephyr-7b-ipo-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. -->
+
+# zephyr-7b-ipo-lora
+
+This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on the None dataset.
+It achieves the following results on the evaluation set:
+- Loss: 18.0888
+- Rewards/chosen: 0.0201
+- Rewards/rejected: -0.1194
+- Rewards/accuracies: 0.7320
+- Rewards/margins: 0.1395
+- Logps/rejected: -212.8609
+- Logps/chosen: -268.5415
+- Logits/rejected: -2.3003
+- Logits/chosen: -2.6873
+
+## 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: 1
+- eval_batch_size: 1
+- seed: 42
+- distributed_type: multi-GPU
+- num_devices: 8
+- gradient_accumulation_steps: 32
+- total_train_batch_size: 256
+- total_eval_batch_size: 8
+- 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 |
+|:-------------:|:-----:|:----:|:---------------:|:--------------:|:----------------:|:------------------:|:---------------:|:--------------:|:------------:|:---------------:|:-------------:|
+| 19.1303       | 1.0   | 242  | 18.9124         | 0.0302         | -0.0939          | 0.7360             | 0.1241          | -212.6061      | -268.4410    | -2.3029         | -2.6884       |
+| 19.0824       | 2.0   | 484  | 18.2266         | 0.0186         | -0.1163          | 0.7600             | 0.1349          | -212.8297      | -268.5567    | -2.3011         | -2.6877       |
+| 18.0366       | 3.0   | 726  | 18.0888         | 0.0201         | -0.1194          | 0.7320             | 0.1395          | -212.8609      | -268.5415    | -2.3003         | -2.6873       |
+
+
+### Framework versions
+
+- Transformers 4.36.1
+- Pytorch 2.1.2+cu121
+- Datasets 2.14.6
+- Tokenizers 0.15.0
+## Training procedure
+
+
+### Framework versions
+
+
+- PEFT 0.6.1

adapter_config.json

@@ -0,0 +1,25 @@
+{
+  "alpha_pattern": {},
+  "auto_mapping": null,
+  "base_model_name_or_path": "mistralai/Mistral-7B-v0.1",
+  "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": [
+    "k_proj",
+    "o_proj",
+    "q_proj",
+    "v_proj"
+  ],
+  "task_type": "CAUSAL_LM"
+}

adapter_model.safetensors

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

all_results.json

@@ -0,0 +1,21 @@
+{
+    "epoch": 3.0,
+    "eval_logits/chosen": -2.687349796295166,
+    "eval_logits/rejected": -2.300260305404663,
+    "eval_logps/chosen": -268.54150390625,
+    "eval_logps/rejected": -212.86087036132812,
+    "eval_loss": 18.088815689086914,
+    "eval_rewards/accuracies": 0.7319999933242798,
+    "eval_rewards/chosen": 0.02013408951461315,
+    "eval_rewards/margins": 0.13951165974140167,
+    "eval_rewards/rejected": -0.11937756091356277,
+    "eval_runtime": 350.1967,
+    "eval_samples": 2000,
+    "eval_samples_per_second": 5.711,
+    "eval_steps_per_second": 0.714,
+    "train_loss": 19.467516181882747,
+    "train_runtime": 42806.6761,
+    "train_samples": 61966,
+    "train_samples_per_second": 4.343,
+    "train_steps_per_second": 0.017
+}

eval_results.json

@@ -0,0 +1,16 @@
+{
+    "epoch": 3.0,
+    "eval_logits/chosen": -2.687349796295166,
+    "eval_logits/rejected": -2.300260305404663,
+    "eval_logps/chosen": -268.54150390625,
+    "eval_logps/rejected": -212.86087036132812,
+    "eval_loss": 18.088815689086914,
+    "eval_rewards/accuracies": 0.7319999933242798,
+    "eval_rewards/chosen": 0.02013408951461315,
+    "eval_rewards/margins": 0.13951165974140167,
+    "eval_rewards/rejected": -0.11937756091356277,
+    "eval_runtime": 350.1967,
+    "eval_samples": 2000,
+    "eval_samples_per_second": 5.711,
+    "eval_steps_per_second": 0.714
+}

runs/Dec23_06-36-03_ip-172-16-86-203.ec2.internal/events.out.tfevents.1703313736.ip-172-16-86-203.ec2.internal.47042.0

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

runs/Dec23_06-36-03_ip-172-16-86-203.ec2.internal/events.out.tfevents.1703356892.ip-172-16-86-203.ec2.internal.47042.1

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

special_tokens_map.json

@@ -0,0 +1,30 @@
+{
+  "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_config.json

@@ -0,0 +1,43 @@
+{
+  "add_bos_token": true,
+  "add_eos_token": false,
+  "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": [],
+  "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

@@ -0,0 +1,8 @@
+{
+    "epoch": 3.0,
+    "train_loss": 19.467516181882747,
+    "train_runtime": 42806.6761,
+    "train_samples": 61966,
+    "train_samples_per_second": 4.343,
+    "train_steps_per_second": 0.017
+}

trainer_state.json

@@ -0,0 +1,1100 @@
+{
+  "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.55204439163208,
+      "logits/rejected": -2.328748941421509,
+      "logps/chosen": -243.65313720703125,
+      "logps/rejected": -203.76687622070312,
+      "loss": 25.0,
+      "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.6284379959106445,
+      "logits/rejected": -2.217939853668213,
+      "logps/chosen": -283.397705078125,
+      "logps/rejected": -223.02455139160156,
+      "loss": 25.2356,
+      "rewards/accuracies": 0.4652777910232544,
+      "rewards/chosen": 0.000989701016806066,
+      "rewards/margins": 0.0014238969888538122,
+      "rewards/rejected": -0.0004341956228017807,
+      "step": 10
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 1.36986301369863e-07,
+      "logits/chosen": -2.6449296474456787,
+      "logits/rejected": -2.2949368953704834,
+      "logps/chosen": -276.48468017578125,
+      "logps/rejected": -204.2726593017578,
+      "loss": 25.4323,
+      "rewards/accuracies": 0.5093749761581421,
+      "rewards/chosen": 0.0012504293117672205,
+      "rewards/margins": 0.0009743297705426812,
+      "rewards/rejected": 0.0002760997449513525,
+      "step": 20
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 2.054794520547945e-07,
+      "logits/chosen": -2.61030912399292,
+      "logits/rejected": -2.28235125541687,
+      "logps/chosen": -242.44509887695312,
+      "logps/rejected": -195.07154846191406,
+      "loss": 25.3196,
+      "rewards/accuracies": 0.5093749761581421,
+      "rewards/chosen": -0.0013605057029053569,
+      "rewards/margins": 0.0007983079412952065,
+      "rewards/rejected": -0.0021588136442005634,
+      "step": 30
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 2.73972602739726e-07,
+      "logits/chosen": -2.676435947418213,
+      "logits/rejected": -2.2325174808502197,
+      "logps/chosen": -276.4580383300781,
+      "logps/rejected": -195.8528289794922,
+      "loss": 24.8252,
+      "rewards/accuracies": 0.546875,
+      "rewards/chosen": 0.004295675549656153,
+      "rewards/margins": 0.006117778830230236,
+      "rewards/rejected": -0.001822102814912796,
+      "step": 40
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 3.424657534246575e-07,
+      "logits/chosen": -2.5881507396698,
+      "logits/rejected": -2.340297222137451,
+      "logps/chosen": -254.146484375,
+      "logps/rejected": -227.5242919921875,
+      "loss": 24.8309,
+      "rewards/accuracies": 0.484375,
+      "rewards/chosen": 0.0033471225760877132,
+      "rewards/margins": -0.0003331190091557801,
+      "rewards/rejected": 0.003680241061374545,
+      "step": 50
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 4.10958904109589e-07,
+      "logits/chosen": -2.6254591941833496,
+      "logits/rejected": -2.264179229736328,
+      "logps/chosen": -268.96685791015625,
+      "logps/rejected": -212.42770385742188,
+      "loss": 23.9609,
+      "rewards/accuracies": 0.6156250238418579,
+      "rewards/chosen": 0.015944072976708412,
+      "rewards/margins": 0.016598206013441086,
+      "rewards/rejected": -0.0006541347247548401,
+      "step": 60
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 4.794520547945205e-07,
+      "logits/chosen": -2.596461534500122,
+      "logits/rejected": -2.2788121700286865,
+      "logps/chosen": -244.193115234375,
+      "logps/rejected": -198.00149536132812,
+      "loss": 23.4551,
+      "rewards/accuracies": 0.550000011920929,
+      "rewards/chosen": 0.01318939495831728,
+      "rewards/margins": 0.014477437362074852,
+      "rewards/rejected": -0.0012880431022495031,
+      "step": 70
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 4.946401225114854e-07,
+      "logits/chosen": -2.6231789588928223,
+      "logits/rejected": -2.248992443084717,
+      "logps/chosen": -256.1118469238281,
+      "logps/rejected": -212.2821502685547,
+      "loss": 22.8216,
+      "rewards/accuracies": 0.6625000238418579,
+      "rewards/chosen": 0.021900424733757973,
+      "rewards/margins": 0.034875936806201935,
+      "rewards/rejected": -0.012975512072443962,
+      "step": 80
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 4.869831546707504e-07,
+      "logits/chosen": -2.6711127758026123,
+      "logits/rejected": -2.2961134910583496,
+      "logps/chosen": -265.5664978027344,
+      "logps/rejected": -222.6741180419922,
+      "loss": 22.2504,
+      "rewards/accuracies": 0.621874988079071,
+      "rewards/chosen": 0.037670500576496124,
+      "rewards/margins": 0.04158278554677963,
+      "rewards/rejected": -0.003912287298589945,
+      "step": 90
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 4.793261868300153e-07,
+      "logits/chosen": -2.6314404010772705,
+      "logits/rejected": -2.367823362350464,
+      "logps/chosen": -272.08331298828125,
+      "logps/rejected": -240.5725555419922,
+      "loss": 21.7851,
+      "rewards/accuracies": 0.675000011920929,
+      "rewards/chosen": 0.04255244508385658,
+      "rewards/margins": 0.05080147832632065,
+      "rewards/rejected": -0.00824903603643179,
+      "step": 100
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 4.7166921898928023e-07,
+      "logits/chosen": -2.665757417678833,
+      "logits/rejected": -2.239075183868408,
+      "logps/chosen": -236.4951934814453,
+      "logps/rejected": -197.68630981445312,
+      "loss": 21.284,
+      "rewards/accuracies": 0.715624988079071,
+      "rewards/chosen": 0.03496263176202774,
+      "rewards/margins": 0.06847874075174332,
+      "rewards/rejected": -0.03351610153913498,
+      "step": 110
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 4.640122511485451e-07,
+      "logits/chosen": -2.61603045463562,
+      "logits/rejected": -2.319265127182007,
+      "logps/chosen": -263.4698791503906,
+      "logps/rejected": -221.0123748779297,
+      "loss": 20.8846,
+      "rewards/accuracies": 0.6312500238418579,
+      "rewards/chosen": 0.03651345893740654,
+      "rewards/margins": 0.06544128805398941,
+      "rewards/rejected": -0.02892782725393772,
+      "step": 120
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 4.563552833078101e-07,
+      "logits/chosen": -2.642108201980591,
+      "logits/rejected": -2.2785372734069824,
+      "logps/chosen": -265.32977294921875,
+      "logps/rejected": -217.34768676757812,
+      "loss": 20.4002,
+      "rewards/accuracies": 0.671875,
+      "rewards/chosen": 0.039137523621320724,
+      "rewards/margins": 0.07437606900930405,
+      "rewards/rejected": -0.035238541662693024,
+      "step": 130
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 4.4869831546707505e-07,
+      "logits/chosen": -2.6350350379943848,
+      "logits/rejected": -2.323685646057129,
+      "logps/chosen": -266.7952575683594,
+      "logps/rejected": -226.29617309570312,
+      "loss": 20.3918,
+      "rewards/accuracies": 0.706250011920929,
+      "rewards/chosen": 0.04871717840433121,
+      "rewards/margins": 0.0838700383901596,
+      "rewards/rejected": -0.0351528637111187,
+      "step": 140
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 4.4104134762633994e-07,
+      "logits/chosen": -2.6673989295959473,
+      "logits/rejected": -2.305692195892334,
+      "logps/chosen": -267.29180908203125,
+      "logps/rejected": -207.8380889892578,
+      "loss": 19.8348,
+      "rewards/accuracies": 0.703125,
+      "rewards/chosen": 0.04811176657676697,
+      "rewards/margins": 0.09522075951099396,
+      "rewards/rejected": -0.047109004110097885,
+      "step": 150
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 4.333843797856049e-07,
+      "logits/chosen": -2.664452314376831,
+      "logits/rejected": -2.3129963874816895,
+      "logps/chosen": -264.9499816894531,
+      "logps/rejected": -236.6409454345703,
+      "loss": 19.7381,
+      "rewards/accuracies": 0.6937500238418579,
+      "rewards/chosen": 0.045278288424015045,
+      "rewards/margins": 0.0909709632396698,
+      "rewards/rejected": -0.04569266736507416,
+      "step": 160
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 4.257274119448698e-07,
+      "logits/chosen": -2.581209421157837,
+      "logits/rejected": -2.3164689540863037,
+      "logps/chosen": -269.43072509765625,
+      "logps/rejected": -230.83047485351562,
+      "loss": 19.8434,
+      "rewards/accuracies": 0.640625,
+      "rewards/chosen": 0.03498955816030502,
+      "rewards/margins": 0.08386311680078506,
+      "rewards/rejected": -0.048873551189899445,
+      "step": 170
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 4.180704441041347e-07,
+      "logits/chosen": -2.648585557937622,
+      "logits/rejected": -2.3146426677703857,
+      "logps/chosen": -254.5563507080078,
+      "logps/rejected": -222.23922729492188,
+      "loss": 20.3028,
+      "rewards/accuracies": 0.6875,
+      "rewards/chosen": 0.03642386570572853,
+      "rewards/margins": 0.10294415056705475,
+      "rewards/rejected": -0.06652027368545532,
+      "step": 180
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 4.1041347626339966e-07,
+      "logits/chosen": -2.684251308441162,
+      "logits/rejected": -2.282341241836548,
+      "logps/chosen": -262.4200744628906,
+      "logps/rejected": -213.4808807373047,
+      "loss": 20.0038,
+      "rewards/accuracies": 0.699999988079071,
+      "rewards/chosen": 0.03100644052028656,
+      "rewards/margins": 0.10564477741718292,
+      "rewards/rejected": -0.07463832199573517,
+      "step": 190
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 4.027565084226646e-07,
+      "logits/chosen": -2.6774110794067383,
+      "logits/rejected": -2.31669020652771,
+      "logps/chosen": -242.54110717773438,
+      "logps/rejected": -201.0673370361328,
+      "loss": 18.8904,
+      "rewards/accuracies": 0.6812499761581421,
+      "rewards/chosen": 0.029109233990311623,
+      "rewards/margins": 0.09885577857494354,
+      "rewards/rejected": -0.06974655389785767,
+      "step": 200
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 3.9509954058192954e-07,
+      "logits/chosen": -2.6493515968322754,
+      "logits/rejected": -2.2453055381774902,
+      "logps/chosen": -269.30072021484375,
+      "logps/rejected": -215.35482788085938,
+      "loss": 18.7438,
+      "rewards/accuracies": 0.6937500238418579,
+      "rewards/chosen": 0.03713355213403702,
+      "rewards/margins": 0.12264309078454971,
+      "rewards/rejected": -0.0855095237493515,
+      "step": 210
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 3.874425727411945e-07,
+      "logits/chosen": -2.6454288959503174,
+      "logits/rejected": -2.2814364433288574,
+      "logps/chosen": -264.0574035644531,
+      "logps/rejected": -206.1600341796875,
+      "loss": 19.3111,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": 0.03974270448088646,
+      "rewards/margins": 0.12600289285182953,
+      "rewards/rejected": -0.08626019209623337,
+      "step": 220
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 3.797856049004594e-07,
+      "logits/chosen": -2.6513967514038086,
+      "logits/rejected": -2.437913179397583,
+      "logps/chosen": -261.82855224609375,
+      "logps/rejected": -220.0791778564453,
+      "loss": 19.1461,
+      "rewards/accuracies": 0.731249988079071,
+      "rewards/chosen": 0.04797480255365372,
+      "rewards/margins": 0.12008707225322723,
+      "rewards/rejected": -0.07211227715015411,
+      "step": 230
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 3.7212863705972436e-07,
+      "logits/chosen": -2.664517879486084,
+      "logits/rejected": -2.3380627632141113,
+      "logps/chosen": -268.38922119140625,
+      "logps/rejected": -203.86264038085938,
+      "loss": 19.1303,
+      "rewards/accuracies": 0.7124999761581421,
+      "rewards/chosen": 0.037844784557819366,
+      "rewards/margins": 0.11463721841573715,
+      "rewards/rejected": -0.07679243385791779,
+      "step": 240
+    },
+    {
+      "epoch": 1.0,
+      "eval_logits/chosen": -2.688377618789673,
+      "eval_logits/rejected": -2.3029286861419678,
+      "eval_logps/chosen": -268.4410400390625,
+      "eval_logps/rejected": -212.60609436035156,
+      "eval_loss": 18.912391662597656,
+      "eval_rewards/accuracies": 0.7360000014305115,
+      "eval_rewards/chosen": 0.03018273413181305,
+      "eval_rewards/margins": 0.12408097833395004,
+      "eval_rewards/rejected": -0.09389825165271759,
+      "eval_runtime": 352.0426,
+      "eval_samples_per_second": 5.681,
+      "eval_steps_per_second": 0.71,
+      "step": 242
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 3.6447166921898925e-07,
+      "logits/chosen": -2.585019588470459,
+      "logits/rejected": -2.2706055641174316,
+      "logps/chosen": -248.5968017578125,
+      "logps/rejected": -192.33859252929688,
+      "loss": 19.3292,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": 0.026646222919225693,
+      "rewards/margins": 0.12298872321844101,
+      "rewards/rejected": -0.09634250402450562,
+      "step": 250
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 3.568147013782542e-07,
+      "logits/chosen": -2.6509532928466797,
+      "logits/rejected": -2.2828612327575684,
+      "logps/chosen": -250.4876708984375,
+      "logps/rejected": -202.44654846191406,
+      "loss": 18.5341,
+      "rewards/accuracies": 0.734375,
+      "rewards/chosen": 0.04957858473062515,
+      "rewards/margins": 0.13254784047603607,
+      "rewards/rejected": -0.08296926319599152,
+      "step": 260
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 3.4915773353751913e-07,
+      "logits/chosen": -2.6814422607421875,
+      "logits/rejected": -2.252516269683838,
+      "logps/chosen": -273.83038330078125,
+      "logps/rejected": -227.3008270263672,
+      "loss": 18.3748,
+      "rewards/accuracies": 0.768750011920929,
+      "rewards/chosen": 0.038934461772441864,
+      "rewards/margins": 0.14948731660842896,
+      "rewards/rejected": -0.11055286228656769,
+      "step": 270
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 3.41500765696784e-07,
+      "logits/chosen": -2.6279919147491455,
+      "logits/rejected": -2.3474836349487305,
+      "logps/chosen": -230.10598754882812,
+      "logps/rejected": -215.902587890625,
+      "loss": 18.5872,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": 0.03518626466393471,
+      "rewards/margins": 0.108550526201725,
+      "rewards/rejected": -0.0733642652630806,
+      "step": 280
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 3.33843797856049e-07,
+      "logits/chosen": -2.625840663909912,
+      "logits/rejected": -2.388363838195801,
+      "logps/chosen": -287.68280029296875,
+      "logps/rejected": -237.7518768310547,
+      "loss": 18.1373,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": 0.03980474919080734,
+      "rewards/margins": 0.125565305352211,
+      "rewards/rejected": -0.08576056361198425,
+      "step": 290
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 3.2618683001531396e-07,
+      "logits/chosen": -2.592059373855591,
+      "logits/rejected": -2.242305278778076,
+      "logps/chosen": -239.9512939453125,
+      "logps/rejected": -212.0597686767578,
+      "loss": 18.2727,
+      "rewards/accuracies": 0.746874988079071,
+      "rewards/chosen": 0.030053243041038513,
+      "rewards/margins": 0.1290222704410553,
+      "rewards/rejected": -0.09896902740001678,
+      "step": 300
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 3.1852986217457885e-07,
+      "logits/chosen": -2.6135897636413574,
+      "logits/rejected": -2.2893946170806885,
+      "logps/chosen": -252.56002807617188,
+      "logps/rejected": -218.57937622070312,
+      "loss": 18.9516,
+      "rewards/accuracies": 0.7124999761581421,
+      "rewards/chosen": 0.027646947652101517,
+      "rewards/margins": 0.11891412734985352,
+      "rewards/rejected": -0.0912671834230423,
+      "step": 310
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 3.108728943338438e-07,
+      "logits/chosen": -2.6720099449157715,
+      "logits/rejected": -2.3789660930633545,
+      "logps/chosen": -255.5870361328125,
+      "logps/rejected": -232.6374969482422,
+      "loss": 18.993,
+      "rewards/accuracies": 0.684374988079071,
+      "rewards/chosen": 0.03462428227066994,
+      "rewards/margins": 0.11329145729541779,
+      "rewards/rejected": -0.07866715639829636,
+      "step": 320
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 3.0321592649310873e-07,
+      "logits/chosen": -2.636521816253662,
+      "logits/rejected": -2.260267496109009,
+      "logps/chosen": -242.1477508544922,
+      "logps/rejected": -213.030029296875,
+      "loss": 18.1642,
+      "rewards/accuracies": 0.737500011920929,
+      "rewards/chosen": 0.04163060337305069,
+      "rewards/margins": 0.13628455996513367,
+      "rewards/rejected": -0.09465396404266357,
+      "step": 330
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 2.955589586523736e-07,
+      "logits/chosen": -2.6310007572174072,
+      "logits/rejected": -2.3438334465026855,
+      "logps/chosen": -268.5300598144531,
+      "logps/rejected": -229.24057006835938,
+      "loss": 19.3254,
+      "rewards/accuracies": 0.734375,
+      "rewards/chosen": 0.03425261378288269,
+      "rewards/margins": 0.1407606601715088,
+      "rewards/rejected": -0.1065080538392067,
+      "step": 340
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 2.8790199081163856e-07,
+      "logits/chosen": -2.6794886589050293,
+      "logits/rejected": -2.297974109649658,
+      "logps/chosen": -271.8599853515625,
+      "logps/rejected": -222.39688110351562,
+      "loss": 19.1924,
+      "rewards/accuracies": 0.684374988079071,
+      "rewards/chosen": 0.022753948345780373,
+      "rewards/margins": 0.10870112478733063,
+      "rewards/rejected": -0.0859471783041954,
+      "step": 350
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 2.802450229709035e-07,
+      "logits/chosen": -2.635558843612671,
+      "logits/rejected": -2.196741819381714,
+      "logps/chosen": -250.3131103515625,
+      "logps/rejected": -201.41946411132812,
+      "loss": 18.4515,
+      "rewards/accuracies": 0.7250000238418579,
+      "rewards/chosen": 0.018490195274353027,
+      "rewards/margins": 0.12475041300058365,
+      "rewards/rejected": -0.10626020282506943,
+      "step": 360
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.725880551301684e-07,
+      "logits/chosen": -2.6633524894714355,
+      "logits/rejected": -2.3473284244537354,
+      "logps/chosen": -271.9929504394531,
+      "logps/rejected": -228.14157104492188,
+      "loss": 19.0148,
+      "rewards/accuracies": 0.684374988079071,
+      "rewards/chosen": 0.015023022890090942,
+      "rewards/margins": 0.12562182545661926,
+      "rewards/rejected": -0.11059880256652832,
+      "step": 370
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 2.649310872894334e-07,
+      "logits/chosen": -2.6451079845428467,
+      "logits/rejected": -2.3189663887023926,
+      "logps/chosen": -245.8695831298828,
+      "logps/rejected": -210.0269775390625,
+      "loss": 18.4524,
+      "rewards/accuracies": 0.7250000238418579,
+      "rewards/chosen": 0.007006034255027771,
+      "rewards/margins": 0.11740052700042725,
+      "rewards/rejected": -0.11039450019598007,
+      "step": 380
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 2.572741194486983e-07,
+      "logits/chosen": -2.6605451107025146,
+      "logits/rejected": -2.206817626953125,
+      "logps/chosen": -275.4052429199219,
+      "logps/rejected": -203.68002319335938,
+      "loss": 18.4799,
+      "rewards/accuracies": 0.746874988079071,
+      "rewards/chosen": 0.017081182450056076,
+      "rewards/margins": 0.14246238768100739,
+      "rewards/rejected": -0.1253812164068222,
+      "step": 390
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 2.496171516079632e-07,
+      "logits/chosen": -2.74629282951355,
+      "logits/rejected": -2.338416337966919,
+      "logps/chosen": -276.91033935546875,
+      "logps/rejected": -221.68307495117188,
+      "loss": 18.5723,
+      "rewards/accuracies": 0.7281249761581421,
+      "rewards/chosen": 0.018960032612085342,
+      "rewards/margins": 0.1522599160671234,
+      "rewards/rejected": -0.13329990208148956,
+      "step": 400
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 2.4196018376722816e-07,
+      "logits/chosen": -2.616495370864868,
+      "logits/rejected": -2.2858164310455322,
+      "logps/chosen": -260.0771484375,
+      "logps/rejected": -216.95919799804688,
+      "loss": 18.8108,
+      "rewards/accuracies": 0.731249988079071,
+      "rewards/chosen": 0.022241920232772827,
+      "rewards/margins": 0.14108815789222717,
+      "rewards/rejected": -0.11884623765945435,
+      "step": 410
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 2.343032159264931e-07,
+      "logits/chosen": -2.651947259902954,
+      "logits/rejected": -2.266605854034424,
+      "logps/chosen": -264.1819152832031,
+      "logps/rejected": -221.7666473388672,
+      "loss": 18.6108,
+      "rewards/accuracies": 0.746874988079071,
+      "rewards/chosen": 0.019768675789237022,
+      "rewards/margins": 0.14661307632923126,
+      "rewards/rejected": -0.1268443912267685,
+      "step": 420
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 2.26646248085758e-07,
+      "logits/chosen": -2.6224217414855957,
+      "logits/rejected": -2.2512006759643555,
+      "logps/chosen": -242.04757690429688,
+      "logps/rejected": -202.48556518554688,
+      "loss": 18.291,
+      "rewards/accuracies": 0.753125011920929,
+      "rewards/chosen": 0.020827816799283028,
+      "rewards/margins": 0.13378821313381195,
+      "rewards/rejected": -0.11296038329601288,
+      "step": 430
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 2.1898928024502298e-07,
+      "logits/chosen": -2.658205509185791,
+      "logits/rejected": -2.3186447620391846,
+      "logps/chosen": -247.4537353515625,
+      "logps/rejected": -198.62149047851562,
+      "loss": 18.2582,
+      "rewards/accuracies": 0.706250011920929,
+      "rewards/chosen": 0.008651083335280418,
+      "rewards/margins": 0.12013272196054459,
+      "rewards/rejected": -0.11148162931203842,
+      "step": 440
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.113323124042879e-07,
+      "logits/chosen": -2.6413986682891846,
+      "logits/rejected": -2.323753595352173,
+      "logps/chosen": -265.66241455078125,
+      "logps/rejected": -233.0228729248047,
+      "loss": 18.5032,
+      "rewards/accuracies": 0.734375,
+      "rewards/chosen": 0.01642756350338459,
+      "rewards/margins": 0.13231472671031952,
+      "rewards/rejected": -0.11588716506958008,
+      "step": 450
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 2.036753445635528e-07,
+      "logits/chosen": -2.6705307960510254,
+      "logits/rejected": -2.3020169734954834,
+      "logps/chosen": -286.9468078613281,
+      "logps/rejected": -232.2310028076172,
+      "loss": 18.0061,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": 0.01540865283459425,
+      "rewards/margins": 0.14157035946846008,
+      "rewards/rejected": -0.12616169452667236,
+      "step": 460
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 1.9601837672281775e-07,
+      "logits/chosen": -2.645066022872925,
+      "logits/rejected": -2.2868847846984863,
+      "logps/chosen": -275.22344970703125,
+      "logps/rejected": -230.9308624267578,
+      "loss": 18.0886,
+      "rewards/accuracies": 0.7593749761581421,
+      "rewards/chosen": 0.025068962946534157,
+      "rewards/margins": 0.1567041575908661,
+      "rewards/rejected": -0.13163520395755768,
+      "step": 470
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 1.883614088820827e-07,
+      "logits/chosen": -2.6855411529541016,
+      "logits/rejected": -2.2669596672058105,
+      "logps/chosen": -257.3984680175781,
+      "logps/rejected": -210.7415771484375,
+      "loss": 19.0824,
+      "rewards/accuracies": 0.737500011920929,
+      "rewards/chosen": 0.01750902086496353,
+      "rewards/margins": 0.12825700640678406,
+      "rewards/rejected": -0.11074797064065933,
+      "step": 480
+    },
+    {
+      "epoch": 2.0,
+      "eval_logits/chosen": -2.687664747238159,
+      "eval_logits/rejected": -2.3010897636413574,
+      "eval_logps/chosen": -268.55670166015625,
+      "eval_logps/rejected": -212.82968139648438,
+      "eval_loss": 18.22655487060547,
+      "eval_rewards/accuracies": 0.7599999904632568,
+      "eval_rewards/chosen": 0.01861482858657837,
+      "eval_rewards/margins": 0.13487249612808228,
+      "eval_rewards/rejected": -0.1162576824426651,
+      "eval_runtime": 352.6258,
+      "eval_samples_per_second": 5.672,
+      "eval_steps_per_second": 0.709,
+      "step": 484
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 1.807044410413476e-07,
+      "logits/chosen": -2.63578200340271,
+      "logits/rejected": -2.335254430770874,
+      "logps/chosen": -252.0665740966797,
+      "logps/rejected": -221.12002563476562,
+      "loss": 18.6586,
+      "rewards/accuracies": 0.734375,
+      "rewards/chosen": 0.00841272622346878,
+      "rewards/margins": 0.12638814747333527,
+      "rewards/rejected": -0.11797541379928589,
+      "step": 490
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 1.7304747320061255e-07,
+      "logits/chosen": -2.576451063156128,
+      "logits/rejected": -2.343707799911499,
+      "logps/chosen": -253.4521484375,
+      "logps/rejected": -228.43115234375,
+      "loss": 18.2798,
+      "rewards/accuracies": 0.706250011920929,
+      "rewards/chosen": 0.0265754796564579,
+      "rewards/margins": 0.1290186494588852,
+      "rewards/rejected": -0.1024431437253952,
+      "step": 500
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 1.6539050535987747e-07,
+      "logits/chosen": -2.672646999359131,
+      "logits/rejected": -2.302988290786743,
+      "logps/chosen": -275.57440185546875,
+      "logps/rejected": -221.45455932617188,
+      "loss": 18.1,
+      "rewards/accuracies": 0.737500011920929,
+      "rewards/chosen": 0.02416495978832245,
+      "rewards/margins": 0.14235563576221466,
+      "rewards/rejected": -0.11819068342447281,
+      "step": 510
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 1.5773353751914243e-07,
+      "logits/chosen": -2.759183406829834,
+      "logits/rejected": -2.3177244663238525,
+      "logps/chosen": -282.8927307128906,
+      "logps/rejected": -217.7075958251953,
+      "loss": 18.4222,
+      "rewards/accuracies": 0.7093750238418579,
+      "rewards/chosen": 0.021920911967754364,
+      "rewards/margins": 0.13253231346607208,
+      "rewards/rejected": -0.11061139404773712,
+      "step": 520
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 1.5007656967840735e-07,
+      "logits/chosen": -2.6353068351745605,
+      "logits/rejected": -2.291774272918701,
+      "logps/chosen": -260.0713806152344,
+      "logps/rejected": -232.3697052001953,
+      "loss": 18.084,
+      "rewards/accuracies": 0.7749999761581421,
+      "rewards/chosen": 0.020517101511359215,
+      "rewards/margins": 0.15263554453849792,
+      "rewards/rejected": -0.13211843371391296,
+      "step": 530
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 1.4241960183767226e-07,
+      "logits/chosen": -2.665095806121826,
+      "logits/rejected": -2.3143036365509033,
+      "logps/chosen": -270.5109558105469,
+      "logps/rejected": -228.31759643554688,
+      "loss": 18.2089,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": 0.020305803045630455,
+      "rewards/margins": 0.12145675718784332,
+      "rewards/rejected": -0.10115096718072891,
+      "step": 540
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 1.347626339969372e-07,
+      "logits/chosen": -2.6581876277923584,
+      "logits/rejected": -2.402988910675049,
+      "logps/chosen": -276.17437744140625,
+      "logps/rejected": -239.04080200195312,
+      "loss": 18.2431,
+      "rewards/accuracies": 0.753125011920929,
+      "rewards/chosen": 0.04555383324623108,
+      "rewards/margins": 0.13798871636390686,
+      "rewards/rejected": -0.09243487566709518,
+      "step": 550
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 1.2710566615620215e-07,
+      "logits/chosen": -2.6366963386535645,
+      "logits/rejected": -2.2572150230407715,
+      "logps/chosen": -272.67034912109375,
+      "logps/rejected": -214.80441284179688,
+      "loss": 17.8127,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": 0.020294126123189926,
+      "rewards/margins": 0.14420394599437714,
+      "rewards/rejected": -0.1239098310470581,
+      "step": 560
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 1.1944869831546706e-07,
+      "logits/chosen": -2.663076162338257,
+      "logits/rejected": -2.3145666122436523,
+      "logps/chosen": -273.4615478515625,
+      "logps/rejected": -232.04428100585938,
+      "loss": 18.2306,
+      "rewards/accuracies": 0.753125011920929,
+      "rewards/chosen": 0.03313525393605232,
+      "rewards/margins": 0.13954436779022217,
+      "rewards/rejected": -0.10640911012887955,
+      "step": 570
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 1.11791730474732e-07,
+      "logits/chosen": -2.637507915496826,
+      "logits/rejected": -2.294003963470459,
+      "logps/chosen": -254.7287139892578,
+      "logps/rejected": -214.94534301757812,
+      "loss": 17.9129,
+      "rewards/accuracies": 0.762499988079071,
+      "rewards/chosen": 0.023206666111946106,
+      "rewards/margins": 0.14315126836299896,
+      "rewards/rejected": -0.11994459480047226,
+      "step": 580
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 1.0413476263399694e-07,
+      "logits/chosen": -2.5812573432922363,
+      "logits/rejected": -2.254936695098877,
+      "logps/chosen": -267.64752197265625,
+      "logps/rejected": -209.09072875976562,
+      "loss": 18.3503,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": 0.01215180940926075,
+      "rewards/margins": 0.13401289284229279,
+      "rewards/rejected": -0.12186107784509659,
+      "step": 590
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 9.647779479326186e-08,
+      "logits/chosen": -2.645982265472412,
+      "logits/rejected": -2.3950445652008057,
+      "logps/chosen": -253.10336303710938,
+      "logps/rejected": -217.19949340820312,
+      "loss": 19.0115,
+      "rewards/accuracies": 0.699999988079071,
+      "rewards/chosen": 0.022207045927643776,
+      "rewards/margins": 0.11882482469081879,
+      "rewards/rejected": -0.09661778062582016,
+      "step": 600
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 8.88208269525268e-08,
+      "logits/chosen": -2.606689929962158,
+      "logits/rejected": -2.274811029434204,
+      "logps/chosen": -254.597412109375,
+      "logps/rejected": -207.0297088623047,
+      "loss": 18.634,
+      "rewards/accuracies": 0.721875011920929,
+      "rewards/chosen": 0.020870240405201912,
+      "rewards/margins": 0.11707819998264313,
+      "rewards/rejected": -0.09620794653892517,
+      "step": 610
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 8.116385911179173e-08,
+      "logits/chosen": -2.6234843730926514,
+      "logits/rejected": -2.277554750442505,
+      "logps/chosen": -265.1994934082031,
+      "logps/rejected": -217.2080841064453,
+      "loss": 18.3686,
+      "rewards/accuracies": 0.7124999761581421,
+      "rewards/chosen": 0.03428371995687485,
+      "rewards/margins": 0.13564249873161316,
+      "rewards/rejected": -0.1013587936758995,
+      "step": 620
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 7.350689127105667e-08,
+      "logits/chosen": -2.681215286254883,
+      "logits/rejected": -2.2755188941955566,
+      "logps/chosen": -263.49530029296875,
+      "logps/rejected": -226.5316619873047,
+      "loss": 18.1639,
+      "rewards/accuracies": 0.706250011920929,
+      "rewards/chosen": 0.024900907650589943,
+      "rewards/margins": 0.14352695643901825,
+      "rewards/rejected": -0.11862604320049286,
+      "step": 630
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 6.584992343032159e-08,
+      "logits/chosen": -2.589116096496582,
+      "logits/rejected": -2.3119215965270996,
+      "logps/chosen": -255.6792449951172,
+      "logps/rejected": -225.32754516601562,
+      "loss": 18.7407,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": 0.022887857630848885,
+      "rewards/margins": 0.13587646186351776,
+      "rewards/rejected": -0.11298861354589462,
+      "step": 640
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 5.819295558958652e-08,
+      "logits/chosen": -2.664092779159546,
+      "logits/rejected": -2.245487928390503,
+      "logps/chosen": -287.26129150390625,
+      "logps/rejected": -202.8896942138672,
+      "loss": 17.9121,
+      "rewards/accuracies": 0.765625,
+      "rewards/chosen": 0.03479045629501343,
+      "rewards/margins": 0.17097266018390656,
+      "rewards/rejected": -0.13618221879005432,
+      "step": 650
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 5.0535987748851455e-08,
+      "logits/chosen": -2.6955533027648926,
+      "logits/rejected": -2.341909646987915,
+      "logps/chosen": -276.48583984375,
+      "logps/rejected": -231.2989959716797,
+      "loss": 18.0214,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": 0.03698002174496651,
+      "rewards/margins": 0.14650048315525055,
+      "rewards/rejected": -0.10952047258615494,
+      "step": 660
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 4.287901990811638e-08,
+      "logits/chosen": -2.691222667694092,
+      "logits/rejected": -2.269242286682129,
+      "logps/chosen": -257.7073974609375,
+      "logps/rejected": -232.1117706298828,
+      "loss": 18.0439,
+      "rewards/accuracies": 0.7124999761581421,
+      "rewards/chosen": 0.030216818675398827,
+      "rewards/margins": 0.14609763026237488,
+      "rewards/rejected": -0.1158808022737503,
+      "step": 670
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 3.522205206738132e-08,
+      "logits/chosen": -2.6416354179382324,
+      "logits/rejected": -2.310128688812256,
+      "logps/chosen": -271.9048767089844,
+      "logps/rejected": -223.27969360351562,
+      "loss": 18.4412,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": 0.02033752016723156,
+      "rewards/margins": 0.12728351354599,
+      "rewards/rejected": -0.10694599151611328,
+      "step": 680
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.7565084226646246e-08,
+      "logits/chosen": -2.660130739212036,
+      "logits/rejected": -2.3302083015441895,
+      "logps/chosen": -269.9329833984375,
+      "logps/rejected": -232.0976104736328,
+      "loss": 18.3673,
+      "rewards/accuracies": 0.746874988079071,
+      "rewards/chosen": 0.030508223921060562,
+      "rewards/margins": 0.14415839314460754,
+      "rewards/rejected": -0.11365016549825668,
+      "step": 690
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 1.9908116385911178e-08,
+      "logits/chosen": -2.6514875888824463,
+      "logits/rejected": -2.333763599395752,
+      "logps/chosen": -251.9337158203125,
+      "logps/rejected": -223.32437133789062,
+      "loss": 18.373,
+      "rewards/accuracies": 0.684374988079071,
+      "rewards/chosen": 0.020953018218278885,
+      "rewards/margins": 0.114670529961586,
+      "rewards/rejected": -0.09371750801801682,
+      "step": 700
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 1.225114854517611e-08,
+      "logits/chosen": -2.6763319969177246,
+      "logits/rejected": -2.2669501304626465,
+      "logps/chosen": -274.1488037109375,
+      "logps/rejected": -220.2371063232422,
+      "loss": 18.4056,
+      "rewards/accuracies": 0.7250000238418579,
+      "rewards/chosen": 0.005063548684120178,
+      "rewards/margins": 0.12724530696868896,
+      "rewards/rejected": -0.12218175083398819,
+      "step": 710
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 4.594180704441042e-09,
+      "logits/chosen": -2.5772223472595215,
+      "logits/rejected": -2.3834898471832275,
+      "logps/chosen": -257.3438720703125,
+      "logps/rejected": -220.6257781982422,
+      "loss": 18.0366,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": 0.019895819947123528,
+      "rewards/margins": 0.12051371484994888,
+      "rewards/rejected": -0.10061788558959961,
+      "step": 720
+    },
+    {
+      "epoch": 3.0,
+      "eval_logits/chosen": -2.687349796295166,
+      "eval_logits/rejected": -2.300260305404663,
+      "eval_logps/chosen": -268.54150390625,
+      "eval_logps/rejected": -212.86087036132812,
+      "eval_loss": 18.088815689086914,
+      "eval_rewards/accuracies": 0.7319999933242798,
+      "eval_rewards/chosen": 0.02013408951461315,
+      "eval_rewards/margins": 0.13951165974140167,
+      "eval_rewards/rejected": -0.11937756091356277,
+      "eval_runtime": 350.3279,
+      "eval_samples_per_second": 5.709,
+      "eval_steps_per_second": 0.714,
+      "step": 726
+    },
+    {
+      "epoch": 3.0,
+      "step": 726,
+      "total_flos": 0.0,
+      "train_loss": 19.467516181882747,
+      "train_runtime": 42806.6761,
+      "train_samples_per_second": 4.343,
+      "train_steps_per_second": 0.017
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 726,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 3,
+  "save_steps": 500,
+  "total_flos": 0.0,
+  "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:dcef5bbdb0029b8a95cad5773a8a63ce2c0d87560605567e421e5d172bfb3de9
+size 4856