Model save

README.md

@@ -0,0 +1,75 @@
+---
+license: mit
+library_name: peft
+tags:
+- trl
+- dpo
+- generated_from_trainer
+base_model: microsoft/phi-2
+model-index:
+- name: phi-2-gpo-ultrachat-lora-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. -->
+
+# phi-2-gpo-ultrachat-lora-2
+
+This model is a fine-tuned version of [microsoft/phi-2](https://huggingface.co/microsoft/phi-2) on the None dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.0100
+- Rewards/chosen: -0.0005
+- Rewards/rejected: -0.0009
+- Rewards/accuracies: 0.2620
+- Rewards/margins: 0.0004
+- Logps/rejected: -94.2882
+- Logps/chosen: -91.7769
+- Logits/rejected: 0.8176
+- Logits/chosen: 0.7994
+
+## 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: 4
+- eval_batch_size: 4
+- seed: 42
+- distributed_type: multi-GPU
+- num_devices: 4
+- gradient_accumulation_steps: 4
+- total_train_batch_size: 64
+- total_eval_batch_size: 16
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: cosine
+- lr_scheduler_warmup_ratio: 0.1
+- num_epochs: 2
+
+### 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.01          | 1.04  | 100  | 0.0100          | -0.0004        | -0.0007          | 0.25               | 0.0003          | -94.2623       | -91.7671     | 0.8188          | 0.8011        |
+
+
+### Framework versions
+
+- PEFT 0.7.1
+- Transformers 4.36.2
+- Pytorch 2.1.2+cu118
+- Datasets 2.14.6
+- Tokenizers 0.15.2

adapter_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a2873a4714d851cf961d555fe0d5c4e003dc74444f2ddeae1bf4bbd7e4316daf
+oid sha256:716f6cfffe6c0290a3d3693b4e351f4358939bc90324c7e09eb6bb9add2bd60a
 size 41977616

all_results.json

@@ -0,0 +1,21 @@
+{
+    "epoch": 1.98,
+    "eval_logits/chosen": 0.7993869781494141,
+    "eval_logits/rejected": 0.8175749778747559,
+    "eval_logps/chosen": -91.77693176269531,
+    "eval_logps/rejected": -94.28819274902344,
+    "eval_loss": 0.009953780099749565,
+    "eval_rewards/accuracies": 0.2619999945163727,
+    "eval_rewards/chosen": -0.0004955825861543417,
+    "eval_rewards/margins": 0.00041623329161666334,
+    "eval_rewards/rejected": -0.0009118159650824964,
+    "eval_runtime": 274.2974,
+    "eval_samples": 2000,
+    "eval_samples_per_second": 7.291,
+    "eval_steps_per_second": 0.456,
+    "train_loss": 0.00995884225458691,
+    "train_runtime": 2568.9627,
+    "train_samples": 6113,
+    "train_samples_per_second": 4.759,
+    "train_steps_per_second": 0.074
+}

eval_results.json

@@ -0,0 +1,16 @@
+{
+    "epoch": 1.98,
+    "eval_logits/chosen": 0.7993869781494141,
+    "eval_logits/rejected": 0.8175749778747559,
+    "eval_logps/chosen": -91.77693176269531,
+    "eval_logps/rejected": -94.28819274902344,
+    "eval_loss": 0.009953780099749565,
+    "eval_rewards/accuracies": 0.2619999945163727,
+    "eval_rewards/chosen": -0.0004955825861543417,
+    "eval_rewards/margins": 0.00041623329161666334,
+    "eval_rewards/rejected": -0.0009118159650824964,
+    "eval_runtime": 274.2974,
+    "eval_samples": 2000,
+    "eval_samples_per_second": 7.291,
+    "eval_steps_per_second": 0.456
+}

runs/Feb29_17-39-53_gpu4-119-4/events.out.tfevents.1709188872.gpu4-119-4.1114052.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a800ba3c4715539d01cc1aaa4d56fad15d7ca235e787025beffd5ce2b502790d
-size 12414
+oid sha256:c8196c904d5639636ba91b3950985d80901f1666329a62d3941c837156254e13
+size 18452

runs/Feb29_17-39-53_gpu4-119-4/events.out.tfevents.1709191715.gpu4-119-4.1114052.1

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

train_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 1.98,
+    "train_loss": 0.00995884225458691,
+    "train_runtime": 2568.9627,
+    "train_samples": 6113,
+    "train_samples_per_second": 4.759,
+    "train_steps_per_second": 0.074
+}

trainer_state.json

@@ -0,0 +1,326 @@
+{
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 1.9843342036553526,
+  "eval_steps": 100,
+  "global_step": 190,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.01,
+      "learning_rate": 2.6315789473684213e-07,
+      "logits/chosen": 0.8826487064361572,
+      "logits/rejected": 0.921362042427063,
+      "logps/chosen": -36.58121871948242,
+      "logps/rejected": -54.902320861816406,
+      "loss": 0.01,
+      "rewards/accuracies": 0.0,
+      "rewards/chosen": 0.0,
+      "rewards/margins": 0.0,
+      "rewards/rejected": 0.0,
+      "step": 1
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 2.631578947368421e-06,
+      "logits/chosen": 0.8915393352508545,
+      "logits/rejected": 0.8742258548736572,
+      "logps/chosen": -87.77196502685547,
+      "logps/rejected": -96.38507843017578,
+      "loss": 0.0101,
+      "rewards/accuracies": 0.2569444477558136,
+      "rewards/chosen": 0.0003006549668498337,
+      "rewards/margins": 0.0004423653008416295,
+      "rewards/rejected": -0.00014171031943988055,
+      "step": 10
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 4.999578104083307e-06,
+      "logits/chosen": 0.7802013158798218,
+      "logits/rejected": 0.8469624519348145,
+      "logps/chosen": -91.75413513183594,
+      "logps/rejected": -85.1173095703125,
+      "loss": 0.01,
+      "rewards/accuracies": 0.24375000596046448,
+      "rewards/chosen": 3.31846640619915e-05,
+      "rewards/margins": -0.00015664812235627323,
+      "rewards/rejected": 0.00018983279005624354,
+      "step": 20
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 4.949122667718935e-06,
+      "logits/chosen": 0.8652933835983276,
+      "logits/rejected": 0.848902702331543,
+      "logps/chosen": -85.29698944091797,
+      "logps/rejected": -78.0544204711914,
+      "loss": 0.0099,
+      "rewards/accuracies": 0.28125,
+      "rewards/chosen": 0.0002226830692961812,
+      "rewards/margins": 0.00038711068918928504,
+      "rewards/rejected": -0.00016442763444501907,
+      "step": 30
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 4.8162351680370046e-06,
+      "logits/chosen": 0.7649837136268616,
+      "logits/rejected": 0.841802716255188,
+      "logps/chosen": -122.76881408691406,
+      "logps/rejected": -108.8086166381836,
+      "loss": 0.01,
+      "rewards/accuracies": 0.26875001192092896,
+      "rewards/chosen": -0.000362707010935992,
+      "rewards/margins": -7.97106167738093e-06,
+      "rewards/rejected": -0.0003547359665390104,
+      "step": 40
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 4.605388304968915e-06,
+      "logits/chosen": 0.8386613130569458,
+      "logits/rejected": 0.8677732348442078,
+      "logps/chosen": -65.77490997314453,
+      "logps/rejected": -71.66779327392578,
+      "loss": 0.01,
+      "rewards/accuracies": 0.21250000596046448,
+      "rewards/chosen": -0.00019812444224953651,
+      "rewards/margins": -0.00015129637904465199,
+      "rewards/rejected": -4.682801591116004e-05,
+      "step": 50
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 4.323678718546552e-06,
+      "logits/chosen": 0.8152744174003601,
+      "logits/rejected": 0.8866288065910339,
+      "logps/chosen": -117.6309814453125,
+      "logps/rejected": -110.9274673461914,
+      "loss": 0.0101,
+      "rewards/accuracies": 0.26875001192092896,
+      "rewards/chosen": 2.0224386389600113e-05,
+      "rewards/margins": -1.7978531104745343e-05,
+      "rewards/rejected": 3.820297933998518e-05,
+      "step": 60
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 3.980588131662451e-06,
+      "logits/chosen": 0.8049997091293335,
+      "logits/rejected": 0.8617580533027649,
+      "logps/chosen": -82.61628723144531,
+      "logps/rejected": -83.92156982421875,
+      "loss": 0.01,
+      "rewards/accuracies": 0.3187499940395355,
+      "rewards/chosen": 0.00046920054592192173,
+      "rewards/margins": 0.000604915083386004,
+      "rewards/rejected": -0.00013571445015259087,
+      "step": 70
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 3.5876642162051833e-06,
+      "logits/chosen": 0.7660447955131531,
+      "logits/rejected": 0.8296969532966614,
+      "logps/chosen": -105.82564544677734,
+      "logps/rejected": -104.93913269042969,
+      "loss": 0.01,
+      "rewards/accuracies": 0.23749999701976776,
+      "rewards/chosen": -0.0001158899613074027,
+      "rewards/margins": -0.00039075990207493305,
+      "rewards/rejected": 0.0002748699625954032,
+      "step": 80
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 3.1581319239114983e-06,
+      "logits/chosen": 0.8339036107063293,
+      "logits/rejected": 0.8548374176025391,
+      "logps/chosen": -77.63902282714844,
+      "logps/rejected": -90.03132629394531,
+      "loss": 0.0099,
+      "rewards/accuracies": 0.36250001192092896,
+      "rewards/chosen": 0.0005246674409136176,
+      "rewards/margins": 0.0012112573022022843,
+      "rewards/rejected": -0.0006865898030810058,
+      "step": 90
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 2.7064483636808314e-06,
+      "logits/chosen": 0.7685378789901733,
+      "logits/rejected": 0.8903138041496277,
+      "logps/chosen": -91.08631896972656,
+      "logps/rejected": -101.58308410644531,
+      "loss": 0.01,
+      "rewards/accuracies": 0.29374998807907104,
+      "rewards/chosen": -0.00034642056562006474,
+      "rewards/margins": 0.00024903417215682566,
+      "rewards/rejected": -0.0005954547086730599,
+      "step": 100
+    },
+    {
+      "epoch": 1.04,
+      "eval_logits/chosen": 0.8011398911476135,
+      "eval_logits/rejected": 0.8187842965126038,
+      "eval_logps/chosen": -91.76709747314453,
+      "eval_logps/rejected": -94.26233673095703,
+      "eval_loss": 0.009981811977922916,
+      "eval_rewards/accuracies": 0.25,
+      "eval_rewards/chosen": -0.00039719167398288846,
+      "eval_rewards/margins": 0.00025612558238208294,
+      "eval_rewards/rejected": -0.0006533172563649714,
+      "eval_runtime": 274.2663,
+      "eval_samples_per_second": 7.292,
+      "eval_steps_per_second": 0.456,
+      "step": 100
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 2.2478162071993296e-06,
+      "logits/chosen": 0.8606807589530945,
+      "logits/rejected": 0.9067083597183228,
+      "logps/chosen": -103.346435546875,
+      "logps/rejected": -102.74019622802734,
+      "loss": 0.0099,
+      "rewards/accuracies": 0.3125,
+      "rewards/chosen": 0.0002891735057346523,
+      "rewards/margins": 0.0007119966903701425,
+      "rewards/rejected": -0.00042282306822016835,
+      "step": 110
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 1.797672000566077e-06,
+      "logits/chosen": 0.8887661099433899,
+      "logits/rejected": 0.8524330258369446,
+      "logps/chosen": -92.60103607177734,
+      "logps/rejected": -70.87162780761719,
+      "loss": 0.0099,
+      "rewards/accuracies": 0.2874999940395355,
+      "rewards/chosen": 0.00012959113519173115,
+      "rewards/margins": 0.0010213626082986593,
+      "rewards/rejected": -0.0008917713421396911,
+      "step": 120
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 1.3711666042227772e-06,
+      "logits/chosen": 0.8812958598136902,
+      "logits/rejected": 0.9191433191299438,
+      "logps/chosen": -99.47935485839844,
+      "logps/rejected": -102.90433502197266,
+      "loss": 0.0099,
+      "rewards/accuracies": 0.33125001192092896,
+      "rewards/chosen": -0.00027921958826482296,
+      "rewards/margins": 0.00039964643656276166,
+      "rewards/rejected": -0.0006788660539314151,
+      "step": 130
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 9.826552484321086e-07,
+      "logits/chosen": 0.8100296258926392,
+      "logits/rejected": 0.8721168637275696,
+      "logps/chosen": -98.98599243164062,
+      "logps/rejected": -95.76078033447266,
+      "loss": 0.0099,
+      "rewards/accuracies": 0.29374998807907104,
+      "rewards/chosen": -0.0003388571203686297,
+      "rewards/margins": 0.0009376562084071338,
+      "rewards/rejected": -0.0012765133287757635,
+      "step": 140
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 6.452143679117965e-07,
+      "logits/chosen": 0.8696478009223938,
+      "logits/rejected": 0.884810745716095,
+      "logps/chosen": -74.74392700195312,
+      "logps/rejected": -80.85291290283203,
+      "loss": 0.0099,
+      "rewards/accuracies": 0.2874999940395355,
+      "rewards/chosen": -3.058182119275443e-05,
+      "rewards/margins": 0.0010159575613215566,
+      "rewards/rejected": -0.0010465392842888832,
+      "step": 150
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 3.7020147790418266e-07,
+      "logits/chosen": 0.8866029977798462,
+      "logits/rejected": 0.8834725618362427,
+      "logps/chosen": -105.7184066772461,
+      "logps/rejected": -99.87519836425781,
+      "loss": 0.0099,
+      "rewards/accuracies": 0.28125,
+      "rewards/chosen": -0.0007261586142703891,
+      "rewards/margins": 2.431169559713453e-05,
+      "rewards/rejected": -0.000750470208004117,
+      "step": 160
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 1.6687290528135725e-07,
+      "logits/chosen": 0.7714171409606934,
+      "logits/rejected": 0.8409671783447266,
+      "logps/chosen": -102.75898742675781,
+      "logps/rejected": -96.38053894042969,
+      "loss": 0.0099,
+      "rewards/accuracies": 0.3375000059604645,
+      "rewards/chosen": -0.0006845382740721107,
+      "rewards/margins": 0.00033002972486428916,
+      "rewards/rejected": -0.0010145680280402303,
+      "step": 170
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 4.207224101311247e-08,
+      "logits/chosen": 0.8822728395462036,
+      "logits/rejected": 0.935335636138916,
+      "logps/chosen": -119.57994079589844,
+      "logps/rejected": -112.05470275878906,
+      "loss": 0.0099,
+      "rewards/accuracies": 0.3125,
+      "rewards/chosen": -0.0007582681137137115,
+      "rewards/margins": 0.0003905483172275126,
+      "rewards/rejected": -0.001148816430941224,
+      "step": 180
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 0.0,
+      "logits/chosen": 0.8246952295303345,
+      "logits/rejected": 0.8706514239311218,
+      "logps/chosen": -73.35179138183594,
+      "logps/rejected": -76.91303253173828,
+      "loss": 0.0099,
+      "rewards/accuracies": 0.25,
+      "rewards/chosen": -0.0005546126631088555,
+      "rewards/margins": 0.0004530520236585289,
+      "rewards/rejected": -0.0010076647158712149,
+      "step": 190
+    },
+    {
+      "epoch": 1.98,
+      "step": 190,
+      "total_flos": 0.0,
+      "train_loss": 0.00995884225458691,
+      "train_runtime": 2568.9627,
+      "train_samples_per_second": 4.759,
+      "train_steps_per_second": 0.074
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 190,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2,
+  "save_steps": 100,
+  "total_flos": 0.0,
+  "train_batch_size": 4,
+  "trial_name": null,
+  "trial_params": null
+}