End of training
Browse files
README.md
CHANGED
@@ -1,17 +1,20 @@
|
|
1 |
---
|
2 |
base_model: mistralai/Mistral-7B-v0.1
|
|
|
3 |
library_name: transformers
|
4 |
-
model_name:
|
5 |
tags:
|
6 |
- generated_from_trainer
|
|
|
|
|
7 |
- trl
|
8 |
- dpo
|
9 |
licence: license
|
10 |
---
|
11 |
|
12 |
-
# Model Card for
|
13 |
|
14 |
-
This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1).
|
15 |
It has been trained using [TRL](https://github.com/huggingface/trl).
|
16 |
|
17 |
## Quick start
|
|
|
1 |
---
|
2 |
base_model: mistralai/Mistral-7B-v0.1
|
3 |
+
datasets: HuggingFaceH4/ultrafeedback_binarized
|
4 |
library_name: transformers
|
5 |
+
model_name: mistralai/Mistral-7B-v0.1
|
6 |
tags:
|
7 |
- generated_from_trainer
|
8 |
+
- alignment-handbook
|
9 |
+
- HuggingFaceH4/ultrafeedback_binarized
|
10 |
- trl
|
11 |
- dpo
|
12 |
licence: license
|
13 |
---
|
14 |
|
15 |
+
# Model Card for mistralai/Mistral-7B-v0.1
|
16 |
|
17 |
+
This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on the [HuggingFaceH4/ultrafeedback_binarized](https://huggingface.co/datasets/HuggingFaceH4/ultrafeedback_binarized) dataset.
|
18 |
It has been trained using [TRL](https://github.com/huggingface/trl).
|
19 |
|
20 |
## Quick start
|
all_results.json
CHANGED
@@ -1,5 +1,18 @@
|
|
1 |
{
|
2 |
"epoch": 1.0,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
"total_flos": 0.0,
|
4 |
"train_loss": 0.6713338216145833,
|
5 |
"train_runtime": 13324.2114,
|
|
|
1 |
{
|
2 |
"epoch": 1.0,
|
3 |
+
"eval_logits/chosen": -3.0115926265716553,
|
4 |
+
"eval_logits/rejected": -3.022681474685669,
|
5 |
+
"eval_logps/chosen": -266.7742004394531,
|
6 |
+
"eval_logps/rejected": -253.80645751953125,
|
7 |
+
"eval_loss": 0.6836252808570862,
|
8 |
+
"eval_rewards/accuracies": 0.5225093364715576,
|
9 |
+
"eval_rewards/chosen": -0.020729802548885345,
|
10 |
+
"eval_rewards/margins": 0.020151691511273384,
|
11 |
+
"eval_rewards/rejected": -0.040862053632736206,
|
12 |
+
"eval_runtime": 165.8484,
|
13 |
+
"eval_samples": 2000,
|
14 |
+
"eval_samples_per_second": 12.21,
|
15 |
+
"eval_steps_per_second": 0.386,
|
16 |
"total_flos": 0.0,
|
17 |
"train_loss": 0.6713338216145833,
|
18 |
"train_runtime": 13324.2114,
|
eval_results.json
ADDED
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"epoch": 1.0,
|
3 |
+
"eval_logits/chosen": -3.0115926265716553,
|
4 |
+
"eval_logits/rejected": -3.022681474685669,
|
5 |
+
"eval_logps/chosen": -266.7742004394531,
|
6 |
+
"eval_logps/rejected": -253.80645751953125,
|
7 |
+
"eval_loss": 0.6836252808570862,
|
8 |
+
"eval_rewards/accuracies": 0.5225093364715576,
|
9 |
+
"eval_rewards/chosen": -0.020729802548885345,
|
10 |
+
"eval_rewards/margins": 0.020151691511273384,
|
11 |
+
"eval_rewards/rejected": -0.040862053632736206,
|
12 |
+
"eval_runtime": 165.8484,
|
13 |
+
"eval_samples": 2000,
|
14 |
+
"eval_samples_per_second": 12.21,
|
15 |
+
"eval_steps_per_second": 0.386
|
16 |
+
}
|
runs/May25_22-42-33_cn-g009.server.mila.quebec/events.out.tfevents.1748242384.cn-g009.server.mila.quebec.3277230.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e2760dec0e9d94f9e27fb889aba097dc13697229c09137188360215ed2a77de5
|
3 |
+
size 815
|