Minbyul commited on
Commit
398deda
1 Parent(s): 5a4a1e6

End of training

Browse files
Files changed (4) hide show
  1. README.md +17 -1
  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-healthsearch_qa-ep3
9
  results: []
@@ -14,7 +20,17 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # llama2-7b-dpo-full-wo-healthsearch_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
 
19
  ## Model description
20
 
 
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-healthsearch_qa-ep3
15
  results: []
 
20
 
21
  # llama2-7b-dpo-full-wo-healthsearch_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.6764
26
+ - Rewards/chosen: 0.0307
27
+ - Rewards/rejected: -0.0060
28
+ - Rewards/accuracies: 0.7173
29
+ - Rewards/margins: 0.0367
30
+ - Logps/rejected: -867.9213
31
+ - Logps/chosen: -604.5544
32
+ - Logits/rejected: -0.5562
33
+ - Logits/chosen: 0.0445
34
 
35
  ## Model description
36
 
all_results.json CHANGED
@@ -1,5 +1,18 @@
1
  {
2
  "epoch": 1.0,
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  "train_loss": 0.6753969060050117,
4
  "train_runtime": 1121.4796,
5
  "train_samples": 5168,
 
1
  {
2
  "epoch": 1.0,
3
+ "eval_logits/chosen": 0.04453304782509804,
4
+ "eval_logits/rejected": -0.5562469959259033,
5
+ "eval_logps/chosen": -604.554443359375,
6
+ "eval_logps/rejected": -867.9213256835938,
7
+ "eval_loss": 0.6763553023338318,
8
+ "eval_rewards/accuracies": 0.7173295617103577,
9
+ "eval_rewards/chosen": 0.030703268945217133,
10
+ "eval_rewards/margins": 0.03669063001871109,
11
+ "eval_rewards/rejected": -0.005987361539155245,
12
+ "eval_runtime": 622.4746,
13
+ "eval_samples": 5631,
14
+ "eval_samples_per_second": 9.046,
15
+ "eval_steps_per_second": 0.283,
16
  "train_loss": 0.6753969060050117,
17
  "train_runtime": 1121.4796,
18
  "train_samples": 5168,
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.04453304782509804,
4
+ "eval_logits/rejected": -0.5562469959259033,
5
+ "eval_logps/chosen": -604.554443359375,
6
+ "eval_logps/rejected": -867.9213256835938,
7
+ "eval_loss": 0.6763553023338318,
8
+ "eval_rewards/accuracies": 0.7173295617103577,
9
+ "eval_rewards/chosen": 0.030703268945217133,
10
+ "eval_rewards/margins": 0.03669063001871109,
11
+ "eval_rewards/rejected": -0.005987361539155245,
12
+ "eval_runtime": 622.4746,
13
+ "eval_samples": 5631,
14
+ "eval_samples_per_second": 9.046,
15
+ "eval_steps_per_second": 0.283
16
+ }