Kareem Amr commited on
Commit
2d73bb2
·
verified ·
1 Parent(s): 90f5dfa

End of training

Browse files
Files changed (2) hide show
  1. README.md +151 -0
  2. adapter_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,151 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: peft
4
+ tags:
5
+ - axolotl
6
+ - generated_from_trainer
7
+ base_model: TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
8
+ model-index:
9
+ - name: tinyllama-1.1B_dolly-3k_lora
10
+ results: []
11
+ ---
12
+
13
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
14
+ should probably proofread and complete it, then remove this comment. -->
15
+
16
+ [<img src="https://raw.githubusercontent.com/OpenAccess-AI-Collective/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/OpenAccess-AI-Collective/axolotl)
17
+ <details><summary>See axolotl config</summary>
18
+
19
+ axolotl version: `0.4.0`
20
+ ```yaml
21
+ base_model: TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
22
+ model_type: LlamaForCausalLM
23
+ tokenizer_type: LlamaTokenizer
24
+
25
+ load_in_8bit: true
26
+ load_in_4bit: false
27
+ strict: false
28
+
29
+ datasets:
30
+ - path: kareemamrr/databricks-dolly-3k
31
+ type: alpaca
32
+ dataset_prepared_path:
33
+ val_set_size: 0.05
34
+ output_dir: ./outputs/lora-out
35
+
36
+ sequence_len: 4096
37
+ sample_packing: true
38
+ eval_sample_packing: false
39
+ pad_to_sequence_len: true
40
+
41
+ adapter: lora
42
+ lora_model_dir:
43
+ lora_r: 32
44
+ lora_alpha: 16
45
+ lora_dropout: 0.05
46
+ lora_target_linear: true
47
+ lora_fan_in_fan_out:
48
+
49
+ hub_model_id: kareemamrr/tinyllama-1.1B_dolly-3k_lora
50
+
51
+ wandb_project: tinyllama-1.1B_dolly-3k
52
+ wandb_entity: kamr54
53
+ wandb_name: lora
54
+
55
+ gradient_accumulation_steps: 4
56
+ micro_batch_size: 2
57
+ num_epochs: 4
58
+ optimizer: adamw_bnb_8bit
59
+ lr_scheduler: cosine
60
+ learning_rate: 0.0002
61
+
62
+ train_on_inputs: false
63
+ group_by_length: false
64
+ bf16: auto
65
+ fp16:
66
+ tf32: false
67
+
68
+ gradient_checkpointing: true
69
+ early_stopping_patience:
70
+ resume_from_checkpoint:
71
+ local_rank:
72
+ logging_steps: 1
73
+ xformers_attention:
74
+ flash_attention: true
75
+
76
+ warmup_steps: 10
77
+ evals_per_epoch: 4
78
+ saves_per_epoch: 1
79
+ debug:
80
+ deepspeed:
81
+ weight_decay: 0.0
82
+ fsdp:
83
+ fsdp_config:
84
+ special_tokens:
85
+
86
+ ```
87
+
88
+ </details><br>
89
+
90
+ # tinyllama-1.1B_dolly-3k_lora
91
+
92
+ This model is a fine-tuned version of [TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T](https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T) on the None dataset.
93
+ It achieves the following results on the evaluation set:
94
+ - Loss: 1.7510
95
+
96
+ ## Model description
97
+
98
+ More information needed
99
+
100
+ ## Intended uses & limitations
101
+
102
+ More information needed
103
+
104
+ ## Training and evaluation data
105
+
106
+ More information needed
107
+
108
+ ## Training procedure
109
+
110
+ ### Training hyperparameters
111
+
112
+ The following hyperparameters were used during training:
113
+ - learning_rate: 0.0002
114
+ - train_batch_size: 2
115
+ - eval_batch_size: 2
116
+ - seed: 42
117
+ - gradient_accumulation_steps: 4
118
+ - total_train_batch_size: 8
119
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
120
+ - lr_scheduler_type: cosine
121
+ - lr_scheduler_warmup_steps: 10
122
+ - num_epochs: 4
123
+
124
+ ### Training results
125
+
126
+ | Training Loss | Epoch | Step | Validation Loss |
127
+ |:-------------:|:------:|:----:|:---------------:|
128
+ | 1.9375 | 0.0476 | 1 | 2.1187 |
129
+ | 1.9297 | 0.2857 | 6 | 2.0186 |
130
+ | 1.8257 | 0.5714 | 12 | 1.8111 |
131
+ | 1.7285 | 0.8571 | 18 | 1.7862 |
132
+ | 1.752 | 1.1071 | 24 | 1.7848 |
133
+ | 1.8686 | 1.3929 | 30 | 1.7808 |
134
+ | 1.6451 | 1.6786 | 36 | 1.7594 |
135
+ | 1.7861 | 1.9643 | 42 | 1.7582 |
136
+ | 1.6606 | 2.2024 | 48 | 1.7536 |
137
+ | 1.6168 | 2.4881 | 54 | 1.7543 |
138
+ | 1.7087 | 2.7738 | 60 | 1.7527 |
139
+ | 1.7276 | 3.0119 | 66 | 1.7525 |
140
+ | 1.8117 | 3.2976 | 72 | 1.7487 |
141
+ | 1.6897 | 3.5833 | 78 | 1.7502 |
142
+ | 1.7861 | 3.8690 | 84 | 1.7510 |
143
+
144
+
145
+ ### Framework versions
146
+
147
+ - PEFT 0.10.0
148
+ - Transformers 4.40.2
149
+ - Pytorch 2.1.2+cu118
150
+ - Datasets 2.19.1
151
+ - Tokenizers 0.19.1
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ec61afd5eb06084d7db7db405cc986fb6411060589b2aa8148635e761e78268
3
+ size 101036698