Views
No views yet
0.4.11base_model: macadeliccc/magistrate-3.2-3b-base
2model_type: LlamaForCausalLM
3tokenizer_type: AutoTokenizer
4
5load_in_8bit: false
6load_in_4bit: false
7strict: false
8
9datasets:
10 - path: json
11 type: sharegpt
12 conversation: chatml
13 data_files: train/hermes-2.5.jsonl
14 # - path: json
15 # type: sharegpt
16 # conversation: chatml
17 # data_files: train/financial_instructions_cleaned_2.json
18 - path: json
19 type: sharegpt
20 conversation: chatml
21 data_files: train/glaive-function-calling-5k.json
22 - path: json
23 type: sharegpt
24 conversation: chatml
25 data_files: train/func-calling-singleturn.json
26 - path: json
27 type: sharegpt
28 conversation: chatml
29 data_files: train/func-calling.json
30 - path: json
31 type: sharegpt
32 conversation: chatml
33 data_files: train/json-mode-agentic.json
34 - path: json
35 type: sharegpt
36 conversation: chatml
37 data_files: train/json-mode-singleturn.json
38 - path: json
39 type: sharegpt
40 conversation: chatml
41 data_files: train/reasoning_sharegpt.json
42 - path: json
43 type: sharegpt
44 conversation: chatml
45 data_files: train/systemchat_2_0_small.json
46 - path: json
47 type: sharegpt
48 conversation: chatml
49 data_files: train/argument_dataset/303_creative_llc_v__elenis_sharegpt.json
50 - path: json
51 type: sharegpt
52 conversation: chatml
53 data_files: train/argument_dataset/abitron_austria_gmbh_v__hetronic_international__inc__sharegpt.json
54 - path: json
55 type: sharegpt
56 conversation: chatml
57 data_files: train/argument_dataset/acheson_hotels__llc_v__laufer_sharegpt.json
58 - path: json
59 type: sharegpt
60 conversation: chatml
61 data_files: train/argument_dataset/alexander_v__sc_conference_of_naacp_sharegpt.json
62 - path: json
63 type: sharegpt
64 conversation: chatml
65 data_files: train/argument_dataset/amgen_inc__v__sanofi_sharegpt.json
66 - path: json
67 type: sharegpt
68 conversation: chatml
69 data_files: train/argument_dataset/andy_warhol_found___inc__v__goldsmith_sharegpt.json
70 - path: json
71 type: sharegpt
72 conversation: chatml
73 data_files: train/argument_dataset/arizona_v__navajo_nation_sharegpt.json
74 - path: json
75 type: sharegpt
76 conversation: chatml
77 data_files: train/argument_dataset/becerra__sec__of_h_hs_v__san_carlos_apache_tribe_sharegpt.json
78 - path: json
79 type: sharegpt
80 conversation: chatml
81 data_files: train/argument_dataset/biden_v__nebraska_sharegpt.json
82 - path: json
83 type: sharegpt
84 conversation: chatml
85 data_files: train/argument_dataset/bissonnette_v__lepage_bakeries_park_st___llc_sharegpt.json
86 - path: json
87 type: sharegpt
88 conversation: chatml
89 data_files: train/argument_dataset/bittner_v__united_states_sharegpt.json
90 - path: json
91 type: sharegpt
92 conversation: chatml
93 data_files: train/argument_dataset/brown_v__united_states_sharegpt.json
94 - path: json
95 type: sharegpt
96 conversation: chatml
97 data_files: train/argument_dataset/cantero_v__bank_of_america__n_a__sharegpt.json
98 - path: json
99 type: sharegpt
100 conversation: chatml
101 data_files: train/argument_dataset/cfpb_v__com__fin__services_assn__sharegpt.json
102 - path: json
103 type: sharegpt
104 conversation: chatml
105 data_files: train/argument_dataset/chiaverini_v__city_of_napoleon_sharegpt.json
106 - path: json
107 type: sharegpt
108 conversation: chatml
109 data_files: train/argument_dataset/ciminelli_v__united_state_sharegpt.json
110 - path: json
111 type: sharegpt
112 conversation: chatml
113 data_files: train/argument_dataset/city_of_grants_pass_v__johnson_sharegpt.json
114 - path: json
115 type: sharegpt
116 conversation: chatml
117 data_files: train/argument_dataset/coinbase__inc__v__bielski_sharegpt.json
118 - path: json
119 type: sharegpt
120 conversation: chatml
121 data_files: train/argument_dataset/coinbase__inc__v__suski_sharegpt.json
122 - path: json
123 type: sharegpt
124 conversation: chatml
125 data_files: train/argument_dataset/connelly_v__united_states_sharegpt.json
126 - path: json
127 type: sharegpt
128 conversation: chatml
129 data_files: train/argument_dataset/corner_post__inc__v__bd__of_governors__frs_sharegpt.json
130 - path: json
131 type: sharegpt
132 conversation: chatml
133 data_files: train/argument_dataset/counterman_v__colorado_sharegpt.json
134 - path: json
135 type: sharegpt
136 conversation: chatml
137 data_files: train/argument_dataset/cruz_v__arizona_sharegpt.json
138 - path: json
139 type: sharegpt
140 conversation: chatml
141 data_files: train/argument_dataset/culley_v__marshall_sharegpt.json
142 - path: json
143 type: sharegpt
144 conversation: chatml
145 data_files: train/argument_dataset/dept__of_agric__rural_dev__v__kirtz_sharegpt.json
146 - path: json
147 type: sharegpt
148 conversation: chatml
149 data_files: train/argument_dataset/dept__of_education_v__brown_sharegpt.json
150 - path: json
151 type: sharegpt
152 conversation: chatml
153 data_files: train/argument_dataset/dept__of_state_v__munoz_sharegpt.json
154 - path: json
155 type: sharegpt
156 conversation: chatml
157 data_files: train/argument_dataset/devillier_v__texas_sharegpt.json
158 - path: json
159 type: sharegpt
160 conversation: chatml
161 data_files: train/argument_dataset/diaz_v__united_states_sharegpt.json
162 - path: json
163 type: sharegpt
164 conversation: chatml
165 data_files: train/argument_dataset/dubin_v__united_states_sharegpt.json
166 - path: json
167 type: sharegpt
168 conversation: chatml
169 data_files: train/argument_dataset/dupree_v__younger_sharegpt.json
170 - path: json
171 type: sharegpt
172 conversation: chatml
173 data_files: train/argument_dataset/erlinger_v__united_states_sharegpt.json
174 - path: json
175 type: sharegpt
176 conversation: chatml
177 data_files: train/argument_dataset/fbi_v__fikre_sharegpt.json
178 - path: json
179 type: sharegpt
180 conversation: chatml
181 data_files: train/argument_dataset/fda_v__alliance_hippocratic_medicine_sharegpt.json
182 - path: json
183 type: sharegpt
184 conversation: chatml
185 data_files: train/argument_dataset/financial_oversight_board_v__cpi_sharegpt.json
186 - path: json
187 type: sharegpt
188 conversation: chatml
189 data_files: train/argument_dataset/fischer_v__united_states_sharegpt.json
190 - path: json
191 type: sharegpt
192 conversation: chatml
193 data_files: train/argument_dataset/garland__att_y_gen__v__cargill_sharegpt.json
194 - path: json
195 type: sharegpt
196 conversation: chatml
197 data_files: train/argument_dataset/glacier_northwest__inc__v__int_l_brotherhood_of_teamsters_sharegpt.json
198 - path: json
199 type: sharegpt
200 conversation: chatml
201 data_files: train/argument_dataset/gonzalez_v__google_llc_sharegpt.json
202 - path: json
203 type: sharegpt
204 conversation: chatml
205 data_files: train/argument_dataset/gonzalez_v__trevino_sharegpt.json
206 - path: json
207 type: sharegpt
208 conversation: chatml
209 data_files: train/argument_dataset/great_lakes_insurance_se_v__raiders_retreat_realty_co___llc_sharegpt.json
210 - path: json
211 type: sharegpt
212 conversation: chatml
213 data_files: train/argument_dataset/groff_v__dejoy_sharegpt.json
214 - path: json
215 type: sharegpt
216 conversation: chatml
217 data_files: train/argument_dataset/harrington_v__purdue_pharma_l_p__sharegpt.json
218 - path: json
219 type: sharegpt
220 conversation: chatml
221 data_files: train/argument_dataset/harrow_v__dept__of_defense_sharegpt.json
222 - path: json
223 type: sharegpt
224 conversation: chatml
225 data_files: train/argument_dataset/health_and_hospital_corp__v__talevski_sharegpt.json
226 - path: json
227 type: sharegpt
228 conversation: chatml
229 data_files: train/argument_dataset/helix_energy_solutions_v__hewitt_sharegpt.json
230 - path: json
231 type: sharegpt
232 conversation: chatml
233 data_files: train/argument_dataset/in_re_grand_jury_sharegpt.json
234 - path: json
235 type: sharegpt
236 conversation: chatml
237 data_files: train/argument_dataset/jack_daniel_s_properties__inc__v__vip_products_sharegpt.json
238 - path: json
239 type: sharegpt
240 conversation: chatml
241 data_files: train/argument_dataset/jones_v__hendrix_sharegpt.json
242 - path: json
243 type: sharegpt
244 conversation: chatml
245 data_files: train/argument_dataset/karcho_polselli_v__irs_sharegpt.json
246 - path: json
247 type: sharegpt
248 conversation: chatml
249 data_files: train/argument_dataset/lac_du_flambeau_band_v__coughlin_sharegpt.json
250 - path: json
251 type: sharegpt
252 conversation: chatml
253 data_files: train/argument_dataset/lindke_v__freed_sharegpt.json
254 - path: json
255 type: sharegpt
256 conversation: chatml
257 data_files: train/argument_dataset/loper_bright_enterprises__inc__v__raimondo__sec__of_comm__sharegpt.json
258 - path: json
259 type: sharegpt
260 conversation: chatml
261 data_files: train/argument_dataset/lora_v__united_states_sharegpt.json
262 - path: json
263 type: sharegpt
264 conversation: chatml
265 data_files: train/argument_dataset/macquarie_infrastructure_corp__v__moab_partners__l_p__sharegpt.json
266 - path: json
267 type: sharegpt
268 conversation: chatml
269 data_files: train/argument_dataset/mallory_v__norfolk_southern_railway_co__sharegpt.json
270 - path: json
271 type: sharegpt
272 conversation: chatml
273 data_files: train/argument_dataset/mcintosh_v__united_states_sharegpt.json
274 - path: json
275 type: sharegpt
276 conversation: chatml
277 data_files: train/argument_dataset/merrill_v__milligan_sharegpt.json
278 - path: json
279 type: sharegpt
280 conversation: chatml
281 data_files: train/argument_dataset/moore_v__harper_sharegpt.json
282 - path: json
283 type: sharegpt
284 conversation: chatml
285 data_files: train/argument_dataset/moore_v__united_states_sharegpt.json
286 - path: json
287 type: sharegpt
288 conversation: chatml
289 data_files: train/argument_dataset/moyle_v__united_states_sharegpt.json
290 - path: json
291 type: sharegpt
292 conversation: chatml
293 data_files: train/argument_dataset/muldrow_v__st__louis_sharegpt.json
294 - path: json
295 type: sharegpt
296 conversation: chatml
297 data_files: train/argument_dataset/murray_v__ubs_securities__llc_sharegpt.json
298 - path: json
299 type: sharegpt
300 conversation: chatml
301 data_files: train/argument_dataset/murthy__surgeon_gen__v__missouri_sharegpt.json
302 - path: json
303 type: sharegpt
304 conversation: chatml
305 data_files: train/argument_dataset/netchoice__llc_v__paxton_sharegpt.json
306 - path: json
307 type: sharegpt
308 conversation: chatml
309 data_files: train/argument_dataset/new_york_v__new_jersey_sharegpt.json
310 - path: json
311 type: sharegpt
312 conversation: chatml
313 data_files: train/argument_dataset/nra_v__vullo_sharegpt.json
314 - path: json
315 type: sharegpt
316 conversation: chatml
317 data_files: train/argument_dataset/o_connor_ratcliff_v__garnier_sharegpt.json
318 - path: json
319 type: sharegpt
320 conversation: chatml
321 data_files: train/argument_dataset/oh_adjutant_gen__s_dept__v__flra_sharegpt.json
322 - path: json
323 type: sharegpt
324 conversation: chatml
325 data_files: train/argument_dataset/ohio_v__epa_sharegpt.json
326 - path: json
327 type: sharegpt
328 conversation: chatml
329 data_files: train/argument_dataset/perez_v__sturgis_public_schools_sharegpt.json
330 - path: json
331 type: sharegpt
332 conversation: chatml
333 data_files: train/argument_dataset/pugin_v__garland_sharegpt.json
334 - path: json
335 type: sharegpt
336 conversation: chatml
337 data_files: train/argument_dataset/pulsifer_v__united_states_sharegpt.json
338 - path: json
339 type: sharegpt
340 conversation: chatml
341 data_files: train/argument_dataset/relentless__inc__v__dept__of_commerce_sharegpt.json
342 - path: json
343 type: sharegpt
344 conversation: chatml
345 data_files: train/argument_dataset/rudisill_v__mcdonough__sec__of_va_sharegpt.json
346 - path: json
347 type: sharegpt
348 conversation: chatml
349 data_files: train/argument_dataset/sackett_v__epa_sharegpt.json
350 - path: json
351 type: sharegpt
352 conversation: chatml
353 data_files: train/argument_dataset/samia_v__united_states_sharegpt.json
354 - path: json
355 type: sharegpt
356 conversation: chatml
357 data_files: train/argument_dataset/santos_zacaria_v__garland__att_y_gen__sharegpt.json
358 - path: json
359 type: sharegpt
360 conversation: chatml
361 data_files: train/argument_dataset/sec_v__cochran_sharegpt.json
362 - path: json
363 type: sharegpt
364 conversation: chatml
365 data_files: train/argument_dataset/sec_v__jarkesy_sharegpt.json
366 - path: json
367 type: sharegpt
368 conversation: chatml
369 data_files: train/argument_dataset/sheetz_v__county_of_el_dorado_sharegpt.json
370 - path: json
371 type: sharegpt
372 conversation: chatml
373 data_files: train/argument_dataset/slack_technologies__llc_v__pirani_sharegpt.json
374 - path: json
375 type: sharegpt
376 conversation: chatml
377 data_files: train/argument_dataset/smith_v__arizona_sharegpt.json
378 - path: json
379 type: sharegpt
380 conversation: chatml
381 data_files: train/argument_dataset/smith_v__spizzirri_sharegpt.json
382 - path: json
383 type: sharegpt
384 conversation: chatml
385 data_files: train/argument_dataset/smith_v__united_states_sharegpt.json
386 - path: json
387 type: sharegpt
388 conversation: chatml
389 data_files: train/argument_dataset/snyder_v__united_states_sharegpt.json
390 - path: json
391 type: sharegpt
392 conversation: chatml
393 data_files: train/argument_dataset/starbucks_corp__v__mckinney_sharegpt.json
394 - path: json
395 type: sharegpt
396 conversation: chatml
397 data_files: train/argument_dataset/students_for_fair_admissions_v__university_of_nc_sharegpt.json
398 - path: json
399 type: sharegpt
400 conversation: chatml
401 data_files: train/argument_dataset/texas_v__new_mexico_and_colorado_sharegpt.json
402 - path: json
403 type: sharegpt
404 conversation: chatml
405 data_files: train/argument_dataset/thornell_v__jones_sharegpt.json
406 - path: json
407 type: sharegpt
408 conversation: chatml
409 data_files: train/argument_dataset/truck_insurance_exchange_v__kaiser_gypsum_co__inc__sharegpt.json
410 - path: json
411 type: sharegpt
412 conversation: chatml
413 data_files: train/argument_dataset/trump_v__anderson_sharegpt.json
414 - path: json
415 type: sharegpt
416 conversation: chatml
417 data_files: train/argument_dataset/turkiye_halk_bankasi_a_s__v__united_states_sharegpt.json
418 - path: json
419 type: sharegpt
420 conversation: chatml
421 data_files: train/argument_dataset/twitter__inc__v__taamneh_sharegpt.json
422 - path: json
423 type: sharegpt
424 conversation: chatml
425 data_files: train/argument_dataset/tyler_v__hennepin_county_sharegpt.json
426 - path: json
427 type: sharegpt
428 conversation: chatml
429 data_files: train/argument_dataset/u_s___ex_rel__polansky_v__executive_health_sharegpt.json
430 - path: json
431 type: sharegpt
432 conversation: chatml
433 data_files: train/argument_dataset/u_s___ex_rel__schutte_v__supervalu_inc__sharegpt.json
434 - path: json
435 type: sharegpt
436 conversation: chatml
437 data_files: train/argument_dataset/united_states_trustee_v__john_q__hammons_fall_2006__llc_sharegpt.json
438 - path: json
439 type: sharegpt
440 conversation: chatml
441 data_files: train/argument_dataset/united_states_v__hansen_sharegpt.json
442 - path: json
443 type: sharegpt
444 conversation: chatml
445 data_files: train/argument_dataset/united_states_v__rahimi_sharegpt.json
446 - path: json
447 type: sharegpt
448 conversation: chatml
449 data_files: train/argument_dataset/united_states_v__texas_sharegpt.json
450 - path: json
451 type: sharegpt
452 conversation: chatml
453 data_files: train/argument_dataset/vidal__under_sec__of_comm__v__elster_sharegpt.json
454 - path: json
455 type: sharegpt
456 conversation: chatml
457 data_files: train/argument_dataset/warner_chappell_music__inc__v__nealy_sharegpt.json
458 - path: json
459 type: sharegpt
460 conversation: chatml
461 data_files: train/argument_dataset/wilkins_v__united_states_sharegpt.json
462 - path: json
463 type: sharegpt
464 conversation: chatml
465 data_files: train/argument_dataset/wilkinson_v__garland__att_y_gen__sharegpt.json
466 - path: json
467 type: sharegpt
468 conversation: chatml
469 data_files: train/argument_dataset/yegiazaryan_v__smagin_sharegpt.json
470
471chat_template: chatml
472
473unfrozen_parameters:
474- ^lm_head.weight$
475- ^model.embed_tokens.weight$
476# input_layernorm layers
477- model.layers.0.input_layernorm
478- model.layers.1.input_layernorm
479- model.layers.2.input_layernorm
480- model.layers.3.input_layernorm
481- model.layers.4.input_layernorm
482- model.layers.5.input_layernorm
483- model.layers.6.input_layernorm
484- model.layers.7.input_layernorm
485- model.layers.8.input_layernorm
486- model.layers.9.input_layernorm
487- model.layers.10.input_layernorm
488- model.layers.11.input_layernorm
489- model.layers.12.input_layernorm
490- model.layers.13.input_layernorm
491# mlp.down_proj layers
492- model.layers.0.mlp.down_proj
493- model.layers.1.mlp.down_proj
494- model.layers.17.mlp.down_proj
495- model.layers.19.mlp.down_proj
496- model.layers.18.mlp.down_proj
497- model.layers.5.mlp.down_proj
498- model.layers.20.mlp.down_proj
499- model.layers.2.mlp.down_proj
500- model.layers.4.mlp.down_proj
501- model.layers.6.mlp.down_proj
502- model.layers.3.mlp.down_proj
503- model.layers.16.mlp.down_proj
504- model.layers.15.mlp.down_proj
505- model.layers.13.mlp.down_proj
506# mlp.gate_proj layers
507- model.layers.0.mlp.gate_proj
508- model.layers.1.mlp.gate_proj
509- model.layers.2.mlp.gate_proj
510- model.layers.3.mlp.gate_proj
511- model.layers.22.mlp.gate_proj
512- model.layers.21.mlp.gate_proj
513- model.layers.20.mlp.gate_proj
514- model.layers.23.mlp.gate_proj
515- model.layers.19.mlp.gate_proj
516- model.layers.4.mlp.gate_proj
517- model.layers.18.mlp.gate_proj
518- model.layers.17.mlp.gate_proj
519- model.layers.5.mlp.gate_proj
520- model.layers.24.mlp.gate_proj
521# mlp.up_proj layers
522- model.layers.4.mlp.up_proj
523- model.layers.3.mlp.up_proj
524- model.layers.5.mlp.up_proj
525- model.layers.6.mlp.up_proj
526- model.layers.7.mlp.up_proj
527- model.layers.2.mlp.up_proj
528- model.layers.8.mlp.up_proj
529- model.layers.14.mlp.up_proj
530- model.layers.13.mlp.up_proj
531- model.layers.11.mlp.up_proj
532- model.layers.9.mlp.up_proj
533- model.layers.1.mlp.up_proj
534- model.layers.15.mlp.up_proj
535- model.layers.12.mlp.up_proj
536# post_attention_layernorm layers
537- model.layers.0.post_attention_layernorm
538- model.layers.1.post_attention_layernorm
539- model.layers.2.post_attention_layernorm
540- model.layers.3.post_attention_layernorm
541- model.layers.4.post_attention_layernorm
542- model.layers.5.post_attention_layernorm
543- model.layers.6.post_attention_layernorm
544- model.layers.7.post_attention_layernorm
545- model.layers.8.post_attention_layernorm
546- model.layers.9.post_attention_layernorm
547- model.layers.10.post_attention_layernorm
548- model.layers.11.post_attention_layernorm
549- model.layers.12.post_attention_layernorm
550- model.layers.13.post_attention_layernorm
551# self_attn.k_proj layers
552- model.layers.25.self_attn.k_proj
553- model.layers.22.self_attn.k_proj
554- model.layers.19.self_attn.k_proj
555- model.layers.20.self_attn.k_proj
556- model.layers.17.self_attn.k_proj
557- model.layers.24.self_attn.k_proj
558- model.layers.23.self_attn.k_proj
559- model.layers.18.self_attn.k_proj
560- model.layers.21.self_attn.k_proj
561- model.layers.27.self_attn.k_proj
562- model.layers.15.self_attn.k_proj
563- model.layers.10.self_attn.k_proj
564- model.layers.6.self_attn.k_proj
565- model.layers.5.self_attn.k_proj
566# self_attn.o_proj layers
567- model.layers.13.self_attn.o_proj
568- model.layers.7.self_attn.o_proj
569- model.layers.12.self_attn.o_proj
570- model.layers.10.self_attn.o_proj
571- model.layers.5.self_attn.o_proj
572- model.layers.21.self_attn.o_proj
573- model.layers.6.self_attn.o_proj
574- model.layers.19.self_attn.o_proj
575- model.layers.8.self_attn.o_proj
576- model.layers.20.self_attn.o_proj
577- model.layers.22.self_attn.o_proj
578- model.layers.9.self_attn.o_proj
579- model.layers.17.self_attn.o_proj
580- model.layers.11.self_attn.o_proj
581# self_attn.q_proj layers
582- model.layers.12.self_attn.q_proj
583- model.layers.13.self_attn.q_proj
584- model.layers.9.self_attn.q_proj
585- model.layers.8.self_attn.q_proj
586- model.layers.10.self_attn.q_proj
587- model.layers.14.self_attn.q_proj
588- model.layers.11.self_attn.q_proj
589- model.layers.15.self_attn.q_proj
590- model.layers.26.self_attn.q_proj
591- model.layers.6.self_attn.q_proj
592- model.layers.7.self_attn.q_proj
593- model.layers.16.self_attn.q_proj
594- model.layers.5.self_attn.q_proj
595- model.layers.25.self_attn.q_proj
596# model.norm layers
597# self_attn.v_proj layers
598- model.layers.23.self_attn.v_proj
599- model.layers.14.self_attn.v_proj
600- model.layers.15.self_attn.v_proj
601- model.layers.19.self_attn.v_proj
602- model.layers.3.self_attn.v_proj
603- model.layers.18.self_attn.v_proj
604- model.layers.25.self_attn.v_proj
605- model.layers.4.self_attn.v_proj
606- model.layers.17.self_attn.v_proj
607- model.layers.22.self_attn.v_proj
608- model.layers.20.self_attn.v_proj
609- model.layers.13.self_attn.v_proj
610- model.layers.6.self_attn.v_proj
611- model.layers.27.self_attn.v_proj
612
613val_set_size: 0.05
614output_dir: ./outputs/magistrate-3.2-3b
615
616sequence_len: 8192
617sample_packing: true
618eval_sample_packing: false
619pad_to_sequence_len: true
620
621adapter:
622
623wandb_project:
624wandb_entity:
625wandb_watch:
626wandb_name:
627wandb_log_model:
628
629gradient_accumulation_steps: 8
630micro_batch_size: 1
631num_epochs: 3
632optimizer: paged_adamw_32bit
633lr_scheduler: cosine
634learning_rate: 2e-4
635
636train_on_inputs: false
637group_by_length: false
638bf16: auto
639fp16:
640tf32: false
641
642gradient_checkpointing: true
643early_stopping_patience:
644resume_from_checkpoint:
645local_rank:
646logging_steps: 1
647xformers_attention:
648flash_attention: true
649s2_attention:
650
651warmup_steps: 1000
652evals_per_epoch: 2
653eval_table_size:
654eval_max_new_tokens: 128
655saves_per_epoch: 1
656debug:
657deepspeed: deepspeed_configs/zero3.json
658weight_decay: 0.0
659fsdp:
660fsdp_config:
661special_tokens:
662 eos_token: "<|im_end|>"
663 pad_token: "<|end_of_text|>"
664tokens:
665 - "<|im_start|>"
666 - "<|im_end|>"| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 1.3754 | 0.0005 | 1 | 1.7429 |
| 1.0 | 0.5002 | 1017 | 0.8864 |
| 0.9482 | 1.0005 | 2034 | 0.8395 |
| 0.6817 | 1.4987 | 3051 | 0.8063 |
| 0.697 | 1.9991 | 4068 | 0.7580 |
| 0.3769 | 2.4966 | 5085 | 0.8140 |
| 0.4278 | 2.9965 | 6102 | 0.8067 |