Views
No views yet
1scripts/train.py \
2 --verifier-name-or-path RedHatAI/GLM-5.2-NVFP4-FP8 \
3 --data-path ./output/dspark_glm52 \
4 --vllm-endpoint http://localhost:8000/v1 \
5 --save-path ./output/dspark_glm52/checkpoints \
6 --epochs 1 \
7 --lr 3e-4 \
8 --total-seq-len 8192 \
9 --speculator-type dspark \
10 --block-size 8 \
11 --max-anchors 1024 \
12 --num-layers 3 \
13 --target-layer-ids 2 20 39 58 75 \
14 --markov-rank 256 \
15 --markov-head-type vanilla \
16 --enable-confidence-head \
17 --confidence-head-with-markov \
18 --loss-fn '{"ce": 0.1, "tv": 0.9}' \
19 --confidence-head-alpha 1.0 \
20 --fsdp-shard \
21 --logger tensorboard \
22 --log-dir ./output/dspark_glm52/logs \
23 --on-missing generate \
24 --on-generate delete \
25 --checkpoint-freq 0.01 \
26 --request-timeout 60 \
27 --max-retries 21vllm serve RedHatAI/GLM-5.2-NVFP4-FP8 \
2 --max-model-len 20480 \
3 --spec-model RedHatAI/GLM-5.2-speculator.dspark \
4 --spec-method dspark \
5 --spec-tokens 7 \
6 -tp 8| Dataset | Pos 0 | Pos 1 | Pos 2 | Pos 3 | Pos 4 | Pos 5 | Pos 6 | Avg. Length |
|---|---|---|---|---|---|---|---|---|
| HumanEval | 77.8% | 59.1% | 44.6% | 33.9% | 26.1% | 20.3% | 15.8% | 3.78 |
| math_reasoning | 87.5% | 76.6% | 67.6% | 59.2% | 51.8% | 45.1% | 39.5% | 5.27 |
| qa | 70.7% | 48.5% | 33.7% | 23.7% | 17.1% | 11.9% | 8.4% | 3.14 |
| question | 73.7% | 52.6% | 38.1% | 28.5% | 21.6% | 16.7% | 13.2% | 3.44 |
| rag | 78.5% | 61.3% | 48.7% | 38.9% | 30.7% | 24.2% | 19.2% | 4.01 |
| summarization | 77.9% | 58.0% | 43.0% | 31.7% | 22.8% | 16.2% | 11.4% | 3.61 |
| tool_call | 71.9% | 51.1% | 37.1% | 27.6% | 21.4% | 16.8% | 13.2% | 3.39 |
| translation | 76.8% | 57.4% | 43.5% | 33.6% | 26.1% | 20.0% | 15.1% | 3.72 |
| writing | 74.0% | 52.2% | 38.0% | 28.1% | 21.3% | 16.7% | 13.1% | 3.44 |