erbacher commited on
Commit
aec00ac
1 Parent(s): 347e301

Model save

Browse files
Files changed (5) hide show
  1. README.md +71 -0
  2. all_results.json +13 -0
  3. eval_results.json +8 -0
  4. train_results.json +8 -0
  5. trainer_state.json +0 -0
README.md ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: mistralai/Mistral-7B-v0.1
4
+ tags:
5
+ - generated_from_trainer
6
+ model-index:
7
+ - name: mistral-convsearch-7b-v4
8
+ results: []
9
+ ---
10
+
11
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
12
+ should probably proofread and complete it, then remove this comment. -->
13
+
14
+ # mistral-convsearch-7b-v4
15
+
16
+ This model is a fine-tuned version of [mistralai/Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1) on an unknown dataset.
17
+ It achieves the following results on the evaluation set:
18
+ - Loss: 0.4812
19
+
20
+ ## Model description
21
+
22
+ More information needed
23
+
24
+ ## Intended uses & limitations
25
+
26
+ More information needed
27
+
28
+ ## Training and evaluation data
29
+
30
+ More information needed
31
+
32
+ ## Training procedure
33
+
34
+ ### Training hyperparameters
35
+
36
+ The following hyperparameters were used during training:
37
+ - learning_rate: 0.0002
38
+ - train_batch_size: 8
39
+ - eval_batch_size: 4
40
+ - seed: 42
41
+ - distributed_type: multi-GPU
42
+ - num_devices: 2
43
+ - gradient_accumulation_steps: 8
44
+ - total_train_batch_size: 128
45
+ - total_eval_batch_size: 8
46
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
+ - lr_scheduler_type: cosine
48
+ - num_epochs: 10
49
+
50
+ ### Training results
51
+
52
+ | Training Loss | Epoch | Step | Validation Loss |
53
+ |:-------------:|:-----:|:----:|:---------------:|
54
+ | 0.5117 | 0.63 | 137 | 0.5122 |
55
+ | 0.4733 | 1.63 | 275 | 0.4869 |
56
+ | 0.4585 | 2.63 | 413 | 0.4761 |
57
+ | 0.4408 | 3.63 | 551 | 0.4705 |
58
+ | 0.4266 | 4.63 | 689 | 0.4692 |
59
+ | 0.4087 | 5.63 | 827 | 0.4686 |
60
+ | 0.399 | 6.63 | 965 | 0.4724 |
61
+ | 0.3875 | 7.63 | 1103 | 0.4756 |
62
+ | 0.3832 | 8.63 | 1240 | 0.4780 |
63
+ | 0.3682 | 9.63 | 1378 | 0.4814 |
64
+
65
+
66
+ ### Framework versions
67
+
68
+ - Transformers 4.35.0
69
+ - Pytorch 2.1.1+cu118
70
+ - Datasets 2.14.6
71
+ - Tokenizers 0.14.1
all_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.63,
3
+ "eval_loss": 0.4812127947807312,
4
+ "eval_runtime": 27.8382,
5
+ "eval_samples": 200,
6
+ "eval_samples_per_second": 7.184,
7
+ "eval_steps_per_second": 0.898,
8
+ "train_loss": 0.4462547982638737,
9
+ "train_runtime": 110414.5531,
10
+ "train_samples": 27870,
11
+ "train_samples_per_second": 2.524,
12
+ "train_steps_per_second": 0.02
13
+ }
eval_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.63,
3
+ "eval_loss": 0.4812127947807312,
4
+ "eval_runtime": 27.8382,
5
+ "eval_samples": 200,
6
+ "eval_samples_per_second": 7.184,
7
+ "eval_steps_per_second": 0.898
8
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 9.63,
3
+ "train_loss": 0.4462547982638737,
4
+ "train_runtime": 110414.5531,
5
+ "train_samples": 27870,
6
+ "train_samples_per_second": 2.524,
7
+ "train_steps_per_second": 0.02
8
+ }
trainer_state.json ADDED
The diff for this file is too large to render. See raw diff