WooYoungSeok commited on
Commit
f40682c
·
verified ·
1 Parent(s): 3625f98

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +56 -0
README.md ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # grpo-llama3.1-8b-0623-final
2
+
3
+ **Base model:** `meta-llama/Llama-3.1-8B-Instruct`
4
+ **Method:** GRPO (Group Relative Policy Optimization, TRL) with dual-RM reward signal.
5
+ **Final checkpoint** = `load_best_model_at_end=True`로 골라진 best step 5,185 가중치.
6
+
7
+ ## Reward models used during training (frozen)
8
+
9
+ | Slot | Model |
10
+ |---|---|
11
+ | RM 1 | [`WooYoungSeok/rm-qwen2.5-math-7b-0622`](https://huggingface.co/WooYoungSeok/rm-qwen2.5-math-7b-0622) — verifier-set acc 0.8354 |
12
+ | RM 2 | [`WooYoungSeok/rm-deepseek-r1-qwen3-8b-0622`](https://huggingface.co/WooYoungSeok/rm-deepseek-r1-qwen3-8b-0622) — verifier-set acc 0.8497 |
13
+
14
+ ## Training
15
+
16
+ | Hyperparameter | Value |
17
+ |---|---|
18
+ | Algorithm | GRPO (TRL) |
19
+ | Step at best ckpt (saved as final) | **5,185** |
20
+ | Epoch at best ckpt | 1.6989 / 2 |
21
+ | Total training step | 6,100 (≈ 2 epochs completed) |
22
+ | Group size (G) | 8 |
23
+ | Per-device batch | 8 |
24
+ | Gradient accumulation | 2 |
25
+ | Learning rate | 1e-6 |
26
+ | Warmup ratio | 0.1 |
27
+ | KL coefficient (β) | 0.04 |
28
+ | Clip ε | 0.2 |
29
+ | λ (verifiable penalty) | 0.75 |
30
+ | Format penalty | 0.5 |
31
+ | Answer-tag c | 0.5 |
32
+ | Max completion length | 1024 |
33
+ | Optimizer | DeepSpeed ZeRO-2 + CPU offload |
34
+ | Generation backend | vLLM (server mode, port 8001) |
35
+ | Seed | 42 |
36
+
37
+ ## Data (in-distribution 90/10 split of `train_subset.json`)
38
+
39
+ | Split | Source | Rows |
40
+ |---|---|---|
41
+ | Train | `train_subset_90pct.json` (90% stratified by error_type, 5-cluster taxonomy from MathEDU + Stepwise Verification + MathClean + EIC) | 6,105 |
42
+ | Eval (during training) | `train_subset_10pct.json` (in-distribution 10% held-out, stratified, seed=42) | 678 |
43
+
44
+ ## Performance
45
+
46
+ | Metric | Value | Step |
47
+ |---|---|---|
48
+ | **eval_reward (best, saved as final)** | **1.2560** | 5,185 |
49
+ | eval_loss (at best) | 0.0187 | 5,185 |
50
+ | eval_reward (last) | 1.2236 | 6,100 |
51
+ | eval_loss (last) | 0.0308 | 6,100 |
52
+ | train reward (last 100-step window) | ≈ 1.20 | — |
53
+
54
+ ## License
55
+
56
+ Apache-2.0. Base model is governed by the Llama 3.1 Community License — verify compliance for your use case.