dabrown commited on
Commit
cecc939
·
verified ·
1 Parent(s): e250c6b

End of training

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