vincentmin commited on
Commit
d5ec288
1 Parent(s): 5d0d1e5

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +65 -13
README.md CHANGED
@@ -1,20 +1,72 @@
1
  ---
2
- library_name: peft
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  ## Training procedure
5
 
 
6
 
7
- The following `bitsandbytes` quantization config was used during training:
8
- - load_in_8bit: False
9
- - load_in_4bit: True
10
- - llm_int8_threshold: 6.0
11
- - llm_int8_skip_modules: None
12
- - llm_int8_enable_fp32_cpu_offload: False
13
- - llm_int8_has_fp16_weight: False
14
- - bnb_4bit_quant_type: nf4
15
- - bnb_4bit_use_double_quant: False
16
- - bnb_4bit_compute_dtype: float16
17
- ### Framework versions
 
18
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
 
20
- - PEFT 0.5.0.dev0
 
 
 
 
1
  ---
2
+ base_model: meta-llama/Llama-2-13b-chat-hf
3
+ tags:
4
+ - generated_from_trainer
5
+ metrics:
6
+ - accuracy
7
+ model-index:
8
+ - name: llama-2-13b-reward-oasst1
9
+ results: []
10
  ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # llama-2-13b-reward-oasst1
16
+
17
+ This model is a fine-tuned version of [meta-llama/Llama-2-13b-chat-hf](https://huggingface.co/meta-llama/Llama-2-13b-chat-hf) on the None dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 0.4810
20
+ - Accuracy: 0.7869
21
+
22
+ ## Model description
23
+
24
+ More information needed
25
+
26
+ ## Intended uses & limitations
27
+
28
+ More information needed
29
+
30
+ ## Training and evaluation data
31
+
32
+ More information needed
33
+
34
  ## Training procedure
35
 
36
+ ### Training hyperparameters
37
 
38
+ The following hyperparameters were used during training:
39
+ - learning_rate: 2e-05
40
+ - train_batch_size: 1
41
+ - eval_batch_size: 1
42
+ - seed: 42
43
+ - gradient_accumulation_steps: 4
44
+ - total_train_batch_size: 4
45
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
46
+ - lr_scheduler_type: linear
47
+ - num_epochs: 1
48
+
49
+ ### Training results
50
 
51
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
+ | 0.5602 | 0.08 | 250 | 0.5436 | 0.7388 |
54
+ | 0.6166 | 0.17 | 500 | 0.5340 | 0.7468 |
55
+ | 0.6545 | 0.25 | 750 | 0.4899 | 0.7644 |
56
+ | 0.5635 | 0.33 | 1000 | 0.4877 | 0.7532 |
57
+ | 0.5933 | 0.42 | 1250 | 0.4930 | 0.7660 |
58
+ | 0.5758 | 0.5 | 1500 | 0.4851 | 0.7740 |
59
+ | 0.5212 | 0.58 | 1750 | 0.5021 | 0.7788 |
60
+ | 0.5251 | 0.67 | 2000 | 0.4893 | 0.7804 |
61
+ | 0.5145 | 0.75 | 2250 | 0.4924 | 0.7853 |
62
+ | 0.5085 | 0.83 | 2500 | 0.4934 | 0.7853 |
63
+ | 0.617 | 0.92 | 2750 | 0.4803 | 0.7821 |
64
+ | 0.5525 | 1.0 | 3000 | 0.4810 | 0.7869 |
65
+
66
+
67
+ ### Framework versions
68
 
69
+ - Transformers 4.32.0.dev0
70
+ - Pytorch 2.0.1+cu118
71
+ - Datasets 2.14.0
72
+ - Tokenizers 0.13.3