sedrickkeh commited on
Commit
9b7d0e5
·
verified ·
1 Parent(s): 79e9614

End of training

Browse files
README.md CHANGED
@@ -3,6 +3,8 @@ library_name: transformers
3
  license: gemma
4
  base_model: google/gemma-2-9b-it
5
  tags:
 
 
6
  - trl
7
  - dpo
8
  - llama-factory
@@ -17,7 +19,7 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  # gemma-oh-preferences
19
 
20
- This model is a fine-tuned version of [google/gemma-2-9b-it](https://huggingface.co/google/gemma-2-9b-it) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
  - Loss: 3.9424
23
  - Rewards/chosen: -46.0706
 
3
  license: gemma
4
  base_model: google/gemma-2-9b-it
5
  tags:
6
+ - llama-factory
7
+ - full
8
  - trl
9
  - dpo
10
  - llama-factory
 
19
 
20
  # gemma-oh-preferences
21
 
22
+ This model is a fine-tuned version of [google/gemma-2-9b-it](https://huggingface.co/google/gemma-2-9b-it) on the cdc0b2d9-493b-4cb1-87e8-8fb1e3f4b247 dataset.
23
  It achieves the following results on the evaluation set:
24
  - Loss: 3.9424
25
  - Rewards/chosen: -46.0706
all_results.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_logits/chosen": 14.471516609191895,
4
+ "eval_logits/rejected": 14.6997652053833,
5
+ "eval_logps/chosen": -4.607059001922607,
6
+ "eval_logps/rejected": -4.770802974700928,
7
+ "eval_loss": 3.942422866821289,
8
+ "eval_rewards/accuracies": 0.6459948420524597,
9
+ "eval_rewards/chosen": -46.07059097290039,
10
+ "eval_rewards/margins": 1.6374331712722778,
11
+ "eval_rewards/rejected": -47.7080192565918,
12
+ "eval_runtime": 439.7489,
13
+ "eval_samples_per_second": 112.507,
14
+ "eval_steps_per_second": 1.76,
15
+ "total_flos": 4181538261958656.0,
16
+ "train_loss": 4.105394068821323,
17
+ "train_runtime": 31326.6745,
18
+ "train_samples_per_second": 30.007,
19
+ "train_steps_per_second": 0.234
20
+ }
eval_results.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_logits/chosen": 14.471516609191895,
4
+ "eval_logits/rejected": 14.6997652053833,
5
+ "eval_logps/chosen": -4.607059001922607,
6
+ "eval_logps/rejected": -4.770802974700928,
7
+ "eval_loss": 3.942422866821289,
8
+ "eval_rewards/accuracies": 0.6459948420524597,
9
+ "eval_rewards/chosen": -46.07059097290039,
10
+ "eval_rewards/margins": 1.6374331712722778,
11
+ "eval_rewards/rejected": -47.7080192565918,
12
+ "eval_runtime": 439.7489,
13
+ "eval_samples_per_second": 112.507,
14
+ "eval_steps_per_second": 1.76
15
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "total_flos": 4181538261958656.0,
4
+ "train_loss": 4.105394068821323,
5
+ "train_runtime": 31326.6745,
6
+ "train_samples_per_second": 30.007,
7
+ "train_steps_per_second": 0.234
8
+ }
trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
training_eval_loss.png ADDED
training_loss.png ADDED
training_rewards_accuracies.png ADDED