ViuAI commited on
Commit
46be641
Β·
verified Β·
1 Parent(s): f4401f4

docs: ROADMAP_SCRATCH Step 0-10 with status

Browse files
Files changed (2) hide show
  1. docs/PROGRESS.md +1 -0
  2. docs/ROADMAP_SCRATCH.md +50 -0
docs/PROGRESS.md CHANGED
@@ -29,6 +29,7 @@ Har kaam ke baad yaha entry karo taaki yaad rahe kab kya kiya tha.
29
  - [x] 2026-09-17: Kaggle smoke FAIL (`data/raw` HF me excluded) -> `train.py` smoke fallback (built-in dummy lines), Kaggle re-test PASS (24 blocks, loss 5.38->5.37), `docs/TEST_REPORT.md` banaya, HF push OK
30
  - [x] 2026-09-17: full code-phase suite 5/5 PASS (params 241.7M/166.2M, sliding unit, toy router balanced max 0.203 + loss 6.18->5.95, tok fertility 1.38 unk 0, train smoke), TEST_REPORT update, HF push
31
  - [x] 2026-09-17: POLICY LOCK β€” `docs/TESTING_POLICY.md` (saare tests Kaggle pe, local pe koi test nahi), TEST_REPORT header update, HF push
 
32
  - [ ] TODO: toy router-balance test (Kaggle), FINAL tokenizer (500k lines wait), dense baseline
33
 
34
  ## Kaise note likhna hai
 
29
  - [x] 2026-09-17: Kaggle smoke FAIL (`data/raw` HF me excluded) -> `train.py` smoke fallback (built-in dummy lines), Kaggle re-test PASS (24 blocks, loss 5.38->5.37), `docs/TEST_REPORT.md` banaya, HF push OK
30
  - [x] 2026-09-17: full code-phase suite 5/5 PASS (params 241.7M/166.2M, sliding unit, toy router balanced max 0.203 + loss 6.18->5.95, tok fertility 1.38 unk 0, train smoke), TEST_REPORT update, HF push
31
  - [x] 2026-09-17: POLICY LOCK β€” `docs/TESTING_POLICY.md` (saare tests Kaggle pe, local pe koi test nahi), TEST_REPORT header update, HF push
32
+ - [x] 2026-09-17: `docs/ROADMAP_SCRATCH.md` banaya (Step 0-10, status: 0-2 done, agla Step 3 toy-Kaggle), HF push
33
  - [ ] TODO: toy router-balance test (Kaggle), FINAL tokenizer (500k lines wait), dense baseline
34
 
35
  ## Kaise note likhna hai
docs/ROADMAP_SCRATCH.md ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # ROADMAP β€” Model Scratch se Deployment tak (ViuMini-MoE-242M status ke saath)
2
+ > Har step ka status: [x] done / [ ] bacha. Evidence file bracket me.
3
+ > Update rule: step complete hote hi tick + PROGRESS + TEST_REPORT entry + HF push.
4
+
5
+ ## Step 0 β€” Goal + design lock [x] DONE
6
+ - Model type, size, language mix fix (Hinglish 40:30:30, 28L MoE 241.7M/166.2M LOCK)
7
+ - Evidence: `docs/VIU1_MOE_PLAN.md`, `model/configs/viu1_moe_config.yaml`
8
+
9
+ ## Step 1 β€” Repo + code skeleton [x] DONE
10
+ - Folder structure, arch code (`viu1_moe.py`), config, requirements, HF card
11
+ - Evidence: `README.md`, `model/scripts/viu1_moe.py`, HF https://huggingface.co/ViuAI/ViuMini-MoE-242M
12
+
13
+ ## Step 2 β€” Audit + bugfix [x] DONE
14
+ - Sliding-mask, MoE dispatch, aux-loss, batch, ckpt, paths β€” 10 critical fix verify
15
+ - Evidence: `docs/TEST_REPORT.md` (5/5 suite PASS)
16
+
17
+ ## Step 3 β€” Toy test (Kaggle) [ ] BACHA
18
+ - 2L tiny MoE pe router-balance (max expert <60%), balance-loss lock, switch-gate on/off
19
+ - Success: usage report + loss-down curve TEST_REPORT me
20
+ - Fail rule: yaha fail = design badlo, aage mat badho
21
+
22
+ ## Step 4 β€” Real data collect [ ] BACHA (sabse important)
23
+ - 4-5B tokens 40:30:30 (2B Hinglish + 1.5B Hindi + 1.5B English) + 200k CoT Hinglish
24
+ - Clean + dedup + eval split, 0 leakage
25
+ - Evidence: `data/processed/` + dedup report + `ViuAI/viu-mini-pretrain-40-30-30` dataset repo
26
+
27
+ ## Step 5 β€” FINAL tokenizer [ ] BACHA
28
+ - 500k lines pe 48k BPE train, fertility Hinglish ≀1.5 / Hindi ≀1.8 / unk=0
29
+ - Evidence: `tokenizer/outputs/tokenizer.json` (FINAL tag) + coverage log
30
+
31
+ ## Step 6 β€” Dense baseline (GPU) [ ] BACHA
32
+ - MoE off karke ~135M dense train, loss smooth down + Hinglish fluent sample
33
+ - Evidence: ckpt + val PPL + sample log (TEST_REPORT)
34
+
35
+ ## Step 7 β€” Full MoE pretrain (GPU 2-4 hafte) [ ] BACHA
36
+ - 9 MoE layers on, BF16, ckpt har 5k, early-stop (expert >60% for 2k steps β†’ roko)
37
+ - Success: dense se +5% better + router balanced
38
+
39
+ ## Step 8 β€” Eval + inference [ ] BACHA
40
+ - PPL (3 langs), CoT accuracy, human rating + `generate.py` (top-k/temp)
41
+ - Evidence: eval numbers + samples PROGRESS me
42
+
43
+ ## Step 9 β€” SFT / alignment (optional) [ ] BACHA
44
+ - Instruction + DPO data pe finetune (plan me baad me)
45
+
46
+ ## Step 10 β€” Release [ ] BACHA
47
+ - FINAL ckpt + model card + public HF + version tag
48
+
49
+ ## Abhi kaha hai?
50
+ Step 0-2 done. Agla: **Step 3 toy test (Kaggle)** β†’ Step 4 data (tumhara kaam) β†’ Step 5 tokenizer.