VitaliiVrublevskyi commited on
Commit
81de10c
1 Parent(s): e41147f

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -26
README.md CHANGED
@@ -10,7 +10,6 @@ metrics:
10
  model-index:
11
  - name: Llama-2-7b-hf-finetuned-mrpc
12
  results: []
13
- library_name: peft
14
  ---
15
 
16
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -20,9 +19,9 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  This model is a fine-tuned version of [meta-llama/Llama-2-7b-hf](https://huggingface.co/meta-llama/Llama-2-7b-hf) on the glue dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.4479
24
  - Accuracy: 0.7941
25
  - F1: 0.8571
 
26
 
27
  ## Model description
28
 
@@ -38,27 +37,16 @@ More information needed
38
 
39
  ## Training procedure
40
 
41
-
42
- The following `bitsandbytes` quantization config was used during training:
43
- - load_in_8bit: True
44
- - load_in_4bit: False
45
- - llm_int8_threshold: 6.0
46
- - llm_int8_skip_modules: None
47
- - llm_int8_enable_fp32_cpu_offload: False
48
- - llm_int8_has_fp16_weight: False
49
- - bnb_4bit_quant_type: fp4
50
- - bnb_4bit_use_double_quant: False
51
- - bnb_4bit_compute_dtype: float32
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
- - learning_rate: 0.0001
56
  - train_batch_size: 16
57
  - eval_batch_size: 16
58
  - seed: 42
59
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
60
  - lr_scheduler_type: linear
61
- - num_epochs: 20
62
 
63
  ### Training results
64
 
@@ -74,21 +62,20 @@ The following hyperparameters were used during training:
74
  | 0.5139 | 8.0 | 1840 | 0.7402 | 0.8307 | 0.4989 |
75
  | 0.4754 | 9.0 | 2070 | 0.7745 | 0.8435 | 0.4732 |
76
  | 0.4754 | 10.0 | 2300 | 0.7672 | 0.8403 | 0.4716 |
77
- | 0.5407 | 11.0 | 2530 | 0.4823 | 0.7598 | 0.8393 |
78
- | 0.5407 | 12.0 | 2760 | 0.4782 | 0.7451 | 0.8333 |
79
- | 0.5407 | 13.0 | 2990 | 0.4713 | 0.7451 | 0.8333 |
80
- | 0.4951 | 14.0 | 3220 | 0.4553 | 0.7819 | 0.8489 |
81
- | 0.4951 | 15.0 | 3450 | 0.4591 | 0.7745 | 0.8506 |
82
- | 0.4724 | 16.0 | 3680 | 0.4631 | 0.7770 | 0.8423 |
83
- | 0.4724 | 17.0 | 3910 | 0.4581 | 0.8015 | 0.8576 |
84
- | 0.4455 | 18.0 | 4140 | 0.4548 | 0.7819 | 0.8468 |
85
- | 0.4455 | 19.0 | 4370 | 0.4511 | 0.7819 | 0.8484 |
86
- | 0.4354 | 20.0 | 4600 | 0.4479 | 0.7941 | 0.8571 |
87
 
88
 
89
  ### Framework versions
90
 
91
- - PEFT 0.4.0
92
  - Transformers 4.31.0
93
  - Pytorch 2.0.1+cu118
94
  - Datasets 2.14.5
 
10
  model-index:
11
  - name: Llama-2-7b-hf-finetuned-mrpc
12
  results: []
 
13
  ---
14
 
15
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
19
 
20
  This model is a fine-tuned version of [meta-llama/Llama-2-7b-hf](https://huggingface.co/meta-llama/Llama-2-7b-hf) on the glue dataset.
21
  It achieves the following results on the evaluation set:
 
22
  - Accuracy: 0.7941
23
  - F1: 0.8571
24
+ - Loss: 0.4479
25
 
26
  ## Model description
27
 
 
37
 
38
  ## Training procedure
39
 
 
 
 
 
 
 
 
 
 
 
 
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
+ - learning_rate: 5e-05
44
  - train_batch_size: 16
45
  - eval_batch_size: 16
46
  - seed: 42
47
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
48
  - lr_scheduler_type: linear
49
+ - num_epochs: 10
50
 
51
  ### Training results
52
 
 
62
  | 0.5139 | 8.0 | 1840 | 0.7402 | 0.8307 | 0.4989 |
63
  | 0.4754 | 9.0 | 2070 | 0.7745 | 0.8435 | 0.4732 |
64
  | 0.4754 | 10.0 | 2300 | 0.7672 | 0.8403 | 0.4716 |
65
+ | 0.5407 | 11.0 | 2530 | 0.7598 | 0.8393 | 0.4823 |
66
+ | 0.5407 | 12.0 | 2760 | 0.7451 | 0.8333 | 0.4782 |
67
+ | 0.5407 | 13.0 | 2990 | 0.7451 | 0.8333 | 0.4713 |
68
+ | 0.4951 | 14.0 | 3220 | 0.7819 | 0.8489 | 0.4553 |
69
+ | 0.4951 | 15.0 | 3450 | 0.7745 | 0.8506 | 0.4591 |
70
+ | 0.4724 | 16.0 | 3680 | 0.7770 | 0.8423 | 0.4631 |
71
+ | 0.4724 | 17.0 | 3910 | 0.8015 | 0.8576 | 0.4581 |
72
+ | 0.4455 | 18.0 | 4140 | 0.7819 | 0.8468 | 0.4548 |
73
+ | 0.4455 | 19.0 | 4370 | 0.7819 | 0.8484 | 0.4511 |
74
+ | 0.4354 | 20.0 | 4600 | 0.7941 | 0.8571 | 0.4479 |
75
 
76
 
77
  ### Framework versions
78
 
 
79
  - Transformers 4.31.0
80
  - Pytorch 2.0.1+cu118
81
  - Datasets 2.14.5