bastienp commited on
Commit
394dfc8
1 Parent(s): e82d058

End of training

Browse files
Files changed (2) hide show
  1. README.md +146 -0
  2. adapter_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,146 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: pytiny-hf
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.3.0`
20
+ ```yaml
21
+ base_model: TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
22
+ model_type: LlamaForCausalLM
23
+ tokenizer_type: LlamaTokenizer
24
+ is_llama_derived_model: true
25
+ hub_model_id: pytiny-hf
26
+
27
+ load_in_8bit: false
28
+ load_in_4bit: true
29
+ strict: false
30
+
31
+ datasets:
32
+ - path: mhenrichsen/alpaca_2k_test
33
+ type: alpaca
34
+ dataset_prepared_path:
35
+ val_set_size: 0.05
36
+ output_dir: ./qlora-out
37
+
38
+ adapter: qlora
39
+ lora_model_dir:
40
+
41
+ sequence_len: 2048
42
+ sample_packing: true
43
+ pad_to_sequence_len: true
44
+
45
+ lora_r: 32
46
+ lora_alpha: 16
47
+ lora_dropout: 0.05
48
+ lora_target_modules:
49
+ lora_target_linear: true
50
+ lora_fan_in_fan_out:
51
+
52
+ wandb_project: axolotl-pytiny-hf
53
+ wandb_entity:
54
+ wandb_watch:
55
+ wandb_name:
56
+ wandb_log_model:
57
+
58
+ gradient_accumulation_steps: 4
59
+ micro_batch_size: 2
60
+ num_epochs: 3
61
+ optimizer: paged_adamw_32bit
62
+ lr_scheduler: cosine
63
+ learning_rate: 0.0002
64
+
65
+ train_on_inputs: false
66
+ group_by_length: false
67
+ bf16: false
68
+ fp16: true
69
+ tf32: false
70
+
71
+ gradient_checkpointing: true
72
+ early_stopping_patience:
73
+ resume_from_checkpoint:
74
+ local_rank:
75
+ logging_steps: 1
76
+ xformers_attention:
77
+ flash_attention: false
78
+
79
+ warmup_steps: 10
80
+ evals_per_epoch: 2
81
+ saves_per_epoch: 1
82
+ debug:
83
+ deepspeed:
84
+ weight_decay: 0.0
85
+ fsdp:
86
+ fsdp_config:
87
+ special_tokens:
88
+ ```
89
+
90
+ </details><br>
91
+
92
+ # pytiny-hf
93
+
94
+ 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.
95
+ It achieves the following results on the evaluation set:
96
+ - Loss: 3.7089
97
+
98
+ ## Model description
99
+
100
+ More information needed
101
+
102
+ ## Intended uses & limitations
103
+
104
+ More information needed
105
+
106
+ ## Training and evaluation data
107
+
108
+ More information needed
109
+
110
+ ## Training procedure
111
+
112
+ ### Training hyperparameters
113
+
114
+ The following hyperparameters were used during training:
115
+ - learning_rate: 0.0002
116
+ - train_batch_size: 2
117
+ - eval_batch_size: 2
118
+ - seed: 42
119
+ - gradient_accumulation_steps: 4
120
+ - total_train_batch_size: 8
121
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
122
+ - lr_scheduler_type: cosine
123
+ - lr_scheduler_warmup_steps: 10
124
+ - num_epochs: 3
125
+ - mixed_precision_training: Native AMP
126
+
127
+ ### Training results
128
+
129
+ | Training Loss | Epoch | Step | Validation Loss |
130
+ |:-------------:|:-----:|:----:|:---------------:|
131
+ | 6.8735 | 0.04 | 1 | 6.7693 |
132
+ | 5.594 | 0.5 | 13 | 5.4061 |
133
+ | 4.7525 | 1.0 | 26 | 4.6806 |
134
+ | 4.404 | 1.46 | 39 | 4.3381 |
135
+ | 4.0121 | 1.96 | 52 | 3.9443 |
136
+ | 3.7604 | 2.42 | 65 | 3.7493 |
137
+ | 3.6504 | 2.92 | 78 | 3.7089 |
138
+
139
+
140
+ ### Framework versions
141
+
142
+ - PEFT 0.7.2.dev0
143
+ - Transformers 4.37.0
144
+ - Pytorch 2.0.1+cu117
145
+ - Datasets 2.16.1
146
+ - Tokenizers 0.15.0
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e452479777f63e3d57c70cefeb75d74c4be806a0f9bd900d01f144eda33c9a3
3
+ size 101036253