Chengheng commited on
Commit
3d67ea4
·
verified ·
1 Parent(s): 26190cc

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +50 -0
README.md ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: Qwen/Qwen3.5-2B
4
+ tags:
5
+ - gui-grounding
6
+ - sdft
7
+ - self-distillation
8
+ - qwen3.5
9
+ ---
10
+
11
+ # WebOS SDFT 2B
12
+
13
+ Full-parameter SDFT (Self-Distillation Fine-Tuning, paper-faithful Shenfeld et al. 2601.19897 Algorithm 1) of Qwen3.5-2B on the WebOS GUI-grounding dataset.
14
+
15
+ ## Layout
16
+
17
+ | Subfolder | Description | Click-acc on full test set |
18
+ |---|---|---|
19
+ | `checkpoint-200/` | Step 200 / 616 (32% of training) | 18.41% |
20
+ | `checkpoint-400/` | Step 400 / 616 (65%) | 18.75% |
21
+ | `checkpoint-600/` | Step 600 / 616 (97%) | **19.95%** |
22
+ | `final/` | Step 616 / 616 (100%) | ~20% |
23
+
24
+ For comparison: base Qwen3.5-2B on the same eval is 18.84%, and Qwen3.5-2B + privileged GT-demo teacher prompt is 26.46%.
25
+
26
+ ## Training config (16b_train_sdft_full_ft.py)
27
+ - bf16, full-param FT (vision frozen)
28
+ - bs=1, ga=32 (effective batch 32)
29
+ - lr=5e-6, warmup=10 steps, cosine
30
+ - AdamW, wd=0, max_grad_norm=1.0
31
+ - ema-α=0.01, kl-temperature=2.0, reverse KL
32
+ - 2 epochs × 9,864 train samples = 616 optimizer steps
33
+ - on-policy max_new_tokens=96, temperature=1.0
34
+
35
+ ## Loading
36
+
37
+ ```python
38
+ from transformers import AutoModelForImageTextToText, AutoProcessor
39
+ model = AutoModelForImageTextToText.from_pretrained(
40
+ "Chengheng/webos-sdft-2b", subfolder="checkpoint-600",
41
+ torch_dtype="bfloat16", device_map="cuda:0",
42
+ )
43
+ processor = AutoProcessor.from_pretrained(
44
+ "Chengheng/webos-sdft-2b", subfolder="checkpoint-600",
45
+ )
46
+ ```
47
+
48
+ ## Code
49
+
50
+ Training and eval scripts: https://github.com/ChenghengLi/WebOS