Hugging Face's logo

EllieS
/
zephyr-7b-dpo-lora-pubmedqa-mix2

PEFT
TensorBoard
Safetensors
mistral
alignment-handbook
Generated from Trainer
trl
dpo
Model card Files Metrics Community
EllieS commited on
Commit
8f17533
•
1 Parent(s): c056b96

Model save

Browse files
Files changed (8) hide show
  1. README.md +101 -0
  2. adapter_model.safetensors +1 -1
  3. all_results.json +21 -0
  4. eval_results.json +16 -0
  5. runs/Mar03_06-19-23_586cb8b6da8c/events.out.tfevents.1709446868.586cb8b6da8c.24054.0 +2 -2
  6. runs/Mar03_06-19-23_586cb8b6da8c/events.out.tfevents.1709583061.586cb8b6da8c.24054.1 +3 -0
  7. train_results.json +8 -0
  8. trainer_state.json +0 -0
README.md ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: peft
4
+ tags:
5
+ - trl
6
+ - dpo
7
+ - generated_from_trainer
8
+ base_model: alignment-handbook/zephyr-7b-sft-full
9
+ model-index:
10
+ - name: zephyr-7b-dpo-lora-pubmedqa-mix2
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # zephyr-7b-dpo-lora-pubmedqa-mix2
18
+
19
+ 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.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.0013
22
+ - Rewards/chosen: -1.8126
23
+ - Rewards/rejected: -10.9731
24
+ - Rewards/accuracies: 1.0
25
+ - Rewards/margins: 9.1605
26
+ - Logps/rejected: -1144.0397
27
+ - Logps/chosen: -242.4412
28
+ - Logits/rejected: -1.7638
29
+ - Logits/chosen: -2.8841
30
+
31
+ ## Model description
32
+
33
+ More information needed
34
+
35
+ ## Intended uses & limitations
36
+
37
+ More information needed
38
+
39
+ ## Training and evaluation data
40
+
41
+ More information needed
42
+
43
+ ## Training procedure
44
+
45
+ ### Training hyperparameters
46
+
47
+ The following hyperparameters were used during training:
48
+ - learning_rate: 5e-06
49
+ - train_batch_size: 1
50
+ - eval_batch_size: 1
51
+ - seed: 42
52
+ - distributed_type: multi-GPU
53
+ - num_devices: 2
54
+ - gradient_accumulation_steps: 2
55
+ - total_train_batch_size: 4
56
+ - total_eval_batch_size: 2
57
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
58
+ - lr_scheduler_type: cosine
59
+ - lr_scheduler_warmup_ratio: 0.1
60
+ - num_epochs: 1
61
+
62
+ ### Training results
63
+
64
+ | Training Loss | Epoch | Step | Validation Loss | Rewards/chosen | Rewards/rejected | Rewards/accuracies | Rewards/margins | Logps/rejected | Logps/chosen | Logits/rejected | Logits/chosen |
65
+ |:-------------:|:-----:|:-----:|:---------------:|:--------------:|:----------------:|:------------------:|:---------------:|:--------------:|:------------:|:---------------:|:-------------:|
66
+ | 0.2697 | 0.04 | 3000 | 0.3396 | 0.2213 | -0.6386 | 1.0 | 0.8599 | -110.5876 | -39.0518 | -3.0278 | -3.0862 |
67
+ | 0.1599 | 0.07 | 6000 | 0.0750 | -0.5884 | -3.6673 | 1.0 | 3.0789 | -413.4546 | -120.0204 | -2.9055 | -3.0346 |
68
+ | 0.0563 | 0.11 | 9000 | 0.0204 | -0.6260 | -5.6712 | 1.0 | 5.0452 | -613.8441 | -123.7819 | -3.0269 | -3.1136 |
69
+ | 0.0463 | 0.14 | 12000 | 0.0287 | -0.7209 | -7.9224 | 1.0 | 7.2014 | -838.9609 | -133.2740 | -3.0642 | -3.1628 |
70
+ | 0.1206 | 0.18 | 15000 | 0.0030 | -0.9209 | -8.8089 | 1.0 | 7.8880 | -927.6118 | -153.2670 | -3.0802 | -3.1766 |
71
+ | 0.0508 | 0.22 | 18000 | 0.4964 | -0.4026 | -8.0330 | 1.0 | 7.6304 | -850.0245 | -101.4397 | -3.1314 | -3.2075 |
72
+ | 0.0323 | 0.25 | 21000 | 0.0872 | -1.4713 | -10.3437 | 1.0 | 8.8723 | -1081.0913 | -208.3129 | -2.6496 | -3.1189 |
73
+ | 0.4534 | 0.29 | 24000 | 0.0077 | -2.3507 | -12.1827 | 1.0 | 9.8320 | -1264.9957 | -296.2491 | -1.6282 | -2.8665 |
74
+ | 0.0013 | 0.32 | 27000 | 0.0019 | -2.1480 | -10.6645 | 1.0 | 8.5166 | -1113.1797 | -275.9768 | -1.7614 | -2.8604 |
75
+ | 0.1404 | 0.36 | 30000 | 0.0002 | -2.4964 | -12.4101 | 1.0 | 9.9138 | -1287.7384 | -310.8155 | -1.5907 | -2.8352 |
76
+ | 0.0198 | 0.4 | 33000 | 0.0009 | -3.0802 | -13.3347 | 1.0 | 10.2545 | -1380.1964 | -369.1991 | -1.6628 | -2.8372 |
77
+ | 0.0041 | 0.43 | 36000 | 0.0004 | -2.7800 | -12.5815 | 1.0 | 9.8014 | -1304.8732 | -339.1852 | -1.6282 | -2.8242 |
78
+ | 0.0007 | 0.47 | 39000 | 0.0007 | -2.9921 | -13.2089 | 1.0 | 10.2168 | -1367.6129 | -360.3922 | -1.6672 | -2.8403 |
79
+ | 0.0008 | 0.5 | 42000 | 0.0013 | -2.3107 | -11.8754 | 1.0 | 9.5647 | -1234.2609 | -292.2454 | -1.6475 | -2.8400 |
80
+ | 0.0024 | 0.54 | 45000 | 0.0010 | -3.3769 | -13.2333 | 1.0 | 9.8564 | -1370.0538 | -398.8731 | -1.6937 | -2.8403 |
81
+ | 0.0019 | 0.57 | 48000 | 0.0013 | -2.8151 | -12.4427 | 1.0 | 9.6277 | -1290.9999 | -342.6892 | -1.7047 | -2.8503 |
82
+ | 0.2266 | 0.61 | 51000 | 0.0014 | -1.9532 | -11.0212 | 1.0 | 9.0680 | -1148.8468 | -256.4992 | -1.6745 | -2.8650 |
83
+ | 0.0016 | 0.65 | 54000 | 0.0014 | -1.8077 | -10.7512 | 1.0 | 8.9435 | -1121.8423 | -241.9466 | -1.8328 | -2.8946 |
84
+ | 0.0019 | 0.68 | 57000 | 0.0013 | -1.8159 | -10.8808 | 1.0 | 9.0649 | -1134.8024 | -242.7715 | -1.7644 | -2.8860 |
85
+ | 0.0013 | 0.72 | 60000 | 0.0013 | -1.7356 | -10.8007 | 1.0 | 9.0651 | -1126.8002 | -234.7419 | -1.7574 | -2.8871 |
86
+ | 0.0014 | 0.75 | 63000 | 0.0013 | -1.8249 | -10.9773 | 1.0 | 9.1524 | -1144.4586 | -243.6743 | -1.7699 | -2.8867 |
87
+ | 0.0014 | 0.79 | 66000 | 0.0013 | -1.8308 | -10.9698 | 1.0 | 9.1389 | -1143.7017 | -244.2651 | -1.7597 | -2.8841 |
88
+ | 0.0011 | 0.83 | 69000 | 0.0013 | -1.8034 | -10.9390 | 1.0 | 9.1356 | -1140.6276 | -241.5220 | -1.7619 | -2.8858 |
89
+ | 0.0016 | 0.86 | 72000 | 0.0013 | -1.7971 | -10.9097 | 1.0 | 9.1126 | -1137.6914 | -240.8868 | -1.7608 | -2.8852 |
90
+ | 0.0239 | 0.9 | 75000 | 0.0013 | -1.7976 | -10.9400 | 1.0 | 9.1424 | -1140.7238 | -240.9355 | -1.7773 | -2.8872 |
91
+ | 0.0024 | 0.93 | 78000 | 0.0013 | -1.7862 | -10.9196 | 1.0 | 9.1334 | -1138.6901 | -239.8036 | -1.7733 | -2.8861 |
92
+ | 0.0018 | 0.97 | 81000 | 0.0013 | -1.8228 | -10.9802 | 1.0 | 9.1574 | -1144.7491 | -243.4639 | -1.7594 | -2.8860 |
93
+
94
+
95
+ ### Framework versions
96
+
97
+ - PEFT 0.7.1
98
+ - Transformers 4.36.2
99
+ - Pytorch 2.1.2+cu121
100
+ - Datasets 2.14.6
101
+ - Tokenizers 0.15.2
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:081c384c47555af4b4c93b0d32893b4a2575c6f4deae65893a8a80ae19dc9cfe
3
  size 83946192
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7aadcb01398f9926ef3ab1b7d873281709bc3d83632b96a18ef7d6293f2b69e2
3
  size 83946192
all_results.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_logits/chosen": -2.884117841720581,
4
+ "eval_logits/rejected": -1.7638154029846191,
5
+ "eval_logps/chosen": -242.4412078857422,
6
+ "eval_logps/rejected": -1144.0396728515625,
7
+ "eval_loss": 0.0013236103113740683,
8
+ "eval_rewards/accuracies": 1.0,
9
+ "eval_rewards/chosen": -1.812609314918518,
10
+ "eval_rewards/margins": 9.16053295135498,
11
+ "eval_rewards/rejected": -10.97314167022705,
12
+ "eval_runtime": 1.0748,
13
+ "eval_samples": 5,
14
+ "eval_samples_per_second": 4.652,
15
+ "eval_steps_per_second": 2.791,
16
+ "train_loss": 0.05805659896943479,
17
+ "train_runtime": 136192.1568,
18
+ "train_samples": 334157,
19
+ "train_samples_per_second": 2.454,
20
+ "train_steps_per_second": 0.613
21
+ }
eval_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_logits/chosen": -2.884117841720581,
4
+ "eval_logits/rejected": -1.7638154029846191,
5
+ "eval_logps/chosen": -242.4412078857422,
6
+ "eval_logps/rejected": -1144.0396728515625,
7
+ "eval_loss": 0.0013236103113740683,
8
+ "eval_rewards/accuracies": 1.0,
9
+ "eval_rewards/chosen": -1.812609314918518,
10
+ "eval_rewards/margins": 9.16053295135498,
11
+ "eval_rewards/rejected": -10.97314167022705,
12
+ "eval_runtime": 1.0748,
13
+ "eval_samples": 5,
14
+ "eval_samples_per_second": 4.652,
15
+ "eval_steps_per_second": 2.791
16
+ }
runs/Mar03_06-19-23_586cb8b6da8c/events.out.tfevents.1709446868.586cb8b6da8c.24054.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c2bc491a60f304d41702d7fb5dc4ad25d080803d8df2828bd60c0997e51fb177
3
- size 5360843
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0bff45a07510cf196e3d4dc4e23a7860c21531d42ea8ce4a03c4e77b7ed22c4
3
+ size 5395388
runs/Mar03_06-19-23_586cb8b6da8c/events.out.tfevents.1709583061.586cb8b6da8c.24054.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b8055e1cca072bfefdbb826264c4f782e3453eef1a9243cd85dc756f5941960a
3
+ size 841
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "train_loss": 0.05805659896943479,
4
+ "train_runtime": 136192.1568,
5
+ "train_samples": 334157,
6
+ "train_samples_per_second": 2.454,
7
+ "train_steps_per_second": 0.613
8
+ }
trainer_state.json ADDED
The diff for this file is too large to render. See raw diff