mrsandip commited on
Commit
cb58cd0
·
verified ·
1 Parent(s): 789bbf4

Add best 4GB scratch config

Browse files
Files changed (1) hide show
  1. configs/best-4gb-scratch.yaml +25 -0
configs/best-4gb-scratch.yaml ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Best data-only results on 4GB (no base model)
2
+ data:
3
+ train: data/processed/train.jsonl # now 2005 rows; grow to full 20k + filtered CoT
4
+ dedupe: true
5
+ fim_rate: 0.5 # fill-in-middle for code
6
+ instruct_mix: 0.3
7
+ tokenizer:
8
+ vocab_size: 32000
9
+ corpus: data/processed/train.jsonl # retrain on full data
10
+ training:
11
+ max_seq_len: 2048
12
+ batch_size: 2
13
+ grad_accum: 16 # eff batch 32
14
+ lr: 3e-4
15
+ scheduler: cosine
16
+ warmup_ratio: 0.05
17
+ epochs: 3
18
+ gradient_checkpointing: true
19
+ weight_decay: 0.1
20
+ eval_every: 200
21
+ ctx_curriculum: [2048, 4096, 8192]
22
+ gguf:
23
+ quant_best_quality: Q8_0 # best, bigger
24
+ quant_best_size: Q4_K_M # best fit for 4GB
25
+ keep_both: true