silanm commited on
Commit
0e1cc42
·
verified ·
1 Parent(s): 9c2eccf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +80 -80
README.md CHANGED
@@ -1,80 +1,80 @@
1
- ---
2
- library_name: transformers
3
- license: mit
4
- base_model: gpt2
5
- tags:
6
- - trl
7
- - dpo
8
- - generated_from_trainer
9
- model-index:
10
- - name: nlp-a5
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
- # nlp-a5
18
-
19
- This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on an unknown dataset.
20
- It achieves the following results on the evaluation set:
21
- - Loss: 0.6409
22
- - Rewards/chosen: 0.9778
23
- - Rewards/rejected: -2.1491
24
- - Rewards/accuracies: 0.8235
25
- - Rewards/margins: 3.1270
26
- - Logps/rejected: -410.6469
27
- - Logps/chosen: -337.3829
28
- - Logits/rejected: -66.9816
29
- - Logits/chosen: -67.8481
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: 5.38e-05
49
- - train_batch_size: 8
50
- - eval_batch_size: 8
51
- - seed: 42
52
- - gradient_accumulation_steps: 4
53
- - total_train_batch_size: 32
54
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
55
- - lr_scheduler_type: linear
56
- - lr_scheduler_warmup_steps: 50
57
- - training_steps: 500
58
-
59
- ### Training results
60
-
61
- | Training Loss | Epoch | Step | Validation Loss | Rewards/chosen | Rewards/rejected | Rewards/accuracies | Rewards/margins | Logps/rejected | Logps/chosen | Logits/rejected | Logits/chosen |
62
- |:-------------:|:------:|:----:|:---------------:|:--------------:|:----------------:|:------------------:|:---------------:|:--------------:|:------------:|:---------------:|:-------------:|
63
- | 0.6454 | 0.1382 | 50 | 0.7701 | 0.4667 | -1.3878 | 0.7591 | 1.8546 | -406.8403 | -339.9385 | -95.7163 | -95.3393 |
64
- | 0.7265 | 0.2764 | 100 | 0.7531 | 0.2791 | -2.1548 | 0.7777 | 2.4339 | -410.6752 | -340.8765 | -85.4456 | -85.2691 |
65
- | 0.5317 | 0.4147 | 150 | 0.7164 | 0.0401 | -2.6230 | 0.7743 | 2.6631 | -413.0164 | -342.0717 | -77.7900 | -78.4781 |
66
- | 0.8947 | 0.5529 | 200 | 0.7223 | -0.0327 | -3.1585 | 0.7961 | 3.1258 | -415.6938 | -342.4356 | -73.7223 | -74.3845 |
67
- | 0.6882 | 0.6911 | 250 | 0.6677 | 0.6186 | -2.0402 | 0.7904 | 2.6588 | -410.1023 | -339.1790 | -66.4183 | -67.2267 |
68
- | 0.4596 | 0.8293 | 300 | 0.6199 | 0.5863 | -2.4937 | 0.8116 | 3.0800 | -412.3698 | -339.3405 | -66.5151 | -67.2825 |
69
- | 0.6719 | 0.9675 | 350 | 0.6214 | 1.1018 | -1.4390 | 0.7842 | 2.5408 | -407.0965 | -336.7633 | -64.9415 | -65.8130 |
70
- | 0.119 | 1.1057 | 400 | 0.6442 | 0.4069 | -2.8694 | 0.8282 | 3.2763 | -414.2482 | -340.2375 | -64.6611 | -65.4554 |
71
- | 0.1427 | 1.2440 | 450 | 0.6730 | 1.1133 | -1.9897 | 0.8131 | 3.1030 | -409.8499 | -336.7056 | -65.8348 | -66.7287 |
72
- | 0.1022 | 1.3822 | 500 | 0.6409 | 0.9778 | -2.1491 | 0.8235 | 3.1270 | -410.6469 | -337.3829 | -66.9816 | -67.8481 |
73
-
74
-
75
- ### Framework versions
76
-
77
- - Transformers 4.45.0
78
- - Pytorch 2.4.0+cu124
79
- - Datasets 3.2.0
80
- - Tokenizers 0.20.3
 
1
+ ---
2
+ library_name: transformers
3
+ license: mit
4
+ base_model: gpt2
5
+ tags:
6
+ - trl
7
+ - dpo
8
+ - generated_from_trainer
9
+ model-index:
10
+ - name: nlp-a5
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
+ # nlp-a5
18
+
19
+ This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on [`distilabel-intel-orca-dpo-pairs`](https://huggingface.co/datasets/argilla/distilabel-intel-orca-dpo-pairs) dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.6409
22
+ - Rewards/chosen: 0.9778
23
+ - Rewards/rejected: -2.1491
24
+ - Rewards/accuracies: 0.8235
25
+ - Rewards/margins: 3.1270
26
+ - Logps/rejected: -410.6469
27
+ - Logps/chosen: -337.3829
28
+ - Logits/rejected: -66.9816
29
+ - Logits/chosen: -67.8481
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: 5.38e-05
49
+ - train_batch_size: 8
50
+ - eval_batch_size: 8
51
+ - seed: 42
52
+ - gradient_accumulation_steps: 4
53
+ - total_train_batch_size: 32
54
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
55
+ - lr_scheduler_type: linear
56
+ - lr_scheduler_warmup_steps: 50
57
+ - training_steps: 500
58
+
59
+ ### Training results
60
+
61
+ | Training Loss | Epoch | Step | Validation Loss | Rewards/chosen | Rewards/rejected | Rewards/accuracies | Rewards/margins | Logps/rejected | Logps/chosen | Logits/rejected | Logits/chosen |
62
+ |:-------------:|:------:|:----:|:---------------:|:--------------:|:----------------:|:------------------:|:---------------:|:--------------:|:------------:|:---------------:|:-------------:|
63
+ | 0.6454 | 0.1382 | 50 | 0.7701 | 0.4667 | -1.3878 | 0.7591 | 1.8546 | -406.8403 | -339.9385 | -95.7163 | -95.3393 |
64
+ | 0.7265 | 0.2764 | 100 | 0.7531 | 0.2791 | -2.1548 | 0.7777 | 2.4339 | -410.6752 | -340.8765 | -85.4456 | -85.2691 |
65
+ | 0.5317 | 0.4147 | 150 | 0.7164 | 0.0401 | -2.6230 | 0.7743 | 2.6631 | -413.0164 | -342.0717 | -77.7900 | -78.4781 |
66
+ | 0.8947 | 0.5529 | 200 | 0.7223 | -0.0327 | -3.1585 | 0.7961 | 3.1258 | -415.6938 | -342.4356 | -73.7223 | -74.3845 |
67
+ | 0.6882 | 0.6911 | 250 | 0.6677 | 0.6186 | -2.0402 | 0.7904 | 2.6588 | -410.1023 | -339.1790 | -66.4183 | -67.2267 |
68
+ | 0.4596 | 0.8293 | 300 | 0.6199 | 0.5863 | -2.4937 | 0.8116 | 3.0800 | -412.3698 | -339.3405 | -66.5151 | -67.2825 |
69
+ | 0.6719 | 0.9675 | 350 | 0.6214 | 1.1018 | -1.4390 | 0.7842 | 2.5408 | -407.0965 | -336.7633 | -64.9415 | -65.8130 |
70
+ | 0.119 | 1.1057 | 400 | 0.6442 | 0.4069 | -2.8694 | 0.8282 | 3.2763 | -414.2482 | -340.2375 | -64.6611 | -65.4554 |
71
+ | 0.1427 | 1.2440 | 450 | 0.6730 | 1.1133 | -1.9897 | 0.8131 | 3.1030 | -409.8499 | -336.7056 | -65.8348 | -66.7287 |
72
+ | 0.1022 | 1.3822 | 500 | 0.6409 | 0.9778 | -2.1491 | 0.8235 | 3.1270 | -410.6469 | -337.3829 | -66.9816 | -67.8481 |
73
+
74
+
75
+ ### Framework versions
76
+
77
+ - Transformers 4.45.0
78
+ - Pytorch 2.4.0+cu124
79
+ - Datasets 3.2.0
80
+ - Tokenizers 0.20.3