YeongminKim commited on
Commit
4bf1a1a
·
verified ·
1 Parent(s): 00937f2

End of training

Browse files
Files changed (4) hide show
  1. README.md +15 -9
  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-dpo-full-alpha_0.5_batch64_0.03
11
  results: []
@@ -16,17 +22,17 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  # zephyr-7b-dpo-full-alpha_0.5_batch64_0.03
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.7788
22
- - Rewards/chosen: -1.3317
23
- - Rewards/rejected: -2.3491
24
  - Rewards/accuracies: 0.7857
25
- - Rewards/margins: 1.0174
26
- - Logps/rejected: -495.1149
27
- - Logps/chosen: -415.1518
28
- - Logits/rejected: 0.3728
29
- - Logits/chosen: -0.7858
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-dpo-full-alpha_0.5_batch64_0.03
17
  results: []
 
22
 
23
  # zephyr-7b-dpo-full-alpha_0.5_batch64_0.03
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.7789
28
+ - Rewards/chosen: -1.3273
29
+ - Rewards/rejected: -2.3422
30
  - Rewards/accuracies: 0.7857
31
+ - Rewards/margins: 1.0149
32
+ - Logps/rejected: -494.4228
33
+ - Logps/chosen: -414.7076
34
+ - Logits/rejected: 0.3577
35
+ - Logits/chosen: -0.7982
36
 
37
  ## Model description
38
 
all_results.json CHANGED
@@ -1,5 +1,18 @@
1
  {
2
  "epoch": 0.9994767137624281,
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  "total_flos": 0.0,
4
  "train_loss": 0.8168348412239115,
5
  "train_runtime": 34624.4791,
 
1
  {
2
  "epoch": 0.9994767137624281,
3
+ "eval_logits/chosen": -0.7981740236282349,
4
+ "eval_logits/rejected": 0.35771894454956055,
5
+ "eval_logps/chosen": -414.70758056640625,
6
+ "eval_logps/rejected": -494.42279052734375,
7
+ "eval_loss": 0.7788758277893066,
8
+ "eval_rewards/accuracies": 0.7857142686843872,
9
+ "eval_rewards/chosen": -1.3273053169250488,
10
+ "eval_rewards/margins": 1.0148944854736328,
11
+ "eval_rewards/rejected": -2.3421998023986816,
12
+ "eval_runtime": 408.2742,
13
+ "eval_samples": 2000,
14
+ "eval_samples_per_second": 4.899,
15
+ "eval_steps_per_second": 0.154,
16
  "total_flos": 0.0,
17
  "train_loss": 0.8168348412239115,
18
  "train_runtime": 34624.4791,
config.json CHANGED
@@ -22,6 +22,6 @@
22
  "tie_word_embeddings": false,
23
  "torch_dtype": "bfloat16",
24
  "transformers_version": "4.44.2",
25
- "use_cache": false,
26
  "vocab_size": 32000
27
  }
 
22
  "tie_word_embeddings": false,
23
  "torch_dtype": "bfloat16",
24
  "transformers_version": "4.44.2",
25
+ "use_cache": true,
26
  "vocab_size": 32000
27
  }
eval_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 0.9994767137624281,
3
+ "eval_logits/chosen": -0.7981740236282349,
4
+ "eval_logits/rejected": 0.35771894454956055,
5
+ "eval_logps/chosen": -414.70758056640625,
6
+ "eval_logps/rejected": -494.42279052734375,
7
+ "eval_loss": 0.7788758277893066,
8
+ "eval_rewards/accuracies": 0.7857142686843872,
9
+ "eval_rewards/chosen": -1.3273053169250488,
10
+ "eval_rewards/margins": 1.0148944854736328,
11
+ "eval_rewards/rejected": -2.3421998023986816,
12
+ "eval_runtime": 408.2742,
13
+ "eval_samples": 2000,
14
+ "eval_samples_per_second": 4.899,
15
+ "eval_steps_per_second": 0.154
16
+ }