dabrown commited on
Commit
9fa6e0f
·
verified ·
1 Parent(s): 82cb8ae

End of training

Browse files
Files changed (3) hide show
  1. README.md +175 -0
  2. adapter_model.bin +3 -0
  3. adapter_model.safetensors +1 -1
README.md ADDED
@@ -0,0 +1,175 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: ac205c4f-ba6f-4fb4-81fb-dbb3561fd3af
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.6.0`
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: 4
55
+ gradient_checkpointing: true
56
+ group_by_length: true
57
+ hub_model_id: dabrown/ac205c4f-ba6f-4fb4-81fb-dbb3561fd3af
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: null
69
+ lora_model_dir: null
70
+ lora_r: 64
71
+ lora_target_modules:
72
+ - q_proj 0.2
73
+ - k_proj 0.2
74
+ - v_proj 0.2
75
+ - o_proj 0.1
76
+ - gate_proj 0.1
77
+ - down_proj 0.1
78
+ - up_proj 0.1
79
+ - lm_head
80
+ - embed_tokens
81
+ lr_scheduler: cosine
82
+ max_grad_norm: 1.0
83
+ max_memory:
84
+ 0: 75GB
85
+ max_steps: 200
86
+ micro_batch_size: 8
87
+ mlflow_experiment_name: /tmp/012ab4813cc99fb8_train_data.json
88
+ model_type: AutoModelForCausalLM
89
+ modules_to_save: lm_head
90
+ num_epochs: 3
91
+ optim_args:
92
+ adam_beta1: 0.9
93
+ adam_beta2: 0.95
94
+ adam_epsilon: 1e-5
95
+ optimizer: adamw_bnb_8bit
96
+ output_dir: miner_id_24
97
+ pad_to_sequence_len: true
98
+ resume_from_checkpoint: null
99
+ s2_attention: null
100
+ sample_packing: false
101
+ save_steps: 50
102
+ saves_per_epoch: null
103
+ sequence_len: 1024
104
+ shuffle_merged_datasets: true
105
+ special_tokens:
106
+ pad_token: <|endoftext|>
107
+ strict: false
108
+ tf32: true
109
+ tokenizer_type: AutoTokenizer
110
+ train_on_inputs: false
111
+ trust_remote_code: true
112
+ val_set_size: 0.05
113
+ wandb_entity: null
114
+ wandb_mode: online
115
+ wandb_name: b1e23278-252e-44d7-9491-1b28d344421c
116
+ wandb_project: Gradients-On-Demand
117
+ wandb_run: your_name
118
+ wandb_runid: b1e23278-252e-44d7-9491-1b28d344421c
119
+ warmup_steps: 10
120
+ weight_decay: 0.0
121
+ xformers_attention: null
122
+
123
+ ```
124
+
125
+ </details><br>
126
+
127
+ # ac205c4f-ba6f-4fb4-81fb-dbb3561fd3af
128
+
129
+ This model is a fine-tuned version of [bigcode/starcoder2-3b](https://huggingface.co/bigcode/starcoder2-3b) on the None dataset.
130
+ It achieves the following results on the evaluation set:
131
+ - Loss: 0.9061
132
+
133
+ ## Model description
134
+
135
+ More information needed
136
+
137
+ ## Intended uses & limitations
138
+
139
+ More information needed
140
+
141
+ ## Training and evaluation data
142
+
143
+ More information needed
144
+
145
+ ## Training procedure
146
+
147
+ ### Training hyperparameters
148
+
149
+ The following hyperparameters were used during training:
150
+ - learning_rate: 0.0001
151
+ - train_batch_size: 8
152
+ - eval_batch_size: 4
153
+ - seed: 42
154
+ - gradient_accumulation_steps: 4
155
+ - total_train_batch_size: 32
156
+ - 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
157
+ - lr_scheduler_type: cosine
158
+ - lr_scheduler_warmup_steps: 10
159
+ - training_steps: 149
160
+
161
+ ### Training results
162
+
163
+ | Training Loss | Epoch | Step | Validation Loss |
164
+ |:-------------:|:------:|:----:|:---------------:|
165
+ | 2.0237 | 1.0101 | 50 | 0.9812 |
166
+ | 2.0966 | 2.0202 | 100 | 0.9061 |
167
+
168
+
169
+ ### Framework versions
170
+
171
+ - PEFT 0.14.0
172
+ - Transformers 4.46.3
173
+ - Pytorch 2.5.1+cu124
174
+ - Datasets 3.1.0
175
+ - Tokenizers 0.20.3
adapter_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e6a7d5a19b8867008ddcd48a72d6f904e321233a195c799a88c3378fd312b7b
3
+ size 328731266
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:56a01c6a3c9c79036b68b4042fe85b84e092caaed4e0b3ecf7cf532d16e09864
3
  size 630719216
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7f0cfc4d628da4020481e327bec8430bfebf3db135190cda4a56f6684eab24fe
3
  size 630719216