haoranxu commited on
Commit
4f36d1f
1 Parent(s): 15c51bf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -75
README.md CHANGED
@@ -1,75 +1 @@
1
- ---
2
- license: llama3
3
- base_model: meta-llama/Meta-Llama-3-8B-Instruct
4
- tags:
5
- - alignment-handbook
6
- - generated_from_trainer
7
- datasets:
8
- - princeton-nlp/llama3-ultrafeedback
9
- model-index:
10
- - name: llama-3-8b-instruct-cpo-beta-0.1
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
- # llama-3-8b-instruct-cpo-beta-0.1
18
-
19
- This model is a fine-tuned version of [meta-llama/Meta-Llama-3-8B-Instruct](https://huggingface.co/meta-llama/Meta-Llama-3-8B-Instruct) on the princeton-nlp/llama3-ultrafeedback dataset.
20
- It achieves the following results on the evaluation set:
21
- - Loss: 1.3274
22
- - Rewards/chosen: -13.1885
23
- - Rewards/rejected: -14.3989
24
- - Rewards/accuracies: 0.7016
25
- - Rewards/margins: 1.2104
26
- - Logps/rejected: -143.9887
27
- - Logps/chosen: -131.8848
28
- - Logits/rejected: 0.2135
29
- - Logits/chosen: 0.1708
30
- - Nll Loss: 0.3287
31
-
32
- ## Model description
33
-
34
- More information needed
35
-
36
- ## Intended uses & limitations
37
-
38
- More information needed
39
-
40
- ## Training and evaluation data
41
-
42
- More information needed
43
-
44
- ## Training procedure
45
-
46
- ### Training hyperparameters
47
-
48
- The following hyperparameters were used during training:
49
- - learning_rate: 1e-06
50
- - train_batch_size: 2
51
- - eval_batch_size: 4
52
- - seed: 42
53
- - distributed_type: multi-GPU
54
- - num_devices: 16
55
- - gradient_accumulation_steps: 4
56
- - total_train_batch_size: 128
57
- - total_eval_batch_size: 64
58
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
59
- - lr_scheduler_type: cosine
60
- - lr_scheduler_warmup_ratio: 0.1
61
- - num_epochs: 1
62
-
63
- ### Training results
64
-
65
- | Training Loss | Epoch | Step | Validation Loss | Rewards/chosen | Rewards/rejected | Rewards/accuracies | Rewards/margins | Logps/rejected | Logps/chosen | Logits/rejected | Logits/chosen | Nll Loss |
66
- |:-------------:|:------:|:----:|:---------------:|:--------------:|:----------------:|:------------------:|:---------------:|:--------------:|:------------:|:---------------:|:-------------:|:--------:|
67
- | 1.2968 | 0.8547 | 400 | 1.3274 | -13.1885 | -14.3989 | 0.7016 | 1.2104 | -143.9887 | -131.8848 | 0.2135 | 0.1708 | 0.3287 |
68
-
69
-
70
- ### Framework versions
71
-
72
- - Transformers 4.41.2
73
- - Pytorch 2.3.0+rocm6.0
74
- - Datasets 2.19.2
75
- - Tokenizers 0.19.1
 
1
+ This repository contains the model release for the use of [CPO](https://arxiv.org/abs/2401.08417), please find more details in [our github](https://github.com/fe1ixxu/CPO_SIMPO)!