ZhangShenao commited on
Commit
6f2aeea
1 Parent(s): 319216b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -27
README.md CHANGED
@@ -1,47 +1,46 @@
1
  ---
2
  license: mit
3
- base_model: ZhangShenao/0.001_2SELM_Zephyr_iter_1
4
  tags:
5
  - alignment-handbook
6
- - trl
7
  - dpo
8
- - generated_from_trainer
9
  - trl
10
- - dpo
11
- - generated_from_trainer
12
  datasets:
13
- - updated
14
- - original
15
  model-index:
16
- - name: 0.001_2SELM_Zephyr_iter_2
17
  results: []
18
  ---
19
 
20
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
21
  should probably proofread and complete it, then remove this comment. -->
22
 
23
- # 0.001_2SELM_Zephyr_iter_2
24
-
25
- This model is a fine-tuned version of [ZhangShenao/0.001_2SELM_Zephyr_iter_1](https://huggingface.co/ZhangShenao/0.001_2SELM_Zephyr_iter_1) on the updated and the original datasets.
26
-
27
- ## Model description
28
 
29
- More information needed
30
 
31
- ## Intended uses & limitations
32
 
33
- More information needed
34
 
35
- ## Training and evaluation data
 
36
 
37
- More information needed
38
 
39
- ## Training procedure
 
 
 
 
 
40
 
41
  ### Training hyperparameters
42
 
43
  The following hyperparameters were used during training:
44
- - learning_rate: 5e-07
 
45
  - train_batch_size: 8
46
  - eval_batch_size: 8
47
  - seed: 42
@@ -49,16 +48,9 @@ The following hyperparameters were used during training:
49
  - num_devices: 8
50
  - gradient_accumulation_steps: 4
51
  - total_train_batch_size: 256
52
- - total_eval_batch_size: 64
53
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
54
- - lr_scheduler_type: cosine
55
- - lr_scheduler_warmup_ratio: 0.1
56
  - num_epochs: 1
57
 
58
- ### Training results
59
-
60
-
61
-
62
  ### Framework versions
63
 
64
  - Transformers 4.40.2
 
1
  ---
2
  license: mit
3
+ base_model: ZhangShenao/DPO-Zephyr-7B
4
  tags:
5
  - alignment-handbook
 
6
  - dpo
 
7
  - trl
8
+ - selm
 
9
  datasets:
10
+ - HuggingFaceH4/ultrafeedback_binarized
 
11
  model-index:
12
+ - name: SELM-Zephyr-7B-iter-1
13
  results: []
14
  ---
15
 
16
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
  should probably proofread and complete it, then remove this comment. -->
18
 
19
+ Self-Exploring Language Models: Active Preference Elicitation for Online Alignment.
 
 
 
 
20
 
21
+ # SELM-Zephyr-7B-iter-1
22
 
23
+ This model is a fine-tuned version of [ZhangShenao/DPO-Zephyr-7B](https://huggingface.co/ZhangShenao/DPO-Zephyr-7B) using synthetic data based on on the HuggingFaceH4/ultrafeedback_binarized dataset.
24
 
25
+ ## Model description
26
 
27
+ - Model type: A 7B parameter Zephyr-based Self-Exploring Language Models (SELM).
28
+ - License: MIT
29
 
30
+ ## Results
31
 
32
+ | | AlpacaEval 2.0 (LC WR) | MT-Bench (Average) |
33
+ |----------------------------------------|------------------------|--------------------|
34
+ | [SELM-Zephyr-7B-iter-3](https://huggingface.co/ZhangShenao/SELM-Zephyr-7B-iter-3) | &emsp; &emsp; &emsp;&emsp; 24.00 | &emsp; &emsp; &emsp; 7.48 |
35
+ | [SELM-Zephyr-7B-iter-2](https://huggingface.co/ZhangShenao/SELM-Zephyr-7B-iter-2) | &emsp; &emsp; &emsp;&emsp; 23.40 | &emsp; &emsp; &emsp; 7.72 |
36
+ | [SELM-Zephyr-7B-iter-1](https://huggingface.co/ZhangShenao/SELM-Zephyr-7B-iter-1) | &emsp; &emsp; &emsp;&emsp; 20.28 | &emsp; &emsp; &emsp; 7.42 |
37
+ | [DPO-Zephyr-7B](https://huggingface.co/ZhangShenao/DPO-Zephyr-7B) | &emsp; &emsp; &emsp;&emsp; 14.45 | &emsp; &emsp; &emsp; 7.28 |
38
 
39
  ### Training hyperparameters
40
 
41
  The following hyperparameters were used during training:
42
+ - alpha: 0.001
43
+ - beta: 0.01
44
  - train_batch_size: 8
45
  - eval_batch_size: 8
46
  - seed: 42
 
48
  - num_devices: 8
49
  - gradient_accumulation_steps: 4
50
  - total_train_batch_size: 256
 
51
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 
 
52
  - num_epochs: 1
53
 
 
 
 
 
54
  ### Framework versions
55
 
56
  - Transformers 4.40.2