sert121 commited on
Commit
458f27a
1 Parent(s): a53b99b

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +60 -14
README.md CHANGED
@@ -1,21 +1,67 @@
1
  ---
2
- license: mit
3
- library_name: peft
 
 
 
4
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  ## Training procedure
6
 
 
7
 
8
- The following `bitsandbytes` quantization config was used during training:
9
- - load_in_8bit: False
10
- - load_in_4bit: True
11
- - llm_int8_threshold: 6.0
12
- - llm_int8_skip_modules: None
13
- - llm_int8_enable_fp32_cpu_offload: False
14
- - llm_int8_has_fp16_weight: False
15
- - bnb_4bit_quant_type: nf4
16
- - bnb_4bit_use_double_quant: False
17
- - bnb_4bit_compute_dtype: float16
18
- ### Framework versions
 
 
19
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
 
21
- - PEFT 0.4.0.dev0
 
 
 
 
1
  ---
2
+ tags:
3
+ - generated_from_trainer
4
+ model-index:
5
+ - name: falcon_spider_v1
6
+ results: []
7
  ---
8
+
9
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
10
+ should probably proofread and complete it, then remove this comment. -->
11
+
12
+ # falcon_spider_v1
13
+
14
+ This model is a fine-tuned version of [ybelkada/falcon-7b-sharded-bf16](https://huggingface.co/ybelkada/falcon-7b-sharded-bf16) on an unknown dataset.
15
+ It achieves the following results on the evaluation set:
16
+ - Loss: 0.5862
17
+
18
+ ## Model description
19
+
20
+ More information needed
21
+
22
+ ## Intended uses & limitations
23
+
24
+ More information needed
25
+
26
+ ## Training and evaluation data
27
+
28
+ More information needed
29
+
30
  ## Training procedure
31
 
32
+ ### Training hyperparameters
33
 
34
+ The following hyperparameters were used during training:
35
+ - learning_rate: 0.0002
36
+ - train_batch_size: 4
37
+ - eval_batch_size: 8
38
+ - seed: 42
39
+ - gradient_accumulation_steps: 4
40
+ - total_train_batch_size: 16
41
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
+ - lr_scheduler_type: constant
43
+ - lr_scheduler_warmup_ratio: 0.03
44
+ - training_steps: 100
45
+
46
+ ### Training results
47
 
48
+ | Training Loss | Epoch | Step | Validation Loss |
49
+ |:-------------:|:-----:|:----:|:---------------:|
50
+ | 1.1754 | 0.03 | 10 | 0.6316 |
51
+ | 0.5011 | 0.05 | 20 | 0.4848 |
52
+ | 0.4355 | 0.08 | 30 | 0.4586 |
53
+ | 0.3921 | 0.1 | 40 | 0.4555 |
54
+ | 0.3327 | 0.13 | 50 | 0.4674 |
55
+ | 0.2662 | 0.15 | 60 | 0.4507 |
56
+ | 0.2258 | 0.18 | 70 | 0.4813 |
57
+ | 0.1569 | 0.2 | 80 | 0.5407 |
58
+ | 0.1313 | 0.23 | 90 | 0.5924 |
59
+ | 0.1215 | 0.25 | 100 | 0.5862 |
60
+
61
+
62
+ ### Framework versions
63
 
64
+ - Transformers 4.30.2
65
+ - Pytorch 2.0.1+cu117
66
+ - Datasets 2.13.1
67
+ - Tokenizers 0.13.3