Jon-Biz commited on
Commit
86d6cc0
1 Parent(s): fd06a4c

End of training

Browse files
Files changed (2) hide show
  1. README.md +159 -0
  2. adapter_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,159 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: axolotl-example-A
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
+ # Upload the final model to Huggingface
22
+ hub_model_id: Jon-Biz/axolotl-example-A
23
+
24
+ # Store the training logs in weights and biases
25
+ wandb_entity: jonbiz
26
+ wandb_project: tinyllama-1.1B_alpaca_2k_lora
27
+ base_model: TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
28
+ model_type: LlamaForCausalLM
29
+ tokenizer_type: LlamaTokenizer
30
+
31
+ load_in_8bit: true
32
+ load_in_4bit: false
33
+ strict: false
34
+
35
+ datasets:
36
+ - path: mhenrichsen/alpaca_2k_test
37
+ type: alpaca
38
+ dataset_prepared_path:
39
+ val_set_size: 0.05
40
+ output_dir: ./outputs/lora-out
41
+
42
+ sequence_len: 4096
43
+ sample_packing: true
44
+ eval_sample_packing: false
45
+ pad_to_sequence_len: true
46
+
47
+ adapter: lora
48
+ lora_model_dir:
49
+ lora_r: 32
50
+ lora_alpha: 16
51
+ lora_dropout: 0.05
52
+ lora_target_linear: true
53
+ lora_fan_in_fan_out:
54
+
55
+ wandb_project:
56
+ wandb_entity:
57
+ wandb_watch:
58
+ wandb_name:
59
+ wandb_log_model:
60
+
61
+ gradient_accumulation_steps: 4
62
+ micro_batch_size: 2
63
+ num_epochs: 4
64
+ optimizer: adamw_bnb_8bit
65
+ lr_scheduler: cosine
66
+ learning_rate: 0.0002
67
+
68
+ train_on_inputs: false
69
+ group_by_length: false
70
+ bf16: auto
71
+ fp16:
72
+ tf32: false
73
+
74
+ gradient_checkpointing: true
75
+ early_stopping_patience:
76
+ resume_from_checkpoint:
77
+ local_rank:
78
+ logging_steps: 1
79
+ xformers_attention:
80
+ flash_attention: true
81
+
82
+ warmup_steps: 10
83
+ evals_per_epoch: 4
84
+ saves_per_epoch: 1
85
+ debug:
86
+ deepspeed:
87
+ weight_decay: 0.0
88
+ fsdp:
89
+ fsdp_config:
90
+ special_tokens:
91
+
92
+ ```
93
+
94
+ </details><br>
95
+
96
+ # axolotl-example-A
97
+
98
+ 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.
99
+ It achieves the following results on the evaluation set:
100
+ - Loss: 1.2126
101
+
102
+ ## Model description
103
+
104
+ More information needed
105
+
106
+ ## Intended uses & limitations
107
+
108
+ More information needed
109
+
110
+ ## Training and evaluation data
111
+
112
+ More information needed
113
+
114
+ ## Training procedure
115
+
116
+ ### Training hyperparameters
117
+
118
+ The following hyperparameters were used during training:
119
+ - learning_rate: 0.0002
120
+ - train_batch_size: 2
121
+ - eval_batch_size: 2
122
+ - seed: 42
123
+ - gradient_accumulation_steps: 4
124
+ - total_train_batch_size: 8
125
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
126
+ - lr_scheduler_type: cosine
127
+ - lr_scheduler_warmup_steps: 10
128
+ - num_epochs: 4
129
+
130
+ ### Training results
131
+
132
+ | Training Loss | Epoch | Step | Validation Loss |
133
+ |:-------------:|:------:|:----:|:---------------:|
134
+ | 1.4615 | 0.08 | 1 | 1.4899 |
135
+ | 1.3853 | 0.24 | 3 | 1.4875 |
136
+ | 1.3666 | 0.48 | 6 | 1.4399 |
137
+ | 1.2683 | 0.72 | 9 | 1.3400 |
138
+ | 1.2268 | 0.96 | 12 | 1.2948 |
139
+ | 1.2512 | 1.16 | 15 | 1.2798 |
140
+ | 1.2258 | 1.4 | 18 | 1.2542 |
141
+ | 1.1351 | 1.6400 | 21 | 1.2364 |
142
+ | 1.2706 | 1.88 | 24 | 1.2267 |
143
+ | 1.1476 | 2.08 | 27 | 1.2245 |
144
+ | 1.1514 | 2.32 | 30 | 1.2243 |
145
+ | 1.1941 | 2.56 | 33 | 1.2191 |
146
+ | 1.1119 | 2.8 | 36 | 1.2150 |
147
+ | 1.1515 | 3.04 | 39 | 1.2142 |
148
+ | 1.1881 | 3.24 | 42 | 1.2114 |
149
+ | 1.1006 | 3.48 | 45 | 1.2152 |
150
+ | 1.1884 | 3.7200 | 48 | 1.2126 |
151
+
152
+
153
+ ### Framework versions
154
+
155
+ - PEFT 0.10.0
156
+ - Transformers 4.40.2
157
+ - Pytorch 2.1.2+cu118
158
+ - Datasets 2.19.1
159
+ - Tokenizers 0.19.1
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a0ceb30c3b0e9e853f011c46d426c454f2f232baa526a69af0da6e0a7eadcb6
3
+ size 101036698