Minbyul commited on
Commit
6f74788
1 Parent(s): b820d93

End of training

Browse files
Files changed (4) hide show
  1. README.md +16 -10
  2. all_results.json +13 -0
  3. config.json +1 -1
  4. eval_results.json +16 -0
README.md CHANGED
@@ -1,9 +1,15 @@
1
  ---
2
  base_model: meta-llama/Llama-2-7b-hf
3
  tags:
 
4
  - trl
5
  - dpo
6
  - generated_from_trainer
 
 
 
 
 
7
  model-index:
8
  - name: llama2-7b-dpo-full-wo-live_qa-ep3
9
  results: []
@@ -14,17 +20,17 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # llama2-7b-dpo-full-wo-live_qa-ep3
16
 
17
- This model is a fine-tuned version of [meta-llama/Llama-2-7b-hf](https://huggingface.co/meta-llama/Llama-2-7b-hf) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 0.6297
20
- - Rewards/chosen: 0.0796
21
- - Rewards/rejected: -0.0955
22
- - Rewards/accuracies: 0.7679
23
- - Rewards/margins: 0.1751
24
- - Logps/rejected: -812.6376
25
- - Logps/chosen: -523.0222
26
- - Logits/rejected: -0.3805
27
- - Logits/chosen: -0.1364
28
 
29
  ## Model description
30
 
 
1
  ---
2
  base_model: meta-llama/Llama-2-7b-hf
3
  tags:
4
+ - alignment-handbook
5
  - trl
6
  - dpo
7
  - generated_from_trainer
8
+ - trl
9
+ - dpo
10
+ - generated_from_trainer
11
+ datasets:
12
+ - HuggingFaceH4/ultrafeedback_binarized
13
  model-index:
14
  - name: llama2-7b-dpo-full-wo-live_qa-ep3
15
  results: []
 
20
 
21
  # llama2-7b-dpo-full-wo-live_qa-ep3
22
 
23
+ This model is a fine-tuned version of [meta-llama/Llama-2-7b-hf](https://huggingface.co/meta-llama/Llama-2-7b-hf) on the HuggingFaceH4/ultrafeedback_binarized dataset.
24
  It achieves the following results on the evaluation set:
25
+ - Loss: 0.6073
26
+ - Rewards/chosen: 0.0705
27
+ - Rewards/rejected: -0.1795
28
+ - Rewards/accuracies: 0.8214
29
+ - Rewards/margins: 0.2499
30
+ - Logps/rejected: -821.0361
31
+ - Logps/chosen: -523.9339
32
+ - Logits/rejected: -0.3815
33
+ - Logits/chosen: -0.1586
34
 
35
  ## Model description
36
 
all_results.json CHANGED
@@ -1,5 +1,18 @@
1
  {
2
  "epoch": 1.0,
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  "train_loss": 0.6189163742643414,
4
  "train_runtime": 2381.9724,
5
  "train_samples": 10591,
 
1
  {
2
  "epoch": 1.0,
3
+ "eval_logits/chosen": -0.15863379836082458,
4
+ "eval_logits/rejected": -0.3815014362335205,
5
+ "eval_logps/chosen": -523.9338989257812,
6
+ "eval_logps/rejected": -821.0360717773438,
7
+ "eval_loss": 0.6073186993598938,
8
+ "eval_rewards/accuracies": 0.8214285969734192,
9
+ "eval_rewards/chosen": 0.07047966867685318,
10
+ "eval_rewards/margins": 0.24993744492530823,
11
+ "eval_rewards/rejected": -0.17945778369903564,
12
+ "eval_runtime": 22.8385,
13
+ "eval_samples": 208,
14
+ "eval_samples_per_second": 9.107,
15
+ "eval_steps_per_second": 0.307,
16
  "train_loss": 0.6189163742643414,
17
  "train_runtime": 2381.9724,
18
  "train_samples": 10591,
config.json CHANGED
@@ -23,6 +23,6 @@
23
  "tie_word_embeddings": false,
24
  "torch_dtype": "bfloat16",
25
  "transformers_version": "4.39.0.dev0",
26
- "use_cache": false,
27
  "vocab_size": 32000
28
  }
 
23
  "tie_word_embeddings": false,
24
  "torch_dtype": "bfloat16",
25
  "transformers_version": "4.39.0.dev0",
26
+ "use_cache": true,
27
  "vocab_size": 32000
28
  }
eval_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_logits/chosen": -0.15863379836082458,
4
+ "eval_logits/rejected": -0.3815014362335205,
5
+ "eval_logps/chosen": -523.9338989257812,
6
+ "eval_logps/rejected": -821.0360717773438,
7
+ "eval_loss": 0.6073186993598938,
8
+ "eval_rewards/accuracies": 0.8214285969734192,
9
+ "eval_rewards/chosen": 0.07047966867685318,
10
+ "eval_rewards/margins": 0.24993744492530823,
11
+ "eval_rewards/rejected": -0.17945778369903564,
12
+ "eval_runtime": 22.8385,
13
+ "eval_samples": 208,
14
+ "eval_samples_per_second": 9.107,
15
+ "eval_steps_per_second": 0.307
16
+ }