hllj commited on
Commit
58d0581
·
1 Parent(s): 61d81d3

Model save

Browse files
README.md ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: hllj/mistral-vi-math
3
+ tags:
4
+ - generated_from_trainer
5
+ model-index:
6
+ - name: sft-mistral-v2-rank-64-alpha-128
7
+ results: []
8
+ ---
9
+
10
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
11
+ should probably proofread and complete it, then remove this comment. -->
12
+
13
+ # sft-mistral-v2-rank-64-alpha-128
14
+
15
+ This model is a fine-tuned version of [hllj/mistral-vi-math](https://huggingface.co/hllj/mistral-vi-math) on an unknown dataset.
16
+ It achieves the following results on the evaluation set:
17
+ - Loss: 0.4965
18
+
19
+ ## Model description
20
+
21
+ More information needed
22
+
23
+ ## Intended uses & limitations
24
+
25
+ More information needed
26
+
27
+ ## Training and evaluation data
28
+
29
+ More information needed
30
+
31
+ ## Training procedure
32
+
33
+ ### Training hyperparameters
34
+
35
+ The following hyperparameters were used during training:
36
+ - learning_rate: 5e-05
37
+ - train_batch_size: 8
38
+ - eval_batch_size: 8
39
+ - seed: 42
40
+ - distributed_type: multi-GPU
41
+ - gradient_accumulation_steps: 2
42
+ - total_train_batch_size: 16
43
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
+ - lr_scheduler_type: cosine
45
+ - lr_scheduler_warmup_ratio: 0.05
46
+ - num_epochs: 2
47
+ - mixed_precision_training: Native AMP
48
+
49
+ ### Training results
50
+
51
+
52
+
53
+ ### Framework versions
54
+
55
+ - Transformers 4.35.2
56
+ - Pytorch 2.1.0
57
+ - Datasets 2.15.0
58
+ - Tokenizers 0.15.0
adapter_config.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "hllj/mistral-vi-math",
5
+ "bias": "none",
6
+ "fan_in_fan_out": false,
7
+ "inference_mode": true,
8
+ "init_lora_weights": true,
9
+ "layers_pattern": null,
10
+ "layers_to_transform": null,
11
+ "lora_alpha": 128,
12
+ "lora_dropout": 0.05,
13
+ "modules_to_save": null,
14
+ "peft_type": "LORA",
15
+ "r": 64,
16
+ "rank_pattern": {},
17
+ "revision": null,
18
+ "target_modules": [
19
+ "k_proj",
20
+ "q_proj",
21
+ "o_proj",
22
+ "v_proj"
23
+ ],
24
+ "task_type": "CAUSAL_LM"
25
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:585f67a6b98fe9814d09e92c9151e25abfcc00fc86b0481098669a6db9d2098b
3
+ size 218138576
all_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.45,
3
+ "eval_loss": 0.4965249300003052,
4
+ "eval_runtime": 80.8754,
5
+ "eval_samples": 852,
6
+ "eval_samples_per_second": 10.535,
7
+ "eval_steps_per_second": 1.323,
8
+ "train_loss": 0.33035796076752416,
9
+ "train_runtime": 4854.2663,
10
+ "train_samples": 7665,
11
+ "train_samples_per_second": 3.158,
12
+ "train_steps_per_second": 0.197
13
+ }
config_argument.yaml ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ cache_dir: ./cache
2
+ ddp_find_unused_parameters: false
3
+ ddp_timeout: 30000
4
+ device_map: auto
5
+ do_eval: true
6
+ do_train: true
7
+ eval_steps: 500
8
+ evaluation_strategy: steps
9
+ fp16: true
10
+ gradient_accumulation_steps: 2
11
+ gradient_checkpointing: true
12
+ gradient_checkpointing_kwargs:
13
+ use_reentrant: false
14
+ hub_model_id: hllj/sft-mistral-v2-rank-64-alpha-128
15
+ hub_strategy: every_save
16
+ learning_rate: 5.0e-05
17
+ log_level: info
18
+ logging_first_step: true
19
+ logging_steps: 10
20
+ logging_strategy: steps
21
+ lora_alpha: 128
22
+ lora_dropout: 0.05
23
+ lora_r: 64
24
+ lora_target_modules:
25
+ - q_proj
26
+ - k_proj
27
+ - v_proj
28
+ - o_proj
29
+ lr_scheduler_type: cosine
30
+ max_seq_length: 1024
31
+ model_name_or_path: hllj/mistral-vi-math
32
+ model_type: auto
33
+ num_train_epochs: 2
34
+ output_dir: outputs-sft-mistral-v2-rank-64-alpha-128
35
+ overwrite_output_dir: true
36
+ per_device_eval_batch_size: 8
37
+ per_device_train_batch_size: 8
38
+ preprocessing_num_workers: 4
39
+ push_to_hub: true
40
+ report_to: wandb
41
+ run_name: sft-mistral-v2-rank-64-alpha-128
42
+ save_steps: 500
43
+ save_strategy: steps
44
+ save_total_limit: 13
45
+ seed: 42
46
+ torch_dtype: float16
47
+ train_file_dir: datasets/finetune
48
+ use_peft: true
49
+ warmup_ratio: 0.05
50
+ weight_decay: 0.05
eval_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.45,
3
+ "eval_loss": 0.4965249300003052,
4
+ "eval_runtime": 80.8754,
5
+ "eval_samples": 852,
6
+ "eval_samples_per_second": 10.535,
7
+ "eval_steps_per_second": 1.323
8
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "</s>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": "<unk>",
17
+ "unk_token": {
18
+ "content": "<unk>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ }
24
+ }
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dadfd56d766715c61d2ef780a525ab43b8e6da4de6865bda3d95fdef5e134055
3
+ size 493443
tokenizer_config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": true,
3
+ "add_eos_token": false,
4
+ "added_tokens_decoder": {
5
+ "0": {
6
+ "content": "<unk>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false,
11
+ "special": true
12
+ },
13
+ "1": {
14
+ "content": "<s>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false,
19
+ "special": true
20
+ },
21
+ "2": {
22
+ "content": "</s>",
23
+ "lstrip": false,
24
+ "normalized": false,
25
+ "rstrip": false,
26
+ "single_word": false,
27
+ "special": true
28
+ }
29
+ },
30
+ "additional_special_tokens": [],
31
+ "bos_token": "<s>",
32
+ "clean_up_tokenization_spaces": false,
33
+ "eos_token": "</s>",
34
+ "legacy": true,
35
+ "model_max_length": 1000000000000000019884624838656,
36
+ "pad_token": "<unk>",
37
+ "padding_side": "right",
38
+ "sp_model_kwargs": {},
39
+ "spaces_between_special_tokens": false,
40
+ "tokenizer_class": "LlamaTokenizer",
41
+ "unk_token": "<unk>",
42
+ "use_default_system_prompt": true
43
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.45,
3
+ "train_loss": 0.33035796076752416,
4
+ "train_runtime": 4854.2663,
5
+ "train_samples": 7665,
6
+ "train_samples_per_second": 3.158,
7
+ "train_steps_per_second": 0.197
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,292 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 1.448383733055266,
5
+ "eval_steps": 500,
6
+ "global_step": 430,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.0,
13
+ "learning_rate": 1.0416666666666667e-06,
14
+ "loss": 0.7671,
15
+ "step": 1
16
+ },
17
+ {
18
+ "epoch": 0.02,
19
+ "learning_rate": 1.0416666666666668e-05,
20
+ "loss": 0.6676,
21
+ "step": 10
22
+ },
23
+ {
24
+ "epoch": 0.04,
25
+ "learning_rate": 2.0833333333333336e-05,
26
+ "loss": 0.5586,
27
+ "step": 20
28
+ },
29
+ {
30
+ "epoch": 0.06,
31
+ "learning_rate": 3.125e-05,
32
+ "loss": 0.4224,
33
+ "step": 30
34
+ },
35
+ {
36
+ "epoch": 0.08,
37
+ "learning_rate": 4.166666666666667e-05,
38
+ "loss": 0.3747,
39
+ "step": 40
40
+ },
41
+ {
42
+ "epoch": 0.1,
43
+ "learning_rate": 4.999940408373438e-05,
44
+ "loss": 0.3632,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 0.13,
49
+ "learning_rate": 4.997854999726187e-05,
50
+ "loss": 0.3615,
51
+ "step": 60
52
+ },
53
+ {
54
+ "epoch": 0.15,
55
+ "learning_rate": 4.99279285007165e-05,
56
+ "loss": 0.346,
57
+ "step": 70
58
+ },
59
+ {
60
+ "epoch": 0.17,
61
+ "learning_rate": 4.984759992069235e-05,
62
+ "loss": 0.3425,
63
+ "step": 80
64
+ },
65
+ {
66
+ "epoch": 0.19,
67
+ "learning_rate": 4.973765998627628e-05,
68
+ "loss": 0.3341,
69
+ "step": 90
70
+ },
71
+ {
72
+ "epoch": 0.21,
73
+ "learning_rate": 4.959823971496574e-05,
74
+ "loss": 0.3289,
75
+ "step": 100
76
+ },
77
+ {
78
+ "epoch": 0.23,
79
+ "learning_rate": 4.942950525653282e-05,
80
+ "loss": 0.3351,
81
+ "step": 110
82
+ },
83
+ {
84
+ "epoch": 0.25,
85
+ "learning_rate": 4.923165769502033e-05,
86
+ "loss": 0.3279,
87
+ "step": 120
88
+ },
89
+ {
90
+ "epoch": 0.27,
91
+ "learning_rate": 4.900493280910623e-05,
92
+ "loss": 0.3255,
93
+ "step": 130
94
+ },
95
+ {
96
+ "epoch": 0.29,
97
+ "learning_rate": 4.874960079112168e-05,
98
+ "loss": 0.3365,
99
+ "step": 140
100
+ },
101
+ {
102
+ "epoch": 0.31,
103
+ "learning_rate": 4.846596592505776e-05,
104
+ "loss": 0.3486,
105
+ "step": 150
106
+ },
107
+ {
108
+ "epoch": 0.33,
109
+ "learning_rate": 4.815436622394441e-05,
110
+ "loss": 0.3294,
111
+ "step": 160
112
+ },
113
+ {
114
+ "epoch": 0.35,
115
+ "learning_rate": 4.781517302703405e-05,
116
+ "loss": 0.3316,
117
+ "step": 170
118
+ },
119
+ {
120
+ "epoch": 0.38,
121
+ "learning_rate": 4.744879055726944e-05,
122
+ "loss": 0.3176,
123
+ "step": 180
124
+ },
125
+ {
126
+ "epoch": 0.4,
127
+ "learning_rate": 4.7055655439563656e-05,
128
+ "loss": 0.327,
129
+ "step": 190
130
+ },
131
+ {
132
+ "epoch": 0.42,
133
+ "learning_rate": 4.6636236180465876e-05,
134
+ "loss": 0.3203,
135
+ "step": 200
136
+ },
137
+ {
138
+ "epoch": 0.44,
139
+ "learning_rate": 4.619103260983328e-05,
140
+ "loss": 0.3225,
141
+ "step": 210
142
+ },
143
+ {
144
+ "epoch": 1.01,
145
+ "learning_rate": 4.572057528517441e-05,
146
+ "loss": 0.3125,
147
+ "step": 220
148
+ },
149
+ {
150
+ "epoch": 1.03,
151
+ "learning_rate": 4.522542485937369e-05,
152
+ "loss": 0.3141,
153
+ "step": 230
154
+ },
155
+ {
156
+ "epoch": 1.05,
157
+ "learning_rate": 4.4706171412550815e-05,
158
+ "loss": 0.2982,
159
+ "step": 240
160
+ },
161
+ {
162
+ "epoch": 1.07,
163
+ "learning_rate": 4.416343374885116e-05,
164
+ "loss": 0.2867,
165
+ "step": 250
166
+ },
167
+ {
168
+ "epoch": 1.09,
169
+ "learning_rate": 4.359785865900513e-05,
170
+ "loss": 0.2975,
171
+ "step": 260
172
+ },
173
+ {
174
+ "epoch": 1.11,
175
+ "learning_rate": 4.301012014953544e-05,
176
+ "loss": 0.3021,
177
+ "step": 270
178
+ },
179
+ {
180
+ "epoch": 1.14,
181
+ "learning_rate": 4.240091863953084e-05,
182
+ "loss": 0.3082,
183
+ "step": 280
184
+ },
185
+ {
186
+ "epoch": 1.16,
187
+ "learning_rate": 4.177098012594341e-05,
188
+ "loss": 0.306,
189
+ "step": 290
190
+ },
191
+ {
192
+ "epoch": 1.18,
193
+ "learning_rate": 4.1121055318404264e-05,
194
+ "loss": 0.2859,
195
+ "step": 300
196
+ },
197
+ {
198
+ "epoch": 1.2,
199
+ "learning_rate": 4.045191874458876e-05,
200
+ "loss": 0.2982,
201
+ "step": 310
202
+ },
203
+ {
204
+ "epoch": 1.22,
205
+ "learning_rate": 3.976436782719713e-05,
206
+ "loss": 0.2956,
207
+ "step": 320
208
+ },
209
+ {
210
+ "epoch": 1.24,
211
+ "learning_rate": 3.905922193365089e-05,
212
+ "loss": 0.2958,
213
+ "step": 330
214
+ },
215
+ {
216
+ "epoch": 1.26,
217
+ "learning_rate": 3.8337321399637125e-05,
218
+ "loss": 0.2952,
219
+ "step": 340
220
+ },
221
+ {
222
+ "epoch": 1.28,
223
+ "learning_rate": 3.75995265276646e-05,
224
+ "loss": 0.2918,
225
+ "step": 350
226
+ },
227
+ {
228
+ "epoch": 1.3,
229
+ "learning_rate": 3.6846716561824965e-05,
230
+ "loss": 0.2909,
231
+ "step": 360
232
+ },
233
+ {
234
+ "epoch": 1.32,
235
+ "learning_rate": 3.6079788639981036e-05,
236
+ "loss": 0.2927,
237
+ "step": 370
238
+ },
239
+ {
240
+ "epoch": 1.34,
241
+ "learning_rate": 3.529965672463056e-05,
242
+ "loss": 0.2889,
243
+ "step": 380
244
+ },
245
+ {
246
+ "epoch": 1.36,
247
+ "learning_rate": 3.450725051371987e-05,
248
+ "loss": 0.2752,
249
+ "step": 390
250
+ },
251
+ {
252
+ "epoch": 1.39,
253
+ "learning_rate": 3.370351433270529e-05,
254
+ "loss": 0.2883,
255
+ "step": 400
256
+ },
257
+ {
258
+ "epoch": 1.41,
259
+ "learning_rate": 3.288940600918266e-05,
260
+ "loss": 0.2761,
261
+ "step": 410
262
+ },
263
+ {
264
+ "epoch": 1.43,
265
+ "learning_rate": 3.2065895731425995e-05,
266
+ "loss": 0.2824,
267
+ "step": 420
268
+ },
269
+ {
270
+ "epoch": 1.45,
271
+ "learning_rate": 3.1233964892195884e-05,
272
+ "loss": 0.2919,
273
+ "step": 430
274
+ },
275
+ {
276
+ "epoch": 1.45,
277
+ "step": 430,
278
+ "total_flos": 3.022612632268964e+17,
279
+ "train_loss": 0.33035796076752416,
280
+ "train_runtime": 4854.2663,
281
+ "train_samples_per_second": 3.158,
282
+ "train_steps_per_second": 0.197
283
+ }
284
+ ],
285
+ "logging_steps": 10,
286
+ "max_steps": 958,
287
+ "num_train_epochs": 2,
288
+ "save_steps": 500,
289
+ "total_flos": 3.022612632268964e+17,
290
+ "trial_name": null,
291
+ "trial_params": null
292
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d1d023ff15474507d51f7190d602417774ec3256dd9c52464567a168763c921
3
+ size 4664