nttx commited on
Commit
501a7b7
·
verified ·
1 Parent(s): 2b0ec09

End of training

Browse files
Files changed (2) hide show
  1. README.md +181 -0
  2. adapter_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,181 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ base_model: TitanML/tiny-mixtral
4
+ tags:
5
+ - axolotl
6
+ - generated_from_trainer
7
+ model-index:
8
+ - name: b279808d-b2c3-4b2c-a69a-9d224adb42fe
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ [<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
16
+ <details><summary>See axolotl config</summary>
17
+
18
+ axolotl version: `0.4.1`
19
+ ```yaml
20
+ adapter: lora
21
+ base_model: TitanML/tiny-mixtral
22
+ bf16: true
23
+ chat_template: llama3
24
+ dataset_prepared_path: null
25
+ datasets:
26
+ - data_files:
27
+ - 1626557c75bc8215_train_data.json
28
+ ds_type: json
29
+ format: custom
30
+ path: /workspace/input_data/1626557c75bc8215_train_data.json
31
+ type:
32
+ field_instruction: instruction
33
+ field_output: output
34
+ format: '{instruction}'
35
+ no_input_format: '{instruction}'
36
+ system_format: '{system}'
37
+ system_prompt: ''
38
+ debug: null
39
+ deepspeed: null
40
+ device_map: auto
41
+ do_eval: true
42
+ early_stopping_patience: 2
43
+ eval_batch_size: 4
44
+ eval_max_new_tokens: 128
45
+ eval_steps: 150
46
+ eval_table_size: null
47
+ evals_per_epoch: null
48
+ flash_attention: true
49
+ fp16: false
50
+ fsdp: null
51
+ fsdp_config: null
52
+ gradient_accumulation_steps: 4
53
+ gradient_checkpointing: true
54
+ group_by_length: true
55
+ hub_model_id: nttx/b279808d-b2c3-4b2c-a69a-9d224adb42fe
56
+ hub_repo: null
57
+ hub_strategy: checkpoint
58
+ hub_token: null
59
+ learning_rate: 0.0002
60
+ load_in_4bit: false
61
+ load_in_8bit: false
62
+ local_rank: null
63
+ logging_steps: 50
64
+ lora_alpha: 128
65
+ lora_dropout: 0.3
66
+ lora_fan_in_fan_out: null
67
+ lora_model_dir: null
68
+ lora_r: 64
69
+ lora_target_linear: true
70
+ lr_scheduler: cosine
71
+ max_grad_norm: 1.0
72
+ max_memory:
73
+ 0: 75GB
74
+ max_steps: 3000
75
+ micro_batch_size: 4
76
+ mlflow_experiment_name: /tmp/1626557c75bc8215_train_data.json
77
+ model_type: AutoModelForCausalLM
78
+ num_epochs: 10
79
+ optim_args:
80
+ adam_beta1: 0.9
81
+ adam_beta2: 0.95
82
+ adam_epsilon: 1e-5
83
+ optimizer: adamw_bnb_8bit
84
+ output_dir: miner_id_24
85
+ pad_to_sequence_len: true
86
+ resume_from_checkpoint: null
87
+ s2_attention: null
88
+ sample_packing: false
89
+ save_steps: 150
90
+ saves_per_epoch: null
91
+ sequence_len: 512
92
+ special_tokens:
93
+ pad_token: </s>
94
+ strict: false
95
+ tf32: true
96
+ tokenizer_type: AutoTokenizer
97
+ train_on_inputs: false
98
+ trust_remote_code: true
99
+ val_set_size: 0.05
100
+ wandb_entity: null
101
+ wandb_mode: online
102
+ wandb_name: 7950e38d-4ae2-4e1f-a6a0-0107e2b31c7d
103
+ wandb_project: Gradients-On-Demand
104
+ wandb_run: your_name
105
+ wandb_runid: 7950e38d-4ae2-4e1f-a6a0-0107e2b31c7d
106
+ warmup_steps: 50
107
+ weight_decay: 0.0
108
+ xformers_attention: null
109
+
110
+ ```
111
+
112
+ </details><br>
113
+
114
+ # b279808d-b2c3-4b2c-a69a-9d224adb42fe
115
+
116
+ This model is a fine-tuned version of [TitanML/tiny-mixtral](https://huggingface.co/TitanML/tiny-mixtral) on the None dataset.
117
+ It achieves the following results on the evaluation set:
118
+ - Loss: 4.8374
119
+
120
+ ## Model description
121
+
122
+ More information needed
123
+
124
+ ## Intended uses & limitations
125
+
126
+ More information needed
127
+
128
+ ## Training and evaluation data
129
+
130
+ More information needed
131
+
132
+ ## Training procedure
133
+
134
+ ### Training hyperparameters
135
+
136
+ The following hyperparameters were used during training:
137
+ - learning_rate: 0.0002
138
+ - train_batch_size: 4
139
+ - eval_batch_size: 4
140
+ - seed: 42
141
+ - gradient_accumulation_steps: 4
142
+ - total_train_batch_size: 16
143
+ - optimizer: Use OptimizerNames.ADAMW_BNB with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=adam_beta1=0.9,adam_beta2=0.95,adam_epsilon=1e-5
144
+ - lr_scheduler_type: cosine
145
+ - lr_scheduler_warmup_steps: 50
146
+ - training_steps: 3000
147
+
148
+ ### Training results
149
+
150
+ | Training Loss | Epoch | Step | Validation Loss |
151
+ |:-------------:|:------:|:----:|:---------------:|
152
+ | No log | 0.0003 | 1 | 10.5753 |
153
+ | 8.3438 | 0.0492 | 150 | 8.0755 |
154
+ | 6.9827 | 0.0983 | 300 | 6.9109 |
155
+ | 6.5081 | 0.1475 | 450 | 6.4205 |
156
+ | 6.207 | 0.1966 | 600 | 6.1038 |
157
+ | 5.9758 | 0.2458 | 750 | 5.9017 |
158
+ | 5.7978 | 0.2950 | 900 | 5.7243 |
159
+ | 5.5757 | 0.3441 | 1050 | 5.5193 |
160
+ | 5.5036 | 0.3933 | 1200 | 5.3982 |
161
+ | 5.3336 | 0.4424 | 1350 | 5.2550 |
162
+ | 5.2624 | 0.4916 | 1500 | 5.1578 |
163
+ | 5.1754 | 0.5408 | 1650 | 5.0719 |
164
+ | 5.1936 | 0.5899 | 1800 | 5.0090 |
165
+ | 5.0071 | 0.6391 | 1950 | 4.9581 |
166
+ | 5.0504 | 0.6882 | 2100 | 4.9146 |
167
+ | 4.9615 | 0.7374 | 2250 | 4.8807 |
168
+ | 4.9688 | 0.7866 | 2400 | 4.8611 |
169
+ | 4.9255 | 0.8357 | 2550 | 4.8456 |
170
+ | 5.0045 | 0.8849 | 2700 | 4.8406 |
171
+ | 4.9882 | 0.9340 | 2850 | 4.8395 |
172
+ | 4.9035 | 0.9832 | 3000 | 4.8374 |
173
+
174
+
175
+ ### Framework versions
176
+
177
+ - PEFT 0.13.2
178
+ - Transformers 4.46.0
179
+ - Pytorch 2.5.0+cu124
180
+ - Datasets 3.0.1
181
+ - Tokenizers 0.20.1
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5388c4964bf191dc087bb7888ea8438b59d4682f620c0b011331e19b6c368281
3
+ size 60602650