rdubwiley commited on
Commit
ffdb43e
·
verified ·
1 Parent(s): 0b97131

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -46,20 +46,20 @@ It reads the tool's result, then emits the next action, until it calls `final_an
46
 
47
  ## How it was trained
48
 
49
- 1. **Teacher traces.** Two strong teacher models — **Kimi k2.6** and **DeepSeek 4 pro** (via [OpenCode Go](https://opencode.ai)) — drove the *real* agent loop over 7 public agenda packets and a set of local-government legal questions. Tools executed live, so every observation is grounded.
50
  2. **Judge filtering.** Each completed trace's final answer was scored for **faithfulness** against the text the agent actually retrieved (fast OpenCode-Go judge); only high-faithfulness traces were kept. One accepted agent step = one training example.
51
- 3. **SFT.** LoRA on the base's attention projections (q/k/v/o), 3 epochs over **204 examples** (held-out packet excluded — see Evaluation), full-sequence loss (the Gemma chat template lacks `{% generation %}` markers for assistant-only loss), bf16 + gradient checkpointing, then **merged** and converted to GGUF.
52
 
53
  | hyperparameter | value |
54
  |---|---|
55
- | LoRA rank / α / dropout | 16 / 32 / 0.05 |
56
- | target modules | attention `q_proj,k_proj,v_proj,o_proj` (auto-detected real `nn.Linear`) |
57
- | epochs | 3 |
58
  | learning rate | 1e-4 (cosine, 3% warmup) |
59
  | batch × grad-accum | 1 × 16 |
60
  | max sequence length | 4096 |
61
  | precision / GPU | bf16 / H100 |
62
- | final in-training token accuracy | ~0.74 |
63
 
64
  The full training/generation pipeline (trace capture, judge, LoRA, merge, GGUF) is reproducible from the dataset card.
65
 
 
46
 
47
  ## How it was trained
48
 
49
+ 1. **Teacher traces.** Two strong teacher models — **Kimi k2.6** and **DeepSeek 4 pro** (via [OpenCode Go](https://opencode.ai)) — drove the *real* agent loop over 11 public agenda packets and a set of local-government legal questions. Tools executed live, so every observation is grounded.
50
  2. **Judge filtering.** Each completed trace's final answer was scored for **faithfulness** against the text the agent actually retrieved (fast OpenCode-Go judge); only high-faithfulness traces were kept. One accepted agent step = one training example.
51
+ 3. **SFT.** LoRA on the base's attention projections (q/k/v/o), 4 epochs over **974 examples** (held-out packet excluded — see Evaluation), full-sequence loss (the Gemma chat template lacks `{% generation %}` markers for assistant-only loss), bf16 + gradient checkpointing, then **merged** and converted to GGUF.
52
 
53
  | hyperparameter | value |
54
  |---|---|
55
+ | LoRA rank / α / dropout | 32 / 64 / 0.05 |
56
+ | target modules | attention + MLP `q,k,v,o,gate,up,down_proj` (auto-detected real `nn.Linear`) |
57
+ | epochs | 4 |
58
  | learning rate | 1e-4 (cosine, 3% warmup) |
59
  | batch × grad-accum | 1 × 16 |
60
  | max sequence length | 4096 |
61
  | precision / GPU | bf16 / H100 |
62
+ | final in-training token accuracy | ~0.96 |
63
 
64
  The full training/generation pipeline (trace capture, judge, LoRA, merge, GGUF) is reproducible from the dataset card.
65