btherien commited on
Commit
bca1e7d
1 Parent(s): 22385c4

End of training

Browse files
Files changed (2) hide show
  1. README.md +5 -1
  2. config.json +1 -1
README.md CHANGED
@@ -1,9 +1,13 @@
1
  ---
2
  base_model: btherien/410M_it-132366_Tr-slim-pajama-300B-replay5_finetune_sft-full
3
  tags:
 
 
4
  - trl
5
  - dpo
6
  - generated_from_trainer
 
 
7
  model-index:
8
  - name: 410M_it-132366_Tr-slim-pajama-300B-replay5_finetune_dpo-full
9
  results: []
@@ -14,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # 410M_it-132366_Tr-slim-pajama-300B-replay5_finetune_dpo-full
16
 
17
- This model is a fine-tuned version of [btherien/410M_it-132366_Tr-slim-pajama-300B-replay5_finetune_sft-full](https://huggingface.co/btherien/410M_it-132366_Tr-slim-pajama-300B-replay5_finetune_sft-full) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
  - Loss: 0.6500
20
  - Rewards/chosen: -0.4023
 
1
  ---
2
  base_model: btherien/410M_it-132366_Tr-slim-pajama-300B-replay5_finetune_sft-full
3
  tags:
4
+ - alignment-handbook
5
+ - generated_from_trainer
6
  - trl
7
  - dpo
8
  - generated_from_trainer
9
+ datasets:
10
+ - HuggingFaceH4/ultrafeedback_binarized
11
  model-index:
12
  - name: 410M_it-132366_Tr-slim-pajama-300B-replay5_finetune_dpo-full
13
  results: []
 
18
 
19
  # 410M_it-132366_Tr-slim-pajama-300B-replay5_finetune_dpo-full
20
 
21
+ This model is a fine-tuned version of [btherien/410M_it-132366_Tr-slim-pajama-300B-replay5_finetune_sft-full](https://huggingface.co/btherien/410M_it-132366_Tr-slim-pajama-300B-replay5_finetune_sft-full) on the HuggingFaceH4/ultrafeedback_binarized dataset.
22
  It achieves the following results on the evaluation set:
23
  - Loss: 0.6500
24
  - Rewards/chosen: -0.4023
config.json CHANGED
@@ -23,7 +23,7 @@
23
  "tie_word_embeddings": false,
24
  "torch_dtype": "bfloat16",
25
  "transformers_version": "4.36.2",
26
- "use_cache": false,
27
  "use_parallel_residual": true,
28
  "vocab_size": 50304
29
  }
 
23
  "tie_word_embeddings": false,
24
  "torch_dtype": "bfloat16",
25
  "transformers_version": "4.36.2",
26
+ "use_cache": true,
27
  "use_parallel_residual": true,
28
  "vocab_size": 50304
29
  }