NicholasCorrado commited on
Commit
778d555
·
verified ·
1 Parent(s): 1a23120

End of training

Browse files
Files changed (4) hide show
  1. README.md +14 -8
  2. all_results.json +13 -0
  3. config.json +1 -1
  4. eval_results.json +16 -0
README.md CHANGED
@@ -3,9 +3,15 @@ library_name: transformers
3
  license: apache-2.0
4
  base_model: alignment-handbook/zephyr-7b-sft-full
5
  tags:
 
6
  - trl
7
  - dpo
8
  - generated_from_trainer
 
 
 
 
 
9
  model-index:
10
  - name: zephyr-7b-group-dpo-full-4
11
  results: []
@@ -16,17 +22,17 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  # zephyr-7b-group-dpo-full-4
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.3329
22
- - Rewards/chosen: -3.5971
23
- - Rewards/rejected: -6.6557
24
  - Rewards/accuracies: 0.8281
25
- - Rewards/margins: 3.0586
26
- - Logps/rejected: -955.4623
27
- - Logps/chosen: -642.2396
28
- - Logits/rejected: -0.6052
29
- - Logits/chosen: -1.1035
30
 
31
  ## Model description
32
 
 
3
  license: apache-2.0
4
  base_model: alignment-handbook/zephyr-7b-sft-full
5
  tags:
6
+ - alignment-handbook
7
  - trl
8
  - dpo
9
  - generated_from_trainer
10
+ - trl
11
+ - dpo
12
+ - generated_from_trainer
13
+ datasets:
14
+ - HuggingFaceH4/ultrafeedback_binarized
15
  model-index:
16
  - name: zephyr-7b-group-dpo-full-4
17
  results: []
 
22
 
23
  # zephyr-7b-group-dpo-full-4
24
 
25
+ 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 HuggingFaceH4/ultrafeedback_binarized dataset.
26
  It achieves the following results on the evaluation set:
27
  - Loss: 0.3329
28
+ - Rewards/chosen: -3.5718
29
+ - Rewards/rejected: -6.6220
30
  - Rewards/accuracies: 0.8281
31
+ - Rewards/margins: 3.0502
32
+ - Logps/rejected: -952.0885
33
+ - Logps/chosen: -639.7113
34
+ - Logits/rejected: -0.6186
35
+ - Logits/chosen: -1.1189
36
 
37
  ## Model description
38
 
all_results.json CHANGED
@@ -1,5 +1,18 @@
1
  {
2
  "epoch": 0.9994242947610823,
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  "total_flos": 0.0,
4
  "train_loss": 0.17276417467451316,
5
  "train_runtime": 14185.6914,
 
1
  {
2
  "epoch": 0.9994242947610823,
3
+ "eval_logits/chosen": -1.1188652515411377,
4
+ "eval_logits/rejected": -0.6186092495918274,
5
+ "eval_logps/chosen": -639.7113037109375,
6
+ "eval_logps/rejected": -952.0885009765625,
7
+ "eval_loss": 0.3328971564769745,
8
+ "eval_rewards/accuracies": 0.828125,
9
+ "eval_rewards/chosen": -3.571798324584961,
10
+ "eval_rewards/margins": 3.05015230178833,
11
+ "eval_rewards/rejected": -6.621951103210449,
12
+ "eval_runtime": 330.4232,
13
+ "eval_samples": 7126,
14
+ "eval_samples_per_second": 21.566,
15
+ "eval_steps_per_second": 0.339,
16
  "total_flos": 0.0,
17
  "train_loss": 0.17276417467451316,
18
  "train_runtime": 14185.6914,
config.json CHANGED
@@ -22,6 +22,6 @@
22
  "tie_word_embeddings": false,
23
  "torch_dtype": "bfloat16",
24
  "transformers_version": "4.44.1",
25
- "use_cache": false,
26
  "vocab_size": 32000
27
  }
 
22
  "tie_word_embeddings": false,
23
  "torch_dtype": "bfloat16",
24
  "transformers_version": "4.44.1",
25
+ "use_cache": true,
26
  "vocab_size": 32000
27
  }
eval_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 0.9994242947610823,
3
+ "eval_logits/chosen": -1.1188652515411377,
4
+ "eval_logits/rejected": -0.6186092495918274,
5
+ "eval_logps/chosen": -639.7113037109375,
6
+ "eval_logps/rejected": -952.0885009765625,
7
+ "eval_loss": 0.3328971564769745,
8
+ "eval_rewards/accuracies": 0.828125,
9
+ "eval_rewards/chosen": -3.571798324584961,
10
+ "eval_rewards/margins": 3.05015230178833,
11
+ "eval_rewards/rejected": -6.621951103210449,
12
+ "eval_runtime": 330.4232,
13
+ "eval_samples": 7126,
14
+ "eval_samples_per_second": 21.566,
15
+ "eval_steps_per_second": 0.339
16
+ }