dabrown commited on
Commit
4c44d0a
·
verified ·
1 Parent(s): b4956ea

End of training

Browse files
Files changed (2) hide show
  1. README.md +172 -0
  2. adapter_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,172 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ license: bigcode-openrail-m
4
+ base_model: bigcode/starcoder2-3b
5
+ tags:
6
+ - axolotl
7
+ - generated_from_trainer
8
+ model-index:
9
+ - name: 59f7578c-7311-4882-b526-b801cc6a269b
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/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)
17
+ <details><summary>See axolotl config</summary>
18
+
19
+ axolotl version: `0.5.2`
20
+ ```yaml
21
+ adapter: lora
22
+ base_model: bigcode/starcoder2-3b
23
+ bf16: true
24
+ chat_template: llama3
25
+ dataset_prepared_path: null
26
+ datasets:
27
+ - data_files:
28
+ - 012ab4813cc99fb8_train_data.json
29
+ ds_type: json
30
+ format: custom
31
+ path: /workspace/input_data/012ab4813cc99fb8_train_data.json
32
+ type:
33
+ field_input: evidence
34
+ field_instruction: question
35
+ field_output: SQL
36
+ format: '{instruction} {input}'
37
+ no_input_format: '{instruction}'
38
+ system_format: '{system}'
39
+ system_prompt: ''
40
+ debug: null
41
+ deepspeed: null
42
+ device_map: auto
43
+ do_eval: true
44
+ early_stopping_patience: 5
45
+ eval_batch_size: 4
46
+ eval_max_new_tokens: 128
47
+ eval_steps: 50
48
+ eval_table_size: null
49
+ evals_per_epoch: null
50
+ flash_attention: false
51
+ fp16: false
52
+ fsdp: null
53
+ fsdp_config: null
54
+ gradient_accumulation_steps: 2
55
+ gradient_checkpointing: true
56
+ group_by_length: true
57
+ hub_model_id: dabrown/59f7578c-7311-4882-b526-b801cc6a269b
58
+ hub_repo: null
59
+ hub_strategy: checkpoint
60
+ hub_token: null
61
+ learning_rate: 0.0001
62
+ load_in_4bit: false
63
+ load_in_8bit: false
64
+ local_rank: null
65
+ logging_steps: 1
66
+ lora_alpha: 128
67
+ lora_dropout: 0.2
68
+ lora_fan_in_fan_out: false
69
+ lora_inference_mode: true
70
+ lora_model_dir: null
71
+ lora_modules_to_save:
72
+ - lm_head
73
+ lora_r: 64
74
+ lora_target_linear: true
75
+ lr_scheduler: cosine
76
+ max_grad_norm: 1.0
77
+ max_memory:
78
+ 0: 75GB
79
+ max_steps: 200
80
+ micro_batch_size: 4
81
+ mlflow_experiment_name: /tmp/012ab4813cc99fb8_train_data.json
82
+ model_type: AutoModelForCausalLM
83
+ num_epochs: 10
84
+ optim_args:
85
+ adam_beta1: 0.9
86
+ adam_beta2: 0.95
87
+ adam_epsilon: 1e-5
88
+ optimizer: adamw_bnb_8bit
89
+ output_dir: miner_id_24
90
+ pad_to_sequence_len: true
91
+ peft_use_rslora: true
92
+ resume_from_checkpoint: null
93
+ s2_attention: null
94
+ sample_packing: false
95
+ save_steps: 50
96
+ saves_per_epoch: null
97
+ sequence_len: 1024
98
+ shuffle_merged_datasets: true
99
+ special_tokens:
100
+ pad_token: <|endoftext|>
101
+ strict: false
102
+ tf32: true
103
+ tokenizer_type: AutoTokenizer
104
+ train_on_inputs: false
105
+ trust_remote_code: true
106
+ val_set_size: 0.05
107
+ wandb_entity: null
108
+ wandb_mode: online
109
+ wandb_name: b1e23278-252e-44d7-9491-1b28d344421c
110
+ wandb_project: Gradients-On-Demand
111
+ wandb_run: your_name
112
+ wandb_runid: b1e23278-252e-44d7-9491-1b28d344421c
113
+ warmup_steps: 10
114
+ weight_decay: 0.0
115
+ xformers_attention: null
116
+
117
+ ```
118
+
119
+ </details><br>
120
+
121
+ # 59f7578c-7311-4882-b526-b801cc6a269b
122
+
123
+ This model is a fine-tuned version of [bigcode/starcoder2-3b](https://huggingface.co/bigcode/starcoder2-3b) on the None dataset.
124
+ It achieves the following results on the evaluation set:
125
+ - Loss: 0.2604
126
+
127
+ ## Model description
128
+
129
+ More information needed
130
+
131
+ ## Intended uses & limitations
132
+
133
+ More information needed
134
+
135
+ ## Training and evaluation data
136
+
137
+ More information needed
138
+
139
+ ## Training procedure
140
+
141
+ ### Training hyperparameters
142
+
143
+ The following hyperparameters were used during training:
144
+ - learning_rate: 0.0001
145
+ - train_batch_size: 4
146
+ - eval_batch_size: 4
147
+ - seed: 42
148
+ - gradient_accumulation_steps: 2
149
+ - total_train_batch_size: 8
150
+ - optimizer: Use adamw_bnb_8bit 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
151
+ - lr_scheduler_type: cosine
152
+ - lr_scheduler_warmup_steps: 10
153
+ - training_steps: 200
154
+
155
+ ### Training results
156
+
157
+ | Training Loss | Epoch | Step | Validation Loss |
158
+ |:-------------:|:------:|:----:|:---------------:|
159
+ | 1.6102 | 0.0051 | 1 | 0.9090 |
160
+ | 0.6222 | 0.2525 | 50 | 0.4225 |
161
+ | 0.5434 | 0.5051 | 100 | 0.3194 |
162
+ | 0.3708 | 0.7576 | 150 | 0.2683 |
163
+ | 0.2308 | 1.0101 | 200 | 0.2604 |
164
+
165
+
166
+ ### Framework versions
167
+
168
+ - PEFT 0.13.2
169
+ - Transformers 4.46.3
170
+ - Pytorch 2.5.1+cu124
171
+ - Datasets 3.1.0
172
+ - Tokenizers 0.20.3
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78a25cddf6adf28b3a4abe1bd6be19f1136e33ba027e73434e3c8677009cfc3b
3
+ size 683539566