Model save

README.md

@@ -0,0 +1,75 @@
+---
+license: apache-2.0
+library_name: peft
+tags:
+- trl
+- dpo
+- generated_from_trainer
+base_model: alignment-handbook/zephyr-7b-sft-full
+model-index:
+- name: nash_ave_pi_iter_2
+  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. -->
+
+# nash_ave_pi_iter_2
+
+This model is a fine-tuned version of [alignment-handbook/zephyr-7b-sft-full](https://huggingface.co/alignment-handbook/zephyr-7b-sft-full) on the None dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.6930
+- Rewards/chosen: -0.1247
+- Rewards/rejected: -0.1296
+- Rewards/accuracies: 0.5140
+- Rewards/margins: 0.0049
+- Logps/rejected: -302.4665
+- Logps/chosen: -310.5244
+- Logits/rejected: -2.4580
+- Logits/chosen: -2.5554
+
+## 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-06
+- train_batch_size: 2
+- eval_batch_size: 2
+- seed: 42
+- distributed_type: multi-GPU
+- num_devices: 4
+- gradient_accumulation_steps: 16
+- total_train_batch_size: 128
+- total_eval_batch_size: 8
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: cosine
+- lr_scheduler_warmup_ratio: 0.1
+- num_epochs: 1
+
+### 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.6902        | 0.64  | 100  | 0.6930          | -0.1247        | -0.1296          | 0.5140             | 0.0049          | -302.4665      | -310.5244    | -2.4580         | -2.5554       |
+
+
+### Framework versions
+
+- PEFT 0.7.1
+- Transformers 4.36.2
+- Pytorch 2.3.0+cu121
+- Datasets 2.14.6
+- Tokenizers 0.15.2

all_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 1.0,
+    "train_loss": 0.6914212252849188,
+    "train_runtime": 7065.3773,
+    "train_samples": 20000,
+    "train_samples_per_second": 2.831,
+    "train_steps_per_second": 0.022
+}

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 1.0,
+    "train_loss": 0.6914212252849188,
+    "train_runtime": 7065.3773,
+    "train_samples": 20000,
+    "train_samples_per_second": 2.831,
+    "train_steps_per_second": 0.022
+}

trainer_state.json

@@ -0,0 +1,270 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.9984,
+  "eval_steps": 100,
+  "global_step": 156,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.01,
+      "learning_rate": 3.125e-07,
+      "logits/chosen": -2.62762188911438,
+      "logits/rejected": -2.586808681488037,
+      "logps/chosen": -233.10275268554688,
+      "logps/rejected": -220.01724243164062,
+      "loss": 0.6931,
+      "rewards/accuracies": 0.0,
+      "rewards/chosen": 0.0,
+      "rewards/margins": 0.0,
+      "rewards/rejected": 0.0,
+      "step": 1
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 3.125e-06,
+      "logits/chosen": -2.640223741531372,
+      "logits/rejected": -2.6421992778778076,
+      "logps/chosen": -172.71710205078125,
+      "logps/rejected": -174.91177368164062,
+      "loss": 0.6932,
+      "rewards/accuracies": 0.4270833432674408,
+      "rewards/chosen": -0.0005670329555869102,
+      "rewards/margins": -0.00034490120015107095,
+      "rewards/rejected": -0.00022213184274733067,
+      "step": 10
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 4.989935734988098e-06,
+      "logits/chosen": -2.6506519317626953,
+      "logits/rejected": -2.6606290340423584,
+      "logps/chosen": -163.25784301757812,
+      "logps/rejected": -166.31251525878906,
+      "loss": 0.6923,
+      "rewards/accuracies": 0.546875,
+      "rewards/chosen": -0.005096795037388802,
+      "rewards/margins": 0.002602009102702141,
+      "rewards/rejected": -0.007698804140090942,
+      "step": 20
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 4.8776412907378845e-06,
+      "logits/chosen": -2.6732003688812256,
+      "logits/rejected": -2.6337249279022217,
+      "logps/chosen": -179.1455535888672,
+      "logps/rejected": -177.36630249023438,
+      "loss": 0.6928,
+      "rewards/accuracies": 0.5218750238418579,
+      "rewards/chosen": -0.07040824741125107,
+      "rewards/margins": 0.0008022676920518279,
+      "rewards/rejected": -0.07121052592992783,
+      "step": 30
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 4.646121984004666e-06,
+      "logits/chosen": -2.592510223388672,
+      "logits/rejected": -2.569596767425537,
+      "logps/chosen": -176.0773162841797,
+      "logps/rejected": -177.98220825195312,
+      "loss": 0.6939,
+      "rewards/accuracies": 0.5249999761581421,
+      "rewards/chosen": -0.06681038439273834,
+      "rewards/margins": -0.002174162771552801,
+      "rewards/rejected": -0.0646362155675888,
+      "step": 40
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 4.3069871595684795e-06,
+      "logits/chosen": -2.6304595470428467,
+      "logits/rejected": -2.6132376194000244,
+      "logps/chosen": -170.5052947998047,
+      "logps/rejected": -170.62728881835938,
+      "loss": 0.6925,
+      "rewards/accuracies": 0.5218750238418579,
+      "rewards/chosen": -0.03540392965078354,
+      "rewards/margins": 0.004562810063362122,
+      "rewards/rejected": -0.03996673971414566,
+      "step": 50
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 3.8772424536302565e-06,
+      "logits/chosen": -2.5987067222595215,
+      "logits/rejected": -2.573859691619873,
+      "logps/chosen": -180.9357452392578,
+      "logps/rejected": -177.1747589111328,
+      "loss": 0.6923,
+      "rewards/accuracies": 0.48750001192092896,
+      "rewards/chosen": -0.09346936643123627,
+      "rewards/margins": 0.0016549427527934313,
+      "rewards/rejected": -0.09512430429458618,
+      "step": 60
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 3.3784370602033572e-06,
+      "logits/chosen": -2.5544121265411377,
+      "logits/rejected": -2.570075750350952,
+      "logps/chosen": -172.70663452148438,
+      "logps/rejected": -173.13766479492188,
+      "loss": 0.692,
+      "rewards/accuracies": 0.49687498807907104,
+      "rewards/chosen": -0.15676796436309814,
+      "rewards/margins": -0.002442942699417472,
+      "rewards/rejected": -0.15432503819465637,
+      "step": 70
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 2.835583164544139e-06,
+      "logits/chosen": -2.604128122329712,
+      "logits/rejected": -2.5767688751220703,
+      "logps/chosen": -184.4633331298828,
+      "logps/rejected": -183.540283203125,
+      "loss": 0.6936,
+      "rewards/accuracies": 0.5562499761581421,
+      "rewards/chosen": -0.18969281017780304,
+      "rewards/margins": -0.0006604200461879373,
+      "rewards/rejected": -0.18903236091136932,
+      "step": 80
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 2.2759017277414165e-06,
+      "logits/chosen": -2.661609172821045,
+      "logits/rejected": -2.5941507816314697,
+      "logps/chosen": -191.8031463623047,
+      "logps/rejected": -189.98135375976562,
+      "loss": 0.6914,
+      "rewards/accuracies": 0.565625011920929,
+      "rewards/chosen": -0.08525966852903366,
+      "rewards/margins": 0.013879667036235332,
+      "rewards/rejected": -0.09913934022188187,
+      "step": 90
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 1.7274575140626318e-06,
+      "logits/chosen": -2.6250574588775635,
+      "logits/rejected": -2.6189353466033936,
+      "logps/chosen": -178.07730102539062,
+      "logps/rejected": -179.85862731933594,
+      "loss": 0.6902,
+      "rewards/accuracies": 0.5625,
+      "rewards/chosen": -0.038252077996730804,
+      "rewards/margins": 0.017853444442152977,
+      "rewards/rejected": -0.05610552430152893,
+      "step": 100
+    },
+    {
+      "epoch": 0.64,
+      "eval_logits/chosen": -2.555354595184326,
+      "eval_logits/rejected": -2.4580423831939697,
+      "eval_logps/chosen": -310.5243835449219,
+      "eval_logps/rejected": -302.46649169921875,
+      "eval_loss": 0.6929731369018555,
+      "eval_rewards/accuracies": 0.5139999985694885,
+      "eval_rewards/chosen": -0.12465862929821014,
+      "eval_rewards/margins": 0.00494543369859457,
+      "eval_rewards/rejected": -0.12960407137870789,
+      "eval_runtime": 383.981,
+      "eval_samples_per_second": 5.209,
+      "eval_steps_per_second": 0.651,
+      "step": 100
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 1.217751806485235e-06,
+      "logits/chosen": -2.651780128479004,
+      "logits/rejected": -2.581507921218872,
+      "logps/chosen": -194.07461547851562,
+      "logps/rejected": -183.43362426757812,
+      "loss": 0.6898,
+      "rewards/accuracies": 0.4781250059604645,
+      "rewards/chosen": -0.11336056143045425,
+      "rewards/margins": 0.0036790375597774982,
+      "rewards/rejected": -0.11703959852457047,
+      "step": 110
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 7.723433775328385e-07,
+      "logits/chosen": -2.6290206909179688,
+      "logits/rejected": -2.6049842834472656,
+      "logps/chosen": -186.02259826660156,
+      "logps/rejected": -193.3793487548828,
+      "loss": 0.6902,
+      "rewards/accuracies": 0.534375011920929,
+      "rewards/chosen": -0.11839202791452408,
+      "rewards/margins": 0.004132995847612619,
+      "rewards/rejected": -0.12252502143383026,
+      "step": 120
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 4.1356686569674344e-07,
+      "logits/chosen": -2.5702383518218994,
+      "logits/rejected": -2.5472493171691895,
+      "logps/chosen": -169.70834350585938,
+      "logps/rejected": -174.14761352539062,
+      "loss": 0.6891,
+      "rewards/accuracies": 0.5562499761581421,
+      "rewards/chosen": -0.09281644970178604,
+      "rewards/margins": 0.012840485200285912,
+      "rewards/rejected": -0.1056569367647171,
+      "step": 130
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 1.59412823400657e-07,
+      "logits/chosen": -2.5276107788085938,
+      "logits/rejected": -2.5228161811828613,
+      "logps/chosen": -174.78802490234375,
+      "logps/rejected": -176.2909698486328,
+      "loss": 0.6878,
+      "rewards/accuracies": 0.5625,
+      "rewards/chosen": -0.05636407807469368,
+      "rewards/margins": 0.011297663673758507,
+      "rewards/rejected": -0.06766173988580704,
+      "step": 140
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 2.262559558016325e-08,
+      "logits/chosen": -2.568333864212036,
+      "logits/rejected": -2.586660623550415,
+      "logps/chosen": -166.882080078125,
+      "logps/rejected": -171.17140197753906,
+      "loss": 0.6892,
+      "rewards/accuracies": 0.565625011920929,
+      "rewards/chosen": -0.061885736882686615,
+      "rewards/margins": 0.011388251557946205,
+      "rewards/rejected": -0.07327400147914886,
+      "step": 150
+    },
+    {
+      "epoch": 1.0,
+      "step": 156,
+      "total_flos": 0.0,
+      "train_loss": 0.6914212252849188,
+      "train_runtime": 7065.3773,
+      "train_samples_per_second": 2.831,
+      "train_steps_per_second": 0.022
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 156,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "save_steps": 100,
+  "total_flos": 0.0,
+  "train_batch_size": 2,
+  "trial_name": null,
+  "trial_params": null
+}