huluhuluu commited on
Commit
6da28d9
·
verified ·
1 Parent(s): 280e947

Document training parameters

Browse files
Files changed (1) hide show
  1. README.md +10 -8
README.md CHANGED
@@ -18,9 +18,9 @@ base_model:
18
  - Qwen/Qwen3-1.7B
19
  ---
20
 
21
- # Qwen3-1.7B EAGLE3 K2 SW256 MHA ShareGPT
22
 
23
- This repository contains one SpecForge EAGLE3 draft-model checkpoint for use with `Qwen/Qwen3-1.7B`.
24
  It is a draft model for speculative decoding, not a standalone target language model.
25
 
26
  ## Checkpoint
@@ -31,16 +31,16 @@ It is a draft model for speculative decoding, not a standalone target language m
31
  | Checkpoint | `epoch_9_step_171560` |
32
  | Epoch | `9` |
33
  | Global step | `171560` |
34
- | Files | `config.json`, `model.safetensors`, `training_state.pt` |
35
 
36
  ## Training Parameters
37
 
38
  | Parameter | Value |
39
  |---|---|
40
  | Base model | `Qwen/Qwen3-1.7B` |
41
- | Method | SpecForge EAGLE3 online training |
42
- | Dataset | ShareGPT-format SpecForge training set |
43
- | Learning rate | `1e-4` |
44
  | Batch size | `1` |
45
  | Target batch size | `1` |
46
  | Epochs configured | `10` |
@@ -48,6 +48,7 @@ It is a draft model for speculative decoding, not a standalone target language m
48
  | Warmup ratio | `0.015` |
49
  | Max grad norm | `0.5` |
50
  | TTT length | `5` |
 
51
  | Draft sliding window | `256` |
52
  | Save interval | `5000` |
53
  | Eval interval | `5000` |
@@ -74,10 +75,11 @@ It is a draft model for speculative decoding, not a standalone target language m
74
  | Max position embeddings | `40960` |
75
  | Sliding window | `256` |
76
  | Max window layers | `2` |
 
77
 
78
  ## Notes
79
 
80
- - This is the multi-head-attention variant: `num_key_value_heads=16`.
81
  - The checkpoint is intended to be loaded by SpecForge/EAGLE3-compatible code.
82
- - `training_state.pt` is included for provenance and training-state inspection.
83
  - No benchmark claim is made in this card.
 
18
  - Qwen/Qwen3-1.7B
19
  ---
20
 
21
+ # Qwen3 1.7B EAGLE3 K2 SW256 MHA ShareGPT
22
 
23
+ This repository contains a SpecForge EAGLE3 draft-model checkpoint for use with `Qwen/Qwen3-1.7B`.
24
  It is a draft model for speculative decoding, not a standalone target language model.
25
 
26
  ## Checkpoint
 
31
  | Checkpoint | `epoch_9_step_171560` |
32
  | Epoch | `9` |
33
  | Global step | `171560` |
34
+ | Files | `config.json`, `model.safetensors`, `training_state.pt` (when present) |
35
 
36
  ## Training Parameters
37
 
38
  | Parameter | Value |
39
  |---|---|
40
  | Base model | `Qwen/Qwen3-1.7B` |
41
+ | Method | `SpecForge EAGLE3 online training` |
42
+ | Training data | `sharegpt_train.jsonl` |
43
+ | Learning rate | `0.0001` |
44
  | Batch size | `1` |
45
  | Target batch size | `1` |
46
  | Epochs configured | `10` |
 
48
  | Warmup ratio | `0.015` |
49
  | Max grad norm | `0.5` |
50
  | TTT length | `5` |
51
+ | Draft layers | `2` |
52
  | Draft sliding window | `256` |
53
  | Save interval | `5000` |
54
  | Eval interval | `5000` |
 
75
  | Max position embeddings | `40960` |
76
  | Sliding window | `256` |
77
  | Max window layers | `2` |
78
+ | Future hidden | `Not recorded` |
79
 
80
  ## Notes
81
 
82
+ - This is the highest-step local checkpoint available when this repository was published.
83
  - The checkpoint is intended to be loaded by SpecForge/EAGLE3-compatible code.
84
+ - `training_state.pt` is included when available for provenance and training-state inspection.
85
  - No benchmark claim is made in this card.