NeelContractor commited on
Commit
de36c2d
·
verified ·
1 Parent(s): 5c0931f

Model card: update GRPO eval with reasoning warm-up results

Browse files
Files changed (1) hide show
  1. README.md +24 -17
README.md CHANGED
@@ -57,7 +57,7 @@ not from the DPO adapter.
57
  | `models/sft_qwen06/final` + `checkpoint-100/150/198` | LoRA adapter | Instruction SFT | `cpt_qwen06_merged` |
58
  | `models/sft_qwen06_merged` | Full model (fp) | merged SFT | Qwen3-0.6B |
59
  | `models/dpo_qwen06/final` + `checkpoint-50/77` | LoRA adapter (+ `ref/`) | DPO preference tuning | `sft_qwen06_merged` |
60
- | `models/reasoning_warmup_qwen06/final` + `checkpoint-3` | LoRA adapter | Reasoning-format SFT (`think`/answer) | `sft_qwen06_merged` |
61
  | `models/reasoning_warmup_qwen06_merged` | Full model (fp) | merged warm-up | Qwen3-0.6B |
62
  | `models/grpo_qwen06/llm` | LoRA adapter | GRPO (latest) | `reasoning_warmup_qwen06_merged` |
63
  | `models/grpo_qwen06/llm_best_0` | LoRA adapter | GRPO (best eval reward) | `reasoning_warmup_qwen06_merged` |
@@ -154,6 +154,8 @@ print(tokenizer.decode(
154
  | Hardware | Kaggle, 2× Tesla T4 (15 GB), DDP |
155
  | Precision | fp16 training on T4; merged checkpoints saved as bf16 |
156
  | Max sequence length | 1024 (SFT / reasoning / GRPO) |
 
 
157
 
158
  ### Data
159
 
@@ -173,20 +175,22 @@ From `models/grpo_qwen06/logs/eval_summary_5.json` (64 held-out prompts):
173
 
174
  | Metric | Mean |
175
  |---|---|
176
- | `total_reward` | **0.440** |
177
- | `neuraltxt_reward` | **0.486** |
178
- | `length_penalty_reward` | −0.094 |
179
- | `num_tokens` | 101 |
180
- | `think_format_reward` | 0.000 |
181
- | `output_datatype_reward` | 0.000 |
182
- | `output_schema_reward` | 0.000 |
183
-
184
- **Known limitation:** this run's reasoning warm-up was only 3 optimizer steps, so
185
- the model never learned to emit `think...` blocks and the format-related rewards
186
- stayed at zero. The pipeline has since added a dedicated `reasoning` stage that
187
- trains the format on ~10k rows for 3 epochs; the artifacts in this repo predate
188
- that change. Do not treat these checkpoints as strong reasoning models — they
189
- are research/teaching artifacts from a short, small-model run.
 
 
190
 
191
  ---
192
 
@@ -208,7 +212,9 @@ python cloud_train.py --stages reasoning grpo --num-gpus 2
208
  ```
209
 
210
  See the [project README](https://github.com/NeelContractor/finetuning_recipes)
211
- for stage-by-stage details, dataset prep, and evaluation scripts.
 
 
212
 
213
  ---
214
 
@@ -217,6 +223,7 @@ for stage-by-stage details, dataset prep, and evaluation scripts.
217
  - Research / educational checkpoints from a short, small-model run.
218
  - Not instruction-tuned for production, not safety-tuned, not evaluated for
219
  harmful content.
220
- - Reasoning format is unreliable in these artifacts (see metrics above).
 
221
  - Derived from Qwen3-0.6B (Apache-2.0); check the licenses of the training
222
  datasets and the reward model before redistributing or commercial use.
 
57
  | `models/sft_qwen06/final` + `checkpoint-100/150/198` | LoRA adapter | Instruction SFT | `cpt_qwen06_merged` |
58
  | `models/sft_qwen06_merged` | Full model (fp) | merged SFT | Qwen3-0.6B |
59
  | `models/dpo_qwen06/final` + `checkpoint-50/77` | LoRA adapter (+ `ref/`) | DPO preference tuning | `sft_qwen06_merged` |
60
+ | `models/reasoning_warmup_qwen06/final` + `checkpoint-250/300/324` | LoRA adapter | Reasoning-format SFT (`think`/answer) | `sft_qwen06_merged` |
61
  | `models/reasoning_warmup_qwen06_merged` | Full model (fp) | merged warm-up | Qwen3-0.6B |
62
  | `models/grpo_qwen06/llm` | LoRA adapter | GRPO (latest) | `reasoning_warmup_qwen06_merged` |
63
  | `models/grpo_qwen06/llm_best_0` | LoRA adapter | GRPO (best eval reward) | `reasoning_warmup_qwen06_merged` |
 
154
  | Hardware | Kaggle, 2× Tesla T4 (15 GB), DDP |
155
  | Precision | fp16 training on T4; merged checkpoints saved as bf16 |
156
  | Max sequence length | 1024 (SFT / reasoning / GRPO) |
157
+ | Reasoning warm-up | 10,000 sampled → 6,911 after the 1024-token filter; 3 epochs, 324 steps, ~46 min on 2×T4 |
158
+ | GRPO | 1 epoch, 8 train events (384 optimizer steps), ~50 min on 2×T4 |
159
 
160
  ### Data
161
 
 
175
 
176
  | Metric | Mean |
177
  |---|---|
178
+ | `total_reward` | **2.355** |
179
+ | `neuraltxt_reward` | **0.519** |
180
+ | `length_penalty_reward` | −0.078 |
181
+ | `num_tokens` | 190.6 |
182
+ | `think_format_reward` | **0.953** |
183
+ | `output_datatype_reward` | 0.461 |
184
+ | `output_schema_reward` | 0.461 |
185
+
186
+ Datatype and schema match rate: **93.8%** each. Best inference score: **2.355**.
187
+
188
+ An earlier run used a reasoning warm-up of only 3 optimizer steps, so the model
189
+ never emitted `think...` blocks and every format reward stayed at zero. The
190
+ current checkpoints come from the dedicated `reasoning` stage (10,000 rows,
191
+ 3 epochs), which fixed it: 95.3% of eval responses contain a valid
192
+ `think...` block and 93.8% match the requested datatype/schema. Semantic
193
+ quality is still modest — this is a 0.6B research model, not a strong reasoner.
194
 
195
  ---
196
 
 
212
  ```
213
 
214
  See the [project README](https://github.com/NeelContractor/finetuning_recipes)
215
+ for stage-by-stage details, dataset prep, and evaluation scripts. The
216
+ Kaggle runbook for the reasoning branch is
217
+ [REASON_TRAINING.md](https://github.com/NeelContractor/finetuning_recipes/blob/main/REASON_TRAINING.md).
218
 
219
  ---
220
 
 
223
  - Research / educational checkpoints from a short, small-model run.
224
  - Not instruction-tuned for production, not safety-tuned, not evaluated for
225
  harmful content.
226
+ - Format compliance is now high (95.3% valid think blocks, 93.8% structural
227
+ match), but semantic reasoning quality is limited by the 0.6B model.
228
  - Derived from Qwen3-0.6B (Apache-2.0); check the licenses of the training
229
  datasets and the reward model before redistributing or commercial use.